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.
7 // 503 optional subfeature not supported
9 // LIST EXTENSIONS is probably incomplete
10 // pull mode (ie suck)
11 // control message processing?
13 package org.ibex.mail;
14 import org.ibex.util.*;
16 import org.ibex.net.*;
17 import org.ibex.mail.target.*;
18 import org.ibex.jinetd.*;
24 /** NNTP send/recieve */
27 public static final DateFormat dateFormat = new SimpleDateFormat("yyyyMMDDhhmmss");
29 public static class No extends RuntimeException { int code = 400; } // 4xx response codes
30 public static class Bad extends RuntimeException { int code = 500; public Bad(String s) { super(s); } } // 5xx response codes
32 public static class Group {
33 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;}
34 public final String name; // case insensitive
35 public final boolean post;
36 public final int first;
37 public final int last;
38 public final int count; // an approximation; must be >= actual number
41 public static class Article {
42 public Article(int num, Message message) { this.message = message; this.num = num; }
44 public final Message message;
47 public static interface Server {
48 public Group group(String s);
49 public boolean ihave(String messageid);
50 public Article next();
51 public Article last();
52 public boolean postok();
53 public void post(Message m) throws IOException;
54 public Article article(String messageid, boolean head, boolean body);
55 public Article article(int messagenum, boolean head, boolean body);
56 public Group[] list();
57 public Group[] newgroups(Date d, String[] distributions);
58 public String[] newnews(String[] groups, Date d, String[] distributions);
61 public static class MailboxWrapper implements Server {
62 private final MailboxTree root;
63 private Mailbox current;
66 public MailboxWrapper(MailboxTree root) { this(root, false); }
67 public MailboxWrapper(MailboxTree root, boolean post) { this.root = root; this.post = post; }
68 public boolean postok() { return post; }
69 public void post(Message m) throws IOException { current.post(m); }
70 public Group group(String s) {
72 Group g = getgroup(s);
73 if (g==null) return null;
78 public boolean ihave(String messageid) { /* FEATURE */ return false; }
80 public Article next() { return article(ptr++, false, false); }
81 public Article last() { return article(ptr--, false, false); }
82 public Article article(String i, boolean h, boolean b) { return article(Query.header("message-id",i),h,b); }
83 public Article article(int n, boolean h, boolean b) { ptr = n; return article(Query.nntpNumber(n,n),h,b); }
84 private Article article(Query q, boolean head, boolean body) {
85 Mailbox.Iterator it = current.iterator(q);
86 if (!it.next()) return null;
89 Message m = body ? it.cur() : Message.newMessage(new Fountain.StringFountain(it.head() + "\r\n"));
90 return new Article(it.nntpNumber(), m);
91 } catch (Exception e) { return null; }
93 public Group[] list() { return list(root, ""); }
94 private Group[] list(MailboxTree who, String prefix) {
96 if (who == null) who = root;
97 String[] s = who.children();
98 for(int i=0; i<s.length; i++) {
99 v.addElement(new Group(prefix + s[i], true, 0, 0, 0)); // FIXME numbers
100 Group[] g2 = list(who.slash(s[i], false), prefix + s[i] + ".");
101 for(int j=0; j<g2.length; j++) v.addElement(g2[j]);
103 Group[] ret = new Group[v.size()];
108 private void setgroup(String s) {
109 MailboxTree ncurrent = root;
110 for(StringTokenizer st = new StringTokenizer(s, ".");
111 ncurrent != null && st.hasMoreTokens();
112 ncurrent = ncurrent.slash(st.nextToken(), false));
113 if (ncurrent!=null) current=ncurrent.getMailbox();
115 private Group getgroup(String s) {
116 MailboxTree box = root;
117 for(StringTokenizer st = new StringTokenizer(s, ".");
118 box!=null && st.hasMoreTokens();
119 box = box.slash(st.nextToken(), false));
120 if (box==null) return null;
121 return new Group(s, true, 1, box.getMailbox().count(Query.all()), box.getMailbox().count(Query.all()));
124 public Group[] newgroups(Date d, String[] distributions) { /* FEATURE */ return new Group[] { }; }
125 public String[] newnews(String[] groups, Date d, String[] distributions) { /* FIXME */ return null; }
128 public static class Listener {
129 private Server api = null;
131 private Connection conn;
132 public Listener(Login l) { this.login = l; }
134 private void println(String s) { Log.warn("[nntp-write]", s); conn.println(s); }
135 private void println() { Log.warn("[nntp-write]", ""); conn.println(""); }
136 private void print(String s) { Log.warn("[nntp-write]", s); conn.print(s); }
138 private void article(String numOrMessageId, boolean head, boolean body) {
139 String s = numOrMessageId.trim();
141 if (s.startsWith("<")) a = api.article(s.substring(0, s.length() - 1), head, body);
142 else a = api.article(Integer.parseInt(s), head, body);
144 println("423 No such article.");
147 int code = (head && body) ? 220 : head ? 221 : body ? 222 : 223;
148 println(code + " " + a.num + " <" + a.message.messageid + "> get ready for some stuff...");
149 if (head) { a.message.headers.getStream().transcribe(conn); println(); }
150 if (head && body) println();
152 Stream stream = a.message.getBody().getStream();
155 if (s == null) break;
156 if (s.startsWith(".")) print(".");
162 public void handleRequest(Connection conn) {
164 conn.setTimeout(30 * 60 * 1000);
165 conn.setNewline("\r\n");
166 println("200 " + conn.vhost + " [" + NNTP.class.getName() + "]");
169 Account account = login.anonymous();
170 this.api = account == null ? null : new MailboxWrapper(account.getMailbox(NNTP.class), true);
171 for(String line = conn.readln(); line != null; line = conn.readln()) try {
172 Log.warn("[nntp-read]", line);
173 StringTokenizer st = new StringTokenizer(line, " ");
174 String command = st.nextToken().toUpperCase();
175 if (command.equals("AUTHINFO")) {
176 // FIXME technically the RFC says we need to use this info to generate a SEnder: header...
177 String uop = st.nextToken().toUpperCase();
178 if (uop.equals("USER")) {
179 user = st.nextToken();
180 println("381 More authentication required");
182 } else if (uop.equals("PASS")) {
183 pass = st.nextToken();
184 account = login.login(user, pass);
185 if (account == null) { println("502 Invalid"); continue; }
186 this.api = new MailboxWrapper(account.getMailbox(NNTP.class), true);
187 println("281 Good to go");
190 throw new Bad("wtf are you talking about?");
192 if (this.api == null) {
193 if (user == null) { println("480 Authentication required"); continue; }
194 if (pass == null) { println("381 Password required"); continue; }
196 if (command.equals("ARTICLE")) { article(st.hasMoreTokens() ? st.nextToken() : null, true, true);
197 } else if (command.equals("HEAD")) { article(st.hasMoreTokens() ? st.nextToken() : null, true, false);
198 } else if (command.equals("DATE")) {
200 println("111 " + dateFormat.format(new Date()));
201 } else if (command.equals("MODE")) {
202 if (st.hasMoreTokens()) {
203 String arg = st.nextToken();
204 if (arg.equalsIgnoreCase("STREAM"));
206 println("203 Streaming permitted");
208 println("201 Hello, you can post.");
210 } else if (command.equals("BODY")) { article(st.hasMoreTokens() ? st.nextToken() : null, false, true);
211 } else if (command.equals("STAT")) { article(st.hasMoreTokens() ? st.nextToken() : null, false, false);
212 } else if (command.equals("HELP")) { println("100 you are beyond help."); println(".");
213 } else if (command.equals("SLAVE")) { println("220 SLAVE was removed in RFC3977, you should not use it");
214 } else if (command.equals("XOVER")) {
215 println("224 Overview information follows");
216 MailboxWrapper api = (MailboxWrapper)this.api;
217 String range = st.hasMoreTokens() ? st.nextToken() : (api.ptr+"-"+api.ptr);
218 int start = Integer.parseInt(range.substring(0, range.indexOf('-')));
219 int end = Integer.parseInt(range.substring(range.indexOf('-') + 1));
220 Mailbox.Iterator it = api.current.iterator(Query.nntpNumber(start, end));
223 Message m = it.cur();
224 println(it.nntpNumber()+"\t"+m.subject+"\t"+m.from+"\t"+m.date+"\t"+m.messageid+"\t"+
225 m.headers.get("references") + "\t" + m.getLength() + "\t" + m.getNumLines());
226 } catch (Exception e) { Log.error(this, e); }
229 } else if (command.equals("LAST")) {
230 Article a = api.last(); println("223 "+a.num+" "+a.message.messageid+" ok");
231 } else if (command.equals("NEXT")) {
232 Article a = api.next(); println("223 "+a.num+" "+a.message.messageid+" ok");
233 } else if (command.equals("QUIT")) { println("205 Bye."); conn.close(); return;
234 } else if (command.equals("GROUP")) {
235 Group g = api.group(st.nextToken().toLowerCase());
236 if (g==null) println("411 no such group");
237 else println("211 " + g.count + " " + g.first + " " + g.last + " " + g.name);
238 } else if (command.equals("NEWGROUPS") || command.equals("NEWNEWS")) {
239 // FIXME: * and ! unsupported
240 // NEWNEWS is often not supported
241 String groups = command.equals("NEWNEWS") ? st.nextToken() : null;
242 String datetime = st.nextToken() + " " + st.nextToken();
243 String gmt = st.nextToken();
244 String distributions = gmt.equals("GMT") ? (st.hasMoreTokens() ? st.nextToken() : "") : gmt;
245 while(st.hasMoreTokens()) distributions += " " + st.nextToken();
247 // FIXME deal with GMT
250 d = new SimpleDateFormat("yyMMDD HHMMSS").parse(datetime);
251 } catch (ParseException pe) {
254 distributions = distributions.trim();
255 if (distributions.startsWith("<")) distributions = distributions.substring(1, distributions.length() - 1);
257 st = new StringTokenizer(distributions, ",");
258 String[] dists = new String[st.countTokens()];
259 for(int i=0; st.hasMoreTokens(); i++) dists[i] = st.nextToken();
261 if (command.equals("NEWGROUPS")) {
262 Group[] g = api.newgroups(d, dists);
263 println("231 list of groups follows");
264 for(int i=0; i<g.length; i++)
265 println(g[i].name + " " + g[i].last + " " + g[i].first + " " + (g[i].post ? "y" : "n"));
268 st = new StringTokenizer(groups, ",");
269 String[] g = new String[st.countTokens()];
270 for(int i=0; st.hasMoreTokens(); i++) g[i] = st.nextToken();
271 String[] a = api.newnews(g, d, dists);
272 println("230 list of article messageids follows");
273 for(int i=0; i<a.length; i++) println(a[i]);
277 } else if (command.equals("POST")) {
279 // add NNTP-Posting-Host header
280 // Path header: prepend <myname>, (any punctuation separates the list)
281 // Expires header: the date when expiration happens (??) should we ignore this?
282 // Control header: body is the command. Inteprert posts to all.all.ctl as control messages,
283 // use Subject line if no Cntrol line
284 // "Approved" line is used for moderaion
285 // Xref: drop this header if you see it
288 // cancel <Message-ID> (do not forward if I am unable to cancel locally)
289 // ihave/sendme: do not support
290 // newgroup <groupname> [moderated] -- body of message is a description of the group
291 // rmgroup <groupname>
293 boolean postok = api.postok();
295 println("440 no posting allowed");
297 println("340 send the article");
298 StringBuffer buf = new StringBuffer();
301 String s = conn.readln();
302 if (s == null) throw new RuntimeException("connection closed");
303 if (s.equals(".")) break;
304 if (s.startsWith(".")) s = s.substring(1);
305 buf.append(s + "\r\n");
307 String body = buf.toString();
309 Message m = Message.newMessage(new Fountain.StringFountain(body));
310 if (m.headers.get("newsgroups")==null)
311 println("441 posted messages must have a Newsgroups header per RFC 977");
312 else if (m.headers.get("newsgroups").indexOf('*')!=-1)
313 println("441 Newsgroups header in posted messages may not contain wildcards (*) per RFC 977");
314 else if (m.headers.get("subject")==null)
315 println("441 posted messages must have a Subject header per RFC 977");
316 // else if (m.headers.get("path")==null)
317 //println("441 posted messages must have a Path header per RFC 977");
318 else if (m.headers.get("from")==null)
319 println("441 posted messages must have a From header per RFC 977");
320 else if (m.headers.get("date")==null)
321 println("441 posted messages must have a Date header per RFC 977");
324 println("240 article posted ok");
326 } catch (Exception e) {
328 println("441 posting failed: " + e);
332 } else if (command.equals("XROVER")) {
333 // equivalent to "XHDR References"
334 } else if (command.equals("XHDR")) {
335 // argument: header name
336 // argument: 1 | 1- | 1-2 | <mid> | nothing (use current article)
338 // print art#+header for all matching messages
340 // 412 if no group selected and numeric form used
341 // 430 if <mid> and not found
342 // 420 if no messages in range
343 } else if (command.equals("XPAT")) {
344 // just like XHDR, but a pattern follows the last argument (may contain whitespace)
348 } else if (command.equals("LIST")) {
349 if (st.hasMoreTokens()) {
350 String argument = st.nextToken().toUpperCase();
351 if (argument.equalsIgnoreCase("EXTENSIONS")) {
352 println("202 Extensions supported:");
353 println("STREAMING");
356 } else if (argument.equals("ACTIVE")) {
357 String wildmat = st.hasMoreTokens() ? st.nextToken() : null;
358 // FIXME: deal with wildmat
359 // just like list, but only show active groups
360 throw new Bad("not implemented yet");
361 } else if (argument.equals("SUBSCRIPTIONS")) {
362 // FIXME: show 215, default subscription list for new users, period
363 } else if (argument.equals("OVERVIEW.FMT")) {
364 println("215 Overview format:");
368 println("Message-ID:");
369 println("References:");
372 //println("Xref:full");
374 } else if (argument.equals("NEWSGROUPS")) {
375 String wildmat = st.hasMoreTokens() ? st.nextToken() : null;
376 // respond 215, print each newsgroup, a space, and the description; end with lone period
381 Group[] g = api.list();
382 println("215 list of groups follows");
383 for(int i=0; i<g.length; i++)
384 println(g[i].name + " " + g[i].last + " " + g[i].first + " " + (g[i].post ? "y" : "n"));
388 } else if (command.equals("LISTGROUP")) {
389 String groupname = st.hasMoreTokens() ? st.nextToken() : null;
390 // 211, all article numbers in group, period. Set article ptr to first item in group
392 } else if (command.equals("XGTITLE")) {
393 String wildmat = st.hasMoreTokens() ? st.nextToken() : null;
394 // 282, then identical to LIST NEWSGROUP
396 } else if (command.equals("CHECK")) {
397 // FIXME: may be pipelined; must spawn threads
398 String mid = st.nextToken();
399 boolean want = api.ihave(mid);
401 println("438 "+ mid+" No thanks");
403 println("238 "+mid+" Yes, I'd like that");
406 } else if (command.equals("TAKETHIS")) {
407 // FIXME: may be pipelined
408 String mid = st.nextToken();
409 // MUST read message here
412 println("439 "+ mid+" Transfer failed");
414 println("239 "+mid+" Rock on.");
418 } else if (command.equals("IHAVE")) {
419 boolean want = api.ihave(st.nextToken());
421 println("435 No thanks");
423 println("335 Proceed");
424 // FIXME read article here
425 println("235 Got it");
428 throw new Bad("wtf are you talking about?");
430 } catch (No n) { println(n.code + " " + n.getMessage());
431 } catch (Bad b) { println(b.code + " " + b.getMessage()); Log.warn(this, b); }