net: wimax/i2400m: fix NULL-deref at probe

[karo-tx-linux.git] / drivers / iommu / intel-svm.c

diff --git a/drivers/iommu/intel-svm.c b/drivers/iommu/intel-svm.c
index cb72e0011310d1fbd04cb1560a861c8e5ec2fd55..23c427602c55ba10a546737edabb1bfb24497304 100644 (file)
--- a/drivers/iommu/intel-svm.c
+++ b/drivers/iommu/intel-svm.c
@@ -16,6 +16,7 @@
 #include <linux/intel-iommu.h>
 #include <linux/mmu_notifier.h>
 #include <linux/sched.h>
+#include <linux/sched/mm.h>
 #include <linux/slab.h>
 #include <linux/intel-svm.h>
 #include <linux/rculist.h>
@@ -579,7 +580,7 @@ static irqreturn_t prq_event_thread(int irq, void *d)
                if (!svm->mm)
                        goto bad_req;
                /* If the mm is already defunct, don't handle faults. */
-               if (!atomic_inc_not_zero(&svm->mm->mm_users))
+               if (!mmget_not_zero(svm->mm))
                        goto bad_req;
                down_read(&svm->mm->mmap_sem);
                vma = find_extend_vma(svm->mm, address);