// BOOT_ARGS: dyld_flags=2 // BUILD: mkdir -p $BUILD_DIR/lib // BUILD: $CC main.cpp -std=c++11 -o $BUILD_DIR/rpath_insert_main.exe -Wl,-rpath,$RUN_DIR/lib // BUILD: $CC foo.c -dynamiclib -install_name $RUN_DIR/libfoo.dylib -o $BUILD_DIR/lib/libfoo.dylib // BUILD: $CC bar.c -dynamiclib -install_name $RUN_DIR/libbar.dylib -o $BUILD_DIR/libbar.dylib // BUILD: $CC baz.c -dynamiclib -install_name $RUN_DIR/libbaz.dylib -o $BUILD_DIR/libbaz.dylib // BUILD: $DYLD_ENV_VARS_ENABLE $BUILD_DIR/rpath_insert_main.exe // Load foo with @rpath, bar with @executable_path, and baz with @loader_path // Note, dyld2 only supports DYLD_INSERT_LIBRARIES with @executable path so we expect failures on @rpath and @loader_path // RUN: DYLD_INSERT_LIBRARIES="@rpath/libfoo.dylib" DYLD_AMFI_FAKE=0xFF ./rpath_insert_main.exe libfoo.dylib // RUN: DYLD_INSERT_LIBRARIES="@executable_path/libbar.dylib" ./rpath_insert_main.exe libbar.dylib // RUN: DYLD_INSERT_LIBRARIES="@loader_path/libbaz.dylib" DYLD_AMFI_FAKE=0xFF ./rpath_insert_main.exe libbaz.dylib #include <stdlib.h> #include <stdio.h> #include <stdbool.h> #include <string.h> #include <mach-o/dyld_priv.h> bool gFoundLibrary = false; const char* gLibraryName = NULL; bool wasImageLoaded(const char* libraryName) { gFoundLibrary = false; gLibraryName = libraryName; _dyld_register_for_image_loads([](const mach_header* mh, const char* path, bool unloadable) { if ( strstr(path, gLibraryName) != NULL ) { gFoundLibrary = true; } }); return gFoundLibrary; } int main(int argc, const char* argv[]) { printf("[BEGIN] dyld-insert-library-rpath\n"); if (argc != 2) { printf("[FAIL] dyld-insert-library-rpath: expected library name\n"); return 0; } bool expectInsertFailure = getenv("DYLD_AMFI_FAKE") != NULL; if (wasImageLoaded(argv[1])) { // Image was loaded, but make sure that is what we wanted to happen if ( expectInsertFailure ) { printf("[FAIL] dyld-insert-library-rpath: expected insert to fail for '%s'\n", argv[1]); return 0; } } else { // Image was not loaded, so make sure we are ok with that if ( !expectInsertFailure ) { printf("[FAIL] dyld-insert-library-rpath: expected insert to pass for '%s'\n", argv[1]); return 0; } } printf("[PASS] dyld-insert-library-rpath\n"); return 0; }