Add test yielding anchor-related events

Author: wpaulino

lightning/src/ln/monitor_tests.rs

@@ -9,17 +9,29 @@
 
 //! Further functional tests which test blockchain reorganizations.
 
+#[cfg(anchors)]
+use crate::chain::keysinterface::BaseSign;
+#[cfg(anchors)]
+use crate::chain::channelmonitor::LATENCY_GRACE_PERIOD_BLOCKS;
 use crate::chain::channelmonitor::{ANTI_REORG_DELAY, Balance};
 use crate::chain::transaction::OutPoint;
 use crate::chain::chaininterface::LowerBoundedFeeEstimator;
 use crate::ln::channel;
+#[cfg(anchors)]
+use crate::ln::chan_utils;
 use crate::ln::channelmanager::{self, BREAKDOWN_TIMEOUT};
 use crate::ln::msgs::ChannelMessageHandler;
+#[cfg(anchors)]
+use crate::util::config::UserConfig;
+#[cfg(anchors)]
+use crate::util::events::BumpTransactionEvent;
 use crate::util::events::{Event, MessageSendEvent, MessageSendEventsProvider, ClosureReason, HTLCDestination};
 
 use bitcoin::blockdata::script::Builder;
 use bitcoin::blockdata::opcodes;
 use bitcoin::secp256k1::Secp256k1;
+#[cfg(anchors)]
+use bitcoin::{Amount, Script, TxIn, TxOut, PackedLockTime};
 use bitcoin::Transaction;
 
 use crate::prelude::*;
@@ -1629,3 +1641,132 @@ fn test_revoked_counterparty_aggregated_claims() {
 	test_spendable_output(&nodes[1], &claim_txn_2[0]);
 	assert!(nodes[1].chain_monitor.chain_monitor.get_monitor(funding_outpoint).unwrap().get_claimable_balances().is_empty());
 }
