Adam Ierymenko
12 years ago
18 changed files with 419 additions and 136 deletions
@ -0,0 +1,83 @@
|
||||
/*
|
||||
* 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 "LinuxEthernetTapFactory.hpp" |
||||
#include "LinuxEthernetTap.hpp" |
||||
|
||||
namespace ZeroTier { |
||||
|
||||
LinuxEthernetTapFactory::LinuxEthernetTapFactory() |
||||
{ |
||||
} |
||||
|
||||
LinuxEthernetTapFactory::~LinuxEthernetTapFactory() |
||||
{ |
||||
Mutex::Lock _l(_devices_m); |
||||
for(std::vector<EthernetTap *>::iterator d(_devices.begin());d!=_devices.end();++d) |
||||
delete *d; |
||||
} |
||||
|
||||
EthernetTap *LinuxEthernetTapFactory::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 LinuxEthernetTap(mac,mtu,metric,nwid,desiredDevice,friendlyName,handler,arg); |
||||
_devices.push_back(t); |
||||
return t; |
||||
} |
||||
|
||||
void LinuxEthernetTapFactory::close(EthernetTap *tap) |
||||
{ |
||||
{ |
||||
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; |
||||
} |
||||
|
||||
std::vector<std::string> allTapDeviceNames() const |
||||
{ |
||||
std::vector<std::string> dn; |
||||
Mutex::Lock _l(_devices_m); |
||||
for(std::vector<EthernetTap *>::const_iterator d(_devices.begin());d!=_devices.end();++d) |
||||
dn.push_back(d->deviceName()); |
||||
return dn; |
||||
} |
||||
|
||||
} // namespace ZeroTier
|
||||
@ -0,0 +1,64 @@
|
||||
/*
|
||||
* 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_LINUXETHERNETTAPFACTORY_HPP |
||||
#define ZT_LINUXETHERNETTAPFACTORY_HPP |
||||
|
||||
#include <vector> |
||||
#include <string> |
||||
|
||||
#include "../EthernetTapFactory.hpp" |
||||
#include "../Mutex.hpp" |
||||
|
||||
namespace ZeroTier { |
||||
|
||||
class LinuxEthernetTapFactory : public EthernetTapFactory |
||||
{ |
||||
public: |
||||
LinuxEthernetTapFactory(); |
||||
virtual ~LinuxEthernetTapFactory(); |
||||
|
||||
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); |
||||
virtual std::vector<std::string> allTapDeviceNames() const; |
||||
|
||||
private: |
||||
std::vector<EthernetTap *> _devices; |
||||
Mutex _devices_m; |
||||
}; |
||||
|
||||
} // namespace ZeroTier
|
||||
|
||||
#endif |
||||
@ -0,0 +1,126 @@
|
||||
/*
|
||||
* 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 <stdio.h> |
||||
#include <stdlib.h> |
||||
#include <string.h> |
||||
#include <unistd.h> |
||||
|
||||
#include "OSXEthernetTapFactory.hpp" |
||||
#include "OSXEthernetTap.hpp" |
||||
|
||||
namespace ZeroTier { |
||||
|
||||
OSXEthernetTapFactory::OSXEthernetTapFactory(const char *pathToTapKext,const char *tapKextName) |
||||
_pathToTapKext((pathToTapKext) ? pathToTapKext : ""), |
||||
_tapKextName((tapKextName) ? tapKextName : "") |
||||
{ |
||||
struct stat stattmp; |
||||
|
||||
if ((_pathToTapKext.length())&&(_tapKextName.length())) { |
||||
if (stat("/dev/zt0",&stattmp)) { |
||||
long kextpid = (long)vfork(); |
||||
if (kextpid == 0) { |
||||
::chdir(_pathToTapKext.c_str()); |
||||
::execl("/sbin/kextload","/sbin/kextload","-q","-repository",_pathToTapKext.c_str(),_tapKextName.c_str(),(const char *)0); |
||||
::_exit(-1); |
||||
} else if (kextpid > 0) { |
||||
int exitcode = -1; |
||||
::waitpid(kextpid,&exitcode,0); |
||||
} else throw std::runtime_error("unable to create subprocess with fork()"); |
||||
} |
||||
} |
||||
|
||||
if (stat("/dev/zt0",&stattmp)) { |
||||
::usleep(500); // give tap device driver time to start up and try again
|
||||
if (stat("/dev/zt0",&stattmp)) |
||||
throw std::runtime_error("/dev/zt# tap devices do not exist and unable to load kernel extension"); |
||||
} |
||||
} |
||||
|
||||
OSXEthernetTapFactory::~OSXEthernetTapFactory() |
||||
{ |
||||
Mutex::Lock _l(_devices_m); |
||||
for(std::vector<EthernetTap *>::iterator d(_devices.begin());d!=_devices.end();++d) |
||||
delete *d; |
||||
|
||||
if ((_pathToTapKext.length())&&(_tapKextName.length())) { |
||||
// Attempt to unload kext. If anything else is using a /dev/zt# node, this
|
||||
// fails and the kext stays in the kernel.
|
||||
char tmp[16384]; |
||||
sprintf(tmp,"%s/%s",_pathToTapKext.c_str(),_tapKextName.c_str()); |
||||
long kextpid = (long)vfork(); |
||||
if (kextpid == 0) { |
||||
::execl("/sbin/kextunload","/sbin/kextunload",tmp,(const char *)0); |
||||
::_exit(-1); |
||||
} else if (kextpid > 0) { |
||||
int exitcode = -1; |
||||
::waitpid(kextpid,&exitcode,0); |
||||
} |
||||
} |
||||
} |
||||
|
||||
EthernetTap *OSXEthernetTapFactory::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 OSXEthernetTap(mac,mtu,metric,nwid,desiredDevice,friendlyName,handler,arg); |
||||
_devices.push_back(t); |
||||
return t; |
||||
} |
||||
|
||||
void OSXEthernetTapFactory::close(EthernetTap *tap) |
||||
{ |
||||
{ |
||||
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; |
||||
} |
||||
|
||||
std::vector<std::string> allTapDeviceNames() const |
||||
{ |
||||
std::vector<std::string> dn; |
||||
Mutex::Lock _l(_devices_m); |
||||
for(std::vector<EthernetTap *>::const_iterator d(_devices.begin());d!=_devices.end();++d) |
||||
dn.push_back(d->deviceName()); |
||||
return dn; |
||||
} |
||||
|
||||
} // namespace ZeroTier
|
||||
@ -0,0 +1,77 @@
|
||||
/*
|
||||
* 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_LINUXETHERNETTAPFACTORY_HPP |
||||
#define ZT_LINUXETHERNETTAPFACTORY_HPP |
||||
|
||||
#include <vector> |
||||
#include <string> |
||||
|
||||
#include "../EthernetTapFactory.hpp" |
||||
#include "../Mutex.hpp" |
||||
|
||||
namespace ZeroTier { |
||||
|
||||
class OSXEthernetTapFactory : public EthernetTapFactory |
||||
{ |
||||
public: |
||||
/**
|
||||
* Create OSX ethernet tap factory |
||||
* |
||||
* If kext paths are specified, an attempt will be made to load the kext |
||||
* on launch if not present and unload it on shutdown. |
||||
* |
||||
* @param pathToTapKext Full path to the location of the tap kext |
||||
* @param tapKextName Name of tap kext as found within tap kext path (usually "tap.kext") |
||||
* @throws std::runtime_error Tap not available and unable to load kext |
||||
*/ |
||||
OSXEthernetTapFactory(const char *pathToTapKext,const char *tapKextName); |
||||
|
||||
virtual ~OSXEthernetTapFactory(); |
||||
|
||||
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); |
||||
virtual std::vector<std::string> allTapDeviceNames() const; |
||||
|
||||
private: |
||||
std::vector<EthernetTap *> _devices; |
||||
Mutex _devices_m; |
||||
std::string _pathToTapKext; |
||||
std::string _tapKextName; |
||||
}; |
||||
|
||||
} // namespace ZeroTier
|
||||
|
||||
#endif |
||||
Loading…
Reference in new issue