don't copy RouterInfos and LeaseSets

LeaseSet.cpp

@@ -12,6 +12,17 @@ namespace data
 {
 	
 	LeaseSet::LeaseSet (const uint8_t * buf, int len)
+	{
+		ReadFromBuffer (buf, len);
+	}
+
+	void LeaseSet::Update (const uint8_t * buf, int len)
+	{	
+		m_Leases.clear ();
+		ReadFromBuffer (buf, len);
+	}
+	
+	void LeaseSet::ReadFromBuffer (const uint8_t * buf, int len)	
 	{	
 #pragma pack(1)
 		struct H

LeaseSet.h

@@ -37,6 +37,7 @@ namespace data
 			LeaseSet (const uint8_t * buf, int len);
 			LeaseSet (const LeaseSet& ) = default;
 			LeaseSet& operator=(const LeaseSet& ) = default;
+			void Update (const uint8_t * buf, int len);
 			
 			// implements RoutingDestination
 			const Identity& GetIdentity () const { return m_Identity; };
@@ -48,6 +49,10 @@ namespace data
 			const uint8_t * GetEncryptionPublicKey () const { return m_EncryptionKey; };
 			bool IsDestination () const { return true; };
 
+		private:
+
+			void ReadFromBuffer (const uint8_t * buf, int len);
+			
 		private:
 
 			std::vector<Lease> m_Leases;

NetDb.cpp

@@ -155,44 +155,40 @@ namespace data
 		}	
 	}	
 	
-	void NetDb::AddRouterInfo (uint8_t * buf, int len)
+	void NetDb::AddRouterInfo (const IdentHash& ident, uint8_t * buf, int len)
 	{	
-		RouterInfo * r = new RouterInfo (buf, len);
+		DeleteRequestedDestination (ident);
-		DeleteRequestedDestination (r->GetIdentHash ());
+		auto it = m_RouterInfos.find(ident);
-		auto it = m_RouterInfos.find(r->GetIdentHash ());
 		if (it != m_RouterInfos.end ())
 		{
-			if (r->GetTimestamp () > it->second->GetTimestamp ())
+			auto ts = it->second->GetTimestamp ();
-			{
+			it->second->Update (buf, len);
+			if (it->second->GetTimestamp () > ts)
 				LogPrint ("RouterInfo updated");
-				*(it->second) = *r; // we can't replace pointer because it's used by tunnels
-			}				
-			delete r;
 		}	
 		else	
 		{	
 			LogPrint ("New RouterInfo added");
+			RouterInfo * r = new RouterInfo (buf, len);
 			m_RouterInfos[r->GetIdentHash ()] = r;
 			if (r->IsFloodfill ())
 				m_Floodfills.push_back (r);
 		}	
 	}	
 
-	void NetDb::AddLeaseSet (uint8_t * buf, int len)
+	void NetDb::AddLeaseSet (const IdentHash& ident, uint8_t * buf, int len)
 	{
-		LeaseSet * l = new LeaseSet (buf, len);
+		DeleteRequestedDestination (ident);
-		DeleteRequestedDestination (l->GetIdentHash ());
+		auto it = m_LeaseSets.find(ident);
-		auto it = m_LeaseSets.find(l->GetIdentHash ());
 		if (it != m_LeaseSets.end ())
 		{
+			it->second->Update (buf, len); 
 			LogPrint ("LeaseSet updated");
-			*(it->second) = *l; // we can't replace pointer because it's used by streams
-			delete l;
 		}
 		else
 		{	
 			LogPrint ("New LeaseSet added");
-			m_LeaseSets[l->GetIdentHash ()] = l;
+			m_LeaseSets[ident] = new LeaseSet (buf, len);
 		}	
 	}	
 
@@ -395,7 +391,7 @@ namespace data
 		if (msg->type)
 		{
 			LogPrint ("LeaseSet");
-			AddLeaseSet (buf + offset, len - offset);
+			AddLeaseSet (msg->key, buf + offset, len - offset);
 		}	
 		else
 		{
@@ -413,7 +409,7 @@ namespace data
 			uint8_t uncompressed[2048];
 			size_t uncomressedSize = decompressor.MaxRetrievable ();
 			decompressor.Get (uncompressed, uncomressedSize);
-			AddRouterInfo (uncompressed, uncomressedSize);
+			AddRouterInfo (msg->key, uncompressed, uncomressedSize);
 		}	
 	}	
 

NetDb.h

@@ -58,8 +58,8 @@ namespace data
 			void Start ();
 			void Stop ();
 			
-			void AddRouterInfo (uint8_t * buf, int len);
+			void AddRouterInfo (const IdentHash& ident, uint8_t * buf, int len);
-			void AddLeaseSet (uint8_t * buf, int len);
+			void AddLeaseSet (const IdentHash& ident, uint8_t * buf, int len);
 			RouterInfo * FindRouter (const IdentHash& ident) const;
 			LeaseSet * FindLeaseSet (const IdentHash& destination) const;
 			const IdentHash * FindAddress (const std::string& address) { return m_AddressBook.FindAddress (address); }; // TODO: move AddressBook away from NetDb

RouterInfo.cpp

@@ -31,6 +31,19 @@ namespace data
 		ReadFromBuffer ();
 	}	
 
+	void RouterInfo::Update (const uint8_t * buf, int len)
+	{
+		m_IsUpdated = true;
+		m_IsUnreachable = false;
+		m_SupportedTransports = 0;
+		m_Caps = 0;
+		m_Addresses.clear ();
+		m_Properties.clear ();
+		memcpy (m_Buffer, buf, len);
+		m_BufferLen = len;
+		ReadFromBuffer ();
+	}	
+		
 	void RouterInfo::SetRouterIdentity (const Identity& identity)
 	{	
 		m_RouterIdentity = identity;

RouterInfo.h

@@ -68,6 +68,7 @@ namespace data
 			RouterInfo (const RouterInfo& ) = default;
 			RouterInfo& operator=(const RouterInfo& ) = default;
 			RouterInfo (const uint8_t * buf, int len);
+			void Update (const uint8_t * buf, int len);
 			
 			const Identity& GetRouterIdentity () const { return m_RouterIdentity; };
 			void SetRouterIdentity (const Identity& identity);