From: Suren Baghdasaryan <surenb@google.com>
Date: Tue, 8 Nov 2022 17:40:09 +0000 (+0000)
Subject: enable maple-tree RCU mode
X-Git-Url: https://www.infradead.org/git/?a=commitdiff_plain;h=e99359a3cc2cf28032e8dbf156c24532bc4cbadd;p=users%2Fjedix%2Flinux-maple.git

enable maple-tree RCU mode

Signed-off-by: Suren Baghdasaryan <surenb@google.com>
---

diff --git a/kernel/fork.c b/kernel/fork.c
index 08969f5aa38d..f8ce1fc21378 100644
--- a/kernel/fork.c
+++ b/kernel/fork.c
@@ -700,6 +700,8 @@ static __latent_entropy int dup_mmap(struct mm_struct *mm,
 	retval = arch_dup_mmap(oldmm, mm);
 loop_out:
 	mas_destroy(&mas);
+	if (!retval)
+		mt_set_in_rcu(mas.tree);
 out:
 	mmap_write_unlock(mm);
 	flush_tlb_mm(oldmm);
diff --git a/mm/mmap.c b/mm/mmap.c
index c3c5c1d6103d..374e0da0f711 100644
--- a/mm/mmap.c
+++ b/mm/mmap.c
@@ -3100,6 +3100,7 @@ void exit_mmap(struct mm_struct *mm)
 	 */
 	set_bit(MMF_OOM_SKIP, &mm->flags);
 	mmap_write_lock(mm);
+	mt_clear_in_rcu(&mm->mm_mt);
 	free_pgtables(&tlb, &mm->mm_mt, vma, FIRST_USER_ADDRESS,
 		      USER_PGTABLES_CEILING);
 	tlb_finish_mmu(&tlb);