Rim/t7x
Rim/t7x
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases 1 Wiki Activity

rcon: use netadr_t as key

Browse Source
This commit is contained in:
Edo 2023-04-26 10:36:34 +01:00 committed by GitHub
parent 8369d6f15f
commit 3a8e096657
1 changed files with 5 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
src/client/component/rcon.cpp
View File

@ -16,7 +16,7 @@ namespace rcon
{ {
const game::dvar_t* rcon_timeout; const game::dvar_t* rcon_timeout;
std::unordered_map<std::size_t, int> rate_limit_map; std::unordered_map<game::netadr_t, int> rate_limit_map;
std::optional<std::string> get_and_validate_rcon_command(const std::string& data) std::optional<std::string> get_and_validate_rcon_command(const std::string& data)
{ {
@ -57,20 +57,19 @@ namespace rcon
bool rate_limit_check(const game::netadr_t& address, const int time) bool rate_limit_check(const game::netadr_t& address, const int time)
{ {
const auto hash = std::hash<game::netadr_t>()(address); if (!rate_limit_map.contains(address))
if (!rate_limit_map.contains(hash))
{ {
rate_limit_map[hash] = 0; rate_limit_map[address] = 0;
} }
const auto last_time = rate_limit_map[hash]; const auto last_time = rate_limit_map[address];
if (last_time && (time - last_time) < rcon_timeout->current.value.integer) if (last_time && (time - last_time) < rcon_timeout->current.value.integer)
{ {
return false; // Flooding return false; // Flooding
} }
rate_limit_map[hash] = time; rate_limit_map[address] = time;
return true; return true;
} }