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() {
main_body = quote! {loop{}};
}
@ -167,6 +172,7 @@ pub fn run(args: TokenStream, item: TokenStream, arch: &Arch) -> TokenStream {
}
#entry
#main_attrs
fn main() -> #main_ret {
#main_body
}