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
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
|
// **************************************************************************
// * 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) *
// **************************************************************************
#ifndef DLLLOADER_H_INCLUDED
#define DLLLOADER_H_INCLUDED
#include <memory>
#include <string>
#include "scope_guard.h"
#include "win.h" //includes "windows.h"
namespace zen
{
/*
Manage DLL function and library ownership
- thread safety: like built-in type
- full value semantics
Usage:
typedef BOOL (WINAPI *IsWow64ProcessFun)(HANDLE hProcess, PBOOL Wow64Process);
const zen::DllFun<IsWow64ProcessFun> isWow64Process(L"kernel32.dll", "IsWow64Process");
if (isWow64Process) ... use function ptr ...
*/
template <class Func>
class DllFun
{
public:
DllFun() : fun(NULL) {}
DllFun(const wchar_t* libraryName, const char* functionName) :
hLibRef(new HMODULE(::LoadLibrary(libraryName)), deleter),
fun(*hLibRef ? reinterpret_cast<Func>(::GetProcAddress(*hLibRef, functionName)) : NULL) {}
operator Func() const { return fun; }
private:
static void deleter(HMODULE* ptr) { if (*ptr) ::FreeLibrary(*ptr); delete ptr; }
std::shared_ptr<const HMODULE> hLibRef;
Func fun;
};
//if the dll is already part of the process space, e.g. "kernel32.dll" or "shell32.dll", we can use a faster variant:
//NOTE: since the lifetime of the referenced library is *not* controlled, this is safe to use only for permanently loaded libraries like these!
template <class Func>
class SysDllFun
{
public:
SysDllFun() : fun(NULL) {}
SysDllFun(const wchar_t* systemLibrary, const char* functionName) :
fun(reinterpret_cast<Func>(::GetProcAddress(::GetModuleHandle(systemLibrary), functionName))) {}
operator Func() const { return fun; }
private:
Func fun;
};
/*
extract binary resources from .exe/.dll:
-- resource.h --
#define MY_BINARY_RESOURCE 1337
-- resource.rc --
MY_BINARY_RESOURCE RCDATA "filename.dat"
*/
std::string getResourceStream(const std::wstring& libraryName, size_t resourceId);
//---------------Inline Implementation---------------------------------------------------
inline
std::string getResourceStream(const wchar_t* libraryName, size_t resourceId)
{
std::string output;
HMODULE module = ::LoadLibrary(libraryName);
if (module)
{
ZEN_ON_BLOCK_EXIT(::FreeLibrary(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;
}
}
#endif // DLLLOADER_H_INCLUDED
|
