place.rs 30.6 KB
Newer Older
1 2 3 4 5 6 7 8 9 10
// Copyright 2018 The Rust Project Developers. See the COPYRIGHT
// file at the top-level directory of this distribution and at
// http://rust-lang.org/COPYRIGHT.
//
// Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
// http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
// <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
// option. This file may not be copied, modified, or distributed
// except according to those terms.

R
Ralf Jung 已提交
11 12 13 14 15 16 17
//! Computations on places -- field projections, going from mir::Place, and writing
//! into a place.
//! All high-level functions to write to memory work on places as destinations.

use std::hash::{Hash, Hasher};
use std::convert::TryFrom;

18
use rustc::mir;
R
Ralf Jung 已提交
19
use rustc::ty::{self, Ty};
20
use rustc::ty::layout::{self, Size, Align, LayoutOf, TyLayout, HasDataLayout};
O
Oliver Schneider 已提交
21 22
use rustc_data_structures::indexed_vec::Idx;

R
Ralf Jung 已提交
23 24 25 26 27 28 29 30 31 32 33 34 35 36
use rustc::mir::interpret::{
    GlobalId, Scalar, EvalResult, Pointer, ScalarMaybeUndef
};
use super::{EvalContext, Machine, Value, ValTy, Operand, OpTy, MemoryKind};

#[derive(Copy, Clone, Debug, Hash, PartialEq, Eq)]
pub struct MemPlace {
    /// A place may have an integral pointer for ZSTs, and since it might
    /// be turned back into a reference before ever being dereferenced.
    /// However, it may never be undef.
    pub ptr: Scalar,
    pub align: Align,
    pub extra: PlaceExtra,
}
O
Oliver Schneider 已提交
37

38
#[derive(Copy, Clone, Debug, Hash, PartialEq, Eq)]
O
Oliver Schneider 已提交
39
pub enum Place {
40
    /// A place referring to a value allocated in the `Memory` system.
R
Ralf Jung 已提交
41
    Ptr(MemPlace),
O
Oliver Schneider 已提交
42

R
Ralf Jung 已提交
43 44 45 46 47 48
    /// To support alloc-free locals, we are able to write directly to a local.
    /// (Without that optimization, we'd just always be a `MemPlace`.)
    Local {
        frame: usize,
        local: mir::Local,
    },
O
Oliver Schneider 已提交
49 50
}

R
Ralf Jung 已提交
51
// Extra information for fat pointers / places
52
#[derive(Copy, Clone, Debug, Hash, PartialEq, Eq)]
O
Oliver Schneider 已提交
53 54 55
pub enum PlaceExtra {
    None,
    Length(u64),
56
    Vtable(Pointer),
O
Oliver Schneider 已提交
57 58
}

R
Ralf Jung 已提交
59 60 61 62 63 64 65 66
#[derive(Copy, Clone, Debug)]
pub struct PlaceTy<'tcx> {
    place: Place,
    pub layout: TyLayout<'tcx>,
}

impl<'tcx> ::std::ops::Deref for PlaceTy<'tcx> {
    type Target = Place;
67
    #[inline(always)]
R
Ralf Jung 已提交
68 69 70 71 72 73 74 75 76 77 78 79 80 81
    fn deref(&self) -> &Place {
        &self.place
    }
}

/// A MemPlace with its layout. Constructing it is only possible in this module.
#[derive(Copy, Clone, Debug)]
pub struct MPlaceTy<'tcx> {
    mplace: MemPlace,
    pub layout: TyLayout<'tcx>,
}

impl<'tcx> ::std::ops::Deref for MPlaceTy<'tcx> {
    type Target = MemPlace;
82
    #[inline(always)]
R
Ralf Jung 已提交
83 84
    fn deref(&self) -> &MemPlace {
        &self.mplace
O
Oliver Schneider 已提交
85
    }
R
Ralf Jung 已提交
86 87 88
}

impl<'tcx> From<MPlaceTy<'tcx>> for PlaceTy<'tcx> {
89
    #[inline(always)]
R
Ralf Jung 已提交
90 91 92 93 94 95 96
    fn from(mplace: MPlaceTy<'tcx>) -> Self {
        PlaceTy {
            place: Place::Ptr(mplace.mplace),
            layout: mplace.layout
        }
    }
}
O
Oliver Schneider 已提交
97

R
Ralf Jung 已提交
98 99 100 101
impl MemPlace {
    #[inline(always)]
    pub fn from_scalar_ptr(ptr: Scalar, align: Align) -> Self {
        MemPlace {
102 103
            ptr,
            align,
O
Oliver Schneider 已提交
104 105 106 107
            extra: PlaceExtra::None,
        }
    }

R
Ralf Jung 已提交
108 109 110 111 112 113 114 115 116 117 118 119 120 121
    #[inline(always)]
    pub fn from_ptr(ptr: Pointer, align: Align) -> Self {
        Self::from_scalar_ptr(ptr.into(), align)
    }

    #[inline(always)]
    pub fn to_scalar_ptr_align(self) -> (Scalar, Align) {
        assert_eq!(self.extra, PlaceExtra::None);
        (self.ptr, self.align)
    }

