xref: /openbmc/linux/tools/perf/builtin-script.c (revision 565d76cb)
1 #include "builtin.h"
2 
3 #include "perf.h"
4 #include "util/cache.h"
5 #include "util/debug.h"
6 #include "util/exec_cmd.h"
7 #include "util/header.h"
8 #include "util/parse-options.h"
9 #include "util/session.h"
10 #include "util/symbol.h"
11 #include "util/thread.h"
12 #include "util/trace-event.h"
13 #include "util/parse-options.h"
14 #include "util/util.h"
15 #include "util/evlist.h"
16 #include "util/evsel.h"
17 
18 static char const		*script_name;
19 static char const		*generate_script_lang;
20 static bool			debug_mode;
21 static u64			last_timestamp;
22 static u64			nr_unordered;
23 extern const struct option	record_options[];
24 static bool			no_callchain;
25 
26 enum perf_output_field {
27 	PERF_OUTPUT_COMM            = 1U << 0,
28 	PERF_OUTPUT_TID             = 1U << 1,
29 	PERF_OUTPUT_PID             = 1U << 2,
30 	PERF_OUTPUT_TIME            = 1U << 3,
31 	PERF_OUTPUT_CPU             = 1U << 4,
32 	PERF_OUTPUT_EVNAME          = 1U << 5,
33 	PERF_OUTPUT_TRACE           = 1U << 6,
34 	PERF_OUTPUT_SYM             = 1U << 7,
35 };
36 
37 struct output_option {
38 	const char *str;
39 	enum perf_output_field field;
40 } all_output_options[] = {
41 	{.str = "comm",  .field = PERF_OUTPUT_COMM},
42 	{.str = "tid",   .field = PERF_OUTPUT_TID},
43 	{.str = "pid",   .field = PERF_OUTPUT_PID},
44 	{.str = "time",  .field = PERF_OUTPUT_TIME},
45 	{.str = "cpu",   .field = PERF_OUTPUT_CPU},
46 	{.str = "event", .field = PERF_OUTPUT_EVNAME},
47 	{.str = "trace", .field = PERF_OUTPUT_TRACE},
48 	{.str = "sym",   .field = PERF_OUTPUT_SYM},
49 };
50 
51 /* default set to maintain compatibility with current format */
52 static u64 output_fields[PERF_TYPE_MAX] = {
53 	[PERF_TYPE_HARDWARE] = PERF_OUTPUT_COMM | PERF_OUTPUT_TID | \
54 			       PERF_OUTPUT_CPU | PERF_OUTPUT_TIME | \
55 			       PERF_OUTPUT_EVNAME | PERF_OUTPUT_SYM,
56 
57 	[PERF_TYPE_SOFTWARE] = PERF_OUTPUT_COMM | PERF_OUTPUT_TID | \
58 			       PERF_OUTPUT_CPU | PERF_OUTPUT_TIME | \
59 			       PERF_OUTPUT_EVNAME | PERF_OUTPUT_SYM,
60 
61 	[PERF_TYPE_TRACEPOINT] = PERF_OUTPUT_COMM | PERF_OUTPUT_TID | \
62 				 PERF_OUTPUT_CPU | PERF_OUTPUT_TIME | \
63 				 PERF_OUTPUT_EVNAME | PERF_OUTPUT_TRACE,
64 };
65 
66 static bool output_set_by_user;
67 
68 #define PRINT_FIELD(x)  (output_fields[attr->type] & PERF_OUTPUT_##x)
69 
70 static int perf_session__check_attr(struct perf_session *session,
71 				    struct perf_event_attr *attr)
72 {
73 	if (PRINT_FIELD(TRACE) &&
74 		!perf_session__has_traces(session, "record -R"))
75 		return -EINVAL;
76 
77 	if (PRINT_FIELD(SYM)) {
78 		if (!(session->sample_type & PERF_SAMPLE_IP)) {
79 			pr_err("Samples do not contain IP data.\n");
80 			return -EINVAL;
81 		}
82 		if (!no_callchain &&
83 		    !(session->sample_type & PERF_SAMPLE_CALLCHAIN))
84 			symbol_conf.use_callchain = false;
85 	}
86 
87 	if ((PRINT_FIELD(PID) || PRINT_FIELD(TID)) &&
88 		!(session->sample_type & PERF_SAMPLE_TID)) {
89 		pr_err("Samples do not contain TID/PID data.\n");
90 		return -EINVAL;
91 	}
92 
93 	if (PRINT_FIELD(TIME) &&
94 		!(session->sample_type & PERF_SAMPLE_TIME)) {
95 		pr_err("Samples do not contain timestamps.\n");
96 		return -EINVAL;
97 	}
98 
99 	if (PRINT_FIELD(CPU) &&
100 		!(session->sample_type & PERF_SAMPLE_CPU)) {
101 		pr_err("Samples do not contain cpu.\n");
102 		return -EINVAL;
103 	}
104 
105 	return 0;
106 }
107 
108 static void print_sample_start(struct perf_sample *sample,
109 			       struct thread *thread,
110 			       struct perf_event_attr *attr)
111 {
112 	int type;
113 	struct event *event;
114 	const char *evname = NULL;
115 	unsigned long secs;
116 	unsigned long usecs;
117 	unsigned long long nsecs;
118 
119 	if (PRINT_FIELD(COMM)) {
120 		if (latency_format)
121 			printf("%8.8s ", thread->comm);
122 		else if (PRINT_FIELD(SYM) && symbol_conf.use_callchain)
123 			printf("%s ", thread->comm);
124 		else
125 			printf("%16s ", thread->comm);
126 	}
127 
128 	if (PRINT_FIELD(PID) && PRINT_FIELD(TID))
129 		printf("%5d/%-5d ", sample->pid, sample->tid);
130 	else if (PRINT_FIELD(PID))
131 		printf("%5d ", sample->pid);
132 	else if (PRINT_FIELD(TID))
133 		printf("%5d ", sample->tid);
134 
135 	if (PRINT_FIELD(CPU)) {
136 		if (latency_format)
137 			printf("%3d ", sample->cpu);
138 		else
139 			printf("[%03d] ", sample->cpu);
140 	}
141 
142 	if (PRINT_FIELD(TIME)) {
143 		nsecs = sample->time;
144 		secs = nsecs / NSECS_PER_SEC;
145 		nsecs -= secs * NSECS_PER_SEC;
146 		usecs = nsecs / NSECS_PER_USEC;
147 		printf("%5lu.%06lu: ", secs, usecs);
148 	}
149 
150 	if (PRINT_FIELD(EVNAME)) {
151 		if (attr->type == PERF_TYPE_TRACEPOINT) {
152 			type = trace_parse_common_type(sample->raw_data);
153 			event = trace_find_event(type);
154 			if (event)
155 				evname = event->name;
156 		} else
157 			evname = __event_name(attr->type, attr->config);
158 
159 		printf("%s: ", evname ? evname : "(unknown)");
160 	}
161 }
162 
163 static void process_event(union perf_event *event __unused,
164 			  struct perf_sample *sample,
165 			  struct perf_session *session,
166 			  struct thread *thread)
167 {
168 	struct perf_event_attr *attr;
169 	struct perf_evsel *evsel;
170 
171 	evsel = perf_evlist__id2evsel(session->evlist, sample->id);
172 	if (evsel == NULL) {
173 		pr_err("Invalid data. Contains samples with id not in "
174 		       "its header!\n");
175 		return;
176 	}
177 	attr = &evsel->attr;
178 
179 	if (output_fields[attr->type] == 0)
180 		return;
181 
182 	if (perf_session__check_attr(session, attr) < 0)
183 		return;
184 
185 	print_sample_start(sample, thread, attr);
186 
187 	if (PRINT_FIELD(TRACE))
188 		print_trace_event(sample->cpu, sample->raw_data,
189 				  sample->raw_size);
190 
191 	if (PRINT_FIELD(SYM)) {
192 		if (!symbol_conf.use_callchain)
193 			printf(" ");
194 		else
195 			printf("\n");
196 		perf_session__print_symbols(event, sample, session);
197 	}
198 
199 	printf("\n");
200 }
201 
202 static int default_start_script(const char *script __unused,
203 				int argc __unused,
204 				const char **argv __unused)
205 {
206 	return 0;
207 }
208 
209 static int default_stop_script(void)
210 {
211 	return 0;
212 }
213 
214 static int default_generate_script(const char *outfile __unused)
215 {
216 	return 0;
217 }
218 
219 static struct scripting_ops default_scripting_ops = {
220 	.start_script		= default_start_script,
221 	.stop_script		= default_stop_script,
222 	.process_event		= process_event,
223 	.generate_script	= default_generate_script,
224 };
225 
226 static struct scripting_ops	*scripting_ops;
227 
228 static void setup_scripting(void)
229 {
230 	setup_perl_scripting();
231 	setup_python_scripting();
232 
233 	scripting_ops = &default_scripting_ops;
234 }
235 
236 static int cleanup_scripting(void)
237 {
238 	pr_debug("\nperf script stopped\n");
239 
240 	return scripting_ops->stop_script();
241 }
242 
243 static char const		*input_name = "perf.data";
244 
245 static int process_sample_event(union perf_event *event,
246 				struct perf_sample *sample,
247 				struct perf_session *session)
248 {
249 	struct thread *thread = perf_session__findnew(session, event->ip.pid);
250 
251 	if (thread == NULL) {
252 		pr_debug("problem processing %d event, skipping it.\n",
253 			 event->header.type);
254 		return -1;
255 	}
256 
257 	if (debug_mode) {
258 		if (sample->time < last_timestamp) {
259 			pr_err("Samples misordered, previous: %" PRIu64
260 				" this: %" PRIu64 "\n", last_timestamp,
261 				sample->time);
262 			nr_unordered++;
263 		}
264 		last_timestamp = sample->time;
265 		return 0;
266 	}
267 	scripting_ops->process_event(event, sample, session, thread);
268 
269 	session->hists.stats.total_period += sample->period;
270 	return 0;
271 }
272 
273 static struct perf_event_ops event_ops = {
274 	.sample		 = process_sample_event,
275 	.mmap		 = perf_event__process_mmap,
276 	.comm		 = perf_event__process_comm,
277 	.exit		 = perf_event__process_task,
278 	.fork		 = perf_event__process_task,
279 	.attr		 = perf_event__process_attr,
280 	.event_type	 = perf_event__process_event_type,
281 	.tracing_data	 = perf_event__process_tracing_data,
282 	.build_id	 = perf_event__process_build_id,
283 	.ordered_samples = true,
284 	.ordering_requires_timestamps = true,
285 };
286 
287 extern volatile int session_done;
288 
289 static void sig_handler(int sig __unused)
290 {
291 	session_done = 1;
292 }
293 
294 static int __cmd_script(struct perf_session *session)
295 {
296 	int ret;
297 
298 	signal(SIGINT, sig_handler);
299 
300 	ret = perf_session__process_events(session, &event_ops);
301 
302 	if (debug_mode)
303 		pr_err("Misordered timestamps: %" PRIu64 "\n", nr_unordered);
304 
305 	return ret;
306 }
307 
308 struct script_spec {
309 	struct list_head	node;
310 	struct scripting_ops	*ops;
311 	char			spec[0];
312 };
313 
314 static LIST_HEAD(script_specs);
315 
316 static struct script_spec *script_spec__new(const char *spec,
317 					    struct scripting_ops *ops)
318 {
319 	struct script_spec *s = malloc(sizeof(*s) + strlen(spec) + 1);
320 
321 	if (s != NULL) {
322 		strcpy(s->spec, spec);
323 		s->ops = ops;
324 	}
325 
326 	return s;
327 }
328 
329 static void script_spec__delete(struct script_spec *s)
330 {
331 	free(s->spec);
332 	free(s);
333 }
334 
335 static void script_spec__add(struct script_spec *s)
336 {
337 	list_add_tail(&s->node, &script_specs);
338 }
339 
340 static struct script_spec *script_spec__find(const char *spec)
341 {
342 	struct script_spec *s;
343 
344 	list_for_each_entry(s, &script_specs, node)
345 		if (strcasecmp(s->spec, spec) == 0)
346 			return s;
347 	return NULL;
348 }
349 
350 static struct script_spec *script_spec__findnew(const char *spec,
351 						struct scripting_ops *ops)
352 {
353 	struct script_spec *s = script_spec__find(spec);
354 
355 	if (s)
356 		return s;
357 
358 	s = script_spec__new(spec, ops);
359 	if (!s)
360 		goto out_delete_spec;
361 
362 	script_spec__add(s);
363 
364 	return s;
365 
366 out_delete_spec:
367 	script_spec__delete(s);
368 
369 	return NULL;
370 }
371 
372 int script_spec_register(const char *spec, struct scripting_ops *ops)
373 {
374 	struct script_spec *s;
375 
376 	s = script_spec__find(spec);
377 	if (s)
378 		return -1;
379 
380 	s = script_spec__findnew(spec, ops);
381 	if (!s)
382 		return -1;
383 
384 	return 0;
385 }
386 
387 static struct scripting_ops *script_spec__lookup(const char *spec)
388 {
389 	struct script_spec *s = script_spec__find(spec);
390 	if (!s)
391 		return NULL;
392 
393 	return s->ops;
394 }
395 
396 static void list_available_languages(void)
397 {
398 	struct script_spec *s;
399 
400 	fprintf(stderr, "\n");
401 	fprintf(stderr, "Scripting language extensions (used in "
402 		"perf script -s [spec:]script.[spec]):\n\n");
403 
404 	list_for_each_entry(s, &script_specs, node)
405 		fprintf(stderr, "  %-42s [%s]\n", s->spec, s->ops->name);
406 
407 	fprintf(stderr, "\n");
408 }
409 
410 static int parse_scriptname(const struct option *opt __used,
411 			    const char *str, int unset __used)
412 {
413 	char spec[PATH_MAX];
414 	const char *script, *ext;
415 	int len;
416 
417 	if (strcmp(str, "lang") == 0) {
418 		list_available_languages();
419 		exit(0);
420 	}
421 
422 	script = strchr(str, ':');
423 	if (script) {
424 		len = script - str;
425 		if (len >= PATH_MAX) {
426 			fprintf(stderr, "invalid language specifier");
427 			return -1;
428 		}
429 		strncpy(spec, str, len);
430 		spec[len] = '\0';
431 		scripting_ops = script_spec__lookup(spec);
432 		if (!scripting_ops) {
433 			fprintf(stderr, "invalid language specifier");
434 			return -1;
435 		}
436 		script++;
437 	} else {
438 		script = str;
439 		ext = strrchr(script, '.');
440 		if (!ext) {
441 			fprintf(stderr, "invalid script extension");
442 			return -1;
443 		}
444 		scripting_ops = script_spec__lookup(++ext);
445 		if (!scripting_ops) {
446 			fprintf(stderr, "invalid script extension");
447 			return -1;
448 		}
449 	}
450 
451 	script_name = strdup(script);
452 
453 	return 0;
454 }
455 
456 static int parse_output_fields(const struct option *opt __used,
457 			    const char *arg, int unset __used)
458 {
459 	char *tok;
460 	int i, imax = sizeof(all_output_options) / sizeof(struct output_option);
461 	int rc = 0;
462 	char *str = strdup(arg);
463 	int type = -1;
464 
465 	if (!str)
466 		return -ENOMEM;
467 
468 	tok = strtok(str, ":");
469 	if (!tok) {
470 		fprintf(stderr,
471 			"Invalid field string - not prepended with type.");
472 		return -EINVAL;
473 	}
474 
475 	/* first word should state which event type user
476 	 * is specifying the fields
477 	 */
478 	if (!strcmp(tok, "hw"))
479 		type = PERF_TYPE_HARDWARE;
480 	else if (!strcmp(tok, "sw"))
481 		type = PERF_TYPE_SOFTWARE;
482 	else if (!strcmp(tok, "trace"))
483 		type = PERF_TYPE_TRACEPOINT;
484 	else {
485 		fprintf(stderr, "Invalid event type in field string.");
486 		return -EINVAL;
487 	}
488 
489 	output_fields[type] = 0;
490 	while (1) {
491 		tok = strtok(NULL, ",");
492 		if (!tok)
493 			break;
494 		for (i = 0; i < imax; ++i) {
495 			if (strcmp(tok, all_output_options[i].str) == 0) {
496 				output_fields[type] |= all_output_options[i].field;
497 				break;
498 			}
499 		}
500 		if (i == imax) {
501 			fprintf(stderr, "Invalid field requested.");
502 			rc = -EINVAL;
503 			break;
504 		}
505 	}
506 
507 	if (output_fields[type] == 0) {
508 		pr_debug("No fields requested for %s type. "
509 			 "Events will not be displayed\n", event_type(type));
510 	}
511 
512 	output_set_by_user = true;
513 
514 	free(str);
515 	return rc;
516 }
517 
518 /* Helper function for filesystems that return a dent->d_type DT_UNKNOWN */
519 static int is_directory(const char *base_path, const struct dirent *dent)
520 {
521 	char path[PATH_MAX];
522 	struct stat st;
523 
524 	sprintf(path, "%s/%s", base_path, dent->d_name);
525 	if (stat(path, &st))
526 		return 0;
527 
528 	return S_ISDIR(st.st_mode);
529 }
530 
531 #define for_each_lang(scripts_path, scripts_dir, lang_dirent, lang_next)\
532 	while (!readdir_r(scripts_dir, &lang_dirent, &lang_next) &&	\
533 	       lang_next)						\
534 		if ((lang_dirent.d_type == DT_DIR ||			\
535 		     (lang_dirent.d_type == DT_UNKNOWN &&		\
536 		      is_directory(scripts_path, &lang_dirent))) &&	\
537 		    (strcmp(lang_dirent.d_name, ".")) &&		\
538 		    (strcmp(lang_dirent.d_name, "..")))
539 
540 #define for_each_script(lang_path, lang_dir, script_dirent, script_next)\
541 	while (!readdir_r(lang_dir, &script_dirent, &script_next) &&	\
542 	       script_next)						\
543 		if (script_dirent.d_type != DT_DIR &&			\
544 		    (script_dirent.d_type != DT_UNKNOWN ||		\
545 		     !is_directory(lang_path, &script_dirent)))
546 
547 
548 #define RECORD_SUFFIX			"-record"
549 #define REPORT_SUFFIX			"-report"
550 
551 struct script_desc {
552 	struct list_head	node;
553 	char			*name;
554 	char			*half_liner;
555 	char			*args;
556 };
557 
558 static LIST_HEAD(script_descs);
559 
560 static struct script_desc *script_desc__new(const char *name)
561 {
562 	struct script_desc *s = zalloc(sizeof(*s));
563 
564 	if (s != NULL && name)
565 		s->name = strdup(name);
566 
567 	return s;
568 }
569 
570 static void script_desc__delete(struct script_desc *s)
571 {
572 	free(s->name);
573 	free(s->half_liner);
574 	free(s->args);
575 	free(s);
576 }
577 
578 static void script_desc__add(struct script_desc *s)
579 {
580 	list_add_tail(&s->node, &script_descs);
581 }
582 
583 static struct script_desc *script_desc__find(const char *name)
584 {
585 	struct script_desc *s;
586 
587 	list_for_each_entry(s, &script_descs, node)
588 		if (strcasecmp(s->name, name) == 0)
589 			return s;
590 	return NULL;
591 }
592 
593 static struct script_desc *script_desc__findnew(const char *name)
594 {
595 	struct script_desc *s = script_desc__find(name);
596 
597 	if (s)
598 		return s;
599 
600 	s = script_desc__new(name);
601 	if (!s)
602 		goto out_delete_desc;
603 
604 	script_desc__add(s);
605 
606 	return s;
607 
608 out_delete_desc:
609 	script_desc__delete(s);
610 
611 	return NULL;
612 }
613 
614 static const char *ends_with(const char *str, const char *suffix)
615 {
616 	size_t suffix_len = strlen(suffix);
617 	const char *p = str;
618 
619 	if (strlen(str) > suffix_len) {
620 		p = str + strlen(str) - suffix_len;
621 		if (!strncmp(p, suffix, suffix_len))
622 			return p;
623 	}
624 
625 	return NULL;
626 }
627 
628 static char *ltrim(char *str)
629 {
630 	int len = strlen(str);
631 
632 	while (len && isspace(*str)) {
633 		len--;
634 		str++;
635 	}
636 
637 	return str;
638 }
639 
640 static int read_script_info(struct script_desc *desc, const char *filename)
641 {
642 	char line[BUFSIZ], *p;
643 	FILE *fp;
644 
645 	fp = fopen(filename, "r");
646 	if (!fp)
647 		return -1;
648 
649 	while (fgets(line, sizeof(line), fp)) {
650 		p = ltrim(line);
651 		if (strlen(p) == 0)
652 			continue;
653 		if (*p != '#')
654 			continue;
655 		p++;
656 		if (strlen(p) && *p == '!')
657 			continue;
658 
659 		p = ltrim(p);
660 		if (strlen(p) && p[strlen(p) - 1] == '\n')
661 			p[strlen(p) - 1] = '\0';
662 
663 		if (!strncmp(p, "description:", strlen("description:"))) {
664 			p += strlen("description:");
665 			desc->half_liner = strdup(ltrim(p));
666 			continue;
667 		}
668 
669 		if (!strncmp(p, "args:", strlen("args:"))) {
670 			p += strlen("args:");
671 			desc->args = strdup(ltrim(p));
672 			continue;
673 		}
674 	}
675 
676 	fclose(fp);
677 
678 	return 0;
679 }
680 
681 static int list_available_scripts(const struct option *opt __used,
682 				  const char *s __used, int unset __used)
683 {
684 	struct dirent *script_next, *lang_next, script_dirent, lang_dirent;
685 	char scripts_path[MAXPATHLEN];
686 	DIR *scripts_dir, *lang_dir;
687 	char script_path[MAXPATHLEN];
688 	char lang_path[MAXPATHLEN];
689 	struct script_desc *desc;
690 	char first_half[BUFSIZ];
691 	char *script_root;
692 	char *str;
693 
694 	snprintf(scripts_path, MAXPATHLEN, "%s/scripts", perf_exec_path());
695 
696 	scripts_dir = opendir(scripts_path);
697 	if (!scripts_dir)
698 		return -1;
699 
700 	for_each_lang(scripts_path, scripts_dir, lang_dirent, lang_next) {
701 		snprintf(lang_path, MAXPATHLEN, "%s/%s/bin", scripts_path,
702 			 lang_dirent.d_name);
703 		lang_dir = opendir(lang_path);
704 		if (!lang_dir)
705 			continue;
706 
707 		for_each_script(lang_path, lang_dir, script_dirent, script_next) {
708 			script_root = strdup(script_dirent.d_name);
709 			str = (char *)ends_with(script_root, REPORT_SUFFIX);
710 			if (str) {
711 				*str = '\0';
712 				desc = script_desc__findnew(script_root);
713 				snprintf(script_path, MAXPATHLEN, "%s/%s",
714 					 lang_path, script_dirent.d_name);
715 				read_script_info(desc, script_path);
716 			}
717 			free(script_root);
718 		}
719 	}
720 
721 	fprintf(stdout, "List of available trace scripts:\n");
722 	list_for_each_entry(desc, &script_descs, node) {
723 		sprintf(first_half, "%s %s", desc->name,
724 			desc->args ? desc->args : "");
725 		fprintf(stdout, "  %-36s %s\n", first_half,
726 			desc->half_liner ? desc->half_liner : "");
727 	}
728 
729 	exit(0);
730 }
731 
732 static char *get_script_path(const char *script_root, const char *suffix)
733 {
734 	struct dirent *script_next, *lang_next, script_dirent, lang_dirent;
735 	char scripts_path[MAXPATHLEN];
736 	char script_path[MAXPATHLEN];
737 	DIR *scripts_dir, *lang_dir;
738 	char lang_path[MAXPATHLEN];
739 	char *str, *__script_root;
740 	char *path = NULL;
741 
742 	snprintf(scripts_path, MAXPATHLEN, "%s/scripts", perf_exec_path());
743 
744 	scripts_dir = opendir(scripts_path);
745 	if (!scripts_dir)
746 		return NULL;
747 
748 	for_each_lang(scripts_path, scripts_dir, lang_dirent, lang_next) {
749 		snprintf(lang_path, MAXPATHLEN, "%s/%s/bin", scripts_path,
750 			 lang_dirent.d_name);
751 		lang_dir = opendir(lang_path);
752 		if (!lang_dir)
753 			continue;
754 
755 		for_each_script(lang_path, lang_dir, script_dirent, script_next) {
756 			__script_root = strdup(script_dirent.d_name);
757 			str = (char *)ends_with(__script_root, suffix);
758 			if (str) {
759 				*str = '\0';
760 				if (strcmp(__script_root, script_root))
761 					continue;
762 				snprintf(script_path, MAXPATHLEN, "%s/%s",
763 					 lang_path, script_dirent.d_name);
764 				path = strdup(script_path);
765 				free(__script_root);
766 				break;
767 			}
768 			free(__script_root);
769 		}
770 	}
771 
772 	return path;
773 }
774 
775 static bool is_top_script(const char *script_path)
776 {
777 	return ends_with(script_path, "top") == NULL ? false : true;
778 }
779 
780 static int has_required_arg(char *script_path)
781 {
782 	struct script_desc *desc;
783 	int n_args = 0;
784 	char *p;
785 
786 	desc = script_desc__new(NULL);
787 
788 	if (read_script_info(desc, script_path))
789 		goto out;
790 
791 	if (!desc->args)
792 		goto out;
793 
794 	for (p = desc->args; *p; p++)
795 		if (*p == '<')
796 			n_args++;
797 out:
798 	script_desc__delete(desc);
799 
800 	return n_args;
801 }
802 
803 static const char * const script_usage[] = {
804 	"perf script [<options>]",
805 	"perf script [<options>] record <script> [<record-options>] <command>",
806 	"perf script [<options>] report <script> [script-args]",
807 	"perf script [<options>] <script> [<record-options>] <command>",
808 	"perf script [<options>] <top-script> [script-args]",
809 	NULL
810 };
811 
812 static const struct option options[] = {
813 	OPT_BOOLEAN('D', "dump-raw-trace", &dump_trace,
814 		    "dump raw trace in ASCII"),
815 	OPT_INCR('v', "verbose", &verbose,
816 		    "be more verbose (show symbol address, etc)"),
817 	OPT_BOOLEAN('L', "Latency", &latency_format,
818 		    "show latency attributes (irqs/preemption disabled, etc)"),
819 	OPT_CALLBACK_NOOPT('l', "list", NULL, NULL, "list available scripts",
820 			   list_available_scripts),
821 	OPT_CALLBACK('s', "script", NULL, "name",
822 		     "script file name (lang:script name, script name, or *)",
823 		     parse_scriptname),
824 	OPT_STRING('g', "gen-script", &generate_script_lang, "lang",
825 		   "generate perf-script.xx script in specified language"),
826 	OPT_STRING('i', "input", &input_name, "file",
827 		    "input file name"),
828 	OPT_BOOLEAN('d', "debug-mode", &debug_mode,
829 		   "do various checks like samples ordering and lost events"),
830 	OPT_STRING('k', "vmlinux", &symbol_conf.vmlinux_name,
831 		   "file", "vmlinux pathname"),
832 	OPT_STRING(0, "kallsyms", &symbol_conf.kallsyms_name,
833 		   "file", "kallsyms pathname"),
834 	OPT_BOOLEAN('G', "hide-call-graph", &no_callchain,
835 		    "When printing symbols do not display call chain"),
836 	OPT_STRING(0, "symfs", &symbol_conf.symfs, "directory",
837 		    "Look for files with symbols relative to this directory"),
838 	OPT_CALLBACK('f', "fields", NULL, "str",
839 		     "comma separated output fields prepend with 'type:'. Valid types: hw,sw,trace. Fields: comm,tid,pid,time,cpu,event,trace,sym",
840 		     parse_output_fields),
841 
842 	OPT_END()
843 };
844 
845 static bool have_cmd(int argc, const char **argv)
846 {
847 	char **__argv = malloc(sizeof(const char *) * argc);
848 
849 	if (!__argv)
850 		die("malloc");
851 	memcpy(__argv, argv, sizeof(const char *) * argc);
852 	argc = parse_options(argc, (const char **)__argv, record_options,
853 			     NULL, PARSE_OPT_STOP_AT_NON_OPTION);
854 	free(__argv);
855 
856 	return argc != 0;
857 }
858 
859 int cmd_script(int argc, const char **argv, const char *prefix __used)
860 {
861 	char *rec_script_path = NULL;
862 	char *rep_script_path = NULL;
863 	struct perf_session *session;
864 	char *script_path = NULL;
865 	const char **__argv;
866 	bool system_wide;
867 	int i, j, err;
868 
869 	setup_scripting();
870 
871 	argc = parse_options(argc, argv, options, script_usage,
872 			     PARSE_OPT_STOP_AT_NON_OPTION);
873 
874 	if (argc > 1 && !strncmp(argv[0], "rec", strlen("rec"))) {
875 		rec_script_path = get_script_path(argv[1], RECORD_SUFFIX);
876 		if (!rec_script_path)
877 			return cmd_record(argc, argv, NULL);
878 	}
879 
880 	if (argc > 1 && !strncmp(argv[0], "rep", strlen("rep"))) {
881 		rep_script_path = get_script_path(argv[1], REPORT_SUFFIX);
882 		if (!rep_script_path) {
883 			fprintf(stderr,
884 				"Please specify a valid report script"
885 				"(see 'perf script -l' for listing)\n");
886 			return -1;
887 		}
888 	}
889 
890 	/* make sure PERF_EXEC_PATH is set for scripts */
891 	perf_set_argv_exec_path(perf_exec_path());
892 
893 	if (argc && !script_name && !rec_script_path && !rep_script_path) {
894 		int live_pipe[2];
895 		int rep_args;
896 		pid_t pid;
897 
898 		rec_script_path = get_script_path(argv[0], RECORD_SUFFIX);
899 		rep_script_path = get_script_path(argv[0], REPORT_SUFFIX);
900 
901 		if (!rec_script_path && !rep_script_path) {
902 			fprintf(stderr, " Couldn't find script %s\n\n See perf"
903 				" script -l for available scripts.\n", argv[0]);
904 			usage_with_options(script_usage, options);
905 		}
906 
907 		if (is_top_script(argv[0])) {
908 			rep_args = argc - 1;
909 		} else {
910 			int rec_args;
911 
912 			rep_args = has_required_arg(rep_script_path);
913 			rec_args = (argc - 1) - rep_args;
914 			if (rec_args < 0) {
915 				fprintf(stderr, " %s script requires options."
916 					"\n\n See perf script -l for available "
917 					"scripts and options.\n", argv[0]);
918 				usage_with_options(script_usage, options);
919 			}
920 		}
921 
922 		if (pipe(live_pipe) < 0) {
923 			perror("failed to create pipe");
924 			exit(-1);
925 		}
926 
927 		pid = fork();
928 		if (pid < 0) {
929 			perror("failed to fork");
930 			exit(-1);
931 		}
932 
933 		if (!pid) {
934 			system_wide = true;
935 			j = 0;
936 
937 			dup2(live_pipe[1], 1);
938 			close(live_pipe[0]);
939 
940 			if (!is_top_script(argv[0]))
941 				system_wide = !have_cmd(argc - rep_args,
942 							&argv[rep_args]);
943 
944 			__argv = malloc((argc + 6) * sizeof(const char *));
945 			if (!__argv)
946 				die("malloc");
947 
948 			__argv[j++] = "/bin/sh";
949 			__argv[j++] = rec_script_path;
950 			if (system_wide)
951 				__argv[j++] = "-a";
952 			__argv[j++] = "-q";
953 			__argv[j++] = "-o";
954 			__argv[j++] = "-";
955 			for (i = rep_args + 1; i < argc; i++)
956 				__argv[j++] = argv[i];
957 			__argv[j++] = NULL;
958 
959 			execvp("/bin/sh", (char **)__argv);
960 			free(__argv);
961 			exit(-1);
962 		}
963 
964 		dup2(live_pipe[0], 0);
965 		close(live_pipe[1]);
966 
967 		__argv = malloc((argc + 4) * sizeof(const char *));
968 		if (!__argv)
969 			die("malloc");
970 		j = 0;
971 		__argv[j++] = "/bin/sh";
972 		__argv[j++] = rep_script_path;
973 		for (i = 1; i < rep_args + 1; i++)
974 			__argv[j++] = argv[i];
975 		__argv[j++] = "-i";
976 		__argv[j++] = "-";
977 		__argv[j++] = NULL;
978 
979 		execvp("/bin/sh", (char **)__argv);
980 		free(__argv);
981 		exit(-1);
982 	}
983 
984 	if (rec_script_path)
985 		script_path = rec_script_path;
986 	if (rep_script_path)
987 		script_path = rep_script_path;
988 
989 	if (script_path) {
990 		system_wide = false;
991 		j = 0;
992 
993 		if (rec_script_path)
994 			system_wide = !have_cmd(argc - 1, &argv[1]);
995 
996 		__argv = malloc((argc + 2) * sizeof(const char *));
997 		if (!__argv)
998 			die("malloc");
999 		__argv[j++] = "/bin/sh";
1000 		__argv[j++] = script_path;
1001 		if (system_wide)
1002 			__argv[j++] = "-a";
1003 		for (i = 2; i < argc; i++)
1004 			__argv[j++] = argv[i];
1005 		__argv[j++] = NULL;
1006 
1007 		execvp("/bin/sh", (char **)__argv);
1008 		free(__argv);
1009 		exit(-1);
1010 	}
1011 
1012 	if (symbol__init() < 0)
1013 		return -1;
1014 	if (!script_name)
1015 		setup_pager();
1016 
1017 	session = perf_session__new(input_name, O_RDONLY, 0, false, &event_ops);
1018 	if (session == NULL)
1019 		return -ENOMEM;
1020 
1021 	if (!no_callchain)
1022 		symbol_conf.use_callchain = true;
1023 	else
1024 		symbol_conf.use_callchain = false;
1025 
1026 	if (generate_script_lang) {
1027 		struct stat perf_stat;
1028 		int input;
1029 
1030 		if (output_set_by_user) {
1031 			fprintf(stderr,
1032 				"custom fields not supported for generated scripts");
1033 			return -1;
1034 		}
1035 
1036 		input = open(input_name, O_RDONLY);
1037 		if (input < 0) {
1038 			perror("failed to open file");
1039 			exit(-1);
1040 		}
1041 
1042 		err = fstat(input, &perf_stat);
1043 		if (err < 0) {
1044 			perror("failed to stat file");
1045 			exit(-1);
1046 		}
1047 
1048 		if (!perf_stat.st_size) {
1049 			fprintf(stderr, "zero-sized file, nothing to do!\n");
1050 			exit(0);
1051 		}
1052 
1053 		scripting_ops = script_spec__lookup(generate_script_lang);
1054 		if (!scripting_ops) {
1055 			fprintf(stderr, "invalid language specifier");
1056 			return -1;
1057 		}
1058 
1059 		err = scripting_ops->generate_script("perf-script");
1060 		goto out;
1061 	}
1062 
1063 	if (script_name) {
1064 		err = scripting_ops->start_script(script_name, argc, argv);
1065 		if (err)
1066 			goto out;
1067 		pr_debug("perf script started with script %s\n\n", script_name);
1068 	}
1069 
1070 	err = __cmd_script(session);
1071 
1072 	perf_session__delete(session);
1073 	cleanup_scripting();
1074 out:
1075 	return err;
1076 }
1077