1*c54f35caSApple OSS Distributions #include <stdint.h> 2*c54f35caSApple OSS Distributions #include <stdio.h> 3*c54f35caSApple OSS Distributions #include <unistd.h> 4*c54f35caSApple OSS Distributions 5*c54f35caSApple OSS Distributions #include <mach/mach.h> 6*c54f35caSApple OSS Distributions #include <mach/mk_timer.h> 7*c54f35caSApple OSS Distributions 8*c54f35caSApple OSS Distributions #include <darwintest.h> 9*c54f35caSApple OSS Distributions 10*c54f35caSApple OSS Distributions T_GLOBAL_META(T_META_RUN_CONCURRENTLY(true), 11*c54f35caSApple OSS Distributions T_META_NAMESPACE("xnu.ipc"), 12*c54f35caSApple OSS Distributions T_META_RADAR_COMPONENT_NAME("xnu"), 13*c54f35caSApple OSS Distributions T_META_RADAR_COMPONENT_VERSION("IPC")); 14*c54f35caSApple OSS Distributions 15*c54f35caSApple OSS Distributions T_DECL(mktimer_kobject, "mktimer_kobject()", T_META_ALL_VALID_ARCHS(true), T_META_IGNORECRASHES(".*mktimer_kobject.*")) 16*c54f35caSApple OSS Distributions { 17*c54f35caSApple OSS Distributions mach_port_t timer_port = MACH_PORT_NULL; 18*c54f35caSApple OSS Distributions mach_port_t notify_port = MACH_PORT_NULL; 19*c54f35caSApple OSS Distributions 20*c54f35caSApple OSS Distributions kern_return_t kr = KERN_SUCCESS; 21*c54f35caSApple OSS Distributions task_exc_guard_behavior_t old, new; 22*c54f35caSApple OSS Distributions 23*c54f35caSApple OSS Distributions /* 24*c54f35caSApple OSS Distributions * Disable [optional] Mach port guard exceptions to avoid fatal crash 25*c54f35caSApple OSS Distributions */ 26*c54f35caSApple OSS Distributions kr = task_get_exc_guard_behavior(mach_task_self(), &old); 27*c54f35caSApple OSS Distributions T_ASSERT_MACH_SUCCESS(kr, "task_get_exc_guard_behavior"); 28*c54f35caSApple OSS Distributions new = (old & ~TASK_EXC_GUARD_MP_DELIVER); 29*c54f35caSApple OSS Distributions kr = task_set_exc_guard_behavior(mach_task_self(), new); 30*c54f35caSApple OSS Distributions T_ASSERT_MACH_SUCCESS(kr, "task_set_exc_guard_behavior new"); 31*c54f35caSApple OSS Distributions 32*c54f35caSApple OSS Distributions /* 33*c54f35caSApple OSS Distributions * timer port 34*c54f35caSApple OSS Distributions * This is a receive right which is also a kobject 35*c54f35caSApple OSS Distributions */ 36*c54f35caSApple OSS Distributions timer_port = mk_timer_create(); 37*c54f35caSApple OSS Distributions T_ASSERT_NE(timer_port, (mach_port_t)MACH_PORT_NULL, "mk_timer_create: %s", mach_error_string(kr)); 38*c54f35caSApple OSS Distributions 39*c54f35caSApple OSS Distributions mach_port_set_context(mach_task_self(), timer_port, (mach_port_context_t) 0x1); 40*c54f35caSApple OSS Distributions T_ASSERT_EQ(kr, KERN_SUCCESS, "mach_port_set_context(timer_port): %s", mach_error_string(kr)); 41*c54f35caSApple OSS Distributions 42*c54f35caSApple OSS Distributions /* notification port for the mk_timer port to come back on */ 43*c54f35caSApple OSS Distributions kr = mach_port_allocate(mach_task_self(), MACH_PORT_RIGHT_RECEIVE, ¬ify_port); 44*c54f35caSApple OSS Distributions T_ASSERT_EQ(kr, KERN_SUCCESS, "mach_port_allocate(notify_port): %s", mach_error_string(kr)); 45*c54f35caSApple OSS Distributions 46*c54f35caSApple OSS Distributions kr = mach_port_set_context(mach_task_self(), notify_port, (mach_port_context_t) 0x2); 47*c54f35caSApple OSS Distributions T_ASSERT_EQ(kr, KERN_SUCCESS, "mach_port_set_context(notify_port): %s", mach_error_string(kr)); 48*c54f35caSApple OSS Distributions 49*c54f35caSApple OSS Distributions T_LOG("timer: 0x%x, notify: 0x%x", timer_port, notify_port); 50*c54f35caSApple OSS Distributions 51*c54f35caSApple OSS Distributions /* 52*c54f35caSApple OSS Distributions * This code generates a mach port guard exception and should be tested with an exception catcher. 53*c54f35caSApple OSS Distributions * Will be updated in <rdar://problem/70971318> 54*c54f35caSApple OSS Distributions */ 55*c54f35caSApple OSS Distributions mach_port_t previous = MACH_PORT_NULL; 56*c54f35caSApple OSS Distributions 57*c54f35caSApple OSS Distributions /* request a port-destroyed notification on the timer port */ 58*c54f35caSApple OSS Distributions kr = mach_port_request_notification(mach_task_self(), timer_port, MACH_NOTIFY_PORT_DESTROYED, 59*c54f35caSApple OSS Distributions 0, notify_port, MACH_MSG_TYPE_MAKE_SEND_ONCE, &previous); 60*c54f35caSApple OSS Distributions /* this will ordinarily fail with a guard exception! */ 61*c54f35caSApple OSS Distributions T_ASSERT_MACH_ERROR(kr, KERN_INVALID_RIGHT, "notifications should NOT work on mk_timer ports!"); 62*c54f35caSApple OSS Distributions 63*c54f35caSApple OSS Distributions /* restore the old guard behavior */ 64*c54f35caSApple OSS Distributions kr = task_set_exc_guard_behavior(mach_task_self(), old); 65*c54f35caSApple OSS Distributions T_ASSERT_MACH_SUCCESS(kr, "task_set_exc_guard_behavior old"); 66*c54f35caSApple OSS Distributions 67*c54f35caSApple OSS Distributions T_LOG("done"); 68*c54f35caSApple OSS Distributions } 69