f handle non-latest transactions confirming correctly and update test

TheBlueMatt · TheBlueMatt · commit 6074cf52efe7 · 2025-08-15T23:26:01.000Z
diff --git a/lightning/src/chain/channelmonitor.rs b/lightning/src/chain/channelmonitor.rs
@@ -2986,35 +2986,9 @@ impl<Signer: EcdsaChannelSigner> ChannelMonitor<Signer> {
 	/// Gets the set of outbound HTLCs which hit the chain and ultimately were claimed by us via
 	/// the timeout path and reached [`ANTI_REORG_DELAY`] confirmations. This is used to determine
 	/// if an HTLC has failed without the `ChannelManager` having seen it prior to being persisted.
-	pub(crate) fn get_onchain_failed_outbound_htlcs(
-		&self,
-	) -> HashMap<HTLCSource, HTLCOutputInCommitment> {
+	pub(crate) fn get_onchain_failed_outbound_htlcs(&self) -> HashMap<HTLCSource, PaymentHash> {
 		let mut res = new_hash_map();
 		let us = self.inner.lock().unwrap();
-		macro_rules! walk_htlcs {
-			($holder_commitment: expr, $htlc_iter: expr) => {
-				for (htlc, source) in $htlc_iter {
-					let filter = |v: &&IrrevocablyResolvedHTLC| {
-						v.commitment_tx_output_idx == htlc.transaction_output_index
-					};
-					if htlc.transaction_output_index.is_none() {
-						if let Some(source) = source {
-							// Dust HTLCs are always implicitly failed once the commitment
-							// transaction reaches ANTI_REORG_DELAY confirmations.
-							res.insert(source.clone(), htlc.clone());
-						}
-					} else if let Some(state) =
-						us.htlcs_resolved_on_chain.iter().filter(filter).next()
-					{
-						if let Some(source) = source {
-							if state.payment_preimage.is_none() {
-								res.insert(source.clone(), htlc.clone());
-							}
-						}
-					}
-				}
-			};
-		}
 
 		// We only want HTLCs with ANTI_REORG_DELAY confirmations, which implies the commitment
 		// transaction has least ANTI_REORG_DELAY confirmations for any dependent HTLC transactions
@@ -3033,18 +3007,75 @@ impl<Signer: EcdsaChannelSigner> ChannelMonitor<Signer> {
 			})
 		});
 
