1 package org.cacert.gigi.ping;
3 import static org.junit.Assert.*;
5 import java.io.IOException;
6 import java.net.HttpURLConnection;
7 import java.net.MalformedURLException;
9 import java.net.URLConnection;
10 import java.net.URLEncoder;
11 import java.sql.PreparedStatement;
12 import java.sql.ResultSet;
13 import java.sql.SQLException;
14 import java.util.regex.Matcher;
15 import java.util.regex.Pattern;
17 import javax.naming.NamingException;
19 import org.cacert.gigi.database.DatabaseConnection;
20 import org.cacert.gigi.pages.account.DomainOverview;
21 import org.cacert.gigi.testUtils.IOUtils;
22 import org.cacert.gigi.testUtils.ManagedTest;
23 import org.cacert.gigi.testUtils.TestEmailReciever.TestMail;
24 import org.cacert.gigi.util.DNSUtil;
25 import org.cacert.gigi.util.RandomToken;
26 import org.junit.After;
27 import org.junit.Test;
29 public class TestDNS extends ManagedTest {
32 public void testDNSSanity() throws IOException, NamingException {
34 String token = RandomToken.generateToken(16);
35 String value = RandomToken.generateToken(16);
37 String reRead = updateDNS(token, value);
38 assertEquals(value, reRead);
43 public void testEmailAndDNSSuccess() throws IOException, InterruptedException, SQLException, NamingException {
44 testEmailAndDNS(0, 0, true, true);
48 public void test() throws SQLException, IOException {
53 public void testEmailAndDNSFail() throws IOException, InterruptedException, SQLException, NamingException {
54 testEmailAndDNS(1, 0, false, true);
56 testEmailAndDNS(2, 0, false, true);
59 public void testEmailAndDNS(int dnsVariant, int emailVariant, boolean successDNS, boolean successMail) throws IOException, InterruptedException, SQLException, NamingException {
60 String email = createUniqueName() + "@example.org";
61 createVerifiedUser("a", "b", email, TEST_PASSWORD);
62 String cookie = login(email, TEST_PASSWORD);
64 String test = getTestProps().getProperty("domain.dnstest");
65 URL u = new URL("https://" + getServerName() + DomainOverview.PATH);
66 URLConnection openConnection = u.openConnection();
67 openConnection.setRequestProperty("Cookie", cookie);
68 String content1 = IOUtils.readURL(openConnection);
69 String csrf = getCSRF(1, content1);
71 Pattern p = Pattern.compile("cacert-([A-Za-z0-9]+) IN TXT ([A-Za-z0-9]+)");
72 Matcher m = p.matcher(content1);
74 updateDNS(m.group(1) + (dnsVariant == 1 ? "a" : ""), m.group(2) + (dnsVariant == 2 ? "a" : ""));
76 String content = "newdomain=" + URLEncoder.encode(test, "UTF-8") + //
77 "&emailType=y&email=2&DNSType=y" + //
78 "&ssl-type-0=direct&ssl-port-0=" + //
79 "&ssl-type-1=direct&ssl-port-1=" + //
80 "&ssl-type-2=direct&ssl-port-2=" + //
81 "&ssl-type-3=direct&ssl-port-3=" + //
82 "&adddomain&csrf=" + csrf;
83 openConnection = u.openConnection();
84 openConnection.setRequestProperty("Cookie", cookie);
85 openConnection.setDoOutput(true);
86 openConnection.getOutputStream().write(content.getBytes());
87 openConnection.getHeaderField("Location");
89 String newcontent = IOUtils.readURL(cookie(u.openConnection(), cookie));
90 Pattern dlink = Pattern.compile(DomainOverview.PATH + "([0-9]+)'>");
91 Matcher m1 = dlink.matcher(newcontent);
93 URL u2 = new URL(u.toString() + m1.group(1));
95 TestMail mail = getMailReciever().recieve();
96 String link = mail.extractLink();
97 new URL(link).openConnection().getHeaderField("");
99 PreparedStatement ps = DatabaseConnection.getInstance().prepare("SELECT COUNT(*) FROM domainPinglog");
101 ResultSet rs = ps.executeQuery();
103 if (rs.getInt(1) >= 2) {
109 newcontent = IOUtils.readURL(cookie(u2.openConnection(), cookie));
110 Pattern pat = Pattern.compile("<td>dns</td>\\s*<td>success</td>");
111 assertTrue(newcontent, !successDNS ^ pat.matcher(newcontent).find());
112 pat = Pattern.compile("<td>email</td>\\s*<td>success</td>");
113 assertTrue(newcontent, !successMail ^ pat.matcher(newcontent).find());
116 private String updateDNS(String token, String value) throws IOException, MalformedURLException, NamingException {
117 String test = getTestProps().getProperty("domain.dnstest");
118 String targetDomain = "cacert-" + token + "." + test;
119 String manage = getTestProps().getProperty("domain.dnsmanage");
120 String url = manage + "t1=" + token + "&t2=" + value;
121 assertEquals(200, ((HttpURLConnection) new URL(url).openConnection()).getResponseCode());
122 String[] data = DNSUtil.getTXTEntries(targetDomain, getTestProps().getProperty("domain.testns"));
123 assertEquals(1, data.length);