ClassiCube/ClassicalSharp/Network/Utils/NetReader.cs

using System;
using System.Net.Sockets;

namespace ClassicalSharp {

	internal class NetReader {
		
		public byte[] buffer = new byte[4096 * 4];
		public int size = 0;
		public NetworkStream Stream;
		
		public NetReader( NetworkStream stream ) {
			Stream = stream;
		}
		
		public void ReadPendingData() {
			if( !Stream.DataAvailable ) return;
			// NOTE: Always using a read call that is a multiple of 4096
			// (appears to?) improve read performance.
			int received = Stream.Read( buffer, size, 4096 * 3 );
			size += received;
		}
		
		public void Remove( int byteCount ) {
			size -= byteCount;
			Buffer.BlockCopy( buffer, byteCount, buffer, 0, size );
			// We don't need to zero the old bytes, since they will be overwritten when ReadData() is called.
		}
		
		public int ReadInt32() {
			int value = buffer[0] << 24 | buffer[1] << 16 | buffer[2] << 8 | buffer[3];
			Remove( 4 );
			return value;
		}
		
		public short ReadInt16() {
			short value = (short)( buffer[0] << 8 | buffer[1] );
			Remove( 2 );
			return value;
		}
		
		public sbyte ReadInt8() {
			sbyte value = (sbyte)buffer[0];
			Remove( 1 );
			return value;
		}
		
		public byte ReadUInt8() {
			byte value = buffer[0];
			Remove( 1 );
			return value;
		}
		
		public byte[] ReadBytes( int length ) {
			byte[] data = new byte[length];
			Buffer.BlockCopy( buffer, 0, data, 0, length );
			Remove( length );
			return data;
		}

		public string ReadCp437String() {
			int length = GetString( buffer, false );
			Remove( 64 );
			return new String( characters, 0, length );
		}
		
		public string ReadAsciiString() {
			int length = GetString( buffer, true );
			Remove( 64 );
			return new String( characters, 0, length );
		}
		
		internal string ReadChatString( ref byte messageType, bool useMessageTypes ) {
			if( useMessageTypes ) 
				return ReadCp437String();
					
			messageType = (byte)CpeMessage.Normal;
			int length = GetString( buffer, false );		
			Remove( 64 );
			
			int offset = 0;
			if( length >= womDetail.Length && IsWomDetailString() ) {
				length -= womDetail.Length;
				offset = womDetail.Length;
				messageType = (byte)CpeMessage.Status3;
			}
			return new String( characters, offset, length );
		}
		
		static char[] characters = new char[64];
		const string womDetail = "^detail.user=";		
		static bool IsWomDetailString() {
			for( int i = 0; i < womDetail.Length; i++ ) {
				if( characters[i] != womDetail[i] )
					return false;
			}
			return true;
		}
		
		static int GetString( byte[] data, bool ascii ) {			
			int length = 0;
			for( int i = 63; i >= 0; i-- ) {
				byte code = data[i];
				if( length == 0 && !( code == 0 || code == 0x20 ) )
				   length = i + 1;
				
				if( ascii ) {
					characters[i] = code >= 0x7F ? '?' : (char)code;
					continue;
				}
				
				// Treat code as an index in code page 437
				if( code < 0x20 ) {
					characters[i] = Utils.ControlCharReplacements[code];
				} else if( code < 0x7F ) {
					characters[i] = (char)code;
				} else {
					characters[i] = Utils.ExtendedCharReplacements[code - 0x7F];
				}
			}
			return length;
		}
	}
}