From 71f7fe3c888f0d71b1a4af131934bd66fb49b2a7 Mon Sep 17 00:00:00 2001
From: Revertron <r@revertron.com>
Date: Fri, 19 Feb 2021 23:00:55 +0100
Subject: [PATCH] Minor network fixes.

---
 src/p2p/network.rs | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/src/p2p/network.rs b/src/p2p/network.rs
index 1bca9dc..ca080ac 100644
--- a/src/p2p/network.rs
+++ b/src/p2p/network.rs
@@ -222,7 +222,7 @@ fn handle_connection_event(context: Arc<Mutex<Context>>, peers: &mut Peers, regi
     Ok(true)
 }
 
-fn read_message(stream: &mut TcpStream) -> Result<Vec<u8>, Vec<u8>> {
+fn read_message(stream: &mut TcpStream) -> Result<Vec<u8>, ()> {
     let instant = Instant::now();
     let data_size = match stream.read_u32::<BigEndian>() {
         Ok(size) => { size as usize }
@@ -233,7 +233,7 @@ fn read_message(stream: &mut TcpStream) -> Result<Vec<u8>, Vec<u8>> {
     };
     println!("Payload size is {}", data_size);
     if data_size > MAX_PACKET_SIZE {
-        return Err(Vec::new());
+        return Err(());
     }
 
     let mut buf = vec![0u8; data_size];
@@ -256,14 +256,14 @@ fn read_message(stream: &mut TcpStream) -> Result<Vec<u8>, Vec<u8>> {
             // Other errors we'll consider fatal.
             Err(_) => {
                 println!("Error reading message, only {} bytes read", bytes_read);
-                return Err(buf)
+                return Err(())
             },
         }
     }
     if buf.len() == data_size {
         Ok(buf)
     } else {
-        Err(buf)
+        Err(())
     }
 }