summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
Diffstat (limited to 'Swiften/FileTransfer/SOCKS5BytestreamRegistry.cpp')
-rw-r--r--Swiften/FileTransfer/SOCKS5BytestreamRegistry.cpp50
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())));
-}
-
}