From: Rick Edgecombe <redgecombe.l...@gmail.com>

Calling vm_unmap_alias() in vm_remove_mappings() could potentially be a
lot of work to do on a free operation. Simply flushing the TLB instead of
the whole vm_unmap_alias() operation makes the frees faster and pushes
the heavy work to happen on allocation where it would be more expected.
In addition to the extra work, vm_unmap_alias() takes some locks including
a long hold of vmap_purge_lock, which will make all other
VM_FLUSH_RESET_PERMS vfrees wait while the purge operation happens.

Lastly, page_address() can involve locking and lookups on some
configurations, so skip calling this by exiting out early when
!CONFIG_ARCH_HAS_SET_DIRECT_MAP.

Cc: Meelis Roos <mr...@linux.ee>
Cc: Peter Zijlstra <pet...@infradead.org>
Cc: "David S. Miller" <da...@davemloft.net>
Cc: Dave Hansen <dave.han...@intel.com>
Cc: Borislav Petkov <b...@alien8.de>
Cc: Andy Lutomirski <l...@kernel.org>
Cc: Ingo Molnar <mi...@redhat.com>
Cc: Nadav Amit <na...@vmware.com>
Signed-off-by: Rick Edgecombe <rick.p.edgeco...@intel.com>
---
 mm/vmalloc.c | 18 ++++++++++--------
 1 file changed, 10 insertions(+), 8 deletions(-)

diff --git a/mm/vmalloc.c b/mm/vmalloc.c
index 836888ae01f6..8d03427626dc 100644
--- a/mm/vmalloc.c
+++ b/mm/vmalloc.c
@@ -2122,9 +2122,10 @@ 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)
 {
+       const bool has_set_direct = IS_ENABLED(CONFIG_ARCH_HAS_SET_DIRECT_MAP);
+       const bool flush_reset = area->flags & VM_FLUSH_RESET_PERMS;
        unsigned long addr = (unsigned long)area->addr;
-       unsigned long start = ULONG_MAX, end = 0;
-       int flush_reset = area->flags & VM_FLUSH_RESET_PERMS;
+       unsigned long start = addr, end = addr + area->size;
        int i;
 
        /*
@@ -2133,7 +2134,7 @@ static void vm_remove_mappings(struct vm_struct *area, 
int deallocate_pages)
         * This is concerned with resetting the direct map any an vm alias with
         * execute permissions, without leaving a RW+X window.
         */
-       if (flush_reset && !IS_ENABLED(CONFIG_ARCH_HAS_SET_DIRECT_MAP)) {
+       if (flush_reset && !has_set_direct) {
                set_memory_nx(addr, area->nr_pages);
                set_memory_rw(addr, area->nr_pages);
        }
@@ -2146,17 +2147,18 @@ static void vm_remove_mappings(struct vm_struct *area, 
int deallocate_pages)
 
        /*
         * If not deallocating pages, just do the flush of the VM area and
-        * return.
+        * return. If the arch doesn't have set_direct_map_(), also skip the
+        * below work.
         */
-       if (!deallocate_pages) {
-               vm_unmap_aliases();
+       if (!deallocate_pages || !has_set_direct) {
+               flush_tlb_kernel_range(start, end);
                return;
        }
 
        /*
         * If execution gets here, flush the vm mapping and reset the direct
         * map. Find the start and end range of the direct mappings to make sure
-        * the vm_unmap_aliases() flush includes the direct map.
+        * the flush_tlb_kernel_range() includes the direct map.
         */
        for (i = 0; i < area->nr_pages; i++) {
                addr = (unsigned long)page_address(area->pages[i]);
@@ -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);
+       flush_tlb_kernel_range(start, end);
        set_area_direct_map(area, set_direct_map_default_noflush);
 }
 
-- 
2.20.1

Reply via email to