- * Copyright (c) 2010 Remko Tronçon

- fileTransferManager = new FileTransferManagerImpl(getJID(), jingleSessionManager, getIQRouter(), getEntityCapsProvider(), presenceOracle, getNetworkFactories()->getConnectionFactory(), getNetworkFactories()->getConnectionServerFactory(), getNetworkFactories()->getTimerFactory(), getNetworkFactories()->getNATTraverser(), getNetworkFactories()->getCryptoProvider());

- unsigned int getBlockSize() const {

- unsigned int blockSize;

-#include <Swiften/FileTransfer/DefaultLocalJingleTransportCandidateGeneratorFactory.h>

-#include <Swiften/FileTransfer/DefaultRemoteJingleTransportCandidateSelectorFactory.h>

- FileReceiver(const JID& jid, const std::string& password) : jid(jid), password(password), jingleSessionManager(NULL), incomingFileTransferManager(NULL) {

- client->getFileTransferManager()->startListeningOnPort(9999);

- transfer->accept(boost::make_shared<FileWriteBytestream>("out"));

- //transfer->onFinished.connect(boost::bind(&FileReceiver::handleFileTransferFinished, this, _1));

- //transfer->start();

- /*

- void handleFileTransferFinished(const boost::optional<FileTransferError>& error) {

- }*/

- JingleSessionManager* jingleSessionManager;

- IncomingFileTransferManager* incomingFileTransferManager;

-#include <Swiften/FileTransfer/DefaultLocalJingleTransportCandidateGeneratorFactory.h>

-#include <Swiften/FileTransfer/DefaultRemoteJingleTransportCandidateSelectorFactory.h>

- client->getFileTransferManager()->startListeningOnPort(19999);

-/*

- * Copyright (c) 2011 Tobias Markmann

- * Licensed under the simplified BSD license.

- * See Documentation/Licenses/BSD-simplified.txt for more information.

- */

-

-#include "ConnectivityManager.h"

-

-#include <boost/bind.hpp>

-

-#include <Swiften/Base/foreach.h>

-#include <Swiften/Base/Log.h>

-#include <Swiften/Network/NetworkInterface.h>

-#include <Swiften/Network/NATTraversalGetPublicIPRequest.h>

-#include <Swiften/Network/NATTraversalRemovePortForwardingRequest.h>

-#include <Swiften/Network/NATTraverser.h>

-#include <Swiften/Network/PlatformNetworkEnvironment.h>

-

-namespace Swift {

-

-ConnectivityManager::ConnectivityManager(NATTraverser* worker) : natTraversalWorker(worker) {

-

-}

-

-ConnectivityManager::~ConnectivityManager() {

- std::set<int> leftOpenPorts = ports;

- foreach(int port, leftOpenPorts) {

- removeListeningPort(port);

- }

-}

-

-void ConnectivityManager::addListeningPort(int port) {

- ports.insert(port);

- boost::shared_ptr<NATTraversalGetPublicIPRequest> getIPRequest = natTraversalWorker->createGetPublicIPRequest();

- if (getIPRequest) {

- getIPRequest->onResult.connect(boost::bind(&ConnectivityManager::natTraversalGetPublicIPResult, this, _1));

- getIPRequest->start();

- }

-

- boost::shared_ptr<NATTraversalForwardPortRequest> forwardPortRequest = natTraversalWorker->createForwardPortRequest(port, port);

- if (forwardPortRequest) {

- forwardPortRequest->onResult.connect(boost::bind(&ConnectivityManager::natTraversalForwardPortResult, this, _1));

- forwardPortRequest->start();

- }

-}

-

-void ConnectivityManager::removeListeningPort(int port) {

- SWIFT_LOG(debug) << "remove listening port " << port << std::endl;

- ports.erase(port);

- boost::shared_ptr<NATTraversalRemovePortForwardingRequest> removePortForwardingRequest = natTraversalWorker->createRemovePortForwardingRequest(port, port);

- if (removePortForwardingRequest) {

- removePortForwardingRequest->start();

- }

-}

-

-std::vector<HostAddressPort> ConnectivityManager::getHostAddressPorts() const {

- PlatformNetworkEnvironment env;

- std::vector<HostAddressPort> results;

-

- std::vector<HostAddress> addresses;

-

- std::vector<NetworkInterface> networkInterfaces;

- foreach (const NetworkInterface& iface, networkInterfaces) {

- foreach (const HostAddress& address, iface.getAddresses()) {

- foreach (int port, ports) {

- results.push_back(HostAddressPort(address, port));

- }

- }

- }

-

- return results;

-}

-

-std::vector<HostAddressPort> ConnectivityManager::getAssistedHostAddressPorts() const {

- std::vector<HostAddressPort> results;

-

- if (publicAddress) {

- foreach (int port, ports) {

- results.push_back(HostAddressPort(publicAddress.get(), port));

- }

- }

-

- return results;

-}

-

-void ConnectivityManager::natTraversalGetPublicIPResult(boost::optional<HostAddress> address) {

- if (address) {

- publicAddress = address;

- SWIFT_LOG(debug) << "Public IP discovered as " << publicAddress.get().toString() << "." << std::endl;

- } else {

- SWIFT_LOG(debug) << "No public IP discoverable." << std::endl;

- }

-}

-

-void ConnectivityManager::natTraversalForwardPortResult(boost::optional<NATPortMapping> mapping) {

- if (mapping) {

- SWIFT_LOG(debug) << "Mapping port was successful." << std::endl;

- } else {

- SWIFT_LOG(debug) << "Mapping port has failed." << std::endl;

- }

-}

-

-

-}

-/*

- * Copyright (c) 2011 Tobias Markmann

- * Licensed under the simplified BSD license.

- * See Documentation/Licenses/BSD-simplified.txt for more information.

- */

-

-#pragma once

-

-#include <vector>

-#include <set>

-

-#include <boost/optional.hpp>

-

-#include <Swiften/Network/HostAddressPort.h>

-#include <Swiften/Network/NATTraverser.h>

-#include <Swiften/Network/NATTraversalForwardPortRequest.h>

-#include <Swiften/Network/NATPortMapping.h>

-

-namespace Swift {

-

-class NATTraverser;

-

-class ConnectivityManager {

-public:

- ConnectivityManager(NATTraverser*);

- ~ConnectivityManager();

-public:

- void addListeningPort(int port);

- void removeListeningPort(int port);

-

- std::vector<HostAddressPort> getHostAddressPorts() const;

- std::vector<HostAddressPort> getAssistedHostAddressPorts() const;

-

-private:

- void natTraversalGetPublicIPResult(boost::optional<HostAddress> address);

- void natTraversalForwardPortResult(boost::optional<NATPortMapping> mapping);

-

-private:

- NATTraverser* natTraversalWorker;

-

- std::set<int> ports;

- boost::optional<HostAddress> publicAddress;

-};

-

-}

-/*

- * Copyright (c) 2011 Tobias Markmann

- * Licensed under the simplified BSD license.

- * See Documentation/Licenses/BSD-simplified.txt for more information.

- */

-

-#include "DefaultLocalJingleTransportCandidateGenerator.h"

-

-#include <vector>

-

-#include <boost/shared_ptr.hpp>

-#include <boost/smart_ptr/make_shared.hpp>

-

-#include <Swiften/Base/foreach.h>

-#include <Swiften/Base/Log.h>

-#include <Swiften/Elements/JingleIBBTransportPayload.h>

-#include <Swiften/Elements/JingleS5BTransportPayload.h>

-#include <Swiften/FileTransfer/ConnectivityManager.h>

-#include <Swiften/FileTransfer/SOCKS5BytestreamRegistry.h>

-#include <Swiften/FileTransfer/SOCKS5BytestreamProxy.h>

-

-namespace Swift {

-

-DefaultLocalJingleTransportCandidateGenerator::DefaultLocalJingleTransportCandidateGenerator(ConnectivityManager* connectivityManager, SOCKS5BytestreamRegistry* s5bRegistry, SOCKS5BytestreamProxy* s5bProxy, JID& ownJID) : connectivityManager(connectivityManager), s5bRegistry(s5bRegistry), s5bProxy(s5bProxy), ownJID(ownJID) {

-}

-

-DefaultLocalJingleTransportCandidateGenerator::~DefaultLocalJingleTransportCandidateGenerator() {

-}

-

-void DefaultLocalJingleTransportCandidateGenerator::start(JingleTransportPayload::ref transportPayload) {

- if (boost::dynamic_pointer_cast<JingleIBBTransportPayload>(transportPayload)) {

- JingleTransportPayload::ref payL = boost::make_shared<JingleTransportPayload>();

- payL->setSessionID(transportPayload->getSessionID());

- onLocalTransportCandidatesGenerated(payL);

- }

- if (boost::dynamic_pointer_cast<JingleS5BTransportPayload>(transportPayload)) {

- JingleS5BTransportPayload::ref payL = boost::make_shared<JingleS5BTransportPayload>();

- payL->setSessionID(transportPayload->getSessionID());

- payL->setMode(JingleS5BTransportPayload::TCPMode);

-

- const unsigned long localPreference = 0;

-

- // get direct candidates

- std::vector<HostAddressPort> directCandidates = connectivityManager->getHostAddressPorts();

- foreach(HostAddressPort addressPort, directCandidates) {

- JingleS5BTransportPayload::Candidate candidate;

- candidate.type = JingleS5BTransportPayload::Candidate::DirectType;

- candidate.jid = ownJID;

- candidate.hostPort = addressPort;

- candidate.priority = 65536 * 126 + localPreference;

- candidate.cid = idGenerator.generateID();

- payL->addCandidate(candidate);

- }

-

- // get assissted candidates

- std::vector<HostAddressPort> assisstedCandidates = connectivityManager->getAssistedHostAddressPorts();

- foreach(HostAddressPort addressPort, assisstedCandidates) {

- JingleS5BTransportPayload::Candidate candidate;

- candidate.type = JingleS5BTransportPayload::Candidate::AssistedType;

- candidate.jid = ownJID;

- candidate.hostPort = addressPort;

- candidate.priority = 65536 * 120 + localPreference;

- candidate.cid = idGenerator.generateID();

- payL->addCandidate(candidate);

- }

-

- // get proxy candidates

- std::vector<S5BProxyRequest::ref> proxyCandidates = s5bProxy->getS5BProxies();

- foreach(S5BProxyRequest::ref proxy, proxyCandidates) {

- if (proxy->getStreamHost()) { // FIXME: Added this test, because there were cases where this wasn't initialized. Investigate this. (Remko)

- JingleS5BTransportPayload::Candidate candidate;

- candidate.type = JingleS5BTransportPayload::Candidate::ProxyType;

- candidate.jid = (*proxy->getStreamHost()).jid;

- candidate.hostPort = (*proxy->getStreamHost()).addressPort;

- candidate.priority = 65536 * 10 + localPreference;

- candidate.cid = idGenerator.generateID();

- payL->addCandidate(candidate);

- }

- }

-

- onLocalTransportCandidatesGenerated(payL);

- }

-

-}

-

-void DefaultLocalJingleTransportCandidateGenerator::stop() {

-}

-

-bool DefaultLocalJingleTransportCandidateGenerator::isActualCandidate(JingleTransportPayload::ref transportPayload) {

- if (!transportPayload.get()) return false;

- return false;

-}

-

-int DefaultLocalJingleTransportCandidateGenerator::getPriority(JingleTransportPayload::ref /* transportPayload */) {

- return 0;

-}

-

-JingleTransport::ref DefaultLocalJingleTransportCandidateGenerator::selectTransport(JingleTransportPayload::ref /* transportPayload */) {

- return JingleTransport::ref();

-}

-

-}

-/*

- * Copyright (c) 2011 Tobias Markmann

- * Licensed under the simplified BSD license.

- * See Documentation/Licenses/BSD-simplified.txt for more information.

- */

-

-#pragma once

-

-#include <Swiften/FileTransfer/LocalJingleTransportCandidateGenerator.h>

-

-#include <Swiften/Base/IDGenerator.h>

-#include <Swiften/JID/JID.h>

-

-namespace Swift {

-

-class SOCKS5BytestreamRegistry;

-class SOCKS5BytestreamProxy;

-class ConnectivityManager;

-

-class DefaultLocalJingleTransportCandidateGenerator : public LocalJingleTransportCandidateGenerator {

-public:

- DefaultLocalJingleTransportCandidateGenerator(ConnectivityManager* connectivityManager, SOCKS5BytestreamRegistry* s5bRegistry, SOCKS5BytestreamProxy* s5bProxy, JID& ownJID);

- virtual ~DefaultLocalJingleTransportCandidateGenerator();

-

- virtual void start(JingleTransportPayload::ref);

- virtual void stop();

-

- virtual bool isActualCandidate(JingleTransportPayload::ref);

- virtual int getPriority(JingleTransportPayload::ref);

- virtual JingleTransport::ref selectTransport(JingleTransportPayload::ref);

-

-private:

- IDGenerator idGenerator;

- ConnectivityManager* connectivityManager;

- SOCKS5BytestreamRegistry* s5bRegistry;

- SOCKS5BytestreamProxy* s5bProxy;

- JID ownJID;

-};

-

-}

-/*

- * Copyright (c) 2011 Tobias Markmann

- * Licensed under the simplified BSD license.

- * See Documentation/Licenses/BSD-simplified.txt for more information.

- */

-

-#include "DefaultLocalJingleTransportCandidateGeneratorFactory.h"

-

-#include <Swiften/FileTransfer/DefaultLocalJingleTransportCandidateGenerator.h>

-#include <Swiften/Base/Log.h>

-

-namespace Swift {

-

-DefaultLocalJingleTransportCandidateGeneratorFactory::DefaultLocalJingleTransportCandidateGeneratorFactory(ConnectivityManager* connectivityManager, SOCKS5BytestreamRegistry* s5bRegistry, SOCKS5BytestreamProxy* s5bProxy, const JID& ownJID) : connectivityManager(connectivityManager), s5bRegistry(s5bRegistry), s5bProxy(s5bProxy), ownJID(ownJID) {

-}

-

-DefaultLocalJingleTransportCandidateGeneratorFactory::~DefaultLocalJingleTransportCandidateGeneratorFactory() {

-}

-

-LocalJingleTransportCandidateGenerator* DefaultLocalJingleTransportCandidateGeneratorFactory::createCandidateGenerator() {

- return new DefaultLocalJingleTransportCandidateGenerator(connectivityManager, s5bRegistry, s5bProxy, ownJID);

-}

-

-

-}

-/*

- * Copyright (c) 2011 Tobias Markmann

- * Licensed under the simplified BSD license.

- * See Documentation/Licenses/BSD-simplified.txt for more information.

- */

-

-#pragma once

-

-#include <Swiften/FileTransfer/LocalJingleTransportCandidateGeneratorFactory.h>

-

-#include <Swiften/JID/JID.h>

-

-namespace Swift {

-

-class ConnectivityManager;

-class SOCKS5BytestreamRegistry;

-class SOCKS5BytestreamProxy;

-

-class DefaultLocalJingleTransportCandidateGeneratorFactory : public LocalJingleTransportCandidateGeneratorFactory{

-public:

- DefaultLocalJingleTransportCandidateGeneratorFactory(ConnectivityManager* connectivityManager, SOCKS5BytestreamRegistry* s5bRegistry, SOCKS5BytestreamProxy* s5bProxy, const JID& ownJID);

- virtual ~DefaultLocalJingleTransportCandidateGeneratorFactory();

-

- LocalJingleTransportCandidateGenerator* createCandidateGenerator();

-

-private:

- ConnectivityManager* connectivityManager;

- SOCKS5BytestreamRegistry* s5bRegistry;

- SOCKS5BytestreamProxy* s5bProxy;

- JID ownJID;

-};

-

-}

-/*

- * Copyright (c) 2011 Tobias Markmann

- * Licensed under the simplified BSD license.

- * See Documentation/Licenses/BSD-simplified.txt for more information.

- */

-

-/*

- * Copyright (c) 2013 Remko Tronçon

- * Licensed under the GNU General Public License.

- * See the COPYING file for more information.

- */

-

-#include <Swiften/FileTransfer/DefaultRemoteJingleTransportCandidateSelector.h>

-

-#include <boost/smart_ptr/make_shared.hpp>

-#include <boost/bind.hpp>

-

-#include <Swiften/Base/Log.h>

-#include <Swiften/Base/boost_bsignals.h>

-#include <Swiften/Base/foreach.h>

-#include <Swiften/Elements/JingleS5BTransportPayload.h>

-#include <Swiften/Network/ConnectionFactory.h>

-#include <Swiften/FileTransfer/SOCKS5BytestreamRegistry.h>

-

-namespace Swift {

-

-DefaultRemoteJingleTransportCandidateSelector::DefaultRemoteJingleTransportCandidateSelector(ConnectionFactory* connectionFactory, TimerFactory* timerFactory, CryptoProvider* crypto) : connectionFactory(connectionFactory), timerFactory(timerFactory), crypto(crypto) {

-}

-

-DefaultRemoteJingleTransportCandidateSelector::~DefaultRemoteJingleTransportCandidateSelector() {

-}

-

-void DefaultRemoteJingleTransportCandidateSelector::addRemoteTransportCandidates(JingleTransportPayload::ref transportPayload) {

- JingleS5BTransportPayload::ref s5bPayload;

- transportSID = transportPayload->getSessionID();

- if ((s5bPayload = boost::dynamic_pointer_cast<JingleS5BTransportPayload>(transportPayload))) {

- foreach(JingleS5BTransportPayload::Candidate c, s5bPayload->getCandidates()) {

- candidates.push(c);

- }

- }

-}

-

-void DefaultRemoteJingleTransportCandidateSelector::selectCandidate() {

- tryNextCandidate(true);

-}

-

-void DefaultRemoteJingleTransportCandidateSelector::tryNextCandidate(bool error) {

- if (error) {

- if (s5bSession) {

- SWIFT_LOG(debug) << "failed to connect" << std::endl;

- }

- if (candidates.empty()) {

- // failed to connect to any of the candidates

- // issue an error

- SWIFT_LOG(debug) << "out of candidates )=" << std::endl;

- JingleS5BTransportPayload::ref failed = boost::make_shared<JingleS5BTransportPayload>();

- failed->setCandidateError(true);

- failed->setSessionID(transportSID);

- onRemoteTransportCandidateSelectFinished(failed);

- } else {

- lastCandidate = candidates.top();

- // only try direct or assisted for now

- if (lastCandidate.type == JingleS5BTransportPayload::Candidate::DirectType ||

- lastCandidate.type == JingleS5BTransportPayload::Candidate::AssistedType || lastCandidate.type == JingleS5BTransportPayload::Candidate::ProxyType ) {

- // create connection

- connection = connectionFactory->createConnection();

- s5bSession = boost::make_shared<SOCKS5BytestreamClientSession>(connection, lastCandidate.hostPort, SOCKS5BytestreamRegistry::getHostname(transportSID, requester, target, crypto), timerFactory);

-

- // bind onReady to this method

- s5bSession->onSessionReady.connect(boost::bind(&DefaultRemoteJingleTransportCandidateSelector::tryNextCandidate, this, _1));

-

- std::string candidateType;

- if (lastCandidate.type == JingleS5BTransportPayload::Candidate::DirectType) {

- candidateType = "direct";

- } else if (lastCandidate.type == JingleS5BTransportPayload::Candidate::AssistedType) {

- candidateType = "assisted";

- } else if (lastCandidate.type == JingleS5BTransportPayload::Candidate::ProxyType) {

- candidateType = "proxy";

- }

-

- // initiate connect

- SWIFT_LOG(debug) << "try to connect to candidate of type " << candidateType << " : " << lastCandidate.hostPort.toString() << std::endl;

- s5bSession->start();

-

- // that's it. we're gonna be called back

- candidates.pop();

- } else {

- s5bSession.reset();

- candidates.pop();

- tryNextCandidate(true);

- }

- }

- } else {

- // we have a working connection, hooray

- JingleS5BTransportPayload::ref success = boost::make_shared<JingleS5BTransportPayload>();

- success->setCandidateUsed(lastCandidate.cid);

- success->setSessionID(transportSID);

- onRemoteTransportCandidateSelectFinished(success);

- }

-}

-

-void DefaultRemoteJingleTransportCandidateSelector::setMinimumPriority(int priority) {

- SWIFT_LOG(debug) << "priority: " << priority << std::endl;

-}

-

-void DefaultRemoteJingleTransportCandidateSelector::setRequesterTarget(const JID& requester, const JID& target) {

- this->requester = requester;

- this->target = target;

-}

-

-SOCKS5BytestreamClientSession::ref DefaultRemoteJingleTransportCandidateSelector::getS5BSession() {

- return s5bSession;

-}

-

-bool DefaultRemoteJingleTransportCandidateSelector::isActualCandidate(JingleTransportPayload::ref /* transportPayload */) {

- return false;

-}

-

-int DefaultRemoteJingleTransportCandidateSelector::getPriority(JingleTransportPayload::ref /* transportPayload */) {

- return 0;

-}

-

-JingleTransport::ref DefaultRemoteJingleTransportCandidateSelector::selectTransport(JingleTransportPayload::ref /* transportPayload */) {

- return JingleTransport::ref();

-}

-

-}

-/*

- * Copyright (c) 2011 Tobias Markmann

- * Licensed under the simplified BSD license.

- * See Documentation/Licenses/BSD-simplified.txt for more information.

- */

-

-/*

- * Copyright (c) 2013 Remko Tronçon

- * Licensed under the GNU General Public License.

- * See the COPYING file for more information.

- */

-

-#pragma once

-

-#include <queue>

-#include <vector>

-

-#include <boost/shared_ptr.hpp>

-

-#include <Swiften/Base/Override.h>

-#include <Swiften/JID/JID.h>

-#include <Swiften/Network/Connection.h>

-#include <Swiften/FileTransfer/SOCKS5BytestreamClientSession.h>

-#include <Swiften/FileTransfer/RemoteJingleTransportCandidateSelector.h>

-#include <Swiften/Elements/JingleS5BTransportPayload.h>

-

-

-namespace Swift {

-

-class ConnectionFactory;

-class TimerFactory;

-class CryptoProvider;

-

-class DefaultRemoteJingleTransportCandidateSelector : public RemoteJingleTransportCandidateSelector {

-public:

- DefaultRemoteJingleTransportCandidateSelector(ConnectionFactory*, TimerFactory*, CryptoProvider*);

- virtual ~DefaultRemoteJingleTransportCandidateSelector();

-

- virtual void addRemoteTransportCandidates(JingleTransportPayload::ref) SWIFTEN_OVERRIDE;

- virtual void selectCandidate() SWIFTEN_OVERRIDE;

- virtual void setMinimumPriority(int) SWIFTEN_OVERRIDE;

- virtual void setRequesterTarget(const JID& requester, const JID& target) SWIFTEN_OVERRIDE;

- virtual SOCKS5BytestreamClientSession::ref getS5BSession() SWIFTEN_OVERRIDE;

-

- virtual bool isActualCandidate(JingleTransportPayload::ref) SWIFTEN_OVERRIDE;

- virtual int getPriority(JingleTransportPayload::ref) SWIFTEN_OVERRIDE;

- virtual JingleTransport::ref selectTransport(JingleTransportPayload::ref) SWIFTEN_OVERRIDE;

-

-private:

- void tryNextCandidate(bool error);

-

-private:

- ConnectionFactory* connectionFactory;

- TimerFactory* timerFactory;

- CryptoProvider* crypto;

-

- std::priority_queue<JingleS5BTransportPayload::Candidate, std::vector<JingleS5BTransportPayload::Candidate>, JingleS5BTransportPayload::CompareCandidate> candidates;

-

- std::string transportSID;

- boost::shared_ptr<Connection> connection;

- boost::shared_ptr<SOCKS5BytestreamClientSession> s5bSession;

- JingleS5BTransportPayload::Candidate lastCandidate;

- JID requester;

- JID target;

-};

-

-}

-/*

- * Copyright (c) 2011 Tobias Markmann

- * Licensed under the simplified BSD license.

- * See Documentation/Licenses/BSD-simplified.txt for more information.

- */

-

-/*

- * Copyright (c) 2013 Remko Tronçon

- * Licensed under the GNU General Public License.

- * See the COPYING file for more information.

- */

-

-#include "DefaultRemoteJingleTransportCandidateSelectorFactory.h"

-

-#include <Swiften/FileTransfer/DefaultRemoteJingleTransportCandidateSelector.h>

-

-#include <Swiften/Base/Log.h>

-

-namespace Swift {

-

-DefaultRemoteJingleTransportCandidateSelectorFactory::DefaultRemoteJingleTransportCandidateSelectorFactory(ConnectionFactory* connectionFactory, TimerFactory* timerFactory, CryptoProvider* crypto) : connectionFactory(connectionFactory), timerFactory(timerFactory), crypto(crypto) {

-}

-

-DefaultRemoteJingleTransportCandidateSelectorFactory::~DefaultRemoteJingleTransportCandidateSelectorFactory() {

-}

-

-RemoteJingleTransportCandidateSelector* DefaultRemoteJingleTransportCandidateSelectorFactory::createCandidateSelector() {

- return new DefaultRemoteJingleTransportCandidateSelector(connectionFactory, timerFactory, crypto);

-}

-

-}

-/*

- * Copyright (c) 2011 Tobias Markmann

- * Licensed under the simplified BSD license.

- * See Documentation/Licenses/BSD-simplified.txt for more information.

- */

-

-/*

- * Copyright (c) 2013 Remko Tronçon

- * Licensed under the GNU General Public License.

- * See the COPYING file for more information.

- */

-

-#pragma once

-

-#include <Swiften/FileTransfer/RemoteJingleTransportCandidateSelectorFactory.h>

-

-namespace Swift {

-

-class ConnectionFactory;

-class TimerFactory;

-class CryptoProvider;

-

-class DefaultRemoteJingleTransportCandidateSelectorFactory : public RemoteJingleTransportCandidateSelectorFactory {

-public:

- DefaultRemoteJingleTransportCandidateSelectorFactory(ConnectionFactory*, TimerFactory*, CryptoProvider*);

- virtual ~DefaultRemoteJingleTransportCandidateSelectorFactory();

-

- RemoteJingleTransportCandidateSelector* createCandidateSelector();

-

-private:

- ConnectionFactory* connectionFactory;

- TimerFactory* timerFactory;

- CryptoProvider* crypto;

-};

-

-}

-class FileTransfer {

-public:

- struct State {

- enum FTState {

- Canceled,

- Failed,

- Finished,

- Negotiating,

- Transferring,

- WaitingForStart,

- WaitingForAccept

- };

-

- FTState state;

- std::string message;

-

- State(FTState state) : state(state), message("") {}

- State(FTState state, std::string message) : state(state), message(message) {}

- };

-public:

- typedef boost::shared_ptr<FileTransfer> ref;

-public:

- boost::uintmax_t fileSizeInBytes;

- std::string filename;

- std::string algo;

- std::string hash;

-public:

- virtual void cancel() = 0;

-public:

- boost::signal<void (size_t /* proccessedBytes */)> onProcessedBytes;

- boost::signal<void (State)> onStateChange;

- boost::signal<void (boost::optional<FileTransferError>)> onFinished;

-public:

- virtual ~FileTransfer() {}

-};

- class S5BProxyRequest;

- virtual void startListeningOnPort(int port) = 0;

-

- virtual OutgoingFileTransfer::ref createOutgoingFileTransfer(const JID& to, const boost::filesystem::path& filepath, const std::string& description, boost::shared_ptr<ReadBytestream> bytestream) = 0;

- virtual OutgoingFileTransfer::ref createOutgoingFileTransfer(const JID& to, const std::string& filename, const std::string& description, const boost::uintmax_t sizeInBytes, const boost::posix_time::ptime& lastModified, boost::shared_ptr<ReadBytestream> bytestream) = 0;

-#include <Swiften/FileTransfer/SOCKS5BytestreamProxyFinder.h>

-#include <Swiften/FileTransfer/ConnectivityManager.h>

-#include <Swiften/FileTransfer/DefaultLocalJingleTransportCandidateGeneratorFactory.h>

-#include <Swiften/FileTransfer/DefaultRemoteJingleTransportCandidateSelectorFactory.h>

-#include <Swiften/FileTransfer/SOCKS5BytestreamServer.h>

-#include <Swiften/FileTransfer/SOCKS5BytestreamProxy.h>

-FileTransferManagerImpl::FileTransferManagerImpl(const JID& ownFullJID, JingleSessionManager* jingleSessionManager, IQRouter* router, EntityCapsProvider* capsProvider, PresenceOracle* presOracle, ConnectionFactory* connectionFactory, ConnectionServerFactory* connectionServerFactory, TimerFactory* timerFactory, NATTraverser* natTraverser, CryptoProvider* crypto) : ownJID(ownFullJID), jingleSM(jingleSessionManager), iqRouter(router), capsProvider(capsProvider), presenceOracle(presOracle), connectionServerFactory(connectionServerFactory), crypto(crypto), bytestreamServer(NULL), s5bProxyFinder(NULL) {

- connectivityManager = new ConnectivityManager(natTraverser);

- bytestreamProxy = new SOCKS5BytestreamProxy(connectionFactory, timerFactory);

-

- localCandidateGeneratorFactory = new DefaultLocalJingleTransportCandidateGeneratorFactory(connectivityManager, bytestreamRegistry, bytestreamProxy, ownFullJID);

- remoteCandidateSelectorFactory = new DefaultRemoteJingleTransportCandidateSelectorFactory(connectionFactory, timerFactory, crypto);

- outgoingFTManager = new OutgoingFileTransferManager(jingleSM, iqRouter, capsProvider, remoteCandidateSelectorFactory, localCandidateGeneratorFactory, bytestreamRegistry, bytestreamProxy, crypto);

- incomingFTManager = new IncomingFileTransferManager(jingleSM, iqRouter, remoteCandidateSelectorFactory, localCandidateGeneratorFactory, bytestreamRegistry, bytestreamProxy, timerFactory, crypto);

- if (s5bProxyFinder) {

- s5bProxyFinder->stop();

- delete s5bProxyFinder;

- }

- if (bytestreamServer) {

- bytestreamServer->stop();

- delete bytestreamServer;

- }

- delete remoteCandidateSelectorFactory;

- delete localCandidateGeneratorFactory;

- delete connectivityManager;

-void FileTransferManagerImpl::startListeningOnPort(int port) {

- // TODO: create a server for each interface we're on

- SWIFT_LOG(debug) << "Start listening on port " << port << " and hope it's not in use." << std::endl;

- boost::shared_ptr<ConnectionServer> server = connectionServerFactory->createConnectionServer(HostAddress("0.0.0.0"), port);

- server->start();

- bytestreamServer = new SOCKS5BytestreamServer(server, bytestreamRegistry, crypto);

- bytestreamServer->start();

- connectivityManager->addListeningPort(port);

-

- s5bProxyFinder = new SOCKS5BytestreamProxyFinder(ownJID.getDomain(), iqRouter);

- s5bProxyFinder->onProxyFound.connect(boost::bind(&FileTransferManagerImpl::addS5BProxy, this, _1));

- s5bProxyFinder->start();

-void FileTransferManagerImpl::addS5BProxy(S5BProxyRequest::ref proxy) {

- bytestreamProxy->addS5BProxy(proxy);

-OutgoingFileTransfer::ref FileTransferManagerImpl::createOutgoingFileTransfer(const JID& to, const boost::filesystem::path& filepath, const std::string& description, boost::shared_ptr<ReadBytestream> bytestream) {

- return createOutgoingFileTransfer(to, filename, description, sizeInBytes, lastModified, bytestream);

-OutgoingFileTransfer::ref FileTransferManagerImpl::createOutgoingFileTransfer(const JID& to, const std::string& filename, const std::string& description, const boost::uintmax_t sizeInBytes, const boost::posix_time::ptime& lastModified, boost::shared_ptr<ReadBytestream> bytestream) {

- return outgoingFTManager->createOutgoingFileTransfer(ownJID, receipient, bytestream, fileInfo);

- class Client;

- class ConnectivityManager;

- class LocalJingleTransportCandidateGeneratorFactory;

- class RemoteJingleTransportCandidateSelectorFactory;

- class SOCKS5BytestreamServer;

- class SOCKS5BytestreamProxy;

- class SOCKS5BytestreamProxyFinder;

- class FileTransferManagerImpl : public FileTransferManager {

- FileTransferManagerImpl(const JID& ownFullJID, JingleSessionManager* jingleSessionManager, IQRouter* router, EntityCapsProvider* capsProvider, PresenceOracle* presOracle, ConnectionFactory* connectionFactory, ConnectionServerFactory* connectionServerFactory, TimerFactory* timerFactory, NATTraverser* natTraverser, CryptoProvider* crypto);

- void startListeningOnPort(int port);

- void addS5BProxy(S5BProxyRequest::ref proxy);

- OutgoingFileTransfer::ref createOutgoingFileTransfer(const JID& to, const boost::filesystem::path& filepath, const std::string& description, boost::shared_ptr<ReadBytestream> bytestream);

- OutgoingFileTransfer::ref createOutgoingFileTransfer(const JID& to, const std::string& filename, const std::string& description, const boost::uintmax_t sizeInBytes, const boost::posix_time::ptime& lastModified, boost::shared_ptr<ReadBytestream> bytestream);

- RemoteJingleTransportCandidateSelectorFactory* remoteCandidateSelectorFactory;

- LocalJingleTransportCandidateGeneratorFactory* localCandidateGeneratorFactory;

- JingleSessionManager* jingleSM;

-

- ConnectionServerFactory* connectionServerFactory;

- CryptoProvider* crypto;

- SOCKS5BytestreamServer* bytestreamServer;

- SOCKS5BytestreamProxy* bytestreamProxy;

- ConnectivityManager* connectivityManager;

- SOCKS5BytestreamProxyFinder* s5bProxyFinder;

-

- session->onDataReceived(ibb->getData());

- size_t receivedSize;

- size_t size,

- size_t size,

- boost::signal<void (const std::vector<unsigned char>&)> onDataReceived;

- size_t size;

-IBBSendSession::IBBSendSession(const std::string& id, const JID& from, const JID& to, boost::shared_ptr<ReadBytestream> bytestream, IQRouter* router) : id(id), from(from), to(to), bytestream(bytestream), router(router), blockSize(4096), sequenceNumber(0), active(false), waitingForData(false) {

- IBBRequest::ref request = IBBRequest::create(from, to, IBB::createIBBOpen(id, boost::numeric_cast<int>(blockSize)), router);

- IBBSendSession(const std::string& id, const JID& from, const JID& to, boost::shared_ptr<ReadBytestream> bytestream, IQRouter* router);

- boost::signal<void (unsigned long long)> onBytesSent;

-#include <Swiften/JID/JID.h>

-#include <Swiften/FileTransfer/WriteBytestream.h>

- virtual void accept(WriteBytestream::ref) = 0;

-IncomingFileTransferManager::IncomingFileTransferManager(JingleSessionManager* jingleSessionManager, IQRouter* router,

- RemoteJingleTransportCandidateSelectorFactory* remoteFactory,

- LocalJingleTransportCandidateGeneratorFactory* localFactory, SOCKS5BytestreamRegistry* bytestreamRegistry, SOCKS5BytestreamProxy* bytestreamProxy, TimerFactory* timerFactory, CryptoProvider* crypto) : jingleSessionManager(jingleSessionManager), router(router), remoteFactory(remoteFactory), localFactory(localFactory), bytestreamRegistry(bytestreamRegistry), bytestreamProxy(bytestreamProxy), timerFactory(timerFactory), crypto(crypto) {

-bool IncomingFileTransferManager::handleIncomingJingleSession(JingleSession::ref session, const std::vector<JingleContentPayload::ref>& contents, const JID& recipient) {

- if (content->getTransport<JingleIBBTransportPayload>() || content->getTransport<JingleS5BTransportPayload>()) {

-

-

- IncomingJingleFileTransfer::ref transfer = boost::shared_ptr<IncomingJingleFileTransfer>(new IncomingJingleFileTransfer(recipient, session, content, remoteFactory, localFactory, router, bytestreamRegistry, bytestreamProxy, timerFactory, crypto));

- } else {

- class RemoteJingleTransportCandidateSelectorFactory;

- class LocalJingleTransportCandidateGeneratorFactory;

- class SOCKS5BytestreamRegistry;

- class SOCKS5BytestreamProxy;

- IncomingFileTransferManager(JingleSessionManager* jingleSessionManager, IQRouter* router, RemoteJingleTransportCandidateSelectorFactory* remoteFactory, LocalJingleTransportCandidateGeneratorFactory* localFactory, SOCKS5BytestreamRegistry* bytestreamRegistry, SOCKS5BytestreamProxy* bytestreamProxy, TimerFactory* timerFactory, CryptoProvider* crypto);

- bool handleIncomingJingleSession(JingleSession::ref session, const std::vector<JingleContentPayload::ref>& contents, const JID& recipient);

- RemoteJingleTransportCandidateSelectorFactory* remoteFactory;

- LocalJingleTransportCandidateGeneratorFactory* localFactory;

- SOCKS5BytestreamRegistry* bytestreamRegistry;

- SOCKS5BytestreamProxy* bytestreamProxy;

- * Copyright (c) 2011 Remko Tronçon

-#include <Swiften/Elements/S5BProxyRequest.h>

-#include <Swiften/FileTransfer/JingleIncomingIBBTransport.h>

-#include <Swiften/FileTransfer/LocalJingleTransportCandidateGenerator.h>

-#include <Swiften/FileTransfer/LocalJingleTransportCandidateGeneratorFactory.h>

-#include <Swiften/FileTransfer/RemoteJingleTransportCandidateSelector.h>

-#include <Swiften/FileTransfer/RemoteJingleTransportCandidateSelectorFactory.h>

-#include <Swiften/FileTransfer/SOCKS5BytestreamRegistry.h>

-#include <Swiften/FileTransfer/SOCKS5BytestreamProxy.h>

-namespace Swift {

- const JID& ourJID,

- RemoteJingleTransportCandidateSelectorFactory* candidateSelectorFactory,

- LocalJingleTransportCandidateGeneratorFactory* candidateGeneratorFactory,

- IQRouter* router,

- SOCKS5BytestreamRegistry* registry,

- SOCKS5BytestreamProxy* proxy,

- ourJID(ourJID),

- session(session),

- router(router),

- s5bRegistry(registry),

- s5bProxy(proxy),

- remoteTransportCandidateSelectFinished(false),

- localTransportCandidateSelectFinished(false),

- serverSession(0) {

-

- candidateSelector = candidateSelectorFactory->createCandidateSelector();

- candidateSelector->onRemoteTransportCandidateSelectFinished.connect(boost::bind(&IncomingJingleFileTransfer::handleRemoteTransportCandidateSelectFinished, this, _1));

-

- candidateGenerator = candidateGeneratorFactory->createCandidateGenerator();

- candidateGenerator->onLocalTransportCandidatesGenerated.connect(boost::bind(&IncomingJingleFileTransfer::handleLocalTransportCandidatesGenerated, this, _1));

-

- session->onTransportInfoReceived.connect(boost::bind(&IncomingJingleFileTransfer::handleTransportInfoReceived, this, _1, _2));

- session->onTransportReplaceReceived.connect(boost::bind(&IncomingJingleFileTransfer::handleTransportReplaceReceived, this, _1, _2));

- session->onSessionTerminateReceived.connect(boost::bind(&IncomingJingleFileTransfer::handleSessionTerminateReceived, this, _1));

- session->onSessionInfoReceived.connect(boost::bind(&IncomingJingleFileTransfer::handleSessionInfoReceived, this, _1));

-

- fileSizeInBytes = fileInfo.getSize();

- filename = fileInfo.getName();

- algo = fileInfo.getAlgo();

- waitOnHashTimer->onTick.connect(boost::bind(&IncomingJingleFileTransfer::finishOffTransfer, this));

- stream->onWrite.disconnect(boost::bind(&IncrementalBytestreamHashCalculator::feedData, hashCalculator, _1));

- delete hashCalculator;

-

- session->onSessionTerminateReceived.disconnect(boost::bind(&IncomingJingleFileTransfer::handleSessionTerminateReceived, this, _1));

- session->onTransportReplaceReceived.disconnect(boost::bind(&IncomingJingleFileTransfer::handleTransportReplaceReceived, this, _1, _2));

- session->onTransportInfoReceived.disconnect(boost::bind(&IncomingJingleFileTransfer::handleTransportInfoReceived, this, _1, _2));

-

- candidateGenerator->onLocalTransportCandidatesGenerated.disconnect(boost::bind(&IncomingJingleFileTransfer::handleLocalTransportCandidatesGenerated, this, _1));

- delete candidateGenerator;

-

- candidateSelector->onRemoteTransportCandidateSelectFinished.disconnect(boost::bind(&IncomingJingleFileTransfer::handleRemoteTransportCandidateSelectFinished, this, _1));

- delete candidateSelector;

-void IncomingJingleFileTransfer::accept(WriteBytestream::ref stream) {

- hashCalculator = new IncrementalBytestreamHashCalculator( algo == "md5" || hash.empty() , algo == "sha-1" || hash.empty(), crypto);

- stream->onWrite.connect(boost::bind(&IncrementalBytestreamHashCalculator::feedData, hashCalculator, _1));

- stream->onWrite.connect(boost::bind(&IncomingJingleFileTransfer::handleWriteStreamDataReceived, this, _1));

- onStateChange(FileTransfer::State(FileTransfer::State::Negotiating));

- if (JingleIBBTransportPayload::ref ibbTransport = initialContent->getTransport<JingleIBBTransportPayload>()) {

- SWIFT_LOG(debug) << "Got IBB transport payload!" << std::endl;

- setActiveTransport(createIBBTransport(ibbTransport));

- session->sendAccept(getContentID(), initialContent->getDescriptions()[0], ibbTransport);

- }

- else if (JingleS5BTransportPayload::ref s5bTransport = initialContent->getTransport<JingleS5BTransportPayload>()) {

- state = CreatingInitialTransports;

- s5bSessionID = s5bTransport->getSessionID().empty() ? idGenerator.generateID() : s5bTransport->getSessionID();

- s5bDestination = SOCKS5BytestreamRegistry::getHostname(s5bSessionID, ourJID, session->getInitiator(), crypto);

- s5bRegistry->addWriteBytestream(s5bDestination, stream);

- fillCandidateMap(theirCandidates, s5bTransport);

- candidateSelector->addRemoteTransportCandidates(s5bTransport);

- candidateSelector->setRequesterTarget(session->getInitiator(), ourJID);

- s5bTransport->setSessionID(s5bSessionID);

- candidateGenerator->start(s5bTransport);

-const JID& IncomingJingleFileTransfer::getSender() const {

- return session->getInitiator();

-}

-

-const JID& IncomingJingleFileTransfer::getRecipient() const {

- return ourJID;

-}

-

- session->sendTerminate(JinglePayload::Reason::Cancel);

-

- if (activeTransport) activeTransport->stop();

- if (serverSession) serverSession->stop();

- if (clientSession) clientSession->stop();

- onStateChange(FileTransfer::State(FileTransfer::State::Canceled));

-}

-

-void IncomingJingleFileTransfer::handleLocalTransportCandidatesGenerated(JingleTransportPayload::ref candidates) {

- if (state == CreatingInitialTransports) {

- if (JingleS5BTransportPayload::ref s5bCandidates = boost::dynamic_pointer_cast<JingleS5BTransportPayload>(candidates)) {

- //localTransportCandidateSelectFinished = true;

- //JingleS5BTransportPayload::ref emptyCandidates = boost::make_shared<JingleS5BTransportPayload>();

- //emptyCandidates->setSessionID(s5bCandidates->getSessionID());

- fillCandidateMap(ourCandidates, s5bCandidates);

- session->sendAccept(getContentID(), initialContent->getDescriptions()[0], s5bCandidates);

-

- state = NegotiatingTransport;

- candidateSelector->selectCandidate();

- }

- }

- else {

- SWIFT_LOG(debug) << "Unhandled state!" << std::endl;

- }

-}

-

-

-void IncomingJingleFileTransfer::handleRemoteTransportCandidateSelectFinished(JingleTransportPayload::ref transport) {

- if (state == Terminated) {

- return;

- }

- if (JingleS5BTransportPayload::ref s5bPayload = boost::dynamic_pointer_cast<JingleS5BTransportPayload>(transport)) {

- //remoteTransportCandidateSelectFinished = true;

- //selectedRemoteTransportCandidate = transport;

- ourCandidate = s5bPayload;

- //checkCandidateSelected();

- decideOnUsedTransport();

- session->sendTransportInfo(getContentID(), s5bPayload);

- }

- else {

- SWIFT_LOG(debug) << "Expected something different here." << std::endl;

- }

-// TODO: Why was assert(false) there? Is this method no longer used perhaps? Delete it if not

-void IncomingJingleFileTransfer::checkCandidateSelected() {

- //assert(false);

- if (localTransportCandidateSelectFinished && remoteTransportCandidateSelectFinished) {

- if (candidateGenerator->isActualCandidate(selectedLocalTransportCandidate) && candidateSelector->isActualCandidate(selectedRemoteTransportCandidate)) {

- if (candidateGenerator->getPriority(selectedLocalTransportCandidate) > candidateSelector->getPriority(selectedRemoteTransportCandidate)) {

- setActiveTransport(candidateGenerator->selectTransport(selectedLocalTransportCandidate));

- }

- else {

- setActiveTransport(candidateSelector->selectTransport(selectedRemoteTransportCandidate));

- }

- }

- else if (candidateSelector->isActualCandidate(selectedRemoteTransportCandidate)) {

- setActiveTransport(candidateSelector->selectTransport(selectedRemoteTransportCandidate));

- }

- else if (candidateGenerator->isActualCandidate(selectedLocalTransportCandidate)) {

- setActiveTransport(candidateGenerator->selectTransport(selectedLocalTransportCandidate));

- }

- else {

- state = WaitingForFallbackOrTerminate;

- }

- }

-}

-void IncomingJingleFileTransfer::setActiveTransport(JingleTransport::ref transport) {

- state = Transferring;

- onStateChange(FileTransfer::State(FileTransfer::State::Transferring));

- activeTransport = transport;

- activeTransport->onDataReceived.connect(boost::bind(&IncomingJingleFileTransfer::handleTransportDataReceived, this, _1));

- activeTransport->onFinished.connect(boost::bind(&IncomingJingleFileTransfer::handleTransferFinished, this, _1));

- activeTransport->start();

-}

-bool IncomingJingleFileTransfer::verifyReceviedData() {

- if (algo.empty() || hash.empty()) {

- SWIFT_LOG(debug) << "no verification possible, skipping" << std::endl;

- return true;

- } else {

- if (algo == "sha-1") {

- SWIFT_LOG(debug) << "verify data via SHA-1 hash: " << (hash == hashCalculator->getSHA1String()) << std::endl;

- return hash == hashCalculator->getSHA1String();

- }

- else if (algo == "md5") {

- SWIFT_LOG(debug) << "verify data via MD5 hash: " << (hash == hashCalculator->getMD5String()) << std::endl;

- return hash == hashCalculator->getMD5String();

- }

- else {

- SWIFT_LOG(debug) << "no verification possible, skipping" << std::endl;

- return true;

- }

-}

-void IncomingJingleFileTransfer::finishOffTransfer() {

- if (verifyReceviedData()) {

- onStateChange(FileTransfer::State(FileTransfer::State::Finished));

- session->sendTerminate(JinglePayload::Reason::Success);

- } else {

- onStateChange(FileTransfer::State(FileTransfer::State::Failed, "Verification failed."));

- session->sendTerminate(JinglePayload::Reason::MediaError);

- }

- state = Terminated;

- waitOnHashTimer->stop();

- if (state == Terminated) {

- return;

- }

- SWIFT_LOG(debug) << "Recevied hash information." << std::endl;

- algo = "sha-1";

- algo = "md5";

- checkIfAllDataReceived();

- SWIFT_LOG(debug) << "session terminate received" << std::endl;

- if (activeTransport) activeTransport->stop();

- if (reason && reason.get().type == JinglePayload::Reason::Cancel) {

- onStateChange(FileTransfer::State(FileTransfer::State::Canceled, "Other user canceled the transfer."));

- else if (reason && reason.get().type == JinglePayload::Reason::Success) {

- /*if (verifyReceviedData()) {

- onStateChange(FileTransfer::State(FileTransfer::State::Finished));

- } else {

- onStateChange(FileTransfer::State(FileTransfer::State::Failed, "Verification failed."));

- }*/

- state = Terminated;

- if (receivedBytes == fileSizeInBytes) {

- SWIFT_LOG(debug) << "No hash information yet. Waiting 5 seconds on hash info." << std::endl;

- } else {

- SWIFT_LOG(debug) << "We already have hash info using " << algo << " algorithm. Finishing off transfer." << std::endl;

- finishOffTransfer();

- else if (receivedBytes > fileSizeInBytes) {

-void IncomingJingleFileTransfer::handleTransportDataReceived(const std::vector<unsigned char>& data) {

- SWIFT_LOG(debug) << data.size() << " bytes received" << std::endl;

- onProcessedBytes(data.size());

- stream->write(data);

-void IncomingJingleFileTransfer::handleWriteStreamDataReceived(const std::vector<unsigned char>& data) {

- receivedBytes += data.size();

- checkIfAllDataReceived();

-}

-

-void IncomingJingleFileTransfer::useOurCandidateChoiceForTransfer(JingleS5BTransportPayload::Candidate candidate) {

- if (candidate.type == JingleS5BTransportPayload::Candidate::ProxyType) {

- // get proxy client session from remoteCandidateSelector

- clientSession = candidateSelector->getS5BSession();

-

- // wait on <activated/> transport-info

- } else {

- // ask s5b client

- clientSession = candidateSelector->getS5BSession();

- if (clientSession) {

- state = Transferring;

- SWIFT_LOG(debug) << clientSession->getAddressPort().toString() << std::endl;

- clientSession->onBytesReceived.connect(boost::bind(boost::ref(onProcessedBytes), _1));

- clientSession->onFinished.connect(boost::bind(&IncomingJingleFileTransfer::handleTransferFinished, this, _1));

- clientSession->startReceiving(stream);

- } else {

- SWIFT_LOG(debug) << "No S5B client session found!!!" << std::endl;

- }

-}

-void IncomingJingleFileTransfer::useTheirCandidateChoiceForTransfer(JingleS5BTransportPayload::Candidate candidate) {

- SWIFT_LOG(debug) << std::endl;

- if (candidate.type == JingleS5BTransportPayload::Candidate::ProxyType) {

- // get proxy client session from s5bRegistry

- clientSession = s5bProxy->createSOCKS5BytestreamClientSession(candidate.hostPort, SOCKS5BytestreamRegistry::getHostname(s5bSessionID, ourJID, session->getInitiator(), crypto));

- clientSession->onSessionReady.connect(boost::bind(&IncomingJingleFileTransfer::proxySessionReady, this, candidate.jid, _1));

- clientSession->start();

-

- // on reply send activate

- } else {

- // ask s5b server

- serverSession = s5bRegistry->getConnectedSession(s5bDestination);

- if (serverSession) {

- state = Transferring;

- serverSession->onBytesReceived.connect(boost::bind(boost::ref(onProcessedBytes), _1));

- serverSession->onFinished.connect(boost::bind(&IncomingJingleFileTransfer::handleTransferFinished, this, _1));

- serverSession->startTransfer();

- } else {

- SWIFT_LOG(debug) << "No S5B server session found!!!" << std::endl;

- }

-void IncomingJingleFileTransfer::fillCandidateMap(CandidateMap& map, JingleS5BTransportPayload::ref s5bPayload) {

- map.clear();

- foreach (JingleS5BTransportPayload::Candidate candidate, s5bPayload->getCandidates()) {

- map[candidate.cid] = candidate;

- }

-

-void IncomingJingleFileTransfer::decideOnUsedTransport() {

- if (ourCandidate && theirCandidate) {

- if (ourCandidate->hasCandidateError() && theirCandidate->hasCandidateError()) {

- state = WaitingForFallbackOrTerminate;

- return;

- }

- std::string our_cid = ourCandidate->getCandidateUsed();

- std::string their_cid = theirCandidate->getCandidateUsed();

- if (ourCandidate->hasCandidateError() && !their_cid.empty()) {

- useTheirCandidateChoiceForTransfer(ourCandidates[their_cid]);

- onStateChange(FileTransfer::State(FileTransfer::State::Transferring));

- }

- else if (theirCandidate->hasCandidateError() && !our_cid.empty()) {

- useOurCandidateChoiceForTransfer(theirCandidates[our_cid]);

- onStateChange(FileTransfer::State(FileTransfer::State::Transferring));

- }

- else if (!our_cid.empty() && !their_cid.empty()) {

- // compare priorites, if same they win

- if (ourCandidates.find(their_cid) == ourCandidates.end() || theirCandidates.find(our_cid) == theirCandidates.end()) {

- SWIFT_LOG(debug) << "Didn't recognize candidate IDs!" << std::endl;

- session->sendTerminate(JinglePayload::Reason::FailedTransport);

- onStateChange(FileTransfer::State(FileTransfer::State::Canceled, "Failed to negotiate candidate."));

- onFinished(FileTransferError(FileTransferError::PeerError));

- return;

- }

-

- JingleS5BTransportPayload::Candidate our_candidate = theirCandidates[our_cid];

- JingleS5BTransportPayload::Candidate their_candidate = ourCandidates[their_cid];

- if (our_candidate.priority > their_candidate.priority) {

- useOurCandidateChoiceForTransfer(our_candidate);

- }

- else if (our_candidate.priority < their_candidate.priority) {

- useTheirCandidateChoiceForTransfer(their_candidate);

- }

- else {

- useTheirCandidateChoiceForTransfer(their_candidate);

- }

- onStateChange(FileTransfer::State(FileTransfer::State::Transferring));

- }

- else {

- assert(false);

- }

- } else {

- SWIFT_LOG(debug) << "Can't make a transport decision yet." << std::endl;

-void IncomingJingleFileTransfer::proxySessionReady(const JID& proxy, bool error) {

- if (error) {

- // indicate proxy error

- } else {

- // activate proxy

- activateProxySession(proxy);

- }

-void IncomingJingleFileTransfer::activateProxySession(const JID &proxy) {

- S5BProxyRequest::ref proxyRequest = boost::make_shared<S5BProxyRequest>();

- proxyRequest->setSID(s5bSessionID);

- proxyRequest->setActivate(session->getInitiator());

- boost::shared_ptr<GenericRequest<S5BProxyRequest> > request = boost::make_shared<GenericRequest<S5BProxyRequest> >(IQ::Set, proxy, proxyRequest, router);

- request->onResponse.connect(boost::bind(&IncomingJingleFileTransfer::handleActivateProxySessionResult, this, _1, _2));

- request->send();

-void IncomingJingleFileTransfer::handleActivateProxySessionResult(boost::shared_ptr<S5BProxyRequest> /*request*/, ErrorPayload::ref error) {

- if (error) {

- SWIFT_LOG(debug) << "ERROR" << std::endl;

- } else {

- // send activated to other jingle party

- JingleS5BTransportPayload::ref proxyActivate = boost::make_shared<JingleS5BTransportPayload>();

- proxyActivate->setActivated(theirCandidate->getCandidateUsed());

- session->sendTransportInfo(getContentID(), proxyActivate);

-

- // start transferring

- clientSession->onBytesReceived.connect(boost::bind(boost::ref(onProcessedBytes), _1));

- clientSession->onFinished.connect(boost::bind(&IncomingJingleFileTransfer::handleTransferFinished, this, _1));

- clientSession->startReceiving(stream);

- onStateChange(FileTransfer::State(FileTransfer::State::Transferring));

-void IncomingJingleFileTransfer::handleTransportInfoReceived(const JingleContentID&, JingleTransportPayload::ref transport) {

- SWIFT_LOG(debug) << "transport info received" << std::endl;

- if (state == Terminated) {

- return;

- if (JingleS5BTransportPayload::ref s5bPayload = boost::dynamic_pointer_cast<JingleS5BTransportPayload>(transport)) {

- if (!s5bPayload->getActivated().empty()) {

- if (ourCandidate->getCandidateUsed() == s5bPayload->getActivated()) {

- clientSession->onBytesReceived.connect(boost::bind(boost::ref(onProcessedBytes), _1));

- clientSession->onFinished.connect(boost::bind(&IncomingJingleFileTransfer::handleTransferFinished, this, _1));

- clientSession->startReceiving(stream);

- onStateChange(FileTransfer::State(FileTransfer::State::Transferring));

- } else {

- SWIFT_LOG(debug) << "ourCandidateChoice doesn't match activated proxy candidate!" << std::endl;

- JingleS5BTransportPayload::ref proxyError = boost::make_shared<JingleS5BTransportPayload>();

- proxyError->setProxyError(true);

- proxyError->setSessionID(s5bSessionID);

- session->sendTransportInfo(getContentID(), proxyError);

- }

- } else {

- theirCandidate = s5bPayload;

- decideOnUsedTransport();

- }

- else {

- SWIFT_LOG(debug) << "Expected something different here." << std::endl;

- /*localTransportCandidateSelectFinished = true;

- selectedLocalTransportCandidate = transport;

- if (candidateGenerator->isActualCandidate(transport)) {

- candidateSelector->setMinimumPriority(candidateGenerator->getPriority(transport));

- }*/

- //checkCandidateSelected();

-void IncomingJingleFileTransfer::handleTransportReplaceReceived(const JingleContentID& content, JingleTransportPayload::ref transport) {

- if (state == Terminated) {

- return;

- if (JingleIBBTransportPayload::ref ibbTransport = boost::dynamic_pointer_cast<JingleIBBTransportPayload>(transport)) {

- SWIFT_LOG(debug) << "transport replaced with IBB" << std::endl;

- setActiveTransport(createIBBTransport(ibbTransport));

- session->sendTransportAccept(content, ibbTransport);

- } else {

- SWIFT_LOG(debug) << "transport replaced failed" << std::endl;

- session->sendTransportReject(content, transport);

-void IncomingJingleFileTransfer::stopActiveTransport() {

- if (activeTransport) {

- activeTransport->stop();

- activeTransport->onDataReceived.disconnect(boost::bind(&IncomingJingleFileTransfer::handleTransportDataReceived, this, _1));

-JingleIncomingIBBTransport::ref IncomingJingleFileTransfer::createIBBTransport(JingleIBBTransportPayload::ref ibbTransport) {

- // TODO: getOffer() -> getOffers correction

- return boost::make_shared<JingleIncomingIBBTransport>(session->getInitiator(), getRecipient(), ibbTransport->getSessionID(), description->getOffers()[0].getSize(), router);

-JingleContentID IncomingJingleFileTransfer::getContentID() const {

- return JingleContentID(initialContent->getName(), initialContent->getCreator());

-void IncomingJingleFileTransfer::handleTransferFinished(boost::optional<FileTransferError> error) {

- if (state == Terminated) {

- return;

- }

- if (error) {

- session->sendTerminate(JinglePayload::Reason::ConnectivityError);

- onStateChange(FileTransfer::State(FileTransfer::State::Failed));

- onFinished(error);

- }

- //

- * Copyright (c) 2010 Remko Tronçon

-#include <Swiften/Base/IDGenerator.h>

-#include <Swiften/Network/Timer.h>

-#include <Swiften/Jingle/JingleSession.h>

-#include <Swiften/FileTransfer/JingleTransport.h>

-#include <Swiften/FileTransfer/JingleIncomingIBBTransport.h>

-#include <Swiften/FileTransfer/SOCKS5BytestreamClientSession.h>

-#include <Swiften/FileTransfer/SOCKS5BytestreamServerSession.h>

-#include <Swiften/Elements/JingleContentPayload.h>

-#include <Swiften/Elements/JingleFileTransferDescription.h>

-#include <Swiften/Elements/JingleIBBTransportPayload.h>

-#include <Swiften/Elements/S5BProxyRequest.h>

-#include <Swiften/Elements/ErrorPayload.h>

- class IQRouter;

- class RemoteJingleTransportCandidateSelectorFactory;

- class LocalJingleTransportCandidateGeneratorFactory;

- class RemoteJingleTransportCandidateSelector;

- class LocalJingleTransportCandidateGenerator;

- class SOCKS5BytestreamRegistry;

- class SOCKS5BytestreamProxy;

- class IncrementalBytestreamHashCalculator;

- class SWIFTEN_API IncomingJingleFileTransfer : public IncomingFileTransfer {

- enum State {

- Initial,

- CreatingInitialTransports,

- NegotiatingTransport,

- Transferring,

- WaitingForFallbackOrTerminate,

- Terminated

- };

- const JID& recipient,

- JingleSession::ref,

- JingleContentPayload::ref content,

- RemoteJingleTransportCandidateSelectorFactory*,

- LocalJingleTransportCandidateGeneratorFactory*,

- IQRouter* router,

- SOCKS5BytestreamRegistry* bytestreamRegistry,

- SOCKS5BytestreamProxy* bytestreamProxy,

- TimerFactory*,

- CryptoProvider*);

- virtual void accept(WriteBytestream::ref);

- virtual const JID& getSender() const;

- virtual const JID& getRecipient() const;

- void handleSessionTerminateReceived(boost::optional<JinglePayload::Reason>);

- void handleSessionInfoReceived(JinglePayload::ref);

- void handleTransportReplaceReceived(const JingleContentID&, JingleTransportPayload::ref);

- void handleTransportInfoReceived(const JingleContentID&, JingleTransportPayload::ref);

- void handleLocalTransportCandidatesGenerated(JingleTransportPayload::ref candidates);

- void handleRemoteTransportCandidateSelectFinished(JingleTransportPayload::ref candidate);

- void setActiveTransport(JingleTransport::ref transport);

- void handleTransportDataReceived(const std::vector<unsigned char>& data);

- JingleIncomingIBBTransport::ref createIBBTransport(JingleIBBTransportPayload::ref ibbTransport);

- JingleContentID getContentID() const;

- bool verifyReceviedData();

- void finishOffTransfer();

- typedef std::map<std::string, JingleS5BTransportPayload::Candidate> CandidateMap;

- private:

- void activateProxySession(const JID &proxy);

- void handleActivateProxySessionResult(boost::shared_ptr<S5BProxyRequest> request, ErrorPayload::ref error);

- void proxySessionReady(const JID& proxy, bool error);

-

- private:

- void useOurCandidateChoiceForTransfer(JingleS5BTransportPayload::Candidate candidate);

- void useTheirCandidateChoiceForTransfer(JingleS5BTransportPayload::Candidate candidate);

- void decideOnUsedTransport();

- void fillCandidateMap(CandidateMap& map, JingleS5BTransportPayload::ref s5bPayload);

- JID ourJID;

- JingleSession::ref session;

- IQRouter* router;

- JingleContentPayload::ref initialContent;

- JingleFileTransferDescription::ref description;

- WriteBytestream::ref stream;

- Timer::ref waitOnHashTimer;

- IDGenerator idGenerator;

-

- RemoteJingleTransportCandidateSelector* candidateSelector;

- LocalJingleTransportCandidateGenerator* candidateGenerator;

- SOCKS5BytestreamRegistry* s5bRegistry;

- SOCKS5BytestreamProxy* s5bProxy;

- bool remoteTransportCandidateSelectFinished;

- JingleTransportPayload::ref selectedRemoteTransportCandidate;

- bool localTransportCandidateSelectFinished;

- JingleTransportPayload::ref selectedLocalTransportCandidate;

-

- JingleS5BTransportPayload::ref ourCandidate;

- JingleS5BTransportPayload::ref theirCandidate;

- CandidateMap ourCandidates;

- CandidateMap theirCandidates;

- SOCKS5BytestreamClientSession::ref clientSession;

- std::string s5bDestination;

- std::string s5bSessionID;

- SOCKS5BytestreamServerSession* serverSession;

- JingleTransport::ref activeTransport;

- if (sha1Hasher) {

- ByteArray result = sha1Hasher->getHash();

- return Hexify::hexify(result);

- } else {

- return std::string();

- if (md5Hasher) {

- ByteArray result = md5Hasher->getHash();

- return Hexify::hexify(result);

- } else {

- return std::string();

-/*

- * Copyright (c) 2011 Remko Tronçon

- * Licensed under the GNU General Public License v3.

- * See Documentation/Licenses/GPLv3.txt for more information.

- */

-

-#include <Swiften/FileTransfer/JingleIncomingIBBTransport.h>

-

-namespace Swift {

-

-JingleIncomingIBBTransport::JingleIncomingIBBTransport(const JID& from, const JID& to, const std::string& id, size_t size, IQRouter* router) : ibbSession(id, from, to, size, router) {

- ibbSession.onDataReceived.connect(boost::ref(onDataReceived));

- ibbSession.onFinished.connect(boost::ref(onFinished));

-}

-

-void JingleIncomingIBBTransport::start() {

- ibbSession.start();

-}

-

-void JingleIncomingIBBTransport::stop() {

- ibbSession.stop();

-}

-

-}

-/*

- * Copyright (c) 2011 Remko Tronçon

- * Licensed under the GNU General Public License v3.

- * See Documentation/Licenses/GPLv3.txt for more information.

- */

-

-#pragma once

-

-#include <boost/shared_ptr.hpp>

-

-#include <Swiften/FileTransfer/JingleTransport.h>

-#include <Swiften/FileTransfer/IBBReceiveSession.h>

-

-namespace Swift {

- class JingleIncomingIBBTransport : public JingleTransport {

- public:

- typedef boost::shared_ptr<JingleIncomingIBBTransport> ref;

-

- JingleIncomingIBBTransport(const JID& from, const JID& to, const std::string& id, size_t size, IQRouter* router);

-

- virtual void start();

- virtual void stop();

-

- private:

- IBBReceiveSession ibbSession;

- };

-}

-/*

- * Copyright (c) 2011 Remko Tronçon

- * Licensed under the GNU General Public License v3.

- * See Documentation/Licenses/GPLv3.txt for more information.

- */

-

-#include <Swiften/FileTransfer/JingleTransport.h>

-

-namespace Swift {

-

-JingleTransport::~JingleTransport() {

-

-}

-

-}

-/*

- * Copyright (c) 2011 Remko Tronçon

- * Licensed under the GNU General Public License v3.

- * See Documentation/Licenses/GPLv3.txt for more information.

- */

-

-#pragma once

-

-#include <boost/shared_ptr.hpp>

-

-#include <Swiften/Base/boost_bsignals.h>

-#include <Swiften/FileTransfer/FileTransfer.h>

-

-namespace Swift {

- class JingleTransport {

- public:

- typedef boost::shared_ptr<JingleTransport> ref;

-

- virtual ~JingleTransport();

-

- virtual void start() = 0;

- virtual void stop() = 0;

-

- boost::signal<void (const std::vector<unsigned char>&)> onDataReceived;

- boost::signal<void (boost::optional<FileTransferError>)> onFinished;

- };

-}

- * Copyright (c) 2011 Remko Tronçon

- * Licensed under the GNU General Public License v3.

- * See Documentation/Licenses/GPLv3.txt for more information.

- * Copyright (c) 2011 Remko Tronçon

- * Licensed under the GNU General Public License v3.

- * See Documentation/Licenses/GPLv3.txt for more information.

-#include <Swiften/Base/API.h>

-#include <Swiften/Elements/JingleTransportPayload.h>

-#include <Swiften/FileTransfer/JingleTransport.h>

- class SWIFTEN_API LocalJingleTransportCandidateGenerator {

- /**

- * Should call onLocalTransportCandidatesGenerated if it has finished discovering local candidates.

- */

- virtual void start(JingleTransportPayload::ref) = 0;

- virtual void stop() = 0;

- virtual bool isActualCandidate(JingleTransportPayload::ref) = 0;

- virtual int getPriority(JingleTransportPayload::ref) = 0;

- virtual JingleTransport::ref selectTransport(JingleTransportPayload::ref) = 0;

- boost::signal<void (JingleTransportPayload::ref)> onLocalTransportCandidatesGenerated;

-/*

- * Copyright (c) 2011 Remko Tronçon

- * Licensed under the GNU General Public License v3.

- * See Documentation/Licenses/GPLv3.txt for more information.

- */

-

-#include <Swiften/FileTransfer/LocalJingleTransportCandidateGeneratorFactory.h>

-

-namespace Swift {

-

-LocalJingleTransportCandidateGeneratorFactory::~LocalJingleTransportCandidateGeneratorFactory() {

-}

-

-}

-/*

- * Copyright (c) 2011 Remko Tronçon

- * Licensed under the GNU General Public License v3.

- * See Documentation/Licenses/GPLv3.txt for more information.

- */

-

-#pragma once

-

-#include <Swiften/Base/API.h>

-

-namespace Swift {

- class LocalJingleTransportCandidateGenerator;

-

- class SWIFTEN_API LocalJingleTransportCandidateGeneratorFactory {

- public:

- virtual ~LocalJingleTransportCandidateGeneratorFactory();

-

- virtual LocalJingleTransportCandidateGenerator* createCandidateGenerator() = 0;

- };

-}

- virtual void stop() = 0;

-#include "OutgoingFileTransferManager.h"

-OutgoingFileTransferManager::OutgoingFileTransferManager(JingleSessionManager* jingleSessionManager, IQRouter* router, EntityCapsProvider* capsProvider, RemoteJingleTransportCandidateSelectorFactory* remoteFactory, LocalJingleTransportCandidateGeneratorFactory* localFactory, SOCKS5BytestreamRegistry* bytestreamRegistry, SOCKS5BytestreamProxy* bytestreamProxy, CryptoProvider* crypto) : jsManager(jingleSessionManager), iqRouter(router), capsProvider(capsProvider), remoteFactory(remoteFactory), localFactory(localFactory), bytestreamRegistry(bytestreamRegistry), bytestreamProxy(bytestreamProxy), crypto(crypto) {

-boost::shared_ptr<OutgoingFileTransfer> OutgoingFileTransferManager::createOutgoingFileTransfer(const JID& from, const JID& receipient, boost::shared_ptr<ReadBytestream> readBytestream, const StreamInitiationFileInfo& fileInfo) {

- // check if receipient support Jingle FT

-

-

- JingleSessionImpl::ref jingleSession = boost::make_shared<JingleSessionImpl>(from, receipient, idGenerator->generateID(), iqRouter);

-

- //jsManager->getSession(receipient, idGenerator->generateID());

- assert(jingleSession);

- jsManager->registerOutgoingSession(from, jingleSession);

- boost::shared_ptr<OutgoingJingleFileTransfer> jingleFT = boost::shared_ptr<OutgoingJingleFileTransfer>(new OutgoingJingleFileTransfer(jingleSession, remoteFactory, localFactory, iqRouter, idGenerator, from, receipient, readBytestream, fileInfo, bytestreamRegistry, bytestreamProxy, crypto));

-

- // otherwise try SI

-

- // else fail

-

- return jingleFT;

-#include <Swiften/JID/JID.h>

-

-

-class JingleSessionManager;

-class IQRouter;

-class EntityCapsProvider;

-class RemoteJingleTransportCandidateSelectorFactory;

-class LocalJingleTransportCandidateGeneratorFactory;

-class OutgoingFileTransfer;

-class JID;

-class IDGenerator;

-class ReadBytestream;

-class StreamInitiationFileInfo;

-class SOCKS5BytestreamRegistry;

-class SOCKS5BytestreamProxy;

-class CryptoProvider;

-

-class OutgoingFileTransferManager {

-public:

- OutgoingFileTransferManager(JingleSessionManager* jingleSessionManager, IQRouter* router, EntityCapsProvider* capsProvider, RemoteJingleTransportCandidateSelectorFactory* remoteFactory, LocalJingleTransportCandidateGeneratorFactory* localFactory, SOCKS5BytestreamRegistry* bytestreamRegistry, SOCKS5BytestreamProxy* bytestreamProxy, CryptoProvider* crypto);

- ~OutgoingFileTransferManager();

-

- boost::shared_ptr<OutgoingFileTransfer> createOutgoingFileTransfer(const JID& from, const JID& to, boost::shared_ptr<ReadBytestream>, const StreamInitiationFileInfo&);

-

-private:

- JingleSessionManager* jsManager;

- IQRouter* iqRouter;

- EntityCapsProvider* capsProvider;

- RemoteJingleTransportCandidateSelectorFactory* remoteFactory;

- LocalJingleTransportCandidateGeneratorFactory* localFactory;

- IDGenerator *idGenerator;

- SOCKS5BytestreamRegistry* bytestreamRegistry;

- SOCKS5BytestreamProxy* bytestreamProxy;

- CryptoProvider* crypto;

-};

-

-#include "OutgoingJingleFileTransfer.h"

-#include <Swiften/Base/boost_bsignals.h>

-#include <Swiften/Queries/GenericRequest.h>

-#include <Swiften/FileTransfer/IBBSendSession.h>

-#include <Swiften/FileTransfer/LocalJingleTransportCandidateGenerator.h>

-#include <Swiften/FileTransfer/LocalJingleTransportCandidateGeneratorFactory.h>

-#include <Swiften/FileTransfer/RemoteJingleTransportCandidateSelector.h>

-#include <Swiften/FileTransfer/RemoteJingleTransportCandidateSelectorFactory.h>

-#include <Swiften/FileTransfer/SOCKS5BytestreamRegistry.h>

-#include <Swiften/FileTransfer/SOCKS5BytestreamProxy.h>

-#include <Swiften/StringCodecs/Hexify.h>

-namespace Swift {

-

-OutgoingJingleFileTransfer::OutgoingJingleFileTransfer(JingleSession::ref session,

- RemoteJingleTransportCandidateSelectorFactory* remoteFactory,

- LocalJingleTransportCandidateGeneratorFactory* localFactory,

- IQRouter* router,

- IDGenerator *idGenerator,

- const JID& fromJID,

- const JID& toJID,

- boost::shared_ptr<ReadBytestream> readStream,

- const StreamInitiationFileInfo &fileInfo,

- SOCKS5BytestreamRegistry* bytestreamRegistry,

- SOCKS5BytestreamProxy* bytestreamProxy,

- CryptoProvider* crypto) :

- session(session), router(router), idGenerator(idGenerator), fromJID(fromJID), toJID(toJID), readStream(readStream), fileInfo(fileInfo), s5bRegistry(bytestreamRegistry), s5bProxy(bytestreamProxy), crypto(crypto), serverSession(NULL), contentID(JingleContentID(idGenerator->generateID(), JingleContentPayload::InitiatorCreator)), canceled(false) {

- session->onSessionAcceptReceived.connect(boost::bind(&OutgoingJingleFileTransfer::handleSessionAcceptReceived, this, _1, _2, _3));

- session->onSessionTerminateReceived.connect(boost::bind(&OutgoingJingleFileTransfer::handleSessionTerminateReceived, this, _1));

- session->onTransportInfoReceived.connect(boost::bind(&OutgoingJingleFileTransfer::handleTransportInfoReceived, this, _1, _2));

- session->onTransportAcceptReceived.connect(boost::bind(&OutgoingJingleFileTransfer::handleTransportAcceptReceived, this, _1, _2));

- fileSizeInBytes = fileInfo.getSize();

- filename = fileInfo.getName();

-

- localCandidateGenerator = localFactory->createCandidateGenerator();

- localCandidateGenerator->onLocalTransportCandidatesGenerated.connect(

- boost::bind(&OutgoingJingleFileTransfer::handleLocalTransportCandidatesGenerated, this, _1));

-

- remoteCandidateSelector = remoteFactory->createCandidateSelector();

- remoteCandidateSelector->onRemoteTransportCandidateSelectFinished.connect(boost::bind(&OutgoingJingleFileTransfer::handleRemoteTransportCandidateSelectFinished, this, _1));

- this->readStream->onRead.connect(boost::bind(&IncrementalBytestreamHashCalculator::feedData, hashCalculator, _1));

- readStream->onRead.disconnect(boost::bind(&IncrementalBytestreamHashCalculator::feedData, hashCalculator, _1));

- onStateChange(FileTransfer::State(FileTransfer::State::WaitingForStart));

-

- s5bSessionID = s5bRegistry->generateSessionID();

- SWIFT_LOG(debug) << "S5B SessionID: " << s5bSessionID << std::endl;

-

- //s5bProxy->connectToProxies(s5bSessionID);

- JingleS5BTransportPayload::ref transport = boost::make_shared<JingleS5BTransportPayload>();

- localCandidateGenerator->start(transport);

-void OutgoingJingleFileTransfer::stop() {

-

-void OutgoingJingleFileTransfer::cancel() {

- canceled = true;

- session->sendTerminate(JinglePayload::Reason::Cancel);

- if (ibbSession) {

- ibbSession->stop();

- SOCKS5BytestreamServerSession *serverSession = s5bRegistry->getConnectedSession(SOCKS5BytestreamRegistry::getHostname(s5bSessionID, session->getInitiator(), toJID, crypto));

- if (serverSession) {

- serverSession->stop();

- }

- if (clientSession) {

- clientSession->stop();

- }

- onStateChange(FileTransfer::State(FileTransfer::State::Canceled));

-void OutgoingJingleFileTransfer::handleSessionAcceptReceived(const JingleContentID& id, JingleDescription::ref /* decription */, JingleTransportPayload::ref transportPayload) {

- if (canceled) {

- return;

- }

- onStateChange(FileTransfer::State(FileTransfer::State::Negotiating));

-

- JingleIBBTransportPayload::ref ibbPayload;

- JingleS5BTransportPayload::ref s5bPayload;

- if ((ibbPayload = boost::dynamic_pointer_cast<JingleIBBTransportPayload>(transportPayload))) {

- ibbSession = boost::make_shared<IBBSendSession>(ibbPayload->getSessionID(), fromJID, toJID, readStream, router);

- ibbSession->setBlockSize(ibbPayload->getBlockSize());

- ibbSession->onBytesSent.connect(boost::bind(boost::ref(onProcessedBytes), _1));

- ibbSession->onFinished.connect(boost::bind(&OutgoingJingleFileTransfer::handleTransferFinished, this, _1));

- ibbSession->start();

- onStateChange(FileTransfer::State(FileTransfer::State::Transferring));

- }

- else if ((s5bPayload = boost::dynamic_pointer_cast<JingleS5BTransportPayload>(transportPayload))) {

- fillCandidateMap(theirCandidates, s5bPayload);

- remoteCandidateSelector->setRequesterTarget(toJID, session->getInitiator());

- remoteCandidateSelector->addRemoteTransportCandidates(s5bPayload);

- remoteCandidateSelector->selectCandidate();

- // TODO: error handling

- SWIFT_LOG(debug) << "Unknown transport payload! Try replaceing with IBB." << std::endl;

- replaceTransportWithIBB(id);

- if (canceled) {

- return;

- }

- if (ibbSession) {

- ibbSession->stop();

- }

- if (clientSession) {

- clientSession->stop();

- if (serverSession) {

- serverSession->stop();

- }

-

- if (reason.is_initialized() && reason.get().type == JinglePayload::Reason::Cancel) {

- onStateChange(FileTransfer::State(FileTransfer::State::Canceled));

- onFinished(FileTransferError(FileTransferError::PeerError));

- } else if (reason.is_initialized() && reason.get().type == JinglePayload::Reason::Success) {

- onStateChange(FileTransfer::State(FileTransfer::State::Finished));

- onFinished(boost::optional<FileTransferError>());

- } else {

- onStateChange(FileTransfer::State(FileTransfer::State::Failed));

- onFinished(FileTransferError(FileTransferError::PeerError));

- canceled = true;

-void OutgoingJingleFileTransfer::handleTransportAcceptReceived(const JingleContentID& /* contentID */, JingleTransportPayload::ref transport) {

- if (canceled) {

- return;

- }

- ibbSession = boost::make_shared<IBBSendSession>(ibbPayload->getSessionID(), fromJID, toJID, readStream, router);

- ibbSession->setBlockSize(ibbPayload->getBlockSize());

- ibbSession->onBytesSent.connect(boost::bind(boost::ref(onProcessedBytes), _1));

- ibbSession->onFinished.connect(boost::bind(&OutgoingJingleFileTransfer::handleTransferFinished, this, _1));

- ibbSession->start();

- onStateChange(FileTransfer::State(FileTransfer::State::Transferring));

- } else {

- // error handling

- SWIFT_LOG(debug) << "Replacing with anything other than IBB isn't supported yet." << std::endl;

- session->sendTerminate(JinglePayload::Reason::FailedTransport);

- onStateChange(FileTransfer::State(FileTransfer::State::Failed));

-void OutgoingJingleFileTransfer::startTransferViaOurCandidateChoice(JingleS5BTransportPayload::Candidate candidate) {

- SWIFT_LOG(debug) << "Transferring data using our candidate." << std::endl;

- if (candidate.type == JingleS5BTransportPayload::Candidate::ProxyType) {

- // get proxy client session from remoteCandidateSelector

- clientSession = remoteCandidateSelector->getS5BSession();

-

- // wait on <activated/> transport-info

- } else {

- clientSession = remoteCandidateSelector->getS5BSession();

- clientSession->onBytesSent.connect(boost::bind(boost::ref(onProcessedBytes), _1));

- clientSession->onFinished.connect(boost::bind(&OutgoingJingleFileTransfer::handleTransferFinished, this, _1));

- clientSession->startSending(readStream);

- onStateChange(FileTransfer::State(FileTransfer::State::Transferring));

- }

- assert(clientSession);

-}

-void OutgoingJingleFileTransfer::startTransferViaTheirCandidateChoice(JingleS5BTransportPayload::Candidate candidate) {

- SWIFT_LOG(debug) << "Transferring data using their candidate." << std::endl;

- if (candidate.type == JingleS5BTransportPayload::Candidate::ProxyType) {

- // connect to proxy

- clientSession = s5bProxy->createSOCKS5BytestreamClientSession(candidate.hostPort, SOCKS5BytestreamRegistry::getHostname(s5bSessionID, session->getInitiator(), toJID, crypto));

- clientSession->onSessionReady.connect(boost::bind(&OutgoingJingleFileTransfer::proxySessionReady, this, candidate.jid, _1));

- clientSession->start();

-

- // on reply send activate

-

- } else {

- serverSession = s5bRegistry->getConnectedSession(SOCKS5BytestreamRegistry::getHostname(s5bSessionID, session->getInitiator(), toJID, crypto));

- serverSession->onBytesSent.connect(boost::bind(boost::ref(onProcessedBytes), _1));

- serverSession->onFinished.connect(boost::bind(&OutgoingJingleFileTransfer::handleTransferFinished, this, _1));

- serverSession->startTransfer();

- }

- onStateChange(FileTransfer::State(FileTransfer::State::Transferring));

-// decide on candidates according to http://xmpp.org/extensions/xep-0260.html#complete

-void OutgoingJingleFileTransfer::decideOnCandidates() {

- if (ourCandidateChoice && theirCandidateChoice) {

- std::string our_cid = ourCandidateChoice->getCandidateUsed();

- std::string their_cid = theirCandidateChoice->getCandidateUsed();

- if (ourCandidateChoice->hasCandidateError() && theirCandidateChoice->hasCandidateError()) {

- replaceTransportWithIBB(contentID);

- }

- else if (!our_cid.empty() && theirCandidateChoice->hasCandidateError()) {

- // use our candidate

- startTransferViaOurCandidateChoice(theirCandidates[our_cid]);

- }

- else if (!their_cid.empty() && ourCandidateChoice->hasCandidateError()) {

- // use their candidate

- startTransferViaTheirCandidateChoice(ourCandidates[their_cid]);

- }

- else if (!our_cid.empty() && !their_cid.empty()) {

- // compare priorites, if same we win

- if (ourCandidates.find(their_cid) == ourCandidates.end() || theirCandidates.find(our_cid) == theirCandidates.end()) {

- SWIFT_LOG(debug) << "Didn't recognize candidate IDs!" << std::endl;

- session->sendTerminate(JinglePayload::Reason::FailedTransport);

- onStateChange(FileTransfer::State(FileTransfer::State::Failed));

- onFinished(FileTransferError(FileTransferError::PeerError));

- return;

- }

-

- JingleS5BTransportPayload::Candidate ourCandidate = theirCandidates[our_cid];

- JingleS5BTransportPayload::Candidate theirCandidate = ourCandidates[their_cid];

- if (ourCandidate.priority > theirCandidate.priority) {

- startTransferViaOurCandidateChoice(ourCandidate);

- }

- else if (ourCandidate.priority < theirCandidate.priority) {

- startTransferViaTheirCandidateChoice(theirCandidate);

- }

- else {

- startTransferViaOurCandidateChoice(ourCandidate);

- }

- }

- } else {

- SWIFT_LOG(debug) << "Can't make a decision yet!" << std::endl;

-void OutgoingJingleFileTransfer::fillCandidateMap(CandidateMap& map, JingleS5BTransportPayload::ref s5bPayload) {

- map.clear();

- foreach (JingleS5BTransportPayload::Candidate candidate, s5bPayload->getCandidates()) {

- map[candidate.cid] = candidate;

-void OutgoingJingleFileTransfer::proxySessionReady(const JID& proxy, bool error) {

- // indicate proxy error

- } else {

- // activate proxy

- activateProxySession(proxy);

-void OutgoingJingleFileTransfer::activateProxySession(const JID& proxy) {

- S5BProxyRequest::ref proxyRequest = boost::make_shared<S5BProxyRequest>();

- proxyRequest->setSID(s5bSessionID);

- proxyRequest->setActivate(toJID);

- boost::shared_ptr<GenericRequest<S5BProxyRequest> > request = boost::make_shared<GenericRequest<S5BProxyRequest> >(IQ::Set, proxy, proxyRequest, router);

- request->onResponse.connect(boost::bind(&OutgoingJingleFileTransfer::handleActivateProxySessionResult, this, _1, _2));

- request->send();

-void OutgoingJingleFileTransfer::handleActivateProxySessionResult(boost::shared_ptr<S5BProxyRequest> /*request*/, ErrorPayload::ref error) {

- if (error) {

- SWIFT_LOG(debug) << "ERROR" << std::endl;

- } else {

- // send activated to other jingle party

- JingleS5BTransportPayload::ref proxyActivate = boost::make_shared<JingleS5BTransportPayload>();

- proxyActivate->setActivated(theirCandidateChoice->getCandidateUsed());

- session->sendTransportInfo(contentID, proxyActivate);

-

- // start transferring

- clientSession->onBytesSent.connect(boost::bind(boost::ref(onProcessedBytes), _1));

- clientSession->onFinished.connect(boost::bind(&OutgoingJingleFileTransfer::handleTransferFinished, this, _1));

- clientSession->startSending(readStream);

- onStateChange(FileTransfer::State(FileTransfer::State::Transferring));

- }

-void OutgoingJingleFileTransfer::sendSessionInfoHash() {

- JingleFileTransferHash::ref hashElement = boost::make_shared<JingleFileTransferHash>();

- hashElement->setHash("sha-1", hashCalculator->getSHA1String());

- hashElement->setHash("md5", hashCalculator->getMD5String());

- session->sendInfo(hashElement);

-void OutgoingJingleFileTransfer::handleTransportInfoReceived(const JingleContentID& /* contentID */, JingleTransportPayload::ref transport) {

- if (canceled) {

- return;

- }

- if (JingleS5BTransportPayload::ref s5bPayload = boost::dynamic_pointer_cast<JingleS5BTransportPayload>(transport)) {

- if (s5bPayload->hasCandidateError() || !s5bPayload->getCandidateUsed().empty()) {

- theirCandidateChoice = s5bPayload;

- decideOnCandidates();

- } else if(!s5bPayload->getActivated().empty()) {

- if (ourCandidateChoice->getCandidateUsed() == s5bPayload->getActivated()) {

- clientSession->onBytesSent.connect(boost::bind(boost::ref(onProcessedBytes), _1));

- clientSession->onFinished.connect(boost::bind(&OutgoingJingleFileTransfer::handleTransferFinished, this, _1));

- clientSession->startSending(readStream);

- onStateChange(FileTransfer::State(FileTransfer::State::Transferring));

- } else {

- SWIFT_LOG(debug) << "ourCandidateChoice doesn't match activated proxy candidate!" << std::endl;

- JingleS5BTransportPayload::ref proxyError = boost::make_shared<JingleS5BTransportPayload>();

- proxyError->setProxyError(true);

- proxyError->setSessionID(s5bSessionID);

- session->sendTransportInfo(contentID, proxyError);

- }

- }

-void OutgoingJingleFileTransfer::handleLocalTransportCandidatesGenerated(JingleTransportPayload::ref payload) {

- if (canceled) {

- return;

- JingleFileTransferDescription::ref description = boost::make_shared<JingleFileTransferDescription>();

- description->addOffer(fileInfo);

- JingleTransportPayload::ref transport;

- if (JingleIBBTransportPayload::ref ibbTransport = boost::dynamic_pointer_cast<JingleIBBTransportPayload>(payload)) {

- ibbTransport->setBlockSize(4096);

- ibbTransport->setSessionID(idGenerator->generateID());

- transport = ibbTransport;

- else if (JingleS5BTransportPayload::ref s5bTransport = boost::dynamic_pointer_cast<JingleS5BTransportPayload>(payload)) {

- //fillCandidateMap(ourCandidates, s5bTransport);

- //s5bTransport->setSessionID(s5bSessionID);

- JingleS5BTransportPayload::ref emptyCandidates = boost::make_shared<JingleS5BTransportPayload>();

- emptyCandidates->setSessionID(s5bTransport->getSessionID());

- fillCandidateMap(ourCandidates, emptyCandidates);

- transport = emptyCandidates;

- s5bRegistry->addReadBytestream(SOCKS5BytestreamRegistry::getHostname(s5bSessionID, session->getInitiator(), toJID, crypto), readStream);

- SWIFT_LOG(debug) << "Unknown tranport payload: " << typeid(*payload).name() << std::endl;

- return;

- session->sendInitiate(contentID, description, transport);

- onStateChange(FileTransfer::State(FileTransfer::State::WaitingForAccept));

-void OutgoingJingleFileTransfer::handleRemoteTransportCandidateSelectFinished(JingleTransportPayload::ref payload) {

- if (canceled) {

- return;

- if (JingleS5BTransportPayload::ref s5bPayload = boost::dynamic_pointer_cast<JingleS5BTransportPayload>(payload)) {

- ourCandidateChoice = s5bPayload;

- session->sendTransportInfo(contentID, s5bPayload);

- decideOnCandidates();

-void OutgoingJingleFileTransfer::replaceTransportWithIBB(const JingleContentID& id) {

- SWIFT_LOG(debug) << "Both parties failed. Replace transport with IBB." << std::endl;

- JingleIBBTransportPayload::ref ibbTransport = boost::make_shared<JingleIBBTransportPayload>();

- ibbTransport->setBlockSize(4096);

- ibbTransport->setSessionID(idGenerator->generateID());

- session->sendTransportReplace(id, ibbTransport);

-void OutgoingJingleFileTransfer::handleTransferFinished(boost::optional<FileTransferError> error) {

- if (error) {

- session->sendTerminate(JinglePayload::Reason::ConnectivityError);

- onStateChange(FileTransfer::State(FileTransfer::State::Failed));

- onFinished(error);

- } else {

- sendSessionInfoHash();

- /*

- session->terminate(JinglePayload::Reason::Success);

- onStateChange(FileTransfer::State(FileTransfer::State::Finished));

- */

- }

- //

-#include <Swiften/Elements/JingleContentPayload.h>

-#include <Swiften/Elements/JingleS5BTransportPayload.h>

-#include <Swiften/Elements/S5BProxyRequest.h>

-#include <Swiften/Elements/ErrorPayload.h>

-#include <Swiften/FileTransfer/SOCKS5BytestreamClientSession.h>

-#include <Swiften/FileTransfer/SOCKS5BytestreamServerSession.h>

-#include <Swiften/Jingle/JingleContentID.h>

-#include <Swiften/Jingle/JingleSession.h>

-

-class RemoteJingleTransportCandidateSelectorFactory;

-class RemoteJingleTransportCandidateSelector;

-class LocalJingleTransportCandidateGeneratorFactory;

-class LocalJingleTransportCandidateGenerator;

-class IQRouter;

-class ReadBytestream;

-class IBBSendSession;

-class IDGenerator;

-class IncrementalBytestreamHashCalculator;

-class SOCKS5BytestreamRegistry;

-class SOCKS5BytestreamProxy;

-class CryptoProvider;

-

-class SWIFTEN_API OutgoingJingleFileTransfer : public OutgoingFileTransfer {

-public:

- OutgoingJingleFileTransfer(JingleSession::ref,

- RemoteJingleTransportCandidateSelectorFactory*,

- LocalJingleTransportCandidateGeneratorFactory*,

- IQRouter*,

- IDGenerator*,

- const JID& from,

- const JID& to,

- boost::shared_ptr<ReadBytestream>,

- const StreamInitiationFileInfo&,

- SOCKS5BytestreamRegistry*,

- SOCKS5BytestreamProxy*,

- CryptoProvider*);

- virtual ~OutgoingJingleFileTransfer();

-

- void start();

- void stop();

-

- void cancel();

-

-private:

- void handleSessionAcceptReceived(const JingleContentID&, JingleDescription::ref, JingleTransportPayload::ref);

- void handleSessionTerminateReceived(boost::optional<JinglePayload::Reason> reason);

- void handleTransportAcceptReceived(const JingleContentID&, JingleTransportPayload::ref);

- void handleTransportInfoReceived(const JingleContentID&, JingleTransportPayload::ref);

-

- void handleLocalTransportCandidatesGenerated(JingleTransportPayload::ref);

- void handleRemoteTransportCandidateSelectFinished(JingleTransportPayload::ref);

-

-private:

- void replaceTransportWithIBB(const JingleContentID&);

- void handleTransferFinished(boost::optional<FileTransferError>);

- void activateProxySession(const JID &proxy);

- void handleActivateProxySessionResult(boost::shared_ptr<S5BProxyRequest> request, ErrorPayload::ref error);

- void proxySessionReady(const JID& proxy, bool error);

-

-private:

- typedef std::map<std::string, JingleS5BTransportPayload::Candidate> CandidateMap;

-

-private:

- void startTransferViaOurCandidateChoice(JingleS5BTransportPayload::Candidate);

- void startTransferViaTheirCandidateChoice(JingleS5BTransportPayload::Candidate);

- void decideOnCandidates();

- void fillCandidateMap(CandidateMap& map, JingleS5BTransportPayload::ref s5bPayload);

-

-private:

- void sendSessionInfoHash();

-

-private:

- JingleSession::ref session;

- RemoteJingleTransportCandidateSelector* remoteCandidateSelector;

- LocalJingleTransportCandidateGenerator* localCandidateGenerator;

-

- IQRouter* router;

- IDGenerator* idGenerator;

- JID fromJID;

- JID toJID;

- boost::shared_ptr<ReadBytestream> readStream;

- StreamInitiationFileInfo fileInfo;

- IncrementalBytestreamHashCalculator *hashCalculator;

-

- boost::shared_ptr<IBBSendSession> ibbSession;

-

- JingleS5BTransportPayload::ref ourCandidateChoice;

- JingleS5BTransportPayload::ref theirCandidateChoice;

- CandidateMap ourCandidates;

- CandidateMap theirCandidates;

-

- SOCKS5BytestreamRegistry* s5bRegistry;

- SOCKS5BytestreamProxy* s5bProxy;

- CryptoProvider* crypto;

- SOCKS5BytestreamClientSession::ref clientSession;

- SOCKS5BytestreamServerSession* serverSession;

- JingleContentID contentID;

- std::string s5bSessionID;

-

- bool canceled;

-};

-void OutgoingSIFileTransfer::handleStreamInitiationRequestResponse(StreamInitiation::ref response, ErrorPayload::ref error) {

-void OutgoingSIFileTransfer::handleBytestreamsRequestResponse(Bytestreams::ref, ErrorPayload::ref error) {

-void OutgoingSIFileTransfer::finish(boost::optional<FileTransferError> error) {

-void OutgoingSIFileTransfer::handleIBBSessionFinished(boost::optional<FileTransferError> error) {

- finish(error);

- * Copyright (c) 2011 Remko Tronçon

- * Licensed under the GNU General Public License v3.

- * See Documentation/Licenses/GPLv3.txt for more information.

-namespace Swift {

- * Copyright (c) 2011 Remko Tronçon

- * Licensed under the GNU General Public License v3.

- * See Documentation/Licenses/GPLv3.txt for more information.

-#include <Swiften/Base/boost_bsignals.h>

-#include <Swiften/Base/API.h>

-#include <Swiften/Elements/JingleTransportPayload.h>

-#include <Swiften/FileTransfer/JingleTransport.h>

- class SWIFTEN_API RemoteJingleTransportCandidateSelector {

- virtual void addRemoteTransportCandidates(JingleTransportPayload::ref) = 0;

- virtual void selectCandidate() = 0;

- virtual void setMinimumPriority(int) = 0;

- virtual void setRequesterTarget(const JID&, const JID&) {}

- virtual SOCKS5BytestreamClientSession::ref getS5BSession() { return SOCKS5BytestreamClientSession::ref(); }

- virtual bool isActualCandidate(JingleTransportPayload::ref) = 0;

- virtual int getPriority(JingleTransportPayload::ref) = 0;

- virtual JingleTransport::ref selectTransport(JingleTransportPayload::ref) = 0;

- boost::signal<void (JingleTransportPayload::ref)> onRemoteTransportCandidateSelectFinished;

- };

-/*

- * Copyright (c) 2011 Remko Tronçon

- * Licensed under the GNU General Public License v3.

- * See Documentation/Licenses/GPLv3.txt for more information.

- */

-

-#include <Swiften/FileTransfer/RemoteJingleTransportCandidateSelectorFactory.h>

-

-namespace Swift {

-

-RemoteJingleTransportCandidateSelectorFactory::~RemoteJingleTransportCandidateSelectorFactory() {

-}

-

-}

-/*

- * Copyright (c) 2011 Remko Tronçon

- * Licensed under the GNU General Public License v3.

- * See Documentation/Licenses/GPLv3.txt for more information.

- */

-

-#pragma once

-

-#include <Swiften/Base/API.h>

-

-namespace Swift {

- class RemoteJingleTransportCandidateSelector;

-

- class SWIFTEN_API RemoteJingleTransportCandidateSelectorFactory {

- public:

- virtual ~RemoteJingleTransportCandidateSelectorFactory();

-

- virtual RemoteJingleTransportCandidateSelector* createCandidateSelector() = 0;

- };

-}

- "RemoteJingleTransportCandidateSelectorFactory.cpp",

- "LocalJingleTransportCandidateGeneratorFactory.cpp",

- "DefaultRemoteJingleTransportCandidateSelectorFactory.cpp",

- "DefaultLocalJingleTransportCandidateGeneratorFactory.cpp",

- "DefaultRemoteJingleTransportCandidateSelector.cpp",

- "DefaultLocalJingleTransportCandidateGenerator.cpp",

- "JingleTransport.cpp",

- "JingleIncomingIBBTransport.cpp",

- "SOCKS5BytestreamProxy.cpp",

- "ConnectivityManager.cpp",

-SOCKS5BytestreamClientSession::SOCKS5BytestreamClientSession(boost::shared_ptr<Connection> connection, const HostAddressPort& addressPort, const std::string& destination, TimerFactory* timerFactory) :

- connection(connection), addressPort(addressPort), destination(destination), state(Initial), chunkSize(131072) {

- connection->onConnectFinished.connect(boost::bind(&SOCKS5BytestreamClientSession::handleConnectFinished, this, _1));

- connection->onDisconnected.connect(boost::bind(&SOCKS5BytestreamClientSession::handleDisconnected, this, _1));

- weFailedTimeout->onTick.connect(boost::bind(&SOCKS5BytestreamClientSession::handleWeFailedTimeout, this));

- connection->disconnect();

- connection->onDataWritten.disconnect(boost::bind(&SOCKS5BytestreamClientSession::sendData, this));

- connection->onDataRead.disconnect(boost::bind(&SOCKS5BytestreamClientSession::handleDataRead, this, _1));

- connection->onDataWritten.connect(boost::bind(&SOCKS5BytestreamClientSession::sendData, this));

- connection->disconnect();

- connection->onDataWritten.disconnect(boost::bind(&SOCKS5BytestreamClientSession::sendData, this));

- connection->onDataRead.disconnect(boost::bind(&SOCKS5BytestreamClientSession::handleDataRead, this, _1));

- connection->onDataRead.connect(boost::bind(&SOCKS5BytestreamClientSession::handleDataRead, this, _1));

- SOCKS5BytestreamClientSession(boost::shared_ptr<Connection> connection, const HostAddressPort&, const std::string&, TimerFactory*);

-#include "SOCKS5BytestreamProxy.h"

-SOCKS5BytestreamProxy::SOCKS5BytestreamProxy(ConnectionFactory *connFactory, TimerFactory *timeFactory) : connectionFactory(connFactory), timerFactory(timeFactory) {

-void SOCKS5BytestreamProxy::addS5BProxy(S5BProxyRequest::ref proxy) {

-const std::vector<S5BProxyRequest::ref>& SOCKS5BytestreamProxy::getS5BProxies() const {

-void SOCKS5BytestreamProxy::connectToProxies(const std::string& sessionID) {

-boost::shared_ptr<SOCKS5BytestreamClientSession> SOCKS5BytestreamProxy::getProxySessionAndCloseOthers(const JID& proxyJID, const std::string& sessionID) {

-boost::shared_ptr<SOCKS5BytestreamClientSession> SOCKS5BytestreamProxy::createSOCKS5BytestreamClientSession(HostAddressPort addressPort, const std::string& destAddr) {

-/*

- * Copyright (c) 2011 Tobias Markmann

- * Licensed under the simplified BSD license.

- * See Documentation/Licenses/BSD-simplified.txt for more information.

- */

-

-#pragma once

-

-#include <string>

-#include <vector>

-

-#include <Swiften/Base/API.h>

-#include <Swiften/Elements/S5BProxyRequest.h>

-#include <Swiften/FileTransfer/SOCKS5BytestreamClientSession.h>

-#include <Swiften/JID/JID.h>

-#include <Swiften/Network/ConnectionFactory.h>

-#include <Swiften/Network/TimerFactory.h>

-

-namespace Swift {

-

-/**

- * - manages list of working S5B proxies

- * - creates initial connections (for the candidates you provide)

- */

-class SWIFTEN_API SOCKS5BytestreamProxy {

-public:

- SOCKS5BytestreamProxy(ConnectionFactory*, TimerFactory*);

-

- void addS5BProxy(S5BProxyRequest::ref);

- const std::vector<S5BProxyRequest::ref>& getS5BProxies() const;

-

- void connectToProxies(const std::string& sessionID);

- boost::shared_ptr<SOCKS5BytestreamClientSession> getProxySessionAndCloseOthers(const JID& proxyJID, const std::string& sessionID);

-

- boost::shared_ptr<SOCKS5BytestreamClientSession> createSOCKS5BytestreamClientSession(HostAddressPort addressPort, const std::string& destAddr);

-

-private:

- ConnectionFactory* connectionFactory;

- TimerFactory* timerFactory;

-

- typedef std::map<JID, boost::shared_ptr<SOCKS5BytestreamClientSession> > ProxyJIDClientSessionMap;

- std::map<std::string, ProxyJIDClientSessionMap> proxySessions;

-

- std::vector<S5BProxyRequest::ref> localS5BProxies;

-};

-

-}

-public:

- SOCKS5BytestreamProxyFinder(const JID& service, IQRouter *iqRouter);

- ~SOCKS5BytestreamProxyFinder();

-

- void start();

- void stop();

-

- boost::signal<void(boost::shared_ptr<S5BProxyRequest>)> onProxyFound;

-

-private:

- void sendBytestreamQuery(const JID&);

-

- void handleServiceFound(const JID&, boost::shared_ptr<DiscoInfo>);

- void handleProxyResponse(boost::shared_ptr<S5BProxyRequest>, ErrorPayload::ref);

-private:

- JID service;

- IQRouter* iqRouter;

- boost::shared_ptr<DiscoServiceWalker> serviceWalker;

- std::vector<boost::shared_ptr<GenericRequest<S5BProxyRequest> > > requests;

-};

-#include <Swiften/StringCodecs/Hexify.h>

-#include <Swiften/Crypto/CryptoProvider.h>

-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;

- 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>();

-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())));

-}

-

-#include <boost/shared_ptr.hpp>

-

-#include <vector>

-#include <Swiften/JID/JID.h>

-#include <Swiften/Elements/S5BProxyRequest.h>

-#include <Swiften/FileTransfer/ReadBytestream.h>

-#include <Swiften/FileTransfer/WriteBytestream.h>

-#include <Swiften/FileTransfer/SOCKS5BytestreamServerSession.h>

-#include <Swiften/FileTransfer/SOCKS5BytestreamClientSession.h>

-#include <Swiften/Network/HostAddressPort.h>

- class CryptoProvider;

- boost::shared_ptr<ReadBytestream> getReadBytestream(const std::string& destination) const;

- void addReadBytestream(const std::string& destination, boost::shared_ptr<ReadBytestream> byteStream);

- void removeReadBytestream(const std::string& destination);

-

- boost::shared_ptr<WriteBytestream> getWriteBytestream(const std::string& destination) const;

- void addWriteBytestream(const std::string& destination, boost::shared_ptr<WriteBytestream> byteStream);

- void removeWriteBytestream(const std::string& destination);

- /**

- * Start an actual transfer.

- */

- SOCKS5BytestreamServerSession* getConnectedSession(const std::string& destination);

-

- public:

- static std::string getHostname(const std::string& sessionID, const JID& requester, const JID& target, CryptoProvider* crypto);

-

- friend class SOCKS5BytestreamServerSession;

-

- typedef std::map<std::string, boost::shared_ptr<ReadBytestream> > ReadBytestreamMap;

- ReadBytestreamMap readBytestreams;

-

- typedef std::map<std::string, boost::shared_ptr<WriteBytestream> > WriteBytestreamMap;

- WriteBytestreamMap writeBytestreams;

-

- std::map<std::string, SOCKS5BytestreamServerSession*> serverSessions;

-

-SOCKS5BytestreamServer::SOCKS5BytestreamServer(boost::shared_ptr<ConnectionServer> connectionServer, SOCKS5BytestreamRegistry* registry, CryptoProvider* crypto) : connectionServer(connectionServer), registry(registry), crypto(crypto) {

-}

-

-void SOCKS5BytestreamServer::addReadBytestream(const std::string& id, const JID& from, const JID& to, boost::shared_ptr<ReadBytestream> byteStream) {

- registry->addReadBytestream(getSOCKSDestinationAddress(id, from, to), byteStream);

-}

-

-void SOCKS5BytestreamServer::removeReadBytestream(const std::string& id, const JID& from, const JID& to) {

- registry->removeReadBytestream(getSOCKSDestinationAddress(id, from, to));

-}

-

-std::string SOCKS5BytestreamServer::getSOCKSDestinationAddress(const std::string& id, const JID& from, const JID& to) {

- return Hexify::hexify(crypto->getSHA1Hash(createByteArray(id + from.toString() + to.toString())));

- boost::shared_ptr<SOCKS5BytestreamServerSession> session(new SOCKS5BytestreamServerSession(connection, registry));

- SOCKS5BytestreamServer(boost::shared_ptr<ConnectionServer> connectionServer, SOCKS5BytestreamRegistry* registry, CryptoProvider* crypto);

- 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::string getSOCKSDestinationAddress(const std::string& id, const JID& from, const JID& to);

- CryptoProvider* crypto;