1 files changed, 7 insertions, 9 deletions
diff --git a/Swiften/FileTransfer/SOCKS5BytestreamServer.h b/Swiften/FileTransfer/SOCKS5BytestreamServer.h
index 6bb598e..71605c4 100644
--- a/Swiften/FileTransfer/SOCKS5BytestreamServer.h
+++ b/Swiften/FileTransfer/SOCKS5BytestreamServer.h
@@ -1,5 +1,5 @@
 /*
- * Copyright (c) 2010 Remko Tronçon
+ * Copyright (c) 2010-2013 Remko Tronçon
  * Licensed under the GNU General Public License v3.
  * See Documentation/Licenses/GPLv3.txt for more information.
  */
@@ -17,26 +17,24 @@
 
 namespace Swift {
 	class SOCKS5BytestreamServerSession;
+	class CryptoProvider;
 
 	class SOCKS5BytestreamServer {
 		public:
-			SOCKS5BytestreamServer(boost::shared_ptr<ConnectionServer> connectionServer, SOCKS5BytestreamRegistry* registry);
+			SOCKS5BytestreamServer(
+					boost::shared_ptr<ConnectionServer> connectionServer, 
+					SOCKS5BytestreamRegistry* registry);
 
 			HostAddressPort getAddressPort() const;
 
 			void start();
 			void stop();
 
-			void addReadBytestream(const std::string& id, const JID& from, const JID& to, boost::shared_ptr<ReadBytestream> byteStream);
-			void removeReadBytestream(const std::string& id, const JID& from, const JID& to);
-
-		/*protected:
-			boost::shared_ptr<ReadBytestream> getBytestream(const std::string& dest);*/
+			std::vector< boost::shared_ptr<SOCKS5BytestreamServerSession> > getSessions(const std::string& id) const;
 
 		private:
 			void handleNewConnection(boost::shared_ptr<Connection> connection);
-
-			static std::string getSOCKSDestinationAddress(const std::string& id, const JID& from, const JID& to);
+			void handleSessionFinished(boost::shared_ptr<SOCKS5BytestreamServerSession>);
 
 		private:
 			friend class SOCKS5BytestreamServerSession;