make defmt attributes forward input attributes

Cargo.toml

@@ -21,6 +21,9 @@ unstable-test = ["defmt-macros/unstable-test"]
 defmt-macros = { path = "macros", version = "0.1.1" }
 heapless = "0.5.6"
 
+[dev-dependencies]
+trybuild = "1.0.37"
+
 [workspace]
 members = [
   ".",

macros/src/lib.rs

@@ -12,11 +12,35 @@ use syn::{
     parse_macro_input,
     punctuated::Punctuated,
     spanned::Spanned as _,
-    Data, DeriveInput, Expr, ExprPath, Fields, FieldsNamed, FieldsUnnamed, GenericParam, ItemFn,
-    ItemStruct, LitStr, Path, PathArguments, PathSegment, ReturnType, Token, Type, WhereClause,
-    WherePredicate,
+    Attribute, Data, DeriveInput, Expr, ExprPath, Fields, FieldsNamed, FieldsUnnamed, GenericParam,
+    ItemFn, ItemStruct, LitStr, Path, PathArguments, PathSegment, ReturnType, Token, Type,
+    WhereClause, WherePredicate,
 };
 
+fn reject_attributes(
+    // appears in the error message
+    attr_name: &str,
+    attrs_to_check: &[Attribute],
+    reject_list: &[&str],
+) -> parse::Result<()> {
+    for attr in attrs_to_check {
+        if let Some(ident) = attr.path.get_ident() {
+            let ident = ident.to_string();
+            if reject_list.contains(&&*ident) {
+                return Err(parse::Error::new(
+                    attr.span(),
+                    format!(
+                        "`#[{}]` attribute cannot be used together with `#[{}]`",
+                        attr_name, ident
+                    ),
+                ));
+            }
+        }
+    }
+
+    Ok(())
+}
+
 #[proc_macro_attribute]
 pub fn global_logger(args: TokenStream, input: TokenStream) -> TokenStream {
     if !args.is_empty() {
@@ -42,8 +66,10 @@ pub fn global_logger(args: TokenStream, input: TokenStream) -> TokenStream {
         .into();
     }
 
+    let attrs = &s.attrs;
     let vis = &s.vis;
     quote!(
+        #(#attrs)*
         #vis struct #ident;
 
         #[no_mangle]
@@ -95,8 +121,13 @@ pub fn panic_handler(args: TokenStream, input: TokenStream) -> TokenStream {
             .into();
     }
 
+    let attrs = &f.attrs;
+    if let Err(e) = reject_attributes("panic_handler", attrs, &["export_name", "no_mangle"]) {
+        return e.to_compile_error().into();
+    }
     let block = &f.block;
     quote!(
+        #(#attrs)*
         #[export_name = "_defmt_panic"]
         fn #ident() -> ! {
             #block
@@ -141,9 +172,14 @@ pub fn timestamp(args: TokenStream, input: TokenStream) -> TokenStream {
             .into();
     }
 
+    let attrs = &f.attrs;
+    if let Err(e) = reject_attributes("timestamp", attrs, &["export_name", "no_mangle"]) {
+        return e.to_compile_error().into();
+    }
     let block = &f.block;
     quote!(
         #[export_name = "_defmt_timestamp"]
+        #(#attrs)*
         fn #ident() -> u64 {
             #block
         }

tests/ui.rs

@@ -0,0 +1,5 @@
+#[test]
+fn ui() {
+    let t = trybuild::TestCases::new();
+    t.compile_fail("tests/ui/*.rs");
+}

tests/ui/panic-handler-export-name.rs

@@ -0,0 +1,8 @@
+#![no_main]
+#![no_std]
+
+#[defmt::panic_handler]
+#[export_name = "hello"]
+fn foo() -> ! {
+    loop {}
+}

tests/ui/panic-handler-export-name.stderr

@@ -0,0 +1,5 @@
+error: `#[panic_handler]` attribute cannot be used together with `#[export_name]`
+ --> $DIR/panic-handler-export-name.rs:5:1
+  |
+5 | #[export_name = "hello"]
+  | ^^^^^^^^^^^^^^^^^^^^^^^^

tests/ui/panic-handler-no-mangle.rs

@@ -0,0 +1,8 @@
+#![no_main]
+#![no_std]
+
+#[defmt::panic_handler]
+#[no_mangle]
+fn panic() -> ! {
+    loop {}
+}

tests/ui/panic-handler-no-mangle.stderr

@@ -0,0 +1,5 @@
+error: `#[panic_handler]` attribute cannot be used together with `#[no_mangle]`
+ --> $DIR/panic-handler-no-mangle.rs:5:1
+  |
+5 | #[no_mangle]
+  | ^^^^^^^^^^^^

tests/ui/timestamp-export-name.rs

@@ -0,0 +1,8 @@
+#![no_main]
+#![no_std]
+
+#[defmt::timestamp]
+#[export_name = "hello"]
+fn foo() -> u64 {
+    0
+}

tests/ui/timestamp-export-name.stderr

@@ -0,0 +1,5 @@
+error: `#[timestamp]` attribute cannot be used together with `#[export_name]`
+ --> $DIR/timestamp-export-name.rs:5:1
+  |
+5 | #[export_name = "hello"]
+  | ^^^^^^^^^^^^^^^^^^^^^^^^

tests/ui/timestamp-no-mangle.rs

@@ -0,0 +1,8 @@
+#![no_main]
+#![no_std]
+
+#[defmt::timestamp]
+#[no_mangle]
+fn foo() -> u64 {
+    0
+}

tests/ui/timestamp-no-mangle.stderr

@@ -0,0 +1,5 @@
+error: `#[timestamp]` attribute cannot be used together with `#[no_mangle]`
+ --> $DIR/timestamp-no-mangle.rs:5:1
+  |
+5 | #[no_mangle]
+  | ^^^^^^^^^^^^