compiler/rustc_borrowck/src/diagnostics/mutability_errors.rs

   1 use rustc_hir as hir;
   2 use rustc_hir::Node;
   3 use rustc_middle::hir::map::Map;
   4 use rustc_middle::mir::{Mutability, Place, PlaceRef, ProjectionElem};
   5 use rustc_middle::ty::{self, Ty, TyCtxt};
   6 use rustc_middle::{
   7     hir::place::PlaceBase,
   8     mir::{
   9         self, BindingForm, ClearCrossCrate, ImplicitSelfKind, Local, LocalDecl, LocalInfo,
  10         LocalKind, Location,
  11     },
  12 };
  13 use rustc_span::source_map::DesugaringKind;
  14 use rustc_span::symbol::{kw, Symbol};
  15 use rustc_span::{BytePos, Span};
  16
  17 use crate::diagnostics::BorrowedContentSource;
  18 use crate::MirBorrowckCtxt;
  19 use rustc_const_eval::util::collect_writes::FindAssignments;
  20 use rustc_errors::{Applicability, DiagnosticBuilder};
  21
  22 #[derive(Copy, Clone, Debug, Eq, PartialEq)]
  23 pub(crate) enum AccessKind {
  24     MutableBorrow,
  25     Mutate,
  26 }
  27
  28 impl<'a, 'tcx> MirBorrowckCtxt<'a, 'tcx> {
  29     pub(crate) fn report_mutability_error(
  30         &mut self,
  31         access_place: Place<'tcx>,
  32         span: Span,
  33         the_place_err: PlaceRef<'tcx>,
  34         error_access: AccessKind,
  35         location: Location,
  36     ) {
  37         debug!(
  38             "report_mutability_error(\
  39                 access_place={:?}, span={:?}, the_place_err={:?}, error_access={:?}, location={:?},\
  40             )",
  41             access_place, span, the_place_err, error_access, location,
  42         );
  43
  44         let mut err;
  45         let item_msg;
  46         let reason;
  47         let mut opt_source = None;
  48         let access_place_desc = self.describe_any_place(access_place.as_ref());
  49         debug!("report_mutability_error: access_place_desc={:?}", access_place_desc);
  50
  51         match the_place_err {
  52             PlaceRef { local, projection: [] } => {
  53                 item_msg = access_place_desc;
  54                 if access_place.as_local().is_some() {
  55                     reason = ", as it is not declared as mutable".to_string();
  56                 } else {
  57                     let name = self.local_names[local].expect("immutable unnamed local");
  58                     reason = format!(", as `{}` is not declared as mutable", name);
  59                 }
  60             }
  61
  62             PlaceRef {
  63                 local,
  64                 projection: [proj_base @ .., ProjectionElem::Field(upvar_index, _)],
  65             } => {
  66                 debug_assert!(is_closure_or_generator(
  67                     Place::ty_from(local, proj_base, self.body, self.infcx.tcx).ty
  68                 ));
  69
  70                 let imm_borrow_derefed = self.upvars[upvar_index.index()]
  71                     .place
  72                     .place
  73                     .deref_tys()
  74                     .any(|ty| matches!(ty.kind(), ty::Ref(.., hir::Mutability::Not)));
  75
  76                 // If the place is immutable then:
  77                 //
  78                 // - Either we deref an immutable ref to get to our final place.
  79                 //    - We don't capture derefs of raw ptrs
  80                 // - Or the final place is immut because the root variable of the capture
  81                 //   isn't marked mut and we should suggest that to the user.
  82                 if imm_borrow_derefed {
  83                     // If we deref an immutable ref then the suggestion here doesn't help.
  84                     return;
  85                 } else {
  86                     item_msg = access_place_desc;
  87                     if self.is_upvar_field_projection(access_place.as_ref()).is_some() {
  88                         reason = ", as it is not declared as mutable".to_string();
  89                     } else {
  90                         let name = self.upvars[upvar_index.index()].place.to_string(self.infcx.tcx);
  91                         reason = format!(", as `{}` is not declared as mutable", name);
  92                     }
  93                 }
  94             }
  95
  96             PlaceRef { local, projection: [ProjectionElem::Deref] }
  97                 if self.body.local_decls[local].is_ref_for_guard() =>
  98             {
  99                 item_msg = access_place_desc;
 100                 reason = ", as it is immutable for the pattern guard".to_string();
 101             }
 102             PlaceRef { local, projection: [ProjectionElem::Deref] }
 103                 if self.body.local_decls[local].is_ref_to_static() =>
 104             {
 105                 if access_place.projection.len() == 1 {
 106                     item_msg = format!("immutable static item {}", access_place_desc);
 107                     reason = String::new();
 108                 } else {
 109                     item_msg = access_place_desc;
 110                     let local_info = &self.body.local_decls[local].local_info;
 111                     if let Some(box LocalInfo::StaticRef { def_id, .. }) = *local_info {
 112                         let static_name = &self.infcx.tcx.item_name(def_id);
 113                         reason = format!(", as `{}` is an immutable static item", static_name);
 114                     } else {
 115                         bug!("is_ref_to_static return true, but not ref to static?");
 116                     }
 117                 }
 118             }
 119             PlaceRef { local: _, projection: [proj_base @ .., ProjectionElem::Deref] } => {
 120                 if the_place_err.local == ty::CAPTURE_STRUCT_LOCAL
 121                     && proj_base.is_empty()
 122                     && !self.upvars.is_empty()
 123                 {
 124                     item_msg = access_place_desc;
 125                     debug_assert!(
 126                         self.body.local_decls[ty::CAPTURE_STRUCT_LOCAL].ty.is_region_ptr()
 127                     );
 128                     debug_assert!(is_closure_or_generator(
 129                         Place::ty_from(
 130                             the_place_err.local,
 131                             the_place_err.projection,
 132                             self.body,
 133                             self.infcx.tcx
 134                         )
 135                         .ty
 136                     ));
 137
 138                     reason = if self.is_upvar_field_projection(access_place.as_ref()).is_some() {
 139                         ", as it is a captured variable in a `Fn` closure".to_string()
 140                     } else {
 141                         ", as `Fn` closures cannot mutate their captured variables".to_string()
 142                     }
 143                 } else {
 144                     let source = self.borrowed_content_source(PlaceRef {
 145                         local: the_place_err.local,
 146                         projection: proj_base,
 147                     });
 148                     let pointer_type = source.describe_for_immutable_place(self.infcx.tcx);
 149                     opt_source = Some(source);
 150                     if let Some(desc) = self.describe_place(access_place.as_ref()) {
 151                         item_msg = format!("`{}`", desc);
 152                         reason = match error_access {
 153                             AccessKind::Mutate => format!(", which is behind {}", pointer_type),
 154                             AccessKind::MutableBorrow => {
 155                                 format!(", as it is behind {}", pointer_type)
 156                             }
 157                         }
 158                     } else {
 159                         item_msg = format!("data in {}", pointer_type);
 160                         reason = String::new();
 161                     }
 162                 }
 163             }
 164
 165             PlaceRef {
 166                 local: _,
 167                 projection:
 168                     [
 169                         ..,
 170                         ProjectionElem::Index(_)
 171                         | ProjectionElem::ConstantIndex { .. }
 172                         | ProjectionElem::Subslice { .. }
 173                         | ProjectionElem::Downcast(..),
 174                     ],
 175             } => bug!("Unexpected immutable place."),
 176         }
 177
 178         debug!("report_mutability_error: item_msg={:?}, reason={:?}", item_msg, reason);
 179
 180         // `act` and `acted_on` are strings that let us abstract over
 181         // the verbs used in some diagnostic messages.
 182         let act;
 183         let acted_on;
 184
 185         let span = match error_access {
 186             AccessKind::Mutate => {
 187                 err = self.cannot_assign(span, &(item_msg + &reason));
 188                 act = "assign";
 189                 acted_on = "written";
 190                 span
 191             }
 192             AccessKind::MutableBorrow => {
 193                 act = "borrow as mutable";
 194                 acted_on = "borrowed as mutable";
 195
 196                 let borrow_spans = self.borrow_spans(span, location);
 197                 let borrow_span = borrow_spans.args_or_use();
 198                 err = self.cannot_borrow_path_as_mutable_because(borrow_span, &item_msg, &reason);
 199                 borrow_spans.var_span_label(
 200                     &mut err,
 201                     format!(
 202                         "mutable borrow occurs due to use of {} in closure",
 203                         self.describe_any_place(access_place.as_ref()),
 204                     ),
 205                     "mutable",
 206                 );
 207                 borrow_span
 208             }
 209         };
 210
 211         debug!("report_mutability_error: act={:?}, acted_on={:?}", act, acted_on);
 212
 213         match the_place_err {
 214             // Suggest making an existing shared borrow in a struct definition a mutable borrow.
 215             //
 216             // This is applicable when we have a deref of a field access to a deref of a local -
 217             // something like `*((*_1).0`. The local that we get will be a reference to the
 218             // struct we've got a field access of (it must be a reference since there's a deref
 219             // after the field access).
 220             PlaceRef {
 221                 local,
 222                 projection:
 223                     [
 224                         proj_base @ ..,
 225                         ProjectionElem::Deref,
 226                         ProjectionElem::Field(field, _),
 227                         ProjectionElem::Deref,
 228                     ],
 229             } => {
 230                 err.span_label(span, format!("cannot {ACT}", ACT = act));
 231
 232                 if let Some(span) = get_mut_span_in_struct_field(
 233                     self.infcx.tcx,
 234                     Place::ty_from(local, proj_base, self.body, self.infcx.tcx).ty,
 235                     field,
 236                 ) {
 237                     err.span_suggestion_verbose(
 238                         span,
 239                         "consider changing this to be mutable",
 240                         " mut ".into(),
 241                         Applicability::MaybeIncorrect,
 242                     );
 243                 }
 244             }
 245
 246             // Suggest removing a `&mut` from the use of a mutable reference.
 247             PlaceRef { local, projection: [] }
 248                 if self
 249                     .body
 250                     .local_decls
 251                     .get(local)
 252                     .map(|l| mut_borrow_of_mutable_ref(l, self.local_names[local]))
 253                     .unwrap_or(false) =>
 254             {
 255                 let decl = &self.body.local_decls[local];
 256                 err.span_label(span, format!("cannot {ACT}", ACT = act));
 257                 if let Some(mir::Statement {
 258                     source_info,
 259                     kind:
 260                         mir::StatementKind::Assign(box (
 261                             _,
 262                             mir::Rvalue::Ref(
 263                                 _,
 264                                 mir::BorrowKind::Mut { allow_two_phase_borrow: false },
 265                                 _,
 266                             ),
 267                         )),
 268                     ..
 269                 }) = &self.body[location.block].statements.get(location.statement_index)
 270                 {
 271                     match decl.local_info {
 272                         Some(box LocalInfo::User(ClearCrossCrate::Set(BindingForm::Var(
 273                             mir::VarBindingForm {
 274                                 binding_mode: ty::BindingMode::BindByValue(Mutability::Not),
 275                                 opt_ty_info: Some(sp),
 276                                 opt_match_place: _,
 277                                 pat_span: _,
 278                             },
 279                         )))) => {
 280                             err.span_note(sp, "the binding is already a mutable borrow");
 281                         }
 282                         _ => {
 283                             err.span_note(
 284                                 decl.source_info.span,
 285                                 "the binding is already a mutable borrow",
 286                             );
 287                         }
 288                     }
 289                     if let Ok(snippet) =
 290                         self.infcx.tcx.sess.source_map().span_to_snippet(source_info.span)
 291                     {
 292                         if snippet.starts_with("&mut ") {
 293                             // We don't have access to the HIR to get accurate spans, but we can
 294                             // give a best effort structured suggestion.
 295                             err.span_suggestion_verbose(
 296                                 source_info.span.with_hi(source_info.span.lo() + BytePos(5)),
 297                                 "try removing `&mut` here",
 298                                 String::new(),
 299                                 Applicability::MachineApplicable,
 300                             );
 301                         } else {
 302                             // This can occur with things like `(&mut self).foo()`.
 303                             err.span_help(source_info.span, "try removing `&mut` here");
 304                         }
 305                     } else {
 306                         err.span_help(source_info.span, "try removing `&mut` here");
 307                     }
 308                 } else if decl.mutability == Mutability::Not
 309                     && !matches!(
 310                         decl.local_info,
 311                         Some(box LocalInfo::User(ClearCrossCrate::Set(BindingForm::ImplicitSelf(
 312                             ImplicitSelfKind::MutRef
 313                         ))))
 314                     )
 315                 {
 316                     err.span_suggestion_verbose(
 317                         decl.source_info.span.shrink_to_lo(),
 318                         "consider making the binding mutable",
 319                         "mut ".to_string(),
 320                         Applicability::MachineApplicable,
 321                     );
 322                 }
 323             }
 324
 325             // We want to suggest users use `let mut` for local (user
 326             // variable) mutations...
 327             PlaceRef { local, projection: [] }
 328                 if self.body.local_decls[local].can_be_made_mutable() =>
 329             {
 330                 // ... but it doesn't make sense to suggest it on
 331                 // variables that are `ref x`, `ref mut x`, `&self`,
 332                 // or `&mut self` (such variables are simply not
 333                 // mutable).
 334                 let local_decl = &self.body.local_decls[local];
 335                 assert_eq!(local_decl.mutability, Mutability::Not);
 336
 337                 err.span_label(span, format!("cannot {ACT}", ACT = act));
 338                 err.span_suggestion(
 339                     local_decl.source_info.span,
 340                     "consider changing this to be mutable",
 341                     format!("mut {}", self.local_names[local].unwrap()),
 342                     Applicability::MachineApplicable,
 343                 );
 344                 let tcx = self.infcx.tcx;
 345                 if let ty::Closure(id, _) = the_place_err.ty(self.body, tcx).ty.kind() {
 346                     self.show_mutating_upvar(tcx, id, the_place_err, &mut err);
 347                 }
 348             }
 349
 350             // Also suggest adding mut for upvars
 351             PlaceRef {
 352                 local,
 353                 projection: [proj_base @ .., ProjectionElem::Field(upvar_index, _)],
 354             } => {
 355                 debug_assert!(is_closure_or_generator(
 356                     Place::ty_from(local, proj_base, self.body, self.infcx.tcx).ty
 357                 ));
 358
 359                 let captured_place = &self.upvars[upvar_index.index()].place;
 360
 361                 err.span_label(span, format!("cannot {ACT}", ACT = act));
 362
 363                 let upvar_hir_id = captured_place.get_root_variable();
 364
 365                 if let Some(Node::Binding(pat)) = self.infcx.tcx.hir().find(upvar_hir_id) {
 366                     if let hir::PatKind::Binding(
 367                         hir::BindingAnnotation::Unannotated,
 368                         _,
 369                         upvar_ident,
 370                         _,
 371                     ) = pat.kind
 372                     {
 373                         err.span_suggestion(
 374                             upvar_ident.span,
 375                             "consider changing this to be mutable",
 376                             format!("mut {}", upvar_ident.name),
 377                             Applicability::MachineApplicable,
 378                         );
 379                     }
 380                 }
 381
 382                 let tcx = self.infcx.tcx;
 383                 if let ty::Ref(_, ty, Mutability::Mut) = the_place_err.ty(self.body, tcx).ty.kind()
 384                 {
 385                     if let ty::Closure(id, _) = ty.kind() {
 386                         self.show_mutating_upvar(tcx, id, the_place_err, &mut err);
 387                     }
 388                 }
 389             }
 390
 391             // complete hack to approximate old AST-borrowck
 392             // diagnostic: if the span starts with a mutable borrow of
 393             // a local variable, then just suggest the user remove it.
 394             PlaceRef { local: _, projection: [] }
 395                 if {
 396                     if let Ok(snippet) = self.infcx.tcx.sess.source_map().span_to_snippet(span) {
 397                         snippet.starts_with("&mut ")
 398                     } else {
 399                         false
 400                     }
 401                 } =>
 402             {
 403                 err.span_label(span, format!("cannot {ACT}", ACT = act));
 404                 err.span_suggestion(
 405                     span,
 406                     "try removing `&mut` here",
 407                     String::new(),
 408                     Applicability::MaybeIncorrect,
 409                 );
 410             }
 411
 412             PlaceRef { local, projection: [ProjectionElem::Deref] }
 413                 if self.body.local_decls[local].is_ref_for_guard() =>
 414             {
 415                 err.span_label(span, format!("cannot {ACT}", ACT = act));
 416                 err.note(
 417                     "variables bound in patterns are immutable until the end of the pattern guard",
 418                 );
 419             }
 420
 421             // We want to point out when a `&` can be readily replaced
 422             // with an `&mut`.
 423             //
 424             // FIXME: can this case be generalized to work for an
 425             // arbitrary base for the projection?
 426             PlaceRef { local, projection: [ProjectionElem::Deref] }
 427                 if self.body.local_decls[local].is_user_variable() =>
 428             {
 429                 let local_decl = &self.body.local_decls[local];
 430
 431                 let (pointer_sigil, pointer_desc) = if local_decl.ty.is_region_ptr() {
 432                     ("&", "reference")
 433                 } else {
 434                     ("*const", "pointer")
 435                 };
 436
 437                 match self.local_names[local] {
 438                     Some(name) if !local_decl.from_compiler_desugaring() => {
 439                         let label = match local_decl.local_info.as_ref().unwrap() {
 440                             box LocalInfo::User(ClearCrossCrate::Set(
 441                                 mir::BindingForm::ImplicitSelf(_),
 442                             )) => {
 443                                 let (span, suggestion) =
 444                                     suggest_ampmut_self(self.infcx.tcx, local_decl);
 445                                 Some((true, span, suggestion))
 446                             }
 447
 448                             box LocalInfo::User(ClearCrossCrate::Set(mir::BindingForm::Var(
 449                                 mir::VarBindingForm {
 450                                     binding_mode: ty::BindingMode::BindByValue(_),
 451                                     opt_ty_info,
 452                                     ..
 453                                 },
 454                             ))) => {
 455                                 // check if the RHS is from desugaring
 456                                 let opt_assignment_rhs_span =
 457                                     self.body.find_assignments(local).first().map(|&location| {
 458                                         if let Some(mir::Statement {
 459                                             source_info: _,
 460                                             kind:
 461                                                 mir::StatementKind::Assign(box (
 462                                                     _,
 463                                                     mir::Rvalue::Use(mir::Operand::Copy(place)),
 464                                                 )),
 465                                         }) = self.body[location.block]
 466                                             .statements
 467                                             .get(location.statement_index)
 468                                         {
 469                                             self.body.local_decls[place.local].source_info.span
 470                                         } else {
 471                                             self.body.source_info(location).span
 472                                         }
 473                                     });
 474                                 match opt_assignment_rhs_span.and_then(|s| s.desugaring_kind()) {
 475                                     // on for loops, RHS points to the iterator part
 476                                     Some(DesugaringKind::ForLoop) => {
 477                                         self.suggest_similar_mut_method_for_for_loop(&mut err);
 478                                         Some((
 479                                             false,
 480                                             opt_assignment_rhs_span.unwrap(),
 481                                             format!(
 482                                                 "this iterator yields `{SIGIL}` {DESC}s",
 483                                                 SIGIL = pointer_sigil,
 484                                                 DESC = pointer_desc
 485                                             ),
 486                                         ))
 487                                     }
 488                                     // don't create labels for compiler-generated spans
 489                                     Some(_) => None,
 490                                     None => {
 491                                         let (span, suggestion) = if name != kw::SelfLower {
 492                                             suggest_ampmut(
 493                                                 self.infcx.tcx,
 494                                                 local_decl,
 495                                                 opt_assignment_rhs_span,
 496                                                 *opt_ty_info,
 497                                             )
 498                                         } else {
 499                                             match local_decl.local_info.as_deref() {
 500                                                 Some(LocalInfo::User(ClearCrossCrate::Set(
 501                                                     mir::BindingForm::Var(mir::VarBindingForm {
 502                                                         opt_ty_info: None,
 503                                                         ..
 504                                                     }),
 505                                                 ))) => {
 506                                                     suggest_ampmut_self(self.infcx.tcx, local_decl)
 507                                                 }
 508                                                 // explicit self (eg `self: &'a Self`)
 509                                                 _ => suggest_ampmut(
 510                                                     self.infcx.tcx,
 511                                                     local_decl,
 512                                                     opt_assignment_rhs_span,
 513                                                     *opt_ty_info,
 514                                                 ),
 515                                             }
 516                                         };
 517                                         Some((true, span, suggestion))
 518                                     }
 519                                 }
 520                             }
 521
 522                             box LocalInfo::User(ClearCrossCrate::Set(mir::BindingForm::Var(
 523                                 mir::VarBindingForm {
 524                                     binding_mode: ty::BindingMode::BindByReference(_),
 525                                     ..
 526                                 },
 527                             ))) => {
 528                                 let pattern_span = local_decl.source_info.span;
 529                                 suggest_ref_mut(self.infcx.tcx, pattern_span)
 530                                     .map(|replacement| (true, pattern_span, replacement))
 531                             }
 532
 533                             box LocalInfo::User(ClearCrossCrate::Clear) => {
 534                                 bug!("saw cleared local state")
 535                             }
 536
 537                             _ => unreachable!(),
 538                         };
 539
 540                         match label {
 541                             Some((true, err_help_span, suggested_code)) => {
 542                                 let (is_trait_sig, local_trait) = self.is_error_in_trait(local);
 543                                 if !is_trait_sig {
 544                                     err.span_suggestion(
 545                                         err_help_span,
 546                                         &format!(
 547                                             "consider changing this to be a mutable {}",
 548                                             pointer_desc
 549                                         ),
 550                                         suggested_code,
 551                                         Applicability::MachineApplicable,
 552                                     );
 553                                 } else if let Some(x) = local_trait {
 554                                     err.span_suggestion(
 555                                         x,
 556                                         &format!(
 557                                             "consider changing that to be a mutable {}",
 558                                             pointer_desc
 559                                         ),
 560                                         suggested_code,
 561                                         Applicability::MachineApplicable,
 562                                     );
 563                                 }
 564                             }
 565                             Some((false, err_label_span, message)) => {
 566                                 err.span_label(err_label_span, &message);
 567                             }
 568                             None => {}
 569                         }
 570                         err.span_label(
 571                             span,
 572                             format!(
 573                                 "`{NAME}` is a `{SIGIL}` {DESC}, \
 574                                 so the data it refers to cannot be {ACTED_ON}",
 575                                 NAME = name,
 576                                 SIGIL = pointer_sigil,
 577                                 DESC = pointer_desc,
 578                                 ACTED_ON = acted_on
 579                             ),
 580                         );
 581                     }
 582                     _ => {
 583                         err.span_label(
 584                             span,
 585                             format!(
 586                                 "cannot {ACT} through `{SIGIL}` {DESC}",
 587                                 ACT = act,
 588                                 SIGIL = pointer_sigil,
 589                                 DESC = pointer_desc
 590                             ),
 591                         );
 592                     }
 593                 }
 594             }
 595
 596             PlaceRef { local, projection: [ProjectionElem::Deref] }
 597                 if local == ty::CAPTURE_STRUCT_LOCAL && !self.upvars.is_empty() =>
 598             {
 599                 self.expected_fn_found_fn_mut_call(&mut err, span, act);
 600             }
 601
 602             PlaceRef { local: _, projection: [.., ProjectionElem::Deref] } => {
 603                 err.span_label(span, format!("cannot {ACT}", ACT = act));
 604
 605                 match opt_source {
 606                     Some(BorrowedContentSource::OverloadedDeref(ty)) => {
 607                         err.help(&format!(
 608                             "trait `DerefMut` is required to modify through a dereference, \
 609                                 but it is not implemented for `{}`",
 610                             ty,
 611                         ));
 612                     }
 613                     Some(BorrowedContentSource::OverloadedIndex(ty)) => {
 614                         err.help(&format!(
 615                             "trait `IndexMut` is required to modify indexed content, \
 616                                 but it is not implemented for `{}`",
 617                             ty,
 618                         ));
 619                     }
 620                     _ => (),
 621                 }
 622             }
 623
 624             _ => {
 625                 err.span_label(span, format!("cannot {ACT}", ACT = act));
 626             }
 627         }
 628
 629         self.buffer_error(err);
 630     }
 631
 632     /// User cannot make signature of a trait mutable without changing the
 633     /// trait. So we find if this error belongs to a trait and if so we move
 634     /// suggestion to the trait or disable it if it is out of scope of this crate
 635     fn is_error_in_trait(&self, local: Local) -> (bool, Option<Span>) {
 636         if self.body.local_kind(local) != LocalKind::Arg {
 637             return (false, None);
 638         }
 639         let hir_map = self.infcx.tcx.hir();
 640         let my_def = self.body.source.def_id();
 641         let my_hir = hir_map.local_def_id_to_hir_id(my_def.as_local().unwrap());
 642         let Some(td) =
 643             self.infcx.tcx.impl_of_method(my_def).and_then(|x| self.infcx.tcx.trait_id_of_impl(x))
 644         else {
 645             return (false, None);
 646         };
 647         (
 648             true,
 649             td.as_local().and_then(|tld| match hir_map.find_by_def_id(tld) {
 650                 Some(Node::Item(hir::Item {
 651                     kind: hir::ItemKind::Trait(_, _, _, _, items),
 652                     ..
 653                 })) => {
 654                     let mut f_in_trait_opt = None;
 655                     for hir::TraitItemRef { id: fi, kind: k, .. } in *items {
 656                         let hi = fi.hir_id();
 657                         if !matches!(k, hir::AssocItemKind::Fn { .. }) {
 658                             continue;
 659                         }
 660                         if hir_map.name(hi) != hir_map.name(my_hir) {
 661                             continue;
 662                         }
 663                         f_in_trait_opt = Some(hi);
 664                         break;
 665                     }
 666                     f_in_trait_opt.and_then(|f_in_trait| match hir_map.find(f_in_trait) {
 667                         Some(Node::TraitItem(hir::TraitItem {
 668                             kind:
 669                                 hir::TraitItemKind::Fn(
 670                                     hir::FnSig { decl: hir::FnDecl { inputs, .. }, .. },
 671                                     _,
 672                                 ),
 673                             ..
 674                         })) => {
 675                             let hir::Ty { span, .. } = inputs[local.index() - 1];
 676                             Some(span)
 677                         }
 678                         _ => None,
 679                     })
 680                 }
 681                 _ => None,
 682             }),
 683         )
 684     }
 685
 686     // point to span of upvar making closure call require mutable borrow
 687     fn show_mutating_upvar(
 688         &self,
 689         tcx: TyCtxt<'_>,
 690         id: &hir::def_id::DefId,
 691         the_place_err: PlaceRef<'tcx>,
 692         err: &mut DiagnosticBuilder<'_>,
 693     ) {
 694         let closure_local_def_id = id.expect_local();
 695         let tables = tcx.typeck(closure_local_def_id);
 696         let closure_hir_id = tcx.hir().local_def_id_to_hir_id(closure_local_def_id);
 697         if let Some((span, closure_kind_origin)) =
 698             &tables.closure_kind_origins().get(closure_hir_id)
 699         {
 700             let reason = if let PlaceBase::Upvar(upvar_id) = closure_kind_origin.base {
 701                 let upvar = ty::place_to_string_for_capture(tcx, closure_kind_origin);
 702                 let root_hir_id = upvar_id.var_path.hir_id;
 703                 // we have an origin for this closure kind starting at this root variable so it's safe to unwrap here
 704                 let captured_places = tables.closure_min_captures[id].get(&root_hir_id).unwrap();
 705
 706                 let origin_projection = closure_kind_origin
 707                     .projections
 708                     .iter()
 709                     .map(|proj| proj.kind)
 710                     .collect::<Vec<_>>();
 711                 let mut capture_reason = String::new();
 712                 for captured_place in captured_places {
 713                     let captured_place_kinds = captured_place
 714                         .place
 715                         .projections
 716                         .iter()
 717                         .map(|proj| proj.kind)
 718                         .collect::<Vec<_>>();
 719                     if rustc_middle::ty::is_ancestor_or_same_capture(
 720                         &captured_place_kinds,
 721                         &origin_projection,
 722                     ) {
 723                         match captured_place.info.capture_kind {
 724                             ty::UpvarCapture::ByRef(
 725                                 ty::BorrowKind::MutBorrow | ty::BorrowKind::UniqueImmBorrow,
 726                             ) => {
 727                                 capture_reason = format!("mutable borrow of `{}`", upvar);
 728                             }
 729                             ty::UpvarCapture::ByValue => {
 730                                 capture_reason = format!("possible mutation of `{}`", upvar);
 731                             }
 732                             _ => bug!("upvar `{}` borrowed, but not mutably", upvar),
 733                         }
 734                         break;
 735                     }
 736                 }
 737                 if capture_reason.is_empty() {
 738                     bug!("upvar `{}` borrowed, but cannot find reason", upvar);
 739                 }
 740                 capture_reason
 741             } else {
 742                 bug!("not an upvar")
 743             };
 744             err.span_label(
 745                 *span,
 746                 format!(
 747                     "calling `{}` requires mutable binding due to {}",
 748                     self.describe_place(the_place_err).unwrap(),
 749                     reason
 750                 ),
 751             );
 752         }
 753     }
 754
 755     // Attempt to search similar mutable associated items for suggestion.
 756     // In the future, attempt in all path but initially for RHS of for_loop
 757     fn suggest_similar_mut_method_for_for_loop(&self, err: &mut DiagnosticBuilder<'_>) {
 758         use hir::{
 759             BodyId, Expr,
 760             ExprKind::{Block, Call, DropTemps, Match, MethodCall},
 761             HirId, ImplItem, ImplItemKind, Item, ItemKind,
 762         };
 763
 764         fn maybe_body_id_of_fn(hir_map: Map<'_>, id: HirId) -> Option<BodyId> {
 765             match hir_map.find(id) {
 766                 Some(Node::Item(Item { kind: ItemKind::Fn(_, _, body_id), .. }))
 767                 | Some(Node::ImplItem(ImplItem { kind: ImplItemKind::Fn(_, body_id), .. })) => {
 768                     Some(*body_id)
 769                 }
 770                 _ => None,
 771             }
 772         }
 773         let hir_map = self.infcx.tcx.hir();
 774         let mir_body_hir_id = self.mir_hir_id();
 775         if let Some(fn_body_id) = maybe_body_id_of_fn(hir_map, mir_body_hir_id) {
 776             if let Block(
 777                 hir::Block {
 778                     expr:
 779                         Some(Expr {
 780                             kind:
 781                                 DropTemps(Expr {
 782                                     kind:
 783                                         Match(
 784                                             Expr {
 785                                                 kind:
 786                                                     Call(
 787                                                         _,
 788                                                         [
 789                                                             Expr {
 790                                                                 kind: MethodCall(path_segment, ..),
 791                                                                 hir_id,
 792                                                                 ..
 793                                                             },
 794                                                             ..,
 795                                                         ],
 796                                                     ),
 797                                                 ..
 798                                             },
 799                                             ..,
 800                                         ),
 801                                     ..
 802                                 }),
 803                             ..
 804                         }),
 805                     ..
 806                 },
 807                 _,
 808             ) = hir_map.body(fn_body_id).value.kind
 809             {
 810                 let opt_suggestions = path_segment
 811                     .hir_id
 812                     .map(|path_hir_id| self.infcx.tcx.typeck(path_hir_id.owner))
 813                     .and_then(|typeck| typeck.type_dependent_def_id(*hir_id))
 814                     .and_then(|def_id| self.infcx.tcx.impl_of_method(def_id))
 815                     .map(|def_id| self.infcx.tcx.associated_items(def_id))
 816                     .map(|assoc_items| {
 817                         assoc_items
 818                             .in_definition_order()
 819                             .map(|assoc_item_def| assoc_item_def.ident(self.infcx.tcx))
 820                             .filter(|&ident| {
 821                                 let original_method_ident = path_segment.ident;
 822                                 original_method_ident != ident
 823                                     && ident
 824                                         .as_str()
 825                                         .starts_with(&original_method_ident.name.to_string())
 826                             })
 827                             .map(|ident| format!("{}()", ident))
 828                             .peekable()
 829                     });
 830
 831                 if let Some(mut suggestions) = opt_suggestions {
 832                     if suggestions.peek().is_some() {
 833                         err.span_suggestions(
 834                             path_segment.ident.span,
 835                             "use mutable method",
 836                             suggestions,
 837                             Applicability::MaybeIncorrect,
 838                         );
 839                     }
 840                 }
 841             }
 842         };
 843     }
 844
 845     /// Targeted error when encountering an `FnMut` closure where an `Fn` closure was expected.
 846     fn expected_fn_found_fn_mut_call(&self, err: &mut DiagnosticBuilder<'_>, sp: Span, act: &str) {
 847         err.span_label(sp, format!("cannot {}", act));
 848
 849         let hir = self.infcx.tcx.hir();
 850         let closure_id = self.mir_hir_id();
 851         let fn_call_id = hir.get_parent_node(closure_id);
 852         let node = hir.get(fn_call_id);
 853         let item_id = hir.enclosing_body_owner(fn_call_id);
 854         let mut look_at_return = true;
 855         // If we can detect the expression to be an `fn` call where the closure was an argument,
 856         // we point at the `fn` definition argument...
 857         if let hir::Node::Expr(hir::Expr { kind: hir::ExprKind::Call(func, args), .. }) = node {
 858             let arg_pos = args
 859                 .iter()
 860                 .enumerate()
 861                 .filter(|(_, arg)| arg.span == self.body.span)
 862                 .map(|(pos, _)| pos)
 863                 .next();
 864             let def_id = hir.local_def_id(item_id);
 865             let tables = self.infcx.tcx.typeck(def_id);
 866             if let Some(ty::FnDef(def_id, _)) =
 867                 tables.node_type_opt(func.hir_id).as_ref().map(|ty| ty.kind())
 868             {
 869                 let arg = match hir.get_if_local(*def_id) {
 870                     Some(
 871                         hir::Node::Item(hir::Item {
 872                             ident, kind: hir::ItemKind::Fn(sig, ..), ..
 873                         })
 874                         | hir::Node::TraitItem(hir::TraitItem {
 875                             ident,
 876                             kind: hir::TraitItemKind::Fn(sig, _),
 877                             ..
 878                         })
 879                         | hir::Node::ImplItem(hir::ImplItem {
 880                             ident,
 881                             kind: hir::ImplItemKind::Fn(sig, _),
 882                             ..
 883                         }),
 884                     ) => Some(
 885                         arg_pos
 886                             .and_then(|pos| {
 887                                 sig.decl.inputs.get(
 888                                     pos + if sig.decl.implicit_self.has_implicit_self() {
 889                                         1
 890                                     } else {
 891                                         0
 892                                     },
 893                                 )
 894                             })
 895                             .map(|arg| arg.span)
 896                             .unwrap_or(ident.span),
 897                     ),
 898                     _ => None,
 899                 };
 900                 if let Some(span) = arg {
 901                     err.span_label(span, "change this to accept `FnMut` instead of `Fn`");
 902                     err.span_label(func.span, "expects `Fn` instead of `FnMut`");
 903                     if self.infcx.tcx.sess.source_map().is_multiline(self.body.span) {
 904                         err.span_label(self.body.span, "in this closure");
 905                     }
 906                     look_at_return = false;
 907                 }
 908             }
 909         }
 910
 911         if look_at_return && hir.get_return_block(closure_id).is_some() {
 912             // ...otherwise we are probably in the tail expression of the function, point at the
 913             // return type.
 914             match hir.get_by_def_id(hir.get_parent_item(fn_call_id)) {
 915                 hir::Node::Item(hir::Item { ident, kind: hir::ItemKind::Fn(sig, ..), .. })
 916                 | hir::Node::TraitItem(hir::TraitItem {
 917                     ident,
 918                     kind: hir::TraitItemKind::Fn(sig, _),
 919                     ..
 920                 })
 921                 | hir::Node::ImplItem(hir::ImplItem {
 922                     ident,
 923                     kind: hir::ImplItemKind::Fn(sig, _),
 924                     ..
 925                 }) => {
 926                     err.span_label(ident.span, "");
 927                     err.span_label(
 928                         sig.decl.output.span(),
 929                         "change this to return `FnMut` instead of `Fn`",
 930                     );
 931                     err.span_label(self.body.span, "in this closure");
 932                 }
 933                 _ => {}
 934             }
 935         }
 936     }
 937 }
 938
 939 fn mut_borrow_of_mutable_ref(local_decl: &LocalDecl<'_>, local_name: Option<Symbol>) -> bool {
 940     debug!("local_info: {:?}, ty.kind(): {:?}", local_decl.local_info, local_decl.ty.kind());
 941
 942     match local_decl.local_info.as_deref() {
 943         // Check if mutably borrowing a mutable reference.
 944         Some(LocalInfo::User(ClearCrossCrate::Set(mir::BindingForm::Var(
 945             mir::VarBindingForm {
 946                 binding_mode: ty::BindingMode::BindByValue(Mutability::Not), ..
 947             },
 948         )))) => matches!(local_decl.ty.kind(), ty::Ref(_, _, hir::Mutability::Mut)),
 949         Some(LocalInfo::User(ClearCrossCrate::Set(mir::BindingForm::ImplicitSelf(kind)))) => {
 950             // Check if the user variable is a `&mut self` and we can therefore
 951             // suggest removing the `&mut`.
 952             //
 953             // Deliberately fall into this case for all implicit self types,
 954             // so that we don't fall in to the next case with them.
 955             *kind == mir::ImplicitSelfKind::MutRef
 956         }
 957         _ if Some(kw::SelfLower) == local_name => {
 958             // Otherwise, check if the name is the `self` keyword - in which case
 959             // we have an explicit self. Do the same thing in this case and check
 960             // for a `self: &mut Self` to suggest removing the `&mut`.
 961             matches!(local_decl.ty.kind(), ty::Ref(_, _, hir::Mutability::Mut))
 962         }
 963         _ => false,
 964     }
 965 }
 966
 967 fn suggest_ampmut_self<'tcx>(
 968     tcx: TyCtxt<'tcx>,
 969     local_decl: &mir::LocalDecl<'tcx>,
 970 ) -> (Span, String) {
 971     let sp = local_decl.source_info.span;
 972     (
 973         sp,
 974         match tcx.sess.source_map().span_to_snippet(sp) {
 975             Ok(snippet) => {
 976                 let lt_pos = snippet.find('\'');
 977                 if let Some(lt_pos) = lt_pos {
 978                     format!("&{}mut self", &snippet[lt_pos..snippet.len() - 4])
 979                 } else {
 980                     "&mut self".to_string()
 981                 }
 982             }
 983             _ => "&mut self".to_string(),
 984         },
 985     )
 986 }
 987
 988 // When we want to suggest a user change a local variable to be a `&mut`, there
 989 // are three potential "obvious" things to highlight:
 990 //
 991 // let ident [: Type] [= RightHandSideExpression];
 992 //     ^^^^^    ^^^^     ^^^^^^^^^^^^^^^^^^^^^^^
 993 //     (1.)     (2.)              (3.)
 994 //
 995 // We can always fallback on highlighting the first. But chances are good that
 996 // the user experience will be better if we highlight one of the others if possible;
 997 // for example, if the RHS is present and the Type is not, then the type is going to
 998 // be inferred *from* the RHS, which means we should highlight that (and suggest
 999 // that they borrow the RHS mutably).
