Merge proces, enable signing of the assembly.

2026-10-18 02:13:00 +08:00 · 2017-11-25 19:55:26 +01:00
parent ebfc6aab6b
commit f9d67e2cd2
153 changed files with 323 additions and 383 deletions
@@ -0,0 +1,109 @@
+using System;
+using System.Collections.Generic;
+using System.IO;
+using System.Text;
+using System.Threading;
+using System.Threading.Tasks;
+using MQTTnet.Adapter;
+using MQTTnet.Exceptions;
+using MQTTnet.Packets;
+using MQTTnet.Protocol;
+
+namespace MQTTnet.Serializer
+{
+    public sealed class MqttPacketReader : BinaryReader
+    {
+        private readonly ReceivedMqttPacket _receivedMqttPacket;
+
+        public MqttPacketReader(ReceivedMqttPacket receivedMqttPacket)
+            : base(receivedMqttPacket.Body, Encoding.UTF8, true)
+        {
+            _receivedMqttPacket = receivedMqttPacket;
+        }
+
+        public bool EndOfRemainingData => BaseStream.Position == _receivedMqttPacket.Header.BodyLength;
+
+        public static MqttPacketHeader ReadHeaderFromSource(Stream stream, CancellationToken cancellationToken)
+        {
+            var buffer = stream.ReadByte();
+            if (buffer == -1)
+            {
+                return null;
+            }
+
+            var fixedHeader = (byte)buffer;
+            var controlPacketType = (MqttControlPacketType)(fixedHeader >> 4);
+            var bodyLength = ReadBodyLengthFromSource(stream, cancellationToken);
+
+            return new MqttPacketHeader
+            {
+                FixedHeader = fixedHeader,
+                ControlPacketType = controlPacketType,
+                BodyLength = bodyLength
+            };
+        }
+
+        public override ushort ReadUInt16()
+        {
+            var buffer = ReadBytes(2);
+
+            var temp = buffer[0];
+            buffer[0] = buffer[1];
+            buffer[1] = temp;
+
+            return BitConverter.ToUInt16(buffer, 0);
+        }
+
+        public string ReadStringWithLengthPrefix()
+        {
+            var buffer = ReadWithLengthPrefix();
+            return Encoding.UTF8.GetString(buffer, 0, buffer.Length);
+        }
+
+        public byte[] ReadWithLengthPrefix()
+        {
+            var length = ReadUInt16();
+            return ReadBytes(length);
+        }
+
+        public byte[] ReadRemainingData()
+        {
+            return ReadBytes(_receivedMqttPacket.Header.BodyLength - (int)BaseStream.Position);
+        }
+
+        private static int ReadBodyLengthFromSource(Stream stream, CancellationToken cancellationToken)
+        {
+            // Alorithm taken from http://docs.oasis-open.org/mqtt/mqtt/v3.1.1/os/mqtt-v3.1.1-os.html.
+            var multiplier = 1;
+            var value = 0;
+            byte encodedByte;
+
+            var readBytes = new List<int>();
+            do
+            {
+                if (cancellationToken.IsCancellationRequested)
+                {
+                    throw new TaskCanceledException();
+                }
+
+                var buffer = stream.ReadByte();
+                readBytes.Add(buffer);
+
+                if (buffer == -1)
+                {
+                    throw new MqttCommunicationException("Connection closed while reading remaining length data.");
+                }
+
+                encodedByte = (byte)buffer;
+                value += (byte)(encodedByte & 127) * multiplier;
+                multiplier *= 128;
+                if (multiplier > 128 * 128 * 128)
+                {
+                    throw new MqttProtocolViolationException($"Remaining length is invalid (Data={string.Join(",", readBytes)}).");
+                }
+            } while ((encodedByte & 128) != 0);
+
+            return value;
+        }
+    }
+}