blob: 3d81be97e39825f970aec2c646fd9a9c08c82b84 (
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
|
#pragma once
#include <wintun/wintun.h>
#include <libcommon/error.h>
#include "util.h"
class WintunDll
{
public:
WintunDll() : dllHandle(nullptr)
{
auto wintunPath = GetProcessModulePath().replace_filename(L"wintun.dll");
dllHandle = LoadLibraryExW(wintunPath.c_str(), nullptr, LOAD_WITH_ALTERED_SEARCH_PATH);
if (nullptr == dllHandle)
{
THROW_WINDOWS_ERROR(GetLastError(), "LoadLibraryExW");
}
try
{
createAdapter = getProcAddressOrThrow<WINTUN_CREATE_ADAPTER_FUNC>("WintunCreateAdapter");
openAdapter = getProcAddressOrThrow<WINTUN_OPEN_ADAPTER_FUNC>("WintunOpenAdapter");
freeAdapter = getProcAddressOrThrow<WINTUN_FREE_ADAPTER_FUNC>("WintunFreeAdapter");
deletePoolDriver = getProcAddressOrThrow<WINTUN_DELETE_POOL_DRIVER_FUNC>("WintunDeletePoolDriver");
}
catch (...)
{
FreeLibrary(dllHandle);
throw;
}
}
~WintunDll()
{
if (nullptr != dllHandle)
{
FreeLibrary(dllHandle);
}
}
WINTUN_CREATE_ADAPTER_FUNC createAdapter;
WINTUN_OPEN_ADAPTER_FUNC openAdapter;
WINTUN_FREE_ADAPTER_FUNC freeAdapter;
WINTUN_DELETE_POOL_DRIVER_FUNC deletePoolDriver;
private:
template<typename T>
T getProcAddressOrThrow(const char *procName)
{
const T result = reinterpret_cast<T>(GetProcAddress(dllHandle, procName));
if (nullptr == result)
{
THROW_WINDOWS_ERROR(GetLastError(), "GetProcAddress");
}
return result;
}
HMODULE dllHandle;
};
|
