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
11 package org.ibex.mail.protocol;
12 import org.ibex.util.*;
14 import org.ibex.mail.*;
15 import org.ibex.mail.target.*;
16 import org.ibex.jinetd.*;
22 /** NNTP send/recieve */
25 public static final DateFormat dateFormat = new SimpleDateFormat("yyyyMMDDhhmmss");
27 public static class No extends RuntimeException { int code = 400; } // 4xx response codes
28 public static class Bad extends RuntimeException { int code = 500; public Bad(String s) { super(s); } } // 5xx response codes
30 public static class Group {
31 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;}
32 public final String name; // case insensitive
33 public final boolean post;
34 public final int first;
35 public final int last;
36 public final int count; // an approximation; must be >= actual number
39 public static class Article {
40 public Article(int num, Message message) { this.message = message; this.num = num;}
42 public final Message message;
45 public static interface Server {
46 public Group group(String s);
47 public boolean ihave(String messageid);
48 public Article next();
49 public Article last();
50 public boolean post(Message m);
51 public Article article(String messageid, boolean head, boolean body);
52 public Article article(int messagenum, boolean head, boolean body);
53 public Group[] list();
54 public Group[] newgroups(Date d, String[] distributions);
55 public String[] newnews(String[] groups, Date d, String[] distributions);
58 public static class MailboxWrapper implements Server {
59 private final Mailbox root;
60 private Mailbox current;
62 public MailboxWrapper(Mailbox root) { this.root = root; }
63 public Group group(String s) { ptr = 0; setgroup(s); return getgroup(s); }
65 public boolean ihave(String messageid) { /* FEATURE */ return false; }
66 public boolean post(Message m) { /* FEATURE */ return false; }
68 public Article next() { return article(ptr++, false, false); }
69 public Article last() { return article(ptr--, false, false); }
70 public Article article(String i, boolean h, boolean b) { return article(Query.header("message-id",i),h,b); }
71 public Article article(int n, boolean h, boolean b) { ptr = n; return article(Query.messagenum(n,n),h,b); }
72 private Article article(Query q, boolean head, boolean body) {
73 Mailbox.Iterator it = current.iterator(q);
74 if (!it.next()) return null;
76 Message m = body ? it.cur() : Message.newMessage(new Fountain.StringFountain(it.head() + "\r\n"));
77 return new Article(it.num(), m);
78 } catch (Exception e) { return null; }
80 public Group[] list() { return list(root, ""); }
81 private Group[] list(Mailbox who, String prefix) {
83 if (who == null) who = root;
84 String[] s = who.children();
85 for(int i=0; i<s.length; i++) {
86 v.addElement(new Group(prefix + s[i], true, 0, 0, 0)); // FIXME numbers
87 Group[] g2 = list(who.slash(s[i], false), prefix + s[i] + ".");
88 for(int j=0; j<g2.length; j++) v.addElement(g2[j]);
90 Group[] ret = new Group[v.size()];
95 private void setgroup(String s) {
97 for(StringTokenizer st = new StringTokenizer(s, ".");
99 current = current.slash(st.nextToken(), false));
101 private Group getgroup(String s) {
103 for(StringTokenizer st = new StringTokenizer(s, "."); st.hasMoreTokens(); box = box.slash(st.nextToken(), false));
104 return new Group(s, true, 1, box.count(Query.all()), box.count(Query.all()));
107 public Group[] newgroups(Date d, String[] distributions) { /* FEATURE */ return new Group[] { }; }
108 public String[] newnews(String[] groups, Date d, String[] distributions) { /* FIXME */ return null; }
111 public static class Listener {
112 private Server api = null;
114 private Connection conn;
115 public Listener(Login l) { this.login = l; }
117 private void println(String s) { Log.warn("[nntp-write]", s); conn.println(s); }
118 private void println() { Log.warn("[nntp-write]", ""); conn.println(""); }
119 private void print(String s) { Log.warn("[nntp-write]", s); conn.print(s); }
121 private void article(String numOrMessageId, boolean head, boolean body) {
122 String s = numOrMessageId.trim();
124 if (s.startsWith("<")) a = api.article(s.substring(0, s.length() - 1), head, body);
125 else a = api.article(Integer.parseInt(s), head, body);
127 println("423 No such article.");
130 int code = (head && body) ? 220 : head ? 221 : body ? 222 : 223;
131 println(code + " " + a.num + " <" + a.message.messageid + "> get ready for some stuff...");
132 if (head) println(a.message.headers.getString());
133 if (head && body) println();
135 Stream stream = a.message.getBody().getStream();
138 if (s == null) break;
139 if (s.startsWith(".")) print(".");
145 public void handleRequest(Connection conn) {
147 conn.setTimeout(30 * 60 * 1000);
148 conn.setNewline("\r\n");
149 println("200 " + conn.vhost + " [" + NNTP.class.getName() + "]");
152 Account account = login.anonymous();
153 this.api = account == null ? null : new MailboxWrapper(account.getMailbox(NNTP.class));
154 for(String line = conn.readln(); line != null; line = conn.readln()) try {
155 Log.warn("[nntp-read]", line);
156 StringTokenizer st = new StringTokenizer(line, " ");
157 String command = st.nextToken().toUpperCase();
158 if (command.equals("AUTHINFO")) {
159 // FIXME technically the RFC says we need to use this info to generate a SEnder: header...
160 String uop = st.nextToken().toUpperCase();
161 if (uop.equals("USER")) user = st.nextToken();
162 else if (uop.equals("PASS")) pass = st.nextToken();
165 if (this.api == null) {
166 if (user == null) { println("480 Authentication required"); continue; }
167 if (pass == null) { println("381 Password required"); continue; }
168 account = login.login(user, pass);
169 if (account == null) { println("502 Invalid"); continue; }
170 this.api = new MailboxWrapper(account.getMailbox(NNTP.class));
171 println("281 Good to go");
174 if (command.equals("ARTICLE")) { article(st.hasMoreTokens() ? st.nextToken() : null, true, true);
175 } else if (command.equals("HEAD")) { article(st.hasMoreTokens() ? st.nextToken() : null, true, false);
176 } else if (command.equals("DATE")) {
178 println("111 " + dateFormat.format(new Date()));
179 } else if (command.equals("MODE")) {
180 if (st.hasMoreTokens()) {
181 String arg = st.nextToken();
182 if (arg.equalsIgnoreCase("STREAM"));
184 println("203 Streaming permitted");
186 println("201 Hello, you can post.");
188 } else if (command.equals("BODY")) { article(st.hasMoreTokens() ? st.nextToken() : null, false, true);
189 } else if (command.equals("STAT")) { article(st.hasMoreTokens() ? st.nextToken() : null, false, false);
190 } else if (command.equals("HELP")) { println("100 you are beyond help."); println(".");
191 } else if (command.equals("SLAVE")) { println("220 I don't care");
192 } else if (command.equals("XOVER")) {
193 println("224 Overview information follows");
194 MailboxWrapper api = (MailboxWrapper)this.api;
195 String range = st.hasMoreTokens() ? st.nextToken() : (api.ptr+"-"+api.ptr);
196 int start = Integer.parseInt(range.substring(0, range.indexOf('-')));
197 int end = Integer.parseInt(range.substring(range.indexOf('-') + 1));
198 Mailbox.Iterator it = api.current.iterator(Query.messagenum(start, end));
201 Message m = it.cur();
202 println(it.num()+"\t"+m.subject+"\t"+m.from+"\t"+m.date+"\t"+m.messageid+"\t"+
203 m.headers.get("references") + "\t" + m.getLength() + "\t" + m.getNumLines());
204 } catch (Exception e) { Log.error(this, e); }
207 } else if (command.equals("LAST")) { Article a = api.last(); println("223 "+a.num+" "+a.message.messageid+" ok");
208 } else if (command.equals("NEXT")) { Article a = api.next(); println("223 "+a.num+" "+a.message.messageid+" ok");
209 } else if (command.equals("QUIT")) { println("205 Bye."); conn.close(); return;
210 } else if (command.equals("GROUP")) {
211 Group g = api.group(st.nextToken().toLowerCase());
212 println("211 " + g.count + " " + g.first + " " + g.last + " " + g.name);
213 } else if (command.equals("NEWGROUPS") || command.equals("NEWNEWS")) {
214 // FIXME: * and ! unsupported
215 // NEWNEWS is often not supported
216 String groups = command.equals("NEWNEWS") ? st.nextToken() : null;
217 String datetime = st.nextToken() + " " + st.nextToken();
218 String gmt = st.nextToken();
219 String distributions = gmt.equals("GMT") ? (st.hasMoreTokens() ? st.nextToken() : null) : gmt;
220 while(st.hasMoreTokens()) distributions += " " + st.nextToken();
221 // FIXME deal with GMT
224 d = new SimpleDateFormat("yyMMDD HHMMSS").parse(gmt);
225 } catch (ParseException pe) {
228 distributions = distributions.trim();
229 if (distributions.startsWith("<")) distributions = distributions.substring(1, distributions.length() - 1);
231 st = new StringTokenizer(distributions, ",");
232 String[] dists = new String[st.countTokens()];
233 for(int i=0; st.hasMoreTokens(); i++) dists[i] = st.nextToken();
235 if (command.equals("NEWGROUPS")) {
236 Group[] g = api.newgroups(d, dists);
237 println("231 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 st = new StringTokenizer(groups, ",");
243 String[] g = new String[st.countTokens()];
244 for(int i=0; st.hasMoreTokens(); i++) g[i] = st.nextToken();
245 String[] a = api.newnews(g, d, dists);
246 println("230 list of article messageids follows");
247 for(int i=0; i<a.length; i++) println(a[i]);
251 } else if (command.equals("POST")) {
252 // add NNTP-Posting-Host header
254 // required headers: Newsgroups, Subject, Message-ID, Path, From, Date. No wildcars in newsgroups list
255 // Path header: prepend <myname>, (any punctuation separates the list)
256 // Expires header: the date when expiration happens (??) should we ignore this?
257 // Control header: body is the command. Inteprert posts to all.all.ctl as control messages, use Subject line if no Cntrol line
258 // "Approved" line is used for moderaion
259 // Xref: drop this header if you see it
262 // cancel <Message-ID> (do not forward if I am unable to cancel locally)
263 // ihave/sendme: do not support
264 // newgroup <groupname> [moderated] -- body of message is a description of the group
265 // rmgroup <groupname>
268 boolean postok = api.post();
270 println("440 no posting allowed");
273 println("340 send the article");
274 // FIME read the article here
275 println("240 article posted ok");
278 } else if (command.equals("XROVER")) {
279 // equivalent to "XHDR References"
280 } else if (command.equals("XHDR")) {
281 // argument: header name
282 // argument: 1 | 1- | 1-2 | <mid> | nothing (use current article)
284 // print art#+header for all matching messages
286 // 412 if no group selected and numeric form used
287 // 430 if <mid> and not found
288 // 420 if no messages in range
289 } else if (command.equals("XPAT")) {
290 // just like XHDR, but a pattern follows the last argument (may contain whitespace)
294 } else if (command.equals("LIST")) {
295 if (st.hasMoreTokens()) {
296 String argument = st.nextToken().toUpperCase();
297 if (argument.equalsIgnoreCase("EXTENSIONS")) {
298 println("202 Extensions supported:");
299 println("STREAMING");
302 } else if (argument.equals("ACTIVE")) {
303 String wildmat = st.hasMoreTokens() ? st.nextToken() : null;
304 // FIXME: deal with wildmat
305 // just like list, but only show active groups
306 throw new Bad("not implemented yet");
307 } else if (argument.equals("SUBSCRIPTIONS")) {
308 // FIXME: show 215, default subscription list for new users, period
309 } else if (argument.equals("OVERVIEW.FMT")) {
310 println("215 Overview format:");
314 println("Message-ID:");
315 println("References:");
318 //println("Xref:full");
320 } else if (argument.equals("NEWSGROUPS")) {
321 String wildmat = st.hasMoreTokens() ? st.nextToken() : null;
322 // respond 215, print each newsgroup, a space, and the description; end with lone period
327 Group[] g = api.list();
328 println("215 list of groups follows");
329 for(int i=0; i<g.length; i++)
330 println(g[i].name + " " + g[i].last + " " + g[i].first + " " + (g[i].post ? "y" : "n"));
334 } else if (command.equals("LISTGROUP")) {
335 String groupname = st.hasMoreTokens() ? st.nextToken() : null;
336 // 211, all article numbers in group, period. Set article ptr to first item in group
338 } else if (command.equals("XGTITLE")) {
339 String wildmat = st.hasMoreTokens() ? st.nextToken() : null;
340 // 282, then identical to LIST NEWSGROUP
342 } else if (command.equals("CHECK")) {
343 // FIXME: may be pipelined; must spawn threads
344 String mid = st.nextToken();
345 boolean want = api.ihave(mid);
347 println("438 "+ mid+" No thanks");
349 println("238 "+mid+" Yes, I'd like that");
352 } else if (command.equals("TAKETHIS")) {
353 // FIXME: may be pipelined
354 String mid = st.nextToken();
355 // MUST read message here
358 println("439 "+ mid+" Transfer failed");
360 println("239 "+mid+" Rock on.");
364 } else if (command.equals("IHAVE")) {
365 boolean want = api.ihave(st.nextToken());
367 println("435 No thanks");
369 println("335 Proceed");
370 // FIXME read article here
371 println("235 Got it");
374 throw new Bad("wtf are you talking about?");
376 } catch (No n) { println(n.code + " " + n.getMessage());
377 } catch (Bad b) { println(b.code + " " + b.getMessage()); Log.warn(this, b); }