Move to new Build System

MailServer/Connection.cpp

@@ -0,0 +1,61 @@
+#include "Connection.h"
+#include <functional>
+#include <sstream>
+
+
+namespace MailServer {
+
+Connection::Connection(asio::io_service& ioService, const Util::Context& context) :
+	m_ioService(ioService),
+	m_socket(ioService),
+	m_protocol(context)
+{
+}
+
+asio::ip::tcp::socket& Connection::Socket()
+{
+	return m_socket;
+}
+
+void Connection::Start()
+{
+	StartWrite(m_protocol.OpenConnection(m_socket.remote_endpoint().address().to_string()).second);
+}
+
+void Connection::StartRead()
+{
+	m_socket.async_read_some(asio::buffer(m_data), std::bind(&Connection::HandleRead, this, std::placeholders::_1, std::placeholders::_2));
+}
+
+void Connection::HandleRead(const asio::error_code& ec, std::size_t length)
+{
+	if (!ec)
+	{
+		Protocol::Protocol::Result result = m_protocol.ProcessMessage(std::string(m_data.begin(), m_data.begin() + length));
+		if (result.second.size() > 0)
+			StartWrite(result.second);
+		else
+			StartRead();
+
+		if (result.first == Protocol::Protocol::State::Disconnected)
+			m_socket.close();
+	}
+}
+
+void Connection::StartWrite(const std::string& message)
+{
+	std::copy(message.begin(), message.end(), m_data.data());
+
+	size_t length = message.size() + 1;
+	m_data[message.size()] = '\n';
+
+	asio::async_write(m_socket, asio::buffer(m_data, length), std::bind(&Connection::HandleWrite, this, std::placeholders::_1));
+}
+
+void Connection::HandleWrite(const asio::error_code& ec)
+{
+	if (!ec)
+		StartRead();
+}
+
+} // namespace MailServer