Support for socket options to get/set TOS/TTL info

changelog/2203.added.md

@@ -0,0 +1,17 @@
+Added support for getting/setting IPv4 type-of-service (TOS) information via
+`nix::sys::socket::sockopt::IpRecvTos` and `nix::sys::socket::sockopt::IpTos`.
+
+Added support for getting/setting IPv6 traffic class information via
+`nix::sys::socket::sockopt::Ipv6RecvTClass` and
+`nix::sys::socket::sockopt::Ipv6TClass`.
+
+Added support for getting/setting IPv4 time-to-live (TTL) information via
+`nix::sys::socket::sockopt::IpRecvTtl` and `nix::sys::socket::sockopt::IpTtl`.
+
+Added support for getting/setting IPv6 hop limit information via
+`nix::sys::socket::sockopt::Ipv6RecvHopLimit` and
+`nix::sys::socket::sockopt::Ipv6HopLimit`.
+
+Added support for preventing IP fragmentation via
+`nix::sys::socket::sockopt::IpDontFrag` for IPv4 on Linux, Android and FreeBSD, and
+`nix::sys::socket::sockopt::Ipv6DontFrag` for IPv6 for FreeBSD.

src/sys/socket/mod.rs

@@ -38,10 +38,20 @@ pub use self::addr::{SockaddrLike, SockaddrStorage};
 pub use self::addr::{AddressFamily, UnixAddr};
 #[cfg(not(solarish))]
 pub use self::addr::{AddressFamily, UnixAddr};
-#[cfg(not(any(solarish, target_os = "haiku", target_os = "hurd", target_os = "redox")))]
+#[cfg(not(any(
+    solarish,
+    target_os = "haiku",
+    target_os = "hurd",
+    target_os = "redox"
+)))]
 #[cfg(feature = "net")]
 pub use self::addr::{LinkAddr, SockaddrIn, SockaddrIn6};
-#[cfg(any(solarish, target_os = "haiku", target_os = "hurd", target_os = "redox"))]
+#[cfg(any(
+    solarish,
+    target_os = "haiku",
+    target_os = "hurd",
+    target_os = "redox"
+))]
 #[cfg(feature = "net")]
 pub use self::addr::{SockaddrIn, SockaddrIn6};
 
@@ -803,6 +813,34 @@ pub enum ControlMessageOwned {
     #[cfg(any(target_os = "linux"))]
     TlsGetRecordType(TlsGetRecordType),
 
+    #[cfg(any(
+        apple_targets,
+        linux_android,
+        target_os = "freebsd",
+    ))]
+    #[cfg(feature = "net")]
+    IpTos(libc::c_int),
+
+    #[cfg(any(
+        apple_targets,
+        linux_android,
+        target_os = "freebsd",
+    ))]
+    #[cfg(feature = "net")]
+    Ipv6TClass(libc::c_int),
+
+    #[cfg(any(
+        apple_targets,
+        linux_android,
+        target_os = "freebsd",
+    ))]
+    #[cfg(feature = "net")]
+    IpTtl(libc::c_int),
+
+    #[cfg(any(linux_android, freebsdlike, apple_targets, target_os = "haiku"))]
+    #[cfg(feature = "net")]
+    Ipv6HopLimit(libc::c_int),
+
     /// Catch-all variant for unimplemented cmsg types.
     #[doc(hidden)]
     Unknown(UnknownCmsg),
@@ -982,6 +1020,53 @@ impl ControlMessageOwned {
                 let content_type = unsafe { ptr::read_unaligned(p as *const u8) };
                 ControlMessageOwned::TlsGetRecordType(content_type.into())
             },
+            #[cfg(any(
+                apple_targets,
+                target_os = "freebsd",
+            ))]
+            #[cfg(feature = "net")]
+            (libc::IPPROTO_IP, libc::IP_RECVTOS) => {
+                let tos = unsafe { ptr::read_unaligned(p as *const u8) };
+                ControlMessageOwned::IpTos(tos as libc::c_int)
+            },
+            #[cfg(linux_android)]
+            #[cfg(feature = "net")]
+            (libc::IPPROTO_IP, libc::IP_TOS) => {
+                let tos = unsafe { ptr::read_unaligned(p as *const u8) };
+                ControlMessageOwned::IpTos(tos as libc::c_int)
+            },
+            #[cfg(any(
+                apple_targets,
+                linux_android,
+                target_os = "freebsd",
+            ))]
+            #[cfg(feature = "net")]
+            (libc::IPPROTO_IPV6, libc::IPV6_TCLASS) => {
+                let tclass = unsafe { ptr::read_unaligned(p as *const libc::c_int) };
+                ControlMessageOwned::Ipv6TClass(tclass)
+            },
+            #[cfg(any(apple_targets))]
+            #[cfg(feature = "net")]
+            (libc::IPPROTO_IP, libc::IP_RECVTTL) => {
+                let ttl = unsafe { ptr::read_unaligned(p as *const libc::c_int) };
+                ControlMessageOwned::IpTtl(ttl)
+            },
+            #[cfg(linux_android)]
+            #[cfg(feature = "net")]
+            (libc::IPPROTO_IP, libc::IP_TTL) => {
+                let ttl = unsafe { ptr::read_unaligned(p as *const libc::c_int) };
+                ControlMessageOwned::IpTtl(ttl)
+            },
+            #[cfg(any(
+                apple_targets,
+                linux_android,
+                target_os = "freebsd",
+            ))]
+            #[cfg(feature = "net")]
+            (libc::IPPROTO_IPV6, libc::IPV6_HOPLIMIT) => {
+                let hop_limit = unsafe { ptr::read_unaligned(p as *const libc::c_int) };
+                ControlMessageOwned::Ipv6HopLimit(hop_limit)
+            },
             (_, _) => {
                 let sl = unsafe { std::slice::from_raw_parts(p, len) };
                 let ucmsg = UnknownCmsg(*header, Vec::<u8>::from(sl));
@@ -1147,6 +1232,30 @@ pub enum ControlMessage<'a> {
     /// page.
     #[cfg(target_os = "linux")]
     TxTime(&'a u64),
+
+    #[cfg(any(
+        apple_targets,
+        linux_android,
+        target_os = "freebsd",
+    ))]
+    #[cfg(feature = "net")]
+    IpTos(&'a libc::c_int),
+
+    #[cfg(any(
+        apple_targets,
+        linux_android,
+        target_os = "freebsd",
+    ))]
+    #[cfg(feature = "net")]
+    Ipv6TClass(&'a libc::c_int),
+
+    #[cfg(any(
+        apple_targets,
+        linux_android,
+        target_os = "freebsd",
+    ))]
+    #[cfg(feature = "net")]
+    IpTtl(&'a libc::c_int),
 }
 
 // An opaque structure used to prevent cmsghdr from being a public type
