[thirdparty/git.git] / run-command.c

#include "cache.h"
#include "run-command.h"
#include "exec_cmd.h"

static inline void close_pair(int fd[2])
{
	close(fd[0]);
	close(fd[1]);
}

static inline void dup_devnull(int to)
{
	int fd = open("/dev/null", O_RDWR);
	dup2(fd, to);
	close(fd);
}

int start_command(struct child_process *cmd)
{
	int need_in, need_out, need_err;
	int fdin[2], fdout[2], fderr[2];

	/*
	 * In case of errors we must keep the promise to close FDs
	 * that have been passed in via ->in and ->out.
	 */

	need_in = !cmd->no_stdin && cmd->in < 0;
	if (need_in) {
		if (pipe(fdin) < 0) {
			if (cmd->out > 0)
				close(cmd->out);
			return -ERR_RUN_COMMAND_PIPE;
		}
		cmd->in = fdin[1];
	}

	need_out = !cmd->no_stdout
		&& !cmd->stdout_to_stderr
		&& cmd->out < 0;
	if (need_out) {
		if (pipe(fdout) < 0) {
			if (need_in)
				close_pair(fdin);
			else if (cmd->in)
				close(cmd->in);
			return -ERR_RUN_COMMAND_PIPE;
		}
		cmd->out = fdout[0];
	}

	need_err = !cmd->no_stderr && cmd->err < 0;
	if (need_err) {
		if (pipe(fderr) < 0) {
			if (need_in)
				close_pair(fdin);
			else if (cmd->in)
				close(cmd->in);
			if (need_out)
				close_pair(fdout);
			else if (cmd->out)
				close(cmd->out);
			return -ERR_RUN_COMMAND_PIPE;
		}
		cmd->err = fderr[0];
	}

	trace_argv_printf(cmd->argv, "trace: run_command:");

#ifndef __MINGW32__
	cmd->pid = fork();
	if (!cmd->pid) {
		if (cmd->no_stdin)
			dup_devnull(0);
		else if (need_in) {
			dup2(fdin[0], 0);
			close_pair(fdin);
		} else if (cmd->in) {
			dup2(cmd->in, 0);
			close(cmd->in);
		}

		if (cmd->no_stderr)
			dup_devnull(2);
		else if (need_err) {
			dup2(fderr[1], 2);
			close_pair(fderr);
		}

		if (cmd->no_stdout)
			dup_devnull(1);
		else if (cmd->stdout_to_stderr)
			dup2(2, 1);
		else if (need_out) {
			dup2(fdout[1], 1);
			close_pair(fdout);
		} else if (cmd->out > 1) {
			dup2(cmd->out, 1);
			close(cmd->out);
		}

		if (cmd->dir && chdir(cmd->dir))
			die("exec %s: cd to %s failed (%s)", cmd->argv[0],
			    cmd->dir, strerror(errno));
		if (cmd->env) {
			for (; *cmd->env; cmd->env++) {
				if (strchr(*cmd->env, '='))
					putenv((char*)*cmd->env);
				else
					unsetenv(*cmd->env);
			}
		}
		if (cmd->git_cmd) {
			execv_git_cmd(cmd->argv);
		} else {
			execvp(cmd->argv[0], (char *const*) cmd->argv);
		}
		die("exec %s failed.", cmd->argv[0]);
	}
#else
	int s0 = -1, s1 = -1, s2 = -1;	/* backups of stdin, stdout, stderr */
	const char **sargv = cmd->argv;
	char **env = environ;

	if (cmd->no_stdin) {
		s0 = dup(0);
		dup_devnull(0);
	} else if (need_in) {
		s0 = dup(0);
		dup2(fdin[0], 0);
	} else if (cmd->in) {
		s0 = dup(0);
		dup2(cmd->in, 0);
	}

	if (cmd->no_stderr) {
		s2 = dup(2);
		dup_devnull(2);
	} else if (need_err) {
		s2 = dup(2);
		dup2(fderr[1], 2);
	}

	if (cmd->no_stdout) {
		s1 = dup(1);
		dup_devnull(1);
	} else if (cmd->stdout_to_stderr) {
		s1 = dup(1);
		dup2(2, 1);
	} else if (need_out) {
		s1 = dup(1);
		dup2(fdout[1], 1);
	} else if (cmd->out > 1) {
		s1 = dup(1);
		dup2(cmd->out, 1);
	}

	if (cmd->dir)
		die("chdir in start_command() not implemented");
	if (cmd->env) {
		env = copy_environ();
		for (; *cmd->env; cmd->env++)
			env = env_setenv(env, *cmd->env);
	}

	if (cmd->git_cmd) {
		cmd->argv = prepare_git_cmd(cmd->argv);
	}

	cmd->pid = mingw_spawnvpe(cmd->argv[0], cmd->argv, env);

	if (cmd->env)
		free_environ(env);
	if (cmd->git_cmd)
		free(cmd->argv);

	cmd->argv = sargv;
	if (s0 >= 0)
		dup2(s0, 0), close(s0);
	if (s1 >= 0)
		dup2(s1, 1), close(s1);
	if (s2 >= 0)
		dup2(s2, 2), close(s2);
#endif

	if (cmd->pid < 0) {
		if (need_in)
			close_pair(fdin);
		else if (cmd->in)
			close(cmd->in);
		if (need_out)
			close_pair(fdout);
		else if (cmd->out)
			close(cmd->out);
		if (need_err)
			close_pair(fderr);
		return -ERR_RUN_COMMAND_FORK;
	}

	if (need_in)
		close(fdin[0]);
	else if (cmd->in)
		close(cmd->in);

	if (need_out)
		close(fdout[1]);
	else if (cmd->out)
		close(cmd->out);

	if (need_err)
		close(fderr[1]);

	return 0;
}

