如何在 boost tcp/udp 服务器中处理 control-c

Question

我如何处理 control-C 事件或停止我的 boost::asio 服务器。我有一个 TCP & udp 组合服务器并希望能够在我按 ctrl-c 时干净地退出。我得到了未处理的 control-C 的第一次机会异常。这是我的代码

void startTCP()
{
  http::syncServer::server serv( 2);

 // Set console control handler to allow server to be stopped.
 // console_ctrl_function = boost::bind(&http::syncServer::server::stop, &serv);
 //SetConsoleCtrlHandler(console_ctrl_handler, TRUE);
 // Run the server until stopped.
 serv.run();
}


void startUDP()
{
  boost::asio::io_service io_service;
  http::syncServer::udp_server server(io_service);
 // console_ctrl_function = boost::bind(&http::syncServer::udp_server::stop, &server);
 // SetConsoleCtrlHandler(console_ctrl_handler, TRUE);
 io_service.run();
}

int main(int argc, char* argv[])
{
  try
  {
     boost::shared_ptr<boost::thread> tcpThread( new boost::thread(startTCP));
     boost::shared_ptr<boost::thread> udpThread (new boost::thread(startUDP));

     /*console_ctrl_function = boost::bind(&http::syncServer::udp_server::stop, &server);
     SetConsoleCtrlHandler(console_ctrl_handler, FALSE);*/

    tcpThread->join();
    udpThread->join();
}
catch (std::exception& e)
{
   std::cerr << "exception: " << e.what() << "\n";
}

return 0;
}

Answer 1

从 boost 1.47 开始，您可以使用 asio 的 signal_set：

class Server {
public:
  Server(...) {
    _signals.add(SIGINT);
    _signals.add(SIGTERM);
    _signals.async_wait(bind(&Server::handle_stop, this));

  }
  void handle_stop() {
    // do what you need to kill the Server - usually you just have to cancel all waits (incl deadline_timers), which will make the io_service exit gracefully
  }
private:
  boost::asio::signal_set _signals;
};

注意，最新 Ubuntu 的 boost 版本是 1.46 - 所以这是前沿（primo 2012）。

干杯，

迈克尔

Answer 2

C 标准库包含 （例如 参见此处)，您可以使用它注册 SIGINT (Ctrl-C) 的信号处理程序。假设您的平台支持信号，这应该可以解决问题。

您可能还想为 SIGTERM 注册一个处理程序，以优雅地响应被kill(1)ed。

#include <signal.h> // or <csignal> in C++

void ctrlchandler(int) { /*...*/ WE_MUST_STOP = 1; }
void killhandler(int) { /*...*/ WE_MUST_STOP = 2; }

int WE_MUST_STOP = 0;

int main() {
  signal(SIGINT, ctrlchandler);
  signal(SIGTERM, killhandler);
  /* ... */

  // e.g. main loop like this:
  while(pump_loop() && 0 == WE_MUST_STOP) { }

}

---

正如 Sam Miller 所建议的，假设您的主循环是一个带有一些 m_io_service.run() 的单线程 boost.asio 循环。然后，您可以代替全局标志（假设 m_io_service 可见）从信号处理程序中向 io 服务发布停止处理程序。

Answer 3

您可以使用 asio 中的 signal_set，如下所示

// stop on ctrlC signal

boost::asio::signal_set signals(
    io_service,
    SIGINT,
    SIGTERM );
signals.async_wait(
    boost::bind(
        &boost::asio::io_service::stop,
        &io_service));

文档在这里