feat: add dont_frag option

Signed-off-by: loongtao.zhang <[email protected]>

Author: cathaysia

src/socket.rs

@@ -1615,6 +1615,53 @@ impl Socket {
                 .map(|recv_tos| recv_tos > 0)
         }
     }
+
+    /// Set value for `IP_DONTFRAG` option of this socket
+    #[cfg(any(
+        target_os = "macos",
+        target_os = "ios",
+        target_os = "linux",
+        target_os = "windows",
+    ))]
+    pub fn dont_frag(&self) -> io::Result<bool> {
+        unsafe {
+            #[cfg(any(target_os = "macos", target_os = "ios",))]
+            return getsockopt::<c_int>(self.as_raw(), sys::IPPROTO_IP, libc::IP_DONTFRAG)
+                .map(|dont_frag| dont_frag > 0);
+            #[cfg(any(target_os = "linux", target_os = "windows",))]
+            return getsockopt::<c_int>(self.as_raw(), sys::IPPROTO_IP, sys::IP_MTU_DISCOVER)
+                .map(|dont_frag| dont_frag == 0);
+        }
+    }
+    /// Set value for `IP_DONTFRAG` option on this socket
+    #[cfg(any(
+        target_os = "macos",
+        target_os = "ios",
+        target_os = "linux",
+        target_os = "windows",
+    ))]
+    pub fn set_dont_frag(&self, dont_frag: bool) -> io::Result<()> {
+        unsafe {
+            #[cfg(any(target_os = "macos", target_os = "ios",))]
+            return setsockopt(
+                self.as_raw(),
+                sys::IPPROTO_IP,
+                libc::IP_DONTFRAG,
+                dont_frag as c_int,
+            );
+            #[cfg(any(target_os = "linux", target_os = "windows",))]
+            return setsockopt(
+                self.as_raw(),
+                sys::IPPROTO_IP,
+                sys::IP_MTU_DISCOVER,
+                if dont_frag {
+                    sys::IP_PMTUDISC_DONT
+                } else {
+                    sys::IP_PMTUDISC_DO
+                },
+            );
+        };
+    }
 }
 
 /// Socket options for IPv6 sockets, get/set using `IPPROTO_IPV6`.
@@ -2015,6 +2062,54 @@ impl Socket {
             )
         }
     }
+
+    /// Get value for `IP_DONTFRAG` option on this socket.
+    #[cfg(any(
+        target_os = "linux",
+        target_os = "windows",
+        target_os = "macos",
+        target_os = "ios",
+    ))]
+    pub fn dont_frag_v6(&self) -> io::Result<bool> {
+        unsafe {
+            #[cfg(any(target_os = "macos", target_os = "ios",))]
+            return getsockopt::<c_int>(self.as_raw(), sys::IPPROTO_IPV6, libc::IPV6_DONTFRAG)
+                .map(|dont_frag| dont_frag > 0);
+            #[cfg(any(target_os = "linux", target_os = "windows",))]
+            return getsockopt::<c_int>(self.as_raw(), sys::IPPROTO_IPV6, sys::IPV6_MTU_DISCOVER)
+                .map(|dont_frag| dont_frag == sys::IP_PMTUDISC_DONT);
+        }
+    }
+
+    /// Set value for `IP_DONTFRAG` option on this socket
+    #[cfg(any(
+        target_os = "macos",
+        target_os = "ios",
+        target_os = "linux",
+        target_os = "windows",
+    ))]
+    pub fn set_dont_frag_v6(&self, dont_frag: bool) -> io::Result<()> {
+        unsafe {
+            #[cfg(any(target_os = "macos", target_os = "ios",))]
+            return setsockopt(
+                self.as_raw(),
+                sys::IPPROTO_IPV6,
+                libc::IPV6_DONTFRAG,
+                dont_frag as c_int,
+            );
+            #[cfg(any(target_os = "linux", target_os = "windows",))]
+            setsockopt(
+                self.as_raw(),
+                sys::IPPROTO_IPV6,
+                sys::IPV6_MTU_DISCOVER,
+                if dont_frag {
+                    sys::IP_PMTUDISC_DONT
+                } else {
+                    sys::IP_PMTUDISC_DO
+                },
+            )
+        }
+    }
 }
 
 /// Socket options for TCP sockets, get/set using `IPPROTO_TCP`.

src/sys/unix.rs

@@ -204,6 +204,10 @@ pub(crate) use libc::{
     SO_BROADCAST, SO_ERROR, SO_KEEPALIVE, SO_RCVBUF, SO_RCVTIMEO, SO_REUSEADDR, SO_SNDBUF,
     SO_SNDTIMEO, SO_TYPE, TCP_NODELAY,
 };
+
+#[cfg(target_os = "linux")]
+pub(crate) use libc::{IPV6_MTU_DISCOVER, IP_MTU_DISCOVER, IP_PMTUDISC_DO, IP_PMTUDISC_DONT};
+
 #[cfg(not(any(
     target_os = "dragonfly",
     target_os = "haiku",

src/sys/windows.rs

@@ -84,6 +84,14 @@ pub(crate) use windows_sys::Win32::Networking::WinSock::{
 pub(crate) const IPPROTO_IP: c_int = windows_sys::Win32::Networking::WinSock::IPPROTO_IP as c_int;
 pub(crate) const SOL_SOCKET: c_int = windows_sys::Win32::Networking::WinSock::SOL_SOCKET as c_int;
 
+pub(crate) const IPV6_MTU_DISCOVER: c_int =
+    windows_sys::Win32::Networking::WinSock::IPV6_MTU_DISCOVER as c_int;
+pub(crate) const IP_MTU_DISCOVER: c_int =
+    windows_sys::Win32::Networking::WinSock::IP_MTU_DISCOVER as c_int;
+pub(crate) const IP_PMTUDISC_DO: c_int =
+    windows_sys::Win32::Networking::WinSock::IP_PMTUDISC_DO as c_int;
+pub(crate) const IP_PMTUDISC_DONT: c_int =
+    windows_sys::Win32::Networking::WinSock::IP_PMTUDISC_DONT as c_int;
 /// Type used in set/getsockopt to retrieve the `TCP_NODELAY` option.
 ///
 /// NOTE: <https://docs.microsoft.com/en-us/windows/win32/api/winsock2/nf-winsock2-getsockopt>

tests/socket.rs

@@ -1350,6 +1350,18 @@ test!(
 #[cfg(not(target_os = "redox"))]
 test!(out_of_band_inline, set_out_of_band_inline(true));
 test!(reuse_address, set_reuse_address(true));
+
+#[cfg(any(
+    target_os = "macos",
+    target_os = "ios",
+    target_os = "linux",
+    target_os = "windows",
+))]
+test!(IPv4 dont_frag, set_dont_frag(cfg!(not(target_os = "windows"))));
+// TODO: macos can only set on udp socket.
+#[cfg(any(target_os = "linux", target_os = "windows",))]
+test!(IPv6 dont_frag_v6, set_dont_frag_v6(cfg!(not(target_os = "freebsd"))));
+
 #[cfg(all(
     feature = "all",
     not(any(windows, target_os = "solaris", target_os = "illumos"))