Skip to content

Commit

Permalink
feat: Add UDP GRO option
Browse files Browse the repository at this point in the history
  • Loading branch information
XAMPPRocky committed Nov 7, 2024
1 parent 3a93893 commit 7c987e7
Show file tree
Hide file tree
Showing 2 changed files with 45 additions and 1 deletion.
40 changes: 39 additions & 1 deletion src/sys/unix.rs
Original file line number Diff line number Diff line change
Expand Up @@ -179,7 +179,7 @@ pub(crate) use libc::SO_LINGER;
))]
pub(crate) use libc::SO_LINGER_SEC as SO_LINGER;
#[cfg(target_os = "linux")]
pub(crate) use libc::SO_PASSCRED;
pub(crate) use libc::{SO_PASSCRED, SOL_UDP, UDP_GRO};
pub(crate) use libc::{
ip_mreq as IpMreq, linger, IPPROTO_IP, IPPROTO_IPV6, IPV6_MULTICAST_HOPS, IPV6_MULTICAST_IF,
IPV6_MULTICAST_LOOP, IPV6_UNICAST_HOPS, IPV6_V6ONLY, IP_ADD_MEMBERSHIP, IP_DROP_MEMBERSHIP,
Expand Down Expand Up @@ -896,6 +896,44 @@ impl SockAddr {
#[cfg(not(any(target_os = "linux", target_os = "android")))]
None
}

/// Get the value of the `UDP_GRO` option on this socket.
///
/// For more information about this option, see [`set_udp_gro`].
///
/// [`set_udp_gro`]: Socket::set_udp_gro
#[cfg(all(feature = "all", any(target_os = "android", target_os = "linux")))]
#[cfg_attr(
docsrs,
doc(cfg(all(feature = "all", any(target_os = "android", target_os = "linux"))))
)]
pub fn udp_gro(&self) -> io::Result<bool> {
unsafe {
getsockopt::<c_int>(self.as_raw(), sys::SOL_UDP, sys::UDP_GRO)
.map(|reuse| reuse != 0)
}
}

/// Set value for the `UDP_GRO` option on this socket.
///
/// This indicates that the kernel can combine multiple datagrams into a
/// single buffer, this needs to be used in combination with [`Self::recvmsg`]
/// to get the number of segments in the buffer from the [`MsgHdr`].
#[cfg(all(feature = "all", any(target_os = "android", target_os = "linux")))]
#[cfg_attr(
docsrs,
doc(cfg(all(feature = "all", any(target_os = "android", target_os = "linux"))))
)]
pub fn set_udp_gro(&self, reuse: bool) -> io::Result<()> {
unsafe {
setsockopt(
self.as_raw(),
sys::SOL_UDP,
sys::UDP_GRO,
reuse as c_int,
)
}
}
}

pub(crate) type Socket = c_int;
Expand Down
6 changes: 6 additions & 0 deletions tests/socket.rs
Original file line number Diff line number Diff line change
Expand Up @@ -1355,6 +1355,12 @@ test!(reuse_address, set_reuse_address(true));
not(any(windows, target_os = "solaris", target_os = "illumos"))
))]
test!(reuse_port, set_reuse_port(true));
#[cfg(all(feature = "all", any(target_os = "android", target_os = "linux")))]
#[cfg_attr(
docsrs,
doc(cfg(all(feature = "all", any(target_os = "android", target_os = "linux"))))
)]
test!(udp_gro, set_udp_gro(true));
#[cfg(all(feature = "all", target_os = "freebsd"))]
test!(reuse_port_lb, set_reuse_port_lb(true));
#[cfg(all(feature = "all", unix, not(target_os = "redox")))]
Expand Down

0 comments on commit 7c987e7

Please sign in to comment.