src/ldpreload.h

   1 /*
   2  * Helper header for LD_PRELOAD related headers macros. Must be loaded _first_
   3  * (for RTLD_NEXT)!
   4  *
   5  * Copyright (C) 2012-2013  Simon Ruderich
   6  *
   7  * This program is free software: you can redistribute it and/or modify
   8  * it under the terms of the GNU General Public License as published by
   9  * the Free Software Foundation, either version 3 of the License, or
  10  * (at your option) any later version.
  11  *
  12  * This program is distributed in the hope that it will be useful,
  13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15  * GNU General Public License for more details.
  16  *
  17  * You should have received a copy of the GNU General Public License
  18  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
  19  */
  20
  21 #ifndef LDPRELOAD_H
  22 #define LDPRELOAD_H 1
  23
  24 /* Necessary for RTLD_NEXT. */
  25 #define _GNU_SOURCE
  26
  27 /* abort() */
  28 #include <stdlib.h>
  29 /* dlsym() */
  30 #include <dlfcn.h>
  31
  32
  33 /* Load the function name using dlsym() if necessary and store it in pointer.
  34  * Terminate program on failure. */
  35 #define DLSYM_FUNCTION(pointer, name) \
  36     if (NULL == (pointer)) { \
  37         char *error; \
  38         dlerror(); /* Clear possibly existing error. */ \
  39         \
  40         *(void **) (&(pointer)) = dlsym(RTLD_NEXT, (name)); \
  41         \
  42         if (NULL != (error = dlerror())) { \
  43             /* Not much we can do. Most likely the other output functions \
  44              * failed to load too. */ \
  45             abort(); \
  46         } \
  47     }
  48
  49 #endif