/*
*
*    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
*
*    In addition, as a special exception, the author gives permission to
*    link the code of this program with the Half-Life Game Engine ("HL
*    Engine") and Modified Game Libraries ("MODs") developed by Valve,
*    L.L.C ("Valve").  You must obey the GNU General Public License in all
*    respects for all of the code used other than the HL Engine and MODs
*    from Valve.  If you modify this file, you may extend this exception
*    to your version of the file, but you are not obligated to do so.  If
*    you do not wish to do so, delete this exception statement from your
*    version.
*
*/

#pragma once

#include "maintypes.h"
#include "net.h"
#include "utlrbtree.h"

class CIPRateLimit
{
public:
	CIPRateLimit();
	~CIPRateLimit();

	// updates an ip entry, return true if the ip is allowed, false otherwise
	bool CheckIP(netadr_t adr);

private:
	enum
	{
		MAX_TREE_SIZE = 512,
		START_TREE_SIZE = 256,
		FLUSH_TIMEOUT = 120,
	};

	using ip_t = int;
	struct iprate_t
	{
		ip_t ip;
		long lastTime;
		int count;
	};

private:
	CUtlRBTree<iprate_t, int> m_IPTree;
	int m_iGlobalCount;
	long m_lLastTime;

	static bool LessIP(const iprate_t &lhs, const iprate_t &rhs);
};

#ifdef SHARED_GAME_DATA
extern cvar_t& max_queries_sec;
extern cvar_t& max_queries_sec_global;
extern cvar_t& max_queries_window;
extern cvar_t& sv_logblocks;
#else
extern cvar_t max_queries_sec;
extern cvar_t max_queries_sec_global;
extern cvar_t max_queries_window;
extern cvar_t sv_logblocks;
#endif
bool SV_CheckConnectionLessRateLimits(netadr_t &adr);
