feat(s2n-quic-platform): make GSO/GRO configuration more ergonomic

quic/s2n-quic-platform/src/io/tokio/builder.rs

@@ -122,6 +122,18 @@ impl Builder {
         Ok(self)
     }
 
+    /// Configures Generic Segmentation Offload (GSO)
+    ///
+    /// By default, GSO will be used unless the platform does not support it or an attempt to use
+    /// GSO fails. If it is known that GSO is not available, set this option to explicitly disable it.
+    pub fn with_gso(self, enabled: bool) -> io::Result<Self> {
+        if enabled {
+            Ok(self)
+        } else {
+            self.with_gso_disabled()
+        }
+    }
+
     /// Disables Generic Receive Offload (GRO)
     ///
     /// By default, GRO will be used unless the platform does not support it. If it is known that
@@ -131,6 +143,18 @@ impl Builder {
         Ok(self)
     }
 
+    /// Configures Generic Receive Offload (GRO)
+    ///
+    /// By default, GRO will be used unless the platform does not support it. If it is known that
+    /// GRO is not available, set this option to explicitly disable it.
+    pub fn with_gro(self, enabled: bool) -> io::Result<Self> {
+        if enabled {
+            Ok(self)
+        } else {
+            self.with_gro_disabled()
+        }
+    }
+
     /// Enables the port reuse (SO_REUSEPORT) socket option
     pub fn with_reuse_port(mut self) -> io::Result<Self> {
         if !cfg!(unix) {

quic/s2n-quic-qns/src/io.rs

@@ -46,11 +46,8 @@ impl Server {
     pub fn build(&self) -> Result<impl io::Provider> {
         let mut io_builder = io::Default::builder()
             .with_receive_address((self.ip, self.port).into())?
-            .with_max_mtu(self.max_mtu)?;
-
-        if self.disable_gso {
-            io_builder = io_builder.with_gso_disabled()?;
-        }
+            .with_max_mtu(self.max_mtu)?
+            .with_gso(!self.disable_gso)?;
 
         if let Some(size) = self.queue_send_buffer_size {
             io_builder = io_builder.with_internal_send_buffer_size(size)?;
@@ -102,11 +99,8 @@ impl Client {
     pub fn build(&self) -> Result<impl io::Provider> {
         let mut io_builder = io::Default::builder()
             .with_receive_address((self.local_ip, 0u16).into())?
-            .with_max_mtu(self.max_mtu)?;
-
-        if self.disable_gso {
-            io_builder = io_builder.with_gso_disabled()?;
-        }
+            .with_max_mtu(self.max_mtu)?
+            .with_gso(!self.disable_gso)?;
 
         if let Some(size) = self.queue_send_buffer_size {
             io_builder = io_builder.with_internal_send_buffer_size(size)?;