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

changelog/2203.added.md

@@ -0,0 +1,13 @@
+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`.

src/sys/socket/mod.rs

@@ -803,6 +803,38 @@ 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(
+        apple_targets,
+        linux_android,
+        target_os = "freebsd",
+    ))]
+    #[cfg(feature = "net")]
+    Ipv6HopLimit(libc::c_int),
+
     /// Catch-all variant for unimplemented cmsg types.
     #[doc(hidden)]
     Unknown(UnknownCmsg),
@@ -982,6 +1014,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 +1226,34 @@ pub enum ControlMessage<'a> {
     /// page.
     #[cfg(target_os = "linux")]
     TxTime(&'a u64),
+
+    #[cfg(any(
+        apple_targets,
+        target_os = "freebsd",
+    ))]
+    #[cfg(feature = "net")]
+    IpTos(&'a libc::c_int),
+    #[cfg(any(
+        linux_android,
+    ))]
+    #[cfg(feature = "net")]
+    IpTos(&'a u8),
+
+    #[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 +1358,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 +1450,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 +1511,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 +1581,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,
         }
     }