From: Matt Corallo <git@bluematt.me>
Date: Sun, 13 Mar 2022 21:35:46 +0000 (+0000)
Subject: Generate a funding_locked on relevant transactions_confirmed calls
X-Git-Tag: v0.0.106~21^2
X-Git-Url: http://git.bitcoin.ninja/index.cgi?a=commitdiff_plain;h=ea769427fe31c622f2ae7e48b7d97c97631355b2;p=rust-lightning

Generate a funding_locked on relevant transactions_confirmed calls

Previously, if we were offline when a funding transaction was
locked in, and then we came back online, calling
`best_block_updated` once followed by `transactions_confirmed`,
we'd not generate a funding_locked until the next
`best_block_updated`.

We address this by re-calling `best_block_updated` in
`transactions_confirmed`, similar to how `ChannelMonitor` works.
---

diff --git a/lightning/src/ln/channelmanager.rs b/lightning/src/ln/channelmanager.rs
index f3913c516..10404195c 100644
--- a/lightning/src/ln/channelmanager.rs
+++ b/lightning/src/ln/channelmanager.rs
@@ -5505,6 +5505,12 @@ where
 		let _persistence_guard = PersistenceNotifierGuard::notify_on_drop(&self.total_consistency_lock, &self.persistence_notifier);
 		self.do_chain_event(Some(height), |channel| channel.transactions_confirmed(&block_hash, height, txdata, self.genesis_hash.clone(), self.get_our_node_id(), &self.logger)
 			.map(|(a, b)| (a, Vec::new(), b)));
+
+		let last_best_block_height = self.best_block.read().unwrap().height();
+		if height < last_best_block_height {
+			let timestamp = self.highest_seen_timestamp.load(Ordering::Acquire);
+			self.do_chain_event(Some(last_best_block_height), |channel| channel.best_block_updated(last_best_block_height, timestamp as u32, self.genesis_hash.clone(), self.get_our_node_id(), &self.logger));
+		}
 	}
 
 	fn best_block_updated(&self, header: &BlockHeader, height: u32) {
diff --git a/lightning/src/ln/functional_tests.rs b/lightning/src/ln/functional_tests.rs
index 1f64ef0a9..3e2862cc4 100644
--- a/lightning/src/ln/functional_tests.rs
+++ b/lightning/src/ln/functional_tests.rs
@@ -3940,6 +3940,32 @@ fn test_funding_peer_disconnect() {
 	assert!(found_announcement);
 }
 
+#[test]
+fn test_funding_locked_without_best_block_updated() {
+	// Previously, if we were offline when a funding transaction was locked in, and then we came
+	// back online, calling best_block_updated once followed by transactions_confirmed, we'd not
+	// generate a funding_locked until a later best_block_updated. This tests that we generate the
+	// funding_locked immediately instead.
+	let chanmon_cfgs = create_chanmon_cfgs(2);
+	let node_cfgs = create_node_cfgs(2, &chanmon_cfgs);
+	let node_chanmgrs = create_node_chanmgrs(2, &node_cfgs, &[None, None]);
+	let mut nodes = create_network(2, &node_cfgs, &node_chanmgrs);
+	*nodes[0].connect_style.borrow_mut() = ConnectStyle::BestBlockFirstSkippingBlocks;
+
+	let funding_tx = create_chan_between_nodes_with_value_init(&nodes[0], &nodes[1], 1_000_000, 0, InitFeatures::known(), InitFeatures::known());
+
+	let conf_height = nodes[0].best_block_info().1 + 1;
+	connect_blocks(&nodes[0], CHAN_CONFIRM_DEPTH);
+	let block_txn = [funding_tx];
+	let conf_txn: Vec<_> = block_txn.iter().enumerate().collect();
+	let conf_block_header = nodes[0].get_block_header(conf_height);
+	nodes[0].node.transactions_confirmed(&conf_block_header, &conf_txn[..], conf_height);
+
+	// Ensure nodes[0] generates a funding_locked after the transactions_confirmed
+	let as_funding_locked = get_event_msg!(nodes[0], MessageSendEvent::SendFundingLocked, nodes[1].node.get_our_node_id());
+	nodes[1].node.handle_funding_locked(&nodes[0].node.get_our_node_id(), &as_funding_locked);
+}
+
 #[test]
 fn test_drop_messages_peer_disconnect_dual_htlc() {
 	// Test that we can handle reconnecting when both sides of a channel have pending