Uzyskaj liczbę bajtów dostępnych w gnieździe przez "recv" z "MSG_PEEK" w C++

Question

C++ ma następującą funkcję do odbierania bajtów z gniazda, może sprawdzić liczbę bajtów dostępnych z flagą MSG_PEEK. Z MSG_PEEK, zwracana wartość „recv” jest liczba bajtów dostępnych w gnieździe:

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

Potrzebuję uzyskać liczbę bajtów dostępnych w gnieździe bez tworzenia buffer (bez przydzielania pamięci dla buffer). Czy to możliwe i jak?

Answer 1

Poszukujesz: ioctl(fd,FIONREAD,&bytes_available) i pod Windows ioctlsocket(socket,FIONREAD,&bytes_available).

Należy pamiętać, że system operacyjny niekoniecznie gwarantuje ilość buforowanych danych, więc jeśli czekasz na bardzo dużo danych, lepiej będzie odczytać dane w miarę ich dostarczania i przechowywania. w swoim własnym buforze, dopóki nie będziesz mieć wszystkiego, czego potrzebujesz do przetworzenia czegoś.

Aby to zrobić, co jest zwykle wykonywane jest po prostu czytać kawałki na raz, takie jak

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); 

A jeśli nie chcesz siedzieć tam czeka na dane, należy spojrzeć na select lub epoll, aby określić, kiedy dane są gotowe do odczytania, a flaga O_NONBLOCK dla gniazd jest bardzo przydatna, jeśli chcesz mieć pewność, że nigdy nie blokujesz recv.

Answer 2

W systemie Windows można użyć funkcji ioctlsocket() z flagą FIONREAD, aby zapytać gniazdo, ile bajtów jest dostępnych, bez konieczności odczytywania/przeglądania samych bajtów. Zwrócona wartość to minimalna liczba bajtów recv() może zostać zwrócona bez blokowania. Zanim zadzwonisz na numer recv(), może przybyć więcej bajtów.