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 static bool doneRegistering = false;
  35
  36 //
  37 // By returning a string, we prevent that image from loading.
  38 // We just prevent any image with "bar" in its name from loading.
  39 //
  40
  41 static const char* batchMappedHandler(enum dyld_image_states state, uint32_t infoCount, const struct dyld_image_info info[])
  42 {
  43         for (uint32_t i=0; i < infoCount; ++i) {
  44                 //fprintf(stderr, "batch mapped: %d %s\n", i, info[i].imageFilePath);
  45                 if ( strstr(info[i].imageFilePath, "bar") != NULL )
  46                         return "can't load bar";
  47         }
  48         return NULL;
  49 }
  50
  51 static const char* singleMappedHandler(enum dyld_image_states state, uint32_t infoCount, const struct dyld_image_info info[])
  52 {
  53         //fprintf(stderr, "single mapped: %s\n", info[0].imageFilePath);
  54         if ( strstr(info[0].imageFilePath, "bar") != NULL )
  55                 return "can't load bar";
  56         return NULL;
  57 }
  58
  59
  60
  61 static const char* singleInitializedHandler(enum dyld_image_states state, uint32_t infoCount, const struct dyld_image_info info[])
  62 {
  63         //fprintf(stderr, "singleInitializedHandler(%s)\n", info[0].imageFilePath);
  64         if ( doneRegistering ) {
  65                 FAIL("image-state-deny something loaded");
  66                 exit(0);
  67         }
  68         return NULL;
  69 }
  70
  71 //static const char* batchBoundHandler(enum dyld_image_states state, uint32_t infoCount, const struct dyld_image_info info[])
  72 //{
  73 //      //for (uint32_t i=0; i < infoCount; ++i)
  74 //      //      fprintf(stderr, "bound: %u %s\n", i, info[i].imageFilePath);
  75 //      return NULL;
  76 //}
  77
  78 static void load(const char* name)
  79 {
  80         NSObjectFileImage ofi;
  81         if ( NSCreateObjectFileImageFromFile(name, &ofi) == NSObjectFileImageSuccess ) {
  82                 NSModule mod = NSLinkModule(ofi, name, NSLINKMODULE_OPTION_PRIVATE | NSLINKMODULE_OPTION_RETURN_ON_ERROR);
  83                 if ( mod != NULL ) {
  84                         FAIL("neither NSCreateObjectFileImageFromFile nor NSLinkModule failed but should have");
  85                         if ( !NSDestroyObjectFileImage(ofi) ) {
  86                                 FAIL("NSDestroyObjectFileImage failed");
  87                         }
  88                         exit(0);
  89                 }
  90                 else {
  91                         NSDestroyObjectFileImage(ofi);
  92                 }
  93         }
  94 }
  95
  96
  97 int main(int argc, const char* argv[])
  98 {
  99         // tell dyld we want to know when images successfully loaded
 100         dyld_register_image_state_change_handler(dyld_image_state_initialized, false, singleInitializedHandler);
 101         doneRegistering = true;
 102
 103         // tell dyld we want to know when images successfully loaded
 104         //dyld_register_image_state_change_handler(dyld_image_state_bound, true, batchBoundHandler);
 105
 106         // tell dyld we want to know when images are mapped
 107         if ( argc > 1 )
 108                 dyld_register_image_state_change_handler(dyld_image_state_dependents_mapped, true, batchMappedHandler);
 109         else
 110                 dyld_register_image_state_change_handler(dyld_image_state_mapped, false, singleMappedHandler);
 111
 112         NSObjectFileImage ofi;
 113         if ( NSCreateObjectFileImageFromFile("foo2.bundle", &ofi) != NSObjectFileImageSuccess ) {
 114                 FAIL("NSCreateObjectFileImageFromFile failed for foo2.bundle");
 115                 exit(0);
 116         }
 117
 118         load("foo.bundle");
 119
 120         load("bar.bundle");
 121
 122         if ( !NSDestroyObjectFileImage(ofi) ) {
 123                 FAIL("NSDestroyObjectFileImage failed for foo2.bundle");
 124                 exit(0);
 125         }
 126
 127 //      dlopen("/usr/lib/libz.1.2.3.dylib", RTLD_LAZY);
 128
 129         PASS("image-state-deny");
 130
 131         return EXIT_SUCCESS;
 132 }