Changes to QuicListener.

Author: Vect0rZ

QuicNet/Context/QuicContext.cs

@@ -0,0 +1,20 @@
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using System.Net;
+using System.Text;
+using System.Threading.Tasks;
+
+namespace QuicNet.Context
+{
+    public class QuicContext
+    {
+        public IPEndPoint EndPoint { get; set; }
+        public byte[] Data { get; set; }
+
+        public QuicContext()
+        {
+
+        }
+    }
+}

QuicNet/QuicListener.cs

@@ -1,4 +1,5 @@
-using QuicNet.Infrastructure;
+using QuicNet.Context;
+using QuicNet.Infrastructure;
 using QuicNet.Infrastructure.Packets;
 using System;
 using System.Collections.Generic;
@@ -13,43 +14,57 @@ namespace QuicNet
     public class QuicListener
     {
         private UdpClient _client;
-        private IPEndPoint _endPoint;
 
         private Unpacker _unpacker;
         private PacketCreator _packetCreator;
         private Dispatcher _dispatcher;
 
+        private int _port;
+        private bool _started;
+
+        public event Action<QuicContext> OnDataReceived;
+
         public QuicListener(int port)
         {
-            _endPoint = new IPEndPoint(IPAddress.Any, port);
-            _client = new UdpClient(port);
+            _started = false;
+            _port = port;
+            
             _unpacker = new Unpacker();
             _packetCreator = new PacketCreator();
             _dispatcher = new Dispatcher(_packetCreator);
         }
 
         public void Start()
         {
-            Receive();
+            _client = new UdpClient(_port);
+        }
+
+        public void Close()
+        {
+            _client.Close();
         }
 
         public void Receive()
         {
-            byte[] data = _client.Receive(ref _endPoint);
+            while (true)
+            {
+                IPEndPoint endpoint = new IPEndPoint(IPAddress.Any, _port);
+                byte[] data = _client.Receive(ref endpoint);
 
-            Packet packet = _unpacker.Unpack(data);
+                Packet packet = _unpacker.Unpack(data);
 
-            // Discard unknown packets
-            if (packet == null)
-                return;
+                // Discard unknown packets
+                if (packet == null)
+                    continue;
 
-            // TODO: Validate packet before dispatching
-            Packet result = _dispatcher.Dispatch(packet);
+                // TODO: Validate packet before dispatching
+                Packet result = _dispatcher.Dispatch(packet);
 
-            // Send a response packet if the dispatcher has information for the peer.
-            if (result != null)
-            {
-                
+                // Send a response packet if the dispatcher has information for the peer. (1-RTT)
+                if (result != null)
+                {
+
+                }
             }
         }
     }

QuicNet/QuicNet.csproj

@@ -43,6 +43,7 @@
     <Compile Include="Connections\ConnectionPool.cs" />
     <Compile Include="Connections\ConnectionState.cs" />
     <Compile Include="Connections\QuicConnection.cs" />
+    <Compile Include="Context\QuicContext.cs" />
     <Compile Include="Dispatcher.cs" />
     <Compile Include="QuicClient.cs" />
     <Compile Include="QuicListener.cs" />

QuickNet.Console/Program.cs

@@ -1,5 +1,6 @@
 using QuickNet.Utilities;
 using QuicNet;
+using QuicNet.Context;
 using QuicNet.Infrastructure;
 using QuicNet.Infrastructure.Frames;
 using QuicNet.Infrastructure.Packets;
@@ -45,9 +46,15 @@ static void Main(string[] args)
             StreamId streamId = streamIdData;
 
             QuicListener listener = new QuicListener(11000);
+            listener.OnDataReceived += Listener_OnDataReceived;
             listener.Start();
         }
 
+        private static void Listener_OnDataReceived(QuicContext obj)
+        {
+            throw new NotImplementedException();
+        }
+
         static string ToBase64(byte[] data)
         {
             return Convert.ToBase64String(data);