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