Skip to content

Commit

Permalink
wip: add channelmanager dual-funding support
Browse files Browse the repository at this point in the history
  • Loading branch information
dunxen committed May 22, 2023
1 parent 7c5321f commit 5bd6e42
Show file tree
Hide file tree
Showing 2 changed files with 268 additions and 2 deletions.
254 changes: 252 additions & 2 deletions lightning/src/ln/channelmanager.rs
Original file line number Diff line number Diff line change
Expand Up @@ -28,7 +28,7 @@ use bitcoin::hash_types::{BlockHash, Txid};

use bitcoin::secp256k1::{SecretKey,PublicKey};
use bitcoin::secp256k1::Secp256k1;
use bitcoin::{LockTime, secp256k1, Sequence};
use bitcoin::{LockTime, secp256k1, Sequence, Script};

use crate::chain;
use crate::chain::{Confirm, ChannelMonitorUpdateStatus, Watch, BestBlock};
Expand All @@ -40,7 +40,7 @@ use crate::events::{Event, EventHandler, EventsProvider, MessageSendEvent, Messa
// Since this struct is returned in `list_channels` methods, expose it here in case users want to
// construct one themselves.
use crate::ln::{inbound_payment, PaymentHash, PaymentPreimage, PaymentSecret};
use crate::ln::channel::{Channel, OutboundV1Channel, InboundV1Channel, ChannelInterface, ChannelError, ChannelUpdateStatus, UpdateFulfillCommitFetch};
use crate::ln::channel::{Channel, OutboundV1Channel, InboundV1Channel, OutboundV2Channel, InboundV2Channel, ChannelInterface, ChannelError, ChannelUpdateStatus, UpdateFulfillCommitFetch};
use crate::ln::features::{ChannelFeatures, ChannelTypeFeatures, InitFeatures, NodeFeatures};
#[cfg(any(feature = "_test_utils", test))]
use crate::ln::features::InvoiceFeatures;
Expand Down Expand Up @@ -80,6 +80,8 @@ use core::ops::Deref;
pub use crate::ln::outbound_payment::{PaymentSendFailure, Retry, RetryableSendFailure, RecipientOnionFields};
use crate::ln::script::ShutdownScript;

use super::channel::DualFundingUtxo;

// We hold various information about HTLC relay in the HTLC objects in Channel itself:
//
// Upon receipt of an HTLC from a peer, we'll give it a PendingHTLCStatus indicating if it should
Expand Down Expand Up @@ -556,6 +558,27 @@ pub(super) struct PeerState<Signer: ChannelSigner> {
/// been assigned a `channel_id`, the entry in this map is removed and one is created in
/// `channel_by_id`.
pub(super) inbound_v1_channel_by_id: HashMap<[u8; 32], InboundV1Channel<Signer>>,
/// `(temporary_)channel_id` -> `OutboundV2Channel`.
///
/// Holds all outbound V2 channels where the peer is the counterparty. V2 channels are assigned
/// a `channel_id` before a funding transaction is created interactively as it's derived from
/// both parties' revocation basepoints once these are known. Hence, this map's keys are either
/// temporary channel IDs or channel IDs.
///
/// The entries in this map are only moved to `channel_by_id` once interactive transaction
/// construction completes successfully.
pub(super) outbound_v2_channel_by_id: HashMap<[u8; 32], OutboundV2Channel<Signer>>,
/// `channel_id` -> `InboundV2Channel`.
///
/// Holds all inbound V2 channels where the peer is the counterparty. V2 channels are assigned
/// a `channel_id` before a funding transaction is created interactively as it's derived from
/// both parties' revocation basepoints once these are known. At the stage of receiving an
/// `open_channel2` request, we have enough information to derive the `channel_id`. Hence, this
/// map's keys are always `channel_id`s.
///
/// The entries in this map are only moved to `channel_by_id` once interactive transaction
/// construction completes successfully.
pub(super) inbound_v2_channel_by_id: HashMap<[u8; 32], InboundV2Channel<Signer>>,
/// The latest `InitFeatures` we heard from the peer.
latest_features: InitFeatures,
/// Messages to send to the peer - pushed to in the same lock that they are generated in (except
Expand Down Expand Up @@ -1913,6 +1936,102 @@ where
Ok(temporary_channel_id)
}


/// Creates a new outbound dual-funded channel to the given remote node and with the given value
/// contributed by us.
///
/// `user_channel_id` will be provided back as in
/// [`Event::FundingGenerationReady::user_channel_id`] to allow tracking of which events
/// correspond with which `create_channel` call. Note that the `user_channel_id` defaults to a
/// randomized value for inbound channels. `user_channel_id` has no meaning inside of LDK, it
/// is simply copied to events and otherwise ignored.
///
/// `funnding_satoshis` is the amount we are contributing to the channel.
/// Raises [`APIError::APIMisuseError`] when `funding_satoshis` > 2**24.
///
/// The `funding_inputs` parameter accepts UTXOs in the form of [`DualFundingUtxo`] which will
/// be used to contribute `funding_satoshis` towards the channel (minus any mining fees due).
/// Raises [`APIError::APIMisuseError`] if the total value of the provided `funding_inputs` is
/// less than `funding_satoshis`.
// TODO(dual_funding): Describe error relating to inputs not being able to cover fees payable by us.
///
/// The `change_script_pubkey` parameter provides a destination for the change output if any value
/// is remaining (greater than dust) after `funding_satoshis` and fees payable are satisfied by
/// `funding_inputs`
// TODO(dual_funding): We could allow a list of such outputs to be provided so that the user may
/// be able to do some more interesting things at the same time as funding a channel, like making
/// some low priority on-chain payment.
///
/// Raises [`APIError::ChannelUnavailable`] if the channel cannot be opened due to failing to
/// generate a shutdown scriptpubkey or destination script set by
/// [`SignerProvider::get_shutdown_scriptpubkey`] or [`SignerProvider::get_destination_script`].
///
/// Note that we do not check if you are currently connected to the given peer. If no
/// connection is available, the outbound `open_channel` message may fail to send, resulting in
/// the channel eventually being silently forgotten (dropped on reload).
///
/// Returns the new Channel's temporary `channel_id`. This ID will appear as
/// [`Event::FundingGenerationReady::temporary_channel_id`] and in
/// [`ChannelDetails::channel_id`] until after
/// [`ChannelManager::funding_transaction_generated`] is called, swapping the Channel's ID for
/// one derived from the funding transaction's TXID. If the counterparty rejects the channel
/// immediately, this temporary ID will appear in [`Event::ChannelClosed::channel_id`].
///
/// [`Event::FundingGenerationReady::user_channel_id`]: events::Event::FundingGenerationReady::user_channel_id
/// [`Event::FundingGenerationReady::temporary_channel_id`]: events::Event::FundingGenerationReady::temporary_channel_id
/// [`Event::ChannelClosed::channel_id`]: events::Event::ChannelClosed::channel_id
pub fn create_dual_funded_channel(&self, their_network_key: PublicKey, funding_satoshis: u64,
funding_inputs: Vec<DualFundingUtxo>, change_script_pubkey: Script,
user_channel_id: u128, override_config: Option<UserConfig>) -> Result<[u8; 32], APIError>
{
Self::dual_funding_amount_checks(funding_satoshis, &funding_inputs);

let _persistence_guard = PersistenceNotifierGuard::notify_on_drop(&self.total_consistency_lock, &self.persistence_notifier);
// We want to make sure the lock is actually acquired by PersistenceNotifierGuard.
debug_assert!(&self.total_consistency_lock.try_write().is_err());

let per_peer_state = self.per_peer_state.read().unwrap();

let peer_state_mutex = per_peer_state.get(&their_network_key)
.ok_or_else(|| APIError::APIMisuseError{ err: format!("Not connected to node: {}", their_network_key) })?;

let mut peer_state = peer_state_mutex.lock().unwrap();
let channel = {
let outbound_scid_alias = self.create_and_insert_outbound_scid_alias();
let their_features = &peer_state.latest_features;
let config = if override_config.is_some() { override_config.as_ref().unwrap() } else { &self.default_configuration };
match OutboundV2Channel::new(&self.fee_estimator, &self.entropy_source, &self.signer_provider, their_network_key,
their_features, funding_satoshis, funding_inputs, change_script_pubkey, user_channel_id, config,
self.best_block.read().unwrap().height(), outbound_scid_alias, true)
{
Ok(res) => res,
Err(e) => {
self.outbound_scid_aliases.lock().unwrap().remove(&outbound_scid_alias);
return Err(e);
},
}
};
let res = channel.get_open_channel_v2(self.genesis_hash.clone());

let temporary_channel_id = channel.channel_id();
match peer_state.outbound_v2_channel_by_id.entry(temporary_channel_id) {
hash_map::Entry::Occupied(_) => {
if cfg!(fuzzing) {
return Err(APIError::APIMisuseError { err: "Fuzzy bad RNG".to_owned() });
} else {
panic!("RNG is bad???");
}
},
hash_map::Entry::Vacant(entry) => { entry.insert(channel); }
}

peer_state.pending_msg_events.push(events::MessageSendEvent::SendOpenChannelV2 {
node_id: their_network_key,
msg: res,
});
Ok(temporary_channel_id)
}

fn list_funded_channels_with_filter<Fn: FnMut(&(&[u8; 32], &Channel<<SP::Target as SignerProvider>::Signer>)) -> bool + Copy>(&self, f: Fn) -> Vec<ChannelDetails> {
// Allocate our best estimate of the number of channels we have in the `res`
// Vec. Sadly the `short_to_chan_info` map doesn't cover channels without
Expand Down Expand Up @@ -4687,6 +4806,128 @@ where
Ok(())
}

/// Accepts a request to open a dual-funded channel after an [`Event::OpenChannelV2Request`].
///
/// The `temporary_channel_id` parameter indicates which inbound channel should be accepted,
/// and the `counterparty_node_id` parameter is the id of the peer which has requested to open
/// the channel.
///
/// The `user_channel_id` parameter will be provided back in
/// [`Event::ChannelClosed::user_channel_id`] to allow tracking of which events correspond
/// with which `accept_inbound_dual_funded_channel`/`accept_inbound_dual_funded_channel_from_trusted_peer_0conf` call.
///
/// `funnding_satoshis` is the amount we are contributing to the channel.
/// Raises [`APIError::APIMisuseError`] when `funding_satoshis` > 2**24.
///
/// The `funding_inputs` parameter accepts UTXOs in the form of [`DualFundingUtxo`] which will
/// be used to contribute `funding_satoshis` towards the channel (minus any mining fees due).
/// Raises [`APIError::APIMisuseError`] if the total value of the provided `funding_inputs` is
/// less than `funding_satoshis`.
// TODO(dual_funding): Describe error relating to inputs not being able to cover fees payable by us.
///
/// The `change_script_pubkey` parameter provides a destination for the change output if any value
/// is remaining (greater than dust) after `funding_satoshis` and fees payable are satisfied by
/// `funding_inputs`
// TODO(dual_funding): We could allow a list of such outputs to be provided so that the user may
/// be able to do some more interesting things at the same time as funding.
///
/// Note that this method will return an error and reject the channel, if it requires support
/// for zero confirmations.
// TODO(dual_funding): Discussion on complications with 0conf dual-funded channels where "locking"
// of UTXOs used for funding would be required and other issues.
// See: https://lists.linuxfoundation.org/pipermail/lightning-dev/2023-May/003920.html
///
///
/// [`Event::OpenChannelV2Request`]: events::Event::OpenChannelV2Request
/// [`Event::ChannelClosed::user_channel_id`]: events::Event::ChannelClosed::user_channel_id
pub fn accept_inbound_dual_funded_channel(&self, temporary_channel_id: &[u8; 32],
counterparty_node_id: &PublicKey, user_channel_id: u128, funding_satoshis: u64,
funding_inputs: Vec<DualFundingUtxo>, change_script_pubkey: Script) -> Result<(), APIError> {
self.do_accept_inbound_dual_funded_channel(temporary_channel_id, counterparty_node_id, false,
user_channel_id, funding_satoshis, funding_inputs, change_script_pubkey)
}

fn do_accept_inbound_dual_funded_channel(&self, temporary_channel_id: &[u8; 32],
counterparty_node_id: &PublicKey, accept_0conf: bool, user_channel_id: u128,
funding_satoshis: u64, funding_inputs: Vec<DualFundingUtxo>, change_script_pubkey: Script,
) -> Result<(), APIError> {
Self::dual_funding_amount_checks(funding_satoshis, &funding_inputs)?;

let _persistence_guard = PersistenceNotifierGuard::notify_on_drop(&self.total_consistency_lock, &self.persistence_notifier);

let peers_without_funded_channels =
self.peers_without_funded_channels(|peer| { peer.total_channel_count() > 0 });
let per_peer_state = self.per_peer_state.read().unwrap();
let peer_state_mutex = per_peer_state.get(counterparty_node_id)
.ok_or_else(|| APIError::ChannelUnavailable { err: format!("Can't find a peer matching the passed counterparty node_id {}", counterparty_node_id) })?;
let mut peer_state_lock = peer_state_mutex.lock().unwrap();
let peer_state = &mut *peer_state_lock;
let is_only_peer_channel = peer_state.total_channel_count() == 1;
match peer_state.inbound_v2_channel_by_id.entry(temporary_channel_id.clone()) {
hash_map::Entry::Occupied(mut channel) => {
if !channel.get().inbound_is_awaiting_accept() {
return Err(APIError::APIMisuseError { err: "The channel isn't currently awaiting to be accepted.".to_owned() });
}
if accept_0conf {
channel.get_mut().set_0conf();
} else if channel.get().get_channel_type().requires_zero_conf() {
let send_msg_err_event = events::MessageSendEvent::HandleError {
node_id: channel.get().get_counterparty_node_id(),
action: msgs::ErrorAction::SendErrorMessage{
msg: msgs::ErrorMessage { channel_id: temporary_channel_id.clone(), data: "No zero confirmation channels accepted".to_owned(), }
}
};
peer_state.pending_msg_events.push(send_msg_err_event);
let _ = remove_channel!(self, channel);
return Err(APIError::APIMisuseError { err: "Please use accept_inbound_channel_from_trusted_peer_0conf to accept channels with zero confirmations.".to_owned() });
} else {
// If this peer already has some channels, a new channel won't increase our number of peers
// with unfunded channels, so as long as we aren't over the maximum number of unfunded
// channels per-peer we can accept channels from a peer with existing ones.
if is_only_peer_channel && peers_without_funded_channels >= MAX_UNFUNDED_CHANNEL_PEERS {
let send_msg_err_event = events::MessageSendEvent::HandleError {
node_id: channel.get().get_counterparty_node_id(),
action: msgs::ErrorAction::SendErrorMessage{
msg: msgs::ErrorMessage { channel_id: temporary_channel_id.clone(), data: "Have too many peers with unfunded channels, not accepting new ones".to_owned(), }
}
};
peer_state.pending_msg_events.push(send_msg_err_event);
let _ = remove_channel!(self, channel);
return Err(APIError::APIMisuseError { err: "Too many peers with unfunded channels, refusing to accept new ones".to_owned() });
}
}

peer_state.pending_msg_events.push(events::MessageSendEvent::SendAcceptChannel {
node_id: channel.get().get_counterparty_node_id(),
msg: channel.get_mut().accept_inbound_channel(user_channel_id),
});
}
hash_map::Entry::Vacant(_) => {
return Err(APIError::ChannelUnavailable { err: format!("Channel with id {} not found for the passed counterparty node_id {}", log_bytes!(*temporary_channel_id), counterparty_node_id) });
}
}
Ok(())
}

/// Checks related to inputs and their amounts related to establishing dual-funded channels.
fn dual_funding_amount_checks(funding_satoshis: u64, funding_inputs: &Vec<DualFundingUtxo>)
-> Result<(), APIError> {
if funding_satoshis < 1000 {
return Err(APIError::APIMisuseError {
err: format!("Funding amount must be at least 1000 satoshis. It was {} sats", funding_satoshis),
});
}

let total_input_satoshis: u64 = funding_inputs.iter().map(|input| input.output.value).sum();
if total_input_satoshis < funding_satoshis {
Err(APIError::APIMisuseError {
err: format!("Total value of funding inputs must be at least funding amount. It was {} sats",
total_input_satoshis) })
} else {
Ok(())
}
}

/// Gets the number of peers which match the given filter and do not have any funded, outbound,
/// or 0-conf channels.
///
Expand Down Expand Up @@ -4726,6 +4967,11 @@ where
num_unfunded_channels += 1;
}
}
for (_, chan) in peer.inbound_v2_channel_by_id.iter() {
if chan.minimum_depth().unwrap_or(1) != 0 {
num_unfunded_channels += 1;
}
}
num_unfunded_channels
}

