MyBank

JCO

JWAA

Java+

ALE

Academia

Mideast

Cox

SecretKeyAESTest.java

package edu.virtualschool.jco.test;

import java.security.spec.InvalidKeySpecException;

import junit.framework.TestCase;
import edu.virtualschool.jco.JCOGenericBytes;
import edu.virtualschool.jco.JCOFault;
import edu.virtualschool.jco.JCOSealedBytes;
import edu.virtualschool.jco.JCOSecretKeyAES;
import edu.virtualschool.jco.JCOSymmetricKey;
import edu.virtualschool.jwaa.IntervalTimer;
import edu.virtualschool.jwaa.RuntimeFault;

public class SecretKeyAESTest extends TestCase
{
  private final static byte[] key16 = "1234567890123456".getBytes();
  private final static byte[] key24 = "123456789012345678901234".getBytes();
  private final static byte[] key32 = "12345678901234567890123456789012".getBytes();
  
  public static void main(String[] args)
  {
    junit.textui.TestRunner.run(SecretKeyAESTest.class);
  }
  public SecretKeyAESTest(String arg0)
  {
    super(arg0);
  }
  final public void testSecretKey() throws JCOFault
  {
    checkKey(new JCOSecretKeyAES());
  }
  final public void testEncoding() throws InvalidKeySpecException
  {
    JCOSecretKeyAES key = new JCOSecretKeyAES();
    assertEquals(key.getFormat(), "RAW");
    assertEquals( "AES", key.getAlgorithm());
    
    JCOSymmetricKey originalKey = new JCOSecretKeyAES(key16);
    JCOSymmetricKey newKey = new JCOSecretKeyAES(originalKey.getEncoded());
    assertEquals(originalKey.encode(), newKey.encode());
    
    originalKey = new JCOSecretKeyAES(key24);
    newKey = new JCOSecretKeyAES(originalKey.getEncoded());
    assertEquals(originalKey.encode(), newKey.encode());
    
    originalKey = new JCOSecretKeyAES(key32);
    newKey = new JCOSecretKeyAES(originalKey.getEncoded());
    assertEquals(originalKey.encode(), newKey.encode());
  }
  final public void testEqualsObject()
    throws InvalidKeySpecException
  {
    JCOSymmetricKey key1 = new JCOSecretKeyAES(key16);
    JCOSymmetricKey key2 = new JCOSecretKeyAES(key16);
    assertEquals(key1.encode(), key2.encode());
    
    key1 = new JCOSecretKeyAES(key24);
    key2 = new JCOSecretKeyAES(key24);
    assertEquals(key1.encode(), key2.encode());
    
    key1 = new JCOSecretKeyAES(key32);
    key2 = new JCOSecretKeyAES(key32);
    assertEquals(key1.encode(), key2.encode());
  }
  final public void testEncryptDecryptObjects()
  throws JCOFault
  {
    String object = "test string";
    IntervalTimer t = new IntervalTimer();
    JCOSecretKeyAES key = new JCOSecretKeyAES(16);
    System.out.println(t + " JCOSecretKeyAES: new JCOSecretKeyAES(16)");
    byte[] ciphertext = key.encrypt(object);
    System.out.println(t + " JCOSecretKeyAES.encrypt(object)");
    Object newObject = key.decrypt(ciphertext);
    System.out.println(t + " JCOSecretKeyAES.decrypt(ciphertext)");
    assertEquals(object, newObject);
    
    key = new JCOSecretKeyAES(32);
    System.out.println(t + " new JCOSecretKeyAES(24)");
    ciphertext = key.encrypt(object);
    System.out.println(t + " JCOSecretKeyAES.encrypt(object)");
    newObject = key.decrypt(ciphertext);
    System.out.println(t + " JCOSecretKeyAES.decrypt(ciphertext)");
    assertEquals(object, newObject);
    
    key = new JCOSecretKeyAES(32);
    System.out.println(t + " new JCOSecretKeyAES(32)");
    ciphertext = key.encrypt(object);
    System.out.println(t + " JCOSecretKeyAES.encrypt(object)");
    newObject = key.decrypt(ciphertext);
    System.out.println(t + " JCOSecretKeyAES.decrypt(ciphertext)");
    assertEquals(object, newObject);
  }
  final public void testEncryptDecryptBytes()
  throws JCOFault
  {
    String object = "test string";
    JCOSecretKeyAES key = new JCOSecretKeyAES();
    byte[] input = object.getBytes();
    byte[] ciphertext = key.encryptBytes(input);
    byte[] cleartext = key.decryptBytes(ciphertext);
    assertTrue(JCOGenericBytes.isEqualBytes(input, cleartext));
  }
  final public void testValidLengths() throws Exception
  {
//    throw new IllegalArgumentException("Key length not 128/192/256 bits.");
    int[] validLengths = new int[] { 16, 24, 32 };
    for (int i = 0; i < validLengths.length; i++)
      checkKey(new JCOSecretKeyAES(validLengths[i]));
  }
  final public void testInvalidLengths() throws Exception
  {
      int[] invalidLengths = new int[] { -1, 0, 15, 23, 31, 33 };
      for (int i = 0; i < invalidLengths.length; i++)
      {
        try
        {
          checkKey(new JCOSecretKeyAES(invalidLengths[i]));
          assertTrue(invalidLengths[i] + " should not be accepted as valid", false);
        }
        catch (NegativeArraySizeException e)
        {
          continue;
        }
        catch (RuntimeFault e)
        {
          continue;
        }
      }
  }

  final static void checkKey(JCOSymmetricKey key)
  throws JCOFault
  {
    String input = "some cleartext";
    JCOSealedBytes ciphertext = key.seal(input);
    Object cleartext = key.unseal(ciphertext);
    assertEquals(cleartext, input);
  }
}