]> git.lizzy.rs Git - rust.git/blob - src/base.rs
Fix discriminant_value intrinsic
[rust.git] / src / base.rs
1 use crate::prelude::*;
2
3 struct PrintOnPanic<F: Fn() -> String>(F);
4 impl<F: Fn() -> String> Drop for PrintOnPanic<F> {
5     fn drop(&mut self) {
6         if ::std::thread::panicking() {
7             println!("{}", (self.0)());
8         }
9     }
10 }
11
12 pub fn trans_mono_item<'a, 'clif, 'tcx: 'a, B: Backend + 'static>(
13     cx: &mut crate::CodegenCx<'a, 'clif, 'tcx, B>,
14     mono_item: MonoItem<'tcx>,
15     linkage: Linkage,
16 ) {
17     let tcx = cx.tcx;
18     match mono_item {
19         MonoItem::Fn(inst) => {
20             let _inst_guard =
21                 PrintOnPanic(|| format!("{:?} {}", inst, tcx.symbol_name(inst).as_str()));
22             debug_assert!(!inst.substs.needs_infer());
23             let _mir_guard = PrintOnPanic(|| {
24                 match inst.def {
25                     InstanceDef::Item(_)
26                     | InstanceDef::DropGlue(_, _)
27                     | InstanceDef::Virtual(_, _)
28                         if inst.def_id().krate == LOCAL_CRATE =>
29                     {
30                         let mut mir = ::std::io::Cursor::new(Vec::new());
31                         crate::rustc_mir::util::write_mir_pretty(
32                             tcx,
33                             Some(inst.def_id()),
34                             &mut mir,
35                         )
36                         .unwrap();
37                         String::from_utf8(mir.into_inner()).unwrap()
38                     }
39                     _ => {
40                         // FIXME fix write_mir_pretty for these instances
41                         format!("{:#?}", tcx.instance_mir(inst.def))
42                     }
43                 }
44             });
45
46             trans_fn(cx, inst, linkage);
47         }
48         MonoItem::Static(def_id) => {
49             crate::constant::codegen_static(&mut cx.ccx, def_id);
50         }
51         MonoItem::GlobalAsm(node_id) => tcx
52             .sess
53             .fatal(&format!("Unimplemented global asm mono item {:?}", node_id)),
54     }
55 }
56
57 fn trans_fn<'a, 'clif, 'tcx: 'a, B: Backend + 'static>(
58     cx: &mut crate::CodegenCx<'a, 'clif, 'tcx, B>,
59     instance: Instance<'tcx>,
60     linkage: Linkage,
61 ) {
62     let tcx = cx.tcx;
63
64     // Step 1. Get mir
65     let mir = tcx.instance_mir(instance.def);
66
67     // Step 2. Declare function
68     let (name, sig) = get_function_name_and_sig(tcx, instance, false);
69     let func_id = cx.module
70         .declare_function(&name, linkage, &sig)
71         .unwrap();
72     let mut debug_context = cx.debug_context.as_mut().map(|debug_context| FunctionDebugContext::new(
73         tcx,
74         debug_context,
75         mir,
76         &name,
77         &sig,
78     ));
79
80     // Step 3. Make FunctionBuilder
81     let mut func = Function::with_name_signature(ExternalName::user(0, 0), sig);
82     let mut func_ctx = FunctionBuilderContext::new();
83     let mut bcx = FunctionBuilder::new(&mut func, &mut func_ctx);
84
85     // Step 4. Predefine ebb's
86     let start_ebb = bcx.create_ebb();
87     let mut ebb_map: HashMap<BasicBlock, Ebb> = HashMap::new();
88     for (bb, _bb_data) in mir.basic_blocks().iter_enumerated() {
89         ebb_map.insert(bb, bcx.create_ebb());
90     }
91
92     // Step 5. Make FunctionCx
93     let pointer_type = cx.module.target_config().pointer_type();
94     let clif_comments = crate::pretty_clif::CommentWriter::new(tcx, instance);
95
96     let mut fx = FunctionCx {
97         tcx,
98         module: cx.module,
99         pointer_type,
100
101         instance,
102         mir,
103
104         bcx,
105         ebb_map,
106         local_map: HashMap::new(),
107
108         clif_comments,
109         constants: &mut cx.ccx,
110         caches: &mut cx.caches,
111         source_info_set: indexmap::IndexSet::new(),
112     };
113
114     // Step 6. Codegen function
115     with_unimpl_span(fx.mir.span, || {
116         crate::abi::codegen_fn_prelude(&mut fx, start_ebb);
117         codegen_fn_content(&mut fx);
118     });
119     let source_info_set = fx.source_info_set.clone();
120
121     // Step 7. Write function to file for debugging
122     #[cfg(debug_assertions)]
123     fx.write_clif_file();
124
125     // Step 8. Verify function
126     verify_func(tcx, fx.clif_comments, &func);
127
128     // Step 9. Define function
129     cx.caches.context.func = func;
130     cx.module
131         .define_function_peek_compiled(func_id, &mut cx.caches.context, |size, context, isa| {
132             debug_context.as_mut().map(|x| x.define(tcx, size, context, isa, &source_info_set));
133         })
134         .unwrap();
135     //let module = &mut cx.module;
136     //let caches = &cx.caches;
137     cx.caches.context.clear();
138 }
139
140 fn verify_func(tcx: TyCtxt, writer: crate::pretty_clif::CommentWriter, func: &Function) {
141     let flags = settings::Flags::new(settings::builder());
142     match ::cranelift::codegen::verify_function(&func, &flags) {
143         Ok(_) => {}
144         Err(err) => {
145             tcx.sess.err(&format!("{:?}", err));
146             let pretty_error = ::cranelift::codegen::print_errors::pretty_verifier_error(
147                 &func,
148                 None,
149                 Some(Box::new(&writer)),
150                 err,
151             );
152             tcx.sess
153                 .fatal(&format!("cranelift verify error:\n{}", pretty_error));
154         }
155     }
156 }
157
158 fn codegen_fn_content<'a, 'tcx: 'a>(fx: &mut FunctionCx<'a, 'tcx, impl Backend>) {
159     for (bb, bb_data) in fx.mir.basic_blocks().iter_enumerated() {
160         if bb_data.is_cleanup {
161             // Unwinding after panicking is not supported
162             continue;
163         }
164
165         let ebb = fx.get_ebb(bb);
166         fx.bcx.switch_to_block(ebb);
167
168         fx.bcx.ins().nop();
169         for stmt in &bb_data.statements {
170             fx.set_debug_loc(stmt.source_info);
171             trans_stmt(fx, ebb, stmt);
172         }
173
174         #[cfg(debug_assertions)]
175         {
176             let mut terminator_head = "\n".to_string();
177             bb_data
178                 .terminator()
179                 .kind
180                 .fmt_head(&mut terminator_head)
181                 .unwrap();
182             let inst = fx.bcx.func.layout.last_inst(ebb).unwrap();
183             fx.add_comment(inst, terminator_head);
184         }
185
186         fx.set_debug_loc(bb_data.terminator().source_info);
187
188         match &bb_data.terminator().kind {
189             TerminatorKind::Goto { target } => {
190                 let ebb = fx.get_ebb(*target);
191                 fx.bcx.ins().jump(ebb, &[]);
192             }
193             TerminatorKind::Return => {
194                 crate::abi::codegen_return(fx);
195             }
196             TerminatorKind::Assert {
197                 cond,
198                 expected,
199                 msg: _,
200                 target,
201                 cleanup: _,
202             } => {
203                 let cond = trans_operand(fx, cond).load_scalar(fx);
204                 // TODO HACK brz/brnz for i8/i16 is not yet implemented
205                 let cond = fx.bcx.ins().uextend(types::I32, cond);
206                 let target = fx.get_ebb(*target);
207                 if *expected {
208                     fx.bcx.ins().brnz(cond, target, &[]);
209                 } else {
210                     fx.bcx.ins().brz(cond, target, &[]);
211                 };
212                 trap_panic(&mut fx.bcx);
213             }
214
215             TerminatorKind::SwitchInt {
216                 discr,
217                 switch_ty: _,
218                 values,
219                 targets,
220             } => {
221                 let discr = trans_operand(fx, discr).load_scalar(fx);
222                 let mut switch = ::cranelift::frontend::Switch::new();
223                 for (i, value) in values.iter().enumerate() {
224                     let ebb = fx.get_ebb(targets[i]);
225                     switch.set_entry(*value as u64, ebb);
226                 }
227                 let otherwise_ebb = fx.get_ebb(targets[targets.len() - 1]);
228                 switch.emit(&mut fx.bcx, discr, otherwise_ebb);
229             }
230             TerminatorKind::Call {
231                 func,
232                 args,
233                 destination,
234                 cleanup: _,
235                 from_hir_call: _,
236             } => {
237                 crate::abi::codegen_terminator_call(fx, func, args, destination);
238             }
239             TerminatorKind::Resume | TerminatorKind::Abort | TerminatorKind::Unreachable => {
240                 trap_unreachable(&mut fx.bcx);
241             }
242             TerminatorKind::Yield { .. }
243             | TerminatorKind::FalseEdges { .. }
244             | TerminatorKind::FalseUnwind { .. }
245             | TerminatorKind::DropAndReplace { .. }
246             | TerminatorKind::GeneratorDrop => {
247                 bug!("shouldn't exist at trans {:?}", bb_data.terminator());
248             }
249             TerminatorKind::Drop {
250                 location,
251                 target,
252                 unwind: _,
253             } => {
254                 let ty = location.ty(fx.mir, fx.tcx).to_ty(fx.tcx);
255                 let ty = fx.monomorphize(&ty);
256                 let drop_fn = crate::rustc_mir::monomorphize::resolve_drop_in_place(fx.tcx, ty);
257
258                 if let ty::InstanceDef::DropGlue(_, None) = drop_fn.def {
259                     // we don't actually need to drop anything
260                 } else {
261                     let drop_place = trans_place(fx, location);
262                     let drop_fn_ty = drop_fn.ty(fx.tcx);
263                     match ty.sty {
264                         ty::Dynamic(..) => {
265                             crate::abi::codegen_drop(fx, drop_place, drop_fn_ty);
266                         }
267                         _ => {
268                             let arg_place = CPlace::new_stack_slot(
269                                 fx,
270                                 fx.tcx.mk_ref(
271                                     &ty::RegionKind::ReErased,
272                                     TypeAndMut {
273                                         ty,
274                                         mutbl: crate::rustc::hir::Mutability::MutMutable,
275                                     },
276                                 ),
277                             );
278                             drop_place.write_place_ref(fx, arg_place);
279                             let arg_value = arg_place.to_cvalue(fx);
280                             crate::abi::codegen_call_inner(
281                                 fx,
282                                 None,
283                                 drop_fn_ty,
284                                 vec![arg_value],
285                                 None,
286                             );
287                         }
288                     }
289                 }
290
291                 let target_ebb = fx.get_ebb(*target);
292                 fx.bcx.ins().jump(target_ebb, &[]);
293             }
294         };
295     }
296
297     fx.bcx.seal_all_blocks();
298     fx.bcx.finalize();
299 }
300
301 fn trans_stmt<'a, 'tcx: 'a>(
302     fx: &mut FunctionCx<'a, 'tcx, impl Backend>,
303     cur_ebb: Ebb,
304     stmt: &Statement<'tcx>,
305 ) {
306     let _print_guard = PrintOnPanic(|| format!("stmt {:?}", stmt));
307
308     fx.set_debug_loc(stmt.source_info);
309
310     #[cfg(debug_assertions)]
311     match &stmt.kind {
312         StatementKind::StorageLive(..) | StatementKind::StorageDead(..) => {} // Those are not very useful
313         _ => {
314             let inst = fx.bcx.func.layout.last_inst(cur_ebb).unwrap();
315             fx.add_comment(inst, format!("{:?}", stmt));
316         }
317     }
318
319     match &stmt.kind {
320         StatementKind::SetDiscriminant {
321             place,
322             variant_index,
323         } => {
324             let place = trans_place(fx, place);
325             let layout = place.layout();
326             if layout.for_variant(&*fx, *variant_index).abi == layout::Abi::Uninhabited {
327                 return;
328             }
329             match layout.variants {
330                 layout::Variants::Single { index } => {
331                     assert_eq!(index, *variant_index);
332                 }
333                 layout::Variants::Tagged { .. } => {
334                     let ptr = place.place_field(fx, mir::Field::new(0));
335                     let to = layout
336                         .ty
337                         .ty_adt_def()
338                         .unwrap()
339                         .discriminant_for_variant(fx.tcx, *variant_index)
340                         .val;
341                     let discr = CValue::const_val(fx, ptr.layout().ty, to as u64 as i64);
342                     ptr.write_cvalue(fx, discr);
343                 }
344                 layout::Variants::NicheFilling {
345                     dataful_variant,
346                     ref niche_variants,
347                     niche_start,
348                     ..
349                 } => {
350                     if *variant_index != dataful_variant {
351                         let niche = place.place_field(fx, mir::Field::new(0));
352                         //let niche_llty = niche.layout.immediate_llvm_type(bx.cx);
353                         let niche_value =
354                             ((variant_index.as_u32() - niche_variants.start().as_u32()) as u128)
355                                 .wrapping_add(niche_start);
356                         // FIXME(eddyb) Check the actual primitive type here.
357                         let niche_llval = if niche_value == 0 {
358                             CValue::const_val(fx, niche.layout().ty, 0)
359                         } else {
360                             CValue::const_val(fx, niche.layout().ty, niche_value as u64 as i64)
361                         };
362                         niche.write_cvalue(fx, niche_llval);
363                     }
364                 }
365             }
366         }
367         StatementKind::Assign(to_place, rval) => {
368             let lval = trans_place(fx, to_place);
369             let dest_layout = lval.layout();
370             match &**rval {
371                 Rvalue::Use(operand) => {
372                     let val = trans_operand(fx, operand);
373                     lval.write_cvalue(fx, val);
374                 }
375                 Rvalue::Ref(_, _, place) => {
376                     let place = trans_place(fx, place);
377                     place.write_place_ref(fx, lval);
378                 }
379                 Rvalue::BinaryOp(bin_op, lhs, rhs) => {
380                     let ty = fx.monomorphize(&lhs.ty(fx.mir, fx.tcx));
381                     let lhs = trans_operand(fx, lhs);
382                     let rhs = trans_operand(fx, rhs);
383
384                     let res = match ty.sty {
385                         ty::Bool => trans_bool_binop(fx, *bin_op, lhs, rhs, lval.layout().ty),
386                         ty::Uint(_) => {
387                             trans_int_binop(fx, *bin_op, lhs, rhs, lval.layout().ty, false)
388                         }
389                         ty::Int(_) => {
390                             trans_int_binop(fx, *bin_op, lhs, rhs, lval.layout().ty, true)
391                         }
392                         ty::Float(_) => trans_float_binop(fx, *bin_op, lhs, rhs, lval.layout().ty),
393                         ty::Char => trans_char_binop(fx, *bin_op, lhs, rhs, lval.layout().ty),
394                         ty::RawPtr(..) => trans_ptr_binop(fx, *bin_op, lhs, rhs, lval.layout().ty),
395                         ty::FnPtr(..) => trans_ptr_binop(fx, *bin_op, lhs, rhs, lval.layout().ty),
396                         _ => unimplemented!("binop {:?} for {:?}", bin_op, ty),
397                     };
398                     lval.write_cvalue(fx, res);
399                 }
400                 Rvalue::CheckedBinaryOp(bin_op, lhs, rhs) => {
401                     let ty = fx.monomorphize(&lhs.ty(fx.mir, fx.tcx));
402                     let lhs = trans_operand(fx, lhs);
403                     let rhs = trans_operand(fx, rhs);
404
405                     let res = match ty.sty {
406                         ty::Uint(_) => {
407                             trans_checked_int_binop(fx, *bin_op, lhs, rhs, lval.layout().ty, false)
408                         }
409                         ty::Int(_) => {
410                             trans_checked_int_binop(fx, *bin_op, lhs, rhs, lval.layout().ty, true)
411                         }
412                         _ => unimplemented!("checked binop {:?} for {:?}", bin_op, ty),
413                     };
414                     lval.write_cvalue(fx, res);
415                 }
416                 Rvalue::UnaryOp(un_op, operand) => {
417                     let operand = trans_operand(fx, operand);
418                     let layout = operand.layout();
419                     let val = operand.load_scalar(fx);
420                     let res = match un_op {
421                         UnOp::Not => {
422                             match layout.ty.sty {
423                                 ty::Bool => {
424                                     let val = fx.bcx.ins().uextend(types::I32, val); // WORKAROUND for CraneStation/cranelift#466
425                                     let res = fx.bcx.ins().icmp_imm(IntCC::Equal, val, 0);
426                                     fx.bcx.ins().bint(types::I8, res)
427                                 }
428                                 ty::Uint(_) | ty::Int(_) => fx.bcx.ins().bnot(val),
429                                 _ => unimplemented!("un op Not for {:?}", layout.ty),
430                             }
431                         }
432                         UnOp::Neg => match layout.ty.sty {
433                             ty::Int(_) => {
434                                 let clif_ty = fx.clif_type(layout.ty).unwrap();
435                                 let zero = fx.bcx.ins().iconst(clif_ty, 0);
436                                 fx.bcx.ins().isub(zero, val)
437                             }
438                             ty::Float(_) => fx.bcx.ins().fneg(val),
439                             _ => unimplemented!("un op Neg for {:?}", layout.ty),
440                         },
441                     };
442                     lval.write_cvalue(fx, CValue::ByVal(res, layout));
443                 }
444                 Rvalue::Cast(CastKind::ReifyFnPointer, operand, ty) => {
445                     let layout = fx.layout_of(ty);
446                     match fx.monomorphize(&operand.ty(&fx.mir.local_decls, fx.tcx)).sty {
447                         ty::FnDef(def_id, substs) => {
448                             let func_ref = fx.get_function_ref(
449                                 Instance::resolve(fx.tcx, ParamEnv::reveal_all(), def_id, substs).unwrap(),
450                             );
451                             let func_addr = fx.bcx.ins().func_addr(fx.pointer_type, func_ref);
452                             lval.write_cvalue(fx, CValue::ByVal(func_addr, layout));
453                         }
454                         _ => bug!("Trying to ReifyFnPointer on non FnDef {:?}", ty),
455                     }
456                 }
457                 Rvalue::Cast(CastKind::UnsafeFnPointer, operand, ty) => {
458                     let operand = trans_operand(fx, operand);
459                     let layout = fx.layout_of(ty);
460                     lval.write_cvalue(fx, operand.unchecked_cast_to(layout));
461                 }
462                 Rvalue::Cast(CastKind::Misc, operand, to_ty) => {
463                     let operand = trans_operand(fx, operand);
464                     let from_ty = operand.layout().ty;
465                     match (&from_ty.sty, &to_ty.sty) {
466                         (ty::Ref(..), ty::Ref(..))
467                         | (ty::Ref(..), ty::RawPtr(..))
468                         | (ty::RawPtr(..), ty::Ref(..))
469                         | (ty::RawPtr(..), ty::RawPtr(..))
470                         | (ty::FnPtr(..), ty::RawPtr(..)) => {
471                             lval.write_cvalue(fx, operand.unchecked_cast_to(dest_layout));
472                         }
473                         (ty::RawPtr(..), ty::Uint(_))
474                         | (ty::RawPtr(..), ty::Int(_))
475                         | (ty::FnPtr(..), ty::Uint(_))
476                             if to_ty.sty == fx.tcx.types.usize.sty
477                                 || to_ty.sty == fx.tcx.types.isize.sty
478                                 || fx.clif_type(to_ty).unwrap() == pointer_ty(fx.tcx) =>
479                         {
480                             lval.write_cvalue(fx, operand.unchecked_cast_to(dest_layout));
481                         }
482                         (ty::Uint(_), ty::RawPtr(..)) if from_ty.sty == fx.tcx.types.usize.sty => {
483                             lval.write_cvalue(fx, operand.unchecked_cast_to(dest_layout));
484                         }
485                         (ty::Int(_), ty::RawPtr(..)) if from_ty.sty == fx.tcx.types.isize.sty => {
486                             lval.write_cvalue(fx, operand.unchecked_cast_to(dest_layout));
487                         }
488                         (ty::Char, ty::Uint(_))
489                         | (ty::Uint(_), ty::Char)
490                         | (ty::Uint(_), ty::Int(_))
491                         | (ty::Uint(_), ty::Uint(_)) => {
492                             let from = operand.load_scalar(fx);
493                             let res = crate::common::clif_intcast(
494                                 fx,
495                                 from,
496                                 fx.clif_type(to_ty).unwrap(),
497                                 false,
498                             );
499                             lval.write_cvalue(fx, CValue::ByVal(res, dest_layout));
500                         }
501                         (ty::Int(_), ty::Int(_)) | (ty::Int(_), ty::Uint(_)) => {
502                             let from = operand.load_scalar(fx);
503                             let res = crate::common::clif_intcast(
504                                 fx,
505                                 from,
506                                 fx.clif_type(to_ty).unwrap(),
507                                 true,
508                             );
509                             lval.write_cvalue(fx, CValue::ByVal(res, dest_layout));
510                         }
511                         (ty::Float(from_flt), ty::Float(to_flt)) => {
512                             let from = operand.load_scalar(fx);
513                             let res = match (from_flt, to_flt) {
514                                 (FloatTy::F32, FloatTy::F64) => {
515                                     fx.bcx.ins().fpromote(types::F64, from)
516                                 }
517                                 (FloatTy::F64, FloatTy::F32) => {
518                                     fx.bcx.ins().fdemote(types::F32, from)
519                                 }
520                                 _ => from,
521                             };
522                             lval.write_cvalue(fx, CValue::ByVal(res, dest_layout));
523                         }
524                         (ty::Float(_), ty::Int(_)) => {
525                             let from = operand.load_scalar(fx);
526                             let i_type = fx.clif_type(to_ty).unwrap();
527                             let res = fx.bcx.ins().fcvt_to_sint_sat(i_type, from);
528                             lval.write_cvalue(fx, CValue::ByVal(res, dest_layout));
529                         }
530                         (ty::Float(_), ty::Uint(_)) => {
531                             let from = operand.load_scalar(fx);
532                             let i_type = fx.clif_type(to_ty).unwrap();
533                             let res = fx.bcx.ins().fcvt_to_uint_sat(i_type, from);
534                             lval.write_cvalue(fx, CValue::ByVal(res, dest_layout));
535                         }
536                         (ty::Int(_), ty::Float(_)) => {
537                             let from_ty = fx.clif_type(from_ty).unwrap();
538                             let from = operand.load_scalar(fx);
539                             // FIXME missing encoding for fcvt_from_sint.f32.i8
540                             let from = if from_ty == types::I8 || from_ty == types::I16 {
541                                 fx.bcx.ins().sextend(types::I32, from)
542                             } else {
543                                 from
544                             };
545                             let f_type = fx.clif_type(to_ty).unwrap();
546                             let res = fx.bcx.ins().fcvt_from_sint(f_type, from);
547                             lval.write_cvalue(fx, CValue::ByVal(res, dest_layout));
548                         }
549                         (ty::Uint(_), ty::Float(_)) => {
550                             let from_ty = fx.clif_type(from_ty).unwrap();
551                             let from = operand.load_scalar(fx);
552                             // FIXME missing encoding for fcvt_from_uint.f32.i8
553                             let from = if from_ty == types::I8 || from_ty == types::I16 {
554                                 fx.bcx.ins().uextend(types::I32, from)
555                             } else {
556                                 from
557                             };
558                             let f_type = fx.clif_type(to_ty).unwrap();
559                             let res = fx.bcx.ins().fcvt_from_uint(f_type, from);
560                             lval.write_cvalue(fx, CValue::ByVal(res, dest_layout));
561                         }
562                         (ty::Bool, ty::Uint(_)) | (ty::Bool, ty::Int(_)) => {
563                             let to_ty = fx.clif_type(to_ty).unwrap();
564                             let from = operand.load_scalar(fx);
565                             let res = if to_ty != types::I8 {
566                                 fx.bcx.ins().uextend(to_ty, from)
567                             } else {
568                                 from
569                             };
570                             lval.write_cvalue(fx, CValue::ByVal(res, dest_layout));
571                         }
572                         (ty::Adt(adt_def, _substs), ty::Uint(_)) | (ty::Adt(adt_def, _substs), ty::Int(_)) if adt_def.is_enum() => {
573                             // FIXME avoid forcing to stack
574                             let place = CPlace::Addr(operand.force_stack(fx), None, operand.layout());
575                             let discr = trans_get_discriminant(fx, place, fx.layout_of(to_ty));
576                             lval.write_cvalue(fx, discr);
577                         }
578                         _ => unimpl!("rval misc {:?} {:?}", from_ty, to_ty),
579                     }
580                 }
581                 Rvalue::Cast(CastKind::ClosureFnPointer, operand, ty) => {
582                     unimplemented!("rval closure_fn_ptr {:?} {:?}", operand, ty)
583                 }
584                 Rvalue::Cast(CastKind::Unsize, operand, _ty) => {
585                     let operand = trans_operand(fx, operand);
586                     operand.unsize_value(fx, lval);
587                 }
588                 Rvalue::Discriminant(place) => {
589                     let place = trans_place(fx, place);
590                     let discr = trans_get_discriminant(fx, place, dest_layout);
591                     lval.write_cvalue(fx, discr);
592                 }
593                 Rvalue::Repeat(operand, times) => {
594                     let operand = trans_operand(fx, operand);
595                     for i in 0..*times {
596                         let index = fx.bcx.ins().iconst(fx.pointer_type, i as i64);
597                         let to = lval.place_index(fx, index);
598                         to.write_cvalue(fx, operand);
599                     }
600                 }
601                 Rvalue::Len(place) => {
602                     let place = trans_place(fx, place);
603                     let usize_layout = fx.layout_of(fx.tcx.types.usize);
604                     let len = codegen_array_len(fx, place);
605                     lval.write_cvalue(fx, CValue::ByVal(len, usize_layout));
606                 }
607                 Rvalue::NullaryOp(NullOp::Box, content_ty) => {
608                     use rustc::middle::lang_items::ExchangeMallocFnLangItem;
609
610                     let usize_type = fx.clif_type(fx.tcx.types.usize).unwrap();
611                     let layout = fx.layout_of(content_ty);
612                     let llsize = fx.bcx.ins().iconst(usize_type, layout.size.bytes() as i64);
613                     let llalign = fx
614                         .bcx
615                         .ins()
616                         .iconst(usize_type, layout.align.abi.bytes() as i64);
617                     let box_layout = fx.layout_of(fx.tcx.mk_box(content_ty));
618
619                     // Allocate space:
620                     let def_id = match fx.tcx.lang_items().require(ExchangeMallocFnLangItem) {
621                         Ok(id) => id,
622                         Err(s) => {
623                             fx.tcx
624                                 .sess
625                                 .fatal(&format!("allocation of `{}` {}", box_layout.ty, s));
626                         }
627                     };
628                     let instance = ty::Instance::mono(fx.tcx, def_id);
629                     let func_ref = fx.get_function_ref(instance);
630                     let call = fx.bcx.ins().call(func_ref, &[llsize, llalign]);
631                     let ptr = fx.bcx.inst_results(call)[0];
632                     lval.write_cvalue(fx, CValue::ByVal(ptr, box_layout));
633                 }
634                 Rvalue::NullaryOp(NullOp::SizeOf, ty) => {
635                     assert!(lval
636                         .layout()
637                         .ty
638                         .is_sized(fx.tcx.at(DUMMY_SP), ParamEnv::reveal_all()));
639                     let ty_size = fx.layout_of(ty).size.bytes();
640                     let val = CValue::const_val(fx, fx.tcx.types.usize, ty_size as i64);
641                     lval.write_cvalue(fx, val);
642                 }
643                 Rvalue::Aggregate(kind, operands) => match **kind {
644                     AggregateKind::Array(_ty) => {
645                         for (i, operand) in operands.into_iter().enumerate() {
646                             let operand = trans_operand(fx, operand);
647                             let index = fx.bcx.ins().iconst(fx.pointer_type, i as i64);
648                             let to = lval.place_index(fx, index);
649                             to.write_cvalue(fx, operand);
650                         }
651                     }
652                     _ => unimpl!("shouldn't exist at trans {:?}", rval),
653                 },
654             }
655         }
656         StatementKind::StorageLive(_)
657         | StatementKind::StorageDead(_)
658         | StatementKind::Nop
659         | StatementKind::FakeRead(..)
660         | StatementKind::Retag { .. }
661         | StatementKind::AscribeUserType(..) => {}
662
663         StatementKind::InlineAsm { .. } => unimpl!("Inline assembly is not supported"),
664     }
665 }
666
667 fn codegen_array_len<'a, 'tcx: 'a>(
668     fx: &mut FunctionCx<'a, 'tcx, impl Backend>,
669     place: CPlace<'tcx>,
670 ) -> Value {
671     match place.layout().ty.sty {
672         ty::Array(_elem_ty, len) => {
673             let len = crate::constant::force_eval_const(fx, len).unwrap_usize(fx.tcx) as i64;
674             fx.bcx.ins().iconst(fx.pointer_type, len)
675         }
676         ty::Slice(_elem_ty) => {
677             place.to_addr_maybe_unsized(fx).1.expect("Length metadata for slice place")
678         }
679         _ => bug!("Rvalue::Len({:?})", place),
680     }
681 }
682
683 pub fn trans_get_discriminant<'a, 'tcx: 'a>(
684     fx: &mut FunctionCx<'a, 'tcx, impl Backend>,
685     place: CPlace<'tcx>,
686     dest_layout: TyLayout<'tcx>,
687 ) -> CValue<'tcx> {
688     let layout = place.layout();
689
690     if layout.abi == layout::Abi::Uninhabited {
691         trap_unreachable(&mut fx.bcx);
692     }
693     match layout.variants {
694         layout::Variants::Single { index } => {
695             let discr_val = layout
696                 .ty
697                 .ty_adt_def()
698                 .map_or(index.as_u32() as u128, |def| {
699                     def.discriminant_for_variant(fx.tcx, index).val
700                 });
701             return CValue::const_val(fx, dest_layout.ty, discr_val as u64 as i64);
702         }
703         layout::Variants::Tagged { .. } | layout::Variants::NicheFilling { .. } => {}
704     }
705
706     let discr = place.place_field(fx, mir::Field::new(0)).to_cvalue(fx);
707     let discr_ty = discr.layout().ty;
708     let lldiscr = discr.load_scalar(fx);
709     match layout.variants {
710         layout::Variants::Single { .. } => bug!(),
711         layout::Variants::Tagged { ref tag, .. } => {
712             let signed = match tag.value {
713                 layout::Int(_, signed) => signed,
714                 _ => false,
715             };
716             let val = clif_intcast(fx, lldiscr, fx.clif_type(dest_layout.ty).unwrap(), signed);
717             return CValue::ByVal(val, dest_layout);
718         }
719         layout::Variants::NicheFilling {
720             dataful_variant,
721             ref niche_variants,
722             niche_start,
723             ..
724         } => {
725             let niche_llty = fx.clif_type(discr_ty).unwrap();
726             let dest_clif_ty = fx.clif_type(dest_layout.ty).unwrap();
727             if niche_variants.start() == niche_variants.end() {
728                 let b = fx
729                     .bcx
730                     .ins()
731                     .icmp_imm(IntCC::Equal, lldiscr, niche_start as u64 as i64);
732                 let if_true = fx
733                     .bcx
734                     .ins()
735                     .iconst(dest_clif_ty, niche_variants.start().as_u32() as i64);
736                 let if_false = fx
737                     .bcx
738                     .ins()
739                     .iconst(dest_clif_ty, dataful_variant.as_u32() as i64);
740                 let val = fx.bcx.ins().select(b, if_true, if_false);
741                 return CValue::ByVal(val, dest_layout);
742             } else {
743                 // Rebase from niche values to discriminant values.
744                 let delta = niche_start.wrapping_sub(niche_variants.start().as_u32() as u128);
745                 let delta = fx.bcx.ins().iconst(niche_llty, delta as u64 as i64);
746                 let lldiscr = fx.bcx.ins().isub(lldiscr, delta);
747                 let b = fx.bcx.ins().icmp_imm(
748                     IntCC::UnsignedLessThanOrEqual,
749                     lldiscr,
750                     niche_variants.end().as_u32() as i64,
751                 );
752                 let if_true =
753                     clif_intcast(fx, lldiscr, fx.clif_type(dest_layout.ty).unwrap(), false);
754                 let if_false = fx
755                     .bcx
756                     .ins()
757                     .iconst(dest_clif_ty, dataful_variant.as_u32() as i64);
758                 let val = fx.bcx.ins().select(b, if_true, if_false);
759                 return CValue::ByVal(val, dest_layout);
760             }
761         }
762     }
763 }
764
765 macro_rules! binop_match {
766     (@single $fx:expr, $bug_fmt:expr, $var:expr, $signed:expr, $lhs:expr, $rhs:expr, $ret_ty:expr, bug) => {
767         bug!("binop {} on {} lhs: {:?} rhs: {:?}", stringify!($var), $bug_fmt, $lhs, $rhs)
768     };
769     (@single $fx:expr, $bug_fmt:expr, $var:expr, $signed:expr, $lhs:expr, $rhs:expr, $ret_ty:expr, icmp($cc:ident)) => {{
770         assert_eq!($fx.tcx.types.bool, $ret_ty);
771         let ret_layout = $fx.layout_of($ret_ty);
772
773         let b = $fx.bcx.ins().icmp(IntCC::$cc, $lhs, $rhs);
774         CValue::ByVal($fx.bcx.ins().bint(types::I8, b), ret_layout)
775     }};
776     (@single $fx:expr, $bug_fmt:expr, $var:expr, $signed:expr, $lhs:expr, $rhs:expr, $ret_ty:expr, fcmp($cc:ident)) => {{
777         assert_eq!($fx.tcx.types.bool, $ret_ty);
778         let ret_layout = $fx.layout_of($ret_ty);
779         let b = $fx.bcx.ins().fcmp(FloatCC::$cc, $lhs, $rhs);
780         CValue::ByVal($fx.bcx.ins().bint(types::I8, b), ret_layout)
781     }};
782     (@single $fx:expr, $bug_fmt:expr, $var:expr, $signed:expr, $lhs:expr, $rhs:expr, $ret_ty:expr, custom(|| $body:expr)) => {{
783         $body
784     }};
785     (@single $fx:expr, $bug_fmt:expr, $var:expr, $signed:expr, $lhs:expr, $rhs:expr, $ret_ty:expr, $name:ident) => {{
786         let ret_layout = $fx.layout_of($ret_ty);
787         CValue::ByVal($fx.bcx.ins().$name($lhs, $rhs), ret_layout)
788     }};
789     (
790         $fx:expr, $bin_op:expr, $signed:expr, $lhs:expr, $rhs:expr, $ret_ty:expr, $bug_fmt:expr;
791         $(
792             $var:ident ($sign:pat) $name:tt $( ( $($next:tt)* ) )? ;
793         )*
794     ) => {{
795         let lhs = $lhs.load_scalar($fx);
796         let rhs = $rhs.load_scalar($fx);
797         match ($bin_op, $signed) {
798             $(
799                 (BinOp::$var, $sign) => binop_match!(@single $fx, $bug_fmt, $var, $signed, lhs, rhs, $ret_ty, $name $( ( $($next)* ) )?),
800             )*
801         }
802     }}
803 }
804
805 fn trans_bool_binop<'a, 'tcx: 'a>(
806     fx: &mut FunctionCx<'a, 'tcx, impl Backend>,
807     bin_op: BinOp,
808     lhs: CValue<'tcx>,
809     rhs: CValue<'tcx>,
810     ty: Ty<'tcx>,
811 ) -> CValue<'tcx> {
812     let res = binop_match! {
813         fx, bin_op, false, lhs, rhs, ty, "bool";
814         Add (_) bug;
815         Sub (_) bug;
816         Mul (_) bug;
817         Div (_) bug;
818         Rem (_) bug;
819         BitXor (_) bxor;
820         BitAnd (_) band;
821         BitOr (_) bor;
822         Shl (_) bug;
823         Shr (_) bug;
824
825         Eq (_) icmp(Equal);
826         Lt (_) icmp(UnsignedLessThan);
827         Le (_) icmp(UnsignedLessThanOrEqual);
828         Ne (_) icmp(NotEqual);
829         Ge (_) icmp(UnsignedGreaterThanOrEqual);
830         Gt (_) icmp(UnsignedGreaterThan);
831
832         Offset (_) bug;
833     };
834
835     res
836 }
837
838 pub fn trans_int_binop<'a, 'tcx: 'a>(
839     fx: &mut FunctionCx<'a, 'tcx, impl Backend>,
840     bin_op: BinOp,
841     lhs: CValue<'tcx>,
842     rhs: CValue<'tcx>,
843     out_ty: Ty<'tcx>,
844     signed: bool,
845 ) -> CValue<'tcx> {
846     if bin_op != BinOp::Shl && bin_op != BinOp::Shr {
847         assert_eq!(
848             lhs.layout().ty,
849             rhs.layout().ty,
850             "int binop requires lhs and rhs of same type"
851         );
852     }
853     binop_match! {
854         fx, bin_op, signed, lhs, rhs, out_ty, "int/uint";
855         Add (_) iadd;
856         Sub (_) isub;
857         Mul (_) imul;
858         Div (false) udiv;
859         Div (true) sdiv;
860         Rem (false) urem;
861         Rem (true) srem;
862         BitXor (_) bxor;
863         BitAnd (_) band;
864         BitOr (_) bor;
865         Shl (_) ishl;
866         Shr (false) ushr;
867         Shr (true) sshr;
868
869         Eq (_) icmp(Equal);
870         Lt (false) icmp(UnsignedLessThan);
871         Lt (true) icmp(SignedLessThan);
872         Le (false) icmp(UnsignedLessThanOrEqual);
873         Le (true) icmp(SignedLessThanOrEqual);
874         Ne (_) icmp(NotEqual);
875         Ge (false) icmp(UnsignedGreaterThanOrEqual);
876         Ge (true) icmp(SignedGreaterThanOrEqual);
877         Gt (false) icmp(UnsignedGreaterThan);
878         Gt (true) icmp(SignedGreaterThan);
879
880         Offset (_) bug;
881     }
882 }
883
884 pub fn trans_checked_int_binop<'a, 'tcx: 'a>(
885     fx: &mut FunctionCx<'a, 'tcx, impl Backend>,
886     bin_op: BinOp,
887     in_lhs: CValue<'tcx>,
888     in_rhs: CValue<'tcx>,
889     out_ty: Ty<'tcx>,
890     signed: bool,
891 ) -> CValue<'tcx> {
892     if bin_op != BinOp::Shl && bin_op != BinOp::Shr {
893         assert_eq!(
894             in_lhs.layout().ty,
895             in_rhs.layout().ty,
896             "checked int binop requires lhs and rhs of same type"
897         );
898     }
899
900     let lhs = in_lhs.load_scalar(fx);
901     let rhs = in_rhs.load_scalar(fx);
902     let res = match bin_op {
903         BinOp::Add => fx.bcx.ins().iadd(lhs, rhs),
904         BinOp::Sub => fx.bcx.ins().isub(lhs, rhs),
905         BinOp::Mul => fx.bcx.ins().imul(lhs, rhs),
906         BinOp::Shl => fx.bcx.ins().ishl(lhs, rhs),
907         BinOp::Shr => {
908             if !signed {
909                 fx.bcx.ins().ushr(lhs, rhs)
910             } else {
911                 fx.bcx.ins().sshr(lhs, rhs)
912             }
913         }
914         _ => bug!(
915             "binop {:?} on checked int/uint lhs: {:?} rhs: {:?}",
916             bin_op,
917             in_lhs,
918             in_rhs
919         ),
920     };
921
922     // TODO: check for overflow
923     let has_overflow = fx.bcx.ins().iconst(types::I8, 0);
924
925     let out_place = CPlace::new_stack_slot(fx, out_ty);
926     let out_layout = out_place.layout();
927     out_place.write_cvalue(fx, CValue::ByValPair(res, has_overflow, out_layout));
928
929     out_place.to_cvalue(fx)
930 }
931
932 fn trans_float_binop<'a, 'tcx: 'a>(
933     fx: &mut FunctionCx<'a, 'tcx, impl Backend>,
934     bin_op: BinOp,
935     lhs: CValue<'tcx>,
936     rhs: CValue<'tcx>,
937     ty: Ty<'tcx>,
938 ) -> CValue<'tcx> {
939     let res = binop_match! {
940         fx, bin_op, false, lhs, rhs, ty, "float";
941         Add (_) fadd;
942         Sub (_) fsub;
943         Mul (_) fmul;
944         Div (_) fdiv;
945         Rem (_) custom(|| {
946             assert_eq!(lhs.layout().ty, ty);
947             assert_eq!(rhs.layout().ty, ty);
948             match ty.sty {
949                 ty::Float(FloatTy::F32) => fx.easy_call("fmodf", &[lhs, rhs], ty),
950                 ty::Float(FloatTy::F64) => fx.easy_call("fmod", &[lhs, rhs], ty),
951                 _ => bug!(),
952             }
953         });
954         BitXor (_) bxor;
955         BitAnd (_) band;
956         BitOr (_) bor;
957         Shl (_) bug;
958         Shr (_) bug;
959
960         Eq (_) fcmp(Equal);
961         Lt (_) fcmp(LessThan);
962         Le (_) fcmp(LessThanOrEqual);
963         Ne (_) fcmp(NotEqual);
964         Ge (_) fcmp(GreaterThanOrEqual);
965         Gt (_) fcmp(GreaterThan);
966
967         Offset (_) bug;
968     };
969
970     res
971 }
972
973 fn trans_char_binop<'a, 'tcx: 'a>(
974     fx: &mut FunctionCx<'a, 'tcx, impl Backend>,
975     bin_op: BinOp,
976     lhs: CValue<'tcx>,
977     rhs: CValue<'tcx>,
978     ty: Ty<'tcx>,
979 ) -> CValue<'tcx> {
980     let res = binop_match! {
981         fx, bin_op, false, lhs, rhs, ty, "char";
982         Add (_) bug;
983         Sub (_) bug;
984         Mul (_) bug;
985         Div (_) bug;
986         Rem (_) bug;
987         BitXor (_) bug;
988         BitAnd (_) bug;
989         BitOr (_) bug;
990         Shl (_) bug;
991         Shr (_) bug;
992
993         Eq (_) icmp(Equal);
994         Lt (_) icmp(UnsignedLessThan);
995         Le (_) icmp(UnsignedLessThanOrEqual);
996         Ne (_) icmp(NotEqual);
997         Ge (_) icmp(UnsignedGreaterThanOrEqual);
998         Gt (_) icmp(UnsignedGreaterThan);
999
1000         Offset (_) bug;
1001     };
1002
1003     res
1004 }
1005
1006 fn trans_ptr_binop<'a, 'tcx: 'a>(
1007     fx: &mut FunctionCx<'a, 'tcx, impl Backend>,
1008     bin_op: BinOp,
1009     lhs: CValue<'tcx>,
1010     rhs: CValue<'tcx>,
1011     ret_ty: Ty<'tcx>,
1012 ) -> CValue<'tcx> {
1013     let not_fat = match lhs.layout().ty.sty {
1014         ty::RawPtr(TypeAndMut { ty, mutbl: _ }) => ty.is_sized(fx.tcx.at(DUMMY_SP), ParamEnv::reveal_all()),
1015         ty::FnPtr(..) => true,
1016         _ => bug!("trans_ptr_binop on non ptr"),
1017     };
1018     if not_fat {
1019         if let BinOp::Offset = bin_op {
1020             let (base, offset) = (lhs, rhs.load_scalar(fx));
1021             let pointee_ty = base.layout().ty.builtin_deref(true).unwrap().ty;
1022             let pointee_size = fx.layout_of(pointee_ty).size.bytes();
1023             let ptr_diff = fx.bcx.ins().imul_imm(offset, pointee_size as i64);
1024             let base_val = base.load_scalar(fx);
1025             let res = fx.bcx.ins().iadd(base_val, ptr_diff);
1026             return CValue::ByVal(res, base.layout());
1027         }
1028
1029         binop_match! {
1030             fx, bin_op, false, lhs, rhs, ret_ty, "ptr";
1031             Add (_) bug;
1032             Sub (_) bug;
1033             Mul (_) bug;
1034             Div (_) bug;
1035             Rem (_) bug;
1036             BitXor (_) bug;
1037             BitAnd (_) bug;
1038             BitOr (_) bug;
1039             Shl (_) bug;
1040             Shr (_) bug;
1041
1042             Eq (_) icmp(Equal);
1043             Lt (_) icmp(UnsignedLessThan);
1044             Le (_) icmp(UnsignedLessThanOrEqual);
1045             Ne (_) icmp(NotEqual);
1046             Ge (_) icmp(UnsignedGreaterThanOrEqual);
1047             Gt (_) icmp(UnsignedGreaterThan);
1048
1049             Offset (_) bug; // Handled above
1050         }
1051     } else {
1052         let (lhs_ptr, lhs_extra) = lhs.load_value_pair(fx);
1053         let (rhs_ptr, rhs_extra) = rhs.load_value_pair(fx);
1054         let res = match bin_op {
1055             BinOp::Eq => {
1056                 let ptr_eq = fx.bcx.ins().icmp(IntCC::Equal, lhs_ptr, rhs_ptr);
1057                 let extra_eq = fx.bcx.ins().icmp(IntCC::Equal, lhs_extra, rhs_extra);
1058                 fx.bcx.ins().band(ptr_eq, extra_eq)
1059             }
1060             BinOp::Ne => {
1061                 let ptr_ne = fx.bcx.ins().icmp(IntCC::NotEqual, lhs_ptr, rhs_ptr);
1062                 let extra_ne = fx.bcx.ins().icmp(IntCC::NotEqual, lhs_extra, rhs_extra);
1063                 fx.bcx.ins().bor(ptr_ne, extra_ne)
1064             }
1065             _ => unimplemented!(
1066                 "trans_ptr_binop({:?}, <fat ptr>, <fat ptr>) not implemented",
1067                 bin_op
1068             ),
1069         };
1070
1071         assert_eq!(fx.tcx.types.bool, ret_ty);
1072         let ret_layout = fx.layout_of(ret_ty);
1073         CValue::ByVal(fx.bcx.ins().bint(types::I8, res), ret_layout)
1074     }
1075 }
1076
1077 pub fn trans_place<'a, 'tcx: 'a>(
1078     fx: &mut FunctionCx<'a, 'tcx, impl Backend>,
1079     place: &Place<'tcx>,
1080 ) -> CPlace<'tcx> {
1081     match place {
1082         Place::Local(local) => fx.get_local_place(*local),
1083         Place::Promoted(promoted) => crate::constant::trans_promoted(fx, promoted.0),
1084         Place::Static(static_) => crate::constant::codegen_static_ref(fx, static_),
1085         Place::Projection(projection) => {
1086             let base = trans_place(fx, &projection.base);
1087             match projection.elem {
1088                 ProjectionElem::Deref => base.place_deref(fx),
1089                 ProjectionElem::Field(field, _ty) => base.place_field(fx, field),
1090                 ProjectionElem::Index(local) => {
1091                     let index = fx.get_local_place(local).to_cvalue(fx).load_scalar(fx);
1092                     base.place_index(fx, index)
1093                 }
1094                 ProjectionElem::ConstantIndex {
1095                     offset,
1096                     min_length: _,
1097                     from_end,
1098                 } => {
1099                     let index = if !from_end {
1100                         fx.bcx.ins().iconst(fx.pointer_type, offset as i64)
1101                     } else {
1102                         let len = codegen_array_len(fx, base);
1103                         fx.bcx.ins().iadd_imm(len, -(offset as i64))
1104                     };
1105                     base.place_index(fx, index)
1106                 }
1107                 ProjectionElem::Subslice { from, to } => unimpl!(
1108                     "projection subslice {:?} from {} to {}",
1109                     projection.base,
1110                     from,
1111                     to
1112                 ),
1113                 ProjectionElem::Downcast(_adt_def, variant) => base.downcast_variant(fx, variant),
1114             }
1115         }
1116     }
1117 }
1118
1119 pub fn trans_operand<'a, 'tcx>(
1120     fx: &mut FunctionCx<'a, 'tcx, impl Backend>,
1121     operand: &Operand<'tcx>,
1122 ) -> CValue<'tcx> {
1123     match operand {
1124         Operand::Move(place) | Operand::Copy(place) => {
1125             let cplace = trans_place(fx, place);
1126             cplace.to_cvalue(fx)
1127         }
1128         Operand::Constant(const_) => crate::constant::trans_constant(fx, const_),
1129     }
1130 }