library/std/src/sys/wasi/io.rs

   1 #![deny(unsafe_op_in_unsafe_fn)]
   2
   3 use crate::marker::PhantomData;
   4 use crate::os::fd::{AsFd, AsRawFd};
   5 use crate::slice;
   6
   7 #[derive(Copy, Clone)]
   8 #[repr(transparent)]
   9 pub struct IoSlice<'a> {
  10     vec: wasi::Ciovec,
  11     _p: PhantomData<&'a [u8]>,
  12 }
  13
  14 impl<'a> IoSlice<'a> {
  15     #[inline]
  16     pub fn new(buf: &'a [u8]) -> IoSlice<'a> {
  17         IoSlice { vec: wasi::Ciovec { buf: buf.as_ptr(), buf_len: buf.len() }, _p: PhantomData }
  18     }
  19
  20     #[inline]
  21     pub fn advance(&mut self, n: usize) {
  22         if self.vec.buf_len < n {
  23             panic!("advancing IoSlice beyond its length");
  24         }
  25
  26         unsafe {
  27             self.vec.buf_len -= n;
  28             self.vec.buf = self.vec.buf.add(n);
  29         }
  30     }
  31
  32     #[inline]
  33     pub fn as_slice(&self) -> &[u8] {
  34         unsafe { slice::from_raw_parts(self.vec.buf as *const u8, self.vec.buf_len) }
  35     }
  36 }
  37
  38 #[repr(transparent)]
  39 pub struct IoSliceMut<'a> {
  40     vec: wasi::Iovec,
  41     _p: PhantomData<&'a mut [u8]>,
  42 }
  43
  44 impl<'a> IoSliceMut<'a> {
  45     #[inline]
  46     pub fn new(buf: &'a mut [u8]) -> IoSliceMut<'a> {
  47         IoSliceMut {
  48             vec: wasi::Iovec { buf: buf.as_mut_ptr(), buf_len: buf.len() },
  49             _p: PhantomData,
  50         }
  51     }
  52
  53     #[inline]
  54     pub fn advance(&mut self, n: usize) {
  55         if self.vec.buf_len < n {
  56             panic!("advancing IoSlice beyond its length");
  57         }
  58
  59         unsafe {
  60             self.vec.buf_len -= n;
  61             self.vec.buf = self.vec.buf.add(n);
  62         }
  63     }
  64
  65     #[inline]
  66     pub fn as_slice(&self) -> &[u8] {
  67         unsafe { slice::from_raw_parts(self.vec.buf as *const u8, self.vec.buf_len) }
  68     }
  69
  70     #[inline]
  71     pub fn as_mut_slice(&mut self) -> &mut [u8] {
  72         unsafe { slice::from_raw_parts_mut(self.vec.buf as *mut u8, self.vec.buf_len) }
  73     }
  74 }
  75
  76 pub fn is_terminal(fd: &impl AsFd) -> bool {
  77     let fd = fd.as_fd();
  78     unsafe { libc::isatty(fd.as_raw_fd()) != 0 }
  79 }