2018-07-20 06:36:53 -05:00
|
|
|
use std::iter;
|
2018-07-19 12:33:42 -05:00
|
|
|
|
2018-07-20 06:36:53 -05:00
|
|
|
use rustc::hir;
|
|
|
|
use rustc_target::spec::abi::Abi;
|
2018-07-19 12:33:42 -05:00
|
|
|
|
2018-07-30 09:57:40 -05:00
|
|
|
use crate::prelude::*;
|
2018-07-19 12:33:42 -05:00
|
|
|
|
2018-08-11 05:34:46 -05:00
|
|
|
#[derive(Debug)]
|
|
|
|
enum PassMode {
|
|
|
|
NoPass,
|
|
|
|
ByVal(Type),
|
|
|
|
ByRef,
|
|
|
|
}
|
|
|
|
|
|
|
|
impl PassMode {
|
2018-08-18 10:10:02 -05:00
|
|
|
fn get_param_ty(self, fx: &FunctionCx<impl Backend>) -> Type {
|
2018-08-11 05:34:46 -05:00
|
|
|
match self {
|
|
|
|
PassMode::NoPass => unimplemented!("pass mode nopass"),
|
|
|
|
PassMode::ByVal(cton_type) => cton_type,
|
2018-08-18 10:10:02 -05:00
|
|
|
PassMode::ByRef => fx.module.pointer_type(),
|
2018-08-11 05:34:46 -05:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
fn get_pass_mode<'a, 'tcx: 'a>(
|
|
|
|
tcx: TyCtxt<'a, 'tcx, 'tcx>,
|
|
|
|
abi: Abi,
|
|
|
|
ty: Ty<'tcx>,
|
|
|
|
is_return: bool,
|
|
|
|
) -> PassMode {
|
|
|
|
if ty.sty == tcx.mk_nil().sty {
|
|
|
|
if is_return {
|
2018-08-11 06:59:08 -05:00
|
|
|
//if false {
|
2018-08-11 05:34:46 -05:00
|
|
|
PassMode::NoPass
|
|
|
|
} else {
|
|
|
|
PassMode::ByRef
|
|
|
|
}
|
|
|
|
} else if let Some(ret_ty) = crate::common::cton_type_from_ty(tcx, ty) {
|
|
|
|
PassMode::ByVal(ret_ty)
|
|
|
|
} else {
|
|
|
|
if abi == Abi::C {
|
|
|
|
unimplemented!("Non scalars are not yet supported for \"C\" abi");
|
|
|
|
}
|
|
|
|
PassMode::ByRef
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-07-31 05:25:16 -05:00
|
|
|
pub fn cton_sig_from_fn_ty<'a, 'tcx: 'a>(
|
|
|
|
tcx: TyCtxt<'a, 'tcx, 'tcx>,
|
|
|
|
fn_ty: Ty<'tcx>,
|
|
|
|
) -> Signature {
|
2018-07-20 06:36:53 -05:00
|
|
|
let sig = ty_fn_sig(tcx, fn_ty);
|
2018-07-19 12:33:42 -05:00
|
|
|
assert!(!sig.variadic, "Variadic function are not yet supported");
|
2018-08-11 05:34:46 -05:00
|
|
|
let (call_conv, inputs, output): (CallConv, Vec<Ty>, Ty) = match sig.abi {
|
2018-07-30 11:20:37 -05:00
|
|
|
Abi::Rust => (CallConv::Fast, sig.inputs().to_vec(), sig.output()),
|
|
|
|
Abi::C => (CallConv::SystemV, sig.inputs().to_vec(), sig.output()),
|
2018-07-20 06:36:53 -05:00
|
|
|
Abi::RustCall => {
|
2018-07-27 12:01:38 -05:00
|
|
|
assert_eq!(sig.inputs().len(), 2);
|
2018-07-20 06:38:49 -05:00
|
|
|
let extra_args = match sig.inputs().last().unwrap().sty {
|
|
|
|
ty::TyTuple(ref tupled_arguments) => tupled_arguments,
|
|
|
|
_ => bug!("argument to function with \"rust-call\" ABI is not a tuple"),
|
|
|
|
};
|
2018-07-27 12:01:38 -05:00
|
|
|
let mut inputs: Vec<Ty> = vec![sig.inputs()[0]];
|
2018-07-20 06:38:49 -05:00
|
|
|
inputs.extend(extra_args.into_iter());
|
2018-07-31 05:25:16 -05:00
|
|
|
(CallConv::Fast, inputs, sig.output())
|
2018-07-20 06:36:53 -05:00
|
|
|
}
|
|
|
|
Abi::System => bug!("system abi should be selected elsewhere"),
|
2018-07-20 07:20:37 -05:00
|
|
|
Abi::RustIntrinsic => (CallConv::SystemV, sig.inputs().to_vec(), sig.output()),
|
2018-07-20 06:36:53 -05:00
|
|
|
_ => unimplemented!("unsupported abi {:?}", sig.abi),
|
2018-07-19 12:33:42 -05:00
|
|
|
};
|
2018-08-11 05:34:46 -05:00
|
|
|
|
|
|
|
let inputs = inputs
|
|
|
|
.into_iter()
|
|
|
|
.filter_map(|ty| match get_pass_mode(tcx, sig.abi, ty, false) {
|
|
|
|
PassMode::ByVal(cton_ty) => Some(cton_ty),
|
|
|
|
PassMode::NoPass => unimplemented!("pass mode nopass"),
|
2018-08-18 10:10:02 -05:00
|
|
|
PassMode::ByRef => Some(pointer_ty(tcx)),
|
2018-08-11 05:34:46 -05:00
|
|
|
});
|
|
|
|
|
|
|
|
let (params, returns) = match get_pass_mode(tcx, sig.abi, output, true) {
|
|
|
|
PassMode::NoPass => (inputs.map(AbiParam::new).collect(), vec![]),
|
|
|
|
PassMode::ByVal(ret_ty) => (
|
|
|
|
inputs.map(AbiParam::new).collect(),
|
|
|
|
vec![AbiParam::new(ret_ty)],
|
|
|
|
),
|
|
|
|
PassMode::ByRef => {
|
|
|
|
(
|
2018-08-18 10:10:02 -05:00
|
|
|
Some(pointer_ty(tcx)).into_iter() // First param is place to put return val
|
2018-08-11 05:34:46 -05:00
|
|
|
.chain(inputs)
|
|
|
|
.map(AbiParam::new)
|
|
|
|
.collect(),
|
|
|
|
vec![],
|
|
|
|
)
|
|
|
|
}
|
|
|
|
};
|
|
|
|
|
2018-07-19 12:33:42 -05:00
|
|
|
Signature {
|
2018-08-11 05:34:46 -05:00
|
|
|
params,
|
|
|
|
returns,
|
2018-07-19 12:33:42 -05:00
|
|
|
call_conv,
|
|
|
|
argument_bytes: None,
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-07-31 05:25:16 -05:00
|
|
|
fn ty_fn_sig<'a, 'tcx>(tcx: TyCtxt<'a, 'tcx, 'tcx>, ty: Ty<'tcx>) -> ty::FnSig<'tcx> {
|
2018-07-20 06:38:49 -05:00
|
|
|
let sig = match ty.sty {
|
2018-07-20 06:36:53 -05:00
|
|
|
ty::TyFnDef(..) |
|
|
|
|
// Shims currently have type TyFnPtr. Not sure this should remain.
|
|
|
|
ty::TyFnPtr(_) => ty.fn_sig(tcx),
|
|
|
|
ty::TyClosure(def_id, substs) => {
|
|
|
|
let sig = substs.closure_sig(def_id, tcx);
|
|
|
|
|
|
|
|
let env_ty = tcx.closure_env_ty(def_id, substs).unwrap();
|
|
|
|
sig.map_bound(|sig| tcx.mk_fn_sig(
|
|
|
|
iter::once(*env_ty.skip_binder()).chain(sig.inputs().iter().cloned()),
|
|
|
|
sig.output(),
|
|
|
|
sig.variadic,
|
|
|
|
sig.unsafety,
|
|
|
|
sig.abi
|
|
|
|
))
|
|
|
|
}
|
|
|
|
ty::TyGenerator(def_id, substs, _) => {
|
|
|
|
let sig = substs.poly_sig(def_id, tcx);
|
|
|
|
|
|
|
|
let env_region = ty::ReLateBound(ty::INNERMOST, ty::BrEnv);
|
|
|
|
let env_ty = tcx.mk_mut_ref(tcx.mk_region(env_region), ty);
|
|
|
|
|
|
|
|
sig.map_bound(|sig| {
|
|
|
|
let state_did = tcx.lang_items().gen_state().unwrap();
|
|
|
|
let state_adt_ref = tcx.adt_def(state_did);
|
|
|
|
let state_substs = tcx.intern_substs(&[
|
|
|
|
sig.yield_ty.into(),
|
|
|
|
sig.return_ty.into(),
|
|
|
|
]);
|
|
|
|
let ret_ty = tcx.mk_adt(state_adt_ref, state_substs);
|
|
|
|
|
|
|
|
tcx.mk_fn_sig(iter::once(env_ty),
|
|
|
|
ret_ty,
|
|
|
|
false,
|
|
|
|
hir::Unsafety::Normal,
|
|
|
|
Abi::Rust
|
|
|
|
)
|
|
|
|
})
|
|
|
|
}
|
|
|
|
_ => bug!("unexpected type {:?} to ty_fn_sig", ty)
|
2018-07-20 06:38:49 -05:00
|
|
|
};
|
|
|
|
tcx.normalize_erasing_late_bound_regions(ParamEnv::reveal_all(), &sig)
|
2018-07-20 06:36:53 -05:00
|
|
|
}
|
|
|
|
|
2018-08-11 07:10:00 -05:00
|
|
|
pub fn get_function_name_and_sig<'a, 'tcx>(
|
2018-08-11 06:59:08 -05:00
|
|
|
tcx: TyCtxt<'a, 'tcx, 'tcx>,
|
|
|
|
inst: Instance<'tcx>,
|
|
|
|
) -> (String, Signature) {
|
|
|
|
assert!(!inst.substs.needs_infer() && !inst.substs.has_param_types());
|
|
|
|
let fn_ty = inst.ty(tcx);
|
|
|
|
let sig = cton_sig_from_fn_ty(tcx, fn_ty);
|
2018-08-11 07:29:32 -05:00
|
|
|
(tcx.symbol_name(inst).as_str().to_string(), sig)
|
2018-08-11 06:59:08 -05:00
|
|
|
}
|
|
|
|
|
2018-08-14 13:31:16 -05:00
|
|
|
impl<'a, 'tcx: 'a, B: Backend + 'a> FunctionCx<'a, 'tcx, B> {
|
2018-07-20 06:36:53 -05:00
|
|
|
/// Instance must be monomorphized
|
2018-07-19 12:33:42 -05:00
|
|
|
pub fn get_function_ref(&mut self, inst: Instance<'tcx>) -> FuncRef {
|
2018-08-11 06:59:08 -05:00
|
|
|
let (name, sig) = get_function_name_and_sig(self.tcx, inst);
|
2018-07-31 05:25:16 -05:00
|
|
|
let func_id = self
|
|
|
|
.module
|
|
|
|
.declare_function(&name, Linkage::Import, &sig)
|
|
|
|
.unwrap();
|
|
|
|
self.module
|
|
|
|
.declare_func_in_func(func_id, &mut self.bcx.func)
|
2018-07-19 12:33:42 -05:00
|
|
|
}
|
2018-07-20 06:36:53 -05:00
|
|
|
|
2018-07-30 08:58:46 -05:00
|
|
|
fn lib_call(
|
2018-07-30 08:34:34 -05:00
|
|
|
&mut self,
|
|
|
|
name: &str,
|
|
|
|
input_tys: Vec<types::Type>,
|
2018-07-30 08:58:46 -05:00
|
|
|
output_ty: Option<types::Type>,
|
2018-07-30 08:34:34 -05:00
|
|
|
args: &[Value],
|
2018-07-30 08:58:46 -05:00
|
|
|
) -> Option<Value> {
|
2018-07-30 08:34:34 -05:00
|
|
|
let sig = Signature {
|
|
|
|
params: input_tys.iter().cloned().map(AbiParam::new).collect(),
|
2018-07-30 08:58:46 -05:00
|
|
|
returns: vec![AbiParam::new(output_ty.unwrap_or(types::VOID))],
|
2018-07-30 08:34:34 -05:00
|
|
|
call_conv: CallConv::SystemV,
|
|
|
|
argument_bytes: None,
|
|
|
|
};
|
2018-07-31 05:25:16 -05:00
|
|
|
let func_id = self
|
|
|
|
.module
|
|
|
|
.declare_function(&name, Linkage::Import, &sig)
|
|
|
|
.unwrap();
|
|
|
|
let func_ref = self
|
|
|
|
.module
|
|
|
|
.declare_func_in_func(func_id, &mut self.bcx.func);
|
2018-07-30 08:34:34 -05:00
|
|
|
let call_inst = self.bcx.ins().call(func_ref, args);
|
2018-07-30 08:58:46 -05:00
|
|
|
if output_ty.is_none() {
|
|
|
|
return None;
|
|
|
|
}
|
2018-07-30 08:34:34 -05:00
|
|
|
let results = self.bcx.inst_results(call_inst);
|
|
|
|
assert_eq!(results.len(), 1);
|
2018-07-30 08:58:46 -05:00
|
|
|
Some(results[0])
|
2018-07-30 08:34:34 -05:00
|
|
|
}
|
|
|
|
|
2018-07-31 05:25:16 -05:00
|
|
|
pub fn easy_call(
|
|
|
|
&mut self,
|
|
|
|
name: &str,
|
|
|
|
args: &[CValue<'tcx>],
|
|
|
|
return_ty: Ty<'tcx>,
|
|
|
|
) -> CValue<'tcx> {
|
|
|
|
let (input_tys, args): (Vec<_>, Vec<_>) = args
|
|
|
|
.into_iter()
|
|
|
|
.map(|arg| {
|
|
|
|
(
|
|
|
|
self.cton_type(arg.layout().ty).unwrap(),
|
|
|
|
arg.load_value(self),
|
|
|
|
)
|
|
|
|
}).unzip();
|
2018-07-30 08:34:34 -05:00
|
|
|
let return_layout = self.layout_of(return_ty);
|
2018-07-30 08:58:46 -05:00
|
|
|
let return_ty = if let TypeVariants::TyTuple(tup) = return_ty.sty {
|
|
|
|
if !tup.is_empty() {
|
|
|
|
bug!("easy_call( (...) -> <non empty tuple> ) is not allowed");
|
|
|
|
}
|
|
|
|
None
|
|
|
|
} else {
|
|
|
|
Some(self.cton_type(return_ty).unwrap())
|
|
|
|
};
|
|
|
|
if let Some(val) = self.lib_call(name, input_tys, return_ty, &args) {
|
|
|
|
CValue::ByVal(val, return_layout)
|
|
|
|
} else {
|
2018-08-22 05:31:45 -05:00
|
|
|
CValue::ByRef(
|
|
|
|
self.bcx.ins().iconst(self.module.pointer_type(), 0),
|
|
|
|
return_layout,
|
|
|
|
)
|
2018-07-30 08:58:46 -05:00
|
|
|
}
|
2018-07-30 08:34:34 -05:00
|
|
|
}
|
|
|
|
|
2018-07-20 06:36:53 -05:00
|
|
|
fn self_sig(&self) -> FnSig<'tcx> {
|
2018-07-20 06:38:49 -05:00
|
|
|
ty_fn_sig(self.tcx, self.instance.ty(self.tcx))
|
2018-07-20 06:36:53 -05:00
|
|
|
}
|
|
|
|
|
|
|
|
fn return_type(&self) -> Ty<'tcx> {
|
|
|
|
self.self_sig().output()
|
|
|
|
}
|
2018-07-19 12:33:42 -05:00
|
|
|
}
|
|
|
|
|
2018-08-14 13:31:16 -05:00
|
|
|
pub fn codegen_fn_prelude<'a, 'tcx: 'a>(
|
|
|
|
fx: &mut FunctionCx<'a, 'tcx, impl Backend>,
|
|
|
|
start_ebb: Ebb,
|
|
|
|
) {
|
2018-08-09 03:46:56 -05:00
|
|
|
let ssa_analyzed = crate::analyze::analyze(fx);
|
|
|
|
|
2018-07-30 11:20:37 -05:00
|
|
|
match fx.self_sig().abi {
|
|
|
|
Abi::Rust | Abi::RustCall => {}
|
|
|
|
_ => unimplemented!("declared function with non \"rust\" or \"rust-call\" abi"),
|
|
|
|
}
|
|
|
|
|
2018-08-11 05:34:46 -05:00
|
|
|
let ret_layout = fx.layout_of(fx.return_type());
|
|
|
|
let output_pass_mode = get_pass_mode(fx.tcx, fx.self_sig().abi, fx.return_type(), true);
|
|
|
|
let ret_param = match output_pass_mode {
|
2018-08-11 06:59:34 -05:00
|
|
|
PassMode::NoPass => None,
|
2018-08-11 08:21:26 -05:00
|
|
|
PassMode::ByVal(_) => None,
|
2018-08-18 10:10:02 -05:00
|
|
|
PassMode::ByRef => Some(fx.bcx.append_ebb_param(start_ebb, fx.module.pointer_type())),
|
2018-08-11 05:34:46 -05:00
|
|
|
};
|
2018-07-19 12:33:42 -05:00
|
|
|
|
2018-07-27 12:27:20 -05:00
|
|
|
enum ArgKind {
|
|
|
|
Normal(Value),
|
|
|
|
Spread(Vec<Value>),
|
|
|
|
}
|
|
|
|
|
2018-08-14 05:13:07 -05:00
|
|
|
let func_params = fx
|
|
|
|
.mir
|
|
|
|
.args_iter()
|
|
|
|
.map(|local| {
|
|
|
|
let arg_ty = fx.monomorphize(&fx.mir.local_decls[local].ty);
|
|
|
|
|
|
|
|
// Adapted from https://github.com/rust-lang/rust/blob/145155dc96757002c7b2e9de8489416e2fdbbd57/src/librustc_codegen_llvm/mir/mod.rs#L442-L482
|
|
|
|
if Some(local) == fx.mir.spread_arg {
|
|
|
|
// This argument (e.g. the last argument in the "rust-call" ABI)
|
|
|
|
// is a tuple that was spread at the ABI level and now we have
|
|
|
|
// to reconstruct it into a tuple local variable, from multiple
|
|
|
|
// individual function arguments.
|
|
|
|
|
|
|
|
let tupled_arg_tys = match arg_ty.sty {
|
|
|
|
ty::TyTuple(ref tys) => tys,
|
|
|
|
_ => bug!("spread argument isn't a tuple?! but {:?}", arg_ty),
|
|
|
|
};
|
|
|
|
|
|
|
|
let mut ebb_params = Vec::new();
|
|
|
|
for arg_ty in tupled_arg_tys.iter() {
|
|
|
|
let cton_type =
|
|
|
|
get_pass_mode(fx.tcx, fx.self_sig().abi, arg_ty, false).get_param_ty(fx);
|
|
|
|
ebb_params.push(fx.bcx.append_ebb_param(start_ebb, cton_type));
|
|
|
|
}
|
|
|
|
|
|
|
|
(local, ArgKind::Spread(ebb_params), arg_ty)
|
|
|
|
} else {
|
|
|
|
let cton_type =
|
|
|
|
get_pass_mode(fx.tcx, fx.self_sig().abi, arg_ty, false).get_param_ty(fx);
|
|
|
|
(
|
|
|
|
local,
|
|
|
|
ArgKind::Normal(fx.bcx.append_ebb_param(start_ebb, cton_type)),
|
|
|
|
arg_ty,
|
|
|
|
)
|
2018-07-27 12:27:20 -05:00
|
|
|
}
|
2018-08-14 05:13:07 -05:00
|
|
|
}).collect::<Vec<(Local, ArgKind, Ty)>>();
|
2018-07-27 12:27:20 -05:00
|
|
|
|
2018-08-14 11:52:43 -05:00
|
|
|
fx.bcx.switch_to_block(start_ebb);
|
|
|
|
|
2018-08-15 09:17:59 -05:00
|
|
|
fx.top_nop = Some(fx.bcx.ins().nop());
|
|
|
|
fx.add_global_comment(format!("ssa {:?}", ssa_analyzed));
|
|
|
|
|
2018-08-11 05:34:46 -05:00
|
|
|
match output_pass_mode {
|
|
|
|
PassMode::NoPass => {
|
2018-08-18 10:10:02 -05:00
|
|
|
let null = fx.bcx.ins().iconst(fx.module.pointer_type(), 0);
|
2018-08-11 05:34:46 -05:00
|
|
|
//unimplemented!("pass mode nopass");
|
2018-08-11 06:59:34 -05:00
|
|
|
fx.local_map.insert(
|
|
|
|
RETURN_PLACE,
|
|
|
|
CPlace::Addr(null, fx.layout_of(fx.return_type())),
|
|
|
|
);
|
2018-08-11 05:34:46 -05:00
|
|
|
}
|
|
|
|
PassMode::ByVal(ret_ty) => {
|
2018-08-15 05:36:13 -05:00
|
|
|
fx.bcx.declare_var(mir_var(RETURN_PLACE), ret_ty);
|
2018-08-11 05:34:46 -05:00
|
|
|
fx.local_map
|
2018-08-15 05:36:13 -05:00
|
|
|
.insert(RETURN_PLACE, CPlace::Var(RETURN_PLACE, ret_layout));
|
2018-08-11 05:34:46 -05:00
|
|
|
}
|
|
|
|
PassMode::ByRef => {
|
|
|
|
fx.local_map
|
|
|
|
.insert(RETURN_PLACE, CPlace::Addr(ret_param.unwrap(), ret_layout));
|
|
|
|
}
|
|
|
|
}
|
2018-07-27 12:27:20 -05:00
|
|
|
|
|
|
|
for (local, arg_kind, ty) in func_params {
|
|
|
|
let layout = fx.layout_of(ty);
|
2018-08-09 03:46:56 -05:00
|
|
|
|
|
|
|
if let ArgKind::Normal(ebb_param) = arg_kind {
|
2018-08-09 04:25:14 -05:00
|
|
|
if !ssa_analyzed
|
|
|
|
.get(&local)
|
|
|
|
.unwrap()
|
|
|
|
.contains(crate::analyze::Flags::NOT_SSA)
|
|
|
|
{
|
2018-08-15 07:45:32 -05:00
|
|
|
fx.bcx
|
|
|
|
.declare_var(mir_var(local), fx.cton_type(ty).unwrap());
|
2018-08-11 05:34:46 -05:00
|
|
|
match get_pass_mode(fx.tcx, fx.self_sig().abi, ty, false) {
|
|
|
|
PassMode::NoPass => unimplemented!("pass mode nopass"),
|
2018-08-15 05:36:13 -05:00
|
|
|
PassMode::ByVal(_) => fx.bcx.def_var(mir_var(local), ebb_param),
|
2018-08-11 05:34:46 -05:00
|
|
|
PassMode::ByRef => {
|
|
|
|
let val = CValue::ByRef(ebb_param, fx.layout_of(ty)).load_value(fx);
|
2018-08-15 05:36:13 -05:00
|
|
|
fx.bcx.def_var(mir_var(local), val);
|
2018-08-11 05:34:46 -05:00
|
|
|
}
|
|
|
|
}
|
2018-08-15 05:36:13 -05:00
|
|
|
fx.local_map.insert(local, CPlace::Var(local, layout));
|
2018-08-09 03:46:56 -05:00
|
|
|
continue;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-07-19 12:33:42 -05:00
|
|
|
let stack_slot = fx.bcx.create_stack_slot(StackSlotData {
|
|
|
|
kind: StackSlotKind::ExplicitSlot,
|
|
|
|
size: layout.size.bytes() as u32,
|
|
|
|
offset: None,
|
|
|
|
});
|
|
|
|
|
|
|
|
let place = CPlace::from_stack_slot(fx, stack_slot, ty);
|
2018-07-27 12:27:20 -05:00
|
|
|
|
|
|
|
match arg_kind {
|
2018-08-11 06:59:34 -05:00
|
|
|
ArgKind::Normal(ebb_param) => match get_pass_mode(fx.tcx, fx.self_sig().abi, ty, false)
|
|
|
|
{
|
|
|
|
PassMode::NoPass => unimplemented!("pass mode nopass"),
|
|
|
|
PassMode::ByVal(_) => {
|
|
|
|
place.write_cvalue(fx, CValue::ByVal(ebb_param, place.layout()))
|
2018-07-27 12:27:20 -05:00
|
|
|
}
|
2018-08-11 06:59:34 -05:00
|
|
|
PassMode::ByRef => place.write_cvalue(fx, CValue::ByRef(ebb_param, place.layout())),
|
|
|
|
},
|
2018-07-27 12:27:20 -05:00
|
|
|
ArgKind::Spread(ebb_params) => {
|
|
|
|
for (i, ebb_param) in ebb_params.into_iter().enumerate() {
|
|
|
|
let sub_place = place.place_field(fx, mir::Field::new(i));
|
2018-08-11 05:34:46 -05:00
|
|
|
match get_pass_mode(fx.tcx, fx.self_sig().abi, sub_place.layout().ty, false) {
|
|
|
|
PassMode::NoPass => unimplemented!("pass mode nopass"),
|
2018-08-11 06:59:34 -05:00
|
|
|
PassMode::ByVal(_) => {
|
|
|
|
sub_place.write_cvalue(fx, CValue::ByVal(ebb_param, sub_place.layout()))
|
|
|
|
}
|
|
|
|
PassMode::ByRef => {
|
|
|
|
sub_place.write_cvalue(fx, CValue::ByRef(ebb_param, sub_place.layout()))
|
|
|
|
}
|
2018-07-27 12:27:20 -05:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
2018-07-19 12:33:42 -05:00
|
|
|
}
|
|
|
|
fx.local_map.insert(local, place);
|
|
|
|
}
|
|
|
|
|
|
|
|
for local in fx.mir.vars_and_temps_iter() {
|
|
|
|
let ty = fx.mir.local_decls[local].ty;
|
|
|
|
let layout = fx.layout_of(ty);
|
2018-08-09 03:46:56 -05:00
|
|
|
|
2018-08-09 04:25:14 -05:00
|
|
|
let place = if ssa_analyzed
|
|
|
|
.get(&local)
|
|
|
|
.unwrap()
|
|
|
|
.contains(crate::analyze::Flags::NOT_SSA)
|
|
|
|
{
|
2018-08-09 03:46:56 -05:00
|
|
|
let stack_slot = fx.bcx.create_stack_slot(StackSlotData {
|
|
|
|
kind: StackSlotKind::ExplicitSlot,
|
|
|
|
size: layout.size.bytes() as u32,
|
|
|
|
offset: None,
|
|
|
|
});
|
|
|
|
CPlace::from_stack_slot(fx, stack_slot, ty)
|
|
|
|
} else {
|
2018-08-15 07:45:32 -05:00
|
|
|
fx.bcx
|
|
|
|
.declare_var(mir_var(local), fx.cton_type(ty).unwrap());
|
2018-08-15 05:36:13 -05:00
|
|
|
CPlace::Var(local, layout)
|
2018-08-09 03:46:56 -05:00
|
|
|
};
|
|
|
|
|
2018-07-19 12:33:42 -05:00
|
|
|
fx.local_map.insert(local, place);
|
|
|
|
}
|
2018-08-14 11:52:43 -05:00
|
|
|
|
|
|
|
fx.bcx
|
|
|
|
.ins()
|
|
|
|
.jump(*fx.ebb_map.get(&START_BLOCK).unwrap(), &[]);
|
2018-07-19 12:33:42 -05:00
|
|
|
}
|
|
|
|
|
|
|
|
pub fn codegen_call<'a, 'tcx: 'a>(
|
2018-08-14 13:31:16 -05:00
|
|
|
fx: &mut FunctionCx<'a, 'tcx, impl Backend>,
|
2018-07-19 12:33:42 -05:00
|
|
|
func: &Operand<'tcx>,
|
|
|
|
args: &[Operand<'tcx>],
|
|
|
|
destination: &Option<(Place<'tcx>, BasicBlock)>,
|
2018-07-20 06:51:34 -05:00
|
|
|
) {
|
2018-08-11 07:24:05 -05:00
|
|
|
let fn_ty = fx.monomorphize(&func.ty(&fx.mir.local_decls, fx.tcx));
|
2018-07-20 06:38:49 -05:00
|
|
|
let sig = ty_fn_sig(fx.tcx, fn_ty);
|
2018-07-21 11:38:08 -05:00
|
|
|
|
2018-07-20 06:38:49 -05:00
|
|
|
// Unpack arguments tuple for closures
|
|
|
|
let args = if sig.abi == Abi::RustCall {
|
|
|
|
assert_eq!(args.len(), 2, "rust-call abi requires two arguments");
|
2018-07-30 09:57:40 -05:00
|
|
|
let self_arg = trans_operand(fx, &args[0]);
|
|
|
|
let pack_arg = trans_operand(fx, &args[1]);
|
2018-07-20 06:38:49 -05:00
|
|
|
let mut args = Vec::new();
|
|
|
|
args.push(self_arg);
|
|
|
|
match pack_arg.layout().ty.sty {
|
|
|
|
ty::TyTuple(ref tupled_arguments) => {
|
|
|
|
for (i, _) in tupled_arguments.iter().enumerate() {
|
|
|
|
args.push(pack_arg.value_field(fx, mir::Field::new(i)));
|
|
|
|
}
|
2018-07-31 05:25:16 -05:00
|
|
|
}
|
2018-07-20 06:38:49 -05:00
|
|
|
_ => bug!("argument to function with \"rust-call\" ABI is not a tuple"),
|
|
|
|
}
|
|
|
|
args
|
|
|
|
} else {
|
2018-07-31 05:25:16 -05:00
|
|
|
args.into_iter()
|
|
|
|
.map(|arg| trans_operand(fx, arg))
|
2018-07-20 06:38:49 -05:00
|
|
|
.collect::<Vec<_>>()
|
|
|
|
};
|
2018-07-21 11:38:08 -05:00
|
|
|
|
2018-08-11 06:59:34 -05:00
|
|
|
let destination = destination
|
|
|
|
.as_ref()
|
|
|
|
.map(|(place, bb)| (trans_place(fx, place), *bb));
|
2018-08-11 03:58:31 -05:00
|
|
|
|
2018-08-10 12:43:01 -05:00
|
|
|
if codegen_intrinsic_call(fx, fn_ty, sig, &args, destination) {
|
|
|
|
return;
|
2018-07-21 11:38:08 -05:00
|
|
|
}
|
|
|
|
|
2018-08-11 05:34:46 -05:00
|
|
|
let ret_layout = fx.layout_of(sig.output());
|
|
|
|
|
|
|
|
let output_pass_mode = get_pass_mode(fx.tcx, sig.abi, sig.output(), true);
|
|
|
|
let return_ptr = match output_pass_mode {
|
|
|
|
PassMode::NoPass => None,
|
|
|
|
PassMode::ByRef => match destination {
|
|
|
|
Some((place, _)) => Some(place.expect_addr()),
|
2018-08-18 10:10:02 -05:00
|
|
|
None => Some(fx.bcx.ins().iconst(fx.module.pointer_type(), 0)),
|
2018-08-11 05:34:46 -05:00
|
|
|
},
|
|
|
|
PassMode::ByVal(_) => None,
|
2018-07-21 11:38:08 -05:00
|
|
|
};
|
|
|
|
|
2018-08-11 05:34:46 -05:00
|
|
|
let call_args: Vec<Value> = return_ptr
|
2018-07-31 05:25:16 -05:00
|
|
|
.into_iter()
|
2018-08-11 06:59:34 -05:00
|
|
|
.chain(args.into_iter().map(|arg| {
|
|
|
|
match get_pass_mode(fx.tcx, sig.abi, arg.layout().ty, false) {
|
2018-08-11 06:59:08 -05:00
|
|
|
PassMode::NoPass => unimplemented!("pass mode nopass"),
|
|
|
|
PassMode::ByVal(_) => arg.load_value(fx),
|
|
|
|
PassMode::ByRef => arg.force_stack(fx),
|
2018-08-11 06:59:34 -05:00
|
|
|
}
|
|
|
|
})).collect::<Vec<_>>();
|
2018-08-11 05:34:46 -05:00
|
|
|
|
2018-08-19 03:50:39 -05:00
|
|
|
let call_inst = match fn_ty.sty {
|
|
|
|
TypeVariants::TyFnDef(def_id, substs) => {
|
|
|
|
let func_ref = fx.get_function_ref(
|
|
|
|
Instance::resolve(fx.tcx, ParamEnv::reveal_all(), def_id, substs).unwrap(),
|
|
|
|
);
|
|
|
|
fx.bcx.ins().call(func_ref, &call_args)
|
|
|
|
}
|
|
|
|
_ => {
|
|
|
|
let func = trans_operand(fx, func);
|
2018-07-19 12:33:42 -05:00
|
|
|
let func = func.load_value(fx);
|
2018-08-11 05:34:46 -05:00
|
|
|
let sig = fx.bcx.import_signature(cton_sig_from_fn_ty(fx.tcx, fn_ty));
|
|
|
|
fx.bcx.ins().call_indirect(sig, func, &call_args)
|
2018-07-19 12:33:42 -05:00
|
|
|
}
|
2018-08-11 05:34:46 -05:00
|
|
|
};
|
|
|
|
|
|
|
|
match output_pass_mode {
|
|
|
|
PassMode::NoPass => {}
|
|
|
|
PassMode::ByVal(_) => {
|
|
|
|
if let Some((ret_place, _)) = destination {
|
2018-08-19 03:50:39 -05:00
|
|
|
let results = fx.bcx.inst_results(call_inst);
|
2018-08-11 05:34:46 -05:00
|
|
|
ret_place.write_cvalue(fx, CValue::ByVal(results[0], ret_layout));
|
|
|
|
}
|
|
|
|
}
|
|
|
|
PassMode::ByRef => {}
|
2018-07-20 06:51:34 -05:00
|
|
|
}
|
2018-08-11 03:58:31 -05:00
|
|
|
if let Some((_, dest)) = destination {
|
2018-07-19 12:33:42 -05:00
|
|
|
let ret_ebb = fx.get_ebb(dest);
|
|
|
|
fx.bcx.ins().jump(ret_ebb, &[]);
|
|
|
|
} else {
|
|
|
|
fx.bcx.ins().trap(TrapCode::User(!0));
|
|
|
|
}
|
|
|
|
}
|
2018-08-10 12:20:13 -05:00
|
|
|
|
2018-08-14 13:31:16 -05:00
|
|
|
pub fn codegen_return(fx: &mut FunctionCx<impl Backend>) {
|
2018-08-11 05:34:46 -05:00
|
|
|
match get_pass_mode(fx.tcx, fx.self_sig().abi, fx.return_type(), true) {
|
|
|
|
PassMode::NoPass | PassMode::ByRef => {
|
|
|
|
fx.bcx.ins().return_(&[]);
|
2018-08-11 06:59:34 -05:00
|
|
|
}
|
2018-08-11 05:34:46 -05:00
|
|
|
PassMode::ByVal(_) => {
|
|
|
|
let place = fx.get_local_place(RETURN_PLACE);
|
|
|
|
let ret_val = place.to_cvalue(fx).load_value(fx);
|
|
|
|
fx.bcx.ins().return_(&[ret_val]);
|
|
|
|
}
|
|
|
|
}
|
2018-08-11 04:01:48 -05:00
|
|
|
}
|
|
|
|
|
2018-08-10 12:20:13 -05:00
|
|
|
fn codegen_intrinsic_call<'a, 'tcx: 'a>(
|
2018-08-14 13:31:16 -05:00
|
|
|
fx: &mut FunctionCx<'a, 'tcx, impl Backend>,
|
2018-08-10 12:43:01 -05:00
|
|
|
fn_ty: Ty<'tcx>,
|
|
|
|
sig: FnSig<'tcx>,
|
|
|
|
args: &[CValue<'tcx>],
|
2018-08-11 03:58:31 -05:00
|
|
|
destination: Option<(CPlace<'tcx>, BasicBlock)>,
|
2018-08-10 12:43:01 -05:00
|
|
|
) -> bool {
|
|
|
|
if let TypeVariants::TyFnDef(def_id, substs) = fn_ty.sty {
|
|
|
|
if sig.abi == Abi::RustIntrinsic {
|
|
|
|
let intrinsic = fx.tcx.item_name(def_id).as_str();
|
|
|
|
let intrinsic = &intrinsic[..];
|
2018-08-10 12:20:13 -05:00
|
|
|
|
2018-08-10 12:43:01 -05:00
|
|
|
let ret = match destination {
|
2018-08-11 03:58:31 -05:00
|
|
|
Some((place, _)) => place,
|
2018-08-10 12:43:01 -05:00
|
|
|
None => {
|
|
|
|
// Insert non returning intrinsics here
|
|
|
|
match intrinsic {
|
|
|
|
"abort" => {
|
|
|
|
fx.bcx.ins().trap(TrapCode::User(!0 - 1));
|
|
|
|
}
|
|
|
|
"unreachable" => {
|
|
|
|
fx.bcx.ins().trap(TrapCode::User(!0 - 1));
|
|
|
|
}
|
|
|
|
_ => unimplemented!("unsupported instrinsic {}", intrinsic),
|
|
|
|
}
|
|
|
|
return true;
|
|
|
|
}
|
2018-08-10 12:20:13 -05:00
|
|
|
};
|
|
|
|
|
2018-08-10 12:43:01 -05:00
|
|
|
let nil_ty = fx.tcx.mk_nil();
|
|
|
|
let u64_layout = fx.layout_of(fx.tcx.types.u64);
|
|
|
|
let usize_layout = fx.layout_of(fx.tcx.types.usize);
|
|
|
|
|
|
|
|
match intrinsic {
|
|
|
|
"assume" => {
|
|
|
|
assert_eq!(args.len(), 1);
|
|
|
|
}
|
|
|
|
"arith_offset" => {
|
|
|
|
assert_eq!(args.len(), 2);
|
|
|
|
let base = args[0].load_value(fx);
|
|
|
|
let offset = args[1].load_value(fx);
|
|
|
|
let res = fx.bcx.ins().iadd(base, offset);
|
|
|
|
let res = CValue::ByVal(res, ret.layout());
|
|
|
|
ret.write_cvalue(fx, res);
|
|
|
|
}
|
|
|
|
"likely" | "unlikely" => {
|
|
|
|
assert_eq!(args.len(), 1);
|
|
|
|
ret.write_cvalue(fx, args[0]);
|
|
|
|
}
|
|
|
|
"copy" | "copy_nonoverlapping" => {
|
|
|
|
let elem_ty = substs.type_at(0);
|
|
|
|
let elem_size: u64 = fx.layout_of(elem_ty).size.bytes();
|
2018-08-22 05:31:45 -05:00
|
|
|
let elem_size = fx
|
|
|
|
.bcx
|
|
|
|
.ins()
|
|
|
|
.iconst(fx.module.pointer_type(), elem_size as i64);
|
2018-08-10 12:43:01 -05:00
|
|
|
assert_eq!(args.len(), 3);
|
|
|
|
let src = args[0];
|
|
|
|
let dst = args[1];
|
|
|
|
let count = args[2].load_value(fx);
|
|
|
|
let byte_amount = fx.bcx.ins().imul(count, elem_size);
|
|
|
|
fx.easy_call(
|
|
|
|
"memmove",
|
|
|
|
&[dst, src, CValue::ByVal(byte_amount, usize_layout)],
|
|
|
|
nil_ty,
|
|
|
|
);
|
|
|
|
}
|
|
|
|
"discriminant_value" => {
|
|
|
|
assert_eq!(args.len(), 1);
|
|
|
|
let discr = crate::base::trans_get_discriminant(fx, args[0], ret.layout());
|
|
|
|
ret.write_cvalue(fx, discr);
|
|
|
|
}
|
|
|
|
"size_of" => {
|
|
|
|
assert_eq!(args.len(), 0);
|
|
|
|
let size_of = fx.layout_of(substs.type_at(0)).size.bytes();
|
|
|
|
let size_of = CValue::const_val(fx, usize_layout.ty, size_of as i64);
|
|
|
|
ret.write_cvalue(fx, size_of);
|
|
|
|
}
|
|
|
|
"type_id" => {
|
|
|
|
assert_eq!(args.len(), 0);
|
|
|
|
let type_id = fx.tcx.type_id_hash(substs.type_at(0));
|
|
|
|
let type_id = CValue::const_val(fx, u64_layout.ty, type_id as i64);
|
|
|
|
ret.write_cvalue(fx, type_id);
|
|
|
|
}
|
|
|
|
"min_align_of" => {
|
|
|
|
assert_eq!(args.len(), 0);
|
|
|
|
let min_align = fx.layout_of(substs.type_at(0)).align.abi();
|
|
|
|
let min_align = CValue::const_val(fx, usize_layout.ty, min_align as i64);
|
|
|
|
ret.write_cvalue(fx, min_align);
|
|
|
|
}
|
|
|
|
_ if intrinsic.starts_with("unchecked_") => {
|
|
|
|
assert_eq!(args.len(), 2);
|
|
|
|
let bin_op = match intrinsic {
|
|
|
|
"unchecked_div" => BinOp::Div,
|
|
|
|
"unchecked_rem" => BinOp::Rem,
|
|
|
|
"unchecked_shl" => BinOp::Shl,
|
|
|
|
"unchecked_shr" => BinOp::Shr,
|
|
|
|
_ => unimplemented!("intrinsic {}", intrinsic),
|
|
|
|
};
|
|
|
|
let res = match ret.layout().ty.sty {
|
|
|
|
TypeVariants::TyUint(_) => crate::base::trans_int_binop(
|
|
|
|
fx,
|
|
|
|
bin_op,
|
|
|
|
args[0],
|
|
|
|
args[1],
|
|
|
|
ret.layout().ty,
|
|
|
|
false,
|
|
|
|
),
|
|
|
|
TypeVariants::TyInt(_) => crate::base::trans_int_binop(
|
|
|
|
fx,
|
|
|
|
bin_op,
|
|
|
|
args[0],
|
|
|
|
args[1],
|
|
|
|
ret.layout().ty,
|
|
|
|
true,
|
|
|
|
),
|
|
|
|
_ => panic!(),
|
|
|
|
};
|
|
|
|
ret.write_cvalue(fx, res);
|
|
|
|
}
|
|
|
|
_ if intrinsic.ends_with("_with_overflow") => {
|
|
|
|
assert_eq!(args.len(), 2);
|
|
|
|
assert_eq!(args[0].layout().ty, args[1].layout().ty);
|
|
|
|
let bin_op = match intrinsic {
|
|
|
|
"add_with_overflow" => BinOp::Add,
|
|
|
|
"sub_with_overflow" => BinOp::Sub,
|
|
|
|
"mul_with_overflow" => BinOp::Mul,
|
|
|
|
_ => unimplemented!("intrinsic {}", intrinsic),
|
|
|
|
};
|
|
|
|
let res = match args[0].layout().ty.sty {
|
|
|
|
TypeVariants::TyUint(_) => crate::base::trans_checked_int_binop(
|
|
|
|
fx,
|
|
|
|
bin_op,
|
|
|
|
args[0],
|
|
|
|
args[1],
|
|
|
|
ret.layout().ty,
|
|
|
|
false,
|
|
|
|
),
|
|
|
|
TypeVariants::TyInt(_) => crate::base::trans_checked_int_binop(
|
|
|
|
fx,
|
|
|
|
bin_op,
|
|
|
|
args[0],
|
|
|
|
args[1],
|
|
|
|
ret.layout().ty,
|
|
|
|
true,
|
|
|
|
),
|
|
|
|
_ => panic!(),
|
|
|
|
};
|
|
|
|
ret.write_cvalue(fx, res);
|
|
|
|
}
|
|
|
|
_ if intrinsic.starts_with("overflowing_") => {
|
|
|
|
assert_eq!(args.len(), 2);
|
|
|
|
assert_eq!(args[0].layout().ty, args[1].layout().ty);
|
|
|
|
let bin_op = match intrinsic {
|
|
|
|
"overflowing_add" => BinOp::Add,
|
|
|
|
"overflowing_sub" => BinOp::Sub,
|
|
|
|
"overflowing_mul" => BinOp::Mul,
|
|
|
|
_ => unimplemented!("intrinsic {}", intrinsic),
|
|
|
|
};
|
|
|
|
let res = match args[0].layout().ty.sty {
|
|
|
|
TypeVariants::TyUint(_) => crate::base::trans_int_binop(
|
|
|
|
fx,
|
|
|
|
bin_op,
|
|
|
|
args[0],
|
|
|
|
args[1],
|
|
|
|
ret.layout().ty,
|
|
|
|
false,
|
|
|
|
),
|
|
|
|
TypeVariants::TyInt(_) => crate::base::trans_int_binop(
|
|
|
|
fx,
|
|
|
|
bin_op,
|
|
|
|
args[0],
|
|
|
|
args[1],
|
|
|
|
ret.layout().ty,
|
|
|
|
true,
|
|
|
|
),
|
|
|
|
_ => panic!(),
|
|
|
|
};
|
|
|
|
ret.write_cvalue(fx, res);
|
|
|
|
}
|
|
|
|
"offset" => {
|
|
|
|
assert_eq!(args.len(), 2);
|
|
|
|
let base = args[0].load_value(fx);
|
|
|
|
let offset = args[1].load_value(fx);
|
|
|
|
let res = fx.bcx.ins().iadd(base, offset);
|
|
|
|
ret.write_cvalue(fx, CValue::ByVal(res, args[0].layout()));
|
|
|
|
}
|
|
|
|
"transmute" => {
|
|
|
|
assert_eq!(args.len(), 1);
|
|
|
|
let src_ty = substs.type_at(0);
|
|
|
|
let dst_ty = substs.type_at(1);
|
|
|
|
assert_eq!(args[0].layout().ty, src_ty);
|
|
|
|
let addr = args[0].force_stack(fx);
|
|
|
|
let dst_layout = fx.layout_of(dst_ty);
|
|
|
|
ret.write_cvalue(fx, CValue::ByRef(addr, dst_layout))
|
|
|
|
}
|
|
|
|
"uninit" => {
|
|
|
|
assert_eq!(args.len(), 0);
|
|
|
|
let ty = substs.type_at(0);
|
|
|
|
let layout = fx.layout_of(ty);
|
|
|
|
let stack_slot = fx.bcx.create_stack_slot(StackSlotData {
|
|
|
|
kind: StackSlotKind::ExplicitSlot,
|
|
|
|
size: layout.size.bytes() as u32,
|
|
|
|
offset: None,
|
|
|
|
});
|
|
|
|
|
|
|
|
let uninit_place = CPlace::from_stack_slot(fx, stack_slot, ty);
|
|
|
|
let uninit_val = uninit_place.to_cvalue(fx);
|
|
|
|
ret.write_cvalue(fx, uninit_val);
|
|
|
|
}
|
|
|
|
"ctlz" | "ctlz_nonzero" => {
|
|
|
|
assert_eq!(args.len(), 1);
|
|
|
|
let arg = args[0].load_value(fx);
|
|
|
|
let res = CValue::ByVal(fx.bcx.ins().clz(arg), args[0].layout());
|
|
|
|
ret.write_cvalue(fx, res);
|
|
|
|
}
|
|
|
|
"cttz" | "cttz_nonzero" => {
|
|
|
|
assert_eq!(args.len(), 1);
|
|
|
|
let arg = args[0].load_value(fx);
|
|
|
|
let res = CValue::ByVal(fx.bcx.ins().clz(arg), args[0].layout());
|
|
|
|
ret.write_cvalue(fx, res);
|
|
|
|
}
|
|
|
|
"ctpop" => {
|
|
|
|
assert_eq!(args.len(), 1);
|
|
|
|
let arg = args[0].load_value(fx);
|
|
|
|
let res = CValue::ByVal(fx.bcx.ins().popcnt(arg), args[0].layout());
|
|
|
|
ret.write_cvalue(fx, res);
|
|
|
|
}
|
|
|
|
_ => unimpl!("unsupported intrinsic {}", intrinsic),
|
|
|
|
}
|
|
|
|
|
2018-08-11 03:58:31 -05:00
|
|
|
if let Some((_, dest)) = destination {
|
2018-08-10 12:43:01 -05:00
|
|
|
let ret_ebb = fx.get_ebb(dest);
|
|
|
|
fx.bcx.ins().jump(ret_ebb, &[]);
|
|
|
|
} else {
|
|
|
|
fx.bcx.ins().trap(TrapCode::User(!0));
|
|
|
|
}
|
|
|
|
return true;
|
2018-08-10 12:20:13 -05:00
|
|
|
}
|
|
|
|
}
|
2018-08-10 12:43:01 -05:00
|
|
|
|
|
|
|
false
|
2018-08-10 12:20:13 -05:00
|
|
|
}
|