blob: 4e2c0e65641f421946ba6b568f55e281d304fe02 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
|
// **************************************************************************
// * This file is part of the FreeFileSync project. It is distributed under *
// * GNU General Public License: http://www.gnu.org/licenses/gpl.html *
// * Copyright (C) 2008-2011 ZenJu (zhnmju123 AT gmx.de) *
// **************************************************************************
#include "dll_loader.h"
#include <map>
namespace
{
class DllHandler
{
public:
static DllHandler& getInstance()
{
static DllHandler instance;
return instance;
}
HMODULE getHandle(const std::wstring& libraryName)
{
if (libraryName.empty())
return ::GetModuleHandle(NULL); //return handle to calling executable
HandleMap::const_iterator iter = handles.find(libraryName);
if (iter != handles.end())
return iter->second;
HMODULE newHandle = ::LoadLibrary(libraryName.c_str());
if (newHandle != NULL)
handles.insert(std::make_pair(libraryName, newHandle));
return newHandle;
}
private:
DllHandler() {}
DllHandler(const DllHandler&);
DllHandler& operator=(const DllHandler&);
~DllHandler()
{
for (HandleMap::const_iterator i = handles.begin(); i != handles.end(); ++i)
::FreeLibrary(i->second);
}
typedef std::map<std::wstring, HMODULE> HandleMap;
HandleMap handles; //only valid handles here!
};
}
FARPROC util::loadSymbol(const std::wstring& libraryName, const std::string& functionName)
{
const HMODULE libHandle = DllHandler::getInstance().getHandle(libraryName);
return libHandle != NULL ? ::GetProcAddress(libHandle, functionName.c_str()) : NULL;
}
std::string util::getResourceStream(const std::wstring& libraryName, size_t resourceId)
{
std::string output;
const HMODULE module = DllHandler::getInstance().getHandle(libraryName);
if (module)
{
const HRSRC res = ::FindResource(module, MAKEINTRESOURCE(resourceId), RT_RCDATA);
if (res != NULL)
{
const HGLOBAL resHandle = ::LoadResource(module, res);
if (resHandle != NULL)
{
const char* stream = static_cast<const char*>(::LockResource(resHandle));
if (stream)
{
const DWORD streamSize = ::SizeofResource(module, res);
output.assign(stream, streamSize);
}
}
}
}
return output;
}
|
