| 1 | use std::io; |
| 2 | use std::mem; |
| 3 | use std::net::{self, SocketAddr}; |
| 4 | #[cfg (not(target_os = "hermit" ))] |
| 5 | use 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" )] |
| 9 | use std::os::hermit::io::{AsRawFd, FromRawFd}; |
| 10 | |
| 11 | use crate::sys::unix::net::{new_ip_socket, socket_addr}; |
| 12 | |
| 13 | pub fn bind(addr: SocketAddr) -> io::Result<net::UdpSocket> { |
| 14 | let fd: i32 = new_ip_socket(addr, socket_type:libc::SOCK_DGRAM)?; |
| 15 | let socket: UdpSocket = unsafe { net::UdpSocket::from_raw_fd(fd) }; |
| 16 | |
| 17 | let (raw_addr: SocketAddrCRepr, raw_addr_length: u32) = socket_addr(&addr); |
| 18 | syscall!(bind(fd, raw_addr.as_ptr(), raw_addr_length))?; |
| 19 | |
| 20 | Ok(socket) |
| 21 | } |
| 22 | |
| 23 | pub(crate) fn only_v6(socket: &net::UdpSocket) -> io::Result<bool> { |
| 24 | let mut optval: libc::c_int = 0; |
| 25 | let mut optlen: u32 = 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 | } |
| 37 | |