h1-mod/src/client/component/rcon.cpp

#include <std_include.hpp>
#include "loader/component_loader.hpp"

#include "game/game.hpp"
#include "game/dvars.hpp"

#include "command.hpp"
#include "console.hpp"
#include "network.hpp"
#include "scheduler.hpp"
#include "rcon.hpp"

#include <utils/hook.hpp>
#include <utils/string.hpp>

namespace rcon
{
	namespace
	{
		bool is_redirecting_ = false;
		bool has_redirected_ = false;
		game::netadr_s redirect_target_ = {};
		std::recursive_mutex redirect_lock;

		void setup_redirect(const game::netadr_s& target)
		{
			std::lock_guard<std::recursive_mutex> $(redirect_lock);

			has_redirected_ = false;
			is_redirecting_ = true;
			redirect_target_ = target;
		}

		void clear_redirect()
		{
			std::lock_guard<std::recursive_mutex> $(redirect_lock);

			has_redirected_ = false;
			is_redirecting_ = false;
			redirect_target_ = {};
		}

		void send_rcon_command(const std::string& password, const std::string& data)
		{
			// If you are the server, don't bother with rcon and just execute the command
			if (game::Dvar_FindVar("sv_running")->current.enabled)
			{
				game::Cbuf_AddText(0, 0, data.data());
				return;
			}

			if (password.empty())
			{
				console::info("You must login first to use RCON\n");
				return;
			}

			if (*game::mp::connect_state != nullptr && *game::connectionState >= game::CA_CONNECTED)
			{
				const auto target = (*game::mp::connect_state)->address;
				const auto buffer = password + " " + data;
				network::send(target, "rcon", buffer);
			}
			else
			{
				console::warn("You need to be connected to a server!\n");
			}
		}

		std::string build_status_buffer()
		{
			const auto sv_maxclients = game::Dvar_FindVar("sv_maxclients");
			const auto mapname = game::Dvar_FindVar("mapname");

			std::string buffer{};
			buffer.append(utils::string::va("map: %s\n", mapname->current.string));
			buffer.append(
				"num score bot ping guid                             name             address               qport\n");
			buffer.append(
				"--- ----- --- ---- -------------------------------- ---------------- --------------------- -----\n");

			const auto svs_clients = *game::mp::svs_clients;
			if (svs_clients == nullptr)
			{
				return buffer;
			}

			for (int i = 0; i < sv_maxclients->current.integer; i++)
			{
				const auto client = &svs_clients[i];

				char clean_name[32] = {0};
				strncpy_s(clean_name, client->name, sizeof(clean_name));
				game::I_CleanStr(clean_name);

				if (client->header.state >= 1)
				{
					buffer.append(utils::string::va("%3i %5i %3s %s %32s %16s %21s %5i\n",
						i,
						game::G_GetClientScore(i),
						game::SV_BotIsBot(i) ? "Yes" : "No",
						(client->header.state == 2)
							? "CNCT"
							: (client->header.state == 1)
								? "ZMBI"
								: utils::string::va("%4i", game::SV_GetClientPing(i)),
						game::SV_GetGuid(i),
						clean_name,
						network::net_adr_to_string(client->header.remoteAddress),
						client->header.remoteAddress.port)
					);
				}
			}

			return buffer;
		}
	}

	bool message_redirect(const std::string& message)
	{
		std::lock_guard<std::recursive_mutex> $(redirect_lock);

		if (is_redirecting_)
		{
			has_redirected_ = true;
			network::send(redirect_target_, "print", message, '\n');
			return true;
		}
		return false;
	}

	class component final : public component_interface
	{
	public:
		void post_unpack() override
		{
			if (game::environment::is_sp())
			{
				return;
			}

			scheduler::once([]()
			{
				dvars::register_string("rcon_password", "", game::DvarFlags::DVAR_FLAG_NONE,
				                          "The password for remote console");
			}, scheduler::pipeline::main);

			command::add("status", []()
			{
				const auto sv_running = game::Dvar_FindVar("sv_running");
				if (game::VirtualLobby_Loaded() || !sv_running || !sv_running->current.enabled)
				{
					console::error("Server is not running\n");
					return;
				}

				auto status_buffer = build_status_buffer();
				console::info(status_buffer.data());
			});

			if (!game::environment::is_dedi())
			{
				command::add("rcon", [&](const command::params& params)
				{
					static std::string rcon_password{};

					if (params.size() < 2) return;

					const auto operation = params.get(1);
					if (operation == "login"s)
					{
						if (params.size() < 3) return;

						rcon_password = params.get(2);
					}
					else if (operation == "logout"s)
					{
						rcon_password.clear();
					}
					else
					{
						send_rcon_command(rcon_password, params.join(1));
					}
				});
			}
			else
			{
				network::on("rcon", [](const game::netadr_s& addr, const std::string_view& data)
				{
					const auto message = std::string{data};
					const auto pos = message.find_first_of(" ");
					if (pos == std::string::npos)
					{
						network::send(addr, "print", "Invalid RCon request", '\n');
						console::info("Invalid RCon request from %s\n", network::net_adr_to_string(addr));
						return;
					}

					const auto password = message.substr(0, pos);
					const auto command = message.substr(pos + 1);
					const auto rcon_password = game::Dvar_FindVar("rcon_password");
					if (command.empty() || !rcon_password || !rcon_password->current.string || !strlen(
						rcon_password->current.string))
					{
						return;
					}

					setup_redirect(addr);

					if (password != rcon_password->current.string)
					{
						network::send(redirect_target_, "print", "Invalid rcon password", '\n');
						console::error("Invalid rcon password\n");
					}
					else
					{
						command::execute(command, true);
					}

					if (!has_redirected_)
					{
						network::send(redirect_target_, "print", "", '\n');
					}

					clear_redirect();
				});
			}
		}
	};
}

REGISTER_COMPONENT(rcon::component)