X-Git-Url: https://code.wpia.club/?p=gigi.git;a=blobdiff_plain;f=tests%2Forg%2Fcacert%2Fgigi%2FtestUtils%2FTestEmailReciever.java;h=b7cbc38999420b208af82d78c9f21846d99fc90a;hp=849bf927f234165190080c0ddba955a44a52cc76;hb=9474df27407dd99134876f2c8a2a3b6e7d1fae6a;hpb=af1651c1ebb7fe2633181653a2b66a11ca6926fd

diff --git a/tests/org/cacert/gigi/testUtils/TestEmailReciever.java b/tests/org/cacert/gigi/testUtils/TestEmailReciever.java
index 849bf927..b7cbc389 100644
--- a/tests/org/cacert/gigi/testUtils/TestEmailReciever.java
+++ b/tests/org/cacert/gigi/testUtils/TestEmailReciever.java
@@ -3,142 +3,185 @@ package org.cacert.gigi.testUtils;
 import java.io.DataInputStream;
 import java.io.DataOutputStream;
 import java.io.IOException;
-import java.net.SocketAddress;
 import java.net.Socket;
+import java.net.SocketAddress;
+import java.net.URL;
+import java.net.URLConnection;
 import java.util.concurrent.LinkedBlockingQueue;
 import java.util.concurrent.TimeUnit;
 import java.util.regex.Matcher;
 import java.util.regex.Pattern;
 
