src/org/ibex/mail/protocol/NNTP.java

   1 package org.ibex.mail.protocol;
   2 import org.ibex.util.*;
   3 import org.ibex.io.*;
   4 import org.ibex.mail.*;
   5 import org.ibex.mail.target.*;
   6 import org.ibex.jinetd.*;
   7 import java.io.*;
   8 import java.net.*;
   9 import java.util.*;
  10 import java.text.*;
  11
  12 /** NNTP send/recieve */
  13 public class NNTP {
  14
  15     // FIXME: command lines limited to 512 chars
  16
  17     public static class No  extends RuntimeException { int code = 0; }    // 4xx response codes
  18     public static class Bad extends RuntimeException { int code = 0; }    // 5xx response codes
  19
  20     public static class Group {
  21         public Group(String n, boolean p, int f, int l, int c) { this.name=n;this.post=p;this.first=f;this.last=l;this.count=c;}
  22         public final String  name;
  23         public final boolean post;
  24         public final int     first;
  25         public final int     last;
  26         public final int     count;
  27     }
  28
  29     public static class Article {
  30         public Article(String messageid, int num, Message message) { this.message=message;this.messageid=messageid;this.num=num;}
  31         public final String  messageid;
  32         public final int     num;
  33         public final Message message;
  34     }
  35
  36     public static interface Server {
  37         public Group    group(String s);
  38         public boolean  ihave(String messageid);
  39         public Article  next();
  40         public Article  last();
  41         public boolean  post(Message m);
  42         public Article  article(String messageid,  boolean head, boolean body);
  43         public Article  article(int    messagenum, boolean head, boolean body);
  44         public Group[]  list();
  45         public Group[]  newgroups(Date d, String[] distributions);
  46         public String[] newnews(String[] groups, Date d, String[] distributions);
  47     }
  48
  49     public static class MailboxWrapper implements Server {
  50         private final Mailbox root;
  51         private Mailbox current;
  52         private int ptr = 0;
  53         public MailboxWrapper(Mailbox root)     { this.root = root; }
  54         public Group    group(String s)         { ptr = 0; setgroup(s); return getgroup(s); }
  55         public boolean  ihave(String messageid) { /* FEATURE */ return false; }
  56         public boolean  post(Message m)         { /* FEATURE */ return false; }
  57         public Article  next()                  { return article(ptr++, false, false); }
  58         public Article  last()                  { return article(ptr--, false, false); }
  59         public Article  article(String i, boolean h, boolean b) { return article(Query.header("message-id",i),h,b); }
  60         public Article  article(int    n, boolean h, boolean b) { ptr = n; return article(Query.messagenum(n,n),h,b); }
  61         private Article article(Query q, boolean head, boolean body) {
  62             Mailbox.Iterator it = current.iterator(q);
  63             if (!it.next()) return null;
  64             Message m = body ? it.cur() : it.head();
  65             return new Article(m.messageid, it.num(), m);
  66         }
  67         public Group[]  list() { return list(root, ""); }
  68         private Group[] list(Mailbox who, String prefix) {
  69             Vec v = new Vec();
  70             String[] s = who.children();
  71             for(int i=0; i<s.length; i++) {
  72                 v.addElement(new Group(prefix + s[i], true, 0, 0, 0)); // FIXME numbers
  73                 Group[] g2 = list(who.slash(s[i], false), prefix + s[i] + ".");
  74                 for(int j=0; j<g2.length; j++) v.addElement(g2[j]);
  75             }
  76             Group[] ret = new Group[v.size()];
  77             v.copyInto(ret);
  78             return ret;
  79         }
  80
  81         private void setgroup(String s) {
  82             current = root;
  83             for(StringTokenizer st = new StringTokenizer(s, ".");
  84                 st.hasMoreTokens();
  85                 current = current.slash(st.nextToken(), false));
  86         }
  87         private Group getgroup(String s) {
  88             Mailbox box = root;
  89             for(StringTokenizer st = new StringTokenizer(s, "."); st.hasMoreTokens(); box = box.slash(st.nextToken(), false));
  90             return new Group(s, true, 0, box.count(Query.all()), box.count(Query.all()));
  91         }
  92
  93         public Group[]  newgroups(Date d, String[] distributions) { /* FEATURE */ return new Group[] { }; }
  94         public String[] newnews(String[] groups, Date d, String[] distributions) { /* FIXME */  return null; }
  95     }
  96
  97     public static class Listener implements Worker {
  98         private Server api;
  99         private Connection conn;
 100         private void article(String numOrMessageId, boolean head, boolean body) {
 101             String s = numOrMessageId.trim();
 102             Article a;
 103             if (s.startsWith("<")) a = api.article(s.substring(0, s.length() - 1), head, body);
 104             else                   a = api.article(Integer.parseInt(s), head, body);
 105             int code = (head && body) ? 220 : head ? 221 : body ? 222 : 223;
 106             conn.println(code + " " + a.num + " <" + a.messageid + "> get ready for some stuff...");
 107             if (head) conn.println(a.message.headers.raw);
 108             if (head && body) conn.println();
 109             if (body) conn.println(a.message.body);
 110             conn.println(".");
 111         }
 112         public void handleRequest(Connection conn) {
 113             this.conn = conn;
 114             api = new NNTP.MailboxWrapper(null);  // FIXME
 115             conn.setTimeout(30 * 60 * 1000);
 116             conn.println("200 " + conn.vhost + " [" + NNTP.class.getName() + "]");
 117             for(String line = conn.readln(); line != null; line = conn.readln()) try {
 118                 StringTokenizer st = new StringTokenizer(line, " ");
 119                 String command = st.nextToken().toUpperCase();
 120                 if        (command.equals("ARTICLE"))   { article(st.hasMoreTokens() ? st.nextToken() : null, true,  true);
 121                 } else if (command.equals("HEAD"))      { article(st.hasMoreTokens() ? st.nextToken() : null, true,  false);
 122                 } else if (command.equals("BODY"))      { article(st.hasMoreTokens() ? st.nextToken() : null, false, true);
 123                 } else if (command.equals("STAT"))      { article(st.hasMoreTokens() ? st.nextToken() : null, false, false);
 124                 } else if (command.equals("HELP"))      { conn.println("100 you are beyond help."); conn.println(".");
 125                 } else if (command.equals("SLAVE"))     { conn.println("220 I don't care");
 126                 } else if (command.equals("LAST"))      { Article a = api.last(); conn.println("223 "+a.num+" "+a.messageid+" ok");
 127                 } else if (command.equals("NEXT"))      { Article a = api.next(); conn.println("223 "+a.num+" "+a.messageid+" ok");
 128                 } else if (command.equals("QUIT"))      { conn.println("205 Bye."); conn.close(); return;
 129                 } else if (command.equals("GROUP"))     {
 130                     Group g = api.group(st.nextToken().toLowerCase());
 131                     conn.println("221 " + g.count + " " + g.first + " " + g.last + " " + g.name);
 132                 } else if (command.equals("NEWGROUPS") || command.equals("NEWNEWS")) {
 133                     String groups = command.equals("NEWNEWS") ? st.nextToken() : null;
 134                     String datetime = st.nextToken() + " " + st.nextToken();
 135                     String gmt = st.nextToken();
 136                     String distributions = gmt.equals("GMT") ? (st.hasMoreTokens() ? st.nextToken() : null) : gmt;
 137                     while(st.hasMoreTokens()) distributions += " " + st.nextToken();
 138                     // FIXME deal with GMT
 139                     Date d = new Date();
 140                     try {
 141                         d = new SimpleDateFormat("YYMMDD HHMMSS").parse(gmt);
 142                     } catch (ParseException pe) {
 143                         Log.warn(this, pe);
 144                     }
 145                     distributions = distributions.trim();
 146                     if (distributions.startsWith("<")) distributions = distributions.substring(1, distributions.length() - 1);
 147
 148                     st = new StringTokenizer(distributions, ",");
 149                     String[] dists = new String[st.countTokens()];
 150                     for(int i=0; st.hasMoreTokens(); i++) dists[i] = st.nextToken();
 151
 152                     if (command.equals("NEWGROUPS")) {
 153                         Group[] g = api.newgroups(d, dists);
 154                         conn.println("231 list of groups follows");
 155                         for(int i=0; i<g.length; i++)
 156                             conn.println(g[i].name + " " + g[i].last + " " + g[i].first + " " + (g[i].post ? "y" : "n"));
 157                         conn.println(".");
 158                     } else {
 159                         st = new StringTokenizer(groups, ",");
 160                         String[] g = new String[st.countTokens()];
 161                         for(int i=0; st.hasMoreTokens(); i++) g[i] = st.nextToken();
 162                         String[] a = api.newnews(g, d, dists);
 163                         conn.println("230 list of article messageids follows");
 164                         for(int i=0; i<a.length; i++) conn.println(a[i]);
 165                         conn.println(".");
 166                     }
 167
 168                 } else if (command.equals("POST"))      {
 169                     // FIXME
 170                     /*
 171                     boolean postok = api.post();
 172                     if (!postok) {
 173                         conn.println("440 no posting allowed");
 174                     } else {
 175                     */
 176                         conn.println("340 send the article");
 177                         // FIME read the article here
 178                         conn.println("240 article posted ok");
 179                         //}
 180
 181                 } else if (command.equals("LIST"))      {
 182                     Group[] g = api.list();
 183                     conn.println("215 list of groups follows");
 184                     for(int i=0; i<g.length; i++)
 185                         conn.println(g[i].name + " " + g[i].last + " " + g[i].first + " " + (g[i].post ? "y" : "n"));
 186                     conn.println(".");
 187
 188                 } else if (command.equals("IHAVE"))     {
 189                     boolean want = api.ihave(st.nextToken());
 190                     if (!want) {
 191                         conn.println("435 No thanks");
 192                     } else {
 193                         conn.println("335 Proceed");
 194                         // FIXME read article here
 195                         conn.println("235 Got it");
 196                     }
 197                 }
 198             } catch (No n)  { conn.println(n.code + " " + n.getMessage());
 199             } catch (Bad b) { conn.println(b.code + " " + b.getMessage()); Log.warn(this, b); }
 200             conn.close();
 201         }
 202     }
 203 }