2020-07-03 12:26:06 +02:00
|
|
|
/* SPDX-License-Identifier: GPL-2.0
|
|
|
|
*
|
|
|
|
* Copyright (C) 2018-2020 WireGuard LLC. All Rights Reserved.
|
|
|
|
*/
|
|
|
|
|
2020-07-21 16:57:36 +02:00
|
|
|
#include "pch.h"
|
2020-07-03 12:26:06 +02:00
|
|
|
|
|
|
|
HINSTANCE ResourceModule;
|
|
|
|
|
2020-07-21 18:19:15 +02:00
|
|
|
WINTUN_STATUS WINAPI
|
|
|
|
WintunGetVersion(
|
|
|
|
_Out_ DWORD *DriverVersionMaj,
|
|
|
|
_Out_ DWORD *DriverVersionMin,
|
|
|
|
_Out_ DWORD *NdisVersionMaj,
|
|
|
|
_Out_ DWORD *NdisVersionMin)
|
|
|
|
{
|
|
|
|
HKEY Key;
|
|
|
|
DWORD Result =
|
|
|
|
RegOpenKeyExW(HKEY_LOCAL_MACHINE, L"SYSTEM\\CurrentControlSet\\Services\\Wintun", 0, KEY_QUERY_VALUE, &Key);
|
|
|
|
if (Result != ERROR_SUCCESS)
|
2020-10-14 13:04:29 +02:00
|
|
|
return LOG_ERROR(L"Failed to open registry key", Result);
|
2020-07-21 18:19:15 +02:00
|
|
|
Result = RegistryQueryDWORD(Key, L"DriverMajorVersion", DriverVersionMaj);
|
|
|
|
if (Result != ERROR_SUCCESS)
|
2020-10-13 19:40:52 +02:00
|
|
|
{
|
2020-10-15 12:21:55 +02:00
|
|
|
LOG(WINTUN_LOG_ERR, L"Failed to query DriverMajorVersion value");
|
2020-07-21 18:19:15 +02:00
|
|
|
goto cleanupKey;
|
2020-10-13 19:40:52 +02:00
|
|
|
}
|
2020-07-21 18:19:15 +02:00
|
|
|
Result = RegistryQueryDWORD(Key, L"DriverMinorVersion", DriverVersionMin);
|
|
|
|
if (Result != ERROR_SUCCESS)
|
2020-10-13 19:40:52 +02:00
|
|
|
{
|
2020-10-15 12:21:55 +02:00
|
|
|
LOG(WINTUN_LOG_ERR, L"Failed to query DriverMinorVersion value");
|
2020-07-21 18:19:15 +02:00
|
|
|
goto cleanupKey;
|
2020-10-13 19:40:52 +02:00
|
|
|
}
|
2020-07-21 18:19:15 +02:00
|
|
|
Result = RegistryQueryDWORD(Key, L"NdisMajorVersion", NdisVersionMaj);
|
|
|
|
if (Result != ERROR_SUCCESS)
|
2020-10-13 19:40:52 +02:00
|
|
|
{
|
2020-10-15 12:21:55 +02:00
|
|
|
LOG(WINTUN_LOG_ERR, L"Failed to query NdisMajorVersion value");
|
2020-07-21 18:19:15 +02:00
|
|
|
goto cleanupKey;
|
2020-10-13 19:40:52 +02:00
|
|
|
}
|
2020-07-21 18:19:15 +02:00
|
|
|
Result = RegistryQueryDWORD(Key, L"NdisMinorVersion", NdisVersionMin);
|
2020-10-13 19:40:52 +02:00
|
|
|
if (Result != ERROR_SUCCESS)
|
2020-10-15 12:21:55 +02:00
|
|
|
LOG(WINTUN_LOG_ERR, L"Failed to query NdisMinorVersion value");
|
2020-07-21 18:19:15 +02:00
|
|
|
cleanupKey:
|
|
|
|
RegCloseKey(Key);
|
|
|
|
return Result;
|
|
|
|
}
|
|
|
|
|
2020-07-03 12:26:06 +02:00
|
|
|
BOOL APIENTRY
|
|
|
|
DllMain(_In_ HINSTANCE hinstDLL, _In_ DWORD fdwReason, _In_ LPVOID lpvReserved)
|
|
|
|
{
|
|
|
|
UNREFERENCED_PARAMETER(lpvReserved);
|
|
|
|
|
|
|
|
switch (fdwReason)
|
|
|
|
{
|
|
|
|
case DLL_PROCESS_ATTACH:
|
|
|
|
ResourceModule = hinstDLL;
|
2020-10-15 11:32:06 +02:00
|
|
|
AdapterInit();
|
2020-07-03 16:49:47 +02:00
|
|
|
NamespaceInit();
|
2020-07-07 15:32:06 +02:00
|
|
|
NciInit();
|
2020-07-03 16:49:47 +02:00
|
|
|
break;
|
|
|
|
|
2020-07-03 12:26:06 +02:00
|
|
|
case DLL_PROCESS_DETACH:
|
2020-07-07 15:32:06 +02:00
|
|
|
NciCleanup();
|
2020-07-03 16:49:47 +02:00
|
|
|
NamespaceCleanup();
|
2020-10-15 11:32:06 +02:00
|
|
|
AdapterCleanup();
|
2020-07-03 12:26:06 +02:00
|
|
|
break;
|
|
|
|
}
|
|
|
|
return TRUE;
|
|
|
|
}
|