From 9fe8f22a8451fa511069f9daa4dc0c20dccd57ba Mon Sep 17 00:00:00 2001
From: Henry de Valence <hdevalence@hdevalence.ca>
Date: Thu, 19 Sep 2019 13:17:22 -0700
Subject: [PATCH] Add verack, ping, pong serialization.

---
 zebra-network/src/message.rs | 24 +++++++++++++++++-------
 1 file changed, 17 insertions(+), 7 deletions(-)

diff --git a/zebra-network/src/message.rs b/zebra-network/src/message.rs
index 7509a5ab..617076cb 100644
--- a/zebra-network/src/message.rs
+++ b/zebra-network/src/message.rs
@@ -93,13 +93,16 @@ pub enum Message {
     /// A `ping` message.
     ///
     /// [Bitcoin reference](https://en.bitcoin.it/wiki/Protocol_documentation#ping)
-    Ping(Nonce),
+    Ping(
+        /// A nonce unique to this [`Ping`] message.
+        Nonce,
+    ),
 
     /// A `pong` message.
     ///
     /// [Bitcoin reference](https://en.bitcoin.it/wiki/Protocol_documentation#pong)
     Pong(
-        /// The nonce from the `Ping` message this was in response to.
+        /// The nonce from the [`Ping`] message this was in response to.
         Nonce,
     ),
 
@@ -450,6 +453,13 @@ impl Message {
                 writer.write_u32::<LittleEndian>(start_height.0)?;
                 writer.write_u8(*relay as u8)?;
             }
+            Verack => { /* Empty payload -- no-op */ }
+            Ping(nonce) => {
+                writer.write_u64::<LittleEndian>(nonce.0)?;
+            }
+            Pong(nonce) => {
+                writer.write_u64::<LittleEndian>(nonce.0)?;
+            }
             _ => unimplemented!(),
         }
         Ok(())
@@ -487,21 +497,21 @@ fn try_read_verack<R: io::Read>(
     mut _reader: R,
     _version: Version,
 ) -> Result<Message, SerializationError> {
-    unimplemented!()
+    Ok(Message::Verack)
 }
 
 fn try_read_ping<R: io::Read>(
-    mut _reader: R,
+    mut reader: R,
     _version: Version,
 ) -> Result<Message, SerializationError> {
-    unimplemented!()
+    Ok(Message::Ping(Nonce(reader.read_u64::<LittleEndian>()?)))
 }
 
 fn try_read_pong<R: io::Read>(
-    mut _reader: R,
+    mut reader: R,
     _version: Version,
 ) -> Result<Message, SerializationError> {
-    unimplemented!()
+    Ok(Message::Pong(Nonce(reader.read_u64::<LittleEndian>()?)))
 }
 
 fn try_read_reject<R: io::Read>(