CryptUtils: add I/O streams

[nikiroo-utils.git] / src / be / nikiroo / utils / CryptUtils.java

diff --git a/src/be/nikiroo/utils/CryptUtils.java b/src/be/nikiroo/utils/CryptUtils.java
index d5413d7c392f6856405b3b25e0fcab989ff4de68..b82a169ac21fffd4eb6d95f6df86c46a43566bfd 100644 (file)
--- a/src/be/nikiroo/utils/CryptUtils.java
+++ b/src/be/nikiroo/utils/CryptUtils.java
@@ -1,12 +1,16 @@
 package be.nikiroo.utils;
 
 import java.io.IOException;
+import java.io.InputStream;
+import java.io.OutputStream;
 import java.io.UnsupportedEncodingException;
 import java.security.InvalidKeyException;
 import java.security.NoSuchAlgorithmException;
 
 import javax.crypto.BadPaddingException;
 import javax.crypto.Cipher;
+import javax.crypto.CipherInputStream;
+import javax.crypto.CipherOutputStream;
 import javax.crypto.IllegalBlockSizeException;
 import javax.crypto.NoSuchPaddingException;
 import javax.crypto.SecretKey;
@@ -60,6 +64,54 @@ public class CryptUtils {
                init(bytes32);
        }
 
+       /**
+        * Wrap the given {@link InputStream} so it is transparently encrypted by
+        * the current {@link CryptUtils}.
+        * 
+        * @param in
+        *            the {@link InputStream} to wrap
+        * @return the auto-encode {@link InputStream}
+        */
+       public InputStream encryptInputStream(InputStream in) {
+               return new CipherInputStream(in, ecipher);
+       }
+
+       /**
+        * Wrap the given {@link OutputStream} so it is transparently encrypted by
+        * the current {@link CryptUtils}.
+        * 
+        * @param in
+        *            the {@link OutputStream} to wrap
+        * @return the auto-encode {@link OutputStream}
+        */
+       public OutputStream encryptOutpuStream(OutputStream out) {
+               return new CipherOutputStream(out, ecipher);
+       }
+
+       /**
+        * Wrap the given {@link OutStream} so it is transparently decoded by the
+        * current {@link CryptUtils}.
+        * 
+        * @param in
+        *            the {@link InputStream} to wrap
+        * @return the auto-decode {@link InputStream}
+        */
+       public InputStream decryptInputStream(InputStream in) {
+               return new CipherInputStream(in, dcipher);
+       }
+
+       /**
+        * Wrap the given {@link OutStream} so it is transparently decoded by the
+        * current {@link CryptUtils}.
+        * 
+        * @param out
+        *            the {@link OutputStream} to wrap
+        * @return the auto-decode {@link OutputStream}
+        */
+       public OutputStream decryptOutputStream(OutputStream out) {
+               return new CipherOutputStream(out, dcipher);
+       }
+
        /**
         * This method required an array of 128 bytes.
         * 
@@ -279,6 +331,6 @@ public class CryptUtils {
         * @return a 128 bits key computed from the given input
         */
        static private byte[] key2key(String input) {
-               return StringUtils.getMd5Hash(input).getBytes();
+               return StringUtils.getMd5Hash("" + input).getBytes();
        }
 }