1 //
2 // server.cpp
3 // ~~~~~~~~~~
4 //
5 // Copyright (c) 2003-2021 Christopher M. Kohlhoff (chris at kohlhoff dot com)
6 //
7 // Distributed under the Boost Software License, Version 1.0. (See accompanying
8 // file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
9 //
10
11 #include "server.hpp"
12 #include <boost/bind/bind.hpp>
13 #include <signal.h>
14
15 namespace http {
16 namespace server {
17
server(const std::string & address,const std::string & port,const std::string & doc_root)18 server::server(const std::string& address, const std::string& port,
19 const std::string& doc_root)
20 : io_context_(),
21 signals_(io_context_),
22 acceptor_(io_context_),
23 connection_manager_(),
24 new_connection_(),
25 request_handler_(doc_root)
26 {
27 // Register to handle the signals that indicate when the server should exit.
28 // It is safe to register for the same signal multiple times in a program,
29 // provided all registration for the specified signal is made through Asio.
30 signals_.add(SIGINT);
31 signals_.add(SIGTERM);
32 #if defined(SIGQUIT)
33 signals_.add(SIGQUIT);
34 #endif // defined(SIGQUIT)
35 signals_.async_wait(boost::bind(&server::handle_stop, this));
36
37 // Open the acceptor with the option to reuse the address (i.e. SO_REUSEADDR).
38 boost::asio::ip::tcp::resolver resolver(io_context_);
39 boost::asio::ip::tcp::endpoint endpoint =
40 *resolver.resolve(address, port).begin();
41 acceptor_.open(endpoint.protocol());
42 acceptor_.set_option(boost::asio::ip::tcp::acceptor::reuse_address(true));
43 acceptor_.bind(endpoint);
44 acceptor_.listen();
45
46 start_accept();
47 }
48
run()49 void server::run()
50 {
51 // The io_context::run() call will block until all asynchronous operations
52 // have finished. While the server is running, there is always at least one
53 // asynchronous operation outstanding: the asynchronous accept call waiting
54 // for new incoming connections.
55 io_context_.run();
56 }
57
start_accept()58 void server::start_accept()
59 {
60 new_connection_.reset(new connection(io_context_,
61 connection_manager_, request_handler_));
62 acceptor_.async_accept(new_connection_->socket(),
63 boost::bind(&server::handle_accept, this,
64 boost::asio::placeholders::error));
65 }
66
handle_accept(const boost::system::error_code & e)67 void server::handle_accept(const boost::system::error_code& e)
68 {
69 // Check whether the server was stopped by a signal before this completion
70 // handler had a chance to run.
71 if (!acceptor_.is_open())
72 {
73 return;
74 }
75
76 if (!e)
77 {
78 connection_manager_.start(new_connection_);
79 }
80
81 start_accept();
82 }
83
handle_stop()84 void server::handle_stop()
85 {
86 // The server is stopped by cancelling all outstanding asynchronous
87 // operations. Once all operations have finished the io_context::run() call
88 // will exit.
89 acceptor_.close();
90 connection_manager_.stop_all();
91 }
92
93 } // namespace server
94 } // namespace http
95