Merge pull request #24 from F4FXL/UDPHolePunching

Allow to compile without NAT traversal, disabled by default

Common/G2ProtocolHandler.cpp

@@ -144,6 +144,20 @@ CAMBEData* CG2ProtocolHandler::readAMBE()
 	return data;
 }
 
+#if defined(ENABLE_NAT_TRAVERSAL)
+void CG2ProtocolHandler::traverseNat(const wxString& address)
+{
+	unsigned char buffer[1];
+	::memset(buffer, 0, 1);
+	
+	in_addr addr = CUDPReaderWriter::lookup(address);
+
+	//wxLogError(wxT("Punching hole to %s"), address.mb_str());
+
+	m_socket.write(buffer, 1, addr, G2_DV_PORT);
+}
+#endif
+
 void CG2ProtocolHandler::close()
 {
 	m_socket.close();

Common/G2ProtocolHandler.h

@@ -52,6 +52,10 @@ public:
 	CHeaderData* readHeader();
 	CAMBEData*   readAMBE();
 
+#if defined(ENABLE_NAT_TRAVERSAL)
+	void traverseNat(const wxString& addr);
+#endif
+
 	void close();
 
 private:

Common/Makefile

@@ -4,7 +4,7 @@ OBJECTS = AMBEData.o AnnouncementUnit.o APRSCollector.o APRSWriter.o APRSWriterT
 	  DPlusAuthenticator.o DPlusHandler.o DPlusProtocolHandler.o DPlusProtocolHandlerPool.o DRATSServer.o DTMF.o \
 	  DummyRepeaterProtocolHandler.o DVTOOLFileReader.o EchoUnit.o G2Handler.o G2ProtocolHandler.o GatewayCache.o \
 	  HBRepeaterProtocolHandler.o HeaderData.o HeaderLogger.o HeardData.o HostFile.o IcomRepeaterProtocolHandler.o IRCDDBGatewayConfig.o \
-	  LogEvent.o Logger.o PollData.o RemoteHandler.o RemoteLinkData.o RemoteProtocolHandler.o RemoteRepeaterData.o RemoteStarNetGroup.o \
+	  LogEvent.o Logger.o NatTraversalHandler.o PollData.o RemoteHandler.o RemoteLinkData.o RemoteProtocolHandler.o RemoteRepeaterData.o RemoteStarNetGroup.o \
 	  RemoteStarNetUser.o RepeaterCache.o RepeaterHandler.o SHA256.o SlowDataEncoder.o StarNetHandler.o StatusData.o \
 	  TCPReaderWriterClient.o TCPReaderWriterServer.o TextCollector.o TextData.o Timer.o UDPReaderWriter.o UserCache.o Utils.o \
 	  VersionUnit.o XLXHostsFileDownloader.o

Common/NatTraversalHandler.cpp

@@ -0,0 +1,61 @@
+/*
+ *   Copyright (C) 2010,2011,2012,2013,2014,2015,2016,2017,2018 by Jonathan Naylor G4KLX
+ *
+ *   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 2 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, write to the Free Software
+ *   Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
+ */
+
+#if defined(ENABLE_NAT_TRAVERSAL)
+
+#include "NatTraversalHandler.h"
+
+const unsigned int CACHE_SIZE = 500U;
+
+CNatTraversalHandler::CNatTraversalHandler() :
+m_g2cache(CACHE_SIZE),
+m_g2Handler(NULL)
+{
+
+}
+
+CNatTraversalHandler::~CNatTraversalHandler()
+{
+    for (CNatTraversalCache_t::iterator it = m_g2cache.begin(); it != m_g2cache.end(); ++it)
+		delete it->second;
+}
+
+void CNatTraversalHandler::setG2Handler(CG2ProtocolHandler * handler)
+{
+    m_g2Handler = handler;
+}
+
+void CNatTraversalHandler::traverseNatG2(const wxString& address)
+{
+    if(m_g2Handler != NULL){
+        CNatTraversalRecord* record = m_g2cache[address];
+        
+        if(record == NULL) {
+            record = new CNatTraversalRecord(address);
+            m_g2cache[address] = record;
+        }
+
+        std::time_t currentTime = std::time(NULL);
+        if(currentTime - record->getTimestamp() > G2_TRAVERSAL_TIMEOUT) {
+            record->setTimestamp(currentTime);
+            m_g2Handler->traverseNat(address);
+        }
+    }
+}
+
+#endif

Common/NatTraversalHandler.h

@@ -0,0 +1,80 @@
+/*
+ *   Copyright (C) 2010,2011,2012,2013,2014,2015,2016,2017,2018 by Jonathan Naylor G4KLX
+ *
+ *   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 2 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, write to the Free Software
+ *   Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
+ */
+
+#if defined(ENABLE_NAT_TRAVERSAL)
+#ifndef NatTraversalHandler_H
+#define NatTraversalHandler_H
+
+#define G2_TRAVERSAL_TIMEOUT 29 //seconds
+
+#include "G2ProtocolHandler.h"
+
+#include <wx/wx.h>
+#include <ctime>
+
+enum NAT_TRAVERSAL_TYPE {
+    NTT_G2,
+    //NTT_DEXTRA
+    //NTT_DCS
+    //NTT_DPLUS
+};
+
+class CNatTraversalRecord {
+public:
+    CNatTraversalRecord(const wxString& address) :
+    m_address(address),
+    m_timestamp(0)
+    {
+    }
+
+    std::time_t getTimestamp() const
+    {
+        return m_timestamp;
+    }
+
+    void setTimestamp(std::time_t timestamp)
+    {
+        m_timestamp = timestamp;
+    }
+
+private:
+    wxString m_address;
+    std::time_t m_timestamp;
+};
+
+WX_DECLARE_STRING_HASH_MAP(CNatTraversalRecord*, CNatTraversalCache_t);
+
+/*
+* This keeps track of when we UDP punched to one destination so to avoid unnecessary traffic on each ircddb reporting
+*/
+class CNatTraversalHandler {
+public:
+    CNatTraversalHandler();
+    ~CNatTraversalHandler();
+
+    void setG2Handler(CG2ProtocolHandler* handler);
+    void traverseNatG2(const wxString& address);
+
+private:
+    CNatTraversalCache_t m_g2cache;    
+    CG2ProtocolHandler* m_g2Handler;
+};
+
+#endif
+
+#endif

ircDDBGateway/IRCDDBGatewayThread.cpp

@@ -72,6 +72,9 @@ m_dextraPool(NULL),
 m_dplusPool(NULL),
 m_dcsPool(NULL),
 m_g2Handler(NULL),
+#if defined(ENABLE_NAT_TRAVERSAL)
+m_natTraversal(NULL),
+#endif
 m_aprsWriter(NULL),
 m_irc(NULL),
 m_cache(),
@@ -217,6 +220,13 @@ void CIRCDDBGatewayThread::run()
 		m_g2Handler = NULL;
 	}
 
