#include <stdint.h>
#include <stdbool.h>
#include <stdlib.h>
#include <stdatomic.h>
#include <machine/cpu_capabilities.h>
#include <sys/kdebug.h>
#include <sys/kdebug_signpost.h>
#include <sys/errno.h>
#include <sys/param.h>
#include <sys/sysctl.h>
#include <mach/mach.h>
#include <mach/mach_vm.h>
extern int __kdebug_typefilter(void** addr, size_t* size);
extern int __kdebug_trace64(uint32_t code, uint64_t arg1, uint64_t arg2,
uint64_t arg3, uint64_t arg4);
extern uint64_t __kdebug_trace_string(uint32_t debugid, uint64_t str_id,
const char *str);
static int kdebug_signpost_internal(uint32_t debugid, uintptr_t arg1,
uintptr_t arg2, uintptr_t arg3, uintptr_t arg4);
void *
kdebug_typefilter(void)
{
static void* typefilter;
if (__builtin_expect(!typefilter, 0)) {
void* ptr = NULL;
size_t ptr_size = 0;
if (__kdebug_typefilter(&ptr, &ptr_size) == 0) {
void* old_value = NULL;
if (ptr && !atomic_compare_exchange_strong((void* _Atomic volatile *)&typefilter, &old_value, ptr)) {
mach_vm_deallocate(mach_task_self(), (mach_vm_offset_t)ptr, KDBG_TYPEFILTER_BITMAP_SIZE);
}
}
}
return typefilter;
}
bool
kdebug_is_enabled(uint32_t debugid)
{
uint32_t state = COMM_PAGE_READ(uint32_t, KDEBUG_ENABLE);
if (state == 0) {
return FALSE;
}
if ((state & KDEBUG_COMMPAGE_ENABLE_TYPEFILTER) > 0) {
uint8_t* typefilter = kdebug_typefilter();
if (typefilter && isset(typefilter, KDBG_EXTRACT_CSC(debugid)) == 0) {
return FALSE;
}
}
return TRUE;
}
bool
kdebug_using_continuous_time(void)
{
uint32_t state = COMM_PAGE_READ(uint32_t, KDEBUG_ENABLE);
return state & KDEBUG_ENABLE_CONT_TIME;
}
int
kdebug_trace(uint32_t debugid, uint64_t arg1, uint64_t arg2, uint64_t arg3,
uint64_t arg4)
{
if (!kdebug_is_enabled(debugid)) {
return 0;
}
return __kdebug_trace64(debugid, arg1, arg2, arg3, arg4);
}
uint64_t
kdebug_trace_string(uint32_t debugid, uint64_t str_id, const char *str)
{
if (!kdebug_is_enabled(debugid)) {
return 0;
}
if ((int64_t)str_id == -1) {
errno = EINVAL;
return (uint64_t)-1;
}
if (str_id == 0 && str == NULL) {
errno = EINVAL;
return (uint64_t)-1;
}
return __kdebug_trace_string(debugid, str_id, str);
}
static int
kdebug_signpost_internal(uint32_t debugid, uintptr_t arg1, uintptr_t arg2, uintptr_t arg3, uintptr_t arg4)
{
if (KDBG_EXTRACT_CSC(debugid) != 0) {
errno = EINVAL;
return -1;
}
debugid |= APPSDBG_CODE(DBG_APP_SIGNPOST, 0);
return kdebug_trace(debugid, arg1, arg2, arg3, arg4);
}
int
kdebug_signpost(uint32_t code, uintptr_t arg1, uintptr_t arg2, uintptr_t arg3, uintptr_t arg4)
{
return kdebug_signpost_internal(code << KDBG_CODE_OFFSET, arg1, arg2, arg3, arg4);
}
int
kdebug_signpost_start(uint32_t code, uintptr_t arg1, uintptr_t arg2, uintptr_t arg3, uintptr_t arg4)
{
return kdebug_signpost_internal((code << KDBG_CODE_OFFSET) | DBG_FUNC_START, arg1, arg2, arg3, arg4);
}
int
kdebug_signpost_end(uint32_t code, uintptr_t arg1, uintptr_t arg2, uintptr_t arg3, uintptr_t arg4)
{
return kdebug_signpost_internal((code << KDBG_CODE_OFFSET) | DBG_FUNC_END, arg1, arg2, arg3, arg4);
}