2013-07-04 20:56:19 +00:00
|
|
|
/*
|
|
|
|
* ZeroTier One - Global Peer to Peer Ethernet
|
2014-02-16 20:40:22 +00:00
|
|
|
* Copyright (C) 2011-2014 ZeroTier Networks LLC
|
2013-07-04 20:56:19 +00:00
|
|
|
*
|
|
|
|
* This program is free software: you can redistribute it and/or modify
|
|
|
|
* it under the terms of the GNU General Public License as published by
|
|
|
|
* the Free Software Foundation, either version 3 of the License, or
|
|
|
|
* (at your option) any later version.
|
|
|
|
*
|
|
|
|
* This program is distributed in the hope that it will be useful,
|
|
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
|
|
* GNU General Public License for more details.
|
|
|
|
*
|
|
|
|
* You should have received a copy of the GNU General Public License
|
|
|
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
*
|
|
|
|
* --
|
|
|
|
*
|
|
|
|
* ZeroTier may be used and distributed under the terms of the GPLv3, which
|
|
|
|
* are available at: http://www.gnu.org/licenses/gpl-3.0.html
|
|
|
|
*
|
|
|
|
* If you would like to embed ZeroTier into a commercial application or
|
|
|
|
* redistribute it in a modified binary form, please contact ZeroTier Networks
|
|
|
|
* LLC. Start here: http://www.zerotier.com/
|
|
|
|
*/
|
|
|
|
|
2013-08-13 18:42:51 +00:00
|
|
|
#include "Constants.hpp"
|
2014-04-08 19:00:21 +00:00
|
|
|
|
|
|
|
#include <stdio.h>
|
|
|
|
#include <stdlib.h>
|
|
|
|
#include <stdint.h>
|
|
|
|
#include <string.h>
|
|
|
|
#include <WinSock2.h>
|
|
|
|
#include <Windows.h>
|
|
|
|
#include <tchar.h>
|
|
|
|
#include <winreg.h>
|
|
|
|
#include <wchar.h>
|
|
|
|
#include <ws2ipdef.h>
|
|
|
|
#include <WS2tcpip.h>
|
|
|
|
#include <IPHlpApi.h>
|
|
|
|
#include <nldef.h>
|
|
|
|
#include <netioapi.h>
|
|
|
|
|
2013-07-04 20:56:19 +00:00
|
|
|
#include "EthernetTap.hpp"
|
2014-04-08 19:00:21 +00:00
|
|
|
#include "WindowsEthernetTap.hpp"
|
2013-07-04 20:56:19 +00:00
|
|
|
#include "Logger.hpp"
|
|
|
|
#include "RuntimeEnvironment.hpp"
|
2013-07-09 18:06:55 +00:00
|
|
|
#include "Utils.hpp"
|
2013-07-04 20:56:19 +00:00
|
|
|
#include "Mutex.hpp"
|
2014-04-08 19:00:21 +00:00
|
|
|
|
|
|
|
#include "..\windows\TapDriver\tap-windows.h"
|
2013-07-06 19:56:12 +00:00
|
|
|
|
|
|
|
// ff:ff:ff:ff:ff:ff with no ADI
|
|
|
|
static const ZeroTier::MulticastGroup _blindWildcardMulticastGroup(ZeroTier::MAC(0xff),0);
|
2013-07-04 20:56:19 +00:00
|
|
|
|
|
|
|
namespace ZeroTier {
|
|
|
|
|
2014-04-08 19:00:21 +00:00
|
|
|
// Helper function to get an adapter's LUID and index from its GUID. The LUID is
|
|
|
|
// constant but the index can change, so go ahead and just look them both up by
|
|
|
|
// the GUID which is constant. (The GUID is the instance ID in the registry.)
|
|
|
|
static inline std::pair<NET_LUID,NET_IFINDEX> _findAdapterByGuid(const GUID &guid)
|
|
|
|
throw(std::runtime_error)
|
2013-07-04 20:56:19 +00:00
|
|
|
{
|
2014-04-08 19:00:21 +00:00
|
|
|
MIB_IF_TABLE2 *ift = (MIB_IF_TABLE2 *)0;
|
2013-07-04 20:56:19 +00:00
|
|
|
|
2014-04-08 19:00:21 +00:00
|
|
|
if (GetIfTable2Ex(MibIfTableRaw,&ift) != NO_ERROR)
|
|
|
|
throw std::runtime_error("GetIfTable2Ex() failed");
|
2013-07-04 20:56:19 +00:00
|
|
|
|
2014-04-08 19:00:21 +00:00
|
|
|
for(ULONG i=0;i<ift->NumEntries;++i) {
|
|
|
|
if (ift->Table[i].InterfaceGuid == guid) {
|
|
|
|
std::pair<NET_LUID,NET_IFINDEX> tmp(ift->Table[i].InterfaceLuid,ift->Table[i].InterfaceIndex);
|
|
|
|
FreeMibTable(ift);
|
|
|
|
return tmp;
|
2014-02-02 07:10:04 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2014-04-08 19:00:21 +00:00
|
|
|
FreeMibTable(&ift);
|
2013-07-04 20:56:19 +00:00
|
|
|
|
2014-04-08 19:00:21 +00:00
|
|
|
throw std::runtime_error("interface not found");
|
2013-07-09 18:06:55 +00:00
|
|
|
}
|
2013-07-04 20:56:19 +00:00
|
|
|
|
2014-04-08 23:10:48 +00:00
|
|
|
// Only create or delete devices one at a time
|
2014-04-08 19:00:21 +00:00
|
|
|
static Mutex _systemTapInitLock;
|
2014-01-27 06:47:08 +00:00
|
|
|
|
2014-04-08 19:00:21 +00:00
|
|
|
// Compute some basic environment stuff on startup
|
|
|
|
class _WinSysEnv
|
2013-07-09 18:06:55 +00:00
|
|
|
{
|
2014-04-08 19:00:21 +00:00
|
|
|
public:
|
|
|
|
_WinSysEnv()
|
2013-07-09 18:06:55 +00:00
|
|
|
{
|
2014-04-08 19:00:21 +00:00
|
|
|
#ifdef _WIN64
|
|
|
|
is64Bit = TRUE;
|
|
|
|
devcon = "\\devcon_x64.exe";
|
|
|
|
tapDriver = "\\tap-windows\\x64\\zttap200.inf";
|
|
|
|
#else
|
|
|
|
is64Bit = FALSE;
|
|
|
|
IsWow64Process(GetCurrentProcess(),&is64Bit);
|
|
|
|
devcon = ((is64Bit == TRUE) ? "\\devcon_x64.exe" : "\\devcon_x86.exe");
|
|
|
|
tapDriver = ((is64Bit == TRUE) ? "\\tap-windows\\x64\\zttap200.inf" : "\\tap-windows\\x86\\zttap200.inf");
|
|
|
|
#endif
|
2013-07-09 18:06:55 +00:00
|
|
|
}
|
2014-04-08 19:00:21 +00:00
|
|
|
BOOL is64Bit;
|
|
|
|
const char *devcon;
|
|
|
|
const char *tapDriver;
|
2014-02-02 07:18:31 +00:00
|
|
|
};
|
2014-04-08 19:00:21 +00:00
|
|
|
static const _WinSysEnv _winEnv;
|
2014-02-02 07:18:31 +00:00
|
|
|
|
2014-04-08 19:00:21 +00:00
|
|
|
static bool _disableTapDevice(const RuntimeEnvironment *_r,const std::string deviceInstanceId)
|
2014-01-16 23:11:59 +00:00
|
|
|
{
|
2014-04-08 19:00:21 +00:00
|
|
|
HANDLE devconLog = CreateFileA((_r->homePath + "\\devcon.log").c_str(),GENERIC_WRITE,FILE_SHARE_READ|FILE_SHARE_WRITE,NULL,OPEN_ALWAYS,FILE_ATTRIBUTE_NORMAL,NULL);
|
|
|
|
if (devconLog != INVALID_HANDLE_VALUE)
|
|
|
|
SetFilePointer(devconLog,0,0,FILE_END);
|
2014-01-16 23:11:59 +00:00
|
|
|
|
2014-04-08 19:00:21 +00:00
|
|
|
STARTUPINFOA startupInfo;
|
|
|
|
startupInfo.cb = sizeof(startupInfo);
|
|
|
|
if (devconLog != INVALID_HANDLE_VALUE) {
|
|
|
|
startupInfo.hStdOutput = devconLog;
|
|
|
|
startupInfo.hStdError = devconLog;
|
2014-01-16 23:11:59 +00:00
|
|
|
}
|
2014-04-08 19:00:21 +00:00
|
|
|
PROCESS_INFORMATION processInfo;
|
|
|
|
memset(&startupInfo,0,sizeof(STARTUPINFOA));
|
|
|
|
memset(&processInfo,0,sizeof(PROCESS_INFORMATION));
|
|
|
|
if (!CreateProcessA(NULL,(LPSTR)(std::string("\"") + _r->homePath + _winEnv.devcon + "\" disable @" + deviceInstanceId).c_str(),NULL,NULL,FALSE,0,NULL,NULL,&startupInfo,&processInfo)) {
|
|
|
|
if (devconLog != INVALID_HANDLE_VALUE)
|
|
|
|
CloseHandle(devconLog);
|
|
|
|
return false;
|
2014-01-16 23:11:59 +00:00
|
|
|
}
|
2014-04-08 19:00:21 +00:00
|
|
|
WaitForSingleObject(processInfo.hProcess,INFINITE);
|
|
|
|
CloseHandle(processInfo.hProcess);
|
|
|
|
CloseHandle(processInfo.hThread);
|
2014-01-16 23:11:59 +00:00
|
|
|
|
2014-04-08 19:00:21 +00:00
|
|
|
if (devconLog != INVALID_HANDLE_VALUE)
|
|
|
|
CloseHandle(devconLog);
|
2014-01-16 23:11:59 +00:00
|
|
|
|
2014-04-08 19:00:21 +00:00
|
|
|
return true;
|
2014-01-16 23:11:59 +00:00
|
|
|
}
|
2014-04-08 19:00:21 +00:00
|
|
|
static bool _enableTapDevice(const RuntimeEnvironment *_r,const std::string deviceInstanceId)
|
2014-01-16 23:11:59 +00:00
|
|
|
{
|
2014-04-08 19:00:21 +00:00
|
|
|
HANDLE devconLog = CreateFileA((_r->homePath + "\\devcon.log").c_str(),GENERIC_WRITE,FILE_SHARE_READ|FILE_SHARE_WRITE,NULL,OPEN_ALWAYS,FILE_ATTRIBUTE_NORMAL,NULL);
|
|
|
|
if (devconLog != INVALID_HANDLE_VALUE)
|
|
|
|
SetFilePointer(devconLog,0,0,FILE_END);
|
2014-01-16 23:11:59 +00:00
|
|
|
|
2014-04-08 19:00:21 +00:00
|
|
|
STARTUPINFOA startupInfo;
|
|
|
|
startupInfo.cb = sizeof(startupInfo);
|
|
|
|
if (devconLog != INVALID_HANDLE_VALUE) {
|
|
|
|
startupInfo.hStdOutput = devconLog;
|
|
|
|
startupInfo.hStdError = devconLog;
|
2013-07-04 20:56:19 +00:00
|
|
|
}
|
2014-04-08 19:00:21 +00:00
|
|
|
PROCESS_INFORMATION processInfo;
|
|
|
|
memset(&startupInfo,0,sizeof(STARTUPINFOA));
|
|
|
|
memset(&processInfo,0,sizeof(PROCESS_INFORMATION));
|
|
|
|
if (!CreateProcessA(NULL,(LPSTR)(std::string("\"") + _r->homePath + _winEnv.devcon + "\" enable @" + deviceInstanceId).c_str(),NULL,NULL,FALSE,0,NULL,NULL,&startupInfo,&processInfo)) {
|
|
|
|
if (devconLog != INVALID_HANDLE_VALUE)
|
|
|
|
CloseHandle(devconLog);
|
|
|
|
return false;
|
2013-07-04 20:56:19 +00:00
|
|
|
}
|
2014-04-08 19:00:21 +00:00
|
|
|
WaitForSingleObject(processInfo.hProcess,INFINITE);
|
|
|
|
CloseHandle(processInfo.hProcess);
|
|
|
|
CloseHandle(processInfo.hThread);
|
2013-07-04 20:56:19 +00:00
|
|
|
|
2014-04-08 19:00:21 +00:00
|
|
|
if (devconLog != INVALID_HANDLE_VALUE)
|
|
|
|
CloseHandle(devconLog);
|
2013-07-04 20:56:19 +00:00
|
|
|
|
2014-04-08 19:00:21 +00:00
|
|
|
return true;
|
2013-07-04 20:56:19 +00:00
|
|
|
}
|
2013-07-09 18:06:55 +00:00
|
|
|
|
2014-04-08 19:00:21 +00:00
|
|
|
static void _syncIpsWithRegistry(const std::set<InetAddress> &haveIps,const std::string netCfgInstanceId)
|
2013-07-09 18:06:55 +00:00
|
|
|
{
|
2014-04-08 19:00:21 +00:00
|
|
|
// Update registry to contain all non-link-local IPs for this interface
|
|
|
|
std::string regMultiIps,regMultiNetmasks;
|
|
|
|
for(std::set<InetAddress>::const_iterator i(haveIps.begin());i!=haveIps.end();++i) {
|
|
|
|
if (!i->isLinkLocal()) {
|
|
|
|
regMultiIps.append(i->toIpString());
|
|
|
|
regMultiIps.push_back((char)0);
|
|
|
|
regMultiNetmasks.append(i->netmask().toIpString());
|
|
|
|
regMultiNetmasks.push_back((char)0);
|
2013-07-09 18:06:55 +00:00
|
|
|
}
|
|
|
|
}
|
2014-04-08 19:00:21 +00:00
|
|
|
HKEY tcpIpInterfaces;
|
|
|
|
if (RegOpenKeyExA(HKEY_LOCAL_MACHINE,"SYSTEM\\CurrentControlSet\\services\\Tcpip\\Parameters\\Interfaces",0,KEY_READ|KEY_WRITE,&tcpIpInterfaces) == ERROR_SUCCESS) {
|
|
|
|
if (regMultiIps.length()) {
|
|
|
|
regMultiIps.push_back((char)0);
|
|
|
|
regMultiNetmasks.push_back((char)0);
|
|
|
|
RegSetKeyValueA(tcpIpInterfaces,netCfgInstanceId.c_str(),"IPAddress",REG_MULTI_SZ,regMultiIps.data(),(DWORD)regMultiIps.length());
|
|
|
|
RegSetKeyValueA(tcpIpInterfaces,netCfgInstanceId.c_str(),"SubnetMask",REG_MULTI_SZ,regMultiNetmasks.data(),(DWORD)regMultiNetmasks.length());
|
|
|
|
} else {
|
|
|
|
RegDeleteKeyValueA(tcpIpInterfaces,netCfgInstanceId.c_str(),"IPAddress");
|
|
|
|
RegDeleteKeyValueA(tcpIpInterfaces,netCfgInstanceId.c_str(),"SubnetMask");
|
2013-08-26 21:22:20 +00:00
|
|
|
}
|
|
|
|
}
|
2014-04-08 19:00:21 +00:00
|
|
|
RegCloseKey(tcpIpInterfaces);
|
2013-08-26 21:22:20 +00:00
|
|
|
}
|
|
|
|
|
2014-04-08 19:00:21 +00:00
|
|
|
WindowsEthernetTap::WindowsEthernetTap(
|
2013-08-14 17:23:25 +00:00
|
|
|
const RuntimeEnvironment *renv,
|
2013-08-21 14:18:05 +00:00
|
|
|
const char *tag,
|
2013-08-14 17:23:25 +00:00
|
|
|
const MAC &mac,
|
|
|
|
unsigned int mtu,
|
|
|
|
void (*handler)(void *,const MAC &,const MAC &,unsigned int,const Buffer<4096> &),
|
|
|
|
void *arg)
|
|
|
|
throw(std::runtime_error) :
|
2014-04-08 19:00:21 +00:00
|
|
|
EthernetTap("WindowsEthernetTap",mac,mtu),
|
2013-08-14 17:23:25 +00:00
|
|
|
_r(renv),
|
|
|
|
_handler(handler),
|
2013-08-25 22:18:02 +00:00
|
|
|
_arg(arg),
|
|
|
|
_tap(INVALID_HANDLE_VALUE),
|
|
|
|
_injectSemaphore(INVALID_HANDLE_VALUE),
|
2014-01-28 07:13:36 +00:00
|
|
|
_run(true),
|
2014-04-08 19:00:21 +00:00
|
|
|
_initialized(false),
|
2014-04-08 22:47:33 +00:00
|
|
|
_enabled(true)
|
2013-08-14 17:23:25 +00:00
|
|
|
{
|
2013-08-23 21:39:21 +00:00
|
|
|
char subkeyName[4096];
|
|
|
|
char subkeyClass[4096];
|
|
|
|
char data[4096];
|
2013-08-22 18:30:55 +00:00
|
|
|
|
2013-08-26 21:22:20 +00:00
|
|
|
if (mtu > ZT_IF_MTU)
|
|
|
|
throw std::runtime_error("MTU too large for Windows tap");
|
|
|
|
|
2014-04-08 23:10:48 +00:00
|
|
|
Mutex::Lock _l(_systemTapInitLock);
|
2013-08-22 18:30:55 +00:00
|
|
|
|
2013-08-21 14:18:05 +00:00
|
|
|
HKEY nwAdapters;
|
2013-08-22 18:30:55 +00:00
|
|
|
if (RegOpenKeyExA(HKEY_LOCAL_MACHINE,"SYSTEM\\CurrentControlSet\\Control\\Class\\{4D36E972-E325-11CE-BFC1-08002BE10318}",0,KEY_READ|KEY_WRITE,&nwAdapters) != ERROR_SUCCESS)
|
2013-08-21 14:18:05 +00:00
|
|
|
throw std::runtime_error("unable to open registry key for network adapter enumeration");
|
|
|
|
|
2013-08-22 18:30:55 +00:00
|
|
|
std::set<std::string> existingDeviceInstances;
|
2013-08-23 21:39:21 +00:00
|
|
|
std::string mySubkeyName;
|
2013-08-22 18:30:55 +00:00
|
|
|
|
2014-03-06 22:06:31 +00:00
|
|
|
// Look for the tap instance that corresponds with our interface tag (network ID)
|
|
|
|
for(DWORD subkeyIndex=0;;++subkeyIndex) {
|
2013-08-22 18:30:55 +00:00
|
|
|
DWORD type;
|
|
|
|
DWORD dataLen;
|
2013-08-21 14:18:05 +00:00
|
|
|
DWORD subkeyNameLen = sizeof(subkeyName);
|
|
|
|
DWORD subkeyClassLen = sizeof(subkeyClass);
|
|
|
|
FILETIME lastWriteTime;
|
2014-03-06 22:06:31 +00:00
|
|
|
if (RegEnumKeyExA(nwAdapters,subkeyIndex,subkeyName,&subkeyNameLen,(DWORD *)0,subkeyClass,&subkeyClassLen,&lastWriteTime) == ERROR_SUCCESS) {
|
|
|
|
type = 0;
|
|
|
|
dataLen = sizeof(data);
|
|
|
|
if (RegGetValueA(nwAdapters,subkeyName,"ComponentId",RRF_RT_ANY,&type,(PVOID)data,&dataLen) == ERROR_SUCCESS) {
|
|
|
|
data[dataLen] = '\0';
|
|
|
|
if (!strnicmp(data,"zttap",5)) {
|
|
|
|
std::string instanceId;
|
|
|
|
type = 0;
|
|
|
|
dataLen = sizeof(data);
|
|
|
|
if (RegGetValueA(nwAdapters,subkeyName,"NetCfgInstanceId",RRF_RT_ANY,&type,(PVOID)data,&dataLen) == ERROR_SUCCESS) {
|
|
|
|
instanceId.assign(data,dataLen);
|
|
|
|
existingDeviceInstances.insert(instanceId);
|
|
|
|
}
|
2013-08-22 18:30:55 +00:00
|
|
|
|
2014-03-06 22:06:31 +00:00
|
|
|
std::string instanceIdPath;
|
|
|
|
type = 0;
|
|
|
|
dataLen = sizeof(data);
|
|
|
|
if (RegGetValueA(nwAdapters,subkeyName,"DeviceInstanceID",RRF_RT_ANY,&type,(PVOID)data,&dataLen) == ERROR_SUCCESS)
|
|
|
|
instanceIdPath.assign(data,dataLen);
|
|
|
|
|
2014-04-08 19:00:21 +00:00
|
|
|
if ((_netCfgInstanceId.length() == 0)&&(instanceId.length() != 0)&&(instanceIdPath.length() != 0)) {
|
2013-08-26 21:48:47 +00:00
|
|
|
type = 0;
|
|
|
|
dataLen = sizeof(data);
|
2014-03-06 22:06:31 +00:00
|
|
|
if (RegGetValueA(nwAdapters,subkeyName,"_ZeroTierTapIdentifier",RRF_RT_ANY,&type,(PVOID)data,&dataLen) == ERROR_SUCCESS) {
|
|
|
|
data[dataLen] = '\0';
|
|
|
|
if (!strcmp(data,tag)) {
|
2014-04-08 19:00:21 +00:00
|
|
|
_netCfgInstanceId = instanceId;
|
|
|
|
_deviceInstanceId = instanceIdPath;
|
2014-03-06 22:06:31 +00:00
|
|
|
mySubkeyName = subkeyName;
|
|
|
|
break; // found it!
|
2013-08-22 18:30:55 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
2013-08-21 14:18:05 +00:00
|
|
|
}
|
2014-03-06 22:06:31 +00:00
|
|
|
}
|
|
|
|
} else break; // no more subkeys or error occurred enumerating them
|
2013-08-22 18:30:55 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
// If there is no device, try to create one
|
2014-04-08 19:00:21 +00:00
|
|
|
if (_netCfgInstanceId.length() == 0) {
|
2014-03-06 22:06:31 +00:00
|
|
|
// Log devcon output to a file
|
|
|
|
HANDLE devconLog = CreateFileA((_r->homePath + "\\devcon.log").c_str(),GENERIC_WRITE,FILE_SHARE_READ|FILE_SHARE_WRITE,NULL,OPEN_ALWAYS,FILE_ATTRIBUTE_NORMAL,NULL);
|
|
|
|
if (devconLog == INVALID_HANDLE_VALUE) {
|
|
|
|
LOG("WARNING: unable to open devcon.log");
|
|
|
|
} else {
|
|
|
|
SetFilePointer(devconLog,0,0,FILE_END);
|
|
|
|
}
|
|
|
|
|
2013-08-22 18:30:55 +00:00
|
|
|
// Execute devcon to install an instance of the Microsoft Loopback Adapter
|
|
|
|
STARTUPINFOA startupInfo;
|
|
|
|
startupInfo.cb = sizeof(startupInfo);
|
2014-03-06 22:06:31 +00:00
|
|
|
if (devconLog != INVALID_HANDLE_VALUE) {
|
|
|
|
SetFilePointer(devconLog,0,0,FILE_END);
|
|
|
|
startupInfo.hStdOutput = devconLog;
|
|
|
|
startupInfo.hStdError = devconLog;
|
|
|
|
}
|
2013-08-22 18:30:55 +00:00
|
|
|
PROCESS_INFORMATION processInfo;
|
|
|
|
memset(&startupInfo,0,sizeof(STARTUPINFOA));
|
|
|
|
memset(&processInfo,0,sizeof(PROCESS_INFORMATION));
|
2014-04-08 19:00:21 +00:00
|
|
|
if (!CreateProcessA(NULL,(LPSTR)(std::string("\"") + _r->homePath + _winEnv.devcon + "\" install \"" + _r->homePath + _winEnv.tapDriver + "\" zttap200").c_str(),NULL,NULL,FALSE,0,NULL,NULL,&startupInfo,&processInfo)) {
|
2013-08-22 18:30:55 +00:00
|
|
|
RegCloseKey(nwAdapters);
|
2014-03-06 22:06:31 +00:00
|
|
|
if (devconLog != INVALID_HANDLE_VALUE)
|
|
|
|
CloseHandle(devconLog);
|
2014-04-08 19:00:21 +00:00
|
|
|
throw std::runtime_error(std::string("unable to find or execute devcon at ") + _winEnv.devcon);
|
2013-08-22 18:30:55 +00:00
|
|
|
}
|
|
|
|
WaitForSingleObject(processInfo.hProcess,INFINITE);
|
|
|
|
CloseHandle(processInfo.hProcess);
|
|
|
|
CloseHandle(processInfo.hThread);
|
2014-03-06 22:06:31 +00:00
|
|
|
|
|
|
|
if (devconLog != INVALID_HANDLE_VALUE)
|
|
|
|
CloseHandle(devconLog);
|
|
|
|
|
2013-08-25 22:18:02 +00:00
|
|
|
// Scan for the new instance by simply looking for taps that weren't
|
2014-03-06 22:06:31 +00:00
|
|
|
// there originally. The static mutex we lock ensures this can't step
|
|
|
|
// on its own toes.
|
|
|
|
for(DWORD subkeyIndex=0;;++subkeyIndex) {
|
2013-08-22 18:30:55 +00:00
|
|
|
DWORD type;
|
|
|
|
DWORD dataLen;
|
|
|
|
DWORD subkeyNameLen = sizeof(subkeyName);
|
|
|
|
DWORD subkeyClassLen = sizeof(subkeyClass);
|
|
|
|
FILETIME lastWriteTime;
|
2014-03-06 22:06:31 +00:00
|
|
|
if (RegEnumKeyExA(nwAdapters,subkeyIndex,subkeyName,&subkeyNameLen,(DWORD *)0,subkeyClass,&subkeyClassLen,&lastWriteTime) == ERROR_SUCCESS) {
|
|
|
|
type = 0;
|
|
|
|
dataLen = sizeof(data);
|
|
|
|
if (RegGetValueA(nwAdapters,subkeyName,"ComponentId",RRF_RT_ANY,&type,(PVOID)data,&dataLen) == ERROR_SUCCESS) {
|
|
|
|
data[dataLen] = '\0';
|
|
|
|
if (!strnicmp(data,"zttap",5)) {
|
|
|
|
type = 0;
|
|
|
|
dataLen = sizeof(data);
|
|
|
|
if (RegGetValueA(nwAdapters,subkeyName,"NetCfgInstanceId",RRF_RT_ANY,&type,(PVOID)data,&dataLen) == ERROR_SUCCESS) {
|
|
|
|
if (existingDeviceInstances.count(std::string(data,dataLen)) == 0) {
|
|
|
|
RegSetKeyValueA(nwAdapters,subkeyName,"_ZeroTierTapIdentifier",REG_SZ,tag,(DWORD)(strlen(tag)+1));
|
2014-04-08 19:00:21 +00:00
|
|
|
_netCfgInstanceId.assign(data,dataLen);
|
2014-03-06 22:06:31 +00:00
|
|
|
type = 0;
|
|
|
|
dataLen = sizeof(data);
|
|
|
|
if (RegGetValueA(nwAdapters,subkeyName,"DeviceInstanceID",RRF_RT_ANY,&type,(PVOID)data,&dataLen) == ERROR_SUCCESS)
|
2014-04-08 19:00:21 +00:00
|
|
|
_deviceInstanceId.assign(data,dataLen);
|
2014-03-06 22:06:31 +00:00
|
|
|
mySubkeyName = subkeyName;
|
|
|
|
|
|
|
|
// Disable DHCP by default on newly created devices
|
|
|
|
HKEY tcpIpInterfaces;
|
|
|
|
if (RegOpenKeyExA(HKEY_LOCAL_MACHINE,"SYSTEM\\CurrentControlSet\\services\\Tcpip\\Parameters\\Interfaces",0,KEY_READ|KEY_WRITE,&tcpIpInterfaces) == ERROR_SUCCESS) {
|
|
|
|
DWORD enable = 0;
|
2014-04-08 19:00:21 +00:00
|
|
|
RegSetKeyValueA(tcpIpInterfaces,_netCfgInstanceId.c_str(),"EnableDHCP",REG_DWORD,&enable,sizeof(enable));
|
2014-03-06 22:06:31 +00:00
|
|
|
RegCloseKey(tcpIpInterfaces);
|
2013-08-22 18:30:55 +00:00
|
|
|
}
|
2014-03-06 22:06:31 +00:00
|
|
|
|
|
|
|
break; // found it!
|
2013-08-22 18:30:55 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
2014-03-06 22:06:31 +00:00
|
|
|
}
|
|
|
|
} else break; // no more keys or error occurred
|
2013-08-21 14:18:05 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2014-04-08 19:00:21 +00:00
|
|
|
if (_netCfgInstanceId.length() > 0) {
|
2013-08-23 21:39:21 +00:00
|
|
|
char tmps[4096];
|
2014-05-23 23:21:57 +00:00
|
|
|
unsigned int tmpsl = Utils::snprintf(tmps,sizeof(tmps),"%.2X-%.2X-%.2X-%.2X-%.2X-%.2X",(unsigned int)mac[0],(unsigned int)mac[1],(unsigned int)mac[2],(unsigned int)mac[3],(unsigned int)mac[4],(unsigned int)mac[5]) + 1;
|
2013-08-25 22:18:02 +00:00
|
|
|
RegSetKeyValueA(nwAdapters,mySubkeyName.c_str(),"NetworkAddress",REG_SZ,tmps,tmpsl);
|
|
|
|
RegSetKeyValueA(nwAdapters,mySubkeyName.c_str(),"MAC",REG_SZ,tmps,tmpsl);
|
2013-08-23 21:39:21 +00:00
|
|
|
DWORD tmp = mtu;
|
|
|
|
RegSetKeyValueA(nwAdapters,mySubkeyName.c_str(),"MTU",REG_DWORD,(LPCVOID)&tmp,sizeof(tmp));
|
|
|
|
tmp = 0;
|
|
|
|
RegSetKeyValueA(nwAdapters,mySubkeyName.c_str(),"EnableDHCP",REG_DWORD,(LPCVOID)&tmp,sizeof(tmp));
|
2014-04-08 23:10:48 +00:00
|
|
|
RegCloseKey(nwAdapters);
|
2014-03-06 22:06:31 +00:00
|
|
|
} else {
|
|
|
|
RegCloseKey(nwAdapters);
|
|
|
|
throw std::runtime_error("unable to find or create tap adapter");
|
2013-08-23 21:39:21 +00:00
|
|
|
}
|
|
|
|
|
2014-03-06 22:06:31 +00:00
|
|
|
// Convert device GUID junk... blech... is there an easier way to do this?
|
2013-08-26 21:22:20 +00:00
|
|
|
{
|
|
|
|
char nobraces[128];
|
2014-04-08 19:00:21 +00:00
|
|
|
const char *nbtmp1 = _netCfgInstanceId.c_str();
|
2013-08-26 21:22:20 +00:00
|
|
|
char *nbtmp2 = nobraces;
|
|
|
|
while (*nbtmp1) {
|
|
|
|
if ((*nbtmp1 != '{')&&(*nbtmp1 != '}'))
|
|
|
|
*nbtmp2++ = *nbtmp1;
|
|
|
|
++nbtmp1;
|
|
|
|
}
|
|
|
|
*nbtmp2 = (char)0;
|
|
|
|
if (UuidFromStringA((RPC_CSTR)nobraces,&_deviceGuid) != RPC_S_OK)
|
|
|
|
throw std::runtime_error("unable to convert instance ID GUID to native GUID (invalid NetCfgInstanceId in registry?)");
|
|
|
|
}
|
|
|
|
|
2013-08-27 15:55:56 +00:00
|
|
|
// Start background thread that actually performs I/O
|
2013-08-25 22:18:02 +00:00
|
|
|
_injectSemaphore = CreateSemaphore(NULL,0,1,NULL);
|
|
|
|
_thread = Thread::start(this);
|
2014-01-28 07:13:36 +00:00
|
|
|
|
|
|
|
// Certain functions can now work (e.g. ips())
|
|
|
|
_initialized = true;
|
2013-08-14 17:23:25 +00:00
|
|
|
}
|
|
|
|
|
2014-04-08 19:00:21 +00:00
|
|
|
WindowsEthernetTap::~WindowsEthernetTap()
|
2013-08-14 17:23:25 +00:00
|
|
|
{
|
2013-08-25 22:18:02 +00:00
|
|
|
_run = false;
|
|
|
|
ReleaseSemaphore(_injectSemaphore,1,NULL);
|
|
|
|
Thread::join(_thread);
|
|
|
|
CloseHandle(_injectSemaphore);
|
2014-04-08 19:00:21 +00:00
|
|
|
_disableTapDevice(_r,_deviceInstanceId);
|
|
|
|
}
|
2014-01-21 21:07:22 +00:00
|
|
|
|
2014-04-08 19:00:21 +00:00
|
|
|
void WindowsEthernetTap::setEnabled(bool en)
|
|
|
|
{
|
2014-04-08 22:47:33 +00:00
|
|
|
_enabled = en;
|
2013-08-14 17:23:25 +00:00
|
|
|
}
|
|
|
|
|
2014-04-08 19:00:21 +00:00
|
|
|
bool WindowsEthernetTap::enabled() const
|
|
|
|
{
|
|
|
|
return _enabled;
|
|
|
|
}
|
|
|
|
|
|
|
|
void WindowsEthernetTap::setDisplayName(const char *dn)
|
2013-08-27 19:55:32 +00:00
|
|
|
{
|
2014-01-28 07:13:36 +00:00
|
|
|
if (!_initialized)
|
|
|
|
return;
|
2013-08-27 19:55:32 +00:00
|
|
|
HKEY ifp;
|
2014-04-08 19:00:21 +00:00
|
|
|
if (RegOpenKeyExA(HKEY_LOCAL_MACHINE,(std::string("SYSTEM\\CurrentControlSet\\Control\\Network\\{4D36E972-E325-11CE-BFC1-08002BE10318}\\") + _netCfgInstanceId).c_str(),0,KEY_READ|KEY_WRITE,&ifp) == ERROR_SUCCESS) {
|
2013-08-27 20:11:39 +00:00
|
|
|
RegSetKeyValueA(ifp,"Connection","Name",REG_SZ,(LPCVOID)dn,(DWORD)(strlen(dn)+1));
|
2013-08-27 19:55:32 +00:00
|
|
|
RegCloseKey(ifp);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2014-04-08 19:00:21 +00:00
|
|
|
bool WindowsEthernetTap::addIP(const InetAddress &ip)
|
2013-08-14 17:23:25 +00:00
|
|
|
{
|
2014-01-28 07:13:36 +00:00
|
|
|
if (!_initialized)
|
|
|
|
return false;
|
2014-01-23 22:15:00 +00:00
|
|
|
if (!ip.netmaskBits()) // sanity check... netmask of 0.0.0.0 is WUT?
|
2013-08-26 21:22:20 +00:00
|
|
|
return false;
|
|
|
|
|
2014-01-23 22:15:00 +00:00
|
|
|
std::set<InetAddress> haveIps(ips());
|
2013-08-26 21:22:20 +00:00
|
|
|
|
2014-01-23 22:15:00 +00:00
|
|
|
try {
|
|
|
|
// Add IP to interface at the netlink level if not already assigned.
|
|
|
|
if (!haveIps.count(ip)) {
|
|
|
|
std::pair<NET_LUID,NET_IFINDEX> ifidx = _findAdapterByGuid(_deviceGuid);
|
|
|
|
MIB_UNICASTIPADDRESS_ROW ipr;
|
|
|
|
|
|
|
|
InitializeUnicastIpAddressEntry(&ipr);
|
|
|
|
if (ip.isV4()) {
|
|
|
|
ipr.Address.Ipv4.sin_family = AF_INET;
|
|
|
|
ipr.Address.Ipv4.sin_addr.S_un.S_addr = *((const uint32_t *)ip.rawIpData());
|
|
|
|
ipr.OnLinkPrefixLength = ip.port();
|
2014-04-08 23:10:48 +00:00
|
|
|
if (ipr.OnLinkPrefixLength >= 32)
|
|
|
|
return false;
|
2014-01-23 22:15:00 +00:00
|
|
|
} else if (ip.isV6()) {
|
2014-04-08 19:00:21 +00:00
|
|
|
ipr.Address.Ipv6.sin6_family = AF_INET6;
|
|
|
|
memcpy(ipr.Address.Ipv6.sin6_addr.u.Byte,ip.rawIpData(),16);
|
|
|
|
ipr.OnLinkPrefixLength = ip.port();
|
2014-04-08 23:10:48 +00:00
|
|
|
if (ipr.OnLinkPrefixLength >= 128)
|
|
|
|
return false;
|
2014-01-23 22:15:00 +00:00
|
|
|
} else return false;
|
|
|
|
|
|
|
|
ipr.PrefixOrigin = IpPrefixOriginManual;
|
|
|
|
ipr.SuffixOrigin = IpSuffixOriginManual;
|
|
|
|
ipr.ValidLifetime = 0xffffffff;
|
|
|
|
ipr.PreferredLifetime = 0xffffffff;
|
|
|
|
|
|
|
|
ipr.InterfaceLuid = ifidx.first;
|
|
|
|
ipr.InterfaceIndex = ifidx.second;
|
|
|
|
|
|
|
|
if (CreateUnicastIpAddressEntry(&ipr) == NO_ERROR) {
|
|
|
|
haveIps.insert(ip);
|
|
|
|
} else {
|
|
|
|
LOG("unable to add IP address %s to interface %s: %d",ip.toString().c_str(),deviceName().c_str(),(int)GetLastError());
|
|
|
|
return false;
|
|
|
|
}
|
2013-08-26 21:22:20 +00:00
|
|
|
}
|
|
|
|
|
2014-04-08 19:00:21 +00:00
|
|
|
_syncIpsWithRegistry(haveIps,_netCfgInstanceId);
|
2014-01-23 22:15:00 +00:00
|
|
|
} catch (std::exception &exc) {
|
2014-01-27 06:47:08 +00:00
|
|
|
LOG("unexpected exception adding IP address %s to %s: %s",ip.toString().c_str(),deviceName().c_str(),exc.what());
|
2014-04-08 23:10:48 +00:00
|
|
|
return false;
|
2014-01-23 22:15:00 +00:00
|
|
|
} catch ( ... ) {
|
|
|
|
LOG("unexpected exception adding IP address %s to %s: unknown exception",ip.toString().c_str(),deviceName().c_str());
|
2014-04-08 23:10:48 +00:00
|
|
|
return false;
|
2014-01-23 22:15:00 +00:00
|
|
|
}
|
2014-04-08 23:10:48 +00:00
|
|
|
return true;
|
2013-08-14 17:23:25 +00:00
|
|
|
}
|
|
|
|
|
2014-04-08 19:00:21 +00:00
|
|
|
bool WindowsEthernetTap::removeIP(const InetAddress &ip)
|
2013-08-14 17:23:25 +00:00
|
|
|
{
|
2014-01-28 07:13:36 +00:00
|
|
|
if (!_initialized)
|
|
|
|
return false;
|
2013-08-26 21:22:20 +00:00
|
|
|
try {
|
|
|
|
MIB_UNICASTIPADDRESS_TABLE *ipt = (MIB_UNICASTIPADDRESS_TABLE *)0;
|
|
|
|
std::pair<NET_LUID,NET_IFINDEX> ifidx = _findAdapterByGuid(_deviceGuid);
|
|
|
|
if (GetUnicastIpAddressTable(AF_UNSPEC,&ipt) == NO_ERROR) {
|
|
|
|
for(DWORD i=0;i<ipt->NumEntries;++i) {
|
|
|
|
if ((ipt->Table[i].InterfaceLuid.Value == ifidx.first.Value)&&(ipt->Table[i].InterfaceIndex == ifidx.second)) {
|
|
|
|
InetAddress addr;
|
|
|
|
switch(ipt->Table[i].Address.si_family) {
|
|
|
|
case AF_INET:
|
|
|
|
addr.set(&(ipt->Table[i].Address.Ipv4.sin_addr.S_un.S_addr),4,ipt->Table[i].OnLinkPrefixLength);
|
|
|
|
break;
|
|
|
|
case AF_INET6:
|
|
|
|
addr.set(ipt->Table[i].Address.Ipv6.sin6_addr.u.Byte,16,ipt->Table[i].OnLinkPrefixLength);
|
2014-01-23 22:15:00 +00:00
|
|
|
if (addr.isLinkLocal())
|
|
|
|
continue; // can't remove link-local IPv6 addresses
|
2013-08-26 21:22:20 +00:00
|
|
|
break;
|
|
|
|
}
|
|
|
|
if (addr == ip) {
|
|
|
|
DeleteUnicastIpAddressEntry(&(ipt->Table[i]));
|
2013-08-27 15:15:14 +00:00
|
|
|
FreeMibTable(ipt);
|
2014-04-08 19:00:21 +00:00
|
|
|
_syncIpsWithRegistry(ips(),_netCfgInstanceId);
|
2013-08-26 21:22:20 +00:00
|
|
|
return true;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
2014-01-23 22:15:00 +00:00
|
|
|
FreeMibTable((PVOID)ipt);
|
2013-08-26 21:22:20 +00:00
|
|
|
}
|
2014-01-27 06:47:08 +00:00
|
|
|
} catch (std::exception &exc) {
|
|
|
|
LOG("unexpected exception removing IP address %s from %s: %s",ip.toString().c_str(),deviceName().c_str(),exc.what());
|
|
|
|
} catch ( ... ) {
|
|
|
|
LOG("unexpected exception removing IP address %s from %s: unknown exception",ip.toString().c_str(),deviceName().c_str());
|
|
|
|
}
|
2013-08-14 17:23:25 +00:00
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
2014-04-08 19:00:21 +00:00
|
|
|
std::set<InetAddress> WindowsEthernetTap::ips() const
|
2013-08-26 21:22:20 +00:00
|
|
|
{
|
2014-01-23 22:15:00 +00:00
|
|
|
static const InetAddress linkLocalLoopback("fe80::1",64); // what is this and why does Windows assign it?
|
2013-08-26 21:22:20 +00:00
|
|
|
std::set<InetAddress> addrs;
|
|
|
|
|
2014-01-28 07:13:36 +00:00
|
|
|
if (!_initialized)
|
|
|
|
return addrs;
|
|
|
|
|
2013-08-26 21:22:20 +00:00
|
|
|
try {
|
|
|
|
MIB_UNICASTIPADDRESS_TABLE *ipt = (MIB_UNICASTIPADDRESS_TABLE *)0;
|
|
|
|
std::pair<NET_LUID,NET_IFINDEX> ifidx = _findAdapterByGuid(_deviceGuid);
|
|
|
|
|
|
|
|
if (GetUnicastIpAddressTable(AF_UNSPEC,&ipt) == NO_ERROR) {
|
|
|
|
for(DWORD i=0;i<ipt->NumEntries;++i) {
|
|
|
|
if ((ipt->Table[i].InterfaceLuid.Value == ifidx.first.Value)&&(ipt->Table[i].InterfaceIndex == ifidx.second)) {
|
|
|
|
switch(ipt->Table[i].Address.si_family) {
|
2014-01-23 22:15:00 +00:00
|
|
|
case AF_INET: {
|
|
|
|
InetAddress ip(&(ipt->Table[i].Address.Ipv4.sin_addr.S_un.S_addr),4,ipt->Table[i].OnLinkPrefixLength);
|
|
|
|
if (ip != InetAddress::LO4)
|
|
|
|
addrs.insert(ip);
|
|
|
|
} break;
|
2013-08-27 20:45:22 +00:00
|
|
|
case AF_INET6: {
|
|
|
|
InetAddress ip(ipt->Table[i].Address.Ipv6.sin6_addr.u.Byte,16,ipt->Table[i].OnLinkPrefixLength);
|
2014-01-23 22:15:00 +00:00
|
|
|
if ((ip != linkLocalLoopback)&&(ip != InetAddress::LO6))
|
2013-08-27 20:45:22 +00:00
|
|
|
addrs.insert(ip);
|
|
|
|
} break;
|
2013-08-26 21:22:20 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
2013-08-27 15:15:14 +00:00
|
|
|
FreeMibTable(ipt);
|
2013-08-26 21:22:20 +00:00
|
|
|
}
|
2014-04-08 19:00:21 +00:00
|
|
|
} catch ( ... ) {} // sanity check, shouldn't happen unless out of memory
|
2013-08-26 21:22:20 +00:00
|
|
|
|
|
|
|
return addrs;
|
|
|
|
}
|
|
|
|
|
2014-04-08 19:00:21 +00:00
|
|
|
void WindowsEthernetTap::put(const MAC &from,const MAC &to,unsigned int etherType,const void *data,unsigned int len)
|
2013-08-14 17:23:25 +00:00
|
|
|
{
|
2014-04-08 23:10:48 +00:00
|
|
|
if ((!_initialized)||(!_enabled)||(_tap == INVALID_HANDLE_VALUE)||(len > (ZT_IF_MTU)))
|
2014-01-28 07:13:36 +00:00
|
|
|
return;
|
2013-08-23 02:33:32 +00:00
|
|
|
{
|
|
|
|
Mutex::Lock _l(_injectPending_m);
|
|
|
|
_injectPending.push( std::pair<Array<char,ZT_IF_MTU + 32>,unsigned int>(Array<char,ZT_IF_MTU + 32>(),len + 14) );
|
|
|
|
char *d = _injectPending.back().first.data;
|
2014-05-23 23:21:57 +00:00
|
|
|
to.copyTo(d,6);
|
|
|
|
from.copyTo(d + 6,6);
|
2014-01-24 00:10:24 +00:00
|
|
|
d[12] = (char)((etherType >> 8) & 0xff);
|
|
|
|
d[13] = (char)(etherType & 0xff);
|
2013-08-23 02:33:32 +00:00
|
|
|
memcpy(d + 14,data,len);
|
|
|
|
}
|
2013-08-25 22:18:02 +00:00
|
|
|
ReleaseSemaphore(_injectSemaphore,1,NULL);
|
2013-08-14 17:23:25 +00:00
|
|
|
}
|
|
|
|
|
2014-04-08 19:00:21 +00:00
|
|
|
std::string WindowsEthernetTap::deviceName() const
|
2013-08-14 17:23:25 +00:00
|
|
|
{
|
2014-04-08 19:00:21 +00:00
|
|
|
return _netCfgInstanceId;
|
2013-08-14 17:23:25 +00:00
|
|
|
}
|
|
|
|
|
2014-04-08 19:00:21 +00:00
|
|
|
std::string WindowsEthernetTap::persistentId() const
|
2014-01-27 06:47:08 +00:00
|
|
|
{
|
2014-04-08 19:00:21 +00:00
|
|
|
return _deviceInstanceId;
|
2014-01-27 06:47:08 +00:00
|
|
|
}
|
|
|
|
|
2014-04-08 19:00:21 +00:00
|
|
|
bool WindowsEthernetTap::updateMulticastGroups(std::set<MulticastGroup> &groups)
|
2013-08-14 17:23:25 +00:00
|
|
|
{
|
2014-01-28 07:13:36 +00:00
|
|
|
if (!_initialized)
|
|
|
|
return false;
|
2014-04-08 22:47:33 +00:00
|
|
|
HANDLE t = _tap;
|
|
|
|
if (t == INVALID_HANDLE_VALUE)
|
|
|
|
return false;
|
2014-01-28 07:13:36 +00:00
|
|
|
|
2013-08-27 20:45:22 +00:00
|
|
|
std::set<MulticastGroup> newGroups;
|
|
|
|
|
2014-01-21 21:07:22 +00:00
|
|
|
// Ensure that groups are added for each IP... this handles the MAC:ADI
|
|
|
|
// groups that are created from IPv4 addresses. Some of these may end
|
|
|
|
// up being duplicates of what the IOCTL returns but that's okay since
|
2014-01-23 22:15:00 +00:00
|
|
|
// the set<> will filter that.
|
|
|
|
std::set<InetAddress> ipaddrs(ips());
|
2013-08-27 20:45:22 +00:00
|
|
|
for(std::set<InetAddress>::const_iterator i(ipaddrs.begin());i!=ipaddrs.end();++i)
|
|
|
|
newGroups.insert(MulticastGroup::deriveMulticastGroupForAddressResolution(*i));
|
|
|
|
|
2014-01-21 21:07:22 +00:00
|
|
|
// The ZT1 tap driver supports an IOCTL to get multicast memberships at the L2
|
|
|
|
// level... something Windows does not seem to expose ordinarily. This lets
|
|
|
|
// pretty much anything work... IPv4, IPv6, IPX, oldskool Netbios, who knows...
|
|
|
|
unsigned char mcastbuf[TAP_WIN_IOCTL_GET_MULTICAST_MEMBERSHIPS_OUTPUT_BUF_SIZE];
|
|
|
|
DWORD bytesReturned = 0;
|
2014-04-08 22:47:33 +00:00
|
|
|
if (DeviceIoControl(t,TAP_WIN_IOCTL_GET_MULTICAST_MEMBERSHIPS,(LPVOID)0,0,(LPVOID)mcastbuf,sizeof(mcastbuf),&bytesReturned,NULL)) {
|
2014-01-21 21:07:22 +00:00
|
|
|
MAC mac;
|
|
|
|
DWORD i = 0;
|
|
|
|
while ((i + 6) <= bytesReturned) {
|
2014-05-23 23:21:57 +00:00
|
|
|
mac.setTo(mcastbuf + i,6);
|
|
|
|
i += 6;
|
|
|
|
if ((mac.isMulticast())&&(!mac.isBroadcast())) {
|
2014-01-23 07:46:33 +00:00
|
|
|
// exclude the nulls that may be returned or any other junk Windows puts in there
|
2014-01-21 21:07:22 +00:00
|
|
|
newGroups.insert(MulticastGroup(mac,0));
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
2013-08-27 20:45:22 +00:00
|
|
|
|
2014-01-21 21:07:22 +00:00
|
|
|
bool changed = false;
|
|
|
|
|
2013-08-27 20:45:22 +00:00
|
|
|
for(std::set<MulticastGroup>::iterator mg(newGroups.begin());mg!=newGroups.end();++mg) {
|
|
|
|
if (!groups.count(*mg)) {
|
|
|
|
groups.insert(*mg);
|
|
|
|
changed = true;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
for(std::set<MulticastGroup>::iterator mg(groups.begin());mg!=groups.end();) {
|
2014-05-23 22:13:34 +00:00
|
|
|
if ((!newGroups.count(*mg))&&(*mg != _blindWildcardMulticastGroup)) {
|
2013-08-27 20:45:22 +00:00
|
|
|
groups.erase(mg++);
|
|
|
|
changed = true;
|
|
|
|
} else ++mg;
|
|
|
|
}
|
|
|
|
|
|
|
|
return changed;
|
2013-08-14 17:23:25 +00:00
|
|
|
}
|
|
|
|
|
2014-04-08 19:00:21 +00:00
|
|
|
void WindowsEthernetTap::threadMain()
|
2013-08-23 21:39:21 +00:00
|
|
|
throw()
|
2013-08-14 17:23:25 +00:00
|
|
|
{
|
2014-04-08 22:47:33 +00:00
|
|
|
char tapPath[256];
|
|
|
|
OVERLAPPED tapOvlRead,tapOvlWrite;
|
2013-08-25 22:18:02 +00:00
|
|
|
HANDLE wait4[3];
|
2014-04-08 22:47:33 +00:00
|
|
|
char *tapReadBuf = (char *)0;
|
|
|
|
|
|
|
|
// Shouldn't be needed, but Windows does not overcommit. This Windows
|
|
|
|
// tap code is defensive to schizoid paranoia degrees.
|
|
|
|
while (!tapReadBuf) {
|
|
|
|
tapReadBuf = (char *)::malloc(ZT_IF_MTU + 32);
|
|
|
|
if (!tapReadBuf)
|
|
|
|
Sleep(1000);
|
|
|
|
}
|
|
|
|
|
|
|
|
// Tap is in this weird Windows global pseudo file space
|
|
|
|
Utils::snprintf(tapPath,sizeof(tapPath),"\\\\.\\Global\\%s.tap",_netCfgInstanceId.c_str());
|
|
|
|
|
|
|
|
// More insanity: repetatively try to enable/disable tap device. The first
|
|
|
|
// time we succeed, close it and do it again. This is to fix a driver init
|
|
|
|
// bug that seems to be extremely non-deterministic and to only occur after
|
|
|
|
// headless MSI upgrade. It cannot be reproduced in any other circumstance.
|
|
|
|
bool throwOneAway = true;
|
|
|
|
while (_run) {
|
|
|
|
_disableTapDevice(_r,_deviceInstanceId);
|
|
|
|
Sleep(250);
|
|
|
|
if (!_enableTapDevice(_r,_deviceInstanceId)) {
|
|
|
|
::free(tapReadBuf);
|
|
|
|
_enabled = false;
|
|
|
|
return; // only happens if devcon is missing or totally fails
|
|
|
|
}
|
|
|
|
Sleep(250);
|
|
|
|
|
|
|
|
_tap = CreateFileA(tapPath,GENERIC_READ|GENERIC_WRITE,0,NULL,OPEN_EXISTING,FILE_ATTRIBUTE_SYSTEM|FILE_FLAG_OVERLAPPED,NULL);
|
|
|
|
if (_tap == INVALID_HANDLE_VALUE) {
|
|
|
|
Sleep(500);
|
|
|
|
continue;
|
|
|
|
}
|
|
|
|
|
|
|
|
uint32_t tmpi = 1;
|
|
|
|
DWORD bytesReturned = 0;
|
|
|
|
DeviceIoControl(_tap,TAP_WIN_IOCTL_SET_MEDIA_STATUS,&tmpi,sizeof(tmpi),&tmpi,sizeof(tmpi),&bytesReturned,NULL);
|
|
|
|
|
|
|
|
if (throwOneAway) {
|
|
|
|
throwOneAway = false;
|
|
|
|
CloseHandle(_tap);
|
|
|
|
_tap = INVALID_HANDLE_VALUE;
|
|
|
|
Sleep(250);
|
|
|
|
continue;
|
|
|
|
} else break;
|
|
|
|
}
|
|
|
|
|
|
|
|
memset(&tapOvlRead,0,sizeof(tapOvlRead));
|
|
|
|
tapOvlRead.hEvent = CreateEvent(NULL,TRUE,FALSE,NULL);
|
|
|
|
memset(&tapOvlWrite,0,sizeof(tapOvlWrite));
|
|
|
|
tapOvlWrite.hEvent = CreateEvent(NULL,TRUE,FALSE,NULL);
|
|
|
|
|
2013-08-25 22:18:02 +00:00
|
|
|
wait4[0] = _injectSemaphore;
|
2014-04-08 22:47:33 +00:00
|
|
|
wait4[1] = tapOvlRead.hEvent;
|
|
|
|
wait4[2] = tapOvlWrite.hEvent; // only included if writeInProgress is true
|
2013-08-25 22:18:02 +00:00
|
|
|
|
2014-04-08 22:47:33 +00:00
|
|
|
// Start overlapped read, which is always active
|
|
|
|
ReadFile(_tap,tapReadBuf,sizeof(tapReadBuf),NULL,&tapOvlRead);
|
2013-08-25 22:18:02 +00:00
|
|
|
bool writeInProgress = false;
|
|
|
|
|
|
|
|
for(;;) {
|
|
|
|
if (!_run) break;
|
2014-04-08 22:47:33 +00:00
|
|
|
DWORD r = WaitForMultipleObjectsEx(writeInProgress ? 3 : 2,wait4,FALSE,5000,TRUE);
|
2013-08-25 22:18:02 +00:00
|
|
|
if (!_run) break;
|
|
|
|
|
2014-04-08 22:47:33 +00:00
|
|
|
if ((r == WAIT_TIMEOUT)||(r == WAIT_FAILED))
|
|
|
|
continue;
|
|
|
|
|
|
|
|
if (HasOverlappedIoCompleted(&tapOvlRead)) {
|
2013-08-25 22:18:02 +00:00
|
|
|
DWORD bytesRead = 0;
|
2014-04-08 22:47:33 +00:00
|
|
|
if (GetOverlappedResult(_tap,&tapOvlRead,&bytesRead,FALSE)) {
|
2014-04-08 19:00:21 +00:00
|
|
|
if ((bytesRead > 14)&&(_enabled)) {
|
2014-05-23 23:21:57 +00:00
|
|
|
MAC to(tapReadBuf,6);
|
|
|
|
MAC from(tapReadBuf + 6,6);
|
2014-04-08 22:47:33 +00:00
|
|
|
unsigned int etherType = ((((unsigned int)tapReadBuf[12]) & 0xff) << 8) | (((unsigned int)tapReadBuf[13]) & 0xff);
|
2014-04-08 19:00:21 +00:00
|
|
|
try {
|
2014-04-08 22:47:33 +00:00
|
|
|
Buffer<4096> tmp(tapReadBuf + 14,bytesRead - 14);
|
2014-04-08 19:00:21 +00:00
|
|
|
_handler(_arg,from,to,etherType,tmp);
|
|
|
|
} catch ( ... ) {} // handlers should not throw
|
2013-08-25 22:18:02 +00:00
|
|
|
}
|
|
|
|
}
|
2014-04-08 22:47:33 +00:00
|
|
|
ReadFile(_tap,tapReadBuf,ZT_IF_MTU + 32,NULL,&tapOvlRead);
|
2013-08-25 22:18:02 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
if (writeInProgress) {
|
2014-04-08 22:47:33 +00:00
|
|
|
if (HasOverlappedIoCompleted(&tapOvlWrite)) {
|
2013-08-25 22:18:02 +00:00
|
|
|
writeInProgress = false;
|
|
|
|
_injectPending_m.lock();
|
|
|
|
_injectPending.pop();
|
|
|
|
} else continue; // still writing, so skip code below and wait
|
|
|
|
} else _injectPending_m.lock();
|
|
|
|
|
|
|
|
if (!_injectPending.empty()) {
|
2014-04-08 22:47:33 +00:00
|
|
|
WriteFile(_tap,_injectPending.front().first.data,_injectPending.front().second,NULL,&tapOvlWrite);
|
2013-08-25 22:18:02 +00:00
|
|
|
writeInProgress = true;
|
|
|
|
}
|
|
|
|
|
|
|
|
_injectPending_m.unlock();
|
|
|
|
}
|
|
|
|
|
|
|
|
CancelIo(_tap);
|
2014-04-08 22:47:33 +00:00
|
|
|
|
|
|
|
CloseHandle(tapOvlRead.hEvent);
|
|
|
|
CloseHandle(tapOvlWrite.hEvent);
|
|
|
|
CloseHandle(_tap);
|
|
|
|
_tap = INVALID_HANDLE_VALUE;
|
|
|
|
|
|
|
|
::free(tapReadBuf);
|
2013-08-14 17:23:25 +00:00
|
|
|
}
|
|
|
|
|
2014-04-08 19:00:21 +00:00
|
|
|
bool WindowsEthernetTap::deletePersistentTapDevice(const RuntimeEnvironment *_r,const char *pid)
|
2014-01-27 06:47:08 +00:00
|
|
|
{
|
2014-02-28 23:04:50 +00:00
|
|
|
Mutex::Lock _l(_systemTapInitLock); // only one thread may mess with taps at a time, process-wide
|
|
|
|
|
2014-03-06 22:06:31 +00:00
|
|
|
HANDLE devconLog = CreateFileA((_r->homePath + "\\devcon.log").c_str(),GENERIC_WRITE,FILE_SHARE_READ|FILE_SHARE_WRITE,NULL,OPEN_ALWAYS,FILE_ATTRIBUTE_NORMAL,NULL);
|
2014-01-27 06:47:08 +00:00
|
|
|
STARTUPINFOA startupInfo;
|
|
|
|
startupInfo.cb = sizeof(startupInfo);
|
2014-03-06 22:06:31 +00:00
|
|
|
if (devconLog != INVALID_HANDLE_VALUE) {
|
|
|
|
SetFilePointer(devconLog,0,0,FILE_END);
|
|
|
|
startupInfo.hStdOutput = devconLog;
|
|
|
|
startupInfo.hStdError = devconLog;
|
|
|
|
}
|
2014-01-27 06:47:08 +00:00
|
|
|
PROCESS_INFORMATION processInfo;
|
|
|
|
memset(&startupInfo,0,sizeof(STARTUPINFOA));
|
|
|
|
memset(&processInfo,0,sizeof(PROCESS_INFORMATION));
|
2014-04-08 19:00:21 +00:00
|
|
|
if (CreateProcessA(NULL,(LPSTR)(std::string("\"") + _r->homePath + _winEnv.devcon + "\" remove @" + pid).c_str(),NULL,NULL,FALSE,0,NULL,NULL,&startupInfo,&processInfo)) {
|
2014-01-27 06:47:08 +00:00
|
|
|
WaitForSingleObject(processInfo.hProcess,INFINITE);
|
|
|
|
CloseHandle(processInfo.hProcess);
|
|
|
|
CloseHandle(processInfo.hThread);
|
2014-03-06 22:06:31 +00:00
|
|
|
if (devconLog != INVALID_HANDLE_VALUE)
|
|
|
|
CloseHandle(devconLog);
|
2014-01-27 06:47:08 +00:00
|
|
|
return true;
|
|
|
|
}
|
2014-03-06 22:06:31 +00:00
|
|
|
if (devconLog != INVALID_HANDLE_VALUE)
|
|
|
|
CloseHandle(devconLog);
|
2014-01-27 06:47:08 +00:00
|
|
|
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
2014-04-08 19:00:21 +00:00
|
|
|
int WindowsEthernetTap::cleanPersistentTapDevices(const RuntimeEnvironment *_r,const std::set<std::string> &exceptThese,bool alsoRemoveUnassociatedDevices)
|
2014-02-28 23:04:50 +00:00
|
|
|
{
|
|
|
|
char subkeyName[4096];
|
|
|
|
char subkeyClass[4096];
|
|
|
|
char data[4096];
|
|
|
|
|
|
|
|
std::set<std::string> instanceIdPathsToRemove;
|
|
|
|
{
|
|
|
|
Mutex::Lock _l(_systemTapInitLock); // only one thread may mess with taps at a time, process-wide
|
|
|
|
|
|
|
|
HKEY nwAdapters;
|
|
|
|
if (RegOpenKeyExA(HKEY_LOCAL_MACHINE,"SYSTEM\\CurrentControlSet\\Control\\Class\\{4D36E972-E325-11CE-BFC1-08002BE10318}",0,KEY_READ|KEY_WRITE,&nwAdapters) != ERROR_SUCCESS)
|
|
|
|
return -1;
|
|
|
|
|
2014-03-06 22:06:31 +00:00
|
|
|
for(DWORD subkeyIndex=0;;++subkeyIndex) {
|
2014-02-28 23:04:50 +00:00
|
|
|
DWORD type;
|
|
|
|
DWORD dataLen;
|
|
|
|
DWORD subkeyNameLen = sizeof(subkeyName);
|
|
|
|
DWORD subkeyClassLen = sizeof(subkeyClass);
|
|
|
|
FILETIME lastWriteTime;
|
2014-03-06 22:06:31 +00:00
|
|
|
if (RegEnumKeyExA(nwAdapters,subkeyIndex,subkeyName,&subkeyNameLen,(DWORD *)0,subkeyClass,&subkeyClassLen,&lastWriteTime) == ERROR_SUCCESS) {
|
|
|
|
type = 0;
|
|
|
|
dataLen = sizeof(data);
|
|
|
|
if (RegGetValueA(nwAdapters,subkeyName,"ComponentId",RRF_RT_ANY,&type,(PVOID)data,&dataLen) == ERROR_SUCCESS) {
|
|
|
|
data[dataLen] = '\0';
|
|
|
|
if (!strnicmp(data,"zttap",5)) {
|
|
|
|
std::string instanceIdPath;
|
|
|
|
type = 0;
|
|
|
|
dataLen = sizeof(data);
|
|
|
|
if (RegGetValueA(nwAdapters,subkeyName,"DeviceInstanceID",RRF_RT_ANY,&type,(PVOID)data,&dataLen) == ERROR_SUCCESS)
|
|
|
|
instanceIdPath.assign(data,dataLen);
|
|
|
|
if (instanceIdPath.length() != 0) {
|
2014-02-28 23:04:50 +00:00
|
|
|
type = 0;
|
|
|
|
dataLen = sizeof(data);
|
2014-03-06 22:06:31 +00:00
|
|
|
if (RegGetValueA(nwAdapters,subkeyName,"_ZeroTierTapIdentifier",RRF_RT_ANY,&type,(PVOID)data,&dataLen) == ERROR_SUCCESS) {
|
|
|
|
if (dataLen <= 0) {
|
|
|
|
if (alsoRemoveUnassociatedDevices)
|
|
|
|
instanceIdPathsToRemove.insert(instanceIdPath);
|
|
|
|
} else {
|
|
|
|
if (!exceptThese.count(std::string(data,dataLen)))
|
|
|
|
instanceIdPathsToRemove.insert(instanceIdPath);
|
|
|
|
}
|
|
|
|
} else if (alsoRemoveUnassociatedDevices)
|
|
|
|
instanceIdPathsToRemove.insert(instanceIdPath);
|
2014-02-28 23:04:50 +00:00
|
|
|
}
|
|
|
|
}
|
2014-03-06 22:06:31 +00:00
|
|
|
}
|
|
|
|
} else break; // end of list or failure
|
2014-02-28 23:04:50 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
RegCloseKey(nwAdapters);
|
|
|
|
}
|
|
|
|
|
|
|
|
int removed = 0;
|
|
|
|
for(std::set<std::string>::iterator iidp(instanceIdPathsToRemove.begin());iidp!=instanceIdPathsToRemove.end();++iidp) {
|
|
|
|
if (deletePersistentTapDevice(_r,iidp->c_str()))
|
|
|
|
++removed;
|
|
|
|
}
|
|
|
|
return removed;
|
|
|
|
}
|
|
|
|
|
2013-08-14 17:23:25 +00:00
|
|
|
} // namespace ZeroTier
|