probe-event.c 38.3 KB
Newer Older
1
/*
2
 * probe-event.c : perf-probe definition to probe_events format converter
3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31
 *
 * Written by Masami Hiramatsu <mhiramat@redhat.com>
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
 *
 */

#define _GNU_SOURCE
#include <sys/utsname.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <errno.h>
#include <stdio.h>
#include <unistd.h>
#include <stdlib.h>
#include <string.h>
32 33
#include <stdarg.h>
#include <limits.h>
34 35

#undef _GNU_SOURCE
36
#include "util.h"
37
#include "event.h"
38
#include "string.h"
39
#include "strlist.h"
40
#include "debug.h"
41
#include "cache.h"
42
#include "color.h"
43 44
#include "symbol.h"
#include "thread.h"
45
#include "debugfs.h"
46
#include "trace-event.h"	/* For __unused */
47
#include "probe-event.h"
48
#include "probe-finder.h"
49 50 51 52 53

#define MAX_CMDLEN 256
#define MAX_PROBE_ARGS 128
#define PERFPROBE_GROUP "probe"

54 55
bool probe_event_dry_run;	/* Dry run flag */

56
#define semantic_error(msg ...) pr_err("Semantic error :" msg)
57

58
/* If there is no space to write, returns -E2BIG. */
59 60 61
static int e_snprintf(char *str, size_t size, const char *format, ...)
	__attribute__((format(printf, 3, 4)));

62 63 64 65 66 67 68 69 70 71 72 73
static int e_snprintf(char *str, size_t size, const char *format, ...)
{
	int ret;
	va_list ap;
	va_start(ap, format);
	ret = vsnprintf(str, size, format, ap);
	va_end(ap);
	if (ret >= (int)size)
		ret = -E2BIG;
	return ret;
}

74
static char *synthesize_perf_probe_point(struct perf_probe_point *pp);
75
static struct machine machine;
76

77
/* Initialize symbol maps and path of vmlinux */
78
static int init_vmlinux(void)
79
{
80
	struct dso *kernel;
81 82
	int ret;

83 84 85 86 87
	symbol_conf.sort_by_name = true;
	if (symbol_conf.vmlinux_name == NULL)
		symbol_conf.try_vmlinux_path = true;
	else
		pr_debug("Use vmlinux: %s\n", symbol_conf.vmlinux_name);
88 89 90 91 92
	ret = symbol__init();
	if (ret < 0) {
		pr_debug("Failed to init symbol map.\n");
		goto out;
	}
93

94 95 96 97
	ret = machine__init(&machine, "/", 0);
	if (ret < 0)
		goto out;

98 99 100 101
	kernel = dso__new_kernel(symbol_conf.vmlinux_name);
	if (kernel == NULL)
		die("Failed to create kernel dso.");

102
	ret = __machine__create_kernel_maps(&machine, kernel);
103 104 105 106 107 108 109
	if (ret < 0)
		pr_debug("Failed to create kernel maps.\n");

out:
	if (ret < 0)
		pr_warning("Failed to init vmlinux path.\n");
	return ret;
110 111
}

112
#ifdef DWARF_SUPPORT
113 114
static int open_vmlinux(void)
{
115
	if (map__load(machine.vmlinux_maps[MAP__FUNCTION], NULL) < 0) {
116 117 118
		pr_debug("Failed to load kernel map.\n");
		return -EINVAL;
	}
119 120
	pr_debug("Try to open %s\n", machine.vmlinux_maps[MAP__FUNCTION]->dso->long_name);
	return open(machine.vmlinux_maps[MAP__FUNCTION]->dso->long_name, O_RDONLY);
121
}
122

123 124 125 126 127
/*
 * Convert trace point to probe point with debuginfo
 * Currently only handles kprobes.
 */
static int kprobe_convert_to_perf_probe(struct probe_trace_point *tp,
128
				       struct perf_probe_point *pp)
129 130
{
	struct symbol *sym;
131
	int fd, ret = -ENOENT;
132

133
	sym = map__find_symbol_by_name(machine.vmlinux_maps[MAP__FUNCTION],
134 135 136
				       tp->symbol, NULL);
	if (sym) {
		fd = open_vmlinux();
137 138 139 140 141
		if (fd >= 0) {
			ret = find_perf_probe_point(fd,
						 sym->start + tp->offset, pp);
			close(fd);
		}
142 143
	}
	if (ret <= 0) {
144 145
		pr_debug("Failed to find corresponding probes from "
			 "debuginfo. Use kprobe event information.\n");
146 147 148
		pp->function = strdup(tp->symbol);
		if (pp->function == NULL)
			return -ENOMEM;
149 150 151
		pp->offset = tp->offset;
	}
	pp->retprobe = tp->retprobe;
152 153

	return 0;
154 155 156
}

/* Try to find perf_probe_event with debuginfo */
157 158
static int try_to_find_probe_trace_events(struct perf_probe_event *pev,
					   struct probe_trace_event **tevs,
159
					   int max_tevs)
160 161 162 163 164 165
{
	bool need_dwarf = perf_probe_event_need_dwarf(pev);
	int fd, ntevs;

	fd = open_vmlinux();
	if (fd < 0) {
166 167 168 169
		if (need_dwarf) {
			pr_warning("Failed to open debuginfo file.\n");
			return fd;
		}
170 171 172 173 174
		pr_debug("Could not open vmlinux. Try to use symbols.\n");
		return 0;
	}

	/* Searching trace events corresponding to probe event */
175
	ntevs = find_probe_trace_events(fd, pev, tevs, max_tevs);
176 177
	close(fd);

178
	if (ntevs > 0) {	/* Succeeded to find trace events */
179
		pr_debug("find %d probe_trace_events.\n", ntevs);
180
		return ntevs;
181
	}
182

183 184 185 186 187 188
	if (ntevs == 0)	{	/* No error but failed to find probe point. */
		pr_warning("Probe point '%s' not found.\n",
			   synthesize_perf_probe_point(&pev->point));
		return -ENOENT;
	}
	/* Error path : ntevs < 0 */
189 190 191 192 193 194 195 196
	pr_debug("An error occurred in debuginfo analysis (%d).\n", ntevs);
	if (ntevs == -EBADF) {
		pr_warning("Warning: No dwarf info found in the vmlinux - "
			"please rebuild kernel with CONFIG_DEBUG_INFO=y.\n");
		if (!need_dwarf) {
			pr_debug("Trying to use symbols.\nn");
			return 0;
		}
197
	}
198
	return ntevs;
199 200
}

201 202 203 204 205 206
/*
 * Find a src file from a DWARF tag path. Prepend optional source path prefix
 * and chop off leading directories that do not exist. Result is passed back as
 * a newly allocated path on success.
 * Return 0 if file was found and readable, -errno otherwise.
 */
207 208
static int get_real_path(const char *raw_path, const char *comp_dir,
			 char **new_path)
