probe-event.c 38.3 KB
Newer Older
1 2 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
/*
 * probe-event.c : perf-probe definition to kprobe_events format converter
 *
 * 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
/* Convert trace point to probe point with debuginfo */
static int convert_to_perf_probe_point(struct kprobe_trace_point *tp,
				       struct perf_probe_point *pp)
126 127
{
	struct symbol *sym;
128
	int fd, ret = -ENOENT;
129

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

	return 0;
151 152 153 154
}

/* Try to find perf_probe_event with debuginfo */
static int try_to_find_kprobe_trace_events(struct perf_probe_event *pev,
155 156
					   struct kprobe_trace_event **tevs,
					   int max_tevs)
157 158 159 160 161 162
{
	bool need_dwarf = perf_probe_event_need_dwarf(pev);
	int fd, ntevs;

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

	/* Searching trace events corresponding to probe event */
172
	ntevs = find_kprobe_trace_events(fd, pev, tevs, max_tevs);
173 174
	close(fd);

175 176
	if (ntevs > 0) {	/* Succeeded to find trace events */
		pr_debug("find %d kprobe_trace_events.\n", ntevs);
177
		return ntevs;
178
	}
179

180 181 182 183 184 185
	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 */
186 187 188 189 190 191 192 193
	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;
		}
194
	}
195
	return ntevs;
196 197
}

198 199 200 201 202 203
/*
 * 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.
 */
204 205
static int get_real_path(const char *raw_path, const char *comp_dir,
			 char **new_path)
206
{
207 208 209 210 211 212 213 214 215 216 217 218 219
	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;
		}
220 221
	}

222
	*new_path = malloc((strlen(prefix) + strlen(raw_path) + 2));
223 224 225 226
	if (!*new_path)
		return -ENOMEM;

	for (;;) {
227
		sprintf(*new_path, "%s/%s", prefix, raw_path);
228 229 230 231

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

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

236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256
		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;
		}
	}
}

257 258 259
#define LINEBUF_SIZE 256
#define NR_ADDITIONAL_LINES 2

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

	if (fgets(buf, LINEBUF_SIZE, fp) == NULL)
		goto error;
	if (!skip) {
		if (show_num)
269
			fprintf(stdout, "%7d  %s", l, buf);
270 271 272 273 274 275 276 277 278 279 280 281 282 283 284
		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);
		}
	}
285 286

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

	return -1;
294 295 296 297 298 299
}

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

	/* Search a line range */
309 310 311 312
	ret = init_vmlinux();
	if (ret < 0)
		return ret;

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

319 320
	ret = find_line_range(fd, lr);
	close(fd);
321 322 323 324 325 326 327
	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;
	}
328

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

338 339 340 341 342 343 344 345 346
	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");
347 348 349 350 351
	if (fp == NULL) {
		pr_warning("Failed to open %s: %s\n", lr->path,
			   strerror(errno));
		return -errno;
	}
352
	/* Skip to starting line number */
353 354 355 356
	while (l < lr->start && ret >= 0)
		ret = show_one_line(fp, l++, true, false);
	if (ret < 0)
		goto end;
357 358

	list_for_each_entry(ln, &lr->line_list, list) {
359 360 361 362 363 364 365 366
		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;
367 368 369 370
	}

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

#else	/* !DWARF_SUPPORT */

380
static int convert_to_perf_probe_point(struct kprobe_trace_point *tp,
381 382
					struct perf_probe_point *pp)
{
383 384 385
	pp->function = strdup(tp->symbol);
	if (pp->function == NULL)
		return -ENOMEM;
386 387
	pp->offset = tp->offset;
	pp->retprobe = tp->retprobe;
388 389

	return 0;
390 391 392
}

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

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

409 410
#endif

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

	if (tmp == NULL)
		return -ENOMEM;
454 455 456 457 458

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

	return 0;
461 462
}

463 464 465 466 467 468 469 470 471 472 473 474
/* 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;
}

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

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

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

515 516 517 518
	tmp = strdup(arg);
	if (tmp == NULL)
		return -ENOMEM;

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

	do {
671 672 673
		*fieldp = zalloc(sizeof(struct perf_probe_arg_field));
		if (*fieldp == NULL)
			return -ENOMEM;
674 675 676
		if (*tmp == '[') {	/* Array */
			str = tmp;
			(*fieldp)->index = strtol(str + 1, &tmp, 0);
