test(rsjudge-runner): 🚚 normalize test names

crates/rsjudge-runner/examples/exploit.rs → crates/rsjudge-runner/examples/cap_test.rs

@@ -2,7 +2,7 @@
 
 use std::path::PathBuf;
 
-use capctl::{Cap, FullCapState};
+use capctl::{Cap, CapState};
 use rsjudge_runner::{use_caps, user::builder, RunAs};
 use rsjudge_utils::command::check_output;
 use tokio::process::Command;
@@ -19,11 +19,11 @@ use tokio::process::Command;
 /// ```
 #[tokio::main]
 async fn main() -> anyhow::Result<()> {
-    dbg!(FullCapState::get_current().unwrap());
+    dbg!(CapState::get_current().unwrap());
 
     use_caps!(Cap::SETUID, Cap::SETGID, Cap::DAC_READ_SEARCH);
 
-    dbg!(FullCapState::get_current().unwrap());
+    dbg!(CapState::get_current().unwrap());
 
     // Get the path to the examples.
     // This crate is located at crates/rsjudge-runner,

crates/rsjudge-runner/examples/exploit_inner.rs

@@ -1,6 +1,6 @@
 // SPDX-License-Identifier: Apache-2.0
 
-use capctl::FullCapState;
+use capctl::CapState;
 use nix::unistd::{setuid, Uid};
 
 /// This example should be called by the `exploit` example.
@@ -11,7 +11,7 @@ use nix::unistd::{setuid, Uid};
 /// have no permitted capabilities.
 fn main() {
     eprintln!("Start exploit_inner binary");
-    dbg!(FullCapState::get_current().unwrap());
+    dbg!(CapState::get_current().unwrap());
 
     eprintln!("Starting setuid syscall.");
     let result = setuid(Uid::from_raw(0)).expect_err("Should fail to set UID");

crates/rsjudge-runner/examples/normal.rs

@@ -1,9 +1,9 @@
 // SPDX-License-Identifier: Apache-2.0
 
-use capctl::FullCapState;
+use capctl::CapState;
 
 fn main() {
     eprintln!("Start normal binary");
-    dbg!(FullCapState::get_current().unwrap());
+    dbg!(CapState::get_current().unwrap());
     eprintln!("End normal binary");
 }

crates/rsjudge-runner/examples/rusage_test.rs

@@ -0,0 +1,43 @@
+use std::{path::PathBuf, time::Duration};
+
+use rsjudge_runner::utils::resources::{rusage::WaitForResourceUsage, RunWithResourceLimit};
+use rsjudge_traits::resource::ResourceLimit;
+use tokio::{process::Command, time::Instant};
+
+#[tokio::main]
+async fn main() -> anyhow::Result<()> {
+    let examples = PathBuf::from(env!("CARGO_MANIFEST_DIR"))
+        .parent()
+        .and_then(|p| p.parent())
+        .ok_or_else(|| anyhow::anyhow!("cannot find crate root"))?
+        .join("target/debug/examples");
+    let spin_lock = examples.join("spin_lock");
+    eprintln!("Starting spin_lock with CPU time limit of 1s, wall time limit 2s:");
+    let start_time = Instant::now();
+    let _ = Command::new(spin_lock)
+        .spawn_with_resource_limit(ResourceLimit::new(
+            Some(Duration::from_secs(1)),
+            Some(Duration::from_secs(2)),
+            None,
+            None,
+        ))?
+        .wait_for_resource_usage()
+        .await;
+
+    dbg!(start_time.elapsed());
+    let sleep = examples.join("sleep");
+    eprintln!("Starting sleep with CPU time limit of 1s, wall time limit 2s:");
+    let start_time = Instant::now();
+    let _ = Command::new(sleep)
+        .spawn_with_resource_limit(ResourceLimit::new(
+            Some(Duration::from_secs(1)),
+            Some(Duration::from_secs(2)),
+            None,
+            None,
+        ))?
+        .wait_for_resource_usage()
+        .await;
+
+    dbg!(start_time.elapsed());
+    Ok(())
+}

crates/rsjudge-runner/examples/sleep.rs

@@ -1,34 +1,6 @@
-use std::{path::PathBuf, time::Duration};
+use std::{thread::sleep, time::Duration};
 
-use capctl::{Cap, CapState};
-use rsjudge_runner::{
-    use_caps,
-    utils::resources::{rusage::WaitForResourceUsage, RunWithResourceLimit},
-};
-use rsjudge_traits::resource::ResourceLimit;
-use tokio::process::Command;
-
-#[tokio::main]
-async fn main() -> anyhow::Result<()> {
-    use_caps!(Cap::DAC_READ_SEARCH);
-    dbg!(CapState::get_current().unwrap());
-
-    let examples = PathBuf::from(env!("CARGO_MANIFEST_DIR"))
-        .parent()
-        .and_then(|p| p.parent())
-        .ok_or_else(|| anyhow::anyhow!("cannot find crate root"))?
-        .join("target/debug/examples");
-    let sleep_inner = examples.join("sleep_inner");
-    dbg!(&sleep_inner);
-    let status = Command::new(sleep_inner)
-        .spawn_with_resource_limit(ResourceLimit::new(
-            Some(Duration::from_secs(1)),
-            Some(Duration::from_secs(2)),
-            None,
-            None,
-        ))?
-        .wait_for_resource_usage()
-        .await?;
-    dbg!(status);
-    Ok(())
+fn main() {
+    println!("Trying to sleep for 10s.");
+    sleep(Duration::from_secs(10));
 }

crates/rsjudge-runner/examples/spin_lock.rs

@@ -0,0 +1,7 @@
+use std::time::{Duration, Instant};
+
+fn main() {
+    println!("Trying to spin for 10s.");
+    let start_time = Instant::now();
+    while start_time.elapsed() < Duration::from_secs(10) {}
+}

xtask/src/main.rs

@@ -31,7 +31,7 @@ enum Command {
     /// Requires `sudo` to set capabilities.
     CapTest,
     /// Run integrated sleep test.
-    SleepTest,
+    RusageTest,
     /// Debug a command.
     #[cfg(feature = "dbg")]
     Debug,
@@ -55,22 +55,18 @@ fn main() -> anyhow::Result<()> {
         Command::CapTest => cmd! {
             cargo build "--examples" "--workspace";
 
-            echo "Setting capabilities for demo and exploit binaries with sudo.";
+            echo "Setting capabilities for get_user_info and cap_test binaries with sudo.";
 
-            sudo setcap "cap_setuid,cap_setgid=p" "target/debug/examples/demo";
-            sudo setcap "cap_setuid,cap_setgid,cap_dac_read_search=p" "target/debug/examples/exploit";
+            sudo setcap "cap_setuid,cap_setgid=p" "target/debug/examples/get_user_info";
+            sudo setcap "cap_setuid,cap_setgid,cap_dac_read_search=p" "target/debug/examples/cap_test";
 
-            "target/debug/examples/demo";
-            "target/debug/examples/exploit";
+            "target/debug/examples/get_user_info";
+            "target/debug/examples/cap_test";
         },
-        Command::SleepTest => cmd! {
+        Command::RusageTest => cmd! {
             cargo build "--examples" "--workspace";
 
-            echo "Setting capabilities for sleep binary with sudo.";
-
-            sudo setcap "cap_dac_read_search=p" "target/debug/examples/sleep";
-
-            "target/debug/examples/sleep";
+            "target/debug/examples/rusage_test";
         },
 
         #[cfg(feature = "dbg")]