git.cworth.org Git - apitrace/blob - glproc_gl.cpp

   1 /**************************************************************************
   2  *
   3  * Copyright 2011 Jose Fonseca
   4  * All Rights Reserved.
   5  *
   6  * Permission is hereby granted, free of charge, to any person obtaining a copy
   7  * of this software and associated documentation files (the "Software"), to deal
   8  * in the Software without restriction, including without limitation the rights
   9  * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
  10  * copies of the Software, and to permit persons to whom the Software is
  11  * furnished to do so, subject to the following conditions:
  12  *
  13  * The above copyright notice and this permission notice shall be included in
  14  * all copies or substantial portions of the Software.
  15  *
  16  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  17  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  18  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  19  * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  20  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  21  * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
  22  * THE SOFTWARE.
  23  *
  24  **************************************************************************/
  25
  26
  27 #include "glproc.hpp"
  28
  29
  30 #if !defined(_WIN32)
  31 #ifndef _GNU_SOURCE
  32 #define _GNU_SOURCE // for dladdr
  33 #endif
  34 #include <dlfcn.h>
  35 #endif
  36
  37
  38 /*
  39  * Handle to the true OpenGL library.
  40  */
  41 #if defined(_WIN32)
  42 HINSTANCE __libGlHandle = NULL;
  43 #else
  44 void *__libGlHandle = NULL;
  45 #endif
  46
  47
  48
  49 #if defined(_WIN32)
  50
  51 void *
  52 __getPublicProcAddress(const char *procName)
  53 {
  54     if (!__libGlHandle) {
  55         char szDll[MAX_PATH] = {0};
  56
  57         if (!GetSystemDirectoryA(szDll, MAX_PATH)) {
  58             return NULL;
  59         }
  60
  61         strcat(szDll, "\\\\opengl32.dll");
  62
  63         __libGlHandle = LoadLibraryA(szDll);
  64         if (!__libGlHandle) {
  65             os::log("apitrace: error: couldn't load %s\n", szDll);
  66             return NULL;
  67         }
  68     }
  69
  70     return (void *)GetProcAddress(__libGlHandle, procName);
  71 }
  72
  73
  74 void *
  75 __getPrivateProcAddress(const char *procName) {
  76     return (void *)__wglGetProcAddress(procName);
  77 }
  78
  79
  80 #elif defined(__APPLE__)
  81
  82
  83 /*
  84  * Path to the true OpenGL framework
  85  */
  86 static const char *libgl_filename = "/System/Library/Frameworks/OpenGL.framework/OpenGL";
  87
  88
  89 /*
  90  * Lookup a libGL symbol
  91  */
  92 void * __libgl_sym(const char *symbol)
  93 {
  94     void *result;
  95
  96     if (!__libGlHandle) {
  97         /*
  98          * Unfortunately we can't just dlopen the true dynamic library because
  99          * DYLD_LIBRARY_PATH/DYLD_FRAMEWORK_PATH take precedence, even for
 100          * absolute paths.  So we create a temporary symlink, and dlopen that
 101          * instead.
 102          */
 103
 104         char temp_filename[] = "/tmp/tmp.XXXXXX";
 105
 106         if (mktemp(temp_filename) != NULL) {
 107             if (symlink(libgl_filename, temp_filename) == 0) {
 108                 __libGlHandle = dlopen(temp_filename, RTLD_LOCAL | RTLD_NOW | RTLD_FIRST);
 109                 remove(temp_filename);
 110             }
 111         }
 112
 113         if (!__libGlHandle) {
 114             os::log("apitrace: error: couldn't load %s\n", libgl_filename);
 115             os::abort();
 116             return NULL;
 117         }
 118     }
 119
 120     result = dlsym(__libGlHandle, symbol);
 121
 122     if (result == dlsym(RTLD_SELF, symbol)) {
 123         os::log("apitrace: error: symbol lookup recursion\n");
 124         os::abort();
 125         return NULL;
 126     }
 127
 128     return result;
 129 }
 130
 131
 132 void *
 133 __getPublicProcAddress(const char *procName)
 134 {
 135     return __libgl_sym(procName);
 136 }
 137
 138 void *
 139 __getPrivateProcAddress(const char *procName)
 140 {
 141     return __libgl_sym(procName);
 142 }
 143
 144
 145 #else
 146
 147
 148 /*
 149  * Invoke the true dlopen() function.
 150  */
 151 static void *
 152 __dlopen(const char *filename, int flag)
 153 {
 154     typedef void * (*PFNDLOPEN)(const char *, int);
 155     static PFNDLOPEN dlopen_ptr = NULL;
 156
 157     if (!dlopen_ptr) {
 158         dlopen_ptr = (PFNDLOPEN)dlsym(RTLD_NEXT, "dlopen");
 159         if (!dlopen_ptr) {
 160             os::log("apitrace: error: dlsym(RTLD_NEXT, \"dlopen\") failed\n");
 161             return NULL;
 162         }
 163     }
 164
 165     return dlopen_ptr(filename, flag);
 166 }
 167
 168
 169 /*
 170  * Lookup a libGL symbol
 171  */
 172 void * __libgl_sym(const char *symbol)
 173 {
 174     void *result;
 175
 176     if (!__libGlHandle) {
 177         /*
 178          * The app doesn't directly link against libGL.so, nor does it directly
 179          * dlopen it.  So we have to load it ourselves.
 180          */
 181
 182         const char * libgl_filename = getenv("TRACE_LIBGL");
 183
 184         if (!libgl_filename) {
 185             /*
 186              * Try to use whatever libGL.so the library is linked against.
 187              */
 188
 189             result = dlsym(RTLD_NEXT, symbol);
 190             if (result) {
 191                 __libGlHandle = RTLD_NEXT;
 192                 return result;
 193             }
 194
 195             libgl_filename = "libGL.so.1";
 196         }
 197
 198         /*
 199          * It would have been preferable to use RTLD_LOCAL to ensure that the
 200          * application can never access libGL.so symbols directly, but this
 201          * won't work, given libGL.so often loads a driver specific SO and
 202          * exposes symbols to it.
 203          */
 204
 205         __libGlHandle = __dlopen(libgl_filename, RTLD_GLOBAL | RTLD_LAZY);
 206         if (!__libGlHandle) {
 207             os::log("apitrace: error: couldn't find libGL.so\n");
 208             return NULL;
 209         }
 210     }
 211
 212     return dlsym(__libGlHandle, symbol);
 213 }
 214
 215
 216 void *
 217 __getPublicProcAddress(const char *procName)
 218 {
 219     return __libgl_sym(procName);
 220 }
 221
 222 void *
 223 __getPrivateProcAddress(const char *procName)
 224 {
 225     return (void *)__glXGetProcAddressARB((const GLubyte *)procName);
 226 }
 227
 228
 229 #endif
 230