PasswordBasedEncryptionTest.java

   1 package org.argeo.util.crypto;
   2
   3 import java.io.ByteArrayInputStream;
   4 import java.io.ByteArrayOutputStream;
   5 import java.io.InputStream;
   6 import java.security.AlgorithmParameters;
   7
   8 import javax.crypto.Cipher;
   9 import javax.crypto.CipherInputStream;
  10 import javax.crypto.CipherOutputStream;
  11 import javax.crypto.SecretKey;
  12 import javax.crypto.SecretKeyFactory;
  13 import javax.crypto.spec.IvParameterSpec;
  14 import javax.crypto.spec.PBEKeySpec;
  15 import javax.crypto.spec.PBEParameterSpec;
  16 import javax.crypto.spec.SecretKeySpec;
  17
  18 import junit.framework.TestCase;
  19
  20 import org.argeo.StreamUtils;
  21 import org.argeo.util.crypto.PasswordBasedEncryption;
  22
  23 public class PasswordBasedEncryptionTest{// extends TestCase {
  24 //      public void testEncryptDecrypt() {
  25 //              final String password = "test long password since they are more powerful";
  26 //              PasswordBasedEncryption pbeEnc = new PasswordBasedEncryption(
  27 //                              password.toCharArray());
  28 //              String message = "Hello World!";
  29 //              byte[] encrypted = pbeEnc.encryptString(message);
  30 //              // System.out.println("Encrypted: '" + new String(encrypted) + "'");
  31 //              PasswordBasedEncryption pbeDec = new PasswordBasedEncryption(
  32 //                              password.toCharArray());
  33 //              InputStream in = null;
  34 //              in = new ByteArrayInputStream(encrypted);
  35 //              String decrypted = pbeDec.decryptAsString(in);
  36 //              // System.out.println("Decrypted: '" + decrypted + "'");
  37 //              StreamUtils.closeQuietly(in);
  38 //              assertEquals(message, decrypted);
  39 //      }
  40 //
  41 //      public void testPBEWithMD5AndDES() throws Exception {
  42 //              String password = "test";
  43 //              String message = "Hello World!";
  44 //
  45 //              byte[] salt = { (byte) 0xc7, (byte) 0x73, (byte) 0x21, (byte) 0x8c,
  46 //                              (byte) 0x7e, (byte) 0xc8, (byte) 0xee, (byte) 0x99 };
  47 //
  48 //              int count = 1024;
  49 //
  50 //              String cipherAlgorithm = "PBEWithMD5AndDES";
  51 //              String secretKeyAlgorithm = "PBEWithMD5AndDES";
  52 //              SecretKeyFactory keyFac = SecretKeyFactory
  53 //                              .getInstance(secretKeyAlgorithm);
  54 //              PBEKeySpec pbeKeySpec = new PBEKeySpec(password.toCharArray());
  55 //              PBEParameterSpec pbeParamSpec = new PBEParameterSpec(salt, count);
  56 //              SecretKey pbeKey = keyFac.generateSecret(pbeKeySpec);
  57 //              Cipher ecipher = Cipher.getInstance(cipherAlgorithm);
  58 //              ecipher.init(Cipher.ENCRYPT_MODE, pbeKey, pbeParamSpec);
  59 //              Cipher dcipher = Cipher.getInstance(cipherAlgorithm);
  60 //              dcipher.init(Cipher.DECRYPT_MODE, pbeKey, pbeParamSpec);
  61 //
  62 //              byte[] encrypted = ecipher.doFinal(message.getBytes());
  63 //              byte[] decrypted = dcipher.doFinal(encrypted);
  64 //              assertEquals(message, new String(decrypted));
  65 //
  66 //      }
  67 //
  68 //      public void testPBEWithSHA1AndAES() throws Exception {
  69 //              String password = "test";
  70 //              String message = "Hello World!";
  71 //
  72 //              byte[] salt = { (byte) 0xc7, (byte) 0x73, (byte) 0x21, (byte) 0x8c,
  73 //                              (byte) 0x7e, (byte) 0xc8, (byte) 0xee, (byte) 0x99 };
  74 //              byte[] iv = { (byte) 0xc7, (byte) 0x73, (byte) 0x21, (byte) 0x8c,
  75 //                              (byte) 0x7e, (byte) 0xc8, (byte) 0xee, (byte) 0x99,
  76 //                              (byte) 0xc7, (byte) 0x73, (byte) 0x21, (byte) 0x8c,
  77 //                              (byte) 0x7e, (byte) 0xc8, (byte) 0xee, (byte) 0x99 };
  78 //
  79 //              int count = 1024;
  80 //              // int keyLength = 256;
  81 //              int keyLength = 128;
  82 //
  83 //              String cipherAlgorithm = "AES/CBC/PKCS5Padding";
  84 //              String secretKeyAlgorithm = "PBKDF2WithHmacSHA1";
  85 //              SecretKeyFactory keyFac = SecretKeyFactory
  86 //                              .getInstance(secretKeyAlgorithm);
  87 //              PBEKeySpec pbeKeySpec = new PBEKeySpec(password.toCharArray(), salt,
  88 //                              count, keyLength);
  89 //              SecretKey tmp = keyFac.generateSecret(pbeKeySpec);
  90 //              SecretKey secret = new SecretKeySpec(tmp.getEncoded(), "AES");
  91 //              Cipher ecipher = Cipher.getInstance(cipherAlgorithm);
  92 //              ecipher.init(Cipher.ENCRYPT_MODE, secret, new IvParameterSpec(iv));
  93 //
  94 //              // decrypt
  95 //              keyFac = SecretKeyFactory.getInstance(secretKeyAlgorithm);
  96 //              pbeKeySpec = new PBEKeySpec(password.toCharArray(), salt, count,
  97 //                              keyLength);
  98 //              tmp = keyFac.generateSecret(pbeKeySpec);
  99 //              secret = new SecretKeySpec(tmp.getEncoded(), "AES");
 100 //              // AlgorithmParameters params = ecipher.getParameters();
 101 //              // byte[] iv = params.getParameterSpec(IvParameterSpec.class).getIV();
 102 //              Cipher dcipher = Cipher.getInstance(cipherAlgorithm);
 103 //              dcipher.init(Cipher.DECRYPT_MODE, secret, new IvParameterSpec(iv));
 104 //
 105 //              byte[] encrypted = ecipher.doFinal(message.getBytes());
 106 //              byte[] decrypted = dcipher.doFinal(encrypted);
 107 //              assertEquals(message, new String(decrypted));
 108 //
 109 //              ByteArrayOutputStream out = new ByteArrayOutputStream();
 110 //              CipherOutputStream cipherOut = new CipherOutputStream(out, ecipher);
 111 //              cipherOut.write(message.getBytes());
 112 //              StreamUtils.closeQuietly(cipherOut);
 113 //              byte[] enc = out.toByteArray();
 114 //
 115 //              ByteArrayInputStream in = new ByteArrayInputStream(enc);
 116 //              CipherInputStream cipherIn = new CipherInputStream(in, dcipher);
 117 //              ByteArrayOutputStream dec = new ByteArrayOutputStream();
 118 //              StreamUtils.copy(cipherIn, dec);
 119 //              assertEquals(message, new String(dec.toByteArray()));
 120 //      }
 121 }