-public class TestEmailReciever implements Runnable {
-	public class TestMail {
-		String to;
-		String subject;
-		String message;
-		String from;
-		String replyto;
-
-		public TestMail(String to, String subject, String message, String from, String replyto) {
-			this.to = to;
-			this.subject = subject;
-			this.message = message;
-			this.from = from;
-			this.replyto = replyto;
-		}
-
-		public String getTo() {
-			return to;
-		}
-
-		public String getSubject() {
-			return subject;
-		}
-
-		public String getMessage() {
-			return message;
-		}
-
-		public String getFrom() {
-			return from;
-		}
-
-		public String getReplyto() {
-			return replyto;
-		}
-
-		public String extractLink() {
-			Pattern link = Pattern.compile("https?://[^\\s]+(?=\\s)");
-			Matcher m = link.matcher(getMessage());
-			m.find();
-			return m.group(0);
-		}
-
-	}
-
-	private Socket s;
-	private DataInputStream dis;
-	private DataOutputStream dos;
-
-	public TestEmailReciever(SocketAddress target) throws IOException {
-		s = new Socket();
-		s.connect(target);
-		s.setKeepAlive(true);
-		s.setSoTimeout(1000 * 60 * 60);
-		dis = new DataInputStream(s.getInputStream());
-		dos = new DataOutputStream(s.getOutputStream());
-		new Thread(this).start();
-	}
-
-	LinkedBlockingQueue<TestMail> mails = new LinkedBlockingQueue<TestEmailReciever.TestMail>();
-
-	public TestMail recieve() throws InterruptedException {
-		return mails.poll(5, TimeUnit.SECONDS);
-	}
-
-	@Override
-	public void run() {
-		try {
-			while (true) {
-				String type = dis.readUTF();
-				if (type.equals("mail")) {
-					String to = dis.readUTF();
-					String subject = dis.readUTF();
-					String message = dis.readUTF();
-					String from = dis.readUTF();
-					String replyto = dis.readUTF();
-					mails.add(new TestMail(to, subject, message, from, replyto));
-				} else if (type.equals("challengeAddrBox")) {
-					String email = dis.readUTF();
-					if (approveRegex.matcher(email).matches()) {
-						dos.writeUTF("OK");
-					} else {
-						dos.writeUTF(error);
-					}
-					dos.flush();
-				} else if (type.equals("ping")) {
-				} else {
-					System.err.println("Unknown type: " + type);
-				}
-			}
-		} catch (IOException e) {
-			if (!closed) {
-				e.printStackTrace();
-			}
-		}
-
-	}
-
-	String error = "FAIL";
-
-	public void setEmailCheckError(String error) {
-		this.error = error;
-	}
-
-	Pattern approveRegex = Pattern.compile(".*");
-
-	public void setApproveRegex(Pattern approveRegex) {
-		this.approveRegex = approveRegex;
-	}
-
-	public void clearMails() {
-		mails.clear();
-	}
-
-	public void reset() {
-		clearMails();
-		error = "FAIL";
-		approveRegex = Pattern.compile(".*");
-	}
-
-	boolean closed = false;
-
-	public void destroy() {
-		try {
-			closed = true;
-			s.close();
-		} catch (IOException e) {
-			e.printStackTrace();
-		}
-	}
+import org.cacert.gigi.email.EmailProvider;
+
+public class TestEmailReciever extends EmailProvider implements Runnable {
+
+    public class TestMail {
+
+        String to;
+
+        String subject;
+
+        String message;
+
+        String from;
+
+        String replyto;
+
+        public TestMail(String to, String subject, String message, String from, String replyto) {
+            this.to = to;
+            this.subject = subject;
+            this.message = message;
+            this.from = from;
+            this.replyto = replyto;
+        }
+
+        public String getTo() {
+            return to;
+        }
+
+        public String getSubject() {
+            return subject;
+        }
+
+        public String getMessage() {
+            return message;
+        }
+
+        public String getFrom() {
+            return from;
+        }
+
+        public String getReplyto() {
+            return replyto;
+        }
+
+        public String extractLink() {
+            Pattern link = Pattern.compile("https?://[^\\s]+(?=\\s)");
+            Matcher m = link.matcher(getMessage());
+            m.find();
+            return m.group(0);
+        }
+
+        public void verify() throws IOException {
+            String[] parts = extractLink().split("\\?");
+            URL u = new URL("https://" + ManagedTest.getServerName() + "/verify?" + parts[1]);
+
+            URLConnection csrfConn = u.openConnection();
+            String csrf = ManagedTest.getCSRF(csrfConn, 0);
+
+            u = new URL("https://" + ManagedTest.getServerName() + "/verify");
+            URLConnection uc = u.openConnection();
+            ManagedTest.cookie(uc, ManagedTest.stripCookie(csrfConn.getHeaderField("Set-Cookie")));
+            uc.setDoOutput(true);
+            uc.getOutputStream().write((parts[1] + "&csrf=" + csrf).getBytes("UTF-8"));
+            uc.connect();
+            uc.getInputStream().close();
+        }
+
+    }
+
+    private Socket s;
+
+    private DataInputStream dis;
+
+    private DataOutputStream dos;
+
+    public TestEmailReciever(SocketAddress target) throws IOException {
+        s = new Socket();
+        s.connect(target);
+        s.setKeepAlive(true);
+        s.setSoTimeout(1000 * 60 * 60);
+        dis = new DataInputStream(s.getInputStream());
+        dos = new DataOutputStream(s.getOutputStream());
+        new Thread(this).start();
+        setInstance(this);
+    }
+
+    LinkedBlockingQueue<TestMail> mails = new LinkedBlockingQueue<TestEmailReciever.TestMail>();
+
+    public TestMail recieve() throws InterruptedException {
+        return mails.poll(5, TimeUnit.SECONDS);
+    }
+
+    @Override
+    public void run() {
+        try {
+            while (true) {
+                String type = dis.readUTF();
+                if (type.equals("mail")) {
+                    String to = dis.readUTF();
+                    String subject = dis.readUTF();
+                    String message = dis.readUTF();
+                    String from = dis.readUTF();
+                    String replyto = dis.readUTF();
+                    mails.add(new TestMail(to, subject, message, from, replyto));
+                } else if (type.equals("challengeAddrBox")) {
+                    String email = dis.readUTF();
+                    dos.writeUTF(quickEmailCheck(email));
+                    dos.flush();
+                } else if (type.equals("ping")) {
+                } else {
+                    System.err.println("Unknown type: " + type);
+                }
+            }
+        } catch (IOException e) {
+            if ( !closed) {
+                e.printStackTrace();
+            }
+        }
+
+    }
+
+    private String quickEmailCheck(String email) throws IOException {
+        if (approveRegex.matcher(email).matches()) {
+            return "OK";
+        } else {
+            return error;
+        }
+    }
+
+    String error = "FAIL";
+
+    public void setEmailCheckError(String error) {
+        this.error = error;
+    }
+
+    Pattern approveRegex = Pattern.compile(".*");
+
+    public void setApproveRegex(Pattern approveRegex) {
+        this.approveRegex = approveRegex;
+    }
+
+    public void clearMails() {
+        mails.clear();
+    }
+
+    public void reset() {
+        clearMails();
+        error = "FAIL";
+        approveRegex = Pattern.compile(".*");
+    }
+
+    boolean closed = false;
+
+    public void destroy() {
+        try {
+            closed = true;
+            s.close();
+        } catch (IOException e) {
+            e.printStackTrace();
+        }
+    }
+
+    @Override
+    public String checkEmailServer(int forUid, String address) throws IOException {
+        return quickEmailCheck(address);
+    }
+
+    @Override
+    public void sendmail(String to, String subject, String message, String from, String replyto, String toname, String fromname, String errorsto, boolean extra) throws IOException {
+        mails.add(new TestMail(to, subject, message, from, replyto));
+    }
 
 }