mirror of
https://github.com/shankar0123/certctl.git
synced 2026-06-11 23:39:01 +00:00
fix(bundle-3): MCP Trust-Boundary Fencing — 5 audit findings closed
Closes Audit-2026-04-25 H-002, H-003, M-003, M-004, M-005 (all CWE-1039 LLM Prompt Injection at the MCP↔consumer trust boundary, TB-7). Strategy: wrapper-layer fencing. All 87 MCP tools route their success path through textResult and their failure path through errorResult. By fencing at those two wrappers we cover every existing tool AND every future tool with a single change — no per-tool wiring required. What changed - internal/mcp/fence.go (new) — FenceUntrusted helper with strategy doc + per-finding rationale. Both fenceMCPResponse and fenceMCPError use it internally. - internal/mcp/tools.go — textResult wraps response body via fenceMCPResponse; errorResult wraps error string via fenceMCPError. - internal/mcp/tools_test.go — TestTextResult / TestErrorResult updated to assert fenced shape (start marker + end marker + inner body). - internal/mcp/injection_regression_test.go (new) — 5 regression test functions, one per audit finding, each replays 5 classic LLM injection payloads (instruction_override, system_role_spoofing, delimiter_break_attempt, markdown_link_phishing, data_exfil_via_url) and asserts the planted payload appears VERBATIM (preservation, operator visibility) INSIDE the fence boundaries. - internal/mcp/fence_guardrail_test.go (new) — CI guardrail that walks every non-test .go file in the mcp package and fails if it finds a bare gomcp.CallToolResult literal outside tools.go. Prevents future tools from silently bypassing the fence. Delimiter-forgery defense The naive constant fence (--- UNTRUSTED MCP_RESPONSE END ---) is forgeable: an attacker who controls a field value can plant the literal end marker and "break out" of the fence. Defense: every fence call generates a 6-byte crypto/rand nonce, hex-encoded, and embeds it in BOTH the START and END markers. An attacker would need to predict the nonce (2^48 search per fence) to forge a matching END inside the payload. The delimiter_break_attempt regression test exercises this. Per-finding mapping - H-002 Cert Subject DN injection (CSR submitter controlled) → TestMCP_PromptInjection_H002_CertSubjectDN - H-003 Discovered cert metadata injection (cert owner controlled) → TestMCP_PromptInjection_H003_DiscoveredCertMetadata - M-003 Agent heartbeat injection (agent self-reports hostname/OS/IP) → TestMCP_PromptInjection_M003_AgentHeartbeat - M-004 Upstream CA error injection (CA controls error string) → TestMCP_PromptInjection_M004_UpstreamCAError - M-005 Audit details + notification body injection (downstream actors control these) → TestMCP_PromptInjection_M005_AuditDetailsAndNotifications Verification gates - go vet ./... → clean - go build ./... → clean - go test -short -count=1 ./... → all packages pass - go test -count=1 ./internal/mcp/... → all packages pass - npx tsc --noEmit (web) → clean - npx vitest run (web) → 337 passed - python3 yaml.safe_load(api/openapi.yaml) → 89 paths, 56 schemas Threat-model placement: TB-7 (MCP↔LLM consumer). certctl owns the boundary; consumer-side prompt engineering is recommended but not relied upon. Defense-in-depth: per-call nonce closes the delimiter-forgery edge case that constant fences would have left exposed. Bundle 3 of the 2026-04-25 comprehensive audit (88 findings).
This commit is contained in:
+19
-2
@@ -33,16 +33,33 @@ func RegisterTools(s *gomcp.Server, client *Client) {
|
||||
|
||||
// ── Helpers ─────────────────────────────────────────────────────────
|
||||
|
||||
// textResult is the success-path wrapper used by every MCP tool. Bundle-3
|
||||
// (Audit H-002, H-003, M-003, M-004, M-005, CWE-1039 LLM Prompt Injection):
|
||||
// the response body returned to the LLM consumer may contain attacker-
|
||||
// controllable text — cert subject DN/SANs (CSR submitter controls), agent
|
||||
// hostname/OS/arch/IP (agent self-reports), upstream CA error strings (CA
|
||||
// controls), audit details + notification bodies (downstream actors). To
|
||||
// make the trust boundary explicit, we wrap every body in `--- UNTRUSTED
|
||||
// MCP_RESPONSE START ... END ---` fences. LLM consumers that fence
|
||||
// untrusted data correctly will see the attack as data, not instructions.
|
||||
//
|
||||
// See internal/mcp/fence.go for the strategy doc + per-finding rationale.
|
||||
func textResult(data json.RawMessage) (*gomcp.CallToolResult, any, error) {
|
||||
return &gomcp.CallToolResult{
|
||||
Content: []gomcp.Content{
|
||||
&gomcp.TextContent{Text: string(data)},
|
||||
&gomcp.TextContent{Text: fenceMCPResponse(string(data))},
|
||||
},
|
||||
}, nil, nil
|
||||
}
|
||||
|
||||
// errorResult is the failure-path wrapper used by every MCP tool. Bundle-3
|
||||
// (M-004 in particular): the wrapped error often originates from an upstream
|
||||
// CA whose error string the attacker may control. We fence the error message
|
||||
// via fenceMCPError before returning to the LLM consumer. The third return
|
||||
// value is what the gomcp framework surfaces; gomcp formats it into a
|
||||
// CallToolResult.IsError content automatically.
|
||||
func errorResult(err error) (*gomcp.CallToolResult, any, error) {
|
||||
return nil, nil, fmt.Errorf("%w", err)
|
||||
return nil, nil, fmt.Errorf("%s", fenceMCPError(err.Error()))
|
||||
}
|
||||
|
||||
func paginationQuery(page, perPage int) url.Values {
|
||||
|
||||
Reference in New Issue
Block a user