static int wait_or_whine(pid_t pid)
{
	for (;;) {
		int status, code;
		pid_t waiting = waitpid(pid, &status, 0);

		if (waiting < 0) {
			if (errno == EINTR)
				continue;
			error("waitpid failed (%s)", strerror(errno));
			return -ERR_RUN_COMMAND_WAITPID;
		}
		if (waiting != pid)
			return -ERR_RUN_COMMAND_WAITPID_WRONG_PID;
		if (WIFSIGNALED(status))
			return -ERR_RUN_COMMAND_WAITPID_SIGNAL;

		if (!WIFEXITED(status))
			return -ERR_RUN_COMMAND_WAITPID_NOEXIT;
		code = WEXITSTATUS(status);
		if (code)
			return -code;
		return 0;
	}
}

int finish_command(struct child_process *cmd)
{
	return wait_or_whine(cmd->pid);
}

int run_command(struct child_process *cmd)
{
	int code = start_command(cmd);
	if (code)
		return code;
	return finish_command(cmd);
}

static void prepare_run_command_v_opt(struct child_process *cmd,
				      const char **argv,
				      int opt)
{
	memset(cmd, 0, sizeof(*cmd));
	cmd->argv = argv;
	cmd->no_stdin = opt & RUN_COMMAND_NO_STDIN ? 1 : 0;
	cmd->git_cmd = opt & RUN_GIT_CMD ? 1 : 0;
	cmd->stdout_to_stderr = opt & RUN_COMMAND_STDOUT_TO_STDERR ? 1 : 0;
}

int run_command_v_opt(const char **argv, int opt)
{
	struct child_process cmd;
	prepare_run_command_v_opt(&cmd, argv, opt);
	return run_command(&cmd);
}

int run_command_v_opt_cd(const char **argv, int opt, const char *dir)
{
	struct child_process cmd;
	prepare_run_command_v_opt(&cmd, argv, opt);
	cmd.dir = dir;
	return run_command(&cmd);
}

