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