bug(wrap): add oom_score_adj to exec specs (#124)

This value can be forwarded from a CRI and ensures an appropriate OOM
score is set for processes in cases where high memory use is seen.

Signed-off-by: Alexander Merritt <alexander@edera.dev>

Author: alexandermerritt

src/config.rs

@@ -85,6 +85,9 @@ pub struct ExecutableSpec {
     /// Requires `no_new_privs = true`.
     #[serde(default)]
     pub seccomp: Option<SeccompFilter>,
+
+    /// An optional out-of-memory score adjustment value.
+    pub oom_score_adj: Option<i32>,
 }
 
 #[derive(Default, Debug, Serialize, Deserialize)]

src/runner.rs

@@ -97,6 +97,11 @@ impl AttachRequestBuilder {
         self
     }
 
+    pub fn set_oom_score_adj(mut self, score: i32) -> AttachRequestBuilder {
+        self.config.exec.oom_score_adj = Some(score);
+        self
+    }
+
     pub fn push_namespace(mut self, ns: Namespace) -> AttachRequestBuilder {
         if self.config.namespaces.is_none() {
             self.config.namespaces = vec![].into();
@@ -211,6 +216,11 @@ impl CreateRequestBuilder {
         self
     }
 
+    pub fn set_oom_score_adj(mut self, score: i32) -> CreateRequestBuilder {
+        self.config.exec.oom_score_adj = Some(score);
+        self
+    }
+
     pub fn set_hostname(mut self, hostname: &str) -> CreateRequestBuilder {
         self.config.hostname = hostname.to_string().into();
         self

src/wrap.rs

@@ -639,6 +639,11 @@ impl Wrappable for CreateRequest {
 
         debug!("mount tree finalized, doing final prep");
 
+        // Ensure the process receives the desired out-of-memory score adjustment.
+        if let Some(score) = self.exec.oom_score_adj {
+            fs::write("/proc/self/oom_score_adj", score.to_string())?;
+        }
+
         // We need to toggle SECBIT before we change UID/GID,
         // or else changing UID/GID may cause us to lose the capabilities
         // we need to explicitly drop capabilities later on.
@@ -841,6 +846,11 @@ impl Wrappable for AttachRequest {
 
         apply_capabilities(self.capabilities.as_ref())?;
 
+        // Ensure the process receives the desired out-of-memory score adjustment.
+        if let Some(score) = self.exec.oom_score_adj {
+            fs::write("/proc/self/oom_score_adj", score.to_string())?;
+        }
+
         debug!("all namespaces joined -- forking child");
         fork_and_wait()?;