Add node id to custom message callback

Author: Tibo-lg

lightning/src/ln/peer_handler.rs

@@ -47,7 +47,7 @@ use bitcoin::hashes::{HashEngine, Hash};
 pub trait CustomMessageHandler: wire::CustomMessageReader {
 	/// Called with the message type that was received and the buffer to be read.
 	/// Can return a `MessageHandlingError` if the message could not be handled.
-	fn handle_custom_message(&self, msg: Self::CustomMessage) -> Result<(), LightningError>;
+	fn handle_custom_message(&self, msg: Self::CustomMessage, org: &PublicKey) -> Result<(), LightningError>;
 
 	/// Gets the list of pending messages which were generated by the custom message
 	/// handler, clearing the list in the process. The first tuple element must
@@ -102,7 +102,7 @@ impl wire::CustomMessageReader for IgnoringMessageHandler {
 }
 
 impl CustomMessageHandler for IgnoringMessageHandler {
-	fn handle_custom_message(&self, _msg: Self::CustomMessage) -> Result<(), LightningError> {
+	fn handle_custom_message(&self, _msg: Self::CustomMessage, _org: &PublicKey) -> Result<(), LightningError> {
 		// Since we always return `None` in the read the handle method should never be called.
 		unreachable!();
 	}
@@ -1087,7 +1087,7 @@ impl<Descriptor: SocketDescriptor, CM: Deref, RM: Deref, L: Deref, CMH: Deref> P
 				log_trace!(self.logger, "Received unknown odd message of type {}, ignoring", type_id);
 			},
 			wire::Message::Custom(custom) => {
-				self.custom_message_handler.handle_custom_message(custom)?;
+				self.custom_message_handler.handle_custom_message(custom, &peer.their_node_id.unwrap())?;
 			},
 		};
 		Ok(should_forward)