通过boost::asio::tcp::ip发送的大正文

Large body sent via boost::asio::tcp::ip

本文关键字：正文 ip boost asio tcp 通过      更新时间：2023-10-16

我的协议很简单:我发送了一个size_t，指示正文的大小，然后是正文本身。

代码基于官方Boost示例，如下:

class tcp_conn : 
        public std::enable_shared_from_this<tcp_conn>, 
        private boost::noncopyable 
    {
    public:
        tcp_conn(ba::io_service& io_service);
        void start();
        void stop();
        tcp::socket& socket();
    private:
        void handle_read_header(const error_code& e, std::size_t bytes_transferred);
        void handle_read(const error_code& e, std::size_t bytes_transferred);
        ba::streambuf::mutable_buffers_type buffer_wrapper_;
        ba::streambuf buffer_;
        std::size_t buffer_size_;
        tcp::socket socket_;
};

实现为:

void tcp_conn::start() {
    socket_.async_read_some(
        ba::buffer(&buffer_size_, sizeof(buffer_size_)),
        boost::bind(
          &tcp_conn::handle_read_header,
          this,
          ba::placeholders::error,
          ba::placeholders::bytes_transferred
          )
        );
}
void tcp_conn::handle_read_header(const boost::system::error_code& e, std::size_t bytes_transferred){
    if(!e) {
        buffer_wrapper_ =  buffer_.prepare(buffer_size_);
        socket_.async_read_some(
            ba::buffer(buffer_wrapper_),
            boost::bind(
              &tcp_conn::handle_read,
              this,
              ba::placeholders::error,
              ba::placeholders::bytes_transferred
              )
            ); 
    } else {
        //stop connection
    }
}

void tcp_conn::handle_read(const boost::system::error_code& e, std::size_t bytes_transferred)   {
    if (!e ) {
        buffer_.commit(bytes_transferred);
        if(buffer_.size() == buffer_size_) {
            std::cout << "WHOLE BODY TRANSFERED NOW" << std::endl;
            //has all data in buffer, handle it.
        }
        else {
            std::cout << "NOT WHOLE BODY TRANSFERED, still got "<< (buffer_size_ - bytes_transferred) << " to go" << std::endl;
            socket_.async_read_some(
                ba::buffer(buffer_wrapper_),
                boost::bind(
                  &tcp_conn::handle_read,
                  this,
                  ba::placeholders::error,
                  ba::placeholders::bytes_transferred
                  )
                ); 
        }

    }
    else {
        //handle error
    }   
}

这个不太管用。我猜我的错误是在buffer_wrapper_的递归更新的某个地方-一些数据丢失，覆盖?

buffer_.commit()使prepare()的结果无效，如文档所述:"返回的对象被任何修改输入序列或输出序列的basic_streambuf成员函数无效。"

实际上，不需要手动准备缓冲区并链接async_read_some()，您只需使用streambuf和完成条件的适当的helper free函数:

asio::async_read(
   socket_,
   buffer_, 
   asio::transfer_exactly(buffer_size_),
   boost::bind(&tcp_conn::handle_read, this, _1, _2)
);

这样，当您获得整个消息正文或发生错误时，就会调用handle_read。

相关文章：