Adam Ierymenko
12 years ago
7 changed files with 402 additions and 295 deletions
@ -0,0 +1,170 @@ |
|||||||
|
/*
|
||||||
|
* ZeroTier One - Global Peer to Peer Ethernet |
||||||
|
* Copyright (C) 2011-2014 ZeroTier Networks LLC |
||||||
|
* |
||||||
|
* 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/
|
||||||
|
*/ |
||||||
|
|
||||||
|
#include "WindowsEthernetTapFactory.hpp" |
||||||
|
#include "WindowsEthernetTap.hpp" |
||||||
|
|
||||||
|
namespace ZeroTier { |
||||||
|
|
||||||
|
WindowsEthernetTapFactory::Env::Env() |
||||||
|
{ |
||||||
|
#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 |
||||||
|
} |
||||||
|
|
||||||
|
WindowsEthernetTapFactory::WindowsEthernetTapFactory(const char *pathToHelpers) : |
||||||
|
_pathToHelpers(pathToHelpers) |
||||||
|
{ |
||||||
|
} |
||||||
|
|
||||||
|
WindowsEthernetTapFactory::~WindowsEthernetTapFactory() |
||||||
|
{ |
||||||
|
Mutex::Lock _l(_devices_m); |
||||||
|
for(std::vector<EthernetTap *>::iterator d(_devices.begin());d!=_devices.end();++d) |
||||||
|
delete *d; |
||||||
|
} |
||||||
|
|
||||||
|
EthernetTap *WindowsEthernetTapFactory::open( |
||||||
|
const MAC &mac, |
||||||
|
unsigned int mtu, |
||||||
|
unsigned int metric, |
||||||
|
uint64_t nwid, |
||||||
|
const char *desiredDevice, |
||||||
|
const char *friendlyName, |
||||||
|
void (*handler)(void *,const MAC &,const MAC &,unsigned int,const Buffer<4096> &), |
||||||
|
void *arg) |
||||||
|
{ |
||||||
|
Mutex::Lock _l(_devices_m); |
||||||
|
EthernetTap *t = new WindowsEthernetTap(_pathToHelpers.c_str(),mac,mtu,metric,nwid,desiredDevice,friendlyName,handler,arg); |
||||||
|
_devices.push_back(t); |
||||||
|
return t; |
||||||
|
} |
||||||
|
|
||||||
|
void WindowsEthernetTapFactory::close(EthernetTap *tap,bool destroyPersistentDevices) |
||||||
|
{ |
||||||
|
std::string instanceId(((WindowsEthernetTap *)tap)->instanceId()); |
||||||
|
Mutex::Lock _l(_devices_m); |
||||||
|
|
||||||
|
for(std::vector<EthernetTap *>::iterator d(_devices.begin());d!=_devices.end();++d) { |
||||||
|
if (*d == tap) { |
||||||
|
_devices.erase(d); |
||||||
|
break; |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
delete tap; |
||||||
|
|
||||||
|
if (destroyPersistentDevices) |
||||||
|
_deletePersistentTapDevice(_pathToHelpers.c_str(),instanceId.c_str()); |
||||||
|
} |
||||||
|
|
||||||
|
void WindowsEthernetTapFactory::destroyAllPersistentTapDevices(const char *pathToHelpers) |
||||||
|
{ |
||||||
|
char subkeyName[4096]; |
||||||
|
char subkeyClass[4096]; |
||||||
|
char data[4096]; |
||||||
|
|
||||||
|
std::set<std::string> instanceIdPathsToRemove; |
||||||
|
{ |
||||||
|
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; |
||||||
|
|
||||||
|
for(DWORD subkeyIndex=0;;++subkeyIndex) { |
||||||
|
DWORD type; |
||||||
|
DWORD dataLen; |
||||||
|
DWORD subkeyNameLen = sizeof(subkeyName); |
||||||
|
DWORD subkeyClassLen = sizeof(subkeyClass); |
||||||
|
FILETIME lastWriteTime; |
||||||
|
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) { |
||||||
|
instanceIdPathsToRemove.insert(instanceIdPath); |
||||||
|
/*
|
||||||
|
type = 0; |
||||||
|
dataLen = sizeof(data); |
||||||
|
if (RegGetValueA(nwAdapters,subkeyName,"_ZeroTierTapIdentifier",RRF_RT_ANY,&type,(PVOID)data,&dataLen) == ERROR_SUCCESS) { |
||||||
|
if (dataLen <= 0) { |
||||||
|
instanceIdPathsToRemove.insert(instanceIdPath); |
||||||
|
} else { |
||||||
|
instanceIdPathsToRemove.insert(instanceIdPath); |
||||||
|
} |
||||||
|
} |
||||||
|
*/ |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
} else break; // end of list or failure
|
||||||
|
} |
||||||
|
|
||||||
|
RegCloseKey(nwAdapters); |
||||||
|
} |
||||||
|
|
||||||
|
for(std::set<std::string>::iterator iidp(instanceIdPathsToRemove.begin());iidp!=instanceIdPathsToRemove.end();++iidp) |
||||||
|
_deletePersistentTapDevice(pathToHelpers,iidp->c_str()); |
||||||
|
} |
||||||
|
|
||||||
|
void WindowsEthernetTapFactory::_deletePersistentTapDevice(const char *pathToHelpers,const char *instanceId) |
||||||
|
{ |
||||||
|
HANDLE devconLog = CreateFileA((std::string(pathToHelpers) + "\\devcon.log").c_str(),GENERIC_WRITE,FILE_SHARE_READ|FILE_SHARE_WRITE,NULL,OPEN_ALWAYS,FILE_ATTRIBUTE_NORMAL,NULL); |
||||||
|
STARTUPINFOA startupInfo; |
||||||
|
startupInfo.cb = sizeof(startupInfo); |
||||||
|
if (devconLog != INVALID_HANDLE_VALUE) { |
||||||
|
SetFilePointer(devconLog,0,0,FILE_END); |
||||||
|
startupInfo.hStdOutput = devconLog; |
||||||
|
startupInfo.hStdError = devconLog; |
||||||
|
} |
||||||
|
PROCESS_INFORMATION processInfo; |
||||||
|
memset(&startupInfo,0,sizeof(STARTUPINFOA)); |
||||||
|
memset(&processInfo,0,sizeof(PROCESS_INFORMATION)); |
||||||
|
if (CreateProcessA(NULL,(LPSTR)(std::string("\"") + pathToHelpers + WINENV.devcon + "\" remove @" + instanceId).c_str(),NULL,NULL,FALSE,0,NULL,NULL,&startupInfo,&processInfo)) { |
||||||
|
WaitForSingleObject(processInfo.hProcess,INFINITE); |
||||||
|
CloseHandle(processInfo.hProcess); |
||||||
|
CloseHandle(processInfo.hThread); |
||||||
|
} |
||||||
|
if (devconLog != INVALID_HANDLE_VALUE) |
||||||
|
CloseHandle(devconLog); |
||||||
|
} |
||||||
|
|
||||||
|
} // namespace ZeroTier
|
||||||
@ -0,0 +1,90 @@ |
|||||||
|
/*
|
||||||
|
* ZeroTier One - Global Peer to Peer Ethernet |
||||||
|
* Copyright (C) 2011-2014 ZeroTier Networks LLC |
||||||
|
* |
||||||
|
* 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/
|
||||||
|
*/ |
||||||
|
|
||||||
|
#ifndef ZT_WINDOWSETHERNETTAPFACTORY_HPP |
||||||
|
#define ZT_WINDOWSETHERNETTAPFACTORY_HPP |
||||||
|
|
||||||
|
#include <vector> |
||||||
|
#include <string> |
||||||
|
|
||||||
|
#include "../node/EthernetTapFactory.hpp" |
||||||
|
#include "../node/Mutex.hpp" |
||||||
|
|
||||||
|
namespace ZeroTier { |
||||||
|
|
||||||
|
class WindowsEthernetTapFactory : public EthernetTapFactory |
||||||
|
{ |
||||||
|
public: |
||||||
|
class Env |
||||||
|
{ |
||||||
|
public: |
||||||
|
Env(); |
||||||
|
BOOL is64Bit; // true if WIN64 or WoW64 (32-bit binary on 64-bit architecture)
|
||||||
|
const char *devcon; // name of devcon binary in pathToHelpers to use
|
||||||
|
const char *tapDriver; // relative path to driver under pathToHelpers to use
|
||||||
|
}; |
||||||
|
|
||||||
|
/**
|
||||||
|
* Constants related to Windows environment, computed on program start |
||||||
|
*/ |
||||||
|
static const Env WINENV; |
||||||
|
|
||||||
|
/**
|
||||||
|
* @param pathToHelpers Path to devcon32.exe, devcon64.exe, and other required helper binaries (ZeroTier running directory) |
||||||
|
*/ |
||||||
|
WindowsEthernetTapFactory(const char *pathToHelpers); |
||||||
|
virtual ~WindowsEthernetTapFactory(); |
||||||
|
|
||||||
|
virtual EthernetTap *open( |
||||||
|
const MAC &mac, |
||||||
|
unsigned int mtu, |
||||||
|
unsigned int metric, |
||||||
|
uint64_t nwid, |
||||||
|
const char *desiredDevice, |
||||||
|
const char *friendlyName, |
||||||
|
void (*handler)(void *,const MAC &,const MAC &,unsigned int,const Buffer<4096> &), |
||||||
|
void *arg); |
||||||
|
virtual void close(EthernetTap *tap,bool destroyPersistentDevices); |
||||||
|
|
||||||
|
/**
|
||||||
|
* Uninstalls all persistent tap devices in the system belonging to ZeroTier |
||||||
|
* |
||||||
|
* This is for uninstallation. Do not call this while tap devices are active. |
||||||
|
*/ |
||||||
|
static void destroyAllPersistentTapDevices(const char *pathToHelpers); |
||||||
|
|
||||||
|
private: |
||||||
|
static void _deletePersistentTapDevice(const char *pathToHelpers,const char *instanceId); |
||||||
|
|
||||||
|
std::string _pathToHelpers; |
||||||
|
std::vector<EthernetTap *> _devices; |
||||||
|
Mutex _devices_m; |
||||||
|
}; |
||||||
|
|
||||||
|
} // namespace ZeroTier
|
||||||
|
|
||||||
|
#endif |
||||||
Loading…
Reference in new issue