1 files changed, 44 insertions, 0 deletions
diff --git a/crypto/src/tls/DefaultTlsHeartbeat.cs b/crypto/src/tls/DefaultTlsHeartbeat.cs
new file mode 100644
index 000000000..9090d2cc0
--- /dev/null
+++ b/crypto/src/tls/DefaultTlsHeartbeat.cs
@@ -0,0 +1,44 @@
+using System;
+
+using Org.BouncyCastle.Crypto.Utilities;
+
+namespace Org.BouncyCastle.Tls
+{
+    public class DefaultTlsHeartbeat
+        : TlsHeartbeat
+    {
+        private readonly int idleMillis, timeoutMillis;
+
+        private uint counter = 0U;
+
+        public DefaultTlsHeartbeat(int idleMillis, int timeoutMillis)
+        {
+            if (idleMillis <= 0)
+                throw new ArgumentException("must be > 0", "idleMillis");
+            if (timeoutMillis <= 0)
+                throw new ArgumentException("must be > 0", "timeoutMillis");
+
+            this.idleMillis = idleMillis;
+            this.timeoutMillis = timeoutMillis;
+        }
+
+        public virtual byte[] GeneratePayload()
+        {
+            lock (this)
+            {
+                // NOTE: The counter naturally wraps back to 0
+                return Pack.UInt32_To_BE(++counter);
+            }
+        }
+
+        public virtual int IdleMillis
+        {
+            get { return idleMillis; }
+        }
+
+        public virtual int TimeoutMillis
+        {
+            get { return timeoutMillis; }
+        }
+    }
+}