Josef Weidendorfer | b1bf95b | 2005-07-31 21:17:43 +0200 | [diff] [blame] | 1 | #ifndef RUN_COMMAND_H |
| 2 | #define RUN_COMMAND_H |
| 3 | |
Johannes Sixt | f6b6098 | 2010-03-09 21:00:36 +0100 | [diff] [blame] | 4 | #ifndef NO_PTHREADS |
Johannes Sixt | 200a76b | 2010-03-06 16:40:42 +0100 | [diff] [blame] | 5 | #include <pthread.h> |
| 6 | #endif |
| 7 | |
Jeff King | c460c0e | 2014-05-15 04:33:26 -0400 | [diff] [blame] | 8 | #include "argv-array.h" |
| 9 | |
Shawn O. Pearce | f100089 | 2007-03-10 03:28:00 -0500 | [diff] [blame] | 10 | struct child_process { |
| 11 | const char **argv; |
Jeff King | c460c0e | 2014-05-15 04:33:26 -0400 | [diff] [blame] | 12 | struct argv_array args; |
René Scharfe | 19a583d | 2014-10-19 13:13:55 +0200 | [diff] [blame] | 13 | struct argv_array env_array; |
Shawn O. Pearce | ebcb5d1 | 2007-03-10 03:28:05 -0500 | [diff] [blame] | 14 | pid_t pid; |
Johannes Sixt | c20181e | 2008-02-21 23:42:56 +0100 | [diff] [blame] | 15 | /* |
| 16 | * Using .in, .out, .err: |
| 17 | * - Specify 0 for no redirections (child inherits stdin, stdout, |
| 18 | * stderr from parent). |
| 19 | * - Specify -1 to have a pipe allocated as follows: |
| 20 | * .in: returns the writable pipe end; parent writes to it, |
| 21 | * the readable pipe end becomes child's stdin |
| 22 | * .out, .err: returns the readable pipe end; parent reads from |
| 23 | * it, the writable pipe end becomes child's stdout/stderr |
| 24 | * The caller of start_command() must close the returned FDs |
| 25 | * after it has completed reading from/writing to it! |
| 26 | * - Specify > 0 to set a channel to a particular FD as follows: |
| 27 | * .in: a readable FD, becomes child's stdin |
| 28 | * .out: a writable FD, becomes child's stdout/stderr |
Shawn O. Pearce | 4f41b61 | 2010-02-05 12:57:37 -0800 | [diff] [blame] | 29 | * .err: a writable FD, becomes child's stderr |
Johannes Sixt | c20181e | 2008-02-21 23:42:56 +0100 | [diff] [blame] | 30 | * The specified FD is closed by start_command(), even in case |
| 31 | * of errors! |
| 32 | */ |
Shawn O. Pearce | 4919bf0 | 2007-03-10 03:28:08 -0500 | [diff] [blame] | 33 | int in; |
Shawn O. Pearce | f4bba25 | 2007-03-12 14:37:45 -0400 | [diff] [blame] | 34 | int out; |
Johannes Sixt | f3b33f1 | 2007-10-19 21:47:58 +0200 | [diff] [blame] | 35 | int err; |
Alex Riesen | 1568fea | 2007-05-22 23:48:23 +0200 | [diff] [blame] | 36 | const char *dir; |
Alex Riesen | ee49314 | 2007-05-22 23:48:47 +0200 | [diff] [blame] | 37 | const char *const *env; |
Shawn O. Pearce | f100089 | 2007-03-10 03:28:00 -0500 | [diff] [blame] | 38 | unsigned no_stdin:1; |
Shawn O. Pearce | e4507ae | 2007-03-12 14:37:55 -0400 | [diff] [blame] | 39 | unsigned no_stdout:1; |
Shawn O. Pearce | b73a439 | 2007-11-11 02:29:37 -0500 | [diff] [blame] | 40 | unsigned no_stderr:1; |
Shawn O. Pearce | f100089 | 2007-03-10 03:28:00 -0500 | [diff] [blame] | 41 | unsigned git_cmd:1; /* if this is to be git sub-command */ |
Johannes Sixt | c024beb | 2009-07-04 21:26:42 +0200 | [diff] [blame] | 42 | unsigned silent_exec_failure:1; |
Shawn O. Pearce | f100089 | 2007-03-10 03:28:00 -0500 | [diff] [blame] | 43 | unsigned stdout_to_stderr:1; |
Jeff King | 8dba1e6 | 2009-12-30 05:53:16 -0500 | [diff] [blame] | 44 | unsigned use_shell:1; |
Jeff King | afe19ff | 2012-01-07 12:42:43 +0100 | [diff] [blame] | 45 | unsigned clean_on_exit:1; |
Shawn O. Pearce | f100089 | 2007-03-10 03:28:00 -0500 | [diff] [blame] | 46 | }; |
| 47 | |
René Scharfe | 19a583d | 2014-10-19 13:13:55 +0200 | [diff] [blame] | 48 | #define CHILD_PROCESS_INIT { NULL, ARGV_ARRAY_INIT, ARGV_ARRAY_INIT } |
René Scharfe | 483bbd4 | 2014-08-19 21:10:48 +0200 | [diff] [blame] | 49 | void child_process_init(struct child_process *); |
René Scharfe | 2d71608 | 2015-10-24 14:11:27 +0200 | [diff] [blame] | 50 | void child_process_clear(struct child_process *); |
René Scharfe | d318027 | 2014-08-19 21:09:35 +0200 | [diff] [blame] | 51 | |
Shawn O. Pearce | ebcb5d1 | 2007-03-10 03:28:05 -0500 | [diff] [blame] | 52 | int start_command(struct child_process *); |
| 53 | int finish_command(struct child_process *); |
Takashi Iwai | 507d780 | 2015-09-04 11:35:57 +0200 | [diff] [blame] | 54 | int finish_command_in_signal(struct child_process *); |
Shawn O. Pearce | f100089 | 2007-03-10 03:28:00 -0500 | [diff] [blame] | 55 | int run_command(struct child_process *); |
| 56 | |
Jeff King | 03f2c77 | 2015-08-10 05:37:45 -0400 | [diff] [blame] | 57 | /* |
| 58 | * Returns the path to the hook file, or NULL if the hook is missing |
| 59 | * or disabled. Note that this points to static storage that will be |
| 60 | * overwritten by further calls to find_hook and run_hook_*. |
| 61 | */ |
Nguyễn Thái Ngọc Duy | dcf6926 | 2014-11-30 15:24:27 +0700 | [diff] [blame] | 62 | extern const char *find_hook(const char *name); |
Ramsay Jones | 9fe3edc | 2013-07-18 21:02:12 +0100 | [diff] [blame] | 63 | LAST_ARG_MUST_BE_NULL |
Benoit Pierre | 15048f8 | 2014-03-18 11:00:53 +0100 | [diff] [blame] | 64 | extern int run_hook_le(const char *const *env, const char *name, ...); |
| 65 | extern int run_hook_ve(const char *const *env, const char *name, va_list args); |
| 66 | |
Shawn O. Pearce | 95d3c4f | 2006-12-30 21:55:22 -0500 | [diff] [blame] | 67 | #define RUN_COMMAND_NO_STDIN 1 |
Michal Ostrowski | 77cb17e | 2006-01-10 21:12:17 -0500 | [diff] [blame] | 68 | #define RUN_GIT_CMD 2 /*If this is to be git sub-command */ |
Shawn O. Pearce | cd83c74 | 2006-12-30 21:55:19 -0500 | [diff] [blame] | 69 | #define RUN_COMMAND_STDOUT_TO_STDERR 4 |
Johannes Sixt | c024beb | 2009-07-04 21:26:42 +0200 | [diff] [blame] | 70 | #define RUN_SILENT_EXEC_FAILURE 8 |
Jeff King | 8dba1e6 | 2009-12-30 05:53:16 -0500 | [diff] [blame] | 71 | #define RUN_USING_SHELL 16 |
Clemens Buchacher | 10c6cdd | 2012-01-08 21:41:09 +0100 | [diff] [blame] | 72 | #define RUN_CLEAN_ON_EXIT 32 |
Shawn O. Pearce | 9b0b509 | 2006-12-30 21:55:15 -0500 | [diff] [blame] | 73 | int run_command_v_opt(const char **argv, int opt); |
Alex Riesen | 3427b37 | 2007-05-23 22:21:39 +0200 | [diff] [blame] | 74 | |
| 75 | /* |
| 76 | * env (the environment) is to be formatted like environ: "VAR=VALUE". |
| 77 | * To unset an environment variable use just "VAR". |
| 78 | */ |
Alex Riesen | ee49314 | 2007-05-22 23:48:47 +0200 | [diff] [blame] | 79 | int run_command_v_opt_cd_env(const char **argv, int opt, const char *dir, const char *const *env); |
Josef Weidendorfer | b1bf95b | 2005-07-31 21:17:43 +0200 | [diff] [blame] | 80 | |
Jeff King | 911ec99 | 2015-03-22 23:53:43 -0400 | [diff] [blame] | 81 | /** |
| 82 | * Execute the given command, capturing its stdout in the given strbuf. |
| 83 | * Returns -1 if starting the command fails or reading fails, and otherwise |
| 84 | * returns the exit code of the command. The output collected in the |
| 85 | * buffer is kept even if the command returns a non-zero exit. The hint field |
| 86 | * gives a starting size for the strbuf allocation. |
| 87 | * |
| 88 | * The fields of "cmd" should be set up as they would for a normal run_command |
| 89 | * invocation. But note that there is no need to set cmd->out; the function |
| 90 | * sets it up for the caller. |
| 91 | */ |
| 92 | int capture_command(struct child_process *cmd, struct strbuf *buf, size_t hint); |
| 93 | |
Johannes Sixt | 2d22c20 | 2007-10-19 21:48:00 +0200 | [diff] [blame] | 94 | /* |
| 95 | * The purpose of the following functions is to feed a pipe by running |
| 96 | * a function asynchronously and providing output that the caller reads. |
| 97 | * |
| 98 | * It is expected that no synchronization and mutual exclusion between |
| 99 | * the caller and the feed function is necessary so that the function |
| 100 | * can run in a thread without interfering with the caller. |
| 101 | */ |
| 102 | struct async { |
| 103 | /* |
Erik Faye-Lund | ae6a560 | 2010-02-05 12:57:38 -0800 | [diff] [blame] | 104 | * proc reads from in; closes it before return |
| 105 | * proc writes to out; closes it before return |
Johannes Sixt | 2d22c20 | 2007-10-19 21:48:00 +0200 | [diff] [blame] | 106 | * returns 0 on success, non-zero on failure |
| 107 | */ |
Erik Faye-Lund | ae6a560 | 2010-02-05 12:57:38 -0800 | [diff] [blame] | 108 | int (*proc)(int in, int out, void *data); |
Johannes Sixt | 2d22c20 | 2007-10-19 21:48:00 +0200 | [diff] [blame] | 109 | void *data; |
Erik Faye-Lund | ae6a560 | 2010-02-05 12:57:38 -0800 | [diff] [blame] | 110 | int in; /* caller writes here and closes it */ |
Johannes Sixt | 2d22c20 | 2007-10-19 21:48:00 +0200 | [diff] [blame] | 111 | int out; /* caller reads from here and closes it */ |
Johannes Sixt | f6b6098 | 2010-03-09 21:00:36 +0100 | [diff] [blame] | 112 | #ifdef NO_PTHREADS |
Johannes Sixt | 2d22c20 | 2007-10-19 21:48:00 +0200 | [diff] [blame] | 113 | pid_t pid; |
Johannes Sixt | 618ebe9 | 2007-12-08 22:19:14 +0100 | [diff] [blame] | 114 | #else |
Johannes Sixt | 200a76b | 2010-03-06 16:40:42 +0100 | [diff] [blame] | 115 | pthread_t tid; |
Erik Faye-Lund | ae6a560 | 2010-02-05 12:57:38 -0800 | [diff] [blame] | 116 | int proc_in; |
| 117 | int proc_out; |
Johannes Sixt | 618ebe9 | 2007-12-08 22:19:14 +0100 | [diff] [blame] | 118 | #endif |
Jeff King | c792d7b | 2016-04-19 18:49:41 -0400 | [diff] [blame] | 119 | int isolate_sigpipe; |
Johannes Sixt | 2d22c20 | 2007-10-19 21:48:00 +0200 | [diff] [blame] | 120 | }; |
| 121 | |
| 122 | int start_async(struct async *async); |
| 123 | int finish_async(struct async *async); |
Jeff King | 661a8cf | 2015-09-01 16:22:43 -0400 | [diff] [blame] | 124 | int in_async(void); |
Jeff King | 9658846 | 2016-02-24 02:40:16 -0500 | [diff] [blame] | 125 | void NORETURN async_exit(int code); |
Johannes Sixt | 2d22c20 | 2007-10-19 21:48:00 +0200 | [diff] [blame] | 126 | |
Stefan Beller | c553c72 | 2015-12-15 16:04:10 -0800 | [diff] [blame] | 127 | /** |
| 128 | * This callback should initialize the child process and preload the |
| 129 | * error channel if desired. The preloading of is useful if you want to |
| 130 | * have a message printed directly before the output of the child process. |
| 131 | * pp_cb is the callback cookie as passed to run_processes_parallel. |
| 132 | * You can store a child process specific callback cookie in pp_task_cb. |
| 133 | * |
| 134 | * Even after returning 0 to indicate that there are no more processes, |
| 135 | * this function will be called again until there are no more running |
| 136 | * child processes. |
| 137 | * |
| 138 | * Return 1 if the next child is ready to run. |
| 139 | * Return 0 if there are currently no more tasks to be processed. |
| 140 | * To send a signal to other child processes for abortion, |
| 141 | * return the negative signal number. |
| 142 | */ |
| 143 | typedef int (*get_next_task_fn)(struct child_process *cp, |
Stefan Beller | aa71049 | 2016-02-29 18:07:16 -0800 | [diff] [blame] | 144 | struct strbuf *out, |
Stefan Beller | c553c72 | 2015-12-15 16:04:10 -0800 | [diff] [blame] | 145 | void *pp_cb, |
| 146 | void **pp_task_cb); |
| 147 | |
| 148 | /** |
| 149 | * This callback is called whenever there are problems starting |
| 150 | * a new process. |
| 151 | * |
| 152 | * You must not write to stdout or stderr in this function. Add your |
Stefan Beller | aa71049 | 2016-02-29 18:07:16 -0800 | [diff] [blame] | 153 | * message to the strbuf out instead, which will be printed without |
Stefan Beller | c553c72 | 2015-12-15 16:04:10 -0800 | [diff] [blame] | 154 | * messing up the output of the other parallel processes. |
| 155 | * |
| 156 | * pp_cb is the callback cookie as passed into run_processes_parallel, |
| 157 | * pp_task_cb is the callback cookie as passed into get_next_task_fn. |
| 158 | * |
| 159 | * Return 0 to continue the parallel processing. To abort return non zero. |
| 160 | * To send a signal to other child processes for abortion, return |
| 161 | * the negative signal number. |
| 162 | */ |
Stefan Beller | aa71049 | 2016-02-29 18:07:16 -0800 | [diff] [blame] | 163 | typedef int (*start_failure_fn)(struct strbuf *out, |
Stefan Beller | c553c72 | 2015-12-15 16:04:10 -0800 | [diff] [blame] | 164 | void *pp_cb, |
| 165 | void *pp_task_cb); |
| 166 | |
| 167 | /** |
| 168 | * This callback is called on every child process that finished processing. |
| 169 | * |
| 170 | * You must not write to stdout or stderr in this function. Add your |
Stefan Beller | aa71049 | 2016-02-29 18:07:16 -0800 | [diff] [blame] | 171 | * message to the strbuf out instead, which will be printed without |
Stefan Beller | c553c72 | 2015-12-15 16:04:10 -0800 | [diff] [blame] | 172 | * messing up the output of the other parallel processes. |
| 173 | * |
| 174 | * pp_cb is the callback cookie as passed into run_processes_parallel, |
| 175 | * pp_task_cb is the callback cookie as passed into get_next_task_fn. |
| 176 | * |
| 177 | * Return 0 to continue the parallel processing. To abort return non zero. |
| 178 | * To send a signal to other child processes for abortion, return |
| 179 | * the negative signal number. |
| 180 | */ |
| 181 | typedef int (*task_finished_fn)(int result, |
Stefan Beller | aa71049 | 2016-02-29 18:07:16 -0800 | [diff] [blame] | 182 | struct strbuf *out, |
Stefan Beller | c553c72 | 2015-12-15 16:04:10 -0800 | [diff] [blame] | 183 | void *pp_cb, |
| 184 | void *pp_task_cb); |
| 185 | |
| 186 | /** |
| 187 | * Runs up to n processes at the same time. Whenever a process can be |
| 188 | * started, the callback get_next_task_fn is called to obtain the data |
| 189 | * required to start another child process. |
| 190 | * |
| 191 | * The children started via this function run in parallel. Their output |
| 192 | * (both stdout and stderr) is routed to stderr in a manner that output |
| 193 | * from different tasks does not interleave. |
| 194 | * |
Stefan Beller | 2a73b3d | 2016-02-29 13:57:06 -0800 | [diff] [blame] | 195 | * start_failure_fn and task_finished_fn can be NULL to omit any |
| 196 | * special handling. |
Stefan Beller | c553c72 | 2015-12-15 16:04:10 -0800 | [diff] [blame] | 197 | */ |
| 198 | int run_processes_parallel(int n, |
| 199 | get_next_task_fn, |
| 200 | start_failure_fn, |
| 201 | task_finished_fn, |
| 202 | void *pp_cb); |
| 203 | |
Josef Weidendorfer | b1bf95b | 2005-07-31 21:17:43 +0200 | [diff] [blame] | 204 | #endif |