]>
Commit | Line | Data |
---|---|---|
7c673cae FG |
1 | // |
2 | // server.cpp | |
3 | // ~~~~~~~~~~ | |
4 | // | |
11fdf7f2 | 5 | // Copyright (c) 2003-2018 Christopher M. Kohlhoff (chris at kohlhoff dot com) |
7c673cae FG |
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.hpp> | |
13 | #include <signal.h> | |
14 | ||
15 | namespace http { | |
16 | namespace server { | |
17 | ||
18 | server::server(const std::string& address, const std::string& port, | |
19 | const std::string& doc_root) | |
b32b8144 FG |
20 | : io_context_(), |
21 | signals_(io_context_), | |
22 | acceptor_(io_context_), | |
7c673cae FG |
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). | |
b32b8144 FG |
38 | boost::asio::ip::tcp::resolver resolver(io_context_); |
39 | boost::asio::ip::tcp::endpoint endpoint = | |
40 | *resolver.resolve(address, port).begin(); | |
7c673cae FG |
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 | ||
49 | void server::run() | |
50 | { | |
b32b8144 | 51 | // The io_context::run() call will block until all asynchronous operations |
7c673cae FG |
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. | |
b32b8144 | 55 | io_context_.run(); |
7c673cae FG |
56 | } |
57 | ||
58 | void server::start_accept() | |
59 | { | |
b32b8144 | 60 | new_connection_.reset(new connection(io_context_, |
7c673cae FG |
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 | ||
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 | ||
84 | void server::handle_stop() | |
85 | { | |
86 | // The server is stopped by cancelling all outstanding asynchronous | |
b32b8144 | 87 | // operations. Once all operations have finished the io_context::run() call |
7c673cae FG |
88 | // will exit. |
89 | acceptor_.close(); | |
90 | connection_manager_.stop_all(); | |
91 | } | |
92 | ||
93 | } // namespace server | |
94 | } // namespace http |