From f171ac10bc1e79bde57d994c60edba42ba9d3237 Mon Sep 17 00:00:00 2001
From: mii443 <mii@sfc.wide.ad.jp>
Date: Fri, 18 Oct 2024 08:10:15 +0000
Subject: [PATCH] add cgroup enter

---
 src/bin/izoli.rs     | 2 ++
 src/cgroup/cgroup.rs | 6 ++++++
 2 files changed, 8 insertions(+)

diff --git a/src/bin/izoli.rs b/src/bin/izoli.rs
index 8fa6213..56a30f9 100644
--- a/src/bin/izoli.rs
+++ b/src/bin/izoli.rs
@@ -12,4 +12,6 @@ fn main() {
     println!("{:?}", cgroup.get_stat());
     println!("{:?}", cgroup.get_max_depth());
     println!("{:?}", cgroup.get_max_descendants());
+
+    cgroup.enter().unwrap();
 }
diff --git a/src/cgroup/cgroup.rs b/src/cgroup/cgroup.rs
index f7c3440..3f94e1c 100644
--- a/src/cgroup/cgroup.rs
+++ b/src/cgroup/cgroup.rs
@@ -30,6 +30,12 @@ impl CGroup {
         fs::create_dir_all(root)
     }
 
+    pub fn enter(&self) -> Result<(), std::io::Error> {
+        let pid = std::process::id();
+
+        self.add_procs(vec![pid])
+    }
+
     pub fn read(&self, name: &str) -> Result<String, std::io::Error> {
         let path = self.get_file_path(name);
         let mut file = File::open(path)?;