1 files changed, 3 insertions, 3 deletions

diff --git a/src/client.h b/src/client.h
index 4188a49..0b670e2 100644
--- a/src/client.h
+++ b/src/client.h
@@ -44,13 +44,12 @@ namespace Bu
                 void clearProtocol();
 
                 bool isOpen();
+                void close();
 
                 const Bu::Socket *getSocket() const;
 
-                /**
-                 *@todo Make this not suck.
-                 */
                 void disconnect();
+                bool wantsDisconnect();
 
         private:
                 Bu::Socket *pSocket;
@@ -58,6 +57,7 @@ namespace Bu
                 Bu::FString sReadBuf;
                 int nRBOffset;
                 Bu::FString sWriteBuf;
+                bool bWantsDisconnect;
         };
 }