+
+#[cfg(anchors)]
+#[test]
+fn test_yield_anchors_events() {
+	let secp = Secp256k1::new();
+	let mut chanmon_cfgs = create_chanmon_cfgs(2);
+	let node_cfgs = create_node_cfgs(2, &chanmon_cfgs);
+	let mut anchors_config = UserConfig::default();
+	anchors_config.channel_handshake_config.announced_channel = true;
+	anchors_config.channel_handshake_config.negotiate_anchors_zero_fee_htlc_tx = true;
+	let node_chanmgrs = create_node_chanmgrs(2, &node_cfgs, &[Some(anchors_config), Some(anchors_config)]);
+	let nodes = create_network(2, &node_cfgs, &node_chanmgrs);
+
+	let features = channelmanager::provided_init_features();
+	let _ = create_announced_chan_between_nodes_with_value(
+		&nodes, 0, 1, 1_000_000, 100_000_000, features.clone(), features.clone(),
+	);
+	let _ = route_payment(&nodes[0], &[&nodes[1]], 1_000_000).1;
+	let _ = route_payment(&nodes[1], &[&nodes[0]], 1_000_000).1;
+
+	assert!(nodes[0].node.get_and_clear_pending_events().is_empty());
+
+	let _ = connect_blocks(&nodes[0], TEST_FINAL_CLTV + LATENCY_GRACE_PERIOD_BLOCKS + 1);
+	assert!(nodes[0].tx_broadcaster.txn_broadcasted.lock().unwrap().is_empty());
+
+	let mut holder_events = nodes[0].chain_monitor.chain_monitor.get_and_clear_pending_events();
+	assert_eq!(holder_events.len(), 1);
+	let (commitment_tx, anchor_tx) = match holder_events.pop().unwrap() {
+		Event::BumpTransaction(BumpTransactionEvent::ChannelClose { commitment_tx, anchor_descriptor, .. })  => {
+			assert_eq!(commitment_tx.input.len(), 1);
+			assert_eq!(commitment_tx.output.len(), 6);
+			let mut anchor_tx = Transaction {
+				version: 2,
+				lock_time: PackedLockTime::ZERO,
+				input: vec![
+					TxIn { previous_output: anchor_descriptor.outpoint, ..Default::default() },
+					TxIn { ..Default::default() },
+				],
+				output: vec![TxOut {
+					value: Amount::ONE_BTC.to_sat(),
+					script_pubkey: Script::new_op_return(&[]),
+				}],
+			};
+			let signer = nodes[0].keys_manager.derive_channel_keys(
+				anchor_descriptor.channel_value_satoshis, &anchor_descriptor.channel_keys_id,
+			);
+			let funding_sig = signer.sign_holder_anchor_input(&mut anchor_tx, 0, &secp).unwrap();
+			anchor_tx.input[0].witness = chan_utils::build_anchor_input_witness(
+				&signer.pubkeys().funding_pubkey, &funding_sig
+			);
+			(commitment_tx, anchor_tx)
+		},
+		_ => panic!("Unexpected event"),
+	};
+
+	mine_transactions(&nodes[0], &[&commitment_tx, &anchor_tx]);
+	connect_blocks(&nodes[0], ANTI_REORG_DELAY - 1);
+	check_added_monitors!(nodes[0], 1);
+
+	let holder_events = nodes[0].chain_monitor.chain_monitor.get_and_clear_pending_events();
+	assert_eq!(holder_events.len(), 1);
+	let mut htlc_tx = Transaction { version: 2, lock_time: PackedLockTime::ZERO, input: vec![], output: vec![] };
+	for event in holder_events {
+		match event {
+			Event::BumpTransaction(BumpTransactionEvent::HTLCResolution { mut tx_template, htlc_descriptors, .. }) => {
+				tx_template.input.push(TxIn { ..Default::default() });
+				tx_template.output.push(TxOut {
+					value: Amount::ONE_BTC.to_sat(),
+					script_pubkey: Script::new_op_return(&[]),
+				});
+				for (idx, htlc_descriptor) in htlc_descriptors.iter().enumerate() {
+					let signer = nodes[0].keys_manager.derive_channel_keys(
+						htlc_descriptor.channel_value_satoshis, &htlc_descriptor.channel_keys_id
+					);
+					let our_sig = signer.sign_holder_htlc_transaction(
+						&mut tx_template, idx, htlc_descriptor.per_commitment_number,
+						&htlc_descriptor.htlc, &htlc_descriptor.counterparty_base_htlc_key,
+						&htlc_descriptor.counterparty_base_revocation_key, &secp
+					).unwrap();
+					let witness_script = {
+						let per_commitment_point = signer.get_per_commitment_point(
+							htlc_descriptor.per_commitment_number, &secp
+						);
+						let our_htlc_key = chan_utils::derive_public_key(
+							&secp, &per_commitment_point, &signer.pubkeys().htlc_basepoint
+						).unwrap();
+						let counterparty_htlc_key = chan_utils::derive_public_key(
+							&secp, &per_commitment_point, &htlc_descriptor.counterparty_base_htlc_key
+						).unwrap();
+						let revocation_key = chan_utils::derive_public_revocation_key(
+							&secp, &per_commitment_point, &htlc_descriptor.counterparty_base_revocation_key
+						).unwrap();
+						chan_utils::get_htlc_redeemscript_with_explicit_keys(
+							&htlc_descriptor.htlc, true, &our_htlc_key, &counterparty_htlc_key, &revocation_key,
+						)
+					};
+					tx_template.input[idx].witness = chan_utils::build_htlc_input_witness(
+						&our_sig, &htlc_descriptor.counterparty_sig, &htlc_descriptor.preimage,
+						&witness_script, true
+					);
+				}
+				htlc_tx = tx_template;
+			},
+			_ => panic!("Unexpected event"),
+		}
+	}
+
+	mine_transaction(&nodes[0], &htlc_tx);
+	connect_blocks(&nodes[0], ANTI_REORG_DELAY - 1);
+
+	assert!(nodes[0].chain_monitor.chain_monitor.get_and_clear_pending_events().is_empty());
+
+	connect_blocks(&nodes[0], BREAKDOWN_TIMEOUT as u32);
+
+	let holder_events = nodes[0].chain_monitor.chain_monitor.get_and_clear_pending_events();
+	assert_eq!(holder_events.len(), 2);
+	for event in holder_events {
+		match event {
+			Event::SpendableOutputs { .. } => {},
+			_ => panic!("Unexpected event"),
+		}
+	}
+
+	nodes[0].node.get_and_clear_pending_events();
+	nodes[1].node.get_and_clear_pending_events();
+
+	nodes[0].node.get_and_clear_pending_msg_events();
+	nodes[1].node.get_and_clear_pending_msg_events();
+}