1 files changed, 23 insertions, 0 deletions

diff --git a/mm/mmap.c b/mm/mmap.c
index 3f758c7f4c81..1aed183636d7 100644
--- a/mm/mmap.c
+++ b/mm/mmap.c
@@ -30,6 +30,7 @@
 #include <linux/perf_event.h>
 #include <linux/audit.h>
 #include <linux/khugepaged.h>
+#include <linux/uprobes.h>
 
 #include <asm/uaccess.h>
 #include <asm/cacheflush.h>
@@ -616,6 +617,13 @@ again:			remove_next = 1 + (end > next->vm_end);
         if (mapping)
                 mutex_unlock(&mapping->i_mmap_mutex);
 
+        if (root) {
+                mmap_uprobe(vma);
+
+                if (adjust_next)
+                        mmap_uprobe(next);
+        }
+
         if (remove_next) {
                 if (file) {
                         fput(file);
@@ -637,6 +645,8 @@ again:			remove_next = 1 + (end > next->vm_end);
                         goto again;
                 }
         }
+        if (insert && file)
+                mmap_uprobe(insert);
 
         validate_mm(mm);
 
@@ -1329,6 +1339,11 @@ out:
                         mm->locked_vm += (len >> PAGE_SHIFT);
         } else if ((flags & MAP_POPULATE) && !(flags & MAP_NONBLOCK))
                 make_pages_present(addr, addr + len);
+
+        if (file && mmap_uprobe(vma))
+                /* matching probes but cannot insert */
+                goto unmap_and_free_vma;
+
         return addr;
 
 unmap_and_free_vma:
@@ -2285,6 +2300,10 @@ int insert_vm_struct(struct mm_struct * mm, struct vm_area_struct * vma)
         if ((vma->vm_flags & VM_ACCOUNT) &&
              security_vm_enough_memory_mm(mm, vma_pages(vma)))
                 return -ENOMEM;
+
+        if (vma->vm_file && mmap_uprobe(vma))
+                return -EINVAL;
+
         vma_link(mm, vma, prev, rb_link, rb_parent);
         return 0;
 }
@@ -2354,6 +2373,10 @@ struct vm_area_struct *copy_vma(struct vm_area_struct **vmap,
                         new_vma->vm_pgoff = pgoff;
                         if (new_vma->vm_file) {
                                 get_file(new_vma->vm_file);
+
+                                if (mmap_uprobe(new_vma))
+                                        goto out_free_mempol;
+
                                 if (vma->vm_flags & VM_EXECUTABLE)
                                         added_exe_file_vma(mm);
                         }