use crate::grip::RawGrip;
#[cfg(not(windows))]
use crate::os::rustix::{AsRawFd, FromRawFd, IntoRawFd, RawFd};
use io_lifetimes::raw::RawFilelike;
use io_lifetimes::views::FilelikeView;
use std::fmt;
use std::fs::File;
use std::io::{self, IoSlice, IoSliceMut, Read, Write};
#[cfg(all(doc, not(windows)))]
use std::net::TcpStream;
#[cfg(windows)]
use {
crate::os::windows::{
AsRawHandleOrSocket, FromRawHandleOrSocket, IntoRawHandleOrSocket, RawEnum,
RawHandleOrSocket,
},
io_lifetimes::raw::RawSocketlike,
io_lifetimes::views::SocketlikeView,
std::net::TcpStream,
std::os::windows::io::{FromRawHandle, RawHandle},
};
#[derive(Copy, Clone)]
#[repr(transparent)]
pub struct RawReadable(RawGrip);
#[derive(Copy, Clone)]
#[repr(transparent)]
pub struct RawWriteable(RawGrip);
#[cfg(not(windows))]
impl AsRawFd for RawReadable {
#[inline]
fn as_raw_fd(&self) -> RawFd {
self.0
}
}
#[cfg(not(windows))]
impl IntoRawFd for RawReadable {
#[inline]
fn into_raw_fd(self) -> RawFd {
self.0
}
}
#[cfg(not(windows))]
impl FromRawFd for RawReadable {
#[inline]
unsafe fn from_raw_fd(raw_fd: RawFd) -> Self {
Self(raw_fd)
}
}
#[cfg(not(windows))]
impl AsRawFd for RawWriteable {
#[inline]
fn as_raw_fd(&self) -> RawFd {
self.0
}
}
#[cfg(not(windows))]
impl IntoRawFd for RawWriteable {
#[inline]
fn into_raw_fd(self) -> RawFd {
self.0
}
}
#[cfg(not(windows))]
impl FromRawFd for RawWriteable {
#[inline]
unsafe fn from_raw_fd(raw_fd: RawFd) -> Self {
Self(raw_fd)
}
}
#[cfg(windows)]
impl AsRawHandleOrSocket for RawReadable {
#[inline]
fn as_raw_handle_or_socket(&self) -> RawHandleOrSocket {
self.0
}
}
#[cfg(windows)]
impl IntoRawHandleOrSocket for RawReadable {
#[inline]
fn into_raw_handle_or_socket(self) -> RawHandleOrSocket {
self.0
}
}
#[cfg(windows)]
impl FromRawHandleOrSocket for RawReadable {
#[inline]
unsafe fn from_raw_handle_or_socket(raw_handle_or_socket: RawHandleOrSocket) -> Self {
Self(raw_handle_or_socket)
}
}
#[cfg(windows)]
impl FromRawHandle for RawReadable {
#[inline]
unsafe fn from_raw_handle(raw_handle: RawHandle) -> Self {
Self(RawHandleOrSocket::unowned_from_raw_handle(raw_handle))
}
}
#[cfg(windows)]
impl AsRawHandleOrSocket for RawWriteable {
#[inline]
fn as_raw_handle_or_socket(&self) -> RawHandleOrSocket {
self.0
}
}
#[cfg(windows)]
impl IntoRawHandleOrSocket for RawWriteable {
#[inline]
fn into_raw_handle_or_socket(self) -> RawHandleOrSocket {
self.0
}
}
#[cfg(windows)]
impl FromRawHandleOrSocket for RawWriteable {
#[inline]
unsafe fn from_raw_handle_or_socket(raw_handle_or_socket: RawHandleOrSocket) -> Self {
Self(raw_handle_or_socket)
}
}
#[cfg(windows)]
impl FromRawHandle for RawWriteable {
#[inline]
unsafe fn from_raw_handle(raw_handle: RawHandle) -> Self {
Self(RawHandleOrSocket::unowned_from_raw_handle(raw_handle))
}
}
#[inline]
unsafe fn as_file_view<'a>(file: RawFilelike) -> FilelikeView<'a, File> {
FilelikeView::<'a>::view_raw(file)
}
#[cfg(windows)]
#[inline]
unsafe fn as_socket_view<'a>(socket: RawSocketlike) -> SocketlikeView<'a, TcpStream> {
SocketlikeView::<'a>::view_raw(socket)
}
#[cfg(not(windows))]
impl Read for RawReadable {
#[inline]
fn read(&mut self, buf: &mut [u8]) -> io::Result<usize> {
unsafe { &*as_file_view(self.0) }.read(buf)
}
#[inline]
fn read_vectored(&mut self, bufs: &mut [IoSliceMut<'_>]) -> io::Result<usize> {
unsafe { &*as_file_view(self.0) }.read_vectored(bufs)
}
#[cfg(can_vector)]
#[inline]
fn is_read_vectored(&self) -> bool {
unsafe { &*as_file_view(self.0) }.is_read_vectored()
}
#[inline]
fn read_to_end(&mut self, buf: &mut Vec<u8>) -> io::Result<usize> {
unsafe { &*as_file_view(self.0) }.read_to_end(buf)
}
#[inline]
fn read_to_string(&mut self, buf: &mut String) -> io::Result<usize> {
unsafe { &*as_file_view(self.0) }.read_to_string(buf)
}
#[inline]
fn read_exact(&mut self, buf: &mut [u8]) -> io::Result<()> {
unsafe { &*as_file_view(self.0) }.read_exact(buf)
}
}
#[cfg(windows)]
impl Read for RawReadable {
#[inline]
fn read(&mut self, buf: &mut [u8]) -> io::Result<usize> {
match self.0 .0 {
RawEnum::Handle(raw_handle) => unsafe { &*as_file_view(raw_handle) }.read(buf),
RawEnum::Socket(raw_socket) => unsafe { &*as_socket_view(raw_socket) }.read(buf),
RawEnum::Stdio(ref mut stdio) => stdio.read(buf),
}
}
#[inline]
fn read_vectored(&mut self, bufs: &mut [IoSliceMut<'_>]) -> io::Result<usize> {
match self.0 .0 {
RawEnum::Handle(raw_handle) => {
unsafe { &*as_file_view(raw_handle) }.read_vectored(bufs)
}
RawEnum::Socket(raw_socket) => {
unsafe { &*as_socket_view(raw_socket) }.read_vectored(bufs)
}
RawEnum::Stdio(ref mut stdio) => stdio.read_vectored(bufs),
}
}
#[cfg(can_vector)]
#[inline]
fn is_read_vectored(&self) -> bool {
match self.0 .0 {
RawEnum::Handle(raw_handle) => unsafe { &*as_file_view(raw_handle) }.is_read_vectored(),
RawEnum::Socket(raw_socket) => {
unsafe { &*as_socket_view(raw_socket) }.is_read_vectored()
}
RawEnum::Stdio(ref stdio) => stdio.is_read_vectored(),
}
}
#[inline]
fn read_to_end(&mut self, buf: &mut Vec<u8>) -> io::Result<usize> {
match self.0 .0 {
RawEnum::Handle(raw_handle) => unsafe { &*as_file_view(raw_handle) }.read_to_end(buf),
RawEnum::Socket(raw_socket) => unsafe { &*as_socket_view(raw_socket) }.read_to_end(buf),
RawEnum::Stdio(ref mut stdio) => stdio.read_to_end(buf),
}
}
#[inline]
fn read_to_string(&mut self, buf: &mut String) -> io::Result<usize> {
match self.0 .0 {
RawEnum::Handle(raw_handle) => {
unsafe { &*as_file_view(raw_handle) }.read_to_string(buf)
}
RawEnum::Socket(raw_socket) => {
unsafe { &*as_socket_view(raw_socket) }.read_to_string(buf)
}
RawEnum::Stdio(ref mut stdio) => stdio.read_to_string(buf),
}
}
#[inline]
fn read_exact(&mut self, buf: &mut [u8]) -> io::Result<()> {
match self.0 .0 {
RawEnum::Handle(raw_handle) => unsafe { &*as_file_view(raw_handle) }.read_exact(buf),
RawEnum::Socket(raw_socket) => unsafe { &*as_socket_view(raw_socket) }.read_exact(buf),
RawEnum::Stdio(ref mut stdio) => stdio.read_exact(buf),
}
}
}
#[cfg(not(windows))]
impl Write for RawWriteable {
#[inline]
fn write(&mut self, buf: &[u8]) -> io::Result<usize> {
unsafe { &*as_file_view(self.0) }.write(buf)
}
#[inline]
fn flush(&mut self) -> io::Result<()> {
unsafe { &*as_file_view(self.0) }.flush()
}
#[inline]
fn write_vectored(&mut self, bufs: &[IoSlice<'_>]) -> io::Result<usize> {
unsafe { &*as_file_view(self.0) }.write_vectored(bufs)
}
#[cfg(can_vector)]
#[inline]
fn is_write_vectored(&self) -> bool {
unsafe { &*as_file_view(self.0) }.is_write_vectored()
}
#[inline]
fn write_all(&mut self, buf: &[u8]) -> io::Result<()> {
unsafe { &*as_file_view(self.0) }.write_all(buf)
}
#[cfg(write_all_vectored)]
#[inline]
fn write_all_vectored(&mut self, bufs: &mut [IoSlice<'_>]) -> io::Result<()> {
unsafe { &*as_file_view(self.0) }.write_all_vectored(bufs)
}
#[inline]
fn write_fmt(&mut self, fmt: fmt::Arguments<'_>) -> io::Result<()> {
unsafe { &*as_file_view(self.0) }.write_fmt(fmt)
}
}
#[cfg(windows)]
impl Write for RawWriteable {
#[inline]
fn write(&mut self, buf: &[u8]) -> io::Result<usize> {
match self.0 .0 {
RawEnum::Handle(raw_handle) => unsafe { &*as_file_view(raw_handle) }.write(buf),
RawEnum::Socket(raw_socket) => unsafe { &*as_socket_view(raw_socket) }.write(buf),
RawEnum::Stdio(ref mut stdio) => stdio.write(buf),
}
}
#[inline]
fn flush(&mut self) -> io::Result<()> {
match self.0 .0 {
RawEnum::Handle(raw_handle) => unsafe { &*as_file_view(raw_handle) }.flush(),
RawEnum::Socket(raw_socket) => unsafe { &*as_socket_view(raw_socket) }.flush(),
RawEnum::Stdio(ref mut stdio) => stdio.flush(),
}
}
#[inline]
fn write_vectored(&mut self, bufs: &[IoSlice<'_>]) -> io::Result<usize> {
match self.0 .0 {
RawEnum::Handle(raw_handle) => {
unsafe { &*as_file_view(raw_handle) }.write_vectored(bufs)
}
RawEnum::Socket(raw_socket) => {
unsafe { &*as_socket_view(raw_socket) }.write_vectored(bufs)
}
RawEnum::Stdio(ref mut stdio) => stdio.write_vectored(bufs),
}
}
#[cfg(can_vector)]
#[inline]
fn is_write_vectored(&self) -> bool {
match self.0 .0 {
RawEnum::Handle(raw_handle) => {
unsafe { &*as_file_view(raw_handle) }.is_write_vectored()
}
RawEnum::Socket(raw_socket) => {
unsafe { &*as_socket_view(raw_socket) }.is_write_vectored()
}
RawEnum::Stdio(ref stdio) => stdio.is_write_vectored(),
}
}
#[inline]
fn write_all(&mut self, buf: &[u8]) -> io::Result<()> {
match self.0 .0 {
RawEnum::Handle(raw_handle) => unsafe { &*as_file_view(raw_handle) }.write_all(buf),
RawEnum::Socket(raw_socket) => unsafe { &*as_socket_view(raw_socket) }.write_all(buf),
RawEnum::Stdio(ref mut stdio) => stdio.write_all(buf),
}
}
#[cfg(write_all_vectored)]
#[inline]
fn write_all_vectored(&mut self, bufs: &mut [IoSlice<'_>]) -> io::Result<()> {
match self.0 .0 {
RawEnum::Handle(raw_handle) => {
unsafe { &*as_file_view(raw_handle) }.write_all_vectored(bufs)
}
RawEnum::Socket(raw_socket) => {
unsafe { &*as_socket_view(raw_socket) }.write_all_vectored(bufs)
}
RawEnum::Stdio(ref mut stdio) => stdio.write_all_vectored(bufs),
}
}
#[inline]
fn write_fmt(&mut self, fmt: fmt::Arguments<'_>) -> io::Result<()> {
match self.0 .0 {
RawEnum::Handle(raw_handle) => unsafe { &*as_file_view(raw_handle) }.write_fmt(fmt),
RawEnum::Socket(raw_socket) => unsafe { &*as_socket_view(raw_socket) }.write_fmt(fmt),
RawEnum::Stdio(ref mut stdio) => stdio.write_fmt(fmt),
}
}
}
#[cfg(not(windows))]
impl fmt::Debug for RawReadable {
#[allow(clippy::missing_inline_in_public_items)]
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
f.debug_struct("RawReadable")
.field("raw_fd", &self.0)
.finish()
}
}
#[cfg(windows)]
impl fmt::Debug for RawReadable {
#[allow(clippy::missing_inline_in_public_items)]
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
f.debug_struct("RawReadable")
.field("raw_handle_or_socket", &self.0)
.finish()
}
}
#[cfg(not(windows))]
impl fmt::Debug for RawWriteable {
#[allow(clippy::missing_inline_in_public_items)]
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
f.debug_struct("RawWriteable")
.field("raw_fd", &self.0)
.finish()
}
}
#[cfg(windows)]
impl fmt::Debug for RawWriteable {
#[allow(clippy::missing_inline_in_public_items)]
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
f.debug_struct("RawWriteable")
.field("raw_handle_or_socket", &self.0)
.finish()
}
}