blockworks-foundation
/
anchor
mirror of https://github.com/blockworks-foundation/anchor.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Add account(zero_copy(safe_bytemuck_derives)) feature. 

The default account(zero_copy) feature unsafe impls the bytemuck traits

The new one derives it instead, which runs desired sanity checks like
"struct has no padding".
This commit is contained in:
Christian Kamm 2022-12-05 13:54:19 +01:00
parent fbd238fb7c
commit 3f8373770c
1 changed files with 31 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

36
lang/attribute/account/src/lib.rs
View File

@ -66,6 +66,7 @@ pub fn account(
) -> proc_macro::TokenStream {
let mut namespace = "".to_string();
let mut is_zero_copy = false;
let mut safe_bytemuck = false;
let args_str = args.to_string();
let args: Vec<&str> = args_str.split(',').collect();
if args.len() > 2 {
@ -80,6 +81,10 @@ pub fn account(
.collect();
if ns == "zero_copy" {
is_zero_copy = true;
safe_bytemuck = false;
} else if ns == "zero_copy(safe_bytemuck_derives)" {
is_zero_copy = true;
safe_bytemuck = true;
} else {
namespace = ns;
}
@ -123,16 +128,37 @@ pub fn account(
}
};
proc_macro::TokenStream::from({
if is_zero_copy {
let unsafe_bytemuck_impl = {
if !safe_bytemuck {
quote! {
#[zero_copy]
#account_strct
#[automatically_derived]
unsafe impl #impl_gen anchor_lang::__private::bytemuck::Pod for #account_name #type_gen #where_clause {}
#[automatically_derived]
unsafe impl #impl_gen anchor_lang::__private::bytemuck::Zeroable for #account_name #type_gen #where_clause {}
}
} else {
quote! {}
}
};
let safe_bytemuck_derives = {
if safe_bytemuck {
quote! {
#[derive(bytemuck::Pod, bytemuck::Zeroable)]
}
} else {
quote! {}
}
};
proc_macro::TokenStream::from({
if is_zero_copy {
quote! {
#[zero_copy]
#safe_bytemuck_derives
#account_strct
#unsafe_bytemuck_impl
#[automatically_derived]
impl #impl_gen anchor_lang::ZeroCopy for #account_name #type_gen #where_clause {}