boost::asio - 知道何时必须关闭/关闭连接

Question

我实现了一个协议（袜子），该协议要求我的服务器将来自客户端的连接中继到目的地。

我实现中继部分的方式是使用这样的：

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_ 的所有数据都中继到 socket_)

我看到了异步 tcp 服务器回显示例 (http://www.boost.org/doc/libs/1_38_0/doc/html/boost_asio/example/echo/async_tcp_echo_server.cpp)，其中关闭连接的逻辑只是删除​​连接对象 - 删除它 - （关闭它拥有的通信套接字）when 在处理程序中收到了 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();
    }
}

key 是我在处理完成处理程序时使用了 shared_from_this()。通过这种方式，我确保 删除 ProxySocksSession 实例没有由 parentConnection_ 实例完成，该实例在 ProxySocksSession 之前上有一个 shared_ptr > 所有 ProxySocksSession 处理程序在套接字关闭时被调用。