Skip to content

Commit ca3966a

Browse files
scrabshascrabsha
committed
Port #[coverage] to the new attribute system
1 parent ad635e5 commit ca3966a

File tree

15 files changed

+358
-389
lines changed

15 files changed

+358
-389
lines changed

‎compiler/rustc_attr_data_structures/src/attributes.rs

Lines changed: 19 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -109,6 +109,22 @@ pub enum DeprecatedSince {
109109
Err,
110110
}
111111

112+
#[derive(
113+
Copy,
114+
Debug,
115+
Eq,
116+
PartialEq,
117+
Encodable,
118+
Decodable,
119+
Clone,
120+
HashStable_Generic,
121+
PrintAttribute
122+
)]
123+
pub enum CoverageStatus {
124+
On,
125+
Off,
126+
}
127+
112128
impl Deprecation {
113129
/// Whether an item marked with #[deprecated(since = "X")] is currently
114130
/// deprecated (i.e., whether X is not greater than the current rustc
@@ -249,6 +265,9 @@ pub enum AttributeKind {
249265
/// Represents `#[const_trait]`.
250266
ConstTrait(Span),
251267

268+
/// Represents `#[coverage]`.
269+
Coverage(Span, CoverageStatus),
270+
252271
///Represents `#[rustc_deny_explicit_impl]`.
253272
DenyExplicitImpl(Span),
254273

‎compiler/rustc_attr_data_structures/src/encode_cross_crate.rs

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -28,6 +28,7 @@ impl AttributeKind {
2828
ConstStability { .. } => Yes,
2929
ConstStabilityIndirect => No,
3030
ConstTrait(..) => No,
31+
Coverage(..) => No,
3132
DenyExplicitImpl(..) => No,
3233
Deprecation { .. } => Yes,
3334
DoNotImplementViaObject(..) => No,

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

Lines changed: 40 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
use rustc_attr_data_structures::{AttributeKind, OptimizeAttr, UsedBy};
1+
use rustc_attr_data_structures::{AttributeKind, CoverageStatus, OptimizeAttr, UsedBy};
22
use rustc_feature::{AttributeTemplate, template};
33
use rustc_session::parse::feature_err;
44
use rustc_span::{Span, Symbol, sym};
@@ -52,6 +52,45 @@ impl<S: Stage> NoArgsAttributeParser<S> for ColdParser {
5252
const CREATE: fn(Span) -> AttributeKind = AttributeKind::Cold;
5353
}
5454

55+
pub(crate) struct CoverageParser;
56+
57+
impl<S: Stage> SingleAttributeParser<S> for CoverageParser {
58+
const PATH: &[Symbol] = &[sym::coverage];
59+
const ATTRIBUTE_ORDER: AttributeOrder = AttributeOrder::KeepOutermost;
60+
const ON_DUPLICATE: OnDuplicate<S> = OnDuplicate::Error;
61+
const TEMPLATE: AttributeTemplate = template!(OneOf: &[sym::off, sym::on]);
62+
63+
fn convert(cx: &mut AcceptContext<'_, '_, S>, args: &ArgParser<'_>) -> Option<AttributeKind> {
64+
let Some(args) = args.list() else {
65+
cx.expected_specific_argument_and_list(cx.attr_span, vec!["on", "off"]);
66+
return None;
67+
};
68+
69+
let Some(arg) = args.single() else {
70+
cx.expected_single_argument(args.span);
71+
return None;
72+
};
73+
74+
let fail_incorrect_argument = |span| cx.expected_specific_argument(span, vec!["on", "off"]);
75+
76+
let Some(arg) = arg.meta_item() else {
77+
fail_incorrect_argument(args.span);
78+
return None;
79+
};
80+
81+
let status = match arg.path().word_sym() {
82+
Some(sym::off) => CoverageStatus::Off,
83+
Some(sym::on) => CoverageStatus::On,
84+
None | Some(_) => {
85+
fail_incorrect_argument(arg.span());
86+
return None;
87+
}
88+
};
89+
90+
Some(AttributeKind::Coverage(cx.attr_span, status))
91+
}
92+
}
93+
5594
pub(crate) struct ExportNameParser;
5695

5796
impl<S: Stage> SingleAttributeParser<S> for ExportNameParser {

‎compiler/rustc_attr_parsing/src/context.rs

Lines changed: 23 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -16,8 +16,8 @@ use rustc_span::{DUMMY_SP, ErrorGuaranteed, Span, Symbol, sym};
1616

1717
use crate::attributes::allow_unstable::{AllowConstFnUnstableParser, AllowInternalUnstableParser};
1818
use crate::attributes::codegen_attrs::{
19-
ColdParser, ExportNameParser, NakedParser, NoMangleParser, OptimizeParser, TargetFeatureParser,
20-
TrackCallerParser, UsedParser,
19+
ColdParser, CoverageParser, ExportNameParser, NakedParser, NoMangleParser, OptimizeParser,
20+
TargetFeatureParser, TrackCallerParser, UsedParser,
2121
};
2222
use crate::attributes::confusables::ConfusablesParser;
2323
use crate::attributes::deprecation::DeprecationParser;
@@ -137,6 +137,7 @@ attribute_parsers!(
137137
// tidy-alphabetical-end
138138

139139
// tidy-alphabetical-start
140+
Single<CoverageParser>,
140141
Single<DeprecationParser>,
141142
Single<DummyParser>,
142143
Single<ExportNameParser>,
@@ -426,6 +427,25 @@ impl<'f, 'sess: 'f, S: Stage> AcceptContext<'f, 'sess, S> {
426427
reason: AttributeParseErrorReason::ExpectedSpecificArgument {
427428
possibilities,
428429
strings: false,
430+
list: false,
431+
},
432+
})
433+
}
434+
435+
pub(crate) fn expected_specific_argument_and_list(
436+
&self,
437+
span: Span,
438+
possibilities: Vec<&'static str>,
439+
) -> ErrorGuaranteed {
440+
self.emit_err(AttributeParseError {
441+
span,
442+
attr_span: self.attr_span,
443+
template: self.template.clone(),
444+
attribute: self.attr_path.clone(),
445+
reason: AttributeParseErrorReason::ExpectedSpecificArgument {
446+
possibilities,
447+
strings: false,
448+
list: true,
429449
},
430450
})
431451
}
@@ -443,6 +463,7 @@ impl<'f, 'sess: 'f, S: Stage> AcceptContext<'f, 'sess, S> {
443463
reason: AttributeParseErrorReason::ExpectedSpecificArgument {
444464
possibilities,
445465
strings: true,
466+
list: false,
446467
},
447468
})
448469
}

