Open Menu / testing / test-cases / driverkit-dlfcn.dtest / dext.c
Loading...
 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
#include <dlfcn.h>
#include <mach-o/dyld_priv.h>
#include <os/log.h>
#include <string.h>

#if __has_feature(ptrauth_calls)
    #include <ptrauth.h>
#endif

static const void* stripPointer(const void* ptr)
{
#if __has_feature(ptrauth_calls)
    return __builtin_ptrauth_strip(ptr, ptrauth_key_asia);
#else
    return ptr;
#endif
}

extern struct mach_header __dso_handle;

int bar()
{
    return 0;
}

int foo()
{
    Dl_info info;
    if ( dladdr(&bar, &info) == 0 ) {
        os_log(OS_LOG_DEFAULT, "dyld-driverkit-dlfcn: dladdr failed for 'bar'");
        return 2;
    }
    if ( strncmp(info.dli_sname, "bar", 3) != 0 ) {
        os_log(OS_LOG_DEFAULT, "dyld-driverkit-dlfcn: dli_sname is '%s' instead of 'bar'", info.dli_sname);
        return 3;
    }
    if ( info.dli_saddr != stripPointer(&bar) ) {
        os_log(OS_LOG_DEFAULT, "dyld-driverkit-dlfcn: dli_saddr is not &bar");
        return 4;
    }
    if ( info.dli_fbase != &__dso_handle ) {
        os_log(OS_LOG_DEFAULT, "dyld-driverkit-dlfcn: dli_fbase is not image that contains &bar");
        return 5;
    }
    return 0;
}

int dext_main(void)
{
    os_log(OS_LOG_DEFAULT, "dyld-driverkit-dlfcn");

    __typeof(&foo) symFunc = (__typeof(&foo))dlsym(RTLD_DEFAULT, "foo");
    if ( symFunc == NULL ) {
        os_log(OS_LOG_DEFAULT, "dyld-driverkit-dlfcn: could not find symbol 'foo'");
        return 1;
    }

    return symFunc();
}

__attribute__((constructor))
void init(void)
{
    _dyld_register_driverkit_main((void (*)())dext_main);
}