1 files changed, 5 insertions, 5 deletions

diff --git a/mm/vmalloc.c b/mm/vmalloc.c
index 7350a124524b..93b2dca2aadb 100644
--- a/mm/vmalloc.c
+++ b/mm/vmalloc.c
@@ -2123,7 +2123,6 @@ 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 i;
@@ -2135,8 +2134,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 +2159,10 @@ 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);
                 }
         }