From d04f9e723c3c2d94b529376a354cbf37fd3867da Mon Sep 17 00:00:00 2001
From: orignal <romakoshelkin@yandex.ru>
Date: Thu, 6 Feb 2014 10:16:50 -0500
Subject: [PATCH] handle recieved I2NP message

---
 SSU.cpp | 27 ++++++++++++++++++++++++++-
 SSU.h   |  1 +
 2 files changed, 27 insertions(+), 1 deletion(-)

diff --git a/SSU.cpp b/SSU.cpp
index 314a62da..4fbd2696 100644
--- a/SSU.cpp
+++ b/SSU.cpp
@@ -382,7 +382,32 @@ namespace ssu
 			uint16_t fragmentSize = fragmentInfo & 0x1FFF; // bits 0 - 13
 			bool isLast = fragmentInfo & 0x010000; // bit 16	
 			uint8_t fragmentNum = fragmentInfo >> 17; // bits 23 - 17
-			LogPrint ("SSU data fragment ", (int)fragmentNum, " of message ", msgID, " size=", (int)fragmentSize, isLast ? " last" : " non-last"); 
+			LogPrint ("SSU data fragment ", (int)fragmentNum, " of message ", msgID, " size=", (int)fragmentSize, isLast ? " last" : " non-last"); 			
+			I2NPMessage * msg = nullptr;
+			if (fragmentNum > 0) // follow-up fragment
+			{
+				auto it = m_IncomleteMessages.find (msgID);
+				if (it != m_IncomleteMessages.end ())
+					msg = it->second;
+				else
+					// TODO:
+					LogPrint ("Unexpected follow-on fragment ", fragmentNum, " of message ", msgID);	
+			}
+			else // first fragment
+				msg = NewI2NPMessage ();
+			if (msg)
+			{
+				memcpy (msg->buf + msg->len, buf, fragmentSize);
+				msg->len += fragmentSize;	
+				if (!fragmentNum && !isLast)
+					m_IncomleteMessages[msgID] = msg;
+				if (isLast)
+				{
+					if (fragmentNum > 0)	
+						m_IncomleteMessages.erase (msgID);
+					i2p::HandleI2NPMessage (msg, false);	
+				}
+			}
 			buf += fragmentSize;
 		}	
 	}
diff --git a/SSU.h b/SSU.h
index aee160ae..a27743a0 100644
--- a/SSU.h
+++ b/SSU.h
@@ -88,6 +88,7 @@ namespace ssu
 			CryptoPP::CBC_Mode<CryptoPP::AES>::Encryption m_Encryption;	
 			CryptoPP::CBC_Mode<CryptoPP::AES>::Decryption m_Decryption;	
 			uint8_t m_SessionKey[32], m_MacKey[32];
+			std::map<uint32_t, I2NPMessage *> m_IncomleteMessages;
 	};
 
 	class SSUServer