src/dso_utils.c

   1 /*
   2  * Copyright (C) 2012 Intel Corporation. All Rights Reserved.
   3  *
   4  * Permission is hereby granted, free of charge, to any person obtaining a
   5  * copy of this software and associated documentation files (the
   6  * "Software"), to deal in the Software without restriction, including
   7  * without limitation the rights to use, copy, modify, merge, publish,
   8  * distribute, sub license, and/or sell copies of the Software, and to
   9  * permit persons to whom the Software is furnished to do so, subject to
  10  * the following conditions:
  11  *
  12  * The above copyright notice and this permission notice (including the
  13  * next paragraph) shall be included in all copies or substantial portions
  14  * of the Software.
  15  *
  16  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
  17  * OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
  18  * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT.
  19  * IN NO EVENT SHALL PRECISION INSIGHT AND/OR ITS SUPPLIERS BE LIABLE FOR
  20  * ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
  21  * TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
  22  * SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
  23  */
  24
  25 #define _GNU_SOURCE 1
  26 #include <stdio.h>
  27 #include <stdlib.h>
  28 #include <dlfcn.h>
  29 #include "dso_utils.h"
  30
  31 struct dso_handle {
  32     void       *handle;
  33 };
  34
  35 /* Opens the named shared library */
  36 struct dso_handle *
  37 dso_open(const char *path)
  38 {
  39     struct dso_handle *h;
  40
  41     h = calloc(1, sizeof(*h));
  42     if (!h)
  43         return NULL;
  44
  45     if (path) {
  46         h->handle = dlopen(path, RTLD_LAZY | RTLD_LOCAL);
  47         if (!h->handle)
  48             goto error;
  49     } else
  50         h->handle = RTLD_DEFAULT;
  51     return h;
  52
  53 error:
  54     dso_close(h);
  55     return NULL;
  56 }
  57
  58 /* Closes and disposed any allocated data */
  59 void
  60 dso_close(struct dso_handle *h)
  61 {
  62     if (!h)
  63         return;
  64
  65     if (h->handle) {
  66         if (h->handle != RTLD_DEFAULT)
  67             dlclose(h->handle);
  68         h->handle = NULL;
  69     }
  70     free(h);
  71 }
  72
  73 /* Load symbol into the supplied location */
  74 static bool
  75 get_symbol(struct dso_handle *h, void *func_vptr, const char *name)
  76 {
  77     dso_generic_func func, * const func_ptr = func_vptr;
  78     const char *error;
  79
  80     dlerror();
  81     func = (dso_generic_func)dlsym(h->handle, name);
  82     error = dlerror();
  83     if (error) {
  84         fprintf(stderr, "error: failed to resolve %s(): %s\n", name, error);
  85         return false;
  86     }
  87     *func_ptr = func;
  88     return true;
  89 }
  90
  91 /* Loads symbols into the supplied vtable */
  92 bool
  93 dso_get_symbols(
  94     struct dso_handle          *h,
  95     void                       *vtable,
  96     unsigned int                vtable_length,
  97     const struct dso_symbol    *symbols
  98 )
  99 {
 100     const struct dso_symbol *s;
 101
 102     for (s = symbols; s->name != NULL; s++) {
 103         if (s->offset + sizeof(dso_generic_func) > vtable_length)
 104             return false;
 105         if (!get_symbol(h, ((char *)vtable) + s->offset, s->name))
 106             return false;
 107     }
 108     return true;
 109 }