Add support for cgroups managed by systemd

Cargo.toml

@@ -22,3 +22,4 @@ once_cell = "1.6.0"
 futures = { version = "0.3", features = ["thread-pool"] }
 regex = "1.5"
 oci_spec = { version = "0.1.0", path = "./oci_spec" }
+systemd = "0.8.2"

src/cgroups/common.rs

@@ -10,6 +10,7 @@ use anyhow::{bail, Result};
 use nix::unistd::Pid;
 use oci_spec::LinuxResources;
 use procfs::process::Process;
+use systemd::daemon::booted;
 
 use crate::cgroups::v1;
 use crate::cgroups::v2;
@@ -50,7 +51,10 @@ pub fn write_cgroup_file<P: AsRef<Path>>(path: P, data: &str) -> Result<()> {
  Ok(())
 }
 
-pub fn create_cgroup_manager<P: Into<PathBuf>>(cgroup_path: P) -> Result<Box<dyn CgroupManager>> {
+pub fn create_cgroup_manager<P: Into<PathBuf>>(
+ cgroup_path: P,
+ systemd_cgroup: bool,
+) -> Result<Box<dyn CgroupManager>> {
  let cgroup_mount = Process::myself()?
  .mountinfo()?
  .into_iter()
@@ -78,13 +82,23 @@ pub fn create_cgroup_manager<P: Into<PathBuf>>(cgroup_path: P) -> Result<Box<dyn
  match cgroup_override {
  Ok(v) if v == "true" => {
  log::info!("cgroup manager V2 will be used");
+ if systemd_cgroup {
+ if !booted()? {
+ bail!("systemd cgroup flag passed, but systemd support for managing cgroups is not available");
+ }
+ log::info!("systemd cgroup manager will be used");
+ return Ok(Box::new(v2::SystemDCGroupManager::new(
+ cgroup2.mount_point,
+ cgroup_path.into(),
+ )?));
+ }
  Ok(Box::new(v2::manager::Manager::new(
  cgroup2.mount_point,
  cgroup_path.into(),
  )?))
  }
  _ => Ok(Box::new(v1::manager::Manager::new(cgroup_path.into())?)),
- } 
+ }
  }
  _ => bail!("could not find cgroup filesystem"),
  }

src/cgroups/v2/mod.rs

@@ -7,3 +7,5 @@ mod io;
 pub mod manager;
 mod memory;
 mod pids;
+pub mod systemd_manager;
+pub use systemd_manager::SystemDCGroupManager;