use std::borrow::Cow;
use std::cell::RefCell;
+use std::collections::HashSet;
use std::fmt;
use std::num::NonZeroU64;
use std::time::Instant;
use rand::rngs::StdRng;
use rand::SeedableRng;
+use rustc_ast::ast::Mutability;
use rustc_data_structures::fx::FxHashMap;
+#[allow(unused)]
+use rustc_data_structures::static_assert_size;
use rustc_middle::{
mir,
ty::{
self,
layout::{LayoutCx, LayoutError, LayoutOf, TyAndLayout},
- Instance, TyCtxt,
+ Instance, TyCtxt, TypeAndMut,
},
};
-use rustc_span::def_id::DefId;
-use rustc_span::symbol::{sym, Symbol};
+use rustc_span::def_id::{CrateNum, DefId};
+use rustc_span::Symbol;
use rustc_target::abi::Size;
use rustc_target::spec::abi::Abi;
-use crate::*;
+use crate::{
+ concurrency::{data_race, weak_memory},
+ shims::unix::FileHandler,
+ *,
+};
// Some global facts about the emulated machine.
pub const PAGE_SIZE: u64 = 4 * 1024; // FIXME: adjust to target architecture
/// Memory for args, errno, and other parts of the machine-managed environment.
/// This memory may leak.
Machine,
- /// Memory for env vars. Separate from `Machine` because we clean it up and leak-check it.
- Env,
+ /// Memory allocated by the runtime (e.g. env vars). Separate from `Machine`
+ /// because we clean it up and leak-check it.
+ Runtime,
/// Globals copied from `tcx`.
/// This memory may leak.
Global,
Tls,
}
-impl Into<MemoryKind<MiriMemoryKind>> for MiriMemoryKind {
+impl From<MiriMemoryKind> for MemoryKind<MiriMemoryKind> {
#[inline(always)]
- fn into(self) -> MemoryKind<MiriMemoryKind> {
- MemoryKind::Machine(self)
+ fn from(kind: MiriMemoryKind) -> MemoryKind<MiriMemoryKind> {
+ MemoryKind::Machine(kind)
}
}
fn may_leak(self) -> bool {
use self::MiriMemoryKind::*;
match self {
- Rust | C | WinHeap | Env => false,
+ Rust | C | WinHeap | Runtime => false,
Machine | Global | ExternStatic | Tls => true,
}
}
C => write!(f, "C heap"),
WinHeap => write!(f, "Windows heap"),
Machine => write!(f, "machine-managed memory"),
- Env => write!(f, "environment variable"),
+ Runtime => write!(f, "language runtime memory"),
Global => write!(f, "global (static or const)"),
ExternStatic => write!(f, "extern static"),
Tls => write!(f, "thread-local static"),
/// Pointer provenance (tag).
#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
-pub struct Tag {
+pub enum Tag {
+ Concrete(ConcreteTag),
+ Wildcard,
+}
+
+#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
+pub struct ConcreteTag {
pub alloc_id: AllocId,
/// Stacked Borrows tag.
pub sb: SbTag,
}
+#[cfg(all(target_arch = "x86_64", target_pointer_width = "64"))]
+static_assert_size!(Pointer<Tag>, 24);
+// #[cfg(all(target_arch = "x86_64", target_pointer_width = "64"))]
+// static_assert_size!(Pointer<Option<Tag>>, 24);
+#[cfg(all(target_arch = "x86_64", target_pointer_width = "64"))]
+static_assert_size!(ScalarMaybeUninit<Tag>, 32);
+
impl Provenance for Tag {
/// We use absolute addresses in the `offset` of a `Pointer<Tag>`.
const OFFSET_IS_ADDR: bool = true;
fn fmt(ptr: &Pointer<Self>, f: &mut fmt::Formatter<'_>) -> fmt::Result {
let (tag, addr) = ptr.into_parts(); // address is absolute
write!(f, "0x{:x}", addr.bytes())?;
- // Forward `alternate` flag to `alloc_id` printing.
- if f.alternate() {
- write!(f, "[{:#?}]", tag.alloc_id)?;
- } else {
- write!(f, "[{:?}]", tag.alloc_id)?;
+
+ match tag {
+ Tag::Concrete(tag) => {
+ // Forward `alternate` flag to `alloc_id` printing.
+ if f.alternate() {
+ write!(f, "[{:#?}]", tag.alloc_id)?;
+ } else {
+ write!(f, "[{:?}]", tag.alloc_id)?;
+ }
+ // Print Stacked Borrows tag.
+ write!(f, "{:?}", tag.sb)?;
+ }
+ Tag::Wildcard => {
+ write!(f, "[Wildcard]")?;
+ }
}
- // Print Stacked Borrows tag.
- write!(f, "{:?}", tag.sb)
+
+ Ok(())
}
- fn get_alloc_id(self) -> AllocId {
- self.alloc_id
+ fn get_alloc_id(self) -> Option<AllocId> {
+ match self {
+ Tag::Concrete(concrete) => Some(concrete.alloc_id),
+ Tag::Wildcard => None,
+ }
}
}
/// Data race detection via the use of a vector-clock,
/// this is only added if it is enabled.
pub data_race: Option<data_race::AllocExtra>,
-}
-
-/// Extra global memory data
-#[derive(Debug)]
-pub struct MemoryExtra {
- pub stacked_borrows: Option<stacked_borrows::MemoryExtra>,
- pub data_race: Option<data_race::MemoryExtra>,
- pub intptrcast: intptrcast::MemoryExtra,
-
- /// Mapping extern static names to their base pointer.
- extern_statics: FxHashMap<Symbol, Pointer<Tag>>,
-
- /// The random number generator used for resolving non-determinism.
- /// Needs to be queried by ptr_to_int, hence needs interior mutability.
- pub(crate) rng: RefCell<StdRng>,
-
- /// An allocation ID to report when it is being allocated
- /// (helps for debugging memory leaks and use after free bugs).
- tracked_alloc_id: Option<AllocId>,
-
- /// Controls whether alignment of memory accesses is being checked.
- pub(crate) check_alignment: AlignmentCheck,
-
- /// Failure rate of compare_exchange_weak, between 0.0 and 1.0
- pub(crate) cmpxchg_weak_failure_rate: f64,
-}
-
-impl MemoryExtra {
- pub fn new(config: &MiriConfig) -> Self {
- let rng = StdRng::seed_from_u64(config.seed.unwrap_or(0));
- let stacked_borrows = if config.stacked_borrows {
- Some(RefCell::new(stacked_borrows::GlobalState::new(
- config.tracked_pointer_tag,
- config.tracked_call_id,
- config.tag_raw,
- )))
- } else {
- None
- };
- let data_race =
- if config.data_race_detector { Some(data_race::GlobalState::new()) } else { None };
- MemoryExtra {
- stacked_borrows,
- data_race,
- intptrcast: Default::default(),
- extern_statics: FxHashMap::default(),
- rng: RefCell::new(rng),
- tracked_alloc_id: config.tracked_alloc_id,
- check_alignment: config.check_alignment,
- cmpxchg_weak_failure_rate: config.cmpxchg_weak_failure_rate,
- }
- }
-
- fn add_extern_static<'tcx, 'mir>(
- this: &mut MiriEvalContext<'mir, 'tcx>,
- name: &str,
- ptr: Pointer<Option<Tag>>,
- ) {
- let ptr = ptr.into_pointer_or_addr().unwrap();
- this.memory.extra.extern_statics.try_insert(Symbol::intern(name), ptr).unwrap();
- }
-
- /// Sets up the "extern statics" for this machine.
- pub fn init_extern_statics<'tcx, 'mir>(
- this: &mut MiriEvalContext<'mir, 'tcx>,
- ) -> InterpResult<'tcx> {
- match this.tcx.sess.target.os.as_str() {
- "linux" => {
- // "environ"
- Self::add_extern_static(
- this,
- "environ",
- this.machine.env_vars.environ.unwrap().ptr,
- );
- // A couple zero-initialized pointer-sized extern statics.
- // Most of them are for weak symbols, which we all set to null (indicating that the
- // symbol is not supported, and triggering fallback code which ends up calling a
- // syscall that we do support).
- for name in &["__cxa_thread_atexit_impl", "getrandom", "statx"] {
- let layout = this.machine.layouts.usize;
- let place = this.allocate(layout, MiriMemoryKind::ExternStatic.into())?;
- this.write_scalar(Scalar::from_machine_usize(0, this), &place.into())?;
- Self::add_extern_static(this, name, place.ptr);
- }
- }
- "windows" => {
- // "_tls_used"
- // This is some obscure hack that is part of the Windows TLS story. It's a `u8`.
- let layout = this.machine.layouts.u8;
- let place = this.allocate(layout, MiriMemoryKind::ExternStatic.into())?;
- this.write_scalar(Scalar::from_u8(0), &place.into())?;
- Self::add_extern_static(this, "_tls_used", place.ptr);
- }
- _ => {} // No "extern statics" supported on this target
- }
- Ok(())
- }
+ /// Weak memory emulation via the use of store buffers,
+ /// this is only added if it is enabled.
+ pub weak_memory: Option<weak_memory::AllocExtra>,
}
/// Precomputed layouts of primitive types
pub struct PrimitiveLayouts<'tcx> {
pub unit: TyAndLayout<'tcx>,
pub i8: TyAndLayout<'tcx>,
+ pub i16: TyAndLayout<'tcx>,
pub i32: TyAndLayout<'tcx>,
pub isize: TyAndLayout<'tcx>,
pub u8: TyAndLayout<'tcx>,
+ pub u16: TyAndLayout<'tcx>,
pub u32: TyAndLayout<'tcx>,
pub usize: TyAndLayout<'tcx>,
pub bool: TyAndLayout<'tcx>,
+ pub mut_raw_ptr: TyAndLayout<'tcx>,
}
impl<'mir, 'tcx: 'mir> PrimitiveLayouts<'tcx> {
fn new(layout_cx: LayoutCx<'tcx, TyCtxt<'tcx>>) -> Result<Self, LayoutError<'tcx>> {
+ let tcx = layout_cx.tcx;
+ let mut_raw_ptr = tcx.mk_ptr(TypeAndMut { ty: tcx.types.unit, mutbl: Mutability::Mut });
Ok(Self {
- unit: layout_cx.layout_of(layout_cx.tcx.mk_unit())?,
- i8: layout_cx.layout_of(layout_cx.tcx.types.i8)?,
- i32: layout_cx.layout_of(layout_cx.tcx.types.i32)?,
- isize: layout_cx.layout_of(layout_cx.tcx.types.isize)?,
- u8: layout_cx.layout_of(layout_cx.tcx.types.u8)?,
- u32: layout_cx.layout_of(layout_cx.tcx.types.u32)?,
- usize: layout_cx.layout_of(layout_cx.tcx.types.usize)?,
- bool: layout_cx.layout_of(layout_cx.tcx.types.bool)?,
+ unit: layout_cx.layout_of(tcx.mk_unit())?,
+ i8: layout_cx.layout_of(tcx.types.i8)?,
+ i16: layout_cx.layout_of(tcx.types.i16)?,
+ i32: layout_cx.layout_of(tcx.types.i32)?,
+ isize: layout_cx.layout_of(tcx.types.isize)?,
+ u8: layout_cx.layout_of(tcx.types.u8)?,
+ u16: layout_cx.layout_of(tcx.types.u16)?,
+ u32: layout_cx.layout_of(tcx.types.u32)?,
+ usize: layout_cx.layout_of(tcx.types.usize)?,
+ bool: layout_cx.layout_of(tcx.types.bool)?,
+ mut_raw_ptr: layout_cx.layout_of(mut_raw_ptr)?,
})
}
}
/// The machine itself.
pub struct Evaluator<'mir, 'tcx> {
+ pub stacked_borrows: Option<stacked_borrows::GlobalState>,
+ pub data_race: Option<data_race::GlobalState>,
+ pub intptrcast: intptrcast::GlobalState,
+
/// Environment variables set by `setenv`.
/// Miri does not expose env vars from the host to the emulated program.
pub(crate) env_vars: EnvVars<'tcx>,
/// Whether to enforce the validity invariant.
pub(crate) validate: bool,
- /// Whether to enforce validity (e.g., initialization) of integers and floats.
- pub(crate) enforce_number_validity: bool,
+ /// Whether to allow uninitialized numbers (integers and floats).
+ pub(crate) allow_uninit_numbers: bool,
+
+ /// Whether to allow ptr2int transmutes, and whether to allow *dereferencing* the result of an
+ /// int2ptr transmute.
+ pub(crate) allow_ptr_int_transmute: bool,
/// Whether to enforce [ABI](Abi) of function calls.
pub(crate) enforce_abi: bool,
- pub(crate) file_handler: shims::posix::FileHandler,
- pub(crate) dir_handler: shims::posix::DirHandler,
+ /// The table of file descriptors.
+ pub(crate) file_handler: shims::unix::FileHandler,
+ /// The table of directory descriptors.
+ pub(crate) dir_handler: shims::unix::DirHandler,
/// The "time anchor" for this machine's monotone clock (for `Instant` simulation).
pub(crate) time_anchor: Instant,
/// Equivalent setting as RUST_BACKTRACE on encountering an error.
pub(crate) backtrace_style: BacktraceStyle,
+
+ /// Crates which are considered local for the purposes of error reporting.
+ pub(crate) local_crates: Vec<CrateNum>,
+
+ /// Mapping extern static names to their base pointer.
+ extern_statics: FxHashMap<Symbol, Pointer<Tag>>,
+
+ /// The random number generator used for resolving non-determinism.
+ /// Needs to be queried by ptr_to_int, hence needs interior mutability.
+ pub(crate) rng: RefCell<StdRng>,
+
+ /// The allocation IDs to report when they are being allocated
+ /// (helps for debugging memory leaks and use after free bugs).
+ tracked_alloc_ids: HashSet<AllocId>,
+
+ /// Controls whether alignment of memory accesses is being checked.
+ pub(crate) check_alignment: AlignmentCheck,
+
+ /// Failure rate of compare_exchange_weak, between 0.0 and 1.0
+ pub(crate) cmpxchg_weak_failure_rate: f64,
+
+ /// Corresponds to -Zmiri-mute-stdout-stderr and doesn't write the output but acts as if it succeeded.
+ pub(crate) mute_stdout_stderr: bool,
+
+ /// Whether weak memory emulation is enabled
+ pub(crate) weak_memory: bool,
}
impl<'mir, 'tcx> Evaluator<'mir, 'tcx> {
pub(crate) fn new(config: &MiriConfig, layout_cx: LayoutCx<'tcx, TyCtxt<'tcx>>) -> Self {
+ let local_crates = helpers::get_local_crates(layout_cx.tcx);
let layouts =
PrimitiveLayouts::new(layout_cx).expect("Couldn't get layouts of primitive types");
let profiler = config.measureme_out.as_ref().map(|out| {
measureme::Profiler::new(out).expect("Couldn't create `measureme` profiler")
});
+ let rng = StdRng::seed_from_u64(config.seed.unwrap_or(0));
+ let stacked_borrows = if config.stacked_borrows {
+ Some(RefCell::new(stacked_borrows::GlobalStateInner::new(
+ config.tracked_pointer_tags.clone(),
+ config.tracked_call_ids.clone(),
+ config.tag_raw,
+ )))
+ } else {
+ None
+ };
+ let data_race =
+ if config.data_race_detector { Some(data_race::GlobalState::new()) } else { None };
Evaluator {
- // `env_vars` could be initialized properly here if `Memory` were available before
- // calling this method.
+ stacked_borrows,
+ data_race,
+ intptrcast: RefCell::new(intptrcast::GlobalStateInner::new(config)),
+ // `env_vars` depends on a full interpreter so we cannot properly initialize it yet.
env_vars: EnvVars::default(),
argc: None,
argv: None,
tls: TlsData::default(),
isolated_op: config.isolated_op,
validate: config.validate,
- enforce_number_validity: config.check_number_validity,
+ allow_uninit_numbers: config.allow_uninit_numbers,
+ allow_ptr_int_transmute: config.allow_ptr_int_transmute,
enforce_abi: config.check_abi,
- file_handler: Default::default(),
+ file_handler: FileHandler::new(config.mute_stdout_stderr),
dir_handler: Default::default(),
time_anchor: Instant::now(),
layouts,
exported_symbols_cache: FxHashMap::default(),
panic_on_unsupported: config.panic_on_unsupported,
backtrace_style: config.backtrace_style,
+ local_crates,
+ extern_statics: FxHashMap::default(),
+ rng: RefCell::new(rng),
+ tracked_alloc_ids: config.tracked_alloc_ids.clone(),
+ check_alignment: config.check_alignment,
+ cmpxchg_weak_failure_rate: config.cmpxchg_weak_failure_rate,
+ mute_stdout_stderr: config.mute_stdout_stderr,
+ weak_memory: config.weak_memory_emulation,
+ }
+ }
+
+ pub(crate) fn late_init(
+ this: &mut MiriEvalContext<'mir, 'tcx>,
+ config: &MiriConfig,
+ ) -> InterpResult<'tcx> {
+ EnvVars::init(this, config)?;
+ Evaluator::init_extern_statics(this)?;
+ Ok(())
+ }
+
+ fn add_extern_static(
+ this: &mut MiriEvalContext<'mir, 'tcx>,
+ name: &str,
+ ptr: Pointer<Option<Tag>>,
+ ) {
+ // This got just allocated, so there definitely is a pointer here.
+ let ptr = ptr.into_pointer_or_addr().unwrap();
+ this.machine.extern_statics.try_insert(Symbol::intern(name), ptr).unwrap();
+ }
+
+ /// Sets up the "extern statics" for this machine.
+ fn init_extern_statics(this: &mut MiriEvalContext<'mir, 'tcx>) -> InterpResult<'tcx> {
+ match this.tcx.sess.target.os.as_ref() {
+ "linux" => {
+ // "environ"
+ Self::add_extern_static(
+ this,
+ "environ",
+ this.machine.env_vars.environ.unwrap().ptr,
+ );
+ // A couple zero-initialized pointer-sized extern statics.
+ // Most of them are for weak symbols, which we all set to null (indicating that the
+ // symbol is not supported, and triggering fallback code which ends up calling a
+ // syscall that we do support).
+ for name in &["__cxa_thread_atexit_impl", "getrandom", "statx", "__clock_gettime64"]
+ {
+ let layout = this.machine.layouts.usize;
+ let place = this.allocate(layout, MiriMemoryKind::ExternStatic.into())?;
+ this.write_scalar(Scalar::from_machine_usize(0, this), &place.into())?;
+ Self::add_extern_static(this, name, place.ptr);
+ }
+ }
+ "windows" => {
+ // "_tls_used"
+ // This is some obscure hack that is part of the Windows TLS story. It's a `u8`.
+ let layout = this.machine.layouts.u8;
+ let place = this.allocate(layout, MiriMemoryKind::ExternStatic.into())?;
+ this.write_scalar(Scalar::from_u8(0), &place.into())?;
+ Self::add_extern_static(this, "_tls_used", place.ptr);
+ }
+ _ => {} // No "extern statics" supported on this target
}
+ Ok(())
}
pub(crate) fn communicate(&self) -> bool {
self.isolated_op == IsolatedOp::Allow
}
+
+ /// Check whether the stack frame that this `FrameInfo` refers to is part of a local crate.
+ pub(crate) fn is_local(&self, frame: &FrameInfo<'_>) -> bool {
+ let def_id = frame.instance.def_id();
+ def_id.is_local() || self.local_crates.contains(&def_id.krate)
+ }
}
/// A rustc InterpCx for Miri.
/// Machine hook implementations.
impl<'mir, 'tcx> Machine<'mir, 'tcx> for Evaluator<'mir, 'tcx> {
type MemoryKind = MiriMemoryKind;
+ type ExtraFnVal = Dlsym;
type FrameExtra = FrameData<'tcx>;
- type MemoryExtra = MemoryExtra;
type AllocExtra = AllocExtra;
+
type PointerTag = Tag;
- type ExtraFnVal = Dlsym;
+ type TagExtra = SbTag;
type MemoryMap =
MonoHashMap<AllocId, (MemoryKind<MiriMemoryKind>, Allocation<Tag, Self::AllocExtra>)>;
const PANIC_ON_ALLOC_FAIL: bool = false;
#[inline(always)]
- fn enforce_alignment(memory_extra: &MemoryExtra) -> bool {
- memory_extra.check_alignment != AlignmentCheck::None
+ fn enforce_alignment(ecx: &MiriEvalContext<'mir, 'tcx>) -> bool {
+ ecx.machine.check_alignment != AlignmentCheck::None
}
#[inline(always)]
- fn force_int_for_alignment_check(memory_extra: &Self::MemoryExtra) -> bool {
- memory_extra.check_alignment == AlignmentCheck::Int
+ fn force_int_for_alignment_check(ecx: &MiriEvalContext<'mir, 'tcx>) -> bool {
+ ecx.machine.check_alignment == AlignmentCheck::Int
}
#[inline(always)]
- fn enforce_validity(ecx: &InterpCx<'mir, 'tcx, Self>) -> bool {
+ fn enforce_validity(ecx: &MiriEvalContext<'mir, 'tcx>) -> bool {
ecx.machine.validate
}
#[inline(always)]
- fn enforce_number_validity(ecx: &InterpCx<'mir, 'tcx, Self>) -> bool {
- ecx.machine.enforce_number_validity
+ fn enforce_number_init(ecx: &MiriEvalContext<'mir, 'tcx>) -> bool {
+ !ecx.machine.allow_uninit_numbers
}
#[inline(always)]
- fn enforce_abi(ecx: &InterpCx<'mir, 'tcx, Self>) -> bool {
+ fn enforce_number_no_provenance(ecx: &MiriEvalContext<'mir, 'tcx>) -> bool {
+ !ecx.machine.allow_ptr_int_transmute
+ }
+
+ #[inline(always)]
+ fn enforce_abi(ecx: &MiriEvalContext<'mir, 'tcx>) -> bool {
ecx.machine.enforce_abi
}
#[inline(always)]
fn find_mir_or_eval_fn(
- ecx: &mut InterpCx<'mir, 'tcx, Self>,
+ ecx: &mut MiriEvalContext<'mir, 'tcx>,
instance: ty::Instance<'tcx>,
abi: Abi,
args: &[OpTy<'tcx, Tag>],
- ret: Option<(&PlaceTy<'tcx, Tag>, mir::BasicBlock)>,
+ dest: &PlaceTy<'tcx, Tag>,
+ ret: Option<mir::BasicBlock>,
unwind: StackPopUnwind,
) -> InterpResult<'tcx, Option<(&'mir mir::Body<'tcx>, ty::Instance<'tcx>)>> {
- ecx.find_mir_or_eval_fn(instance, abi, args, ret, unwind)
+ ecx.find_mir_or_eval_fn(instance, abi, args, dest, ret, unwind)
}
#[inline(always)]
fn call_extra_fn(
- ecx: &mut InterpCx<'mir, 'tcx, Self>,
+ ecx: &mut MiriEvalContext<'mir, 'tcx>,
fn_val: Dlsym,
abi: Abi,
args: &[OpTy<'tcx, Tag>],
- ret: Option<(&PlaceTy<'tcx, Tag>, mir::BasicBlock)>,
+ dest: &PlaceTy<'tcx, Tag>,
+ ret: Option<mir::BasicBlock>,
_unwind: StackPopUnwind,
) -> InterpResult<'tcx> {
- ecx.call_dlsym(fn_val, abi, args, ret)
+ ecx.call_dlsym(fn_val, abi, args, dest, ret)
}
#[inline(always)]
fn call_intrinsic(
- ecx: &mut rustc_const_eval::interpret::InterpCx<'mir, 'tcx, Self>,
+ ecx: &mut MiriEvalContext<'mir, 'tcx>,
instance: ty::Instance<'tcx>,
args: &[OpTy<'tcx, Tag>],
- ret: Option<(&PlaceTy<'tcx, Tag>, mir::BasicBlock)>,
+ dest: &PlaceTy<'tcx, Tag>,
+ ret: Option<mir::BasicBlock>,
unwind: StackPopUnwind,
) -> InterpResult<'tcx> {
- ecx.call_intrinsic(instance, args, ret, unwind)
+ ecx.call_intrinsic(instance, args, dest, ret, unwind)
}
#[inline(always)]
fn assert_panic(
- ecx: &mut InterpCx<'mir, 'tcx, Self>,
+ ecx: &mut MiriEvalContext<'mir, 'tcx>,
msg: &mir::AssertMessage<'tcx>,
unwind: Option<mir::BasicBlock>,
) -> InterpResult<'tcx> {
}
#[inline(always)]
- fn abort(_ecx: &mut InterpCx<'mir, 'tcx, Self>, msg: String) -> InterpResult<'tcx, !> {
+ fn abort(_ecx: &mut MiriEvalContext<'mir, 'tcx>, msg: String) -> InterpResult<'tcx, !> {
throw_machine_stop!(TerminationInfo::Abort(msg))
}
#[inline(always)]
fn binary_ptr_op(
- ecx: &rustc_const_eval::interpret::InterpCx<'mir, 'tcx, Self>,
+ ecx: &MiriEvalContext<'mir, 'tcx>,
bin_op: mir::BinOp,
left: &ImmTy<'tcx, Tag>,
right: &ImmTy<'tcx, Tag>,
}
fn thread_local_static_base_pointer(
- ecx: &mut InterpCx<'mir, 'tcx, Self>,
+ ecx: &mut MiriEvalContext<'mir, 'tcx>,
def_id: DefId,
) -> InterpResult<'tcx, Pointer<Tag>> {
ecx.get_or_create_thread_local_alloc(def_id)
}
fn extern_static_base_pointer(
- memory: &Memory<'mir, 'tcx, Self>,
+ ecx: &MiriEvalContext<'mir, 'tcx>,
def_id: DefId,
) -> InterpResult<'tcx, Pointer<Tag>> {
- let attrs = memory.tcx.get_attrs(def_id);
- let link_name = match memory.tcx.sess.first_attr_value_str_by_name(&attrs, sym::link_name) {
- Some(name) => name,
- None => memory.tcx.item_name(def_id),
- };
- if let Some(&ptr) = memory.extra.extern_statics.get(&link_name) {
+ let link_name = ecx.item_link_name(def_id);
+ if let Some(&ptr) = ecx.machine.extern_statics.get(&link_name) {
Ok(ptr)
} else {
- throw_unsup_format!("`extern` static {:?} is not supported by Miri", def_id)
+ throw_unsup_format!(
+ "`extern` static `{}` from crate `{}` is not supported by Miri",
+ ecx.tcx.def_path_str(def_id),
+ ecx.tcx.crate_name(def_id.krate),
+ )
}
}
fn init_allocation_extra<'b>(
- mem: &Memory<'mir, 'tcx, Self>,
+ ecx: &MiriEvalContext<'mir, 'tcx>,
id: AllocId,
alloc: Cow<'b, Allocation>,
kind: Option<MemoryKind<Self::MemoryKind>>,
) -> Cow<'b, Allocation<Self::PointerTag, Self::AllocExtra>> {
- if Some(id) == mem.extra.tracked_alloc_id {
+ if ecx.machine.tracked_alloc_ids.contains(&id) {
register_diagnostic(NonHaltingDiagnostic::CreatedAlloc(id));
}
let kind = kind.expect("we set our STATIC_KIND so this cannot be None");
let alloc = alloc.into_owned();
- let stacks = if let Some(stacked_borrows) = &mem.extra.stacked_borrows {
- Some(Stacks::new_allocation(id, alloc.size(), stacked_borrows, kind))
+ let stacks = if let Some(stacked_borrows) = &ecx.machine.stacked_borrows {
+ Some(Stacks::new_allocation(
+ id,
+ alloc.size(),
+ stacked_borrows,
+ kind,
+ ecx.machine.current_span(),
+ ))
+ } else {
+ None
+ };
+ let race_alloc = if let Some(data_race) = &ecx.machine.data_race {
+ Some(data_race::AllocExtra::new_allocation(data_race, alloc.size(), kind))
} else {
None
};
- let race_alloc = if let Some(data_race) = &mem.extra.data_race {
- Some(data_race::AllocExtra::new_allocation(&data_race, alloc.size(), kind))
+ let buffer_alloc = if ecx.machine.weak_memory {
+ // FIXME: if this is an atomic obejct, we want to supply its initial value
+ // while allocating the store buffer here.
+ Some(weak_memory::AllocExtra::new_allocation())
} else {
None
};
let alloc: Allocation<Tag, Self::AllocExtra> = alloc.convert_tag_add_extra(
- &mem.tcx,
- AllocExtra { stacked_borrows: stacks, data_race: race_alloc },
- |ptr| Evaluator::tag_alloc_base_pointer(mem, ptr),
+ &ecx.tcx,
+ AllocExtra {
+ stacked_borrows: stacks,
+ data_race: race_alloc,
+ weak_memory: buffer_alloc,
+ },
+ |ptr| Evaluator::tag_alloc_base_pointer(ecx, ptr),
);
Cow::Owned(alloc)
}
fn tag_alloc_base_pointer(
- mem: &Memory<'mir, 'tcx, Self>,
+ ecx: &MiriEvalContext<'mir, 'tcx>,
ptr: Pointer<AllocId>,
) -> Pointer<Tag> {
- let absolute_addr = intptrcast::GlobalState::rel_ptr_to_addr(&mem, ptr);
- let sb_tag = if let Some(stacked_borrows) = &mem.extra.stacked_borrows {
+ let absolute_addr = intptrcast::GlobalStateInner::rel_ptr_to_addr(ecx, ptr);
+ let sb_tag = if let Some(stacked_borrows) = &ecx.machine.stacked_borrows {
stacked_borrows.borrow_mut().base_tag(ptr.provenance)
} else {
SbTag::Untagged
};
- Pointer::new(Tag { alloc_id: ptr.provenance, sb: sb_tag }, Size::from_bytes(absolute_addr))
+ Pointer::new(
+ Tag::Concrete(ConcreteTag { alloc_id: ptr.provenance, sb: sb_tag }),
+ Size::from_bytes(absolute_addr),
+ )
}
#[inline(always)]
- fn ptr_from_addr(
- mem: &Memory<'mir, 'tcx, Self>,
+ fn ptr_from_addr_cast(
+ ecx: &MiriEvalContext<'mir, 'tcx>,
addr: u64,
) -> Pointer<Option<Self::PointerTag>> {
- intptrcast::GlobalState::ptr_from_addr(addr, mem)
+ intptrcast::GlobalStateInner::ptr_from_addr_cast(ecx, addr)
+ }
+
+ #[inline(always)]
+ fn ptr_from_addr_transmute(
+ ecx: &MiriEvalContext<'mir, 'tcx>,
+ addr: u64,
+ ) -> Pointer<Option<Self::PointerTag>> {
+ intptrcast::GlobalStateInner::ptr_from_addr_transmute(ecx, addr)
+ }
+
+ fn expose_ptr(
+ ecx: &mut InterpCx<'mir, 'tcx, Self>,
+ ptr: Pointer<Self::PointerTag>,
+ ) -> InterpResult<'tcx> {
+ match ptr.provenance {
+ Tag::Concrete(concrete) =>
+ intptrcast::GlobalStateInner::expose_addr(ecx, concrete.alloc_id),
+ Tag::Wildcard => {
+ // No need to do anything for wildcard pointers as
+ // their provenances have already been previously exposed.
+ }
+ }
+ Ok(())
}
/// Convert a pointer with provenance into an allocation-offset pair,
/// or a `None` with an absolute address if that conversion is not possible.
fn ptr_get_alloc(
- mem: &Memory<'mir, 'tcx, Self>,
+ ecx: &MiriEvalContext<'mir, 'tcx>,
ptr: Pointer<Self::PointerTag>,
- ) -> (AllocId, Size) {
- let rel = intptrcast::GlobalState::abs_ptr_to_rel(mem, ptr);
- (ptr.provenance.alloc_id, rel)
+ ) -> Option<(AllocId, Size, Self::TagExtra)> {
+ let rel = intptrcast::GlobalStateInner::abs_ptr_to_rel(ecx, ptr);
+
+ rel.map(|(alloc_id, size)| {
+ let sb = match ptr.provenance {
+ Tag::Concrete(ConcreteTag { sb, .. }) => sb,
+ Tag::Wildcard => SbTag::Untagged,
+ };
+ (alloc_id, size, sb)
+ })
}
#[inline(always)]
fn memory_read(
- memory_extra: &Self::MemoryExtra,
+ _tcx: TyCtxt<'tcx>,
+ machine: &Self,
alloc_extra: &AllocExtra,
- tag: Tag,
+ (alloc_id, tag): (AllocId, Self::TagExtra),
range: AllocRange,
) -> InterpResult<'tcx> {
if let Some(data_race) = &alloc_extra.data_race {
- data_race.read(tag.alloc_id, range, memory_extra.data_race.as_ref().unwrap())?;
+ data_race.read(alloc_id, range, machine.data_race.as_ref().unwrap())?;
}
if let Some(stacked_borrows) = &alloc_extra.stacked_borrows {
stacked_borrows.memory_read(
- tag.alloc_id,
- tag.sb,
+ alloc_id,
+ tag,
range,
- memory_extra.stacked_borrows.as_ref().unwrap(),
- )
- } else {
- Ok(())
+ machine.stacked_borrows.as_ref().unwrap(),
+ machine.current_span(),
+ )?;
}
+ if let Some(weak_memory) = &alloc_extra.weak_memory {
+ if !machine.data_race.as_ref().unwrap().ongoing_atomic_access() {
+ // This is a non-atomic access. And if we are accessing a previously atomically
+ // accessed location without racing with them, then the location no longer needs
+ // to exhibit weak-memory behaviours until a fresh atomic access happens
+ weak_memory.destroy_atomicity(range);
+ }
+ }
+ Ok(())
}
#[inline(always)]
fn memory_written(
- memory_extra: &mut Self::MemoryExtra,
+ _tcx: TyCtxt<'tcx>,
+ machine: &mut Self,
alloc_extra: &mut AllocExtra,
- tag: Tag,
+ (alloc_id, tag): (AllocId, Self::TagExtra),
range: AllocRange,
) -> InterpResult<'tcx> {
if let Some(data_race) = &mut alloc_extra.data_race {
- data_race.write(tag.alloc_id, range, memory_extra.data_race.as_mut().unwrap())?;
+ data_race.write(alloc_id, range, machine.data_race.as_mut().unwrap())?;
}
if let Some(stacked_borrows) = &mut alloc_extra.stacked_borrows {
stacked_borrows.memory_written(
- tag.alloc_id,
- tag.sb,
+ alloc_id,
+ tag,
range,
- memory_extra.stacked_borrows.as_mut().unwrap(),
- )
- } else {
- Ok(())
+ machine.stacked_borrows.as_ref().unwrap(),
+ machine.current_span(),
+ )?;
}
+ if let Some(weak_memory) = &alloc_extra.weak_memory {
+ if !machine.data_race.as_ref().unwrap().ongoing_atomic_access() {
+ weak_memory.destroy_atomicity(range);
+ }
+ }
+ Ok(())
}
#[inline(always)]
fn memory_deallocated(
- memory_extra: &mut Self::MemoryExtra,
+ _tcx: TyCtxt<'tcx>,
+ machine: &mut Self,
alloc_extra: &mut AllocExtra,
- tag: Tag,
+ (alloc_id, tag): (AllocId, Self::TagExtra),
range: AllocRange,
) -> InterpResult<'tcx> {
- if Some(tag.alloc_id) == memory_extra.tracked_alloc_id {
- register_diagnostic(NonHaltingDiagnostic::FreedAlloc(tag.alloc_id));
+ if machine.tracked_alloc_ids.contains(&alloc_id) {
+ register_diagnostic(NonHaltingDiagnostic::FreedAlloc(alloc_id));
}
if let Some(data_race) = &mut alloc_extra.data_race {
- data_race.deallocate(tag.alloc_id, range, memory_extra.data_race.as_mut().unwrap())?;
+ data_race.deallocate(alloc_id, range, machine.data_race.as_mut().unwrap())?;
}
if let Some(stacked_borrows) = &mut alloc_extra.stacked_borrows {
stacked_borrows.memory_deallocated(
- tag.alloc_id,
- tag.sb,
+ alloc_id,
+ tag,
range,
- memory_extra.stacked_borrows.as_mut().unwrap(),
+ machine.stacked_borrows.as_ref().unwrap(),
)
} else {
Ok(())
kind: mir::RetagKind,
place: &PlaceTy<'tcx, Tag>,
) -> InterpResult<'tcx> {
- if ecx.memory.extra.stacked_borrows.is_some() { ecx.retag(kind, place) } else { Ok(()) }
+ if ecx.machine.stacked_borrows.is_some() { ecx.retag(kind, place) } else { Ok(()) }
}
#[inline(always)]
None
};
- let stacked_borrows = ecx.memory.extra.stacked_borrows.as_ref();
+ let stacked_borrows = ecx.machine.stacked_borrows.as_ref();
let call_id = stacked_borrows.map_or(NonZeroU64::new(1).unwrap(), |stacked_borrows| {
stacked_borrows.borrow_mut().new_call()
});
#[inline(always)]
fn after_stack_push(ecx: &mut InterpCx<'mir, 'tcx, Self>) -> InterpResult<'tcx> {
- if ecx.memory.extra.stacked_borrows.is_some() { ecx.retag_return_place() } else { Ok(()) }
+ if ecx.machine.stacked_borrows.is_some() { ecx.retag_return_place() } else { Ok(()) }
}
#[inline(always)]