Skip to content

Commit f42da01

Browse files
Port #[must_use] to new attribute parsing infrastructure
Signed-off-by: Jonathan Brouwer <[email protected]>
1 parent 8051f01 commit f42da01

File tree

20 files changed

+151
-69
lines changed

20 files changed

+151
-69
lines changed

compiler/rustc_attr_data_structures/src/attributes.rs

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -237,6 +237,13 @@ pub enum AttributeKind {
237237
/// Represents [`#[may_dangle]`](https://std-dev-guide.rust-lang.org/tricky/may-dangle.html).
238238
MayDangle(Span),
239239

240+
/// Represents `#[must_use]`.
241+
MustUse {
242+
span: Span,
243+
/// must_use can optionally have a reason: `#[must_use = "reason this must be used"]`
244+
reason: Option<Symbol>,
245+
},
246+
240247
/// Represents `#[optimize(size|speed)]`
241248
Optimize(OptimizeAttr, Span),
242249

compiler/rustc_attr_parsing/src/attributes/mod.rs

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -33,6 +33,7 @@ pub(crate) mod confusables;
3333
pub(crate) mod deprecation;
3434
pub(crate) mod inline;
3535
pub(crate) mod lint_helpers;
36+
pub(crate) mod must_use;
3637
pub(crate) mod repr;
3738
pub(crate) mod semantics;
3839
pub(crate) mod stability;
Lines changed: 40 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,40 @@
1+
use rustc_attr_data_structures::AttributeKind;
2+
use rustc_errors::DiagArgValue;
3+
use rustc_feature::{AttributeTemplate, template};
4+
use rustc_span::{Symbol, sym};
5+
6+
use crate::attributes::{AttributeOrder, OnDuplicate, SingleAttributeParser};
7+
use crate::context::{AcceptContext, Stage};
8+
use crate::parser::ArgParser;
9+
use crate::session_diagnostics;
10+
11+
pub(crate) struct MustUseParser;
12+
13+
impl<S: Stage> SingleAttributeParser<S> for MustUseParser {
14+
const PATH: &[Symbol] = &[sym::must_use];
15+
const ATTRIBUTE_ORDER: AttributeOrder = AttributeOrder::KeepLast;
16+
const ON_DUPLICATE: OnDuplicate<S> = OnDuplicate::WarnButFutureError;
17+
const TEMPLATE: AttributeTemplate = template!(Word, NameValueStr: "reason");
18+
19+
fn convert(cx: &mut AcceptContext<'_, '_, S>, args: &ArgParser<'_>) -> Option<AttributeKind> {
20+
Some(AttributeKind::MustUse {
21+
span: cx.attr_span,
22+
reason: match args {
23+
ArgParser::NoArgs => None,
24+
ArgParser::NameValue(name_value) => name_value.value_as_str(),
25+
ArgParser::List(_) => {
26+
let suggestions =
27+
<Self as SingleAttributeParser<S>>::TEMPLATE.suggestions(false, "must_use");
28+
cx.emit_err(session_diagnostics::MustUseIllFormedAttributeInput {
29+
num_suggestions: suggestions.len(),
30+
suggestions: DiagArgValue::StrListSepByAnd(
31+
suggestions.into_iter().map(|s| format!("`{s}`").into()).collect(),
32+
),
33+
span: cx.attr_span,
34+
});
35+
return None;
36+
}
37+
},
38+
})
39+
}
40+
}

compiler/rustc_attr_parsing/src/context.rs

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,7 @@ use crate::attributes::confusables::ConfusablesParser;
2020
use crate::attributes::deprecation::DeprecationParser;
2121
use crate::attributes::inline::{InlineParser, RustcForceInlineParser};
2222
use crate::attributes::lint_helpers::AsPtrParser;
23+
use crate::attributes::must_use::MustUseParser;
2324
use crate::attributes::repr::{AlignParser, ReprParser};
2425
use crate::attributes::semantics::MayDangleParser;
2526
use crate::attributes::stability::{
@@ -112,6 +113,7 @@ attribute_parsers!(
112113
Single<DeprecationParser>,
113114
Single<InlineParser>,
114115
Single<MayDangleParser>,
116+
Single<MustUseParser>,
115117
Single<OptimizeParser>,
116118
Single<RustcForceInlineParser>,
117119
Single<TransparencyParser>,

compiler/rustc_attr_parsing/src/session_diagnostics.rs

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -436,6 +436,15 @@ pub(crate) struct IllFormedAttributeInput {
436436
pub suggestions: DiagArgValue,
437437
}
438438

439+
#[derive(Diagnostic)]
440+
#[diag(attr_parsing_ill_formed_attribute_input)]
441+
pub(crate) struct MustUseIllFormedAttributeInput {
442+
#[primary_span]
443+
pub span: Span,
444+
pub num_suggestions: usize,
445+
pub suggestions: DiagArgValue,
446+
}
447+
439448
#[derive(Diagnostic)]
440449
#[diag(attr_parsing_stability_outside_std, code = E0734)]
441450
pub(crate) struct StabilityOutsideStd {

compiler/rustc_lint/src/unused.rs

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,7 @@ use std::iter;
22

33
use rustc_ast::util::{classify, parser};
44
use rustc_ast::{self as ast, ExprKind, HasAttrs as _, StmtKind};
5+
use rustc_attr_data_structures::{AttributeKind, find_attr};
56
use rustc_errors::{MultiSpan, pluralize};
67
use rustc_hir::def::{DefKind, Res};
78
use rustc_hir::def_id::DefId;
@@ -368,10 +369,12 @@ impl<'tcx> LateLintPass<'tcx> for UnusedResults {
368369
}
369370

370371
fn is_def_must_use(cx: &LateContext<'_>, def_id: DefId, span: Span) -> Option<MustUsePath> {
371-
if let Some(attr) = cx.tcx.get_attr(def_id, sym::must_use) {
372+
if let Some(reason) = find_attr!(
373+
cx.tcx.get_all_attrs(def_id),
374+
AttributeKind::MustUse { reason, .. } => reason
375+
) {
372376
// check for #[must_use = "..."]
373-
let reason = attr.value_str();
374-
Some(MustUsePath::Def(span, def_id, reason))
377+
Some(MustUsePath::Def(span, def_id, *reason))
375378
} else {
376379
None
377380
}

compiler/rustc_parse/src/validate_attr.rs

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -293,6 +293,7 @@ fn emit_malformed_attribute(
293293
| sym::deprecated
294294
| sym::optimize
295295
| sym::cold
296+
| sym::must_use
296297
) {
297298
return;
298299
}

compiler/rustc_passes/src/check_attr.rs

Lines changed: 7 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -166,6 +166,9 @@ impl<'tcx> CheckAttrVisitor<'tcx> {
166166
&Attribute::Parsed(AttributeKind::MayDangle(attr_span)) => {
167167
self.check_may_dangle(hir_id, attr_span)
168168
}
169+
Attribute::Parsed(AttributeKind::MustUse { span, .. }) => {
170+
self.check_must_use(hir_id, *span, target)
171+
}
169172
Attribute::Unparsed(_) => {
170173
match attr.path().as_slice() {
171174
[sym::diagnostic, sym::do_not_recommend, ..] => {
@@ -238,7 +241,6 @@ impl<'tcx> CheckAttrVisitor<'tcx> {
238241
| [sym::const_trait, ..] => self.check_must_be_applied_to_trait(attr, span, target),
239242
[sym::collapse_debuginfo, ..] => self.check_collapse_debuginfo(attr, span, target),
240243
[sym::must_not_suspend, ..] => self.check_must_not_suspend(attr, span, target),
241-
[sym::must_use, ..] => self.check_must_use(hir_id, attr, target),
242244
[sym::rustc_pass_by_value, ..] => self.check_pass_by_value(attr, span, target),
243245
[sym::rustc_allow_incoherent_impl, ..] => {
244246
self.check_allow_incoherent_impl(attr, span, target)
@@ -690,7 +692,8 @@ impl<'tcx> CheckAttrVisitor<'tcx> {
690692
AttributeKind::Deprecation { .. }
691693
| AttributeKind::Repr { .. }
692694
| AttributeKind::Align { .. }
693-
| AttributeKind::Cold(..),
695+
| AttributeKind::Cold(..)
696+
| AttributeKind::MustUse { .. },
694697
) => {
695698
continue;
696699
}
@@ -1565,7 +1568,7 @@ impl<'tcx> CheckAttrVisitor<'tcx> {
15651568
}
15661569

15671570
/// Warns against some misuses of `#[must_use]`
1568-
fn check_must_use(&self, hir_id: HirId, attr: &Attribute, target: Target) {
1571+
fn check_must_use(&self, hir_id: HirId, attr_span: Span, target: Target) {
15691572
if matches!(
15701573
target,
15711574
Target::Fn
@@ -1605,7 +1608,7 @@ impl<'tcx> CheckAttrVisitor<'tcx> {
16051608
self.tcx.emit_node_span_lint(
16061609
UNUSED_ATTRIBUTES,
16071610
hir_id,
1608-
attr.span(),
1611+
attr_span,
16091612
errors::MustUseNoEffect { article, target },
16101613
);
16111614
}

src/rustdoc-json-types/lib.rs

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -37,8 +37,8 @@ pub type FxHashMap<K, V> = HashMap<K, V>; // re-export for use in src/librustdoc
3737
// will instead cause conflicts. See #94591 for more. (This paragraph and the "Latest feature" line
3838
// are deliberately not in a doc comment, because they need not be in public docs.)
3939
//
40-
// Latest feature: improve handling of generic args
41-
pub const FORMAT_VERSION: u32 = 51;
40+
// Latest feature: Pretty printing of must_use attributes changed
41+
pub const FORMAT_VERSION: u32 = 52;
4242

4343
/// The root of the emitted JSON blob.
4444
///

src/tools/clippy/clippy_lints/src/functions/must_use.rs

Lines changed: 17 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -15,14 +15,17 @@ use clippy_utils::ty::is_must_use_ty;
1515
use clippy_utils::visitors::for_each_expr_without_closures;
1616
use clippy_utils::{return_ty, trait_ref_of_method};
1717
use rustc_trait_selection::error_reporting::InferCtxtErrorExt;
18+
use rustc_attr_data_structures::AttributeKind;
19+
use rustc_span::Symbol;
20+
use rustc_attr_data_structures::find_attr;
1821

1922
use core::ops::ControlFlow;
2023

2124
use super::{DOUBLE_MUST_USE, MUST_USE_CANDIDATE, MUST_USE_UNIT};
2225

2326
pub(super) fn check_item<'tcx>(cx: &LateContext<'tcx>, item: &'tcx hir::Item<'_>) {
2427
let attrs = cx.tcx.hir_attrs(item.hir_id());
25-
let attr = cx.tcx.get_attr(item.owner_id, sym::must_use);
28+
let attr = find_attr!(cx.tcx.hir_attrs(item.hir_id()), AttributeKind::MustUse { span, reason } => (span, reason));
2629
if let hir::ItemKind::Fn {
2730
ref sig,
2831
body: ref body_id,
@@ -31,8 +34,8 @@ pub(super) fn check_item<'tcx>(cx: &LateContext<'tcx>, item: &'tcx hir::Item<'_>
3134
{
3235
let is_public = cx.effective_visibilities.is_exported(item.owner_id.def_id);
3336
let fn_header_span = item.span.with_hi(sig.decl.output.span().hi());
34-
if let Some(attr) = attr {
35-
check_needless_must_use(cx, sig.decl, item.owner_id, item.span, fn_header_span, attr, attrs, sig);
37+
if let Some((attr_span, reason)) = attr {
38+
check_needless_must_use(cx, sig.decl, item.owner_id, item.span, fn_header_span, *attr_span, *reason, attrs, sig);
3639
} else if is_public && !is_proc_macro(attrs) && !attrs.iter().any(|a| a.has_name(sym::no_mangle)) {
3740
check_must_use_candidate(
3841
cx,
@@ -52,9 +55,9 @@ pub(super) fn check_impl_item<'tcx>(cx: &LateContext<'tcx>, item: &'tcx hir::Imp
5255
let is_public = cx.effective_visibilities.is_exported(item.owner_id.def_id);
5356
let fn_header_span = item.span.with_hi(sig.decl.output.span().hi());
5457
let attrs = cx.tcx.hir_attrs(item.hir_id());
55-
let attr = cx.tcx.get_attr(item.owner_id, sym::must_use);
56-
if let Some(attr) = attr {
57-
check_needless_must_use(cx, sig.decl, item.owner_id, item.span, fn_header_span, attr, attrs, sig);
58+
let attr = find_attr!(cx.tcx.hir_attrs(item.hir_id()), AttributeKind::MustUse { span, reason } => (span, reason));
59+
if let Some((attr_span, reason)) = attr {
60+
check_needless_must_use(cx, sig.decl, item.owner_id, item.span, fn_header_span, *attr_span, *reason, attrs, sig);
5861
} else if is_public && !is_proc_macro(attrs) && trait_ref_of_method(cx, item.owner_id).is_none() {
5962
check_must_use_candidate(
6063
cx,
@@ -75,9 +78,9 @@ pub(super) fn check_trait_item<'tcx>(cx: &LateContext<'tcx>, item: &'tcx hir::Tr
7578
let fn_header_span = item.span.with_hi(sig.decl.output.span().hi());
7679

7780
let attrs = cx.tcx.hir_attrs(item.hir_id());
78-
let attr = cx.tcx.get_attr(item.owner_id, sym::must_use);
79-
if let Some(attr) = attr {
80-
check_needless_must_use(cx, sig.decl, item.owner_id, item.span, fn_header_span, attr, attrs, sig);
81+
let attr = find_attr!(cx.tcx.hir_attrs(item.hir_id()), AttributeKind::MustUse { span, reason } => (span, reason));
82+
if let Some((attr_span, reason)) = attr {
83+
check_needless_must_use(cx, sig.decl, item.owner_id, item.span, fn_header_span, *attr_span, *reason, attrs, sig);
8184
} else if let hir::TraitFn::Provided(eid) = *eid {
8285
let body = cx.tcx.hir_body(eid);
8386
if attr.is_none() && is_public && !is_proc_macro(attrs) {
@@ -103,7 +106,8 @@ fn check_needless_must_use(
103106
item_id: hir::OwnerId,
104107
item_span: Span,
105108
fn_header_span: Span,
106-
attr: &Attribute,
109+
attr_span: Span,
110+
reason: Option<Symbol>,
107111
attrs: &[Attribute],
108112
sig: &FnSig<'_>,
109113
) {
@@ -119,7 +123,7 @@ fn check_needless_must_use(
119123
"this unit-returning function has a `#[must_use]` attribute",
120124
|diag| {
121125
diag.span_suggestion(
122-
attr.span(),
126+
attr_span,
123127
"remove the attribute",
124128
"",
125129
Applicability::MachineApplicable,
@@ -137,11 +141,11 @@ fn check_needless_must_use(
137141
MUST_USE_UNIT,
138142
fn_header_span,
139143
"this unit-returning function has a `#[must_use]` attribute",
140-
Some(attr.span()),
144+
Some(attr_span),
141145
"remove `must_use`",
142146
);
143147
}
144-
} else if attr.value_str().is_none() && is_must_use_ty(cx, return_ty(cx, item_id)) {
148+
} else if reason.is_none() && is_must_use_ty(cx, return_ty(cx, item_id)) {
145149
// Ignore async functions unless Future::Output type is a must_use type
146150
if sig.header.is_async() {
147151
let infcx = cx.tcx.infer_ctxt().build(cx.typing_mode());

0 commit comments

Comments
 (0)