ProtocolRecognizer.h

Go to the documentation of this file.

#pragma once
 
#include "Protocol.h"
 
 
 
 
 
// Adjust these if a new protocol is added or an old one is removed:
#define MCS_CLIENT_VERSIONS "1.8.x-1.14.4"
#define MCS_PROTOCOL_VERSIONS "47, 107, 108, 109, 110, 210, 315, 316, 335, 338, 340, 393, 401, 477, 480, 485, 490, 498"
#define MCS_LATEST_PROTOCOL_VERSION 498
 
 
 
 
 
class cMultiVersionProtocol
{
public:
 
    cMultiVersionProtocol();
 
    static AString GetVersionTextFromInt(cProtocol::Version a_ProtocolVersion);
 
    bool VersionRecognitionSuccessful()
    {
        return m_Protocol != nullptr;
    }
 
    auto & operator->()
    {
        return m_Protocol;
    }
 
    void HandleIncomingData(cClientHandle & a_Client, ContiguousByteBuffer & a_Data);
 
    void HandleOutgoingData(ContiguousByteBuffer & a_Data);
 
    void SendDisconnect(cClientHandle & a_Client, const AString & a_Reason);
 
private:
 
    void HandleIncomingDataInRecognitionStage(cClientHandle & a_Client, ContiguousByteBuffer & a_Data);
 
    void HandleIncomingDataInOldPingResponseStage(cClientHandle & a_Client, ContiguousByteBufferView a_Data);
 
    /* Checks if incoming data is an HTTP request and handles it if it is. */
    bool TryHandleHTTPRequest(cClientHandle & a_Client, ContiguousByteBuffer & a_Data);
 
    std::unique_ptr<cProtocol> TryRecognizeLengthedProtocol(cClientHandle & a_Client);
 
    static void SendPacket(cClientHandle & a_Client, cByteBuffer & a_OutPacketBuffer);
 
    static UInt32 GetPacketID(cProtocol::ePacketType a_PacketType);
 
    /* Status handler for unrecognised versions. */
    void HandlePacketStatusRequest(cClientHandle & a_Client);
 
    /* Ping handler for unrecognised versions. */
    void HandlePacketStatusPing(cClientHandle & a_Client);
 
    cByteBuffer m_Buffer;
 
    std::unique_ptr<cProtocol> m_Protocol;
 
    bool m_WaitingForData;
} ;