Macro-out checking a tx validly spends another (and add one more)

[rust-lightning] / src / util / ser_macros.rs

diff --git a/src/util/ser_macros.rs b/src/util/ser_macros.rs
index 2e1386a9c304e34f80beedd4b23a82ffac2c49da..09a319fb45951a043106f9ae6d7fe4d13b94779b 100644 (file)
--- a/src/util/ser_macros.rs
+++ b/src/util/ser_macros.rs
@@ -1,7 +1,29 @@
 macro_rules! impl_writeable {
-       ($st:ident, {$($field:ident),*}) => {
-               impl<W: Writer> Writeable<W> for $st {
-                       fn write(&self, w: &mut W) -> Result<(), DecodeError> {
+       ($st:ident, $len: expr, {$($field:ident),*}) => {
+               impl Writeable for $st {
+                       fn write<W: Writer>(&self, w: &mut W) -> Result<(), ::std::io::Error> {
+                               w.size_hint($len);
+                               $( self.$field.write(w)?; )*
+                               Ok(())
+                       }
+               }
+
+               impl<R: Read> Readable<R> for $st {
+                       fn read(r: &mut R) -> Result<Self, DecodeError> {
+                               Ok(Self {
+                                       $($field: Readable::read(r)?),*
+                               })
+                       }
+               }
+       }
+}
+macro_rules! impl_writeable_len_match {
+       ($st:ident, {$({$m: pat, $l: expr}),*}, {$($field:ident),*}) => {
+               impl Writeable for $st {
+                       fn write<W: Writer>(&self, w: &mut W) -> Result<(), ::std::io::Error> {
+                               w.size_hint(match *self {
+                                       $($m => $l,)*
+                               });
                                $( self.$field.write(w)?; )*
                                Ok(())
                        }