1 files changed, 111 insertions, 0 deletions
diff --git a/Swiften/Network/HTTPConnectProxiedConnection.cpp b/Swiften/Network/HTTPConnectProxiedConnection.cpp
new file mode 100644
index 0000000..4e05f29
--- /dev/null
+++ b/Swiften/Network/HTTPConnectProxiedConnection.cpp
@@ -0,0 +1,111 @@
+/*
+ * Copyright (c) 2010-2011 Thilo Cestonaro
+ * Licensed under the simplified BSD license.
+ * See Documentation/Licenses/BSD-simplified.txt for more information.
+ */
+
+#include "Swiften/Network/HTTPConnectProxiedConnection.h"
+
+#include <iostream>
+#include <boost/bind.hpp>
+#include <boost/thread.hpp>
+#include <boost/lexical_cast.hpp>
+#include <math.h>
+
+#include "Swiften/Base/Log.h"
+#include "Swiften/Base/String.h"
+#include "Swiften/Base/ByteArray.h"
+#include "Swiften/Network/HostAddressPort.h"
+
+namespace Swift {
+
+HTTPConnectProxiedConnection::HTTPConnectProxiedConnection(ConnectionFactory* connectionFactory, HostAddressPort proxy)
+: connectionFactory_(connectionFactory), proxy_(proxy), server_(HostAddressPort(HostAddress("0.0.0.0"), 0))
+{
+	connected_ = false;
+}
+
+HTTPConnectProxiedConnection::~HTTPConnectProxiedConnection() {
+	if(connection_) {
+		connection_->onDataRead.disconnect(boost::bind(&HTTPConnectProxiedConnection::handleDataRead, shared_from_this(), _1));
+		connection_->onDisconnected.disconnect(boost::bind(&HTTPConnectProxiedConnection::handleDisconnected, shared_from_this(), _1));
+	}
+
+	if(connected_) {
+		std::cerr << "Warning: Connection was still established." << std::endl;
+	}
+}
+
+void HTTPConnectProxiedConnection::connect(const HostAddressPort& server) {
+	server_ = server;
+	connection_ = connectionFactory_->createConnection();
+	connection_->onConnectFinished.connect(boost::bind(&HTTPConnectProxiedConnection::handleConnectionConnectFinished, shared_from_this(), _1));
+	connection_->onDataRead.connect(boost::bind(&HTTPConnectProxiedConnection::handleDataRead, shared_from_this(), _1));
+	connection_->onDisconnected.connect(boost::bind(&HTTPConnectProxiedConnection::handleDisconnected, shared_from_this(), _1));
+	connection_->connect(proxy_);
+}
+
+void HTTPConnectProxiedConnection::listen() {
+	assert(false);
+	connection_->listen();
+}
+
+void HTTPConnectProxiedConnection::disconnect() {
+	connected_ = false;
+	connection_->disconnect();
+}
+
+void HTTPConnectProxiedConnection::handleDisconnected(const boost::optional<Error>& error) {
+	onDisconnected(error);
+}
+
+void HTTPConnectProxiedConnection::write(const ByteArray& data) {
+	connection_->write(data);
+}
+
+void HTTPConnectProxiedConnection::handleConnectionConnectFinished(bool error) {
+	connection_->onConnectFinished.disconnect(boost::bind(&HTTPConnectProxiedConnection::handleConnectionConnectFinished, shared_from_this(), _1));
+	if(!error) {
+		proxyState_ = ProxyConnecting;
+		std::stringstream connect;
+		connect << "CONNECT " << server_.getAddress().toString() << ":" << server_.getPort() << " HTTP/1.1\r\n\r\n";
+		connection_->write(ByteArray(connect.str()));
+	}
+	else {
+		onConnectFinished(true);
+	}
+}
+
+void HTTPConnectProxiedConnection::handleDataRead(const ByteArray& data) {
+	if (!connected_) {
+		if (proxyState_ == ProxyConnecting) {
+			SWIFT_LOG(debug) << data.toString() << std::endl;
+			std::vector<std::string> tmp = String::split(data.toString(), ' ');
+			if(tmp.size() > 1) {
+				int status = boost::lexical_cast<int> (tmp[1].c_str()); 
+				SWIFT_LOG(debug) << "Proxy Status: " << status << std::endl;
+				if (status / 100 == 2) { // all 2XX states are OK
+					connected_ = true;
+					onConnectFinished(false);
+					return;
+				}
+			}
+
+			std::cerr << "HTTP Proxy returned an error: " << data.toString() << std::endl;
+			return;
+		}
+	}
+	else {
+		onDataRead(data);
+		return;
+	}
+	disconnect();
+	onConnectFinished(true);
+}
+
+HostAddressPort HTTPConnectProxiedConnection::getLocalAddress() const {
+	return connection_->getLocalAddress();
+}
+
+//namespace
+}