csocketsbufferrecvpeek

Get the number of bytes available in socket by 'recv' with 'MSG_PEEK' in C++


C++ has the following function to receive bytes from socket, it can check for number of bytes available with the MSG_PEEK flag. With MSG_PEEK, the returned value of 'recv' is the number of bytes available in socket:

#include <sys/socket.h>
ssize_t recv(int socket, void *buffer, size_t length, int flags); 

I need to get the number of bytes available in the socket without creating buffer (without allocating memory for buffer). Is it possible and how?


Solution

  • You're looking for is ioctl(fd,FIONREAD,&bytes_available) , and under windows ioctlsocket(socket,FIONREAD,&bytes_available).

    Be warned though, the OS doesn't necessarily guarantee how much data it will buffer for you, so if you are waiting for very much data you are going to be better off reading in data as it comes in and storing it in your own buffer until you have everything you need to process something.

    To do this, what is normally done is you simply read chunks at a time, such as

    char buf[4096];
    ssize_t bytes_read;
    do {
         bytes_read = recv(socket, buf, sizeof(buf), 0);
         if (bytes_read > 0) {
             /* do something with buf, such as append it to a larger buffer or
              * process it */
         }
    } while (bytes_read > 0);
    

    And if you don't want to sit there waiting for data, you should look into select or epoll to determine when data is ready to be read or not, and the O_NONBLOCK flag for sockets is very handy if you want to ensure you never block on a recv.