aboutsummaryrefslogtreecommitdiff
path: root/lib/System/Win32/DynamicLibrary.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'lib/System/Win32/DynamicLibrary.cpp')
-rw-r--r--lib/System/Win32/DynamicLibrary.cpp14
1 files changed, 9 insertions, 5 deletions
diff --git a/lib/System/Win32/DynamicLibrary.cpp b/lib/System/Win32/DynamicLibrary.cpp
index 15730d43a0..b2add2ffe9 100644
--- a/lib/System/Win32/DynamicLibrary.cpp
+++ b/lib/System/Win32/DynamicLibrary.cpp
@@ -12,6 +12,7 @@
//===----------------------------------------------------------------------===//
#include "Win32.h"
+#include <windef.h>
namespace llvm {
using namespace sys;
@@ -22,9 +23,10 @@ using namespace sys;
//===----------------------------------------------------------------------===//
DynamicLibrary::DynamicLibrary(const char*filename) : handle(0) {
- handle = LoadLibrary(filename);
+ handle = new HMODULE;
+ *((HMODULE*)handle) = LoadLibrary(filename);
- if (handle == 0) {
+ if (*((HMODULE*)handle) == 0) {
char Buffer[100];
// FIXME: This should use FormatMessage
sprintf(Buffer, "Windows error code %d\n", GetLastError());
@@ -33,13 +35,15 @@ DynamicLibrary::DynamicLibrary(const char*filename) : handle(0) {
}
DynamicLibrary::~DynamicLibrary() {
- if (handle)
- FreeLibrary(handle);
+ assert(handle !=0 && "Invalid DynamicLibrary handle");
+ if (*((HMODULE*)handle))
+ FreeLibrary(*((HMODULE*)handle));
+ delete (HMODULE*)handle;
}
void *DynamicLibrary::GetAddressOfSymbol(const char *symbolName) {
assert(handle !=0 && "Invalid DynamicLibrary handle");
- return GetProcAddress(handle, symbolName);
+ return (void*) GetProcAddress(*((HMODULE*)handle), symbolName);
}
}