Prevent infinite frame proecssing.

Author: Vect0rZ

QuicNet.Infrastructure/Exceptions/ProtocolException.cs

@@ -0,0 +1,19 @@
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using System.Text;
+using System.Threading.Tasks;
+
+namespace QuicNet.Infrastructure.Exceptions
+{
+    public class ProtocolException : Exception
+    {
+        public ProtocolException()
+        {
+        }
+
+        public ProtocolException(string message) : base(message)
+        {
+        }
+    }
+}

QuicNet.Infrastructure/Packets/Packet.cs

@@ -1,5 +1,7 @@
 using QuickNet.Utilities;
+using QuicNet.Infrastructure.Exceptions;
 using QuicNet.Infrastructure.Frames;
+using QuicNet.Infrastructure.Settings;
 using System;
 using System.Collections.Generic;
 using System.Text;
@@ -33,14 +35,20 @@ public virtual void DecodeFrames(ByteArray array)
         {
             FrameParser factory = new FrameParser(array);
             Frame result;
-            while (array.HasData())
+            int frames = 0;
+            while (array.HasData() && frames <= QuicSettings.MaximumFramesPerPacket)
             {
                 result = factory.GetFrame();
                 if (result != null)
                     _frames.Add(result);
 
+                frames++;
+
                 // TODO: Possibily handle broken frames.
             }
+
+            if (array.HasData())
+                throw new ProtocolException("Unexpected number of frames or possibly corrupted frame was sent.");
         }
 
         public virtual byte[] EncodeFrames()

QuicNet.Infrastructure/Settings/QuicSettings.cs

@@ -36,5 +36,10 @@ public class QuicSettings
         /// Should the server buffer packets that came before the initial packet?
         /// </summary>
         public const bool ShouldBufferPacketsBeforeConnection = false;
+
+        /// <summary>
+        /// Limit the maximum number of frames a packet can carry.
+        /// </summary>
+        public const int MaximumFramesPerPacket = 10;
     }
 }