fixed small issues

SSU.cpp

@@ -138,23 +138,42 @@ namespace ssu
 
 	void SSUSession::ProcessIntroKeyMessage (uint8_t * buf, size_t len, const boost::asio::ip::udp::endpoint& senderEndpoint)
 	{
-		if (ProcessIntroKeyEncryptedMessage (buf, len))
+		auto introKey = GetIntroKey ();
+		if (!introKey)
 		{
-			SSUHeader * header = (SSUHeader *)buf;
+			LogPrint ("SSU is not supported");
-			switch (header->GetPayloadType ())
+			return;
+		}	
+		// use intro key for verification and decryption
+		if (!Validate (buf, len, introKey))
+		{
+			LogPrint ("MAC verification intro key failed");	
+			Failed ();
+			return;
+		}	
+		
+		Decrypt (buf, len, introKey);
+		CreateAESandMacKey (buf + sizeof (SSUHeader), m_SessionKey, m_MacKey);
+		SSUHeader * header = (SSUHeader *)buf;
+		switch (header->GetPayloadType ())
+		{
+			case PAYLOAD_TYPE_SESSION_REQUEST:
+				ProcessSessionRequest (buf, len, senderEndpoint);				
+			break;
+			case PAYLOAD_TYPE_SESSION_CREATED:
+				ProcessSessionCreated (buf, len);
+			break;
+			case PAYLOAD_TYPE_SESSION_DESTROYED:
 			{
-				case PAYLOAD_TYPE_SESSION_REQUEST:
+				LogPrint ("SSU session destroy with into key received");
-					ProcessSessionRequest (buf, len, senderEndpoint);				
+				m_Server.DeleteSession (this); // delete this 
-				break;
-				case PAYLOAD_TYPE_SESSION_CREATED:
-					ProcessSessionCreated (buf, len);
 				break;
-				case PAYLOAD_TYPE_PEER_TEST:
+			}		
-					// TODO
+			case PAYLOAD_TYPE_PEER_TEST:
-				break;	
+				// TODO
-				default: ;
+			break;	
-			}
+			default: ;
-		}		
+		}	
 	}
 
 	void SSUSession::ProcessSessionRequest (uint8_t * buf, size_t len, const boost::asio::ip::udp::endpoint& senderEndpoint)
@@ -414,29 +433,6 @@ namespace ssu
 		}
 	}
 
-	bool SSUSession::ProcessIntroKeyEncryptedMessage (uint8_t * buf, size_t len)
-	{
-		auto introKey = GetIntroKey ();
-		if (introKey)
-		{
-			// use intro key for verification and decryption
-			if (Validate (buf, len, introKey))
-			{
-				Decrypt (buf, len, introKey);
-				CreateAESandMacKey (buf + sizeof (SSUHeader), m_SessionKey, m_MacKey);	
-				return true;					
-			}
-			else
-			{	
-				LogPrint ("MAC verification failed");	
-				Failed ();
-			}	
-		}
-		else
-			LogPrint ("SSU is not supported");
-		return false;
-	}	
-
 	void SSUSession::FillHeaderAndEncrypt (uint8_t payloadType, uint8_t * buf, size_t len, 
 		const uint8_t * aesKey, const uint8_t * iv, const uint8_t * macKey)
 	{	
@@ -734,12 +730,13 @@ namespace ssu
 
 	void SSUSession::SendPeerTest ()
 	{
-		auto introKey = GetIntroKey ();
+		auto address = i2p::context.GetRouterInfo ().GetSSUAddress ();
-		if (!introKey)
+		if (!address)
 		{
 			LogPrint ("SSU is not supported. Can't send peer test");
 			return;
 		}
+		auto introKey = address->key;
 		uint8_t buf[80 + 18];
 		uint8_t * payload = buf + sizeof (SSUHeader);
 		CryptoPP::RandomNumberGenerator& rnd = i2p::context.GetRandomNumberGenerator ();

SSU.h

@@ -110,7 +110,6 @@ namespace ssu
 			void Send (i2p::I2NPMessage * msg);
 			void Send (uint8_t type, const uint8_t * payload, size_t len); // with session key
 			
-			bool ProcessIntroKeyEncryptedMessage (uint8_t * buf, size_t len);
 			void FillHeaderAndEncrypt (uint8_t payloadType, uint8_t * buf, size_t len, const uint8_t * aesKey, const uint8_t * iv, const uint8_t * macKey);
 			void Decrypt (uint8_t * buf, size_t len, const uint8_t * aesKey);			
 			bool Validate (uint8_t * buf, size_t len, const uint8_t * macKey);