31 lines
835 B
C++
31 lines
835 B
C++
#pragma once
|
|
#include <windows.h>
|
|
|
|
// A DLL runner.
|
|
class DllRunner {
|
|
public:
|
|
virtual ~DllRunner(){}
|
|
virtual void* LoadLibraryA(const char* path, int size = 0) = 0;
|
|
virtual FARPROC GetProcAddress(void* mod, const char* lpProcName) = 0;
|
|
virtual BOOL FreeLibrary(void* mod) = 0;
|
|
};
|
|
|
|
// Default DLL runner.
|
|
class DefaultDllRunner : public DllRunner {
|
|
private:
|
|
std::string m_path;
|
|
HMODULE m_mod;
|
|
public:
|
|
DefaultDllRunner(const std::string &path="") :m_path(path), m_mod(nullptr) {}
|
|
// Load DLL from the disk.
|
|
virtual void* LoadLibraryA(const char* path, int size = 0) {
|
|
return m_mod = ::LoadLibraryA(size ? m_path.c_str() : path);
|
|
}
|
|
virtual FARPROC GetProcAddress(void* mod, const char* lpProcName) {
|
|
return ::GetProcAddress(m_mod, lpProcName);
|
|
}
|
|
virtual BOOL FreeLibrary(void* mod) {
|
|
return ::FreeLibrary(m_mod);
|
|
}
|
|
};
|