Skip to content

Commit c3783c4

Browse files
oli-obkoli-obk
committed
Exhaustively handle expressions in patterns
1 parent 044ce57 commit c3783c4

File tree

36 files changed

+412
-193
lines changed

36 files changed

+412
-193
lines changed

‎compiler/rustc_ast_lowering/src/expr.rs

+28-23
Original file line numberDiff line numberDiff line change
@@ -102,17 +102,7 @@ impl<'hir> LoweringContext<'_, 'hir> {
102102

103103
let kind = match &e.kind {
104104
ExprKind::Array(exprs) => hir::ExprKind::Array(self.lower_exprs(exprs)),
105-
ExprKind::ConstBlock(c) => {
106-
let c = self.with_new_scopes(c.value.span, |this| {
107-
let def_id = this.local_def_id(c.id);
108-
hir::ConstBlock {
109-
def_id,
110-
hir_id: this.lower_node_id(c.id),
111-
body: this.lower_const_body(c.value.span, Some(&c.value)),
112-
}
113-
});
114-
hir::ExprKind::ConstBlock(c)
115-
}
105+
ExprKind::ConstBlock(c) => hir::ExprKind::ConstBlock(self.lower_const_block(c)),
116106
ExprKind::Repeat(expr, count) => {
117107
let expr = self.lower_expr(expr);
118108
let count = self.lower_array_length_to_const_arg(count);
@@ -153,18 +143,7 @@ impl<'hir> LoweringContext<'_, 'hir> {
153143
let ohs = self.lower_expr(ohs);
154144
hir::ExprKind::Unary(op, ohs)
155145
}
156-
ExprKind::Lit(token_lit) => {
157-
let lit_kind = match LitKind::from_token_lit(*token_lit) {
158-
Ok(lit_kind) => lit_kind,
159-
Err(err) => {
160-
let guar =
161-
report_lit_error(&self.tcx.sess.psess, err, *token_lit, e.span);
162-
LitKind::Err(guar)
163-
}
164-
};
165-
let lit = self.arena.alloc(respan(self.lower_span(e.span), lit_kind));
166-
hir::ExprKind::Lit(lit)
167-
}
146+
ExprKind::Lit(token_lit) => hir::ExprKind::Lit(self.lower_lit(token_lit, e.span)),
168147
ExprKind::IncludedBytes(bytes) => {
169148
let lit = self.arena.alloc(respan(
170149
self.lower_span(e.span),
@@ -403,6 +382,32 @@ impl<'hir> LoweringContext<'_, 'hir> {
403382
})
404383
}
405384

385+
pub(crate) fn lower_const_block(&mut self, c: &AnonConst) -> hir::ConstBlock {
386+
self.with_new_scopes(c.value.span, |this| {
387+
let def_id = this.local_def_id(c.id);
388+
hir::ConstBlock {
389+
def_id,
390+
hir_id: this.lower_node_id(c.id),
391+
body: this.lower_const_body(c.value.span, Some(&c.value)),
392+
}
393+
})
394+
}
395+
396+
pub(crate) fn lower_lit(
397+
&mut self,
398+
token_lit: &token::Lit,
399+
span: Span,
400+
) -> &'hir Spanned<LitKind> {
401+
let lit_kind = match LitKind::from_token_lit(*token_lit) {
402+
Ok(lit_kind) => lit_kind,
403+
Err(err) => {
404+
let guar = report_lit_error(&self.tcx.sess.psess, err, *token_lit, span);
405+
LitKind::Err(guar)
406+
}
407+
};
408+
self.arena.alloc(respan(self.lower_span(span), lit_kind))
409+
}
410+
406411
fn lower_unop(&mut self, u: UnOp) -> hir::UnOp {
407412
match u {
408413
UnOp::Deref => hir::UnOp::Deref,

‎compiler/rustc_ast_lowering/src/index.rs

+8
Original file line numberDiff line numberDiff line change
@@ -209,6 +209,14 @@ impl<'a, 'hir> Visitor<'hir> for NodeCollector<'a, 'hir> {
209209
});
210210
}
211211

