Open Menu / tests / atm_diagnostic_flag.c
Loading...
 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
#include <darwintest.h>

#include <mach/mach_error.h>
#include <mach/mach_host.h>

T_GLOBAL_META(T_META_NAMESPACE("xnu.debugging"));

/*
 * The low 8 bits may be in use, so modify one
 * of the upper 8 bits to ensure round-tripping.
 */
#define LIBTRACE_PRIVATE_DATA  0x01000000

extern void drop_priv(void);

static bool _needs_reset;
static uint32_t _original;

static uint32_t
_save_atm_diagnostic_flag(void)
{
	kern_return_t kr;
	kr = host_get_atm_diagnostic_flag(mach_host_self(), &_original);
	T_QUIET; T_ASSERT_MACH_SUCCESS(kr, "host_get_atm_diagnostic_flag()");
	T_LOG("Original ATM diagnostic flag: 0x%08x", _original);
	return _original;
}

static kern_return_t
_mutate_atm_diagnostic_flag(uint32_t v)
{
	T_LOG("Try to set ATM diagnostic flag to: 0x%08x", v);
	kern_return_t kr = host_set_atm_diagnostic_flag(mach_host_self(), v);
	if (kr == KERN_SUCCESS) {
		_needs_reset = true;
	}
	return kr;
}

static void
_reset_atm_diagnostic_flag(void)
{
	if (!_needs_reset) {
		return;
	}
	T_LOG("Reset ATM diagnostic flag to: 0x%08x", _original);
	kern_return_t kr;
	kr = host_set_atm_diagnostic_flag(mach_host_self(), _original);
	if (kr != KERN_SUCCESS) {
		T_ASSERT_FAIL("host_set_atm_diagnostic_flag() failed: %s",
		    mach_error_string(kr));
	}
}

static void
_toggle_atm_diagnostic_flag(void)
{
	T_ATEND(_reset_atm_diagnostic_flag);
	uint32_t f = _save_atm_diagnostic_flag();
	f ^= LIBTRACE_PRIVATE_DATA;
	kern_return_t kr = _mutate_atm_diagnostic_flag(f);
	if (kr == KERN_NOT_SUPPORTED) {
		T_SKIP("Seems ATM is disabled on this platform. "
		    "Ignoring host_set_atm_diagnostic_flag functionality. "
		    "Bailing gracefully.");
	}
	T_EXPECT_MACH_ERROR(KERN_NO_ACCESS, kr,
	    "Deny change to atm_diagnostic_flag");
}

T_DECL(atm_diagnostic_flag_unentitled_privileged,
    "expect to fail to set the atm_diagnostic_flag (unentitled, privileged)",
    T_META_ASROOT(true))
{
	_toggle_atm_diagnostic_flag();
}

T_DECL(atm_diagnostic_flag_unentitled_unprivileged,
    "expect to fail to set the atm_diagnostic_flag (unentitled, unprivileged)",
    T_META_ASROOT(false))
{
	drop_priv();
	_toggle_atm_diagnostic_flag();
}