try hook connect

Author: loongs-zhang

core/src/net/event_loop.rs

@@ -123,7 +123,7 @@ impl<'e> EventLoop<'e> {
             let syscall_mask = <Syscall as Into<&str>>::into(syscall).as_ptr() as usize;
             let token = thread_id as usize ^ syscall_mask;
             if Syscall::nio() != syscall {
-                eprintln!("{syscall} {token}");
+                eprintln!("generate token:{token} for {syscall}");
             }
             token
         }
@@ -255,8 +255,7 @@ impl<'e> EventLoop<'e> {
                         continue;
                     }
                     // resolve completed read/write tasks
-                    let result = cqe.result() as c_longlong;
-                    eprintln!("io_uring finish {token} {result}");
+                    let result = c_longlong::from(cqe.result());
                     if let Some((_, pair)) = self.syscall_wait_table.remove(&token) {
                         let (lock, cvar) = &*pair;
                         let mut pending = lock.lock().expect("lock failed");

core/src/net/mod.rs

@@ -23,7 +23,11 @@ pub type UserFunc = extern "C" fn(usize) -> usize;
 
 mod selector;
 
-#[allow(clippy::cast_possible_truncation, clippy::cast_sign_loss)]
+#[allow(
+    clippy::cast_possible_truncation,
+    clippy::cast_sign_loss,
+    clippy::too_many_arguments
+)]
 #[cfg(all(target_os = "linux", feature = "io_uring"))]
 mod operator;
 

core/src/net/operator/mod.rs

@@ -1,4 +1,3 @@
-#[allow(clippy::too_many_arguments)]
 #[cfg(all(target_os = "linux", feature = "io_uring"))]
 mod linux;
 #[cfg(all(target_os = "linux", feature = "io_uring"))]

hook/src/syscall/windows.rs

@@ -149,11 +149,11 @@ unsafe fn attach() -> std::io::Result<()> {
     //     addresssize: usize,
     //     dwmilliseconds: c_uint
     // ) -> BOOL);
-    // impl_hook!("ws2_32.dll", CONNECT, connect(
-    //     fd: SOCKET,
-    //     address: *const SOCKADDR,
-    //     len: c_int
-    // ) -> c_int);
+    impl_hook!("ws2_32.dll", CONNECT, connect(
+        fd: SOCKET,
+        address: *const SOCKADDR,
+        len: c_int
+    ) -> c_int);
     // Enable the hook
     minhook::MinHook::enable_all_hooks()
         .map_err(|_| Error::new(ErrorKind::Other, "init all hooks failed !"))