unit-tests/test-cases/dlopen-NULL-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(NULL, RTLD_FIRST) is called,
  32 /// any dlsym() looks against that handle should only look in
  33 /// the main executable, 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("dlsym-NULL-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("dlsym-NULL-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("dlsym-NULL-RTLD_FIRST: dlsym(handle-%s, \"%s\") failed", hp.path, symbol);
  71                 exit(0);
  72         }
  73 }
  74
  75
  76 int main()
  77 {
  78         int result;
  79         info mainFirst          = dlopen_or_fail(NULL, RTLD_FIRST);
  80         info mainDefault        = dlopen_or_fail(NULL, 0);
  81
  82         dlsym_should_succeed(mainFirst, "main_foo");
  83         dlsym_should_fail(mainFirst, "foo");
  84
  85         dlsym_should_succeed(mainDefault, "main_foo");
  86         dlsym_should_succeed(mainDefault, "foo");
  87
  88         result = dlclose(mainFirst.handle);
  89         if ( result != 0 ) {
  90                 FAIL("dlsym-NULL-RTLD_FIRST: dlclose(mainFirst.handle) failed: %s", dlerror());
  91                 exit(0);
  92         }
  93         result = dlclose(mainDefault.handle);
  94         if ( result != 0 ) {
  95                 FAIL("dlsym-NULL-RTLD_FIRST: dlclose(mainDefault.handle) failed: %s", dlerror());
  96                 exit(0);
  97         }
  98
  99         PASS("dlsym-NULL-RTLD_FIRST");
 100         return EXIT_SUCCESS;
 101 }
 102
 103
 104 void main_foo() {}
 105
 106