diff mbox series

[v2,09/13] perf mem_info: Add and use map_symbol__exit and addr_map_symbol__exit

Message ID 20231012062359.1616786-10-irogers@google.com (mailing list archive)
State Not Applicable
Headers show
Series Improvements to memory use | expand

Checks

Context Check Description
netdev/tree_selection success Not a local patch

Commit Message

Ian Rogers Oct. 12, 2023, 6:23 a.m. UTC
Fix leak where mem_info__put wouldn't release the maps/map as used by
perf mem. Add exit functions and use elsewhere that the maps and map
are released.

Signed-off-by: Ian Rogers <irogers@google.com>
---
 tools/perf/util/Build        |  1 +
 tools/perf/util/callchain.c  | 15 +++++----------
 tools/perf/util/hist.c       |  6 ++----
 tools/perf/util/machine.c    |  6 ++----
 tools/perf/util/map_symbol.c | 15 +++++++++++++++
 tools/perf/util/map_symbol.h |  4 ++++
 tools/perf/util/symbol.c     |  5 ++++-
 7 files changed, 33 insertions(+), 19 deletions(-)
 create mode 100644 tools/perf/util/map_symbol.c

Comments

Namhyung Kim Oct. 18, 2023, 11:20 p.m. UTC | #1
On Wed, Oct 11, 2023 at 11:24 PM Ian Rogers <irogers@google.com> wrote:
>
> Fix leak where mem_info__put wouldn't release the maps/map as used by
> perf mem. Add exit functions and use elsewhere that the maps and map
> are released.
>
> Signed-off-by: Ian Rogers <irogers@google.com>
> ---
>  tools/perf/util/Build        |  1 +
>  tools/perf/util/callchain.c  | 15 +++++----------
>  tools/perf/util/hist.c       |  6 ++----
>  tools/perf/util/machine.c    |  6 ++----
>  tools/perf/util/map_symbol.c | 15 +++++++++++++++
>  tools/perf/util/map_symbol.h |  4 ++++
>  tools/perf/util/symbol.c     |  5 ++++-
>  7 files changed, 33 insertions(+), 19 deletions(-)
>  create mode 100644 tools/perf/util/map_symbol.c
>
> diff --git a/tools/perf/util/Build b/tools/perf/util/Build
> index 0ea5a9d368d4..96058f949ec9 100644
> --- a/tools/perf/util/Build
> +++ b/tools/perf/util/Build
> @@ -49,6 +49,7 @@ perf-y += dso.o
>  perf-y += dsos.o
>  perf-y += symbol.o
>  perf-y += symbol_fprintf.o
> +perf-y += map_symbol.o
>  perf-y += color.o
>  perf-y += color_config.o
>  perf-y += metricgroup.o
> diff --git a/tools/perf/util/callchain.c b/tools/perf/util/callchain.c
> index 0a7919c2af91..02881d5b822c 100644
> --- a/tools/perf/util/callchain.c
> +++ b/tools/perf/util/callchain.c
> @@ -1496,16 +1496,14 @@ static void free_callchain_node(struct callchain_node *node)
>
>         list_for_each_entry_safe(list, tmp, &node->parent_val, list) {
>                 list_del_init(&list->list);
> -               map__zput(list->ms.map);
> -               maps__zput(list->ms.maps);
> +               map_symbol__exit(&list->ms);
>                 zfree(&list->brtype_stat);
>                 free(list);
>         }
>
>         list_for_each_entry_safe(list, tmp, &node->val, list) {
>                 list_del_init(&list->list);
> -               map__zput(list->ms.map);
> -               maps__zput(list->ms.maps);
> +               map_symbol__exit(&list->ms);
>                 zfree(&list->brtype_stat);
>                 free(list);
>         }
> @@ -1591,8 +1589,7 @@ int callchain_node__make_parent_list(struct callchain_node *node)
>  out:
>         list_for_each_entry_safe(chain, new, &head, list) {
>                 list_del_init(&chain->list);
> -               map__zput(chain->ms.map);
> -               maps__zput(chain->ms.maps);
> +               map_symbol__exit(&chain->ms);
>                 zfree(&chain->brtype_stat);
>                 free(chain);
>         }
> @@ -1676,10 +1673,8 @@ void callchain_cursor_reset(struct callchain_cursor *cursor)
>         cursor->nr = 0;
>         cursor->last = &cursor->first;
>
> -       for (node = cursor->first; node != NULL; node = node->next) {
> -               map__zput(node->ms.map);
> -               maps__zput(node->ms.maps);
> -       }
> +       for (node = cursor->first; node != NULL; node = node->next)
> +               map_symbol__exit(&node->ms);
>  }
>
>  void callchain_param_setup(u64 sample_type, const char *arch)
> diff --git a/tools/perf/util/hist.c b/tools/perf/util/hist.c
> index ac8c0ef48a7f..d62693b8fad8 100644
> --- a/tools/perf/util/hist.c
> +++ b/tools/perf/util/hist.c
> @@ -524,8 +524,7 @@ static int hist_entry__init(struct hist_entry *he,
>                 map__put(he->mem_info->daddr.ms.map);
>         }
>  err:
> -       maps__zput(he->ms.maps);
> -       map__zput(he->ms.map);
> +       map_symbol__exit(&he->ms);
>         zfree(&he->stat_acc);
>         return -ENOMEM;
>  }
> @@ -1317,8 +1316,7 @@ void hist_entry__delete(struct hist_entry *he)
>         struct hist_entry_ops *ops = he->ops;
>
>         thread__zput(he->thread);
> -       maps__zput(he->ms.maps);
> -       map__zput(he->ms.map);
> +       map_symbol__exit(&he->ms);
>
>         if (he->branch_info) {
>                 map__zput(he->branch_info->from.ms.map);

What about he->branch_info and he->mem_info ?

Also I think we can use it in hists__account_cycles() too.

Thanks,
Namhyung


> diff --git a/tools/perf/util/machine.c b/tools/perf/util/machine.c
> index 8e5085b77c7b..6ca7500e2cf4 100644
> --- a/tools/perf/util/machine.c
> +++ b/tools/perf/util/machine.c
> @@ -2389,8 +2389,7 @@ static int add_callchain_ip(struct thread *thread,
>                                       iter_cycles, branch_from, srcline);
>  out:
>         addr_location__exit(&al);
> -       maps__put(ms.maps);
> -       map__put(ms.map);
> +       map_symbol__exit(&ms);
>         return err;
>  }
>
> @@ -3116,8 +3115,7 @@ static int append_inlines(struct callchain_cursor *cursor, struct map_symbol *ms
>                 if (ret != 0)
>                         return ret;
>         }
> -       map__put(ilist_ms.map);
> -       maps__put(ilist_ms.maps);
> +       map_symbol__exit(&ilist_ms);
>
>         return ret;
>  }
> diff --git a/tools/perf/util/map_symbol.c b/tools/perf/util/map_symbol.c
> new file mode 100644
> index 000000000000..bef5079f2403
> --- /dev/null
> +++ b/tools/perf/util/map_symbol.c
> @@ -0,0 +1,15 @@
> +// SPDX-License-Identifier: GPL-2.0
> +#include "map_symbol.h"
> +#include "maps.h"
> +#include "map.h"
> +
> +void map_symbol__exit(struct map_symbol *ms)
> +{
> +       maps__zput(ms->maps);
> +       map__zput(ms->map);
> +}
> +
> +void addr_map_symbol__exit(struct addr_map_symbol *ams)
> +{
> +       map_symbol__exit(&ams->ms);
> +}
> diff --git a/tools/perf/util/map_symbol.h b/tools/perf/util/map_symbol.h
> index e08817b0c30f..72d5ed938ed6 100644
> --- a/tools/perf/util/map_symbol.h
> +++ b/tools/perf/util/map_symbol.h
> @@ -22,4 +22,8 @@ struct addr_map_symbol {
>         u64           phys_addr;
>         u64           data_page_size;
>  };
> +
> +void map_symbol__exit(struct map_symbol *ms);
> +void addr_map_symbol__exit(struct addr_map_symbol *ams);
> +
>  #endif // __PERF_MAP_SYMBOL
> diff --git a/tools/perf/util/symbol.c b/tools/perf/util/symbol.c
> index 2740d4457c13..d67a87072eec 100644
> --- a/tools/perf/util/symbol.c
> +++ b/tools/perf/util/symbol.c
> @@ -2790,8 +2790,11 @@ struct mem_info *mem_info__get(struct mem_info *mi)
>
>  void mem_info__put(struct mem_info *mi)
>  {
> -       if (mi && refcount_dec_and_test(&mi->refcnt))
> +       if (mi && refcount_dec_and_test(&mi->refcnt)) {
> +               addr_map_symbol__exit(&mi->iaddr);
> +               addr_map_symbol__exit(&mi->daddr);
>                 free(mi);
> +       }
>  }
>
>  struct mem_info *mem_info__new(void)
> --
> 2.42.0.609.gbb76f46606-goog
>
Ian Rogers Oct. 24, 2023, 8:52 p.m. UTC | #2
On Wed, Oct 18, 2023 at 4:20 PM Namhyung Kim <namhyung@kernel.org> wrote:
>
> On Wed, Oct 11, 2023 at 11:24 PM Ian Rogers <irogers@google.com> wrote:
> >
> > Fix leak where mem_info__put wouldn't release the maps/map as used by
> > perf mem. Add exit functions and use elsewhere that the maps and map
> > are released.
> >
> > Signed-off-by: Ian Rogers <irogers@google.com>
> > ---
> >  tools/perf/util/Build        |  1 +
> >  tools/perf/util/callchain.c  | 15 +++++----------
> >  tools/perf/util/hist.c       |  6 ++----
> >  tools/perf/util/machine.c    |  6 ++----
> >  tools/perf/util/map_symbol.c | 15 +++++++++++++++
> >  tools/perf/util/map_symbol.h |  4 ++++
> >  tools/perf/util/symbol.c     |  5 ++++-
> >  7 files changed, 33 insertions(+), 19 deletions(-)
> >  create mode 100644 tools/perf/util/map_symbol.c
> >
> > diff --git a/tools/perf/util/Build b/tools/perf/util/Build
> > index 0ea5a9d368d4..96058f949ec9 100644
> > --- a/tools/perf/util/Build
> > +++ b/tools/perf/util/Build
> > @@ -49,6 +49,7 @@ perf-y += dso.o
> >  perf-y += dsos.o
> >  perf-y += symbol.o
> >  perf-y += symbol_fprintf.o
> > +perf-y += map_symbol.o
> >  perf-y += color.o
> >  perf-y += color_config.o
> >  perf-y += metricgroup.o
> > diff --git a/tools/perf/util/callchain.c b/tools/perf/util/callchain.c
> > index 0a7919c2af91..02881d5b822c 100644
> > --- a/tools/perf/util/callchain.c
> > +++ b/tools/perf/util/callchain.c
> > @@ -1496,16 +1496,14 @@ static void free_callchain_node(struct callchain_node *node)
> >
> >         list_for_each_entry_safe(list, tmp, &node->parent_val, list) {
> >                 list_del_init(&list->list);
> > -               map__zput(list->ms.map);
> > -               maps__zput(list->ms.maps);
> > +               map_symbol__exit(&list->ms);
> >                 zfree(&list->brtype_stat);
> >                 free(list);
> >         }
> >
> >         list_for_each_entry_safe(list, tmp, &node->val, list) {
> >                 list_del_init(&list->list);
> > -               map__zput(list->ms.map);
> > -               maps__zput(list->ms.maps);
> > +               map_symbol__exit(&list->ms);
> >                 zfree(&list->brtype_stat);
> >                 free(list);
> >         }
> > @@ -1591,8 +1589,7 @@ int callchain_node__make_parent_list(struct callchain_node *node)
> >  out:
> >         list_for_each_entry_safe(chain, new, &head, list) {
> >                 list_del_init(&chain->list);
> > -               map__zput(chain->ms.map);
> > -               maps__zput(chain->ms.maps);
> > +               map_symbol__exit(&chain->ms);
> >                 zfree(&chain->brtype_stat);
> >                 free(chain);
> >         }
> > @@ -1676,10 +1673,8 @@ void callchain_cursor_reset(struct callchain_cursor *cursor)
> >         cursor->nr = 0;
> >         cursor->last = &cursor->first;
> >
> > -       for (node = cursor->first; node != NULL; node = node->next) {
> > -               map__zput(node->ms.map);
> > -               maps__zput(node->ms.maps);
> > -       }
> > +       for (node = cursor->first; node != NULL; node = node->next)
> > +               map_symbol__exit(&node->ms);
> >  }
> >
> >  void callchain_param_setup(u64 sample_type, const char *arch)
> > diff --git a/tools/perf/util/hist.c b/tools/perf/util/hist.c
> > index ac8c0ef48a7f..d62693b8fad8 100644
> > --- a/tools/perf/util/hist.c
> > +++ b/tools/perf/util/hist.c
> > @@ -524,8 +524,7 @@ static int hist_entry__init(struct hist_entry *he,
> >                 map__put(he->mem_info->daddr.ms.map);
> >         }
> >  err:
> > -       maps__zput(he->ms.maps);
> > -       map__zput(he->ms.map);
> > +       map_symbol__exit(&he->ms);
> >         zfree(&he->stat_acc);
> >         return -ENOMEM;
> >  }
> > @@ -1317,8 +1316,7 @@ void hist_entry__delete(struct hist_entry *he)
> >         struct hist_entry_ops *ops = he->ops;
> >
> >         thread__zput(he->thread);
> > -       maps__zput(he->ms.maps);
> > -       map__zput(he->ms.map);
> > +       map_symbol__exit(&he->ms);
> >
> >         if (he->branch_info) {
> >                 map__zput(he->branch_info->from.ms.map);
>
> What about he->branch_info and he->mem_info ?
>
> Also I think we can use it in hists__account_cycles() too.

Thanks, I addressed the ones I could find. Running perf mem report I still see:

==2948587==ERROR: LeakSanitizer: detected memory leaks

Direct leak of 1296 byte(s) in 162 object(s) allocated from:
    #0 0x7f2a018d85bf in __interceptor_malloc
../../../../src/libsanitizer/asan/asan_malloc_linux.cpp:69
    #1 0x556104500e47 in map__get util/map.h:180
    #2 0x55610450cd6d in ip__resolve_data util/machine.c:2260
    #3 0x55610450d022 in sample__resolve_mem util/machine.c:2275
    #4 0x55610459dc73 in iter_prepare_mem_entry util/hist.c:824
    #5 0x5561045a0175 in hist_entry_iter__add util/hist.c:1238
    #6 0x5561042d8e8e in process_sample_event tools/perf/builtin-report.c:332
    #7 0x556104526a37 in evlist__deliver_sample util/session.c:1518
    #8 0x556104526d18 in machines__deliver_event util/session.c:1557
    #9 0x556104527942 in perf_session__deliver_event util/session.c:1639
    #10 0x55610451d576 in ordered_events__deliver_event util/session.c:188
    #11 0x556104536b6a in do_flush util/ordered-events.c:245
    #12 0x5561045372a8 in __ordered_events__flush util/ordered-events.c:324
    #13 0x556104537544 in ordered_events__flush util/ordered-events.c:342
    #14 0x5561045240b0 in perf_event__process_finished_round util/session.c:1080
    #15 0x556104527fd5 in perf_session__process_user_event util/session.c:1692
    #16 0x556104529396 in perf_session__process_event util/session.c:1861
    #17 0x55610452cf8c in process_simple util/session.c:2436
    #18 0x55610452c69b in reader__read_event util/session.c:2365
    #19 0x55610452cb36 in reader__process_events util/session.c:2414
    #20 0x55610452d467 in __perf_session__process_events util/session.c:2461
    #21 0x55610452ec52 in perf_session__process_events util/session.c:2627
    #22 0x5561042dd613 in __cmd_report tools/perf/builtin-report.c:992
    #23 0x5561042e7ac7 in cmd_report tools/perf/builtin-report.c:1711
    #24 0x556104343967 in report_events tools/perf/builtin-mem.c:374
    #25 0x556104345278 in cmd_mem tools/perf/builtin-mem.c:516
    #26 0x55610441848d in run_builtin tools/perf/perf.c:322
    #27 0x5561044189f9 in handle_internal_command tools/perf/perf.c:375
    #28 0x556104418dc1 in run_argv tools/perf/perf.c:419
    #29 0x556104419329 in main tools/perf/perf.c:535

Direct leak of 1296 byte(s) in 162 object(s) allocated from:
    #0 0x7f2a018d85bf in __interceptor_malloc
../../../../src/libsanitizer/asan/asan_malloc_linux.cpp:69
    #1 0x556104500e47 in map__get util/map.h:180
    #2 0x55610450c97e in ip__resolve_ams util/machine.c:2239
    #3 0x55610450cf1b in sample__resolve_mem util/machine.c:2274
    #4 0x55610459dc73 in iter_prepare_mem_entry util/hist.c:824
    #5 0x5561045a0175 in hist_entry_iter__add util/hist.c:1238
    #6 0x5561042d8e8e in process_sample_event tools/perf/builtin-report.c:332
    #7 0x556104526a37 in evlist__deliver_sample util/session.c:1518
    #8 0x556104526d18 in machines__deliver_event util/session.c:1557
    #9 0x556104527942 in perf_session__deliver_event util/session.c:1639
    #10 0x55610451d576 in ordered_events__deliver_event util/session.c:188
    #11 0x556104536b6a in do_flush util/ordered-events.c:245
    #12 0x5561045372a8 in __ordered_events__flush util/ordered-events.c:324
    #13 0x556104537544 in ordered_events__flush util/ordered-events.c:342
    #14 0x5561045240b0 in perf_event__process_finished_round util/session.c:1080
    #15 0x556104527fd5 in perf_session__process_user_event util/session.c:1692
    #16 0x556104529396 in perf_session__process_event util/session.c:1861
    #17 0x55610452cf8c in process_simple util/session.c:2436
    #18 0x55610452c69b in reader__read_event util/session.c:2365
    #19 0x55610452cb36 in reader__process_events util/session.c:2414
    #20 0x55610452d467 in __perf_session__process_events util/session.c:2461
    #21 0x55610452ec52 in perf_session__process_events util/session.c:2627
    #22 0x5561042dd613 in __cmd_report tools/perf/builtin-report.c:992
    #23 0x5561042e7ac7 in cmd_report tools/perf/builtin-report.c:1711
    #24 0x556104343967 in report_events tools/perf/builtin-mem.c:374
    #25 0x556104345278 in cmd_mem tools/perf/builtin-mem.c:516
    #26 0x55610441848d in run_builtin tools/perf/perf.c:322
    #27 0x5561044189f9 in handle_internal_command tools/perf/perf.c:375
    #28 0x556104418dc1 in run_argv tools/perf/perf.c:419
    #29 0x556104419329 in main tools/perf/perf.c:535

Which looks like iter->priv was freed without puts. The use of void*s
and the general layout of the histogram code I find confusing. It
would be nice to get this resolved but I wasn't able to do it in v3.

Thanks,
Ian

> Thanks,
> Namhyung
>
>
> > diff --git a/tools/perf/util/machine.c b/tools/perf/util/machine.c
> > index 8e5085b77c7b..6ca7500e2cf4 100644
> > --- a/tools/perf/util/machine.c
> > +++ b/tools/perf/util/machine.c
> > @@ -2389,8 +2389,7 @@ static int add_callchain_ip(struct thread *thread,
> >                                       iter_cycles, branch_from, srcline);
> >  out:
> >         addr_location__exit(&al);
> > -       maps__put(ms.maps);
> > -       map__put(ms.map);
> > +       map_symbol__exit(&ms);
> >         return err;
> >  }
> >
> > @@ -3116,8 +3115,7 @@ static int append_inlines(struct callchain_cursor *cursor, struct map_symbol *ms
> >                 if (ret != 0)
> >                         return ret;
> >         }
> > -       map__put(ilist_ms.map);
> > -       maps__put(ilist_ms.maps);
> > +       map_symbol__exit(&ilist_ms);
> >
> >         return ret;
> >  }
> > diff --git a/tools/perf/util/map_symbol.c b/tools/perf/util/map_symbol.c
> > new file mode 100644
> > index 000000000000..bef5079f2403
> > --- /dev/null
> > +++ b/tools/perf/util/map_symbol.c
> > @@ -0,0 +1,15 @@
> > +// SPDX-License-Identifier: GPL-2.0
> > +#include "map_symbol.h"
> > +#include "maps.h"
> > +#include "map.h"
> > +
> > +void map_symbol__exit(struct map_symbol *ms)
> > +{
> > +       maps__zput(ms->maps);
> > +       map__zput(ms->map);
> > +}
> > +
> > +void addr_map_symbol__exit(struct addr_map_symbol *ams)
> > +{
> > +       map_symbol__exit(&ams->ms);
> > +}
> > diff --git a/tools/perf/util/map_symbol.h b/tools/perf/util/map_symbol.h
> > index e08817b0c30f..72d5ed938ed6 100644
> > --- a/tools/perf/util/map_symbol.h
> > +++ b/tools/perf/util/map_symbol.h
> > @@ -22,4 +22,8 @@ struct addr_map_symbol {
> >         u64           phys_addr;
> >         u64           data_page_size;
> >  };
> > +
> > +void map_symbol__exit(struct map_symbol *ms);
> > +void addr_map_symbol__exit(struct addr_map_symbol *ams);
> > +
> >  #endif // __PERF_MAP_SYMBOL
> > diff --git a/tools/perf/util/symbol.c b/tools/perf/util/symbol.c
> > index 2740d4457c13..d67a87072eec 100644
> > --- a/tools/perf/util/symbol.c
> > +++ b/tools/perf/util/symbol.c
> > @@ -2790,8 +2790,11 @@ struct mem_info *mem_info__get(struct mem_info *mi)
> >
> >  void mem_info__put(struct mem_info *mi)
> >  {
> > -       if (mi && refcount_dec_and_test(&mi->refcnt))
> > +       if (mi && refcount_dec_and_test(&mi->refcnt)) {
> > +               addr_map_symbol__exit(&mi->iaddr);
> > +               addr_map_symbol__exit(&mi->daddr);
> >                 free(mi);
> > +       }
> >  }
> >
> >  struct mem_info *mem_info__new(void)
> > --
> > 2.42.0.609.gbb76f46606-goog
> >
diff mbox series

