use core::convert::TryInto;
use core::ops::Deref;
use crate::chain::chaininterface::BroadcasterInterface;
use crate::chain::ClaimId;
use crate::events::Event;
use crate::io_extras::sink;
use crate::ln::chan_utils;
use crate::ln::chan_utils::{
ANCHOR_INPUT_WITNESS_WEIGHT, HTLC_SUCCESS_INPUT_ANCHOR_WITNESS_WEIGHT,
HTLC_TIMEOUT_INPUT_ANCHOR_WITNESS_WEIGHT, ChannelTransactionParameters, HTLCOutputInCommitment
};
use crate::ln::features::ChannelTypeFeatures;
use crate::ln::PaymentPreimage;
use crate::prelude::*;
use crate::sign::{ChannelSigner, EcdsaChannelSigner, SignerProvider};
use crate::sync::Mutex;
use crate::util::logger::Logger;
use bitcoin::{OutPoint, PackedLockTime, PubkeyHash, Sequence, Script, Transaction, Txid, TxIn, TxOut, Witness, WPubkeyHash};
use bitcoin::blockdata::constants::WITNESS_SCALE_FACTOR;
use bitcoin::consensus::Encodable;
use bitcoin::secp256k1;
use bitcoin::secp256k1::{PublicKey, Secp256k1};
use bitcoin::secp256k1::ecdsa::Signature;
const EMPTY_SCRIPT_SIG_WEIGHT: u64 = 1 * WITNESS_SCALE_FACTOR as u64;
const BASE_INPUT_SIZE: u64 = 32 + 4 + 4 ;
const BASE_INPUT_WEIGHT: u64 = BASE_INPUT_SIZE * WITNESS_SCALE_FACTOR as u64;
fn compute_feerate_sat_per_1000_weight(fee_sat: u64, weight: u64) -> u32 {
(fee_sat * 1000 / weight).try_into().unwrap_or(u32::max_value())
}
const fn fee_for_weight(feerate_sat_per_1000_weight: u32, weight: u64) -> u64 {
((feerate_sat_per_1000_weight as u64 * weight) + 1000 - 1) / 1000
}
#[derive(Clone, Debug, PartialEq, Eq)]
pub struct AnchorDescriptor {
pub channel_keys_id: [u8; 32],
pub channel_value_satoshis: u64,
pub outpoint: OutPoint,
}
#[derive(Clone, Debug, PartialEq, Eq)]
pub struct HTLCDescriptor {
pub channel_keys_id: [u8; 32],
pub channel_value_satoshis: u64,
pub channel_parameters: ChannelTransactionParameters,
pub commitment_txid: Txid,
pub per_commitment_number: u64,
pub htlc: HTLCOutputInCommitment,
pub preimage: Option<PaymentPreimage>,
pub counterparty_sig: Signature
}
impl HTLCDescriptor {
pub fn unsigned_tx_input(&self) -> TxIn {
chan_utils::build_htlc_input(&self.commitment_txid, &self.htlc, &ChannelTypeFeatures::anchors_zero_htlc_fee_and_dependencies())
}
pub fn tx_output<C: secp256k1::Signing + secp256k1::Verification>(
&self, per_commitment_point: &PublicKey, secp: &Secp256k1<C>
) -> TxOut {
let channel_params = self.channel_parameters.as_holder_broadcastable();
let broadcaster_keys = channel_params.broadcaster_pubkeys();
let counterparty_keys = channel_params.countersignatory_pubkeys();
let broadcaster_delayed_key = chan_utils::derive_public_key(
secp, per_commitment_point, &broadcaster_keys.delayed_payment_basepoint
);
let counterparty_revocation_key = chan_utils::derive_public_revocation_key(
secp, per_commitment_point, &counterparty_keys.revocation_basepoint
);
chan_utils::build_htlc_output(
0 , channel_params.contest_delay(), &self.htlc,
&ChannelTypeFeatures::anchors_zero_htlc_fee_and_dependencies(), &broadcaster_delayed_key, &counterparty_revocation_key
)
}
pub fn witness_script<C: secp256k1::Signing + secp256k1::Verification>(
&self, per_commitment_point: &PublicKey, secp: &Secp256k1<C>
) -> Script {
let channel_params = self.channel_parameters.as_holder_broadcastable();
let broadcaster_keys = channel_params.broadcaster_pubkeys();
let counterparty_keys = channel_params.countersignatory_pubkeys();
let broadcaster_htlc_key = chan_utils::derive_public_key(
secp, per_commitment_point, &broadcaster_keys.htlc_basepoint
);
let counterparty_htlc_key = chan_utils::derive_public_key(
secp, per_commitment_point, &counterparty_keys.htlc_basepoint
);
let counterparty_revocation_key = chan_utils::derive_public_revocation_key(
secp, per_commitment_point, &counterparty_keys.revocation_basepoint
);
chan_utils::get_htlc_redeemscript_with_explicit_keys(
&self.htlc, &ChannelTypeFeatures::anchors_zero_htlc_fee_and_dependencies(), &broadcaster_htlc_key, &counterparty_htlc_key,
&counterparty_revocation_key,
)
}
pub fn tx_input_witness(&self, signature: &Signature, witness_script: &Script) -> Witness {
chan_utils::build_htlc_input_witness(
signature, &self.counterparty_sig, &self.preimage, witness_script, &ChannelTypeFeatures::anchors_zero_htlc_fee_and_dependencies()
)
}
}
#[derive(Clone, Debug, PartialEq, Eq)]
pub enum BumpTransactionEvent {
ChannelClose {
claim_id: ClaimId,
package_target_feerate_sat_per_1000_weight: u32,
commitment_tx: Transaction,
commitment_tx_fee_satoshis: u64,
anchor_descriptor: AnchorDescriptor,
pending_htlcs: Vec<HTLCOutputInCommitment>,
},
HTLCResolution {
claim_id: ClaimId,
target_feerate_sat_per_1000_weight: u32,
htlc_descriptors: Vec<HTLCDescriptor>,
tx_lock_time: PackedLockTime,
},
}
pub struct Input {
pub outpoint: OutPoint,
pub satisfaction_weight: u64,
}
#[derive(Clone, Debug)]
pub struct Utxo {
pub outpoint: OutPoint,
pub output: TxOut,
pub satisfaction_weight: u64,
}
impl Utxo {
const P2WPKH_WITNESS_WEIGHT: u64 = 1 +
1 +
73 +
1 +
33 ;
pub fn new_p2pkh(outpoint: OutPoint, value: u64, pubkey_hash: &PubkeyHash) -> Self {
let script_sig_size = 1 +
1 +
73 +
1 +
33 ;
Self {
outpoint,
output: TxOut {
value,
script_pubkey: Script::new_p2pkh(pubkey_hash),
},
satisfaction_weight: script_sig_size * WITNESS_SCALE_FACTOR as u64 + 1 ,
}
}
pub fn new_nested_p2wpkh(outpoint: OutPoint, value: u64, pubkey_hash: &WPubkeyHash) -> Self {
let script_sig_size = 1 +
1 +
1 +
20 ;
Self {
outpoint,
output: TxOut {
value,
script_pubkey: Script::new_p2sh(&Script::new_v0_p2wpkh(pubkey_hash).script_hash()),
},
satisfaction_weight: script_sig_size * WITNESS_SCALE_FACTOR as u64 + Self::P2WPKH_WITNESS_WEIGHT,
}
}
pub fn new_v0_p2wpkh(outpoint: OutPoint, value: u64, pubkey_hash: &WPubkeyHash) -> Self {
Self {
outpoint,
output: TxOut {
value,
script_pubkey: Script::new_v0_p2wpkh(pubkey_hash),
},
satisfaction_weight: EMPTY_SCRIPT_SIG_WEIGHT + Self::P2WPKH_WITNESS_WEIGHT,
}
}
}
pub struct CoinSelection {
confirmed_utxos: Vec<Utxo>,
change_output: Option<TxOut>,
}
pub trait CoinSelectionSource {
fn select_confirmed_utxos(
&self, claim_id: ClaimId, must_spend: &[Input], must_pay_to: &[TxOut],
target_feerate_sat_per_1000_weight: u32,
) -> Result<CoinSelection, ()>;
fn sign_tx(&self, tx: &mut Transaction) -> Result<(), ()>;
}
pub trait WalletSource {
fn list_confirmed_utxos(&self) -> Result<Vec<Utxo>, ()>;
fn get_change_script(&self) -> Result<Script, ()>;
fn sign_tx(&self, tx: &mut Transaction) -> Result<(), ()>;
}
pub struct Wallet<W: Deref> where W::Target: WalletSource {
source: W,
locked_utxos: Mutex<HashMap<OutPoint, ClaimId>>,
}
impl<W: Deref> Wallet<W> where W::Target: WalletSource {
pub fn new(source: W) -> Self {
Self { source, locked_utxos: Mutex::new(HashMap::new()) }
}
fn select_confirmed_utxos_internal(
&self, utxos: &[Utxo], claim_id: ClaimId, force_conflicting_utxo_spend: bool,
tolerate_high_network_feerates: bool, target_feerate_sat_per_1000_weight: u32,
preexisting_tx_weight: u64, target_amount_sat: u64,
) -> Result<CoinSelection, ()> {
let mut locked_utxos = self.locked_utxos.lock().unwrap();
let mut eligible_utxos = utxos.iter().filter_map(|utxo| {
if let Some(utxo_claim_id) = locked_utxos.get(&utxo.outpoint) {
if *utxo_claim_id != claim_id && !force_conflicting_utxo_spend {
return None;
}
}
let fee_to_spend_utxo = fee_for_weight(
target_feerate_sat_per_1000_weight, BASE_INPUT_WEIGHT as u64 + utxo.satisfaction_weight,
);
let should_spend = if tolerate_high_network_feerates {
utxo.output.value > fee_to_spend_utxo
} else {
utxo.output.value >= fee_to_spend_utxo * 2
};
if should_spend {
Some((utxo, fee_to_spend_utxo))
} else {
None
}
}).collect::<Vec<_>>();
eligible_utxos.sort_unstable_by_key(|(utxo, _)| utxo.output.value);
let mut selected_amount = 0;
let mut total_fees = fee_for_weight(target_feerate_sat_per_1000_weight, preexisting_tx_weight);
let mut selected_utxos = Vec::new();
for (utxo, fee_to_spend_utxo) in eligible_utxos {
if selected_amount >= target_amount_sat + total_fees {
break;
}
selected_amount += utxo.output.value;
total_fees += fee_to_spend_utxo;
selected_utxos.push(utxo.clone());
}
if selected_amount < target_amount_sat + total_fees {
return Err(());
}
for utxo in &selected_utxos {
locked_utxos.insert(utxo.outpoint, claim_id);
}
core::mem::drop(locked_utxos);
let remaining_amount = selected_amount - target_amount_sat - total_fees;
let change_script = self.source.get_change_script()?;
let change_output_fee = fee_for_weight(
target_feerate_sat_per_1000_weight,
(8 + change_script.consensus_encode(&mut sink()).unwrap() as u64) *
WITNESS_SCALE_FACTOR as u64,
);
let change_output_amount = remaining_amount.saturating_sub(change_output_fee);
let change_output = if change_output_amount < change_script.dust_value().to_sat() {
None
} else {
Some(TxOut { script_pubkey: change_script, value: change_output_amount })
};
Ok(CoinSelection {
confirmed_utxos: selected_utxos,
change_output,
})
}
}
impl<W: Deref> CoinSelectionSource for Wallet<W> where W::Target: WalletSource {
fn select_confirmed_utxos(
&self, claim_id: ClaimId, must_spend: &[Input], must_pay_to: &[TxOut],
target_feerate_sat_per_1000_weight: u32,
) -> Result<CoinSelection, ()> {
let utxos = self.source.list_confirmed_utxos()?;
const BASE_TX_SIZE: u64 = 4 + 1 + 1 + 4 ;
let total_output_size: u64 = must_pay_to.iter().map(|output|
8 + 1 + output.script_pubkey.len() as u64
).sum();
let total_satisfaction_weight: u64 = must_spend.iter().map(|input| input.satisfaction_weight).sum();
let total_input_weight = (BASE_INPUT_WEIGHT * must_spend.len() as u64) + total_satisfaction_weight;
let preexisting_tx_weight = 2 + total_input_weight +
((BASE_TX_SIZE + total_output_size) * WITNESS_SCALE_FACTOR as u64);
let target_amount_sat = must_pay_to.iter().map(|output| output.value).sum();
let do_coin_selection = |force_conflicting_utxo_spend: bool, tolerate_high_network_feerates: bool| {
self.select_confirmed_utxos_internal(
&utxos, claim_id, force_conflicting_utxo_spend, tolerate_high_network_feerates,
target_feerate_sat_per_1000_weight, preexisting_tx_weight, target_amount_sat,
)
};
do_coin_selection(false, false)
.or_else(|_| do_coin_selection(false, true))
.or_else(|_| do_coin_selection(true, false))
.or_else(|_| do_coin_selection(true, true))
}
fn sign_tx(&self, tx: &mut Transaction) -> Result<(), ()> {
self.source.sign_tx(tx)
}
}
pub struct BumpTransactionEventHandler<B: Deref, C: Deref, SP: Deref, L: Deref>
where
B::Target: BroadcasterInterface,
C::Target: CoinSelectionSource,
SP::Target: SignerProvider,
L::Target: Logger,
{
broadcaster: B,
utxo_source: C,
signer_provider: SP,
logger: L,
secp: Secp256k1<secp256k1::All>,
}
impl<B: Deref, C: Deref, SP: Deref, L: Deref> BumpTransactionEventHandler<B, C, SP, L>
where
B::Target: BroadcasterInterface,
C::Target: CoinSelectionSource,
SP::Target: SignerProvider,
L::Target: Logger,
{
pub fn new(broadcaster: B, utxo_source: C, signer_provider: SP, logger: L) -> Self {
Self {
broadcaster,
utxo_source,
signer_provider,
logger,
secp: Secp256k1::new(),
}
}
fn process_coin_selection(&self, tx: &mut Transaction, mut coin_selection: CoinSelection) {
for utxo in coin_selection.confirmed_utxos.drain(..) {
tx.input.push(TxIn {
previous_output: utxo.outpoint,
script_sig: Script::new(),
sequence: Sequence::ZERO,
witness: Witness::new(),
});
}
if let Some(change_output) = coin_selection.change_output.take() {
tx.output.push(change_output);
} else if tx.output.is_empty() {
tx.output.push(TxOut {
value: 0,
script_pubkey: Script::new_op_return(&[]),
});
}
}
fn build_anchor_tx(
&self, claim_id: ClaimId, target_feerate_sat_per_1000_weight: u32,
commitment_tx: &Transaction, anchor_descriptor: &AnchorDescriptor,
) -> Result<Transaction, ()> {
let must_spend = vec![Input {
outpoint: anchor_descriptor.outpoint,
satisfaction_weight: commitment_tx.weight() as u64 + ANCHOR_INPUT_WITNESS_WEIGHT + EMPTY_SCRIPT_SIG_WEIGHT,
}];
let coin_selection = self.utxo_source.select_confirmed_utxos(
claim_id, &must_spend, &[], target_feerate_sat_per_1000_weight,
)?;
let mut tx = Transaction {
version: 2,
lock_time: PackedLockTime::ZERO, input: vec![TxIn {
previous_output: anchor_descriptor.outpoint,
script_sig: Script::new(),
sequence: Sequence::ZERO,
witness: Witness::new(),
}],
output: vec![],
};
self.process_coin_selection(&mut tx, coin_selection);
Ok(tx)
}
fn handle_channel_close(
&self, claim_id: ClaimId, package_target_feerate_sat_per_1000_weight: u32,
commitment_tx: &Transaction, commitment_tx_fee_sat: u64, anchor_descriptor: &AnchorDescriptor,
) -> Result<(), ()> {
let commitment_tx_sat_per_1000_weight: u32 = compute_feerate_sat_per_1000_weight(
commitment_tx_fee_sat, commitment_tx.weight() as u64,
);
if commitment_tx_sat_per_1000_weight >= package_target_feerate_sat_per_1000_weight {
self.broadcaster.broadcast_transactions(&[&commitment_tx]);
return Ok(());
}
let mut anchor_tx = self.build_anchor_tx(
claim_id, package_target_feerate_sat_per_1000_weight, commitment_tx, anchor_descriptor,
)?;
debug_assert_eq!(anchor_tx.output.len(), 1);
self.utxo_source.sign_tx(&mut anchor_tx)?;
let signer = self.signer_provider.derive_channel_signer(
anchor_descriptor.channel_value_satoshis, anchor_descriptor.channel_keys_id,
);
let anchor_sig = signer.sign_holder_anchor_input(&anchor_tx, 0, &self.secp)?;
anchor_tx.input[0].witness =
chan_utils::build_anchor_input_witness(&signer.pubkeys().funding_pubkey, &anchor_sig);
self.broadcaster.broadcast_transactions(&[&commitment_tx, &anchor_tx]);
Ok(())
}
fn build_htlc_tx(
&self, claim_id: ClaimId, target_feerate_sat_per_1000_weight: u32,
htlc_descriptors: &[HTLCDescriptor], tx_lock_time: PackedLockTime,
) -> Result<(Transaction, HashMap<[u8; 32], <SP::Target as SignerProvider>::Signer>), ()> {
let mut tx = Transaction {
version: 2,
lock_time: tx_lock_time,
input: vec![],
output: vec![],
};
let mut signers = HashMap::new();
let mut must_spend = Vec::with_capacity(htlc_descriptors.len());
for htlc_descriptor in htlc_descriptors {
let signer = signers.entry(htlc_descriptor.channel_keys_id)
.or_insert_with(||
self.signer_provider.derive_channel_signer(
htlc_descriptor.channel_value_satoshis, htlc_descriptor.channel_keys_id,
)
);
let per_commitment_point = signer.get_per_commitment_point(
htlc_descriptor.per_commitment_number, &self.secp
);
let htlc_input = htlc_descriptor.unsigned_tx_input();
must_spend.push(Input {
outpoint: htlc_input.previous_output.clone(),
satisfaction_weight: EMPTY_SCRIPT_SIG_WEIGHT + if htlc_descriptor.preimage.is_some() {
HTLC_SUCCESS_INPUT_ANCHOR_WITNESS_WEIGHT
} else {
HTLC_TIMEOUT_INPUT_ANCHOR_WITNESS_WEIGHT
},
});
tx.input.push(htlc_input);
let htlc_output = htlc_descriptor.tx_output(&per_commitment_point, &self.secp);
tx.output.push(htlc_output);
}
let coin_selection = self.utxo_source.select_confirmed_utxos(
claim_id, &must_spend, &tx.output, target_feerate_sat_per_1000_weight,
)?;
self.process_coin_selection(&mut tx, coin_selection);
Ok((tx, signers))
}
fn handle_htlc_resolution(
&self, claim_id: ClaimId, target_feerate_sat_per_1000_weight: u32,
htlc_descriptors: &[HTLCDescriptor], tx_lock_time: PackedLockTime,
) -> Result<(), ()> {
let (mut htlc_tx, signers) = self.build_htlc_tx(
claim_id, target_feerate_sat_per_1000_weight, htlc_descriptors, tx_lock_time,
)?;
self.utxo_source.sign_tx(&mut htlc_tx)?;
for (idx, htlc_descriptor) in htlc_descriptors.iter().enumerate() {
let signer = signers.get(&htlc_descriptor.channel_keys_id).unwrap();
let htlc_sig = signer.sign_holder_htlc_transaction(
&htlc_tx, idx, htlc_descriptor, &self.secp
)?;
let per_commitment_point = signer.get_per_commitment_point(
htlc_descriptor.per_commitment_number, &self.secp
);
let witness_script = htlc_descriptor.witness_script(&per_commitment_point, &self.secp);
htlc_tx.input[idx].witness = htlc_descriptor.tx_input_witness(&htlc_sig, &witness_script);
}
self.broadcaster.broadcast_transactions(&[&htlc_tx]);
Ok(())
}
pub fn handle_event(&self, event: &Event) {
let event = if let Event::BumpTransaction(event) = event {
event
} else {
return;
};
match event {
BumpTransactionEvent::ChannelClose {
claim_id, package_target_feerate_sat_per_1000_weight, commitment_tx,
anchor_descriptor, commitment_tx_fee_satoshis, ..
} => {
if let Err(_) = self.handle_channel_close(
*claim_id, *package_target_feerate_sat_per_1000_weight, commitment_tx,
*commitment_tx_fee_satoshis, anchor_descriptor,
) {
log_error!(self.logger, "Failed bumping commitment transaction fee for {}",
commitment_tx.txid());
}
}
BumpTransactionEvent::HTLCResolution {
claim_id, target_feerate_sat_per_1000_weight, htlc_descriptors, tx_lock_time,
} => {
if let Err(_) = self.handle_htlc_resolution(
*claim_id, *target_feerate_sat_per_1000_weight, htlc_descriptors, *tx_lock_time,
) {
log_error!(self.logger, "Failed bumping HTLC transaction fee for commitment {}",
htlc_descriptors[0].commitment_txid);
}
}
}
}
}