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
121
|
// **************************************************************************
// * 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) Zenju (zenju AT gmx DOT de) - All Rights Reserved *
// **************************************************************************
#ifndef DLLLOADER_H_INCLUDED
#define DLLLOADER_H_INCLUDED
#include <memory>
#ifdef FFS_WIN
#include <string>
#include "scope_guard.h"
#include "win.h" //includes "windows.h"
#elif defined FFS_LINUX || defined FFS_MAC
#include <dlfcn.h>
#endif
namespace zen
{
/*
Manage DLL function and library ownership
- thread safety: like built-in type
- full value semantics
Usage:
typedef BOOL (WINAPI* FunType_IsWow64Process)(HANDLE hProcess, PBOOL Wow64Process);
const zen::SysDllFun<FunType_IsWow64Process> isWow64Process(L"kernel32.dll", "IsWow64Process");
if (isWow64Process) ... use function ptr ...
Usage 2:
#define DEF_DLL_FUN(name) DllFun<dll_ns::FunType_##name> name(dll_ns::getDllName(), dll_ns::funName_##name);
DEF_DLL_FUN(funname1); DEF_DLL_FUN(funname2); DEF_DLL_FUN(funname3);
*/
template <class Func>
class DllFun
{
public:
DllFun() : fun(nullptr) {}
#ifdef FFS_WIN
DllFun(const wchar_t* libraryName, const char* functionName) :
hLibRef(::LoadLibrary(libraryName), ::FreeLibrary),
fun(hLibRef ? reinterpret_cast<Func>(::GetProcAddress(static_cast<HMODULE>(hLibRef.get()), functionName)) : nullptr) {}
#elif defined FFS_LINUX || defined FFS_MAC
DllFun(const char* libraryName, const char* functionName) :
hLibRef(::dlopen(libraryName, RTLD_LAZY), ::dlclose),
fun(hLibRef ? reinterpret_cast<Func>(::dlsym(hLibRef.get(), functionName)) : nullptr) {}
#endif
operator Func() const { return fun; }
private:
std::shared_ptr<void> hLibRef; //we would prefer decltype(*HMODULE()) if only it would work...
Func fun;
};
#ifdef FFS_WIN
//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(nullptr) {}
SysDllFun(const wchar_t* systemLibrary, const char* functionName)
{
HMODULE mod = ::GetModuleHandle(systemLibrary);
fun = mod ? reinterpret_cast<Func>(::GetProcAddress(mod, functionName)) : nullptr;
}
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);
#endif
//--------------- implementation---------------------------------------------------
#ifdef FFS_WIN
inline
std::string getResourceStream(const wchar_t* libraryName, size_t resourceId)
{
if (HMODULE module = ::LoadLibrary(libraryName))
{
ZEN_ON_SCOPE_EXIT(::FreeLibrary(module));
if (HRSRC res = ::FindResource(module, MAKEINTRESOURCE(resourceId), RT_RCDATA))
if (HGLOBAL resHandle = ::LoadResource(module, res))
if (const char* stream = static_cast<const char*>(::LockResource(resHandle)))
return std::string(stream, static_cast<size_t>(::SizeofResource(module, res))); //size is 0 on error
}
return std::string();
}
#endif
}
#endif // DLLLOADER_H_INCLUDED
|
