#544 fixed.
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.daemon;
21 import org.sonews.util.Log;
22 import java.io.IOException;
23 import java.net.InetSocketAddress;
24 import java.net.SocketException;
25 import java.nio.ByteBuffer;
26 import java.nio.CharBuffer;
27 import java.nio.channels.ClosedChannelException;
28 import java.nio.channels.SelectionKey;
29 import java.nio.channels.SocketChannel;
30 import java.nio.charset.Charset;
31 import java.util.Arrays;
32 import java.util.Timer;
33 import java.util.TimerTask;
34 import org.sonews.daemon.command.Command;
35 import org.sonews.storage.Article;
36 import org.sonews.storage.Channel;
37 import org.sonews.util.Stats;
40 * For every SocketChannel (so TCP/IP connection) there is an instance of
42 * @author Christian Lins
45 public final class NNTPConnection
48 public static final String NEWLINE = "\r\n"; // RFC defines this as newline
49 public static final String MESSAGE_ID_PATTERN = "<[^>]+>";
51 private static final Timer cancelTimer = new Timer(true); // Thread-safe? True for run as daemon
53 /** SocketChannel is generally thread-safe */
54 private SocketChannel channel = null;
55 private Charset charset = Charset.forName("UTF-8");
56 private Command command = null;
57 private Article currentArticle = null;
58 private Channel currentGroup = null;
59 private volatile long lastActivity = System.currentTimeMillis();
60 private ChannelLineBuffers lineBuffers = new ChannelLineBuffers();
61 private int readLock = 0;
62 private final Object readLockGate = new Object();
63 private SelectionKey writeSelKey = null;
65 public NNTPConnection(final SocketChannel channel)
70 throw new IllegalArgumentException("channel is null");
73 this.channel = channel;
74 Stats.getInstance().clientConnect();
78 * Tries to get the read lock for this NNTPConnection. This method is Thread-
79 * safe and returns true of the read lock was successfully set. If the lock
80 * is still hold by another Thread the method returns false.
84 // As synchronizing simple types may cause deadlocks,
85 // we use a gate object.
86 synchronized(readLockGate)
94 readLock = Thread.currentThread().hashCode();
101 * Releases the read lock in a Thread-safe way.
102 * @throws IllegalMonitorStateException if a Thread not holding the lock
103 * tries to release it.
105 void unlockReadLock()
107 synchronized(readLockGate)
109 if(readLock == Thread.currentThread().hashCode())
115 throw new IllegalMonitorStateException();
121 * @return Current input buffer of this NNTPConnection instance.
123 public ByteBuffer getInputBuffer()
125 return this.lineBuffers.getInputBuffer();
129 * @return Output buffer of this NNTPConnection which has at least one byte
132 public ByteBuffer getOutputBuffer()
134 return this.lineBuffers.getOutputBuffer();
138 * @return ChannelLineBuffers instance associated with this NNTPConnection.
140 public ChannelLineBuffers getBuffers()
142 return this.lineBuffers;
146 * @return true if this connection comes from a local remote address.
148 public boolean isLocalConnection()
150 return ((InetSocketAddress)this.channel.socket().getRemoteSocketAddress())
151 .getHostName().equalsIgnoreCase("localhost");
154 void setWriteSelectionKey(SelectionKey selKey)
156 this.writeSelKey = selKey;
159 public void shutdownInput()
163 // Closes the input line of the channel's socket, so no new data
164 // will be received and a timeout can be triggered.
165 this.channel.socket().shutdownInput();
167 catch(IOException ex)
169 Log.get().warning("Exception in NNTPConnection.shutdownInput(): " + ex);
173 public void shutdownOutput()
175 cancelTimer.schedule(new TimerTask()
182 // Closes the output line of the channel's socket.
183 channel.socket().shutdownOutput();
186 catch(SocketException ex)
188 // Socket was already disconnected
189 Log.get().info("NNTPConnection.shutdownOutput(): " + ex);
193 Log.get().warning("NNTPConnection.shutdownOutput(): " + ex);
199 public SocketChannel getSocketChannel()
204 public Article getCurrentArticle()
206 return this.currentArticle;
209 public Charset getCurrentCharset()
215 * @return The currently selected communication channel (not SocketChannel)
217 public Channel getCurrentChannel()
219 return this.currentGroup;
222 public void setCurrentArticle(final Article article)
224 this.currentArticle = article;
227 public void setCurrentGroup(final Channel group)
229 this.currentGroup = group;
232 public long getLastActivity()
234 return this.lastActivity;
238 * Due to the readLockGate there is no need to synchronize this method.
240 * @throws IllegalArgumentException if raw is null.
241 * @throws IllegalStateException if calling thread does not own the readLock.
243 void lineReceived(byte[] raw)
247 throw new IllegalArgumentException("raw is null");
250 if(readLock == 0 || readLock != Thread.currentThread().hashCode())
252 throw new IllegalStateException("readLock not properly set");
255 this.lastActivity = System.currentTimeMillis();
257 String line = new String(raw, this.charset);
259 // There might be a trailing \r, but trim() is a bad idea
260 // as it removes also leading spaces from long header lines.
261 if(line.endsWith("\r"))
263 line = line.substring(0, line.length() - 1);
264 raw = Arrays.copyOf(raw, raw.length - 1);
267 Log.get().fine("<< " + line);
271 command = parseCommandLine(line);
272 assert command != null;
277 // The command object will process the line we just received
278 command.processLine(this, line, raw);
280 catch(ClosedChannelException ex0)
284 Log.get().info("Connection to " + channel.socket().getRemoteSocketAddress()
285 + " closed: " + ex0);
287 catch(Exception ex0a)
289 ex0a.printStackTrace();
297 ex1.printStackTrace();
298 println("500 Internal server error");
302 ex2.printStackTrace();
306 if(command == null || command.hasFinished())
309 charset = Charset.forName("UTF-8"); // Reset to default
314 * This method determines the fitting command processing class.
318 private Command parseCommandLine(String line)
320 String cmdStr = line.split(" ")[0];
321 return CommandSelector.getInstance().get(cmdStr);
325 * Puts the given line into the output buffer, adds a newline character
326 * and returns. The method returns immediately and does not block until
327 * the line was sent. If line is longer than 510 octets it is split up in
328 * several lines. Each line is terminated by \r\n (NNTPConnection.NEWLINE).
331 public void println(final CharSequence line, final Charset charset)
334 writeToChannel(CharBuffer.wrap(line), charset, line);
335 writeToChannel(CharBuffer.wrap(NEWLINE), charset, null);
339 * Writes the given raw lines to the output buffers and finishes with
340 * a newline character (\r\n).
343 public void println(final byte[] rawLines)
346 this.lineBuffers.addOutputBuffer(ByteBuffer.wrap(rawLines));
347 writeToChannel(CharBuffer.wrap(NEWLINE), charset, null);
351 * Encodes the given CharBuffer using the given Charset to a bunch of
352 * ByteBuffers (each 512 bytes large) and enqueues them for writing at the
353 * connected SocketChannel.
354 * @throws java.io.IOException
356 private void writeToChannel(CharBuffer characters, final Charset charset,
357 CharSequence debugLine)
360 if(!charset.canEncode())
362 Log.get().severe("FATAL: Charset " + charset + " cannot encode!");
366 // Write characters to output buffers
367 LineEncoder lenc = new LineEncoder(characters, charset);
368 lenc.encode(lineBuffers);
370 enableWriteEvents(debugLine);
373 private void enableWriteEvents(CharSequence debugLine)
375 // Enable OP_WRITE events so that the buffers are processed
378 this.writeSelKey.interestOps(SelectionKey.OP_WRITE);
379 ChannelWriter.getInstance().getSelector().wakeup();
381 catch(Exception ex) // CancelledKeyException and ChannelCloseException
383 Log.get().warning("NNTPConnection.writeToChannel(): " + ex);
387 // Update last activity timestamp
388 this.lastActivity = System.currentTimeMillis();
389 if(debugLine != null)
391 Log.get().fine(">> " + debugLine);
395 public void println(final CharSequence line)
398 println(line, charset);
401 public void print(final String line)
404 writeToChannel(CharBuffer.wrap(line), charset, line);
407 public void setCurrentCharset(final Charset charset)
409 this.charset = charset;