#include <stdio.h>
#include <stdlib.h>
#include <dlfcn.h>
#include <mach-o/dyld.h>
#include <mach-o/dyld_priv.h>
#include <unordered_set>
#include "test_support.h"
extern "C" void foo();
extern mach_header __dso_handle;
static std::unordered_set<const mach_header*> sCurrentImages;
static void notify(unsigned count, const mach_header* mhs[], const char* paths[])
{
LOG("notification:");
for (unsigned i=0; i < count; ++i) {
const mach_header* mh = mhs[i];
const char* path = paths[i];
LOG(" %3d mh=%p, path=%s", i, mh, path);
if ( sCurrentImages.count(mh) != 0 ) {
FAIL("notified twice about %p", mh);
exit(0);
}
sCurrentImages.insert(mh);
}
}
int main(int argc, const char* argv[], const char* envp[], const char* apple[]) {
_dyld_register_for_bulk_image_loads(¬ify);
if ( sCurrentImages.count(&__dso_handle) == 0 ) {
FAIL("did not notify us about main executable");
}
const mach_header* libSysMH = dyld_image_header_containing_address((void*)&printf);
if ( sCurrentImages.count(libSysMH) == 0 ) {
FAIL("did not notify us about libsystem_c.dylib");
}
const mach_header* libFoo = dyld_image_header_containing_address((void*)&foo);
if ( sCurrentImages.count(libFoo) == 0 ) {
FAIL("did not notify us about libfoo.dylib");
}
void* handle2 = dlopen(RUN_DIR "/libfoo2.dylib", RTLD_FIRST);
if ( handle2 == NULL ) {
FAIL("dlopen(\"%s\") failed with: %s", RUN_DIR "/libfoo.dylib", dlerror());
}
const void* libfoo2Foo = dlsym(handle2, "foo");
const mach_header* libfoo2MH = dyld_image_header_containing_address(libfoo2Foo);
if ( sCurrentImages.count(libfoo2MH) == 0 ) {
FAIL("did not notify us about libfoo2.dylib");
}
const void* inflateSym = dlsym(RTLD_DEFAULT, "inflate");
if ( inflateSym == NULL ) {
FAIL("did not load libz.dylib");
}
const mach_header* libzMH = dyld_image_header_containing_address(inflateSym);
if ( sCurrentImages.count(libzMH) == 0 ) {
FAIL("did not notify us about libz.dylib");
}
void* handleBar = dlopen(RUN_DIR "/libbar.dylib", RTLD_FIRST);
if ( handleBar == NULL ) {
FAIL("dlopen(\"%s\") failed with: %s", RUN_DIR "/libbar.dylib", dlerror());
}
PASS("SUuccess");
}