-use std::mem;
+use std::{mem, iter};
use std::ffi::{OsStr, OsString};
use rustc::hir::def_id::{DefId, CRATE_DEF_INDEX};
use rustc::mir;
use rustc::ty::{
self,
- layout::{self, Align, LayoutOf, Size, TyLayout},
+ layout::{self, LayoutOf, Size, TyLayout},
};
use rand::RngCore;
}
let this = self.eval_context_mut();
- let ptr = this.memory.check_ptr_access(
- ptr,
- Size::from_bytes(len as u64),
- Align::from_bytes(1).unwrap()
- )?.expect("we already checked for size 0");
-
let mut data = vec![0; len];
if this.machine.communicate {
rng.fill_bytes(&mut data);
}
- this.memory.get_mut(ptr.alloc_id)?.write_bytes(&*this.tcx, ptr, &data)
+ this.memory.write_bytes(ptr, data.iter().copied())
}
/// Visits the memory covered by `place`, sensitive to freezing: the 3rd parameter
Ok(())
}
- fn read_os_string(&mut self, scalar: Scalar<Tag>) -> InterpResult<'tcx, OsString> {
+ /// Sets the last error variable.
+ fn set_last_error(&mut self, scalar: Scalar<Tag>) -> InterpResult<'tcx> {
+ let this = self.eval_context_mut();
+ let errno_place = this.machine.last_error.unwrap();
+ this.write_scalar(scalar, errno_place.into())
+ }
+
+ /// Gets the last error variable.
+ fn get_last_error(&mut self) -> InterpResult<'tcx, Scalar<Tag>> {
+ let this = self.eval_context_mut();
+ let errno_place = this.machine.last_error.unwrap();
+ this.read_scalar(errno_place.into())?.not_undef()
+ }
+
+ /// Sets the last OS error using a `std::io::Error`. This function tries to produce the most
+ /// similar OS error from the `std::io::ErrorKind` and sets it as the last OS error.
+ fn set_last_error_from_io_error(&mut self, e: std::io::Error) -> InterpResult<'tcx> {
+ use std::io::ErrorKind::*;
+ let this = self.eval_context_mut();
+ let target = &this.tcx.tcx.sess.target.target;
+ let last_error = if target.options.target_family == Some("unix".to_owned()) {
+ this.eval_libc(match e.kind() {
+ ConnectionRefused => "ECONNREFUSED",
+ ConnectionReset => "ECONNRESET",
+ PermissionDenied => "EPERM",
+ BrokenPipe => "EPIPE",
+ NotConnected => "ENOTCONN",
+ ConnectionAborted => "ECONNABORTED",
+ AddrNotAvailable => "EADDRNOTAVAIL",
+ AddrInUse => "EADDRINUSE",
+ NotFound => "ENOENT",
+ Interrupted => "EINTR",
+ InvalidInput => "EINVAL",
+ TimedOut => "ETIMEDOUT",
+ AlreadyExists => "EEXIST",
+ WouldBlock => "EWOULDBLOCK",
+ _ => throw_unsup_format!("The {} error cannot be transformed into a raw os error", e)
+ })?
+ } else {
+ // FIXME: we have to implement the windows' equivalent of this.
+ throw_unsup_format!("Setting the last OS error from an io::Error is unsupported for {}.", target.target_os)
+ };
+ this.set_last_error(last_error)
+ }
+
+ /// Helper function that consumes an `std::io::Result<T>` and returns an
+ /// `InterpResult<'tcx,T>::Ok` instead. In case the result is an error, this function returns
+ /// `Ok(-1)` and sets the last OS error accordingly.
+ ///
+ /// This function uses `T: From<i32>` instead of `i32` directly because some IO related
+ /// functions return different integer types (like `read`, that returns an `i64`)
+ fn try_unwrap_io_result<T: From<i32>>(
+ &mut self,
+ result: std::io::Result<T>,
+ ) -> InterpResult<'tcx, T> {
+ match result {
+ Ok(ok) => Ok(ok),
+ Err(e) => {
+ self.eval_context_mut().set_last_error_from_io_error(e)?;
+ Ok((-1).into())
+ }
+ }
+ }
+
+ /// Helper function to read an OsString from a null-terminated sequence of bytes, which is what
+ /// the Unix APIs usually handle.
+ fn read_os_string_from_c_string(&mut self, scalar: Scalar<Tag>) -> InterpResult<'tcx, OsString> {
let bytes = self.eval_context_mut().memory.read_c_str(scalar)?;
Ok(bytes_to_os_str(bytes)?.into())
}
- fn write_os_str(&mut self, os_str: &OsStr, ptr: Pointer<Tag>, size: u64) -> InterpResult<'tcx> {
+ /// Helper function to write an OsStr as a null-terminated sequence of bytes, which is what
+ /// the Unix APIs usually handle. This function returns `Ok(false)` without trying to write if
+ /// `size` is not large enough to fit the contents of `os_string` plus a null terminator. It
+ /// returns `Ok(true)` if the writing process was successful.
+ fn write_os_str_to_c_string(
+ &mut self,
+ os_str: &OsStr,
+ scalar: Scalar<Tag>,
+ size: u64
+ ) -> InterpResult<'tcx, bool> {
let bytes = os_str_to_bytes(os_str)?;
// If `size` is smaller or equal than `bytes.len()`, writing `bytes` plus the required null
// terminator to memory using the `ptr` pointer would cause an overflow.
- if (bytes.len() as u64) < size {
- let this = self.eval_context_mut();
- let tcx = &{ this.tcx.tcx };
- // This is ok because the buffer was strictly larger than `bytes`, so after adding the
- // null terminator, the buffer size is larger or equal to `bytes.len()`, meaning that
- // `bytes` actually fit inside tbe buffer.
- this.memory
- .get_mut(ptr.alloc_id)?
- .write_bytes(tcx, ptr, &bytes)?;
- // We write the `/0` terminator
- let tail_ptr = ptr.offset(Size::from_bytes(bytes.len() as u64 + 1), this)?;
- this.memory
- .get_mut(ptr.alloc_id)?
- .write_bytes(tcx, tail_ptr, b"0")
- } else {
- throw_unsup_format!("OsString is larger than destination")
+ if size <= bytes.len() as u64 {
+ return Ok(false);
}
+ // FIXME: We should use `Iterator::chain` instead when rust-lang/rust#65704 lands.
+ self.eval_context_mut().memory.write_bytes(scalar, bytes.iter().copied().chain(iter::once(0u8)))?;
+ Ok(true)
}
}
#[cfg(target_os = "unix")]
-fn bytes_to_os_str<'tcx, 'a>(bytes: &'a[u8]) -> InterpResult<'tcx, &'a OsStr> {
- Ok(std::os::unix::ffi::OsStringExt::from_bytes(bytes))
+fn os_str_to_bytes<'tcx, 'a>(os_str: &'a OsStr) -> InterpResult<'tcx, &'a [u8]> {
+ std::os::unix::ffi::OsStringExt::into_bytes(os_str)
}
#[cfg(target_os = "unix")]
-fn os_str_to_bytes<'tcx, 'a>(os_str: &'a OsStr) -> InterpResult<'tcx, &'a [u8]> {
- std::os::unix::ffi::OsStringExt::into_bytes(os_str)
+fn bytes_to_os_str<'tcx, 'a>(bytes: &'a[u8]) -> InterpResult<'tcx, &'a OsStr> {
+ Ok(std::os::unix::ffi::OsStringExt::from_bytes(bytes))
}
// On non-unix platforms the best we can do to transform bytes from/to OS strings is to do the