liblttng-ust-dl/ustdl.c

   1 /*
   2  * Copyright (C) 2013  Paul Woegerer <paul.woegerer@mentor.com>
   3  *
   4  * This library is free software; you can redistribute it and/or
   5  * modify it under the terms of the GNU Lesser General Public
   6  * License as published by the Free Software Foundation; version 2.1 of
   7  * the License.
   8  *
   9  * This library is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12  * Lesser General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU Lesser General Public
  15  * License along with this library; if not, write to the Free Software
  16  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301 USA
  17  */
  18
  19 #define _LGPL_SOURCE
  20 #define _GNU_SOURCE
  21 #include <inttypes.h>
  22 #include <dlfcn.h>
  23 #include <link.h>
  24 #include <unistd.h>
  25 #include <stdio.h>
  26 #include <limits.h>
  27 #include <sys/types.h>
  28 #include <sys/stat.h>
  29 #include <signal.h>
  30 #include <sched.h>
  31 #include <stdarg.h>
  32 #include "usterr.h"
  33
  34 #include <lttng/ust-compiler.h>
  35 #include <lttng/ust.h>
  36
  37 #define TRACEPOINT_DEFINE
  38 #include "ust_baddr.h"
  39
  40 static void *(*__lttng_ust_plibc_dlopen)(const char *filename, int flag);
  41 static int (*__lttng_ust_plibc_dlclose)(void *handle);
  42
  43 static
  44 void *_lttng_ust_dl_libc_dlopen(const char *filename, int flag)
  45 {
  46         if (!__lttng_ust_plibc_dlopen) {
  47                 __lttng_ust_plibc_dlopen = dlsym(RTLD_NEXT, "dlopen");
  48                 if (__lttng_ust_plibc_dlopen == NULL) {
  49                         fprintf(stderr, "%s\n", dlerror());
  50                         return NULL;
  51                 }
  52         }
  53         return __lttng_ust_plibc_dlopen(filename, flag);
  54 }
  55
  56 static
  57 int _lttng_ust_dl_libc_dlclose(void *handle)
  58 {
  59         if (!__lttng_ust_plibc_dlclose) {
  60                 __lttng_ust_plibc_dlclose = dlsym(RTLD_NEXT, "dlclose");
  61                 if (__lttng_ust_plibc_dlclose == NULL) {
  62                         fprintf(stderr, "%s\n", dlerror());
  63                         return -1;
  64                 }
  65         }
  66         return __lttng_ust_plibc_dlclose(handle);
  67 }
  68
  69 static
  70 void lttng_ust_baddr_push(void *so_base, const char *so_name)
  71 {
  72         char resolved_path[PATH_MAX];
  73         struct stat sostat;
  74
  75         if (!realpath(so_name, resolved_path)) {
  76                 ERR("could not resolve path '%s'", so_name);
  77                 return;
  78         }
  79
  80         if (stat(resolved_path, &sostat)) {
  81                 ERR("could not access file status for %s", resolved_path);
  82                 return;
  83         }
  84
  85         tracepoint(ust_baddr, push,
  86                 so_base, resolved_path, sostat.st_size, sostat.st_mtime);
  87         return;
  88 }
  89
  90 void *dlopen(const char *filename, int flag)
  91 {
  92         void *handle = _lttng_ust_dl_libc_dlopen(filename, flag);
  93         if (__tracepoint_ptrs_registered && handle) {
  94                 struct link_map *p = NULL;
  95                 if (dlinfo(handle, RTLD_DI_LINKMAP, &p) != -1 && p != NULL
  96                                 && p->l_addr != 0)
  97                         lttng_ust_baddr_push((void *) p->l_addr, p->l_name);
  98         }
  99         return handle;
 100 }
 101
 102 int dlclose(void *handle)
 103 {
 104         if (__tracepoint_ptrs_registered && handle) {
 105                 struct link_map *p = NULL;
 106                 if (dlinfo(handle, RTLD_DI_LINKMAP, &p) != -1 && p != NULL
 107                                 && p->l_addr != 0)
 108                         tracepoint(ust_baddr, pop, (void *) p->l_addr);
 109         }
 110         return _lttng_ust_dl_libc_dlclose(handle);
 111 }