f unit test

Author: arik-so

lightning-rapid-gossip-sync/src/processing.rs

@@ -9,8 +9,8 @@ use lightning::ln::msgs::{DecodeError, ErrorAction, LightningError, SocketAddres
 use lightning::routing::gossip::{NetworkGraph, NodeAlias, NodeId};
 use lightning::util::logger::Logger;
 use lightning::{log_debug, log_warn, log_trace, log_given_level, log_gossip};
-use lightning::util::ser::{BigSize, Readable};
-use lightning::io;
+use lightning::util::ser::{BigSize, FixedLengthReader, Readable};
+use lightning::{io, io::Read};
 
 use crate::{GraphSyncError, RapidGossipSync};
 
@@ -141,8 +141,9 @@ impl<NG: Deref<Target=NetworkGraph<L>>, L: Deref> RapidGossipSync<NG, L> where L
 					let mut node_addresses: Vec<SocketAddress> = Vec::new();
 					for _ in 0..address_count {
 						let current_byte_count: u8 = Readable::read(read_cursor)?;
-						let mut address_bytes = vec![0u8; current_byte_count as usize];
-						read_cursor.read_exact(&mut address_bytes).unwrap();
+						let mut address_reader = FixedLengthReader::new(&mut read_cursor, current_byte_count as u64);
+						let mut address_bytes = Vec::new();
+						address_reader.read_to_end(&mut address_bytes).unwrap();
 
 						let mut address_cursor = io::Cursor::new(&address_bytes);
 						if let Ok(current_address) = Readable::read(&mut address_cursor) {