walkero
/
nghttp2
1
0
Fork 0
Code Issues Pull Requests Projects Releases 1 Wiki Activity

asio: Listen to all resolved addresses

This commit is contained in:
Tatsuhiro Tsujikawa 2014-11-30 22:39:16 +09:00
parent d23105ccb7
commit b9667fd209
2 changed files with 48 additions and 37 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

83
src/asio_server.cc
View File

@ -50,7 +50,6 @@ server::server(const std::string &address, uint16_t port,
signals_(io_service_pool_.get_io_service()), signals_(io_service_pool_.get_io_service()),
tick_timer_(io_service_pool_.get_io_service(), tick_timer_(io_service_pool_.get_io_service(),
boost::posix_time::seconds(1)), boost::posix_time::seconds(1)),
acceptor_(io_service_pool_.get_io_service()),
ssl_ctx_(std::move(ssl_ctx)), request_cb_(std::move(cb)) { ssl_ctx_(std::move(ssl_ctx)), request_cb_(std::move(cb)) {
// Register to handle the signals that indicate when the server should exit. // Register to handle the signals that indicate when the server should exit.
// It is safe to register for the same signal multiple times in a program, // It is safe to register for the same signal multiple times in a program,
@ -64,16 +63,24 @@ server::server(const std::string &address, uint16_t port,
int signal_number) { io_service_pool_.stop(); }); int signal_number) { io_service_pool_.stop(); });
// Open the acceptor with the option to reuse the address (i.e. SO_REUSEADDR). // Open the acceptor with the option to reuse the address (i.e. SO_REUSEADDR).
boost::asio::ip::tcp::resolver resolver(acceptor_.get_io_service()); boost::asio::ip::tcp::resolver resolver(io_service_pool_.get_io_service());
boost::asio::ip::tcp::resolver::query query(address, std::to_string(port)); boost::asio::ip::tcp::resolver::query query(address, std::to_string(port));
boost::asio::ip::tcp::endpoint endpoint = *resolver.resolve(query);
acceptor_.open(endpoint.protocol()); for (auto itr = resolver.resolve(query);
acceptor_.set_option(boost::asio::ip::tcp::acceptor::reuse_address(true)); itr != boost::asio::ip::tcp::resolver::iterator(); ++itr) {
acceptor_.bind(endpoint); boost::asio::ip::tcp::endpoint endpoint = *itr;
if (backlog == -1) { auto acceptor =
acceptor_.listen(); boost::asio::ip::tcp::acceptor(io_service_pool_.get_io_service());
} else {
acceptor_.listen(backlog); acceptor.open(endpoint.protocol());
acceptor.set_option(boost::asio::ip::tcp::acceptor::reuse_address(true));
acceptor.bind(endpoint);
if (backlog == -1) {
acceptor.listen();
} else {
acceptor.listen(backlog);
}
acceptors_.push_back(std::move(acceptor));
} }
start_accept(); start_accept();
@ -109,40 +116,44 @@ void server::start_accept() {
request_cb_, io_service_pool_.get_task_io_service(), request_cb_, io_service_pool_.get_task_io_service(),
io_service_pool_.get_io_service(), *ssl_ctx_); io_service_pool_.get_io_service(), *ssl_ctx_);
acceptor_.async_accept( for (auto &acceptor : acceptors_) {
new_connection->socket().lowest_layer(), acceptor.async_accept(
[this, new_connection](const boost::system::error_code &e) { new_connection->socket().lowest_layer(),
if (!e) { [this, new_connection](const boost::system::error_code &e) {
new_connection->socket().lowest_layer().set_option( if (!e) {
boost::asio::ip::tcp::no_delay(true)); new_connection->socket().lowest_layer().set_option(
new_connection->socket().async_handshake( boost::asio::ip::tcp::no_delay(true));
boost::asio::ssl::stream_base::server, new_connection->socket().async_handshake(
[new_connection](const boost::system::error_code &e) { boost::asio::ssl::stream_base::server,
if (!e) { [new_connection](const boost::system::error_code &e) {
new_connection->start(); if (!e) {
} new_connection->start();
}); }
} });
}
start_accept(); start_accept();
}); });
}
} else { } else {
auto new_connection = auto new_connection =
std::make_shared<connection<boost::asio::ip::tcp::socket>>( std::make_shared<connection<boost::asio::ip::tcp::socket>>(
request_cb_, io_service_pool_.get_task_io_service(), request_cb_, io_service_pool_.get_task_io_service(),
io_service_pool_.get_io_service()); io_service_pool_.get_io_service());
acceptor_.async_accept( for (auto &acceptor : acceptors_) {
new_connection->socket(), acceptor.async_accept(
[this, new_connection](const boost::system::error_code &e) { new_connection->socket(),
if (!e) { [this, new_connection](const boost::system::error_code &e) {
new_connection->socket().set_option( if (!e) {
boost::asio::ip::tcp::no_delay(true)); new_connection->socket().set_option(
new_connection->start(); boost::asio::ip::tcp::no_delay(true));
} new_connection->start();
}
start_accept(); start_accept();
}); });
}
} }
} }

2
src/asio_server.h
View File

@ -86,7 +86,7 @@ private:
boost::asio::deadline_timer tick_timer_; boost::asio::deadline_timer tick_timer_;
/// Acceptor used to listen for incoming connections. /// Acceptor used to listen for incoming connections.
boost::asio::ip::tcp::acceptor acceptor_; std::vector<boost::asio::ip::tcp::acceptor> acceptors_;
std::unique_ptr<boost::asio::ssl::context> ssl_ctx_; std::unique_ptr<boost::asio::ssl::context> ssl_ctx_;