1 files changed, 8 insertions, 6 deletions

diff --git a/mm/vmalloc.c b/mm/vmalloc.c
index 7350a124524b..4c9e150e5ad3 100644
--- a/mm/vmalloc.c
+++ b/mm/vmalloc.c
@@ -2123,9 +2123,9 @@ static inline void set_area_direct_map(const struct vm_struct *area,
 /* Handle removing and resetting vm mappings related to the vm_struct. */
 static void vm_remove_mappings(struct vm_struct *area, int deallocate_pages)
 {
-        unsigned long addr = (unsigned long)area->addr;
         unsigned long start = ULONG_MAX, end = 0;
         int flush_reset = area->flags & VM_FLUSH_RESET_PERMS;
+        int flush_dmap = 0;
         int i;
 
         /*
@@ -2135,8 +2135,8 @@ static void vm_remove_mappings(struct vm_struct *area, int deallocate_pages)
          * execute permissions, without leaving a RW+X window.
          */
         if (flush_reset && !IS_ENABLED(CONFIG_ARCH_HAS_SET_DIRECT_MAP)) {
-                set_memory_nx(addr, area->nr_pages);
-                set_memory_rw(addr, area->nr_pages);
+                set_memory_nx((unsigned long)area->addr, area->nr_pages);
+                set_memory_rw((unsigned long)area->addr, area->nr_pages);
         }
 
         remove_vm_area(area->addr);
@@ -2160,9 +2160,11 @@ static void vm_remove_mappings(struct vm_struct *area, int deallocate_pages)
          * the vm_unmap_aliases() flush includes the direct map.
          */
         for (i = 0; i < area->nr_pages; i++) {
-                if (page_address(area->pages[i])) {
+                unsigned long addr = (unsigned long)page_address(area->pages[i]);
+                if (addr) {
                         start = min(addr, start);
-                        end = max(addr, end);
+                        end = max(addr + PAGE_SIZE, end);
+                        flush_dmap = 1;
                 }
         }
 
@@ -2172,7 +2174,7 @@ static void vm_remove_mappings(struct vm_struct *area, int deallocate_pages)
          * reset the direct map permissions to the default.
          */
         set_area_direct_map(area, set_direct_map_invalid_noflush);
-        _vm_unmap_aliases(start, end, 1);
+        _vm_unmap_aliases(start, end, flush_dmap);
         set_area_direct_map(area, set_direct_map_default_noflush);
 }