677
			(*fieldp)->ref = true;
678 679 680 681 682 683 684 685 686 687 688 689 690 691 692 693 694 695 696 697 698
			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, "-.[");
699
		}
700
		if (tmp) {
701 702 703
			(*fieldp)->name = strndup(str, tmp - str);
			if ((*fieldp)->name == NULL)
				return -ENOMEM;
704 705
			if (*str != '[')
				goodname = (*fieldp)->name;
706 707 708 709
			pr_debug("%s(%d), ", (*fieldp)->name, (*fieldp)->ref);
			fieldp = &(*fieldp)->next;
		}
	} while (tmp);
710 711 712
	(*fieldp)->name = strdup(str);
	if ((*fieldp)->name == NULL)
		return -ENOMEM;
713 714
	if (*str != '[')
		goodname = (*fieldp)->name;
715
	pr_debug("%s(%d)\n", (*fieldp)->name, (*fieldp)->ref);
716

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

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

732
	argv = argv_split(cmd, &argc);
733 734 735 736 737 738 739 740 741
	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;
	}
742
	/* Parse probe point */
743 744 745
	ret = parse_perf_probe_point(argv[0], pev);
	if (ret < 0)
		goto out;
746

747
	/* Copy arguments and ensure return probe has no C argument */
748
	pev->nargs = argc - 1;
749 750 751 752 753
	pev->args = zalloc(sizeof(struct perf_probe_arg) * pev->nargs);
	if (pev->args == NULL) {
		ret = -ENOMEM;
		goto out;
	}
754 755 756 757
	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) {
758
			semantic_error("You can't specify local variable for"
759 760 761
				       " kretprobe.\n");
			ret = -EINVAL;
		}
762
	}
763
out:
764
	argv_free(argv);
765 766

	return ret;
767 768
}

769 770 771 772 773 774 775 776 777
/* 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++)
778
		if (is_c_varname(pev->args[i].var))
779 780 781 782 783
			return true;

	return false;
}

784
/* Parse kprobe_events event into struct probe_point */
785
int parse_kprobe_trace_command(const char *cmd, struct kprobe_trace_event *tev)
786
{
787
	struct kprobe_trace_point *tp = &tev->point;
788 789 790 791 792
	char pr;
	char *p;
	int ret, i, argc;
	char **argv;

793 794
	pr_debug("Parsing kprobe_events: %s\n", cmd);
	argv = argv_split(cmd, &argc);
795 796 797 798 799 800 801 802 803
	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;
	}
804 805

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

816
	tp->retprobe = (pr == 'r');
817 818

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

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

850 851 852 853 854 855 856
/* 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;

857 858 859 860
	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);
861 862 863 864 865 866
	if (ret <= 0)
		goto error;
	tmp += ret;
	len -= ret;

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

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

887 888
	return tmp - buf;
error:
889 890 891
	pr_debug("Failed to synthesize perf probe argument: %s",
		 strerror(-ret));
	return ret;
892 893
}

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

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

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

	return buf;
938
error:
939 940
	pr_debug("Failed to synthesize perf probe point: %s",
		 strerror(-ret));
941 942
	if (buf)
		free(buf);
943
	return NULL;
944 945
}

946 947
#if 0
char *synthesize_perf_probe_command(struct perf_probe_event *pev)
948 949 950 951
{
	char *buf;
	int i, len, ret;

952 953 954
	buf = synthesize_perf_probe_point(&pev->point);
	if (!buf)
		return NULL;
955

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

967 968 969 970 971 972 973 974 975 976 977 978 979 980 981 982 983 984 985 986 987 988 989 990 991
	return buf;
}
#endif

static int __synthesize_kprobe_trace_arg_ref(struct kprobe_trace_arg_ref *ref,
					     char **buf, size_t *buflen,
					     int depth)
{
	int ret;
	if (ref->next) {
		depth = __synthesize_kprobe_trace_arg_ref(ref->next, buf,
							 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;
992 993 994

}

995 996
static int synthesize_kprobe_trace_arg(struct kprobe_trace_arg *arg,
				       char *buf, size_t buflen)
997
{
998
	struct kprobe_trace_arg_ref *ref = arg->ref;
999 1000 1001 1002 1003 1004 1005 1006 1007 1008 1009 1010 1011
	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;

1012 1013 1014 1015
	/* Special case: @XXX */
	if (arg->value[0] == '@' && arg->ref)
			ref = ref->next;

