Merge pull request #2 from mfalkvidd/configurable-server-timeout

Add configurable server timeout
CreatorDev · Mar 20, 2019 · 9b6f7fd · 9b6f7fd
2 parents 404dd15 + b64da42
commit 9b6f7fd
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/src/DTLS.Net/ServerHandshake.cs b/src/DTLS.Net/ServerHandshake.cs
@@ -44,6 +44,8 @@ internal class ServerHandshake
 		public Org.BouncyCastle.Crypto.AsymmetricKeyParameter PrivateKey { get; set; }
 		public Version ServerVersion { get; set; }
 		public Sessions Sessions { get; set; }
+		private const int HANDSHAKE_DWELL_TIME = 10;
+		public static int HandshakeTimeout { get; set; } = 5000;
 
 
 
@@ -72,15 +74,15 @@ public void ProcessHandshake(DTLSRecord record)
             if ((session != null) && session.IsEncypted(record))
             {
                 int count = 0;
-                while ((session.Cipher == null) && (count < 50))
+                while ((session.Cipher == null) && (count < (HandshakeTimeout / HANDSHAKE_DWELL_TIME)))
                 {
-                    System.Threading.Thread.Sleep(10);
+                    System.Threading.Thread.Sleep(HANDSHAKE_DWELL_TIME);
                     count++;
                 }
 
                 if (session.Cipher == null)
-                {                    
-                    throw new Exception();
+                {
+                    throw new Exception($"HandshakeTimeout: >{HandshakeTimeout}");
                 }
 
                 if (session.Cipher != null)