feat(net): add set_socket_option on Socket

Author: AsakuraMizu

compio-net/src/socket.rs

@@ -10,7 +10,7 @@ use compio_driver::{
         RecvResultExt, RecvVectored, Send, SendMsg, SendTo, SendToVectored, SendVectored,
         ShutdownSocket,
     },
-    ToSharedFd,
+    syscall, AsRawFd, ToSharedFd,
 };
 use compio_runtime::Attacher;
 use socket2::{Domain, Protocol, SockAddr, Socket as Socket2, Type};
@@ -324,6 +324,33 @@ impl Socket {
         let op = SendMsg::new(fd, buffer, control, addr.clone());
         compio_runtime::submit(op).await.into_inner()
     }
+
+    #[cfg(unix)]
+    pub fn set_socket_option<T>(&self, level: i32, name: i32, value: &T) -> io::Result<()> {
+        syscall!(libc::setsockopt(
+            self.socket.as_raw_fd(),
+            level,
+            name,
+            value as *const _ as _,
+            std::mem::size_of::<T>() as _
+        ))
+        .map(|_| ())
+    }
+
+    #[cfg(windows)]
+    pub fn set_socket_option<T>(&self, level: i32, name: i32, value: &T) -> io::Result<()> {
+        syscall!(
+            SOCKET,
+            windows_sys::Win32::Networking::WinSock::setsockopt(
+                self.socket.as_raw_fd() as _,
+                level,
+                name,
+                value as *const _ as _,
+                std::mem::size_of::<T>() as _
+            )
+        )
+        .map(|_| ())
+    }
 }
 
 impl_raw_fd!(Socket, Socket2, socket, socket);

compio-net/src/udp.rs

@@ -315,6 +315,11 @@ impl UdpSocket {
         )
         .await
     }
+
+    /// Sets a socket option.
+    pub fn set_socket_option<T>(&self, level: i32, name: i32, value: &T) -> io::Result<()> {
+        self.inner.set_socket_option(level, name, value)
+    }
 }
 
 impl_raw_fd!(UdpSocket, socket2::Socket, inner, socket);