On Wed, May 18, 2022 at 9:02 PM Ian Rogers <irogers@xxxxxxxxxx> wrote:
>
> On Wed, May 18, 2022 at 3:47 PM Namhyung Kim <namhyung@xxxxxxxxxx> wrote:
> >
> > It should honor cpu and task filtering with -a, -C or -p, -t options.
> >
> > Signed-off-by: Namhyung Kim <namhyung@xxxxxxxxxx>
> > ---
> > tools/perf/builtin-record.c | 2 +-
> > tools/perf/util/bpf_off_cpu.c | 78 +++++++++++++++++++++++---
> > tools/perf/util/bpf_skel/off_cpu.bpf.c | 52 +++++++++++++++--
> > tools/perf/util/off_cpu.h | 6 +-
> > 4 files changed, 123 insertions(+), 15 deletions(-)
> >
> > diff --git a/tools/perf/builtin-record.c b/tools/perf/builtin-record.c
> > index 91f88501412e..7f60d2eac0b4 100644
> > --- a/tools/perf/builtin-record.c
> > +++ b/tools/perf/builtin-record.c
> > @@ -907,7 +907,7 @@ static int record__config_text_poke(struct evlist *evlist)
> >
> > static int record__config_off_cpu(struct record *rec)
> > {
> > - return off_cpu_prepare(rec->evlist);
> > + return off_cpu_prepare(rec->evlist, &rec->opts.target);
> > }
> >
> > static bool record__kcore_readable(struct machine *machine)
> > diff --git a/tools/perf/util/bpf_off_cpu.c b/tools/perf/util/bpf_off_cpu.c
> > index 9ed7aca3f4ac..b5e2d038da50 100644
> > --- a/tools/perf/util/bpf_off_cpu.c
> > +++ b/tools/perf/util/bpf_off_cpu.c
> > @@ -6,6 +6,9 @@
> > #include "util/off_cpu.h"
> > #include "util/perf-hooks.h"
> > #include "util/session.h"
> > +#include "util/target.h"
> > +#include "util/cpumap.h"
> > +#include "util/thread_map.h"
> > #include <bpf/bpf.h>
> >
> > #include "bpf_skel/off_cpu.skel.h"
> > @@ -60,8 +63,23 @@ static int off_cpu_config(struct evlist *evlist)
> > return 0;
> > }
> >
> > -static void off_cpu_start(void *arg __maybe_unused)
> > +static void off_cpu_start(void *arg)
> > {
> > + struct evlist *evlist = arg;
> > +
> > + /* update task filter for the given workload */
> > + if (!skel->bss->has_cpu && !skel->bss->has_task &&
> > + perf_thread_map__pid(evlist->core.threads, 0) != -1) {
> > + int fd;
> > + u32 pid;
> > + u8 val = 1;
> > +
> > + skel->bss->has_task = 1;
> > + fd = bpf_map__fd(skel->maps.task_filter);
> > + pid = perf_thread_map__pid(evlist->core.threads, 0);
> > + bpf_map_update_elem(fd, &pid, &val, BPF_ANY);
> > + }
> > +
> > skel->bss->enabled = 1;
> > }
> >
> > @@ -71,31 +89,75 @@ static void off_cpu_finish(void *arg __maybe_unused)
> > off_cpu_bpf__destroy(skel);
> > }
> >
> > -int off_cpu_prepare(struct evlist *evlist)
> > +int off_cpu_prepare(struct evlist *evlist, struct target *target)
> > {
> > - int err;
> > + int err, fd, i;
> > + int ncpus = 1, ntasks = 1;
> >
> > if (off_cpu_config(evlist) < 0) {
> > pr_err("Failed to config off-cpu BPF event\n");
> > return -1;
> > }
> >
> > - set_max_rlimit();
> > -
> > - skel = off_cpu_bpf__open_and_load();
> > + skel = off_cpu_bpf__open();
> > if (!skel) {
> > pr_err("Failed to open off-cpu BPF skeleton\n");
> > return -1;
> > }
> >
> > + /* don't need to set cpu filter for system-wide mode */
> > + if (target->cpu_list) {
> > + ncpus = perf_cpu_map__nr(evlist->core.user_requested_cpus);
> > + bpf_map__set_max_entries(skel->maps.cpu_filter, ncpus);
> > + }
> > +
> > + if (target__has_task(target)) {
> > + ntasks = perf_thread_map__nr(evlist->core.threads);
> > + bpf_map__set_max_entries(skel->maps.task_filter, ntasks);
> > + }
> > +
> > + set_max_rlimit();
> > +
> > + err = off_cpu_bpf__load(skel);
> > + if (err) {
> > + pr_err("Failed to load off-cpu skeleton\n");
> > + goto out;
> > + }
> > +
> > + if (target->cpu_list) {
> > + u32 cpu;
> > + u8 val = 1;
> > +
> > + skel->bss->has_cpu = 1;
> > + fd = bpf_map__fd(skel->maps.cpu_filter);
> > +
> > + for (i = 0; i < ncpus; i++) {
> > + cpu = perf_cpu_map__cpu(evlist->core.user_requested_cpus, i).cpu;
> > + bpf_map_update_elem(fd, &cpu, &val, BPF_ANY);
>
> Perhaps more concise with a for_each:
>
> perf_cpu_map__for_each_cpu(cpu, idx, evlist->core.user_requested_cpus)
> bpf_map_update_elem(fd, &cpu.cpu, &val, BPF_ANY);
Will change.
Thanks,
Namhyung
