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;
@@ -238,17 +238,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 <[email protected]>", "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>");