我怎样才能获得可用于在Linux上的异步套接字上读取的字节数？

Question

我有一个简单的tcp/ip服务器用C++在linux上编写。我使用异步套接字和epoll。当我收到EPOLLIN事件时，是否有可能找出有多少字节可供阅读？

Answer 1

从man 7 tcp：

int value; 
error = ioctl(sock, FIONREAD, &value); 

或替代地SIOCINQ，这是FIONREAD的同义词。

无论如何，我建议只是在循环中以非阻塞模式使用recv，直到它返回EWOULDBLOCK。

UPDATE：

从您的评论下面我想，这是不是你的问题，相应的解决方案。

想象一下，你的标题是8字节，你只收到4;那么你的poll/select将返回EPOLLIN，你会检查FIONREAD，看到头还没有完成和更多的字节wayt。但是这些字节永远不会到达，因此每次调用poll/select时都会收到EPOLLIN，并且您有一个无操作的忙循环。也就是说，poll/select是级触发的。不是边缘触发功能也可以解决您的问题。

最后，你做得好一点，每个连接添加一个缓冲区，然后排队字节，直到有足够的空间。它并不像看起来那么困难，而且效果更好。例如，类似的东西：

struct ConnectionData 
{ 
    int sck; 
    std::vector<uint8_t> buffer; 
    size_t offset, pending; 
}; 

void OnPollIn(ConnectionData *d) 
{ 
    int res = recv(d->sck, d->buffer.data() + offset, d->pending); 
    if (res < 0) 
     handle_error(); 
    d->offset += res; 
    d->pending -= res; 

    if (d->pending == 0) 
     DoSomethingUseful(d); 
} 

，只要你想得到的字节数：

void PrepareToRecv(ConnectionData *d, size_t size) 
{ 
    d->buffer.resize(size); 
    d->offset = 0; 
    d->pending = size; 
}