]> git.cworth.org Git - glaze/blobdiff - glaze.c
Add specs/README
[glaze] / glaze.c
diff --git a/glaze.c b/glaze.c
index 5ad3af33a59301cb6e4da0857948fd3e49962624..8377d846a149321640e8b3dfc0bcae3c0d3ed1dc 100644 (file)
--- a/glaze.c
+++ b/glaze.c
  * THE SOFTWARE.
  */
 
+#include "config.h"
+
 #define _GNU_SOURCE
 #include <dlfcn.h>
 
+#include "glaze.h"
+
+#include <talloc.h>
+
 #include <stdio.h>
 #include <stdlib.h>
+#include <string.h>
+#include <sys/types.h>
+#include <sys/stat.h>
+#include <unistd.h>
+#include <errno.h>
+
+/* For PATH_MAX */
+#include <linux/limits.h>
+
+void *
+glaze_lookup (char *function)
+{
+       static void *libgl_handle = NULL;
+       void *ret;
+
+       if (libgl_handle == NULL) {
+               const char *path;
+
+               path = getenv ("GLAZE_LIBGL");
+               if (path == NULL) {
+                       fprintf (stderr, "GLAZE_LIBGL unset. "
+                                "Please set to path of libGL.so under glaze.\n"
+                               );
+                       exit (1);
+               }
+
+               libgl_handle = dlopen (path, RTLD_LAZY | RTLD_GLOBAL);
+               if (libgl_handle == NULL) {
+                       fprintf (stderr, "glaze_lookup: Error: Failed to dlopen %s\n", path);
+                       exit (1);
+               }
+       }
 
-void *libgl_handle;
+       ret = dlsym (libgl_handle, function);
 
+       if (ret == NULL) {
+               fprintf (stderr, "Error: glaze_lookup failed to dlsym %s\n",
+                        function);
+               exit (1);
+       }
+
+       return ret;
+}
+
+/* Terminate a string representing a filename at the final '/' to
+ * eliminate the final filename component, (leaving only the directory
+ * portions of the original path).
+ *
+ * Notes: A path containing no '/' character will not be modified.
+ *        A path consisting only of "/" will not be modified.
+ */
 static void