@@ -1251,6 +1360,34 @@ impl<'a> ControlMessage<'a> {
             ControlMessage::TxTime(tx_time) => {
                 tx_time as *const _ as *const u8
             },
+            #[cfg(any(
+                apple_targets,
+                linux_android,
+                target_os = "freebsd",
+            ))]
+            #[cfg(feature = "net")]
+            #[allow(clippy::unnecessary_cast)]
+            ControlMessage::IpTos(tos) => {
+                tos as *const _ as *const u8
+            },
+            #[cfg(any(
+                apple_targets,
+                linux_android,
+                target_os = "freebsd",
+            ))]
+            #[cfg(feature = "net")]
+            ControlMessage::Ipv6TClass(tclass) => {
+                tclass as *const _ as *const u8
+            },
+            #[cfg(any(
+                apple_targets,
+                linux_android,
+                target_os = "freebsd",
+            ))]
+            #[cfg(feature = "net")]
+            ControlMessage::IpTtl(ttl) => {
+                ttl as *const _ as *const u8
+            },
         };
         unsafe {
             ptr::copy_nonoverlapping(
@@ -1315,6 +1452,33 @@ impl<'a> ControlMessage<'a> {
             ControlMessage::TxTime(tx_time) => {
                 mem::size_of_val(tx_time)
             },
+            #[cfg(any(
+                apple_targets,
+                linux_android,
+                target_os = "freebsd",
+            ))]
+            #[cfg(feature = "net")]
+            ControlMessage::IpTos(tos) => {
+                mem::size_of_val(tos)
+            },
+            #[cfg(any(
+                apple_targets,
+                linux_android,
+                target_os = "freebsd",
+            ))]
+            #[cfg(feature = "net")]
+            ControlMessage::Ipv6TClass(tclass) => {
+                mem::size_of_val(tclass)
+            },
+            #[cfg(any(
+                apple_targets,
+                linux_android,
+                target_os = "freebsd",
+            ))]
+            #[cfg(feature = "net")]
+            ControlMessage::IpTtl(ttl) => {
+                mem::size_of_val(ttl)
+            },
         }
     }
 
@@ -1349,6 +1513,27 @@ impl<'a> ControlMessage<'a> {
             ControlMessage::RxqOvfl(_) => libc::SOL_SOCKET,
             #[cfg(target_os = "linux")]
             ControlMessage::TxTime(_) => libc::SOL_SOCKET,
+            #[cfg(any(
+                apple_targets,
+                linux_android,
+                target_os = "freebsd",
+            ))]
+            #[cfg(feature = "net")]
+            ControlMessage::IpTos(_) => libc::IPPROTO_IP,
+            #[cfg(any(
+                apple_targets,
+                linux_android,
+                target_os = "freebsd",
+            ))]
+            #[cfg(feature = "net")]
+            ControlMessage::Ipv6TClass(_) => libc::IPPROTO_IPV6,
+            #[cfg(any(
+                apple_targets,
+                linux_android,
+                target_os = "freebsd",
+            ))]
+            #[cfg(feature = "net")]
+            ControlMessage::IpTtl(_) => libc::IPPROTO_IP,
         }
     }
 
@@ -1398,6 +1583,27 @@ impl<'a> ControlMessage<'a> {
             ControlMessage::TxTime(_) => {
                 libc::SCM_TXTIME
             },
+            #[cfg(any(
+                apple_targets,
+                linux_android,
+                target_os = "freebsd",
+            ))]
+            #[cfg(feature = "net")]
+            ControlMessage::IpTos(_) => libc::IP_TOS,
+            #[cfg(any(
+                apple_targets,
+                linux_android,
+                target_os = "freebsd",
+            ))]
+            #[cfg(feature = "net")]
+            ControlMessage::Ipv6TClass(_) => libc::IPV6_TCLASS,
+            #[cfg(any(
+                apple_targets,
+                linux_android,
+                target_os = "freebsd",
+            ))]
+            #[cfg(feature = "net")]
+            ControlMessage::IpTtl(_) => libc::IP_TTL,
         }
     }
 
@@ -2261,4 +2467,3 @@ pub fn shutdown(df: RawFd, how: Shutdown) -> Result<()> {
         Errno::result(shutdown(df, how)).map(drop)
     }
 }
-