From 376d54397d1a7a7803776ed2f030999caaf56889 Mon Sep 17 00:00:00 2001
From: =?utf8?q?David=20=E2=80=98Bombe=E2=80=99=20Roden?=
 <bombe@freenetproject.org>
Date: Wed, 20 May 2009 07:41:53 +0200
Subject: [PATCH] =?utf8?q?Add=20=E2=80=9Cconnected=E2=80=9D=20flag.=20Chec?=
 =?utf8?q?k=20for=20client=20being=20connected=20before=20trying=20to=20co?=
 =?utf8?q?mmunicate=20with=20the=20node.?=
MIME-Version: 1.0
Content-Type: text/plain; charset=utf8
Content-Transfer-Encoding: 8bit

---
 src/net/pterodactylus/fcp/highlevel/FcpClient.java | 32 ++++++++++++++++++++++
 1 file changed, 32 insertions(+)

diff --git a/src/net/pterodactylus/fcp/highlevel/FcpClient.java b/src/net/pterodactylus/fcp/highlevel/FcpClient.java
index f90ab77..7f2d8bd 100644
--- a/src/net/pterodactylus/fcp/highlevel/FcpClient.java
+++ b/src/net/pterodactylus/fcp/highlevel/FcpClient.java
@@ -81,6 +81,9 @@ public class FcpClient {
 	/** The underlying FCP connection. */
 	private final FcpConnection fcpConnection;
 
+	/** Whether the client is currently connected. */
+	private volatile boolean connected;
+
 	/**
 	 * Creates an FCP client with the given name.
 	 *
@@ -163,6 +166,8 @@ public class FcpClient {
 	 *             if an FCP error occurs
 	 */
 	public void connect() throws IOException, FcpException {
+		checkConnected(false);
+		connected = true;
 		new ExtendedFcpAdapter() {
 
 			/**
@@ -808,6 +813,28 @@ public class FcpClient {
 	}
 
 	/**
+	 * Checks whether the connection is in the required state.
+	 *
+	 * @param connected
+	 *            The required connection state
+	 * @throws FcpException
+	 *             if the connection is not in the required state
+	 */
+	private void checkConnected(boolean connected) throws FcpException {
+		if (this.connected != connected) {
+			throw new FcpException("Client is " + (connected ? "not" : "already") + " connected.");
+		}
+	}
+
+	/**
+	 * Tells the client that it is now disconnected. This method is called by
+	 * {@link ExtendedFcpAdapter} only.
+	 */
+	private void setDisconnected() {
+		connected = false;
+	}
+
+	/**
 	 * Implementation of an {@link FcpListener} that can store an
 	 * {@link FcpException} and wait for the arrival of a certain command.
 	 *
@@ -839,6 +866,7 @@ public class FcpClient {
 		 */
 		@SuppressWarnings("synthetic-access")
 		public void execute() throws IOException, FcpException {
+			checkConnected(true);
 			fcpConnection.addFcpListener(this);
 			try {
 				run();
@@ -850,10 +878,14 @@ public class FcpClient {
 						/* ignore, we’ll loop. */
 					}
 				}
+			} catch (IOException ioe1) {
+				setDisconnected();
+				throw ioe1;
 			} finally {
 				fcpConnection.removeFcpListener(this);
 			}
 			if (fcpException != null) {
+				setDisconnected();
 				throw fcpException;
 			}
 		}
-- 
2.7.4