From 9f1e496fa414a24666d1e33649a9c5a25f05195c Mon Sep 17 00:00:00 2001
From: orignal <i2porignal@yandex.ru>
Date: Tue, 7 Oct 2014 14:06:50 -0400
Subject: [PATCH] create data message inside detination

---
 Destination.cpp | 42 +++++++++++++++++++++---------------------
 Destination.h   |  9 ++++-----
 Streaming.cpp   |  2 +-
 3 files changed, 26 insertions(+), 27 deletions(-)

diff --git a/Destination.cpp b/Destination.cpp
index c2a8913f..732bfd2b 100644
--- a/Destination.cpp
+++ b/Destination.cpp
@@ -197,6 +197,27 @@ namespace stream
 			LogPrint ("Data: unexpected protocol ", buf[9]);
 	}	
 	
+	I2NPMessage * StreamingDestination::CreateDataMessage (const uint8_t * payload, size_t len)
+	{
+		I2NPMessage * msg = NewI2NPShortMessage ();
+		CryptoPP::Gzip compressor; // DEFAULT_DEFLATE_LEVEL
+		if (len <= COMPRESSION_THRESHOLD_SIZE)
+			compressor.SetDeflateLevel (CryptoPP::Gzip::MIN_DEFLATE_LEVEL);
+		compressor.Put (payload, len);
+		compressor.MessageEnd();
+		int size = compressor.MaxRetrievable ();
+		uint8_t * buf = msg->GetPayload ();
+		*(uint32_t *)buf = htobe32 (size); // length
+		buf += 4;
+		compressor.Get (buf, size);
+		memset (buf + 4, 0, 4); // source and destination ports. TODO: fill with proper values later
+		buf[9] = 6; // streaming protocol
+		msg->len += size + 4; 
+		FillI2NPMessageHeader (msg, eI2NPData);
+		
+		return msg;
+	}		
+
 	void StreamingDestination::SetLeaseSetUpdated ()
 	{
 		UpdateLeaseSet ();
@@ -385,26 +406,5 @@ namespace stream
 	{
 		return destinations;
 	}	
-
-	I2NPMessage * CreateDataMessage (Stream * s, const uint8_t * payload, size_t len)
-	{
-		I2NPMessage * msg = NewI2NPShortMessage ();
-		CryptoPP::Gzip compressor; // DEFAULT_DEFLATE_LEVEL
-		if (len <= COMPRESSION_THRESHOLD_SIZE)
-			compressor.SetDeflateLevel (CryptoPP::Gzip::MIN_DEFLATE_LEVEL);
-		compressor.Put (payload, len);
-		compressor.MessageEnd();
-		int size = compressor.MaxRetrievable ();
-		uint8_t * buf = msg->GetPayload ();
-		*(uint32_t *)buf = htobe32 (size); // length
-		buf += 4;
-		compressor.Get (buf, size);
-		memset (buf + 4, 0, 4); // source and destination ports. TODO: fill with proper values later
-		buf[9] = 6; // streaming protocol
-		msg->len += size + 4; 
-		FillI2NPMessageHeader (msg, eI2NPData);
-		
-		return msg;
-	}		
 }		
 }
diff --git a/Destination.h b/Destination.h
index 3c69a40c..c9f36d5d 100644
--- a/Destination.h
+++ b/Destination.h
@@ -33,14 +33,16 @@ namespace stream
 			bool IsAcceptorSet () const { return m_Acceptor != nullptr; };	
 			void HandleNextPacket (Packet * packet);
 			void SendTunnelDataMsgs (const std::vector<i2p::tunnel::TunnelMessageBlock>& msgs);
+			void ResetCurrentOutboundTunnel () { m_CurrentOutboundTunnel = nullptr; };
+			// I2CP
+			void HandleDataMessage (const uint8_t * buf, size_t len);
+			I2NPMessage * CreateDataMessage (const uint8_t * payload, size_t len);
 
 			// implements LocalDestination
 			const i2p::data::PrivateKeys& GetPrivateKeys () const { return m_Keys; };
 			const uint8_t * GetEncryptionPrivateKey () const { return m_EncryptionPrivateKey; };
 			const uint8_t * GetEncryptionPublicKey () const { return m_EncryptionPublicKey; };
 			void SetLeaseSetUpdated ();
-			void HandleDataMessage (const uint8_t * buf, size_t len);
-			void ResetCurrentOutboundTunnel () { m_CurrentOutboundTunnel = nullptr; };
 
 		private:		
 	
@@ -117,9 +119,6 @@ namespace stream
 	StreamingDestination * LoadLocalDestination (const std::string& filename, bool isPublic);
 	// for HTTP
 	const StreamingDestinations& GetLocalDestinations ();	
-	
-	// assuming data is I2CP message
-	I2NPMessage * CreateDataMessage (Stream * s, const uint8_t * payload, size_t len);
 }		
 }	
 
diff --git a/Streaming.cpp b/Streaming.cpp
index 0c43220e..6e3bf260 100644
--- a/Streaming.cpp
+++ b/Streaming.cpp
@@ -426,7 +426,7 @@ namespace stream
 			for (auto it: packets)
 			{ 
 				auto msg = m_RoutingSession->WrapSingleMessage ( 
-					CreateDataMessage (this, it->GetBuffer (), it->GetLength ()), 
+					m_LocalDestination.CreateDataMessage (it->GetBuffer (), it->GetLength ()), 
 				    leaseSet);
 				msgs.push_back (i2p::tunnel::TunnelMessageBlock 
 							{