PKCS11Test.java

/*
 * eID Applet Project.
 * Copyright (C) 2008-2012 FedICT.
 *
 * This is free software; you can redistribute it and/or modify it
 * under the terms of the GNU Lesser General Public License version
 * 3.0 as published by the Free Software Foundation.
 *
 * This software 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
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this software; if not, see 
 * http://www.gnu.org/licenses/.
 */

package test.be.fedict.eid.applet;

import java.io.File;
import java.io.FileOutputStream;
import java.io.PrintWriter;
import java.math.BigInteger;
import java.security.KeyStore;
import java.security.KeyStore.PrivateKeyEntry;
import java.security.PrivateKey;
import java.security.Security;
import java.security.Signature;
import java.security.cert.X509Certificate;
import java.security.interfaces.RSAPublicKey;

import javax.swing.JOptionPane;

import org.apache.commons.codec.binary.Hex;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.junit.Test;

import sun.security.pkcs11.SunPKCS11;

public class PKCS11Test {

	private static final Log LOG = LogFactory.getLog(PKCS11Test.class);

	@Test
	public void testPKCS1viaPKCS11() throws Exception {
		File tmpConfigFile = File.createTempFile("pkcs11-", "conf");
		tmpConfigFile.deleteOnExit();
		PrintWriter configWriter = new PrintWriter(new FileOutputStream(tmpConfigFile), true);
		configWriter.println("name=SmartCard");
		configWriter.println("library=/usr/lib/libbeidpkcs11.so.0");
		configWriter.println("slotListIndex=2");

		SunPKCS11 provider = new SunPKCS11(tmpConfigFile.getAbsolutePath());
		Security.addProvider(provider);
		KeyStore keyStore = KeyStore.getInstance("PKCS11", provider);
		keyStore.load(null, null);
		PrivateKeyEntry privateKeyEntry = (PrivateKeyEntry) keyStore.getEntry("Authentication", null);
		PrivateKey privateKey = privateKeyEntry.getPrivateKey();
		Signature signature = Signature.getInstance("SHA1withRSA");
		signature.initSign(privateKey);
		byte[] toBeSigned = "hello world".getBytes();
		signature.update(toBeSigned);
		byte[] signatureValue = signature.sign();

		X509Certificate certificate = (X509Certificate) privateKeyEntry.getCertificate();
		RSAPublicKey publicKey = (RSAPublicKey) certificate.getPublicKey();
		BigInteger signatureValueBigInteger = new BigInteger(signatureValue);
		BigInteger messageBigInteger = signatureValueBigInteger.modPow(publicKey.getPublicExponent(),
				publicKey.getModulus());
		LOG.debug("original message: " + new String(Hex.encodeHex(messageBigInteger.toByteArray())));

		// LOG.debug("ASN.1 signature: " + ASN1Dump.dumpAsString(obj)
	}

	@Test
	public void testTokenHasBeenRemovedError() throws Exception {
		File tmpConfigFile = File.createTempFile("pkcs11-", "conf");
		tmpConfigFile.deleteOnExit();
		PrintWriter configWriter = new PrintWriter(new FileOutputStream(tmpConfigFile), true);
		configWriter.println("name=SmartCard");
		configWriter.println("library=/usr/lib/libbeidpkcs11.so.0");
		configWriter.println("slotListIndex=1");

		SunPKCS11 provider = new SunPKCS11(tmpConfigFile.getAbsolutePath());
		Security.addProvider(provider);
		KeyStore keyStore = KeyStore.getInstance("PKCS11", provider);
		keyStore.load(null, null);
		{
			PrivateKeyEntry privateKeyEntry = (PrivateKeyEntry) keyStore.getEntry("Authentication", null);
			Signature signature = Signature.getInstance("SHA1withRSA");
			signature.initSign(privateKeyEntry.getPrivateKey());
			byte[] toBeSigned = "hello world".getBytes();
			signature.update(toBeSigned);
			byte[] signatureValue = signature.sign();
		}
		JOptionPane.showMessageDialog(null, "Please remove and re-insert the token...");
		{
			PrivateKeyEntry privateKeyEntry = (PrivateKeyEntry) keyStore.getEntry("Authentication", null);
			Signature signature = Signature.getInstance("SHA1withRSA");
			signature.initSign(privateKeyEntry.getPrivateKey());
			byte[] toBeSigned = "hello world".getBytes();
			signature.update(toBeSigned);
			byte[] signatureValue = signature.sign();
		}
	}

	@Test
	public void testTokenHasBeenRemovedWorkaround() throws Exception {
		File tmpConfigFile = File.createTempFile("pkcs11-", "conf");
		tmpConfigFile.deleteOnExit();
		PrintWriter configWriter = new PrintWriter(new FileOutputStream(tmpConfigFile), true);
		configWriter.println("name=SmartCard");
		configWriter.println("library=/usr/lib/libbeidpkcs11.so.0");
		configWriter.println("slotListIndex=1");

		SunPKCS11 provider = new SunPKCS11(tmpConfigFile.getAbsolutePath());
		Security.addProvider(provider);
		{
			KeyStore keyStore = KeyStore.getInstance("PKCS11", provider);
			keyStore.load(null, null);
			PrivateKeyEntry privateKeyEntry = (PrivateKeyEntry) keyStore.getEntry("Authentication", null);
			Signature signature = Signature.getInstance("SHA1withRSA");
			signature.initSign(privateKeyEntry.getPrivateKey());
			byte[] toBeSigned = "hello world".getBytes();
			signature.update(toBeSigned);
			byte[] signatureValue = signature.sign();

		}
		JOptionPane.showMessageDialog(null, "Please remove and re-insert the token...");
		Security.removeProvider(provider.getName());
		{
			SunPKCS11 provider2 = new SunPKCS11(tmpConfigFile.getAbsolutePath());
			Security.addProvider(provider2);
			KeyStore keyStore = KeyStore.getInstance("PKCS11", provider2);
			keyStore.load(null, null);
			PrivateKeyEntry privateKeyEntry = (PrivateKeyEntry) keyStore.getEntry("Authentication", null);
			Signature signature = Signature.getInstance("SHA1withRSA");
			signature.initSign(privateKeyEntry.getPrivateKey());
			byte[] toBeSigned = "hello world".getBytes();
			signature.update(toBeSigned);
			byte[] signatureValue = signature.sign();
			Security.removeProvider(provider2.getName());
		}
	}
}