Handle errors on remote_endpoint()

lvhuat · Sep 8, 2016 · e27ffde · e27ffde
1 parent 85ba065
commit e27ffde
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/src/served/net/connection.cpp b/src/served/net/connection.cpp
@@ -54,7 +54,15 @@ connection::connection( boost::asio::io_service &    io_service
 void
 connection::start()
 {
-	_request.set_source(_socket.remote_endpoint().address().to_string());
+	boost::system::error_code ec;
+	boost::asio::ip::tcp::endpoint endpoint = _socket.remote_endpoint(ec);
+	if (ec)
+	{
+		_connection_manager.stop(shared_from_this());
+		return;
+	}
+
+	_request.set_source(endpoint.address().to_string());
 	do_read();
 
 	if ( _read_timeout > 0 )