tools/perf/util/bpf-loader.c

   1 // SPDX-License-Identifier: GPL-2.0
   2 /*
   3  * bpf-loader.c
   4  *
   5  * Copyright (C) 2015 Wang Nan <wangnan0@huawei.com>
   6  * Copyright (C) 2015 Huawei Inc.
   7  */
   8
   9 #include <linux/bpf.h>
  10 #include <bpf/libbpf.h>
  11 #include <bpf/bpf.h>
  12 #include <linux/err.h>
  13 #include <linux/kernel.h>
  14 #include <linux/string.h>
  15 #include <linux/zalloc.h>
  16 #include <errno.h>
  17 #include <stdlib.h>
  18 #include "debug.h"
  19 #include "evlist.h"
  20 #include "bpf-loader.h"
  21 #include "bpf-prologue.h"
  22 #include "probe-event.h"
  23 #include "probe-finder.h" // for MAX_PROBES
  24 #include "parse-events.h"
  25 #include "strfilter.h"
  26 #include "util.h"
  27 #include "llvm-utils.h"
  28 #include "c++/clang-c.h"
  29
  30 #include <internal/xyarray.h>
  31
  32 static int libbpf_perf_print(enum libbpf_print_level level __attribute__((unused)),
  33                               const char *fmt, va_list args)
  34 {
  35         return veprintf(1, verbose, pr_fmt(fmt), args);
  36 }
  37
  38 struct bpf_prog_priv {
  39         bool is_tp;
  40         char *sys_name;
  41         char *evt_name;
  42         struct perf_probe_event pev;
  43         bool need_prologue;
  44         struct bpf_insn *insns_buf;
  45         int nr_types;
  46         int *type_mapping;
  47 };
  48
  49 static bool libbpf_initialized;
  50
  51 struct bpf_object *
  52 bpf__prepare_load_buffer(void *obj_buf, size_t obj_buf_sz, const char *name)
  53 {
  54         struct bpf_object *obj;
  55
  56         if (!libbpf_initialized) {
  57                 libbpf_set_print(libbpf_perf_print);
  58                 libbpf_initialized = true;
  59         }
  60
  61         obj = bpf_object__open_buffer(obj_buf, obj_buf_sz, name);
  62         if (IS_ERR_OR_NULL(obj)) {
  63                 pr_debug("bpf: failed to load buffer\n");
  64                 return ERR_PTR(-EINVAL);
  65         }
  66
  67         return obj;
  68 }
  69
  70 struct bpf_object *bpf__prepare_load(const char *filename, bool source)
  71 {
  72         struct bpf_object *obj;
  73
  74         if (!libbpf_initialized) {
  75                 libbpf_set_print(libbpf_perf_print);
  76                 libbpf_initialized = true;
  77         }
  78
  79         if (source) {
  80                 int err;
  81                 void *obj_buf;
  82                 size_t obj_buf_sz;
  83
  84                 perf_clang__init();
  85                 err = perf_clang__compile_bpf(filename, &obj_buf, &obj_buf_sz);
  86                 perf_clang__cleanup();
  87                 if (err) {
  88                         pr_debug("bpf: builtin compilation failed: %d, try external compiler\n", err);
  89                         err = llvm__compile_bpf(filename, &obj_buf, &obj_buf_sz);
  90                         if (err)
  91                                 return ERR_PTR(-BPF_LOADER_ERRNO__COMPILE);
  92                 } else
  93                         pr_debug("bpf: successful builtin compilation\n");
  94                 obj = bpf_object__open_buffer(obj_buf, obj_buf_sz, filename);
  95
  96                 if (!IS_ERR_OR_NULL(obj) && llvm_param.dump_obj)
  97                         llvm__dump_obj(filename, obj_buf, obj_buf_sz);
  98
  99                 free(obj_buf);
 100         } else
 101                 obj = bpf_object__open(filename);
 102
 103         if (IS_ERR_OR_NULL(obj)) {
 104                 pr_debug("bpf: failed to load %s\n", filename);
 105                 return obj;
 106         }
 107
 108         return obj;
 109 }
 110
 111 void bpf__clear(void)
 112 {
 113         struct bpf_object *obj, *tmp;
 114
 115         bpf_object__for_each_safe(obj, tmp) {
 116                 bpf__unprobe(obj);
 117                 bpf_object__close(obj);
 118         }
 119 }
 120
 121 static void
 122 clear_prog_priv(struct bpf_program *prog __maybe_unused,
 123                 void *_priv)
 124 {
 125         struct bpf_prog_priv *priv = _priv;
 126
 127         cleanup_perf_probe_events(&priv->pev, 1);
 128         zfree(&priv->insns_buf);
 129         zfree(&priv->type_mapping);
 130         zfree(&priv->sys_name);
 131         zfree(&priv->evt_name);
 132         free(priv);
 133 }
 134
 135 static int
 136 prog_config__exec(const char *value, struct perf_probe_event *pev)
 137 {
 138         pev->uprobes = true;
 139         pev->target = strdup(value);
 140         if (!pev->target)
 141                 return -ENOMEM;
 142         return 0;
 143 }
 144
 145 static int
 146 prog_config__module(const char *value, struct perf_probe_event *pev)
 147 {
 148         pev->uprobes = false;
 149         pev->target = strdup(value);
 150         if (!pev->target)
 151                 return -ENOMEM;
 152         return 0;
 153 }
 154
 155 static int
 156 prog_config__bool(const char *value, bool *pbool, bool invert)
 157 {
 158         int err;
 159         bool bool_value;
 160
 161         if (!pbool)
 162                 return -EINVAL;
 163
 164         err = strtobool(value, &bool_value);
 165         if (err)
 166                 return err;
 167
 168         *pbool = invert ? !bool_value : bool_value;
 169         return 0;
 170 }
 171
 172 static int
 173 prog_config__inlines(const char *value,
 174                      struct perf_probe_event *pev __maybe_unused)
 175 {
 176         return prog_config__bool(value, &probe_conf.no_inlines, true);
 177 }
 178
 179 static int
 180 prog_config__force(const char *value,
 181                    struct perf_probe_event *pev __maybe_unused)
 182 {
 183         return prog_config__bool(value, &probe_conf.force_add, false);
 184 }
 185
 186 static struct {
 187         const char *key;
 188         const char *usage;
 189         const char *desc;
 190         int (*func)(const char *, struct perf_probe_event *);
 191 } bpf_prog_config_terms[] = {
 192         {
 193                 .key    = "exec",
 194                 .usage  = "exec=<full path of file>",
 195                 .desc   = "Set uprobe target",
 196                 .func   = prog_config__exec,
 197         },
 198         {
 199                 .key    = "module",
 200                 .usage  = "module=<module name>    ",
 201                 .desc   = "Set kprobe module",
 202                 .func   = prog_config__module,
 203         },
 204         {
 205                 .key    = "inlines",
 206                 .usage  = "inlines=[yes|no]        ",
 207                 .desc   = "Probe at inline symbol",
 208                 .func   = prog_config__inlines,
 209         },
 210         {
 211                 .key    = "force",
 212                 .usage  = "force=[yes|no]          ",
 213                 .desc   = "Forcibly add events with existing name",
 214                 .func   = prog_config__force,
 215         },
 216 };
 217
 218 static int
 219 do_prog_config(const char *key, const char *value,
 220                struct perf_probe_event *pev)
 221 {
 222         unsigned int i;
 223
 224         pr_debug("config bpf program: %s=%s\n", key, value);
 225         for (i = 0; i < ARRAY_SIZE(bpf_prog_config_terms); i++)
 226                 if (strcmp(key, bpf_prog_config_terms[i].key) == 0)
 227                         return bpf_prog_config_terms[i].func(value, pev);
 228
 229         pr_debug("BPF: ERROR: invalid program config option: %s=%s\n",
 230                  key, value);
 231
 232         pr_debug("\nHint: Valid options are:\n");
 233         for (i = 0; i < ARRAY_SIZE(bpf_prog_config_terms); i++)
 234                 pr_debug("\t%s:\t%s\n", bpf_prog_config_terms[i].usage,
 235                          bpf_prog_config_terms[i].desc);
 236         pr_debug("\n");
 237
 238         return -BPF_LOADER_ERRNO__PROGCONF_TERM;
 239 }
 240
 241 static const char *
 242 parse_prog_config_kvpair(const char *config_str, struct perf_probe_event *pev)
 243 {
 244         char *text = strdup(config_str);
 245         char *sep, *line;
 246         const char *main_str = NULL;
 247         int err = 0;
 248
 249         if (!text) {
 250                 pr_debug("Not enough memory: dup config_str failed\n");
 251                 return ERR_PTR(-ENOMEM);
 252         }
 253
 254         line = text;
 255         while ((sep = strchr(line, ';'))) {
 256                 char *equ;
 257
 258                 *sep = '\0';
 259                 equ = strchr(line, '=');
 260                 if (!equ) {
 261                         pr_warning("WARNING: invalid config in BPF object: %s\n",
 262                                    line);
 263                         pr_warning("\tShould be 'key=value'.\n");
 264                         goto nextline;
 265                 }
 266                 *equ = '\0';
 267
 268                 err = do_prog_config(line, equ + 1, pev);
 269                 if (err)
 270                         break;
 271 nextline:
 272                 line = sep + 1;
 273         }
 274
 275         if (!err)
 276                 main_str = config_str + (line - text);
 277         free(text);
 278
 279         return err ? ERR_PTR(err) : main_str;
 280 }
 281
 282 static int
 283 parse_prog_config(const char *config_str, const char **p_main_str,
 284                   bool *is_tp, struct perf_probe_event *pev)
 285 {
 286         int err;
 287         const char *main_str = parse_prog_config_kvpair(config_str, pev);
 288
 289         if (IS_ERR(main_str))
 290                 return PTR_ERR(main_str);
 291
 292         *p_main_str = main_str;
 293         if (!strchr(main_str, '=')) {
 294                 /* Is a tracepoint event? */
 295                 const char *s = strchr(main_str, ':');
 296
 297                 if (!s) {
 298                         pr_debug("bpf: '%s' is not a valid tracepoint\n",
 299                                  config_str);
 300                         return -BPF_LOADER_ERRNO__CONFIG;
 301                 }
 302
 303                 *is_tp = true;
 304                 return 0;
 305         }
 306
 307         *is_tp = false;
 308         err = parse_perf_probe_command(main_str, pev);
 309         if (err < 0) {
 310                 pr_debug("bpf: '%s' is not a valid config string\n",
 311                          config_str);
 312                 /* parse failed, don't need clear pev. */
 313                 return -BPF_LOADER_ERRNO__CONFIG;
 314         }
 315         return 0;
 316 }
 317
 318 static int
 319 config_bpf_program(struct bpf_program *prog)
 320 {
 321         struct perf_probe_event *pev = NULL;
 322         struct bpf_prog_priv *priv = NULL;
 323         const char *config_str, *main_str;
 324         bool is_tp = false;
 325         int err;
 326
 327         /* Initialize per-program probing setting */
 328         probe_conf.no_inlines = false;
 329         probe_conf.force_add = false;
 330
 331         priv = calloc(sizeof(*priv), 1);
 332         if (!priv) {
 333                 pr_debug("bpf: failed to alloc priv\n");
 334                 return -ENOMEM;
 335         }
 336         pev = &priv->pev;
 337
 338         config_str = bpf_program__section_name(prog);
 339         pr_debug("bpf: config program '%s'\n", config_str);
 340         err = parse_prog_config(config_str, &main_str, &is_tp, pev);
 341         if (err)
 342                 goto errout;
 343
 344         if (is_tp) {
 345                 char *s = strchr(main_str, ':');
 346
 347                 priv->is_tp = true;
 348                 priv->sys_name = strndup(main_str, s - main_str);
 349                 priv->evt_name = strdup(s + 1);
 350                 goto set_priv;
 351         }
 352
 353         if (pev->group && strcmp(pev->group, PERF_BPF_PROBE_GROUP)) {
 354                 pr_debug("bpf: '%s': group for event is set and not '%s'.\n",
 355                          config_str, PERF_BPF_PROBE_GROUP);
 356                 err = -BPF_LOADER_ERRNO__GROUP;
 357                 goto errout;
 358         } else if (!pev->group)
 359                 pev->group = strdup(PERF_BPF_PROBE_GROUP);
 360
 361         if (!pev->group) {
 362                 pr_debug("bpf: strdup failed\n");
 363                 err = -ENOMEM;
 364                 goto errout;
 365         }
 366
 367         if (!pev->event) {
 368                 pr_debug("bpf: '%s': event name is missing. Section name should be 'key=value'\n",
 369                          config_str);
 370                 err = -BPF_LOADER_ERRNO__EVENTNAME;
 371                 goto errout;
 372         }
 373         pr_debug("bpf: config '%s' is ok\n", config_str);
 374
 375 set_priv:
 376         err = bpf_program__set_priv(prog, priv, clear_prog_priv);
 377         if (err) {
 378                 pr_debug("Failed to set priv for program '%s'\n", config_str);
 379                 goto errout;
 380         }
 381
 382         return 0;
 383
 384 errout:
 385         if (pev)
 386                 clear_perf_probe_event(pev);
 387         free(priv);
 388         return err;
 389 }
 390
 391 static int bpf__prepare_probe(void)
 392 {
 393         static int err = 0;
 394         static bool initialized = false;
 395
 396         /*
 397          * Make err static, so if init failed the first, bpf__prepare_probe()
 398          * fails each time without calling init_probe_symbol_maps multiple
 399          * times.
 400          */
 401         if (initialized)
 402                 return err;
 403
 404         initialized = true;
 405         err = init_probe_symbol_maps(false);
 406         if (err < 0)
 407                 pr_debug("Failed to init_probe_symbol_maps\n");
 408         probe_conf.max_probes = MAX_PROBES;
 409         return err;
 410 }
 411
 412 static int
 413 preproc_gen_prologue(struct bpf_program *prog, int n,
 414                      struct bpf_insn *orig_insns, int orig_insns_cnt,
 415                      struct bpf_prog_prep_result *res)
 416 {
 417         struct bpf_prog_priv *priv = bpf_program__priv(prog);
 418         struct probe_trace_event *tev;
 419         struct perf_probe_event *pev;
 420         struct bpf_insn *buf;
 421         size_t prologue_cnt = 0;
 422         int i, err;
 423
 424         if (IS_ERR(priv) || !priv || priv->is_tp)
 425                 goto errout;
 426
 427         pev = &priv->pev;
 428
 429         if (n < 0 || n >= priv->nr_types)
 430                 goto errout;
 431
 432         /* Find a tev belongs to that type */
 433         for (i = 0; i < pev->ntevs; i++) {
 434                 if (priv->type_mapping[i] == n)
 435                         break;
 436         }
 437
 438         if (i >= pev->ntevs) {
 439                 pr_debug("Internal error: prologue type %d not found\n", n);
 440                 return -BPF_LOADER_ERRNO__PROLOGUE;
 441         }
 442
 443         tev = &pev->tevs[i];
 444
 445         buf = priv->insns_buf;
 446         err = bpf__gen_prologue(tev->args, tev->nargs,
 447                                 buf, &prologue_cnt,
 448                                 BPF_MAXINSNS - orig_insns_cnt);
 449         if (err) {
 450                 const char *title;
 451
 452                 title = bpf_program__section_name(prog);
 453                 pr_debug("Failed to generate prologue for program %s\n",
 454                          title);
 455                 return err;
 456         }
 457
 458         memcpy(&buf[prologue_cnt], orig_insns,
 459                sizeof(struct bpf_insn) * orig_insns_cnt);
 460
 461         res->new_insn_ptr = buf;
 462         res->new_insn_cnt = prologue_cnt + orig_insns_cnt;
 463         res->pfd = NULL;
 464         return 0;
 465
 466 errout:
 467         pr_debug("Internal error in preproc_gen_prologue\n");
 468         return -BPF_LOADER_ERRNO__PROLOGUE;
 469 }
 470
 471 /*
 472  * compare_tev_args is reflexive, transitive and antisymmetric.
 473  * I can proof it but this margin is too narrow to contain.
 474  */
 475 static int compare_tev_args(const void *ptev1, const void *ptev2)
 476 {
 477         int i, ret;
 478         const struct probe_trace_event *tev1 =
 479                 *(const struct probe_trace_event **)ptev1;
 480         const struct probe_trace_event *tev2 =
 481                 *(const struct probe_trace_event **)ptev2;
 482
 483         ret = tev2->nargs - tev1->nargs;
 484         if (ret)
 485                 return ret;
 486
 487         for (i = 0; i < tev1->nargs; i++) {
 488                 struct probe_trace_arg *arg1, *arg2;
 489                 struct probe_trace_arg_ref *ref1, *ref2;
 490
 491                 arg1 = &tev1->args[i];
 492                 arg2 = &tev2->args[i];
 493
 494                 ret = strcmp(arg1->value, arg2->value);
 495                 if (ret)
 496                         return ret;
 497
 498                 ref1 = arg1->ref;
 499                 ref2 = arg2->ref;
 500
 501                 while (ref1 && ref2) {
 502                         ret = ref2->offset - ref1->offset;
 503                         if (ret)
 504                                 return ret;
 505
 506                         ref1 = ref1->next;
 507                         ref2 = ref2->next;
 508                 }
 509
 510                 if (ref1 || ref2)
 511                         return ref2 ? 1 : -1;
 512         }
 513
 514         return 0;
 515 }
 516
 517 /*
 518  * Assign a type number to each tevs in a pev.
 519  * mapping is an array with same slots as tevs in that pev.
 520  * nr_types will be set to number of types.
 521  */
 522 static int map_prologue(struct perf_probe_event *pev, int *mapping,
 523                         int *nr_types)
 524 {
 525         int i, type = 0;
 526         struct probe_trace_event **ptevs;
 527
 528         size_t array_sz = sizeof(*ptevs) * pev->ntevs;
 529
 530         ptevs = malloc(array_sz);
 531         if (!ptevs) {
 532                 pr_debug("Not enough memory: alloc ptevs failed\n");
 533                 return -ENOMEM;
 534         }
 535
 536         pr_debug("In map_prologue, ntevs=%d\n", pev->ntevs);
 537         for (i = 0; i < pev->ntevs; i++)
 538                 ptevs[i] = &pev->tevs[i];
 539
 540         qsort(ptevs, pev->ntevs, sizeof(*ptevs),
 541               compare_tev_args);
 542
 543         for (i = 0; i < pev->ntevs; i++) {
 544                 int n;
 545
 546                 n = ptevs[i] - pev->tevs;
 547                 if (i == 0) {
 548                         mapping[n] = type;
 549                         pr_debug("mapping[%d]=%d\n", n, type);
 550                         continue;
 551                 }
 552
 553                 if (compare_tev_args(ptevs + i, ptevs + i - 1) == 0)
 554                         mapping[n] = type;
 555                 else
 556                         mapping[n] = ++type;
 557
 558                 pr_debug("mapping[%d]=%d\n", n, mapping[n]);
 559         }
 560         free(ptevs);
 561         *nr_types = type + 1;
 562
 563         return 0;
 564 }
 565
 566 static int hook_load_preprocessor(struct bpf_program *prog)
 567 {
 568         struct bpf_prog_priv *priv = bpf_program__priv(prog);
 569         struct perf_probe_event *pev;
 570         bool need_prologue = false;
 571         int err, i;
 572
 573         if (IS_ERR(priv) || !priv) {
 574                 pr_debug("Internal error when hook preprocessor\n");
 575                 return -BPF_LOADER_ERRNO__INTERNAL;
 576         }
 577
 578         if (priv->is_tp) {
 579                 priv->need_prologue = false;
 580                 return 0;
 581         }
 582
 583         pev = &priv->pev;
 584         for (i = 0; i < pev->ntevs; i++) {
 585                 struct probe_trace_event *tev = &pev->tevs[i];
 586
 587                 if (tev->nargs > 0) {
 588                         need_prologue = true;
 589                         break;
 590                 }
 591         }
 592
 593         /*
 594          * Since all tevs don't have argument, we don't need generate
 595          * prologue.
 596          */
 597         if (!need_prologue) {
 598                 priv->need_prologue = false;
 599                 return 0;
 600         }
 601
 602         priv->need_prologue = true;
 603         priv->insns_buf = malloc(sizeof(struct bpf_insn) * BPF_MAXINSNS);
 604         if (!priv->insns_buf) {
 605                 pr_debug("Not enough memory: alloc insns_buf failed\n");
 606                 return -ENOMEM;
 607         }
 608
 609         priv->type_mapping = malloc(sizeof(int) * pev->ntevs);
 610         if (!priv->type_mapping) {
 611                 pr_debug("Not enough memory: alloc type_mapping failed\n");
 612                 return -ENOMEM;
 613         }
 614         memset(priv->type_mapping, -1,
 615                sizeof(int) * pev->ntevs);
 616
 617         err = map_prologue(pev, priv->type_mapping, &priv->nr_types);
 618         if (err)
 619                 return err;
 620
 621         err = bpf_program__set_prep(prog, priv->nr_types,
 622                                     preproc_gen_prologue);
 623         return err;
 624 }
 625
 626 int bpf__probe(struct bpf_object *obj)
 627 {
 628         int err = 0;
 629         struct bpf_program *prog;
 630         struct bpf_prog_priv *priv;
 631         struct perf_probe_event *pev;
 632
 633         err = bpf__prepare_probe();
 634         if (err) {
 635                 pr_debug("bpf__prepare_probe failed\n");
 636                 return err;
 637         }
 638
 639         bpf_object__for_each_program(prog, obj) {
 640                 err = config_bpf_program(prog);
 641                 if (err)
 642                         goto out;
 643
 644                 priv = bpf_program__priv(prog);
 645                 if (IS_ERR(priv) || !priv) {
 646                         err = PTR_ERR(priv);
 647                         goto out;
 648                 }
 649
 650                 if (priv->is_tp) {
 651                         bpf_program__set_tracepoint(prog);
 652                         continue;
 653                 }
 654
 655                 bpf_program__set_kprobe(prog);
 656                 pev = &priv->pev;
 657
 658                 err = convert_perf_probe_events(pev, 1);
 659                 if (err < 0) {
 660                         pr_debug("bpf_probe: failed to convert perf probe events\n");
 661                         goto out;
 662                 }
 663
 664                 err = apply_perf_probe_events(pev, 1);
 665                 if (err < 0) {
 666                         pr_debug("bpf_probe: failed to apply perf probe events\n");
 667                         goto out;
 668                 }
 669
 670                 /*
 671                  * After probing, let's consider prologue, which
 672                  * adds program fetcher to BPF programs.
 673                  *
 674                  * hook_load_preprocessorr() hooks pre-processor
 675                  * to bpf_program, let it generate prologue
 676                  * dynamically during loading.
 677                  */
 678                 err = hook_load_preprocessor(prog);
 679                 if (err)
 680                         goto out;
 681         }
 682 out:
 683         return err < 0 ? err : 0;
 684 }
 685
 686 #define EVENTS_WRITE_BUFSIZE  4096
 687 int bpf__unprobe(struct bpf_object *obj)
 688 {
 689         int err, ret = 0;
 690         struct bpf_program *prog;
 691
 692         bpf_object__for_each_program(prog, obj) {
 693                 struct bpf_prog_priv *priv = bpf_program__priv(prog);
 694                 int i;
 695
 696                 if (IS_ERR(priv) || !priv || priv->is_tp)
 697                         continue;
 698
 699                 for (i = 0; i < priv->pev.ntevs; i++) {
 700                         struct probe_trace_event *tev = &priv->pev.tevs[i];
 701                         char name_buf[EVENTS_WRITE_BUFSIZE];
 702                         struct strfilter *delfilter;
 703
 704                         snprintf(name_buf, EVENTS_WRITE_BUFSIZE,
 705                                  "%s:%s", tev->group, tev->event);
 706                         name_buf[EVENTS_WRITE_BUFSIZE - 1] = '\0';
 707
 708                         delfilter = strfilter__new(name_buf, NULL);
 709                         if (!delfilter) {
 710                                 pr_debug("Failed to create filter for unprobing\n");
 711                                 ret = -ENOMEM;
 712                                 continue;
 713                         }
 714
 715                         err = del_perf_probe_events(delfilter);
 716                         strfilter__delete(delfilter);
 717                         if (err) {
 718                                 pr_debug("Failed to delete %s\n", name_buf);
 719                                 ret = err;
 720                                 continue;
 721                         }
 722                 }
 723         }
 724         return ret;
 725 }
 726
 727 int bpf__load(struct bpf_object *obj)
 728 {
 729         int err;
 730
 731         err = bpf_object__load(obj);
 732         if (err) {
 733                 char bf[128];
 734                 libbpf_strerror(err, bf, sizeof(bf));
 735                 pr_debug("bpf: load objects failed: err=%d: (%s)\n", err, bf);
 736                 return err;
 737         }
 738         return 0;
 739 }
 740
 741 int bpf__foreach_event(struct bpf_object *obj,
 742                        bpf_prog_iter_callback_t func,
 743                        void *arg)
 744 {
 745         struct bpf_program *prog;
 746         int err;
 747
 748         bpf_object__for_each_program(prog, obj) {
 749                 struct bpf_prog_priv *priv = bpf_program__priv(prog);
 750                 struct probe_trace_event *tev;
 751                 struct perf_probe_event *pev;
 752                 int i, fd;
 753
 754                 if (IS_ERR(priv) || !priv) {
 755                         pr_debug("bpf: failed to get private field\n");
 756                         return -BPF_LOADER_ERRNO__INTERNAL;
 757                 }
 758
 759                 if (priv->is_tp) {
 760                         fd = bpf_program__fd(prog);
 761                         err = (*func)(priv->sys_name, priv->evt_name, fd, obj, arg);
 762                         if (err) {
 763                                 pr_debug("bpf: tracepoint call back failed, stop iterate\n");
 764                                 return err;
 765                         }
 766                         continue;
 767                 }
 768
 769                 pev = &priv->pev;
 770                 for (i = 0; i < pev->ntevs; i++) {
 771                         tev = &pev->tevs[i];
 772
 773                         if (priv->need_prologue) {
 774                                 int type = priv->type_mapping[i];
 775
 776                                 fd = bpf_program__nth_fd(prog, type);
 777                         } else {
 778                                 fd = bpf_program__fd(prog);
 779                         }
 780
 781                         if (fd < 0) {
 782                                 pr_debug("bpf: failed to get file descriptor\n");
 783                                 return fd;
 784                         }
 785
 786                         err = (*func)(tev->group, tev->event, fd, obj, arg);
 787                         if (err) {
 788                                 pr_debug("bpf: call back failed, stop iterate\n");
 789                                 return err;
 790                         }
 791                 }
 792         }
 793         return 0;
 794 }
 795
 796 enum bpf_map_op_type {
 797         BPF_MAP_OP_SET_VALUE,
 798         BPF_MAP_OP_SET_EVSEL,
 799 };
 800
 801 enum bpf_map_key_type {
 802         BPF_MAP_KEY_ALL,
 803         BPF_MAP_KEY_RANGES,
 804 };
 805
 806 struct bpf_map_op {
 807         struct list_head list;
 808         enum bpf_map_op_type op_type;
 809         enum bpf_map_key_type key_type;
 810         union {
 811                 struct parse_events_array array;
 812         } k;
 813         union {
 814                 u64 value;
 815                 struct evsel *evsel;
 816         } v;
 817 };
 818
 819 struct bpf_map_priv {
 820         struct list_head ops_list;
 821 };
 822
 823 static void
 824 bpf_map_op__delete(struct bpf_map_op *op)
 825 {
 826         if (!list_empty(&op->list))
 827                 list_del_init(&op->list);
 828         if (op->key_type == BPF_MAP_KEY_RANGES)
 829                 parse_events__clear_array(&op->k.array);
 830         free(op);
 831 }
 832
 833 static void
 834 bpf_map_priv__purge(struct bpf_map_priv *priv)
 835 {
 836         struct bpf_map_op *pos, *n;
 837
 838         list_for_each_entry_safe(pos, n, &priv->ops_list, list) {
 839                 list_del_init(&pos->list);
 840                 bpf_map_op__delete(pos);
 841         }
 842 }
 843
 844 static void
 845 bpf_map_priv__clear(struct bpf_map *map __maybe_unused,
 846                     void *_priv)
 847 {
 848         struct bpf_map_priv *priv = _priv;
 849
 850         bpf_map_priv__purge(priv);
 851         free(priv);
 852 }
 853
 854 static int
 855 bpf_map_op_setkey(struct bpf_map_op *op, struct parse_events_term *term)
 856 {
 857         op->key_type = BPF_MAP_KEY_ALL;
 858         if (!term)
 859                 return 0;
 860
 861         if (term->array.nr_ranges) {
 862                 size_t memsz = term->array.nr_ranges *
 863                                 sizeof(op->k.array.ranges[0]);
 864
 865                 op->k.array.ranges = memdup(term->array.ranges, memsz);
 866                 if (!op->k.array.ranges) {
 867                         pr_debug("Not enough memory to alloc indices for map\n");
 868                         return -ENOMEM;
 869                 }
 870                 op->key_type = BPF_MAP_KEY_RANGES;
 871                 op->k.array.nr_ranges = term->array.nr_ranges;
 872         }
 873         return 0;
 874 }
 875
 876 static struct bpf_map_op *
 877 bpf_map_op__new(struct parse_events_term *term)
 878 {
 879         struct bpf_map_op *op;
 880         int err;
 881
 882         op = zalloc(sizeof(*op));
 883         if (!op) {
 884                 pr_debug("Failed to alloc bpf_map_op\n");
 885                 return ERR_PTR(-ENOMEM);
 886         }
 887         INIT_LIST_HEAD(&op->list);
 888
 889         err = bpf_map_op_setkey(op, term);
 890         if (err) {
 891                 free(op);
 892                 return ERR_PTR(err);
 893         }
 894         return op;
 895 }
 896
 897 static struct bpf_map_op *
 898 bpf_map_op__clone(struct bpf_map_op *op)
 899 {
 900         struct bpf_map_op *newop;
 901
 902         newop = memdup(op, sizeof(*op));
 903         if (!newop) {
 904                 pr_debug("Failed to alloc bpf_map_op\n");
 905                 return NULL;
 906         }
 907
 908         INIT_LIST_HEAD(&newop->list);
 909         if (op->key_type == BPF_MAP_KEY_RANGES) {
 910                 size_t memsz = op->k.array.nr_ranges *
 911                                sizeof(op->k.array.ranges[0]);
 912
 913                 newop->k.array.ranges = memdup(op->k.array.ranges, memsz);
 914                 if (!newop->k.array.ranges) {
 915                         pr_debug("Failed to alloc indices for map\n");
 916                         free(newop);
 917                         return NULL;
 918                 }
 919         }
 920
 921         return newop;
 922 }
 923
 924 static struct bpf_map_priv *
 925 bpf_map_priv__clone(struct bpf_map_priv *priv)
 926 {
 927         struct bpf_map_priv *newpriv;
 928         struct bpf_map_op *pos, *newop;
 929
 930         newpriv = zalloc(sizeof(*newpriv));
 931         if (!newpriv) {
 932                 pr_debug("Not enough memory to alloc map private\n");
 933                 return NULL;
 934         }
 935         INIT_LIST_HEAD(&newpriv->ops_list);
 936
 937         list_for_each_entry(pos, &priv->ops_list, list) {
 938                 newop = bpf_map_op__clone(pos);
 939                 if (!newop) {
 940                         bpf_map_priv__purge(newpriv);
 941                         return NULL;
 942                 }
 943                 list_add_tail(&newop->list, &newpriv->ops_list);
 944         }
 945
 946         return newpriv;
 947 }
 948
 949 static int
 950 bpf_map__add_op(struct bpf_map *map, struct bpf_map_op *op)
 951 {
 952         const char *map_name = bpf_map__name(map);
 953         struct bpf_map_priv *priv = bpf_map__priv(map);
 954
 955         if (IS_ERR(priv)) {
 956                 pr_debug("Failed to get private from map %s\n", map_name);
 957                 return PTR_ERR(priv);
 958         }
 959
 960         if (!priv) {
 961                 priv = zalloc(sizeof(*priv));
 962                 if (!priv) {
 963                         pr_debug("Not enough memory to alloc map private\n");
 964                         return -ENOMEM;
 965                 }
 966                 INIT_LIST_HEAD(&priv->ops_list);
 967
 968                 if (bpf_map__set_priv(map, priv, bpf_map_priv__clear)) {
 969                         free(priv);
 970                         return -BPF_LOADER_ERRNO__INTERNAL;
 971                 }
 972         }
 973
 974         list_add_tail(&op->list, &priv->ops_list);
 975         return 0;
 976 }
 977
 978 static struct bpf_map_op *
 979 bpf_map__add_newop(struct bpf_map *map, struct parse_events_term *term)
 980 {
 981         struct bpf_map_op *op;
 982         int err;
 983
 984         op = bpf_map_op__new(term);
 985         if (IS_ERR(op))
 986                 return op;
 987
 988         err = bpf_map__add_op(map, op);
 989         if (err) {
 990                 bpf_map_op__delete(op);
 991                 return ERR_PTR(err);
 992         }
 993         return op;
 994 }
 995
 996 static int
 997 __bpf_map__config_value(struct bpf_map *map,
 998                         struct parse_events_term *term)
 999 {
1000         struct bpf_map_op *op;
1001         const char *map_name = bpf_map__name(map);
1002         const struct bpf_map_def *def = bpf_map__def(map);
1003
1004         if (IS_ERR(def)) {
1005                 pr_debug("Unable to get map definition from '%s'\n",
1006                          map_name);
1007                 return -BPF_LOADER_ERRNO__INTERNAL;
1008         }
1009
1010         if (def->type != BPF_MAP_TYPE_ARRAY) {
1011                 pr_debug("Map %s type is not BPF_MAP_TYPE_ARRAY\n",
1012                          map_name);
1013                 return -BPF_LOADER_ERRNO__OBJCONF_MAP_TYPE;
1014         }
1015         if (def->key_size < sizeof(unsigned int)) {
1016                 pr_debug("Map %s has incorrect key size\n", map_name);
1017                 return -BPF_LOADER_ERRNO__OBJCONF_MAP_KEYSIZE;
1018         }
1019         switch (def->value_size) {
1020         case 1:
1021         case 2:
1022         case 4:
1023         case 8:
1024                 break;
1025         default:
1026                 pr_debug("Map %s has incorrect value size\n", map_name);
1027                 return -BPF_LOADER_ERRNO__OBJCONF_MAP_VALUESIZE;
1028         }
1029
1030         op = bpf_map__add_newop(map, term);
1031         if (IS_ERR(op))
1032                 return PTR_ERR(op);
1033         op->op_type = BPF_MAP_OP_SET_VALUE;
1034         op->v.value = term->val.num;
1035         return 0;
1036 }
1037
1038 static int
1039 bpf_map__config_value(struct bpf_map *map,
1040                       struct parse_events_term *term,
1041                       struct evlist *evlist __maybe_unused)
1042 {
1043         if (!term->err_val) {
1044                 pr_debug("Config value not set\n");
1045                 return -BPF_LOADER_ERRNO__OBJCONF_CONF;
1046         }
1047
1048         if (term->type_val != PARSE_EVENTS__TERM_TYPE_NUM) {
1049                 pr_debug("ERROR: wrong value type for 'value'\n");
1050                 return -BPF_LOADER_ERRNO__OBJCONF_MAP_VALUE;
1051         }
1052
1053         return __bpf_map__config_value(map, term);
1054 }
1055
1056 static int
1057 __bpf_map__config_event(struct bpf_map *map,
1058                         struct parse_events_term *term,
1059                         struct evlist *evlist)
1060 {
1061         struct evsel *evsel;
1062         const struct bpf_map_def *def;
1063         struct bpf_map_op *op;
1064         const char *map_name = bpf_map__name(map);
1065
1066         evsel = perf_evlist__find_evsel_by_str(evlist, term->val.str);
1067         if (!evsel) {
1068                 pr_debug("Event (for '%s') '%s' doesn't exist\n",
1069                          map_name, term->val.str);
1070                 return -BPF_LOADER_ERRNO__OBJCONF_MAP_NOEVT;
1071         }
1072
1073         def = bpf_map__def(map);
1074         if (IS_ERR(def)) {
1075                 pr_debug("Unable to get map definition from '%s'\n",
1076                          map_name);
1077                 return PTR_ERR(def);
1078         }
1079
1080         /*
1081          * No need to check key_size and value_size:
1082          * kernel has already checked them.
1083          */
1084         if (def->type != BPF_MAP_TYPE_PERF_EVENT_ARRAY) {
1085                 pr_debug("Map %s type is not BPF_MAP_TYPE_PERF_EVENT_ARRAY\n",
1086                          map_name);
1087                 return -BPF_LOADER_ERRNO__OBJCONF_MAP_TYPE;
1088         }
1089
1090         op = bpf_map__add_newop(map, term);
1091         if (IS_ERR(op))
1092                 return PTR_ERR(op);
1093         op->op_type = BPF_MAP_OP_SET_EVSEL;
1094         op->v.evsel = evsel;
1095         return 0;
1096 }
1097
1098 static int
1099 bpf_map__config_event(struct bpf_map *map,
1100                       struct parse_events_term *term,
1101                       struct evlist *evlist)
1102 {
1103         if (!term->err_val) {
1104                 pr_debug("Config value not set\n");
1105                 return -BPF_LOADER_ERRNO__OBJCONF_CONF;
1106         }
1107
1108         if (term->type_val != PARSE_EVENTS__TERM_TYPE_STR) {
1109                 pr_debug("ERROR: wrong value type for 'event'\n");
1110                 return -BPF_LOADER_ERRNO__OBJCONF_MAP_VALUE;
1111         }
1112
1113         return __bpf_map__config_event(map, term, evlist);
1114 }
1115
1116 struct bpf_obj_config__map_func {
1117         const char *config_opt;
1118         int (*config_func)(struct bpf_map *, struct parse_events_term *,
1119                            struct evlist *);
1120 };
1121
1122 struct bpf_obj_config__map_func bpf_obj_config__map_funcs[] = {
1123         {"value", bpf_map__config_value},
1124         {"event", bpf_map__config_event},
1125 };
1126
1127 static int
1128 config_map_indices_range_check(struct parse_events_term *term,
1129                                struct bpf_map *map,
1130                                const char *map_name)
1131 {
1132         struct parse_events_array *array = &term->array;
1133         const struct bpf_map_def *def;
1134         unsigned int i;
1135
1136         if (!array->nr_ranges)
1137                 return 0;
1138         if (!array->ranges) {
1139                 pr_debug("ERROR: map %s: array->nr_ranges is %d but range array is NULL\n",
1140                          map_name, (int)array->nr_ranges);
1141                 return -BPF_LOADER_ERRNO__INTERNAL;
1142         }
1143
1144         def = bpf_map__def(map);
1145         if (IS_ERR(def)) {
1146                 pr_debug("ERROR: Unable to get map definition from '%s'\n",
1147                          map_name);
1148                 return -BPF_LOADER_ERRNO__INTERNAL;
1149         }
1150
1151         for (i = 0; i < array->nr_ranges; i++) {
1152                 unsigned int start = array->ranges[i].start;
1153                 size_t length = array->ranges[i].length;
1154                 unsigned int idx = start + length - 1;
1155
1156                 if (idx >= def->max_entries) {
1157                         pr_debug("ERROR: index %d too large\n", idx);
1158                         return -BPF_LOADER_ERRNO__OBJCONF_MAP_IDX2BIG;
1159                 }
1160         }
1161         return 0;
1162 }
1163
1164 static int
1165 bpf__obj_config_map(struct bpf_object *obj,
1166                     struct parse_events_term *term,
1167                     struct evlist *evlist,
1168                     int *key_scan_pos)
1169 {
1170         /* key is "map:<mapname>.<config opt>" */
1171         char *map_name = strdup(term->config + sizeof("map:") - 1);
1172         struct bpf_map *map;
1173         int err = -BPF_LOADER_ERRNO__OBJCONF_OPT;
1174         char *map_opt;
1175         size_t i;
1176
1177         if (!map_name)
1178                 return -ENOMEM;
1179
1180         map_opt = strchr(map_name, '.');
1181         if (!map_opt) {
1182                 pr_debug("ERROR: Invalid map config: %s\n", map_name);
1183                 goto out;
1184         }
1185
1186         *map_opt++ = '\0';
1187         if (*map_opt == '\0') {
1188                 pr_debug("ERROR: Invalid map option: %s\n", term->config);
1189                 goto out;
1190         }
1191
1192         map = bpf_object__find_map_by_name(obj, map_name);
1193         if (!map) {
1194                 pr_debug("ERROR: Map %s doesn't exist\n", map_name);
1195                 err = -BPF_LOADER_ERRNO__OBJCONF_MAP_NOTEXIST;
1196                 goto out;
1197         }
1198
1199         *key_scan_pos += strlen(map_opt);
1200         err = config_map_indices_range_check(term, map, map_name);
1201         if (err)
1202                 goto out;
1203         *key_scan_pos -= strlen(map_opt);
1204
1205         for (i = 0; i < ARRAY_SIZE(bpf_obj_config__map_funcs); i++) {
1206                 struct bpf_obj_config__map_func *func =
1207                                 &bpf_obj_config__map_funcs[i];
1208
1209                 if (strcmp(map_opt, func->config_opt) == 0) {
1210                         err = func->config_func(map, term, evlist);
1211                         goto out;
1212                 }
1213         }
1214
1215         pr_debug("ERROR: Invalid map config option '%s'\n", map_opt);
1216         err = -BPF_LOADER_ERRNO__OBJCONF_MAP_OPT;
1217 out:
1218         free(map_name);
1219         if (!err)
1220                 *key_scan_pos += strlen(map_opt);
1221         return err;
1222 }
1223
1224 int bpf__config_obj(struct bpf_object *obj,
1225                     struct parse_events_term *term,
1226                     struct evlist *evlist,
1227                     int *error_pos)
1228 {
1229         int key_scan_pos = 0;
1230         int err;
1231
1232         if (!obj || !term || !term->config)
1233                 return -EINVAL;
1234
1235         if (strstarts(term->config, "map:")) {
1236                 key_scan_pos = sizeof("map:") - 1;
1237                 err = bpf__obj_config_map(obj, term, evlist, &key_scan_pos);
1238                 goto out;
1239         }
1240         err = -BPF_LOADER_ERRNO__OBJCONF_OPT;
1241 out:
1242         if (error_pos)
1243                 *error_pos = key_scan_pos;
1244         return err;
1245
1246 }
1247
1248 typedef int (*map_config_func_t)(const char *name, int map_fd,
1249                                  const struct bpf_map_def *pdef,
1250                                  struct bpf_map_op *op,
1251                                  void *pkey, void *arg);
1252
1253 static int
1254 foreach_key_array_all(map_config_func_t func,
1255                       void *arg, const char *name,
1256                       int map_fd, const struct bpf_map_def *pdef,
1257                       struct bpf_map_op *op)
1258 {
1259         unsigned int i;
1260         int err;
1261
1262         for (i = 0; i < pdef->max_entries; i++) {
1263                 err = func(name, map_fd, pdef, op, &i, arg);
1264                 if (err) {
1265                         pr_debug("ERROR: failed to insert value to %s[%u]\n",
1266                                  name, i);
1267                         return err;
1268                 }
1269         }
1270         return 0;
1271 }
1272
1273 static int
1274 foreach_key_array_ranges(map_config_func_t func, void *arg,
1275                          const char *name, int map_fd,
1276                          const struct bpf_map_def *pdef,
1277                          struct bpf_map_op *op)
1278 {
1279         unsigned int i, j;
1280         int err;
1281
1282         for (i = 0; i < op->k.array.nr_ranges; i++) {
1283                 unsigned int start = op->k.array.ranges[i].start;
1284                 size_t length = op->k.array.ranges[i].length;
1285
1286                 for (j = 0; j < length; j++) {
1287                         unsigned int idx = start + j;
1288
1289                         err = func(name, map_fd, pdef, op, &idx, arg);
1290                         if (err) {
1291                                 pr_debug("ERROR: failed to insert value to %s[%u]\n",
1292                                          name, idx);
1293                                 return err;
1294                         }
1295                 }
1296         }
1297         return 0;
1298 }
1299
1300 static int
1301 bpf_map_config_foreach_key(struct bpf_map *map,
1302                            map_config_func_t func,
1303                            void *arg)
1304 {
1305         int err, map_fd;
1306         struct bpf_map_op *op;
1307         const struct bpf_map_def *def;
1308         const char *name = bpf_map__name(map);
1309         struct bpf_map_priv *priv = bpf_map__priv(map);
1310
1311         if (IS_ERR(priv)) {
1312                 pr_debug("ERROR: failed to get private from map %s\n", name);
1313                 return -BPF_LOADER_ERRNO__INTERNAL;
1314         }
1315         if (!priv || list_empty(&priv->ops_list)) {
1316                 pr_debug("INFO: nothing to config for map %s\n", name);
1317                 return 0;
1318         }
1319
1320         def = bpf_map__def(map);
1321         if (IS_ERR(def)) {
1322                 pr_debug("ERROR: failed to get definition from map %s\n", name);
1323                 return -BPF_LOADER_ERRNO__INTERNAL;
1324         }
1325         map_fd = bpf_map__fd(map);
1326         if (map_fd < 0) {
1327                 pr_debug("ERROR: failed to get fd from map %s\n", name);
1328                 return map_fd;
1329         }
1330
1331         list_for_each_entry(op, &priv->ops_list, list) {
1332                 switch (def->type) {
1333                 case BPF_MAP_TYPE_ARRAY:
1334                 case BPF_MAP_TYPE_PERF_EVENT_ARRAY:
1335                         switch (op->key_type) {
1336                         case BPF_MAP_KEY_ALL:
1337                                 err = foreach_key_array_all(func, arg, name,
1338                                                             map_fd, def, op);
1339                                 break;
1340                         case BPF_MAP_KEY_RANGES:
1341                                 err = foreach_key_array_ranges(func, arg, name,
1342                                                                map_fd, def,
1343                                                                op);
1344                                 break;
1345                         default:
1346                                 pr_debug("ERROR: keytype for map '%s' invalid\n",
1347                                          name);
1348                                 return -BPF_LOADER_ERRNO__INTERNAL;
1349                         }
1350                         if (err)
1351                                 return err;
1352                         break;
1353                 default:
1354                         pr_debug("ERROR: type of '%s' incorrect\n", name);
1355                         return -BPF_LOADER_ERRNO__OBJCONF_MAP_TYPE;
1356                 }
1357         }
1358
1359         return 0;
1360 }
1361
1362 static int
1363 apply_config_value_for_key(int map_fd, void *pkey,
1364                            size_t val_size, u64 val)
1365 {
1366         int err = 0;
1367
1368         switch (val_size) {
1369         case 1: {
1370                 u8 _val = (u8)(val);
1371                 err = bpf_map_update_elem(map_fd, pkey, &_val, BPF_ANY);
1372                 break;
1373         }
1374         case 2: {
1375                 u16 _val = (u16)(val);
1376                 err = bpf_map_update_elem(map_fd, pkey, &_val, BPF_ANY);
1377                 break;
1378         }
1379         case 4: {
1380                 u32 _val = (u32)(val);
1381                 err = bpf_map_update_elem(map_fd, pkey, &_val, BPF_ANY);
1382                 break;
1383         }
1384         case 8: {
1385                 err = bpf_map_update_elem(map_fd, pkey, &val, BPF_ANY);
1386                 break;
1387         }
1388         default:
1389                 pr_debug("ERROR: invalid value size\n");
1390                 return -BPF_LOADER_ERRNO__OBJCONF_MAP_VALUESIZE;
1391         }
1392         if (err && errno)
1393                 err = -errno;
1394         return err;
1395 }
1396
1397 static int
1398 apply_config_evsel_for_key(const char *name, int map_fd, void *pkey,
1399                            struct evsel *evsel)
1400 {
1401         struct xyarray *xy = evsel->core.fd;
1402         struct perf_event_attr *attr;
1403         unsigned int key, events;
1404         bool check_pass = false;
1405         int *evt_fd;
1406         int err;
1407
1408         if (!xy) {
1409                 pr_debug("ERROR: evsel not ready for map %s\n", name);
1410                 return -BPF_LOADER_ERRNO__INTERNAL;
1411         }
1412
1413         if (xy->row_size / xy->entry_size != 1) {
1414                 pr_debug("ERROR: Dimension of target event is incorrect for map %s\n",
1415                          name);
1416                 return -BPF_LOADER_ERRNO__OBJCONF_MAP_EVTDIM;
1417         }
1418
1419         attr = &evsel->core.attr;
1420         if (attr->inherit) {
1421                 pr_debug("ERROR: Can't put inherit event into map %s\n", name);
1422                 return -BPF_LOADER_ERRNO__OBJCONF_MAP_EVTINH;
1423         }
1424
1425         if (evsel__is_bpf_output(evsel))
1426                 check_pass = true;
1427         if (attr->type == PERF_TYPE_RAW)
1428                 check_pass = true;
1429         if (attr->type == PERF_TYPE_HARDWARE)
1430                 check_pass = true;
1431         if (!check_pass) {
1432                 pr_debug("ERROR: Event type is wrong for map %s\n", name);
1433                 return -BPF_LOADER_ERRNO__OBJCONF_MAP_EVTTYPE;
1434         }
1435
1436         events = xy->entries / (xy->row_size / xy->entry_size);
1437         key = *((unsigned int *)pkey);
1438         if (key >= events) {
1439                 pr_debug("ERROR: there is no event %d for map %s\n",
1440                          key, name);
1441                 return -BPF_LOADER_ERRNO__OBJCONF_MAP_MAPSIZE;
1442         }
1443         evt_fd = xyarray__entry(xy, key, 0);
1444         err = bpf_map_update_elem(map_fd, pkey, evt_fd, BPF_ANY);
1445         if (err && errno)
1446                 err = -errno;
1447         return err;
1448 }
1449
1450 static int
1451 apply_obj_config_map_for_key(const char *name, int map_fd,
1452                              const struct bpf_map_def *pdef,
1453                              struct bpf_map_op *op,
1454                              void *pkey, void *arg __maybe_unused)
1455 {
1456         int err;
1457
1458         switch (op->op_type) {
1459         case BPF_MAP_OP_SET_VALUE:
1460                 err = apply_config_value_for_key(map_fd, pkey,
1461                                                  pdef->value_size,
1462                                                  op->v.value);
1463                 break;
1464         case BPF_MAP_OP_SET_EVSEL:
1465                 err = apply_config_evsel_for_key(name, map_fd, pkey,
1466                                                  op->v.evsel);
1467                 break;
1468         default:
1469                 pr_debug("ERROR: unknown value type for '%s'\n", name);
1470                 err = -BPF_LOADER_ERRNO__INTERNAL;
1471         }
1472         return err;
1473 }
1474
1475 static int
1476 apply_obj_config_map(struct bpf_map *map)
1477 {
1478         return bpf_map_config_foreach_key(map,
1479                                           apply_obj_config_map_for_key,
1480                                           NULL);
1481 }
1482
1483 static int
1484 apply_obj_config_object(struct bpf_object *obj)
1485 {
1486         struct bpf_map *map;
1487         int err;
1488
1489         bpf_object__for_each_map(map, obj) {
1490                 err = apply_obj_config_map(map);
1491                 if (err)
1492                         return err;
1493         }
1494         return 0;
1495 }
1496
1497 int bpf__apply_obj_config(void)
1498 {
1499         struct bpf_object *obj, *tmp;
1500         int err;
1501
1502         bpf_object__for_each_safe(obj, tmp) {
1503                 err = apply_obj_config_object(obj);
1504                 if (err)
1505                         return err;
1506         }
1507
1508         return 0;
1509 }
1510
1511 #define bpf__for_each_map(pos, obj, objtmp)     \
1512         bpf_object__for_each_safe(obj, objtmp)  \
1513                 bpf_object__for_each_map(pos, obj)
1514
1515 #define bpf__for_each_map_named(pos, obj, objtmp, name) \
1516         bpf__for_each_map(pos, obj, objtmp)             \
1517                 if (bpf_map__name(pos) &&               \
1518                         (strcmp(name,                   \
1519                                 bpf_map__name(pos)) == 0))
1520
1521 struct evsel *bpf__setup_output_event(struct evlist *evlist, const char *name)
1522 {
1523         struct bpf_map_priv *tmpl_priv = NULL;
1524         struct bpf_object *obj, *tmp;
1525         struct evsel *evsel = NULL;
1526         struct bpf_map *map;
1527         int err;
1528         bool need_init = false;
1529
1530         bpf__for_each_map_named(map, obj, tmp, name) {
1531                 struct bpf_map_priv *priv = bpf_map__priv(map);
1532
1533                 if (IS_ERR(priv))
1534                         return ERR_PTR(-BPF_LOADER_ERRNO__INTERNAL);
1535
1536                 /*
1537                  * No need to check map type: type should have been
1538                  * verified by kernel.
1539                  */
1540                 if (!need_init && !priv)
1541                         need_init = !priv;
1542                 if (!tmpl_priv && priv)
1543                         tmpl_priv = priv;
1544         }
1545
1546         if (!need_init)
1547                 return NULL;
1548
1549         if (!tmpl_priv) {
1550                 char *event_definition = NULL;
1551
1552                 if (asprintf(&event_definition, "bpf-output/no-inherit=1,name=%s/", name) < 0)
1553                         return ERR_PTR(-ENOMEM);
1554
1555                 err = parse_events(evlist, event_definition, NULL);
1556                 free(event_definition);
1557
1558                 if (err) {
1559                         pr_debug("ERROR: failed to create the \"%s\" bpf-output event\n", name);
1560                         return ERR_PTR(-err);
1561                 }
1562
1563                 evsel = evlist__last(evlist);
1564         }
1565
1566         bpf__for_each_map_named(map, obj, tmp, name) {
1567                 struct bpf_map_priv *priv = bpf_map__priv(map);
1568
1569                 if (IS_ERR(priv))
1570                         return ERR_PTR(-BPF_LOADER_ERRNO__INTERNAL);
1571                 if (priv)
1572                         continue;
1573
1574                 if (tmpl_priv) {
1575                         priv = bpf_map_priv__clone(tmpl_priv);
1576                         if (!priv)
1577                                 return ERR_PTR(-ENOMEM);
1578
1579                         err = bpf_map__set_priv(map, priv, bpf_map_priv__clear);
1580                         if (err) {
1581                                 bpf_map_priv__clear(map, priv);
1582                                 return ERR_PTR(err);
1583                         }
1584                 } else if (evsel) {
1585                         struct bpf_map_op *op;
1586
1587                         op = bpf_map__add_newop(map, NULL);
1588                         if (IS_ERR(op))
1589                                 return ERR_CAST(op);
1590                         op->op_type = BPF_MAP_OP_SET_EVSEL;
1591                         op->v.evsel = evsel;
1592                 }
1593         }
1594
1595         return evsel;
1596 }
1597
1598 int bpf__setup_stdout(struct evlist *evlist)
1599 {
1600         struct evsel *evsel = bpf__setup_output_event(evlist, "__bpf_stdout__");
1601         return PTR_ERR_OR_ZERO(evsel);
1602 }
1603
1604 #define ERRNO_OFFSET(e)         ((e) - __BPF_LOADER_ERRNO__START)
1605 #define ERRCODE_OFFSET(c)       ERRNO_OFFSET(BPF_LOADER_ERRNO__##c)
1606 #define NR_ERRNO        (__BPF_LOADER_ERRNO__END - __BPF_LOADER_ERRNO__START)
1607
1608 static const char *bpf_loader_strerror_table[NR_ERRNO] = {
1609         [ERRCODE_OFFSET(CONFIG)]        = "Invalid config string",
1610         [ERRCODE_OFFSET(GROUP)]         = "Invalid group name",
1611         [ERRCODE_OFFSET(EVENTNAME)]     = "No event name found in config string",
1612         [ERRCODE_OFFSET(INTERNAL)]      = "BPF loader internal error",
1613         [ERRCODE_OFFSET(COMPILE)]       = "Error when compiling BPF scriptlet",
1614         [ERRCODE_OFFSET(PROGCONF_TERM)] = "Invalid program config term in config string",
1615         [ERRCODE_OFFSET(PROLOGUE)]      = "Failed to generate prologue",
1616         [ERRCODE_OFFSET(PROLOGUE2BIG)]  = "Prologue too big for program",
1617         [ERRCODE_OFFSET(PROLOGUEOOB)]   = "Offset out of bound for prologue",
1618         [ERRCODE_OFFSET(OBJCONF_OPT)]   = "Invalid object config option",
1619         [ERRCODE_OFFSET(OBJCONF_CONF)]  = "Config value not set (missing '=')",
1620         [ERRCODE_OFFSET(OBJCONF_MAP_OPT)]       = "Invalid object map config option",
1621         [ERRCODE_OFFSET(OBJCONF_MAP_NOTEXIST)]  = "Target map doesn't exist",
1622         [ERRCODE_OFFSET(OBJCONF_MAP_VALUE)]     = "Incorrect value type for map",
1623         [ERRCODE_OFFSET(OBJCONF_MAP_TYPE)]      = "Incorrect map type",
1624         [ERRCODE_OFFSET(OBJCONF_MAP_KEYSIZE)]   = "Incorrect map key size",
1625         [ERRCODE_OFFSET(OBJCONF_MAP_VALUESIZE)] = "Incorrect map value size",
1626         [ERRCODE_OFFSET(OBJCONF_MAP_NOEVT)]     = "Event not found for map setting",
1627         [ERRCODE_OFFSET(OBJCONF_MAP_MAPSIZE)]   = "Invalid map size for event setting",
1628         [ERRCODE_OFFSET(OBJCONF_MAP_EVTDIM)]    = "Event dimension too large",
1629         [ERRCODE_OFFSET(OBJCONF_MAP_EVTINH)]    = "Doesn't support inherit event",
1630         [ERRCODE_OFFSET(OBJCONF_MAP_EVTTYPE)]   = "Wrong event type for map",
1631         [ERRCODE_OFFSET(OBJCONF_MAP_IDX2BIG)]   = "Index too large",
1632 };
1633
1634 static int
1635 bpf_loader_strerror(int err, char *buf, size_t size)
1636 {
1637         char sbuf[STRERR_BUFSIZE];
1638         const char *msg;
1639
1640         if (!buf || !size)
1641                 return -1;
1642
1643         err = err > 0 ? err : -err;
1644
1645         if (err >= __LIBBPF_ERRNO__START)
1646                 return libbpf_strerror(err, buf, size);
1647
1648         if (err >= __BPF_LOADER_ERRNO__START && err < __BPF_LOADER_ERRNO__END) {
1649                 msg = bpf_loader_strerror_table[ERRNO_OFFSET(err)];
1650                 snprintf(buf, size, "%s", msg);
1651                 buf[size - 1] = '\0';
1652                 return 0;
1653         }
1654
1655         if (err >= __BPF_LOADER_ERRNO__END)
1656                 snprintf(buf, size, "Unknown bpf loader error %d", err);
1657         else
1658                 snprintf(buf, size, "%s",
1659                          str_error_r(err, sbuf, sizeof(sbuf)));
1660
1661         buf[size - 1] = '\0';
1662         return -1;
1663 }
1664
1665 #define bpf__strerror_head(err, buf, size) \
1666         char sbuf[STRERR_BUFSIZE], *emsg;\
1667         if (!size)\
1668                 return 0;\
1669         if (err < 0)\
1670                 err = -err;\
1671         bpf_loader_strerror(err, sbuf, sizeof(sbuf));\
1672         emsg = sbuf;\
1673         switch (err) {\
1674         default:\
1675                 scnprintf(buf, size, "%s", emsg);\
1676                 break;
1677
1678 #define bpf__strerror_entry(val, fmt...)\
1679         case val: {\
1680                 scnprintf(buf, size, fmt);\
1681                 break;\
1682         }
1683
1684 #define bpf__strerror_end(buf, size)\
1685         }\
1686         buf[size - 1] = '\0';
1687
1688 int bpf__strerror_prepare_load(const char *filename, bool source,
1689                                int err, char *buf, size_t size)
1690 {
1691         size_t n;
1692         int ret;
1693
1694         n = snprintf(buf, size, "Failed to load %s%s: ",
1695                          filename, source ? " from source" : "");
1696         if (n >= size) {
1697                 buf[size - 1] = '\0';
1698                 return 0;
1699         }
1700         buf += n;
1701         size -= n;
1702
1703         ret = bpf_loader_strerror(err, buf, size);
1704         buf[size - 1] = '\0';
1705         return ret;
1706 }
1707
1708 int bpf__strerror_probe(struct bpf_object *obj __maybe_unused,
1709                         int err, char *buf, size_t size)
1710 {
1711         bpf__strerror_head(err, buf, size);
1712         case BPF_LOADER_ERRNO__PROGCONF_TERM: {
1713                 scnprintf(buf, size, "%s (add -v to see detail)", emsg);
1714                 break;
1715         }
1716         bpf__strerror_entry(EEXIST, "Probe point exist. Try 'perf probe -d \"*\"' and set 'force=yes'");
1717         bpf__strerror_entry(EACCES, "You need to be root");
1718         bpf__strerror_entry(EPERM, "You need to be root, and /proc/sys/kernel/kptr_restrict should be 0");
1719         bpf__strerror_entry(ENOENT, "You need to check probing points in BPF file");
1720         bpf__strerror_end(buf, size);
1721         return 0;
1722 }
1723
1724 int bpf__strerror_load(struct bpf_object *obj,
1725                        int err, char *buf, size_t size)
1726 {
1727         bpf__strerror_head(err, buf, size);
1728         case LIBBPF_ERRNO__KVER: {
1729                 unsigned int obj_kver = bpf_object__kversion(obj);
1730                 unsigned int real_kver;
1731
1732                 if (fetch_kernel_version(&real_kver, NULL, 0)) {
1733                         scnprintf(buf, size, "Unable to fetch kernel version");
1734                         break;
1735                 }
1736
1737                 if (obj_kver != real_kver) {
1738                         scnprintf(buf, size,
1739                                   "'version' ("KVER_FMT") doesn't match running kernel ("KVER_FMT")",
1740                                   KVER_PARAM(obj_kver),
1741                                   KVER_PARAM(real_kver));
1742                         break;
1743                 }
1744
1745                 scnprintf(buf, size, "Failed to load program for unknown reason");
1746                 break;
1747         }
1748         bpf__strerror_end(buf, size);
1749         return 0;
1750 }
1751
1752 int bpf__strerror_config_obj(struct bpf_object *obj __maybe_unused,
1753                              struct parse_events_term *term __maybe_unused,
1754                              struct evlist *evlist __maybe_unused,
1755                              int *error_pos __maybe_unused, int err,
1756                              char *buf, size_t size)
1757 {
1758         bpf__strerror_head(err, buf, size);
1759         bpf__strerror_entry(BPF_LOADER_ERRNO__OBJCONF_MAP_TYPE,
1760                             "Can't use this config term with this map type");
1761         bpf__strerror_end(buf, size);
1762         return 0;
1763 }
1764
1765 int bpf__strerror_apply_obj_config(int err, char *buf, size_t size)
1766 {
1767         bpf__strerror_head(err, buf, size);
1768         bpf__strerror_entry(BPF_LOADER_ERRNO__OBJCONF_MAP_EVTDIM,
1769                             "Cannot set event to BPF map in multi-thread tracing");
1770         bpf__strerror_entry(BPF_LOADER_ERRNO__OBJCONF_MAP_EVTINH,
1771                             "%s (Hint: use -i to turn off inherit)", emsg);
1772         bpf__strerror_entry(BPF_LOADER_ERRNO__OBJCONF_MAP_EVTTYPE,
1773                             "Can only put raw, hardware and BPF output event into a BPF map");
1774         bpf__strerror_end(buf, size);
1775         return 0;
1776 }
1777
1778 int bpf__strerror_setup_output_event(struct evlist *evlist __maybe_unused,
1779                                      int err, char *buf, size_t size)
1780 {
1781         bpf__strerror_head(err, buf, size);
1782         bpf__strerror_end(buf, size);
1783         return 0;
1784 }