]> git.lizzy.rs Git - rust.git/blob - src/librustc/mir/mod.rs
9ac1465cb0ba9a0f58dd773a7cdfe0591e28b73e
[rust.git] / src / librustc / mir / mod.rs
1 // ignore-tidy-filelength
2
3 //! MIR datatypes and passes. See the [rustc guide] for more info.
4 //!
5 //! [rustc guide]: https://rust-lang.github.io/rustc-guide/mir/index.html
6
7 use crate::hir::def::{CtorKind, Namespace};
8 use crate::hir::def_id::DefId;
9 use crate::hir::{self, InlineAsm as HirInlineAsm};
10 use crate::mir::interpret::{ConstValue, PanicInfo, Scalar};
11 use crate::mir::visit::MirVisitable;
12 use crate::ty::adjustment::PointerCast;
13 use crate::ty::fold::{TypeFoldable, TypeFolder, TypeVisitor};
14 use crate::ty::layout::VariantIdx;
15 use crate::ty::print::{FmtPrinter, Printer};
16 use crate::ty::subst::{Subst, SubstsRef};
17 use crate::ty::{
18     self, AdtDef, CanonicalUserTypeAnnotations, Region, Ty, TyCtxt,
19     UserTypeAnnotationIndex,
20 };
21
22 use polonius_engine::Atom;
23 use rustc_index::bit_set::BitMatrix;
24 use rustc_data_structures::fx::FxHashSet;
25 use rustc_data_structures::graph::dominators::{dominators, Dominators};
26 use rustc_data_structures::graph::{self, GraphPredecessors, GraphSuccessors};
27 use rustc_index::vec::{Idx, IndexVec};
28 use rustc_data_structures::sync::Lrc;
29 use rustc_data_structures::sync::MappedReadGuard;
30 use rustc_macros::HashStable;
31 use rustc_serialize::{Encodable, Decodable};
32 use smallvec::SmallVec;
33 use std::borrow::Cow;
34 use std::fmt::{self, Debug, Display, Formatter, Write};
35 use std::ops::{Index, IndexMut};
36 use std::slice;
37 use std::vec::IntoIter;
38 use std::{iter, mem, option, u32};
39 use syntax::ast::Name;
40 use syntax::symbol::{InternedString, Symbol};
41 use syntax_pos::{Span, DUMMY_SP};
42
43 pub use crate::mir::interpret::AssertMessage;
44
45 mod cache;
46 pub mod interpret;
47 pub mod mono;
48 pub mod tcx;
49 pub mod traversal;
50 pub mod visit;
51
52 /// Types for locals
53 type LocalDecls<'tcx> = IndexVec<Local, LocalDecl<'tcx>>;
54
55 pub trait HasLocalDecls<'tcx> {
56     fn local_decls(&self) -> &LocalDecls<'tcx>;
57 }
58
59 impl<'tcx> HasLocalDecls<'tcx> for LocalDecls<'tcx> {
60     fn local_decls(&self) -> &LocalDecls<'tcx> {
61         self
62     }
63 }
64
65 impl<'tcx> HasLocalDecls<'tcx> for Body<'tcx> {
66     fn local_decls(&self) -> &LocalDecls<'tcx> {
67         &self.local_decls
68     }
69 }
70
71 /// The various "big phases" that MIR goes through.
72 ///
73 /// Warning: ordering of variants is significant.
74 #[derive(Copy, Clone, RustcEncodable, RustcDecodable, Debug, PartialEq, Eq, PartialOrd, Ord)]
75 pub enum MirPhase {
76     Build = 0,
77     Const = 1,
78     Validated = 2,
79     Optimized = 3,
80 }
81
82 impl MirPhase {
83     /// Gets the index of the current MirPhase within the set of all `MirPhase`s.
84     pub fn phase_index(&self) -> usize {
85         *self as usize
86     }
87 }
88
89 /// The lowered representation of a single function.
90 #[derive(Clone, RustcEncodable, RustcDecodable, Debug)]
91 pub struct Body<'tcx> {
92     /// A list of basic blocks. References to basic block use a newtyped index type `BasicBlock`
93     /// that indexes into this vector.
94     basic_blocks: IndexVec<BasicBlock, BasicBlockData<'tcx>>,
95
96     /// Records how far through the "desugaring and optimization" process this particular
97     /// MIR has traversed. This is particularly useful when inlining, since in that context
98     /// we instantiate the promoted constants and add them to our promoted vector -- but those
99     /// promoted items have already been optimized, whereas ours have not. This field allows
100     /// us to see the difference and forego optimization on the inlined promoted items.
101     pub phase: MirPhase,
102
103     /// A list of source scopes; these are referenced by statements
104     /// and used for debuginfo. Indexed by a `SourceScope`.
105     pub source_scopes: IndexVec<SourceScope, SourceScopeData>,
106
107     /// Crate-local information for each source scope, that can't (and
108     /// needn't) be tracked across crates.
109     pub source_scope_local_data: ClearCrossCrate<IndexVec<SourceScope, SourceScopeLocalData>>,
110
111     /// The yield type of the function, if it is a generator.
112     pub yield_ty: Option<Ty<'tcx>>,
113
114     /// Generator drop glue.
115     pub generator_drop: Option<Box<Body<'tcx>>>,
116
117     /// The layout of a generator. Produced by the state transformation.
118     pub generator_layout: Option<GeneratorLayout<'tcx>>,
119
120     /// Declarations of locals.
121     ///
122     /// The first local is the return value pointer, followed by `arg_count`
123     /// locals for the function arguments, followed by any user-declared
124     /// variables and temporaries.
125     pub local_decls: LocalDecls<'tcx>,
126
127     /// User type annotations.
128     pub user_type_annotations: CanonicalUserTypeAnnotations<'tcx>,
129
130     /// The number of arguments this function takes.
131     ///
132     /// Starting at local 1, `arg_count` locals will be provided by the caller
133     /// and can be assumed to be initialized.
134     ///
135     /// If this MIR was built for a constant, this will be 0.
136     pub arg_count: usize,
137
138     /// Mark an argument local (which must be a tuple) as getting passed as
139     /// its individual components at the LLVM level.
140     ///
141     /// This is used for the "rust-call" ABI.
142     pub spread_arg: Option<Local>,
143
144     /// Names and capture modes of all the closure upvars, assuming
145     /// the first argument is either the closure or a reference to it.
146     //
147     // NOTE(eddyb) This is *strictly* a temporary hack for codegen
148     // debuginfo generation, and will be removed at some point.
149     // Do **NOT** use it for anything else; upvar information should not be
150     // in the MIR, so please rely on local crate HIR or other side-channels.
151     pub __upvar_debuginfo_codegen_only_do_not_use: Vec<UpvarDebuginfo>,
152
153     /// Mark this MIR of a const context other than const functions as having converted a `&&` or
154     /// `||` expression into `&` or `|` respectively. This is problematic because if we ever stop
155     /// this conversion from happening and use short circuiting, we will cause the following code
156     /// to change the value of `x`: `let mut x = 42; false && { x = 55; true };`
157     ///
158     /// List of places where control flow was destroyed. Used for error reporting.
159     pub control_flow_destroyed: Vec<(Span, String)>,
160
161     /// A span representing this MIR, for error reporting.
162     pub span: Span,
163
164     /// A cache for various calculations.
165     cache: cache::Cache,
166 }
167
168 impl<'tcx> Body<'tcx> {
169     pub fn new(
170         basic_blocks: IndexVec<BasicBlock, BasicBlockData<'tcx>>,
171         source_scopes: IndexVec<SourceScope, SourceScopeData>,
172         source_scope_local_data: ClearCrossCrate<IndexVec<SourceScope, SourceScopeLocalData>>,
173         yield_ty: Option<Ty<'tcx>>,
174         local_decls: LocalDecls<'tcx>,
175         user_type_annotations: CanonicalUserTypeAnnotations<'tcx>,
176         arg_count: usize,
177         __upvar_debuginfo_codegen_only_do_not_use: Vec<UpvarDebuginfo>,
178         span: Span,
179         control_flow_destroyed: Vec<(Span, String)>,
180     ) -> Self {
181         // We need `arg_count` locals, and one for the return place.
182         assert!(
183             local_decls.len() >= arg_count + 1,
184             "expected at least {} locals, got {}",
185             arg_count + 1,
186             local_decls.len()
187         );
188
189         Body {
190             phase: MirPhase::Build,
191             basic_blocks,
192             source_scopes,
193             source_scope_local_data,
194             yield_ty,
195             generator_drop: None,
196             generator_layout: None,
197             local_decls,
198             user_type_annotations,
199             arg_count,
200             __upvar_debuginfo_codegen_only_do_not_use,
201             spread_arg: None,
202             span,
203             cache: cache::Cache::new(),
204             control_flow_destroyed,
205         }
206     }
207
208     #[inline]
209     pub fn basic_blocks(&self) -> &IndexVec<BasicBlock, BasicBlockData<'tcx>> {
210         &self.basic_blocks
211     }
212
213     #[inline]
214     pub fn basic_blocks_mut(&mut self) -> &mut IndexVec<BasicBlock, BasicBlockData<'tcx>> {
215         self.cache.invalidate();
216         &mut self.basic_blocks
217     }
218
219     #[inline]
220     pub fn basic_blocks_and_local_decls_mut(
221         &mut self,
222     ) -> (&mut IndexVec<BasicBlock, BasicBlockData<'tcx>>, &mut LocalDecls<'tcx>) {
223         self.cache.invalidate();
224         (&mut self.basic_blocks, &mut self.local_decls)
225     }
226
227     #[inline]
228     pub fn predecessors(&self) -> MappedReadGuard<'_, IndexVec<BasicBlock, Vec<BasicBlock>>> {
229         self.cache.predecessors(self)
230     }
231
232     #[inline]
233     pub fn predecessors_for(&self, bb: BasicBlock) -> MappedReadGuard<'_, Vec<BasicBlock>> {
234         MappedReadGuard::map(self.predecessors(), |p| &p[bb])
235     }
236
237     #[inline]
238     pub fn predecessor_locations(&self, loc: Location) -> impl Iterator<Item = Location> + '_ {
239         let if_zero_locations = if loc.statement_index == 0 {
240             let predecessor_blocks = self.predecessors_for(loc.block);
241             let num_predecessor_blocks = predecessor_blocks.len();
242             Some(
243                 (0..num_predecessor_blocks)
244                     .map(move |i| predecessor_blocks[i])
245                     .map(move |bb| self.terminator_loc(bb)),
246             )
247         } else {
248             None
249         };
250
251         let if_not_zero_locations = if loc.statement_index == 0 {
252             None
253         } else {
254             Some(Location { block: loc.block, statement_index: loc.statement_index - 1 })
255         };
256
257         if_zero_locations.into_iter().flatten().chain(if_not_zero_locations)
258     }
259
260     #[inline]
261     pub fn dominators(&self) -> Dominators<BasicBlock> {
262         dominators(self)
263     }
264
265     /// Returns `true` if a cycle exists in the control-flow graph that is reachable from the
266     /// `START_BLOCK`.
267     pub fn is_cfg_cyclic(&self) -> bool {
268         graph::is_cyclic(self)
269     }
270
271     #[inline]
272     pub fn local_kind(&self, local: Local) -> LocalKind {
273         let index = local.as_usize();
274         if index == 0 {
275             debug_assert!(
276                 self.local_decls[local].mutability == Mutability::Mut,
277                 "return place should be mutable"
278             );
279
280             LocalKind::ReturnPointer
281         } else if index < self.arg_count + 1 {
282             LocalKind::Arg
283         } else if self.local_decls[local].name.is_some() {
284             LocalKind::Var
285         } else {
286             LocalKind::Temp
287         }
288     }
289
290     /// Returns an iterator over all temporaries.
291     #[inline]
292     pub fn temps_iter<'a>(&'a self) -> impl Iterator<Item = Local> + 'a {
293         (self.arg_count + 1..self.local_decls.len()).filter_map(move |index| {
294             let local = Local::new(index);
295             if self.local_decls[local].is_user_variable.is_some() {
296                 None
297             } else {
298                 Some(local)
299             }
300         })
301     }
302
303     /// Returns an iterator over all user-declared locals.
304     #[inline]
305     pub fn vars_iter<'a>(&'a self) -> impl Iterator<Item = Local> + 'a {
306         (self.arg_count + 1..self.local_decls.len()).filter_map(move |index| {
307             let local = Local::new(index);
308             if self.local_decls[local].is_user_variable.is_some() {
309                 Some(local)
310             } else {
311                 None
312             }
313         })
314     }
315
316     /// Returns an iterator over all user-declared mutable locals.
317     #[inline]
318     pub fn mut_vars_iter<'a>(&'a self) -> impl Iterator<Item = Local> + 'a {
319         (self.arg_count + 1..self.local_decls.len()).filter_map(move |index| {
320             let local = Local::new(index);
321             let decl = &self.local_decls[local];
322             if decl.is_user_variable.is_some() && decl.mutability == Mutability::Mut {
323                 Some(local)
324             } else {
325                 None
326             }
327         })
328     }
329
330     /// Returns an iterator over all user-declared mutable arguments and locals.
331     #[inline]
332     pub fn mut_vars_and_args_iter<'a>(&'a self) -> impl Iterator<Item = Local> + 'a {
333         (1..self.local_decls.len()).filter_map(move |index| {
334             let local = Local::new(index);
335             let decl = &self.local_decls[local];
336             if (decl.is_user_variable.is_some() || index < self.arg_count + 1)
337                 && decl.mutability == Mutability::Mut
338             {
339                 Some(local)
340             } else {
341                 None
342             }
343         })
344     }
345
346     /// Returns an iterator over all function arguments.
347     #[inline]
348     pub fn args_iter(&self) -> impl Iterator<Item = Local> {
349         let arg_count = self.arg_count;
350         (1..=arg_count).map(Local::new)
351     }
352
353     /// Returns an iterator over all user-defined variables and compiler-generated temporaries (all
354     /// locals that are neither arguments nor the return place).
355     #[inline]
356     pub fn vars_and_temps_iter(&self) -> impl Iterator<Item = Local> {
357         let arg_count = self.arg_count;
358         let local_count = self.local_decls.len();
359         (arg_count + 1..local_count).map(Local::new)
360     }
361
362     /// Changes a statement to a nop. This is both faster than deleting instructions and avoids
363     /// invalidating statement indices in `Location`s.
364     pub fn make_statement_nop(&mut self, location: Location) {
365         let block = &mut self[location.block];
366         debug_assert!(location.statement_index < block.statements.len());
367         block.statements[location.statement_index].make_nop()
368     }
369
370     /// Returns the source info associated with `location`.
371     pub fn source_info(&self, location: Location) -> &SourceInfo {
372         let block = &self[location.block];
373         let stmts = &block.statements;
374         let idx = location.statement_index;
375         if idx < stmts.len() {
376             &stmts[idx].source_info
377         } else {
378             assert_eq!(idx, stmts.len());
379             &block.terminator().source_info
380         }
381     }
382
383     /// Checks if `sub` is a sub scope of `sup`
384     pub fn is_sub_scope(&self, mut sub: SourceScope, sup: SourceScope) -> bool {
385         while sub != sup {
386             match self.source_scopes[sub].parent_scope {
387                 None => return false,
388                 Some(p) => sub = p,
389             }
390         }
391         true
392     }
393
394     /// Returns the return type; it always return first element from `local_decls` array.
395     pub fn return_ty(&self) -> Ty<'tcx> {
396         self.local_decls[RETURN_PLACE].ty
397     }
398
399     /// Gets the location of the terminator for the given block.
400     pub fn terminator_loc(&self, bb: BasicBlock) -> Location {
401         Location { block: bb, statement_index: self[bb].statements.len() }
402     }
403 }
404
405 #[derive(Copy, Clone, Debug, RustcEncodable, RustcDecodable, HashStable)]
406 pub enum Safety {
407     Safe,
408     /// Unsafe because of a PushUnsafeBlock
409     BuiltinUnsafe,
410     /// Unsafe because of an unsafe fn
411     FnUnsafe,
412     /// Unsafe because of an `unsafe` block
413     ExplicitUnsafe(hir::HirId),
414 }
415
416 impl_stable_hash_for!(struct Body<'tcx> {
417     phase,
418     basic_blocks,
419     source_scopes,
420     source_scope_local_data,
421     yield_ty,
422     generator_drop,
423     generator_layout,
424     local_decls,
425     user_type_annotations,
426     arg_count,
427     __upvar_debuginfo_codegen_only_do_not_use,
428     spread_arg,
429     control_flow_destroyed,
430     span,
431     cache
432 });
433
434 impl<'tcx> Index<BasicBlock> for Body<'tcx> {
435     type Output = BasicBlockData<'tcx>;
436
437     #[inline]
438     fn index(&self, index: BasicBlock) -> &BasicBlockData<'tcx> {
439         &self.basic_blocks()[index]
440     }
441 }
442
443 impl<'tcx> IndexMut<BasicBlock> for Body<'tcx> {
444     #[inline]
445     fn index_mut(&mut self, index: BasicBlock) -> &mut BasicBlockData<'tcx> {
446         &mut self.basic_blocks_mut()[index]
447     }
448 }
449
450 #[derive(Copy, Clone, Debug, HashStable)]
451 pub enum ClearCrossCrate<T> {
452     Clear,
453     Set(T),
454 }
455
456 impl<T> ClearCrossCrate<T> {
457     pub fn assert_crate_local(self) -> T {
458         match self {
459             ClearCrossCrate::Clear => bug!("unwrapping cross-crate data"),
460             ClearCrossCrate::Set(v) => v,
461         }
462     }
463 }
464
465 impl<T: Encodable> rustc_serialize::UseSpecializedEncodable for ClearCrossCrate<T> {}
466 impl<T: Decodable> rustc_serialize::UseSpecializedDecodable for ClearCrossCrate<T> {}
467
468 /// Grouped information about the source code origin of a MIR entity.
469 /// Intended to be inspected by diagnostics and debuginfo.
470 /// Most passes can work with it as a whole, within a single function.
471 #[derive(Copy, Clone, Debug, PartialEq, Eq, RustcEncodable, RustcDecodable, Hash, HashStable)]
472 pub struct SourceInfo {
473     /// The source span for the AST pertaining to this MIR entity.
474     pub span: Span,
475
476     /// The source scope, keeping track of which bindings can be
477     /// seen by debuginfo, active lint levels, `unsafe {...}`, etc.
478     pub scope: SourceScope,
479 }
480
481 ///////////////////////////////////////////////////////////////////////////
482 // Mutability and borrow kinds
483
484 #[derive(Copy, Clone, Debug, PartialEq, Eq, RustcEncodable, RustcDecodable, HashStable)]
485 pub enum Mutability {
486     Mut,
487     Not,
488 }
489
490 impl From<Mutability> for hir::Mutability {
491     fn from(m: Mutability) -> Self {
492         match m {
493             Mutability::Mut => hir::MutMutable,
494             Mutability::Not => hir::MutImmutable,
495         }
496     }
497 }
498
499 #[derive(
500     Copy, Clone, Debug, PartialEq, Eq, PartialOrd, Ord, RustcEncodable, RustcDecodable, HashStable,
501 )]
502 pub enum BorrowKind {
503     /// Data must be immutable and is aliasable.
504     Shared,
505
506     /// The immediately borrowed place must be immutable, but projections from
507     /// it don't need to be. For example, a shallow borrow of `a.b` doesn't
508     /// conflict with a mutable borrow of `a.b.c`.
509     ///
510     /// This is used when lowering matches: when matching on a place we want to
511     /// ensure that place have the same value from the start of the match until
512     /// an arm is selected. This prevents this code from compiling:
513     ///
514     ///     let mut x = &Some(0);
515     ///     match *x {
516     ///         None => (),
517     ///         Some(_) if { x = &None; false } => (),
518     ///         Some(_) => (),
519     ///     }
520     ///
521     /// This can't be a shared borrow because mutably borrowing (*x as Some).0
522     /// should not prevent `if let None = x { ... }`, for example, because the
523     /// mutating `(*x as Some).0` can't affect the discriminant of `x`.
524     /// We can also report errors with this kind of borrow differently.
525     Shallow,
526
527     /// Data must be immutable but not aliasable. This kind of borrow
528     /// cannot currently be expressed by the user and is used only in
529     /// implicit closure bindings. It is needed when the closure is
530     /// borrowing or mutating a mutable referent, e.g.:
531     ///
532     ///     let x: &mut isize = ...;
533     ///     let y = || *x += 5;
534     ///
535     /// If we were to try to translate this closure into a more explicit
536     /// form, we'd encounter an error with the code as written:
537     ///
538     ///     struct Env { x: & &mut isize }
539     ///     let x: &mut isize = ...;
540     ///     let y = (&mut Env { &x }, fn_ptr);  // Closure is pair of env and fn
541     ///     fn fn_ptr(env: &mut Env) { **env.x += 5; }
542     ///
543     /// This is then illegal because you cannot mutate an `&mut` found
544     /// in an aliasable location. To solve, you'd have to translate with
545     /// an `&mut` borrow:
546     ///
547     ///     struct Env { x: & &mut isize }
548     ///     let x: &mut isize = ...;
549     ///     let y = (&mut Env { &mut x }, fn_ptr); // changed from &x to &mut x
550     ///     fn fn_ptr(env: &mut Env) { **env.x += 5; }
551     ///
552     /// Now the assignment to `**env.x` is legal, but creating a
553     /// mutable pointer to `x` is not because `x` is not mutable. We
554     /// could fix this by declaring `x` as `let mut x`. This is ok in
555     /// user code, if awkward, but extra weird for closures, since the
556     /// borrow is hidden.
557     ///
558     /// So we introduce a "unique imm" borrow -- the referent is
559     /// immutable, but not aliasable. This solves the problem. For
560     /// simplicity, we don't give users the way to express this
561     /// borrow, it's just used when translating closures.
562     Unique,
563
564     /// Data is mutable and not aliasable.
565     Mut {
566         /// `true` if this borrow arose from method-call auto-ref
567         /// (i.e., `adjustment::Adjust::Borrow`).
568         allow_two_phase_borrow: bool,
569     },
570 }
571
572 impl BorrowKind {
573     pub fn allows_two_phase_borrow(&self) -> bool {
574         match *self {
575             BorrowKind::Shared | BorrowKind::Shallow | BorrowKind::Unique => false,
576             BorrowKind::Mut { allow_two_phase_borrow } => allow_two_phase_borrow,
577         }
578     }
579 }
580
581 ///////////////////////////////////////////////////////////////////////////
582 // Variables and temps
583
584 rustc_index::newtype_index! {
585     pub struct Local {
586         derive [HashStable]
587         DEBUG_FORMAT = "_{}",
588         const RETURN_PLACE = 0,
589     }
590 }
591
592 impl Atom for Local {
593     fn index(self) -> usize {
594         Idx::index(self)
595     }
596 }
597
598 /// Classifies locals into categories. See `Body::local_kind`.
599 #[derive(PartialEq, Eq, Debug, HashStable)]
600 pub enum LocalKind {
601     /// User-declared variable binding.
602     Var,
603     /// Compiler-introduced temporary.
604     Temp,
605     /// Function argument.
606     Arg,
607     /// Location of function's return value.
608     ReturnPointer,
609 }
610
611 #[derive(Clone, PartialEq, Eq, Hash, Debug, RustcEncodable, RustcDecodable)]
612 pub struct VarBindingForm<'tcx> {
613     /// Is variable bound via `x`, `mut x`, `ref x`, or `ref mut x`?
614     pub binding_mode: ty::BindingMode,
615     /// If an explicit type was provided for this variable binding,
616     /// this holds the source Span of that type.
617     ///
618     /// NOTE: if you want to change this to a `HirId`, be wary that
619     /// doing so breaks incremental compilation (as of this writing),
620     /// while a `Span` does not cause our tests to fail.
621     pub opt_ty_info: Option<Span>,
622     /// Place of the RHS of the =, or the subject of the `match` where this
623     /// variable is initialized. None in the case of `let PATTERN;`.
624     /// Some((None, ..)) in the case of and `let [mut] x = ...` because
625     /// (a) the right-hand side isn't evaluated as a place expression.
626     /// (b) it gives a way to separate this case from the remaining cases
627     ///     for diagnostics.
628     pub opt_match_place: Option<(Option<Place<'tcx>>, Span)>,
629     /// The span of the pattern in which this variable was bound.
630     pub pat_span: Span,
631 }
632
633 #[derive(Clone, PartialEq, Eq, Hash, Debug, RustcEncodable, RustcDecodable)]
634 pub enum BindingForm<'tcx> {
635     /// This is a binding for a non-`self` binding, or a `self` that has an explicit type.
636     Var(VarBindingForm<'tcx>),
637     /// Binding for a `self`/`&self`/`&mut self` binding where the type is implicit.
638     ImplicitSelf(ImplicitSelfKind),
639     /// Reference used in a guard expression to ensure immutability.
640     RefForGuard,
641 }
642
643 /// Represents what type of implicit self a function has, if any.
644 #[derive(Clone, Copy, PartialEq, Eq, Hash, Debug, RustcEncodable, RustcDecodable)]
645 pub enum ImplicitSelfKind {
646     /// Represents a `fn x(self);`.
647     Imm,
648     /// Represents a `fn x(mut self);`.
649     Mut,
650     /// Represents a `fn x(&self);`.
651     ImmRef,
652     /// Represents a `fn x(&mut self);`.
653     MutRef,
654     /// Represents when a function does not have a self argument or
655     /// when a function has a `self: X` argument.
656     None,
657 }
658
659 CloneTypeFoldableAndLiftImpls! { BindingForm<'tcx>, }
660
661 impl_stable_hash_for!(struct self::VarBindingForm<'tcx> {
662     binding_mode,
663     opt_ty_info,
664     opt_match_place,
665     pat_span
666 });
667
668 impl_stable_hash_for!(enum self::ImplicitSelfKind {
669     Imm,
670     Mut,
671     ImmRef,
672     MutRef,
673     None
674 });
675
676 impl_stable_hash_for!(enum self::MirPhase {
677     Build,
678     Const,
679     Validated,
680     Optimized,
681 });
682
683 mod binding_form_impl {
684     use crate::ich::StableHashingContext;
685     use rustc_data_structures::stable_hasher::{HashStable, StableHasher};
686
687     impl<'a, 'tcx> HashStable<StableHashingContext<'a>> for super::BindingForm<'tcx> {
688         fn hash_stable(&self, hcx: &mut StableHashingContext<'a>, hasher: &mut StableHasher) {
689             use super::BindingForm::*;
690             ::std::mem::discriminant(self).hash_stable(hcx, hasher);
691
692             match self {
693                 Var(binding) => binding.hash_stable(hcx, hasher),
694                 ImplicitSelf(kind) => kind.hash_stable(hcx, hasher),
695                 RefForGuard => (),
696             }
697         }
698     }
699 }
700
701 /// `BlockTailInfo` is attached to the `LocalDecl` for temporaries
702 /// created during evaluation of expressions in a block tail
703 /// expression; that is, a block like `{ STMT_1; STMT_2; EXPR }`.
704 ///
705 /// It is used to improve diagnostics when such temporaries are
706 /// involved in borrow_check errors, e.g., explanations of where the
707 /// temporaries come from, when their destructors are run, and/or how
708 /// one might revise the code to satisfy the borrow checker's rules.
709 #[derive(Clone, Debug, RustcEncodable, RustcDecodable)]
710 pub struct BlockTailInfo {
711     /// If `true`, then the value resulting from evaluating this tail
712     /// expression is ignored by the block's expression context.
713     ///
714     /// Examples include `{ ...; tail };` and `let _ = { ...; tail };`
715     /// but not e.g., `let _x = { ...; tail };`
716     pub tail_result_is_ignored: bool,
717 }
718
719 impl_stable_hash_for!(struct BlockTailInfo { tail_result_is_ignored });
720
721 /// A MIR local.
722 ///
723 /// This can be a binding declared by the user, a temporary inserted by the compiler, a function
724 /// argument, or the return place.
725 #[derive(Clone, Debug, RustcEncodable, RustcDecodable, HashStable)]
726 pub struct LocalDecl<'tcx> {
727     /// Whether this is a mutable minding (i.e., `let x` or `let mut x`).
728     ///
729     /// Temporaries and the return place are always mutable.
730     pub mutability: Mutability,
731
732     /// `Some(binding_mode)` if this corresponds to a user-declared local variable.
733     ///
734     /// This is solely used for local diagnostics when generating
735     /// warnings/errors when compiling the current crate, and
736     /// therefore it need not be visible across crates. pnkfelix
737     /// currently hypothesized we *need* to wrap this in a
738     /// `ClearCrossCrate` as long as it carries as `HirId`.
739     pub is_user_variable: Option<ClearCrossCrate<BindingForm<'tcx>>>,
740
741     /// `true` if this is an internal local.
742     ///
743     /// These locals are not based on types in the source code and are only used
744     /// for a few desugarings at the moment.
745     ///
746     /// The generator transformation will sanity check the locals which are live
747     /// across a suspension point against the type components of the generator
748     /// which type checking knows are live across a suspension point. We need to
749     /// flag drop flags to avoid triggering this check as they are introduced
750     /// after typeck.
751     ///
752     /// Unsafety checking will also ignore dereferences of these locals,
753     /// so they can be used for raw pointers only used in a desugaring.
754     ///
755     /// This should be sound because the drop flags are fully algebraic, and
756     /// therefore don't affect the OIBIT or outlives properties of the
757     /// generator.
758     pub internal: bool,
759
760     /// If this local is a temporary and `is_block_tail` is `Some`,
761     /// then it is a temporary created for evaluation of some
762     /// subexpression of some block's tail expression (with no
763     /// intervening statement context).
764     pub is_block_tail: Option<BlockTailInfo>,
765
766     /// The type of this local.
767     pub ty: Ty<'tcx>,
768
769     /// If the user manually ascribed a type to this variable,
770     /// e.g., via `let x: T`, then we carry that type here. The MIR
771     /// borrow checker needs this information since it can affect
772     /// region inference.
773     pub user_ty: UserTypeProjections,
774
775     /// The name of the local, used in debuginfo and pretty-printing.
776     ///
777     /// Note that function arguments can also have this set to `Some(_)`
778     /// to generate better debuginfo.
779     pub name: Option<Name>,
780
781     /// The *syntactic* (i.e., not visibility) source scope the local is defined
782     /// in. If the local was defined in a let-statement, this
783     /// is *within* the let-statement, rather than outside
784     /// of it.
785     ///
786     /// This is needed because the visibility source scope of locals within
787     /// a let-statement is weird.
788     ///
789     /// The reason is that we want the local to be *within* the let-statement
790     /// for lint purposes, but we want the local to be *after* the let-statement
791     /// for names-in-scope purposes.
792     ///
793     /// That's it, if we have a let-statement like the one in this
794     /// function:
795     ///
796     /// ```
797     /// fn foo(x: &str) {
798     ///     #[allow(unused_mut)]
799     ///     let mut x: u32 = { // <- one unused mut
800     ///         let mut y: u32 = x.parse().unwrap();
801     ///         y + 2
802     ///     };
803     ///     drop(x);
804     /// }
805     /// ```
806     ///
807     /// Then, from a lint point of view, the declaration of `x: u32`
808     /// (and `y: u32`) are within the `#[allow(unused_mut)]` scope - the
809     /// lint scopes are the same as the AST/HIR nesting.
810     ///
811     /// However, from a name lookup point of view, the scopes look more like
812     /// as if the let-statements were `match` expressions:
813     ///
814     /// ```
815     /// fn foo(x: &str) {
816     ///     match {
817     ///         match x.parse().unwrap() {
818     ///             y => y + 2
819     ///         }
820     ///     } {
821     ///         x => drop(x)
822     ///     };
823     /// }
824     /// ```
825     ///
826     /// We care about the name-lookup scopes for debuginfo - if the
827     /// debuginfo instruction pointer is at the call to `x.parse()`, we
828     /// want `x` to refer to `x: &str`, but if it is at the call to
829     /// `drop(x)`, we want it to refer to `x: u32`.
830     ///
831     /// To allow both uses to work, we need to have more than a single scope
832     /// for a local. We have the `source_info.scope` represent the
833     /// "syntactic" lint scope (with a variable being under its let
834     /// block) while the `visibility_scope` represents the "local variable"
835     /// scope (where the "rest" of a block is under all prior let-statements).
836     ///
837     /// The end result looks like this:
838     ///
839     /// ```text
840     /// ROOT SCOPE
841     ///  │{ argument x: &str }
842     ///  │
843     ///  │ │{ #[allow(unused_mut)] } // This is actually split into 2 scopes
844     ///  │ │                         // in practice because I'm lazy.
845     ///  │ │
846     ///  │ │← x.source_info.scope
847     ///  │ │← `x.parse().unwrap()`
848     ///  │ │
849     ///  │ │ │← y.source_info.scope
850     ///  │ │
851     ///  │ │ │{ let y: u32 }
852     ///  │ │ │
853     ///  │ │ │← y.visibility_scope
854     ///  │ │ │← `y + 2`
855     ///  │
856     ///  │ │{ let x: u32 }
857     ///  │ │← x.visibility_scope
858     ///  │ │← `drop(x)` // This accesses `x: u32`.
859     /// ```
860     pub source_info: SourceInfo,
861
862     /// Source scope within which the local is visible (for debuginfo)
863     /// (see `source_info` for more details).
864     pub visibility_scope: SourceScope,
865 }
866
867 impl<'tcx> LocalDecl<'tcx> {
868     /// Returns `true` only if local is a binding that can itself be
869     /// made mutable via the addition of the `mut` keyword, namely
870     /// something like the occurrences of `x` in:
871     /// - `fn foo(x: Type) { ... }`,
872     /// - `let x = ...`,
873     /// - or `match ... { C(x) => ... }`
874     pub fn can_be_made_mutable(&self) -> bool {
875         match self.is_user_variable {
876             Some(ClearCrossCrate::Set(BindingForm::Var(VarBindingForm {
877                 binding_mode: ty::BindingMode::BindByValue(_),
878                 opt_ty_info: _,
879                 opt_match_place: _,
880                 pat_span: _,
881             }))) => true,
882
883             Some(ClearCrossCrate::Set(BindingForm::ImplicitSelf(ImplicitSelfKind::Imm))) => true,
884
885             _ => false,
886         }
887     }
888
889     /// Returns `true` if local is definitely not a `ref ident` or
890     /// `ref mut ident` binding. (Such bindings cannot be made into
891     /// mutable bindings, but the inverse does not necessarily hold).
892     pub fn is_nonref_binding(&self) -> bool {
893         match self.is_user_variable {
894             Some(ClearCrossCrate::Set(BindingForm::Var(VarBindingForm {
895                 binding_mode: ty::BindingMode::BindByValue(_),
896                 opt_ty_info: _,
897                 opt_match_place: _,
898                 pat_span: _,
899             }))) => true,
900
901             Some(ClearCrossCrate::Set(BindingForm::ImplicitSelf(_))) => true,
902
903             _ => false,
904         }
905     }
906
907     /// Returns `true` if this is a reference to a variable bound in a `match`
908     /// expression that is used to access said variable for the guard of the
909     /// match arm.
910     pub fn is_ref_for_guard(&self) -> bool {
911         match self.is_user_variable {
912             Some(ClearCrossCrate::Set(BindingForm::RefForGuard)) => true,
913             _ => false,
914         }
915     }
916
917     /// Returns `true` is the local is from a compiler desugaring, e.g.,
918     /// `__next` from a `for` loop.
919     #[inline]
920     pub fn from_compiler_desugaring(&self) -> bool {
921         self.source_info.span.desugaring_kind().is_some()
922     }
923
924     /// Creates a new `LocalDecl` for a temporary.
925     #[inline]
926     pub fn new_temp(ty: Ty<'tcx>, span: Span) -> Self {
927         Self::new_local(ty, Mutability::Mut, false, span)
928     }
929
930     /// Converts `self` into same `LocalDecl` except tagged as immutable.
931     #[inline]
932     pub fn immutable(mut self) -> Self {
933         self.mutability = Mutability::Not;
934         self
935     }
936
937     /// Converts `self` into same `LocalDecl` except tagged as internal temporary.
938     #[inline]
939     pub fn block_tail(mut self, info: BlockTailInfo) -> Self {
940         assert!(self.is_block_tail.is_none());
941         self.is_block_tail = Some(info);
942         self
943     }
944
945     /// Creates a new `LocalDecl` for a internal temporary.
946     #[inline]
947     pub fn new_internal(ty: Ty<'tcx>, span: Span) -> Self {
948         Self::new_local(ty, Mutability::Mut, true, span)
949     }
950
951     #[inline]
952     fn new_local(ty: Ty<'tcx>, mutability: Mutability, internal: bool, span: Span) -> Self {
953         LocalDecl {
954             mutability,
955             ty,
956             user_ty: UserTypeProjections::none(),
957             name: None,
958             source_info: SourceInfo { span, scope: OUTERMOST_SOURCE_SCOPE },
959             visibility_scope: OUTERMOST_SOURCE_SCOPE,
960             internal,
961             is_user_variable: None,
962             is_block_tail: None,
963         }
964     }
965
966     /// Builds a `LocalDecl` for the return place.
967     ///
968     /// This must be inserted into the `local_decls` list as the first local.
969     #[inline]
970     pub fn new_return_place(return_ty: Ty<'_>, span: Span) -> LocalDecl<'_> {
971         LocalDecl {
972             mutability: Mutability::Mut,
973             ty: return_ty,
974             user_ty: UserTypeProjections::none(),
975             source_info: SourceInfo { span, scope: OUTERMOST_SOURCE_SCOPE },
976             visibility_scope: OUTERMOST_SOURCE_SCOPE,
977             internal: false,
978             is_block_tail: None,
979             name: None, // FIXME maybe we do want some name here?
980             is_user_variable: None,
981         }
982     }
983 }
984
985 /// A closure capture, with its name and mode.
986 #[derive(Clone, Debug, RustcEncodable, RustcDecodable, HashStable)]
987 pub struct UpvarDebuginfo {
988     pub debug_name: Name,
989
990     /// If true, the capture is behind a reference.
991     pub by_ref: bool,
992 }
993
994 ///////////////////////////////////////////////////////////////////////////
995 // BasicBlock
996
997 rustc_index::newtype_index! {
998     pub struct BasicBlock {
999         derive [HashStable]
1000         DEBUG_FORMAT = "bb{}",
1001         const START_BLOCK = 0,
1002     }
1003 }
1004
1005 impl BasicBlock {
1006     pub fn start_location(self) -> Location {
1007         Location { block: self, statement_index: 0 }
1008     }
1009 }
1010
1011 ///////////////////////////////////////////////////////////////////////////
1012 // BasicBlockData and Terminator
1013
1014 #[derive(Clone, Debug, RustcEncodable, RustcDecodable, HashStable)]
1015 pub struct BasicBlockData<'tcx> {
1016     /// List of statements in this block.
1017     pub statements: Vec<Statement<'tcx>>,
1018
1019     /// Terminator for this block.
1020     ///
1021     /// N.B., this should generally ONLY be `None` during construction.
1022     /// Therefore, you should generally access it via the
1023     /// `terminator()` or `terminator_mut()` methods. The only
1024     /// exception is that certain passes, such as `simplify_cfg`, swap
1025     /// out the terminator temporarily with `None` while they continue
1026     /// to recurse over the set of basic blocks.
1027     pub terminator: Option<Terminator<'tcx>>,
1028
1029     /// If true, this block lies on an unwind path. This is used
1030     /// during codegen where distinct kinds of basic blocks may be
1031     /// generated (particularly for MSVC cleanup). Unwind blocks must
1032     /// only branch to other unwind blocks.
1033     pub is_cleanup: bool,
1034 }
1035
1036 #[derive(Clone, Debug, RustcEncodable, RustcDecodable, HashStable)]
1037 pub struct Terminator<'tcx> {
1038     pub source_info: SourceInfo,
1039     pub kind: TerminatorKind<'tcx>,
1040 }
1041
1042 #[derive(Clone, RustcEncodable, RustcDecodable, HashStable)]
1043 pub enum TerminatorKind<'tcx> {
1044     /// Block should have one successor in the graph; we jump there.
1045     Goto { target: BasicBlock },
1046
1047     /// Operand evaluates to an integer; jump depending on its value
1048     /// to one of the targets, and otherwise fallback to `otherwise`.
1049     SwitchInt {
1050         /// The discriminant value being tested.
1051         discr: Operand<'tcx>,
1052
1053         /// The type of value being tested.
1054         switch_ty: Ty<'tcx>,
1055
1056         /// Possible values. The locations to branch to in each case
1057         /// are found in the corresponding indices from the `targets` vector.
1058         values: Cow<'tcx, [u128]>,
1059
1060         /// Possible branch sites. The last element of this vector is used
1061         /// for the otherwise branch, so targets.len() == values.len() + 1
1062         /// should hold.
1063         //
1064         // This invariant is quite non-obvious and also could be improved.
1065         // One way to make this invariant is to have something like this instead:
1066         //
1067         // branches: Vec<(ConstInt, BasicBlock)>,
1068         // otherwise: Option<BasicBlock> // exhaustive if None
1069         //
1070         // However we’ve decided to keep this as-is until we figure a case
1071         // where some other approach seems to be strictly better than other.
1072         targets: Vec<BasicBlock>,
1073     },
1074
1075     /// Indicates that the landing pad is finished and unwinding should
1076     /// continue. Emitted by `build::scope::diverge_cleanup`.
1077     Resume,
1078
1079     /// Indicates that the landing pad is finished and that the process
1080     /// should abort. Used to prevent unwinding for foreign items.
1081     Abort,
1082
1083     /// Indicates a normal return. The return place should have
1084     /// been filled in by now. This should occur at most once.
1085     Return,
1086
1087     /// Indicates a terminator that can never be reached.
1088     Unreachable,
1089
1090     /// Drop the `Place`.
1091     Drop { location: Place<'tcx>, target: BasicBlock, unwind: Option<BasicBlock> },
1092
1093     /// Drop the `Place` and assign the new value over it. This ensures
1094     /// that the assignment to `P` occurs *even if* the destructor for
1095     /// place unwinds. Its semantics are best explained by the
1096     /// elaboration:
1097     ///
1098     /// ```
1099     /// BB0 {
1100     ///   DropAndReplace(P <- V, goto BB1, unwind BB2)
1101     /// }
1102     /// ```
1103     ///
1104     /// becomes
1105     ///
1106     /// ```
1107     /// BB0 {
1108     ///   Drop(P, goto BB1, unwind BB2)
1109     /// }
1110     /// BB1 {
1111     ///   // P is now uninitialized
1112     ///   P <- V
1113     /// }
1114     /// BB2 {
1115     ///   // P is now uninitialized -- its dtor panicked
1116     ///   P <- V
1117     /// }
1118     /// ```
1119     DropAndReplace {
1120         location: Place<'tcx>,
1121         value: Operand<'tcx>,
1122         target: BasicBlock,
1123         unwind: Option<BasicBlock>,
1124     },
1125
1126     /// Block ends with a call of a converging function.
1127     Call {
1128         /// The function that’s being called.
1129         func: Operand<'tcx>,
1130         /// Arguments the function is called with.
1131         /// These are owned by the callee, which is free to modify them.
1132         /// This allows the memory occupied by "by-value" arguments to be
1133         /// reused across function calls without duplicating the contents.
1134         args: Vec<Operand<'tcx>>,
1135         /// Destination for the return value. If some, the call is converging.
1136         destination: Option<(Place<'tcx>, BasicBlock)>,
1137         /// Cleanups to be done if the call unwinds.
1138         cleanup: Option<BasicBlock>,
1139         /// `true` if this is from a call in HIR rather than from an overloaded
1140         /// operator. True for overloaded function call.
1141         from_hir_call: bool,
1142     },
1143
1144     /// Jump to the target if the condition has the expected value,
1145     /// otherwise panic with a message and a cleanup target.
1146     Assert {
1147         cond: Operand<'tcx>,
1148         expected: bool,
1149         msg: AssertMessage<'tcx>,
1150         target: BasicBlock,
1151         cleanup: Option<BasicBlock>,
1152     },
1153
1154     /// A suspend point.
1155     Yield {
1156         /// The value to return.
1157         value: Operand<'tcx>,
1158         /// Where to resume to.
1159         resume: BasicBlock,
1160         /// Cleanup to be done if the generator is dropped at this suspend point.
1161         drop: Option<BasicBlock>,
1162     },
1163
1164     /// Indicates the end of the dropping of a generator.
1165     GeneratorDrop,
1166
1167     /// A block where control flow only ever takes one real path, but borrowck
1168     /// needs to be more conservative.
1169     FalseEdges {
1170         /// The target normal control flow will take.
1171         real_target: BasicBlock,
1172         /// A block control flow could conceptually jump to, but won't in
1173         /// practice.
1174         imaginary_target: BasicBlock,
1175     },
1176     /// A terminator for blocks that only take one path in reality, but where we
1177     /// reserve the right to unwind in borrowck, even if it won't happen in practice.
1178     /// This can arise in infinite loops with no function calls for example.
1179     FalseUnwind {
1180         /// The target normal control flow will take.
1181         real_target: BasicBlock,
1182         /// The imaginary cleanup block link. This particular path will never be taken
1183         /// in practice, but in order to avoid fragility we want to always
1184         /// consider it in borrowck. We don't want to accept programs which
1185         /// pass borrowck only when `panic=abort` or some assertions are disabled
1186         /// due to release vs. debug mode builds. This needs to be an `Option` because
1187         /// of the `remove_noop_landing_pads` and `no_landing_pads` passes.
1188         unwind: Option<BasicBlock>,
1189     },
1190 }
1191
1192 pub type Successors<'a> =
1193     iter::Chain<option::IntoIter<&'a BasicBlock>, slice::Iter<'a, BasicBlock>>;
1194 pub type SuccessorsMut<'a> =
1195     iter::Chain<option::IntoIter<&'a mut BasicBlock>, slice::IterMut<'a, BasicBlock>>;
1196
1197 impl<'tcx> Terminator<'tcx> {
1198     pub fn successors(&self) -> Successors<'_> {
1199         self.kind.successors()
1200     }
1201
1202     pub fn successors_mut(&mut self) -> SuccessorsMut<'_> {
1203         self.kind.successors_mut()
1204     }
1205
1206     pub fn unwind(&self) -> Option<&Option<BasicBlock>> {
1207         self.kind.unwind()
1208     }
1209
1210     pub fn unwind_mut(&mut self) -> Option<&mut Option<BasicBlock>> {
1211         self.kind.unwind_mut()
1212     }
1213 }
1214
1215 impl<'tcx> TerminatorKind<'tcx> {
1216     pub fn if_(
1217         tcx: TyCtxt<'tcx>,
1218         cond: Operand<'tcx>,
1219         t: BasicBlock,
1220         f: BasicBlock,
1221     ) -> TerminatorKind<'tcx> {
1222         static BOOL_SWITCH_FALSE: &'static [u128] = &[0];
1223         TerminatorKind::SwitchInt {
1224             discr: cond,
1225             switch_ty: tcx.types.bool,
1226             values: From::from(BOOL_SWITCH_FALSE),
1227             targets: vec![f, t],
1228         }
1229     }
1230
1231     pub fn successors(&self) -> Successors<'_> {
1232         use self::TerminatorKind::*;
1233         match *self {
1234             Resume
1235             | Abort
1236             | GeneratorDrop
1237             | Return
1238             | Unreachable
1239             | Call { destination: None, cleanup: None, .. } => None.into_iter().chain(&[]),
1240             Goto { target: ref t }
1241             | Call { destination: None, cleanup: Some(ref t), .. }
1242             | Call { destination: Some((_, ref t)), cleanup: None, .. }
1243             | Yield { resume: ref t, drop: None, .. }
1244             | DropAndReplace { target: ref t, unwind: None, .. }
1245             | Drop { target: ref t, unwind: None, .. }
1246             | Assert { target: ref t, cleanup: None, .. }
1247             | FalseUnwind { real_target: ref t, unwind: None } => Some(t).into_iter().chain(&[]),
1248             Call { destination: Some((_, ref t)), cleanup: Some(ref u), .. }
1249             | Yield { resume: ref t, drop: Some(ref u), .. }
1250             | DropAndReplace { target: ref t, unwind: Some(ref u), .. }
1251             | Drop { target: ref t, unwind: Some(ref u), .. }
1252             | Assert { target: ref t, cleanup: Some(ref u), .. }
1253             | FalseUnwind { real_target: ref t, unwind: Some(ref u) } => {
1254                 Some(t).into_iter().chain(slice::from_ref(u))
1255             }
1256             SwitchInt { ref targets, .. } => None.into_iter().chain(&targets[..]),
1257             FalseEdges { ref real_target, ref imaginary_target } => {
1258                 Some(real_target).into_iter().chain(slice::from_ref(imaginary_target))
1259             }
1260         }
1261     }
1262
1263     pub fn successors_mut(&mut self) -> SuccessorsMut<'_> {
1264         use self::TerminatorKind::*;
1265         match *self {
1266             Resume
1267             | Abort
1268             | GeneratorDrop
1269             | Return
1270             | Unreachable
1271             | Call { destination: None, cleanup: None, .. } => None.into_iter().chain(&mut []),
1272             Goto { target: ref mut t }
1273             | Call { destination: None, cleanup: Some(ref mut t), .. }
1274             | Call { destination: Some((_, ref mut t)), cleanup: None, .. }
1275             | Yield { resume: ref mut t, drop: None, .. }
1276             | DropAndReplace { target: ref mut t, unwind: None, .. }
1277             | Drop { target: ref mut t, unwind: None, .. }
1278             | Assert { target: ref mut t, cleanup: None, .. }
1279             | FalseUnwind { real_target: ref mut t, unwind: None } => {
1280                 Some(t).into_iter().chain(&mut [])
1281             }
1282             Call { destination: Some((_, ref mut t)), cleanup: Some(ref mut u), .. }
1283             | Yield { resume: ref mut t, drop: Some(ref mut u), .. }
1284             | DropAndReplace { target: ref mut t, unwind: Some(ref mut u), .. }
1285             | Drop { target: ref mut t, unwind: Some(ref mut u), .. }
1286             | Assert { target: ref mut t, cleanup: Some(ref mut u), .. }
1287             | FalseUnwind { real_target: ref mut t, unwind: Some(ref mut u) } => {
1288                 Some(t).into_iter().chain(slice::from_mut(u))
1289             }
1290             SwitchInt { ref mut targets, .. } => None.into_iter().chain(&mut targets[..]),
1291             FalseEdges { ref mut real_target, ref mut imaginary_target } => {
1292                 Some(real_target).into_iter().chain(slice::from_mut(imaginary_target))
1293             }
1294         }
1295     }
1296
1297     pub fn unwind(&self) -> Option<&Option<BasicBlock>> {
1298         match *self {
1299             TerminatorKind::Goto { .. }
1300             | TerminatorKind::Resume
1301             | TerminatorKind::Abort
1302             | TerminatorKind::Return
1303             | TerminatorKind::Unreachable
1304             | TerminatorKind::GeneratorDrop
1305             | TerminatorKind::Yield { .. }
1306             | TerminatorKind::SwitchInt { .. }
1307             | TerminatorKind::FalseEdges { .. } => None,
1308             TerminatorKind::Call { cleanup: ref unwind, .. }
1309             | TerminatorKind::Assert { cleanup: ref unwind, .. }
1310             | TerminatorKind::DropAndReplace { ref unwind, .. }
1311             | TerminatorKind::Drop { ref unwind, .. }
1312             | TerminatorKind::FalseUnwind { ref unwind, .. } => Some(unwind),
1313         }
1314     }
1315
1316     pub fn unwind_mut(&mut self) -> Option<&mut Option<BasicBlock>> {
1317         match *self {
1318             TerminatorKind::Goto { .. }
1319             | TerminatorKind::Resume
1320             | TerminatorKind::Abort
1321             | TerminatorKind::Return
1322             | TerminatorKind::Unreachable
1323             | TerminatorKind::GeneratorDrop
1324             | TerminatorKind::Yield { .. }
1325             | TerminatorKind::SwitchInt { .. }
1326             | TerminatorKind::FalseEdges { .. } => None,
1327             TerminatorKind::Call { cleanup: ref mut unwind, .. }
1328             | TerminatorKind::Assert { cleanup: ref mut unwind, .. }
1329             | TerminatorKind::DropAndReplace { ref mut unwind, .. }
1330             | TerminatorKind::Drop { ref mut unwind, .. }
1331             | TerminatorKind::FalseUnwind { ref mut unwind, .. } => Some(unwind),
1332         }
1333     }
1334 }
1335
1336 impl<'tcx> BasicBlockData<'tcx> {
1337     pub fn new(terminator: Option<Terminator<'tcx>>) -> BasicBlockData<'tcx> {
1338         BasicBlockData { statements: vec![], terminator, is_cleanup: false }
1339     }
1340
1341     /// Accessor for terminator.
1342     ///
1343     /// Terminator may not be None after construction of the basic block is complete. This accessor
1344     /// provides a convenience way to reach the terminator.
1345     pub fn terminator(&self) -> &Terminator<'tcx> {
1346         self.terminator.as_ref().expect("invalid terminator state")
1347     }
1348
1349     pub fn terminator_mut(&mut self) -> &mut Terminator<'tcx> {
1350         self.terminator.as_mut().expect("invalid terminator state")
1351     }
1352
1353     pub fn retain_statements<F>(&mut self, mut f: F)
1354     where
1355         F: FnMut(&mut Statement<'_>) -> bool,
1356     {
1357         for s in &mut self.statements {
1358             if !f(s) {
1359                 s.make_nop();
1360             }
1361         }
1362     }
1363
1364     pub fn expand_statements<F, I>(&mut self, mut f: F)
1365     where
1366         F: FnMut(&mut Statement<'tcx>) -> Option<I>,
1367         I: iter::TrustedLen<Item = Statement<'tcx>>,
1368     {
1369         // Gather all the iterators we'll need to splice in, and their positions.
1370         let mut splices: Vec<(usize, I)> = vec![];
1371         let mut extra_stmts = 0;
1372         for (i, s) in self.statements.iter_mut().enumerate() {
1373             if let Some(mut new_stmts) = f(s) {
1374                 if let Some(first) = new_stmts.next() {
1375                     // We can already store the first new statement.
1376                     *s = first;
1377
1378                     // Save the other statements for optimized splicing.
1379                     let remaining = new_stmts.size_hint().0;
1380                     if remaining > 0 {
1381                         splices.push((i + 1 + extra_stmts, new_stmts));
1382                         extra_stmts += remaining;
1383                     }
1384                 } else {
1385                     s.make_nop();
1386                 }
1387             }
1388         }
1389
1390         // Splice in the new statements, from the end of the block.
1391         // FIXME(eddyb) This could be more efficient with a "gap buffer"
1392         // where a range of elements ("gap") is left uninitialized, with
1393         // splicing adding new elements to the end of that gap and moving
1394         // existing elements from before the gap to the end of the gap.
1395         // For now, this is safe code, emulating a gap but initializing it.
1396         let mut gap = self.statements.len()..self.statements.len() + extra_stmts;
1397         self.statements.resize(
1398             gap.end,
1399             Statement {
1400                 source_info: SourceInfo { span: DUMMY_SP, scope: OUTERMOST_SOURCE_SCOPE },
1401                 kind: StatementKind::Nop,
1402             },
1403         );
1404         for (splice_start, new_stmts) in splices.into_iter().rev() {
1405             let splice_end = splice_start + new_stmts.size_hint().0;
1406             while gap.end > splice_end {
1407                 gap.start -= 1;
1408                 gap.end -= 1;
1409                 self.statements.swap(gap.start, gap.end);
1410             }
1411             self.statements.splice(splice_start..splice_end, new_stmts);
1412             gap.end = splice_start;
1413         }
1414     }
1415
1416     pub fn visitable(&self, index: usize) -> &dyn MirVisitable<'tcx> {
1417         if index < self.statements.len() {
1418             &self.statements[index]
1419         } else {
1420             &self.terminator
1421         }
1422     }
1423 }
1424
1425 impl<'tcx> Debug for TerminatorKind<'tcx> {
1426     fn fmt(&self, fmt: &mut Formatter<'_>) -> fmt::Result {
1427         self.fmt_head(fmt)?;
1428         let successor_count = self.successors().count();
1429         let labels = self.fmt_successor_labels();
1430         assert_eq!(successor_count, labels.len());
1431
1432         match successor_count {
1433             0 => Ok(()),
1434
1435             1 => write!(fmt, " -> {:?}", self.successors().nth(0).unwrap()),
1436
1437             _ => {
1438                 write!(fmt, " -> [")?;
1439                 for (i, target) in self.successors().enumerate() {
1440                     if i > 0 {
1441                         write!(fmt, ", ")?;
1442                     }
1443                     write!(fmt, "{}: {:?}", labels[i], target)?;
1444                 }
1445                 write!(fmt, "]")
1446             }
1447         }
1448     }
1449 }
1450
1451 impl<'tcx> TerminatorKind<'tcx> {
1452     /// Writes the "head" part of the terminator; that is, its name and the data it uses to pick the
1453     /// successor basic block, if any. The only information not included is the list of possible
1454     /// successors, which may be rendered differently between the text and the graphviz format.
1455     pub fn fmt_head<W: Write>(&self, fmt: &mut W) -> fmt::Result {
1456         use self::TerminatorKind::*;
1457         match *self {
1458             Goto { .. } => write!(fmt, "goto"),
1459             SwitchInt { discr: ref place, .. } => write!(fmt, "switchInt({:?})", place),
1460             Return => write!(fmt, "return"),
1461             GeneratorDrop => write!(fmt, "generator_drop"),
1462             Resume => write!(fmt, "resume"),
1463             Abort => write!(fmt, "abort"),
1464             Yield { ref value, .. } => write!(fmt, "_1 = suspend({:?})", value),
1465             Unreachable => write!(fmt, "unreachable"),
1466             Drop { ref location, .. } => write!(fmt, "drop({:?})", location),
1467             DropAndReplace { ref location, ref value, .. } => {
1468                 write!(fmt, "replace({:?} <- {:?})", location, value)
1469             }
1470             Call { ref func, ref args, ref destination, .. } => {
1471                 if let Some((ref destination, _)) = *destination {
1472                     write!(fmt, "{:?} = ", destination)?;
1473                 }
1474                 write!(fmt, "{:?}(", func)?;
1475                 for (index, arg) in args.iter().enumerate() {
1476                     if index > 0 {
1477                         write!(fmt, ", ")?;
1478                     }
1479                     write!(fmt, "{:?}", arg)?;
1480                 }
1481                 write!(fmt, ")")
1482             }
1483             Assert { ref cond, expected, ref msg, .. } => {
1484                 write!(fmt, "assert(")?;
1485                 if !expected {
1486                     write!(fmt, "!")?;
1487                 }
1488                 write!(fmt, "{:?}, \"{:?}\")", cond, msg)
1489             }
1490             FalseEdges { .. } => write!(fmt, "falseEdges"),
1491             FalseUnwind { .. } => write!(fmt, "falseUnwind"),
1492         }
1493     }
1494
1495     /// Returns the list of labels for the edges to the successor basic blocks.
1496     pub fn fmt_successor_labels(&self) -> Vec<Cow<'static, str>> {
1497         use self::TerminatorKind::*;
1498         match *self {
1499             Return | Resume | Abort | Unreachable | GeneratorDrop => vec![],
1500             Goto { .. } => vec!["".into()],
1501             SwitchInt { ref values, switch_ty, .. } => ty::tls::with(|tcx| {
1502                 let param_env = ty::ParamEnv::empty();
1503                 let switch_ty = tcx.lift(&switch_ty).unwrap();
1504                 let size = tcx.layout_of(param_env.and(switch_ty)).unwrap().size;
1505                 values
1506                     .iter()
1507                     .map(|&u| {
1508                         tcx.mk_const(ty::Const {
1509                             val: ConstValue::Scalar(Scalar::from_uint(u, size).into()),
1510                             ty: switch_ty,
1511                         })
1512                         .to_string()
1513                         .into()
1514                     })
1515                     .chain(iter::once("otherwise".into()))
1516                     .collect()
1517             }),
1518             Call { destination: Some(_), cleanup: Some(_), .. } => {
1519                 vec!["return".into(), "unwind".into()]
1520             }
1521             Call { destination: Some(_), cleanup: None, .. } => vec!["return".into()],
1522             Call { destination: None, cleanup: Some(_), .. } => vec!["unwind".into()],
1523             Call { destination: None, cleanup: None, .. } => vec![],
1524             Yield { drop: Some(_), .. } => vec!["resume".into(), "drop".into()],
1525             Yield { drop: None, .. } => vec!["resume".into()],
1526             DropAndReplace { unwind: None, .. } | Drop { unwind: None, .. } => {
1527                 vec!["return".into()]
1528             }
1529             DropAndReplace { unwind: Some(_), .. } | Drop { unwind: Some(_), .. } => {
1530                 vec!["return".into(), "unwind".into()]
1531             }
1532             Assert { cleanup: None, .. } => vec!["".into()],
1533             Assert { .. } => vec!["success".into(), "unwind".into()],
1534             FalseEdges { .. } => vec!["real".into(), "imaginary".into()],
1535             FalseUnwind { unwind: Some(_), .. } => vec!["real".into(), "cleanup".into()],
1536             FalseUnwind { unwind: None, .. } => vec!["real".into()],
1537         }
1538     }
1539 }
1540
1541 ///////////////////////////////////////////////////////////////////////////
1542 // Statements
1543
1544 #[derive(Clone, RustcEncodable, RustcDecodable, HashStable)]
1545 pub struct Statement<'tcx> {
1546     pub source_info: SourceInfo,
1547     pub kind: StatementKind<'tcx>,
1548 }
1549
1550 // `Statement` is used a lot. Make sure it doesn't unintentionally get bigger.
1551 #[cfg(target_arch = "x86_64")]
1552 static_assert_size!(Statement<'_>, 32);
1553
1554 impl Statement<'_> {
1555     /// Changes a statement to a nop. This is both faster than deleting instructions and avoids
1556     /// invalidating statement indices in `Location`s.
1557     pub fn make_nop(&mut self) {
1558         self.kind = StatementKind::Nop
1559     }
1560
1561     /// Changes a statement to a nop and returns the original statement.
1562     pub fn replace_nop(&mut self) -> Self {
1563         Statement {
1564             source_info: self.source_info,
1565             kind: mem::replace(&mut self.kind, StatementKind::Nop),
1566         }
1567     }
1568 }
1569
1570 #[derive(Clone, Debug, RustcEncodable, RustcDecodable, HashStable)]
1571 pub enum StatementKind<'tcx> {
1572     /// Write the RHS Rvalue to the LHS Place.
1573     Assign(Box<(Place<'tcx>, Rvalue<'tcx>)>),
1574
1575     /// This represents all the reading that a pattern match may do
1576     /// (e.g., inspecting constants and discriminant values), and the
1577     /// kind of pattern it comes from. This is in order to adapt potential
1578     /// error messages to these specific patterns.
1579     ///
1580     /// Note that this also is emitted for regular `let` bindings to ensure that locals that are
1581     /// never accessed still get some sanity checks for, e.g., `let x: ! = ..;`
1582     FakeRead(FakeReadCause, Box<Place<'tcx>>),
1583
1584     /// Write the discriminant for a variant to the enum Place.
1585     SetDiscriminant { place: Box<Place<'tcx>>, variant_index: VariantIdx },
1586
1587     /// Start a live range for the storage of the local.
1588     StorageLive(Local),
1589
1590     /// End the current live range for the storage of the local.
1591     StorageDead(Local),
1592
1593     /// Executes a piece of inline Assembly. Stored in a Box to keep the size
1594     /// of `StatementKind` low.
1595     InlineAsm(Box<InlineAsm<'tcx>>),
1596
1597     /// Retag references in the given place, ensuring they got fresh tags. This is
1598     /// part of the Stacked Borrows model. These statements are currently only interpreted
1599     /// by miri and only generated when "-Z mir-emit-retag" is passed.
1600     /// See <https://internals.rust-lang.org/t/stacked-borrows-an-aliasing-model-for-rust/8153/>
1601     /// for more details.
1602     Retag(RetagKind, Box<Place<'tcx>>),
1603
1604     /// Encodes a user's type ascription. These need to be preserved
1605     /// intact so that NLL can respect them. For example:
1606     ///
1607     ///     let a: T = y;
1608     ///
1609     /// The effect of this annotation is to relate the type `T_y` of the place `y`
1610     /// to the user-given type `T`. The effect depends on the specified variance:
1611     ///
1612     /// - `Covariant` -- requires that `T_y <: T`
1613     /// - `Contravariant` -- requires that `T_y :> T`
1614     /// - `Invariant` -- requires that `T_y == T`
1615     /// - `Bivariant` -- no effect
1616     AscribeUserType(Box<(Place<'tcx>, UserTypeProjection)>, ty::Variance),
1617
1618     /// No-op. Useful for deleting instructions without affecting statement indices.
1619     Nop,
1620 }
1621
1622 /// Describes what kind of retag is to be performed.
1623 #[derive(Copy, Clone, RustcEncodable, RustcDecodable, Debug, PartialEq, Eq, HashStable)]
1624 pub enum RetagKind {
1625     /// The initial retag when entering a function.
1626     FnEntry,
1627     /// Retag preparing for a two-phase borrow.
1628     TwoPhase,
1629     /// Retagging raw pointers.
1630     Raw,
1631     /// A "normal" retag.
1632     Default,
1633 }
1634
1635 /// The `FakeReadCause` describes the type of pattern why a FakeRead statement exists.
1636 #[derive(Copy, Clone, RustcEncodable, RustcDecodable, Debug, HashStable)]
1637 pub enum FakeReadCause {
1638     /// Inject a fake read of the borrowed input at the end of each guards
1639     /// code.
1640     ///
1641     /// This should ensure that you cannot change the variant for an enum while
1642     /// you are in the midst of matching on it.
1643     ForMatchGuard,
1644
1645     /// `let x: !; match x {}` doesn't generate any read of x so we need to
1646     /// generate a read of x to check that it is initialized and safe.
1647     ForMatchedPlace,
1648
1649     /// A fake read of the RefWithinGuard version of a bind-by-value variable
1650     /// in a match guard to ensure that it's value hasn't change by the time
1651     /// we create the OutsideGuard version.
1652     ForGuardBinding,
1653
1654     /// Officially, the semantics of
1655     ///
1656     /// `let pattern = <expr>;`
1657     ///
1658     /// is that `<expr>` is evaluated into a temporary and then this temporary is
1659     /// into the pattern.
1660     ///
1661     /// However, if we see the simple pattern `let var = <expr>`, we optimize this to
1662     /// evaluate `<expr>` directly into the variable `var`. This is mostly unobservable,
1663     /// but in some cases it can affect the borrow checker, as in #53695.
1664     /// Therefore, we insert a "fake read" here to ensure that we get
1665     /// appropriate errors.
1666     ForLet,
1667 }
1668
1669 #[derive(Clone, Debug, RustcEncodable, RustcDecodable, HashStable)]
1670 pub struct InlineAsm<'tcx> {
1671     pub asm: HirInlineAsm,
1672     pub outputs: Box<[Place<'tcx>]>,
1673     pub inputs: Box<[(Span, Operand<'tcx>)]>,
1674 }
1675
1676 impl Debug for Statement<'_> {
1677     fn fmt(&self, fmt: &mut Formatter<'_>) -> fmt::Result {
1678         use self::StatementKind::*;
1679         match self.kind {
1680             Assign(box(ref place, ref rv)) => write!(fmt, "{:?} = {:?}", place, rv),
1681             FakeRead(ref cause, ref place) => write!(fmt, "FakeRead({:?}, {:?})", cause, place),
1682             Retag(ref kind, ref place) => write!(
1683                 fmt,
1684                 "Retag({}{:?})",
1685                 match kind {
1686                     RetagKind::FnEntry => "[fn entry] ",
1687                     RetagKind::TwoPhase => "[2phase] ",
1688                     RetagKind::Raw => "[raw] ",
1689                     RetagKind::Default => "",
1690                 },
1691                 place,
1692             ),
1693             StorageLive(ref place) => write!(fmt, "StorageLive({:?})", place),
1694             StorageDead(ref place) => write!(fmt, "StorageDead({:?})", place),
1695             SetDiscriminant { ref place, variant_index } => {
1696                 write!(fmt, "discriminant({:?}) = {:?}", place, variant_index)
1697             }
1698             InlineAsm(ref asm) => {
1699                 write!(fmt, "asm!({:?} : {:?} : {:?})", asm.asm, asm.outputs, asm.inputs)
1700             }
1701             AscribeUserType(box(ref place, ref c_ty), ref variance) => {
1702                 write!(fmt, "AscribeUserType({:?}, {:?}, {:?})", place, variance, c_ty)
1703             }
1704             Nop => write!(fmt, "nop"),
1705         }
1706     }
1707 }
1708
1709 ///////////////////////////////////////////////////////////////////////////
1710 // Places
1711
1712 /// A path to a value; something that can be evaluated without
1713 /// changing or disturbing program state.
1714 #[derive(
1715     Clone, PartialEq, Eq, PartialOrd, Ord, Hash, RustcEncodable, RustcDecodable, HashStable,
1716 )]
1717 pub struct Place<'tcx> {
1718     pub base: PlaceBase<'tcx>,
1719
1720     /// projection out of a place (access a field, deref a pointer, etc)
1721     pub projection: Box<[PlaceElem<'tcx>]>,
1722 }
1723
1724 #[derive(
1725     Clone, PartialEq, Eq, PartialOrd, Ord, Hash, RustcEncodable, RustcDecodable, HashStable,
1726 )]
1727 pub enum PlaceBase<'tcx> {
1728     /// local variable
1729     Local(Local),
1730
1731     /// static or static mut variable
1732     Static(Box<Static<'tcx>>),
1733 }
1734
1735 /// We store the normalized type to avoid requiring normalization when reading MIR
1736 #[derive(Clone, Debug, PartialEq, Eq, PartialOrd, Ord, Hash, RustcEncodable, RustcDecodable)]
1737 pub struct Static<'tcx> {
1738     pub ty: Ty<'tcx>,
1739     pub kind: StaticKind<'tcx>,
1740     /// The `DefId` of the item this static was declared in. For promoted values, usually, this is
1741     /// the same as the `DefId` of the `mir::Body` containing the `Place` this promoted appears in.
1742     /// However, after inlining, that might no longer be the case as inlined `Place`s are copied
1743     /// into the calling frame.
1744     pub def_id: DefId,
1745 }
1746
1747 #[derive(
1748     Clone, Debug, PartialEq, Eq, PartialOrd, Ord, Hash, HashStable, RustcEncodable, RustcDecodable,
1749 )]
1750 pub enum StaticKind<'tcx> {
1751     /// Promoted references consist of an id (`Promoted`) and the substs necessary to monomorphize
1752     /// it. Usually, these substs are just the identity substs for the item. However, the inliner
1753     /// will adjust these substs when it inlines a function based on the substs at the callsite.
1754     Promoted(Promoted, SubstsRef<'tcx>),
1755     Static,
1756 }
1757
1758 impl_stable_hash_for!(struct Static<'tcx> {
1759     ty,
1760     kind,
1761     def_id
1762 });
1763
1764 #[derive(
1765     Clone, Debug, PartialEq, Eq, PartialOrd, Ord, Hash, RustcEncodable, RustcDecodable, HashStable,
1766 )]
1767 pub enum ProjectionElem<V, T> {
1768     Deref,
1769     Field(Field, T),
1770     Index(V),
1771
1772     /// These indices are generated by slice patterns. Easiest to explain
1773     /// by example:
1774     ///
1775     /// ```
1776     /// [X, _, .._, _, _] => { offset: 0, min_length: 4, from_end: false },
1777     /// [_, X, .._, _, _] => { offset: 1, min_length: 4, from_end: false },
1778     /// [_, _, .._, X, _] => { offset: 2, min_length: 4, from_end: true },
1779     /// [_, _, .._, _, X] => { offset: 1, min_length: 4, from_end: true },
1780     /// ```
1781     ConstantIndex {
1782         /// index or -index (in Python terms), depending on from_end
1783         offset: u32,
1784         /// thing being indexed must be at least this long
1785         min_length: u32,
1786         /// counting backwards from end?
1787         from_end: bool,
1788     },
1789
1790     /// These indices are generated by slice patterns.
1791     ///
1792     /// slice[from:-to] in Python terms.
1793     Subslice {
1794         from: u32,
1795         to: u32,
1796     },
1797
1798     /// "Downcast" to a variant of an ADT. Currently, we only introduce
1799     /// this for ADTs with more than one variant. It may be better to
1800     /// just introduce it always, or always for enums.
1801     ///
1802     /// The included Symbol is the name of the variant, used for printing MIR.
1803     Downcast(Option<Symbol>, VariantIdx),
1804 }
1805
1806 impl<V, T> ProjectionElem<V, T> {
1807     /// Returns `true` if the target of this projection may refer to a different region of memory
1808     /// than the base.
1809     fn is_indirect(&self) -> bool {
1810         match self {
1811             Self::Deref => true,
1812
1813             | Self::Field(_, _)
1814             | Self::Index(_)
1815             | Self::ConstantIndex { .. }
1816             | Self::Subslice { .. }
1817             | Self::Downcast(_, _)
1818             => false
1819         }
1820     }
1821 }
1822
1823 /// Alias for projections as they appear in places, where the base is a place
1824 /// and the index is a local.
1825 pub type PlaceElem<'tcx> = ProjectionElem<Local, Ty<'tcx>>;
1826
1827 // At least on 64 bit systems, `PlaceElem` should not be larger than two pointers.
1828 #[cfg(target_arch = "x86_64")]
1829 static_assert_size!(PlaceElem<'_>, 16);
1830
1831 /// Alias for projections as they appear in `UserTypeProjection`, where we
1832 /// need neither the `V` parameter for `Index` nor the `T` for `Field`.
1833 pub type ProjectionKind = ProjectionElem<(), ()>;
1834
1835 rustc_index::newtype_index! {
1836     pub struct Field {
1837         derive [HashStable]
1838         DEBUG_FORMAT = "field[{}]"
1839     }
1840 }
1841
1842 #[derive(Clone, Copy, Debug, PartialEq, Eq, PartialOrd, Ord, Hash)]
1843 pub struct PlaceRef<'a, 'tcx> {
1844     pub base: &'a PlaceBase<'tcx>,
1845     pub projection: &'a [PlaceElem<'tcx>],
1846 }
1847
1848 impl<'tcx> Place<'tcx> {
1849     // FIXME change this back to a const when projection is a shared slice.
1850     //
1851     // pub const RETURN_PLACE: Place<'tcx> = Place {
1852     //     base: PlaceBase::Local(RETURN_PLACE),
1853     //     projection: &[],
1854     // };
1855     pub fn return_place() -> Place<'tcx> {
1856         Place {
1857             base: PlaceBase::Local(RETURN_PLACE),
1858             projection: Box::new([]),
1859         }
1860     }
1861
1862     pub fn field(self, f: Field, ty: Ty<'tcx>) -> Place<'tcx> {
1863         self.elem(ProjectionElem::Field(f, ty))
1864     }
1865
1866     pub fn deref(self) -> Place<'tcx> {
1867         self.elem(ProjectionElem::Deref)
1868     }
1869
1870     pub fn downcast(self, adt_def: &'tcx AdtDef, variant_index: VariantIdx) -> Place<'tcx> {
1871         self.elem(ProjectionElem::Downcast(
1872             Some(adt_def.variants[variant_index].ident.name),
1873             variant_index,
1874         ))
1875     }
1876
1877     pub fn downcast_unnamed(self, variant_index: VariantIdx) -> Place<'tcx> {
1878         self.elem(ProjectionElem::Downcast(None, variant_index))
1879     }
1880
1881     pub fn index(self, index: Local) -> Place<'tcx> {
1882         self.elem(ProjectionElem::Index(index))
1883     }
1884
1885     pub fn elem(self, elem: PlaceElem<'tcx>) -> Place<'tcx> {
1886         // FIXME(spastorino): revisit this again once projection is not a Box<[T]> anymore
1887         let mut projection = self.projection.into_vec();
1888         projection.push(elem);
1889
1890         Place {
1891             base: self.base,
1892             projection: projection.into_boxed_slice(),
1893         }
1894     }
1895
1896     /// Returns `true` if this `Place` contains a `Deref` projection.
1897     ///
1898     /// If `Place::is_indirect` returns false, the caller knows that the `Place` refers to the
1899     /// same region of memory as its base.
1900     pub fn is_indirect(&self) -> bool {
1901         self.projection.iter().any(|elem| elem.is_indirect())
1902     }
1903
1904     /// Finds the innermost `Local` from this `Place`, *if* it is either a local itself or
1905     /// a single deref of a local.
1906     //
1907     // FIXME: can we safely swap the semantics of `fn base_local` below in here instead?
1908     pub fn local_or_deref_local(&self) -> Option<Local> {
1909         match self {
1910             Place {
1911                 base: PlaceBase::Local(local),
1912                 projection: box [],
1913             } |
1914             Place {
1915                 base: PlaceBase::Local(local),
1916                 projection: box [ProjectionElem::Deref],
1917             } => Some(*local),
1918             _ => None,
1919         }
1920     }
1921
1922     /// If this place represents a local variable like `_X` with no
1923     /// projections, return `Some(_X)`.
1924     pub fn as_local(&self) -> Option<Local> {
1925         match self {
1926             Place { projection: box [], base: PlaceBase::Local(l) } => Some(*l),
1927             _ => None,
1928         }
1929     }
1930
1931     pub fn as_ref(&self) -> PlaceRef<'_, 'tcx> {
1932         PlaceRef {
1933             base: &self.base,
1934             projection: &self.projection,
1935         }
1936     }
1937 }
1938
1939 impl From<Local> for Place<'_> {
1940     fn from(local: Local) -> Self {
1941         Place {
1942             base: local.into(),
1943             projection: Box::new([]),
1944         }
1945     }
1946 }
1947
1948 impl From<Local> for PlaceBase<'_> {
1949     fn from(local: Local) -> Self {
1950         PlaceBase::Local(local)
1951     }
1952 }
1953
1954 impl<'a, 'tcx> PlaceRef<'a, 'tcx> {
1955     /// Finds the innermost `Local` from this `Place`, *if* it is either a local itself or
1956     /// a single deref of a local.
1957     //
1958     // FIXME: can we safely swap the semantics of `fn base_local` below in here instead?
1959     pub fn local_or_deref_local(&self) -> Option<Local> {
1960         match self {
1961             PlaceRef {
1962                 base: PlaceBase::Local(local),
1963                 projection: [],
1964             } |
1965             PlaceRef {
1966                 base: PlaceBase::Local(local),
1967                 projection: [ProjectionElem::Deref],
1968             } => Some(*local),
1969             _ => None,
1970         }
1971     }
1972 }
1973
1974 impl Debug for Place<'_> {
1975     fn fmt(&self, fmt: &mut Formatter<'_>) -> fmt::Result {
1976         for elem in self.projection.iter().rev() {
1977             match elem {
1978                 ProjectionElem::Downcast(_, _) | ProjectionElem::Field(_, _) => {
1979                     write!(fmt, "(").unwrap();
1980                 }
1981                 ProjectionElem::Deref => {
1982                     write!(fmt, "(*").unwrap();
1983                 }
1984                 ProjectionElem::Index(_)
1985                 | ProjectionElem::ConstantIndex { .. }
1986                 | ProjectionElem::Subslice { .. } => {}
1987             }
1988         }
1989
1990         write!(fmt, "{:?}", self.base)?;
1991
1992         for elem in self.projection.iter() {
1993             match elem {
1994                 ProjectionElem::Downcast(Some(name), _index) => {
1995                     write!(fmt, " as {})", name)?;
1996                 }
1997                 ProjectionElem::Downcast(None, index) => {
1998                     write!(fmt, " as variant#{:?})", index)?;
1999                 }
2000                 ProjectionElem::Deref => {
2001                     write!(fmt, ")")?;
2002                 }
2003                 ProjectionElem::Field(field, ty) => {
2004                     write!(fmt, ".{:?}: {:?})", field.index(), ty)?;
2005                 }
2006                 ProjectionElem::Index(ref index) => {
2007                     write!(fmt, "[{:?}]", index)?;
2008                 }
2009                 ProjectionElem::ConstantIndex { offset, min_length, from_end: false } => {
2010                     write!(fmt, "[{:?} of {:?}]", offset, min_length)?;
2011                 }
2012                 ProjectionElem::ConstantIndex { offset, min_length, from_end: true } => {
2013                     write!(fmt, "[-{:?} of {:?}]", offset, min_length)?;
2014                 }
2015                 ProjectionElem::Subslice { from, to } if *to == 0 => {
2016                     write!(fmt, "[{:?}:]", from)?;
2017                 }
2018                 ProjectionElem::Subslice { from, to } if *from == 0 => {
2019                     write!(fmt, "[:-{:?}]", to)?;
2020                 }
2021                 ProjectionElem::Subslice { from, to } => {
2022                     write!(fmt, "[{:?}:-{:?}]", from, to)?;
2023                 }
2024             }
2025         }
2026
2027         Ok(())
2028     }
2029 }
2030
2031 impl Debug for PlaceBase<'_> {
2032     fn fmt(&self, fmt: &mut Formatter<'_>) -> fmt::Result {
2033         match *self {
2034             PlaceBase::Local(id) => write!(fmt, "{:?}", id),
2035             PlaceBase::Static(box self::Static { ty, kind: StaticKind::Static, def_id }) => {
2036                 write!(fmt, "({}: {:?})", ty::tls::with(|tcx| tcx.def_path_str(def_id)), ty)
2037             }
2038             PlaceBase::Static(box self::Static {
2039                 ty, kind: StaticKind::Promoted(promoted, _), def_id: _
2040             }) => {
2041                 write!(fmt, "({:?}: {:?})", promoted, ty)
2042             }
2043         }
2044     }
2045 }
2046
2047 ///////////////////////////////////////////////////////////////////////////
2048 // Scopes
2049
2050 rustc_index::newtype_index! {
2051     pub struct SourceScope {
2052         derive [HashStable]
2053         DEBUG_FORMAT = "scope[{}]",
2054         const OUTERMOST_SOURCE_SCOPE = 0,
2055     }
2056 }
2057
2058 #[derive(Clone, Debug, RustcEncodable, RustcDecodable, HashStable)]
2059 pub struct SourceScopeData {
2060     pub span: Span,
2061     pub parent_scope: Option<SourceScope>,
2062 }
2063
2064 #[derive(Clone, Debug, RustcEncodable, RustcDecodable, HashStable)]
2065 pub struct SourceScopeLocalData {
2066     /// An `HirId` with lint levels equivalent to this scope's lint levels.
2067     pub lint_root: hir::HirId,
2068     /// The unsafe block that contains this node.
2069     pub safety: Safety,
2070 }
2071
2072 ///////////////////////////////////////////////////////////////////////////
2073 // Operands
2074
2075 /// These are values that can appear inside an rvalue. They are intentionally
2076 /// limited to prevent rvalues from being nested in one another.
2077 #[derive(Clone, PartialEq, RustcEncodable, RustcDecodable, HashStable)]
2078 pub enum Operand<'tcx> {
2079     /// Copy: The value must be available for use afterwards.
2080     ///
2081     /// This implies that the type of the place must be `Copy`; this is true
2082     /// by construction during build, but also checked by the MIR type checker.
2083     Copy(Place<'tcx>),
2084
2085     /// Move: The value (including old borrows of it) will not be used again.
2086     ///
2087     /// Safe for values of all types (modulo future developments towards `?Move`).
2088     /// Correct usage patterns are enforced by the borrow checker for safe code.
2089     /// `Copy` may be converted to `Move` to enable "last-use" optimizations.
2090     Move(Place<'tcx>),
2091
2092     /// Synthesizes a constant value.
2093     Constant(Box<Constant<'tcx>>),
2094 }
2095
2096 impl<'tcx> Debug for Operand<'tcx> {
2097     fn fmt(&self, fmt: &mut Formatter<'_>) -> fmt::Result {
2098         use self::Operand::*;
2099         match *self {
2100             Constant(ref a) => write!(fmt, "{:?}", a),
2101             Copy(ref place) => write!(fmt, "{:?}", place),
2102             Move(ref place) => write!(fmt, "move {:?}", place),
2103         }
2104     }
2105 }
2106
2107 impl<'tcx> Operand<'tcx> {
2108     /// Convenience helper to make a constant that refers to the fn
2109     /// with given `DefId` and substs. Since this is used to synthesize
2110     /// MIR, assumes `user_ty` is None.
2111     pub fn function_handle(
2112         tcx: TyCtxt<'tcx>,
2113         def_id: DefId,
2114         substs: SubstsRef<'tcx>,
2115         span: Span,
2116     ) -> Self {
2117         let ty = tcx.type_of(def_id).subst(tcx, substs);
2118         Operand::Constant(box Constant {
2119             span,
2120             user_ty: None,
2121             literal: ty::Const::zero_sized(tcx, ty),
2122         })
2123     }
2124
2125     pub fn to_copy(&self) -> Self {
2126         match *self {
2127             Operand::Copy(_) | Operand::Constant(_) => self.clone(),
2128             Operand::Move(ref place) => Operand::Copy(place.clone()),
2129         }
2130     }
2131 }
2132
2133 ///////////////////////////////////////////////////////////////////////////
2134 /// Rvalues
2135
2136 #[derive(Clone, RustcEncodable, RustcDecodable, HashStable)]
2137 pub enum Rvalue<'tcx> {
2138     /// x (either a move or copy, depending on type of x)
2139     Use(Operand<'tcx>),
2140
2141     /// [x; 32]
2142     Repeat(Operand<'tcx>, u64),
2143
2144     /// &x or &mut x
2145     Ref(Region<'tcx>, BorrowKind, Place<'tcx>),
2146
2147     /// length of a [X] or [X;n] value
2148     Len(Place<'tcx>),
2149
2150     Cast(CastKind, Operand<'tcx>, Ty<'tcx>),
2151
2152     BinaryOp(BinOp, Operand<'tcx>, Operand<'tcx>),
2153     CheckedBinaryOp(BinOp, Operand<'tcx>, Operand<'tcx>),
2154
2155     NullaryOp(NullOp, Ty<'tcx>),
2156     UnaryOp(UnOp, Operand<'tcx>),
2157
2158     /// Read the discriminant of an ADT.
2159     ///
2160     /// Undefined (i.e., no effort is made to make it defined, but there’s no reason why it cannot
2161     /// be defined to return, say, a 0) if ADT is not an enum.
2162     Discriminant(Place<'tcx>),
2163
2164     /// Creates an aggregate value, like a tuple or struct. This is
2165     /// only needed because we want to distinguish `dest = Foo { x:
2166     /// ..., y: ... }` from `dest.x = ...; dest.y = ...;` in the case
2167     /// that `Foo` has a destructor. These rvalues can be optimized
2168     /// away after type-checking and before lowering.
2169     Aggregate(Box<AggregateKind<'tcx>>, Vec<Operand<'tcx>>),
2170 }
2171
2172 #[derive(Clone, Copy, Debug, PartialEq, Eq, RustcEncodable, RustcDecodable, HashStable)]
2173 pub enum CastKind {
2174     Misc,
2175     Pointer(PointerCast),
2176 }
2177
2178 #[derive(Clone, Debug, PartialEq, Eq, RustcEncodable, RustcDecodable, HashStable)]
2179 pub enum AggregateKind<'tcx> {
2180     /// The type is of the element
2181     Array(Ty<'tcx>),
2182     Tuple,
2183
2184     /// The second field is the variant index. It's equal to 0 for struct
2185     /// and union expressions. The fourth field is
2186     /// active field number and is present only for union expressions
2187     /// -- e.g., for a union expression `SomeUnion { c: .. }`, the
2188     /// active field index would identity the field `c`
2189     Adt(&'tcx AdtDef, VariantIdx, SubstsRef<'tcx>, Option<UserTypeAnnotationIndex>, Option<usize>),
2190
2191     Closure(DefId, SubstsRef<'tcx>),
2192     Generator(DefId, SubstsRef<'tcx>, hir::GeneratorMovability),
2193 }
2194
2195 #[derive(Copy, Clone, Debug, PartialEq, Eq, RustcEncodable, RustcDecodable, HashStable)]
2196 pub enum BinOp {
2197     /// The `+` operator (addition)
2198     Add,
2199     /// The `-` operator (subtraction)
2200     Sub,
2201     /// The `*` operator (multiplication)
2202     Mul,
2203     /// The `/` operator (division)
2204     Div,
2205     /// The `%` operator (modulus)
2206     Rem,
2207     /// The `^` operator (bitwise xor)
2208     BitXor,
2209     /// The `&` operator (bitwise and)
2210     BitAnd,
2211     /// The `|` operator (bitwise or)
2212     BitOr,
2213     /// The `<<` operator (shift left)
2214     Shl,
2215     /// The `>>` operator (shift right)
2216     Shr,
2217     /// The `==` operator (equality)
2218     Eq,
2219     /// The `<` operator (less than)
2220     Lt,
2221     /// The `<=` operator (less than or equal to)
2222     Le,
2223     /// The `!=` operator (not equal to)
2224     Ne,
2225     /// The `>=` operator (greater than or equal to)
2226     Ge,
2227     /// The `>` operator (greater than)
2228     Gt,
2229     /// The `ptr.offset` operator
2230     Offset,
2231 }
2232
2233 impl BinOp {
2234     pub fn is_checkable(self) -> bool {
2235         use self::BinOp::*;
2236         match self {
2237             Add | Sub | Mul | Shl | Shr => true,
2238             _ => false,
2239         }
2240     }
2241 }
2242
2243 #[derive(Copy, Clone, Debug, PartialEq, Eq, RustcEncodable, RustcDecodable, HashStable)]
2244 pub enum NullOp {
2245     /// Returns the size of a value of that type
2246     SizeOf,
2247     /// Creates a new uninitialized box for a value of that type
2248     Box,
2249 }
2250
2251 #[derive(Copy, Clone, Debug, PartialEq, Eq, RustcEncodable, RustcDecodable, HashStable)]
2252 pub enum UnOp {
2253     /// The `!` operator for logical inversion
2254     Not,
2255     /// The `-` operator for negation
2256     Neg,
2257 }
2258
2259 impl<'tcx> Debug for Rvalue<'tcx> {
2260     fn fmt(&self, fmt: &mut Formatter<'_>) -> fmt::Result {
2261         use self::Rvalue::*;
2262
2263         match *self {
2264             Use(ref place) => write!(fmt, "{:?}", place),
2265             Repeat(ref a, ref b) => write!(fmt, "[{:?}; {:?}]", a, b),
2266             Len(ref a) => write!(fmt, "Len({:?})", a),
2267             Cast(ref kind, ref place, ref ty) => {
2268                 write!(fmt, "{:?} as {:?} ({:?})", place, ty, kind)
2269             }
2270             BinaryOp(ref op, ref a, ref b) => write!(fmt, "{:?}({:?}, {:?})", op, a, b),
2271             CheckedBinaryOp(ref op, ref a, ref b) => {
2272                 write!(fmt, "Checked{:?}({:?}, {:?})", op, a, b)
2273             }
2274             UnaryOp(ref op, ref a) => write!(fmt, "{:?}({:?})", op, a),
2275             Discriminant(ref place) => write!(fmt, "discriminant({:?})", place),
2276             NullaryOp(ref op, ref t) => write!(fmt, "{:?}({:?})", op, t),
2277             Ref(region, borrow_kind, ref place) => {
2278                 let kind_str = match borrow_kind {
2279                     BorrowKind::Shared => "",
2280                     BorrowKind::Shallow => "shallow ",
2281                     BorrowKind::Mut { .. } | BorrowKind::Unique => "mut ",
2282                 };
2283
2284                 // When printing regions, add trailing space if necessary.
2285                 let print_region = ty::tls::with(|tcx| {
2286                     tcx.sess.verbose() || tcx.sess.opts.debugging_opts.identify_regions
2287                 });
2288                 let region = if print_region {
2289                     let mut region = region.to_string();
2290                     if region.len() > 0 {
2291                         region.push(' ');
2292                     }
2293                     region
2294                 } else {
2295                     // Do not even print 'static
2296                     String::new()
2297                 };
2298                 write!(fmt, "&{}{}{:?}", region, kind_str, place)
2299             }
2300
2301             Aggregate(ref kind, ref places) => {
2302                 fn fmt_tuple(fmt: &mut Formatter<'_>, places: &[Operand<'_>]) -> fmt::Result {
2303                     let mut tuple_fmt = fmt.debug_tuple("");
2304                     for place in places {
2305                         tuple_fmt.field(place);
2306                     }
2307                     tuple_fmt.finish()
2308                 }
2309
2310                 match **kind {
2311                     AggregateKind::Array(_) => write!(fmt, "{:?}", places),
2312
2313                     AggregateKind::Tuple => match places.len() {
2314                         0 => write!(fmt, "()"),
2315                         1 => write!(fmt, "({:?},)", places[0]),
2316                         _ => fmt_tuple(fmt, places),
2317                     },
2318
2319                     AggregateKind::Adt(adt_def, variant, substs, _user_ty, _) => {
2320                         let variant_def = &adt_def.variants[variant];
2321
2322                         let f = &mut *fmt;
2323                         ty::tls::with(|tcx| {
2324                             let substs = tcx.lift(&substs).expect("could not lift for printing");
2325                             FmtPrinter::new(tcx, f, Namespace::ValueNS)
2326                                 .print_def_path(variant_def.def_id, substs)?;
2327                             Ok(())
2328                         })?;
2329
2330                         match variant_def.ctor_kind {
2331                             CtorKind::Const => Ok(()),
2332                             CtorKind::Fn => fmt_tuple(fmt, places),
2333                             CtorKind::Fictive => {
2334                                 let mut struct_fmt = fmt.debug_struct("");
2335                                 for (field, place) in variant_def.fields.iter().zip(places) {
2336                                     struct_fmt.field(&field.ident.as_str(), place);
2337                                 }
2338                                 struct_fmt.finish()
2339                             }
2340                         }
2341                     }
2342
2343                     AggregateKind::Closure(def_id, _) => ty::tls::with(|tcx| {
2344                         if let Some(hir_id) = tcx.hir().as_local_hir_id(def_id) {
2345                             let name = if tcx.sess.opts.debugging_opts.span_free_formats {
2346                                 format!("[closure@{:?}]", hir_id)
2347                             } else {
2348                                 format!("[closure@{:?}]", tcx.hir().span(hir_id))
2349                             };
2350                             let mut struct_fmt = fmt.debug_struct(&name);
2351
2352                             if let Some(upvars) = tcx.upvars(def_id) {
2353                                 for (&var_id, place) in upvars.keys().zip(places) {
2354                                     let var_name = tcx.hir().name(var_id);
2355                                     struct_fmt.field(&var_name.as_str(), place);
2356                                 }
2357                             }
2358
2359                             struct_fmt.finish()
2360                         } else {
2361                             write!(fmt, "[closure]")
2362                         }
2363                     }),
2364
2365                     AggregateKind::Generator(def_id, _, _) => ty::tls::with(|tcx| {
2366                         if let Some(hir_id) = tcx.hir().as_local_hir_id(def_id) {
2367                             let name = format!("[generator@{:?}]", tcx.hir().span(hir_id));
2368                             let mut struct_fmt = fmt.debug_struct(&name);
2369
2370                             if let Some(upvars) = tcx.upvars(def_id) {
2371                                 for (&var_id, place) in upvars.keys().zip(places) {
2372                                     let var_name = tcx.hir().name(var_id);
2373                                     struct_fmt.field(&var_name.as_str(), place);
2374                                 }
2375                             }
2376
2377                             struct_fmt.finish()
2378                         } else {
2379                             write!(fmt, "[generator]")
2380                         }
2381                     }),
2382                 }
2383             }
2384         }
2385     }
2386 }
2387
2388 ///////////////////////////////////////////////////////////////////////////
2389 /// Constants
2390 ///
2391 /// Two constants are equal if they are the same constant. Note that
2392 /// this does not necessarily mean that they are "==" in Rust -- in
2393 /// particular one must be wary of `NaN`!
2394
2395 #[derive(Copy, Clone, PartialEq, Eq, Hash, RustcEncodable, RustcDecodable, HashStable)]
2396 pub struct Constant<'tcx> {
2397     pub span: Span,
2398
2399     /// Optional user-given type: for something like
2400     /// `collect::<Vec<_>>`, this would be present and would
2401     /// indicate that `Vec<_>` was explicitly specified.
2402     ///
2403     /// Needed for NLL to impose user-given type constraints.
2404     pub user_ty: Option<UserTypeAnnotationIndex>,
2405
2406     pub literal: &'tcx ty::Const<'tcx>,
2407 }
2408
2409 /// A collection of projections into user types.
2410 ///
2411 /// They are projections because a binding can occur a part of a
2412 /// parent pattern that has been ascribed a type.
2413 ///
2414 /// Its a collection because there can be multiple type ascriptions on
2415 /// the path from the root of the pattern down to the binding itself.
2416 ///
2417 /// An example:
2418 ///
2419 /// ```rust
2420 /// struct S<'a>((i32, &'a str), String);
2421 /// let S((_, w): (i32, &'static str), _): S = ...;
2422 /// //    ------  ^^^^^^^^^^^^^^^^^^^ (1)
2423 /// //  ---------------------------------  ^ (2)
2424 /// ```
2425 ///
2426 /// The highlights labelled `(1)` show the subpattern `(_, w)` being
2427 /// ascribed the type `(i32, &'static str)`.
2428 ///
2429 /// The highlights labelled `(2)` show the whole pattern being
2430 /// ascribed the type `S`.
2431 ///
2432 /// In this example, when we descend to `w`, we will have built up the
2433 /// following two projected types:
2434 ///
2435 ///   * base: `S`,                   projection: `(base.0).1`
2436 ///   * base: `(i32, &'static str)`, projection: `base.1`
2437 ///
2438 /// The first will lead to the constraint `w: &'1 str` (for some
2439 /// inferred region `'1`). The second will lead to the constraint `w:
2440 /// &'static str`.
2441 #[derive(Clone, Debug, PartialEq, Eq, Hash, RustcEncodable, RustcDecodable, HashStable)]
2442 pub struct UserTypeProjections {
2443     pub(crate) contents: Vec<(UserTypeProjection, Span)>,
2444 }
2445
2446 BraceStructTypeFoldableImpl! {
2447     impl<'tcx> TypeFoldable<'tcx> for UserTypeProjections {
2448         contents
2449     }
2450 }
2451
2452 impl<'tcx> UserTypeProjections {
2453     pub fn none() -> Self {
2454         UserTypeProjections { contents: vec![] }
2455     }
2456
2457     pub fn from_projections(projs: impl Iterator<Item = (UserTypeProjection, Span)>) -> Self {
2458         UserTypeProjections { contents: projs.collect() }
2459     }
2460
2461     pub fn projections_and_spans(&self) -> impl Iterator<Item = &(UserTypeProjection, Span)> {
2462         self.contents.iter()
2463     }
2464
2465     pub fn projections(&self) -> impl Iterator<Item = &UserTypeProjection> {
2466         self.contents.iter().map(|&(ref user_type, _span)| user_type)
2467     }
2468
2469     pub fn push_projection(mut self, user_ty: &UserTypeProjection, span: Span) -> Self {
2470         self.contents.push((user_ty.clone(), span));
2471         self
2472     }
2473
2474     fn map_projections(
2475         mut self,
2476         mut f: impl FnMut(UserTypeProjection) -> UserTypeProjection,
2477     ) -> Self {
2478         self.contents = self.contents.drain(..).map(|(proj, span)| (f(proj), span)).collect();
2479         self
2480     }
2481
2482     pub fn index(self) -> Self {
2483         self.map_projections(|pat_ty_proj| pat_ty_proj.index())
2484     }
2485
2486     pub fn subslice(self, from: u32, to: u32) -> Self {
2487         self.map_projections(|pat_ty_proj| pat_ty_proj.subslice(from, to))
2488     }
2489
2490     pub fn deref(self) -> Self {
2491         self.map_projections(|pat_ty_proj| pat_ty_proj.deref())
2492     }
2493
2494     pub fn leaf(self, field: Field) -> Self {
2495         self.map_projections(|pat_ty_proj| pat_ty_proj.leaf(field))
2496     }
2497
2498     pub fn variant(self, adt_def: &'tcx AdtDef, variant_index: VariantIdx, field: Field) -> Self {
2499         self.map_projections(|pat_ty_proj| pat_ty_proj.variant(adt_def, variant_index, field))
2500     }
2501 }
2502
2503 /// Encodes the effect of a user-supplied type annotation on the
2504 /// subcomponents of a pattern. The effect is determined by applying the
2505 /// given list of proejctions to some underlying base type. Often,
2506 /// the projection element list `projs` is empty, in which case this
2507 /// directly encodes a type in `base`. But in the case of complex patterns with
2508 /// subpatterns and bindings, we want to apply only a *part* of the type to a variable,
2509 /// in which case the `projs` vector is used.
2510 ///
2511 /// Examples:
2512 ///
2513 /// * `let x: T = ...` -- here, the `projs` vector is empty.
2514 ///
2515 /// * `let (x, _): T = ...` -- here, the `projs` vector would contain
2516 ///   `field[0]` (aka `.0`), indicating that the type of `s` is
2517 ///   determined by finding the type of the `.0` field from `T`.
2518 #[derive(Clone, Debug, PartialEq, Eq, Hash, RustcEncodable, RustcDecodable, HashStable)]
2519 pub struct UserTypeProjection {
2520     pub base: UserTypeAnnotationIndex,
2521     pub projs: Vec<ProjectionKind>,
2522 }
2523
2524 impl Copy for ProjectionKind {}
2525
2526 impl UserTypeProjection {
2527     pub(crate) fn index(mut self) -> Self {
2528         self.projs.push(ProjectionElem::Index(()));
2529         self
2530     }
2531
2532     pub(crate) fn subslice(mut self, from: u32, to: u32) -> Self {
2533         self.projs.push(ProjectionElem::Subslice { from, to });
2534         self
2535     }
2536
2537     pub(crate) fn deref(mut self) -> Self {
2538         self.projs.push(ProjectionElem::Deref);
2539         self
2540     }
2541
2542     pub(crate) fn leaf(mut self, field: Field) -> Self {
2543         self.projs.push(ProjectionElem::Field(field, ()));
2544         self
2545     }
2546
2547     pub(crate) fn variant(
2548         mut self,
2549         adt_def: &'tcx AdtDef,
2550         variant_index: VariantIdx,
2551         field: Field,
2552     ) -> Self {
2553         self.projs.push(ProjectionElem::Downcast(
2554             Some(adt_def.variants[variant_index].ident.name),
2555             variant_index,
2556         ));
2557         self.projs.push(ProjectionElem::Field(field, ()));
2558         self
2559     }
2560 }
2561
2562 CloneTypeFoldableAndLiftImpls! { ProjectionKind, }
2563
2564 impl<'tcx> TypeFoldable<'tcx> for UserTypeProjection {
2565     fn super_fold_with<F: TypeFolder<'tcx>>(&self, folder: &mut F) -> Self {
2566         use crate::mir::ProjectionElem::*;
2567
2568         let base = self.base.fold_with(folder);
2569         let projs: Vec<_> = self
2570             .projs
2571             .iter()
2572             .map(|elem| match elem {
2573                 Deref => Deref,
2574                 Field(f, ()) => Field(f.clone(), ()),
2575                 Index(()) => Index(()),
2576                 elem => elem.clone(),
2577             })
2578             .collect();
2579
2580         UserTypeProjection { base, projs }
2581     }
2582
2583     fn super_visit_with<Vs: TypeVisitor<'tcx>>(&self, visitor: &mut Vs) -> bool {
2584         self.base.visit_with(visitor)
2585         // Note: there's nothing in `self.proj` to visit.
2586     }
2587 }
2588
2589 rustc_index::newtype_index! {
2590     pub struct Promoted {
2591         derive [HashStable]
2592         DEBUG_FORMAT = "promoted[{}]"
2593     }
2594 }
2595
2596 impl<'tcx> Debug for Constant<'tcx> {
2597     fn fmt(&self, fmt: &mut Formatter<'_>) -> fmt::Result {
2598         write!(fmt, "{}", self)
2599     }
2600 }
2601
2602 impl<'tcx> Display for Constant<'tcx> {
2603     fn fmt(&self, fmt: &mut Formatter<'_>) -> fmt::Result {
2604         write!(fmt, "const ")?;
2605         // FIXME make the default pretty printing of raw pointers more detailed. Here we output the
2606         // debug representation of raw pointers, so that the raw pointers in the mir dump output are
2607         // detailed and just not '{pointer}'.
2608         if let ty::RawPtr(_) = self.literal.ty.kind {
2609             write!(fmt, "{:?} : {}", self.literal.val, self.literal.ty)
2610         } else {
2611             write!(fmt, "{}", self.literal)
2612         }
2613     }
2614 }
2615
2616 impl<'tcx> graph::DirectedGraph for Body<'tcx> {
2617     type Node = BasicBlock;
2618 }
2619
2620 impl<'tcx> graph::WithNumNodes for Body<'tcx> {
2621     fn num_nodes(&self) -> usize {
2622         self.basic_blocks.len()
2623     }
2624 }
2625
2626 impl<'tcx> graph::WithStartNode for Body<'tcx> {
2627     fn start_node(&self) -> Self::Node {
2628         START_BLOCK
2629     }
2630 }
2631
2632 impl<'tcx> graph::WithPredecessors for Body<'tcx> {
2633     fn predecessors(
2634         &self,
2635         node: Self::Node,
2636     ) -> <Self as GraphPredecessors<'_>>::Iter {
2637         self.predecessors_for(node).clone().into_iter()
2638     }
2639 }
2640
2641 impl<'tcx> graph::WithSuccessors for Body<'tcx> {
2642     fn successors(
2643         &self,
2644         node: Self::Node,
2645     ) -> <Self as GraphSuccessors<'_>>::Iter {
2646         self.basic_blocks[node].terminator().successors().cloned()
2647     }
2648 }
2649
2650 impl<'a, 'b> graph::GraphPredecessors<'b> for Body<'a> {
2651     type Item = BasicBlock;
2652     type Iter = IntoIter<BasicBlock>;
2653 }
2654
2655 impl<'a, 'b> graph::GraphSuccessors<'b> for Body<'a> {
2656     type Item = BasicBlock;
2657     type Iter = iter::Cloned<Successors<'b>>;
2658 }
2659
2660 #[derive(Copy, Clone, PartialEq, Eq, Hash, Ord, PartialOrd, HashStable)]
2661 pub struct Location {
2662     /// The block that the location is within.
2663     pub block: BasicBlock,
2664
2665     /// The location is the position of the start of the statement; or, if
2666     /// `statement_index` equals the number of statements, then the start of the
2667     /// terminator.
2668     pub statement_index: usize,
2669 }
2670
2671 impl fmt::Debug for Location {
2672     fn fmt(&self, fmt: &mut fmt::Formatter<'_>) -> fmt::Result {
2673         write!(fmt, "{:?}[{}]", self.block, self.statement_index)
2674     }
2675 }
2676
2677 impl Location {
2678     pub const START: Location = Location { block: START_BLOCK, statement_index: 0 };
2679
2680     /// Returns the location immediately after this one within the enclosing block.
2681     ///
2682     /// Note that if this location represents a terminator, then the
2683     /// resulting location would be out of bounds and invalid.
2684     pub fn successor_within_block(&self) -> Location {
2685         Location { block: self.block, statement_index: self.statement_index + 1 }
2686     }
2687
2688     /// Returns `true` if `other` is earlier in the control flow graph than `self`.
2689     pub fn is_predecessor_of<'tcx>(&self, other: Location, body: &Body<'tcx>) -> bool {
2690         // If we are in the same block as the other location and are an earlier statement
2691         // then we are a predecessor of `other`.
2692         if self.block == other.block && self.statement_index < other.statement_index {
2693             return true;
2694         }
2695
2696         // If we're in another block, then we want to check that block is a predecessor of `other`.
2697         let mut queue: Vec<BasicBlock> = body.predecessors_for(other.block).clone();
2698         let mut visited = FxHashSet::default();
2699
2700         while let Some(block) = queue.pop() {
2701             // If we haven't visited this block before, then make sure we visit it's predecessors.
2702             if visited.insert(block) {
2703                 queue.append(&mut body.predecessors_for(block).clone());
2704             } else {
2705                 continue;
2706             }
2707
2708             // If we found the block that `self` is in, then we are a predecessor of `other` (since
2709             // we found that block by looking at the predecessors of `other`).
2710             if self.block == block {
2711                 return true;
2712             }
2713         }
2714
2715         false
2716     }
2717
2718     pub fn dominates(&self, other: Location, dominators: &Dominators<BasicBlock>) -> bool {
2719         if self.block == other.block {
2720             self.statement_index <= other.statement_index
2721         } else {
2722             dominators.is_dominated_by(other.block, self.block)
2723         }
2724     }
2725 }
2726
2727 #[derive(Copy, Clone, Debug, PartialEq, Eq, Hash, RustcEncodable, RustcDecodable, HashStable)]
2728 pub enum UnsafetyViolationKind {
2729     General,
2730     /// Permitted both in `const fn`s and regular `fn`s.
2731     GeneralAndConstFn,
2732     ExternStatic(hir::HirId),
2733     BorrowPacked(hir::HirId),
2734 }
2735
2736 #[derive(Copy, Clone, Debug, PartialEq, Eq, Hash, RustcEncodable, RustcDecodable, HashStable)]
2737 pub struct UnsafetyViolation {
2738     pub source_info: SourceInfo,
2739     pub description: InternedString,
2740     pub details: InternedString,
2741     pub kind: UnsafetyViolationKind,
2742 }
2743
2744 #[derive(Clone, Debug, PartialEq, Eq, Hash, RustcEncodable, RustcDecodable, HashStable)]
2745 pub struct UnsafetyCheckResult {
2746     /// Violations that are propagated *upwards* from this function.
2747     pub violations: Lrc<[UnsafetyViolation]>,
2748     /// `unsafe` blocks in this function, along with whether they are used. This is
2749     /// used for the "unused_unsafe" lint.
2750     pub unsafe_blocks: Lrc<[(hir::HirId, bool)]>,
2751 }
2752
2753 rustc_index::newtype_index! {
2754     pub struct GeneratorSavedLocal {
2755         derive [HashStable]
2756         DEBUG_FORMAT = "_{}",
2757     }
2758 }
2759
2760 /// The layout of generator state.
2761 #[derive(Clone, Debug, RustcEncodable, RustcDecodable, HashStable)]
2762 pub struct GeneratorLayout<'tcx> {
2763     /// The type of every local stored inside the generator.
2764     pub field_tys: IndexVec<GeneratorSavedLocal, Ty<'tcx>>,
2765
2766     /// Which of the above fields are in each variant. Note that one field may
2767     /// be stored in multiple variants.
2768     pub variant_fields: IndexVec<VariantIdx, IndexVec<Field, GeneratorSavedLocal>>,
2769
2770     /// Which saved locals are storage-live at the same time. Locals that do not
2771     /// have conflicts with each other are allowed to overlap in the computed
2772     /// layout.
2773     pub storage_conflicts: BitMatrix<GeneratorSavedLocal, GeneratorSavedLocal>,
2774
2775     /// The names and scopes of all the stored generator locals.
2776     ///
2777     /// N.B., this is *strictly* a temporary hack for codegen
2778     /// debuginfo generation, and will be removed at some point.
2779     /// Do **NOT** use it for anything else, local information should not be
2780     /// in the MIR, please rely on local crate HIR or other side-channels.
2781     //
2782     // FIXME(tmandry): see above.
2783     pub __local_debuginfo_codegen_only_do_not_use: IndexVec<GeneratorSavedLocal, LocalDecl<'tcx>>,
2784 }
2785
2786 #[derive(Clone, Debug, RustcEncodable, RustcDecodable, HashStable)]
2787 pub struct BorrowCheckResult<'tcx> {
2788     pub closure_requirements: Option<ClosureRegionRequirements<'tcx>>,
2789     pub used_mut_upvars: SmallVec<[Field; 8]>,
2790 }
2791
2792 /// After we borrow check a closure, we are left with various
2793 /// requirements that we have inferred between the free regions that
2794 /// appear in the closure's signature or on its field types. These
2795 /// requirements are then verified and proved by the closure's
2796 /// creating function. This struct encodes those requirements.
2797 ///
2798 /// The requirements are listed as being between various
2799 /// `RegionVid`. The 0th region refers to `'static`; subsequent region
2800 /// vids refer to the free regions that appear in the closure (or
2801 /// generator's) type, in order of appearance. (This numbering is
2802 /// actually defined by the `UniversalRegions` struct in the NLL
2803 /// region checker. See for example
2804 /// `UniversalRegions::closure_mapping`.) Note that we treat the free
2805 /// regions in the closure's type "as if" they were erased, so their
2806 /// precise identity is not important, only their position.
2807 ///
2808 /// Example: If type check produces a closure with the closure substs:
2809 ///
2810 /// ```text
2811 /// ClosureSubsts = [
2812 ///     i8,                                  // the "closure kind"
2813 ///     for<'x> fn(&'a &'x u32) -> &'x u32,  // the "closure signature"
2814 ///     &'a String,                          // some upvar
2815 /// ]
2816 /// ```
2817 ///
2818 /// here, there is one unique free region (`'a`) but it appears
2819 /// twice. We would "renumber" each occurrence to a unique vid, as follows:
2820 ///
2821 /// ```text
2822 /// ClosureSubsts = [
2823 ///     i8,                                  // the "closure kind"
2824 ///     for<'x> fn(&'1 &'x u32) -> &'x u32,  // the "closure signature"
2825 ///     &'2 String,                          // some upvar
2826 /// ]
2827 /// ```
2828 ///
2829 /// Now the code might impose a requirement like `'1: '2`. When an
2830 /// instance of the closure is created, the corresponding free regions
2831 /// can be extracted from its type and constrained to have the given
2832 /// outlives relationship.
2833 ///
2834 /// In some cases, we have to record outlives requirements between
2835 /// types and regions as well. In that case, if those types include
2836 /// any regions, those regions are recorded as `ReClosureBound`
2837 /// instances assigned one of these same indices. Those regions will
2838 /// be substituted away by the creator. We use `ReClosureBound` in
2839 /// that case because the regions must be allocated in the global
2840 /// `TyCtxt`, and hence we cannot use `ReVar` (which is what we use
2841 /// internally within the rest of the NLL code).
2842 #[derive(Clone, Debug, RustcEncodable, RustcDecodable, HashStable)]
2843 pub struct ClosureRegionRequirements<'tcx> {
2844     /// The number of external regions defined on the closure. In our
2845     /// example above, it would be 3 -- one for `'static`, then `'1`
2846     /// and `'2`. This is just used for a sanity check later on, to
2847     /// make sure that the number of regions we see at the callsite
2848     /// matches.
2849     pub num_external_vids: usize,
2850
2851     /// Requirements between the various free regions defined in
2852     /// indices.
2853     pub outlives_requirements: Vec<ClosureOutlivesRequirement<'tcx>>,
2854 }
2855
2856 /// Indicates an outlives-constraint between a type or between two
2857 /// free regions declared on the closure.
2858 #[derive(Copy, Clone, Debug, RustcEncodable, RustcDecodable, HashStable)]
2859 pub struct ClosureOutlivesRequirement<'tcx> {
2860     // This region or type ...
2861     pub subject: ClosureOutlivesSubject<'tcx>,
2862
2863     // ... must outlive this one.
2864     pub outlived_free_region: ty::RegionVid,
2865
2866     // If not, report an error here ...
2867     pub blame_span: Span,
2868
2869     // ... due to this reason.
2870     pub category: ConstraintCategory,
2871 }
2872
2873 /// Outlives-constraints can be categorized to determine whether and why they
2874 /// are interesting (for error reporting). Order of variants indicates sort
2875 /// order of the category, thereby influencing diagnostic output.
2876 ///
2877 /// See also [rustc_mir::borrow_check::nll::constraints].
2878 #[derive(
2879     Copy,
2880     Clone,
2881     Debug,
2882     Eq,
2883     PartialEq,
2884     PartialOrd,
2885     Ord,
2886     Hash,
2887     RustcEncodable,
2888     RustcDecodable,
2889     HashStable,
2890 )]
2891 pub enum ConstraintCategory {
2892     Return,
2893     Yield,
2894     UseAsConst,
2895     UseAsStatic,
2896     TypeAnnotation,
2897     Cast,
2898
2899     /// A constraint that came from checking the body of a closure.
2900     ///
2901     /// We try to get the category that the closure used when reporting this.
2902     ClosureBounds,
2903     CallArgument,
2904     CopyBound,
2905     SizedBound,
2906     Assignment,
2907     OpaqueType,
2908
2909     /// A "boring" constraint (caused by the given location) is one that
2910     /// the user probably doesn't want to see described in diagnostics,
2911     /// because it is kind of an artifact of the type system setup.
2912     /// Example: `x = Foo { field: y }` technically creates
2913     /// intermediate regions representing the "type of `Foo { field: y
2914     /// }`", and data flows from `y` into those variables, but they
2915     /// are not very interesting. The assignment into `x` on the other
2916     /// hand might be.
2917     Boring,
2918     // Boring and applicable everywhere.
2919     BoringNoLocation,
2920
2921     /// A constraint that doesn't correspond to anything the user sees.
2922     Internal,
2923 }
2924
2925 /// The subject of a `ClosureOutlivesRequirement` -- that is, the thing
2926 /// that must outlive some region.
2927 #[derive(Copy, Clone, Debug, RustcEncodable, RustcDecodable, HashStable)]
2928 pub enum ClosureOutlivesSubject<'tcx> {
2929     /// Subject is a type, typically a type parameter, but could also
2930     /// be a projection. Indicates a requirement like `T: 'a` being
2931     /// passed to the caller, where the type here is `T`.
2932     ///
2933     /// The type here is guaranteed not to contain any free regions at
2934     /// present.
2935     Ty(Ty<'tcx>),
2936
2937     /// Subject is a free region from the closure. Indicates a requirement
2938     /// like `'a: 'b` being passed to the caller; the region here is `'a`.
2939     Region(ty::RegionVid),
2940 }
2941
2942 /*
2943  * `TypeFoldable` implementations for MIR types
2944 */
2945
2946 CloneTypeFoldableAndLiftImpls! {
2947     BlockTailInfo,
2948     MirPhase,
2949     Mutability,
2950     SourceInfo,
2951     UpvarDebuginfo,
2952     FakeReadCause,
2953     RetagKind,
2954     SourceScope,
2955     SourceScopeData,
2956     SourceScopeLocalData,
2957     UserTypeAnnotationIndex,
2958 }
2959
2960 BraceStructTypeFoldableImpl! {
2961     impl<'tcx> TypeFoldable<'tcx> for Body<'tcx> {
2962         phase,
2963         basic_blocks,
2964         source_scopes,
2965         source_scope_local_data,
2966         yield_ty,
2967         generator_drop,
2968         generator_layout,
2969         local_decls,
2970         user_type_annotations,
2971         arg_count,
2972         __upvar_debuginfo_codegen_only_do_not_use,
2973         spread_arg,
2974         control_flow_destroyed,
2975         span,
2976         cache,
2977     }
2978 }
2979
2980 BraceStructTypeFoldableImpl! {
2981     impl<'tcx> TypeFoldable<'tcx> for GeneratorLayout<'tcx> {
2982         field_tys,
2983         variant_fields,
2984         storage_conflicts,
2985         __local_debuginfo_codegen_only_do_not_use,
2986     }
2987 }
2988
2989 BraceStructTypeFoldableImpl! {
2990     impl<'tcx> TypeFoldable<'tcx> for LocalDecl<'tcx> {
2991         mutability,
2992         is_user_variable,
2993         internal,
2994         ty,
2995         user_ty,
2996         name,
2997         source_info,
2998         is_block_tail,
2999         visibility_scope,
3000     }
3001 }
3002
3003 BraceStructTypeFoldableImpl! {
3004     impl<'tcx> TypeFoldable<'tcx> for BasicBlockData<'tcx> {
3005         statements,
3006         terminator,
3007         is_cleanup,
3008     }
3009 }
3010
3011 BraceStructTypeFoldableImpl! {
3012     impl<'tcx> TypeFoldable<'tcx> for Statement<'tcx> {
3013         source_info, kind
3014     }
3015 }
3016
3017 EnumTypeFoldableImpl! {
3018     impl<'tcx> TypeFoldable<'tcx> for StatementKind<'tcx> {
3019         (StatementKind::Assign)(a),
3020         (StatementKind::FakeRead)(cause, place),
3021         (StatementKind::SetDiscriminant) { place, variant_index },
3022         (StatementKind::StorageLive)(a),
3023         (StatementKind::StorageDead)(a),
3024         (StatementKind::InlineAsm)(a),
3025         (StatementKind::Retag)(kind, place),
3026         (StatementKind::AscribeUserType)(a, v),
3027         (StatementKind::Nop),
3028     }
3029 }
3030
3031 BraceStructTypeFoldableImpl! {
3032     impl<'tcx> TypeFoldable<'tcx> for InlineAsm<'tcx> {
3033         asm,
3034         outputs,
3035         inputs,
3036     }
3037 }
3038
3039 EnumTypeFoldableImpl! {
3040     impl<'tcx, T> TypeFoldable<'tcx> for ClearCrossCrate<T> {
3041         (ClearCrossCrate::Clear),
3042         (ClearCrossCrate::Set)(a),
3043     } where T: TypeFoldable<'tcx>
3044 }
3045
3046 impl<'tcx> TypeFoldable<'tcx> for Terminator<'tcx> {
3047     fn super_fold_with<F: TypeFolder<'tcx>>(&self, folder: &mut F) -> Self {
3048         use crate::mir::TerminatorKind::*;
3049
3050         let kind = match self.kind {
3051             Goto { target } => Goto { target },
3052             SwitchInt { ref discr, switch_ty, ref values, ref targets } => SwitchInt {
3053                 discr: discr.fold_with(folder),
3054                 switch_ty: switch_ty.fold_with(folder),
3055                 values: values.clone(),
3056                 targets: targets.clone(),
3057             },
3058             Drop { ref location, target, unwind } => {
3059                 Drop { location: location.fold_with(folder), target, unwind }
3060             }
3061             DropAndReplace { ref location, ref value, target, unwind } => DropAndReplace {
3062                 location: location.fold_with(folder),
3063                 value: value.fold_with(folder),
3064                 target,
3065                 unwind,
3066             },
3067             Yield { ref value, resume, drop } => {
3068                 Yield { value: value.fold_with(folder), resume: resume, drop: drop }
3069             }
3070             Call { ref func, ref args, ref destination, cleanup, from_hir_call } => {
3071                 let dest =
3072                     destination.as_ref().map(|&(ref loc, dest)| (loc.fold_with(folder), dest));
3073
3074                 Call {
3075                     func: func.fold_with(folder),
3076                     args: args.fold_with(folder),
3077                     destination: dest,
3078                     cleanup,
3079                     from_hir_call,
3080                 }
3081             }
3082             Assert { ref cond, expected, ref msg, target, cleanup } => {
3083                 use PanicInfo::*;
3084                 let msg = match msg {
3085                     BoundsCheck { ref len, ref index } =>
3086                         BoundsCheck {
3087                             len: len.fold_with(folder),
3088                             index: index.fold_with(folder),
3089                         },
3090                     Panic { .. } | Overflow(_) | OverflowNeg | DivisionByZero | RemainderByZero |
3091                     GeneratorResumedAfterReturn | GeneratorResumedAfterPanic =>
3092                         msg.clone(),
3093                 };
3094                 Assert { cond: cond.fold_with(folder), expected, msg, target, cleanup }
3095             }
3096             GeneratorDrop => GeneratorDrop,
3097             Resume => Resume,
3098             Abort => Abort,
3099             Return => Return,
3100             Unreachable => Unreachable,
3101             FalseEdges { real_target, imaginary_target } => {
3102                 FalseEdges { real_target, imaginary_target }
3103             }
3104             FalseUnwind { real_target, unwind } => FalseUnwind { real_target, unwind },
3105         };
3106         Terminator { source_info: self.source_info, kind }
3107     }
3108
3109     fn super_visit_with<V: TypeVisitor<'tcx>>(&self, visitor: &mut V) -> bool {
3110         use crate::mir::TerminatorKind::*;
3111
3112         match self.kind {
3113             SwitchInt { ref discr, switch_ty, .. } => {
3114                 discr.visit_with(visitor) || switch_ty.visit_with(visitor)
3115             }
3116             Drop { ref location, .. } => location.visit_with(visitor),
3117             DropAndReplace { ref location, ref value, .. } => {
3118                 location.visit_with(visitor) || value.visit_with(visitor)
3119             }
3120             Yield { ref value, .. } => value.visit_with(visitor),
3121             Call { ref func, ref args, ref destination, .. } => {
3122                 let dest = if let Some((ref loc, _)) = *destination {
3123                     loc.visit_with(visitor)
3124                 } else {
3125                     false
3126                 };
3127                 dest || func.visit_with(visitor) || args.visit_with(visitor)
3128             }
3129             Assert { ref cond, ref msg, .. } => {
3130                 if cond.visit_with(visitor) {
3131                     use PanicInfo::*;
3132                     match msg {
3133                         BoundsCheck { ref len, ref index } =>
3134                             len.visit_with(visitor) || index.visit_with(visitor),
3135                         Panic { .. } | Overflow(_) | OverflowNeg |
3136                         DivisionByZero | RemainderByZero |
3137                         GeneratorResumedAfterReturn | GeneratorResumedAfterPanic =>
3138                             false
3139                     }
3140                 } else {
3141                     false
3142                 }
3143             }
3144             Goto { .. }
3145             | Resume
3146             | Abort
3147             | Return
3148             | GeneratorDrop
3149             | Unreachable
3150             | FalseEdges { .. }
3151             | FalseUnwind { .. } => false,
3152         }
3153     }
3154 }
3155
3156 impl<'tcx> TypeFoldable<'tcx> for Place<'tcx> {
3157     fn super_fold_with<F: TypeFolder<'tcx>>(&self, folder: &mut F) -> Self {
3158         Place {
3159             base: self.base.fold_with(folder),
3160             projection: self.projection.fold_with(folder),
3161         }
3162     }
3163
3164     fn super_visit_with<V: TypeVisitor<'tcx>>(&self, visitor: &mut V) -> bool {
3165         self.base.visit_with(visitor) || self.projection.visit_with(visitor)
3166     }
3167 }
3168
3169 impl<'tcx> TypeFoldable<'tcx> for PlaceBase<'tcx> {
3170     fn super_fold_with<F: TypeFolder<'tcx>>(&self, folder: &mut F) -> Self {
3171         match self {
3172             PlaceBase::Local(local) => PlaceBase::Local(local.fold_with(folder)),
3173             PlaceBase::Static(static_) => PlaceBase::Static(static_.fold_with(folder)),
3174         }
3175     }
3176
3177     fn super_visit_with<V: TypeVisitor<'tcx>>(&self, visitor: &mut V) -> bool {
3178         match self {
3179             PlaceBase::Local(local) => local.visit_with(visitor),
3180             PlaceBase::Static(static_) => (**static_).visit_with(visitor),
3181         }
3182     }
3183 }
3184
3185 impl<'tcx> TypeFoldable<'tcx> for Static<'tcx> {
3186     fn super_fold_with<F: TypeFolder<'tcx>>(&self, folder: &mut F) -> Self {
3187         Static {
3188             ty: self.ty.fold_with(folder),
3189             kind: self.kind.fold_with(folder),
3190             def_id: self.def_id,
3191         }
3192     }
3193
3194     fn super_visit_with<V: TypeVisitor<'tcx>>(&self, visitor: &mut V) -> bool {
3195         let Static { ty, kind, def_id: _ } = self;
3196
3197         ty.visit_with(visitor) || kind.visit_with(visitor)
3198     }
3199 }
3200
3201 impl<'tcx> TypeFoldable<'tcx> for StaticKind<'tcx> {
3202     fn super_fold_with<F: TypeFolder<'tcx>>(&self, folder: &mut F) -> Self {
3203         match self {
3204             StaticKind::Promoted(promoted, substs) =>
3205                 StaticKind::Promoted(promoted.fold_with(folder), substs.fold_with(folder)),
3206             StaticKind::Static => StaticKind::Static
3207         }
3208     }
3209
3210     fn super_visit_with<V: TypeVisitor<'tcx>>(&self, visitor: &mut V) -> bool {
3211         match self {
3212             StaticKind::Promoted(promoted, substs) =>
3213                 promoted.visit_with(visitor) || substs.visit_with(visitor),
3214             StaticKind::Static => { false }
3215         }
3216     }
3217 }
3218
3219 impl<'tcx> TypeFoldable<'tcx> for Rvalue<'tcx> {
3220     fn super_fold_with<F: TypeFolder<'tcx>>(&self, folder: &mut F) -> Self {
3221         use crate::mir::Rvalue::*;
3222         match *self {
3223             Use(ref op) => Use(op.fold_with(folder)),
3224             Repeat(ref op, len) => Repeat(op.fold_with(folder), len),
3225             Ref(region, bk, ref place) => {
3226                 Ref(region.fold_with(folder), bk, place.fold_with(folder))
3227             }
3228             Len(ref place) => Len(place.fold_with(folder)),
3229             Cast(kind, ref op, ty) => Cast(kind, op.fold_with(folder), ty.fold_with(folder)),
3230             BinaryOp(op, ref rhs, ref lhs) => {
3231                 BinaryOp(op, rhs.fold_with(folder), lhs.fold_with(folder))
3232             }
3233             CheckedBinaryOp(op, ref rhs, ref lhs) => {
3234                 CheckedBinaryOp(op, rhs.fold_with(folder), lhs.fold_with(folder))
3235             }
3236             UnaryOp(op, ref val) => UnaryOp(op, val.fold_with(folder)),
3237             Discriminant(ref place) => Discriminant(place.fold_with(folder)),
3238             NullaryOp(op, ty) => NullaryOp(op, ty.fold_with(folder)),
3239             Aggregate(ref kind, ref fields) => {
3240                 let kind = box match **kind {
3241                     AggregateKind::Array(ty) => AggregateKind::Array(ty.fold_with(folder)),
3242                     AggregateKind::Tuple => AggregateKind::Tuple,
3243                     AggregateKind::Adt(def, v, substs, user_ty, n) => AggregateKind::Adt(
3244                         def,
3245                         v,
3246                         substs.fold_with(folder),
3247                         user_ty.fold_with(folder),
3248                         n,
3249                     ),
3250                     AggregateKind::Closure(id, substs) => {
3251                         AggregateKind::Closure(id, substs.fold_with(folder))
3252                     }
3253                     AggregateKind::Generator(id, substs, movablity) => {
3254                         AggregateKind::Generator(id, substs.fold_with(folder), movablity)
3255                     }
3256                 };
3257                 Aggregate(kind, fields.fold_with(folder))
3258             }
3259         }
3260     }
3261
3262     fn super_visit_with<V: TypeVisitor<'tcx>>(&self, visitor: &mut V) -> bool {
3263         use crate::mir::Rvalue::*;
3264         match *self {
3265             Use(ref op) => op.visit_with(visitor),
3266             Repeat(ref op, _) => op.visit_with(visitor),
3267             Ref(region, _, ref place) => region.visit_with(visitor) || place.visit_with(visitor),
3268             Len(ref place) => place.visit_with(visitor),
3269             Cast(_, ref op, ty) => op.visit_with(visitor) || ty.visit_with(visitor),
3270             BinaryOp(_, ref rhs, ref lhs) | CheckedBinaryOp(_, ref rhs, ref lhs) => {
3271                 rhs.visit_with(visitor) || lhs.visit_with(visitor)
3272             }
3273             UnaryOp(_, ref val) => val.visit_with(visitor),
3274             Discriminant(ref place) => place.visit_with(visitor),
3275             NullaryOp(_, ty) => ty.visit_with(visitor),
3276             Aggregate(ref kind, ref fields) => {
3277                 (match **kind {
3278                     AggregateKind::Array(ty) => ty.visit_with(visitor),
3279                     AggregateKind::Tuple => false,
3280                     AggregateKind::Adt(_, _, substs, user_ty, _) => {
3281                         substs.visit_with(visitor) || user_ty.visit_with(visitor)
3282                     }
3283                     AggregateKind::Closure(_, substs) => substs.visit_with(visitor),
3284                     AggregateKind::Generator(_, substs, _) => substs.visit_with(visitor),
3285                 }) || fields.visit_with(visitor)
3286             }
3287         }
3288     }
3289 }
3290
3291 impl<'tcx> TypeFoldable<'tcx> for Operand<'tcx> {
3292     fn super_fold_with<F: TypeFolder<'tcx>>(&self, folder: &mut F) -> Self {
3293         match *self {
3294             Operand::Copy(ref place) => Operand::Copy(place.fold_with(folder)),
3295             Operand::Move(ref place) => Operand::Move(place.fold_with(folder)),
3296             Operand::Constant(ref c) => Operand::Constant(c.fold_with(folder)),
3297         }
3298     }
3299
3300     fn super_visit_with<V: TypeVisitor<'tcx>>(&self, visitor: &mut V) -> bool {
3301         match *self {
3302             Operand::Copy(ref place) | Operand::Move(ref place) => place.visit_with(visitor),
3303             Operand::Constant(ref c) => c.visit_with(visitor),
3304         }
3305     }
3306 }
3307
3308 impl<'tcx> TypeFoldable<'tcx> for PlaceElem<'tcx> {
3309     fn super_fold_with<F: TypeFolder<'tcx>>(&self, folder: &mut F) -> Self {
3310         use crate::mir::ProjectionElem::*;
3311
3312         match self {
3313             Deref => Deref,
3314             Field(f, ty) => Field(*f, ty.fold_with(folder)),
3315             Index(v) => Index(v.fold_with(folder)),
3316             elem => elem.clone(),
3317         }
3318     }
3319
3320     fn super_visit_with<Vs: TypeVisitor<'tcx>>(&self, visitor: &mut Vs) -> bool {
3321         use crate::mir::ProjectionElem::*;
3322
3323         match self {
3324             Field(_, ty) => ty.visit_with(visitor),
3325             Index(v) => v.visit_with(visitor),
3326             _ => false,
3327         }
3328     }
3329 }
3330
3331 impl<'tcx> TypeFoldable<'tcx> for Field {
3332     fn super_fold_with<F: TypeFolder<'tcx>>(&self, _: &mut F) -> Self {
3333         *self
3334     }
3335     fn super_visit_with<V: TypeVisitor<'tcx>>(&self, _: &mut V) -> bool {
3336         false
3337     }
3338 }
3339
3340 impl<'tcx> TypeFoldable<'tcx> for GeneratorSavedLocal {
3341     fn super_fold_with<F: TypeFolder<'tcx>>(&self, _: &mut F) -> Self {
3342         *self
3343     }
3344     fn super_visit_with<V: TypeVisitor<'tcx>>(&self, _: &mut V) -> bool {
3345         false
3346     }
3347 }
3348
3349 impl<'tcx, R: Idx, C: Idx> TypeFoldable<'tcx> for BitMatrix<R, C> {
3350     fn super_fold_with<F: TypeFolder<'tcx>>(&self, _: &mut F) -> Self {
3351         self.clone()
3352     }
3353     fn super_visit_with<V: TypeVisitor<'tcx>>(&self, _: &mut V) -> bool {
3354         false
3355     }
3356 }
3357
3358 impl<'tcx> TypeFoldable<'tcx> for Constant<'tcx> {
3359     fn super_fold_with<F: TypeFolder<'tcx>>(&self, folder: &mut F) -> Self {
3360         Constant {
3361             span: self.span.clone(),
3362             user_ty: self.user_ty.fold_with(folder),
3363             literal: self.literal.fold_with(folder),
3364         }
3365     }
3366     fn super_visit_with<V: TypeVisitor<'tcx>>(&self, visitor: &mut V) -> bool {
3367         self.literal.visit_with(visitor)
3368     }
3369 }