mirror of
https://github.com/Mauler125/r5sdk.git
synced 2025-02-09 19:15:03 +01:00
85f586bd2e
* Prevent attacker from being able to abuse and overflow the banned list vector. * Improved IPv6 comparison performance. * Change size fields of payload frame from unsigned to signed. * Close all accepted sockets on RCON server shutdown.
69 lines
2.6 KiB
C++
69 lines
2.6 KiB
C++
#pragma once
|
|
#include "tier1/NetAdr.h"
|
|
#include "tier2/socketcreator.h"
|
|
#include "protoc/sv_rcon.pb.h"
|
|
#include "protoc/cl_rcon.pb.h"
|
|
|
|
constexpr char s_pszNoAuthMessage[] = "This server is password protected for console access; authenticate with 'PASS <password>' command.\n";
|
|
constexpr char s_pszWrongPwMessage[] = "Admin password incorrect.\n";
|
|
constexpr char s_pszBannedMessage[] = "Go away.\n";
|
|
constexpr char s_pszAuthMessage[] = "Authentication successful.\n";
|
|
|
|
#define RCON_MIN_PASSWORD_LEN 8
|
|
#define RCON_MAX_BANNEDLIST_SIZE 512
|
|
|
|
class CRConServer
|
|
{
|
|
public:
|
|
CRConServer(void);
|
|
~CRConServer(void);
|
|
|
|
void Init(void);
|
|
void Shutdown(void);
|
|
|
|
bool SetPassword(const char* pszPassword);
|
|
bool SetWhiteListAddress(const char* pszAddress);
|
|
|
|
void Think(void);
|
|
void RunFrame(void);
|
|
|
|
void Send(const std::string& responseMsg, const std::string& responseVal, const sv_rcon::response_t responseType,
|
|
const int nMessageId = static_cast<int>(eDLL_T::NETCON), const int nMessageType = static_cast<int>(LogType_t::LOG_NET));
|
|
void Send(const SocketHandle_t hSocket, const std::string& responseMsg, const std::string& responseVal,
|
|
const sv_rcon::response_t responseType, const int nMessageId = static_cast<int>(eDLL_T::NETCON), const int nMessageType = static_cast<int>(LogType_t::LOG_NET));
|
|
void Recv(void);
|
|
|
|
std::string Serialize(const std::string& responseMsg, const std::string& responseVal, const sv_rcon::response_t responseType,
|
|
const int nMessageId = static_cast<int>(eDLL_T::NETCON), const int nMessageType = static_cast<int>(LogType_t::LOG_NET)) const;
|
|
cl_rcon::request Deserialize(const std::string& svBuf) const;
|
|
|
|
void Authenticate(const cl_rcon::request& cl_request, CConnectedNetConsoleData* pData);
|
|
bool Comparator(std::string svPassword) const;
|
|
|
|
void ProcessBuffer(const char* pszIn, int nRecvLen, CConnectedNetConsoleData* pData);
|
|
void ProcessMessage(const cl_rcon::request& cl_request);
|
|
|
|
void Execute(const cl_rcon::request& cl_request, const bool bConVar) const;
|
|
bool CheckForBan(CConnectedNetConsoleData* pData);
|
|
|
|
void CloseConnection(void);
|
|
void CloseNonAuthConnection(void);
|
|
|
|
bool ShouldSend(const sv_rcon::response_t responseType) const;
|
|
bool IsInitialized(void) const;
|
|
|
|
private:
|
|
void Send(const std::string& svMessage) const;
|
|
void Send(const SocketHandle_t hSocket, const std::string& svMessage) const;
|
|
|
|
bool m_bInitialized;
|
|
int m_nConnIndex;
|
|
std::unordered_set<std::string> m_BannedList;
|
|
std::string m_svPasswordHash;
|
|
netadr_t m_Address;
|
|
netadr_t m_WhiteListAddress;
|
|
CSocketCreator m_Socket;
|
|
};
|
|
|
|
CRConServer* RCONServer();
|