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.Test;
28 public class TestDNS extends ManagedTest {
31 public void testDNSSanity() throws IOException, NamingException {
33 String token = RandomToken.generateToken(16);
34 String value = RandomToken.generateToken(16);
36 String reRead = updateDNS(token, value);
37 assertEquals(value, reRead);
42 public void testEmailAndDNS() throws IOException, InterruptedException, SQLException, NamingException {
43 String email = createUniqueName() + "@example.org";
44 int uid = createVerifiedUser("a", "b", email, TEST_PASSWORD);
45 String cookie = login(email, TEST_PASSWORD);
47 String test = getTestProps().getProperty("domain.dnstest");
48 URL u = new URL("https://" + getServerName() + DomainOverview.PATH);
49 URLConnection openConnection = u.openConnection();
50 openConnection.setRequestProperty("Cookie", cookie);
51 String content1 = IOUtils.readURL(openConnection);
52 String csrf = getCSRF(1, content1);
54 Pattern p = Pattern.compile("cacert-([A-Za-z0-9]+) IN TXT ([A-Za-z0-9]+)");
55 Matcher m = p.matcher(content1);
57 updateDNS(m.group(1), m.group(2));
59 String content = "newdomain=" + URLEncoder.encode(test, "UTF-8") + //
60 "&emailType=y&email=2&DNSType=y" + //
61 "&ssl-type-0=direct&ssl-port-0=" + //
62 "&ssl-type-1=direct&ssl-port-1=" + //
63 "&ssl-type-2=direct&ssl-port-2=" + //
64 "&ssl-type-3=direct&ssl-port-3=" + //
65 "&adddomain&csrf=" + csrf;
66 openConnection = u.openConnection();
67 openConnection.setRequestProperty("Cookie", cookie);
68 openConnection.setDoOutput(true);
69 openConnection.getOutputStream().write(content.getBytes());
70 openConnection.getHeaderField("Location");
72 String newcontent = IOUtils.readURL(cookie(u.openConnection(), cookie));
73 Pattern dlink = Pattern.compile(DomainOverview.PATH + "([0-9]+)'>");
74 Matcher m1 = dlink.matcher(newcontent);
76 URL u2 = new URL(u.toString() + m1.group(1));
78 TestMail mail = getMailReciever().recieve();
79 String link = mail.extractLink();
80 new URL(link).openConnection().getHeaderField("");
82 PreparedStatement ps = DatabaseConnection.getInstance().prepare("SELECT COUNT(*) FROM domainPinglog");
84 ResultSet rs = ps.executeQuery();
86 if (rs.getInt(1) >= 2) {
92 newcontent = IOUtils.readURL(cookie(u2.openConnection(), cookie));
93 Pattern pat = Pattern.compile("<td>dns</td>\\s*<td>success</td>");
94 assertTrue(newcontent, pat.matcher(newcontent).find());
95 pat = Pattern.compile("<td>email</td>\\s*<td>success</td>");
96 assertTrue(newcontent, pat.matcher(newcontent).find());
99 private String updateDNS(String token, String value) throws IOException, MalformedURLException, NamingException {
100 String test = getTestProps().getProperty("domain.dnstest");
101 String targetDomain = "cacert-" + token + "." + test;
102 String manage = getTestProps().getProperty("domain.dnsmanage");
103 String url = manage + "t1=" + token + "&t2=" + value;
104 assertEquals(200, ((HttpURLConnection) new URL(url).openConnection()).getResponseCode());
105 String[] data = DNSUtil.getTXTEntries(targetDomain, getTestProps().getProperty("domain.testns"));
106 assertEquals(1, data.length);