]> git.lizzy.rs Git - rust.git/blob - src/librustc_mir/borrow_check.rs
mir-borrowck: Add method to MIR borrowck context to retrieve the span of a given...
[rust.git] / src / librustc_mir / borrow_check.rs
1 // Copyright 2017 The Rust Project Developers. See the COPYRIGHT
2 // file at the top-level directory of this distribution and at
3 // http://rust-lang.org/COPYRIGHT.
4 //
5 // Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
6 // http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
7 // <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
8 // option. This file may not be copied, modified, or distributed
9 // except according to those terms.
10
11 //! This query borrow-checks the MIR to (further) ensure it is not broken.
12
13 use rustc::hir::def_id::{DefId};
14 use rustc::infer::{InferCtxt};
15 use rustc::ty::{self, TyCtxt, ParamEnv};
16 use rustc::ty::maps::Providers;
17 use rustc::mir::{AssertMessage, BasicBlock, BorrowKind, Location, Lvalue};
18 use rustc::mir::{Mir, Mutability, Operand, Projection, ProjectionElem, Rvalue};
19 use rustc::mir::{Statement, StatementKind, Terminator, TerminatorKind};
20 use rustc::mir::transform::{MirSource};
21
22 use rustc_data_structures::indexed_set::{self, IdxSetBuf};
23 use rustc_data_structures::indexed_vec::{Idx};
24
25 use syntax::ast::{self};
26 use syntax_pos::{DUMMY_SP, Span};
27
28 use dataflow::{do_dataflow};
29 use dataflow::{MoveDataParamEnv};
30 use dataflow::{BitDenotation, BlockSets, DataflowResults, DataflowResultsConsumer};
31 use dataflow::{MaybeInitializedLvals, MaybeUninitializedLvals};
32 use dataflow::{Borrows, BorrowData, BorrowIndex};
33 use dataflow::move_paths::{HasMoveData, MoveData, MovePathIndex, LookupResult};
34 use util::borrowck_errors::{BorrowckErrors, Origin};
35
36 use self::MutateMode::{JustWrite, WriteAndRead};
37 use self::ConsumeKind::{Consume};
38
39
40 pub fn provide(providers: &mut Providers) {
41     *providers = Providers {
42         mir_borrowck,
43         ..*providers
44     };
45 }
46
47 fn mir_borrowck<'a, 'tcx>(tcx: TyCtxt<'a, 'tcx, 'tcx>, def_id: DefId) {
48     let mir = tcx.mir_validated(def_id);
49     let src = MirSource::from_local_def_id(tcx, def_id);
50     debug!("run query mir_borrowck: {}", tcx.node_path_str(src.item_id()));
51
52     let mir: &Mir<'tcx> = &mir.borrow();
53     if !tcx.has_attr(def_id, "rustc_mir_borrowck") && !tcx.sess.opts.debugging_opts.borrowck_mir {
54         return;
55     }
56
57     let id = src.item_id();
58     let attributes = tcx.get_attrs(def_id);
59     let param_env = tcx.param_env(def_id);
60     tcx.infer_ctxt().enter(|_infcx| {
61
62         let move_data = MoveData::gather_moves(mir, tcx, param_env);
63         let mdpe = MoveDataParamEnv { move_data: move_data, param_env: param_env };
64         let dead_unwinds = IdxSetBuf::new_empty(mir.basic_blocks().len());
65         let flow_borrows = do_dataflow(tcx, mir, id, &attributes, &dead_unwinds,
66                                        Borrows::new(tcx, mir),
67                                        |bd, i| bd.location(i));
68         let flow_inits = do_dataflow(tcx, mir, id, &attributes, &dead_unwinds,
69                                      MaybeInitializedLvals::new(tcx, mir, &mdpe),
70                                      |bd, i| &bd.move_data().move_paths[i]);
71         let flow_uninits = do_dataflow(tcx, mir, id, &attributes, &dead_unwinds,
72                                        MaybeUninitializedLvals::new(tcx, mir, &mdpe),
73                                        |bd, i| &bd.move_data().move_paths[i]);
74
75         let mut mbcx = MirBorrowckCtxt {
76             tcx: tcx,
77             mir: mir,
78             node_id: id,
79             move_data: &mdpe.move_data,
80             param_env: param_env,
81             fake_infer_ctxt: &_infcx,
82         };
83
84         let mut state = InProgress::new(flow_borrows,
85                                         flow_inits,
86                                         flow_uninits);
87
88         mbcx.analyze_results(&mut state); // entry point for DataflowResultsConsumer
89     });
90
91     debug!("mir_borrowck done");
92 }
93
94 #[allow(dead_code)]
95 pub struct MirBorrowckCtxt<'c, 'b, 'a: 'b+'c, 'gcx: 'a+'tcx, 'tcx: 'a> {
96     tcx: TyCtxt<'a, 'gcx, 'gcx>,
97     mir: &'b Mir<'gcx>,
98     node_id: ast::NodeId,
99     move_data: &'b MoveData<'gcx>,
100     param_env: ParamEnv<'tcx>,
101     fake_infer_ctxt: &'c InferCtxt<'c, 'gcx, 'tcx>,
102 }
103
104 // (forced to be `pub` due to its use as an associated type below.)
105 pub struct InProgress<'b, 'tcx: 'b> {
106     borrows: FlowInProgress<Borrows<'b, 'tcx>>,
107     inits: FlowInProgress<MaybeInitializedLvals<'b, 'tcx>>,
108     uninits: FlowInProgress<MaybeUninitializedLvals<'b, 'tcx>>,
109 }
110
111 struct FlowInProgress<BD> where BD: BitDenotation {
112     base_results: DataflowResults<BD>,
113     curr_state: IdxSetBuf<BD::Idx>,
114     stmt_gen: IdxSetBuf<BD::Idx>,
115     stmt_kill: IdxSetBuf<BD::Idx>,
116 }
117
118 // Check that:
119 // 1. assignments are always made to mutable locations (FIXME: does that still really go here?)
120 // 2. loans made in overlapping scopes do not conflict
121 // 3. assignments do not affect things loaned out as immutable
122 // 4. moves do not affect things loaned out in any way
123 impl<'c, 'b, 'a: 'b+'c, 'gcx, 'tcx: 'a> DataflowResultsConsumer<'b, 'gcx>
124     for MirBorrowckCtxt<'c, 'b, 'a, 'gcx, 'tcx>
125 {
126     type FlowState = InProgress<'b, 'gcx>;
127
128     fn mir(&self) -> &'b Mir<'gcx> { self.mir }
129
130     fn reset_to_entry_of(&mut self, bb: BasicBlock, flow_state: &mut Self::FlowState) {
131         flow_state.each_flow(|b| b.reset_to_entry_of(bb),
132                              |i| i.reset_to_entry_of(bb),
133                              |u| u.reset_to_entry_of(bb));
134     }
135
136     fn reconstruct_statement_effect(&mut self,
137                                     location: Location,
138                                     flow_state: &mut Self::FlowState) {
139         flow_state.each_flow(|b| b.reconstruct_statement_effect(location),
140                              |i| i.reconstruct_statement_effect(location),
141                              |u| u.reconstruct_statement_effect(location));
142     }
143
144     fn apply_local_effect(&mut self,
145                           _location: Location,
146                           flow_state: &mut Self::FlowState) {
147         flow_state.each_flow(|b| b.apply_local_effect(),
148                              |i| i.apply_local_effect(),
149                              |u| u.apply_local_effect());
150     }
151
152     fn reconstruct_terminator_effect(&mut self,
153                                      location: Location,
154                                      flow_state: &mut Self::FlowState) {
155         flow_state.each_flow(|b| b.reconstruct_terminator_effect(location),
156                              |i| i.reconstruct_terminator_effect(location),
157                              |u| u.reconstruct_terminator_effect(location));
158     }
159
160     fn visit_block_entry(&mut self,
161                          bb: BasicBlock,
162                          flow_state: &Self::FlowState) {
163         let summary = flow_state.summary();
164         debug!("MirBorrowckCtxt::process_block({:?}): {}", bb, summary);
165     }
166
167     fn visit_statement_entry(&mut self,
168                              location: Location,
169                              stmt: &Statement<'gcx>,
170                              flow_state: &Self::FlowState) {
171         let summary = flow_state.summary();
172         debug!("MirBorrowckCtxt::process_statement({:?}, {:?}): {}", location, stmt, summary);
173         let span = stmt.source_info.span;
174         match stmt.kind {
175             StatementKind::Assign(ref lhs, ref rhs) => {
176                 self.mutate_lvalue(ContextKind::AssignLhs.new(location),
177                                    (lhs, span), JustWrite, flow_state);
178                 self.consume_rvalue(ContextKind::AssignRhs.new(location),
179                                     (rhs, span), location, flow_state);
180             }
181             StatementKind::SetDiscriminant { ref lvalue, variant_index: _ } => {
182                 self.mutate_lvalue(ContextKind::SetDiscrim.new(location),
183                                    (lvalue, span), JustWrite, flow_state);
184             }
185             StatementKind::InlineAsm { ref asm, ref outputs, ref inputs } => {
186                 for (o, output) in asm.outputs.iter().zip(outputs) {
187                     if o.is_indirect {
188                         self.consume_lvalue(ContextKind::InlineAsm.new(location),
189                                             Consume,
190                                             (output, span),
191                                             flow_state);
192                     } else {
193                         self.mutate_lvalue(ContextKind::InlineAsm.new(location),
194                                            (output, span),
195                                            if o.is_rw { WriteAndRead } else { JustWrite },
196                                            flow_state);
197                     }
198                 }
199                 for input in inputs {
200                     self.consume_operand(ContextKind::InlineAsm.new(location),
201                                          Consume,
202                                          (input, span), flow_state);
203                 }
204             }
205             StatementKind::EndRegion(ref _rgn) => {
206                 // ignored when consuming results (update to
207                 // flow_state already handled).
208             }
209             StatementKind::Nop |
210             StatementKind::Validate(..) |
211             StatementKind::StorageLive(..) => {
212                 // ignored by borrowck
213             }
214
215             StatementKind::StorageDead(local) => {
216                 // causes non-drop values to be dropped.
217                 self.consume_lvalue(ContextKind::StorageDead.new(location),
218                                     ConsumeKind::Consume,
219                                     (&Lvalue::Local(local), span),
220                                     flow_state)
221             }
222         }
223     }
224
225     fn visit_terminator_entry(&mut self,
226                               location: Location,
227                               term: &Terminator<'gcx>,
228                               flow_state: &Self::FlowState) {
229         let loc = location;
230         let summary = flow_state.summary();
231         debug!("MirBorrowckCtxt::process_terminator({:?}, {:?}): {}", location, term, summary);
232         let span = term.source_info.span;
233         match term.kind {
234             TerminatorKind::SwitchInt { ref discr, switch_ty: _, values: _, targets: _ } => {
235                 self.consume_operand(ContextKind::SwitchInt.new(loc),
236                                      Consume,
237                                      (discr, span), flow_state);
238             }
239             TerminatorKind::Drop { location: ref drop_lvalue, target: _, unwind: _ } => {
240                 self.consume_lvalue(ContextKind::Drop.new(loc),
241                                     ConsumeKind::Drop,
242                                     (drop_lvalue, span), flow_state);
243             }
244             TerminatorKind::DropAndReplace { location: ref drop_lvalue,
245                                              value: ref new_value,
246                                              target: _,
247                                              unwind: _ } => {
248                 self.mutate_lvalue(ContextKind::DropAndReplace.new(loc),
249                                    (drop_lvalue, span), JustWrite, flow_state);
250                 self.consume_operand(ContextKind::DropAndReplace.new(loc),
251                                      ConsumeKind::Drop,
252                                      (new_value, span), flow_state);
253             }
254             TerminatorKind::Call { ref func, ref args, ref destination, cleanup: _ } => {
255                 self.consume_operand(ContextKind::CallOperator.new(loc),
256                                      Consume,
257                                      (func, span), flow_state);
258                 for arg in args {
259                     self.consume_operand(ContextKind::CallOperand.new(loc),
260                                          Consume,
261                                          (arg, span), flow_state);
262                 }
263                 if let Some((ref dest, _/*bb*/)) = *destination {
264                     self.mutate_lvalue(ContextKind::CallDest.new(loc),
265                                        (dest, span), JustWrite, flow_state);
266                 }
267             }
268             TerminatorKind::Assert { ref cond, expected: _, ref msg, target: _, cleanup: _ } => {
269                 self.consume_operand(ContextKind::Assert.new(loc),
270                                      Consume,
271                                      (cond, span), flow_state);
272                 match *msg {
273                     AssertMessage::BoundsCheck { ref len, ref index } => {
274                         self.consume_operand(ContextKind::Assert.new(loc),
275                                              Consume,
276                                              (len, span), flow_state);
277                         self.consume_operand(ContextKind::Assert.new(loc),
278                                              Consume,
279                                              (index, span), flow_state);
280                     }
281                     AssertMessage::Math(_/*const_math_err*/) => {}
282                     AssertMessage::GeneratorResumedAfterReturn => {}
283                     AssertMessage::GeneratorResumedAfterPanic => {}
284                 }
285             }
286
287             TerminatorKind::Yield { ref value, resume: _, drop: _} => {
288                 self.consume_operand(ContextKind::Yield.new(loc),
289                                      Consume, (value, span), flow_state);
290             }
291
292             TerminatorKind::Goto { target: _ } |
293             TerminatorKind::Resume |
294             TerminatorKind::Return |
295             TerminatorKind::GeneratorDrop |
296             TerminatorKind::Unreachable => {
297                 // no data used, thus irrelevant to borrowck
298             }
299         }
300     }
301 }
302
303 #[derive(Copy, Clone, PartialEq, Eq, Debug)]
304 enum MutateMode { JustWrite, WriteAndRead }
305
306 #[derive(Copy, Clone, PartialEq, Eq, Debug)]
307 enum ConsumeKind { Drop, Consume }
308
309 #[derive(Copy, Clone, PartialEq, Eq, Debug)]
310 enum Control { Continue, Break }
311
312 impl<'c, 'b, 'a: 'b+'c, 'gcx, 'tcx: 'a> MirBorrowckCtxt<'c, 'b, 'a, 'gcx, 'tcx> {
313     fn mutate_lvalue(&mut self,
314                      context: Context,
315                      lvalue_span: (&Lvalue<'gcx>, Span),
316                      mode: MutateMode,
317                      flow_state: &InProgress<'b, 'gcx>) {
318         // Write of P[i] or *P, or WriteAndRead of any P, requires P init'd.
319         match mode {
320             MutateMode::WriteAndRead => {
321                 self.check_if_path_is_moved(context, lvalue_span, flow_state);
322             }
323             MutateMode::JustWrite => {
324                 self.check_if_assigned_path_is_moved(context, lvalue_span, flow_state);
325             }
326         }
327
328         // check we don't invalidate any outstanding loans
329         self.each_borrow_involving_path(context,
330                                         lvalue_span.0, flow_state, |this, _index, _data| {
331                                             this.report_illegal_mutation_of_borrowed(context,
332                                                                                      lvalue_span);
333                                             Control::Break
334                                         });
335
336         // check for reassignments to immutable local variables
337         self.check_if_reassignment_to_immutable_state(context, lvalue_span, flow_state);
338     }
339
340     fn consume_rvalue(&mut self,
341                       context: Context,
342                       (rvalue, span): (&Rvalue<'gcx>, Span),
343                       location: Location,
344                       flow_state: &InProgress<'b, 'gcx>) {
345         match *rvalue {
346             Rvalue::Ref(_/*rgn*/, bk, ref lvalue) => {
347                 self.borrow(context, location, bk, (lvalue, span), flow_state)
348             }
349
350             Rvalue::Use(ref operand) |
351             Rvalue::Repeat(ref operand, _) |
352             Rvalue::UnaryOp(_/*un_op*/, ref operand) |
353             Rvalue::Cast(_/*cast_kind*/, ref operand, _/*ty*/) => {
354                 self.consume_operand(context, Consume, (operand, span), flow_state)
355             }
356
357             Rvalue::Len(ref lvalue) |
358             Rvalue::Discriminant(ref lvalue) => {
359                 // len(_)/discriminant(_) merely read, not consume.
360                 self.check_if_path_is_moved(context, (lvalue, span), flow_state);
361             }
362
363             Rvalue::BinaryOp(_bin_op, ref operand1, ref operand2) |
364             Rvalue::CheckedBinaryOp(_bin_op, ref operand1, ref operand2) => {
365                 self.consume_operand(context, Consume, (operand1, span), flow_state);
366                 self.consume_operand(context, Consume, (operand2, span), flow_state);
367             }
368
369             Rvalue::NullaryOp(_op, _ty) => {
370                 // nullary ops take no dynamic input; no borrowck effect.
371                 //
372                 // FIXME: is above actually true? Do we want to track
373                 // the fact that uninitialized data can be created via
374                 // `NullOp::Box`?
375             }
376
377             Rvalue::Aggregate(ref _aggregate_kind, ref operands) => {
378                 for operand in operands {
379                     self.consume_operand(context, Consume, (operand, span), flow_state);
380                 }
381             }
382         }
383     }
384
385     fn consume_operand(&mut self,
386                        context: Context,
387                        consume_via_drop: ConsumeKind,
388                        (operand, span): (&Operand<'gcx>, Span),
389                        flow_state: &InProgress<'b, 'gcx>) {
390         match *operand {
391             Operand::Consume(ref lvalue) =>
392                 self.consume_lvalue(context, consume_via_drop, (lvalue, span), flow_state),
393             Operand::Constant(_) => {}
394         }
395     }
396
397     fn consume_lvalue(&mut self,
398                       context: Context,
399                       consume_via_drop: ConsumeKind,
400                       lvalue_span: (&Lvalue<'gcx>, Span),
401                       flow_state: &InProgress<'b, 'gcx>) {
402         let lvalue = lvalue_span.0;
403         let ty = lvalue.ty(self.mir, self.tcx).to_ty(self.tcx);
404         let moves_by_default =
405             self.fake_infer_ctxt.type_moves_by_default(self.param_env, ty, DUMMY_SP);
406         if moves_by_default {
407             // move of lvalue: check if this is move of already borrowed path
408             self.each_borrow_involving_path(
409                 context, lvalue_span.0, flow_state, |this, _idx, borrow| {
410                     if !borrow.compatible_with(BorrowKind::Mut) {
411                         this.report_move_out_while_borrowed(context, lvalue_span);
412                         Control::Break
413                     } else {
414                         Control::Continue
415                     }
416                 });
417         } else {
418             // copy of lvalue: check if this is "copy of frozen path" (FIXME: see check_loans.rs)
419             self.each_borrow_involving_path(
420                 context, lvalue_span.0, flow_state, |this, _idx, borrow| {
421                     if !borrow.compatible_with(BorrowKind::Shared) {
422                         this.report_use_while_mutably_borrowed(context, lvalue_span);
423                         Control::Break
424                     } else {
425                         Control::Continue
426                     }
427                 });
428         }
429
430         // Finally, check if path was already moved.
431         match consume_via_drop {
432             ConsumeKind::Drop => {
433                 // If path is merely being dropped, then we'll already
434                 // check the drop flag to see if it is moved (thus we
435                 // skip this check in that case).
436             }
437             ConsumeKind::Consume => {
438                 self.check_if_path_is_moved(context, lvalue_span, flow_state);
439             }
440         }
441     }
442
443     fn borrow(&mut self,
444               context: Context,
445               location: Location,
446               bk: BorrowKind,
447               lvalue_span: (&Lvalue<'gcx>, Span),
448               flow_state: &InProgress<'b, 'gcx>) {
449         debug!("borrow location: {:?} lvalue: {:?} span: {:?}",
450                location, lvalue_span.0, lvalue_span.1);
451         self.check_if_path_is_moved(context, lvalue_span, flow_state);
452         self.check_for_conflicting_loans(context, location, bk, lvalue_span, flow_state);
453     }
454 }
455
456 impl<'c, 'b, 'a: 'b+'c, 'gcx, 'tcx: 'a> MirBorrowckCtxt<'c, 'b, 'a, 'gcx, 'tcx> {
457     fn check_if_reassignment_to_immutable_state(&mut self,
458                                                 context: Context,
459                                                 (lvalue, span): (&Lvalue<'gcx>, Span),
460                                                 flow_state: &InProgress<'b, 'gcx>) {
461         let move_data = flow_state.inits.base_results.operator().move_data();
462
463         // determine if this path has a non-mut owner (and thus needs checking).
464         let mut l = lvalue;
465         loop {
466             match *l {
467                 Lvalue::Projection(ref proj) => {
468                     l = &proj.base;
469                     continue;
470                 }
471                 Lvalue::Local(local) => {
472                     match self.mir.local_decls[local].mutability {
473                         Mutability::Not => break, // needs check
474                         Mutability::Mut => return,
475                     }
476                 }
477                 Lvalue::Static(_) => {
478                     // mutation of non-mut static is always illegal,
479                     // independent of dataflow.
480                     self.report_assignment_to_static(context, (lvalue, span));
481                     return;
482                 }
483             }
484         }
485
486         if let Some(mpi) = self.move_path_for_lvalue(context, move_data, lvalue) {
487             if flow_state.inits.curr_state.contains(&mpi) {
488                 // may already be assigned before reaching this statement;
489                 // report error.
490                 self.report_illegal_reassignment(context, (lvalue, span));
491             }
492         }
493     }
494
495     fn check_if_path_is_moved(&mut self,
496                               context: Context,
497                               lvalue_span: (&Lvalue<'gcx>, Span),
498                               flow_state: &InProgress<'b, 'gcx>) {
499         // FIXME: analogous code in check_loans first maps `lvalue` to
500         // its base_path ... but is that what we want here?
501         let lvalue = self.base_path(lvalue_span.0);
502
503         let maybe_uninits = &flow_state.uninits;
504         let move_data = maybe_uninits.base_results.operator().move_data();
505         if let Some(mpi) = self.move_path_for_lvalue(context, move_data, lvalue) {
506             if maybe_uninits.curr_state.contains(&mpi) {
507                 // find and report move(s) that could cause this to be uninitialized
508                 self.report_use_of_moved(context, lvalue_span);
509             } else {
510                 // sanity check: initialized on *some* path, right?
511                 assert!(flow_state.inits.curr_state.contains(&mpi));
512             }
513         }
514     }
515
516     fn move_path_for_lvalue(&mut self,
517                             _context: Context,
518                             move_data: &MoveData<'gcx>,
519                             lvalue: &Lvalue<'gcx>)
520                             -> Option<MovePathIndex>
521     {
522         // If returns None, then there is no move path corresponding
523         // to a direct owner of `lvalue` (which means there is nothing
524         // that borrowck tracks for its analysis).
525
526         match move_data.rev_lookup.find(lvalue) {
527             LookupResult::Parent(_) => None,
528             LookupResult::Exact(mpi) => Some(mpi),
529         }
530     }
531
532     fn check_if_assigned_path_is_moved(&mut self,
533                                        context: Context,
534                                        (lvalue, span): (&Lvalue<'gcx>, Span),
535                                        flow_state: &InProgress<'b, 'gcx>) {
536         // recur down lvalue; dispatch to check_if_path_is_moved when necessary
537         let mut lvalue = lvalue;
538         loop {
539             match *lvalue {
540                 Lvalue::Local(_) | Lvalue::Static(_) => {
541                     // assigning to `x` does not require `x` be initialized.
542                     break;
543                 }
544                 Lvalue::Projection(ref proj) => {
545                     let Projection { ref base, ref elem } = **proj;
546                     match *elem {
547                         ProjectionElem::Deref |
548                         // assigning to *P requires `P` initialized.
549                         ProjectionElem::Index(_/*operand*/) |
550                         ProjectionElem::ConstantIndex { .. } |
551                         // assigning to P[i] requires `P` initialized.
552                         ProjectionElem::Downcast(_/*adt_def*/, _/*variant_idx*/) =>
553                         // assigning to (P->variant) is okay if assigning to `P` is okay
554                         //
555                         // FIXME: is this true even if P is a adt with a dtor?
556                         { }
557
558                         ProjectionElem::Subslice { .. } => {
559                             panic!("we dont allow assignments to subslices, context: {:?}",
560                                    context);
561                         }
562
563                         ProjectionElem::Field(..) => {
564                             // if type of `P` has a dtor, then
565                             // assigning to `P.f` requires `P` itself
566                             // be already initialized
567                             let tcx = self.tcx;
568                             match base.ty(self.mir, tcx).to_ty(tcx).sty {
569                                 ty::TyAdt(def, _) if def.has_dtor(tcx) => {
570
571                                     // FIXME: analogous code in
572                                     // check_loans.rs first maps
573                                     // `base` to its base_path.
574
575                                     self.check_if_path_is_moved(context,
576                                                                 (base, span), flow_state);
577
578                                     // (base initialized; no need to
579                                     // recur further)
580                                     break;
581                                 }
582                                 _ => {}
583                             }
584                         }
585                     }
586
587                     lvalue = base;
588                     continue;
589                 }
590             }
591         }
592     }
593
594     fn check_for_conflicting_loans(&mut self,
595                                    context: Context,
596                                    _location: Location,
597                                    _bk: BorrowKind,
598                                    lvalue_span: (&Lvalue<'gcx>, Span),
599                                    flow_state: &InProgress<'b, 'gcx>) {
600         // NOTE FIXME: The analogous code in old borrowck
601         // check_loans.rs is careful to iterate over every *issued*
602         // loan, as opposed to just the in scope ones.
603         //
604         // (Or if you prefer, all the *other* iterations over loans
605         // only consider loans that are in scope of some given
606         // region::Scope)
607         //
608         // The (currently skeletal) code here does not encode such a
609         // distinction, which means it is almost certainly over
610         // looking something.
611         //
612         // (It is probably going to reject code that should be
613         // accepted, I suspect, by treated issued-but-out-of-scope
614         // loans as issued-and-in-scope, and thus causing them to
615         // interfere with other loans.)
616         //
617         // However, I just want to get something running, especially
618         // since I am trying to move into new territory with NLL, so
619         // lets get this going first, and then address the issued vs
620         // in-scope distinction later.
621
622         let state = &flow_state.borrows;
623         let data = &state.base_results.operator().borrows();
624
625         debug!("check_for_conflicting_loans location: {:?}", _location);
626
627         // does any loan generated here conflict with a previously issued loan?
628         let mut loans_generated = 0;
629         for (g, gen) in state.elems_generated().map(|g| (g, &data[g])) {
630             loans_generated += 1;
631             for (i, issued) in state.elems_incoming().map(|i| (i, &data[i])) {
632                 debug!("check_for_conflicting_loans gen: {:?} issued: {:?} conflicts: {}",
633                        (g, gen, self.base_path(&gen.lvalue),
634                         self.restrictions(&gen.lvalue).collect::<Vec<_>>()),
635                        (i, issued, self.base_path(&issued.lvalue),
636                         self.restrictions(&issued.lvalue).collect::<Vec<_>>()),
637                        self.conflicts_with(gen, issued));
638                 if self.conflicts_with(gen, issued) {
639                     self.report_conflicting_borrow(context, lvalue_span, gen, issued);
640                 }
641             }
642         }
643
644         // MIR statically ensures each statement gens *at most one*
645         // loan; mutual conflict (within a statement) can't arise.
646         //
647         // As safe-guard, assert that above property actually holds.
648         assert!(loans_generated <= 1);
649     } }
650
651 impl<'c, 'b, 'a: 'b+'c, 'gcx, 'tcx: 'a> MirBorrowckCtxt<'c, 'b, 'a, 'gcx, 'tcx> {
652     fn each_borrow_involving_path<F>(&mut self,
653                                      _context: Context,
654                                      lvalue: &Lvalue<'gcx>,
655                                      flow_state: &InProgress<'b, 'gcx>,
656                                      mut op: F)
657         where F: FnMut(&mut Self, BorrowIndex, &BorrowData<'gcx>) -> Control
658     {
659         // FIXME: analogous code in check_loans first maps `lvalue` to
660         // its base_path.
661
662         let domain = flow_state.borrows.base_results.operator();
663         let data = domain.borrows();
664
665         // check for loan restricting path P being used. Accounts for
666         // borrows of P, P.a.b, etc.
667         for i in flow_state.borrows.elems_incoming() {
668             // FIXME: check_loans.rs filtered this to "in scope"
669             // loans; i.e. it took a scope S and checked that each
670             // restriction's kill_scope was a superscope of S.
671             let borrowed = &data[i];
672             for restricted in self.restrictions(&borrowed.lvalue) {
673                 if restricted == lvalue {
674                     let ctrl = op(self, i, borrowed);
675                     if ctrl == Control::Break { return; }
676                 }
677             }
678         }
679
680         // check for loans (not restrictions) on any base path.
681         // e.g. Rejects `{ let x = &mut a.b; let y = a.b.c; }`,
682         // since that moves out of borrowed path `a.b`.
683         //
684         // Limiting to loans (not restrictions) keeps this one
685         // working: `{ let x = &mut a.b; let y = a.c; }`
686         let mut cursor = lvalue;
687         loop {
688             // FIXME: check_loans.rs invoked `op` *before* cursor
689             // shift here.  Might just work (and even avoid redundant
690             // errors?) given code above?  But for now, I want to try
691             // doing what I think is more "natural" check.
692             for i in flow_state.borrows.elems_incoming() {
693                 let borrowed = &data[i];
694                 if borrowed.lvalue == *cursor {
695                     let ctrl = op(self, i, borrowed);
696                     if ctrl == Control::Break { return; }
697                 }
698             }
699
700             match *cursor {
701                 Lvalue::Local(_) | Lvalue::Static(_) => break,
702                 Lvalue::Projection(ref proj) => cursor = &proj.base,
703             }
704         }
705     }
706 }
707
708 mod restrictions {
709     use super::MirBorrowckCtxt;
710
711     use rustc::hir;
712     use rustc::ty::{self, TyCtxt};
713     use rustc::mir::{Lvalue, Mir, ProjectionElem};
714
715     pub(super) struct Restrictions<'c, 'tcx: 'c> {
716         mir: &'c Mir<'tcx>,
717         tcx: TyCtxt<'c, 'tcx, 'tcx>,
718         lvalue_stack: Vec<&'c Lvalue<'tcx>>,
719     }
720
721     impl<'c, 'b, 'a: 'b+'c, 'gcx, 'tcx: 'a> MirBorrowckCtxt<'c, 'b, 'a, 'gcx, 'tcx> {
722         pub(super) fn restrictions<'d>(&self,
723                                        lvalue: &'d Lvalue<'gcx>)
724                                        -> Restrictions<'d, 'gcx> where 'b: 'd
725         {
726             let lvalue_stack = if self.has_restrictions(lvalue) { vec![lvalue] } else { vec![] };
727             Restrictions { lvalue_stack: lvalue_stack, mir: self.mir, tcx: self.tcx }
728         }
729
730         fn has_restrictions(&self, lvalue: &Lvalue<'gcx>) -> bool {
731             let mut cursor = lvalue;
732             loop {
733                 let proj = match *cursor {
734                     Lvalue::Local(_) => return true,
735                     Lvalue::Static(_) => return false,
736                     Lvalue::Projection(ref proj) => proj,
737                 };
738                 match proj.elem {
739                     ProjectionElem::Index(..) |
740                     ProjectionElem::ConstantIndex { .. } |
741                     ProjectionElem::Downcast(..) |
742                     ProjectionElem::Subslice { .. } |
743                     ProjectionElem::Field(_/*field*/, _/*ty*/) => {
744                         cursor = &proj.base;
745                         continue;
746                     }
747                     ProjectionElem::Deref => {
748                         let ty = proj.base.ty(self.mir, self.tcx).to_ty(self.tcx);
749                         match ty.sty {
750                             ty::TyRawPtr(_) => {
751                                 return false;
752                             }
753                             ty::TyRef(_, ty::TypeAndMut { ty: _, mutbl: hir::MutImmutable }) => {
754                                 // FIXME: do I need to check validity of
755                                 // region here though? (I think the original
756                                 // check_loans code did, like readme says)
757                                 return false;
758                             }
759                             ty::TyRef(_, ty::TypeAndMut { ty: _, mutbl: hir::MutMutable }) => {
760                                 cursor = &proj.base;
761                                 continue;
762                             }
763                             ty::TyAdt(..) if ty.is_box() => {
764                                 cursor = &proj.base;
765                                 continue;
766                             }
767                             _ => {
768                                 panic!("unknown type fed to Projection Deref.");
769                             }
770                         }
771                     }
772                 }
773             }
774         }
775     }
776
777     impl<'c, 'tcx> Iterator for Restrictions<'c, 'tcx> {
778         type Item = &'c Lvalue<'tcx>;
779         fn next(&mut self) -> Option<Self::Item> {
780             'pop: loop {
781                 let lvalue = match self.lvalue_stack.pop() {
782                     None => return None,
783                     Some(lvalue) => lvalue,
784                 };
785
786                 // `lvalue` may not be a restriction itself, but may
787                 // hold one further down (e.g. we never return
788                 // downcasts here, but may return a base of a
789                 // downcast).
790                 //
791                 // Also, we need to enqueue any additional
792                 // subrestrictions that it implies, since we can only
793                 // return from from this call alone.
794
795                 let mut cursor = lvalue;
796                 'cursor: loop {
797                     let proj = match *cursor {
798                         Lvalue::Local(_) => return Some(cursor), // search yielded this leaf
799                         Lvalue::Static(_) => continue 'pop, // fruitless leaf; try next on stack
800                         Lvalue::Projection(ref proj) => proj,
801                     };
802
803                     match proj.elem {
804                         ProjectionElem::Field(_/*field*/, _/*ty*/) => {
805                             // FIXME: add union handling
806                             self.lvalue_stack.push(&proj.base);
807                             return Some(cursor);
808                         }
809                         ProjectionElem::Downcast(..) |
810                         ProjectionElem::Subslice { .. } |
811                         ProjectionElem::ConstantIndex { .. } |
812                         ProjectionElem::Index(_) => {
813                             cursor = &proj.base;
814                             continue 'cursor;
815                         }
816                         ProjectionElem::Deref => {
817                             // (handled below)
818                         }
819                     }
820
821                     assert_eq!(proj.elem, ProjectionElem::Deref);
822
823                     let ty = proj.base.ty(self.mir, self.tcx).to_ty(self.tcx);
824                     match ty.sty {
825                         ty::TyRawPtr(_) => {
826                             // borrowck ignores raw ptrs; treat analogous to imm borrow
827                             continue 'pop;
828                         }
829                         // R-Deref-Imm-Borrowed
830                         ty::TyRef(_/*rgn*/, ty::TypeAndMut { ty: _, mutbl: hir::MutImmutable }) => {
831                             // immutably-borrowed referents do not
832                             // have recursively-implied restrictions
833                             // (because preventing actions on `*LV`
834                             // does nothing about aliases like `*LV1`)
835
836                             // FIXME: do I need to check validity of
837                             // `_r` here though? (I think the original
838                             // check_loans code did, like the readme
839                             // says)
840
841                             // (And do I *really* not have to
842                             // recursively process the `base` as a
843                             // further search here? Leaving this `if
844                             // false` here as a hint to look at this
845                             // again later.
846                             //
847                             // Ah, it might be because the
848                             // restrictions are distinct from the path
849                             // substructure. Note that there is a
850                             // separate loop over the path
851                             // substructure in fn
852                             // each_borrow_involving_path, for better
853                             // or for worse.
854
855                             if false {
856                                 cursor = &proj.base;
857                                 continue 'cursor;
858                             } else {
859                                 continue 'pop;
860                             }
861                         }
862
863                         // R-Deref-Mut-Borrowed
864                         ty::TyRef(_/*rgn*/, ty::TypeAndMut { ty: _, mutbl: hir::MutMutable }) => {
865                             // mutably-borrowed referents are
866                             // themselves restricted.
867
868                             // FIXME: do I need to check validity of
869                             // `_r` here though? (I think the original
870                             // check_loans code did, like the readme
871                             // says)
872
873                             // schedule base for future iteration.
874                             self.lvalue_stack.push(&proj.base);
875                             return Some(cursor); // search yielded interior node
876                         }
877
878                         // R-Deref-Send-Pointer
879                         ty::TyAdt(..) if ty.is_box() => {
880                             // borrowing interior of a box implies that
881                             // its base can no longer be mutated (o/w box
882                             // storage would be freed)
883                             self.lvalue_stack.push(&proj.base);
884                             return Some(cursor); // search yielded interior node
885                         }
886
887                         _ => panic!("unknown type fed to Projection Deref."),
888                     }
889                 }
890             }
891         }
892     }
893 }
894
895 impl<'c, 'b, 'a: 'b+'c, 'gcx, 'tcx: 'a> MirBorrowckCtxt<'c, 'b, 'a, 'gcx, 'tcx> {
896     fn report_use_of_moved(&mut self,
897                            _context: Context,
898                            (lvalue, span): (&Lvalue, Span)) {
899         let mut err = self.tcx.cannot_act_on_uninitialized_variable(
900             span, "use", &self.describe_lvalue(lvalue), Origin::Mir);
901         // FIXME: add span_label for use of uninitialized variable
902         err.emit();
903     }
904
905     fn report_move_out_while_borrowed(&mut self,
906                                       _context: Context,
907                                       (lvalue, span): (&Lvalue, Span)) {
908         let mut err = self.tcx.cannot_move_when_borrowed(
909             span, &self.describe_lvalue(lvalue), Origin::Mir);
910         // FIXME 1: add span_label for "borrow of `()` occurs here"
911         // FIXME 2: add span_label for "move out of `{}` occurs here"
912         err.emit();
913     }
914
915     fn report_use_while_mutably_borrowed(&mut self,
916                                          _context: Context,
917                                          (lvalue, span): (&Lvalue, Span)) {
918         let mut err = self.tcx.cannot_use_when_mutably_borrowed(
919             span, &self.describe_lvalue(lvalue), Origin::Mir);
920         // FIXME 1: add span_label for "borrow of `()` occurs here"
921         // FIXME 2: add span_label for "use of `{}` occurs here"
922         err.emit();
923     }
924
925     fn report_conflicting_borrow(&mut self,
926                                  _context: Context,
927                                  (lvalue, span): (&Lvalue, Span),
928                                  loan1: &BorrowData,
929                                  loan2: &BorrowData) {
930         // FIXME: obviously falsifiable. Generalize for non-eq lvalues later.
931         assert_eq!(loan1.lvalue, loan2.lvalue);
932
933         // FIXME: supply non-"" `opt_via` when appropriate
934         let mut err = match (loan1.kind, "immutable", "mutable",
935                              loan2.kind, "immutable", "mutable") {
936             (BorrowKind::Shared, lft, _, BorrowKind::Mut, _, rgt) |
937             (BorrowKind::Mut, _, lft, BorrowKind::Shared, rgt, _) |
938             (BorrowKind::Mut, _, lft, BorrowKind::Mut, _, rgt) =>
939                 self.tcx.cannot_reborrow_already_borrowed(
940                     span, &self.describe_lvalue(lvalue),
941                     "", lft, "it", rgt, "", Origin::Mir),
942
943             _ =>  self.tcx.cannot_mutably_borrow_multiply(
944                 span, &self.describe_lvalue(lvalue), "", Origin::Mir),
945             // FIXME: add span labels for first and second mutable borrows, as well as
946             // end point for first.
947         };
948         err.emit();
949     }
950
951     fn report_illegal_mutation_of_borrowed(&mut self, _: Context, (lvalue, span): (&Lvalue, Span)) {
952         let mut err = self.tcx.cannot_assign_to_borrowed(
953             span, &self.describe_lvalue(lvalue), Origin::Mir);
954         // FIXME: add span labels for borrow and assignment points
955         err.emit();
956     }
957
958     fn report_illegal_reassignment(&mut self, _context: Context, (lvalue, span): (&Lvalue, Span)) {
959         let mut err = self.tcx.cannot_reassign_immutable(
960             span, &self.describe_lvalue(lvalue), Origin::Mir);
961         // FIXME: add span labels for borrow and assignment points
962         err.emit();
963     }
964
965     fn report_assignment_to_static(&mut self, _context: Context, (lvalue, span): (&Lvalue, Span)) {
966         let mut err = self.tcx.cannot_assign_static(
967             span, &self.describe_lvalue(lvalue), Origin::Mir);
968         // FIXME: add span labels for borrow and assignment points
969         err.emit();
970     }
971 }
972
973 impl<'c, 'b, 'a: 'b+'c, 'gcx, 'tcx: 'a> MirBorrowckCtxt<'c, 'b, 'a, 'gcx, 'tcx> {
974     // End-user visible description of `lvalue`
975     fn describe_lvalue(&self, lvalue: &Lvalue) -> String {
976         let mut buf = String::new();
977         self.append_lvalue_to_string(lvalue, &mut buf);
978         buf
979     }
980
981     // Appends end-user visible description of `lvalue` to `buf`.
982     fn append_lvalue_to_string(&self, lvalue: &Lvalue, buf: &mut String) {
983         match *lvalue {
984             Lvalue::Local(local) => {
985                 let local = &self.mir.local_decls[local];
986                 match local.name {
987                     Some(name) => buf.push_str(&format!("{}", name)),
988                     None => buf.push_str("_"),
989                 }
990             }
991             Lvalue::Static(ref static_) => {
992                 buf.push_str(&format!("{}", &self.tcx.item_name(static_.def_id)));
993             }
994             Lvalue::Projection(ref proj) => {
995                 let (prefix, suffix, index_operand) = match proj.elem {
996                     ProjectionElem::Deref =>
997                         ("(*", format!(")"), None),
998                     ProjectionElem::Downcast(..) =>
999                         ("",   format!(""), None), // (dont emit downcast info)
1000                     ProjectionElem::Field(field, _ty) =>
1001                         ("",   format!(".{}", field.index()), None),
1002                     ProjectionElem::Index(index) =>
1003                         ("",   format!(""), Some(index)),
1004                     ProjectionElem::ConstantIndex { offset, min_length, from_end: true } =>
1005                         ("",   format!("[{} of {}]", offset, min_length), None),
1006                     ProjectionElem::ConstantIndex { offset, min_length, from_end: false } =>
1007                         ("",   format!("[-{} of {}]", offset, min_length), None),
1008                     ProjectionElem::Subslice { from, to: 0 } =>
1009                         ("",   format!("[{}:]", from), None),
1010                     ProjectionElem::Subslice { from: 0, to } =>
1011                         ("",   format!("[:-{}]", to), None),
1012                     ProjectionElem::Subslice { from, to } =>
1013                         ("",   format!("[{}:-{}]", from, to), None),
1014                 };
1015                 buf.push_str(prefix);
1016                 self.append_lvalue_to_string(&proj.base, buf);
1017                 if let Some(index) = index_operand {
1018                     buf.push_str("[");
1019                     self.append_lvalue_to_string(&Lvalue::Local(index), buf);
1020                     buf.push_str("]");
1021                 } else {
1022                     buf.push_str(&suffix);
1023                 }
1024             }
1025         }
1026     }
1027
1028     // Retrieve span of given borrow from the current MIR representation
1029     fn retrieve_borrow_span(&self, borrow: &BorrowData) -> Span {
1030         self.mir.basic_blocks()[borrow.location.block]
1031             .statements[borrow.location.statement_index]
1032             .source_info.span
1033     }
1034 }
1035
1036 impl<'c, 'b, 'a: 'b+'c, 'gcx, 'tcx: 'a> MirBorrowckCtxt<'c, 'b, 'a, 'gcx, 'tcx> {
1037     // FIXME: needs to be able to express errors analogous to check_loans.rs
1038     fn conflicts_with(&self, loan1: &BorrowData<'gcx>, loan2: &BorrowData<'gcx>) -> bool {
1039         if loan1.compatible_with(loan2.kind) { return false; }
1040
1041         let loan2_base_path = self.base_path(&loan2.lvalue);
1042         for restricted in self.restrictions(&loan1.lvalue) {
1043             if restricted != loan2_base_path { continue; }
1044             return true;
1045         }
1046
1047         let loan1_base_path = self.base_path(&loan1.lvalue);
1048         for restricted in self.restrictions(&loan2.lvalue) {
1049             if restricted != loan1_base_path { continue; }
1050             return true;
1051         }
1052
1053         return false;
1054     }
1055
1056     // FIXME (#16118): function intended to allow the borrow checker
1057     // to be less precise in its handling of Box while still allowing
1058     // moves out of a Box. They should be removed when/if we stop
1059     // treating Box specially (e.g. when/if DerefMove is added...)
1060
1061     fn base_path<'d>(&self, lvalue: &'d Lvalue<'gcx>) -> &'d Lvalue<'gcx> {
1062         //! Returns the base of the leftmost (deepest) dereference of an
1063         //! Box in `lvalue`. If there is no dereference of an Box
1064         //! in `lvalue`, then it just returns `lvalue` itself.
1065
1066         let mut cursor = lvalue;
1067         let mut deepest = lvalue;
1068         loop {
1069             let proj = match *cursor {
1070                 Lvalue::Local(..) | Lvalue::Static(..) => return deepest,
1071                 Lvalue::Projection(ref proj) => proj,
1072             };
1073             if proj.elem == ProjectionElem::Deref &&
1074                 lvalue.ty(self.mir, self.tcx).to_ty(self.tcx).is_box()
1075             {
1076                 deepest = &proj.base;
1077             }
1078             cursor = &proj.base;
1079         }
1080     }
1081 }
1082
1083 #[derive(Copy, Clone, PartialEq, Eq, Debug)]
1084 struct Context {
1085     kind: ContextKind,
1086     loc: Location,
1087 }
1088
1089 #[derive(Copy, Clone, PartialEq, Eq, Debug)]
1090 enum ContextKind {
1091     AssignLhs,
1092     AssignRhs,
1093     SetDiscrim,
1094     InlineAsm,
1095     SwitchInt,
1096     Drop,
1097     DropAndReplace,
1098     CallOperator,
1099     CallOperand,
1100     CallDest,
1101     Assert,
1102     StorageDead,
1103     Yield,
1104 }
1105
1106 impl ContextKind {
1107     fn new(self, loc: Location) -> Context { Context { kind: self, loc: loc } }
1108 }
1109
1110 impl<'b, 'tcx: 'b> InProgress<'b, 'tcx> {
1111     pub(super) fn new(borrows: DataflowResults<Borrows<'b, 'tcx>>,
1112                       inits: DataflowResults<MaybeInitializedLvals<'b, 'tcx>>,
1113                       uninits: DataflowResults<MaybeUninitializedLvals<'b, 'tcx>>)
1114                       -> Self {
1115         InProgress {
1116             borrows: FlowInProgress::new(borrows),
1117             inits: FlowInProgress::new(inits),
1118             uninits: FlowInProgress::new(uninits),
1119         }
1120     }
1121
1122     fn each_flow<XB, XI, XU>(&mut self,
1123                              mut xform_borrows: XB,
1124                              mut xform_inits: XI,
1125                              mut xform_uninits: XU) where
1126         XB: FnMut(&mut FlowInProgress<Borrows<'b, 'tcx>>),
1127         XI: FnMut(&mut FlowInProgress<MaybeInitializedLvals<'b, 'tcx>>),
1128         XU: FnMut(&mut FlowInProgress<MaybeUninitializedLvals<'b, 'tcx>>),
1129     {
1130         xform_borrows(&mut self.borrows);
1131         xform_inits(&mut self.inits);
1132         xform_uninits(&mut self.uninits);
1133     }
1134
1135     fn summary(&self) -> String {
1136         let mut s = String::new();
1137
1138         s.push_str("borrows in effect: [");
1139         let mut saw_one = false;
1140         self.borrows.each_state_bit(|borrow| {
1141             if saw_one { s.push_str(", "); };
1142             saw_one = true;
1143             let borrow_data = &self.borrows.base_results.operator().borrows()[borrow];
1144             s.push_str(&format!("{}", borrow_data));
1145         });
1146         s.push_str("] ");
1147
1148         s.push_str("borrows generated: [");
1149         let mut saw_one = false;
1150         self.borrows.each_gen_bit(|borrow| {
1151             if saw_one { s.push_str(", "); };
1152             saw_one = true;
1153             let borrow_data = &self.borrows.base_results.operator().borrows()[borrow];
1154             s.push_str(&format!("{}", borrow_data));
1155         });
1156         s.push_str("] ");
1157
1158         s.push_str("inits: [");
1159         let mut saw_one = false;
1160         self.inits.each_state_bit(|mpi_init| {
1161             if saw_one { s.push_str(", "); };
1162             saw_one = true;
1163             let move_path =
1164                 &self.inits.base_results.operator().move_data().move_paths[mpi_init];
1165             s.push_str(&format!("{}", move_path));
1166         });
1167         s.push_str("] ");
1168
1169         s.push_str("uninits: [");
1170         let mut saw_one = false;
1171         self.uninits.each_state_bit(|mpi_uninit| {
1172             if saw_one { s.push_str(", "); };
1173             saw_one = true;
1174             let move_path =
1175                 &self.uninits.base_results.operator().move_data().move_paths[mpi_uninit];
1176             s.push_str(&format!("{}", move_path));
1177         });
1178         s.push_str("]");
1179
1180         return s;
1181     }
1182 }
1183
1184 impl<BD> FlowInProgress<BD> where BD: BitDenotation {
1185     fn each_state_bit<F>(&self, f: F) where F: FnMut(BD::Idx) {
1186         self.curr_state.each_bit(self.base_results.operator().bits_per_block(), f)
1187     }
1188
1189     fn each_gen_bit<F>(&self, f: F) where F: FnMut(BD::Idx) {
1190         self.stmt_gen.each_bit(self.base_results.operator().bits_per_block(), f)
1191     }
1192
1193     fn new(results: DataflowResults<BD>) -> Self {
1194         let bits_per_block = results.sets().bits_per_block();
1195         let curr_state = IdxSetBuf::new_empty(bits_per_block);
1196         let stmt_gen = IdxSetBuf::new_empty(bits_per_block);
1197         let stmt_kill = IdxSetBuf::new_empty(bits_per_block);
1198         FlowInProgress {
1199             base_results: results,
1200             curr_state: curr_state,
1201             stmt_gen: stmt_gen,
1202             stmt_kill: stmt_kill,
1203         }
1204     }
1205
1206     fn reset_to_entry_of(&mut self, bb: BasicBlock) {
1207         (*self.curr_state).clone_from(self.base_results.sets().on_entry_set_for(bb.index()));
1208     }
1209
1210     fn reconstruct_statement_effect(&mut self, loc: Location) {
1211         self.stmt_gen.reset_to_empty();
1212         self.stmt_kill.reset_to_empty();
1213         let mut ignored = IdxSetBuf::new_empty(0);
1214         let mut sets = BlockSets {
1215             on_entry: &mut ignored, gen_set: &mut self.stmt_gen, kill_set: &mut self.stmt_kill,
1216         };
1217         self.base_results.operator().statement_effect(&mut sets, loc);
1218     }
1219
1220     fn reconstruct_terminator_effect(&mut self, loc: Location) {
1221         self.stmt_gen.reset_to_empty();
1222         self.stmt_kill.reset_to_empty();
1223         let mut ignored = IdxSetBuf::new_empty(0);
1224         let mut sets = BlockSets {
1225             on_entry: &mut ignored, gen_set: &mut self.stmt_gen, kill_set: &mut self.stmt_kill,
1226         };
1227         self.base_results.operator().terminator_effect(&mut sets, loc);
1228     }
1229
1230     fn apply_local_effect(&mut self) {
1231         self.curr_state.union(&self.stmt_gen);
1232         self.curr_state.subtract(&self.stmt_kill);
1233     }
1234
1235     fn elems_generated(&self) -> indexed_set::Elems<BD::Idx> {
1236         let univ = self.base_results.sets().bits_per_block();
1237         self.stmt_gen.elems(univ)
1238     }
1239
1240     fn elems_incoming(&self) -> indexed_set::Elems<BD::Idx> {
1241         let univ = self.base_results.sets().bits_per_block();
1242         self.curr_state.elems(univ)
1243     }
1244 }
1245
1246 impl<'tcx> BorrowData<'tcx> {
1247     fn compatible_with(&self, bk: BorrowKind) -> bool {
1248         match (self.kind, bk) {
1249             (BorrowKind::Shared, BorrowKind::Shared) => true,
1250
1251             (BorrowKind::Mut, _) |
1252             (BorrowKind::Unique, _) |
1253             (_, BorrowKind::Mut) |
1254             (_, BorrowKind::Unique) => false,
1255         }
1256     }
1257 }