Adam Ierymenko
11 years ago
5 changed files with 477 additions and 20 deletions
@ -0,0 +1,319 @@
|
||||
/*
|
||||
* ZeroTier One - Network Virtualization Everywhere |
||||
* Copyright (C) 2011-2015 ZeroTier, Inc. |
||||
* |
||||
* 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 "../version.h" |
||||
#include "../include/ZeroTierOne.h" |
||||
|
||||
#include "../node/Constants.hpp" |
||||
#include "../node/Mutex.hpp" |
||||
#include "../node/Node.hpp" |
||||
#include "../node/Utils.hpp" |
||||
#include "../node/InetAddress.hpp" |
||||
|
||||
#include "../osdep/Phy.hpp" |
||||
#include "../osdep/Thread.hpp" |
||||
#include "../osdep/OSUtils.hpp" |
||||
|
||||
#include "One.hpp" |
||||
|
||||
namespace ZeroTier { |
||||
|
||||
static void SphyOnDatagramFunction(PhySocket *sock,void **uptr,const struct sockaddr *from,void *data,unsigned long len); |
||||
static void SphyOnTcpConnectFunction(PhySocket *sock,void **uptr,bool success); |
||||
static void SphyOnTcpAcceptFunction(PhySocket *sockL,PhySocket *sockN,void **uptrL,void **uptrN,const struct sockaddr *from); |
||||
static void SphyOnTcpCloseFunction(PhySocket *sock,void **uptr); |
||||
static void SphyOnTcpDataFunction(PhySocket *sock,void **uptr,void *data,unsigned long len); |
||||
static void SphyOnTcpWritableFunction(PhySocket *sock,void **uptr); |
||||
|
||||
static int SnodeVirtualNetworkConfigFunction(ZT1_Node *node,void *uptr,uint64_t nwid,enum ZT1_VirtualNetworkConfigOperation op,const ZT1_VirtualNetworkConfig *nwconf); |
||||
static void SnodeEventCallback(ZT1_Node *node,void *uptr,enum ZT1_Event event,const void *metaData); |
||||
static long SnodeDataStoreGetFunction(ZT1_Node *node,void *uptr,const char *name,void *buf,unsigned long bufSize,unsigned long readIndex,unsigned long *totalSize); |
||||
static int SnodeDataStorePutFunction(ZT1_Node *node,void *uptr,const char *name,const void *data,unsigned long len,int secure); |
||||
static int SnodeWirePacketSendFunction(ZT1_Node *node,void *uptr,const struct sockaddr_storage *addr,unsigned int desperation,const void *data,unsigned int len); |
||||
static void SnodeVirtualNetworkFrameFunction(ZT1_Node *node,void *uptr,uint64_t nwid,uint64_t sourceMac,uint64_t destMac,unsigned int etherType,unsigned int vlanId,const void *data,unsigned int len); |
||||
|
||||
class OneImpl : public One |
||||
{ |
||||
public: |
||||
OneImpl(const char *hp,unsigned int port,NetworkConfigMaster *master,const char *overrideRootTopology) : |
||||
_phy(SphyOnDatagramFunction,SphyOnTcpWritableFunction,SphyOnTcpAcceptFunction,SphyOnTcpCloseFunction,SphyOnTcpDataFunction,SphyOnTcpWritableFunction,true), |
||||
_master(master), |
||||
_overrideRootTopology((overrideRootTopology) ? overrideRootTopology : ""), |
||||
_node((Node *)0), |
||||
_nextBackgroundTaskDeadline(0), |
||||
_termReason(ONE_STILL_RUNNING), |
||||
_run(true) |
||||
{ |
||||
struct sockaddr_in in4; |
||||
struct sockaddr_in6 in6; |
||||
|
||||
::memset(&in4,0,sizeof(in4)); |
||||
in4.sin_family = AF_INET; |
||||
in4.sin_port = Utils::hton(port); |
||||
_v4UdpSocket = _phy.udpBind((const struct sockaddr *)&in4,this,131072); |
||||
if (!_v4UdpSocket) |
||||
throw std::runtime_error("cannot bind to port (UDP/IPv4)"); |
||||
_v4TcpListenSocket = _phy.tcpListen((const struct sockaddr *)&in4,this); |
||||
if (!_v4TcpListenSocket) { |
||||
_phy.close(_v4UdpSocket); |
||||
throw std::runtime_error("cannot bind to port (TCP/IPv4)"); |
||||
} |
||||
|
||||
::memset(in6,0,sizeof(in6)); |
||||
in6.sin_family = AF_INET6; |
||||
in6.sin_port = in4.sin_port; |
||||
_v6UdpSocket = _phy.udpBind((const struct sockaddr *)&in6,this,131072); |
||||
_v6TcpListenSocket = _phy.tcpListen((const struct sockaddr *)&in6,this); |
||||
|
||||
_thread = Thread::start(this); |
||||
} |
||||
|
||||
virtual ~OneImpl() |
||||
{ |
||||
if (reasonForTermination() == ONE_STILL_RUNNING) { |
||||
terminate(); |
||||
waitForTermination(); |
||||
} |
||||
_phy.close(_v4UdpSocket); |
||||
_phy.close(_v6UdpSocket); |
||||
_phy.close(_v4TcpListenSocket); |
||||
_phy.close(_v6TcpListenSocket); |
||||
} |
||||
|
||||
virtual ReasonForTermination reasonForTermination() |
||||
{ |
||||
Mutex::Lock _l(_termReason_m); |
||||
return _termReason; |
||||
} |
||||
|
||||
virtual void waitForTermination() |
||||
{ |
||||
if (reasonForTermination() == ONE_STILL_RUNNING) |
||||
Thread::join(_thread); |
||||
} |
||||
|
||||
virtual std::string fatalErrorMessage() |
||||
{ |
||||
Mutex::Lock _l(_termReason_m); |
||||
return _fatalErrorMessage; |
||||
} |
||||
|
||||
virtual void terminate() |
||||
{ |
||||
_run_m.lock(); |
||||
_run = false; |
||||
_run_m.unlock(); |
||||
_phy.whack(); |
||||
} |
||||
|
||||
// Begin private implementation methods
|
||||
|
||||
inline void phyOnDatagramFunction(PhySocket *sock,const struct sockaddr *from,void *data,unsigned long len) |
||||
{ |
||||
InetAddress fromss(from); |
||||
ZT1_ResultCode rc = _node->processWirePacket(OSUtils::now(),(const struct sockaddr_storage *)&fromss,0,reinterpret_cast<uint64_t *>(&_nextBackgroundTaskDeadline)); |
||||
if (ZT1_ResultCode_isFatal(rc)) { |
||||
char tmp[256]; |
||||
Utils::snprintf(tmp,sizeof(tmp),"fatal error code from processWirePacket(%d)",(int)rc); |
||||
Mutex::Lock _termReason_m; |
||||
_termReason = ONE_UNRECOVERABLE_ERROR; |
||||
_fatalErrorMessage = tmp; |
||||
this->terminate(); |
||||
} |
||||
} |
||||
|
||||
inline void phyOnTcpConnectFunction(PhySocket *sock,bool success) |
||||
{ |
||||
} |
||||
|
||||
inline void phyOnTcpAcceptFunction(PhySocket *sockN,const struct sockaddr *from) |
||||
{ |
||||
} |
||||
|
||||
inline void phyOnTcpCloseFunction(PhySocket *sock) |
||||
{ |
||||
} |
||||
|
||||
inline void phyOnTcpDataFunction(PhySocket *sock,void *data,unsigned long len) |
||||
{ |
||||
} |
||||
|
||||
inline void phyOnTcpWritableFunction(PhySocket *sock) |
||||
{ |
||||
} |
||||
|
||||
inline int nodeVirtualNetworkConfigFunction(uint64_t nwid,enum ZT1_VirtualNetworkConfigOperation op,const ZT1_VirtualNetworkConfig *nwconf) |
||||
{ |
||||
} |
||||
|
||||
inline void nodeEventCallback(enum ZT1_Event event,const void *metaData) |
||||
{ |
||||
switch(event) { |
||||
case ZT1_EVENT_FATAL_ERROR_IDENTITY_COLLISION: { |
||||
Mutex::Lock _termReason_m; |
||||
_termReason = ONE_IDENTITY_COLLISION; |
||||
_fatalErrorMessage = "identity/address collision"; |
||||
this->terminate(); |
||||
} break; |
||||
|
||||
case ZT1_EVENT_SAW_MORE_RECENT_VERSION: { |
||||
} break; |
||||
|
||||
case ZT1_EVENT_TRACE: { |
||||
if (metaData) { |
||||
::fprintf(stderr,"%s"ZT_EOL_S,(const char *)metaData); |
||||
::fflush(stderr); |
||||
} |
||||
} break; |
||||
|
||||
default: |
||||
break; |
||||
} |
||||
} |
||||
|
||||
inline long nodeDataStoreGetFunction(const char *name,void *buf,unsigned long bufSize,unsigned long readIndex,unsigned long *totalSize) |
||||
{ |
||||
} |
||||
|
||||
inline int nodeDataStorePutFunction(const char *name,const void *data,unsigned long len,int secure) |
||||
{ |
||||
} |
||||
|
||||
inline int nodeWirePacketSendFunction(const struct sockaddr_storage *addr,unsigned int desperation,const void *data,unsigned int len) |
||||
{ |
||||
switch(addr->ss_family) { |
||||
case AF_INET: |
||||
if (_v4UdpSocket) |
||||
_phy.udpSend(_v4UdpSocket,(const struct sockaddr *)addr,data,len); |
||||
break; |
||||
case AF_INET6: |
||||
if (_v6UdpSocket) |
||||
_phy.udpSend(_v6UdpSocket,(const struct sockaddr *)addr,data,len); |
||||
break; |
||||
} |
||||
} |
||||
|
||||
inline void nodeVirtualNetworkFrameFunction(uint64_t nwid,uint64_t sourceMac,uint64_t destMac,unsigned int etherType,unsigned int vlanId,const void *data,unsigned int len) |
||||
{ |
||||
} |
||||
|
||||
void threadMain() |
||||
throw() |
||||
{ |
||||
_nextBackgroundTaskDeadline = OSUtils::now(); |
||||
try { |
||||
_node = new Node( |
||||
OSUtils::now(), |
||||
this, |
||||
SnodeDataStoreGetFunction, |
||||
SnodeDataStorePutFunction, |
||||
SnodeWirePacketSendFunction, |
||||
SnodeVirtualNetworkFrameFunction, |
||||
SnodeVirtualNetworkConfigFunction, |
||||
SnodeEventCallback, |
||||
((_overrideRootTopology.length() > 0) ? _overrideRootTopology.c_str() : (const char *)0)); |
||||
|
||||
if (_master) |
||||
_node->setNetconfMaster((void *)_master); |
||||
|
||||
for(;;) { |
||||
_run_m.lock(); |
||||
if (!_run) { |
||||
_run_m.unlock(); |
||||
break; |
||||
} else _run_m.unlock(); |
||||
|
||||
const uint64_t dl = _nextBackgroundTaskDeadline; |
||||
const uint64_t now = OSUtils::now(); |
||||
_phy.poll((dl > now) ? (unsigned long)(dl - now) : 500); |
||||
} |
||||
} catch (std::exception &exc) { |
||||
Mutex::Lock _l(_termReason_m); |
||||
_termReason = ONE_UNRECOVERABLE_ERROR; |
||||
_fatalErrorMessage = exc.what(); |
||||
} catch ( ... ) { |
||||
Mutex::Lock _l(_termReason_m); |
||||
_termReason = ONE_UNRECOVERABLE_ERROR; |
||||
_fatalErrorMessage = "unexpected exception in main thread"; |
||||
} |
||||
delete _node; |
||||
} |
||||
|
||||
private: |
||||
SimpleFunctionPhy _phy; |
||||
NetworkConfigMaster *_master; |
||||
std::string _overrideRootTopology; |
||||
Node *_node; |
||||
PhySocket *_v4UdpSocket; |
||||
PhySocket *_v6UdpSocket; |
||||
PhySocket *_v4TcpListenSocket; |
||||
PhySocket *_v6TcpListenSocket; |
||||
Thread _thread; |
||||
volatile uint64_t _nextBackgroundTaskDeadline; |
||||
|
||||
ReasonForTermination _termReason; |
||||
std::string _fatalErrorMessage; |
||||
Mutex _termReason_m; |
||||
|
||||
bool _run; |
||||
Mutex _run_m; |
||||
}; |
||||
|
||||
static void SphyOnDatagramFunction(PhySocket *sock,void **uptr,const struct sockaddr *from,void *data,unsigned long len) |
||||
{ reinterpret_cast<OneImpl *>(*uptr)->phyOnDatagramFunction(sock,data,len); } |
||||
static void SphyOnTcpConnectFunction(PhySocket *sock,void **uptr,bool success) |
||||
{ reinterpret_cast<OneImpl *>(*uptr)->phyOnTcpConnectFunction(sock,success); } |
||||
static void SphyOnTcpAcceptFunction(PhySocket *sockL,PhySocket *sockN,void **uptrL,void **uptrN,const struct sockaddr *from) |
||||
{ *uptrN = *uptrL; reinterpret_cast<OneImpl *>(*uptrL)->phyOnTcpAcceptFunction(sockN,from); } |
||||
static void SphyOnTcpCloseFunction(PhySocket *sock,void **uptr) |
||||
{ reinterpret_cast<OneImpl *>(*uptr)->phyOnTcpCloseFunction(sock); } |
||||
static void SphyOnTcpDataFunction(PhySocket *sock,void **uptr,void *data,unsigned long len) |
||||
{ reinterpret_cast<OneImpl *>(*uptr)->phyOnTcpDataFunction(sock,data,len); } |
||||
static void SphyOnTcpWritableFunction(PhySocket *sock,void **uptr) |
||||
{ reinterpret_cast<OneImpl *>(*uptr)->phyOnTcpWritableFunction(sock); } |
||||
|
||||
static int SnodeVirtualNetworkConfigFunction(ZT1_Node *node,void *uptr,uint64_t nwid,enum ZT1_VirtualNetworkConfigOperation op,const ZT1_VirtualNetworkConfig *nwconf) |
||||
{ reinterpret_cast<OneImpl *>(uptr)->nodeVirtualNetworkConfigFunction(nwid,op,nwconf); } |
||||
static void SnodeEventCallback(ZT1_Node *node,void *uptr,enum ZT1_Event event,const void *metaData) |
||||
{ reinterpret_cast<OneImpl *>(uptr)->nodeEventCallback(event,metaData); } |
||||
static long SnodeDataStoreGetFunction(ZT1_Node *node,void *uptr,const char *name,void *buf,unsigned long bufSize,unsigned long readIndex,unsigned long *totalSize) |
||||
{ reinterpret_cast<OneImpl *>(uptr)->nodeDataStoreGetFunction(name,buf,bufSize,readIndex,totalSize); } |
||||
static int SnodeDataStorePutFunction(ZT1_Node *node,void *uptr,const char *name,const void *data,unsigned long len,int secure) |
||||
{ reinterpret_cast<OneImpl *>(uptr)->nodeDataStorePutFunction(name,data,len,secure); } |
||||
static int SnodeWirePacketSendFunction(ZT1_Node *node,void *uptr,const struct sockaddr_storage *addr,unsigned int desperation,const void *data,unsigned int len) |
||||
{ reinterpret_cast<OneImpl *>(uptr)->nodeWirePacketSendFunction(addr,desperation,data,len); } |
||||
static void SnodeVirtualNetworkFrameFunction(ZT1_Node *node,void *uptr,uint64_t nwid,uint64_t sourceMac,uint64_t destMac,unsigned int etherType,unsigned int vlanId,const void *data,unsigned int len) |
||||
{ reinterpret_cast<OneImpl *>(uptr)->nodeVirtualNetworkFrameFunction(nwid,sourceMac,destMac,etherType,vlanId,data,len); } |
||||
|
||||
One *One::newInstance(const char *hp,unsigned int port,NetworkConfigMaster *master,const char *overrideRootTopology) { return new OneImpl(hp,port,master,overrideRootTopology); } |
||||
One::~One() {} |
||||
|
||||
} // namespace ZeroTier
|
||||
@ -0,0 +1,126 @@
|
||||
/*
|
||||
* ZeroTier One - Network Virtualization Everywhere |
||||
* Copyright (C) 2011-2015 ZeroTier, Inc. |
||||
* |
||||
* 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_ONE_HPP |
||||
#define ZT_ONE_HPP |
||||
|
||||
namespace ZeroTier { |
||||
|
||||
class NetworkConfigMaster; |
||||
|
||||
/**
|
||||
* ZeroTier One -- local VPN/NVF service built around ZeroTier core |
||||
* |
||||
* Actual implementation is under the fold, hence the pure virtual |
||||
* interface. |
||||
*/ |
||||
class One |
||||
{ |
||||
public: |
||||
/**
|
||||
* Returned by node main if/when it terminates |
||||
*/ |
||||
enum ReasonForTermination |
||||
{ |
||||
/**
|
||||
* Instance is still running |
||||
*/ |
||||
ONE_STILL_RUNNING = 0, |
||||
|
||||
/**
|
||||
* Normal shutdown |
||||
*/ |
||||
ONE_NORMAL_TERMINATION = 1, |
||||
|
||||
/**
|
||||
* A serious unrecoverable error has occurred |
||||
*/ |
||||
ONE_UNRECOVERABLE_ERROR = 2, |
||||
|
||||
/**
|
||||
* Your identity has collided with another |
||||
*/ |
||||
ONE_IDENTITY_COLLISION = 3 |
||||
}; |
||||
|
||||
/**
|
||||
* Create and start a new instance of the service |
||||
* |
||||
* @param hp Home path |
||||
* @param port TCP and UDP port for packets and HTTP control |
||||
* @param master Instance of network config master if this instance is to act as one (default: NULL) |
||||
* @param overrideRootTopology String-serialized root topology (for testing, default: NULL) |
||||
*/ |
||||
static One *newInstance( |
||||
const char *hp, |
||||
unsigned int port, |
||||
NetworkConfigMaster *master = (NetworkConfigMaster *)0), |
||||
const char *overrideRootTopology = (const char *)0); |
||||
|
||||
/**
|
||||
* Deletion will block until service stops if it's still running |
||||
*/ |
||||
virtual ~One(); |
||||
|
||||
/**
|
||||
* @return Reason for terminating or ONE_STILL_RUNNING if running |
||||
*/ |
||||
virtual ReasonForTermination reasonForTermination() = 0; |
||||
|
||||
/**
|
||||
* @return Fatal error message or empty string if none |
||||
*/ |
||||
virtual std::string fatalErrorMessage() = 0; |
||||
|
||||
/**
|
||||
* Block until service terminates |
||||
*/ |
||||
virtual void waitForTermination() = 0; |
||||
|
||||
/**
|
||||
* Terminate background service |
||||
* |
||||
* Actual shutdown might take a few seconds. |
||||
*/ |
||||
virtual void terminate() = 0; |
||||
|
||||
/**
|
||||
* @return True if service is still running |
||||
*/ |
||||
inline isRunning() const { return (this->reasonForTermination() == ONE_STILL_RUNNING); } |
||||
|
||||
protected: |
||||
One() {} |
||||
|
||||
private: |
||||
One(const One &one) {} |
||||
inline One &operator=(const One &one) { return *this; } |
||||
}; |
||||
|
||||
} // namespace ZeroTier
|
||||
|
||||
#endif |
||||
Loading…
Reference in new issue