Article(javax.mail.Message) now has a safe method to read the byte body from the given message object (fixes #16).
3 * see AUTHORS for the list of contributors
5 * This program is free software: you can redistribute it and/or modify
6 * it under the terms of the GNU General Public License as published by
7 * the Free Software Foundation, either version 3 of the License, or
8 * (at your option) any later version.
10 * This program is distributed in the hope that it will be useful,
11 * but WITHOUT ANY WARRANTY; without even the implied warranty of
12 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
13 * GNU General Public License for more details.
15 * You should have received a copy of the GNU General Public License
16 * along with this program. If not, see <http://www.gnu.org/licenses/>.
19 package org.sonews.util.io;
21 import java.io.BufferedInputStream;
22 import java.io.BufferedOutputStream;
23 import java.io.ByteArrayOutputStream;
24 import java.io.IOException;
25 import java.io.InputStream;
26 import java.io.UnsupportedEncodingException;
27 import java.net.Socket;
28 import java.net.UnknownHostException;
29 import org.sonews.config.Config;
30 import org.sonews.util.Log;
33 * Reads an news article from a NNTP server.
34 * @author Christian Lins
37 public class ArticleReader
40 private BufferedOutputStream out;
41 private BufferedInputStream in;
42 private String messageID;
44 public ArticleReader(String host, int port, String messageID)
45 throws IOException, UnknownHostException
47 this.messageID = messageID;
49 // Connect to NNTP server
50 Socket socket = new Socket(host, port);
51 this.out = new BufferedOutputStream(socket.getOutputStream());
52 this.in = new BufferedInputStream(socket.getInputStream());
53 String line = readln(this.in);
54 if(!line.startsWith("200 "))
56 throw new IOException("Invalid hello from server: " + line);
60 private boolean eofArticle(byte[] buf)
67 int l = buf.length - 1;
68 return buf[l-3] == 10 // '*\n'
69 && buf[l-2] == '.' // '.'
70 && buf[l-1] == 13 && buf[l] == 10; // '\r\n'
73 public byte[] getArticleData()
74 throws IOException, UnsupportedEncodingException
76 long maxSize = Config.inst().get(Config.ARTICLE_MAXSIZE, 1024) * 1024L;
80 this.out.write(("ARTICLE " + this.messageID + "\r\n").getBytes("UTF-8"));
83 String line = readln(this.in);
84 if(line.startsWith("220 "))
86 ByteArrayOutputStream buf = new ByteArrayOutputStream();
88 while(!eofArticle(buf.toByteArray()))
90 for(int b = in.read(); b != 10; b = in.read())
96 if(buf.size() > maxSize)
98 Log.get().warning("Skipping message that is too large: " + buf.size());
103 return buf.toByteArray();
107 Log.get().warning("ArticleReader: " + line);
111 catch(IOException ex)
117 this.out.write("QUIT\r\n".getBytes("UTF-8"));
123 private String readln(InputStream in)
126 ByteArrayOutputStream buf = new ByteArrayOutputStream();
127 for(int b = in.read(); b != 10 /* \n */; b = in.read())
132 return new String(buf.toByteArray());