1 package club.wpia.gigi.pages.main;
3 import static org.junit.Assert.*;
5 import java.io.IOException;
6 import java.io.UnsupportedEncodingException;
7 import java.net.HttpURLConnection;
8 import java.net.URLEncoder;
9 import java.security.GeneralSecurityException;
10 import java.security.KeyPair;
11 import java.security.PrivateKey;
12 import java.security.cert.CertificateEncodingException;
14 import org.hamcrest.CoreMatchers;
15 import org.junit.Test;
17 import club.wpia.gigi.GigiApiException;
18 import club.wpia.gigi.dbObjects.Certificate;
19 import club.wpia.gigi.dbObjects.Certificate.CertificateStatus;
20 import club.wpia.gigi.dbObjects.Digest;
21 import club.wpia.gigi.dbObjects.Job;
22 import club.wpia.gigi.pages.account.certs.CertificateRequest;
23 import club.wpia.gigi.testUtils.ClientTest;
24 import club.wpia.gigi.testUtils.IOUtils;
25 import club.wpia.gigi.testUtils.TestEmailReceiver.TestMail;
26 import club.wpia.gigi.util.AuthorizationContext;
27 import club.wpia.gigi.util.HTMLEncoder;
28 import club.wpia.gigi.util.PEM;
30 public class KeyCompromiseTestMessage extends ClientTest {
32 private Certificate cert;
34 private PrivateKey priv;
36 public KeyCompromiseTestMessage() throws GeneralSecurityException, IOException, GigiApiException, InterruptedException {
37 KeyPair kp = generateKeypair();
38 priv = kp.getPrivate();
39 String csr = generatePEMCSR(kp, "CN=test");
40 CertificateRequest cr = new CertificateRequest(new AuthorizationContext(u, u, false), csr);
41 cr.update(CertificateRequest.DEFAULT_CN, Digest.SHA512.toString(), "client", null, null, "email:" + email + "\n");
43 Job j = cert.issue(null, "2y", u);
48 public void testExecution() throws IOException, InterruptedException, GigiApiException, GeneralSecurityException {
49 reportCompromiseAndCheck("");
53 public void testNoConfidential() throws IOException, InterruptedException, GigiApiException, GeneralSecurityException {
54 TestMail rc = reportCompromiseAndCheck("message=test+message");
55 assertThat(rc.getMessage(), CoreMatchers.containsString("test message"));
59 public void testNoConfidentialButMarker() throws IOException, InterruptedException, GigiApiException, GeneralSecurityException {
60 TestMail rc = reportCompromiseAndCheck("message=" + URLEncoder.encode(KeyCompromiseForm.CONFIDENTIAL_MARKER + "\ntest message", "UTF-8"));
61 assertThat(rc.getMessage(), CoreMatchers.containsString("test message"));
62 assertThat(rc.getMessage(), CoreMatchers.containsString(" " + KeyCompromiseForm.CONFIDENTIAL_MARKER));
66 public void testConfidential() throws IOException, InterruptedException, GigiApiException, GeneralSecurityException {
67 TestMail rc = reportCompromiseAndCheck("message=test+message&confidential=on");
68 assertThat(rc.getMessage(), CoreMatchers.not(CoreMatchers.containsString("test message")));
72 public void testCR() throws IOException, InterruptedException, GigiApiException, GeneralSecurityException {
73 TestMail rc = reportCompromiseAndCheck("message=test%0Dmessage&confidential=on");
74 assertThat(rc.getMessage(), CoreMatchers.not(CoreMatchers.containsString("test\r\nmessage")));
78 public void testLF() throws IOException, InterruptedException, GigiApiException, GeneralSecurityException {
79 TestMail rc = reportCompromiseAndCheck("message=test%0Amessage&confidential=on");
80 assertThat(rc.getMessage(), CoreMatchers.not(CoreMatchers.containsString("test\r\nmessage")));
84 public void testCRLF() throws IOException, InterruptedException, GigiApiException, GeneralSecurityException {
85 TestMail rc = reportCompromiseAndCheck("message=test%0D%0Amessage&confidential=on");
86 assertThat(rc.getMessage(), CoreMatchers.not(CoreMatchers.containsString("test\r\nmessage")));
90 public void testIllegalContent() throws IOException, InterruptedException, GigiApiException, GeneralSecurityException {
91 HttpURLConnection rc = reportCompromise("message=test+message+---&confidential=on");
92 String data = IOUtils.readURL(rc);
93 assertThat(data, hasError());
94 assertThat(data, CoreMatchers.containsString(HTMLEncoder.encodeHTML("message may not contain '---'")));
95 assertNull(getMailReceiver().poll(null));
96 assertEquals(CertificateStatus.ISSUED, cert.getStatus());
100 public void testIllegalChars() throws IOException, InterruptedException, GigiApiException, GeneralSecurityException {
101 HttpURLConnection rc = reportCompromise("message=" + URLEncoder.encode("ยง", "UTF-8"));
102 String data = IOUtils.readURL(rc);
103 assertThat(data, hasError());
104 assertThat(data, CoreMatchers.containsString("may only contain printable ASCII characters"));
105 assertEquals(CertificateStatus.ISSUED, cert.getStatus());
108 private TestMail reportCompromiseAndCheck(String params) throws IOException, UnsupportedEncodingException, CertificateEncodingException, GeneralSecurityException, GigiApiException {
109 HttpURLConnection huc = reportCompromise(params);
110 assertThat(IOUtils.readURL(huc), hasNoError());
111 TestMail rc = getMailReceiver().receive(email);
112 assertEquals(u.getEmail(), rc.getTo());
113 assertThat(rc.getMessage(), CoreMatchers.containsString(cert.getSerial()));
114 assertEquals(CertificateStatus.REVOKED, cert.getStatus());
118 private HttpURLConnection reportCompromise(String params) throws IOException, UnsupportedEncodingException, CertificateEncodingException, GeneralSecurityException, GigiApiException {
119 if ( !params.isEmpty() && !params.startsWith("&")) {
120 params = "&" + params;
122 HttpURLConnection huc = post(KeyCompromisePage.PATH, "cert=" + URLEncoder.encode(PEM.encode("CERTIFICATE", cert.cert().getEncoded()), "UTF-8")//
123 + "&priv=" + URLEncoder.encode(PEM.encode("PRIVATE KEY", priv.getEncoded()), "UTF-8") + params);