1#include "cache.h"
  2#include "exec_cmd.h"
  3#include "quote.h"
  4
  5#include <string.h>
  6
  7#define MAX_ARGS	32
  8
  9static const char *argv_exec_path;
 10static const char *argv0_path;
 11
 12const char *system_path(const char *path)
 13{
 14	static const char *prefix = PREFIX;
 15	struct strbuf d = STRBUF_INIT;
 16
 17	if (is_absolute_path(path))
 18		return path;
 19
 20	strbuf_addf(&d, "%s/%s", prefix, path);
 21	path = strbuf_detach(&d, NULL);
 22	return path;
 23}
 24
 25const char *perf_extract_argv0_path(const char *argv0)
 26{
 27	const char *slash;
 28
 29	if (!argv0 || !*argv0)
 30		return NULL;
 31	slash = argv0 + strlen(argv0);
 32
 33	while (argv0 <= slash && !is_dir_sep(*slash))
 34		slash--;
 35
 36	if (slash >= argv0) {
 37		argv0_path = strndup(argv0, slash - argv0);
 38		return argv0_path ? slash + 1 : NULL;
 39	}
 40
 41	return argv0;
 42}
 43
 44void perf_set_argv_exec_path(const char *exec_path)
 45{
 46	argv_exec_path = exec_path;
 47	/*
 48	 * Propagate this setting to external programs.
 49	 */
 50	setenv(EXEC_PATH_ENVIRONMENT, exec_path, 1);
 51}
 52
 53
 54/* Returns the highest-priority, location to look for perf programs. */
 55const char *perf_exec_path(void)
 56{
 57	const char *env;
 58
 59	if (argv_exec_path)
 60		return argv_exec_path;
 61
 62	env = getenv(EXEC_PATH_ENVIRONMENT);
 63	if (env && *env) {
 64		return env;
 65	}
 66
 67	return system_path(PERF_EXEC_PATH);
 68}
 69
 70static void add_path(struct strbuf *out, const char *path)
 71{
 72	if (path && *path) {
 73		if (is_absolute_path(path))
 74			strbuf_addstr(out, path);
 75		else
 76			strbuf_addstr(out, make_nonrelative_path(path));
 77
 78		strbuf_addch(out, PATH_SEP);
 79	}
 80}
 81
 82void setup_path(void)
 83{
 84	const char *old_path = getenv("PATH");
 85	struct strbuf new_path = STRBUF_INIT;
 86
 87	add_path(&new_path, perf_exec_path());
 88	add_path(&new_path, argv0_path);
 89
 90	if (old_path)
 91		strbuf_addstr(&new_path, old_path);
 92	else
 93		strbuf_addstr(&new_path, "/usr/local/bin:/usr/bin:/bin");
 94
 95	setenv("PATH", new_path.buf, 1);
 96
 97	strbuf_release(&new_path);
 98}
 99
100static const char **prepare_perf_cmd(const char **argv)
101{
102	int argc;
103	const char **nargv;
104
105	for (argc = 0; argv[argc]; argc++)
106		; /* just counting */
107	nargv = malloc(sizeof(*nargv) * (argc + 2));
108
109	nargv[0] = "perf";
110	for (argc = 0; argv[argc]; argc++)
111		nargv[argc + 1] = argv[argc];
112	nargv[argc + 1] = NULL;
113	return nargv;
114}
115
116int execv_perf_cmd(const char **argv) {
117	const char **nargv = prepare_perf_cmd(argv);
118
119	/* execvp() can only ever return if it fails */
120	execvp("perf", (char **)nargv);
121
122	free(nargv);
123	return -1;
124}
125
126
127int execl_perf_cmd(const char *cmd,...)
128{
129	int argc;
130	const char *argv[MAX_ARGS + 1];
131	const char *arg;
132	va_list param;
133
134	va_start(param, cmd);
135	argv[0] = cmd;
136	argc = 1;
137	while (argc < MAX_ARGS) {
138		arg = argv[argc++] = va_arg(param, char *);
139		if (!arg)
140			break;
141	}
142	va_end(param);
143	if (MAX_ARGS <= argc)
144		return error("too many args to run %s", cmd);
145
146	argv[argc] = NULL;
147	return execv_perf_cmd(argv);
148}