209
{
210 211 212 213 214 215 216 217 218 219 220 221 222
	const char *prefix = symbol_conf.source_prefix;

	if (!prefix) {
		if (raw_path[0] != '/' && comp_dir)
			/* If not an absolute path, try to use comp_dir */
			prefix = comp_dir;
		else {
			if (access(raw_path, R_OK) == 0) {
				*new_path = strdup(raw_path);
				return 0;
			} else
				return -errno;
		}
223 224
	}

225
	*new_path = malloc((strlen(prefix) + strlen(raw_path) + 2));
226 227 228 229
	if (!*new_path)
		return -ENOMEM;

	for (;;) {
230
		sprintf(*new_path, "%s/%s", prefix, raw_path);
231 232 233 234

		if (access(*new_path, R_OK) == 0)
			return 0;

235 236 237 238
		if (!symbol_conf.source_prefix)
			/* In case of searching comp_dir, don't retry */
			return -errno;

239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259
		switch (errno) {
		case ENAMETOOLONG:
		case ENOENT:
		case EROFS:
		case EFAULT:
			raw_path = strchr(++raw_path, '/');
			if (!raw_path) {
				free(*new_path);
				*new_path = NULL;
				return -ENOENT;
			}
			continue;

		default:
			free(*new_path);
			*new_path = NULL;
			return -errno;
		}
	}
}

260 261 262
#define LINEBUF_SIZE 256
#define NR_ADDITIONAL_LINES 2

263
static int show_one_line(FILE *fp, int l, bool skip, bool show_num)
264 265 266 267 268 269 270 271
{
	char buf[LINEBUF_SIZE];
	const char *color = PERF_COLOR_BLUE;

	if (fgets(buf, LINEBUF_SIZE, fp) == NULL)
		goto error;
	if (!skip) {
		if (show_num)
272
			fprintf(stdout, "%7d  %s", l, buf);
273 274 275 276 277 278 279 280 281 282 283 284 285 286 287
		else
			color_fprintf(stdout, color, "         %s", buf);
	}

	while (strlen(buf) == LINEBUF_SIZE - 1 &&
	       buf[LINEBUF_SIZE - 2] != '\n') {
		if (fgets(buf, LINEBUF_SIZE, fp) == NULL)
			goto error;
		if (!skip) {
			if (show_num)
				fprintf(stdout, "%s", buf);
			else
				color_fprintf(stdout, color, "%s", buf);
		}
	}
288 289

	return 0;
290 291
error:
	if (feof(fp))
292
		pr_warning("Source file is shorter than expected.\n");
293
	else
294 295 296
		pr_warning("File read error: %s\n", strerror(errno));

	return -1;
297 298 299 300 301 302
}

/*
 * Show line-range always requires debuginfo to find source file and
 * line number.
 */
303
int show_line_range(struct line_range *lr)
304
{
305
	int l = 1;
306 307 308
	struct line_node *ln;
	FILE *fp;
	int fd, ret;
309
	char *tmp;
310 311

	/* Search a line range */
312 313 314 315
	ret = init_vmlinux();
	if (ret < 0)
		return ret;

316
	fd = open_vmlinux();
317 318 319 320 321
	if (fd < 0) {
		pr_warning("Failed to open debuginfo file.\n");
		return fd;
	}

322 323
	ret = find_line_range(fd, lr);
	close(fd);
324 325 326 327 328 329 330
	if (ret == 0) {
		pr_warning("Specified source line is not found.\n");
		return -ENOENT;
	} else if (ret < 0) {
		pr_warning("Debuginfo analysis failed. (%d)\n", ret);
		return ret;
	}
331

332 333
	/* Convert source file path */
	tmp = lr->path;
334
	ret = get_real_path(tmp, lr->comp_dir, &lr->path);
335 336 337 338 339 340
	free(tmp);	/* Free old path */
	if (ret < 0) {
		pr_warning("Failed to find source file. (%d)\n", ret);
		return ret;
	}

341 342 343 344 345 346 347 348 349
	setup_pager();

	if (lr->function)
		fprintf(stdout, "<%s:%d>\n", lr->function,
			lr->start - lr->offset);
	else
		fprintf(stdout, "<%s:%d>\n", lr->file, lr->start);

	fp = fopen(lr->path, "r");
350 351 352 353 354
	if (fp == NULL) {
		pr_warning("Failed to open %s: %s\n", lr->path,
			   strerror(errno));
		return -errno;
	}
355
	/* Skip to starting line number */
356 357 358 359
	while (l < lr->start && ret >= 0)
		ret = show_one_line(fp, l++, true, false);
	if (ret < 0)
		goto end;
360 361

	list_for_each_entry(ln, &lr->line_list, list) {
362 363 364 365 366 367 368 369
		while (ln->line > l && ret >= 0)
			ret = show_one_line(fp, (l++) - lr->offset,
					    false, false);
		if (ret >= 0)
			ret = show_one_line(fp, (l++) - lr->offset,
					    false, true);
		if (ret < 0)
			goto end;
370 371 372 373
	}

	if (lr->end == INT_MAX)
		lr->end = l + NR_ADDITIONAL_LINES;
374
	while (l <= lr->end && !feof(fp) && ret >= 0)
375 376
		ret = show_one_line(fp, (l++) - lr->offset, false, false);
end:
377
	fclose(fp);
378
	return ret;
379 380 381 382
}

#else	/* !DWARF_SUPPORT */

383 384
static int kprobe_convert_to_perf_probe(struct probe_trace_point *tp,
				       struct perf_probe_point *pp)
385
{
386 387 388
	pp->function = strdup(tp->symbol);
	if (pp->function == NULL)
		return -ENOMEM;
389 390
	pp->offset = tp->offset;
	pp->retprobe = tp->retprobe;
391 392

	return 0;
393 394
}

395 396
static int try_to_find_probe_trace_events(struct perf_probe_event *pev,
				struct probe_trace_event **tevs __unused,
397
				int max_tevs __unused)
398
{
399 400 401 402
	if (perf_probe_event_need_dwarf(pev)) {
		pr_warning("Debuginfo-analysis is not supported.\n");
		return -ENOSYS;
	}
403 404 405
	return 0;
}

406
int show_line_range(struct line_range *lr __unused)
407
{
408 409
	pr_warning("Debuginfo-analysis is not supported.\n");
	return -ENOSYS;
410 411
}

412 413
#endif

414
int parse_line_range_desc(const char *arg, struct line_range *lr)
415 416 417 418 419 420 421 422 423 424
{
	const char *ptr;
	char *tmp;
	/*
	 * <Syntax>
	 * SRC:SLN[+NUM|-ELN]
	 * FUNC[:SLN[+NUM|-ELN]]
	 */
	ptr = strchr(arg, ':');
	if (ptr) {
425
		lr->start = (int)strtoul(ptr + 1, &tmp, 0);
426
		if (*tmp == '+') {
427
			lr->end = lr->start + (int)strtoul(tmp + 1, &tmp, 0);
428 429 430 431 432 433 434
			lr->end--;	/*
					 * Adjust the number of lines here.
					 * If the number of lines == 1, the
					 * the end of line should be equal to
					 * the start of line.
					 */
		} else if (*tmp == '-')
435
			lr->end = (int)strtoul(tmp + 1, &tmp, 0);
436
		else
437 438 439
			lr->end = INT_MAX;
		pr_debug("Line range is %d to %d\n", lr->start, lr->end);
		if (lr->start > lr->end) {
440
			semantic_error("Start line must be smaller"
441 442 443 444 445
				       " than end line.\n");
			return -EINVAL;
		}
		if (*tmp != '\0') {
			semantic_error("Tailing with invalid character '%d'.\n",
446
				       *tmp);
447 448
			return -EINVAL;
		}
449
		tmp = strndup(arg, (ptr - arg));
450
	} else {
451
		tmp = strdup(arg);
452 453
		lr->end = INT_MAX;
	}
454 455 456

	if (tmp == NULL)
		return -ENOMEM;
457 458 459 460 461

	if (strchr(tmp, '.'))
		lr->file = tmp;
	else
		lr->function = tmp;
462 463

	return 0;
464 465
}

