Clean parsing via ReadZcashExt read-array helpers.

This adds convenience methods to `ReadZcashExt` that read 4 and 12 byte
fixed size arrays from the `Reader`, making the actual parsing code more
legible.

Closes #10.

zebra-chain/src/serialization.rs

@@ -220,6 +220,22 @@ pub trait ReadZcashExt: io::Read {
         self.read_exact(&mut buf)?;
         String::from_utf8(buf).map_err(|_| io::ErrorKind::InvalidData.into())
     }
+
+    /// Convenience method to read a `[u8; 4]`.
+    #[inline]
+    fn read_4_bytes(&mut self) -> io::Result<[u8; 4]> {
+        let mut bytes = [0; 4];
+        self.read_exact(&mut bytes)?;
+        Ok(bytes)
+    }
+
+    /// Convenience method to read a `[u8; 12]`.
+    #[inline]
+    fn read_12_bytes(&mut self) -> io::Result<[u8; 12]> {
+        let mut bytes = [0; 12];
+        self.read_exact(&mut bytes)?;
+        Ok(bytes)
+    }
 }
 
 /// Mark all types implementing `Read` as implementing the extension.

zebra-network/src/message.rs

@@ -392,29 +392,18 @@ impl Message {
         version: Version,
     ) -> Result<Self, SerializationError> {
         use SerializationError::ParseError;
-        let message_magic = {
+
-            let mut bytes = [0u8; 4];
+        // Read header data
-            reader.read_exact(&mut bytes)?;
+        let message_magic = Magic(reader.read_4_bytes()?);
-            Magic(bytes)
+        let command = reader.read_12_bytes()?;
-        };
+        let body_len = reader.read_u32::<LittleEndian>()? as usize;
+        let checksum = Sha256dChecksum(reader.read_4_bytes()?);
+
         if magic != message_magic {
             return Err(ParseError("Message has incorrect magic value"));
         }
 
-        let command = {
+        // XXX bound the body_len value to avoid large attacker-controlled allocs
-            let mut bytes = [0u8; 12];
-            reader.read_exact(&mut bytes)?;
-            bytes
-        };
-
-        let body_len = reader.read_u32::<LittleEndian>()? as usize;
-        // XXX ugly
-        let checksum = {
-            let mut bytes = [0u8; 4];
-            reader.read_exact(&mut bytes)?;
-            Sha256dChecksum(bytes)
-        };
-
         // XXX add a ChecksumReader<R: Read>(R) wrapper and avoid this
         let body = {
             let mut bytes = vec![0; body_len];