summaryrefslogtreecommitdiffstats
path: root/drivers/misc/habanalabs/common/mmu.c
diff options
context:
space:
mode:
authorOfir Bitton <obitton@habana.ai>2020-08-11 08:57:45 +0300
committerOded Gabbay <oded.gabbay@gmail.com>2020-08-22 12:47:58 +0300
commit0839152f8c1efc1cc2d515d1ff1e253ca9402ad3 (patch)
tree5f9289b6926b13486ef957581741f134a4753f72 /drivers/misc/habanalabs/common/mmu.c
parentbce382a8bb080ed5f2f3a06754526dc58b91cca2 (diff)
downloadlinux-0839152f8c1efc1cc2d515d1ff1e253ca9402ad3.tar.bz2
habanalabs: check correct vmalloc return code
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>
Diffstat (limited to 'drivers/misc/habanalabs/common/mmu.c')
-rw-r--r--drivers/misc/habanalabs/common/mmu.c2
1 files changed, 1 insertions, 1 deletions
diff --git a/drivers/misc/habanalabs/common/mmu.c b/drivers/misc/habanalabs/common/mmu.c
index edcc11d5eaf1..3fc0f497fab3 100644
--- a/drivers/misc/habanalabs/common/mmu.c
+++ b/drivers/misc/habanalabs/common/mmu.c
@@ -450,7 +450,7 @@ int hl_mmu_init(struct hl_device *hdev)
hdev->mmu_shadow_hop0 = kvmalloc_array(prop->max_asid,
prop->mmu_hop_table_size,
GFP_KERNEL | __GFP_ZERO);
- if (!hdev->mmu_shadow_hop0) {
+ if (ZERO_OR_NULL_PTR(hdev->mmu_shadow_hop0)) {
rc = -ENOMEM;
goto err_pool_add;
}