1 // SPDX-License-Identifier: GPL-2.0
2 #include "../../builtin.h"
3 #include "../../perf.h"
4 #include "../../util/util.h" // perf_exe()
5 #include "../util.h"
6 #include "../../util/hist.h"
7 #include "../../util/debug.h"
8 #include "../../util/symbol.h"
9 #include "../browser.h"
10 #include "../libslang.h"
11 #include "config.h"
12 #include <linux/string.h>
13 #include <linux/zalloc.h>
14 #include <stdlib.h>
15 
16 #define SCRIPT_NAMELEN	128
17 #define SCRIPT_MAX_NO	64
18 /*
19  * Usually the full path for a script is:
20  *	/home/username/libexec/perf-core/scripts/python/xxx.py
21  *	/home/username/libexec/perf-core/scripts/perl/xxx.pl
22  * So 256 should be long enough to contain the full path.
23  */
24 #define SCRIPT_FULLPATH_LEN	256
25 
26 struct script_config {
27 	const char **names;
28 	char **paths;
29 	int index;
30 	const char *perf;
31 	char extra_format[256];
32 };
33 
attr_to_script(char * extra_format,struct perf_event_attr * attr)34 void attr_to_script(char *extra_format, struct perf_event_attr *attr)
35 {
36 	extra_format[0] = 0;
37 	if (attr->read_format & PERF_FORMAT_GROUP)
38 		strcat(extra_format, " -F +metric");
39 	if (attr->sample_type & PERF_SAMPLE_BRANCH_STACK)
40 		strcat(extra_format, " -F +brstackinsn --xed");
41 	if (attr->sample_type & PERF_SAMPLE_REGS_INTR)
42 		strcat(extra_format, " -F +iregs");
43 	if (attr->sample_type & PERF_SAMPLE_REGS_USER)
44 		strcat(extra_format, " -F +uregs");
45 	if (attr->sample_type & PERF_SAMPLE_PHYS_ADDR)
46 		strcat(extra_format, " -F +phys_addr");
47 }
48 
add_script_option(const char * name,const char * opt,struct script_config * c)49 static int add_script_option(const char *name, const char *opt,
50 			     struct script_config *c)
51 {
52 	c->names[c->index] = name;
53 	if (asprintf(&c->paths[c->index],
54 		     "%s script %s -F +metric %s %s",
55 		     c->perf, opt, symbol_conf.inline_name ? " --inline" : "",
56 		     c->extra_format) < 0)
57 		return -1;
58 	c->index++;
59 	return 0;
60 }
61 
scripts_config(const char * var,const char * value,void * data)62 static int scripts_config(const char *var, const char *value, void *data)
63 {
64 	struct script_config *c = data;
65 
66 	if (!strstarts(var, "scripts."))
67 		return -1;
68 	if (c->index >= SCRIPT_MAX_NO)
69 		return -1;
70 	c->names[c->index] = strdup(var + 7);
71 	if (!c->names[c->index])
72 		return -1;
73 	if (asprintf(&c->paths[c->index], "%s %s", value,
74 		     c->extra_format) < 0)
75 		return -1;
76 	c->index++;
77 	return 0;
78 }
79 
80 /*
81  * When success, will copy the full path of the selected script
82  * into  the buffer pointed by script_name, and return 0.
83  * Return -1 on failure.
84  */
list_scripts(char * script_name,bool * custom,struct evsel * evsel)85 static int list_scripts(char *script_name, bool *custom,
86 			struct evsel *evsel)
87 {
88 	char *buf, *paths[SCRIPT_MAX_NO], *names[SCRIPT_MAX_NO];
89 	int i, num, choice;
90 	int ret = 0;
91 	int max_std, custom_perf;
92 	char pbuf[256];
93 	const char *perf = perf_exe(pbuf, sizeof pbuf);
94 	struct script_config scriptc = {
95 		.names = (const char **)names,
96 		.paths = paths,
97 		.perf = perf
98 	};
99 
100 	script_name[0] = 0;
101 
102 	/* Preset the script name to SCRIPT_NAMELEN */
103 	buf = malloc(SCRIPT_MAX_NO * (SCRIPT_NAMELEN + SCRIPT_FULLPATH_LEN));
104 	if (!buf)
105 		return -1;
106 
107 	if (evsel)
108 		attr_to_script(scriptc.extra_format, &evsel->core.attr);
109 	add_script_option("Show individual samples", "", &scriptc);
110 	add_script_option("Show individual samples with assembler", "-F +insn --xed",
111 			  &scriptc);
112 	add_script_option("Show individual samples with source", "-F +srcline,+srccode",
113 			  &scriptc);
114 	perf_config(scripts_config, &scriptc);
115 	custom_perf = scriptc.index;
116 	add_script_option("Show samples with custom perf script arguments", "", &scriptc);
117 	i = scriptc.index;
118 	max_std = i;
119 
120 	for (; i < SCRIPT_MAX_NO; i++) {
121 		names[i] = buf + (i - max_std) * (SCRIPT_NAMELEN + SCRIPT_FULLPATH_LEN);
122 		paths[i] = names[i] + SCRIPT_NAMELEN;
123 	}
124 
125 	num = find_scripts(names + max_std, paths + max_std, SCRIPT_MAX_NO - max_std,
126 			SCRIPT_FULLPATH_LEN);
127 	if (num < 0)
128 		num = 0;
129 	choice = ui__popup_menu(num + max_std, (char * const *)names, NULL);
130 	if (choice < 0) {
131 		ret = -1;
132 		goto out;
133 	}
134 	if (choice == custom_perf) {
135 		char script_args[50];
136 		int key = ui_browser__input_window("perf script command",
137 				"Enter perf script command line (without perf script prefix)",
138 				script_args, "", 0);
139 		if (key != K_ENTER) {
140 			ret = -1;
141 			goto out;
142 		}
143 		sprintf(script_name, "%s script %s", perf, script_args);
144 	} else if (choice < num + max_std) {
145 		strcpy(script_name, paths[choice]);
146 	}
147 	*custom = choice >= max_std;
148 
149 out:
150 	free(buf);
151 	for (i = 0; i < max_std; i++)
152 		zfree(&paths[i]);
153 	return ret;
154 }
155 
run_script(char * cmd)156 void run_script(char *cmd)
157 {
158 	pr_debug("Running %s\n", cmd);
159 	SLang_reset_tty();
160 	if (system(cmd) < 0)
161 		pr_warning("Cannot run %s\n", cmd);
162 	/*
163 	 * SLang doesn't seem to reset the whole terminal, so be more
164 	 * forceful to get back to the original state.
165 	 */
166 	printf("\033[c\033[H\033[J");
167 	fflush(stdout);
168 	SLang_init_tty(0, 0, 0);
169 	SLsmg_refresh();
170 }
171 
script_browse(const char * script_opt,struct evsel * evsel)172 int script_browse(const char *script_opt, struct evsel *evsel)
173 {
174 	char *cmd, script_name[SCRIPT_FULLPATH_LEN];
175 	bool custom = false;
176 
177 	memset(script_name, 0, SCRIPT_FULLPATH_LEN);
178 	if (list_scripts(script_name, &custom, evsel))
179 		return -1;
180 
181 	if (asprintf(&cmd, "%s%s %s %s%s 2>&1 | less",
182 			custom ? "perf script -s " : "",
183 			script_name,
184 			script_opt ? script_opt : "",
185 			input_name ? "-i " : "",
186 			input_name ? input_name : "") < 0)
187 		return -1;
188 
189 	run_script(cmd);
190 	free(cmd);
191 
192 	return 0;
193 }
194