2018-07-14 05:21:45 -05:00
|
|
|
use std::fmt;
|
2018-07-14 04:59:42 -05:00
|
|
|
|
2018-09-17 11:49:10 -05:00
|
|
|
use crate::rustc_target::spec::{HasTargetSpec, Target};
|
2018-06-22 12:18:53 -05:00
|
|
|
|
2018-08-13 05:13:43 -05:00
|
|
|
use cranelift_module::Module;
|
2018-06-22 12:18:53 -05:00
|
|
|
|
2018-07-30 09:57:40 -05:00
|
|
|
use crate::prelude::*;
|
2018-06-22 12:18:53 -05:00
|
|
|
|
2018-08-15 05:36:13 -05:00
|
|
|
pub fn mir_var(loc: Local) -> Variable {
|
|
|
|
Variable::with_u32(loc.index() as u32)
|
2018-06-22 12:18:53 -05:00
|
|
|
}
|
|
|
|
|
2018-08-18 10:10:02 -05:00
|
|
|
pub fn pointer_ty(tcx: TyCtxt) -> types::Type {
|
|
|
|
match tcx.data_layout.pointer_size.bits() {
|
|
|
|
16 => types::I16,
|
|
|
|
32 => types::I32,
|
|
|
|
64 => types::I64,
|
|
|
|
bits => bug!("ptr_sized_integer: unknown pointer bit size {}", bits),
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-08-22 08:38:10 -05:00
|
|
|
fn scalar_to_cton_type(tcx: TyCtxt, scalar: &Scalar) -> Type {
|
2018-11-05 11:03:47 -06:00
|
|
|
match scalar.value.size(&tcx).bits() {
|
2018-08-22 08:38:10 -05:00
|
|
|
8 => types::I8,
|
|
|
|
16 => types::I16,
|
|
|
|
32 => types::I32,
|
|
|
|
64 => types::I64,
|
|
|
|
size => bug!("Unsupported scalar size {}", size),
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-07-31 05:25:16 -05:00
|
|
|
pub fn cton_type_from_ty<'a, 'tcx: 'a>(
|
|
|
|
tcx: TyCtxt<'a, 'tcx, 'tcx>,
|
|
|
|
ty: Ty<'tcx>,
|
|
|
|
) -> Option<types::Type> {
|
2018-06-22 12:18:53 -05:00
|
|
|
Some(match ty.sty {
|
2018-08-24 07:51:02 -05:00
|
|
|
ty::Bool => types::I8,
|
|
|
|
ty::Uint(size) => match size {
|
2018-07-31 05:25:16 -05:00
|
|
|
UintTy::U8 => types::I8,
|
|
|
|
UintTy::U16 => types::I16,
|
|
|
|
UintTy::U32 => types::I32,
|
|
|
|
UintTy::U64 => types::I64,
|
2018-08-09 08:08:54 -05:00
|
|
|
UintTy::U128 => unimpl!("u128"),
|
2018-08-18 10:10:02 -05:00
|
|
|
UintTy::Usize => pointer_ty(tcx),
|
2018-07-31 05:25:16 -05:00
|
|
|
},
|
2018-08-24 07:51:02 -05:00
|
|
|
ty::Int(size) => match size {
|
2018-07-31 05:25:16 -05:00
|
|
|
IntTy::I8 => types::I8,
|
|
|
|
IntTy::I16 => types::I16,
|
|
|
|
IntTy::I32 => types::I32,
|
|
|
|
IntTy::I64 => types::I64,
|
2018-08-09 08:08:54 -05:00
|
|
|
IntTy::I128 => unimpl!("i128"),
|
2018-08-22 05:31:45 -05:00
|
|
|
IntTy::Isize => pointer_ty(tcx),
|
2018-07-31 05:25:16 -05:00
|
|
|
},
|
2018-08-24 07:51:02 -05:00
|
|
|
ty::Char => types::I32,
|
|
|
|
ty::Float(size) => match size {
|
2018-08-08 03:26:25 -05:00
|
|
|
FloatTy::F32 => types::F32,
|
|
|
|
FloatTy::F64 => types::F64,
|
2018-07-31 05:25:16 -05:00
|
|
|
},
|
2018-08-24 07:51:02 -05:00
|
|
|
ty::FnPtr(_) => pointer_ty(tcx),
|
|
|
|
ty::RawPtr(TypeAndMut { ty, mutbl: _ }) | ty::Ref(_, ty, _) => {
|
2018-07-14 09:39:49 -05:00
|
|
|
if ty.is_sized(tcx.at(DUMMY_SP), ParamEnv::reveal_all()) {
|
2018-08-18 10:10:02 -05:00
|
|
|
pointer_ty(tcx)
|
2018-07-14 09:39:49 -05:00
|
|
|
} else {
|
|
|
|
return None;
|
|
|
|
}
|
|
|
|
}
|
2018-08-24 07:51:02 -05:00
|
|
|
ty::Param(_) => bug!("{:?}: {:?}", ty, ty.sty),
|
2018-06-22 12:18:53 -05:00
|
|
|
_ => return None,
|
|
|
|
})
|
|
|
|
}
|
|
|
|
|
2018-10-07 04:18:08 -05:00
|
|
|
pub fn codegen_select(bcx: &mut FunctionBuilder, cond: Value, lhs: Value, rhs: Value) -> Value {
|
|
|
|
let lhs_ty = bcx.func.dfg.value_type(lhs);
|
|
|
|
let rhs_ty = bcx.func.dfg.value_type(rhs);
|
|
|
|
assert_eq!(lhs_ty, rhs_ty);
|
|
|
|
if lhs_ty == types::I8 || lhs_ty == types::I16 {
|
|
|
|
// FIXME workaround for missing enocding for select.i8
|
|
|
|
let lhs = bcx.ins().uextend(types::I32, lhs);
|
|
|
|
let rhs = bcx.ins().uextend(types::I32, rhs);
|
|
|
|
let res = bcx.ins().select(cond, lhs, rhs);
|
|
|
|
bcx.ins().ireduce(lhs_ty, res)
|
|
|
|
} else {
|
|
|
|
bcx.ins().select(cond, lhs, rhs)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-07-27 12:27:20 -05:00
|
|
|
fn codegen_field<'a, 'tcx: 'a>(
|
2018-08-14 13:31:16 -05:00
|
|
|
fx: &mut FunctionCx<'a, 'tcx, impl Backend>,
|
2018-07-27 12:27:20 -05:00
|
|
|
base: Value,
|
|
|
|
layout: TyLayout<'tcx>,
|
2018-07-31 05:25:16 -05:00
|
|
|
field: mir::Field,
|
2018-07-27 12:27:20 -05:00
|
|
|
) -> (Value, TyLayout<'tcx>) {
|
|
|
|
let field_offset = layout.fields.offset(field.index());
|
|
|
|
let field_ty = layout.field(&*fx, field.index());
|
|
|
|
if field_offset.bytes() > 0 {
|
2018-08-22 05:31:45 -05:00
|
|
|
(
|
|
|
|
fx.bcx.ins().iadd_imm(base, field_offset.bytes() as i64),
|
|
|
|
field_ty,
|
|
|
|
)
|
2018-07-27 12:27:20 -05:00
|
|
|
} else {
|
|
|
|
(base, field_ty)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-07-18 04:55:32 -05:00
|
|
|
/// A read-only value
|
2018-06-23 11:26:54 -05:00
|
|
|
#[derive(Debug, Copy, Clone)]
|
2018-06-26 12:44:19 -05:00
|
|
|
pub enum CValue<'tcx> {
|
|
|
|
ByRef(Value, TyLayout<'tcx>),
|
|
|
|
ByVal(Value, TyLayout<'tcx>),
|
2018-08-22 05:35:07 -05:00
|
|
|
ByValPair(Value, Value, TyLayout<'tcx>),
|
2018-06-22 12:18:53 -05:00
|
|
|
}
|
|
|
|
|
2018-06-26 12:44:19 -05:00
|
|
|
impl<'tcx> CValue<'tcx> {
|
|
|
|
pub fn layout(&self) -> TyLayout<'tcx> {
|
|
|
|
match *self {
|
2018-08-22 05:35:07 -05:00
|
|
|
CValue::ByRef(_, layout)
|
|
|
|
| CValue::ByVal(_, layout)
|
|
|
|
| CValue::ByValPair(_, _, layout) => layout,
|
2018-06-26 12:44:19 -05:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-08-14 13:31:16 -05:00
|
|
|
pub fn force_stack<'a>(self, fx: &mut FunctionCx<'a, 'tcx, impl Backend>) -> Value
|
2018-07-31 05:25:16 -05:00
|
|
|
where
|
|
|
|
'tcx: 'a,
|
|
|
|
{
|
2018-06-22 12:18:53 -05:00
|
|
|
match self {
|
2018-06-26 12:44:19 -05:00
|
|
|
CValue::ByRef(value, _layout) => value,
|
|
|
|
CValue::ByVal(value, layout) => {
|
2018-06-22 12:18:53 -05:00
|
|
|
let stack_slot = fx.bcx.create_stack_slot(StackSlotData {
|
|
|
|
kind: StackSlotKind::ExplicitSlot,
|
|
|
|
size: layout.size.bytes() as u32,
|
|
|
|
offset: None,
|
|
|
|
});
|
2018-10-10 12:07:13 -05:00
|
|
|
let addr = fx
|
|
|
|
.bcx
|
2018-08-22 05:31:45 -05:00
|
|
|
.ins()
|
2018-11-03 07:14:28 -05:00
|
|
|
.stack_addr(fx.pointer_type, stack_slot, 0);
|
2018-09-30 09:33:55 -05:00
|
|
|
fx.bcx.ins().store(MemFlags::new(), value, addr, 0);
|
|
|
|
addr
|
2018-06-22 12:18:53 -05:00
|
|
|
}
|
2018-08-22 05:35:07 -05:00
|
|
|
CValue::ByValPair(value, extra, layout) => {
|
|
|
|
let stack_slot = fx.bcx.create_stack_slot(StackSlotData {
|
|
|
|
kind: StackSlotKind::ExplicitSlot,
|
|
|
|
size: layout.size.bytes() as u32,
|
|
|
|
offset: None,
|
|
|
|
});
|
|
|
|
let base = fx.bcx.ins().stack_addr(types::I64, stack_slot, 0);
|
|
|
|
let a_addr = codegen_field(fx, base, layout, mir::Field::new(0)).0;
|
|
|
|
let b_addr = codegen_field(fx, base, layout, mir::Field::new(1)).0;
|
|
|
|
fx.bcx.ins().store(MemFlags::new(), value, a_addr, 0);
|
|
|
|
fx.bcx.ins().store(MemFlags::new(), extra, b_addr, 0);
|
|
|
|
base
|
|
|
|
}
|
2018-06-22 12:18:53 -05:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-08-14 13:31:16 -05:00
|
|
|
pub fn load_value<'a>(self, fx: &mut FunctionCx<'a, 'tcx, impl Backend>) -> Value
|
2018-07-31 05:25:16 -05:00
|
|
|
where
|
|
|
|
'tcx: 'a,
|
|
|
|
{
|
2018-06-22 12:18:53 -05:00
|
|
|
match self {
|
2018-06-30 09:27:11 -05:00
|
|
|
CValue::ByRef(addr, layout) => {
|
2018-09-08 10:24:52 -05:00
|
|
|
let cton_ty = fx.cton_type(layout.ty).unwrap_or_else(|| {
|
|
|
|
if layout.ty.is_box() && !fx
|
|
|
|
.layout_of(layout.ty.builtin_deref(true).unwrap().ty)
|
|
|
|
.is_unsized()
|
|
|
|
{
|
|
|
|
// Consider sized box to be a ptr
|
|
|
|
pointer_ty(fx.tcx)
|
|
|
|
} else {
|
|
|
|
panic!("load_value of type {:?}", layout.ty);
|
|
|
|
}
|
|
|
|
});
|
2018-06-30 09:27:11 -05:00
|
|
|
fx.bcx.ins().load(cton_ty, MemFlags::new(), addr, 0)
|
2018-06-22 12:18:53 -05:00
|
|
|
}
|
2018-06-26 12:44:19 -05:00
|
|
|
CValue::ByVal(value, _layout) => value,
|
2018-08-22 05:35:07 -05:00
|
|
|
CValue::ByValPair(_, _, _layout) => bug!("Please use load_value_pair for ByValPair"),
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
pub fn load_value_pair<'a>(self, fx: &mut FunctionCx<'a, 'tcx, impl Backend>) -> (Value, Value)
|
|
|
|
where
|
|
|
|
'tcx: 'a,
|
|
|
|
{
|
|
|
|
match self {
|
|
|
|
CValue::ByRef(addr, layout) => {
|
|
|
|
assert_eq!(
|
|
|
|
layout.size.bytes(),
|
|
|
|
fx.tcx.data_layout.pointer_size.bytes() * 2
|
|
|
|
);
|
|
|
|
let val1_offset = layout.fields.offset(0).bytes() as i32;
|
|
|
|
let val2_offset = layout.fields.offset(1).bytes() as i32;
|
|
|
|
let val1 =
|
|
|
|
fx.bcx
|
|
|
|
.ins()
|
2018-11-03 07:14:28 -05:00
|
|
|
.load(fx.pointer_type, MemFlags::new(), addr, val1_offset);
|
2018-08-22 05:35:07 -05:00
|
|
|
let val2 =
|
|
|
|
fx.bcx
|
|
|
|
.ins()
|
2018-11-03 07:14:28 -05:00
|
|
|
.load(fx.pointer_type, MemFlags::new(), addr, val2_offset);
|
2018-08-22 05:35:07 -05:00
|
|
|
(val1, val2)
|
|
|
|
}
|
|
|
|
CValue::ByVal(_, _layout) => bug!("Please use load_value for ByVal"),
|
|
|
|
CValue::ByValPair(val1, val2, _layout) => (val1, val2),
|
2018-06-22 12:18:53 -05:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-06-26 12:44:19 -05:00
|
|
|
pub fn expect_byref(self) -> (Value, TyLayout<'tcx>) {
|
2018-06-22 12:18:53 -05:00
|
|
|
match self {
|
2018-06-26 12:44:19 -05:00
|
|
|
CValue::ByRef(value, layout) => (value, layout),
|
2018-08-08 02:34:56 -05:00
|
|
|
CValue::ByVal(_, _) => bug!("Expected CValue::ByRef, found CValue::ByVal: {:?}", self),
|
2018-08-22 05:35:07 -05:00
|
|
|
CValue::ByValPair(_, _, _) => bug!(
|
|
|
|
"Expected CValue::ByRef, found CValue::ByValPair: {:?}",
|
|
|
|
self
|
|
|
|
),
|
2018-06-22 12:18:53 -05:00
|
|
|
}
|
|
|
|
}
|
2018-06-23 11:26:54 -05:00
|
|
|
|
2018-08-14 13:31:16 -05:00
|
|
|
pub fn value_field<'a>(
|
|
|
|
self,
|
|
|
|
fx: &mut FunctionCx<'a, 'tcx, impl Backend>,
|
|
|
|
field: mir::Field,
|
|
|
|
) -> CValue<'tcx>
|
2018-07-31 05:25:16 -05:00
|
|
|
where
|
|
|
|
'tcx: 'a,
|
|
|
|
{
|
2018-06-26 12:44:19 -05:00
|
|
|
let (base, layout) = match self {
|
|
|
|
CValue::ByRef(addr, layout) => (addr, layout),
|
2018-06-23 11:26:54 -05:00
|
|
|
_ => bug!("place_field for {:?}", self),
|
|
|
|
};
|
2018-07-27 12:27:20 -05:00
|
|
|
|
|
|
|
let (field_ptr, field_layout) = codegen_field(fx, base, layout, field);
|
|
|
|
CValue::ByRef(field_ptr, field_layout)
|
2018-06-23 11:26:54 -05:00
|
|
|
}
|
|
|
|
|
2018-08-22 08:38:56 -05:00
|
|
|
pub fn unsize_value<'a>(self, fx: &mut FunctionCx<'a, 'tcx, impl Backend>, dest: CPlace<'tcx>) {
|
|
|
|
if self.layout().ty == dest.layout().ty {
|
|
|
|
dest.write_cvalue(fx, self); // FIXME this shouldn't happen (rust-lang/rust#53602)
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
match &self.layout().ty.sty {
|
|
|
|
ty::Ref(_, ty, _) | ty::RawPtr(TypeAndMut { ty, mutbl: _ }) => {
|
2018-09-04 12:04:25 -05:00
|
|
|
let (ptr, extra) = match dest.layout().ty.builtin_deref(true).unwrap().ty.sty {
|
2018-08-22 08:38:56 -05:00
|
|
|
ty::Slice(slice_elem_ty) => match ty.sty {
|
|
|
|
ty::Array(array_elem_ty, size) => {
|
|
|
|
assert_eq!(slice_elem_ty, array_elem_ty);
|
|
|
|
let ptr = self.load_value(fx);
|
|
|
|
let extra = fx
|
|
|
|
.bcx
|
|
|
|
.ins()
|
2018-11-03 07:14:28 -05:00
|
|
|
.iconst(fx.pointer_type, size.unwrap_usize(fx.tcx) as i64);
|
2018-08-22 08:38:56 -05:00
|
|
|
(ptr, extra)
|
|
|
|
}
|
|
|
|
_ => bug!("unsize non array {:?} to slice", ty),
|
|
|
|
},
|
2018-09-08 11:00:06 -05:00
|
|
|
ty::Dynamic(data, _) => match ty.sty {
|
2018-08-22 08:38:56 -05:00
|
|
|
ty::Dynamic(_, _) => self.load_value_pair(fx),
|
2018-09-08 11:00:06 -05:00
|
|
|
_ => {
|
|
|
|
let ptr = self.load_value(fx);
|
|
|
|
let vtable = crate::vtable::get_vtable(fx, ty, data.principal());
|
|
|
|
(ptr, vtable)
|
|
|
|
}
|
2018-08-22 08:38:56 -05:00
|
|
|
},
|
|
|
|
_ => bug!(
|
|
|
|
"unsize of type {:?} to {:?}",
|
|
|
|
self.layout().ty,
|
|
|
|
dest.layout().ty
|
|
|
|
),
|
|
|
|
};
|
|
|
|
dest.write_cvalue(fx, CValue::ByValPair(ptr, extra, dest.layout()));
|
|
|
|
}
|
2018-09-16 04:57:27 -05:00
|
|
|
_ => {
|
|
|
|
assert!(!self.layout().ty.is_enum(), "Tried to unsize enum");
|
|
|
|
let field_count = self.layout().fields.count();
|
|
|
|
let mut found_unsize_field = false;
|
|
|
|
for idx in 0..field_count {
|
|
|
|
let field_dest = dest.place_field(fx, mir::Field::new(idx));
|
|
|
|
let field_src = self.value_field(fx, mir::Field::new(idx));
|
2018-09-16 07:49:45 -05:00
|
|
|
if field_src.layout().ty.is_phantom_data() {
|
|
|
|
// Ignore PhantomData so for example `Unique<()>` can coerce to `Unique<Debug>`
|
|
|
|
//
|
|
|
|
// ```rust
|
|
|
|
// struct Unique<T: ?Sized> {
|
|
|
|
// pointer: NonZero<*const T>,
|
|
|
|
// _marker: PhantomData<T>,
|
|
|
|
// }
|
|
|
|
// ```
|
|
|
|
continue;
|
|
|
|
}
|
2018-09-16 04:57:27 -05:00
|
|
|
if field_src.layout().ty != field_dest.layout().ty {
|
|
|
|
assert!(!found_unsize_field);
|
|
|
|
found_unsize_field = true;
|
|
|
|
field_src.unsize_value(fx, field_dest);
|
|
|
|
} else {
|
|
|
|
field_dest.write_cvalue(fx, field_src);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
2018-08-22 08:38:56 -05:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-07-31 05:25:16 -05:00
|
|
|
pub fn const_val<'a>(
|
2018-08-14 13:31:16 -05:00
|
|
|
fx: &mut FunctionCx<'a, 'tcx, impl Backend>,
|
2018-07-31 05:25:16 -05:00
|
|
|
ty: Ty<'tcx>,
|
|
|
|
const_val: i64,
|
|
|
|
) -> CValue<'tcx>
|
|
|
|
where
|
|
|
|
'tcx: 'a,
|
|
|
|
{
|
2018-06-26 12:44:19 -05:00
|
|
|
let cton_ty = fx.cton_type(ty).unwrap();
|
|
|
|
let layout = fx.layout_of(ty);
|
|
|
|
CValue::ByVal(fx.bcx.ins().iconst(cton_ty, const_val), layout)
|
|
|
|
}
|
|
|
|
|
|
|
|
pub fn unchecked_cast_to(self, layout: TyLayout<'tcx>) -> Self {
|
|
|
|
match self {
|
|
|
|
CValue::ByRef(addr, _) => CValue::ByRef(addr, layout),
|
|
|
|
CValue::ByVal(val, _) => CValue::ByVal(val, layout),
|
2018-08-22 05:35:07 -05:00
|
|
|
CValue::ByValPair(val, extra, _) => CValue::ByValPair(val, extra, layout),
|
2018-06-26 12:44:19 -05:00
|
|
|
}
|
2018-06-23 11:26:54 -05:00
|
|
|
}
|
2018-06-22 12:18:53 -05:00
|
|
|
}
|
|
|
|
|
2018-07-18 04:55:32 -05:00
|
|
|
/// A place where you can write a value to or read a value from
|
2018-06-23 11:26:54 -05:00
|
|
|
#[derive(Debug, Copy, Clone)]
|
2018-06-26 12:44:19 -05:00
|
|
|
pub enum CPlace<'tcx> {
|
2018-08-15 05:36:13 -05:00
|
|
|
Var(Local, TyLayout<'tcx>),
|
2018-08-22 08:38:10 -05:00
|
|
|
Addr(Value, Option<Value>, TyLayout<'tcx>),
|
2018-06-22 12:18:53 -05:00
|
|
|
}
|
|
|
|
|
2018-06-26 12:44:19 -05:00
|
|
|
impl<'a, 'tcx: 'a> CPlace<'tcx> {
|
|
|
|
pub fn layout(&self) -> TyLayout<'tcx> {
|
|
|
|
match *self {
|
2018-08-22 08:38:10 -05:00
|
|
|
CPlace::Var(_, layout) | CPlace::Addr(_, _, layout) => layout,
|
2018-06-26 12:44:19 -05:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-08-14 13:31:16 -05:00
|
|
|
pub fn temp(fx: &mut FunctionCx<'a, 'tcx, impl Backend>, ty: Ty<'tcx>) -> CPlace<'tcx> {
|
2018-08-08 02:34:56 -05:00
|
|
|
let layout = fx.layout_of(ty);
|
2018-08-22 08:38:10 -05:00
|
|
|
assert!(!layout.is_unsized());
|
2018-08-08 02:34:56 -05:00
|
|
|
let stack_slot = fx.bcx.create_stack_slot(StackSlotData {
|
|
|
|
kind: StackSlotKind::ExplicitSlot,
|
|
|
|
size: layout.size.bytes() as u32,
|
|
|
|
offset: None,
|
|
|
|
});
|
2018-08-22 05:31:45 -05:00
|
|
|
CPlace::Addr(
|
|
|
|
fx.bcx
|
|
|
|
.ins()
|
2018-11-03 07:14:28 -05:00
|
|
|
.stack_addr(fx.pointer_type, stack_slot, 0),
|
2018-08-22 08:38:10 -05:00
|
|
|
None,
|
2018-08-22 05:31:45 -05:00
|
|
|
layout,
|
|
|
|
)
|
2018-08-08 02:34:56 -05:00
|
|
|
}
|
|
|
|
|
2018-07-31 05:25:16 -05:00
|
|
|
pub fn from_stack_slot(
|
2018-08-14 13:31:16 -05:00
|
|
|
fx: &mut FunctionCx<'a, 'tcx, impl Backend>,
|
2018-07-31 05:25:16 -05:00
|
|
|
stack_slot: StackSlot,
|
|
|
|
ty: Ty<'tcx>,
|
|
|
|
) -> CPlace<'tcx> {
|
2018-06-26 12:44:19 -05:00
|
|
|
let layout = fx.layout_of(ty);
|
2018-08-22 08:38:10 -05:00
|
|
|
assert!(!layout.is_unsized());
|
2018-08-22 05:31:45 -05:00
|
|
|
CPlace::Addr(
|
|
|
|
fx.bcx
|
|
|
|
.ins()
|
2018-11-03 07:14:28 -05:00
|
|
|
.stack_addr(fx.pointer_type, stack_slot, 0),
|
2018-08-22 08:38:10 -05:00
|
|
|
None,
|
2018-08-22 05:31:45 -05:00
|
|
|
layout,
|
|
|
|
)
|
2018-06-22 12:18:53 -05:00
|
|
|
}
|
|
|
|
|
2018-08-14 13:31:16 -05:00
|
|
|
pub fn to_cvalue(self, fx: &mut FunctionCx<'a, 'tcx, impl Backend>) -> CValue<'tcx> {
|
2018-06-22 12:18:53 -05:00
|
|
|
match self {
|
2018-08-15 05:36:13 -05:00
|
|
|
CPlace::Var(var, layout) => CValue::ByVal(fx.bcx.use_var(mir_var(var)), layout),
|
2018-08-22 08:38:10 -05:00
|
|
|
CPlace::Addr(addr, extra, layout) => {
|
|
|
|
assert!(extra.is_none(), "unsized values are not yet supported");
|
|
|
|
CValue::ByRef(addr, layout)
|
|
|
|
}
|
2018-06-22 12:18:53 -05:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
pub fn expect_addr(self) -> Value {
|
|
|
|
match self {
|
2018-08-22 08:38:10 -05:00
|
|
|
CPlace::Addr(addr, None, _layout) => addr,
|
|
|
|
CPlace::Addr(_, _, _) => bug!("Expected sized CPlace::Addr, found {:?}", self),
|
2018-06-26 12:44:19 -05:00
|
|
|
CPlace::Var(_, _) => bug!("Expected CPlace::Addr, found CPlace::Var"),
|
2018-06-22 12:18:53 -05:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-08-14 13:31:16 -05:00
|
|
|
pub fn write_cvalue(self, fx: &mut FunctionCx<'a, 'tcx, impl Backend>, from: CValue<'tcx>) {
|
2018-06-29 11:57:59 -05:00
|
|
|
match (&self.layout().ty.sty, &from.layout().ty.sty) {
|
2018-08-24 07:51:02 -05:00
|
|
|
(ty::Ref(_, t, dest_mut), ty::Ref(_, u, src_mut))
|
2018-09-26 08:40:11 -05:00
|
|
|
if (if *dest_mut != crate::rustc::hir::Mutability::MutImmutable
|
|
|
|
&& src_mut != dest_mut
|
|
|
|
{
|
2018-06-29 11:57:59 -05:00
|
|
|
false
|
|
|
|
} else if t != u {
|
|
|
|
false
|
|
|
|
} else {
|
|
|
|
true
|
2018-07-31 05:25:16 -05:00
|
|
|
}) =>
|
|
|
|
{
|
2018-06-29 11:57:59 -05:00
|
|
|
// &mut T -> &T is allowed
|
|
|
|
// &'a T -> &'b T is allowed
|
|
|
|
}
|
|
|
|
_ => {
|
|
|
|
assert_eq!(
|
2018-07-31 05:25:16 -05:00
|
|
|
self.layout().ty,
|
|
|
|
from.layout().ty,
|
2018-06-29 11:57:59 -05:00
|
|
|
"Can't write value of incompatible type to place {:?} {:?}\n\n{:#?}",
|
2018-07-31 05:25:16 -05:00
|
|
|
self.layout().ty.sty,
|
|
|
|
from.layout().ty.sty,
|
2018-06-29 11:57:59 -05:00
|
|
|
fx,
|
|
|
|
);
|
|
|
|
}
|
|
|
|
}
|
2018-06-26 12:44:19 -05:00
|
|
|
|
2018-06-22 12:18:53 -05:00
|
|
|
match self {
|
2018-06-26 12:44:19 -05:00
|
|
|
CPlace::Var(var, _) => {
|
|
|
|
let data = from.load_value(fx);
|
2018-08-15 05:36:13 -05:00
|
|
|
fx.bcx.def_var(mir_var(var), data)
|
2018-07-31 05:25:16 -05:00
|
|
|
}
|
2018-10-25 10:36:29 -05:00
|
|
|
CPlace::Addr(addr, None, dst_layout) => {
|
2018-08-22 05:35:07 -05:00
|
|
|
match from {
|
2018-10-25 10:36:29 -05:00
|
|
|
CValue::ByVal(val, _src_layout) => {
|
2018-08-22 05:35:07 -05:00
|
|
|
fx.bcx.ins().store(MemFlags::new(), val, addr, 0);
|
2018-07-14 09:39:49 -05:00
|
|
|
}
|
2018-10-25 10:36:29 -05:00
|
|
|
CValue::ByValPair(val1, val2, _src_layout) => {
|
|
|
|
let val1_offset = dst_layout.fields.offset(0).bytes() as i32;
|
|
|
|
let val2_offset = dst_layout.fields.offset(1).bytes() as i32;
|
2018-08-22 05:35:07 -05:00
|
|
|
fx.bcx.ins().store(MemFlags::new(), val1, addr, val1_offset);
|
|
|
|
fx.bcx.ins().store(MemFlags::new(), val2, addr, val2_offset);
|
2018-07-14 09:39:49 -05:00
|
|
|
}
|
2018-10-25 10:36:29 -05:00
|
|
|
CValue::ByRef(from, _src_layout) => {
|
|
|
|
let size = dst_layout.size.bytes() as i32;
|
|
|
|
// FIXME emit_small_memcpy has a bug as of commit CraneStation/cranelift@b2281ed
|
2018-11-03 07:14:28 -05:00
|
|
|
// fx.bcx.emit_small_memcpy(fx.module.target_config(), addr, from, size, layout.align.abi() as u8, src_layout.align.abi() as u8);
|
2018-10-25 10:36:29 -05:00
|
|
|
|
2018-08-22 05:35:07 -05:00
|
|
|
let mut offset = 0;
|
|
|
|
while size - offset >= 8 {
|
|
|
|
let byte = fx.bcx.ins().load(
|
2018-11-03 07:14:28 -05:00
|
|
|
fx.pointer_type,
|
2018-08-22 05:35:07 -05:00
|
|
|
MemFlags::new(),
|
|
|
|
from,
|
|
|
|
offset,
|
|
|
|
);
|
|
|
|
fx.bcx.ins().store(MemFlags::new(), byte, addr, offset);
|
|
|
|
offset += 8;
|
|
|
|
}
|
|
|
|
while size - offset >= 4 {
|
|
|
|
let byte = fx.bcx.ins().load(types::I32, MemFlags::new(), from, offset);
|
|
|
|
fx.bcx.ins().store(MemFlags::new(), byte, addr, offset);
|
|
|
|
offset += 4;
|
|
|
|
}
|
|
|
|
while offset < size {
|
|
|
|
let byte = fx.bcx.ins().load(types::I8, MemFlags::new(), from, offset);
|
|
|
|
fx.bcx.ins().store(MemFlags::new(), byte, addr, offset);
|
|
|
|
offset += 1;
|
|
|
|
}
|
2018-06-22 12:18:53 -05:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
2018-08-22 08:38:10 -05:00
|
|
|
CPlace::Addr(_, _, _) => bug!("Can't write value to unsized place {:?}", self),
|
2018-06-22 12:18:53 -05:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-08-14 13:31:16 -05:00
|
|
|
pub fn place_field(
|
|
|
|
self,
|
|
|
|
fx: &mut FunctionCx<'a, 'tcx, impl Backend>,
|
|
|
|
field: mir::Field,
|
|
|
|
) -> CPlace<'tcx> {
|
2018-09-16 08:28:27 -05:00
|
|
|
match self {
|
|
|
|
CPlace::Var(var, layout) => {
|
2018-09-26 08:40:11 -05:00
|
|
|
bug!(
|
|
|
|
"Tried to project {:?}, which is put in SSA var {:?}",
|
|
|
|
layout.ty,
|
|
|
|
var
|
|
|
|
);
|
2018-09-16 08:28:27 -05:00
|
|
|
}
|
|
|
|
CPlace::Addr(base, extra, layout) => {
|
|
|
|
let (field_ptr, field_layout) = codegen_field(fx, base, layout, field);
|
2018-10-06 04:21:18 -05:00
|
|
|
let extra = if field_layout.is_unsized() {
|
|
|
|
assert!(extra.is_some());
|
|
|
|
extra
|
|
|
|
} else {
|
|
|
|
None
|
|
|
|
};
|
2018-09-16 08:28:27 -05:00
|
|
|
CPlace::Addr(field_ptr, extra, field_layout)
|
|
|
|
}
|
2018-08-22 08:38:10 -05:00
|
|
|
}
|
2018-06-23 11:26:54 -05:00
|
|
|
}
|
2018-06-26 12:44:19 -05:00
|
|
|
|
2018-08-14 13:31:16 -05:00
|
|
|
pub fn place_index(
|
|
|
|
self,
|
|
|
|
fx: &mut FunctionCx<'a, 'tcx, impl Backend>,
|
|
|
|
index: Value,
|
|
|
|
) -> CPlace<'tcx> {
|
2018-08-25 04:22:48 -05:00
|
|
|
let (elem_layout, addr) = match self.layout().ty.sty {
|
|
|
|
ty::Array(elem_ty, _) => (fx.layout_of(elem_ty), self.expect_addr()),
|
|
|
|
ty::Slice(elem_ty) => (
|
|
|
|
fx.layout_of(elem_ty),
|
|
|
|
match self {
|
|
|
|
CPlace::Addr(addr, _, _) => addr,
|
|
|
|
CPlace::Var(_, _) => bug!("Expected CPlace::Addr found CPlace::Var"),
|
|
|
|
},
|
|
|
|
),
|
|
|
|
_ => bug!("place_index({:?})", self.layout().ty),
|
|
|
|
};
|
2018-08-22 10:58:25 -05:00
|
|
|
|
2018-08-25 04:22:48 -05:00
|
|
|
let offset = fx
|
|
|
|
.bcx
|
|
|
|
.ins()
|
|
|
|
.imul_imm(index, elem_layout.size.bytes() as i64);
|
2018-08-22 10:58:25 -05:00
|
|
|
|
2018-08-25 04:22:48 -05:00
|
|
|
CPlace::Addr(fx.bcx.ins().iadd(addr, offset), None, elem_layout)
|
2018-08-08 05:22:16 -05:00
|
|
|
}
|
|
|
|
|
2018-08-22 08:38:10 -05:00
|
|
|
pub fn place_deref(self, fx: &mut FunctionCx<'a, 'tcx, impl Backend>) -> CPlace<'tcx> {
|
2018-09-04 12:04:25 -05:00
|
|
|
let inner_layout = fx.layout_of(self.layout().ty.builtin_deref(true).unwrap().ty);
|
2018-08-22 08:38:10 -05:00
|
|
|
if !inner_layout.is_unsized() {
|
|
|
|
CPlace::Addr(self.to_cvalue(fx).load_value(fx), None, inner_layout)
|
|
|
|
} else {
|
|
|
|
match self.layout().abi {
|
|
|
|
Abi::ScalarPair(ref a, ref b) => {
|
|
|
|
let addr = self.expect_addr();
|
|
|
|
let ptr =
|
|
|
|
fx.bcx
|
|
|
|
.ins()
|
|
|
|
.load(scalar_to_cton_type(fx.tcx, a), MemFlags::new(), addr, 0);
|
|
|
|
let extra = fx.bcx.ins().load(
|
|
|
|
scalar_to_cton_type(fx.tcx, b),
|
|
|
|
MemFlags::new(),
|
|
|
|
addr,
|
2018-11-05 11:03:47 -06:00
|
|
|
a.value.size(&fx.tcx).bytes() as u32 as i32,
|
2018-08-22 08:38:10 -05:00
|
|
|
);
|
|
|
|
CPlace::Addr(ptr, Some(extra), inner_layout)
|
|
|
|
}
|
|
|
|
_ => bug!(
|
|
|
|
"Fat ptr doesn't have abi ScalarPair, but it has {:?}",
|
|
|
|
self.layout().abi
|
|
|
|
),
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
pub fn write_place_ref(self, fx: &mut FunctionCx<'a, 'tcx, impl Backend>, dest: CPlace<'tcx>) {
|
|
|
|
if !self.layout().is_unsized() {
|
|
|
|
let ptr = CValue::ByVal(self.expect_addr(), dest.layout());
|
|
|
|
dest.write_cvalue(fx, ptr);
|
|
|
|
} else {
|
|
|
|
match self {
|
|
|
|
CPlace::Var(_, _) => bug!("expected CPlace::Addr found CPlace::Var"),
|
|
|
|
CPlace::Addr(value, extra, _) => match dest.layout().abi {
|
|
|
|
Abi::ScalarPair(ref a, _) => {
|
|
|
|
fx.bcx
|
|
|
|
.ins()
|
|
|
|
.store(MemFlags::new(), value, dest.expect_addr(), 0);
|
|
|
|
fx.bcx.ins().store(
|
|
|
|
MemFlags::new(),
|
|
|
|
extra.expect("unsized type without metadata"),
|
|
|
|
dest.expect_addr(),
|
2018-11-05 11:03:47 -06:00
|
|
|
a.value.size(&fx.tcx).bytes() as u32 as i32,
|
2018-08-22 08:38:10 -05:00
|
|
|
);
|
|
|
|
}
|
|
|
|
_ => bug!(
|
|
|
|
"Non ScalarPair abi {:?} in write_place_ref dest",
|
|
|
|
dest.layout().abi
|
|
|
|
),
|
|
|
|
},
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-06-26 12:44:19 -05:00
|
|
|
pub fn unchecked_cast_to(self, layout: TyLayout<'tcx>) -> Self {
|
|
|
|
match self {
|
|
|
|
CPlace::Var(var, _) => CPlace::Var(var, layout),
|
2018-08-22 08:38:10 -05:00
|
|
|
CPlace::Addr(addr, extra, _) => {
|
|
|
|
assert!(!layout.is_unsized());
|
|
|
|
CPlace::Addr(addr, extra, layout)
|
|
|
|
}
|
2018-06-26 12:44:19 -05:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-08-14 13:31:16 -05:00
|
|
|
pub fn downcast_variant(self, fx: &FunctionCx<'a, 'tcx, impl Backend>, variant: usize) -> Self {
|
2018-06-26 12:44:19 -05:00
|
|
|
let layout = self.layout().for_variant(fx, variant);
|
|
|
|
self.unchecked_cast_to(layout)
|
|
|
|
}
|
2018-06-22 12:18:53 -05:00
|
|
|
}
|
|
|
|
|
2018-07-31 05:25:16 -05:00
|
|
|
pub fn cton_intcast<'a, 'tcx: 'a>(
|
2018-08-14 13:31:16 -05:00
|
|
|
fx: &mut FunctionCx<'a, 'tcx, impl Backend>,
|
2018-07-31 05:25:16 -05:00
|
|
|
val: Value,
|
2018-08-14 04:58:39 -05:00
|
|
|
to: Type,
|
2018-07-31 05:25:16 -05:00
|
|
|
signed: bool,
|
|
|
|
) -> Value {
|
2018-08-14 04:58:39 -05:00
|
|
|
let from = fx.bcx.func.dfg.value_type(val);
|
2018-06-23 11:26:54 -05:00
|
|
|
if from == to {
|
|
|
|
return val;
|
|
|
|
}
|
2018-07-18 09:22:29 -05:00
|
|
|
if to.wider_or_equal(from) {
|
2018-06-23 11:26:54 -05:00
|
|
|
if signed {
|
|
|
|
fx.bcx.ins().sextend(to, val)
|
|
|
|
} else {
|
|
|
|
fx.bcx.ins().uextend(to, val)
|
|
|
|
}
|
|
|
|
} else {
|
|
|
|
fx.bcx.ins().ireduce(to, val)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-08-14 13:31:16 -05:00
|
|
|
pub struct FunctionCx<'a, 'tcx: 'a, B: Backend + 'a> {
|
2018-06-22 12:18:53 -05:00
|
|
|
pub tcx: TyCtxt<'a, 'tcx, 'tcx>,
|
2018-08-14 13:31:16 -05:00
|
|
|
pub module: &'a mut Module<B>,
|
2018-11-03 07:14:28 -05:00
|
|
|
pub pointer_type: Type, // Cached from module
|
2018-06-23 11:54:15 -05:00
|
|
|
pub instance: Instance<'tcx>,
|
2018-06-22 12:18:53 -05:00
|
|
|
pub mir: &'tcx Mir<'tcx>,
|
2018-06-23 11:54:15 -05:00
|
|
|
pub param_substs: &'tcx Substs<'tcx>,
|
2018-08-15 05:36:13 -05:00
|
|
|
pub bcx: FunctionBuilder<'a>,
|
2018-06-22 12:18:53 -05:00
|
|
|
pub ebb_map: HashMap<BasicBlock, Ebb>,
|
2018-06-26 12:44:19 -05:00
|
|
|
pub local_map: HashMap<Local, CPlace<'tcx>>,
|
2018-07-14 04:59:42 -05:00
|
|
|
pub comments: HashMap<Inst, String>,
|
2018-08-13 05:13:43 -05:00
|
|
|
pub constants: &'a mut crate::constant::ConstantCx,
|
2018-09-08 11:00:06 -05:00
|
|
|
pub caches: &'a mut Caches<'tcx>,
|
2018-08-15 09:17:59 -05:00
|
|
|
|
|
|
|
/// add_global_comment inserts a comment here
|
|
|
|
pub top_nop: Option<Inst>,
|
2018-06-22 12:18:53 -05:00
|
|
|
}
|
|
|
|
|
2018-08-14 13:31:16 -05:00
|
|
|
impl<'a, 'tcx: 'a, B: Backend + 'a> fmt::Debug for FunctionCx<'a, 'tcx, B> {
|
2018-07-14 05:21:45 -05:00
|
|
|
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
|
|
|
|
writeln!(f, "{:?}", self.param_substs)?;
|
|
|
|
writeln!(f, "{:?}", self.local_map)?;
|
|
|
|
|
|
|
|
let mut clif = String::new();
|
2018-07-30 09:57:40 -05:00
|
|
|
let mut writer = crate::pretty_clif::CommentWriter(self.comments.clone());
|
2018-07-14 05:21:45 -05:00
|
|
|
::cranelift::codegen::write::decorate_function(
|
|
|
|
&mut writer,
|
|
|
|
&mut clif,
|
|
|
|
&self.bcx.func,
|
|
|
|
None,
|
2018-10-10 12:07:13 -05:00
|
|
|
)
|
|
|
|
.unwrap();
|
2018-07-14 05:21:45 -05:00
|
|
|
writeln!(f, "\n{}", clif)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-11-05 11:03:47 -06:00
|
|
|
impl<'a, 'tcx: 'a, B: Backend> LayoutOf for FunctionCx<'a, 'tcx, B> {
|
2018-06-24 07:01:41 -05:00
|
|
|
type Ty = Ty<'tcx>;
|
|
|
|
type TyLayout = TyLayout<'tcx>;
|
|
|
|
|
2018-11-05 11:03:47 -06:00
|
|
|
fn layout_of(&self, ty: Ty<'tcx>) -> TyLayout<'tcx> {
|
2018-06-24 07:01:41 -05:00
|
|
|
let ty = self.monomorphize(&ty);
|
|
|
|
self.tcx.layout_of(ParamEnv::reveal_all().and(&ty)).unwrap()
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-11-05 11:03:47 -06:00
|
|
|
impl<'a, 'tcx, B: Backend + 'a> layout::HasTyCtxt<'tcx> for FunctionCx<'a, 'tcx, B> {
|
2018-06-24 07:25:29 -05:00
|
|
|
fn tcx<'b>(&'b self) -> TyCtxt<'b, 'tcx, 'tcx> {
|
|
|
|
self.tcx
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-11-05 11:03:47 -06:00
|
|
|
impl<'a, 'tcx, B: Backend + 'a> layout::HasDataLayout for FunctionCx<'a, 'tcx, B> {
|
2018-06-24 07:25:29 -05:00
|
|
|
fn data_layout(&self) -> &layout::TargetDataLayout {
|
|
|
|
&self.tcx.data_layout
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-11-05 11:03:47 -06:00
|
|
|
impl<'a, 'tcx, B: Backend + 'a> HasTargetSpec for FunctionCx<'a, 'tcx, B> {
|
2018-06-24 07:25:29 -05:00
|
|
|
fn target_spec(&self) -> &Target {
|
|
|
|
&self.tcx.sess.target.target
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-08-14 13:31:16 -05:00
|
|
|
impl<'a, 'tcx: 'a, B: Backend + 'a> FunctionCx<'a, 'tcx, B> {
|
2018-06-23 11:54:15 -05:00
|
|
|
pub fn monomorphize<T>(&self, value: &T) -> T
|
2018-07-31 05:25:16 -05:00
|
|
|
where
|
|
|
|
T: TypeFoldable<'tcx>,
|
2018-06-23 11:54:15 -05:00
|
|
|
{
|
|
|
|
self.tcx.subst_and_normalize_erasing_regions(
|
|
|
|
self.param_substs,
|
|
|
|
ty::ParamEnv::reveal_all(),
|
|
|
|
value,
|
|
|
|
)
|
|
|
|
}
|
|
|
|
|
2018-06-24 07:01:41 -05:00
|
|
|
pub fn cton_type(&self, ty: Ty<'tcx>) -> Option<Type> {
|
2018-07-14 09:39:49 -05:00
|
|
|
cton_type_from_ty(self.tcx, self.monomorphize(&ty))
|
2018-06-24 07:01:41 -05:00
|
|
|
}
|
|
|
|
|
2018-06-22 12:18:53 -05:00
|
|
|
pub fn get_ebb(&self, bb: BasicBlock) -> Ebb {
|
|
|
|
*self.ebb_map.get(&bb).unwrap()
|
|
|
|
}
|
|
|
|
|
2018-06-26 12:44:19 -05:00
|
|
|
pub fn get_local_place(&mut self, local: Local) -> CPlace<'tcx> {
|
2018-06-22 12:18:53 -05:00
|
|
|
*self.local_map.get(&local).unwrap()
|
|
|
|
}
|
|
|
|
}
|