    /// Extract the ptr part of the mplace
    #[inline(always)]
    pub fn to_ptr(self) -> EvalResult<'tcx, Pointer> {
B
Bernardo Meurer 已提交
122 123 124
        // At this point, we forget about the alignment information --
        // the place has been turned into a reference, and no matter where it came from,
        // it now must be aligned.
R
Ralf Jung 已提交
125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176
        self.to_scalar_ptr_align().0.to_ptr()
    }

    /// Turn a mplace into a (thin or fat) pointer, as a reference, pointing to the same space.
    /// This is the inverse of `ref_to_mplace`.
    pub fn to_ref(self, cx: impl HasDataLayout) -> Value {
        // We ignore the alignment of the place here -- special handling for packed structs ends
        // at the `&` operator.
        match self.extra {
            PlaceExtra::None => Value::Scalar(self.ptr.into()),
            PlaceExtra::Length(len) => Value::new_slice(self.ptr.into(), len, cx),
            PlaceExtra::Vtable(vtable) => Value::new_dyn_trait(self.ptr.into(), vtable),
        }
    }
}

impl<'tcx> MPlaceTy<'tcx> {
    #[inline]
    fn from_aligned_ptr(ptr: Pointer, layout: TyLayout<'tcx>) -> Self {
        MPlaceTy { mplace: MemPlace::from_ptr(ptr, layout.align), layout }
    }

    #[inline]
    pub(super) fn len(self) -> u64 {
        // Sanity check
        let ty_len = match self.layout.fields {
            layout::FieldPlacement::Array { count, .. } => count,
            _ => bug!("Length for non-array layout {:?} requested", self.layout),
        };
        if let PlaceExtra::Length(len) = self.extra {
            len
        } else {
            ty_len
        }
    }
}

// Validation needs to hash MPlaceTy, but we cannot hash Layout -- so we just hash the type
impl<'tcx> Hash for MPlaceTy<'tcx> {
    fn hash<H: Hasher>(&self, state: &mut H) {
        self.mplace.hash(state);
        self.layout.ty.hash(state);
    }
}
impl<'tcx> PartialEq for MPlaceTy<'tcx> {
    fn eq(&self, other: &Self) -> bool {
        self.mplace == other.mplace && self.layout.ty == other.layout.ty
    }
}
impl<'tcx> Eq for MPlaceTy<'tcx> {}

impl<'tcx> OpTy<'tcx> {
177
    #[inline(always)]
R
Ralf Jung 已提交
178 179 180 181 182 183 184
    pub fn try_as_mplace(self) -> Result<MPlaceTy<'tcx>, Value> {
        match *self {
            Operand::Indirect(mplace) => Ok(MPlaceTy { mplace, layout: self.layout }),
            Operand::Immediate(value) => Err(value),
        }
    }

185
    #[inline(always)]
R
Ralf Jung 已提交
186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203
    pub fn to_mem_place(self) -> MPlaceTy<'tcx> {
        self.try_as_mplace().unwrap()
    }
}

impl<'tcx> Place {
    /// Produces a Place that will error if attempted to be read from or written to
    #[inline]
    pub fn null(cx: impl HasDataLayout) -> Self {
        Self::from_scalar_ptr(Scalar::ptr_null(cx), Align::from_bytes(1, 1).unwrap())
    }

    #[inline]
    pub fn from_scalar_ptr(ptr: Scalar, align: Align) -> Self {
        Place::Ptr(MemPlace::from_scalar_ptr(ptr, align))
    }

    #[inline]
204
    pub fn from_ptr(ptr: Pointer, align: Align) -> Self {
R
Ralf Jung 已提交
205
        Place::Ptr(MemPlace::from_ptr(ptr, align))
O
Oliver Schneider 已提交
206 207
    }

R
Ralf Jung 已提交
208 209
    #[inline]
    pub fn to_mem_place(self) -> MemPlace {
O
Oliver Schneider 已提交
210
        match self {
R
Ralf Jung 已提交
211 212
            Place::Ptr(mplace) => mplace,
            _ => bug!("to_mem_place: expected Place::Ptr, got {:?}", self),
O
Oliver Schneider 已提交
213 214 215 216

        }
    }

R
Ralf Jung 已提交
217 218 219
    #[inline]
    pub fn to_scalar_ptr_align(self) -> (Scalar, Align) {
        self.to_mem_place().to_scalar_ptr_align()
220
    }
221

R
Ralf Jung 已提交
222
    #[inline]
223
    pub fn to_ptr(self) -> EvalResult<'tcx, Pointer> {
R
Ralf Jung 已提交
224 225 226
        self.to_mem_place().to_ptr()
    }
}
O
Oliver Schneider 已提交
227

R
Ralf Jung 已提交
228 229 230 231 232 233 234 235 236 237
impl<'tcx> PlaceTy<'tcx> {
    /// Produces a Place that will error if attempted to be read from or written to
    #[inline]
    pub fn null(cx: impl HasDataLayout, layout: TyLayout<'tcx>) -> Self {
        PlaceTy { place: Place::from_scalar_ptr(Scalar::ptr_null(cx), layout.align), layout }
    }

