Open Menu / testing / test-cases / dyld-insert-library-double.dtest / main.cpp
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

// BOOT_ARGS: dyld_flags=2

// BUILD:  $CXX main.cpp -std=c++11 -o $BUILD_DIR/double_insert_main.exe
// BUILD:  $CC foo.c -dynamiclib -install_name $RUN_DIR/libfoo.dylib -o $BUILD_DIR/libfoo.dylib
// BUILD:  $CC bar.c -dynamiclib -install_name $RUN_DIR/libbar.dylib -o $BUILD_DIR/libbar.dylib
// BUILD:  $DYLD_ENV_VARS_ENABLE $BUILD_DIR/double_insert_main.exe

// Load foo and bar at the same time to ensure that paths separated by ':'' are working

// RUN:  DYLD_INSERT_LIBRARIES="$RUN_DIR/libfoo.dylib:$RUN_DIR/libbar.dylib" ./double_insert_main.exe

#include <stdlib.h>
#include <stdio.h>
#include <stdbool.h>
#include <string.h>
#include <mach-o/dyld_priv.h>

#include "test_support.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;
		}
	});
    if (!gFoundLibrary)
        FAIL("Expected insert to pass for '%s'", libraryName);
	return gFoundLibrary;
}

int main(int argc, const char* argv[], const char* envp[], const char* apple[]) {
    if (!wasImageLoaded("libfoo.dylib") || !wasImageLoaded("libbar.dylib")) {
        return 0;
    }

    PASS("Success");
}