main.cxx   [plain text]

// BUILD:  $CC  foo.c -dynamiclib  -install_name $RUN_DIR/libfoo.dylib -o $BUILD_DIR/libfoo.dylib
// BUILD:  $CXX main.cxx -o $BUILD_DIR/dyld_register_test.exe $BUILD_DIR/libfoo.dylib -DRUN_DIR="$RUN_DIR"
// BUILD:  $CC  foo.c -dynamiclib  -install_name $RUN_DIR/libfoo2.dylib -o $BUILD_DIR/libfoo2.dylib
// BUILD:  $CC  foo.c -bundle -o $BUILD_DIR/foo.bundle

// RUN:  ./dyld_register_test.exe

#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(&notify);

    // verify we were notified about already loaded images
    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);
    }

    // verify we were notified about load of libfoo2.dylib
	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);
    }

    // verify we were notified about load of foo.bundle
    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;
}