SSLeayEncoderTest.java

/*******************************************************************************
 * Copyright (C) 2010 Marco Sandrini
 * 
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License as
 * published by the Free Software Foundation, either version 3 of the
 * License, or (at your option) any later version.
 * 
 * This program is distributed in the hope that it will be useful, but
 * WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Affero General Public License for more details.
 * 
 * You should have received a copy of the GNU Affero General Public
 * License along with this program.
 * If not, see <http://www.gnu.org/licenses/>.
 ******************************************************************************/
package org.casbah.provider;

import static org.junit.Assert.assertArrayEquals;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertTrue;

import java.io.ByteArrayOutputStream;
import java.io.File;
import java.io.IOException;
import java.security.GeneralSecurityException;
import java.security.KeyFactory;
import java.security.PrivateKey;
import java.security.interfaces.RSAPrivateCrtKey;
import java.security.interfaces.RSAPrivateKey;
import java.security.spec.RSAPrivateCrtKeySpec;

import org.apache.commons.codec.DecoderException;
import org.apache.commons.codec.binary.Hex;
import org.apache.commons.io.FileUtils;
import org.apache.commons.io.IOUtils;
import org.junit.Test;

public class SSLeayEncoderTest extends TestKeyValues {
	
	@Test
	public void testDecryptKey() throws IOException, GeneralSecurityException, DecoderException {
		byte[] salt = Hex.decodeHex("73EE9A1CEEFA817D".toCharArray());
		ByteArrayOutputStream baos = new ByteArrayOutputStream();
		IOUtils.copy(this.getClass().getResourceAsStream("/caplaintext.key"), baos);
		byte[] expected = baos.toByteArray();
		String pemData = IOUtils.toString(this.getClass().getResourceAsStream("/caencrypted.key"));
		byte[] decData = SSLeayEncoder.decryptKey(pemData, salt, "casbah");
		System.out.println(Hex.encodeHexString(decData));
		assertArrayEquals(expected, decData);
	}
	
	@Test
	public void testDecodeKey() throws CAProviderException, IOException {
		 
		File keyFile = new File(this.getClass().getResource("/ca.key").getFile());
		String pemData = FileUtils.readFileToString(keyFile);
		PrivateKey key = SSLeayEncoder.decodeKey(pemData, "casbah");
		assertTrue(key instanceof RSAPrivateKey);
		assertEquals(TestKeyValues.PRIVATE_EXPONENT, ((RSAPrivateKey) key).getPrivateExponent());
	}
	
	private static RSAPrivateCrtKey generateKey() throws GeneralSecurityException {
		KeyFactory kf = KeyFactory.getInstance("RSA");
		RSAPrivateCrtKeySpec keyspec = new RSAPrivateCrtKeySpec(MODULUS, PUBLIC_EXPONENT,
				PRIVATE_EXPONENT, PRIME1, PRIME2, EXPONENT1, EXPONENT2, COEFFICIENT);
		return (RSAPrivateCrtKey) kf.generatePrivate(keyspec);		
	}
	
	@Test
	public void testEncodeKey() throws CAProviderException, GeneralSecurityException, IOException {
		
		RSAPrivateCrtKey key = generateKey();
		String encodedKey = SSLeayEncoder.encodeKey(key, "casbah");
		RSAPrivateCrtKey key2 = SSLeayEncoder.decodeKey(encodedKey, "casbah");
		assertEquals(key, key2);	
	}
	
	@Test
	public void testEncryptKey() throws GeneralSecurityException, DecoderException, IOException {
		
		PKCS1EncodedKey key = new PKCS1EncodedKey(generateKey());
		byte[] derEncoded = key.getEncoded();
		
		byte[] salt = Hex.decodeHex("73EE9A1CEEFA817D".toCharArray());
		String expected = FileUtils.readFileToString(new File(this.getClass().getResource("/caencrypted.key").getFile()));
		
		String pemData = SSLeayEncoder.encryptKey(derEncoded, salt, "casbah");
		assertEquals(expected, pemData);
		
	}
	
}