2013-07-29 07:13:03 -04:00
// Connection.h
// Interfaces to the cConnection class representing a single pair of connected sockets
# pragma once
# include "ByteBuffer.h"
2013-08-30 04:10:56 -04:00
# include "../../source/OSSupport/Timer.h"
2013-07-29 07:13:03 -04:00
class cServer ;
class cConnection
{
AString m_LogNameBase ; ///< Base for the log filename and all files connected to this log
int m_ItemIdx ; ///< Index for the next file into which item metadata should be written (ParseSlot() function)
cCriticalSection m_CSLog ;
FILE * m_LogFile ;
cServer & m_Server ;
SOCKET m_ClientSocket ;
SOCKET m_ServerSocket ;
2013-08-30 04:10:56 -04:00
cTimer m_Timer ;
long long m_BeginTick ; // Tick when the relative time was first retrieved (used for GetRelativeTime())
2013-07-29 07:13:03 -04:00
enum eConnectionState
{
csUnencrypted , // The connection is not encrypted. Packets must be decoded in order to be able to start decryption.
csEncryptedUnderstood , // The communication is encrypted and so far all packets have been understood, so they can be still decoded
csEncryptedUnknown , // The communication is encrypted, but an unknown packet has been received, so packets cannot be decoded anymore
csWaitingForEncryption , // The communication is waiting for the other line to establish encryption
} ;
eConnectionState m_ClientState ;
eConnectionState m_ServerState ;
int m_Nonce ;
public :
cConnection ( SOCKET a_ClientSocket , cServer & a_Server ) ;
~ cConnection ( ) ;
void Run ( void ) ;
void Log ( const char * a_Format , . . . ) ;
void DataLog ( const void * a_Data , int a_Size , const char * a_Format , . . . ) ;
void LogFlush ( void ) ;
protected :
typedef CFB_Mode < AES > : : Encryption Encryptor ;
typedef CFB_Mode < AES > : : Decryption Decryptor ;
cByteBuffer m_ClientBuffer ;
cByteBuffer m_ServerBuffer ;
Decryptor m_ServerDecryptor ;
Encryptor m_ServerEncryptor ;
Decryptor m_ClientDecryptor ;
Encryptor m_ClientEncryptor ;
AString m_ClientEncryptionBuffer ; // Buffer for the data to be sent to the client once encryption is established
AString m_ServerEncryptionBuffer ; // Buffer for the data to be sent to the server once encryption is established
/// Set to true when PACKET_PING is received from the client; will cause special parsing for server kick
bool m_HasClientPinged ;
2013-10-28 18:05:53 -04:00
/// State the protocol is in (as defined by the initial handshake), -1 if no initial handshake received yet
int m_ProtocolState ;
2013-07-29 07:13:03 -04:00
bool ConnectToServer ( void ) ;
/// Relays data from server to client; returns false if connection aborted
bool RelayFromServer ( void ) ;
/// Relays data from client to server; returns false if connection aborted
bool RelayFromClient ( void ) ;
/// Returns the time relative to the first call of this function, in the fractional seconds elapsed
double GetRelativeTime ( void ) ;
/// Sends data to the specified socket. If sending fails, prints a fail message using a_Peer and returns false.
bool SendData ( SOCKET a_Socket , const char * a_Data , int a_Size , const char * a_Peer ) ;
/// Sends data to the specified socket. If sending fails, prints a fail message using a_Peer and returns false.
bool SendData ( SOCKET a_Socket , cByteBuffer & a_Data , const char * a_Peer ) ;
/// Sends data to the specfied socket, after encrypting it using a_Encryptor. If sending fails, prints a fail message using a_Peer and returns false
bool SendEncryptedData ( SOCKET a_Socket , Encryptor & a_Encryptor , const char * a_Data , int a_Size , const char * a_Peer ) ;
/// Sends data to the specfied socket, after encrypting it using a_Encryptor. If sending fails, prints a fail message using a_Peer and returns false
bool SendEncryptedData ( SOCKET a_Socket , Encryptor & a_Encryptor , cByteBuffer & a_Data , const char * a_Peer ) ;
/// Decodes packets coming from the client, sends appropriate counterparts to the server; returns false if the connection is to be dropped
bool DecodeClientsPackets ( const char * a_Data , int a_Size ) ;
/// Decodes packets coming from the server, sends appropriate counterparts to the client; returns false if the connection is to be dropped
bool DecodeServersPackets ( const char * a_Data , int a_Size ) ;
// Packet handling, client-side:
bool HandleClientAnimation ( void ) ;
bool HandleClientBlockDig ( void ) ;
bool HandleClientBlockPlace ( void ) ;
bool HandleClientChatMessage ( void ) ;
bool HandleClientClientStatuses ( void ) ;
bool HandleClientCreativeInventoryAction ( void ) ;
2013-08-25 15:57:12 -04:00
bool HandleClientDisconnect ( void ) ;
2013-07-29 07:13:03 -04:00
bool HandleClientEncryptionKeyResponse ( void ) ;
bool HandleClientEntityAction ( void ) ;
bool HandleClientHandshake ( void ) ;
bool HandleClientKeepAlive ( void ) ;
bool HandleClientLocaleAndView ( void ) ;
bool HandleClientPing ( void ) ;
bool HandleClientPlayerAbilities ( void ) ;
bool HandleClientPlayerLook ( void ) ;
bool HandleClientPlayerOnGround ( void ) ;
bool HandleClientPlayerPosition ( void ) ;
bool HandleClientPlayerPositionLook ( void ) ;
bool HandleClientPluginMessage ( void ) ;
bool HandleClientSlotSelect ( void ) ;
2013-10-28 18:05:53 -04:00
bool HandleClientStatusPing ( void ) ;
bool HandleClientStatusRequest ( void ) ;
2013-07-31 08:32:05 -04:00
bool HandleClientTabCompletion ( void ) ;
2013-07-29 07:13:03 -04:00
bool HandleClientUpdateSign ( void ) ;
bool HandleClientUseEntity ( void ) ;
bool HandleClientWindowClick ( void ) ;
bool HandleClientWindowClose ( void ) ;
// Packet handling, server-side:
bool HandleServerAttachEntity ( void ) ;
bool HandleServerBlockAction ( void ) ;
bool HandleServerBlockChange ( void ) ;
bool HandleServerChangeGameState ( void ) ;
bool HandleServerChatMessage ( void ) ;
bool HandleServerCollectPickup ( void ) ;
bool HandleServerCompass ( void ) ;
bool HandleServerDestroyEntities ( void ) ;
bool HandleServerEncryptionKeyRequest ( void ) ;
bool HandleServerEncryptionKeyResponse ( void ) ;
bool HandleServerEntity ( void ) ;
bool HandleServerEntityEquipment ( void ) ;
bool HandleServerEntityHeadLook ( void ) ;
bool HandleServerEntityLook ( void ) ;
bool HandleServerEntityMetadata ( void ) ;
bool HandleServerEntityProperties ( void ) ;
bool HandleServerEntityRelativeMove ( void ) ;
bool HandleServerEntityRelativeMoveLook ( void ) ;
bool HandleServerEntityStatus ( void ) ;
bool HandleServerEntityTeleport ( void ) ;
bool HandleServerEntityVelocity ( void ) ;
2013-09-19 15:51:59 -04:00
bool HandleServerExplosion ( void ) ;
2013-07-29 07:13:03 -04:00
bool HandleServerIncrementStatistic ( void ) ;
bool HandleServerKeepAlive ( void ) ;
bool HandleServerKick ( void ) ;
bool HandleServerLogin ( void ) ;
bool HandleServerMapChunk ( void ) ;
bool HandleServerMapChunkBulk ( void ) ;
bool HandleServerMultiBlockChange ( void ) ;
bool HandleServerNamedSoundEffect ( void ) ;
bool HandleServerPlayerAbilities ( void ) ;
bool HandleServerPlayerAnimation ( void ) ;
bool HandleServerPlayerListItem ( void ) ;
bool HandleServerPlayerPositionLook ( void ) ;
bool HandleServerPluginMessage ( void ) ;
bool HandleServerSetExperience ( void ) ;
bool HandleServerSetSlot ( void ) ;
bool HandleServerSlotSelect ( void ) ;
bool HandleServerSoundEffect ( void ) ;
bool HandleServerSpawnMob ( void ) ;
bool HandleServerSpawnNamedEntity ( void ) ;
bool HandleServerSpawnObjectVehicle ( void ) ;
bool HandleServerSpawnPainting ( void ) ;
bool HandleServerSpawnPickup ( void ) ;
2013-10-28 18:05:53 -04:00
bool HandleServerStatusPing ( void ) ;
bool HandleServerStatusResponse ( void ) ;
2013-07-31 08:32:05 -04:00
bool HandleServerTabCompletion ( void ) ;
2013-07-29 07:13:03 -04:00
bool HandleServerTimeUpdate ( void ) ;
bool HandleServerUpdateHealth ( void ) ;
bool HandleServerUpdateSign ( void ) ;
bool HandleServerUpdateTileEntity ( void ) ;
bool HandleServerWindowClose ( void ) ;
bool HandleServerWindowContents ( void ) ;
bool HandleServerWindowOpen ( void ) ;
/// Parses the slot data in a_Buffer into item description; returns true if successful, false if not enough data
bool ParseSlot ( cByteBuffer & a_Buffer , AString & a_ItemDesc ) ;
/// Parses the metadata in a_Buffer into raw metadata in an AString; returns true if successful, false if not enough data
bool ParseMetadata ( cByteBuffer & a_Buffer , AString & a_Metadata ) ;
/// Logs the contents of the metadata in the AString, using Log(). Assumes a_Metadata is valid (parsed by ParseMetadata()). The log is indented by a_IndentCount spaces
void LogMetadata ( const AString & a_Metadata , size_t a_IndentCount ) ;
/// Send EKResp to the server:
void SendEncryptionKeyResponse ( const AString & a_ServerPublicKey , const AString & a_Nonce ) ;
/// Starts client encryption based on the parameters received
void StartClientEncryption ( const AString & a_EncryptedSecret , const AString & a_EncryptedNonce ) ;
} ;