unit-tests/test-cases/image-state-deny-OFI/main.c

   1 /*
   2  * Copyright (c) 2005 Apple Computer, Inc. All rights reserved.
   3  *
   4  * @APPLE_LICENSE_HEADER_START@
   5  *
   6  * This file contains Original Code and/or Modifications of Original Code
   7  * as defined in and that are subject to the Apple Public Source License
   8  * Version 2.0 (the 'License'). You may not use this file except in
   9  * compliance with the License. Please obtain a copy of the License at
  10  * http://www.opensource.apple.com/apsl/ and read it before using this
  11  * file.
  12  *
  13  * The Original Code and all software distributed under the License are
  14  * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
  15  * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
  16  * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY,
  17  * FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT.
  18  * Please see the License for the specific language governing rights and
  19  * limitations under the License.
  20  *
  21  * @APPLE_LICENSE_HEADER_END@
  22  */
  23 #include <stdbool.h>  // fprintf(), NULL
  24 #include <stdio.h>  // fprintf(), NULL
  25 #include <stdlib.h> // exit(), EXIT_SUCCESS
  26 #include <stdbool.h>
  27 #include <string.h>
  28 #include <mach-o/dyld.h>
  29 #include <mach-o/dyld_priv.h>
  30 #include <dlfcn.h>
  31
  32 #include "test.h" // PASS(), FAIL(), XPASS(), XFAIL()
  33
  34
  35 #if __MAC_OS_X_VERSION_MIN_REQUIRED
  36
  37 static bool doneRegistering = false;
  38
  39 //
  40 // By returning a string, we prevent that image from loading.
  41 // We just prevent any image with "bar" in its name from loading.
  42 //
  43
  44 static const char* batchMappedHandler(enum dyld_image_states state, uint32_t infoCount, const struct dyld_image_info info[])
  45 {
  46         for (uint32_t i=0; i < infoCount; ++i) {
  47                 //fprintf(stderr, "batch mapped: %d %s\n", i, info[i].imageFilePath);
  48                 if ( strstr(info[i].imageFilePath, "bar") != NULL )
  49                         return "can't load bar";
  50         }
  51         return NULL;
  52 }
  53
  54 static const char* singleMappedHandler(enum dyld_image_states state, uint32_t infoCount, const struct dyld_image_info info[])
  55 {
  56         //fprintf(stderr, "single mapped: %s\n", info[0].imageFilePath);
  57         if ( strstr(info[0].imageFilePath, "bar") != NULL )
  58                 return "can't load bar";
  59         return NULL;
  60 }
  61
  62
  63
  64 static const char* singleInitializedHandler(enum dyld_image_states state, uint32_t infoCount, const struct dyld_image_info info[])
  65 {
  66         //fprintf(stderr, "singleInitializedHandler(%s)\n", info[0].imageFilePath);
  67         if ( doneRegistering ) {
  68                 FAIL("image-state-deny something loaded");
  69                 exit(0);
  70         }
  71         return NULL;
  72 }
  73
  74 //static const char* batchBoundHandler(enum dyld_image_states state, uint32_t infoCount, const struct dyld_image_info info[])
  75 //{
  76 //      //for (uint32_t i=0; i < infoCount; ++i)
  77 //      //      fprintf(stderr, "bound: %u %s\n", i, info[i].imageFilePath);
  78 //      return NULL;
  79 //}
  80
  81 static void load(const char* name)
  82 {
  83         NSObjectFileImage ofi;
  84         if ( NSCreateObjectFileImageFromFile(name, &ofi) == NSObjectFileImageSuccess ) {
  85                 NSModule mod = NSLinkModule(ofi, name, NSLINKMODULE_OPTION_PRIVATE | NSLINKMODULE_OPTION_RETURN_ON_ERROR);
  86                 if ( mod != NULL ) {
  87                         FAIL("neither NSCreateObjectFileImageFromFile nor NSLinkModule failed but should have");
  88                         if ( !NSDestroyObjectFileImage(ofi) ) {
  89                                 FAIL("NSDestroyObjectFileImage failed");
  90                         }
  91                         exit(0);
  92                 }
  93                 else {
  94                         NSDestroyObjectFileImage(ofi);
  95                 }
  96         }
  97 }
  98 #endif
  99
 100
 101 int main(int argc, const char* argv[])
 102 {
 103 // NSCreateObjectFileImageFromMemory is only available on Mac OS X - not iPhone OS
 104 #if __MAC_OS_X_VERSION_MIN_REQUIRED
 105         // tell dyld we want to know when images successfully loaded
 106         dyld_register_image_state_change_handler(dyld_image_state_initialized, false, singleInitializedHandler);
 107         doneRegistering = true;
 108
 109         // tell dyld we want to know when images successfully loaded
 110         //dyld_register_image_state_change_handler(dyld_image_state_bound, true, batchBoundHandler);
 111
 112         // tell dyld we want to know when images are mapped
 113         if ( argc > 1 )
 114                 dyld_register_image_state_change_handler(dyld_image_state_dependents_mapped, true, batchMappedHandler);
 115         else
 116                 dyld_register_image_state_change_handler(dyld_image_state_mapped, false, singleMappedHandler);
 117
 118         NSObjectFileImage ofi;
 119         if ( NSCreateObjectFileImageFromFile("foo2.bundle", &ofi) != NSObjectFileImageSuccess ) {
 120                 FAIL("NSCreateObjectFileImageFromFile failed for foo2.bundle");
 121                 exit(0);
 122         }
 123
 124         load("foo.bundle");
 125
 126         load("bar.bundle");
 127
 128         if ( !NSDestroyObjectFileImage(ofi) ) {
 129                 FAIL("NSDestroyObjectFileImage failed for foo2.bundle");
 130                 exit(0);
 131         }
 132
 133 //      dlopen("/usr/lib/libz.1.2.3.dylib", RTLD_LAZY);
 134 #endif
 135
 136         PASS("image-state-deny");
 137
 138         return EXIT_SUCCESS;
 139 }