| // |
| // connection.cpp |
| // ~~~~~~~~~~~~~~ |
| // |
| // Copyright (c) 2003-2016 Christopher M. Kohlhoff (chris at kohlhoff dot com) |
| // |
| // Distributed under the Boost Software License, Version 1.0. (See accompanying |
| // file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt) |
| // |
| |
| #include "connection.hpp" |
| #include <vector> |
| #include <boost/bind.hpp> |
| #include "request_handler.hpp" |
| |
| namespace http { |
| namespace server2 { |
| |
| connection::connection(asio::io_context& io_context, |
| request_handler& handler) |
| : socket_(io_context), |
| request_handler_(handler) |
| { |
| } |
| |
| asio::ip::tcp::socket& connection::socket() |
| { |
| return socket_; |
| } |
| |
| void connection::start() |
| { |
| socket_.async_read_some(asio::buffer(buffer_), |
| boost::bind(&connection::handle_read, shared_from_this(), |
| asio::placeholders::error, |
| asio::placeholders::bytes_transferred)); |
| } |
| |
| void connection::handle_read(const asio::error_code& e, |
| std::size_t bytes_transferred) |
| { |
| if (!e) |
| { |
| boost::tribool result; |
| boost::tie(result, boost::tuples::ignore) = request_parser_.parse( |
| request_, buffer_.data(), buffer_.data() + bytes_transferred); |
| |
| if (result) |
| { |
| request_handler_.handle_request(request_, reply_); |
| asio::async_write(socket_, reply_.to_buffers(), |
| boost::bind(&connection::handle_write, shared_from_this(), |
| asio::placeholders::error)); |
| } |
| else if (!result) |
| { |
| reply_ = reply::stock_reply(reply::bad_request); |
| asio::async_write(socket_, reply_.to_buffers(), |
| boost::bind(&connection::handle_write, shared_from_this(), |
| asio::placeholders::error)); |
| } |
| else |
| { |
| socket_.async_read_some(asio::buffer(buffer_), |
| boost::bind(&connection::handle_read, shared_from_this(), |
| asio::placeholders::error, |
| asio::placeholders::bytes_transferred)); |
| } |
| } |
| |
| // If an error occurs then no new asynchronous operations are started. This |
| // means that all shared_ptr references to the connection object will |
| // disappear and the object will be destroyed automatically after this |
| // handler returns. The connection class's destructor closes the socket. |
| } |
| |
| void connection::handle_write(const asio::error_code& e) |
| { |
| if (!e) |
| { |
| // Initiate graceful connection closure. |
| asio::error_code ignored_ec; |
| socket_.shutdown(asio::ip::tcp::socket::shutdown_both, ignored_ec); |
| } |
| |
| // No new asynchronous operations are started. This means that all shared_ptr |
| // references to the connection object will disappear and the object will be |
| // destroyed automatically after this handler returns. The connection class's |
| // destructor closes the socket. |
| } |
| |
| } // namespace server2 |
| } // namespace http |