summaryrefslogtreecommitdiffstats
path: root/src/RCONServer.h
blob: 18e92a2aafa75db0b4b53caabb84652a68d6c48f (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91

// RCONServer.h

// Declares the cRCONServer class representing the RCON server





#pragma once

#include "OSSupport/Network.h"





// fwd:
class cServer;
class cSettingsRepositoryInterface;





class cRCONServer
{
  public:
	cRCONServer(cServer & a_Server);
	virtual ~cRCONServer();

	void Initialize(cSettingsRepositoryInterface & a_Settings);

  protected:
	friend class cRCONCommandOutput;
	friend class cRCONListenCallbacks;

	class cConnection : public cTCPLink::cCallbacks
	{
	  public:
		cConnection(cRCONServer & a_RCONServer, const AString & a_IPAddress);

	  protected:
		friend class cRCONCommandOutput;

		/** Set to true if the client has successfully authenticated */
		bool m_IsAuthenticated;

		/** Buffer for the incoming data */
		AString m_Buffer;

		/** Server that owns this connection and processes requests */
		cRCONServer & m_RCONServer;

		/** The TCP link to the client */
		cTCPLinkPtr m_Link;

		/** Address of the client */
		AString m_IPAddress;


		// cTCPLink::cCallbacks overrides:
		virtual void OnLinkCreated(cTCPLinkPtr a_Link) override;
		virtual void OnReceivedData(const char * a_Data, size_t a_Length) override;
		virtual void OnRemoteClosed(void) override;
		virtual void OnError(int a_ErrorCode, const AString & a_ErrorMsg) override;

		/** Processes the given packet and sends the response; returns true if successful, false if the connection is to
		 * be dropped */
		bool ProcessPacket(UInt32 a_RequestID, UInt32 a_PacketType, UInt32 a_PayloadLength, const char * a_Payload);

		/** Reads 4 bytes from a_Buffer and returns the LE UInt32 they represent */
		UInt32 UIntFromBuffer(const char * a_Buffer);

		/** Puts 4 bytes representing the int into the buffer */
		void UIntToBuffer(UInt32 a_Value, char * a_Buffer);

		/** Sends a RCON packet back to the client */
		void SendResponse(UInt32 a_RequestID, UInt32 a_PacketType, UInt32 a_PayloadLength, const char * a_Payload);
	};


	/** The server object that will process the commands received */
	cServer & m_Server;

	/** The sockets for accepting RCON connections (one socket per port). */
	cServerHandlePtrs m_ListenServers;

	/** Password for authentication */
	AString m_Password;
};