1#![allow(clippy::needless_lifetimes)]
10
11#![deny(missing_docs, missing_debug_implementations, rust_2018_idioms)]
56#![cfg_attr(docsrs, feature(doc_cfg))]
58#![cfg_attr(test, deny(warnings))]
60#![doc(test(attr(deny(warnings))))]
62
63use std::fmt;
64#[cfg(not(any(target_os = "redox", target_os = "wasi")))]
65use std::io::IoSlice;
66#[cfg(not(any(target_os = "redox", target_os = "wasi")))]
67use std::marker::PhantomData;
68#[cfg(not(any(target_os = "redox", target_os = "wasi")))]
69use std::mem;
70use std::mem::MaybeUninit;
71use std::net::SocketAddr;
72use std::ops::{Deref, DerefMut};
73use std::time::Duration;
74
75macro_rules! impl_debug {
81 (
82 $type: path,
84 $(
85 $(#[$target: meta])*
86 $libc: ident :: $flag: ident
90 ),+ $(,)*
91 ) => {
92 impl std::fmt::Debug for $type {
93 fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
94 let string = match self.0 {
95 $(
96 $(#[$target])*
97 $libc :: $flag => stringify!($flag),
98 )+
99 n => return write!(f, "{n}"),
100 };
101 f.write_str(string)
102 }
103 }
104 };
105}
106
107macro_rules! from {
109 ($from: ty, $for: ty) => {
110 impl From<$from> for $for {
111 fn from(socket: $from) -> $for {
112 #[cfg(any(unix, all(target_os = "wasi", not(target_env = "p1"))))]
113 unsafe {
114 <$for>::from_raw_fd(socket.into_raw_fd())
115 }
116 #[cfg(windows)]
117 unsafe {
118 <$for>::from_raw_socket(socket.into_raw_socket())
119 }
120 }
121 }
122 };
123}
124
125#[rustfmt::skip]
127macro_rules! man_links {
128 ($syscall: tt ( $section: tt ) ) => {
130 concat!(
131 man_links!(__ intro),
132 man_links!(__ unix $syscall($section)),
133 man_links!(__ windows $syscall($section)),
134 )
135 };
136 (unix: $syscall: tt ( $section: tt ) ) => {
138 concat!(
139 man_links!(__ intro),
140 man_links!(__ unix $syscall($section)),
141 )
142 };
143 (windows: $syscall: tt ( $section: tt ) ) => {
145 concat!(
146 man_links!(__ intro),
147 man_links!(__ windows $syscall($section)),
148 )
149 };
150 (__ intro) => {
152 "\n\nAdditional documentation can be found in manual of the OS:\n\n"
153 };
154 (__ unix $syscall: tt ( $section: tt ) ) => {
156 concat!(
157 " * DragonFly BSD: <https://man.dragonflybsd.org/?command=", stringify!($syscall), "§ion=", stringify!($section), ">\n",
158 " * FreeBSD: <https://www.freebsd.org/cgi/man.cgi?query=", stringify!($syscall), "&sektion=", stringify!($section), ">\n",
159 " * Linux: <https://man7.org/linux/man-pages/man", stringify!($section), "/", stringify!($syscall), ".", stringify!($section), ".html>\n",
160 " * macOS: <https://developer.apple.com/library/archive/documentation/System/Conceptual/ManPages_iPhoneOS/man2/", stringify!($syscall), ".", stringify!($section), ".html> (archived, actually for iOS)\n",
161 " * NetBSD: <https://man.netbsd.org/", stringify!($syscall), ".", stringify!($section), ">\n",
162 " * OpenBSD: <https://man.openbsd.org/", stringify!($syscall), ".", stringify!($section), ">\n",
163 " * iOS: <https://developer.apple.com/library/archive/documentation/System/Conceptual/ManPages_iPhoneOS/man2/", stringify!($syscall), ".", stringify!($section), ".html> (archived)\n",
164 " * illumos: <https://illumos.org/man/3SOCKET/", stringify!($syscall), ">\n",
165 )
166 };
167 (__ windows $syscall: tt ( $section: tt ) ) => {
169 concat!(
170 " * Windows: <https://docs.microsoft.com/en-us/windows/win32/api/winsock2/nf-winsock2-", stringify!($syscall), ">\n",
171 )
172 };
173}
174
175mod sockaddr;
176mod socket;
177mod sockref;
178
179#[cfg_attr(
180 any(unix, all(target_os = "wasi", not(target_env = "p1"))),
181 path = "sys/unix.rs"
182)]
183#[cfg_attr(windows, path = "sys/windows.rs")]
184mod sys;
185
186#[cfg(not(any(windows, unix, all(target_os = "wasi", not(target_env = "p1")))))]
187compile_error!("Socket2 doesn't support the compile target");
188
189use sys::c_int;
190
191pub use sockaddr::{sa_family_t, socklen_t, SockAddr, SockAddrStorage};
192#[cfg(not(any(
193 target_os = "haiku",
194 target_os = "illumos",
195 target_os = "netbsd",
196 target_os = "redox",
197 target_os = "solaris",
198 target_os = "wasi",
199)))]
200pub use socket::InterfaceIndexOrAddress;
201pub use socket::Socket;
202pub use sockref::SockRef;
203#[cfg(all(feature = "all", target_os = "linux"))]
204pub use sys::CcidEndpoints;
205#[cfg(all(feature = "all", any(target_os = "linux", target_os = "android")))]
206pub use sys::SockFilter;
207
208#[derive(Copy, Clone, Eq, PartialEq)]
218pub struct Domain(c_int);
219
220impl Domain {
221 pub const IPV4: Domain = Domain(sys::AF_INET);
223
224 pub const IPV6: Domain = Domain(sys::AF_INET6);
226
227 #[cfg(not(target_os = "wasi"))]
229 pub const UNIX: Domain = Domain(sys::AF_UNIX);
230
231 pub const fn for_address(address: SocketAddr) -> Domain {
233 match address {
234 SocketAddr::V4(_) => Domain::IPV4,
235 SocketAddr::V6(_) => Domain::IPV6,
236 }
237 }
238}
239
240impl From<c_int> for Domain {
241 fn from(d: c_int) -> Domain {
242 Domain(d)
243 }
244}
245
246impl From<Domain> for c_int {
247 fn from(d: Domain) -> c_int {
248 d.0
249 }
250}
251
252#[derive(Copy, Clone, Eq, PartialEq)]
262pub struct Type(c_int);
263
264impl Type {
265 pub const STREAM: Type = Type(sys::SOCK_STREAM);
269
270 pub const DGRAM: Type = Type(sys::SOCK_DGRAM);
274
275 #[cfg(all(feature = "all", target_os = "linux"))]
279 pub const DCCP: Type = Type(sys::SOCK_DCCP);
280
281 #[cfg(all(feature = "all", not(any(target_os = "espidf", target_os = "wasi"))))]
283 pub const SEQPACKET: Type = Type(sys::SOCK_SEQPACKET);
284
285 #[cfg(all(
287 feature = "all",
288 not(any(target_os = "redox", target_os = "espidf", target_os = "wasi"))
289 ))]
290 pub const RAW: Type = Type(sys::SOCK_RAW);
291}
292
293impl From<c_int> for Type {
294 fn from(t: c_int) -> Type {
295 Type(t)
296 }
297}
298
299impl From<Type> for c_int {
300 fn from(t: Type) -> c_int {
301 t.0
302 }
303}
304
305#[derive(Copy, Clone, Eq, PartialEq)]
313pub struct Protocol(c_int);
314
315impl Protocol {
316 #[cfg(not(target_os = "wasi"))]
318 pub const ICMPV4: Protocol = Protocol(sys::IPPROTO_ICMP);
319
320 #[cfg(not(target_os = "wasi"))]
322 pub const ICMPV6: Protocol = Protocol(sys::IPPROTO_ICMPV6);
323
324 pub const TCP: Protocol = Protocol(sys::IPPROTO_TCP);
326
327 pub const UDP: Protocol = Protocol(sys::IPPROTO_UDP);
329
330 #[cfg(target_os = "linux")]
331 pub const MPTCP: Protocol = Protocol(sys::IPPROTO_MPTCP);
333
334 #[cfg(all(feature = "all", target_os = "linux"))]
336 pub const DCCP: Protocol = Protocol(sys::IPPROTO_DCCP);
337
338 #[cfg(all(feature = "all", any(target_os = "freebsd", target_os = "linux")))]
340 pub const SCTP: Protocol = Protocol(sys::IPPROTO_SCTP);
341
342 #[cfg(all(
344 feature = "all",
345 any(
346 target_os = "android",
347 target_os = "freebsd",
348 target_os = "fuchsia",
349 target_os = "linux",
350 )
351 ))]
352 pub const UDPLITE: Protocol = Protocol(sys::IPPROTO_UDPLITE);
353
354 #[cfg(all(feature = "all", any(target_os = "freebsd", target_os = "openbsd")))]
356 pub const DIVERT: Protocol = Protocol(sys::IPPROTO_DIVERT);
357}
358
359impl From<c_int> for Protocol {
360 fn from(p: c_int) -> Protocol {
361 Protocol(p)
362 }
363}
364
365impl From<Protocol> for c_int {
366 fn from(p: Protocol) -> c_int {
367 p.0
368 }
369}
370
371#[cfg(not(any(target_os = "redox", target_os = "wasi")))]
375#[derive(Copy, Clone, Eq, PartialEq)]
376pub struct RecvFlags(c_int);
377
378#[cfg(not(any(target_os = "redox", target_os = "wasi")))]
379impl RecvFlags {
380 #[cfg(not(target_os = "espidf"))]
388 pub const fn is_truncated(self) -> bool {
389 self.0 & sys::MSG_TRUNC != 0
390 }
391}
392
393#[repr(transparent)]
397pub struct MaybeUninitSlice<'a>(sys::MaybeUninitSlice<'a>);
398
399impl<'a> fmt::Debug for MaybeUninitSlice<'a> {
400 fn fmt(&self, fmt: &mut fmt::Formatter<'_>) -> fmt::Result {
401 fmt::Debug::fmt(self.0.as_slice(), fmt)
402 }
403}
404
405impl<'a> MaybeUninitSlice<'a> {
406 pub fn new(buf: &'a mut [MaybeUninit<u8>]) -> MaybeUninitSlice<'a> {
412 MaybeUninitSlice(sys::MaybeUninitSlice::new(buf))
413 }
414}
415
416impl<'a> Deref for MaybeUninitSlice<'a> {
417 type Target = [MaybeUninit<u8>];
418
419 fn deref(&self) -> &[MaybeUninit<u8>] {
420 self.0.as_slice()
421 }
422}
423
424impl<'a> DerefMut for MaybeUninitSlice<'a> {
425 fn deref_mut(&mut self) -> &mut [MaybeUninit<u8>] {
426 self.0.as_mut_slice()
427 }
428}
429
430#[derive(Debug, Clone)]
434pub struct TcpKeepalive {
435 #[cfg_attr(
436 any(target_os = "openbsd", target_os = "haiku", target_os = "vita"),
437 allow(dead_code)
438 )]
439 time: Option<Duration>,
440 #[cfg(not(any(
441 target_os = "openbsd",
442 target_os = "redox",
443 target_os = "solaris",
444 target_os = "nto",
445 target_os = "espidf",
446 target_os = "vita",
447 target_os = "haiku",
448 )))]
449 interval: Option<Duration>,
450 #[cfg(not(any(
451 target_os = "openbsd",
452 target_os = "redox",
453 target_os = "solaris",
454 target_os = "nto",
455 target_os = "espidf",
456 target_os = "vita",
457 target_os = "haiku",
458 )))]
459 retries: Option<u32>,
460}
461
462impl TcpKeepalive {
463 #[allow(clippy::new_without_default)]
465 pub const fn new() -> TcpKeepalive {
466 TcpKeepalive {
467 time: None,
468 #[cfg(not(any(
469 target_os = "openbsd",
470 target_os = "redox",
471 target_os = "solaris",
472 target_os = "nto",
473 target_os = "espidf",
474 target_os = "vita",
475 target_os = "haiku",
476 )))]
477 interval: None,
478 #[cfg(not(any(
479 target_os = "openbsd",
480 target_os = "redox",
481 target_os = "solaris",
482 target_os = "nto",
483 target_os = "espidf",
484 target_os = "vita",
485 target_os = "haiku",
486 )))]
487 retries: None,
488 }
489 }
490
491 pub const fn with_time(self, time: Duration) -> Self {
503 Self {
504 time: Some(time),
505 ..self
506 }
507 }
508
509 #[cfg(any(
517 target_os = "android",
518 target_os = "dragonfly",
519 target_os = "freebsd",
520 target_os = "fuchsia",
521 target_os = "illumos",
522 target_os = "ios",
523 target_os = "visionos",
524 target_os = "linux",
525 target_os = "macos",
526 target_os = "netbsd",
527 target_os = "tvos",
528 target_os = "watchos",
529 target_os = "windows",
530 target_os = "cygwin",
531 all(target_os = "wasi", not(target_env = "p1")),
532 ))]
533 pub const fn with_interval(self, interval: Duration) -> Self {
534 Self {
535 interval: Some(interval),
536 ..self
537 }
538 }
539
540 #[cfg(all(
545 feature = "all",
546 any(
547 target_os = "android",
548 target_os = "dragonfly",
549 target_os = "freebsd",
550 target_os = "fuchsia",
551 target_os = "illumos",
552 target_os = "ios",
553 target_os = "visionos",
554 target_os = "linux",
555 target_os = "macos",
556 target_os = "netbsd",
557 target_os = "tvos",
558 target_os = "watchos",
559 target_os = "cygwin",
560 target_os = "windows",
561 all(target_os = "wasi", not(target_env = "p1")),
562 )
563 ))]
564 pub const fn with_retries(self, retries: u32) -> Self {
565 Self {
566 retries: Some(retries),
567 ..self
568 }
569 }
570}
571
572#[cfg(not(any(target_os = "redox", target_os = "wasi")))]
577#[repr(transparent)]
578pub struct MsgHdr<'addr, 'bufs, 'control> {
579 inner: sys::msghdr,
580 #[allow(clippy::type_complexity)]
581 _lifetimes: PhantomData<(&'addr SockAddr, &'bufs IoSlice<'bufs>, &'control [u8])>,
582}
583
584#[cfg(not(any(target_os = "redox", target_os = "wasi")))]
585impl<'addr, 'bufs, 'control> MsgHdr<'addr, 'bufs, 'control> {
586 #[allow(clippy::new_without_default)]
588 pub fn new() -> MsgHdr<'addr, 'bufs, 'control> {
589 MsgHdr {
591 inner: unsafe { mem::zeroed() },
592 _lifetimes: PhantomData,
593 }
594 }
595
596 pub fn with_addr(mut self, addr: &'addr SockAddr) -> Self {
601 sys::set_msghdr_name(&mut self.inner, addr);
602 self
603 }
604
605 pub fn with_buffers(mut self, bufs: &'bufs [IoSlice<'_>]) -> Self {
610 let ptr = bufs.as_ptr() as *mut _;
611 sys::set_msghdr_iov(&mut self.inner, ptr, bufs.len());
612 self
613 }
614
615 pub fn with_control(mut self, buf: &'control [u8]) -> Self {
620 let ptr = buf.as_ptr() as *mut _;
621 sys::set_msghdr_control(&mut self.inner, ptr, buf.len());
622 self
623 }
624
625 pub fn with_flags(mut self, flags: sys::c_int) -> Self {
629 sys::set_msghdr_flags(&mut self.inner, flags);
630 self
631 }
632}
633
634#[cfg(not(any(target_os = "redox", target_os = "wasi")))]
635impl<'name, 'bufs, 'control> fmt::Debug for MsgHdr<'name, 'bufs, 'control> {
636 fn fmt(&self, fmt: &mut fmt::Formatter<'_>) -> fmt::Result {
637 "MsgHdr".fmt(fmt)
638 }
639}
640
641#[cfg(not(any(target_os = "redox", target_os = "wasi")))]
646#[repr(transparent)]
647pub struct MsgHdrMut<'addr, 'bufs, 'control> {
648 inner: sys::msghdr,
649 #[allow(clippy::type_complexity)]
650 _lifetimes: PhantomData<(
651 &'addr mut SockAddr,
652 &'bufs mut MaybeUninitSlice<'bufs>,
653 &'control mut [u8],
654 )>,
655}
656
657#[cfg(not(any(target_os = "redox", target_os = "wasi")))]
658impl<'addr, 'bufs, 'control> MsgHdrMut<'addr, 'bufs, 'control> {
659 #[allow(clippy::new_without_default)]
661 pub fn new() -> MsgHdrMut<'addr, 'bufs, 'control> {
662 MsgHdrMut {
664 inner: unsafe { mem::zeroed() },
665 _lifetimes: PhantomData,
666 }
667 }
668
669 #[allow(clippy::needless_pass_by_ref_mut)]
674 pub fn with_addr(mut self, addr: &'addr mut SockAddr) -> Self {
675 sys::set_msghdr_name(&mut self.inner, addr);
676 self
677 }
678
679 pub fn with_buffers(mut self, bufs: &'bufs mut [MaybeUninitSlice<'_>]) -> Self {
684 sys::set_msghdr_iov(&mut self.inner, bufs.as_mut_ptr().cast(), bufs.len());
685 self
686 }
687
688 pub fn with_control(mut self, buf: &'control mut [MaybeUninit<u8>]) -> Self {
693 sys::set_msghdr_control(&mut self.inner, buf.as_mut_ptr().cast(), buf.len());
694 self
695 }
696
697 pub fn flags(&self) -> RecvFlags {
699 sys::msghdr_flags(&self.inner)
700 }
701
702 pub fn control_len(&self) -> usize {
708 sys::msghdr_control_len(&self.inner)
709 }
710}
711
712#[cfg(not(any(target_os = "redox", target_os = "wasi")))]
713impl<'name, 'bufs, 'control> fmt::Debug for MsgHdrMut<'name, 'bufs, 'control> {
714 fn fmt(&self, fmt: &mut fmt::Formatter<'_>) -> fmt::Result {
715 "MsgHdrMut".fmt(fmt)
716 }
717}