#include <stdio.h>
#include <stdlib.h>
#include <dlfcn.h>
#include <mach-o/dyld.h>
#include <mach-o/dyld_priv.h>
#include <unordered_set>
extern "C" void foo();
extern mach_header __dso_handle;
static std::unordered_set<const mach_header*> sCurrentImages;
static void notify(const mach_header* mh, const char* path, bool unloadable)
{
fprintf(stderr, "mh=%p, path=%s, unloadable=%d\n", mh, path, unloadable);
if ( sCurrentImages.count(mh) != 0 ) {
printf("[FAIL] _dyld_register_for_image_loads: notified twice about %p\n", mh);
exit(0);
}
sCurrentImages.insert(mh);
const char* leaf = strrchr(path, '/');
if ( unloadable ) {
if ( (strcmp(leaf, "/libfoo2.dylib") != 0) && (strcmp(leaf, "/foo.bundle") != 0) ) {
printf("[FAIL] _dyld_register_for_image_loads: image incorrectly marked unloadable %p %s\n", mh, path);
exit(0);
}
}
else {
if ( (strcmp(leaf, "/libfoo2.dylib") == 0) || (strcmp(leaf, "/foo.bundle") == 0) ) {
printf("[FAIL] _dyld_register_for_image_loads: image incorrectly marked as not unloadable %p %s\n", mh, path);
exit(0);
}
}
}
int main()
{
printf("[BEGIN] _dyld_register_for_image_loads\n");
_dyld_register_for_image_loads(¬ify);
if ( sCurrentImages.count(&__dso_handle) == 0 ) {
printf("[FAIL] _dyld_register_for_image_loads() did not notify us about main executable\n");
exit(0);
}
const mach_header* libSysMH = dyld_image_header_containing_address((void*)&printf);
if ( sCurrentImages.count(libSysMH) == 0 ) {
printf("[FAIL] _dyld_register_for_image_loads() did not notify us about libsystem_c.dylib\n");
exit(0);
}
const mach_header* libFoo = dyld_image_header_containing_address((void*)&foo);
if ( sCurrentImages.count(libFoo) == 0 ) {
printf("[FAIL] _dyld_register_for_image_loads() did not notify us about libfoo.dylib\n");
exit(0);
}
void* handle2 = dlopen(RUN_DIR "/libfoo2.dylib", RTLD_FIRST);
if ( handle2 == NULL ) {
printf("[FAIL] dlopen(\"%s\") failed with: %s\n", RUN_DIR "/libfoo.dylib", dlerror());
exit(0);
}
const void* libfoo2Foo = dlsym(handle2, "foo");
const mach_header* libfoo2MH = dyld_image_header_containing_address(libfoo2Foo);
if ( sCurrentImages.count(libfoo2MH) == 0 ) {
printf("[FAIL] _dyld_register_for_image_loads() did not notify us about libfoo2.dylib\n");
exit(0);
}
void* handleB = dlopen(RUN_DIR "/foo.bundle", RTLD_FIRST);
if ( handleB == NULL ) {
printf("[FAIL] dlopen(\"%s\") failed with: %s\n", RUN_DIR "/foo.bundle", dlerror());
exit(0);
}
const void* libfooBFoo = dlsym(handle2, "foo");
const mach_header* libfooB = dyld_image_header_containing_address(libfooBFoo);
if ( sCurrentImages.count(libfooB) == 0 ) {
printf("[FAIL] _dyld_register_for_image_loads() did not notify us about foo.bundle\n");
exit(0);
}
printf("[PASS] _dyld_register_for_image_loads\n");
return 0;
}