X-Git-Url: https://git.netwichtig.de/gitweb/?a=blobdiff_plain;f=include%2Fbancache.h;h=944d69816e5450d19d1de264a8ac5a8e81d0bf9a;hb=8c2d96013084de950e3a63be4ae6ed626c4093ab;hp=19fc0d9528909dcc58f89603e2033c8dbe5c0252;hpb=cce3669a3a6c2011f202795c6333db77009cedd7;p=user%2Fhenk%2Fcode%2Finspircd.git

diff --git a/include/bancache.h b/include/bancache.h
index 19fc0d952..944d69816 100644
--- a/include/bancache.h
+++ b/include/bancache.h
@@ -2,7 +2,7 @@
  *       | Inspire Internet Relay Chat Daemon |
  *       +------------------------------------+
  *
- *  InspIRCd: (C) 2002-2007 InspIRCd Development Team
+ *  InspIRCd: (C) 2002-2008 InspIRCd Development Team
  * See: http://www.inspircd.org/wiki/index.php/Credits
  *
  * This program is free but copyrighted software; see
@@ -14,8 +14,6 @@
 #ifndef __BANCACHE_H
 #define __BANCACHE_H
 
-#include <string>
-
 class CoreExport BanCacheHit : public classbase
 {
  private:
@@ -24,6 +22,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 +30,17 @@ class CoreExport BanCacheHit : public classbase
 		this->Type = type;
 		this->Reason = reason;
 		this->IP = ip;
+		this->Expiry = time(NULL) + 86400; // a day. this might seem long, but entries will be removed as glines/etc expire.
+	}
+
+	// overridden to allow custom time
+	BanCacheHit(InspIRCd *Instance, const std::string &ip, const std::string &type, const std::string &reason, time_t seconds)
+	{
+		ServerInstance = Instance;
+		this->Type = type;
+		this->Reason = reason;
+		this->IP = ip;
+		this->Expiry = time(NULL) + seconds;
 	}
 };
 
@@ -38,7 +48,7 @@ class CoreExport BanCacheHit : public classbase
 #ifndef WIN32
 typedef nspace::hash_map<std::string, BanCacheHit *, nspace::hash<std::string> > BanCacheHash;
 #else
-typedef nspace::hash_map<std::string, BanCacheHit*, nspace::hash_compare<string, less<string> > > BanCacheHash;
+typedef nspace::hash_map<std::string, BanCacheHit*, nspace::hash_compare<std::string, std::less<std::string> > > BanCacheHash;
 #endif
 
 class CoreExport BanCacheManager : public classbase
@@ -54,14 +64,25 @@ class CoreExport BanCacheManager : public classbase
 	 * @param reason The reason for the ban. Left .empty() if it's a negative match.
 	 */
 	BanCacheHit *AddHit(const std::string &ip, const std::string &type, const std::string &reason);
+
+	// Overridden to allow an optional number of seconds before expiry
+	BanCacheHit *AddHit(const std::string &ip, const std::string &type, const std::string &reason, time_t seconds);
 	BanCacheHit *GetHit(const std::string &ip);
 	bool RemoveHit(BanCacheHit *b);
 
+	/** Removes all entries of a given type, either positive or negative. Returns the number of hits removed.
+	 * @param type The type of bancache entries to remove (e.g. 'G')
+	 * @param positive Remove either positive (true) or negative (false) hits.
+	 */
+	unsigned int RemoveEntries(const std::string &type, bool positive);
+
 	BanCacheManager(InspIRCd *Instance)
 	{
 		this->ServerInstance = Instance;
 		this->BanHash = new BanCacheHash();
 	}
+
+	void RehashCache();
 };
 
 #endif