git.cworth.org Git - glaze/blob - glaze.c

   1 /* Copyright © 2013, Intel Corporation
   2  *
   3  * Permission is hereby granted, free of charge, to any person obtaining a copy
   4  * of this software and associated documentation files (the "Software"), to deal
   5  * in the Software without restriction, including without limitation the rights
   6  * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
   7  * copies of the Software, and to permit persons to whom the Software is
   8  * furnished to do so, subject to the following conditions:
   9  *
  10  * The above copyright notice and this permission notice shall be included in
  11  * all copies or substantial portions of the Software.
  12  *
  13  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  14  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  15  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  16  * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  17  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  18  * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
  19  * THE SOFTWARE.
  20  */
  21
  22 #define _GNU_SOURCE
  23 #include <dlfcn.h>
  24
  25 #include <stdio.h>
  26 #include <stdlib.h>
  27
  28 void *libgl_handle;
  29 void *wrapper_handle;
  30
  31 static void
  32 open_libgl_handle (void)
  33 {
  34         const char *path;
  35
  36         if (libgl_handle)
  37                 return;
  38
  39         path = getenv ("GLAZE_LIBGL");
  40         if (path == NULL) {
  41                 fprintf (stderr, "GLAZE_LIBGL unset. Please set to path of real libGL.so under glaze.\n");
  42                 exit (1);
  43         }
  44
  45         libgl_handle = dlopen (path, RTLD_LAZY | RTLD_GLOBAL);
  46         if (libgl_handle == NULL) {
  47                 fprintf (stderr, "Error: Failed to dlopen %s\n", path);
  48                 exit (1);
  49         }
  50 }
  51
  52 static void
  53 open_wrapper_handle (void)
  54 {
  55         const char *path;
  56
  57         if (wrapper_handle)
  58                 return;
  59
  60         path = getenv ("GLAZE_WRAPPER");
  61         if (path == NULL) {
  62                 fprintf (stderr, "GLAZE_WRAPPER unset. Please set to path of real libGL.so under glaze.\n");
  63                 exit (1);
  64         }
  65
  66         wrapper_handle = dlopen (path, RTLD_LAZY);
  67         if (wrapper_handle == NULL) {
  68                 fprintf (stderr, "Error: Failed to dlopen %s\n", path);
  69                 exit (1);
  70         }
  71 }
  72
  73 static void
  74 glaze_init (void) __attribute__((constructor));
  75
  76 static void
  77 glaze_init (void)
  78 {
  79         open_libgl_handle ();
  80         open_wrapper_handle ();
  81 }
  82
  83 static void *
  84 resolve (const char *name)
  85 {
  86         void *symbol;
  87
  88         /* The wrapper gets first choice on all symbols. */
  89         symbol = dlsym (wrapper_handle, name);
  90         if (symbol)
  91                 return symbol;
  92
  93         return dlsym (libgl_handle, name);
  94 }
  95
  96 #define GLAZE_API(name)                                                 \
  97 void * name() __attribute__((ifunc(#name "_resolver")));                \
  98 static void *                                                           \
  99 name ## _resolver (void) { return resolve (#name); }
 100
 101 #include "specs/gl.def"
 102 #include "specs/glx.def"