From 0d51f240d0e41bfda33afb968ce5641ce03c1266 Mon Sep 17 00:00:00 2001
From: orignal <romakoshelkin@yandex.ru>
Date: Thu, 15 May 2014 18:58:26 -0400
Subject: [PATCH] TunnelDecryption for tunnels

---
 I2NPProtocol.cpp |  9 ++++-----
 Tunnel.cpp       | 28 +++++++++++++---------------
 TunnelConfig.h   |  5 +----
 3 files changed, 18 insertions(+), 24 deletions(-)

diff --git a/I2NPProtocol.cpp b/I2NPProtocol.cpp
index 76f6e7d4..778d9197 100644
--- a/I2NPProtocol.cpp
+++ b/I2NPProtocol.cpp
@@ -1,8 +1,6 @@
 #include <string.h>
 #include "I2PEndian.h"
 #include <cryptopp/sha.h>
-#include <cryptopp/modes.h>
-#include <cryptopp/aes.h>
 #include <cryptopp/gzip.h>
 #include "ElGamal.h"
 #include "Timestamp.h"
@@ -259,11 +257,12 @@ namespace i2p
 				//TODO: fill filler
 				CryptoPP::SHA256().CalculateDigest(reply->hash, reply->padding, sizeof (reply->padding) + 1); // + 1 byte of ret
 				// encrypt reply
-				CryptoPP::CBC_Mode<CryptoPP::AES>::Encryption encryption;
+				i2p::crypto::CBCEncryption encryption;
 				for (int j = 0; j < num; j++)
 				{
-					encryption.SetKeyWithIV (clearText.replyKey, 32, clearText.replyIV);
-					encryption.ProcessData((uint8_t *)(records + j), (uint8_t *)(records + j), sizeof (records[j])); 
+					encryption.SetKey (clearText.replyKey);
+					encryption.SetIV (clearText.replyIV);
+					encryption.Encrypt((uint8_t *)(records + j), sizeof (records[j]), (uint8_t *)(records + j)); 
 				}
 				return true;
 			}	
diff --git a/Tunnel.cpp b/Tunnel.cpp
index 8529df4b..30bbee80 100644
--- a/Tunnel.cpp
+++ b/Tunnel.cpp
@@ -51,14 +51,17 @@ namespace tunnel
 			i++;
 			hop = hop->next;
 		}	
-		
+
+		i2p::crypto::CBCDecryption decryption;
 		hop = m_Config->GetLastHop ()->prev;
 		size_t ind = numRecords - 1;
 		while (hop)
 		{
-			for (size_t i = ind; i < numRecords; i++)
-				hop->decryption.Decrypt((uint8_t *)&records[i], 
-					sizeof (I2NPBuildRequestRecordElGamalEncrypted), (uint8_t *)&records[i]);
+			decryption.SetKey (hop->replyKey);
+			decryption.SetIV (hop->replyIV);
+			for (size_t i = ind; i < numRecords; i++)	
+				decryption.Decrypt((uint8_t *)&records[i], 
+					sizeof (I2NPBuildRequestRecordElGamalEncrypted), (uint8_t *)&records[i]);	
 			hop = hop->prev;
 			ind--;
 		}	
@@ -74,14 +77,17 @@ namespace tunnel
 	{
 		LogPrint ("TunnelBuildResponse ", (int)msg[0], " records.");
 
+		i2p::crypto::CBCDecryption decryption;
 		TunnelHopConfig * hop = m_Config->GetLastHop (); 
 		int num = msg[0];
 		while (hop)
 		{	
+			decryption.SetKey (hop->replyKey);
+			decryption.SetIV (hop->replyIV);
 			for (int i = 0; i < num; i++)
 			{			
 				uint8_t * record = msg + 1 + i*sizeof (I2NPBuildResponseRecord);
-				hop->decryption.Decrypt(record, sizeof (I2NPBuildResponseRecord), record);
+				decryption.Decrypt(record, sizeof (I2NPBuildResponseRecord), record);
 			}
 			hop = hop->prev;
 			num--;
@@ -102,8 +108,7 @@ namespace tunnel
 			TunnelHopConfig * hop = m_Config->GetFirstHop ();
 			while (hop)
 			{
-				hop->decryption.SetKey (hop->layerKey);
-				hop->ivDecryption.SetKey (hop->ivKey);
+				hop->decryption.SetKeys (hop->layerKey, hop->ivKey);
 				hop = hop->next;
 			}	
 		}	
@@ -116,14 +121,7 @@ namespace tunnel
 		TunnelHopConfig * hop = m_Config->GetLastHop (); 
 		while (hop)
 		{	
-			// iv
-			hop->ivDecryption.Decrypt ((i2p::crypto::ChipherBlock *)payload, (i2p::crypto::ChipherBlock *)payload);
-			// data
-			hop->decryption.SetIV (payload);
-			hop->decryption.Decrypt (payload + 16, TUNNEL_DATA_ENCRYPTED_SIZE, payload+16);
-			// double iv ecncryption
-			hop->ivDecryption.Decrypt ((i2p::crypto::ChipherBlock *)payload, (i2p::crypto::ChipherBlock *)payload);
-
+			hop->decryption.Decrypt (payload);
 			hop = hop->prev;
 		}
 	}	
diff --git a/TunnelConfig.h b/TunnelConfig.h
index 3db6c30e..5008d4ea 100644
--- a/TunnelConfig.h
+++ b/TunnelConfig.h
@@ -23,8 +23,7 @@ namespace tunnel
 		bool isGateway, isEndpoint;	
 		
 		TunnelHopConfig * next, * prev;
-		i2p::crypto::CBCDecryption decryption;	
-		i2p::crypto::ECBDecryption ivDecryption;
+		i2p::crypto::TunnelDecryption decryption;	
 		
 		TunnelHopConfig (const i2p::data::RouterInfo * r)
 		{
@@ -41,8 +40,6 @@ namespace tunnel
 
 			next = 0;
 			prev = 0;
-			decryption.SetKey (replyKey);
-			decryption.SetIV (replyIV);
 		}	
 
 		void SetNextRouter (const i2p::data::RouterInfo * r)