mirror of
https://github.com/pchuan98/codex.git
synced 2026-07-01 00:31:56 +08:00
Michael Bolin
58f0e5ab74
As described in detail in `codex-rs/execpolicy/README.md` introduced in this PR, `execpolicy` is a tool that lets you define a set of _patterns_ used to match [`execv(3)`](https://linux.die.net/man/3/execv) invocations. When a pattern is matched, `execpolicy` returns the parsed version in a structured form that is amenable to static analysis. The primary use case is to define patterns match commands that should be auto-approved by a tool such as Codex. This supports a richer pattern matching mechanism that the sort of prefix-matching we have done to date, e.g.: https://github.com/openai/codex/blob/5e40d9d2211737f46136610497bcd9a8271009e0/codex-cli/src/approvals.ts#L333-L354 Note we are still playing with the API and the `system_path` option in particular still needs some work.
51 lines
1.6 KiB
Rust
51 lines
1.6 KiB
Rust
use codex_execpolicy::ArgType;
|
|
use codex_execpolicy::Error;
|
|
use codex_execpolicy::ExecCall;
|
|
use codex_execpolicy::MatchedArg;
|
|
use codex_execpolicy::MatchedExec;
|
|
use codex_execpolicy::PolicyParser;
|
|
use codex_execpolicy::Result;
|
|
use codex_execpolicy::ValidExec;
|
|
|
|
extern crate codex_execpolicy;
|
|
|
|
#[test]
|
|
fn test_invalid_subcommand() -> Result<()> {
|
|
let unparsed_policy = r#"
|
|
define_program(
|
|
program="fake_executable",
|
|
args=["subcommand", "sub-subcommand"],
|
|
)
|
|
"#;
|
|
let parser = PolicyParser::new("test_invalid_subcommand", unparsed_policy);
|
|
let policy = parser.parse().expect("failed to parse policy");
|
|
let valid_call = ExecCall::new("fake_executable", &["subcommand", "sub-subcommand"]);
|
|
assert_eq!(
|
|
Ok(MatchedExec::Match {
|
|
exec: ValidExec::new(
|
|
"fake_executable",
|
|
vec![
|
|
MatchedArg::new(0, ArgType::Literal("subcommand".to_string()), "subcommand")?,
|
|
MatchedArg::new(
|
|
1,
|
|
ArgType::Literal("sub-subcommand".to_string()),
|
|
"sub-subcommand"
|
|
)?,
|
|
],
|
|
&[]
|
|
)
|
|
}),
|
|
policy.check(&valid_call)
|
|
);
|
|
|
|
let invalid_call = ExecCall::new("fake_executable", &["subcommand", "not-a-real-subcommand"]);
|
|
assert_eq!(
|
|
Err(Error::LiteralValueDidNotMatch {
|
|
expected: "sub-subcommand".to_string(),
|
|
actual: "not-a-real-subcommand".to_string()
|
|
}),
|
|
policy.check(&invalid_call)
|
|
);
|
|
Ok(())
|
|
}
|