diff options
Diffstat (limited to 'netcon/NetconEthernetTap.cpp')
| -rw-r--r-- | netcon/NetconEthernetTap.cpp | 45 |
1 files changed, 21 insertions, 24 deletions
diff --git a/netcon/NetconEthernetTap.cpp b/netcon/NetconEthernetTap.cpp index b882d62e..acbccef3 100644 --- a/netcon/NetconEthernetTap.cpp +++ b/netcon/NetconEthernetTap.cpp @@ -51,7 +51,7 @@ #define APPLICATION_POLL_FREQ 5 #define ZT_LWIP_TCP_TIMER_INTERVAL 10 -#define STATUS_TMR_INTERVAL 500 +#define STATUS_TMR_INTERVAL 500 // How often we check connection statuses namespace ZeroTier { @@ -293,7 +293,7 @@ void NetconEthernetTap::closeConnection(TcpConnection *conn) */ void NetconEthernetTap::closeClient(PhySocket *sock) { - for(int i=0; i<rpc_sockets.size(); i++) { + for(size_t i=0; i<rpc_sockets.size(); i++) { if(rpc_sockets[i] == sock){ rpc_sockets.erase(rpc_sockets.begin() + i); break; @@ -314,10 +314,11 @@ void NetconEthernetTap::closeAll() closeConnection(tcp_connections.front()); } +#include <sys/resource.h> + void NetconEthernetTap::threadMain() throw() { - fprintf(stderr, "_threadMain()\n"); uint64_t prev_tcp_time = 0; uint64_t prev_status_time = 0; uint64_t prev_etharp_time = 0; @@ -353,29 +354,25 @@ void NetconEthernetTap::threadMain() if (since_status >= STATUS_TMR_INTERVAL) { prev_status_time = now; - fprintf(stderr, "tcp_conns = %d, rpc_socks = %d\n", tcp_connections.size(), rpc_sockets.size()); - - /* Here we will periodically check the list of rpc_sockets for any which - do not currently have any data connection associated with them. If they are - unused, then we will try to read from them, if they fail, we can safely assumes - that the client has closed their and we can close ours */ - - for(int i=0, associated = 0; i<rpc_sockets.size(); i++, associated = 0) { - for(int j=0; j<tcp_connections.size(); j++) { - if(tcp_connections[j]->rpcSock == rpc_sockets[i]) - associated++; - } - if(!associated){ - // No TCP connections are associated, this is a candidate for removal - //fprintf(stderr, "removing RPC connection... \n"); - char c; - if(read(_phy.getDescriptor(rpc_sockets[i]),&c,1) < 0) { - closeClient(rpc_sockets[i]); + if(rpc_sockets.size() || tcp_connections.size()) { + /* Here we will periodically check the list of rpc_sockets for those that + do not currently have any data connection associated with them. If they are + unused, then we will try to read from them, if they fail, we can safely assume + that the client has closed their end and we can close ours */ + for(size_t i=0, associated = 0; i<rpc_sockets.size(); i++, associated = 0) { + for(size_t j=0; j<tcp_connections.size(); j++) { + if(tcp_connections[j]->rpcSock == rpc_sockets[i]) + associated++; } - else { - //fprintf(stderr, "this appears to be in use.\n"); + if(!associated){ + // No TCP connections are associated, this is a candidate for removal + char c; + if(read(_phy.getDescriptor(rpc_sockets[i]),&c,1) < 0) { + closeClient(rpc_sockets[i]); + } } } + fprintf(stderr, "tcp_conns = %d, rpc_socks = %d\n", tcp_connections.size(), rpc_sockets.size()); } } if (since_tcp >= ZT_LWIP_TCP_TIMER_INTERVAL) { @@ -393,7 +390,7 @@ void NetconEthernetTap::threadMain() _phy.poll((unsigned long)std::min(tcp_remaining,etharp_remaining)); } closeAll(); - // TODO: cleanup -- destroy LWIP state, kill any clients, unload .so, etc. + dlclose(lwipstack->_libref); } void NetconEthernetTap::phyOnUnixClose(PhySocket *sock,void **uptr) |