1 //! The BSD sockets API requires us to read the `ss_family` field before we can
2 //! interpret the rest of a `sockaddr` produced by the kernel.
3
4 use super::addr::SocketAddrStorage;
5 #[cfg(unix)]
6 use super::addr::SocketAddrUnix;
7 use super::ext::{in6_addr_new, in_addr_new, sockaddr_in6_new};
8 use crate::backend::c;
9 #[cfg(target_os = "linux")]
10 use crate::net::xdp::SocketAddrXdp;
11 use crate::net::{SocketAddrAny, SocketAddrV4, SocketAddrV6};
12 use core::mem::size_of;
13
write_sockaddr( addr: &SocketAddrAny, storage: *mut SocketAddrStorage, ) -> usize14 pub(crate) unsafe fn write_sockaddr(
15 addr: &SocketAddrAny,
16 storage: *mut SocketAddrStorage,
17 ) -> usize {
18 match addr {
19 SocketAddrAny::V4(v4) => write_sockaddr_v4(v4, storage),
20 SocketAddrAny::V6(v6) => write_sockaddr_v6(v6, storage),
21 #[cfg(unix)]
22 SocketAddrAny::Unix(unix) => write_sockaddr_unix(unix, storage),
23 #[cfg(target_os = "linux")]
24 SocketAddrAny::Xdp(xdp) => write_sockaddr_xdp(xdp, storage),
25 }
26 }
27
encode_sockaddr_v4(v4: &SocketAddrV4) -> c::sockaddr_in28 pub(crate) fn encode_sockaddr_v4(v4: &SocketAddrV4) -> c::sockaddr_in {
29 c::sockaddr_in {
30 #[cfg(any(
31 bsd,
32 target_os = "aix",
33 target_os = "espidf",
34 target_os = "haiku",
35 target_os = "nto",
36 target_os = "vita",
37 ))]
38 sin_len: size_of::<c::sockaddr_in>() as _,
39 sin_family: c::AF_INET as _,
40 sin_port: u16::to_be(v4.port()),
41 sin_addr: in_addr_new(u32::from_ne_bytes(v4.ip().octets())),
42 #[cfg(not(any(target_os = "haiku", target_os = "vita")))]
43 sin_zero: [0; 8_usize],
44 #[cfg(target_os = "haiku")]
45 sin_zero: [0; 24_usize],
46 #[cfg(target_os = "vita")]
47 sin_zero: [0; 6_usize],
48 #[cfg(target_os = "vita")]
49 sin_vport: 0,
50 }
51 }
52
write_sockaddr_v4(v4: &SocketAddrV4, storage: *mut SocketAddrStorage) -> usize53 unsafe fn write_sockaddr_v4(v4: &SocketAddrV4, storage: *mut SocketAddrStorage) -> usize {
54 let encoded = encode_sockaddr_v4(v4);
55 core::ptr::write(storage.cast(), encoded);
56 size_of::<c::sockaddr_in>()
57 }
58
encode_sockaddr_v6(v6: &SocketAddrV6) -> c::sockaddr_in659 pub(crate) fn encode_sockaddr_v6(v6: &SocketAddrV6) -> c::sockaddr_in6 {
60 #[cfg(any(
61 bsd,
62 target_os = "aix",
63 target_os = "espidf",
64 target_os = "haiku",
65 target_os = "nto",
66 target_os = "vita"
67 ))]
68 {
69 sockaddr_in6_new(
70 size_of::<c::sockaddr_in6>() as _,
71 c::AF_INET6 as _,
72 u16::to_be(v6.port()),
73 u32::to_be(v6.flowinfo()),
74 in6_addr_new(v6.ip().octets()),
75 v6.scope_id(),
76 )
77 }
78 #[cfg(not(any(
79 bsd,
80 target_os = "aix",
81 target_os = "espidf",
82 target_os = "haiku",
83 target_os = "nto",
84 target_os = "vita"
85 )))]
86 {
87 sockaddr_in6_new(
88 c::AF_INET6 as _,
89 u16::to_be(v6.port()),
90 u32::to_be(v6.flowinfo()),
91 in6_addr_new(v6.ip().octets()),
92 v6.scope_id(),
93 )
94 }
95 }
96
write_sockaddr_v6(v6: &SocketAddrV6, storage: *mut SocketAddrStorage) -> usize97 unsafe fn write_sockaddr_v6(v6: &SocketAddrV6, storage: *mut SocketAddrStorage) -> usize {
98 let encoded = encode_sockaddr_v6(v6);
99 core::ptr::write(storage.cast(), encoded);
100 size_of::<c::sockaddr_in6>()
101 }
102
103 #[cfg(unix)]
write_sockaddr_unix(unix: &SocketAddrUnix, storage: *mut SocketAddrStorage) -> usize104 unsafe fn write_sockaddr_unix(unix: &SocketAddrUnix, storage: *mut SocketAddrStorage) -> usize {
105 core::ptr::write(storage.cast(), unix.unix);
106 unix.len()
107 }
108
109 #[cfg(target_os = "linux")]
encode_sockaddr_xdp(xdp: &SocketAddrXdp) -> c::sockaddr_xdp110 pub(crate) fn encode_sockaddr_xdp(xdp: &SocketAddrXdp) -> c::sockaddr_xdp {
111 c::sockaddr_xdp {
112 sxdp_family: c::AF_XDP as _,
113 sxdp_flags: xdp.flags().bits(),
114 sxdp_ifindex: xdp.interface_index(),
115 sxdp_queue_id: xdp.queue_id(),
116 sxdp_shared_umem_fd: xdp.shared_umem_fd(),
117 }
118 }
119
120 #[cfg(target_os = "linux")]
write_sockaddr_xdp(xdp: &SocketAddrXdp, storage: *mut SocketAddrStorage) -> usize121 unsafe fn write_sockaddr_xdp(xdp: &SocketAddrXdp, storage: *mut SocketAddrStorage) -> usize {
122 let encoded = encode_sockaddr_xdp(xdp);
123 core::ptr::write(storage.cast(), encoded);
124 size_of::<c::sockaddr_xdp>()
125 }
126