1 files changed, 20 insertions, 12 deletions

diff --git a/tools/perf/util/machine.c b/tools/perf/util/machine.c
index 61959aba7e27..3c520baa198c 100644
--- a/tools/perf/util/machine.c
+++ b/tools/perf/util/machine.c
@@ -1421,6 +1421,20 @@ static void machine__set_kernel_mmap(struct machine *machine,
                 machine->vmlinux_map->end = ~0ULL;
 }
 
+static void machine__update_kernel_mmap(struct machine *machine,
+                                     u64 start, u64 end)
+{
+        struct map *map = machine__kernel_map(machine);
+
+        map__get(map);
+        map_groups__remove(&machine->kmaps, map);
+
+        machine__set_kernel_mmap(machine, start, end);
+
+        map_groups__insert(&machine->kmaps, map);
+        map__put(map);
+}
+
 int machine__create_kernel_maps(struct machine *machine)
 {
         struct dso *kernel = machine__get_kernel(machine);
@@ -1453,17 +1467,11 @@ int machine__create_kernel_maps(struct machine *machine)
                         goto out_put;
                 }
 
-                /* we have a real start address now, so re-order the kmaps */
-                map = machine__kernel_map(machine);
-
-                map__get(map);
-                map_groups__remove(&machine->kmaps, map);
-
-                /* assume it's the last in the kmaps */
-                machine__set_kernel_mmap(machine, addr, ~0ULL);
-
-                map_groups__insert(&machine->kmaps, map);
-                map__put(map);
+                /*
+                 * we have a real start address now, so re-order the kmaps
+                 * assume it's the last in the kmaps
+                 */
+                machine__update_kernel_mmap(machine, addr, ~0ULL);
         }
 
         if (machine__create_extra_kernel_maps(machine, kernel))
@@ -1599,7 +1607,7 @@ static int machine__process_kernel_mmap_event(struct machine *machine,
                 if (strstr(kernel->long_name, "vmlinux"))
                         dso__set_short_name(kernel, "[kernel.vmlinux]", false);
 
-                machine__set_kernel_mmap(machine, event->mmap.start,
+                machine__update_kernel_mmap(machine, event->mmap.start,
                                          event->mmap.start + event->mmap.len);
 
                 /*