1*d4514f0bSApple OSS Distributions #include <stdio.h> 2*d4514f0bSApple OSS Distributions #include <mach/mach_vm.h> 3*d4514f0bSApple OSS Distributions #include <mach/mach_port.h> 4*d4514f0bSApple OSS Distributions #include <mach/mach_error.h> 5*d4514f0bSApple OSS Distributions #include <sys/sysctl.h> 6*d4514f0bSApple OSS Distributions #include <sys/wait.h> 7*d4514f0bSApple OSS Distributions #include <unistd.h> 8*d4514f0bSApple OSS Distributions 9*d4514f0bSApple OSS Distributions #ifdef T_NAMESPACE 10*d4514f0bSApple OSS Distributions #undef T_NAMESPACE 11*d4514f0bSApple OSS Distributions #endif 12*d4514f0bSApple OSS Distributions 13*d4514f0bSApple OSS Distributions #include <darwintest.h> 14*d4514f0bSApple OSS Distributions #include <darwintest_utils.h> 15*d4514f0bSApple OSS Distributions 16*d4514f0bSApple OSS Distributions T_GLOBAL_META(T_META_RUN_CONCURRENTLY(true), 17*d4514f0bSApple OSS Distributions T_META_NAMESPACE("xnu.ipc"), 18*d4514f0bSApple OSS Distributions T_META_RADAR_COMPONENT_NAME("xnu"), 19*d4514f0bSApple OSS Distributions T_META_RADAR_COMPONENT_VERSION("IPC"), 20*d4514f0bSApple OSS Distributions T_META_TAG_VM_PREFERRED); 21*d4514f0bSApple OSS Distributions 22*d4514f0bSApple OSS Distributions T_DECL(test_task_filter_msg_flag, "Set the filter msg flag on the task and check if the forked child inherits it", 23*d4514f0bSApple OSS Distributions T_META_ASROOT(true), T_META_CHECK_LEAKS(false)) 24*d4514f0bSApple OSS Distributions { 25*d4514f0bSApple OSS Distributions int ret, dev; 26*d4514f0bSApple OSS Distributions size_t sysctl_size; 27*d4514f0bSApple OSS Distributions 28*d4514f0bSApple OSS Distributions T_SETUPBEGIN; 29*d4514f0bSApple OSS Distributions 30*d4514f0bSApple OSS Distributions dev = 0; 31*d4514f0bSApple OSS Distributions sysctl_size = sizeof(dev); 32*d4514f0bSApple OSS Distributions ret = sysctlbyname("kern.development", &dev, &sysctl_size, NULL, 0); 33*d4514f0bSApple OSS Distributions T_QUIET; T_ASSERT_POSIX_SUCCESS(ret, "sysctl kern.development failed"); 34*d4514f0bSApple OSS Distributions if (dev == 0) { 35*d4514f0bSApple OSS Distributions T_SKIP("Skipping test on release kernel"); 36*d4514f0bSApple OSS Distributions } 37*d4514f0bSApple OSS Distributions 38*d4514f0bSApple OSS Distributions T_SETUPEND; 39*d4514f0bSApple OSS Distributions 40*d4514f0bSApple OSS Distributions int cur_filter_flag = 0; 41*d4514f0bSApple OSS Distributions int new_filter_flag = 1; 42*d4514f0bSApple OSS Distributions ret = sysctlbyname("kern.task_set_filter_msg_flag", NULL, 0, &new_filter_flag, sizeof(new_filter_flag)); 43*d4514f0bSApple OSS Distributions T_ASSERT_POSIX_SUCCESS(ret, "sysctlbyname"); 44*d4514f0bSApple OSS Distributions ret = sysctlbyname("kern.task_set_filter_msg_flag", &cur_filter_flag, &sysctl_size, NULL, 0); 45*d4514f0bSApple OSS Distributions T_ASSERT_POSIX_SUCCESS(ret, "sysctlbyname"); 46*d4514f0bSApple OSS Distributions T_ASSERT_EQ(cur_filter_flag, 1, "Task should have filtering on"); 47*d4514f0bSApple OSS Distributions 48*d4514f0bSApple OSS Distributions pid_t pid = fork(); 49*d4514f0bSApple OSS Distributions if (pid == 0) { 50*d4514f0bSApple OSS Distributions cur_filter_flag = 0; 51*d4514f0bSApple OSS Distributions ret = sysctlbyname("kern.task_set_filter_msg_flag", &cur_filter_flag, &sysctl_size, NULL, 0); 52*d4514f0bSApple OSS Distributions if (ret == 0) { 53*d4514f0bSApple OSS Distributions if (cur_filter_flag == 1) { 54*d4514f0bSApple OSS Distributions exit(0); 55*d4514f0bSApple OSS Distributions } 56*d4514f0bSApple OSS Distributions } 57*d4514f0bSApple OSS Distributions exit(1); 58*d4514f0bSApple OSS Distributions } 59*d4514f0bSApple OSS Distributions 60*d4514f0bSApple OSS Distributions int status; 61*d4514f0bSApple OSS Distributions ret = waitpid(pid, &status, 0); 62*d4514f0bSApple OSS Distributions T_ASSERT_POSIX_SUCCESS(ret, "waitpid"); 63*d4514f0bSApple OSS Distributions 64*d4514f0bSApple OSS Distributions if (WIFEXITED(status)) { 65*d4514f0bSApple OSS Distributions const int exit_code = WEXITSTATUS(status); 66*d4514f0bSApple OSS Distributions T_ASSERT_EQ(exit_code, 0, "Child inherited the filter msg flag"); 67*d4514f0bSApple OSS Distributions } 68*d4514f0bSApple OSS Distributions 69*d4514f0bSApple OSS Distributions /* Turn off task msg filtering */ 70*d4514f0bSApple OSS Distributions cur_filter_flag = 1; 71*d4514f0bSApple OSS Distributions new_filter_flag = 0; 72*d4514f0bSApple OSS Distributions ret = sysctlbyname("kern.task_set_filter_msg_flag", &cur_filter_flag, &sysctl_size, &new_filter_flag, sizeof(new_filter_flag)); 73*d4514f0bSApple OSS Distributions T_ASSERT_POSIX_SUCCESS(ret, "sysctlbyname"); 74*d4514f0bSApple OSS Distributions T_ASSERT_EQ(cur_filter_flag, 1, "Task should have filtering on"); 75*d4514f0bSApple OSS Distributions 76*d4514f0bSApple OSS Distributions T_END; 77*d4514f0bSApple OSS Distributions } 78