f - side effect in check_get_splice_locked

Author: jkczyz

lightning/src/ln/channel.rs

@@ -2156,6 +2156,41 @@ struct PendingSplice {
 	received_funding_txid: Option<Txid>,
 }
 
+#[cfg(splicing)]
+impl PendingSplice {
+	fn check_get_splice_locked<SP: Deref>(
+		&mut self, context: &ChannelContext<SP>, funding: &FundingScope, height: u32,
+	) -> Option<msgs::SpliceLocked>
+	where
+		SP::Target: SignerProvider,
+	{
+		if !context.check_funding_meets_minimum_depth(funding, height) {
+			return None;
+		}
+
+		let confirmed_funding_txid = match funding.get_funding_txid() {
+			Some(funding_txid) => funding_txid,
+			None => {
+				debug_assert!(false);
+				return None;
+			},
+		};
+
+		match self.sent_funding_txid {
+			Some(sent_funding_txid) if confirmed_funding_txid == sent_funding_txid => None,
+			_ => {
+				let splice_locked = msgs::SpliceLocked {
+					channel_id: context.channel_id(),
+					splice_txid: confirmed_funding_txid,
+				};
+				self.sent_funding_txid = Some(splice_locked.splice_txid);
+				Some(splice_locked)
+			},
+		}
+	}
+
+}
+
 /// Wrapper around a [`Transaction`] useful for caching the result of [`Transaction::compute_txid`].
 struct ConfirmedTransaction<'a> {
 	tx: &'a Transaction,
@@ -5525,6 +5560,29 @@ where
 		self.get_initial_counterparty_commitment_signature(funding, logger)
 	}
 
+	fn check_funding_meets_minimum_depth(&self, funding: &FundingScope, height: u32) -> bool {
+		let minimum_depth = self
+			.minimum_depth(funding)
+			.expect("ChannelContext::minimum_depth should be set for FundedChannel");
+
+		// Zero-conf channels always meet the minimum depth.
+		if minimum_depth == 0 {
+			return true;
+		}
+
+		if funding.funding_tx_confirmation_height == 0 {
+			return false;
+		}
+
+		let funding_tx_confirmations =
+			height as i64 - funding.funding_tx_confirmation_height as i64 + 1;
+		if funding_tx_confirmations < minimum_depth as i64 {
+			return false;
+		}
+
+		return true;
+	}
+
 	#[rustfmt::skip]
 	fn check_for_funding_tx_confirmed<L: Deref>(
 		&mut self, funding: &mut FundingScope, block_hash: &BlockHash, height: u32,
@@ -9074,53 +9132,8 @@ where
 		}
 	}
 
-	#[cfg(splicing)]
-	fn check_get_splice_locked(
-		&self, pending_splice: &PendingSplice, funding: &FundingScope, height: u32,
-	) -> Option<msgs::SpliceLocked> {
-		if !self.check_funding_meets_minimum_depth(funding, height) {
-			return None;
-		}
-
-		let confirmed_funding_txid = match funding.get_funding_txid() {
-			Some(funding_txid) => funding_txid,
-			None => {
-				debug_assert!(false);
-				return None;
-			},
-		};
-
-		match pending_splice.sent_funding_txid {
-			Some(sent_funding_txid) if confirmed_funding_txid == sent_funding_txid => None,
-			_ => Some(msgs::SpliceLocked {
-				channel_id: self.context.channel_id(),
-				splice_txid: confirmed_funding_txid,
-			}),
-		}
-	}
-
 	fn check_funding_meets_minimum_depth(&self, funding: &FundingScope, height: u32) -> bool {
-		let minimum_depth = self
-			.context
-			.minimum_depth(funding)
-			.expect("ChannelContext::minimum_depth should be set for FundedChannel");
-
-		// Zero-conf channels always meet the minimum depth.
-		if minimum_depth == 0 {
-			return true;
-		}
-
-		if funding.funding_tx_confirmation_height == 0 {
-			return false;
-		}
-
-		let funding_tx_confirmations =
-			height as i64 - funding.funding_tx_confirmation_height as i64 + 1;
-		if funding_tx_confirmations < minimum_depth as i64 {
-			return false;
-		}
-
-		return true;
+		self.context.check_funding_meets_minimum_depth(funding, height)
 	}
 
 	#[cfg(splicing)]
@@ -9241,7 +9254,7 @@ where
 
 			#[cfg(splicing)]
 			if let Some(confirmed_funding_index) = confirmed_funding_index {
-				let pending_splice = match self.pending_splice.as_ref() {
+				let pending_splice = match self.pending_splice.as_mut() {
 					Some(pending_splice) => pending_splice,
 					None => {
 						// TODO: Move pending_funding into pending_splice
@@ -9252,16 +9265,13 @@ where
 				};
 				let funding = self.pending_funding.get(confirmed_funding_index).unwrap();
 
-				if let Some(splice_locked) = self.check_get_splice_locked(pending_splice, funding, height) {
+				if let Some(splice_locked) = pending_splice.check_get_splice_locked(&self.context, funding, height) {
 					for &(idx, tx) in txdata.iter() {
 						if idx > index_in_block {
 							self.context.check_for_funding_tx_spent(funding, tx, logger)?;
 						}
 					}
 
-					let pending_splice = self.pending_splice.as_mut().unwrap();
-					pending_splice.sent_funding_txid = Some(splice_locked.splice_txid);
-
 					log_info!(
 						logger,
 						"Sending splice_locked txid {} to our peer for channel {}",
@@ -9427,12 +9437,9 @@ where
 				}
 			}
 
-			let pending_splice = self.pending_splice.as_ref().unwrap();
+			let pending_splice = self.pending_splice.as_mut().unwrap();
 			let funding = self.pending_funding.get(confirmed_funding_index).unwrap();
-			if let Some(splice_locked) = self.check_get_splice_locked(pending_splice, funding, height) {
-				let pending_splice = self.pending_splice.as_mut().unwrap();
-				pending_splice.sent_funding_txid = Some(splice_locked.splice_txid);
-
+			if let Some(splice_locked) = pending_splice.check_get_splice_locked(&self.context, funding, height) {
 				log_info!(logger, "Sending a splice_locked to our peer for channel {}", &self.context.channel_id);
 
 				let announcement_sigs = self