    #[inline]
    pub fn to_mem_place(self) -> MPlaceTy<'tcx> {
        MPlaceTy { mplace: self.place.to_mem_place(), layout: self.layout }
O
Oliver Schneider 已提交
238 239 240
    }
}

O
Oliver Schneider 已提交
241
impl<'a, 'mir, 'tcx, M: Machine<'mir, 'tcx>> EvalContext<'a, 'mir, 'tcx, M> {
R
Ralf Jung 已提交
242 243 244 245 246 247 248 249
    /// Take a value, which represents a (thin or fat) reference, and make it a place.
    /// Alignment is just based on the type.  This is the inverse of `MemPlace::to_ref`.
    pub fn ref_to_mplace(
        &self, val: ValTy<'tcx>
    ) -> EvalResult<'tcx, MPlaceTy<'tcx>> {
        let pointee_type = val.layout.ty.builtin_deref(true).unwrap().ty;
        let layout = self.layout_of(pointee_type)?;
        let mplace = match self.tcx.struct_tail(pointee_type).sty {
R
Ralf Jung 已提交
250 251
            // Matching on the type is okay here, because we used `struct_tail` to get to
            // the "core" of what makes this unsized.
V
varkor 已提交
252
            ty::Dynamic(..) => {
R
Ralf Jung 已提交
253 254 255 256 257 258 259
                let (ptr, vtable) = val.to_scalar_dyn_trait()?;
                MemPlace {
                    ptr,
                    align: layout.align,
                    extra: PlaceExtra::Vtable(vtable),
                }
            }
V
varkor 已提交
260
            ty::Str | ty::Slice(_) => {
R
Ralf Jung 已提交
261 262 263 264 265 266 267
                let (ptr, len) = val.to_scalar_slice(self)?;
                MemPlace {
                    ptr,
                    align: layout.align,
                    extra: PlaceExtra::Length(len),
                }
            }
R
Ralf Jung 已提交
268 269 270 271 272 273 274 275
            _ => {
                assert!(!layout.is_unsized(), "Unhandled unsized type {:?}", pointee_type);
                MemPlace {
                    ptr: val.to_scalar()?,
                    align: layout.align,
                    extra: PlaceExtra::None,
                }
            }
R
Ralf Jung 已提交
276 277 278 279 280 281 282 283 284
        };
        Ok(MPlaceTy { mplace, layout })
    }

