network: added function to set socket receiving buffer size

Increasing socket receiving buffers can avoid losing datagram packets in case of a burst of incoming packets. The value is capped by /proc/sys/net/core/rmem_max on Linux (rmem_default is the default value if flb_net_socket_rcv_buffer is not called). Signed-off-by: Benoît GARNIER <[email protected]>
fluent · Nov 8, 2022 · 636e083 · 636e083
1 parent 4208622
commit 636e083
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/include/fluent-bit/flb_network.h b/include/fluent-bit/flb_network.h
@@ -143,6 +143,7 @@ int flb_net_socket_reset(flb_sockfd_t fd);
 int flb_net_socket_tcp_nodelay(flb_sockfd_t fd);
 int flb_net_socket_blocking(flb_sockfd_t fd);
 int flb_net_socket_nonblocking(flb_sockfd_t fd);
+int flb_net_socket_rcv_buffer(flb_sockfd_t fd, int rcvbuf);
 int flb_net_socket_tcp_fastopen(flb_sockfd_t sockfd);
 
 /* Socket handling */

diff --git a/src/flb_network.c b/src/flb_network.c
@@ -212,6 +212,16 @@ int flb_net_socket_nonblocking(flb_sockfd_t fd)
     return 0;
 }
 
+int flb_net_socket_rcv_buffer(flb_sockfd_t fd, int rcvbuf)
+{
+    if (setsockopt(fd, SOL_SOCKET, SO_RCVBUF, &rcvbuf, sizeof(rcvbuf)) != 0) {
+        flb_errno();
+        return -1;
+    }
+
+    return 0;
+}
+
 int flb_net_socket_blocking(flb_sockfd_t fd)
 {
 #ifdef _WIN32