mirrors/rust-lightning
1
0
Fork 0
mirror of https://github.com/lightningdevkit/rust-lightning.git synced 2025-03-13 14:52:21 +01:00
Code Issues Projects Releases Packages Wiki Activity

Parse inbound TrampolineEntrypoint payload

Browse source 
Check inbound onion payloads for the presence of an inner onion, which
we will later also need to decrypt to handle Trampoline payments.
This commit is contained in:
Arik Sosman 2025-02-24 12:12:49 -08:00
parent ee1514a955
commit a11e66d156
No known key found for this signature in database
GPG key ID: CFF795E7811C0093
1 changed files with 52 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

52
lightning/src/ln/msgs.rs
View file

@ -2073,6 +2073,17 @@ mod fuzzy_internal_msgs {
pub outgoing_cltv_value: u32,
}
#[cfg(trampoline)]
pub struct InboundTrampolineEntrypointPayload {
pub amt_to_forward: u64,
pub outgoing_cltv_value: u32,
pub multipath_trampoline_data: FinalOnionHopData,
pub trampoline_packet: TrampolineOnionPacket,
/// The blinding point this hop needs to decrypt its Trampoline onion.
/// This is used for Trampoline hops that are not the blinded path intro hop.
pub current_path_key: Option<PublicKey>
}
pub struct InboundOnionReceivePayload {
pub payment_data: Option<FinalOnionHopData>,
pub payment_metadata: Option<Vec<u8>>,
@ -2104,6 +2115,8 @@ mod fuzzy_internal_msgs {
pub enum InboundOnionPayload {
Forward(InboundOnionForwardPayload),
#[cfg(trampoline)]
TrampolineEntrypoint(InboundTrampolineEntrypointPayload),
Receive(InboundOnionReceivePayload),
BlindedForward(InboundOnionBlindedForwardPayload),
BlindedReceive(InboundOnionBlindedReceivePayload),
@ -3194,11 +3207,36 @@ impl<NS: Deref> ReadableArgs<(Option<PublicKey>, NS)> for InboundOnionPayload wh
let mut payment_metadata: Option<WithoutLength<Vec<u8>>> = None;
let mut total_msat = None;
let mut keysend_preimage: Option<PaymentPreimage> = None;
#[cfg(trampoline)]
let mut trampoline_onion_packet: Option<TrampolineOnionPacket> = None;
let mut invoice_request: Option<InvoiceRequest> = None;
let mut custom_tlvs = Vec::new();
let tlv_len = BigSize::read(r)?;
let mut rd = FixedLengthReader::new(r, tlv_len.0);
#[cfg(trampoline)]
decode_tlv_stream_with_custom_tlv_decode!(&mut rd, {
(2, amt, (option, encoding: (u64, HighZeroBytesDroppedBigSize))),
(4, cltv_value, (option, encoding: (u32, HighZeroBytesDroppedBigSize))),
(6, short_id, option),
(8, payment_data, option),
(10, encrypted_tlvs_opt, option),
(12, intro_node_blinding_point, option),
(16, payment_metadata, option),
(18, total_msat, (option, encoding: (u64, HighZeroBytesDroppedBigSize))),
(20, trampoline_onion_packet, option),
(77_777, invoice_request, option),
// See https://github.com/lightning/blips/blob/master/blip-0003.md
(5482373484, keysend_preimage, option)
}, |msg_type: u64, msg_reader: &mut FixedLengthReader<_>| -> Result<bool, DecodeError> {
if msg_type < 1 << 16 { return Ok(false) }
let mut value = Vec::new();
msg_reader.read_to_limit(&mut value, u64::MAX)?;
custom_tlvs.push((msg_type, value));
Ok(true)
});
#[cfg(not(trampoline))]
decode_tlv_stream_with_custom_tlv_decode!(&mut rd, {
(2, amt, (option, encoding: (u64, HighZeroBytesDroppedBigSize))),
(4, cltv_value, (option, encoding: (u32, HighZeroBytesDroppedBigSize))),
@ -3224,6 +3262,20 @@ impl<NS: Deref> ReadableArgs<(Option<PublicKey>, NS)> for InboundOnionPayload wh
return Err(DecodeError::InvalidValue)
}
#[cfg(trampoline)]
if let Some(trampoline_onion_packet) = trampoline_onion_packet {
if payment_metadata.is_some() || encrypted_tlvs_opt.is_some() ||
total_msat.is_some()
{ return Err(DecodeError::InvalidValue) }
return Ok(Self::TrampolineEntrypoint(InboundTrampolineEntrypointPayload {
amt_to_forward: amt.ok_or(DecodeError::InvalidValue)?,
outgoing_cltv_value: cltv_value.ok_or(DecodeError::InvalidValue)?,
multipath_trampoline_data: payment_data.ok_or(DecodeError::InvalidValue)?,
trampoline_packet: trampoline_onion_packet,
current_path_key: intro_node_blinding_point,
}))
}
if let Some(blinding_point) = intro_node_blinding_point.or(update_add_blinding_point) {
if short_id.is_some() || payment_data.is_some() || payment_metadata.is_some() {
return Err(DecodeError::InvalidValue)