#include <fcntl.h>
#include <stdarg.h>
#include <stdlib.h>
#include <string.h>
#include <sys/stat.h>
#include <sys/types.h>
#include <unistd.h>
#include <stdbool.h>
#include <time.h>
#include "testmore.h"
#include "testenv.h"
#include <utilities/debugging.h>
#include <utilities/SecCFRelease.h>
#include <utilities/SecFileLocations.h>
int test_verbose = 0;
#ifdef NO_SERVER
#include <securityd/spi.h>
static int current_dir = -1;
static char scratch_dir[50];
static char *home_var;
static bool keep_scratch_dir = false;
static int
rmdir_recursive(const char *path)
{
char command_buf[256];
if (strlen(path) + 10 > sizeof(command_buf) || strchr(path, '\''))
{
fprintf(stderr, "# rmdir_recursive: invalid path: %s", path);
return -1;
}
sprintf(command_buf, "/bin/rm -rf '%s'", path);
return system(command_buf);
}
#endif
static int tests_init(void) {
int ok = 0;
#ifdef NO_SERVER
char preferences_dir[80];
char library_dir[70];
setup("tests_init");
sprintf(scratch_dir, "/tmp/tst-%d", getpid());
if (keep_scratch_dir) {
printf("running tests with HOME=%s\n", scratch_dir);
}
sprintf(library_dir, "%s/Library", scratch_dir);
sprintf(preferences_dir, "%s/Preferences", library_dir);
ok = (ok_unix(mkdir(scratch_dir, 0755), "mkdir") &&
ok_unix(current_dir = open(".", O_RDONLY), "open") &&
ok_unix(chdir(scratch_dir), "chdir") &&
ok_unix(setenv("HOME", scratch_dir, 1), "setenv") &&
ok_unix(mkdir(library_dir, 0755), "mkdir") &&
ok_unix(mkdir(preferences_dir, 0755), "mkdir") &&
ok(home_var = getenv("HOME"), "getenv"));
if (ok > 0)
securityd_init(scratch_dir);
#endif
return ok;
}
static int
tests_end(void)
{
#ifdef NO_SERVER
setup("tests_end");
int ok = ok_unix(fchdir(current_dir), "fchdir");
if (ok)
ok = ok_unix(close(current_dir), "close");
if (ok) {
if (!keep_scratch_dir) {
ok = ok_unix(rmdir_recursive(scratch_dir), "rmdir_recursive");
}
}
return ok;
#else
return 0;
#endif
}
static void usage(const char *progname)
{
fprintf(stderr, "usage: %s [-k][-w][testname [testargs] ...]\n", progname);
exit(1);
}
static int tests_run_index(int i, int argc, char * const *argv)
{
int ch;
while ((ch = getopt(argc, argv, "v")) != -1)
{
switch (ch)
{
case 'v':
test_verbose++;
break;
default:
usage(argv[0]);
}
}
fprintf(stderr, "TEST: Test Case '%s' started.\n", testlist[i].name);
run_one_test(&testlist[i], argc, argv);
if(testlist[i].failed_tests) {
fprintf(stderr, "FAIL: Test Case '%s' failed.\n", testlist[i].name);
} else {
fprintf(stderr, "PASS: Test Case '%s' passed. (%lu ms)\n", testlist[i].name, testlist[i].duration);
}
return testlist[i].failed_tests;
}
static int strcmp_under_is_dash(const char *s, const char *t) {
for (;;) {
char a = *s++, b = *t++;
if (a != b) {
if (a != '_' || b != '-')
return a - b;
} else if (a == 0) {
return 0;
}
}
}
static int tests_named_index(const char *testcase)
{
int i;
for (i = 0; testlist[i].name; ++i) {
if (strcmp_under_is_dash(testlist[i].name, testcase) == 0) {
return i;
}
}
return -1;
}
static int tests_run_all(int argc, char * const *argv)
{
int curroptind = optind;
int i;
int failcount=0;
for (i = 0; testlist[i].name; ++i) {
if(!testlist[i].off) {
failcount+=tests_run_index(i, argc, argv);
optind = curroptind;
}
}
return failcount;
}
int
tests_begin(int argc, char * const *argv)
{
const char *testcase = NULL;
bool initialized = false;
bool print_security_logs = false;
int testix = -1;
int failcount = 0;
int ch;
int loop = 0;
for (;;) {
while (!testcase && (ch = getopt(argc, argv, "klvws")) != -1)
{
switch (ch)
{
#ifdef NO_SERVER
case 'k':
keep_scratch_dir = true;
break;
#endif
case 's':
print_security_logs = true;
break;
case 'v':
test_verbose++;
break;
case 'w':
sleep(100);
break;
case 'l':
loop=1;
break;
case '?':
default:
printf("invalid option %c\n",ch);
usage(argv[0]);
}
}
if (optind < argc) {
testix = tests_named_index(argv[optind]);
if(testix<0) {
printf("invalid test %s\n",argv[optind]);
usage(argv[0]);
}
}
if (print_security_logs) {
add_security_log_hanlder(^(const char *level, CFStringRef scope, const char *function, const char *file, int line, CFStringRef message) {
time_t now = time(NULL);
char *date = ctime(&now);
date[19] = '\0';
CFStringRef logStr = CFStringCreateWithFormat(kCFAllocatorDefault, NULL,
CFSTR("%s %@ %s %@\n"), date + 4,
scope ? scope : CFSTR(""), function, message);
CFShow(logStr);
CFReleaseSafe(logStr);
});
}
if (testix < 0) {
if (!initialized) {
initialized = true;
tests_init();
failcount+=tests_run_all(argc, argv);
}
break;
} else {
if (!initialized) {
tests_init();
initialized = true;
}
optind++;
failcount+=tests_run_index(testix, argc, argv);
testix = -1;
}
}
printf("Total failcount = %d\n", failcount);
tests_end();
if(loop) {
printf("Looping until key press 'q'. You can run leaks now.\n");
while(getchar()!='q');
}
return failcount;
}