X-Git-Url: https://git.saurik.com/apple/dyld.git/blobdiff_plain/dd1e3476105179eb3efd5ebd2af97f6de91170b3..16b475fcb248267b8b51f759bc62a49ec2afa88d:/testing/test-cases/dyld_process_info_unload.dtest/main.cpp

diff --git a/testing/test-cases/dyld_process_info_unload.dtest/main.cpp b/testing/test-cases/dyld_process_info_unload.dtest/main.cpp
new file mode 100644
index 0000000..961ed8a
--- /dev/null
+++ b/testing/test-cases/dyld_process_info_unload.dtest/main.cpp
@@ -0,0 +1,60 @@
+
+// BUILD:  $CC target.c       -o $BUILD_DIR/target.exe
+// BUILD:  $CC foo.c          -o $BUILD_DIR/libfoo.dylib -dynamiclib -install_name $RUN_DIR/libfoo.dylib
+// BUILD:  $CXX main.cpp         -o $BUILD_DIR/dyld_process_info_unload.exe -DRUN_DIR="$RUN_DIR"
+// BUILD:  $TASK_FOR_PID_ENABLE  $BUILD_DIR/dyld_process_info_unload.exe
+
+// RUN:  $SUDO ./dyld_process_info_unload.exe $RUN_DIR/target.exe
+
+#include <stdio.h>
+#include <stdlib.h>
+#include <dlfcn.h>
+#include <unistd.h>
+#include <signal.h>
+#include <spawn.h>
+#include <errno.h>
+#include <mach/mach.h>
+#include <mach/machine.h>
+#include <mach-o/dyld_process_info.h>
+
+#include "test_support.h"
+
+int main(int argc, const char* argv[], const char* envp[], const char* apple[]) {
+    _process process;
+    process.set_executable_path(RUN_DIR "/target.exe");
+    process.set_launch_suspended(true);
+    process.set_launch_async(true);
+    const char* env[] = { "TEST_OUTPUT=None", NULL};
+    process.set_env(env);
+    pid_t pid = process.launch();
+    task_t task;
+    if (task_for_pid(mach_task_self(), pid, &task) != KERN_SUCCESS) {
+        FAIL("task_for_pid() failed");
+    }
+
+    dispatch_async(dispatch_get_main_queue(), ^{
+        int failCount = 0;
+        for (int i=0; i < 100; ++i ) {
+            kern_return_t result;
+            dyld_process_info info = _dyld_process_info_create(task, 0, &result);
+            LOG("info=%p, result=%08X", info, result);
+            if ( i == 0 )
+                (void)kill(pid, SIGCONT);
+            if ( info == NULL ) {
+                //FIXME: Compact info will never fail, make this a FAIL()
+                failCount++;
+                // ideally the fail count would be zero.  But the target is dlopen/dlclosing in a tight loop, so there may never be a stable set of images.
+                // The real bug driving this test case was _dyld_process_info_create() crashing when the the image list changed too fast.
+                // The important thing is to not crash.  Getting NULL back is ok.
+                LOG("info=%p, result=%08X", info, result);
+            }
+            else {
+                usleep(100);
+                _dyld_process_info_release(info);
+            }
+        }
+        PASS("Success");
+    });
+
+    dispatch_main();
+}