init

src/server.cpp

@@ -0,0 +1,105 @@
+#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());
+	}
+}