mio/sys/unix/
udp.rs

1use std::io;
2use std::mem;
3use std::net::{self, SocketAddr};
4#[cfg(not(target_os = "hermit"))]
5use std::os::fd::{AsRawFd, FromRawFd};
6// TODO: once <https://github.com/rust-lang/rust/issues/126198> is fixed this
7// can use `std::os::fd` and be merged with the above.
8#[cfg(target_os = "hermit")]
9use std::os::hermit::io::{AsRawFd, FromRawFd};
10
11use crate::sys::unix::net::{new_ip_socket, socket_addr};
12
13pub fn bind(addr: SocketAddr) -> io::Result<net::UdpSocket> {
14    let fd = new_ip_socket(addr, libc::SOCK_DGRAM)?;
15    let socket = unsafe { net::UdpSocket::from_raw_fd(fd) };
16
17    let (raw_addr, raw_addr_length) = socket_addr(&addr);
18    syscall!(bind(fd, raw_addr.as_ptr(), raw_addr_length))?;
19
20    Ok(socket)
21}
22
23pub(crate) fn only_v6(socket: &net::UdpSocket) -> io::Result<bool> {
24    let mut optval: libc::c_int = 0;
25    let mut optlen = mem::size_of::<libc::c_int>() as libc::socklen_t;
26
27    syscall!(getsockopt(
28        socket.as_raw_fd(),
29        libc::IPPROTO_IPV6,
30        libc::IPV6_V6ONLY,
31        &mut optval as *mut _ as *mut _,
32        &mut optlen,
33    ))?;
34
35    Ok(optval != 0)
36}