From e0a44fb3bc250197e30c5023fa53fc5c27a5cdcf Mon Sep 17 00:00:00 2001
From: peterbell10 <peterbell10@live.co.uk>
Date: Mon, 17 Jul 2017 00:19:09 +0100
Subject: Handle incomplete packets in cProtocolRecognizer

---
 src/Protocol/ProtocolRecognizer.h | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

(limited to 'src/Protocol/ProtocolRecognizer.h')

diff --git a/src/Protocol/ProtocolRecognizer.h b/src/Protocol/ProtocolRecognizer.h
index 1a638fb21..32dcca940 100644
--- a/src/Protocol/ProtocolRecognizer.h
+++ b/src/Protocol/ProtocolRecognizer.h
@@ -143,8 +143,6 @@ public:
 
 	virtual void SendData(const char * a_Data, size_t a_Size) override;
 
-	void SendPingStatusResponse(void);
-
 protected:
 	/** The recognized protocol */
 	cProtocol * m_Protocol;
@@ -155,6 +153,9 @@ protected:
 	/** Is a server list ping for an unrecognized version currently occuring? */
 	bool m_InPingForUnrecognizedVersion;
 
+	// Packet handlers while in status state (m_InPingForUnrecognizedVersion == true)
+	void HandlePacketStatusRequest();
+	void HandlePacketStatusPing();
 
 	/** Tries to recognize protocol based on m_Buffer contents; returns true if recognized */
 	bool TryRecognizeProtocol(void);
-- 
cgit v1.2.3