Compare commits

...

3 Commits

Author SHA1 Message Date
zhao-oai
b81cb7ceb3 Fix sandbox detection for user shell commands (#6094) 2025-11-01 17:27:03 -04:00
kevin zhao
c7a3428986 fix 2025-10-31 17:27:35 -04:00
kevin zhao
d609dfa2fc escalating permissions 2025-10-31 17:15:50 -04:00
5 changed files with 21 additions and 6 deletions

View File

@@ -89,7 +89,10 @@ impl SessionTask for UserShellCommandTask {
let tool_call = ToolCall { let tool_call = ToolCall {
tool_name: USER_SHELL_TOOL_NAME.to_string(), tool_name: USER_SHELL_TOOL_NAME.to_string(),
call_id: Uuid::new_v4().to_string(), call_id: Uuid::new_v4().to_string(),
payload: ToolPayload::LocalShell { params }, payload: ToolPayload::LocalShell {
params,
is_user_shell_command: true,
},
}; };
let router = Arc::new(ToolRouter::from_config(&turn_context.tools_config, None)); let router = Arc::new(ToolRouter::from_config(&turn_context.tools_config, None));

View File

@@ -40,6 +40,7 @@ pub enum ToolPayload {
}, },
LocalShell { LocalShell {
params: ShellToolCallParams, params: ShellToolCallParams,
is_user_shell_command: bool,
}, },
UnifiedExec { UnifiedExec {
arguments: String, arguments: String,
@@ -56,7 +57,7 @@ impl ToolPayload {
match self { match self {
ToolPayload::Function { arguments } => Cow::Borrowed(arguments), ToolPayload::Function { arguments } => Cow::Borrowed(arguments),
ToolPayload::Custom { input } => Cow::Borrowed(input), ToolPayload::Custom { input } => Cow::Borrowed(input),
ToolPayload::LocalShell { params } => Cow::Owned(params.command.join(" ")), ToolPayload::LocalShell { params, .. } => Cow::Owned(params.command.join(" ")),
ToolPayload::UnifiedExec { arguments } => Cow::Borrowed(arguments), ToolPayload::UnifiedExec { arguments } => Cow::Borrowed(arguments),
ToolPayload::Mcp { raw_arguments, .. } => Cow::Borrowed(raw_arguments), ToolPayload::Mcp { raw_arguments, .. } => Cow::Borrowed(raw_arguments),
} }

View File

@@ -82,7 +82,10 @@ impl ToolHandler for ShellHandler {
) )
.await .await
} }
ToolPayload::LocalShell { params } => { ToolPayload::LocalShell {
params,
is_user_shell_command,
} => {
let exec_params = Self::to_exec_params(params, turn.as_ref()); let exec_params = Self::to_exec_params(params, turn.as_ref());
Self::run_exec_like( Self::run_exec_like(
tool_name.as_str(), tool_name.as_str(),
@@ -91,7 +94,7 @@ impl ToolHandler for ShellHandler {
turn, turn,
tracker, tracker,
call_id, call_id,
true, is_user_shell_command,
) )
.await .await
} }
@@ -219,6 +222,7 @@ impl ShellHandler {
env: exec_params.env.clone(), env: exec_params.env.clone(),
with_escalated_permissions: exec_params.with_escalated_permissions, with_escalated_permissions: exec_params.with_escalated_permissions,
justification: exec_params.justification.clone(), justification: exec_params.justification.clone(),
is_user_shell_command,
}; };
let mut orchestrator = ToolOrchestrator::new(); let mut orchestrator = ToolOrchestrator::new();
let mut runtime = ShellRuntime::new(); let mut runtime = ShellRuntime::new();

View File

@@ -120,7 +120,10 @@ impl ToolRouter {
Ok(Some(ToolCall { Ok(Some(ToolCall {
tool_name: "local_shell".to_string(), tool_name: "local_shell".to_string(),
call_id, call_id,
payload: ToolPayload::LocalShell { params }, payload: ToolPayload::LocalShell {
params,
is_user_shell_command: false,
},
})) }))
} }
} }

View File

@@ -34,6 +34,7 @@ pub struct ShellRequest {
pub env: std::collections::HashMap<String, String>, pub env: std::collections::HashMap<String, String>,
pub with_escalated_permissions: Option<bool>, pub with_escalated_permissions: Option<bool>,
pub justification: Option<String>, pub justification: Option<String>,
pub is_user_shell_command: bool,
} }
impl ProvidesSandboxRetryData for ShellRequest { impl ProvidesSandboxRetryData for ShellRequest {
@@ -121,6 +122,9 @@ impl Approvable<ShellRequest> for ShellRuntime {
policy: AskForApproval, policy: AskForApproval,
sandbox_policy: &SandboxPolicy, sandbox_policy: &SandboxPolicy,
) -> bool { ) -> bool {
if req.is_user_shell_command {
return false;
}
if is_known_safe_command(&req.command) { if is_known_safe_command(&req.command) {
return false; return false;
} }
@@ -146,7 +150,7 @@ impl Approvable<ShellRequest> for ShellRuntime {
} }
fn wants_escalated_first_attempt(&self, req: &ShellRequest) -> bool { fn wants_escalated_first_attempt(&self, req: &ShellRequest) -> bool {
req.with_escalated_permissions.unwrap_or(false) req.is_user_shell_command || req.with_escalated_permissions.unwrap_or(false)
} }
} }