| @@ -12,21 +12,23 @@ namespace Discord.ETF | |||
| { | |||
| public unsafe class ETFWriter : IDisposable | |||
| { | |||
| private readonly static byte[] _nilBytes = new byte[] { (byte)ETFType.SMALL_ATOM_EXT, 3, (byte)'n', (byte)'i', (byte)'l' }; | |||
| private readonly static byte[] _falseBytes = new byte[] { (byte)ETFType.SMALL_ATOM_EXT, 5, (byte)'f', (byte)'a', (byte)'l', (byte)'s', (byte)'e' }; | |||
| private readonly static byte[] _trueBytes = new byte[] { (byte)ETFType.SMALL_ATOM_EXT, 4, (byte)'t', (byte)'r', (byte)'u', (byte)'e' }; | |||
| private static readonly ConcurrentDictionary<Type, Delegate> _serializers = new ConcurrentDictionary<Type, Delegate>(); | |||
| private static readonly ConcurrentDictionary<Type, Delegate> _indirectSerializers = new ConcurrentDictionary<Type, Delegate>(); | |||
| private readonly static MethodInfo _writeTMethod = GetGenericWriteMethod(null); | |||
| private readonly static MethodInfo _writeNullableTMethod = GetGenericWriteMethod(typeof(Nullable<>)); | |||
| private readonly static MethodInfo _writeDictionaryTMethod = GetGenericWriteMethod(typeof(IDictionary<,>)); | |||
| private readonly static MethodInfo _writeEnumerableTMethod = GetGenericWriteMethod(typeof(IEnumerable<>)); | |||
| private readonly static DateTime _epochTime = new DateTime(1970, 1, 1, 0, 0, 0, DateTimeKind.Utc); | |||
| private static readonly byte[] _nilBytes = new byte[] { (byte)ETFType.SMALL_ATOM_EXT, 3, (byte)'n', (byte)'i', (byte)'l' }; | |||
| private static readonly byte[] _falseBytes = new byte[] { (byte)ETFType.SMALL_ATOM_EXT, 5, (byte)'f', (byte)'a', (byte)'l', (byte)'s', (byte)'e' }; | |||
| private static readonly byte[] _trueBytes = new byte[] { (byte)ETFType.SMALL_ATOM_EXT, 4, (byte)'t', (byte)'r', (byte)'u', (byte)'e' }; | |||
| private static readonly MethodInfo _writeTMethod = GetGenericWriteMethod(null); | |||
| private static readonly MethodInfo _writeNullableTMethod = GetGenericWriteMethod(typeof(Nullable<>)); | |||
| private static readonly MethodInfo _writeDictionaryTMethod = GetGenericWriteMethod(typeof(IDictionary<,>)); | |||
| private static readonly MethodInfo _writeEnumerableTMethod = GetGenericWriteMethod(typeof(IEnumerable<>)); | |||
| private static readonly DateTime _epochTime = new DateTime(1970, 1, 1, 0, 0, 0, DateTimeKind.Utc); | |||
| private readonly Stream _stream; | |||
| private readonly byte[] _buffer; | |||
| private readonly bool _leaveOpen; | |||
| private readonly Encoding _encoding; | |||
| private readonly ConcurrentDictionary<Type, Delegate> _serializers, _indirectSerializers; | |||
| public virtual Stream BaseStream | |||
| { | |||
| @@ -45,8 +47,6 @@ namespace Discord.ETF | |||
| _leaveOpen = leaveOpen; | |||
| _buffer = new byte[11]; | |||
| _encoding = Encoding.UTF8; | |||
| _serializers = new ConcurrentDictionary<Type, Delegate>(); | |||
| _indirectSerializers = new ConcurrentDictionary<Type, Delegate>(); | |||
| } | |||
| public void Write(bool value) | |||
| @@ -72,6 +72,7 @@ namespace Discord.ETF | |||
| } | |||
| else if (value >= int.MinValue && value <= int.MaxValue) | |||
| { | |||
| //TODO: Does this encode negatives correctly? | |||
| _buffer[0] = (byte)ETFType.INTEGER_EXT; | |||
| _buffer[1] = (byte)(value >> 24); | |||
| _buffer[2] = (byte)(value >> 16); | |||
| @@ -145,7 +146,7 @@ namespace Discord.ETF | |||
| public void Write(DateTime value) => Write((ulong)((value.Ticks - _epochTime.Ticks) / TimeSpan.TicksPerSecond)); | |||
| public void Write(bool? value) { if (value.HasValue) Write((bool)value.Value); else WriteNil(); } | |||
| public void Write(bool? value) { if (value.HasValue) Write(value.Value); else WriteNil(); } | |||
| public void Write(sbyte? value) { if (value.HasValue) Write((long)value.Value); else WriteNil(); } | |||
| public void Write(byte? value) { if (value.HasValue) Write((ulong)value.Value); else WriteNil(); } | |||
| public void Write(short? value) { if (value.HasValue) Write((long)value.Value); else WriteNil(); } | |||
| @@ -158,35 +159,35 @@ namespace Discord.ETF | |||
| public void Write(float? value) { if (value.HasValue) Write((double)value.Value); else WriteNil(); } | |||
| public void Write(DateTime? value) { if (value.HasValue) Write(value.Value); else WriteNil(); } | |||
| public void Write(byte[] value) | |||
| public void Write(string value) | |||
| { | |||
| if (value != null) | |||
| { | |||
| int count = value.Length; | |||
| var bytes = _encoding.GetBytes(value); | |||
| int count = bytes.Length; | |||
| _buffer[0] = (byte)ETFType.BINARY_EXT; | |||
| _buffer[1] = (byte)(count >> 24); | |||
| _buffer[2] = (byte)(count >> 16); | |||
| _buffer[3] = (byte)(count >> 8); | |||
| _buffer[4] = (byte)count; | |||
| _stream.Write(_buffer, 0, 5); | |||
| _stream.Write(value, 0, value.Length); | |||
| _stream.Write(bytes, 0, bytes.Length); | |||
| } | |||
| else | |||
| WriteNil(); | |||
| } | |||
| public void Write(string value) | |||
| public void Write(byte[] value) | |||
| { | |||
| if (value != null) | |||
| { | |||
| var bytes = _encoding.GetBytes(value); | |||
| int count = bytes.Length; | |||
| int count = value.Length; | |||
| _buffer[0] = (byte)ETFType.BINARY_EXT; | |||
| _buffer[1] = (byte)(count >> 24); | |||
| _buffer[2] = (byte)(count >> 16); | |||
| _buffer[3] = (byte)(count >> 8); | |||
| _buffer[4] = (byte)count; | |||
| _stream.Write(_buffer, 0, 5); | |||
| _stream.Write(bytes, 0, bytes.Length); | |||
| _stream.Write(value, 0, value.Length); | |||
| } | |||
| else | |||
| WriteNil(); | |||
| @@ -269,7 +270,7 @@ namespace Discord.ETF | |||
| public virtual long Seek(int offset, SeekOrigin origin) => _stream.Seek(offset, origin); | |||
| #region Emit | |||
| private Action<ETFWriter, T> CreateSerializer<T>(Type type, TypeInfo typeInfo, bool isDirect) | |||
| private static Action<ETFWriter, T> CreateSerializer<T>(Type type, TypeInfo typeInfo, bool isDirect) | |||
| { | |||
| var method = new DynamicMethod(isDirect ? "SerializeETF" : "SerializeIndirectETF", | |||
| null, new[] { typeof(ETFWriter), isDirect ? type : typeof(object) }, true); | |||
| @@ -291,7 +292,7 @@ namespace Discord.ETF | |||
| generator.Emit(OpCodes.Ret); | |||
| return method.CreateDelegate(typeof(Action<ETFWriter, T>)) as Action<ETFWriter, T>; | |||
| } | |||
| private void EmitWriteValue(ILGenerator generator, Type type, TypeInfo typeInfo, bool isTop) | |||
| private static void EmitWriteValue(ILGenerator generator, Type type, TypeInfo typeInfo, bool isTop) | |||
| { | |||
| //Convert enum types to their base type | |||
| if (typeInfo.IsEnum) | |||
| @@ -335,18 +336,21 @@ namespace Discord.ETF | |||
| //Dictionaries | |||
| else if (!typeInfo.IsValueType && typeInfo.ImplementedInterfaces | |||
| .Any(x => x.IsConstructedGenericType && x.GetGenericTypeDefinition() == typeof(IDictionary<,>))) | |||
| { | |||
| generator.EmitCall(OpCodes.Call, _writeDictionaryTMethod.MakeGenericMethod(typeInfo.GenericTypeParameters), null); | |||
| } | |||
| //Enumerable | |||
| else if (!typeInfo.IsValueType && typeInfo.ImplementedInterfaces | |||
| .Any(x => x.IsConstructedGenericType && x.GetGenericTypeDefinition() == typeof(IEnumerable<>))) | |||
| { | |||
| generator.EmitCall(OpCodes.Call, _writeEnumerableTMethod.MakeGenericMethod(typeInfo.GenericTypeParameters), null); | |||
| } | |||
| //Nullable Structs | |||
| else if (typeInfo.IsGenericType && typeInfo.GetGenericTypeDefinition() == typeof(Nullable<>) && | |||
| else if (typeInfo.IsGenericType && typeInfo.GetGenericTypeDefinition() == typeof(Nullable<>) && | |||
| typeInfo.GenericTypeParameters[0].GetTypeInfo().IsValueType) | |||
| { | |||
| generator.EmitCall(OpCodes.Call, _writeNullableTMethod.MakeGenericMethod(typeInfo.GenericTypeParameters), null); | |||
| } | |||
| //Structs/Classes | |||
| else if (typeInfo.IsClass || (typeInfo.IsValueType && !typeInfo.IsPrimitive)) | |||
| { | |||
| @@ -387,7 +391,6 @@ namespace Discord.ETF | |||
| generator.EmitCall(OpCodes.Call, _writeTMethod.MakeGenericMethod(typeInfo.GenericTypeParameters), null); | |||
| } | |||
| } | |||
| //Unsupported (decimal, char) | |||
| else | |||
| throw new InvalidOperationException($"Serializing {type.Name} is not supported."); | |||
| @@ -429,7 +432,7 @@ namespace Discord.ETF | |||
| { | |||
| return typeof(ETFWriter).GetTypeInfo().GetDeclaredMethods(nameof(Write)) | |||
| .Where(x => x.GetParameters()[0].ParameterType == paramType) | |||
| .FirstOrDefault(); | |||
| .Single(); | |||
| } | |||
| private static MethodInfo GetGenericWriteMethod(Type genericType) | |||
| { | |||