From 5e5b87a18797967345d4933981c8500176de3870 Mon Sep 17 00:00:00 2001
From: "madmaxoft@gmail.com"
 <madmaxoft@gmail.com@0a769ca7-a7f5-676a-18bf-c427514a06d6>
Date: Tue, 4 Sep 2012 19:06:46 +0000
Subject: Debugging in cByteBuffer::Write(); added the ReadAgain() method to
 allow ProtoProxy re-send the data it has parsed.

git-svn-id: http://mc-server.googlecode.com/svn/trunk@833 0a769ca7-a7f5-676a-18bf-c427514a06d6
---
 source/ByteBuffer.cpp | 28 ++++++++++++++++++++++++++++
 source/ByteBuffer.h   |  3 +++
 2 files changed, 31 insertions(+)

(limited to 'source')

diff --git a/source/ByteBuffer.cpp b/source/ByteBuffer.cpp
index 0f360a104..4f74ca19f 100644
--- a/source/ByteBuffer.cpp
+++ b/source/ByteBuffer.cpp
@@ -48,6 +48,11 @@ cByteBuffer::~cByteBuffer()
 
 bool cByteBuffer::Write(const char * a_Bytes, int a_Count)
 {
+	// DEBUG: Store the current free space for a check after writing
+	int CurFreeSpace = GetFreeSpace();
+	int CurReadableSpace = GetReadableSpace();
+	int WrittenBytes = 0;
+	
 	if (GetFreeSpace() < a_Count)
 	{
 		return false;
@@ -60,11 +65,16 @@ bool cByteBuffer::Write(const char * a_Bytes, int a_Count)
 		m_WritePos = 0;
 		a_Bytes += TillEnd;
 		a_Count -= TillEnd;
+		WrittenBytes = TillEnd;
 	}
 	
 	// We're guaranteed that we'll fit in a single write op
 	memcpy(m_Buffer + m_WritePos, a_Bytes, a_Count);
 	m_WritePos += a_Count;
+	WrittenBytes += a_Count;
+	
+	ASSERT(GetFreeSpace() == CurFreeSpace - WrittenBytes);
+	ASSERT(GetReadableSpace() == CurReadableSpace + WrittenBytes);
 	return true;
 }
 
@@ -467,6 +477,24 @@ void cByteBuffer::ResetRead(void)
 
 
 
+void cByteBuffer::ReadAgain(AString & a_Out)
+{
+	// Return the data between m_DataStart and m_ReadPos (the data that has been read but not committed)
+	// Used by ProtoProxy to repeat communication twice, once for parsing and the other time for the remote party
+	int DataStart = m_DataStart;
+	if (m_ReadPos < m_DataStart)
+	{
+		// Across the ringbuffer end, read the first part and adjust next part's start:
+		a_Out.append(m_Buffer + m_DataStart, m_BufferSize - m_DataStart);
+		DataStart = 0;
+	}
+	a_Out.append(m_Buffer + DataStart, m_ReadPos - DataStart);
+}
+
+
+
+
+
 void cByteBuffer::AdvanceReadPos(int a_Count)
 {
 	m_ReadPos += a_Count;
diff --git a/source/ByteBuffer.h b/source/ByteBuffer.h
index c0ffed43c..3981ab066 100644
--- a/source/ByteBuffer.h
+++ b/source/ByteBuffer.h
@@ -94,6 +94,9 @@ public:
 	/// Restarts next reading operation at the start of the ringbuffer
 	void ResetRead(void);
 	
+	/// Re-reads the data that has been read since the last commit to the current readpos. Used by ProtoProxy to duplicate communication
+	void ReadAgain(AString & a_Out);
+	
 protected:
 	char * m_Buffer;
 	int    m_BufferSize;  // Total size of the ringbuffer
-- 
cgit v1.2.3