-open_libgl_handle (void)
+chop_trailing_path_component (char *path)
 {
-       const char *path;
+       char *slash;
+
+       slash = strrchr (path, '/');
 
-       if (libgl_handle)
+       if (slash == NULL)
                return;
 
-       path = getenv ("GLAZE_LIBGL");
-       if (path == NULL) {
-               fprintf (stderr, "GLAZE_LIBGL unset. Please set to path of real libGL.so under glaze.\n");
+       if (slash == path)
+               return;
+
+       *slash = '\0';
+}
+
+/* Find the absolute path of the currently executing binary.
+ *
+ * Returns: a string talloc'ed to 'ctx'
+ */
+static char *
+get_bin_name (void *ctx)
+{
+       const char *link = "/proc/self/exe";
+       char *name;
+
+       /* Yes, PATH_MAX is cheesy. I would have preferred to have
+        * used lstat and read the resulting st_size, but everytime I
+        * did that with /proc/self/exe I got a value of 0, (whereas
+        * with a "real" symbolic link I make myself I get the length
+        * of the filename being linked to). Go figure. */
+       int name_len = PATH_MAX + 1;
+
+       name = talloc_size (ctx, name_len);
+       if (name == NULL) {
+               fprintf (stderr, "Out of memory.\n");
                exit (1);
        }
 
-       libgl_handle = dlopen (path, RTLD_LAZY | RTLD_GLOBAL);
-       if (libgl_handle == NULL) {
-               fprintf (stderr, "Error: Failed to dlopen %s\n", path);
+       name_len = readlink (link, name, name_len - 1);
+       if (name_len < 0) {
+               fprintf (stderr, "Failed to readlink %s: %s\n", link,
+                        strerror (errno));
                exit (1);
        }
+
+       name[name_len] = '\0';
+
+       return name;
 }
 
-static void
-glaze_init (void) __attribute__((constructor));
+/* Does path exist? */
+static int
+exists (const char *path)
+{
+       struct stat st;
+       int err;
 
-static void
-glaze_init (void)
+       err = stat (path, &st);
+
+       /* Failed to stat. It either doesn't exist, or might as well not. */
+       if (err == -1)
+               return 0;
+
+       return 1;
+}
+
+/* Execute "program" in a pipe, reads its first line of output on
+ * stdout, and returns that as a string (discarding any further
+ * output).
+ *
+ * Returns NULL if the program failed to execute for any reason.
+ *
+ * NOTE: The caller should free() the returned string when done with
+ * it.
+ */
+static char *
+read_process_output_one_line (const char *program)
+{
+       FILE *process;
+       int status;
+       char *line = NULL;
+       size_t len = 0;
+       ssize_t bytes_read;
+
+       process = popen (program, "r");
+       if (process == NULL)
+               return NULL;
+
+       bytes_read = getline (&line, &len, process);
+
+       status = pclose (process);
+       if (! WIFEXITED (status))
+               return NULL;
+
+       if (WEXITSTATUS (status))
+               return NULL;
+
+       if (bytes_read == -1)
+               return NULL;
+
+       if (bytes_read) {
+               if (line[strlen(line)-1] == '\n')
+                       line[strlen(line)-1] = '\0';
+               return line;
+       } else {
+               return NULL;
+       }
+}
+
+
+/* Look for "wrapper" library next to currently executing binary.
+ *
+ * If "wrapper" is an absolute path, return it directly.
+ *
+ * Otherwise, ("wrapper" is relative), look for an existing file named
+ * "wrapper" in the same directory as the currently executing binary,
+ * (as determined by /proc/self/exe). If that file exists, return its
+ * path.
+ *
+ * Otherwise, return the original, relative "wrapper".
+ */
+static const char *
+resolve_wrapper_path (void *ctx, const char *wrapper)
 {
-       open_libgl_handle ();
+       char *bin_path, *lib_path;
+
+       if (*wrapper == '/')
+               return wrapper;
+
+       bin_path = get_bin_name (ctx);
+
+       chop_trailing_path_component (bin_path);
+
+       lib_path = talloc_asprintf (ctx, "%s/%s", bin_path, wrapper);
+
+       talloc_free (bin_path);
+
+       if (exists (lib_path))
+               return lib_path;
+
+       talloc_free (lib_path);
+
+       return wrapper;
 }
 
-static void *
-resolve (const char *name)
+/* Return path to directory containing Glaze wrapper's libGL.so.1
+ * suitable for use in LD_PRELOAD or LD_LIBRARY_PATH. Note that the
+ * path returned may not be a full path to the directory but may end
+ * with "$LIB" which will be expanded by the Linux dynamic linker to
+ * an architecture specific string (such as "lib/i386-linux-gnu" or
+ * "lib/x86_64-linux-gnu"). */
+static const char *
+find_glaze_libgl_dir (void)
 {
-       return dlsym (libgl_handle, name);
+       return CONFIG_LIBDIR "/glaze/$LIB";
 }
 
-#define GLAZE_API(name)                                                        \
-void * name() __attribute__((ifunc(#name "_resolver")));               \
-static void *                                                          \
-name ## _resolver (void) { return resolve (#name); }
+void
+glaze_execute (int argc, char *argv[], const char *wrapper)
+{
+       void *ctx = talloc_new (NULL);
+       int i;
 
-#include "glapi.def"
+       /* Set GLAZE_WRAPPER to absolute path of wrapper library */
+       if (wrapper == NULL || *wrapper == '\0') {
+               fprintf (stderr, "Error: glaze_execute called with empty wrapper library.\n");
+               return;
+       }
+
+       wrapper = resolve_wrapper_path (ctx, wrapper);
+
+       setenv ("GLAZE_WRAPPER", wrapper, 1);
+
+       /* Ensure GLAZE_LIBGL is set. If not, set GLAZE_LIBGL_32_AUTO
+        * and GLAZE_LIBGL_64_AUTO
+        *
+        * Note that we must do this before setting LD_LIBRARY_PATH,
+        * since after that, of course we would find Glaze's wrapper
+        * libGL.so.1. */
+       if (getenv ("GLAZE_LIBGL") == NULL) {
+               char *libgl_path;
+
+               libgl_path = read_process_output_one_line ("glaze-find-libgl-32");
+               if (libgl_path) {
+                       setenv ("GLAZE_LIBGL_32_AUTO", libgl_path, 1);
+                       free (libgl_path);
+               }
+
+               libgl_path = read_process_output_one_line ("glaze-find-libgl-64");
+               if (libgl_path) {
+                       setenv ("GLAZE_LIBGL_64_AUTO", libgl_path, 1);
+                       free (libgl_path);
+               }
+       }
+
+       /* Set LD_LIBRARY_PATH to include glaze's own libGL.so */
+       const char *glaze_libgl_dir, *ld_library_path;
+
+       glaze_libgl_dir = find_glaze_libgl_dir ();
+
+       ld_library_path = getenv ("LD_LIBRARY_PATH");
+
+       if (ld_library_path == NULL)
+               ld_library_path = glaze_libgl_dir;
+       else
+               ld_library_path = talloc_asprintf (ctx, "%s:%s",
+                                                  glaze_libgl_dir,
+                                                  ld_library_path);
+
+       setenv ("LD_LIBRARY_PATH", ld_library_path, 1);
+
+       talloc_free (ctx);
+
+       /* Execute program */
+       execvp (argv[0], argv);
+
+       /* If execvp returns, something went wrong. */
+       fprintf (stderr, "Error: Failed to exec:");
+       for (i = 0; i < argc; i++)
+               fprintf (stderr, " %s", argv[i]);
+       fprintf (stderr, "\n");
+
+       return;
+}
+
+void
+glaze_set_first_gl_call_callback (const char *function_name)
+{
+       setenv ("GLAZE_FIRST_GL_CALL_CALLBACK", function_name, 1);
+}