Patch

diff --git a/tools/perf/util/Build b/tools/perf/util/Build
index 0ea5a9d368d4..96058f949ec9 100644
--- a/tools/perf/util/Build
+++ b/tools/perf/util/Build
@@ -49,6 +49,7 @@  perf-y += dso.o
 perf-y += dsos.o
 perf-y += symbol.o
 perf-y += symbol_fprintf.o
+perf-y += map_symbol.o
 perf-y += color.o
 perf-y += color_config.o
 perf-y += metricgroup.o
diff --git a/tools/perf/util/callchain.c b/tools/perf/util/callchain.c
index 0a7919c2af91..02881d5b822c 100644
--- a/tools/perf/util/callchain.c
+++ b/tools/perf/util/callchain.c
@@ -1496,16 +1496,14 @@  static void free_callchain_node(struct callchain_node *node)
 
 	list_for_each_entry_safe(list, tmp, &node->parent_val, list) {
 		list_del_init(&list->list);
-		map__zput(list->ms.map);
-		maps__zput(list->ms.maps);
+		map_symbol__exit(&list->ms);
 		zfree(&list->brtype_stat);
 		free(list);
 	}
 
 	list_for_each_entry_safe(list, tmp, &node->val, list) {
 		list_del_init(&list->list);
-		map__zput(list->ms.map);
-		maps__zput(list->ms.maps);
+		map_symbol__exit(&list->ms);
 		zfree(&list->brtype_stat);
 		free(list);
 	}
