Implement #243, separate core/client (PCH support dropped for now)

core/tunnel/TransitTunnel.h

@@ -0,0 +1,111 @@
+#ifndef TRANSIT_TUNNEL_H__
+#define TRANSIT_TUNNEL_H__
+
+#include <inttypes.h>
+#include <vector>
+#include <mutex>
+#include <memory>
+#include "TunnelCrypto.h"
+#include "I2NPProtocol.h"
+#include "TunnelEndpoint.h"
+#include "TunnelGateway.h"
+#include "TunnelBase.h"
+
+namespace i2p
+{
+namespace tunnel
+{   
+    class TransitTunnel: public TunnelBase 
+    {
+        public:
+
+            TransitTunnel (uint32_t receiveTunnelID,
+                const uint8_t * nextIdent, uint32_t nextTunnelID, 
+                const uint8_t * layerKey,const uint8_t * ivKey); 
+            
+            virtual size_t GetNumTransmittedBytes () const { return 0; };
+            
+            uint32_t GetTunnelID () const { return m_TunnelID; };
+
+            // implements TunnelBase
+            void SendTunnelDataMsg (std::shared_ptr<i2p::I2NPMessage> msg);
+            void HandleTunnelDataMsg (std::shared_ptr<const i2p::I2NPMessage> tunnelMsg);
+            void EncryptTunnelMsg (std::shared_ptr<const I2NPMessage> in, std::shared_ptr<I2NPMessage> out); 
+            uint32_t GetNextTunnelID () const { return m_NextTunnelID; };
+            const i2p::data::IdentHash& GetNextIdentHash () const { return m_NextIdent; };
+            
+        private:
+
+            uint32_t m_TunnelID, m_NextTunnelID;
+            i2p::data::IdentHash m_NextIdent;
+            
+            i2p::crypto::TunnelEncryption m_Encryption;
+    };  
+
+    class TransitTunnelParticipant: public TransitTunnel
+    {
+        public:
+
+            TransitTunnelParticipant (uint32_t receiveTunnelID,
+                const uint8_t * nextIdent, uint32_t nextTunnelID, 
+                const uint8_t * layerKey,const uint8_t * ivKey):
+                TransitTunnel (receiveTunnelID, nextIdent, nextTunnelID, 
+                layerKey, ivKey), m_NumTransmittedBytes (0) {};
+            ~TransitTunnelParticipant ();
+
+            size_t GetNumTransmittedBytes () const { return m_NumTransmittedBytes; };
+            void HandleTunnelDataMsg (std::shared_ptr<const i2p::I2NPMessage> tunnelMsg);
+            void FlushTunnelDataMsgs ();
+
+        private:
+
+            size_t m_NumTransmittedBytes;
+            std::vector<std::shared_ptr<i2p::I2NPMessage> > m_TunnelDataMsgs;
+    };  
+    
+    class TransitTunnelGateway: public TransitTunnel
+    {
+        public:
+
+            TransitTunnelGateway (uint32_t receiveTunnelID,
+                const uint8_t * nextIdent, uint32_t nextTunnelID, 
+                const uint8_t * layerKey,const uint8_t * ivKey):
+                TransitTunnel (receiveTunnelID, nextIdent, nextTunnelID, 
+                layerKey, ivKey), m_Gateway(this) {};
+
+            void SendTunnelDataMsg (std::shared_ptr<i2p::I2NPMessage> msg);
+            void FlushTunnelDataMsgs ();
+            size_t GetNumTransmittedBytes () const { return m_Gateway.GetNumSentBytes (); };
+            
+        private:
+
+            std::mutex m_SendMutex;
+            TunnelGateway m_Gateway;
+    };  
+
+    class TransitTunnelEndpoint: public TransitTunnel
+    {
+        public:
+
+            TransitTunnelEndpoint (uint32_t receiveTunnelID,
+                const uint8_t * nextIdent, uint32_t nextTunnelID, 
+                const uint8_t * layerKey,const uint8_t * ivKey):
+                TransitTunnel (receiveTunnelID, nextIdent, nextTunnelID, layerKey, ivKey),
+                m_Endpoint (false) {}; // transit endpoint is always outbound
+
+            void HandleTunnelDataMsg (std::shared_ptr<const i2p::I2NPMessage> tunnelMsg);
+            size_t GetNumTransmittedBytes () const { return m_Endpoint.GetNumReceivedBytes (); }
+            
+        private:
+
+            TunnelEndpoint m_Endpoint;
+    };
+    
+    TransitTunnel * CreateTransitTunnel (uint32_t receiveTunnelID,
+        const uint8_t * nextIdent, uint32_t nextTunnelID, 
+        const uint8_t * layerKey,const uint8_t * ivKey, 
+        bool isGateway, bool isEndpoint);
+}
+}
+
+#endif