Merge pull request #661 from bytedance/feat-heatbeat-info

heartbeat report add info field

Author: yoloyyh

rasp/librasp/src/process.rs

@@ -44,6 +44,7 @@ pub struct ProcessInfo {
     pub start_time: Option<f32>,
     pub try_attach_count: u16,
     pub attached_count: u16,
+    pub failed_reason:Option<String>,
 }
 
 #[allow(non_camel_case_types)]
@@ -219,6 +220,13 @@ impl ProcessInfo {
         }
         Ok(self.namespace_info.clone().unwrap())
     }
+
+    pub fn update_failed_reason(&mut self, reason: &String) -> AnyhowResult<()> {
+        if self.failed_reason.is_none() {
+            self.failed_reason = Some(reason.clone());
+        }
+        Ok(())
+    }
     pub fn get_mnt_ns(&self) -> AnyhowResult<String> {
         if let Some(ref ns) = self.namespace_info {
             return match ns.mnt.clone() {

rasp/plugin/src/monitor.rs

@@ -486,6 +486,11 @@ fn internal_main(
                     ) {
                         warn!("operation thread send command to receiver err: {}, pid: {}", e, process.pid);
                     }
+                    let _ = process.update_failed_reason(&e.to_string());
+                    let mut opp = operation_process_rw.write();
+                    opp.insert(process.pid, process.clone());
+                    drop(opp);
+
                     continue;
                 }
             };

rasp/plugin/src/utils.rs

@@ -47,6 +47,7 @@ pub fn generate_heartbeat(watched_process: &ProcessInfo) -> HashMap<&'static str
     message.insert("try_attach_count", watched_process.try_attach_count.to_string());
     message.insert("attached_count", watched_process.attached_count.to_string());
     message.insert("probe_version", RASP_VERSION.to_string());
+    message.insert("info", watched_process.failed_reason.clone().unwrap_or("".to_string()));
     message.insert("uptime", match count_uptime(watched_process.start_time.unwrap_or(0 as f32)) {
         Ok(t) => t.to_string(),
         Err(e) => {