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 <stdio.h> 24#include <stdlib.h> 25#include <mach-o/dyld.h> 26#include <Availability.h> 27 28#include "test.h" // PASS(), FAIL() 29 30/// 31/// The point of this test is to load and unload a bundle and 32/// verify that the address->image cache is properly invalidated. rdar://problem/4212667 33/// 34 35typedef void (*fooProc)(); 36 37// test.bundle 38void doit() 39{ 40// NSCreateObjectFileImageFromMemory is only available on Mac OS X - not iPhone OS 41#if __MAC_OS_X_VERSION_MIN_REQUIRED 42 NSObjectFileImage ofi; 43 if ( NSCreateObjectFileImageFromFile("test.bundle", &ofi) != NSObjectFileImageSuccess ) { 44 FAIL("NSCreateObjectFileImageFromFile failed"); 45 exit(0); 46 } 47 48 NSModule mod = NSLinkModule(ofi,"test.bundle", NSLINKMODULE_OPTION_PRIVATE | NSLINKMODULE_OPTION_BINDNOW | NSLINKMODULE_OPTION_RETURN_ON_ERROR); 49 if ( mod == NULL ) { 50 FAIL("NSLinkModule failed"); 51 exit(0); 52 } 53 54 NSSymbol sym = NSLookupSymbolInModule(mod, "_foo"); 55 if ( sym == NULL ) { 56 FAIL("NSLookupSymbolInModule failed"); 57 exit(0); 58 } 59 60 fooProc func = (fooProc)NSAddressOfSymbol(sym); 61 (*func)(); 62 63 if ( !NSUnLinkModule(mod, NSUNLINKMODULE_OPTION_NONE) ) { 64 FAIL("NSUnLinkModule failed"); 65 exit(0); 66 } 67 68 if ( !NSDestroyObjectFileImage(ofi) ) { 69 FAIL("NSDestroyObjectFileImage failed"); 70 exit(0); 71 } 72#endif 73} 74 75 76static void myRemoveImage(const struct mach_header *mh, intptr_t vmaddr_slide) 77{ 78#if __MAC_OS_X_VERSION_MIN_REQUIRED 79 // calling _dyld_get_image_header_containing_address() during the remove image hook 80 // could cause dyld to not flush the address->image cache 81 _dyld_get_image_header_containing_address(mh); 82#endif 83} 84 85 86int main() 87{ 88 _dyld_register_func_for_remove_image(&myRemoveImage); 89 90 doit(); 91 doit(); 92 doit(); 93 94 PASS("bundle-reload"); 95 return 0; 96} 97 98