Initial Commit

database/FileZillaFTP/source/autobanmanager.h

@@ -0,0 +1,55 @@
+// FileZilla Server - a Windows ftp server
+
+// Copyright (C) 2002-2007 - Tim Kosse <tim.kosse@gmx.de>
+
+// This program is free software; you can redistribute it and/or
+// modify it under the terms of the GNU General Public License
+// as published by the Free Software Foundation; either version 2
+// of the License, or (at your option) any later version.
+
+// This program is distributed in the hope that it will be useful,
+// but WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+// GNU General Public License for more details.
+
+// You should have received a copy of the GNU General Public License
+// along with this program; if not, write to the Free Software
+// Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
+
+#ifndef __AUTOBANMANAGER_H__
+#define __AUTOBANMANAGER_H__
+
+class COptions;
+class CAutoBanManager
+{
+public:
+
+	CAutoBanManager(COptions* pOptions);
+	virtual ~CAutoBanManager();
+
+	void PurgeOutdated();
+
+	bool IsBanned(const CStdString& ip);
+
+	// Returns true if address got banned
+	bool RegisterAttempt(const CStdString& ip);
+
+protected:
+
+	static int m_refCount;
+	
+	struct t_attemptInfo
+	{
+		int attempts;
+		time_t time;
+	};
+
+	static std::map<CStdString, time_t> m_banMap;
+	static std::map<CStdString, t_attemptInfo> m_attemptMap;
+
+	static CCriticalSectionWrapper m_sync;
+
+	COptions* m_pOptions;
+};
+
+#endif //__AUTOBANMANAGER_H__