add to Handle peer storage message

Author: Aditya Sharma

lightning/src/ln/channel.rs

@@ -2109,11 +2109,6 @@ impl<SP: Deref> ChannelContext<SP> where SP::Target: SignerProvider  {
 		self.channel_keys_id
 	}
 
-	pub fn increment_and_fetch_monitor_update_id(&mut self) -> u64 {
-		self.latest_monitor_update_id +=1;
-		self.latest_monitor_update_id
-	}
-
 	pub fn get_latest_monitor_update_id(&self) -> u64 {
 		self.latest_monitor_update_id
 	}
@@ -4544,6 +4539,22 @@ impl<SP: Deref> Channel<SP> where
 		return Ok(self.push_ret_blockable_mon_update(monitor_update));
 	}
 
+	/// Used to send LatestPeerStorage to the corresponding `ChannelMonitor`
+	pub fn update_peer_storage(&mut self, data: Vec<u8>) -> ChannelMonitorUpdate {
+		self.context.latest_monitor_update_id += 1;
+		let monitor_update = ChannelMonitorUpdate {
+			update_id: self.context.latest_monitor_update_id,
+			counterparty_node_id: None,
+			updates: vec![ChannelMonitorUpdateStep::LatestPeerStorage { 
+				data,
+			}],
+			channel_id: Some(self.context.channel_id()),
+		};
+		self.monitor_updating_paused(false, false, false, Vec::new(), Vec::new(), Vec::new());
+
+		monitor_update
+	}
+
 	/// Public version of the below, checking relevant preconditions first.
 	/// If we're not in a state where freeing the holding cell makes sense, this is a no-op and
 	/// returns `(None, Vec::new())`.

lightning/src/ln/channelmanager.rs

@@ -6789,17 +6789,9 @@ where
 			.iter_mut()
 			.min_by_key(|s| s.context.get_funding_txo().unwrap().get_txid())
 			.unwrap();
-		
+
 		// Send ChannelMonitor Update.
-		let peer_storage_update = ChannelMonitorUpdate {
-			update_id: min_funded_chan.context.increment_and_fetch_monitor_update_id(),
-			counterparty_node_id: None,
-			updates: vec![ChannelMonitorUpdateStep::LatestPeerStorage { 
-				data: msg.data.clone(),
-			}],
-			channel_id: Some(min_funded_chan.context.channel_id()),
-		};
-		
+		let peer_storage_update = min_funded_chan.update_peer_storage(msg.data.clone());
 		// Update the store.
 		peer_state.peer_storage = msg.data.clone();