-		let txid = if let Some(txid) = confirmed_txid {
+		let confirmed_txid = if let Some(txid) = confirmed_txid {
 			txid
 		} else {
 			return res;
 		};
 
-		if Some(txid) == us.funding.current_counterparty_commitment_txid
-			|| Some(txid) == us.funding.prev_counterparty_commitment_txid
+		macro_rules! walk_htlcs {
+			($holder_commitment: expr, $htlc_iter: expr) => {
+				let mut walk_candidate_htlcs = |htlcs| {
+					for &(ref candidate_htlc, ref candidate_source) in htlcs {
+						let candidate_htlc: &HTLCOutputInCommitment = &candidate_htlc;
+						let candidate_source: &Option<Box<HTLCSource>> = &candidate_source;
+
+						let source: &HTLCSource = if let Some(source) = candidate_source {
+							&*source
+						} else {
+							continue;
+						};
+						let confirmed = $htlc_iter.find(|(_, conf_src)| Some(source) == *conf_src);
+						if let Some((confirmed_htlc, _)) = confirmed {
+							let filter = |v: &&IrrevocablyResolvedHTLC| {
+								v.commitment_tx_output_idx
+									== confirmed_htlc.transaction_output_index
+							};
+
+							// The HTLC was included in the confirmed commitment transaction, so we
+							// need to see if it has been irrevocably failed yet.
+							if confirmed_htlc.transaction_output_index.is_none() {
+								// Dust HTLCs are always implicitly failed once the commitment
+								// transaction reaches ANTI_REORG_DELAY confirmations.
+								res.insert(source.clone(), confirmed_htlc.payment_hash);
+							} else if let Some(state) =
+								us.htlcs_resolved_on_chain.iter().filter(filter).next()
+							{
+								if state.payment_preimage.is_none() {
+									res.insert(source.clone(), confirmed_htlc.payment_hash);
+								}
+							}
+						} else {
+							// The HTLC was not included in the confirmed commitment transaction,
+							// which has now reached ANTI_REORG_DELAY confirmations and thus the
+							// HTLC has been failed.
+							res.insert(source.clone(), candidate_htlc.payment_hash);
+						}
+					}
+				};
+
+				// We walk the set of HTLCs in the unrevoked counterparty commitment transactions (see
+				// `fail_unbroadcast_htlcs` for a description of why).
+				if let Some(ref txid) = us.funding.current_counterparty_commitment_txid {
+					if let Some(htlcs) = us.funding.counterparty_claimable_outpoints.get(txid) {
+						walk_candidate_htlcs(htlcs);
+					}
+				}
+				if let Some(ref txid) = us.funding.prev_counterparty_commitment_txid {
+					if let Some(htlcs) = us.funding.counterparty_claimable_outpoints.get(txid) {
+						walk_candidate_htlcs(htlcs);
+					}
+				}
+			};
+		}
+
+		if Some(confirmed_txid) == us.funding.current_counterparty_commitment_txid
+			|| Some(confirmed_txid) == us.funding.prev_counterparty_commitment_txid
 		{
+			let htlcs = funding.counterparty_claimable_outpoints.get(&confirmed_txid).unwrap();
 			walk_htlcs!(
 				false,
-				us.funding.counterparty_claimable_outpoints.get(&txid).unwrap().iter().filter_map(
+				htlcs.iter().filter_map(
 					|(a, b)| {
 						if let &Some(ref source) = b {
 							Some((a, Some(&**source)))
@@ -3054,12 +3085,18 @@ impl<Signer: EcdsaChannelSigner> ChannelMonitor<Signer> {
 					}
 				)
 			);
-		} else if txid == us.funding.current_holder_commitment_tx.trust().txid() {
+		} else if confirmed_txid == us.funding.current_holder_commitment_tx.trust().txid() {
 			walk_htlcs!(true, holder_commitment_htlcs!(us, CURRENT_WITH_SOURCES));
 		} else if let Some(prev_commitment_tx) = &us.funding.prev_holder_commitment_tx {
-			if txid == prev_commitment_tx.trust().txid() {
+			if confirmed_txid == prev_commitment_tx.trust().txid() {
 				walk_htlcs!(true, holder_commitment_htlcs!(us, PREV_WITH_SOURCES).unwrap());
+			} else {
+				let htlcs_confirmed: &[(&HTLCOutputInCommitment, _)] = &[];
+				walk_htlcs!(false, htlcs_confirmed.iter());
 			}
+		} else {
+			let htlcs_confirmed: &[(&HTLCOutputInCommitment, _)] = &[];
+			walk_htlcs!(false, htlcs_confirmed.iter());
 		}
 
 		res
diff --git a/lightning/src/ln/channelmanager.rs b/lightning/src/ln/channelmanager.rs
@@ -16541,15 +16541,15 @@ where
 							},
 						}
 					}
-					for (htlc_source, htlc) in monitor.get_onchain_failed_outbound_htlcs() {
+					for (htlc_source, payment_hash) in monitor.get_onchain_failed_outbound_htlcs() {
 						log_info!(
 							args.logger,
 							"Failing HTLC with payment hash {} as it was resolved on-chain.",
-							htlc.payment_hash
+							payment_hash
 						);
 						failed_htlcs.push((
 							htlc_source,
-							htlc.payment_hash,
+							payment_hash,
 							monitor.get_counterparty_node_id(),
 							monitor.channel_id(),
 							LocalHTLCFailureReason::OnChainTimeout,
diff --git a/lightning/src/ln/monitor_tests.rs b/lightning/src/ln/monitor_tests.rs
@@ -3464,7 +3464,16 @@ fn test_lost_preimage_monitor_events() {
 	do_test_lost_preimage_monitor_events(false);
 }
 
-fn do_test_lost_timeout_monitor_events(on_counterparty_tx: bool, dust_htlcs: bool) {
+#[derive(PartialEq)]
+enum CommitmentType {
+	RevokedCounterparty,
+	LatestCounterparty,
+	PreviousCounterparty,
+	LocalWithoutLastHTLC,
+	LocalWithLastHTLC,
+}
+
+fn do_test_lost_timeout_monitor_events(confirm_tx: CommitmentType, dust_htlcs: bool) {
 	// `MonitorEvent`s aren't delivered to the `ChannelManager` in a durable fasion - if the
 	// `ChannelManager` fetches the pending `MonitorEvent`s, then the `ChannelMonitor` gets
 	// persisted (i.e. due to a block update) then the node crashes, prior to persisting the
@@ -3511,9 +3520,37 @@ fn do_test_lost_timeout_monitor_events(on_counterparty_tx: bool, dust_htlcs: boo
 
 	send_payment(&nodes[0], &[&nodes[1], &nodes[2]], 25_000_000);
 
+	let cs_revoked_commit = get_local_commitment_txn!(nodes[2], chan_b);
+	assert_eq!(cs_revoked_commit.len(), 1);
+
 	let amt = if dust_htlcs { 1_000 } else { 10_000_000 };
 	let (_, hash_a, ..) = route_payment(&nodes[0], &[&nodes[1], &nodes[2]], amt);
-	let (_, hash_b, ..) = route_payment(&nodes[1], &[&nodes[2]], amt);
+
+	let cs_previous_commit = get_local_commitment_txn!(nodes[2], chan_b);
+	assert_eq!(cs_previous_commit.len(), 1);
+
+	let (route, hash_b, _, payment_secret_b) =
+		get_route_and_payment_hash!(nodes[1], nodes[2], amt);
+	let onion = RecipientOnionFields::secret_only(payment_secret_b);
+	nodes[1].node.send_payment_with_route(route, hash_b, onion, PaymentId(hash_b.0)).unwrap();
+	check_added_monitors(&nodes[1], 1);
+
+	let updates = get_htlc_update_msgs(&nodes[1], &node_c_id);
+	nodes[2].node.handle_update_add_htlc(node_b_id, &updates.update_add_htlcs[0]);
+	nodes[2].node.handle_commitment_signed_batch_test(node_b_id, &updates.commitment_signed);
+	check_added_monitors(&nodes[2], 1);
+
+	let (cs_raa, cs_cs) = get_revoke_commit_msgs!(nodes[2], node_b_id);
+	if confirm_tx == CommitmentType::LocalWithLastHTLC {
+		// Only deliver the last RAA + CS if we need to update the local commitment with the third
+		// HTLC.
+		nodes[1].node.handle_revoke_and_ack(node_c_id, &cs_raa);
+		check_added_monitors(&nodes[1], 1);
+		nodes[1].node.handle_commitment_signed_batch_test(node_c_id, &cs_cs);
+		check_added_monitors(&nodes[1], 1);
+
+		let _bs_raa = get_event_msg!(nodes[1], MessageSendEvent::SendRevokeAndACK, node_c_id);
+	}
 
 	nodes[1].node.peer_disconnected(nodes[2].node.get_our_node_id());
 	nodes[2].node.peer_disconnected(nodes[1].node.get_our_node_id());
@@ -3547,44 +3584,68 @@ fn do_test_lost_timeout_monitor_events(on_counterparty_tx: bool, dust_htlcs: boo
 	let bs_commit_tx = nodes[1].tx_broadcaster.txn_broadcasted.lock().unwrap().split_off(0);
 	assert_eq!(bs_commit_tx.len(), 1);
 
-	let selected_commit_tx = if on_counterparty_tx {
-		&cs_commit_tx[0]
-	} else {
-		&bs_commit_tx[0]
+	let selected_commit_tx = match confirm_tx {
+		CommitmentType::RevokedCounterparty => &cs_revoked_commit[0],
+		CommitmentType::PreviousCounterparty => &cs_previous_commit[0],
+		CommitmentType::LatestCounterparty => &cs_commit_tx[0],
+		CommitmentType::LocalWithoutLastHTLC|CommitmentType::LocalWithLastHTLC => &bs_commit_tx[0],
 	};
 
 	mine_transaction(&nodes[1], selected_commit_tx);
 	// If the block gets connected first we may re-broadcast B's commitment transaction before
-	// seeing the C's confirm.
-	nodes[1].tx_broadcaster.txn_broadcasted.lock().unwrap().clear();
+	// seeing the C's confirm. In any case, if we confirmed the revoked counterparty commitment
+	// transaction, we want to go ahead and confirm the spend of it.
+	let bs_transactions = nodes[1].tx_broadcaster.txn_broadcasted.lock().unwrap().split_off(0);
+	if confirm_tx == CommitmentType::RevokedCounterparty {
+		assert!(bs_transactions.len() == 1 || bs_transactions.len() == 2);
+		mine_transaction(&nodes[1], bs_transactions.last().unwrap());
+	} else {
+		assert!(bs_transactions.len() == 1 || bs_transactions.len() == 0);
+	}
+
 	connect_blocks(&nodes[1], ANTI_REORG_DELAY - 1);
 	let mut events = nodes[1].chain_monitor.chain_monitor.get_and_clear_pending_events();
-	if on_counterparty_tx {
-		assert_eq!(events.len(), 1, "{events:?}");
-		match events[0] {
-			Event::SpendableOutputs { .. } => {},
-			_ => panic!("Unexpected event {events:?}"),
-		}
-	} else {
-		assert_eq!(events.len(), 0);
+	match confirm_tx {
+		CommitmentType::LocalWithoutLastHTLC|CommitmentType::LocalWithLastHTLC => {
+			assert_eq!(events.len(), 0, "{events:?}");
+		},
+		CommitmentType::PreviousCounterparty|CommitmentType::LatestCounterparty => {
+			assert_eq!(events.len(), 1, "{events:?}");
+			match events[0] {
+				Event::SpendableOutputs { .. } => {},
+				_ => panic!("Unexpected event {events:?}"),
+			}
+		},
+		CommitmentType::RevokedCounterparty => {
+			assert_eq!(events.len(), 2, "{events:?}");
+			for event in events {
+				match event {
+					Event::SpendableOutputs { .. } => {},
+					_ => panic!("Unexpected event {event:?}"),
+				}
+			}
+		},
 	}
 
-	connect_blocks(&nodes[1], TEST_FINAL_CLTV - ANTI_REORG_DELAY + 1);
-	if !on_counterparty_tx {
-		if !dust_htlcs {
-			handle_bump_events(&nodes[1], false, 1);
+	if confirm_tx != CommitmentType::RevokedCounterparty {
+		connect_blocks(&nodes[1], TEST_FINAL_CLTV - ANTI_REORG_DELAY + 1);
+		if confirm_tx == CommitmentType::LocalWithoutLastHTLC || confirm_tx == CommitmentType::LocalWithLastHTLC {
+			if !dust_htlcs {
+				handle_bump_events(&nodes[1], false, 1);
+			}
 		}
 	}
+
 	let bs_htlc_timeouts =
 		nodes[1].tx_broadcaster.txn_broadcasted.lock().unwrap().split_off(0);
-	if dust_htlcs {
+	if dust_htlcs || confirm_tx == CommitmentType::RevokedCounterparty {
 		assert_eq!(bs_htlc_timeouts.len(), 0);
 	} else {
 		assert_eq!(bs_htlc_timeouts.len(), 1);
 
 		// Now replay the timeouts on node B, which after 6 confirmations should fail the HTLCs via
 		// `MonitorUpdate`s
-		mine_transactions(&nodes[1], &bs_htlc_timeouts.iter().collect::<Vec<_>>());
+		mine_transaction(&nodes[1], &bs_htlc_timeouts[0]);
 		connect_blocks(&nodes[1], ANTI_REORG_DELAY - 1);
 	}
 
@@ -3637,8 +3698,14 @@ fn do_test_lost_timeout_monitor_events(on_counterparty_tx: bool, dust_htlcs: boo
 
 #[test]
 fn test_lost_timeout_monitor_events() {
-	do_test_lost_timeout_monitor_events(true, false);
-	do_test_lost_timeout_monitor_events(false, false);
-	do_test_lost_timeout_monitor_events(true, true);
-	do_test_lost_timeout_monitor_events(false, true);
+	do_test_lost_timeout_monitor_events(CommitmentType::RevokedCounterparty, false);
+	do_test_lost_timeout_monitor_events(CommitmentType::RevokedCounterparty, true);
+	do_test_lost_timeout_monitor_events(CommitmentType::PreviousCounterparty, false);
+	do_test_lost_timeout_monitor_events(CommitmentType::PreviousCounterparty, true);
+	do_test_lost_timeout_monitor_events(CommitmentType::LatestCounterparty, false);
+	do_test_lost_timeout_monitor_events(CommitmentType::LatestCounterparty, true);
+	do_test_lost_timeout_monitor_events(CommitmentType::LocalWithoutLastHTLC, false);
+	do_test_lost_timeout_monitor_events(CommitmentType::LocalWithoutLastHTLC, true);
+	do_test_lost_timeout_monitor_events(CommitmentType::LocalWithLastHTLC, false);
+	do_test_lost_timeout_monitor_events(CommitmentType::LocalWithLastHTLC, true);
 }
