1218885Sdim//===- Win32/DynamicLibrary.cpp - Win32 DL Implementation -------*- C++ -*-===// 2218885Sdim// 3218885Sdim// The LLVM Compiler Infrastructure 4218885Sdim// 5218885Sdim// This file is distributed under the University of Illinois Open Source 6218885Sdim// License. See LICENSE.TXT for details. 7218885Sdim// 8218885Sdim//===----------------------------------------------------------------------===// 9218885Sdim// 10218885Sdim// This file provides the Win32 specific implementation of DynamicLibrary. 11218885Sdim// 12218885Sdim//===----------------------------------------------------------------------===// 13218885Sdim 14276479Sdim#include "WindowsSupport.h" 15218885Sdim 16218885Sdim#ifdef __MINGW32__ 17218885Sdim #include <imagehlp.h> 18218885Sdim#else 19218885Sdim #include <dbghelp.h> 20218885Sdim#endif 21218885Sdim 22218885Sdim#ifdef _MSC_VER 23218885Sdim #include <ntverp.h> 24218885Sdim#endif 25218885Sdim 26218885Sdimnamespace llvm { 27218885Sdimusing namespace sys; 28218885Sdim 29218885Sdim//===----------------------------------------------------------------------===// 30218885Sdim//=== WARNING: Implementation here must contain only Win32 specific code 31218885Sdim//=== and must not be UNIX code. 32218885Sdim//===----------------------------------------------------------------------===// 33218885Sdim 34288943Sdimtypedef BOOL (WINAPI *fpEnumerateLoadedModules)(HANDLE,PENUMLOADED_MODULES_CALLBACK64,PVOID); 35288943Sdimstatic fpEnumerateLoadedModules fEnumerateLoadedModules; 36226633Sdimstatic DenseSet<HMODULE> *OpenedHandles; 37218885Sdim 38288943Sdimstatic bool loadDebugHelp(void) { 39288943Sdim HMODULE hLib = ::LoadLibraryW(L"Dbghelp.dll"); 40288943Sdim if (hLib) { 41288943Sdim fEnumerateLoadedModules = (fpEnumerateLoadedModules) 42288943Sdim ::GetProcAddress(hLib, "EnumerateLoadedModules64"); 43288943Sdim } 44288943Sdim return fEnumerateLoadedModules != 0; 45288943Sdim} 46288943Sdim 47280031Sdimstatic BOOL CALLBACK 48288943SdimELM_Callback(WIN32_ELMCB_PCSTR ModuleName, DWORD64 ModuleBase, 49280031Sdim ULONG ModuleSize, PVOID UserContext) { 50280031Sdim OpenedHandles->insert((HMODULE)ModuleBase); 51280031Sdim return TRUE; 52218885Sdim} 53218885Sdim 54226633SdimDynamicLibrary DynamicLibrary::getPermanentLibrary(const char *filename, 55226633Sdim std::string *errMsg) { 56261991Sdim SmartScopedLock<true> lock(*SymbolsMutex); 57218885Sdim 58226633Sdim if (!filename) { 59226633Sdim // When no file is specified, enumerate all DLLs and EXEs in the process. 60226633Sdim if (OpenedHandles == 0) 61226633Sdim OpenedHandles = new DenseSet<HMODULE>(); 62218885Sdim 63288943Sdim if (!fEnumerateLoadedModules) { 64288943Sdim if (!loadDebugHelp()) { 65288943Sdim assert(false && "These APIs should always be available"); 66288943Sdim return DynamicLibrary(); 67288943Sdim } 68288943Sdim } 69288943Sdim 70288943Sdim fEnumerateLoadedModules(GetCurrentProcess(), ELM_Callback, 0); 71226633Sdim // Dummy library that represents "search all handles". 72226633Sdim // This is mostly to ensure that the return value still shows up as "valid". 73226633Sdim return DynamicLibrary(&OpenedHandles); 74218885Sdim } 75261991Sdim 76261991Sdim SmallVector<wchar_t, MAX_PATH> filenameUnicode; 77276479Sdim if (std::error_code ec = windows::UTF8ToUTF16(filename, filenameUnicode)) { 78261991Sdim SetLastError(ec.value()); 79296417Sdim MakeErrMsg(errMsg, std::string(filename) + ": Can't convert to UTF-16"); 80261991Sdim return DynamicLibrary(); 81261991Sdim } 82226633Sdim 83261991Sdim HMODULE a_handle = LoadLibraryW(filenameUnicode.data()); 84218885Sdim 85226633Sdim if (a_handle == 0) { 86296417Sdim MakeErrMsg(errMsg, std::string(filename) + ": Can't open"); 87226633Sdim return DynamicLibrary(); 88226633Sdim } 89226633Sdim 90226633Sdim if (OpenedHandles == 0) 91226633Sdim OpenedHandles = new DenseSet<HMODULE>(); 92226633Sdim 93226633Sdim // If we've already loaded this library, FreeLibrary() the handle in order to 94226633Sdim // keep the internal refcount at +1. 95226633Sdim if (!OpenedHandles->insert(a_handle).second) 96226633Sdim FreeLibrary(a_handle); 97226633Sdim 98226633Sdim return DynamicLibrary(a_handle); 99218885Sdim} 100218885Sdim 101218885Sdim// Stack probing routines are in the support library (e.g. libgcc), but we don't 102218885Sdim// have dynamic linking on windows. Provide a hook. 103218885Sdim#define EXPLICIT_SYMBOL(SYM) \ 104218885Sdim extern "C" { extern void *SYM; } 105218885Sdim#define EXPLICIT_SYMBOL2(SYMFROM, SYMTO) EXPLICIT_SYMBOL(SYMTO) 106218885Sdim 107280031Sdim#ifdef _M_IX86 108280031Sdim// Win32 on x86 implements certain single-precision math functions as macros. 109280031Sdim// These functions are not exported by the DLL, but will still be needed 110280031Sdim// for symbol-resolution by the JIT loader. Therefore, this Support libray 111280031Sdim// provides helper functions with the same implementation. 112280031Sdim 113280031Sdim#define INLINE_DEF_SYMBOL1(TYP, SYM) \ 114280031Sdim extern "C" TYP inline_##SYM(TYP _X) { return SYM(_X); } 115280031Sdim#define INLINE_DEF_SYMBOL2(TYP, SYM) \ 116280031Sdim extern "C" TYP inline_##SYM(TYP _X, TYP _Y) { return SYM(_X, _Y); } 117280031Sdim#endif 118280031Sdim 119218885Sdim#include "explicit_symbols.inc" 120218885Sdim 121218885Sdim#undef EXPLICIT_SYMBOL 122218885Sdim#undef EXPLICIT_SYMBOL2 123280031Sdim#undef INLINE_DEF_SYMBOL1 124280031Sdim#undef INLINE_DEF_SYMBOL2 125218885Sdim 126218885Sdimvoid* DynamicLibrary::SearchForAddressOfSymbol(const char* symbolName) { 127261991Sdim SmartScopedLock<true> Lock(*SymbolsMutex); 128226633Sdim 129218885Sdim // First check symbols added via AddSymbol(). 130261991Sdim if (ExplicitSymbols.isConstructed()) { 131226633Sdim StringMap<void *>::iterator i = ExplicitSymbols->find(symbolName); 132226633Sdim 133226633Sdim if (i != ExplicitSymbols->end()) 134226633Sdim return i->second; 135218885Sdim } 136218885Sdim 137218885Sdim // Now search the libraries. 138226633Sdim if (OpenedHandles) { 139226633Sdim for (DenseSet<HMODULE>::iterator I = OpenedHandles->begin(), 140226633Sdim E = OpenedHandles->end(); I != E; ++I) { 141226633Sdim FARPROC ptr = GetProcAddress((HMODULE)*I, symbolName); 142226633Sdim if (ptr) { 143226633Sdim return (void *)(intptr_t)ptr; 144226633Sdim } 145218885Sdim } 146218885Sdim } 147218885Sdim 148280031Sdim#define EXPLICIT_SYMBOL(SYM) \ 149280031Sdim if (!strcmp(symbolName, #SYM)) \ 150280031Sdim return (void *)&SYM; 151280031Sdim#define EXPLICIT_SYMBOL2(SYMFROM, SYMTO) \ 152280031Sdim if (!strcmp(symbolName, #SYMFROM)) \ 153280031Sdim return (void *)&SYMTO; 154218885Sdim 155280031Sdim#ifdef _M_IX86 156280031Sdim#define INLINE_DEF_SYMBOL1(TYP, SYM) \ 157280031Sdim if (!strcmp(symbolName, #SYM)) \ 158280031Sdim return (void *)&inline_##SYM; 159280031Sdim#define INLINE_DEF_SYMBOL2(TYP, SYM) INLINE_DEF_SYMBOL1(TYP, SYM) 160280031Sdim#endif 161280031Sdim 162218885Sdim { 163280031Sdim#include "explicit_symbols.inc" 164218885Sdim } 165218885Sdim 166280031Sdim#undef EXPLICIT_SYMBOL 167280031Sdim#undef EXPLICIT_SYMBOL2 168280031Sdim#undef INLINE_DEF_SYMBOL1 169280031Sdim#undef INLINE_DEF_SYMBOL2 170218885Sdim 171218885Sdim return 0; 172218885Sdim} 173218885Sdim 174226633Sdimvoid *DynamicLibrary::getAddressOfSymbol(const char *symbolName) { 175226633Sdim if (!isValid()) 176226633Sdim return NULL; 177226633Sdim if (Data == &OpenedHandles) 178226633Sdim return SearchForAddressOfSymbol(symbolName); 179226633Sdim return (void *)(intptr_t)GetProcAddress((HMODULE)Data, symbolName); 180218885Sdim} 181226633Sdim 182226633Sdim} 183