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, Diagnostic};
  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 `{name}` is not declared as mutable");
  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 `{name}` is not declared as mutable");
  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 `{static_name}` is an immutable static item");
 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                         ref 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 ",
 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                                 "",
 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 ",
 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::Pat(pat)) = self.infcx.tcx.hir().find(upvar_hir_id)
 366                     && 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                 let tcx = self.infcx.tcx;
 382                 if let ty::Ref(_, ty, Mutability::Mut) = the_place_err.ty(self.body, tcx).ty.kind()
 383                     && let ty::Closure(id, _) = *ty.kind()
 384                 {
 385                     self.show_mutating_upvar(tcx, id, the_place_err, &mut err);
 386                 }
 387             }
 388
 389             // complete hack to approximate old AST-borrowck
 390             // diagnostic: if the span starts with a mutable borrow of
 391             // a local variable, then just suggest the user remove it.
 392             PlaceRef { local: _, projection: [] }
 393                 if {
 394                     if let Ok(snippet) = self.infcx.tcx.sess.source_map().span_to_snippet(span) {
 395                         snippet.starts_with("&mut ")
 396                     } else {
 397                         false
 398                     }
 399                 } =>
 400             {
 401                 err.span_label(span, format!("cannot {ACT}", ACT = act));
 402                 err.span_suggestion(
 403                     span,
 404                     "try removing `&mut` here",
 405                     "",
 406                     Applicability::MaybeIncorrect,
 407                 );
 408             }
 409
 410             PlaceRef { local, projection: [ProjectionElem::Deref] }
 411                 if self.body.local_decls[local].is_ref_for_guard() =>
 412             {
 413                 err.span_label(span, format!("cannot {ACT}", ACT = act));
 414                 err.note(
 415                     "variables bound in patterns are immutable until the end of the pattern guard",
 416                 );
 417             }
 418
 419             // We want to point out when a `&` can be readily replaced
 420             // with an `&mut`.
 421             //
 422             // FIXME: can this case be generalized to work for an
 423             // arbitrary base for the projection?
 424             PlaceRef { local, projection: [ProjectionElem::Deref] }
 425                 if self.body.local_decls[local].is_user_variable() =>
 426             {
 427                 let local_decl = &self.body.local_decls[local];
 428
 429                 let (pointer_sigil, pointer_desc) = if local_decl.ty.is_region_ptr() {
 430                     ("&", "reference")
 431                 } else {
 432                     ("*const", "pointer")
 433                 };
 434
 435                 match self.local_names[local] {
 436                     Some(name) if !local_decl.from_compiler_desugaring() => {
 437                         let label = match local_decl.local_info.as_deref().unwrap() {
 438                             LocalInfo::User(ClearCrossCrate::Set(
 439                                 mir::BindingForm::ImplicitSelf(_),
 440                             )) => {
 441                                 let (span, suggestion) =
 442                                     suggest_ampmut_self(self.infcx.tcx, local_decl);
 443                                 Some((true, span, suggestion))
 444                             }
 445
 446                             LocalInfo::User(ClearCrossCrate::Set(mir::BindingForm::Var(
 447                                 mir::VarBindingForm {
 448                                     binding_mode: ty::BindingMode::BindByValue(_),
 449                                     opt_ty_info,
 450                                     ..
 451                                 },
 452                             ))) => {
 453                                 // check if the RHS is from desugaring
 454                                 let opt_assignment_rhs_span =
 455                                     self.body.find_assignments(local).first().map(|&location| {
 456                                         if let Some(mir::Statement {
 457                                             source_info: _,
 458                                             kind:
 459                                                 mir::StatementKind::Assign(box (
 460                                                     _,
 461                                                     mir::Rvalue::Use(mir::Operand::Copy(place)),
 462                                                 )),
 463                                         }) = self.body[location.block]
 464                                             .statements
 465                                             .get(location.statement_index)
 466                                         {
 467                                             self.body.local_decls[place.local].source_info.span
 468                                         } else {
 469                                             self.body.source_info(location).span
 470                                         }
 471                                     });
 472                                 match opt_assignment_rhs_span.and_then(|s| s.desugaring_kind()) {
 473                                     // on for loops, RHS points to the iterator part
 474                                     Some(DesugaringKind::ForLoop) => {
 475                                         self.suggest_similar_mut_method_for_for_loop(&mut err);
 476                                         err.span_label(opt_assignment_rhs_span.unwrap(), format!(
 477                                             "this iterator yields `{pointer_sigil}` {pointer_desc}s",
 478                                         ));
 479                                         None
 480                                     }
 481                                     // don't create labels for compiler-generated spans
 482                                     Some(_) => None,
 483                                     None => {
 484                                         let label = if name != kw::SelfLower {
 485                                             suggest_ampmut(
 486                                                 self.infcx.tcx,
 487                                                 local_decl,
 488                                                 opt_assignment_rhs_span,
 489                                                 *opt_ty_info,
 490                                             )
 491                                         } else {
 492                                             match local_decl.local_info.as_deref() {
 493                                                 Some(LocalInfo::User(ClearCrossCrate::Set(
 494                                                     mir::BindingForm::Var(mir::VarBindingForm {
 495                                                         opt_ty_info: None,
 496                                                         ..
 497                                                     }),
 498                                                 ))) => {
 499                                                     let (span, sugg) = suggest_ampmut_self(
 500                                                         self.infcx.tcx,
 501                                                         local_decl,
 502                                                     );
 503                                                     (true, span, sugg)
 504                                                 }
 505                                                 // explicit self (eg `self: &'a Self`)
 506                                                 _ => suggest_ampmut(
 507                                                     self.infcx.tcx,
 508                                                     local_decl,
 509                                                     opt_assignment_rhs_span,
 510                                                     *opt_ty_info,
 511                                                 ),
 512                                             }
 513                                         };
 514                                         Some(label)
 515                                     }
 516                                 }
 517                             }
 518
 519                             LocalInfo::User(ClearCrossCrate::Set(mir::BindingForm::Var(
 520                                 mir::VarBindingForm {
 521                                     binding_mode: ty::BindingMode::BindByReference(_),
 522                                     ..
 523                                 },
 524                             ))) => {
 525                                 let pattern_span = local_decl.source_info.span;
 526                                 suggest_ref_mut(self.infcx.tcx, pattern_span)
 527                                     .map(|replacement| (true, pattern_span, replacement))
 528                             }
 529
 530                             LocalInfo::User(ClearCrossCrate::Clear) => {
 531                                 bug!("saw cleared local state")
 532                             }
 533
 534                             _ => unreachable!(),
 535                         };
 536
 537                         match label {
 538                             Some((true, err_help_span, suggested_code)) => {
 539                                 let (is_trait_sig, local_trait) = self.is_error_in_trait(local);
 540                                 if !is_trait_sig {
 541                                     err.span_suggestion(
 542                                         err_help_span,
 543                                         &format!(
 544                                             "consider changing this to be a mutable {pointer_desc}"
 545                                         ),
 546                                         suggested_code,
 547                                         Applicability::MachineApplicable,
 548                                     );
 549                                 } else if let Some(x) = local_trait {
 550                                     err.span_suggestion(
 551                                         x,
 552                                         &format!(
 553                                             "consider changing that to be a mutable {pointer_desc}"
 554                                         ),
 555                                         suggested_code,
 556                                         Applicability::MachineApplicable,
 557                                     );
 558                                 }
 559                             }
 560                             Some((false, err_label_span, message)) => {
 561                                 err.span_label(
 562                                     err_label_span,
 563                                     &format!(
 564                                         "consider changing this binding's type to be: `{message}`"
 565                                     ),
 566                                 );
 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 `{ty}`",
 610                         ));
 611                     }
 612                     Some(BorrowedContentSource::OverloadedIndex(ty)) => {
 613                         err.help(&format!(
 614                             "trait `IndexMut` is required to modify indexed content, \
 615                                 but it is not implemented for `{ty}`",
 616                         ));
 617                     }
 618                     _ => (),
 619                 }
 620             }
 621
 622             _ => {
 623                 err.span_label(span, format!("cannot {ACT}", ACT = act));
 624             }
 625         }
 626
 627         self.buffer_error(err);
 628     }
 629
 630     /// User cannot make signature of a trait mutable without changing the
 631     /// trait. So we find if this error belongs to a trait and if so we move
 632     /// suggestion to the trait or disable it if it is out of scope of this crate
 633     fn is_error_in_trait(&self, local: Local) -> (bool, Option<Span>) {
 634         if self.body.local_kind(local) != LocalKind::Arg {
 635             return (false, None);
 636         }
 637         let hir_map = self.infcx.tcx.hir();
 638         let my_def = self.body.source.def_id();
 639         let my_hir = hir_map.local_def_id_to_hir_id(my_def.as_local().unwrap());
 640         let Some(td) =
 641             self.infcx.tcx.impl_of_method(my_def).and_then(|x| self.infcx.tcx.trait_id_of_impl(x))
 642         else {
 643             return (false, None);
 644         };
 645         (
 646             true,
 647             td.as_local().and_then(|tld| match hir_map.find_by_def_id(tld) {
 648                 Some(Node::Item(hir::Item {
 649                     kind: hir::ItemKind::Trait(_, _, _, _, items),
 650                     ..
 651                 })) => {
 652                     let mut f_in_trait_opt = None;
 653                     for hir::TraitItemRef { id: fi, kind: k, .. } in *items {
 654                         let hi = fi.hir_id();
 655                         if !matches!(k, hir::AssocItemKind::Fn { .. }) {
 656                             continue;
 657                         }
 658                         if hir_map.name(hi) != hir_map.name(my_hir) {
 659                             continue;
 660                         }
 661                         f_in_trait_opt = Some(hi);
 662                         break;
 663                     }
 664                     f_in_trait_opt.and_then(|f_in_trait| match hir_map.find(f_in_trait) {
 665                         Some(Node::TraitItem(hir::TraitItem {
 666                             kind:
 667                                 hir::TraitItemKind::Fn(
 668                                     hir::FnSig { decl: hir::FnDecl { inputs, .. }, .. },
 669                                     _,
 670                                 ),
 671                             ..
 672                         })) => {
 673                             let hir::Ty { span, .. } = inputs[local.index() - 1];
 674                             Some(span)
 675                         }
 676                         _ => None,
 677                     })
 678                 }
 679                 _ => None,
 680             }),
 681         )
 682     }
 683
 684     // point to span of upvar making closure call require mutable borrow
 685     fn show_mutating_upvar(
 686         &self,
 687         tcx: TyCtxt<'_>,
 688         id: hir::def_id::DefId,
 689         the_place_err: PlaceRef<'tcx>,
 690         err: &mut Diagnostic,
 691     ) {
 692         let closure_local_def_id = id.expect_local();
 693         let tables = tcx.typeck(closure_local_def_id);
 694         let closure_hir_id = tcx.hir().local_def_id_to_hir_id(closure_local_def_id);
 695         if let Some((span, closure_kind_origin)) =
 696             &tables.closure_kind_origins().get(closure_hir_id)
 697         {
 698             let reason = if let PlaceBase::Upvar(upvar_id) = closure_kind_origin.base {
 699                 let upvar = ty::place_to_string_for_capture(tcx, closure_kind_origin);
 700                 let root_hir_id = upvar_id.var_path.hir_id;
 701                 // we have an origin for this closure kind starting at this root variable so it's safe to unwrap here
 702                 let captured_places = tables.closure_min_captures[&id].get(&root_hir_id).unwrap();
 703
 704                 let origin_projection = closure_kind_origin
 705                     .projections
 706                     .iter()
 707                     .map(|proj| proj.kind)
 708                     .collect::<Vec<_>>();
 709                 let mut capture_reason = String::new();
 710                 for captured_place in captured_places {
 711                     let captured_place_kinds = captured_place
 712                         .place
 713                         .projections
 714                         .iter()
 715                         .map(|proj| proj.kind)
 716                         .collect::<Vec<_>>();
 717                     if rustc_middle::ty::is_ancestor_or_same_capture(
 718                         &captured_place_kinds,
 719                         &origin_projection,
 720                     ) {
 721                         match captured_place.info.capture_kind {
 722                             ty::UpvarCapture::ByRef(
 723                                 ty::BorrowKind::MutBorrow | ty::BorrowKind::UniqueImmBorrow,
 724                             ) => {
 725                                 capture_reason = format!("mutable borrow of `{upvar}`");
 726                             }
 727                             ty::UpvarCapture::ByValue => {
 728                                 capture_reason = format!("possible mutation of `{upvar}`");
 729                             }
 730                             _ => bug!("upvar `{upvar}` borrowed, but not mutably"),
 731                         }
 732                         break;
 733                     }
 734                 }
 735                 if capture_reason.is_empty() {
 736                     bug!("upvar `{upvar}` borrowed, but cannot find reason");
 737                 }
 738                 capture_reason
 739             } else {
 740                 bug!("not an upvar")
 741             };
 742             err.span_label(
 743                 *span,
 744                 format!(
 745                     "calling `{}` requires mutable binding due to {}",
 746                     self.describe_place(the_place_err).unwrap(),
 747                     reason
 748                 ),
 749             );
 750         }
 751     }
 752
 753     // Attempt to search similar mutable associated items for suggestion.
 754     // In the future, attempt in all path but initially for RHS of for_loop
 755     fn suggest_similar_mut_method_for_for_loop(&self, err: &mut Diagnostic) {
 756         use hir::{
 757             BodyId, Expr,
 758             ExprKind::{Block, Call, DropTemps, Match, MethodCall},
 759             HirId, ImplItem, ImplItemKind, Item, ItemKind,
 760         };
 761
 762         fn maybe_body_id_of_fn(hir_map: Map<'_>, id: HirId) -> Option<BodyId> {
 763             match hir_map.find(id) {
 764                 Some(Node::Item(Item { kind: ItemKind::Fn(_, _, body_id), .. }))
 765                 | Some(Node::ImplItem(ImplItem { kind: ImplItemKind::Fn(_, body_id), .. })) => {
 766                     Some(*body_id)
 767                 }
 768                 _ => None,
 769             }
 770         }
 771         let hir_map = self.infcx.tcx.hir();
 772         let mir_body_hir_id = self.mir_hir_id();
 773         if let Some(fn_body_id) = maybe_body_id_of_fn(hir_map, mir_body_hir_id) {
 774             if let Block(
 775                 hir::Block {
 776                     expr:
 777                         Some(Expr {
 778                             kind:
 779                                 DropTemps(Expr {
 780                                     kind:
 781                                         Match(
 782                                             Expr {
 783                                                 kind:
 784                                                     Call(
 785                                                         _,
 786                                                         [
 787                                                             Expr {
 788                                                                 kind:
 789                                                                     MethodCall(
 790                                                                         path_segment,
 791                                                                         _args,
 792                                                                         span,
 793                                                                     ),
 794                                                                 hir_id,
 795                                                                 ..
 796                                                             },
 797                                                             ..,
 798                                                         ],
 799                                                     ),
 800                                                 ..
 801                                             },
 802                                             ..,
 803                                         ),
 804                                     ..
 805                                 }),
 806                             ..
 807                         }),
 808                     ..
 809                 },
 810                 _,
 811             ) = hir_map.body(fn_body_id).value.kind
 812             {
 813                 let opt_suggestions = path_segment
 814                     .hir_id
 815                     .map(|path_hir_id| self.infcx.tcx.typeck(path_hir_id.owner))
 816                     .and_then(|typeck| typeck.type_dependent_def_id(*hir_id))
 817                     .and_then(|def_id| self.infcx.tcx.impl_of_method(def_id))
 818                     .map(|def_id| self.infcx.tcx.associated_items(def_id))
 819                     .map(|assoc_items| {
 820                         assoc_items
 821                             .in_definition_order()
 822                             .map(|assoc_item_def| assoc_item_def.ident(self.infcx.tcx))
 823                             .filter(|&ident| {
 824                                 let original_method_ident = path_segment.ident;
 825                                 original_method_ident != ident
 826                                     && ident
 827                                         .as_str()
 828                                         .starts_with(&original_method_ident.name.to_string())
 829                             })
 830                             .map(|ident| format!("{ident}()"))
 831                             .peekable()
 832                     });
 833
 834                 if let Some(mut suggestions) = opt_suggestions
 835                     && suggestions.peek().is_some()
 836                 {
 837                     err.span_suggestions(
 838                         *span,
 839                         "use mutable method",
 840                         suggestions,
 841                         Applicability::MaybeIncorrect,
 842                     );
 843                 }
 844             }
 845         };
 846     }
 847
 848     /// Targeted error when encountering an `FnMut` closure where an `Fn` closure was expected.
 849     fn expected_fn_found_fn_mut_call(&self, err: &mut Diagnostic, sp: Span, act: &str) {
 850         err.span_label(sp, format!("cannot {act}"));
 851
 852         let hir = self.infcx.tcx.hir();
 853         let closure_id = self.mir_hir_id();
 854         let fn_call_id = hir.get_parent_node(closure_id);
 855         let node = hir.get(fn_call_id);
 856         let item_id = hir.enclosing_body_owner(fn_call_id);
 857         let mut look_at_return = true;
 858         // If we can detect the expression to be an `fn` call where the closure was an argument,
 859         // we point at the `fn` definition argument...
 860         if let hir::Node::Expr(hir::Expr { kind: hir::ExprKind::Call(func, args), .. }) = node {
 861             let arg_pos = args
 862                 .iter()
 863                 .enumerate()
 864                 .filter(|(_, arg)| arg.hir_id == closure_id)
 865                 .map(|(pos, _)| pos)
 866                 .next();
 867             let def_id = hir.local_def_id(item_id);
 868             let tables = self.infcx.tcx.typeck(def_id);
 869             if let Some(ty::FnDef(def_id, _)) =
 870                 tables.node_type_opt(func.hir_id).as_ref().map(|ty| ty.kind())
 871             {
 872                 let arg = match hir.get_if_local(*def_id) {
 873                     Some(
 874                         hir::Node::Item(hir::Item {
 875                             ident, kind: hir::ItemKind::Fn(sig, ..), ..
 876                         })
 877                         | hir::Node::TraitItem(hir::TraitItem {
 878                             ident,
 879                             kind: hir::TraitItemKind::Fn(sig, _),
 880                             ..
 881                         })
 882                         | hir::Node::ImplItem(hir::ImplItem {
 883                             ident,
 884                             kind: hir::ImplItemKind::Fn(sig, _),
 885                             ..
 886                         }),
 887                     ) => Some(
 888                         arg_pos
 889                             .and_then(|pos| {
 890                                 sig.decl.inputs.get(
 891                                     pos + if sig.decl.implicit_self.has_implicit_self() {
 892                                         1
 893                                     } else {
 894                                         0
 895                                     },
 896                                 )
 897                             })
 898                             .map(|arg| arg.span)
 899                             .unwrap_or(ident.span),
 900                     ),
 901                     _ => None,
 902                 };
 903                 if let Some(span) = arg {
 904                     err.span_label(span, "change this to accept `FnMut` instead of `Fn`");
 905                     err.span_label(func.span, "expects `Fn` instead of `FnMut`");
 906                     err.span_label(self.body.span, "in this closure");
 907                     look_at_return = false;
 908                 }
 909             }
 910         }
 911
 912         if look_at_return && hir.get_return_block(closure_id).is_some() {
 913             // ...otherwise we are probably in the tail expression of the function, point at the
 914             // return type.
 915             match hir.get_by_def_id(hir.get_parent_item(fn_call_id)) {
 916                 hir::Node::Item(hir::Item { ident, kind: hir::ItemKind::Fn(sig, ..), .. })
 917                 | hir::Node::TraitItem(hir::TraitItem {
 918                     ident,
 919                     kind: hir::TraitItemKind::Fn(sig, _),
 920                     ..
 921                 })
 922                 | hir::Node::ImplItem(hir::ImplItem {
 923                     ident,
 924                     kind: hir::ImplItemKind::Fn(sig, _),
 925                     ..
 926                 }) => {
 927                     err.span_label(ident.span, "");
 928                     err.span_label(
 929                         sig.decl.output.span(),
 930                         "change this to return `FnMut` instead of `Fn`",
 931                     );
 932                     err.span_label(self.body.span, "in this closure");
 933                 }
 934                 _ => {}
 935             }
 936         }
 937     }
 938 }
 939
 940 fn mut_borrow_of_mutable_ref(local_decl: &LocalDecl<'_>, local_name: Option<Symbol>) -> bool {
 941     debug!("local_info: {:?}, ty.kind(): {:?}", local_decl.local_info, local_decl.ty.kind());
 942
 943     match local_decl.local_info.as_deref() {
 944         // Check if mutably borrowing a mutable reference.
 945         Some(LocalInfo::User(ClearCrossCrate::Set(mir::BindingForm::Var(
 946             mir::VarBindingForm {
 947                 binding_mode: ty::BindingMode::BindByValue(Mutability::Not), ..
 948             },
 949         )))) => matches!(local_decl.ty.kind(), ty::Ref(_, _, hir::Mutability::Mut)),
 950         Some(LocalInfo::User(ClearCrossCrate::Set(mir::BindingForm::ImplicitSelf(kind)))) => {
 951             // Check if the user variable is a `&mut self` and we can therefore
 952             // suggest removing the `&mut`.
 953             //
 954             // Deliberately fall into this case for all implicit self types,
 955             // so that we don't fall in to the next case with them.
 956             *kind == mir::ImplicitSelfKind::MutRef
 957         }
 958         _ if Some(kw::SelfLower) == local_name => {
 959             // Otherwise, check if the name is the `self` keyword - in which case
 960             // we have an explicit self. Do the same thing in this case and check
 961             // for a `self: &mut Self` to suggest removing the `&mut`.
 962             matches!(local_decl.ty.kind(), ty::Ref(_, _, hir::Mutability::Mut))
 963         }
 964         _ => false,
 965     }
 966 }
 967
 968 fn suggest_ampmut_self<'tcx>(
 969     tcx: TyCtxt<'tcx>,
 970     local_decl: &mir::LocalDecl<'tcx>,
 971 ) -> (Span, String) {
 972     let sp = local_decl.source_info.span;
 973     (
 974         sp,
 975         match tcx.sess.source_map().span_to_snippet(sp) {
 976             Ok(snippet) => {
 977                 let lt_pos = snippet.find('\'');
 978                 if let Some(lt_pos) = lt_pos {
 979                     format!("&{}mut self", &snippet[lt_pos..snippet.len() - 4])
 980                 } else {
 981                     "&mut self".to_string()
 982                 }
 983             }
 984             _ => "&mut self".to_string(),
 985         },
 986     )
 987 }
 988
 989 // When we want to suggest a user change a local variable to be a `&mut`, there
 990 // are three potential "obvious" things to highlight:
 991 //
 992 // let ident [: Type] [= RightHandSideExpression];
 993 //     ^^^^^    ^^^^     ^^^^^^^^^^^^^^^^^^^^^^^
 994 //     (1.)     (2.)              (3.)
 995 //
 996 // We can always fallback on highlighting the first. But chances are good that
 997 // the user experience will be better if we highlight one of the others if possible;
 998 // for example, if the RHS is present and the Type is not, then the type is going to
 999 // be inferred *from* the RHS, which means we should highlight that (and suggest
