int main( int argc, const char * argv[] ) { #pragma unused(argc) #pragma unused(argv) int my_tests_count, i; int err; int my_failures = 0; int list_the_tests = 0; const char * my_targetp; time_t my_start_time, my_end_time; struct stat my_stat_buf; char my_buffer[64]; /* vars for XILog */ #if !TARGET_OS_EMBEDDED XILogRef logRef; char *logPath = ""; char *config = NULL; int echo = 0; int xml = 0; #endif sranddev( ); /* set up seed for our random name generator */ g_cmd_namep = argv[0]; /* NOTE - code in create_target_directory will append '/' if it is necessary */ my_targetp = getenv("TMPDIR"); if ( my_targetp == NULL ) my_targetp = "/tmp"; /* make sure our executable is owned by root and has set uid bit */ err = stat( g_cmd_namep, &my_stat_buf ); if ( err != 0 ) { err = errno; printf( "stat call on our executable failed - \"%s\" \n", g_cmd_namep ); printf( " failed with error %d - \"%s\" \n", err, strerror( err) ); exit( -1 ); } if ( my_stat_buf.st_uid != 0 || (my_stat_buf.st_mode & S_ISUID) == 0 ) { printf( "executable file - \"%s\" \n", g_cmd_namep ); printf( "does not have correct owner (must be root) or setuid bit is not set \n" ); exit( -1 ); } /* parse input parameters */ for ( i = 1; i < argc; i++ ) { if ( strcmp( argv[i], "-u" ) == 0 ) { usage( ); } if ( strcmp( argv[i], "-t" ) == 0 || strcmp( argv[i], "-target" ) == 0 ) { if ( ++i >= argc ) { printf( "invalid target parameter \n" ); usage( ); } /* verify our target directory exists */ my_targetp = argv[i]; err = stat( my_targetp, &my_stat_buf ); if ( err != 0 || S_ISDIR(my_stat_buf.st_mode) == 0 ) { printf( "invalid target path \n" ); if ( err != 0 ) { printf( "stat call failed with error %d - \"%s\" \n", errno, strerror( errno) ); } usage( ); } continue; } if ( strcmp( argv[i], "-f" ) == 0 || strcmp( argv[i], "-failures" ) == 0 ) { if ( ++i >= argc ) { printf( "invalid failures parameter \n" ); usage( ); } /* get our max number of failures */ g_max_failures = strtol( argv[i], NULL, 10 ); continue; } if ( strcmp( argv[i], "-l" ) == 0 || strcmp( argv[i], "-list" ) == 0 ) { /* list all the tests this tool will do. */ list_the_tests = 1; continue; } if ( strcmp( argv[i], "-r" ) == 0 || strcmp( argv[i], "-run" ) == 0 ) { if ( ++i >= argc ) { printf( "invalid run tests parameter \n" ); usage( ); } /* get which tests to run */ if ( parse_tests_to_run( argc, argv, &i ) != 0 ) { printf( "invalid run tests parameter \n" ); usage( ); } continue; } if ( strcmp( argv[i], "-s" ) == 0 || strcmp( argv[i], "-skip" ) == 0 ) { /* set that want to skip the setuid related tests - this is useful for debgugging since since I can't * get setuid tests to work while in gdb. */ g_skip_setuid_tests = 1; continue; } #if !TARGET_OS_EMBEDDED if ( strcmp( argv[i], "-x" ) == 0 || strcmp( argv[i], "-xilog" ) == 0 ) { g_xilog_active = 1; continue; } #endif printf( "invalid argument \"%s\" \n", argv[i] ); usage( ); } /* done parsing. */ /* Check if we are running under testbots */ #if RUN_UNDER_TESTBOTS g_testbots_active = 1; #endif /* Code added to run xnu_quick_test under testbots */ if ( g_testbots_active == 1 ) { printf("[TEST] xnu_quick_test \n"); /* Declare the beginning of test suite */ } /* Populate groups list if we're in single user mode */ if (setgroups_if_single_user()) { return 1; } if ( list_the_tests != 0 ) { list_all_tests( ); return 0; } #if !TARGET_OS_EMBEDDED if (g_xilog_active == 1) { logRef = XILogOpenLogExtended( logPath, "xnu_quick_test", "com.apple.coreos", config, xml, echo, NULL, "ResultOwner", "com.apple.coreos", NULL ); if( logRef == NULL ) { fprintf(stderr,"Couldn't create log: %s",logPath); exit(-1); } } #endif /* build a test target directory that we use as our path to create any test * files and directories. */ create_target_directory( my_targetp ); printf( "Will allow %ld failures before testing is aborted \n", g_max_failures ); my_start_time = time( NULL ); printf( "\nBegin testing - %s \n", ctime_r( &my_start_time, &my_buffer[0] ) ); printf( "Current architecture is %s\n", current_arch() ); /* Code added to run xnu_quick_test under testbots */ if ( g_testbots_active == 1 ) { printf("[PASS] xnu_quick_test started\n"); } /* run each test that is marked to run in our table until we complete all of them or * hit the maximum number of failures. */ my_tests_count = (sizeof( g_tests ) / sizeof( g_tests[0] )); for ( i = 0; i < (my_tests_count - 1); i++ ) { int my_err; test_entryp my_testp; my_testp = &g_tests[i]; if ( my_testp->test_run_it == 0 || my_testp->test_routine == NULL ) continue; #if !TARGET_OS_EMBEDDED if (g_xilog_active == 1) { XILogBeginTestCase( logRef, my_testp->test_infop, my_testp->test_infop ); XILogMsg( "test #%d - %s \n", (i + 1), my_testp->test_infop ); } #endif printf( "test #%d - %s \n", (i + 1), my_testp->test_infop ); fflush(stdout); my_err = my_testp->test_routine( my_testp->test_input ); if ( my_err != 0 ) { printf("\t--> FAILED \n"); #if !TARGET_OS_EMBEDDED if (g_xilog_active == 1) { XILogMsg("SysCall %s failed", my_testp->test_infop); XILogErr("Result %d", my_err); } #endif my_failures++; if ( my_failures > g_max_failures ) { #if !TARGET_OS_EMBEDDED if (g_xilog_active == 1) { XILogMsg("Reached the maximum number of failures - Aborting xnu_quick_test."); XILogEndTestCase( logRef, kXILogTestPassOnErrorLevel ); } #endif printf( "\n Reached the maximum number of failures - Aborting xnu_quick_test. \n" ); /* Code added to run xnu_quick_test under testbots */ if ( g_testbots_active == 1 ) { printf("[FAIL] %s \n", my_testp->test_infop); } goto exit_this_routine; } /* Code added to run xnu_quick_test under testbots */ if ( g_testbots_active == 1 ) { printf("[FAIL] %s \n", my_testp->test_infop); } #if !TARGET_OS_EMBEDDED if (g_xilog_active == 1) { XILogEndTestCase( logRef, kXILogTestPassOnErrorLevel ); } #endif continue; } #if !TARGET_OS_EMBEDDED if (g_xilog_active == 1) { XILogEndTestCase(logRef, kXILogTestPassOnErrorLevel); } #endif /* Code added to run xnu_quick_test under testbots */ if ( g_testbots_active == 1 ) { printf("[PASS] %s \n", my_testp->test_infop); } } exit_this_routine: my_end_time = time( NULL ); printf( "\nEnd testing - %s \n", ctime_r( &my_end_time, &my_buffer[0] ) ); /* clean up our test directory */ rmdir( &g_target_path[0] ); #if !TARGET_OS_EMBEDDED if (g_xilog_active == 1) { XILogCloseLog(logRef); } #endif return 0; } /* main */
int main( int argc, const char * argv[] ) { #pragma unused(argc) #pragma unused(argv) int my_tests_count, i; int err; int my_failures = 0; int list_the_tests = 0; const char * my_targetp; time_t my_start_time, my_end_time; struct stat my_stat_buf; char my_buffer[64]; uid_t sudo_uid = 0; const char * sudo_uid_env; gid_t sudo_gid; const char * sudo_gid_env; sranddev( ); /* set up seed for our random name generator */ g_cmd_namep = argv[0]; /* make sure SIGCHLD is not ignored, so wait4 calls work */ signal(SIGCHLD, SIG_DFL); /* NOTE - code in create_target_directory will append '/' if it is necessary */ my_targetp = getenv("TMPDIR"); if ( my_targetp == NULL ) my_targetp = "/tmp"; /* make sure we are running as root */ if ( ( getuid() != 0 ) || ( geteuid() != 0 ) ) { printf( "Test must be run as root\n", g_cmd_namep ); exit( -1 ); } sudo_uid_env = getenv("SUDO_UID"); if ( sudo_uid_env ) { sudo_uid = strtol(sudo_uid_env, NULL, 10); } /* switch real uid to a non_root user, while keeping effective uid as root */ if ( sudo_uid != 0 ) { setreuid( sudo_uid, 0 ); } else { /* Default to 501 if no sudo uid found */ setreuid( 501, 0 ); } /* restore the gid if run through sudo */ sudo_gid_env = getenv("SUDO_GID"); if ( sudo_gid_env ) { sudo_gid = strtol(sudo_gid_env, NULL, 10); } if ( getgid() == 0 ) { if ( sudo_gid != 0 ) { setgid( sudo_gid ); } else { /* Default to 20 if no sudo gid found */ setgid( 20 ); } } /* parse input parameters */ for ( i = 1; i < argc; i++ ) { if ( strcmp( argv[i], "-u" ) == 0 ) { usage( ); } if ( strcmp( argv[i], "-t" ) == 0 || strcmp( argv[i], "-target" ) == 0 ) { if ( ++i >= argc ) { printf( "invalid target parameter \n" ); usage( ); } /* verify our target directory exists */ my_targetp = argv[i]; err = stat( my_targetp, &my_stat_buf ); if ( err != 0 || S_ISDIR(my_stat_buf.st_mode) == 0 ) { printf( "invalid target path \n" ); if ( err != 0 ) { printf( "stat call failed with error %d - \"%s\" \n", errno, strerror( errno) ); } usage( ); } continue; } if ( strcmp( argv[i], "-f" ) == 0 || strcmp( argv[i], "-failures" ) == 0 ) { if ( ++i >= argc ) { printf( "invalid failures parameter \n" ); usage( ); } /* get our max number of failures */ g_max_failures = strtol( argv[i], NULL, 10 ); continue; } if ( strcmp( argv[i], "-l" ) == 0 || strcmp( argv[i], "-list" ) == 0 ) { /* list all the tests this tool will do. */ list_the_tests = 1; continue; } if ( strcmp( argv[i], "-r" ) == 0 || strcmp( argv[i], "-run" ) == 0 ) { if ( ++i >= argc ) { printf( "invalid run tests parameter \n" ); usage( ); } /* get which tests to run */ if ( parse_tests_to_run( argc, argv, &i ) != 0 ) { printf( "invalid run tests parameter \n" ); usage( ); } continue; } if ( strcmp( argv[i], "-s" ) == 0 || strcmp( argv[i], "-skip" ) == 0 ) { /* set that want to skip the setuid related tests - this is useful for debgugging since since I can't * get setuid tests to work while in gdb. */ g_skip_setuid_tests = 1; continue; } if ( strcmp( argv[i], "-testbot" ) == 0 ) { g_testbots_active = 1; continue; } printf( "invalid argument \"%s\" \n", argv[i] ); usage( ); } /* done parsing. */ /* Check if we are running under testbots */ #if RUN_UNDER_TESTBOTS g_testbots_active = 1; #endif /* Code added to run xnu_quick_test under testbots */ if ( g_testbots_active == 1 ) { printf("[TEST] xnu_quick_test \n"); /* Declare the beginning of test suite */ } /* Populate groups list if we're in single user mode */ if (setgroups_if_single_user()) { return 1; } if ( list_the_tests != 0 ) { list_all_tests( ); return 0; } /* build a test target directory that we use as our path to create any test * files and directories. */ create_target_directory( my_targetp ); printf( "Will allow %ld failures before testing is aborted \n", g_max_failures ); my_start_time = time( NULL ); printf( "\nBegin testing - %s \n", ctime_r( &my_start_time, &my_buffer[0] ) ); printf( "Current architecture is %s\n", current_arch() ); /* Code added to run xnu_quick_test under testbots */ /* run each test that is marked to run in our table until we complete all of them or * hit the maximum number of failures. */ my_tests_count = (sizeof( g_tests ) / sizeof( g_tests[0] )); for ( i = 0; i < (my_tests_count - 1); i++ ) { int my_err; test_entryp my_testp; my_testp = &g_tests[i]; if ( my_testp->test_run_it == 0 || my_testp->test_routine == NULL ) continue; if ( g_testbots_active == 1 ) { printf("[BEGIN] %s \n", my_testp->test_infop); } printf( "test #%d - %s \n", (i + 1), my_testp->test_infop ); fflush(stdout); my_err = my_testp->test_routine( my_testp->test_input ); if ( my_err != 0 ) { printf("\t--> FAILED \n"); printf("SysCall %s failed", my_testp->test_infop); printf("Result %d", my_err); my_failures++; if ( my_failures > g_max_failures ) { printf( "\n Reached the maximum number of failures - Aborting xnu_quick_test. \n" ); /* Code added to run xnu_quick_test under testbots */ if ( g_testbots_active == 1 ) { printf("[FAIL] %s \n", my_testp->test_infop); } goto exit_this_routine; } /* Code added to run xnu_quick_test under testbots */ if ( g_testbots_active == 1 ) { printf("\n[FAIL] %s \n", my_testp->test_infop); } continue; } /* Code added to run xnu_quick_test under testbots */ if ( g_testbots_active == 1 ) { printf("[PASS] %s \n", my_testp->test_infop); } } exit_this_routine: my_end_time = time( NULL ); printf( "\nEnd testing - %s \n", ctime_r( &my_end_time, &my_buffer[0] ) ); /* clean up our test directory */ rmdir( &g_target_path[0] ); /* exit non zero if there are any failures */ return my_failures != 0; } /* main */