graylist/whitelist updates

[org.ibex.mail.git] / src / org / ibex / mail / Whitelist.java

diff --git a/src/org/ibex/mail/Whitelist.java b/src/org/ibex/mail/Whitelist.java
index 1fe679e..a9de5c5 100644 (file)
--- a/src/org/ibex/mail/Whitelist.java
+++ b/src/org/ibex/mail/Whitelist.java
@@ -3,6 +3,7 @@ package org.ibex.mail;
 import org.ibex.io.*;
 import org.ibex.mail.protocol.*;
 import org.ibex.util.*;
+import org.ibex.net.*;
 import java.sql.*;
 import java.net.*;
 import java.io.*;
@@ -10,78 +11,58 @@ import java.util.*;
 import java.sql.Timestamp;
 import java.sql.Connection;
 
-// now all I need is the click-through page
+public class Whitelist extends SqliteTable {
 
-// FIXME: periodic cleanup
-public class Whitelist {
-
-    private Connection conn;
-
-    // FIXME very ugly
-    static {
-        new Thread() { public void run() { startWebServer(); } }.start();
+    public Whitelist(String filename) {
+        super(filename,
+              new String[] {
+                  "create table if not exists 'whitelist' (email)",
+                  "create table if not exists 'pending' (spamid,email,message,date)"
+              },
+              "pending",
+              "date");
     }
-    public static void startWebServer() {
+
+    public synchronized boolean handleRequest(org.ibex.io.Connection c) {
         try {
-            ServerSocket ss = new ServerSocket(8025);
-            while(true) {
-                final Socket sock = ss.accept();
-                new Thread() {
-                    public void run() {
-                        try {
-                            BufferedReader br = new BufferedReader(new InputStreamReader(sock.getInputStream()));
-                            String s = br.readLine();
-                            String url = s.substring(s.indexOf(' ')+1);
-                            url = url.substring(0, url.indexOf(' '));
-                            while(s!=null && !s.equals(""))
-                                s = br.readLine();
-                            PrintWriter pw = new PrintWriter(new OutputStreamWriter(sock.getOutputStream()));
-                            if (!url.startsWith("/whitelist/")) {
-                                pw.print("HTTP/1.0 404 Not FoundK\r\n");
-                                pw.print("Content-Type: text/plain\r\n");
-                                pw.print("\r\n");
-                                pw.println("you are lost.");
-                            } else {
-                                url = url.substring("/whitelist/".length());
-                                url = URLDecoder.decode(url);
-                                pw.print("HTTP/1.0 200 OK\r\n");
-                                pw.print("Content-Type: text/plain\r\n");
-                                pw.print("\r\n");
-                                try {
-                                    SMTP.whitelist.response(url);
-                                    pw.println("Thanks!  You've been added to my list of non-spammers and your message");
-                                    pw.println("has been moved to my inbox.");
-                                    pw.println("email id " + url);
-                                    pw.println("");
-                                } catch (Exception e) {
-                                    e.printStackTrace(pw);
-                                }
-                            }
-                            pw.flush();
-                            sock.close();
-                        } catch (Exception e) { throw new RuntimeException(e); }
-                    }
-                }.start();
+            Socket sock = c.getSocket();
+            BufferedReader br = new BufferedReader(new InputStreamReader(sock.getInputStream()));
+            String s = br.readLine();
+            String url = s.substring(s.indexOf(' ')+1);
+            url = url.substring(0, url.indexOf(' '));
+            while(s!=null && !s.equals(""))
+                s = br.readLine();
+            PrintWriter pw = new PrintWriter(new OutputStreamWriter(sock.getOutputStream()));
+            if (!url.startsWith("/whitelist/")) {
+                pw.print("HTTP/1.0 404 Not FoundK\r\n");
+                pw.print("Content-Type: text/plain\r\n");
+                pw.print("\r\n");
+                pw.println("you are lost.");
+            } else {
+                url = url.substring("/whitelist/".length());
+                url = URLDecoder.decode(url);
+                pw.print("HTTP/1.0 200 OK\r\n");
+                pw.print("Content-Type: text/plain\r\n");
+                pw.print("\r\n");
+                try {
+                    SMTP.whitelist.response(url);
+                    pw.println("Thanks!  You've been added to my list of non-spammers and your message");
+                    pw.println("has been moved to my inbox.");
+                    pw.println("email id " + url);
+                    pw.println("");
+                } catch (Exception e) {
+                    e.printStackTrace(pw);
+                }
             }
+            pw.flush();
+            sock.close();
         } catch (Exception e) { throw new RuntimeException(e); }
-    }
-
-
-    public Whitelist(String filename) {
-        try {
-            Class.forName("org.sqlite.JDBC");
-            conn = DriverManager.getConnection("jdbc:sqlite:"+filename);
-            conn.prepareStatement("create table if not exists "+
-                                  "'whitelist' (email)").executeUpdate();
-            conn.prepareStatement("create table if not exists "+
-                                  "'pending' (spamid,email,message,date)").executeUpdate();
-        }
-        catch (SQLException e) { throw new RuntimeException(e); }
-        catch (ClassNotFoundException e) { throw new RuntimeException(e); }
+        return true;
     }
 
     public synchronized boolean isWhitelisted(Address a) {
         try {
+            if (a==null) return false;
             PreparedStatement check = conn.prepareStatement("select * from 'whitelist' where email=?");
             check.setString(1, a.toString(false).toLowerCase());
             ResultSet rs = check.executeQuery();