mikros/rust
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Print unsafety of attribute in AST unpretty

Browse Source
This commit is contained in:
Urgau 2024-10-26 13:33:36 +02:00
parent f249fdd962
commit f5b6f938ce
2 changed files with 12 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
compiler/rustc_ast_pretty/src/pprust/state.rs
View File

@ -627,6 +627,13 @@ fn print_attribute_inline(&mut self, attr: &ast::Attribute, is_inline: bool) {
fn print_attr_item(&mut self, item: &ast::AttrItem, span: Span) { fn print_attr_item(&mut self, item: &ast::AttrItem, span: Span) {
self.ibox(0); self.ibox(0);
match item.unsafety {
ast::Safety::Unsafe(_) => {
self.word("unsafe");
self.popen();
}
ast::Safety::Default | ast::Safety::Safe(_) => {}
}
match &item.args { match &item.args {
AttrArgs::Delimited(DelimArgs { dspan: _, delim, tokens }) => self.print_mac_common( AttrArgs::Delimited(DelimArgs { dspan: _, delim, tokens }) => self.print_mac_common(
Some(MacHeader::Path(&item.path)), Some(MacHeader::Path(&item.path)),
@ -655,6 +662,10 @@ fn print_attr_item(&mut self, item: &ast::AttrItem, span: Span) {
self.word(token_str); self.word(token_str);
} }
} }
match item.unsafety {
ast::Safety::Unsafe(_) => self.pclose(),
ast::Safety::Default | ast::Safety::Safe(_) => {}
}
self.end(); self.end();
} }

2
tests/ui/unpretty/unsafe-attr.stdout
View File

@ -4,7 +4,7 @@
#[no_mangle] #[no_mangle]
extern "C" fn foo() {} extern "C" fn foo() {}
#[no_mangle] #[unsafe(no_mangle)]
extern "C" fn bar() {} extern "C" fn bar() {}
#[cfg_attr(FALSE, unsafe(no_mangle))] #[cfg_attr(FALSE, unsafe(no_mangle))]