add support for oom detection

Signed-off-by: Ryan Phillips <[email protected]>

conmon-rs/server/Cargo.toml

@@ -31,6 +31,8 @@ strum = { version = "0.24.0", features = ["derive"] }
 systemd-journal-logger = "0.5.0"
 shadow-rs = "0.11.0"
 multimap = "0.8.3"
+regex = "1.5.5"
+notify = "5.0.0-pre.14"
 
 [build-dependencies]
 shadow-rs = "0.11.0"

conmon-rs/server/src/child_reaper.rs

@@ -2,6 +2,7 @@
 use crate::{
     child::Child,
     container_io::{ContainerIO, ContainerIOType, SharedContainerIO},
+    oom_watcher::OOMWatcher,
 };
 use anyhow::{anyhow, format_err, Context, Result};
 use getset::{CopyGetters, Getters, Setters};
@@ -204,15 +205,20 @@ impl ReapableChild {
 
     fn watch(&self) -> Result<(Sender<ExitChannelData>, Receiver<ExitChannelData>)> {
         let exit_paths = self.exit_paths().clone();
+        let oom_exit_paths = self.exit_paths().clone();
         let pid = self.pid();
         // Only one exit code will be written.
         let (exit_tx, exit_rx) = broadcast::channel(1);
         let exit_tx_clone = exit_tx.clone();
+        let mut cancel_rx = exit_tx.subscribe();
         let timeout = *self.timeout();
 
         task::spawn(async move {
             let exit_code: i32;
             let mut timed_out = false;
+            task::spawn(async move {
+                OOMWatcher::wait_for_oom(&mut cancel_rx, pid, &oom_exit_paths).await
+            });
             let wait_for_exit_code = task::spawn_blocking(move || Self::wait_for_exit_code(pid));
             if let Some(timeout) = timeout {
                 match time::timeout_at(timeout, wait_for_exit_code).await {

conmon-rs/server/src/lib.rs

@@ -10,6 +10,7 @@ mod container_log;
 mod cri_logger;
 mod init;
 mod listener;
+mod oom_watcher;
 mod rpc;
 mod server;
 mod streams;