max/lean4-htt
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 2

fix: search all loaded modules for symbols on Windows

Browse source
This commit is contained in:
Sebastian Ullrich 2021-08-04 15:57:42 +02:00
parent d9337fa39c
commit 58d6f3b817
1 changed files with 11 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
src/library/compiler/ir_interpreter.cpp
View file

@ -31,6 +31,7 @@ functions, which have a (relatively) homogeneous ABI that we can use without run
#ifdef LEAN_WINDOWS
#include <windows.h>
#undef ERROR // thanks, wingdi.h
#include <psapi.h>
#else
#include <dlfcn.h>
#endif
@ -286,7 +287,16 @@ void print_value(std::ostream & ios, value const & v, type t) {
void * lookup_symbol_in_cur_exe(char const * sym) {
#ifdef LEAN_WINDOWS
return reinterpret_cast<void *>(GetProcAddress(GetModuleHandle(nullptr), sym));
HMODULE hmods[128]; // 128 modules should be enough for everyone
DWORD bytes_needed;
lean_always_assert(EnumProcessModules(GetCurrentProcess(), hmods, sizeof(hmods), &bytes_needed));
for (int i = 0; i < bytes_needed / sizeof(HMODULE); i++) {
void * addr = reinterpret_cast<void *>(GetProcAddress(hmods[i], sym));
if (addr) {
return addr;
}
}
return nullptr;
#else
return dlsym(RTLD_DEFAULT, sym);
#endif