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