Loading...
1/*
2 * perf.c
3 *
4 * Performance analysis utility.
5 *
6 * This is the main hub from which the sub-commands (perf stat,
7 * perf top, perf record, perf report, etc.) are started.
8 */
9#include "builtin.h"
10
11#include "util/exec_cmd.h"
12#include "util/cache.h"
13#include "util/quote.h"
14#include "util/run-command.h"
15#include "util/parse-events.h"
16#include <api/fs/debugfs.h>
17#include <pthread.h>
18
19const char perf_usage_string[] =
20 "perf [--version] [--help] COMMAND [ARGS]";
21
22const char perf_more_info_string[] =
23 "See 'perf help COMMAND' for more information on a specific command.";
24
25int use_browser = -1;
26static int use_pager = -1;
27const char *input_name;
28
29struct cmd_struct {
30 const char *cmd;
31 int (*fn)(int, const char **, const char *);
32 int option;
33};
34
35static struct cmd_struct commands[] = {
36 { "buildid-cache", cmd_buildid_cache, 0 },
37 { "buildid-list", cmd_buildid_list, 0 },
38 { "diff", cmd_diff, 0 },
39 { "evlist", cmd_evlist, 0 },
40 { "help", cmd_help, 0 },
41 { "list", cmd_list, 0 },
42 { "record", cmd_record, 0 },
43 { "report", cmd_report, 0 },
44 { "bench", cmd_bench, 0 },
45 { "stat", cmd_stat, 0 },
46 { "timechart", cmd_timechart, 0 },
47 { "top", cmd_top, 0 },
48 { "annotate", cmd_annotate, 0 },
49 { "version", cmd_version, 0 },
50 { "script", cmd_script, 0 },
51 { "sched", cmd_sched, 0 },
52#ifdef HAVE_LIBELF_SUPPORT
53 { "probe", cmd_probe, 0 },
54#endif
55 { "kmem", cmd_kmem, 0 },
56 { "lock", cmd_lock, 0 },
57 { "kvm", cmd_kvm, 0 },
58 { "test", cmd_test, 0 },
59#ifdef HAVE_LIBAUDIT_SUPPORT
60 { "trace", cmd_trace, 0 },
61#endif
62 { "inject", cmd_inject, 0 },
63 { "mem", cmd_mem, 0 },
64};
65
66struct pager_config {
67 const char *cmd;
68 int val;
69};
70
71static int pager_command_config(const char *var, const char *value, void *data)
72{
73 struct pager_config *c = data;
74 if (!prefixcmp(var, "pager.") && !strcmp(var + 6, c->cmd))
75 c->val = perf_config_bool(var, value);
76 return 0;
77}
78
79/* returns 0 for "no pager", 1 for "use pager", and -1 for "not specified" */
80int check_pager_config(const char *cmd)
81{
82 struct pager_config c;
83 c.cmd = cmd;
84 c.val = -1;
85 perf_config(pager_command_config, &c);
86 return c.val;
87}
88
89static int browser_command_config(const char *var, const char *value, void *data)
90{
91 struct pager_config *c = data;
92 if (!prefixcmp(var, "tui.") && !strcmp(var + 4, c->cmd))
93 c->val = perf_config_bool(var, value);
94 if (!prefixcmp(var, "gtk.") && !strcmp(var + 4, c->cmd))
95 c->val = perf_config_bool(var, value) ? 2 : 0;
96 return 0;
97}
98
99/*
100 * returns 0 for "no tui", 1 for "use tui", 2 for "use gtk",
101 * and -1 for "not specified"
102 */
103static int check_browser_config(const char *cmd)
104{
105 struct pager_config c;
106 c.cmd = cmd;
107 c.val = -1;
108 perf_config(browser_command_config, &c);
109 return c.val;
110}
111
112static void commit_pager_choice(void)
113{
114 switch (use_pager) {
115 case 0:
116 setenv("PERF_PAGER", "cat", 1);
117 break;
118 case 1:
119 /* setup_pager(); */
120 break;
121 default:
122 break;
123 }
124}
125
126static int handle_options(const char ***argv, int *argc, int *envchanged)
127{
128 int handled = 0;
129
130 while (*argc > 0) {
131 const char *cmd = (*argv)[0];
132 if (cmd[0] != '-')
133 break;
134
135 /*
136 * For legacy reasons, the "version" and "help"
137 * commands can be written with "--" prepended
138 * to make them look like flags.
139 */
140 if (!strcmp(cmd, "--help") || !strcmp(cmd, "--version"))
141 break;
142
143 /*
144 * Check remaining flags.
145 */
146 if (!prefixcmp(cmd, CMD_EXEC_PATH)) {
147 cmd += strlen(CMD_EXEC_PATH);
148 if (*cmd == '=')
149 perf_set_argv_exec_path(cmd + 1);
150 else {
151 puts(perf_exec_path());
152 exit(0);
153 }
154 } else if (!strcmp(cmd, "--html-path")) {
155 puts(system_path(PERF_HTML_PATH));
156 exit(0);
157 } else if (!strcmp(cmd, "-p") || !strcmp(cmd, "--paginate")) {
158 use_pager = 1;
159 } else if (!strcmp(cmd, "--no-pager")) {
160 use_pager = 0;
161 if (envchanged)
162 *envchanged = 1;
163 } else if (!strcmp(cmd, "--perf-dir")) {
164 if (*argc < 2) {
165 fprintf(stderr, "No directory given for --perf-dir.\n");
166 usage(perf_usage_string);
167 }
168 setenv(PERF_DIR_ENVIRONMENT, (*argv)[1], 1);
169 if (envchanged)
170 *envchanged = 1;
171 (*argv)++;
172 (*argc)--;
173 handled++;
174 } else if (!prefixcmp(cmd, CMD_PERF_DIR)) {
175 setenv(PERF_DIR_ENVIRONMENT, cmd + strlen(CMD_PERF_DIR), 1);
176 if (envchanged)
177 *envchanged = 1;
178 } else if (!strcmp(cmd, "--work-tree")) {
179 if (*argc < 2) {
180 fprintf(stderr, "No directory given for --work-tree.\n");
181 usage(perf_usage_string);
182 }
183 setenv(PERF_WORK_TREE_ENVIRONMENT, (*argv)[1], 1);
184 if (envchanged)
185 *envchanged = 1;
186 (*argv)++;
187 (*argc)--;
188 } else if (!prefixcmp(cmd, CMD_WORK_TREE)) {
189 setenv(PERF_WORK_TREE_ENVIRONMENT, cmd + strlen(CMD_WORK_TREE), 1);
190 if (envchanged)
191 *envchanged = 1;
192 } else if (!strcmp(cmd, "--debugfs-dir")) {
193 if (*argc < 2) {
194 fprintf(stderr, "No directory given for --debugfs-dir.\n");
195 usage(perf_usage_string);
196 }
197 perf_debugfs_set_path((*argv)[1]);
198 if (envchanged)
199 *envchanged = 1;
200 (*argv)++;
201 (*argc)--;
202 } else if (!prefixcmp(cmd, CMD_DEBUGFS_DIR)) {
203 perf_debugfs_set_path(cmd + strlen(CMD_DEBUGFS_DIR));
204 fprintf(stderr, "dir: %s\n", debugfs_mountpoint);
205 if (envchanged)
206 *envchanged = 1;
207 } else if (!strcmp(cmd, "--list-cmds")) {
208 unsigned int i;
209
210 for (i = 0; i < ARRAY_SIZE(commands); i++) {
211 struct cmd_struct *p = commands+i;
212 printf("%s ", p->cmd);
213 }
214 exit(0);
215 } else {
216 fprintf(stderr, "Unknown option: %s\n", cmd);
217 usage(perf_usage_string);
218 }
219
220 (*argv)++;
221 (*argc)--;
222 handled++;
223 }
224 return handled;
225}
226
227static int handle_alias(int *argcp, const char ***argv)
228{
229 int envchanged = 0, ret = 0, saved_errno = errno;
230 int count, option_count;
231 const char **new_argv;
232 const char *alias_command;
233 char *alias_string;
234
235 alias_command = (*argv)[0];
236 alias_string = alias_lookup(alias_command);
237 if (alias_string) {
238 if (alias_string[0] == '!') {
239 if (*argcp > 1) {
240 struct strbuf buf;
241
242 strbuf_init(&buf, PATH_MAX);
243 strbuf_addstr(&buf, alias_string);
244 sq_quote_argv(&buf, (*argv) + 1, PATH_MAX);
245 free(alias_string);
246 alias_string = buf.buf;
247 }
248 ret = system(alias_string + 1);
249 if (ret >= 0 && WIFEXITED(ret) &&
250 WEXITSTATUS(ret) != 127)
251 exit(WEXITSTATUS(ret));
252 die("Failed to run '%s' when expanding alias '%s'",
253 alias_string + 1, alias_command);
254 }
255 count = split_cmdline(alias_string, &new_argv);
256 if (count < 0)
257 die("Bad alias.%s string", alias_command);
258 option_count = handle_options(&new_argv, &count, &envchanged);
259 if (envchanged)
260 die("alias '%s' changes environment variables\n"
261 "You can use '!perf' in the alias to do this.",
262 alias_command);
263 memmove(new_argv - option_count, new_argv,
264 count * sizeof(char *));
265 new_argv -= option_count;
266
267 if (count < 1)
268 die("empty alias for %s", alias_command);
269
270 if (!strcmp(alias_command, new_argv[0]))
271 die("recursive alias: %s", alias_command);
272
273 new_argv = realloc(new_argv, sizeof(char *) *
274 (count + *argcp + 1));
275 /* insert after command name */
276 memcpy(new_argv + count, *argv + 1, sizeof(char *) * *argcp);
277 new_argv[count + *argcp] = NULL;
278
279 *argv = new_argv;
280 *argcp += count - 1;
281
282 ret = 1;
283 }
284
285 errno = saved_errno;
286
287 return ret;
288}
289
290const char perf_version_string[] = PERF_VERSION;
291
292#define RUN_SETUP (1<<0)
293#define USE_PAGER (1<<1)
294/*
295 * require working tree to be present -- anything uses this needs
296 * RUN_SETUP for reading from the configuration file.
297 */
298#define NEED_WORK_TREE (1<<2)
299
300static int run_builtin(struct cmd_struct *p, int argc, const char **argv)
301{
302 int status;
303 struct stat st;
304 const char *prefix;
305
306 prefix = NULL;
307 if (p->option & RUN_SETUP)
308 prefix = NULL; /* setup_perf_directory(); */
309
310 if (use_browser == -1)
311 use_browser = check_browser_config(p->cmd);
312
313 if (use_pager == -1 && p->option & RUN_SETUP)
314 use_pager = check_pager_config(p->cmd);
315 if (use_pager == -1 && p->option & USE_PAGER)
316 use_pager = 1;
317 commit_pager_choice();
318
319 status = p->fn(argc, argv, prefix);
320 exit_browser(status);
321
322 if (status)
323 return status & 0xff;
324
325 /* Somebody closed stdout? */
326 if (fstat(fileno(stdout), &st))
327 return 0;
328 /* Ignore write errors for pipes and sockets.. */
329 if (S_ISFIFO(st.st_mode) || S_ISSOCK(st.st_mode))
330 return 0;
331
332 status = 1;
333 /* Check for ENOSPC and EIO errors.. */
334 if (fflush(stdout)) {
335 fprintf(stderr, "write failure on standard output: %s", strerror(errno));
336 goto out;
337 }
338 if (ferror(stdout)) {
339 fprintf(stderr, "unknown write failure on standard output");
340 goto out;
341 }
342 if (fclose(stdout)) {
343 fprintf(stderr, "close failed on standard output: %s", strerror(errno));
344 goto out;
345 }
346 status = 0;
347out:
348 return status;
349}
350
351static void handle_internal_command(int argc, const char **argv)
352{
353 const char *cmd = argv[0];
354 unsigned int i;
355 static const char ext[] = STRIP_EXTENSION;
356
357 if (sizeof(ext) > 1) {
358 i = strlen(argv[0]) - strlen(ext);
359 if (i > 0 && !strcmp(argv[0] + i, ext)) {
360 char *argv0 = strdup(argv[0]);
361 argv[0] = cmd = argv0;
362 argv0[i] = '\0';
363 }
364 }
365
366 /* Turn "perf cmd --help" into "perf help cmd" */
367 if (argc > 1 && !strcmp(argv[1], "--help")) {
368 argv[1] = argv[0];
369 argv[0] = cmd = "help";
370 }
371
372 for (i = 0; i < ARRAY_SIZE(commands); i++) {
373 struct cmd_struct *p = commands+i;
374 if (strcmp(p->cmd, cmd))
375 continue;
376 exit(run_builtin(p, argc, argv));
377 }
378}
379
380static void execv_dashed_external(const char **argv)
381{
382 struct strbuf cmd = STRBUF_INIT;
383 const char *tmp;
384 int status;
385
386 strbuf_addf(&cmd, "perf-%s", argv[0]);
387
388 /*
389 * argv[0] must be the perf command, but the argv array
390 * belongs to the caller, and may be reused in
391 * subsequent loop iterations. Save argv[0] and
392 * restore it on error.
393 */
394 tmp = argv[0];
395 argv[0] = cmd.buf;
396
397 /*
398 * if we fail because the command is not found, it is
399 * OK to return. Otherwise, we just pass along the status code.
400 */
401 status = run_command_v_opt(argv, 0);
402 if (status != -ERR_RUN_COMMAND_EXEC) {
403 if (IS_RUN_COMMAND_ERR(status))
404 die("unable to run '%s'", argv[0]);
405 exit(-status);
406 }
407 errno = ENOENT; /* as if we called execvp */
408
409 argv[0] = tmp;
410
411 strbuf_release(&cmd);
412}
413
414static int run_argv(int *argcp, const char ***argv)
415{
416 int done_alias = 0;
417
418 while (1) {
419 /* See if it's an internal command */
420 handle_internal_command(*argcp, *argv);
421
422 /* .. then try the external ones */
423 execv_dashed_external(*argv);
424
425 /* It could be an alias -- this works around the insanity
426 * of overriding "perf log" with "perf show" by having
427 * alias.log = show
428 */
429 if (done_alias || !handle_alias(argcp, argv))
430 break;
431 done_alias = 1;
432 }
433
434 return done_alias;
435}
436
437static void pthread__block_sigwinch(void)
438{
439 sigset_t set;
440
441 sigemptyset(&set);
442 sigaddset(&set, SIGWINCH);
443 pthread_sigmask(SIG_BLOCK, &set, NULL);
444}
445
446void pthread__unblock_sigwinch(void)
447{
448 sigset_t set;
449
450 sigemptyset(&set);
451 sigaddset(&set, SIGWINCH);
452 pthread_sigmask(SIG_UNBLOCK, &set, NULL);
453}
454
455int main(int argc, const char **argv)
456{
457 const char *cmd;
458
459 /* The page_size is placed in util object. */
460 page_size = sysconf(_SC_PAGE_SIZE);
461
462 cmd = perf_extract_argv0_path(argv[0]);
463 if (!cmd)
464 cmd = "perf-help";
465 /* get debugfs mount point from /proc/mounts */
466 perf_debugfs_mount(NULL);
467 /*
468 * "perf-xxxx" is the same as "perf xxxx", but we obviously:
469 *
470 * - cannot take flags in between the "perf" and the "xxxx".
471 * - cannot execute it externally (since it would just do
472 * the same thing over again)
473 *
474 * So we just directly call the internal command handler, and
475 * die if that one cannot handle it.
476 */
477 if (!prefixcmp(cmd, "perf-")) {
478 cmd += 5;
479 argv[0] = cmd;
480 handle_internal_command(argc, argv);
481 fprintf(stderr, "cannot handle %s internally", cmd);
482 goto out;
483 }
484#ifdef HAVE_LIBAUDIT_SUPPORT
485 if (!prefixcmp(cmd, "trace")) {
486 set_buildid_dir();
487 setup_path();
488 argv[0] = "trace";
489 return cmd_trace(argc, argv, NULL);
490 }
491#endif
492 /* Look for flags.. */
493 argv++;
494 argc--;
495 handle_options(&argv, &argc, NULL);
496 commit_pager_choice();
497 set_buildid_dir();
498
499 if (argc > 0) {
500 if (!prefixcmp(argv[0], "--"))
501 argv[0] += 2;
502 } else {
503 /* The user didn't specify a command; give them help */
504 printf("\n usage: %s\n\n", perf_usage_string);
505 list_common_cmds_help();
506 printf("\n %s\n\n", perf_more_info_string);
507 goto out;
508 }
509 cmd = argv[0];
510
511 test_attr__init();
512
513 /*
514 * We use PATH to find perf commands, but we prepend some higher
515 * precedence paths: the "--exec-path" option, the PERF_EXEC_PATH
516 * environment, and the $(perfexecdir) from the Makefile at build
517 * time.
518 */
519 setup_path();
520 /*
521 * Block SIGWINCH notifications so that the thread that wants it can
522 * unblock and get syscalls like select interrupted instead of waiting
523 * forever while the signal goes to some other non interested thread.
524 */
525 pthread__block_sigwinch();
526
527 while (1) {
528 static int done_help;
529 int was_alias = run_argv(&argc, &argv);
530
531 if (errno != ENOENT)
532 break;
533
534 if (was_alias) {
535 fprintf(stderr, "Expansion of alias '%s' failed; "
536 "'%s' is not a perf-command\n",
537 cmd, argv[0]);
538 goto out;
539 }
540 if (!done_help) {
541 cmd = argv[0] = help_unknown_cmd(cmd);
542 done_help = 1;
543 } else
544 break;
545 }
546
547 fprintf(stderr, "Failed to run command '%s': %s\n",
548 cmd, strerror(errno));
549out:
550 return 1;
551}
1/*
2 * perf.c
3 *
4 * Performance analysis utility.
5 *
6 * This is the main hub from which the sub-commands (perf stat,
7 * perf top, perf record, perf report, etc.) are started.
8 */
9#include "builtin.h"
10
11#include "util/exec_cmd.h"
12#include "util/cache.h"
13#include "util/quote.h"
14#include "util/run-command.h"
15#include "util/parse-events.h"
16#include "util/debugfs.h"
17
18const char perf_usage_string[] =
19 "perf [--version] [--help] COMMAND [ARGS]";
20
21const char perf_more_info_string[] =
22 "See 'perf help COMMAND' for more information on a specific command.";
23
24int use_browser = -1;
25static int use_pager = -1;
26
27struct pager_config {
28 const char *cmd;
29 int val;
30};
31
32static int pager_command_config(const char *var, const char *value, void *data)
33{
34 struct pager_config *c = data;
35 if (!prefixcmp(var, "pager.") && !strcmp(var + 6, c->cmd))
36 c->val = perf_config_bool(var, value);
37 return 0;
38}
39
40/* returns 0 for "no pager", 1 for "use pager", and -1 for "not specified" */
41int check_pager_config(const char *cmd)
42{
43 struct pager_config c;
44 c.cmd = cmd;
45 c.val = -1;
46 perf_config(pager_command_config, &c);
47 return c.val;
48}
49
50static int tui_command_config(const char *var, const char *value, void *data)
51{
52 struct pager_config *c = data;
53 if (!prefixcmp(var, "tui.") && !strcmp(var + 4, c->cmd))
54 c->val = perf_config_bool(var, value);
55 return 0;
56}
57
58/* returns 0 for "no tui", 1 for "use tui", and -1 for "not specified" */
59static int check_tui_config(const char *cmd)
60{
61 struct pager_config c;
62 c.cmd = cmd;
63 c.val = -1;
64 perf_config(tui_command_config, &c);
65 return c.val;
66}
67
68static void commit_pager_choice(void)
69{
70 switch (use_pager) {
71 case 0:
72 setenv("PERF_PAGER", "cat", 1);
73 break;
74 case 1:
75 /* setup_pager(); */
76 break;
77 default:
78 break;
79 }
80}
81
82static int handle_options(const char ***argv, int *argc, int *envchanged)
83{
84 int handled = 0;
85
86 while (*argc > 0) {
87 const char *cmd = (*argv)[0];
88 if (cmd[0] != '-')
89 break;
90
91 /*
92 * For legacy reasons, the "version" and "help"
93 * commands can be written with "--" prepended
94 * to make them look like flags.
95 */
96 if (!strcmp(cmd, "--help") || !strcmp(cmd, "--version"))
97 break;
98
99 /*
100 * Check remaining flags.
101 */
102 if (!prefixcmp(cmd, CMD_EXEC_PATH)) {
103 cmd += strlen(CMD_EXEC_PATH);
104 if (*cmd == '=')
105 perf_set_argv_exec_path(cmd + 1);
106 else {
107 puts(perf_exec_path());
108 exit(0);
109 }
110 } else if (!strcmp(cmd, "--html-path")) {
111 puts(system_path(PERF_HTML_PATH));
112 exit(0);
113 } else if (!strcmp(cmd, "-p") || !strcmp(cmd, "--paginate")) {
114 use_pager = 1;
115 } else if (!strcmp(cmd, "--no-pager")) {
116 use_pager = 0;
117 if (envchanged)
118 *envchanged = 1;
119 } else if (!strcmp(cmd, "--perf-dir")) {
120 if (*argc < 2) {
121 fprintf(stderr, "No directory given for --perf-dir.\n");
122 usage(perf_usage_string);
123 }
124 setenv(PERF_DIR_ENVIRONMENT, (*argv)[1], 1);
125 if (envchanged)
126 *envchanged = 1;
127 (*argv)++;
128 (*argc)--;
129 handled++;
130 } else if (!prefixcmp(cmd, CMD_PERF_DIR)) {
131 setenv(PERF_DIR_ENVIRONMENT, cmd + strlen(CMD_PERF_DIR), 1);
132 if (envchanged)
133 *envchanged = 1;
134 } else if (!strcmp(cmd, "--work-tree")) {
135 if (*argc < 2) {
136 fprintf(stderr, "No directory given for --work-tree.\n");
137 usage(perf_usage_string);
138 }
139 setenv(PERF_WORK_TREE_ENVIRONMENT, (*argv)[1], 1);
140 if (envchanged)
141 *envchanged = 1;
142 (*argv)++;
143 (*argc)--;
144 } else if (!prefixcmp(cmd, CMD_WORK_TREE)) {
145 setenv(PERF_WORK_TREE_ENVIRONMENT, cmd + strlen(CMD_WORK_TREE), 1);
146 if (envchanged)
147 *envchanged = 1;
148 } else if (!strcmp(cmd, "--debugfs-dir")) {
149 if (*argc < 2) {
150 fprintf(stderr, "No directory given for --debugfs-dir.\n");
151 usage(perf_usage_string);
152 }
153 debugfs_set_path((*argv)[1]);
154 if (envchanged)
155 *envchanged = 1;
156 (*argv)++;
157 (*argc)--;
158 } else if (!prefixcmp(cmd, CMD_DEBUGFS_DIR)) {
159 debugfs_set_path(cmd + strlen(CMD_DEBUGFS_DIR));
160 fprintf(stderr, "dir: %s\n", debugfs_mountpoint);
161 if (envchanged)
162 *envchanged = 1;
163 } else {
164 fprintf(stderr, "Unknown option: %s\n", cmd);
165 usage(perf_usage_string);
166 }
167
168 (*argv)++;
169 (*argc)--;
170 handled++;
171 }
172 return handled;
173}
174
175static int handle_alias(int *argcp, const char ***argv)
176{
177 int envchanged = 0, ret = 0, saved_errno = errno;
178 int count, option_count;
179 const char **new_argv;
180 const char *alias_command;
181 char *alias_string;
182
183 alias_command = (*argv)[0];
184 alias_string = alias_lookup(alias_command);
185 if (alias_string) {
186 if (alias_string[0] == '!') {
187 if (*argcp > 1) {
188 struct strbuf buf;
189
190 strbuf_init(&buf, PATH_MAX);
191 strbuf_addstr(&buf, alias_string);
192 sq_quote_argv(&buf, (*argv) + 1, PATH_MAX);
193 free(alias_string);
194 alias_string = buf.buf;
195 }
196 ret = system(alias_string + 1);
197 if (ret >= 0 && WIFEXITED(ret) &&
198 WEXITSTATUS(ret) != 127)
199 exit(WEXITSTATUS(ret));
200 die("Failed to run '%s' when expanding alias '%s'",
201 alias_string + 1, alias_command);
202 }
203 count = split_cmdline(alias_string, &new_argv);
204 if (count < 0)
205 die("Bad alias.%s string", alias_command);
206 option_count = handle_options(&new_argv, &count, &envchanged);
207 if (envchanged)
208 die("alias '%s' changes environment variables\n"
209 "You can use '!perf' in the alias to do this.",
210 alias_command);
211 memmove(new_argv - option_count, new_argv,
212 count * sizeof(char *));
213 new_argv -= option_count;
214
215 if (count < 1)
216 die("empty alias for %s", alias_command);
217
218 if (!strcmp(alias_command, new_argv[0]))
219 die("recursive alias: %s", alias_command);
220
221 new_argv = realloc(new_argv, sizeof(char *) *
222 (count + *argcp + 1));
223 /* insert after command name */
224 memcpy(new_argv + count, *argv + 1, sizeof(char *) * *argcp);
225 new_argv[count + *argcp] = NULL;
226
227 *argv = new_argv;
228 *argcp += count - 1;
229
230 ret = 1;
231 }
232
233 errno = saved_errno;
234
235 return ret;
236}
237
238const char perf_version_string[] = PERF_VERSION;
239
240#define RUN_SETUP (1<<0)
241#define USE_PAGER (1<<1)
242/*
243 * require working tree to be present -- anything uses this needs
244 * RUN_SETUP for reading from the configuration file.
245 */
246#define NEED_WORK_TREE (1<<2)
247
248struct cmd_struct {
249 const char *cmd;
250 int (*fn)(int, const char **, const char *);
251 int option;
252};
253
254static int run_builtin(struct cmd_struct *p, int argc, const char **argv)
255{
256 int status;
257 struct stat st;
258 const char *prefix;
259
260 prefix = NULL;
261 if (p->option & RUN_SETUP)
262 prefix = NULL; /* setup_perf_directory(); */
263
264 if (use_browser == -1)
265 use_browser = check_tui_config(p->cmd);
266
267 if (use_pager == -1 && p->option & RUN_SETUP)
268 use_pager = check_pager_config(p->cmd);
269 if (use_pager == -1 && p->option & USE_PAGER)
270 use_pager = 1;
271 commit_pager_choice();
272
273 status = p->fn(argc, argv, prefix);
274 exit_browser(status);
275
276 if (status)
277 return status & 0xff;
278
279 /* Somebody closed stdout? */
280 if (fstat(fileno(stdout), &st))
281 return 0;
282 /* Ignore write errors for pipes and sockets.. */
283 if (S_ISFIFO(st.st_mode) || S_ISSOCK(st.st_mode))
284 return 0;
285
286 /* Check for ENOSPC and EIO errors.. */
287 if (fflush(stdout))
288 die("write failure on standard output: %s", strerror(errno));
289 if (ferror(stdout))
290 die("unknown write failure on standard output");
291 if (fclose(stdout))
292 die("close failed on standard output: %s", strerror(errno));
293 return 0;
294}
295
296static void handle_internal_command(int argc, const char **argv)
297{
298 const char *cmd = argv[0];
299 static struct cmd_struct commands[] = {
300 { "buildid-cache", cmd_buildid_cache, 0 },
301 { "buildid-list", cmd_buildid_list, 0 },
302 { "diff", cmd_diff, 0 },
303 { "evlist", cmd_evlist, 0 },
304 { "help", cmd_help, 0 },
305 { "list", cmd_list, 0 },
306 { "record", cmd_record, 0 },
307 { "report", cmd_report, 0 },
308 { "bench", cmd_bench, 0 },
309 { "stat", cmd_stat, 0 },
310 { "timechart", cmd_timechart, 0 },
311 { "top", cmd_top, 0 },
312 { "annotate", cmd_annotate, 0 },
313 { "version", cmd_version, 0 },
314 { "script", cmd_script, 0 },
315 { "sched", cmd_sched, 0 },
316 { "probe", cmd_probe, 0 },
317 { "kmem", cmd_kmem, 0 },
318 { "lock", cmd_lock, 0 },
319 { "kvm", cmd_kvm, 0 },
320 { "test", cmd_test, 0 },
321 { "inject", cmd_inject, 0 },
322 };
323 unsigned int i;
324 static const char ext[] = STRIP_EXTENSION;
325
326 if (sizeof(ext) > 1) {
327 i = strlen(argv[0]) - strlen(ext);
328 if (i > 0 && !strcmp(argv[0] + i, ext)) {
329 char *argv0 = strdup(argv[0]);
330 argv[0] = cmd = argv0;
331 argv0[i] = '\0';
332 }
333 }
334
335 /* Turn "perf cmd --help" into "perf help cmd" */
336 if (argc > 1 && !strcmp(argv[1], "--help")) {
337 argv[1] = argv[0];
338 argv[0] = cmd = "help";
339 }
340
341 for (i = 0; i < ARRAY_SIZE(commands); i++) {
342 struct cmd_struct *p = commands+i;
343 if (strcmp(p->cmd, cmd))
344 continue;
345 exit(run_builtin(p, argc, argv));
346 }
347}
348
349static void execv_dashed_external(const char **argv)
350{
351 struct strbuf cmd = STRBUF_INIT;
352 const char *tmp;
353 int status;
354
355 strbuf_addf(&cmd, "perf-%s", argv[0]);
356
357 /*
358 * argv[0] must be the perf command, but the argv array
359 * belongs to the caller, and may be reused in
360 * subsequent loop iterations. Save argv[0] and
361 * restore it on error.
362 */
363 tmp = argv[0];
364 argv[0] = cmd.buf;
365
366 /*
367 * if we fail because the command is not found, it is
368 * OK to return. Otherwise, we just pass along the status code.
369 */
370 status = run_command_v_opt(argv, 0);
371 if (status != -ERR_RUN_COMMAND_EXEC) {
372 if (IS_RUN_COMMAND_ERR(status))
373 die("unable to run '%s'", argv[0]);
374 exit(-status);
375 }
376 errno = ENOENT; /* as if we called execvp */
377
378 argv[0] = tmp;
379
380 strbuf_release(&cmd);
381}
382
383static int run_argv(int *argcp, const char ***argv)
384{
385 int done_alias = 0;
386
387 while (1) {
388 /* See if it's an internal command */
389 handle_internal_command(*argcp, *argv);
390
391 /* .. then try the external ones */
392 execv_dashed_external(*argv);
393
394 /* It could be an alias -- this works around the insanity
395 * of overriding "perf log" with "perf show" by having
396 * alias.log = show
397 */
398 if (done_alias || !handle_alias(argcp, argv))
399 break;
400 done_alias = 1;
401 }
402
403 return done_alias;
404}
405
406static void pthread__block_sigwinch(void)
407{
408 sigset_t set;
409
410 sigemptyset(&set);
411 sigaddset(&set, SIGWINCH);
412 pthread_sigmask(SIG_BLOCK, &set, NULL);
413}
414
415void pthread__unblock_sigwinch(void)
416{
417 sigset_t set;
418
419 sigemptyset(&set);
420 sigaddset(&set, SIGWINCH);
421 pthread_sigmask(SIG_UNBLOCK, &set, NULL);
422}
423
424int main(int argc, const char **argv)
425{
426 const char *cmd;
427
428 cmd = perf_extract_argv0_path(argv[0]);
429 if (!cmd)
430 cmd = "perf-help";
431 /* get debugfs mount point from /proc/mounts */
432 debugfs_mount(NULL);
433 /*
434 * "perf-xxxx" is the same as "perf xxxx", but we obviously:
435 *
436 * - cannot take flags in between the "perf" and the "xxxx".
437 * - cannot execute it externally (since it would just do
438 * the same thing over again)
439 *
440 * So we just directly call the internal command handler, and
441 * die if that one cannot handle it.
442 */
443 if (!prefixcmp(cmd, "perf-")) {
444 cmd += 5;
445 argv[0] = cmd;
446 handle_internal_command(argc, argv);
447 die("cannot handle %s internally", cmd);
448 }
449
450 /* Look for flags.. */
451 argv++;
452 argc--;
453 handle_options(&argv, &argc, NULL);
454 commit_pager_choice();
455 set_buildid_dir();
456
457 if (argc > 0) {
458 if (!prefixcmp(argv[0], "--"))
459 argv[0] += 2;
460 } else {
461 /* The user didn't specify a command; give them help */
462 printf("\n usage: %s\n\n", perf_usage_string);
463 list_common_cmds_help();
464 printf("\n %s\n\n", perf_more_info_string);
465 exit(1);
466 }
467 cmd = argv[0];
468
469 /*
470 * We use PATH to find perf commands, but we prepend some higher
471 * precedence paths: the "--exec-path" option, the PERF_EXEC_PATH
472 * environment, and the $(perfexecdir) from the Makefile at build
473 * time.
474 */
475 setup_path();
476 /*
477 * Block SIGWINCH notifications so that the thread that wants it can
478 * unblock and get syscalls like select interrupted instead of waiting
479 * forever while the signal goes to some other non interested thread.
480 */
481 pthread__block_sigwinch();
482
483 while (1) {
484 static int done_help;
485 static int was_alias;
486
487 was_alias = run_argv(&argc, &argv);
488 if (errno != ENOENT)
489 break;
490
491 if (was_alias) {
492 fprintf(stderr, "Expansion of alias '%s' failed; "
493 "'%s' is not a perf-command\n",
494 cmd, argv[0]);
495 exit(1);
496 }
497 if (!done_help) {
498 cmd = argv[0] = help_unknown_cmd(cmd);
499 done_help = 1;
500 } else
501 break;
502 }
503
504 fprintf(stderr, "Failed to run command '%s': %s\n",
505 cmd, strerror(errno));
506
507 return 1;
508}