bcsmime_demo/src/test/java/de/communardo/jdi/bcsmime_demo/EncryptDecryptTest.java

/*
 * Copyright (c) 2011 Jan Dittberner
 *
 * Permission is hereby granted, free of charge, to any person obtaining
 * a copy of this software and associated documentation files (the
 * "Software"), to deal in the Software without restriction, including
 * without limitation the rights to use, copy, modify, merge, publish,
 * distribute, sublicense, and/or sell copies of the Software, and to
 * permit persons to whom the Software is furnished to do so, subject to
 * the following conditions:
 *
 * The above copyright notice and this permission notice shall be
 * included in all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
 * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
 * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.
 * IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY
 * CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
 * TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
 * SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
 */
package de.communardo.jdi.bcsmime_demo;

import java.io.IOException;
import java.math.BigInteger;
import java.security.KeyPair;
import java.security.KeyPairGenerator;
import java.security.KeyStore;
import java.security.Security;
import java.security.cert.Certificate;
import java.util.Date;

import javax.mail.BodyPart;
import javax.mail.MessagingException;
import javax.mail.Multipart;
import javax.mail.Session;
import javax.mail.Message.RecipientType;
import javax.mail.internet.InternetAddress;
import javax.mail.internet.MimeBodyPart;
import javax.mail.internet.MimeMessage;
import javax.mail.internet.MimeMultipart;

import junit.framework.TestCase;

import org.bouncycastle.asn1.x500.X500Name;
import org.bouncycastle.asn1.x509.BasicConstraints;
import org.bouncycastle.asn1.x509.ExtendedKeyUsage;
import org.bouncycastle.asn1.x509.GeneralName;
import org.bouncycastle.asn1.x509.GeneralNames;
import org.bouncycastle.asn1.x509.KeyPurposeId;
import org.bouncycastle.asn1.x509.KeyUsage;
import org.bouncycastle.asn1.x509.X509Extension;
import org.bouncycastle.cert.X509CertificateHolder;
import org.bouncycastle.cert.X509v3CertificateBuilder;
import org.bouncycastle.cert.jcajce.JcaX509CertificateConverter;
import org.bouncycastle.cert.jcajce.JcaX509v3CertificateBuilder;
import org.bouncycastle.jce.provider.BouncyCastleProvider;
import org.bouncycastle.operator.ContentSigner;
import org.bouncycastle.operator.jcajce.JcaContentSignerBuilder;

/**
 * Test Encryption and Decryption.
 * 
 * @author Jan Dittberner &lt;<a href=
 *         "mailto:jan.dittberner@t-systems.com>jan.dittberner@t-systems.com</a>
 *         & g t ;
 */
public class EncryptDecryptTest extends TestCase {
    private KeyStore keystore;

    /**
     * {@inheritDoc}
     * 
     * @see junit.framework.TestCase#setUp()
     */
    @Override
    public void setUp() throws Exception {
        if (Security.getProvider(BouncyCastleProvider.PROVIDER_NAME) == null) {
            Security.addProvider(new BouncyCastleProvider());
        }

        if (this.keystore == null) {
            this.keystore = KeyStore.getInstance("JKS");
            keystore.load(null, null);

            KeyPairGenerator kpg = KeyPairGenerator.getInstance("RSA");
            kpg.initialize(2048);
            KeyPair keypair = kpg.generateKeyPair();

            X500Name issuer = new X500Name(
                    "CN=Test Recipient,emailAddress=testrecpt@example.org");
            X500Name subject = issuer;
            X509v3CertificateBuilder certbuilder = new JcaX509v3CertificateBuilder(
                    issuer, BigInteger.valueOf(System.currentTimeMillis()),
                    new Date(System.currentTimeMillis() - 50000), new Date(
                            System.currentTimeMillis() + 50000), subject,
                    keypair.getPublic());
            certbuilder.addExtension(X509Extension.basicConstraints, true,
                    new BasicConstraints(true));
            certbuilder.addExtension(X509Extension.keyUsage, true,
                    new KeyUsage(KeyUsage.digitalSignature
                            | KeyUsage.keyEncipherment));
            certbuilder.addExtension(X509Extension.extendedKeyUsage, true,
                    new ExtendedKeyUsage(KeyPurposeId.id_kp_emailProtection));
            certbuilder.addExtension(X509Extension.subjectAlternativeName,
                    false, new GeneralNames(new GeneralName(
                            GeneralName.rfc822Name, "testrecpt@example.org")));

            ContentSigner signer = new JcaContentSignerBuilder("SHA256WithRSA")
                    .build(keypair.getPrivate());
            X509CertificateHolder certholder = certbuilder.build(signer);

            keystore.setKeyEntry("test", keypair.getPrivate(), "changeit"
                    .toCharArray(),
                    new Certificate[] { (new JcaX509CertificateConverter())
                            .getCertificate(certholder) });
        }
    }

    /**
     * Test of {@link SMIMEEncrypt} and {@link SMIMEDecrypt}.
     */
    public void testEncryptDecryptMail() throws Exception {
        MimeMessage message = getNewMultipartMessage();
        assertNotNull(message);

        SMIMEEncrypt encrypt = new SMIMEEncrypt(keystore);
        MimeMessage encrypted = encrypt.encryptMessage(message);
        assertNotNull(encrypted);
        encrypted.writeTo(System.err);

        SMIMEDecrypt decrypt = new SMIMEDecrypt(keystore);
        MimeMessage decrypted = decrypt.decryptMessage(encrypted);
        assertNotNull(decrypted);
        decrypted.writeTo(System.err);
    }

    /**
     * Creates a new MimeMessage with one Bodypart.
     * 
     * @return MimeMessage instance
     * @throws MessagingException
     *             on error creating the message
     */
    private MimeMessage getNewMultipartMessage() throws MessagingException,
            IOException {
        MimeMessage message = new MimeMessage(Session.getDefaultInstance(System
                .getProperties()));
        message.setFrom(new InternetAddress("testsender@example.org",
                "Test Sender"));
        message.addRecipient(RecipientType.TO, new InternetAddress(
                "testrecpt@example.org", "Test Recipient"));
        message.setSubject("Test subject");
        Multipart multipart = new MimeMultipart();
        BodyPart textpart = new MimeBodyPart();
        textpart.setText("Das ist ein Text");
        multipart.addBodyPart(textpart);
        message.setContent(multipart);
        return message;
    }
}