Add wrapper for LoadLibrary, FreeLibrary.

1 files changed, 26 insertions, 0 deletions

diff --git a/c/common.cpp b/c/common.cpp
index 63adb53..93f97eb 100644
--- a/c/common.cpp
+++ b/c/common.cpp
@@ -1,3 +1,4 @@
+#include <memory>
 #include <windows.h>
 #include <SWI-Prolog.h>
 
@@ -27,3 +28,28 @@ TszFromSz(const char *sz, int iCp)
 
 	return tsz;
 }
+
+Library::Library(const TCHAR *tszLibrary)
+{
+	m_hModule = LoadLibrary(tszLibrary);
+}
+
+Library::~Library()
+{
+	if (m_hModule)
+		FreeLibrary(m_hModule);
+}
+
+FARPROC
+Library::GetProcAddress(const char *szProc)
+{
+	return m_hModule? ::GetProcAddress(m_hModule, szProc): NULL;
+}
+
+std::unique_ptr<Library>
+Library::Load(const TCHAR *tszLibrary)
+{
+	auto upLib = std::make_unique<Library>(tszLibrary);
+	if (!upLib->m_hModule) upLib = nullptr;
+	return upLib;
+}