2019-06-29 07:15:05 -05:00
|
|
|
//! Global machine state as well as implementation of the interpreter engine
|
|
|
|
//! `Machine` trait.
|
|
|
|
|
2019-06-27 16:59:00 -05:00
|
|
|
use std::borrow::Cow;
|
2019-06-28 03:16:10 -05:00
|
|
|
use std::cell::RefCell;
|
2021-05-16 04:28:01 -05:00
|
|
|
use std::fmt;
|
2020-02-24 09:22:02 -06:00
|
|
|
use std::num::NonZeroU64;
|
2020-03-19 17:00:02 -05:00
|
|
|
use std::time::Instant;
|
2019-06-27 16:59:00 -05:00
|
|
|
|
2020-03-30 04:07:32 -05:00
|
|
|
use log::trace;
|
2019-06-27 16:59:00 -05:00
|
|
|
use rand::rngs::StdRng;
|
2020-10-28 06:23:35 -05:00
|
|
|
use rand::SeedableRng;
|
2019-06-27 16:59:00 -05:00
|
|
|
|
2020-04-05 16:03:44 -05:00
|
|
|
use rustc_data_structures::fx::FxHashMap;
|
|
|
|
use rustc_middle::{
|
|
|
|
mir,
|
|
|
|
ty::{
|
|
|
|
self,
|
|
|
|
layout::{LayoutCx, LayoutError, TyAndLayout},
|
|
|
|
TyCtxt,
|
|
|
|
},
|
|
|
|
};
|
2020-06-01 12:23:54 -05:00
|
|
|
use rustc_span::def_id::DefId;
|
2021-05-16 04:28:01 -05:00
|
|
|
use rustc_span::symbol::{sym, Symbol};
|
2020-04-05 16:03:44 -05:00
|
|
|
use rustc_target::abi::{LayoutOf, Size};
|
2021-01-11 05:35:13 -06:00
|
|
|
use rustc_target::spec::abi::Abi;
|
2019-06-27 16:59:00 -05:00
|
|
|
|
|
|
|
use crate::*;
|
|
|
|
|
2019-06-29 07:37:41 -05:00
|
|
|
// Some global facts about the emulated machine.
|
2019-10-07 08:39:59 -05:00
|
|
|
pub const PAGE_SIZE: u64 = 4 * 1024; // FIXME: adjust to target architecture
|
|
|
|
pub const STACK_ADDR: u64 = 32 * PAGE_SIZE; // not really about the "stack", but where we start assigning integer addresses to allocations
|
|
|
|
pub const STACK_SIZE: u64 = 16 * PAGE_SIZE; // whatever
|
2019-06-29 07:37:41 -05:00
|
|
|
pub const NUM_CPUS: u64 = 1;
|
|
|
|
|
2019-04-14 20:02:55 -05:00
|
|
|
/// Extra data stored with each stack frame
|
|
|
|
pub struct FrameData<'tcx> {
|
|
|
|
/// Extra data for Stacked Borrows.
|
|
|
|
pub call_id: stacked_borrows::CallId,
|
2019-11-19 07:51:08 -06:00
|
|
|
|
2020-03-14 05:53:09 -05:00
|
|
|
/// If this is Some(), then this is a special "catch unwind" frame (the frame of `try_fn`
|
|
|
|
/// called by `try`). When this frame is popped during unwinding a panic,
|
|
|
|
/// we stop unwinding, use the `CatchUnwindData` to handle catching.
|
|
|
|
pub catch_unwind: Option<CatchUnwindData<'tcx>>,
|
2021-05-08 11:20:51 -05:00
|
|
|
|
2021-05-29 17:09:46 -05:00
|
|
|
/// If `measureme` profiling is enabled, holds timing information
|
|
|
|
/// for the start of this frame. When we finish executing this frame,
|
|
|
|
/// we use this to register a completed event with `measureme`.
|
2021-05-30 10:04:57 -05:00
|
|
|
pub timing: Option<measureme::DetachedTiming>,
|
|
|
|
}
|
|
|
|
|
|
|
|
impl<'tcx> std::fmt::Debug for FrameData<'tcx> {
|
|
|
|
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
|
2021-05-30 10:13:49 -05:00
|
|
|
// Omitting `timing`, it does not support `Debug`.
|
2021-05-30 10:04:57 -05:00
|
|
|
f.debug_struct("FrameData")
|
|
|
|
.field("call_id", &self.call_id)
|
|
|
|
.field("catch_unwind", &self.catch_unwind)
|
|
|
|
.finish()
|
|
|
|
}
|
2019-04-14 20:02:55 -05:00
|
|
|
}
|
|
|
|
|
2019-06-27 16:59:00 -05:00
|
|
|
/// Extra memory kinds
|
|
|
|
#[derive(Debug, Copy, Clone, PartialEq, Eq)]
|
|
|
|
pub enum MiriMemoryKind {
|
|
|
|
/// `__rust_alloc` memory.
|
|
|
|
Rust,
|
|
|
|
/// `malloc` memory.
|
|
|
|
C,
|
2019-07-02 02:03:45 -05:00
|
|
|
/// Windows `HeapAlloc` memory.
|
|
|
|
WinHeap,
|
2020-07-08 04:05:22 -05:00
|
|
|
/// Memory for args, errno, and other parts of the machine-managed environment.
|
2020-03-28 05:06:56 -05:00
|
|
|
/// This memory may leak.
|
2020-02-23 14:55:02 -06:00
|
|
|
Machine,
|
2020-03-28 05:06:56 -05:00
|
|
|
/// Memory for env vars. Separate from `Machine` because we clean it up and leak-check it.
|
|
|
|
Env,
|
2020-03-25 03:05:24 -05:00
|
|
|
/// Globals copied from `tcx`.
|
2020-03-28 05:06:56 -05:00
|
|
|
/// This memory may leak.
|
2020-03-25 03:05:24 -05:00
|
|
|
Global,
|
2020-07-08 04:05:22 -05:00
|
|
|
/// Memory for extern statics.
|
|
|
|
/// This memory may leak.
|
2020-07-27 05:53:39 -05:00
|
|
|
ExternStatic,
|
|
|
|
/// Memory for thread-local statics.
|
|
|
|
/// This memory may leak.
|
|
|
|
Tls,
|
2019-06-27 16:59:00 -05:00
|
|
|
}
|
|
|
|
|
|
|
|
impl Into<MemoryKind<MiriMemoryKind>> for MiriMemoryKind {
|
|
|
|
#[inline(always)]
|
|
|
|
fn into(self) -> MemoryKind<MiriMemoryKind> {
|
|
|
|
MemoryKind::Machine(self)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2020-04-04 06:35:30 -05:00
|
|
|
impl MayLeak for MiriMemoryKind {
|
|
|
|
#[inline(always)]
|
|
|
|
fn may_leak(self) -> bool {
|
|
|
|
use self::MiriMemoryKind::*;
|
|
|
|
match self {
|
|
|
|
Rust | C | WinHeap | Env => false,
|
2020-07-27 05:53:39 -05:00
|
|
|
Machine | Global | ExternStatic | Tls => true,
|
2020-04-04 06:35:30 -05:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
impl fmt::Display for MiriMemoryKind {
|
|
|
|
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
|
|
|
|
use self::MiriMemoryKind::*;
|
|
|
|
match self {
|
|
|
|
Rust => write!(f, "Rust heap"),
|
|
|
|
C => write!(f, "C heap"),
|
|
|
|
WinHeap => write!(f, "Windows heap"),
|
|
|
|
Machine => write!(f, "machine-managed memory"),
|
|
|
|
Env => write!(f, "environment variable"),
|
2020-07-27 05:53:39 -05:00
|
|
|
Global => write!(f, "global (static or const)"),
|
|
|
|
ExternStatic => write!(f, "extern static"),
|
2021-05-16 04:28:01 -05:00
|
|
|
Tls => write!(f, "thread-local static"),
|
2020-04-04 06:35:30 -05:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2019-06-27 16:59:00 -05:00
|
|
|
/// Extra per-allocation data
|
|
|
|
#[derive(Debug, Clone)]
|
|
|
|
pub struct AllocExtra {
|
2020-02-24 09:22:02 -06:00
|
|
|
/// Stacked Borrows state is only added if it is enabled.
|
2019-07-03 03:19:55 -05:00
|
|
|
pub stacked_borrows: Option<stacked_borrows::AllocExtra>,
|
2020-11-15 12:30:26 -06:00
|
|
|
/// 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>,
|
2019-06-27 16:59:00 -05:00
|
|
|
}
|
|
|
|
|
|
|
|
/// Extra global memory data
|
2021-05-22 06:24:08 -05:00
|
|
|
#[derive(Debug)]
|
2020-03-08 11:54:47 -05:00
|
|
|
pub struct MemoryExtra {
|
2020-02-24 09:22:02 -06:00
|
|
|
pub stacked_borrows: Option<stacked_borrows::MemoryExtra>,
|
2020-11-15 12:30:26 -06:00
|
|
|
pub data_race: Option<data_race::MemoryExtra>,
|
2019-06-27 16:59:00 -05:00
|
|
|
pub intptrcast: intptrcast::MemoryExtra,
|
2019-07-03 03:19:55 -05:00
|
|
|
|
2020-02-23 15:32:37 -06:00
|
|
|
/// Mapping extern static names to their canonical allocation.
|
2020-03-06 02:06:23 -06:00
|
|
|
extern_statics: FxHashMap<Symbol, AllocId>,
|
2020-02-23 15:32:37 -06:00
|
|
|
|
2019-07-23 14:38:53 -05:00
|
|
|
/// The random number generator used for resolving non-determinism.
|
2020-02-23 15:32:37 -06:00
|
|
|
/// Needs to be queried by ptr_to_int, hence needs interior mutability.
|
2019-07-23 14:38:53 -05:00
|
|
|
pub(crate) rng: RefCell<StdRng>,
|
2020-03-06 02:06:23 -06:00
|
|
|
|
|
|
|
/// An allocation ID to report when it is being allocated
|
2020-04-14 18:00:56 -05:00
|
|
|
/// (helps for debugging memory leaks and use after free bugs).
|
2020-03-06 02:06:23 -06:00
|
|
|
tracked_alloc_id: Option<AllocId>,
|
2020-04-13 10:51:22 -05:00
|
|
|
|
|
|
|
/// Controls whether alignment of memory accesses is being checked.
|
2020-08-16 10:16:53 -05:00
|
|
|
pub(crate) check_alignment: AlignmentCheck,
|
2021-01-26 02:07:43 -06:00
|
|
|
|
|
|
|
/// Failure rate of compare_exchange_weak, between 0.0 and 1.0
|
|
|
|
pub(crate) cmpxchg_weak_failure_rate: f64,
|
2019-06-27 16:59:00 -05:00
|
|
|
}
|
|
|
|
|
2020-03-08 11:54:47 -05:00
|
|
|
impl MemoryExtra {
|
2020-10-28 06:23:35 -05:00
|
|
|
pub fn new(config: &MiriConfig) -> Self {
|
|
|
|
let rng = StdRng::seed_from_u64(config.seed.unwrap_or(0));
|
|
|
|
let stacked_borrows = if config.stacked_borrows {
|
2021-05-22 06:24:08 -05:00
|
|
|
Some(RefCell::new(stacked_borrows::GlobalState::new(
|
2020-10-28 06:23:35 -05:00
|
|
|
config.tracked_pointer_tag,
|
|
|
|
config.tracked_call_id,
|
|
|
|
config.track_raw,
|
2021-05-22 06:24:08 -05:00
|
|
|
)))
|
2020-02-24 09:22:02 -06:00
|
|
|
} else {
|
|
|
|
None
|
|
|
|
};
|
2021-05-22 07:55:33 -05:00
|
|
|
let data_race =
|
|
|
|
if config.data_race_detector { Some(data_race::GlobalState::new()) } else { None };
|
2020-02-23 15:32:37 -06:00
|
|
|
MemoryExtra {
|
|
|
|
stacked_borrows,
|
2020-11-01 18:23:27 -06:00
|
|
|
data_race,
|
2020-02-23 15:32:37 -06:00
|
|
|
intptrcast: Default::default(),
|
2020-03-02 15:36:15 -06:00
|
|
|
extern_statics: FxHashMap::default(),
|
2020-02-23 15:32:37 -06:00
|
|
|
rng: RefCell::new(rng),
|
2020-10-28 06:23:35 -05:00
|
|
|
tracked_alloc_id: config.tracked_alloc_id,
|
|
|
|
check_alignment: config.check_alignment,
|
2021-01-26 02:07:43 -06:00
|
|
|
cmpxchg_weak_failure_rate: config.cmpxchg_weak_failure_rate,
|
2020-02-23 15:32:37 -06:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2020-03-28 04:07:23 -05:00
|
|
|
fn add_extern_static<'tcx, 'mir>(
|
|
|
|
this: &mut MiriEvalContext<'mir, 'tcx>,
|
|
|
|
name: &str,
|
|
|
|
ptr: Scalar<Tag>,
|
|
|
|
) {
|
|
|
|
let ptr = ptr.assert_ptr();
|
|
|
|
assert_eq!(ptr.offset, Size::ZERO);
|
2021-05-16 04:28:01 -05:00
|
|
|
this.memory.extra.extern_statics.try_insert(Symbol::intern(name), ptr.alloc_id).unwrap();
|
2020-03-28 04:07:23 -05:00
|
|
|
}
|
|
|
|
|
2020-02-23 15:32:37 -06:00
|
|
|
/// Sets up the "extern statics" for this machine.
|
2020-03-08 11:54:47 -05:00
|
|
|
pub fn init_extern_statics<'tcx, 'mir>(
|
2020-02-23 15:32:37 -06:00
|
|
|
this: &mut MiriEvalContext<'mir, 'tcx>,
|
|
|
|
) -> InterpResult<'tcx> {
|
2020-11-11 03:29:10 -06:00
|
|
|
match this.tcx.sess.target.os.as_str() {
|
2020-03-07 14:33:27 -06:00
|
|
|
"linux" => {
|
|
|
|
// "__cxa_thread_atexit_impl"
|
|
|
|
// This should be all-zero, pointer-sized.
|
2020-04-18 10:53:54 -05:00
|
|
|
let layout = this.machine.layouts.usize;
|
2020-07-27 05:53:39 -05:00
|
|
|
let place = this.allocate(layout, MiriMemoryKind::ExternStatic.into());
|
2021-02-19 18:00:00 -06:00
|
|
|
this.write_scalar(Scalar::from_machine_usize(0, this), &place.into())?;
|
2020-03-28 04:07:23 -05:00
|
|
|
Self::add_extern_static(this, "__cxa_thread_atexit_impl", place.ptr);
|
2020-03-07 14:33:27 -06:00
|
|
|
// "environ"
|
2021-05-16 04:28:01 -05:00
|
|
|
Self::add_extern_static(
|
|
|
|
this,
|
|
|
|
"environ",
|
|
|
|
this.machine.env_vars.environ.unwrap().ptr,
|
|
|
|
);
|
2020-03-28 04:07:23 -05:00
|
|
|
}
|
|
|
|
"windows" => {
|
|
|
|
// "_tls_used"
|
|
|
|
// This is some obscure hack that is part of the Windows TLS story. It's a `u8`.
|
2020-04-18 10:53:54 -05:00
|
|
|
let layout = this.machine.layouts.u8;
|
2020-07-27 05:53:39 -05:00
|
|
|
let place = this.allocate(layout, MiriMemoryKind::ExternStatic.into());
|
2021-02-19 18:00:00 -06:00
|
|
|
this.write_scalar(Scalar::from_u8(0), &place.into())?;
|
2020-03-28 04:07:23 -05:00
|
|
|
Self::add_extern_static(this, "_tls_used", place.ptr);
|
2020-02-23 15:32:37 -06:00
|
|
|
}
|
2020-03-22 02:51:15 -05:00
|
|
|
_ => {} // No "extern statics" supported on this target
|
2020-02-23 15:32:37 -06:00
|
|
|
}
|
|
|
|
Ok(())
|
2019-06-27 16:59:00 -05:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2020-04-05 16:03:44 -05:00
|
|
|
/// Precomputed layouts of primitive types
|
2020-04-18 10:53:54 -05:00
|
|
|
pub struct PrimitiveLayouts<'tcx> {
|
|
|
|
pub unit: TyAndLayout<'tcx>,
|
|
|
|
pub i8: TyAndLayout<'tcx>,
|
|
|
|
pub i32: TyAndLayout<'tcx>,
|
|
|
|
pub isize: TyAndLayout<'tcx>,
|
|
|
|
pub u8: TyAndLayout<'tcx>,
|
|
|
|
pub u32: TyAndLayout<'tcx>,
|
|
|
|
pub usize: TyAndLayout<'tcx>,
|
2020-03-29 01:38:34 -05:00
|
|
|
}
|
|
|
|
|
|
|
|
impl<'mir, 'tcx: 'mir> PrimitiveLayouts<'tcx> {
|
2020-04-05 16:03:44 -05:00
|
|
|
fn new(layout_cx: LayoutCx<'tcx, TyCtxt<'tcx>>) -> Result<Self, LayoutError<'tcx>> {
|
|
|
|
Ok(Self {
|
2020-04-18 10:53:54 -05:00
|
|
|
unit: layout_cx.layout_of(layout_cx.tcx.mk_unit())?,
|
|
|
|
i8: layout_cx.layout_of(layout_cx.tcx.types.i8)?,
|
2020-04-05 16:03:44 -05:00
|
|
|
i32: layout_cx.layout_of(layout_cx.tcx.types.i32)?,
|
2020-04-18 10:53:54 -05:00
|
|
|
isize: layout_cx.layout_of(layout_cx.tcx.types.isize)?,
|
|
|
|
u8: layout_cx.layout_of(layout_cx.tcx.types.u8)?,
|
2020-04-05 16:03:44 -05:00
|
|
|
u32: layout_cx.layout_of(layout_cx.tcx.types.u32)?,
|
2020-04-18 10:53:54 -05:00
|
|
|
usize: layout_cx.layout_of(layout_cx.tcx.types.usize)?,
|
2020-04-05 16:03:44 -05:00
|
|
|
})
|
2020-03-29 01:38:34 -05:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2019-06-27 16:59:00 -05:00
|
|
|
/// The machine itself.
|
2020-04-01 18:55:52 -05:00
|
|
|
pub struct Evaluator<'mir, 'tcx> {
|
2019-06-27 16:59:00 -05:00
|
|
|
/// Environment variables set by `setenv`.
|
|
|
|
/// Miri does not expose env vars from the host to the emulated program.
|
2020-03-08 11:54:47 -05:00
|
|
|
pub(crate) env_vars: EnvVars<'tcx>,
|
2019-06-27 16:59:00 -05:00
|
|
|
|
|
|
|
/// Program arguments (`Option` because we can only initialize them after creating the ecx).
|
|
|
|
/// These are *pointers* to argc/argv because macOS.
|
|
|
|
/// We also need the full command line as one string because of Windows.
|
2019-11-02 05:50:21 -05:00
|
|
|
pub(crate) argc: Option<Scalar<Tag>>,
|
|
|
|
pub(crate) argv: Option<Scalar<Tag>>,
|
|
|
|
pub(crate) cmd_line: Option<Scalar<Tag>>,
|
2019-06-27 16:59:00 -05:00
|
|
|
|
|
|
|
/// TLS state.
|
|
|
|
pub(crate) tls: TlsData<'tcx>,
|
2019-08-06 15:32:57 -05:00
|
|
|
|
2019-08-20 10:47:57 -05:00
|
|
|
/// If enabled, the `env_vars` field is populated with the host env vars during initialization
|
|
|
|
/// and random number generation is delegated to the host.
|
2019-08-06 15:32:57 -05:00
|
|
|
pub(crate) communicate: bool,
|
2019-09-24 17:28:00 -05:00
|
|
|
|
2020-02-24 09:25:29 -06:00
|
|
|
/// Whether to enforce the validity invariant.
|
|
|
|
pub(crate) validate: bool,
|
|
|
|
|
2021-05-29 12:36:06 -05:00
|
|
|
/// Whether to enforce [ABI](Abi) of function calls.
|
|
|
|
pub(crate) enforce_abi: bool,
|
|
|
|
|
2020-08-13 04:31:52 -05:00
|
|
|
pub(crate) file_handler: shims::posix::FileHandler,
|
2020-06-27 06:19:35 -05:00
|
|
|
pub(crate) dir_handler: shims::posix::DirHandler,
|
2019-04-14 20:02:55 -05:00
|
|
|
|
2020-03-19 17:00:02 -05:00
|
|
|
/// The "time anchor" for this machine's monotone clock (for `Instant` simulation).
|
|
|
|
pub(crate) time_anchor: Instant,
|
2020-03-29 01:38:34 -05:00
|
|
|
|
2020-03-16 18:48:44 -05:00
|
|
|
/// The set of threads.
|
2020-04-09 14:06:33 -05:00
|
|
|
pub(crate) threads: ThreadManager<'mir, 'tcx>,
|
2020-04-01 18:55:52 -05:00
|
|
|
|
2020-04-05 16:03:44 -05:00
|
|
|
/// Precomputed `TyLayout`s for primitive data types that are commonly used inside Miri.
|
|
|
|
pub(crate) layouts: PrimitiveLayouts<'tcx>,
|
2020-07-23 08:47:33 -05:00
|
|
|
|
|
|
|
/// Allocations that are considered roots of static memory (that may leak).
|
|
|
|
pub(crate) static_roots: Vec<AllocId>,
|
2021-05-08 11:20:51 -05:00
|
|
|
|
2021-05-29 17:09:46 -05:00
|
|
|
/// The `measureme` profiler used to record timing information about
|
|
|
|
/// the emulated program.
|
2021-05-30 10:04:57 -05:00
|
|
|
profiler: Option<measureme::Profiler>,
|
2021-05-29 17:09:46 -05:00
|
|
|
/// Used with `profiler` to cache the `StringId`s for event names
|
|
|
|
/// uesd with `measureme`.
|
2021-05-30 10:04:57 -05:00
|
|
|
string_cache: FxHashMap<String, measureme::StringId>,
|
2019-06-27 16:59:00 -05:00
|
|
|
}
|
|
|
|
|
2020-04-01 18:55:52 -05:00
|
|
|
impl<'mir, 'tcx> Evaluator<'mir, 'tcx> {
|
2021-05-29 17:16:12 -05:00
|
|
|
pub(crate) fn new(config: &MiriConfig, layout_cx: LayoutCx<'tcx, TyCtxt<'tcx>>) -> Self {
|
2021-05-16 04:28:01 -05:00
|
|
|
let layouts =
|
|
|
|
PrimitiveLayouts::new(layout_cx).expect("Couldn't get layouts of primitive types");
|
2021-05-30 10:04:57 -05:00
|
|
|
let profiler = config.measureme_out.as_ref().map(|out| {
|
|
|
|
measureme::Profiler::new(out).expect("Couldn't create `measureme` profiler")
|
|
|
|
});
|
2019-06-27 16:59:00 -05:00
|
|
|
Evaluator {
|
2019-08-13 16:17:41 -05:00
|
|
|
// `env_vars` could be initialized properly here if `Memory` were available before
|
|
|
|
// calling this method.
|
2019-08-14 10:24:35 -05:00
|
|
|
env_vars: EnvVars::default(),
|
2019-06-27 16:59:00 -05:00
|
|
|
argc: None,
|
|
|
|
argv: None,
|
|
|
|
cmd_line: None,
|
|
|
|
tls: TlsData::default(),
|
2021-05-08 11:20:51 -05:00
|
|
|
communicate: config.communicate,
|
|
|
|
validate: config.validate,
|
2021-05-29 12:36:06 -05:00
|
|
|
enforce_abi: config.check_abi,
|
2019-09-24 17:28:00 -05:00
|
|
|
file_handler: Default::default(),
|
2020-01-25 12:57:15 -06:00
|
|
|
dir_handler: Default::default(),
|
2020-03-19 17:00:02 -05:00
|
|
|
time_anchor: Instant::now(),
|
2020-04-05 16:03:44 -05:00
|
|
|
layouts,
|
2020-04-19 23:03:23 -05:00
|
|
|
threads: ThreadManager::default(),
|
2020-07-23 08:47:33 -05:00
|
|
|
static_roots: Vec::new(),
|
2021-05-08 11:20:51 -05:00
|
|
|
profiler,
|
|
|
|
string_cache: Default::default(),
|
2019-06-27 16:59:00 -05:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2019-07-05 16:47:10 -05:00
|
|
|
/// A rustc InterpCx for Miri.
|
2020-04-01 18:55:52 -05:00
|
|
|
pub type MiriEvalContext<'mir, 'tcx> = InterpCx<'mir, 'tcx, Evaluator<'mir, 'tcx>>;
|
2019-06-27 16:59:00 -05:00
|
|
|
|
|
|
|
/// A little trait that's useful to be inherited by extension traits.
|
|
|
|
pub trait MiriEvalContextExt<'mir, 'tcx> {
|
2019-12-04 03:16:08 -06:00
|
|
|
fn eval_context_ref<'a>(&'a self) -> &'a MiriEvalContext<'mir, 'tcx>;
|
|
|
|
fn eval_context_mut<'a>(&'a mut self) -> &'a mut MiriEvalContext<'mir, 'tcx>;
|
2019-06-27 16:59:00 -05:00
|
|
|
}
|
|
|
|
impl<'mir, 'tcx> MiriEvalContextExt<'mir, 'tcx> for MiriEvalContext<'mir, 'tcx> {
|
|
|
|
#[inline(always)]
|
|
|
|
fn eval_context_ref(&self) -> &MiriEvalContext<'mir, 'tcx> {
|
|
|
|
self
|
|
|
|
}
|
|
|
|
#[inline(always)]
|
|
|
|
fn eval_context_mut(&mut self) -> &mut MiriEvalContext<'mir, 'tcx> {
|
|
|
|
self
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
/// Machine hook implementations.
|
2020-04-01 18:55:52 -05:00
|
|
|
impl<'mir, 'tcx> Machine<'mir, 'tcx> for Evaluator<'mir, 'tcx> {
|
2020-03-25 03:05:24 -05:00
|
|
|
type MemoryKind = MiriMemoryKind;
|
2019-06-27 16:59:00 -05:00
|
|
|
|
2019-04-14 20:02:55 -05:00
|
|
|
type FrameExtra = FrameData<'tcx>;
|
2020-03-08 11:54:47 -05:00
|
|
|
type MemoryExtra = MemoryExtra;
|
2019-06-27 16:59:00 -05:00
|
|
|
type AllocExtra = AllocExtra;
|
|
|
|
type PointerTag = Tag;
|
2019-06-30 09:03:13 -05:00
|
|
|
type ExtraFnVal = Dlsym;
|
2019-06-27 16:59:00 -05:00
|
|
|
|
2019-12-23 05:56:23 -06:00
|
|
|
type MemoryMap =
|
|
|
|
MonoHashMap<AllocId, (MemoryKind<MiriMemoryKind>, Allocation<Tag, Self::AllocExtra>)>;
|
2019-06-27 16:59:00 -05:00
|
|
|
|
2020-03-25 03:05:24 -05:00
|
|
|
const GLOBAL_KIND: Option<MiriMemoryKind> = Some(MiriMemoryKind::Global);
|
2019-06-27 16:59:00 -05:00
|
|
|
|
2020-04-13 10:51:22 -05:00
|
|
|
#[inline(always)]
|
|
|
|
fn enforce_alignment(memory_extra: &MemoryExtra) -> bool {
|
2020-08-16 10:08:34 -05:00
|
|
|
memory_extra.check_alignment != AlignmentCheck::None
|
|
|
|
}
|
|
|
|
|
|
|
|
#[inline(always)]
|
|
|
|
fn force_int_for_alignment_check(memory_extra: &Self::MemoryExtra) -> bool {
|
|
|
|
memory_extra.check_alignment == AlignmentCheck::Int
|
2020-04-13 10:51:22 -05:00
|
|
|
}
|
2019-08-05 08:49:19 -05:00
|
|
|
|
2019-06-27 16:59:00 -05:00
|
|
|
#[inline(always)]
|
2019-07-05 16:47:10 -05:00
|
|
|
fn enforce_validity(ecx: &InterpCx<'mir, 'tcx, Self>) -> bool {
|
2020-02-24 09:25:29 -06:00
|
|
|
ecx.machine.validate
|
2019-06-27 16:59:00 -05:00
|
|
|
}
|
|
|
|
|
2021-05-29 12:36:06 -05:00
|
|
|
#[inline(always)]
|
|
|
|
fn enforce_abi(ecx: &InterpCx<'mir, 'tcx, Self>) -> bool {
|
|
|
|
ecx.machine.enforce_abi
|
|
|
|
}
|
|
|
|
|
2019-06-27 16:59:00 -05:00
|
|
|
#[inline(always)]
|
2019-12-04 16:31:39 -06:00
|
|
|
fn find_mir_or_eval_fn(
|
2019-07-05 16:47:10 -05:00
|
|
|
ecx: &mut InterpCx<'mir, 'tcx, Self>,
|
2019-06-27 16:59:00 -05:00
|
|
|
instance: ty::Instance<'tcx>,
|
2021-01-21 20:45:39 -06:00
|
|
|
abi: Abi,
|
2019-06-27 16:59:00 -05:00
|
|
|
args: &[OpTy<'tcx, Tag>],
|
2021-02-19 18:00:00 -06:00
|
|
|
ret: Option<(&PlaceTy<'tcx, Tag>, mir::BasicBlock)>,
|
2021-05-22 09:45:00 -05:00
|
|
|
unwind: StackPopUnwind,
|
2019-06-27 16:59:00 -05:00
|
|
|
) -> InterpResult<'tcx, Option<&'mir mir::Body<'tcx>>> {
|
2021-01-21 20:45:39 -06:00
|
|
|
ecx.find_mir_or_eval_fn(instance, abi, args, ret, unwind)
|
2019-06-27 16:59:00 -05:00
|
|
|
}
|
|
|
|
|
2019-06-30 09:03:13 -05:00
|
|
|
#[inline(always)]
|
|
|
|
fn call_extra_fn(
|
|
|
|
ecx: &mut InterpCx<'mir, 'tcx, Self>,
|
|
|
|
fn_val: Dlsym,
|
2021-01-21 20:45:39 -06:00
|
|
|
abi: Abi,
|
2019-06-30 09:03:13 -05:00
|
|
|
args: &[OpTy<'tcx, Tag>],
|
2021-02-19 18:00:00 -06:00
|
|
|
ret: Option<(&PlaceTy<'tcx, Tag>, mir::BasicBlock)>,
|
2021-05-22 09:45:00 -05:00
|
|
|
_unwind: StackPopUnwind,
|
2019-06-30 09:03:13 -05:00
|
|
|
) -> InterpResult<'tcx> {
|
2021-01-21 20:45:39 -06:00
|
|
|
ecx.call_dlsym(fn_val, abi, args, ret)
|
2019-06-30 09:03:13 -05:00
|
|
|
}
|
|
|
|
|
2019-06-27 16:59:00 -05:00
|
|
|
#[inline(always)]
|
|
|
|
fn call_intrinsic(
|
2019-07-05 16:47:10 -05:00
|
|
|
ecx: &mut rustc_mir::interpret::InterpCx<'mir, 'tcx, Self>,
|
2019-06-27 16:59:00 -05:00
|
|
|
instance: ty::Instance<'tcx>,
|
|
|
|
args: &[OpTy<'tcx, Tag>],
|
2021-02-19 18:00:00 -06:00
|
|
|
ret: Option<(&PlaceTy<'tcx, Tag>, mir::BasicBlock)>,
|
2021-05-22 09:45:00 -05:00
|
|
|
unwind: StackPopUnwind,
|
2019-06-27 16:59:00 -05:00
|
|
|
) -> InterpResult<'tcx> {
|
2020-03-30 15:54:49 -05:00
|
|
|
ecx.call_intrinsic(instance, args, ret, unwind)
|
2019-06-27 16:59:00 -05:00
|
|
|
}
|
|
|
|
|
2019-11-29 03:16:03 -06:00
|
|
|
#[inline(always)]
|
|
|
|
fn assert_panic(
|
|
|
|
ecx: &mut InterpCx<'mir, 'tcx, Self>,
|
2020-02-13 07:01:35 -06:00
|
|
|
msg: &mir::AssertMessage<'tcx>,
|
2019-11-29 03:16:03 -06:00
|
|
|
unwind: Option<mir::BasicBlock>,
|
|
|
|
) -> InterpResult<'tcx> {
|
2020-03-11 14:05:44 -05:00
|
|
|
ecx.assert_panic(msg, unwind)
|
2019-11-29 03:16:03 -06:00
|
|
|
}
|
|
|
|
|
2020-03-12 14:46:58 -05:00
|
|
|
#[inline(always)]
|
2020-12-10 12:53:45 -06:00
|
|
|
fn abort(_ecx: &mut InterpCx<'mir, 'tcx, Self>, msg: String) -> InterpResult<'tcx, !> {
|
|
|
|
throw_machine_stop!(TerminationInfo::Abort(msg))
|
2020-03-12 14:46:58 -05:00
|
|
|
}
|
|
|
|
|
2019-06-27 16:59:00 -05:00
|
|
|
#[inline(always)]
|
2019-07-24 09:17:49 -05:00
|
|
|
fn binary_ptr_op(
|
2019-07-05 16:47:10 -05:00
|
|
|
ecx: &rustc_mir::interpret::InterpCx<'mir, 'tcx, Self>,
|
2019-06-27 16:59:00 -05:00
|
|
|
bin_op: mir::BinOp,
|
2021-02-19 18:00:00 -06:00
|
|
|
left: &ImmTy<'tcx, Tag>,
|
|
|
|
right: &ImmTy<'tcx, Tag>,
|
2020-04-02 17:05:35 -05:00
|
|
|
) -> InterpResult<'tcx, (Scalar<Tag>, bool, ty::Ty<'tcx>)> {
|
2019-07-24 09:17:49 -05:00
|
|
|
ecx.binary_ptr_op(bin_op, left, right)
|
2019-06-27 16:59:00 -05:00
|
|
|
}
|
|
|
|
|
|
|
|
fn box_alloc(
|
2019-07-05 16:47:10 -05:00
|
|
|
ecx: &mut InterpCx<'mir, 'tcx, Self>,
|
2021-02-19 18:00:00 -06:00
|
|
|
dest: &PlaceTy<'tcx, Tag>,
|
2019-06-27 16:59:00 -05:00
|
|
|
) -> InterpResult<'tcx> {
|
|
|
|
trace!("box_alloc for {:?}", dest.layout.ty);
|
2019-11-28 16:42:10 -06:00
|
|
|
let layout = ecx.layout_of(dest.layout.ty.builtin_deref(false).unwrap().ty)?;
|
|
|
|
// First argument: `size`.
|
|
|
|
// (`0` is allowed here -- this is expected to be handled by the lang item).
|
2020-03-28 11:35:40 -05:00
|
|
|
let size = Scalar::from_machine_usize(layout.size.bytes(), ecx);
|
2019-11-28 16:42:10 -06:00
|
|
|
|
|
|
|
// Second argument: `align`.
|
2020-03-28 11:35:40 -05:00
|
|
|
let align = Scalar::from_machine_usize(layout.align.abi.bytes(), ecx);
|
2019-11-28 16:42:10 -06:00
|
|
|
|
2019-06-27 16:59:00 -05:00
|
|
|
// Call the `exchange_malloc` lang item.
|
|
|
|
let malloc = ecx.tcx.lang_items().exchange_malloc_fn().unwrap();
|
|
|
|
let malloc = ty::Instance::mono(ecx.tcx.tcx, malloc);
|
2019-11-28 16:42:10 -06:00
|
|
|
ecx.call_function(
|
2019-06-27 16:59:00 -05:00
|
|
|
malloc,
|
2021-03-14 09:30:37 -05:00
|
|
|
Abi::Rust,
|
2019-11-29 04:17:44 -06:00
|
|
|
&[size.into(), align.into()],
|
2019-06-27 16:59:00 -05:00
|
|
|
Some(dest),
|
|
|
|
// Don't do anything when we are done. The `statement()` function will increment
|
|
|
|
// the old stack frame's stmt counter to the next statement, which means that when
|
|
|
|
// `exchange_malloc` returns, we go on evaluating exactly where we want to be.
|
|
|
|
StackPopCleanup::None { cleanup: true },
|
|
|
|
)?;
|
|
|
|
Ok(())
|
|
|
|
}
|
|
|
|
|
2020-07-26 04:15:01 -05:00
|
|
|
fn thread_local_static_alloc_id(
|
2020-06-01 12:23:54 -05:00
|
|
|
ecx: &mut InterpCx<'mir, 'tcx, Self>,
|
|
|
|
def_id: DefId,
|
|
|
|
) -> InterpResult<'tcx, AllocId> {
|
|
|
|
ecx.get_or_create_thread_local_alloc_id(def_id)
|
|
|
|
}
|
|
|
|
|
2020-07-26 04:15:01 -05:00
|
|
|
fn extern_static_alloc_id(
|
|
|
|
memory: &Memory<'mir, 'tcx, Self>,
|
|
|
|
def_id: DefId,
|
|
|
|
) -> InterpResult<'tcx, AllocId> {
|
|
|
|
let attrs = memory.tcx.get_attrs(def_id);
|
2020-08-08 03:27:23 -05:00
|
|
|
let link_name = match memory.tcx.sess.first_attr_value_str_by_name(&attrs, sym::link_name) {
|
2020-04-14 19:21:52 -05:00
|
|
|
Some(name) => name,
|
2020-07-26 04:15:01 -05:00
|
|
|
None => memory.tcx.item_name(def_id),
|
2020-04-14 19:21:52 -05:00
|
|
|
};
|
2020-07-26 04:15:01 -05:00
|
|
|
if let Some(&id) = memory.extra.extern_statics.get(&link_name) {
|
|
|
|
Ok(id)
|
2020-04-14 19:21:52 -05:00
|
|
|
} else {
|
2020-07-26 04:15:01 -05:00
|
|
|
throw_unsup_format!("`extern` static {:?} is not supported by Miri", def_id)
|
2020-02-23 15:32:37 -06:00
|
|
|
}
|
2019-06-27 16:59:00 -05:00
|
|
|
}
|
|
|
|
|
2019-11-29 12:50:37 -06:00
|
|
|
fn init_allocation_extra<'b>(
|
2020-03-08 11:54:47 -05:00
|
|
|
memory_extra: &MemoryExtra,
|
2019-06-27 16:59:00 -05:00
|
|
|
id: AllocId,
|
|
|
|
alloc: Cow<'b, Allocation>,
|
2020-03-25 03:05:24 -05:00
|
|
|
kind: Option<MemoryKind<Self::MemoryKind>>,
|
2019-12-01 03:18:41 -06:00
|
|
|
) -> (Cow<'b, Allocation<Self::PointerTag, Self::AllocExtra>>, Self::PointerTag) {
|
2020-03-06 02:11:41 -06:00
|
|
|
if Some(id) == memory_extra.tracked_alloc_id {
|
|
|
|
register_diagnostic(NonHaltingDiagnostic::CreatedAlloc(id));
|
|
|
|
}
|
|
|
|
|
2019-06-27 16:59:00 -05:00
|
|
|
let kind = kind.expect("we set our STATIC_KIND so this cannot be None");
|
|
|
|
let alloc = alloc.into_owned();
|
2021-05-16 04:28:01 -05:00
|
|
|
let (stacks, base_tag) = if let Some(stacked_borrows) = &memory_extra.stacked_borrows {
|
|
|
|
let (stacks, base_tag) =
|
2021-05-22 06:24:08 -05:00
|
|
|
Stacks::new_allocation(id, alloc.size(), stacked_borrows, kind);
|
2021-05-16 04:28:01 -05:00
|
|
|
(Some(stacks), base_tag)
|
|
|
|
} else {
|
|
|
|
// No stacks, no tag.
|
|
|
|
(None, Tag::Untagged)
|
|
|
|
};
|
2020-11-15 12:30:26 -06:00
|
|
|
let race_alloc = if let Some(data_race) = &memory_extra.data_race {
|
2021-05-17 17:11:05 -05:00
|
|
|
Some(data_race::AllocExtra::new_allocation(&data_race, alloc.size(), kind))
|
2020-11-15 12:30:26 -06:00
|
|
|
} else {
|
|
|
|
None
|
|
|
|
};
|
2020-02-24 09:22:02 -06:00
|
|
|
let mut stacked_borrows = memory_extra.stacked_borrows.as_ref().map(|sb| sb.borrow_mut());
|
2019-09-17 05:30:14 -05:00
|
|
|
let alloc: Allocation<Tag, Self::AllocExtra> = alloc.with_tags_and_extra(
|
2019-10-07 08:39:59 -05:00
|
|
|
|alloc| {
|
2020-04-12 03:32:36 -05:00
|
|
|
if let Some(stacked_borrows) = &mut stacked_borrows {
|
2020-03-25 03:05:24 -05:00
|
|
|
// Only globals may already contain pointers at this point
|
|
|
|
assert_eq!(kind, MiriMemoryKind::Global.into());
|
|
|
|
stacked_borrows.global_base_ptr(alloc)
|
2020-02-24 09:22:02 -06:00
|
|
|
} else {
|
|
|
|
Tag::Untagged
|
2019-10-07 08:39:59 -05:00
|
|
|
}
|
2019-09-05 11:17:58 -05:00
|
|
|
},
|
2020-11-01 18:23:27 -06:00
|
|
|
AllocExtra { stacked_borrows: stacks, data_race: race_alloc },
|
2019-09-05 11:17:58 -05:00
|
|
|
);
|
2019-12-01 03:18:41 -06:00
|
|
|
(Cow::Owned(alloc), base_tag)
|
2019-06-27 16:59:00 -05:00
|
|
|
}
|
|
|
|
|
2020-04-14 18:00:56 -05:00
|
|
|
#[inline(always)]
|
2021-05-17 06:50:45 -05:00
|
|
|
fn memory_read(
|
2021-05-22 06:24:08 -05:00
|
|
|
memory_extra: &Self::MemoryExtra,
|
2021-05-20 06:32:18 -05:00
|
|
|
alloc_extra: &AllocExtra,
|
2021-05-17 06:50:45 -05:00
|
|
|
ptr: Pointer<Tag>,
|
|
|
|
size: Size,
|
2020-04-14 18:00:56 -05:00
|
|
|
) -> InterpResult<'tcx> {
|
2021-05-20 06:32:18 -05:00
|
|
|
if let Some(data_race) = &alloc_extra.data_race {
|
2021-05-22 07:47:14 -05:00
|
|
|
data_race.read(ptr, size, memory_extra.data_race.as_ref().unwrap())?;
|
2020-04-14 18:00:56 -05:00
|
|
|
}
|
2021-05-20 06:32:18 -05:00
|
|
|
if let Some(stacked_borrows) = &alloc_extra.stacked_borrows {
|
2021-05-22 06:24:08 -05:00
|
|
|
stacked_borrows.memory_read(ptr, size, memory_extra.stacked_borrows.as_ref().unwrap())
|
2021-05-17 06:50:45 -05:00
|
|
|
} else {
|
|
|
|
Ok(())
|
|
|
|
}
|
|
|
|
}
|
2020-05-08 06:56:10 -05:00
|
|
|
|
2021-05-17 06:50:45 -05:00
|
|
|
#[inline(always)]
|
|
|
|
fn memory_written(
|
2021-05-22 06:24:08 -05:00
|
|
|
memory_extra: &mut Self::MemoryExtra,
|
2021-05-20 06:32:18 -05:00
|
|
|
alloc_extra: &mut AllocExtra,
|
2021-05-17 06:50:45 -05:00
|
|
|
ptr: Pointer<Tag>,
|
|
|
|
size: Size,
|
|
|
|
) -> InterpResult<'tcx> {
|
2021-05-20 06:32:18 -05:00
|
|
|
if let Some(data_race) = &mut alloc_extra.data_race {
|
2021-05-22 07:47:14 -05:00
|
|
|
data_race.write(ptr, size, memory_extra.data_race.as_mut().unwrap())?;
|
2021-05-17 06:50:45 -05:00
|
|
|
}
|
2021-05-20 06:32:18 -05:00
|
|
|
if let Some(stacked_borrows) = &mut alloc_extra.stacked_borrows {
|
2021-05-22 07:55:33 -05:00
|
|
|
stacked_borrows.memory_written(
|
|
|
|
ptr,
|
|
|
|
size,
|
|
|
|
memory_extra.stacked_borrows.as_mut().unwrap(),
|
|
|
|
)
|
2021-05-17 06:50:45 -05:00
|
|
|
} else {
|
|
|
|
Ok(())
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
#[inline(always)]
|
|
|
|
fn memory_deallocated(
|
|
|
|
memory_extra: &mut Self::MemoryExtra,
|
2021-05-20 06:32:18 -05:00
|
|
|
alloc_extra: &mut AllocExtra,
|
2021-05-17 06:50:45 -05:00
|
|
|
ptr: Pointer<Tag>,
|
2021-05-20 06:32:18 -05:00
|
|
|
size: Size,
|
2021-05-17 06:50:45 -05:00
|
|
|
) -> InterpResult<'tcx> {
|
|
|
|
if Some(ptr.alloc_id) == memory_extra.tracked_alloc_id {
|
|
|
|
register_diagnostic(NonHaltingDiagnostic::FreedAlloc(ptr.alloc_id));
|
|
|
|
}
|
2021-05-20 06:32:18 -05:00
|
|
|
if let Some(data_race) = &mut alloc_extra.data_race {
|
2021-05-22 07:47:14 -05:00
|
|
|
data_race.deallocate(ptr, size, memory_extra.data_race.as_mut().unwrap())?;
|
2021-05-17 06:50:45 -05:00
|
|
|
}
|
2021-05-20 06:32:18 -05:00
|
|
|
if let Some(stacked_borrows) = &mut alloc_extra.stacked_borrows {
|
2021-05-22 07:55:33 -05:00
|
|
|
stacked_borrows.memory_deallocated(
|
|
|
|
ptr,
|
|
|
|
size,
|
|
|
|
memory_extra.stacked_borrows.as_mut().unwrap(),
|
|
|
|
)
|
2021-05-17 06:50:45 -05:00
|
|
|
} else {
|
|
|
|
Ok(())
|
|
|
|
}
|
2020-04-14 18:00:56 -05:00
|
|
|
}
|
|
|
|
|
2020-12-11 13:32:25 -06:00
|
|
|
fn after_static_mem_initialized(
|
|
|
|
ecx: &mut InterpCx<'mir, 'tcx, Self>,
|
|
|
|
ptr: Pointer<Self::PointerTag>,
|
|
|
|
size: Size,
|
|
|
|
) -> InterpResult<'tcx> {
|
|
|
|
if ecx.memory.extra.data_race.is_some() {
|
|
|
|
ecx.reset_vector_clocks(ptr, size)?;
|
|
|
|
}
|
|
|
|
Ok(())
|
|
|
|
}
|
|
|
|
|
2019-06-27 16:59:00 -05:00
|
|
|
#[inline(always)]
|
2020-03-25 03:05:24 -05:00
|
|
|
fn tag_global_base_pointer(memory_extra: &MemoryExtra, id: AllocId) -> Self::PointerTag {
|
2020-04-12 03:32:36 -05:00
|
|
|
if let Some(stacked_borrows) = &memory_extra.stacked_borrows {
|
2020-03-25 03:05:24 -05:00
|
|
|
stacked_borrows.borrow_mut().global_base_ptr(id)
|
2019-07-03 03:19:55 -05:00
|
|
|
} else {
|
2020-02-24 09:22:02 -06:00
|
|
|
Tag::Untagged
|
2019-07-03 03:19:55 -05:00
|
|
|
}
|
2019-06-27 16:59:00 -05:00
|
|
|
}
|
|
|
|
|
|
|
|
#[inline(always)]
|
|
|
|
fn retag(
|
2019-07-05 16:47:10 -05:00
|
|
|
ecx: &mut InterpCx<'mir, 'tcx, Self>,
|
2019-06-27 16:59:00 -05:00
|
|
|
kind: mir::RetagKind,
|
2021-02-19 18:00:00 -06:00
|
|
|
place: &PlaceTy<'tcx, Tag>,
|
2019-06-27 16:59:00 -05:00
|
|
|
) -> InterpResult<'tcx> {
|
2021-05-16 04:28:01 -05:00
|
|
|
if ecx.memory.extra.stacked_borrows.is_some() { ecx.retag(kind, place) } else { Ok(()) }
|
2019-06-27 16:59:00 -05:00
|
|
|
}
|
|
|
|
|
|
|
|
#[inline(always)]
|
2020-04-13 09:08:12 -05:00
|
|
|
fn init_frame_extra(
|
|
|
|
ecx: &mut InterpCx<'mir, 'tcx, Self>,
|
|
|
|
frame: Frame<'mir, 'tcx, Tag>,
|
|
|
|
) -> InterpResult<'tcx, Frame<'mir, 'tcx, Tag, FrameData<'tcx>>> {
|
2021-05-29 17:09:46 -05:00
|
|
|
// Start recording our event before doing anything else
|
2021-05-08 11:20:51 -05:00
|
|
|
let timing = if let Some(profiler) = ecx.machine.profiler.as_ref() {
|
|
|
|
let fn_name = frame.instance.to_string();
|
|
|
|
let entry = ecx.machine.string_cache.entry(fn_name.clone());
|
2021-05-29 17:16:12 -05:00
|
|
|
let name = entry.or_insert_with(|| profiler.alloc_string(&*fn_name));
|
2021-05-08 11:20:51 -05:00
|
|
|
|
|
|
|
Some(profiler.start_recording_interval_event_detached(
|
2021-05-29 17:09:46 -05:00
|
|
|
*name,
|
2021-05-30 10:04:57 -05:00
|
|
|
measureme::EventId::from_label(*name),
|
2021-05-29 17:09:46 -05:00
|
|
|
ecx.get_active_thread().to_u32(),
|
2021-05-08 11:20:51 -05:00
|
|
|
))
|
|
|
|
} else {
|
|
|
|
None
|
|
|
|
};
|
|
|
|
|
2021-05-29 17:09:46 -05:00
|
|
|
let stacked_borrows = ecx.memory.extra.stacked_borrows.as_ref();
|
|
|
|
let call_id = stacked_borrows.map_or(NonZeroU64::new(1).unwrap(), |stacked_borrows| {
|
|
|
|
stacked_borrows.borrow_mut().new_call()
|
|
|
|
});
|
|
|
|
|
2021-05-08 11:20:51 -05:00
|
|
|
let extra = FrameData { call_id, catch_unwind: None, timing };
|
2020-04-13 09:08:12 -05:00
|
|
|
Ok(frame.with_extra(extra))
|
2019-06-27 16:59:00 -05:00
|
|
|
}
|
|
|
|
|
2020-04-14 19:21:52 -05:00
|
|
|
fn stack<'a>(
|
2021-05-16 04:28:01 -05:00
|
|
|
ecx: &'a InterpCx<'mir, 'tcx, Self>,
|
2020-04-14 19:21:52 -05:00
|
|
|
) -> &'a [Frame<'mir, 'tcx, Self::PointerTag, Self::FrameExtra>] {
|
|
|
|
ecx.active_thread_stack()
|
|
|
|
}
|
|
|
|
|
|
|
|
fn stack_mut<'a>(
|
2021-05-16 04:28:01 -05:00
|
|
|
ecx: &'a mut InterpCx<'mir, 'tcx, Self>,
|
2020-04-14 19:21:52 -05:00
|
|
|
) -> &'a mut Vec<Frame<'mir, 'tcx, Self::PointerTag, Self::FrameExtra>> {
|
|
|
|
ecx.active_thread_stack_mut()
|
|
|
|
}
|
|
|
|
|
2020-04-13 10:31:19 -05:00
|
|
|
#[inline(always)]
|
|
|
|
fn after_stack_push(ecx: &mut InterpCx<'mir, 'tcx, Self>) -> InterpResult<'tcx> {
|
2021-05-16 04:28:01 -05:00
|
|
|
if ecx.memory.extra.stacked_borrows.is_some() { ecx.retag_return_place() } else { Ok(()) }
|
2020-04-13 10:31:19 -05:00
|
|
|
}
|
|
|
|
|
2019-06-27 16:59:00 -05:00
|
|
|
#[inline(always)]
|
2020-04-13 09:08:12 -05:00
|
|
|
fn after_stack_pop(
|
2019-07-05 16:47:10 -05:00
|
|
|
ecx: &mut InterpCx<'mir, 'tcx, Self>,
|
2021-05-08 11:20:51 -05:00
|
|
|
mut frame: Frame<'mir, 'tcx, Tag, FrameData<'tcx>>,
|
2019-12-23 05:56:23 -06:00
|
|
|
unwinding: bool,
|
2020-03-14 05:53:09 -05:00
|
|
|
) -> InterpResult<'tcx, StackPopJump> {
|
2021-05-08 11:20:51 -05:00
|
|
|
let timing = frame.extra.timing.take();
|
|
|
|
let res = ecx.handle_stack_pop(frame.extra, unwinding);
|
|
|
|
if let Some(profiler) = ecx.machine.profiler.as_ref() {
|
|
|
|
profiler.finish_recording_interval_event(timing.unwrap());
|
|
|
|
}
|
|
|
|
res
|
2019-06-27 16:59:00 -05:00
|
|
|
}
|
|
|
|
|
2019-07-23 14:38:53 -05:00
|
|
|
#[inline(always)]
|
2019-06-27 16:59:00 -05:00
|
|
|
fn int_to_ptr(
|
|
|
|
memory: &Memory<'mir, 'tcx, Self>,
|
2019-07-06 02:51:20 -05:00
|
|
|
int: u64,
|
2019-06-27 16:59:00 -05:00
|
|
|
) -> InterpResult<'tcx, Pointer<Self::PointerTag>> {
|
2019-07-23 14:38:53 -05:00
|
|
|
intptrcast::GlobalState::int_to_ptr(int, memory)
|
2019-06-27 16:59:00 -05:00
|
|
|
}
|
|
|
|
|
2019-07-23 14:38:53 -05:00
|
|
|
#[inline(always)]
|
2019-06-27 16:59:00 -05:00
|
|
|
fn ptr_to_int(
|
|
|
|
memory: &Memory<'mir, 'tcx, Self>,
|
2019-07-06 02:51:20 -05:00
|
|
|
ptr: Pointer<Self::PointerTag>,
|
2019-06-27 16:59:00 -05:00
|
|
|
) -> InterpResult<'tcx, u64> {
|
2019-07-23 14:38:53 -05:00
|
|
|
intptrcast::GlobalState::ptr_to_int(ptr, memory)
|
2019-06-27 16:59:00 -05:00
|
|
|
}
|
|
|
|
}
|