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.
5 package org.ibex.mail.protocol;
6 import org.ibex.util.*;
8 import org.ibex.mail.*;
9 import org.ibex.mail.target.*;
10 import org.ibex.jinetd.*;
16 /** NNTP send/recieve */
19 // FIXME: command lines limited to 512 chars
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
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;
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;
37 public final Message message;
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);
53 public static class MailboxWrapper implements Server {
54 private final Mailbox root;
55 private Mailbox current;
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;
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; }
74 public Group[] list() { return list(root, ""); }
75 private Group[] list(Mailbox who, String prefix) {
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]);
84 Group[] ret = new Group[v.size()];
89 private void setgroup(String s) {
91 for(StringTokenizer st = new StringTokenizer(s, ".");
93 current = current.slash(st.nextToken(), false));
95 private Group getgroup(String s) {
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()));
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; }
105 public static class Listener implements Worker {
106 private Server api = null;
108 private Connection conn;
109 public Listener(Login l) { this.login = l; }
111 private void println(String s) { Log.warn("[nntp-write]", s); conn.println(s); }
112 private void println() { Log.warn("[nntp-write]", ""); conn.println(""); }
114 private void article(String numOrMessageId, boolean head, boolean body) {
115 String s = numOrMessageId.trim();
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);
120 println("423 No such article.");
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);
130 public void handleRequest(Connection conn) {
132 conn.setTimeout(30 * 60 * 1000);
133 println("200 " + conn.vhost + " [" + NNTP.class.getName() + "]");
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();
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");
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));
173 Message m = Message.newMessage(new Stream(it.head() + "\r\n"));
174 println(it.num()+"\t"+m.subject+"\t"+m.from+"\t"+m.date+"\t"+m.messageid+"\t"+
175 m.headers.gets("references") + "\t" + m.size() + "\t" + m.lines);
176 } catch (Exception e) { Log.error(this, e); }
179 } else if (command.equals("LAST")) { Article a = api.last(); println("223 "+a.num+" "+a.messageid+" ok");
180 } else if (command.equals("NEXT")) { Article a = api.next(); println("223 "+a.num+" "+a.messageid+" ok");
181 } else if (command.equals("QUIT")) { println("205 Bye."); conn.close(); return;
182 } else if (command.equals("GROUP")) {
183 Group g = api.group(st.nextToken().toLowerCase());
184 println("211 " + g.count + " " + g.first + " " + g.last + " " + g.name);
185 } else if (command.equals("NEWGROUPS") || command.equals("NEWNEWS")) {
186 String groups = command.equals("NEWNEWS") ? st.nextToken() : null;
187 String datetime = st.nextToken() + " " + st.nextToken();
188 String gmt = st.nextToken();
189 String distributions = gmt.equals("GMT") ? (st.hasMoreTokens() ? st.nextToken() : null) : gmt;
190 while(st.hasMoreTokens()) distributions += " " + st.nextToken();
191 // FIXME deal with GMT
194 d = new SimpleDateFormat("YYMMDD HHMMSS").parse(gmt);
195 } catch (ParseException pe) {
198 distributions = distributions.trim();
199 if (distributions.startsWith("<")) distributions = distributions.substring(1, distributions.length() - 1);
201 st = new StringTokenizer(distributions, ",");
202 String[] dists = new String[st.countTokens()];
203 for(int i=0; st.hasMoreTokens(); i++) dists[i] = st.nextToken();
205 if (command.equals("NEWGROUPS")) {
206 Group[] g = api.newgroups(d, dists);
207 println("231 list of groups follows");
208 for(int i=0; i<g.length; i++)
209 println(g[i].name + " " + g[i].last + " " + g[i].first + " " + (g[i].post ? "y" : "n"));
212 st = new StringTokenizer(groups, ",");
213 String[] g = new String[st.countTokens()];
214 for(int i=0; st.hasMoreTokens(); i++) g[i] = st.nextToken();
215 String[] a = api.newnews(g, d, dists);
216 println("230 list of article messageids follows");
217 for(int i=0; i<a.length; i++) println(a[i]);
221 } else if (command.equals("POST")) {
224 boolean postok = api.post();
226 println("440 no posting allowed");
229 println("340 send the article");
230 // FIME read the article here
231 println("240 article posted ok");
234 } else if (command.equals("LIST")) {
235 if (st.hasMoreTokens()) throw new Bad("LIST " + st.nextToken() + " not supported");
236 Group[] g = api.list();
237 println("215 list of groups follows");
238 for(int i=0; i<g.length; i++)
239 println(g[i].name + " " + g[i].last + " " + g[i].first + " " + (g[i].post ? "y" : "n"));
242 } else if (command.equals("IHAVE")) {
243 boolean want = api.ihave(st.nextToken());
245 println("435 No thanks");
247 println("335 Proceed");
248 // FIXME read article here
249 println("235 Got it");
252 throw new Bad("wtf are you talking about?");
254 } catch (No n) { println(n.code + " " + n.getMessage());
255 } catch (Bad b) { println(b.code + " " + b.getMessage()); Log.warn(this, b); }