int run_command_v_opt_cd_env(const char **argv, int opt, const char *dir, const char *const *env)
{
	struct child_process cmd;
	prepare_run_command_v_opt(&cmd, argv, opt);
	cmd.dir = dir;
	cmd.env = env;
	return run_command(&cmd);
}

#ifdef __MINGW32__
static __stdcall unsigned run_thread(void *data)
{
	struct async *async = data;
	return async->proc(async->fd_for_proc, async->data);
}
#endif

int start_async(struct async *async)
{
	int pipe_out[2];

	if (pipe(pipe_out) < 0)
		return error("cannot create pipe: %s", strerror(errno));
	async->out = pipe_out[0];

#ifndef __MINGW32__
	async->pid = fork();
	if (async->pid < 0) {
		error("fork (async) failed: %s", strerror(errno));
		close_pair(pipe_out);
		return -1;
	}
	if (!async->pid) {
		close(pipe_out[0]);
		exit(!!async->proc(pipe_out[1], async->data));
	}
	close(pipe_out[1]);
#else
	async->fd_for_proc = pipe_out[1];
	async->tid = (HANDLE) _beginthreadex(NULL, 0, run_thread, async, 0, NULL);
	if (!async->tid) {
		error("cannot create thread: %s", strerror(errno));
		close_pair(pipe_out);
		return -1;
	}
#endif
	return 0;
}

int finish_async(struct async *async)
{
#ifndef __MINGW32__
	int ret = 0;

	if (wait_or_whine(async->pid))
		ret = error("waitpid (async) failed");
#else
	DWORD ret = 0;
	if (WaitForSingleObject(async->tid, INFINITE) != WAIT_OBJECT_0)
		ret = error("waiting for thread failed: %lu", GetLastError());
	else if (!GetExitCodeThread(async->tid, &ret))
		ret = error("cannot get thread exit code: %lu", GetLastError());
	CloseHandle(async->tid);
#endif
	return ret;
}
Commit	Line	Data
b1bf95bb JW	1	#include "cache.h"
b1bf95bb JW	2	#include "run-command.h"
77cb17e9	3	#include "exec_cmd.h"
b1bf95bb	4
9dc09c76 SP	5	static inline void close_pair(int fd[2])
	6	{
	7	close(fd[0]);
	8	close(fd[1]);
	9	}
	10
e4507ae8 SP	11	static inline void dup_devnull(int to)
	12	{
	13	int fd = open("/dev/null", O_RDWR);
	14	dup2(fd, to);
	15	close(fd);
	16	}
	17
