MyBank

JCO

JWAA

Java+

ALE

Academia

Mideast

Cox

KeyPairTest.java

/*
 * Created on Jun 19, 2003
 *
 * To change the template for this generated file go to
 * Window>Preferences>Java>Code Generation>Code and Comments
 */
package edu.virtualschool.jco.test;

import junit.framework.TestCase;
import edu.virtualschool.jco.JCOFault;
import edu.virtualschool.jco.JCOKey;
import edu.virtualschool.jco.JCOKeyPair;
import edu.virtualschool.jco.JCOPrivateKey;
import edu.virtualschool.jco.JCOPublicKey;
import edu.virtualschool.jco.JCOSignedBytes;
import edu.virtualschool.jwaa.IntervalTimer;

public class KeyPairTest extends TestCase
{
  final JCOKeyPair keypair;
    
  public static void main(String[] args)
  {
    junit.textui.TestRunner.run(KeyPairTest.class);
  }
  public KeyPairTest(String arg0)
  {
    super(arg0);
    this.keypair = new JCOKeyPair();
  }
  public void testKeyPair() 
    throws JCOFault
  {
    int[] validLengths = new int[] { 40/8, 1026/8, 2048/8 };
    for (int j = 0; j < validLengths.length; j++)
    {
      int i = validLengths[j];
//      JCOKeyPair pair = null;
      try
      {
        checkKeyOfLength(i);
      }
      catch (JCOFault e)
      {
        System.out.println("key length error:"+i);
      }
    }
//    int[] invalidLengths = new int[] { (40-1)/8, (2048+1)/8, 0 };
//    for (int j = 0; j < 10; j++)
//    {
//      int length = invalidLengths[j];
//      try
//      {
//        JCOKeyPair pair1 = new JCOKeyPair(length);
//        assertTrue("keylength " + length + " should not be accepted", false);
//      }
//      catch (RuntimeFault e) {}
//    }
//  }
  }
  private static void checkKeyOfLength(int length) 
    throws JCOFault
  {
    String object = "123456789012345678901234567890";
    
    IntervalTimer t = new IntervalTimer();
    JCOKeyPair pair = new JCOKeyPair(length);
    System.out.println(t + "  new JCOKeyPair("+length+")");
    JCOPrivateKey privateKey = pair.getPrivate();
    JCOPublicKey publicKey = pair.getPublic();

    JCOSignedBytes signedObject = privateKey.sign(object);
    System.out.println(t + "  JCOPrivateKey.sign");
    Object unsignedObject = publicKey.verify(signedObject);
    System.out.println(t + "  JCOPublicKey.verify");
    assertEquals(object, unsignedObject);
  }
  public void testKeyEncodeDecode()
    throws Exception
  {
    checkEncodeDecode(keypair.getPublic());
    checkEncodeDecode(keypair.getPrivate());
  }
  private void checkEncodeDecode(JCOKey key)
    throws Exception
  {
    byte[] keyBytes = key.getEncoded();
    JCOKey decodedKey = null;
    if (key instanceof JCOPublicKey)
      decodedKey = new JCOPublicKey(keyBytes);
    else if (key instanceof JCOPrivateKey)
      decodedKey = new JCOPrivateKey(keyBytes);
    else 
      throw new Exception("unexpected key type:"+key);
    assertEquals(key.encode(), decodedKey.encode());
  }
  public void testSignature() throws Exception
  {
    JCOPrivateKey privateKey = keypair.getPrivate();
    JCOPublicKey publicKey = keypair.getPublic();

    String originalObject = "foobar";
    IntervalTimer t = new IntervalTimer();
    JCOSignedBytes signedObject = privateKey.sign(originalObject);
    System.out.println(t + "  JCOPrivateKey.sign(object)");
    Object verifiedObject = publicKey.verify(signedObject);
    System.out.println(t + "  JCOPublicKey.veritfy(signedObject)");
    assertEquals(originalObject, verifiedObject);
  }
}