Merge branch 'for-linus' of git://git.kernel.org/pub/scm/linux/kernel/git/kyle/parisc-2.6
[linux-2.6] / tools / perf / util / exec_cmd.c
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 }