tests/atm_diagnostic_flag_entitled.c

   1 #include <darwintest.h>
   2
   3 #include <mach/mach_error.h>
   4 #include <mach/mach_host.h>
   5
   6 T_GLOBAL_META(T_META_NAMESPACE("xnu.debugging"));
   7
   8 /*
   9  * The low 8 bits may be in use, so modify one
  10  * of the upper 8 bits to ensure round-tripping.
  11  */
  12 #define LIBTRACE_PRIVATE_DATA  0x01000000
  13
  14 extern void drop_priv(void);
  15
  16 static bool _needs_reset;
  17 static uint32_t _original;
  18
  19 static uint32_t
  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
  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
  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
  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 }