@@ -1591,8 +1589,7 @@  int callchain_node__make_parent_list(struct callchain_node *node)
 out:
 	list_for_each_entry_safe(chain, new, &head, list) {
 		list_del_init(&chain->list);
-		map__zput(chain->ms.map);
-		maps__zput(chain->ms.maps);
+		map_symbol__exit(&chain->ms);
 		zfree(&chain->brtype_stat);
 		free(chain);
 	}
@@ -1676,10 +1673,8 @@  void callchain_cursor_reset(struct callchain_cursor *cursor)
 	cursor->nr = 0;
 	cursor->last = &cursor->first;
 
-	for (node = cursor->first; node != NULL; node = node->next) {
-		map__zput(node->ms.map);
-		maps__zput(node->ms.maps);
-	}
+	for (node = cursor->first; node != NULL; node = node->next)
+		map_symbol__exit(&node->ms);
 }
 
 void callchain_param_setup(u64 sample_type, const char *arch)
diff --git a/tools/perf/util/hist.c b/tools/perf/util/hist.c
index ac8c0ef48a7f..d62693b8fad8 100644
--- a/tools/perf/util/hist.c
+++ b/tools/perf/util/hist.c
@@ -524,8 +524,7 @@  static int hist_entry__init(struct hist_entry *he,
 		map__put(he->mem_info->daddr.ms.map);
 	}
 err:
