1 files changed, 42 insertions, 0 deletions
diff --git a/core/src/main/java/org/spongycastle/crypto/io/DigestOutputStream.java b/core/src/main/java/org/spongycastle/crypto/io/DigestOutputStream.java
new file mode 100644
index 00000000..d280b514
--- /dev/null
+++ b/core/src/main/java/org/spongycastle/crypto/io/DigestOutputStream.java
@@ -0,0 +1,42 @@
+package org.spongycastle.crypto.io;
+
+import java.io.IOException;
+import java.io.OutputStream;
+
+import org.spongycastle.crypto.Digest;
+
+public class DigestOutputStream
+    extends OutputStream
+{
+    protected Digest digest;
+
+    public DigestOutputStream(
+        Digest          Digest)
+    {
+        this.digest = Digest;
+    }
+
+    public void write(int b)
+        throws IOException
+    {
+        digest.update((byte)b);
+    }
+
+    public void write(
+        byte[] b,
+        int off,
+        int len)
+        throws IOException
+    {
+        digest.update(b, off, len);
+    }
+
+    public byte[] getDigest()
+    {
+        byte[] res = new byte[digest.getDigestSize()];
+        
+        digest.doFinal(res, 0);
+        
+        return res;
+    }
+}