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> // fprintf(), NULL 24#include <stdlib.h> // exit(), EXIT_SUCCESS 25#include <stdbool.h> 26#include <mach-o/dyld.h> 27#include <dlfcn.h> 28 29#include "test.h" // PASS(), FAIL(), XPASS(), XFAIL() 30 31 32static void* sExpectedBundleBaseAddress = 0; 33static bool sNotified = false; 34 35static void removeNotification(const struct mach_header *mh, intptr_t vmaddr_slide) 36{ 37 if ( mh == sExpectedBundleBaseAddress ) 38 sNotified = true; 39} 40 41 42 43int main(int argc, const char* argv[]) 44{ 45 // tell dyld we want to know when images go away 46 _dyld_register_func_for_remove_image(removeNotification); 47 48 void* handle = dlopen("foo.bundle", RTLD_LAZY); 49 if ( handle == NULL ) { 50 FAIL("image-remove-notification: dlopen(foo.bundle) failed: %s", dlerror()); 51 exit(0); 52 } 53 54 void* sym = dlsym(handle, "foo"); 55 if ( sym == NULL ) { 56 FAIL("image-remove-notification: dlsym(handle, \"foo\") failed: %s", dlerror()); 57 exit(0); 58 } 59 60 Dl_info info; 61 if ( dladdr(sym, &info) == 0 ) { 62 FAIL("image-remove-notification: dladdr() failed"); 63 exit(0); 64 } 65 66 // record the base address we expect to get notification about 67 sExpectedBundleBaseAddress = info.dli_fbase; 68 69 int result = dlclose(handle); 70 if ( result != 0 ) { 71 FAIL("image-remove-notification: dlclose(handle) returned %d, %s", result, dlerror()); 72 exit(0); 73 } 74 75 if ( sNotified ) 76 PASS("image-remove-notification"); 77 else 78 FAIL("image-remove-notification"); 79 80 return EXIT_SUCCESS; 81} 82