134 lines
3.3 KiB
C++
134 lines
3.3 KiB
C++
// This file is distributed under the BSD License.
|
|
// See "license.txt" for details.
|
|
// Copyright 2009-2012, Jonathan Turner (jonathan@emptycrate.com)
|
|
// Copyright 2009-2016, Jason Turner (jason@emptycrate.com)
|
|
// http://www.chaiscript.com
|
|
|
|
#ifndef CHAISCRIPT_WINDOWS_HPP_
|
|
#define CHAISCRIPT_WINDOWS_HPP_
|
|
|
|
#include <string>
|
|
|
|
#ifdef CHAISCRIPT_WINDOWS
|
|
#define VC_EXTRA_LEAN
|
|
#if !defined(WIN32_LEAN_AND_MEAN)
|
|
#define WIN32_LEAN_AND_MEAN
|
|
#endif
|
|
#include <windows.h>
|
|
#endif
|
|
|
|
|
|
namespace chaiscript
|
|
{
|
|
namespace detail
|
|
{
|
|
struct Loadable_Module
|
|
{
|
|
template<typename T>
|
|
static std::wstring to_wstring(const T &t_str)
|
|
{
|
|
return std::wstring(t_str.begin(), t_str.end());
|
|
}
|
|
|
|
template<typename T>
|
|
static std::string to_string(const T &t_str)
|
|
{
|
|
return std::string(t_str.begin(), t_str.end());
|
|
}
|
|
|
|
#if defined(_UNICODE) || defined(UNICODE)
|
|
template<typename T>
|
|
static std::wstring to_proper_string(const T &t_str)
|
|
{
|
|
return to_wstring(t_str);
|
|
}
|
|
#else
|
|
template<typename T>
|
|
static std::string to_proper_string(const T &t_str)
|
|
{
|
|
return to_string(t_str);
|
|
}
|
|
#endif
|
|
|
|
static std::string get_error_message(DWORD t_err)
|
|
{
|
|
typedef LPTSTR StringType;
|
|
|
|
#if defined(_UNICODE) || defined(UNICODE)
|
|
std::wstring retval = L"Unknown Error";
|
|
#else
|
|
std::string retval = "Unknown Error";
|
|
#endif
|
|
StringType lpMsgBuf = nullptr;
|
|
|
|
if (FormatMessage(
|
|
FORMAT_MESSAGE_ALLOCATE_BUFFER |
|
|
FORMAT_MESSAGE_FROM_SYSTEM |
|
|
FORMAT_MESSAGE_IGNORE_INSERTS,
|
|
nullptr,
|
|
t_err,
|
|
MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT),
|
|
reinterpret_cast<StringType>(&lpMsgBuf),
|
|
0, nullptr ) != 0 && lpMsgBuf)
|
|
{
|
|
retval = lpMsgBuf;
|
|
LocalFree(lpMsgBuf);
|
|
}
|
|
|
|
return to_string(retval);
|
|
}
|
|
|
|
struct DLModule
|
|
{
|
|
DLModule(const std::string &t_filename)
|
|
: m_data(LoadLibrary(to_proper_string(t_filename).c_str()))
|
|
{
|
|
if (!m_data)
|
|
{
|
|
throw chaiscript::exception::load_module_error(get_error_message(GetLastError()));
|
|
}
|
|
}
|
|
|
|
DLModule(DLModule &&) = default;
|
|
DLModule &operator=(DLModule &&) = default;
|
|
DLModule(const DLModule &) = delete;
|
|
DLModule &operator=(const DLModule &) = delete;
|
|
|
|
~DLModule()
|
|
{
|
|
FreeLibrary(m_data);
|
|
}
|
|
|
|
HMODULE m_data;
|
|
};
|
|
|
|
template<typename T>
|
|
struct DLSym
|
|
{
|
|
DLSym(DLModule &t_mod, const std::string &t_symbol)
|
|
: m_symbol(reinterpret_cast<T>(GetProcAddress(t_mod.m_data, t_symbol.c_str())))
|
|
{
|
|
if (!m_symbol)
|
|
{
|
|
throw chaiscript::exception::load_module_error(get_error_message(GetLastError()));
|
|
}
|
|
}
|
|
|
|
T m_symbol;
|
|
};
|
|
|
|
Loadable_Module(const std::string &t_module_name, const std::string &t_filename)
|
|
: m_dlmodule(t_filename), m_func(m_dlmodule, "create_chaiscript_module_" + t_module_name),
|
|
m_moduleptr(m_func.m_symbol())
|
|
{
|
|
}
|
|
|
|
DLModule m_dlmodule;
|
|
DLSym<Create_Module_Func> m_func;
|
|
ModulePtr m_moduleptr;
|
|
};
|
|
}
|
|
}
|
|
#endif
|
|
|