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_images.h> 30#include <mach-o/dyld_priv.h> 31#include <dlfcn.h> 32 33#include "test.h" // PASS(), FAIL(), XPASS(), XFAIL() 34 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 42static const char* singleMappedHandler(enum dyld_image_states state, uint32_t infoCount, const struct dyld_image_info info[]) 43{ 44 //fprintf(stderr, "single mapped: %s\n", info[0].imageFilePath); 45 if ( strstr(info[0].imageFilePath, "bar") != NULL ) 46 return "can't load bar"; 47 return NULL; 48} 49 50 51static void load(const char* name) 52{ 53 void* handle = dlopen(name, RTLD_LAZY); 54 if ( handle != NULL ) { 55 FAIL("image-state-deny-all_image_infos: dlopen(%s) should have failed", name); 56 exit(0); 57 } 58} 59 60 61int main(int argc, const char* argv[]) 62{ 63 // tell dyld we want to know when images are mapped 64 dyld_register_image_state_change_handler(dyld_image_state_mapped, false, singleMappedHandler); 65 66 const struct dyld_all_image_infos* infos = _dyld_get_all_image_infos(); 67 const uint32_t initialCount = infos->infoArrayCount; 68 69 load("bar.bundle"); 70 71 if ( infos->infoArrayCount != initialCount ){ 72 FAIL("image-state-deny-all_image_infos: infoArrayCount should not have changed"); 73 exit(0); 74 } 75 76 //for (int i=0; i < infos->infoArrayCount; ++i) 77 // printf("mh=%p, path=%s\n", infos->infoArray[i].imageLoadAddress, infos->infoArray[i].imageFilePath); 78 79 PASS("image-state-deny-all_image_infos"); 80 81 return EXIT_SUCCESS; 82} 83