]> git.saurik.com Git - apple/dyld.git/blob - testing/test-cases/dyld_abort_payload.dtest/main.c
dyld-733.8.tar.gz
[apple/dyld.git] / testing / test-cases / dyld_abort_payload.dtest / main.c
1
2 // BUILD: $CC foo.c -dynamiclib -install_name /cant/find/me.dylib -o $BUILD_DIR/libmissing.dylib
3 // BUILD: $CC foo.c -dynamiclib $BUILD_DIR/libmissing.dylib -install_name $RUN_DIR/libMissingDylib.dylib -o $BUILD_DIR/libMissingDylib.dylib
4 // BUILD: $CC emptyMain.c $BUILD_DIR/libMissingDylib.dylib -o $BUILD_DIR/prog_missing_dylib.exe
5 // BUILD: $CC defSymbol.c -dynamiclib -install_name $RUN_DIR/libMissingSymbols.dylib -o $BUILD_DIR/libMissingSymbols.dylib
6 // BUILD: $CC defSymbol.c -dynamiclib -install_name $RUN_DIR/libMissingSymbols.dylib -o $BUILD_DIR/libHasSymbols.dylib -DHAS_SYMBOL
7 // BUILD: $CC useSymbol.c $BUILD_DIR/libHasSymbols.dylib -o $BUILD_DIR/prog_missing_symbol.exe
8 // BUILD: $CC main.c -o $BUILD_DIR/dyld_abort_tests.exe
9
10 // NO_CRASH_LOG: prog_missing_dylib.exe
11 // NO_CRASH_LOG: prog_missing_symbol.exe
12
13 // RUN: ./dyld_abort_tests.exe
14
15 #include <stdio.h>
16 #include <stdlib.h>
17 #include <dlfcn.h>
18 #include <unistd.h>
19 #include <signal.h>
20 #include <errno.h>
21 #include <mach/mach.h>
22 #include <mach/machine.h>
23 #include <err.h>
24 #include <System/sys/reason.h>
25 #include <System/sys/proc_info.h>
26 #include <System/kern/kern_cdata.h>
27 #include <libproc.h>
28 #include <mach-o/dyld_priv.h>
29
30
31 static bool sSignalCaught = false;
32 static bool sChildAbortInfoCorrect = false;
33 static pid_t sChildPid = 0;
34 static uint64_t sExpectedDyldReason = 0;
35 static const char* sExpectedDylibPath = NULL;
36 static const char* sExpectedSymbol = NULL;
37
38
39 static void childDied(int sig)
40 {
41 sSignalCaught = true;
42 //printf("sigchld for pid=%d\n", sChildPid);
43
44 struct proc_exitreasoninfo info;
45 bzero(&info, sizeof(info));
46 uint8_t packReasonData[OS_REASON_BUFFER_MAX_SIZE];
47 bzero(packReasonData, OS_REASON_BUFFER_MAX_SIZE);
48 info.eri_reason_buf_size = OS_REASON_BUFFER_MAX_SIZE;
49 info.eri_kcd_buf = (user_addr_t)packReasonData;
50 //fprintf(stderr, "info=%p\n", &info);
51 int procResult = proc_pidinfo(sChildPid, PROC_PIDEXITREASONINFO, 1, &info, PROC_PIDEXITREASONINFO_SIZE);
52 if ( procResult != sizeof(struct proc_exitreasoninfo) ) {
53 printf("bad return size from proc_pidinfo(), %d expected %lu\n", procResult, PROC_PIDEXITREASONINFO_SIZE);
54 return;
55 }
56 if ( info.eri_namespace != OS_REASON_DYLD ) {
57 printf("eri_namespace (%d) != OS_REASON_DYLD\n", info.eri_namespace);
58 return;
59 }
60 if ( info.eri_code != sExpectedDyldReason ) {
61 printf("eri_code (%llu) != %lld\n", sExpectedDyldReason, info.eri_code);
62 return;
63 }
64 kcdata_iter_t iter = kcdata_iter(packReasonData, info.eri_reason_buf_size);
65
66 if ( !kcdata_iter_valid(iter) ) {
67 printf("invalid kcdata iterator from payload data\n");
68 return;
69 }
70
71 if ( kcdata_iter_type(iter) != KCDATA_BUFFER_BEGIN_OS_REASON ){
72 printf("first kcdata from payload data is not KCDATA_BUFFER_BEGIN_OS_REASON\n");
73 return;
74 }
75
76 kcdata_iter_t payloadIter = kcdata_iter_find_type(iter, EXIT_REASON_USER_PAYLOAD);
77 if ( !kcdata_iter_valid(payloadIter) ) {
78 printf("invalid kcdata payload iterator from payload data\n");
79 return;
80 }
81 const dyld_abort_payload* dyldInfo = (dyld_abort_payload*)kcdata_iter_payload(payloadIter);
82
83 if ( dyldInfo->version != 1 ) {
84 printf("dyld payload is not version 1\n");
85 return;
86 }
87
88 if ( (dyldInfo->flags & 1) == 0 ) {
89 printf("dyld flags should have low bit set to me process terminated at launch\n");
90 return;
91 }
92
93 if ( sExpectedDylibPath != NULL ) {
94 if ( dyldInfo->targetDylibPathOffset != 0 ) {
95 const char* targetDylib = (char*)dyldInfo + dyldInfo->targetDylibPathOffset;
96 if ( strstr(targetDylib, sExpectedDylibPath) == NULL ) {
97 printf("dylib path (%s) not what expected (%s)\n", targetDylib, sExpectedDylibPath);
98 return;
99 }
100 }
101 else {
102 printf("dylib path (%s) not provided by dyld\n", sExpectedDylibPath);
103 return;
104 }
105 }
106
107 if ( sExpectedSymbol != NULL ) {
108 if ( dyldInfo->targetDylibPathOffset != 0 ) {
109 const char* missingSymbol = (char*)dyldInfo + dyldInfo->symbolOffset;
110 if ( strcmp(sExpectedSymbol, missingSymbol) != 0 ) {
111 printf("symbol (%s) not what expected (%s)\n", missingSymbol, sExpectedSymbol);
112 return;
113 }
114 }
115 else {
116 printf("symbol (%s) not provided by dyld\n", sExpectedSymbol);
117 return;
118 }
119 }
120
121 sChildAbortInfoCorrect = true;
122 }
123
124
125 bool runTest(const char* prog, uint64_t dyldReason, const char* expectedDylibPath, const char* expectedSymbol)
126 {
127 sSignalCaught = false;
128 sChildAbortInfoCorrect = false;
129 sExpectedDyldReason = dyldReason;
130 sExpectedDylibPath = expectedDylibPath;
131 sExpectedSymbol = expectedSymbol;
132
133 // fork and exec child
134 sChildPid = fork();
135 if ( sChildPid < 0 )
136 err(EXIT_FAILURE, "fork");
137 if ( sChildPid == 0 ) {
138 // child side
139 char* childArgv[] = { (char*)prog, NULL };
140 int result = execvp(prog, childArgv);
141 err(EXIT_FAILURE, "exec(\"%s\",...)", prog);
142 }
143 for(int i=0; i < 10; ++i) {
144 if ( sSignalCaught )
145 break;
146 sleep(1);
147 }
148
149 return sChildAbortInfoCorrect;
150 }
151
152
153 int main(int argc, const char* argv[])
154 {
155 bool someTestFailed = false;
156 printf("[BEGIN] dyld_abort_payload\n");
157
158 // set up signal handler for catching child terminations
159 signal(SIGCHLD, childDied);
160
161 // test launch program with missing library
162 if ( !runTest("./prog_missing_dylib.exe", DYLD_EXIT_REASON_DYLIB_MISSING, "/cant/find/me.dylib", NULL) ) {
163 printf("[FAIL] dyld_abort_payload DYLD_EXIT_REASON_DYLIB_MISSING\n");
164 someTestFailed = true;
165 }
166
167 // test launch program with missing symbol
168 if ( !runTest("./prog_missing_symbol.exe", DYLD_EXIT_REASON_SYMBOL_MISSING, "libMissingSymbols.dylib", "_slipperySymbol") ) {
169 printf("[FAIL] dyld_abort_payload DYLD_EXIT_REASON_SYMBOL_MISSING\n");
170 someTestFailed = true;
171 }
172
173 if ( !someTestFailed )
174 printf("[PASS] dyld_abort_payload\n");
175
176 return 0;
177 }
178