feat: set THP_DISABLE=true in shim, and restore it before starting runc

I think shim does

containerd
    |
    |--shim1   --start
        |
        |--shim2    (this shim will on host)
            |
            |--runc create (when containerd send create request by ttrpc)
                |
                |--runc init (this is the pid 1 in container)

    we should set thp_disabled=1 in shim1 --start, because if we set this
    in shim 2, the huge page has been setted while func main() running,
    we set thp_disabled cannot change the setted huge pages.
    So We need to set thp_disabled=1 in shim1 so that shim2 inherits the
    settings of the parent process shim1, and shim2 has closed the
    hugepage when it starts.

    For runc processes, we need to set thp_disabled='before' in shim2 after
    fork() and before execve(). So we use cmd.pre_exec to do this.

crates/runc-shim/Cargo.toml

@@ -33,6 +33,7 @@ serde_json = "1.0.74"
 oci-spec = "0.6.0"
 crossbeam = "0.8.1"
 uuid = { version = "1.0.0", features = ["v4"] }
+prctl = "1.0.0"
 
 # Async dependencies
 async-trait = { workspace = true }

crates/runc-shim/src/service.rs

@@ -79,7 +79,17 @@ impl Shim for Service {
             None => {}
         }
 
-        let address = spawn(opts, &grouping, Vec::new()).await?;
+        let mut vars: Vec<(&str, &str)> = Vec::new();
+        let thp_disabled = match prctl::get_thp_disable() {
+            Ok(x) => {
+                x.to_string()
+            }
+            _ => {"".to_string()}
+        };
+        match prctl::set_thp_disable(true) { _=> {} }
+        vars.push(("THP_DISABLED", thp_disabled.as_str()));
+
+        let address = spawn(opts, &grouping, vars).await?;
         write_str_to_file("address", &address).await?;
         Ok(address)
     }

crates/runc/Cargo.toml

@@ -28,6 +28,7 @@ thiserror = "1.0.30"
 time = { version = "0.3.7", features = ["serde", "std"] }
 uuid = { version = "1.0.0", features = ["v4"] }
 os_pipe = "1.0.0"
+prctl = "1.0.0"
 
 # Async dependencies
 tokio = { workspace = true, features = ["full"], optional = true }

crates/runc/src/lib.rs

@@ -366,8 +366,23 @@ pub trait Spawner: Debug {
 /// and some other utilities.
 #[cfg(feature = "async")]
 impl Runc {
-    async fn launch(&self, cmd: Command, combined_output: bool) -> Result<Response> {
+    async fn launch(&self, mut cmd: Command, combined_output: bool) -> Result<Response> {
         debug!("Execute command {:?}", cmd);
+        if let Ok(thp) = std::env::var("THP_DISABLED") {
+            if let Ok(thp_disabled) = thp.parse::<bool>() {
+                unsafe {
+                    cmd.pre_exec(move || {
+                        match prctl::set_thp_disable(thp_disabled) {
+                            Err(e) => {
+                                log::debug!("set_thp_disable err: {}", e);
+                            }
+                            _ => {}
+                        };
+                        Ok(())
+                    });
+                }
+            }
+        }
         let (status, pid, stdout, stderr) = self.spawner.execute(cmd).await?;
         if status.success() {
             let output = if combined_output {