    /// Offset a pointer to project to a field. Unlike place_field, this is always
    /// possible without allocating, so it can take &self. Also return the field's layout.
    /// This supports both struct and array fields.
    #[inline(always)]
    pub fn mplace_field(
O
Oliver Schneider 已提交
285
        &self,
R
Ralf Jung 已提交
286 287 288 289 290 291 292 293 294
        base: MPlaceTy<'tcx>,
        field: u64,
    ) -> EvalResult<'tcx, MPlaceTy<'tcx>> {
        // Not using the layout method because we want to compute on u64
        let offset = match base.layout.fields {
            layout::FieldPlacement::Arbitrary { ref offsets, .. } =>
                offsets[usize::try_from(field).unwrap()],
            layout::FieldPlacement::Array { stride, .. } => {
                let len = base.len();
B
Bernardo Meurer 已提交
295 296
                assert!(field < len,
                        "Tried to access element {} of array/slice with length {}", field, len);
R
Ralf Jung 已提交
297 298
                stride * field
            }
299
            layout::FieldPlacement::Union(count) => {
B
Bernardo Meurer 已提交
300 301
                assert!(field < count as u64,
                        "Tried to access field {} of union with {} fields", field, count);
302 303 304
                // Offset is always 0
                Size::from_bytes(0)
            }
R
Ralf Jung 已提交
305 306 307
        };
        // the only way conversion can fail if is this is an array (otherwise we already panicked
        // above). In that case, all fields are equal.
R
Ralf Jung 已提交
308
        let field_layout = base.layout.field(self, usize::try_from(field).unwrap_or(0))?;
R
Ralf Jung 已提交
309 310

        // Adjust offset
R
Ralf Jung 已提交
311 312 313 314 315 316 317 318 319 320 321
        let offset = match base.extra {
            PlaceExtra::Vtable(vtable) => {
                let (_, align) = self.read_size_and_align_from_vtable(vtable)?;
                // FIXME: Is this right? Should we always do this, or only when actually
                // accessing the field to which the vtable applies?
                offset.abi_align(align)
            }
            _ => {
                // No adjustment needed
                offset
            }
R
Ralf Jung 已提交
322 323 324
        };

        let ptr = base.ptr.ptr_offset(offset, self)?;
R
Ralf Jung 已提交
325 326
        let align = base.align.min(field_layout.align);
        let extra = if !field_layout.is_unsized() {
R
Ralf Jung 已提交
327 328 329 330 331 332
            PlaceExtra::None
        } else {
            assert!(base.extra != PlaceExtra::None, "Expected fat ptr");
            base.extra
        };

R
Ralf Jung 已提交
333
        Ok(MPlaceTy { mplace: MemPlace { ptr, align, extra }, layout: field_layout })
O
Oliver Schneider 已提交
334 335
    }

336 337 338 339 340 341 342 343 344 345 346 347 348 349 350 351 352 353 354 355 356 357
    // Iterates over all fields of an array. Much more efficient than doing the
    // same by repeatedly calling `mplace_array`.
    pub fn mplace_array_fields(
        &self,
        base: MPlaceTy<'tcx>,
    ) -> EvalResult<'tcx, impl Iterator<Item=EvalResult<'tcx, MPlaceTy<'tcx>>> + 'a> {
        let len = base.len();
        let stride = match base.layout.fields {
            layout::FieldPlacement::Array { stride, .. } => stride,
            _ => bug!("mplace_array_fields: expected an array layout"),
        };
        let layout = base.layout.field(self, 0)?;
        let dl = &self.tcx.data_layout;
        Ok((0..len).map(move |i| {
            let ptr = base.ptr.ptr_offset(i * stride, dl)?;
            Ok(MPlaceTy {
                mplace: MemPlace { ptr, align: base.align, extra: PlaceExtra::None },
                layout
            })
        }))
    }

R
Ralf Jung 已提交
358
    pub fn mplace_subslice(
O
Oliver Schneider 已提交
359
        &self,
R
Ralf Jung 已提交
360 361 362 363 364 365 366 367 368 369 370 371 372
        base: MPlaceTy<'tcx>,
        from: u64,
        to: u64,
    ) -> EvalResult<'tcx, MPlaceTy<'tcx>> {
        let len = base.len();
        assert!(from <= len - to);

        // Not using layout method because that works with usize, and does not work with slices
        // (that have count 0 in their layout).
        let from_offset = match base.layout.fields {
            layout::FieldPlacement::Array { stride, .. } =>
                stride * from,
            _ => bug!("Unexpected layout of index access: {:#?}", base.layout),
O
Oliver Schneider 已提交
373
        };
R
Ralf Jung 已提交
374 375 376 377 378
        let ptr = base.ptr.ptr_offset(from_offset, self)?;

        // Compute extra and new layout
        let inner_len = len - to - from;
        let (extra, ty) = match base.layout.ty.sty {
R
Ralf Jung 已提交
379 380
            // It is not nice to match on the type, but that seems to be the only way to
            // implement this.
V
varkor 已提交
381
            ty::Array(inner, _) =>
R
Ralf Jung 已提交
382
                (PlaceExtra::None, self.tcx.mk_array(inner, inner_len)),
V
varkor 已提交
383
            ty::Slice(..) =>
R
Ralf Jung 已提交
384 385 386 387 388 389 390 391 392 393 394 395 396 397 398 399 400 401 402 403
                (PlaceExtra::Length(inner_len), base.layout.ty),
            _ =>
                bug!("cannot subslice non-array type: `{:?}`", base.layout.ty),
        };
        let layout = self.layout_of(ty)?;

        Ok(MPlaceTy {
            mplace: MemPlace { ptr, align: base.align, extra },
            layout
        })
    }

