Merge pull request #3115 from alecchendev/2024-06-specific-async-sign
[rust-lightning] / lightning / src / util / logger.rs
index 92ea8ffed55a91c218c4481e01a83f924536585d..33ef9b1f7838694bcc749903a1a39ad7d3fae8f2 100644 (file)
@@ -20,7 +20,8 @@ use core::cmp;
 use core::fmt;
 use core::ops::Deref;
 
-use crate::ln::ChannelId;
+use crate::ln::types::ChannelId;
+use crate::ln::PaymentHash;
 #[cfg(c_bindings)]
 use crate::prelude::*; // Needed for String
 
@@ -91,10 +92,12 @@ impl Level {
        }
 }
 
+macro_rules! impl_record {
+       ($($args: lifetime)?, $($nonstruct_args: lifetime)?) => {
 /// A Record, unit of logging output with Metadata to enable filtering
 /// Module_path, file, line to inform on log's source
 #[derive(Clone, Debug)]
-pub struct Record<'a> {
+pub struct Record<$($args)?> {
        /// The verbosity level of the message.
        pub level: Level,
        /// The node id of the peer pertaining to the logged record.
@@ -118,22 +121,23 @@ pub struct Record<'a> {
        pub file: &'static str,
        /// The line containing the message.
        pub line: u32,
-
-       #[cfg(c_bindings)]
-       /// We don't actually use the lifetime parameter in C bindings (as there is no good way to
-       /// communicate a lifetime to a C, or worse, Java user).
-       _phantom: core::marker::PhantomData<&'a ()>,
+       /// The payment hash.
+       ///
+       /// Note that this is only filled in for logs pertaining to a specific payment, and will be
+       /// `None` for logs which are not directly related to a payment.
+       pub payment_hash: Option<PaymentHash>,
 }
 
-impl<'a> Record<'a> {
+impl<$($args)?> Record<$($args)?> {
        /// Returns a new Record.
        ///
        /// This is not exported to bindings users as fmt can't be used in C
        #[inline]
-       pub fn new(
+       pub fn new<$($nonstruct_args)?>(
                level: Level, peer_id: Option<PublicKey>, channel_id: Option<ChannelId>,
-               args: fmt::Arguments<'a>, module_path: &'static str, file: &'static str, line: u32
-       ) -> Record<'a> {
+               args: fmt::Arguments<'a>, module_path: &'static str, file: &'static str, line: u32,
+               payment_hash: Option<PaymentHash>
+       ) -> Record<$($args)?> {
                Record {
                        level,
                        peer_id,
@@ -145,11 +149,15 @@ impl<'a> Record<'a> {
                        module_path,
                        file,
                        line,
-                       #[cfg(c_bindings)]
-                       _phantom: core::marker::PhantomData,
+                       payment_hash,
                }
        }
 }
+} }
+#[cfg(not(c_bindings))]
+impl_record!('a, );
+#[cfg(c_bindings)]
+impl_record!(, 'a);
 
 /// A trait encapsulating the operations required of a logger.
 pub trait Logger {
@@ -158,6 +166,9 @@ pub trait Logger {
 }
 
 /// Adds relevant context to a [`Record`] before passing it to the wrapped [`Logger`].
+///
+/// This is not exported to bindings users as lifetimes are problematic and there's little reason
+/// for this to be used downstream anyway.
 pub struct WithContext<'a, L: Deref> where L::Target: Logger {
        /// The logger to delegate to after adding context to the record.
        logger: &'a L,
@@ -165,6 +176,8 @@ pub struct WithContext<'a, L: Deref> where L::Target: Logger {
        peer_id: Option<PublicKey>,
        /// The channel id of the channel pertaining to the logged record.
        channel_id: Option<ChannelId>,
+       /// The payment hash of the payment pertaining to the logged record.
+       payment_hash: Option<PaymentHash>
 }
 
 impl<'a, L: Deref> Logger for WithContext<'a, L> where L::Target: Logger {
@@ -175,17 +188,21 @@ impl<'a, L: Deref> Logger for WithContext<'a, L> where L::Target: Logger {
                if self.channel_id.is_some() {
                        record.channel_id = self.channel_id;
                }
+               if self.payment_hash.is_some() {
+                       record.payment_hash = self.payment_hash;
+               }
                self.logger.log(record)
        }
 }
 
 impl<'a, L: Deref> WithContext<'a, L> where L::Target: Logger {
        /// Wraps the given logger, providing additional context to any logged records.
-       pub fn from(logger: &'a L, peer_id: Option<PublicKey>, channel_id: Option<ChannelId>) -> Self {
+       pub fn from(logger: &'a L, peer_id: Option<PublicKey>, channel_id: Option<ChannelId>, payment_hash: Option<PaymentHash>) -> Self {
                WithContext {
                        logger,
                        peer_id,
                        channel_id,
+                       payment_hash,
                }
        }
 }
@@ -241,7 +258,8 @@ impl<T: fmt::Display, I: core::iter::Iterator<Item = T> + Clone> fmt::Display fo
 #[cfg(test)]
 mod tests {
        use bitcoin::secp256k1::{PublicKey, SecretKey, Secp256k1};
-       use crate::ln::ChannelId;
+       use crate::ln::types::ChannelId;
+       use crate::ln::PaymentHash;
        use crate::util::logger::{Logger, Level, WithContext};
        use crate::util::test_utils::TestLogger;
        use crate::sync::Arc;
@@ -288,7 +306,8 @@ mod tests {
                let logger = &TestLogger::new();
                let secp_ctx = Secp256k1::new();
                let pk = PublicKey::from_secret_key(&secp_ctx, &SecretKey::from_slice(&[42; 32]).unwrap());
-               let context_logger = WithContext::from(&logger, Some(pk), Some(ChannelId([0; 32])));
+               let payment_hash = PaymentHash([0; 32]);
+               let context_logger = WithContext::from(&logger, Some(pk), Some(ChannelId([0; 32])), Some(payment_hash));
                log_error!(context_logger, "This is an error");
                log_warn!(context_logger, "This is an error");
                log_debug!(context_logger, "This is an error");
@@ -305,8 +324,9 @@ mod tests {
                let logger = &TestLogger::new();
                let secp_ctx = Secp256k1::new();
                let pk = PublicKey::from_secret_key(&secp_ctx, &SecretKey::from_slice(&[42; 32]).unwrap());
-               let context_logger = &WithContext::from(&logger, None, Some(ChannelId([0; 32])));
-               let full_context_logger = WithContext::from(&context_logger, Some(pk), None);
+               let payment_hash = PaymentHash([0; 32]);
+               let context_logger = &WithContext::from(&logger, None, Some(ChannelId([0; 32])), Some(payment_hash));
+               let full_context_logger = WithContext::from(&context_logger, Some(pk), None, None);
                log_error!(full_context_logger, "This is an error");
                log_warn!(full_context_logger, "This is an error");
                log_debug!(full_context_logger, "This is an error");