1 files changed, 42 insertions, 0 deletions
diff --git a/Swiften/Network/BoostConnection.h b/Swiften/Network/BoostConnection.h
new file mode 100644
index 0000000..ae09fb8
--- /dev/null
+++ b/Swiften/Network/BoostConnection.h
@@ -0,0 +1,42 @@
+#pragma once
+
+#include <boost/asio.hpp>
+#include <boost/enable_shared_from_this.hpp>
+
+#include "Swiften/Network/Connection.h"
+#include "Swiften/EventLoop/EventOwner.h"
+
+namespace boost {
+	class thread;
+	namespace system {
+		class error_code;
+	}
+}
+
+namespace Swift {
+	class BoostConnection : public Connection, public EventOwner, public boost::enable_shared_from_this<BoostConnection> {
+		public:
+			BoostConnection(boost::asio::io_service* ioService);
+			~BoostConnection();
+
+			virtual void listen();
+			virtual void connect(const HostAddressPort& address);
+			virtual void disconnect();
+			virtual void write(const ByteArray& data);
+
+			boost::asio::ip::tcp::socket& getSocket() {
+				return socket_;
+			}
+
+		private:
+			void handleConnectFinished(const boost::system::error_code& error);
+			void handleSocketRead(const boost::system::error_code& error, size_t bytesTransferred);
+			void handleDataWritten(const boost::system::error_code& error);
+			void doRead();
+
+		private:
+			boost::asio::ip::tcp::socket socket_;
+			std::vector<char> readBuffer_;
+			bool disconnecting_;
+	};
+}