-	maps__zput(he->ms.maps);
-	map__zput(he->ms.map);
+	map_symbol__exit(&he->ms);
 	zfree(&he->stat_acc);
 	return -ENOMEM;
 }
@@ -1317,8 +1316,7 @@  void hist_entry__delete(struct hist_entry *he)
 	struct hist_entry_ops *ops = he->ops;
 
 	thread__zput(he->thread);
-	maps__zput(he->ms.maps);
-	map__zput(he->ms.map);
+	map_symbol__exit(&he->ms);
 
 	if (he->branch_info) {
 		map__zput(he->branch_info->from.ms.map);
diff --git a/tools/perf/util/machine.c b/tools/perf/util/machine.c
index 8e5085b77c7b..6ca7500e2cf4 100644
--- a/tools/perf/util/machine.c
+++ b/tools/perf/util/machine.c
@@ -2389,8 +2389,7 @@  static int add_callchain_ip(struct thread *thread,
 				      iter_cycles, branch_from, srcline);
 out:
 	addr_location__exit(&al);
-	maps__put(ms.maps);
-	map__put(ms.map);
+	map_symbol__exit(&ms);
 	return err;
 }
 
@@ -3116,8 +3115,7 @@  static int append_inlines(struct callchain_cursor *cursor, struct map_symbol *ms
 		if (ret != 0)
 			return ret;
 	}
