1/* 2 * Copyright (c) 2007-2008 Apple 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 <dlfcn.h> 26#include <unistd.h> 27#include <stdlib.h> 28#include <string.h> 29#include <mach-o/dyld.h> 30#include <mach-o/dyld_priv.h> 31#include <dlfcn.h> 32 33#include "test.h" 34 35static bool doCheck = false; 36 37static const char* batchMappedHandler(enum dyld_image_states state, uint32_t infoCount, const struct dyld_image_info info[]) 38{ 39 if ( doCheck ) 40 return "can't load"; 41 return NULL; 42} 43 44 45int main() 46{ 47 // tell dyld we want to know when images are mapped 48 dyld_register_image_state_change_handler(dyld_image_state_mapped, false, batchMappedHandler); 49 doCheck = true; 50 51 for (int i=0; i < 10; ++i) { 52 dlopen_preflight("libfoo.dylib"); 53 } 54 55 // execute leaks command on myself 56 char cmd[512]; 57 sprintf(cmd, "sudo leaks %u > /dev/null\n", getpid()); 58 int result = system(cmd); 59 if ( result == EXIT_SUCCESS ) 60 PASS("dlopen_preflight-leak-image-deny-single"); 61 else 62 FAIL("dlopen_preflight-leak-image-deny-single"); 63 64 return EXIT_SUCCESS; 65} 66