From 01d7b8e2463c3cb8e1f220e64abd5d560b886bc7 Mon Sep 17 00:00:00 2001
From: w00t <w00t@e03df62e-2008-0410-955e-edbf42e46eb7>
Date: Sun, 6 Jan 2008 01:15:58 +0000
Subject: [PATCH] Add bancache expiry stuff, currently records expire 60
 seconds after creation. Live, we will want perhaps an hour before expiry.

git-svn-id: http://svn.inspircd.org/repository/trunk/inspircd@8640 e03df62e-2008-0410-955e-edbf42e46eb7
---
 include/bancache.h |  4 +++-
 src/bancache.cpp   | 29 ++++++++++++++++++++++++-----
 2 files changed, 27 insertions(+), 6 deletions(-)

diff --git a/include/bancache.h b/include/bancache.h
index cabd85365..75d1a67b5 100644
--- a/include/bancache.h
+++ b/include/bancache.h
@@ -24,6 +24,7 @@ class CoreExport BanCacheHit : public classbase
 	std::string Type;
 	std::string Reason;
 	std::string IP;
+	time_t Expiry;
 
 	BanCacheHit(InspIRCd *Instance, const std::string &ip, const std::string &type, const std::string &reason)
 	{
@@ -31,6 +32,7 @@ class CoreExport BanCacheHit : public classbase
 		this->Type = type;
 		this->Reason = reason;
 		this->IP = ip;
+		this->Expiry = time(NULL) + 60; // XXX changeme
 	}
 };
 
@@ -61,7 +63,7 @@ class CoreExport BanCacheManager : public classbase
 	 * @param type The type of bancache entries to remove (e.g. 'G')
 	 * @param positive Remove either positive (true) or negative (false) hits.
 	 */
-	int RemoveEntries(const std::string &type, bool positive);
+	unsigned int RemoveEntries(const std::string &type, bool positive);
 
 	BanCacheManager(InspIRCd *Instance)
 	{
diff --git a/src/bancache.cpp b/src/bancache.cpp
index 05c014de3..24578b379 100644
--- a/src/bancache.cpp
+++ b/src/bancache.cpp
@@ -20,7 +20,6 @@ BanCacheHit *BanCacheManager::AddHit(const std::string &ip, const std::string &t
 {
 	BanCacheHit *b;
 
-
 	if (this->BanHash->find(ip) != this->BanHash->end()) // can't have two cache entries on the same IP, sorry..
 		return NULL;
 
@@ -37,7 +36,16 @@ BanCacheHit *BanCacheManager::GetHit(const std::string &ip)
 	if (i == this->BanHash->end())
 		return NULL; // free and safe
 	else
+	{
+		if (time(NULL) > i->second->Expiry)
+		{
+			ServerInstance->Log(DEBUG, "Hit on " + ip + " is out of date, removing!");
+			RemoveHit(i->second);
+			return NULL; // out of date
+		}
+
 		return i->second; // hit.
+	}
 }
 
 bool BanCacheManager::RemoveHit(BanCacheHit *b)
@@ -63,7 +71,7 @@ bool BanCacheManager::RemoveHit(BanCacheHit *b)
 	return true;
 }
 
-int BanCacheManager::RemoveEntries(const std::string &type, bool positive)
+unsigned int BanCacheManager::RemoveEntries(const std::string &type, bool positive)
 {
 	int removed = 0;
 
@@ -84,7 +92,7 @@ int BanCacheManager::RemoveEntries(const std::string &type, bool positive)
 				/* we need to remove this one. */
 				delete b;
 				b = NULL;
-				BanHash->erase(n);
+				BanHash->erase(n); // WORD TO THE WISE: don't use RemoveHit here, because we MUST remove the iterator in a safe way.
 				removed++;
 			}
 		}
@@ -108,9 +116,20 @@ void BanCacheManager::RehashCache()
 		safei++;
 
 		/* Safe to delete items here through iterator 'n' */
+		BanCacheHit *b = n->second;
 
-		/* Actually inserts a std::pair */
-		NewHash->insert(*n);
+		if (time(NULL) > b->Expiry)
+		{
+			/* we need to remove this one. */
+			delete b;
+			b = NULL;
+			BanHash->erase(n); // WORD TO THE WISE: don't use RemoveHit here, because we MUST remove the iterator in a safe way.
+		}
+		else
+		{
+			/* Actually inserts a std::pair */
+			NewHash->insert(*n);
+		}
 
 		/* End of safe section */
 
-- 
2.39.2