tests/org/cacert/gigi/TestCertificate.java

   1 package org.cacert.gigi;
   2
   3 import static org.junit.Assert.*;
   4
   5 import java.io.IOException;
   6 import java.security.GeneralSecurityException;
   7 import java.security.KeyPair;
   8 import java.security.PrivateKey;
   9 import java.security.cert.X509Certificate;
  10 import java.sql.SQLException;
  11 import java.util.Collection;
  12 import java.util.List;
  13
  14 import org.cacert.gigi.dbObjects.Certificate;
  15 import org.cacert.gigi.dbObjects.Certificate.CSRType;
  16 import org.cacert.gigi.dbObjects.Certificate.CertificateStatus;
  17 import org.cacert.gigi.dbObjects.Certificate.SANType;
  18 import org.cacert.gigi.dbObjects.Certificate.SubjectAlternateName;
  19 import org.cacert.gigi.dbObjects.CertificateProfile;
  20 import org.cacert.gigi.dbObjects.Digest;
  21 import org.cacert.gigi.dbObjects.User;
  22 import org.cacert.gigi.testUtils.ManagedTest;
  23 import org.junit.Test;
  24
  25 import sun.security.x509.GeneralNameInterface;
  26
  27 public class TestCertificate extends ManagedTest {
  28
  29     User u = User.getById(createVerifiedUser("fn", "ln", createUniqueName() + "@example.com", TEST_PASSWORD));
  30
  31     @Test
  32     public void testClientCertLoginStates() throws IOException, GeneralSecurityException, SQLException, InterruptedException, GigiApiException {
  33         KeyPair kp = generateKeypair();
  34         String key1 = generatePEMCSR(kp, "CN=testmail@example.com");
  35         Certificate c = new Certificate(u, u, Certificate.buildDN("CN", "testmail@example.com"), Digest.SHA256, key1, CSRType.CSR, CertificateProfile.getById(1));
  36         final PrivateKey pk = kp.getPrivate();
  37         c.issue(null, "2y", u).waitFor(60000);
  38         final X509Certificate ce = c.cert();
  39         assertNotNull(login(pk, ce));
  40     }
  41
  42     @Test
  43     public void testSANs() throws IOException, GeneralSecurityException, SQLException, InterruptedException, GigiApiException {
  44         KeyPair kp = generateKeypair();
  45         String key = generatePEMCSR(kp, "CN=testmail@example.com");
  46         Certificate c = new Certificate(u, u, Certificate.buildDN("CN", "testmail@example.com"), Digest.SHA256, key, CSRType.CSR, CertificateProfile.getById(1),//
  47                 new SubjectAlternateName(SANType.EMAIL, "testmail@example.com"), new SubjectAlternateName(SANType.DNS, "testmail.example.com"));
  48
  49         testFails(CertificateStatus.DRAFT, c);
  50         c.issue(null, "2y", u).waitFor(60000);
  51         X509Certificate cert = c.cert();
  52         Collection<List<?>> sans = cert.getSubjectAlternativeNames();
  53         assertEquals(2, sans.size());
  54         boolean hadDNS = false;
  55         boolean hadEmail = false;
  56         for (List<?> list : sans) {
  57             assertEquals(2, list.size());
  58             Integer type = (Integer) list.get(0);
  59             switch (type) {
  60             case GeneralNameInterface.NAME_RFC822:
  61                 hadEmail = true;
  62                 assertEquals("testmail@example.com", list.get(1));
  63                 break;
  64             case GeneralNameInterface.NAME_DNS:
  65                 hadDNS = true;
  66                 assertEquals("testmail.example.com", list.get(1));
  67                 break;
  68             default:
  69                 fail("Unknown type");
  70
  71             }
  72         }
  73         assertTrue(hadDNS);
  74         assertTrue(hadEmail);
  75
  76         testFails(CertificateStatus.ISSUED, c);
  77
  78         Certificate c2 = Certificate.getBySerial(c.getSerial());
  79         assertNotNull(c2);
  80         assertEquals(2, c2.getSANs().size());
  81         assertEquals(c.getSANs().get(0).getName(), c2.getSANs().get(0).getName());
  82         assertEquals(c.getSANs().get(0).getType(), c2.getSANs().get(0).getType());
  83         assertEquals(c.getSANs().get(1).getName(), c2.getSANs().get(1).getName());
  84         assertEquals(c.getSANs().get(1).getType(), c2.getSANs().get(1).getType());
  85
  86         try {
  87             c2.getSANs().remove(0);
  88             fail("the list should not be modifiable");
  89         } catch (UnsupportedOperationException e) {
  90             // expected
  91         }
  92     }
  93
  94     @Test
  95     public void testCertLifeCycle() throws IOException, GeneralSecurityException, SQLException, InterruptedException, GigiApiException {
  96         KeyPair kp = generateKeypair();
  97         String key = generatePEMCSR(kp, "CN=testmail@example.com");
  98         Certificate c = new Certificate(u, u, Certificate.buildDN("CN", "testmail@example.com"), Digest.SHA256, key, CSRType.CSR, CertificateProfile.getById(1));
  99         final PrivateKey pk = kp.getPrivate();
 100
 101         testFails(CertificateStatus.DRAFT, c);
 102         c.issue(null, "2y", u).waitFor(60000);
 103
 104         testFails(CertificateStatus.ISSUED, c);
 105         X509Certificate cert = c.cert();
 106         assertNotNull(login(pk, cert));
 107         c.revoke().waitFor(60000);
 108
 109         testFails(CertificateStatus.REVOKED, c);
 110         assertNull(login(pk, cert));
 111
 112     }
 113
 114     private void testFails(CertificateStatus status, Certificate c) throws IOException, GeneralSecurityException, SQLException, GigiApiException {
 115         assertEquals(status, c.getStatus());
 116         if (status != CertificateStatus.ISSUED) {
 117             try {
 118                 c.revoke();
 119                 fail(status + " is in invalid state");
 120             } catch (IllegalStateException ise) {
 121
 122             }
 123         }
 124         if (status != CertificateStatus.DRAFT) {
 125             try {
 126                 c.issue(null, "2y", u);
 127                 fail(status + " is in invalid state");
 128             } catch (IllegalStateException ise) {
 129
 130             }
 131         }
 132         if (status != CertificateStatus.ISSUED) {
 133             try {
 134                 c.cert();
 135                 if (status != CertificateStatus.REVOKED) {
 136                     fail(status + " is in invalid state");
 137                 }
 138             } catch (IllegalStateException ise) {
 139
 140             }
 141         }
 142     }
 143 }