master-server/src/server.cpp
2023-05-26 16:09:29 +02:00

106 lines
2.8 KiB
C++

#include "std_include.hpp"
#include "server.hpp"
#include "service.hpp"
#include "console.hpp"
#include "services/getbots_command.hpp"
#include "services/getservers_command.hpp"
#include "services/heartbeat_command.hpp"
#include "services/info_response_command.hpp"
#include "services/patch_kill_list_command.hpp"
#include "services/ping_handler.hpp"
#include "services/elimination_handler.hpp"
#include "services/statistics_handler.hpp"
#include "services/kill_list.hpp"
server::server(const network::address& bind_addr)
: server_base(bind_addr)
{
this->register_service<getbots_command>();
this->register_service<getservers_command>();
this->register_service<heartbeat_command>();
this->register_service<info_response_command>();
this->register_service<patch_kill_list_command>();
this->register_service<ping_handler>();
this->register_service<elimination_handler>();
this->register_service<statistics_handler>();
this->register_service<kill_list>();
}
server_list& server::get_server_list()
{
return this->server_list_;
}
const server_list& server::get_server_list() const
{
return this->server_list_;
}
client_list& server::get_client_list()
{
return this->client_list_;
}
const client_list& server::get_client_list() const
{
return this->client_list_;
}
void server::run_frame()
{
for (const auto& service : services_)
{
try
{
service->run_frame();
}
catch (const service::execution_exception& e)
{
console::warn("Execption in service: %s", e.what());
}
catch (const std::exception& e)
{
console::error("Fatal execption in service: %s", e.what());
}
}
}
void server::handle_command(const network::address& target, const std::string_view& command,
const std::string_view& data)
{
const auto handler = this->command_services_.find(std::string{command});
if (handler == this->command_services_.end())
{
console::warn("Unhandled command (%s): %.*s", target.to_string().data(), command.size(), command.data());
return;
}
std::string ban_reason;
if (this->get_service<kill_list>()->contains(target, ban_reason))
{
console::log("Refused command from server %s - target is on the kill list (%s)", target.to_string().data(), ban_reason.data());
return;
}
#ifdef DEBUG
console::log("Handling command (%s): %.*s - %.*s", target.to_string().data(), command.size(), command.data(),
data.size(), data.data());
#endif
try
{
handler->second->handle_command(target, data);
}
catch (const service::execution_exception& e)
{
console::warn("Execption in command %.*s (%s): %s", command.size(), command.data(), target.to_string().data(),
e.what());
}
catch (const std::exception& e)
{
console::error("Fatal execption in command %.*s (%s): %s", command.size(), command.data(),
target.to_string().data(), e.what());
}
}