robin/bitbuffer
1
0
Fork 0
mirror of https://codeberg.org/icewind/bitbuffer.git synced 2026-06-03 16:44:06 +02:00
Code Issues Releases Wiki Activity

rountrip tests

Browse source
This commit is contained in:
Robin Appelman 2021-07-12 20:19:36 +02:00
commit f704ee9f9a
4 changed files with 117 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

24
bitbuffer_derive/src/lib.rs
View file

@ -374,9 +374,14 @@ fn parse(data: Data, struct_name: &Ident, attrs: &[Attribute], unchecked: bool)
}
}
Data::Enum(data) => {
let discriminant_bits: u64 = get_attribute_value(attrs, &["discriminant_bits"]).expect(
"'discriminant_bits' attribute is required when deriving `BinRead` for enums",
);
let discriminant_bits: u64 = match get_attribute_value(attrs, &["discriminant_bits"]) {
Some(attr) => attr,
None => {
return quote! {span=>
compile_error!("'discriminant_bits' attribute is required when deriving `BinRead` for enums");
}
}
};
let mut last_discriminant = -1;
let match_arms = data.variants.iter().map(|variant| {
@ -484,10 +489,15 @@ fn size(data: Data, struct_name: &Ident, attrs: &[Attribute], has_input_size: bo
}
}
Data::Enum(data) => {
let discriminant_bits = get_attribute_value::<u64>(attrs, &["discriminant_bits"])
.expect(
"'discriminant_bits' attribute is required when deriving `BinRead` for enums",
) as usize;
let discriminant_bits = match get_attribute_value::<u64>(attrs, &["discriminant_bits"])
{
Some(attr) => attr as usize,
None => {
return quote! {span=>
compile_error!("'discriminant_bits' attribute is required when deriving `BinRead` for enums");
}
}
};
let is_unit = data
.variants

11
bitbuffer_derive/src/write.rs
View file

@ -137,9 +137,14 @@ fn write(data: Data, struct_name: &Ident, attrs: &[Attribute]) -> TokenStream {
}
}
Data::Enum(data) => {
let discriminant_bits: u64 = get_attribute_value(attrs, &["discriminant_bits"]).expect(
"'discriminant_bits' attribute is required when deriving `BinRead` for enums",
);
let discriminant_bits: u64 = match get_attribute_value(attrs, &["discriminant_bits"]) {
Some(attr) => attr,
None => {
return quote! {span=>
compile_error!("'discriminant_bits' attribute is required when deriving `BinWrite` for enums");
}
}
};
let mut last_discriminant = -1;