-	map__put(ilist_ms.map);
-	maps__put(ilist_ms.maps);
+	map_symbol__exit(&ilist_ms);
 
 	return ret;
 }
diff --git a/tools/perf/util/map_symbol.c b/tools/perf/util/map_symbol.c
new file mode 100644
index 000000000000..bef5079f2403
--- /dev/null
+++ b/tools/perf/util/map_symbol.c
@@ -0,0 +1,15 @@ 
+// SPDX-License-Identifier: GPL-2.0
+#include "map_symbol.h"
+#include "maps.h"
+#include "map.h"
+
+void map_symbol__exit(struct map_symbol *ms)
+{
+	maps__zput(ms->maps);
+	map__zput(ms->map);
+}
+
+void addr_map_symbol__exit(struct addr_map_symbol *ams)
+{
+	map_symbol__exit(&ams->ms);
+}
diff --git a/tools/perf/util/map_symbol.h b/tools/perf/util/map_symbol.h
index e08817b0c30f..72d5ed938ed6 100644
--- a/tools/perf/util/map_symbol.h
+++ b/tools/perf/util/map_symbol.h
@@ -22,4 +22,8 @@  struct addr_map_symbol {
 	u64	      phys_addr;
 	u64	      data_page_size;
 };
+
+void map_symbol__exit(struct map_symbol *ms);
+void addr_map_symbol__exit(struct addr_map_symbol *ams);
+
 #endif // __PERF_MAP_SYMBOL
diff --git a/tools/perf/util/symbol.c b/tools/perf/util/symbol.c
index 2740d4457c13..d67a87072eec 100644
--- a/tools/perf/util/symbol.c
+++ b/tools/perf/util/symbol.c
@@ -2790,8 +2790,11 @@  struct mem_info *mem_info__get(struct mem_info *mi)
 
 void mem_info__put(struct mem_info *mi)
 {
-	if (mi && refcount_dec_and_test(&mi->refcnt))
+	if (mi && refcount_dec_and_test(&mi->refcnt)) {
+		addr_map_symbol__exit(&mi->iaddr);
+		addr_map_symbol__exit(&mi->daddr);
 		free(mi);
+	}
 }
 
 struct mem_info *mem_info__new(void)