Implement hdrincl for raw socket

kernel/src/net/structs.rs

@@ -31,6 +31,10 @@ pub trait Socket: Send + Sync {
     fn remote_endpoint(&self) -> Option<IpEndpoint> {
         None
     }
+    fn setsockopt(&mut self, level: usize, opt: usize, data: &[u8]) -> SysResult {
+        warn!("setsockopt is unimplemented");
+        Ok(0)
+    }
     fn box_clone(&self) -> Box<dyn Socket>;
 }
 
@@ -65,6 +69,7 @@ pub struct UdpSocketState {
 #[derive(Debug, Clone)]
 pub struct RawSocketState {
     handle: GlobalSocketHandle,
+    header_included: bool,
 }
 
 /// A wrapper for `SocketHandle`.
@@ -469,7 +474,10 @@ impl RawSocketState {
         );
         let handle = GlobalSocketHandle(SOCKETS.lock().add(socket));
 
-        RawSocketState { handle }
+        RawSocketState {
+            handle,
+            header_included: false,
+        }
     }
 }
 
@@ -499,6 +507,15 @@ impl Socket for RawSocketState {
     }
 
     fn write(&self, data: &[u8], sendto_endpoint: Option<IpEndpoint>) -> SysResult {
+        if self.header_included {
+            let mut sockets = SOCKETS.lock();
+            let mut socket = sockets.get::<RawSocket>(self.handle.0);
+
+            match socket.send_slice(&data) {
+                Ok(()) => Ok(data.len()),
+                Err(_) => Err(SysError::ENOBUFS),
+            }
+        } else {
             if let Some(endpoint) = sendto_endpoint {
                 // temporary solution
                 let iface = &*(NET_DRIVERS.read()[0]);
@@ -536,6 +553,7 @@ impl Socket for RawSocketState {
                 Err(SysError::ENOTCONN)
             }
         }
+    }
 
     fn poll(&self) -> (bool, bool, bool) {
         unimplemented!()
@@ -548,6 +566,19 @@ impl Socket for RawSocketState {
     fn box_clone(&self) -> Box<dyn Socket> {
         Box::new(self.clone())
     }
+
+    fn setsockopt(&mut self, level: usize, opt: usize, data: &[u8]) -> SysResult {
+        match (level, opt) {
+            (IPPROTO_IP, IP_HDRINCL) => {
+                if let Some(arg) = data.first() {
+                    self.header_included = *arg > 0;
+                    debug!("hdrincl set to {}", self.header_included);
+                }
+            }
+            _ => {}
+        }
+        Ok(0)
+    }
 }
 
 fn get_ephemeral_port() -> u16 {

kernel/src/syscall/net.rs

@@ -34,15 +34,18 @@ pub fn sys_setsockopt(
     fd: usize,
     level: usize,
     optname: usize,
-    _optval: *const u8,
-    _optlen: usize,
+    optval: *const u8,
+    optlen: usize,
 ) -> SysResult {
     info!(
         "setsockopt: fd: {}, level: {}, optname: {}",
         fd, level, optname
     );
-    warn!("sys_setsockopt is unimplemented");
-    Ok(0)
+    let mut proc = process();
+    proc.vm.check_read_array(optval, optlen)?;
+    let data = unsafe { slice::from_raw_parts(optval, optlen) };
+    let socket = proc.get_socket(fd)?;
+    socket.setsockopt(level, optname, data)
 }
 
 pub fn sys_getsockopt(
@@ -387,3 +390,5 @@ const SO_RCVBUF: usize = 8;
 const SO_LINGER: usize = 13;
 
 const TCP_CONGESTION: usize = 13;
+
+const IP_HDRINCL: usize = 3;