1016
	/* Dereferencing arguments */
1017 1018
	if (ref) {
		depth = __synthesize_kprobe_trace_arg_ref(ref, &buf,
1019 1020 1021 1022 1023 1024
							  &buflen, 1);
		if (depth < 0)
			return depth;
	}

	/* Print argument value */
1025 1026 1027 1028 1029
	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);
1030 1031 1032 1033 1034 1035 1036 1037 1038 1039 1040 1041 1042
	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;
	}
1043 1044 1045 1046 1047 1048 1049
	/* Print argument type */
	if (arg->type) {
		ret = e_snprintf(buf, buflen, ":%s", arg->type);
		if (ret <= 0)
			return ret;
		buf += ret;
	}
1050 1051 1052 1053 1054 1055 1056

	return buf - tmp;
}

char *synthesize_kprobe_trace_command(struct kprobe_trace_event *tev)
{
	struct kprobe_trace_point *tp = &tev->point;
1057 1058 1059
	char *buf;
	int i, len, ret;

1060 1061 1062 1063
	buf = zalloc(MAX_CMDLEN);
	if (buf == NULL)
		return NULL;

1064 1065 1066 1067 1068
	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)
1069 1070
		goto error;

1071 1072 1073
	for (i = 0; i < tev->nargs; i++) {
		ret = synthesize_kprobe_trace_arg(&tev->args[i], buf + len,
						  MAX_CMDLEN - len);
1074
		if (ret <= 0)
1075 1076 1077 1078
			goto error;
		len += ret;
	}

1079
	return buf;
1080
error:
1081 1082 1083
	free(buf);
	return NULL;
}
1084

1085 1086
int convert_to_perf_probe_event(struct kprobe_trace_event *tev,
				struct perf_probe_event *pev)
1087
{
1088
	char buf[64] = "";
1089
	int i, ret;
1090

1091
	/* Convert event/group name */
1092 1093 1094 1095
	pev->event = strdup(tev->event);
	pev->group = strdup(tev->group);
	if (pev->event == NULL || pev->group == NULL)
		return -ENOMEM;
1096

1097
	/* Convert trace_point to probe_point */
1098 1099 1100
	ret = convert_to_perf_probe_point(&tev->point, &pev->point);
	if (ret < 0)
		return ret;
1101

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

	if (ret < 0)
		clear_perf_probe_event(pev);

	return ret;
1123 1124 1125 1126 1127
}

void clear_perf_probe_event(struct perf_probe_event *pev)
{
	struct perf_probe_point *pp = &pev->point;
1128
	struct perf_probe_arg_field *field, *next;
1129 1130 1131 1132 1133 1134 1135 1136 1137 1138 1139 1140
	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);
1141
	for (i = 0; i < pev->nargs; i++) {
1142 1143
		if (pev->args[i].name)
			free(pev->args[i].name);
1144 1145
		if (pev->args[i].var)
			free(pev->args[i].var);
1146 1147
		if (pev->args[i].type)
			free(pev->args[i].type);
1148 1149 1150 1151 1152 1153 1154 1155 1156
		field = pev->args[i].field;
		while (field) {
			next = field->next;
			if (field->name)
				free(field->name);
			free(field);
			field = next;
		}
	}
1157 1158 1159 1160 1161 1162 1163 1164 1165 1166 1167 1168 1169 1170 1171 1172 1173 1174 1175 1176 1177
	if (pev->args)
		free(pev->args);
	memset(pev, 0, sizeof(*pev));
}

void clear_kprobe_trace_event(struct kprobe_trace_event *tev)
{
	struct kprobe_trace_arg_ref *ref, *next;
	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);
1178 1179
		if (tev->args[i].type)
			free(tev->args[i].type);
1180 1181 1182 1183 1184 1185 1186 1187 1188 1189
		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));
1190 1191
}

1192
static int open_kprobe_events(bool readwrite)
1193 1194
{
	char buf[PATH_MAX];
1195
	const char *__debugfs;
1196 1197
	int ret;

1198 1199 1200 1201 1202 1203 1204
	__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);