‎compiler/rustc_attr_parsing/src/session_diagnostics.rs

Lines changed: 46 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -516,15 +516,22 @@ pub(crate) struct NakedFunctionIncompatibleAttribute {
516516

517517
pub(crate) enum AttributeParseErrorReason {
518518
ExpectedNoArgs,
519-
ExpectedStringLiteral { byte_string: Option<Span> },
519+
ExpectedStringLiteral {
520+
byte_string: Option<Span>,
521+
},
520522
ExpectedIntegerLiteral,
521523
ExpectedAtLeastOneArgument,
522524
ExpectedSingleArgument,
523525
ExpectedList,
524526
UnexpectedLiteral,
525527
ExpectedNameValue(Option<Symbol>),
526528
DuplicateKey(Symbol),
527-
ExpectedSpecificArgument { possibilities: Vec<&'static str>, strings: bool },
529+
ExpectedSpecificArgument {
530+
possibilities: Vec<&'static str>,
531+
strings: bool,
532+
/// Should we tell the user to write a list when they didn't?
533+
list: bool,
534+
},
528535
}
529536

530537
pub(crate) struct AttributeParseError {
@@ -592,7 +599,11 @@ impl<'a, G: EmissionGuarantee> Diagnostic<'a, G> for AttributeParseError {
592599
format!("expected this to be of the form `{name} = \"...\"`"),
593600
);
594601
}
595-
AttributeParseErrorReason::ExpectedSpecificArgument { possibilities, strings } => {
602+
AttributeParseErrorReason::ExpectedSpecificArgument {
603+
possibilities,
604+
strings,
605+
list: false,
606+
} => {
596607
let quote = if strings { '"' } else { '`' };
597608
match possibilities.as_slice() {
598609
&[] => {}
@@ -618,6 +629,38 @@ impl<'a, G: EmissionGuarantee> Diagnostic<'a, G> for AttributeParseError {
618629
}
619630
}
620631
}
632+
AttributeParseErrorReason::ExpectedSpecificArgument {
633+
possibilities,
634+
strings,
635+
list: true,
636+
} => {
637+
let quote = if strings { '"' } else { '`' };
638+
match possibilities.as_slice() {
639+
&[] => {}
640+
&[x] => {
641+
diag.span_label(
642+
self.span,
643+
format!(
644+
"this attribute is only valid with {quote}{x}{quote} as an argument"
645+
),
646+
);
647+
}
648+
[first, second] => {
649+
diag.span_label(self.span, format!("this attribute is only valid with either {quote}{first}{quote} or {quote}{second}{quote} as an argument"));
650+
}
651+
[first @ .., second_to_last, last] => {
652+
let mut res = String::new();
653+
for i in first {
654+
res.push_str(&format!("{quote}{i}{quote}, "));
655+
}
656+
res.push_str(&format!(
657+
"{quote}{second_to_last}{quote} or {quote}{last}{quote}"
658+
));
659+
660+
diag.span_label(self.span, format!("this attribute is only valid with one of the following arguments: {res}"));
661+
}
662+
}
663+
}
621664
}
622665

623666
let suggestions = self.template.suggestions(false, &name);

‎compiler/rustc_mir_transform/src/coverage/query.rs

Lines changed: 13 additions & 19 deletions
Original file line numberDiff line numberDiff line change
@@ -1,11 +1,11 @@
1+
use rustc_attr_data_structures::{AttributeKind, CoverageStatus, find_attr};
12
use rustc_index::bit_set::DenseBitSet;
23
use rustc_middle::middle::codegen_fn_attrs::CodegenFnAttrFlags;
34
use rustc_middle::mir::coverage::{BasicCoverageBlock, CoverageIdsInfo, CoverageKind, MappingKind};
45
use rustc_middle::mir::{Body, Statement, StatementKind};
56
use rustc_middle::ty::{self, TyCtxt};
67
use rustc_middle::util::Providers;
78
use rustc_span::def_id::LocalDefId;
8-
use rustc_span::sym;
99
use tracing::trace;
1010

1111
use crate::coverage::counters::node_flow::make_node_counters;
@@ -58,26 +58,20 @@ fn is_eligible_for_coverage(tcx: TyCtxt<'_>, def_id: LocalDefId) -> bool {
5858
/// Query implementation for `coverage_attr_on`.
5959
fn coverage_attr_on(tcx: TyCtxt<'_>, def_id: LocalDefId) -> bool {
6060
// Check for annotations directly on this def.
61-
if let Some(attr) = tcx.get_attr(def_id, sym::coverage) {
62-
match attr.meta_item_list().as_deref() {
63-
Some([item]) if item.has_name(sym::off) => return false,
64-
Some([item]) if item.has_name(sym::on) => return true,
65-
Some(_) | None => {
66-
// Other possibilities should have been rejected by `rustc_parse::validate_attr`.
67-
// Use `span_delayed_bug` to avoid an ICE in failing builds (#127880).
68-
tcx.dcx().span_delayed_bug(attr.span(), "unexpected value of coverage attribute");
69-
}
61+
if let Some(coverage_status) =
62+
find_attr!(tcx.get_all_attrs(def_id), AttributeKind::Coverage(_, status) => status)
63+
{
64+
*coverage_status == CoverageStatus::On
65+
} else {
66+
match tcx.opt_local_parent(def_id) {
67+
// Check the parent def (and so on recursively) until we find an
68+
// enclosing attribute or reach the crate root.
69+
Some(parent) => tcx.coverage_attr_on(parent),
70+
// We reached the crate root without seeing a coverage attribute, so
71+
// allow coverage instrumentation by default.
72+
None => true,
7073
}
7174
}
72-
73-
match tcx.opt_local_parent(def_id) {
74-
// Check the parent def (and so on recursively) until we find an
75-
// enclosing attribute or reach the crate root.
76-
Some(parent) => tcx.coverage_attr_on(parent),
77-
// We reached the crate root without seeing a coverage attribute, so
78-
// allow coverage instrumentation by default.
79-
None => true,
80-
}
8175
}
8276

8377
/// Query implementation for `coverage_ids_info`.

‎compiler/rustc_parse/src/validate_attr.rs

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -317,6 +317,7 @@ pub fn check_builtin_meta_item(
317317
| sym::rustc_layout_scalar_valid_range_end
318318
| sym::no_implicit_prelude
319319
| sym::automatically_derived
320+
| sym::coverage
320321
) {
321322
return;
322323
}

‎compiler/rustc_passes/src/check_attr.rs

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -279,6 +279,9 @@ impl<'tcx> CheckAttrVisitor<'tcx> {
279279
&Attribute::Parsed(AttributeKind::StdInternalSymbol(attr_span)) => {
280280
self.check_rustc_std_internal_symbol(attr_span, span, target)
281281
}
282+
&Attribute::Parsed(AttributeKind::Coverage(attr_span, _)) => {
283+
self.check_coverage(attr_span, span, target)
284+
}
282285
Attribute::Unparsed(attr_item) => {
283286
style = Some(attr_item.style);
284287
match attr.path().as_slice() {
@@ -288,7 +291,6 @@ impl<'tcx> CheckAttrVisitor<'tcx> {
288291
[sym::diagnostic, sym::on_unimplemented, ..] => {
289292
self.check_diagnostic_on_unimplemented(attr.span(), hir_id, target)
290293
}
291-
[sym::coverage, ..] => self.check_coverage(attr, span, target),
292294
[sym::no_sanitize, ..] => {
293295
self.check_no_sanitize(attr, span, target)
294296
}
@@ -580,7 +582,7 @@ impl<'tcx> CheckAttrVisitor<'tcx> {
580582

581583
/// Checks that `#[coverage(..)]` is applied to a function/closure/method,
582584
/// or to an impl block or module.
583-
fn check_coverage(&self, attr: &Attribute, target_span: Span, target: Target) {
585+
fn check_coverage(&self, attr_span: Span, target_span: Span, target: Target) {
584586
let mut not_fn_impl_mod = None;
585587
let mut no_body = None;
586588

@@ -603,7 +605,7 @@ impl<'tcx> CheckAttrVisitor<'tcx> {
603605
}
604606

605607
self.dcx().emit_err(errors::CoverageAttributeNotAllowed {
606-
attr_span: attr.span(),
608+
attr_span,
607609
not_fn_impl_mod,
608610
no_body,
609611
help: (),

‎tests/ui/attributes/malformed-attrs.stderr

Lines changed: 13 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -40,19 +40,6 @@ error: malformed `crate_name` attribute input
4040
LL | #[crate_name]
4141
| ^^^^^^^^^^^^^ help: must be of the form: `#[crate_name = "name"]`
4242

43-
error: malformed `coverage` attribute input
44-
--> $DIR/malformed-attrs.rs:90:1
45-
|
46-
LL | #[coverage]
47-
| ^^^^^^^^^^^
48-
|
49-
help: the following are the possible correct uses
50-
|
51-
LL | #[coverage(off)]
52-
| +++++
53-
LL | #[coverage(on)]
54-
| ++++
55-
5643
error: malformed `no_sanitize` attribute input
5744
--> $DIR/malformed-attrs.rs:92:1
5845
|
@@ -460,6 +447,19 @@ error[E0539]: malformed `link_section` attribute input
460447
LL | #[link_section]
461448
| ^^^^^^^^^^^^^^^ help: must be of the form: `#[link_section = "name"]`
462449

450+
error[E0539]: malformed `coverage` attribute input
451+
--> $DIR/malformed-attrs.rs:90:1
452+
|
453+
LL | #[coverage]
454+
| ^^^^^^^^^^^ this attribute is only valid with either `on` or `off` as an argument
455+
|
456+
help: try changing it to one of the following valid forms of the attribute
457+
|
458+
LL | #[coverage(off)]
459+
| +++++
460+
LL | #[coverage(on)]
461+
| ++++
462+
463463
error[E0565]: malformed `no_implicit_prelude` attribute input
464464
--> $DIR/malformed-attrs.rs:97:1
465465
|

0 commit comments

Comments
 (0)