From 00c5a9f234111fff6a97aaa149b551fdb3078bea Mon Sep 17 00:00:00 2001
From: xPaw <github@xpaw.me>
Date: Fri, 20 Feb 2015 14:35:45 +0200
Subject: [PATCH] Partial fix for multi-packet source rcon

---
 SourceQuery/SourceRcon.class.php | 32 +++++++++++++++++++++++++-------
 1 file changed, 25 insertions(+), 7 deletions(-)

diff --git a/SourceQuery/SourceRcon.class.php b/SourceQuery/SourceRcon.class.php
index d6c9b41..3669eec 100644
--- a/SourceQuery/SourceRcon.class.php
+++ b/SourceQuery/SourceRcon.class.php
@@ -75,9 +75,9 @@
 			return $Length === FWrite( $this->RconSocket, $Command, $Length );
 		}
 		
-		public function Read( $Length = 1400 )
+		public function Read( )
 		{
-			$this->Buffer->Set( FRead( $this->RconSocket, $Length ) );
+			$this->Buffer->Set( FRead( $this->RconSocket, 4 ) );
 			
 			if( $this->Buffer->Remaining( ) < 4 )
 			{
@@ -86,16 +86,27 @@
 			
 			$PacketSize = $this->Buffer->GetLong( );
 			
+			$this->Buffer->Set( FRead( $this->RconSocket, $PacketSize ) );
+			
 			$Buffer = $this->Buffer->Get( );
 			
 			$Remaining = $PacketSize - StrLen( $Buffer );
 			
 			while( $Remaining > 0 )
 			{
-				$Buffer2 = FRead( $this->RconSocket, $Length );
-				$Buffer .= $Buffer2;
+				$Buffer2 = FRead( $this->RconSocket, $Remaining );
+				
+				$PacketSize = StrLen( $Buffer2 );
+				
+				if( $PacketSize === 0 )
+				{
+					throw new InvalidPacketException( 'Read ' . strlen( $Buffer ) . ' bytes from socket, ' . $Remaining . ' remaining', InvalidPacketException::BUFFER_EMPTY );
+					
+					break;
+				}
 				
-				$Remaining -= StrLen( $Buffer2 );
+				$Buffer .= $Buffer2;
+				$Remaining -= $PacketSize;
 			}
 			
 			$this->Buffer->Set( $Buffer );
@@ -139,9 +150,16 @@
 						break;
 					}
 					
-					$Buffer .= $this->Buffer->Get( );
+					$Buffer2 = $this->Buffer->Get( );
+					
+					if( $Buffer2 === "\x00\x01\x00\x00\x00\x00" )
+					{
+						break;
+					}
+					
+					$Buffer .= $Buffer2;
 				}
-				while( false ); // TODO: This is so broken that we don't even try to read multiple times, needs to be revised
+				while( true );
 			}
 			
 			// TODO: It should use GetString, but there are no null bytes at the end, why?