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