unit-tests/test-cases/dlopen-RTLD_FIRST/main.c

   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
  27 #include "test.h" // PASS(), FAIL(), XPASS(), XFAIL()
  28
  29
  30 ///
  31 /// When dlopen() is called with the RTLD_FIRST option,
  32 /// any dlsym() looks against that handle should only look in
  33 /// that handle, and not subsequent images.
  34 ///
  35
  36
  37 struct info
  38 {
  39         const char* path;
  40         void*           handle;
  41 };
  42 typedef struct info info;
  43
  44 static info dlopen_or_fail(const char* path, int options)
  45 {
  46         info result;
  47         result.path = path;
  48         result.handle = dlopen(path, options);
  49         if ( result.handle == NULL ) {
  50                 FAIL("dlopen-RTLD_FIRST: dlopen(\"%s\") failed: %s", path, dlerror());
  51                 exit(0);
  52         }
  53         //fprintf(stderr, "dlopen(%s, 0x%0X) => %p\n", path, options, result.handle);
  54         return result;
  55 }
  56
  57 static void dlsym_should_fail(info hp, const char* symbol)
  58 {
  59         void* sym = dlsym(hp.handle, symbol);
  60         if ( sym != NULL ) {
  61                 FAIL("dlopen-RTLD_FIRST: dlsym(handle-%s, \"%s\") should have failed", hp.path, symbol);
  62                 exit(0);
  63         }
  64 }
  65
  66 static void dlsym_should_succeed(info hp, const char* symbol)
  67 {
  68         void* sym = dlsym(hp.handle, symbol);
  69         if ( sym == NULL ) {
  70                 FAIL("dlopen-RTLD_FIRST: dlsym(handle-%s, \"%s\") failed", hp.path, symbol);
  71                 exit(0);
  72         }
  73 }
  74
  75
  76 int main()
  77 {
  78         info libFooFirst = dlopen_or_fail("libfoo.dylib", RTLD_FIRST);
  79         info libFoo      = dlopen_or_fail("libfoo.dylib", 0);
  80
  81         info libBarFirst = dlopen_or_fail("libbar.dylib", RTLD_FIRST);
  82         info libBar      = dlopen_or_fail("libbar.dylib", 0);
  83
  84         dlsym_should_succeed(libFooFirst, "foo");
  85         dlsym_should_fail(libFooFirst, "base");
  86         dlsym_should_fail(libFooFirst, "bar");
  87
  88         dlsym_should_succeed(libFoo, "foo");
  89         dlsym_should_succeed(libFoo, "base");
  90
  91         dlsym_should_succeed(libBarFirst, "bar");
  92         dlsym_should_succeed(libBarFirst, "base");      // libbar re-exports libbase
  93         dlsym_should_fail(libBarFirst, "foo");
  94
  95         dlsym_should_succeed(libBar, "bar");
  96         dlsym_should_succeed(libBar, "base");
  97
  98
  99
 100         info bundleFooFirst = dlopen_or_fail("foo.bundle", RTLD_FIRST);
 101         info bundleFoo      = dlopen_or_fail("foo.bundle", 0);
 102
 103         info bundleBarFirst = dlopen_or_fail("bar.bundle", RTLD_FIRST);
 104         info bundleBar      = dlopen_or_fail("bar.bundle", 0);
 105
 106         dlsym_should_succeed(bundleFooFirst, "foo");
 107         dlsym_should_fail(bundleFooFirst, "base");
 108         dlsym_should_fail(bundleFooFirst, "bar");
 109
 110         dlsym_should_succeed(bundleFoo, "foo");
 111         dlsym_should_succeed(bundleFoo, "base");
 112
 113         dlsym_should_succeed(bundleBarFirst, "bar");
 114         dlsym_should_fail(bundleBarFirst, "base");
 115         dlsym_should_fail(bundleBarFirst, "foo");
 116
 117         dlsym_should_succeed(bundleBar, "bar");
 118         dlsym_should_succeed(bundleBar, "base");
 119
 120
 121         PASS("dlsym-RTLD_FIRST bundle and dylib");
 122         return EXIT_SUCCESS;
 123 }