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

   1 package org.ibex.mail.protocol;
   2 import org.ibex.io.*;
   3 import org.ibex.crypto.*;
   4 import org.ibex.jinetd.Listener;
   5 import org.ibex.jinetd.Worker;
   6 import org.ibex.mail.*;
   7 import org.ibex.util.*;
   8 import org.ibex.mail.target.*;
   9 import java.util.*;
  10 import java.net.*;
  11 import java.text.*;
  12 import java.io.*;
  13
  14 // FIXME: this is valid    LSUB "" asdfdas%*%*%*%*SFEFGWEF
  15 // FIXME: be very careful about where/when we quotify stuff
  16 // FIXME: 'UID FOO 100:*' must match at least one message even if all UIDs less than 100
  17 // FIXME: LIST should return INBOX if applicable
  18
  19 // Relevant RFC's:
  20 //   RFC 2060: IMAPv4
  21 //   RFC 3501: IMAPv4 with clarifications
  22 //   RFC 3691: UNSELECT
  23 //   RFC 2971: ID
  24
  25 // FEATURE: make sure we're being case-insensitive enough; probably want to upcase atom()
  26 // FEATURE: MIME-queries and BODYSTRUCTURE
  27 // FEATURE: READ-WRITE / READ-ONLY status on SELECT
  28 // FEATURE: pipelining
  29 // FEATURE: support [charset]
  30 // FEATURE: \Noselect
  31 // FEATURE: subscriptions
  32 // FEATURE: tune for efficiency
  33 // FEATURE: STARTTLS
  34 // FEATURE: asynchronous client notifications (need to re-read RFC)
  35
  36 public class IMAP {
  37
  38     public IMAP() { }
  39     public static final float version = (float)0.2;
  40
  41     // API Class //////////////////////////////////////////////////////////////////////////////
  42
  43     public static interface Client {
  44         public void expunge(int uid);
  45         public void list(char separator, String mailbox, boolean lsub, boolean phantom);
  46         public void fetch(int num, int flags, int size, Message m, int muid);  // m may be null or incomplete
  47     }
  48
  49     public static interface Server {
  50         public void      setClient(IMAP.Client client);
  51         public String[]  capability();
  52         public Hashtable id(Hashtable clientId);
  53         public void      copy(Query q, String to);
  54         public void      logout();
  55         public void      unselect();
  56         public void      delete(String m);
  57         public void      create(String m);
  58         public void      append(String m, int flags, Date arrival, String body);
  59         public void      check();
  60         public void      noop();
  61         public void      close();
  62         public void      subscribe(String mailbox);
  63         public void      unsubscribe(String mailbox);
  64         public int       unseen(String mailbox);
  65         public int       recent(String mailbox);
  66         public int       count(String mailbox);
  67         public int       uidNext(String mailbox);
  68         public int       uidValidity(String mailbox);
  69         public int[]     search(Query q, boolean uid);
  70         public void      rename(String from, String to);
  71         public void      select(String mailbox, boolean examineOnly);
  72         public void      setFlags(Query q, int flags, boolean uid, boolean silent);
  73         public void      removeFlags(Query q, int flags, boolean uid, boolean silent);
  74         public void      addFlags(Query q, int flags, boolean uid, boolean silent);
  75         public void      expunge();
  76         public void      fetch(Query q, int spec, String[] headers, int start, int end, boolean uid);
  77         public void      lsub(String start, String ref);
  78         public void      list(String start, String ref);
  79         public static class Exn extends MailException { public Exn(String s) { super(s); } }
  80         public static class Bad extends Exn { public Bad(String s) { super(s); } }
  81         public static class No extends Exn { public No(String s) { super(s); } }
  82     }
  83
  84
  85     // MailboxWrapper //////////////////////////////////////////////////////////////////////////////
  86
  87     /** wraps an IMAP.Server interface around a Mailbox */
  88     public static class MailboxWrapper implements Server {
  89
  90         public static final char sep = '.';
  91
  92         Mailbox inbox = null;
  93         Mailbox selected = null;
  94         Mailbox root = null;
  95         Mailbox selected() { if (selected == null) throw new Bad("no mailbox selected"); return selected; }
  96         final Login auth;
  97         final Client client;
  98
  99         public MailboxWrapper(Login auth, Client c) { this.auth=auth; this.client=c;}
 100         public void setClient(IMAP.Client client) { }
 101
 102         private Mailbox mailbox(String name, boolean create) { return mailbox(name, create, true); }
 103         private Mailbox mailbox(String name, boolean create, boolean throwexn) {
 104             if (name.equalsIgnoreCase("inbox")) return inbox;
 105             Mailbox m = root;
 106             for(StringTokenizer st = new StringTokenizer(name, sep + ""); st.hasMoreTokens();)
 107                 if ((m = m.slash(st.nextToken(), create)) == null) {
 108                     if (throwexn) throw new Server.No("no such mailbox " + name);
 109                     return null;
 110                 }
 111             return m;
 112         }
 113
 114         // FEATURE: not accurate when a wildcard and subsequent non-wildcards both match a single component
 115         public void lsub(String start, String ref) { list(start, ref, true); }
 116         public void list(String start, String ref) { list(start, ref, false); }
 117         public void list(String start, String ref, boolean lsub) {
 118             if (ref.length() == 0) { client.list(sep, start, lsub, false); return; }
 119             while (start.endsWith(""+sep)) start = start.substring(0, start.length() - 1);
 120             if (ref.endsWith("%")) ref = ref + sep;
 121             String[] children = (start.length() == 0 ? root : mailbox(start, false)).children();
 122             for(int i=0; i<children.length; i++) {
 123                 String s = children[i], pre = ref, kid = start + (start.length() > 0 ? sep+"" : "") + s;
 124                 if (mailbox(kid, false) == null) continue;
 125                 boolean phantom = mailbox(kid, false).phantom();
 126                 while(true) {
 127                     if (pre.length() == 0) {
 128                         if (s.length() == 0)       client.list(sep, kid, lsub, phantom);
 129                     } else switch(pre.charAt(0)) {
 130                         case sep:        if (s.length() == 0) list(kid, pre.substring(1), lsub);                    break;
 131                         case '%':        client.list(sep,kid,lsub,phantom);pre=pre.substring(1); s = "";            continue;
 132                         case '*':        client.list(sep,kid,lsub,phantom);list(kid,pre,lsub);pre=pre.substring(1); break;
 133                         default:         if (s.length()==0)                                                         break;
 134                                          if (s.charAt(0) != pre.charAt(0))                                          break;
 135                                          s = s.substring(1); pre = pre.substring(1);                                continue;
 136                     }
 137                     break;
 138                 }
 139             }
 140         }
 141
 142         public String[] capability() { return new String[] { "IMAP4rev1" , "UNSELECT", "ID" }; }
 143         public Hashtable id(Hashtable clientId) {
 144             Hashtable response = new Hashtable();
 145             response.put("name", IMAP.class.getName());
 146             response.put("version", version + "");
 147             response.put("os", System.getProperty("os.name", null));
 148             response.put("os-version", System.getProperty("os.version", null));
 149             response.put("vendor", "none");
 150             response.put("support-url", "http://mail.ibex.org/");
 151             return response;
 152         }
 153
 154         public void copy(Query q, String to) { copy(q, mailbox(to, false)); }
 155         /*  */ void copy(Query q, Mailbox to) {
 156             for(Mailbox.Iterator it=selected().iterator(q);it.next();) to.add(it.cur(), it.flags() | Mailbox.Flag.RECENT); }
 157
 158         public void unselect() { selected = null; }
 159         public void delete(String m0) { delete(mailbox(m0,false)); }
 160         public void delete(Mailbox m) { if (!m.equals(inbox)) m.destroy(false); else throw new Bad("can't delete inbox"); }
 161         public void create(String m) { mailbox(m, true, false); }
 162         public void append(String m,int f,Date a,String b) { try {
 163             mailbox(m,false).add(new Message(new Stream(b), new Message.Envelope(null,null,a)),f|Mailbox.Flag.RECENT);
 164         } catch (Message.Malformed e) { throw new No(e.getMessage()); } }
 165         public void check() { }
 166         public void noop() { }
 167         public void logout() { }
 168         public void close() { for(Mailbox.Iterator it=selected().iterator(Query.deleted()); it.next();) it.delete(); }
 169         public void expunge() { for(Mailbox.Iterator it = selected().iterator(Query.deleted());it.next();) expunge(it); }
 170         public void expunge(Mailbox.Iterator it) { client.expunge(it.uid()); it.delete(); }
 171         public void subscribe(String mailbox) { }
 172         public void unsubscribe(String mailbox) { }
 173         public int unseen(String mailbox)      { return mailbox(mailbox, false).count(Query.not(Query.seen())); }
 174         public int recent(String mailbox)      { return mailbox(mailbox, false).count(Query.recent()); }
 175         public int count(String mailbox)       { return mailbox(mailbox, false).count(Query.all()); }
 176         public int uidNext(String mailbox)     { return mailbox(mailbox, false).uidNext(); }
 177         public int uidValidity(String mailbox) { return mailbox(mailbox, false).uidValidity(); }
 178         public void select(String mailbox, boolean examineOnly) {
 179             selected = mailbox(mailbox, false);
 180         }
 181
 182         public int[] search(Query q, boolean uid) {
 183             Vec.Int vec = new Vec.Int();
 184             for(Mailbox.Iterator it = selected().iterator(q); it.next();) {
 185                 vec.addElement(uid ? it.uid() : it.num());
 186                 it.recent(false);
 187             }
 188             return vec.dump();
 189         }
 190
 191         public void setFlags(Query q, int f, boolean uid, boolean silent)    { doFlags(q, f, uid, 0, silent); }
 192         public void addFlags(Query q, int f, boolean uid, boolean silent)    { doFlags(q, f, uid, 1, silent); }
 193         public void removeFlags(Query q, int f, boolean uid, boolean silent) { doFlags(q, f, uid, -1, silent); }
 194
 195         private void doFlags(Query q, int flags, boolean uid, int style, boolean silent) {
 196             for(Mailbox.Iterator it = selected().iterator(q);it.next();) {
 197                 boolean recent = it.recent();
 198                 if (style == -1)     it.removeFlags(flags);
 199                 else if (style == 0) it.setFlags(flags);
 200                 else if (style == 1) it.addFlags(flags);
 201                 it.recent(recent);
 202                 // FIXME
 203                 //if (!silent) client.fetch(it.num(), it.flags(), -1, null, it.uid());
 204             }
 205         }
 206         public void rename(String from0, String to) {
 207             Mailbox from = mailbox(from0, false);
 208             if (from.equals(inbox))                { from.copy(Query.all(), mailbox(to, true)); }
 209             else if (to.equalsIgnoreCase("inbox")) { from.copy(Query.all(), mailbox(to, true)); from.destroy(false); }
 210             else from.rename(to);
 211         }
 212         public void fetch(Query q, int spec, String[] headers, int start, int end, boolean uid) {
 213             for(Mailbox.Iterator it = selected().iterator(q); it.next(); ) {
 214                 Message message = ((spec & (BODYSTRUCTURE | ENVELOPE | INTERNALDATE | FIELDS | FIELDSNOT | RFC822 |
 215                                             RFC822TEXT | RFC822SIZE | HEADERNOT | HEADER)) != 0) ? it.cur() : null;
 216                 int size = message == null ? 0 : message.size();
 217                 client.fetch(it.num(), it.flags(), size, message, it.uid());
 218                 it.recent(false);
 219             }
 220         }
 221     }
 222
 223
 224     // Single Session Handler //////////////////////////////////////////////////////////////////////////////
 225
 226     /** takes an IMAP.Server and exposes it to the world as an IMAP server on a TCP socket */
 227     public static class Listener implements Worker, Client {
 228         String selectedName = null;
 229         Mailbox inbox = null, root = null;
 230         Server api;
 231         Parser parser = null;
 232         Connection conn = null;
 233         Login auth;
 234         public Listener(Login auth) { api = new IMAP.MailboxWrapper(this.auth = auth, this); }
 235         Parser.Token token() { return parser.token(); }
 236         void println(String s) { conn.println(s); }
 237         void newline() { parser.newline(); }
 238         Query query() { return parser.query(); }
 239
 240         public void login(String u, String p) {
 241             Object ret;
 242             if ((ret = auth.login(u,p,IMAP.class)) == null) throw new Server.No("Login failed.");
 243             if (ret instanceof IMAP.Server) {
 244                 api = (IMAP.Server)ret;
 245                 api.setClient(this);
 246             } else {
 247                 Account account = (Account)ret;
 248                 root = account.getMailbox(IMAP.class);
 249                 inbox = root.slash("INBOX", false);
 250                 if (inbox == null) inbox = root;
 251             }
 252         }
 253
 254         public void handleRequest(Connection conn) {
 255             this.conn = conn;
 256             parser = new Parser(conn);
 257             conn.setTimeout(30 * 60 * 1000);
 258             println("* OK " + conn.vhost + " " + IMAP.class.getName() + " IMAP4rev1 [RFC3501] v" + version + " server ready");
 259             for(String tag = null;; newline()) try {
 260                 conn.flush();
 261                 boolean uid = false;
 262                 tag = null; Parser.Token tok = token(); if (tok == null) return; tag = tok.astring();
 263                 String command = token().atom();
 264                 if (command.equalsIgnoreCase("UID")) { uid = true; command = token().atom(); }
 265                 int commandKey = ((Integer)commands.get(command.toUpperCase())).intValue();
 266                 switch(commandKey) {
 267                     case LOGIN:        login(token().astring(), token().astring()); break;
 268                     case CAPABILITY:   println("* CAPABILITY " + Printer.join(" ", api.capability())); break;
 269                     case AUTHENTICATE: throw new Server.No("AUTHENTICATE not supported");
 270                     case LOGOUT:       api.logout(); println("* BYE"); conn.close(); return;
 271                     case LIST:         api.list(token().q(), token().astring()); break; /*astring is a hack for EUDORA*/
 272                     case LSUB:         api.lsub(token().q(), token().q()); break;
 273                     case SUBSCRIBE:    api.subscribe(token().astring()); break;
 274                     case UNSUBSCRIBE:  api.unsubscribe(token().astring()); break;
 275                     case RENAME:       api.rename(token().astring(), token().astring()); break;
 276                     case COPY:         selected(); api.copy(Query.set(uid, token().set()), token().astring()); break;
 277                     case DELETE:       api.delete(token().atom()); break;
 278                     case CHECK:        selected(); api.check(); break;
 279                     case NOOP:         api.noop(); break;
 280                     case CLOSE:        selected(); api.close(); break;
 281                     case EXPUNGE:      selected(); api.expunge(); break;
 282                     case UNSELECT:     selected(); api.unselect(); selected = false; break;
 283                     case CREATE:       api.create(token().astring()); break;
 284                     case FETCH:        selected(); fetch(Query.set(lastuid=uid, token().set()),
 285                                                          lastfetch=token().lx(), 0, 0, 0, uid, 0); break;
 286                     case SEARCH:       selected(); println("* SEARCH " + Printer.join(api.search(query(), uid))); break;
 287                     case EXAMINE:
 288                     case SELECT: {
 289                         String mailbox = token().astring();
 290                         api.select(mailbox, commandKey==EXAMINE);
 291                         println("* FLAGS (\\Answered \\Flagged \\Deleted \\Seen \\Draft)");
 292                         println("* " + api.count(mailbox)  + " EXISTS");
 293                         println("* " + api.recent(mailbox) + " RECENT");
 294                         println("* OK [UIDVALIDITY " + api.uidValidity(mailbox) + "] UIDs valid");
 295                         println("* OK [UIDNEXT " + api.uidNext(mailbox) + "]");
 296                         println("* OK [PERMANENTFLAGS (\\Seen \\Draft \\Answered \\Deleted)]");
 297                         selected = true;
 298                         break; }
 299                     case STATUS: {
 300                         String mailbox = token().astring();  // hack for GNUS buggy client
 301                         Parser.Token[] list = token().l();
 302                         String response = "";
 303                         for(int i=0; i<list.length; i++) {
 304                             String s = list[i].atom().toUpperCase();
 305                             if (i>0) response += " ";
 306                             if (s.equals("MESSAGES"))    response += "MESSAGES "    + api.count(mailbox);
 307                             if (s.equals("RECENT"))      response += "RECENT "      + api.recent(mailbox);
 308                             if (s.equals("UIDNEXT"))     response += "UIDNEXT "     + api.uidNext(mailbox);
 309                             if (s.equals("UIDVALIDITY")) response += "UIDVALIDITY " + api.uidValidity(mailbox);
 310                             if (s.equals("UNSEEN"))      response += "UNSEEN "      + api.unseen(mailbox);
 311                         }
 312                         println("* STATUS " + mailbox + " (" + response + ")");
 313                         break;
 314                     }
 315                     case APPEND: {
 316                         String m = token().astring();
 317                         int flags = 0;
 318                         Date arrival = new Date();
 319                         Parser.Token t = token();
 320                         if (t.type == t.LIST)   { flags = t.flags();      t = token(); }
 321                         if (t.type != t.QUOTED) { arrival = t.datetime(); t = token(); }
 322                         api.append(m, flags, arrival, t.q());
 323                         break; }
 324                     case STORE: {
 325                         selected();
 326                         Query q = uid ? Query.uid(token().set()) : Query.num(token().set());
 327                         String s = token().atom().toUpperCase();
 328                         int flags = token().flags();
 329                         if (s.equals("FLAGS"))              api.setFlags(q,    flags, uid, false);
 330                         else if (s.equals("+FLAGS"))        api.addFlags(q,    flags, uid, false);
 331                         else if (s.equals("-FLAGS"))        api.removeFlags(q, flags, uid, false);
 332                         else if (s.equals("FLAGS.SILENT"))  api.setFlags(q,    flags, uid, true);
 333                         else if (s.equals("+FLAGS.SILENT")) api.addFlags(q,    flags, uid, true);
 334                         else if (s.equals("-FLAGS.SILENT")) api.removeFlags(q, flags, uid, true);
 335                         else throw new Server.Bad("unknown STORE specifier " + s);
 336                         break; }
 337                     default: throw new Server.Bad("unrecognized command \"" + command + "\"");
 338                 }
 339                 println(tag+" OK "+command+" Completed. " +
 340                         (commandKey == LOGIN ? ("[CAPABILITY "+Printer.join(" ", api.capability())+"]") : ""));
 341             } catch (Server.Bad b) { println(tag==null ? "* BAD Invalid tag":(tag + " Bad " + b.toString())); Log.warn(this,b);
 342             } catch (Server.No n)  { println(tag==null?"* BAD Invalid tag":(tag+" No "  + n.toString())); Log.warn(this,n); }
 343         }
 344
 345         private Parser.Token[] lastfetch = null; // hack
 346         private boolean lastuid = false;  // hack
 347         private boolean selected = false;
 348         private void selected() { if (!selected) throw new Server.Bad("no mailbox selected"); }
 349
 350         // Callbacks //////////////////////////////////////////////////////////////////////////////
 351
 352         public void expunge(int uid) { println("* " + uid + " EXPUNGE"); }
 353         public void fetch(int n, int f, int size, Message m, int muid) { fetch(m, lastfetch, n, f, size, lastuid, muid); }
 354         public void list(char sep, String mb, boolean sub, boolean p) {
 355             println("* " + (sub?"LSUB":"LIST")+" ("+(p?"\\Noselect":"")+") \""+sep+"\" \""+mb+"\"");}
 356
 357         /**
 358          *   Parse a fetch request <i>or</i> emit a fetch reply.
 359          *
 360          *   To avoid duplicating tedious parsing logic, this function
 361          *   performs both of the following tasks:
 362          *      - parse the fetch request in Token[] t and return a fetch spec
 363          *      - emit a fetch reply for the parsed spec with respect to message m
 364          */
 365         private void fetch(Object o, Parser.Token[] t, int num, int flags, int size, boolean uid, int muid) {
 366             Query q   = o instanceof Query ? (Query)o : null;
 367             Message m = o instanceof Message ? (Message)o : null;
 368             boolean e = q == null;
 369
 370             lastfetch = t;
 371             int spec = 0;                              // spec; see constants for flags
 372             String[] headers = null;
 373             int start = -1, end = -1;
 374             StringBuffer r = new StringBuffer();
 375             if (e) { r.append(uid ? muid : num); r.append(" FETCH ("); }
 376             int initlen = r.length();
 377             if (uid) {
 378                 boolean good = false;
 379                 for(int i=0; i<t.length; i++)
 380                     if ((t[i].type == Parser.Token.QUOTED || t[i].type == Parser.Token.ATOM) &&
 381                         t[i].astring().equalsIgnoreCase("UID")) good = true;
 382                 if (!good) {
 383                     Parser.Token[] t2 = new Parser.Token[t.length + 1];
 384                     System.arraycopy(t, 0, t2, 0, t.length);
 385                     t2[t2.length - 1] = parser.token("UID");
 386                     lastfetch = (t = t2);
 387                 }
 388             }
 389             if (t.length == 0 && (t[0].type == Parser.Token.QUOTED || t[0].type == Parser.Token.ATOM)) {
 390                 if (t[0].astring().equalsIgnoreCase("ALL"))
 391                     t = new Parser.Token[] { parser.token("FLAGS"), parser.token("INTERNALDATE"),
 392                                       parser.token("ENVELOPE"), parser.token("RFC822.SIZE") };
 393                 else if (t[0].astring().equalsIgnoreCase("FULL"))
 394                     t = new Parser.Token[] { parser.token("FLAGS"), parser.token("INTERNALDATE"), parser.token("BODY"),
 395                                       parser.token("ENVELOPE"), parser.token("RFC822.SIZE") };
 396                 else if (t[0].astring().equalsIgnoreCase("FAST"))
 397                     t = new Parser.Token[] { parser.token("FLAGS"), parser.token("INTERNALDATE"),
 398                                       parser.token("RFC822.SIZE") };
 399             }
 400             for(int i=0; i<t.length; i++) {
 401                 if (r.length() > initlen) r.append(" ");
 402                 if (t[i] == null || t[i].s == null) continue;
 403                 String s = t[i].s.toUpperCase();
 404                 r.append(s.equalsIgnoreCase("BODY.PEEK")?"BODY":s);
 405                 if (s.equals("BODYSTRUCTURE")) {       spec|=BODYSTRUCTURE;if(e){r.append(" ");r.append(Printer.bodystructure(m));}
 406                 } else if (s.equals("ENVELOPE")) {     spec|=ENVELOPE;     if(e){r.append(" ");r.append(Printer.envelope(m));}
 407                 } else if (s.equals("FLAGS")) {        spec|=FLAGS;        if(e){r.append(" ");r.append(Printer.flags(flags));}
 408                 } else if (s.equals("INTERNALDATE")) { spec|=INTERNALDATE; if(e){r.append(" ");r.append(Printer.date(m.envelope.arrival));}
 409                 } else if (s.equals("RFC822")) {       spec|=RFC822;       if(e){r.append(" ");r.append(Printer.message(m));}
 410                 } else if (s.equals("RFC822.TEXT")) {  spec|=RFC822TEXT;   if(e){r.append(" ");r.append(Printer.qq(m.body));}
 411                 } else if (s.equals("RFC822.HEADER")){ spec|=HEADER;if(e){r.append(" ");r.append(Printer.qq(m.headers.raw+"\r\n"));}
 412                 } else if (s.equals("RFC822.SIZE")) {  spec|=RFC822SIZE;   if(e){r.append(" ");r.append(m.size());}
 413                 } else if (s.equals("UID")) {          spec|=UID;          if(e){r.append(" ");r.append(muid); }
 414                 } else if (!(s.equals("BODY.PEEK") || s.equals("BODY"))) { throw new Server.No("unknown fetch argument: " + s);
 415                 } else {
 416                     if (s.equalsIgnoreCase("BODY.PEEK"))   spec |= PEEK;
 417                     //else if (e) api.addFlags(Query.num(new int[] { num, num }), Mailbox.Flag.SEEN, false, false);
 418                     if (i >= t.length - 1 || t[i+1].type != Parser.Token.LIST) {
 419                         spec |= BODYSTRUCTURE;
 420                         if (e) { r.append(" "); r.append(Printer.bodystructure(m)); } continue;
 421                         //{ if (e) { r.append(" "); r.append(Printer.qq(m.body)); } continue; }
 422                     }
 423                     String payload = "";
 424                     r.append("[");
 425                     Parser.Token[] list = t[++i].l();
 426                     s = list.length == 0 ? "" : list[0].s.toUpperCase();
 427                     r.append(s);
 428                     if (list.length == 0)                   { spec |= RFC822TEXT;   if(e) payload = m.headers.raw+"\r\n"+m.body; }
 429                     else if (s.equals("") || s.equals("1")) { spec |= RFC822TEXT;   if(e) payload = m.headers.raw+"\r\n"+m.body; }
 430                     else if (s.equals("TEXT"))              { spec |= RFC822TEXT;   if(e) payload = m.body; }
 431                     else if (s.equals("HEADER"))            { spec |= HEADER;       if(e) payload = m.headers.raw+"\r\n"; }
 432                     else if (s.equals("HEADER.FIELDS"))     { spec |= FIELDS;     payload=headers(r,t[i].l()[1].sl(),false,m,e); }
 433                     else if (s.equals("HEADER.FIELDS.NOT")) { spec |= FIELDSNOT;  payload=headers(r,t[i].l()[1].sl(),true,m,e); }
 434                     else if (s.equals("MIME")) {              throw new Server.Bad("MIME not supported"); }
 435                     else                                      throw new Server.Bad("unknown section type " + s);
 436                     if (i<t.length - 1 && (t[i+1].s != null && t[i+1].s.startsWith("<"))) {
 437                         i++;
 438                         s = t[i].s.substring(1, t[i].s.indexOf('>'));
 439                         int dot = s.indexOf('.');
 440                         start = dot == -1 ? Integer.parseInt(s) : Integer.parseInt(s.substring(0, s.indexOf('.')));
 441                         end = dot == -1 ? -1 : Integer.parseInt(s.substring(s.indexOf('.') + 1));
 442                         if (e) { payload = payload.substring(start, Math.min(end+1,payload.length())); r.append("<"+start+">"); }
 443                     }
 444                     if (e) { r.append("] "); r.append(Printer.qq(payload)); }
 445                 }
 446             }
 447             if (e) {
 448                 r.append(")");
 449                 println("* " + r.toString());
 450             } else {
 451                 api.fetch(q, spec, headers, start, end, uid);
 452             }
 453         }
 454
 455         private String headers(StringBuffer r, String[] headers, boolean negate, Message m, boolean e) {
 456             String payload = "";
 457             if (e) r.append(" (");
 458             if (!negate) {
 459                 if(e) for(int j=0; j<headers.length; j++) {
 460                     r.append(headers[j] + (j<headers.length-1?" ":""));
 461                     if (m.headers.gets(headers[j]) != null) payload += headers[j]+": "+m.headers.gets(headers[j])+"\r\n";
 462                 }
 463             } else {
 464                 throw new Server.No("HEADERS.NOT temporarily disaled");
 465                 /*
 466                 if (e) for(int j=0; j<headers.length; j++) r.append(headers[j] + (j<headers.length-1?" ":""));
 467                 if(e) { OUTER: for(Enumeration x=m.headers.keys(); x.hasMoreElements();) {
 468                     String key = (String)x.nextElement();
 469                     for(int j=0; j<headers.length; j++) if (key.equalsIgnoreCase(headers[j])) continue OUTER;
 470                     payload += key + ": " + m.headers.gets(key)+"\r\n";
 471                 } }
 472                 */
 473             }
 474             if (e) r.append(")");
 475             return payload + "\r\n";
 476         }
 477
 478         private static final Hashtable commands = new Hashtable();
 479         private static final int UID = 0;          static { commands.put("UID", new Integer(UID)); }
 480         private static final int AUTHENTICATE = 1; static { commands.put("AUTHENTICATE", new Integer(AUTHENTICATE)); }
 481         private static final int LIST = 2;         static { commands.put("LIST", new Integer(LIST)); }
 482         private static final int LSUB = 3;         static { commands.put("LSUB", new Integer(LSUB)); }
 483         private static final int SUBSCRIBE = 4;    static { commands.put("SUBSCRIBE", new Integer(SUBSCRIBE)); }
 484         private static final int UNSUBSCRIBE = 5;  static { commands.put("UNSUBSCRIBE", new Integer(UNSUBSCRIBE)); }
 485         private static final int CAPABILITY = 6;   static { commands.put("CAPABILITY", new Integer(CAPABILITY)); }
 486         private static final int ID = 7;           static { commands.put("ID", new Integer(ID)); }
 487         private static final int LOGIN = 8;        static { commands.put("LOGIN", new Integer(LOGIN)); }
 488         private static final int LOGOUT = 9;       static { commands.put("LOGOUT", new Integer(LOGOUT)); }
 489         private static final int RENAME = 10;      static { commands.put("RENAME", new Integer(RENAME)); }
 490         private static final int EXAMINE = 11;     static { commands.put("EXAMINE", new Integer(EXAMINE)); }
 491         private static final int SELECT = 12;      static { commands.put("SELECT", new Integer(SELECT)); }
 492         private static final int COPY = 13;        static { commands.put("COPY", new Integer(COPY)); }
 493         private static final int DELETE = 14;      static { commands.put("DELETE", new Integer(DELETE)); }
 494         private static final int CHECK = 15;       static { commands.put("CHECK", new Integer(CHECK)); }
 495         private static final int NOOP = 16;        static { commands.put("NOOP", new Integer(NOOP)); }
 496         private static final int CLOSE = 17;       static { commands.put("CLOSE", new Integer(CLOSE)); }
 497         private static final int EXPUNGE = 18;     static { commands.put("EXPUNGE", new Integer(EXPUNGE)); }
 498         private static final int UNSELECT = 19;    static { commands.put("UNSELECT", new Integer(UNSELECT)); }
 499         private static final int CREATE = 20;      static { commands.put("CREATE", new Integer(CREATE)); }
 500         private static final int STATUS = 21;      static { commands.put("STATUS", new Integer(STATUS)); }
 501         private static final int FETCH = 22;       static { commands.put("FETCH", new Integer(FETCH)); }
 502         private static final int APPEND = 23;      static { commands.put("APPEND", new Integer(APPEND)); }
 503         private static final int STORE = 24;       static { commands.put("STORE", new Integer(STORE)); }
 504         private static final int SEARCH = 25;      static { commands.put("SEARCH", new Integer(SEARCH)); }
 505     }
 506
 507     public static class Parser {
 508         private Stream stream;
 509         public Parser(Stream from) { this.stream = from; }
 510         public Token token(String s) { return new Token(s); }
 511         protected Query query() {
 512             String s = null;
 513             boolean not = false;
 514             Query q = null;
 515             while(true) {
 516                 Parser.Token t = token(false);
 517                 if (t == null) break;
 518                 if (t.type == t.LIST) throw new Server.No("nested queries not yet supported FIXME");
 519                 else if (t.type == t.SET) return Query.num(t.set());
 520                 s = t.atom().toUpperCase();
 521                 if (s.equals("NOT")) { not = true; continue; }
 522                 if (s.equals("OR"))    return Query.or(query(), query());    // FIXME parse rest of list
 523                 if (s.equals("AND"))   return Query.and(query(), query());
 524             }
 525             if (s.startsWith("UN"))        { not = true; s = s.substring(2); }
 526             if (s.equals("ANSWERED"))        q = Query.answered();
 527             else if (s.equals("DELETED"))    q = Query.deleted();
 528             else if (s.equals("ALL"))        q = Query.all();
 529             else if (s.equals("DRAFT"))      q = Query.draft();
 530             else if (s.equals("FLAGGED"))    q = Query.flagged();
 531             else if (s.equals("RECENT"))     q = Query.recent();
 532             else if (s.equals("SEEN"))       q = Query.seen();
 533             else if (s.equals("OLD"))      { not = true; q = Query.recent(); }
 534             else if (s.equals("NEW"))        q = Query.and(Query.recent(), Query.not(Query.seen()));
 535             else if (s.equals("KEYWORD"))    q = Query.header("keyword", token().flag());
 536             else if (s.equals("HEADER"))     q = Query.header(token().astring(), token().astring());
 537             else if (s.equals("BCC"))        q = Query.header("bcc", token().astring());
 538             else if (s.equals("CC"))         q = Query.header("cc", token().astring());
 539             else if (s.equals("FROM"))       q = Query.header("from", token().astring());
 540             else if (s.equals("TO"))         q = Query.header("to", token().astring());
 541             else if (s.equals("SUBJECT"))    q = Query.header("subject", token().astring());
 542             else if (s.equals("LARGER"))     q = Query.size(token().n(), Integer.MAX_VALUE);
 543             else if (s.equals("SMALLER"))    q = Query.size(Integer.MIN_VALUE, token().n());
 544             else if (s.equals("BODY"))       q = Query.body(token().astring());
 545             else if (s.equals("TEXT"))       q = Query.full(token().astring());
 546             else if (s.equals("BEFORE"))     q = Query.arrival(new Date(0), token().date());
 547             else if (s.equals("SINCE"))      q = Query.arrival(token().date(), new Date(Long.MAX_VALUE));
 548             else if (s.equals("ON"))       { Date d = token().date(); q = Query.arrival(d, new Date(d.getTime() + 24 * 60 * 60)); }
 549             else if (s.equals("SENTBEFORE")) q = Query.sent(new Date(0), token().date());
 550             else if (s.equals("SENTSINCE"))  q = Query.sent(token().date(), new Date(Long.MAX_VALUE));
 551             else if (s.equals("SENTON"))   { Date d = token().date(); q = Query.sent(d, new Date(d.getTime() + 24 * 60 * 60)); }
 552             else if (s.equals("UID"))        q = Query.uid(token().set());
 553             return q;
 554         }
 555
 556         private static void bad(String s) { throw new Server.Bad(s); }
 557         class Token {
 558             public final byte type;
 559             private final String s;
 560             private final Parser.Token[] l;
 561             private final int n;
 562             private static final byte NIL = 0, LIST = 1, QUOTED = 2, NUMBER = 3, ATOM = 4, BAREWORD = 5, SET = 6;
 563             public Token()                         { this.s = null; n = 0;      l = null; type = NIL; }
 564             public Token(String s)                 { this(s, false); }
 565             public Token(String s, boolean quoted) { this.s = s;    n = 0;      l = null; type = quoted ? QUOTED : ATOM;  }
 566             public Token(Parser.Token[] list)      { this.s = null; n = 0;      l = list; type = LIST; }
 567             public Token(int number)               { this.s = null; n = number; l = null; type = NUMBER; }
 568
 569             public String   flag()    { if (type != ATOM) bad("expected a flag"); return s; }
 570             public int      n()       { if (type != NUMBER) bad("expected number"); return n; }
 571             public int      nz()      { int n = n(); if (n == 0) bad("expected nonzero number"); return n; }
 572             public String   q()       { if (type == NIL) return null; if (type != QUOTED) bad("expected qstring"); return s; }
 573             public Parser.Token[]  l()       { if (type == NIL) return null; if (type != LIST) bad("expected list"); return l; }
 574             public Parser.Token[]  lx()      {
 575                 if (type == LIST) return l;
 576                 Vec v = new Vec();
 577                 v.addElement(this);
 578                 while(true) {
 579                     Parser.Token t = token(false);
 580                     if (t == null) break;
 581                     v.addElement(t);
 582                 }
 583                 Parser.Token[] ret = new Parser.Token[v.size()];
 584                 v.copyInto(ret);
 585                 return ret;
 586             }
 587             public String   nstring() { if (type==NIL) return null; if (type!=QUOTED) bad("expected nstring"); return s; }
 588             public String   astring() {
 589                 if (type != ATOM && type != QUOTED) bad("expected atom or string");
 590                 if (s == null) bad("astring cannot be null");
 591                 return s; }
 592             public String[] sl() {
 593                 if (type == NIL) return null;
 594                 if (type != LIST) bad("expected list");
 595                 String[] ret = new String[l.length];
 596                 for(int i=0; i<ret.length; i++) ret[i] = l[i].s;
 597                 return ret;
 598             }
 599             public int flags() {
 600                 if (type != LIST) bad("expected flag list");
 601                 int ret = 0;
 602                 for(int i=0; i<l.length; i++) {
 603                     String flag = l[i].s;
 604                     if (flag.equals("\\Deleted"))       ret |= Mailbox.Flag.DELETED;
 605                     else if (flag.equals("\\Seen"))     ret |= Mailbox.Flag.SEEN;
 606                     else if (flag.equals("\\Flagged"))  ret |= Mailbox.Flag.FLAGGED;
 607                     else if (flag.equals("\\Draft"))    ret |= Mailbox.Flag.DRAFT;
 608                     else if (flag.equals("\\Answered")) ret |= Mailbox.Flag.ANSWERED;
 609                     else if (flag.equals("\\Recent"))   ret |= Mailbox.Flag.RECENT;
 610                 }
 611                 return ret;
 612             }
 613             public int[] set() {
 614                 if (type != ATOM) bad("expected a messageid set");
 615                 Vec.Int ids = new Vec.Int();
 616                 StringTokenizer st = new StringTokenizer(s, ",");
 617                 while(st.hasMoreTokens()) {
 618                     String s = st.nextToken();
 619                     if (s.indexOf(':') == -1) {
 620                         if (s.equals("*")) {
 621                             ids.addElement(0);
 622                             ids.addElement(Integer.MAX_VALUE);
 623                         } else {
 624                             ids.addElement(Integer.parseInt(s));
 625                             ids.addElement(Integer.parseInt(s));
 626                         }
 627                         continue; }
 628                     int start = Integer.parseInt(s.substring(0, s.indexOf(':')));
 629                     String end_s = s.substring(s.indexOf(':')+1);
 630                     if (end_s.equals("*")) { ids.addElement(start); ids.addElement(Integer.MAX_VALUE); }
 631                     else {
 632                         int end = Integer.parseInt(end_s);
 633                         for(int j=Math.min(start,end); j<=Math.max(start,end); j++) {
 634                             ids.addElement(j);
 635                             ids.addElement(j);
 636                         }
 637                     }
 638                 }
 639                 return ids.dump();
 640             }
 641             public Date date() {
 642                 if (type != QUOTED && type != ATOM) bad("Expected quoted or unquoted date");
 643                 try { return new SimpleDateFormat("dd-MMM-yyyy").parse(s);
 644                 } catch (ParseException p) { throw new Server.Bad("invalid date format; " + p); }
 645             }
 646             public Date datetime() {
 647                 if (type != QUOTED) bad("Expected quoted datetime");
 648                 try { return new SimpleDateFormat("dd-MM-yyyy hh:mm:ss zzzz").parse(s.trim());
 649                 } catch (ParseException p) { throw new Server.Bad("invalid datetime format " + s + " : " + p); }
 650             }
 651             public String atom() {
 652                 if (type != ATOM) bad("expected atom");
 653                 for(int i=0; i<s.length(); i++) {
 654                     char c = s.charAt(i);
 655                     if (c == '(' || c == ')' || c == '{' || c == ' ' || c == '%' || c == '*' || c == '\"' | c == '\\')
 656                         bad("invalid char in atom: " + c);
 657                 }
 658                 return s;
 659             }
 660         }
 661
 662         public void newline() {
 663             while (stream.peekc() == '\r' || stream.peekc() == '\n' || stream.peekc() == ' ') {
 664                 for(char c = stream.peekc(); c == ' ';) { stream.getc(); c = stream.peekc(); };
 665                 for(char c = stream.peekc(); c == '\r' || c == '\n';) { stream.getc(); c = stream.peekc(); };
 666             }
 667         }
 668
 669         public Token token() { return token(true); }
 670         public Token token(boolean freak) {
 671             Vec toks = new Vec();
 672             StringBuffer sb = new StringBuffer();
 673             char c = stream.getc(); while (c == ' ') c = stream.getc();
 674             if (c == '\r' || c == '\n') { if (freak) bad("unexpected end of line"); return null; }
 675             else if (c == '{') {
 676                 while(stream.peekc() != '}') sb.append(stream.getc());
 677                 stream.getc();
 678                 stream.println("+ Ready when you are...");
 679                 int octets = Integer.parseInt(sb.toString());
 680                 while(stream.peekc() == ' ') stream.getc();   // whitespace
 681                 while(stream.peekc() == '\n' || stream.peekc() == '\r') stream.getc();
 682                 byte[] bytes = new byte[octets];
 683                 int numread = 0;
 684                 while(numread < bytes.length) {
 685                     int n = stream.read(bytes, numread, bytes.length - numread);
 686                     if (n == -1) bad("end of stream while reading IMAP qstring");
 687                     numread += n;
 688                 }
 689                 return new Token(new String(bytes), true);
 690             } else if (c == '\"') {
 691                 while(true) {
 692                     c = stream.getc();
 693                     if (c == '\\') sb.append(stream.getc());
 694                     else if (c == '\"') break;
 695                     else sb.append(c);
 696                 }
 697                 return new Token(sb.toString(), true);
 698
 699                 // NOTE: this is technically a violation of the IMAP grammar, since atoms like FOO[BAR should be legal
 700             } else if (c == ']' || c == ')') { return null;
 701             } else if (c == '[' || c == '(') {
 702                 Token t;
 703                 do { t = token(); if (t != null) toks.addElement(t); } while (t != null);
 704                 Token[] ret = new Token[toks.size()];
 705                 toks.copyInto(ret);
 706                 return new Token(ret);
 707
 708             } else while(true) {
 709                 sb.append(c);
 710                 c = stream.peekc();
 711                 if (c == ' ' || c == '\"' || c == '(' || c == ')' || c == '[' || c == ']' ||
 712                     c == '{' || c == '\n' || c == '\r')
 713                     return new Token(sb.toString(), false);
 714                 stream.getc();
 715             }
 716         }
 717     }
 718
 719     public static class Printer {
 720             static String quotify(String s){
 721                 return s==null?"NIL":"\""+s.replaceAll("\\\\","\\\\").replaceAll("\"","\\\\\"")+"\"";}
 722             static String quotify(Date d) {
 723                 return new SimpleDateFormat("dd-MMM-yyyy HH:mm:ss +zzzz").format(d); }
 724             static String address(Address a) {
 725                 return"("+quotify(a.description)+" NIL "+quotify(a.user)+" "+quotify(a.host)+")"; }
 726             public static String addressList(Object a) {
 727                 if (a == null) return "NIL";
 728                 if (a instanceof Address) return "("+address((Address)a)+")";
 729                 if (a instanceof String) return "("+address(Address.parse((String)a))+")";
 730                 Address[] aa = (Address[])a;
 731                 StringBuffer ret = new StringBuffer();
 732                 ret.append("(");
 733                 for(int i=0; i<aa.length; i++) { ret.append(aa[i]); if (i < aa.length - 1) ret.append(" "); }
 734                 ret.append(")");
 735                 return ret.toString();
 736             }
 737             static String flags(Mailbox.Iterator it) {
 738                 return
 739                     (it.deleted()  ? "\\Deleted "  : "") +
 740                     (it.seen()     ? "\\Seen "     : "") +
 741                     (it.flagged()  ? "\\Flagged "  : "") +
 742                     (it.draft()    ? "\\Draft "    : "") +
 743                     (it.answered() ? "\\Answered " : "") +
 744                     (it.recent()   ? "\\Recent "   : "");
 745         }
 746         static String flags(int flags) {
 747             String ret = "(" +
 748                 (((flags & Mailbox.Flag.DELETED) == Mailbox.Flag.DELETED) ? "\\Deleted "  : "") +
 749                 (((flags & Mailbox.Flag.SEEN) == Mailbox.Flag.SEEN)    ? "\\Seen "     : "") +
 750                 (((flags & Mailbox.Flag.FLAGGED) == Mailbox.Flag.FLAGGED) ? "\\Flagged "  : "") +
 751                 (((flags & Mailbox.Flag.DRAFT) == Mailbox.Flag.DRAFT)   ? "\\Draft "    : "") +
 752                 (((flags & Mailbox.Flag.ANSWERED) == Mailbox.Flag.ANSWERED)? "\\Answered " : "") +
 753                 (((flags & Mailbox.Flag.RECENT) == Mailbox.Flag.RECENT)  ? "\\Recent "   : "");
 754             if (ret.endsWith(" ")) ret = ret.substring(0, ret.length() - 1);
 755             return ret + ")";
 756         }
 757         static String bodystructure(Message m) {
 758             // FIXME
 759             return "(\"TEXT\" \"PLAIN\" (\"CHARSET\" \"ISO-8859-1\") NIL NIL \"7BIT\" "+m.size()+" "+m.lines+")";
 760         }
 761         static String message(Message m) { return m.toString(); }
 762         static String date(Date d) { return "\""+d.toString()+"\""; }
 763         static String envelope(Message m) {
 764             return
 765                 "(" + quotify(m.envelope.arrival.toString()) +
 766                 " " + quotify(m.subject) +
 767                 " " + addressList(m.from) +
 768                 " " + addressList(m.headers.gets("sender")) +
 769                 " " + addressList(m.replyto) +
 770                 " " + addressList(m.to) +
 771                 " " + addressList(m.cc) +
 772                 " " + addressList(m.bcc) +
 773                 " " + quotify((String)m.headers.gets("in-reply-to")) +
 774                 " " + quotify(m.messageid) +
 775                 ")";
 776         }
 777
 778         public static String qq(String s) {
 779             StringBuffer ret = new StringBuffer();
 780             ret.append('{');
 781             ret.append(s.length());
 782             ret.append('}');
 783             ret.append('\r');
 784             ret.append('\n');
 785             ret.append(s);
 786             return ret.toString();
 787         }
 788
 789         private static String join(int[] nums) {
 790             StringBuffer ret = new StringBuffer();
 791             for(int i=0; i<nums.length; i++) {
 792                 ret.append(nums[i]);
 793                 if (i<nums.length-1) ret.append(' ');
 794             }
 795             return ret.toString();
 796         }
 797         private static String join(String delimit, String[] stuff) {
 798             StringBuffer ret = new StringBuffer();
 799             for(int i=0; i<stuff.length; i++) {
 800                 ret.append(stuff[i]);
 801                 if (i<stuff.length - 1) ret.append(delimit);
 802             }
 803             return ret.toString();
 804         }
 805         }
 806
 807
 808     // Main //////////////////////////////////////////////////////////////////////////////
 809
 810     public static final int
 811         PEEK=0x1, BODYSTRUCTURE=0x2, ENVELOPE=0x4, FLAGS=0x8, INTERNALDATE=0x10, FIELDS=0x800, FIELDSNOT=0x1000,
 812         RFC822=0x20, RFC822TEXT=0x40, RFC822SIZE=0x80, HEADERNOT=0x100, UID=0x200, HEADER=0x400;
 813
 814 }