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 org.cacert.gigi.database.DatabaseConnection;
  18 import org.cacert.gigi.pages.account.DomainOverview;
  19 import org.cacert.gigi.testUtils.IOUtils;
  20 import org.cacert.gigi.testUtils.ManagedTest;
  21 import org.cacert.gigi.testUtils.TestEmailReciever.TestMail;
  22 import org.cacert.gigi.util.RandomToken;
  23 import org.junit.Test;
  24
  25 public class TestDNS extends ManagedTest {
  26
  27     @Test
  28     public void testDNSSanity() throws IOException {
  29
  30         String token = RandomToken.generateToken(16);
  31         String value = RandomToken.generateToken(16);
  32
  33         Process p = updateDNS(token, value);
  34         String reRead = new String(IOUtils.readURL(p.getInputStream()));
  35         reRead = reRead.trim();
  36         reRead = reRead.substring(1, reRead.length() - 1);
  37         assertEquals(value, reRead);
  38
  39     }
  40
  41     @Test
  42     public void testEmailAndDNS() throws IOException, InterruptedException, SQLException {
  43         String email = createUniqueName() + "@example.org";
  44         int uid = createVerifiedUser("a", "b", email, TEST_PASSWORD);
  45         String cookie = login(email, TEST_PASSWORD);
  46
  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);
  53
  54         Pattern p = Pattern.compile("cacert-([A-Za-z0-9]+) IN TXT ([A-Za-z0-9]+)");
  55         Matcher m = p.matcher(content1);
  56         m.find();
  57         updateDNS(m.group(1), m.group(2));
  58
  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");
  71
  72         String newcontent = IOUtils.readURL(cookie(u.openConnection(), cookie));
  73         Pattern dlink = Pattern.compile(DomainOverview.PATH + "([0-9]+)'>");
  74         Matcher m1 = dlink.matcher(newcontent);
  75         m1.find();
  76         URL u2 = new URL(u.toString() + m1.group(1));
  77
  78         TestMail mail = getMailReciever().recieve();
  79         String link = mail.extractLink();
  80         new URL(link).openConnection().getHeaderField("");
  81
  82         PreparedStatement ps = DatabaseConnection.getInstance().prepare("SELECT COUNT(*) FROM domainPinglog");
  83         while (true) {
  84             ResultSet rs = ps.executeQuery();
  85             rs.next();
  86             if (rs.getInt(1) >= 2) {
  87                 break;
  88             }
  89             Thread.sleep(200);
  90         }
  91
  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());
  97     }
  98
  99     private Process updateDNS(String token, String value) throws IOException, MalformedURLException {
 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
 106         Process p = Runtime.getRuntime().exec(new String[] {
 107                 "dig", "@" + getTestProps().getProperty("domain.testns"), "+short", "TXT", targetDomain
 108         });
 109         return p;
 110     }
 111
 112 }