dyld-353.2.1.tar.gz

[apple/dyld.git] / unit-tests / test-cases / image-state-change / main.c

diff --git a/unit-tests/test-cases/image-state-change/main.c b/unit-tests/test-cases/image-state-change/main.c
index 0a6673d8f0ffc2bb265131198cfed6bf94e212ca..6ee6a622cba5897f8cd04f05534823d537133a6e 100644 (file)
--- a/unit-tests/test-cases/image-state-change/main.c
+++ b/unit-tests/test-cases/image-state-change/main.c
@@ -34,7 +34,7 @@
 
 static int singleMappedCount = 0;
 static int batchMappedCount = 0;
-
+static int singleUnMappedCount = 0;
 
 
 static const char* batchMappedHandler(enum dyld_image_states state, uint32_t infoCount, const struct dyld_image_info info[])
@@ -64,6 +64,21 @@ static const char* singleMappedHandler(enum dyld_image_states state, uint32_t in
        return NULL;
 }
 
+static const char* singleUnmappedHandler(enum dyld_image_states state, uint32_t infoCount, const struct dyld_image_info info[])
+{
+       printf("singleUnmappedHandler(%s)\n", info[0].imageFilePath);
+       if ( state != dyld_image_state_terminated ) {
+               FAIL("image-state-change: singleUnmappedHandler passed state %d", state);
+               exit(0);
+       }
+       if ( infoCount != 1 ) {
+               FAIL("image-state-change: singleUnmappedHandler given %d images", infoCount);
+               exit(0);
+       }
+       ++singleUnMappedCount;
+       return NULL;
+}
+
 static void loadAndUnLoad()
 {
        void* handle = dlopen("foo.bundle", RTLD_LAZY);
@@ -85,6 +100,7 @@ int main(int argc, const char* argv[])
        // tell dyld we want to know when images are mapped
        dyld_register_image_state_change_handler(dyld_image_state_dependents_mapped, true, batchMappedHandler);
        dyld_register_image_state_change_handler(dyld_image_state_mapped, false, singleMappedHandler);
+       dyld_register_image_state_change_handler(dyld_image_state_terminated, false, singleUnmappedHandler);
        // with batch mode we get notified of existing images, but not with single mode, so re-sync counts
        batchMappedCount=0;
        
@@ -92,7 +108,7 @@ int main(int argc, const char* argv[])
 
        loadAndUnLoad();
        
-       if ( singleMappedCount == batchMappedCount )
+       if ( (singleMappedCount == batchMappedCount) && (singleMappedCount == singleUnMappedCount) )
                PASS("image-state-change");
        else
                FAIL("image-state-change: batch count=%d, single count=%d", batchMappedCount, singleMappedCount);