draft KeysInterface.sign_holder_commitment_phase2

Author: devrandom

lightning/src/chain/keysinterface.rs

@@ -33,12 +33,13 @@ use util::ser::{Writeable, Writer, Readable};
 
 use chain::transaction::OutPoint;
 use ln::chan_utils;
-use ln::chan_utils::{HTLCOutputInCommitment, make_funding_redeemscript, ChannelPublicKeys, HolderCommitmentTransaction, PreCalculatedTxCreationKeys};
+use ln::chan_utils::{HTLCOutputInCommitment, make_funding_redeemscript, ChannelPublicKeys, HolderCommitmentTransaction, PreCalculatedTxCreationKeys, derive_public_key, derive_public_revocation_key, TxCreationKeys};
 use ln::msgs::UnsignedChannelAnnouncement;
 
 use std::sync::atomic::{AtomicUsize, Ordering};
 use std::io::Error;
 use ln::msgs::DecodeError;
+use ln::transaction::{CommitmentInfo, build_commitment_tx, get_commitment_transaction_number_obscure_factor};
 
 /// When on-chain outputs are created by rust-lightning (which our counterparty is not able to
 /// claim at any point in the future) an event is generated which you must track and be able to
@@ -244,6 +245,17 @@ pub trait ChannelKeys : Send+Clone {
 	// TODO: Add more input vars to enable better checking (preferably removing commitment_tx and
 	fn sign_holder_commitment<T: secp256k1::Signing + secp256k1::Verification>(&self, holder_commitment_tx: &HolderCommitmentTransaction, secp_ctx: &Secp256k1<T>) -> Result<Signature, ()>;
 
+	/// Create a signature for a holder's commitment transaction and attached HTLC transactions.
+	fn sign_holder_commitment_phase2<T: secp256k1::Signing + secp256k1::Verification>(
+		&self,
+		commitment_number: u64,
+		feerate_per_kw: u32,
+		to_holder_value_sat: u64,
+		to_counterparty_value_sat: u64,
+		htlcs: &Vec<HTLCOutputInCommitment>,
+		secp_ctx: &Secp256k1<T>,
+	) -> Result<(Signature, Vec<Signature>), ()>;
+
 	/// Same as sign_holder_commitment, but exists only for tests to get access to holder commitment
 	/// transactions which will be broadcasted later, after the channel has moved on to a newer
 	/// state. Thus, needs its own method as sign_holder_commitment may enforce that we only ever
@@ -325,7 +337,12 @@ pub trait ChannelKeys : Send+Clone {
 	/// We bind holder_selected_contest_delay late here for API convenience.
 	///
 	/// Will be called before any signatures are applied.
-	fn on_accept(&mut self, channel_points: &ChannelPublicKeys, counterparty_selected_contest_delay: u16, holder_selected_contest_delay: u16);
+	fn on_accept(&mut self, channel_points: &ChannelPublicKeys, counterparty_selected_contest_delay: u16, holder_selected_contest_delay: u16, is_outbound: bool);
+
+	/// Set the funding outpoint.
+	///
+	/// Used for re-constructing transactions as part of signing.
+	fn on_funding_created(&mut self, funding_outpoint: &OutPoint);
 }
 
 /// A trait to describe an object which can get user secrets and key material.
@@ -365,6 +382,8 @@ struct AcceptedChannelData {
 	/// by our counterparty, ie the amount of time that they have to wait to recover their funds
 	/// if they broadcast a transaction.
 	holder_selected_contest_delay: u16,
+	/// Whether the holder is the initiator of this channel
+	is_outbound: bool,
 }
 
 #[derive(Clone)]
@@ -386,6 +405,8 @@ pub struct InMemoryChannelKeys {
 	pub(crate) holder_channel_pubkeys: ChannelPublicKeys,
 	/// Counterparty public keys and counterparty/holder selected_contest_delay, populated on channel acceptance
 	accepted_channel_data: Option<AcceptedChannelData>,
+	/// Funding outpoint, populated on channel funding creation
+	funding_outpoint: Option<OutPoint>,
 	/// The total value of this channel
 	channel_value_satoshis: u64,
 	/// Key derivation parameters
@@ -418,6 +439,7 @@ impl InMemoryChannelKeys {
 			channel_value_satoshis,
 			holder_channel_pubkeys,
 			accepted_channel_data: None,
+			funding_outpoint: None,
 			key_derivation_params,
 		}
 	}
@@ -454,6 +476,77 @@ impl InMemoryChannelKeys {
 	/// if they broadcast a transaction.
 	/// Will panic if on_accept wasn't called.
 	pub fn holder_selected_contest_delay(&self) -> u16 { self.accepted_channel_data.as_ref().unwrap().holder_selected_contest_delay }
+
+	/// Whether the holder is the initiator
+	pub fn is_outbound(&self) -> bool { self.accepted_channel_data.as_ref().unwrap().is_outbound }
+
+	/// Funding outpoint
+	/// Will panic if on_funding_created wasn't called.
+	pub fn funding_outpoint(&self) -> &OutPoint { self.funding_outpoint.as_ref().unwrap() }
+
+	/// Prepare build information for a holder-broadcast commitment tx
+	pub fn build_holder_commitment_info<T: secp256k1::Signing + secp256k1::Verification>(
+		&self,
+		per_commitment_point: &PublicKey,
+		to_holder_value_sat: u64,
+		to_counterparty_value_sat: u64,
+		htlcs: &Vec<HTLCOutputInCommitment>,
+		secp_ctx: &Secp256k1<T>,
+	) -> Result<CommitmentInfo, ()> {
+		let counterparty_points = self.counterparty_pubkeys();
+
+		let to_holder_delayed_pubkey = derive_public_key(
+			&secp_ctx,
+			&per_commitment_point,
+			&self.pubkeys().delayed_payment_basepoint,
+		).map_err(|_| ())?;
+
+		let revocation_pubkey = derive_public_revocation_key(
+			&secp_ctx,
+			&per_commitment_point,
+			&counterparty_points.revocation_basepoint,
+		).map_err(|_| ())?;
+
+		Ok(CommitmentInfo {
+			is_counterparty_broadcaster: false,
+			to_countersigner_pubkey: counterparty_points.payment_point,
+			to_countersigner_value_sat: to_counterparty_value_sat,
+			revocation_pubkey,
+			to_broadcaster_delayed_pubkey: to_holder_delayed_pubkey,
+			to_broadcaster_value_sat: to_holder_value_sat,
+			to_self_delay: self.counterparty_selected_contest_delay(),
+			htlcs: htlcs.clone(),
+		})
+	}
+
+	fn get_commitment_transaction_number_obscure_factor(&self) -> u64 {
+		get_commitment_transaction_number_obscure_factor(
+			&self.pubkeys().payment_point,
+			&self.counterparty_pubkeys().payment_point,
+			self.is_outbound(),
+		)
+	}
+
+	/// Build a commitment tx
+	pub fn build_commitment_tx<T: secp256k1::Signing + secp256k1::Verification>(&self, per_commitment_point: &PublicKey,
+																				commitment_number: u64, info: &CommitmentInfo,
+																				secp_ctx: &Secp256k1<T>) -> Result<(bitcoin::Transaction, Vec<HTLCOutputInCommitment>, TxCreationKeys, Vec<Script>), ()> {
+		let keys = if !info.is_counterparty_broadcaster {
+			TxCreationKeys::make_tx_keys(per_commitment_point, self.pubkeys(), self.counterparty_pubkeys(), secp_ctx)
+		} else {
+			TxCreationKeys::make_tx_keys(per_commitment_point, self.counterparty_pubkeys(), self.pubkeys(), secp_ctx)
+		}.unwrap();
+
+		let obscured_commitment_transaction_number =
+			self.get_commitment_transaction_number_obscure_factor() ^ commitment_number;
+		let (tx, htlcs, scripts) = build_commitment_tx(
+			&keys,
+			info,
+			obscured_commitment_transaction_number,
+			self.funding_outpoint().into_bitcoin_outpoint(),
+		);
+		Ok((tx, htlcs, keys, scripts))
+	}
 }
 
 impl ChannelKeys for InMemoryChannelKeys {
@@ -507,6 +600,47 @@ impl ChannelKeys for InMemoryChannelKeys {
 		Ok(holder_commitment_tx.get_holder_sig(&self.funding_key, &channel_funding_redeemscript, self.channel_value_satoshis, secp_ctx))
 	}
 
+	fn sign_holder_commitment_phase2<T: secp256k1::Signing + secp256k1::Verification>(&self, commitment_number: u64, feerate_per_kw: u32, to_holder_value_sat: u64, to_counterparty_value_sat: u64, htlcs: &Vec<HTLCOutputInCommitment>, secp_ctx: &Secp256k1<T>) -> Result<(Signature, Vec<Signature>), ()> {
+		let per_commitment_point = self.get_per_commitment_point(commitment_number, secp_ctx);
+		let info = self.build_holder_commitment_info(
+			&per_commitment_point,
+			to_holder_value_sat,
+			to_counterparty_value_sat,
+			htlcs,
+			secp_ctx,
+		)?;
+		let (tx, htlcs, keys, _) =
+			self.build_commitment_tx(&per_commitment_point, commitment_number, &info, secp_ctx)?;
+
+		let funding_pubkey = PublicKey::from_secret_key(secp_ctx, &self.funding_key);
+		let funding_redeemscript = make_funding_redeemscript(&funding_pubkey, &self.counterparty_pubkeys().funding_pubkey);
+		let sighash = hash_to_message!(&bip143::SigHashCache::new(&tx)
+			.signature_hash(0, &funding_redeemscript, self.channel_value_satoshis, SigHashType::All)[..]);
+		let sig = secp_ctx.sign(&sighash, &self.funding_key);
+
+		// We provide a dummy signature for the remote, since we don't require that sig
+		// to be passed in to this call.  It would have been better if HolderCommitmentTransaction
+		// didn't require the remote sig.
+		let dummy_sig = Secp256k1::new().sign(
+			&secp256k1::Message::from_slice(&[42; 32]).unwrap(),
+			&SecretKey::from_slice(&[42; 32]).unwrap(),
+		);
+		let htlcs_with_sig = htlcs.iter().map(|h| (h.clone(), Some(dummy_sig.clone()))).collect();
+		let commitment_tx = HolderCommitmentTransaction::new_missing_holder_sig(
+			tx,
+			dummy_sig,
+			&self.pubkeys().funding_pubkey,
+			&self.counterparty_pubkeys().funding_pubkey,
+			keys,
+			feerate_per_kw,
+			htlcs_with_sig,
+		);
+		let htlc_sigs_o = self.sign_holder_commitment_htlc_transactions(&commitment_tx, secp_ctx)?;
+		let htlc_sigs = htlc_sigs_o.iter().map(|o| o.unwrap()).collect();
+
+		Ok((sig, htlc_sigs))
+	}
+
 	#[cfg(any(test,feature = "unsafe_revoked_tx_signing"))]
 	fn unsafe_sign_holder_commitment<T: secp256k1::Signing + secp256k1::Verification>(&self, holder_commitment_tx: &HolderCommitmentTransaction, secp_ctx: &Secp256k1<T>) -> Result<Signature, ()> {
 		let funding_pubkey = PublicKey::from_secret_key(secp_ctx, &self.funding_key);
@@ -588,18 +722,24 @@ impl ChannelKeys for InMemoryChannelKeys {
 		Ok(secp_ctx.sign(&msghash, &self.funding_key))
 	}
 
-	fn on_accept(&mut self, channel_pubkeys: &ChannelPublicKeys, counterparty_selected_contest_delay: u16, holder_selected_contest_delay: u16) {
-		assert!(self.accepted_channel_data.is_none(), "Already accepted");
+	fn on_accept(&mut self, channel_pubkeys: &ChannelPublicKeys, counterparty_selected_contest_delay: u16, holder_selected_contest_delay: u16, is_outbound: bool) {
+		assert!(self.accepted_channel_data.is_none(), "Acceptance already noted");
 		self.accepted_channel_data = Some(AcceptedChannelData {
 			counterparty_channel_pubkeys: channel_pubkeys.clone(),
 			counterparty_selected_contest_delay,
 			holder_selected_contest_delay,
+			is_outbound,
 		});
 	}
+
+	fn on_funding_created(&mut self, funding_outpoint: &OutPoint) {
+		assert!(self.funding_outpoint.is_none(), "Funding creation already noted");
+		self.funding_outpoint = Some(funding_outpoint.clone());
+	}
 }
 
 impl_writeable!(AcceptedChannelData, 0,
- { counterparty_channel_pubkeys, counterparty_selected_contest_delay, holder_selected_contest_delay });
+ { counterparty_channel_pubkeys, counterparty_selected_contest_delay, holder_selected_contest_delay, is_outbound });
 
 impl Writeable for InMemoryChannelKeys {
 	fn write<W: Writer>(&self, writer: &mut W) -> Result<(), Error> {
@@ -627,6 +767,7 @@ impl Readable for InMemoryChannelKeys {
 		let htlc_base_key = Readable::read(reader)?;
 		let commitment_seed = Readable::read(reader)?;
 		let counterparty_channel_data = Readable::read(reader)?;
+		let funding_outpoint = Readable::read(reader)?;
 		let channel_value_satoshis = Readable::read(reader)?;
 		let secp_ctx = Secp256k1::signing_only();
 		let holder_channel_pubkeys =
@@ -646,6 +787,7 @@ impl Readable for InMemoryChannelKeys {
 			channel_value_satoshis,
 			holder_channel_pubkeys,
 			accepted_channel_data: counterparty_channel_data,
+			funding_outpoint,
 			key_derivation_params: (params_1, params_2),
 		})
 	}

lightning/src/lib.rs

@@ -30,6 +30,7 @@
 extern crate bitcoin;
 #[cfg(test)] extern crate hex;
 #[cfg(test)] extern crate regex;
+extern crate core;
 
 #[macro_use]
 pub mod util;

lightning/src/ln/chan_utils.rs

@@ -382,6 +382,18 @@ impl TxCreationKeys {
 			broadcaster_delayed_payment_key: derive_public_key(&secp_ctx, &per_commitment_point, &broadcaster_delayed_payment_base)?,
 		})
 	}
+
+	/// Make transaction creation keys from channel keys
+	pub fn make_tx_keys<T: secp256k1::Signing + secp256k1::Verification>(per_commitment_point: &PublicKey, broadcaster_keys: &ChannelPublicKeys, countersignatory_keys: &ChannelPublicKeys, secp_ctx: &Secp256k1<T>) -> Result<TxCreationKeys, SecpError> {
+		TxCreationKeys::derive_new(
+			&secp_ctx,
+			&per_commitment_point,
+			&broadcaster_keys.delayed_payment_basepoint,
+			&broadcaster_keys.htlc_basepoint,
+			&countersignatory_keys.revocation_basepoint,
+			&countersignatory_keys.htlc_basepoint,
+		)
+	}
 }
 
 /// A script either spendable by the revocation

