#include <stdio.h>
#include <stdlib.h>
#include <dlfcn.h>
#include <unistd.h>
#include <signal.h>
#include <spawn.h>
#include <errno.h>
#include <mach/mach.h>
#include <mach/machine.h>
#include <err.h>
#include <System/sys/reason.h>
#include <System/sys/proc_info.h>
#include <System/kern/kern_cdata.h>
#include <libproc.h>
#include <mach-o/dyld_priv.h>
#include "test_support.h"
void runTest(const char* prog, uint64_t dyldReason, const char* expectedDylibPath, const char* expectedSymbol) {
dispatch_block_t oneShotSemaphoreBlock = dispatch_block_create(DISPATCH_BLOCK_INHERIT_QOS_CLASS, ^{});
LOG("Running test for %s / %s / %s", prog, expectedDylibPath, expectedSymbol);
_process process;
process.set_executable_path(prog);
process.set_crash_handler(^(task_t task) {
LOG("Crash for task=%u", task);
mach_vm_address_t corpse_data;
mach_vm_size_t corpse_size;
kern_return_t kr = task_map_corpse_info_64(mach_task_self(), task, &corpse_data, &corpse_size);
if (kr != KERN_SUCCESS) {
FAIL("Could not read corpse data kr=(%d)", kr);
}
kcdata_iter_t autopsyData = kcdata_iter((void*)corpse_data, corpse_size);
if (!kcdata_iter_valid(autopsyData)) {
FAIL("Corpse Data Invalid");
}
kcdata_iter_t pidIter = kcdata_iter_find_type(autopsyData, TASK_CRASHINFO_PID);
pid_t pid = *(pid_t *) (kcdata_iter_payload(pidIter));
LOG("Crash for pid=%u", pid);
kcdata_iter_t exitReasonData = kcdata_iter_find_type(autopsyData, EXIT_REASON_SNAPSHOT);
if (!kcdata_iter_valid(exitReasonData)) {
FAIL("Could not find exit data");
}
struct exit_reason_snapshot *ers = (struct exit_reason_snapshot *)kcdata_iter_payload(exitReasonData);
if ( ers->ers_namespace != OS_REASON_DYLD ) {
FAIL("eri_namespace (%d) != OS_REASON_DYLD", ers->ers_namespace);
}
if ( ers->ers_code != dyldReason ) {
FAIL("eri_code (%llu) != dyldReason (%lld)", ers->ers_code, dyldReason);
}
kcdata_iter_t iter = kcdata_iter((void*)corpse_data, corpse_size);
bool foundOSReason = false;
KCDATA_ITER_FOREACH(iter) {
if (kcdata_iter_type(iter) == KCDATA_TYPE_NESTED_KCDATA) {
kcdata_iter_t nestedIter = kcdata_iter(kcdata_iter_payload(iter), kcdata_iter_size(iter));
if ( kcdata_iter_type(nestedIter) != KCDATA_BUFFER_BEGIN_OS_REASON ){
return;
}
foundOSReason = true;
kcdata_iter_t payloadIter = kcdata_iter_find_type(nestedIter, EXIT_REASON_USER_PAYLOAD);
if ( !kcdata_iter_valid(payloadIter) ) {
FAIL("invalid kcdata payload iterator from payload data");
}
const dyld_abort_payload* dyldInfo = (dyld_abort_payload*)kcdata_iter_payload(payloadIter);
if ( dyldInfo->version != 1 ) {
FAIL("dyld payload is not version 1");
}
if ( (dyldInfo->flags & 1) == 0 ) {
FAIL("dyld flags should have low bit set to indicate process terminated during launch");
}
if ( expectedDylibPath != NULL ) {
if ( dyldInfo->targetDylibPathOffset != 0 ) {
const char* targetDylib = (char*)dyldInfo + dyldInfo->targetDylibPathOffset;
if ( strstr(targetDylib, expectedDylibPath) == NULL ) {
FAIL("dylib path (%s) not what expected (%s)", targetDylib, expectedDylibPath);
}
} else {
FAIL("dylib path (%s) not provided by dyld", expectedDylibPath);
}
}
if ( expectedSymbol != NULL ) {
if ( dyldInfo->targetDylibPathOffset != 0 ) {
const char* missingSymbol = (char*)dyldInfo + dyldInfo->symbolOffset;
if ( strcmp(expectedSymbol, missingSymbol) != 0 ) {
FAIL("symbol (%s) not what expected (%s)", missingSymbol, expectedSymbol);
}
} else {
FAIL("symbol (%s) not provided by dyld", expectedSymbol);
}
}
}
}
if (!foundOSReason) {
FAIL("Did not find KCDATA_BUFFER_BEGIN_OS_REASON");
}
oneShotSemaphoreBlock();
});
pid_t pid = process.launch();
LOG("Launch pid: %u", pid);
dispatch_block_wait(oneShotSemaphoreBlock, DISPATCH_TIME_FOREVER);
}
int main(int argc, const char* argv[], const char* envp[], const char* apple[]) {
runTest(RUN_DIR "/prog_missing_dylib.exe", DYLD_EXIT_REASON_DYLIB_MISSING, "/cant/find/me.dylib", NULL);
runTest(RUN_DIR "/prog_missing_symbol.exe", DYLD_EXIT_REASON_SYMBOL_MISSING, "libMissingSymbols.dylib", "_slipperySymbol");
PASS("Success");
}