c++ - boost::asio - 知道什么时候必须关闭/关闭连接

Question

我实现了一个协议(protocol) (socks)，该协议(protocol)要求我的服务器将来自客户端的连接中继到目的地。

我实现中继部分的方式是使用类似这样的东西:

socket_.async_read_some(boost::asio::buffer(dataClient_, 1024),
boost::bind(&ProxySocksSession::HandleClientProxyRead, this,
 boost::asio::placeholders::error,
 boost::asio::placeholders::bytes_transferred));

remoteSock_.async_read_some(boost::asio::buffer(dataRemote_, 1024),
boost::bind(&ProxySocksSession::HandleRemoteProxyRead, this,
 boost::asio::placeholders::error,
 boost::asio::placeholders::bytes_transferred));

当然还有更多的代码 - 那里有处理程序中继来自 socket_ 的数据并将其发送到 remoteSock_ 反之亦然(所有来自 remoteSock_ 的数据都被中继到套接字_)

我看到了异步 tcp 服务器回显示例 (http://www.boost.org/doc/libs/1_38_0/doc/html/boost_asio/example/echo/async_tcp_echo_server.cpp) 并且关闭连接的逻辑只是删除​​连接对象 - 删除它 - (关闭它拥有的通信套接字)当 在处理程序中收到一个 boost::system::error_code。

我该如何自己处理这个案子？这次我有 2 个套接字上的数据，如果我没有干净地关闭，我可能会在所有数据传输之前关闭(例如来自客户端的套接字 - socket_ - 可能会关闭连接但是 - remoteSock - 可能仍在尝试发送数据)。

编辑 我已经更新了我的代码，如果我检测到其中一个套接字(remoteSock_ 或 socket_)读/写处理程序报告了一个 boost::system::error_code，我将执行以下操作为了关闭通信:

void ProxySocksSession::Shutdown()
{
    if (!shutdownInProgress_)
    {
        std::cout << "Shuting down ..." << std::endl;
        shutdownInProgress_ = true;
        remoteSock_.shutdown((boost::asio::ip::tcp::socket::shutdown_both));
        remoteSock_.close();
        socket_.shutdown((boost::asio::ip::tcp::socket::shutdown_both));
        socket_.close();
        parentConnection_.Shutdown();
    }
}

问题是，即使我在套接字上调用 shutdown() 和 close()，我仍然会收到对套接字处理程序的调用(它们在同一个类 ProxySocksSession 中)。当这些出现时，我的 ProxySocksSession 实例已经被删除(删除是由上面的 parentConnection_.Shutdown() 完成的)

Answer 1

我设法提出了一个有效的解决方案(不会导致所描述的问题)。为了了解这个想法，我还在下面包含了处理函数的框架:

void ProxySocksSession::Start()
{
    socket_.async_read_some(boost::asio::buffer(dataClient_, 1024),
       boost::bind(&ProxySocksSession::HandleClientProxyRead, shared_from_this(),
           boost::asio::placeholders::error,
           boost::asio::placeholders::bytes_transferred));

    remoteSock_.async_read_some(boost::asio::buffer(dataRemote_, 1024),
           boost::bind(&ProxySocksSession::HandleRemoteProxyRead, shared_from_this(),
           boost::asio::placeholders::error,
           boost::asio::placeholders::bytes_transferred));
}

// received data from socks5 client - completion handler
void ProxySocksSession::HandleClientProxyRead(const boost::system::error_code& error,
      size_t bytes_transferred)
{
     if (!error && !this->shutdownInProgress_)
     {
         // relay data coming from client to remote endpoint -> async write to remoteSock_
         // async read some more data from socket_
      }
      else
      {
         Shutdown();
      }
 }

//received data from socks5 remote endpoint (socks5 client destination)
void ProxySocksSession::HandleRemoteProxyRead(const boost::system::error_code& error,
      size_t bytes_transferred)
{
     if (!error && !this->shutdownInProgress_)
     {
         // relay data coming from remote endpoint to client -> async write to socket__
         // async read some more data from remoteSock_
      }
      else
      {
         Shutdown();
      }
 }

void ProxySocksSession::Shutdown()
{
    if (!shutdownInProgress_)
    {
        std::cout << "Shuting down ..." << std::endl;
        shutdownInProgress_ = true;
        //remoteSock_.close(); -- no need as it is closed automatically as part of parentConnection_ shutdown/deletion
        //socket_.close(); -- no need as it is closed automatically as part of parentConnection_ shutdown/deletion
        parentConnection_.Shutdown();
    }
}

这里的关键是我在绑定(bind)完成处理程序时使用了shared_from_this()。通过这种方式，我确保 ProxySocksSession 实例的删除 未完成 parentConnection_ 实例之前具有到 ProxySocksSession 的 shared_ptr > 在套接字关闭时调用了所有 ProxySocksSession 处理程序。