1 #include <darwintest.h>
2
3 #include <mach/mach_error.h>
4 #include <mach/mach_host.h>
5
6 #include "drop_priv.h"
7
8 T_GLOBAL_META(T_META_NAMESPACE("xnu.debugging"));
9
10 /*
11 * The low 8 bits may be in use, so modify one
12 * of the upper 8 bits to ensure round-tripping.
13 */
14 #define LIBTRACE_PRIVATE_DATA 0x01000000
15
16 static bool _needs_reset;
17 static uint32_t _original;
18
19 static uint32_t
_save_atm_diagnostic_flag(void)20 _save_atm_diagnostic_flag(void)
21 {
22 kern_return_t kr;
23 kr = host_get_atm_diagnostic_flag(mach_host_self(), &_original);
24 T_QUIET; T_ASSERT_MACH_SUCCESS(kr, "host_get_atm_diagnostic_flag()");
25 T_LOG("Original ATM diagnostic flag: 0x%08x", _original);
26 return _original;
27 }
28
29 static kern_return_t
_mutate_atm_diagnostic_flag(uint32_t v)30 _mutate_atm_diagnostic_flag(uint32_t v)
31 {
32 T_LOG("Try to set ATM diagnostic flag to: 0x%08x", v);
33 kern_return_t kr = host_set_atm_diagnostic_flag(mach_host_self(), v);
34 if (kr == KERN_SUCCESS) {
35 _needs_reset = true;
36 }
37 return kr;
38 }
39
40 static void
_reset_atm_diagnostic_flag(void)41 _reset_atm_diagnostic_flag(void)
42 {
43 if (!_needs_reset) {
44 return;
45 }
46 T_LOG("Reset ATM diagnostic flag to: 0x%08x", _original);
47 kern_return_t kr;
48 kr = host_set_atm_diagnostic_flag(mach_host_self(), _original);
49 if (kr != KERN_SUCCESS) {
50 T_ASSERT_FAIL("host_set_atm_diagnostic_flag() failed: %s",
51 mach_error_string(kr));
52 }
53 }
54
55 static void
_toggle_atm_diagnostic_flag(void)56 _toggle_atm_diagnostic_flag(void)
57 {
58 T_ATEND(_reset_atm_diagnostic_flag);
59 uint32_t f = _save_atm_diagnostic_flag();
60 f ^= LIBTRACE_PRIVATE_DATA;
61 kern_return_t kr = _mutate_atm_diagnostic_flag(f);
62 if (kr == KERN_NOT_SUPPORTED) {
63 T_SKIP("Seems ATM is disabled on this platform. "
64 "Ignoring host_set_atm_diagnostic_flag functionality. "
65 "Bailing gracefully.");
66 }
67 T_EXPECT_MACH_SUCCESS(kr, "Set atm_diagnostic_flag");
68 }
69
70 T_DECL(atm_diagnostic_flag_entitled_privileged,
71 "change the atm_diagnostic_flag (entitled, privileged)",
72 T_META_ASROOT(true))
73 {
74 _toggle_atm_diagnostic_flag();
75 }
76
77 T_DECL(atm_diagnostic_flag_entitled_unprivileged,
78 "change the atm_diagnostic_flag (entitled, unprivileged)",
79 T_META_ASROOT(false))
80 {
81 drop_priv();
82 _toggle_atm_diagnostic_flag();
83 }
84