Initial commit

Http/HttpConnection.cpp

@@ -0,0 +1,92 @@
+#include "HttpConnection.h"
+#include <functional>
+
+
+namespace Http {
+
+HttpConnection::HttpConnection(asio::io_service& ioService, HttpRequestHandler& handler) :
+	m_strand(ioService),
+	m_socket(ioService),
+	m_requestHandler(handler),
+	m_dataState(false, 0),
+	m_dataIncomplete(false)
+{
+}
+
+HttpConnection::~HttpConnection()
+{
+}
+
+asio::ip::tcp::socket& HttpConnection::Socket()
+{
+	return m_socket;
+}
+
+void HttpConnection::Start()
+{
+	m_socket.async_read_some(asio::buffer(m_data), m_strand.wrap(std::bind(&HttpConnection::HandleRead, shared_from_this(), std::placeholders::_1, std::placeholders::_2)));
+}
+
+void HttpConnection::HandleRead(const asio::error_code& ec, std::size_t length)
+{
+	if (!ec)
+	{
+		size_t dataLocation;
+		Utilities::Tribool result;
+		Buffer::iterator dataIterator;
+
+		if (!m_dataIncomplete)
+		{
+			m_dataState = m_requestParser.Parse(m_request, m_data.data(), m_data.data() + length);
+			dataLocation = std::get<1>(m_dataState) - m_data.data();
+		}
+		else
+		{
+			std::get<1>(m_dataState) = m_data.data();
+			dataLocation = 0;
+		}
+
+		std::tie(result, dataIterator) = m_dataState;
+
+		if (result && length < m_request.headers.GetContentLength() + dataLocation)
+		{
+			m_dataIncomplete = true;
+			result = Utilities::Tribool::Indeterminate;
+		}
+		else
+		{
+			m_dataIncomplete = false;
+		}
+
+		if (result)
+		{
+			std::string data;
+
+			while (dataIterator != m_data.data() + length)
+				data.push_back(*dataIterator++);
+
+			HttpReply reply = m_requestHandler.HandleRequest(m_request, data);
+			asio::async_write(m_socket, reply.ToBuffers(), m_strand.wrap(std::bind(&HttpConnection::HandleWrite, shared_from_this(), std::placeholders::_1)));
+		}
+		else if (!result)
+		{
+			HttpReply reply = HttpReply::StockReply(HttpReply::Status::BadRequest);
+			asio::async_write(m_socket, reply.ToBuffers(), m_strand.wrap(std::bind(&HttpConnection::HandleWrite, shared_from_this(), std::placeholders::_1)));
+		}
+		else
+		{
+			m_socket.async_read_some(asio::buffer(m_data), m_strand.wrap(std::bind(&HttpConnection::HandleRead, shared_from_this(), std::placeholders::_1, std::placeholders::_2)));
+		}
+	}
+}
+
+void HttpConnection::HandleWrite(const asio::error_code& ec)
+{
+	if (!ec)
+	{
+		asio::error_code ignored_ec;
+		m_socket.shutdown(asio::ip::tcp::socket::shutdown_both, ignored_ec);
+	}
+}
+
+} // namespace Http