1use bitcoin::secp256k1::PublicKey;
28use bitcoin::secp256k1::ecdsa::Signature;
29use bitcoin::secp256k1;
30use bitcoin::blockdata::script::Script;
31use bitcoin::hash_types::{Txid, BlockHash};
32
33use crate::ln::features::{ChannelFeatures, ChannelTypeFeatures, InitFeatures, NodeFeatures};
34use crate::ln::onion_utils;
35use crate::onion_message;
36
37use crate::prelude::*;
38use core::fmt;
39use core::fmt::Debug;
40use crate::io::{self, Read};
41use crate::io_extras::read_to_end;
42
43use crate::util::events::{MessageSendEventsProvider, OnionMessageProvider};
44use crate::util::logger;
45use crate::util::ser::{LengthReadable, Readable, ReadableArgs, Writeable, Writer, FixedLengthReader, HighZeroBytesDroppedBigSize, Hostname};
46
47use crate::ln::{PaymentPreimage, PaymentHash, PaymentSecret};
48
49pub(crate) const MAX_VALUE_MSAT: u64 = 21_000_000_0000_0000_000;
51
52#[derive(Clone, Debug, PartialEq, Eq)]
54pub enum DecodeError {
55 UnknownVersion,
58 UnknownRequiredFeature,
60 InvalidValue,
64 ShortRead,
66 BadLengthDescriptor,
68 Io(io::ErrorKind),
70 UnsupportedCompression,
72}
73
74#[derive(Clone, Debug, PartialEq, Eq)]
76pub struct Init {
77 pub features: InitFeatures,
79 pub remote_network_address: Option<NetAddress>,
84}
85
86#[derive(Clone, Debug, PartialEq, Eq)]
88pub struct ErrorMessage {
89 pub channel_id: [u8; 32],
94 pub data: String,
99}
100
101#[derive(Clone, Debug, PartialEq, Eq)]
103pub struct WarningMessage {
104 pub channel_id: [u8; 32],
108 pub data: String,
113}
114
115#[derive(Clone, Debug, PartialEq, Eq)]
117pub struct Ping {
118 pub ponglen: u16,
120 pub byteslen: u16,
123}
124
125#[derive(Clone, Debug, PartialEq, Eq)]
127pub struct Pong {
128 pub byteslen: u16,
131}
132
133#[derive(Clone, Debug, PartialEq, Eq)]
135pub struct OpenChannel {
136 pub chain_hash: BlockHash,
138 pub temporary_channel_id: [u8; 32],
140 pub funding_satoshis: u64,
142 pub push_msat: u64,
144 pub dust_limit_satoshis: u64,
146 pub max_htlc_value_in_flight_msat: u64,
148 pub channel_reserve_satoshis: u64,
150 pub htlc_minimum_msat: u64,
152 pub feerate_per_kw: u32,
154 pub to_self_delay: u16,
156 pub max_accepted_htlcs: u16,
158 pub funding_pubkey: PublicKey,
160 pub revocation_basepoint: PublicKey,
162 pub payment_point: PublicKey,
164 pub delayed_payment_basepoint: PublicKey,
166 pub htlc_basepoint: PublicKey,
168 pub first_per_commitment_point: PublicKey,
170 pub channel_flags: u8,
172 pub shutdown_scriptpubkey: OptionalField<Script>,
174 pub channel_type: Option<ChannelTypeFeatures>,
178}
179
180#[derive(Clone, Debug, PartialEq, Eq)]
182pub struct AcceptChannel {
183 pub temporary_channel_id: [u8; 32],
185 pub dust_limit_satoshis: u64,
187 pub max_htlc_value_in_flight_msat: u64,
189 pub channel_reserve_satoshis: u64,
191 pub htlc_minimum_msat: u64,
193 pub minimum_depth: u32,
195 pub to_self_delay: u16,
197 pub max_accepted_htlcs: u16,
199 pub funding_pubkey: PublicKey,
201 pub revocation_basepoint: PublicKey,
203 pub payment_point: PublicKey,
205 pub delayed_payment_basepoint: PublicKey,
207 pub htlc_basepoint: PublicKey,
209 pub first_per_commitment_point: PublicKey,
211 pub shutdown_scriptpubkey: OptionalField<Script>,
213 pub channel_type: Option<ChannelTypeFeatures>,
219}
220
221#[derive(Clone, Debug, PartialEq, Eq)]
223pub struct FundingCreated {
224 pub temporary_channel_id: [u8; 32],
226 pub funding_txid: Txid,
228 pub funding_output_index: u16,
230 pub signature: Signature,
232}
233
234#[derive(Clone, Debug, PartialEq, Eq)]
236pub struct FundingSigned {
237 pub channel_id: [u8; 32],
239 pub signature: Signature,
241}
242
243#[derive(Clone, Debug, PartialEq, Eq)]
245pub struct ChannelReady {
246 pub channel_id: [u8; 32],
248 pub next_per_commitment_point: PublicKey,
250 pub short_channel_id_alias: Option<u64>,
253}
254
255#[derive(Clone, Debug, PartialEq, Eq)]
257pub struct Shutdown {
258 pub channel_id: [u8; 32],
260 pub scriptpubkey: Script,
263}
264
265#[derive(Clone, Debug, PartialEq, Eq)]
269pub struct ClosingSignedFeeRange {
270 pub min_fee_satoshis: u64,
273 pub max_fee_satoshis: u64,
276}
277
278#[derive(Clone, Debug, PartialEq, Eq)]
280pub struct ClosingSigned {
281 pub channel_id: [u8; 32],
283 pub fee_satoshis: u64,
285 pub signature: Signature,
287 pub fee_range: Option<ClosingSignedFeeRange>,
290}
291
292#[derive(Clone, Debug, PartialEq, Eq)]
294pub struct UpdateAddHTLC {
295 pub channel_id: [u8; 32],
297 pub htlc_id: u64,
299 pub amount_msat: u64,
301 pub payment_hash: PaymentHash,
303 pub cltv_expiry: u32,
305 pub(crate) onion_routing_packet: OnionPacket,
306}
307
308 #[derive(Clone, Debug, PartialEq, Eq)]
310pub struct OnionMessage {
311 pub blinding_point: PublicKey,
313 pub(crate) onion_routing_packet: onion_message::Packet,
314}
315
316#[derive(Clone, Debug, PartialEq, Eq)]
318pub struct UpdateFulfillHTLC {
319 pub channel_id: [u8; 32],
321 pub htlc_id: u64,
323 pub payment_preimage: PaymentPreimage,
325}
326
327#[derive(Clone, Debug, PartialEq, Eq)]
329pub struct UpdateFailHTLC {
330 pub channel_id: [u8; 32],
332 pub htlc_id: u64,
334 pub(crate) reason: OnionErrorPacket,
335}
336
337#[derive(Clone, Debug, PartialEq, Eq)]
339pub struct UpdateFailMalformedHTLC {
340 pub channel_id: [u8; 32],
342 pub htlc_id: u64,
344 pub(crate) sha256_of_onion: [u8; 32],
345 pub failure_code: u16,
347}
348
349#[derive(Clone, Debug, PartialEq, Eq)]
351pub struct CommitmentSigned {
352 pub channel_id: [u8; 32],
354 pub signature: Signature,
356 pub htlc_signatures: Vec<Signature>,
358}
359
360#[derive(Clone, Debug, PartialEq, Eq)]
362pub struct RevokeAndACK {
363 pub channel_id: [u8; 32],
365 pub per_commitment_secret: [u8; 32],
367 pub next_per_commitment_point: PublicKey,
369}
370
371#[derive(Clone, Debug, PartialEq, Eq)]
373pub struct UpdateFee {
374 pub channel_id: [u8; 32],
376 pub feerate_per_kw: u32,
378}
379
380#[derive(Clone, Debug, PartialEq, Eq)]
381pub struct DataLossProtect {
386 pub your_last_per_commitment_secret: [u8; 32],
389 pub my_current_per_commitment_point: PublicKey,
391}
392
393#[derive(Clone, Debug, PartialEq, Eq)]
395pub struct ChannelReestablish {
396 pub channel_id: [u8; 32],
398 pub next_local_commitment_number: u64,
400 pub next_remote_commitment_number: u64,
402 pub data_loss_protect: OptionalField<DataLossProtect>,
404}
405
406#[derive(Clone, Debug, PartialEq, Eq)]
408pub struct AnnouncementSignatures {
409 pub channel_id: [u8; 32],
411 pub short_channel_id: u64,
413 pub node_signature: Signature,
415 pub bitcoin_signature: Signature,
417}
418
419#[derive(Clone, Debug, PartialEq, Eq)]
421pub enum NetAddress {
422 IPv4 {
424 addr: [u8; 4],
426 port: u16,
428 },
429 IPv6 {
431 addr: [u8; 16],
433 port: u16,
435 },
436 OnionV2([u8; 12]),
441 OnionV3 {
445 ed25519_pubkey: [u8; 32],
447 checksum: u16,
449 version: u8,
451 port: u16,
453 },
454 Hostname {
456 hostname: Hostname,
458 port: u16,
460 },
461}
462impl NetAddress {
463 pub(crate) fn get_id(&self) -> u8 {
466 match self {
467 &NetAddress::IPv4 {..} => { 1 },
468 &NetAddress::IPv6 {..} => { 2 },
469 &NetAddress::OnionV2(_) => { 3 },
470 &NetAddress::OnionV3 {..} => { 4 },
471 &NetAddress::Hostname {..} => { 5 },
472 }
473 }
474
475 fn len(&self) -> u16 {
477 match self {
478 &NetAddress::IPv4 { .. } => { 6 },
479 &NetAddress::IPv6 { .. } => { 18 },
480 &NetAddress::OnionV2(_) => { 12 },
481 &NetAddress::OnionV3 { .. } => { 37 },
482 &NetAddress::Hostname { ref hostname, .. } => { u16::from(hostname.len()) + 3 },
484 }
485 }
486
487 pub(crate) const MAX_LEN: u16 = 258;
491}
492
493impl Writeable for NetAddress {
494 fn write<W: Writer>(&self, writer: &mut W) -> Result<(), io::Error> {
495 match self {
496 &NetAddress::IPv4 { ref addr, ref port } => {
497 1u8.write(writer)?;
498 addr.write(writer)?;
499 port.write(writer)?;
500 },
501 &NetAddress::IPv6 { ref addr, ref port } => {
502 2u8.write(writer)?;
503 addr.write(writer)?;
504 port.write(writer)?;
505 },
506 &NetAddress::OnionV2(bytes) => {
507 3u8.write(writer)?;
508 bytes.write(writer)?;
509 },
510 &NetAddress::OnionV3 { ref ed25519_pubkey, ref checksum, ref version, ref port } => {
511 4u8.write(writer)?;
512 ed25519_pubkey.write(writer)?;
513 checksum.write(writer)?;
514 version.write(writer)?;
515 port.write(writer)?;
516 },
517 &NetAddress::Hostname { ref hostname, ref port } => {
518 5u8.write(writer)?;
519 hostname.write(writer)?;
520 port.write(writer)?;
521 },
522 }
523 Ok(())
524 }
525}
526
527impl Readable for Result<NetAddress, u8> {
528 fn read<R: Read>(reader: &mut R) -> Result<Result<NetAddress, u8>, DecodeError> {
529 let byte = <u8 as Readable>::read(reader)?;
530 match byte {
531 1 => {
532 Ok(Ok(NetAddress::IPv4 {
533 addr: Readable::read(reader)?,
534 port: Readable::read(reader)?,
535 }))
536 },
537 2 => {
538 Ok(Ok(NetAddress::IPv6 {
539 addr: Readable::read(reader)?,
540 port: Readable::read(reader)?,
541 }))
542 },
543 3 => Ok(Ok(NetAddress::OnionV2(Readable::read(reader)?))),
544 4 => {
545 Ok(Ok(NetAddress::OnionV3 {
546 ed25519_pubkey: Readable::read(reader)?,
547 checksum: Readable::read(reader)?,
548 version: Readable::read(reader)?,
549 port: Readable::read(reader)?,
550 }))
551 },
552 5 => {
553 Ok(Ok(NetAddress::Hostname {
554 hostname: Readable::read(reader)?,
555 port: Readable::read(reader)?,
556 }))
557 },
558 _ => return Ok(Err(byte)),
559 }
560 }
561}
562
563impl Readable for NetAddress {
564 fn read<R: Read>(reader: &mut R) -> Result<NetAddress, DecodeError> {
565 match Readable::read(reader) {
566 Ok(Ok(res)) => Ok(res),
567 Ok(Err(_)) => Err(DecodeError::UnknownVersion),
568 Err(e) => Err(e),
569 }
570 }
571}
572
573
574#[derive(Clone, Debug, PartialEq, Eq)]
576pub struct UnsignedNodeAnnouncement {
577 pub features: NodeFeatures,
579 pub timestamp: u32,
581 pub node_id: PublicKey,
584 pub rgb: [u8; 3],
586 pub alias: [u8; 32],
589 pub addresses: Vec<NetAddress>,
591 pub(crate) excess_address_data: Vec<u8>,
592 pub(crate) excess_data: Vec<u8>,
593}
594#[derive(Clone, Debug, PartialEq, Eq)]
595pub struct NodeAnnouncement {
597 pub signature: Signature,
599 pub contents: UnsignedNodeAnnouncement,
601}
602
603#[derive(Clone, Debug, PartialEq, Eq)]
605pub struct UnsignedChannelAnnouncement {
606 pub features: ChannelFeatures,
608 pub chain_hash: BlockHash,
610 pub short_channel_id: u64,
612 pub node_id_1: PublicKey,
614 pub node_id_2: PublicKey,
616 pub bitcoin_key_1: PublicKey,
618 pub bitcoin_key_2: PublicKey,
620 pub(crate) excess_data: Vec<u8>,
621}
622#[derive(Clone, Debug, PartialEq, Eq)]
624pub struct ChannelAnnouncement {
625 pub node_signature_1: Signature,
627 pub node_signature_2: Signature,
629 pub bitcoin_signature_1: Signature,
631 pub bitcoin_signature_2: Signature,
633 pub contents: UnsignedChannelAnnouncement,
635}
636
637#[derive(Clone, Debug, PartialEq, Eq)]
639pub struct UnsignedChannelUpdate {
640 pub chain_hash: BlockHash,
642 pub short_channel_id: u64,
644 pub timestamp: u32,
646 pub flags: u8,
648 pub cltv_expiry_delta: u16,
657 pub htlc_minimum_msat: u64,
659 pub htlc_maximum_msat: u64,
661 pub fee_base_msat: u32,
663 pub fee_proportional_millionths: u32,
665 pub excess_data: Vec<u8>,
669}
670#[derive(Clone, Debug, PartialEq, Eq)]
672pub struct ChannelUpdate {
673 pub signature: Signature,
675 pub contents: UnsignedChannelUpdate,
677}
678
679#[derive(Clone, Debug, PartialEq, Eq)]
684pub struct QueryChannelRange {
685 pub chain_hash: BlockHash,
687 pub first_blocknum: u32,
689 pub number_of_blocks: u32,
691}
692
693#[derive(Clone, Debug, PartialEq, Eq)]
701pub struct ReplyChannelRange {
702 pub chain_hash: BlockHash,
704 pub first_blocknum: u32,
706 pub number_of_blocks: u32,
708 pub sync_complete: bool,
710 pub short_channel_ids: Vec<u64>,
712}
713
714#[derive(Clone, Debug, PartialEq, Eq)]
723pub struct QueryShortChannelIds {
724 pub chain_hash: BlockHash,
726 pub short_channel_ids: Vec<u64>,
728}
729
730#[derive(Clone, Debug, PartialEq, Eq)]
735pub struct ReplyShortChannelIdsEnd {
736 pub chain_hash: BlockHash,
738 pub full_information: bool,
741}
742
743#[derive(Clone, Debug, PartialEq, Eq)]
747pub struct GossipTimestampFilter {
748 pub chain_hash: BlockHash,
750 pub first_timestamp: u32,
752 pub timestamp_range: u32,
754}
755
756enum EncodingType {
759 Uncompressed = 0x00,
760}
761
762#[derive(Clone, Debug)]
764pub enum ErrorAction {
765 DisconnectPeer {
767 msg: Option<ErrorMessage>
769 },
770 IgnoreError,
773 IgnoreAndLog(logger::Level),
776 IgnoreDuplicateGossip,
780 SendErrorMessage {
782 msg: ErrorMessage,
784 },
785 SendWarningMessage {
787 msg: WarningMessage,
789 log_level: logger::Level,
793 },
794}
795
796#[derive(Clone, Debug)]
798pub struct LightningError {
799 pub err: String,
801 pub action: ErrorAction,
803}
804
805#[derive(Clone, Debug, PartialEq, Eq)]
808pub struct CommitmentUpdate {
809 pub update_add_htlcs: Vec<UpdateAddHTLC>,
811 pub update_fulfill_htlcs: Vec<UpdateFulfillHTLC>,
813 pub update_fail_htlcs: Vec<UpdateFailHTLC>,
815 pub update_fail_malformed_htlcs: Vec<UpdateFailMalformedHTLC>,
817 pub update_fee: Option<UpdateFee>,
819 pub commitment_signed: CommitmentSigned,
821}
822
823#[derive(Clone, Debug, PartialEq, Eq)]
829pub enum OptionalField<T> {
830 Present(T),
832 Absent
834}
835
836pub trait ChannelMessageHandler : MessageSendEventsProvider {
841 fn handle_open_channel(&self, their_node_id: &PublicKey, their_features: InitFeatures, msg: &OpenChannel);
844 fn handle_accept_channel(&self, their_node_id: &PublicKey, their_features: InitFeatures, msg: &AcceptChannel);
846 fn handle_funding_created(&self, their_node_id: &PublicKey, msg: &FundingCreated);
848 fn handle_funding_signed(&self, their_node_id: &PublicKey, msg: &FundingSigned);
850 fn handle_channel_ready(&self, their_node_id: &PublicKey, msg: &ChannelReady);
852
853 fn handle_shutdown(&self, their_node_id: &PublicKey, their_features: &InitFeatures, msg: &Shutdown);
856 fn handle_closing_signed(&self, their_node_id: &PublicKey, msg: &ClosingSigned);
858
859 fn handle_update_add_htlc(&self, their_node_id: &PublicKey, msg: &UpdateAddHTLC);
862 fn handle_update_fulfill_htlc(&self, their_node_id: &PublicKey, msg: &UpdateFulfillHTLC);
864 fn handle_update_fail_htlc(&self, their_node_id: &PublicKey, msg: &UpdateFailHTLC);
866 fn handle_update_fail_malformed_htlc(&self, their_node_id: &PublicKey, msg: &UpdateFailMalformedHTLC);
868 fn handle_commitment_signed(&self, their_node_id: &PublicKey, msg: &CommitmentSigned);
870 fn handle_revoke_and_ack(&self, their_node_id: &PublicKey, msg: &RevokeAndACK);
872
873 fn handle_update_fee(&self, their_node_id: &PublicKey, msg: &UpdateFee);
875
876 fn handle_announcement_signatures(&self, their_node_id: &PublicKey, msg: &AnnouncementSignatures);
879
880 fn peer_disconnected(&self, their_node_id: &PublicKey, no_connection_possible: bool);
889
890 fn peer_connected(&self, their_node_id: &PublicKey, msg: &Init) -> Result<(), ()>;
896 fn handle_channel_reestablish(&self, their_node_id: &PublicKey, msg: &ChannelReestablish);
898
899 fn handle_channel_update(&self, their_node_id: &PublicKey, msg: &ChannelUpdate);
901
902 fn handle_error(&self, their_node_id: &PublicKey, msg: &ErrorMessage);
905
906 fn provided_node_features(&self) -> NodeFeatures;
911
912 fn provided_init_features(&self, their_node_id: &PublicKey) -> InitFeatures;
918}
919
920pub trait RoutingMessageHandler : MessageSendEventsProvider {
928 fn handle_node_announcement(&self, msg: &NodeAnnouncement) -> Result<bool, LightningError>;
931 fn handle_channel_announcement(&self, msg: &ChannelAnnouncement) -> Result<bool, LightningError>;
934 fn handle_channel_update(&self, msg: &ChannelUpdate) -> Result<bool, LightningError>;
937 fn get_next_channel_announcement(&self, starting_point: u64) -> Option<(ChannelAnnouncement, Option<ChannelUpdate>, Option<ChannelUpdate>)>;
941 fn get_next_node_announcement(&self, starting_point: Option<&PublicKey>) -> Option<NodeAnnouncement>;
946 fn peer_connected(&self, their_node_id: &PublicKey, init: &Init) -> Result<(), ()>;
954 fn handle_reply_channel_range(&self, their_node_id: &PublicKey, msg: ReplyChannelRange) -> Result<(), LightningError>;
958 fn handle_reply_short_channel_ids_end(&self, their_node_id: &PublicKey, msg: ReplyShortChannelIdsEnd) -> Result<(), LightningError>;
963 fn handle_query_channel_range(&self, their_node_id: &PublicKey, msg: QueryChannelRange) -> Result<(), LightningError>;
966 fn handle_query_short_channel_ids(&self, their_node_id: &PublicKey, msg: QueryShortChannelIds) -> Result<(), LightningError>;
969
970 fn provided_node_features(&self) -> NodeFeatures;
975 fn provided_init_features(&self, their_node_id: &PublicKey) -> InitFeatures;
981}
982
983pub trait OnionMessageHandler : OnionMessageProvider {
985 fn handle_onion_message(&self, peer_node_id: &PublicKey, msg: &OnionMessage);
987 fn peer_connected(&self, their_node_id: &PublicKey, init: &Init) -> Result<(), ()>;
994 fn peer_disconnected(&self, their_node_id: &PublicKey, no_connection_possible: bool);
1000
1001 fn provided_node_features(&self) -> NodeFeatures;
1006
1007 fn provided_init_features(&self, their_node_id: &PublicKey) -> InitFeatures;
1013}
1014
1015mod fuzzy_internal_msgs {
1016 use crate::prelude::*;
1017 use crate::ln::{PaymentPreimage, PaymentSecret};
1018
1019 #[derive(Clone)]
1022 pub(crate) struct FinalOnionHopData {
1023 pub(crate) payment_secret: PaymentSecret,
1024 pub(crate) total_msat: u64,
1027 }
1028
1029 pub(crate) enum OnionHopDataFormat {
1030 NonFinalNode {
1031 short_channel_id: u64,
1032 },
1033 FinalNode {
1034 payment_data: Option<FinalOnionHopData>,
1035 keysend_preimage: Option<PaymentPreimage>,
1036 },
1037 }
1038
1039 pub struct OnionHopData {
1040 pub(crate) format: OnionHopDataFormat,
1041 pub(crate) amt_to_forward: u64,
1044 pub(crate) outgoing_cltv_value: u32,
1045 }
1046
1047 pub struct DecodedOnionErrorPacket {
1048 pub(crate) hmac: [u8; 32],
1049 pub(crate) failuremsg: Vec<u8>,
1050 pub(crate) pad: Vec<u8>,
1051 }
1052}
1053#[cfg(fuzzing)]
1054pub use self::fuzzy_internal_msgs::*;
1055#[cfg(not(fuzzing))]
1056pub(crate) use self::fuzzy_internal_msgs::*;
1057
1058#[derive(Clone)]
1059pub(crate) struct OnionPacket {
1060 pub(crate) version: u8,
1061 pub(crate) public_key: Result<PublicKey, secp256k1::Error>,
1065 pub(crate) hop_data: [u8; 20*65],
1066 pub(crate) hmac: [u8; 32],
1067}
1068
1069impl onion_utils::Packet for OnionPacket {
1070 type Data = onion_utils::FixedSizeOnionPacket;
1071 fn new(pubkey: PublicKey, hop_data: onion_utils::FixedSizeOnionPacket, hmac: [u8; 32]) -> Self {
1072 Self {
1073 version: 0,
1074 public_key: Ok(pubkey),
1075 hop_data: hop_data.0,
1076 hmac,
1077 }
1078 }
1079}
1080
1081impl Eq for OnionPacket { }
1082impl PartialEq for OnionPacket {
1083 fn eq(&self, other: &OnionPacket) -> bool {
1084 for (i, j) in self.hop_data.iter().zip(other.hop_data.iter()) {
1085 if i != j { return false; }
1086 }
1087 self.version == other.version &&
1088 self.public_key == other.public_key &&
1089 self.hmac == other.hmac
1090 }
1091}
1092
1093impl fmt::Debug for OnionPacket {
1094 fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
1095 f.write_fmt(format_args!("OnionPacket version {} with hmac {:?}", self.version, &self.hmac[..]))
1096 }
1097}
1098
1099#[derive(Clone, Debug, PartialEq, Eq)]
1100pub(crate) struct OnionErrorPacket {
1101 pub(crate) data: Vec<u8>,
1104}
1105
1106impl fmt::Display for DecodeError {
1107 fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
1108 match *self {
1109 DecodeError::UnknownVersion => f.write_str("Unknown realm byte in Onion packet"),
1110 DecodeError::UnknownRequiredFeature => f.write_str("Unknown required feature preventing decode"),
1111 DecodeError::InvalidValue => f.write_str("Nonsense bytes didn't map to the type they were interpreted as"),
1112 DecodeError::ShortRead => f.write_str("Packet extended beyond the provided bytes"),
1113 DecodeError::BadLengthDescriptor => f.write_str("A length descriptor in the packet didn't describe the later data correctly"),
1114 DecodeError::Io(ref e) => fmt::Debug::fmt(e, f),
1115 DecodeError::UnsupportedCompression => f.write_str("We don't support receiving messages with zlib-compressed fields"),
1116 }
1117 }
1118}
1119
1120impl From<io::Error> for DecodeError {
1121 fn from(e: io::Error) -> Self {
1122 if e.kind() == io::ErrorKind::UnexpectedEof {
1123 DecodeError::ShortRead
1124 } else {
1125 DecodeError::Io(e.kind())
1126 }
1127 }
1128}
1129
1130impl Writeable for OptionalField<Script> {
1131 fn write<W: Writer>(&self, w: &mut W) -> Result<(), io::Error> {
1132 match *self {
1133 OptionalField::Present(ref script) => {
1134 script.write(w)?;
1136 },
1137 OptionalField::Absent => {}
1138 }
1139 Ok(())
1140 }
1141}
1142
1143impl Readable for OptionalField<Script> {
1144 fn read<R: Read>(r: &mut R) -> Result<Self, DecodeError> {
1145 match <u16 as Readable>::read(r) {
1146 Ok(len) => {
1147 let mut buf = vec![0; len as usize];
1148 r.read_exact(&mut buf)?;
1149 Ok(OptionalField::Present(Script::from(buf)))
1150 },
1151 Err(DecodeError::ShortRead) => Ok(OptionalField::Absent),
1152 Err(e) => Err(e)
1153 }
1154 }
1155}
1156
1157impl Writeable for OptionalField<u64> {
1158 fn write<W: Writer>(&self, w: &mut W) -> Result<(), io::Error> {
1159 match *self {
1160 OptionalField::Present(ref value) => {
1161 value.write(w)?;
1162 },
1163 OptionalField::Absent => {}
1164 }
1165 Ok(())
1166 }
1167}
1168
1169impl Readable for OptionalField<u64> {
1170 fn read<R: Read>(r: &mut R) -> Result<Self, DecodeError> {
1171 let value: u64 = Readable::read(r)?;
1172 Ok(OptionalField::Present(value))
1173 }
1174}
1175
1176
1177impl_writeable_msg!(AcceptChannel, {
1178 temporary_channel_id,
1179 dust_limit_satoshis,
1180 max_htlc_value_in_flight_msat,
1181 channel_reserve_satoshis,
1182 htlc_minimum_msat,
1183 minimum_depth,
1184 to_self_delay,
1185 max_accepted_htlcs,
1186 funding_pubkey,
1187 revocation_basepoint,
1188 payment_point,
1189 delayed_payment_basepoint,
1190 htlc_basepoint,
1191 first_per_commitment_point,
1192 shutdown_scriptpubkey
1193}, {
1194 (1, channel_type, option),
1195});
1196
1197impl_writeable_msg!(AnnouncementSignatures, {
1198 channel_id,
1199 short_channel_id,
1200 node_signature,
1201 bitcoin_signature
1202}, {});
1203
1204impl Writeable for ChannelReestablish {
1205 fn write<W: Writer>(&self, w: &mut W) -> Result<(), io::Error> {
1206 self.channel_id.write(w)?;
1207 self.next_local_commitment_number.write(w)?;
1208 self.next_remote_commitment_number.write(w)?;
1209 match self.data_loss_protect {
1210 OptionalField::Present(ref data_loss_protect) => {
1211 (*data_loss_protect).your_last_per_commitment_secret.write(w)?;
1212 (*data_loss_protect).my_current_per_commitment_point.write(w)?;
1213 },
1214 OptionalField::Absent => {}
1215 }
1216 Ok(())
1217 }
1218}
1219
1220impl Readable for ChannelReestablish{
1221 fn read<R: Read>(r: &mut R) -> Result<Self, DecodeError> {
1222 Ok(Self {
1223 channel_id: Readable::read(r)?,
1224 next_local_commitment_number: Readable::read(r)?,
1225 next_remote_commitment_number: Readable::read(r)?,
1226 data_loss_protect: {
1227 match <[u8; 32] as Readable>::read(r) {
1228 Ok(your_last_per_commitment_secret) =>
1229 OptionalField::Present(DataLossProtect {
1230 your_last_per_commitment_secret,
1231 my_current_per_commitment_point: Readable::read(r)?,
1232 }),
1233 Err(DecodeError::ShortRead) => OptionalField::Absent,
1234 Err(e) => return Err(e)
1235 }
1236 }
1237 })
1238 }
1239}
1240
1241impl_writeable_msg!(ClosingSigned,
1242 { channel_id, fee_satoshis, signature },
1243 { (1, fee_range, option) }
1244);
1245
1246impl_writeable!(ClosingSignedFeeRange, {
1247 min_fee_satoshis,
1248 max_fee_satoshis
1249});
1250
1251impl_writeable_msg!(CommitmentSigned, {
1252 channel_id,
1253 signature,
1254 htlc_signatures
1255}, {});
1256
1257impl_writeable!(DecodedOnionErrorPacket, {
1258 hmac,
1259 failuremsg,
1260 pad
1261});
1262
1263impl_writeable_msg!(FundingCreated, {
1264 temporary_channel_id,
1265 funding_txid,
1266 funding_output_index,
1267 signature
1268}, {});
1269
1270impl_writeable_msg!(FundingSigned, {
1271 channel_id,
1272 signature
1273}, {});
1274
1275impl_writeable_msg!(ChannelReady, {
1276 channel_id,
1277 next_per_commitment_point,
1278}, {
1279 (1, short_channel_id_alias, option),
1280});
1281
1282impl Writeable for Init {
1283 fn write<W: Writer>(&self, w: &mut W) -> Result<(), io::Error> {
1284 self.features.write_up_to_13(w)?;
1287 self.features.write(w)?;
1288 encode_tlv_stream!(w, {
1289 (3, self.remote_network_address, option)
1290 });
1291 Ok(())
1292 }
1293}
1294
1295impl Readable for Init {
1296 fn read<R: Read>(r: &mut R) -> Result<Self, DecodeError> {
1297 let global_features: InitFeatures = Readable::read(r)?;
1298 let features: InitFeatures = Readable::read(r)?;
1299 let mut remote_network_address: Option<NetAddress> = None;
1300 decode_tlv_stream!(r, {
1301 (3, remote_network_address, option)
1302 });
1303 Ok(Init {
1304 features: features.or(global_features),
1305 remote_network_address,
1306 })
1307 }
1308}
1309
1310impl_writeable_msg!(OpenChannel, {
1311 chain_hash,
1312 temporary_channel_id,
1313 funding_satoshis,
1314 push_msat,
1315 dust_limit_satoshis,
1316 max_htlc_value_in_flight_msat,
1317 channel_reserve_satoshis,
1318 htlc_minimum_msat,
1319 feerate_per_kw,
1320 to_self_delay,
1321 max_accepted_htlcs,
1322 funding_pubkey,
1323 revocation_basepoint,
1324 payment_point,
1325 delayed_payment_basepoint,
1326 htlc_basepoint,
1327 first_per_commitment_point,
1328 channel_flags,
1329 shutdown_scriptpubkey
1330}, {
1331 (1, channel_type, option),
1332});
1333
1334impl_writeable_msg!(RevokeAndACK, {
1335 channel_id,
1336 per_commitment_secret,
1337 next_per_commitment_point
1338}, {});
1339
1340impl_writeable_msg!(Shutdown, {
1341 channel_id,
1342 scriptpubkey
1343}, {});
1344
1345impl_writeable_msg!(UpdateFailHTLC, {
1346 channel_id,
1347 htlc_id,
1348 reason
1349}, {});
1350
1351impl_writeable_msg!(UpdateFailMalformedHTLC, {
1352 channel_id,
1353 htlc_id,
1354 sha256_of_onion,
1355 failure_code
1356}, {});
1357
1358impl_writeable_msg!(UpdateFee, {
1359 channel_id,
1360 feerate_per_kw
1361}, {});
1362
1363impl_writeable_msg!(UpdateFulfillHTLC, {
1364 channel_id,
1365 htlc_id,
1366 payment_preimage
1367}, {});
1368
1369impl_writeable!(OnionErrorPacket, {
1373 data
1374});
1375
1376impl Writeable for OnionPacket {
1380 fn write<W: Writer>(&self, w: &mut W) -> Result<(), io::Error> {
1381 self.version.write(w)?;
1382 match self.public_key {
1383 Ok(pubkey) => pubkey.write(w)?,
1384 Err(_) => [0u8;33].write(w)?,
1385 }
1386 w.write_all(&self.hop_data)?;
1387 self.hmac.write(w)?;
1388 Ok(())
1389 }
1390}
1391
1392impl Readable for OnionPacket {
1393 fn read<R: Read>(r: &mut R) -> Result<Self, DecodeError> {
1394 Ok(OnionPacket {
1395 version: Readable::read(r)?,
1396 public_key: {
1397 let mut buf = [0u8;33];
1398 r.read_exact(&mut buf)?;
1399 PublicKey::from_slice(&buf)
1400 },
1401 hop_data: Readable::read(r)?,
1402 hmac: Readable::read(r)?,
1403 })
1404 }
1405}
1406
1407impl_writeable_msg!(UpdateAddHTLC, {
1408 channel_id,
1409 htlc_id,
1410 amount_msat,
1411 payment_hash,
1412 cltv_expiry,
1413 onion_routing_packet
1414}, {});
1415
1416impl Readable for OnionMessage {
1417 fn read<R: Read>(r: &mut R) -> Result<Self, DecodeError> {
1418 let blinding_point: PublicKey = Readable::read(r)?;
1419 let len: u16 = Readable::read(r)?;
1420 let mut packet_reader = FixedLengthReader::new(r, len as u64);
1421 let onion_routing_packet: onion_message::Packet = <onion_message::Packet as LengthReadable>::read(&mut packet_reader)?;
1422 Ok(Self {
1423 blinding_point,
1424 onion_routing_packet,
1425 })
1426 }
1427}
1428
1429impl Writeable for OnionMessage {
1430 fn write<W: Writer>(&self, w: &mut W) -> Result<(), io::Error> {
1431 self.blinding_point.write(w)?;
1432 let onion_packet_len = self.onion_routing_packet.serialized_length();
1433 (onion_packet_len as u16).write(w)?;
1434 self.onion_routing_packet.write(w)?;
1435 Ok(())
1436 }
1437}
1438
1439impl Writeable for FinalOnionHopData {
1440 fn write<W: Writer>(&self, w: &mut W) -> Result<(), io::Error> {
1441 self.payment_secret.0.write(w)?;
1442 HighZeroBytesDroppedBigSize(self.total_msat).write(w)
1443 }
1444}
1445
1446impl Readable for FinalOnionHopData {
1447 fn read<R: Read>(r: &mut R) -> Result<Self, DecodeError> {
1448 let secret: [u8; 32] = Readable::read(r)?;
1449 let amt: HighZeroBytesDroppedBigSize<u64> = Readable::read(r)?;
1450 Ok(Self { payment_secret: PaymentSecret(secret), total_msat: amt.0 })
1451 }
1452}
1453
1454impl Writeable for OnionHopData {
1455 fn write<W: Writer>(&self, w: &mut W) -> Result<(), io::Error> {
1456 match self.format {
1457 OnionHopDataFormat::NonFinalNode { short_channel_id } => {
1458 encode_varint_length_prefixed_tlv!(w, {
1459 (2, HighZeroBytesDroppedBigSize(self.amt_to_forward), required),
1460 (4, HighZeroBytesDroppedBigSize(self.outgoing_cltv_value), required),
1461 (6, short_channel_id, required)
1462 });
1463 },
1464 OnionHopDataFormat::FinalNode { ref payment_data, ref keysend_preimage } => {
1465 encode_varint_length_prefixed_tlv!(w, {
1466 (2, HighZeroBytesDroppedBigSize(self.amt_to_forward), required),
1467 (4, HighZeroBytesDroppedBigSize(self.outgoing_cltv_value), required),
1468 (8, payment_data, option),
1469 (5482373484, keysend_preimage, option)
1470 });
1471 },
1472 }
1473 Ok(())
1474 }
1475}
1476
1477impl Readable for OnionHopData {
1478 fn read<R: Read>(r: &mut R) -> Result<Self, DecodeError> {
1479 let mut amt = HighZeroBytesDroppedBigSize(0u64);
1480 let mut cltv_value = HighZeroBytesDroppedBigSize(0u32);
1481 let mut short_id: Option<u64> = None;
1482 let mut payment_data: Option<FinalOnionHopData> = None;
1483 let mut keysend_preimage: Option<PaymentPreimage> = None;
1484 read_tlv_fields!(r, {
1485 (2, amt, required),
1486 (4, cltv_value, required),
1487 (6, short_id, option),
1488 (8, payment_data, option),
1489 (5482373484, keysend_preimage, option)
1491 });
1492
1493 let format = if let Some(short_channel_id) = short_id {
1494 if payment_data.is_some() { return Err(DecodeError::InvalidValue); }
1495 OnionHopDataFormat::NonFinalNode {
1496 short_channel_id,
1497 }
1498 } else {
1499 if let &Some(ref data) = &payment_data {
1500 if data.total_msat > MAX_VALUE_MSAT {
1501 return Err(DecodeError::InvalidValue);
1502 }
1503 }
1504 OnionHopDataFormat::FinalNode {
1505 payment_data,
1506 keysend_preimage,
1507 }
1508 };
1509
1510 if amt.0 > MAX_VALUE_MSAT {
1511 return Err(DecodeError::InvalidValue);
1512 }
1513 Ok(OnionHopData {
1514 format,
1515 amt_to_forward: amt.0,
1516 outgoing_cltv_value: cltv_value.0,
1517 })
1518 }
1519}
1520
1521impl ReadableArgs<()> for OnionHopData {
1524 fn read<R: Read>(r: &mut R, _arg: ()) -> Result<Self, DecodeError> {
1525 <Self as Readable>::read(r)
1526 }
1527}
1528
1529impl Writeable for Ping {
1530 fn write<W: Writer>(&self, w: &mut W) -> Result<(), io::Error> {
1531 self.ponglen.write(w)?;
1532 vec![0u8; self.byteslen as usize].write(w)?; Ok(())
1534 }
1535}
1536
1537impl Readable for Ping {
1538 fn read<R: Read>(r: &mut R) -> Result<Self, DecodeError> {
1539 Ok(Ping {
1540 ponglen: Readable::read(r)?,
1541 byteslen: {
1542 let byteslen = Readable::read(r)?;
1543 r.read_exact(&mut vec![0u8; byteslen as usize][..])?;
1544 byteslen
1545 }
1546 })
1547 }
1548}
1549
1550impl Writeable for Pong {
1551 fn write<W: Writer>(&self, w: &mut W) -> Result<(), io::Error> {
1552 vec![0u8; self.byteslen as usize].write(w)?; Ok(())
1554 }
1555}
1556
1557impl Readable for Pong {
1558 fn read<R: Read>(r: &mut R) -> Result<Self, DecodeError> {
1559 Ok(Pong {
1560 byteslen: {
1561 let byteslen = Readable::read(r)?;
1562 r.read_exact(&mut vec![0u8; byteslen as usize][..])?;
1563 byteslen
1564 }
1565 })
1566 }
1567}
1568
1569impl Writeable for UnsignedChannelAnnouncement {
1570 fn write<W: Writer>(&self, w: &mut W) -> Result<(), io::Error> {
1571 self.features.write(w)?;
1572 self.chain_hash.write(w)?;
1573 self.short_channel_id.write(w)?;
1574 self.node_id_1.write(w)?;
1575 self.node_id_2.write(w)?;
1576 self.bitcoin_key_1.write(w)?;
1577 self.bitcoin_key_2.write(w)?;
1578 w.write_all(&self.excess_data[..])?;
1579 Ok(())
1580 }
1581}
1582
1583impl Readable for UnsignedChannelAnnouncement {
1584 fn read<R: Read>(r: &mut R) -> Result<Self, DecodeError> {
1585 Ok(Self {
1586 features: Readable::read(r)?,
1587 chain_hash: Readable::read(r)?,
1588 short_channel_id: Readable::read(r)?,
1589 node_id_1: Readable::read(r)?,
1590 node_id_2: Readable::read(r)?,
1591 bitcoin_key_1: Readable::read(r)?,
1592 bitcoin_key_2: Readable::read(r)?,
1593 excess_data: read_to_end(r)?,
1594 })
1595 }
1596}
1597
1598impl_writeable!(ChannelAnnouncement, {
1599 node_signature_1,
1600 node_signature_2,
1601 bitcoin_signature_1,
1602 bitcoin_signature_2,
1603 contents
1604});
1605
1606impl Writeable for UnsignedChannelUpdate {
1607 fn write<W: Writer>(&self, w: &mut W) -> Result<(), io::Error> {
1608 const MESSAGE_FLAGS: u8 = 1;
1610 self.chain_hash.write(w)?;
1611 self.short_channel_id.write(w)?;
1612 self.timestamp.write(w)?;
1613 let all_flags = self.flags as u16 | ((MESSAGE_FLAGS as u16) << 8);
1614 all_flags.write(w)?;
1615 self.cltv_expiry_delta.write(w)?;
1616 self.htlc_minimum_msat.write(w)?;
1617 self.fee_base_msat.write(w)?;
1618 self.fee_proportional_millionths.write(w)?;
1619 self.htlc_maximum_msat.write(w)?;
1620 w.write_all(&self.excess_data[..])?;
1621 Ok(())
1622 }
1623}
1624
1625impl Readable for UnsignedChannelUpdate {
1626 fn read<R: Read>(r: &mut R) -> Result<Self, DecodeError> {
1627 Ok(Self {
1628 chain_hash: Readable::read(r)?,
1629 short_channel_id: Readable::read(r)?,
1630 timestamp: Readable::read(r)?,
1631 flags: {
1632 let flags: u16 = Readable::read(r)?;
1633 flags as u8
1635 },
1636 cltv_expiry_delta: Readable::read(r)?,
1637 htlc_minimum_msat: Readable::read(r)?,
1638 fee_base_msat: Readable::read(r)?,
1639 fee_proportional_millionths: Readable::read(r)?,
1640 htlc_maximum_msat: Readable::read(r)?,
1641 excess_data: read_to_end(r)?,
1642 })
1643 }
1644}
1645
1646impl_writeable!(ChannelUpdate, {
1647 signature,
1648 contents
1649});
1650
1651impl Writeable for ErrorMessage {
1652 fn write<W: Writer>(&self, w: &mut W) -> Result<(), io::Error> {
1653 self.channel_id.write(w)?;
1654 (self.data.len() as u16).write(w)?;
1655 w.write_all(self.data.as_bytes())?;
1656 Ok(())
1657 }
1658}
1659
1660impl Readable for ErrorMessage {
1661 fn read<R: Read>(r: &mut R) -> Result<Self, DecodeError> {
1662 Ok(Self {
1663 channel_id: Readable::read(r)?,
1664 data: {
1665 let sz: usize = <u16 as Readable>::read(r)? as usize;
1666 let mut data = Vec::with_capacity(sz);
1667 data.resize(sz, 0);
1668 r.read_exact(&mut data)?;
1669 match String::from_utf8(data) {
1670 Ok(s) => s,
1671 Err(_) => return Err(DecodeError::InvalidValue),
1672 }
1673 }
1674 })
1675 }
1676}
1677
1678impl Writeable for WarningMessage {
1679 fn write<W: Writer>(&self, w: &mut W) -> Result<(), io::Error> {
1680 self.channel_id.write(w)?;
1681 (self.data.len() as u16).write(w)?;
1682 w.write_all(self.data.as_bytes())?;
1683 Ok(())
1684 }
1685}
1686
1687impl Readable for WarningMessage {
1688 fn read<R: Read>(r: &mut R) -> Result<Self, DecodeError> {
1689 Ok(Self {
1690 channel_id: Readable::read(r)?,
1691 data: {
1692 let sz: usize = <u16 as Readable>::read(r)? as usize;
1693 let mut data = Vec::with_capacity(sz);
1694 data.resize(sz, 0);
1695 r.read_exact(&mut data)?;
1696 match String::from_utf8(data) {
1697 Ok(s) => s,
1698 Err(_) => return Err(DecodeError::InvalidValue),
1699 }
1700 }
1701 })
1702 }
1703}
1704
1705impl Writeable for UnsignedNodeAnnouncement {
1706 fn write<W: Writer>(&self, w: &mut W) -> Result<(), io::Error> {
1707 self.features.write(w)?;
1708 self.timestamp.write(w)?;
1709 self.node_id.write(w)?;
1710 w.write_all(&self.rgb)?;
1711 self.alias.write(w)?;
1712
1713 let mut addr_len = 0;
1714 for addr in self.addresses.iter() {
1715 addr_len += 1 + addr.len();
1716 }
1717 (addr_len + self.excess_address_data.len() as u16).write(w)?;
1718 for addr in self.addresses.iter() {
1719 addr.write(w)?;
1720 }
1721 w.write_all(&self.excess_address_data[..])?;
1722 w.write_all(&self.excess_data[..])?;
1723 Ok(())
1724 }
1725}
1726
1727impl Readable for UnsignedNodeAnnouncement {
1728 fn read<R: Read>(r: &mut R) -> Result<Self, DecodeError> {
1729 let features: NodeFeatures = Readable::read(r)?;
1730 let timestamp: u32 = Readable::read(r)?;
1731 let node_id: PublicKey = Readable::read(r)?;
1732 let mut rgb = [0; 3];
1733 r.read_exact(&mut rgb)?;
1734 let alias: [u8; 32] = Readable::read(r)?;
1735
1736 let addr_len: u16 = Readable::read(r)?;
1737 let mut addresses: Vec<NetAddress> = Vec::new();
1738 let mut addr_readpos = 0;
1739 let mut excess = false;
1740 let mut excess_byte = 0;
1741 loop {
1742 if addr_len <= addr_readpos { break; }
1743 match Readable::read(r) {
1744 Ok(Ok(addr)) => {
1745 if addr_len < addr_readpos + 1 + addr.len() {
1746 return Err(DecodeError::BadLengthDescriptor);
1747 }
1748 addr_readpos += (1 + addr.len()) as u16;
1749 addresses.push(addr);
1750 },
1751 Ok(Err(unknown_descriptor)) => {
1752 excess = true;
1753 excess_byte = unknown_descriptor;
1754 break;
1755 },
1756 Err(DecodeError::ShortRead) => return Err(DecodeError::BadLengthDescriptor),
1757 Err(e) => return Err(e),
1758 }
1759 }
1760
1761 let mut excess_data = vec![];
1762 let excess_address_data = if addr_readpos < addr_len {
1763 let mut excess_address_data = vec![0; (addr_len - addr_readpos) as usize];
1764 r.read_exact(&mut excess_address_data[if excess { 1 } else { 0 }..])?;
1765 if excess {
1766 excess_address_data[0] = excess_byte;
1767 }
1768 excess_address_data
1769 } else {
1770 if excess {
1771 excess_data.push(excess_byte);
1772 }
1773 Vec::new()
1774 };
1775 excess_data.extend(read_to_end(r)?.iter());
1776 Ok(UnsignedNodeAnnouncement {
1777 features,
1778 timestamp,
1779 node_id,
1780 rgb,
1781 alias,
1782 addresses,
1783 excess_address_data,
1784 excess_data,
1785 })
1786 }
1787}
1788
1789impl_writeable!(NodeAnnouncement, {
1790 signature,
1791 contents
1792});
1793
1794impl Readable for QueryShortChannelIds {
1795 fn read<R: Read>(r: &mut R) -> Result<Self, DecodeError> {
1796 let chain_hash: BlockHash = Readable::read(r)?;
1797
1798 let encoding_len: u16 = Readable::read(r)?;
1799 let encoding_type: u8 = Readable::read(r)?;
1800
1801 if encoding_type != EncodingType::Uncompressed as u8 {
1804 return Err(DecodeError::UnsupportedCompression);
1805 }
1806
1807 if encoding_len == 0 || (encoding_len - 1) % 8 != 0 {
1810 return Err(DecodeError::InvalidValue);
1811 }
1812
1813 let short_channel_id_count: u16 = (encoding_len - 1)/8;
1816 let mut short_channel_ids = Vec::with_capacity(short_channel_id_count as usize);
1817 for _ in 0..short_channel_id_count {
1818 short_channel_ids.push(Readable::read(r)?);
1819 }
1820
1821 Ok(QueryShortChannelIds {
1822 chain_hash,
1823 short_channel_ids,
1824 })
1825 }
1826}
1827
1828impl Writeable for QueryShortChannelIds {
1829 fn write<W: Writer>(&self, w: &mut W) -> Result<(), io::Error> {
1830 let encoding_len: u16 = 1 + self.short_channel_ids.len() as u16 * 8;
1832
1833 self.chain_hash.write(w)?;
1834 encoding_len.write(w)?;
1835
1836 (EncodingType::Uncompressed as u8).write(w)?;
1838
1839 for scid in self.short_channel_ids.iter() {
1840 scid.write(w)?;
1841 }
1842
1843 Ok(())
1844 }
1845}
1846
1847impl_writeable_msg!(ReplyShortChannelIdsEnd, {
1848 chain_hash,
1849 full_information,
1850}, {});
1851
1852impl QueryChannelRange {
1853 pub fn end_blocknum(&self) -> u32 {
1858 match self.first_blocknum.checked_add(self.number_of_blocks) {
1859 Some(block) => block,
1860 None => u32::max_value(),
1861 }
1862 }
1863}
1864
1865impl_writeable_msg!(QueryChannelRange, {
1866 chain_hash,
1867 first_blocknum,
1868 number_of_blocks
1869}, {});
1870
1871impl Readable for ReplyChannelRange {
1872 fn read<R: Read>(r: &mut R) -> Result<Self, DecodeError> {
1873 let chain_hash: BlockHash = Readable::read(r)?;
1874 let first_blocknum: u32 = Readable::read(r)?;
1875 let number_of_blocks: u32 = Readable::read(r)?;
1876 let sync_complete: bool = Readable::read(r)?;
1877
1878 let encoding_len: u16 = Readable::read(r)?;
1879 let encoding_type: u8 = Readable::read(r)?;
1880
1881 if encoding_type != EncodingType::Uncompressed as u8 {
1884 return Err(DecodeError::UnsupportedCompression);
1885 }
1886
1887 if encoding_len == 0 || (encoding_len - 1) % 8 != 0 {
1890 return Err(DecodeError::InvalidValue);
1891 }
1892
1893 let short_channel_id_count: u16 = (encoding_len - 1)/8;
1896 let mut short_channel_ids = Vec::with_capacity(short_channel_id_count as usize);
1897 for _ in 0..short_channel_id_count {
1898 short_channel_ids.push(Readable::read(r)?);
1899 }
1900
1901 Ok(ReplyChannelRange {
1902 chain_hash,
1903 first_blocknum,
1904 number_of_blocks,
1905 sync_complete,
1906 short_channel_ids
1907 })
1908 }
1909}
1910
1911impl Writeable for ReplyChannelRange {
1912 fn write<W: Writer>(&self, w: &mut W) -> Result<(), io::Error> {
1913 let encoding_len: u16 = 1 + self.short_channel_ids.len() as u16 * 8;
1914 self.chain_hash.write(w)?;
1915 self.first_blocknum.write(w)?;
1916 self.number_of_blocks.write(w)?;
1917 self.sync_complete.write(w)?;
1918
1919 encoding_len.write(w)?;
1920 (EncodingType::Uncompressed as u8).write(w)?;
1921 for scid in self.short_channel_ids.iter() {
1922 scid.write(w)?;
1923 }
1924
1925 Ok(())
1926 }
1927}
1928
1929impl_writeable_msg!(GossipTimestampFilter, {
1930 chain_hash,
1931 first_timestamp,
1932 timestamp_range,
1933}, {});
1934
1935#[cfg(test)]
1936mod tests {
1937 use hex;
1938 use crate::ln::{PaymentPreimage, PaymentHash, PaymentSecret};
1939 use crate::ln::features::{ChannelFeatures, ChannelTypeFeatures, InitFeatures, NodeFeatures};
1940 use crate::ln::msgs;
1941 use crate::ln::msgs::{FinalOnionHopData, OptionalField, OnionErrorPacket, OnionHopDataFormat};
1942 use crate::util::ser::{Writeable, Readable, Hostname};
1943
1944 use bitcoin::hashes::hex::FromHex;
1945 use bitcoin::util::address::Address;
1946 use bitcoin::network::constants::Network;
1947 use bitcoin::blockdata::script::Builder;
1948 use bitcoin::blockdata::opcodes;
1949 use bitcoin::hash_types::{Txid, BlockHash};
1950
1951 use bitcoin::secp256k1::{PublicKey,SecretKey};
1952 use bitcoin::secp256k1::{Secp256k1, Message};
1953
1954 use crate::io::{self, Cursor};
1955 use crate::prelude::*;
1956 use core::convert::TryFrom;
1957
1958 #[test]
1959 fn encoding_channel_reestablish_no_secret() {
1960 let cr = msgs::ChannelReestablish {
1961 channel_id: [4, 0, 0, 0, 0, 0, 0, 0, 5, 0, 0, 0, 0, 0, 0, 0, 6, 0, 0, 0, 0, 0, 0, 0, 7, 0, 0, 0, 0, 0, 0, 0],
1962 next_local_commitment_number: 3,
1963 next_remote_commitment_number: 4,
1964 data_loss_protect: OptionalField::Absent,
1965 };
1966
1967 let encoded_value = cr.encode();
1968 assert_eq!(
1969 encoded_value,
1970 vec![4, 0, 0, 0, 0, 0, 0, 0, 5, 0, 0, 0, 0, 0, 0, 0, 6, 0, 0, 0, 0, 0, 0, 0, 7, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 3, 0, 0, 0, 0, 0, 0, 0, 4]
1971 );
1972 }
1973
1974 #[test]
1975 fn encoding_channel_reestablish_with_secret() {
1976 let public_key = {
1977 let secp_ctx = Secp256k1::new();
1978 PublicKey::from_secret_key(&secp_ctx, &SecretKey::from_slice(&hex::decode("0101010101010101010101010101010101010101010101010101010101010101").unwrap()[..]).unwrap())
1979 };
1980
1981 let cr = msgs::ChannelReestablish {
1982 channel_id: [4, 0, 0, 0, 0, 0, 0, 0, 5, 0, 0, 0, 0, 0, 0, 0, 6, 0, 0, 0, 0, 0, 0, 0, 7, 0, 0, 0, 0, 0, 0, 0],
1983 next_local_commitment_number: 3,
1984 next_remote_commitment_number: 4,
1985 data_loss_protect: OptionalField::Present(msgs::DataLossProtect { your_last_per_commitment_secret: [9;32], my_current_per_commitment_point: public_key}),
1986 };
1987
1988 let encoded_value = cr.encode();
1989 assert_eq!(
1990 encoded_value,
1991 vec![4, 0, 0, 0, 0, 0, 0, 0, 5, 0, 0, 0, 0, 0, 0, 0, 6, 0, 0, 0, 0, 0, 0, 0, 7, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 3, 0, 0, 0, 0, 0, 0, 0, 4, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 3, 27, 132, 197, 86, 123, 18, 100, 64, 153, 93, 62, 213, 170, 186, 5, 101, 215, 30, 24, 52, 96, 72, 25, 255, 156, 23, 245, 233, 213, 221, 7, 143]
1992 );
1993 }
1994
1995 macro_rules! get_keys_from {
1996 ($slice: expr, $secp_ctx: expr) => {
1997 {
1998 let privkey = SecretKey::from_slice(&hex::decode($slice).unwrap()[..]).unwrap();
1999 let pubkey = PublicKey::from_secret_key(&$secp_ctx, &privkey);
2000 (privkey, pubkey)
2001 }
2002 }
2003 }
2004
2005 macro_rules! get_sig_on {
2006 ($privkey: expr, $ctx: expr, $string: expr) => {
2007 {
2008 let sighash = Message::from_slice(&$string.into_bytes()[..]).unwrap();
2009 $ctx.sign_ecdsa(&sighash, &$privkey)
2010 }
2011 }
2012 }
2013
2014 #[test]
2015 fn encoding_announcement_signatures() {
2016 let secp_ctx = Secp256k1::new();
2017 let (privkey, _) = get_keys_from!("0101010101010101010101010101010101010101010101010101010101010101", secp_ctx);
2018 let sig_1 = get_sig_on!(privkey, secp_ctx, String::from("01010101010101010101010101010101"));
2019 let sig_2 = get_sig_on!(privkey, secp_ctx, String::from("02020202020202020202020202020202"));
2020 let announcement_signatures = msgs::AnnouncementSignatures {
2021 channel_id: [4, 0, 0, 0, 0, 0, 0, 0, 5, 0, 0, 0, 0, 0, 0, 0, 6, 0, 0, 0, 0, 0, 0, 0, 7, 0, 0, 0, 0, 0, 0, 0],
2022 short_channel_id: 2316138423780173,
2023 node_signature: sig_1,
2024 bitcoin_signature: sig_2,
2025 };
2026
2027 let encoded_value = announcement_signatures.encode();
2028 assert_eq!(encoded_value, hex::decode("040000000000000005000000000000000600000000000000070000000000000000083a840000034dd977cb9b53d93a6ff64bb5f1e158b4094b66e798fb12911168a3ccdf80a83096340a6a95da0ae8d9f776528eecdbb747eb6b545495a4319ed5378e35b21e073acf9953cef4700860f5967838eba2bae89288ad188ebf8b20bf995c3ea53a26df1876d0a3a0e13172ba286a673140190c02ba9da60a2e43a745188c8a83c7f3ef").unwrap());
2029 }
2030
2031 fn do_encoding_channel_announcement(unknown_features_bits: bool, excess_data: bool) {
2032 let secp_ctx = Secp256k1::new();
2033 let (privkey_1, pubkey_1) = get_keys_from!("0101010101010101010101010101010101010101010101010101010101010101", secp_ctx);
2034 let (privkey_2, pubkey_2) = get_keys_from!("0202020202020202020202020202020202020202020202020202020202020202", secp_ctx);
2035 let (privkey_3, pubkey_3) = get_keys_from!("0303030303030303030303030303030303030303030303030303030303030303", secp_ctx);
2036 let (privkey_4, pubkey_4) = get_keys_from!("0404040404040404040404040404040404040404040404040404040404040404", secp_ctx);
2037 let sig_1 = get_sig_on!(privkey_1, secp_ctx, String::from("01010101010101010101010101010101"));
2038 let sig_2 = get_sig_on!(privkey_2, secp_ctx, String::from("01010101010101010101010101010101"));
2039 let sig_3 = get_sig_on!(privkey_3, secp_ctx, String::from("01010101010101010101010101010101"));
2040 let sig_4 = get_sig_on!(privkey_4, secp_ctx, String::from("01010101010101010101010101010101"));
2041 let mut features = ChannelFeatures::empty();
2042 if unknown_features_bits {
2043 features = ChannelFeatures::from_le_bytes(vec![0xFF, 0xFF]);
2044 }
2045 let unsigned_channel_announcement = msgs::UnsignedChannelAnnouncement {
2046 features,
2047 chain_hash: BlockHash::from_hex("6fe28c0ab6f1b372c1a6a246ae63f74f931e8365e15a089c68d6190000000000").unwrap(),
2048 short_channel_id: 2316138423780173,
2049 node_id_1: pubkey_1,
2050 node_id_2: pubkey_2,
2051 bitcoin_key_1: pubkey_3,
2052 bitcoin_key_2: pubkey_4,
2053 excess_data: if excess_data { vec![10, 0, 0, 20, 0, 0, 30, 0, 0, 40] } else { Vec::new() },
2054 };
2055 let channel_announcement = msgs::ChannelAnnouncement {
2056 node_signature_1: sig_1,
2057 node_signature_2: sig_2,
2058 bitcoin_signature_1: sig_3,
2059 bitcoin_signature_2: sig_4,
2060 contents: unsigned_channel_announcement,
2061 };
2062 let encoded_value = channel_announcement.encode();
2063 let mut target_value = hex::decode("d977cb9b53d93a6ff64bb5f1e158b4094b66e798fb12911168a3ccdf80a83096340a6a95da0ae8d9f776528eecdbb747eb6b545495a4319ed5378e35b21e073a1735b6a427e80d5fe7cd90a2f4ee08dc9c27cda7c35a4172e5d85b12c49d4232537e98f9b1f3c5e6989a8b9644e90e8918127680dbd0d4043510840fc0f1e11a216c280b5395a2546e7e4b2663e04f811622f15a4f91e83aa2e92ba2a573c139142c54ae63072a1ec1ee7dc0c04bde5c847806172aa05c92c22ae8e308d1d2692b12cc195ce0a2d1bda6a88befa19fa07f51caa75ce83837f28965600b8aacab0855ffb0e741ec5f7c41421e9829a9d48611c8c831f71be5ea73e66594977ffd").unwrap();
2064 if unknown_features_bits {
2065 target_value.append(&mut hex::decode("0002ffff").unwrap());
2066 } else {
2067 target_value.append(&mut hex::decode("0000").unwrap());
2068 }
2069 target_value.append(&mut hex::decode("000000000019d6689c085ae165831e934ff763ae46a2a6c172b3f1b60a8ce26f").unwrap());
2070 target_value.append(&mut hex::decode("00083a840000034d031b84c5567b126440995d3ed5aaba0565d71e1834604819ff9c17f5e9d5dd078f024d4b6cd1361032ca9bd2aeb9d900aa4d45d9ead80ac9423374c451a7254d076602531fe6068134503d2723133227c867ac8fa6c83c537e9a44c3c5bdbdcb1fe33703462779ad4aad39514614751a71085f2f10e1c7a593e4e030efb5b8721ce55b0b").unwrap());
2071 if excess_data {
2072 target_value.append(&mut hex::decode("0a00001400001e000028").unwrap());
2073 }
2074 assert_eq!(encoded_value, target_value);
2075 }
2076
2077 #[test]
2078 fn encoding_channel_announcement() {
2079 do_encoding_channel_announcement(true, false);
2080 do_encoding_channel_announcement(false, true);
2081 do_encoding_channel_announcement(false, false);
2082 do_encoding_channel_announcement(true, true);
2083 }
2084
2085 fn do_encoding_node_announcement(unknown_features_bits: bool, ipv4: bool, ipv6: bool, onionv2: bool, onionv3: bool, hostname: bool, excess_address_data: bool, excess_data: bool) {
2086 let secp_ctx = Secp256k1::new();
2087 let (privkey_1, pubkey_1) = get_keys_from!("0101010101010101010101010101010101010101010101010101010101010101", secp_ctx);
2088 let sig_1 = get_sig_on!(privkey_1, secp_ctx, String::from("01010101010101010101010101010101"));
2089 let features = if unknown_features_bits {
2090 NodeFeatures::from_le_bytes(vec![0xFF, 0xFF])
2091 } else {
2092 NodeFeatures::from_le_bytes(vec![2 | 1 << 5])
2094 };
2095 let mut addresses = Vec::new();
2096 if ipv4 {
2097 addresses.push(msgs::NetAddress::IPv4 {
2098 addr: [255, 254, 253, 252],
2099 port: 9735
2100 });
2101 }
2102 if ipv6 {
2103 addresses.push(msgs::NetAddress::IPv6 {
2104 addr: [255, 254, 253, 252, 251, 250, 249, 248, 247, 246, 245, 244, 243, 242, 241, 240],
2105 port: 9735
2106 });
2107 }
2108 if onionv2 {
2109 addresses.push(msgs::NetAddress::OnionV2(
2110 [255, 254, 253, 252, 251, 250, 249, 248, 247, 246, 38, 7]
2111 ));
2112 }
2113 if onionv3 {
2114 addresses.push(msgs::NetAddress::OnionV3 {
2115 ed25519_pubkey: [255, 254, 253, 252, 251, 250, 249, 248, 247, 246, 245, 244, 243, 242, 241, 240, 239, 238, 237, 236, 235, 234, 233, 232, 231, 230, 229, 228, 227, 226, 225, 224],
2116 checksum: 32,
2117 version: 16,
2118 port: 9735
2119 });
2120 }
2121 if hostname {
2122 addresses.push(msgs::NetAddress::Hostname {
2123 hostname: Hostname::try_from(String::from("host")).unwrap(),
2124 port: 9735,
2125 });
2126 }
2127 let mut addr_len = 0;
2128 for addr in &addresses {
2129 addr_len += addr.len() + 1;
2130 }
2131 let unsigned_node_announcement = msgs::UnsignedNodeAnnouncement {
2132 features,
2133 timestamp: 20190119,
2134 node_id: pubkey_1,
2135 rgb: [32; 3],
2136 alias: [16;32],
2137 addresses,
2138 excess_address_data: if excess_address_data { vec![33, 108, 40, 11, 83, 149, 162, 84, 110, 126, 75, 38, 99, 224, 79, 129, 22, 34, 241, 90, 79, 146, 232, 58, 162, 233, 43, 162, 165, 115, 193, 57, 20, 44, 84, 174, 99, 7, 42, 30, 193, 238, 125, 192, 192, 75, 222, 92, 132, 120, 6, 23, 42, 160, 92, 146, 194, 42, 232, 227, 8, 209, 210, 105] } else { Vec::new() },
2139 excess_data: if excess_data { vec![59, 18, 204, 25, 92, 224, 162, 209, 189, 166, 168, 139, 239, 161, 159, 160, 127, 81, 202, 167, 92, 232, 56, 55, 242, 137, 101, 96, 11, 138, 172, 171, 8, 85, 255, 176, 231, 65, 236, 95, 124, 65, 66, 30, 152, 41, 169, 212, 134, 17, 200, 200, 49, 247, 27, 229, 234, 115, 230, 101, 148, 151, 127, 253] } else { Vec::new() },
2140 };
2141 addr_len += unsigned_node_announcement.excess_address_data.len() as u16;
2142 let node_announcement = msgs::NodeAnnouncement {
2143 signature: sig_1,
2144 contents: unsigned_node_announcement,
2145 };
2146 let encoded_value = node_announcement.encode();
2147 let mut target_value = hex::decode("d977cb9b53d93a6ff64bb5f1e158b4094b66e798fb12911168a3ccdf80a83096340a6a95da0ae8d9f776528eecdbb747eb6b545495a4319ed5378e35b21e073a").unwrap();
2148 if unknown_features_bits {
2149 target_value.append(&mut hex::decode("0002ffff").unwrap());
2150 } else {
2151 target_value.append(&mut hex::decode("000122").unwrap());
2152 }
2153 target_value.append(&mut hex::decode("013413a7031b84c5567b126440995d3ed5aaba0565d71e1834604819ff9c17f5e9d5dd078f2020201010101010101010101010101010101010101010101010101010101010101010").unwrap());
2154 target_value.append(&mut vec![(addr_len >> 8) as u8, addr_len as u8]);
2155 if ipv4 {
2156 target_value.append(&mut hex::decode("01fffefdfc2607").unwrap());
2157 }
2158 if ipv6 {
2159 target_value.append(&mut hex::decode("02fffefdfcfbfaf9f8f7f6f5f4f3f2f1f02607").unwrap());
2160 }
2161 if onionv2 {
2162 target_value.append(&mut hex::decode("03fffefdfcfbfaf9f8f7f62607").unwrap());
2163 }
2164 if onionv3 {
2165 target_value.append(&mut hex::decode("04fffefdfcfbfaf9f8f7f6f5f4f3f2f1f0efeeedecebeae9e8e7e6e5e4e3e2e1e00020102607").unwrap());
2166 }
2167 if hostname {
2168 target_value.append(&mut hex::decode("0504686f73742607").unwrap());
2169 }
2170 if excess_address_data {
2171 target_value.append(&mut hex::decode("216c280b5395a2546e7e4b2663e04f811622f15a4f92e83aa2e92ba2a573c139142c54ae63072a1ec1ee7dc0c04bde5c847806172aa05c92c22ae8e308d1d269").unwrap());
2172 }
2173 if excess_data {
2174 target_value.append(&mut hex::decode("3b12cc195ce0a2d1bda6a88befa19fa07f51caa75ce83837f28965600b8aacab0855ffb0e741ec5f7c41421e9829a9d48611c8c831f71be5ea73e66594977ffd").unwrap());
2175 }
2176 assert_eq!(encoded_value, target_value);
2177 }
2178
2179 #[test]
2180 fn encoding_node_announcement() {
2181 do_encoding_node_announcement(true, true, true, true, true, true, true, true);
2182 do_encoding_node_announcement(false, false, false, false, false, false, false, false);
2183 do_encoding_node_announcement(false, true, false, false, false, false, false, false);
2184 do_encoding_node_announcement(false, false, true, false, false, false, false, false);
2185 do_encoding_node_announcement(false, false, false, true, false, false, false, false);
2186 do_encoding_node_announcement(false, false, false, false, true, false, false, false);
2187 do_encoding_node_announcement(false, false, false, false, false, true, false, false);
2188 do_encoding_node_announcement(false, false, false, false, false, false, true, false);
2189 do_encoding_node_announcement(false, true, false, true, false, false, true, false);
2190 do_encoding_node_announcement(false, false, true, false, true, false, false, false);
2191 }
2192
2193 fn do_encoding_channel_update(direction: bool, disable: bool, excess_data: bool) {
2194 let secp_ctx = Secp256k1::new();
2195 let (privkey_1, _) = get_keys_from!("0101010101010101010101010101010101010101010101010101010101010101", secp_ctx);
2196 let sig_1 = get_sig_on!(privkey_1, secp_ctx, String::from("01010101010101010101010101010101"));
2197 let unsigned_channel_update = msgs::UnsignedChannelUpdate {
2198 chain_hash: BlockHash::from_hex("6fe28c0ab6f1b372c1a6a246ae63f74f931e8365e15a089c68d6190000000000").unwrap(),
2199 short_channel_id: 2316138423780173,
2200 timestamp: 20190119,
2201 flags: if direction { 1 } else { 0 } | if disable { 1 << 1 } else { 0 },
2202 cltv_expiry_delta: 144,
2203 htlc_minimum_msat: 1000000,
2204 htlc_maximum_msat: 131355275467161,
2205 fee_base_msat: 10000,
2206 fee_proportional_millionths: 20,
2207 excess_data: if excess_data { vec![0, 0, 0, 0, 59, 154, 202, 0] } else { Vec::new() }
2208 };
2209 let channel_update = msgs::ChannelUpdate {
2210 signature: sig_1,
2211 contents: unsigned_channel_update
2212 };
2213 let encoded_value = channel_update.encode();
2214 let mut target_value = hex::decode("d977cb9b53d93a6ff64bb5f1e158b4094b66e798fb12911168a3ccdf80a83096340a6a95da0ae8d9f776528eecdbb747eb6b545495a4319ed5378e35b21e073a").unwrap();
2215 target_value.append(&mut hex::decode("000000000019d6689c085ae165831e934ff763ae46a2a6c172b3f1b60a8ce26f").unwrap());
2216 target_value.append(&mut hex::decode("00083a840000034d013413a7").unwrap());
2217 target_value.append(&mut hex::decode("01").unwrap());
2218 target_value.append(&mut hex::decode("00").unwrap());
2219 if direction {
2220 let flag = target_value.last_mut().unwrap();
2221 *flag = 1;
2222 }
2223 if disable {
2224 let flag = target_value.last_mut().unwrap();
2225 *flag = *flag | 1 << 1;
2226 }
2227 target_value.append(&mut hex::decode("009000000000000f42400000271000000014").unwrap());
2228 target_value.append(&mut hex::decode("0000777788889999").unwrap());
2229 if excess_data {
2230 target_value.append(&mut hex::decode("000000003b9aca00").unwrap());
2231 }
2232 assert_eq!(encoded_value, target_value);
2233 }
2234
2235 #[test]
2236 fn encoding_channel_update() {
2237 do_encoding_channel_update(false, false, false);
2238 do_encoding_channel_update(false, false, true);
2239 do_encoding_channel_update(true, false, false);
2240 do_encoding_channel_update(true, false, true);
2241 do_encoding_channel_update(false, true, false);
2242 do_encoding_channel_update(false, true, true);
2243 do_encoding_channel_update(true, true, false);
2244 do_encoding_channel_update(true, true, true);
2245 }
2246
2247 fn do_encoding_open_channel(random_bit: bool, shutdown: bool, incl_chan_type: bool) {
2248 let secp_ctx = Secp256k1::new();
2249 let (_, pubkey_1) = get_keys_from!("0101010101010101010101010101010101010101010101010101010101010101", secp_ctx);
2250 let (_, pubkey_2) = get_keys_from!("0202020202020202020202020202020202020202020202020202020202020202", secp_ctx);
2251 let (_, pubkey_3) = get_keys_from!("0303030303030303030303030303030303030303030303030303030303030303", secp_ctx);
2252 let (_, pubkey_4) = get_keys_from!("0404040404040404040404040404040404040404040404040404040404040404", secp_ctx);
2253 let (_, pubkey_5) = get_keys_from!("0505050505050505050505050505050505050505050505050505050505050505", secp_ctx);
2254 let (_, pubkey_6) = get_keys_from!("0606060606060606060606060606060606060606060606060606060606060606", secp_ctx);
2255 let open_channel = msgs::OpenChannel {
2256 chain_hash: BlockHash::from_hex("6fe28c0ab6f1b372c1a6a246ae63f74f931e8365e15a089c68d6190000000000").unwrap(),
2257 temporary_channel_id: [2; 32],
2258 funding_satoshis: 1311768467284833366,
2259 push_msat: 2536655962884945560,
2260 dust_limit_satoshis: 3608586615801332854,
2261 max_htlc_value_in_flight_msat: 8517154655701053848,
2262 channel_reserve_satoshis: 8665828695742877976,
2263 htlc_minimum_msat: 2316138423780173,
2264 feerate_per_kw: 821716,
2265 to_self_delay: 49340,
2266 max_accepted_htlcs: 49340,
2267 funding_pubkey: pubkey_1,
2268 revocation_basepoint: pubkey_2,
2269 payment_point: pubkey_3,
2270 delayed_payment_basepoint: pubkey_4,
2271 htlc_basepoint: pubkey_5,
2272 first_per_commitment_point: pubkey_6,
2273 channel_flags: if random_bit { 1 << 5 } else { 0 },
2274 shutdown_scriptpubkey: if shutdown { OptionalField::Present(Address::p2pkh(&::bitcoin::PublicKey{compressed: true, inner: pubkey_1}, Network::Testnet).script_pubkey()) } else { OptionalField::Absent },
2275 channel_type: if incl_chan_type { Some(ChannelTypeFeatures::empty()) } else { None },
2276 };
2277 let encoded_value = open_channel.encode();
2278 let mut target_value = Vec::new();
2279 target_value.append(&mut hex::decode("000000000019d6689c085ae165831e934ff763ae46a2a6c172b3f1b60a8ce26f").unwrap());
2280 target_value.append(&mut hex::decode("02020202020202020202020202020202020202020202020202020202020202021234567890123456233403289122369832144668701144767633030896203198784335490624111800083a840000034d000c89d4c0bcc0bc031b84c5567b126440995d3ed5aaba0565d71e1834604819ff9c17f5e9d5dd078f024d4b6cd1361032ca9bd2aeb9d900aa4d45d9ead80ac9423374c451a7254d076602531fe6068134503d2723133227c867ac8fa6c83c537e9a44c3c5bdbdcb1fe33703462779ad4aad39514614751a71085f2f10e1c7a593e4e030efb5b8721ce55b0b0362c0a046dacce86ddd0343c6d3c7c79c2208ba0d9c9cf24a6d046d21d21f90f703f006a18d5653c4edf5391ff23a61f03ff83d237e880ee61187fa9f379a028e0a").unwrap());
2281 if random_bit {
2282 target_value.append(&mut hex::decode("20").unwrap());
2283 } else {
2284 target_value.append(&mut hex::decode("00").unwrap());
2285 }
2286 if shutdown {
2287 target_value.append(&mut hex::decode("001976a91479b000887626b294a914501a4cd226b58b23598388ac").unwrap());
2288 }
2289 if incl_chan_type {
2290 target_value.append(&mut hex::decode("0100").unwrap());
2291 }
2292 assert_eq!(encoded_value, target_value);
2293 }
2294
2295 #[test]
2296 fn encoding_open_channel() {
2297 do_encoding_open_channel(false, false, false);
2298 do_encoding_open_channel(false, false, true);
2299 do_encoding_open_channel(false, true, false);
2300 do_encoding_open_channel(false, true, true);
2301 do_encoding_open_channel(true, false, false);
2302 do_encoding_open_channel(true, false, true);
2303 do_encoding_open_channel(true, true, false);
2304 do_encoding_open_channel(true, true, true);
2305 }
2306
2307 fn do_encoding_accept_channel(shutdown: bool) {
2308 let secp_ctx = Secp256k1::new();
2309 let (_, pubkey_1) = get_keys_from!("0101010101010101010101010101010101010101010101010101010101010101", secp_ctx);
2310 let (_, pubkey_2) = get_keys_from!("0202020202020202020202020202020202020202020202020202020202020202", secp_ctx);
2311 let (_, pubkey_3) = get_keys_from!("0303030303030303030303030303030303030303030303030303030303030303", secp_ctx);
2312 let (_, pubkey_4) = get_keys_from!("0404040404040404040404040404040404040404040404040404040404040404", secp_ctx);
2313 let (_, pubkey_5) = get_keys_from!("0505050505050505050505050505050505050505050505050505050505050505", secp_ctx);
2314 let (_, pubkey_6) = get_keys_from!("0606060606060606060606060606060606060606060606060606060606060606", secp_ctx);
2315 let accept_channel = msgs::AcceptChannel {
2316 temporary_channel_id: [2; 32],
2317 dust_limit_satoshis: 1311768467284833366,
2318 max_htlc_value_in_flight_msat: 2536655962884945560,
2319 channel_reserve_satoshis: 3608586615801332854,
2320 htlc_minimum_msat: 2316138423780173,
2321 minimum_depth: 821716,
2322 to_self_delay: 49340,
2323 max_accepted_htlcs: 49340,
2324 funding_pubkey: pubkey_1,
2325 revocation_basepoint: pubkey_2,
2326 payment_point: pubkey_3,
2327 delayed_payment_basepoint: pubkey_4,
2328 htlc_basepoint: pubkey_5,
2329 first_per_commitment_point: pubkey_6,
2330 shutdown_scriptpubkey: if shutdown { OptionalField::Present(Address::p2pkh(&::bitcoin::PublicKey{compressed: true, inner: pubkey_1}, Network::Testnet).script_pubkey()) } else { OptionalField::Absent },
2331 channel_type: None,
2332 };
2333 let encoded_value = accept_channel.encode();
2334 let mut target_value = hex::decode("020202020202020202020202020202020202020202020202020202020202020212345678901234562334032891223698321446687011447600083a840000034d000c89d4c0bcc0bc031b84c5567b126440995d3ed5aaba0565d71e1834604819ff9c17f5e9d5dd078f024d4b6cd1361032ca9bd2aeb9d900aa4d45d9ead80ac9423374c451a7254d076602531fe6068134503d2723133227c867ac8fa6c83c537e9a44c3c5bdbdcb1fe33703462779ad4aad39514614751a71085f2f10e1c7a593e4e030efb5b8721ce55b0b0362c0a046dacce86ddd0343c6d3c7c79c2208ba0d9c9cf24a6d046d21d21f90f703f006a18d5653c4edf5391ff23a61f03ff83d237e880ee61187fa9f379a028e0a").unwrap();
2335 if shutdown {
2336 target_value.append(&mut hex::decode("001976a91479b000887626b294a914501a4cd226b58b23598388ac").unwrap());
2337 }
2338 assert_eq!(encoded_value, target_value);
2339 }
2340
2341 #[test]
2342 fn encoding_accept_channel() {
2343 do_encoding_accept_channel(false);
2344 do_encoding_accept_channel(true);
2345 }
2346
2347 #[test]
2348 fn encoding_funding_created() {
2349 let secp_ctx = Secp256k1::new();
2350 let (privkey_1, _) = get_keys_from!("0101010101010101010101010101010101010101010101010101010101010101", secp_ctx);
2351 let sig_1 = get_sig_on!(privkey_1, secp_ctx, String::from("01010101010101010101010101010101"));
2352 let funding_created = msgs::FundingCreated {
2353 temporary_channel_id: [2; 32],
2354 funding_txid: Txid::from_hex("c2d4449afa8d26140898dd54d3390b057ba2a5afcf03ba29d7dc0d8b9ffe966e").unwrap(),
2355 funding_output_index: 255,
2356 signature: sig_1,
2357 };
2358 let encoded_value = funding_created.encode();
2359 let target_value = hex::decode("02020202020202020202020202020202020202020202020202020202020202026e96fe9f8b0ddcd729ba03cfafa5a27b050b39d354dd980814268dfa9a44d4c200ffd977cb9b53d93a6ff64bb5f1e158b4094b66e798fb12911168a3ccdf80a83096340a6a95da0ae8d9f776528eecdbb747eb6b545495a4319ed5378e35b21e073a").unwrap();
2360 assert_eq!(encoded_value, target_value);
2361 }
2362
2363 #[test]
2364 fn encoding_funding_signed() {
2365 let secp_ctx = Secp256k1::new();
2366 let (privkey_1, _) = get_keys_from!("0101010101010101010101010101010101010101010101010101010101010101", secp_ctx);
2367 let sig_1 = get_sig_on!(privkey_1, secp_ctx, String::from("01010101010101010101010101010101"));
2368 let funding_signed = msgs::FundingSigned {
2369 channel_id: [2; 32],
2370 signature: sig_1,
2371 };
2372 let encoded_value = funding_signed.encode();
2373 let target_value = hex::decode("0202020202020202020202020202020202020202020202020202020202020202d977cb9b53d93a6ff64bb5f1e158b4094b66e798fb12911168a3ccdf80a83096340a6a95da0ae8d9f776528eecdbb747eb6b545495a4319ed5378e35b21e073a").unwrap();
2374 assert_eq!(encoded_value, target_value);
2375 }
2376
2377 #[test]
2378 fn encoding_channel_ready() {
2379 let secp_ctx = Secp256k1::new();
2380 let (_, pubkey_1,) = get_keys_from!("0101010101010101010101010101010101010101010101010101010101010101", secp_ctx);
2381 let channel_ready = msgs::ChannelReady {
2382 channel_id: [2; 32],
2383 next_per_commitment_point: pubkey_1,
2384 short_channel_id_alias: None,
2385 };
2386 let encoded_value = channel_ready.encode();
2387 let target_value = hex::decode("0202020202020202020202020202020202020202020202020202020202020202031b84c5567b126440995d3ed5aaba0565d71e1834604819ff9c17f5e9d5dd078f").unwrap();
2388 assert_eq!(encoded_value, target_value);
2389 }
2390
2391 fn do_encoding_shutdown(script_type: u8) {
2392 let secp_ctx = Secp256k1::new();
2393 let (_, pubkey_1) = get_keys_from!("0101010101010101010101010101010101010101010101010101010101010101", secp_ctx);
2394 let script = Builder::new().push_opcode(opcodes::OP_TRUE).into_script();
2395 let shutdown = msgs::Shutdown {
2396 channel_id: [2; 32],
2397 scriptpubkey:
2398 if script_type == 1 { Address::p2pkh(&::bitcoin::PublicKey{compressed: true, inner: pubkey_1}, Network::Testnet).script_pubkey() }
2399 else if script_type == 2 { Address::p2sh(&script, Network::Testnet).unwrap().script_pubkey() }
2400 else if script_type == 3 { Address::p2wpkh(&::bitcoin::PublicKey{compressed: true, inner: pubkey_1}, Network::Testnet).unwrap().script_pubkey() }
2401 else { Address::p2wsh(&script, Network::Testnet).script_pubkey() },
2402 };
2403 let encoded_value = shutdown.encode();
2404 let mut target_value = hex::decode("0202020202020202020202020202020202020202020202020202020202020202").unwrap();
2405 if script_type == 1 {
2406 target_value.append(&mut hex::decode("001976a91479b000887626b294a914501a4cd226b58b23598388ac").unwrap());
2407 } else if script_type == 2 {
2408 target_value.append(&mut hex::decode("0017a914da1745e9b549bd0bfa1a569971c77eba30cd5a4b87").unwrap());
2409 } else if script_type == 3 {
2410 target_value.append(&mut hex::decode("0016001479b000887626b294a914501a4cd226b58b235983").unwrap());
2411 } else if script_type == 4 {
2412 target_value.append(&mut hex::decode("002200204ae81572f06e1b88fd5ced7a1a000945432e83e1551e6f721ee9c00b8cc33260").unwrap());
2413 }
2414 assert_eq!(encoded_value, target_value);
2415 }
2416
2417 #[test]
2418 fn encoding_shutdown() {
2419 do_encoding_shutdown(1);
2420 do_encoding_shutdown(2);
2421 do_encoding_shutdown(3);
2422 do_encoding_shutdown(4);
2423 }
2424
2425 #[test]
2426 fn encoding_closing_signed() {
2427 let secp_ctx = Secp256k1::new();
2428 let (privkey_1, _) = get_keys_from!("0101010101010101010101010101010101010101010101010101010101010101", secp_ctx);
2429 let sig_1 = get_sig_on!(privkey_1, secp_ctx, String::from("01010101010101010101010101010101"));
2430 let closing_signed = msgs::ClosingSigned {
2431 channel_id: [2; 32],
2432 fee_satoshis: 2316138423780173,
2433 signature: sig_1,
2434 fee_range: None,
2435 };
2436 let encoded_value = closing_signed.encode();
2437 let target_value = hex::decode("020202020202020202020202020202020202020202020202020202020202020200083a840000034dd977cb9b53d93a6ff64bb5f1e158b4094b66e798fb12911168a3ccdf80a83096340a6a95da0ae8d9f776528eecdbb747eb6b545495a4319ed5378e35b21e073a").unwrap();
2438 assert_eq!(encoded_value, target_value);
2439 assert_eq!(msgs::ClosingSigned::read(&mut Cursor::new(&target_value)).unwrap(), closing_signed);
2440
2441 let closing_signed_with_range = msgs::ClosingSigned {
2442 channel_id: [2; 32],
2443 fee_satoshis: 2316138423780173,
2444 signature: sig_1,
2445 fee_range: Some(msgs::ClosingSignedFeeRange {
2446 min_fee_satoshis: 0xdeadbeef,
2447 max_fee_satoshis: 0x1badcafe01234567,
2448 }),
2449 };
2450 let encoded_value_with_range = closing_signed_with_range.encode();
2451 let target_value_with_range = hex::decode("020202020202020202020202020202020202020202020202020202020202020200083a840000034dd977cb9b53d93a6ff64bb5f1e158b4094b66e798fb12911168a3ccdf80a83096340a6a95da0ae8d9f776528eecdbb747eb6b545495a4319ed5378e35b21e073a011000000000deadbeef1badcafe01234567").unwrap();
2452 assert_eq!(encoded_value_with_range, target_value_with_range);
2453 assert_eq!(msgs::ClosingSigned::read(&mut Cursor::new(&target_value_with_range)).unwrap(),
2454 closing_signed_with_range);
2455 }
2456
2457 #[test]
2458 fn encoding_update_add_htlc() {
2459 let secp_ctx = Secp256k1::new();
2460 let (_, pubkey_1) = get_keys_from!("0101010101010101010101010101010101010101010101010101010101010101", secp_ctx);
2461 let onion_routing_packet = msgs::OnionPacket {
2462 version: 255,
2463 public_key: Ok(pubkey_1),
2464 hop_data: [1; 20*65],
2465 hmac: [2; 32]
2466 };
2467 let update_add_htlc = msgs::UpdateAddHTLC {
2468 channel_id: [2; 32],
2469 htlc_id: 2316138423780173,
2470 amount_msat: 3608586615801332854,
2471 payment_hash: PaymentHash([1; 32]),
2472 cltv_expiry: 821716,
2473 onion_routing_packet
2474 };
2475 let encoded_value = update_add_htlc.encode();
2476 let target_value = hex::decode("020202020202020202020202020202020202020202020202020202020202020200083a840000034d32144668701144760101010101010101010101010101010101010101010101010101010101010101000c89d4ff031b84c5567b126440995d3ed5aaba0565d71e1834604819ff9c17f5e9d5dd078f010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010101010202020202020202020202020202020202020202020202020202020202020202").unwrap();
2477 assert_eq!(encoded_value, target_value);
2478 }
2479
2480 #[test]
2481 fn encoding_update_fulfill_htlc() {
2482 let update_fulfill_htlc = msgs::UpdateFulfillHTLC {
2483 channel_id: [2; 32],
2484 htlc_id: 2316138423780173,
2485 payment_preimage: PaymentPreimage([1; 32]),
2486 };
2487 let encoded_value = update_fulfill_htlc.encode();
2488 let target_value = hex::decode("020202020202020202020202020202020202020202020202020202020202020200083a840000034d0101010101010101010101010101010101010101010101010101010101010101").unwrap();
2489 assert_eq!(encoded_value, target_value);
2490 }
2491
2492 #[test]
2493 fn encoding_update_fail_htlc() {
2494 let reason = OnionErrorPacket {
2495 data: [1; 32].to_vec(),
2496 };
2497 let update_fail_htlc = msgs::UpdateFailHTLC {
2498 channel_id: [2; 32],
2499 htlc_id: 2316138423780173,
2500 reason
2501 };
2502 let encoded_value = update_fail_htlc.encode();
2503 let target_value = hex::decode("020202020202020202020202020202020202020202020202020202020202020200083a840000034d00200101010101010101010101010101010101010101010101010101010101010101").unwrap();
2504 assert_eq!(encoded_value, target_value);
2505 }
2506
2507 #[test]
2508 fn encoding_update_fail_malformed_htlc() {
2509 let update_fail_malformed_htlc = msgs::UpdateFailMalformedHTLC {
2510 channel_id: [2; 32],
2511 htlc_id: 2316138423780173,
2512 sha256_of_onion: [1; 32],
2513 failure_code: 255
2514 };
2515 let encoded_value = update_fail_malformed_htlc.encode();
2516 let target_value = hex::decode("020202020202020202020202020202020202020202020202020202020202020200083a840000034d010101010101010101010101010101010101010101010101010101010101010100ff").unwrap();
2517 assert_eq!(encoded_value, target_value);
2518 }
2519
2520 fn do_encoding_commitment_signed(htlcs: bool) {
2521 let secp_ctx = Secp256k1::new();
2522 let (privkey_1, _) = get_keys_from!("0101010101010101010101010101010101010101010101010101010101010101", secp_ctx);
2523 let (privkey_2, _) = get_keys_from!("0202020202020202020202020202020202020202020202020202020202020202", secp_ctx);
2524 let (privkey_3, _) = get_keys_from!("0303030303030303030303030303030303030303030303030303030303030303", secp_ctx);
2525 let (privkey_4, _) = get_keys_from!("0404040404040404040404040404040404040404040404040404040404040404", secp_ctx);
2526 let sig_1 = get_sig_on!(privkey_1, secp_ctx, String::from("01010101010101010101010101010101"));
2527 let sig_2 = get_sig_on!(privkey_2, secp_ctx, String::from("01010101010101010101010101010101"));
2528 let sig_3 = get_sig_on!(privkey_3, secp_ctx, String::from("01010101010101010101010101010101"));
2529 let sig_4 = get_sig_on!(privkey_4, secp_ctx, String::from("01010101010101010101010101010101"));
2530 let commitment_signed = msgs::CommitmentSigned {
2531 channel_id: [2; 32],
2532 signature: sig_1,
2533 htlc_signatures: if htlcs { vec![sig_2, sig_3, sig_4] } else { Vec::new() },
2534 };
2535 let encoded_value = commitment_signed.encode();
2536 let mut target_value = hex::decode("0202020202020202020202020202020202020202020202020202020202020202d977cb9b53d93a6ff64bb5f1e158b4094b66e798fb12911168a3ccdf80a83096340a6a95da0ae8d9f776528eecdbb747eb6b545495a4319ed5378e35b21e073a").unwrap();
2537 if htlcs {
2538 target_value.append(&mut hex::decode("00031735b6a427e80d5fe7cd90a2f4ee08dc9c27cda7c35a4172e5d85b12c49d4232537e98f9b1f3c5e6989a8b9644e90e8918127680dbd0d4043510840fc0f1e11a216c280b5395a2546e7e4b2663e04f811622f15a4f91e83aa2e92ba2a573c139142c54ae63072a1ec1ee7dc0c04bde5c847806172aa05c92c22ae8e308d1d2692b12cc195ce0a2d1bda6a88befa19fa07f51caa75ce83837f28965600b8aacab0855ffb0e741ec5f7c41421e9829a9d48611c8c831f71be5ea73e66594977ffd").unwrap());
2539 } else {
2540 target_value.append(&mut hex::decode("0000").unwrap());
2541 }
2542 assert_eq!(encoded_value, target_value);
2543 }
2544
2545 #[test]
2546 fn encoding_commitment_signed() {
2547 do_encoding_commitment_signed(true);
2548 do_encoding_commitment_signed(false);
2549 }
2550
2551 #[test]
2552 fn encoding_revoke_and_ack() {
2553 let secp_ctx = Secp256k1::new();
2554 let (_, pubkey_1) = get_keys_from!("0101010101010101010101010101010101010101010101010101010101010101", secp_ctx);
2555 let raa = msgs::RevokeAndACK {
2556 channel_id: [2; 32],
2557 per_commitment_secret: [1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1],
2558 next_per_commitment_point: pubkey_1,
2559 };
2560 let encoded_value = raa.encode();
2561 let target_value = hex::decode("02020202020202020202020202020202020202020202020202020202020202020101010101010101010101010101010101010101010101010101010101010101031b84c5567b126440995d3ed5aaba0565d71e1834604819ff9c17f5e9d5dd078f").unwrap();
2562 assert_eq!(encoded_value, target_value);
2563 }
2564
2565 #[test]
2566 fn encoding_update_fee() {
2567 let update_fee = msgs::UpdateFee {
2568 channel_id: [2; 32],
2569 feerate_per_kw: 20190119,
2570 };
2571 let encoded_value = update_fee.encode();
2572 let target_value = hex::decode("0202020202020202020202020202020202020202020202020202020202020202013413a7").unwrap();
2573 assert_eq!(encoded_value, target_value);
2574 }
2575
2576 #[test]
2577 fn encoding_init() {
2578 assert_eq!(msgs::Init {
2579 features: InitFeatures::from_le_bytes(vec![0xFF, 0xFF, 0xFF]),
2580 remote_network_address: None,
2581 }.encode(), hex::decode("00023fff0003ffffff").unwrap());
2582 assert_eq!(msgs::Init {
2583 features: InitFeatures::from_le_bytes(vec![0xFF]),
2584 remote_network_address: None,
2585 }.encode(), hex::decode("0001ff0001ff").unwrap());
2586 assert_eq!(msgs::Init {
2587 features: InitFeatures::from_le_bytes(vec![]),
2588 remote_network_address: None,
2589 }.encode(), hex::decode("00000000").unwrap());
2590
2591 let init_msg = msgs::Init { features: InitFeatures::from_le_bytes(vec![]),
2592 remote_network_address: Some(msgs::NetAddress::IPv4 {
2593 addr: [127, 0, 0, 1],
2594 port: 1000,
2595 }),
2596 };
2597 let encoded_value = init_msg.encode();
2598 let target_value = hex::decode("000000000307017f00000103e8").unwrap();
2599 assert_eq!(encoded_value, target_value);
2600 assert_eq!(msgs::Init::read(&mut Cursor::new(&target_value)).unwrap(), init_msg);
2601 }
2602
2603 #[test]
2604 fn encoding_error() {
2605 let error = msgs::ErrorMessage {
2606 channel_id: [2; 32],
2607 data: String::from("rust-lightning"),
2608 };
2609 let encoded_value = error.encode();
2610 let target_value = hex::decode("0202020202020202020202020202020202020202020202020202020202020202000e727573742d6c696768746e696e67").unwrap();
2611 assert_eq!(encoded_value, target_value);
2612 }
2613
2614 #[test]
2615 fn encoding_warning() {
2616 let error = msgs::WarningMessage {
2617 channel_id: [2; 32],
2618 data: String::from("rust-lightning"),
2619 };
2620 let encoded_value = error.encode();
2621 let target_value = hex::decode("0202020202020202020202020202020202020202020202020202020202020202000e727573742d6c696768746e696e67").unwrap();
2622 assert_eq!(encoded_value, target_value);
2623 }
2624
2625 #[test]
2626 fn encoding_ping() {
2627 let ping = msgs::Ping {
2628 ponglen: 64,
2629 byteslen: 64
2630 };
2631 let encoded_value = ping.encode();
2632 let target_value = hex::decode("0040004000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000").unwrap();
2633 assert_eq!(encoded_value, target_value);
2634 }
2635
2636 #[test]
2637 fn encoding_pong() {
2638 let pong = msgs::Pong {
2639 byteslen: 64
2640 };
2641 let encoded_value = pong.encode();
2642 let target_value = hex::decode("004000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000").unwrap();
2643 assert_eq!(encoded_value, target_value);
2644 }
2645
2646 #[test]
2647 fn encoding_nonfinal_onion_hop_data() {
2648 let mut msg = msgs::OnionHopData {
2649 format: OnionHopDataFormat::NonFinalNode {
2650 short_channel_id: 0xdeadbeef1bad1dea,
2651 },
2652 amt_to_forward: 0x0badf00d01020304,
2653 outgoing_cltv_value: 0xffffffff,
2654 };
2655 let encoded_value = msg.encode();
2656 let target_value = hex::decode("1a02080badf00d010203040404ffffffff0608deadbeef1bad1dea").unwrap();
2657 assert_eq!(encoded_value, target_value);
2658 msg = Readable::read(&mut Cursor::new(&target_value[..])).unwrap();
2659 if let OnionHopDataFormat::NonFinalNode { short_channel_id } = msg.format {
2660 assert_eq!(short_channel_id, 0xdeadbeef1bad1dea);
2661 } else { panic!(); }
2662 assert_eq!(msg.amt_to_forward, 0x0badf00d01020304);
2663 assert_eq!(msg.outgoing_cltv_value, 0xffffffff);
2664 }
2665
2666 #[test]
2667 fn encoding_final_onion_hop_data() {
2668 let mut msg = msgs::OnionHopData {
2669 format: OnionHopDataFormat::FinalNode {
2670 payment_data: None,
2671 keysend_preimage: None,
2672 },
2673 amt_to_forward: 0x0badf00d01020304,
2674 outgoing_cltv_value: 0xffffffff,
2675 };
2676 let encoded_value = msg.encode();
2677 let target_value = hex::decode("1002080badf00d010203040404ffffffff").unwrap();
2678 assert_eq!(encoded_value, target_value);
2679 msg = Readable::read(&mut Cursor::new(&target_value[..])).unwrap();
2680 if let OnionHopDataFormat::FinalNode { payment_data: None, .. } = msg.format { } else { panic!(); }
2681 assert_eq!(msg.amt_to_forward, 0x0badf00d01020304);
2682 assert_eq!(msg.outgoing_cltv_value, 0xffffffff);
2683 }
2684
2685 #[test]
2686 fn encoding_final_onion_hop_data_with_secret() {
2687 let expected_payment_secret = PaymentSecret([0x42u8; 32]);
2688 let mut msg = msgs::OnionHopData {
2689 format: OnionHopDataFormat::FinalNode {
2690 payment_data: Some(FinalOnionHopData {
2691 payment_secret: expected_payment_secret,
2692 total_msat: 0x1badca1f
2693 }),
2694 keysend_preimage: None,
2695 },
2696 amt_to_forward: 0x0badf00d01020304,
2697 outgoing_cltv_value: 0xffffffff,
2698 };
2699 let encoded_value = msg.encode();
2700 let target_value = hex::decode("3602080badf00d010203040404ffffffff082442424242424242424242424242424242424242424242424242424242424242421badca1f").unwrap();
2701 assert_eq!(encoded_value, target_value);
2702 msg = Readable::read(&mut Cursor::new(&target_value[..])).unwrap();
2703 if let OnionHopDataFormat::FinalNode {
2704 payment_data: Some(FinalOnionHopData {
2705 payment_secret,
2706 total_msat: 0x1badca1f
2707 }),
2708 keysend_preimage: None,
2709 } = msg.format {
2710 assert_eq!(payment_secret, expected_payment_secret);
2711 } else { panic!(); }
2712 assert_eq!(msg.amt_to_forward, 0x0badf00d01020304);
2713 assert_eq!(msg.outgoing_cltv_value, 0xffffffff);
2714 }
2715
2716 #[test]
2717 fn query_channel_range_end_blocknum() {
2718 let tests: Vec<(u32, u32, u32)> = vec![
2719 (10000, 1500, 11500),
2720 (0, 0xffffffff, 0xffffffff),
2721 (1, 0xffffffff, 0xffffffff),
2722 ];
2723
2724 for (first_blocknum, number_of_blocks, expected) in tests.into_iter() {
2725 let sut = msgs::QueryChannelRange {
2726 chain_hash: BlockHash::from_hex("06226e46111a0b59caaf126043eb5bbf28c34f3a5e332a1fc7b2b73cf188910f").unwrap(),
2727 first_blocknum,
2728 number_of_blocks,
2729 };
2730 assert_eq!(sut.end_blocknum(), expected);
2731 }
2732 }
2733
2734 #[test]
2735 fn encoding_query_channel_range() {
2736 let mut query_channel_range = msgs::QueryChannelRange {
2737 chain_hash: BlockHash::from_hex("06226e46111a0b59caaf126043eb5bbf28c34f3a5e332a1fc7b2b73cf188910f").unwrap(),
2738 first_blocknum: 100000,
2739 number_of_blocks: 1500,
2740 };
2741 let encoded_value = query_channel_range.encode();
2742 let target_value = hex::decode("0f9188f13cb7b2c71f2a335e3a4fc328bf5beb436012afca590b1a11466e2206000186a0000005dc").unwrap();
2743 assert_eq!(encoded_value, target_value);
2744
2745 query_channel_range = Readable::read(&mut Cursor::new(&target_value[..])).unwrap();
2746 assert_eq!(query_channel_range.first_blocknum, 100000);
2747 assert_eq!(query_channel_range.number_of_blocks, 1500);
2748 }
2749
2750 #[test]
2751 fn encoding_reply_channel_range() {
2752 do_encoding_reply_channel_range(0);
2753 do_encoding_reply_channel_range(1);
2754 }
2755
2756 fn do_encoding_reply_channel_range(encoding_type: u8) {
2757 let mut target_value = hex::decode("0f9188f13cb7b2c71f2a335e3a4fc328bf5beb436012afca590b1a11466e2206000b8a06000005dc01").unwrap();
2758 let expected_chain_hash = BlockHash::from_hex("06226e46111a0b59caaf126043eb5bbf28c34f3a5e332a1fc7b2b73cf188910f").unwrap();
2759 let mut reply_channel_range = msgs::ReplyChannelRange {
2760 chain_hash: expected_chain_hash,
2761 first_blocknum: 756230,
2762 number_of_blocks: 1500,
2763 sync_complete: true,
2764 short_channel_ids: vec![0x000000000000008e, 0x0000000000003c69, 0x000000000045a6c4],
2765 };
2766
2767 if encoding_type == 0 {
2768 target_value.append(&mut hex::decode("001900000000000000008e0000000000003c69000000000045a6c4").unwrap());
2769 let encoded_value = reply_channel_range.encode();
2770 assert_eq!(encoded_value, target_value);
2771
2772 reply_channel_range = Readable::read(&mut Cursor::new(&target_value[..])).unwrap();
2773 assert_eq!(reply_channel_range.chain_hash, expected_chain_hash);
2774 assert_eq!(reply_channel_range.first_blocknum, 756230);
2775 assert_eq!(reply_channel_range.number_of_blocks, 1500);
2776 assert_eq!(reply_channel_range.sync_complete, true);
2777 assert_eq!(reply_channel_range.short_channel_ids[0], 0x000000000000008e);
2778 assert_eq!(reply_channel_range.short_channel_ids[1], 0x0000000000003c69);
2779 assert_eq!(reply_channel_range.short_channel_ids[2], 0x000000000045a6c4);
2780 } else {
2781 target_value.append(&mut hex::decode("001601789c636000833e08659309a65878be010010a9023a").unwrap());
2782 let result: Result<msgs::ReplyChannelRange, msgs::DecodeError> = Readable::read(&mut Cursor::new(&target_value[..]));
2783 assert!(result.is_err(), "Expected decode failure with unsupported zlib encoding");
2784 }
2785 }
2786
2787 #[test]
2788 fn encoding_query_short_channel_ids() {
2789 do_encoding_query_short_channel_ids(0);
2790 do_encoding_query_short_channel_ids(1);
2791 }
2792
2793 fn do_encoding_query_short_channel_ids(encoding_type: u8) {
2794 let mut target_value = hex::decode("0f9188f13cb7b2c71f2a335e3a4fc328bf5beb436012afca590b1a11466e2206").unwrap();
2795 let expected_chain_hash = BlockHash::from_hex("06226e46111a0b59caaf126043eb5bbf28c34f3a5e332a1fc7b2b73cf188910f").unwrap();
2796 let mut query_short_channel_ids = msgs::QueryShortChannelIds {
2797 chain_hash: expected_chain_hash,
2798 short_channel_ids: vec![0x0000000000008e, 0x0000000000003c69, 0x000000000045a6c4],
2799 };
2800
2801 if encoding_type == 0 {
2802 target_value.append(&mut hex::decode("001900000000000000008e0000000000003c69000000000045a6c4").unwrap());
2803 let encoded_value = query_short_channel_ids.encode();
2804 assert_eq!(encoded_value, target_value);
2805
2806 query_short_channel_ids = Readable::read(&mut Cursor::new(&target_value[..])).unwrap();
2807 assert_eq!(query_short_channel_ids.chain_hash, expected_chain_hash);
2808 assert_eq!(query_short_channel_ids.short_channel_ids[0], 0x000000000000008e);
2809 assert_eq!(query_short_channel_ids.short_channel_ids[1], 0x0000000000003c69);
2810 assert_eq!(query_short_channel_ids.short_channel_ids[2], 0x000000000045a6c4);
2811 } else {
2812 target_value.append(&mut hex::decode("001601789c636000833e08659309a65878be010010a9023a").unwrap());
2813 let result: Result<msgs::QueryShortChannelIds, msgs::DecodeError> = Readable::read(&mut Cursor::new(&target_value[..]));
2814 assert!(result.is_err(), "Expected decode failure with unsupported zlib encoding");
2815 }
2816 }
2817
2818 #[test]
2819 fn encoding_reply_short_channel_ids_end() {
2820 let expected_chain_hash = BlockHash::from_hex("06226e46111a0b59caaf126043eb5bbf28c34f3a5e332a1fc7b2b73cf188910f").unwrap();
2821 let mut reply_short_channel_ids_end = msgs::ReplyShortChannelIdsEnd {
2822 chain_hash: expected_chain_hash,
2823 full_information: true,
2824 };
2825 let encoded_value = reply_short_channel_ids_end.encode();
2826 let target_value = hex::decode("0f9188f13cb7b2c71f2a335e3a4fc328bf5beb436012afca590b1a11466e220601").unwrap();
2827 assert_eq!(encoded_value, target_value);
2828
2829 reply_short_channel_ids_end = Readable::read(&mut Cursor::new(&target_value[..])).unwrap();
2830 assert_eq!(reply_short_channel_ids_end.chain_hash, expected_chain_hash);
2831 assert_eq!(reply_short_channel_ids_end.full_information, true);
2832 }
2833
2834 #[test]
2835 fn encoding_gossip_timestamp_filter(){
2836 let expected_chain_hash = BlockHash::from_hex("06226e46111a0b59caaf126043eb5bbf28c34f3a5e332a1fc7b2b73cf188910f").unwrap();
2837 let mut gossip_timestamp_filter = msgs::GossipTimestampFilter {
2838 chain_hash: expected_chain_hash,
2839 first_timestamp: 1590000000,
2840 timestamp_range: 0xffff_ffff,
2841 };
2842 let encoded_value = gossip_timestamp_filter.encode();
2843 let target_value = hex::decode("0f9188f13cb7b2c71f2a335e3a4fc328bf5beb436012afca590b1a11466e22065ec57980ffffffff").unwrap();
2844 assert_eq!(encoded_value, target_value);
2845
2846 gossip_timestamp_filter = Readable::read(&mut Cursor::new(&target_value[..])).unwrap();
2847 assert_eq!(gossip_timestamp_filter.chain_hash, expected_chain_hash);
2848 assert_eq!(gossip_timestamp_filter.first_timestamp, 1590000000);
2849 assert_eq!(gossip_timestamp_filter.timestamp_range, 0xffff_ffff);
2850 }
2851
2852 #[test]
2853 fn decode_onion_hop_data_len_as_bigsize() {
2854 let big_payload = encode_big_payload().unwrap();
2862 let mut rd = Cursor::new(&big_payload[..]);
2863 <msgs::OnionHopData as Readable>::read(&mut rd).unwrap();
2864 }
2865 fn encode_big_payload() -> Result<Vec<u8>, io::Error> {
2867 use crate::util::ser::HighZeroBytesDroppedBigSize;
2868 let payload = msgs::OnionHopData {
2869 format: OnionHopDataFormat::NonFinalNode {
2870 short_channel_id: 0xdeadbeef1bad1dea,
2871 },
2872 amt_to_forward: 1000,
2873 outgoing_cltv_value: 0xffffffff,
2874 };
2875 let mut encoded_payload = Vec::new();
2876 let test_bytes = vec![42u8; 1000];
2877 if let OnionHopDataFormat::NonFinalNode { short_channel_id } = payload.format {
2878 encode_varint_length_prefixed_tlv!(&mut encoded_payload, {
2879 (1, test_bytes, vec_type),
2880 (2, HighZeroBytesDroppedBigSize(payload.amt_to_forward), required),
2881 (4, HighZeroBytesDroppedBigSize(payload.outgoing_cltv_value), required),
2882 (6, short_channel_id, required)
2883 });
2884 }
2885 Ok(encoded_payload)
2886 }
2887}