+#if defined(ENABLE_NAT_TRAVERSAL)
+	if(m_g2Handler != NULL) {
+		m_natTraversal = new CNatTraversalHandler();
+		m_natTraversal->setG2Handler(m_g2Handler);
+	}
+#endif
+
 	// Wait here until we have the essentials to run
 	while (!m_killed && (m_dextraPool == NULL || m_dplusPool == NULL || m_dcsPool == NULL || m_g2Handler == NULL || (m_icomRepeaterHandler == NULL && m_hbRepeaterHandler == NULL && m_dummyRepeaterHandler == NULL) || m_gatewayCallsign.IsEmpty()))
 		::wxMilliSleep(500UL);		// 1/2 sec
@@ -719,6 +729,9 @@ void CIRCDDBGatewayThread::processIrcDDB()
 					if (!address.IsEmpty()) {
 						wxLogMessage(wxT("USER: %s %s %s %s"), user.c_str(), repeater.c_str(), gateway.c_str(), address.c_str());
 						m_cache.updateUser(user, repeater, gateway, address, timestamp, DP_DEXTRA, false, false);
+#if defined(ENABLE_NAT_TRAVERSAL)
+						m_natTraversal->traverseNatG2(address);
+#endif
 					} else {
 						wxLogMessage(wxT("USER: %s NOT FOUND"), user.c_str());
 					}
@@ -735,6 +748,9 @@ void CIRCDDBGatewayThread::processIrcDDB()
 					if (!address.IsEmpty()) {
 						wxLogMessage(wxT("REPEATER: %s %s %s"), repeater.c_str(), gateway.c_str(), address.c_str());
 						m_cache.updateRepeater(repeater, gateway, address, DP_DEXTRA, false, false);
+#if defined(ENABLE_NAT_TRAVERSAL)
+						m_natTraversal->traverseNatG2(address);
+#endif
 					} else {
 						wxLogMessage(wxT("REPEATER: %s NOT FOUND"), repeater.c_str());
 					}
@@ -752,6 +768,9 @@ void CIRCDDBGatewayThread::processIrcDDB()
 					if (!address.IsEmpty()) {
 						wxLogMessage(wxT("GATEWAY: %s %s"), gateway.c_str(), address.c_str());
 						m_cache.updateGateway(gateway, address, DP_DEXTRA, false, false);
+#if defined(ENABLE_NAT_TRAVERSAL)						
+						m_natTraversal->traverseNatG2(address);
+#endif
 					} else {
 						wxLogMessage(wxT("GATEWAY: %s NOT FOUND"), gateway.c_str());
 					}

ircDDBGateway/IRCDDBGatewayThread.h

@@ -35,6 +35,9 @@
 #include "IRCDDB.h"
 #include "Timer.h"
 #include "Defs.h"
+#if defined(ENABLE_NAT_TRAVERSAL)
+#include "NatTraversalHandler.h"
+#endif
 
 #include <wx/wx.h>
 
@@ -92,6 +95,9 @@ private:
 	CDPlusProtocolHandlerPool*     m_dplusPool;
 	CDCSProtocolHandlerPool*       m_dcsPool;
 	CG2ProtocolHandler*       m_g2Handler;
+#if defined(ENABLE_NAT_TRAVERSAL)
+	CNatTraversalHandler*     m_natTraversal;
+#endif
 	CAPRSWriter*              m_aprsWriter;
 	CIRCDDB*                  m_irc;
 	CCacheManager             m_cache;