212+
fn visit_pat_lit(&mut self, lit: &'hir PatLit<'hir>) {
213+
self.insert(lit.span, lit.hir_id, Node::PatLit(lit));
214+
215+
self.with_parent(lit.hir_id, |this| {
216+
intravisit::walk_pat_lit(this, lit);
217+
});
218+
}
219+
212220
fn visit_pat_field(&mut self, field: &'hir PatField<'hir>) {
213221
self.insert(field.span, field.hir_id, Node::PatField(field));
214222
self.with_parent(field.hir_id, |this| {

‎compiler/rustc_ast_lowering/src/lib.rs

+1
Original file line numberDiff line numberDiff line change
@@ -35,6 +35,7 @@
3535
#![doc(rust_logo)]
3636
#![feature(assert_matches)]
3737
#![feature(box_patterns)]
38+
#![feature(if_let_guard)]
3839
#![feature(let_chains)]
3940
#![feature(rustdoc_internals)]
4041
#![warn(unreachable_pub)]

‎compiler/rustc_ast_lowering/src/pat.rs

+38-13
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,12 @@
1+
use std::sync::Arc;
2+
13
use rustc_ast::ptr::P;
24
use rustc_ast::*;
35
use rustc_data_structures::stack::ensure_sufficient_stack;
46
use rustc_hir as hir;
57
use rustc_hir::def::Res;
6-
use rustc_span::source_map::Spanned;
8+
use rustc_middle::span_bug;
9+
use rustc_span::source_map::{Spanned, respan};
710
use rustc_span::{Ident, Span};
811

912
use super::errors::{
@@ -366,24 +369,46 @@ impl<'a, 'hir> LoweringContext<'a, 'hir> {
366369
// }
367370
// m!(S);
368371
// ```
369-
fn lower_expr_within_pat(&mut self, expr: &Expr, allow_paths: bool) -> &'hir hir::Expr<'hir> {
370-
match &expr.kind {
371-
ExprKind::Lit(..)
372-
| ExprKind::ConstBlock(..)
373-
| ExprKind::IncludedBytes(..)
374-
| ExprKind::Err(_)
375-
| ExprKind::Dummy => {}
376-
ExprKind::Path(..) if allow_paths => {}
377-
ExprKind::Unary(UnOp::Neg, inner) if matches!(inner.kind, ExprKind::Lit(_)) => {}
372+
fn lower_expr_within_pat(&mut self, expr: &Expr, allow_paths: bool) -> &'hir hir::PatLit<'hir> {
373+
let err = |guar| hir::PatLitKind::Lit {
374+
lit: self.arena.alloc(respan(self.lower_span(expr.span), LitKind::Err(guar))),
375+
negated: false,
376+
};
377+
let kind = match &expr.kind {
378+
ExprKind::Lit(lit) => {
379+
hir::PatLitKind::Lit { lit: self.lower_lit(lit, expr.span), negated: false }
380+
}
381+
ExprKind::ConstBlock(c) => hir::PatLitKind::ConstBlock(self.lower_const_block(c)),
382+
ExprKind::IncludedBytes(bytes) => hir::PatLitKind::Lit {
383+
lit: self.arena.alloc(respan(
384+
self.lower_span(expr.span),
385+
LitKind::ByteStr(Arc::clone(bytes), StrStyle::Cooked),
386+
)),
387+
negated: false,
388+
},
389+
ExprKind::Err(guar) => err(*guar),
390+
ExprKind::Dummy => span_bug!(expr.span, "lowered ExprKind::Dummy"),
391+
ExprKind::Path(qself, path) if allow_paths => hir::PatLitKind::Path(self.lower_qpath(
392+
expr.id,
393+
qself,
394+
path,
395+
ParamMode::Optional,
396+
AllowReturnTypeNotation::No,
397+
ImplTraitContext::Disallowed(ImplTraitPosition::Path),
398+
None,
399+
)),
400+
ExprKind::Unary(UnOp::Neg, inner) if let ExprKind::Lit(lit) = &inner.kind => {
401+
hir::PatLitKind::Lit { lit: self.lower_lit(lit, expr.span), negated: true }
402+
}
378403
_ => {
379404
let pattern_from_macro = expr.is_approximately_pattern();
380405
let guar = self.dcx().emit_err(ArbitraryExpressionInPattern {
381406
span: expr.span,
382407
pattern_from_macro_note: pattern_from_macro,
383408
});
384-
return self.arena.alloc(self.expr_err(expr.span, guar));
409+
err(guar)
385410
}
386-
}
387-
self.lower_expr(expr)
411+
};
412+
self.arena.alloc(hir::PatLit { hir_id: self.lower_node_id(expr.id), span: expr.span, kind })
388413
}
389414
}

‎compiler/rustc_hir/src/hir.rs

+27-2
Original file line numberDiff line numberDiff line change
@@ -1517,6 +1517,26 @@ impl fmt::Debug for DotDotPos {
15171517
}
15181518
}
15191519

1520+
#[derive(Debug, Clone, Copy, HashStable_Generic)]
1521+
pub struct PatLit<'hir> {
1522+
pub hir_id: HirId,
1523+
pub span: Span,
1524+
pub kind: PatLitKind<'hir>,
1525+
}
1526+
1527+
#[derive(Debug, Clone, Copy, HashStable_Generic)]
1528+
pub enum PatLitKind<'hir> {
1529+
Lit {
1530+
lit: &'hir Lit,
1531+
// FIXME: move this into `Lit` and handle negated literal expressions
1532+
// once instead of matching on unop neg expressions everywhere.
1533+
negated: bool,
1534+
},
1535+
ConstBlock(ConstBlock),
1536+
/// A path pattern for a unit struct/variant or a (maybe-associated) constant.
1537+
Path(QPath<'hir>),
1538+
}
1539+
15201540
#[derive(Debug, Clone, Copy, HashStable_Generic)]
15211541
pub enum PatKind<'hir> {
15221542
/// Represents a wildcard pattern (i.e., `_`).
@@ -1562,10 +1582,10 @@ pub enum PatKind<'hir> {
15621582
Ref(&'hir Pat<'hir>, Mutability),
15631583

15641584
/// A literal.
1565-
Lit(&'hir Expr<'hir>),
1585+
Lit(&'hir PatLit<'hir>),
15661586

15671587
/// A range pattern (e.g., `1..=2` or `1..2`).
1568-
Range(Option<&'hir Expr<'hir>>, Option<&'hir Expr<'hir>>, RangeEnd),
1588+
Range(Option<&'hir PatLit<'hir>>, Option<&'hir PatLit<'hir>>, RangeEnd),
15691589

15701590
/// A slice pattern, `[before_0, ..., before_n, (slice, after_0, ..., after_n)?]`.
15711591
///
@@ -4126,6 +4146,10 @@ pub enum Node<'hir> {
41264146
OpaqueTy(&'hir OpaqueTy<'hir>),
41274147
Pat(&'hir Pat<'hir>),
41284148
PatField(&'hir PatField<'hir>),
4149+
/// Needed as its own node with its own HirId for tracking
4150+
/// the unadjusted type of literals within patterns
4151+
/// (e.g. byte str literals not being of slice type).
4152+
PatLit(&'hir PatLit<'hir>),
41294153
Arm(&'hir Arm<'hir>),
41304154
Block(&'hir Block<'hir>),
41314155
LetStmt(&'hir LetStmt<'hir>),
@@ -4182,6 +4206,7 @@ impl<'hir> Node<'hir> {
41824206
| Node::Block(..)
41834207
| Node::Ctor(..)
41844208
| Node::Pat(..)
4209+
| Node::PatLit(..)
41854210
| Node::Arm(..)
41864211
| Node::LetStmt(..)
41874212
| Node::Crate(..)

‎compiler/rustc_hir/src/intravisit.rs

+15-3
Original file line numberDiff line numberDiff line change
@@ -342,6 +342,9 @@ pub trait Visitor<'v>: Sized {
342342
fn visit_pat_field(&mut self, f: &'v PatField<'v>) -> Self::Result {
343343
walk_pat_field(self, f)
344344
}
345+
fn visit_pat_lit(&mut self, lit: &'v PatLit<'v>) -> Self::Result {
346+
walk_pat_lit(self, lit)
347+
}
345348
fn visit_anon_const(&mut self, c: &'v AnonConst) -> Self::Result {
346349
walk_anon_const(self, c)
347350
}
@@ -685,10 +688,10 @@ pub fn walk_pat<'v, V: Visitor<'v>>(visitor: &mut V, pattern: &'v Pat<'v>) -> V:
685688
try_visit!(visitor.visit_ident(ident));
686689
visit_opt!(visitor, visit_pat, optional_subpattern);
687690
}
688-
PatKind::Lit(ref expression) => try_visit!(visitor.visit_expr(expression)),
691+
PatKind::Lit(lit) => try_visit!(visitor.visit_pat_lit(lit)),
689692
PatKind::Range(ref lower_bound, ref upper_bound, _) => {
690-
visit_opt!(visitor, visit_expr, lower_bound);
691-
visit_opt!(visitor, visit_expr, upper_bound);
693+
visit_opt!(visitor, visit_pat_lit, lower_bound);
694+
visit_opt!(visitor, visit_pat_lit, upper_bound);
692695
}
693696
PatKind::Never | PatKind::Wild | PatKind::Err(_) => (),
694697
PatKind::Slice(prepatterns, ref slice_pattern, postpatterns) => {
@@ -706,6 +709,15 @@ pub fn walk_pat_field<'v, V: Visitor<'v>>(visitor: &mut V, field: &'v PatField<'
706709
visitor.visit_pat(field.pat)
707710
}
708711

712+
pub fn walk_pat_lit<'v, V: Visitor<'v>>(visitor: &mut V, lit: &'v PatLit<'v>) -> V::Result {
713+
try_visit!(visitor.visit_id(lit.hir_id));
714+
match &lit.kind {
715+
PatLitKind::Lit { .. } => V::Result::output(),
716+
PatLitKind::ConstBlock(c) => visitor.visit_inline_const(c),
717+
PatLitKind::Path(qpath) => visitor.visit_qpath(qpath, lit.hir_id, lit.span),
718+
}
719+
}
720+
709721
pub fn walk_anon_const<'v, V: Visitor<'v>>(visitor: &mut V, constant: &'v AnonConst) -> V::Result {
710722
try_visit!(visitor.visit_id(constant.hir_id));
711723
visitor.visit_nested_body(constant.body)

‎compiler/rustc_hir_analysis/src/hir_ty_lowering/mod.rs

+20-22
Original file line numberDiff line numberDiff line change
@@ -2449,17 +2449,11 @@ impl<'tcx> dyn HirTyLowerer<'tcx> + '_ {
24492449
Ty::new_error(tcx, err)
24502450
}
24512451
hir::PatKind::Range(start, end, include_end) => {
2452-
let expr_to_const = |expr: &'tcx hir::Expr<'tcx>| -> ty::Const<'tcx> {
2453-
let (expr, neg) = match expr.kind {
2454-
hir::ExprKind::Unary(hir::UnOp::Neg, negated) => {
2455-
(negated, Some((expr.hir_id, expr.span)))
2456-
}
2457-
_ => (expr, None),
2458-
};
2459-
let (c, c_ty) = match &expr.kind {
2460-
hir::ExprKind::Lit(lit) => {
2452+
let expr_to_const = |expr: &'tcx hir::PatLit<'tcx>| -> ty::Const<'tcx> {
2453+
let (c, c_ty) = match expr.kind {
2454+
hir::PatLitKind::Lit { lit, negated } => {
24612455
let lit_input =
2462-
LitToConstInput { lit: &lit.node, ty, neg: neg.is_some() };
2456+
LitToConstInput { lit: &lit.node, ty, neg: negated };
24632457
let ct = match tcx.lit_to_const(lit_input) {
24642458
Ok(c) => c,
24652459
Err(LitToConstError::Reported(err)) => {
@@ -2470,23 +2464,30 @@ impl<'tcx> dyn HirTyLowerer<'tcx> + '_ {
24702464
(ct, ty)
24712465
}
24722466

2473-
hir::ExprKind::Path(hir::QPath::Resolved(
2467+
hir::PatLitKind::Path(hir::QPath::Resolved(
24742468
_,
24752469
path @ &hir::Path {
24762470
res: Res::Def(DefKind::ConstParam, def_id),
24772471
..
24782472
},
24792473
)) => {
2480-
let _ = self.prohibit_generic_args(
2474+
match self.prohibit_generic_args(
24812475
path.segments.iter(),
24822476
GenericsArgsErrExtend::Param(def_id),
2483-
);
2484-
let ty = tcx
2485-
.type_of(def_id)
2486-
.no_bound_vars()
2487-
.expect("const parameter types cannot be generic");
2488-
let ct = self.lower_const_param(def_id, expr.hir_id);
2489-
(ct, ty)
2477+
) {
2478+
Ok(()) => {
2479+
let ty = tcx
2480+
.type_of(def_id)
2481+
.no_bound_vars()
2482+
.expect("const parameter types cannot be generic");
2483+
let ct = self.lower_const_param(def_id, expr.hir_id);
2484+
(ct, ty)
2485+
}
2486+
Err(guar) => (
2487+
ty::Const::new_error(tcx, guar),
2488+
Ty::new_error(tcx, guar),
2489+
),
2490+
}
24902491
}
24912492

24922493
_ => {
@@ -2497,9 +2498,6 @@ impl<'tcx> dyn HirTyLowerer<'tcx> + '_ {
24972498
}
24982499
};
24992500
self.record_ty(expr.hir_id, c_ty, expr.span);
2500-
if let Some((id, span)) = neg {
2501-
self.record_ty(id, c_ty, span);
2502-
}
25032501
c
25042502
};
25052503

‎compiler/rustc_hir_pretty/src/lib.rs

+17-3
Original file line numberDiff line numberDiff line change
@@ -199,6 +199,7 @@ impl<'a> State<'a> {
199199
Node::OpaqueTy(o) => self.print_opaque_ty(o),
200200
Node::Pat(a) => self.print_pat(a),
201201
Node::PatField(a) => self.print_patfield(a),
202+
Node::PatLit(a) => self.print_pat_lit(a),
202203
Node::Arm(a) => self.print_arm(a),
203204
Node::Infer(_) => self.word("_"),
204205
Node::PreciseCapturingNonLifetimeArg(param) => self.print_ident(param.ident),
@@ -1847,6 +1848,19 @@ impl<'a> State<'a> {
18471848
}
18481849
}
18491850

1851+
fn print_pat_lit(&mut self, lit: &hir::PatLit<'_>) {
1852+
match &lit.kind {
1853+
hir::PatLitKind::Lit { lit, negated } => {
1854+
if *negated {
1855+
self.word("-");
1856+
}
1857+
self.print_literal(lit);
1858+
}
1859+
hir::PatLitKind::ConstBlock(c) => self.print_inline_const(c),
1860+
hir::PatLitKind::Path(qpath) => self.print_qpath(qpath, true),
1861+
}
1862+
}
1863+
18501864
fn print_pat(&mut self, pat: &hir::Pat<'_>) {
18511865
self.maybe_print_comment(pat.span.lo());
18521866
self.ann.pre(self, AnnNode::Pat(pat));
@@ -1964,17 +1978,17 @@ impl<'a> State<'a> {
19641978
self.pclose();
19651979
}
19661980
}
1967-
PatKind::Lit(e) => self.print_expr(e),
1981+
PatKind::Lit(e) => self.print_pat_lit(e),
19681982
PatKind::Range(begin, end, end_kind) => {
19691983
if let Some(expr) = begin {
1970-
self.print_expr(expr);
1984+
self.print_pat_lit(expr);
19711985
}
19721986
match end_kind {
19731987
RangeEnd::Included => self.word("..."),
19741988
RangeEnd::Excluded => self.word(".."),
19751989
}
19761990
if let Some(expr) = end {
1977-
self.print_expr(expr);
1991+
self.print_pat_lit(expr);
19781992
}
19791993
}
19801994
PatKind::Slice(before, slice, after) => {

0 commit comments

Comments
 (0)