Enable gVisor tcp_socket_test

kernel/src/net/socket/ip/stream/util.rs

@@ -1,5 +1,7 @@
 // SPDX-License-Identifier: MPL-2.0
 
+use aster_bigtcp::time::Duration;
+
 use crate::prelude::*;
 
 #[derive(Debug, Clone, Copy, CopyGetters, Setters)]
@@ -9,12 +11,17 @@ pub struct TcpOptionSet {
     no_delay: bool,
     maxseg: u32,
     keep_idle: u32,
+    syn_cnt: u8,
+    defer_accept: Retrans,
     window_clamp: u32,
     congestion: CongestionControl,
+    user_timeout: u32,
+    receive_inq: bool,
 }
 
 pub const DEFAULT_MAXSEG: u32 = 536;
 pub const DEFAULT_KEEP_IDLE: u32 = 7200;
+pub const DEFAULT_SYN_CNT: u8 = 6;
 pub const DEFAULT_WINDOW_CLAMP: u32 = 0x8000_0000;
 
 impl TcpOptionSet {
@@ -23,8 +30,12 @@ impl TcpOptionSet {
             no_delay: false,
             maxseg: DEFAULT_MAXSEG,
             keep_idle: DEFAULT_KEEP_IDLE,
+            syn_cnt: DEFAULT_SYN_CNT,
+            defer_accept: Retrans(0),
             window_clamp: DEFAULT_WINDOW_CLAMP,
             congestion: CongestionControl::Reno,
+            user_timeout: 0,
+            receive_inq: false,
         }
     }
 }
@@ -35,6 +46,63 @@ impl Default for TcpOptionSet {
     }
 }
 
+/// Initial RTO value
+const TCP_TIMEOUT_INIT: Duration = Duration::from_secs(1);
+const TCP_RTO_MAX: Duration = Duration::from_secs(120);
+
+/// The number of retransmits.
+#[derive(Debug, Clone, Copy)]
+pub struct Retrans(u8);
+
+impl Retrans {
+    /// Converts seconds to retransmits.
+    pub const fn from_secs(seconds: u32) -> Self {
+        if seconds == 0 {
+            return Self(0);
+        }
+
+        let mut timeout = TCP_TIMEOUT_INIT.secs() as u32;
+        let rto_max = TCP_RTO_MAX.secs() as u32;
+        let mut period = timeout;
+        let mut res = 1;
+
+        while seconds > period && res < 255 {
+            res += 1;
+            timeout <<= 1;
+            if timeout > rto_max {
+                timeout = rto_max;
+            }
+            period += timeout;
+        }
+
+        Self(res)
+    }
+
+    /// Converts retransmits to seconds.
+    pub const fn to_secs(self) -> u32 {
+        let mut retrans = self.0;
+
+        if retrans == 0 {
+            return 0;
+        }
+
+        let mut timeout = TCP_TIMEOUT_INIT.secs() as u32;
+        let rto_max = TCP_RTO_MAX.secs() as u32;
+        let mut period = timeout;
+
+        while retrans > 1 {
+            retrans -= 1;
+            timeout <<= 1;
+            if timeout > rto_max {
+                timeout = rto_max;
+            }
+            period += timeout;
+        }
+
+        period
+    }
+}
+
 #[derive(Debug, Clone, Copy)]
 pub enum CongestionControl {
     Reno,
@@ -49,7 +117,7 @@ impl CongestionControl {
         let congestion = match name {
             Self::RENO => Self::Reno,
             Self::CUBIC => Self::Cubic,
-            _ => return_errno_with_message!(Errno::EINVAL, "unsupported congestion name"),
+            _ => return_errno_with_message!(Errno::ENOENT, "unsupported congestion name"),
         };
 
         Ok(congestion)