1 files changed, 112 insertions, 0 deletions
diff --git a/Swiften/Network/ProxiedConnection.cpp b/Swiften/Network/ProxiedConnection.cpp
new file mode 100644
index 0000000..0061820
--- /dev/null
+++ b/Swiften/Network/ProxiedConnection.cpp
@@ -0,0 +1,112 @@
+/*
+ * Copyright (c) 2012 Remko Tronçon
+ * Licensed under the GNU General Public License v3.
+ * See Documentation/Licenses/GPLv3.txt for more information.
+ */
+
+
+#include <Swiften/Network/ProxiedConnection.h>
+
+#include <iostream>
+#include <boost/bind.hpp>
+
+#include <Swiften/Base/ByteArray.h>
+#include <Swiften/Network/HostAddressPort.h>
+#include <Swiften/Network/ConnectionFactory.h>
+
+using namespace Swift;
+
+ProxiedConnection::ProxiedConnection(
+		DomainNameResolver* resolver, 
+		ConnectionFactory* connectionFactory, 
+		TimerFactory* timerFactory, 
+		const std::string& proxyHost, 
+		int proxyPort) : 
+			resolver_(resolver),
+			connectionFactory_(connectionFactory), 
+			timerFactory_(timerFactory), 
+			proxyHost_(proxyHost), 
+			proxyPort_(proxyPort), 
+			server_(HostAddressPort(HostAddress("0.0.0.0"), 0)) {
+	connected_ = false;
+}
+
+ProxiedConnection::~ProxiedConnection() {
+	cancelConnector();
+	if (connection_) {
+		connection_->onDataRead.disconnect(boost::bind(&ProxiedConnection::handleDataRead, shared_from_this(), _1));
+		connection_->onDisconnected.disconnect(boost::bind(&ProxiedConnection::handleDisconnected, shared_from_this(), _1));
+	}
+	if (connected_) {
+		std::cerr << "Warning: Connection was still established." << std::endl;
+	}
+}
+
+void ProxiedConnection::cancelConnector() {
+	if (connector_) {
+		connector_->onConnectFinished.disconnect(boost::bind(&ProxiedConnection::handleConnectFinished, shared_from_this(), _1));
+		connector_->stop();
+		connector_.reset();
+	}
+}
+
+void ProxiedConnection::connect(const HostAddressPort& server) {
+	server_ = server;
+
+	connector_ = Connector::create(proxyHost_, proxyPort_, boost::optional<std::string>(), resolver_, connectionFactory_, timerFactory_);
+	connector_->onConnectFinished.connect(boost::bind(&ProxiedConnection::handleConnectFinished, shared_from_this(), _1));
+	connector_->start();
+}
+
+void ProxiedConnection::listen() {
+	assert(false);
+	connection_->listen();
+}
+
+void ProxiedConnection::disconnect() {
+	connected_ = false;
+	connection_->disconnect();
+}
+
+void ProxiedConnection::handleDisconnected(const boost::optional<Error>& error) {
+	onDisconnected(error);
+}
+
+void ProxiedConnection::write(const SafeByteArray& data) {
+	connection_->write(data);
+}
+
+void ProxiedConnection::handleConnectFinished(Connection::ref connection) {
+	cancelConnector();
+	if (connection) {
+		connection_ = connection;
+		connection_->onDataRead.connect(boost::bind(&ProxiedConnection::handleDataRead, shared_from_this(), _1));
+		connection_->onDisconnected.connect(boost::bind(&ProxiedConnection::handleDisconnected, shared_from_this(), _1));
+		
+		initializeProxy();
+	}
+	else {
+		onConnectFinished(true);
+	}
+}
+
+void ProxiedConnection::handleDataRead(boost::shared_ptr<SafeByteArray> data) {
+	if (!connected_) {
+		handleProxyInitializeData(data);
+	}
+	else {
+		onDataRead(data);
+	}
+}
+
+HostAddressPort ProxiedConnection::getLocalAddress() const {
+	return connection_->getLocalAddress();
+}
+
+void ProxiedConnection::setProxyInitializeFinished(bool success) {
+	connected_ = success;
+	if (!success) {
+		disconnect();
+	}
+	onConnectFinished(!success);
+}