1 files changed, 9 insertions, 7 deletions
diff --git a/crypto/src/crypto/io/DigestSink.cs b/crypto/src/crypto/io/DigestSink.cs
index 98307e5f7..c2a168bfe 100644
--- a/crypto/src/crypto/io/DigestSink.cs
+++ b/crypto/src/crypto/io/DigestSink.cs
@@ -19,17 +19,19 @@ namespace Org.BouncyCastle.Crypto.IO
             get { return mDigest; }
         }
 
-        public override void WriteByte(byte b)
+        public override void Write(byte[] buffer, int offset, int count)
         {
-            mDigest.Update(b);
-        }
+            Streams.ValidateBufferArguments(buffer, offset, count);
 
-        public override void Write(byte[] buf, int off, int len)
-        {
-            if (len > 0)
+            if (count > 0)
             {
-                mDigest.BlockUpdate(buf, off, len);
+                mDigest.BlockUpdate(buffer, offset, count);
             }
         }
+
+        public override void WriteByte(byte value)
+        {
+            mDigest.Update(value);
+        }
     }
 }