336 lines
		
	
	
		
			6.5 KiB
		
	
	
	
		
			C#
		
	
	
	
	
	
			
		
		
	
	
			336 lines
		
	
	
		
			6.5 KiB
		
	
	
	
		
			C#
		
	
	
	
	
	
| //
 | |
| // Author:
 | |
| //   Jb Evain (jbevain@gmail.com)
 | |
| //
 | |
| // Copyright (c) 2008 - 2015 Jb Evain
 | |
| // Copyright (c) 2008 - 2011 Novell, Inc.
 | |
| //
 | |
| // Licensed under the MIT/X11 license.
 | |
| //
 | |
| 
 | |
| using System;
 | |
| 
 | |
| namespace MonoFN.Cecil.PE {
 | |
| 
 | |
| 	class ByteBuffer {
 | |
| 
 | |
| 		internal byte [] buffer;
 | |
| 		internal int length;
 | |
| 		internal int position;
 | |
| 
 | |
| 		public ByteBuffer ()
 | |
| 		{
 | |
| 			this.buffer = Empty<byte>.Array;
 | |
| 		}
 | |
| 
 | |
| 		public ByteBuffer (int length)
 | |
| 		{
 | |
| 			this.buffer = new byte [length];
 | |
| 		}
 | |
| 
 | |
| 		public ByteBuffer (byte [] buffer)
 | |
| 		{
 | |
| 			this.buffer = buffer ?? Empty<byte>.Array;
 | |
| 			this.length = this.buffer.Length;
 | |
| 		}
 | |
| 
 | |
| 		public void Advance (int length)
 | |
| 		{
 | |
| 			position += length;
 | |
| 		}
 | |
| 
 | |
| 		public byte ReadByte ()
 | |
| 		{
 | |
| 			return buffer [position++];
 | |
| 		}
 | |
| 
 | |
| 		public sbyte ReadSByte ()
 | |
| 		{
 | |
| 			return (sbyte)ReadByte ();
 | |
| 		}
 | |
| 
 | |
| 		public byte [] ReadBytes (int length)
 | |
| 		{
 | |
| 			var bytes = new byte [length];
 | |
| 			Buffer.BlockCopy (buffer, position, bytes, 0, length);
 | |
| 			position += length;
 | |
| 			return bytes;
 | |
| 		}
 | |
| 
 | |
| 		public ushort ReadUInt16 ()
 | |
| 		{
 | |
| 			ushort value = (ushort)(buffer [position]
 | |
| 				| (buffer [position + 1] << 8));
 | |
| 			position += 2;
 | |
| 			return value;
 | |
| 		}
 | |
| 
 | |
| 		public short ReadInt16 ()
 | |
| 		{
 | |
| 			return (short)ReadUInt16 ();
 | |
| 		}
 | |
| 
 | |
| 		public uint ReadUInt32 ()
 | |
| 		{
 | |
| 			uint value = (uint)(buffer [position]
 | |
| 				| (buffer [position + 1] << 8)
 | |
| 				| (buffer [position + 2] << 16)
 | |
| 				| (buffer [position + 3] << 24));
 | |
| 			position += 4;
 | |
| 			return value;
 | |
| 		}
 | |
| 
 | |
| 		public int ReadInt32 ()
 | |
| 		{
 | |
| 			return (int)ReadUInt32 ();
 | |
| 		}
 | |
| 
 | |
| 		public ulong ReadUInt64 ()
 | |
| 		{
 | |
| 			uint low = ReadUInt32 ();
 | |
| 			uint high = ReadUInt32 ();
 | |
| 
 | |
| 			return (((ulong)high) << 32) | low;
 | |
| 		}
 | |
| 
 | |
| 		public long ReadInt64 ()
 | |
| 		{
 | |
| 			return (long)ReadUInt64 ();
 | |
| 		}
 | |
| 
 | |
| 		public uint ReadCompressedUInt32 ()
 | |
| 		{
 | |
| 			byte first = ReadByte ();
 | |
| 			if ((first & 0x80) == 0)
 | |
| 				return first;
 | |
| 
 | |
| 			if ((first & 0x40) == 0)
 | |
| 				return ((uint)(first & ~0x80) << 8)
 | |
| 					| ReadByte ();
 | |
| 
 | |
| 			return ((uint)(first & ~0xc0) << 24)
 | |
| 				| (uint)ReadByte () << 16
 | |
| 				| (uint)ReadByte () << 8
 | |
| 				| ReadByte ();
 | |
| 		}
 | |
| 
 | |
| 		public int ReadCompressedInt32 ()
 | |
| 		{
 | |
| 			var b = buffer [position];
 | |
| 			var u = (int)ReadCompressedUInt32 ();
 | |
| 			var v = u >> 1;
 | |
| 			if ((u & 1) == 0)
 | |
| 				return v;
 | |
| 
 | |
| 			switch (b & 0xc0) {
 | |
| 			case 0:
 | |
| 			case 0x40:
 | |
| 				return v - 0x40;
 | |
| 			case 0x80:
 | |
| 				return v - 0x2000;
 | |
| 			default:
 | |
| 				return v - 0x10000000;
 | |
| 			}
 | |
| 		}
 | |
| 
 | |
| 		public float ReadSingle ()
 | |
| 		{
 | |
| 			if (!BitConverter.IsLittleEndian) {
 | |
| 				var bytes = ReadBytes (4);
 | |
| 				Array.Reverse (bytes);
 | |
| 				return BitConverter.ToSingle (bytes, 0);
 | |
| 			}
 | |
| 
 | |
| 			float value = BitConverter.ToSingle (buffer, position);
 | |
| 			position += 4;
 | |
| 			return value;
 | |
| 		}
 | |
| 
 | |
| 		public double ReadDouble ()
 | |
| 		{
 | |
| 			if (!BitConverter.IsLittleEndian) {
 | |
| 				var bytes = ReadBytes (8);
 | |
| 				Array.Reverse (bytes);
 | |
| 				return BitConverter.ToDouble (bytes, 0);
 | |
| 			}
 | |
| 
 | |
| 			double value = BitConverter.ToDouble (buffer, position);
 | |
| 			position += 8;
 | |
| 			return value;
 | |
| 		}
 | |
| 
 | |
| 		public void WriteByte (byte value)
 | |
| 		{
 | |
| 			if (position == buffer.Length)
 | |
| 				Grow (1);
 | |
| 
 | |
| 			buffer [position++] = value;
 | |
| 
 | |
| 			if (position > length)
 | |
| 				length = position;
 | |
| 		}
 | |
| 
 | |
| 		public void WriteSByte (sbyte value)
 | |
| 		{
 | |
| 			WriteByte ((byte)value);
 | |
| 		}
 | |
| 
 | |
| 		public void WriteUInt16 (ushort value)
 | |
| 		{
 | |
| 			if (position + 2 > buffer.Length)
 | |
| 				Grow (2);
 | |
| 
 | |
| 			buffer [position++] = (byte)value;
 | |
| 			buffer [position++] = (byte)(value >> 8);
 | |
| 
 | |
| 			if (position > length)
 | |
| 				length = position;
 | |
| 		}
 | |
| 
 | |
| 		public void WriteInt16 (short value)
 | |
| 		{
 | |
| 			WriteUInt16 ((ushort)value);
 | |
| 		}
 | |
| 
 | |
| 		public void WriteUInt32 (uint value)
 | |
| 		{
 | |
| 			if (position + 4 > buffer.Length)
 | |
| 				Grow (4);
 | |
| 
 | |
| 			buffer [position++] = (byte)value;
 | |
| 			buffer [position++] = (byte)(value >> 8);
 | |
| 			buffer [position++] = (byte)(value >> 16);
 | |
| 			buffer [position++] = (byte)(value >> 24);
 | |
| 
 | |
| 			if (position > length)
 | |
| 				length = position;
 | |
| 		}
 | |
| 
 | |
| 		public void WriteInt32 (int value)
 | |
| 		{
 | |
| 			WriteUInt32 ((uint)value);
 | |
| 		}
 | |
| 
 | |
| 		public void WriteUInt64 (ulong value)
 | |
| 		{
 | |
| 			if (position + 8 > buffer.Length)
 | |
| 				Grow (8);
 | |
| 
 | |
| 			buffer [position++] = (byte)value;
 | |
| 			buffer [position++] = (byte)(value >> 8);
 | |
| 			buffer [position++] = (byte)(value >> 16);
 | |
| 			buffer [position++] = (byte)(value >> 24);
 | |
| 			buffer [position++] = (byte)(value >> 32);
 | |
| 			buffer [position++] = (byte)(value >> 40);
 | |
| 			buffer [position++] = (byte)(value >> 48);
 | |
| 			buffer [position++] = (byte)(value >> 56);
 | |
| 
 | |
| 			if (position > length)
 | |
| 				length = position;
 | |
| 		}
 | |
| 
 | |
| 		public void WriteInt64 (long value)
 | |
| 		{
 | |
| 			WriteUInt64 ((ulong)value);
 | |
| 		}
 | |
| 
 | |
| 		public void WriteCompressedUInt32 (uint value)
 | |
| 		{
 | |
| 			if (value < 0x80)
 | |
| 				WriteByte ((byte)value);
 | |
| 			else if (value < 0x4000) {
 | |
| 				WriteByte ((byte)(0x80 | (value >> 8)));
 | |
| 				WriteByte ((byte)(value & 0xff));
 | |
| 			} else {
 | |
| 				WriteByte ((byte)((value >> 24) | 0xc0));
 | |
| 				WriteByte ((byte)((value >> 16) & 0xff));
 | |
| 				WriteByte ((byte)((value >> 8) & 0xff));
 | |
| 				WriteByte ((byte)(value & 0xff));
 | |
| 			}
 | |
| 		}
 | |
| 
 | |
| 		public void WriteCompressedInt32 (int value)
 | |
| 		{
 | |
| 			if (value >= 0) {
 | |
| 				WriteCompressedUInt32 ((uint)(value << 1));
 | |
| 				return;
 | |
| 			}
 | |
| 
 | |
| 			if (value > -0x40)
 | |
| 				value = 0x40 + value;
 | |
| 			else if (value >= -0x2000)
 | |
| 				value = 0x2000 + value;
 | |
| 			else if (value >= -0x20000000)
 | |
| 				value = 0x20000000 + value;
 | |
| 
 | |
| 			WriteCompressedUInt32 ((uint)((value << 1) | 1));
 | |
| 		}
 | |
| 
 | |
| 		public void WriteBytes (byte [] bytes)
 | |
| 		{
 | |
| 			var length = bytes.Length;
 | |
| 			if (position + length > buffer.Length)
 | |
| 				Grow (length);
 | |
| 
 | |
| 			Buffer.BlockCopy (bytes, 0, buffer, position, length);
 | |
| 			position += length;
 | |
| 
 | |
| 			if (position > this.length)
 | |
| 				this.length = position;
 | |
| 		}
 | |
| 
 | |
| 		public void WriteBytes (int length)
 | |
| 		{
 | |
| 			if (position + length > buffer.Length)
 | |
| 				Grow (length);
 | |
| 
 | |
| 			position += length;
 | |
| 
 | |
| 			if (position > this.length)
 | |
| 				this.length = position;
 | |
| 		}
 | |
| 
 | |
| 		public void WriteBytes (ByteBuffer buffer)
 | |
| 		{
 | |
| 			if (position + buffer.length > this.buffer.Length)
 | |
| 				Grow (buffer.length);
 | |
| 
 | |
| 			Buffer.BlockCopy (buffer.buffer, 0, this.buffer, position, buffer.length);
 | |
| 			position += buffer.length;
 | |
| 
 | |
| 			if (position > this.length)
 | |
| 				this.length = position;
 | |
| 		}
 | |
| 
 | |
| 		public void WriteSingle (float value)
 | |
| 		{
 | |
| 			var bytes = BitConverter.GetBytes (value);
 | |
| 
 | |
| 			if (!BitConverter.IsLittleEndian)
 | |
| 				Array.Reverse (bytes);
 | |
| 
 | |
| 			WriteBytes (bytes);
 | |
| 		}
 | |
| 
 | |
| 		public void WriteDouble (double value)
 | |
| 		{
 | |
| 			var bytes = BitConverter.GetBytes (value);
 | |
| 
 | |
| 			if (!BitConverter.IsLittleEndian)
 | |
| 				Array.Reverse (bytes);
 | |
| 
 | |
| 			WriteBytes (bytes);
 | |
| 		}
 | |
| 
 | |
| 		void Grow (int desired)
 | |
| 		{
 | |
| 			var current = this.buffer;
 | |
| 			var current_length = current.Length;
 | |
| 
 | |
| 			var buffer = new byte [System.Math.Max (current_length + desired, current_length * 2)];
 | |
| 			Buffer.BlockCopy (current, 0, buffer, 0, current_length);
 | |
| 			this.buffer = buffer;
 | |
| 		}
 | |
| 	}
 | |
| }
 |