Rollup of 14 pull requests

compiler/rustc_ast/src/attr/mod.rs

@@ -235,14 +235,14 @@ impl AttributeExt for Attribute {
         }
     }
 
-    fn deprecation_note(&self) -> Option<Symbol> {
+    fn deprecation_note(&self) -> Option<Ident> {
         match &self.kind {
             AttrKind::Normal(normal) if normal.item.path == sym::deprecated => {
                 let meta = &normal.item;
 
                 // #[deprecated = "..."]
                 if let Some(s) = meta.value_str() {
-                    return Some(s);
+                    return Some(Ident { name: s, span: meta.span() });
                 }
 
                 // #[deprecated(note = "...")]
@@ -252,7 +252,7 @@ impl AttributeExt for Attribute {
                             && mi.path == sym::note
                             && let Some(s) = mi.value_str()
                         {
-                            return Some(s);
+                            return Some(Ident { name: s, span: mi.span });
                         }
                     }
                 }
@@ -905,7 +905,7 @@ pub trait AttributeExt: Debug {
     /// Returns the deprecation note if this is deprecation attribute.
     /// * `#[deprecated = "note"]` returns `Some("note")`.
     /// * `#[deprecated(note = "note", ...)]` returns `Some("note")`.
-    fn deprecation_note(&self) -> Option<Symbol>;
+    fn deprecation_note(&self) -> Option<Ident>;
 
     fn is_proc_macro_attr(&self) -> bool {
         [sym::proc_macro, sym::proc_macro_attribute, sym::proc_macro_derive]

compiler/rustc_attr_parsing/messages.ftl

@@ -23,6 +23,9 @@ attr_parsing_doc_alias_malformed =
 attr_parsing_doc_alias_start_end =
     {$attr_str} cannot start or end with ' '
 
+attr_parsing_doc_attr_not_crate_level =
+    `#![doc({$attr_name} = "...")]` isn't allowed as a crate-level attribute
+
 attr_parsing_doc_attribute_not_attribute =
     nonexistent builtin attribute `{$attribute}` used in `#[doc(attribute = "...")]`
     .help = only existing builtin attributes are allowed in core/std

compiler/rustc_attr_parsing/src/attributes/deprecation.rs

@@ -13,14 +13,14 @@ fn get<S: Stage>(
     name: Symbol,
     param_span: Span,
     arg: &ArgParser,
-    item: &Option<Symbol>,
-) -> Option<Symbol> {
+    item: Option<Symbol>,
+) -> Option<Ident> {
     if item.is_some() {
         cx.duplicate_key(param_span, name);
         return None;
     }
     if let Some(v) = arg.name_value() {
-        if let Some(value_str) = v.value_as_str() {
+        if let Some(value_str) = v.value_as_ident() {
             Some(value_str)
         } else {
             cx.expected_string_literal(v.value_span, Some(&v.value_as_lit()));
@@ -72,7 +72,7 @@ impl<S: Stage> SingleAttributeParser<S> for DeprecationParser {
         let features = cx.features();
 
         let mut since = None;
-        let mut note = None;
+        let mut note: Option<Ident> = None;
         let mut suggestion = None;
 
         let is_rustc = features.staged_api();
@@ -92,10 +92,16 @@ impl<S: Stage> SingleAttributeParser<S> for DeprecationParser {
 
                     match ident_name {
                         Some(name @ sym::since) => {
-                            since = Some(get(cx, name, param.span(), param.args(), &since)?);
+                            since = Some(get(cx, name, param.span(), param.args(), since)?.name);
                         }
                         Some(name @ sym::note) => {
-                            note = Some(get(cx, name, param.span(), param.args(), &note)?);
+                            note = Some(get(
+                                cx,
+                                name,
+                                param.span(),
+                                param.args(),
+                                note.map(|ident| ident.name),
+                            )?);
                         }
                         Some(name @ sym::suggestion) => {
                             if !features.deprecated_suggestion() {
@@ -107,7 +113,7 @@ impl<S: Stage> SingleAttributeParser<S> for DeprecationParser {
                             }
 
                             suggestion =
-                                Some(get(cx, name, param.span(), param.args(), &suggestion)?);
+                                Some(get(cx, name, param.span(), param.args(), suggestion)?.name);
                         }
                         _ => {
                             cx.expected_specific_argument(
@@ -124,7 +130,7 @@ impl<S: Stage> SingleAttributeParser<S> for DeprecationParser {
                 }
             }
             ArgParser::NameValue(v) => {
-                let Some(value) = v.value_as_str() else {
+                let Some(value) = v.value_as_ident() else {
                     cx.expected_string_literal(v.value_span, Some(v.value_as_lit()));
                     return None;
                 };

compiler/rustc_attr_parsing/src/attributes/doc.rs

@@ -1,5 +1,6 @@
 use rustc_ast::ast::{AttrStyle, LitKind, MetaItemLit};
 use rustc_feature::template;
+use rustc_hir::Target;
 use rustc_hir::attrs::{
     AttributeKind, CfgEntry, CfgHideShow, CfgInfo, DocAttribute, DocInline, HideOrShow,
 };
@@ -12,8 +13,8 @@ use super::{AcceptMapping, AttributeParser};
 use crate::context::{AcceptContext, FinalizeContext, Stage};
 use crate::parser::{ArgParser, MetaItemOrLitParser, MetaItemParser, OwnedPathParser};
 use crate::session_diagnostics::{
-    DocAliasBadChar, DocAliasEmpty, DocAliasMalformed, DocAliasStartEnd, DocAttributeNotAttribute,
-    DocKeywordNotKeyword,
+    DocAliasBadChar, DocAliasEmpty, DocAliasMalformed, DocAliasStartEnd, DocAttrNotCrateLevel,
+    DocAttributeNotAttribute, DocKeywordNotKeyword,
 };
 
 fn check_keyword<S: Stage>(cx: &mut AcceptContext<'_, '_, S>, keyword: Symbol, span: Span) -> bool {
@@ -43,16 +44,39 @@ fn check_attribute<S: Stage>(
     false
 }
 
-fn parse_keyword_and_attribute<S, F>(
+/// Checks that an attribute is *not* used at the crate level. Returns `true` if valid.
+fn check_attr_not_crate_level<S: Stage>(
+    cx: &mut AcceptContext<'_, '_, S>,
+    span: Span,
+    attr_name: Symbol,
+) -> bool {
+    if cx.shared.target.is_some_and(|target| target == Target::Crate) {
+        cx.emit_err(DocAttrNotCrateLevel { span, attr_name });
+        return false;
+    }
+    true
+}
+
+/// Checks that an attribute is used at the crate level. Returns `true` if valid.
+fn check_attr_crate_level<S: Stage>(cx: &mut AcceptContext<'_, '_, S>, span: Span) -> bool {
+    if cx.shared.target.is_some_and(|target| target != Target::Crate) {
+        cx.emit_lint(
+            rustc_session::lint::builtin::INVALID_DOC_ATTRIBUTES,
+            AttributeLintKind::AttrCrateLevelOnly,
+            span,
+        );
+        return false;
+    }
+    true
+}
+
+fn parse_keyword_and_attribute<S: Stage>(
     cx: &mut AcceptContext<'_, '_, S>,
     path: &OwnedPathParser,
     args: &ArgParser,
     attr_value: &mut Option<(Symbol, Span)>,
-    callback: F,
-) where
-    S: Stage,
-    F: FnOnce(&mut AcceptContext<'_, '_, S>, Symbol, Span) -> bool,
-{
+    attr_name: Symbol,
+) {
     let Some(nv) = args.name_value() else {
         cx.expected_name_value(args.span().unwrap_or(path.span()), path.word_sym());
         return;
@@ -63,16 +87,26 @@ fn parse_keyword_and_attribute<S, F>(
         return;
     };
 
-    if !callback(cx, value, nv.value_span) {
+    let ret = if attr_name == sym::keyword {
+        check_keyword(cx, value, nv.value_span)
+    } else {
+        check_attribute(cx, value, nv.value_span)
+    };
+    if !ret {
         return;
     }
 
+    let span = path.span();
     if attr_value.is_some() {
-        cx.duplicate_key(path.span(), path.word_sym().unwrap());
+        cx.duplicate_key(span, path.word_sym().unwrap());
         return;
     }
 
-    *attr_value = Some((value, path.span()));
+    if !check_attr_not_crate_level(cx, span, attr_name) {
+        return;
+    }
+
+    *attr_value = Some((value, span));
 }
 
 #[derive(Default, Debug)]
@@ -102,6 +136,10 @@ impl DocParser {
                     return;
                 }
 
+                if !check_attr_crate_level(cx, path.span()) {
+                    return;
+                }
+
                 self.attribute.no_crate_inject = Some(path.span())
             }
             Some(sym::attr) => {
@@ -155,6 +193,9 @@ impl DocParser {
             cx.emit_err(DocAliasStartEnd { span, attr_str });
             return;
         }
+        if !check_attr_not_crate_level(cx, span, sym::alias) {
+            return;
+        }
 
         if let Some(first_definition) = self.attribute.aliases.get(&alias).copied() {
             cx.emit_lint(
@@ -366,7 +407,33 @@ impl DocParser {
                 self.attribute.$ident = Some(path.span());
             }};
         }
-        macro_rules! string_arg {
+        macro_rules! no_args_and_not_crate_level {
+            ($ident: ident) => {{
+                if let Err(span) = args.no_args() {
+                    cx.expected_no_args(span);
+                    return;
+                }
+                let span = path.span();
+                if !check_attr_not_crate_level(cx, span, sym::$ident) {
+                    return;
+                }
+                self.attribute.$ident = Some(span);
+            }};
+        }
+        macro_rules! no_args_and_crate_level {
+            ($ident: ident) => {{
+                if let Err(span) = args.no_args() {
+                    cx.expected_no_args(span);
+                    return;
+                }
+                let span = path.span();
+                if !check_attr_crate_level(cx, span) {
+                    return;
+                }
+                self.attribute.$ident = Some(span);
+            }};
+        }
+        macro_rules! string_arg_and_crate_level {
             ($ident: ident) => {{
                 let Some(nv) = args.name_value() else {
                     cx.expected_name_value(args.span().unwrap_or(path.span()), path.word_sym());
@@ -378,6 +445,10 @@ impl DocParser {
                     return;
                 };
 
+                if !check_attr_crate_level(cx, path.span()) {
+                    return;
+                }
+
                 // FIXME: It's errorring when the attribute is passed multiple times on the command
                 // line.
                 // The right fix for this would be to only check this rule if the attribute is
@@ -394,12 +465,14 @@ impl DocParser {
         match path.word_sym() {
             Some(sym::alias) => self.parse_alias(cx, path, args),
             Some(sym::hidden) => no_args!(hidden),
-            Some(sym::html_favicon_url) => string_arg!(html_favicon_url),
-            Some(sym::html_logo_url) => string_arg!(html_logo_url),
-            Some(sym::html_no_source) => no_args!(html_no_source),
-            Some(sym::html_playground_url) => string_arg!(html_playground_url),
-            Some(sym::html_root_url) => string_arg!(html_root_url),
-            Some(sym::issue_tracker_base_url) => string_arg!(issue_tracker_base_url),
+            Some(sym::html_favicon_url) => string_arg_and_crate_level!(html_favicon_url),
+            Some(sym::html_logo_url) => string_arg_and_crate_level!(html_logo_url),
+            Some(sym::html_no_source) => no_args_and_crate_level!(html_no_source),
+            Some(sym::html_playground_url) => string_arg_and_crate_level!(html_playground_url),
+            Some(sym::html_root_url) => string_arg_and_crate_level!(html_root_url),
+            Some(sym::issue_tracker_base_url) => {
+                string_arg_and_crate_level!(issue_tracker_base_url)
+            }
             Some(sym::inline) => self.parse_inline(cx, path, args, DocInline::Inline),
             Some(sym::no_inline) => self.parse_inline(cx, path, args, DocInline::NoInline),
             Some(sym::masked) => no_args!(masked),
@@ -410,18 +483,18 @@ impl DocParser {
                 path,
                 args,
                 &mut self.attribute.keyword,
-                check_keyword,
+                sym::keyword,
             ),
             Some(sym::attribute) => parse_keyword_and_attribute(
                 cx,
                 path,
                 args,
                 &mut self.attribute.attribute,
-                check_attribute,
+                sym::attribute,
             ),
-            Some(sym::fake_variadic) => no_args!(fake_variadic),
-            Some(sym::search_unbox) => no_args!(search_unbox),
-            Some(sym::rust_logo) => no_args!(rust_logo),
+            Some(sym::fake_variadic) => no_args_and_not_crate_level!(fake_variadic),
+            Some(sym::search_unbox) => no_args_and_not_crate_level!(search_unbox),
+            Some(sym::rust_logo) => no_args_and_crate_level!(rust_logo),
             Some(sym::auto_cfg) => self.parse_auto_cfg(cx, path, args),
             Some(sym::test) => {
                 let Some(list) = args.list() else {

compiler/rustc_attr_parsing/src/attributes/mod.rs

@@ -47,6 +47,7 @@ pub(crate) mod link_attrs;
 pub(crate) mod lint_helpers;
 pub(crate) mod loop_match;
 pub(crate) mod macro_attrs;
+pub(crate) mod must_not_suspend;
 pub(crate) mod must_use;
 pub(crate) mod no_implicit_prelude;
 pub(crate) mod no_link;

compiler/rustc_attr_parsing/src/attributes/must_not_suspend.rs

@@ -0,0 +1,35 @@
+use super::prelude::*;
+
+pub(crate) struct MustNotSuspendParser;
+
+impl<S: Stage> SingleAttributeParser<S> for MustNotSuspendParser {
+    const PATH: &[rustc_span::Symbol] = &[sym::must_not_suspend];
+    const ATTRIBUTE_ORDER: AttributeOrder = AttributeOrder::KeepInnermost;
+    const ON_DUPLICATE: OnDuplicate<S> = OnDuplicate::Error;
+    const ALLOWED_TARGETS: AllowedTargets = AllowedTargets::AllowList(&[
+        Allow(Target::Struct),
+        Allow(Target::Enum),
+        Allow(Target::Union),
+        Allow(Target::Trait),
+    ]);
+    const TEMPLATE: AttributeTemplate = template!(Word, List: &["count"]);
+
+    fn convert(cx: &mut AcceptContext<'_, '_, S>, args: &ArgParser) -> Option<AttributeKind> {
+        let reason = match args {
+            ArgParser::NameValue(reason) => match reason.value_as_str() {
+                Some(val) => Some(val),
+                None => {
+                    cx.expected_nv_or_no_args(reason.value_span);
+                    return None;
+                }
+            },
+            ArgParser::NoArgs => None,
+            ArgParser::List(list) => {
+                cx.expected_nv_or_no_args(list.span);
+                return None;
+            }
+        };
+
+        Some(AttributeKind::MustNotSupend { reason })
+    }
+}

compiler/rustc_attr_parsing/src/context.rs

@@ -51,6 +51,7 @@ use crate::attributes::macro_attrs::{
     AllowInternalUnsafeParser, CollapseDebugInfoParser, MacroEscapeParser, MacroExportParser,
     MacroUseParser,
 };
+use crate::attributes::must_not_suspend::MustNotSuspendParser;
 use crate::attributes::must_use::MustUseParser;
 use crate::attributes::no_implicit_prelude::NoImplicitPreludeParser;
 use crate::attributes::no_link::NoLinkParser;
@@ -89,7 +90,6 @@ use crate::session_diagnostics::{
     AttributeParseError, AttributeParseErrorReason, ParsedDescription,
 };
 use crate::target_checking::AllowedTargets;
-
 type GroupType<S> = LazyLock<GroupTypeInner<S>>;
 
 pub(super) struct GroupTypeInner<S: Stage> {
@@ -208,6 +208,7 @@ attribute_parsers!(
         Single<LinkageParser>,
         Single<MacroExportParser>,
         Single<MoveSizeLimitParser>,
+        Single<MustNotSuspendParser>,
         Single<MustUseParser>,
         Single<ObjcClassParser>,
         Single<ObjcSelectorParser>,
@@ -655,6 +656,7 @@ pub struct SharedContext<'p, 'sess, S: Stage> {
     pub(crate) target_span: Span,
     /// The id ([`NodeId`] if `S` is `Early`, [`HirId`] if `S` is `Late`) of the syntactical component this attribute was applied to
     pub(crate) target_id: S::Id,
+    pub(crate) target: Option<rustc_hir::Target>,
 
     pub(crate) emit_lint: &'p mut dyn FnMut(AttributeLint<S::Id>),
 }