use bitcoin::amount::Amount;
use bitcoin::hash_types::{BlockHash, Txid};
use bitcoin::hashes::Hash;
use bitcoin::locktime::absolute::LockTime;
use bitcoin::script::{Script, ScriptBuf};
use bitcoin::secp256k1;
use bitcoin::secp256k1::{ecdsa::Signature, Secp256k1};
use bitcoin::transaction::OutPoint as BitcoinOutPoint;
use bitcoin::transaction::Transaction;
use crate::chain::chaininterface::ConfirmationTarget;
use crate::chain::chaininterface::{BroadcasterInterface, FeeEstimator, LowerBoundedFeeEstimator};
use crate::chain::channelmonitor::ANTI_REORG_DELAY;
use crate::chain::package::{PackageSolvingData, PackageTemplate};
use crate::chain::transaction::MaybeSignedTransaction;
use crate::chain::ClaimId;
use crate::ln::chan_utils::{
get_keyed_anchor_redeemscript, shared_anchor_script_pubkey, ChannelTransactionParameters,
HTLCOutputInCommitment, HolderCommitmentTransaction,
};
use crate::ln::msgs::DecodeError;
use crate::sign::{ecdsa::EcdsaChannelSigner, EntropySource, HTLCDescriptor, SignerProvider};
use crate::util::logger::Logger;
use crate::util::ser::{
MaybeReadable, Readable, ReadableArgs, UpgradableRequired, Writeable, Writer,
};
use crate::io;
use crate::prelude::*;
use alloc::collections::BTreeMap;
use core::cmp;
use core::mem::replace;
use core::mem::swap;
use core::ops::Deref;
const MAX_ALLOC_SIZE: usize = 64 * 1024;
#[derive(Clone, PartialEq, Eq)]
struct OnchainEventEntry {
txid: Txid,
height: u32,
block_hash: Option<BlockHash>, event: OnchainEvent,
}
impl OnchainEventEntry {
fn confirmation_threshold(&self) -> u32 {
self.height + ANTI_REORG_DELAY - 1
}
fn has_reached_confirmation_threshold(&self, height: u32) -> bool {
height >= self.confirmation_threshold()
}
}
#[derive(Clone, PartialEq, Eq)]
enum OnchainEvent {
Claim { claim_id: ClaimId },
ContentiousOutpoint { package: PackageTemplate },
}
impl Writeable for OnchainEventEntry {
fn write<W: Writer>(&self, writer: &mut W) -> Result<(), io::Error> {
write_tlv_fields!(writer, {
(0, self.txid, required),
(1, self.block_hash, option),
(2, self.height, required),
(4, self.event, required),
});
Ok(())
}
}
impl MaybeReadable for OnchainEventEntry {
#[rustfmt::skip]
fn read<R: io::Read>(reader: &mut R) -> Result<Option<Self>, DecodeError> {
let mut txid = Txid::all_zeros();
let mut height = 0;
let mut block_hash = None;
let mut event = UpgradableRequired(None);
read_tlv_fields!(reader, {
(0, txid, required),
(1, block_hash, option),
(2, height, required),
(4, event, upgradable_required),
});
Ok(Some(Self { txid, height, block_hash, event: _init_tlv_based_struct_field!(event, upgradable_required) }))
}
}
impl_writeable_tlv_based_enum_upgradable!(OnchainEvent,
(0, Claim) => {
(0, claim_id, required),
},
(1, ContentiousOutpoint) => {
(0, package, required),
},
);
impl Readable for Option<Vec<Option<(usize, Signature)>>> {
#[rustfmt::skip]
fn read<R: io::Read>(reader: &mut R) -> Result<Self, DecodeError> {
match Readable::read(reader)? {
0u8 => Ok(None),
1u8 => {
let vlen: u64 = Readable::read(reader)?;
let mut ret = Vec::with_capacity(cmp::min(vlen as usize, MAX_ALLOC_SIZE / ::core::mem::size_of::<Option<(usize, Signature)>>()));
for _ in 0..vlen {
ret.push(match Readable::read(reader)? {
0u8 => None,
1u8 => Some((<u64 as Readable>::read(reader)? as usize, Readable::read(reader)?)),
_ => return Err(DecodeError::InvalidValue)
});
}
Ok(Some(ret))
},
_ => Err(DecodeError::InvalidValue),
}
}
}
impl Writeable for Option<Vec<Option<(usize, Signature)>>> {
fn write<W: Writer>(&self, writer: &mut W) -> Result<(), io::Error> {
match self {
&Some(ref vec) => {
1u8.write(writer)?;
(vec.len() as u64).write(writer)?;
for opt in vec.iter() {
match opt {
&Some((ref idx, ref sig)) => {
1u8.write(writer)?;
(*idx as u64).write(writer)?;
sig.write(writer)?;
},
&None => 0u8.write(writer)?,
}
}
},
&None => 0u8.write(writer)?,
}
Ok(())
}
}
#[derive(Clone, PartialEq, Eq)]
pub(crate) enum ClaimEvent {
BumpCommitment {
package_target_feerate_sat_per_1000_weight: u32,
commitment_tx: Transaction,
commitment_tx_fee_satoshis: u64,
pending_nondust_htlcs: Vec<HTLCOutputInCommitment>,
anchor_output_idx: u32,
channel_parameters: ChannelTransactionParameters,
},
BumpHTLC {
target_feerate_sat_per_1000_weight: u32,
htlcs: Vec<HTLCDescriptor>,
tx_lock_time: LockTime,
},
}
pub(crate) enum OnchainClaim {
Tx(MaybeSignedTransaction),
Event(ClaimEvent),
}
#[derive(Debug)]
pub(crate) enum FeerateStrategy {
RetryPrevious,
HighestOfPreviousOrNew,
ForceBump,
}
#[derive(Clone)]
pub struct OnchainTxHandler<ChannelSigner: EcdsaChannelSigner> {
channel_value_satoshis: u64, channel_keys_id: [u8; 32], destination_script: ScriptBuf, holder_commitment: HolderCommitmentTransaction,
prev_holder_commitment: Option<HolderCommitmentTransaction>,
pub(super) signer: ChannelSigner,
channel_transaction_parameters: ChannelTransactionParameters,
#[cfg(any(test, feature = "_test_utils"))]
pub(crate) pending_claim_requests: HashMap<ClaimId, PackageTemplate>,
#[cfg(not(any(test, feature = "_test_utils")))]
pending_claim_requests: HashMap<ClaimId, PackageTemplate>,
pending_claim_events: Vec<(ClaimId, ClaimEvent)>,
#[cfg(any(test, feature = "_test_utils"))]
pub(crate) claimable_outpoints: HashMap<BitcoinOutPoint, (ClaimId, u32)>,
#[cfg(not(any(test, feature = "_test_utils")))]
claimable_outpoints: HashMap<BitcoinOutPoint, (ClaimId, u32)>,
#[cfg(any(test, feature = "_test_utils"))]
pub(crate) locktimed_packages: BTreeMap<u32, Vec<PackageTemplate>>,
#[cfg(not(any(test, feature = "_test_utils")))]
locktimed_packages: BTreeMap<u32, Vec<PackageTemplate>>,
onchain_events_awaiting_threshold_conf: Vec<OnchainEventEntry>,
pub(super) secp_ctx: Secp256k1<secp256k1::All>,
}
impl<ChannelSigner: EcdsaChannelSigner> PartialEq for OnchainTxHandler<ChannelSigner> {
#[rustfmt::skip]
fn eq(&self, other: &Self) -> bool {
self.channel_value_satoshis == other.channel_value_satoshis &&
self.channel_keys_id == other.channel_keys_id &&
self.destination_script == other.destination_script &&
self.holder_commitment == other.holder_commitment &&
self.prev_holder_commitment == other.prev_holder_commitment &&
self.channel_transaction_parameters == other.channel_transaction_parameters &&
self.pending_claim_requests == other.pending_claim_requests &&
self.claimable_outpoints == other.claimable_outpoints &&
self.locktimed_packages == other.locktimed_packages &&
self.onchain_events_awaiting_threshold_conf == other.onchain_events_awaiting_threshold_conf
}
}
const SERIALIZATION_VERSION: u8 = 1;
const MIN_SERIALIZATION_VERSION: u8 = 1;
impl<ChannelSigner: EcdsaChannelSigner> OnchainTxHandler<ChannelSigner> {
#[rustfmt::skip]
pub(crate) fn write<W: Writer>(&self, writer: &mut W) -> Result<(), io::Error> {
write_ver_prefix!(writer, SERIALIZATION_VERSION, MIN_SERIALIZATION_VERSION);
self.destination_script.write(writer)?;
self.holder_commitment.write(writer)?;
None::<Option<Vec<Option<(usize, Signature)>>>>.write(writer)?; self.prev_holder_commitment.write(writer)?;
None::<Option<Vec<Option<(usize, Signature)>>>>.write(writer)?;
self.channel_transaction_parameters.write(writer)?;
0u32.write(writer)?;
writer.write_all(&(self.pending_claim_requests.len() as u64).to_be_bytes())?;
for (ref ancestor_claim_txid, request) in self.pending_claim_requests.iter() {
ancestor_claim_txid.write(writer)?;
request.write(writer)?;
}
writer.write_all(&(self.claimable_outpoints.len() as u64).to_be_bytes())?;
for (ref outp, ref claim_and_height) in self.claimable_outpoints.iter() {
outp.write(writer)?;
claim_and_height.0.write(writer)?;
claim_and_height.1.write(writer)?;
}
writer.write_all(&(self.locktimed_packages.len() as u64).to_be_bytes())?;
for (ref locktime, ref packages) in self.locktimed_packages.iter() {
locktime.write(writer)?;
writer.write_all(&(packages.len() as u64).to_be_bytes())?;
for ref package in packages.iter() {
package.write(writer)?;
}
}
writer.write_all(&(self.onchain_events_awaiting_threshold_conf.len() as u64).to_be_bytes())?;
for ref entry in self.onchain_events_awaiting_threshold_conf.iter() {
entry.write(writer)?;
}
write_tlv_fields!(writer, {});
Ok(())
}
}
impl<'a, 'b, ES: EntropySource, SP: SignerProvider> ReadableArgs<(&'a ES, &'b SP, u64, [u8; 32])>
for OnchainTxHandler<SP::EcdsaSigner>
{
#[rustfmt::skip]
fn read<R: io::Read>(reader: &mut R, args: (&'a ES, &'b SP, u64, [u8; 32])) -> Result<Self, DecodeError> {
let entropy_source = args.0;
let signer_provider = args.1;
let channel_value_satoshis = args.2;
let channel_keys_id = args.3;
let _ver = read_ver_prefix!(reader, SERIALIZATION_VERSION);
let destination_script = Readable::read(reader)?;
let holder_commitment = Readable::read(reader)?;
let _holder_htlc_sigs: Option<Vec<Option<(usize, Signature)>>> = Readable::read(reader)?;
let prev_holder_commitment = Readable::read(reader)?;
let _prev_holder_htlc_sigs: Option<Vec<Option<(usize, Signature)>>> = Readable::read(reader)?;
let channel_parameters = ReadableArgs::<Option<u64>>::read(reader, Some(channel_value_satoshis))?;
let keys_len: u32 = Readable::read(reader)?;
let mut bytes_read = 0;
while bytes_read != keys_len as usize {
let mut data = [0; 1024];
let bytes_to_read = cmp::min(1024, keys_len as usize - bytes_read);
let read_slice = &mut data[0..bytes_to_read];
reader.read_exact(read_slice)?;
bytes_read += bytes_to_read;
}
let signer = signer_provider.derive_channel_signer(channel_keys_id);
let pending_claim_requests_len: u64 = Readable::read(reader)?;
let mut pending_claim_requests = hash_map_with_capacity(cmp::min(pending_claim_requests_len as usize, MAX_ALLOC_SIZE / 128));
for _ in 0..pending_claim_requests_len {
pending_claim_requests.insert(Readable::read(reader)?, Readable::read(reader)?);
}
let claimable_outpoints_len: u64 = Readable::read(reader)?;
let mut claimable_outpoints = hash_map_with_capacity(cmp::min(pending_claim_requests_len as usize, MAX_ALLOC_SIZE / 128));
for _ in 0..claimable_outpoints_len {
let outpoint = Readable::read(reader)?;
let ancestor_claim_txid = Readable::read(reader)?;
let height = Readable::read(reader)?;
claimable_outpoints.insert(outpoint, (ancestor_claim_txid, height));
}
let locktimed_packages_len: u64 = Readable::read(reader)?;
let mut locktimed_packages = BTreeMap::new();
for _ in 0..locktimed_packages_len {
let locktime = Readable::read(reader)?;
let packages_len: u64 = Readable::read(reader)?;
let mut packages = Vec::with_capacity(cmp::min(packages_len as usize, MAX_ALLOC_SIZE / core::mem::size_of::<PackageTemplate>()));
for _ in 0..packages_len {
packages.push(Readable::read(reader)?);
}
locktimed_packages.insert(locktime, packages);
}
let waiting_threshold_conf_len: u64 = Readable::read(reader)?;
let mut onchain_events_awaiting_threshold_conf = Vec::with_capacity(cmp::min(waiting_threshold_conf_len as usize, MAX_ALLOC_SIZE / 128));
for _ in 0..waiting_threshold_conf_len {
if let Some(val) = MaybeReadable::read(reader)? {
onchain_events_awaiting_threshold_conf.push(val);
}
}
read_tlv_fields!(reader, {});
let mut secp_ctx = Secp256k1::new();
secp_ctx.seeded_randomize(&entropy_source.get_secure_random_bytes());
Ok(OnchainTxHandler {
channel_value_satoshis,
channel_keys_id,
destination_script,
holder_commitment,
prev_holder_commitment,
signer,
channel_transaction_parameters: channel_parameters,
claimable_outpoints,
locktimed_packages,
pending_claim_requests,
onchain_events_awaiting_threshold_conf,
pending_claim_events: Vec::new(),
secp_ctx,
})
}
}
impl<ChannelSigner: EcdsaChannelSigner> OnchainTxHandler<ChannelSigner> {
pub(crate) fn new(
channel_value_satoshis: u64, channel_keys_id: [u8; 32], destination_script: ScriptBuf,
signer: ChannelSigner, channel_parameters: ChannelTransactionParameters,
holder_commitment: HolderCommitmentTransaction, secp_ctx: Secp256k1<secp256k1::All>,
) -> Self {
OnchainTxHandler {
channel_value_satoshis,
channel_keys_id,
destination_script,
holder_commitment,
prev_holder_commitment: None,
signer,
channel_transaction_parameters: channel_parameters,
pending_claim_requests: new_hash_map(),
claimable_outpoints: new_hash_map(),
locktimed_packages: BTreeMap::new(),
onchain_events_awaiting_threshold_conf: Vec::new(),
pending_claim_events: Vec::new(),
secp_ctx,
}
}
pub(crate) fn prev_holder_commitment_tx(&self) -> Option<&HolderCommitmentTransaction> {
self.prev_holder_commitment.as_ref()
}
pub(crate) fn current_holder_commitment_tx(&self) -> &HolderCommitmentTransaction {
&self.holder_commitment
}
pub(crate) fn get_and_clear_pending_claim_events(&mut self) -> Vec<(ClaimId, ClaimEvent)> {
let mut events = Vec::new();
swap(&mut events, &mut self.pending_claim_events);
events
}
#[rustfmt::skip]
pub(super) fn rebroadcast_pending_claims<B: Deref, F: Deref, L: Logger>(
&mut self, current_height: u32, feerate_strategy: FeerateStrategy, broadcaster: &B,
conf_target: ConfirmationTarget, destination_script: &Script,
fee_estimator: &LowerBoundedFeeEstimator<F>, logger: &L,
)
where
B::Target: BroadcasterInterface,
F::Target: FeeEstimator,
{
let mut bump_requests = Vec::with_capacity(self.pending_claim_requests.len());
for (claim_id, request) in self.pending_claim_requests.iter() {
let inputs = request.outpoints();
log_info!(logger, "Triggering rebroadcast/fee-bump for request with inputs {:?}", inputs);
bump_requests.push((*claim_id, request.clone()));
}
for (claim_id, request) in bump_requests {
self.generate_claim(
current_height, &request, &feerate_strategy, conf_target, destination_script,
fee_estimator, logger,
)
.map(|(_, new_feerate, claim)| {
let mut feerate_was_bumped = false;
if let Some(mut_request) = self.pending_claim_requests.get_mut(&claim_id) {
feerate_was_bumped = new_feerate > request.previous_feerate();
mut_request.set_feerate(new_feerate);
}
match claim {
OnchainClaim::Tx(tx) => {
if tx.is_fully_signed() {
let log_start = if feerate_was_bumped { "Broadcasting RBF-bumped" } else { "Rebroadcasting" };
log_info!(logger, "{} onchain {}", log_start, log_tx!(tx.0));
broadcaster.broadcast_transactions(&[&tx.0]);
} else {
log_info!(logger, "Waiting for signature of unsigned onchain transaction {}", tx.0.compute_txid());
}
},
OnchainClaim::Event(event) => {
let log_start = if feerate_was_bumped { "Yielding fee-bumped" } else { "Replaying" };
log_info!(logger, "{} onchain event to spend inputs {:?}", log_start,
request.outpoints());
#[cfg(debug_assertions)] {
debug_assert!(request.requires_external_funding());
let num_existing = self.pending_claim_events.iter()
.filter(|entry| entry.0 == claim_id).count();
assert!(num_existing == 0 || num_existing == 1);
}
self.pending_claim_events.retain(|event| event.0 != claim_id);
self.pending_claim_events.push((claim_id, event));
}
}
});
}
}
pub(super) fn has_pending_claims(&self) -> bool {
self.pending_claim_requests.len() != 0
}
#[rustfmt::skip]
fn generate_claim<F: Deref, L: Logger>(
&mut self, cur_height: u32, cached_request: &PackageTemplate,
feerate_strategy: &FeerateStrategy, conf_target: ConfirmationTarget,
destination_script: &Script, fee_estimator: &LowerBoundedFeeEstimator<F>, logger: &L,
) -> Option<(u32, u64, OnchainClaim)>
where F::Target: FeeEstimator,
{
let request_outpoints = cached_request.outpoints();
if request_outpoints.is_empty() {
debug_assert!(cached_request.is_malleable());
return None;
}
let mut all_inputs_have_confirmed_spend = true;
for outpoint in request_outpoints.iter() {
if let Some((request_claim_id, _)) = self.claimable_outpoints.get(*outpoint) {
if !self.onchain_events_awaiting_threshold_conf.iter()
.any(|event_entry| if let OnchainEvent::Claim { claim_id } = event_entry.event {
*request_claim_id == claim_id
} else {
false
})
{
all_inputs_have_confirmed_spend = false;
}
} else {
all_inputs_have_confirmed_spend = false;
}
}
if all_inputs_have_confirmed_spend {
return None;
}
let new_timer = cached_request.get_height_timer(cur_height);
if cached_request.is_malleable() {
if cached_request.requires_external_funding() {
let target_feerate_sat_per_1000_weight = cached_request.compute_package_feerate(
fee_estimator, conf_target, feerate_strategy,
);
let htlcs = cached_request.construct_malleable_package_with_external_funding(self)?;
return Some((
new_timer,
target_feerate_sat_per_1000_weight as u64,
OnchainClaim::Event(ClaimEvent::BumpHTLC {
target_feerate_sat_per_1000_weight,
htlcs,
tx_lock_time: LockTime::from_consensus(cached_request.package_locktime(cur_height)),
}),
));
}
let predicted_weight = cached_request.package_weight(destination_script);
if let Some((output_value, new_feerate)) = cached_request.compute_package_output(
predicted_weight, destination_script.minimal_non_dust().to_sat(),
feerate_strategy, conf_target, fee_estimator, logger,
) {
assert!(new_feerate != 0);
let transaction = cached_request.maybe_finalize_malleable_package(
cur_height, self, Amount::from_sat(output_value), destination_script.into(), logger
).unwrap();
assert!(predicted_weight >= transaction.0.weight().to_wu());
return Some((new_timer, new_feerate, OnchainClaim::Tx(transaction)));
}
} else {
let mut inputs = cached_request.inputs();
debug_assert_eq!(inputs.len(), 1);
if !cached_request.requires_external_funding() {
return cached_request.maybe_finalize_untractable_package(self, logger)
.map(|tx| (new_timer, 0, OnchainClaim::Tx(tx)))
}
return inputs.find_map(|input| match input {
PackageSolvingData::HolderFundingOutput(output) => {
let maybe_signed_commitment_tx = output.get_maybe_signed_commitment_tx(self);
let tx = if maybe_signed_commitment_tx.is_fully_signed() {
maybe_signed_commitment_tx.0
} else {
return Some((new_timer, 0, OnchainClaim::Tx(maybe_signed_commitment_tx)));
};
let holder_commitment = output.commitment_tx.as_ref()
.unwrap_or(self.current_holder_commitment_tx());
let input_amount_sats = if let Some(funding_amount_sats) = output.funding_amount_sats {
funding_amount_sats
} else {
debug_assert!(false, "Funding amount should always exist for anchor-based claims");
self.channel_value_satoshis
};
let fee_sat = input_amount_sats - tx.output.iter()
.map(|output| output.value.to_sat()).sum::<u64>();
let package_target_feerate_sat_per_1000_weight = cached_request
.compute_package_feerate(fee_estimator, conf_target, feerate_strategy);
let channel_parameters = output.channel_parameters.as_ref()
.unwrap_or(self.channel_parameters());
let funding_pubkey = &channel_parameters.holder_pubkeys.funding_pubkey;
let script_pubkey = if channel_parameters.channel_type_features.supports_anchors_zero_fee_htlc_tx() {
get_keyed_anchor_redeemscript(funding_pubkey).to_p2wsh()
} else {
debug_assert!(channel_parameters.channel_type_features.supports_anchor_zero_fee_commitments());
shared_anchor_script_pubkey()
};
let anchor_output = tx.output.iter().enumerate()
.find(|(_, txout)| txout.script_pubkey == script_pubkey)
.map(|(idx, txout)| (idx as u32, txout));
match anchor_output {
Some((idx, _)) => {
Some((
new_timer,
package_target_feerate_sat_per_1000_weight as u64,
OnchainClaim::Event(ClaimEvent::BumpCommitment {
package_target_feerate_sat_per_1000_weight,
commitment_tx: tx,
pending_nondust_htlcs: holder_commitment.nondust_htlcs().to_vec(),
commitment_tx_fee_satoshis: fee_sat,
anchor_output_idx: idx,
channel_parameters: channel_parameters.clone(),
}),
))
},
None => Some((new_timer, 0, OnchainClaim::Tx(MaybeSignedTransaction(tx)))),
}
},
_ => {
debug_assert!(false, "Only HolderFundingOutput inputs should be untractable and require external funding");
None
},
})
}
None
}
#[rustfmt::skip]
pub fn abandon_claim(&mut self, outpoint: &BitcoinOutPoint) -> bool {
let mut found_claim = false;
let claim_id = self.claimable_outpoints.get(outpoint).map(|(claim_id, _)| *claim_id)
.or_else(|| {
self.pending_claim_requests.iter()
.find(|(_, claim)| claim.outpoints().contains(&outpoint))
.map(|(claim_id, _)| *claim_id)
});
if let Some(claim_id) = claim_id {
if let Some(claim) = self.pending_claim_requests.remove(&claim_id) {
for outpoint in claim.outpoints() {
if self.claimable_outpoints.remove(outpoint).is_some() {
found_claim = true;
}
}
}
} else {
self.locktimed_packages.values_mut().for_each(|claims| {
claims.retain(|claim| {
let includes_outpoint = claim.outpoints().contains(&outpoint);
if includes_outpoint {
found_claim = true;
}
!includes_outpoint
})
});
}
found_claim
}
#[rustfmt::skip]
pub(super) fn update_claims_view_from_requests<B: Deref, F: Deref, L: Logger>(
&mut self, mut requests: Vec<PackageTemplate>, conf_height: u32, cur_height: u32,
broadcaster: &B, conf_target: ConfirmationTarget, destination_script: &Script,
fee_estimator: &LowerBoundedFeeEstimator<F>, logger: &L
) where
B::Target: BroadcasterInterface,
F::Target: FeeEstimator,
{
if !requests.is_empty() {
log_debug!(logger, "Updating claims view at height {} with {} claim requests", cur_height, requests.len());
}
requests.retain(|req| {
debug_assert_eq!(
req.outpoints().len(),
1,
"Claims passed to `update_claims_view_from_requests` should not be aggregated"
);
let mut all_outpoints_claiming = true;
for outpoint in req.outpoints() {
if self.claimable_outpoints.get(outpoint).is_none() {
all_outpoints_claiming = false;
}
}
if all_outpoints_claiming {
log_info!(logger, "Ignoring second claim for outpoint {}:{}, already registered its claiming request",
req.outpoints()[0].txid, req.outpoints()[0].vout);
false
} else {
let timelocked_equivalent_package = self.locktimed_packages.iter().map(|v| v.1.iter()).flatten()
.find(|locked_package| locked_package.outpoints() == req.outpoints());
if let Some(package) = timelocked_equivalent_package {
log_info!(logger, "Ignoring second claim for outpoint {}:{}, we already have one which we're waiting on a timelock at {} for.",
req.outpoints()[0].txid, req.outpoints()[0].vout, package.package_locktime(cur_height));
false
} else {
true
}
}
});
for i in (1..requests.len()).rev() {
for j in 0..i {
if requests[i].can_merge_with(&requests[j], cur_height) {
let merge = requests.remove(i);
if let Err(rejected) = requests[j].merge_package(merge, cur_height) {
debug_assert!(false, "Merging package should not be rejected after verifying can_merge_with.");
requests.insert(i, rejected);
} else {
break;
}
}
}
}
let mut preprocessed_requests = Vec::with_capacity(requests.len());
for req in requests {
let package_locktime = req.package_locktime(cur_height);
if package_locktime > cur_height {
log_info!(logger, "Delaying claim of package until its timelock at {} (current height {}), the following outpoints are spent:", package_locktime, cur_height);
for outpoint in req.outpoints() {
log_info!(logger, " Outpoint {}", outpoint);
}
self.locktimed_packages.entry(package_locktime).or_default().push(req);
} else {
preprocessed_requests.push(req);
}
}
let remaining_locked_packages = self.locktimed_packages.split_off(&(cur_height + 1));
if !self.locktimed_packages.is_empty() {
log_debug!(logger,
"Updating claims view at height {} with {} locked packages available for claim",
cur_height,
self.locktimed_packages.len());
}
for (pop_height, mut entry) in self.locktimed_packages.iter_mut() {
log_trace!(logger, "Restoring delayed claim of package(s) at their timelock at {}.", pop_height);
preprocessed_requests.append(&mut entry);
}
self.locktimed_packages = remaining_locked_packages;
for mut req in preprocessed_requests {
if let Some((new_timer, new_feerate, claim)) = self.generate_claim(
cur_height, &req, &FeerateStrategy::ForceBump, conf_target, destination_script,
&*fee_estimator, &*logger,
) {
req.set_timer(new_timer);
req.set_feerate(new_feerate);
let claim_id = match claim {
OnchainClaim::Tx(tx) => {
if tx.is_fully_signed() {
log_info!(logger, "Broadcasting onchain {}", log_tx!(tx.0));
broadcaster.broadcast_transactions(&[&tx.0]);
} else {
log_info!(logger, "Waiting for signature of unsigned onchain transaction {}", tx.0.compute_txid());
}
ClaimId(tx.0.compute_txid().to_byte_array())
},
OnchainClaim::Event(claim_event) => {
log_info!(logger, "Yielding onchain event to spend inputs {:?}", req.outpoints());
let claim_id = match claim_event {
ClaimEvent::BumpCommitment { ref commitment_tx, .. } =>
ClaimId(commitment_tx.compute_txid().to_byte_array()),
ClaimEvent::BumpHTLC { ref htlcs, .. } => {
ClaimId::from_htlcs(htlcs)
},
};
debug_assert!(self.pending_claim_requests.get(&claim_id).is_none());
debug_assert_eq!(self.pending_claim_events.iter().filter(|entry| entry.0 == claim_id).count(), 0);
self.pending_claim_events.push((claim_id, claim_event));
claim_id
},
};
debug_assert!(cfg!(fuzzing) || self.pending_claim_requests.get(&claim_id).is_none());
for (k, outpoint_confirmation_height) in req.outpoints_and_creation_heights() {
let creation_height = outpoint_confirmation_height.unwrap_or(conf_height);
log_info!(logger, "Registering claiming request for {}:{}, which exists as of height {creation_height}", k.txid, k.vout);
self.claimable_outpoints.insert(k.clone(), (claim_id, creation_height));
}
self.pending_claim_requests.insert(claim_id, req);
}
}
}
#[rustfmt::skip]
pub(super) fn update_claims_view_from_matched_txn<B: Deref, F: Deref, L: Logger>(
&mut self, txn_matched: &[&Transaction], conf_height: u32, conf_hash: BlockHash,
cur_height: u32, broadcaster: &B, conf_target: ConfirmationTarget,
destination_script: &Script, fee_estimator: &LowerBoundedFeeEstimator<F>, logger: &L
) where
B::Target: BroadcasterInterface,
F::Target: FeeEstimator,
{
let mut have_logged_intro = false;
let mut maybe_log_intro = || {
if !have_logged_intro {
log_debug!(logger, "Updating claims view at height {} with {} matched transactions in block {}", cur_height, txn_matched.len(), conf_height);
have_logged_intro = true;
}
};
let mut bump_candidates = new_hash_map();
if !txn_matched.is_empty() { maybe_log_intro(); }
for tx in txn_matched {
let mut claimed_outputs_material = Vec::new();
for inp in &tx.input {
if let Some((claim_id, _)) = self.claimable_outpoints.get(&inp.previous_output) {
if let Some(request) = self.pending_claim_requests.get_mut(claim_id) {
let mut is_claim_subset_of_tx = true;
let mut tx_inputs = tx.input.iter().map(|input| &input.previous_output).collect::<Vec<_>>();
tx_inputs.sort_unstable();
for request_input in request.outpoints() {
if tx_inputs.binary_search(&request_input).is_err() {
is_claim_subset_of_tx = false;
break;
}
}
macro_rules! clean_claim_request_after_safety_delay {
() => {
let entry = OnchainEventEntry {
txid: tx.compute_txid(),
height: conf_height,
block_hash: Some(conf_hash),
event: OnchainEvent::Claim { claim_id: *claim_id }
};
if !self.onchain_events_awaiting_threshold_conf.contains(&entry) {
self.onchain_events_awaiting_threshold_conf.push(entry);
}
}
}
if is_claim_subset_of_tx {
clean_claim_request_after_safety_delay!();
} else { let mut at_least_one_drop = false;
for input in tx.input.iter() {
if let Some(package) = request.split_package(&input.previous_output) {
claimed_outputs_material.push(package);
at_least_one_drop = true;
}
if request.outpoints().is_empty() {
clean_claim_request_after_safety_delay!();
}
}
if at_least_one_drop {
bump_candidates.insert(*claim_id, request.clone());
#[cfg(debug_assertions)] {
let existing = self.pending_claim_events.iter()
.filter(|entry| entry.0 == *claim_id).count();
assert!(existing == 0 || existing == 1);
}
self.pending_claim_events.retain(|entry| entry.0 != *claim_id);
}
}
} else {
panic!("Inconsistencies between pending_claim_requests map and claimable_outpoints map");
}
}
self.locktimed_packages.retain(|_locktime, packages|{
packages.retain_mut(|package| {
if let Some(p) = package.split_package(&inp.previous_output) {
claimed_outputs_material.push(p);
}
!package.outpoints().is_empty()
});
!packages.is_empty()
});
}
for package in claimed_outputs_material.drain(..) {
let entry = OnchainEventEntry {
txid: tx.compute_txid(),
height: conf_height,
block_hash: Some(conf_hash),
event: OnchainEvent::ContentiousOutpoint { package },
};
if !self.onchain_events_awaiting_threshold_conf.contains(&entry) {
self.onchain_events_awaiting_threshold_conf.push(entry);
}
}
}
let onchain_events_awaiting_threshold_conf =
self.onchain_events_awaiting_threshold_conf.drain(..).collect::<Vec<_>>();
for entry in onchain_events_awaiting_threshold_conf {
if entry.has_reached_confirmation_threshold(cur_height) {
maybe_log_intro();
match entry.event {
OnchainEvent::Claim { claim_id } => {
if let Some(request) = self.pending_claim_requests.remove(&claim_id) {
for outpoint in request.outpoints() {
log_debug!(logger, "Removing claim tracking for {} due to maturation of claim package {}.",
outpoint, log_bytes!(claim_id.0));
self.claimable_outpoints.remove(outpoint);
}
#[cfg(debug_assertions)] {
let num_existing = self.pending_claim_events.iter()
.filter(|entry| entry.0 == claim_id).count();
assert!(num_existing == 0 || num_existing == 1);
}
self.pending_claim_events.retain(|(id, _)| *id != claim_id);
}
},
OnchainEvent::ContentiousOutpoint { package } => {
log_debug!(logger, "Removing claim tracking due to maturation of claim tx for outpoints:");
log_debug!(logger, " {:?}", package.outpoints());
self.claimable_outpoints.remove(package.outpoints()[0]);
}
}
} else {
self.onchain_events_awaiting_threshold_conf.push(entry);
}
}
for (claim_id, request) in self.pending_claim_requests.iter() {
if cur_height >= request.timer() {
bump_candidates.insert(*claim_id, request.clone());
}
}
if !bump_candidates.is_empty() {
maybe_log_intro();
log_trace!(logger, "Bumping {} candidates", bump_candidates.len());
}
for (claim_id, request) in bump_candidates.iter() {
if let Some((new_timer, new_feerate, bump_claim)) = self.generate_claim(
cur_height, &request, &FeerateStrategy::ForceBump, conf_target, destination_script,
&*fee_estimator, &*logger,
) {
match bump_claim {
OnchainClaim::Tx(bump_tx) => {
if bump_tx.is_fully_signed() {
log_info!(logger, "Broadcasting RBF-bumped onchain {}", log_tx!(bump_tx.0));
broadcaster.broadcast_transactions(&[&bump_tx.0]);
} else {
log_info!(logger, "Waiting for signature of RBF-bumped unsigned onchain transaction {}",
bump_tx.0.compute_txid());
}
},
OnchainClaim::Event(claim_event) => {
log_info!(logger, "Yielding RBF-bumped onchain event to spend inputs {:?}", request.outpoints());
#[cfg(debug_assertions)] {
let num_existing = self.pending_claim_events.iter().
filter(|entry| entry.0 == *claim_id).count();
assert!(num_existing == 0 || num_existing == 1);
}
self.pending_claim_events.retain(|event| event.0 != *claim_id);
self.pending_claim_events.push((*claim_id, claim_event));
},
}
if let Some(request) = self.pending_claim_requests.get_mut(claim_id) {
request.set_timer(new_timer);
request.set_feerate(new_feerate);
}
}
}
}
#[rustfmt::skip]
pub(super) fn transaction_unconfirmed<B: Deref, F: Deref, L: Logger>(
&mut self,
txid: &Txid,
broadcaster: &B,
conf_target: ConfirmationTarget,
destination_script: &Script,
fee_estimator: &LowerBoundedFeeEstimator<F>,
logger: &L,
) where
B::Target: BroadcasterInterface,
F::Target: FeeEstimator,
{
let mut height = None;
for entry in self.onchain_events_awaiting_threshold_conf.iter() {
if entry.txid == *txid {
height = Some(entry.height);
break;
}
}
if let Some(height) = height {
self.blocks_disconnected(
height - 1, broadcaster, conf_target, destination_script, fee_estimator, logger,
);
}
}
#[rustfmt::skip]
pub(super) fn blocks_disconnected<B: Deref, F: Deref, L: Logger>(
&mut self, new_best_height: u32, broadcaster: &B, conf_target: ConfirmationTarget,
destination_script: &Script, fee_estimator: &LowerBoundedFeeEstimator<F>, logger: &L,
)
where B::Target: BroadcasterInterface,
F::Target: FeeEstimator,
{
let mut bump_candidates = new_hash_map();
let onchain_events_awaiting_threshold_conf =
self.onchain_events_awaiting_threshold_conf.drain(..).collect::<Vec<_>>();
for entry in onchain_events_awaiting_threshold_conf {
if entry.height > new_best_height {
match entry.event {
OnchainEvent::ContentiousOutpoint { package } => {
let package_locktime = package.package_locktime(0);
if package_locktime > new_best_height {
self.locktimed_packages.entry(package_locktime).or_default().push(package);
continue;
}
if let Some(pending_claim) = self.claimable_outpoints.get(package.outpoints()[0]) {
if let Some(request) = self.pending_claim_requests.get_mut(&pending_claim.0) {
assert!(request.merge_package(package, new_best_height + 1).is_ok());
bump_candidates.insert(pending_claim.clone(), request.clone());
}
}
},
_ => {},
}
} else {
self.onchain_events_awaiting_threshold_conf.push(entry);
}
}
for ((_claim_id, _), ref mut request) in bump_candidates.iter_mut() {
if let Some((new_timer, new_feerate, bump_claim)) = self.generate_claim(
new_best_height, &request, &FeerateStrategy::ForceBump, conf_target,
destination_script, fee_estimator, logger
) {
request.set_timer(new_timer);
request.set_feerate(new_feerate);
match bump_claim {
OnchainClaim::Tx(bump_tx) => {
if bump_tx.is_fully_signed() {
log_info!(logger, "Broadcasting onchain {}", log_tx!(bump_tx.0));
broadcaster.broadcast_transactions(&[&bump_tx.0]);
} else {
log_info!(logger, "Waiting for signature of unsigned onchain transaction {}", bump_tx.0.compute_txid());
}
},
OnchainClaim::Event(claim_event) => {
log_info!(logger, "Yielding onchain event after reorg to spend inputs {:?}", request.outpoints());
#[cfg(debug_assertions)] {
let num_existing = self.pending_claim_events.iter()
.filter(|entry| entry.0 == *_claim_id).count();
assert!(num_existing == 0 || num_existing == 1);
}
self.pending_claim_events.retain(|event| event.0 != *_claim_id);
self.pending_claim_events.push((*_claim_id, claim_event));
},
}
}
}
for (ancestor_claim_txid, request) in bump_candidates.drain() {
self.pending_claim_requests.insert(ancestor_claim_txid.0, request);
}
let mut remove_request = Vec::new();
self.claimable_outpoints.retain(|_, ref v|
if v.1 > new_best_height {
remove_request.push(v.0.clone());
false
} else { true });
for req in remove_request {
self.pending_claim_requests.remove(&req);
}
}
pub(crate) fn is_output_spend_pending(&self, outpoint: &BitcoinOutPoint) -> bool {
self.claimable_outpoints.get(outpoint).is_some()
}
#[rustfmt::skip]
pub(crate) fn get_relevant_txids(&self) -> Vec<(Txid, u32, Option<BlockHash>)> {
let mut txids: Vec<(Txid, u32, Option<BlockHash>)> = self.onchain_events_awaiting_threshold_conf
.iter()
.map(|entry| (entry.txid, entry.height, entry.block_hash))
.collect();
txids.sort_unstable_by(|a, b| a.0.cmp(&b.0).then(b.1.cmp(&a.1)));
txids.dedup_by_key(|(txid, _, _)| *txid);
txids
}
pub(crate) fn provide_latest_holder_tx(&mut self, tx: HolderCommitmentTransaction) {
self.prev_holder_commitment = Some(replace(&mut self.holder_commitment, tx));
}
pub(crate) fn update_after_renegotiated_funding_locked(
&mut self, channel_parameters: ChannelTransactionParameters,
current: HolderCommitmentTransaction, prev: Option<HolderCommitmentTransaction>,
) {
self.channel_value_satoshis = channel_parameters.channel_value_satoshis;
self.channel_transaction_parameters = channel_parameters;
self.holder_commitment = current;
self.prev_holder_commitment = prev;
}
pub(crate) fn channel_parameters(&self) -> &ChannelTransactionParameters {
&self.channel_transaction_parameters
}
pub(crate) fn channel_keys_id(&self) -> [u8; 32] {
self.channel_keys_id
}
}
#[cfg(test)]
mod tests {
use bitcoin::hash_types::Txid;
use bitcoin::hashes::sha256::Hash as Sha256;
use bitcoin::hashes::Hash;
use bitcoin::Network;
use bitcoin::{key::Secp256k1, secp256k1::PublicKey, secp256k1::SecretKey, ScriptBuf};
use types::features::ChannelTypeFeatures;
use crate::chain::chaininterface::{ConfirmationTarget, LowerBoundedFeeEstimator};
use crate::chain::package::{HolderHTLCOutput, PackageSolvingData, PackageTemplate};
use crate::chain::transaction::OutPoint;
use crate::ln::chan_utils::{
ChannelPublicKeys, ChannelTransactionParameters, CounterpartyChannelTransactionParameters,
HTLCOutputInCommitment, HolderCommitmentTransaction,
};
use crate::ln::channel_keys::{DelayedPaymentBasepoint, HtlcBasepoint, RevocationBasepoint};
use crate::ln::functional_test_utils::create_dummy_block;
use crate::sign::{ChannelDerivationParameters, ChannelSigner, HTLCDescriptor, InMemorySigner};
use crate::types::payment::{PaymentHash, PaymentPreimage};
use crate::util::test_utils::{TestBroadcaster, TestFeeEstimator, TestLogger};
use super::OnchainTxHandler;
#[test]
#[rustfmt::skip]
fn test_broadcast_height() {
let secp_ctx = Secp256k1::new();
let signer = InMemorySigner::new(
SecretKey::from_slice(&[41; 32]).unwrap(),
SecretKey::from_slice(&[41; 32]).unwrap(),
SecretKey::from_slice(&[41; 32]).unwrap(),
SecretKey::from_slice(&[41; 32]).unwrap(),
true,
SecretKey::from_slice(&[41; 32]).unwrap(),
SecretKey::from_slice(&[41; 32]).unwrap(),
[41; 32],
[0; 32],
[0; 32],
);
let counterparty_pubkeys = ChannelPublicKeys {
funding_pubkey: PublicKey::from_secret_key(
&secp_ctx,
&SecretKey::from_slice(&[44; 32]).unwrap(),
),
revocation_basepoint: RevocationBasepoint::from(PublicKey::from_secret_key(
&secp_ctx,
&SecretKey::from_slice(&[45; 32]).unwrap(),
)),
payment_point: PublicKey::from_secret_key(
&secp_ctx,
&SecretKey::from_slice(&[46; 32]).unwrap(),
),
delayed_payment_basepoint: DelayedPaymentBasepoint::from(PublicKey::from_secret_key(
&secp_ctx,
&SecretKey::from_slice(&[47; 32]).unwrap(),
)),
htlc_basepoint: HtlcBasepoint::from(PublicKey::from_secret_key(
&secp_ctx,
&SecretKey::from_slice(&[48; 32]).unwrap(),
)),
};
let funding_outpoint = OutPoint { txid: Txid::all_zeros(), index: u16::MAX };
let chan_params = ChannelTransactionParameters {
holder_pubkeys: signer.pubkeys(&secp_ctx),
holder_selected_contest_delay: 66,
is_outbound_from_holder: true,
counterparty_parameters: Some(CounterpartyChannelTransactionParameters {
pubkeys: counterparty_pubkeys,
selected_contest_delay: 67,
}),
funding_outpoint: Some(funding_outpoint),
splice_parent_funding_txid: None,
channel_type_features: ChannelTypeFeatures::only_static_remote_key(),
channel_value_satoshis: 0,
};
let mut nondust_htlcs = Vec::new();
for i in 0..3 {
let preimage = PaymentPreimage([i; 32]);
let hash = PaymentHash(Sha256::hash(&preimage.0[..]).to_byte_array());
nondust_htlcs.push(
HTLCOutputInCommitment {
offered: true,
amount_msat: 10000,
cltv_expiry: i as u32,
payment_hash: hash,
transaction_output_index: Some(i as u32),
}
);
}
let holder_commit = HolderCommitmentTransaction::dummy(1000000, funding_outpoint, nondust_htlcs);
let destination_script = ScriptBuf::new();
let mut tx_handler = OnchainTxHandler::new(
1000000,
[0; 32],
destination_script.clone(),
signer,
chan_params,
holder_commit,
secp_ctx,
);
let broadcaster = TestBroadcaster::new(Network::Testnet);
{
let mut blocks = broadcaster.blocks.lock().unwrap();
let genesis_hash = blocks[0].0.block_hash();
blocks.push((create_dummy_block(genesis_hash, 0, Vec::new()), 1));
}
let fee_estimator = TestFeeEstimator::new(253);
let fee_estimator = LowerBoundedFeeEstimator::new(&fee_estimator);
let logger = TestLogger::new();
let holder_commit = tx_handler.current_holder_commitment_tx();
let holder_commit_txid = holder_commit.trust().txid();
let mut requests = Vec::new();
for (htlc, counterparty_sig) in holder_commit.nondust_htlcs().iter().zip(holder_commit.counterparty_htlc_sigs.iter()) {
requests.push(PackageTemplate::build_package(
holder_commit_txid,
htlc.transaction_output_index.unwrap(),
PackageSolvingData::HolderHTLCOutput(HolderHTLCOutput::build(HTLCDescriptor {
channel_derivation_parameters: ChannelDerivationParameters {
value_satoshis: tx_handler.channel_value_satoshis,
keys_id: tx_handler.channel_keys_id,
transaction_parameters: tx_handler.channel_transaction_parameters.clone(),
},
commitment_txid: holder_commit_txid,
per_commitment_number: holder_commit.commitment_number(),
per_commitment_point: holder_commit.per_commitment_point(),
feerate_per_kw: holder_commit.negotiated_feerate_per_kw(),
htlc: htlc.clone(),
preimage: None,
counterparty_sig: *counterparty_sig,
},
0
)),
0,
));
}
tx_handler.update_claims_view_from_requests(
requests,
1,
1,
&&broadcaster,
ConfirmationTarget::UrgentOnChainSweep,
&destination_script,
&fee_estimator,
&logger,
);
let txs_broadcasted = broadcaster.txn_broadcast();
assert_eq!(txs_broadcasted.len(), 2);
assert!(txs_broadcasted[0].lock_time.to_consensus_u32() <= 1);
assert!(txs_broadcasted[1].lock_time.to_consensus_u32() <= 1);
{
let mut blocks = broadcaster.blocks.lock().unwrap();
let block1_hash = blocks[1].0.block_hash();
blocks.push((create_dummy_block(block1_hash, 0, Vec::new()), 2));
}
tx_handler.update_claims_view_from_requests(
Vec::new(),
2,
2,
&&broadcaster,
ConfirmationTarget::UrgentOnChainSweep,
&destination_script,
&fee_estimator,
&logger,
);
let txs_broadcasted = broadcaster.txn_broadcast();
assert_eq!(txs_broadcasted.len(), 1);
assert_eq!(txs_broadcasted[0].lock_time.to_consensus_u32(), 2);
}
}