/*
* Copyright (c) 2000-2024 Apple Inc. All rights reserved.
*
* @APPLE_OSREFERENCE_LICENSE_HEADER_START@
*
* This file contains Original Code and/or Modifications of Original Code
* as defined in and that are subject to the Apple Public Source License
* Version 2.0 (the 'License'). You may not use this file except in
* compliance with the License. The rights granted to you under the License
* may not be used to create, or enable the creation or redistribution of,
* unlawful or unlicensed copies of an Apple operating system, or to
* circumvent, violate, or enable the circumvention or violation of, any
* terms of an Apple operating system software license agreement.
*
* Please obtain a copy of the License at
* http://www.opensource.apple.com/apsl/ and read it before using this file.
*
* The Original Code and all software distributed under the License are
* distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
* EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
* INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY,
* FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT.
* Please see the License for the specific language governing rights and
* limitations under the License.
*
* @APPLE_OSREFERENCE_LICENSE_HEADER_END@
*/
#include "std_safe.h"
#include "dt_proxy.h"
#include "unit_test_utils.h"
#include <mach/arm/vm_types.h> // for vm_map_offset_t
#include <mach/mach_types.h>
#include <kern/assert.h>
#include <kern/debug.h>
// This file is linked to the same .dylib as the XNU code static library
// to provide some utilities that XNU code relies on when building for unit-test
// This symbol normally comes from lastkernelconstructor.o but this object is not linked to libkernel since it is
// a zero length symbol, which fails to prelink with ld -r
// It is defined here again to resolve the external.
// It is used for debugging and in kext related functions which are not needed by the tester
void* last_kernel_symbol = NULL;
// In normal XNU build this is a global of type mach_header_64 that the linker adds. The user-space linker
// doesn't add it so it's added here to resolve the external. It's not needed by the tester. see <mach-o/ldsyms.h>
int _mh_execute_header;
// called from fake_init, setup proxies for darwintest asserts
struct dt_proxy_callbacks *dt_proxy = NULL;
void
set_dt_proxy_attached(struct dt_proxy_callbacks *p)
{
dt_proxy = p;
}
struct dt_proxy_callbacks *
get_dt_proxy_attached(void)
{
return dt_proxy;
}
// check if panic/assert were expected by the test using T_ASSERT_PANIC
struct ut_expected_panic_s ut_expected_panic;
void
ut_check_expected_panic(const char* panic_str)
{
if (!ut_expected_panic.expect_panic) {
return;
}
ut_expected_panic.expect_panic = false;
if (ut_expected_panic.str_contains != NULL) {
if (strstr(panic_str, ut_expected_panic.str_contains) == NULL) {
PT_LOG_FMTSTR("Panic with unexpected panic-string, expected: `%s`", ut_expected_panic.str_contains);
return;
}
}
PT_LOG("Panic was expected");
longjmp(ut_expected_panic.jb, 1);
}
// This function is called on an assert instead of invoking a brk instruction which would trap the kernel
__attribute__((noreturn)) void
ut_assert_trap(int code, long a, long b, long c)
{
struct kernel_panic_reason pr = {};
if (code == MACH_ASSERT_TRAP_CODE) {
panic_assert_format(pr.buf, sizeof(pr.buf), (struct mach_assert_hdr *)a, b, c);
PT_LOG_OR_RAW_FMTSTR("%s", pr.buf);
} else {
snprintf(pr.buf, sizeof(pr.buf), "%x", code);
PT_LOG_OR_RAW_FMTSTR("Unknown assert code %s", pr.buf);
}
ut_check_expected_panic(pr.buf); // may not return
PT_FAIL("Unexpected assert fail, exiting");
abort();
}
// This function can be called from the tested code to force a context switch when using fibers
// See the mock implementation
void
ut_fibers_ctxswitch(void)
{
}
// This function can be called from the tested code to force a context switch to a specific fiber
// See the mock implementation
void
ut_fibers_ctxswitch_to(int fiber_id)
{
}
// This function can be called from the tested code to get the current fiber id when using fibers, -1 otherwise
// See the mock implementation
int
ut_fibers_current_id(void)
{
return -1;
}
static void
fail_not_mocked()
{
PT_FAIL("This function should never be called since it is mocked by the mocks dylib");
}
// This function is changed from being a macro. It needs to have an implementation
// in the code attached to XNU so that it can be mocked
__mockable void
lock_disable_preemption_for_thread(thread_t t)
{
fail_not_mocked();
}
__mockable __attribute__((const)) thread_t
current_thread_fast(void)
{
fail_not_mocked();
return NULL;
}
extern vm_map_t vm_map_create_external(
pmap_t pmap,
vm_map_offset_t min,
vm_map_offset_t max,
boolean_t pageable);
// this function alias is done during linking of XNU, which doesn't happen when building the library
vm_map_t
vm_map_create(
pmap_t pmap,
vm_map_offset_t min,
vm_map_offset_t max,
boolean_t pageable)
{
return vm_map_create_external(pmap, min, max, pageable);
}