--- /dev/null
+/*
+ * Copyright (c) 2005-2007,2009-2014 Apple Inc. All Rights Reserved.
+ *
+ * @APPLE_LICENSE_HEADER_START@
+ *
+ * This file contains Original Code and/or Modifications of Original Code
+ * as defined in and that are subject to the Apple Public Source License
+ * Version 2.0 (the 'License'). You may not use this file except in
+ * compliance with the License. Please obtain a copy of the License at
+ * http://www.opensource.apple.com/apsl/ and read it before using this
+ * file.
+ *
+ * The Original Code and all software distributed under the License are
+ * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
+ * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
+ * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY,
+ * FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT.
+ * Please see the License for the specific language governing rights and
+ * limitations under the License.
+ *
+ * @APPLE_LICENSE_HEADER_END@
+ *
+ * testenv.c
+ */
+
+#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_strict_bats = 1;
+int test_verbose = 0;
+int test_onebatstest = 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(stdout, "# 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");
+
+ /* Create scratch dir for tests to run in. */
+ 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") &&
+ /* @@@ Work around a bug that the prefs code in
+ libsecurity_keychain never creates the Library/Preferences
+ dir. */
+ 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");
+ /* Restore previous cwd and remove scratch dir. */
+ 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]);
+ }
+ }
+
+ if (test_onebatstest)
+ fprintf(stdout, "[TEST] %s\n", testlist[i].name);
+ else
+ fprintf(stdout, "[BEGIN] %s\n", testlist[i].name);
+ run_one_test(&testlist[i], argc, argv);
+ if(testlist[i].failed_tests) {
+ fprintf(stdout, "[FAIL] %s\n", testlist[i].name);
+ } else {
+ fprintf(stdout, "[PASS] %s\n", testlist[i].name);
+ if (test_verbose)
+ fprintf(stdout, "(%lu ms)\n", 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;
+ int list = 0;
+
+ // TODO Currently our callers do this, but we can move this here together with the build date info.
+ const char *progname = strrchr(argv[0], '/');
+ progname = progname ? progname + 1 : argv[0];
+
+ for (;;) {
+ while (!testcase && (ch = getopt(argc, argv, "bklL1vwqs")) != -1)
+ {
+ switch (ch)
+ {
+#ifdef NO_SERVER
+ case 'k':
+ keep_scratch_dir = true;
+ break;
+#endif
+ case 's':
+ print_security_logs = true;
+ break;
+
+ case 'b':
+ test_strict_bats = 0;
+ break;
+
+ case 'v':
+ test_verbose++;
+ break;
+
+ case 'w':
+ sleep(100);
+ break;
+ case 'l':
+ loop=1;
+ break;
+ case 'L':
+ list=1;
+ break;
+ case '1':
+ test_onebatstest=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_handler(^(int 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 (!list && !initialized && !test_onebatstest)
+ fprintf(stdout, "[TEST] %s\n", progname);
+ if (testix < 0) {
+ if (!initialized) {
+ tests_init();
+ initialized = true;
+ (void)initialized;
+ if (!list)
+ failcount+=tests_run_all(argc, argv);
+ }
+ break;
+ } else {
+ if (!initialized) {
+ tests_init();
+ initialized = true;
+ for (int i = 0; testlist[i].name; ++i) {
+ testlist[i].off = 1;
+ }
+ }
+ optind++;
+ testlist[testix].off = 0;
+ if (!list)
+ failcount+=tests_run_index(testix, argc, argv);
+ testix = -1;
+ }
+ }
+ if (!list) {
+ if (!test_onebatstest) {
+ fprintf(stdout, "[%s] %s\n", failcount ? "FAIL" : "PASS", progname);
+ }
+ fprintf(stdout, "[SUMMARY]\n");
+ }
+ int sub_tests = 0;
+ if (failcount) {
+ fprintf(stdout, "Test name failed total\n");
+ fprintf(stdout, "================================================================\n");
+ }
+ for (int i = 0; testlist[i].name; ++i) {
+ if (!testlist[i].off) {
+ sub_tests += testlist[i].sub_tests;
+ if (testlist[i].failed_tests) {
+ fprintf(stdout, "%-50s %6d %6d\n", testlist[i].name, testlist[i].failed_tests, testlist[i].sub_tests);
+ } else if (list) {
+ fprintf(stdout, "%s\n", testlist[i].name);
+ }
+ }
+ }
+ if (failcount)
+ fprintf(stdout, "================================================================\n");
+ if (!list)
+ fprintf(stdout, "Total failcount: %6d %6d\n", failcount, sub_tests);
+ fflush(stdout);
+
+ /* Cleanups */
+ tests_end();
+
+ if (loop) {
+ printf("Looping until key press 'q'. You can run leaks now.\n");
+ while(getchar()!='q');
+ }
+
+ return failcount;
+}
+
projects / apple / security.git / blobdiff