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