1/* 2 * Copyright (c) 2014 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 <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_priv.h> 30#include <dlfcn.h> 31 32#include "test.h" // PASS(), FAIL(), XPASS(), XFAIL() 33 34 35// 36// By returning a string, we prevent that image from loading. 37// We just prevent any image with "libz" in its name from loading. 38// 39static const char* singleMappedHandler(enum dyld_image_states state, uint32_t infoCount, const struct dyld_image_info info[]) 40{ 41 //fprintf(stderr, "single mapped: %s\n", info[0].imageFilePath); 42 if ( strstr(info[0].imageFilePath, "libz") != NULL ) 43 return "can't load libz"; 44 return NULL; 45} 46 47 48int main(int argc, const char* argv[]) 49{ 50 // tell dyld we want to know when images are mapped 51 dyld_register_image_state_change_handler(dyld_image_state_mapped, false, singleMappedHandler); 52 53 void* handle = dlopen("libfoo1.dylib", RTLD_LAZY); 54 if ( handle != NULL ) { 55 FAIL("image-state-deny-cache-leak: dlopen(libfoo.dylib) should have failed"); 56 return EXIT_SUCCESS; 57 } 58 59 handle = dlopen("libfoo2.dylib", RTLD_LAZY); 60 if ( handle == NULL ) { 61 FAIL("image-state-deny-cache-leak: dlopen(libfoo.dylib) should have succeeded"); 62 return EXIT_SUCCESS; 63 } 64 dlclose(handle); 65 66 // execute leaks command on myself 67 char cmd[512]; 68 sprintf(cmd, "sudo leaks %u > /dev/null\n", getpid()); 69 int result = system(cmd); 70 if ( result == EXIT_SUCCESS ) 71 PASS("image-state-deny-cache-leak"); 72 else 73 FAIL("image-state-deny-cache-leak"); 74 75 76 return EXIT_SUCCESS; 77} 78