Compare commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
a633db6186 | ||
|
|
2b3f5baaeb |
@@ -589,6 +589,47 @@ When dynamic profile switching is enabled and a profile is activated via `gitea_
|
|||||||
2. Call `gitea_whoami` with the target remote to prove and verify the fresh Gitea authenticated identity.
|
2. Call `gitea_whoami` with the target remote to prove and verify the fresh Gitea authenticated identity.
|
||||||
This guarantees the active profile operations align with the actual Gitea authenticated user credential.
|
This guarantees the active profile operations align with the actual Gitea authenticated user credential.
|
||||||
|
|
||||||
|
### 4. Review-State Invalidation on Profile Switch (#690)
|
||||||
|
|
||||||
|
A cross-profile activation (e.g. reviewer → author → reviewer) is a session
|
||||||
|
boundary for formal review state. On any switch where the activated profile
|
||||||
|
differs from the previous one, `gitea_activate_profile` invalidates, in
|
||||||
|
memory **and** in durable session state (for both the old and new profile
|
||||||
|
identities):
|
||||||
|
|
||||||
|
- preflight identity/capability stamps (`gitea_whoami` / `gitea_resolve_task_capability` proof),
|
||||||
|
- review workflow-load proof (`gitea_load_review_workflow`),
|
||||||
|
- the review decision lock (including `final_review_decision_ready` markers),
|
||||||
|
- the reviewer PR session lease binding,
|
||||||
|
- live namespace-health assessments.
|
||||||
|
|
||||||
|
Before any formal verdict (`gitea_mark_final_review_decision` /
|
||||||
|
`gitea_submit_pr_review`) the full reviewer preflight must be re-established
|
||||||
|
under the new profile: `gitea_whoami`, `gitea_load_review_workflow`,
|
||||||
|
`gitea_resolve_task_capability(review_pr)`, live head re-pin, and lease
|
||||||
|
re-acquire/adopt. Switching back to the earlier profile does **not**
|
||||||
|
resurrect the prior run — durable state keyed by either profile identity is
|
||||||
|
cleared at switch time.
|
||||||
|
|
||||||
|
The supported pattern remains **separate session/namespace per role**
|
||||||
|
(dual-namespace, §2): file author-side follow-ups from an author session,
|
||||||
|
not by hopping profiles inside a formal review run. Runtime profile
|
||||||
|
switching is the operator-approved exception and always carries the
|
||||||
|
re-preflight cost above.
|
||||||
|
|
||||||
|
### 5. Namespace Provenance (#690)
|
||||||
|
|
||||||
|
The server cannot derive its own client-managed MCP namespace name, so a
|
||||||
|
launcher may declare it via the `GITEA_MCP_NAMESPACE` environment variable
|
||||||
|
(e.g. `gitea-reviewer`). `gitea_whoami`, `gitea_get_runtime_context`, and
|
||||||
|
`gitea_resolve_task_capability` report `namespace_provenance` — the
|
||||||
|
configured client namespace, the active execution profile, and, for tasks
|
||||||
|
with a required namespace (`review_pr` → `gitea-reviewer`, `merge_pr` →
|
||||||
|
`gitea-merger`), a mismatch verdict. A declared namespace that disagrees
|
||||||
|
with the requested task's required namespace **fails closed**. An
|
||||||
|
undeclared namespace is reported as `unknown` and is never treated as
|
||||||
|
proof either way.
|
||||||
|
|
||||||
## Gitea MCP Runtime Isolation and Worktree Safety
|
## Gitea MCP Runtime Isolation and Worktree Safety
|
||||||
|
|
||||||
To ensure high availability and prevent broken feature worktrees from disabling essential security/identity controls, the Gitea MCP server implements runtime isolation:
|
To ensure high availability and prevent broken feature worktrees from disabling essential security/identity controls, the Gitea MCP server implements runtime isolation:
|
||||||
|
|||||||
@@ -86,3 +86,24 @@ When a namespace returns EOF, follow
|
|||||||
|
|
||||||
When blocked, repair the IDE namespace and re-record a healthy
|
When blocked, repair the IDE namespace and re-record a healthy
|
||||||
`client_namespace` assessment before retrying the mutation.
|
`client_namespace` assessment before retrying the mutation.
|
||||||
|
|
||||||
|
## Namespace provenance (#690)
|
||||||
|
|
||||||
|
A server process cannot derive the name of the client-managed namespace it is
|
||||||
|
registered under, so the launcher may declare it with the
|
||||||
|
`GITEA_MCP_NAMESPACE` environment variable (e.g. `GITEA_MCP_NAMESPACE=gitea-reviewer`).
|
||||||
|
|
||||||
|
- `gitea_whoami`, `gitea_get_runtime_context`, and
|
||||||
|
`gitea_resolve_task_capability` report `namespace_provenance`: the declared
|
||||||
|
client namespace, the active execution profile, and — for tasks with a
|
||||||
|
required namespace (`review_pr`/`submit_review` → `gitea-reviewer`,
|
||||||
|
`merge_pr` → `gitea-merger`) — a `mismatch` verdict.
|
||||||
|
- A declared namespace that disagrees with the requested task's required
|
||||||
|
namespace **fails closed** (`allowed_in_current_session=false` with a STOP
|
||||||
|
guidance entry).
|
||||||
|
- An undeclared namespace is reported as `namespace_source="unknown"` and is
|
||||||
|
never treated as proof either way.
|
||||||
|
- A profile switch via `gitea_activate_profile` clears all recorded live
|
||||||
|
namespace-health assessments; re-probe through the client before further
|
||||||
|
review/merge mutations.
|
||||||
|
|
||||||
|
|||||||
@@ -1,93 +0,0 @@
|
|||||||
# MCP restart audit events and incidents (#665)
|
|
||||||
|
|
||||||
Restarts and recovery attempts leave a forensic trail. Failed drains and
|
|
||||||
break-glass paths also raise durable Gitea incident issues so unsafe restarts
|
|
||||||
cannot be silently repeated.
|
|
||||||
|
|
||||||
Parent umbrella: **#655**. Related: impact coordinator **#658**, drain proof
|
|
||||||
**#661**, restart classes **#663**, break-glass **#664**, post-restart reconcile
|
|
||||||
**#662**, vision **#652**, roadmap **#653**, console recovery **#642**.
|
|
||||||
|
|
||||||
## Components
|
|
||||||
|
|
||||||
| Piece | Where | Responsibility |
|
|
||||||
|-------|-------|----------------|
|
|
||||||
| Event schema + emission | `restart_audit.py` | `mcp.restart.*` vocabulary, redacted payload builder, append-only sink via `gitea_audit` |
|
|
||||||
| Fail-closed privileged gate | `restart_audit.require_audit_or_deny` | When `GITEA_AUDIT_LOG` is set and the write fails, privileged apply is denied |
|
|
||||||
| Incident descriptors | `restart_audit.build_incident_descriptor` | Durable follow-up issues (failed drain, break-glass, reconcile unresolved, unguarded) |
|
|
||||||
| Materializer | `restart_audit.materialize_incident` | Injected `create_issue_fn` (network kept out of pure tests) |
|
|
||||||
| Wiring | `gitea_request_mcp_restart` | Correlation id, impact-preview audit, apply-gate / break-glass audit + incident creation |
|
|
||||||
|
|
||||||
## Event vocabulary
|
|
||||||
|
|
||||||
| Event type | When |
|
|
||||||
|------------|------|
|
|
||||||
| `mcp.restart.impact_preview` | Every `gitea_request_mcp_restart` evaluation |
|
|
||||||
| `mcp.restart.drain_enter` | Drain window starts (schema reserved; emit from drain path) |
|
|
||||||
| `mcp.restart.drain_exit` | Drain window ends |
|
|
||||||
| `mcp.restart.drain_proof` | Drain-proof verification result |
|
|
||||||
| `mcp.restart.apply_gate` | Apply hard gate (`dry_run=False`) |
|
|
||||||
| `mcp.restart.break_glass` | Authorized break-glass bypass |
|
|
||||||
| `mcp.restart.post_restart_reconcile` | Post-restart reconcile outcome |
|
|
||||||
| `mcp.restart.narrower_recovery` | Narrower recovery attempt recorded |
|
|
||||||
| `mcp.restart.unguarded_detected` | Unguarded restart path detected |
|
|
||||||
|
|
||||||
All free text is redacted before sink write or issue body assembly. Emission
|
|
||||||
never raises; callers decide fail-closed policy.
|
|
||||||
|
|
||||||
## Correlation
|
|
||||||
|
|
||||||
Each restart lifecycle mints a short `correlation_id` (`rst-` + 16 hex) shared
|
|
||||||
across impact preview → apply gate → incident descriptors so operators can join
|
|
||||||
the trail.
|
|
||||||
|
|
||||||
## Privileged deny-on-audit-fail
|
|
||||||
|
|
||||||
Rollout policy (issue #665):
|
|
||||||
|
|
||||||
1. Configure `GITEA_AUDIT_LOG` so writes land.
|
|
||||||
2. Only then enforce deny when a privileged restart path cannot audit.
|
|
||||||
|
|
||||||
When audit is **not** configured, privileged apply still proceeds (no false
|
|
||||||
denials during rollout). When audit **is** configured and the write fails,
|
|
||||||
`apply_authorized` is cleared.
|
|
||||||
|
|
||||||
## Incidents
|
|
||||||
|
|
||||||
| Kind | Trigger |
|
|
||||||
|------|---------|
|
|
||||||
| `restart_failed_drain` | Apply denied by drain hard gate / failed proof |
|
|
||||||
| `restart_break_glass` | Any authorized break-glass apply |
|
|
||||||
| `restart_reconcile_unresolved` | Post-restart reconcile left work unresolved |
|
|
||||||
| `restart_unguarded_detected` | Unguarded restart attempt detected |
|
|
||||||
|
|
||||||
Break-glass **always** creates an incident descriptor (and a Gitea issue when
|
|
||||||
the create path is available). Failed drain does the same. Incident bodies
|
|
||||||
include correlation id, session, class, scope, proof id, and redacted reasons.
|
|
||||||
|
|
||||||
Default labels: `mcp-health`, `safety`, `observability`, `status:ready`,
|
|
||||||
`type:bug`, `workflow-hardening`.
|
|
||||||
|
|
||||||
## Tool payload surface
|
|
||||||
|
|
||||||
`gitea_request_mcp_restart` returns:
|
|
||||||
|
|
||||||
* `correlation_id` — lifecycle join key
|
|
||||||
* `restart_audit.impact_preview_written` — sink success for the preview event
|
|
||||||
* `restart_audit.apply_gate_written` — sink success for apply/break-glass (apply only)
|
|
||||||
* `restart_audit.incident_result` — materialization outcome when an incident was required
|
|
||||||
* `incident` — durable descriptor (when gate requires follow-up)
|
|
||||||
|
|
||||||
## Security
|
|
||||||
|
|
||||||
* No secrets in audit payloads or issue bodies.
|
|
||||||
* This module never restarts a process.
|
|
||||||
* Drain proof verification remains #661; audit only records the decision.
|
|
||||||
* Incident creation failures are recorded in `incident_result.reasons` and never
|
|
||||||
crash the restart evaluation path (audit write failure still fails closed for
|
|
||||||
privileged apply when the sink is enabled).
|
|
||||||
|
|
||||||
## Tests
|
|
||||||
|
|
||||||
See `tests/test_restart_audit.py`: schema, redaction, emission, deny policy,
|
|
||||||
incident materialization mocks, break-glass / failed-drain selection.
|
|
||||||
@@ -33,7 +33,6 @@ recovery behavior for all nine classes.
|
|||||||
| `ControlPlaneDB.list_sessions` | `control_plane_db.py` | Read-only session inventory (the process-level unit a restart kills). |
|
| `ControlPlaneDB.list_sessions` | `control_plane_db.py` | Read-only session inventory (the process-level unit a restart kills). |
|
||||||
| `gitea_request_mcp_restart` | `gitea_mcp_server.py` | MCP tool: gathers inventory from the #613 DB, calls the coordinator, returns the report, and on `dry_run=False` runs the #661 drain-proof hard gate. Never restarts a process. |
|
| `gitea_request_mcp_restart` | `gitea_mcp_server.py` | MCP tool: gathers inventory from the #613 DB, calls the coordinator, returns the report, and on `dry_run=False` runs the #661 drain-proof hard gate. Never restarts a process. |
|
||||||
| `drain_proof.gate_apply_restart` | `drain_proof.py` | The #661 hard gate: verifies a drain proof against the current impact fingerprint, or records an authorized break-glass bypass. |
|
| `drain_proof.gate_apply_restart` | `drain_proof.py` | The #661 hard gate: verifies a drain proof against the current impact fingerprint, or records an authorized break-glass bypass. |
|
||||||
| `restart_audit` | `restart_audit.py` | #665 forensic trail: `mcp.restart.*` events via `gitea_audit`, correlation ids, durable incidents for failed drain / break-glass. See [`mcp-restart-audit.md`](./mcp-restart-audit.md). |
|
|
||||||
|
|
||||||
## Dimensions evaluated
|
## Dimensions evaluated
|
||||||
|
|
||||||
|
|||||||
+119
-114
@@ -839,6 +839,75 @@ def _invalidate_preflight_identity_state() -> None:
|
|||||||
_clear_preflight_capability_state()
|
_clear_preflight_capability_state()
|
||||||
|
|
||||||
|
|
||||||
|
# #690: session-boundary invalidation record for the most recent cross-profile
|
||||||
|
# activation. Surfaced in runtime diagnostics so a formal review run can prove
|
||||||
|
# its state was reset by a profile switch and must be fully re-established.
|
||||||
|
_PROFILE_SWITCH_INVALIDATION: dict | None = None
|
||||||
|
|
||||||
|
|
||||||
|
def _invalidate_review_state_on_profile_switch(
|
||||||
|
before_profile: str,
|
||||||
|
after_profile: str,
|
||||||
|
) -> dict:
|
||||||
|
"""Invalidate review-critical session state on a profile switch (#690).
|
||||||
|
|
||||||
|
Workflow-load proof, reviewer lease binding, the review decision lock,
|
||||||
|
live namespace health, and preflight identity/capability stamps recorded
|
||||||
|
under the prior profile are contaminated for the new role. Durable state
|
||||||
|
keyed by *either* profile identity is cleared so a reviewer → author →
|
||||||
|
reviewer hop cannot resurrect a stale review run: the full reviewer
|
||||||
|
preflight (gitea_whoami, gitea_load_review_workflow,
|
||||||
|
gitea_resolve_task_capability(review_pr), live head re-pin, lease
|
||||||
|
re-acquire/adopt) must be re-established before any formal verdict.
|
||||||
|
"""
|
||||||
|
global _PROFILE_SWITCH_INVALIDATION
|
||||||
|
invalidated: list[str] = []
|
||||||
|
|
||||||
|
_invalidate_preflight_identity_state()
|
||||||
|
invalidated.append("preflight_identity_capability")
|
||||||
|
|
||||||
|
review_workflow_load.clear_review_workflow_load()
|
||||||
|
invalidated.append("review_workflow_load")
|
||||||
|
|
||||||
|
_save_review_decision_lock(None)
|
||||||
|
invalidated.append("review_decision_lock")
|
||||||
|
|
||||||
|
reviewer_pr_lease.clear_session_lease()
|
||||||
|
invalidated.append("reviewer_session_lease")
|
||||||
|
|
||||||
|
if _LIVE_NAMESPACE_HEALTH:
|
||||||
|
_LIVE_NAMESPACE_HEALTH.clear()
|
||||||
|
invalidated.append("live_namespace_health")
|
||||||
|
|
||||||
|
# Durable records keyed by either profile identity must not survive the
|
||||||
|
# switch, or activating author → reviewer → author could revive a stale
|
||||||
|
# review run without re-preflight.
|
||||||
|
for identity in {before_profile, after_profile}:
|
||||||
|
if not identity:
|
||||||
|
continue
|
||||||
|
try:
|
||||||
|
mcp_session_state.clear_state(
|
||||||
|
kind=mcp_session_state.KIND_DECISION_LOCK,
|
||||||
|
profile_identity=identity,
|
||||||
|
)
|
||||||
|
mcp_session_state.clear_state(
|
||||||
|
kind=mcp_session_state.KIND_WORKFLOW_LOAD,
|
||||||
|
profile_identity=identity,
|
||||||
|
)
|
||||||
|
except Exception:
|
||||||
|
pass # best-effort durable cleanup; in-memory state already reset
|
||||||
|
invalidated.append("durable_profile_state")
|
||||||
|
|
||||||
|
_PROFILE_SWITCH_INVALIDATION = {
|
||||||
|
"from_profile": before_profile,
|
||||||
|
"to_profile": after_profile,
|
||||||
|
"invalidated": invalidated,
|
||||||
|
"invalidated_at": datetime.now(timezone.utc).isoformat(),
|
||||||
|
"re_preflight_required": True,
|
||||||
|
}
|
||||||
|
return dict(_PROFILE_SWITCH_INVALIDATION)
|
||||||
|
|
||||||
|
|
||||||
def record_preflight_check(
|
def record_preflight_check(
|
||||||
type_name: str,
|
type_name: str,
|
||||||
resolved_role: str | None = None,
|
resolved_role: str | None = None,
|
||||||
@@ -2100,7 +2169,6 @@ import lease_policy # noqa: E402
|
|||||||
import workflow_dashboard # noqa: E402 # #605 live queue/lease dashboard
|
import workflow_dashboard # noqa: E402 # #605 live queue/lease dashboard
|
||||||
import restart_coordinator # noqa: E402 # #658 MCP restart coordinator/impact
|
import restart_coordinator # noqa: E402 # #658 MCP restart coordinator/impact
|
||||||
import drain_proof # noqa: E402 # #661 pre-restart drain proof and hard gate
|
import drain_proof # noqa: E402 # #661 pre-restart drain proof and hard gate
|
||||||
import restart_audit # noqa: E402 # #665 restart audit events + incidents
|
|
||||||
import incident_bridge # noqa: E402
|
import incident_bridge # noqa: E402
|
||||||
import sentry_observability # noqa: E402 (#606 optional Sentry observability)
|
import sentry_observability # noqa: E402 (#606 optional Sentry observability)
|
||||||
import sentry_incident_bridge # noqa: E402 (#607 Sentry→Gitea incident bridge)
|
import sentry_incident_bridge # noqa: E402 (#607 Sentry→Gitea incident bridge)
|
||||||
@@ -18109,6 +18177,11 @@ def gitea_whoami(
|
|||||||
"session_context_audit": session_ctx.mutation_context_audit_fields(),
|
"session_context_audit": session_ctx.mutation_context_audit_fields(),
|
||||||
"identity_match": not id_match.get("block"),
|
"identity_match": not id_match.get("block"),
|
||||||
"identity_match_reasons": id_match.get("reasons") or [],
|
"identity_match_reasons": id_match.get("reasons") or [],
|
||||||
|
# #690 AC4: report launcher-declared client namespace alongside the
|
||||||
|
# active execution profile so drift is visible in diagnostics.
|
||||||
|
"namespace_provenance": mcp_namespace_health.namespace_provenance(
|
||||||
|
active_profile=profile["profile_name"]
|
||||||
|
),
|
||||||
}
|
}
|
||||||
if id_match.get("block"):
|
if id_match.get("block"):
|
||||||
_invalidate_preflight_identity_state()
|
_invalidate_preflight_identity_state()
|
||||||
@@ -19013,6 +19086,11 @@ def gitea_get_runtime_context(
|
|||||||
"shell_health": native_mcp_preference.shell_health_status(),
|
"shell_health": native_mcp_preference.shell_health_status(),
|
||||||
"workflow_load_proof": review_workflow_load.workflow_load_status(
|
"workflow_load_proof": review_workflow_load.workflow_load_status(
|
||||||
PROJECT_ROOT),
|
PROJECT_ROOT),
|
||||||
|
# #690: namespace provenance + profile-switch invalidation evidence.
|
||||||
|
"namespace_provenance": mcp_namespace_health.namespace_provenance(
|
||||||
|
active_profile=profile["profile_name"]
|
||||||
|
),
|
||||||
|
"profile_switch_invalidation": _PROFILE_SWITCH_INVALIDATION,
|
||||||
}
|
}
|
||||||
|
|
||||||
if not is_client_managed:
|
if not is_client_managed:
|
||||||
@@ -19523,6 +19601,17 @@ def gitea_activate_profile(
|
|||||||
source="gitea_activate_profile",
|
source="gitea_activate_profile",
|
||||||
)
|
)
|
||||||
|
|
||||||
|
# 4.7 #690: a profile switch is a session-boundary event for review state.
|
||||||
|
# Any workflow-load proof, reviewer lease, decision lock, namespace
|
||||||
|
# health, or preflight stamp recorded under the prior profile is
|
||||||
|
# contaminated for the new role and must be re-established under the new
|
||||||
|
# profile before any formal review verdict.
|
||||||
|
switch_invalidation = None
|
||||||
|
if before_profile != after_profile:
|
||||||
|
switch_invalidation = _invalidate_review_state_on_profile_switch(
|
||||||
|
before_profile, after_profile
|
||||||
|
)
|
||||||
|
|
||||||
# 5. Audit the switch if auditing is on
|
# 5. Audit the switch if auditing is on
|
||||||
_audit(
|
_audit(
|
||||||
"activate_profile",
|
"activate_profile",
|
||||||
@@ -19533,11 +19622,12 @@ def gitea_activate_profile(
|
|||||||
"before": before_profile,
|
"before": before_profile,
|
||||||
"after": after_profile,
|
"after": after_profile,
|
||||||
"session_context": session_ctx.mutation_context_audit_fields(),
|
"session_context": session_ctx.mutation_context_audit_fields(),
|
||||||
|
"review_state_invalidated": bool(switch_invalidation),
|
||||||
},
|
},
|
||||||
username=after_identity,
|
username=after_identity,
|
||||||
)
|
)
|
||||||
|
|
||||||
return {
|
result = {
|
||||||
"success": True,
|
"success": True,
|
||||||
"message": f"Successfully activated profile '{profile_name}' (fresh identity verification complete).",
|
"message": f"Successfully activated profile '{profile_name}' (fresh identity verification complete).",
|
||||||
"before_profile": before_profile,
|
"before_profile": before_profile,
|
||||||
@@ -19547,6 +19637,18 @@ def gitea_activate_profile(
|
|||||||
"session_context_audit": session_ctx.mutation_context_audit_fields(),
|
"session_context_audit": session_ctx.mutation_context_audit_fields(),
|
||||||
"auto_profile_substitution": False,
|
"auto_profile_substitution": False,
|
||||||
}
|
}
|
||||||
|
if switch_invalidation is not None:
|
||||||
|
result["review_state_invalidation"] = switch_invalidation
|
||||||
|
result["re_preflight_required"] = True
|
||||||
|
result["exact_next_action"] = (
|
||||||
|
"Profile switch invalidated workflow-load proof, reviewer lease, "
|
||||||
|
"decision lock, and preflight stamps (#690). Before any formal "
|
||||||
|
"review verdict, re-run the full reviewer preflight: "
|
||||||
|
"gitea_whoami, gitea_load_review_workflow, "
|
||||||
|
"gitea_resolve_task_capability(review_pr), live head re-pin, and "
|
||||||
|
"lease re-acquire/adopt."
|
||||||
|
)
|
||||||
|
return result
|
||||||
|
|
||||||
|
|
||||||
@mcp.tool()
|
@mcp.tool()
|
||||||
@@ -21836,12 +21938,22 @@ def gitea_resolve_task_capability(
|
|||||||
f"{required_role} task '{task}' even if nearby permissions are "
|
f"{required_role} task '{task}' even if nearby permissions are "
|
||||||
"present (fail closed)."
|
"present (fail closed)."
|
||||||
)
|
)
|
||||||
|
# #690 AC4: when the launcher declares a client namespace, a task with a
|
||||||
|
# required namespace must fail closed on mismatch (e.g. review_pr served
|
||||||
|
# from an author namespace).
|
||||||
|
ns_provenance = mcp_namespace_health.namespace_provenance(
|
||||||
|
task=task_key, active_profile=profile.get("profile_name")
|
||||||
|
)
|
||||||
|
ns_mismatch_reason = None
|
||||||
|
if ns_provenance.get("mismatch"):
|
||||||
|
ns_mismatch_reason = "; ".join(ns_provenance.get("reasons") or [])
|
||||||
cross_host_block = bool(remote_assess.get("block"))
|
cross_host_block = bool(remote_assess.get("block"))
|
||||||
identity_block = bool(id_assess.get("block"))
|
identity_block = bool(id_assess.get("block"))
|
||||||
drift_block = bool(ctx_assess.get("block"))
|
drift_block = bool(ctx_assess.get("block"))
|
||||||
allowed_in_current_session = (
|
allowed_in_current_session = (
|
||||||
permission_allowed_in_current_session
|
permission_allowed_in_current_session
|
||||||
and role_matches_current_session
|
and role_matches_current_session
|
||||||
|
and not ns_provenance.get("mismatch")
|
||||||
and not cross_host_block
|
and not cross_host_block
|
||||||
and not identity_block
|
and not identity_block
|
||||||
and not drift_block
|
and not drift_block
|
||||||
@@ -21892,6 +22004,8 @@ def gitea_resolve_task_capability(
|
|||||||
)
|
)
|
||||||
if role_mismatch_reason:
|
if role_mismatch_reason:
|
||||||
deny_parts.append(role_mismatch_reason)
|
deny_parts.append(role_mismatch_reason)
|
||||||
|
if ns_mismatch_reason:
|
||||||
|
deny_parts.append(ns_mismatch_reason)
|
||||||
if deny_parts:
|
if deny_parts:
|
||||||
reason_msg = "; ".join(deny_parts)
|
reason_msg = "; ".join(deny_parts)
|
||||||
elif configured and switching:
|
elif configured and switching:
|
||||||
@@ -21974,6 +22088,8 @@ def gitea_resolve_task_capability(
|
|||||||
task_role_guidance = []
|
task_role_guidance = []
|
||||||
if role_mismatch_reason:
|
if role_mismatch_reason:
|
||||||
task_role_guidance.append(f"STOP: {role_mismatch_reason}")
|
task_role_guidance.append(f"STOP: {role_mismatch_reason}")
|
||||||
|
if ns_mismatch_reason:
|
||||||
|
task_role_guidance.append(f"STOP: {ns_mismatch_reason}")
|
||||||
if required_role == "reviewer":
|
if required_role == "reviewer":
|
||||||
if allowed_in_current_session:
|
if allowed_in_current_session:
|
||||||
task_role_guidance.append(
|
task_role_guidance.append(
|
||||||
@@ -22040,6 +22156,7 @@ def gitea_resolve_task_capability(
|
|||||||
"session_context_audit": session_ctx.mutation_context_audit_fields(),
|
"session_context_audit": session_ctx.mutation_context_audit_fields(),
|
||||||
"profile_remote_compatible": not cross_host_block,
|
"profile_remote_compatible": not cross_host_block,
|
||||||
"identity_match": not identity_block,
|
"identity_match": not identity_block,
|
||||||
|
"namespace_provenance": ns_provenance,
|
||||||
"auto_profile_substitution": False,
|
"auto_profile_substitution": False,
|
||||||
}
|
}
|
||||||
# #685: report typed reconnect blocker without mutating config or exiting.
|
# #685: report typed reconnect blocker without mutating config or exiting.
|
||||||
@@ -23868,38 +23985,6 @@ def gitea_request_mcp_restart(
|
|||||||
# and a durable incident is raised. Break-glass is the only bypass and its
|
# and a durable incident is raised. Break-glass is the only bypass and its
|
||||||
# authorization is read from the environment, never self-asserted.
|
# authorization is read from the environment, never self-asserted.
|
||||||
payload["apply_supported"] = False
|
payload["apply_supported"] = False
|
||||||
# #665: correlation id threads impact preview → apply gate → incidents.
|
|
||||||
correlation_id = restart_audit.new_correlation_id()
|
|
||||||
payload["correlation_id"] = correlation_id
|
|
||||||
auth_user = None
|
|
||||||
try:
|
|
||||||
# remote-only: host override is for operator diagnostics, not required here
|
|
||||||
auth_user = (gitea_whoami(remote=remote) or {}).get("username")
|
|
||||||
except Exception: # noqa: BLE001 — identity is best-effort for audit
|
|
||||||
auth_user = None
|
|
||||||
preview_audit = restart_audit.record_restart_lifecycle(
|
|
||||||
event_type=restart_audit.EVENT_IMPACT_PREVIEW,
|
|
||||||
outcome=str(report.verdict or "unknown"),
|
|
||||||
correlation_id=correlation_id,
|
|
||||||
remote=remote,
|
|
||||||
org=o,
|
|
||||||
repo=r,
|
|
||||||
requesting_session_id=sid,
|
|
||||||
restart_class=restart_class,
|
|
||||||
profile_name=profile_name,
|
|
||||||
authenticated_username=auth_user,
|
|
||||||
reasons=list(report.reasons or []),
|
|
||||||
details={
|
|
||||||
"dry_run": True,
|
|
||||||
"allow_restart": bool(report.allow_restart),
|
|
||||||
"inventory_complete": inventory_complete,
|
|
||||||
},
|
|
||||||
privileged=False,
|
|
||||||
)
|
|
||||||
payload["restart_audit"] = {
|
|
||||||
"correlation_id": correlation_id,
|
|
||||||
"impact_preview_written": preview_audit["audit_written"],
|
|
||||||
}
|
|
||||||
if not dry_run:
|
if not dry_run:
|
||||||
proof_obj: dict | None = None
|
proof_obj: dict | None = None
|
||||||
proof_parse_error: str | None = None
|
proof_parse_error: str | None = None
|
||||||
@@ -23952,86 +24037,6 @@ def gitea_request_mcp_restart(
|
|||||||
)
|
)
|
||||||
if not gate.allow and gate.incident is not None:
|
if not gate.allow and gate.incident is not None:
|
||||||
payload["incident"] = gate.incident
|
payload["incident"] = gate.incident
|
||||||
|
|
||||||
# #665: audit apply-gate + materialize durable incidents for failed
|
|
||||||
# drain and break-glass. Privileged apply denies if audit is enabled
|
|
||||||
# and the sink write fails.
|
|
||||||
incident_desc = restart_audit.incident_from_apply_gate(
|
|
||||||
gate_payload={
|
|
||||||
**gate_payload,
|
|
||||||
"incident": gate.incident,
|
|
||||||
"allow": gate.allow,
|
|
||||||
},
|
|
||||||
break_glass=break_glass,
|
|
||||||
correlation_id=correlation_id,
|
|
||||||
requesting_session_id=sid,
|
|
||||||
restart_class=restart_class,
|
|
||||||
remote=remote,
|
|
||||||
org=o,
|
|
||||||
repo=r,
|
|
||||||
)
|
|
||||||
if incident_desc is not None:
|
|
||||||
payload["incident"] = incident_desc
|
|
||||||
|
|
||||||
def _create_restart_incident_issue(
|
|
||||||
*, title, body, labels, org=None, repo=None, **_kw
|
|
||||||
):
|
|
||||||
return gitea_create_issue(
|
|
||||||
title=title,
|
|
||||||
body=body,
|
|
||||||
labels=labels,
|
|
||||||
remote=remote,
|
|
||||||
host=h,
|
|
||||||
org=org or o,
|
|
||||||
repo=repo or r,
|
|
||||||
)
|
|
||||||
|
|
||||||
apply_audit = restart_audit.record_restart_lifecycle(
|
|
||||||
event_type=(
|
|
||||||
restart_audit.EVENT_BREAK_GLASS
|
|
||||||
if break_glass
|
|
||||||
else restart_audit.EVENT_APPLY_GATE
|
|
||||||
),
|
|
||||||
outcome=(
|
|
||||||
"break_glass"
|
|
||||||
if break_glass
|
|
||||||
else ("allow" if payload["apply_authorized"] else "deny")
|
|
||||||
),
|
|
||||||
correlation_id=correlation_id,
|
|
||||||
remote=remote,
|
|
||||||
org=o,
|
|
||||||
repo=r,
|
|
||||||
requesting_session_id=sid,
|
|
||||||
restart_class=restart_class,
|
|
||||||
profile_name=profile_name,
|
|
||||||
authenticated_username=auth_user,
|
|
||||||
reasons=list(gate_payload.get("reasons") or []),
|
|
||||||
details={
|
|
||||||
"apply_authorized": payload["apply_authorized"],
|
|
||||||
"drain_gate_allow": gate_payload.get("drain_gate_allow"),
|
|
||||||
"restart_class_authorized": restart_class_authorized,
|
|
||||||
"break_glass": break_glass,
|
|
||||||
"proof_id": gate_payload.get("proof_id"),
|
|
||||||
},
|
|
||||||
privileged=True,
|
|
||||||
create_incident=incident_desc,
|
|
||||||
create_issue_fn=_create_restart_incident_issue
|
|
||||||
if incident_desc is not None
|
|
||||||
else None,
|
|
||||||
dry_run_incident=False,
|
|
||||||
)
|
|
||||||
payload["restart_audit"] = {
|
|
||||||
"correlation_id": correlation_id,
|
|
||||||
"impact_preview_written": preview_audit["audit_written"],
|
|
||||||
"apply_gate_written": apply_audit["audit_written"],
|
|
||||||
"incident_result": apply_audit.get("incident_result"),
|
|
||||||
}
|
|
||||||
if apply_audit["deny_reasons"]:
|
|
||||||
payload["apply_authorized"] = False
|
|
||||||
payload["reasons"] = list(payload.get("reasons") or []) + list(
|
|
||||||
apply_audit["deny_reasons"]
|
|
||||||
)
|
|
||||||
payload["success"] = True
|
|
||||||
return payload
|
return payload
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
@@ -16,6 +16,7 @@ Probe sources
|
|||||||
|
|
||||||
from __future__ import annotations
|
from __future__ import annotations
|
||||||
|
|
||||||
|
import os
|
||||||
from typing import Any
|
from typing import Any
|
||||||
|
|
||||||
|
|
||||||
@@ -57,8 +58,56 @@ SAFE_ENV_KEYS = (
|
|||||||
"GITEA_SERVICE",
|
"GITEA_SERVICE",
|
||||||
"GITEA_EXECUTION_ROLE",
|
"GITEA_EXECUTION_ROLE",
|
||||||
"GITEA_MCP_CONFIG",
|
"GITEA_MCP_CONFIG",
|
||||||
|
"GITEA_MCP_NAMESPACE",
|
||||||
)
|
)
|
||||||
|
|
||||||
|
# Optional launcher-provided env declaring the client-managed MCP namespace
|
||||||
|
# this process is registered under (e.g. ``gitea-reviewer``). The server
|
||||||
|
# cannot derive its own IDE namespace name, so the launcher declares it; when
|
||||||
|
# declared, reviewers/mergers can fail closed on a namespace/task mismatch
|
||||||
|
# (#690 AC4). Absence means "unknown" — reported, never guessed.
|
||||||
|
NAMESPACE_ENV = "GITEA_MCP_NAMESPACE"
|
||||||
|
|
||||||
|
|
||||||
|
def configured_client_namespace(env: dict[str, str] | None = None) -> str | None:
|
||||||
|
"""Return the launcher-declared client namespace, or None when unknown."""
|
||||||
|
source = os.environ if env is None else env
|
||||||
|
value = (source.get(NAMESPACE_ENV) or "").strip()
|
||||||
|
return value or None
|
||||||
|
|
||||||
|
|
||||||
|
def namespace_provenance(
|
||||||
|
task: str | None = None,
|
||||||
|
*,
|
||||||
|
active_profile: str | None = None,
|
||||||
|
env: dict[str, str] | None = None,
|
||||||
|
) -> dict[str, Any]:
|
||||||
|
"""Report configured client namespace vs active execution profile (#690).
|
||||||
|
|
||||||
|
When *task* carries a required namespace (``TASK_REQUIRED_NAMESPACES``)
|
||||||
|
and the launcher declared a different one, ``mismatch`` is True and the
|
||||||
|
caller must fail closed for that task. An undeclared namespace is
|
||||||
|
reported as unknown — never treated as proof either way.
|
||||||
|
"""
|
||||||
|
configured = configured_client_namespace(env)
|
||||||
|
required = TASK_REQUIRED_NAMESPACES.get(task or "")
|
||||||
|
mismatch = bool(configured and required and configured != required)
|
||||||
|
reasons: list[str] = []
|
||||||
|
if mismatch:
|
||||||
|
reasons.append(
|
||||||
|
f"configured client namespace '{configured}' does not match "
|
||||||
|
f"required namespace '{required}' for task '{task}' (fail closed)"
|
||||||
|
)
|
||||||
|
return {
|
||||||
|
"configured_namespace": configured,
|
||||||
|
"namespace_source": NAMESPACE_ENV if configured else "unknown",
|
||||||
|
"active_profile": active_profile,
|
||||||
|
"requested_task": task,
|
||||||
|
"required_namespace": required,
|
||||||
|
"mismatch": mismatch,
|
||||||
|
"reasons": reasons,
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
def _as_list(value: Any) -> list[str] | None:
|
def _as_list(value: Any) -> list[str] | None:
|
||||||
if value is None:
|
if value is None:
|
||||||
|
|||||||
@@ -1,426 +0,0 @@
|
|||||||
"""MCP restart lifecycle audit events and incident materialization (#665).
|
|
||||||
|
|
||||||
Restarts and recovery attempts must leave a forensic trail: impact previews,
|
|
||||||
drain enter/exit, drain-proof results, apply gate verdicts, break-glass, and
|
|
||||||
post-restart reconcile outcomes. Failed drains and break-glass must also raise
|
|
||||||
durable Gitea incident issues so they cannot be silently repeated.
|
|
||||||
|
|
||||||
This module is the pure + sink layer for that trail:
|
|
||||||
|
|
||||||
* **Schema** — ``mcp.restart.*`` event names and a redacted payload builder.
|
|
||||||
* **Emission** — append-only via :mod:`gitea_audit` (off when ``GITEA_AUDIT_LOG``
|
|
||||||
is unset; privileged apply can still *require* a successful write).
|
|
||||||
* **Incidents** — descriptors for failed drain / break-glass / unguarded restart,
|
|
||||||
plus an optional materializer that creates a Gitea issue through an injected
|
|
||||||
``create_issue_fn`` (keeps this module free of network I/O in tests).
|
|
||||||
|
|
||||||
Design rules:
|
|
||||||
|
|
||||||
* **No secrets.** All free text is redacted before write or issue body assembly.
|
|
||||||
* **Never raises from emission.** ``emit_restart_event`` returns False on sink
|
|
||||||
failure so callers can decide fail-closed policy for privileged restarts.
|
|
||||||
* **Does not restart.** Audit never executes a process restart.
|
|
||||||
* **Drain proof stays #661.** This module records what the gate decided; it
|
|
||||||
does not re-verify proofs.
|
|
||||||
"""
|
|
||||||
|
|
||||||
from __future__ import annotations
|
|
||||||
|
|
||||||
import uuid
|
|
||||||
from datetime import datetime, timezone
|
|
||||||
from typing import Any, Callable, Mapping, Sequence
|
|
||||||
|
|
||||||
import gitea_audit
|
|
||||||
|
|
||||||
# ── Event vocabulary (stable identifiers for operators + tests) ───────────────
|
|
||||||
|
|
||||||
EVENT_IMPACT_PREVIEW = "mcp.restart.impact_preview"
|
|
||||||
EVENT_DRAIN_ENTER = "mcp.restart.drain_enter"
|
|
||||||
EVENT_DRAIN_EXIT = "mcp.restart.drain_exit"
|
|
||||||
EVENT_DRAIN_PROOF = "mcp.restart.drain_proof"
|
|
||||||
EVENT_APPLY_GATE = "mcp.restart.apply_gate"
|
|
||||||
EVENT_BREAK_GLASS = "mcp.restart.break_glass"
|
|
||||||
EVENT_POST_RESTART_RECONCILE = "mcp.restart.post_restart_reconcile"
|
|
||||||
EVENT_NARROWER_RECOVERY = "mcp.restart.narrower_recovery"
|
|
||||||
EVENT_UNGUARDED_DETECTED = "mcp.restart.unguarded_detected"
|
|
||||||
|
|
||||||
RESTART_EVENT_TYPES: frozenset[str] = frozenset(
|
|
||||||
{
|
|
||||||
EVENT_IMPACT_PREVIEW,
|
|
||||||
EVENT_DRAIN_ENTER,
|
|
||||||
EVENT_DRAIN_EXIT,
|
|
||||||
EVENT_DRAIN_PROOF,
|
|
||||||
EVENT_APPLY_GATE,
|
|
||||||
EVENT_BREAK_GLASS,
|
|
||||||
EVENT_POST_RESTART_RECONCILE,
|
|
||||||
EVENT_NARROWER_RECOVERY,
|
|
||||||
EVENT_UNGUARDED_DETECTED,
|
|
||||||
}
|
|
||||||
)
|
|
||||||
|
|
||||||
# Incident kinds (durable Gitea issues).
|
|
||||||
INCIDENT_FAILED_DRAIN = "restart_failed_drain"
|
|
||||||
INCIDENT_BREAK_GLASS = "restart_break_glass"
|
|
||||||
INCIDENT_RECONCILE_UNRESOLVED = "restart_reconcile_unresolved"
|
|
||||||
INCIDENT_UNGUARDED = "restart_unguarded_detected"
|
|
||||||
|
|
||||||
DEFAULT_INCIDENT_LABELS: tuple[str, ...] = (
|
|
||||||
"mcp-health",
|
|
||||||
"safety",
|
|
||||||
"observability",
|
|
||||||
"status:ready",
|
|
||||||
"type:bug",
|
|
||||||
"workflow-hardening",
|
|
||||||
)
|
|
||||||
|
|
||||||
CreateIssueFn = Callable[..., dict[str, Any]]
|
|
||||||
|
|
||||||
|
|
||||||
def _utc_now_iso() -> str:
|
|
||||||
return datetime.now(timezone.utc).isoformat()
|
|
||||||
|
|
||||||
|
|
||||||
def new_correlation_id() -> str:
|
|
||||||
"""Mint a short correlation id shared across a restart lifecycle."""
|
|
||||||
return f"rst-{uuid.uuid4().hex[:16]}"
|
|
||||||
|
|
||||||
|
|
||||||
def build_restart_event(
|
|
||||||
*,
|
|
||||||
event_type: str,
|
|
||||||
outcome: str,
|
|
||||||
correlation_id: str | None = None,
|
|
||||||
remote: str | None = None,
|
|
||||||
org: str | None = None,
|
|
||||||
repo: str | None = None,
|
|
||||||
requesting_session_id: str | None = None,
|
|
||||||
restart_class: str | None = None,
|
|
||||||
profile_name: str | None = None,
|
|
||||||
authenticated_username: str | None = None,
|
|
||||||
reasons: Sequence[str] | None = None,
|
|
||||||
details: Mapping[str, Any] | None = None,
|
|
||||||
now: str | None = None,
|
|
||||||
) -> dict[str, Any]:
|
|
||||||
"""Build a redacted ``mcp.restart.*`` audit event.
|
|
||||||
|
|
||||||
Raises ``ValueError`` on unknown event types so a typo cannot silently land
|
|
||||||
under a free-form action name.
|
|
||||||
"""
|
|
||||||
name = str(event_type or "").strip()
|
|
||||||
if name not in RESTART_EVENT_TYPES:
|
|
||||||
raise ValueError(
|
|
||||||
f"unknown restart audit event_type {name!r}; expected one of "
|
|
||||||
f"{sorted(RESTART_EVENT_TYPES)}"
|
|
||||||
)
|
|
||||||
redacted_reasons = [
|
|
||||||
gitea_audit.redact(str(r)) for r in (reasons or []) if str(r).strip()
|
|
||||||
]
|
|
||||||
redacted_details = gitea_audit.redact(dict(details or {}))
|
|
||||||
if not isinstance(redacted_details, dict):
|
|
||||||
redacted_details = {"value": redacted_details}
|
|
||||||
|
|
||||||
event = gitea_audit.build_event(
|
|
||||||
action=name,
|
|
||||||
result=str(outcome or "unknown"),
|
|
||||||
remote=remote,
|
|
||||||
repository=f"{org}/{repo}" if org and repo else None,
|
|
||||||
profile_name=profile_name,
|
|
||||||
authenticated_username=authenticated_username,
|
|
||||||
reason="; ".join(redacted_reasons) if redacted_reasons else None,
|
|
||||||
request_metadata={
|
|
||||||
"event_family": "mcp.restart",
|
|
||||||
"correlation_id": correlation_id or new_correlation_id(),
|
|
||||||
"restart_class": restart_class,
|
|
||||||
"requesting_session_id": requesting_session_id,
|
|
||||||
"org": org,
|
|
||||||
"repo": repo,
|
|
||||||
"details": redacted_details,
|
|
||||||
"reasons": redacted_reasons,
|
|
||||||
},
|
|
||||||
now=now or _utc_now_iso(),
|
|
||||||
operation=name,
|
|
||||||
)
|
|
||||||
event["action_type"] = "restart_lifecycle"
|
|
||||||
event["event_type"] = name
|
|
||||||
event["correlation_id"] = (event.get("request_metadata") or {}).get(
|
|
||||||
"correlation_id"
|
|
||||||
)
|
|
||||||
return event
|
|
||||||
|
|
||||||
|
|
||||||
def emit_restart_event(event: Mapping[str, Any], *, path: str | None = None) -> bool:
|
|
||||||
"""Append *event* to the audit sink. Never raises. Returns write success."""
|
|
||||||
try:
|
|
||||||
return bool(gitea_audit.write_event(dict(event), path=path))
|
|
||||||
except Exception:
|
|
||||||
return False
|
|
||||||
|
|
||||||
|
|
||||||
def require_audit_or_deny(
|
|
||||||
*,
|
|
||||||
privileged: bool,
|
|
||||||
written: bool,
|
|
||||||
audit_enabled: bool | None = None,
|
|
||||||
) -> list[str]:
|
|
||||||
"""Return deny reasons when a privileged restart path fails to audit.
|
|
||||||
|
|
||||||
When audit is not configured (``GITEA_AUDIT_LOG`` unset), privileged apply
|
|
||||||
still proceeds under the rollout policy "enable audit before enforcing
|
|
||||||
deny-on-audit-fail" — but *if* audit is enabled and the write fails,
|
|
||||||
privileged apply is denied (fail closed).
|
|
||||||
"""
|
|
||||||
enabled = (
|
|
||||||
gitea_audit.audit_enabled() if audit_enabled is None else bool(audit_enabled)
|
|
||||||
)
|
|
||||||
if not privileged:
|
|
||||||
return []
|
|
||||||
if not enabled:
|
|
||||||
return []
|
|
||||||
if written:
|
|
||||||
return []
|
|
||||||
return [
|
|
||||||
"privileged restart path requires a successful audit write; "
|
|
||||||
"audit sink failed (fail closed, #665)"
|
|
||||||
]
|
|
||||||
|
|
||||||
|
|
||||||
# ── Incident descriptors ──────────────────────────────────────────────────────
|
|
||||||
|
|
||||||
|
|
||||||
def build_incident_descriptor(
|
|
||||||
*,
|
|
||||||
kind: str,
|
|
||||||
reasons: Sequence[str],
|
|
||||||
correlation_id: str | None = None,
|
|
||||||
requesting_session_id: str | None = None,
|
|
||||||
restart_class: str | None = None,
|
|
||||||
remote: str | None = None,
|
|
||||||
org: str | None = None,
|
|
||||||
repo: str | None = None,
|
|
||||||
proof_id: str | None = None,
|
|
||||||
at: str | None = None,
|
|
||||||
) -> dict[str, Any]:
|
|
||||||
"""Build a durable incident descriptor (no network)."""
|
|
||||||
titles = {
|
|
||||||
INCIDENT_FAILED_DRAIN: "Restart denied: drain proof failed the hard gate",
|
|
||||||
INCIDENT_BREAK_GLASS: "Break-glass MCP restart authorized",
|
|
||||||
INCIDENT_RECONCILE_UNRESOLVED: "Post-restart reconcile left unresolved work",
|
|
||||||
INCIDENT_UNGUARDED: "Unguarded MCP restart attempt detected",
|
|
||||||
}
|
|
||||||
title = titles.get(kind, f"MCP restart incident ({kind})")
|
|
||||||
redacted_reasons = [
|
|
||||||
gitea_audit.redact(str(r)) for r in reasons if str(r).strip()
|
|
||||||
]
|
|
||||||
return {
|
|
||||||
"kind": kind,
|
|
||||||
"title": title,
|
|
||||||
"labels": list(DEFAULT_INCIDENT_LABELS),
|
|
||||||
"reasons": redacted_reasons,
|
|
||||||
"correlation_id": correlation_id,
|
|
||||||
"requesting_session_id": requesting_session_id,
|
|
||||||
"restart_class": restart_class,
|
|
||||||
"remote": remote,
|
|
||||||
"org": org,
|
|
||||||
"repo": repo,
|
|
||||||
"proof_id": proof_id,
|
|
||||||
"at": at or _utc_now_iso(),
|
|
||||||
"source": "restart_audit#665",
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
def incident_body(descriptor: Mapping[str, Any]) -> str:
|
|
||||||
"""Render a redacted markdown body for a Gitea incident issue."""
|
|
||||||
reasons = descriptor.get("reasons") or []
|
|
||||||
reason_lines = "\n".join(f"- {gitea_audit.redact(str(r))}" for r in reasons) or (
|
|
||||||
"- (no reasons recorded)"
|
|
||||||
)
|
|
||||||
return "\n".join(
|
|
||||||
[
|
|
||||||
"<!-- mcp-restart-incident:v1 -->",
|
|
||||||
f"## MCP restart incident (`{descriptor.get('kind')}`)",
|
|
||||||
"",
|
|
||||||
f"**Correlation:** `{descriptor.get('correlation_id') or 'none'}`",
|
|
||||||
f"**Session:** `{descriptor.get('requesting_session_id') or 'none'}`",
|
|
||||||
f"**Class:** `{descriptor.get('restart_class') or 'none'}`",
|
|
||||||
f"**Scope:** `{descriptor.get('remote')}/{descriptor.get('org')}/"
|
|
||||||
f"{descriptor.get('repo')}`",
|
|
||||||
f"**At:** `{descriptor.get('at')}`",
|
|
||||||
f"**Proof id:** `{descriptor.get('proof_id') or 'none'}`",
|
|
||||||
"",
|
|
||||||
"### Reasons",
|
|
||||||
reason_lines,
|
|
||||||
"",
|
|
||||||
"### Operator next steps",
|
|
||||||
"- Treat this as durable follow-up work under the restart-governance umbrella (#655).",
|
|
||||||
"- Do not invent a second restart path; use sanctioned coordinator tools only.",
|
|
||||||
"- Raw secrets must never appear in this issue (already redacted).",
|
|
||||||
"",
|
|
||||||
f"_Source: {descriptor.get('source')}_",
|
|
||||||
]
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
def materialize_incident(
|
|
||||||
descriptor: Mapping[str, Any],
|
|
||||||
*,
|
|
||||||
create_issue_fn: CreateIssueFn | None,
|
|
||||||
dry_run: bool = False,
|
|
||||||
) -> dict[str, Any]:
|
|
||||||
"""Create a Gitea issue from *descriptor* when *create_issue_fn* is provided.
|
|
||||||
|
|
||||||
Returns a result dict with ``created`` / ``issue_number`` / ``dry_run`` /
|
|
||||||
``reasons``. Never raises.
|
|
||||||
"""
|
|
||||||
base: dict[str, Any] = {
|
|
||||||
"created": False,
|
|
||||||
"dry_run": bool(dry_run),
|
|
||||||
"issue_number": None,
|
|
||||||
"kind": descriptor.get("kind"),
|
|
||||||
"reasons": [],
|
|
||||||
"descriptor": dict(descriptor),
|
|
||||||
}
|
|
||||||
if dry_run:
|
|
||||||
base["reasons"] = ["dry-run only; no Gitea issue created"]
|
|
||||||
return base
|
|
||||||
if create_issue_fn is None:
|
|
||||||
base["reasons"] = [
|
|
||||||
"create_issue_fn not provided; incident descriptor retained only"
|
|
||||||
]
|
|
||||||
return base
|
|
||||||
try:
|
|
||||||
result = create_issue_fn(
|
|
||||||
title=str(descriptor.get("title") or "MCP restart incident"),
|
|
||||||
body=incident_body(descriptor),
|
|
||||||
labels=list(descriptor.get("labels") or DEFAULT_INCIDENT_LABELS),
|
|
||||||
org=descriptor.get("org"),
|
|
||||||
repo=descriptor.get("repo"),
|
|
||||||
)
|
|
||||||
number = None
|
|
||||||
if isinstance(result, dict):
|
|
||||||
number = result.get("number") or result.get("issue_number")
|
|
||||||
if number is not None:
|
|
||||||
base["created"] = True
|
|
||||||
base["issue_number"] = int(number)
|
|
||||||
base["reasons"] = [f"created incident issue #{int(number)}"]
|
|
||||||
else:
|
|
||||||
base["reasons"] = ["create_issue_fn returned no issue number"]
|
|
||||||
except Exception as exc: # noqa: BLE001 — never break restart path here
|
|
||||||
base["reasons"] = [
|
|
||||||
f"incident issue creation failed: {gitea_audit.redact(str(exc))}"
|
|
||||||
]
|
|
||||||
return base
|
|
||||||
|
|
||||||
|
|
||||||
def record_restart_lifecycle(
|
|
||||||
*,
|
|
||||||
event_type: str,
|
|
||||||
outcome: str,
|
|
||||||
correlation_id: str,
|
|
||||||
remote: str | None = None,
|
|
||||||
org: str | None = None,
|
|
||||||
repo: str | None = None,
|
|
||||||
requesting_session_id: str | None = None,
|
|
||||||
restart_class: str | None = None,
|
|
||||||
profile_name: str | None = None,
|
|
||||||
authenticated_username: str | None = None,
|
|
||||||
reasons: Sequence[str] | None = None,
|
|
||||||
details: Mapping[str, Any] | None = None,
|
|
||||||
privileged: bool = False,
|
|
||||||
create_incident: Mapping[str, Any] | None = None,
|
|
||||||
create_issue_fn: CreateIssueFn | None = None,
|
|
||||||
dry_run_incident: bool = False,
|
|
||||||
audit_path: str | None = None,
|
|
||||||
) -> dict[str, Any]:
|
|
||||||
"""Emit one restart audit event and optionally materialize an incident.
|
|
||||||
|
|
||||||
Returns ``{event, audit_written, deny_reasons, incident_result}``.
|
|
||||||
"""
|
|
||||||
event = build_restart_event(
|
|
||||||
event_type=event_type,
|
|
||||||
outcome=outcome,
|
|
||||||
correlation_id=correlation_id,
|
|
||||||
remote=remote,
|
|
||||||
org=org,
|
|
||||||
repo=repo,
|
|
||||||
requesting_session_id=requesting_session_id,
|
|
||||||
restart_class=restart_class,
|
|
||||||
profile_name=profile_name,
|
|
||||||
authenticated_username=authenticated_username,
|
|
||||||
reasons=reasons,
|
|
||||||
details=details,
|
|
||||||
)
|
|
||||||
written = emit_restart_event(event, path=audit_path)
|
|
||||||
deny = require_audit_or_deny(privileged=privileged, written=written)
|
|
||||||
incident_result = None
|
|
||||||
if create_incident is not None:
|
|
||||||
incident_result = materialize_incident(
|
|
||||||
create_incident,
|
|
||||||
create_issue_fn=create_issue_fn,
|
|
||||||
dry_run=dry_run_incident,
|
|
||||||
)
|
|
||||||
return {
|
|
||||||
"event": event,
|
|
||||||
"audit_written": written,
|
|
||||||
"deny_reasons": deny,
|
|
||||||
"incident_result": incident_result,
|
|
||||||
"correlation_id": correlation_id,
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
def incident_from_apply_gate(
|
|
||||||
*,
|
|
||||||
gate_payload: Mapping[str, Any],
|
|
||||||
break_glass: bool,
|
|
||||||
correlation_id: str,
|
|
||||||
requesting_session_id: str | None,
|
|
||||||
restart_class: str | None,
|
|
||||||
remote: str | None,
|
|
||||||
org: str | None,
|
|
||||||
repo: str | None,
|
|
||||||
) -> dict[str, Any] | None:
|
|
||||||
"""Choose an incident descriptor from an apply-gate payload, if required."""
|
|
||||||
reasons = list(gate_payload.get("reasons") or [])
|
|
||||||
proof_id = gate_payload.get("proof_id")
|
|
||||||
if break_glass:
|
|
||||||
return build_incident_descriptor(
|
|
||||||
kind=INCIDENT_BREAK_GLASS,
|
|
||||||
reasons=reasons
|
|
||||||
or ["break-glass restart path used; durable incident required (#665)"],
|
|
||||||
correlation_id=correlation_id,
|
|
||||||
requesting_session_id=requesting_session_id,
|
|
||||||
restart_class=restart_class,
|
|
||||||
remote=remote,
|
|
||||||
org=org,
|
|
||||||
repo=repo,
|
|
||||||
proof_id=proof_id if isinstance(proof_id, str) else None,
|
|
||||||
)
|
|
||||||
# Failed drain / deny path.
|
|
||||||
incident = gate_payload.get("incident")
|
|
||||||
if isinstance(incident, Mapping) and incident:
|
|
||||||
# Normalize gate-provided descriptor into our schema.
|
|
||||||
return build_incident_descriptor(
|
|
||||||
kind=INCIDENT_FAILED_DRAIN,
|
|
||||||
reasons=list(incident.get("reasons") or reasons),
|
|
||||||
correlation_id=correlation_id,
|
|
||||||
requesting_session_id=requesting_session_id
|
|
||||||
or incident.get("requesting_session_id"),
|
|
||||||
restart_class=restart_class,
|
|
||||||
remote=remote,
|
|
||||||
org=org,
|
|
||||||
repo=repo,
|
|
||||||
proof_id=incident.get("proof_id") or proof_id,
|
|
||||||
at=incident.get("at"),
|
|
||||||
)
|
|
||||||
if not gate_payload.get("allow") and not gate_payload.get("drain_gate_allow", True):
|
|
||||||
return build_incident_descriptor(
|
|
||||||
kind=INCIDENT_FAILED_DRAIN,
|
|
||||||
reasons=reasons or ["restart apply denied"],
|
|
||||||
correlation_id=correlation_id,
|
|
||||||
requesting_session_id=requesting_session_id,
|
|
||||||
restart_class=restart_class,
|
|
||||||
remote=remote,
|
|
||||||
org=org,
|
|
||||||
repo=repo,
|
|
||||||
proof_id=proof_id if isinstance(proof_id, str) else None,
|
|
||||||
)
|
|
||||||
return None
|
|
||||||
@@ -41,6 +41,7 @@ def _reset_mutation_authority(monkeypatch):
|
|||||||
"GITEA_REVIEWER_WORKTREE",
|
"GITEA_REVIEWER_WORKTREE",
|
||||||
"GITEA_MERGER_WORKTREE",
|
"GITEA_MERGER_WORKTREE",
|
||||||
"GITEA_RECONCILER_WORKTREE",
|
"GITEA_RECONCILER_WORKTREE",
|
||||||
|
"GITEA_MCP_NAMESPACE",
|
||||||
]:
|
]:
|
||||||
monkeypatch.delenv(env_key, raising=False)
|
monkeypatch.delenv(env_key, raising=False)
|
||||||
|
|
||||||
@@ -117,6 +118,7 @@ def _reset_mutation_authority(monkeypatch):
|
|||||||
monkeypatch.setattr(mcp_server, "_ACTOR_IDENTITY_CACHE", {})
|
monkeypatch.setattr(mcp_server, "_ACTOR_IDENTITY_CACHE", {})
|
||||||
monkeypatch.setattr(mcp_server, "_REVIEW_DECISION_LOCK", None)
|
monkeypatch.setattr(mcp_server, "_REVIEW_DECISION_LOCK", None)
|
||||||
monkeypatch.setattr(mcp_server, "_LIVE_NAMESPACE_HEALTH", {})
|
monkeypatch.setattr(mcp_server, "_LIVE_NAMESPACE_HEALTH", {})
|
||||||
|
monkeypatch.setattr(mcp_server, "_PROFILE_SWITCH_INVALIDATION", None)
|
||||||
monkeypatch.setattr(mcp_server, "_preflight_whoami_called", False)
|
monkeypatch.setattr(mcp_server, "_preflight_whoami_called", False)
|
||||||
monkeypatch.setattr(mcp_server, "_preflight_capability_called", False)
|
monkeypatch.setattr(mcp_server, "_preflight_capability_called", False)
|
||||||
monkeypatch.setattr(mcp_server, "_preflight_resolved_role", None)
|
monkeypatch.setattr(mcp_server, "_preflight_resolved_role", None)
|
||||||
|
|||||||
@@ -0,0 +1,274 @@
|
|||||||
|
"""Regression coverage for #690: cross-role profile activation invalidation.
|
||||||
|
|
||||||
|
A mid-run profile switch (e.g. reviewer → author → reviewer) must invalidate
|
||||||
|
workflow-load proof, reviewer lease binding, review decision lock, live
|
||||||
|
namespace health, and preflight identity/capability stamps, and must require
|
||||||
|
a full reviewer preflight before any formal verdict. Namespace provenance
|
||||||
|
must be reported and fail closed on task/namespace mismatch.
|
||||||
|
"""
|
||||||
|
import json
|
||||||
|
import os
|
||||||
|
import sys
|
||||||
|
import tempfile
|
||||||
|
import unittest
|
||||||
|
from unittest.mock import patch
|
||||||
|
|
||||||
|
sys.path.insert(0, str(__import__("pathlib").Path(__file__).resolve().parent.parent))
|
||||||
|
|
||||||
|
import gitea_config
|
||||||
|
import mcp_namespace_health
|
||||||
|
import mcp_server
|
||||||
|
import mcp_session_state
|
||||||
|
import review_workflow_load
|
||||||
|
import reviewer_pr_lease
|
||||||
|
|
||||||
|
from tests.test_runtime_clarity import CONFIG_SWITCHING_ENABLED
|
||||||
|
|
||||||
|
|
||||||
|
class TestProfileSwitchReviewGuard(unittest.TestCase):
|
||||||
|
def setUp(self):
|
||||||
|
self._remotes_patch = patch.dict(mcp_server.REMOTES, {
|
||||||
|
"dadeschools": {"host": "gitea.example.com", "org": "Example-Org", "repo": "Example-Repo"},
|
||||||
|
"prgs": {"host": "gitea.example.com", "org": "Example-Org", "repo": "Example-Repo"},
|
||||||
|
})
|
||||||
|
self._remotes_patch.start()
|
||||||
|
mcp_server._IDENTITY_CACHE.clear()
|
||||||
|
gitea_config._active_profile_override = None
|
||||||
|
self._dir = tempfile.TemporaryDirectory()
|
||||||
|
self.config_path = os.path.join(self._dir.name, "profiles.json")
|
||||||
|
with open(self.config_path, "w", encoding="utf-8") as fh:
|
||||||
|
fh.write(json.dumps(CONFIG_SWITCHING_ENABLED))
|
||||||
|
|
||||||
|
def tearDown(self):
|
||||||
|
self._remotes_patch.stop()
|
||||||
|
mcp_server._IDENTITY_CACHE.clear()
|
||||||
|
gitea_config._active_profile_override = None
|
||||||
|
self._dir.cleanup()
|
||||||
|
|
||||||
|
def _env(self, profile="reviewer-profile"):
|
||||||
|
return {
|
||||||
|
"GITEA_MCP_CONFIG": self.config_path,
|
||||||
|
"GITEA_MCP_PROFILE": profile,
|
||||||
|
"GITEA_TOKEN_AUTHOR": "author-pass",
|
||||||
|
"GITEA_TOKEN_REVIEWER": "reviewer-pass",
|
||||||
|
"GITEA_TOKEN_MERGER": "merger-pass",
|
||||||
|
}
|
||||||
|
|
||||||
|
def _seed_contaminated_review_state(self):
|
||||||
|
"""Simulate an in-flight reviewer run under reviewer-profile."""
|
||||||
|
mcp_server._preflight_whoami_called = True
|
||||||
|
mcp_server._preflight_capability_called = True
|
||||||
|
mcp_server._preflight_resolved_role = "reviewer"
|
||||||
|
mcp_server._preflight_resolved_task = "review_pr"
|
||||||
|
review_workflow_load._REVIEW_WORKFLOW_LOAD = {"loaded": True}
|
||||||
|
mcp_server._REVIEW_DECISION_LOCK = {
|
||||||
|
"session_profile": "reviewer-profile",
|
||||||
|
"final_review_decision_ready": True,
|
||||||
|
"ready_pr_number": 688,
|
||||||
|
}
|
||||||
|
reviewer_pr_lease.record_session_lease(
|
||||||
|
{"session_id": "lease-session-1", "pr_number": 688}
|
||||||
|
)
|
||||||
|
mcp_server._LIVE_NAMESPACE_HEALTH["gitea-reviewer"] = {
|
||||||
|
"namespace": "gitea-reviewer",
|
||||||
|
"healthy": True,
|
||||||
|
"ide_namespace_proven": True,
|
||||||
|
}
|
||||||
|
# Durable records keyed by the reviewer identity must also be cleared.
|
||||||
|
mcp_session_state.save_state(
|
||||||
|
kind=mcp_session_state.KIND_WORKFLOW_LOAD,
|
||||||
|
payload={"loaded": True},
|
||||||
|
profile_identity="reviewer-profile",
|
||||||
|
)
|
||||||
|
mcp_session_state.save_state(
|
||||||
|
kind=mcp_session_state.KIND_DECISION_LOCK,
|
||||||
|
payload={"final_review_decision_ready": True, "ready_pr_number": 688},
|
||||||
|
profile_identity="reviewer-profile",
|
||||||
|
)
|
||||||
|
|
||||||
|
def _activate(self, target, logins):
|
||||||
|
with patch.object(
|
||||||
|
mcp_server, "get_auth_header", side_effect=[f"token p" for _ in logins]
|
||||||
|
), patch.object(
|
||||||
|
mcp_server, "api_request", side_effect=[{"login": l} for l in logins]
|
||||||
|
), patch.object(
|
||||||
|
mcp_server,
|
||||||
|
"_workspace_repository_slug",
|
||||||
|
return_value="Example-Org/Example-Repo",
|
||||||
|
), patch.object(
|
||||||
|
mcp_server, "_canonical_repository_slug", return_value=(None, [])
|
||||||
|
):
|
||||||
|
return mcp_server.gitea_activate_profile(profile_name=target)
|
||||||
|
|
||||||
|
# -----------------------------------------------------------------
|
||||||
|
# AC1/AC2/AC3: switch invalidates review state; re-preflight required
|
||||||
|
# -----------------------------------------------------------------
|
||||||
|
def test_switch_invalidates_review_state_and_blocks_verdict(self):
|
||||||
|
with patch.dict(os.environ, self._env("reviewer-profile"), clear=True):
|
||||||
|
self._seed_contaminated_review_state()
|
||||||
|
res = self._activate("author-profile", ["reviewer-user", "author-user"])
|
||||||
|
|
||||||
|
self.assertTrue(res["success"])
|
||||||
|
self.assertTrue(res["re_preflight_required"])
|
||||||
|
inv = res["review_state_invalidation"]
|
||||||
|
self.assertEqual(inv["from_profile"], "reviewer-profile")
|
||||||
|
self.assertEqual(inv["to_profile"], "author-profile")
|
||||||
|
for item in (
|
||||||
|
"preflight_identity_capability",
|
||||||
|
"review_workflow_load",
|
||||||
|
"review_decision_lock",
|
||||||
|
"reviewer_session_lease",
|
||||||
|
"live_namespace_health",
|
||||||
|
):
|
||||||
|
self.assertIn(item, inv["invalidated"])
|
||||||
|
|
||||||
|
# In-memory state cleared.
|
||||||
|
self.assertFalse(mcp_server._preflight_whoami_called)
|
||||||
|
self.assertFalse(mcp_server._preflight_capability_called)
|
||||||
|
self.assertIsNone(mcp_server._preflight_resolved_task)
|
||||||
|
self.assertIsNone(review_workflow_load._REVIEW_WORKFLOW_LOAD)
|
||||||
|
self.assertIsNone(mcp_server._REVIEW_DECISION_LOCK)
|
||||||
|
self.assertIsNone(reviewer_pr_lease.get_session_lease())
|
||||||
|
self.assertEqual(mcp_server._LIVE_NAMESPACE_HEALTH, {})
|
||||||
|
self.assertIsNotNone(mcp_server._PROFILE_SWITCH_INVALIDATION)
|
||||||
|
|
||||||
|
# Durable records keyed by the reviewer identity are gone.
|
||||||
|
self.assertIsNone(
|
||||||
|
mcp_session_state.load_state(
|
||||||
|
kind=mcp_session_state.KIND_WORKFLOW_LOAD,
|
||||||
|
profile_identity="reviewer-profile",
|
||||||
|
)
|
||||||
|
)
|
||||||
|
self.assertIsNone(
|
||||||
|
mcp_session_state.load_state(
|
||||||
|
kind=mcp_session_state.KIND_DECISION_LOCK,
|
||||||
|
profile_identity="reviewer-profile",
|
||||||
|
)
|
||||||
|
)
|
||||||
|
|
||||||
|
# A formal verdict without re-preflight fails closed.
|
||||||
|
reasons = mcp_server.check_review_decision_gate(
|
||||||
|
688, "APPROVE", final_review_decision_ready=True
|
||||||
|
)
|
||||||
|
self.assertTrue(reasons)
|
||||||
|
|
||||||
|
def test_switch_back_cannot_resurrect_stale_review_run(self):
|
||||||
|
with patch.dict(os.environ, self._env("reviewer-profile"), clear=True):
|
||||||
|
self._seed_contaminated_review_state()
|
||||||
|
self._activate("author-profile", ["reviewer-user", "author-user"])
|
||||||
|
res = self._activate("reviewer-profile", ["author-user", "reviewer-user"])
|
||||||
|
|
||||||
|
self.assertTrue(res["success"])
|
||||||
|
# The pre-switch review run must not reappear.
|
||||||
|
self.assertIsNone(mcp_server._REVIEW_DECISION_LOCK)
|
||||||
|
self.assertIsNone(review_workflow_load._REVIEW_WORKFLOW_LOAD)
|
||||||
|
self.assertIsNone(reviewer_pr_lease.get_session_lease())
|
||||||
|
status = review_workflow_load.workflow_load_status()
|
||||||
|
self.assertFalse(status["workflow_load_valid"])
|
||||||
|
reasons = mcp_server.check_review_decision_gate(
|
||||||
|
688, "APPROVE", final_review_decision_ready=True
|
||||||
|
)
|
||||||
|
self.assertTrue(reasons)
|
||||||
|
|
||||||
|
def test_same_profile_reactivation_keeps_state(self):
|
||||||
|
with patch.dict(os.environ, self._env("reviewer-profile"), clear=True):
|
||||||
|
self._seed_contaminated_review_state()
|
||||||
|
res = self._activate("reviewer-profile", ["reviewer-user", "reviewer-user"])
|
||||||
|
self.assertTrue(res["success"], res)
|
||||||
|
self.assertNotIn("review_state_invalidation", res)
|
||||||
|
self.assertIsNotNone(mcp_server._REVIEW_DECISION_LOCK)
|
||||||
|
self.assertTrue(mcp_server._preflight_whoami_called)
|
||||||
|
|
||||||
|
def test_clean_repreflight_after_switch_allows_gate(self):
|
||||||
|
with patch.dict(os.environ, self._env("reviewer-profile"), clear=True):
|
||||||
|
self._seed_contaminated_review_state()
|
||||||
|
self._activate("author-profile", ["reviewer-user", "author-user"])
|
||||||
|
self._activate("reviewer-profile", ["author-user", "reviewer-user"])
|
||||||
|
|
||||||
|
# Re-establish the full reviewer preflight under the new profile.
|
||||||
|
mcp_server.record_preflight_check("whoami")
|
||||||
|
mcp_server.record_preflight_check(
|
||||||
|
"capability", resolved_role="reviewer", resolved_task="review_pr"
|
||||||
|
)
|
||||||
|
mcp_server.init_review_decision_lock("dadeschools", "review_pr")
|
||||||
|
lock = mcp_server._load_review_decision_lock()
|
||||||
|
self.assertIsNotNone(lock)
|
||||||
|
lock.update(
|
||||||
|
{
|
||||||
|
"final_review_decision_ready": True,
|
||||||
|
"ready_pr_number": 688,
|
||||||
|
"ready_action": "APPROVE",
|
||||||
|
"ready_remote": "dadeschools",
|
||||||
|
"ready_org": "Example-Org",
|
||||||
|
"ready_repo": "Example-Repo",
|
||||||
|
}
|
||||||
|
)
|
||||||
|
mcp_server._save_review_decision_lock(lock)
|
||||||
|
|
||||||
|
with patch.object(
|
||||||
|
mcp_server, "_review_workflow_load_gate_reasons", return_value=[]
|
||||||
|
):
|
||||||
|
reasons = mcp_server.check_review_decision_gate(
|
||||||
|
688,
|
||||||
|
"APPROVE",
|
||||||
|
final_review_decision_ready=True,
|
||||||
|
remote="dadeschools",
|
||||||
|
)
|
||||||
|
self.assertEqual(reasons, [])
|
||||||
|
|
||||||
|
# -----------------------------------------------------------------
|
||||||
|
# AC4: namespace provenance reporting + fail-closed mismatch
|
||||||
|
# -----------------------------------------------------------------
|
||||||
|
def test_namespace_provenance_mismatch_detection(self):
|
||||||
|
prov = mcp_namespace_health.namespace_provenance(
|
||||||
|
task="review_pr",
|
||||||
|
active_profile="reviewer-profile",
|
||||||
|
env={"GITEA_MCP_NAMESPACE": "gitea-author"},
|
||||||
|
)
|
||||||
|
self.assertTrue(prov["mismatch"])
|
||||||
|
self.assertEqual(prov["required_namespace"], "gitea-reviewer")
|
||||||
|
|
||||||
|
prov_ok = mcp_namespace_health.namespace_provenance(
|
||||||
|
task="review_pr",
|
||||||
|
active_profile="reviewer-profile",
|
||||||
|
env={"GITEA_MCP_NAMESPACE": "gitea-reviewer"},
|
||||||
|
)
|
||||||
|
self.assertFalse(prov_ok["mismatch"])
|
||||||
|
|
||||||
|
prov_unknown = mcp_namespace_health.namespace_provenance(
|
||||||
|
task="review_pr", active_profile="reviewer-profile", env={}
|
||||||
|
)
|
||||||
|
self.assertIsNone(prov_unknown["configured_namespace"])
|
||||||
|
self.assertFalse(prov_unknown["mismatch"])
|
||||||
|
self.assertEqual(prov_unknown["namespace_source"], "unknown")
|
||||||
|
|
||||||
|
@patch("mcp_server.api_request", return_value={"login": "reviewer-user"})
|
||||||
|
@patch("mcp_server.get_auth_header", return_value="token reviewer-pass")
|
||||||
|
def test_whoami_reports_namespace_provenance(self, _auth, _api):
|
||||||
|
env = self._env("reviewer-profile")
|
||||||
|
env["GITEA_MCP_NAMESPACE"] = "gitea-reviewer"
|
||||||
|
with patch.dict(os.environ, env, clear=True):
|
||||||
|
res = mcp_server.gitea_whoami(remote="dadeschools")
|
||||||
|
prov = res["namespace_provenance"]
|
||||||
|
self.assertEqual(prov["configured_namespace"], "gitea-reviewer")
|
||||||
|
self.assertEqual(prov["active_profile"], "reviewer-profile")
|
||||||
|
self.assertFalse(prov["mismatch"])
|
||||||
|
|
||||||
|
@patch("mcp_server.api_request", return_value={"login": "reviewer-user"})
|
||||||
|
@patch("mcp_server.get_auth_header", return_value="token reviewer-pass")
|
||||||
|
def test_resolve_fails_closed_on_namespace_mismatch(self, _auth, _api):
|
||||||
|
env = self._env("reviewer-profile")
|
||||||
|
env["GITEA_MCP_NAMESPACE"] = "gitea-author"
|
||||||
|
with patch.dict(os.environ, env, clear=True):
|
||||||
|
res = mcp_server.gitea_resolve_task_capability(
|
||||||
|
task="review_pr", kwargs="{}", remote="dadeschools"
|
||||||
|
)
|
||||||
|
self.assertFalse(res["allowed_in_current_session"])
|
||||||
|
self.assertTrue(res["namespace_provenance"]["mismatch"])
|
||||||
|
self.assertTrue(
|
||||||
|
any("namespace" in g for g in res["task_role_guidance"])
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
unittest.main()
|
||||||
@@ -1,342 +0,0 @@
|
|||||||
"""Tests for MCP restart lifecycle audit events and incidents (#665)."""
|
|
||||||
|
|
||||||
from __future__ import annotations
|
|
||||||
|
|
||||||
import json
|
|
||||||
import os
|
|
||||||
import tempfile
|
|
||||||
import unittest
|
|
||||||
from unittest.mock import patch
|
|
||||||
|
|
||||||
import gitea_audit
|
|
||||||
import restart_audit as ra
|
|
||||||
|
|
||||||
|
|
||||||
class TestEventSchema(unittest.TestCase):
|
|
||||||
def test_all_lifecycle_event_types_are_named(self):
|
|
||||||
expected = {
|
|
||||||
"mcp.restart.impact_preview",
|
|
||||||
"mcp.restart.drain_enter",
|
|
||||||
"mcp.restart.drain_exit",
|
|
||||||
"mcp.restart.drain_proof",
|
|
||||||
"mcp.restart.apply_gate",
|
|
||||||
"mcp.restart.break_glass",
|
|
||||||
"mcp.restart.post_restart_reconcile",
|
|
||||||
"mcp.restart.narrower_recovery",
|
|
||||||
"mcp.restart.unguarded_detected",
|
|
||||||
}
|
|
||||||
self.assertEqual(set(ra.RESTART_EVENT_TYPES), expected)
|
|
||||||
|
|
||||||
def test_build_restart_event_core_fields(self):
|
|
||||||
event = ra.build_restart_event(
|
|
||||||
event_type=ra.EVENT_IMPACT_PREVIEW,
|
|
||||||
outcome="safe",
|
|
||||||
correlation_id="rst-abc123",
|
|
||||||
remote="prgs",
|
|
||||||
org="Scaled-Tech-Consulting",
|
|
||||||
repo="Gitea-Tools",
|
|
||||||
requesting_session_id="sess-1",
|
|
||||||
restart_class="full_mcp_restart",
|
|
||||||
profile_name="prgs-author",
|
|
||||||
authenticated_username="bot",
|
|
||||||
reasons=["inventory complete"],
|
|
||||||
details={"allow_restart": True},
|
|
||||||
now="2026-07-25T12:00:00+00:00",
|
|
||||||
)
|
|
||||||
self.assertEqual(event["event_type"], ra.EVENT_IMPACT_PREVIEW)
|
|
||||||
self.assertEqual(event["action"], ra.EVENT_IMPACT_PREVIEW)
|
|
||||||
self.assertEqual(event["action_type"], "restart_lifecycle")
|
|
||||||
self.assertEqual(event["result"], "safe")
|
|
||||||
self.assertEqual(event["correlation_id"], "rst-abc123")
|
|
||||||
self.assertEqual(event["profile_name"], "prgs-author")
|
|
||||||
self.assertEqual(event["authenticated_username"], "bot")
|
|
||||||
meta = event["request_metadata"]
|
|
||||||
self.assertEqual(meta["event_family"], "mcp.restart")
|
|
||||||
self.assertEqual(meta["correlation_id"], "rst-abc123")
|
|
||||||
self.assertEqual(meta["restart_class"], "full_mcp_restart")
|
|
||||||
self.assertEqual(meta["details"]["allow_restart"], True)
|
|
||||||
|
|
||||||
def test_unknown_event_type_raises(self):
|
|
||||||
with self.assertRaises(ValueError) as ctx:
|
|
||||||
ra.build_restart_event(
|
|
||||||
event_type="mcp.restart.not_a_real_event",
|
|
||||||
outcome="x",
|
|
||||||
correlation_id="rst-1",
|
|
||||||
)
|
|
||||||
self.assertIn("unknown restart audit event_type", str(ctx.exception))
|
|
||||||
|
|
||||||
def test_reasons_and_details_are_redacted(self):
|
|
||||||
event = ra.build_restart_event(
|
|
||||||
event_type=ra.EVENT_APPLY_GATE,
|
|
||||||
outcome="deny",
|
|
||||||
correlation_id="rst-sec",
|
|
||||||
reasons=["token secret-xyz rejected", "ok"],
|
|
||||||
details={"token": "leak-token", "status": "denied"},
|
|
||||||
)
|
|
||||||
self.assertNotIn("secret-xyz", event.get("reason") or "")
|
|
||||||
meta = event["request_metadata"]
|
|
||||||
self.assertEqual(meta["details"]["token"], gitea_audit.REDACTED)
|
|
||||||
self.assertEqual(meta["details"]["status"], "denied")
|
|
||||||
for reason in meta["reasons"]:
|
|
||||||
self.assertNotIn("secret-xyz", reason)
|
|
||||||
|
|
||||||
def test_new_correlation_id_shape(self):
|
|
||||||
cid = ra.new_correlation_id()
|
|
||||||
self.assertTrue(cid.startswith("rst-"))
|
|
||||||
self.assertEqual(len(cid), len("rst-") + 16)
|
|
||||||
|
|
||||||
|
|
||||||
class TestEmitAndRequire(unittest.TestCase):
|
|
||||||
def test_emit_appends_json_line(self):
|
|
||||||
with tempfile.TemporaryDirectory() as d:
|
|
||||||
path = os.path.join(d, "audit.log")
|
|
||||||
event = ra.build_restart_event(
|
|
||||||
event_type=ra.EVENT_DRAIN_PROOF,
|
|
||||||
outcome="pass",
|
|
||||||
correlation_id="rst-write",
|
|
||||||
)
|
|
||||||
self.assertTrue(ra.emit_restart_event(event, path=path))
|
|
||||||
with open(path, encoding="utf-8") as fh:
|
|
||||||
lines = fh.read().splitlines()
|
|
||||||
self.assertEqual(len(lines), 1)
|
|
||||||
loaded = json.loads(lines[0])
|
|
||||||
self.assertEqual(loaded["event_type"], ra.EVENT_DRAIN_PROOF)
|
|
||||||
self.assertEqual(loaded["correlation_id"], "rst-write")
|
|
||||||
|
|
||||||
def test_emit_never_raises(self):
|
|
||||||
self.assertFalse(
|
|
||||||
ra.emit_restart_event({"action": "x"}, path="/no/such/dir/audit.log")
|
|
||||||
)
|
|
||||||
|
|
||||||
def test_require_audit_denies_privileged_when_write_fails_and_enabled(self):
|
|
||||||
deny = ra.require_audit_or_deny(
|
|
||||||
privileged=True, written=False, audit_enabled=True
|
|
||||||
)
|
|
||||||
self.assertEqual(len(deny), 1)
|
|
||||||
self.assertIn("fail closed", deny[0])
|
|
||||||
|
|
||||||
def test_require_audit_allows_when_audit_disabled(self):
|
|
||||||
# Rollout policy: enable audit before enforcing deny-on-audit-fail.
|
|
||||||
deny = ra.require_audit_or_deny(
|
|
||||||
privileged=True, written=False, audit_enabled=False
|
|
||||||
)
|
|
||||||
self.assertEqual(deny, [])
|
|
||||||
|
|
||||||
def test_require_audit_noop_for_non_privileged(self):
|
|
||||||
deny = ra.require_audit_or_deny(
|
|
||||||
privileged=False, written=False, audit_enabled=True
|
|
||||||
)
|
|
||||||
self.assertEqual(deny, [])
|
|
||||||
|
|
||||||
def test_require_audit_allows_when_written(self):
|
|
||||||
deny = ra.require_audit_or_deny(
|
|
||||||
privileged=True, written=True, audit_enabled=True
|
|
||||||
)
|
|
||||||
self.assertEqual(deny, [])
|
|
||||||
|
|
||||||
|
|
||||||
class TestIncidents(unittest.TestCase):
|
|
||||||
def test_break_glass_descriptor(self):
|
|
||||||
desc = ra.build_incident_descriptor(
|
|
||||||
kind=ra.INCIDENT_BREAK_GLASS,
|
|
||||||
reasons=["break-glass authorized"],
|
|
||||||
correlation_id="rst-bg",
|
|
||||||
requesting_session_id="s1",
|
|
||||||
restart_class="full_mcp_restart",
|
|
||||||
remote="prgs",
|
|
||||||
org="O",
|
|
||||||
repo="R",
|
|
||||||
)
|
|
||||||
self.assertEqual(desc["kind"], ra.INCIDENT_BREAK_GLASS)
|
|
||||||
self.assertIn("Break-glass", desc["title"])
|
|
||||||
self.assertIn("mcp-health", desc["labels"])
|
|
||||||
self.assertEqual(desc["source"], "restart_audit#665")
|
|
||||||
|
|
||||||
def test_incident_body_redacts_and_includes_correlation(self):
|
|
||||||
desc = ra.build_incident_descriptor(
|
|
||||||
kind=ra.INCIDENT_FAILED_DRAIN,
|
|
||||||
reasons=["token secret-xyz failed proof"],
|
|
||||||
correlation_id="rst-body",
|
|
||||||
remote="prgs",
|
|
||||||
org="O",
|
|
||||||
repo="R",
|
|
||||||
proof_id="proof-1",
|
|
||||||
)
|
|
||||||
body = ra.incident_body(desc)
|
|
||||||
self.assertIn("rst-body", body)
|
|
||||||
self.assertIn("proof-1", body)
|
|
||||||
self.assertIn("mcp-restart-incident:v1", body)
|
|
||||||
self.assertNotIn("secret-xyz", body)
|
|
||||||
|
|
||||||
def test_materialize_dry_run(self):
|
|
||||||
desc = ra.build_incident_descriptor(
|
|
||||||
kind=ra.INCIDENT_FAILED_DRAIN,
|
|
||||||
reasons=["denied"],
|
|
||||||
correlation_id="rst-dr",
|
|
||||||
)
|
|
||||||
result = ra.materialize_incident(desc, create_issue_fn=lambda **k: {}, dry_run=True)
|
|
||||||
self.assertFalse(result["created"])
|
|
||||||
self.assertTrue(result["dry_run"])
|
|
||||||
self.assertIn("dry-run", result["reasons"][0])
|
|
||||||
|
|
||||||
def test_materialize_without_create_fn(self):
|
|
||||||
desc = ra.build_incident_descriptor(
|
|
||||||
kind=ra.INCIDENT_FAILED_DRAIN,
|
|
||||||
reasons=["denied"],
|
|
||||||
correlation_id="rst-nfn",
|
|
||||||
)
|
|
||||||
result = ra.materialize_incident(desc, create_issue_fn=None)
|
|
||||||
self.assertFalse(result["created"])
|
|
||||||
self.assertIn("create_issue_fn not provided", result["reasons"][0])
|
|
||||||
|
|
||||||
def test_materialize_creates_issue(self):
|
|
||||||
created = {}
|
|
||||||
|
|
||||||
def _create(*, title, body, labels, org=None, repo=None, **_kw):
|
|
||||||
created["title"] = title
|
|
||||||
created["body"] = body
|
|
||||||
created["labels"] = labels
|
|
||||||
created["org"] = org
|
|
||||||
created["repo"] = repo
|
|
||||||
return {"number": 999}
|
|
||||||
|
|
||||||
desc = ra.build_incident_descriptor(
|
|
||||||
kind=ra.INCIDENT_BREAK_GLASS,
|
|
||||||
reasons=["break-glass"],
|
|
||||||
correlation_id="rst-create",
|
|
||||||
org="O",
|
|
||||||
repo="R",
|
|
||||||
)
|
|
||||||
result = ra.materialize_incident(desc, create_issue_fn=_create)
|
|
||||||
self.assertTrue(result["created"])
|
|
||||||
self.assertEqual(result["issue_number"], 999)
|
|
||||||
self.assertIn("Break-glass", created["title"])
|
|
||||||
self.assertIn("rst-create", created["body"])
|
|
||||||
self.assertEqual(created["org"], "O")
|
|
||||||
|
|
||||||
def test_materialize_never_raises_on_create_failure(self):
|
|
||||||
def _boom(**_kw):
|
|
||||||
raise RuntimeError("token secret-xyz network")
|
|
||||||
|
|
||||||
desc = ra.build_incident_descriptor(
|
|
||||||
kind=ra.INCIDENT_FAILED_DRAIN,
|
|
||||||
reasons=["x"],
|
|
||||||
correlation_id="rst-boom",
|
|
||||||
)
|
|
||||||
result = ra.materialize_incident(desc, create_issue_fn=_boom)
|
|
||||||
self.assertFalse(result["created"])
|
|
||||||
self.assertIn("failed", result["reasons"][0])
|
|
||||||
self.assertNotIn("secret-xyz", result["reasons"][0])
|
|
||||||
|
|
||||||
|
|
||||||
class TestIncidentFromApplyGate(unittest.TestCase):
|
|
||||||
def test_break_glass_always_incident(self):
|
|
||||||
desc = ra.incident_from_apply_gate(
|
|
||||||
gate_payload={"allow": True, "reasons": [], "proof_id": None},
|
|
||||||
break_glass=True,
|
|
||||||
correlation_id="rst-bg2",
|
|
||||||
requesting_session_id="s",
|
|
||||||
restart_class="full_mcp_restart",
|
|
||||||
remote="prgs",
|
|
||||||
org="O",
|
|
||||||
repo="R",
|
|
||||||
)
|
|
||||||
self.assertIsNotNone(desc)
|
|
||||||
self.assertEqual(desc["kind"], ra.INCIDENT_BREAK_GLASS)
|
|
||||||
|
|
||||||
def test_failed_drain_from_gate_incident(self):
|
|
||||||
desc = ra.incident_from_apply_gate(
|
|
||||||
gate_payload={
|
|
||||||
"allow": False,
|
|
||||||
"drain_gate_allow": False,
|
|
||||||
"reasons": ["proof expired"],
|
|
||||||
"incident": {
|
|
||||||
"reasons": ["proof expired"],
|
|
||||||
"proof_id": "p1",
|
|
||||||
},
|
|
||||||
},
|
|
||||||
break_glass=False,
|
|
||||||
correlation_id="rst-fd",
|
|
||||||
requesting_session_id="s",
|
|
||||||
restart_class="full_mcp_restart",
|
|
||||||
remote="prgs",
|
|
||||||
org="O",
|
|
||||||
repo="R",
|
|
||||||
)
|
|
||||||
self.assertIsNotNone(desc)
|
|
||||||
self.assertEqual(desc["kind"], ra.INCIDENT_FAILED_DRAIN)
|
|
||||||
self.assertEqual(desc["proof_id"], "p1")
|
|
||||||
|
|
||||||
def test_allow_without_break_glass_no_incident(self):
|
|
||||||
desc = ra.incident_from_apply_gate(
|
|
||||||
gate_payload={
|
|
||||||
"allow": True,
|
|
||||||
"drain_gate_allow": True,
|
|
||||||
"reasons": [],
|
|
||||||
},
|
|
||||||
break_glass=False,
|
|
||||||
correlation_id="rst-ok",
|
|
||||||
requesting_session_id="s",
|
|
||||||
restart_class="full_mcp_restart",
|
|
||||||
remote="prgs",
|
|
||||||
org="O",
|
|
||||||
repo="R",
|
|
||||||
)
|
|
||||||
self.assertIsNone(desc)
|
|
||||||
|
|
||||||
|
|
||||||
class TestRecordLifecycle(unittest.TestCase):
|
|
||||||
def test_record_emits_and_materializes(self):
|
|
||||||
created = []
|
|
||||||
|
|
||||||
def _create(**kwargs):
|
|
||||||
created.append(kwargs)
|
|
||||||
return {"number": 42}
|
|
||||||
|
|
||||||
with tempfile.TemporaryDirectory() as d:
|
|
||||||
path = os.path.join(d, "audit.log")
|
|
||||||
with patch.dict(os.environ, {"GITEA_AUDIT_LOG": path}, clear=False):
|
|
||||||
incident = ra.build_incident_descriptor(
|
|
||||||
kind=ra.INCIDENT_BREAK_GLASS,
|
|
||||||
reasons=["bg"],
|
|
||||||
correlation_id="rst-lc",
|
|
||||||
org="O",
|
|
||||||
repo="R",
|
|
||||||
)
|
|
||||||
out = ra.record_restart_lifecycle(
|
|
||||||
event_type=ra.EVENT_BREAK_GLASS,
|
|
||||||
outcome="break_glass",
|
|
||||||
correlation_id="rst-lc",
|
|
||||||
remote="prgs",
|
|
||||||
org="O",
|
|
||||||
repo="R",
|
|
||||||
privileged=True,
|
|
||||||
create_incident=incident,
|
|
||||||
create_issue_fn=_create,
|
|
||||||
audit_path=path,
|
|
||||||
)
|
|
||||||
self.assertTrue(out["audit_written"])
|
|
||||||
self.assertEqual(out["deny_reasons"], [])
|
|
||||||
self.assertTrue(out["incident_result"]["created"])
|
|
||||||
self.assertEqual(out["incident_result"]["issue_number"], 42)
|
|
||||||
self.assertEqual(len(created), 1)
|
|
||||||
|
|
||||||
def test_privileged_deny_when_audit_write_fails(self):
|
|
||||||
with patch.dict(
|
|
||||||
os.environ, {"GITEA_AUDIT_LOG": "/no/such/dir/a.log"}, clear=False
|
|
||||||
):
|
|
||||||
with patch("restart_audit.emit_restart_event", return_value=False):
|
|
||||||
with patch("gitea_audit.audit_enabled", return_value=True):
|
|
||||||
out = ra.record_restart_lifecycle(
|
|
||||||
event_type=ra.EVENT_APPLY_GATE,
|
|
||||||
outcome="deny",
|
|
||||||
correlation_id="rst-deny",
|
|
||||||
privileged=True,
|
|
||||||
audit_path="/no/such/dir/a.log",
|
|
||||||
)
|
|
||||||
self.assertFalse(out["audit_written"])
|
|
||||||
self.assertEqual(len(out["deny_reasons"]), 1)
|
|
||||||
|
|
||||||
|
|
||||||
if __name__ == "__main__":
|
|
||||||
unittest.main()
|
|
||||||
Reference in New Issue
Block a user