ebcb5d16	18	int start_command(struct child_process *cmd)
b1bf95bb	19	{
f3b33f1d JS	20	int need_in, need_out, need_err;
f3b33f1d JS	21	int fdin[2], fdout[2], fderr[2];
4919bf03	22
c20181e3 JS	23	/*
	24	* In case of errors we must keep the promise to close FDs
	25	* that have been passed in via ->in and ->out.
	26	*/
	27
e4507ae8	28	need_in = !cmd->no_stdin && cmd->in < 0;
4919bf03	29	if (need_in) {
c20181e3 JS	30	if (pipe(fdin) < 0) {
	31	if (cmd->out > 0)
	32	close(cmd->out);
4919bf03	33	return -ERR_RUN_COMMAND_PIPE;
c20181e3	34	}
4919bf03	35	cmd->in = fdin[1];
4919bf03 SP	36	}
4919bf03 SP	37
e4507ae8 SP	38	need_out = !cmd->no_stdout
	39	&& !cmd->stdout_to_stderr
	40	&& cmd->out < 0;
f4bba25b SP	41	if (need_out) {
	42	if (pipe(fdout) < 0) {
	43	if (need_in)
	44	close_pair(fdin);
c20181e3 JS	45	else if (cmd->in)
c20181e3 JS	46	close(cmd->in);
f4bba25b SP	47	return -ERR_RUN_COMMAND_PIPE;
	48	}
	49	cmd->out = fdout[0];
f4bba25b SP	50	}
f4bba25b SP	51
b73a4397	52	need_err = !cmd->no_stderr && cmd->err < 0;
f3b33f1d JS	53	if (need_err) {
	54	if (pipe(fderr) < 0) {
	55	if (need_in)
	56	close_pair(fdin);
c20181e3 JS	57	else if (cmd->in)
c20181e3 JS	58	close(cmd->in);
f3b33f1d JS	59	if (need_out)
f3b33f1d JS	60	close_pair(fdout);
c20181e3 JS	61	else if (cmd->out)
c20181e3 JS	62	close(cmd->out);
f3b33f1d JS	63	return -ERR_RUN_COMMAND_PIPE;
	64	}
	65	cmd->err = fderr[0];
	66	}
	67
8852f5d7 JS	68	trace_argv_printf(cmd->argv, "trace: run_command:");
8852f5d7 JS	69
ba26f296	70	#ifndef __MINGW32__
ebcb5d16	71	cmd->pid = fork();
ebcb5d16	72	if (!cmd->pid) {
e4507ae8 SP	73	if (cmd->no_stdin)
	74	dup_devnull(0);
	75	else if (need_in) {
4919bf03	76	dup2(fdin[0], 0);
9dc09c76	77	close_pair(fdin);
4919bf03 SP	78	} else if (cmd->in) {
	79	dup2(cmd->in, 0);
	80	close(cmd->in);
95d3c4f5	81	}
4919bf03	82
ce2cf27a CC	83	if (cmd->no_stderr)
	84	dup_devnull(2);
	85	else if (need_err) {
	86	dup2(fderr[1], 2);
	87	close_pair(fderr);
	88	}
	89
e4507ae8 SP	90	if (cmd->no_stdout)
	91	dup_devnull(1);
	92	else if (cmd->stdout_to_stderr)
cd83c74c	93	dup2(2, 1);
f4bba25b SP	94	else if (need_out) {
	95	dup2(fdout[1], 1);
	96	close_pair(fdout);
	97	} else if (cmd->out > 1) {
	98	dup2(cmd->out, 1);
	99	close(cmd->out);
	100	}
	101
1568fea0 AR	102	if (cmd->dir && chdir(cmd->dir))
	103	die("exec %s: cd to %s failed (%s)", cmd->argv[0],
	104	cmd->dir, strerror(errno));
ee493148	105	if (cmd->env) {
3427b375 AR	106	for (; *cmd->env; cmd->env++) {
	107	if (strchr(*cmd->env, '='))
	108	putenv((char)cmd->env);
	109	else
	110	unsetenv(*cmd->env);
	111	}
ee493148	112	}
f1000898 SP	113	if (cmd->git_cmd) {
f1000898 SP	114	execv_git_cmd(cmd->argv);
77cb17e9	115	} else {
f1000898	116	execvp(cmd->argv[0], (char const) cmd->argv);
128aed68	117	}
f1000898	118	die("exec %s failed.", cmd->argv[0]);
b1bf95bb	119	}
ba26f296 JS	120	#else
ba26f296 JS	121	int s0 = -1, s1 = -1, s2 = -1; /* backups of stdin, stdout, stderr */
108ac313	122	const char **sargv = cmd->argv;
ba26f296	123	char **env = environ;
ba26f296 JS	124
	125	if (cmd->no_stdin) {
	126	s0 = dup(0);
	127	dup_devnull(0);
	128	} else if (need_in) {
	129	s0 = dup(0);
	130	dup2(fdin[0], 0);
	131	} else if (cmd->in) {
	132	s0 = dup(0);
	133	dup2(cmd->in, 0);
	134	}
	135
	136	if (cmd->no_stderr) {
	137	s2 = dup(2);
	138	dup_devnull(2);
	139	} else if (need_err) {
	140	s2 = dup(2);
	141	dup2(fderr[1], 2);
	142	}
	143
	144	if (cmd->no_stdout) {
	145	s1 = dup(1);
	146	dup_devnull(1);
	147	} else if (cmd->stdout_to_stderr) {
	148	s1 = dup(1);
	149	dup2(2, 1);
	150	} else if (need_out) {
	151	s1 = dup(1);
	152	dup2(fdout[1], 1);
	153	} else if (cmd->out > 1) {
	154	s1 = dup(1);
	155	dup2(cmd->out, 1);
	156	}
	157
	158	if (cmd->dir)
	159	die("chdir in start_command() not implemented");
	160	if (cmd->env) {
	161	env = copy_environ();
	162	for (; *cmd->env; cmd->env++)
	163	env = env_setenv(env, *cmd->env);
	164	}
	165
	166	if (cmd->git_cmd) {
108ac313	167	cmd->argv = prepare_git_cmd(cmd->argv);
ba26f296 JS	168	}
ba26f296 JS	169
7e5d7768	170	cmd->pid = mingw_spawnvpe(cmd->argv[0], cmd->argv, env);
ba26f296 JS	171
	172	if (cmd->env)
	173	free_environ(env);
	174	if (cmd->git_cmd)
108ac313	175	free(cmd->argv);
ba26f296	176
108ac313	177	cmd->argv = sargv;
ba26f296 JS	178	if (s0 >= 0)
	179	dup2(s0, 0), close(s0);
	180	if (s1 >= 0)
	181	dup2(s1, 1), close(s1);
	182	if (s2 >= 0)
	183	dup2(s2, 2), close(s2);
	184	#endif
	185
	186	if (cmd->pid < 0) {
	187	if (need_in)
	188	close_pair(fdin);
	189	else if (cmd->in)
	190	close(cmd->in);
	191	if (need_out)
	192	close_pair(fdout);
	193	else if (cmd->out)
	194	close(cmd->out);
	195	if (need_err)
	196	close_pair(fderr);
	197	return -ERR_RUN_COMMAND_FORK;
	198	}
4919bf03 SP	199
	200	if (need_in)
	201	close(fdin[0]);
	202	else if (cmd->in)
	203	close(cmd->in);
	204
f4bba25b SP	205	if (need_out)
f4bba25b SP	206	close(fdout[1]);
c20181e3	207	else if (cmd->out)
f4bba25b SP	208	close(cmd->out);
f4bba25b SP	209
f3b33f1d JS	210	if (need_err)
	211	close(fderr[1]);
	212
ebcb5d16 SP	213	return 0;
	214	}
	215
