1 files changed, 4 insertions, 1 deletions

diff --git a/mm/util.c b/mm/util.c
index 3cb2164f4099..b8f538863b5a 100644
--- a/mm/util.c
+++ b/mm/util.c
@@ -11,6 +11,7 @@
 #include <linux/mman.h>
 #include <linux/hugetlb.h>
 #include <linux/vmalloc.h>
+#include <linux/userfaultfd_k.h>
 
 #include <asm/sections.h>
 #include <linux/uaccess.h>
@@ -297,14 +298,16 @@ unsigned long vm_mmap_pgoff(struct file *file, unsigned long addr,
         unsigned long ret;
         struct mm_struct *mm = current->mm;
         unsigned long populate;
+        LIST_HEAD(uf);
 
         ret = security_mmap_file(file, prot, flag);
         if (!ret) {
                 if (down_write_killable(&mm->mmap_sem))
                         return -EINTR;
                 ret = do_mmap_pgoff(file, addr, len, prot, flag, pgoff,
-                                    &populate);
+                                    &populate, &uf);
                 up_write(&mm->mmap_sem);
+                userfaultfd_unmap_complete(mm, &uf);
                 if (populate)
                         mm_populate(ret, populate);
         }