ede/ede-launch/ede-launch.cpp

#ifdef HAVE_CONFIG_H
#include <config.h>
#endif

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/types.h>
#include <sys/wait.h>
#include <sys/resource.h>
#include <unistd.h>
#include <signal.h>
#include <errno.h>

#include <edelib/Run.h>

EDELIB_NS_USING(run_program_fmt)

void help(void) {
	puts("Usage: ede-launch [OPTIONS] program");
	puts("EDE program launcher");
}

static char** cmd_split(const char* cmd) {
	int   sz = 10;
	int   i = 0;
	char* c = strdup(cmd);

	char** arr = (char**)malloc(sizeof(char*) * sz);

	for(char* p = strtok(c, " "); p; p = strtok(NULL, " ")) {
		if(i >= sz) {
			sz *= 2;
			arr = (char**)realloc(arr, sizeof(char*) * sz);
		}
		arr[i++] = strdup(p);
	}

	arr[i] = NULL;
	free(c);

	return arr;
}

static void start_crasher(const char* cmd, int sig) {
	/* 
	 * call edelib implementation instead start_child_process()
	 * to prevents loops if 'ede-crasher' crashes
	 */
	run_program_fmt(true, "ede-crasher --apppath %s --signal %i", cmd, sig);
}

static int start_child_process(const char* cmd) {
	int pid, in[2], out[2], err[2];
	char** params = cmd_split(cmd);

	pipe(in);
	pipe(out);
	pipe(err);

	signal(SIGCHLD, SIG_DFL);

	pid = fork();
	switch(pid) {
		case 0:
			/* child process */
			close(0);
			dup(in[0]);
			close(in[0]);
			close(in[1]);

			close(1);
			dup(out[1]);
			close(out[0]);
			close(out[1]);

			close(2);
			dup(err[1]);
			close(err[0]);
			close(err[1]);

			errno = 0;

			/* start it */
			execvp(params[0], params);

			if(errno == 2)
				_exit(199);
			else
				_exit(errno);
			break;
		case -1:
			fprintf(stderr, "ede-launcher: fork() failed\n");
			/* close the pipes */
			close(in[0]);
			close(in[1]);

			close(out[0]);
			close(out[1]);

			close(err[0]);
			close(err[1]);
			break;
		default:
			/* perent */
			close(in[0]);
			close(out[1]);
			close(err[1]);
			break;
	}

	int status;
	if(waitpid(pid, &status, 0) < 0) {
		fprintf(stderr, "ede-launcher: waitpid() failed\n");
		return 1;
	}

	if(WIFEXITED(status)) {
		int v = WEXITSTATUS(status);
		fprintf(stderr, "ede-launcher: child exited with '%i'\n", v);
	} else if(WIFSIGNALED(status) && WTERMSIG(status) == SIGSEGV) {
		start_crasher(cmd, SIGSEGV);
	} else {
		fprintf(stderr, "ede-launcher: child '%s' killed!\n", cmd);
	}

	return 0;
}

static int start_child_process_with_core(const char* cmd) {
	struct rlimit r;
	if(getrlimit(RLIMIT_CORE, &r) == -1)
		return -1; /* TODO: core dump error message */

	rlim_t old = r.rlim_cur;
	r.rlim_cur = RLIM_INFINITY;

	if(setrlimit(RLIMIT_CORE, &r) == -1)
		return -1; /* TODO: core dump error message */

	int ret = start_child_process(cmd);

	r.rlim_cur = old;
	if(setrlimit(RLIMIT_CORE, &r) == -1)
		return -1; /* TODO: core dump error message */

	return ret;
}

int main(int argc, char** argv) {
	if(argc != 2) {
		help();
		return 0;
	}

	return start_child_process_with_core(argv[1]);
}