1 // Copyright 2012 The Rust Project Developers. See the COPYRIGHT
2 // file at the top-level directory of this distribution and at
3 // http://rust-lang.org/COPYRIGHT.
5 // Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
6 // http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
7 // <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
8 // option. This file may not be copied, modified, or distributed
9 // except according to those terms.
12 use rustc::infer::InferOk;
13 use rustc::traits::ObligationCause;
16 use syntax::util::parser::PREC_POSTFIX;
19 use rustc::hir::def::Def;
21 use rustc::hir::{Item, ItemKind, print};
22 use rustc::ty::{self, Ty, AssociatedItem};
23 use rustc::ty::adjustment::AllowTwoPhase;
24 use errors::{Applicability, DiagnosticBuilder, SourceMapper};
26 use super::method::probe;
28 impl<'a, 'gcx, 'tcx> FnCtxt<'a, 'gcx, 'tcx> {
29 // Requires that the two types unify, and prints an error message if
31 pub fn demand_suptype(&self, sp: Span, expected: Ty<'tcx>, actual: Ty<'tcx>) {
32 self.demand_suptype_diag(sp, expected, actual).map(|mut e| e.emit());
35 pub fn demand_suptype_diag(&self,
38 actual: Ty<'tcx>) -> Option<DiagnosticBuilder<'tcx>> {
39 let cause = &self.misc(sp);
40 match self.at(cause, self.param_env).sup(expected, actual) {
41 Ok(InferOk { obligations, value: () }) => {
42 self.register_predicates(obligations);
46 Some(self.report_mismatched_types(&cause, expected, actual, e))
51 pub fn demand_eqtype(&self, sp: Span, expected: Ty<'tcx>, actual: Ty<'tcx>) {
52 if let Some(mut err) = self.demand_eqtype_diag(sp, expected, actual) {
57 pub fn demand_eqtype_diag(&self,
60 actual: Ty<'tcx>) -> Option<DiagnosticBuilder<'tcx>> {
61 self.demand_eqtype_with_origin(&self.misc(sp), expected, actual)
64 pub fn demand_eqtype_with_origin(&self,
65 cause: &ObligationCause<'tcx>,
67 actual: Ty<'tcx>) -> Option<DiagnosticBuilder<'tcx>> {
68 match self.at(cause, self.param_env).eq(expected, actual) {
69 Ok(InferOk { obligations, value: () }) => {
70 self.register_predicates(obligations);
74 Some(self.report_mismatched_types(cause, expected, actual, e))
79 pub fn demand_coerce(&self,
83 allow_two_phase: AllowTwoPhase)
85 let (ty, err) = self.demand_coerce_diag(expr, checked_ty, expected, allow_two_phase);
86 if let Some(mut err) = err {
92 // Checks that the type of `expr` can be coerced to `expected`.
94 // NB: This code relies on `self.diverges` to be accurate. In
95 // particular, assignments to `!` will be permitted if the
96 // diverges flag is currently "always".
97 pub fn demand_coerce_diag(&self,
101 allow_two_phase: AllowTwoPhase)
102 -> (Ty<'tcx>, Option<DiagnosticBuilder<'tcx>>) {
103 let expected = self.resolve_type_vars_with_obligations(expected);
105 let e = match self.try_coerce(expr, checked_ty, expected, allow_two_phase) {
106 Ok(ty) => return (ty, None),
110 let cause = self.misc(expr.span);
111 let expr_ty = self.resolve_type_vars_with_obligations(checked_ty);
112 let mut err = self.report_mismatched_types(&cause, expected, expr_ty, e);
114 // If the expected type is an enum with any variants whose sole
115 // field is of the found type, suggest such variants. See Issue
117 if let ty::Adt(expected_adt, substs) = expected.sty {
118 let compatible_variants = expected_adt.variants
120 .filter(|variant| variant.fields.len() == 1)
121 .filter_map(|variant| {
122 let sole_field = &variant.fields[0];
123 let sole_field_ty = sole_field.ty(self.tcx, substs);
124 if self.can_coerce(expr_ty, sole_field_ty) {
125 let variant_path = self.tcx.item_path_str(variant.did);
126 Some(variant_path.trim_left_matches("std::prelude::v1::").to_string())
130 }).collect::<Vec<_>>();
132 if !compatible_variants.is_empty() {
133 let expr_text = print::to_string(print::NO_ANN, |s| s.print_expr(expr));
134 let suggestions = compatible_variants.iter()
135 .map(|v| format!("{}({})", v, expr_text)).collect::<Vec<_>>();
136 err.span_suggestions_with_applicability(
138 "try using a variant of the expected type",
140 Applicability::MaybeIncorrect,
145 self.suggest_ref_or_into(&mut err, expr, expected, expr_ty);
147 (expected, Some(err))
150 pub fn get_conversion_methods(&self, span: Span, expected: Ty<'tcx>, checked_ty: Ty<'tcx>)
151 -> Vec<AssociatedItem> {
152 let mut methods = self.probe_for_return_type(span,
153 probe::Mode::MethodCall,
158 self.has_no_input_arg(m) &&
159 self.tcx.get_attrs(m.def_id).iter()
160 // This special internal attribute is used to whitelist
161 // "identity-like" conversion methods to be suggested here.
163 // FIXME (#46459 and #46460): ideally
164 // `std::convert::Into::into` and `std::borrow:ToOwned` would
165 // also be `#[rustc_conversion_suggestion]`, if not for
166 // method-probing false-positives and -negatives (respectively).
168 // FIXME? Other potential candidate methods: `as_ref` and
170 .find(|a| a.check_name("rustc_conversion_suggestion")).is_some()
176 // This function checks if the method isn't static and takes other arguments than `self`.
177 fn has_no_input_arg(&self, method: &AssociatedItem) -> bool {
179 Def::Method(def_id) => {
180 self.tcx.fn_sig(def_id).inputs().skip_binder().len() == 1
186 /// Identify some cases where `as_ref()` would be appropriate and suggest it.
188 /// Given the following code:
191 /// fn takes_ref(_: &Foo) {}
192 /// let ref opt = Some(Foo);
194 /// opt.map(|arg| takes_ref(arg));
196 /// Suggest using `opt.as_ref().map(|arg| takes_ref(arg));` instead.
198 /// It only checks for `Option` and `Result` and won't work with
200 /// opt.map(|arg| { takes_ref(arg) });
202 fn can_use_as_ref(&self, expr: &hir::Expr) -> Option<(Span, &'static str, String)> {
203 if let hir::ExprKind::Path(hir::QPath::Resolved(_, ref path)) = expr.node {
204 if let hir::def::Def::Local(id) = path.def {
205 let parent = self.tcx.hir.get_parent_node(id);
206 if let Some(Node::Expr(hir::Expr {
208 node: hir::ExprKind::Closure(_, decl, ..),
210 })) = self.tcx.hir.find(parent) {
211 let parent = self.tcx.hir.get_parent_node(*id);
212 if let (Some(Node::Expr(hir::Expr {
213 node: hir::ExprKind::MethodCall(path, span, expr),
215 })), 1) = (self.tcx.hir.find(parent), decl.inputs.len()) {
216 let self_ty = self.tables.borrow().node_id_to_type(expr[0].hir_id);
217 let self_ty = format!("{:?}", self_ty);
218 let name = path.ident.as_str();
219 let is_as_ref_able = (
220 self_ty.starts_with("&std::option::Option") ||
221 self_ty.starts_with("&std::result::Result") ||
222 self_ty.starts_with("std::option::Option") ||
223 self_ty.starts_with("std::result::Result")
224 ) && (name == "map" || name == "and_then");
226 return Some((span.shrink_to_lo(),
227 "consider using `as_ref` instead",
228 "as_ref().".into()));
237 /// This function is used to determine potential "simple" improvements or users' errors and
238 /// provide them useful help. For example:
241 /// fn some_fn(s: &str) {}
243 /// let x = "hey!".to_owned();
244 /// some_fn(x); // error
247 /// No need to find every potential function which could make a coercion to transform a
248 /// `String` into a `&str` since a `&` would do the trick!
250 /// In addition of this check, it also checks between references mutability state. If the
251 /// expected is mutable but the provided isn't, maybe we could just say "Hey, try with
253 pub fn check_ref(&self,
255 checked_ty: Ty<'tcx>,
257 -> Option<(Span, &'static str, String)> {
258 let cm = self.sess().source_map();
259 // Use the callsite's span if this is a macro call. #41858
260 let sp = cm.call_span_if_macro(expr.span);
261 if !cm.span_to_filename(sp).is_real() {
265 match (&expected.sty, &checked_ty.sty) {
266 (&ty::Ref(_, exp, _), &ty::Ref(_, check, _)) => match (&exp.sty, &check.sty) {
267 (&ty::Str, &ty::Array(arr, _)) |
268 (&ty::Str, &ty::Slice(arr)) if arr == self.tcx.types.u8 => {
269 if let hir::ExprKind::Lit(_) = expr.node {
270 if let Ok(src) = cm.span_to_snippet(sp) {
271 if src.starts_with("b\"") {
273 "consider removing the leading `b`",
274 src[1..].to_string()));
279 (&ty::Array(arr, _), &ty::Str) |
280 (&ty::Slice(arr), &ty::Str) if arr == self.tcx.types.u8 => {
281 if let hir::ExprKind::Lit(_) = expr.node {
282 if let Ok(src) = cm.span_to_snippet(sp) {
283 if src.starts_with("\"") {
285 "consider adding a leading `b`",
286 format!("b{}", src)));
293 (&ty::Ref(_, _, mutability), _) => {
294 // Check if it can work when put into a ref. For example:
297 // fn bar(x: &mut i32) {}
300 // bar(&x); // error, expected &mut
302 let ref_ty = match mutability {
303 hir::Mutability::MutMutable => self.tcx.mk_mut_ref(
304 self.tcx.mk_region(ty::ReStatic),
306 hir::Mutability::MutImmutable => self.tcx.mk_imm_ref(
307 self.tcx.mk_region(ty::ReStatic),
310 if self.can_coerce(ref_ty, expected) {
311 if let Ok(src) = cm.span_to_snippet(sp) {
312 let needs_parens = match expr.node {
313 // parenthesize if needed (Issue #46756)
314 hir::ExprKind::Cast(_, _) |
315 hir::ExprKind::Binary(_, _, _) => true,
316 // parenthesize borrows of range literals (Issue #54505)
317 _ if self.is_range_literal(expr) => true,
320 let sugg_expr = if needs_parens { format!("({})", src) } else { src };
322 if let Some(sugg) = self.can_use_as_ref(expr) {
325 return Some(match mutability {
326 hir::Mutability::MutMutable => {
327 (sp, "consider mutably borrowing here", format!("&mut {}",
330 hir::Mutability::MutImmutable => {
331 (sp, "consider borrowing here", format!("&{}", sugg_expr))
337 (_, &ty::Ref(_, checked, _)) => {
338 // We have `&T`, check if what was expected was `T`. If so,
339 // we may want to suggest adding a `*`, or removing
342 // (But, also check check the `expn_info()` to see if this is
343 // a macro; if so, it's hard to extract the text and make a good
344 // suggestion, so don't bother.)
345 if self.infcx.can_sub(self.param_env, checked, &expected).is_ok() &&
346 sp.ctxt().outer().expn_info().is_none() {
349 hir::ExprKind::AddrOf(_, ref expr) => {
350 if !cm.span_to_filename(expr.span).is_real() {
353 if let Ok(code) = cm.span_to_snippet(expr.span) {
354 return Some((sp, "consider removing the borrow", code));
358 // Maybe add `*`? Only if `T: Copy`.
360 if !self.infcx.type_moves_by_default(self.param_env,
363 // do not suggest if the span comes from a macro (#52783)
365 true) = (cm.span_to_snippet(sp), sp == expr.span) {
368 "consider dereferencing the borrow",
369 format!("*{}", code),
382 // This function checks if the specified expression is a built-in range literal
383 // (See: librustc/hir/lowering.rs::LoweringContext::lower_expr() )
384 fn is_range_literal(&self, expr: &hir::Expr) -> bool {
385 use hir::{Path, QPath, ExprKind, TyKind};
387 // we support `::std::ops::Range` and `::std::core::Range` prefixes
388 // (via split on "|")
389 let ops_path = ["{{root}}", "std|core", "ops"];
391 let is_range_path = |path: &Path| {
392 let ident_names: Vec<_> = path.segments
394 .map(|seg| seg.ident.as_str())
397 if let Some((last, preceding)) = ident_names.split_last() {
398 last.starts_with("Range") &&
399 preceding.len() == 3 &&
401 .zip(ops_path.iter())
402 .all(|(seg, match_seg)| {
405 .any(|ref spl_seg| seg == spl_seg)
412 let is_range_struct_snippet = |span: &Span| {
413 // Tell if expression span snippet looks like an explicit
414 // Range struct or new() call. This is to allow rejecting
415 // Ranges constructed with non-literals.
416 let source_map = self.tcx.sess.source_map();
417 let end_point = source_map.end_point(*span);
419 if let Ok(end_string) = source_map.span_to_snippet(end_point) {
420 end_string.ends_with("}") || end_string.ends_with(")")
428 // all built-in range literals but `..=` and `..`
429 // desugar to Structs, `..` desugars to its struct path
430 ExprKind::Struct(QPath::Resolved(None, ref path), _, _) |
431 ExprKind::Path(QPath::Resolved(None, ref path)) => {
432 return is_range_path(&path) && !is_range_struct_snippet(&expr.span);
435 // `..=` desugars into RangeInclusive::new(...)
436 ExprKind::Call(ref func, _) => {
437 if let ExprKind::Path(QPath::TypeRelative(ref ty, ref segment)) = func.node {
438 if let TyKind::Path(QPath::Resolved(None, ref path)) = ty.node {
439 let calls_new = segment.ident.as_str() == "new";
441 return is_range_path(&path) && calls_new &&
442 !is_range_struct_snippet(&expr.span);
453 pub fn check_for_cast(&self,
454 err: &mut DiagnosticBuilder<'tcx>,
456 checked_ty: Ty<'tcx>,
457 expected_ty: Ty<'tcx>)
459 let parent_id = self.tcx.hir.get_parent_node(expr.id);
460 if let Some(parent) = self.tcx.hir.find(parent_id) {
461 // Shouldn't suggest `.into()` on `const`s.
462 if let Node::Item(Item { node: ItemKind::Const(_, _), .. }) = parent {
463 // FIXME(estebank): modify once we decide to suggest `as` casts
468 let will_truncate = "will truncate the source value";
469 let depending_on_isize = "will truncate or zero-extend depending on the bit width of \
471 let depending_on_usize = "will truncate or zero-extend depending on the bit width of \
473 let will_sign_extend = "will sign-extend the source value";
474 let will_zero_extend = "will zero-extend the source value";
476 // If casting this expression to a given numeric type would be appropriate in case of a type
479 // We want to minimize the amount of casting operations that are suggested, as it can be a
480 // lossy operation with potentially bad side effects, so we only suggest when encountering
481 // an expression that indicates that the original type couldn't be directly changed.
483 // For now, don't suggest casting with `as`.
484 let can_cast = false;
486 let needs_paren = expr.precedence().order() < (PREC_POSTFIX as i8);
488 if let Ok(src) = self.tcx.sess.source_map().span_to_snippet(expr.span) {
489 let msg = format!("you can cast an `{}` to `{}`", checked_ty, expected_ty);
490 let cast_suggestion = format!("{}{}{} as {}",
491 if needs_paren { "(" } else { "" },
493 if needs_paren { ")" } else { "" },
495 let into_suggestion = format!(
497 if needs_paren { "(" } else { "" },
499 if needs_paren { ")" } else { "" },
501 let literal_is_ty_suffixed = |expr: &hir::Expr| {
502 if let hir::ExprKind::Lit(lit) = &expr.node {
503 lit.node.is_suffixed()
509 let into_sugg = into_suggestion.clone();
510 let suggest_to_change_suffix_or_into = |err: &mut DiagnosticBuilder,
511 note: Option<&str>| {
512 let suggest_msg = if literal_is_ty_suffixed(expr) {
514 "change the type of the numeric literal from `{}` to `{}`",
520 Some(note) => format!("{}, which {}", msg, note),
521 _ => format!("{} in a lossless way", msg),
525 let suffix_suggestion = format!(
527 if needs_paren { "(" } else { "" },
528 src.trim_right_matches(&checked_ty.to_string()),
530 if needs_paren { ")" } else { "" },
533 err.span_suggestion_with_applicability(
536 if literal_is_ty_suffixed(expr) {
541 Applicability::MachineApplicable,
545 match (&expected_ty.sty, &checked_ty.sty) {
546 (&ty::Int(ref exp), &ty::Int(ref found)) => {
547 match (found.bit_width(), exp.bit_width()) {
548 (Some(found), Some(exp)) if found > exp => {
550 err.span_suggestion_with_applicability(
552 &format!("{}, which {}", msg, will_truncate),
554 Applicability::MaybeIncorrect // lossy conversion
558 (None, _) | (_, None) => {
560 err.span_suggestion_with_applicability(
562 &format!("{}, which {}", msg, depending_on_isize),
564 Applicability::MaybeIncorrect // lossy conversion
569 suggest_to_change_suffix_or_into(
571 Some(will_sign_extend),
577 (&ty::Uint(ref exp), &ty::Uint(ref found)) => {
578 match (found.bit_width(), exp.bit_width()) {
579 (Some(found), Some(exp)) if found > exp => {
581 err.span_suggestion_with_applicability(
583 &format!("{}, which {}", msg, will_truncate),
585 Applicability::MaybeIncorrect // lossy conversion
589 (None, _) | (_, None) => {
591 err.span_suggestion_with_applicability(
593 &format!("{}, which {}", msg, depending_on_usize),
595 Applicability::MaybeIncorrect // lossy conversion
600 suggest_to_change_suffix_or_into(
602 Some(will_zero_extend),
608 (&ty::Int(ref exp), &ty::Uint(ref found)) => {
610 match (found.bit_width(), exp.bit_width()) {
611 (Some(found), Some(exp)) if found > exp - 1 => {
612 err.span_suggestion_with_applicability(
614 &format!("{}, which {}", msg, will_truncate),
616 Applicability::MaybeIncorrect // lossy conversion
620 err.span_suggestion_with_applicability(
622 &format!("{}, which {}", msg, will_truncate),
624 Applicability::MaybeIncorrect // lossy conversion
628 err.span_suggestion_with_applicability(
630 &format!("{}, which {}", msg, depending_on_isize),
632 Applicability::MaybeIncorrect // lossy conversion
636 err.span_suggestion_with_applicability(
638 &format!("{}, which {}", msg, depending_on_usize),
640 Applicability::MaybeIncorrect // lossy conversion
644 err.span_suggestion_with_applicability(
646 &format!("{}, which {}", msg, will_zero_extend),
648 Applicability::MachineApplicable
655 (&ty::Uint(ref exp), &ty::Int(ref found)) => {
657 match (found.bit_width(), exp.bit_width()) {
658 (Some(found), Some(exp)) if found - 1 > exp => {
659 err.span_suggestion_with_applicability(
661 &format!("{}, which {}", msg, will_truncate),
663 Applicability::MaybeIncorrect // lossy conversion
667 err.span_suggestion_with_applicability(
669 &format!("{}, which {}", msg, will_sign_extend),
671 Applicability::MachineApplicable // lossy conversion
675 err.span_suggestion_with_applicability(
677 &format!("{}, which {}", msg, depending_on_usize),
679 Applicability::MaybeIncorrect // lossy conversion
683 err.span_suggestion_with_applicability(
685 &format!("{}, which {}", msg, depending_on_isize),
687 Applicability::MaybeIncorrect // lossy conversion
691 err.span_suggestion_with_applicability(
693 &format!("{}, which {}", msg, will_sign_extend),
695 Applicability::MachineApplicable
702 (&ty::Float(ref exp), &ty::Float(ref found)) => {
703 if found.bit_width() < exp.bit_width() {
704 suggest_to_change_suffix_or_into(
709 err.span_suggestion_with_applicability(
711 &format!("{}, producing the closest possible value", msg),
713 Applicability::MaybeIncorrect // lossy conversion
718 (&ty::Uint(_), &ty::Float(_)) | (&ty::Int(_), &ty::Float(_)) => {
720 err.span_suggestion_with_applicability(
722 &format!("{}, rounding the float towards zero", msg),
724 Applicability::MaybeIncorrect // lossy conversion
726 err.warn("casting here will cause undefined behavior if the rounded value \
727 cannot be represented by the target integer type, including \
728 `Inf` and `NaN` (this is a bug and will be fixed)");
732 (&ty::Float(ref exp), &ty::Uint(ref found)) => {
733 // if `found` is `None` (meaning found is `usize`), don't suggest `.into()`
734 if exp.bit_width() > found.bit_width().unwrap_or(256) {
735 err.span_suggestion_with_applicability(
737 &format!("{}, producing the floating point representation of the \
741 Applicability::MachineApplicable
744 err.span_suggestion_with_applicability(expr.span,
745 &format!("{}, producing the floating point representation of the \
746 integer, rounded if necessary",
749 Applicability::MaybeIncorrect // lossy conversion
754 (&ty::Float(ref exp), &ty::Int(ref found)) => {
755 // if `found` is `None` (meaning found is `isize`), don't suggest `.into()`
756 if exp.bit_width() > found.bit_width().unwrap_or(256) {
757 err.span_suggestion_with_applicability(
759 &format!("{}, producing the floating point representation of the \
763 Applicability::MachineApplicable
766 err.span_suggestion_with_applicability(
768 &format!("{}, producing the floating point representation of the \
769 integer, rounded if necessary",
772 Applicability::MaybeIncorrect // lossy conversion