summaryrefslogtreecommitdiffstats
path: root/llvm/lib/System/Win32/DynamicLibrary.inc
diff options
context:
space:
mode:
authorReid Spencer <rspencer@reidspencer.com>2005-01-09 23:29:00 +0000
committerReid Spencer <rspencer@reidspencer.com>2005-01-09 23:29:00 +0000
commitc892a0db9faab4c36f2276f4675661875b2c84b9 (patch)
treea6c30a9befae0918f0a62d7a354cd339a711c91e /llvm/lib/System/Win32/DynamicLibrary.inc
parentfde3a212e2f7f983e676d374e9a7bd445071e5d8 (diff)
downloadbcm5719-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.inc127
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
OpenPOWER on IntegriCloud