通过boost发送大型身体:: asio :: tcp :: ip

Question

我的协议非常简单：我发送了一个size_t，指示身体的大小，然后是身体本身。

的代码是基于official Boost examples，那就是：

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_的递归更新中的某处 - 某些数据丢失，被覆盖？

Answer 1

buffer_.commit()无效的prepare()的结果，如记录：其实"The returned object is invalidated by any basic_streambuf member function that modifies the input sequence or output sequence."

，而不是手动准备缓冲区和链async_read_some()，你可以只使用与streambuf适当的无辅助功能和完成情况：

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

这种方式handle_read在获得整个消息正文或发生错误时被调用。