]> git.lizzy.rs Git - rust.git/blob - compiler/rustc_codegen_ssa/src/mir/block.rs
2ca7e8fd7e8e377e05a8a89a52a6c276137625ed
[rust.git] / compiler / rustc_codegen_ssa / src / mir / block.rs
1 use super::operand::OperandRef;
2 use super::operand::OperandValue::{Immediate, Pair, Ref};
3 use super::place::PlaceRef;
4 use super::{FunctionCx, LocalRef};
5
6 use crate::base;
7 use crate::common::{self, IntPredicate};
8 use crate::meth;
9 use crate::traits::*;
10 use crate::MemFlags;
11
12 use rustc_ast as ast;
13 use rustc_hir::lang_items::LangItem;
14 use rustc_index::vec::Idx;
15 use rustc_middle::mir::AssertKind;
16 use rustc_middle::mir::{self, SwitchTargets};
17 use rustc_middle::ty::layout::{FnAbiExt, HasTyCtxt, LayoutOf};
18 use rustc_middle::ty::print::with_no_trimmed_paths;
19 use rustc_middle::ty::{self, Instance, Ty, TypeFoldable};
20 use rustc_span::source_map::Span;
21 use rustc_span::{sym, Symbol};
22 use rustc_target::abi::call::{ArgAbi, FnAbi, PassMode};
23 use rustc_target::abi::{self, HasDataLayout};
24 use rustc_target::spec::abi::Abi;
25
26 /// Used by `FunctionCx::codegen_terminator` for emitting common patterns
27 /// e.g., creating a basic block, calling a function, etc.
28 struct TerminatorCodegenHelper<'tcx> {
29     bb: mir::BasicBlock,
30     terminator: &'tcx mir::Terminator<'tcx>,
31     funclet_bb: Option<mir::BasicBlock>,
32 }
33
34 impl<'a, 'tcx> TerminatorCodegenHelper<'tcx> {
35     /// Returns the appropriate `Funclet` for the current funclet, if on MSVC,
36     /// either already previously cached, or newly created, by `landing_pad_for`.
37     fn funclet<'b, Bx: BuilderMethods<'a, 'tcx>>(
38         &self,
39         fx: &'b mut FunctionCx<'a, 'tcx, Bx>,
40     ) -> Option<&'b Bx::Funclet> {
41         let funclet_bb = self.funclet_bb?;
42         if base::wants_msvc_seh(fx.cx.tcx().sess) {
43             // If `landing_pad_for` hasn't been called yet to create the `Funclet`,
44             // it has to be now. This may not seem necessary, as RPO should lead
45             // to all the unwind edges being visited (and so to `landing_pad_for`
46             // getting called for them), before building any of the blocks inside
47             // the funclet itself - however, if MIR contains edges that end up not
48             // being needed in the LLVM IR after monomorphization, the funclet may
49             // be unreachable, and we don't have yet a way to skip building it in
50             // such an eventuality (which may be a better solution than this).
51             if fx.funclets[funclet_bb].is_none() {
52                 fx.landing_pad_for(funclet_bb);
53             }
54
55             Some(
56                 fx.funclets[funclet_bb]
57                     .as_ref()
58                     .expect("landing_pad_for didn't also create funclets entry"),
59             )
60         } else {
61             None
62         }
63     }
64
65     fn lltarget<Bx: BuilderMethods<'a, 'tcx>>(
66         &self,
67         fx: &mut FunctionCx<'a, 'tcx, Bx>,
68         target: mir::BasicBlock,
69     ) -> (Bx::BasicBlock, bool) {
70         let span = self.terminator.source_info.span;
71         let lltarget = fx.llbb(target);
72         let target_funclet = fx.cleanup_kinds[target].funclet_bb(target);
73         match (self.funclet_bb, target_funclet) {
74             (None, None) => (lltarget, false),
75             (Some(f), Some(t_f)) if f == t_f || !base::wants_msvc_seh(fx.cx.tcx().sess) => {
76                 (lltarget, false)
77             }
78             // jump *into* cleanup - need a landing pad if GNU, cleanup pad if MSVC
79             (None, Some(_)) => (fx.landing_pad_for(target), false),
80             (Some(_), None) => span_bug!(span, "{:?} - jump out of cleanup?", self.terminator),
81             (Some(_), Some(_)) => (fx.landing_pad_for(target), true),
82         }
83     }
84
85     /// Create a basic block.
86     fn llblock<Bx: BuilderMethods<'a, 'tcx>>(
87         &self,
88         fx: &mut FunctionCx<'a, 'tcx, Bx>,
89         target: mir::BasicBlock,
90     ) -> Bx::BasicBlock {
91         let (lltarget, is_cleanupret) = self.lltarget(fx, target);
92         if is_cleanupret {
93             // MSVC cross-funclet jump - need a trampoline
94
95             debug!("llblock: creating cleanup trampoline for {:?}", target);
96             let name = &format!("{:?}_cleanup_trampoline_{:?}", self.bb, target);
97             let mut trampoline = fx.new_block(name);
98             trampoline.cleanup_ret(self.funclet(fx).unwrap(), Some(lltarget));
99             trampoline.llbb()
100         } else {
101             lltarget
102         }
103     }
104
105     fn funclet_br<Bx: BuilderMethods<'a, 'tcx>>(
106         &self,
107         fx: &mut FunctionCx<'a, 'tcx, Bx>,
108         bx: &mut Bx,
109         target: mir::BasicBlock,
110     ) {
111         let (lltarget, is_cleanupret) = self.lltarget(fx, target);
112         if is_cleanupret {
113             // micro-optimization: generate a `ret` rather than a jump
114             // to a trampoline.
115             bx.cleanup_ret(self.funclet(fx).unwrap(), Some(lltarget));
116         } else {
117             bx.br(lltarget);
118         }
119     }
120
121     /// Call `fn_ptr` of `fn_abi` with the arguments `llargs`, the optional
122     /// return destination `destination` and the cleanup function `cleanup`.
123     fn do_call<Bx: BuilderMethods<'a, 'tcx>>(
124         &self,
125         fx: &mut FunctionCx<'a, 'tcx, Bx>,
126         bx: &mut Bx,
127         fn_abi: FnAbi<'tcx, Ty<'tcx>>,
128         fn_ptr: Bx::Value,
129         llargs: &[Bx::Value],
130         destination: Option<(ReturnDest<'tcx, Bx::Value>, mir::BasicBlock)>,
131         cleanup: Option<mir::BasicBlock>,
132     ) {
133         // If there is a cleanup block and the function we're calling can unwind, then
134         // do an invoke, otherwise do a call.
135         let fn_ty = bx.fn_decl_backend_type(&fn_abi);
136         if let Some(cleanup) = cleanup.filter(|_| fn_abi.can_unwind) {
137             let ret_llbb = if let Some((_, target)) = destination {
138                 fx.llbb(target)
139             } else {
140                 fx.unreachable_block()
141             };
142             let invokeret = bx.invoke(
143                 fn_ty,
144                 fn_ptr,
145                 &llargs,
146                 ret_llbb,
147                 self.llblock(fx, cleanup),
148                 self.funclet(fx),
149             );
150             bx.apply_attrs_callsite(&fn_abi, invokeret);
151
152             if let Some((ret_dest, target)) = destination {
153                 let mut ret_bx = fx.build_block(target);
154                 fx.set_debug_loc(&mut ret_bx, self.terminator.source_info);
155                 fx.store_return(&mut ret_bx, ret_dest, &fn_abi.ret, invokeret);
156             }
157         } else {
158             let llret = bx.call(fn_ty, fn_ptr, &llargs, self.funclet(fx));
159             bx.apply_attrs_callsite(&fn_abi, llret);
160             if fx.mir[self.bb].is_cleanup {
161                 // Cleanup is always the cold path. Don't inline
162                 // drop glue. Also, when there is a deeply-nested
163                 // struct, there are "symmetry" issues that cause
164                 // exponential inlining - see issue #41696.
165                 bx.do_not_inline(llret);
166             }
167
168             if let Some((ret_dest, target)) = destination {
169                 fx.store_return(bx, ret_dest, &fn_abi.ret, llret);
170                 self.funclet_br(fx, bx, target);
171             } else {
172                 bx.unreachable();
173             }
174         }
175     }
176 }
177
178 /// Codegen implementations for some terminator variants.
179 impl<'a, 'tcx, Bx: BuilderMethods<'a, 'tcx>> FunctionCx<'a, 'tcx, Bx> {
180     /// Generates code for a `Resume` terminator.
181     fn codegen_resume_terminator(&mut self, helper: TerminatorCodegenHelper<'tcx>, mut bx: Bx) {
182         if let Some(funclet) = helper.funclet(self) {
183             bx.cleanup_ret(funclet, None);
184         } else {
185             let slot = self.get_personality_slot(&mut bx);
186             let lp0 = slot.project_field(&mut bx, 0);
187             let lp0 = bx.load_operand(lp0).immediate();
188             let lp1 = slot.project_field(&mut bx, 1);
189             let lp1 = bx.load_operand(lp1).immediate();
190             slot.storage_dead(&mut bx);
191
192             let mut lp = bx.const_undef(self.landing_pad_type());
193             lp = bx.insert_value(lp, lp0, 0);
194             lp = bx.insert_value(lp, lp1, 1);
195             bx.resume(lp);
196         }
197     }
198
199     fn codegen_switchint_terminator(
200         &mut self,
201         helper: TerminatorCodegenHelper<'tcx>,
202         mut bx: Bx,
203         discr: &mir::Operand<'tcx>,
204         switch_ty: Ty<'tcx>,
205         targets: &SwitchTargets,
206     ) {
207         let discr = self.codegen_operand(&mut bx, &discr);
208         // `switch_ty` is redundant, sanity-check that.
209         assert_eq!(discr.layout.ty, switch_ty);
210         let mut target_iter = targets.iter();
211         if target_iter.len() == 1 {
212             // If there are two targets (one conditional, one fallback), emit br instead of switch
213             let (test_value, target) = target_iter.next().unwrap();
214             let lltrue = helper.llblock(self, target);
215             let llfalse = helper.llblock(self, targets.otherwise());
216             if switch_ty == bx.tcx().types.bool {
217                 // Don't generate trivial icmps when switching on bool
218                 match test_value {
219                     0 => bx.cond_br(discr.immediate(), llfalse, lltrue),
220                     1 => bx.cond_br(discr.immediate(), lltrue, llfalse),
221                     _ => bug!(),
222                 }
223             } else {
224                 let switch_llty = bx.immediate_backend_type(bx.layout_of(switch_ty));
225                 let llval = bx.const_uint_big(switch_llty, test_value);
226                 let cmp = bx.icmp(IntPredicate::IntEQ, discr.immediate(), llval);
227                 bx.cond_br(cmp, lltrue, llfalse);
228             }
229         } else {
230             bx.switch(
231                 discr.immediate(),
232                 helper.llblock(self, targets.otherwise()),
233                 target_iter.map(|(value, target)| (value, helper.llblock(self, target))),
234             );
235         }
236     }
237
238     fn codegen_return_terminator(&mut self, mut bx: Bx) {
239         // Call `va_end` if this is the definition of a C-variadic function.
240         if self.fn_abi.c_variadic {
241             // The `VaList` "spoofed" argument is just after all the real arguments.
242             let va_list_arg_idx = self.fn_abi.args.len();
243             match self.locals[mir::Local::new(1 + va_list_arg_idx)] {
244                 LocalRef::Place(va_list) => {
245                     bx.va_end(va_list.llval);
246                 }
247                 _ => bug!("C-variadic function must have a `VaList` place"),
248             }
249         }
250         if self.fn_abi.ret.layout.abi.is_uninhabited() {
251             // Functions with uninhabited return values are marked `noreturn`,
252             // so we should make sure that we never actually do.
253             // We play it safe by using a well-defined `abort`, but we could go for immediate UB
254             // if that turns out to be helpful.
255             bx.abort();
256             // `abort` does not terminate the block, so we still need to generate
257             // an `unreachable` terminator after it.
258             bx.unreachable();
259             return;
260         }
261         let llval = match self.fn_abi.ret.mode {
262             PassMode::Ignore | PassMode::Indirect { .. } => {
263                 bx.ret_void();
264                 return;
265             }
266
267             PassMode::Direct(_) | PassMode::Pair(..) => {
268                 let op = self.codegen_consume(&mut bx, mir::Place::return_place().as_ref());
269                 if let Ref(llval, _, align) = op.val {
270                     bx.load(bx.backend_type(op.layout), llval, align)
271                 } else {
272                     op.immediate_or_packed_pair(&mut bx)
273                 }
274             }
275
276             PassMode::Cast(cast_ty) => {
277                 let op = match self.locals[mir::RETURN_PLACE] {
278                     LocalRef::Operand(Some(op)) => op,
279                     LocalRef::Operand(None) => bug!("use of return before def"),
280                     LocalRef::Place(cg_place) => OperandRef {
281                         val: Ref(cg_place.llval, None, cg_place.align),
282                         layout: cg_place.layout,
283                     },
284                     LocalRef::UnsizedPlace(_) => bug!("return type must be sized"),
285                 };
286                 let llslot = match op.val {
287                     Immediate(_) | Pair(..) => {
288                         let scratch = PlaceRef::alloca(&mut bx, self.fn_abi.ret.layout);
289                         op.val.store(&mut bx, scratch);
290                         scratch.llval
291                     }
292                     Ref(llval, _, align) => {
293                         assert_eq!(align, op.layout.align.abi, "return place is unaligned!");
294                         llval
295                     }
296                 };
297                 let ty = bx.cast_backend_type(&cast_ty);
298                 let addr = bx.pointercast(llslot, bx.type_ptr_to(ty));
299                 bx.load(ty, addr, self.fn_abi.ret.layout.align.abi)
300             }
301         };
302         bx.ret(llval);
303     }
304
305     fn codegen_drop_terminator(
306         &mut self,
307         helper: TerminatorCodegenHelper<'tcx>,
308         mut bx: Bx,
309         location: mir::Place<'tcx>,
310         target: mir::BasicBlock,
311         unwind: Option<mir::BasicBlock>,
312     ) {
313         let ty = location.ty(self.mir, bx.tcx()).ty;
314         let ty = self.monomorphize(ty);
315         let drop_fn = Instance::resolve_drop_in_place(bx.tcx(), ty);
316
317         if let ty::InstanceDef::DropGlue(_, None) = drop_fn.def {
318             // we don't actually need to drop anything.
319             helper.funclet_br(self, &mut bx, target);
320             return;
321         }
322
323         let place = self.codegen_place(&mut bx, location.as_ref());
324         let (args1, args2);
325         let mut args = if let Some(llextra) = place.llextra {
326             args2 = [place.llval, llextra];
327             &args2[..]
328         } else {
329             args1 = [place.llval];
330             &args1[..]
331         };
332         let (drop_fn, fn_abi) = match ty.kind() {
333             // FIXME(eddyb) perhaps move some of this logic into
334             // `Instance::resolve_drop_in_place`?
335             ty::Dynamic(..) => {
336                 let virtual_drop = Instance {
337                     def: ty::InstanceDef::Virtual(drop_fn.def_id(), 0),
338                     substs: drop_fn.substs,
339                 };
340                 let fn_abi = FnAbi::of_instance(&bx, virtual_drop, &[]);
341                 let vtable = args[1];
342                 args = &args[..1];
343                 (
344                     meth::VirtualIndex::from_index(ty::COMMON_VTABLE_ENTRIES_DROPINPLACE)
345                         .get_fn(&mut bx, vtable, &fn_abi),
346                     fn_abi,
347                 )
348             }
349             _ => (bx.get_fn_addr(drop_fn), FnAbi::of_instance(&bx, drop_fn, &[])),
350         };
351         helper.do_call(
352             self,
353             &mut bx,
354             fn_abi,
355             drop_fn,
356             args,
357             Some((ReturnDest::Nothing, target)),
358             unwind,
359         );
360     }
361
362     fn codegen_assert_terminator(
363         &mut self,
364         helper: TerminatorCodegenHelper<'tcx>,
365         mut bx: Bx,
366         terminator: &mir::Terminator<'tcx>,
367         cond: &mir::Operand<'tcx>,
368         expected: bool,
369         msg: &mir::AssertMessage<'tcx>,
370         target: mir::BasicBlock,
371         cleanup: Option<mir::BasicBlock>,
372     ) {
373         let span = terminator.source_info.span;
374         let cond = self.codegen_operand(&mut bx, cond).immediate();
375         let mut const_cond = bx.const_to_opt_u128(cond, false).map(|c| c == 1);
376
377         // This case can currently arise only from functions marked
378         // with #[rustc_inherit_overflow_checks] and inlined from
379         // another crate (mostly core::num generic/#[inline] fns),
380         // while the current crate doesn't use overflow checks.
381         // NOTE: Unlike binops, negation doesn't have its own
382         // checked operation, just a comparison with the minimum
383         // value, so we have to check for the assert message.
384         if !bx.check_overflow() {
385             if let AssertKind::OverflowNeg(_) = *msg {
386                 const_cond = Some(expected);
387             }
388         }
389
390         // Don't codegen the panic block if success if known.
391         if const_cond == Some(expected) {
392             helper.funclet_br(self, &mut bx, target);
393             return;
394         }
395
396         // Pass the condition through llvm.expect for branch hinting.
397         let cond = bx.expect(cond, expected);
398
399         // Create the failure block and the conditional branch to it.
400         let lltarget = helper.llblock(self, target);
401         let panic_block = bx.build_sibling_block("panic");
402         if expected {
403             bx.cond_br(cond, lltarget, panic_block.llbb());
404         } else {
405             bx.cond_br(cond, panic_block.llbb(), lltarget);
406         }
407
408         // After this point, bx is the block for the call to panic.
409         bx = panic_block;
410         self.set_debug_loc(&mut bx, terminator.source_info);
411
412         // Get the location information.
413         let location = self.get_caller_location(&mut bx, terminator.source_info).immediate();
414
415         // Put together the arguments to the panic entry point.
416         let (lang_item, args) = match msg {
417             AssertKind::BoundsCheck { ref len, ref index } => {
418                 let len = self.codegen_operand(&mut bx, len).immediate();
419                 let index = self.codegen_operand(&mut bx, index).immediate();
420                 // It's `fn panic_bounds_check(index: usize, len: usize)`,
421                 // and `#[track_caller]` adds an implicit third argument.
422                 (LangItem::PanicBoundsCheck, vec![index, len, location])
423             }
424             _ => {
425                 let msg_str = Symbol::intern(msg.description());
426                 let msg = bx.const_str(msg_str);
427                 // It's `pub fn panic(expr: &str)`, with the wide reference being passed
428                 // as two arguments, and `#[track_caller]` adds an implicit third argument.
429                 (LangItem::Panic, vec![msg.0, msg.1, location])
430             }
431         };
432
433         // Obtain the panic entry point.
434         let def_id = common::langcall(bx.tcx(), Some(span), "", lang_item);
435         let instance = ty::Instance::mono(bx.tcx(), def_id);
436         let fn_abi = FnAbi::of_instance(&bx, instance, &[]);
437         let llfn = bx.get_fn_addr(instance);
438
439         // Codegen the actual panic invoke/call.
440         helper.do_call(self, &mut bx, fn_abi, llfn, &args, None, cleanup);
441     }
442
443     /// Returns `true` if this is indeed a panic intrinsic and codegen is done.
444     fn codegen_panic_intrinsic(
445         &mut self,
446         helper: &TerminatorCodegenHelper<'tcx>,
447         bx: &mut Bx,
448         intrinsic: Option<Symbol>,
449         instance: Option<Instance<'tcx>>,
450         source_info: mir::SourceInfo,
451         destination: &Option<(mir::Place<'tcx>, mir::BasicBlock)>,
452         cleanup: Option<mir::BasicBlock>,
453     ) -> bool {
454         // Emit a panic or a no-op for `assert_*` intrinsics.
455         // These are intrinsics that compile to panics so that we can get a message
456         // which mentions the offending type, even from a const context.
457         #[derive(Debug, PartialEq)]
458         enum AssertIntrinsic {
459             Inhabited,
460             ZeroValid,
461             UninitValid,
462         }
463         let panic_intrinsic = intrinsic.and_then(|i| match i {
464             sym::assert_inhabited => Some(AssertIntrinsic::Inhabited),
465             sym::assert_zero_valid => Some(AssertIntrinsic::ZeroValid),
466             sym::assert_uninit_valid => Some(AssertIntrinsic::UninitValid),
467             _ => None,
468         });
469         if let Some(intrinsic) = panic_intrinsic {
470             use AssertIntrinsic::*;
471             let ty = instance.unwrap().substs.type_at(0);
472             let layout = bx.layout_of(ty);
473             let do_panic = match intrinsic {
474                 Inhabited => layout.abi.is_uninhabited(),
475                 ZeroValid => !layout.might_permit_raw_init(bx, /*zero:*/ true),
476                 UninitValid => !layout.might_permit_raw_init(bx, /*zero:*/ false),
477             };
478             if do_panic {
479                 let msg_str = with_no_trimmed_paths(|| {
480                     if layout.abi.is_uninhabited() {
481                         // Use this error even for the other intrinsics as it is more precise.
482                         format!("attempted to instantiate uninhabited type `{}`", ty)
483                     } else if intrinsic == ZeroValid {
484                         format!("attempted to zero-initialize type `{}`, which is invalid", ty)
485                     } else {
486                         format!("attempted to leave type `{}` uninitialized, which is invalid", ty)
487                     }
488                 });
489                 let msg = bx.const_str(Symbol::intern(&msg_str));
490                 let location = self.get_caller_location(bx, source_info).immediate();
491
492                 // Obtain the panic entry point.
493                 // FIXME: dedup this with `codegen_assert_terminator` above.
494                 let def_id =
495                     common::langcall(bx.tcx(), Some(source_info.span), "", LangItem::Panic);
496                 let instance = ty::Instance::mono(bx.tcx(), def_id);
497                 let fn_abi = FnAbi::of_instance(bx, instance, &[]);
498                 let llfn = bx.get_fn_addr(instance);
499
500                 // Codegen the actual panic invoke/call.
501                 helper.do_call(
502                     self,
503                     bx,
504                     fn_abi,
505                     llfn,
506                     &[msg.0, msg.1, location],
507                     destination.as_ref().map(|(_, bb)| (ReturnDest::Nothing, *bb)),
508                     cleanup,
509                 );
510             } else {
511                 // a NOP
512                 let target = destination.as_ref().unwrap().1;
513                 helper.funclet_br(self, bx, target)
514             }
515             true
516         } else {
517             false
518         }
519     }
520
521     fn codegen_call_terminator(
522         &mut self,
523         helper: TerminatorCodegenHelper<'tcx>,
524         mut bx: Bx,
525         terminator: &mir::Terminator<'tcx>,
526         func: &mir::Operand<'tcx>,
527         args: &[mir::Operand<'tcx>],
528         destination: &Option<(mir::Place<'tcx>, mir::BasicBlock)>,
529         cleanup: Option<mir::BasicBlock>,
530         fn_span: Span,
531     ) {
532         let source_info = terminator.source_info;
533         let span = source_info.span;
534
535         // Create the callee. This is a fn ptr or zero-sized and hence a kind of scalar.
536         let callee = self.codegen_operand(&mut bx, func);
537
538         let (instance, mut llfn) = match *callee.layout.ty.kind() {
539             ty::FnDef(def_id, substs) => (
540                 Some(
541                     ty::Instance::resolve(bx.tcx(), ty::ParamEnv::reveal_all(), def_id, substs)
542                         .unwrap()
543                         .unwrap()
544                         .polymorphize(bx.tcx()),
545                 ),
546                 None,
547             ),
548             ty::FnPtr(_) => (None, Some(callee.immediate())),
549             _ => bug!("{} is not callable", callee.layout.ty),
550         };
551         let def = instance.map(|i| i.def);
552
553         if let Some(ty::InstanceDef::DropGlue(_, None)) = def {
554             // Empty drop glue; a no-op.
555             let &(_, target) = destination.as_ref().unwrap();
556             helper.funclet_br(self, &mut bx, target);
557             return;
558         }
559
560         // FIXME(eddyb) avoid computing this if possible, when `instance` is
561         // available - right now `sig` is only needed for getting the `abi`
562         // and figuring out how many extra args were passed to a C-variadic `fn`.
563         let sig = callee.layout.ty.fn_sig(bx.tcx());
564         let abi = sig.abi();
565
566         // Handle intrinsics old codegen wants Expr's for, ourselves.
567         let intrinsic = match def {
568             Some(ty::InstanceDef::Intrinsic(def_id)) => Some(bx.tcx().item_name(def_id)),
569             _ => None,
570         };
571
572         let extra_args = &args[sig.inputs().skip_binder().len()..];
573         let extra_args = extra_args
574             .iter()
575             .map(|op_arg| {
576                 let op_ty = op_arg.ty(self.mir, bx.tcx());
577                 self.monomorphize(op_ty)
578             })
579             .collect::<Vec<_>>();
580
581         let fn_abi = match instance {
582             Some(instance) => FnAbi::of_instance(&bx, instance, &extra_args),
583             None => FnAbi::of_fn_ptr(&bx, sig, &extra_args),
584         };
585
586         if intrinsic == Some(sym::transmute) {
587             if let Some(destination_ref) = destination.as_ref() {
588                 let &(dest, target) = destination_ref;
589                 self.codegen_transmute(&mut bx, &args[0], dest);
590                 helper.funclet_br(self, &mut bx, target);
591             } else {
592                 // If we are trying to transmute to an uninhabited type,
593                 // it is likely there is no allotted destination. In fact,
594                 // transmuting to an uninhabited type is UB, which means
595                 // we can do what we like. Here, we declare that transmuting
596                 // into an uninhabited type is impossible, so anything following
597                 // it must be unreachable.
598                 assert_eq!(fn_abi.ret.layout.abi, abi::Abi::Uninhabited);
599                 bx.unreachable();
600             }
601             return;
602         }
603
604         if self.codegen_panic_intrinsic(
605             &helper,
606             &mut bx,
607             intrinsic,
608             instance,
609             source_info,
610             destination,
611             cleanup,
612         ) {
613             return;
614         }
615
616         // The arguments we'll be passing. Plus one to account for outptr, if used.
617         let arg_count = fn_abi.args.len() + fn_abi.ret.is_indirect() as usize;
618         let mut llargs = Vec::with_capacity(arg_count);
619
620         // Prepare the return value destination
621         let ret_dest = if let Some((dest, _)) = *destination {
622             let is_intrinsic = intrinsic.is_some();
623             self.make_return_dest(&mut bx, dest, &fn_abi.ret, &mut llargs, is_intrinsic)
624         } else {
625             ReturnDest::Nothing
626         };
627
628         if intrinsic == Some(sym::caller_location) {
629             if let Some((_, target)) = destination.as_ref() {
630                 let location = self
631                     .get_caller_location(&mut bx, mir::SourceInfo { span: fn_span, ..source_info });
632
633                 if let ReturnDest::IndirectOperand(tmp, _) = ret_dest {
634                     location.val.store(&mut bx, tmp);
635                 }
636                 self.store_return(&mut bx, ret_dest, &fn_abi.ret, location.immediate());
637                 helper.funclet_br(self, &mut bx, *target);
638             }
639             return;
640         }
641
642         match intrinsic {
643             None | Some(sym::drop_in_place) => {}
644             Some(sym::copy_nonoverlapping) => unreachable!(),
645             Some(intrinsic) => {
646                 let dest = match ret_dest {
647                     _ if fn_abi.ret.is_indirect() => llargs[0],
648                     ReturnDest::Nothing => {
649                         bx.const_undef(bx.type_ptr_to(bx.arg_memory_ty(&fn_abi.ret)))
650                     }
651                     ReturnDest::IndirectOperand(dst, _) | ReturnDest::Store(dst) => dst.llval,
652                     ReturnDest::DirectOperand(_) => {
653                         bug!("Cannot use direct operand with an intrinsic call")
654                     }
655                 };
656
657                 let args: Vec<_> = args
658                     .iter()
659                     .enumerate()
660                     .map(|(i, arg)| {
661                         // The indices passed to simd_shuffle* in the
662                         // third argument must be constant. This is
663                         // checked by const-qualification, which also
664                         // promotes any complex rvalues to constants.
665                         if i == 2 && intrinsic.as_str().starts_with("simd_shuffle") {
666                             if let mir::Operand::Constant(constant) = arg {
667                                 let c = self.eval_mir_constant(constant);
668                                 let (llval, ty) =
669                                     self.simd_shuffle_indices(&bx, constant.span, constant.ty(), c);
670                                 return OperandRef {
671                                     val: Immediate(llval),
672                                     layout: bx.layout_of(ty),
673                                 };
674                             } else {
675                                 span_bug!(span, "shuffle indices must be constant");
676                             }
677                         }
678
679                         self.codegen_operand(&mut bx, arg)
680                     })
681                     .collect();
682
683                 Self::codegen_intrinsic_call(
684                     &mut bx,
685                     *instance.as_ref().unwrap(),
686                     &fn_abi,
687                     &args,
688                     dest,
689                     span,
690                 );
691
692                 if let ReturnDest::IndirectOperand(dst, _) = ret_dest {
693                     self.store_return(&mut bx, ret_dest, &fn_abi.ret, dst.llval);
694                 }
695
696                 if let Some((_, target)) = *destination {
697                     helper.funclet_br(self, &mut bx, target);
698                 } else {
699                     bx.unreachable();
700                 }
701
702                 return;
703             }
704         }
705
706         // Split the rust-call tupled arguments off.
707         let (first_args, untuple) = if abi == Abi::RustCall && !args.is_empty() {
708             let (tup, args) = args.split_last().unwrap();
709             (args, Some(tup))
710         } else {
711             (args, None)
712         };
713
714         'make_args: for (i, arg) in first_args.iter().enumerate() {
715             let mut op = self.codegen_operand(&mut bx, arg);
716
717             if let (0, Some(ty::InstanceDef::Virtual(_, idx))) = (i, def) {
718                 if let Pair(..) = op.val {
719                     // In the case of Rc<Self>, we need to explicitly pass a
720                     // *mut RcBox<Self> with a Scalar (not ScalarPair) ABI. This is a hack
721                     // that is understood elsewhere in the compiler as a method on
722                     // `dyn Trait`.
723                     // To get a `*mut RcBox<Self>`, we just keep unwrapping newtypes until
724                     // we get a value of a built-in pointer type
725                     'descend_newtypes: while !op.layout.ty.is_unsafe_ptr()
726                         && !op.layout.ty.is_region_ptr()
727                     {
728                         for i in 0..op.layout.fields.count() {
729                             let field = op.extract_field(&mut bx, i);
730                             if !field.layout.is_zst() {
731                                 // we found the one non-zero-sized field that is allowed
732                                 // now find *its* non-zero-sized field, or stop if it's a
733                                 // pointer
734                                 op = field;
735                                 continue 'descend_newtypes;
736                             }
737                         }
738
739                         span_bug!(span, "receiver has no non-zero-sized fields {:?}", op);
740                     }
741
742                     // now that we have `*dyn Trait` or `&dyn Trait`, split it up into its
743                     // data pointer and vtable. Look up the method in the vtable, and pass
744                     // the data pointer as the first argument
745                     match op.val {
746                         Pair(data_ptr, meta) => {
747                             llfn = Some(
748                                 meth::VirtualIndex::from_index(idx).get_fn(&mut bx, meta, &fn_abi),
749                             );
750                             llargs.push(data_ptr);
751                             continue 'make_args;
752                         }
753                         other => bug!("expected a Pair, got {:?}", other),
754                     }
755                 } else if let Ref(data_ptr, Some(meta), _) = op.val {
756                     // by-value dynamic dispatch
757                     llfn = Some(meth::VirtualIndex::from_index(idx).get_fn(&mut bx, meta, &fn_abi));
758                     llargs.push(data_ptr);
759                     continue;
760                 } else {
761                     span_bug!(span, "can't codegen a virtual call on {:?}", op);
762                 }
763             }
764
765             // The callee needs to own the argument memory if we pass it
766             // by-ref, so make a local copy of non-immediate constants.
767             match (arg, op.val) {
768                 (&mir::Operand::Copy(_), Ref(_, None, _))
769                 | (&mir::Operand::Constant(_), Ref(_, None, _)) => {
770                     let tmp = PlaceRef::alloca(&mut bx, op.layout);
771                     op.val.store(&mut bx, tmp);
772                     op.val = Ref(tmp.llval, None, tmp.align);
773                 }
774                 _ => {}
775             }
776
777             self.codegen_argument(&mut bx, op, &mut llargs, &fn_abi.args[i]);
778         }
779         if let Some(tup) = untuple {
780             self.codegen_arguments_untupled(
781                 &mut bx,
782                 tup,
783                 &mut llargs,
784                 &fn_abi.args[first_args.len()..],
785             )
786         }
787
788         let needs_location =
789             instance.map_or(false, |i| i.def.requires_caller_location(self.cx.tcx()));
790         if needs_location {
791             assert_eq!(
792                 fn_abi.args.len(),
793                 args.len() + 1,
794                 "#[track_caller] fn's must have 1 more argument in their ABI than in their MIR",
795             );
796             let location =
797                 self.get_caller_location(&mut bx, mir::SourceInfo { span: fn_span, ..source_info });
798             debug!(
799                 "codegen_call_terminator({:?}): location={:?} (fn_span {:?})",
800                 terminator, location, fn_span
801             );
802
803             let last_arg = fn_abi.args.last().unwrap();
804             self.codegen_argument(&mut bx, location, &mut llargs, last_arg);
805         }
806
807         let fn_ptr = match (llfn, instance) {
808             (Some(llfn), _) => llfn,
809             (None, Some(instance)) => bx.get_fn_addr(instance),
810             _ => span_bug!(span, "no llfn for call"),
811         };
812
813         helper.do_call(
814             self,
815             &mut bx,
816             fn_abi,
817             fn_ptr,
818             &llargs,
819             destination.as_ref().map(|&(_, target)| (ret_dest, target)),
820             cleanup,
821         );
822     }
823
824     fn codegen_asm_terminator(
825         &mut self,
826         helper: TerminatorCodegenHelper<'tcx>,
827         mut bx: Bx,
828         terminator: &mir::Terminator<'tcx>,
829         template: &[ast::InlineAsmTemplatePiece],
830         operands: &[mir::InlineAsmOperand<'tcx>],
831         options: ast::InlineAsmOptions,
832         line_spans: &[Span],
833         destination: Option<mir::BasicBlock>,
834     ) {
835         let span = terminator.source_info.span;
836
837         let operands: Vec<_> = operands
838             .iter()
839             .map(|op| match *op {
840                 mir::InlineAsmOperand::In { reg, ref value } => {
841                     let value = self.codegen_operand(&mut bx, value);
842                     InlineAsmOperandRef::In { reg, value }
843                 }
844                 mir::InlineAsmOperand::Out { reg, late, ref place } => {
845                     let place = place.map(|place| self.codegen_place(&mut bx, place.as_ref()));
846                     InlineAsmOperandRef::Out { reg, late, place }
847                 }
848                 mir::InlineAsmOperand::InOut { reg, late, ref in_value, ref out_place } => {
849                     let in_value = self.codegen_operand(&mut bx, in_value);
850                     let out_place =
851                         out_place.map(|out_place| self.codegen_place(&mut bx, out_place.as_ref()));
852                     InlineAsmOperandRef::InOut { reg, late, in_value, out_place }
853                 }
854                 mir::InlineAsmOperand::Const { ref value } => {
855                     let const_value = self
856                         .eval_mir_constant(value)
857                         .unwrap_or_else(|_| span_bug!(span, "asm const cannot be resolved"));
858                     let string = common::asm_const_to_str(
859                         bx.tcx(),
860                         span,
861                         const_value,
862                         bx.layout_of(value.ty()),
863                     );
864                     InlineAsmOperandRef::Const { string }
865                 }
866                 mir::InlineAsmOperand::SymFn { ref value } => {
867                     let literal = self.monomorphize(value.literal);
868                     if let ty::FnDef(def_id, substs) = *literal.ty().kind() {
869                         let instance = ty::Instance::resolve_for_fn_ptr(
870                             bx.tcx(),
871                             ty::ParamEnv::reveal_all(),
872                             def_id,
873                             substs,
874                         )
875                         .unwrap();
876                         InlineAsmOperandRef::SymFn { instance }
877                     } else {
878                         span_bug!(span, "invalid type for asm sym (fn)");
879                     }
880                 }
881                 mir::InlineAsmOperand::SymStatic { def_id } => {
882                     InlineAsmOperandRef::SymStatic { def_id }
883                 }
884             })
885             .collect();
886
887         bx.codegen_inline_asm(template, &operands, options, line_spans);
888
889         if let Some(target) = destination {
890             helper.funclet_br(self, &mut bx, target);
891         } else {
892             bx.unreachable();
893         }
894     }
895 }
896
897 impl<'a, 'tcx, Bx: BuilderMethods<'a, 'tcx>> FunctionCx<'a, 'tcx, Bx> {
898     pub fn codegen_block(&mut self, bb: mir::BasicBlock) {
899         let mut bx = self.build_block(bb);
900         let mir = self.mir;
901         let data = &mir[bb];
902
903         debug!("codegen_block({:?}={:?})", bb, data);
904
905         for statement in &data.statements {
906             bx = self.codegen_statement(bx, statement);
907         }
908
909         self.codegen_terminator(bx, bb, data.terminator());
910     }
911
912     fn codegen_terminator(
913         &mut self,
914         mut bx: Bx,
915         bb: mir::BasicBlock,
916         terminator: &'tcx mir::Terminator<'tcx>,
917     ) {
918         debug!("codegen_terminator: {:?}", terminator);
919
920         // Create the cleanup bundle, if needed.
921         let funclet_bb = self.cleanup_kinds[bb].funclet_bb(bb);
922         let helper = TerminatorCodegenHelper { bb, terminator, funclet_bb };
923
924         self.set_debug_loc(&mut bx, terminator.source_info);
925         match terminator.kind {
926             mir::TerminatorKind::Resume => self.codegen_resume_terminator(helper, bx),
927
928             mir::TerminatorKind::Abort => {
929                 bx.abort();
930                 // `abort` does not terminate the block, so we still need to generate
931                 // an `unreachable` terminator after it.
932                 bx.unreachable();
933             }
934
935             mir::TerminatorKind::Goto { target } => {
936                 if bb == target {
937                     // This is an unconditional branch back to this same basic block. That means we
938                     // have something like a `loop {}` statement. LLVM versions before 12.0
939                     // miscompile this because they assume forward progress. For older versions
940                     // try to handle just this specific case which comes up commonly in practice
941                     // (e.g., in embedded code).
942                     //
943                     // NB: the `sideeffect` currently checks for the LLVM version used internally.
944                     bx.sideeffect();
945                 }
946
947                 helper.funclet_br(self, &mut bx, target);
948             }
949
950             mir::TerminatorKind::SwitchInt { ref discr, switch_ty, ref targets } => {
951                 self.codegen_switchint_terminator(helper, bx, discr, switch_ty, targets);
952             }
953
954             mir::TerminatorKind::Return => {
955                 self.codegen_return_terminator(bx);
956             }
957
958             mir::TerminatorKind::Unreachable => {
959                 bx.unreachable();
960             }
961
962             mir::TerminatorKind::Drop { place, target, unwind } => {
963                 self.codegen_drop_terminator(helper, bx, place, target, unwind);
964             }
965
966             mir::TerminatorKind::Assert { ref cond, expected, ref msg, target, cleanup } => {
967                 self.codegen_assert_terminator(
968                     helper, bx, terminator, cond, expected, msg, target, cleanup,
969                 );
970             }
971
972             mir::TerminatorKind::DropAndReplace { .. } => {
973                 bug!("undesugared DropAndReplace in codegen: {:?}", terminator);
974             }
975
976             mir::TerminatorKind::Call {
977                 ref func,
978                 ref args,
979                 ref destination,
980                 cleanup,
981                 from_hir_call: _,
982                 fn_span,
983             } => {
984                 self.codegen_call_terminator(
985                     helper,
986                     bx,
987                     terminator,
988                     func,
989                     args,
990                     destination,
991                     cleanup,
992                     fn_span,
993                 );
994             }
995             mir::TerminatorKind::GeneratorDrop | mir::TerminatorKind::Yield { .. } => {
996                 bug!("generator ops in codegen")
997             }
998             mir::TerminatorKind::FalseEdge { .. } | mir::TerminatorKind::FalseUnwind { .. } => {
999                 bug!("borrowck false edges in codegen")
1000             }
1001
1002             mir::TerminatorKind::InlineAsm {
1003                 template,
1004                 ref operands,
1005                 options,
1006                 line_spans,
1007                 destination,
1008             } => {
1009                 self.codegen_asm_terminator(
1010                     helper,
1011                     bx,
1012                     terminator,
1013                     template,
1014                     operands,
1015                     options,
1016                     line_spans,
1017                     destination,
1018                 );
1019             }
1020         }
1021     }
1022
1023     fn codegen_argument(
1024         &mut self,
1025         bx: &mut Bx,
1026         op: OperandRef<'tcx, Bx::Value>,
1027         llargs: &mut Vec<Bx::Value>,
1028         arg: &ArgAbi<'tcx, Ty<'tcx>>,
1029     ) {
1030         // Fill padding with undef value, where applicable.
1031         if let Some(ty) = arg.pad {
1032             llargs.push(bx.const_undef(bx.reg_backend_type(&ty)))
1033         }
1034
1035         if arg.is_ignore() {
1036             return;
1037         }
1038
1039         if let PassMode::Pair(..) = arg.mode {
1040             match op.val {
1041                 Pair(a, b) => {
1042                     llargs.push(a);
1043                     llargs.push(b);
1044                     return;
1045                 }
1046                 _ => bug!("codegen_argument: {:?} invalid for pair argument", op),
1047             }
1048         } else if arg.is_unsized_indirect() {
1049             match op.val {
1050                 Ref(a, Some(b), _) => {
1051                     llargs.push(a);
1052                     llargs.push(b);
1053                     return;
1054                 }
1055                 _ => bug!("codegen_argument: {:?} invalid for unsized indirect argument", op),
1056             }
1057         }
1058
1059         // Force by-ref if we have to load through a cast pointer.
1060         let (mut llval, align, by_ref) = match op.val {
1061             Immediate(_) | Pair(..) => match arg.mode {
1062                 PassMode::Indirect { .. } | PassMode::Cast(_) => {
1063                     let scratch = PlaceRef::alloca(bx, arg.layout);
1064                     op.val.store(bx, scratch);
1065                     (scratch.llval, scratch.align, true)
1066                 }
1067                 _ => (op.immediate_or_packed_pair(bx), arg.layout.align.abi, false),
1068             },
1069             Ref(llval, _, align) => {
1070                 if arg.is_indirect() && align < arg.layout.align.abi {
1071                     // `foo(packed.large_field)`. We can't pass the (unaligned) field directly. I
1072                     // think that ATM (Rust 1.16) we only pass temporaries, but we shouldn't
1073                     // have scary latent bugs around.
1074
1075                     let scratch = PlaceRef::alloca(bx, arg.layout);
1076                     base::memcpy_ty(
1077                         bx,
1078                         scratch.llval,
1079                         scratch.align,
1080                         llval,
1081                         align,
1082                         op.layout,
1083                         MemFlags::empty(),
1084                     );
1085                     (scratch.llval, scratch.align, true)
1086                 } else {
1087                     (llval, align, true)
1088                 }
1089             }
1090         };
1091
1092         if by_ref && !arg.is_indirect() {
1093             // Have to load the argument, maybe while casting it.
1094             if let PassMode::Cast(ty) = arg.mode {
1095                 let llty = bx.cast_backend_type(&ty);
1096                 let addr = bx.pointercast(llval, bx.type_ptr_to(llty));
1097                 llval = bx.load(llty, addr, align.min(arg.layout.align.abi));
1098             } else {
1099                 // We can't use `PlaceRef::load` here because the argument
1100                 // may have a type we don't treat as immediate, but the ABI
1101                 // used for this call is passing it by-value. In that case,
1102                 // the load would just produce `OperandValue::Ref` instead
1103                 // of the `OperandValue::Immediate` we need for the call.
1104                 llval = bx.load(bx.backend_type(arg.layout), llval, align);
1105                 if let abi::Abi::Scalar(ref scalar) = arg.layout.abi {
1106                     if scalar.is_bool() {
1107                         bx.range_metadata(llval, 0..2);
1108                     }
1109                 }
1110                 // We store bools as `i8` so we need to truncate to `i1`.
1111                 llval = bx.to_immediate(llval, arg.layout);
1112             }
1113         }
1114
1115         llargs.push(llval);
1116     }
1117
1118     fn codegen_arguments_untupled(
1119         &mut self,
1120         bx: &mut Bx,
1121         operand: &mir::Operand<'tcx>,
1122         llargs: &mut Vec<Bx::Value>,
1123         args: &[ArgAbi<'tcx, Ty<'tcx>>],
1124     ) {
1125         let tuple = self.codegen_operand(bx, operand);
1126
1127         // Handle both by-ref and immediate tuples.
1128         if let Ref(llval, None, align) = tuple.val {
1129             let tuple_ptr = PlaceRef::new_sized_aligned(llval, tuple.layout, align);
1130             for i in 0..tuple.layout.fields.count() {
1131                 let field_ptr = tuple_ptr.project_field(bx, i);
1132                 let field = bx.load_operand(field_ptr);
1133                 self.codegen_argument(bx, field, llargs, &args[i]);
1134             }
1135         } else if let Ref(_, Some(_), _) = tuple.val {
1136             bug!("closure arguments must be sized")
1137         } else {
1138             // If the tuple is immediate, the elements are as well.
1139             for i in 0..tuple.layout.fields.count() {
1140                 let op = tuple.extract_field(bx, i);
1141                 self.codegen_argument(bx, op, llargs, &args[i]);
1142             }
1143         }
1144     }
1145
1146     fn get_caller_location(
1147         &mut self,
1148         bx: &mut Bx,
1149         mut source_info: mir::SourceInfo,
1150     ) -> OperandRef<'tcx, Bx::Value> {
1151         let tcx = bx.tcx();
1152
1153         let mut span_to_caller_location = |span: Span| {
1154             let topmost = span.ctxt().outer_expn().expansion_cause().unwrap_or(span);
1155             let caller = tcx.sess.source_map().lookup_char_pos(topmost.lo());
1156             let const_loc = tcx.const_caller_location((
1157                 Symbol::intern(&caller.file.name.prefer_remapped().to_string_lossy()),
1158                 caller.line as u32,
1159                 caller.col_display as u32 + 1,
1160             ));
1161             OperandRef::from_const(bx, const_loc, bx.tcx().caller_location_ty())
1162         };
1163
1164         // Walk up the `SourceScope`s, in case some of them are from MIR inlining.
1165         // If so, the starting `source_info.span` is in the innermost inlined
1166         // function, and will be replaced with outer callsite spans as long
1167         // as the inlined functions were `#[track_caller]`.
1168         loop {
1169             let scope_data = &self.mir.source_scopes[source_info.scope];
1170
1171             if let Some((callee, callsite_span)) = scope_data.inlined {
1172                 // Stop inside the most nested non-`#[track_caller]` function,
1173                 // before ever reaching its caller (which is irrelevant).
1174                 if !callee.def.requires_caller_location(tcx) {
1175                     return span_to_caller_location(source_info.span);
1176                 }
1177                 source_info.span = callsite_span;
1178             }
1179
1180             // Skip past all of the parents with `inlined: None`.
1181             match scope_data.inlined_parent_scope {
1182                 Some(parent) => source_info.scope = parent,
1183                 None => break,
1184             }
1185         }
1186
1187         // No inlined `SourceScope`s, or all of them were `#[track_caller]`.
1188         self.caller_location.unwrap_or_else(|| span_to_caller_location(source_info.span))
1189     }
1190
1191     fn get_personality_slot(&mut self, bx: &mut Bx) -> PlaceRef<'tcx, Bx::Value> {
1192         let cx = bx.cx();
1193         if let Some(slot) = self.personality_slot {
1194             slot
1195         } else {
1196             let layout = cx.layout_of(
1197                 cx.tcx().intern_tup(&[cx.tcx().mk_mut_ptr(cx.tcx().types.u8), cx.tcx().types.i32]),
1198             );
1199             let slot = PlaceRef::alloca(bx, layout);
1200             self.personality_slot = Some(slot);
1201             slot
1202         }
1203     }
1204
1205     /// Returns the landing/cleanup pad wrapper around the given basic block.
1206     // FIXME(eddyb) rename this to `eh_pad_for`.
1207     fn landing_pad_for(&mut self, bb: mir::BasicBlock) -> Bx::BasicBlock {
1208         if let Some(landing_pad) = self.landing_pads[bb] {
1209             return landing_pad;
1210         }
1211
1212         let landing_pad = self.landing_pad_for_uncached(bb);
1213         self.landing_pads[bb] = Some(landing_pad);
1214         landing_pad
1215     }
1216
1217     // FIXME(eddyb) rename this to `eh_pad_for_uncached`.
1218     fn landing_pad_for_uncached(&mut self, bb: mir::BasicBlock) -> Bx::BasicBlock {
1219         let llbb = self.llbb(bb);
1220         if base::wants_msvc_seh(self.cx.sess()) {
1221             let funclet;
1222             let ret_llbb;
1223             match self.mir[bb].terminator.as_ref().map(|t| &t.kind) {
1224                 // This is a basic block that we're aborting the program for,
1225                 // notably in an `extern` function. These basic blocks are inserted
1226                 // so that we assert that `extern` functions do indeed not panic,
1227                 // and if they do we abort the process.
1228                 //
1229                 // On MSVC these are tricky though (where we're doing funclets). If
1230                 // we were to do a cleanuppad (like below) the normal functions like
1231                 // `longjmp` would trigger the abort logic, terminating the
1232                 // program. Instead we insert the equivalent of `catch(...)` for C++
1233                 // which magically doesn't trigger when `longjmp` files over this
1234                 // frame.
1235                 //
1236                 // Lots more discussion can be found on #48251 but this codegen is
1237                 // modeled after clang's for:
1238                 //
1239                 //      try {
1240                 //          foo();
1241                 //      } catch (...) {
1242                 //          bar();
1243                 //      }
1244                 Some(&mir::TerminatorKind::Abort) => {
1245                     let mut cs_bx = self.new_block(&format!("cs_funclet{:?}", bb));
1246                     let mut cp_bx = self.new_block(&format!("cp_funclet{:?}", bb));
1247                     ret_llbb = cs_bx.llbb();
1248
1249                     let cs = cs_bx.catch_switch(None, None, 1);
1250                     cs_bx.add_handler(cs, cp_bx.llbb());
1251
1252                     // The "null" here is actually a RTTI type descriptor for the
1253                     // C++ personality function, but `catch (...)` has no type so
1254                     // it's null. The 64 here is actually a bitfield which
1255                     // represents that this is a catch-all block.
1256                     let null = cp_bx.const_null(
1257                         cp_bx.type_i8p_ext(cp_bx.cx().data_layout().instruction_address_space),
1258                     );
1259                     let sixty_four = cp_bx.const_i32(64);
1260                     funclet = cp_bx.catch_pad(cs, &[null, sixty_four, null]);
1261                     cp_bx.br(llbb);
1262                 }
1263                 _ => {
1264                     let mut cleanup_bx = self.new_block(&format!("funclet_{:?}", bb));
1265                     ret_llbb = cleanup_bx.llbb();
1266                     funclet = cleanup_bx.cleanup_pad(None, &[]);
1267                     cleanup_bx.br(llbb);
1268                 }
1269             }
1270             self.funclets[bb] = Some(funclet);
1271             ret_llbb
1272         } else {
1273             let mut bx = self.new_block("cleanup");
1274
1275             let llpersonality = self.cx.eh_personality();
1276             let llretty = self.landing_pad_type();
1277             let lp = bx.landing_pad(llretty, llpersonality, 1);
1278             bx.set_cleanup(lp);
1279
1280             let slot = self.get_personality_slot(&mut bx);
1281             slot.storage_live(&mut bx);
1282             Pair(bx.extract_value(lp, 0), bx.extract_value(lp, 1)).store(&mut bx, slot);
1283
1284             bx.br(llbb);
1285             bx.llbb()
1286         }
1287     }
1288
1289     fn landing_pad_type(&self) -> Bx::Type {
1290         let cx = self.cx;
1291         cx.type_struct(&[cx.type_i8p(), cx.type_i32()], false)
1292     }
1293
1294     fn unreachable_block(&mut self) -> Bx::BasicBlock {
1295         self.unreachable_block.unwrap_or_else(|| {
1296             let mut bx = self.new_block("unreachable");
1297             bx.unreachable();
1298             self.unreachable_block = Some(bx.llbb());
1299             bx.llbb()
1300         })
1301     }
1302
1303     // FIXME(eddyb) replace with `build_sibling_block`/`append_sibling_block`
1304     // (which requires having a `Bx` already, and not all callers do).
1305     fn new_block(&self, name: &str) -> Bx {
1306         let llbb = Bx::append_block(self.cx, self.llfn, name);
1307         Bx::build(self.cx, llbb)
1308     }
1309
1310     /// Get the backend `BasicBlock` for a MIR `BasicBlock`, either already
1311     /// cached in `self.cached_llbbs`, or created on demand (and cached).
1312     // FIXME(eddyb) rename `llbb` and other `ll`-prefixed things to use a
1313     // more backend-agnostic prefix such as `cg` (i.e. this would be `cgbb`).
1314     pub fn llbb(&mut self, bb: mir::BasicBlock) -> Bx::BasicBlock {
1315         self.cached_llbbs[bb].unwrap_or_else(|| {
1316             // FIXME(eddyb) only name the block if `fewer_names` is `false`.
1317             let llbb = Bx::append_block(self.cx, self.llfn, &format!("{:?}", bb));
1318             self.cached_llbbs[bb] = Some(llbb);
1319             llbb
1320         })
1321     }
1322
1323     pub fn build_block(&mut self, bb: mir::BasicBlock) -> Bx {
1324         let llbb = self.llbb(bb);
1325         Bx::build(self.cx, llbb)
1326     }
1327
1328     fn make_return_dest(
1329         &mut self,
1330         bx: &mut Bx,
1331         dest: mir::Place<'tcx>,
1332         fn_ret: &ArgAbi<'tcx, Ty<'tcx>>,
1333         llargs: &mut Vec<Bx::Value>,
1334         is_intrinsic: bool,
1335     ) -> ReturnDest<'tcx, Bx::Value> {
1336         // If the return is ignored, we can just return a do-nothing `ReturnDest`.
1337         if fn_ret.is_ignore() {
1338             return ReturnDest::Nothing;
1339         }
1340         let dest = if let Some(index) = dest.as_local() {
1341             match self.locals[index] {
1342                 LocalRef::Place(dest) => dest,
1343                 LocalRef::UnsizedPlace(_) => bug!("return type must be sized"),
1344                 LocalRef::Operand(None) => {
1345                     // Handle temporary places, specifically `Operand` ones, as
1346                     // they don't have `alloca`s.
1347                     return if fn_ret.is_indirect() {
1348                         // Odd, but possible, case, we have an operand temporary,
1349                         // but the calling convention has an indirect return.
1350                         let tmp = PlaceRef::alloca(bx, fn_ret.layout);
1351                         tmp.storage_live(bx);
1352                         llargs.push(tmp.llval);
1353                         ReturnDest::IndirectOperand(tmp, index)
1354                     } else if is_intrinsic {
1355                         // Currently, intrinsics always need a location to store
1356                         // the result, so we create a temporary `alloca` for the
1357                         // result.
1358                         let tmp = PlaceRef::alloca(bx, fn_ret.layout);
1359                         tmp.storage_live(bx);
1360                         ReturnDest::IndirectOperand(tmp, index)
1361                     } else {
1362                         ReturnDest::DirectOperand(index)
1363                     };
1364                 }
1365                 LocalRef::Operand(Some(_)) => {
1366                     bug!("place local already assigned to");
1367                 }
1368             }
1369         } else {
1370             self.codegen_place(
1371                 bx,
1372                 mir::PlaceRef { local: dest.local, projection: &dest.projection },
1373             )
1374         };
1375         if fn_ret.is_indirect() {
1376             if dest.align < dest.layout.align.abi {
1377                 // Currently, MIR code generation does not create calls
1378                 // that store directly to fields of packed structs (in
1379                 // fact, the calls it creates write only to temps).
1380                 //
1381                 // If someone changes that, please update this code path
1382                 // to create a temporary.
1383                 span_bug!(self.mir.span, "can't directly store to unaligned value");
1384             }
1385             llargs.push(dest.llval);
1386             ReturnDest::Nothing
1387         } else {
1388             ReturnDest::Store(dest)
1389         }
1390     }
1391
1392     fn codegen_transmute(&mut self, bx: &mut Bx, src: &mir::Operand<'tcx>, dst: mir::Place<'tcx>) {
1393         if let Some(index) = dst.as_local() {
1394             match self.locals[index] {
1395                 LocalRef::Place(place) => self.codegen_transmute_into(bx, src, place),
1396                 LocalRef::UnsizedPlace(_) => bug!("transmute must not involve unsized locals"),
1397                 LocalRef::Operand(None) => {
1398                     let dst_layout = bx.layout_of(self.monomorphized_place_ty(dst.as_ref()));
1399                     assert!(!dst_layout.ty.has_erasable_regions(self.cx.tcx()));
1400                     let place = PlaceRef::alloca(bx, dst_layout);
1401                     place.storage_live(bx);
1402                     self.codegen_transmute_into(bx, src, place);
1403                     let op = bx.load_operand(place);
1404                     place.storage_dead(bx);
1405                     self.locals[index] = LocalRef::Operand(Some(op));
1406                     self.debug_introduce_local(bx, index);
1407                 }
1408                 LocalRef::Operand(Some(op)) => {
1409                     assert!(op.layout.is_zst(), "assigning to initialized SSAtemp");
1410                 }
1411             }
1412         } else {
1413             let dst = self.codegen_place(bx, dst.as_ref());
1414             self.codegen_transmute_into(bx, src, dst);
1415         }
1416     }
1417
1418     fn codegen_transmute_into(
1419         &mut self,
1420         bx: &mut Bx,
1421         src: &mir::Operand<'tcx>,
1422         dst: PlaceRef<'tcx, Bx::Value>,
1423     ) {
1424         let src = self.codegen_operand(bx, src);
1425
1426         // Special-case transmutes between scalars as simple bitcasts.
1427         match (&src.layout.abi, &dst.layout.abi) {
1428             (abi::Abi::Scalar(src_scalar), abi::Abi::Scalar(dst_scalar)) => {
1429                 // HACK(eddyb) LLVM doesn't like `bitcast`s between pointers and non-pointers.
1430                 if (src_scalar.value == abi::Pointer) == (dst_scalar.value == abi::Pointer) {
1431                     assert_eq!(src.layout.size, dst.layout.size);
1432
1433                     // NOTE(eddyb) the `from_immediate` and `to_immediate_scalar`
1434                     // conversions allow handling `bool`s the same as `u8`s.
1435                     let src = bx.from_immediate(src.immediate());
1436                     let src_as_dst = bx.bitcast(src, bx.backend_type(dst.layout));
1437                     Immediate(bx.to_immediate_scalar(src_as_dst, dst_scalar)).store(bx, dst);
1438                     return;
1439                 }
1440             }
1441             _ => {}
1442         }
1443
1444         let llty = bx.backend_type(src.layout);
1445         let cast_ptr = bx.pointercast(dst.llval, bx.type_ptr_to(llty));
1446         let align = src.layout.align.abi.min(dst.align);
1447         src.val.store(bx, PlaceRef::new_sized_aligned(cast_ptr, src.layout, align));
1448     }
1449
1450     // Stores the return value of a function call into it's final location.
1451     fn store_return(
1452         &mut self,
1453         bx: &mut Bx,
1454         dest: ReturnDest<'tcx, Bx::Value>,
1455         ret_abi: &ArgAbi<'tcx, Ty<'tcx>>,
1456         llval: Bx::Value,
1457     ) {
1458         use self::ReturnDest::*;
1459
1460         match dest {
1461             Nothing => (),
1462             Store(dst) => bx.store_arg(&ret_abi, llval, dst),
1463             IndirectOperand(tmp, index) => {
1464                 let op = bx.load_operand(tmp);
1465                 tmp.storage_dead(bx);
1466                 self.locals[index] = LocalRef::Operand(Some(op));
1467                 self.debug_introduce_local(bx, index);
1468             }
1469             DirectOperand(index) => {
1470                 // If there is a cast, we have to store and reload.
1471                 let op = if let PassMode::Cast(_) = ret_abi.mode {
1472                     let tmp = PlaceRef::alloca(bx, ret_abi.layout);
1473                     tmp.storage_live(bx);
1474                     bx.store_arg(&ret_abi, llval, tmp);
1475                     let op = bx.load_operand(tmp);
1476                     tmp.storage_dead(bx);
1477                     op
1478                 } else {
1479                     OperandRef::from_immediate_or_packed_pair(bx, llval, ret_abi.layout)
1480                 };
1481                 self.locals[index] = LocalRef::Operand(Some(op));
1482                 self.debug_introduce_local(bx, index);
1483             }
1484         }
1485     }
1486 }
1487
1488 enum ReturnDest<'tcx, V> {
1489     // Do nothing; the return value is indirect or ignored.
1490     Nothing,
1491     // Store the return value to the pointer.
1492     Store(PlaceRef<'tcx, V>),
1493     // Store an indirect return value to an operand local place.
1494     IndirectOperand(PlaceRef<'tcx, V>, mir::Local),
1495     // Store a direct return value to an operand local place.
1496     DirectOperand(mir::Local),
1497 }