Donate to e Foundation | Murena handsets with /e/OS | Own a part of Murena! Learn more

Commit 16b34e53 authored by Lee Jones's avatar Lee Jones Committed by Greg Kroah-Hartman
Browse files

staging: ion: Prevent incorrect reference counting behavour 



Supply additional checks in order to prevent unexpected results.

Fixes: b892bf75 ("ion: Switch ion to use dma-buf")
Suggested-by: default avatarDan Carpenter <dan.carpenter@oracle.com>
Signed-off-by: default avatarLee Jones <lee.jones@linaro.org>
Signed-off-by: default avatarGreg Kroah-Hartman <gregkh@linuxfoundation.org>
parent 5fd8594b

drivers/staging/android/ion/ion.c

+6 −0
Original line number Diff line number Diff line
@@ -489,6 +489,9 @@ static void *ion_buffer_kmap_get(struct ion_buffer *buffer) 
	void *vaddr;



	if (buffer->kmap_cnt) {

		if (buffer->kmap_cnt == INT_MAX)

			return ERR_PTR(-EOVERFLOW);



		buffer->kmap_cnt++;

		return buffer->vaddr;

	}
@@ -509,6 +512,9 @@ static void *ion_handle_kmap_get(struct ion_handle *handle) 
	void *vaddr;



	if (handle->kmap_cnt) {

		if (handle->kmap_cnt == INT_MAX)

			return ERR_PTR(-EOVERFLOW);



		handle->kmap_cnt++;

		return buffer->vaddr;

	}