unit-tests/test-cases/dlopen-from-anonymous-code/main.c

   1 /*
   2  * Copyright (c) 2006-2007 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>  // fprintf(), NULL
  24 #include <string.h> // memcpy
  25 #include <stdlib.h> // exit(), EXIT_SUCCESS
  26 #include <dlfcn.h>
  27 #include <libkern/OSCacheControl.h> // sys_icache_invalidate
  28 #include <sys/mman.h> // for mprotext
  29
  30 #include "test.h" // PASS(), FAIL(), XPASS(), XFAIL()
  31
  32
  33
  34 void* calldlopen(const char* path, int mode, void* (*dlopen_proc)(const char* path, int mode))
  35 {
  36         return (*dlopen_proc)(path, mode);
  37 }
  38
  39 //
  40 // try calling dlopen() from code not owned by dyld
  41 //
  42 int main()
  43 {
  44         void* codeBlock = malloc(4096);
  45         memcpy(codeBlock, &calldlopen, 4096);
  46         sys_icache_invalidate(codeBlock, 4096);
  47         mprotect(codeBlock, 4096, PROT_READ | PROT_EXEC);
  48         //fprintf(stderr, "codeBlock=%p\n", codeBlock);
  49
  50         void* (*caller)(const char* path, int mode, void* (*dlopen_proc)(const char* path, int mode)) = codeBlock;
  51
  52         void* handle = (*caller)("foo.bundle", RTLD_LAZY, &dlopen);
  53         if ( handle == NULL ) {
  54                 FAIL("dlopen(\"%s\") failed with: %s", "foo.bundle", dlerror());
  55                 exit(0);
  56         }
  57
  58         void* sym = dlsym(handle, "foo");
  59         if ( sym == NULL ) {
  60                 FAIL("dlsym(handle, \"foo\") failed");
  61                 exit(0);
  62         }
  63
  64         int result = dlclose(handle);
  65         if ( result != 0 ) {
  66                 FAIL("dlclose(handle) returned %d", result);
  67                 exit(0);
  68         }
  69
  70         PASS("dlopen-from-anonymous-code");
  71         return EXIT_SUCCESS;
  72 }