fix: fix TCP_FASTOPEN on macos (#165)

4 files changed, 29 insertions, 2 deletions

diff --git a/monoio/src/driver/legacy/mod.rs b/monoio/src/driver/legacy/mod.rs
index 98dd3a1..3bcbaac 100644
--- a/monoio/src/driver/legacy/mod.rs
+++ b/monoio/src/driver/legacy/mod.rs
@@ -24,7 +24,7 @@ mod waker;
 pub(crate) use waker::UnparkHandle;
 
 pub(crate) struct LegacyInner {
-    io_dispatch: Slab<ScheduledIo>,
+    pub(crate) io_dispatch: Slab<ScheduledIo>,
     events: Option<mio::Events>,
     poll: mio::Poll,
 
diff --git a/monoio/src/driver/legacy/scheduled_io.rs b/monoio/src/driver/legacy/scheduled_io.rs
index 16a91dc..ce6ab94 100644
--- a/monoio/src/driver/legacy/scheduled_io.rs
+++ b/monoio/src/driver/legacy/scheduled_io.rs
@@ -22,6 +22,11 @@ impl Default for ScheduledIo {
 }
 
 impl ScheduledIo {
+    #[allow(unused)]
+    pub(crate) fn set_writable(&mut self) {
+        self.readiness |= Ready::WRITABLE;
+    }
+
     pub(crate) fn set_readiness(&mut self, f: impl Fn(Ready) -> Ready) {
         self.readiness = f(self.readiness);
     }
diff --git a/monoio/src/net/tcp/listener.rs b/monoio/src/net/tcp/listener.rs
index 75f2a34..476e679 100644
--- a/monoio/src/net/tcp/listener.rs
+++ b/monoio/src/net/tcp/listener.rs
@@ -74,11 +74,16 @@ impl TcpListener {
             #[cfg(any(target_os = "linux", target_os = "android"))]
             super::tfo::set_tcp_fastopen(&sys_listener, opts.backlog)?;
             #[cfg(any(target_os = "ios", target_os = "macos"))]
-            super::tfo::set_tcp_fastopen(&sys_listener)?;
+            let _ = super::tfo::set_tcp_fastopen_force_enable(&sys_listener);
         }
         sys_listener.bind(&addr)?;
         sys_listener.listen(opts.backlog)?;
 
+        #[cfg(any(target_os = "ios", target_os = "macos"))]
+        if opts.tcp_fast_open {
+            super::tfo::set_tcp_fastopen(&sys_listener)?;
+        }
+
         #[cfg(unix)]
         let fd = SharedFd::new(sys_listener.into_raw_fd())?;
 
diff --git a/monoio/src/net/tcp/stream.rs b/monoio/src/net/tcp/stream.rs
index 52e0746..359fc83 100644
--- a/monoio/src/net/tcp/stream.rs
+++ b/monoio/src/net/tcp/stream.rs
@@ -136,6 +136,23 @@ impl TcpStream {
         let stream = TcpStream::from_shared_fd(completion.data.fd);
         // wait write ready on epoll branch
         if crate::driver::op::is_legacy() {
+            #[cfg(any(target_os = "ios", target_os = "macos"))]
+            if !tfo {
+                stream.writable(true).await?;
+            } else {
+                // set writable as init state
+                crate::driver::CURRENT.with(|inner| match inner {
+                    crate::driver::Inner::Legacy(inner) => {
+                        let idx = stream.fd.registered_index().unwrap();
+                        if let Some(mut readiness) =
+                            unsafe { &mut *inner.get() }.io_dispatch.get(idx)
+                        {
+                            readiness.set_writable();
+                        }
+                    }
+                })
+            }
+            #[cfg(not(any(target_os = "ios", target_os = "macos")))]
             stream.writable(true).await?;
 
             // getsockopt libc::SO_ERROR