From e55e15693d6e1676ad10f035bab4c959637cf038 Mon Sep 17 00:00:00 2001
From: orignal <i2porignal@yandex.ru>
Date: Wed, 2 Jan 2019 15:40:48 -0500
Subject: [PATCH] update LS2 leases

---
 libi2pd/LeaseSet.cpp | 52 ++++++++++++++++++++++++++------------------
 libi2pd/LeaseSet.h   |  2 ++
 2 files changed, 33 insertions(+), 21 deletions(-)

diff --git a/libi2pd/LeaseSet.cpp b/libi2pd/LeaseSet.cpp
index d6a92174..e5d7ca5b 100644
--- a/libi2pd/LeaseSet.cpp
+++ b/libi2pd/LeaseSet.cpp
@@ -89,26 +89,7 @@ namespace data
 			leases += 4; // tunnel ID
 			lease.endDate = bufbe64toh (leases);
 			leases += 8; // end date
-			if (ts < lease.endDate + LEASE_ENDDATE_THRESHOLD)
-			{
-				if (lease.endDate > m_ExpirationTime)
-					m_ExpirationTime = lease.endDate;
-				if (m_StoreLeases)
-				{
-					auto ret = m_Leases.insert (std::make_shared<Lease>(lease));
-					if (!ret.second) (*ret.first)->endDate = lease.endDate; // update existing
-					(*ret.first)->isUpdated = true;
-					// check if lease's gateway is in our netDb
-					if (!netdb.FindRouter (lease.tunnelGateway))
-					{
-						// if not found request it
-						LogPrint (eLogInfo, "LeaseSet: Lease's tunnel gateway not found, requesting");
-						netdb.RequestDestination (lease.tunnelGateway);
-					}
-				}
-			}
-			else
-				LogPrint (eLogWarning, "LeaseSet: Lease is expired already ");
+			UpdateLease (lease, ts);
 		}
 		if (!m_ExpirationTime)
 		{
@@ -140,6 +121,30 @@ namespace data
 		}
 	}
 
+	void LeaseSet::UpdateLease (const Lease& lease, uint64_t ts)
+	{
+		if (ts < lease.endDate + LEASE_ENDDATE_THRESHOLD)
+		{
+			if (lease.endDate > m_ExpirationTime)
+				m_ExpirationTime = lease.endDate;
+			if (m_StoreLeases)
+			{
+				auto ret = m_Leases.insert (std::make_shared<Lease>(lease));
+				if (!ret.second) (*ret.first)->endDate = lease.endDate; // update existing
+				(*ret.first)->isUpdated = true;
+				// check if lease's gateway is in our netDb
+				if (!netdb.FindRouter (lease.tunnelGateway))
+				{
+					// if not found request it
+					LogPrint (eLogInfo, "LeaseSet: Lease's tunnel gateway not found, requesting");
+					netdb.RequestDestination (lease.tunnelGateway);
+				}
+			}
+		}
+		else
+			LogPrint (eLogWarning, "LeaseSet: Lease is expired already ");
+	}
+
 	uint64_t LeaseSet::ExtractTimestamp (const uint8_t * buf, size_t len) const
 	{
 		if (!m_Identity) return 0;
@@ -311,10 +316,15 @@ namespace data
 		// leases
 		if (offset + 1 >= len) return 0;	
 		int numLeases = buf[offset]; offset++;
+		auto ts = i2p::util::GetMillisecondsSinceEpoch ();
 		for (int i = 0; i < numLeases; i++)
 		{
 			if (offset + 40 > len) return 0;
-			offset += 40; // lease
+			Lease lease;
+			lease.tunnelGateway = buf + offset; offset += 32; // gateway
+			lease.tunnelID = bufbe32toh (buf + offset); offset += 4; // tunnel ID
+			lease.endDate = bufbe32toh (buf + offset)*1000LL; offset += 4; // end date
+			UpdateLease (lease, ts);
 		}
 		return offset;
 	}
diff --git a/libi2pd/LeaseSet.h b/libi2pd/LeaseSet.h
index 606672f2..89ef011d 100644
--- a/libi2pd/LeaseSet.h
+++ b/libi2pd/LeaseSet.h
@@ -84,6 +84,8 @@ namespace data
 
 		protected:
 
+			void UpdateLease (const Lease& lease, uint64_t ts);
+
 			// called from LeaseSet2
 			LeaseSet ();
 			void SetBuffer (const uint8_t * buf, size_t len);