Skip to content

Commit f7ad406

Browse files
JonathanBrouwerJonathanBrouwer
committed
Port #[should_panic] to the new attribute parsing infrastructure
Signed-off-by: Jonathan Brouwer <[email protected]>
1 parent 8fb40f7 commit f7ad406

File tree

7 files changed

+78
-38
lines changed

7 files changed

+78
-38
lines changed

‎compiler/rustc_attr_parsing/src/attributes/test_attrs.rs

Lines changed: 52 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -44,3 +44,55 @@ impl<S: Stage> SingleAttributeParser<S> for IgnoreParser {
4444
})
4545
}
4646
}
47+
48+
pub(crate) struct ShouldPanicParser;
49+
50+
impl<S: Stage> SingleAttributeParser<S> for ShouldPanicParser {
51+
const PATH: &[Symbol] = &[sym::should_panic];
52+
const ATTRIBUTE_ORDER: AttributeOrder = AttributeOrder::KeepOutermost;
53+
const ON_DUPLICATE: OnDuplicate<S> = OnDuplicate::WarnButFutureError;
54+
const TEMPLATE: AttributeTemplate =
55+
template!(Word, List: r#"expected = "reason""#, NameValueStr: "reason");
56+
57+
fn convert(cx: &mut AcceptContext<'_, '_, S>, args: &ArgParser<'_>) -> Option<AttributeKind> {
58+
Some(AttributeKind::ShouldPanic {
59+
span: cx.attr_span,
60+
reason: match args {
61+
ArgParser::NoArgs => None,
62+
ArgParser::NameValue(name_value) => {
63+
let Some(str_value) = name_value.value_as_str() else {
64+
cx.expected_string_literal(
65+
name_value.value_span,
66+
Some(name_value.value_as_lit()),
67+
);
68+
return None;
69+
};
70+
Some(str_value)
71+
}
72+
ArgParser::List(list) => {
73+
let Some(single) = list.single() else {
74+
cx.expected_single_argument(list.span);
75+
return None;
76+
};
77+
let Some(single) = single.meta_item() else {
78+
cx.expected_name_value(single.span(), Some(sym::expected));
79+
return None;
80+
};
81+
if !single.path().word_is(sym::expected) {
82+
cx.expected_specific_argument_strings(list.span, vec!["expected"]);
83+
return None;
84+
}
85+
let Some(nv) = single.args().name_value() else {
86+
cx.expected_name_value(single.span(), Some(sym::expected));
87+
return None;
88+
};
89+
let Some(expected) = nv.value_as_str() else {
90+
cx.expected_string_literal(nv.value_span, Some(nv.value_as_lit()));
91+
return None;
92+
};
93+
Some(expected)
94+
}
95+
},
96+
})
97+
}
98+
}

‎compiler/rustc_attr_parsing/src/context.rs

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -49,7 +49,7 @@ use crate::attributes::semantics::MayDangleParser;
4949
use crate::attributes::stability::{
5050
BodyStabilityParser, ConstStabilityIndirectParser, ConstStabilityParser, StabilityParser,
5151
};
52-
use crate::attributes::test_attrs::IgnoreParser;
52+
use crate::attributes::test_attrs::{IgnoreParser, ShouldPanicParser};
5353
use crate::attributes::traits::{
5454
AllowIncoherentImplParser, CoherenceIsCoreParser, CoinductiveParser, ConstTraitParser,
5555
DenyExplicitImplParser, DoNotImplementViaObjectParser, FundamentalParser, MarkerParser,
@@ -173,6 +173,7 @@ attribute_parsers!(
173173
Single<RustcLayoutScalarValidRangeEnd>,
174174
Single<RustcLayoutScalarValidRangeStart>,
175175
Single<RustcObjectLifetimeDefaultParser>,
176+
Single<ShouldPanicParser>,
176177
Single<SkipDuringMethodDispatchParser>,
177178
Single<TransparencyParser>,
178179
Single<WithoutArgs<AllowIncoherentImplParser>>,

‎compiler/rustc_builtin_macros/src/test.rs

Lines changed: 17 additions & 34 deletions
Original file line numberDiff line numberDiff line change
@@ -5,10 +5,13 @@ use std::assert_matches::assert_matches;
55
use std::iter;
66

77
use rustc_ast::ptr::P;
8-
use rustc_ast::{self as ast, GenericParamKind, attr, join_path_idents};
8+
use rustc_ast::{self as ast, GenericParamKind, HasNodeId, attr, join_path_idents};
99
use rustc_ast_pretty::pprust;
10+
use rustc_attr_parsing::AttributeParser;
1011
use rustc_errors::{Applicability, Diag, Level};
1112
use rustc_expand::base::*;
13+
use rustc_hir::Attribute;
14+
use rustc_hir::attrs::AttributeKind;
1215
use rustc_span::{ErrorGuaranteed, FileNameDisplayPreference, Ident, Span, Symbol, sym};
1316
use thin_vec::{ThinVec, thin_vec};
1417
use tracing::debug;
@@ -473,39 +476,19 @@ fn should_ignore_message(i: &ast::Item) -> Option<Symbol> {
473476
}
474477

475478
fn should_panic(cx: &ExtCtxt<'_>, i: &ast::Item) -> ShouldPanic {
476-
match attr::find_by_name(&i.attrs, sym::should_panic) {
477-
Some(attr) => {
478-
match attr.meta_item_list() {
479-
// Handle #[should_panic(expected = "foo")]
480-
Some(list) => {
481-
let msg = list
482-
.iter()
483-
.find(|mi| mi.has_name(sym::expected))
484-
.and_then(|mi| mi.meta_item())
485-
.and_then(|mi| mi.value_str());
486-
if list.len() != 1 || msg.is_none() {
487-
cx.dcx()
488-
.struct_span_warn(
489-
attr.span,
490-
"argument must be of the form: \
491-
`expected = \"error message\"`",
492-
)
493-
.with_note(
494-
"errors in this attribute were erroneously \
495-
allowed and will become a hard error in a \
496-
future release",
497-
)
498-
.emit();
499-
ShouldPanic::Yes(None)
500-
} else {
501-
ShouldPanic::Yes(msg)
502-
}
503-
}
504-
// Handle #[should_panic] and #[should_panic = "expected"]
505-
None => ShouldPanic::Yes(attr.value_str()),
506-
}
507-
}
508-
None => ShouldPanic::No,
479+
if let Some(Attribute::Parsed(AttributeKind::ShouldPanic { reason, .. })) =
480+
AttributeParser::parse_limited(
481+
cx.sess,
482+
&i.attrs,
483+
sym::should_panic,
484+
i.span,
485+
i.node_id(),
486+
None,
487+
)
488+
{
489+
ShouldPanic::Yes(reason)
490+
} else {
491+
ShouldPanic::No
509492
}
510493
}
511494

‎compiler/rustc_hir/src/attrs/data_structures.rs

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -433,6 +433,9 @@ pub enum AttributeKind {
433433
/// Represents `#[rustc_object_lifetime_default]`.
434434
RustcObjectLifetimeDefault,
435435

436+
/// Represents `#[should_panic]`
437+
ShouldPanic { reason: Option<Symbol>, span: Span },
438+
436439
/// Represents `#[rustc_skip_during_method_dispatch]`.
437440
SkipDuringMethodDispatch { array: bool, boxed_slice: bool, span: Span },
438441

‎compiler/rustc_hir/src/attrs/encode_cross_crate.rs

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -69,6 +69,7 @@ impl AttributeKind {
6969
RustcLayoutScalarValidRangeEnd(..) => Yes,
7070
RustcLayoutScalarValidRangeStart(..) => Yes,
7171
RustcObjectLifetimeDefault => No,
72+
ShouldPanic { .. } => No,
7273
SkipDuringMethodDispatch { .. } => No,
7374
SpecializationTrait(..) => No,
7475
Stability { .. } => Yes,

‎compiler/rustc_hir/src/hir.rs

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1308,6 +1308,7 @@ impl AttributeExt for Attribute {
13081308
Attribute::Parsed(AttributeKind::MacroUse { span, .. }) => *span,
13091309
Attribute::Parsed(AttributeKind::MayDangle(span)) => *span,
13101310
Attribute::Parsed(AttributeKind::Ignore { span, .. }) => *span,
1311+
Attribute::Parsed(AttributeKind::ShouldPanic { span, .. }) => *span,
13111312
Attribute::Parsed(AttributeKind::AutomaticallyDerived(span)) => *span,
13121313
a => panic!("can't get the span of an arbitrary parsed attribute: {a:?}"),
13131314
}

‎compiler/rustc_passes/src/check_attr.rs

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -315,6 +315,8 @@ impl<'tcx> CheckAttrVisitor<'tcx> {
315315
Attribute::Parsed(AttributeKind::Used { span: attr_span, .. }) => {
316316
self.check_used(*attr_span, target, span);
317317
}
318+
Attribute::Parsed(AttributeKind::ShouldPanic { span: attr_span, .. }) => self
319+
.check_generic_attr(hir_id, sym::should_panic, *attr_span, target, Target::Fn),
318320
&Attribute::Parsed(AttributeKind::PassByValue(attr_span)) => {
319321
self.check_pass_by_value(attr_span, span, target)
320322
}
@@ -384,9 +386,6 @@ impl<'tcx> CheckAttrVisitor<'tcx> {
384386
[sym::link, ..] => self.check_link(hir_id, attr, span, target),
385387
[sym::path, ..] => self.check_generic_attr_unparsed(hir_id, attr, target, Target::Mod),
386388
[sym::macro_export, ..] => self.check_macro_export(hir_id, attr, target),
387-
[sym::should_panic, ..] => {
388-
self.check_generic_attr_unparsed(hir_id, attr, target, Target::Fn)
389-
}
390389
[sym::autodiff_forward, ..] | [sym::autodiff_reverse, ..] => {
391390
self.check_autodiff(hir_id, attr, span, target)
392391
}

0 commit comments

Comments
 (0)