1 files changed, 12 insertions, 1 deletions

diff --git a/src/client.cpp b/src/client.cpp
index d416700..19d2778 100644
--- a/src/client.cpp
+++ b/src/client.cpp
@@ -11,7 +11,8 @@
 Bu::Client::Client( Bu::Socket *pSocket ) :
         pSocket( pSocket ),
         pProto( NULL ),
-        nRBOffset( 0 )
+        nRBOffset( 0 ),
+        bWantsDisconnect( false )
 {
 }
 
@@ -205,5 +206,15 @@ const Bu::Socket *Bu::Client::getSocket() const
 
 void Bu::Client::disconnect()
 {
+        bWantsDisconnect = true;
 }
 
+bool Bu::Client::wantsDisconnect()
+{
+        return bWantsDisconnect;
+}
+
+void Bu::Client::close()
+{
+        pSocket->close();
+}