2d22c208	216	static int wait_or_whine(pid_t pid)
ebcb5d16	217	{
b1bf95bb JW	218	for (;;) {
b1bf95bb JW	219	int status, code;
2d22c208	220	pid_t waiting = waitpid(pid, &status, 0);
b1bf95bb	221
6f002f98	222	if (waiting < 0) {
b1bf95bb JW	223	if (errno == EINTR)
b1bf95bb JW	224	continue;
6f002f98	225	error("waitpid failed (%s)", strerror(errno));
b1bf95bb JW	226	return -ERR_RUN_COMMAND_WAITPID;
b1bf95bb JW	227	}
2d22c208	228	if (waiting != pid)
b1bf95bb JW	229	return -ERR_RUN_COMMAND_WAITPID_WRONG_PID;
	230	if (WIFSIGNALED(status))
	231	return -ERR_RUN_COMMAND_WAITPID_SIGNAL;
	232
	233	if (!WIFEXITED(status))
	234	return -ERR_RUN_COMMAND_WAITPID_NOEXIT;
	235	code = WEXITSTATUS(status);
	236	if (code)
	237	return -code;
	238	return 0;
	239	}
	240	}
f1000898	241
2d22c208 JS	242	int finish_command(struct child_process *cmd)
2d22c208 JS	243	{
2d22c208 JS	244	return wait_or_whine(cmd->pid);
	245	}
	246
ebcb5d16 SP	247	int run_command(struct child_process *cmd)
	248	{
	249	int code = start_command(cmd);
	250	if (code)
	251	return code;
	252	return finish_command(cmd);
	253	}
	254
