You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

144 lines
3.8 KiB
C++

#include "webcc/http_connection.h"
#include <utility> // for move()
#include "boost/asio/write.hpp"
#include "webcc/http_connection_pool.h"
#include "webcc/http_request_handler.h"
#include "webcc/logger.h"
using boost::asio::ip::tcp;
namespace webcc {
HttpConnection::HttpConnection(tcp::socket socket, HttpConnectionPool* pool,
HttpRequestHandler* handler)
: socket_(std::move(socket)),
pool_(pool),
buffer_(kBufferSize),
request_handler_(handler) {
}
void HttpConnection::Start() {
request_.reset(new HttpRequest{});
request_parser_.Init(request_.get());
DoRead();
}
void HttpConnection::Close() {
LOG_INFO("Close socket...");
boost::system::error_code ec;
socket_.close(ec);
if (ec) {
LOG_ERRO("Socket close error (%s).", ec.message().c_str());
}
}
void HttpConnection::SendResponse(HttpResponsePtr response) {
assert(response);
response_ = response;
if (request_->IsConnectionKeepAlive()) {
response_->SetHeader(http::headers::kConnection, "Keep-Alive");
} else {
response_->SetHeader(http::headers::kConnection, "Close");
}
response_->Prepare();
DoWrite();
}
void HttpConnection::SendResponse(http::Status status) {
SendResponse(std::make_shared<HttpResponse>(status));
}
void HttpConnection::DoRead() {
socket_.async_read_some(boost::asio::buffer(buffer_),
std::bind(&HttpConnection::OnRead, shared_from_this(),
std::placeholders::_1,
std::placeholders::_2));
}
void HttpConnection::OnRead(boost::system::error_code ec, std::size_t length) {
if (ec) {
LOG_ERRO("Socket read error (%s).", ec.message().c_str());
if (ec != boost::asio::error::operation_aborted) {
pool_->Close(shared_from_this());
}
return;
}
if (!request_parser_.Parse(buffer_.data(), length)) {
// Bad request.
LOG_ERRO("Failed to parse HTTP request.");
SendResponse(http::Status::kBadRequest);
return;
}
if (!request_parser_.finished()) {
// Continue to read the request.
DoRead();
return;
}
LOG_VERB("HTTP request:\n%s", request_->Dump(4, "> ").c_str());
// Enqueue this connection.
// Some worker thread will handle it later.
request_handler_->Enqueue(shared_from_this());
}
void HttpConnection::DoWrite() {
LOG_VERB("HTTP response:\n%s", response_->Dump(4, "> ").c_str());
boost::asio::async_write(socket_, response_->payload(),
std::bind(&HttpConnection::OnWrite, shared_from_this(),
std::placeholders::_1,
std::placeholders::_2));
}
// NOTE:
// This write handler will be called from main thread (the thread calling
// io_context.run), even though AsyncWrite() is invoked by worker threads.
// This is ensured by Asio.
void HttpConnection::OnWrite(boost::system::error_code ec, std::size_t length) {
if (ec) {
LOG_ERRO("Socket write error (%s).", ec.message().c_str());
if (ec != boost::asio::error::operation_aborted) {
pool_->Close(shared_from_this());
}
} else {
LOG_INFO("Response has been sent back, length: %u.", length);
if (request_->IsConnectionKeepAlive()) {
LOG_INFO("The client asked for a keep-alive connection.");
LOG_INFO("Continue to read the next request...");
Start();
} else {
Shutdown();
pool_->Close(shared_from_this());
}
}
}
// Socket close VS. shutdown:
// https://stackoverflow.com/questions/4160347/close-vs-shutdown-socket
void HttpConnection::Shutdown() {
LOG_INFO("Shutdown socket...");
// Initiate graceful connection closure.
boost::system::error_code ec;
socket_.shutdown(tcp::socket::shutdown_both, ec);
if (ec) {
LOG_ERRO("Socket shutdown error (%s).", ec.message().c_str());
}
}
} // namespace webcc