From: Arnaldo Carvalho de Melo <a...@redhat.com>

We can conceivably have multiple BPF object files for other purposes, so
better look just on the BPF object containing the __augmented_syscalls__
map for all things augmented_syscalls related.

Cc: Adrian Hunter <adrian.hun...@intel.com>
Cc: Brendan Gregg <brendan.d.gr...@gmail.com>
Cc: Jiri Olsa <jo...@kernel.org>
Cc: Luis Cláudio Gonçalves <lclau...@redhat.com>
Cc: Namhyung Kim <namhy...@kernel.org>
Link: https://lkml.kernel.org/n/tip-3jt8knkuae9lt705r1lns...@git.kernel.org
Signed-off-by: Arnaldo Carvalho de Melo <a...@redhat.com>
---
 tools/perf/builtin-trace.c | 20 +++++++-------------
 1 file changed, 7 insertions(+), 13 deletions(-)

diff --git a/tools/perf/builtin-trace.c b/tools/perf/builtin-trace.c
index 6aa080845a84..bfd739a321d1 100644
--- a/tools/perf/builtin-trace.c
+++ b/tools/perf/builtin-trace.c
@@ -3669,28 +3669,22 @@ static int trace__parse_cgroups(const struct option 
*opt, const char *str, int u
        return 0;
 }
 
-static struct bpf_map *bpf__find_map_by_name(const char *name)
+static struct bpf_map *trace__find_bpf_map_by_name(struct trace *trace, const 
char *name)
 {
-       struct bpf_object *obj, *tmp;
-
-       bpf_object__for_each_safe(obj, tmp) {
-               struct bpf_map *map = bpf_object__find_map_by_name(obj, name);
-               if (map)
-                       return map;
-
-       }
+       if (trace->bpf_obj == NULL)
+               return NULL;
 
-       return NULL;
+       return bpf_object__find_map_by_name(trace->bpf_obj, name);
 }
 
 static void trace__set_bpf_map_filtered_pids(struct trace *trace)
 {
-       trace->filter_pids.map = bpf__find_map_by_name("pids_filtered");
+       trace->filter_pids.map = trace__find_bpf_map_by_name(trace, 
"pids_filtered");
 }
 
 static void trace__set_bpf_map_syscalls(struct trace *trace)
 {
-       trace->syscalls.map = bpf__find_map_by_name("syscalls");
+       trace->syscalls.map = trace__find_bpf_map_by_name(trace, "syscalls");
 }
 
 static int trace__config(const char *var, const char *value, void *arg)
@@ -3924,7 +3918,7 @@ int cmd_trace(int argc, const char **argv)
        err = -1;
 
        if (map_dump_str) {
-               trace.dump.map = bpf__find_map_by_name(map_dump_str);
+               trace.dump.map = trace__find_bpf_map_by_name(&trace, 
map_dump_str);
                if (trace.dump.map == NULL) {
                        pr_err("ERROR: BPF map \"%s\" not found\n", 
map_dump_str);
                        goto out;
-- 
2.21.0

Reply via email to