466 467 468 469 470 471 472 473 474 475 476 477
/* Check the name is good for event/group */
static bool check_event_name(const char *name)
{
	if (!isalpha(*name) && *name != '_')
		return false;
	while (*++name != '\0') {
		if (!isalpha(*name) && !isdigit(*name) && *name != '_')
			return false;
	}
	return true;
}

478
/* Parse probepoint definition. */
479
static int parse_perf_probe_point(char *arg, struct perf_probe_event *pev)
480
{
481
	struct perf_probe_point *pp = &pev->point;
482 483 484 485
	char *ptr, *tmp;
	char c, nc = 0;
	/*
	 * <Syntax>
486 487
	 * perf probe [EVENT=]SRC[:LN|;PTN]
	 * perf probe [EVENT=]FUNC[@SRC][+OFFS|%return|:LN|;PAT]
488 489
	 *
	 * TODO:Group name support
490 491
	 */

492 493
	ptr = strpbrk(arg, ";=@+%");
	if (ptr && *ptr == '=') {	/* Event name */
494 495
		*ptr = '\0';
		tmp = ptr + 1;
496 497 498 499 500
		if (strchr(arg, ':')) {
			semantic_error("Group name is not supported yet.\n");
			return -ENOTSUP;
		}
		if (!check_event_name(arg)) {
501
			semantic_error("%s is bad for event name -it must "
502 503 504
				       "follow C symbol-naming rule.\n", arg);
			return -EINVAL;
		}
505 506 507
		pev->event = strdup(arg);
		if (pev->event == NULL)
			return -ENOMEM;
508
		pev->group = NULL;
509 510 511
		arg = tmp;
	}

512
	ptr = strpbrk(arg, ";:+@%");
513 514 515 516 517
	if (ptr) {
		nc = *ptr;
		*ptr++ = '\0';
	}

518 519 520 521
	tmp = strdup(arg);
	if (tmp == NULL)
		return -ENOMEM;

522
	/* Check arg is function or file and copy it */
523 524
	if (strchr(tmp, '.'))	/* File */
		pp->file = tmp;
525
	else			/* Function */
526
		pp->function = tmp;
527 528 529 530 531

	/* Parse other options */
	while (ptr) {
		arg = ptr;
		c = nc;
532
		if (c == ';') {	/* Lazy pattern must be the last part */
533 534 535
			pp->lazy_line = strdup(arg);
			if (pp->lazy_line == NULL)
				return -ENOMEM;
536 537 538
			break;
		}
		ptr = strpbrk(arg, ";:+@%");
539 540 541 542 543 544 545
		if (ptr) {
			nc = *ptr;
			*ptr++ = '\0';
		}
		switch (c) {
		case ':':	/* Line number */
			pp->line = strtoul(arg, &tmp, 0);
546
			if (*tmp != '\0') {
547
				semantic_error("There is non-digit char"
548 549 550
					       " in line number.\n");
				return -EINVAL;
			}
551 552 553
			break;
		case '+':	/* Byte offset from a symbol */
			pp->offset = strtoul(arg, &tmp, 0);
554
			if (*tmp != '\0') {
555
				semantic_error("There is non-digit character"
556 557 558
						" in offset.\n");
				return -EINVAL;
			}
559 560
			break;
		case '@':	/* File name */
561 562 563 564
			if (pp->file) {
				semantic_error("SRC@SRC is not allowed.\n");
				return -EINVAL;
			}
565 566 567
			pp->file = strdup(arg);
			if (pp->file == NULL)
				return -ENOMEM;
568 569 570 571
			break;
		case '%':	/* Probe places */
			if (strcmp(arg, "return") == 0) {
				pp->retprobe = 1;
572 573 574 575
			} else {	/* Others not supported yet */
				semantic_error("%%%s is not supported.\n", arg);
				return -ENOTSUP;
			}
576
			break;
577 578 579 580
		default:	/* Buggy case */
			pr_err("This program has a bug at %s:%d.\n",
				__FILE__, __LINE__);
			return -ENOTSUP;
581 582 583 584 585
			break;
		}
	}

	/* Exclusion check */
586
	if (pp->lazy_line && pp->line) {
587
		semantic_error("Lazy pattern can't be used with line number.");
588 589
		return -EINVAL;
	}
590

591
	if (pp->lazy_line && pp->offset) {
592
		semantic_error("Lazy pattern can't be used with offset.");
593 594
		return -EINVAL;
	}
595

596
	if (pp->line && pp->offset) {
597
		semantic_error("Offset can't be used with line number.");
598 599
		return -EINVAL;
	}
600

601
	if (!pp->line && !pp->lazy_line && pp->file && !pp->function) {
602 603
		semantic_error("File always requires line number or "
			       "lazy pattern.");
604 605
		return -EINVAL;
	}
606

607
	if (pp->offset && !pp->function) {
608
		semantic_error("Offset requires an entry function.");
609 610
		return -EINVAL;
	}
611

612
	if (pp->retprobe && !pp->function) {
613
		semantic_error("Return probe requires an entry function.");
614 615
		return -EINVAL;
	}
616

617
	if ((pp->offset || pp->line || pp->lazy_line) && pp->retprobe) {
618 619
		semantic_error("Offset/Line/Lazy pattern can't be used with "
			       "return probe.");
620 621
		return -EINVAL;
	}
622

623
	pr_debug("symbol:%s file:%s line:%d offset:%lu return:%d lazy:%s\n",
624 625
		 pp->function, pp->file, pp->line, pp->offset, pp->retprobe,
		 pp->lazy_line);
626
	return 0;
627 628
}

