Skip to content

Commit

Permalink
lang: repr override for zero copy (#1273)
Browse files Browse the repository at this point in the history
  • Loading branch information
armaniferrante committed Jan 9, 2022
1 parent e3d733c commit d7e72b3
Show file tree
Hide file tree
Showing 2 changed files with 14 additions and 1 deletion.
14 changes: 13 additions & 1 deletion lang/attribute/account/src/lib.rs
Expand Up @@ -291,9 +291,21 @@ pub fn zero_copy(
) -> proc_macro::TokenStream {
let account_strct = parse_macro_input!(item as syn::ItemStruct);

// Takes the first repr. It's assumed that more than one are not on the
// struct.
let attr = account_strct
.attrs
.iter()
.find(|attr| anchor_syn::parser::tts_to_string(&attr.path) == "repr");

let repr = match attr {
Some(_attr) => quote! {},
None => quote! {#[repr(C)]},
};

proc_macro::TokenStream::from(quote! {
#[derive(anchor_lang::__private::ZeroCopyAccessor, Copy, Clone)]
#[repr(C)]
#repr
#account_strct
})
}
Expand Down
1 change: 1 addition & 0 deletions tests/zero-copy/programs/zero-copy/src/lib.rs
Expand Up @@ -139,6 +139,7 @@ pub struct UpdateLargeAccount<'info> {

#[account(zero_copy)]
#[derive(Default)]
#[repr(packed)]
pub struct Foo {
pub authority: Pubkey,
pub data: u64,
Expand Down

0 comments on commit d7e72b3

Please sign in to comment.