/*
 * lat_proc.c - process creation tests
 *
 * Usage: lat_proc
 *
 * TODO - linux clone, plan9 rfork, IRIX sproc().
 *
 * Copyright (c) 1994 Larry McVoy.  Distributed under the FSF GPL with
 * additional restriction that results may published only if
 * (1) the benchmark is unmodified, and
 * (2) the version in the sccsid below is included in the report.
 * Support for this development by Sun Microsystems is gratefully acknowledged.
 */
char	*id = "$Id$\n";

#include "bench.h"

#ifdef STATIC
#define	PROG "/tmp/hello-s"
#else
#define	PROG "/tmp/hello"
#endif

void do_shell(void)
{
	int	pid;
	int	status;

	switch (pid = fork()) {
	    case -1:
		perror("fork");
		exit(1);

	    case 0:	/* child */
		close(1);
		execlp("/bin/sh", "sh", "-c", PROG, 0);
		exit(1);

	    default:
		while (wait(&status) != pid)
			;
		if (WEXITSTATUS(status))
			exit(WEXITSTATUS(status));
	}
}

void do_vforkexec(void)
{
	int	pid;
	char	*nav[2];
	int	status;

	nav[0] = PROG;
	nav[1] = 0;
	switch (pid = vfork()) {
	    case -1:
		perror("vfork");
		exit(1);

	    case 0: 	/* child */
		close(1);
		execve(PROG, nav, 0);
		exit(1);

	    default:
		while (wait(&status) != pid)
			;
		if (WEXITSTATUS(status))
			exit(WEXITSTATUS(status));
	}
}

void do_forkexec(void)
{
	int	pid;
	char	*nav[2];
	int	status;

	nav[0] = PROG;
	nav[1] = 0;
	switch (pid = fork()) {
	    case -1:
		perror("fork");
		exit(1);

	    case 0: 	/* child */
		close(1);
		execve(PROG, nav, 0);
		exit(1);

	    default:
		while (wait(&status) != pid)
			;
		if (WEXITSTATUS(status))
			exit(WEXITSTATUS(status));
	}
}
	
void do_fork(void)
{
	int	pid;
	int	status;

	switch (pid = fork()) {
	    case -1:
		perror("fork");
		exit(1);

	    case 0:	/* child */
		exit(0);

	    default:
		while (wait(&status) != pid)
			;
		if (WEXITSTATUS(status))
			exit(WEXITSTATUS(status));
	}
}

void do_dfork(void)
{
	int	pid;
	int	status;

	switch (pid = fork()) {
	    case -1:
		perror("fork");
		exit(1);

	    case 0:	/* child */
		do_fork();
		exit(0);

	    default:
		while (wait(&status) != pid)
			;
		if (WEXITSTATUS(status))
			exit(WEXITSTATUS(status));
	}
}
	
void do_dforkexec(void)
{
	int	pid;
	int	status;

	switch (pid = fork()) {
	    case -1:
		perror("fork");
		exit(1);

	    case 0:	/* child */
		do_forkexec();
		exit(0);

	    default:
		while (wait(&status) != pid)
			;
		if (WEXITSTATUS(status))
			exit(WEXITSTATUS(status));
	}
}
	
void do_procedure(int r)
{
	use_int(r);
}
	
int
main(int ac, char **av)
{
	if (ac < 2) goto usage;

	if (!strcmp("procedure", av[1])) {
		BENCH(do_procedure(ac), 0);
		micro("Procedure call", get_n());
	} else if (!strcmp("fork", av[1])) {
		BENCH(do_fork(), 0);
#ifdef STATIC
		micro("Static Process fork+exit", get_n());
#else
		micro("Process fork+exit", get_n());
#endif
	} else if (!strcmp("fork-size", av[1])) {
		size_t size = bytes(av[2]);
		memset(malloc(size), 0, size);
		BENCH(do_fork(), 0);
#ifdef STATIC
		micro("Static Process fork+exit", get_n());
#else
		micro("Process fork+exit", get_n());
#endif
	} else if (!strcmp("dfork", av[1])) {
		BENCH(do_dfork(), 0);
#ifdef STATIC
		micro("Static Process double fork+exit", get_n());
#else
		micro("Process double fork+exit", get_n());
#endif
	} else if (!strcmp("vfork", av[1])) {
		BENCH(do_vforkexec(), 0);
#ifdef STATIC
		micro("Static Process vfork+execve", get_n());
#else
		micro("Process vfork+execve", get_n());
#endif
	} else if (!strcmp("exec", av[1])) {
		BENCH(do_forkexec(), 0);
#ifdef STATIC
		micro("Static Process fork+execve", get_n());
#else
		micro("Process fork+execve", get_n());
#endif
	} else if (!strcmp("dforkexec", av[1])) {
		BENCH(do_dforkexec(), 0);
#ifdef STATIC
		micro("Static Process double fork+execve", get_n());
#else
		micro("Process double fork+execve", get_n());
#endif
	} else if (!strcmp("shell", av[1])) {
		BENCH(do_shell(), 0);
#ifdef STATIC
		micro("Static Process fork+/bin/sh -c", get_n());
#else
		micro("Process fork+/bin/sh -c", get_n());
#endif
	} else {
usage:		printf("Usage: %s fork|vfork|exec|shell|dfork|dforkexec\n", av[0]);
	}
	return(0);
}