1000 // that they borrow the RHS mutably).
1001 //
1002 // This implementation attempts to emulate AST-borrowck prioritization
1003 // by trying (3.), then (2.) and finally falling back on (1.).
1004 fn suggest_ampmut<'tcx>(
1005     tcx: TyCtxt<'tcx>,
1006     local_decl: &mir::LocalDecl<'tcx>,
1007     opt_assignment_rhs_span: Option<Span>,
1008     opt_ty_info: Option<Span>,
1009 ) -> (bool, Span, String) {
1010     if let Some(assignment_rhs_span) = opt_assignment_rhs_span
1011         && let Ok(src) = tcx.sess.source_map().span_to_snippet(assignment_rhs_span)
1012     {
1013         let is_mutbl = |ty: &str| -> bool {
1014             if let Some(rest) = ty.strip_prefix("mut") {
1015                 match rest.chars().next() {
1016                     // e.g. `&mut x`
1017                     Some(c) if c.is_whitespace() => true,
1018                     // e.g. `&mut(x)`
1019                     Some('(') => true,
1020                     // e.g. `&mut{x}`
1021                     Some('{') => true,
1022                     // e.g. `&mutablevar`
1023                     _ => false,
1024                 }
1025             } else {
1026                 false
1027             }
1028         };
1029         if let (true, Some(ws_pos)) = (src.starts_with("&'"), src.find(char::is_whitespace)) {
1030             let lt_name = &src[1..ws_pos];
1031             let ty = src[ws_pos..].trim_start();
1032             if !is_mutbl(ty) {
1033                 return (true, assignment_rhs_span, format!("&{lt_name} mut {ty}"));
1034             }
1035         } else if let Some(stripped) = src.strip_prefix('&') {
1036             let stripped = stripped.trim_start();
1037             if !is_mutbl(stripped) {
1038                 return (true, assignment_rhs_span, format!("&mut {stripped}"));
1039             }
1040         }
1041     }
1042
1043     let (suggestability, highlight_span) = match opt_ty_info {
1044         // if this is a variable binding with an explicit type,
1045         // try to highlight that for the suggestion.
1046         Some(ty_span) => (true, ty_span),
1047
1048         // otherwise, just highlight the span associated with
1049         // the (MIR) LocalDecl.
1050         None => (false, local_decl.source_info.span),
1051     };
1052
1053     if let Ok(src) = tcx.sess.source_map().span_to_snippet(highlight_span)
1054         && let (true, Some(ws_pos)) = (src.starts_with("&'"), src.find(char::is_whitespace))
1055     {
1056         let lt_name = &src[1..ws_pos];
1057         let ty = &src[ws_pos..];
1058         return (true, highlight_span, format!("&{} mut{}", lt_name, ty));
1059     }
1060
1061     let ty_mut = local_decl.ty.builtin_deref(true).unwrap();
1062     assert_eq!(ty_mut.mutbl, hir::Mutability::Not);
1063     (
1064         suggestability,
1065         highlight_span,
1066         if local_decl.ty.is_region_ptr() {
1067             format!("&mut {}", ty_mut.ty)
1068         } else {
1069             format!("*mut {}", ty_mut.ty)
1070         },
1071     )
1072 }
1073
1074 fn is_closure_or_generator(ty: Ty<'_>) -> bool {
1075     ty.is_closure() || ty.is_generator()
1076 }
1077
1078 /// Given a field that needs to be mutable, returns a span where the " mut " could go.
1079 /// This function expects the local to be a reference to a struct in order to produce a span.
1080 ///
1081 /// ```text
1082 /// LL |     s: &'a   String
1083 ///    |           ^^^ returns a span taking up the space here
1084 /// ```
1085 fn get_mut_span_in_struct_field<'tcx>(
1086     tcx: TyCtxt<'tcx>,
1087     ty: Ty<'tcx>,
1088     field: mir::Field,
1089 ) -> Option<Span> {
1090     // Expect our local to be a reference to a struct of some kind.
1091     if let ty::Ref(_, ty, _) = ty.kind()
1092         && let ty::Adt(def, _) = ty.kind()
1093         && let field = def.all_fields().nth(field.index())?
1094         // Use the HIR types to construct the diagnostic message.
1095         && let node = tcx.hir().find_by_def_id(field.did.as_local()?)?
1096         // Now we're dealing with the actual struct that we're going to suggest a change to,
1097         // we can expect a field that is an immutable reference to a type.
1098         && let hir::Node::Field(field) = node
1099         && let hir::TyKind::Rptr(lt, hir::MutTy { mutbl: hir::Mutability::Not, ty }) = field.ty.kind
1100     {
1101         return Some(lt.span.between(ty.span));
1102     }
1103
1104     None
1105 }
1106
1107 /// If possible, suggest replacing `ref` with `ref mut`.
1108 fn suggest_ref_mut(tcx: TyCtxt<'_>, binding_span: Span) -> Option<String> {
1109     let hi_src = tcx.sess.source_map().span_to_snippet(binding_span).ok()?;
1110     if hi_src.starts_with("ref") && hi_src["ref".len()..].starts_with(rustc_lexer::is_whitespace) {
1111         let replacement = format!("ref mut{}", &hi_src["ref".len()..]);
1112         Some(replacement)
1113     } else {
1114         None
1115     }
1116 }