Add test_static_spendable_outputs_justice_tx_revoked_htlc*

Cover both HTLC-Timeout/Success cases

Author: Antoine Riard

src/ln/channelmanager.rs

@@ -7785,4 +7785,189 @@ mod tests {
 		};
 		check_spends!(spend_tx, node_txn[0].clone());
 	}
+
+	#[test]
+	fn test_static_spendable_outputs_justice_tx_revoked_htlc_timeout_tx() {
+		let nodes = create_network(2);
+
+		// Create some initial channels
+		let chan_1 = create_announced_chan_between_nodes(&nodes, 0, 1);
+
+		let payment_preimage = route_payment(&nodes[0], &vec!(&nodes[1])[..], 3000000).0;
+		let revoked_local_txn = nodes[0].node.channel_state.lock().unwrap().by_id.get(&chan_1.2).unwrap().last_local_commitment_txn.clone();
+		assert_eq!(revoked_local_txn[0].input.len(), 1);
+		assert_eq!(revoked_local_txn[0].input[0].previous_output.txid, chan_1.3.txid());
+
+		claim_payment(&nodes[0], &vec!(&nodes[1])[..], payment_preimage);
+
+		let header = BlockHeader { version: 0x20000000, prev_blockhash: Default::default(), merkle_root: Default::default(), time: 42, bits: 42, nonce: 42 };
+		// A will generate HTLC-Timeout from revoked commitment tx
+		nodes[0].chain_monitor.block_connected_with_filtering(&Block { header, txdata: vec![revoked_local_txn[0].clone()] }, 1);
+		let events = nodes[0].node.get_and_clear_pending_msg_events();
+		match events[0] {
+			MessageSendEvent::BroadcastChannelUpdate { .. } => {},
+			_ => panic!("Unexpected event"),
+		}
+		let revoked_htlc_txn = nodes[0].tx_broadcaster.txn_broadcasted.lock().unwrap();
+		assert_eq!(revoked_htlc_txn.len(), 2);
+		assert_eq!(revoked_htlc_txn[0].input.len(), 1);
+		assert_eq!(revoked_htlc_txn[0].input[0].witness.last().unwrap().len(), 133);
+		check_spends!(revoked_htlc_txn[0], revoked_local_txn[0].clone());
+
+		// B will generate justice tx from A's revoked commitment/HTLC tx
+		nodes[1].chain_monitor.block_connected_with_filtering(&Block { header, txdata: vec![revoked_local_txn[0].clone(), revoked_htlc_txn[0].clone()] }, 1);
+		let events = nodes[1].node.get_and_clear_pending_msg_events();
+		match events[0] {
+			MessageSendEvent::BroadcastChannelUpdate { .. } => {},
+			_ => panic!("Unexpected event"),
+		}
+
+		let node_txn = nodes[1].tx_broadcaster.txn_broadcasted.lock().unwrap();
+		assert_eq!(node_txn.len(), 4);
+		assert_eq!(node_txn[3].input.len(), 1);
+		check_spends!(node_txn[3], revoked_htlc_txn[0].clone());
+
+		let events = nodes[1].chan_monitor.simple_monitor.get_and_clear_pending_events();
+		// Check B's ChannelMonitor was able to generate the right spendable output descriptor
+		let spend_tx = match events[1] {
+			Event::SpendableOutputs { ref outputs } => {
+				match outputs[1] {
+					SpendableOutputDescriptor::StaticOutput { ref outpoint, ref output } => {
+						let secp_ctx = Secp256k1::new();
+						let input = TxIn {
+							previous_output: outpoint.clone(),
+							script_sig: Script::new(),
+							sequence: 0,
+							witness: Vec::new(),
+						};
+						let outp = TxOut {
+							script_pubkey: Builder::new().push_opcode(opcodes::All::OP_RETURN).into_script(),
+							value: output.value,
+						};
+						let mut spend_tx = Transaction {
+							version: 2,
+							lock_time: 0,
+							input: vec![input],
+							output: vec![outp.clone()],
+						};
+						let destination_secret = {
+							match ExtendedPrivKey::new_master(&secp_ctx, Network::Testnet, &nodes[1].node_seed) {
+								Ok(master_key) => {
+									match master_key.ckd_priv(&secp_ctx, ChildNumber::from_hardened_idx(1)) {
+										Ok(destination_key) => destination_key,
+										Err(_) => panic!("Your RNG is busted"),
+									}
+								}
+								Err(_) => panic!("Your rng is busted"),
+							}
+						};
+						let destination_pubkey = ExtendedPubKey::from_private(&secp_ctx, &destination_secret).public_key;
+						let witness_script = Address::p2pkh(&destination_pubkey, Network::Testnet).script_pubkey();
+						let sighash = Message::from_slice(&bip143::SighashComponents::new(&spend_tx).sighash_all(&spend_tx.input[0], &witness_script, output.value)[..]).unwrap();
+						let sig = secp_ctx.sign(&sighash, &destination_secret.secret_key);
+						spend_tx.input[0].witness.push(sig.serialize_der(&secp_ctx).to_vec());
+						spend_tx.input[0].witness[0].push(SigHashType::All as u8);
+						spend_tx.input[0].witness.push(destination_pubkey.serialize().to_vec());
+						spend_tx
+					},
+					_ => panic!("Unexpected event !"),
+				}
+			},
+			_ => panic!("Unexpected event !"),
+		};
+		check_spends!(spend_tx, node_txn[3].clone());
+	}
+
+	#[test]
+	fn test_static_spendable_outputs_justice_tx_revoked_htlc_success_tx() {
+		let nodes = create_network(2);
+
+		// Create some initial channels
+		let chan_1 = create_announced_chan_between_nodes(&nodes, 0, 1);
+
+		let payment_preimage = route_payment(&nodes[0], &vec!(&nodes[1])[..], 3000000).0;
+		let revoked_local_txn = nodes[1].node.channel_state.lock().unwrap().by_id.get(&chan_1.2).unwrap().last_local_commitment_txn.clone();
+		assert_eq!(revoked_local_txn[0].input.len(), 1);
+		assert_eq!(revoked_local_txn[0].input[0].previous_output.txid, chan_1.3.txid());
+
+		claim_payment(&nodes[0], &vec!(&nodes[1])[..], payment_preimage);
+
+		let header = BlockHeader { version: 0x20000000, prev_blockhash: Default::default(), merkle_root: Default::default(), time: 42, bits: 42, nonce: 42 };
+		// B will generate HTLC-Success from revoked commitment tx
+		nodes[1].chain_monitor.block_connected_with_filtering(&Block { header, txdata: vec![revoked_local_txn[0].clone()] }, 1);
+		let events = nodes[1].node.get_and_clear_pending_msg_events();
+		match events[0] {
+			MessageSendEvent::BroadcastChannelUpdate { .. } => {},
+			_ => panic!("Unexpected event"),
+		}
+		let revoked_htlc_txn = nodes[1].tx_broadcaster.txn_broadcasted.lock().unwrap();
+
+		assert_eq!(revoked_htlc_txn.len(), 2);
+		assert_eq!(revoked_htlc_txn[0].input.len(), 1);
+		assert_eq!(revoked_htlc_txn[0].input[0].witness.last().unwrap().len(), 138);
+		check_spends!(revoked_htlc_txn[0], revoked_local_txn[0].clone());
+
+		// A will generate justice tx from B's revoked commitment/HTLC tx
+		nodes[0].chain_monitor.block_connected_with_filtering(&Block { header, txdata: vec![revoked_local_txn[0].clone(), revoked_htlc_txn[0].clone()] }, 1);
+		let events = nodes[0].node.get_and_clear_pending_msg_events();
+		match events[0] {
+			MessageSendEvent::BroadcastChannelUpdate { .. } => {},
+			_ => panic!("Unexpected event"),
+		}
+
+		let node_txn = nodes[0].tx_broadcaster.txn_broadcasted.lock().unwrap();
+		assert_eq!(node_txn.len(), 4);
+		assert_eq!(node_txn[3].input.len(), 1);
+		check_spends!(node_txn[3], revoked_htlc_txn[0].clone());
+
+		let events = nodes[0].chan_monitor.simple_monitor.get_and_clear_pending_events();
+		// Check A's ChannelMonitor was able to generate the right spendable output descriptor
+		let spend_tx = match events[1] {
+			Event::SpendableOutputs { ref outputs } => {
+				match outputs[2] {
+					SpendableOutputDescriptor::StaticOutput { ref outpoint, ref output } => {
+						let secp_ctx = Secp256k1::new();
+						let input = TxIn {
+							previous_output: outpoint.clone(),
+							script_sig: Script::new(),
+							sequence: 0,
+							witness: Vec::new(),
+						};
+						let outp = TxOut {
+							script_pubkey: Builder::new().push_opcode(opcodes::All::OP_RETURN).into_script(),
+							value: output.value,
+						};
+						let mut spend_tx = Transaction {
+							version: 2,
+							lock_time: 0,
+							input: vec![input],
+							output: vec![outp.clone()],
+						};
+						let destination_secret = {
+							match ExtendedPrivKey::new_master(&secp_ctx, Network::Testnet, &nodes[0].node_seed) {
+								Ok(master_key) => {
+									match master_key.ckd_priv(&secp_ctx, ChildNumber::from_hardened_idx(1)) {
+										Ok(destination_key) => destination_key,
+										Err(_) => panic!("Your RNG is busted"),
+									}
+								}
+								Err(_) => panic!("Your rng is busted"),
+							}
+						};
+						let destination_pubkey = ExtendedPubKey::from_private(&secp_ctx, &destination_secret).public_key;
+						let witness_script = Address::p2pkh(&destination_pubkey, Network::Testnet).script_pubkey();
+						let sighash = Message::from_slice(&bip143::SighashComponents::new(&spend_tx).sighash_all(&spend_tx.input[0], &witness_script, output.value)[..]).unwrap();
+						let sig = secp_ctx.sign(&sighash, &destination_secret.secret_key);
+						spend_tx.input[0].witness.push(sig.serialize_der(&secp_ctx).to_vec());
+						spend_tx.input[0].witness[0].push(SigHashType::All as u8);
+						spend_tx.input[0].witness.push(destination_pubkey.serialize().to_vec());
+						spend_tx
+					},
+					_ => panic!("Unexpected event !"),
+				}
+			},
+			_ => panic!("Unexpected event !"),
+		};
+		check_spends!(spend_tx, node_txn[3].clone());
+	}
 }