2 * jSite2 - FpcConnection.java -
3 * Copyright © 2008 David Roden
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 2 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, write to the Free Software
17 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
20 package net.pterodactylus.util.fcp;
22 import java.io.IOException;
23 import java.io.InputStream;
24 import java.io.OutputStream;
25 import java.net.InetAddress;
26 import java.net.Socket;
27 import java.net.UnknownHostException;
28 import java.util.ArrayList;
29 import java.util.Collections;
30 import java.util.HashMap;
31 import java.util.List;
34 import net.pterodactylus.util.io.Closer;
35 import net.pterodactylus.util.io.LimitedInputStream;
38 * An FCP connection to a Freenet node.
40 * @author David ‘Bombe’ Roden <bombe@freenetproject.org>
43 public class FcpConnection {
45 /** The default port for FCP v2. */
46 public static final int DEFAULT_PORT = 9481;
48 /** The list of FCP listeners. */
49 private final List<FcpListener> fcpListeners = new ArrayList<FcpListener>();
51 /** The address of the node. */
52 private final InetAddress address;
54 /** The port number of the node’s FCP port. */
55 private final int port;
57 /** The remote socket. */
58 private Socket remoteSocket;
60 /** The input stream from the node. */
61 private InputStream remoteInputStream;
63 /** The output stream to the node. */
64 private OutputStream remoteOutputStream;
66 /** The connection handler. */
67 private FcpConnectionHandler connectionHandler;
69 /** Incoming message statistics. */
70 private Map<String, Integer> incomingMessageStatistics = Collections.synchronizedMap(new HashMap<String, Integer>());
73 * Creates a new FCP connection to the Freenet node running on the given
74 * host, listening on the default port.
77 * The hostname of the Freenet node
78 * @throws UnknownHostException
79 * if <code>host</code> can not be resolved
81 public FcpConnection(String host) throws UnknownHostException {
82 this(host, DEFAULT_PORT);
86 * Creates a new FCP connection to the Freenet node running on the given
87 * host, listening on the given port.
90 * The hostname of the Freenet node
92 * The port number of the node’s FCP port
93 * @throws UnknownHostException
94 * if <code>host</code> can not be resolved
96 public FcpConnection(String host, int port) throws UnknownHostException {
97 this(InetAddress.getByName(host), port);
101 * Creates a new FCP connection to the Freenet node running at the given
102 * address, listening on the default port.
105 * The address of the Freenet node
107 public FcpConnection(InetAddress address) {
108 this(address, DEFAULT_PORT);
112 * Creates a new FCP connection to the Freenet node running at the given
113 * address, listening on the given port.
116 * The address of the Freenet node
118 * The port number of the node’s FCP port
120 public FcpConnection(InetAddress address, int port) {
121 this.address = address;
126 // LISTENER MANAGEMENT
130 * Adds the given listener to the list of listeners.
133 * The listener to add
135 public void addFcpListener(FcpListener fcpListener) {
136 fcpListeners.add(fcpListener);
140 * Removes the given listener from the list of listeners.
143 * The listener to remove
145 public void removeFcpListener(FcpListener fcpListener) {
146 fcpListeners.remove(fcpListener);
150 * Notifies listeners that a “NodeHello” message was received.
152 * @see FcpListener#receivedNodeHello(FcpConnection, NodeHello)
154 * The “NodeHello” message
156 private void fireReceivedNodeHello(NodeHello nodeHello) {
157 for (FcpListener fcpListener: fcpListeners) {
158 fcpListener.receivedNodeHello(this, nodeHello);
163 * Notifies listeners that a “CloseConnectionDuplicateClientName” message
166 * @see FcpListener#receivedCloseConnectionDuplicateClientName(FcpConnection,
167 * CloseConnectionDuplicateClientName)
168 * @param closeConnectionDuplicateClientName
169 * The “CloseConnectionDuplicateClientName” message
171 private void fireReceivedCloseConnectionDuplicateClientName(CloseConnectionDuplicateClientName closeConnectionDuplicateClientName) {
172 for (FcpListener fcpListener: fcpListeners) {
173 fcpListener.receivedCloseConnectionDuplicateClientName(this, closeConnectionDuplicateClientName);
178 * Notifies listeners that a “SSKKeypair” message was received.
180 * @see FcpListener#receivedSSKKeypair(FcpConnection, SSKKeypair)
182 * The “SSKKeypair” message
184 private void fireReceivedSSKKeypair(SSKKeypair sskKeypair) {
185 for (FcpListener fcpListener: fcpListeners) {
186 fcpListener.receivedSSKKeypair(this, sskKeypair);
191 * Notifies listeners that a “Peer” message was received.
193 * @see FcpListener#receivedPeer(FcpConnection, Peer)
197 private void fireReceivedPeer(Peer peer) {
198 for (FcpListener fcpListener: fcpListeners) {
199 fcpListener.receivedPeer(this, peer);
204 * Notifies all listeners that an “EndListPeers” message was received.
206 * @see FcpListener#receivedEndListPeers(FcpConnection, EndListPeers)
207 * @param endListPeers
208 * The “EndListPeers” message
210 private void fireReceivedEndListPeers(EndListPeers endListPeers) {
211 for (FcpListener fcpListener: fcpListeners) {
212 fcpListener.receivedEndListPeers(this, endListPeers);
217 * Notifies all listeners that a “PeerNote” message was received.
219 * @see FcpListener#receivedPeerNote(FcpConnection, PeerNote)
222 private void fireReceivedPeerNote(PeerNote peerNote) {
223 for (FcpListener fcpListener: fcpListeners) {
224 fcpListener.receivedPeerNote(this, peerNote);
229 * Notifies all listeners that an “EndListPeerNotes” message was received.
231 * @see FcpListener#receivedEndListPeerNotes(FcpConnection,
233 * @param endListPeerNotes
234 * The “EndListPeerNotes” message
236 private void fireReceivedEndListPeerNotes(EndListPeerNotes endListPeerNotes) {
237 for (FcpListener fcpListener: fcpListeners) {
238 fcpListener.receivedEndListPeerNotes(this, endListPeerNotes);
243 * Notifies all listeners that a “PeerRemoved” message was received.
245 * @see FcpListener#receivedPeerRemoved(FcpConnection, PeerRemoved)
247 * The “PeerRemoved” message
249 private void fireReceivedPeerRemoved(PeerRemoved peerRemoved) {
250 for (FcpListener fcpListener: fcpListeners) {
251 fcpListener.receivedPeerRemoved(this, peerRemoved);
256 * Notifies all listeners that a “NodeData” message was received.
258 * @see FcpListener#receivedNodeData(FcpConnection, NodeData)
260 * The “NodeData” message
262 private void fireReceivedNodeData(NodeData nodeData) {
263 for (FcpListener fcpListener: fcpListeners) {
264 fcpListener.receivedNodeData(this, nodeData);
269 * Notifies all listeners that a “TestDDAReply” message was received.
271 * @see FcpListener#receivedTestDDAReply(FcpConnection, TestDDAReply)
272 * @param testDDAReply
273 * The “TestDDAReply” message
275 private void fireReceivedTestDDAReply(TestDDAReply testDDAReply) {
276 for (FcpListener fcpListener: fcpListeners) {
277 fcpListener.receivedTestDDAReply(this, testDDAReply);
282 * Notifies all listeners that a “TestDDAComplete” message was received.
284 * @see FcpListener#receivedTestDDAComplete(FcpConnection, TestDDAComplete)
285 * @param testDDAComplete
286 * The “TestDDAComplete” message
288 private void fireReceivedTestDDAComplete(TestDDAComplete testDDAComplete) {
289 for (FcpListener fcpListener: fcpListeners) {
290 fcpListener.receivedTestDDAComplete(this, testDDAComplete);
295 * Notifies all listeners that a “PersistentPut” message was received.
297 * @see FcpListener#receivedPersistentPut(FcpConnection, PersistentPut)
298 * @param persistentPut
299 * The “PersistentPut” message
301 private void fireReceivedPersistentPut(PersistentPut persistentPut) {
302 for (FcpListener fcpListener: fcpListeners) {
303 fcpListener.receivedPersistentPut(this, persistentPut);
308 * Notifies all listeners that a “EndListPersistentRequests” message was
311 * @param endListPersistentRequests
312 * The “EndListPersistentRequests” message
314 private void fireReceivedEndListPersistentRequests(EndListPersistentRequests endListPersistentRequests) {
315 for (FcpListener fcpListener: fcpListeners) {
316 fcpListener.receivedEndListPersistentRequests(this, endListPersistentRequests);
321 * Notifies all listeners that a “URIGenerated” message was received.
323 * @param uriGenerated
324 * The “URIGenerated” message
326 private void fireReceivedURIGenerated(URIGenerated uriGenerated) {
327 for (FcpListener fcpListener: fcpListeners) {
328 fcpListener.receivedURIGenerated(this, uriGenerated);
333 * Notifies all listeners that an “AllData” message was received.
336 * The “AllData” message
338 private void fireReceivedAllData(AllData allData) {
339 for (FcpListener fcpListener: fcpListeners) {
340 fcpListener.receivedAllData(this, allData);
345 * Notifies all listeners that a “SimpleProgress” message was received.
347 * @param simpleProgress
348 * The “SimpleProgress” message
350 private void fireReceivedSimpleProgress(SimpleProgress simpleProgress) {
351 for (FcpListener fcpListener: fcpListeners) {
352 fcpListener.receivedSimpleProgress(this, simpleProgress);
357 * Notifies all listeners that a “StartedCompression” message was received.
359 * @param startedCompression
360 * The “StartedCompression” message
362 private void fireReceivedStartedCompression(StartedCompression startedCompression) {
363 for (FcpListener fcpListener: fcpListeners) {
364 fcpListener.receivedStartedCompression(this, startedCompression);
369 * Notifies all listeners that a “FinishedCompression” message was received.
371 * @param finishedCompression
372 * The “FinishedCompression” message
374 private void fireReceivedFinishedCompression(FinishedCompression finishedCompression) {
375 for (FcpListener fcpListener: fcpListeners) {
376 fcpListener.receviedFinishedCompression(this, finishedCompression);
381 * Notifies all listeners that a “ProtocolError” message was received.
383 * @param protocolError
384 * The “ProtocolError” message
386 private void fireReceivedProtocolError(ProtocolError protocolError) {
387 for (FcpListener fcpListener: fcpListeners) {
388 fcpListener.receivedProtocolError(this, protocolError);
393 * Notifies all registered listeners that a message has been received.
395 * @see FcpListener#receivedMessage(FcpConnection, FcpMessage)
397 * The message that was received
399 private void fireMessageReceived(FcpMessage fcpMessage) {
400 for (FcpListener fcpListener: fcpListeners) {
401 fcpListener.receivedMessage(this, fcpMessage);
410 * Connects to the node.
412 * @throws IOException
413 * if an I/O error occurs
414 * @throws IllegalStateException
415 * if there is already a connection to the node
417 public synchronized void connect() throws IOException, IllegalStateException {
418 if (connectionHandler != null) {
419 throw new IllegalStateException("already connected, disconnect first");
421 remoteSocket = new Socket(address, port);
422 remoteInputStream = remoteSocket.getInputStream();
423 remoteOutputStream = remoteSocket.getOutputStream();
424 new Thread(connectionHandler = new FcpConnectionHandler(this, remoteInputStream)).start();
428 * Disconnects from the node. If there is no connection to the node, this
429 * method does nothing.
431 public synchronized void disconnect() {
432 if (connectionHandler == null) {
435 Closer.close(remoteSocket);
436 connectionHandler.stop();
437 connectionHandler = null;
441 * Sends the given FCP message.
444 * The FCP message to send
445 * @throws IOException
446 * if an I/O error occurs
448 public synchronized void sendMessage(FcpMessage fcpMessage) throws IOException {
449 System.out.println("sending message: " + fcpMessage.getName());
450 fcpMessage.write(remoteOutputStream);
454 // PACKAGE-PRIVATE METHODS
458 * Handles the given message, notifying listeners. This message should only
459 * be called by {@link FcpConnectionHandler}.
462 * The received message
464 void handleMessage(FcpMessage fcpMessage) {
465 String messageName = fcpMessage.getName();
466 countMessage(messageName);
467 if ("SimpleProgress".equals(messageName)) {
468 fireReceivedSimpleProgress(new SimpleProgress(fcpMessage));
469 } else if ("ProtocolError".equals(messageName)) {
470 fireReceivedProtocolError(new ProtocolError(fcpMessage));
471 } else if ("PersistentPut".equals(messageName)) {
472 fireReceivedPersistentPut(new PersistentPut(fcpMessage));
473 } else if ("URIGenerated".equals(messageName)) {
474 fireReceivedURIGenerated(new URIGenerated(fcpMessage));
475 } else if ("EndListPersistentRequests".equals(messageName)) {
476 fireReceivedEndListPersistentRequests(new EndListPersistentRequests(fcpMessage));
477 } else if ("Peer".equals(messageName)) {
478 fireReceivedPeer(new Peer(fcpMessage));
479 } else if ("PeerNote".equals(messageName)) {
480 fireReceivedPeerNote(new PeerNote(fcpMessage));
481 } else if ("StartedCompression".equals(messageName)) {
482 fireReceivedStartedCompression(new StartedCompression(fcpMessage));
483 } else if ("FinishedCompression".equals(messageName)) {
484 fireReceivedFinishedCompression(new FinishedCompression(fcpMessage));
485 } else if ("AllData".equals(messageName)) {
488 dataLength = Long.valueOf(fcpMessage.getField("DataLength"));
489 } catch (NumberFormatException nfe1) {
492 LimitedInputStream payloadInputStream = new LimitedInputStream(remoteInputStream, dataLength);
493 fireReceivedAllData(new AllData(fcpMessage, payloadInputStream));
495 payloadInputStream.consume();
496 } catch (IOException ioe1) {
497 /* FIXME - what now? */
498 /* well, ignore. when the connection handler fails, all fails. */
500 } else if ("EndListPeerNotes".equals(messageName)) {
501 fireReceivedEndListPeerNotes(new EndListPeerNotes(fcpMessage));
502 } else if ("EndListPeers".equals(messageName)) {
503 fireReceivedEndListPeers(new EndListPeers(fcpMessage));
504 } else if ("SSKKeypair".equals(messageName)) {
505 fireReceivedSSKKeypair(new SSKKeypair(fcpMessage));
506 } else if ("PeerRemoved".equals(messageName)) {
507 fireReceivedPeerRemoved(new PeerRemoved(fcpMessage));
508 } else if ("NodeData".equals(messageName)) {
509 fireReceivedNodeData(new NodeData(fcpMessage));
510 } else if ("TestDDAReply".equals(messageName)) {
511 fireReceivedTestDDAReply(new TestDDAReply(fcpMessage));
512 } else if ("TestDDAComplete".equals(messageName)) {
513 fireReceivedTestDDAComplete(new TestDDAComplete(fcpMessage));
514 } else if ("NodeHello".equals(messageName)) {
515 fireReceivedNodeHello(new NodeHello(fcpMessage));
516 } else if ("CloseConnectionDuplicateClientName".equals(messageName)) {
517 fireReceivedCloseConnectionDuplicateClientName(new CloseConnectionDuplicateClientName(fcpMessage));
519 fireMessageReceived(fcpMessage);
528 * Incremets the counter in {@link #incomingMessageStatistics} by <cod>1</code>
529 * for the given message name.
532 * The name of the message to count
534 private void countMessage(String name) {
536 if (incomingMessageStatistics.containsKey(name)) {
537 oldValue = incomingMessageStatistics.get(name);
539 incomingMessageStatistics.put(name, oldValue + 1);