1 files changed, 59 insertions, 0 deletions
diff --git a/Swiften/Network/SOCKS5ProxiedConnection.h b/Swiften/Network/SOCKS5ProxiedConnection.h
new file mode 100644
index 0000000..592ce7d
--- /dev/null
+++ b/Swiften/Network/SOCKS5ProxiedConnection.h
@@ -0,0 +1,59 @@
+/*
+ * Copyright (c) 2010-2011 Thilo Cestonaro
+ * Licensed under the simplified BSD license.
+ * See Documentation/Licenses/BSD-simplified.txt for more information.
+ */
+
+#pragma once
+
+#include <boost/enable_shared_from_this.hpp>
+
+#include <Swiften/Network/Connection.h>
+#include <Swiften/Network/HostAddressPort.h>
+
+namespace boost {
+	class thread;
+	namespace system {
+		class error_code;
+	}
+}
+
+namespace Swift {
+	class ConnectionFactory;
+
+	class SOCKS5ProxiedConnection : public Connection, public boost::enable_shared_from_this<SOCKS5ProxiedConnection> {
+		public:
+			typedef boost::shared_ptr<SOCKS5ProxiedConnection> ref;
+
+			~SOCKS5ProxiedConnection();
+
+			static ref create(ConnectionFactory* connectionFactory, const HostAddressPort& proxy) {
+				return ref(new SOCKS5ProxiedConnection(connectionFactory, proxy));
+			}
+
+			virtual void listen();
+			virtual void connect(const HostAddressPort& address);
+			virtual void disconnect();
+			virtual void write(const SafeByteArray& data);
+
+			virtual HostAddressPort getLocalAddress() const;
+
+		private:
+			SOCKS5ProxiedConnection(ConnectionFactory* connectionFactory, const HostAddressPort& proxy);
+
+			void handleConnectionConnectFinished(bool error);
+			void handleDataRead(boost::shared_ptr<SafeByteArray> data);
+			void handleDisconnected(const boost::optional<Error>& error);
+
+		private:
+			enum {
+				ProxyAuthenticating = 0,
+				ProxyConnecting,
+			} proxyState_;
+			bool connected_;
+			ConnectionFactory* connectionFactory_;	
+			HostAddressPort proxy_;
+			HostAddressPort server_;
+			boost::shared_ptr<Connection> connection_;
+	};
+}