Put in RPK and callbacks for supporting certificates (#48)

Setup TLS to have callbacks so that one can look at things like ACE tokens being passed as the PSK identifier.
Com-AugustCellars · Jun 16, 2018 · ddea0e2 · ddea0e2
1 parent 727780f
commit ddea0e2
Show file tree

Hide file tree

Showing 11 changed files with 872 additions and 83 deletions.
diff --git a/CoAP.NET/CoAP.NET45.csproj b/CoAP.NET/CoAP.NET45.csproj
@@ -1,4 +1,4 @@
-<?xml version="1.0" encoding="utf-8"?>
+<?xml version="1.0" encoding="utf-8"?>
 <Project ToolsVersion="12.0" DefaultTargets="Build" xmlns="http://schemas.microsoft.com/developer/msbuild/2003">
   <PropertyGroup>
     <Configuration Condition=" '$(Configuration)' == '' ">Debug</Configuration>
@@ -109,6 +109,7 @@
     <Compile Include="DTLS\DtlsServer.cs" />
     <Compile Include="DTLS\DTLSSession.cs" />
     <Compile Include="DTLS\QueueItem.cs" />
+    <Compile Include="DTLS\TlsEvent.cs" />
     <Compile Include="EmptyMessage.cs" />
     <Compile Include="EndPoint\Resources\RemoteResource.cs" />
     <Compile Include="EndPoint\Resources\Resource.cs" />
@@ -205,4 +206,4 @@
   <Target Name="AfterBuild">
   </Target>
   -->
-</Project>
+</Project>
diff --git a/CoAP.NET/DTLS/DTLSChannel.cs b/CoAP.NET/DTLS/DTLSChannel.cs
@@ -27,6 +27,7 @@ public class DTLSChannel : IChannel
         private UDPChannel _udpChannel;
         private KeySet _serverKeys;
         private KeySet _userKeys;
+
 
         public DTLSChannel(KeySet serverKeys, KeySet userKeys) : this(serverKeys, userKeys, 0)
         {
@@ -86,6 +87,8 @@ public Int32 ReceivePacketSize {
 
         private Int32 _running;
 
+        public EventHandler<TlsEvent> TlsEventHandler;
+
         public void Start()
         {
             if (System.Threading.Interlocked.CompareExchange(ref _running, 1, 0) > 0) {
@@ -164,6 +167,7 @@ public ISession GetSession(System.Net.EndPoint ep)
 
                 session = new DTLSSession(ipEndPoint, DataReceived, _serverKeys, _userKeys);
                 AddSession(session);
+                session.TlsEventHandler += MyTlsEventHandler;
 
 
                 session.Connect(_udpChannel);
@@ -175,6 +179,14 @@ public ISession GetSession(System.Net.EndPoint ep)
             return session;
         }
 
+        private void MyTlsEventHandler(Object o, TlsEvent e)
+        {
+            EventHandler<TlsEvent> handler = TlsEventHandler;
+            if (handler != null) {
+                handler(o, e);
+            }
+        }
+
         public void Send(byte[] data, ISession sessionReceive, System.Net.EndPoint ep)
         {
             try {
@@ -184,6 +196,7 @@ public void Send(byte[] data, ISession sessionReceive, System.Net.EndPoint ep)
                 if (session == null) {
 
                     session = new DTLSSession(ipEP, DataReceived, _serverKeys, _userKeys);
+                    session.TlsEventHandler += MyTlsEventHandler;
                     AddSession(session);
                     session.Connect(_udpChannel);
                 }
@@ -211,6 +224,7 @@ private void ReceiveData(Object sender, DataReceivedEventArgs e)
                 }
 
                 DTLSSession sessionNew = new DTLSSession((IPEndPoint) e.EndPoint, DataReceived, _serverKeys, _userKeys);
+                sessionNew.TlsEventHandler = MyTlsEventHandler;
                 _sessionList.Add(sessionNew);
                 new Thread(() => Accept(sessionNew, e.Data)).Start();
             }