_ Git - rust-lightning/blob - lightning/src/util/logger.rs

   1 // Pruned copy of crate rust log, without global logger
   2 // https://github.com/rust-lang-nursery/log #7a60286
   3 //
   4 // This file is licensed under the Apache License, Version 2.0 <LICENSE-APACHE
   5 // or http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
   6 // <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your option.
   7 // You may not use this file except in accordance with one or both of these
   8 // licenses.
   9
  10 //! Log traits live here, which are called throughout the library to provide useful information for
  11 //! debugging purposes.
  12 //!
  13 //! There is currently 2 ways to filter log messages. First one, by using compilation features, e.g "max_level_off".
  14 //! The second one, client-side by implementing check against Record Level field.
  15 //! Each module may have its own Logger or share one.
  16
  17 use bitcoin::secp256k1::PublicKey;
  18
  19 use core::cmp;
  20 use core::fmt;
  21 use core::ops::Deref;
  22
  23 use crate::ln::ChannelId;
  24 #[cfg(c_bindings)]
  25 use crate::prelude::*; // Needed for String
  26
  27 static LOG_LEVEL_NAMES: [&'static str; 6] = ["GOSSIP", "TRACE", "DEBUG", "INFO", "WARN", "ERROR"];
  28
  29 /// An enum representing the available verbosity levels of the logger.
  30 #[derive(Copy, Clone, PartialEq, Eq, Debug, Hash)]
  31 pub enum Level {
  32         /// Designates extremely verbose information, including gossip-induced messages
  33         Gossip,
  34         /// Designates very low priority, often extremely verbose, information
  35         Trace,
  36         /// Designates lower priority information
  37         Debug,
  38         /// Designates useful information
  39         Info,
  40         /// Designates hazardous situations
  41         Warn,
  42         /// Designates very serious errors
  43         Error,
  44 }
  45
  46 impl PartialOrd for Level {
  47         #[inline]
  48         fn partial_cmp(&self, other: &Level) -> Option<cmp::Ordering> {
  49                 Some(self.cmp(other))
  50         }
  51
  52         #[inline]
  53         fn lt(&self, other: &Level) -> bool {
  54                 (*self as usize) < *other as usize
  55         }
  56
  57         #[inline]
  58         fn le(&self, other: &Level) -> bool {
  59                 *self as usize <= *other as usize
  60         }
  61
  62         #[inline]
  63         fn gt(&self, other: &Level) -> bool {
  64                 *self as usize > *other as usize
  65         }
  66
  67         #[inline]
  68         fn ge(&self, other: &Level) -> bool {
  69                 *self as usize >= *other as usize
  70         }
  71 }
  72
  73 impl Ord for Level {
  74         #[inline]
  75         fn cmp(&self, other: &Level) -> cmp::Ordering {
  76                 (*self as usize).cmp(&(*other as usize))
  77         }
  78 }
  79
  80 impl fmt::Display for Level {
  81         fn fmt(&self, fmt: &mut fmt::Formatter) -> fmt::Result {
  82                 fmt.pad(LOG_LEVEL_NAMES[*self as usize])
  83         }
  84 }
  85
  86 impl Level {
  87         /// Returns the most verbose logging level.
  88         #[inline]
  89         pub fn max() -> Level {
  90                 Level::Gossip
  91         }
  92 }
  93
  94 macro_rules! impl_record {
  95         ($($args: lifetime)?, $($nonstruct_args: lifetime)?) => {
  96 /// A Record, unit of logging output with Metadata to enable filtering
  97 /// Module_path, file, line to inform on log's source
  98 #[derive(Clone, Debug)]
  99 pub struct Record<$($args)?> {
 100         /// The verbosity level of the message.
 101         pub level: Level,
 102         /// The node id of the peer pertaining to the logged record.
 103         ///
 104         /// Note that in some cases a [`Self::channel_id`] may be filled in but this may still be
 105         /// `None`, depending on if the peer information is readily available in LDK when the log is
 106         /// generated.
 107         pub peer_id: Option<PublicKey>,
 108         /// The channel id of the channel pertaining to the logged record. May be a temporary id before
 109         /// the channel has been funded.
 110         pub channel_id: Option<ChannelId>,
 111         #[cfg(not(c_bindings))]
 112         /// The message body.
 113         pub args: fmt::Arguments<'a>,
 114         #[cfg(c_bindings)]
 115         /// The message body.
 116         pub args: String,
 117         /// The module path of the message.
 118         pub module_path: &'static str,
 119         /// The source file containing the message.
 120         pub file: &'static str,
 121         /// The line containing the message.
 122         pub line: u32,
 123 }
 124
 125 impl<$($args)?> Record<$($args)?> {
 126         /// Returns a new Record.
 127         ///
 128         /// This is not exported to bindings users as fmt can't be used in C
 129         #[inline]
 130         pub fn new<$($nonstruct_args)?>(
 131                 level: Level, peer_id: Option<PublicKey>, channel_id: Option<ChannelId>,
 132                 args: fmt::Arguments<'a>, module_path: &'static str, file: &'static str, line: u32
 133         ) -> Record<$($args)?> {
 134                 Record {
 135                         level,
 136                         peer_id,
 137                         channel_id,
 138                         #[cfg(not(c_bindings))]
 139                         args,
 140                         #[cfg(c_bindings)]
 141                         args: format!("{}", args),
 142                         module_path,
 143                         file,
 144                         line,
 145                 }
 146         }
 147 }
 148 } }
 149 #[cfg(not(c_bindings))]
 150 impl_record!('a, );
 151 #[cfg(c_bindings)]
 152 impl_record!(, 'a);
 153
 154 /// A trait encapsulating the operations required of a logger.
 155 pub trait Logger {
 156         /// Logs the [`Record`].
 157         fn log(&self, record: Record);
 158 }
 159
 160 /// Adds relevant context to a [`Record`] before passing it to the wrapped [`Logger`].
 161 pub struct WithContext<'a, L: Deref> where L::Target: Logger {
 162         /// The logger to delegate to after adding context to the record.
 163         logger: &'a L,
 164         /// The node id of the peer pertaining to the logged record.
 165         peer_id: Option<PublicKey>,
 166         /// The channel id of the channel pertaining to the logged record.
 167         channel_id: Option<ChannelId>,
 168 }
 169
 170 impl<'a, L: Deref> Logger for WithContext<'a, L> where L::Target: Logger {
 171         fn log(&self, mut record: Record) {
 172                 if self.peer_id.is_some() {
 173                         record.peer_id = self.peer_id
 174                 };
 175                 if self.channel_id.is_some() {
 176                         record.channel_id = self.channel_id;
 177                 }
 178                 self.logger.log(record)
 179         }
 180 }
 181
 182 impl<'a, L: Deref> WithContext<'a, L> where L::Target: Logger {
 183         /// Wraps the given logger, providing additional context to any logged records.
 184         pub fn from(logger: &'a L, peer_id: Option<PublicKey>, channel_id: Option<ChannelId>) -> Self {
 185                 WithContext {
 186                         logger,
 187                         peer_id,
 188                         channel_id,
 189                 }
 190         }
 191 }
 192
 193 /// Wrapper for logging a [`PublicKey`] in hex format.
 194 ///
 195 /// This is not exported to bindings users as fmt can't be used in C
 196 #[doc(hidden)]
 197 pub struct DebugPubKey<'a>(pub &'a PublicKey);
 198 impl<'a> core::fmt::Display for DebugPubKey<'a> {
 199         fn fmt(&self, f: &mut core::fmt::Formatter) -> Result<(), core::fmt::Error> {
 200                 for i in self.0.serialize().iter() {
 201                         write!(f, "{:02x}", i)?;
 202                 }
 203                 Ok(())
 204         }
 205 }
 206
 207 /// Wrapper for logging byte slices in hex format.
 208 ///
 209 /// This is not exported to bindings users as fmt can't be used in C
 210 #[doc(hidden)]
 211 pub struct DebugBytes<'a>(pub &'a [u8]);
 212 impl<'a> core::fmt::Display for DebugBytes<'a> {
 213         fn fmt(&self, f: &mut core::fmt::Formatter) -> Result<(), core::fmt::Error> {
 214                 for i in self.0 {
 215                         write!(f, "{:02x}", i)?;
 216                 }
 217                 Ok(())
 218         }
 219 }
 220
 221 /// Wrapper for logging `Iterator`s.
 222 ///
 223 /// This is not exported to bindings users as fmt can't be used in C
 224 #[doc(hidden)]
 225 pub struct DebugIter<T: fmt::Display, I: core::iter::Iterator<Item = T> + Clone>(pub I);
 226 impl<T: fmt::Display, I: core::iter::Iterator<Item = T> + Clone> fmt::Display for DebugIter<T, I> {
 227         fn fmt(&self, f: &mut fmt::Formatter) -> Result<(), fmt::Error> {
 228                 write!(f, "[")?;
 229                 let mut iter = self.0.clone();
 230                 if let Some(item) = iter.next() {
 231                         write!(f, "{}", item)?;
 232                 }
 233                 while let Some(item) = iter.next() {
 234                         write!(f, ", {}", item)?;
 235                 }
 236                 write!(f, "]")?;
 237                 Ok(())
 238         }
 239 }
 240
 241 #[cfg(test)]
 242 mod tests {
 243         use bitcoin::secp256k1::{PublicKey, SecretKey, Secp256k1};
 244         use crate::ln::ChannelId;
 245         use crate::util::logger::{Logger, Level, WithContext};
 246         use crate::util::test_utils::TestLogger;
 247         use crate::sync::Arc;
 248
 249         #[test]
 250         fn test_level_show() {
 251                 assert_eq!("INFO", Level::Info.to_string());
 252                 assert_eq!("ERROR", Level::Error.to_string());
 253                 assert_ne!("WARN", Level::Error.to_string());
 254         }
 255
 256         struct WrapperLog {
 257                 logger: Arc<dyn Logger>
 258         }
 259
 260         impl WrapperLog {
 261                 fn new(logger: Arc<dyn Logger>) -> WrapperLog {
 262                         WrapperLog {
 263                                 logger,
 264                         }
 265                 }
 266
 267                 fn call_macros(&self) {
 268                         log_error!(self.logger, "This is an error");
 269                         log_warn!(self.logger, "This is a warning");
 270                         log_info!(self.logger, "This is an info");
 271                         log_debug!(self.logger, "This is a debug");
 272                         log_trace!(self.logger, "This is a trace");
 273                         log_gossip!(self.logger, "This is a gossip");
 274                 }
 275         }
 276
 277         #[test]
 278         fn test_logging_macros() {
 279                 let mut logger = TestLogger::new();
 280                 logger.enable(Level::Gossip);
 281                 let logger : Arc<dyn Logger> = Arc::new(logger);
 282                 let wrapper = WrapperLog::new(Arc::clone(&logger));
 283                 wrapper.call_macros();
 284         }
 285
 286         #[test]
 287         fn test_logging_with_context() {
 288                 let logger = &TestLogger::new();
 289                 let secp_ctx = Secp256k1::new();
 290                 let pk = PublicKey::from_secret_key(&secp_ctx, &SecretKey::from_slice(&[42; 32]).unwrap());
 291                 let context_logger = WithContext::from(&logger, Some(pk), Some(ChannelId([0; 32])));
 292                 log_error!(context_logger, "This is an error");
 293                 log_warn!(context_logger, "This is an error");
 294                 log_debug!(context_logger, "This is an error");
 295                 log_trace!(context_logger, "This is an error");
 296                 log_gossip!(context_logger, "This is an error");
 297                 log_info!(context_logger, "This is an error");
 298                 logger.assert_log_context_contains(
 299                         "lightning::util::logger::tests", Some(pk), Some(ChannelId([0;32])), 6
 300                 );
 301         }
 302
 303         #[test]
 304         fn test_logging_with_multiple_wrapped_context() {
 305                 let logger = &TestLogger::new();
 306                 let secp_ctx = Secp256k1::new();
 307                 let pk = PublicKey::from_secret_key(&secp_ctx, &SecretKey::from_slice(&[42; 32]).unwrap());
 308                 let context_logger = &WithContext::from(&logger, None, Some(ChannelId([0; 32])));
 309                 let full_context_logger = WithContext::from(&context_logger, Some(pk), None);
 310                 log_error!(full_context_logger, "This is an error");
 311                 log_warn!(full_context_logger, "This is an error");
 312                 log_debug!(full_context_logger, "This is an error");
 313                 log_trace!(full_context_logger, "This is an error");
 314                 log_gossip!(full_context_logger, "This is an error");
 315                 log_info!(full_context_logger, "This is an error");
 316                 logger.assert_log_context_contains(
 317                         "lightning::util::logger::tests", Some(pk), Some(ChannelId([0;32])), 6
 318                 );
 319         }
 320
 321         #[test]
 322         fn test_log_ordering() {
 323                 assert!(Level::Error > Level::Warn);
 324                 assert!(Level::Error >= Level::Warn);
 325                 assert!(Level::Error >= Level::Error);
 326                 assert!(Level::Warn > Level::Info);
 327                 assert!(Level::Warn >= Level::Info);
 328                 assert!(Level::Warn >= Level::Warn);
 329                 assert!(Level::Info > Level::Debug);
 330                 assert!(Level::Info >= Level::Debug);
 331                 assert!(Level::Info >= Level::Info);
 332                 assert!(Level::Debug > Level::Trace);
 333                 assert!(Level::Debug >= Level::Trace);
 334                 assert!(Level::Debug >= Level::Debug);
 335                 assert!(Level::Trace > Level::Gossip);
 336                 assert!(Level::Trace >= Level::Gossip);
 337                 assert!(Level::Trace >= Level::Trace);
 338                 assert!(Level::Gossip >= Level::Gossip);
 339
 340                 assert!(Level::Error <= Level::Error);
 341                 assert!(Level::Warn < Level::Error);
 342                 assert!(Level::Warn <= Level::Error);
 343                 assert!(Level::Warn <= Level::Warn);
 344                 assert!(Level::Info < Level::Warn);
 345                 assert!(Level::Info <= Level::Warn);
 346                 assert!(Level::Info <= Level::Info);
 347                 assert!(Level::Debug < Level::Info);
 348                 assert!(Level::Debug <= Level::Info);
 349                 assert!(Level::Debug <= Level::Debug);
 350                 assert!(Level::Trace < Level::Debug);
 351                 assert!(Level::Trace <= Level::Debug);
 352                 assert!(Level::Trace <= Level::Trace);
 353                 assert!(Level::Gossip < Level::Trace);
 354                 assert!(Level::Gossip <= Level::Trace);
 355                 assert!(Level::Gossip <= Level::Gossip);
 356         }
 357 }