629
/* Parse perf-probe event argument */
630
static int parse_perf_probe_arg(char *str, struct perf_probe_arg *arg)
631
{
632
	char *tmp, *goodname;
633 634 635 636
	struct perf_probe_arg_field **fieldp;

	pr_debug("parsing arg: %s into ", str);

637 638
	tmp = strchr(str, '=');
	if (tmp) {
639 640 641
		arg->name = strndup(str, tmp - str);
		if (arg->name == NULL)
			return -ENOMEM;
642
		pr_debug("name:%s ", arg->name);
643 644 645
		str = tmp + 1;
	}

646 647 648
	tmp = strchr(str, ':');
	if (tmp) {	/* Type setting */
		*tmp = '\0';
649 650 651
		arg->type = strdup(tmp + 1);
		if (arg->type == NULL)
			return -ENOMEM;
652 653 654
		pr_debug("type:%s ", arg->type);
	}

655
	tmp = strpbrk(str, "-.[");
656 657
	if (!is_c_varname(str) || !tmp) {
		/* A variable, register, symbol or special value */
658 659 660
		arg->var = strdup(str);
		if (arg->var == NULL)
			return -ENOMEM;
661
		pr_debug("%s\n", arg->var);
662
		return 0;
663 664
	}

665
	/* Structure fields or array element */
666 667 668
	arg->var = strndup(str, tmp - str);
	if (arg->var == NULL)
		return -ENOMEM;
669
	goodname = arg->var;
670
	pr_debug("%s, ", arg->var);
671 672 673
	fieldp = &arg->field;

	do {
674 675 676
		*fieldp = zalloc(sizeof(struct perf_probe_arg_field));
		if (*fieldp == NULL)
			return -ENOMEM;
677 678 679
		if (*tmp == '[') {	/* Array */
			str = tmp;
			(*fieldp)->index = strtol(str + 1, &tmp, 0);
680
			(*fieldp)->ref = true;
681 682 683 684 685 686 687 688 689 690 691 692 693 694 695 696 697 698 699 700 701
			if (*tmp != ']' || tmp == str + 1) {
				semantic_error("Array index must be a"
						" number.\n");
				return -EINVAL;
			}
			tmp++;
			if (*tmp == '\0')
				tmp = NULL;
		} else {		/* Structure */
			if (*tmp == '.') {
				str = tmp + 1;
				(*fieldp)->ref = false;
			} else if (tmp[1] == '>') {
				str = tmp + 2;
				(*fieldp)->ref = true;
			} else {
				semantic_error("Argument parse error: %s\n",
					       str);
				return -EINVAL;
			}
			tmp = strpbrk(str, "-.[");
702
		}
703
		if (tmp) {
704 705 706
			(*fieldp)->name = strndup(str, tmp - str);
			if ((*fieldp)->name == NULL)
				return -ENOMEM;
707 708
			if (*str != '[')
				goodname = (*fieldp)->name;
709 710 711 712
			pr_debug("%s(%d), ", (*fieldp)->name, (*fieldp)->ref);
			fieldp = &(*fieldp)->next;
		}
	} while (tmp);
713 714 715
	(*fieldp)->name = strdup(str);
	if ((*fieldp)->name == NULL)
		return -ENOMEM;
716 717
	if (*str != '[')
		goodname = (*fieldp)->name;
718
	pr_debug("%s(%d)\n", (*fieldp)->name, (*fieldp)->ref);
719

720
	/* If no name is specified, set the last field name (not array index)*/
721
	if (!arg->name) {
722
		arg->name = strdup(goodname);
723 724 725
		if (arg->name == NULL)
			return -ENOMEM;
	}
726
	return 0;
727 728
}

729
/* Parse perf-probe event command */
730
int parse_perf_probe_command(const char *cmd, struct perf_probe_event *pev)
731
{
732
	char **argv;
733
	int argc, i, ret = 0;
734

735
	argv = argv_split(cmd, &argc);
736 737 738 739 740 741 742 743 744
	if (!argv) {
		pr_debug("Failed to split arguments.\n");
		return -ENOMEM;
	}
	if (argc - 1 > MAX_PROBE_ARGS) {
		semantic_error("Too many probe arguments (%d).\n", argc - 1);
		ret = -ERANGE;
		goto out;
	}
745
	/* Parse probe point */
746 747 748
	ret = parse_perf_probe_point(argv[0], pev);
	if (ret < 0)
		goto out;
749

750
	/* Copy arguments and ensure return probe has no C argument */
751
	pev->nargs = argc - 1;
752 753 754 755 756
	pev->args = zalloc(sizeof(struct perf_probe_arg) * pev->nargs);
	if (pev->args == NULL) {
		ret = -ENOMEM;
		goto out;
	}
757 758 759 760
	for (i = 0; i < pev->nargs && ret >= 0; i++) {
		ret = parse_perf_probe_arg(argv[i + 1], &pev->args[i]);
		if (ret >= 0 &&
		    is_c_varname(pev->args[i].var) && pev->point.retprobe) {
761
			semantic_error("You can't specify local variable for"
762 763 764
				       " kretprobe.\n");
			ret = -EINVAL;
		}
765
	}
766
out:
767
	argv_free(argv);
768 769

	return ret;
770 771
}

772 773 774 775 776 777 778 779 780
/* Return true if this perf_probe_event requires debuginfo */
bool perf_probe_event_need_dwarf(struct perf_probe_event *pev)
{
	int i;

	if (pev->point.file || pev->point.line || pev->point.lazy_line)
		return true;

	for (i = 0; i < pev->nargs; i++)
781
		if (is_c_varname(pev->args[i].var))
782 783 784 785 786
			return true;

	return false;
}

787 788 789
/* Parse probe_events event into struct probe_point */
static int parse_probe_trace_command(const char *cmd,
					struct probe_trace_event *tev)
790
{
791
	struct probe_trace_point *tp = &tev->point;
792 793 794 795 796
	char pr;
	char *p;
	int ret, i, argc;
	char **argv;

797
	pr_debug("Parsing probe_events: %s\n", cmd);
798
	argv = argv_split(cmd, &argc);
799 800 801 802 803 804 805 806 807
	if (!argv) {
		pr_debug("Failed to split arguments.\n");
		return -ENOMEM;
	}
	if (argc < 2) {
		semantic_error("Too few probe arguments.\n");
		ret = -ERANGE;
		goto out;
	}
808 809

	/* Scan event and group name. */
810
	ret = sscanf(argv[0], "%c:%a[^/ \t]/%a[^ \t]",
811 812
		     &pr, (float *)(void *)&tev->group,
		     (float *)(void *)&tev->event);
813 814 815 816 817
	if (ret != 3) {
		semantic_error("Failed to parse event name: %s\n", argv[0]);
		ret = -EINVAL;
		goto out;
	}
818
	pr_debug("Group:%s Event:%s probe:%c\n", tev->group, tev->event, pr);
819

820
	tp->retprobe = (pr == 'r');
821 822

	/* Scan function name and offset */
823 824
	ret = sscanf(argv[1], "%a[^+]+%lu", (float *)(void *)&tp->symbol,
		     &tp->offset);
825
	if (ret == 1)
826
		tp->offset = 0;
827

828
	tev->nargs = argc - 2;
829
	tev->args = zalloc(sizeof(struct probe_trace_arg) * tev->nargs);
830 831 832 833
	if (tev->args == NULL) {
		ret = -ENOMEM;
		goto out;
	}
834
	for (i = 0; i < tev->nargs; i++) {
835 836
		p = strchr(argv[i + 2], '=');
		if (p)	/* We don't need which register is assigned. */
837 838 839
			*p++ = '\0';
		else
			p = argv[i + 2];
840
		tev->args[i].name = strdup(argv[i + 2]);
841
		/* TODO: parse regs and offset */
842 843 844 845 846
		tev->args[i].value = strdup(p);
		if (tev->args[i].name == NULL || tev->args[i].value == NULL) {
			ret = -ENOMEM;
			goto out;
		}
847
	}
848 849
	ret = 0;
out:
850
	argv_free(argv);
851
	return ret;
852 853
}

854 855 856 857 858 859 860
/* Compose only probe arg */
int synthesize_perf_probe_arg(struct perf_probe_arg *pa, char *buf, size_t len)
{
	struct perf_probe_arg_field *field = pa->field;
	int ret;
	char *tmp = buf;

861 862 863 864
	if (pa->name && pa->var)
		ret = e_snprintf(tmp, len, "%s=%s", pa->name, pa->var);
	else
		ret = e_snprintf(tmp, len, "%s", pa->name ? pa->name : pa->var);
865 866 867 868 869 870
	if (ret <= 0)
		goto error;
	tmp += ret;
	len -= ret;

	while (field) {
871 872 873 874 875
		if (field->name[0] == '[')
			ret = e_snprintf(tmp, len, "%s", field->name);
		else
			ret = e_snprintf(tmp, len, "%s%s",
					 field->ref ? "->" : ".", field->name);
876 877 878 879 880 881
		if (ret <= 0)
			goto error;
		tmp += ret;
		len -= ret;
		field = field->next;
	}
882 883 884 885 886 887 888 889 890

	if (pa->type) {
		ret = e_snprintf(tmp, len, ":%s", pa->type);
		if (ret <= 0)
			goto error;
		tmp += ret;
		len -= ret;
	}

891 892
	return tmp - buf;
error:
893 894 895
	pr_debug("Failed to synthesize perf probe argument: %s",
		 strerror(-ret));
	return ret;
896 897
}

