using System; using System.IO; using System.Linq; using System.Numerics; using System.Text; using SabreTools.IO.Extensions; using Xunit; namespace SabreTools.IO.Test.Extensions { public class BinaryWriterExtensionsTests { /// /// Test pattern from 0x00-0x0F /// private static readonly byte[] _bytes = [ 0x00, 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, 0x08, 0x09, 0x0A, 0x0B, 0x0C, 0x0D, 0x0E, 0x0F, ]; /// /// Represents the decimal value 0.0123456789 /// private static readonly byte[] _decimalBytes = [ 0x15, 0xCD, 0x5B, 0x07, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x0A, 0x00, ]; [Fact] public void WriteByteValueTest() { var stream = new MemoryStream(new byte[16], 0, 16, true, true); var bw = new BinaryWriter(stream); byte[] expected = [.. _bytes.Take(1)]; bw.Write((byte)0x00); ValidateBytes(expected, stream.GetBuffer()); } [Fact] public void WriteByteBothEndianTest() { var stream = new MemoryStream(new byte[16], 0, 16, true, true); var bw = new BinaryWriter(stream); byte[] expected = [.. _bytes.Take(2)]; int offset = 0; bw.WriteBothEndian(_bytes.ReadByteBothEndian(ref offset)); ValidateBytes(expected, stream.GetBuffer()); } [Fact] public void WriteBytesTest() { var stream = new MemoryStream(new byte[16], 0, 16, true, true); var bw = new BinaryWriter(stream); byte[] expected = [.. _bytes.Take(4)]; bw.Write([0x00, 0x01, 0x02, 0x03]); ValidateBytes(expected, stream.GetBuffer()); } [Fact] public void WriteBytesBigEndianTest() { var stream = new MemoryStream(new byte[16], 0, 16, true, true); var bw = new BinaryWriter(stream); byte[] expected = [.. _bytes.Take(4)]; bw.WriteBigEndian([0x03, 0x02, 0x01, 0x00]); ValidateBytes(expected, stream.GetBuffer()); } [Fact] public void WriteSByteTest() { var stream = new MemoryStream(new byte[16], 0, 16, true, true); var bw = new BinaryWriter(stream); byte[] expected = [.. _bytes.Take(1)]; bw.Write((sbyte)0x00); ValidateBytes(expected, stream.GetBuffer()); } [Fact] public void WriteSByteBothEndianTest() { var stream = new MemoryStream(new byte[16], 0, 16, true, true); var bw = new BinaryWriter(stream); byte[] expected = [.. _bytes.Take(2)]; int offset = 0; bw.WriteBothEndian(_bytes.ReadSByteBothEndian(ref offset)); ValidateBytes(expected, stream.GetBuffer()); } [Fact] public void WriteCharTest() { var stream = new MemoryStream(new byte[16], 0, 16, true, true); var bw = new BinaryWriter(stream); byte[] expected = [.. _bytes.Take(1)]; bw.Write('\0'); ValidateBytes(expected, stream.GetBuffer()); } [Fact] public void WriteCharEncodingTest() { var stream = new MemoryStream(new byte[16], 0, 16, true, true); var bw = new BinaryWriter(stream); byte[] expected = [0x00, 0x00]; bw.Write('\0', Encoding.Unicode); ValidateBytes(expected, stream.GetBuffer()); } [Fact] public void WriteInt16Test() { var stream = new MemoryStream(new byte[16], 0, 16, true, true); var bw = new BinaryWriter(stream); byte[] expected = [.. _bytes.Take(2)]; bw.Write((short)0x0100); ValidateBytes(expected, stream.GetBuffer()); } [Fact] public void WriteInt16BigEndianTest() { var stream = new MemoryStream(new byte[16], 0, 16, true, true); var bw = new BinaryWriter(stream); byte[] expected = [.. _bytes.Take(2)]; bool write = bw.WriteBigEndian((short)0x0001); Assert.True(write); ValidateBytes(expected, stream.GetBuffer()); } [Fact] public void WriteInt16BothEndianTest() { var stream = new MemoryStream(new byte[16], 0, 16, true, true); var bw = new BinaryWriter(stream); byte[] expected = [.. _bytes.Take(4)]; int offset = 0; bw.WriteBothEndian(_bytes.ReadInt16BothEndian(ref offset)); ValidateBytes(expected, stream.GetBuffer()); } [Fact] public void WriteUInt16Test() { var stream = new MemoryStream(new byte[16], 0, 16, true, true); var bw = new BinaryWriter(stream); byte[] expected = [.. _bytes.Take(2)]; bw.Write((ushort)0x0100); ValidateBytes(expected, stream.GetBuffer()); } [Fact] public void WriteUInt16BigEndianTest() { var stream = new MemoryStream(new byte[16], 0, 16, true, true); var bw = new BinaryWriter(stream); byte[] expected = [.. _bytes.Take(2)]; bool write = bw.WriteBigEndian((ushort)0x0001); Assert.True(write); ValidateBytes(expected, stream.GetBuffer()); } [Fact] public void WriteUInt16BothEndianTest() { var stream = new MemoryStream(new byte[16], 0, 16, true, true); var bw = new BinaryWriter(stream); byte[] expected = [.. _bytes.Take(4)]; int offset = 0; bw.WriteBothEndian(_bytes.ReadUInt16BothEndian(ref offset)); ValidateBytes(expected, stream.GetBuffer()); } [Fact] public void WriteHalfTest() { var stream = new MemoryStream(new byte[16], 0, 16, true, true); var bw = new BinaryWriter(stream); byte[] expected = [.. _bytes.Take(2)]; bw.Write(BitConverter.Int16BitsToHalf(0x0100)); ValidateBytes(expected, stream.GetBuffer()); } [Fact] public void WriteHalfBigEndianTest() { var stream = new MemoryStream(new byte[16], 0, 16, true, true); var bw = new BinaryWriter(stream); byte[] expected = [.. _bytes.Take(2)]; bool write = bw.WriteBigEndian(BitConverter.Int16BitsToHalf(0x0001)); Assert.True(write); ValidateBytes(expected, stream.GetBuffer()); } [Fact] public void WriteInt24Test() { var stream = new MemoryStream(new byte[16], 0, 16, true, true); var bw = new BinaryWriter(stream); byte[] expected = [.. _bytes.Take(3)]; bw.WriteAsInt24(0x020100); ValidateBytes(expected, stream.GetBuffer()); } [Fact] public void WriteInt24BigEndianTest() { var stream = new MemoryStream(new byte[16], 0, 16, true, true); var bw = new BinaryWriter(stream); byte[] expected = [.. _bytes.Take(3)]; bool write = bw.WriteAsInt24BigEndian(0x000102); Assert.True(write); ValidateBytes(expected, stream.GetBuffer()); } [Fact] public void WriteUInt24Test() { var stream = new MemoryStream(new byte[16], 0, 16, true, true); var bw = new BinaryWriter(stream); byte[] expected = [.. _bytes.Take(3)]; bw.WriteAsUInt24(0x020100); ValidateBytes(expected, stream.GetBuffer()); } [Fact] public void WriteUInt24BigEndianTest() { var stream = new MemoryStream(new byte[16], 0, 16, true, true); var bw = new BinaryWriter(stream); byte[] expected = [.. _bytes.Take(3)]; bool write = bw.WriteAsUInt24BigEndian(0x000102); Assert.True(write); ValidateBytes(expected, stream.GetBuffer()); } [Fact] public void WriteInt32Test() { var stream = new MemoryStream(new byte[16], 0, 16, true, true); var bw = new BinaryWriter(stream); byte[] expected = [.. _bytes.Take(4)]; bw.Write(0x03020100); ValidateBytes(expected, stream.GetBuffer()); } [Fact] public void WriteInt32BigEndianTest() { var stream = new MemoryStream(new byte[16], 0, 16, true, true); var bw = new BinaryWriter(stream); byte[] expected = [.. _bytes.Take(4)]; bool write = bw.WriteBigEndian(0x00010203); Assert.True(write); ValidateBytes(expected, stream.GetBuffer()); } [Fact] public void WriteInt32BothEndianTest() { var stream = new MemoryStream(new byte[16], 0, 16, true, true); var bw = new BinaryWriter(stream); byte[] expected = [.. _bytes.Take(8)]; int offset = 0; bw.WriteBothEndian(_bytes.ReadInt32BothEndian(ref offset)); ValidateBytes(expected, stream.GetBuffer()); } [Fact] public void WriteUInt32Test() { var stream = new MemoryStream(new byte[16], 0, 16, true, true); var bw = new BinaryWriter(stream); byte[] expected = [.. _bytes.Take(4)]; bw.Write((uint)0x03020100); ValidateBytes(expected, stream.GetBuffer()); } [Fact] public void WriteUInt32BigEndianTest() { var stream = new MemoryStream(new byte[16], 0, 16, true, true); var bw = new BinaryWriter(stream); byte[] expected = [.. _bytes.Take(4)]; bool write = bw.WriteBigEndian((uint)0x00010203); Assert.True(write); ValidateBytes(expected, stream.GetBuffer()); } [Fact] public void WriteUInt32BothEndianTest() { var stream = new MemoryStream(new byte[16], 0, 16, true, true); var bw = new BinaryWriter(stream); byte[] expected = [.. _bytes.Take(8)]; int offset = 0; bw.WriteBothEndian(_bytes.ReadUInt32BothEndian(ref offset)); ValidateBytes(expected, stream.GetBuffer()); } [Fact] public void WriteSingleTest() { var stream = new MemoryStream(new byte[16], 0, 16, true, true); var bw = new BinaryWriter(stream); byte[] expected = [.. _bytes.Take(4)]; bw.Write(BitConverter.Int32BitsToSingle(0x03020100)); ValidateBytes(expected, stream.GetBuffer()); } [Fact] public void WriteSingleBigEndianTest() { var stream = new MemoryStream(new byte[16], 0, 16, true, true); var bw = new BinaryWriter(stream); byte[] expected = [.. _bytes.Take(4)]; bool write = bw.WriteBigEndian(BitConverter.Int32BitsToSingle(0x00010203)); Assert.True(write); ValidateBytes(expected, stream.GetBuffer()); } [Fact] public void WriteInt48Test() { var stream = new MemoryStream(new byte[16], 0, 16, true, true); var bw = new BinaryWriter(stream); byte[] expected = [.. _bytes.Take(6)]; bw.WriteAsInt48(0x050403020100); ValidateBytes(expected, stream.GetBuffer()); } [Fact] public void WriteInt48BigEndianTest() { var stream = new MemoryStream(new byte[16], 0, 16, true, true); var bw = new BinaryWriter(stream); byte[] expected = [.. _bytes.Take(6)]; bool write = bw.WriteAsInt48BigEndian(0x000102030405); Assert.True(write); ValidateBytes(expected, stream.GetBuffer()); } [Fact] public void WriteUInt48Test() { var stream = new MemoryStream(new byte[16], 0, 16, true, true); var bw = new BinaryWriter(stream); byte[] expected = [.. _bytes.Take(6)]; bw.WriteAsUInt48(0x050403020100); ValidateBytes(expected, stream.GetBuffer()); } [Fact] public void WriteUInt48BigEndianTest() { var stream = new MemoryStream(new byte[16], 0, 16, true, true); var bw = new BinaryWriter(stream); byte[] expected = [.. _bytes.Take(6)]; bool write = bw.WriteAsUInt48BigEndian(0x000102030405); Assert.True(write); ValidateBytes(expected, stream.GetBuffer()); } [Fact] public void WriteInt64Test() { var stream = new MemoryStream(new byte[16], 0, 16, true, true); var bw = new BinaryWriter(stream); byte[] expected = [.. _bytes.Take(8)]; bw.Write(0x0706050403020100); ValidateBytes(expected, stream.GetBuffer()); } [Fact] public void WriteInt64BigEndianTest() { var stream = new MemoryStream(new byte[16], 0, 16, true, true); var bw = new BinaryWriter(stream); byte[] expected = [.. _bytes.Take(8)]; bool write = bw.WriteBigEndian(0x0001020304050607); Assert.True(write); ValidateBytes(expected, stream.GetBuffer()); } [Fact] public void WriteInt64BothEndianTest() { var stream = new MemoryStream(new byte[16], 0, 16, true, true); var bw = new BinaryWriter(stream); byte[] expected = [.. _bytes.Take(16)]; int offset = 0; bw.WriteBothEndian(_bytes.ReadInt64BothEndian(ref offset)); ValidateBytes(expected, stream.GetBuffer()); } [Fact] public void WriteUInt64Test() { var stream = new MemoryStream(new byte[16], 0, 16, true, true); var bw = new BinaryWriter(stream); byte[] expected = [.. _bytes.Take(8)]; bw.Write((ulong)0x0706050403020100); ValidateBytes(expected, stream.GetBuffer()); } [Fact] public void WriteUInt64BigEndianTest() { var stream = new MemoryStream(new byte[16], 0, 16, true, true); var bw = new BinaryWriter(stream); byte[] expected = [.. _bytes.Take(8)]; bool write = bw.WriteBigEndian((ulong)0x0001020304050607); Assert.True(write); ValidateBytes(expected, stream.GetBuffer()); } [Fact] public void WriteUInt64BothEndianTest() { var stream = new MemoryStream(new byte[16], 0, 16, true, true); var bw = new BinaryWriter(stream); byte[] expected = [.. _bytes.Take(16)]; int offset = 0; bw.WriteBothEndian(_bytes.ReadUInt64BothEndian(ref offset)); ValidateBytes(expected, stream.GetBuffer()); } [Fact] public void WriteDoubleTest() { var stream = new MemoryStream(new byte[16], 0, 16, true, true); var bw = new BinaryWriter(stream); byte[] expected = [.. _bytes.Take(8)]; bw.Write(BitConverter.Int64BitsToDouble(0x0706050403020100)); ValidateBytes(expected, stream.GetBuffer()); } [Fact] public void WriteDoubleBigEndianTest() { var stream = new MemoryStream(new byte[16], 0, 16, true, true); var bw = new BinaryWriter(stream); byte[] expected = [.. _bytes.Take(8)]; bool write = bw.WriteBigEndian(BitConverter.Int64BitsToDouble(0x0001020304050607)); Assert.True(write); ValidateBytes(expected, stream.GetBuffer()); } [Fact] public void WriteDecimalTest() { var stream = new MemoryStream(new byte[16], 0, 16, true, true); var bw = new BinaryWriter(stream); byte[] expected = [.. _decimalBytes.Take(16)]; bw.Write(0.0123456789M); ValidateBytes(expected, stream.GetBuffer()); } [Fact] public void WriteDecimalBigEndianTest() { var stream = new MemoryStream(new byte[16], 0, 16, true, true); var bw = new BinaryWriter(stream); byte[] expected = [.. _decimalBytes.Take(16).Reverse()]; bool write = bw.WriteBigEndian(0.0123456789M); Assert.True(write); ValidateBytes(expected, stream.GetBuffer()); } [Fact] public void WriteGuidTest() { var stream = new MemoryStream(new byte[16], 0, 16, true, true); var bw = new BinaryWriter(stream); byte[] expected = [.. _bytes.Take(16)]; bool write = bw.Write(new Guid(_bytes)); Assert.True(write); ValidateBytes(expected, stream.GetBuffer()); } [Fact] public void WriteGuidBigEndianTest() { var stream = new MemoryStream(new byte[16], 0, 16, true, true); var bw = new BinaryWriter(stream); byte[] expected = [.. _bytes.Take(16)]; bool write = bw.WriteBigEndian(new Guid([.. Enumerable.Reverse(_bytes)])); Assert.True(write); ValidateBytes(expected, stream.GetBuffer()); } [Fact] public void WriteInt128Test() { var stream = new MemoryStream(new byte[16], 0, 16, true, true); var bw = new BinaryWriter(stream); byte[] expected = [.. _bytes.Take(16)]; bool write = bw.Write((Int128)new BigInteger(_bytes)); Assert.True(write); ValidateBytes(expected, stream.GetBuffer()); } [Fact] public void WriteInt128BigEndianTest() { var stream = new MemoryStream(new byte[16], 0, 16, true, true); var bw = new BinaryWriter(stream); byte[] expected = [.. _bytes.Take(16)]; bool write = bw.WriteBigEndian((Int128)new BigInteger(Enumerable.Reverse(_bytes).ToArray())); Assert.True(write); ValidateBytes(expected, stream.GetBuffer()); } [Fact] public void WriteUInt128Test() { var stream = new MemoryStream(new byte[16], 0, 16, true, true); var bw = new BinaryWriter(stream); byte[] expected = [.. _bytes.Take(16)]; bool write = bw.Write((UInt128)new BigInteger(_bytes)); Assert.True(write); ValidateBytes(expected, stream.GetBuffer()); } [Fact] public void WriteUInt128BigEndianTest() { var stream = new MemoryStream(new byte[16], 0, 16, true, true); var bw = new BinaryWriter(stream); byte[] expected = [.. _bytes.Take(16)]; bool write = bw.WriteBigEndian((UInt128)new BigInteger(Enumerable.Reverse(_bytes).ToArray())); Assert.True(write); ValidateBytes(expected, stream.GetBuffer()); } [Fact] public void WriteNullTerminatedAnsiStringTest() { var stream = new MemoryStream(new byte[4], 0, 4, true, true); var bw = new BinaryWriter(stream); byte[] expected = [0x41, 0x42, 0x43, 0x00]; bool write = bw.WriteNullTerminatedAnsiString("ABC"); Assert.True(write); ValidateBytes(expected, stream.GetBuffer()); } [Fact] public void WriteNullTerminatedUTF8StringTest() { var stream = new MemoryStream(new byte[4], 0, 4, true, true); var bw = new BinaryWriter(stream); byte[] expected = [0x41, 0x42, 0x43, 0x00]; bool write = bw.WriteNullTerminatedUTF8String("ABC"); Assert.True(write); ValidateBytes(expected, stream.GetBuffer()); } [Fact] public void WriteNullTerminatedUnicodeStringTest() { var stream = new MemoryStream(new byte[8], 0, 8, true, true); var bw = new BinaryWriter(stream); byte[] expected = [0x41, 0x00, 0x42, 0x00, 0x43, 0x00, 0x00]; bool write = bw.WriteNullTerminatedUnicodeString("ABC"); Assert.True(write); ValidateBytes(expected, stream.GetBuffer()); } [Fact] public void WriteNullTerminatedUTF32StringTest() { var stream = new MemoryStream(new byte[16], 0, 16, true, true); var bw = new BinaryWriter(stream); byte[] expected = [0x41, 0x00, 0x00, 0x00, 0x42, 0x00, 0x00, 0x00, 0x43, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00]; bool write = bw.WriteNullTerminatedUTF32String("ABC"); Assert.True(write); ValidateBytes(expected, stream.GetBuffer()); } [Fact] public void WritePrefixedAnsiStringTest() { var stream = new MemoryStream(new byte[4], 0, 4, true, true); var bw = new BinaryWriter(stream); byte[] expected = [0x03, 0x41, 0x42, 0x43]; bool write = bw.WritePrefixedAnsiString("ABC"); Assert.True(write); ValidateBytes(expected, stream.GetBuffer()); } [Fact] public void WritePrefixedUnicodeStringTest() { var stream = new MemoryStream(new byte[8], 0, 8, true, true); var bw = new BinaryWriter(stream); byte[] expected = [0x03, 0x00, 0x41, 0x00, 0x42, 0x00, 0x43, 0x00]; bool write = bw.WritePrefixedUnicodeString("ABC"); Assert.True(write); ValidateBytes(expected, stream.GetBuffer()); } [Fact] public void WriteTypeTest() { // Guid var stream = new MemoryStream(new byte[16], 0, 16, true, true); var bw = new BinaryWriter(stream); bool actual = bw.WriteType(new Guid(_bytes)); Assert.True(actual); ValidateBytes(_bytes, stream.GetBuffer()); // Half stream = new MemoryStream(new byte[2], 0, 2, true, true); bw = new BinaryWriter(stream); actual = bw.WriteType(BitConverter.Int16BitsToHalf(0x0100)); Assert.True(actual); ValidateBytes([.. _bytes.Take(2)], stream.GetBuffer()); // Int128 stream = new MemoryStream(new byte[16], 0, 16, true, true); bw = new BinaryWriter(stream); actual = bw.WriteType((Int128)new BigInteger(_bytes)); Assert.True(actual); ValidateBytes(_bytes, stream.GetBuffer()); // UInt128 stream = new MemoryStream(new byte[16], 0, 16, true, true); bw = new BinaryWriter(stream); actual = bw.WriteType((UInt128)new BigInteger(_bytes)); Assert.True(actual); ValidateBytes(_bytes, stream.GetBuffer()); // Enum stream = new MemoryStream(new byte[4], 0, 4, true, true); bw = new BinaryWriter(stream); actual = bw.WriteType((TestEnum)0x03020100); Assert.True(actual); ValidateBytes([.. _bytes.Take(4)], stream.GetBuffer()); } [Fact] public void WriteTypeExplicitTest() { byte[] bytesWithString = [ 0x00, 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, 0x41, 0x42, 0x43, 0x00, ]; var stream = new MemoryStream(new byte[16], 0, 16, true, true); var bw = new BinaryWriter(stream); var obj = new TestStructExplicit { FirstValue = TestEnum.RecognizedTestValue, SecondValue = 0x07060504, FifthValue = "ABC", }; byte[] expected = [.. bytesWithString.Take(12)]; bool write = bw.WriteType(obj); Assert.True(write); ValidateBytes(expected, stream.GetBuffer()); } [Fact] public void WriteTypeSequentialTest() { byte[] bytesWithString = [ 0x00, 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, 0x08, 0x09, 0x0A, 0x0B, 0x41, 0x42, 0x43, 0x00, ]; var stream = new MemoryStream(new byte[24], 0, count: 24, true, true); var bw = new BinaryWriter(stream); var obj = new TestStructSequential { FirstValue = TestEnum.RecognizedTestValue, SecondValue = 0x07060504, ThirdValue = 0x0908, FourthValue = 0x0B0A, FifthValue = "ABC", }; byte[] expected = [.. bytesWithString.Take(16)]; bool write = bw.WriteType(obj); Assert.True(write); ValidateBytes(expected, stream.GetBuffer()); } /// /// Validate that a set of actual bytes matches the expected bytes /// private static void ValidateBytes(byte[] expected, byte[] actual) { for (int i = 0; i < expected.Length; i++) { Assert.Equal(expected[i], actual[i]); } } } }