Merge pull request #3958 from us-irs/preserve-user-attrs-on-main

preserve user attributes on main method
This commit is contained in:
Dario Nieuwenhuis 2025-03-10 13:59:33 +00:00 committed by GitHub
commit a704c41a36
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -155,6 +155,11 @@ pub fn run(args: TokenStream, item: TokenStream, arch: &Arch) -> TokenStream {
), ),
}; };
let mut main_attrs = TokenStream::new();
for attr in f.attrs {
main_attrs.extend(quote!(#attr));
}
if !errors.is_empty() { if !errors.is_empty() {
main_body = quote! {loop{}}; main_body = quote! {loop{}};
} }
@ -167,6 +172,7 @@ pub fn run(args: TokenStream, item: TokenStream, arch: &Arch) -> TokenStream {
} }
#entry #entry
#main_attrs
fn main() -> #main_ret { fn main() -> #main_ret {
#main_body #main_body
} }