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.
107 lines
2.5 KiB
C++
107 lines
2.5 KiB
C++
/**
|
|
* \author Pyry Haulos <pyry.haulos@gmail.com>
|
|
* \date 2007-2008
|
|
* For conditions of distribution and use, see copyright notice in elfhacks.h
|
|
*/
|
|
|
|
#include <dlfcn.h>
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include "real_dlsym.h"
|
|
#include "elfhacks.h"
|
|
|
|
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;
|
|
|
|
static void get_real_functions()
|
|
{
|
|
eh_obj_t libdl;
|
|
int ret;
|
|
|
|
const char* libs[] = {
|
|
#if defined(__GLIBC__)
|
|
"*libdl.so*",
|
|
#endif
|
|
"*libc.so*",
|
|
"*libc.*.so*",
|
|
"*ld-musl-*.so*",
|
|
};
|
|
|
|
for (size_t i = 0; i < sizeof(libs) / sizeof(*libs); i++)
|
|
{
|
|
ret = eh_find_obj(&libdl, libs[i]);
|
|
if (ret)
|
|
continue;
|
|
|
|
eh_find_sym(&libdl, "dlopen", (void **) &__dlopen);
|
|
eh_find_sym(&libdl, "dlsym", (void **) &__dlsym);
|
|
eh_destroy_obj(&libdl);
|
|
|
|
if (__dlopen && __dlsym)
|
|
break;
|
|
__dlopen = nullptr;
|
|
__dlsym = nullptr;
|
|
}
|
|
|
|
if (!__dlopen && !__dlsym)
|
|
{
|
|
fprintf(stderr, "MANGOHUD: Can't get dlopen() and dlsym()\n");
|
|
exit(ret ? ret : 1);
|
|
}
|
|
}
|
|
|
|
/**
|
|
* \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)
|
|
#ifdef RTLD_DEEPBIND
|
|
FLAG(RTLD_DEEPBIND)
|
|
#endif
|
|
#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;
|
|
}
|
|
|
|
void* get_proc_address(const char* name) {
|
|
void (*func)() = (void (*)())real_dlsym(RTLD_NEXT, name);
|
|
return (void*)func;
|
|
}
|