898 899
/* Compose only probe point (not argument) */
static char *synthesize_perf_probe_point(struct perf_probe_point *pp)
900
{
901 902 903
	char *buf, *tmp;
	char offs[32] = "", line[32] = "", file[32] = "";
	int ret, len;
904

905 906 907 908 909
	buf = zalloc(MAX_CMDLEN);
	if (buf == NULL) {
		ret = -ENOMEM;
		goto error;
	}
910
	if (pp->offset) {
911
		ret = e_snprintf(offs, 32, "+%lu", pp->offset);
912 913 914 915
		if (ret <= 0)
			goto error;
	}
	if (pp->line) {
916 917 918 919 920
		ret = e_snprintf(line, 32, ":%d", pp->line);
		if (ret <= 0)
			goto error;
	}
	if (pp->file) {
921
		len = strlen(pp->file) - 31;
922 923 924 925
		if (len < 0)
			len = 0;
		tmp = strchr(pp->file + len, '/');
		if (!tmp)
926
			tmp = pp->file + len;
927
		ret = e_snprintf(file, 32, "@%s", tmp + 1);
928 929 930 931 932
		if (ret <= 0)
			goto error;
	}

	if (pp->function)
933 934 935
		ret = e_snprintf(buf, MAX_CMDLEN, "%s%s%s%s%s", pp->function,
				 offs, pp->retprobe ? "%return" : "", line,
				 file);
936
	else
937
		ret = e_snprintf(buf, MAX_CMDLEN, "%s%s", file, line);
938 939 940 941
	if (ret <= 0)
		goto error;

	return buf;
942
error:
943 944
	pr_debug("Failed to synthesize perf probe point: %s",
		 strerror(-ret));
945 946
	if (buf)
		free(buf);
947
	return NULL;
948 949
}

950 951
#if 0
char *synthesize_perf_probe_command(struct perf_probe_event *pev)
952 953 954 955
{
	char *buf;
	int i, len, ret;

956 957 958
	buf = synthesize_perf_probe_point(&pev->point);
	if (!buf)
		return NULL;
959

960 961
	len = strlen(buf);
	for (i = 0; i < pev->nargs; i++) {
962
		ret = e_snprintf(&buf[len], MAX_CMDLEN - len, " %s",
963 964 965 966 967
				 pev->args[i].name);
		if (ret <= 0) {
			free(buf);
			return NULL;
		}
968 969 970
		len += ret;
	}

971 972 973 974
	return buf;
}
#endif

975
static int __synthesize_probe_trace_arg_ref(struct probe_trace_arg_ref *ref,
976 977 978 979 980
					     char **buf, size_t *buflen,
					     int depth)
{
	int ret;
	if (ref->next) {
981
		depth = __synthesize_probe_trace_arg_ref(ref->next, buf,
982 983 984 985 986 987 988 989 990 991 992 993 994 995
							 buflen, depth + 1);
		if (depth < 0)
			goto out;
	}

	ret = e_snprintf(*buf, *buflen, "%+ld(", ref->offset);
	if (ret < 0)
		depth = ret;
	else {
		*buf += ret;
		*buflen -= ret;
	}
out:
	return depth;
996 997 998

}

999
static int synthesize_probe_trace_arg(struct probe_trace_arg *arg,
1000
				       char *buf, size_t buflen)
1001
{
1002
	struct probe_trace_arg_ref *ref = arg->ref;
1003 1004 1005 1006 1007 1008 1009 1010 1011 1012 1013 1014 1015
	int ret, depth = 0;
	char *tmp = buf;

	/* Argument name or separator */
	if (arg->name)
		ret = e_snprintf(buf, buflen, " %s=", arg->name);
	else
		ret = e_snprintf(buf, buflen, " ");
	if (ret < 0)
		return ret;
	buf += ret;
	buflen -= ret;

1016 1017 1018 1019
	/* Special case: @XXX */
	if (arg->value[0] == '@' && arg->ref)
			ref = ref->next;

1020
	/* Dereferencing arguments */
1021
	if (ref) {
1022
		depth = __synthesize_probe_trace_arg_ref(ref, &buf,
1023 1024 1025 1026 1027 1028
							  &buflen, 1);
		if (depth < 0)
			return depth;
	}

	/* Print argument value */
1029 1030 1031 1032 1033
	if (arg->value[0] == '@' && arg->ref)
		ret = e_snprintf(buf, buflen, "%s%+ld", arg->value,
				 arg->ref->offset);
	else
		ret = e_snprintf(buf, buflen, "%s", arg->value);
1034 1035 1036 1037 1038 1039 1040 1041 1042 1043 1044 1045 1046
	if (ret < 0)
		return ret;
	buf += ret;
	buflen -= ret;

	/* Closing */
	while (depth--) {
		ret = e_snprintf(buf, buflen, ")");
		if (ret < 0)
			return ret;
		buf += ret;
		buflen -= ret;
	}
1047 1048 1049 1050 1051 1052 1053
	/* Print argument type */
	if (arg->type) {
		ret = e_snprintf(buf, buflen, ":%s", arg->type);
		if (ret <= 0)
			return ret;
		buf += ret;
	}
1054 1055 1056 1057

	return buf - tmp;
}

1058
char *synthesize_probe_trace_command(struct probe_trace_event *tev)
1059
{
1060
	struct probe_trace_point *tp = &tev->point;
1061 1062 1063
	char *buf;
	int i, len, ret;

1064 1065 1066 1067
	buf = zalloc(MAX_CMDLEN);
	if (buf == NULL)
		return NULL;

1068 1069 1070 1071 1072
	len = e_snprintf(buf, MAX_CMDLEN, "%c:%s/%s %s+%lu",
			 tp->retprobe ? 'r' : 'p',
			 tev->group, tev->event,
			 tp->symbol, tp->offset);
	if (len <= 0)
1073 1074
		goto error;

1075
	for (i = 0; i < tev->nargs; i++) {
1076
		ret = synthesize_probe_trace_arg(&tev->args[i], buf + len,
1077
						  MAX_CMDLEN - len);
1078
		if (ret <= 0)
1079 1080 1081 1082
			goto error;
		len += ret;
	}

1083
	return buf;
1084
error:
1085 1086 1087
	free(buf);
	return NULL;
}
1088

1089
static int convert_to_perf_probe_event(struct probe_trace_event *tev,
1090
				struct perf_probe_event *pev)
1091
{
1092
	char buf[64] = "";
1093
	int i, ret;
1094

1095
	/* Convert event/group name */
1096 1097 1098 1099
	pev->event = strdup(tev->event);
	pev->group = strdup(tev->group);
	if (pev->event == NULL || pev->group == NULL)
		return -ENOMEM;
1100

1101
	/* Convert trace_point to probe_point */
1102
	ret = kprobe_convert_to_perf_probe(&tev->point, &pev->point);
1103 1104
	if (ret < 0)
		return ret;
1105

1106 1107
	/* Convert trace_arg to probe_arg */
	pev->nargs = tev->nargs;
1108 1109 1110
	pev->args = zalloc(sizeof(struct perf_probe_arg) * pev->nargs);
	if (pev->args == NULL)
		return -ENOMEM;
1111
	for (i = 0; i < tev->nargs && ret >= 0; i++) {
1112
		if (tev->args[i].name)
1113
			pev->args[i].name = strdup(tev->args[i].name);
1114
		else {
1115
			ret = synthesize_probe_trace_arg(&tev->args[i],
1116
							  buf, 64);
1117
			pev->args[i].name = strdup(buf);
1118
		}
1119 1120 1121
		if (pev->args[i].name == NULL && ret >= 0)
			ret = -ENOMEM;
	}
1122 1123 1124 1125 1126

	if (ret < 0)
		clear_perf_probe_event(pev);

	return ret;
1127 1128 1129 1130 1131
}