lightning/src/ln/channel.rs

@@ -590,7 +590,7 @@ impl<ChanSigner: ChannelKeys> Channel<ChanSigner> {
 			delayed_payment_basepoint: msg.delayed_payment_basepoint,
 			htlc_basepoint: msg.htlc_basepoint
 		};
-		chan_keys.on_accept(&counterparty_pubkeys, msg.to_self_delay, config.own_channel_config.our_to_self_delay);
+		chan_keys.on_accept(&counterparty_pubkeys, msg.to_self_delay, config.own_channel_config.our_to_self_delay, false);
 		let mut local_config = (*config).channel_options.clone();
 
 		if config.own_channel_config.our_to_self_delay < BREAKDOWN_TIMEOUT {
@@ -1463,7 +1463,7 @@ impl<ChanSigner: ChannelKeys> Channel<ChanSigner> {
 			htlc_basepoint: msg.htlc_basepoint
 		};
 
-		self.holder_keys.on_accept(&counterparty_pubkeys, msg.to_self_delay, self.holder_selected_contest_delay);
+		self.holder_keys.on_accept(&counterparty_pubkeys, msg.to_self_delay, self.holder_selected_contest_delay, true);
 		self.counterparty_pubkeys = Some(counterparty_pubkeys);
 
 		self.counterparty_cur_commitment_point = Some(msg.first_per_commitment_point);
@@ -4656,7 +4656,7 @@ mod tests {
 			delayed_payment_basepoint: public_from_secret_hex(&secp_ctx, "1552dfba4f6cf29a62a0af13c8d6981d36d0ef8d61ba10fb0fe90da7634d7e13"),
 			htlc_basepoint: public_from_secret_hex(&secp_ctx, "4444444444444444444444444444444444444444444444444444444444444444")
 		};
-		chan_keys.on_accept(&counterparty_pubkeys, chan.counterparty_selected_contest_delay, chan.holder_selected_contest_delay);
+		chan_keys.on_accept(&counterparty_pubkeys, chan.counterparty_selected_contest_delay, chan.holder_selected_contest_delay, false);
 
 		assert_eq!(counterparty_pubkeys.payment_point.serialize()[..],
 		           hex::decode("032c0b7cf95324a07d05398b240174dc0c2be444d96b159aa6c7f7b1e668680991").unwrap()[..]);

lightning/src/ln/mod.rs

@@ -25,6 +25,7 @@ pub mod peer_handler;
 pub mod chan_utils;
 pub mod features;
 pub(crate) mod onchaintx;
+pub mod transaction;
 
 #[cfg(feature = "fuzztarget")]
 pub mod peer_channel_encryptor;