    pub fn mplace_downcast(
        &self,
        base: MPlaceTy<'tcx>,
        variant: usize,
    ) -> EvalResult<'tcx, MPlaceTy<'tcx>> {
        // Downcasts only change the layout
        assert_eq!(base.extra, PlaceExtra::None);
        Ok(MPlaceTy { layout: base.layout.for_variant(self, variant), ..base })
O
Oliver Schneider 已提交
404 405
    }

R
Ralf Jung 已提交
406 407
    /// Project into an mplace
    pub fn mplace_projection(
O
Oliver Schneider 已提交
408
        &self,
R
Ralf Jung 已提交
409 410 411
        base: MPlaceTy<'tcx>,
        proj_elem: &mir::PlaceElem<'tcx>,
    ) -> EvalResult<'tcx, MPlaceTy<'tcx>> {
412
        use rustc::mir::ProjectionElem::*;
R
Ralf Jung 已提交
413 414 415 416 417 418 419 420 421 422 423 424 425 426 427 428 429 430 431 432 433 434 435 436 437 438 439 440 441 442 443 444 445
        Ok(match *proj_elem {
            Field(field, _) => self.mplace_field(base, field.index() as u64)?,
            Downcast(_, variant) => self.mplace_downcast(base, variant)?,
            Deref => self.deref_operand(base.into())?,

            Index(local) => {
                let n = *self.frame().locals[local].access()?;
                let n_layout = self.layout_of(self.tcx.types.usize)?;
                let n = self.read_scalar(OpTy { op: n, layout: n_layout })?;
                let n = n.to_bits(self.tcx.data_layout.pointer_size)?;
                self.mplace_field(base, u64::try_from(n).unwrap())?
            }

            ConstantIndex {
                offset,
                min_length,
                from_end,
            } => {
                let n = base.len();
                assert!(n >= min_length as u64);

                let index = if from_end {
                    n - u64::from(offset)
                } else {
                    u64::from(offset)
                };

                self.mplace_field(base, index)?
            }

            Subslice { from, to } =>
                self.mplace_subslice(base, u64::from(from), u64::from(to))?,
        })
O
Oliver Schneider 已提交
446 447
    }

R
Ralf Jung 已提交
448 449 450
    /// Get the place of a field inside the place, and also the field's type.
    /// Just a convenience function, but used quite a bit.
    pub fn place_field(
O
Oliver Schneider 已提交
451
        &mut self,
R
Ralf Jung 已提交
452
        base: PlaceTy<'tcx>,
R
Ralf Jung 已提交
453 454 455 456 457 458
        field: u64,
    ) -> EvalResult<'tcx, PlaceTy<'tcx>> {
        // FIXME: We could try to be smarter and avoid allocation for fields that span the
        // entire place.
        let mplace = self.force_allocation(base)?;
        Ok(self.mplace_field(mplace, field)?.into())
O
Oliver Schneider 已提交
459 460
    }

R
Ralf Jung 已提交
461 462
    pub fn place_downcast(
        &mut self,
R
Ralf Jung 已提交
463
        base: PlaceTy<'tcx>,
R
Ralf Jung 已提交
464 465 466 467 468 469 470 471 472
        variant: usize,
    ) -> EvalResult<'tcx, PlaceTy<'tcx>> {
        // Downcast just changes the layout
        Ok(match base.place {
            Place::Ptr(mplace) =>
                self.mplace_downcast(MPlaceTy { mplace, layout: base.layout }, variant)?.into(),
            Place::Local { .. } => {
                let layout = base.layout.for_variant(&self, variant);
                PlaceTy { layout, ..base }
O
Oliver Schneider 已提交
473
            }
R
Ralf Jung 已提交
474
        })
O
Oliver Schneider 已提交
475 476
    }

R
Ralf Jung 已提交
477 478 479 480 481 482 483 484 485 486 487 488 489 490 491 492 493 494 495 496
    /// Project into a place
    pub fn place_projection(
        &mut self,
        base: PlaceTy<'tcx>,
        proj_elem: &mir::ProjectionElem<'tcx, mir::Local, Ty<'tcx>>,
    ) -> EvalResult<'tcx, PlaceTy<'tcx>> {
        use rustc::mir::ProjectionElem::*;
        Ok(match *proj_elem {
            Field(field, _) =>  self.place_field(base, field.index() as u64)?,
            Downcast(_, variant) => self.place_downcast(base, variant)?,
            Deref => self.deref_operand(self.place_to_op(base)?)?.into(),
            // For the other variants, we have to force an allocation.
            // This matches `operand_projection`.
            Subslice { .. } | ConstantIndex { .. } | Index(_) => {
                let mplace = self.force_allocation(base)?;
                self.mplace_projection(mplace, proj_elem)?.into()
            }
        })
    }

497 498 499 500 501 502
    /// Evaluate statics and promoteds to an `MPlace`.  Used to share some code between
    /// `eval_place` and `eval_place_to_op`.
    pub(super) fn eval_place_to_mplace(
        &self,
        mir_place: &mir::Place<'tcx>
    ) -> EvalResult<'tcx, MPlaceTy<'tcx>> {
503
        use rustc::mir::Place::*;
504
        Ok(match *mir_place {
505 506
            Promoted(ref promoted) => {
                let instance = self.frame().instance;
R
Ralf Jung 已提交
507
                let op = self.global_to_op(GlobalId {
508 509 510
                    instance,
                    promoted: Some(promoted.0),
                })?;
511
                let mplace = op.to_mem_place(); // these are always in memory
R
Ralf Jung 已提交
512
                let ty = self.monomorphize(promoted.1, self.substs());
513 514
                MPlaceTy {
                    mplace,
R
Ralf Jung 已提交
515
                    layout: self.layout_of(ty)?,
516 517 518
                }
            }

O
Oliver Schneider 已提交
519
            Static(ref static_) => {
R
Ralf Jung 已提交
520 521
                let ty = self.monomorphize(static_.ty, self.substs());
                let layout = self.layout_of(ty)?;
522 523 524 525 526
                let instance = ty::Instance::mono(*self.tcx, static_.def_id);
                let cid = GlobalId {
                    instance,
                    promoted: None
                };
R
Ralf Jung 已提交
527 528 529 530 531 532
                // Just create a lazy reference, so we can support recursive statics.
                // When the data here is ever actually used, memory will notice,
                // and it knows how to deal with alloc_id that are present in the
                // global table but not in its local memory.
                let alloc = self.tcx.alloc_map.lock()
                    .intern_static(cid.instance.def_id());
533
                MPlaceTy::from_aligned_ptr(alloc.into(), layout)
O
Oliver Schneider 已提交
534 535
            }

536 537 538 539 540 541 542 543 544 545 546 547 548 549 550 551 552 553 554 555 556
            _ => bug!("eval_place_to_mplace called on {:?}", mir_place),
        })
    }

