X-Git-Url: http://git.bitcoin.ninja/index.cgi?a=blobdiff_plain;f=src%2Futil%2Fser_macros.rs;h=09a319fb45951a043106f9ae6d7fe4d13b94779b;hb=af29adcfdc414ebf6c331ad67936f4ab322a1c02;hp=2e1386a9c304e34f80beedd4b23a82ffac2c49da;hpb=1f1f82569a1774d2f6cb84e1de3c55dc733022c8;p=rust-lightning

diff --git a/src/util/ser_macros.rs b/src/util/ser_macros.rs
index 2e1386a9..09a319fb 100644
--- 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(())
 			}