mirror of
https://github.com/openai/codex.git
synced 2026-11-01 19:33:35 +00:00
## Why MCP tool calls reported every aborted approval as "user cancelled MCP tool call", including interrupted requests and automatic cancellations in `exec` mode. ## What changed Carry `user`, `interrupted`, or `non_interactive` cancellation reasons through MCP approval metadata and resolution. Mark explicit TUI cancellations as user actions, interrupted requests as interruptions, and automatic `exec` cancellations as non-interactive. Use the reason to report when a call was cancelled before approval or because interactive approval is unavailable in `exec` mode. Preserve the existing cancellation message for clients that omit the reason. ## Testing Update tests to verify that a closed approval response carries the interruption reason and that `exec` cancellation responses include the non-interactive reason. GitOrigin-RevId: be2a5f7fef539b5db2add53b46aea27e2b8a66fc
864 lines
31 KiB
Rust
864 lines
31 KiB
Rust
//! Central approval policy-stage execution and reviewer routing.
|
|
|
|
use crate::command_canonicalization::canonicalize_command_for_approval;
|
|
use crate::exec_policy::prompt_is_rejected_by_policy;
|
|
use crate::guardian::GuardianNetworkAccessTrigger;
|
|
use crate::guardian::GuardianReviewContext;
|
|
use crate::guardian::GuardianReviewOptions;
|
|
use crate::guardian::decide_approval;
|
|
use crate::guardian::new_guardian_review_id;
|
|
use crate::guardian::spawn_approval_decision;
|
|
use crate::hook_runtime::run_permission_request_hooks;
|
|
use crate::mcp_tool_call::request_mcp_tool_user_approval;
|
|
use crate::sandboxing::SandboxPermissions;
|
|
use crate::session::session::Session;
|
|
use crate::tools::hook_names::HookToolName;
|
|
use crate::tools::runtimes::apply_patch::ApplyPatchApprovalKey;
|
|
use crate::tools::runtimes::unified_exec::UnifiedExecApprovalKey;
|
|
use crate::tools::sandboxing::ApprovalRequestReasons;
|
|
use crate::tools::sandboxing::PermissionRequestPayload;
|
|
use crate::tools::sandboxing::ToolError;
|
|
use crate::tools::sandboxing::with_cached_approval;
|
|
use codex_analytics::GuardianApprovalRequestSource;
|
|
use codex_config::types::AppToolApproval;
|
|
use codex_hooks::PermissionRequestDecision;
|
|
use codex_otel::ToolDecisionSource;
|
|
use codex_prompts::ResolvedModelMessages;
|
|
use codex_protocol::approvals::ExecApprovalKind;
|
|
use codex_protocol::approvals::ExecPolicyAmendment;
|
|
use codex_protocol::approvals::NetworkApprovalContext;
|
|
use codex_protocol::approvals::NetworkApprovalProtocol;
|
|
use codex_protocol::config_types::ApprovalsReviewer;
|
|
use codex_protocol::error::CodexErr;
|
|
use codex_protocol::mcp_approval_meta::McpApprovalCancellationReason;
|
|
use codex_protocol::models::AdditionalPermissionProfile;
|
|
use codex_protocol::openai_models::ModelInfo;
|
|
use codex_protocol::protocol::AskForApproval;
|
|
use codex_protocol::protocol::FileChange;
|
|
use codex_protocol::protocol::NetworkPolicyRuleAction;
|
|
use codex_protocol::protocol::ReviewDecision;
|
|
use codex_protocol::request_permissions::RequestPermissionProfile;
|
|
use codex_tools::ToolName;
|
|
use codex_utils_absolute_path::AbsolutePathBuf;
|
|
use codex_utils_path_uri::PathConvention;
|
|
use codex_utils_path_uri::PathUri;
|
|
use std::collections::HashMap;
|
|
use std::path::PathBuf;
|
|
use std::sync::Arc;
|
|
use tokio_util::sync::CancellationToken;
|
|
use tracing::error;
|
|
use tracing::warn;
|
|
|
|
#[derive(Clone)]
|
|
pub(crate) struct ApprovalContext {
|
|
pub(crate) review_context: GuardianReviewContext,
|
|
pub(crate) cancellation_token: Option<CancellationToken>,
|
|
pub(crate) call_id: String,
|
|
pub(crate) tool_name: ToolName,
|
|
pub(crate) strict_auto_review: bool,
|
|
pub(crate) approval_reason: Option<String>,
|
|
pub(crate) retry_reason: Option<String>,
|
|
pub(crate) network_approval_context: Option<NetworkApprovalContext>,
|
|
}
|
|
|
|
#[derive(Clone, Debug, PartialEq, serde::Serialize)]
|
|
#[serde(tag = "tool", rename_all = "snake_case")]
|
|
pub(crate) enum ApprovalAction {
|
|
ExecCommand {
|
|
id: String,
|
|
environment_id: String,
|
|
command: Vec<String>,
|
|
#[serde(skip_serializing)]
|
|
hook_command: String,
|
|
cwd: PathUri,
|
|
sandbox_permissions: SandboxPermissions,
|
|
additional_permissions: Option<AdditionalPermissionProfile>,
|
|
justification: Option<String>,
|
|
tty: bool,
|
|
proposed_execpolicy_amendment: Option<ExecPolicyAmendment>,
|
|
},
|
|
WriteStdin {
|
|
id: String,
|
|
approval_id: String,
|
|
environment_id: String,
|
|
process_id: i32,
|
|
input: String,
|
|
cwd: PathUri,
|
|
tty: bool,
|
|
sandbox_permissions: SandboxPermissions,
|
|
additional_permissions: Option<AdditionalPermissionProfile>,
|
|
#[serde(skip)]
|
|
additional_permissions_preapproved: bool,
|
|
},
|
|
ApplyPatch {
|
|
id: String,
|
|
environment_id: String,
|
|
cwd: PathUri,
|
|
files: Vec<PathUri>,
|
|
patch: String,
|
|
#[serde(skip_serializing)]
|
|
changes: Arc<HashMap<PathBuf, FileChange>>,
|
|
permissions_preapproved: bool,
|
|
},
|
|
McpToolCall {
|
|
id: String,
|
|
server: String,
|
|
tool_name: String,
|
|
arguments: Option<serde_json::Value>,
|
|
connector_id: Option<String>,
|
|
connector_name: Option<String>,
|
|
connector_description: Option<String>,
|
|
connected_account_email: Option<String>,
|
|
tool_title: Option<String>,
|
|
tool_description: Option<String>,
|
|
annotations: Option<crate::guardian::GuardianMcpAnnotations>,
|
|
#[serde(skip_serializing)]
|
|
hook_tool_name: HookToolName,
|
|
approval_policy: AskForApproval,
|
|
reviewer: ApprovalsReviewer,
|
|
approval_mode: AppToolApproval,
|
|
allow_session_remember: bool,
|
|
allow_persistent_approval: bool,
|
|
},
|
|
NetworkAccess {
|
|
id: String,
|
|
turn_id: String,
|
|
environment_id: String,
|
|
target: String,
|
|
host: String,
|
|
protocol: NetworkApprovalProtocol,
|
|
port: u16,
|
|
trigger: Option<GuardianNetworkAccessTrigger>,
|
|
#[serde(skip_serializing)]
|
|
hook_command: String,
|
|
#[serde(skip_serializing)]
|
|
hook_run_id: String,
|
|
command: Vec<String>,
|
|
cwd: AbsolutePathBuf,
|
|
},
|
|
RequestPermissions {
|
|
id: String,
|
|
environment_id: String,
|
|
turn_id: String,
|
|
reason: Option<String>,
|
|
permissions: RequestPermissionProfile,
|
|
},
|
|
}
|
|
|
|
#[derive(Clone, Debug, Eq, Hash, PartialEq, serde::Serialize)]
|
|
#[serde(untagged)]
|
|
pub(crate) enum ApprovalCacheKey {
|
|
ExecCommand(UnifiedExecApprovalKey),
|
|
ApplyPatch(ApplyPatchApprovalKey),
|
|
}
|
|
|
|
impl ApprovalAction {
|
|
pub(crate) fn permission_request_payload(&self) -> PermissionRequestPayload {
|
|
match self {
|
|
Self::ExecCommand {
|
|
hook_command,
|
|
justification,
|
|
..
|
|
} => PermissionRequestPayload::bash(hook_command.clone(), justification.clone()),
|
|
Self::WriteStdin {
|
|
id,
|
|
approval_id,
|
|
environment_id,
|
|
process_id,
|
|
input,
|
|
cwd,
|
|
tty,
|
|
sandbox_permissions,
|
|
additional_permissions,
|
|
additional_permissions_preapproved: _,
|
|
} => PermissionRequestPayload {
|
|
tool_name: HookToolName::new("write_stdin"),
|
|
tool_input: serde_json::json!({
|
|
"session_id": process_id,
|
|
"chars": input,
|
|
"parent_call_id": id,
|
|
"approval_id": approval_id,
|
|
"environment_id": environment_id,
|
|
"cwd": cwd,
|
|
"tty": tty,
|
|
"sandbox_permissions": sandbox_permissions,
|
|
"additional_permissions": additional_permissions,
|
|
}),
|
|
},
|
|
Self::ApplyPatch { patch, .. } => PermissionRequestPayload {
|
|
tool_name: HookToolName::apply_patch(),
|
|
tool_input: serde_json::json!({ "command": patch }),
|
|
},
|
|
Self::McpToolCall {
|
|
hook_tool_name,
|
|
arguments,
|
|
..
|
|
} => PermissionRequestPayload {
|
|
tool_name: hook_tool_name.clone(),
|
|
tool_input: arguments
|
|
.clone()
|
|
.unwrap_or_else(|| serde_json::Value::Object(serde_json::Map::new())),
|
|
},
|
|
Self::NetworkAccess {
|
|
hook_command,
|
|
target,
|
|
..
|
|
} => PermissionRequestPayload::bash(
|
|
hook_command.clone(),
|
|
Some(format!("network-access {target}")),
|
|
),
|
|
Self::RequestPermissions {
|
|
reason,
|
|
permissions,
|
|
..
|
|
} => PermissionRequestPayload {
|
|
tool_name: HookToolName::new("request_permissions"),
|
|
tool_input: serde_json::json!({
|
|
"reason": reason,
|
|
"permissions": permissions,
|
|
}),
|
|
},
|
|
}
|
|
}
|
|
|
|
pub(crate) fn cache_keys(&self) -> Vec<ApprovalCacheKey> {
|
|
match self {
|
|
Self::ExecCommand {
|
|
environment_id,
|
|
command,
|
|
cwd,
|
|
tty,
|
|
sandbox_permissions,
|
|
additional_permissions,
|
|
..
|
|
} => vec![ApprovalCacheKey::ExecCommand(UnifiedExecApprovalKey {
|
|
environment_id: environment_id.clone(),
|
|
executable: command.first().cloned(),
|
|
command: canonicalize_command_for_approval(command),
|
|
cwd: cwd.clone(),
|
|
tty: *tty,
|
|
sandbox_permissions: *sandbox_permissions,
|
|
additional_permissions: additional_permissions.clone(),
|
|
})],
|
|
Self::McpToolCall { .. }
|
|
| Self::NetworkAccess { .. }
|
|
| Self::RequestPermissions { .. }
|
|
| Self::WriteStdin { .. } => Vec::new(),
|
|
Self::ApplyPatch {
|
|
environment_id,
|
|
files,
|
|
..
|
|
} => files
|
|
.iter()
|
|
.cloned()
|
|
.map(|path| {
|
|
ApprovalCacheKey::ApplyPatch(ApplyPatchApprovalKey {
|
|
environment_id: environment_id.clone(),
|
|
path,
|
|
})
|
|
})
|
|
.collect(),
|
|
}
|
|
}
|
|
|
|
pub(crate) fn into_guardian_request(
|
|
self,
|
|
exec_command_cwd_convention: Option<PathConvention>,
|
|
) -> std::io::Result<crate::guardian::GuardianApprovalRequest> {
|
|
Ok(match self {
|
|
Self::ExecCommand {
|
|
id,
|
|
environment_id,
|
|
command,
|
|
cwd,
|
|
sandbox_permissions,
|
|
additional_permissions,
|
|
justification,
|
|
tty,
|
|
..
|
|
} => crate::guardian::GuardianApprovalRequest::ExecCommand {
|
|
id,
|
|
environment_id,
|
|
command,
|
|
guardian_cwd: codex_utils_path_uri::LegacyAppPathString::from_path_uri(
|
|
&cwd,
|
|
exec_command_cwd_convention.ok_or_else(|| {
|
|
std::io::Error::other("missing exec command cwd convention")
|
|
})?,
|
|
)
|
|
.map_err(std::io::Error::other)?,
|
|
cwd,
|
|
sandbox_permissions,
|
|
additional_permissions,
|
|
justification,
|
|
tty,
|
|
},
|
|
Self::WriteStdin {
|
|
id,
|
|
approval_id,
|
|
environment_id,
|
|
process_id,
|
|
input,
|
|
cwd,
|
|
tty,
|
|
sandbox_permissions,
|
|
additional_permissions,
|
|
additional_permissions_preapproved: _,
|
|
} => crate::guardian::GuardianApprovalRequest::WriteStdin {
|
|
id,
|
|
approval_id,
|
|
environment_id,
|
|
process_id,
|
|
input,
|
|
cwd,
|
|
tty,
|
|
sandbox_permissions,
|
|
additional_permissions,
|
|
},
|
|
Self::ApplyPatch {
|
|
id,
|
|
environment_id,
|
|
cwd,
|
|
files,
|
|
patch,
|
|
..
|
|
} => crate::guardian::GuardianApprovalRequest::ApplyPatch {
|
|
id,
|
|
environment_id,
|
|
cwd,
|
|
files,
|
|
patch,
|
|
},
|
|
Self::McpToolCall {
|
|
id,
|
|
server,
|
|
tool_name,
|
|
arguments,
|
|
connector_id,
|
|
connector_name,
|
|
connector_description,
|
|
connected_account_email,
|
|
tool_title,
|
|
tool_description,
|
|
annotations,
|
|
..
|
|
} => crate::guardian::GuardianApprovalRequest::McpToolCall {
|
|
id,
|
|
server,
|
|
tool_name,
|
|
arguments,
|
|
connector_id,
|
|
connector_name,
|
|
connector_description,
|
|
connected_account_email,
|
|
tool_title,
|
|
tool_description,
|
|
annotations,
|
|
},
|
|
Self::NetworkAccess {
|
|
id,
|
|
turn_id,
|
|
environment_id,
|
|
target,
|
|
host,
|
|
protocol,
|
|
port,
|
|
trigger,
|
|
..
|
|
} => crate::guardian::GuardianApprovalRequest::NetworkAccess {
|
|
id,
|
|
turn_id,
|
|
environment_id,
|
|
target,
|
|
host,
|
|
protocol,
|
|
port,
|
|
trigger,
|
|
},
|
|
Self::RequestPermissions {
|
|
id,
|
|
environment_id,
|
|
turn_id,
|
|
reason,
|
|
permissions,
|
|
} => crate::guardian::GuardianApprovalRequest::RequestPermissions {
|
|
id,
|
|
environment_id,
|
|
turn_id,
|
|
reason,
|
|
permissions,
|
|
},
|
|
})
|
|
}
|
|
}
|
|
|
|
#[derive(Clone, Copy, Debug, PartialEq, Eq)]
|
|
pub(crate) enum ApprovalResolutionSource {
|
|
Hook,
|
|
Guardian,
|
|
User,
|
|
}
|
|
|
|
/// Preserves the approval path even when the decision becomes a tool error.
|
|
/// The source identifies the reviewer, not the actor that caused cancellation.
|
|
#[must_use]
|
|
pub(crate) struct ApprovalResult {
|
|
pub(crate) result: Result<ReviewDecision, ToolError>,
|
|
pub(crate) source: Option<ApprovalResolutionSource>,
|
|
pub(crate) cancellation_reason: Option<McpApprovalCancellationReason>,
|
|
}
|
|
|
|
#[derive(Clone, Debug, PartialEq, Eq)]
|
|
pub(crate) struct ApprovalResolution {
|
|
pub(crate) decision: ReviewDecision,
|
|
pub(crate) source: ApprovalResolutionSource,
|
|
pub(crate) cancellation_reason: Option<McpApprovalCancellationReason>,
|
|
}
|
|
|
|
impl ApprovalResolution {
|
|
fn into_tool_result(self, model_info: &ModelInfo) -> Result<ReviewDecision, ToolError> {
|
|
let source = self.source;
|
|
match self.decision {
|
|
ReviewDecision::ApprovedMcpPolicyAmendment => {
|
|
error!("Tool approval received ApprovedMcpPolicyAmendment");
|
|
Err(ToolError::Rejected(
|
|
"Error while requesting approval".to_string(),
|
|
))
|
|
}
|
|
ReviewDecision::NetworkPolicyAmendment {
|
|
network_policy_amendment,
|
|
} if network_policy_amendment.action == NetworkPolicyRuleAction::Deny => {
|
|
let rejection = match source {
|
|
ApprovalResolutionSource::Hook => "rejected by configuration",
|
|
ApprovalResolutionSource::Guardian => {
|
|
"automatic approval review denied the action"
|
|
}
|
|
ApprovalResolutionSource::User => "rejected by user",
|
|
};
|
|
Err(ToolError::Rejected(rejection.to_string()))
|
|
}
|
|
ReviewDecision::Denied { rejection } => Err(ToolError::Rejected(rejection)),
|
|
ReviewDecision::TimedOut => Err(ToolError::Rejected(
|
|
ResolvedModelMessages::from_model(model_info)
|
|
.auto_review()
|
|
.timeout_instructions
|
|
.to_string(),
|
|
)),
|
|
ReviewDecision::Abort => Err(ToolError::Codex(CodexErr::TurnAborted)),
|
|
decision => Ok(decision),
|
|
}
|
|
}
|
|
}
|
|
|
|
impl Session {
|
|
pub(crate) async fn request_approval(
|
|
self: &Arc<Self>,
|
|
action: ApprovalAction,
|
|
ctx: ApprovalContext,
|
|
) -> ApprovalResult {
|
|
// Retained authority outside current grants needs a new sandbox approval.
|
|
// Current grants permit review of the input; they never approve the input itself.
|
|
let policy = ctx.review_context.turn().approval_policy();
|
|
if matches!(&action, ApprovalAction::WriteStdin { sandbox_permissions, additional_permissions_preapproved, .. }
|
|
if sandbox_permissions.requests_sandbox_override()
|
|
&& (!additional_permissions_preapproved || matches!(policy, AskForApproval::Never)))
|
|
&& !(ctx.strict_auto_review && matches!(policy, AskForApproval::Never))
|
|
&& let Some(reason) =
|
|
prompt_is_rejected_by_policy(policy, /*prompt_is_rule*/ false)
|
|
{
|
|
return ApprovalResult {
|
|
result: Err(ToolError::Rejected(reason.to_string())),
|
|
source: None,
|
|
cancellation_reason: None,
|
|
};
|
|
}
|
|
let is_mcp_tool_call = matches!(&action, ApprovalAction::McpToolCall { .. });
|
|
let is_network_approval = matches!(&action, ApprovalAction::NetworkAccess { .. });
|
|
let permission_request_run_id = match &action {
|
|
ApprovalAction::NetworkAccess { hook_run_id, .. } => hook_run_id.clone(),
|
|
_ if ctx.retry_reason.is_some() => format!("{}:retry", ctx.call_id),
|
|
_ => ctx.call_id.clone(),
|
|
};
|
|
|
|
// Approval precedence is:
|
|
// 1. Hooks
|
|
// 2. If StrictAutoReview || Guardian enabled, then Guardian. Else, user.
|
|
let resolution = match run_permission_request_hooks(
|
|
self,
|
|
&ctx.review_context,
|
|
&permission_request_run_id,
|
|
action.permission_request_payload(),
|
|
)
|
|
.await
|
|
{
|
|
Some(PermissionRequestDecision::Allow) => ApprovalResolution {
|
|
decision: ReviewDecision::Approved,
|
|
source: ApprovalResolutionSource::Hook,
|
|
cancellation_reason: None,
|
|
},
|
|
Some(PermissionRequestDecision::Deny { message }) => ApprovalResolution {
|
|
decision: ReviewDecision::denied(message),
|
|
source: ApprovalResolutionSource::Hook,
|
|
cancellation_reason: None,
|
|
},
|
|
None => self.request_reviewer_approval(action, &ctx).await,
|
|
};
|
|
// Network approvals record their final telemetry after validation and persistence.
|
|
if !is_network_approval {
|
|
record_resolution(&ctx, &resolution);
|
|
}
|
|
let source = Some(resolution.source);
|
|
let cancellation_reason = resolution.cancellation_reason;
|
|
if is_mcp_tool_call && resolution.decision == ReviewDecision::ApprovedMcpPolicyAmendment {
|
|
return ApprovalResult {
|
|
result: Ok(resolution.decision),
|
|
source,
|
|
cancellation_reason,
|
|
};
|
|
}
|
|
if is_network_approval {
|
|
match (&resolution.decision, resolution.source) {
|
|
(
|
|
ReviewDecision::NetworkPolicyAmendment {
|
|
network_policy_amendment,
|
|
},
|
|
_,
|
|
) if network_policy_amendment.action == NetworkPolicyRuleAction::Deny => {
|
|
return ApprovalResult {
|
|
result: Ok(resolution.decision),
|
|
source,
|
|
cancellation_reason,
|
|
};
|
|
}
|
|
(ReviewDecision::Abort, ApprovalResolutionSource::Guardian) => {
|
|
return ApprovalResult {
|
|
result: Err(ToolError::Rejected(
|
|
"automatic approval review was cancelled".to_string(),
|
|
)),
|
|
source,
|
|
cancellation_reason,
|
|
};
|
|
}
|
|
_ => {}
|
|
}
|
|
}
|
|
ApprovalResult {
|
|
result: resolution.into_tool_result(ctx.review_context.turn().model_info()),
|
|
source,
|
|
cancellation_reason,
|
|
}
|
|
}
|
|
|
|
async fn request_reviewer_approval(
|
|
self: &Arc<Self>,
|
|
action: ApprovalAction,
|
|
ctx: &ApprovalContext,
|
|
) -> ApprovalResolution {
|
|
if let Some(decision) = self.request_guardian_approval(action.clone(), ctx).await {
|
|
ApprovalResolution {
|
|
cancellation_reason: (decision == ReviewDecision::Abort)
|
|
.then_some(McpApprovalCancellationReason::Interrupted),
|
|
decision,
|
|
source: ApprovalResolutionSource::Guardian,
|
|
}
|
|
} else {
|
|
self.request_user_approval(&action, ctx).await
|
|
}
|
|
}
|
|
|
|
pub(crate) async fn request_guardian_approval(
|
|
self: &Arc<Self>,
|
|
action: ApprovalAction,
|
|
ctx: &ApprovalContext,
|
|
) -> Option<ReviewDecision> {
|
|
let mut context = ctx.review_context.clone();
|
|
if let ApprovalAction::McpToolCall {
|
|
approval_policy,
|
|
reviewer,
|
|
..
|
|
} = &action
|
|
{
|
|
context.approval_policy = *approval_policy;
|
|
context.approvals_reviewer = *reviewer;
|
|
}
|
|
let is_network_approval = matches!(&action, ApprovalAction::NetworkAccess { .. });
|
|
let review_id = new_guardian_review_id();
|
|
let exec_command_cwd_convention = match &action {
|
|
ApprovalAction::ExecCommand {
|
|
environment_id,
|
|
cwd,
|
|
..
|
|
} => Some(
|
|
match ctx
|
|
.review_context
|
|
.environments()
|
|
.turn_environments()
|
|
.find(|environment| environment.selection.environment_id == *environment_id)
|
|
.and_then(|environment| environment.executor_platform_os.as_deref())
|
|
{
|
|
Some("windows") => PathConvention::Windows,
|
|
Some(_) => PathConvention::Posix,
|
|
// Legacy executors did not report their OS. Preserve the
|
|
// previous spelling-based behavior for those executors.
|
|
None => cwd
|
|
.infer_path_convention()
|
|
.unwrap_or_else(PathConvention::native),
|
|
},
|
|
),
|
|
_ => None,
|
|
};
|
|
let action = crate::guardian::ReviewAction::from_approval_action(
|
|
action,
|
|
exec_command_cwd_convention,
|
|
);
|
|
|
|
let reasons = ApprovalRequestReasons {
|
|
approval: ctx.approval_reason.clone(),
|
|
retry: ctx.retry_reason.clone(),
|
|
};
|
|
let options = GuardianReviewOptions {
|
|
require_guardian: ctx.strict_auto_review,
|
|
plugin_attribution_override: None,
|
|
approval_request_source: GuardianApprovalRequestSource::MainTurn,
|
|
external_cancel: ctx.cancellation_token.clone(),
|
|
require_synchronous_review: false,
|
|
};
|
|
if ctx.cancellation_token.is_some() {
|
|
spawn_approval_decision(
|
|
Arc::clone(self),
|
|
context,
|
|
review_id,
|
|
action,
|
|
reasons,
|
|
options,
|
|
)
|
|
.await
|
|
.unwrap_or_else(|_| {
|
|
Some(ReviewDecision::denied(
|
|
"automatic approval review could not complete",
|
|
))
|
|
})
|
|
} else if is_network_approval {
|
|
let review_cancel = CancellationToken::new();
|
|
let review_cancel_guard = review_cancel.clone().drop_guard();
|
|
let review = tokio::spawn(decide_approval(
|
|
Arc::clone(self),
|
|
context,
|
|
review_id,
|
|
action,
|
|
reasons,
|
|
GuardianReviewOptions {
|
|
external_cancel: Some(review_cancel),
|
|
..options
|
|
},
|
|
));
|
|
let decision = review.await.unwrap_or_else(|err| {
|
|
warn!("network Guardian review task failed: {err}");
|
|
Some(ReviewDecision::denied(
|
|
"automatic approval review could not complete",
|
|
))
|
|
});
|
|
drop(review_cancel_guard.disarm());
|
|
decision
|
|
} else {
|
|
decide_approval(
|
|
Arc::clone(self),
|
|
context,
|
|
review_id,
|
|
action,
|
|
reasons,
|
|
options,
|
|
)
|
|
.await
|
|
}
|
|
}
|
|
|
|
async fn request_user_approval(
|
|
&self,
|
|
action: &ApprovalAction,
|
|
ctx: &ApprovalContext,
|
|
) -> ApprovalResolution {
|
|
let decision = match action {
|
|
ApprovalAction::ExecCommand {
|
|
environment_id,
|
|
command,
|
|
cwd,
|
|
additional_permissions,
|
|
justification,
|
|
proposed_execpolicy_amendment,
|
|
..
|
|
} => {
|
|
let tool_name = "unified_exec";
|
|
let reason = ctx
|
|
.retry_reason
|
|
.clone()
|
|
.or_else(|| ctx.approval_reason.clone())
|
|
.or_else(|| justification.clone());
|
|
let policy_fingerprint = ctx
|
|
.review_context
|
|
.environments()
|
|
.turn_environments()
|
|
.find(|environment| environment.selection.environment_id == *environment_id)
|
|
.and_then(|environment| environment.config().exec_policy.as_ref())
|
|
.map(codex_execpolicy::RequirementsExecPolicy::fingerprint);
|
|
let cache_keys = action
|
|
.cache_keys()
|
|
.into_iter()
|
|
.map(|key| (key, &policy_fingerprint))
|
|
.collect();
|
|
with_cached_approval(&self.services, tool_name, cache_keys, || async {
|
|
self.request_command_approval(
|
|
ctx.review_context.turn(),
|
|
ExecApprovalKind::Command,
|
|
ctx.review_context.model_context(),
|
|
ctx.call_id.clone(),
|
|
/*approval_id*/ None,
|
|
Some(environment_id.clone()),
|
|
command.clone(),
|
|
cwd.clone(),
|
|
reason,
|
|
ctx.network_approval_context.clone(),
|
|
proposed_execpolicy_amendment.clone(),
|
|
additional_permissions.clone(),
|
|
/*available_decisions*/ None,
|
|
/*plugin_attribution_override*/ None,
|
|
)
|
|
.await
|
|
})
|
|
.await
|
|
}
|
|
ApprovalAction::WriteStdin {
|
|
id,
|
|
approval_id,
|
|
environment_id,
|
|
process_id,
|
|
input,
|
|
cwd,
|
|
additional_permissions,
|
|
..
|
|
} => {
|
|
self.request_command_approval(
|
|
ctx.review_context.turn(),
|
|
ExecApprovalKind::WriteStdin,
|
|
ctx.review_context.model_context(),
|
|
id.clone(),
|
|
Some(approval_id.clone()),
|
|
Some(environment_id.clone()),
|
|
vec![
|
|
"write_stdin".to_string(),
|
|
"--session-id".to_string(),
|
|
process_id.to_string(),
|
|
input.clone(),
|
|
],
|
|
cwd.clone(),
|
|
ctx.approval_reason.clone(),
|
|
/*network_approval_context*/ None,
|
|
/*proposed_execpolicy_amendment*/ None,
|
|
additional_permissions.clone(),
|
|
Some(vec![ReviewDecision::Approved, ReviewDecision::Abort]),
|
|
/*plugin_attribution_override*/ None,
|
|
)
|
|
.await
|
|
}
|
|
ApprovalAction::ApplyPatch {
|
|
changes,
|
|
permissions_preapproved,
|
|
..
|
|
} => {
|
|
let reason = ctx
|
|
.retry_reason
|
|
.clone()
|
|
.or_else(|| ctx.approval_reason.clone());
|
|
if *permissions_preapproved && reason.is_none() {
|
|
ReviewDecision::Approved
|
|
} else if reason.is_some() {
|
|
self.request_patch_approval(
|
|
ctx.review_context.turn(),
|
|
ctx.call_id.clone(),
|
|
changes.as_ref().clone(),
|
|
reason,
|
|
/*grant_root*/ None,
|
|
)
|
|
.await
|
|
} else {
|
|
with_cached_approval(
|
|
&self.services,
|
|
"apply_patch",
|
|
action.cache_keys(),
|
|
|| async {
|
|
self.request_patch_approval(
|
|
ctx.review_context.turn(),
|
|
ctx.call_id.clone(),
|
|
changes.as_ref().clone(),
|
|
/*reason*/ None,
|
|
/*grant_root*/ None,
|
|
)
|
|
.await
|
|
},
|
|
)
|
|
.await
|
|
}
|
|
}
|
|
ApprovalAction::McpToolCall { .. } => {
|
|
return request_mcp_tool_user_approval(
|
|
self,
|
|
ctx.review_context.turn(),
|
|
&ctx.call_id,
|
|
action,
|
|
)
|
|
.await;
|
|
}
|
|
ApprovalAction::NetworkAccess {
|
|
environment_id,
|
|
command,
|
|
cwd,
|
|
..
|
|
} => {
|
|
self.request_command_approval(
|
|
ctx.review_context.turn(),
|
|
ExecApprovalKind::Command,
|
|
ctx.review_context.model_context(),
|
|
ctx.call_id.clone(),
|
|
/*approval_id*/ None,
|
|
Some(environment_id.clone()),
|
|
command.clone(),
|
|
cwd.clone().into(),
|
|
ctx.approval_reason.clone(),
|
|
ctx.network_approval_context.clone(),
|
|
/*proposed_execpolicy_amendment*/ None,
|
|
/*additional_permissions*/ None,
|
|
/*available_decisions*/ None,
|
|
/*plugin_attribution_override*/ None,
|
|
)
|
|
.await
|
|
}
|
|
ApprovalAction::RequestPermissions { .. } => {
|
|
unreachable!("permission requests are routed directly to Guardian")
|
|
}
|
|
};
|
|
ApprovalResolution {
|
|
decision,
|
|
source: ApprovalResolutionSource::User,
|
|
cancellation_reason: None,
|
|
}
|
|
}
|
|
}
|
|
|
|
fn record_resolution(ctx: &ApprovalContext, resolution: &ApprovalResolution) {
|
|
let source = match resolution.source {
|
|
ApprovalResolutionSource::Hook => ToolDecisionSource::Config,
|
|
ApprovalResolutionSource::Guardian => ToolDecisionSource::AutomatedReviewer,
|
|
ApprovalResolutionSource::User => ToolDecisionSource::User,
|
|
};
|
|
ctx.review_context.turn().session_telemetry.tool_decision(
|
|
&ctx.tool_name,
|
|
&ctx.call_id,
|
|
&resolution.decision,
|
|
Some(source),
|
|
);
|
|
}
|
|
|
|
#[cfg(all(test, unix))]
|
|
#[path = "approvals_tests.rs"]
|
|
mod tests;
|