1205
	if (ret >= 0) {
1206
		pr_debug("Opening %s write=%d\n", buf, readwrite);
1207 1208 1209 1210 1211
		if (readwrite && !probe_event_dry_run)
			ret = open(buf, O_RDWR, O_APPEND);
		else
			ret = open(buf, O_RDONLY, 0);
	}
1212

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

/* Get raw string list of current kprobe_events */
1225
static struct strlist *get_kprobe_trace_command_rawlist(int fd)
1226 1227 1228 1229 1230 1231 1232 1233 1234 1235 1236 1237 1238 1239 1240 1241 1242 1243 1244
{
	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);
1245 1246 1247 1248 1249
		if (ret < 0) {
			pr_debug("strlist__add failed: %s\n", strerror(-ret));
			strlist__delete(sl);
			return NULL;
		}
1250 1251 1252 1253 1254 1255
	}
	fclose(fp);

	return sl;
}

1256
/* Show an event */
1257
static int show_perf_probe_event(struct perf_probe_event *pev)
1258
{
1259
	int i, ret;
1260
	char buf[128];
1261
	char *place;
1262

1263 1264
	/* Synthesize only event probe point */
	place = synthesize_perf_probe_point(&pev->point);
1265 1266
	if (!place)
		return -EINVAL;
1267 1268

	ret = e_snprintf(buf, 128, "%s:%s", pev->group, pev->event);
1269
	if (ret < 0)
1270 1271
		return ret;

1272
	printf("  %-20s (on %s", buf, place);
1273

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

1289
/* List up current perf-probe events */
1290
int show_perf_probe_events(void)
1291
{
1292
	int fd, ret;
1293 1294
	struct kprobe_trace_event tev;
	struct perf_probe_event pev;
1295 1296 1297
	struct strlist *rawlist;
	struct str_node *ent;

1298
	setup_pager();
1299 1300 1301
	ret = init_vmlinux();
	if (ret < 0)
		return ret;
1302 1303 1304

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

1306
	fd = open_kprobe_events(false);
1307 1308 1309
	if (fd < 0)
		return fd;

1310
	rawlist = get_kprobe_trace_command_rawlist(fd);
1311
	close(fd);
1312 1313
	if (!rawlist)
		return -ENOENT;
1314

1315
	strlist__for_each(ent, rawlist) {
1316 1317 1318 1319 1320 1321
		ret = parse_kprobe_trace_command(ent->s, &tev);
		if (ret >= 0) {
			ret = convert_to_perf_probe_event(&tev, &pev);
			if (ret >= 0)
				ret = show_perf_probe_event(&pev);
		}
1322 1323
		clear_perf_probe_event(&pev);
		clear_kprobe_trace_event(&tev);
1324 1325
		if (ret < 0)
			break;
1326 1327
	}
	strlist__delete(rawlist);
1328 1329

	return ret;
1330 1331
}

1332
/* Get current perf-probe event names */
1333
static struct strlist *get_kprobe_trace_event_names(int fd, bool include_group)
1334
{
1335
	char buf[128];
1336 1337
	struct strlist *sl, *rawlist;
	struct str_node *ent;
1338
	struct kprobe_trace_event tev;
1339
	int ret = 0;
1340

1341
	memset(&tev, 0, sizeof(tev));
1342

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

1362 1363 1364 1365
	if (ret < 0) {
		strlist__delete(sl);
		return NULL;
	}
1366 1367 1368
	return sl;
}

1369
static int write_kprobe_trace_event(int fd, struct kprobe_trace_event *tev)
1370
{
1371
	int ret = 0;
1372
	char *buf = synthesize_kprobe_trace_command(tev);
1373

1374 1375 1376 1377 1378
	if (!buf) {
		pr_debug("Failed to synthesize kprobe trace event.\n");
		return -EINVAL;
	}

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

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

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

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

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

	return ret;
1426 1427
}

1428 1429 1430
static int __add_kprobe_trace_events(struct perf_probe_event *pev,
				     struct kprobe_trace_event *tevs,
				     int ntevs, bool allow_suffix)
1431
{
1432
	int i, fd, ret;
1433
	struct kprobe_trace_event *tev = NULL;
1434 1435
	char buf[64];
	const char *event, *group;
1436
	struct strlist *namelist;
1437

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

1448
	ret = 0;
1449
	printf("Add new event%s\n", (ntevs > 1) ? "s:" : ":");
1450
	for (i = 0; i < ntevs; i++) {
1451 1452 1453 1454 1455 1456 1457 1458 1459 1460 1461 1462 1463 1464
		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 */
1465 1466 1467 1468
		ret = get_new_event_name(buf, 64, event,
					 namelist, allow_suffix);
		if (ret < 0)
			break;
1469 1470
		event = buf;

1471 1472 1473 1474 1475 1476
		tev->event = strdup(event);
		tev->group = strdup(group);
		if (tev->event == NULL || tev->group == NULL) {
			ret = -ENOMEM;
			break;
		}
1477 1478 1479
		ret = write_kprobe_trace_event(fd, tev);
		if (ret < 0)
			break;
1480 1481 1482 1483 1484 1485 1486 1487 1488 1489 1490 1491 1492 1493 1494 1495 1496 1497 1498 1499
		/* 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;
1500
	}
1501 1502 1503 1504 1505 1506 1507

	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);
	}
1508

1509
	strlist__delete(namelist);
1510
	close(fd);
1511
	return ret;
1512
}
1513

1514
static int convert_to_kprobe_trace_events(struct perf_probe_event *pev,
1515 1516
					  struct kprobe_trace_event **tevs,
					  int max_tevs)
1517 1518
{
	struct symbol *sym;
1519
	int ret = 0, i;
1520 1521
	struct kprobe_trace_event *tev;

1522
	/* Convert perf_probe_event with debuginfo */
1523
	ret = try_to_find_kprobe_trace_events(pev, tevs, max_tevs);
1524 1525
	if (ret != 0)
		return ret;
1526

1527
	/* Allocate trace event buffer */
1528 1529 1530
	tev = *tevs = zalloc(sizeof(struct kprobe_trace_event));
	if (tev == NULL)
		return -ENOMEM;
1531 1532

	/* Copy parameters */
1533 1534 1535 1536 1537
	tev->point.symbol = strdup(pev->point.function);
	if (tev->point.symbol == NULL) {
		ret = -ENOMEM;
		goto error;
	}
1538 1539 1540
	tev->point.offset = pev->point.offset;
	tev->nargs = pev->nargs;
	if (tev->nargs) {
1541 1542 1543
		tev->args = zalloc(sizeof(struct kprobe_trace_arg)
				   * tev->nargs);
		if (tev->args == NULL) {
1544 1545
			ret = -ENOMEM;
			goto error;
1546
		}
1547
		for (i = 0; i < tev->nargs; i++) {
1548 1549 1550 1551 1552 1553 1554 1555 1556 1557 1558 1559 1560 1561 1562 1563 1564 1565 1566
			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;
				}
			}
1567
		}
1568 1569 1570
	}

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

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

struct __event_package {
	struct perf_probe_event		*pev;
	struct kprobe_trace_event	*tevs;
	int				ntevs;
};

1594
int add_perf_probe_events(struct perf_probe_event *pevs, int npevs,
1595
			  bool force_add, int max_tevs)
1596
{
1597
	int i, j, ret;
1598 1599
	struct __event_package *pkgs;

1600 1601 1602
	pkgs = zalloc(sizeof(struct __event_package) * npevs);
	if (pkgs == NULL)
		return -ENOMEM;
1603 1604

	/* Init vmlinux path */
1605 1606 1607
	ret = init_vmlinux();
	if (ret < 0)
		return ret;
1608 1609 1610 1611 1612

	/* Loop 1: convert all events */
	for (i = 0; i < npevs; i++) {
		pkgs[i].pev = &pevs[i];
		/* Convert with or without debuginfo */
1613
		ret  = convert_to_kprobe_trace_events(pkgs[i].pev,
1614
						      &pkgs[i].tevs, max_tevs);
1615 1616 1617
		if (ret < 0)
			goto end;
		pkgs[i].ntevs = ret;
1618 1619
	}

1620
	/* Loop 2: add all events */
1621 1622 1623 1624 1625
	for (i = 0; i < npevs && ret >= 0; i++)
		ret = __add_kprobe_trace_events(pkgs[i].pev, pkgs[i].tevs,
						pkgs[i].ntevs, force_add);
end:
	/* Loop 3: cleanup trace events  */
1626
	for (i = 0; i < npevs; i++)
1627 1628 1629 1630
		for (j = 0; j < pkgs[i].ntevs; j++)
			clear_kprobe_trace_event(&pkgs[i].tevs[j]);

	return ret;
1631 1632
}

1633
static int __del_trace_kprobe_event(int fd, struct str_node *ent)
1634 1635 1636
{
	char *p;
	char buf[128];
1637
	int ret;
1638 1639

	/* Convert from perf-probe event to trace-kprobe event */
1640 1641 1642 1643
	ret = e_snprintf(buf, 128, "-:%s", ent->s);
	if (ret < 0)
		goto error;

1644
	p = strchr(buf + 2, ':');
1645 1646 1647 1648 1649 1650
	if (!p) {
		pr_debug("Internal error: %s should have ':' but not.\n",
			 ent->s);
		ret = -ENOTSUP;
		goto error;
	}
1651 1652
	*p = '/';

1653 1654
	pr_debug("Writing event: %s\n", buf);
	ret = write(fd, buf, strlen(buf));
1655 1656 1657
	if (ret < 0)
		goto error;

1658
	printf("Remove event: %s\n", ent->s);
1659 1660 1661 1662
	return 0;
error:
	pr_warning("Failed to delete event: %s\n", strerror(-ret));
	return ret;
1663 1664
}

1665 1666
static int del_trace_kprobe_event(int fd, const char *group,
				  const char *event, struct strlist *namelist)
1667 1668
{
	char buf[128];
1669
	struct str_node *ent, *n;
1670
	int found = 0, ret = 0;
1671

1672 1673 1674 1675 1676
	ret = e_snprintf(buf, 128, "%s:%s", group, event);
	if (ret < 0) {
		pr_err("Failed to copy event.");
		return ret;
	}
1677

1678 1679 1680 1681
	if (strpbrk(buf, "*?")) { /* Glob-exp */
		strlist__for_each_safe(ent, n, namelist)
			if (strglobmatch(ent->s, buf)) {
				found++;
1682 1683 1684
				ret = __del_trace_kprobe_event(fd, ent);
				if (ret < 0)
					break;
1685 1686 1687 1688 1689 1690
				strlist__remove(namelist, ent);
			}
	} else {
		ent = strlist__find(namelist, buf);
		if (ent) {
			found++;
1691 1692 1693
			ret = __del_trace_kprobe_event(fd, ent);
			if (ret >= 0)
				strlist__remove(namelist, ent);
1694 1695
		}
	}
1696 1697 1698 1699
	if (found == 0 && ret >= 0)
		pr_info("Info: Event \"%s\" does not exist.\n", buf);

	return ret;
1700 1701
}

1702
int del_perf_probe_events(struct strlist *dellist)
1703
{
1704
	int fd, ret = 0;
1705 1706 1707 1708 1709
	const char *group, *event;
	char *p, *str;
	struct str_node *ent;
	struct strlist *namelist;

1710
	fd = open_kprobe_events(true);
1711 1712 1713
	if (fd < 0)
		return fd;

1714
	/* Get current event names */
1715
	namelist = get_kprobe_trace_event_names(fd, true);
1716 1717
	if (namelist == NULL)
		return -EINVAL;
1718

1719
	strlist__for_each(ent, dellist) {
1720 1721 1722 1723 1724
		str = strdup(ent->s);
		if (str == NULL) {
			ret = -ENOMEM;
			break;
		}
1725
		pr_debug("Parsing: %s\n", str);
1726 1727 1728 1729 1730 1731
		p = strchr(str, ':');
		if (p) {
			group = str;
			*p = '\0';
			event = p + 1;
		} else {
1732
			group = "*";
1733 1734
			event = str;
		}
1735
		pr_debug("Group: %s, Event: %s\n", group, event);
1736
		ret = del_trace_kprobe_event(fd, group, event, namelist);
1737
		free(str);
1738 1739
		if (ret < 0)
			break;
1740 1741 1742
	}
	strlist__delete(namelist);
	close(fd);
1743 1744

	return ret;
1745 1746
}