From 8ba1e326f6edf4c87a0e500aa3d3936bbd0e4c32 Mon Sep 17 00:00:00 2001
From: Skyxim <no-reply@skyxim.dev>
Date: Sun, 21 Jan 2024 18:01:24 +0800
Subject: [PATCH] fix: synchronize SYN packet and Socket creation

---
 .../shadowsocks-service/src/local/tun/tcp.rs  | 22 ++++++++++++++-----
 1 file changed, 16 insertions(+), 6 deletions(-)

diff --git a/crates/shadowsocks-service/src/local/tun/tcp.rs b/crates/shadowsocks-service/src/local/tun/tcp.rs
index f0b121dd1d61..5296f073fb8c 100644
--- a/crates/shadowsocks-service/src/local/tun/tcp.rs
+++ b/crates/shadowsocks-service/src/local/tun/tcp.rs
@@ -26,7 +26,7 @@ use smoltcp::{
 use spin::Mutex as SpinMutex;
 use tokio::{
     io::{AsyncRead, AsyncWrite, ReadBuf},
-    sync::mpsc,
+    sync::{mpsc, oneshot},
 };
 
 use crate::{
@@ -88,6 +88,7 @@ type SharedTcpConnectionControl = Arc<SpinMutex<TcpSocketControl>>;
 struct TcpSocketCreation {
     control: SharedTcpConnectionControl,
     socket: TcpSocket<'static>,
+    socket_created_tx: oneshot::Sender<()>,
 }
 
 struct TcpConnection {
@@ -112,7 +113,7 @@ impl Drop for TcpConnection {
 }
 
 impl TcpConnection {
-    fn new(
+    async fn new(
         socket: TcpSocket<'static>,
         socket_creation_tx: &mpsc::UnboundedSender<TcpSocketCreation>,
         manager_notify: Arc<ManagerNotify>,
@@ -129,12 +130,14 @@ impl TcpConnection {
             recv_state: TcpSocketState::Normal,
             send_state: TcpSocketState::Normal,
         }));
-
+        let (tx, rx) = oneshot::channel();
         let _ = socket_creation_tx.send(TcpSocketCreation {
             control: control.clone(),
             socket,
+            socket_created_tx: tx,
         });
-
+        // waiting socket add to SocketSet
+        let _ = rx.await;
         TcpConnection {
             control,
             manager_notify,
@@ -304,8 +307,14 @@ impl TcpTun {
                 let mut socket_set = SocketSet::new(vec![]);
 
                 while manager_running.load(Ordering::Relaxed) {
-                    while let Ok(TcpSocketCreation { control, socket }) = socket_creation_rx.try_recv() {
+                    while let Ok(TcpSocketCreation {
+                        control,
+                        socket,
+                        socket_created_tx: socket_create_tx,
+                    }) = socket_creation_rx.try_recv()
+                    {
                         let handle = socket_set.add(socket);
+                        let _ = socket_create_tx.send(());
                         sockets.insert(handle, control);
                     }
 
@@ -512,7 +521,8 @@ impl TcpTun {
                 &self.manager_socket_creation_tx,
                 self.manager_notify.clone(),
                 &accept_opts.tcp,
-            );
+            )
+            .await;
 
             // establish a tunnel
             let context = self.context.clone();