1000 //
1001 // This implementation attempts to emulate AST-borrowck prioritization
1002 // by trying (3.), then (2.) and finally falling back on (1.).
1003 fn suggest_ampmut<'tcx>(
1004     tcx: TyCtxt<'tcx>,
1005     local_decl: &mir::LocalDecl<'tcx>,
1006     opt_assignment_rhs_span: Option<Span>,
1007     opt_ty_info: Option<Span>,
1008 ) -> (Span, String) {
1009     if let Some(assignment_rhs_span) = opt_assignment_rhs_span {
1010         if let Ok(src) = tcx.sess.source_map().span_to_snippet(assignment_rhs_span) {
1011             let is_mutbl = |ty: &str| -> bool {
1012                 if let Some(rest) = ty.strip_prefix("mut") {
1013                     match rest.chars().next() {
1014                         // e.g. `&mut x`
1015                         Some(c) if c.is_whitespace() => true,
1016                         // e.g. `&mut(x)`
1017                         Some('(') => true,
1018                         // e.g. `&mut{x}`
1019                         Some('{') => true,
1020                         // e.g. `&mutablevar`
1021                         _ => false,
1022                     }
1023                 } else {
1024                     false
1025                 }
1026             };
1027             if let (true, Some(ws_pos)) = (src.starts_with("&'"), src.find(char::is_whitespace)) {
1028                 let lt_name = &src[1..ws_pos];
1029                 let ty = src[ws_pos..].trim_start();
1030                 if !is_mutbl(ty) {
1031                     return (assignment_rhs_span, format!("&{} mut {}", lt_name, ty));
1032                 }
1033             } else if let Some(stripped) = src.strip_prefix('&') {
1034                 let stripped = stripped.trim_start();
1035                 if !is_mutbl(stripped) {
1036                     return (assignment_rhs_span, format!("&mut {}", stripped));
1037                 }
1038             }
1039         }
1040     }
1041
1042     let highlight_span = match opt_ty_info {
1043         // if this is a variable binding with an explicit type,
1044         // try to highlight that for the suggestion.
1045         Some(ty_span) => ty_span,
1046
1047         // otherwise, just highlight the span associated with
1048         // the (MIR) LocalDecl.
1049         None => local_decl.source_info.span,
1050     };
1051
1052     if let Ok(src) = tcx.sess.source_map().span_to_snippet(highlight_span) {
1053         if let (true, Some(ws_pos)) = (src.starts_with("&'"), src.find(char::is_whitespace)) {
1054             let lt_name = &src[1..ws_pos];
1055             let ty = &src[ws_pos..];
1056             return (highlight_span, format!("&{} mut{}", lt_name, ty));
1057         }
1058     }
1059
1060     let ty_mut = local_decl.ty.builtin_deref(true).unwrap();
1061     assert_eq!(ty_mut.mutbl, hir::Mutability::Not);
1062     (
1063         highlight_span,
1064         if local_decl.ty.is_region_ptr() {
1065             format!("&mut {}", ty_mut.ty)
1066         } else {
1067             format!("*mut {}", ty_mut.ty)
1068         },
1069     )
1070 }
1071
1072 fn is_closure_or_generator(ty: Ty<'_>) -> bool {
1073     ty.is_closure() || ty.is_generator()
1074 }
1075
1076 /// Given a field that needs to be mutable, returns a span where the " mut " could go.
1077 /// This function expects the local to be a reference to a struct in order to produce a span.
1078 ///
1079 /// ```text
1080 /// LL |     s: &'a   String
1081 ///    |           ^^^ returns a span taking up the space here
1082 /// ```
1083 fn get_mut_span_in_struct_field<'tcx>(
1084     tcx: TyCtxt<'tcx>,
1085     ty: Ty<'tcx>,
1086     field: &mir::Field,
1087 ) -> Option<Span> {
1088     // Expect our local to be a reference to a struct of some kind.
1089     if let ty::Ref(_, ty, _) = ty.kind() {
1090         if let ty::Adt(def, _) = ty.kind() {
1091             let field = def.all_fields().nth(field.index())?;
1092             // Use the HIR types to construct the diagnostic message.
1093             let node = tcx.hir().find_by_def_id(field.did.as_local()?)?;
1094             // Now we're dealing with the actual struct that we're going to suggest a change to,
1095             // we can expect a field that is an immutable reference to a type.
1096             if let hir::Node::Field(field) = node {
1097                 if let hir::TyKind::Rptr(lifetime, hir::MutTy { mutbl: hir::Mutability::Not, ty }) =
1098                     field.ty.kind
1099                 {
1100                     return Some(lifetime.span.between(ty.span));
1101                 }
1102             }
1103         }
1104     }
1105
1106     None
1107 }
1108
1109 /// If possible, suggest replacing `ref` with `ref mut`.
1110 fn suggest_ref_mut(tcx: TyCtxt<'_>, binding_span: Span) -> Option<String> {
1111     let hi_src = tcx.sess.source_map().span_to_snippet(binding_span).ok()?;
1112     if hi_src.starts_with("ref") && hi_src["ref".len()..].starts_with(rustc_lexer::is_whitespace) {
1113         let replacement = format!("ref mut{}", &hi_src["ref".len()..]);
1114         Some(replacement)
1115     } else {
1116         None
1117     }
1118 }