2016-02-26 86 views
3

我有哪里boost::asio::async_read第二呼叫失败的问题,以一种奇怪的方式:为什么boost :: asio :: async_read无法读取请求的字节数?

std::atomic<error_code> ec(asio::error::would_block); 
size_t len = 0; 

// 1st call 
asio::async_read(socket, 
       asio::buffer(buffer+X), 
       asio::transfer_exactly(512-X), 
       [&] (error_code const& err, size_t bytesTransferred) 
       { 
        len = bytesTransferred; 
        ec.store(err, std::memory_order_release); 
       } 
); 
/////// ... wait for read to complete ... 
// 2nd call 
asio::async_read(socket, 
       asio::buffer(buffer), 
       asio::transfer_exactly(512), 
       [&] (error_code const& err, size_t bytesTransferred) 
       { 
        len = bytesTransferred; 
        ec.store(err, std::memory_order_release); 
       } 
); 

X是因为我有一些数据已经是我以另一种方式得到的,所以 第一次读更小。说X = 364,那么bytesTransferred将是第一次148。然而我的问题是,第二次读取再次返回148个字节,即使读取完全是512字节。

我很困惑。第二个电话没有错误情况(我检查了err)。 bytesTransferred是由aync_read传给我的一个参数,它是148个字节的两倍。第一次,它匹配堆栈中较高的asio::transfer_exactly(148)。第二次callstack显然有asio::transfer_exactly(512)。这里发生了什么?

虽然第二次调用是特别的,但第三次调用再次读取512字节,但也获得512字节。

[MCVE]

#include <iostream> 
#include <atomic> 

#include <boost/asio/buffer.hpp> 
#include <boost/asio/ip/tcp.hpp> 
#include <boost/asio/write.hpp> 
#include <boost/asio/read_until.hpp> 
#include <boost/asio/read.hpp> 

// Minimal example, code that works has error checking removed. Class members turned itno globals etc. 
namespace { 
    boost::asio::io_service io_service; 
    boost::asio::ip::tcp::resolver resolver(io_service); 
    boost::asio::ip::tcp::socket sock(io_service); 
    std::vector<char> data(512); 
    boost::asio::mutable_buffers_1 buffer(&data[0], data.size()); 
    unsigned read_counter = 1; 
    std::atomic<unsigned> read_timeout; 
} 

boost::system::error_code openSocket(const std::string &server, 
            const std::string &port) 
{ 
    boost::system::error_code error = boost::asio::error::host_not_found; 
    using boost::asio::ip::tcp; 
    tcp::resolver::query query(server, port); 
    tcp::resolver::iterator endpoint_iterator = resolver.resolve(query); 
    tcp::resolver::iterator end; 

    while (error && endpoint_iterator != end) 
    { 
     sock.close(); 
     sock.connect(*endpoint_iterator++, error); 
    } 
    if (error) 
    { 
     std::cerr << "No route\n"; 
     sock.close(); // Would be wrong to leave it open. 
    } 
    return error; 
} 

int read(size_t bytesNeeded) 
{ 
    size_t buffer_len = boost::asio::buffer_size(buffer); 
    size_t byteShift = buffer_len - bytesNeeded; // Read into back of buffer. 

    const int timeoutSeconds = 10; 
    boost::asio::deadline_timer deadline(io_service); 
    deadline.expires_from_now(boost::posix_time::seconds(timeoutSeconds)); // This will reset any outstanding timer 
    read_counter += 2; // If we'd use +1, after 4 billion cycles it would reset to 0 
    read_timeout.store(0, std::memory_order_release); // 0 = no timeout. 
    unsigned read_counter_copy = read_counter; // Can't capture global. 
    deadline.async_wait([read_counter_copy](boost::system::error_code const&) { 
     // read_timeout is very intentionally captured by value - timeout events are numbered 
     read_timeout.store(read_counter_copy, std::memory_order_release); } 
    ); 

    // Start reading "asynchronously", wait for completion or timeout: 
    std::atomic<boost::system::error_code> ec(boost::asio::error::would_block); 
    size_t len = 0; 

    boost::asio::async_read(sock, boost::asio::buffer(buffer + byteShift), boost::asio::transfer_exactly(bytesNeeded), 
     [&, bytesNeeded](boost::system::error_code const& err, size_t bytesTransferred) 
     { 
      if (bytesTransferred != bytesNeeded) { 
       std::cout << bytesTransferred << " , " << err.message() << std::endl; 
      } 
      len = bytesTransferred; 
      ec.store(err, std::memory_order_release); 
     } 
    ); 

    do { 
     io_service.run_one(); 
    } while (read_timeout.load(std::memory_order_acquire) != read_counter && // Continue if the **last** read didn't time out 
     (ec.load(std::memory_order_acquire) == boost::asio::error::would_block) && // ec.store() not called, 
     !io_service.stopped()); // and program still running. 

    deadline.cancel(); // This will set read_timeout, if it wasn't set yet. But we ignore it from now on. 

    if (ec.load(std::memory_order_acquire)) 
    { 
     std::cerr << "oops\n"; // Real error handling omitted. 
     throw std::runtime_error(""); 
    } 
    else if (read_timeout == read_counter) 
    { 
     std::cerr << "timeout\n"; 
    } 
    else if (len != bytesNeeded) 
    { 
     // This is the real problem. 
     std::cerr << "Asked " << bytesNeeded << " got " << len; 
    } 
    return (int)len; 
} 

int main(int argc, char* argv[]) 
{ 
    do try { 
     ::openSocket("192.168.6.30", "80"); 

     read(148); // Assume that data[] already has 364 bytes on the first call. 
     for (;;) 
     { 
      read(512); // Full buffers on every subsequent call. 
      // Do something with data[] here. 
     } 
    } 
    catch (std::runtime_error) { } while (true); 
} 

do try catch while是必要的,因为错误仅发生后我拔掉的另一侧。在第二次调用read(148)后,下一次读取(512)`失败。

[更新] 这不仅仅是transfer_exactly。与transfer_at_least(512)我也遇到了同样的问题,一个多余的148字节读取。 (两者应具有相同的行为,如读至少 512字节到缓冲区这 512字节不能读取更多或更少字节)

+2

你能提供[MCVE]吗?你使用哪种增强版本? –

+0

它说它应该填充缓冲区?指定所有读取方法以返回实际传输的字节数,因为它们可能*不填充缓冲区。在套接字的情况下,如果一次只有148个字节到达,那就是你会得到的。 – EJP

+0

@EJP http://www.boost.org/doc/libs/1_60_0/doc/html/booster_asio/reference/transfer_exactly.html – sehe

回答

0

通过忽略不正确的读取操作“解决”现在它。幸运的是,我可以处理未知数量的丢失数据,并在稍后与数据流重新同步。但看起来我将来不得不放弃Boost :: Asio,因为我不能再容忍错过的数据。

相关问题