1 files changed, 15 insertions, 4 deletions
diff --git a/Swiften/FileTransfer/SOCKS5BytestreamClientSession.h b/Swiften/FileTransfer/SOCKS5BytestreamClientSession.h
index 0b9790d..287cf3b 100644
--- a/Swiften/FileTransfer/SOCKS5BytestreamClientSession.h
+++ b/Swiften/FileTransfer/SOCKS5BytestreamClientSession.h
@@ -38,14 +38,19 @@ public:
 		Ready,
 		Writing,
 		Reading,
-		Finished,
+		Finished
 	};
 
 public:
 	typedef boost::shared_ptr<SOCKS5BytestreamClientSession> ref;
 
 public:
-	SOCKS5BytestreamClientSession(boost::shared_ptr<Connection> connection, const HostAddressPort&, const std::string&, TimerFactory*);
+	SOCKS5BytestreamClientSession(
+			boost::shared_ptr<Connection> connection, 
+			const HostAddressPort&, 
+			const std::string&, 
+			TimerFactory*);
+	~SOCKS5BytestreamClientSession();
 
 	void start();
 	void stop();
@@ -58,8 +63,8 @@ public:
 	boost::signal<void (bool /*error*/)> onSessionReady;
 
 	boost::signal<void (boost::optional<FileTransferError>)> onFinished;
-	boost::signal<void (int)> onBytesSent;
-	boost::signal<void (int)> onBytesReceived;
+	boost::signal<void (size_t)> onBytesSent;
+	boost::signal<void (size_t)> onBytesReceived;
 
 private:
 	void process();
@@ -73,6 +78,7 @@ private:
 
 	void finish(bool error);
 	void sendData();
+	void closeConnection();
 
 private:
 	boost::shared_ptr<Connection> connection;
@@ -89,6 +95,11 @@ private:
 	boost::shared_ptr<ReadBytestream> readBytestream;
 
 	Timer::ref weFailedTimeout;
+
+	boost::bsignals::connection connectFinishedConnection;
+	boost::bsignals::connection dataWrittenConnection;
+	boost::bsignals::connection dataReadConnection;
+	boost::bsignals::connection disconnectedConnection;
 };
 
 }