void clear_perf_probe_event(struct perf_probe_event *pev)
{
	struct perf_probe_point *pp = &pev->point;
1132
	struct perf_probe_arg_field *field, *next;
1133 1134 1135 1136 1137 1138 1139 1140 1141 1142 1143 1144
	int i;

	if (pev->event)
		free(pev->event);
	if (pev->group)
		free(pev->group);
	if (pp->file)
		free(pp->file);
	if (pp->function)
		free(pp->function);
	if (pp->lazy_line)
		free(pp->lazy_line);
1145
	for (i = 0; i < pev->nargs; i++) {
1146 1147
		if (pev->args[i].name)
			free(pev->args[i].name);
1148 1149
		if (pev->args[i].var)
			free(pev->args[i].var);
1150 1151
		if (pev->args[i].type)
			free(pev->args[i].type);
1152 1153 1154 1155 1156 1157 1158 1159 1160
		field = pev->args[i].field;
		while (field) {
			next = field->next;
			if (field->name)
				free(field->name);
			free(field);
			field = next;
		}
	}
1161 1162 1163 1164 1165
	if (pev->args)
		free(pev->args);
	memset(pev, 0, sizeof(*pev));
}

1166
static void clear_probe_trace_event(struct probe_trace_event *tev)
1167
{
1168
	struct probe_trace_arg_ref *ref, *next;
1169 1170 1171 1172 1173 1174 1175 1176 1177 1178 1179 1180 1181
	int i;

	if (tev->event)
		free(tev->event);
	if (tev->group)
		free(tev->group);
	if (tev->point.symbol)
		free(tev->point.symbol);
	for (i = 0; i < tev->nargs; i++) {
		if (tev->args[i].name)
			free(tev->args[i].name);
		if (tev->args[i].value)
			free(tev->args[i].value);
1182 1183
		if (tev->args[i].type)
			free(tev->args[i].type);
1184 1185 1186 1187 1188 1189 1190 1191 1192 1193
		ref = tev->args[i].ref;
		while (ref) {
			next = ref->next;
			free(ref);
			ref = next;
		}
	}
	if (tev->args)
		free(tev->args);
	memset(tev, 0, sizeof(*tev));
1194 1195
}

1196
static int open_kprobe_events(bool readwrite)
1197 1198
{
	char buf[PATH_MAX];
1199
	const char *__debugfs;
1200 1201
	int ret;

1202 1203 1204 1205 1206 1207 1208
	__debugfs = debugfs_find_mountpoint();
	if (__debugfs == NULL) {
		pr_warning("Debugfs is not mounted.\n");
		return -ENOENT;
	}

	ret = e_snprintf(buf, PATH_MAX, "%stracing/kprobe_events", __debugfs);
1209
	if (ret >= 0) {
1210
		pr_debug("Opening %s write=%d\n", buf, readwrite);
1211 1212 1213 1214 1215
		if (readwrite && !probe_event_dry_run)
			ret = open(buf, O_RDWR, O_APPEND);
		else
			ret = open(buf, O_RDONLY, 0);
	}
1216

1217 1218
	if (ret < 0) {
		if (errno == ENOENT)
1219 1220
			pr_warning("kprobe_events file does not exist - please"
				 " rebuild kernel with CONFIG_KPROBE_EVENT.\n");
1221
		else
1222 1223
			pr_warning("Failed to open kprobe_events file: %s\n",
				   strerror(errno));
1224 1225 1226 1227 1228
	}
	return ret;
}

/* Get raw string list of current kprobe_events */
1229
static struct strlist *get_probe_trace_command_rawlist(int fd)
1230 1231 1232 1233 1234 1235 1236 1237 1238 1239 1240 1241 1242 1243 1244 1245 1246 1247 1248
{
	int ret, idx;
	FILE *fp;
	char buf[MAX_CMDLEN];
	char *p;
	struct strlist *sl;

	sl = strlist__new(true, NULL);

	fp = fdopen(dup(fd), "r");
	while (!feof(fp)) {
		p = fgets(buf, MAX_CMDLEN, fp);
		if (!p)
			break;

		idx = strlen(p) - 1;
		if (p[idx] == '\n')
			p[idx] = '\0';
		ret = strlist__add(sl, buf);
1249 1250 1251 1252 1253
		if (ret < 0) {
			pr_debug("strlist__add failed: %s\n", strerror(-ret));
			strlist__delete(sl);
			return NULL;
		}
1254 1255 1256 1257 1258 1259
	}
	fclose(fp);

	return sl;
}

1260
/* Show an event */
1261
static int show_perf_probe_event(struct perf_probe_event *pev)
1262
{
1263
	int i, ret;
1264
	char buf[128];
1265
	char *place;
1266

1267 1268
	/* Synthesize only event probe point */
	place = synthesize_perf_probe_point(&pev->point);
1269 1270
	if (!place)
		return -EINVAL;
1271 1272

	ret = e_snprintf(buf, 128, "%s:%s", pev->group, pev->event);
1273
	if (ret < 0)
1274 1275
		return ret;

1276
	printf("  %-20s (on %s", buf, place);
1277

1278
	if (pev->nargs > 0) {
1279
		printf(" with");
1280
		for (i = 0; i < pev->nargs; i++) {
1281 1282 1283 1284
			ret = synthesize_perf_probe_arg(&pev->args[i],
							buf, 128);
			if (ret < 0)
				break;
1285 1286
			printf(" %s", buf);
		}
1287 1288
	}
	printf(")\n");
1289
	free(place);
1290
	return ret;
1291 1292
}

1293
/* List up current perf-probe events */
1294
int show_perf_probe_events(void)
1295
{
1296
	int fd, ret;
1297
	struct probe_trace_event tev;
1298
	struct perf_probe_event pev;
1299 1300 1301
	struct strlist *rawlist;
	struct str_node *ent;

1302
	setup_pager();
1303 1304 1305
	ret = init_vmlinux();
	if (ret < 0)
		return ret;
1306 1307 1308

	memset(&tev, 0, sizeof(tev));
	memset(&pev, 0, sizeof(pev));
1309

1310
	fd = open_kprobe_events(false);
1311 1312 1313
	if (fd < 0)
		return fd;

1314
	rawlist = get_probe_trace_command_rawlist(fd);
1315
	close(fd);
1316 1317
	if (!rawlist)
		return -ENOENT;
1318

1319
	strlist__for_each(ent, rawlist) {
1320
		ret = parse_probe_trace_command(ent->s, &tev);
1321 1322 1323 1324 1325
		if (ret >= 0) {
			ret = convert_to_perf_probe_event(&tev, &pev);
			if (ret >= 0)
				ret = show_perf_probe_event(&pev);
		}
1326
		clear_perf_probe_event(&pev);
1327
		clear_probe_trace_event(&tev);
1328 1329
		if (ret < 0)
			break;
1330 1331
	}
	strlist__delete(rawlist);
1332 1333

	return ret;
1334 1335
}

