QuicListener can now receive stream data.

Author: Vect0rZ

QuicNet.Infrastructure/FrameParser.cs

@@ -26,6 +26,7 @@ public Frame GetFrame()
                 case 0x00: result = new PaddingFrame(); break;
                 case 0x02: result = new ConnectionCloseFrame(); break;
                 case 0x06: result = new MaxStreamIdFrame(); break;
+                case 0x10: result = new StreamFrame(); break;
                 default: result = null; break;
             }
 

QuicNet.Infrastructure/Packets/ShortHeaderPacket.cs

@@ -15,6 +15,7 @@ public override void Decode(byte[] packet)
         {
             ByteArray array = new ByteArray(packet);
             byte type = array.ReadByte();
+            DestinationConnectionId = array.ReadByte();
             PacketNumber = array.ReadUInt32();
 
             DecodeFrames(array);
@@ -26,6 +27,7 @@ public override byte[] Encode()
 
             List<byte> result = new List<byte>();
             result.Add(Type);
+            result.Add(DestinationConnectionId);
             result.AddRange(ByteUtilities.GetBytes(PacketNumber));
             result.AddRange(frames);
 

QuicNet/Connections/QuicConnection.cs

@@ -12,7 +12,7 @@ public class QuicConnection
     {
         private ConnectionState _state;
         private Dictionary<UInt64, QuicStream> _streams;
-        private QuicContext _context;
+        public QuicContext Context { get; private set; }
 
         public QuicConnection(UInt32 id)
         {
@@ -22,7 +22,7 @@ public QuicConnection(UInt32 id)
 
         public void AttachContext(QuicContext context)
         {
-            _context = context;
+            Context = context;
         }
 
         public void ProcessFrames(List<Frame> frames)

QuicNet/Context/QuicContext.cs

@@ -1,4 +1,6 @@
-using QuicNet.Infrastructure.Packets;
+using QuickNet.Utilities;
+using QuicNet.Infrastructure.Packets;
+using QuicNet.Streams;
 using System;
 using System.Collections.Generic;
 using System.Linq;
@@ -18,6 +20,7 @@ public class QuicContext
 
         public IPEndPoint Endpoint { get; }
         public bool IsClosed { get; private set; }
+        public event Action<byte[]> OnDataReceived;
 
         /// <summary>
         /// Internal constructor to prevent createing the context outside the scope of Quic.
@@ -30,6 +33,11 @@ internal QuicContext(UdpClient client, IPEndPoint endpoint)
             Endpoint = endpoint;
         }
 
+        internal void DataReceived(byte[] data, StreamId id)
+        {
+            OnDataReceived?.Invoke(data);
+        }
+
         /// <summary>
         /// Send data to the client.
         /// </summary>

QuicNet/QuicListener.cs

@@ -41,14 +41,15 @@ public void Start()
         {
             _client = new UdpClient(_port);
             _started = true;
+            Receive();
         }
 
         public void Close()
         {
             _client.Close();
         }
 
-        public void Receive()
+        private void Receive()
         {
             if (!_started)
                 throw new QuicListenerNotStartedException("Please call the Start() method before receving data.");

QuicNet/Streams/QuicStream.cs

@@ -27,7 +27,8 @@ public QuicStream(QuicConnection connection, StreamId streamId)
 
         public void ProcessData(StreamFrame frame)
         {
-
+            byte[] data = frame.StreamData;
+            _connection.Context.DataReceived(data, StreamId);
         }
     }
 }

QuickNet.Console/Program.cs

@@ -31,17 +31,25 @@ static void Main(string[] args)
             };
             ConnectionCloseFrame frame = new ConnectionCloseFrame(ErrorCode.SERVER_BUSY, "The server is too busy to process your request.");
             MaxStreamIdFrame msidframe = new MaxStreamIdFrame(144123, StreamType.ClientUnidirectional);
-            packet.AttachFrame(frame);
+            //packet.AttachFrame(frame);
             packet.AttachFrame(msidframe);
 
             byte[] data = packet.Encode();
+            string b64 = ToBase64(data);
+
+            byte[] shpdata = new byte[] { 1, 1, 2, 3, 5, 8, 13, 21 };
+            ShortHeaderPacket shp = new ShortHeaderPacket();
+            shp.DestinationConnectionId = 124;
+            shp.PacketNumber = 2;
+            shp.AttachFrame(new StreamFrame() { StreamId = 1, Length = new VariableInteger((UInt64)shpdata.Length), StreamData = shpdata, Offset = 0 });
+
+            string shpb64 = ToBase64(shp.Encode());
+
             packet.Decode(data);
 
             byte[] ccfData = frame.Encode();
             frame.Decode(new ByteArray(ccfData));
 
-            string b64 = ToBase64(data);
-
             byte[] streamIdData = new StreamId(123, StreamType.ClientUnidirectional);
             StreamId streamId = streamIdData;
 
@@ -52,7 +60,17 @@ static void Main(string[] args)
 
         private static void Listener_OnClientConnected(QuicContext obj)
         {
-            throw new NotImplementedException();
+            System.Console.WriteLine("Client connected.");
+            obj.OnDataReceived += Obj_OnDataReceived;
+        }
+
+        private static void Obj_OnDataReceived(byte[] obj)
+        {
+            System.Console.WriteLine("Data received");
+            foreach (byte b in obj)
+            {
+                System.Console.Write(string.Format("{0},", b));
+            }
         }
 
         static string ToBase64(byte[] data)