habanalabs: check correct vmalloc return code
authorOfir Bitton <obitton@habana.ai>
Tue, 11 Aug 2020 05:57:45 +0000 (08:57 +0300)
committerGreg Kroah-Hartman <gregkh@linuxfoundation.org>
Wed, 9 Sep 2020 17:14:09 +0000 (19:14 +0200)
commit2ad613d97f9675d03b2aba68588b5d950cbf1507
tree998f5e91edfd2082c33a284c12327cf4e3081b55
parent72038388869bc808f219734d111a4121d2ff99d5
habanalabs: check correct vmalloc return code

[ Upstream commit 0839152f8c1efc1cc2d515d1ff1e253ca9402ad3 ]

vmalloc can return different return code than NULL and a valid
pointer. We must validate it in order to dereference a non valid
pointer.

Signed-off-by: Ofir Bitton <obitton@habana.ai>
Signed-off-by: Oded Gabbay <oded.gabbay@gmail.com>
Signed-off-by: Sasha Levin <sashal@kernel.org>
drivers/misc/habanalabs/memory.c
drivers/misc/habanalabs/mmu.c