X-Git-Url: http://git.bitcoin.ninja/index.cgi?a=blobdiff_plain;f=lightning%2Fsrc%2Futil%2Fser_macros.rs;h=afd7fcb2e0fc93b3f536a052c15cec841f8b43b0;hb=3e9727b808814e6a6656a071a247695d9d95c572;hp=8d74a83a37365fc490ccf089449aad42bcb955ea;hpb=d325fa7831c9ce018edc46d1dffa22f84ccd20b3;p=rust-lightning diff --git a/lightning/src/util/ser_macros.rs b/lightning/src/util/ser_macros.rs index 8d74a83a..afd7fcb2 100644 --- a/lightning/src/util/ser_macros.rs +++ b/lightning/src/util/ser_macros.rs @@ -39,6 +39,9 @@ macro_rules! _encode_tlv { field.write($stream)?; } }; + ($stream: expr, $type: expr, $field: expr, ignorable) => { + $crate::_encode_tlv!($stream, $type, $field, required); + }; ($stream: expr, $type: expr, $field: expr, (option, encoding: ($fieldty: ty, $encoding: ident))) => { $crate::_encode_tlv!($stream, $type, $field.map(|f| $encoding(f)), option); }; @@ -155,6 +158,9 @@ macro_rules! _get_varint_length_prefixed_tlv_length { $len.0 += field_len; } }; + ($len: expr, $type: expr, $field: expr, ignorable) => { + $crate::_get_varint_length_prefixed_tlv_length!($len, $type, $field, required); + }; } /// See the documentation of [`write_tlv_fields`]. @@ -581,6 +587,9 @@ macro_rules! _init_tlv_based_struct_field { ($field: ident, option) => { $field }; + ($field: ident, ignorable) => { + if $field.is_none() { return Ok(None); } else { $field.unwrap() } + }; ($field: ident, required) => { $field.0.unwrap() }; @@ -610,6 +619,9 @@ macro_rules! _init_tlv_field_var { ($field: ident, option) => { let mut $field = None; }; + ($field: ident, ignorable) => { + let mut $field = None; + }; } /// Equivalent to running [`_init_tlv_field_var`] then [`read_tlv_fields`].