Set both rcvbuf and sndbuf for accepted and peer clients

IABTechLab · Dec 5, 2024 · f7dced6 · f7dced6
1 parent c326fdd
commit f7dced6
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/vsock-bridge/include/listener.h b/vsock-bridge/include/listener.h
@@ -183,6 +183,12 @@ namespace vsockio
                 throw std::runtime_error("error setting SO_RCVBUF");
             }
 
+            if (setsockopt(clientFd, SOL_SOCKET, SO_SNDBUF, &_sndbuf, sizeof(int)) < 0)
+            {
+                close(clientFd);
+                throw std::runtime_error("error setting SO_SNDBUF");
+            }
+
             auto outPeer = connectToPeer();
 			if (!outPeer)
 			{
@@ -218,6 +224,12 @@ namespace vsockio
                 return nullptr;
             }
 
+            if (setsockopt(fd, SOL_SOCKET, SO_RCVBUF, &_rcvbuf, sizeof(int)) < 0)
+            {
+                close(fd);
+                throw std::runtime_error("error setting SO_RCVBUF");
+            }
+
             if (setsockopt(fd, SOL_SOCKET, SO_SNDBUF, &_sndbuf, sizeof(int)) < 0)
             {
                 close(fd);