#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"
void error__(const char *n, int l, const char *s, ...) {
va_list ap;
+ if (failed == 0 && nfailed == 0)
+ dprintf(1, "FAIL\n");
failed = 1;
- fprintf(stderr, " ERROR %s %s:%d: ", name, n, l);
+ dprintf(1, " ERROR %s %s:%d: ", name, n, l);
va_start(ap, s);
- vfprintf(stderr, s, ap);
+ 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;
failed = 0;
name = n;
if (verbose)
- fprintf(stderr, "running %s:\n", name);
+ dprintf(1, "running %s:\n", name);
pid = fork();
if (pid == 0) {
/* run test in a child process */
+ setrl(RLIMIT_STACK, 1<<16);
+ setrl(RLIMIT_CPU, 2);
f();
exit(failed);
}
if (failed) {
nfailed++;
- fprintf(stderr, "FAILED %s\n", name);
+ dprintf(1, "FAILED %s\n", name);
} else if (verbose)
- fprintf(stderr, "PASSED %s\n", name);
+ dprintf(1, "PASSED %s\n", name);
}
static int summary() {
if (nfailed)
- fprintf(stderr, "FAIL:%d (all: %d)\n", nfailed, count);
+ dprintf(1, "FAIL (%d out of %d tests)\n", nfailed, count);
else
- fprintf(stderr, "PASS (all: %d)\n", count);
+ dprintf(1, "ok (%d tests)\n", count);
return !!nfailed;
}
static void usage() {
- fprintf(stderr, "usage: ./t [-vs]\n");
+ dprintf(1, "usage: ./t [-vs]\n");
exit(1);
}