Pass in buf value

IABTechLab · Dec 5, 2024 · 15a4f10 · 15a4f10
1 parent eaf7274
commit 15a4f10
Show file tree

Hide file tree

Showing 4 changed files with 31 additions and 4 deletions.
diff --git a/vsock-bridge/include/config.h b/vsock-bridge/include/config.h
@@ -32,6 +32,10 @@ namespace vsockproxy
 		ServiceType _type = ServiceType::UNKNOWN;
 		EndpointConfig _listenEndpoint;
 		EndpointConfig _connectEndpoint;
+		int _acceptRcvBuf;
+		int _acceptSndBuf;
+		int _peerRcvBuf;
+		int _peerSndBuf;
 	};
 
 	std::vector<ServiceDescription> loadConfig(const std::string& filepath);

diff --git a/vsock-bridge/include/listener.h b/vsock-bridge/include/listener.h
@@ -70,8 +70,12 @@ namespace vsockio
         const int MAX_POLLER_EVENTS = 256;
         const int SO_BACKLOG = 64;
 
-        Listener(std::unique_ptr<Endpoint>&& listenEndpoint, std::unique_ptr<Endpoint>&& connectEndpoint, Dispatcher& dispatcher)
+        Listener(std::unique_ptr<Endpoint>&& listenEndpoint, std::unique_ptr<Endpoint>&& connectEndpoint, Dispatcher& dispatcher, int acceptRcvBuf, int acceptSndBuf, int peerRcvBuf, int peerSndBuf)
             : _fd(-1)
+            , _acceptRcvBuf(acceptRcvBuf)
+            , _acceptSndBuf(acceptSndBuf)
+            , _peerRcvBuf(peerRcvBuf)
+            , _peerSndBuf(peerSndBuf)
             , _listenEp(std::move(listenEndpoint))
             , _connectEp(std::move(connectEndpoint))
             , _events(new VsbEvent[MAX_POLLER_EVENTS])

diff --git a/vsock-bridge/src/config.cpp b/vsock-bridge/src/config.cpp
@@ -251,6 +251,22 @@ namespace vsockproxy
                         }
                         cs._connectEndpoint = *endpoint;
 					}
+					else if (line._key == "acceptRcvBuf")
+					{
+                        cs._acceptRcvBuf = std::stoi(line._value);
+					}
+					else if (line._key == "acceptSndBuf")
+					{
+                        cs._acceptSndBuf = std::stoi(line._value);
+					}
+					else if (line._key == "peerRcvBuf")
+					{
+                        cs._peerRcvBuf = std::stoi(line._value);
+					}
+					else if (line._key == "peerSndBuf")
+					{
+                        cs._peerSndBuf = std::stoi(line._value);
+					}
 				}
 			}
 		}

diff --git a/vsock-bridge/src/vsock-bridge.cpp b/vsock-bridge/src/vsock-bridge.cpp
@@ -27,7 +27,7 @@ static std::unique_ptr<Endpoint> createEndpoint(EndpointScheme scheme, const std
     }
 }
 
-static std::unique_ptr<Listener> createListener(Dispatcher& dispatcher, EndpointScheme inScheme, const std::string& inAddress, uint16_t inPort, EndpointScheme outScheme, const std::string& outAddress, uint16_t outPort)
+static std::unique_ptr<Listener> createListener(Dispatcher& dispatcher, EndpointScheme inScheme, const std::string& inAddress, uint16_t inPort, EndpointScheme outScheme, const std::string& outAddress, uint16_t outPort, int acceptRcvBuf, int acceptSndBuf, int peerRcvBuf, int peerSndBuf)
 {
     auto listenEp { createEndpoint(inScheme, inAddress, inPort) };
     auto connectEp{ createEndpoint(outScheme, outAddress, outPort) };
@@ -44,7 +44,7 @@ static std::unique_ptr<Listener> createListener(Dispatcher& dispatcher, Endpoint
     }
     else
     {
-        return std::make_unique<Listener>(std::move(listenEp), std::move(connectEp), dispatcher);
+        return std::make_unique<Listener>(std::move(listenEp), std::move(connectEp), dispatcher, acceptRcvBuf, acceptSndBuf, peerRcvBuf, peerSndBuf);
     }
 }
 
@@ -68,7 +68,10 @@ static void startServices(const std::vector<ServiceDescription>& services, int n
             /*inPort:*/     sd._listenEndpoint._port,
             /*outScheme:*/  sd._connectEndpoint._scheme,
             /*outAddress:*/ sd._connectEndpoint._address,
-            /*outPort:*/    sd._connectEndpoint._port
+                            _acceptRcvBuf,
+                            _acceptSndBuf,
+                            _peerRcvBuf,
+                            _peerSndBuf,
         );
 
         if (!listener)