| // Licensed to the Apache Software Foundation (ASF) under one |
| // or more contributor license agreements. See the NOTICE file |
| // distributed with this work for additional information |
| // regarding copyright ownership. The ASF licenses this file |
| // to you under the Apache License, Version 2.0 (the |
| // "License"); you may not use this file except in compliance |
| // with the License. You may obtain a copy of the License at |
| // |
| // http://www.apache.org/licenses/LICENSE-2.0 |
| // |
| // Unless required by applicable law or agreed to in writing, |
| // software distributed under the License is distributed on an |
| // "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY |
| // KIND, either express or implied. See the License for the |
| // specific language governing permissions and limitations |
| // under the License.. |
| |
| use crate::cmp; |
| use crate::io::{self, IoSlice, IoSliceMut, Read, ReadBuf}; |
| use crate::os::unix::io::{AsFd, AsRawFd, BorrowedFd, FromRawFd, IntoRawFd, OwnedFd, RawFd}; |
| use crate::sys::cvt; |
| use crate::sys_common::{AsInner, FromInner, IntoInner}; |
| |
| use sgx_libc::{c_int, c_void}; |
| |
| #[derive(Debug)] |
| pub struct FileDesc(OwnedFd); |
| |
| const READ_LIMIT: usize = libc::ssize_t::MAX as usize; |
| |
| const fn max_iov() -> usize { |
| sgx_libc::UIO_MAXIOV as usize |
| } |
| |
| impl FileDesc { |
| pub fn read(&self, buf: &mut [u8]) -> io::Result<usize> { |
| let ret = cvt(unsafe { |
| libc::read( |
| self.as_raw_fd(), |
| buf.as_mut_ptr() as *mut c_void, |
| cmp::min(buf.len(), READ_LIMIT), |
| ) |
| })?; |
| Ok(ret as usize) |
| } |
| |
| pub fn read_vectored(&self, bufs: &mut [IoSliceMut<'_>]) -> io::Result<usize> { |
| let ret = cvt(unsafe { |
| libc::readv( |
| self.as_raw_fd(), |
| bufs.as_ptr() as *const libc::iovec, |
| cmp::min(bufs.len(), max_iov()) as c_int, |
| ) |
| })?; |
| Ok(ret as usize) |
| } |
| |
| #[inline] |
| pub fn is_read_vectored(&self) -> bool { |
| true |
| } |
| |
| pub fn read_to_end(&self, buf: &mut Vec<u8>) -> io::Result<usize> { |
| let mut me = self; |
| (&mut me).read_to_end(buf) |
| } |
| |
| pub fn read_at(&self, buf: &mut [u8], offset: u64) -> io::Result<usize> { |
| use libc::pread64; |
| |
| unsafe { |
| cvt(pread64( |
| self.as_raw_fd(), |
| buf.as_mut_ptr() as *mut c_void, |
| cmp::min(buf.len(), READ_LIMIT), |
| offset as i64, |
| )) |
| .map(|n| n as usize) |
| } |
| } |
| |
| pub fn read_buf(&self, buf: &mut ReadBuf<'_>) -> io::Result<()> { |
| let ret = cvt(unsafe { |
| libc::read( |
| self.as_raw_fd(), |
| buf.unfilled_mut().as_mut_ptr() as *mut c_void, |
| cmp::min(buf.remaining(), READ_LIMIT), |
| ) |
| })?; |
| |
| // Safety: `ret` bytes were written to the initialized portion of the buffer |
| unsafe { |
| buf.assume_init(ret as usize); |
| } |
| buf.add_filled(ret as usize); |
| Ok(()) |
| } |
| |
| pub fn write(&self, buf: &[u8]) -> io::Result<usize> { |
| let ret = cvt(unsafe { |
| libc::write( |
| self.as_raw_fd(), |
| buf.as_ptr() as *const c_void, |
| cmp::min(buf.len(), READ_LIMIT), |
| ) |
| })?; |
| Ok(ret as usize) |
| } |
| |
| pub fn write_vectored(&self, bufs: &[IoSlice<'_>]) -> io::Result<usize> { |
| let ret = cvt(unsafe { |
| libc::writev( |
| self.as_raw_fd(), |
| bufs.as_ptr() as *const libc::iovec, |
| cmp::min(bufs.len(), max_iov()) as c_int, |
| ) |
| })?; |
| Ok(ret as usize) |
| } |
| |
| #[inline] |
| pub fn is_write_vectored(&self) -> bool { |
| true |
| } |
| |
| pub fn write_at(&self, buf: &[u8], offset: u64) -> io::Result<usize> { |
| use libc::pwrite64; |
| |
| unsafe { |
| cvt(pwrite64( |
| self.as_raw_fd(), |
| buf.as_ptr() as *const c_void, |
| cmp::min(buf.len(), READ_LIMIT), |
| offset as i64, |
| )) |
| .map(|n| n as usize) |
| } |
| } |
| |
| pub fn get_cloexec(&self) -> io::Result<bool> { |
| unsafe { Ok((cvt(libc::fcntl_arg0(self.as_raw_fd(), libc::F_GETFD))? & libc::FD_CLOEXEC) != 0) } |
| } |
| |
| pub fn set_cloexec(&self) -> io::Result<()> { |
| unsafe { |
| let previous = cvt(libc::fcntl_arg0(self.as_raw_fd(), libc::F_GETFD))?; |
| let new = previous | libc::FD_CLOEXEC; |
| if new != previous { |
| cvt(libc::fcntl_arg1(self.as_raw_fd(), libc::F_SETFD, new))?; |
| } |
| Ok(()) |
| } |
| } |
| |
| pub fn set_nonblocking(&self, nonblocking: bool) -> io::Result<()> { |
| unsafe { |
| let mut v = nonblocking as c_int; |
| cvt(libc::ioctl_arg1(self.as_raw_fd(), libc::FIONBIO, &mut v as *mut c_int))?; |
| Ok(()) |
| } |
| } |
| |
| #[inline] |
| pub fn duplicate(&self) -> io::Result<FileDesc> { |
| Ok(Self(self.0.try_clone()?)) |
| } |
| } |
| |
| impl<'a> Read for &'a FileDesc { |
| fn read(&mut self, buf: &mut [u8]) -> io::Result<usize> { |
| (**self).read(buf) |
| } |
| } |
| |
| impl AsInner<OwnedFd> for FileDesc { |
| fn as_inner(&self) -> &OwnedFd { |
| &self.0 |
| } |
| } |
| |
| impl IntoInner<OwnedFd> for FileDesc { |
| fn into_inner(self) -> OwnedFd { |
| self.0 |
| } |
| } |
| |
| impl FromInner<OwnedFd> for FileDesc { |
| fn from_inner(owned_fd: OwnedFd) -> Self { |
| Self(owned_fd) |
| } |
| } |
| |
| impl AsFd for FileDesc { |
| fn as_fd(&self) -> BorrowedFd<'_> { |
| self.0.as_fd() |
| } |
| } |
| |
| impl AsRawFd for FileDesc { |
| fn as_raw_fd(&self) -> RawFd { |
| self.0.as_raw_fd() |
| } |
| } |
| |
| impl IntoRawFd for FileDesc { |
| fn into_raw_fd(self) -> RawFd { |
| self.0.into_raw_fd() |
| } |
| } |
| |
| impl FromRawFd for FileDesc { |
| unsafe fn from_raw_fd(raw_fd: RawFd) -> Self { |
| Self(FromRawFd::from_raw_fd(raw_fd)) |
| } |
| } |
| |
| mod libc { |
| pub use sgx_libc::ocall::{ |
| close, fcntl_arg0, fcntl_arg1, ioctl_arg0, ioctl_arg1, pread64, pwrite64, read, readv, |
| write, writev, |
| }; |
| pub use sgx_libc::*; |
| } |