You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
84 lines
2.0 KiB
C++
84 lines
2.0 KiB
C++
4 years ago
|
/**
|
||
|
* \author Pyry Haulos <pyry.haulos@gmail.com>
|
||
|
* \date 2007-2008
|
||
|
* For conditions of distribution and use, see copyright notice in elfhacks.h
|
||
|
*/
|
||
4 years ago
|
|
||
4 years ago
|
#include <stdio.h>
|
||
4 years ago
|
#include <stdlib.h>
|
||
4 years ago
|
#include "real_dlsym.h"
|
||
|
#include "elfhacks.h"
|
||
4 years ago
|
|
||
4 years ago
|
void *(*__dlopen)(const char *, int) = nullptr;
|
||
|
void *(*__dlsym)(void *, const char *) = nullptr;
|
||
|
static bool print_dlopen = getenv("MANGOHUD_DEBUG_DLOPEN") != nullptr;
|
||
|
static bool print_dlsym = getenv("MANGOHUD_DEBUG_DLSYM") != nullptr;
|
||
4 years ago
|
|
||
4 years ago
|
void get_real_functions()
|
||
4 years ago
|
{
|
||
4 years ago
|
eh_obj_t libdl;
|
||
|
|
||
|
if (eh_find_obj(&libdl, "*libdl.so*")) {
|
||
|
fprintf(stderr, "can't get libdl.so\n");
|
||
|
exit(1);
|
||
|
}
|
||
|
|
||
|
if (eh_find_sym(&libdl, "dlopen", (void **) &__dlopen)) {
|
||
|
fprintf(stderr, "can't get dlopen()\n");
|
||
|
exit(1);
|
||
|
}
|
||
4 years ago
|
|
||
4 years ago
|
if (eh_find_sym(&libdl, "dlsym", (void **) &__dlsym)) {
|
||
|
fprintf(stderr, "can't get dlsym()\n");
|
||
|
exit(1);
|
||
4 years ago
|
}
|
||
|
|
||
4 years ago
|
eh_destroy_obj(&libdl);
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* \brief dlopen() wrapper, just passes calls to real dlopen()
|
||
|
* and writes information to standard output
|
||
|
*/
|
||
|
void *real_dlopen(const char *filename, int flag)
|
||
|
{
|
||
|
if (__dlopen == nullptr)
|
||
|
get_real_functions();
|
||
|
|
||
|
void *result = __dlopen(filename, flag);
|
||
|
|
||
|
if (print_dlopen) {
|
||
|
printf("dlopen(%s, ", filename);
|
||
|
const char *fmt = "%s";
|
||
|
#define FLAG(test) if (flag & test) { printf(fmt, #test); fmt = "|%s"; }
|
||
|
FLAG(RTLD_LAZY)
|
||
|
FLAG(RTLD_NOW)
|
||
|
FLAG(RTLD_GLOBAL)
|
||
|
FLAG(RTLD_LOCAL)
|
||
|
FLAG(RTLD_NODELETE)
|
||
|
FLAG(RTLD_NOLOAD)
|
||
|
FLAG(RTLD_DEEPBIND)
|
||
|
#undef FLAG
|
||
|
printf(") = %p\n", result);
|
||
|
}
|
||
|
|
||
|
return result;
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* \brief dlsym() wrapper, passes calls to real dlsym() and
|
||
|
* writes information to standard output
|
||
|
*/
|
||
|
void *real_dlsym(void *handle, const char *symbol)
|
||
|
{
|
||
|
if (__dlsym == nullptr)
|
||
|
get_real_functions();
|
||
|
|
||
|
void *result = __dlsym(handle, symbol);
|
||
|
|
||
|
if (print_dlsym)
|
||
|
printf("dlsym(%p, %s) = %p\n", handle, symbol, result);
|
||
|
|
||
|
return result;
|
||
4 years ago
|
}
|