93 lines
2.2 KiB
C++
93 lines
2.2 KiB
C++
#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;
|
|
Tribool::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 = Tribool::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
|