#include <atomic>
#include <assert.h>
#include <mach/mach.h>
#include <kern/kcdata.h>
#include <mach-o/dyld_priv.h>
#include "Loading.h"
#include "Tracing.h"
extern "C" int
os_fault_with_payload(uint32_t reason_namespace, uint64_t reason_code,
void *payload, uint32_t payload_size, const char *reason_string,
uint64_t reason_flags) __attribute__((cold));
namespace dyld3 {
VIS_HIDDEN
void kdebug_trace_dyld_image(const uint32_t code,
const char* imagePath,
const uuid_t* uuid_bytes,
const fsobj_id_t fsobjid,
const fsid_t fsid,
const mach_header* load_addr)
{
uint64_t id = kdebug_trace_string(code, 0, imagePath);
#if __ARM_ARCH_7K__
uint32_t *uuid = (uint32_t *)uuid_bytes;
kdebug_trace(KDBG_CODE(DBG_DYLD, DBG_DYLD_UUID, code + 2), uuid[0],
uuid[1], uuid[2], uuid[3]);
kdebug_trace(KDBG_CODE(DBG_DYLD, DBG_DYLD_UUID, code + 3),
(uint32_t)load_addr, fsid.val[0], fsid.val[1],
fsobjid.fid_objno);
kdebug_trace(KDBG_CODE(DBG_DYLD, DBG_DYLD_UUID, code + 4),
fsobjid.fid_generation, id, 0, 0);
#else
uint64_t *uuid = (uint64_t *)uuid_bytes;
kdebug_trace(KDBG_CODE(DBG_DYLD, DBG_DYLD_UUID, code), uuid[0],
uuid[1], (uint64_t)load_addr,
(uint64_t)fsid.val[0] | ((uint64_t)fsid.val[1] << 32));
kdebug_trace(KDBG_CODE(DBG_DYLD, DBG_DYLD_UUID, code + 1),
(uint64_t)fsobjid.fid_objno |
((uint64_t)fsobjid.fid_generation << 32),
id, 0, 0);
#endif
kdebug_trace_string(code, id, nullptr);
}
#if BUILDING_DYLD
static std::atomic<uint64_t> trace_pair_id(0);
#else
static std::atomic<uint64_t> trace_pair_id(1LL<<63);
#endif
VIS_HIDDEN
bool kdebug_trace_dyld_enabled(uint32_t code) {
return kdebug_is_enabled(code);
}
VIS_HIDDEN
void kdebug_trace_dyld_marker(uint32_t code, kt_arg data1, kt_arg data2, kt_arg data3, kt_arg data4) {
if (kdebug_is_enabled(code)) {
data1.prepare(code);
data2.prepare(code);
data3.prepare(code);
data4.prepare(code);
kdebug_trace(code, data1.value(), data2.value(), data3.value(), data4.value());
data4.destroy(code);
data3.destroy(code);
data2.destroy(code);
data1.destroy(code);
}
}
VIS_HIDDEN
uint64_t kdebug_trace_dyld_duration_start(uint32_t code, kt_arg data1, kt_arg data2, kt_arg data3) {
uint64_t result = 0;
if (kdebug_is_enabled(code)) {
result = ++trace_pair_id;
data1.prepare(code);
data2.prepare(code);
data3.prepare(code);
kdebug_trace(code | DBG_FUNC_START, result, data1.value(), data2.value(), data3.value());
data3.destroy(code);
data2.destroy(code);
data1.destroy(code);
}
return result;
}
VIS_HIDDEN
void kdebug_trace_dyld_duration_end(uint64_t trace_id, uint32_t code, kt_arg data1, kt_arg data2, kt_arg data3) {
if (trace_id != 0 && kdebug_is_enabled(code)) {
data1.prepare(code);
data2.prepare(code);
data3.prepare(code);
kdebug_trace(code | DBG_FUNC_END, trace_id, data1.value(), data2.value(), data3.value());
data3.destroy(code);
data2.destroy(code);
data1.destroy(code);
}
}
void ScopedTimer::startTimer() {
current_trace_id = kdebug_trace_dyld_duration_start(code, data1, data2, data3);
}
void ScopedTimer::endTimer() {
kdebug_trace_dyld_duration_end(current_trace_id, code, data4, data5, data6);
}
void syntheticBacktrace(const char *reason, bool enableExternally) {
if (!enableExternally && !internalInstall()) { return; }
char payloadBuffer[EXIT_REASON_PAYLOAD_MAX_LEN];
dyld_abort_payload* payload = (dyld_abort_payload*)payloadBuffer;
payload->version = 1;
payload->flags = 0;
payload->targetDylibPathOffset = 0;
payload->clientPathOffset = 0;
payload->symbolOffset = 0;
int payloadSize = sizeof(dyld_abort_payload);
char truncMessage[EXIT_REASON_USER_DESC_MAX_LEN];
strlcpy(truncMessage, reason, EXIT_REASON_USER_DESC_MAX_LEN);
os_fault_with_payload(OS_REASON_DYLD, DYLD_EXIT_REASON_OTHER, payloadBuffer, payloadSize, truncMessage, 0);
}
};