1568fea0	255	static void prepare_run_command_v_opt(struct child_process *cmd,
ee493148 AR	256	const char **argv,
ee493148 AR	257	int opt)
1568fea0 AR	258	{
	259	memset(cmd, 0, sizeof(*cmd));
	260	cmd->argv = argv;
	261	cmd->no_stdin = opt & RUN_COMMAND_NO_STDIN ? 1 : 0;
	262	cmd->git_cmd = opt & RUN_GIT_CMD ? 1 : 0;
	263	cmd->stdout_to_stderr = opt & RUN_COMMAND_STDOUT_TO_STDERR ? 1 : 0;
	264	}
	265
f1000898 SP	266	int run_command_v_opt(const char **argv, int opt)
	267	{
	268	struct child_process cmd;
1568fea0 AR	269	prepare_run_command_v_opt(&cmd, argv, opt);
	270	return run_command(&cmd);
	271	}
	272
	273	int run_command_v_opt_cd(const char *argv, int opt, const char dir)
	274	{
	275	struct child_process cmd;
	276	prepare_run_command_v_opt(&cmd, argv, opt);
	277	cmd.dir = dir;
f1000898 SP	278	return run_command(&cmd);
f1000898 SP	279	}
ee493148 AR	280
	281	int run_command_v_opt_cd_env(const char *argv, int opt, const char dir, const char const env)
	282	{
	283	struct child_process cmd;
	284	prepare_run_command_v_opt(&cmd, argv, opt);
	285	cmd.dir = dir;
	286	cmd.env = env;
	287	return run_command(&cmd);
	288	}
2d22c208	289
618ebe9f JS	290	#ifdef __MINGW32__
	291	static __stdcall unsigned run_thread(void *data)
	292	{
	293	struct async *async = data;
	294	return async->proc(async->fd_for_proc, async->data);
	295	}
	296	#endif
	297
2d22c208 JS	298	int start_async(struct async *async)
	299	{
	300	int pipe_out[2];
	301
	302	if (pipe(pipe_out) < 0)
	303	return error("cannot create pipe: %s", strerror(errno));
618ebe9f	304	async->out = pipe_out[0];
2d22c208	305
618ebe9f	306	#ifndef __MINGW32__
2d22c208 JS	307	async->pid = fork();
	308	if (async->pid < 0) {
	309	error("fork (async) failed: %s", strerror(errno));
	310	close_pair(pipe_out);
	311	return -1;
	312	}
	313	if (!async->pid) {
	314	close(pipe_out[0]);
	315	exit(!!async->proc(pipe_out[1], async->data));
	316	}
2d22c208	317	close(pipe_out[1]);
618ebe9f JS	318	#else
	319	async->fd_for_proc = pipe_out[1];
	320	async->tid = (HANDLE) _beginthreadex(NULL, 0, run_thread, async, 0, NULL);
	321	if (!async->tid) {
	322	error("cannot create thread: %s", strerror(errno));
	323	close_pair(pipe_out);
	324	return -1;
	325	}
	326	#endif
2d22c208 JS	327	return 0;
	328	}
	329
	330	int finish_async(struct async *async)
	331	{
618ebe9f	332	#ifndef __MINGW32__
2d22c208 JS	333	int ret = 0;
	334
	335	if (wait_or_whine(async->pid))
	336	ret = error("waitpid (async) failed");
618ebe9f JS	337	#else
	338	DWORD ret = 0;
	339	if (WaitForSingleObject(async->tid, INFINITE) != WAIT_OBJECT_0)
	340	ret = error("waiting for thread failed: %lu", GetLastError());
	341	else if (!GetExitCodeThread(async->tid, &ret))
	342	ret = error("cannot get thread exit code: %lu", GetLastError());
	343	CloseHandle(async->tid);
	344	#endif
2d22c208 JS	345	return ret;
2d22c208 JS	346	}