From c757b6d0201f7353807136f7ff07dde070c34c89 Mon Sep 17 00:00:00 2001
From: orignal <i2porignal@yandex.ru>
Date: Fri, 5 May 2023 18:22:11 -0400
Subject: [PATCH] reset floodfill cap for unknown floodfills

---
 libi2pd/NetDb.cpp    | 21 +++++++++++++++------
 libi2pd/RouterInfo.h |  1 +
 2 files changed, 16 insertions(+), 6 deletions(-)

diff --git a/libi2pd/NetDb.cpp b/libi2pd/NetDb.cpp
index 3575b052..18b6694b 100644
--- a/libi2pd/NetDb.cpp
+++ b/libi2pd/NetDb.cpp
@@ -267,7 +267,12 @@ namespace data
 					if (wasFloodfill)
 						m_Floodfills.Remove (r->GetIdentHash ());
 					else if (r->IsEligibleFloodfill ())
-						m_Floodfills.Insert (r);
+					{	
+						if (m_Floodfills.GetSize () < NETDB_NUM_FLOODFILLS_THRESHOLD || r->GetProfile ()->IsReal ())	
+							m_Floodfills.Insert (r);
+						else
+							r->ResetFlooldFill ();
+					}	
 				}
 			}
 			else
@@ -290,12 +295,16 @@ namespace data
 				if (inserted)
 				{
 					LogPrint (eLogInfo, "NetDb: RouterInfo added: ", ident.ToBase64());
-					if (r->IsFloodfill () && r->IsEligibleFloodfill () &&
-						(m_Floodfills.GetSize () < NETDB_NUM_FLOODFILLS_THRESHOLD ||
-						 r->GetProfile ()->IsReal ())) // don't insert floodfill until it's known real if we have enough 
+					if (r->IsFloodfill () && r->IsEligibleFloodfill ())
 					{	
-						std::unique_lock<std::mutex> l(m_FloodfillsMutex);
-						m_Floodfills.Insert (r);
+						if (m_Floodfills.GetSize () < NETDB_NUM_FLOODFILLS_THRESHOLD ||
+						 r->GetProfile ()->IsReal ()) // don't insert floodfill until it's known real if we have enough 
+						{	
+							std::unique_lock<std::mutex> l(m_FloodfillsMutex);
+							m_Floodfills.Insert (r);
+						}
+						else
+							r->ResetFlooldFill ();
 					}
 				}
 				else
diff --git a/libi2pd/RouterInfo.h b/libi2pd/RouterInfo.h
index 35737c9b..47ee1dae 100644
--- a/libi2pd/RouterInfo.h
+++ b/libi2pd/RouterInfo.h
@@ -224,6 +224,7 @@ namespace data
 			void UpdateSupportedTransports ();
 			void UpdateIntroducers (uint64_t ts); // ts in seconds
 			bool IsFloodfill () const { return m_Caps & Caps::eFloodfill; };
+			void ResetFlooldFill () { m_Caps &= ~Caps::eFloodfill; };
 			bool IsECIES () const { return m_RouterIdentity->GetCryptoKeyType () == i2p::data::CRYPTO_KEY_TYPE_ECIES_X25519_AEAD; };
 			bool IsNTCP2 (bool v4only = true) const;
 			bool IsNTCP2V6 () const { return m_SupportedTransports & eNTCP2V6; };