1 package club.wpia.gigi;
3 import static org.junit.Assert.*;
5 import java.io.IOException;
6 import java.math.BigInteger;
7 import java.security.GeneralSecurityException;
8 import java.security.KeyPair;
9 import java.security.PrivateKey;
10 import java.security.cert.X509Certificate;
11 import java.sql.SQLException;
12 import java.util.Collection;
13 import java.util.List;
15 import org.hamcrest.CoreMatchers;
16 import org.junit.Test;
18 import club.wpia.gigi.dbObjects.Certificate;
19 import club.wpia.gigi.dbObjects.Certificate.CSRType;
20 import club.wpia.gigi.dbObjects.Certificate.CertificateStatus;
21 import club.wpia.gigi.dbObjects.Certificate.RevocationType;
22 import club.wpia.gigi.dbObjects.Certificate.SANType;
23 import club.wpia.gigi.dbObjects.Certificate.SubjectAlternateName;
24 import club.wpia.gigi.dbObjects.Digest;
25 import club.wpia.gigi.dbObjects.User;
26 import club.wpia.gigi.pages.account.certs.Certificates;
27 import club.wpia.gigi.testUtils.IOUtils;
28 import club.wpia.gigi.testUtils.ManagedTest;
29 import club.wpia.gigi.util.RandomToken;
30 import sun.security.x509.GeneralNameInterface;
32 public class TestCertificate extends ManagedTest {
34 User u = User.getById(createVerifiedUser("fn", "ln", createUniqueName() + "@example.com", TEST_PASSWORD));
37 public void testClientCertLoginStates() throws IOException, GeneralSecurityException, SQLException, InterruptedException, GigiApiException {
38 KeyPair kp = generateKeypair();
39 String key1 = generatePEMCSR(kp, "CN=testmail@example.com");
40 Certificate c = new Certificate(u, u, Certificate.buildDN("CN", "testmail@example.com"), Digest.SHA256, key1, CSRType.CSR, getClientProfile());
41 final PrivateKey pk = kp.getPrivate();
42 await(c.issue(null, "2y", u));
43 final X509Certificate ce = c.cert();
44 c.setLoginEnabled(true);
45 assertNotNull(login(pk, ce));
49 public void testSANs() throws IOException, GeneralSecurityException, SQLException, InterruptedException, GigiApiException {
50 KeyPair kp = generateKeypair();
51 String key = generatePEMCSR(kp, "CN=testmail@example.com");
52 Certificate c = new Certificate(u, u, Certificate.buildDN("CN", "testmail@example.com"), Digest.SHA256, key, CSRType.CSR, getClientProfile(),//
53 new SubjectAlternateName(SANType.EMAIL, "testmail@example.com"), new SubjectAlternateName(SANType.DNS, "testmail.example.com"));
55 testFails(CertificateStatus.DRAFT, c);
56 await(c.issue(null, "2y", u));
57 X509Certificate cert = c.cert();
58 Collection<List<?>> sans = cert.getSubjectAlternativeNames();
59 assertEquals(2, sans.size());
60 boolean hadDNS = false;
61 boolean hadEmail = false;
62 for (List<?> list : sans) {
63 assertEquals(2, list.size());
64 Integer type = (Integer) list.get(0);
66 case GeneralNameInterface.NAME_RFC822:
68 assertEquals("testmail@example.com", list.get(1));
70 case GeneralNameInterface.NAME_DNS:
72 assertEquals("testmail.example.com", list.get(1));
82 testFails(CertificateStatus.ISSUED, c);
84 Certificate c2 = Certificate.getBySerial(new BigInteger(c.getSerial(), 16));
86 assertEquals(2, c2.getSANs().size());
87 assertEquals(c.getSANs().get(0).getName(), c2.getSANs().get(0).getName());
88 assertEquals(c.getSANs().get(0).getType(), c2.getSANs().get(0).getType());
89 assertEquals(c.getSANs().get(1).getName(), c2.getSANs().get(1).getName());
90 assertEquals(c.getSANs().get(1).getType(), c2.getSANs().get(1).getType());
93 c2.getSANs().remove(0);
94 fail("the list should not be modifiable");
95 } catch (UnsupportedOperationException e) {
101 public void testCertCreateSHA384() throws IOException, GeneralSecurityException, SQLException, InterruptedException, GigiApiException {
102 KeyPair kp = generateKeypair();
103 String key = generatePEMCSR(kp, "CN=testmail@example.com");
104 Certificate c = new Certificate(u, u, Certificate.buildDN("CN", "testmail@example.com"), Digest.SHA384, key, CSRType.CSR, getClientProfile());
105 await(c.issue(null, "2y", u));
106 assertThat(c.cert().getSigAlgName().toLowerCase(), CoreMatchers.containsString("sha384"));
110 public void testCertLifeCycle() throws IOException, GeneralSecurityException, SQLException, InterruptedException, GigiApiException {
111 KeyPair kp = generateKeypair();
112 String key = generatePEMCSR(kp, "CN=testmail@example.com");
113 Certificate c = new Certificate(u, u, Certificate.buildDN("CN", "testmail@example.com"), Digest.SHA256, key, CSRType.CSR, getClientProfile());
114 final PrivateKey pk = kp.getPrivate();
116 testFails(CertificateStatus.DRAFT, c);
117 await(c.issue(null, "2y", u));
119 String cookie = login(u.getEmail(), TEST_PASSWORD);
120 testFails(CertificateStatus.ISSUED, c);
121 X509Certificate cert = c.cert();
122 c.setLoginEnabled(true);
123 assertNotNull(login(pk, cert));
124 assertEquals(1, countRegex(IOUtils.readURL(get(cookie, Certificates.PATH)), "<td>(?:REVOKED|ISSUED)</td>"));
125 assertEquals(1, countRegex(IOUtils.readURL(get(cookie, Certificates.PATH + "?withRevoked")), "<td>(?:REVOKED|ISSUED)</td>"));
126 await(c.revoke(RevocationType.USER));
128 testFails(CertificateStatus.REVOKED, c);
129 assertNull(login(pk, cert));
131 assertEquals(0, countRegex(IOUtils.readURL(get(cookie, Certificates.PATH)), "<td>(?:REVOKED|ISSUED)</td>"));
132 assertEquals(1, countRegex(IOUtils.readURL(get(cookie, Certificates.PATH + "?withRevoked")), "<td>(?:REVOKED|ISSUED)</td>"));
135 private void testFails(CertificateStatus status, Certificate c) throws IOException, GeneralSecurityException, SQLException, GigiApiException {
136 assertEquals(status, c.getStatus());
137 if (status != CertificateStatus.ISSUED) {
139 c.revoke(RevocationType.USER);
140 fail(status + " is in invalid state");
141 } catch (IllegalStateException ise) {
145 if (status != CertificateStatus.DRAFT) {
147 c.issue(null, "2y", u);
148 fail(status + " is in invalid state");
149 } catch (IllegalStateException ise) {
153 if (status != CertificateStatus.ISSUED) {
156 if (status != CertificateStatus.REVOKED) {
157 fail(status + " is in invalid state");
159 } catch (IllegalStateException ise) {
166 public void testClientCertDescription() throws IOException, GeneralSecurityException, SQLException, InterruptedException, GigiApiException {
167 KeyPair kp = generateKeypair();
168 String key1 = generatePEMCSR(kp, "CN=testmail@example.com");
169 Certificate c = new Certificate(u, u, Certificate.buildDN("CN", "testmail@example.com"), Digest.SHA256, key1, CSRType.CSR, getClientProfile());
170 await(c.issue(null, "2y", u));
171 String description = RandomToken.generateToken(95) + DIFFICULT_CHARS;
172 c.setDescription(description);
173 assertEquals(description, c.getDescription());
175 // test that description is entered to db
178 Certificate cn = Certificate.getById(cid);
179 assertEquals(description, cn.getDescription());