    /// Compute a place.  You should only use this if you intend to write into this
    /// place; for reading, a more efficient alternative is `eval_place_for_read`.
    pub fn eval_place(&mut self, mir_place: &mir::Place<'tcx>) -> EvalResult<'tcx, PlaceTy<'tcx>> {
        use rustc::mir::Place::*;
        let place = match *mir_place {
            Local(mir::RETURN_PLACE) => PlaceTy {
                place: self.frame().return_place,
                layout: self.layout_of_local(self.cur_frame(), mir::RETURN_PLACE)?,
            },
            Local(local) => PlaceTy {
                place: Place::Local {
                    frame: self.cur_frame(),
                    local,
                },
                layout: self.layout_of_local(self.cur_frame(), local)?,
            },

O
Oliver Schneider 已提交
557 558
            Projection(ref proj) => {
                let place = self.eval_place(&proj.base)?;
R
Ralf Jung 已提交
559
                self.place_projection(place, &proj.elem)?
O
Oliver Schneider 已提交
560
            }
561 562

            _ => self.eval_place_to_mplace(mir_place)?.into(),
O
Oliver Schneider 已提交
563 564
        };

R
Ralf Jung 已提交
565
        self.dump_place(place.place);
O
Oliver Schneider 已提交
566 567 568
        Ok(place)
    }

R
Ralf Jung 已提交
569 570
    /// Write a scalar to a place
    pub fn write_scalar(
O
Oliver Schneider 已提交
571
        &mut self,
R
Ralf Jung 已提交
572 573 574 575 576
        val: impl Into<ScalarMaybeUndef>,
        dest: PlaceTy<'tcx>,
    ) -> EvalResult<'tcx> {
        self.write_value(Value::Scalar(val.into()), dest)
    }
O
Oliver Schneider 已提交
577

R
Ralf Jung 已提交
578 579 580 581
    /// Write a value to a place
    pub fn write_value(
        &mut self,
        src_val: Value,
R
Ralf Jung 已提交
582
        dest: PlaceTy<'tcx>,
R
Ralf Jung 已提交
583
    ) -> EvalResult<'tcx> {
R
Ralf Jung 已提交
584
        trace!("write_value: {:?} <- {:?}", *dest, src_val);
R
Ralf Jung 已提交
585 586
        // See if we can avoid an allocation. This is the counterpart to `try_read_value`,
        // but not factored as a separate function.
R
Ralf Jung 已提交
587
        let mplace = match dest.place {
O
Oliver Schneider 已提交
588
            Place::Local { frame, local } => {
R
Ralf Jung 已提交
589 590 591 592 593
                match *self.stack[frame].locals[local].access_mut()? {
                    Operand::Immediate(ref mut dest_val) => {
                        // Yay, we can just change the local directly.
                        *dest_val = src_val;
                        return Ok(());
594
                    },
R
Ralf Jung 已提交
595
                    Operand::Indirect(mplace) => mplace, // already in memory
O
Oliver Schneider 已提交
596
                }
R
Ralf Jung 已提交
597
            },
R
Ralf Jung 已提交
598
            Place::Ptr(mplace) => mplace, // already in memory
O
Oliver Schneider 已提交
599 600
        };

R
Ralf Jung 已提交
601 602
        // This is already in memory, write there.
        let dest = MPlaceTy { mplace, layout: dest.layout };
R
Ralf Jung 已提交
603
        self.write_value_to_mplace(src_val, dest)
O
Oliver Schneider 已提交
604 605
    }

R
Ralf Jung 已提交
606 607 608 609 610 611
    /// Write a value to memory
    fn write_value_to_mplace(
        &mut self,
        value: Value,
        dest: MPlaceTy<'tcx>,
    ) -> EvalResult<'tcx> {
612
        let (ptr, ptr_align) = dest.to_scalar_ptr_align();
B
Bernardo Meurer 已提交
613 614 615 616
        // Note that it is really important that the type here is the right one, and matches the
        // type things are read at. In case `src_val` is a `ScalarPair`, we don't do any magic here
        // to handle padding properly, which is only correct if we never look at this data with the
        // wrong type.
617 618 619 620 621 622 623 624

        // Nothing to do for ZSTs, other than checking alignment
        if dest.layout.size.bytes() == 0 {
            self.memory.check_align(ptr, ptr_align)?;
            return Ok(());
        }

        let ptr = ptr.to_ptr()?;
R
Ralf Jung 已提交
625 626 627
        match value {
            Value::Scalar(scalar) => {
                self.memory.write_scalar(
628
                    ptr, ptr_align.min(dest.layout.align), scalar, dest.layout.size
R
Ralf Jung 已提交
629
                )
O
Oliver Schneider 已提交
630
            }
R
Ralf Jung 已提交
631 632 633
            Value::ScalarPair(a_val, b_val) => {
                let (a, b) = match dest.layout.abi {
                    layout::Abi::ScalarPair(ref a, ref b) => (&a.value, &b.value),
B
Bernardo Meurer 已提交
634 635
                    _ => bug!("write_value_to_mplace: invalid ScalarPair layout: {:#?}",
                              dest.layout)
R
Ralf Jung 已提交
636 637 638 639
                };
                let (a_size, b_size) = (a.size(&self), b.size(&self));
                let (a_align, b_align) = (a.align(&self), b.align(&self));
                let b_offset = a_size.abi_align(b_align);
640
                let b_ptr = ptr.offset(b_offset, &self)?.into();
641

642 643
                self.memory.write_scalar(ptr, ptr_align.min(a_align), a_val, a_size)?;
                self.memory.write_scalar(b_ptr, ptr_align.min(b_align), b_val, b_size)
O
Oliver Schneider 已提交
644
            }
R
Ralf Jung 已提交
645
        }
