#include <stdlib.h>
#include <stdio.h>
#include <stdarg.h>
#include <string.h>
#include <errno.h>
#include <sys/types.h>
#include <sys/wait.h>
#include <sys/time.h>
#include <sys/resource.h>
#include <unistd.h>
#include "test.h"

#define B(f)
#define T(f) void f();
#include "tests.h"
#undef T

static int failed;
static const char *name;

static int slow;
static int verbose;
static int count;
static int nfailed;

static void errtimer() { error("use *_timer in benchmarks only\n"); }
void start_timer() { errtimer(); }
void stop_timer() { errtimer(); }
void reset_timer() { errtimer(); }

void error__(const char *n, int l, const char *s, ...) {
	va_list ap;

	if (failed == 0 && nfailed == 0)
		dprintf(1, "FAIL\n");
	failed = 1;
	dprintf(1, " ERROR %s %s:%d: ", name, n, l);
	va_start(ap, s);
	vdprintf(1, s, ap);
	va_end(ap);
}

static void setrl(int r, long lim) {
	struct rlimit rl;

	if (getrlimit(r, &rl))
		error("getrlimit %d: %s\n", r, strerror(errno));
	rl.rlim_cur = lim;
	if (setrlimit(r, &rl))
		error("setrlimit %d: %s\n", r, strerror(errno));
}

static void run(const char *n, void (*f)()) {
	pid_t pid;
	int s;

	count++;
	failed = 0;
	name = n;
	if (verbose)
		dprintf(1, "running %s:\n", name);

	pid = fork();
	if (pid == 0) {
		/* run test in a child process */
		setrl(RLIMIT_CORE, 1<<24);
		setrl(RLIMIT_STACK, 1<<16);
		setrl(RLIMIT_CPU, 2);
		f();
		exit(failed);
	}

	if (pid == -1)
		error("fork failed: %s\n", strerror(errno));
	else {
		if (waitpid(pid, &s, 0) == -1)
			error("waitpid failed: %s\n", strerror(errno));
		else if (!WIFEXITED(s))
			error("abnormal exit: %s\n", WIFSIGNALED(s) ? strsignal(WTERMSIG(s)) : "(unknown)");
		else
			failed = !!WEXITSTATUS(s);
	}

	if (failed) {
		nfailed++;
		dprintf(1, "FAILED %s\n", name);
	} else if (verbose)
		dprintf(1, "PASSED %s\n", name);
}

static int summary() {
	if (nfailed)
		dprintf(1, "FAIL (%d out of %d tests)\n", nfailed, count);
	else
		dprintf(1, "ok (%d tests)\n", count);
	return !!nfailed;
}

static void usage() {
	dprintf(1, "usage: ./t [-vs]\n");
	exit(1);
}

int main(int argc, char *argv[]) {
	int c;

	while((c = getopt(argc, argv, "vs")) != -1)
		switch(c) {
		case 'v':
			verbose = 1;
			break;
		case 's':
			slow = 1; /* TODO */
			break;
		default:
			usage();
		}
	if (optind != argc)
		usage();

#define T(t) run(#t, t);
#include "tests.h"
	return summary();
}