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

   1 // Copyright 2000-2005 the Contributors, as shown in the revision logs.
   2 // Licensed under the Apache Public Source License 2.0 ("the License").
   3 // You may not use this file except in compliance with the License.
   4
   5 package org.ibex.mail.protocol;
   6 import org.ibex.util.*;
   7 import org.ibex.io.*;
   8 import org.ibex.mail.*;
   9 import org.ibex.mail.target.*;
  10 import org.ibex.jinetd.*;
  11 import java.io.*;
  12 import java.net.*;
  13 import java.util.*;
  14 import java.text.*;
  15
  16 /** NNTP send/recieve */
  17 public class NNTP {
  18
  19     // FIXME: command lines limited to 512 chars
  20
  21     public static class No  extends RuntimeException { int code = 400; }    // 4xx response codes
  22     public static class Bad extends RuntimeException { int code = 500; public Bad(String s) { super(s); } }    // 5xx response codes
  23
  24     public static class Group {
  25         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;}
  26         public final String  name;
  27         public final boolean post;
  28         public final int     first;
  29         public final int     last;
  30         public final int     count;
  31     }
  32
  33     public static class Article {
  34         public Article(String messageid, int num, Message message) { this.message=message;this.messageid=messageid;this.num=num;}
  35         public final String  messageid;
  36         public final int     num;
  37         public final Message message;
  38     }
  39
  40     public static interface Server {
  41         public Group    group(String s);
  42         public boolean  ihave(String messageid);
  43         public Article  next();
  44         public Article  last();
  45         public boolean  post(Message m);
  46         public Article  article(String messageid,  boolean head, boolean body);
  47         public Article  article(int    messagenum, boolean head, boolean body);
  48         public Group[]  list();
  49         public Group[]  newgroups(Date d, String[] distributions);
  50         public String[] newnews(String[] groups, Date d, String[] distributions);
  51     }
  52
  53     public static class MailboxWrapper implements Server {
  54         private final Mailbox root;
  55         private Mailbox current;
  56         private int ptr = 0;
  57         public MailboxWrapper(Mailbox root)     { this.root = root; }
  58         public Group    group(String s)         { ptr = 0; setgroup(s); return getgroup(s); }
  59         public boolean  ihave(String messageid) { /* FEATURE */ return false; }
  60         public boolean  post(Message m)         { /* FEATURE */ return false; }
  61         public Article  next()                  { return article(ptr++, false, false); }
  62         public Article  last()                  { return article(ptr--, false, false); }
  63         public Article  article(String i, boolean h, boolean b) { return article(Query.header("message-id",i),h,b); }
  64         public Article  article(int    n, boolean h, boolean b) { ptr = n; return article(Query.messagenum(n,n),h,b); }
  65         private Article article(Query q, boolean head, boolean body) {
  66             Mailbox.Iterator it = current.iterator(q);
  67             if (!it.next()) return null;
  68             try {
  69                 Message m = body ? it.cur() : Message.newMessage(new Stream(it.head() + "\r\n"));
  70                 //Message m = it.cur(); // FIXME
  71                 return new Article(m.messageid, it.num(), m);
  72             } catch (Exception e) { return null; }
  73         }
  74         public Group[]  list() { return list(root, ""); }
  75         private Group[] list(Mailbox who, String prefix) {
  76             Vec v = new Vec();
  77             if (who == null) who = root;
  78             String[] s = who.children();
  79             for(int i=0; i<s.length; i++) {
  80                 v.addElement(new Group(prefix + s[i], true, 0, 0, 0)); // FIXME numbers
  81                 Group[] g2 = list(who.slash(s[i], false), prefix + s[i] + ".");
  82                 for(int j=0; j<g2.length; j++) v.addElement(g2[j]);
  83             }
  84             Group[] ret = new Group[v.size()];
  85             v.copyInto(ret);
  86             return ret;
  87         }
  88
  89         private void setgroup(String s) {
  90             current = root;
  91             for(StringTokenizer st = new StringTokenizer(s, ".");
  92                 st.hasMoreTokens();
  93                 current = current.slash(st.nextToken(), false));
  94         }
  95         private Group getgroup(String s) {
  96             Mailbox box = root;
  97             for(StringTokenizer st = new StringTokenizer(s, "."); st.hasMoreTokens(); box = box.slash(st.nextToken(), false));
  98             return new Group(s, true, 1, box.count(Query.all()), box.count(Query.all()));
  99         }
 100
 101         public Group[]  newgroups(Date d, String[] distributions) { /* FEATURE */ return new Group[] { }; }
 102         public String[] newnews(String[] groups, Date d, String[] distributions) { /* FIXME */  return null; }
 103     }
 104
 105     public static class Listener implements Worker {
 106         private Server api = null;
 107         private Login login;
 108         private Connection conn;
 109         public Listener(Login l) { this.login = l; }
 110
 111         private void println(String s) { Log.warn("[nntp-write]", s); conn.println(s); }
 112         private void println() { Log.warn("[nntp-write]", ""); conn.println(""); }
 113
 114         private void article(String numOrMessageId, boolean head, boolean body) {
 115             String s = numOrMessageId.trim();
 116             Article a;
 117             if (s.startsWith("<")) a = api.article(s.substring(0, s.length() - 1), head, body);
 118             else                   a = api.article(Integer.parseInt(s), head, body);
 119             if (a == null) {
 120                 println("423 No such article.");
 121                 return;
 122             }
 123             int code = (head && body) ? 220 : head ? 221 : body ? 222 : 223;
 124             println(code + " " + a.num + " <" + a.messageid + "> get ready for some stuff...");
 125             if (head) println(a.message.headers.raw);
 126             if (head && body) println();
 127             if (body) println(a.message.body);
 128             println(".");
 129         }
 130         public void handleRequest(Connection conn) {
 131             this.conn = conn;
 132             conn.setTimeout(30 * 60 * 1000);
 133             println("200 " + conn.vhost + " [" + NNTP.class.getName() + "]");
 134             String user = null;
 135             String pass = null;
 136             Account account = login.anonymous();
 137             this.api = account == null ? null : new MailboxWrapper(account.getMailbox(NNTP.class));
 138             for(String line = conn.readln(); line != null; line = conn.readln()) try {
 139                 Log.warn("[nntp-read]", line);
 140                 StringTokenizer st = new StringTokenizer(line, " ");
 141                 String command = st.nextToken().toUpperCase();
 142                 if (command.equals("AUTHINFO")) {
 143                     String uop = st.nextToken().toUpperCase();
 144                     if (uop.equals("USER")) user = st.nextToken();
 145                     else if (uop.equals("PASS")) pass = st.nextToken();
 146                     // FIXME error here
 147                 }
 148                 if (this.api == null) {
 149                     if (user == null) { println("480 Authentication required"); continue; }
 150                     if (pass == null) { println("381 Password required"); continue; }
 151                     account = login.login(user, pass);
 152                     if (account == null) { println("502 Invalid"); continue; }
 153                     this.api = new MailboxWrapper(account.getMailbox(NNTP.class));
 154                     println("281 Good to go");
 155                     continue;
 156                 }
 157                 if        (command.equals("ARTICLE"))   { article(st.hasMoreTokens() ? st.nextToken() : null, true,  true);
 158                 } else if (command.equals("HEAD"))      { article(st.hasMoreTokens() ? st.nextToken() : null, true,  false);
 159                 } else if (command.equals("MODE"))      { println("201 Hello, you can post.");
 160                 } else if (command.equals("BODY"))      { article(st.hasMoreTokens() ? st.nextToken() : null, false, true);
 161                 } else if (command.equals("STAT"))      { article(st.hasMoreTokens() ? st.nextToken() : null, false, false);
 162                 } else if (command.equals("HELP"))      { println("100 you are beyond help."); println(".");
 163                 } else if (command.equals("SLAVE"))     { println("220 I don't care");
 164                 } else if (command.equals("XOVER"))     {
 165                     println("224 Overview information follows");
 166                     MailboxWrapper api = (MailboxWrapper)this.api;
 167                     String range = st.hasMoreTokens() ? st.nextToken() : (api.ptr+"-"+api.ptr);
 168                     int start = Integer.parseInt(range.substring(0, range.indexOf('-')));
 169                     int end   = Integer.parseInt(range.substring(range.indexOf('-') + 1));
 170                     Mailbox.Iterator it = api.current.iterator(Query.messagenum(start, end));
 171                     while(it.next()) {
 172                         Message m = it.cur();
 173                         println(it.num()+"\t"+m.subject+"\t"+m.from+"\t"+m.date+"\t"+m.messageid+"\t"+
 174                                 m.headers.gets("references") + "\t" + m.size() + "\t" + m.lines);
 175                     }
 176                     println(".");
 177                 } else if (command.equals("LAST"))      { Article a = api.last(); println("223 "+a.num+" "+a.messageid+" ok");
 178                 } else if (command.equals("NEXT"))      { Article a = api.next(); println("223 "+a.num+" "+a.messageid+" ok");
 179                 } else if (command.equals("QUIT"))      { println("205 Bye."); conn.close(); return;
 180                 } else if (command.equals("GROUP"))     {
 181                     Group g = api.group(st.nextToken().toLowerCase());
 182                     println("211 " + g.count + " " + g.first + " " + g.last + " " + g.name);
 183                 } else if (command.equals("NEWGROUPS") || command.equals("NEWNEWS")) {
 184                     String groups = command.equals("NEWNEWS") ? st.nextToken() : null;
 185                     String datetime = st.nextToken() + " " + st.nextToken();
 186                     String gmt = st.nextToken();
 187                     String distributions = gmt.equals("GMT") ? (st.hasMoreTokens() ? st.nextToken() : null) : gmt;
 188                     while(st.hasMoreTokens()) distributions += " " + st.nextToken();
 189                     // FIXME deal with GMT
 190                     Date d = new Date();
 191                     try {
 192                         d = new SimpleDateFormat("YYMMDD HHMMSS").parse(gmt);
 193                     } catch (ParseException pe) {
 194                         Log.warn(this, pe);
 195                     }
 196                     distributions = distributions.trim();
 197                     if (distributions.startsWith("<")) distributions = distributions.substring(1, distributions.length() - 1);
 198
 199                     st = new StringTokenizer(distributions, ",");
 200                     String[] dists = new String[st.countTokens()];
 201                     for(int i=0; st.hasMoreTokens(); i++) dists[i] = st.nextToken();
 202
 203                     if (command.equals("NEWGROUPS")) {
 204                         Group[] g = api.newgroups(d, dists);
 205                         println("231 list of groups follows");
 206                         for(int i=0; i<g.length; i++)
 207                             println(g[i].name + " " + g[i].last + " " + g[i].first + " " + (g[i].post ? "y" : "n"));
 208                         println(".");
 209                     } else {
 210                         st = new StringTokenizer(groups, ",");
 211                         String[] g = new String[st.countTokens()];
 212                         for(int i=0; st.hasMoreTokens(); i++) g[i] = st.nextToken();
 213                         String[] a = api.newnews(g, d, dists);
 214                         println("230 list of article messageids follows");
 215                         for(int i=0; i<a.length; i++) println(a[i]);
 216                         println(".");
 217                     }
 218
 219                 } else if (command.equals("POST"))      {
 220                     // FIXME
 221                     /*
 222                     boolean postok = api.post();
 223                     if (!postok) {
 224                         println("440 no posting allowed");
 225                     } else {
 226                     */
 227                         println("340 send the article");
 228                         // FIME read the article here
 229                         println("240 article posted ok");
 230                         //}
 231
 232                 } else if (command.equals("LIST"))      {
 233                     Group[] g = api.list();
 234                     println("215 list of groups follows");
 235                     for(int i=0; i<g.length; i++)
 236                         println(g[i].name + " " + g[i].last + " " + g[i].first + " " + (g[i].post ? "y" : "n"));
 237                     println(".");
 238
 239                 } else if (command.equals("IHAVE"))     {
 240                     boolean want = api.ihave(st.nextToken());
 241                     if (!want) {
 242                         println("435 No thanks");
 243                     } else {
 244                         println("335 Proceed");
 245                         // FIXME read article here
 246                         println("235 Got it");
 247                     }
 248                 }
 249             } catch (No n)  { println(n.code + " " + n.getMessage());
 250             } catch (Bad b) { println(b.code + " " + b.getMessage()); Log.warn(this, b); }
 251             conn.close();
 252         }
 253     }
 254 }