summaryrefslogtreecommitdiff
path: root/node
diff options
context:
space:
mode:
authorAdam Ierymenko <adam.ierymenko@gmail.com>2014-10-21 11:25:31 -0700
committerAdam Ierymenko <adam.ierymenko@gmail.com>2014-10-21 11:25:31 -0700
commit6bc9a938cfde3ff6caabc0a38f6e6d4650e3c384 (patch)
treec5e00198b25de72269a67ddabcf2dd180cfa4c10 /node
parent0e47f13f146ffd3c834ab3b4cecd6e66482cff02 (diff)
downloadinfinitytier-6bc9a938cfde3ff6caabc0a38f6e6d4650e3c384.tar.gz
infinitytier-6bc9a938cfde3ff6caabc0a38f6e6d4650e3c384.zip
Move socket manager implementation into osnet/
Diffstat (limited to 'node')
-rw-r--r--node/SocketManager.cpp720
-rw-r--r--node/SocketManager.hpp190
2 files changed, 0 insertions, 910 deletions
diff --git a/node/SocketManager.cpp b/node/SocketManager.cpp
deleted file mode 100644
index 8e3ee9e0..00000000
--- a/node/SocketManager.cpp
+++ /dev/null
@@ -1,720 +0,0 @@
-/*
- * 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 <string.h>
-#include <stdlib.h>
-#include <fcntl.h>
-#include <time.h>
-#include <sys/types.h>
-
-#include "SocketManager.hpp"
-#include "UdpSocket.hpp"
-#include "TcpSocket.hpp"
-
-#ifndef __WINDOWS__
-#include <errno.h>
-#include <unistd.h>
-#include <sys/socket.h>
-#include <arpa/inet.h>
-#include <signal.h>
-#include <netinet/in.h>
-#include <netinet/tcp.h>
-#endif
-
-// Uncomment to turn off TCP Nagle
-//#define ZT_TCP_NODELAY
-
-// Allow us to use the same value on Windows and *nix
-#ifndef INVALID_SOCKET
-#define INVALID_SOCKET (-1)
-#endif
-
-#ifdef __WINDOWS__
-#define CLOSE_SOCKET(s) ::closesocket(s)
-#else
-#define CLOSE_SOCKET(s) ::close(s)
-#endif
-
-namespace ZeroTier {
-
-#ifdef __WINDOWS__
-// hack copied from StackOverflow, behaves a bit like pipe() on *nix systems
-static inline void winPipeHack(SOCKET fds[2])
-{
- struct sockaddr_in inaddr;
- struct sockaddr addr;
- SOCKET lst=::socket(AF_INET, SOCK_STREAM,IPPROTO_TCP);
- memset(&inaddr, 0, sizeof(inaddr));
- memset(&addr, 0, sizeof(addr));
- inaddr.sin_family = AF_INET;
- inaddr.sin_addr.s_addr = htonl(INADDR_LOOPBACK);
- inaddr.sin_port = 0;
- int yes=1;
- setsockopt(lst,SOL_SOCKET,SO_REUSEADDR,(char*)&yes,sizeof(yes));
- bind(lst,(struct sockaddr *)&inaddr,sizeof(inaddr));
- listen(lst,1);
- int len=sizeof(inaddr);
- getsockname(lst, &addr,&len);
- fds[0]=::socket(AF_INET, SOCK_STREAM,0);
- connect(fds[0],&addr,len);
- fds[1]=accept(lst,0,0);
- closesocket(lst);
-}
-#endif
-
-SocketManager::SocketManager(
- int localUdpPort,
- int localTcpPort,
- void (*packetHandler)(const SharedPtr<Socket> &,void *,const InetAddress &,Buffer<ZT_SOCKET_MAX_MESSAGE_LEN> &),
- void *arg) :
- _whackSendPipe(INVALID_SOCKET),
- _whackReceivePipe(INVALID_SOCKET),
- _tcpV4ListenSocket(INVALID_SOCKET),
- _tcpV6ListenSocket(INVALID_SOCKET),
- _nfds(0),
- _packetHandler(packetHandler),
- _arg(arg)
-{
- FD_ZERO(&_readfds);
- FD_ZERO(&_writefds);
-
- // Create a pipe or socket pair that can be used to interrupt select()
-#ifdef __WINDOWS__
- {
- SOCKET tmps[2] = { INVALID_SOCKET,INVALID_SOCKET };
- winPipeHack(tmps);
- _whackSendPipe = tmps[0];
- _whackReceivePipe = tmps[1];
- u_long iMode=1;
- ioctlsocket(tmps[1],FIONBIO,&iMode);
- }
-#else
- {
- int tmpfds[2];
- if (::pipe(tmpfds))
- throw std::runtime_error("pipe() failed");
- _whackSendPipe = tmpfds[1];
- _whackReceivePipe = tmpfds[0];
- fcntl(_whackReceivePipe,F_SETFL,O_NONBLOCK);
- }
-#endif
- FD_SET(_whackReceivePipe,&_readfds);
-
- if (localTcpPort > 0) {
- if (localTcpPort > 0xffff) {
- _closeSockets();
- throw std::runtime_error("invalid local TCP port number");
- }
-
- { // bind TCP IPv6
- _tcpV6ListenSocket = ::socket(AF_INET6,SOCK_STREAM,0);
-#ifdef __WINDOWS__
- if (_tcpV6ListenSocket != INVALID_SOCKET) {
- {
- BOOL f;
- f = TRUE; ::setsockopt(_tcpV6ListenSocket,IPPROTO_IPV6,IPV6_V6ONLY,(const char *)&f,sizeof(f));
- f = TRUE; ::setsockopt(_tcpV6ListenSocket,SOL_SOCKET,SO_REUSEADDR,(const char *)&f,sizeof(f));
- u_long iMode=1;
- ioctlsocket(_tcpV6ListenSocket,FIONBIO,&iMode);
- }
-#else
- if (_tcpV6ListenSocket > 0) {
- {
- int f;
- f = 1; ::setsockopt(_tcpV6ListenSocket,IPPROTO_IPV6,IPV6_V6ONLY,(void *)&f,sizeof(f));
- f = 1; ::setsockopt(_tcpV6ListenSocket,SOL_SOCKET,SO_REUSEADDR,(void *)&f,sizeof(f));
- fcntl(_tcpV6ListenSocket,F_SETFL,O_NONBLOCK);
- }
-#endif // __WINDOWS__ / not __WINDOWS__
-
- struct sockaddr_in6 sin6;
- memset(&sin6,0,sizeof(sin6));
- sin6.sin6_family = AF_INET6;
- sin6.sin6_port = htons(localTcpPort);
- memcpy(&(sin6.sin6_addr),&in6addr_any,sizeof(struct in6_addr));
- if (::bind(_tcpV6ListenSocket,(const struct sockaddr *)&sin6,sizeof(sin6))) {
- _closeSockets();
- throw std::runtime_error("unable to bind to local TCP port");
- }
-
- if (::listen(_tcpV6ListenSocket,16)) {
- _closeSockets();
- throw std::runtime_error("listen() failed");
- }
-
- FD_SET(_tcpV6ListenSocket,&_readfds);
- }
- }
-
- { // bind TCP IPv4
- _tcpV4ListenSocket = ::socket(AF_INET,SOCK_STREAM,0);
-#ifdef __WINDOWS__
- if (_tcpV4ListenSocket == INVALID_SOCKET) {
-#else
- if (_tcpV4ListenSocket <= 0) {
-#endif
- _closeSockets();
- throw std::runtime_error("unable to create IPv4 SOCK_STREAM socket");
- }
-
-#ifdef __WINDOWS__
- {
- BOOL f = TRUE; ::setsockopt(_tcpV4ListenSocket,SOL_SOCKET,SO_REUSEADDR,(const char *)&f,sizeof(f));
- u_long iMode=1;
- ioctlsocket(_tcpV4ListenSocket,FIONBIO,&iMode);
- }
-#else
- {
- int f = 1; ::setsockopt(_tcpV4ListenSocket,SOL_SOCKET,SO_REUSEADDR,(void *)&f,sizeof(f));
- fcntl(_tcpV4ListenSocket,F_SETFL,O_NONBLOCK);
- }
-#endif
-
- struct sockaddr_in sin4;
- memset(&sin4,0,sizeof(sin4));
- sin4.sin_family = AF_INET;
- sin4.sin_port = htons(localTcpPort);
- sin4.sin_addr.s_addr = INADDR_ANY;
- if (::bind(_tcpV4ListenSocket,(const struct sockaddr *)&sin4,sizeof(sin4))) {
- _closeSockets();
- throw std::runtime_error("unable to bind to local TCP port");
- }
-
- if (::listen(_tcpV4ListenSocket,16)) {
- _closeSockets();
- throw std::runtime_error("listen() failed");
- }
-
- FD_SET(_tcpV4ListenSocket,&_readfds);
- }
- }
-
- if (localUdpPort > 0) {
- if (localUdpPort > 0xffff) {
- _closeSockets();
- throw std::runtime_error("invalid local UDP port number");
- }
-
- { // bind UDP IPv6
-#ifdef __WINDOWS__
- SOCKET s = ::socket(AF_INET6,SOCK_DGRAM,0);
- if (s != INVALID_SOCKET) {
-#else
- int s = ::socket(AF_INET6,SOCK_DGRAM,0);
- if (s > 0) {
-#endif
-
- {
- int bs = 1048576;
- while (bs >= 65536) {
- int tmpbs = bs;
- if (setsockopt(s,SOL_SOCKET,SO_RCVBUF,(const char *)&tmpbs,sizeof(tmpbs)) == 0)
- break;
- bs -= 16384;
- }
- bs = 1048576;
- while (bs >= 65536) {
- int tmpbs = bs;
- if (setsockopt(s,SOL_SOCKET,SO_SNDBUF,(const char *)&tmpbs,sizeof(tmpbs)) == 0)
- break;
- bs -= 16384;
- }
-#ifdef __WINDOWS__
- BOOL f;
- f = TRUE; setsockopt(s,IPPROTO_IPV6,IPV6_V6ONLY,(const char *)&f,sizeof(f));
- f = FALSE; setsockopt(s,SOL_SOCKET,SO_REUSEADDR,(const char *)&f,sizeof(f));
- f = FALSE; setsockopt(s,IPPROTO_IPV6,IPV6_DONTFRAG,(const char *)&f,sizeof(f));
- f = TRUE; setsockopt(s,SOL_SOCKET,SO_BROADCAST,(const char *)&f,sizeof(f));
-#else
- int f;
- f = 1; setsockopt(s,IPPROTO_IPV6,IPV6_V6ONLY,(void *)&f,sizeof(f));
- f = 0; setsockopt(s,SOL_SOCKET,SO_REUSEADDR,(void *)&f,sizeof(f));
- f = 1; setsockopt(s,SOL_SOCKET,SO_BROADCAST,(void *)&f,sizeof(f));
-#ifdef IP_DONTFRAG
- f = 0; setsockopt(s,IPPROTO_IP,IP_DONTFRAG,&f,sizeof(f));
-#endif
-#ifdef IP_MTU_DISCOVER
- f = 0; setsockopt(s,IPPROTO_IP,IP_MTU_DISCOVER,&f,sizeof(f));
-#endif
-#ifdef IPV6_MTU_DISCOVER
- f = 0; setsockopt(s,IPPROTO_IPV6,IPV6_MTU_DISCOVER,&f,sizeof(f));
-#endif
-#endif
- }
-
- struct sockaddr_in6 sin6;
- memset(&sin6,0,sizeof(sin6));
- sin6.sin6_family = AF_INET6;
- sin6.sin6_port = htons(localUdpPort);
- memcpy(&(sin6.sin6_addr),&in6addr_any,sizeof(struct in6_addr));
- if (::bind(s,(const struct sockaddr *)&sin6,sizeof(sin6))) {
- CLOSE_SOCKET(s);
- _closeSockets();
- throw std::runtime_error("unable to bind to port");
- }
-
- _udpV6Socket = SharedPtr<Socket>(new UdpSocket(Socket::ZT_SOCKET_TYPE_UDP_V6,s));
-#ifdef __WINDOWS__
- u_long iMode=1;
- ioctlsocket(s,FIONBIO,&iMode);
-#else
- fcntl(s,F_SETFL,O_NONBLOCK);
-#endif
- FD_SET(s,&_readfds);
- }
- }
-
- { // bind UDP IPv4
-#ifdef __WINDOWS__
- SOCKET s = ::socket(AF_INET,SOCK_DGRAM,0);
- if (s == INVALID_SOCKET) {
- _closeSockets();
- throw std::runtime_error("unable to create IPv4 SOCK_DGRAM socket");
- }
-#else
- int s = ::socket(AF_INET,SOCK_DGRAM,0);
- if (s <= 0) {
- _closeSockets();
- throw std::runtime_error("unable to create IPv4 SOCK_DGRAM socket");
- }
-#endif
-
- {
- int bs = 1048576;
- while (bs >= 65536) {
- int tmpbs = bs;
- if (setsockopt(s,SOL_SOCKET,SO_RCVBUF,(const char *)&tmpbs,sizeof(tmpbs)) == 0)
- break;
- bs -= 16384;
- }
- bs = 1048576;
- while (bs >= 65536) {
- int tmpbs = bs;
- if (setsockopt(s,SOL_SOCKET,SO_SNDBUF,(const char *)&tmpbs,sizeof(tmpbs)) == 0)
- break;
- bs -= 16384;
- }
-#ifdef __WINDOWS__
- BOOL f;
- f = FALSE; setsockopt(s,SOL_SOCKET,SO_REUSEADDR,(const char *)&f,sizeof(f));
- f = FALSE; setsockopt(s,IPPROTO_IP,IP_DONTFRAGMENT,(const char *)&f,sizeof(f));
- f = TRUE; setsockopt(s,SOL_SOCKET,SO_BROADCAST,(const char *)&f,sizeof(f));
-#else
- int f;
- f = 0; setsockopt(s,SOL_SOCKET,SO_REUSEADDR,(void *)&f,sizeof(f));
- f = 1; setsockopt(s,SOL_SOCKET,SO_BROADCAST,(void *)&f,sizeof(f));
-#ifdef IP_DONTFRAG
- f = 0; setsockopt(s,IPPROTO_IP,IP_DONTFRAG,&f,sizeof(f));
-#endif
-#ifdef IP_MTU_DISCOVER
- f = 0; setsockopt(s,IPPROTO_IP,IP_MTU_DISCOVER,&f,sizeof(f));
-#endif
-#endif
- }
-
- struct sockaddr_in sin4;
- memset(&sin4,0,sizeof(sin4));
- sin4.sin_family = AF_INET;
- sin4.sin_port = htons(localUdpPort);
- sin4.sin_addr.s_addr = INADDR_ANY;
- if (::bind(s,(const struct sockaddr *)&sin4,sizeof(sin4))) {
- CLOSE_SOCKET(s);
- _closeSockets();
- throw std::runtime_error("unable to bind to port");
- }
-
- _udpV4Socket = SharedPtr<Socket>(new UdpSocket(Socket::ZT_SOCKET_TYPE_UDP_V4,s));
-#ifdef __WINDOWS__
- u_long iMode=1;
- ioctlsocket(s,FIONBIO,&iMode);
-#else
- fcntl(s,F_SETFL,O_NONBLOCK);
-#endif
- FD_SET(s,&_readfds);
- }
- }
-
- _updateNfds();
-}
-
-SocketManager::~SocketManager()
-{
- Mutex::Lock _l(_pollLock);
- _closeSockets();
-}
-
-bool SocketManager::send(const InetAddress &to,bool tcp,bool autoConnectTcp,const void *msg,unsigned int msglen)
-{
- if (tcp) {
- SharedPtr<Socket> ts;
- {
- Mutex::Lock _l(_tcpSockets_m);
- std::map< InetAddress,SharedPtr<Socket> >::iterator opents(_tcpSockets.find(to));
- if (opents != _tcpSockets.end())
- ts = opents->second;
- }
- if (ts)
- return ts->send(to,msg,msglen);
-
- if (!autoConnectTcp)
- return false;
-
-#ifdef __WINDOWS__
- SOCKET s = ::socket(to.isV4() ? AF_INET : AF_INET6,SOCK_STREAM,0);
- if (s == INVALID_SOCKET)
- return false;
- { u_long iMode=1; ioctlsocket(s,FIONBIO,&iMode); }
-#ifdef ZT_TCP_NODELAY
- { BOOL f = TRUE; setsockopt(s,IPPROTO_TCP,TCP_NODELAY,(char *)&f,sizeof(f)); }
-#endif
-#else
- int s = ::socket(to.isV4() ? AF_INET : AF_INET6,SOCK_STREAM,0);
- if (s <= 0)
- return false;
- if (s >= FD_SETSIZE) {
- ::close(s);
- return false;
- }
- fcntl(s,F_SETFL,O_NONBLOCK);
-#ifdef ZT_TCP_NODELAY
- { int f = 1; setsockopt(s,IPPROTO_TCP,TCP_NODELAY,(char *)&f,sizeof(f)); }
-#endif
-#endif
-
- bool connecting = false;
- if (::connect(s,to.saddr(),to.saddrLen())) {
-#ifdef __WINDOWS__
- if (WSAGetLastError() != WSAEWOULDBLOCK) {
-#else
- if (errno != EINPROGRESS) {
-#endif
- CLOSE_SOCKET(s);
- return false;
- } else connecting = true;
- }
-
- ts = SharedPtr<Socket>(new TcpSocket(this,s,Socket::ZT_SOCKET_TYPE_TCP_OUT,connecting,to));
- if (!ts->send(to,msg,msglen)) {
- _fdSetLock.lock();
- FD_CLR(s,&_readfds);
- FD_CLR(s,&_writefds);
- _fdSetLock.unlock();
- return false;
- }
-
- {
- Mutex::Lock _l(_tcpSockets_m);
- _tcpSockets[to] = ts;
- }
-
- _fdSetLock.lock();
- FD_SET(s,&_readfds);
- if (connecting)
- FD_SET(s,&_writefds);
- _fdSetLock.unlock();
-
- _updateNfds();
- whack();
-
- return true;
- } else if (to.isV4()) {
- if (_udpV4Socket)
- return _udpV4Socket->send(to,msg,msglen);
- } else if (to.isV6()) {
- if (_udpV6Socket)
- return _udpV6Socket->send(to,msg,msglen);
- }
- return false;
-}
-
-void SocketManager::poll(unsigned long timeout)
-{
- fd_set rfds,wfds,efds;
- struct timeval tv;
- std::vector< SharedPtr<Socket> > ts;
-#ifdef __WINDOWS__
- SOCKET sockfd;
-#else
- int sockfd;
-#endif
-
- Mutex::Lock _l(_pollLock);
-
- _fdSetLock.lock();
- memcpy(&rfds,&_readfds,sizeof(rfds));
- memcpy(&wfds,&_writefds,sizeof(wfds));
- _fdSetLock.unlock();
- FD_ZERO(&efds);
-
-#ifdef __WINDOWS__
- // Windows signals failed connects in exceptfds
- {
- Mutex::Lock _l2(_tcpSockets_m);
- for(std::map< InetAddress,SharedPtr<Socket> >::iterator s(_tcpSockets.begin());s!=_tcpSockets.end();++s) {
- if (((TcpSocket *)s->second.ptr())->_connecting)
- FD_SET(s->second->_sock,&efds);
- }
- }
-#endif
-
- tv.tv_sec = (long)(timeout / 1000);
- tv.tv_usec = (long)((timeout % 1000) * 1000);
- select(_nfds + 1,&rfds,&wfds,&efds,(timeout > 0) ? &tv : (struct timeval *)0);
-
- if (FD_ISSET(_whackReceivePipe,&rfds)) {
- char tmp[16];
-#ifdef __WINDOWS__
- ::recv(_whackReceivePipe,tmp,16,0);
-#else
- ::read(_whackReceivePipe,tmp,16);
-#endif
- }
-
- if ((_tcpV4ListenSocket != INVALID_SOCKET)&&(FD_ISSET(_tcpV4ListenSocket,&rfds))) {
- struct sockaddr_in from;
- socklen_t fromlen = sizeof(from);
- sockfd = accept(_tcpV4ListenSocket,(struct sockaddr *)&from,&fromlen);
-#ifdef __WINDOWS__
- if (sockfd != INVALID_SOCKET) {
-#else
- if (sockfd > 0) {
- if (sockfd < FD_SETSIZE) {
-#endif
- InetAddress fromia((const struct sockaddr *)&from);
- Mutex::Lock _l2(_tcpSockets_m);
- try {
- _tcpSockets[fromia] = SharedPtr<Socket>(new TcpSocket(this,sockfd,Socket::ZT_SOCKET_TYPE_TCP_IN,false,fromia));
-#ifdef __WINDOWS__
- { u_long iMode=1; ioctlsocket(sockfd,FIONBIO,&iMode); }
-#ifdef ZT_TCP_NODELAY
- { BOOL f = TRUE; setsockopt(sockfd,IPPROTO_TCP,TCP_NODELAY,(char *)&f,sizeof(f)); }
-#endif
-#else
- fcntl(sockfd,F_SETFL,O_NONBLOCK);
-#ifdef ZT_TCP_NODELAY
- { int f = 1; setsockopt(sockfd,IPPROTO_TCP,TCP_NODELAY,(char *)&f,sizeof(f)); }
-#endif
-#endif
- _fdSetLock.lock();
- FD_SET(sockfd,&_readfds);
- _fdSetLock.unlock();
- if ((int)sockfd > (int)_nfds)
- _nfds = (int)sockfd;
- } catch ( ... ) {
- CLOSE_SOCKET(sockfd);
- }
-#ifndef __WINDOWS__
- } else {
- CLOSE_SOCKET(sockfd);
- }
-#endif
- }
- }
- if ((_tcpV6ListenSocket != INVALID_SOCKET)&&(FD_ISSET(_tcpV6ListenSocket,&rfds))) {
- struct sockaddr_in6 from;
- socklen_t fromlen = sizeof(from);
- sockfd = accept(_tcpV6ListenSocket,(struct sockaddr *)&from,&fromlen);
-#ifdef __WINDOWS__
- if (sockfd != INVALID_SOCKET) {
-#else
- if (sockfd > 0) {
- if (sockfd < FD_SETSIZE) {
-#endif
- InetAddress fromia((const struct sockaddr *)&from);
- Mutex::Lock _l2(_tcpSockets_m);
- try {
- _tcpSockets[fromia] = SharedPtr<Socket>(new TcpSocket(this,sockfd,Socket::ZT_SOCKET_TYPE_TCP_IN,false,fromia));
-#ifdef __WINDOWS__
- { u_long iMode=1; ioctlsocket(sockfd,FIONBIO,&iMode); }
-#ifdef ZT_TCP_NODELAY
- { BOOL f = TRUE; setsockopt(sockfd,IPPROTO_TCP,TCP_NODELAY,(char *)&f,sizeof(f)); }
-#endif
-#else
- fcntl(sockfd,F_SETFL,O_NONBLOCK);
-#ifdef ZT_TCP_NODELAY
- { int f = 1; setsockopt(sockfd,IPPROTO_TCP,TCP_NODELAY,(char *)&f,sizeof(f)); }
-#endif
-#endif
- _fdSetLock.lock();
- FD_SET(sockfd,&_readfds);
- _fdSetLock.unlock();
- if ((int)sockfd > (int)_nfds)
- _nfds = (int)sockfd;
- } catch ( ... ) {
- CLOSE_SOCKET(sockfd);
- }
-#ifndef __WINDOWS__
- } else {
- CLOSE_SOCKET(sockfd);
- }
-#endif
- }
- }
-
- if ((_udpV4Socket)&&(FD_ISSET(_udpV4Socket->_sock,&rfds))) {
- _udpV4Socket->notifyAvailableForRead(_udpV4Socket,this);
- }
- if ((_udpV6Socket)&&(FD_ISSET(_udpV6Socket->_sock,&rfds))) {
- _udpV6Socket->notifyAvailableForRead(_udpV6Socket,this);
- }
-
- bool closedSockets = false;
- { // grab copy of TCP sockets list because _tcpSockets[] might be changed in a handler
- Mutex::Lock _l2(_tcpSockets_m);
- if (!_tcpSockets.empty()) {
- ts.reserve(_tcpSockets.size());
- uint64_t now = Utils::now();
- for(std::map< InetAddress,SharedPtr<Socket> >::iterator s(_tcpSockets.begin());s!=_tcpSockets.end();) {
-#ifdef __WINDOWS__
- if ( ((now - ((TcpSocket *)s->second.ptr())->_lastActivity) < ZT_TCP_TUNNEL_ACTIVITY_TIMEOUT) && (! ((((TcpSocket *)s->second.ptr())->_connecting)&&(FD_ISSET(s->second->_sock,&efds))) ) ) {
-#else
- if ((now - ((TcpSocket *)s->second.ptr())->_lastActivity) < ZT_TCP_TUNNEL_ACTIVITY_TIMEOUT) {
-#endif
- ts.push_back(s->second);
- ++s;
- } else {
- _fdSetLock.lock();
- FD_CLR(s->second->_sock,&_readfds);
- FD_CLR(s->second->_sock,&_writefds);
- _fdSetLock.unlock();
- _tcpSockets.erase(s++);
- closedSockets = true;
- }
- }
- }
- }
- for(std::vector< SharedPtr<Socket> >::iterator s(ts.begin());s!=ts.end();++s) {
- if (FD_ISSET((*s)->_sock,&wfds)) {
- if (!(*s)->notifyAvailableForWrite(*s,this)) {
- {
- Mutex::Lock _l2(_tcpSockets_m);
- _tcpSockets.erase(((TcpSocket *)s->ptr())->_remote);
- }
- _fdSetLock.lock();
- FD_CLR((*s)->_sock,&_readfds);
- FD_CLR((*s)->_sock,&_writefds);
- _fdSetLock.unlock();
- closedSockets = true;
- continue;
- }
- }
- if (FD_ISSET((*s)->_sock,&rfds)) {
- if (!(*s)->notifyAvailableForRead(*s,this)) {
- {
- Mutex::Lock _l2(_tcpSockets_m);
- _tcpSockets.erase(((TcpSocket *)s->ptr())->_remote);
- }
- _fdSetLock.lock();
- FD_CLR((*s)->_sock,&_readfds);
- FD_CLR((*s)->_sock,&_writefds);
- _fdSetLock.unlock();
- closedSockets = true;
- continue;
- }
- }
- }
- if (closedSockets)
- _updateNfds();
-}
-
-void SocketManager::whack()
-{
- _whackSendPipe_m.lock();
-#ifdef __WINDOWS__
- ::send(_whackSendPipe,(const char *)this,1,0);
-#else
- ::write(_whackSendPipe,(const void *)this,1); // data is arbitrary, just send a byte
-#endif
- _whackSendPipe_m.unlock();
-}
-
-void SocketManager::closeTcpSockets()
-{
- {
- Mutex::Lock _l2(_tcpSockets_m);
- _fdSetLock.lock();
- for(std::map< InetAddress,SharedPtr<Socket> >::iterator s(_tcpSockets.begin());s!=_tcpSockets.end();++s) {
- FD_CLR(s->second->_sock,&_readfds);
- FD_CLR(s->second->_sock,&_writefds);
- }
- _fdSetLock.unlock();
- _tcpSockets.clear();
- }
- _updateNfds();
-}
-
-void SocketManager::_closeSockets()
- throw()
-{
-#ifdef __WINDOWS__
- if (_whackSendPipe != INVALID_SOCKET)
- ::closesocket(_whackSendPipe);
- if (_whackReceivePipe != INVALID_SOCKET)
- ::closesocket(_whackReceivePipe);
- if (_tcpV4ListenSocket != INVALID_SOCKET)
- ::closesocket(_tcpV4ListenSocket);
- if (_tcpV6ListenSocket != INVALID_SOCKET)
- ::closesocket(_tcpV6ListenSocket);
-#else
- if (_whackSendPipe > 0)
- ::close(_whackSendPipe);
- if (_whackReceivePipe > 0)
- ::close(_whackReceivePipe);
- if (_tcpV4ListenSocket > 0)
- ::close(_tcpV4ListenSocket);
- if (_tcpV4ListenSocket > 0)
- ::close(_tcpV6ListenSocket);
-#endif
-}
-
-void SocketManager::_updateNfds()
-{
-#ifdef __WINDOWS__
- SOCKET nfds = _whackSendPipe;
-#else
- int nfds = _whackSendPipe;
-#endif
- if (_whackReceivePipe > nfds)
- nfds = _whackReceivePipe;
- if (_tcpV4ListenSocket > nfds)
- nfds = _tcpV4ListenSocket;
- if (_tcpV6ListenSocket > nfds)
- nfds = _tcpV6ListenSocket;
- if ((_udpV4Socket)&&(_udpV4Socket->_sock > nfds))
- nfds = _udpV4Socket->_sock;
- if ((_udpV6Socket)&&(_udpV6Socket->_sock > nfds))
- nfds = _udpV6Socket->_sock;
- Mutex::Lock _l(_tcpSockets_m);
- for(std::map< InetAddress,SharedPtr<Socket> >::const_iterator s(_tcpSockets.begin());s!=_tcpSockets.end();++s) {
- if (s->second->_sock > nfds)
- nfds = s->second->_sock;
- }
- _nfds = (int)nfds;
-}
-
-} // namespace ZeroTier
diff --git a/node/SocketManager.hpp b/node/SocketManager.hpp
deleted file mode 100644
index 81d2e780..00000000
--- a/node/SocketManager.hpp
+++ /dev/null
@@ -1,190 +0,0 @@
-/*
- * 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_SOCKETMANAGER_HPP
-#define ZT_SOCKETMANAGER_HPP
-
-#include <stdio.h>
-#include <stdlib.h>
-
-#include <map>
-#include <stdexcept>
-
-#include "Constants.hpp"
-
-#include "SharedPtr.hpp"
-#include "InetAddress.hpp"
-#include "Socket.hpp"
-#include "Mutex.hpp"
-#include "NonCopyable.hpp"
-#include "Buffer.hpp"
-
-#ifdef __WINDOWS__
-#include <WinSock2.h>
-#include <WS2tcpip.h>
-#include <Windows.h>
-#else
-#include <unistd.h>
-#include <sys/time.h>
-#include <sys/types.h>
-#include <sys/select.h>
-#endif
-
-namespace ZeroTier {
-
-/**
- * Socket I/O multiplexer
- *
- * This wraps select(), epoll(), etc. and handles creation of Sockets.
- */
-class SocketManager : NonCopyable
-{
- friend class Socket;
- friend class UdpSocket;
- friend class TcpSocket;
-
-public:
- /**
- * @param localUdpPort Local UDP port to bind or 0 for no UDP support
- * @param localTcpPort Local TCP port to listen to or 0 for no incoming TCP connect support
- * @param packetHandler Function to call when packets are received by a socket
- * @param arg Second argument to packetHandler()
- * @throws std::runtime_error Could not bind local port(s) or open socket(s)
- */
- SocketManager(
- int localUdpPort,
- int localTcpPort,
- void (*packetHandler)(const SharedPtr<Socket> &,void *,const InetAddress &,Buffer<ZT_SOCKET_MAX_MESSAGE_LEN> &),
- void *arg);
-
- ~SocketManager();
-
- /**
- * Send a message to a remote peer
- *
- * @param to Destination address
- * @param tcp Use TCP?
- * @param autoConnectTcp If true, automatically initiate TCP connection if there is none
- * @param msg Message to send
- * @param msglen Length of message
- */
- bool send(const InetAddress &to,bool tcp,bool autoConnectTcp,const void *msg,unsigned int msglen);
-
- /**
- * Send a message to a remote peer via UDP (shortcut for setting both TCP params to false in send)
- *
- * @param to Destination address
- * @param msg Message to send
- * @param msglen Length of message
- */
- inline bool sendUdp(const InetAddress &to,const void *msg,unsigned int msglen) { return send(to,false,false,msg,msglen); }
-
- /**
- * Perform I/O polling operation (e.g. select())
- *
- * If called concurrently, one will block until the other completes.
- *
- * @param timeout Timeout in milliseconds, may return sooner if whack() is called
- */
- void poll(unsigned long timeout);
-
- /**
- * Cause current or next blocking poll() operation to timeout immediately
- */
- void whack();
-
- /**
- * Close TCP sockets
- */
- void closeTcpSockets();
-
-private:
- // Called by socket implementations when a packet is received
- inline void handleReceivedPacket(const SharedPtr<Socket> &sock,const InetAddress &from,Buffer<ZT_SOCKET_MAX_MESSAGE_LEN> &data)
- throw()
- {
- try {
- _packetHandler(sock,_arg,from,data);
- } catch ( ... ) {} // handlers shouldn't throw
- }
-
- // Used by TcpSocket to register/unregister for write availability notification
- inline void startNotifyWrite(const Socket *sock)
- throw()
- {
- _fdSetLock.lock();
- FD_SET(sock->_sock,&_writefds);
- _fdSetLock.unlock();
- }
- inline void stopNotifyWrite(const Socket *sock)
- throw()
- {
- _fdSetLock.lock();
- FD_CLR(sock->_sock,&_writefds);
- _fdSetLock.unlock();
- }
-
- // Called in SocketManager destructor or in constructor cleanup before exception throwing
- void _closeSockets()
- throw();
-
- // Called in SocketManager to recompute _nfds for select() based implementation
- void _updateNfds();
-
-#ifdef __WINDOWS__
- SOCKET _whackSendPipe;
- SOCKET _whackReceivePipe;
- SOCKET _tcpV4ListenSocket;
- SOCKET _tcpV6ListenSocket;
-#else
- int _whackSendPipe;
- int _whackReceivePipe;
- int _tcpV4ListenSocket;
- int _tcpV6ListenSocket;
-#endif
- Mutex _whackSendPipe_m;
-
- SharedPtr<Socket> _udpV4Socket;
- SharedPtr<Socket> _udpV6Socket;
-
- fd_set _readfds;
- fd_set _writefds;
- volatile int _nfds;
- Mutex _fdSetLock;
-
- std::map< InetAddress,SharedPtr<Socket> > _tcpSockets;
- Mutex _tcpSockets_m;
-
- void (*_packetHandler)(const SharedPtr<Socket> &,void *,const InetAddress &,Buffer<ZT_SOCKET_MAX_MESSAGE_LEN> &);
- void *_arg;
-
- Mutex _pollLock;
-};
-
-} // namespace ZeroTier
-
-#endif