Send an email when a pnic occurred

Author: majecty

Cargo.lock

@@ -28,7 +28,7 @@ use ckey::{Address, NetworkId, PlatformAddress};
 use ckeystore::accounts_dir::RootDiskDirectory;
 use ckeystore::KeyStore;
 use clap::ArgMatches;
-use clogger::{self, EmailAlarmConfig, LoggerConfig};
+use clogger::{self, EmailAlarm, EmailAlarmConfig, LoggerConfig};
 use cnetwork::{Filters, NetworkConfig, NetworkControl, NetworkService, RoutingTable, SocketAddr};
 use csync::{BlockSyncExtension, BlockSyncSender, SnapshotService, TransactionSyncExtension};
 use ctimer::TimerLoop;
@@ -236,17 +236,21 @@ pub fn run_node(matches: &ArgMatches) -> Result<(), String> {
  .expect("Current time should be later than unix epoch")
  .subsec_nanos() as usize,
  );
- let email_alarm_config = if !config.email_alarm.disable.unwrap() {
- match (&config.email_alarm.to, &config.email_alarm.sendgrid_key) {
+ let email_alarm = if !config.email_alarm.disable.unwrap() {
+ let config = match (&config.email_alarm.to, &config.email_alarm.sendgrid_key) {
  (Some(to), Some(sendgrid_key)) => Some(EmailAlarmConfig::new(to.to_string(), sendgrid_key.to_string())),
  (None, _) => return Err("email-alarm-to is not specified".to_string()),
  (_, None) => return Err("email-alarm-sendgrid-key is not specified".to_string()),
- }
+ };
+ config.as_ref().map(EmailAlarm::new)
  } else {
  None
  };
- clogger::init(&LoggerConfig::new(instance_id), &email_alarm_config)
+ clogger::init(&LoggerConfig::new(instance_id), email_alarm.clone())
  .expect("Logger must be successfully initialized");
+ if let Some(email_alarm) = email_alarm {
+ panic_hook::set_with_email_alarm(email_alarm);
+ }
 
  let pf = load_password_file(&config.operating.password_path)?;
  let base_path = config.operating.base_path.as_ref().unwrap().clone();

codechain/run_node.rs

@@ -36,7 +36,7 @@ pub fn run_account_command(matches: &ArgMatches) -> Result<(), String> {
  return Ok(())
  }
 
- clogger::init(&LoggerConfig::new(0), &None).expect("Logger must be successfully initialized");
+ clogger::init(&LoggerConfig::new(0), None).expect("Logger must be successfully initialized");
 
  let keys_path = get_global_argument(matches, "keys-path").unwrap_or_else(|| DEFAULT_KEYS_PATH.into());
  let dir = RootDiskDirectory::create(keys_path).expect("Cannot read key path directory");

codechain/subcommand/account_command.rs

@@ -14,6 +14,7 @@ impl EmailAlarmConfig {
  }
 }
 
+#[derive(Clone)]
 pub struct EmailAlarm {
  pub to: String,
  pub sendgrid_key: String,

util/logger/src/email.rs

@@ -38,8 +38,7 @@ use logger::Logger;
 
 pub use log::Level;
 
-pub fn init(config: &LoggerConfig, email_alarm_config: &Option<EmailAlarmConfig>) -> Result<(), SetLoggerError> {
- let email_alarm = email_alarm_config.as_ref().map(EmailAlarm::new);
+pub fn init(config: &LoggerConfig, email_alarm: Option<EmailAlarm>) -> Result<(), SetLoggerError> {
  let logger = Logger::new(config, email_alarm);
  log::set_max_level(logger.filter());
  log::set_boxed_logger(Box::new(logger))
@@ -53,5 +52,4 @@ lazy_static! {
  pub static ref SLOGGER: StructuredLogger = StructuredLogger::create();
 }
 
-use email::EmailAlarm;
-pub use email::EmailAlarmConfig;
+pub use email::{EmailAlarm, EmailAlarmConfig};

util/logger/src/lib.rs

@@ -8,3 +8,4 @@ authors = ["Parity Technologies <admin@parity.io>", "CodeChain Team <hi@codechai
 
 [dependencies]
 backtrace = "0.3.2"
+codechain-logger = { path = "../logger" }

util/panic_hook/Cargo.toml

@@ -17,8 +17,10 @@
 //! Custom panic hook with bug report link
 
 extern crate backtrace;
+extern crate codechain_logger as clogger;
 
 use backtrace::Backtrace;
+use clogger::EmailAlarm;
 use std::panic::{self, PanicInfo};
 use std::thread;
 
@@ -27,6 +29,10 @@ pub fn set() {
  panic::set_hook(Box::new(panic_hook));
 }
 
+pub fn set_with_email_alarm(email_alarm: clogger::EmailAlarm) {
+ panic::set_hook(Box::new(move |info| panic_hook_with_email_alarm(&email_alarm, info)));
+}
+
 static ABOUT_PANIC: &str = "
 This is a bug. Please report it at:
 
@@ -39,6 +45,14 @@ fn panic_hook(info: &PanicInfo) {
  exit_on_debug_mode();
 }
 
+fn panic_hook_with_email_alarm(email_alarm: &EmailAlarm, info: &PanicInfo) {
+ let message = panic_message(info);
+ eprintln!("{}", message);
+ let message_for_email = message.replace("\n", "<br>");
+ email_alarm.send(&message_for_email);
+ exit_on_debug_mode();
+}
+
 fn panic_message(info: &PanicInfo) -> String {
  let location = info.location();
  let file = location.as_ref().map(|l| l.file()).unwrap_or("<unknown>");