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 | |