1 files changed, 67 insertions, 0 deletions
diff --git a/Swiften/Network/ProxiedConnection.h b/Swiften/Network/ProxiedConnection.h
new file mode 100644
index 0000000..aa8df38
--- /dev/null
+++ b/Swiften/Network/ProxiedConnection.h
@@ -0,0 +1,67 @@
+/*
+ * Copyright (c) 2012 Remko Tronçon
+ * Licensed under the GNU General Public License v3.
+ * See Documentation/Licenses/GPLv3.txt for more information.
+ */
+
+
+#pragma once
+
+#include <boost/enable_shared_from_this.hpp>
+
+#include <Swiften/Network/Connection.h>
+#include <Swiften/Network/Connector.h>
+#include <Swiften/Network/HostAddressPort.h>
+#include <Swiften/Base/SafeString.h>
+
+namespace boost {
+	class thread;
+	namespace system {
+		class error_code;
+	}
+}
+
+namespace Swift {
+	class ConnectionFactory;
+
+	class ProxiedConnection : public Connection, public boost::enable_shared_from_this<ProxiedConnection> {
+		public:
+			ProxiedConnection(DomainNameResolver* resolver, ConnectionFactory* connectionFactory, TimerFactory* timerFactory, const std::string& proxyHost, int proxyPort);
+			~ProxiedConnection();
+
+			virtual void listen();
+			virtual void connect(const HostAddressPort& address);
+			virtual void disconnect();
+			virtual void write(const SafeByteArray& data);
+
+			virtual HostAddressPort getLocalAddress() const;
+
+		private:
+			void handleConnectFinished(Connection::ref connection);
+			void handleDataRead(boost::shared_ptr<SafeByteArray> data);
+			void handleDisconnected(const boost::optional<Error>& error);
+			void cancelConnector();
+
+		protected:
+			void setProxyInitializeFinished(bool success);
+
+			virtual void initializeProxy() = 0;
+			virtual void handleProxyInitializeData(boost::shared_ptr<SafeByteArray> data) = 0;
+
+			const HostAddressPort& getServer() const {
+				return server_;
+			}
+
+		private:
+			bool connected_;
+			DomainNameResolver* resolver_;
+			ConnectionFactory* connectionFactory_;	
+			TimerFactory* timerFactory_;
+			std::string proxyHost_;
+			int proxyPort_;
+			HostAddressPort server_;
+			Connector::ref connector_;
+			boost::shared_ptr<Connection> connection_;
+	};
+}
+