diff options
| author | Reid Spencer <rspencer@reidspencer.com> | 2005-01-09 23:29:00 +0000 |
|---|---|---|
| committer | Reid Spencer <rspencer@reidspencer.com> | 2005-01-09 23:29:00 +0000 |
| commit | c892a0db9faab4c36f2276f4675661875b2c84b9 (patch) | |
| tree | a6c30a9befae0918f0a62d7a354cd339a711c91e /llvm/lib/System/Win32/DynamicLibrary.inc | |
| parent | fde3a212e2f7f983e676d374e9a7bd445071e5d8 (diff) | |
| download | bcm5719-llvm-c892a0db9faab4c36f2276f4675661875b2c84b9.tar.gz bcm5719-llvm-c892a0db9faab4c36f2276f4675661875b2c84b9.zip | |
Rename Unix/*.cpp and Win32/*.cpp to have a *.inc suffix so that the silly
gdb debugger doesn't get confused on which file it is reading (the one in
lib/System or the one in lib/System/{Win32,Unix})
llvm-svn: 19426
Diffstat (limited to 'llvm/lib/System/Win32/DynamicLibrary.inc')
| -rw-r--r-- | llvm/lib/System/Win32/DynamicLibrary.inc | 127 |
1 files changed, 127 insertions, 0 deletions
diff --git a/llvm/lib/System/Win32/DynamicLibrary.inc b/llvm/lib/System/Win32/DynamicLibrary.inc new file mode 100644 index 00000000000..af28f756254 --- /dev/null +++ b/llvm/lib/System/Win32/DynamicLibrary.inc @@ -0,0 +1,127 @@ +//===- Win32/DynamicLibrary.cpp - Win32 DL Implementation -------*- C++ -*-===// +// +// The LLVM Compiler Infrastructure +// +// This file was developed by Jeff Cohen and is distributed under the +// University of Illinois Open Source License. See LICENSE.TXT for details. +// +//===----------------------------------------------------------------------===// +// +// This file provides the Win32 specific implementation of DynamicLibrary. +// +//===----------------------------------------------------------------------===// + +#include "Win32.h" + +#ifdef __MINGW +#include <imagehlp.h> +#else +#include <dbghelp.h> +#endif + +#pragma comment(lib, "dbghelp.lib") + +namespace llvm { +using namespace sys; + +//===----------------------------------------------------------------------===// +//=== WARNING: Implementation here must contain only Win32 specific code +//=== and must not be UNIX code. +//===----------------------------------------------------------------------===// + +static std::vector<HMODULE> OpenedHandles; + +extern "C" { + static BOOL CALLBACK ELM_Callback(PSTR ModuleName, + ULONG ModuleBase, + ULONG ModuleSize, + PVOID UserContext) + { + // Ignore VC++ runtimes prior to 7.1. Somehow some of them get loaded + // into the process. + if (stricmp(ModuleName, "msvci70") != 0 && + stricmp(ModuleName, "msvcirt") != 0 && + stricmp(ModuleName, "msvcp50") != 0 && + stricmp(ModuleName, "msvcp60") != 0 && + stricmp(ModuleName, "msvcp70") != 0 && + stricmp(ModuleName, "msvcr70") != 0 && + stricmp(ModuleName, "msvcrt") != 0 && + stricmp(ModuleName, "msvcrt20") != 0 && + stricmp(ModuleName, "msvcrt40") != 0) { + OpenedHandles.push_back((HMODULE)ModuleBase); + } + return TRUE; + } +} + +DynamicLibrary::DynamicLibrary() : handle(0) { + handle = GetModuleHandle(NULL); + OpenedHandles.push_back((HMODULE)handle); +} + +DynamicLibrary::DynamicLibrary(const char*filename) : handle(0) { + HMODULE a_handle = LoadLibrary(filename); + + if (a_handle == 0) + ThrowError(std::string(filename) + ": Can't open : "); + + handle = a_handle; + OpenedHandles.push_back(a_handle); +} + +DynamicLibrary::~DynamicLibrary() { + if (handle == 0) + return; + + // GetModuleHandle() does not increment the ref count, so we must not free + // the handle to the executable. + if (handle != GetModuleHandle(NULL)) + FreeLibrary((HMODULE)handle); + handle = 0; + + for (std::vector<HMODULE>::iterator I = OpenedHandles.begin(), + E = OpenedHandles.end(); I != E; ++I) { + if (*I == handle) { + // Note: don't use the swap/pop_back trick here. Order is important. + OpenedHandles.erase(I); + } + } +} + +void DynamicLibrary::LoadLibraryPermanently(const char* filename) { + if (filename) { + HMODULE a_handle = LoadLibrary(filename); + + if (a_handle == 0) + ThrowError(std::string(filename) + ": Can't open : "); + + OpenedHandles.push_back(a_handle); + } else { + // When no file is specified, enumerate all DLLs and EXEs in the + // process. + EnumerateLoadedModules(GetCurrentProcess(), ELM_Callback, 0); + } + + // Because we don't remember the handle, we will never free it; hence, + // it is loaded permanently. +} + +void* DynamicLibrary::SearchForAddressOfSymbol(const char* symbolName) { + for (std::vector<HMODULE>::iterator I = OpenedHandles.begin(), + E = OpenedHandles.end(); I != E; ++I) { + FARPROC ptr = GetProcAddress((HMODULE)*I, symbolName); + if (ptr) + return (void *) ptr; + } + + return 0; +} + +void *DynamicLibrary::GetAddressOfSymbol(const char *symbolName) { + assert(handle != 0 && "Invalid DynamicLibrary handle"); + return (void *) GetProcAddress((HMODULE)handle, symbolName); +} + +} + +// vim: sw=2 smartindent smarttab tw=80 autoindent expandtab |

