mm: use VM_BUG_ON_MM where possible

Dump the contents of the relevant struct_mm when we hit the bug condition.

Signed-off-by: Sasha Levin <sasha.levin@oracle.com>
Signed-off-by: Andrew Morton <akpm@linux-foundation.org>
Signed-off-by: Linus Torvalds <torvalds@linux-foundation.org>

4 files changed, 7 insertions, 6 deletions

diff --git a/mm/huge_memory.c b/mm/huge_memory.c
index c13148cc745f..74c78aa8bc2f 100644
--- a/mm/huge_memory.c
+++ b/mm/huge_memory.c
@@ -2048,7 +2048,7 @@ int __khugepaged_enter(struct mm_struct *mm)
                 return -ENOMEM;
 
         /* __khugepaged_exit() must not run from under us */
-        VM_BUG_ON(khugepaged_test_exit(mm));
+        VM_BUG_ON_MM(khugepaged_test_exit(mm), mm);
         if (unlikely(test_and_set_bit(MMF_VM_HUGEPAGE, &mm->flags))) {
                 free_mm_slot(mm_slot);
                 return 0;
diff --git a/mm/mlock.c b/mm/mlock.c
index d5d09d0786ec..03aa8512723b 100644
--- a/mm/mlock.c
+++ b/mm/mlock.c
@@ -235,7 +235,7 @@ long __mlock_vma_pages_range(struct vm_area_struct *vma,
         VM_BUG_ON(end   & ~PAGE_MASK);
         VM_BUG_ON_VMA(start < vma->vm_start, vma);
         VM_BUG_ON_VMA(end   > vma->vm_end, vma);
-        VM_BUG_ON(!rwsem_is_locked(&mm->mmap_sem));
+        VM_BUG_ON_MM(!rwsem_is_locked(&mm->mmap_sem), mm);
 
         gup_flags = FOLL_TOUCH | FOLL_MLOCK;
         /*
diff --git a/mm/mmap.c b/mm/mmap.c
index c9bc285df255..16d19b48e2ad 100644
--- a/mm/mmap.c
+++ b/mm/mmap.c
@@ -410,8 +410,9 @@ static void validate_mm_rb(struct rb_root *root, struct vm_area_struct *ignore)
         for (nd = rb_first(root); nd; nd = rb_next(nd)) {
                 struct vm_area_struct *vma;
                 vma = rb_entry(nd, struct vm_area_struct, vm_rb);
-                BUG_ON(vma != ignore &&
-                       vma->rb_subtree_gap != vma_compute_subtree_gap(vma));
+                VM_BUG_ON_VMA(vma != ignore &&
+                        vma->rb_subtree_gap != vma_compute_subtree_gap(vma),
+                        vma);
         }
 }
 
@@ -448,7 +449,7 @@ static void validate_mm(struct mm_struct *mm)
                         pr_emerg("map_count %d rb %d\n", mm->map_count, i);
                 bug = 1;
         }
-        BUG_ON(bug);
+        VM_BUG_ON_MM(bug, mm);
 }
 #else
 #define validate_mm_rb(root, ignore) do { } while (0)
diff --git a/mm/pagewalk.c b/mm/pagewalk.c
index 2beeabf502c5..ad83195521f2 100644
--- a/mm/pagewalk.c
+++ b/mm/pagewalk.c
@@ -177,7 +177,7 @@ int walk_page_range(unsigned long addr, unsigned long end,
         if (!walk->mm)
                 return -EINVAL;
 
-        VM_BUG_ON(!rwsem_is_locked(&walk->mm->mmap_sem));
+        VM_BUG_ON_MM(!rwsem_is_locked(&walk->mm->mmap_sem), walk->mm);
 
         pgd = pgd_offset(walk->mm, addr);
         do {