O
Oliver Schneider 已提交
646 647
    }

R
Ralf Jung 已提交
648 649
    /// Copy the data from an operand to a place
    pub fn copy_op(
O
Oliver Schneider 已提交
650
        &mut self,
R
Ralf Jung 已提交
651 652 653
        src: OpTy<'tcx>,
        dest: PlaceTy<'tcx>,
    ) -> EvalResult<'tcx> {
R
Ralf Jung 已提交
654 655
        assert_eq!(src.layout.size, dest.layout.size,
            "Size mismatch when copying!\nsrc: {:#?}\ndest: {:#?}", src, dest);
R
Ralf Jung 已提交
656 657 658 659

        // Let us see if the layout is simple so we take a shortcut, avoid force_allocation.
        let (src_ptr, src_align) = match self.try_read_value(src)? {
            Ok(src_val) =>
R
Ralf Jung 已提交
660 661 662 663
                // Yay, we got a value that we can write directly.  We write with the
                // *source layout*, because that was used to load, and if they do not match
                // this is a transmute we want to support.
                return self.write_value(src_val, PlaceTy { place: *dest, layout: src.layout }),
R
Ralf Jung 已提交
664 665 666
            Err(mplace) => mplace.to_scalar_ptr_align(),
        };
        // Slow path, this does not fit into an immediate. Just memcpy.
R
Ralf Jung 已提交
667
        trace!("copy_op: {:?} <- {:?}", *dest, *src);
R
Ralf Jung 已提交
668 669 670 671 672 673
        let (dest_ptr, dest_align) = self.force_allocation(dest)?.to_scalar_ptr_align();
        self.memory.copy(
            src_ptr, src_align,
            dest_ptr, dest_align,
            src.layout.size, false
        )
O
Oliver Schneider 已提交
674 675
    }

R
Ralf Jung 已提交
676 677 678
    /// Make sure that a place is in memory, and return where it is.
    /// This is essentially `force_to_memplace`.
    pub fn force_allocation(
O
Oliver Schneider 已提交
679
        &mut self,
R
Ralf Jung 已提交
680 681 682 683
        place: PlaceTy<'tcx>,
    ) -> EvalResult<'tcx, MPlaceTy<'tcx>> {
        let mplace = match place.place {
            Place::Local { frame, local } => {
684 685 686
                // FIXME: Consider not doing anything for a ZST, and just returning
                // a fake pointer?

R
Ralf Jung 已提交
687 688 689 690 691 692 693 694 695 696 697 698 699 700 701 702
                // We need the layout of the local.  We can NOT use the layout we got,
                // that might e.g. be a downcast variant!
                let local_layout = self.layout_of_local(frame, local)?;
                // Make sure it has a place
                let rval = *self.stack[frame].locals[local].access()?;
                let mplace = self.allocate_op(OpTy { op: rval, layout: local_layout })?.mplace;
                // This might have allocated the flag
                *self.stack[frame].locals[local].access_mut()? =
                    Operand::Indirect(mplace);
                // done
                mplace
            }
            Place::Ptr(mplace) => mplace
        };
        // Return with the original layout, so that the caller can go on
        Ok(MPlaceTy { mplace, layout: place.layout })
O
Oliver Schneider 已提交
703 704
    }

R
Ralf Jung 已提交
705
    pub fn allocate(
O
Oliver Schneider 已提交
706
        &mut self,
R
Ralf Jung 已提交
707 708 709 710 711 712 713
        layout: TyLayout<'tcx>,
        kind: MemoryKind<M::MemoryKinds>,
    ) -> EvalResult<'tcx, MPlaceTy<'tcx>> {
        assert!(!layout.is_unsized(), "cannot alloc memory for unsized type");
        let ptr = self.memory.allocate(layout.size, layout.align, kind)?;
        Ok(MPlaceTy::from_aligned_ptr(ptr, layout))
    }
O
Oliver Schneider 已提交
714

R
Ralf Jung 已提交
715 716 717 718 719
    /// Make a place for an operand, allocating if needed
    pub fn allocate_op(
        &mut self,
        OpTy { op, layout }: OpTy<'tcx>,
    ) -> EvalResult<'tcx, MPlaceTy<'tcx>> {
R
Ralf Jung 已提交
720
        trace!("allocate_op: {:?}", op);
R
Ralf Jung 已提交
721 722 723 724 725 726 727 728 729 730
        Ok(match op {
            Operand::Indirect(mplace) => MPlaceTy { mplace, layout },
            Operand::Immediate(value) => {
                // FIXME: Is stack always right here?
                let ptr = self.allocate(layout, MemoryKind::Stack)?;
                self.write_value_to_mplace(value, ptr)?;
                ptr
            },
        })
    }
O
Oliver Schneider 已提交
731

