From 2200133ce3e481fcb7904521096d7db7a5bf89bf Mon Sep 17 00:00:00 2001
From: orignal <romakoshelkin@yandex.ru>
Date: Tue, 29 Jul 2014 14:56:36 -0400
Subject: [PATCH] moved LeaseSet creation away from StreamingDestination

---
 I2NPProtocol.cpp |  9 ++++-----
 I2NPProtocol.h   |  2 +-
 Streaming.cpp    | 39 ++++-----------------------------------
 3 files changed, 9 insertions(+), 41 deletions(-)

diff --git a/I2NPProtocol.cpp b/I2NPProtocol.cpp
index 8909e556..78817aed 100644
--- a/I2NPProtocol.cpp
+++ b/I2NPProtocol.cpp
@@ -209,16 +209,15 @@ namespace i2p
 		return m;
 	}	
 
-	I2NPMessage * CreateDatabaseStoreMsg (const i2p::data::LeaseSet * leaseSet)
+	I2NPMessage * CreateDatabaseStoreMsg (const i2p::data::LeaseSet& leaseSet)
 	{
-		if (!leaseSet) return nullptr;
 		I2NPMessage * m = NewI2NPMessage ();
 		I2NPDatabaseStoreMsg * msg = (I2NPDatabaseStoreMsg *)m->GetPayload ();
-		memcpy (msg->key, leaseSet->GetIdentHash (), 32);
+		memcpy (msg->key, leaseSet.GetIdentHash (), 32);
 		msg->type = 1; // LeaseSet
 		msg->replyToken = 0;
-		memcpy (m->GetPayload () + sizeof (I2NPDatabaseStoreMsg), leaseSet->GetBuffer (), leaseSet->GetBufferLen ());
-		m->len += leaseSet->GetBufferLen () + sizeof (I2NPDatabaseStoreMsg);
+		memcpy (m->GetPayload () + sizeof (I2NPDatabaseStoreMsg), leaseSet.GetBuffer (), leaseSet.GetBufferLen ());
+		m->len += leaseSet.GetBufferLen () + sizeof (I2NPDatabaseStoreMsg);
 		FillI2NPMessageHeader (m, eI2NPDatabaseStore);
 		return m;
 	}
diff --git a/I2NPProtocol.h b/I2NPProtocol.h
index 07bebc94..4d4c5a9e 100644
--- a/I2NPProtocol.h
+++ b/I2NPProtocol.h
@@ -158,7 +158,7 @@ namespace tunnel
 	I2NPMessage * CreateDatabaseSearchReply (const i2p::data::IdentHash& ident, const i2p::data::RouterInfo * floodfill);
 	
 	I2NPMessage * CreateDatabaseStoreMsg (const i2p::data::RouterInfo * router = nullptr);
-	I2NPMessage * CreateDatabaseStoreMsg (const i2p::data::LeaseSet * leaseSet);		
+	I2NPMessage * CreateDatabaseStoreMsg (const i2p::data::LeaseSet& leaseSet);		
 
 	I2NPBuildRequestRecordClearText CreateBuildRequestRecord (
 		const uint8_t * ourIdent, uint32_t receiveTunnelID, 
diff --git a/Streaming.cpp b/Streaming.cpp
index 7f239331..fc16c5b7 100644
--- a/Streaming.cpp
+++ b/Streaming.cpp
@@ -426,41 +426,10 @@ namespace stream
 		
 	I2NPMessage * StreamingDestination::CreateLeaseSet () const
 	{
-		I2NPMessage * m = NewI2NPMessage ();
-		I2NPDatabaseStoreMsg * msg = (I2NPDatabaseStoreMsg *)m->GetPayload ();
-		memcpy (msg->key, (const uint8_t *)m_IdentHash, 32);
-		msg->type = 1; // LeaseSet
-		msg->replyToken = 0;
-		
-		uint8_t * buf = m->GetPayload () + sizeof (I2NPDatabaseStoreMsg);
-		size_t size = 0;
-		memcpy (buf + size, &m_Keys.pub, sizeof (m_Keys.pub));
-		size += sizeof (m_Keys.pub); // destination
-		memcpy (buf + size, m_Pool->GetEncryptionPublicKey (), 256);
-		size += 256; // encryption key
-		memset (buf + size, 0, 128);
-		size += 128; // signing key
-		auto tunnels = m_Pool->GetInboundTunnels (5); // 5 tunnels maximum
-		buf[size] = tunnels.size (); // num leases
-		size++; // num	
-		for (auto it: tunnels)
-		{	
-			auto tunnel = it;	
-			memcpy (buf + size, (const uint8_t *)tunnel->GetNextIdentHash (), 32);
-			size += 32; // tunnel_gw
-			*(uint32_t *)(buf + size) = htobe32 (tunnel->GetNextTunnelID ());
-			size += 4; // tunnel_id
-			uint64_t ts = tunnel->GetCreationTime () + i2p::tunnel::TUNNEL_EXPIRATION_TIMEOUT - 60; // 1 minute before expiration
-			ts *= 1000; // in milliseconds
-			*(uint64_t *)(buf + size) = htobe64 (ts);
-			size += 8; // end_date
-		}	
-		Sign (buf, size, buf+ size);
-		size += 40; // signature
-		LogPrint ("Local LeaseSet of ", tunnels.size (), " leases created");
-		m->len += size + sizeof (I2NPDatabaseStoreMsg);
-		FillI2NPMessageHeader (m, eI2NPDatabaseStore);
-		return m;
+		// TODO: should store actual LeaseSet rather than msg
+		if (!m_Pool) return nullptr;
+		i2p::data::LeaseSet leaseSet(*m_Pool);	
+		return CreateDatabaseStoreMsg (leaseSet);
 	}	
 
 	void StreamingDestination::Sign (const uint8_t * buf, int len, uint8_t * signature) const