Support argv

pwncollege · Jan 20, 2025 · 4c5d38c · 4c5d38c
1 parent 5eb8bb2
commit 4c5d38c
Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/Dockerfile b/Dockerfile
@@ -16,4 +16,4 @@ COPY tests /tests
 
 RUN chmod 4755 /tests/programs/*
 
-CMD ["pytest", "-vv", "/tests"]
+CMD ["pytest", "-v", "/tests"]
diff --git a/src/main.rs b/src/main.rs
@@ -33,6 +33,7 @@ fn main() {
     let matches = opts.parse(&exec_argv[1..]).unwrap();
 
     script_argv.push(path.to_str().unwrap().to_string());
+    script_argv.extend_from_slice(&args[if args.len() == 2 { 2.. } else { 3.. }]);
 
     let stat = stat::stat(path).unwrap_or_else(|err| {
         eprintln!("{}: {}", path.display(), err);