1
  2
  3
  4
  5
  6
  7
  8
  9
 10
 11
 12
 13
 14
 15
 16
 17
 18
 19
 20
 21
 22
 23
 24
 25
 26
 27
 28
 29
 30
 31
 32
 33
 34
 35
 36
 37
 38
 39
 40
 41
 42
 43
 44
 45
 46
 47
 48
 49
 50
 51
 52
 53
 54
 55
 56
 57
 58
 59
 60
 61
 62
 63
 64
 65
 66
 67
 68
 69
 70
 71
 72
 73
 74
 75
 76
 77
 78
 79
 80
 81
 82
 83
 84
 85
 86
 87
 88
 89
 90
 91
 92
 93
 94
 95
 96
 97
 98
 99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127

//! Define the [`OwnsRaw`] trait and implement it for well-known types.

#[cfg(feature = "os_pipe")]
use os_pipe::{PipeReader, PipeWriter};
#[cfg(unix)]
use std::os::unix::net::{UnixDatagram, UnixListener, UnixStream};
use std::{
    fs::File,
    io::{Stderr, StderrLock, Stdin, StdinLock, Stdout, StdoutLock},
    net::{TcpListener, TcpStream, UdpSocket},
    process::{ChildStderr, ChildStdin, ChildStdout},
};

/// Assert that a type owns its raw file descriptor or handle.
///
/// The `AsRaw*` and `IntoRaw*` traits by themselves are not sufficient to
/// describe the ownership of the file descriptor or handle, as they aren't
/// unsafe to implement. See the discussion in [rust-lang/rust#76969] for
/// additional background. The [`UnsafeHandle`] type provided by this crate
/// serves as an implementation of [this suggestion].
///
/// `OwnsRaw` is a trait that types can implement to declare that they do own
/// their file descriptors or handles.
///
/// [rust-lang/rust#76969]: https://github.com/rust-lang/rust/issues/76969
/// [this suggestion]: https://github.com/rust-lang/rust/pull/76969#issuecomment-696275470
/// [`UnsafeHandle`]: crate::UnsafeHandle
///
/// # Safety
///
/// Types implementing `OwnsRaw` must own the file descriptors or handles they
/// return in their `AsRaw*` and `IntoRaw*` implementations.
pub unsafe trait OwnsRaw {}

// Safety: The following types are all known to own their file descriptors or
// handles.
unsafe impl OwnsRaw for Stdin {}
unsafe impl OwnsRaw for StdinLock<'_> {}
unsafe impl OwnsRaw for Stdout {}
unsafe impl OwnsRaw for StdoutLock<'_> {}
unsafe impl OwnsRaw for Stderr {}
unsafe impl OwnsRaw for StderrLock<'_> {}
unsafe impl OwnsRaw for File {}
unsafe impl OwnsRaw for ChildStdin {}
unsafe impl OwnsRaw for ChildStdout {}
unsafe impl OwnsRaw for ChildStderr {}
unsafe impl OwnsRaw for TcpStream {}
unsafe impl OwnsRaw for TcpListener {}
unsafe impl OwnsRaw for UdpSocket {}
#[cfg(feature = "os_pipe")]
unsafe impl OwnsRaw for PipeReader {}
#[cfg(feature = "os_pipe")]
unsafe impl OwnsRaw for PipeWriter {}
#[cfg(feature = "socket2")]
unsafe impl OwnsRaw for socket2::Socket {}
#[cfg(feature = "use_mio_net")]
unsafe impl OwnsRaw for mio::net::TcpStream {}
#[cfg(feature = "use_mio_net")]
unsafe impl OwnsRaw for mio::net::TcpListener {}
#[cfg(feature = "use_mio_net")]
unsafe impl OwnsRaw for mio::net::TcpSocket {}
#[cfg(feature = "use_mio_net")]
unsafe impl OwnsRaw for mio::net::UdpSocket {}
#[cfg(all(unix, feature = "use_mio_net"))]
unsafe impl OwnsRaw for mio::net::UnixDatagram {}
#[cfg(all(unix, feature = "use_mio_net"))]
unsafe impl OwnsRaw for mio::net::UnixListener {}
#[cfg(all(unix, feature = "use_mio_net"))]
unsafe impl OwnsRaw for mio::net::UnixStream {}
#[cfg(all(unix, feature = "use_mio_os_ext"))]
unsafe impl OwnsRaw for mio::unix::pipe::Receiver {}
#[cfg(all(unix, feature = "use_mio_os_ext"))]
unsafe impl OwnsRaw for mio::unix::pipe::Sender {}
#[cfg(unix)]
unsafe impl OwnsRaw for UnixStream {}
#[cfg(unix)]
unsafe impl OwnsRaw for UnixListener {}
#[cfg(unix)]
unsafe impl OwnsRaw for UnixDatagram {}
#[cfg(feature = "async-std")]
unsafe impl OwnsRaw for async_std::io::Stdin {}
#[cfg(feature = "async-std")]
unsafe impl OwnsRaw for async_std::io::Stdout {}
#[cfg(feature = "async-std")]
unsafe impl OwnsRaw for async_std::io::Stderr {}
#[cfg(feature = "async-std")]
unsafe impl OwnsRaw for async_std::fs::File {}
// async_std's `ChildStdin`, `ChildStdout`, and `ChildStderr` don't implement
// `AsRawFd` or `AsRawHandle`.
#[cfg(feature = "async-std")]
unsafe impl OwnsRaw for async_std::net::TcpStream {}
#[cfg(feature = "async-std")]
unsafe impl OwnsRaw for async_std::net::TcpListener {}
#[cfg(feature = "async-std")]
unsafe impl OwnsRaw for async_std::net::UdpSocket {}
#[cfg(all(feature = "async-std", unix))]
unsafe impl OwnsRaw for async_std::os::unix::net::UnixStream {}
#[cfg(all(feature = "async-std", unix))]
unsafe impl OwnsRaw for async_std::os::unix::net::UnixListener {}
#[cfg(all(feature = "async-std", unix))]
unsafe impl OwnsRaw for async_std::os::unix::net::UnixDatagram {}
#[cfg(feature = "tokio")]
unsafe impl OwnsRaw for tokio::io::Stdin {}
#[cfg(feature = "tokio")]
unsafe impl OwnsRaw for tokio::io::Stdout {}
#[cfg(feature = "tokio")]
unsafe impl OwnsRaw for tokio::io::Stderr {}
#[cfg(feature = "tokio")]
unsafe impl OwnsRaw for tokio::fs::File {}
#[cfg(feature = "tokio")]
unsafe impl OwnsRaw for tokio::net::TcpStream {}
#[cfg(feature = "tokio")]
unsafe impl OwnsRaw for tokio::net::TcpListener {}
#[cfg(feature = "tokio")]
unsafe impl OwnsRaw for tokio::net::UdpSocket {}
#[cfg(all(feature = "tokio", unix))]
unsafe impl OwnsRaw for tokio::net::UnixStream {}
#[cfg(all(feature = "tokio", unix))]
unsafe impl OwnsRaw for tokio::net::UnixListener {}
#[cfg(all(feature = "tokio", unix))]
unsafe impl OwnsRaw for tokio::net::UnixDatagram {}
#[cfg(feature = "tokio")]
unsafe impl OwnsRaw for tokio::process::ChildStdin {}
#[cfg(feature = "tokio")]
unsafe impl OwnsRaw for tokio::process::ChildStdout {}
#[cfg(feature = "tokio")]
unsafe impl OwnsRaw for tokio::process::ChildStderr {}