tools/perf/util/exec_cmd.c

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