From d47bf1bada8621e0ac084bef8639a609f99a83b3 Mon Sep 17 00:00:00 2001
From: orignal <i2porignal@yandex.ru>
Date: Fri, 9 Jul 2021 19:26:14 -0400
Subject: [PATCH] different tunnel build record size

---
 libi2pd/Tunnel.cpp     |  7 ++++---
 libi2pd/TunnelConfig.h | 14 +++++++++++---
 2 files changed, 15 insertions(+), 6 deletions(-)

diff --git a/libi2pd/Tunnel.cpp b/libi2pd/Tunnel.cpp
index 1eb30db2..34105962 100644
--- a/libi2pd/Tunnel.cpp
+++ b/libi2pd/Tunnel.cpp
@@ -42,10 +42,11 @@ namespace tunnel
 	void Tunnel::Build (uint32_t replyMsgID, std::shared_ptr<OutboundTunnel> outboundTunnel)
 	{
 		auto numHops = m_Config->GetNumHops ();
-		int numRecords = numHops <= STANDARD_NUM_RECORDS ? STANDARD_NUM_RECORDS : MAX_NUM_RECORDS;
+		const int numRecords = numHops <= STANDARD_NUM_RECORDS ? STANDARD_NUM_RECORDS : MAX_NUM_RECORDS;
 		auto msg = numRecords <= STANDARD_NUM_RECORDS ? NewI2NPShortMessage () : NewI2NPMessage ();
 		*msg->GetPayload () = numRecords;
-		msg->len += numRecords*TUNNEL_BUILD_RECORD_SIZE + 1;
+		const size_t recordSize = m_Config->IsShort () ? SHORT_TUNNEL_BUILD_RECORD_SIZE : TUNNEL_BUILD_RECORD_SIZE; 
+		msg->len += numRecords*recordSize + 1;
 		// shuffle records
 		std::vector<int> recordIndicies;
 		for (int i = 0; i < numRecords; i++) recordIndicies.push_back(i);
@@ -70,7 +71,7 @@ namespace tunnel
 		for (int i = numHops; i < numRecords; i++)
 		{
 			int idx = recordIndicies[i];
-			RAND_bytes (records + idx*TUNNEL_BUILD_RECORD_SIZE, TUNNEL_BUILD_RECORD_SIZE);
+			RAND_bytes (records + idx*recordSize, recordSize);
 		}
 
 		// decrypt real records
diff --git a/libi2pd/TunnelConfig.h b/libi2pd/TunnelConfig.h
index 591fed52..b33f7668 100644
--- a/libi2pd/TunnelConfig.h
+++ b/libi2pd/TunnelConfig.h
@@ -115,6 +115,8 @@ namespace tunnel
 				}
 			}
 
+			bool IsShort () const { return m_IsShort; }
+			
 			TunnelHopConfig * GetFirstHop () const
 			{
 				return m_FirstHop;
@@ -193,10 +195,15 @@ namespace tunnel
 				for (const auto& it: peers)
 				{
 					TunnelHopConfig * hop;
-					if (it->GetCryptoKeyType () == i2p::data::CRYPTO_KEY_TYPE_ECIES_X25519_AEAD)
-						hop = new LongECIESTunnelHopConfig (it);
+					if (m_IsShort)
+						hop = new ShortECIESTunnelHopConfig (it);
 					else
-						hop = new ElGamalTunnelHopConfig (it);
+					{	
+						if (it->GetCryptoKeyType () == i2p::data::CRYPTO_KEY_TYPE_ECIES_X25519_AEAD)
+							hop = new LongECIESTunnelHopConfig (it);
+						else
+							hop = new ElGamalTunnelHopConfig (it);
+					}	
 					if (prev)
 						prev->SetNext (hop);
 					else
@@ -209,6 +216,7 @@ namespace tunnel
 		private:
 
 			TunnelHopConfig * m_FirstHop, * m_LastHop;
+			bool m_IsShort = false;
 	};
 
 	class ZeroHopsTunnelConfig: public TunnelConfig