| //! Validation of patterns/matches. |
| |
| mod _match; |
| mod check_match; |
| mod const_to_pat; |
| |
| pub(crate) use self::check_match::check_match; |
| |
| use crate::hair::util::UserAnnotatedTyHelpers; |
| use crate::hair::constant::*; |
| |
| use rustc::mir::{Field, BorrowKind, Mutability}; |
| use rustc::mir::{UserTypeProjection}; |
| use rustc::mir::interpret::{GlobalId, ConstValue, get_slice_bytes, sign_extend}; |
| use rustc::ty::{self, Region, TyCtxt, AdtDef, Ty, UserType, DefIdTree}; |
| use rustc::ty::{CanonicalUserType, CanonicalUserTypeAnnotation, CanonicalUserTypeAnnotations}; |
| use rustc::ty::subst::{SubstsRef, GenericArg}; |
| use rustc::ty::layout::VariantIdx; |
| use rustc::hir::{self, RangeEnd}; |
| use rustc::hir::def::{CtorOf, Res, DefKind, CtorKind}; |
| use rustc::hir::pat_util::EnumerateAndAdjustIterator; |
| use rustc::hir::ptr::P; |
| |
| use rustc_index::vec::Idx; |
| |
| use std::cmp::Ordering; |
| use std::fmt; |
| use syntax::ast; |
| use syntax_pos::{Span, DUMMY_SP}; |
| |
| use rustc_error_codes::*; |
| |
| #[derive(Clone, Debug)] |
| pub enum PatternError { |
| AssocConstInPattern(Span), |
| StaticInPattern(Span), |
| FloatBug, |
| NonConstPath(Span), |
| } |
| |
| #[derive(Copy, Clone, Debug)] |
| pub enum BindingMode { |
| ByValue, |
| ByRef(BorrowKind), |
| } |
| |
| #[derive(Clone, Debug)] |
| pub struct FieldPat<'tcx> { |
| pub field: Field, |
| pub pattern: Pat<'tcx>, |
| } |
| |
| #[derive(Clone, Debug)] |
| pub struct Pat<'tcx> { |
| pub ty: Ty<'tcx>, |
| pub span: Span, |
| pub kind: Box<PatKind<'tcx>>, |
| } |
| |
| impl<'tcx> Pat<'tcx> { |
| pub(crate) fn wildcard_from_ty(ty: Ty<'tcx>) -> Self { |
| Pat { ty, span: DUMMY_SP, kind: Box::new(PatKind::Wild) } |
| } |
| } |
| |
| #[derive(Copy, Clone, Debug, PartialEq)] |
| pub struct PatTyProj<'tcx> { |
| pub user_ty: CanonicalUserType<'tcx>, |
| } |
| |
| impl<'tcx> PatTyProj<'tcx> { |
| pub(crate) fn from_user_type(user_annotation: CanonicalUserType<'tcx>) -> Self { |
| Self { |
| user_ty: user_annotation, |
| } |
| } |
| |
| pub(crate) fn user_ty( |
| self, |
| annotations: &mut CanonicalUserTypeAnnotations<'tcx>, |
| inferred_ty: Ty<'tcx>, |
| span: Span, |
| ) -> UserTypeProjection { |
| UserTypeProjection { |
| base: annotations.push(CanonicalUserTypeAnnotation { |
| span, |
| user_ty: self.user_ty, |
| inferred_ty, |
| }), |
| projs: Vec::new(), |
| } |
| } |
| } |
| |
| #[derive(Copy, Clone, Debug, PartialEq)] |
| pub struct Ascription<'tcx> { |
| pub user_ty: PatTyProj<'tcx>, |
| /// Variance to use when relating the type `user_ty` to the **type of the value being |
| /// matched**. Typically, this is `Variance::Covariant`, since the value being matched must |
| /// have a type that is some subtype of the ascribed type. |
| /// |
| /// Note that this variance does not apply for any bindings within subpatterns. The type |
| /// assigned to those bindings must be exactly equal to the `user_ty` given here. |
| /// |
| /// The only place where this field is not `Covariant` is when matching constants, where |
| /// we currently use `Contravariant` -- this is because the constant type just needs to |
| /// be "comparable" to the type of the input value. So, for example: |
| /// |
| /// ```text |
| /// match x { "foo" => .. } |
| /// ``` |
| /// |
| /// requires that `&'static str <: T_x`, where `T_x` is the type of `x`. Really, we should |
| /// probably be checking for a `PartialEq` impl instead, but this preserves the behavior |
| /// of the old type-check for now. See #57280 for details. |
| pub variance: ty::Variance, |
| pub user_ty_span: Span, |
| } |
| |
| #[derive(Clone, Debug)] |
| pub enum PatKind<'tcx> { |
| Wild, |
| |
| AscribeUserType { |
| ascription: Ascription<'tcx>, |
| subpattern: Pat<'tcx>, |
| }, |
| |
| /// `x`, `ref x`, `x @ P`, etc. |
| Binding { |
| mutability: Mutability, |
| name: ast::Name, |
| mode: BindingMode, |
| var: hir::HirId, |
| ty: Ty<'tcx>, |
| subpattern: Option<Pat<'tcx>>, |
| }, |
| |
| /// `Foo(...)` or `Foo{...}` or `Foo`, where `Foo` is a variant name from an ADT with |
| /// multiple variants. |
| Variant { |
| adt_def: &'tcx AdtDef, |
| substs: SubstsRef<'tcx>, |
| variant_index: VariantIdx, |
| subpatterns: Vec<FieldPat<'tcx>>, |
| }, |
| |
| /// `(...)`, `Foo(...)`, `Foo{...}`, or `Foo`, where `Foo` is a variant name from an ADT with |
| /// a single variant. |
| Leaf { |
| subpatterns: Vec<FieldPat<'tcx>>, |
| }, |
| |
| /// `box P`, `&P`, `&mut P`, etc. |
| Deref { |
| subpattern: Pat<'tcx>, |
| }, |
| |
| Constant { |
| value: &'tcx ty::Const<'tcx>, |
| }, |
| |
| Range(PatRange<'tcx>), |
| |
| /// Matches against a slice, checking the length and extracting elements. |
| /// irrefutable when there is a slice pattern and both `prefix` and `suffix` are empty. |
| /// e.g., `&[ref xs @ ..]`. |
| Slice { |
| prefix: Vec<Pat<'tcx>>, |
| slice: Option<Pat<'tcx>>, |
| suffix: Vec<Pat<'tcx>>, |
| }, |
| |
| /// Fixed match against an array; irrefutable. |
| Array { |
| prefix: Vec<Pat<'tcx>>, |
| slice: Option<Pat<'tcx>>, |
| suffix: Vec<Pat<'tcx>>, |
| }, |
| |
| /// An or-pattern, e.g. `p | q`. |
| /// Invariant: `pats.len() >= 2`. |
| Or { |
| pats: Vec<Pat<'tcx>>, |
| }, |
| } |
| |
| #[derive(Copy, Clone, Debug, PartialEq)] |
| pub struct PatRange<'tcx> { |
| pub lo: &'tcx ty::Const<'tcx>, |
| pub hi: &'tcx ty::Const<'tcx>, |
| pub end: RangeEnd, |
| } |
| |
| impl<'tcx> fmt::Display for Pat<'tcx> { |
| fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result { |
| // Printing lists is a chore. |
| let mut first = true; |
| let mut start_or_continue = |s| { |
| if first { |
| first = false; |
| "" |
| } else { |
| s |
| } |
| }; |
| let mut start_or_comma = || start_or_continue(", "); |
| |
| match *self.kind { |
| PatKind::Wild => write!(f, "_"), |
| PatKind::AscribeUserType { ref subpattern, .. } => |
| write!(f, "{}: _", subpattern), |
| PatKind::Binding { mutability, name, mode, ref subpattern, .. } => { |
| let is_mut = match mode { |
| BindingMode::ByValue => mutability == Mutability::Mut, |
| BindingMode::ByRef(bk) => { |
| write!(f, "ref ")?; |
| match bk { BorrowKind::Mut { .. } => true, _ => false } |
| } |
| }; |
| if is_mut { |
| write!(f, "mut ")?; |
| } |
| write!(f, "{}", name)?; |
| if let Some(ref subpattern) = *subpattern { |
| write!(f, " @ {}", subpattern)?; |
| } |
| Ok(()) |
| } |
| PatKind::Variant { ref subpatterns, .. } | |
| PatKind::Leaf { ref subpatterns } => { |
| let variant = match *self.kind { |
| PatKind::Variant { adt_def, variant_index, .. } => { |
| Some(&adt_def.variants[variant_index]) |
| } |
| _ => if let ty::Adt(adt, _) = self.ty.kind { |
| if !adt.is_enum() { |
| Some(&adt.variants[VariantIdx::new(0)]) |
| } else { |
| None |
| } |
| } else { |
| None |
| } |
| }; |
| |
| if let Some(variant) = variant { |
| write!(f, "{}", variant.ident)?; |
| |
| // Only for Adt we can have `S {...}`, |
| // which we handle separately here. |
| if variant.ctor_kind == CtorKind::Fictive { |
| write!(f, " {{ ")?; |
| |
| let mut printed = 0; |
| for p in subpatterns { |
| if let PatKind::Wild = *p.pattern.kind { |
| continue; |
| } |
| let name = variant.fields[p.field.index()].ident; |
| write!(f, "{}{}: {}", start_or_comma(), name, p.pattern)?; |
| printed += 1; |
| } |
| |
| if printed < variant.fields.len() { |
| write!(f, "{}..", start_or_comma())?; |
| } |
| |
| return write!(f, " }}"); |
| } |
| } |
| |
| let num_fields = variant.map_or(subpatterns.len(), |v| v.fields.len()); |
| if num_fields != 0 || variant.is_none() { |
| write!(f, "(")?; |
| for i in 0..num_fields { |
| write!(f, "{}", start_or_comma())?; |
| |
| // Common case: the field is where we expect it. |
| if let Some(p) = subpatterns.get(i) { |
| if p.field.index() == i { |
| write!(f, "{}", p.pattern)?; |
| continue; |
| } |
| } |
| |
| // Otherwise, we have to go looking for it. |
| if let Some(p) = subpatterns.iter().find(|p| p.field.index() == i) { |
| write!(f, "{}", p.pattern)?; |
| } else { |
| write!(f, "_")?; |
| } |
| } |
| write!(f, ")")?; |
| } |
| |
| Ok(()) |
| } |
| PatKind::Deref { ref subpattern } => { |
| match self.ty.kind { |
| ty::Adt(def, _) if def.is_box() => write!(f, "box ")?, |
| ty::Ref(_, _, mutbl) => { |
| write!(f, "&{}", mutbl.prefix_str())?; |
| } |
| _ => bug!("{} is a bad Deref pattern type", self.ty) |
| } |
| write!(f, "{}", subpattern) |
| } |
| PatKind::Constant { value } => { |
| write!(f, "{}", value) |
| } |
| PatKind::Range(PatRange { lo, hi, end }) => { |
| write!(f, "{}", lo)?; |
| write!(f, "{}", end)?; |
| write!(f, "{}", hi) |
| } |
| PatKind::Slice { ref prefix, ref slice, ref suffix } | |
| PatKind::Array { ref prefix, ref slice, ref suffix } => { |
| write!(f, "[")?; |
| for p in prefix { |
| write!(f, "{}{}", start_or_comma(), p)?; |
| } |
| if let Some(ref slice) = *slice { |
| write!(f, "{}", start_or_comma())?; |
| match *slice.kind { |
| PatKind::Wild => {} |
| _ => write!(f, "{}", slice)? |
| } |
| write!(f, "..")?; |
| } |
| for p in suffix { |
| write!(f, "{}{}", start_or_comma(), p)?; |
| } |
| write!(f, "]") |
| } |
| PatKind::Or { ref pats } => { |
| for pat in pats { |
| write!(f, "{}{}", start_or_continue(" | "), pat)?; |
| } |
| Ok(()) |
| } |
| } |
| } |
| } |
| |
| pub struct PatCtxt<'a, 'tcx> { |
| pub tcx: TyCtxt<'tcx>, |
| pub param_env: ty::ParamEnv<'tcx>, |
| pub tables: &'a ty::TypeckTables<'tcx>, |
| pub substs: SubstsRef<'tcx>, |
| pub errors: Vec<PatternError>, |
| include_lint_checks: bool, |
| } |
| |
| impl<'a, 'tcx> Pat<'tcx> { |
| pub fn from_hir( |
| tcx: TyCtxt<'tcx>, |
| param_env_and_substs: ty::ParamEnvAnd<'tcx, SubstsRef<'tcx>>, |
| tables: &'a ty::TypeckTables<'tcx>, |
| pat: &'tcx hir::Pat, |
| ) -> Self { |
| let mut pcx = PatCtxt::new(tcx, param_env_and_substs, tables); |
| let result = pcx.lower_pattern(pat); |
| if !pcx.errors.is_empty() { |
| let msg = format!("encountered errors lowering pattern: {:?}", pcx.errors); |
| tcx.sess.delay_span_bug(pat.span, &msg); |
| } |
| debug!("Pat::from_hir({:?}) = {:?}", pat, result); |
| result |
| } |
| } |
| |
| impl<'a, 'tcx> PatCtxt<'a, 'tcx> { |
| pub fn new( |
| tcx: TyCtxt<'tcx>, |
| param_env_and_substs: ty::ParamEnvAnd<'tcx, SubstsRef<'tcx>>, |
| tables: &'a ty::TypeckTables<'tcx>, |
| ) -> Self { |
| PatCtxt { |
| tcx, |
| param_env: param_env_and_substs.param_env, |
| tables, |
| substs: param_env_and_substs.value, |
| errors: vec![], |
| include_lint_checks: false, |
| } |
| } |
| |
| pub fn include_lint_checks(&mut self) -> &mut Self { |
| self.include_lint_checks = true; |
| self |
| } |
| |
| pub fn lower_pattern(&mut self, pat: &'tcx hir::Pat) -> Pat<'tcx> { |
| // When implicit dereferences have been inserted in this pattern, the unadjusted lowered |
| // pattern has the type that results *after* dereferencing. For example, in this code: |
| // |
| // ``` |
| // match &&Some(0i32) { |
| // Some(n) => { ... }, |
| // _ => { ... }, |
| // } |
| // ``` |
| // |
| // the type assigned to `Some(n)` in `unadjusted_pat` would be `Option<i32>` (this is |
| // determined in rustc_typeck::check::match). The adjustments would be |
| // |
| // `vec![&&Option<i32>, &Option<i32>]`. |
| // |
| // Applying the adjustments, we want to instead output `&&Some(n)` (as a HAIR pattern). So |
| // we wrap the unadjusted pattern in `PatKind::Deref` repeatedly, consuming the |
| // adjustments in *reverse order* (last-in-first-out, so that the last `Deref` inserted |
| // gets the least-dereferenced type). |
| let unadjusted_pat = self.lower_pattern_unadjusted(pat); |
| self.tables |
| .pat_adjustments() |
| .get(pat.hir_id) |
| .unwrap_or(&vec![]) |
| .iter() |
| .rev() |
| .fold(unadjusted_pat, |pat, ref_ty| { |
| debug!("{:?}: wrapping pattern with type {:?}", pat, ref_ty); |
| Pat { |
| span: pat.span, |
| ty: ref_ty, |
| kind: Box::new(PatKind::Deref { subpattern: pat }), |
| } |
| }, |
| ) |
| } |
| |
| fn lower_range_expr( |
| &mut self, |
| expr: &'tcx hir::Expr, |
| ) -> (PatKind<'tcx>, Option<Ascription<'tcx>>) { |
| match self.lower_lit(expr) { |
| PatKind::AscribeUserType { |
| ascription: lo_ascription, |
| subpattern: Pat { kind: box kind, .. }, |
| } => (kind, Some(lo_ascription)), |
| kind => (kind, None), |
| } |
| } |
| |
| fn lower_pattern_unadjusted(&mut self, pat: &'tcx hir::Pat) -> Pat<'tcx> { |
| let mut ty = self.tables.node_type(pat.hir_id); |
| |
| let kind = match pat.kind { |
| hir::PatKind::Wild => PatKind::Wild, |
| |
| hir::PatKind::Lit(ref value) => self.lower_lit(value), |
| |
| hir::PatKind::Range(ref lo_expr, ref hi_expr, end) => { |
| let (lo, lo_ascription) = self.lower_range_expr(lo_expr); |
| let (hi, hi_ascription) = self.lower_range_expr(hi_expr); |
| |
| let mut kind = match (lo, hi) { |
| (PatKind::Constant { value: lo }, PatKind::Constant { value: hi }) => { |
| assert_eq!(lo.ty, ty); |
| assert_eq!(hi.ty, ty); |
| let cmp = compare_const_vals( |
| self.tcx, |
| lo, |
| hi, |
| self.param_env, |
| ty, |
| ); |
| match (end, cmp) { |
| (RangeEnd::Excluded, Some(Ordering::Less)) => |
| PatKind::Range(PatRange { lo, hi, end }), |
| (RangeEnd::Excluded, _) => { |
| span_err!( |
| self.tcx.sess, |
| lo_expr.span, |
| E0579, |
| "lower range bound must be less than upper", |
| ); |
| PatKind::Wild |
| } |
| (RangeEnd::Included, Some(Ordering::Equal)) => { |
| PatKind::Constant { value: lo } |
| } |
| (RangeEnd::Included, Some(Ordering::Less)) => { |
| PatKind::Range(PatRange { lo, hi, end }) |
| } |
| (RangeEnd::Included, _) => { |
| let mut err = struct_span_err!( |
| self.tcx.sess, |
| lo_expr.span, |
| E0030, |
| "lower range bound must be less than or equal to upper" |
| ); |
| err.span_label( |
| lo_expr.span, |
| "lower bound larger than upper bound", |
| ); |
| if self.tcx.sess.teach(&err.get_code().unwrap()) { |
| err.note("When matching against a range, the compiler \ |
| verifies that the range is non-empty. Range \ |
| patterns include both end-points, so this is \ |
| equivalent to requiring the start of the range \ |
| to be less than or equal to the end of the range."); |
| } |
| err.emit(); |
| PatKind::Wild |
| } |
| } |
| }, |
| ref pats => { |
| self.tcx.sess.delay_span_bug( |
| pat.span, |
| &format!( |
| "found bad range pattern `{:?}` outside of error recovery", |
| pats, |
| ), |
| ); |
| |
| PatKind::Wild |
| }, |
| }; |
| |
| // If we are handling a range with associated constants (e.g. |
| // `Foo::<'a>::A..=Foo::B`), we need to put the ascriptions for the associated |
| // constants somewhere. Have them on the range pattern. |
| for ascription in &[lo_ascription, hi_ascription] { |
| if let Some(ascription) = ascription { |
| kind = PatKind::AscribeUserType { |
| ascription: *ascription, |
| subpattern: Pat { span: pat.span, ty, kind: Box::new(kind), }, |
| }; |
| } |
| } |
| |
| kind |
| } |
| |
| hir::PatKind::Path(ref qpath) => { |
| return self.lower_path(qpath, pat.hir_id, pat.span); |
| } |
| |
| hir::PatKind::Ref(ref subpattern, _) | |
| hir::PatKind::Box(ref subpattern) => { |
| PatKind::Deref { subpattern: self.lower_pattern(subpattern) } |
| } |
| |
| hir::PatKind::Slice(ref prefix, ref slice, ref suffix) => { |
| match ty.kind { |
| ty::Ref(_, ty, _) => |
| PatKind::Deref { |
| subpattern: Pat { |
| ty, |
| span: pat.span, |
| kind: Box::new(self.slice_or_array_pattern( |
| pat.span, ty, prefix, slice, suffix)) |
| }, |
| }, |
| ty::Slice(..) | |
| ty::Array(..) => |
| self.slice_or_array_pattern(pat.span, ty, prefix, slice, suffix), |
| ty::Error => { // Avoid ICE |
| return Pat { span: pat.span, ty, kind: Box::new(PatKind::Wild) }; |
| } |
| _ => |
| span_bug!( |
| pat.span, |
| "unexpanded type for vector pattern: {:?}", |
| ty), |
| } |
| } |
| |
| hir::PatKind::Tuple(ref subpatterns, ddpos) => { |
| match ty.kind { |
| ty::Tuple(ref tys) => { |
| let subpatterns = |
| subpatterns.iter() |
| .enumerate_and_adjust(tys.len(), ddpos) |
| .map(|(i, subpattern)| FieldPat { |
| field: Field::new(i), |
| pattern: self.lower_pattern(subpattern) |
| }) |
| .collect(); |
| |
| PatKind::Leaf { subpatterns } |
| } |
| ty::Error => { // Avoid ICE (#50577) |
| return Pat { span: pat.span, ty, kind: Box::new(PatKind::Wild) }; |
| } |
| _ => span_bug!(pat.span, "unexpected type for tuple pattern: {:?}", ty), |
| } |
| } |
| |
| hir::PatKind::Binding(_, id, ident, ref sub) => { |
| let var_ty = self.tables.node_type(pat.hir_id); |
| if let ty::Error = var_ty.kind { |
| // Avoid ICE |
| return Pat { span: pat.span, ty, kind: Box::new(PatKind::Wild) }; |
| }; |
| let bm = *self.tables.pat_binding_modes().get(pat.hir_id) |
| .expect("missing binding mode"); |
| let (mutability, mode) = match bm { |
| ty::BindByValue(hir::Mutability::Mutable) => |
| (Mutability::Mut, BindingMode::ByValue), |
| ty::BindByValue(hir::Mutability::Immutable) => |
| (Mutability::Not, BindingMode::ByValue), |
| ty::BindByReference(hir::Mutability::Mutable) => |
| (Mutability::Not, BindingMode::ByRef( |
| BorrowKind::Mut { allow_two_phase_borrow: false })), |
| ty::BindByReference(hir::Mutability::Immutable) => |
| (Mutability::Not, BindingMode::ByRef( |
| BorrowKind::Shared)), |
| }; |
| |
| // A ref x pattern is the same node used for x, and as such it has |
| // x's type, which is &T, where we want T (the type being matched). |
| if let ty::BindByReference(_) = bm { |
| if let ty::Ref(_, rty, _) = ty.kind { |
| ty = rty; |
| } else { |
| bug!("`ref {}` has wrong type {}", ident, ty); |
| } |
| } |
| |
| PatKind::Binding { |
| mutability, |
| mode, |
| name: ident.name, |
| var: id, |
| ty: var_ty, |
| subpattern: self.lower_opt_pattern(sub), |
| } |
| } |
| |
| hir::PatKind::TupleStruct(ref qpath, ref subpatterns, ddpos) => { |
| let res = self.tables.qpath_res(qpath, pat.hir_id); |
| let adt_def = match ty.kind { |
| ty::Adt(adt_def, _) => adt_def, |
| ty::Error => { // Avoid ICE (#50585) |
| return Pat { span: pat.span, ty, kind: Box::new(PatKind::Wild) }; |
| } |
| _ => span_bug!(pat.span, |
| "tuple struct pattern not applied to an ADT {:?}", |
| ty), |
| }; |
| let variant_def = adt_def.variant_of_res(res); |
| |
| let subpatterns = |
| subpatterns.iter() |
| .enumerate_and_adjust(variant_def.fields.len(), ddpos) |
| .map(|(i, field)| FieldPat { |
| field: Field::new(i), |
| pattern: self.lower_pattern(field), |
| }) |
| .collect(); |
| |
| self.lower_variant_or_leaf(res, pat.hir_id, pat.span, ty, subpatterns) |
| } |
| |
| hir::PatKind::Struct(ref qpath, ref fields, _) => { |
| let res = self.tables.qpath_res(qpath, pat.hir_id); |
| let subpatterns = |
| fields.iter() |
| .map(|field| { |
| FieldPat { |
| field: Field::new(self.tcx.field_index(field.hir_id, |
| self.tables)), |
| pattern: self.lower_pattern(&field.pat), |
| } |
| }) |
| .collect(); |
| |
| self.lower_variant_or_leaf(res, pat.hir_id, pat.span, ty, subpatterns) |
| } |
| |
| hir::PatKind::Or(ref pats) => { |
| PatKind::Or { |
| pats: pats.iter().map(|p| self.lower_pattern(p)).collect(), |
| } |
| } |
| }; |
| |
| Pat { |
| span: pat.span, |
| ty, |
| kind: Box::new(kind), |
| } |
| } |
| |
| fn lower_patterns(&mut self, pats: &'tcx [P<hir::Pat>]) -> Vec<Pat<'tcx>> { |
| pats.iter().map(|p| self.lower_pattern(p)).collect() |
| } |
| |
| fn lower_opt_pattern(&mut self, pat: &'tcx Option<P<hir::Pat>>) -> Option<Pat<'tcx>> |
| { |
| pat.as_ref().map(|p| self.lower_pattern(p)) |
| } |
| |
| fn flatten_nested_slice_patterns( |
| &mut self, |
| prefix: Vec<Pat<'tcx>>, |
| slice: Option<Pat<'tcx>>, |
| suffix: Vec<Pat<'tcx>>) |
| -> (Vec<Pat<'tcx>>, Option<Pat<'tcx>>, Vec<Pat<'tcx>>) |
| { |
| let orig_slice = match slice { |
| Some(orig_slice) => orig_slice, |
| None => return (prefix, slice, suffix) |
| }; |
| let orig_prefix = prefix; |
| let orig_suffix = suffix; |
| |
| // dance because of intentional borrow-checker stupidity. |
| let kind = *orig_slice.kind; |
| match kind { |
| PatKind::Slice { prefix, slice, mut suffix } | |
| PatKind::Array { prefix, slice, mut suffix } => { |
| let mut orig_prefix = orig_prefix; |
| |
| orig_prefix.extend(prefix); |
| suffix.extend(orig_suffix); |
| |
| (orig_prefix, slice, suffix) |
| } |
| _ => { |
| (orig_prefix, Some(Pat { |
| kind: box kind, ..orig_slice |
| }), orig_suffix) |
| } |
| } |
| } |
| |
| fn slice_or_array_pattern( |
| &mut self, |
| span: Span, |
| ty: Ty<'tcx>, |
| prefix: &'tcx [P<hir::Pat>], |
| slice: &'tcx Option<P<hir::Pat>>, |
| suffix: &'tcx [P<hir::Pat>]) |
| -> PatKind<'tcx> |
| { |
| let prefix = self.lower_patterns(prefix); |
| let slice = self.lower_opt_pattern(slice); |
| let suffix = self.lower_patterns(suffix); |
| let (prefix, slice, suffix) = |
| self.flatten_nested_slice_patterns(prefix, slice, suffix); |
| |
| match ty.kind { |
| ty::Slice(..) => { |
| // matching a slice or fixed-length array |
| PatKind::Slice { prefix: prefix, slice: slice, suffix: suffix } |
| } |
| |
| ty::Array(_, len) => { |
| // fixed-length array |
| let len = len.eval_usize(self.tcx, self.param_env); |
| assert!(len >= prefix.len() as u64 + suffix.len() as u64); |
| PatKind::Array { prefix: prefix, slice: slice, suffix: suffix } |
| } |
| |
| _ => { |
| span_bug!(span, "bad slice pattern type {:?}", ty); |
| } |
| } |
| } |
| |
| fn lower_variant_or_leaf( |
| &mut self, |
| res: Res, |
| hir_id: hir::HirId, |
| span: Span, |
| ty: Ty<'tcx>, |
| subpatterns: Vec<FieldPat<'tcx>>, |
| ) -> PatKind<'tcx> { |
| let res = match res { |
| Res::Def(DefKind::Ctor(CtorOf::Variant, ..), variant_ctor_id) => { |
| let variant_id = self.tcx.parent(variant_ctor_id).unwrap(); |
| Res::Def(DefKind::Variant, variant_id) |
| }, |
| res => res, |
| }; |
| |
| let mut kind = match res { |
| Res::Def(DefKind::Variant, variant_id) => { |
| let enum_id = self.tcx.parent(variant_id).unwrap(); |
| let adt_def = self.tcx.adt_def(enum_id); |
| if adt_def.is_enum() { |
| let substs = match ty.kind { |
| ty::Adt(_, substs) | |
| ty::FnDef(_, substs) => substs, |
| ty::Error => { // Avoid ICE (#50585) |
| return PatKind::Wild; |
| } |
| _ => bug!("inappropriate type for def: {:?}", ty), |
| }; |
| PatKind::Variant { |
| adt_def, |
| substs, |
| variant_index: adt_def.variant_index_with_id(variant_id), |
| subpatterns, |
| } |
| } else { |
| PatKind::Leaf { subpatterns } |
| } |
| } |
| |
| Res::Def(DefKind::Struct, _) |
| | Res::Def(DefKind::Ctor(CtorOf::Struct, ..), _) |
| | Res::Def(DefKind::Union, _) |
| | Res::Def(DefKind::TyAlias, _) |
| | Res::Def(DefKind::AssocTy, _) |
| | Res::SelfTy(..) |
| | Res::SelfCtor(..) => { |
| PatKind::Leaf { subpatterns } |
| } |
| |
| _ => { |
| self.errors.push(PatternError::NonConstPath(span)); |
| PatKind::Wild |
| } |
| }; |
| |
| if let Some(user_ty) = self.user_substs_applied_to_ty_of_hir_id(hir_id) { |
| debug!("lower_variant_or_leaf: kind={:?} user_ty={:?} span={:?}", kind, user_ty, span); |
| kind = PatKind::AscribeUserType { |
| subpattern: Pat { |
| span, |
| ty, |
| kind: Box::new(kind), |
| }, |
| ascription: Ascription { |
| user_ty: PatTyProj::from_user_type(user_ty), |
| user_ty_span: span, |
| variance: ty::Variance::Covariant, |
| }, |
| }; |
| } |
| |
| kind |
| } |
| |
| /// Takes a HIR Path. If the path is a constant, evaluates it and feeds |
| /// it to `const_to_pat`. Any other path (like enum variants without fields) |
| /// is converted to the corresponding pattern via `lower_variant_or_leaf`. |
| fn lower_path(&mut self, |
| qpath: &hir::QPath, |
| id: hir::HirId, |
| span: Span) |
| -> Pat<'tcx> { |
| let ty = self.tables.node_type(id); |
| let res = self.tables.qpath_res(qpath, id); |
| let is_associated_const = match res { |
| Res::Def(DefKind::AssocConst, _) => true, |
| _ => false, |
| }; |
| let kind = match res { |
| Res::Def(DefKind::Const, def_id) | Res::Def(DefKind::AssocConst, def_id) => { |
| let substs = self.tables.node_substs(id); |
| match ty::Instance::resolve( |
| self.tcx, |
| self.param_env, |
| def_id, |
| substs, |
| ) { |
| Some(instance) => { |
| let cid = GlobalId { |
| instance, |
| promoted: None, |
| }; |
| match self.tcx.at(span).const_eval(self.param_env.and(cid)) { |
| Ok(value) => { |
| let pattern = self.const_to_pat(value, id, span); |
| if !is_associated_const { |
| return pattern; |
| } |
| |
| let user_provided_types = self.tables().user_provided_types(); |
| return if let Some(u_ty) = user_provided_types.get(id) { |
| let user_ty = PatTyProj::from_user_type(*u_ty); |
| Pat { |
| span, |
| kind: Box::new( |
| PatKind::AscribeUserType { |
| subpattern: pattern, |
| ascription: Ascription { |
| /// Note that use `Contravariant` here. See the |
| /// `variance` field documentation for details. |
| variance: ty::Variance::Contravariant, |
| user_ty, |
| user_ty_span: span, |
| }, |
| } |
| ), |
| ty: value.ty, |
| } |
| } else { |
| pattern |
| } |
| }, |
| Err(_) => { |
| self.tcx.sess.span_err( |
| span, |
| "could not evaluate constant pattern", |
| ); |
| PatKind::Wild |
| } |
| } |
| }, |
| None => { |
| self.errors.push(if is_associated_const { |
| PatternError::AssocConstInPattern(span) |
| } else { |
| PatternError::StaticInPattern(span) |
| }); |
| PatKind::Wild |
| }, |
| } |
| } |
| _ => self.lower_variant_or_leaf(res, id, span, ty, vec![]), |
| }; |
| |
| Pat { |
| span, |
| ty, |
| kind: Box::new(kind), |
| } |
| } |
| |
| /// Converts literals, paths and negation of literals to patterns. |
| /// The special case for negation exists to allow things like `-128_i8` |
| /// which would overflow if we tried to evaluate `128_i8` and then negate |
| /// afterwards. |
| fn lower_lit(&mut self, expr: &'tcx hir::Expr) -> PatKind<'tcx> { |
| match expr.kind { |
| hir::ExprKind::Lit(ref lit) => { |
| let ty = self.tables.expr_ty(expr); |
| match lit_to_const(&lit.node, self.tcx, ty, false) { |
| Ok(val) => { |
| *self.const_to_pat(val, expr.hir_id, lit.span).kind |
| }, |
| Err(LitToConstError::UnparseableFloat) => { |
| self.errors.push(PatternError::FloatBug); |
| PatKind::Wild |
| }, |
| Err(LitToConstError::Reported) => PatKind::Wild, |
| } |
| }, |
| hir::ExprKind::Path(ref qpath) => *self.lower_path(qpath, expr.hir_id, expr.span).kind, |
| hir::ExprKind::Unary(hir::UnNeg, ref expr) => { |
| let ty = self.tables.expr_ty(expr); |
| let lit = match expr.kind { |
| hir::ExprKind::Lit(ref lit) => lit, |
| _ => span_bug!(expr.span, "not a literal: {:?}", expr), |
| }; |
| match lit_to_const(&lit.node, self.tcx, ty, true) { |
| Ok(val) => { |
| *self.const_to_pat(val, expr.hir_id, lit.span).kind |
| }, |
| Err(LitToConstError::UnparseableFloat) => { |
| self.errors.push(PatternError::FloatBug); |
| PatKind::Wild |
| }, |
| Err(LitToConstError::Reported) => PatKind::Wild, |
| } |
| } |
| _ => span_bug!(expr.span, "not a literal: {:?}", expr), |
| } |
| } |
| } |
| |
| impl UserAnnotatedTyHelpers<'tcx> for PatCtxt<'_, 'tcx> { |
| fn tcx(&self) -> TyCtxt<'tcx> { |
| self.tcx |
| } |
| |
| fn tables(&self) -> &ty::TypeckTables<'tcx> { |
| self.tables |
| } |
| } |
| |
| |
| pub trait PatternFoldable<'tcx> : Sized { |
| fn fold_with<F: PatternFolder<'tcx>>(&self, folder: &mut F) -> Self { |
| self.super_fold_with(folder) |
| } |
| |
| fn super_fold_with<F: PatternFolder<'tcx>>(&self, folder: &mut F) -> Self; |
| } |
| |
| pub trait PatternFolder<'tcx> : Sized { |
| fn fold_pattern(&mut self, pattern: &Pat<'tcx>) -> Pat<'tcx> { |
| pattern.super_fold_with(self) |
| } |
| |
| fn fold_pattern_kind(&mut self, kind: &PatKind<'tcx>) -> PatKind<'tcx> { |
| kind.super_fold_with(self) |
| } |
| } |
| |
| |
| impl<'tcx, T: PatternFoldable<'tcx>> PatternFoldable<'tcx> for Box<T> { |
| fn super_fold_with<F: PatternFolder<'tcx>>(&self, folder: &mut F) -> Self { |
| let content: T = (**self).fold_with(folder); |
| box content |
| } |
| } |
| |
| impl<'tcx, T: PatternFoldable<'tcx>> PatternFoldable<'tcx> for Vec<T> { |
| fn super_fold_with<F: PatternFolder<'tcx>>(&self, folder: &mut F) -> Self { |
| self.iter().map(|t| t.fold_with(folder)).collect() |
| } |
| } |
| |
| impl<'tcx, T: PatternFoldable<'tcx>> PatternFoldable<'tcx> for Option<T> { |
| fn super_fold_with<F: PatternFolder<'tcx>>(&self, folder: &mut F) -> Self{ |
| self.as_ref().map(|t| t.fold_with(folder)) |
| } |
| } |
| |
| macro_rules! CloneImpls { |
| (<$lt_tcx:tt> $($ty:ty),+) => { |
| $( |
| impl<$lt_tcx> PatternFoldable<$lt_tcx> for $ty { |
| fn super_fold_with<F: PatternFolder<$lt_tcx>>(&self, _: &mut F) -> Self { |
| Clone::clone(self) |
| } |
| } |
| )+ |
| } |
| } |
| |
| CloneImpls!{ <'tcx> |
| Span, Field, Mutability, ast::Name, hir::HirId, usize, ty::Const<'tcx>, |
| Region<'tcx>, Ty<'tcx>, BindingMode, &'tcx AdtDef, |
| SubstsRef<'tcx>, &'tcx GenericArg<'tcx>, UserType<'tcx>, |
| UserTypeProjection, PatTyProj<'tcx> |
| } |
| |
| impl<'tcx> PatternFoldable<'tcx> for FieldPat<'tcx> { |
| fn super_fold_with<F: PatternFolder<'tcx>>(&self, folder: &mut F) -> Self { |
| FieldPat { |
| field: self.field.fold_with(folder), |
| pattern: self.pattern.fold_with(folder) |
| } |
| } |
| } |
| |
| impl<'tcx> PatternFoldable<'tcx> for Pat<'tcx> { |
| fn fold_with<F: PatternFolder<'tcx>>(&self, folder: &mut F) -> Self { |
| folder.fold_pattern(self) |
| } |
| |
| fn super_fold_with<F: PatternFolder<'tcx>>(&self, folder: &mut F) -> Self { |
| Pat { |
| ty: self.ty.fold_with(folder), |
| span: self.span.fold_with(folder), |
| kind: self.kind.fold_with(folder) |
| } |
| } |
| } |
| |
| impl<'tcx> PatternFoldable<'tcx> for PatKind<'tcx> { |
| fn fold_with<F: PatternFolder<'tcx>>(&self, folder: &mut F) -> Self { |
| folder.fold_pattern_kind(self) |
| } |
| |
| fn super_fold_with<F: PatternFolder<'tcx>>(&self, folder: &mut F) -> Self { |
| match *self { |
| PatKind::Wild => PatKind::Wild, |
| PatKind::AscribeUserType { |
| ref subpattern, |
| ascription: Ascription { |
| variance, |
| ref user_ty, |
| user_ty_span, |
| }, |
| } => PatKind::AscribeUserType { |
| subpattern: subpattern.fold_with(folder), |
| ascription: Ascription { |
| user_ty: user_ty.fold_with(folder), |
| variance, |
| user_ty_span, |
| }, |
| }, |
| PatKind::Binding { |
| mutability, |
| name, |
| mode, |
| var, |
| ty, |
| ref subpattern, |
| } => PatKind::Binding { |
| mutability: mutability.fold_with(folder), |
| name: name.fold_with(folder), |
| mode: mode.fold_with(folder), |
| var: var.fold_with(folder), |
| ty: ty.fold_with(folder), |
| subpattern: subpattern.fold_with(folder), |
| }, |
| PatKind::Variant { |
| adt_def, |
| substs, |
| variant_index, |
| ref subpatterns, |
| } => PatKind::Variant { |
| adt_def: adt_def.fold_with(folder), |
| substs: substs.fold_with(folder), |
| variant_index, |
| subpatterns: subpatterns.fold_with(folder) |
| }, |
| PatKind::Leaf { |
| ref subpatterns, |
| } => PatKind::Leaf { |
| subpatterns: subpatterns.fold_with(folder), |
| }, |
| PatKind::Deref { |
| ref subpattern, |
| } => PatKind::Deref { |
| subpattern: subpattern.fold_with(folder), |
| }, |
| PatKind::Constant { |
| value |
| } => PatKind::Constant { |
| value, |
| }, |
| PatKind::Range(range) => PatKind::Range(range), |
| PatKind::Slice { |
| ref prefix, |
| ref slice, |
| ref suffix, |
| } => PatKind::Slice { |
| prefix: prefix.fold_with(folder), |
| slice: slice.fold_with(folder), |
| suffix: suffix.fold_with(folder) |
| }, |
| PatKind::Array { |
| ref prefix, |
| ref slice, |
| ref suffix |
| } => PatKind::Array { |
| prefix: prefix.fold_with(folder), |
| slice: slice.fold_with(folder), |
| suffix: suffix.fold_with(folder) |
| }, |
| PatKind::Or { ref pats } => PatKind::Or { pats: pats.fold_with(folder) }, |
| } |
| } |
| } |
| |
| pub fn compare_const_vals<'tcx>( |
| tcx: TyCtxt<'tcx>, |
| a: &'tcx ty::Const<'tcx>, |
| b: &'tcx ty::Const<'tcx>, |
| param_env: ty::ParamEnv<'tcx>, |
| ty: Ty<'tcx>, |
| ) -> Option<Ordering> { |
| trace!("compare_const_vals: {:?}, {:?}", a, b); |
| |
| let from_bool = |v: bool| v.then_some(Ordering::Equal); |
| |
| let fallback = || from_bool(a == b); |
| |
| // Use the fallback if any type differs |
| if a.ty != b.ty || a.ty != ty { |
| return fallback(); |
| } |
| |
| let a_bits = a.try_eval_bits(tcx, param_env, ty); |
| let b_bits = b.try_eval_bits(tcx, param_env, ty); |
| |
| if let (Some(a), Some(b)) = (a_bits, b_bits) { |
| use ::rustc_apfloat::Float; |
| return match ty.kind { |
| ty::Float(ast::FloatTy::F32) => { |
| let l = ::rustc_apfloat::ieee::Single::from_bits(a); |
| let r = ::rustc_apfloat::ieee::Single::from_bits(b); |
| l.partial_cmp(&r) |
| } |
| ty::Float(ast::FloatTy::F64) => { |
| let l = ::rustc_apfloat::ieee::Double::from_bits(a); |
| let r = ::rustc_apfloat::ieee::Double::from_bits(b); |
| l.partial_cmp(&r) |
| } |
| ty::Int(ity) => { |
| use rustc::ty::layout::{Integer, IntegerExt}; |
| use syntax::attr::SignedInt; |
| let size = Integer::from_attr(&tcx, SignedInt(ity)).size(); |
| let a = sign_extend(a, size); |
| let b = sign_extend(b, size); |
| Some((a as i128).cmp(&(b as i128))) |
| } |
| _ => Some(a.cmp(&b)), |
| } |
| } |
| |
| if let ty::Str = ty.kind { |
| match (a.val, b.val) { |
| (ty::ConstKind::Value(a_val @ ConstValue::Slice { .. }), |
| ty::ConstKind::Value(b_val @ ConstValue::Slice { .. })) => { |
| let a_bytes = get_slice_bytes(&tcx, a_val); |
| let b_bytes = get_slice_bytes(&tcx, b_val); |
| return from_bool(a_bytes == b_bytes); |
| } |
| _ => (), |
| } |
| } |
| |
| fallback() |
| } |