1336
/* Get current perf-probe event names */
1337
static struct strlist *get_probe_trace_event_names(int fd, bool include_group)
1338
{
1339
	char buf[128];
1340 1341
	struct strlist *sl, *rawlist;
	struct str_node *ent;
1342
	struct probe_trace_event tev;
1343
	int ret = 0;
1344

1345
	memset(&tev, 0, sizeof(tev));
1346
	rawlist = get_probe_trace_command_rawlist(fd);
1347
	sl = strlist__new(true, NULL);
1348
	strlist__for_each(ent, rawlist) {
1349
		ret = parse_probe_trace_command(ent->s, &tev);
1350 1351
		if (ret < 0)
			break;
1352
		if (include_group) {
1353 1354 1355 1356
			ret = e_snprintf(buf, 128, "%s:%s", tev.group,
					tev.event);
			if (ret >= 0)
				ret = strlist__add(sl, buf);
1357
		} else
1358
			ret = strlist__add(sl, tev.event);
1359
		clear_probe_trace_event(&tev);
1360 1361
		if (ret < 0)
			break;
1362 1363 1364
	}
	strlist__delete(rawlist);

1365 1366 1367 1368
	if (ret < 0) {
		strlist__delete(sl);
		return NULL;
	}
1369 1370 1371
	return sl;
}

1372
static int write_probe_trace_event(int fd, struct probe_trace_event *tev)
1373
{
1374
	int ret = 0;
1375
	char *buf = synthesize_probe_trace_command(tev);
1376

1377
	if (!buf) {
1378
		pr_debug("Failed to synthesize probe trace event.\n");
1379 1380 1381
		return -EINVAL;
	}

1382
	pr_debug("Writing event: %s\n", buf);
1383 1384 1385
	if (!probe_event_dry_run) {
		ret = write(fd, buf, strlen(buf));
		if (ret <= 0)
1386 1387
			pr_warning("Failed to write event: %s\n",
				   strerror(errno));
1388
	}
1389
	free(buf);
1390
	return ret;
1391 1392
}

1393 1394
static int get_new_event_name(char *buf, size_t len, const char *base,
			      struct strlist *namelist, bool allow_suffix)
1395 1396
{
	int i, ret;
1397 1398 1399

	/* Try no suffix */
	ret = e_snprintf(buf, len, "%s", base);
1400 1401 1402 1403
	if (ret < 0) {
		pr_debug("snprintf() failed: %s\n", strerror(-ret));
		return ret;
	}
1404
	if (!strlist__has_entry(namelist, buf))
1405
		return 0;
1406

1407 1408 1409
	if (!allow_suffix) {
		pr_warning("Error: event \"%s\" already exists. "
			   "(Use -f to force duplicates.)\n", base);
1410
		return -EEXIST;
1411 1412
	}

1413 1414
	/* Try to add suffix */
	for (i = 1; i < MAX_EVENT_INDEX; i++) {
1415
		ret = e_snprintf(buf, len, "%s_%d", base, i);
1416 1417 1418 1419
		if (ret < 0) {
			pr_debug("snprintf() failed: %s\n", strerror(-ret));
			return ret;
		}
1420 1421 1422
		if (!strlist__has_entry(namelist, buf))
			break;
	}
1423 1424 1425 1426 1427 1428
	if (i == MAX_EVENT_INDEX) {
		pr_warning("Too many events are on the same function.\n");
		ret = -ERANGE;
	}

	return ret;
1429 1430
}

1431 1432
static int __add_probe_trace_events(struct perf_probe_event *pev,
				     struct probe_trace_event *tevs,
1433
				     int ntevs, bool allow_suffix)
1434
{
1435
	int i, fd, ret;
1436
	struct probe_trace_event *tev = NULL;
1437 1438
	char buf[64];
	const char *event, *group;
1439
	struct strlist *namelist;
1440

1441
	fd = open_kprobe_events(true);
1442 1443
	if (fd < 0)
		return fd;
1444
	/* Get current event names */
1445
	namelist = get_probe_trace_event_names(fd, false);
1446 1447 1448 1449
	if (!namelist) {
		pr_debug("Failed to get current event list.\n");
		return -EIO;
	}
1450

1451
	ret = 0;
1452
	printf("Add new event%s\n", (ntevs > 1) ? "s:" : ":");
1453
	for (i = 0; i < ntevs; i++) {
1454 1455 1456 1457 1458 1459 1460 1461 1462 1463 1464 1465 1466 1467
		tev = &tevs[i];
		if (pev->event)
			event = pev->event;
		else
			if (pev->point.function)
				event = pev->point.function;
			else
				event = tev->point.symbol;
		if (pev->group)
			group = pev->group;
		else
			group = PERFPROBE_GROUP;

		/* Get an unused new event name */
1468 1469 1470 1471
		ret = get_new_event_name(buf, 64, event,
					 namelist, allow_suffix);
		if (ret < 0)
			break;
1472 1473
		event = buf;

1474 1475 1476 1477 1478 1479
		tev->event = strdup(event);
		tev->group = strdup(group);
		if (tev->event == NULL || tev->group == NULL) {
			ret = -ENOMEM;
			break;
		}
1480
		ret = write_probe_trace_event(fd, tev);
1481 1482
		if (ret < 0)
			break;
1483 1484 1485 1486 1487 1488 1489 1490 1491 1492 1493 1494 1495 1496 1497 1498 1499 1500 1501 1502
		/* Add added event name to namelist */
		strlist__add(namelist, event);

		/* Trick here - save current event/group */
		event = pev->event;
		group = pev->group;
		pev->event = tev->event;
		pev->group = tev->group;
		show_perf_probe_event(pev);
		/* Trick here - restore current event/group */
		pev->event = (char *)event;
		pev->group = (char *)group;

		/*
		 * Probes after the first probe which comes from same
		 * user input are always allowed to add suffix, because
		 * there might be several addresses corresponding to
		 * one code line.
		 */
		allow_suffix = true;
1503
	}
1504 1505 1506 1507 1508 1509 1510

	if (ret >= 0) {
		/* Show how to use the event. */
		printf("\nYou can now use it on all perf tools, such as:\n\n");
		printf("\tperf record -e %s:%s -aR sleep 1\n\n", tev->group,
			 tev->event);
	}
1511

1512
	strlist__delete(namelist);
1513
	close(fd);
1514
	return ret;
1515
}
1516

1517 1518
static int convert_to_probe_trace_events(struct perf_probe_event *pev,
					  struct probe_trace_event **tevs,
1519
					  int max_tevs)