R
Ralf Jung 已提交
732 733 734 735 736 737 738 739 740 741 742 743 744
    pub fn write_discriminant_value(
        &mut self,
        variant_index: usize,
        dest: PlaceTy<'tcx>,
    ) -> EvalResult<'tcx> {
        match dest.layout.variants {
            layout::Variants::Single { index } => {
                if index != variant_index {
                    // If the layout of an enum is `Single`, all
                    // other variants are necessarily uninhabited.
                    assert_eq!(dest.layout.for_variant(&self, variant_index).abi,
                               layout::Abi::Uninhabited);
                }
O
Oliver Schneider 已提交
745
            }
R
Ralf Jung 已提交
746 747 748 749 750 751 752 753 754 755 756 757 758 759 760 761 762
            layout::Variants::Tagged { ref tag, .. } => {
                let discr_val = dest.layout.ty.ty_adt_def().unwrap()
                    .discriminant_for_variant(*self.tcx, variant_index)
                    .val;

                // raw discriminants for enums are isize or bigger during
                // their computation, but the in-memory tag is the smallest possible
                // representation
                let size = tag.value.size(self.tcx.tcx);
                let shift = 128 - size.bits();
                let discr_val = (discr_val << shift) >> shift;

                let discr_dest = self.place_field(dest, 0)?;
                self.write_scalar(Scalar::Bits {
                    bits: discr_val,
                    size: size.bytes() as u8,
                }, discr_dest)?;
O
Oliver Schneider 已提交
763
            }
R
Ralf Jung 已提交
764 765 766 767 768
            layout::Variants::NicheFilling {
                dataful_variant,
                ref niche_variants,
                niche_start,
                ..
O
Oliver Schneider 已提交
769
            } => {
R
Ralf Jung 已提交
770 771 772 773 774 775 776 777 778 779
                if variant_index != dataful_variant {
                    let niche_dest =
                        self.place_field(dest, 0)?;
                    let niche_value = ((variant_index - niche_variants.start()) as u128)
                        .wrapping_add(niche_start);
                    self.write_scalar(Scalar::Bits {
                        bits: niche_value,
                        size: niche_dest.layout.size.bytes() as u8,
                    }, niche_dest)?;
                }
O
Oliver Schneider 已提交
780
            }
781
        }
R
Ralf Jung 已提交
782 783

        Ok(())
O
Oliver Schneider 已提交
784 785
    }

R
Ralf Jung 已提交
786
    /// Every place can be read from, so we can turm them into an operand
787
    #[inline(always)]
R
Ralf Jung 已提交
788 789 790 791 792 793 794 795 796
    pub fn place_to_op(&self, place: PlaceTy<'tcx>) -> EvalResult<'tcx, OpTy<'tcx>> {
        let op = match place.place {
            Place::Ptr(mplace) => {
                Operand::Indirect(mplace)
            }
            Place::Local { frame, local } =>
                *self.stack[frame].locals[local].access()?
        };
        Ok(OpTy { op, layout: place.layout })
O
Oliver Schneider 已提交
797
    }
798 799 800 801

    /// Turn a place that is a dyn trait (i.e., PlaceExtra::Vtable and the appropriate layout)
    /// or a slice into the specific fixed-size place and layout that is given by the vtable/len.
    /// This "unpacks" the existential quantifier, so to speak.
B
Bernardo Meurer 已提交
802 803 804
    pub fn unpack_unsized_mplace(
        &self,
        mplace: MPlaceTy<'tcx>
805
    ) -> EvalResult<'tcx, MPlaceTy<'tcx>> {
806 807 808 809 810 811 812 813 814 815 816 817 818 819 820
        trace!("Unpacking {:?} ({:?})", *mplace, mplace.layout.ty);
        let layout = match mplace.extra {
            PlaceExtra::Vtable(vtable) => {
                // the drop function signature
                let drop_instance = self.read_drop_type_from_vtable(vtable)?;
                trace!("Found drop fn: {:?}", drop_instance);
                let fn_sig = drop_instance.ty(*self.tcx).fn_sig(*self.tcx);
                let fn_sig = self.tcx.normalize_erasing_late_bound_regions(self.param_env, &fn_sig);
                // the drop function takes *mut T where T is the type being dropped, so get that
                let ty = fn_sig.inputs()[0].builtin_deref(true).unwrap().ty;
                let layout = self.layout_of(ty)?;
                // Sanity checks
                let (size, align) = self.read_size_and_align_from_vtable(vtable)?;
                assert_eq!(size, layout.size);
                assert_eq!(align.abi(), layout.align.abi()); // only ABI alignment is preserved
821 822
                // FIXME: More checks for the vtable? We could make sure it is exactly
                // the one one would expect for this type.
823 824 825 826 827 828 829 830 831 832 833 834 835 836 837
                // Done!
                layout
            },
            PlaceExtra::Length(len) => {
                let ty = self.tcx.mk_array(mplace.layout.field(self, 0)?.ty, len);
                self.layout_of(ty)?
            }
            PlaceExtra::None => bug!("Expected a fat pointer"),
        };
        trace!("Unpacked type: {:?}", layout.ty);
        Ok(MPlaceTy {
            mplace: MemPlace { extra: PlaceExtra::None, ..*mplace },
            layout
        })
    }
O
Oliver Schneider 已提交
838
}