tests/org/cacert/gigi/ping/TestDNS.java

   1 package org.cacert.gigi.ping;
   2
   3 import static org.junit.Assert.*;
   4
   5 import java.io.IOException;
   6 import java.net.HttpURLConnection;
   7 import java.net.MalformedURLException;
   8 import java.net.URL;
   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;
  16
  17 import javax.naming.NamingException;
  18
  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;
  28
  29 public class TestDNS extends ManagedTest {
  30
  31     @Test
  32     public void testDNSSanity() throws IOException, NamingException {
  33
  34         String token = RandomToken.generateToken(16);
  35         String value = RandomToken.generateToken(16);
  36
  37         String reRead = updateDNS(token, value);
  38         assertEquals(value, reRead);
  39
  40     }
  41
  42     @Test
  43     public void testEmailAndDNSSuccess() throws IOException, InterruptedException, SQLException, NamingException {
  44         testEmailAndDNS(0, 0, true, true);
  45     }
  46
  47     @After
  48     public void test() throws SQLException, IOException {
  49         purgeDatabase();
  50     }
  51
  52     @Test
  53     public void testEmailAndDNSFail() throws IOException, InterruptedException, SQLException, NamingException {
  54         testEmailAndDNS(1, 0, false, true);
  55         purgeDatabase();
  56         testEmailAndDNS(2, 0, false, true);
  57     }
  58
  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);
  63
  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);
  70
  71         Pattern p = Pattern.compile("cacert-([A-Za-z0-9]+) IN TXT ([A-Za-z0-9]+)");
  72         Matcher m = p.matcher(content1);
  73         m.find();
  74         updateDNS(m.group(1) + (dnsVariant == 1 ? "a" : ""), m.group(2) + (dnsVariant == 2 ? "a" : ""));
  75
  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");
  88
  89         String newcontent = IOUtils.readURL(cookie(u.openConnection(), cookie));
  90         Pattern dlink = Pattern.compile(DomainOverview.PATH + "([0-9]+)'>");
  91         Matcher m1 = dlink.matcher(newcontent);
  92         m1.find();
  93         URL u2 = new URL(u.toString() + m1.group(1));
  94
  95         TestMail mail = getMailReciever().recieve();
  96         String link = mail.extractLink();
  97         new URL(link).openConnection().getHeaderField("");
  98
  99         PreparedStatement ps = DatabaseConnection.getInstance().prepare("SELECT COUNT(*) FROM domainPinglog");
 100         while (true) {
 101             ResultSet rs = ps.executeQuery();
 102             rs.next();
 103             if (rs.getInt(1) >= 2) {
 104                 break;
 105             }
 106             Thread.sleep(200);
 107         }
 108
 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());
 114     }
 115
 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);
 124         return data[0];
 125     }
 126 }