1520 1521
{
	struct symbol *sym;
1522
	int ret = 0, i;
1523
	struct probe_trace_event *tev;
1524

1525
	/* Convert perf_probe_event with debuginfo */
1526
	ret = try_to_find_probe_trace_events(pev, tevs, max_tevs);
1527 1528
	if (ret != 0)
		return ret;
1529

1530
	/* Allocate trace event buffer */
1531
	tev = *tevs = zalloc(sizeof(struct probe_trace_event));
1532 1533
	if (tev == NULL)
		return -ENOMEM;
1534 1535

	/* Copy parameters */
1536 1537 1538 1539 1540
	tev->point.symbol = strdup(pev->point.function);
	if (tev->point.symbol == NULL) {
		ret = -ENOMEM;
		goto error;
	}
1541 1542 1543
	tev->point.offset = pev->point.offset;
	tev->nargs = pev->nargs;
	if (tev->nargs) {
1544
		tev->args = zalloc(sizeof(struct probe_trace_arg)
1545 1546
				   * tev->nargs);
		if (tev->args == NULL) {
1547 1548
			ret = -ENOMEM;
			goto error;
1549
		}
1550
		for (i = 0; i < tev->nargs; i++) {
1551 1552 1553 1554 1555 1556 1557 1558 1559 1560 1561 1562 1563 1564 1565 1566 1567 1568 1569
			if (pev->args[i].name) {
				tev->args[i].name = strdup(pev->args[i].name);
				if (tev->args[i].name == NULL) {
					ret = -ENOMEM;
					goto error;
				}
			}
			tev->args[i].value = strdup(pev->args[i].var);
			if (tev->args[i].value == NULL) {
				ret = -ENOMEM;
				goto error;
			}
			if (pev->args[i].type) {
				tev->args[i].type = strdup(pev->args[i].type);
				if (tev->args[i].type == NULL) {
					ret = -ENOMEM;
					goto error;
				}
			}
1570
		}
1571 1572 1573
	}

	/* Currently just checking function name from symbol map */
1574
	sym = map__find_symbol_by_name(machine.vmlinux_maps[MAP__FUNCTION],
1575
				       tev->point.symbol, NULL);
1576 1577 1578
	if (!sym) {
		pr_warning("Kernel symbol \'%s\' not found.\n",
			   tev->point.symbol);
1579 1580 1581
		ret = -ENOENT;
		goto error;
	}
1582

1583 1584
	return 1;
error:
1585
	clear_probe_trace_event(tev);
1586 1587
	free(tev);
	*tevs = NULL;
1588
	return ret;
1589 1590 1591 1592
}

struct __event_package {
	struct perf_probe_event		*pev;
1593
	struct probe_trace_event	*tevs;
1594 1595 1596
	int				ntevs;
};

1597
int add_perf_probe_events(struct perf_probe_event *pevs, int npevs,
1598
			  bool force_add, int max_tevs)
1599
{
1600
	int i, j, ret;
1601 1602
	struct __event_package *pkgs;

1603 1604 1605
	pkgs = zalloc(sizeof(struct __event_package) * npevs);
	if (pkgs == NULL)
		return -ENOMEM;
1606 1607

	/* Init vmlinux path */
1608
	ret = init_vmlinux();
1609 1610
	if (ret < 0) {
		free(pkgs);
1611
		return ret;
1612
	}
1613 1614 1615 1616 1617

	/* Loop 1: convert all events */
	for (i = 0; i < npevs; i++) {
		pkgs[i].pev = &pevs[i];
		/* Convert with or without debuginfo */
1618
		ret  = convert_to_probe_trace_events(pkgs[i].pev,
1619
						      &pkgs[i].tevs, max_tevs);
1620 1621 1622
		if (ret < 0)
			goto end;
		pkgs[i].ntevs = ret;
1623 1624
	}

1625
	/* Loop 2: add all events */
1626
	for (i = 0; i < npevs && ret >= 0; i++)
1627
		ret = __add_probe_trace_events(pkgs[i].pev, pkgs[i].tevs,
1628 1629
						pkgs[i].ntevs, force_add);
end:
1630 1631
	/* Loop 3: cleanup and free trace events  */
	for (i = 0; i < npevs; i++) {
1632
		for (j = 0; j < pkgs[i].ntevs; j++)
1633
			clear_probe_trace_event(&pkgs[i].tevs[j]);
1634 1635 1636
		free(pkgs[i].tevs);
	}
	free(pkgs);
1637 1638

	return ret;
1639 1640
}

1641
static int __del_trace_probe_event(int fd, struct str_node *ent)
1642 1643 1644
{
	char *p;
	char buf[128];
1645
	int ret;
1646

1647
	/* Convert from perf-probe event to trace-probe event */
1648 1649 1650 1651
	ret = e_snprintf(buf, 128, "-:%s", ent->s);
	if (ret < 0)
		goto error;

1652
	p = strchr(buf + 2, ':');
1653 1654 1655 1656 1657 1658
	if (!p) {
		pr_debug("Internal error: %s should have ':' but not.\n",
			 ent->s);
		ret = -ENOTSUP;
		goto error;
	}
1659 1660
	*p = '/';

1661 1662
	pr_debug("Writing event: %s\n", buf);
	ret = write(fd, buf, strlen(buf));
1663 1664 1665
	if (ret < 0)
		goto error;

1666
	printf("Remove event: %s\n", ent->s);
1667 1668 1669 1670
	return 0;
error:
	pr_warning("Failed to delete event: %s\n", strerror(-ret));
	return ret;
1671 1672
}

1673
static int del_trace_probe_event(int fd, const char *group,
1674
				  const char *event, struct strlist *namelist)
1675 1676
{
	char buf[128];
1677
	struct str_node *ent, *n;
1678
	int found = 0, ret = 0;
1679

1680 1681 1682 1683 1684
	ret = e_snprintf(buf, 128, "%s:%s", group, event);
	if (ret < 0) {
		pr_err("Failed to copy event.");
		return ret;
	}
1685

1686 1687 1688 1689
	if (strpbrk(buf, "*?")) { /* Glob-exp */
		strlist__for_each_safe(ent, n, namelist)
			if (strglobmatch(ent->s, buf)) {
				found++;
1690
				ret = __del_trace_probe_event(fd, ent);
1691 1692
				if (ret < 0)
					break;
1693 1694 1695 1696 1697 1698
				strlist__remove(namelist, ent);
			}
	} else {
		ent = strlist__find(namelist, buf);
		if (ent) {
			found++;
1699
			ret = __del_trace_probe_event(fd, ent);
1700 1701
			if (ret >= 0)
				strlist__remove(namelist, ent);
1702 1703
		}
	}
1704 1705 1706 1707
	if (found == 0 && ret >= 0)
		pr_info("Info: Event \"%s\" does not exist.\n", buf);

	return ret;
1708 1709
}

1710
int del_perf_probe_events(struct strlist *dellist)
1711
{
1712
	int fd, ret = 0;
1713 1714 1715 1716 1717
	const char *group, *event;
	char *p, *str;
	struct str_node *ent;
	struct strlist *namelist;

1718
	fd = open_kprobe_events(true);
1719 1720 1721
	if (fd < 0)
		return fd;

1722
	/* Get current event names */
1723
	namelist = get_probe_trace_event_names(fd, true);
1724 1725
	if (namelist == NULL)
		return -EINVAL;
1726

1727
	strlist__for_each(ent, dellist) {
1728 1729 1730 1731 1732
		str = strdup(ent->s);
		if (str == NULL) {
			ret = -ENOMEM;
			break;
		}
1733
		pr_debug("Parsing: %s\n", str);
1734 1735 1736 1737 1738 1739
		p = strchr(str, ':');
		if (p) {
			group = str;
			*p = '\0';
			event = p + 1;
		} else {
1740
			group = "*";
1741 1742
			event = str;
		}
1743
		pr_debug("Group: %s, Event: %s\n", group, event);
1744
		ret = del_trace_probe_event(fd, group, event, namelist);
1745
		free(str);
1746 1747
		if (ret < 0)
			break;
1748 1749 1750
	}
	strlist__delete(namelist);
	close(fd);
1751 1752

	return ret;
1753 1754
}