1 files changed, 48 insertions, 0 deletions
diff --git a/NorthstarDedicatedTest/sv_rcon.h b/NorthstarDedicatedTest/sv_rcon.h
new file mode 100644
index 00000000..23ca2118
--- /dev/null
+++ b/NorthstarDedicatedTest/sv_rcon.h
@@ -0,0 +1,48 @@
+#pragma once
+#include "NetAdr2.h"
+#include "socketcreator.h"
+#include "sv_rcon.pb.h"
+#include "cl_rcon.pb.h"
+#include "igameserverdata.h"
+
+constexpr char s_pszNoAuthMessage[] = "This server is password protected for console access. Must send 'PASS <password>' command.\n\r";
+constexpr char s_pszWrongPwMessage[] = "Password incorrect.\n\r";
+constexpr char s_pszBannedMessage[] = "Go away.\n\r";
+constexpr char s_pszAuthMessage[] = "RCON authentication succesfull.\n\r";
+
+class CRConServer
+{
+  public:
+	void Init(void);
+	void Shutdown(void);
+
+	void Think(void);
+	void RunFrame(void);
+
+	void Send(const std::string& svMessage) const;
+	void Recv(void);
+
+	std::string Serialize(const std::string& svRspBuf, const std::string& svRspVal, sv_rcon::response_t response_t) 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 CheckForBan(CConnectedNetConsoleData* pData);
+
+	void CloseConnection(void);
+	void CloseNonAuthConnection(void);
+
+  private:
+	bool m_bInitialized = false;
+	int m_nConnIndex = 0;
+	CNetAdr2* m_pAdr2 = new CNetAdr2();
+	CSocketCreator* m_pSocket = new CSocketCreator();
+	std::vector<std::string> m_vBannedAddress;
+	std::string m_svPasswordHash;
+};
+extern CRConServer* g_pRConServer;
+\ No newline at end of file