X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Fnet%2Fpterodactylus%2Futil%2Ffcp%2FFcpConnection.java;h=d585022f3e17ab472abcabe082b2445514cc531a;hb=2b77426f2c37342d979fb44b8bb89991d61df6d0;hp=f5629ca1166eb85efcecdf1dfa268b52d1d4c728;hpb=5c3d23ad2d5f3567b283e7e86ca55787033cbc0f;p=jSite2.git

diff --git a/src/net/pterodactylus/util/fcp/FcpConnection.java b/src/net/pterodactylus/util/fcp/FcpConnection.java
index f5629ca..d585022 100644
--- a/src/net/pterodactylus/util/fcp/FcpConnection.java
+++ b/src/net/pterodactylus/util/fcp/FcpConnection.java
@@ -70,6 +70,17 @@ public class FcpConnection {
 	private Map<String, Integer> incomingMessageStatistics = Collections.synchronizedMap(new HashMap<String, Integer>());
 
 	/**
+	 * Creates a new FCP connection to the freenet node running on localhost,
+	 * using the default port.
+	 * 
+	 * @throws UnknownHostException
+	 *             if the hostname can not be resolved
+	 */
+	public FcpConnection() throws UnknownHostException {
+		this(InetAddress.getLocalHost());
+	}
+
+	/**
 	 * Creates a new FCP connection to the Freenet node running on the given
 	 * host, listening on the default port.
 	 * 
@@ -292,6 +303,18 @@ public class FcpConnection {
 	}
 
 	/**
+	 * Notifies all listeners that a “PersistentGet” message was received.
+	 * 
+	 * @param persistentGet
+	 *            The “PersistentGet” message
+	 */
+	private void fireReceivedPersistentGet(PersistentGet persistentGet) {
+		for (FcpListener fcpListener: fcpListeners) {
+			fcpListener.receivedPersistentGet(this, persistentGet);
+		}
+	}
+
+	/**
 	 * Notifies all listeners that a “PersistentPut” message was received.
 	 * 
 	 * @see FcpListener#receivedPersistentPut(FcpConnection, PersistentPut)
@@ -330,6 +353,18 @@ public class FcpConnection {
 	}
 
 	/**
+	 * Notifies all listeners that a “DataFound” message was received.
+	 * 
+	 * @param dataFound
+	 *            The “DataFound” message
+	 */
+	private void fireReceivedDataFound(DataFound dataFound) {
+		for (FcpListener fcpListener: fcpListeners) {
+			fcpListener.receivedDataFound(this, dataFound);
+		}
+	}
+
+	/**
 	 * Notifies all listeners that an “AllData” message was received.
 	 * 
 	 * @param allData
@@ -378,6 +413,19 @@ public class FcpConnection {
 	}
 
 	/**
+	 * Notifies all listeners that an “UnknownPeerNoteType” message was
+	 * received.
+	 * 
+	 * @param unknownPeerNoteType
+	 *            The “UnknownPeerNoteType” message
+	 */
+	private void fireReceivedUnknownPeerNoteType(UnknownPeerNoteType unknownPeerNoteType) {
+		for (FcpListener fcpListener: fcpListeners) {
+			fcpListener.receivedUnknownPeerNoteType(this, unknownPeerNoteType);
+		}
+	}
+
+	/**
 	 * Notifies all listeners that a “ProtocolError” message was received.
 	 * 
 	 * @param protocolError
@@ -468,6 +516,8 @@ public class FcpConnection {
 			fireReceivedSimpleProgress(new SimpleProgress(fcpMessage));
 		} else if ("ProtocolError".equals(messageName)) {
 			fireReceivedProtocolError(new ProtocolError(fcpMessage));
+		} else if ("PersistentGet".equals(messageName)) {
+			fireReceivedPersistentGet(new PersistentGet(fcpMessage));
 		} else if ("PersistentPut".equals(messageName)) {
 			fireReceivedPersistentPut(new PersistentPut(fcpMessage));
 		} else if ("URIGenerated".equals(messageName)) {
@@ -482,6 +532,8 @@ public class FcpConnection {
 			fireReceivedStartedCompression(new StartedCompression(fcpMessage));
 		} else if ("FinishedCompression".equals(messageName)) {
 			fireReceivedFinishedCompression(new FinishedCompression(fcpMessage));
+		} else if ("DataFound".equals(messageName)) {
+			fireReceivedDataFound(new DataFound(fcpMessage));
 		} else if ("AllData".equals(messageName)) {
 			long dataLength;
 			try {
@@ -505,6 +557,8 @@ public class FcpConnection {
 			fireReceivedSSKKeypair(new SSKKeypair(fcpMessage));
 		} else if ("PeerRemoved".equals(messageName)) {
 			fireReceivedPeerRemoved(new PeerRemoved(fcpMessage));
+		} else if ("UnknownPeerNoteType".equals(messageName)) {
+			fireReceivedUnknownPeerNoteType(new UnknownPeerNoteType(fcpMessage));
 		} else if ("NodeData".equals(messageName)) {
 			fireReceivedNodeData(new NodeData(fcpMessage));
 		} else if ("TestDDAReply".equals(messageName)) {
@@ -520,6 +574,16 @@ public class FcpConnection {
 		}
 	}
 
+	/**
+	 * Handles a disconnect from the node.
+	 */
+	synchronized void handleDisconnect() {
+		Closer.close(remoteInputStream);
+		Closer.close(remoteOutputStream);
+		Closer.close(remoteSocket);
+		connectionHandler = null;
+	}
+
 	//
 	// PRIVATE METHODS
 	//