From f1c922077bc37b5cb0f1a9998d262fcda434b5f5 Mon Sep 17 00:00:00 2001
From: database64128 <free122448@hotmail.com>
Date: Sun, 5 Mar 2023 17:25:11 +0800
Subject: [PATCH] =?UTF-8?q?=F0=9F=94=AB=20macos:=20set=20TCP=5FFASTOPEN=5F?=
 =?UTF-8?q?FORCE=5FENABLE?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

This disables the absolutely brutal TFO backoff mechanism, so that TFO can actually work.
---
 src/listener.rs           | 21 +++++++++++++++++++++
 src/sys/unix/bsd/macos.rs | 26 +++++++++++++++++++++++++-
 2 files changed, 46 insertions(+), 1 deletion(-)

diff --git a/src/listener.rs b/src/listener.rs
index 472a03d..1b3237b 100644
--- a/src/listener.rs
+++ b/src/listener.rs
@@ -6,10 +6,15 @@ use std::{
     task::{Context, Poll},
 };
 
+use cfg_if::cfg_if;
 use futures::{future, ready};
+#[cfg(any(target_os = "macos", target_os = "ios", target_os = "watchos", target_os = "tvos"))]
+use log::debug;
 use tokio::net::{TcpListener as TokioTcpListener, TcpSocket};
 
 use crate::{stream::TfoStream, sys::set_tcp_fastopen};
+#[cfg(any(target_os = "macos", target_os = "ios", target_os = "watchos", target_os = "tvos"))]
+use crate::sys::set_tcp_fastopen_force_enable;
 
 /// TCP listener with TFO enabled
 pub struct TfoListener {
@@ -36,11 +41,27 @@ impl TfoListener {
         #[cfg(not(windows))]
         socket.set_reuseaddr(true)?;
 
+        // On all other platforms, TCP_FASTOPEN can be set before bind(), between bind() and listen(), or after listen().
+        // We prefer setting it before bind() as this feels most like the natural order of socket initialization sequence.
+        //
+        // On macOS, setting TCP_FASTOPEN_FORCE_ENABLE requires the socket to be in the TCPS_CLOSED state.
+        // TCP_FASTOPEN, on the other hand, can only be set when the socket is in the TCPS_LISTEN state.
+        cfg_if! {
+            if #[cfg(not(any(target_os = "macos", target_os = "ios", target_os = "watchos", target_os = "tvos")))] {
+                set_tcp_fastopen(&socket)?;
+            } else {
+                if let Err(err) = set_tcp_fastopen_force_enable(&socket) {
+                    debug!("failed to set TCP_FASTOPEN_FORCE_ENABLE: {:?}", err);
+                }
+            }
+        }
+
         socket.bind(addr)?;
 
         // mio's default backlog is 1024
         let inner = socket.listen(1024)?;
 
+        #[cfg(any(target_os = "macos", target_os = "ios", target_os = "watchos", target_os = "tvos"))]
         set_tcp_fastopen(&inner)?;
 
         Ok(TfoListener { inner })
diff --git a/src/sys/unix/bsd/macos.rs b/src/sys/unix/bsd/macos.rs
index feaf37d..67d186a 100644
--- a/src/sys/unix/bsd/macos.rs
+++ b/src/sys/unix/bsd/macos.rs
@@ -45,7 +45,7 @@ impl TcpStream {
     }
 
     pub async fn connect_with_socket(socket: TcpSocket, addr: SocketAddr) -> io::Result<TcpStream> {
-        // TFO in macos uses connectx
+        set_tcp_fastopen_force_enable(&socket)?;
 
         unsafe {
             let raddr = SockAddr::from(addr);
@@ -205,6 +205,30 @@ impl AsRawFd for TcpStream {
     }
 }
 
+/// Disables the absolutely brutal TFO backoff mechanism on macOS.
+pub fn set_tcp_fastopen_force_enable<S: AsRawFd>(socket: &S) -> io::Result<()> {
+    const TCP_FASTOPEN_FORCE_ENABLE: libc::c_int = 0x218;
+    let enable: libc::c_int = 1;
+
+    unsafe {
+        let ret = libc::setsockopt(
+            socket.as_raw_fd(),
+            libc::IPPROTO_TCP,
+            TCP_FASTOPEN_FORCE_ENABLE,
+            &enable as *const _ as *const libc::c_void,
+            mem::size_of_val(&enable) as libc::socklen_t,
+        );
+
+        if ret != 0 {
+            let err = io::Error::last_os_error();
+            error!("set TCP_FASTOPEN_FORCE_ENABLE error: {}", err);
+            return Err(err);
+        }
+    }
+
+    Ok(())
+}
+
 /// Enable `TCP_FASTOPEN`
 ///
 /// `TCP_FASTOPEN` was supported since