tools/perf/util/probe-event.c

   1 /*
   2  * probe-event.c : perf-probe definition to probe_events format converter
   3  *
   4  * Written by Masami Hiramatsu <mhiramat@redhat.com>
   5  *
   6  * This program is free software; you can redistribute it and/or modify
   7  * it under the terms of the GNU General Public License as published by
   8  * the Free Software Foundation; either version 2 of the License, or
   9  * (at your option) any later version.
  10  *
  11  * This program is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14  * GNU General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU General Public License
  17  * along with this program; if not, write to the Free Software
  18  * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
  19  *
  20  */
  21
  22 #include <inttypes.h>
  23 #include <sys/utsname.h>
  24 #include <sys/types.h>
  25 #include <sys/stat.h>
  26 #include <fcntl.h>
  27 #include <errno.h>
  28 #include <stdio.h>
  29 #include <unistd.h>
  30 #include <stdlib.h>
  31 #include <string.h>
  32 #include <stdarg.h>
  33 #include <limits.h>
  34 #include <elf.h>
  35
  36 #include "util.h"
  37 #include "event.h"
  38 #include "strlist.h"
  39 #include "strfilter.h"
  40 #include "debug.h"
  41 #include "cache.h"
  42 #include "color.h"
  43 #include "symbol.h"
  44 #include "thread.h"
  45 #include <api/fs/fs.h>
  46 #include "trace-event.h"        /* For __maybe_unused */
  47 #include "probe-event.h"
  48 #include "probe-finder.h"
  49 #include "probe-file.h"
  50 #include "session.h"
  51 #include "string2.h"
  52
  53 #include "sane_ctype.h"
  54
  55 #define PERFPROBE_GROUP "probe"
  56
  57 bool probe_event_dry_run;       /* Dry run flag */
  58 struct probe_conf probe_conf;
  59
  60 #define semantic_error(msg ...) pr_err("Semantic error :" msg)
  61
  62 int e_snprintf(char *str, size_t size, const char *format, ...)
  63 {
  64         int ret;
  65         va_list ap;
  66         va_start(ap, format);
  67         ret = vsnprintf(str, size, format, ap);
  68         va_end(ap);
  69         if (ret >= (int)size)
  70                 ret = -E2BIG;
  71         return ret;
  72 }
  73
  74 static struct machine *host_machine;
  75
  76 /* Initialize symbol maps and path of vmlinux/modules */
  77 int init_probe_symbol_maps(bool user_only)
  78 {
  79         int ret;
  80
  81         symbol_conf.sort_by_name = true;
  82         symbol_conf.allow_aliases = true;
  83         ret = symbol__init(NULL);
  84         if (ret < 0) {
  85                 pr_debug("Failed to init symbol map.\n");
  86                 goto out;
  87         }
  88
  89         if (host_machine || user_only)  /* already initialized */
  90                 return 0;
  91
  92         if (symbol_conf.vmlinux_name)
  93                 pr_debug("Use vmlinux: %s\n", symbol_conf.vmlinux_name);
  94
  95         host_machine = machine__new_host();
  96         if (!host_machine) {
  97                 pr_debug("machine__new_host() failed.\n");
  98                 symbol__exit();
  99                 ret = -1;
 100         }
 101 out:
 102         if (ret < 0)
 103                 pr_warning("Failed to init vmlinux path.\n");
 104         return ret;
 105 }
 106
 107 void exit_probe_symbol_maps(void)
 108 {
 109         machine__delete(host_machine);
 110         host_machine = NULL;
 111         symbol__exit();
 112 }
 113
 114 static struct symbol *__find_kernel_function_by_name(const char *name,
 115                                                      struct map **mapp)
 116 {
 117         return machine__find_kernel_function_by_name(host_machine, name, mapp);
 118 }
 119
 120 static struct symbol *__find_kernel_function(u64 addr, struct map **mapp)
 121 {
 122         return machine__find_kernel_function(host_machine, addr, mapp);
 123 }
 124
 125 static struct ref_reloc_sym *kernel_get_ref_reloc_sym(struct map **pmap)
 126 {
 127         /* kmap->ref_reloc_sym should be set if host_machine is initialized */
 128         struct kmap *kmap;
 129         struct map *map = machine__kernel_map(host_machine);
 130
 131         if (map__load(map) < 0)
 132                 return NULL;
 133
 134         kmap = map__kmap(map);
 135         if (!kmap)
 136                 return NULL;
 137
 138         if (pmap)
 139                 *pmap = map;
 140
 141         return kmap->ref_reloc_sym;
 142 }
 143
 144 static int kernel_get_symbol_address_by_name(const char *name, u64 *addr,
 145                                              bool reloc, bool reladdr)
 146 {
 147         struct ref_reloc_sym *reloc_sym;
 148         struct symbol *sym;
 149         struct map *map;
 150
 151         /* ref_reloc_sym is just a label. Need a special fix*/
 152         reloc_sym = kernel_get_ref_reloc_sym(NULL);
 153         if (reloc_sym && strcmp(name, reloc_sym->name) == 0)
 154                 *addr = (reloc) ? reloc_sym->addr : reloc_sym->unrelocated_addr;
 155         else {
 156                 sym = __find_kernel_function_by_name(name, &map);
 157                 if (!sym)
 158                         return -ENOENT;
 159                 *addr = map->unmap_ip(map, sym->start) -
 160                         ((reloc) ? 0 : map->reloc) -
 161                         ((reladdr) ? map->start : 0);
 162         }
 163         return 0;
 164 }
 165
 166 static struct map *kernel_get_module_map(const char *module)
 167 {
 168         struct map_groups *grp = &host_machine->kmaps;
 169         struct maps *maps = &grp->maps[MAP__FUNCTION];
 170         struct map *pos;
 171
 172         /* A file path -- this is an offline module */
 173         if (module && strchr(module, '/'))
 174                 return dso__new_map(module);
 175
 176         if (!module) {
 177                 pos = machine__kernel_map(host_machine);
 178                 return map__get(pos);
 179         }
 180
 181         for (pos = maps__first(maps); pos; pos = map__next(pos)) {
 182                 /* short_name is "[module]" */
 183                 if (strncmp(pos->dso->short_name + 1, module,
 184                             pos->dso->short_name_len - 2) == 0 &&
 185                     module[pos->dso->short_name_len - 2] == '\0') {
 186                         map__get(pos);
 187                         return pos;
 188                 }
 189         }
 190         return NULL;
 191 }
 192
 193 struct map *get_target_map(const char *target, struct nsinfo *nsi, bool user)
 194 {
 195         /* Init maps of given executable or kernel */
 196         if (user) {
 197                 struct map *map;
 198
 199                 map = dso__new_map(target);
 200                 if (map && map->dso) {
 201                         nsinfo__put(map->dso->nsinfo);
 202                         map->dso->nsinfo = nsinfo__get(nsi);
 203                 }
 204                 return map;
 205         } else {
 206                 return kernel_get_module_map(target);
 207         }
 208 }
 209
 210 static int convert_exec_to_group(const char *exec, char **result)
 211 {
 212         char *ptr1, *ptr2, *exec_copy;
 213         char buf[64];
 214         int ret;
 215
 216         exec_copy = strdup(exec);
 217         if (!exec_copy)
 218                 return -ENOMEM;
 219
 220         ptr1 = basename(exec_copy);
 221         if (!ptr1) {
 222                 ret = -EINVAL;
 223                 goto out;
 224         }
 225
 226         for (ptr2 = ptr1; *ptr2 != '\0'; ptr2++) {
 227                 if (!isalnum(*ptr2) && *ptr2 != '_') {
 228                         *ptr2 = '\0';
 229                         break;
 230                 }
 231         }
 232
 233         ret = e_snprintf(buf, 64, "%s_%s", PERFPROBE_GROUP, ptr1);
 234         if (ret < 0)
 235                 goto out;
 236
 237         *result = strdup(buf);
 238         ret = *result ? 0 : -ENOMEM;
 239
 240 out:
 241         free(exec_copy);
 242         return ret;
 243 }
 244
 245 static void clear_perf_probe_point(struct perf_probe_point *pp)
 246 {
 247         free(pp->file);
 248         free(pp->function);
 249         free(pp->lazy_line);
 250 }
 251
 252 static void clear_probe_trace_events(struct probe_trace_event *tevs, int ntevs)
 253 {
 254         int i;
 255
 256         for (i = 0; i < ntevs; i++)
 257                 clear_probe_trace_event(tevs + i);
 258 }
 259
 260 static bool kprobe_blacklist__listed(unsigned long address);
 261 static bool kprobe_warn_out_range(const char *symbol, unsigned long address)
 262 {
 263         u64 etext_addr = 0;
 264         int ret;
 265
 266         /* Get the address of _etext for checking non-probable text symbol */
 267         ret = kernel_get_symbol_address_by_name("_etext", &etext_addr,
 268                                                 false, false);
 269
 270         if (ret == 0 && etext_addr < address)
 271                 pr_warning("%s is out of .text, skip it.\n", symbol);
 272         else if (kprobe_blacklist__listed(address))
 273                 pr_warning("%s is blacklisted function, skip it.\n", symbol);
 274         else
 275                 return false;
 276
 277         return true;
 278 }
 279
 280 /*
 281  * @module can be module name of module file path. In case of path,
 282  * inspect elf and find out what is actual module name.
 283  * Caller has to free mod_name after using it.
 284  */
 285 static char *find_module_name(const char *module)
 286 {
 287         int fd;
 288         Elf *elf;
 289         GElf_Ehdr ehdr;
 290         GElf_Shdr shdr;
 291         Elf_Data *data;
 292         Elf_Scn *sec;
 293         char *mod_name = NULL;
 294         int name_offset;
 295
 296         fd = open(module, O_RDONLY);
 297         if (fd < 0)
 298                 return NULL;
 299
 300         elf = elf_begin(fd, PERF_ELF_C_READ_MMAP, NULL);
 301         if (elf == NULL)
 302                 goto elf_err;
 303
 304         if (gelf_getehdr(elf, &ehdr) == NULL)
 305                 goto ret_err;
 306
 307         sec = elf_section_by_name(elf, &ehdr, &shdr,
 308                         ".gnu.linkonce.this_module", NULL);
 309         if (!sec)
 310                 goto ret_err;
 311
 312         data = elf_getdata(sec, NULL);
 313         if (!data || !data->d_buf)
 314                 goto ret_err;
 315
 316         /*
 317          * NOTE:
 318          * '.gnu.linkonce.this_module' section of kernel module elf directly
 319          * maps to 'struct module' from linux/module.h. This section contains
 320          * actual module name which will be used by kernel after loading it.
 321          * But, we cannot use 'struct module' here since linux/module.h is not
 322          * exposed to user-space. Offset of 'name' has remained same from long
 323          * time, so hardcoding it here.
 324          */
 325         if (ehdr.e_ident[EI_CLASS] == ELFCLASS32)
 326                 name_offset = 12;
 327         else    /* expect ELFCLASS64 by default */
 328                 name_offset = 24;
 329
 330         mod_name = strdup((char *)data->d_buf + name_offset);
 331
 332 ret_err:
 333         elf_end(elf);
 334 elf_err:
 335         close(fd);
 336         return mod_name;
 337 }
 338
 339 #ifdef HAVE_DWARF_SUPPORT
 340
 341 static int kernel_get_module_dso(const char *module, struct dso **pdso)
 342 {
 343         struct dso *dso;
 344         struct map *map;
 345         const char *vmlinux_name;
 346         int ret = 0;
 347
 348         if (module) {
 349                 char module_name[128];
 350
 351                 snprintf(module_name, sizeof(module_name), "[%s]", module);
 352                 map = map_groups__find_by_name(&host_machine->kmaps, MAP__FUNCTION, module_name);
 353                 if (map) {
 354                         dso = map->dso;
 355                         goto found;
 356                 }
 357                 pr_debug("Failed to find module %s.\n", module);
 358                 return -ENOENT;
 359         }
 360
 361         map = machine__kernel_map(host_machine);
 362         dso = map->dso;
 363
 364         vmlinux_name = symbol_conf.vmlinux_name;
 365         dso->load_errno = 0;
 366         if (vmlinux_name)
 367                 ret = dso__load_vmlinux(dso, map, vmlinux_name, false);
 368         else
 369                 ret = dso__load_vmlinux_path(dso, map);
 370 found:
 371         *pdso = dso;
 372         return ret;
 373 }
 374
 375 /*
 376  * Some binaries like glibc have special symbols which are on the symbol
 377  * table, but not in the debuginfo. If we can find the address of the
 378  * symbol from map, we can translate the address back to the probe point.
 379  */
 380 static int find_alternative_probe_point(struct debuginfo *dinfo,
 381                                         struct perf_probe_point *pp,
 382                                         struct perf_probe_point *result,
 383                                         const char *target, struct nsinfo *nsi,
 384                                         bool uprobes)
 385 {
 386         struct map *map = NULL;
 387         struct symbol *sym;
 388         u64 address = 0;
 389         int ret = -ENOENT;
 390
 391         /* This can work only for function-name based one */
 392         if (!pp->function || pp->file)
 393                 return -ENOTSUP;
 394
 395         map = get_target_map(target, nsi, uprobes);
 396         if (!map)
 397                 return -EINVAL;
 398
 399         /* Find the address of given function */
 400         map__for_each_symbol_by_name(map, pp->function, sym) {
 401                 if (uprobes)
 402                         address = sym->start;
 403                 else
 404                         address = map->unmap_ip(map, sym->start) - map->reloc;
 405                 break;
 406         }
 407         if (!address) {
 408                 ret = -ENOENT;
 409                 goto out;
 410         }
 411         pr_debug("Symbol %s address found : %" PRIx64 "\n",
 412                         pp->function, address);
 413
 414         ret = debuginfo__find_probe_point(dinfo, (unsigned long)address,
 415                                           result);
 416         if (ret <= 0)
 417                 ret = (!ret) ? -ENOENT : ret;
 418         else {
 419                 result->offset += pp->offset;
 420                 result->line += pp->line;
 421                 result->retprobe = pp->retprobe;
 422                 ret = 0;
 423         }
 424
 425 out:
 426         map__put(map);
 427         return ret;
 428
 429 }
 430
 431 static int get_alternative_probe_event(struct debuginfo *dinfo,
 432                                        struct perf_probe_event *pev,
 433                                        struct perf_probe_point *tmp)
 434 {
 435         int ret;
 436
 437         memcpy(tmp, &pev->point, sizeof(*tmp));
 438         memset(&pev->point, 0, sizeof(pev->point));
 439         ret = find_alternative_probe_point(dinfo, tmp, &pev->point, pev->target,
 440                                            pev->nsi, pev->uprobes);
 441         if (ret < 0)
 442                 memcpy(&pev->point, tmp, sizeof(*tmp));
 443
 444         return ret;
 445 }
 446
 447 static int get_alternative_line_range(struct debuginfo *dinfo,
 448                                       struct line_range *lr,
 449                                       const char *target, bool user)
 450 {
 451         struct perf_probe_point pp = { .function = lr->function,
 452                                        .file = lr->file,
 453                                        .line = lr->start };
 454         struct perf_probe_point result;
 455         int ret, len = 0;
 456
 457         memset(&result, 0, sizeof(result));
 458
 459         if (lr->end != INT_MAX)
 460                 len = lr->end - lr->start;
 461         ret = find_alternative_probe_point(dinfo, &pp, &result,
 462                                            target, NULL, user);
 463         if (!ret) {
 464                 lr->function = result.function;
 465                 lr->file = result.file;
 466                 lr->start = result.line;
 467                 if (lr->end != INT_MAX)
 468                         lr->end = lr->start + len;
 469                 clear_perf_probe_point(&pp);
 470         }
 471         return ret;
 472 }
 473
 474 /* Open new debuginfo of given module */
 475 static struct debuginfo *open_debuginfo(const char *module, struct nsinfo *nsi,
 476                                         bool silent)
 477 {
 478         const char *path = module;
 479         char reason[STRERR_BUFSIZE];
 480         struct debuginfo *ret = NULL;
 481         struct dso *dso = NULL;
 482         struct nscookie nsc;
 483         int err;
 484
 485         if (!module || !strchr(module, '/')) {
 486                 err = kernel_get_module_dso(module, &dso);
 487                 if (err < 0) {
 488                         if (!dso || dso->load_errno == 0) {
 489                                 if (!str_error_r(-err, reason, STRERR_BUFSIZE))
 490                                         strcpy(reason, "(unknown)");
 491                         } else
 492                                 dso__strerror_load(dso, reason, STRERR_BUFSIZE);
 493                         if (!silent)
 494                                 pr_err("Failed to find the path for %s: %s\n",
 495                                         module ?: "kernel", reason);
 496                         return NULL;
 497                 }
 498                 path = dso->long_name;
 499         }
 500         nsinfo__mountns_enter(nsi, &nsc);
 501         ret = debuginfo__new(path);
 502         if (!ret && !silent) {
 503                 pr_warning("The %s file has no debug information.\n", path);
 504                 if (!module || !strtailcmp(path, ".ko"))
 505                         pr_warning("Rebuild with CONFIG_DEBUG_INFO=y, ");
 506                 else
 507                         pr_warning("Rebuild with -g, ");
 508                 pr_warning("or install an appropriate debuginfo package.\n");
 509         }
 510         nsinfo__mountns_exit(&nsc);
 511         return ret;
 512 }
 513
 514 /* For caching the last debuginfo */
 515 static struct debuginfo *debuginfo_cache;
 516 static char *debuginfo_cache_path;
 517
 518 static struct debuginfo *debuginfo_cache__open(const char *module, bool silent)
 519 {
 520         const char *path = module;
 521
 522         /* If the module is NULL, it should be the kernel. */
 523         if (!module)
 524                 path = "kernel";
 525
 526         if (debuginfo_cache_path && !strcmp(debuginfo_cache_path, path))
 527                 goto out;
 528
 529         /* Copy module path */
 530         free(debuginfo_cache_path);
 531         debuginfo_cache_path = strdup(path);
 532         if (!debuginfo_cache_path) {
 533                 debuginfo__delete(debuginfo_cache);
 534                 debuginfo_cache = NULL;
 535                 goto out;
 536         }
 537
 538         debuginfo_cache = open_debuginfo(module, NULL, silent);
 539         if (!debuginfo_cache)
 540                 zfree(&debuginfo_cache_path);
 541 out:
 542         return debuginfo_cache;
 543 }
 544
 545 static void debuginfo_cache__exit(void)
 546 {
 547         debuginfo__delete(debuginfo_cache);
 548         debuginfo_cache = NULL;
 549         zfree(&debuginfo_cache_path);
 550 }
 551
 552
 553 static int get_text_start_address(const char *exec, unsigned long *address,
 554                                   struct nsinfo *nsi)
 555 {
 556         Elf *elf;
 557         GElf_Ehdr ehdr;
 558         GElf_Shdr shdr;
 559         int fd, ret = -ENOENT;
 560         struct nscookie nsc;
 561
 562         nsinfo__mountns_enter(nsi, &nsc);
 563         fd = open(exec, O_RDONLY);
 564         nsinfo__mountns_exit(&nsc);
 565         if (fd < 0)
 566                 return -errno;
 567
 568         elf = elf_begin(fd, PERF_ELF_C_READ_MMAP, NULL);
 569         if (elf == NULL) {
 570                 ret = -EINVAL;
 571                 goto out_close;
 572         }
 573
 574         if (gelf_getehdr(elf, &ehdr) == NULL)
 575                 goto out;
 576
 577         if (!elf_section_by_name(elf, &ehdr, &shdr, ".text", NULL))
 578                 goto out;
 579
 580         *address = shdr.sh_addr - shdr.sh_offset;
 581         ret = 0;
 582 out:
 583         elf_end(elf);
 584 out_close:
 585         close(fd);
 586
 587         return ret;
 588 }
 589
 590 /*
 591  * Convert trace point to probe point with debuginfo
 592  */
 593 static int find_perf_probe_point_from_dwarf(struct probe_trace_point *tp,
 594                                             struct perf_probe_point *pp,
 595                                             bool is_kprobe)
 596 {
 597         struct debuginfo *dinfo = NULL;
 598         unsigned long stext = 0;
 599         u64 addr = tp->address;
 600         int ret = -ENOENT;
 601
 602         /* convert the address to dwarf address */
 603         if (!is_kprobe) {
 604                 if (!addr) {
 605                         ret = -EINVAL;
 606                         goto error;
 607                 }
 608                 ret = get_text_start_address(tp->module, &stext, NULL);
 609                 if (ret < 0)
 610                         goto error;
 611                 addr += stext;
 612         } else if (tp->symbol) {
 613                 /* If the module is given, this returns relative address */
 614                 ret = kernel_get_symbol_address_by_name(tp->symbol, &addr,
 615                                                         false, !!tp->module);
 616                 if (ret != 0)
 617                         goto error;
 618                 addr += tp->offset;
 619         }
 620
 621         pr_debug("try to find information at %" PRIx64 " in %s\n", addr,
 622                  tp->module ? : "kernel");
 623
 624         dinfo = debuginfo_cache__open(tp->module, verbose <= 0);
 625         if (dinfo)
 626                 ret = debuginfo__find_probe_point(dinfo,
 627                                                  (unsigned long)addr, pp);
 628         else
 629                 ret = -ENOENT;
 630
 631         if (ret > 0) {
 632                 pp->retprobe = tp->retprobe;
 633                 return 0;
 634         }
 635 error:
 636         pr_debug("Failed to find corresponding probes from debuginfo.\n");
 637         return ret ? : -ENOENT;
 638 }
 639
 640 /* Adjust symbol name and address */
 641 static int post_process_probe_trace_point(struct probe_trace_point *tp,
 642                                            struct map *map, unsigned long offs)
 643 {
 644         struct symbol *sym;
 645         u64 addr = tp->address - offs;
 646
 647         sym = map__find_symbol(map, addr);
 648         if (!sym)
 649                 return -ENOENT;
 650
 651         if (strcmp(sym->name, tp->symbol)) {
 652                 /* If we have no realname, use symbol for it */
 653                 if (!tp->realname)
 654                         tp->realname = tp->symbol;
 655                 else
 656                         free(tp->symbol);
 657                 tp->symbol = strdup(sym->name);
 658                 if (!tp->symbol)
 659                         return -ENOMEM;
 660         }
 661         tp->offset = addr - sym->start;
 662         tp->address -= offs;
 663
 664         return 0;
 665 }
 666
 667 /*
 668  * Rename DWARF symbols to ELF symbols -- gcc sometimes optimizes functions
 669  * and generate new symbols with suffixes such as .constprop.N or .isra.N
 670  * etc. Since those symbols are not recorded in DWARF, we have to find
 671  * correct generated symbols from offline ELF binary.
 672  * For online kernel or uprobes we don't need this because those are
 673  * rebased on _text, or already a section relative address.
 674  */
 675 static int
 676 post_process_offline_probe_trace_events(struct probe_trace_event *tevs,
 677                                         int ntevs, const char *pathname)
 678 {
 679         struct map *map;
 680         unsigned long stext = 0;
 681         int i, ret = 0;
 682
 683         /* Prepare a map for offline binary */
 684         map = dso__new_map(pathname);
 685         if (!map || get_text_start_address(pathname, &stext, NULL) < 0) {
 686                 pr_warning("Failed to get ELF symbols for %s\n", pathname);
 687                 return -EINVAL;
 688         }
 689
 690         for (i = 0; i < ntevs; i++) {
 691                 ret = post_process_probe_trace_point(&tevs[i].point,
 692                                                      map, stext);
 693                 if (ret < 0)
 694                         break;
 695         }
 696         map__put(map);
 697
 698         return ret;
 699 }
 700
 701 static int add_exec_to_probe_trace_events(struct probe_trace_event *tevs,
 702                                           int ntevs, const char *exec,
 703                                           struct nsinfo *nsi)
 704 {
 705         int i, ret = 0;
 706         unsigned long stext = 0;
 707
 708         if (!exec)
 709                 return 0;
 710
 711         ret = get_text_start_address(exec, &stext, nsi);
 712         if (ret < 0)
 713                 return ret;
 714
 715         for (i = 0; i < ntevs && ret >= 0; i++) {
 716                 /* point.address is the addres of point.symbol + point.offset */
 717                 tevs[i].point.address -= stext;
 718                 tevs[i].point.module = strdup(exec);
 719                 if (!tevs[i].point.module) {
 720                         ret = -ENOMEM;
 721                         break;
 722                 }
 723                 tevs[i].uprobes = true;
 724         }
 725
 726         return ret;
 727 }
 728
 729 static int
 730 post_process_module_probe_trace_events(struct probe_trace_event *tevs,
 731                                        int ntevs, const char *module,
 732                                        struct debuginfo *dinfo)
 733 {
 734         Dwarf_Addr text_offs = 0;
 735         int i, ret = 0;
 736         char *mod_name = NULL;
 737         struct map *map;
 738
 739         if (!module)
 740                 return 0;
 741
 742         map = get_target_map(module, NULL, false);
 743         if (!map || debuginfo__get_text_offset(dinfo, &text_offs, true) < 0) {
 744                 pr_warning("Failed to get ELF symbols for %s\n", module);
 745                 return -EINVAL;
 746         }
 747
 748         mod_name = find_module_name(module);
 749         for (i = 0; i < ntevs; i++) {
 750                 ret = post_process_probe_trace_point(&tevs[i].point,
 751                                                 map, (unsigned long)text_offs);
 752                 if (ret < 0)
 753                         break;
 754                 tevs[i].point.module =
 755                         strdup(mod_name ? mod_name : module);
 756                 if (!tevs[i].point.module) {
 757                         ret = -ENOMEM;
 758                         break;
 759                 }
 760         }
 761
 762         free(mod_name);
 763         map__put(map);
 764
 765         return ret;
 766 }
 767
 768 static int
 769 post_process_kernel_probe_trace_events(struct probe_trace_event *tevs,
 770                                        int ntevs)
 771 {
 772         struct ref_reloc_sym *reloc_sym;
 773         struct map *map;
 774         char *tmp;
 775         int i, skipped = 0;
 776
 777         /* Skip post process if the target is an offline kernel */
 778         if (symbol_conf.ignore_vmlinux_buildid)
 779                 return post_process_offline_probe_trace_events(tevs, ntevs,
 780                                                 symbol_conf.vmlinux_name);
 781
 782         reloc_sym = kernel_get_ref_reloc_sym(&map);
 783         if (!reloc_sym) {
 784                 pr_warning("Relocated base symbol is not found!\n");
 785                 return -EINVAL;
 786         }
 787
 788         for (i = 0; i < ntevs; i++) {
 789                 if (!tevs[i].point.address)
 790                         continue;
 791                 if (tevs[i].point.retprobe && !kretprobe_offset_is_supported())
 792                         continue;
 793                 /*
 794                  * If we found a wrong one, mark it by NULL symbol.
 795                  * Since addresses in debuginfo is same as objdump, we need
 796                  * to convert it to addresses on memory.
 797                  */
 798                 if (kprobe_warn_out_range(tevs[i].point.symbol,
 799                         map__objdump_2mem(map, tevs[i].point.address))) {
 800                         tmp = NULL;
 801                         skipped++;
 802                 } else {
 803                         tmp = strdup(reloc_sym->name);
 804                         if (!tmp)
 805                                 return -ENOMEM;
 806                 }
 807                 /* If we have no realname, use symbol for it */
 808                 if (!tevs[i].point.realname)
 809                         tevs[i].point.realname = tevs[i].point.symbol;
 810                 else
 811                         free(tevs[i].point.symbol);
 812                 tevs[i].point.symbol = tmp;
 813                 tevs[i].point.offset = tevs[i].point.address -
 814                                        reloc_sym->unrelocated_addr;
 815         }
 816         return skipped;
 817 }
 818
 819 void __weak
 820 arch__post_process_probe_trace_events(struct perf_probe_event *pev __maybe_unused,
 821                                       int ntevs __maybe_unused)
 822 {
 823 }
 824
 825 /* Post processing the probe events */
 826 static int post_process_probe_trace_events(struct perf_probe_event *pev,
 827                                            struct probe_trace_event *tevs,
 828                                            int ntevs, const char *module,
 829                                            bool uprobe, struct debuginfo *dinfo)
 830 {
 831         int ret;
 832
 833         if (uprobe)
 834                 ret = add_exec_to_probe_trace_events(tevs, ntevs, module,
 835                                                      pev->nsi);
 836         else if (module)
 837                 /* Currently ref_reloc_sym based probe is not for drivers */
 838                 ret = post_process_module_probe_trace_events(tevs, ntevs,
 839                                                              module, dinfo);
 840         else
 841                 ret = post_process_kernel_probe_trace_events(tevs, ntevs);
 842
 843         if (ret >= 0)
 844                 arch__post_process_probe_trace_events(pev, ntevs);
 845
 846         return ret;
 847 }
 848
 849 /* Try to find perf_probe_event with debuginfo */
 850 static int try_to_find_probe_trace_events(struct perf_probe_event *pev,
 851                                           struct probe_trace_event **tevs)
 852 {
 853         bool need_dwarf = perf_probe_event_need_dwarf(pev);
 854         struct perf_probe_point tmp;
 855         struct debuginfo *dinfo;
 856         int ntevs, ret = 0;
 857
 858         dinfo = open_debuginfo(pev->target, pev->nsi, !need_dwarf);
 859         if (!dinfo) {
 860                 if (need_dwarf)
 861                         return -ENOENT;
 862                 pr_debug("Could not open debuginfo. Try to use symbols.\n");
 863                 return 0;
 864         }
 865
 866         pr_debug("Try to find probe point from debuginfo.\n");
 867         /* Searching trace events corresponding to a probe event */
 868         ntevs = debuginfo__find_trace_events(dinfo, pev, tevs);
 869
 870         if (ntevs == 0) {  /* Not found, retry with an alternative */
 871                 ret = get_alternative_probe_event(dinfo, pev, &tmp);
 872                 if (!ret) {
 873                         ntevs = debuginfo__find_trace_events(dinfo, pev, tevs);
 874                         /*
 875                          * Write back to the original probe_event for
 876                          * setting appropriate (user given) event name
 877                          */
 878                         clear_perf_probe_point(&pev->point);
 879                         memcpy(&pev->point, &tmp, sizeof(tmp));
 880                 }
 881         }
 882
 883         if (ntevs > 0) {        /* Succeeded to find trace events */
 884                 pr_debug("Found %d probe_trace_events.\n", ntevs);
 885                 ret = post_process_probe_trace_events(pev, *tevs, ntevs,
 886                                         pev->target, pev->uprobes, dinfo);
 887                 if (ret < 0 || ret == ntevs) {
 888                         pr_debug("Post processing failed or all events are skipped. (%d)\n", ret);
 889                         clear_probe_trace_events(*tevs, ntevs);
 890                         zfree(tevs);
 891                         ntevs = 0;
 892                 }
 893         }
 894
 895         debuginfo__delete(dinfo);
 896
 897         if (ntevs == 0) {       /* No error but failed to find probe point. */
 898                 pr_warning("Probe point '%s' not found.\n",
 899                            synthesize_perf_probe_point(&pev->point));
 900                 return -ENOENT;
 901         } else if (ntevs < 0) {
 902                 /* Error path : ntevs < 0 */
 903                 pr_debug("An error occurred in debuginfo analysis (%d).\n", ntevs);
 904                 if (ntevs == -EBADF)
 905                         pr_warning("Warning: No dwarf info found in the vmlinux - "
 906                                 "please rebuild kernel with CONFIG_DEBUG_INFO=y.\n");
 907                 if (!need_dwarf) {
 908                         pr_debug("Trying to use symbols.\n");
 909                         return 0;
 910                 }
 911         }
 912         return ntevs;
 913 }
 914
 915 #define LINEBUF_SIZE 256
 916 #define NR_ADDITIONAL_LINES 2
 917
 918 static int __show_one_line(FILE *fp, int l, bool skip, bool show_num)
 919 {
 920         char buf[LINEBUF_SIZE], sbuf[STRERR_BUFSIZE];
 921         const char *color = show_num ? "" : PERF_COLOR_BLUE;
 922         const char *prefix = NULL;
 923
 924         do {
 925                 if (fgets(buf, LINEBUF_SIZE, fp) == NULL)
 926                         goto error;
 927                 if (skip)
 928                         continue;
 929                 if (!prefix) {
 930                         prefix = show_num ? "%7d  " : "         ";
 931                         color_fprintf(stdout, color, prefix, l);
 932                 }
 933                 color_fprintf(stdout, color, "%s", buf);
 934
 935         } while (strchr(buf, '\n') == NULL);
 936
 937         return 1;
 938 error:
 939         if (ferror(fp)) {
 940                 pr_warning("File read error: %s\n",
 941                            str_error_r(errno, sbuf, sizeof(sbuf)));
 942                 return -1;
 943         }
 944         return 0;
 945 }
 946
 947 static int _show_one_line(FILE *fp, int l, bool skip, bool show_num)
 948 {
 949         int rv = __show_one_line(fp, l, skip, show_num);
 950         if (rv == 0) {
 951                 pr_warning("Source file is shorter than expected.\n");
 952                 rv = -1;
 953         }
 954         return rv;
 955 }
 956
 957 #define show_one_line_with_num(f,l)     _show_one_line(f,l,false,true)
 958 #define show_one_line(f,l)              _show_one_line(f,l,false,false)
 959 #define skip_one_line(f,l)              _show_one_line(f,l,true,false)
 960 #define show_one_line_or_eof(f,l)       __show_one_line(f,l,false,false)
 961
 962 /*
 963  * Show line-range always requires debuginfo to find source file and
 964  * line number.
 965  */
 966 static int __show_line_range(struct line_range *lr, const char *module,
 967                              bool user)
 968 {
 969         int l = 1;
 970         struct int_node *ln;
 971         struct debuginfo *dinfo;
 972         FILE *fp;
 973         int ret;
 974         char *tmp;
 975         char sbuf[STRERR_BUFSIZE];
 976
 977         /* Search a line range */
 978         dinfo = open_debuginfo(module, NULL, false);
 979         if (!dinfo)
 980                 return -ENOENT;
 981
 982         ret = debuginfo__find_line_range(dinfo, lr);
 983         if (!ret) {     /* Not found, retry with an alternative */
 984                 ret = get_alternative_line_range(dinfo, lr, module, user);
 985                 if (!ret)
 986                         ret = debuginfo__find_line_range(dinfo, lr);
 987         }
 988         debuginfo__delete(dinfo);
 989         if (ret == 0 || ret == -ENOENT) {
 990                 pr_warning("Specified source line is not found.\n");
 991                 return -ENOENT;
 992         } else if (ret < 0) {
 993                 pr_warning("Debuginfo analysis failed.\n");
 994                 return ret;
 995         }
 996
 997         /* Convert source file path */
 998         tmp = lr->path;
 999         ret = get_real_path(tmp, lr->comp_dir, &lr->path);
1000
1001         /* Free old path when new path is assigned */
1002         if (tmp != lr->path)
1003                 free(tmp);
1004
1005         if (ret < 0) {
1006                 pr_warning("Failed to find source file path.\n");
1007                 return ret;
1008         }
1009
1010         setup_pager();
1011
1012         if (lr->function)
1013                 fprintf(stdout, "<%s@%s:%d>\n", lr->function, lr->path,
1014                         lr->start - lr->offset);
1015         else
1016                 fprintf(stdout, "<%s:%d>\n", lr->path, lr->start);
1017
1018         fp = fopen(lr->path, "r");
1019         if (fp == NULL) {
1020                 pr_warning("Failed to open %s: %s\n", lr->path,
1021                            str_error_r(errno, sbuf, sizeof(sbuf)));
1022                 return -errno;
1023         }
1024         /* Skip to starting line number */
1025         while (l < lr->start) {
1026                 ret = skip_one_line(fp, l++);
1027                 if (ret < 0)
1028                         goto end;
1029         }
1030
1031         intlist__for_each_entry(ln, lr->line_list) {
1032                 for (; ln->i > l; l++) {
1033                         ret = show_one_line(fp, l - lr->offset);
1034                         if (ret < 0)
1035                                 goto end;
1036                 }
1037                 ret = show_one_line_with_num(fp, l++ - lr->offset);
1038                 if (ret < 0)
1039                         goto end;
1040         }
1041
1042         if (lr->end == INT_MAX)
1043                 lr->end = l + NR_ADDITIONAL_LINES;
1044         while (l <= lr->end) {
1045                 ret = show_one_line_or_eof(fp, l++ - lr->offset);
1046                 if (ret <= 0)
1047                         break;
1048         }
1049 end:
1050         fclose(fp);
1051         return ret;
1052 }
1053
1054 int show_line_range(struct line_range *lr, const char *module,
1055                     struct nsinfo *nsi, bool user)
1056 {
1057         int ret;
1058         struct nscookie nsc;
1059
1060         ret = init_probe_symbol_maps(user);
1061         if (ret < 0)
1062                 return ret;
1063         nsinfo__mountns_enter(nsi, &nsc);
1064         ret = __show_line_range(lr, module, user);
1065         nsinfo__mountns_exit(&nsc);
1066         exit_probe_symbol_maps();
1067
1068         return ret;
1069 }
1070
1071 static int show_available_vars_at(struct debuginfo *dinfo,
1072                                   struct perf_probe_event *pev,
1073                                   struct strfilter *_filter)
1074 {
1075         char *buf;
1076         int ret, i, nvars;
1077         struct str_node *node;
1078         struct variable_list *vls = NULL, *vl;
1079         struct perf_probe_point tmp;
1080         const char *var;
1081
1082         buf = synthesize_perf_probe_point(&pev->point);
1083         if (!buf)
1084                 return -EINVAL;
1085         pr_debug("Searching variables at %s\n", buf);
1086
1087         ret = debuginfo__find_available_vars_at(dinfo, pev, &vls);
1088         if (!ret) {  /* Not found, retry with an alternative */
1089                 ret = get_alternative_probe_event(dinfo, pev, &tmp);
1090                 if (!ret) {
1091                         ret = debuginfo__find_available_vars_at(dinfo, pev,
1092                                                                 &vls);
1093                         /* Release the old probe_point */
1094                         clear_perf_probe_point(&tmp);
1095                 }
1096         }
1097         if (ret <= 0) {
1098                 if (ret == 0 || ret == -ENOENT) {
1099                         pr_err("Failed to find the address of %s\n", buf);
1100                         ret = -ENOENT;
1101                 } else
1102                         pr_warning("Debuginfo analysis failed.\n");
1103                 goto end;
1104         }
1105
1106         /* Some variables are found */
1107         fprintf(stdout, "Available variables at %s\n", buf);
1108         for (i = 0; i < ret; i++) {
1109                 vl = &vls[i];
1110                 /*
1111                  * A probe point might be converted to
1112                  * several trace points.
1113                  */
1114                 fprintf(stdout, "\t@<%s+%lu>\n", vl->point.symbol,
1115                         vl->point.offset);
1116                 zfree(&vl->point.symbol);
1117                 nvars = 0;
1118                 if (vl->vars) {
1119                         strlist__for_each_entry(node, vl->vars) {
1120                                 var = strchr(node->s, '\t') + 1;
1121                                 if (strfilter__compare(_filter, var)) {
1122                                         fprintf(stdout, "\t\t%s\n", node->s);
1123                                         nvars++;
1124                                 }
1125                         }
1126                         strlist__delete(vl->vars);
1127                 }
1128                 if (nvars == 0)
1129                         fprintf(stdout, "\t\t(No matched variables)\n");
1130         }
1131         free(vls);
1132 end:
1133         free(buf);
1134         return ret;
1135 }
1136
1137 /* Show available variables on given probe point */
1138 int show_available_vars(struct perf_probe_event *pevs, int npevs,
1139                         struct strfilter *_filter)
1140 {
1141         int i, ret = 0;
1142         struct debuginfo *dinfo;
1143
1144         ret = init_probe_symbol_maps(pevs->uprobes);
1145         if (ret < 0)
1146                 return ret;
1147
1148         dinfo = open_debuginfo(pevs->target, pevs->nsi, false);
1149         if (!dinfo) {
1150                 ret = -ENOENT;
1151                 goto out;
1152         }
1153
1154         setup_pager();
1155
1156         for (i = 0; i < npevs && ret >= 0; i++)
1157                 ret = show_available_vars_at(dinfo, &pevs[i], _filter);
1158
1159         debuginfo__delete(dinfo);
1160 out:
1161         exit_probe_symbol_maps();
1162         return ret;
1163 }
1164
1165 #else   /* !HAVE_DWARF_SUPPORT */
1166
1167 static void debuginfo_cache__exit(void)
1168 {
1169 }
1170
1171 static int
1172 find_perf_probe_point_from_dwarf(struct probe_trace_point *tp __maybe_unused,
1173                                  struct perf_probe_point *pp __maybe_unused,
1174                                  bool is_kprobe __maybe_unused)
1175 {
1176         return -ENOSYS;
1177 }
1178
1179 static int try_to_find_probe_trace_events(struct perf_probe_event *pev,
1180                                 struct probe_trace_event **tevs __maybe_unused)
1181 {
1182         if (perf_probe_event_need_dwarf(pev)) {
1183                 pr_warning("Debuginfo-analysis is not supported.\n");
1184                 return -ENOSYS;
1185         }
1186
1187         return 0;
1188 }
1189
1190 int show_line_range(struct line_range *lr __maybe_unused,
1191                     const char *module __maybe_unused,
1192                     struct nsinfo *nsi __maybe_unused,
1193                     bool user __maybe_unused)
1194 {
1195         pr_warning("Debuginfo-analysis is not supported.\n");
1196         return -ENOSYS;
1197 }
1198
1199 int show_available_vars(struct perf_probe_event *pevs __maybe_unused,
1200                         int npevs __maybe_unused,
1201                         struct strfilter *filter __maybe_unused)
1202 {
1203         pr_warning("Debuginfo-analysis is not supported.\n");
1204         return -ENOSYS;
1205 }
1206 #endif
1207
1208 void line_range__clear(struct line_range *lr)
1209 {
1210         free(lr->function);
1211         free(lr->file);
1212         free(lr->path);
1213         free(lr->comp_dir);
1214         intlist__delete(lr->line_list);
1215         memset(lr, 0, sizeof(*lr));
1216 }
1217
1218 int line_range__init(struct line_range *lr)
1219 {
1220         memset(lr, 0, sizeof(*lr));
1221         lr->line_list = intlist__new(NULL);
1222         if (!lr->line_list)
1223                 return -ENOMEM;
1224         else
1225                 return 0;
1226 }
1227
1228 static int parse_line_num(char **ptr, int *val, const char *what)
1229 {
1230         const char *start = *ptr;
1231
1232         errno = 0;
1233         *val = strtol(*ptr, ptr, 0);
1234         if (errno || *ptr == start) {
1235                 semantic_error("'%s' is not a valid number.\n", what);
1236                 return -EINVAL;
1237         }
1238         return 0;
1239 }
1240
1241 /* Check the name is good for event, group or function */
1242 static bool is_c_func_name(const char *name)
1243 {
1244         if (!isalpha(*name) && *name != '_')
1245                 return false;
1246         while (*++name != '\0') {
1247                 if (!isalpha(*name) && !isdigit(*name) && *name != '_')
1248                         return false;
1249         }
1250         return true;
1251 }
1252
1253 /*
1254  * Stuff 'lr' according to the line range described by 'arg'.
1255  * The line range syntax is described by:
1256  *
1257  *         SRC[:SLN[+NUM|-ELN]]
1258  *         FNC[@SRC][:SLN[+NUM|-ELN]]
1259  */
1260 int parse_line_range_desc(const char *arg, struct line_range *lr)
1261 {
1262         char *range, *file, *name = strdup(arg);
1263         int err;
1264
1265         if (!name)
1266                 return -ENOMEM;
1267
1268         lr->start = 0;
1269         lr->end = INT_MAX;
1270
1271         range = strchr(name, ':');
1272         if (range) {
1273                 *range++ = '\0';
1274
1275                 err = parse_line_num(&range, &lr->start, "start line");
1276                 if (err)
1277                         goto err;
1278
1279                 if (*range == '+' || *range == '-') {
1280                         const char c = *range++;
1281
1282                         err = parse_line_num(&range, &lr->end, "end line");
1283                         if (err)
1284                                 goto err;
1285
1286                         if (c == '+') {
1287                                 lr->end += lr->start;
1288                                 /*
1289                                  * Adjust the number of lines here.
1290                                  * If the number of lines == 1, the
1291                                  * the end of line should be equal to
1292                                  * the start of line.
1293                                  */
1294                                 lr->end--;
1295                         }
1296                 }
1297
1298                 pr_debug("Line range is %d to %d\n", lr->start, lr->end);
1299
1300                 err = -EINVAL;
1301                 if (lr->start > lr->end) {
1302                         semantic_error("Start line must be smaller"
1303                                        " than end line.\n");
1304                         goto err;
1305                 }
1306                 if (*range != '\0') {
1307                         semantic_error("Tailing with invalid str '%s'.\n", range);
1308                         goto err;
1309                 }
1310         }
1311
1312         file = strchr(name, '@');
1313         if (file) {
1314                 *file = '\0';
1315                 lr->file = strdup(++file);
1316                 if (lr->file == NULL) {
1317                         err = -ENOMEM;
1318                         goto err;
1319                 }
1320                 lr->function = name;
1321         } else if (strchr(name, '/') || strchr(name, '.'))
1322                 lr->file = name;
1323         else if (is_c_func_name(name))/* We reuse it for checking funcname */
1324                 lr->function = name;
1325         else {  /* Invalid name */
1326                 semantic_error("'%s' is not a valid function name.\n", name);
1327                 err = -EINVAL;
1328                 goto err;
1329         }
1330
1331         return 0;
1332 err:
1333         free(name);
1334         return err;
1335 }
1336
1337 static int parse_perf_probe_event_name(char **arg, struct perf_probe_event *pev)
1338 {
1339         char *ptr;
1340
1341         ptr = strchr(*arg, ':');
1342         if (ptr) {
1343                 *ptr = '\0';
1344                 if (!pev->sdt && !is_c_func_name(*arg))
1345                         goto ng_name;
1346                 pev->group = strdup(*arg);
1347                 if (!pev->group)
1348                         return -ENOMEM;
1349                 *arg = ptr + 1;
1350         } else
1351                 pev->group = NULL;
1352         if (!pev->sdt && !is_c_func_name(*arg)) {
1353 ng_name:
1354                 semantic_error("%s is bad for event name -it must "
1355                                "follow C symbol-naming rule.\n", *arg);
1356                 return -EINVAL;
1357         }
1358         pev->event = strdup(*arg);
1359         if (pev->event == NULL)
1360                 return -ENOMEM;
1361
1362         return 0;
1363 }
1364
1365 /* Parse probepoint definition. */
1366 static int parse_perf_probe_point(char *arg, struct perf_probe_event *pev)
1367 {
1368         struct perf_probe_point *pp = &pev->point;
1369         char *ptr, *tmp;
1370         char c, nc = 0;
1371         bool file_spec = false;
1372         int ret;
1373
1374         /*
1375          * <Syntax>
1376          * perf probe [GRP:][EVENT=]SRC[:LN|;PTN]
1377          * perf probe [GRP:][EVENT=]FUNC[@SRC][+OFFS|%return|:LN|;PAT]
1378          * perf probe %[GRP:]SDT_EVENT
1379          */
1380         if (!arg)
1381                 return -EINVAL;
1382
1383         if (is_sdt_event(arg)) {
1384                 pev->sdt = true;
1385                 if (arg[0] == '%')
1386                         arg++;
1387         }
1388
1389         ptr = strpbrk(arg, ";=@+%");
1390         if (pev->sdt) {
1391                 if (ptr) {
1392                         if (*ptr != '@') {
1393                                 semantic_error("%s must be an SDT name.\n",
1394                                                arg);
1395                                 return -EINVAL;
1396                         }
1397                         /* This must be a target file name or build id */
1398                         tmp = build_id_cache__complement(ptr + 1);
1399                         if (tmp) {
1400                                 pev->target = build_id_cache__origname(tmp);
1401                                 free(tmp);
1402                         } else
1403                                 pev->target = strdup(ptr + 1);
1404                         if (!pev->target)
1405                                 return -ENOMEM;
1406                         *ptr = '\0';
1407                 }
1408                 ret = parse_perf_probe_event_name(&arg, pev);
1409                 if (ret == 0) {
1410                         if (asprintf(&pev->point.function, "%%%s", pev->event) < 0)
1411                                 ret = -errno;
1412                 }
1413                 return ret;
1414         }
1415
1416         if (ptr && *ptr == '=') {       /* Event name */
1417                 *ptr = '\0';
1418                 tmp = ptr + 1;
1419                 ret = parse_perf_probe_event_name(&arg, pev);
1420                 if (ret < 0)
1421                         return ret;
1422
1423                 arg = tmp;
1424         }
1425
1426         /*
1427          * Check arg is function or file name and copy it.
1428          *
1429          * We consider arg to be a file spec if and only if it satisfies
1430          * all of the below criteria::
1431          * - it does not include any of "+@%",
1432          * - it includes one of ":;", and
1433          * - it has a period '.' in the name.
1434          *
1435          * Otherwise, we consider arg to be a function specification.
1436          */
1437         if (!strpbrk(arg, "+@%") && (ptr = strpbrk(arg, ";:")) != NULL) {
1438                 /* This is a file spec if it includes a '.' before ; or : */
1439                 if (memchr(arg, '.', ptr - arg))
1440                         file_spec = true;
1441         }
1442
1443         ptr = strpbrk(arg, ";:+@%");
1444         if (ptr) {
1445                 nc = *ptr;
1446                 *ptr++ = '\0';
1447         }
1448
1449         if (arg[0] == '\0')
1450                 tmp = NULL;
1451         else {
1452                 tmp = strdup(arg);
1453                 if (tmp == NULL)
1454                         return -ENOMEM;
1455         }
1456
1457         if (file_spec)
1458                 pp->file = tmp;
1459         else {
1460                 pp->function = tmp;
1461
1462                 /*
1463                  * Keep pp->function even if this is absolute address,
1464                  * so it can mark whether abs_address is valid.
1465                  * Which make 'perf probe lib.bin 0x0' possible.
1466                  *
1467                  * Note that checking length of tmp is not needed
1468                  * because when we access tmp[1] we know tmp[0] is '0',
1469                  * so tmp[1] should always valid (but could be '\0').
1470                  */
1471                 if (tmp && !strncmp(tmp, "0x", 2)) {
1472                         pp->abs_address = strtoul(pp->function, &tmp, 0);
1473                         if (*tmp != '\0') {
1474                                 semantic_error("Invalid absolute address.\n");
1475                                 return -EINVAL;
1476                         }
1477                 }
1478         }
1479
1480         /* Parse other options */
1481         while (ptr) {
1482                 arg = ptr;
1483                 c = nc;
1484                 if (c == ';') { /* Lazy pattern must be the last part */
1485                         pp->lazy_line = strdup(arg);
1486                         if (pp->lazy_line == NULL)
1487                                 return -ENOMEM;
1488                         break;
1489                 }
1490                 ptr = strpbrk(arg, ";:+@%");
1491                 if (ptr) {
1492                         nc = *ptr;
1493                         *ptr++ = '\0';
1494                 }
1495                 switch (c) {
1496                 case ':':       /* Line number */
1497                         pp->line = strtoul(arg, &tmp, 0);
1498                         if (*tmp != '\0') {
1499                                 semantic_error("There is non-digit char"
1500                                                " in line number.\n");
1501                                 return -EINVAL;
1502                         }
1503                         break;
1504                 case '+':       /* Byte offset from a symbol */
1505                         pp->offset = strtoul(arg, &tmp, 0);
1506                         if (*tmp != '\0') {
1507                                 semantic_error("There is non-digit character"
1508                                                 " in offset.\n");
1509                                 return -EINVAL;
1510                         }
1511                         break;
1512                 case '@':       /* File name */
1513                         if (pp->file) {
1514                                 semantic_error("SRC@SRC is not allowed.\n");
1515                                 return -EINVAL;
1516                         }
1517                         pp->file = strdup(arg);
1518                         if (pp->file == NULL)
1519                                 return -ENOMEM;
1520                         break;
1521                 case '%':       /* Probe places */
1522                         if (strcmp(arg, "return") == 0) {
1523                                 pp->retprobe = 1;
1524                         } else {        /* Others not supported yet */
1525                                 semantic_error("%%%s is not supported.\n", arg);
1526                                 return -ENOTSUP;
1527                         }
1528                         break;
1529                 default:        /* Buggy case */
1530                         pr_err("This program has a bug at %s:%d.\n",
1531                                 __FILE__, __LINE__);
1532                         return -ENOTSUP;
1533                         break;
1534                 }
1535         }
1536
1537         /* Exclusion check */
1538         if (pp->lazy_line && pp->line) {
1539                 semantic_error("Lazy pattern can't be used with"
1540                                " line number.\n");
1541                 return -EINVAL;
1542         }
1543
1544         if (pp->lazy_line && pp->offset) {
1545                 semantic_error("Lazy pattern can't be used with offset.\n");
1546                 return -EINVAL;
1547         }
1548
1549         if (pp->line && pp->offset) {
1550                 semantic_error("Offset can't be used with line number.\n");
1551                 return -EINVAL;
1552         }
1553
1554         if (!pp->line && !pp->lazy_line && pp->file && !pp->function) {
1555                 semantic_error("File always requires line number or "
1556                                "lazy pattern.\n");
1557                 return -EINVAL;
1558         }
1559
1560         if (pp->offset && !pp->function) {
1561                 semantic_error("Offset requires an entry function.\n");
1562                 return -EINVAL;
1563         }
1564
1565         if ((pp->offset || pp->line || pp->lazy_line) && pp->retprobe) {
1566                 semantic_error("Offset/Line/Lazy pattern can't be used with "
1567                                "return probe.\n");
1568                 return -EINVAL;
1569         }
1570
1571         pr_debug("symbol:%s file:%s line:%d offset:%lu return:%d lazy:%s\n",
1572                  pp->function, pp->file, pp->line, pp->offset, pp->retprobe,
1573                  pp->lazy_line);
1574         return 0;
1575 }
1576
1577 /* Parse perf-probe event argument */
1578 static int parse_perf_probe_arg(char *str, struct perf_probe_arg *arg)
1579 {
1580         char *tmp, *goodname;
1581         struct perf_probe_arg_field **fieldp;
1582
1583         pr_debug("parsing arg: %s into ", str);
1584
1585         tmp = strchr(str, '=');
1586         if (tmp) {
1587                 arg->name = strndup(str, tmp - str);
1588                 if (arg->name == NULL)
1589                         return -ENOMEM;
1590                 pr_debug("name:%s ", arg->name);
1591                 str = tmp + 1;
1592         }
1593
1594         tmp = strchr(str, ':');
1595         if (tmp) {      /* Type setting */
1596                 *tmp = '\0';
1597                 arg->type = strdup(tmp + 1);
1598                 if (arg->type == NULL)
1599                         return -ENOMEM;
1600                 pr_debug("type:%s ", arg->type);
1601         }
1602
1603         tmp = strpbrk(str, "-.[");
1604         if (!is_c_varname(str) || !tmp) {
1605                 /* A variable, register, symbol or special value */
1606                 arg->var = strdup(str);
1607                 if (arg->var == NULL)
1608                         return -ENOMEM;
1609                 pr_debug("%s\n", arg->var);
1610                 return 0;
1611         }
1612
1613         /* Structure fields or array element */
1614         arg->var = strndup(str, tmp - str);
1615         if (arg->var == NULL)
1616                 return -ENOMEM;
1617         goodname = arg->var;
1618         pr_debug("%s, ", arg->var);
1619         fieldp = &arg->field;
1620
1621         do {
1622                 *fieldp = zalloc(sizeof(struct perf_probe_arg_field));
1623                 if (*fieldp == NULL)
1624                         return -ENOMEM;
1625                 if (*tmp == '[') {      /* Array */
1626                         str = tmp;
1627                         (*fieldp)->index = strtol(str + 1, &tmp, 0);
1628                         (*fieldp)->ref = true;
1629                         if (*tmp != ']' || tmp == str + 1) {
1630                                 semantic_error("Array index must be a"
1631                                                 " number.\n");
1632                                 return -EINVAL;
1633                         }
1634                         tmp++;
1635                         if (*tmp == '\0')
1636                                 tmp = NULL;
1637                 } else {                /* Structure */
1638                         if (*tmp == '.') {
1639                                 str = tmp + 1;
1640                                 (*fieldp)->ref = false;
1641                         } else if (tmp[1] == '>') {
1642                                 str = tmp + 2;
1643                                 (*fieldp)->ref = true;
1644                         } else {
1645                                 semantic_error("Argument parse error: %s\n",
1646                                                str);
1647                                 return -EINVAL;
1648                         }
1649                         tmp = strpbrk(str, "-.[");
1650                 }
1651                 if (tmp) {
1652                         (*fieldp)->name = strndup(str, tmp - str);
1653                         if ((*fieldp)->name == NULL)
1654                                 return -ENOMEM;
1655                         if (*str != '[')
1656                                 goodname = (*fieldp)->name;
1657                         pr_debug("%s(%d), ", (*fieldp)->name, (*fieldp)->ref);
1658                         fieldp = &(*fieldp)->next;
1659                 }
1660         } while (tmp);
1661         (*fieldp)->name = strdup(str);
1662         if ((*fieldp)->name == NULL)
1663                 return -ENOMEM;
1664         if (*str != '[')
1665                 goodname = (*fieldp)->name;
1666         pr_debug("%s(%d)\n", (*fieldp)->name, (*fieldp)->ref);
1667
1668         /* If no name is specified, set the last field name (not array index)*/
1669         if (!arg->name) {
1670                 arg->name = strdup(goodname);
1671                 if (arg->name == NULL)
1672                         return -ENOMEM;
1673         }
1674         return 0;
1675 }
1676
1677 /* Parse perf-probe event command */
1678 int parse_perf_probe_command(const char *cmd, struct perf_probe_event *pev)
1679 {
1680         char **argv;
1681         int argc, i, ret = 0;
1682
1683         argv = argv_split(cmd, &argc);
1684         if (!argv) {
1685                 pr_debug("Failed to split arguments.\n");
1686                 return -ENOMEM;
1687         }
1688         if (argc - 1 > MAX_PROBE_ARGS) {
1689                 semantic_error("Too many probe arguments (%d).\n", argc - 1);
1690                 ret = -ERANGE;
1691                 goto out;
1692         }
1693         /* Parse probe point */
1694         ret = parse_perf_probe_point(argv[0], pev);
1695         if (ret < 0)
1696                 goto out;
1697
1698         /* Copy arguments and ensure return probe has no C argument */
1699         pev->nargs = argc - 1;
1700         pev->args = zalloc(sizeof(struct perf_probe_arg) * pev->nargs);
1701         if (pev->args == NULL) {
1702                 ret = -ENOMEM;
1703                 goto out;
1704         }
1705         for (i = 0; i < pev->nargs && ret >= 0; i++) {
1706                 ret = parse_perf_probe_arg(argv[i + 1], &pev->args[i]);
1707                 if (ret >= 0 &&
1708                     is_c_varname(pev->args[i].var) && pev->point.retprobe) {
1709                         semantic_error("You can't specify local variable for"
1710                                        " kretprobe.\n");
1711                         ret = -EINVAL;
1712                 }
1713         }
1714 out:
1715         argv_free(argv);
1716
1717         return ret;
1718 }
1719
1720 /* Returns true if *any* ARG is either C variable, $params or $vars. */
1721 bool perf_probe_with_var(struct perf_probe_event *pev)
1722 {
1723         int i = 0;
1724
1725         for (i = 0; i < pev->nargs; i++)
1726                 if (is_c_varname(pev->args[i].var)              ||
1727                     !strcmp(pev->args[i].var, PROBE_ARG_PARAMS) ||
1728                     !strcmp(pev->args[i].var, PROBE_ARG_VARS))
1729                         return true;
1730         return false;
1731 }
1732
1733 /* Return true if this perf_probe_event requires debuginfo */
1734 bool perf_probe_event_need_dwarf(struct perf_probe_event *pev)
1735 {
1736         if (pev->point.file || pev->point.line || pev->point.lazy_line)
1737                 return true;
1738
1739         if (perf_probe_with_var(pev))
1740                 return true;
1741
1742         return false;
1743 }
1744
1745 /* Parse probe_events event into struct probe_point */
1746 int parse_probe_trace_command(const char *cmd, struct probe_trace_event *tev)
1747 {
1748         struct probe_trace_point *tp = &tev->point;
1749         char pr;
1750         char *p;
1751         char *argv0_str = NULL, *fmt, *fmt1_str, *fmt2_str, *fmt3_str;
1752         int ret, i, argc;
1753         char **argv;
1754
1755         pr_debug("Parsing probe_events: %s\n", cmd);
1756         argv = argv_split(cmd, &argc);
1757         if (!argv) {
1758                 pr_debug("Failed to split arguments.\n");
1759                 return -ENOMEM;
1760         }
1761         if (argc < 2) {
1762                 semantic_error("Too few probe arguments.\n");
1763                 ret = -ERANGE;
1764                 goto out;
1765         }
1766
1767         /* Scan event and group name. */
1768         argv0_str = strdup(argv[0]);
1769         if (argv0_str == NULL) {
1770                 ret = -ENOMEM;
1771                 goto out;
1772         }
1773         fmt1_str = strtok_r(argv0_str, ":", &fmt);
1774         fmt2_str = strtok_r(NULL, "/", &fmt);
1775         fmt3_str = strtok_r(NULL, " \t", &fmt);
1776         if (fmt1_str == NULL || fmt2_str == NULL || fmt3_str == NULL) {
1777                 semantic_error("Failed to parse event name: %s\n", argv[0]);
1778                 ret = -EINVAL;
1779                 goto out;
1780         }
1781         pr = fmt1_str[0];
1782         tev->group = strdup(fmt2_str);
1783         tev->event = strdup(fmt3_str);
1784         if (tev->group == NULL || tev->event == NULL) {
1785                 ret = -ENOMEM;
1786                 goto out;
1787         }
1788         pr_debug("Group:%s Event:%s probe:%c\n", tev->group, tev->event, pr);
1789
1790         tp->retprobe = (pr == 'r');
1791
1792         /* Scan module name(if there), function name and offset */
1793         p = strchr(argv[1], ':');
1794         if (p) {
1795                 tp->module = strndup(argv[1], p - argv[1]);
1796                 if (!tp->module) {
1797                         ret = -ENOMEM;
1798                         goto out;
1799                 }
1800                 tev->uprobes = (tp->module[0] == '/');
1801                 p++;
1802         } else
1803                 p = argv[1];
1804         fmt1_str = strtok_r(p, "+", &fmt);
1805         /* only the address started with 0x */
1806         if (fmt1_str[0] == '0') {
1807                 /*
1808                  * Fix a special case:
1809                  * if address == 0, kernel reports something like:
1810                  * p:probe_libc/abs_0 /lib/libc-2.18.so:0x          (null) arg1=%ax
1811                  * Newer kernel may fix that, but we want to
1812                  * support old kernel also.
1813                  */
1814                 if (strcmp(fmt1_str, "0x") == 0) {
1815                         if (!argv[2] || strcmp(argv[2], "(null)")) {
1816                                 ret = -EINVAL;
1817                                 goto out;
1818                         }
1819                         tp->address = 0;
1820
1821                         free(argv[2]);
1822                         for (i = 2; argv[i + 1] != NULL; i++)
1823                                 argv[i] = argv[i + 1];
1824
1825                         argv[i] = NULL;
1826                         argc -= 1;
1827                 } else
1828                         tp->address = strtoul(fmt1_str, NULL, 0);
1829         } else {
1830                 /* Only the symbol-based probe has offset */
1831                 tp->symbol = strdup(fmt1_str);
1832                 if (tp->symbol == NULL) {
1833                         ret = -ENOMEM;
1834                         goto out;
1835                 }
1836                 fmt2_str = strtok_r(NULL, "", &fmt);
1837                 if (fmt2_str == NULL)
1838                         tp->offset = 0;
1839                 else
1840                         tp->offset = strtoul(fmt2_str, NULL, 10);
1841         }
1842
1843         tev->nargs = argc - 2;
1844         tev->args = zalloc(sizeof(struct probe_trace_arg) * tev->nargs);
1845         if (tev->args == NULL) {
1846                 ret = -ENOMEM;
1847                 goto out;
1848         }
1849         for (i = 0; i < tev->nargs; i++) {
1850                 p = strchr(argv[i + 2], '=');
1851                 if (p)  /* We don't need which register is assigned. */
1852                         *p++ = '\0';
1853                 else
1854                         p = argv[i + 2];
1855                 tev->args[i].name = strdup(argv[i + 2]);
1856                 /* TODO: parse regs and offset */
1857                 tev->args[i].value = strdup(p);
1858                 if (tev->args[i].name == NULL || tev->args[i].value == NULL) {
1859                         ret = -ENOMEM;
1860                         goto out;
1861                 }
1862         }
1863         ret = 0;
1864 out:
1865         free(argv0_str);
1866         argv_free(argv);
1867         return ret;
1868 }
1869
1870 /* Compose only probe arg */
1871 char *synthesize_perf_probe_arg(struct perf_probe_arg *pa)
1872 {
1873         struct perf_probe_arg_field *field = pa->field;
1874         struct strbuf buf;
1875         char *ret = NULL;
1876         int err;
1877
1878         if (strbuf_init(&buf, 64) < 0)
1879                 return NULL;
1880
1881         if (pa->name && pa->var)
1882                 err = strbuf_addf(&buf, "%s=%s", pa->name, pa->var);
1883         else
1884                 err = strbuf_addstr(&buf, pa->name ?: pa->var);
1885         if (err)
1886                 goto out;
1887
1888         while (field) {
1889                 if (field->name[0] == '[')
1890                         err = strbuf_addstr(&buf, field->name);
1891                 else
1892                         err = strbuf_addf(&buf, "%s%s", field->ref ? "->" : ".",
1893                                           field->name);
1894                 field = field->next;
1895                 if (err)
1896                         goto out;
1897         }
1898
1899         if (pa->type)
1900                 if (strbuf_addf(&buf, ":%s", pa->type) < 0)
1901                         goto out;
1902
1903         ret = strbuf_detach(&buf, NULL);
1904 out:
1905         strbuf_release(&buf);
1906         return ret;
1907 }
1908
1909 /* Compose only probe point (not argument) */
1910 char *synthesize_perf_probe_point(struct perf_probe_point *pp)
1911 {
1912         struct strbuf buf;
1913         char *tmp, *ret = NULL;
1914         int len, err = 0;
1915
1916         if (strbuf_init(&buf, 64) < 0)
1917                 return NULL;
1918
1919         if (pp->function) {
1920                 if (strbuf_addstr(&buf, pp->function) < 0)
1921                         goto out;
1922                 if (pp->offset)
1923                         err = strbuf_addf(&buf, "+%lu", pp->offset);
1924                 else if (pp->line)
1925                         err = strbuf_addf(&buf, ":%d", pp->line);
1926                 else if (pp->retprobe)
1927                         err = strbuf_addstr(&buf, "%return");
1928                 if (err)
1929                         goto out;
1930         }
1931         if (pp->file) {
1932                 tmp = pp->file;
1933                 len = strlen(tmp);
1934                 if (len > 30) {
1935                         tmp = strchr(pp->file + len - 30, '/');
1936                         tmp = tmp ? tmp + 1 : pp->file + len - 30;
1937                 }
1938                 err = strbuf_addf(&buf, "@%s", tmp);
1939                 if (!err && !pp->function && pp->line)
1940                         err = strbuf_addf(&buf, ":%d", pp->line);
1941         }
1942         if (!err)
1943                 ret = strbuf_detach(&buf, NULL);
1944 out:
1945         strbuf_release(&buf);
1946         return ret;
1947 }
1948
1949 char *synthesize_perf_probe_command(struct perf_probe_event *pev)
1950 {
1951         struct strbuf buf;
1952         char *tmp, *ret = NULL;
1953         int i;
1954
1955         if (strbuf_init(&buf, 64))
1956                 return NULL;
1957         if (pev->event)
1958                 if (strbuf_addf(&buf, "%s:%s=", pev->group ?: PERFPROBE_GROUP,
1959                                 pev->event) < 0)
1960                         goto out;
1961
1962         tmp = synthesize_perf_probe_point(&pev->point);
1963         if (!tmp || strbuf_addstr(&buf, tmp) < 0)
1964                 goto out;
1965         free(tmp);
1966
1967         for (i = 0; i < pev->nargs; i++) {
1968                 tmp = synthesize_perf_probe_arg(pev->args + i);
1969                 if (!tmp || strbuf_addf(&buf, " %s", tmp) < 0)
1970                         goto out;
1971                 free(tmp);
1972         }
1973
1974         ret = strbuf_detach(&buf, NULL);
1975 out:
1976         strbuf_release(&buf);
1977         return ret;
1978 }
1979
1980 static int __synthesize_probe_trace_arg_ref(struct probe_trace_arg_ref *ref,
1981                                             struct strbuf *buf, int depth)
1982 {
1983         int err;
1984         if (ref->next) {
1985                 depth = __synthesize_probe_trace_arg_ref(ref->next, buf,
1986                                                          depth + 1);
1987                 if (depth < 0)
1988                         return depth;
1989         }
1990         err = strbuf_addf(buf, "%+ld(", ref->offset);
1991         return (err < 0) ? err : depth;
1992 }
1993
1994 static int synthesize_probe_trace_arg(struct probe_trace_arg *arg,
1995                                       struct strbuf *buf)
1996 {
1997         struct probe_trace_arg_ref *ref = arg->ref;
1998         int depth = 0, err;
1999
2000         /* Argument name or separator */
2001         if (arg->name)
2002                 err = strbuf_addf(buf, " %s=", arg->name);
2003         else
2004                 err = strbuf_addch(buf, ' ');
2005         if (err)
2006                 return err;
2007
2008         /* Special case: @XXX */
2009         if (arg->value[0] == '@' && arg->ref)
2010                         ref = ref->next;
2011
2012         /* Dereferencing arguments */
2013         if (ref) {
2014                 depth = __synthesize_probe_trace_arg_ref(ref, buf, 1);
2015                 if (depth < 0)
2016                         return depth;
2017         }
2018
2019         /* Print argument value */
2020         if (arg->value[0] == '@' && arg->ref)
2021                 err = strbuf_addf(buf, "%s%+ld", arg->value, arg->ref->offset);
2022         else
2023                 err = strbuf_addstr(buf, arg->value);
2024
2025         /* Closing */
2026         while (!err && depth--)
2027                 err = strbuf_addch(buf, ')');
2028
2029         /* Print argument type */
2030         if (!err && arg->type)
2031                 err = strbuf_addf(buf, ":%s", arg->type);
2032
2033         return err;
2034 }
2035
2036 char *synthesize_probe_trace_command(struct probe_trace_event *tev)
2037 {
2038         struct probe_trace_point *tp = &tev->point;
2039         struct strbuf buf;
2040         char *ret = NULL;
2041         int i, err;
2042
2043         /* Uprobes must have tp->module */
2044         if (tev->uprobes && !tp->module)
2045                 return NULL;
2046
2047         if (strbuf_init(&buf, 32) < 0)
2048                 return NULL;
2049
2050         if (strbuf_addf(&buf, "%c:%s/%s ", tp->retprobe ? 'r' : 'p',
2051                         tev->group, tev->event) < 0)
2052                 goto error;
2053         /*
2054          * If tp->address == 0, then this point must be a
2055          * absolute address uprobe.
2056          * try_to_find_absolute_address() should have made
2057          * tp->symbol to "0x0".
2058          */
2059         if (tev->uprobes && !tp->address) {
2060                 if (!tp->symbol || strcmp(tp->symbol, "0x0"))
2061                         goto error;
2062         }
2063
2064         /* Use the tp->address for uprobes */
2065         if (tev->uprobes)
2066                 err = strbuf_addf(&buf, "%s:0x%lx", tp->module, tp->address);
2067         else if (!strncmp(tp->symbol, "0x", 2))
2068                 /* Absolute address. See try_to_find_absolute_address() */
2069                 err = strbuf_addf(&buf, "%s%s0x%lx", tp->module ?: "",
2070                                   tp->module ? ":" : "", tp->address);
2071         else
2072                 err = strbuf_addf(&buf, "%s%s%s+%lu", tp->module ?: "",
2073                                 tp->module ? ":" : "", tp->symbol, tp->offset);
2074         if (err)
2075                 goto error;
2076
2077         for (i = 0; i < tev->nargs; i++)
2078                 if (synthesize_probe_trace_arg(&tev->args[i], &buf) < 0)
2079                         goto error;
2080
2081         ret = strbuf_detach(&buf, NULL);
2082 error:
2083         strbuf_release(&buf);
2084         return ret;
2085 }
2086
2087 static int find_perf_probe_point_from_map(struct probe_trace_point *tp,
2088                                           struct perf_probe_point *pp,
2089                                           bool is_kprobe)
2090 {
2091         struct symbol *sym = NULL;
2092         struct map *map = NULL;
2093         u64 addr = tp->address;
2094         int ret = -ENOENT;
2095
2096         if (!is_kprobe) {
2097                 map = dso__new_map(tp->module);
2098                 if (!map)
2099                         goto out;
2100                 sym = map__find_symbol(map, addr);
2101         } else {
2102                 if (tp->symbol && !addr) {
2103                         if (kernel_get_symbol_address_by_name(tp->symbol,
2104                                                 &addr, true, false) < 0)
2105                                 goto out;
2106                 }
2107                 if (addr) {
2108                         addr += tp->offset;
2109                         sym = __find_kernel_function(addr, &map);
2110                 }
2111         }
2112
2113         if (!sym)
2114                 goto out;
2115
2116         pp->retprobe = tp->retprobe;
2117         pp->offset = addr - map->unmap_ip(map, sym->start);
2118         pp->function = strdup(sym->name);
2119         ret = pp->function ? 0 : -ENOMEM;
2120
2121 out:
2122         if (map && !is_kprobe) {
2123                 map__put(map);
2124         }
2125
2126         return ret;
2127 }
2128
2129 static int convert_to_perf_probe_point(struct probe_trace_point *tp,
2130                                        struct perf_probe_point *pp,
2131                                        bool is_kprobe)
2132 {
2133         char buf[128];
2134         int ret;
2135
2136         ret = find_perf_probe_point_from_dwarf(tp, pp, is_kprobe);
2137         if (!ret)
2138                 return 0;
2139         ret = find_perf_probe_point_from_map(tp, pp, is_kprobe);
2140         if (!ret)
2141                 return 0;
2142
2143         pr_debug("Failed to find probe point from both of dwarf and map.\n");
2144
2145         if (tp->symbol) {
2146                 pp->function = strdup(tp->symbol);
2147                 pp->offset = tp->offset;
2148         } else {
2149                 ret = e_snprintf(buf, 128, "0x%" PRIx64, (u64)tp->address);
2150                 if (ret < 0)
2151                         return ret;
2152                 pp->function = strdup(buf);
2153                 pp->offset = 0;
2154         }
2155         if (pp->function == NULL)
2156                 return -ENOMEM;
2157
2158         pp->retprobe = tp->retprobe;
2159
2160         return 0;
2161 }
2162
2163 static int convert_to_perf_probe_event(struct probe_trace_event *tev,
2164                                struct perf_probe_event *pev, bool is_kprobe)
2165 {
2166         struct strbuf buf = STRBUF_INIT;
2167         int i, ret;
2168
2169         /* Convert event/group name */
2170         pev->event = strdup(tev->event);
2171         pev->group = strdup(tev->group);
2172         if (pev->event == NULL || pev->group == NULL)
2173                 return -ENOMEM;
2174
2175         /* Convert trace_point to probe_point */
2176         ret = convert_to_perf_probe_point(&tev->point, &pev->point, is_kprobe);
2177         if (ret < 0)
2178                 return ret;
2179
2180         /* Convert trace_arg to probe_arg */
2181         pev->nargs = tev->nargs;
2182         pev->args = zalloc(sizeof(struct perf_probe_arg) * pev->nargs);
2183         if (pev->args == NULL)
2184                 return -ENOMEM;
2185         for (i = 0; i < tev->nargs && ret >= 0; i++) {
2186                 if (tev->args[i].name)
2187                         pev->args[i].name = strdup(tev->args[i].name);
2188                 else {
2189                         if ((ret = strbuf_init(&buf, 32)) < 0)
2190                                 goto error;
2191                         ret = synthesize_probe_trace_arg(&tev->args[i], &buf);
2192                         pev->args[i].name = strbuf_detach(&buf, NULL);
2193                 }
2194                 if (pev->args[i].name == NULL && ret >= 0)
2195                         ret = -ENOMEM;
2196         }
2197 error:
2198         if (ret < 0)
2199                 clear_perf_probe_event(pev);
2200
2201         return ret;
2202 }
2203
2204 void clear_perf_probe_event(struct perf_probe_event *pev)
2205 {
2206         struct perf_probe_arg_field *field, *next;
2207         int i;
2208
2209         free(pev->event);
2210         free(pev->group);
2211         free(pev->target);
2212         clear_perf_probe_point(&pev->point);
2213
2214         for (i = 0; i < pev->nargs; i++) {
2215                 free(pev->args[i].name);
2216                 free(pev->args[i].var);
2217                 free(pev->args[i].type);
2218                 field = pev->args[i].field;
2219                 while (field) {
2220                         next = field->next;
2221                         zfree(&field->name);
2222                         free(field);
2223                         field = next;
2224                 }
2225         }
2226         free(pev->args);
2227         memset(pev, 0, sizeof(*pev));
2228 }
2229
2230 #define strdup_or_goto(str, label)      \
2231 ({ char *__p = NULL; if (str && !(__p = strdup(str))) goto label; __p; })
2232
2233 static int perf_probe_point__copy(struct perf_probe_point *dst,
2234                                   struct perf_probe_point *src)
2235 {
2236         dst->file = strdup_or_goto(src->file, out_err);
2237         dst->function = strdup_or_goto(src->function, out_err);
2238         dst->lazy_line = strdup_or_goto(src->lazy_line, out_err);
2239         dst->line = src->line;
2240         dst->retprobe = src->retprobe;
2241         dst->offset = src->offset;
2242         return 0;
2243
2244 out_err:
2245         clear_perf_probe_point(dst);
2246         return -ENOMEM;
2247 }
2248
2249 static int perf_probe_arg__copy(struct perf_probe_arg *dst,
2250                                 struct perf_probe_arg *src)
2251 {
2252         struct perf_probe_arg_field *field, **ppfield;
2253
2254         dst->name = strdup_or_goto(src->name, out_err);
2255         dst->var = strdup_or_goto(src->var, out_err);
2256         dst->type = strdup_or_goto(src->type, out_err);
2257
2258         field = src->field;
2259         ppfield = &(dst->field);
2260         while (field) {
2261                 *ppfield = zalloc(sizeof(*field));
2262                 if (!*ppfield)
2263                         goto out_err;
2264                 (*ppfield)->name = strdup_or_goto(field->name, out_err);
2265                 (*ppfield)->index = field->index;
2266                 (*ppfield)->ref = field->ref;
2267                 field = field->next;
2268                 ppfield = &((*ppfield)->next);
2269         }
2270         return 0;
2271 out_err:
2272         return -ENOMEM;
2273 }
2274
2275 int perf_probe_event__copy(struct perf_probe_event *dst,
2276                            struct perf_probe_event *src)
2277 {
2278         int i;
2279
2280         dst->event = strdup_or_goto(src->event, out_err);
2281         dst->group = strdup_or_goto(src->group, out_err);
2282         dst->target = strdup_or_goto(src->target, out_err);
2283         dst->uprobes = src->uprobes;
2284
2285         if (perf_probe_point__copy(&dst->point, &src->point) < 0)
2286                 goto out_err;
2287
2288         dst->args = zalloc(sizeof(struct perf_probe_arg) * src->nargs);
2289         if (!dst->args)
2290                 goto out_err;
2291         dst->nargs = src->nargs;
2292
2293         for (i = 0; i < src->nargs; i++)
2294                 if (perf_probe_arg__copy(&dst->args[i], &src->args[i]) < 0)
2295                         goto out_err;
2296         return 0;
2297
2298 out_err:
2299         clear_perf_probe_event(dst);
2300         return -ENOMEM;
2301 }
2302
2303 void clear_probe_trace_event(struct probe_trace_event *tev)
2304 {
2305         struct probe_trace_arg_ref *ref, *next;
2306         int i;
2307
2308         free(tev->event);
2309         free(tev->group);
2310         free(tev->point.symbol);
2311         free(tev->point.realname);
2312         free(tev->point.module);
2313         for (i = 0; i < tev->nargs; i++) {
2314                 free(tev->args[i].name);
2315                 free(tev->args[i].value);
2316                 free(tev->args[i].type);
2317                 ref = tev->args[i].ref;
2318                 while (ref) {
2319                         next = ref->next;
2320                         free(ref);
2321                         ref = next;
2322                 }
2323         }
2324         free(tev->args);
2325         memset(tev, 0, sizeof(*tev));
2326 }
2327
2328 struct kprobe_blacklist_node {
2329         struct list_head list;
2330         unsigned long start;
2331         unsigned long end;
2332         char *symbol;
2333 };
2334
2335 static void kprobe_blacklist__delete(struct list_head *blacklist)
2336 {
2337         struct kprobe_blacklist_node *node;
2338
2339         while (!list_empty(blacklist)) {
2340                 node = list_first_entry(blacklist,
2341                                         struct kprobe_blacklist_node, list);
2342                 list_del(&node->list);
2343                 free(node->symbol);
2344                 free(node);
2345         }
2346 }
2347
2348 static int kprobe_blacklist__load(struct list_head *blacklist)
2349 {
2350         struct kprobe_blacklist_node *node;
2351         const char *__debugfs = debugfs__mountpoint();
2352         char buf[PATH_MAX], *p;
2353         FILE *fp;
2354         int ret;
2355
2356         if (__debugfs == NULL)
2357                 return -ENOTSUP;
2358
2359         ret = e_snprintf(buf, PATH_MAX, "%s/kprobes/blacklist", __debugfs);
2360         if (ret < 0)
2361                 return ret;
2362
2363         fp = fopen(buf, "r");
2364         if (!fp)
2365                 return -errno;
2366
2367         ret = 0;
2368         while (fgets(buf, PATH_MAX, fp)) {
2369                 node = zalloc(sizeof(*node));
2370                 if (!node) {
2371                         ret = -ENOMEM;
2372                         break;
2373                 }
2374                 INIT_LIST_HEAD(&node->list);
2375                 list_add_tail(&node->list, blacklist);
2376                 if (sscanf(buf, "0x%lx-0x%lx", &node->start, &node->end) != 2) {
2377                         ret = -EINVAL;
2378                         break;
2379                 }
2380                 p = strchr(buf, '\t');
2381                 if (p) {
2382                         p++;
2383                         if (p[strlen(p) - 1] == '\n')
2384                                 p[strlen(p) - 1] = '\0';
2385                 } else
2386                         p = (char *)"unknown";
2387                 node->symbol = strdup(p);
2388                 if (!node->symbol) {
2389                         ret = -ENOMEM;
2390                         break;
2391                 }
2392                 pr_debug2("Blacklist: 0x%lx-0x%lx, %s\n",
2393                           node->start, node->end, node->symbol);
2394                 ret++;
2395         }
2396         if (ret < 0)
2397                 kprobe_blacklist__delete(blacklist);
2398         fclose(fp);
2399
2400         return ret;
2401 }
2402
2403 static struct kprobe_blacklist_node *
2404 kprobe_blacklist__find_by_address(struct list_head *blacklist,
2405                                   unsigned long address)
2406 {
2407         struct kprobe_blacklist_node *node;
2408
2409         list_for_each_entry(node, blacklist, list) {
2410                 if (node->start <= address && address < node->end)
2411                         return node;
2412         }
2413
2414         return NULL;
2415 }
2416
2417 static LIST_HEAD(kprobe_blacklist);
2418
2419 static void kprobe_blacklist__init(void)
2420 {
2421         if (!list_empty(&kprobe_blacklist))
2422                 return;
2423
2424         if (kprobe_blacklist__load(&kprobe_blacklist) < 0)
2425                 pr_debug("No kprobe blacklist support, ignored\n");
2426 }
2427
2428 static void kprobe_blacklist__release(void)
2429 {
2430         kprobe_blacklist__delete(&kprobe_blacklist);
2431 }
2432
2433 static bool kprobe_blacklist__listed(unsigned long address)
2434 {
2435         return !!kprobe_blacklist__find_by_address(&kprobe_blacklist, address);
2436 }
2437
2438 static int perf_probe_event__sprintf(const char *group, const char *event,
2439                                      struct perf_probe_event *pev,
2440                                      const char *module,
2441                                      struct strbuf *result)
2442 {
2443         int i, ret;
2444         char *buf;
2445
2446         if (asprintf(&buf, "%s:%s", group, event) < 0)
2447                 return -errno;
2448         ret = strbuf_addf(result, "  %-20s (on ", buf);
2449         free(buf);
2450         if (ret)
2451                 return ret;
2452
2453         /* Synthesize only event probe point */
2454         buf = synthesize_perf_probe_point(&pev->point);
2455         if (!buf)
2456                 return -ENOMEM;
2457         ret = strbuf_addstr(result, buf);
2458         free(buf);
2459
2460         if (!ret && module)
2461                 ret = strbuf_addf(result, " in %s", module);
2462
2463         if (!ret && pev->nargs > 0) {
2464                 ret = strbuf_add(result, " with", 5);
2465                 for (i = 0; !ret && i < pev->nargs; i++) {
2466                         buf = synthesize_perf_probe_arg(&pev->args[i]);
2467                         if (!buf)
2468                                 return -ENOMEM;
2469                         ret = strbuf_addf(result, " %s", buf);
2470                         free(buf);
2471                 }
2472         }
2473         if (!ret)
2474                 ret = strbuf_addch(result, ')');
2475
2476         return ret;
2477 }
2478
2479 /* Show an event */
2480 int show_perf_probe_event(const char *group, const char *event,
2481                           struct perf_probe_event *pev,
2482                           const char *module, bool use_stdout)
2483 {
2484         struct strbuf buf = STRBUF_INIT;
2485         int ret;
2486
2487         ret = perf_probe_event__sprintf(group, event, pev, module, &buf);
2488         if (ret >= 0) {
2489                 if (use_stdout)
2490                         printf("%s\n", buf.buf);
2491                 else
2492                         pr_info("%s\n", buf.buf);
2493         }
2494         strbuf_release(&buf);
2495
2496         return ret;
2497 }
2498
2499 static bool filter_probe_trace_event(struct probe_trace_event *tev,
2500                                      struct strfilter *filter)
2501 {
2502         char tmp[128];
2503
2504         /* At first, check the event name itself */
2505         if (strfilter__compare(filter, tev->event))
2506                 return true;
2507
2508         /* Next, check the combination of name and group */
2509         if (e_snprintf(tmp, 128, "%s:%s", tev->group, tev->event) < 0)
2510                 return false;
2511         return strfilter__compare(filter, tmp);
2512 }
2513
2514 static int __show_perf_probe_events(int fd, bool is_kprobe,
2515                                     struct strfilter *filter)
2516 {
2517         int ret = 0;
2518         struct probe_trace_event tev;
2519         struct perf_probe_event pev;
2520         struct strlist *rawlist;
2521         struct str_node *ent;
2522
2523         memset(&tev, 0, sizeof(tev));
2524         memset(&pev, 0, sizeof(pev));
2525
2526         rawlist = probe_file__get_rawlist(fd);
2527         if (!rawlist)
2528                 return -ENOMEM;
2529
2530         strlist__for_each_entry(ent, rawlist) {
2531                 ret = parse_probe_trace_command(ent->s, &tev);
2532                 if (ret >= 0) {
2533                         if (!filter_probe_trace_event(&tev, filter))
2534                                 goto next;
2535                         ret = convert_to_perf_probe_event(&tev, &pev,
2536                                                                 is_kprobe);
2537                         if (ret < 0)
2538                                 goto next;
2539                         ret = show_perf_probe_event(pev.group, pev.event,
2540                                                     &pev, tev.point.module,
2541                                                     true);
2542                 }
2543 next:
2544                 clear_perf_probe_event(&pev);
2545                 clear_probe_trace_event(&tev);
2546                 if (ret < 0)
2547                         break;
2548         }
2549         strlist__delete(rawlist);
2550         /* Cleanup cached debuginfo if needed */
2551         debuginfo_cache__exit();
2552
2553         return ret;
2554 }
2555
2556 /* List up current perf-probe events */
2557 int show_perf_probe_events(struct strfilter *filter)
2558 {
2559         int kp_fd, up_fd, ret;
2560
2561         setup_pager();
2562
2563         if (probe_conf.cache)
2564                 return probe_cache__show_all_caches(filter);
2565
2566         ret = init_probe_symbol_maps(false);
2567         if (ret < 0)
2568                 return ret;
2569
2570         ret = probe_file__open_both(&kp_fd, &up_fd, 0);
2571         if (ret < 0)
2572                 return ret;
2573
2574         if (kp_fd >= 0)
2575                 ret = __show_perf_probe_events(kp_fd, true, filter);
2576         if (up_fd >= 0 && ret >= 0)
2577                 ret = __show_perf_probe_events(up_fd, false, filter);
2578         if (kp_fd > 0)
2579                 close(kp_fd);
2580         if (up_fd > 0)
2581                 close(up_fd);
2582         exit_probe_symbol_maps();
2583
2584         return ret;
2585 }
2586
2587 static int get_new_event_name(char *buf, size_t len, const char *base,
2588                               struct strlist *namelist, bool allow_suffix)
2589 {
2590         int i, ret;
2591         char *p, *nbase;
2592
2593         if (*base == '.')
2594                 base++;
2595         nbase = strdup(base);
2596         if (!nbase)
2597                 return -ENOMEM;
2598
2599         /* Cut off the dot suffixes (e.g. .const, .isra)*/
2600         p = strchr(nbase, '.');
2601         if (p && p != nbase)
2602                 *p = '\0';
2603
2604         /* Try no suffix number */
2605         ret = e_snprintf(buf, len, "%s", nbase);
2606         if (ret < 0) {
2607                 pr_debug("snprintf() failed: %d\n", ret);
2608                 goto out;
2609         }
2610         if (!strlist__has_entry(namelist, buf))
2611                 goto out;
2612
2613         if (!allow_suffix) {
2614                 pr_warning("Error: event \"%s\" already exists.\n"
2615                            " Hint: Remove existing event by 'perf probe -d'\n"
2616                            "       or force duplicates by 'perf probe -f'\n"
2617                            "       or set 'force=yes' in BPF source.\n",
2618                            buf);
2619                 ret = -EEXIST;
2620                 goto out;
2621         }
2622
2623         /* Try to add suffix */
2624         for (i = 1; i < MAX_EVENT_INDEX; i++) {
2625                 ret = e_snprintf(buf, len, "%s_%d", nbase, i);
2626                 if (ret < 0) {
2627                         pr_debug("snprintf() failed: %d\n", ret);
2628                         goto out;
2629                 }
2630                 if (!strlist__has_entry(namelist, buf))
2631                         break;
2632         }
2633         if (i == MAX_EVENT_INDEX) {
2634                 pr_warning("Too many events are on the same function.\n");
2635                 ret = -ERANGE;
2636         }
2637
2638 out:
2639         free(nbase);
2640
2641         /* Final validation */
2642         if (ret >= 0 && !is_c_func_name(buf)) {
2643                 pr_warning("Internal error: \"%s\" is an invalid event name.\n",
2644                            buf);
2645                 ret = -EINVAL;
2646         }
2647
2648         return ret;
2649 }
2650
2651 /* Warn if the current kernel's uprobe implementation is old */
2652 static void warn_uprobe_event_compat(struct probe_trace_event *tev)
2653 {
2654         int i;
2655         char *buf = synthesize_probe_trace_command(tev);
2656
2657         /* Old uprobe event doesn't support memory dereference */
2658         if (!tev->uprobes || tev->nargs == 0 || !buf)
2659                 goto out;
2660
2661         for (i = 0; i < tev->nargs; i++)
2662                 if (strglobmatch(tev->args[i].value, "[$@+-]*")) {
2663                         pr_warning("Please upgrade your kernel to at least "
2664                                    "3.14 to have access to feature %s\n",
2665                                    tev->args[i].value);
2666                         break;
2667                 }
2668 out:
2669         free(buf);
2670 }
2671
2672 /* Set new name from original perf_probe_event and namelist */
2673 static int probe_trace_event__set_name(struct probe_trace_event *tev,
2674                                        struct perf_probe_event *pev,
2675                                        struct strlist *namelist,
2676                                        bool allow_suffix)
2677 {
2678         const char *event, *group;
2679         char buf[64];
2680         int ret;
2681
2682         /* If probe_event or trace_event already have the name, reuse it */
2683         if (pev->event && !pev->sdt)
2684                 event = pev->event;
2685         else if (tev->event)
2686                 event = tev->event;
2687         else {
2688                 /* Or generate new one from probe point */
2689                 if (pev->point.function &&
2690                         (strncmp(pev->point.function, "0x", 2) != 0) &&
2691                         !strisglob(pev->point.function))
2692                         event = pev->point.function;
2693                 else
2694                         event = tev->point.realname;
2695         }
2696         if (pev->group && !pev->sdt)
2697                 group = pev->group;
2698         else if (tev->group)
2699                 group = tev->group;
2700         else
2701                 group = PERFPROBE_GROUP;
2702
2703         /* Get an unused new event name */
2704         ret = get_new_event_name(buf, 64, event,
2705                                  namelist, allow_suffix);
2706         if (ret < 0)
2707                 return ret;
2708
2709         event = buf;
2710
2711         tev->event = strdup(event);
2712         tev->group = strdup(group);
2713         if (tev->event == NULL || tev->group == NULL)
2714                 return -ENOMEM;
2715
2716         /* Add added event name to namelist */
2717         strlist__add(namelist, event);
2718         return 0;
2719 }
2720
2721 static int __open_probe_file_and_namelist(bool uprobe,
2722                                           struct strlist **namelist)
2723 {
2724         int fd;
2725
2726         fd = probe_file__open(PF_FL_RW | (uprobe ? PF_FL_UPROBE : 0));
2727         if (fd < 0)
2728                 return fd;
2729
2730         /* Get current event names */
2731         *namelist = probe_file__get_namelist(fd);
2732         if (!(*namelist)) {
2733                 pr_debug("Failed to get current event list.\n");
2734                 close(fd);
2735                 return -ENOMEM;
2736         }
2737         return fd;
2738 }
2739
2740 static int __add_probe_trace_events(struct perf_probe_event *pev,
2741                                      struct probe_trace_event *tevs,
2742                                      int ntevs, bool allow_suffix)
2743 {
2744         int i, fd[2] = {-1, -1}, up, ret;
2745         struct probe_trace_event *tev = NULL;
2746         struct probe_cache *cache = NULL;
2747         struct strlist *namelist[2] = {NULL, NULL};
2748         struct nscookie nsc;
2749
2750         up = pev->uprobes ? 1 : 0;
2751         fd[up] = __open_probe_file_and_namelist(up, &namelist[up]);
2752         if (fd[up] < 0)
2753                 return fd[up];
2754
2755         ret = 0;
2756         for (i = 0; i < ntevs; i++) {
2757                 tev = &tevs[i];
2758                 up = tev->uprobes ? 1 : 0;
2759                 if (fd[up] == -1) {     /* Open the kprobe/uprobe_events */
2760                         fd[up] = __open_probe_file_and_namelist(up,
2761                                                                 &namelist[up]);
2762                         if (fd[up] < 0)
2763                                 goto close_out;
2764                 }
2765                 /* Skip if the symbol is out of .text or blacklisted */
2766                 if (!tev->point.symbol && !pev->uprobes)
2767                         continue;
2768
2769                 /* Set new name for tev (and update namelist) */
2770                 ret = probe_trace_event__set_name(tev, pev, namelist[up],
2771                                                   allow_suffix);
2772                 if (ret < 0)
2773                         break;
2774
2775                 nsinfo__mountns_enter(pev->nsi, &nsc);
2776                 ret = probe_file__add_event(fd[up], tev);
2777                 nsinfo__mountns_exit(&nsc);
2778                 if (ret < 0)
2779                         break;
2780
2781                 /*
2782                  * Probes after the first probe which comes from same
2783                  * user input are always allowed to add suffix, because
2784                  * there might be several addresses corresponding to
2785                  * one code line.
2786                  */
2787                 allow_suffix = true;
2788         }
2789         if (ret == -EINVAL && pev->uprobes)
2790                 warn_uprobe_event_compat(tev);
2791         if (ret == 0 && probe_conf.cache) {
2792                 cache = probe_cache__new(pev->target, pev->nsi);
2793                 if (!cache ||
2794                     probe_cache__add_entry(cache, pev, tevs, ntevs) < 0 ||
2795                     probe_cache__commit(cache) < 0)
2796                         pr_warning("Failed to add event to probe cache\n");
2797                 probe_cache__delete(cache);
2798         }
2799
2800 close_out:
2801         for (up = 0; up < 2; up++) {
2802                 strlist__delete(namelist[up]);
2803                 if (fd[up] >= 0)
2804                         close(fd[up]);
2805         }
2806         return ret;
2807 }
2808
2809 static int find_probe_functions(struct map *map, char *name,
2810                                 struct symbol **syms)
2811 {
2812         int found = 0;
2813         struct symbol *sym;
2814         struct rb_node *tmp;
2815         const char *norm, *ver;
2816         char *buf = NULL;
2817
2818         if (map__load(map) < 0)
2819                 return 0;
2820
2821         map__for_each_symbol(map, sym, tmp) {
2822                 norm = arch__normalize_symbol_name(sym->name);
2823                 if (!norm)
2824                         continue;
2825
2826                 /* We don't care about default symbol or not */
2827                 ver = strchr(norm, '@');
2828                 if (ver) {
2829                         buf = strndup(norm, ver - norm);
2830                         if (!buf)
2831                                 return -ENOMEM;
2832                         norm = buf;
2833                 }
2834                 if (strglobmatch(norm, name)) {
2835                         found++;
2836                         if (syms && found < probe_conf.max_probes)
2837                                 syms[found - 1] = sym;
2838                 }
2839                 if (buf)
2840                         zfree(&buf);
2841         }
2842
2843         return found;
2844 }
2845
2846 void __weak arch__fix_tev_from_maps(struct perf_probe_event *pev __maybe_unused,
2847                                 struct probe_trace_event *tev __maybe_unused,
2848                                 struct map *map __maybe_unused,
2849                                 struct symbol *sym __maybe_unused) { }
2850
2851 /*
2852  * Find probe function addresses from map.
2853  * Return an error or the number of found probe_trace_event
2854  */
2855 static int find_probe_trace_events_from_map(struct perf_probe_event *pev,
2856                                             struct probe_trace_event **tevs)
2857 {
2858         struct map *map = NULL;
2859         struct ref_reloc_sym *reloc_sym = NULL;
2860         struct symbol *sym;
2861         struct symbol **syms = NULL;
2862         struct probe_trace_event *tev;
2863         struct perf_probe_point *pp = &pev->point;
2864         struct probe_trace_point *tp;
2865         int num_matched_functions;
2866         int ret, i, j, skipped = 0;
2867         char *mod_name;
2868
2869         map = get_target_map(pev->target, pev->nsi, pev->uprobes);
2870         if (!map) {
2871                 ret = -EINVAL;
2872                 goto out;
2873         }
2874
2875         syms = malloc(sizeof(struct symbol *) * probe_conf.max_probes);
2876         if (!syms) {
2877                 ret = -ENOMEM;
2878                 goto out;
2879         }
2880
2881         /*
2882          * Load matched symbols: Since the different local symbols may have
2883          * same name but different addresses, this lists all the symbols.
2884          */
2885         num_matched_functions = find_probe_functions(map, pp->function, syms);
2886         if (num_matched_functions <= 0) {
2887                 pr_err("Failed to find symbol %s in %s\n", pp->function,
2888                         pev->target ? : "kernel");
2889                 ret = -ENOENT;
2890                 goto out;
2891         } else if (num_matched_functions > probe_conf.max_probes) {
2892                 pr_err("Too many functions matched in %s\n",
2893                         pev->target ? : "kernel");
2894                 ret = -E2BIG;
2895                 goto out;
2896         }
2897
2898         /* Note that the symbols in the kmodule are not relocated */
2899         if (!pev->uprobes && !pev->target &&
2900                         (!pp->retprobe || kretprobe_offset_is_supported())) {
2901                 reloc_sym = kernel_get_ref_reloc_sym(NULL);
2902                 if (!reloc_sym) {
2903                         pr_warning("Relocated base symbol is not found!\n");
2904                         ret = -EINVAL;
2905                         goto out;
2906                 }
2907         }
2908
2909         /* Setup result trace-probe-events */
2910         *tevs = zalloc(sizeof(*tev) * num_matched_functions);
2911         if (!*tevs) {
2912                 ret = -ENOMEM;
2913                 goto out;
2914         }
2915
2916         ret = 0;
2917
2918         for (j = 0; j < num_matched_functions; j++) {
2919                 sym = syms[j];
2920
2921                 tev = (*tevs) + ret;
2922                 tp = &tev->point;
2923                 if (ret == num_matched_functions) {
2924                         pr_warning("Too many symbols are listed. Skip it.\n");
2925                         break;
2926                 }
2927                 ret++;
2928
2929                 if (pp->offset > sym->end - sym->start) {
2930                         pr_warning("Offset %ld is bigger than the size of %s\n",
2931                                    pp->offset, sym->name);
2932                         ret = -ENOENT;
2933                         goto err_out;
2934                 }
2935                 /* Add one probe point */
2936                 tp->address = map->unmap_ip(map, sym->start) + pp->offset;
2937
2938                 /* Check the kprobe (not in module) is within .text  */
2939                 if (!pev->uprobes && !pev->target &&
2940                     kprobe_warn_out_range(sym->name, tp->address)) {
2941                         tp->symbol = NULL;      /* Skip it */
2942                         skipped++;
2943                 } else if (reloc_sym) {
2944                         tp->symbol = strdup_or_goto(reloc_sym->name, nomem_out);
2945                         tp->offset = tp->address - reloc_sym->addr;
2946                 } else {
2947                         tp->symbol = strdup_or_goto(sym->name, nomem_out);
2948                         tp->offset = pp->offset;
2949                 }
2950                 tp->realname = strdup_or_goto(sym->name, nomem_out);
2951
2952                 tp->retprobe = pp->retprobe;
2953                 if (pev->target) {
2954                         if (pev->uprobes) {
2955                                 tev->point.module = strdup_or_goto(pev->target,
2956                                                                    nomem_out);
2957                         } else {
2958                                 mod_name = find_module_name(pev->target);
2959                                 tev->point.module =
2960                                         strdup(mod_name ? mod_name : pev->target);
2961                                 free(mod_name);
2962                                 if (!tev->point.module)
2963                                         goto nomem_out;
2964                         }
2965                 }
2966                 tev->uprobes = pev->uprobes;
2967                 tev->nargs = pev->nargs;
2968                 if (tev->nargs) {
2969                         tev->args = zalloc(sizeof(struct probe_trace_arg) *
2970                                            tev->nargs);
2971                         if (tev->args == NULL)
2972                                 goto nomem_out;
2973                 }
2974                 for (i = 0; i < tev->nargs; i++) {
2975                         if (pev->args[i].name)
2976                                 tev->args[i].name =
2977                                         strdup_or_goto(pev->args[i].name,
2978                                                         nomem_out);
2979
2980                         tev->args[i].value = strdup_or_goto(pev->args[i].var,
2981                                                             nomem_out);
2982                         if (pev->args[i].type)
2983                                 tev->args[i].type =
2984                                         strdup_or_goto(pev->args[i].type,
2985                                                         nomem_out);
2986                 }
2987                 arch__fix_tev_from_maps(pev, tev, map, sym);
2988         }
2989         if (ret == skipped) {
2990                 ret = -ENOENT;
2991                 goto err_out;
2992         }
2993
2994 out:
2995         map__put(map);
2996         free(syms);
2997         return ret;
2998
2999 nomem_out:
3000         ret = -ENOMEM;
3001 err_out:
3002         clear_probe_trace_events(*tevs, num_matched_functions);
3003         zfree(tevs);
3004         goto out;
3005 }
3006
3007 static int try_to_find_absolute_address(struct perf_probe_event *pev,
3008                                         struct probe_trace_event **tevs)
3009 {
3010         struct perf_probe_point *pp = &pev->point;
3011         struct probe_trace_event *tev;
3012         struct probe_trace_point *tp;
3013         int i, err;
3014
3015         if (!(pev->point.function && !strncmp(pev->point.function, "0x", 2)))
3016                 return -EINVAL;
3017         if (perf_probe_event_need_dwarf(pev))
3018                 return -EINVAL;
3019
3020         /*
3021          * This is 'perf probe /lib/libc.so 0xabcd'. Try to probe at
3022          * absolute address.
3023          *
3024          * Only one tev can be generated by this.
3025          */
3026         *tevs = zalloc(sizeof(*tev));
3027         if (!*tevs)
3028                 return -ENOMEM;
3029
3030         tev = *tevs;
3031         tp = &tev->point;
3032
3033         /*
3034          * Don't use tp->offset, use address directly, because
3035          * in synthesize_probe_trace_command() address cannot be
3036          * zero.
3037          */
3038         tp->address = pev->point.abs_address;
3039         tp->retprobe = pp->retprobe;
3040         tev->uprobes = pev->uprobes;
3041
3042         err = -ENOMEM;
3043         /*
3044          * Give it a '0x' leading symbol name.
3045          * In __add_probe_trace_events, a NULL symbol is interpreted as
3046          * invalud.
3047          */
3048         if (asprintf(&tp->symbol, "0x%lx", tp->address) < 0)
3049                 goto errout;
3050
3051         /* For kprobe, check range */
3052         if ((!tev->uprobes) &&
3053             (kprobe_warn_out_range(tev->point.symbol,
3054                                    tev->point.address))) {
3055                 err = -EACCES;
3056                 goto errout;
3057         }
3058
3059         if (asprintf(&tp->realname, "abs_%lx", tp->address) < 0)
3060                 goto errout;
3061
3062         if (pev->target) {
3063                 tp->module = strdup(pev->target);
3064                 if (!tp->module)
3065                         goto errout;
3066         }
3067
3068         if (tev->group) {
3069                 tev->group = strdup(pev->group);
3070                 if (!tev->group)
3071                         goto errout;
3072         }
3073
3074         if (pev->event) {
3075                 tev->event = strdup(pev->event);
3076                 if (!tev->event)
3077                         goto errout;
3078         }
3079
3080         tev->nargs = pev->nargs;
3081         tev->args = zalloc(sizeof(struct probe_trace_arg) * tev->nargs);
3082         if (!tev->args)
3083                 goto errout;
3084
3085         for (i = 0; i < tev->nargs; i++)
3086                 copy_to_probe_trace_arg(&tev->args[i], &pev->args[i]);
3087
3088         return 1;
3089
3090 errout:
3091         clear_probe_trace_events(*tevs, 1);
3092         *tevs = NULL;
3093         return err;
3094 }
3095
3096 /* Concatinate two arrays */
3097 static void *memcat(void *a, size_t sz_a, void *b, size_t sz_b)
3098 {
3099         void *ret;
3100
3101         ret = malloc(sz_a + sz_b);
3102         if (ret) {
3103                 memcpy(ret, a, sz_a);
3104                 memcpy(ret + sz_a, b, sz_b);
3105         }
3106         return ret;
3107 }
3108
3109 static int
3110 concat_probe_trace_events(struct probe_trace_event **tevs, int *ntevs,
3111                           struct probe_trace_event **tevs2, int ntevs2)
3112 {
3113         struct probe_trace_event *new_tevs;
3114         int ret = 0;
3115
3116         if (*ntevs == 0) {
3117                 *tevs = *tevs2;
3118                 *ntevs = ntevs2;
3119                 *tevs2 = NULL;
3120                 return 0;
3121         }
3122
3123         if (*ntevs + ntevs2 > probe_conf.max_probes)
3124                 ret = -E2BIG;
3125         else {
3126                 /* Concatinate the array of probe_trace_event */
3127                 new_tevs = memcat(*tevs, (*ntevs) * sizeof(**tevs),
3128                                   *tevs2, ntevs2 * sizeof(**tevs2));
3129                 if (!new_tevs)
3130                         ret = -ENOMEM;
3131                 else {
3132                         free(*tevs);
3133                         *tevs = new_tevs;
3134                         *ntevs += ntevs2;
3135                 }
3136         }
3137         if (ret < 0)
3138                 clear_probe_trace_events(*tevs2, ntevs2);
3139         zfree(tevs2);
3140
3141         return ret;
3142 }
3143
3144 /*
3145  * Try to find probe_trace_event from given probe caches. Return the number
3146  * of cached events found, if an error occurs return the error.
3147  */
3148 static int find_cached_events(struct perf_probe_event *pev,
3149                               struct probe_trace_event **tevs,
3150                               const char *target)
3151 {
3152         struct probe_cache *cache;
3153         struct probe_cache_entry *entry;
3154         struct probe_trace_event *tmp_tevs = NULL;
3155         int ntevs = 0;
3156         int ret = 0;
3157
3158         cache = probe_cache__new(target, pev->nsi);
3159         /* Return 0 ("not found") if the target has no probe cache. */
3160         if (!cache)
3161                 return 0;
3162
3163         for_each_probe_cache_entry(entry, cache) {
3164                 /* Skip the cache entry which has no name */
3165                 if (!entry->pev.event || !entry->pev.group)
3166                         continue;
3167                 if ((!pev->group || strglobmatch(entry->pev.group, pev->group)) &&
3168                     strglobmatch(entry->pev.event, pev->event)) {
3169                         ret = probe_cache_entry__get_event(entry, &tmp_tevs);
3170                         if (ret > 0)
3171                                 ret = concat_probe_trace_events(tevs, &ntevs,
3172                                                                 &tmp_tevs, ret);
3173                         if (ret < 0)
3174                                 break;
3175                 }
3176         }
3177         probe_cache__delete(cache);
3178         if (ret < 0) {
3179                 clear_probe_trace_events(*tevs, ntevs);
3180                 zfree(tevs);
3181         } else {
3182                 ret = ntevs;
3183                 if (ntevs > 0 && target && target[0] == '/')
3184                         pev->uprobes = true;
3185         }
3186
3187         return ret;
3188 }
3189
3190 /* Try to find probe_trace_event from all probe caches */
3191 static int find_cached_events_all(struct perf_probe_event *pev,
3192                                    struct probe_trace_event **tevs)
3193 {
3194         struct probe_trace_event *tmp_tevs = NULL;
3195         struct strlist *bidlist;
3196         struct str_node *nd;
3197         char *pathname;
3198         int ntevs = 0;
3199         int ret;
3200
3201         /* Get the buildid list of all valid caches */
3202         bidlist = build_id_cache__list_all(true);
3203         if (!bidlist) {
3204                 ret = -errno;
3205                 pr_debug("Failed to get buildids: %d\n", ret);
3206                 return ret;
3207         }
3208
3209         ret = 0;
3210         strlist__for_each_entry(nd, bidlist) {
3211                 pathname = build_id_cache__origname(nd->s);
3212                 ret = find_cached_events(pev, &tmp_tevs, pathname);
3213                 /* In the case of cnt == 0, we just skip it */
3214                 if (ret > 0)
3215                         ret = concat_probe_trace_events(tevs, &ntevs,
3216                                                         &tmp_tevs, ret);
3217                 free(pathname);
3218                 if (ret < 0)
3219                         break;
3220         }
3221         strlist__delete(bidlist);
3222
3223         if (ret < 0) {
3224                 clear_probe_trace_events(*tevs, ntevs);
3225                 zfree(tevs);
3226         } else
3227                 ret = ntevs;
3228
3229         return ret;
3230 }
3231
3232 static int find_probe_trace_events_from_cache(struct perf_probe_event *pev,
3233                                               struct probe_trace_event **tevs)
3234 {
3235         struct probe_cache *cache;
3236         struct probe_cache_entry *entry;
3237         struct probe_trace_event *tev;
3238         struct str_node *node;
3239         int ret, i;
3240
3241         if (pev->sdt) {
3242                 /* For SDT/cached events, we use special search functions */
3243                 if (!pev->target)
3244                         return find_cached_events_all(pev, tevs);
3245                 else
3246                         return find_cached_events(pev, tevs, pev->target);
3247         }
3248         cache = probe_cache__new(pev->target, pev->nsi);
3249         if (!cache)
3250                 return 0;
3251
3252         entry = probe_cache__find(cache, pev);
3253         if (!entry) {
3254                 /* SDT must be in the cache */
3255                 ret = pev->sdt ? -ENOENT : 0;
3256                 goto out;
3257         }
3258
3259         ret = strlist__nr_entries(entry->tevlist);
3260         if (ret > probe_conf.max_probes) {
3261                 pr_debug("Too many entries matched in the cache of %s\n",
3262                          pev->target ? : "kernel");
3263                 ret = -E2BIG;
3264                 goto out;
3265         }
3266
3267         *tevs = zalloc(ret * sizeof(*tev));
3268         if (!*tevs) {
3269                 ret = -ENOMEM;
3270                 goto out;
3271         }
3272
3273         i = 0;
3274         strlist__for_each_entry(node, entry->tevlist) {
3275                 tev = &(*tevs)[i++];
3276                 ret = parse_probe_trace_command(node->s, tev);
3277                 if (ret < 0)
3278                         goto out;
3279                 /* Set the uprobes attribute as same as original */
3280                 tev->uprobes = pev->uprobes;
3281         }
3282         ret = i;
3283
3284 out:
3285         probe_cache__delete(cache);
3286         return ret;
3287 }
3288
3289 static int convert_to_probe_trace_events(struct perf_probe_event *pev,
3290                                          struct probe_trace_event **tevs)
3291 {
3292         int ret;
3293
3294         if (!pev->group && !pev->sdt) {
3295                 /* Set group name if not given */
3296                 if (!pev->uprobes) {
3297                         pev->group = strdup(PERFPROBE_GROUP);
3298                         ret = pev->group ? 0 : -ENOMEM;
3299                 } else
3300                         ret = convert_exec_to_group(pev->target, &pev->group);
3301                 if (ret != 0) {
3302                         pr_warning("Failed to make a group name.\n");
3303                         return ret;
3304                 }
3305         }
3306
3307         ret = try_to_find_absolute_address(pev, tevs);
3308         if (ret > 0)
3309                 return ret;
3310
3311         /* At first, we need to lookup cache entry */
3312         ret = find_probe_trace_events_from_cache(pev, tevs);
3313         if (ret > 0 || pev->sdt)        /* SDT can be found only in the cache */
3314                 return ret == 0 ? -ENOENT : ret; /* Found in probe cache */
3315
3316         /* Convert perf_probe_event with debuginfo */
3317         ret = try_to_find_probe_trace_events(pev, tevs);
3318         if (ret != 0)
3319                 return ret;     /* Found in debuginfo or got an error */
3320
3321         return find_probe_trace_events_from_map(pev, tevs);
3322 }
3323
3324 int convert_perf_probe_events(struct perf_probe_event *pevs, int npevs)
3325 {
3326         int i, ret;
3327
3328         /* Loop 1: convert all events */
3329         for (i = 0; i < npevs; i++) {
3330                 /* Init kprobe blacklist if needed */
3331                 if (!pevs[i].uprobes)
3332                         kprobe_blacklist__init();
3333                 /* Convert with or without debuginfo */
3334                 ret  = convert_to_probe_trace_events(&pevs[i], &pevs[i].tevs);
3335                 if (ret < 0)
3336                         return ret;
3337                 pevs[i].ntevs = ret;
3338         }
3339         /* This just release blacklist only if allocated */
3340         kprobe_blacklist__release();
3341
3342         return 0;
3343 }
3344
3345 static int show_probe_trace_event(struct probe_trace_event *tev)
3346 {
3347         char *buf = synthesize_probe_trace_command(tev);
3348
3349         if (!buf) {
3350                 pr_debug("Failed to synthesize probe trace event.\n");
3351                 return -EINVAL;
3352         }
3353
3354         /* Showing definition always go stdout */
3355         printf("%s\n", buf);
3356         free(buf);
3357
3358         return 0;
3359 }
3360
3361 int show_probe_trace_events(struct perf_probe_event *pevs, int npevs)
3362 {
3363         struct strlist *namelist = strlist__new(NULL, NULL);
3364         struct probe_trace_event *tev;
3365         struct perf_probe_event *pev;
3366         int i, j, ret = 0;
3367
3368         if (!namelist)
3369                 return -ENOMEM;
3370
3371         for (j = 0; j < npevs && !ret; j++) {
3372                 pev = &pevs[j];
3373                 for (i = 0; i < pev->ntevs && !ret; i++) {
3374                         tev = &pev->tevs[i];
3375                         /* Skip if the symbol is out of .text or blacklisted */
3376                         if (!tev->point.symbol && !pev->uprobes)
3377                                 continue;
3378
3379                         /* Set new name for tev (and update namelist) */
3380                         ret = probe_trace_event__set_name(tev, pev,
3381                                                           namelist, true);
3382                         if (!ret)
3383                                 ret = show_probe_trace_event(tev);
3384                 }
3385         }
3386         strlist__delete(namelist);
3387
3388         return ret;
3389 }
3390
3391 int apply_perf_probe_events(struct perf_probe_event *pevs, int npevs)
3392 {
3393         int i, ret = 0;
3394
3395         /* Loop 2: add all events */
3396         for (i = 0; i < npevs; i++) {
3397                 ret = __add_probe_trace_events(&pevs[i], pevs[i].tevs,
3398                                                pevs[i].ntevs,
3399                                                probe_conf.force_add);
3400                 if (ret < 0)
3401                         break;
3402         }
3403         return ret;
3404 }
3405
3406 void cleanup_perf_probe_events(struct perf_probe_event *pevs, int npevs)
3407 {
3408         int i, j;
3409         struct perf_probe_event *pev;
3410
3411         /* Loop 3: cleanup and free trace events  */
3412         for (i = 0; i < npevs; i++) {
3413                 pev = &pevs[i];
3414                 for (j = 0; j < pevs[i].ntevs; j++)
3415                         clear_probe_trace_event(&pevs[i].tevs[j]);
3416                 zfree(&pevs[i].tevs);
3417                 pevs[i].ntevs = 0;
3418                 nsinfo__zput(pev->nsi);
3419                 clear_perf_probe_event(&pevs[i]);
3420         }
3421 }
3422
3423 int add_perf_probe_events(struct perf_probe_event *pevs, int npevs)
3424 {
3425         int ret;
3426
3427         ret = init_probe_symbol_maps(pevs->uprobes);
3428         if (ret < 0)
3429                 return ret;
3430
3431         ret = convert_perf_probe_events(pevs, npevs);
3432         if (ret == 0)
3433                 ret = apply_perf_probe_events(pevs, npevs);
3434
3435         cleanup_perf_probe_events(pevs, npevs);
3436
3437         exit_probe_symbol_maps();
3438         return ret;
3439 }
3440
3441 int del_perf_probe_events(struct strfilter *filter)
3442 {
3443         int ret, ret2, ufd = -1, kfd = -1;
3444         char *str = strfilter__string(filter);
3445
3446         if (!str)
3447                 return -EINVAL;
3448
3449         /* Get current event names */
3450         ret = probe_file__open_both(&kfd, &ufd, PF_FL_RW);
3451         if (ret < 0)
3452                 goto out;
3453
3454         ret = probe_file__del_events(kfd, filter);
3455         if (ret < 0 && ret != -ENOENT)
3456                 goto error;
3457
3458         ret2 = probe_file__del_events(ufd, filter);
3459         if (ret2 < 0 && ret2 != -ENOENT) {
3460                 ret = ret2;
3461                 goto error;
3462         }
3463         ret = 0;
3464
3465 error:
3466         if (kfd >= 0)
3467                 close(kfd);
3468         if (ufd >= 0)
3469                 close(ufd);
3470 out:
3471         free(str);
3472
3473         return ret;
3474 }
3475
3476 int show_available_funcs(const char *target, struct nsinfo *nsi,
3477                          struct strfilter *_filter, bool user)
3478 {
3479         struct rb_node *nd;
3480         struct map *map;
3481         int ret;
3482
3483         ret = init_probe_symbol_maps(user);
3484         if (ret < 0)
3485                 return ret;
3486
3487         /* Get a symbol map */
3488         map = get_target_map(target, nsi, user);
3489         if (!map) {
3490                 pr_err("Failed to get a map for %s\n", (target) ? : "kernel");
3491                 return -EINVAL;
3492         }
3493
3494         ret = map__load(map);
3495         if (ret) {
3496                 if (ret == -2) {
3497                         char *str = strfilter__string(_filter);
3498                         pr_err("Failed to find symbols matched to \"%s\"\n",
3499                                str);
3500                         free(str);
3501                 } else
3502                         pr_err("Failed to load symbols in %s\n",
3503                                (target) ? : "kernel");
3504                 goto end;
3505         }
3506         if (!dso__sorted_by_name(map->dso, map->type))
3507                 dso__sort_by_name(map->dso, map->type);
3508
3509         /* Show all (filtered) symbols */
3510         setup_pager();
3511
3512         for (nd = rb_first(&map->dso->symbol_names[map->type]); nd; nd = rb_next(nd)) {
3513                 struct symbol_name_rb_node *pos = rb_entry(nd, struct symbol_name_rb_node, rb_node);
3514
3515                 if (strfilter__compare(_filter, pos->sym.name))
3516                         printf("%s\n", pos->sym.name);
3517         }
3518
3519 end:
3520         map__put(map);
3521         exit_probe_symbol_maps();
3522
3523         return ret;
3524 }
3525
3526 int copy_to_probe_trace_arg(struct probe_trace_arg *tvar,
3527                             struct perf_probe_arg *pvar)
3528 {
3529         tvar->value = strdup(pvar->var);
3530         if (tvar->value == NULL)
3531                 return -ENOMEM;
3532         if (pvar->type) {
3533                 tvar->type = strdup(pvar->type);
3534                 if (tvar->type == NULL)
3535                         return -ENOMEM;
3536         }
3537         if (pvar->name) {
3538                 tvar->name = strdup(pvar->name);
3539                 if (tvar->name == NULL)
3540                         return -ENOMEM;
3541         } else
3542                 tvar->name = NULL;
3543         return 0;
3544 }