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 <dlfcn.h> 26#include <mach-o/dyld.h> 27#include <set> 28 29#include "test.h" // PASS(), FAIL(), XPASS(), XFAIL() 30 31 32static void trySO(const char* path) 33{ 34 void* handle = dlopen(path, RTLD_LAZY); 35 if ( handle == NULL ) { 36 FAIL("dlopen(\"%s\") failed with: %s", path, dlerror()); 37 exit(0); 38 } 39 40 void* sym = dlsym(handle, "foo"); 41 if ( sym == NULL ) { 42 FAIL("dlsym(handle, \"foo\") failed"); 43 exit(0); 44 } 45 46 int result = dlclose(handle); 47 if ( result != 0 ) { 48 if ( result == 1 ) { 49 // panther dyld returns 1 if you try to dlclose() a dylib 50 XFAIL("dlclose(handle) returned %d", result); 51 } 52 else { 53 FAIL("dlclose(handle) returned %d", result); 54 exit(0); 55 } 56 } 57 58} 59 60static std::set<const struct mach_header*> sCurrentImages; 61 62static void notify(const struct mach_header *mh, intptr_t vmaddr_slide) 63{ 64 //fprintf(stderr, "mh=%p\n", mh); 65 if ( sCurrentImages.count(mh) != 0 ) { 66 FAIL("notified twice about %p", mh); 67 exit(0); 68 } 69 sCurrentImages.insert(mh); 70} 71 72 73 74int main() 75{ 76 _dyld_register_func_for_add_image(¬ify); 77 78 trySO("libfoo.dylib"); 79 80 PASS("dlopen-init-dlopen-notify"); 81 return EXIT_SUCCESS; 82} 83