_ Git - rust-lightning/blob - lightning/src/ln/wire.rs

   1 //! Wire encoding/decoding for Lightning messages according to [BOLT #1].
   2 //!
   3 //! Messages known by this module can be read from the wire using [`read`].
   4 //! The [`Message`] enum returned by [`read`] wraps the decoded message or the message type (if
   5 //! unknown) to use with pattern matching.
   6 //!
   7 //! Messages implementing the [`Encode`] trait define a message type and can be sent over the wire
   8 //! using [`write`].
   9 //!
  10 //! [BOLT #1]: https://github.com/lightningnetwork/lightning-rfc/blob/master/01-messaging.md
  11 //! [`read`]: fn.read.html
  12 //! [`write`]: fn.write.html
  13 //! [`Encode`]: trait.Encode.html
  14 //! [`Message`]: enum.Message.html
  15
  16 use ln::msgs;
  17 use util::ser::{Readable, Writeable, Writer};
  18
  19 /// A Lightning message returned by [`read`] when decoding bytes received over the wire. Each
  20 /// variant contains a message from [`ln::msgs`] or otherwise the message type if unknown.
  21 ///
  22 /// [`read`]: fn.read.html
  23 /// [`ln::msgs`]: ../msgs/index.html
  24 #[allow(missing_docs)]
  25 pub enum Message {
  26         Init(msgs::Init),
  27         Error(msgs::ErrorMessage),
  28         Ping(msgs::Ping),
  29         Pong(msgs::Pong),
  30         OpenChannel(msgs::OpenChannel),
  31         AcceptChannel(msgs::AcceptChannel),
  32         FundingCreated(msgs::FundingCreated),
  33         FundingSigned(msgs::FundingSigned),
  34         FundingLocked(msgs::FundingLocked),
  35         Shutdown(msgs::Shutdown),
  36         ClosingSigned(msgs::ClosingSigned),
  37         UpdateAddHTLC(msgs::UpdateAddHTLC),
  38         UpdateFulfillHTLC(msgs::UpdateFulfillHTLC),
  39         UpdateFailHTLC(msgs::UpdateFailHTLC),
  40         UpdateFailMalformedHTLC(msgs::UpdateFailMalformedHTLC),
  41         CommitmentSigned(msgs::CommitmentSigned),
  42         RevokeAndACK(msgs::RevokeAndACK),
  43         UpdateFee(msgs::UpdateFee),
  44         ChannelReestablish(msgs::ChannelReestablish),
  45         AnnouncementSignatures(msgs::AnnouncementSignatures),
  46         ChannelAnnouncement(msgs::ChannelAnnouncement),
  47         NodeAnnouncement(msgs::NodeAnnouncement),
  48         ChannelUpdate(msgs::ChannelUpdate),
  49         /// A message that could not be decoded because its type is unknown.
  50         Unknown(MessageType),
  51 }
  52
  53 /// A number identifying a message to determine how it is encoded on the wire.
  54 #[derive(Clone, Copy)]
  55 pub struct MessageType(u16);
  56
  57 impl Message {
  58         /// Returns the type that was used to decode the message payload.
  59         pub fn type_id(&self) -> MessageType {
  60                 match self {
  61                         &Message::Init(ref msg) => msg.type_id(),
  62                         &Message::Error(ref msg) => msg.type_id(),
  63                         &Message::Ping(ref msg) => msg.type_id(),
  64                         &Message::Pong(ref msg) => msg.type_id(),
  65                         &Message::OpenChannel(ref msg) => msg.type_id(),
  66                         &Message::AcceptChannel(ref msg) => msg.type_id(),
  67                         &Message::FundingCreated(ref msg) => msg.type_id(),
  68                         &Message::FundingSigned(ref msg) => msg.type_id(),
  69                         &Message::FundingLocked(ref msg) => msg.type_id(),
  70                         &Message::Shutdown(ref msg) => msg.type_id(),
  71                         &Message::ClosingSigned(ref msg) => msg.type_id(),
  72                         &Message::UpdateAddHTLC(ref msg) => msg.type_id(),
  73                         &Message::UpdateFulfillHTLC(ref msg) => msg.type_id(),
  74                         &Message::UpdateFailHTLC(ref msg) => msg.type_id(),
  75                         &Message::UpdateFailMalformedHTLC(ref msg) => msg.type_id(),
  76                         &Message::CommitmentSigned(ref msg) => msg.type_id(),
  77                         &Message::RevokeAndACK(ref msg) => msg.type_id(),
  78                         &Message::UpdateFee(ref msg) => msg.type_id(),
  79                         &Message::ChannelReestablish(ref msg) => msg.type_id(),
  80                         &Message::AnnouncementSignatures(ref msg) => msg.type_id(),
  81                         &Message::ChannelAnnouncement(ref msg) => msg.type_id(),
  82                         &Message::NodeAnnouncement(ref msg) => msg.type_id(),
  83                         &Message::ChannelUpdate(ref msg) => msg.type_id(),
  84                         &Message::Unknown(type_id) => type_id,
  85                 }
  86         }
  87 }
  88
  89 impl MessageType {
  90         /// Returns whether the message type is even, indicating both endpoints must support it.
  91         pub fn is_even(&self) -> bool {
  92                 (self.0 & 1) == 0
  93         }
  94 }
  95
  96 impl ::std::fmt::Display for MessageType {
  97         fn fmt(&self, f: &mut ::std::fmt::Formatter) -> ::std::fmt::Result {
  98                 write!(f, "{}", self.0)
  99         }
 100 }
 101
 102 /// Reads a message from the data buffer consisting of a 2-byte big-endian type and a
 103 /// variable-length payload conforming to the type.
 104 ///
 105 /// # Errors
 106 ///
 107 /// Returns an error if the message payload code not be decoded as the specified type.
 108 pub fn read<R: ::std::io::Read>(buffer: &mut R) -> Result<Message, msgs::DecodeError> {
 109         let message_type = <u16 as Readable>::read(buffer)?;
 110         match message_type {
 111                 msgs::Init::TYPE => {
 112                         Ok(Message::Init(Readable::read(buffer)?))
 113                 },
 114                 msgs::ErrorMessage::TYPE => {
 115                         Ok(Message::Error(Readable::read(buffer)?))
 116                 },
 117                 msgs::Ping::TYPE => {
 118                         Ok(Message::Ping(Readable::read(buffer)?))
 119                 },
 120                 msgs::Pong::TYPE => {
 121                         Ok(Message::Pong(Readable::read(buffer)?))
 122                 },
 123                 msgs::OpenChannel::TYPE => {
 124                         Ok(Message::OpenChannel(Readable::read(buffer)?))
 125                 },
 126                 msgs::AcceptChannel::TYPE => {
 127                         Ok(Message::AcceptChannel(Readable::read(buffer)?))
 128                 },
 129                 msgs::FundingCreated::TYPE => {
 130                         Ok(Message::FundingCreated(Readable::read(buffer)?))
 131                 },
 132                 msgs::FundingSigned::TYPE => {
 133                         Ok(Message::FundingSigned(Readable::read(buffer)?))
 134                 },
 135                 msgs::FundingLocked::TYPE => {
 136                         Ok(Message::FundingLocked(Readable::read(buffer)?))
 137                 },
 138                 msgs::Shutdown::TYPE => {
 139                         Ok(Message::Shutdown(Readable::read(buffer)?))
 140                 },
 141                 msgs::ClosingSigned::TYPE => {
 142                         Ok(Message::ClosingSigned(Readable::read(buffer)?))
 143                 },
 144                 msgs::UpdateAddHTLC::TYPE => {
 145                         Ok(Message::UpdateAddHTLC(Readable::read(buffer)?))
 146                 },
 147                 msgs::UpdateFulfillHTLC::TYPE => {
 148                         Ok(Message::UpdateFulfillHTLC(Readable::read(buffer)?))
 149                 },
 150                 msgs::UpdateFailHTLC::TYPE => {
 151                         Ok(Message::UpdateFailHTLC(Readable::read(buffer)?))
 152                 },
 153                 msgs::UpdateFailMalformedHTLC::TYPE => {
 154                         Ok(Message::UpdateFailMalformedHTLC(Readable::read(buffer)?))
 155                 },
 156                 msgs::CommitmentSigned::TYPE => {
 157                         Ok(Message::CommitmentSigned(Readable::read(buffer)?))
 158                 },
 159                 msgs::RevokeAndACK::TYPE => {
 160                         Ok(Message::RevokeAndACK(Readable::read(buffer)?))
 161                 },
 162                 msgs::UpdateFee::TYPE => {
 163                         Ok(Message::UpdateFee(Readable::read(buffer)?))
 164                 },
 165                 msgs::ChannelReestablish::TYPE => {
 166                         Ok(Message::ChannelReestablish(Readable::read(buffer)?))
 167                 },
 168                 msgs::AnnouncementSignatures::TYPE => {
 169                         Ok(Message::AnnouncementSignatures(Readable::read(buffer)?))
 170                 },
 171                 msgs::ChannelAnnouncement::TYPE => {
 172                         Ok(Message::ChannelAnnouncement(Readable::read(buffer)?))
 173                 },
 174                 msgs::NodeAnnouncement::TYPE => {
 175                         Ok(Message::NodeAnnouncement(Readable::read(buffer)?))
 176                 },
 177                 msgs::ChannelUpdate::TYPE => {
 178                         Ok(Message::ChannelUpdate(Readable::read(buffer)?))
 179                 },
 180                 _ => {
 181                         Ok(Message::Unknown(MessageType(message_type)))
 182                 },
 183         }
 184 }
 185
 186 /// Writes a message to the data buffer encoded as a 2-byte big-endian type and a variable-length
 187 /// payload.
 188 ///
 189 /// # Errors
 190 ///
 191 /// Returns an I/O error if the write could not be completed.
 192 pub fn write<M: Encode + Writeable, W: Writer>(message: &M, buffer: &mut W) -> Result<(), ::std::io::Error> {
 193         M::TYPE.write(buffer)?;
 194         message.write(buffer)
 195 }
 196
 197 /// Defines a type-identified encoding for sending messages over the wire.
 198 ///
 199 /// Messages implementing this trait specify a type and must be [`Writeable`] to use with [`write`].
 200 ///
 201 /// [`Writeable`]: ../../util/ser/trait.Writeable.html
 202 /// [`write`]: fn.write.html
 203 pub trait Encode {
 204         /// The type identifying the message payload.
 205         const TYPE: u16;
 206
 207         /// Returns the type identifying the message payload. Convenience method for accessing
 208         /// [`TYPE`](TYPE).
 209         fn type_id(&self) -> MessageType {
 210                 MessageType(Self::TYPE)
 211         }
 212 }
 213
 214 impl Encode for msgs::Init {
 215         const TYPE: u16 = 16;
 216 }
 217
 218 impl Encode for msgs::ErrorMessage {
 219         const TYPE: u16 = 17;
 220 }
 221
 222 impl Encode for msgs::Ping {
 223         const TYPE: u16 = 18;
 224 }
 225
 226 impl Encode for msgs::Pong {
 227         const TYPE: u16 = 19;
 228 }
 229
 230 impl Encode for msgs::OpenChannel {
 231         const TYPE: u16 = 32;
 232 }
 233
 234 impl Encode for msgs::AcceptChannel {
 235         const TYPE: u16 = 33;
 236 }
 237
 238 impl Encode for msgs::FundingCreated {
 239         const TYPE: u16 = 34;
 240 }
 241
 242 impl Encode for msgs::FundingSigned {
 243         const TYPE: u16 = 35;
 244 }
 245
 246 impl Encode for msgs::FundingLocked {
 247         const TYPE: u16 = 36;
 248 }
 249
 250 impl Encode for msgs::Shutdown {
 251         const TYPE: u16 = 38;
 252 }
 253
 254 impl Encode for msgs::ClosingSigned {
 255         const TYPE: u16 = 39;
 256 }
 257
 258 impl Encode for msgs::UpdateAddHTLC {
 259         const TYPE: u16 = 128;
 260 }
 261
 262 impl Encode for msgs::UpdateFulfillHTLC {
 263         const TYPE: u16 = 130;
 264 }
 265
 266 impl Encode for msgs::UpdateFailHTLC {
 267         const TYPE: u16 = 131;
 268 }
 269
 270 impl Encode for msgs::UpdateFailMalformedHTLC {
 271         const TYPE: u16 = 135;
 272 }
 273
 274 impl Encode for msgs::CommitmentSigned {
 275         const TYPE: u16 = 132;
 276 }
 277
 278 impl Encode for msgs::RevokeAndACK {
 279         const TYPE: u16 = 133;
 280 }
 281
 282 impl Encode for msgs::UpdateFee {
 283         const TYPE: u16 = 134;
 284 }
 285
 286 impl Encode for msgs::ChannelReestablish {
 287         const TYPE: u16 = 136;
 288 }
 289
 290 impl Encode for msgs::AnnouncementSignatures {
 291         const TYPE: u16 = 259;
 292 }
 293
 294 impl Encode for msgs::ChannelAnnouncement {
 295         const TYPE: u16 = 256;
 296 }
 297
 298 impl Encode for msgs::NodeAnnouncement {
 299         const TYPE: u16 = 257;
 300 }
 301
 302 impl Encode for msgs::ChannelUpdate {
 303         const TYPE: u16 = 258;
 304 }
 305
 306 #[cfg(test)]
 307 mod tests {
 308         use super::*;
 309         use util::byte_utils;
 310
 311         // Big-endian wire encoding of Pong message (type = 19, byteslen = 2).
 312         const ENCODED_PONG: [u8; 6] = [0u8, 19u8, 0u8, 2u8, 0u8, 0u8];
 313
 314         #[test]
 315         fn read_empty_buffer() {
 316                 let buffer = [];
 317                 let mut reader = ::std::io::Cursor::new(buffer);
 318                 assert!(read(&mut reader).is_err());
 319         }
 320
 321         #[test]
 322         fn read_incomplete_type() {
 323                 let buffer = &ENCODED_PONG[..1];
 324                 let mut reader = ::std::io::Cursor::new(buffer);
 325                 assert!(read(&mut reader).is_err());
 326         }
 327
 328         #[test]
 329         fn read_empty_payload() {
 330                 let buffer = &ENCODED_PONG[..2];
 331                 let mut reader = ::std::io::Cursor::new(buffer);
 332                 assert!(read(&mut reader).is_err());
 333         }
 334
 335         #[test]
 336         fn read_invalid_message() {
 337                 let buffer = &ENCODED_PONG[..4];
 338                 let mut reader = ::std::io::Cursor::new(buffer);
 339                 assert!(read(&mut reader).is_err());
 340         }
 341
 342         #[test]
 343         fn read_known_message() {
 344                 let buffer = &ENCODED_PONG[..];
 345                 let mut reader = ::std::io::Cursor::new(buffer);
 346                 let message = read(&mut reader).unwrap();
 347                 match message {
 348                         Message::Pong(_) => (),
 349                         _ => panic!("Expected pong message; found message type: {}", message.type_id()),
 350                 }
 351         }
 352
 353         #[test]
 354         fn read_unknown_message() {
 355                 let buffer = &byte_utils::be16_to_array(::std::u16::MAX);
 356                 let mut reader = ::std::io::Cursor::new(buffer);
 357                 let message = read(&mut reader).unwrap();
 358                 match message {
 359                         Message::Unknown(MessageType(::std::u16::MAX)) => (),
 360                         _ => panic!("Expected message type {}; found: {}", ::std::u16::MAX, message.type_id()),
 361                 }
 362         }
 363
 364         #[test]
 365         fn write_message_with_type() {
 366                 let message = msgs::Pong { byteslen: 2u16 };
 367                 let mut buffer = Vec::new();
 368                 assert!(write(&message, &mut buffer).is_ok());
 369
 370                 let type_length = ::std::mem::size_of::<u16>();
 371                 let (type_bytes, payload_bytes) = buffer.split_at(type_length);
 372                 assert_eq!(byte_utils::slice_to_be16(type_bytes), msgs::Pong::TYPE);
 373                 assert_eq!(payload_bytes, &ENCODED_PONG[type_length..]);
 374         }
 375
 376         #[test]
 377         fn read_message_encoded_with_write() {
 378                 let message = msgs::Pong { byteslen: 2u16 };
 379                 let mut buffer = Vec::new();
 380                 assert!(write(&message, &mut buffer).is_ok());
 381
 382                 let mut reader = ::std::io::Cursor::new(buffer);
 383                 let decoded_message = read(&mut reader).unwrap();
 384                 match decoded_message {
 385                         Message::Pong(msgs::Pong { byteslen: 2u16 }) => (),
 386                         Message::Pong(msgs::Pong { byteslen }) => {
 387                                 panic!("Expected byteslen {}; found: {}", message.byteslen, byteslen);
 388                         },
 389                         _ => panic!("Expected pong message; found message type: {}", decoded_message.type_id()),
 390                 }
 391         }
 392
 393         #[test]
 394         fn is_even_message_type() {
 395                 let message = Message::Unknown(MessageType(42));
 396                 assert!(message.type_id().is_even());
 397         }
 398
 399         #[test]
 400         fn is_odd_message_type() {
 401                 let message = Message::Unknown(MessageType(43));
 402                 assert!(!message.type_id().is_even());
 403         }
 404
 405         #[test]
 406         fn read_lnd_init_msg() {
 407                 // Taken from lnd v0.9.0-beta.
 408                 let buffer = vec![0, 16, 0, 2, 34, 0, 0, 3, 2, 162, 161];
 409                 check_init_msg(buffer);
 410         }
 411
 412         #[test]
 413         fn read_clightning_init_msg() {
 414                 // Taken from c-lightning v0.8.0.
 415                 let buffer = vec![0, 16, 0, 2, 34, 0, 0, 3, 2, 170, 162, 1, 32, 6, 34, 110, 70, 17, 26, 11, 89, 202, 175, 18, 96, 67, 235, 91, 191, 40, 195, 79, 58, 94, 51, 42, 31, 199, 178, 183, 60, 241, 136, 145, 15];
 416                 check_init_msg(buffer);
 417         }
 418
 419         fn check_init_msg(buffer: Vec<u8>) {
 420                 let mut reader = ::std::io::Cursor::new(buffer);
 421                 let decoded_msg = read(&mut reader).unwrap();
 422                 match decoded_msg {
 423                         Message::Init(msgs::Init { features }) => {
 424                                 assert!(features.supports_variable_length_onion());
 425                                 assert!(features.supports_upfront_shutdown_script());
 426                                 assert!(features.supports_unknown_bits());
 427                                 assert!(!features.requires_unknown_bits());
 428                                 assert!(!features.initial_routing_sync());
 429                         },
 430                         _ => panic!("Expected init message, found message type: {}", decoded_msg.type_id())
 431                 }
 432         }
 433
 434         #[test]
 435         fn read_lnd_node_announcement() {
 436                 // Taken from lnd v0.9.0-beta.
 437                 let buffer = vec![1, 1, 91, 164, 146, 213, 213, 165, 21, 227, 102, 33, 105, 179, 214, 21, 221, 175, 228, 93, 57, 177, 191, 127, 107, 229, 31, 50, 21, 81, 179, 71, 39, 18, 35, 2, 89, 224, 110, 123, 66, 39, 148, 246, 177, 85, 12, 19, 70, 226, 173, 132, 156, 26, 122, 146, 71, 213, 247, 48, 93, 190, 185, 177, 12, 172, 0, 3, 2, 162, 161, 94, 103, 195, 37, 2, 37, 242, 97, 140, 2, 111, 69, 85, 39, 118, 30, 221, 99, 254, 120, 49, 103, 22, 170, 227, 111, 172, 164, 160, 49, 68, 138, 116, 16, 22, 206, 107, 51, 153, 255, 97, 108, 105, 99, 101, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 7, 1, 172, 21, 0, 2, 38, 7];
 438                 let mut reader = ::std::io::Cursor::new(buffer);
 439                 let decoded_msg = read(&mut reader).unwrap();
 440                 match decoded_msg {
 441                         Message::NodeAnnouncement(msgs::NodeAnnouncement { contents: msgs::UnsignedNodeAnnouncement { features, ..}, ..}) => {
 442                                 assert!(features.supports_variable_length_onion());
 443                                 assert!(features.supports_upfront_shutdown_script());
 444                                 assert!(features.supports_unknown_bits());
 445                                 assert!(!features.requires_unknown_bits());
 446                         },
 447                         _ => panic!("Expected node announcement, found message type: {}", decoded_msg.type_id())
 448                 }
 449         }
 450
 451         #[test]
 452         fn read_lnd_chan_announcement() {
 453                 // Taken from lnd v0.9.0-beta.
 454                 let buffer = vec![1, 0, 82, 238, 153, 33, 128, 87, 215, 2, 28, 241, 140, 250, 98, 255, 56, 5, 79, 240, 214, 231, 172, 35, 240, 171, 44, 9, 78, 91, 8, 193, 102, 5, 17, 178, 142, 106, 180, 183, 46, 38, 217, 212, 25, 236, 69, 47, 92, 217, 181, 221, 161, 205, 121, 201, 99, 38, 158, 216, 186, 193, 230, 86, 222, 6, 206, 67, 22, 255, 137, 212, 141, 161, 62, 134, 76, 48, 241, 54, 50, 167, 187, 247, 73, 27, 74, 1, 129, 185, 197, 153, 38, 90, 255, 138, 39, 161, 102, 172, 213, 74, 107, 88, 150, 90, 0, 49, 104, 7, 182, 184, 194, 219, 181, 172, 8, 245, 65, 226, 19, 228, 101, 145, 25, 159, 52, 31, 58, 93, 53, 59, 218, 91, 37, 84, 103, 17, 74, 133, 33, 35, 2, 203, 101, 73, 19, 94, 175, 122, 46, 224, 47, 168, 128, 128, 25, 26, 25, 214, 52, 247, 43, 241, 117, 52, 206, 94, 135, 156, 52, 164, 143, 234, 58, 185, 50, 185, 140, 198, 174, 71, 65, 18, 105, 70, 131, 172, 137, 0, 164, 51, 215, 143, 117, 119, 217, 241, 197, 177, 227, 227, 170, 199, 114, 7, 218, 12, 107, 30, 191, 236, 203, 21, 61, 242, 48, 192, 90, 233, 200, 199, 111, 162, 68, 234, 54, 219, 1, 233, 66, 5, 82, 74, 84, 211, 95, 199, 245, 202, 89, 223, 102, 124, 62, 166, 253, 253, 90, 180, 118, 21, 61, 110, 37, 5, 96, 167, 0, 0, 6, 34, 110, 70, 17, 26, 11, 89, 202, 175, 18, 96, 67, 235, 91, 191, 40, 195, 79, 58, 94, 51, 42, 31, 199, 178, 183, 60, 241, 136, 145, 15, 0, 2, 65, 0, 0, 1, 0, 0, 2, 37, 242, 97, 140, 2, 111, 69, 85, 39, 118, 30, 221, 99, 254, 120, 49, 103, 22, 170, 227, 111, 172, 164, 160, 49, 68, 138, 116, 16, 22, 206, 107, 3, 54, 61, 144, 88, 171, 247, 136, 208, 99, 9, 135, 37, 201, 178, 253, 136, 0, 185, 235, 68, 160, 106, 110, 12, 46, 21, 125, 204, 18, 75, 234, 16, 3, 42, 171, 28, 52, 224, 11, 30, 30, 253, 156, 148, 175, 203, 121, 250, 111, 122, 195, 84, 122, 77, 183, 56, 135, 101, 88, 41, 60, 191, 99, 232, 85, 2, 36, 17, 156, 11, 8, 12, 189, 177, 68, 88, 28, 15, 207, 21, 179, 151, 56, 226, 158, 148, 3, 120, 113, 177, 243, 184, 17, 173, 37, 46, 222, 16];
 455                 let mut reader = ::std::io::Cursor::new(buffer);
 456                 let decoded_msg = read(&mut reader).unwrap();
 457                 match decoded_msg {
 458                         Message::ChannelAnnouncement(msgs::ChannelAnnouncement { contents: msgs::UnsignedChannelAnnouncement { features, ..}, ..}) => {
 459                                 assert!(!features.requires_unknown_bits());
 460                         },
 461                         _ => panic!("Expected node announcement, found message type: {}", decoded_msg.type_id())
 462                 }
 463         }
 464 }