]> git.lizzy.rs Git - rust.git/blob - src/librustc_mir/borrow_check/mod.rs
improve naming
[rust.git] / src / librustc_mir / borrow_check / mod.rs
1 //! This query borrow-checks the MIR to (further) ensure it is not broken.
2
3 use rustc_data_structures::fx::{FxHashMap, FxHashSet};
4 use rustc_data_structures::graph::dominators::Dominators;
5 use rustc_errors::{Applicability, Diagnostic, DiagnosticBuilder, ErrorReported};
6 use rustc_hir as hir;
7 use rustc_hir::def_id::LocalDefId;
8 use rustc_hir::{HirId, Node};
9 use rustc_index::bit_set::BitSet;
10 use rustc_index::vec::IndexVec;
11 use rustc_infer::infer::{InferCtxt, TyCtxtInferExt};
12 use rustc_middle::mir::{
13     traversal, Body, ClearCrossCrate, Local, Location, Mutability, Operand, Place, PlaceElem,
14     PlaceRef,
15 };
16 use rustc_middle::mir::{AggregateKind, BasicBlock, BorrowCheckResult, BorrowKind};
17 use rustc_middle::mir::{Field, ProjectionElem, Promoted, Rvalue, Statement, StatementKind};
18 use rustc_middle::mir::{InlineAsmOperand, Terminator, TerminatorKind};
19 use rustc_middle::ty::query::Providers;
20 use rustc_middle::ty::{self, InstanceDef, RegionVid, TyCtxt};
21 use rustc_session::lint::builtin::{MUTABLE_BORROW_RESERVATION_CONFLICT, UNUSED_MUT};
22 use rustc_span::{Span, Symbol, DUMMY_SP};
23
24 use either::Either;
25 use smallvec::SmallVec;
26 use std::cell::RefCell;
27 use std::collections::BTreeMap;
28 use std::mem;
29 use std::rc::Rc;
30
31 use crate::dataflow;
32 use crate::dataflow::impls::{
33     Borrows, EverInitializedPlaces, MaybeInitializedPlaces, MaybeUninitializedPlaces,
34 };
35 use crate::dataflow::indexes::{BorrowIndex, InitIndex, MoveOutIndex, MovePathIndex};
36 use crate::dataflow::move_paths::{InitLocation, LookupResult, MoveData, MoveError};
37 use crate::dataflow::MoveDataParamEnv;
38 use crate::dataflow::{Analysis, BorrowckFlowState as Flows, BorrowckResults};
39 use crate::transform::MirSource;
40
41 use self::diagnostics::{AccessKind, RegionName};
42 use self::location::LocationTable;
43 use self::prefixes::PrefixSet;
44 use self::MutateMode::{JustWrite, WriteAndRead};
45
46 use self::path_utils::*;
47
48 mod borrow_set;
49 mod constraint_generation;
50 mod constraints;
51 mod def_use;
52 mod diagnostics;
53 mod facts;
54 mod invalidation;
55 mod location;
56 mod member_constraints;
57 mod nll;
58 mod path_utils;
59 mod place_ext;
60 mod places_conflict;
61 mod prefixes;
62 mod region_infer;
63 mod renumber;
64 mod type_check;
65 mod universal_regions;
66 mod used_muts;
67
68 crate use borrow_set::{BorrowData, BorrowSet};
69 crate use nll::{PoloniusOutput, ToRegionVid};
70 crate use place_ext::PlaceExt;
71 crate use places_conflict::{places_conflict, PlaceConflictBias};
72 crate use region_infer::RegionInferenceContext;
73
74 // FIXME(eddyb) perhaps move this somewhere more centrally.
75 #[derive(Debug)]
76 crate struct Upvar {
77     name: Symbol,
78
79     var_hir_id: HirId,
80
81     /// If true, the capture is behind a reference.
82     by_ref: bool,
83
84     mutability: Mutability,
85 }
86
87 const DEREF_PROJECTION: &[PlaceElem<'_>; 1] = &[ProjectionElem::Deref];
88
89 pub fn provide(providers: &mut Providers) {
90     *providers = Providers {
91         mir_borrowck: |tcx, did| mir_borrowck(tcx, ty::WithOptConstParam::dummy(did)),
92         mir_borrowck_const_arg: |tcx, (did, param_did)| {
93             mir_borrowck(tcx, ty::WithOptConstParam { did, const_param_did: Some(param_did) })
94         },
95         ..*providers
96     };
97 }
98
99 fn mir_borrowck<'tcx>(
100     tcx: TyCtxt<'tcx>,
101     def: ty::WithOptConstParam<LocalDefId>,
102 ) -> &'tcx BorrowCheckResult<'tcx> {
103     if def.const_param_did.is_none() {
104         if let Some(param_did) = tcx.opt_const_param_of(def.did) {
105             return tcx.mir_borrowck_const_arg((def.did, param_did));
106         }
107     }
108
109     let (input_body, promoted) = tcx.mir_validated(def);
110     debug!("run query mir_borrowck: {}", tcx.def_path_str(def.did.to_def_id()));
111
112     let opt_closure_req = tcx.infer_ctxt().enter(|infcx| {
113         let input_body: &Body<'_> = &input_body.borrow();
114         let promoted: &IndexVec<_, _> = &promoted.borrow();
115         do_mir_borrowck(&infcx, input_body, promoted, def)
116     });
117     debug!("mir_borrowck done");
118
119     tcx.arena.alloc(opt_closure_req)
120 }
121
122 fn do_mir_borrowck<'a, 'tcx>(
123     infcx: &InferCtxt<'a, 'tcx>,
124     input_body: &Body<'tcx>,
125     input_promoted: &IndexVec<Promoted, Body<'tcx>>,
126     def: ty::WithOptConstParam<LocalDefId>,
127 ) -> BorrowCheckResult<'tcx> {
128     debug!("do_mir_borrowck(def = {:?})", def);
129
130     let tcx = infcx.tcx;
131     let param_env = tcx.param_env(def.did);
132     let id = tcx.hir().as_local_hir_id(def.did);
133
134     let mut local_names = IndexVec::from_elem(None, &input_body.local_decls);
135     for var_debug_info in &input_body.var_debug_info {
136         if let Some(local) = var_debug_info.place.as_local() {
137             if let Some(prev_name) = local_names[local] {
138                 if var_debug_info.name != prev_name {
139                     span_bug!(
140                         var_debug_info.source_info.span,
141                         "local {:?} has many names (`{}` vs `{}`)",
142                         local,
143                         prev_name,
144                         var_debug_info.name
145                     );
146                 }
147             }
148             local_names[local] = Some(var_debug_info.name);
149         }
150     }
151
152     // Gather the upvars of a closure, if any.
153     let tables = tcx.typeck_tables_of_opt_const_arg(def);
154     if let Some(ErrorReported) = tables.tainted_by_errors {
155         infcx.set_tainted_by_errors();
156     }
157     let upvars: Vec<_> = tables
158         .closure_captures
159         .get(&def.did.to_def_id())
160         .into_iter()
161         .flat_map(|v| v.values())
162         .map(|upvar_id| {
163             let var_hir_id = upvar_id.var_path.hir_id;
164             let capture = tables.upvar_capture(*upvar_id);
165             let by_ref = match capture {
166                 ty::UpvarCapture::ByValue => false,
167                 ty::UpvarCapture::ByRef(..) => true,
168             };
169             let mut upvar = Upvar {
170                 name: tcx.hir().name(var_hir_id),
171                 var_hir_id,
172                 by_ref,
173                 mutability: Mutability::Not,
174             };
175             let bm = *tables.pat_binding_modes().get(var_hir_id).expect("missing binding mode");
176             if bm == ty::BindByValue(hir::Mutability::Mut) {
177                 upvar.mutability = Mutability::Mut;
178             }
179             upvar
180         })
181         .collect();
182
183     // Replace all regions with fresh inference variables. This
184     // requires first making our own copy of the MIR. This copy will
185     // be modified (in place) to contain non-lexical lifetimes. It
186     // will have a lifetime tied to the inference context.
187     let mut body = input_body.clone();
188     let mut promoted = input_promoted.clone();
189     let free_regions = nll::replace_regions_in_mir(infcx, def, param_env, &mut body, &mut promoted);
190     let body = &body; // no further changes
191
192     let location_table = &LocationTable::new(&body);
193
194     let mut errors_buffer = Vec::new();
195     let (move_data, move_errors): (MoveData<'tcx>, Vec<(Place<'tcx>, MoveError<'tcx>)>) =
196         match MoveData::gather_moves(&body, tcx, param_env) {
197             Ok(move_data) => (move_data, Vec::new()),
198             Err((move_data, move_errors)) => (move_data, move_errors),
199         };
200     let promoted_errors = promoted
201         .iter_enumerated()
202         .map(|(idx, body)| (idx, MoveData::gather_moves(&body, tcx, param_env)));
203
204     let mdpe = MoveDataParamEnv { move_data, param_env };
205
206     let mut flow_inits = MaybeInitializedPlaces::new(tcx, &body, &mdpe)
207         .into_engine(tcx, &body, def.did.to_def_id())
208         .iterate_to_fixpoint()
209         .into_results_cursor(&body);
210
211     let locals_are_invalidated_at_exit = tcx.hir().body_owner_kind(id).is_fn_or_closure();
212     let borrow_set =
213         Rc::new(BorrowSet::build(tcx, body, locals_are_invalidated_at_exit, &mdpe.move_data));
214
215     // Compute non-lexical lifetimes.
216     let nll::NllOutput {
217         regioncx,
218         opaque_type_values,
219         polonius_output,
220         opt_closure_req,
221         nll_errors,
222     } = nll::compute_regions(
223         infcx,
224         def.did,
225         free_regions,
226         body,
227         &promoted,
228         location_table,
229         param_env,
230         &mut flow_inits,
231         &mdpe.move_data,
232         &borrow_set,
233         &upvars,
234     );
235
236     // Dump MIR results into a file, if that is enabled. This let us
237     // write unit-tests, as well as helping with debugging.
238     nll::dump_mir_results(
239         infcx,
240         MirSource { instance: InstanceDef::Item(def.to_global()), promoted: None },
241         &body,
242         &regioncx,
243         &opt_closure_req,
244     );
245
246     // We also have a `#[rustc_regions]` annotation that causes us to dump
247     // information.
248     nll::dump_annotation(
249         infcx,
250         &body,
251         def.did.to_def_id(),
252         &regioncx,
253         &opt_closure_req,
254         &opaque_type_values,
255         &mut errors_buffer,
256     );
257
258     // The various `flow_*` structures can be large. We drop `flow_inits` here
259     // so it doesn't overlap with the others below. This reduces peak memory
260     // usage significantly on some benchmarks.
261     drop(flow_inits);
262
263     let regioncx = Rc::new(regioncx);
264
265     let flow_borrows = Borrows::new(tcx, &body, regioncx.clone(), &borrow_set)
266         .into_engine(tcx, &body, def.did.to_def_id())
267         .iterate_to_fixpoint();
268     let flow_uninits = MaybeUninitializedPlaces::new(tcx, &body, &mdpe)
269         .into_engine(tcx, &body, def.did.to_def_id())
270         .iterate_to_fixpoint();
271     let flow_ever_inits = EverInitializedPlaces::new(tcx, &body, &mdpe)
272         .into_engine(tcx, &body, def.did.to_def_id())
273         .iterate_to_fixpoint();
274
275     let movable_generator = match tcx.hir().get(id) {
276         Node::Expr(&hir::Expr {
277             kind: hir::ExprKind::Closure(.., Some(hir::Movability::Static)),
278             ..
279         }) => false,
280         _ => true,
281     };
282
283     for (idx, move_data_results) in promoted_errors {
284         let promoted_body = &promoted[idx];
285         let dominators = promoted_body.dominators();
286
287         if let Err((move_data, move_errors)) = move_data_results {
288             let mut promoted_mbcx = MirBorrowckCtxt {
289                 infcx,
290                 body: promoted_body,
291                 mir_def_id: def.did,
292                 move_data: &move_data,
293                 location_table: &LocationTable::new(promoted_body),
294                 movable_generator,
295                 fn_self_span_reported: Default::default(),
296                 locals_are_invalidated_at_exit,
297                 access_place_error_reported: Default::default(),
298                 reservation_error_reported: Default::default(),
299                 reservation_warnings: Default::default(),
300                 move_error_reported: BTreeMap::new(),
301                 uninitialized_error_reported: Default::default(),
302                 errors_buffer,
303                 regioncx: regioncx.clone(),
304                 used_mut: Default::default(),
305                 used_mut_upvars: SmallVec::new(),
306                 borrow_set: borrow_set.clone(),
307                 dominators,
308                 upvars: Vec::new(),
309                 local_names: IndexVec::from_elem(None, &promoted_body.local_decls),
310                 region_names: RefCell::default(),
311                 next_region_name: RefCell::new(1),
312                 polonius_output: None,
313             };
314             promoted_mbcx.report_move_errors(move_errors);
315             errors_buffer = promoted_mbcx.errors_buffer;
316         };
317     }
318
319     let dominators = body.dominators();
320
321     let mut mbcx = MirBorrowckCtxt {
322         infcx,
323         body,
324         mir_def_id: def.did,
325         move_data: &mdpe.move_data,
326         location_table,
327         movable_generator,
328         locals_are_invalidated_at_exit,
329         fn_self_span_reported: Default::default(),
330         access_place_error_reported: Default::default(),
331         reservation_error_reported: Default::default(),
332         reservation_warnings: Default::default(),
333         move_error_reported: BTreeMap::new(),
334         uninitialized_error_reported: Default::default(),
335         errors_buffer,
336         regioncx,
337         used_mut: Default::default(),
338         used_mut_upvars: SmallVec::new(),
339         borrow_set,
340         dominators,
341         upvars,
342         local_names,
343         region_names: RefCell::default(),
344         next_region_name: RefCell::new(1),
345         polonius_output,
346     };
347
348     // Compute and report region errors, if any.
349     mbcx.report_region_errors(nll_errors);
350
351     let results = BorrowckResults {
352         ever_inits: flow_ever_inits,
353         uninits: flow_uninits,
354         borrows: flow_borrows,
355     };
356
357     mbcx.report_move_errors(move_errors);
358
359     dataflow::visit_results(
360         &body,
361         traversal::reverse_postorder(&body).map(|(bb, _)| bb),
362         &results,
363         &mut mbcx,
364     );
365
366     // Convert any reservation warnings into lints.
367     let reservation_warnings = mem::take(&mut mbcx.reservation_warnings);
368     for (_, (place, span, location, bk, borrow)) in reservation_warnings {
369         let mut initial_diag = mbcx.report_conflicting_borrow(location, (place, span), bk, &borrow);
370
371         let scope = mbcx.body.source_info(location).scope;
372         let lint_root = match &mbcx.body.source_scopes[scope].local_data {
373             ClearCrossCrate::Set(data) => data.lint_root,
374             _ => id,
375         };
376
377         // Span and message don't matter; we overwrite them below anyway
378         mbcx.infcx.tcx.struct_span_lint_hir(
379             MUTABLE_BORROW_RESERVATION_CONFLICT,
380             lint_root,
381             DUMMY_SP,
382             |lint| {
383                 let mut diag = lint.build("");
384
385                 diag.message = initial_diag.styled_message().clone();
386                 diag.span = initial_diag.span.clone();
387
388                 diag.buffer(&mut mbcx.errors_buffer);
389             },
390         );
391         initial_diag.cancel();
392     }
393
394     // For each non-user used mutable variable, check if it's been assigned from
395     // a user-declared local. If so, then put that local into the used_mut set.
396     // Note that this set is expected to be small - only upvars from closures
397     // would have a chance of erroneously adding non-user-defined mutable vars
398     // to the set.
399     let temporary_used_locals: FxHashSet<Local> = mbcx
400         .used_mut
401         .iter()
402         .filter(|&local| !mbcx.body.local_decls[*local].is_user_variable())
403         .cloned()
404         .collect();
405     // For the remaining unused locals that are marked as mutable, we avoid linting any that
406     // were never initialized. These locals may have been removed as unreachable code; or will be
407     // linted as unused variables.
408     let unused_mut_locals =
409         mbcx.body.mut_vars_iter().filter(|local| !mbcx.used_mut.contains(local)).collect();
410     mbcx.gather_used_muts(temporary_used_locals, unused_mut_locals);
411
412     debug!("mbcx.used_mut: {:?}", mbcx.used_mut);
413     let used_mut = mbcx.used_mut;
414     for local in mbcx.body.mut_vars_and_args_iter().filter(|local| !used_mut.contains(local)) {
415         let local_decl = &mbcx.body.local_decls[local];
416         let lint_root = match &mbcx.body.source_scopes[local_decl.source_info.scope].local_data {
417             ClearCrossCrate::Set(data) => data.lint_root,
418             _ => continue,
419         };
420
421         // Skip over locals that begin with an underscore or have no name
422         match mbcx.local_names[local] {
423             Some(name) => {
424                 if name.as_str().starts_with('_') {
425                     continue;
426                 }
427             }
428             None => continue,
429         }
430
431         let span = local_decl.source_info.span;
432         if span.desugaring_kind().is_some() {
433             // If the `mut` arises as part of a desugaring, we should ignore it.
434             continue;
435         }
436
437         tcx.struct_span_lint_hir(UNUSED_MUT, lint_root, span, |lint| {
438             let mut_span = tcx.sess.source_map().span_until_non_whitespace(span);
439             lint.build("variable does not need to be mutable")
440                 .span_suggestion_short(
441                     mut_span,
442                     "remove this `mut`",
443                     String::new(),
444                     Applicability::MachineApplicable,
445                 )
446                 .emit();
447         })
448     }
449
450     // Buffer any move errors that we collected and de-duplicated.
451     for (_, (_, diag)) in mbcx.move_error_reported {
452         diag.buffer(&mut mbcx.errors_buffer);
453     }
454
455     if !mbcx.errors_buffer.is_empty() {
456         mbcx.errors_buffer.sort_by_key(|diag| diag.sort_span);
457
458         for diag in mbcx.errors_buffer.drain(..) {
459             mbcx.infcx.tcx.sess.diagnostic().emit_diagnostic(&diag);
460         }
461     }
462
463     let result = BorrowCheckResult {
464         concrete_opaque_types: opaque_type_values,
465         closure_requirements: opt_closure_req,
466         used_mut_upvars: mbcx.used_mut_upvars,
467     };
468
469     debug!("do_mir_borrowck: result = {:#?}", result);
470
471     result
472 }
473
474 crate struct MirBorrowckCtxt<'cx, 'tcx> {
475     crate infcx: &'cx InferCtxt<'cx, 'tcx>,
476     body: &'cx Body<'tcx>,
477     mir_def_id: LocalDefId,
478     move_data: &'cx MoveData<'tcx>,
479
480     /// Map from MIR `Location` to `LocationIndex`; created
481     /// when MIR borrowck begins.
482     location_table: &'cx LocationTable,
483
484     movable_generator: bool,
485     /// This keeps track of whether local variables are free-ed when the function
486     /// exits even without a `StorageDead`, which appears to be the case for
487     /// constants.
488     ///
489     /// I'm not sure this is the right approach - @eddyb could you try and
490     /// figure this out?
491     locals_are_invalidated_at_exit: bool,
492     /// This field keeps track of when borrow errors are reported in the access_place function
493     /// so that there is no duplicate reporting. This field cannot also be used for the conflicting
494     /// borrow errors that is handled by the `reservation_error_reported` field as the inclusion
495     /// of the `Span` type (while required to mute some errors) stops the muting of the reservation
496     /// errors.
497     access_place_error_reported: FxHashSet<(Place<'tcx>, Span)>,
498     /// This field keeps track of when borrow conflict errors are reported
499     /// for reservations, so that we don't report seemingly duplicate
500     /// errors for corresponding activations.
501     //
502     // FIXME: ideally this would be a set of `BorrowIndex`, not `Place`s,
503     // but it is currently inconvenient to track down the `BorrowIndex`
504     // at the time we detect and report a reservation error.
505     reservation_error_reported: FxHashSet<Place<'tcx>>,
506     /// This fields keeps track of the `Span`s that we have
507     /// used to report extra information for `FnSelfUse`, to avoid
508     /// unnecessarily verbose errors.
509     fn_self_span_reported: FxHashSet<Span>,
510     /// Migration warnings to be reported for #56254. We delay reporting these
511     /// so that we can suppress the warning if there's a corresponding error
512     /// for the activation of the borrow.
513     reservation_warnings:
514         FxHashMap<BorrowIndex, (Place<'tcx>, Span, Location, BorrowKind, BorrowData<'tcx>)>,
515     /// This field keeps track of move errors that are to be reported for given move indices.
516     ///
517     /// There are situations where many errors can be reported for a single move out (see #53807)
518     /// and we want only the best of those errors.
519     ///
520     /// The `report_use_of_moved_or_uninitialized` function checks this map and replaces the
521     /// diagnostic (if there is one) if the `Place` of the error being reported is a prefix of the
522     /// `Place` of the previous most diagnostic. This happens instead of buffering the error. Once
523     /// all move errors have been reported, any diagnostics in this map are added to the buffer
524     /// to be emitted.
525     ///
526     /// `BTreeMap` is used to preserve the order of insertions when iterating. This is necessary
527     /// when errors in the map are being re-added to the error buffer so that errors with the
528     /// same primary span come out in a consistent order.
529     move_error_reported: BTreeMap<Vec<MoveOutIndex>, (PlaceRef<'tcx>, DiagnosticBuilder<'cx>)>,
530     /// This field keeps track of errors reported in the checking of uninitialized variables,
531     /// so that we don't report seemingly duplicate errors.
532     uninitialized_error_reported: FxHashSet<PlaceRef<'tcx>>,
533     /// Errors to be reported buffer
534     errors_buffer: Vec<Diagnostic>,
535     /// This field keeps track of all the local variables that are declared mut and are mutated.
536     /// Used for the warning issued by an unused mutable local variable.
537     used_mut: FxHashSet<Local>,
538     /// If the function we're checking is a closure, then we'll need to report back the list of
539     /// mutable upvars that have been used. This field keeps track of them.
540     used_mut_upvars: SmallVec<[Field; 8]>,
541     /// Region inference context. This contains the results from region inference and lets us e.g.
542     /// find out which CFG points are contained in each borrow region.
543     regioncx: Rc<RegionInferenceContext<'tcx>>,
544
545     /// The set of borrows extracted from the MIR
546     borrow_set: Rc<BorrowSet<'tcx>>,
547
548     /// Dominators for MIR
549     dominators: Dominators<BasicBlock>,
550
551     /// Information about upvars not necessarily preserved in types or MIR
552     upvars: Vec<Upvar>,
553
554     /// Names of local (user) variables (extracted from `var_debug_info`).
555     local_names: IndexVec<Local, Option<Symbol>>,
556
557     /// Record the region names generated for each region in the given
558     /// MIR def so that we can reuse them later in help/error messages.
559     region_names: RefCell<FxHashMap<RegionVid, RegionName>>,
560
561     /// The counter for generating new region names.
562     next_region_name: RefCell<usize>,
563
564     /// Results of Polonius analysis.
565     polonius_output: Option<Rc<PoloniusOutput>>,
566 }
567
568 // Check that:
569 // 1. assignments are always made to mutable locations (FIXME: does that still really go here?)
570 // 2. loans made in overlapping scopes do not conflict
571 // 3. assignments do not affect things loaned out as immutable
572 // 4. moves do not affect things loaned out in any way
573 impl<'cx, 'tcx> dataflow::ResultsVisitor<'cx, 'tcx> for MirBorrowckCtxt<'cx, 'tcx> {
574     type FlowState = Flows<'cx, 'tcx>;
575
576     fn visit_statement_before_primary_effect(
577         &mut self,
578         flow_state: &Flows<'cx, 'tcx>,
579         stmt: &'cx Statement<'tcx>,
580         location: Location,
581     ) {
582         debug!("MirBorrowckCtxt::process_statement({:?}, {:?}): {:?}", location, stmt, flow_state);
583         let span = stmt.source_info.span;
584
585         self.check_activations(location, span, flow_state);
586
587         match &stmt.kind {
588             StatementKind::Assign(box (lhs, ref rhs)) => {
589                 self.consume_rvalue(location, (rhs, span), flow_state);
590
591                 self.mutate_place(location, (*lhs, span), Shallow(None), JustWrite, flow_state);
592             }
593             StatementKind::FakeRead(_, box ref place) => {
594                 // Read for match doesn't access any memory and is used to
595                 // assert that a place is safe and live. So we don't have to
596                 // do any checks here.
597                 //
598                 // FIXME: Remove check that the place is initialized. This is
599                 // needed for now because matches don't have never patterns yet.
600                 // So this is the only place we prevent
601                 //      let x: !;
602                 //      match x {};
603                 // from compiling.
604                 self.check_if_path_or_subpath_is_moved(
605                     location,
606                     InitializationRequiringAction::Use,
607                     (place.as_ref(), span),
608                     flow_state,
609                 );
610             }
611             StatementKind::SetDiscriminant { place, variant_index: _ } => {
612                 self.mutate_place(location, (**place, span), Shallow(None), JustWrite, flow_state);
613             }
614             StatementKind::LlvmInlineAsm(ref asm) => {
615                 for (o, output) in asm.asm.outputs.iter().zip(asm.outputs.iter()) {
616                     if o.is_indirect {
617                         // FIXME(eddyb) indirect inline asm outputs should
618                         // be encoded through MIR place derefs instead.
619                         self.access_place(
620                             location,
621                             (*output, o.span),
622                             (Deep, Read(ReadKind::Copy)),
623                             LocalMutationIsAllowed::No,
624                             flow_state,
625                         );
626                         self.check_if_path_or_subpath_is_moved(
627                             location,
628                             InitializationRequiringAction::Use,
629                             (output.as_ref(), o.span),
630                             flow_state,
631                         );
632                     } else {
633                         self.mutate_place(
634                             location,
635                             (*output, o.span),
636                             if o.is_rw { Deep } else { Shallow(None) },
637                             if o.is_rw { WriteAndRead } else { JustWrite },
638                             flow_state,
639                         );
640                     }
641                 }
642                 for (_, input) in asm.inputs.iter() {
643                     self.consume_operand(location, (input, span), flow_state);
644                 }
645             }
646             StatementKind::Nop
647             | StatementKind::AscribeUserType(..)
648             | StatementKind::Retag { .. }
649             | StatementKind::StorageLive(..) => {
650                 // `Nop`, `AscribeUserType`, `Retag`, and `StorageLive` are irrelevant
651                 // to borrow check.
652             }
653             StatementKind::StorageDead(local) => {
654                 self.access_place(
655                     location,
656                     (Place::from(*local), span),
657                     (Shallow(None), Write(WriteKind::StorageDeadOrDrop)),
658                     LocalMutationIsAllowed::Yes,
659                     flow_state,
660                 );
661             }
662         }
663     }
664
665     fn visit_terminator_before_primary_effect(
666         &mut self,
667         flow_state: &Flows<'cx, 'tcx>,
668         term: &'cx Terminator<'tcx>,
669         loc: Location,
670     ) {
671         debug!("MirBorrowckCtxt::process_terminator({:?}, {:?}): {:?}", loc, term, flow_state);
672         let span = term.source_info.span;
673
674         self.check_activations(loc, span, flow_state);
675
676         match term.kind {
677             TerminatorKind::SwitchInt { ref discr, switch_ty: _, values: _, targets: _ } => {
678                 self.consume_operand(loc, (discr, span), flow_state);
679             }
680             TerminatorKind::Drop { place: ref drop_place, target: _, unwind: _ } => {
681                 let tcx = self.infcx.tcx;
682
683                 // Compute the type with accurate region information.
684                 let drop_place_ty = drop_place.ty(self.body, self.infcx.tcx);
685
686                 // Erase the regions.
687                 let drop_place_ty = self.infcx.tcx.erase_regions(&drop_place_ty).ty;
688
689                 // "Lift" into the tcx -- once regions are erased, this type should be in the
690                 // global arenas; this "lift" operation basically just asserts that is true, but
691                 // that is useful later.
692                 tcx.lift(&drop_place_ty).unwrap();
693
694                 debug!(
695                     "visit_terminator_drop \
696                      loc: {:?} term: {:?} drop_place: {:?} drop_place_ty: {:?} span: {:?}",
697                     loc, term, drop_place, drop_place_ty, span
698                 );
699
700                 self.access_place(
701                     loc,
702                     (*drop_place, span),
703                     (AccessDepth::Drop, Write(WriteKind::StorageDeadOrDrop)),
704                     LocalMutationIsAllowed::Yes,
705                     flow_state,
706                 );
707             }
708             TerminatorKind::DropAndReplace {
709                 place: drop_place,
710                 value: ref new_value,
711                 target: _,
712                 unwind: _,
713             } => {
714                 self.mutate_place(loc, (drop_place, span), Deep, JustWrite, flow_state);
715                 self.consume_operand(loc, (new_value, span), flow_state);
716             }
717             TerminatorKind::Call {
718                 ref func,
719                 ref args,
720                 ref destination,
721                 cleanup: _,
722                 from_hir_call: _,
723                 fn_span: _,
724             } => {
725                 self.consume_operand(loc, (func, span), flow_state);
726                 for arg in args {
727                     self.consume_operand(loc, (arg, span), flow_state);
728                 }
729                 if let Some((dest, _ /*bb*/)) = *destination {
730                     self.mutate_place(loc, (dest, span), Deep, JustWrite, flow_state);
731                 }
732             }
733             TerminatorKind::Assert { ref cond, expected: _, ref msg, target: _, cleanup: _ } => {
734                 self.consume_operand(loc, (cond, span), flow_state);
735                 use rustc_middle::mir::AssertKind;
736                 if let AssertKind::BoundsCheck { ref len, ref index } = *msg {
737                     self.consume_operand(loc, (len, span), flow_state);
738                     self.consume_operand(loc, (index, span), flow_state);
739                 }
740             }
741
742             TerminatorKind::Yield { ref value, resume: _, resume_arg, drop: _ } => {
743                 self.consume_operand(loc, (value, span), flow_state);
744                 self.mutate_place(loc, (resume_arg, span), Deep, JustWrite, flow_state);
745             }
746
747             TerminatorKind::InlineAsm {
748                 template: _,
749                 ref operands,
750                 options: _,
751                 line_spans: _,
752                 destination: _,
753             } => {
754                 for op in operands {
755                     match *op {
756                         InlineAsmOperand::In { reg: _, ref value }
757                         | InlineAsmOperand::Const { ref value } => {
758                             self.consume_operand(loc, (value, span), flow_state);
759                         }
760                         InlineAsmOperand::Out { reg: _, late: _, place, .. } => {
761                             if let Some(place) = place {
762                                 self.mutate_place(
763                                     loc,
764                                     (place, span),
765                                     Shallow(None),
766                                     JustWrite,
767                                     flow_state,
768                                 );
769                             }
770                         }
771                         InlineAsmOperand::InOut { reg: _, late: _, ref in_value, out_place } => {
772                             self.consume_operand(loc, (in_value, span), flow_state);
773                             if let Some(out_place) = out_place {
774                                 self.mutate_place(
775                                     loc,
776                                     (out_place, span),
777                                     Shallow(None),
778                                     JustWrite,
779                                     flow_state,
780                                 );
781                             }
782                         }
783                         InlineAsmOperand::SymFn { value: _ }
784                         | InlineAsmOperand::SymStatic { def_id: _ } => {}
785                     }
786                 }
787             }
788
789             TerminatorKind::Goto { target: _ }
790             | TerminatorKind::Abort
791             | TerminatorKind::Unreachable
792             | TerminatorKind::Resume
793             | TerminatorKind::Return
794             | TerminatorKind::GeneratorDrop
795             | TerminatorKind::FalseEdge { real_target: _, imaginary_target: _ }
796             | TerminatorKind::FalseUnwind { real_target: _, unwind: _ } => {
797                 // no data used, thus irrelevant to borrowck
798             }
799         }
800     }
801
802     fn visit_terminator_after_primary_effect(
803         &mut self,
804         flow_state: &Flows<'cx, 'tcx>,
805         term: &'cx Terminator<'tcx>,
806         loc: Location,
807     ) {
808         let span = term.source_info.span;
809
810         match term.kind {
811             TerminatorKind::Yield { value: _, resume: _, resume_arg: _, drop: _ } => {
812                 if self.movable_generator {
813                     // Look for any active borrows to locals
814                     let borrow_set = self.borrow_set.clone();
815                     for i in flow_state.borrows.iter() {
816                         let borrow = &borrow_set[i];
817                         self.check_for_local_borrow(borrow, span);
818                     }
819                 }
820             }
821
822             TerminatorKind::Resume | TerminatorKind::Return | TerminatorKind::GeneratorDrop => {
823                 // Returning from the function implicitly kills storage for all locals and statics.
824                 // Often, the storage will already have been killed by an explicit
825                 // StorageDead, but we don't always emit those (notably on unwind paths),
826                 // so this "extra check" serves as a kind of backup.
827                 let borrow_set = self.borrow_set.clone();
828                 for i in flow_state.borrows.iter() {
829                     let borrow = &borrow_set[i];
830                     self.check_for_invalidation_at_exit(loc, borrow, span);
831                 }
832             }
833
834             TerminatorKind::Abort
835             | TerminatorKind::Assert { .. }
836             | TerminatorKind::Call { .. }
837             | TerminatorKind::Drop { .. }
838             | TerminatorKind::DropAndReplace { .. }
839             | TerminatorKind::FalseEdge { real_target: _, imaginary_target: _ }
840             | TerminatorKind::FalseUnwind { real_target: _, unwind: _ }
841             | TerminatorKind::Goto { .. }
842             | TerminatorKind::SwitchInt { .. }
843             | TerminatorKind::Unreachable
844             | TerminatorKind::InlineAsm { .. } => {}
845         }
846     }
847 }
848
849 #[derive(Copy, Clone, PartialEq, Eq, Debug)]
850 enum MutateMode {
851     JustWrite,
852     WriteAndRead,
853 }
854
855 use self::AccessDepth::{Deep, Shallow};
856 use self::ReadOrWrite::{Activation, Read, Reservation, Write};
857
858 #[derive(Copy, Clone, PartialEq, Eq, Debug)]
859 enum ArtificialField {
860     ArrayLength,
861     ShallowBorrow,
862 }
863
864 #[derive(Copy, Clone, PartialEq, Eq, Debug)]
865 enum AccessDepth {
866     /// From the RFC: "A *shallow* access means that the immediate
867     /// fields reached at P are accessed, but references or pointers
868     /// found within are not dereferenced. Right now, the only access
869     /// that is shallow is an assignment like `x = ...;`, which would
870     /// be a *shallow write* of `x`."
871     Shallow(Option<ArtificialField>),
872
873     /// From the RFC: "A *deep* access means that all data reachable
874     /// through the given place may be invalidated or accesses by
875     /// this action."
876     Deep,
877
878     /// Access is Deep only when there is a Drop implementation that
879     /// can reach the data behind the reference.
880     Drop,
881 }
882
883 /// Kind of access to a value: read or write
884 /// (For informational purposes only)
885 #[derive(Copy, Clone, PartialEq, Eq, Debug)]
886 enum ReadOrWrite {
887     /// From the RFC: "A *read* means that the existing data may be
888     /// read, but will not be changed."
889     Read(ReadKind),
890
891     /// From the RFC: "A *write* means that the data may be mutated to
892     /// new values or otherwise invalidated (for example, it could be
893     /// de-initialized, as in a move operation).
894     Write(WriteKind),
895
896     /// For two-phase borrows, we distinguish a reservation (which is treated
897     /// like a Read) from an activation (which is treated like a write), and
898     /// each of those is furthermore distinguished from Reads/Writes above.
899     Reservation(WriteKind),
900     Activation(WriteKind, BorrowIndex),
901 }
902
903 /// Kind of read access to a value
904 /// (For informational purposes only)
905 #[derive(Copy, Clone, PartialEq, Eq, Debug)]
906 enum ReadKind {
907     Borrow(BorrowKind),
908     Copy,
909 }
910
911 /// Kind of write access to a value
912 /// (For informational purposes only)
913 #[derive(Copy, Clone, PartialEq, Eq, Debug)]
914 enum WriteKind {
915     StorageDeadOrDrop,
916     MutableBorrow(BorrowKind),
917     Mutate,
918     Move,
919 }
920
921 /// When checking permissions for a place access, this flag is used to indicate that an immutable
922 /// local place can be mutated.
923 //
924 // FIXME: @nikomatsakis suggested that this flag could be removed with the following modifications:
925 // - Merge `check_access_permissions()` and `check_if_reassignment_to_immutable_state()`.
926 // - Split `is_mutable()` into `is_assignable()` (can be directly assigned) and
927 //   `is_declared_mutable()`.
928 // - Take flow state into consideration in `is_assignable()` for local variables.
929 #[derive(Copy, Clone, PartialEq, Eq, Debug)]
930 enum LocalMutationIsAllowed {
931     Yes,
932     /// We want use of immutable upvars to cause a "write to immutable upvar"
933     /// error, not an "reassignment" error.
934     ExceptUpvars,
935     No,
936 }
937
938 #[derive(Copy, Clone, Debug)]
939 enum InitializationRequiringAction {
940     Update,
941     Borrow,
942     MatchOn,
943     Use,
944     Assignment,
945     PartialAssignment,
946 }
947
948 struct RootPlace<'tcx> {
949     place_local: Local,
950     place_projection: &'tcx [PlaceElem<'tcx>],
951     is_local_mutation_allowed: LocalMutationIsAllowed,
952 }
953
954 impl InitializationRequiringAction {
955     fn as_noun(self) -> &'static str {
956         match self {
957             InitializationRequiringAction::Update => "update",
958             InitializationRequiringAction::Borrow => "borrow",
959             InitializationRequiringAction::MatchOn => "use", // no good noun
960             InitializationRequiringAction::Use => "use",
961             InitializationRequiringAction::Assignment => "assign",
962             InitializationRequiringAction::PartialAssignment => "assign to part",
963         }
964     }
965
966     fn as_verb_in_past_tense(self) -> &'static str {
967         match self {
968             InitializationRequiringAction::Update => "updated",
969             InitializationRequiringAction::Borrow => "borrowed",
970             InitializationRequiringAction::MatchOn => "matched on",
971             InitializationRequiringAction::Use => "used",
972             InitializationRequiringAction::Assignment => "assigned",
973             InitializationRequiringAction::PartialAssignment => "partially assigned",
974         }
975     }
976 }
977
978 impl<'cx, 'tcx> MirBorrowckCtxt<'cx, 'tcx> {
979     fn body(&self) -> &'cx Body<'tcx> {
980         self.body
981     }
982
983     /// Checks an access to the given place to see if it is allowed. Examines the set of borrows
984     /// that are in scope, as well as which paths have been initialized, to ensure that (a) the
985     /// place is initialized and (b) it is not borrowed in some way that would prevent this
986     /// access.
987     ///
988     /// Returns `true` if an error is reported.
989     fn access_place(
990         &mut self,
991         location: Location,
992         place_span: (Place<'tcx>, Span),
993         kind: (AccessDepth, ReadOrWrite),
994         is_local_mutation_allowed: LocalMutationIsAllowed,
995         flow_state: &Flows<'cx, 'tcx>,
996     ) {
997         let (sd, rw) = kind;
998
999         if let Activation(_, borrow_index) = rw {
1000             if self.reservation_error_reported.contains(&place_span.0) {
1001                 debug!(
1002                     "skipping access_place for activation of invalid reservation \
1003                      place: {:?} borrow_index: {:?}",
1004                     place_span.0, borrow_index
1005                 );
1006                 return;
1007             }
1008         }
1009
1010         // Check is_empty() first because it's the common case, and doing that
1011         // way we avoid the clone() call.
1012         if !self.access_place_error_reported.is_empty()
1013             && self.access_place_error_reported.contains(&(place_span.0, place_span.1))
1014         {
1015             debug!(
1016                 "access_place: suppressing error place_span=`{:?}` kind=`{:?}`",
1017                 place_span, kind
1018             );
1019             return;
1020         }
1021
1022         let mutability_error = self.check_access_permissions(
1023             place_span,
1024             rw,
1025             is_local_mutation_allowed,
1026             flow_state,
1027             location,
1028         );
1029         let conflict_error =
1030             self.check_access_for_conflict(location, place_span, sd, rw, flow_state);
1031
1032         if let (Activation(_, borrow_idx), true) = (kind.1, conflict_error) {
1033             // Suppress this warning when there's an error being emitted for the
1034             // same borrow: fixing the error is likely to fix the warning.
1035             self.reservation_warnings.remove(&borrow_idx);
1036         }
1037
1038         if conflict_error || mutability_error {
1039             debug!("access_place: logging error place_span=`{:?}` kind=`{:?}`", place_span, kind);
1040
1041             self.access_place_error_reported.insert((place_span.0, place_span.1));
1042         }
1043     }
1044
1045     fn check_access_for_conflict(
1046         &mut self,
1047         location: Location,
1048         place_span: (Place<'tcx>, Span),
1049         sd: AccessDepth,
1050         rw: ReadOrWrite,
1051         flow_state: &Flows<'cx, 'tcx>,
1052     ) -> bool {
1053         debug!(
1054             "check_access_for_conflict(location={:?}, place_span={:?}, sd={:?}, rw={:?})",
1055             location, place_span, sd, rw,
1056         );
1057
1058         let mut error_reported = false;
1059         let tcx = self.infcx.tcx;
1060         let body = self.body;
1061         let borrow_set = self.borrow_set.clone();
1062
1063         // Use polonius output if it has been enabled.
1064         let polonius_output = self.polonius_output.clone();
1065         let borrows_in_scope = if let Some(polonius) = &polonius_output {
1066             let location = self.location_table.start_index(location);
1067             Either::Left(polonius.errors_at(location).iter().copied())
1068         } else {
1069             Either::Right(flow_state.borrows.iter())
1070         };
1071
1072         each_borrow_involving_path(
1073             self,
1074             tcx,
1075             body,
1076             location,
1077             (sd, place_span.0),
1078             &borrow_set,
1079             borrows_in_scope,
1080             |this, borrow_index, borrow| match (rw, borrow.kind) {
1081                 // Obviously an activation is compatible with its own
1082                 // reservation (or even prior activating uses of same
1083                 // borrow); so don't check if they interfere.
1084                 //
1085                 // NOTE: *reservations* do conflict with themselves;
1086                 // thus aren't injecting unsoundenss w/ this check.)
1087                 (Activation(_, activating), _) if activating == borrow_index => {
1088                     debug!(
1089                         "check_access_for_conflict place_span: {:?} sd: {:?} rw: {:?} \
1090                          skipping {:?} b/c activation of same borrow_index",
1091                         place_span,
1092                         sd,
1093                         rw,
1094                         (borrow_index, borrow),
1095                     );
1096                     Control::Continue
1097                 }
1098
1099                 (Read(_), BorrowKind::Shared | BorrowKind::Shallow)
1100                 | (
1101                     Read(ReadKind::Borrow(BorrowKind::Shallow)),
1102                     BorrowKind::Unique | BorrowKind::Mut { .. },
1103                 ) => Control::Continue,
1104
1105                 (Write(WriteKind::Move), BorrowKind::Shallow) => {
1106                     // Handled by initialization checks.
1107                     Control::Continue
1108                 }
1109
1110                 (Read(kind), BorrowKind::Unique | BorrowKind::Mut { .. }) => {
1111                     // Reading from mere reservations of mutable-borrows is OK.
1112                     if !is_active(&this.dominators, borrow, location) {
1113                         assert!(allow_two_phase_borrow(borrow.kind));
1114                         return Control::Continue;
1115                     }
1116
1117                     error_reported = true;
1118                     match kind {
1119                         ReadKind::Copy => {
1120                             this.report_use_while_mutably_borrowed(location, place_span, borrow)
1121                                 .buffer(&mut this.errors_buffer);
1122                         }
1123                         ReadKind::Borrow(bk) => {
1124                             this.report_conflicting_borrow(location, place_span, bk, borrow)
1125                                 .buffer(&mut this.errors_buffer);
1126                         }
1127                     }
1128                     Control::Break
1129                 }
1130
1131                 (
1132                     Reservation(WriteKind::MutableBorrow(bk)),
1133                     BorrowKind::Shallow | BorrowKind::Shared,
1134                 ) if {
1135                     tcx.migrate_borrowck() && this.borrow_set.location_map.contains_key(&location)
1136                 } =>
1137                 {
1138                     let bi = this.borrow_set.location_map[&location];
1139                     debug!(
1140                         "recording invalid reservation of place: {:?} with \
1141                          borrow index {:?} as warning",
1142                         place_span.0, bi,
1143                     );
1144                     // rust-lang/rust#56254 - This was previously permitted on
1145                     // the 2018 edition so we emit it as a warning. We buffer
1146                     // these sepately so that we only emit a warning if borrow
1147                     // checking was otherwise successful.
1148                     this.reservation_warnings
1149                         .insert(bi, (place_span.0, place_span.1, location, bk, borrow.clone()));
1150
1151                     // Don't suppress actual errors.
1152                     Control::Continue
1153                 }
1154
1155                 (Reservation(kind) | Activation(kind, _) | Write(kind), _) => {
1156                     match rw {
1157                         Reservation(..) => {
1158                             debug!(
1159                                 "recording invalid reservation of \
1160                                  place: {:?}",
1161                                 place_span.0
1162                             );
1163                             this.reservation_error_reported.insert(place_span.0);
1164                         }
1165                         Activation(_, activating) => {
1166                             debug!(
1167                                 "observing check_place for activation of \
1168                                  borrow_index: {:?}",
1169                                 activating
1170                             );
1171                         }
1172                         Read(..) | Write(..) => {}
1173                     }
1174
1175                     error_reported = true;
1176                     match kind {
1177                         WriteKind::MutableBorrow(bk) => {
1178                             this.report_conflicting_borrow(location, place_span, bk, borrow)
1179                                 .buffer(&mut this.errors_buffer);
1180                         }
1181                         WriteKind::StorageDeadOrDrop => this
1182                             .report_borrowed_value_does_not_live_long_enough(
1183                                 location,
1184                                 borrow,
1185                                 place_span,
1186                                 Some(kind),
1187                             ),
1188                         WriteKind::Mutate => {
1189                             this.report_illegal_mutation_of_borrowed(location, place_span, borrow)
1190                         }
1191                         WriteKind::Move => {
1192                             this.report_move_out_while_borrowed(location, place_span, borrow)
1193                         }
1194                     }
1195                     Control::Break
1196                 }
1197             },
1198         );
1199
1200         error_reported
1201     }
1202
1203     fn mutate_place(
1204         &mut self,
1205         location: Location,
1206         place_span: (Place<'tcx>, Span),
1207         kind: AccessDepth,
1208         mode: MutateMode,
1209         flow_state: &Flows<'cx, 'tcx>,
1210     ) {
1211         // Write of P[i] or *P, or WriteAndRead of any P, requires P init'd.
1212         match mode {
1213             MutateMode::WriteAndRead => {
1214                 self.check_if_path_or_subpath_is_moved(
1215                     location,
1216                     InitializationRequiringAction::Update,
1217                     (place_span.0.as_ref(), place_span.1),
1218                     flow_state,
1219                 );
1220             }
1221             MutateMode::JustWrite => {
1222                 self.check_if_assigned_path_is_moved(location, place_span, flow_state);
1223             }
1224         }
1225
1226         // Special case: you can assign a immutable local variable
1227         // (e.g., `x = ...`) so long as it has never been initialized
1228         // before (at this point in the flow).
1229         if let Some(local) = place_span.0.as_local() {
1230             if let Mutability::Not = self.body.local_decls[local].mutability {
1231                 // check for reassignments to immutable local variables
1232                 self.check_if_reassignment_to_immutable_state(
1233                     location, local, place_span, flow_state,
1234                 );
1235                 return;
1236             }
1237         }
1238
1239         // Otherwise, use the normal access permission rules.
1240         self.access_place(
1241             location,
1242             place_span,
1243             (kind, Write(WriteKind::Mutate)),
1244             LocalMutationIsAllowed::No,
1245             flow_state,
1246         );
1247     }
1248
1249     fn consume_rvalue(
1250         &mut self,
1251         location: Location,
1252         (rvalue, span): (&'cx Rvalue<'tcx>, Span),
1253         flow_state: &Flows<'cx, 'tcx>,
1254     ) {
1255         match *rvalue {
1256             Rvalue::Ref(_ /*rgn*/, bk, place) => {
1257                 let access_kind = match bk {
1258                     BorrowKind::Shallow => {
1259                         (Shallow(Some(ArtificialField::ShallowBorrow)), Read(ReadKind::Borrow(bk)))
1260                     }
1261                     BorrowKind::Shared => (Deep, Read(ReadKind::Borrow(bk))),
1262                     BorrowKind::Unique | BorrowKind::Mut { .. } => {
1263                         let wk = WriteKind::MutableBorrow(bk);
1264                         if allow_two_phase_borrow(bk) {
1265                             (Deep, Reservation(wk))
1266                         } else {
1267                             (Deep, Write(wk))
1268                         }
1269                     }
1270                 };
1271
1272                 self.access_place(
1273                     location,
1274                     (place, span),
1275                     access_kind,
1276                     LocalMutationIsAllowed::No,
1277                     flow_state,
1278                 );
1279
1280                 let action = if bk == BorrowKind::Shallow {
1281                     InitializationRequiringAction::MatchOn
1282                 } else {
1283                     InitializationRequiringAction::Borrow
1284                 };
1285
1286                 self.check_if_path_or_subpath_is_moved(
1287                     location,
1288                     action,
1289                     (place.as_ref(), span),
1290                     flow_state,
1291                 );
1292             }
1293
1294             Rvalue::AddressOf(mutability, place) => {
1295                 let access_kind = match mutability {
1296                     Mutability::Mut => (
1297                         Deep,
1298                         Write(WriteKind::MutableBorrow(BorrowKind::Mut {
1299                             allow_two_phase_borrow: false,
1300                         })),
1301                     ),
1302                     Mutability::Not => (Deep, Read(ReadKind::Borrow(BorrowKind::Shared))),
1303                 };
1304
1305                 self.access_place(
1306                     location,
1307                     (place, span),
1308                     access_kind,
1309                     LocalMutationIsAllowed::No,
1310                     flow_state,
1311                 );
1312
1313                 self.check_if_path_or_subpath_is_moved(
1314                     location,
1315                     InitializationRequiringAction::Borrow,
1316                     (place.as_ref(), span),
1317                     flow_state,
1318                 );
1319             }
1320
1321             Rvalue::ThreadLocalRef(_) => {}
1322
1323             Rvalue::Use(ref operand)
1324             | Rvalue::Repeat(ref operand, _)
1325             | Rvalue::UnaryOp(_ /*un_op*/, ref operand)
1326             | Rvalue::Cast(_ /*cast_kind*/, ref operand, _ /*ty*/) => {
1327                 self.consume_operand(location, (operand, span), flow_state)
1328             }
1329
1330             Rvalue::Len(place) | Rvalue::Discriminant(place) => {
1331                 let af = match *rvalue {
1332                     Rvalue::Len(..) => Some(ArtificialField::ArrayLength),
1333                     Rvalue::Discriminant(..) => None,
1334                     _ => unreachable!(),
1335                 };
1336                 self.access_place(
1337                     location,
1338                     (place, span),
1339                     (Shallow(af), Read(ReadKind::Copy)),
1340                     LocalMutationIsAllowed::No,
1341                     flow_state,
1342                 );
1343                 self.check_if_path_or_subpath_is_moved(
1344                     location,
1345                     InitializationRequiringAction::Use,
1346                     (place.as_ref(), span),
1347                     flow_state,
1348                 );
1349             }
1350
1351             Rvalue::BinaryOp(_bin_op, ref operand1, ref operand2)
1352             | Rvalue::CheckedBinaryOp(_bin_op, ref operand1, ref operand2) => {
1353                 self.consume_operand(location, (operand1, span), flow_state);
1354                 self.consume_operand(location, (operand2, span), flow_state);
1355             }
1356
1357             Rvalue::NullaryOp(_op, _ty) => {
1358                 // nullary ops take no dynamic input; no borrowck effect.
1359                 //
1360                 // FIXME: is above actually true? Do we want to track
1361                 // the fact that uninitialized data can be created via
1362                 // `NullOp::Box`?
1363             }
1364
1365             Rvalue::Aggregate(ref aggregate_kind, ref operands) => {
1366                 // We need to report back the list of mutable upvars that were
1367                 // moved into the closure and subsequently used by the closure,
1368                 // in order to populate our used_mut set.
1369                 match **aggregate_kind {
1370                     AggregateKind::Closure(def_id, _) | AggregateKind::Generator(def_id, _, _) => {
1371                         let BorrowCheckResult { used_mut_upvars, .. } =
1372                             self.infcx.tcx.mir_borrowck(def_id.expect_local());
1373                         debug!("{:?} used_mut_upvars={:?}", def_id, used_mut_upvars);
1374                         for field in used_mut_upvars {
1375                             self.propagate_closure_used_mut_upvar(&operands[field.index()]);
1376                         }
1377                     }
1378                     AggregateKind::Adt(..)
1379                     | AggregateKind::Array(..)
1380                     | AggregateKind::Tuple { .. } => (),
1381                 }
1382
1383                 for operand in operands {
1384                     self.consume_operand(location, (operand, span), flow_state);
1385                 }
1386             }
1387         }
1388     }
1389
1390     fn propagate_closure_used_mut_upvar(&mut self, operand: &Operand<'tcx>) {
1391         let propagate_closure_used_mut_place = |this: &mut Self, place: Place<'tcx>| {
1392             if !place.projection.is_empty() {
1393                 if let Some(field) = this.is_upvar_field_projection(place.as_ref()) {
1394                     this.used_mut_upvars.push(field);
1395                 }
1396             } else {
1397                 this.used_mut.insert(place.local);
1398             }
1399         };
1400
1401         // This relies on the current way that by-value
1402         // captures of a closure are copied/moved directly
1403         // when generating MIR.
1404         match *operand {
1405             Operand::Move(place) | Operand::Copy(place) => {
1406                 match place.as_local() {
1407                     Some(local) if !self.body.local_decls[local].is_user_variable() => {
1408                         if self.body.local_decls[local].ty.is_mutable_ptr() {
1409                             // The variable will be marked as mutable by the borrow.
1410                             return;
1411                         }
1412                         // This is an edge case where we have a `move` closure
1413                         // inside a non-move closure, and the inner closure
1414                         // contains a mutation:
1415                         //
1416                         // let mut i = 0;
1417                         // || { move || { i += 1; }; };
1418                         //
1419                         // In this case our usual strategy of assuming that the
1420                         // variable will be captured by mutable reference is
1421                         // wrong, since `i` can be copied into the inner
1422                         // closure from a shared reference.
1423                         //
1424                         // As such we have to search for the local that this
1425                         // capture comes from and mark it as being used as mut.
1426
1427                         let temp_mpi = self.move_data.rev_lookup.find_local(local);
1428                         let init = if let [init_index] = *self.move_data.init_path_map[temp_mpi] {
1429                             &self.move_data.inits[init_index]
1430                         } else {
1431                             bug!("temporary should be initialized exactly once")
1432                         };
1433
1434                         let loc = match init.location {
1435                             InitLocation::Statement(stmt) => stmt,
1436                             _ => bug!("temporary initialized in arguments"),
1437                         };
1438
1439                         let body = self.body;
1440                         let bbd = &body[loc.block];
1441                         let stmt = &bbd.statements[loc.statement_index];
1442                         debug!("temporary assigned in: stmt={:?}", stmt);
1443
1444                         if let StatementKind::Assign(box (_, Rvalue::Ref(_, _, source))) = stmt.kind
1445                         {
1446                             propagate_closure_used_mut_place(self, source);
1447                         } else {
1448                             bug!(
1449                                 "closures should only capture user variables \
1450                                  or references to user variables"
1451                             );
1452                         }
1453                     }
1454                     _ => propagate_closure_used_mut_place(self, place),
1455                 }
1456             }
1457             Operand::Constant(..) => {}
1458         }
1459     }
1460
1461     fn consume_operand(
1462         &mut self,
1463         location: Location,
1464         (operand, span): (&'cx Operand<'tcx>, Span),
1465         flow_state: &Flows<'cx, 'tcx>,
1466     ) {
1467         match *operand {
1468             Operand::Copy(place) => {
1469                 // copy of place: check if this is "copy of frozen path"
1470                 // (FIXME: see check_loans.rs)
1471                 self.access_place(
1472                     location,
1473                     (place, span),
1474                     (Deep, Read(ReadKind::Copy)),
1475                     LocalMutationIsAllowed::No,
1476                     flow_state,
1477                 );
1478
1479                 // Finally, check if path was already moved.
1480                 self.check_if_path_or_subpath_is_moved(
1481                     location,
1482                     InitializationRequiringAction::Use,
1483                     (place.as_ref(), span),
1484                     flow_state,
1485                 );
1486             }
1487             Operand::Move(place) => {
1488                 // move of place: check if this is move of already borrowed path
1489                 self.access_place(
1490                     location,
1491                     (place, span),
1492                     (Deep, Write(WriteKind::Move)),
1493                     LocalMutationIsAllowed::Yes,
1494                     flow_state,
1495                 );
1496
1497                 // Finally, check if path was already moved.
1498                 self.check_if_path_or_subpath_is_moved(
1499                     location,
1500                     InitializationRequiringAction::Use,
1501                     (place.as_ref(), span),
1502                     flow_state,
1503                 );
1504             }
1505             Operand::Constant(_) => {}
1506         }
1507     }
1508
1509     /// Checks whether a borrow of this place is invalidated when the function
1510     /// exits
1511     fn check_for_invalidation_at_exit(
1512         &mut self,
1513         location: Location,
1514         borrow: &BorrowData<'tcx>,
1515         span: Span,
1516     ) {
1517         debug!("check_for_invalidation_at_exit({:?})", borrow);
1518         let place = borrow.borrowed_place;
1519         let mut root_place = PlaceRef { local: place.local, projection: &[] };
1520
1521         // FIXME(nll-rfc#40): do more precise destructor tracking here. For now
1522         // we just know that all locals are dropped at function exit (otherwise
1523         // we'll have a memory leak) and assume that all statics have a destructor.
1524         //
1525         // FIXME: allow thread-locals to borrow other thread locals?
1526
1527         let (might_be_alive, will_be_dropped) =
1528             if self.body.local_decls[root_place.local].is_ref_to_thread_local() {
1529                 // Thread-locals might be dropped after the function exits
1530                 // We have to dereference the outer reference because
1531                 // borrows don't conflict behind shared references.
1532                 root_place.projection = DEREF_PROJECTION;
1533                 (true, true)
1534             } else {
1535                 (false, self.locals_are_invalidated_at_exit)
1536             };
1537
1538         if !will_be_dropped {
1539             debug!("place_is_invalidated_at_exit({:?}) - won't be dropped", place);
1540             return;
1541         }
1542
1543         let sd = if might_be_alive { Deep } else { Shallow(None) };
1544
1545         if places_conflict::borrow_conflicts_with_place(
1546             self.infcx.tcx,
1547             &self.body,
1548             place,
1549             borrow.kind,
1550             root_place,
1551             sd,
1552             places_conflict::PlaceConflictBias::Overlap,
1553         ) {
1554             debug!("check_for_invalidation_at_exit({:?}): INVALID", place);
1555             // FIXME: should be talking about the region lifetime instead
1556             // of just a span here.
1557             let span = self.infcx.tcx.sess.source_map().end_point(span);
1558             self.report_borrowed_value_does_not_live_long_enough(
1559                 location,
1560                 borrow,
1561                 (place, span),
1562                 None,
1563             )
1564         }
1565     }
1566
1567     /// Reports an error if this is a borrow of local data.
1568     /// This is called for all Yield expressions on movable generators
1569     fn check_for_local_borrow(&mut self, borrow: &BorrowData<'tcx>, yield_span: Span) {
1570         debug!("check_for_local_borrow({:?})", borrow);
1571
1572         if borrow_of_local_data(borrow.borrowed_place) {
1573             let err = self.cannot_borrow_across_generator_yield(
1574                 self.retrieve_borrow_spans(borrow).var_or_use(),
1575                 yield_span,
1576             );
1577
1578             err.buffer(&mut self.errors_buffer);
1579         }
1580     }
1581
1582     fn check_activations(&mut self, location: Location, span: Span, flow_state: &Flows<'cx, 'tcx>) {
1583         // Two-phase borrow support: For each activation that is newly
1584         // generated at this statement, check if it interferes with
1585         // another borrow.
1586         let borrow_set = self.borrow_set.clone();
1587         for &borrow_index in borrow_set.activations_at_location(location) {
1588             let borrow = &borrow_set[borrow_index];
1589
1590             // only mutable borrows should be 2-phase
1591             assert!(match borrow.kind {
1592                 BorrowKind::Shared | BorrowKind::Shallow => false,
1593                 BorrowKind::Unique | BorrowKind::Mut { .. } => true,
1594             });
1595
1596             self.access_place(
1597                 location,
1598                 (borrow.borrowed_place, span),
1599                 (Deep, Activation(WriteKind::MutableBorrow(borrow.kind), borrow_index)),
1600                 LocalMutationIsAllowed::No,
1601                 flow_state,
1602             );
1603             // We do not need to call `check_if_path_or_subpath_is_moved`
1604             // again, as we already called it when we made the
1605             // initial reservation.
1606         }
1607     }
1608
1609     fn check_if_reassignment_to_immutable_state(
1610         &mut self,
1611         location: Location,
1612         local: Local,
1613         place_span: (Place<'tcx>, Span),
1614         flow_state: &Flows<'cx, 'tcx>,
1615     ) {
1616         debug!("check_if_reassignment_to_immutable_state({:?})", local);
1617
1618         // Check if any of the initializiations of `local` have happened yet:
1619         if let Some(init_index) = self.is_local_ever_initialized(local, flow_state) {
1620             // And, if so, report an error.
1621             let init = &self.move_data.inits[init_index];
1622             let span = init.span(&self.body);
1623             self.report_illegal_reassignment(location, place_span, span, place_span.0);
1624         }
1625     }
1626
1627     fn check_if_full_path_is_moved(
1628         &mut self,
1629         location: Location,
1630         desired_action: InitializationRequiringAction,
1631         place_span: (PlaceRef<'tcx>, Span),
1632         flow_state: &Flows<'cx, 'tcx>,
1633     ) {
1634         let maybe_uninits = &flow_state.uninits;
1635
1636         // Bad scenarios:
1637         //
1638         // 1. Move of `a.b.c`, use of `a.b.c`
1639         // 2. Move of `a.b.c`, use of `a.b.c.d` (without first reinitializing `a.b.c.d`)
1640         // 3. Uninitialized `(a.b.c: &_)`, use of `*a.b.c`; note that with
1641         //    partial initialization support, one might have `a.x`
1642         //    initialized but not `a.b`.
1643         //
1644         // OK scenarios:
1645         //
1646         // 4. Move of `a.b.c`, use of `a.b.d`
1647         // 5. Uninitialized `a.x`, initialized `a.b`, use of `a.b`
1648         // 6. Copied `(a.b: &_)`, use of `*(a.b).c`; note that `a.b`
1649         //    must have been initialized for the use to be sound.
1650         // 7. Move of `a.b.c` then reinit of `a.b.c.d`, use of `a.b.c.d`
1651
1652         // The dataflow tracks shallow prefixes distinctly (that is,
1653         // field-accesses on P distinctly from P itself), in order to
1654         // track substructure initialization separately from the whole
1655         // structure.
1656         //
1657         // E.g., when looking at (*a.b.c).d, if the closest prefix for
1658         // which we have a MovePath is `a.b`, then that means that the
1659         // initialization state of `a.b` is all we need to inspect to
1660         // know if `a.b.c` is valid (and from that we infer that the
1661         // dereference and `.d` access is also valid, since we assume
1662         // `a.b.c` is assigned a reference to a initialized and
1663         // well-formed record structure.)
1664
1665         // Therefore, if we seek out the *closest* prefix for which we
1666         // have a MovePath, that should capture the initialization
1667         // state for the place scenario.
1668         //
1669         // This code covers scenarios 1, 2, and 3.
1670
1671         debug!("check_if_full_path_is_moved place: {:?}", place_span.0);
1672         let (prefix, mpi) = self.move_path_closest_to(place_span.0);
1673         if maybe_uninits.contains(mpi) {
1674             self.report_use_of_moved_or_uninitialized(
1675                 location,
1676                 desired_action,
1677                 (prefix, place_span.0, place_span.1),
1678                 mpi,
1679             );
1680         } // Only query longest prefix with a MovePath, not further
1681         // ancestors; dataflow recurs on children when parents
1682         // move (to support partial (re)inits).
1683         //
1684         // (I.e., querying parents breaks scenario 7; but may want
1685         // to do such a query based on partial-init feature-gate.)
1686     }
1687
1688     /// Subslices correspond to multiple move paths, so we iterate through the
1689     /// elements of the base array. For each element we check
1690     ///
1691     /// * Does this element overlap with our slice.
1692     /// * Is any part of it uninitialized.
1693     fn check_if_subslice_element_is_moved(
1694         &mut self,
1695         location: Location,
1696         desired_action: InitializationRequiringAction,
1697         place_span: (PlaceRef<'tcx>, Span),
1698         maybe_uninits: &BitSet<MovePathIndex>,
1699         from: u32,
1700         to: u32,
1701     ) {
1702         if let Some(mpi) = self.move_path_for_place(place_span.0) {
1703             let move_paths = &self.move_data.move_paths;
1704
1705             let root_path = &move_paths[mpi];
1706             for (child_mpi, child_move_path) in root_path.children(move_paths) {
1707                 let last_proj = child_move_path.place.projection.last().unwrap();
1708                 if let ProjectionElem::ConstantIndex { offset, from_end, .. } = last_proj {
1709                     debug_assert!(!from_end, "Array constant indexing shouldn't be `from_end`.");
1710
1711                     if (from..to).contains(offset) {
1712                         let uninit_child =
1713                             self.move_data.find_in_move_path_or_its_descendants(child_mpi, |mpi| {
1714                                 maybe_uninits.contains(mpi)
1715                             });
1716
1717                         if let Some(uninit_child) = uninit_child {
1718                             self.report_use_of_moved_or_uninitialized(
1719                                 location,
1720                                 desired_action,
1721                                 (place_span.0, place_span.0, place_span.1),
1722                                 uninit_child,
1723                             );
1724                             return; // don't bother finding other problems.
1725                         }
1726                     }
1727                 }
1728             }
1729         }
1730     }
1731
1732     fn check_if_path_or_subpath_is_moved(
1733         &mut self,
1734         location: Location,
1735         desired_action: InitializationRequiringAction,
1736         place_span: (PlaceRef<'tcx>, Span),
1737         flow_state: &Flows<'cx, 'tcx>,
1738     ) {
1739         let maybe_uninits = &flow_state.uninits;
1740
1741         // Bad scenarios:
1742         //
1743         // 1. Move of `a.b.c`, use of `a` or `a.b`
1744         //    partial initialization support, one might have `a.x`
1745         //    initialized but not `a.b`.
1746         // 2. All bad scenarios from `check_if_full_path_is_moved`
1747         //
1748         // OK scenarios:
1749         //
1750         // 3. Move of `a.b.c`, use of `a.b.d`
1751         // 4. Uninitialized `a.x`, initialized `a.b`, use of `a.b`
1752         // 5. Copied `(a.b: &_)`, use of `*(a.b).c`; note that `a.b`
1753         //    must have been initialized for the use to be sound.
1754         // 6. Move of `a.b.c` then reinit of `a.b.c.d`, use of `a.b.c.d`
1755
1756         self.check_if_full_path_is_moved(location, desired_action, place_span, flow_state);
1757
1758         if let [base_proj @ .., ProjectionElem::Subslice { from, to, from_end: false }] =
1759             place_span.0.projection
1760         {
1761             let place_ty =
1762                 Place::ty_from(place_span.0.local, base_proj, self.body(), self.infcx.tcx);
1763             if let ty::Array(..) = place_ty.ty.kind {
1764                 let array_place = PlaceRef { local: place_span.0.local, projection: base_proj };
1765                 self.check_if_subslice_element_is_moved(
1766                     location,
1767                     desired_action,
1768                     (array_place, place_span.1),
1769                     maybe_uninits,
1770                     *from,
1771                     *to,
1772                 );
1773                 return;
1774             }
1775         }
1776
1777         // A move of any shallow suffix of `place` also interferes
1778         // with an attempt to use `place`. This is scenario 3 above.
1779         //
1780         // (Distinct from handling of scenarios 1+2+4 above because
1781         // `place` does not interfere with suffixes of its prefixes,
1782         // e.g., `a.b.c` does not interfere with `a.b.d`)
1783         //
1784         // This code covers scenario 1.
1785
1786         debug!("check_if_path_or_subpath_is_moved place: {:?}", place_span.0);
1787         if let Some(mpi) = self.move_path_for_place(place_span.0) {
1788             let uninit_mpi = self
1789                 .move_data
1790                 .find_in_move_path_or_its_descendants(mpi, |mpi| maybe_uninits.contains(mpi));
1791
1792             if let Some(uninit_mpi) = uninit_mpi {
1793                 self.report_use_of_moved_or_uninitialized(
1794                     location,
1795                     desired_action,
1796                     (place_span.0, place_span.0, place_span.1),
1797                     uninit_mpi,
1798                 );
1799                 return; // don't bother finding other problems.
1800             }
1801         }
1802     }
1803
1804     /// Currently MoveData does not store entries for all places in
1805     /// the input MIR. For example it will currently filter out
1806     /// places that are Copy; thus we do not track places of shared
1807     /// reference type. This routine will walk up a place along its
1808     /// prefixes, searching for a foundational place that *is*
1809     /// tracked in the MoveData.
1810     ///
1811     /// An Err result includes a tag indicated why the search failed.
1812     /// Currently this can only occur if the place is built off of a
1813     /// static variable, as we do not track those in the MoveData.
1814     fn move_path_closest_to(&mut self, place: PlaceRef<'tcx>) -> (PlaceRef<'tcx>, MovePathIndex) {
1815         match self.move_data.rev_lookup.find(place) {
1816             LookupResult::Parent(Some(mpi)) | LookupResult::Exact(mpi) => {
1817                 (self.move_data.move_paths[mpi].place.as_ref(), mpi)
1818             }
1819             LookupResult::Parent(None) => panic!("should have move path for every Local"),
1820         }
1821     }
1822
1823     fn move_path_for_place(&mut self, place: PlaceRef<'tcx>) -> Option<MovePathIndex> {
1824         // If returns None, then there is no move path corresponding
1825         // to a direct owner of `place` (which means there is nothing
1826         // that borrowck tracks for its analysis).
1827
1828         match self.move_data.rev_lookup.find(place) {
1829             LookupResult::Parent(_) => None,
1830             LookupResult::Exact(mpi) => Some(mpi),
1831         }
1832     }
1833
1834     fn check_if_assigned_path_is_moved(
1835         &mut self,
1836         location: Location,
1837         (place, span): (Place<'tcx>, Span),
1838         flow_state: &Flows<'cx, 'tcx>,
1839     ) {
1840         debug!("check_if_assigned_path_is_moved place: {:?}", place);
1841
1842         // None case => assigning to `x` does not require `x` be initialized.
1843         let mut cursor = &*place.projection.as_ref();
1844         while let [proj_base @ .., elem] = cursor {
1845             cursor = proj_base;
1846
1847             match elem {
1848                 ProjectionElem::Index(_/*operand*/) |
1849                 ProjectionElem::ConstantIndex { .. } |
1850                 // assigning to P[i] requires P to be valid.
1851                 ProjectionElem::Downcast(_/*adt_def*/, _/*variant_idx*/) =>
1852                 // assigning to (P->variant) is okay if assigning to `P` is okay
1853                 //
1854                 // FIXME: is this true even if P is a adt with a dtor?
1855                 { }
1856
1857                 // assigning to (*P) requires P to be initialized
1858                 ProjectionElem::Deref => {
1859                     self.check_if_full_path_is_moved(
1860                         location, InitializationRequiringAction::Use,
1861                         (PlaceRef {
1862                             local: place.local,
1863                             projection: proj_base,
1864                         }, span), flow_state);
1865                     // (base initialized; no need to
1866                     // recur further)
1867                     break;
1868                 }
1869
1870                 ProjectionElem::Subslice { .. } => {
1871                     panic!("we don't allow assignments to subslices, location: {:?}",
1872                            location);
1873                 }
1874
1875                 ProjectionElem::Field(..) => {
1876                     // if type of `P` has a dtor, then
1877                     // assigning to `P.f` requires `P` itself
1878                     // be already initialized
1879                     let tcx = self.infcx.tcx;
1880                     let base_ty = Place::ty_from(place.local, proj_base, self.body(), tcx).ty;
1881                     match base_ty.kind {
1882                         ty::Adt(def, _) if def.has_dtor(tcx) => {
1883                             self.check_if_path_or_subpath_is_moved(
1884                                 location, InitializationRequiringAction::Assignment,
1885                                 (PlaceRef {
1886                                     local: place.local,
1887                                     projection: proj_base,
1888                                 }, span), flow_state);
1889
1890                             // (base initialized; no need to
1891                             // recur further)
1892                             break;
1893                         }
1894
1895                         // Once `let s; s.x = V; read(s.x);`,
1896                         // is allowed, remove this match arm.
1897                         ty::Adt(..) | ty::Tuple(..) => {
1898                             check_parent_of_field(self, location, PlaceRef {
1899                                 local: place.local,
1900                                 projection: proj_base,
1901                             }, span, flow_state);
1902
1903                             // rust-lang/rust#21232, #54499, #54986: during period where we reject
1904                             // partial initialization, do not complain about unnecessary `mut` on
1905                             // an attempt to do a partial initialization.
1906                             self.used_mut.insert(place.local);
1907                         }
1908
1909                         _ => {}
1910                     }
1911                 }
1912             }
1913         }
1914
1915         fn check_parent_of_field<'cx, 'tcx>(
1916             this: &mut MirBorrowckCtxt<'cx, 'tcx>,
1917             location: Location,
1918             base: PlaceRef<'tcx>,
1919             span: Span,
1920             flow_state: &Flows<'cx, 'tcx>,
1921         ) {
1922             // rust-lang/rust#21232: Until Rust allows reads from the
1923             // initialized parts of partially initialized structs, we
1924             // will, starting with the 2018 edition, reject attempts
1925             // to write to structs that are not fully initialized.
1926             //
1927             // In other words, *until* we allow this:
1928             //
1929             // 1. `let mut s; s.x = Val; read(s.x);`
1930             //
1931             // we will for now disallow this:
1932             //
1933             // 2. `let mut s; s.x = Val;`
1934             //
1935             // and also this:
1936             //
1937             // 3. `let mut s = ...; drop(s); s.x=Val;`
1938             //
1939             // This does not use check_if_path_or_subpath_is_moved,
1940             // because we want to *allow* reinitializations of fields:
1941             // e.g., want to allow
1942             //
1943             // `let mut s = ...; drop(s.x); s.x=Val;`
1944             //
1945             // This does not use check_if_full_path_is_moved on
1946             // `base`, because that would report an error about the
1947             // `base` as a whole, but in this scenario we *really*
1948             // want to report an error about the actual thing that was
1949             // moved, which may be some prefix of `base`.
1950
1951             // Shallow so that we'll stop at any dereference; we'll
1952             // report errors about issues with such bases elsewhere.
1953             let maybe_uninits = &flow_state.uninits;
1954
1955             // Find the shortest uninitialized prefix you can reach
1956             // without going over a Deref.
1957             let mut shortest_uninit_seen = None;
1958             for prefix in this.prefixes(base, PrefixSet::Shallow) {
1959                 let mpi = match this.move_path_for_place(prefix) {
1960                     Some(mpi) => mpi,
1961                     None => continue,
1962                 };
1963
1964                 if maybe_uninits.contains(mpi) {
1965                     debug!(
1966                         "check_parent_of_field updating shortest_uninit_seen from {:?} to {:?}",
1967                         shortest_uninit_seen,
1968                         Some((prefix, mpi))
1969                     );
1970                     shortest_uninit_seen = Some((prefix, mpi));
1971                 } else {
1972                     debug!("check_parent_of_field {:?} is definitely initialized", (prefix, mpi));
1973                 }
1974             }
1975
1976             if let Some((prefix, mpi)) = shortest_uninit_seen {
1977                 // Check for a reassignment into a uninitialized field of a union (for example,
1978                 // after a move out). In this case, do not report a error here. There is an
1979                 // exception, if this is the first assignment into the union (that is, there is
1980                 // no move out from an earlier location) then this is an attempt at initialization
1981                 // of the union - we should error in that case.
1982                 let tcx = this.infcx.tcx;
1983                 if let ty::Adt(def, _) =
1984                     Place::ty_from(base.local, base.projection, this.body(), tcx).ty.kind
1985                 {
1986                     if def.is_union() {
1987                         if this.move_data.path_map[mpi].iter().any(|moi| {
1988                             this.move_data.moves[*moi].source.is_predecessor_of(location, this.body)
1989                         }) {
1990                             return;
1991                         }
1992                     }
1993                 }
1994
1995                 this.report_use_of_moved_or_uninitialized(
1996                     location,
1997                     InitializationRequiringAction::PartialAssignment,
1998                     (prefix, base, span),
1999                     mpi,
2000                 );
2001             }
2002         }
2003     }
2004
2005     /// Checks the permissions for the given place and read or write kind
2006     ///
2007     /// Returns `true` if an error is reported.
2008     fn check_access_permissions(
2009         &mut self,
2010         (place, span): (Place<'tcx>, Span),
2011         kind: ReadOrWrite,
2012         is_local_mutation_allowed: LocalMutationIsAllowed,
2013         flow_state: &Flows<'cx, 'tcx>,
2014         location: Location,
2015     ) -> bool {
2016         debug!(
2017             "check_access_permissions({:?}, {:?}, is_local_mutation_allowed: {:?})",
2018             place, kind, is_local_mutation_allowed
2019         );
2020
2021         let error_access;
2022         let the_place_err;
2023
2024         match kind {
2025             Reservation(WriteKind::MutableBorrow(
2026                 borrow_kind @ (BorrowKind::Unique | BorrowKind::Mut { .. }),
2027             ))
2028             | Write(WriteKind::MutableBorrow(
2029                 borrow_kind @ (BorrowKind::Unique | BorrowKind::Mut { .. }),
2030             )) => {
2031                 let is_local_mutation_allowed = match borrow_kind {
2032                     BorrowKind::Unique => LocalMutationIsAllowed::Yes,
2033                     BorrowKind::Mut { .. } => is_local_mutation_allowed,
2034                     BorrowKind::Shared | BorrowKind::Shallow => unreachable!(),
2035                 };
2036                 match self.is_mutable(place.as_ref(), is_local_mutation_allowed) {
2037                     Ok(root_place) => {
2038                         self.add_used_mut(root_place, flow_state);
2039                         return false;
2040                     }
2041                     Err(place_err) => {
2042                         error_access = AccessKind::MutableBorrow;
2043                         the_place_err = place_err;
2044                     }
2045                 }
2046             }
2047             Reservation(WriteKind::Mutate) | Write(WriteKind::Mutate) => {
2048                 match self.is_mutable(place.as_ref(), is_local_mutation_allowed) {
2049                     Ok(root_place) => {
2050                         self.add_used_mut(root_place, flow_state);
2051                         return false;
2052                     }
2053                     Err(place_err) => {
2054                         error_access = AccessKind::Mutate;
2055                         the_place_err = place_err;
2056                     }
2057                 }
2058             }
2059
2060             Reservation(
2061                 WriteKind::Move
2062                 | WriteKind::StorageDeadOrDrop
2063                 | WriteKind::MutableBorrow(BorrowKind::Shared)
2064                 | WriteKind::MutableBorrow(BorrowKind::Shallow),
2065             )
2066             | Write(
2067                 WriteKind::Move
2068                 | WriteKind::StorageDeadOrDrop
2069                 | WriteKind::MutableBorrow(BorrowKind::Shared)
2070                 | WriteKind::MutableBorrow(BorrowKind::Shallow),
2071             ) => {
2072                 if let (Err(_), true) = (
2073                     self.is_mutable(place.as_ref(), is_local_mutation_allowed),
2074                     self.errors_buffer.is_empty(),
2075                 ) {
2076                     // rust-lang/rust#46908: In pure NLL mode this code path should be
2077                     // unreachable, but we use `delay_span_bug` because we can hit this when
2078                     // dereferencing a non-Copy raw pointer *and* have `-Ztreat-err-as-bug`
2079                     // enabled. We don't want to ICE for that case, as other errors will have
2080                     // been emitted (#52262).
2081                     self.infcx.tcx.sess.delay_span_bug(
2082                         span,
2083                         &format!(
2084                             "Accessing `{:?}` with the kind `{:?}` shouldn't be possible",
2085                             place, kind,
2086                         ),
2087                     );
2088                 }
2089                 return false;
2090             }
2091             Activation(..) => {
2092                 // permission checks are done at Reservation point.
2093                 return false;
2094             }
2095             Read(
2096                 ReadKind::Borrow(
2097                     BorrowKind::Unique
2098                     | BorrowKind::Mut { .. }
2099                     | BorrowKind::Shared
2100                     | BorrowKind::Shallow,
2101                 )
2102                 | ReadKind::Copy,
2103             ) => {
2104                 // Access authorized
2105                 return false;
2106             }
2107         }
2108
2109         // rust-lang/rust#21232, #54986: during period where we reject
2110         // partial initialization, do not complain about mutability
2111         // errors except for actual mutation (as opposed to an attempt
2112         // to do a partial initialization).
2113         let previously_initialized =
2114             self.is_local_ever_initialized(place.local, flow_state).is_some();
2115
2116         // at this point, we have set up the error reporting state.
2117         if previously_initialized {
2118             self.report_mutability_error(place, span, the_place_err, error_access, location);
2119             true
2120         } else {
2121             false
2122         }
2123     }
2124
2125     fn is_local_ever_initialized(
2126         &self,
2127         local: Local,
2128         flow_state: &Flows<'cx, 'tcx>,
2129     ) -> Option<InitIndex> {
2130         let mpi = self.move_data.rev_lookup.find_local(local);
2131         let ii = &self.move_data.init_path_map[mpi];
2132         for &index in ii {
2133             if flow_state.ever_inits.contains(index) {
2134                 return Some(index);
2135             }
2136         }
2137         None
2138     }
2139
2140     /// Adds the place into the used mutable variables set
2141     fn add_used_mut(&mut self, root_place: RootPlace<'tcx>, flow_state: &Flows<'cx, 'tcx>) {
2142         match root_place {
2143             RootPlace { place_local: local, place_projection: [], is_local_mutation_allowed } => {
2144                 // If the local may have been initialized, and it is now currently being
2145                 // mutated, then it is justified to be annotated with the `mut`
2146                 // keyword, since the mutation may be a possible reassignment.
2147                 if is_local_mutation_allowed != LocalMutationIsAllowed::Yes
2148                     && self.is_local_ever_initialized(local, flow_state).is_some()
2149                 {
2150                     self.used_mut.insert(local);
2151                 }
2152             }
2153             RootPlace {
2154                 place_local: _,
2155                 place_projection: _,
2156                 is_local_mutation_allowed: LocalMutationIsAllowed::Yes,
2157             } => {}
2158             RootPlace {
2159                 place_local,
2160                 place_projection: place_projection @ [.., _],
2161                 is_local_mutation_allowed: _,
2162             } => {
2163                 if let Some(field) = self.is_upvar_field_projection(PlaceRef {
2164                     local: place_local,
2165                     projection: place_projection,
2166                 }) {
2167                     self.used_mut_upvars.push(field);
2168                 }
2169             }
2170         }
2171     }
2172
2173     /// Whether this value can be written or borrowed mutably.
2174     /// Returns the root place if the place passed in is a projection.
2175     fn is_mutable(
2176         &self,
2177         place: PlaceRef<'tcx>,
2178         is_local_mutation_allowed: LocalMutationIsAllowed,
2179     ) -> Result<RootPlace<'tcx>, PlaceRef<'tcx>> {
2180         match place {
2181             PlaceRef { local, projection: [] } => {
2182                 let local = &self.body.local_decls[local];
2183                 match local.mutability {
2184                     Mutability::Not => match is_local_mutation_allowed {
2185                         LocalMutationIsAllowed::Yes => Ok(RootPlace {
2186                             place_local: place.local,
2187                             place_projection: place.projection,
2188                             is_local_mutation_allowed: LocalMutationIsAllowed::Yes,
2189                         }),
2190                         LocalMutationIsAllowed::ExceptUpvars => Ok(RootPlace {
2191                             place_local: place.local,
2192                             place_projection: place.projection,
2193                             is_local_mutation_allowed: LocalMutationIsAllowed::ExceptUpvars,
2194                         }),
2195                         LocalMutationIsAllowed::No => Err(place),
2196                     },
2197                     Mutability::Mut => Ok(RootPlace {
2198                         place_local: place.local,
2199                         place_projection: place.projection,
2200                         is_local_mutation_allowed,
2201                     }),
2202                 }
2203             }
2204             PlaceRef { local: _, projection: [proj_base @ .., elem] } => {
2205                 match elem {
2206                     ProjectionElem::Deref => {
2207                         let base_ty =
2208                             Place::ty_from(place.local, proj_base, self.body(), self.infcx.tcx).ty;
2209
2210                         // Check the kind of deref to decide
2211                         match base_ty.kind {
2212                             ty::Ref(_, _, mutbl) => {
2213                                 match mutbl {
2214                                     // Shared borrowed data is never mutable
2215                                     hir::Mutability::Not => Err(place),
2216                                     // Mutably borrowed data is mutable, but only if we have a
2217                                     // unique path to the `&mut`
2218                                     hir::Mutability::Mut => {
2219                                         let mode = match self.is_upvar_field_projection(place) {
2220                                             Some(field) if self.upvars[field.index()].by_ref => {
2221                                                 is_local_mutation_allowed
2222                                             }
2223                                             _ => LocalMutationIsAllowed::Yes,
2224                                         };
2225
2226                                         self.is_mutable(
2227                                             PlaceRef { local: place.local, projection: proj_base },
2228                                             mode,
2229                                         )
2230                                     }
2231                                 }
2232                             }
2233                             ty::RawPtr(tnm) => {
2234                                 match tnm.mutbl {
2235                                     // `*const` raw pointers are not mutable
2236                                     hir::Mutability::Not => Err(place),
2237                                     // `*mut` raw pointers are always mutable, regardless of
2238                                     // context. The users have to check by themselves.
2239                                     hir::Mutability::Mut => Ok(RootPlace {
2240                                         place_local: place.local,
2241                                         place_projection: place.projection,
2242                                         is_local_mutation_allowed,
2243                                     }),
2244                                 }
2245                             }
2246                             // `Box<T>` owns its content, so mutable if its location is mutable
2247                             _ if base_ty.is_box() => self.is_mutable(
2248                                 PlaceRef { local: place.local, projection: proj_base },
2249                                 is_local_mutation_allowed,
2250                             ),
2251                             // Deref should only be for reference, pointers or boxes
2252                             _ => bug!("Deref of unexpected type: {:?}", base_ty),
2253                         }
2254                     }
2255                     // All other projections are owned by their base path, so mutable if
2256                     // base path is mutable
2257                     ProjectionElem::Field(..)
2258                     | ProjectionElem::Index(..)
2259                     | ProjectionElem::ConstantIndex { .. }
2260                     | ProjectionElem::Subslice { .. }
2261                     | ProjectionElem::Downcast(..) => {
2262                         let upvar_field_projection = self.is_upvar_field_projection(place);
2263                         if let Some(field) = upvar_field_projection {
2264                             let upvar = &self.upvars[field.index()];
2265                             debug!(
2266                                 "upvar.mutability={:?} local_mutation_is_allowed={:?} \
2267                                  place={:?}",
2268                                 upvar, is_local_mutation_allowed, place
2269                             );
2270                             match (upvar.mutability, is_local_mutation_allowed) {
2271                                 (
2272                                     Mutability::Not,
2273                                     LocalMutationIsAllowed::No
2274                                     | LocalMutationIsAllowed::ExceptUpvars,
2275                                 ) => Err(place),
2276                                 (Mutability::Not, LocalMutationIsAllowed::Yes)
2277                                 | (Mutability::Mut, _) => {
2278                                     // Subtle: this is an upvar
2279                                     // reference, so it looks like
2280                                     // `self.foo` -- we want to double
2281                                     // check that the location `*self`
2282                                     // is mutable (i.e., this is not a
2283                                     // `Fn` closure).  But if that
2284                                     // check succeeds, we want to
2285                                     // *blame* the mutability on
2286                                     // `place` (that is,
2287                                     // `self.foo`). This is used to
2288                                     // propagate the info about
2289                                     // whether mutability declarations
2290                                     // are used outwards, so that we register
2291                                     // the outer variable as mutable. Otherwise a
2292                                     // test like this fails to record the `mut`
2293                                     // as needed:
2294                                     //
2295                                     // ```
2296                                     // fn foo<F: FnOnce()>(_f: F) { }
2297                                     // fn main() {
2298                                     //     let var = Vec::new();
2299                                     //     foo(move || {
2300                                     //         var.push(1);
2301                                     //     });
2302                                     // }
2303                                     // ```
2304                                     let _ = self.is_mutable(
2305                                         PlaceRef { local: place.local, projection: proj_base },
2306                                         is_local_mutation_allowed,
2307                                     )?;
2308                                     Ok(RootPlace {
2309                                         place_local: place.local,
2310                                         place_projection: place.projection,
2311                                         is_local_mutation_allowed,
2312                                     })
2313                                 }
2314                             }
2315                         } else {
2316                             self.is_mutable(
2317                                 PlaceRef { local: place.local, projection: proj_base },
2318                                 is_local_mutation_allowed,
2319                             )
2320                         }
2321                     }
2322                 }
2323             }
2324         }
2325     }
2326
2327     /// If `place` is a field projection, and the field is being projected from a closure type,
2328     /// then returns the index of the field being projected. Note that this closure will always
2329     /// be `self` in the current MIR, because that is the only time we directly access the fields
2330     /// of a closure type.
2331     pub fn is_upvar_field_projection(&self, place_ref: PlaceRef<'tcx>) -> Option<Field> {
2332         path_utils::is_upvar_field_projection(self.infcx.tcx, &self.upvars, place_ref, self.body())
2333     }
2334 }
2335
2336 /// The degree of overlap between 2 places for borrow-checking.
2337 enum Overlap {
2338     /// The places might partially overlap - in this case, we give
2339     /// up and say that they might conflict. This occurs when
2340     /// different fields of a union are borrowed. For example,
2341     /// if `u` is a union, we have no way of telling how disjoint
2342     /// `u.a.x` and `a.b.y` are.
2343     Arbitrary,
2344     /// The places have the same type, and are either completely disjoint
2345     /// or equal - i.e., they can't "partially" overlap as can occur with
2346     /// unions. This is the "base case" on which we recur for extensions
2347     /// of the place.
2348     EqualOrDisjoint,
2349     /// The places are disjoint, so we know all extensions of them
2350     /// will also be disjoint.
2351     Disjoint,
2352 }