#include <os/variant_private.h>
#include <darwintest.h>
T_DECL(os_variant_basic, "Just calls all the APIs")
{
T_MAYFAIL;
T_EXPECT_TRUE(os_variant_has_internal_content("com.apple.Libc.tests"), NULL);
T_MAYFAIL;
T_EXPECT_TRUE(os_variant_has_internal_diagnostics("com.apple.Libc.tests"), NULL);
T_MAYFAIL;
T_EXPECT_TRUE(os_variant_has_internal_ui("com.apple.Libc.tests"), NULL);
T_MAYFAIL;
T_EXPECT_TRUE(os_variant_allows_internal_security_policies("com.apple.Libc.tests"), NULL);
}
#define VARIANT_SKIP_EXPORTED
#include "../libdarwin/variant.c"
T_DECL(os_variant_detailed, "Looks at individual checks")
{
T_MAYFAIL;
T_EXPECT_FALSE(_check_disabled(VP_ALL), NULL);
T_MAYFAIL;
T_EXPECT_FALSE(_check_disabled(VP_CONTENT), NULL);
T_MAYFAIL;
T_EXPECT_FALSE(_check_disabled(VP_DIAGNOSTICS), NULL);
T_MAYFAIL;
T_EXPECT_FALSE(_check_disabled(VP_UI), NULL);
T_MAYFAIL;
T_EXPECT_FALSE(_check_disabled(VP_SECURITY), NULL);
#if !TARGET_OS_IPHONE || TARGET_IPHONE_SIMULATOR
T_MAYFAIL;
T_EXPECT_TRUE(_check_internal_content(), NULL);
#endif
#if TARGET_OS_IPHONE
T_MAYFAIL;
T_EXPECT_TRUE(_check_internal_release_type(), NULL);
#else
T_MAYFAIL;
T_EXPECT_FALSE(_check_internal_diags_profile(), NULL);
#endif
T_MAYFAIL;
T_EXPECT_TRUE(_check_can_has_debugger(), NULL);
}
T_DECL(os_variant_override_parse, "Checks the parsing of the override file")
{
_check_disabled(VP_ALL);
T_LOG("Override: NULL"); _parse_disabled_status(NULL);
T_MAYFAIL; T_EXPECT_FALSE(_check_disabled(VP_CONTENT), NULL);
T_MAYFAIL; T_EXPECT_FALSE(_check_disabled(VP_DIAGNOSTICS), NULL);
T_MAYFAIL; T_EXPECT_FALSE(_check_disabled(VP_UI), NULL);
T_MAYFAIL; T_EXPECT_FALSE(_check_disabled(VP_SECURITY), NULL);
T_LOG("Override: \"content\"");
_parse_disabled_status("content");
T_EXPECT_TRUE(_check_disabled(VP_CONTENT), NULL);
T_EXPECT_FALSE(_check_disabled(VP_DIAGNOSTICS), NULL);
T_EXPECT_FALSE(_check_disabled(VP_UI), NULL);
T_EXPECT_FALSE(_check_disabled(VP_SECURITY), NULL);
T_LOG("Override: \"ui\"");
_parse_disabled_status("ui");
T_EXPECT_FALSE(_check_disabled(VP_CONTENT), NULL);
T_EXPECT_FALSE(_check_disabled(VP_DIAGNOSTICS), NULL);
T_EXPECT_TRUE(_check_disabled(VP_UI), NULL);
T_EXPECT_FALSE(_check_disabled(VP_SECURITY), NULL);
T_LOG("Override: \"security,diagnostics\"");
_parse_disabled_status("security,diagnostics");
T_EXPECT_FALSE(_check_disabled(VP_CONTENT), NULL);
T_EXPECT_TRUE(_check_disabled(VP_DIAGNOSTICS), NULL);
T_EXPECT_FALSE(_check_disabled(VP_UI), NULL);
T_EXPECT_TRUE(_check_disabled(VP_SECURITY), NULL);
T_LOG("Override: \"content,diagnostics,ui,security\"");
_parse_disabled_status("content,diagnostics,ui,security");
T_EXPECT_TRUE(_check_disabled(VP_CONTENT), NULL);
T_EXPECT_TRUE(_check_disabled(VP_DIAGNOSTICS), NULL);
T_EXPECT_TRUE(_check_disabled(VP_UI), NULL);
T_EXPECT_TRUE(_check_disabled(VP_SECURITY), NULL);
T_LOG("Override: \"diagnostics\\n"); _parse_disabled_status("diagnostics\n");
T_EXPECT_FALSE(_check_disabled(VP_CONTENT), NULL);
T_EXPECT_TRUE(_check_disabled(VP_DIAGNOSTICS), NULL);
T_EXPECT_FALSE(_check_disabled(VP_UI), NULL);
T_EXPECT_FALSE(_check_disabled(VP_SECURITY), NULL);
T_LOG("Override: \"content,diagnostics\\nui,security\\n\"");
_parse_disabled_status("content,diagnostics\nui,security\n");
T_EXPECT_TRUE(_check_disabled(VP_CONTENT), NULL);
T_EXPECT_TRUE(_check_disabled(VP_DIAGNOSTICS), NULL);
T_EXPECT_TRUE(_check_disabled(VP_UI), NULL);
T_EXPECT_TRUE(_check_disabled(VP_SECURITY), NULL);
}
T_DECL(os_status_cache, "Checks saving and restoring of state")
{
uint64_t status = 0;
size_t status_size = sizeof(status);
int ret = sysctlbyname(CACHE_SYSCTL_NAME, &status, &status_size, NULL, 0);
T_EXPECT_POSIX_ZERO(ret, "sysctlbyname(kern.osvariant_status)");
T_EXPECT_GT(status, 0ULL, "Kernel's status has bits set");
T_EXPECT_EQ(status & STATUS_INITIAL_BITS, STATUS_INITIAL_BITS, "Kernel's status has initial bits set");
T_MAYFAIL;
T_EXPECT_TRUE(_check_can_has_debugger(), NULL);
status = _get_cached_check_status();
T_LOG("Cached status: %llx", status);
T_EXPECT_EQ(status & STATUS_INITIAL_BITS, STATUS_INITIAL_BITS, "Our status has initial bits set");
_restore_cached_check_status(status);
T_MAYFAIL;
T_EXPECT_TRUE(os_variant_allows_internal_security_policies(NULL), NULL);
status = STATUS_INITIAL_BITS |
(S_NO << (SFP_CAN_HAS_DEBUGGER * STATUS_BIT_WIDTH));
T_LOG("Restoring status without can_has_debugger: %llx", status);
_restore_cached_check_status(status);
T_EXPECT_FALSE(_check_can_has_debugger(), NULL);
_check_disabled(VP_SECURITY);
status = STATUS_INITIAL_BITS |
(0x1ULL << (VP_SECURITY + 32));
T_LOG("Restoring status with override: %llx", status);
_restore_cached_check_status(status);
T_EXPECT_TRUE(_check_disabled(VP_SECURITY), NULL);
}