Expand Down Expand Up @@ -6820,6 +7066,8 @@ where
channel_by_id: HashMap::new(),
outbound_v1_channel_by_id: HashMap::new(),
inbound_v1_channel_by_id: HashMap::new(),
outbound_v2_channel_by_id: HashMap::new(),
inbound_v2_channel_by_id: HashMap::new(),
latest_features: init_msg.features.clone(),
pending_msg_events: Vec::new(),
monitor_update_blocked_actions: BTreeMap::new(),
Expand Down Expand Up @@ -8013,6 +8261,8 @@ where
channel_by_id: peer_channels.remove(&peer_pubkey).unwrap_or(HashMap::new()),
outbound_v1_channel_by_id: HashMap::new(),
inbound_v1_channel_by_id: HashMap::new(),
outbound_v2_channel_by_id: HashMap::new(),
inbound_v2_channel_by_id: HashMap::new(),
latest_features: Readable::read(reader)?,
pending_msg_events: Vec::new(),
monitor_update_blocked_actions: BTreeMap::new(),
Expand Down
16 changes: 16 additions & 0 deletions lightning/src/util/config.rs
Original file line number Diff line number Diff line change
Expand Up @@ -552,6 +552,21 @@ pub struct UserConfig {
/// [`ChannelManager::get_intercept_scid`]: crate::ln::channelmanager::ChannelManager::get_intercept_scid
/// [`Event::HTLCIntercepted`]: crate::events::Event::HTLCIntercepted
pub accept_intercept_htlcs: bool,
/// If this is set to true, regardless of `accept_inbound_channels`, the user needs to manually
/// accept inbound requests to open a new dual-funded channel and provide any UTXOs they'd like
/// to contribute to the funding transaction.
///
/// When set to true, [`Event::OpenChannelV2Request`] will be triggered once a request to open a
/// new inbound dual-funded channel is received through a [`msgs::OpenChannelV2`] message. In that
/// case, a [`msgs::AcceptChannelV2`] message will not be sent back to the counterparty node unless
/// the user explicitly chooses to accept the request.
///
/// Default value: false.
///
/// [`Event::OpenChannelV2Request`]: crate::events::Event::OpenChannelV2Request
/// [`msgs::OpenChannelV2`]: crate::ln::msgs::OpenChannelV2
/// [`msgs::AcceptChannelV2`]: crate::ln::msgs::AcceptChannelV2
pub contribute_to_dual_funded_channels: bool,
}

impl Default for UserConfig {
Expand All @@ -564,6 +579,7 @@ impl Default for UserConfig {
accept_inbound_channels: true,
manually_accept_inbound_channels: false,
accept_intercept_htlcs: false,
contribute_to_dual_funded_channels: false,
}
}
}

0 comments on commit 5bd6e42

Please sign in to comment.