diff options
Diffstat (limited to 'Swiften/FileTransfer/SOCKS5BytestreamRegistry.cpp')
| -rw-r--r-- | Swiften/FileTransfer/SOCKS5BytestreamRegistry.cpp | 50 | 
1 files changed, 8 insertions, 42 deletions
| diff --git a/Swiften/FileTransfer/SOCKS5BytestreamRegistry.cpp b/Swiften/FileTransfer/SOCKS5BytestreamRegistry.cpp index 8939579..cf4794c 100644 --- a/Swiften/FileTransfer/SOCKS5BytestreamRegistry.cpp +++ b/Swiften/FileTransfer/SOCKS5BytestreamRegistry.cpp @@ -11,62 +11,28 @@  #include <Swiften/Base/Algorithm.h>  #include <Swiften/Base/Log.h>  #include <Swiften/Base/foreach.h> -#include <Swiften/StringCodecs/Hexify.h> -#include <Swiften/Crypto/CryptoProvider.h>  namespace Swift {  SOCKS5BytestreamRegistry::SOCKS5BytestreamRegistry() {  } -void SOCKS5BytestreamRegistry::addReadBytestream(const std::string& destination, boost::shared_ptr<ReadBytestream> byteStream) { -	readBytestreams[destination] = byteStream; -} - -void SOCKS5BytestreamRegistry::removeReadBytestream(const std::string& destination) { -	readBytestreams.erase(destination); -} - -boost::shared_ptr<ReadBytestream> SOCKS5BytestreamRegistry::getReadBytestream(const std::string& destination) const { -	ReadBytestreamMap::const_iterator i = readBytestreams.find(destination); -	if (i != readBytestreams.end()) { -		return i->second; +void SOCKS5BytestreamRegistry::setHasBytestream(const std::string& destination, bool b) { +	if (b) { +		availableBytestreams.insert(destination); +	} +	else { +		availableBytestreams.erase(destination);  	} -	return boost::shared_ptr<ReadBytestream>(); -} - -void SOCKS5BytestreamRegistry::addWriteBytestream(const std::string& destination, boost::shared_ptr<WriteBytestream> byteStream) { -	writeBytestreams[destination] = byteStream; -} - -void SOCKS5BytestreamRegistry::removeWriteBytestream(const std::string& destination) { -	writeBytestreams.erase(destination);  } -boost::shared_ptr<WriteBytestream> SOCKS5BytestreamRegistry::getWriteBytestream(const std::string& destination) const { -	WriteBytestreamMap::const_iterator i = writeBytestreams.find(destination); -	if (i != writeBytestreams.end()) { -		return i->second; -	} -	return boost::shared_ptr<WriteBytestream>(); +bool SOCKS5BytestreamRegistry::hasBytestream(const std::string& destination) const { +	return availableBytestreams.find(destination) != availableBytestreams.end();  }  std::string SOCKS5BytestreamRegistry::generateSessionID() {  	return idGenerator.generateID();  } -SOCKS5BytestreamServerSession* SOCKS5BytestreamRegistry::getConnectedSession(const std::string& destination) { -	if (serverSessions.find(destination) != serverSessions.end()) { -		return serverSessions[destination]; -	} else { -		SWIFT_LOG(debug) << "No active connction for stream ID " << destination << " found!" << std::endl; -		return NULL; -	} -} - -std::string SOCKS5BytestreamRegistry::getHostname(const std::string& sessionID, const JID& requester, const JID& target, CryptoProvider* crypto) { -	return Hexify::hexify(crypto->getSHA1Hash(createSafeByteArray(sessionID + requester.toString() + target.toString()))); -} -  } | 
 Swift
 Swift