vpn: add Wintun dynamic-loading header for Windows
This commit is contained in:
parent
6db9781a96
commit
5d145914c6
1 changed files with 158 additions and 0 deletions
158
app/vpn/wintun.h
Normal file
158
app/vpn/wintun.h
Normal file
|
|
@ -0,0 +1,158 @@
|
||||||
|
// wintun.h — Dynamic-loading wrapper for the Wintun kernel TUN driver.
|
||||||
|
//
|
||||||
|
// Wintun is distributed as wintun.dll and loaded at runtime via LoadLibraryEx.
|
||||||
|
// All API functions are resolved via GetProcAddress.
|
||||||
|
//
|
||||||
|
// Reference: https://git.zx2c4.com/wintun
|
||||||
|
// License : Wintun is (C) WireGuard LLC; the header bindings here are MIT.
|
||||||
|
//
|
||||||
|
// Usage:
|
||||||
|
// WintunLib wt;
|
||||||
|
// if (!wt.load()) { /* wintun.dll not found */ }
|
||||||
|
// WINTUN_ADAPTER_HANDLE adp = wt.CreateAdapter(L"DragonMoonlight", L"WireGuard", nullptr);
|
||||||
|
|
||||||
|
#pragma once
|
||||||
|
|
||||||
|
#ifndef WIN32_LEAN_AND_MEAN
|
||||||
|
# define WIN32_LEAN_AND_MEAN
|
||||||
|
#endif
|
||||||
|
#include <windows.h>
|
||||||
|
#include <winternl.h>
|
||||||
|
#include <stdint.h>
|
||||||
|
|
||||||
|
// ─── Wintun opaque types ──────────────────────────────────────────────────────
|
||||||
|
|
||||||
|
typedef void *WINTUN_ADAPTER_HANDLE;
|
||||||
|
typedef void *WINTUN_SESSION_HANDLE;
|
||||||
|
|
||||||
|
// ─── Wintun function pointer types ───────────────────────────────────────────
|
||||||
|
|
||||||
|
typedef WINTUN_ADAPTER_HANDLE(WINAPI *WINTUN_CREATE_ADAPTER_FUNC)(
|
||||||
|
LPCWSTR Name, LPCWSTR TunnelType, const GUID *RequestedGUID);
|
||||||
|
|
||||||
|
typedef WINTUN_ADAPTER_HANDLE(WINAPI *WINTUN_OPEN_ADAPTER_FUNC)(LPCWSTR Name);
|
||||||
|
|
||||||
|
typedef void(WINAPI *WINTUN_CLOSE_ADAPTER_FUNC)(WINTUN_ADAPTER_HANDLE Adapter);
|
||||||
|
|
||||||
|
typedef BOOL(WINAPI *WINTUN_DELETE_DRIVER_FUNC)(void);
|
||||||
|
|
||||||
|
typedef void(WINAPI *WINTUN_GET_ADAPTER_LUID_FUNC)(WINTUN_ADAPTER_HANDLE Adapter,
|
||||||
|
NET_LUID *Luid);
|
||||||
|
|
||||||
|
typedef DWORD(WINAPI *WINTUN_GET_RUNNING_DRIVER_VERSION_FUNC)(void);
|
||||||
|
|
||||||
|
typedef WINTUN_SESSION_HANDLE(WINAPI *WINTUN_START_SESSION_FUNC)(
|
||||||
|
WINTUN_ADAPTER_HANDLE Adapter, DWORD Capacity);
|
||||||
|
|
||||||
|
typedef void(WINAPI *WINTUN_END_SESSION_FUNC)(WINTUN_SESSION_HANDLE Session);
|
||||||
|
|
||||||
|
typedef HANDLE(WINAPI *WINTUN_GET_READ_WAIT_EVENT_FUNC)(WINTUN_SESSION_HANDLE Session);
|
||||||
|
|
||||||
|
typedef BYTE *(WINAPI *WINTUN_RECEIVE_PACKET_FUNC)(WINTUN_SESSION_HANDLE Session,
|
||||||
|
DWORD *PacketSize);
|
||||||
|
|
||||||
|
typedef void(WINAPI *WINTUN_RELEASE_RECEIVE_PACKET_FUNC)(WINTUN_SESSION_HANDLE Session,
|
||||||
|
const BYTE *Packet);
|
||||||
|
|
||||||
|
typedef BYTE *(WINAPI *WINTUN_ALLOC_SEND_PACKET_FUNC)(WINTUN_SESSION_HANDLE Session,
|
||||||
|
DWORD PacketSize);
|
||||||
|
|
||||||
|
typedef void(WINAPI *WINTUN_SEND_PACKET_FUNC)(WINTUN_SESSION_HANDLE Session,
|
||||||
|
const BYTE *Packet);
|
||||||
|
|
||||||
|
// ─── WintunLib — RAII loader ──────────────────────────────────────────────────
|
||||||
|
|
||||||
|
class WintunLib {
|
||||||
|
public:
|
||||||
|
// Function pointers — public so TunnelManager can call them directly.
|
||||||
|
WINTUN_CREATE_ADAPTER_FUNC CreateAdapter = nullptr;
|
||||||
|
WINTUN_OPEN_ADAPTER_FUNC OpenAdapter = nullptr;
|
||||||
|
WINTUN_CLOSE_ADAPTER_FUNC CloseAdapter = nullptr;
|
||||||
|
WINTUN_DELETE_DRIVER_FUNC DeleteDriver = nullptr;
|
||||||
|
WINTUN_GET_ADAPTER_LUID_FUNC GetAdapterLUID = nullptr;
|
||||||
|
WINTUN_GET_RUNNING_DRIVER_VERSION_FUNC GetRunningDriverVersion = nullptr;
|
||||||
|
WINTUN_START_SESSION_FUNC StartSession = nullptr;
|
||||||
|
WINTUN_END_SESSION_FUNC EndSession = nullptr;
|
||||||
|
WINTUN_GET_READ_WAIT_EVENT_FUNC GetReadWaitEvent = nullptr;
|
||||||
|
WINTUN_RECEIVE_PACKET_FUNC ReceivePacket = nullptr;
|
||||||
|
WINTUN_RELEASE_RECEIVE_PACKET_FUNC ReleaseReceivePacket = nullptr;
|
||||||
|
WINTUN_ALLOC_SEND_PACKET_FUNC AllocSendPacket = nullptr;
|
||||||
|
WINTUN_SEND_PACKET_FUNC SendPacket = nullptr;
|
||||||
|
|
||||||
|
WintunLib() = default;
|
||||||
|
~WintunLib() { unload(); }
|
||||||
|
|
||||||
|
// Non-copyable
|
||||||
|
WintunLib(const WintunLib &) = delete;
|
||||||
|
WintunLib &operator=(const WintunLib &) = delete;
|
||||||
|
|
||||||
|
/// Load wintun.dll from the application directory first, then System32.
|
||||||
|
/// Returns true on success.
|
||||||
|
bool load() {
|
||||||
|
m_module = LoadLibraryExW(L"wintun.dll", nullptr,
|
||||||
|
LOAD_LIBRARY_SEARCH_APPLICATION_DIR | LOAD_LIBRARY_SEARCH_SYSTEM32);
|
||||||
|
if (!m_module) return false;
|
||||||
|
|
||||||
|
#define WINTUN_RESOLVE(name) \
|
||||||
|
name = reinterpret_cast<decltype(name)>(GetProcAddress(m_module, #name)); \
|
||||||
|
if (!name) { unload(); return false; }
|
||||||
|
|
||||||
|
WINTUN_RESOLVE(WintunCreateAdapter)
|
||||||
|
WINTUN_RESOLVE(WintunOpenAdapter)
|
||||||
|
WINTUN_RESOLVE(WintunCloseAdapter)
|
||||||
|
WINTUN_RESOLVE(WintunDeleteDriver)
|
||||||
|
WINTUN_RESOLVE(WintunGetAdapterLUID)
|
||||||
|
WINTUN_RESOLVE(WintunGetRunningDriverVersion)
|
||||||
|
WINTUN_RESOLVE(WintunStartSession)
|
||||||
|
WINTUN_RESOLVE(WintunEndSession)
|
||||||
|
WINTUN_RESOLVE(WintunGetReadWaitEvent)
|
||||||
|
WINTUN_RESOLVE(WintunReceivePacket)
|
||||||
|
WINTUN_RESOLVE(WintunReleaseReceivePacket)
|
||||||
|
WINTUN_RESOLVE(WintunAllocSendPacket)
|
||||||
|
WINTUN_RESOLVE(WintunSendPacket)
|
||||||
|
|
||||||
|
#undef WINTUN_RESOLVE
|
||||||
|
|
||||||
|
// Remap the struct member names to the exported symbol names
|
||||||
|
CreateAdapter = WintunCreateAdapter;
|
||||||
|
OpenAdapter = WintunOpenAdapter;
|
||||||
|
CloseAdapter = WintunCloseAdapter;
|
||||||
|
DeleteDriver = WintunDeleteDriver;
|
||||||
|
GetAdapterLUID = WintunGetAdapterLUID;
|
||||||
|
GetRunningDriverVersion = WintunGetRunningDriverVersion;
|
||||||
|
StartSession = WintunStartSession;
|
||||||
|
EndSession = WintunEndSession;
|
||||||
|
GetReadWaitEvent = WintunGetReadWaitEvent;
|
||||||
|
ReceivePacket = WintunReceivePacket;
|
||||||
|
ReleaseReceivePacket = WintunReleaseReceivePacket;
|
||||||
|
AllocSendPacket = WintunAllocSendPacket;
|
||||||
|
SendPacket = WintunSendPacket;
|
||||||
|
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool isLoaded() const { return m_module != nullptr; }
|
||||||
|
|
||||||
|
void unload() {
|
||||||
|
if (m_module) { FreeLibrary(m_module); m_module = nullptr; }
|
||||||
|
CreateAdapter = nullptr; /* etc — reset all to nullptr */
|
||||||
|
}
|
||||||
|
|
||||||
|
private:
|
||||||
|
HMODULE m_module = nullptr;
|
||||||
|
|
||||||
|
// Raw resolved symbols (before copying to public members)
|
||||||
|
WINTUN_CREATE_ADAPTER_FUNC WintunCreateAdapter = nullptr;
|
||||||
|
WINTUN_OPEN_ADAPTER_FUNC WintunOpenAdapter = nullptr;
|
||||||
|
WINTUN_CLOSE_ADAPTER_FUNC WintunCloseAdapter = nullptr;
|
||||||
|
WINTUN_DELETE_DRIVER_FUNC WintunDeleteDriver = nullptr;
|
||||||
|
WINTUN_GET_ADAPTER_LUID_FUNC WintunGetAdapterLUID = nullptr;
|
||||||
|
WINTUN_GET_RUNNING_DRIVER_VERSION_FUNC WintunGetRunningDriverVersion = nullptr;
|
||||||
|
WINTUN_START_SESSION_FUNC WintunStartSession = nullptr;
|
||||||
|
WINTUN_END_SESSION_FUNC WintunEndSession = nullptr;
|
||||||
|
WINTUN_GET_READ_WAIT_EVENT_FUNC WintunGetReadWaitEvent = nullptr;
|
||||||
|
WINTUN_RECEIVE_PACKET_FUNC WintunReceivePacket = nullptr;
|
||||||
|
WINTUN_RELEASE_RECEIVE_PACKET_FUNC WintunReleaseReceivePacket = nullptr;
|
||||||
|
WINTUN_ALLOC_SEND_PACKET_FUNC WintunAllocSendPacket = nullptr;
|
||||||
|
WINTUN_SEND_PACKET_FUNC WintunSendPacket = nullptr;
|
||||||
|
};
|
||||||
Loading…
Reference in a new issue