diff --git a/crates/runtime-api/src/lib.rs b/crates/runtime-api/src/lib.rs index 1cde155f9..f24428904 100644 --- a/crates/runtime-api/src/lib.rs +++ b/crates/runtime-api/src/lib.rs @@ -1139,10 +1139,6 @@ pub struct RemainingRuntimeRoute { pub reason: &'static str, } -pub const RUNTIME_ROUTE_VERIFICATION_CHALLENGE: &str = - "/v1/workspace-runtime-verification/challenge"; -pub const RUNTIME_ROUTE_VERIFICATION_ACK: &str = - "/v1/workspace-runtime-verification/acknowledgement"; pub const RUNTIME_ROUTE_CONFIG_BUNDLES: &str = "/v1/config-bundles"; pub const RUNTIME_ROUTE_CONFIG_BUNDLE_AVAILABILITY: &str = "/v1/config-bundles/{bundle_id}/availability"; @@ -1169,16 +1165,6 @@ pub const RUNTIME_ROUTE_WORKER_SESSION_ATTACHMENT: &str = /// Intentionally out-of-contract Runtime routes. This inventory keeps manual paths visible rather /// than letting them be mistaken for management-contract omissions. pub const REMAINING_RUNTIME_ROUTES: &[RemainingRuntimeRoute] = &[ - RemainingRuntimeRoute { - method: "POST", - path: RUNTIME_ROUTE_VERIFICATION_CHALLENGE, - reason: "Workspace verification handshake", - }, - RemainingRuntimeRoute { - method: "POST", - path: RUNTIME_ROUTE_VERIFICATION_ACK, - reason: "Workspace verification handshake", - }, RemainingRuntimeRoute { method: "GET", path: RUNTIME_ROUTE_CONFIG_BUNDLES, @@ -1625,7 +1611,7 @@ mod tests { #[test] fn remaining_route_inventory_does_not_overlap_contract() { - assert_eq!(REMAINING_RUNTIME_ROUTES.len(), 21); + assert_eq!(REMAINING_RUNTIME_ROUTES.len(), 19); for remaining in REMAINING_RUNTIME_ROUTES { assert!(!RuntimeApiMetadata::OPERATIONS.iter().any(|operation| { format!("{:?}", operation.method).eq_ignore_ascii_case(remaining.method) diff --git a/crates/server-api/src/lib.rs b/crates/server-api/src/lib.rs index d5758ad75..7c512b2b3 100644 --- a/crates/server-api/src/lib.rs +++ b/crates/server-api/src/lib.rs @@ -8837,62 +8837,14 @@ pub struct RuntimeSummary { pub diagnostics: Vec, } -#[derive(Debug, Clone, Copy, Serialize, Deserialize, PartialEq, Eq, JsonSchema)] -#[cfg_attr(feature = "typescript", derive(ts_rs::TS))] -#[serde(rename_all = "snake_case")] -pub enum WorkspaceRuntimeBindingState { - Configured, - Verified, - Revoked, -} - -#[derive(Debug, Clone, Copy, Serialize, Deserialize, PartialEq, Eq, JsonSchema)] -#[cfg_attr(feature = "typescript", derive(ts_rs::TS))] -#[serde(rename_all = "snake_case")] -pub enum RuntimeConnectionDisplayState { - Configured, - Verified, - Unavailable, - Revoked, -} - -#[derive(Debug, Clone, Copy, Serialize, Deserialize, PartialEq, Eq, JsonSchema)] -#[cfg_attr(feature = "typescript", derive(ts_rs::TS))] -#[serde(rename_all = "snake_case")] -pub enum RuntimeVerificationOutcome { - Verified, - ChallengeIssued, - VerificationFailed, - ConnectivityFailed, -} - -#[derive(Debug, Clone, Serialize, Deserialize, PartialEq, Eq, JsonSchema)] -#[cfg_attr(feature = "typescript", derive(ts_rs::TS))] -#[serde(deny_unknown_fields)] -pub struct RuntimeVerificationEvidenceSummary { - pub verified_at: Option, - pub last_checked_at: String, - pub last_outcome: RuntimeVerificationOutcome, - pub binding_id: String, - pub workspace_key_id: String, - pub workspace_public_key_fingerprint: String, - pub workspace_trust_id: String, - pub runtime_public_key_fingerprint: String, -} - +/// Binding identity and explicit revocation only. Each signed request authenticates +/// using the current Workspace signing identity, not a stored approval. #[derive(Debug, Clone, Serialize, Deserialize, PartialEq, Eq, JsonSchema)] #[cfg_attr(feature = "typescript", derive(ts_rs::TS))] #[serde(deny_unknown_fields)] pub struct WorkspaceRuntimeBindingSummary { - pub state: WorkspaceRuntimeBindingState, - pub connection_state: RuntimeConnectionDisplayState, pub binding_id: String, - #[serde(default, skip_serializing_if = "Option::is_none")] - pub workspace_key_id: Option, - #[serde(default, skip_serializing_if = "Option::is_none")] - pub workspace_trust_id: Option, - #[serde(default, skip_serializing_if = "Option::is_none")] - pub verification: Option, + pub revoked_at: Option, } #[derive(Debug, Clone, Serialize, Deserialize, PartialEq, Eq, JsonSchema)] @@ -9076,9 +9028,6 @@ pub struct RuntimePublicIdentityBundle { #[serde(deny_unknown_fields)] pub struct CreateRemoteRuntimeRequest { pub public_bundle: RuntimePublicIdentityBundle, - /// Exact Runtime-issued enrollment identity from `trust-workspace add/show`. - /// A key fingerprint alone cannot distinguish revocation and re-enrollment. - pub workspace_trust_id: String, #[serde(default, skip_serializing_if = "Option::is_none")] pub display_name: Option, pub endpoint: String, @@ -9121,12 +9070,11 @@ pub enum RuntimeConnectionTestFailureKind { #[derive(Debug, Clone, Serialize, Deserialize, PartialEq, Eq, JsonSchema)] #[cfg_attr(feature = "typescript", derive(ts_rs::TS))] #[serde(deny_unknown_fields)] +/// Observation of an actual signed ping, not a grant or cached approval. pub struct RuntimeConnectionTestResponse { pub workspace_id: String, pub runtime_id: String, pub binding_id: String, - pub connection_state: RuntimeConnectionDisplayState, - pub verification: Option, pub checked_at: String, pub status: RuntimeConnectionTestStatus, pub failure_kind: Option, @@ -12626,18 +12574,22 @@ mod tests { } #[test] - fn runtime_registration_requires_explicit_runtime_issued_trust_identity() { - let mut value = serde_json::json!({ + fn runtime_registration_uses_runtime_bundle_without_workspace_identity_copies() { + let value = serde_json::json!({ "public_bundle": {"identity_id": "runtime", "public_key": "public-key"}, - "endpoint": "https://runtime.example.test", - "workspace_trust_id": "runtime-issued-enrollment" + "endpoint": "https://runtime.example.test" }); let request: CreateRemoteRuntimeRequest = serde_json::from_value(value.clone()).unwrap(); - assert_eq!(request.workspace_trust_id, "runtime-issued-enrollment"); - value.as_object_mut().unwrap().remove("workspace_trust_id"); - assert!(serde_json::from_value::(value.clone()).is_err()); - value["workspace_key_generation"] = serde_json::json!(1); - assert!(serde_json::from_value::(value).is_err()); + assert_eq!(serde_json::to_value(request).unwrap(), value); + for field in [ + "workspace_trust_id", + "workspace_key_id", + "workspace_key_generation", + ] { + let mut obsolete = value.clone(); + obsolete[field] = serde_json::json!("obsolete"); + assert!(serde_json::from_value::(obsolete).is_err()); + } } #[test] @@ -12646,8 +12598,6 @@ mod tests { "workspace_id": "workspace-test", "runtime_id": "runtime-test", "binding_id": "binding-test", - "connection_state": "verified", - "verification": null, "checked_at": "2026-09-01T12:00:00Z", "status": "compatible", "failure_kind": null, @@ -12665,21 +12615,22 @@ mod tests { } #[test] - fn runtime_trust_contract_binds_concrete_keys_and_binding_lifetimes() { - let evidence = RuntimeVerificationEvidenceSummary { - verified_at: Some("2026-10-06T00:00:00Z".into()), - last_checked_at: "2026-10-06T00:00:00Z".into(), - last_outcome: RuntimeVerificationOutcome::Verified, - binding_id: "binding-new".into(), - workspace_key_id: "workspace-key".into(), - workspace_public_key_fingerprint: "sha256:workspace-key".into(), - workspace_trust_id: "trust-new".into(), - runtime_public_key_fingerprint: "sha256:runtime-key".into(), - }; - round_trip(evidence.clone()); - let mut stale = serde_json::to_value(evidence).unwrap(); - stale["unexpected_state_guard"] = serde_json::json!(1); - assert!(serde_json::from_value::(stale).is_err()); + fn runtime_binding_rejects_cached_authentication_and_requires_observed_identity_for_mutation() { + let binding = serde_json::json!({"binding_id": "binding-new", "revoked_at": null}); + let parsed: WorkspaceRuntimeBindingSummary = + serde_json::from_value(binding.clone()).unwrap(); + assert_eq!(serde_json::to_value(parsed).unwrap(), binding); + for field in [ + "state", + "connection_state", + "workspace_key_id", + "workspace_trust_id", + "verification", + ] { + let mut obsolete = binding.clone(); + obsolete[field] = serde_json::json!("obsolete"); + assert!(serde_json::from_value::(obsolete).is_err()); + } round_trip(RevokeRuntimeTrustKeyRequest { expected_binding_id: "binding-observed".into(), }); diff --git a/crates/worker-runtime/README.md b/crates/worker-runtime/README.md index a48822291..e14d955fc 100644 --- a/crates/worker-runtime/README.md +++ b/crates/worker-runtime/README.md @@ -27,4 +27,19 @@ cargo run -p worker-runtime \ ``` The REST server is intended for a trusted Backend/proxy, not direct browser access. +Workspace requests authenticate the actual HTTP request with the Workspace's current +signing identity. The Runtime verifies signatures against its explicitly enrolled +Workspace public key and binds the Runtime audience, Workspace/Worker scope, +operation, method, path/query, body digest, expiry, and single-use token ID. A +connection test is an ordinary signed `/v1/ping`; no prior handshake or saved +verification evidence is required or consulted. + +Keep `trust-workspace add`, `replace`, and `revoke` as explicit operator actions. +A self-reported public key cannot grant trust. Trust record IDs are administrative +only and must not be copied into Backend registration or request claims. Replay +protection uses Workspace plus the trusted key fingerprint, so re-enrolling the +same key or rotating away and back cannot reset consumed requests. Reaccepting a +key does permit fresh, unconsumed requests signed by that key until their short +expiry; it does not create a separate sender-controlled enrollment namespace. + For authenticated remote Runtime setup, see [`../../docs/development/server-runtime-auth.md`](../../docs/development/server-runtime-auth.md). diff --git a/crates/worker-runtime/src/http_server.rs b/crates/worker-runtime/src/http_server.rs index 381afeb0c..e4eb0a2c6 100644 --- a/crates/worker-runtime/src/http_server.rs +++ b/crates/worker-runtime/src/http_server.rs @@ -31,11 +31,7 @@ use crate::ssh_host_key_probe::{ SshHostKeyProbeRequest, SshHostKeyProbeResponse, probe_ssh_host_keys_with_program, }; use crate::workspace_issuer::{ - RuntimeVerificationSigner, VerifiedWorkspaceCapability, WORKSPACE_VERIFICATION_OPERATION, WorkspaceCapabilityExpectation, WorkspaceCapabilityVerifier, - WorkspaceRuntimeVerificationAcknowledgement, WorkspaceRuntimeVerificationAuthority, - WorkspaceRuntimeVerificationChallenge, WorkspaceRuntimeVerificationReceipt, - WorkspaceRuntimeVerificationRecord, WorkspaceRuntimeVerificationResponse, inspect_workspace_capability_claims, workspace_request_body_digest, }; use crate::{Runtime, RuntimeWorkspaceScope}; @@ -256,14 +252,6 @@ fn runtime_http_router_with_auth_and_ssh_keyscan_program( }; let router = Router::new() - .route( - remaining_route(runtime_api::RUNTIME_ROUTE_VERIFICATION_CHALLENGE, &["POST"]), - post(post_workspace_verification_challenge), - ) - .route( - remaining_route(runtime_api::RUNTIME_ROUTE_VERIFICATION_ACK, &["POST"]), - post(post_workspace_verification_acknowledgement), - ) .route( remaining_route(runtime_api::RUNTIME_ROUTE_CONFIG_BUNDLES, &["GET", "POST"]), get(list_config_bundles).post(store_config_bundle), @@ -377,8 +365,7 @@ struct RuntimeHttpState { #[derive(Clone, Debug)] pub struct WorkspaceRuntimeHttpAuth { pub verifier: WorkspaceCapabilityVerifier, - pub signer: RuntimeVerificationSigner, - pub verifications: Arc, + pub runtime_id: String, } struct RuntimeHttpWorkdirSession { @@ -563,172 +550,6 @@ fn unix_now_i64() -> i64 { i64::try_from(unix_now_seconds()).unwrap_or(i64::MAX) } -async fn post_workspace_verification_challenge( - State(state): State, - Extension(verified): Extension, - Json(challenge): Json, -) -> RestResult { - let auth = state.workspace_auth.as_deref().ok_or_else(|| { - RuntimeHttpRestError::new( - StatusCode::NOT_IMPLEMENTED, - "workspace_runtime_verification_unavailable", - "Workspace Runtime verification is not configured", - ) - })?; - if challenge.workspace_id != verified.workspace_id - || challenge.runtime_id != verified.runtime_id - || challenge.binding_id != verified.binding_id - || challenge.workspace_key_id != verified.issuer_key_id - || challenge.workspace_public_key_fingerprint != verified.issuer_public_key_fingerprint - || challenge.workspace_trust_id != verified.trust_id - || challenge.runtime_id != auth.signer.runtime_id() - || challenge.runtime_public_key_fingerprint != auth.signer.public_key_fingerprint() - { - return Err(RuntimeHttpRestError::new( - StatusCode::UNAUTHORIZED, - "workspace_runtime_verification_rejected", - "Workspace Runtime verification challenge does not match authenticated authority", - )); - } - let response = auth - .signer - .sign_response(&challenge, uuid::Uuid::now_v7().to_string(), unix_now_i64()) - .map_err(|error| { - RuntimeHttpRestError::new( - StatusCode::UNAUTHORIZED, - "workspace_runtime_verification_rejected", - error.to_string(), - ) - })?; - auth.verifications.begin(challenge).map_err(|error| { - RuntimeHttpRestError::new( - StatusCode::SERVICE_UNAVAILABLE, - "workspace_runtime_verification_unavailable", - error.to_string(), - ) - })?; - Ok(Json(response)) -} - -async fn post_workspace_verification_acknowledgement( - State(state): State, - Extension(verified): Extension, - Json(acknowledgement): Json, -) -> RestResult { - let auth = state.workspace_auth.as_deref().ok_or_else(|| { - RuntimeHttpRestError::new( - StatusCode::NOT_IMPLEMENTED, - "workspace_runtime_verification_unavailable", - "Workspace Runtime verification is not configured", - ) - })?; - if acknowledgement.workspace_id != verified.workspace_id - || acknowledgement.runtime_id != verified.runtime_id - || acknowledgement.binding_id != verified.binding_id - || acknowledgement.workspace_key_id != verified.issuer_key_id - || acknowledgement.workspace_public_key_fingerprint - != verified.issuer_public_key_fingerprint - || acknowledgement.workspace_trust_id != verified.trust_id - || acknowledgement.runtime_id != auth.signer.runtime_id() - || acknowledgement.runtime_public_key_fingerprint != auth.signer.public_key_fingerprint() - || acknowledgement.expires_at <= unix_now_i64() - || acknowledgement.binding_id.is_empty() - || acknowledgement.workspace_public_key_fingerprint.is_empty() - || acknowledgement.workspace_trust_id.is_empty() - || acknowledgement.workspace_nonce.is_empty() - || acknowledgement.runtime_nonce.is_empty() - || acknowledgement.response_digest.len() != workspace_request_body_digest(&[]).len() - { - return Err(RuntimeHttpRestError::new( - StatusCode::UNAUTHORIZED, - "workspace_runtime_verification_rejected", - "Workspace Runtime verification acknowledgement is invalid", - )); - } - let challenge = WorkspaceRuntimeVerificationChallenge { - challenge_id: acknowledgement.challenge_id.clone(), - workspace_id: acknowledgement.workspace_id.clone(), - runtime_id: acknowledgement.runtime_id.clone(), - binding_id: acknowledgement.binding_id.clone(), - workspace_key_id: acknowledgement.workspace_key_id.clone(), - workspace_public_key_fingerprint: acknowledgement.workspace_public_key_fingerprint.clone(), - workspace_trust_id: acknowledgement.workspace_trust_id.clone(), - runtime_public_key_fingerprint: acknowledgement.runtime_public_key_fingerprint.clone(), - workspace_nonce: acknowledgement.workspace_nonce.clone(), - expires_at: acknowledgement.expires_at, - }; - let response = acknowledgement.response.clone(); - if response.runtime_nonce != acknowledgement.runtime_nonce - || response.workspace_nonce != acknowledgement.workspace_nonce - || response.response_proof.is_empty() - { - return Err(RuntimeHttpRestError::new( - StatusCode::UNAUTHORIZED, - "workspace_runtime_verification_rejected", - "Workspace Runtime verification acknowledgement does not match its response", - )); - } - let response_bytes = serde_json::to_vec(&response).map_err(|_| { - RuntimeHttpRestError::new( - StatusCode::BAD_REQUEST, - "workspace_runtime_verification_rejected", - "Workspace Runtime verification response could not be canonicalized", - ) - })?; - if workspace_request_body_digest(&response_bytes) != acknowledgement.response_digest { - return Err(RuntimeHttpRestError::new( - StatusCode::UNAUTHORIZED, - "workspace_runtime_verification_rejected", - "Workspace Runtime verification acknowledgement has the wrong response digest", - )); - } - crate::workspace_issuer::verify_runtime_verification_response( - &response, - &challenge, - auth.signer.public_key(), - unix_now_i64(), - ) - .map_err(|error| { - RuntimeHttpRestError::new( - StatusCode::UNAUTHORIZED, - "workspace_runtime_verification_rejected", - error.to_string(), - ) - })?; - auth.verifications - .acknowledge( - &acknowledgement.challenge_id, - WorkspaceRuntimeVerificationRecord { - workspace_id: acknowledgement.workspace_id.clone(), - runtime_id: acknowledgement.runtime_id.clone(), - binding_id: acknowledgement.binding_id.clone(), - workspace_key_id: acknowledgement.workspace_key_id.clone(), - workspace_public_key_fingerprint: acknowledgement - .workspace_public_key_fingerprint - .clone(), - workspace_trust_id: acknowledgement.workspace_trust_id.clone(), - runtime_public_key_fingerprint: acknowledgement - .runtime_public_key_fingerprint - .clone(), - verified_at: unix_now_i64(), - }, - ) - .map_err(|error| { - RuntimeHttpRestError::new( - StatusCode::SERVICE_UNAVAILABLE, - "workspace_runtime_verification_unavailable", - error.to_string(), - ) - })?; - Ok(Json(WorkspaceRuntimeVerificationReceipt { - challenge_id: acknowledgement.challenge_id, - workspace_id: acknowledgement.workspace_id, - runtime_id: acknowledgement.runtime_id, - binding_id: acknowledgement.binding_id.clone(), - accepted_at: unix_now_i64(), - })) -} - #[allow(dead_code)] async fn get_runtime_ping( State(state): State, @@ -757,7 +578,7 @@ async fn get_runtime_ping( let runtime_id = state .workspace_auth .as_ref() - .map(|auth| auth.signer.runtime_id().trim()) + .map(|auth| auth.runtime_id.trim()) .filter(|runtime_id| !runtime_id.is_empty()) .ok_or_else(|| { RuntimeHttpRestError::new( @@ -2123,8 +1944,7 @@ async fn require_runtime_auth( let body_digest = workspace_request_body_digest(&body); let expected = WorkspaceCapabilityExpectation { workspace_id: &claims.issuer_workspace_id, - binding_id: &claims.binding_id, - runtime_id: workspace_auth.signer.runtime_id(), + runtime_id: &workspace_auth.runtime_id, worker_id: expected_worker_id.as_deref(), operation: required_permission, method: &method, @@ -2134,48 +1954,6 @@ async fn require_runtime_auth( }; match workspace_auth.verifier.verify(token, &expected) { Ok(verified) => { - let is_verification = path_and_query - == runtime_api::RUNTIME_ROUTE_VERIFICATION_CHALLENGE - || path_and_query == runtime_api::RUNTIME_ROUTE_VERIFICATION_ACK; - if !is_verification { - let record = match workspace_auth - .verifications - .get(&verified.workspace_id, workspace_auth.signer.runtime_id()) - { - Ok(Some(record)) => record, - Ok(None) => { - return RuntimeHttpRestError::new( - StatusCode::FORBIDDEN, - "workspace_runtime_verification_required", - "Workspace Runtime binding has not completed signed verification", - ) - .into_response(); - } - Err(error) => { - return RuntimeHttpRestError::new( - StatusCode::SERVICE_UNAVAILABLE, - "workspace_runtime_verification_unavailable", - error.to_string(), - ) - .into_response(); - } - }; - if record.binding_id != verified.binding_id - || record.workspace_key_id != verified.issuer_key_id - || record.workspace_public_key_fingerprint - != verified.issuer_public_key_fingerprint - || record.workspace_trust_id != verified.trust_id - || record.runtime_public_key_fingerprint - != workspace_auth.signer.public_key_fingerprint() - { - return RuntimeHttpRestError::new( - StatusCode::FORBIDDEN, - "workspace_runtime_verification_stale", - "Workspace Runtime verification does not match current request authority", - ) - .into_response(); - } - } request = Request::from_parts(parts, Body::from(body)); request.extensions_mut().insert(verified.clone()); request.extensions_mut().insert(RuntimeAuthContext { @@ -2199,13 +1977,7 @@ async fn require_runtime_auth( } } - let workspace_bootstrap_request = request.method() == Method::POST - && matches!( - request.uri().path(), - runtime_api::RUNTIME_ROUTE_VERIFICATION_CHALLENGE - | runtime_api::RUNTIME_ROUTE_VERIFICATION_ACK - ); - if state.workspace_auth.is_some() && !workspace_bootstrap_request { + if state.workspace_auth.is_some() { let local_token_matches = state .local_token .as_deref() @@ -2275,12 +2047,6 @@ fn auth_workspace_scope( } fn workspace_runtime_operation(method: &Method, path: &str) -> &'static str { - if (path == runtime_api::RUNTIME_ROUTE_VERIFICATION_CHALLENGE - || path == runtime_api::RUNTIME_ROUTE_VERIFICATION_ACK) - && *method == Method::POST - { - return WORKSPACE_VERIFICATION_OPERATION; - } required_runtime_permission(method, path).unwrap_or("runtime:read") } @@ -2661,12 +2427,10 @@ mod tests { WorkerExecutionResult, WorkerExecutionSpawnRequest, WorkerExecutionSpawnResult, }; use crate::management::RuntimeOptions; - use crate::retention::{DiagnosticsDisposition, SessionDisposition}; use crate::workspace_issuer::{ - InMemoryWorkspaceClaimReplayProtection, InMemoryWorkspaceRuntimeVerificationAuthority, - WorkspaceCapabilityClaims, WorkspaceCapabilityVerifier, WorkspaceIssuerTrustRecord, - WorkspaceIssuerTrustState, issue_workspace_capability_token, - verify_runtime_verification_response, + InMemoryWorkspaceClaimReplayProtection, WorkspaceCapabilityClaims, + WorkspaceCapabilityVerifier, WorkspaceIssuerTrustRecord, WorkspaceIssuerTrustState, + issue_workspace_capability_token, }; use axum::body::to_bytes; use axum::http::Method; @@ -2741,8 +2505,6 @@ mod tests { .map(|route| route.path) .collect::>(); for path in [ - runtime_api::RUNTIME_ROUTE_VERIFICATION_CHALLENGE, - runtime_api::RUNTIME_ROUTE_VERIFICATION_ACK, runtime_api::RUNTIME_ROUTE_CONFIG_BUNDLES, runtime_api::RUNTIME_ROUTE_CONFIG_BUNDLE_AVAILABILITY, runtime_api::RUNTIME_ROUTE_WORKSPACE_PROMPT_PROJECTIONS, @@ -2766,402 +2528,286 @@ mod tests { } } - #[tokio::test] - async fn workspace_signed_verification_requires_exact_request_and_acknowledges_response() { - let runtime = - Runtime::with_execution_backend(RuntimeOptions::default(), Arc::new(AcceptingBackend)) - .unwrap(); - runtime - .store_config_bundle(test_bundle(ProfileSelector::Builtin( - "builtin:coder".to_string(), - ))) - .unwrap(); - let workspace_identity = RuntimeIdentityMaterial::generate("workspace-key").unwrap(); - let runtime_identity = RuntimeIdentityMaterial::generate("runtime-test").unwrap(); - let workspace_public_key = - crate::auth::decode_public_key(&workspace_identity.public_key).unwrap(); - let workspace_fingerprint = format!( - "sha256:{}", - crate::workspace_issuer::hex_lower(&sha2::Sha256::digest(workspace_public_key)) - ); - let runtime_public_key = - crate::auth::decode_public_key(&runtime_identity.public_key).unwrap(); - let runtime_fingerprint = format!( - "sha256:{}", - crate::workspace_issuer::hex_lower(&sha2::Sha256::digest(runtime_public_key)) - ); - let verifier = WorkspaceCapabilityVerifier::new( - vec![WorkspaceIssuerTrustRecord { - workspace_id: "workspace-a".to_string(), - backend_url: "https://backend.test".to_string(), - key_id: "workspace-key".to_string(), - algorithm: "ed25519".to_string(), - public_key: workspace_identity.public_key.clone(), - public_key_fingerprint: workspace_fingerprint.clone(), - trust_id: "trust-test".into(), - state: WorkspaceIssuerTrustState::Active, - registered_at_unix: 1, - updated_at_unix: 1, - }], - Arc::new(InMemoryWorkspaceClaimReplayProtection::default()), - ) - .unwrap(); - let app = runtime_http_router_with_workspace_auth( - runtime.clone(), - None, - WorkspaceRuntimeHttpAuth { - verifier, - signer: RuntimeVerificationSigner::from_identity(&runtime_identity).unwrap(), - verifications: Arc::new(InMemoryWorkspaceRuntimeVerificationAuthority::default()), - }, - ); - let challenge = WorkspaceRuntimeVerificationChallenge { - challenge_id: "challenge-1".to_string(), - workspace_id: "workspace-a".to_string(), - runtime_id: "runtime-test".to_string(), - binding_id: "binding-test".into(), - workspace_key_id: "workspace-key".to_string(), - workspace_public_key_fingerprint: workspace_fingerprint.clone(), - workspace_trust_id: "trust-test".into(), - runtime_public_key_fingerprint: runtime_fingerprint, - workspace_nonce: "workspace-nonce".to_string(), - expires_at: unix_now_i64() + 60, - }; - let body = serde_json::to_vec(&challenge).unwrap(); - let claims = WorkspaceCapabilityClaims { - issuer: "https://backend.test".to_string(), - issuer_workspace_id: "workspace-a".to_string(), - issuer_key_id: "workspace-key".to_string(), - issuer_public_key_fingerprint: workspace_fingerprint.clone(), - trust_id: "trust-test".into(), - binding_id: "binding-test".into(), - runtime_id: "runtime-test".to_string(), - worker_id: None, - operation: WORKSPACE_VERIFICATION_OPERATION.to_string(), - method: "POST".to_string(), - path_and_query: runtime_api::RUNTIME_ROUTE_VERIFICATION_CHALLENGE.to_string(), - body_digest: workspace_request_body_digest(&body), - iat: unix_now_i64(), - exp: challenge.expires_at, - jti: "challenge-token".to_string(), - }; - let token = - issue_workspace_capability_token(&workspace_identity.signing_key().unwrap(), &claims) - .unwrap(); - let response = app - .clone() - .oneshot( - Request::builder() - .method(Method::POST) - .uri(runtime_api::RUNTIME_ROUTE_VERIFICATION_CHALLENGE) - .header(header::AUTHORIZATION, format!("Bearer {token}")) - .header(header::CONTENT_TYPE, "application/json") - .body(Body::from(body)) - .unwrap(), - ) - .await - .unwrap(); - let status = response.status(); - let response_body = to_bytes(response.into_body(), usize::MAX).await.unwrap(); - assert_eq!( - status, - StatusCode::OK, - "{}", - String::from_utf8_lossy(&response_body) - ); - let verification_response = - serde_json::from_slice::(&response_body).unwrap(); - verify_runtime_verification_response( - &verification_response, - &challenge, - &runtime_identity.public_key, - unix_now_i64(), - ) - .unwrap(); - - let acknowledgement = WorkspaceRuntimeVerificationAcknowledgement { - challenge_id: verification_response.challenge_id.clone(), - workspace_id: verification_response.workspace_id.clone(), - runtime_id: verification_response.runtime_id.clone(), - binding_id: verification_response.binding_id.clone(), - workspace_key_id: verification_response.workspace_key_id.clone(), - workspace_public_key_fingerprint: verification_response - .workspace_public_key_fingerprint - .clone(), - workspace_trust_id: verification_response.workspace_trust_id.clone(), - runtime_public_key_fingerprint: verification_response - .runtime_public_key_fingerprint - .clone(), - workspace_nonce: verification_response.workspace_nonce.clone(), - runtime_nonce: verification_response.runtime_nonce.clone(), - response_digest: workspace_request_body_digest(&response_body), - response: verification_response.clone(), - expires_at: verification_response.expires_at, - }; - let acknowledgement_body = serde_json::to_vec(&acknowledgement).unwrap(); - let acknowledgement_claims = WorkspaceCapabilityClaims { - path_and_query: runtime_api::RUNTIME_ROUTE_VERIFICATION_ACK.to_string(), - body_digest: workspace_request_body_digest(&acknowledgement_body), - jti: "ack-token".to_string(), - ..claims - }; - let acknowledgement_token = issue_workspace_capability_token( - &workspace_identity.signing_key().unwrap(), - &acknowledgement_claims, - ) - .unwrap(); - let response = app - .clone() - .oneshot( - Request::builder() - .method(Method::POST) - .uri(runtime_api::RUNTIME_ROUTE_VERIFICATION_ACK) - .header( - header::AUTHORIZATION, - format!("Bearer {acknowledgement_token}"), - ) - .header(header::CONTENT_TYPE, "application/json") - .body(Body::from(acknowledgement_body)) - .unwrap(), - ) - .await - .unwrap(); - let status = response.status(); - let body = to_bytes(response.into_body(), usize::MAX).await.unwrap(); - assert_eq!(status, StatusCode::OK, "{}", String::from_utf8_lossy(&body)); - - for authorization in [None, Some("Bearer malformed")] { - let mut request = Request::builder() - .method(Method::POST) - .uri(runtime_api::RUNTIME_ROUTE_CONFIG_BUNDLES); - if let Some(authorization) = authorization { - request = request.header(header::AUTHORIZATION, authorization); - } - let response = app - .clone() - .oneshot(request.body(Body::empty()).unwrap()) - .await - .unwrap(); - assert_eq!(response.status(), StatusCode::UNAUTHORIZED); - } - - let ping_claims = WorkspaceCapabilityClaims { - operation: RUNTIME_PING_PERMISSION.to_string(), - method: "GET".to_string(), - path_and_query: "/v1/ping".to_string(), - body_digest: workspace_request_body_digest(&[]), - jti: "ping-token".to_string(), - ..acknowledgement_claims - }; - let ping_token = issue_workspace_capability_token( - &workspace_identity.signing_key().unwrap(), - &ping_claims, - ) - .unwrap(); - let response = app - .clone() - .oneshot( - Request::builder() - .method(Method::GET) - .uri("/v1/ping") - .header(header::AUTHORIZATION, format!("Bearer {ping_token}")) - .header(RUNTIME_WORKSPACE_SCOPE_HEADER, "workspace-a") - .body(Body::empty()) - .unwrap(), - ) - .await - .unwrap(); - assert_eq!(response.status(), StatusCode::OK); - - let mut retention_create = task_request("retention-scope"); - retention_create.workspace_api = Some(WorkspaceApiRef { - workspace_id: "workspace-a".to_string(), - base_url: "https://backend.test".to_string(), - }); - let retention_worker_id = retention_create.worker_id.clone(); - runtime.create_worker(retention_create).unwrap(); - let retention_path = format!("/v1/workers/{retention_worker_id}/retention/execute"); - let retention_request = WorkerRetentionExecutionRequest { - operation_id: "retention-op".to_string(), - input_fingerprint: "fingerprint".to_string(), - archive_id: None, - workspace_id: "workspace-b".to_string(), - source_runtime_id: "runtime-a".to_string(), - worker_id: retention_worker_id, - source_created_at: "2025-01-01T00:00:00Z".to_string(), - removed_at: "2025-01-02T00:00:00Z".to_string(), - effective_profile: None, - retention_class: None, - policy_id: "policy".to_string(), - session_disposition: SessionDisposition::Purge, - diagnostics_disposition: DiagnosticsDisposition::Retain, - }; - let retention_body = serde_json::to_vec(&retention_request).unwrap(); - let retention_claims = WorkspaceCapabilityClaims { - operation: required_runtime_permission(&Method::POST, &retention_path) - .unwrap() - .to_string(), - method: "POST".to_string(), - path_and_query: retention_path.clone(), - worker_id: Some(retention_worker_id.to_string()), - body_digest: workspace_request_body_digest(&retention_body), - jti: "retention-scope-token".to_string(), - ..ping_claims - }; - let retention_token = issue_workspace_capability_token( - &workspace_identity.signing_key().unwrap(), - &retention_claims, - ) - .unwrap(); - let response = app - .oneshot( - Request::builder() - .method(Method::POST) - .uri(retention_path) - .header(header::AUTHORIZATION, format!("Bearer {retention_token}")) - .header(header::CONTENT_TYPE, "application/json") - .body(Body::from(retention_body)) - .unwrap(), - ) - .await - .unwrap(); - assert_eq!(response.status(), StatusCode::NOT_FOUND); - let error: RuntimeHttpErrorResponse = read_json(response).await; - assert_eq!(error.error.code, "worker_not_found"); - } - - #[tokio::test] - async fn reenrolled_binding_trust_or_keys_require_fresh_signed_verification() { - let workspace_identity = RuntimeIdentityMaterial::generate("workspace-key").unwrap(); - let runtime_identity = RuntimeIdentityMaterial::generate("runtime-test").unwrap(); - let fingerprint = |key: &str| { - format!( - "sha256:{}", - crate::workspace_issuer::hex_lower(&sha2::Sha256::digest( - crate::auth::decode_public_key(key).unwrap() - )) - ) - }; - let old_trust = WorkspaceIssuerTrustRecord::from_bundle( + fn workspace_trust(identity: &RuntimeIdentityMaterial) -> WorkspaceIssuerTrustRecord { + WorkspaceIssuerTrustRecord::from_bundle( server_api::WorkspacePublicIdentityBundle { workspace_id: "workspace-a".into(), backend_url: "https://backend.test".into(), - key_id: "workspace-key".into(), + key_id: identity.identity_id.clone(), algorithm: "ed25519".into(), - public_key: workspace_identity.public_key.clone(), - public_key_fingerprint: fingerprint(&workspace_identity.public_key), + public_key: identity.public_key.clone(), + public_key_fingerprint: format!( + "sha256:{}", + crate::workspace_issuer::hex_lower(&sha2::Sha256::digest( + crate::auth::decode_public_key(&identity.public_key).unwrap() + )) + ), }, - "old-trust".into(), + "administrative-trust-id".into(), 1, ) - .unwrap(); - let old_verification = WorkspaceRuntimeVerificationRecord { - workspace_id: old_trust.workspace_id.clone(), + .unwrap() + } + + fn signed_claims( + trust: &WorkspaceIssuerTrustRecord, + method: &str, + path: &str, + body: &[u8], + ) -> WorkspaceCapabilityClaims { + WorkspaceCapabilityClaims { + issuer: trust.backend_url.clone(), + issuer_workspace_id: trust.workspace_id.clone(), + issuer_key_id: trust.key_id.clone(), + issuer_public_key_fingerprint: trust.public_key_fingerprint.clone(), runtime_id: "runtime-test".into(), - binding_id: "old-binding".into(), - workspace_key_id: old_trust.key_id.clone(), - workspace_public_key_fingerprint: old_trust.public_key_fingerprint.clone(), - workspace_trust_id: old_trust.trust_id.clone(), - runtime_public_key_fingerprint: fingerprint(&runtime_identity.public_key), - verified_at: 1, - }; - for changed in ["binding", "trust", "workspace_key", "runtime_key"] { - let mut current = old_verification.clone(); - let mut trust = old_trust.clone(); - let mut workspace_key = workspace_identity.clone(); - let mut runtime_key = runtime_identity.clone(); - match changed { - "binding" => current.binding_id = "reenrolled-binding".into(), - "trust" => { - trust.trust_id = "reaccepted-trust".into(); - current.workspace_trust_id = trust.trust_id.clone(); - } - "workspace_key" => { - workspace_key = RuntimeIdentityMaterial::generate("new-workspace-key").unwrap(); - trust.key_id = workspace_key.identity_id.clone(); - trust.public_key = workspace_key.public_key.clone(); - trust.public_key_fingerprint = fingerprint(&workspace_key.public_key); - current.workspace_key_id = trust.key_id.clone(); - current.workspace_public_key_fingerprint = trust.public_key_fingerprint.clone(); - } - "runtime_key" => { - runtime_key = RuntimeIdentityMaterial::generate("runtime-test").unwrap(); - current.runtime_public_key_fingerprint = fingerprint(&runtime_key.public_key); - } - _ => unreachable!(), + worker_id: worker_id_from_runtime_path(path), + operation: workspace_runtime_operation( + &Method::from_bytes(method.as_bytes()).unwrap(), + path, + ) + .into(), + method: method.into(), + path_and_query: path.into(), + body_digest: workspace_request_body_digest(body), + iat: unix_now_i64(), + exp: unix_now_i64() + 60, + jti: uuid::Uuid::now_v7().to_string(), + } + } + + fn workspace_test_auth(records: Vec) -> WorkspaceRuntimeHttpAuth { + WorkspaceRuntimeHttpAuth { + runtime_id: "runtime-test".into(), + verifier: WorkspaceCapabilityVerifier::new( + records, + Arc::new(InMemoryWorkspaceClaimReplayProtection::default()), + ) + .unwrap(), + } + } + + async fn signed_request( + app: Router, + identity: &RuntimeIdentityMaterial, + claims: &WorkspaceCapabilityClaims, + body: Vec, + workspace: &str, + ) -> Response { + let token = + issue_workspace_capability_token(&identity.signing_key().unwrap(), claims).unwrap(); + app.oneshot( + Request::builder() + .method(claims.method.as_str()) + .uri(&claims.path_and_query) + .header(header::AUTHORIZATION, format!("Bearer {token}")) + .header(header::CONTENT_TYPE, "application/json") + .header(RUNTIME_WORKSPACE_SCOPE_HEADER, workspace) + .body(Body::from(body)) + .unwrap(), + ) + .await + .unwrap() + } + + #[tokio::test] + async fn ordinary_signed_ping_succeeds_without_handshake_and_rejects_replay() { + let identity = RuntimeIdentityMaterial::generate("workspace-key").unwrap(); + let trust = workspace_trust(&identity); + let app = runtime_http_router_with_workspace_auth( + Runtime::new_memory(), + None, + workspace_test_auth(vec![trust.clone()]), + ); + let claims = signed_claims(&trust, "GET", "/v1/ping", &[]); + let response = signed_request(app.clone(), &identity, &claims, vec![], "workspace-a").await; + assert_eq!(response.status(), StatusCode::OK); + let ping: runtime_api::RuntimePingResponse = read_json(response).await; + assert_eq!(ping.runtime_id, "runtime-test"); + assert_eq!(ping.protocol_version, RUNTIME_HTTP_PROTOCOL_VERSION); + let response = signed_request(app, &identity, &claims, vec![], "workspace-a").await; + assert_eq!(response.status(), StatusCode::UNAUTHORIZED); + let error: RuntimeHttpErrorResponse = read_json(response).await; + assert_eq!(error.error.code, "unauthorized"); + assert!(error.error.message.contains("already consumed")); + } + + #[tokio::test] + async fn signed_ping_rejects_untrusted_revoked_wrong_key_signature_and_workspace_scope() { + let identity = RuntimeIdentityMaterial::generate("workspace-key").unwrap(); + let other_key = RuntimeIdentityMaterial::generate("other-key").unwrap(); + let trust = workspace_trust(&identity); + for case in [ + "untrusted", + "revoked", + "bad_signature", + "wrong_key", + "wrong_runtime", + "wrong_operation", + "expired", + "wrong_workspace_scope", + "unsigned", + ] { + let mut current = trust.clone(); + if case == "revoked" { + current.state = WorkspaceIssuerTrustState::Revoked; } - let authority = Arc::new(InMemoryWorkspaceRuntimeVerificationAuthority::default()); - authority.record(old_verification.clone()).unwrap(); + let records = if case == "untrusted" { + vec![] + } else { + vec![current] + }; let app = runtime_http_router_with_workspace_auth( Runtime::new_memory(), None, - WorkspaceRuntimeHttpAuth { - verifier: WorkspaceCapabilityVerifier::new( - vec![trust.clone()], - Arc::new(InMemoryWorkspaceClaimReplayProtection::default()), - ) - .unwrap(), - signer: RuntimeVerificationSigner::from_identity(&runtime_key).unwrap(), - verifications: authority.clone(), - }, + workspace_test_auth(records), ); - let claims = WorkspaceCapabilityClaims { - issuer: trust.backend_url.clone(), - issuer_workspace_id: trust.workspace_id.clone(), - issuer_key_id: trust.key_id.clone(), - issuer_public_key_fingerprint: trust.public_key_fingerprint.clone(), - trust_id: trust.trust_id.clone(), - binding_id: current.binding_id.clone(), - runtime_id: current.runtime_id.clone(), - worker_id: None, - operation: RUNTIME_PING_PERMISSION.into(), - method: "GET".into(), - path_and_query: "/v1/ping".into(), - body_digest: workspace_request_body_digest(&[]), - iat: unix_now_i64(), - exp: unix_now_i64() + 60, - jti: "stale-verification".into(), - }; - for (expected, token_id) in [ - (StatusCode::FORBIDDEN, "stale-verification"), - (StatusCode::OK, "fresh-verification"), - ] { - if expected == StatusCode::OK { - authority.record(current.clone()).unwrap(); + let mut claims = signed_claims(&trust, "GET", "/v1/ping", &[]); + match case { + "wrong_key" => claims.issuer_key_id = "untrusted-key".into(), + "wrong_runtime" => claims.runtime_id = "another-runtime".into(), + "wrong_operation" => claims.operation = "workers:create".into(), + "expired" => { + claims.iat -= 120; + claims.exp -= 120; } - let token = issue_workspace_capability_token( - &workspace_key.signing_key().unwrap(), - &WorkspaceCapabilityClaims { - jti: token_id.into(), - ..claims.clone() + _ => {} + } + let response = if case == "unsigned" { + app.oneshot( + Request::builder() + .uri("/v1/ping") + .body(Body::empty()) + .unwrap(), + ) + .await + .unwrap() + } else { + signed_request( + app, + if case == "bad_signature" { + &other_key + } else { + &identity + }, + &claims, + vec![], + if case == "wrong_workspace_scope" { + "workspace-b" + } else { + "workspace-a" }, ) - .unwrap(); - let response = app - .clone() - .oneshot( - Request::builder() - .uri("/v1/ping") - .header(header::AUTHORIZATION, format!("Bearer {token}")) - .header(RUNTIME_WORKSPACE_SCOPE_HEADER, "workspace-a") - .body(Body::empty()) - .unwrap(), - ) - .await - .unwrap(); - let status = response.status(); - let body = to_bytes(response.into_body(), usize::MAX).await.unwrap(); - assert_eq!( - status, - expected, - "{changed}: {}", - String::from_utf8_lossy(&body) - ); - if expected == StatusCode::FORBIDDEN { - let error: RuntimeHttpErrorResponse = serde_json::from_slice(&body).unwrap(); - assert_eq!(error.error.code, "workspace_runtime_verification_stale"); - } - } + .await + }; + let status = response.status(); + let bytes = to_bytes(response.into_body(), usize::MAX).await.unwrap(); + assert_eq!( + status, + if case == "wrong_workspace_scope" { + StatusCode::FORBIDDEN + } else { + StatusCode::UNAUTHORIZED + }, + "{case}: {}", + String::from_utf8_lossy(&bytes) + ); + let error: RuntimeHttpErrorResponse = serde_json::from_slice(&bytes).unwrap(); + assert_eq!( + error.error.code, + if case == "wrong_workspace_scope" { + "runtime_ping_workspace_scope_mismatch" + } else { + "unauthorized" + }, + "{case}" + ); + } + } + + #[tokio::test] + async fn ordinary_signed_workdir_operation_succeeds_without_evidence_and_rejects_changed_body() + { + let temp = tempfile::tempdir().unwrap(); + std::fs::write(temp.path().join("hello.txt"), "hello").unwrap(); + let identity = RuntimeIdentityMaterial::generate("workspace-key").unwrap(); + let trust = workspace_trust(&identity); + let session: WorkdirSessionHandle = Arc::new(LocalWorkdirSession::materialized_bound( + Workdir::new("wd-1"), + temp.path().to_path_buf(), + temp.path().to_path_buf(), + SharedScope::new(Scope::writable(temp.path()).unwrap()), + WorkdirSessionCapabilities::ALL, + )); + let state = RuntimeHttpState { + runtime: Runtime::new_memory(), + local_token: None, + workspace_auth: Some(Arc::new(workspace_test_auth(vec![trust.clone()]))), + workdir_sessions: Arc::new(Mutex::new(HashMap::from([( + "session-1".into(), + RuntimeHttpWorkdirSession { + owner: RuntimeWorkspaceScope::new("workspace-a", "https://backend.test"), + session, + }, + )]))), + ssh_keyscan_program: Arc::new(PathBuf::from("ssh-keyscan")), + }; + let app = Router::new() + .route( + "/v1/workdir-sessions/{session_id}/operations", + post(run_workdir_session_operation), + ) + .with_state(state.clone()) + .layer(middleware::from_fn_with_state(state, require_runtime_auth)); + let body = serde_json::to_vec(&WorkdirSessionOperationRequest { + operation: WorkdirSessionOperation::Stat(StatRequest { + path: WorkdirPath::new("hello.txt").unwrap(), + }), + }) + .unwrap(); + let claims = signed_claims( + &trust, + "POST", + "/v1/workdir-sessions/session-1/operations", + &body, + ); + let response = signed_request(app.clone(), &identity, &claims, body, "workspace-a").await; + assert_eq!(response.status(), StatusCode::OK); + let result: WorkdirSessionOperationResult = read_json(response).await; + assert!(matches!(result, WorkdirSessionOperationResult::Stat(_))); + let claims = WorkspaceCapabilityClaims { + jti: uuid::Uuid::now_v7().to_string(), + ..claims + }; + let response = signed_request(app, &identity, &claims, b"{}".to_vec(), "workspace-a").await; + assert_eq!(response.status(), StatusCode::UNAUTHORIZED); + let error: RuntimeHttpErrorResponse = read_json(response).await; + assert!(error.error.message.contains("request body")); + } + + #[tokio::test] + async fn removed_handshake_routes_are_not_network_aliases() { + let identity = RuntimeIdentityMaterial::generate("workspace-key").unwrap(); + let trust = workspace_trust(&identity); + let app = runtime_http_router_with_workspace_auth( + Runtime::new_memory(), + None, + workspace_test_auth(vec![trust.clone()]), + ); + for path in [ + "/v1/workspace-runtime-verification/challenge", + "/v1/workspace-runtime-verification/acknowledgement", + ] { + let claims = signed_claims(&trust, "POST", path, b"{}"); + let response = signed_request( + app.clone(), + &identity, + &claims, + b"{}".to_vec(), + "workspace-a", + ) + .await; + assert_eq!(response.status(), StatusCode::NOT_FOUND); } } @@ -4630,8 +4276,7 @@ mod tests { Arc::new(InMemoryWorkspaceClaimReplayProtection::default()), ) .unwrap(), - signer: RuntimeVerificationSigner::from_identity(&identity).unwrap(), - verifications: Arc::new(InMemoryWorkspaceRuntimeVerificationAuthority::default()), + runtime_id: identity.identity_id.clone(), }, ); (runtime, app) diff --git a/crates/worker-runtime/src/http_server/runtime_management_api.rs b/crates/worker-runtime/src/http_server/runtime_management_api.rs index b92e22731..d8a919433 100644 --- a/crates/worker-runtime/src/http_server/runtime_management_api.rs +++ b/crates/worker-runtime/src/http_server/runtime_management_api.rs @@ -164,7 +164,7 @@ impl runtime_api::RuntimeApi for RuntimeManagementApi { .state .workspace_auth .as_ref() - .map(|auth| auth.signer.runtime_id().trim()) + .map(|auth| auth.runtime_id.trim()) .filter(|runtime_id| !runtime_id.is_empty()) .ok_or_else(|| { runtime_api::RuntimeApiError::new( diff --git a/crates/worker-runtime/src/main.rs b/crates/worker-runtime/src/main.rs index 08cdb7e64..d8b985f43 100644 --- a/crates/worker-runtime/src/main.rs +++ b/crates/worker-runtime/src/main.rs @@ -27,11 +27,11 @@ use worker_runtime::http_server::{ use worker_runtime::worker_backend::{ProfileRuntimeWorkerFactory, WorkerRuntimeExecutionBackend}; use worker_runtime::working_directory::RuntimeGitMaterializer; use worker_runtime::workspace_issuer::{ - FileWorkspaceClaimReplayProtection, FileWorkspaceRuntimeVerificationAuthority, - MAX_WORKSPACE_ISSUER_TRUST_RECORDS, RuntimeVerificationSigner, WorkspaceCapabilityVerifier, - WorkspaceIssuerTrustError, WorkspaceIssuerTrustMutation, WorkspaceIssuerTrustRecord, - WorkspaceIssuerTrustState, add_workspace_issuer_trust, replace_workspace_issuer_trust, - revoke_workspace_issuer_trust, validate_workspace_issuer_trust_records, + FileWorkspaceClaimReplayProtection, MAX_WORKSPACE_ISSUER_TRUST_RECORDS, + WorkspaceCapabilityVerifier, WorkspaceIssuerTrustError, WorkspaceIssuerTrustMutation, + WorkspaceIssuerTrustRecord, WorkspaceIssuerTrustState, add_workspace_issuer_trust, + replace_workspace_issuer_trust, revoke_workspace_issuer_trust, + validate_workspace_issuer_trust_records, }; use worker_runtime::workspace_request::RuntimeWorkspaceRequestClient; use worker_runtime::{Runtime, RuntimeOptions}; @@ -972,7 +972,7 @@ fn read_runtime_auth_file(path: &Path) -> Result validate_workspace_issuer_trust_records(&auth.workspace_issuers) .map_err(|_| ProcessError::auth("runtime Workspace issuer trust store is corrupt"))?; if migrated { - // Commit the revoked migration before exposing any issuer authority. + // Persist the validated migration without changing saved trust permissions. write_runtime_auth_file(path, &auth)?; } Ok(auth) @@ -1025,25 +1025,15 @@ fn load_workspace_runtime_http_auth( let Some(identity) = auth.identity else { return Ok(None); }; - if auth.workspace_issuers.is_empty() { - return Ok(None); - } let replay_path = runtime_auth_path(config).with_extension("workspace-replay.json"); let verifier = WorkspaceCapabilityVerifier::new( auth.workspace_issuers, Arc::new(FileWorkspaceClaimReplayProtection::new(replay_path)), ) .map_err(|error| ProcessError::auth(format!("invalid Workspace issuer trust: {error}")))?; - let signer = RuntimeVerificationSigner::from_identity(&identity) - .map_err(|error| ProcessError::auth(format!("invalid Runtime identity: {error}")))?; - let verifications_path = - runtime_auth_path(config).with_extension("workspace-verifications.json"); Ok(Some(WorkspaceRuntimeHttpAuth { verifier, - signer, - verifications: Arc::new(FileWorkspaceRuntimeVerificationAuthority::new( - verifications_path, - )), + runtime_id: identity.identity_id, })) } @@ -1407,6 +1397,34 @@ mod tests { assert_eq!(std::fs::read(root.join("runtime.json")).unwrap(), before); } + #[test] + fn runtime_identity_with_no_trusted_workspace_does_not_disable_request_authentication() { + let temp = tempfile::tempdir().unwrap(); + let config = ProcessConfig { + fs_root: Some(temp.path().to_path_buf()), + ..ProcessConfig::default().unwrap() + }; + let path = runtime_auth_path(&config); + write_runtime_auth_file( + &path, + &RuntimeAuthFile { + identity: Some(RuntimeIdentityMaterial::generate("runtime-test").unwrap()), + ..RuntimeAuthFile::default() + }, + ) + .unwrap(); + // Obsolete evidence is neither a startup prerequisite nor live authority. + let evidence = path.with_extension("workspace-verifications.json"); + std::fs::write(&evidence, "obsolete evidence, not even valid JSON").unwrap(); + let auth = load_workspace_runtime_http_auth(&config).unwrap().unwrap(); + assert_eq!(auth.runtime_id, "runtime-test"); + assert!(!auth.verifier.has_active_workspace_issuer("workspace-a")); + assert_eq!( + std::fs::read_to_string(&evidence).unwrap(), + "obsolete evidence, not even valid JSON" + ); + } + #[test] fn workspace_issuer_trust_cli_persists_across_reload_and_writes_private_mode() { use server_api::WorkspacePublicIdentityBundle; @@ -1454,38 +1472,6 @@ mod tests { let first = read_runtime_auth_file(&path).unwrap(); assert_eq!(first.workspace_issuers.len(), 1); assert!(!first.workspace_issuers[0].trust_id.is_empty()); - // `trust-workspace show` serializes this persisted record. Owner - // registration must forward that exact enrollment ID, not derive one - // from the key fingerprint or an initial counter. - let shown = serde_json::to_value(&first.workspace_issuers[0]).unwrap(); - let runtime_identity = RuntimeIdentityMaterial::generate("remote-runtime").unwrap(); - let mut registration_wire = serde_json::json!({ - "public_bundle": { - "identity_id": runtime_identity.identity_id, - "public_key": runtime_identity.public_key, - }, - "endpoint": "https://runtime.example.test", - "workspace_trust_id": shown["trust_id"], - }); - let registration: server_api::CreateRemoteRuntimeRequest = - serde_json::from_value(registration_wire.clone()).unwrap(); - assert_eq!( - registration.workspace_trust_id, - first.workspace_issuers[0].trust_id - ); - assert_ne!( - registration.workspace_trust_id, - first.workspace_issuers[0].public_key_fingerprint - ); - registration_wire - .as_object_mut() - .unwrap() - .remove("workspace_trust_id"); - assert!( - serde_json::from_value::(registration_wire) - .is_err() - ); - run_trust_workspace_command(VecDeque::from([ "add".to_string(), "--bundle".to_string(), @@ -1645,9 +1631,14 @@ mod tests { assert_eq!(error, "invalid trust-workspace list arguments"); } - #[test] - fn pre_id_issuer_migration_retains_history_without_restoring_authority() { + #[tokio::test] + async fn pre_id_issuer_migration_preserves_saved_permission_for_ordinary_signed_ping() { use sha2::{Digest as _, Sha256}; + use tower::ServiceExt; + use worker_runtime::workspace_issuer::{ + WorkspaceCapabilityClaims, issue_workspace_capability_token, + workspace_request_body_digest, + }; let identity = RuntimeIdentityMaterial::generate("WK-legacy").unwrap(); let public_key = decode_public_key(&identity.public_key).unwrap(); @@ -1680,16 +1671,97 @@ mod tests { std::fs::write(&path, toml::to_string(&document).unwrap()).unwrap(); let migrated = read_runtime_auth_file(&path).unwrap(); - assert_eq!( - migrated.workspace_issuers[0].state, - worker_runtime::workspace_issuer::WorkspaceIssuerTrustState::Revoked + let expected_state = if state == "active" { + WorkspaceIssuerTrustState::Active + } else { + WorkspaceIssuerTrustState::Revoked + }; + let record = &migrated.workspace_issuers[0]; + assert_eq!(record.state, expected_state); + assert_eq!(record.workspace_id, "workspace-1"); + assert_eq!(record.backend_url, "https://backend.example.test"); + assert_eq!(record.key_id, "WK-legacy"); + assert_eq!(record.public_key, identity.public_key); + assert_eq!(record.public_key_fingerprint, fingerprint); + assert_eq!(record.registered_at_unix, 10); + assert_eq!(record.updated_at_unix, 20); + // Migration is durably saved by the loader itself, not just by a later write. + assert_eq!(read_runtime_auth_file(&path).unwrap(), migrated); + let claims = WorkspaceCapabilityClaims { + issuer: record.backend_url.clone(), + issuer_workspace_id: record.workspace_id.clone(), + issuer_key_id: record.key_id.clone(), + issuer_public_key_fingerprint: record.public_key_fingerprint.clone(), + runtime_id: "runtime-test".into(), + worker_id: None, + operation: worker_runtime::http_server::RUNTIME_PING_PERMISSION.into(), + method: "GET".into(), + path_and_query: "/v1/ping".into(), + body_digest: workspace_request_body_digest(&[]), + iat: i64::try_from(worker_runtime::auth::unix_now_seconds()).unwrap(), + exp: i64::try_from(worker_runtime::auth::unix_now_seconds()).unwrap() + 60, + jti: uuid::Uuid::now_v7().to_string(), + }; + let token = issue_workspace_capability_token(&identity.signing_key().unwrap(), &claims) + .unwrap(); + let app = worker_runtime::http_server::runtime_http_router_with_workspace_auth( + Runtime::new_memory(), None, WorkspaceRuntimeHttpAuth { + runtime_id: "runtime-test".into(), + verifier: WorkspaceCapabilityVerifier::new(migrated.workspace_issuers.clone(), + Arc::new(worker_runtime::workspace_issuer::InMemoryWorkspaceClaimReplayProtection::default())).unwrap(), + }, ); + let response = app + .oneshot( + axum::http::Request::builder() + .uri("/v1/ping") + .header(axum::http::header::AUTHORIZATION, format!("Bearer {token}")) + .header( + worker_runtime::http_server::RUNTIME_WORKSPACE_SCOPE_HEADER, + "workspace-1", + ) + .body(axum::body::Body::empty()) + .unwrap(), + ) + .await + .unwrap(); + let status = response.status(); + let body = axum::body::to_bytes(response.into_body(), usize::MAX) + .await + .unwrap(); + assert_eq!( + status, + if state == "active" { + axum::http::StatusCode::OK + } else { + axum::http::StatusCode::UNAUTHORIZED + }, + "{state}: {}", + String::from_utf8_lossy(&body) + ); + if state == "active" { + let ping: runtime_api::RuntimePingResponse = serde_json::from_slice(&body).unwrap(); + assert_eq!(ping.runtime_id, "runtime-test"); + } else { + let error: worker_runtime::http_server::RuntimeHttpErrorResponse = + serde_json::from_slice(&body).unwrap(); + assert_eq!(error.error.code, "unauthorized"); + assert!(error.error.message.contains("revoked")); + } assert_eq!(migrated.archived_legacy_issuers.len(), 1); let archived: toml::Value = toml::from_str(&migrated.archived_legacy_issuers[0]).unwrap(); assert_eq!(archived, original_record); write_runtime_auth_file(&path, &migrated).unwrap(); assert_eq!(read_runtime_auth_file(&path).unwrap(), migrated); + if state == "active" { + // A current saved Revoked record stays revoked even if its archived + // pre-ID source was Active. Never guess revocation provenance. + let mut saved_revoked = migrated.clone(); + saved_revoked.workspace_issuers[0].state = WorkspaceIssuerTrustState::Revoked; + write_runtime_auth_file(&path, &saved_revoked).unwrap(); + assert_eq!(read_runtime_auth_file(&path).unwrap(), saved_revoked); + } } } diff --git a/crates/worker-runtime/src/workspace_issuer.rs b/crates/worker-runtime/src/workspace_issuer.rs index 53d706680..1ae60192f 100644 --- a/crates/worker-runtime/src/workspace_issuer.rs +++ b/crates/worker-runtime/src/workspace_issuer.rs @@ -1,4 +1,4 @@ -use std::collections::{HashMap, HashSet}; +use std::collections::HashMap; use std::fmt; use std::fs; use std::path::{Path, PathBuf}; @@ -13,6 +13,9 @@ use sha2::{Digest, Sha256}; use crate::auth::RuntimeAuthError; +// The v2 signing domain remains, but only the current deny_unknown_fields claim +// shape is accepted. Old binding/trust fields fail decoding even with a valid +// signature; retaining the prefix is not a network compatibility decoder. const WORKSPACE_TOKEN_PREFIX: &str = "yoi-workspace-v2"; const WORKSPACE_SIGNING_INPUT_PREFIX: &str = "yoi.workspace.capability.v2."; const WORKSPACE_SIGNING_ALGORITHM: &str = "ed25519"; @@ -82,7 +85,8 @@ impl WorkspaceIssuerTrustRecord { } } -// Frozen pre-ID enrollment format. It is never accepted as live authority. +// Frozen pre-ID saved enrollment format. Only persisted-data migration reads it; +// network requests never supply trust records or authorize their own migration. #[derive(Deserialize)] #[serde(deny_unknown_fields)] struct LegacyWorkspaceIssuerTrustRecord { @@ -101,10 +105,10 @@ struct LegacyWorkspaceIssuerTrustRecord { updated_at_unix: i64, } -/// Migrate a pre-ID enrollment as revoked. At persisted-data decode, legacy counters -/// must be positive but are never compared with current identity or trust. -/// Only an explicit operator replacement can accept new trust; no legacy -/// capability can authorize its own migration. +/// Migrate validated saved trust without changing its Active/Revoked permission. +/// Legacy counters must be positive at persisted-data decode but are never +/// compared with current identity or trust. The generated trust ID is only an +/// administrative identity; it neither grants nor revokes request authorization. pub fn migrate_legacy_workspace_issuer_trust_record( value: serde_json::Value, ) -> Result { @@ -125,9 +129,7 @@ pub fn migrate_legacy_workspace_issuer_trust_record( uuid::Uuid::now_v7().to_string(), legacy.registered_at_unix, )?; - // Both old Active and Revoked records require new enrollment acceptance. - let _old_state = legacy.state; - record.state = WorkspaceIssuerTrustState::Revoked; + record.state = legacy.state; record.updated_at_unix = legacy.updated_at_unix; Ok(record) } @@ -300,664 +302,6 @@ fn validate_id(value: &str) -> Result<(), WorkspaceIssuerTrustError> { Ok(()) } -pub const WORKSPACE_VERIFICATION_CHALLENGE_PATH: &str = - "/v1/workspace-runtime-verification/challenge"; -pub const WORKSPACE_VERIFICATION_ACK_PATH: &str = - "/v1/workspace-runtime-verification/acknowledgement"; -pub const WORKSPACE_VERIFICATION_OPERATION: &str = "workspace.runtime.verify"; -const RUNTIME_VERIFICATION_TOKEN_PREFIX: &str = "yoi-runtime-verification-v2"; -const RUNTIME_VERIFICATION_SIGNING_INPUT_PREFIX: &str = "yoi.runtime.verification.v2."; - -#[derive(Clone, Debug, PartialEq, Eq, Serialize, Deserialize)] -#[serde(deny_unknown_fields)] -pub struct WorkspaceRuntimeVerificationChallenge { - pub challenge_id: String, - pub workspace_id: String, - pub runtime_id: String, - pub binding_id: String, - pub workspace_key_id: String, - pub workspace_public_key_fingerprint: String, - pub workspace_trust_id: String, - pub runtime_public_key_fingerprint: String, - pub workspace_nonce: String, - pub expires_at: i64, -} - -#[derive(Clone, Debug, PartialEq, Eq, Serialize, Deserialize)] -#[serde(deny_unknown_fields)] -pub struct WorkspaceRuntimeVerificationResponse { - pub challenge_id: String, - pub workspace_id: String, - pub runtime_id: String, - pub binding_id: String, - pub workspace_key_id: String, - pub workspace_public_key_fingerprint: String, - pub workspace_trust_id: String, - pub runtime_public_key_fingerprint: String, - pub workspace_nonce: String, - pub runtime_nonce: String, - pub expires_at: i64, - pub response_proof: String, -} - -#[derive(Clone, Debug, PartialEq, Eq, Serialize, Deserialize)] -#[serde(deny_unknown_fields)] -pub struct WorkspaceRuntimeVerificationAcknowledgement { - pub challenge_id: String, - pub workspace_id: String, - pub runtime_id: String, - pub binding_id: String, - pub workspace_key_id: String, - pub workspace_public_key_fingerprint: String, - pub workspace_trust_id: String, - pub runtime_public_key_fingerprint: String, - pub workspace_nonce: String, - pub runtime_nonce: String, - pub response_digest: String, - pub response: WorkspaceRuntimeVerificationResponse, - pub expires_at: i64, -} - -#[derive(Clone, Debug, PartialEq, Eq, Serialize, Deserialize)] -#[serde(deny_unknown_fields)] -pub struct WorkspaceRuntimeVerificationReceipt { - pub challenge_id: String, - pub workspace_id: String, - pub runtime_id: String, - pub binding_id: String, - pub accepted_at: i64, -} - -#[derive(Clone, Debug, PartialEq, Eq, Serialize, Deserialize)] -#[serde(deny_unknown_fields)] -pub struct WorkspaceRuntimeVerificationRecord { - pub workspace_id: String, - pub runtime_id: String, - pub binding_id: String, - pub workspace_key_id: String, - pub workspace_public_key_fingerprint: String, - pub workspace_trust_id: String, - pub runtime_public_key_fingerprint: String, - pub verified_at: i64, -} - -pub trait WorkspaceRuntimeVerificationAuthority: fmt::Debug + Send + Sync { - fn begin( - &self, - challenge: WorkspaceRuntimeVerificationChallenge, - ) -> Result<(), WorkspaceCapabilityVerificationError>; - fn acknowledge( - &self, - challenge_id: &str, - record: WorkspaceRuntimeVerificationRecord, - ) -> Result<(), WorkspaceCapabilityVerificationError>; - fn get( - &self, - workspace_id: &str, - runtime_id: &str, - ) -> Result, WorkspaceCapabilityVerificationError>; - fn record( - &self, - record: WorkspaceRuntimeVerificationRecord, - ) -> Result<(), WorkspaceCapabilityVerificationError>; -} - -#[derive(Debug, Default)] -pub struct InMemoryWorkspaceRuntimeVerificationAuthority { - state: Mutex, -} - -impl WorkspaceRuntimeVerificationAuthority for InMemoryWorkspaceRuntimeVerificationAuthority { - fn begin( - &self, - challenge: WorkspaceRuntimeVerificationChallenge, - ) -> Result<(), WorkspaceCapabilityVerificationError> { - let mut state = self - .state - .lock() - .map_err(|_| WorkspaceCapabilityVerificationError::VerificationAuthorityUnavailable)?; - begin_verification(&mut state, challenge) - } - fn acknowledge( - &self, - challenge_id: &str, - record: WorkspaceRuntimeVerificationRecord, - ) -> Result<(), WorkspaceCapabilityVerificationError> { - let mut state = self - .state - .lock() - .map_err(|_| WorkspaceCapabilityVerificationError::VerificationAuthorityUnavailable)?; - acknowledge_verification(&mut state, challenge_id, record) - } - fn get( - &self, - workspace_id: &str, - runtime_id: &str, - ) -> Result, WorkspaceCapabilityVerificationError> - { - let state = self - .state - .lock() - .map_err(|_| WorkspaceCapabilityVerificationError::VerificationAuthorityUnavailable)?; - Ok(state - .records - .iter() - .find(|record| record.workspace_id == workspace_id && record.runtime_id == runtime_id) - .cloned()) - } - fn record( - &self, - record: WorkspaceRuntimeVerificationRecord, - ) -> Result<(), WorkspaceCapabilityVerificationError> { - validate_verification_record(&record)?; - let mut state = self - .state - .lock() - .map_err(|_| WorkspaceCapabilityVerificationError::VerificationAuthorityUnavailable)?; - state.records.retain(|current| { - current.workspace_id != record.workspace_id || current.runtime_id != record.runtime_id - }); - if state.records.len() >= MAX_WORKSPACE_ISSUER_TRUST_RECORDS { - return Err(WorkspaceCapabilityVerificationError::TrustRecordLimitExceeded); - } - state.records.push(record); - Ok(()) - } -} - -#[derive(Clone, Debug)] -pub struct FileWorkspaceRuntimeVerificationAuthority { - path: PathBuf, - lock: Arc>, -} - -#[derive(Debug, Default, Serialize, Deserialize)] -#[serde(deny_unknown_fields)] -struct WorkspaceRuntimeVerificationDocument { - version: u32, - records: Vec, - #[serde(default)] - pending: Vec, - /// Exact pre-ID document retained only as history; never used to authorize requests. - #[serde(default, skip_serializing_if = "Option::is_none")] - archived_legacy_document: Option, -} - -// Frozen verification evidence from before trust/binding IDs. Never promote -// counters into IDs or keep a pre-ID acknowledgement usable after migration. -#[derive(Deserialize)] -#[serde(deny_unknown_fields)] -struct LegacyVerificationDocument { - version: u32, - records: Vec, -} -#[derive(Deserialize)] -#[serde(deny_unknown_fields)] -#[allow(dead_code)] -struct LegacyVerificationRecord { - workspace_id: String, - runtime_id: String, - #[serde(rename = "binding_revision")] - legacy_binding_counter: u64, - workspace_key_id: String, - #[serde(rename = "workspace_identity_revision")] - legacy_workspace_identity_counter: u64, - #[serde(rename = "workspace_trust_generation")] - legacy_workspace_trust_counter: u64, - runtime_public_key_fingerprint: String, - #[serde(rename = "runtime_identity_revision")] - legacy_runtime_identity_counter: u64, - verified_at: i64, -} - -impl FileWorkspaceRuntimeVerificationAuthority { - pub fn new(path: impl Into) -> Self { - Self { - path: path.into(), - lock: Arc::new(Mutex::new(())), - } - } - - // No pre-ID acknowledgement is live authority. Keep its exact source bytes - // in the persisted archive while new challenges populate only current records. - // Neither read(), get(), nor capability verification consults that archive. - fn read( - &self, - ) -> Result { - match fs::read(&self.path) { - Ok(bytes) => { - let raw: serde_json::Value = serde_json::from_slice(&bytes).map_err(|_| { - WorkspaceCapabilityVerificationError::VerificationAuthorityUnavailable - })?; - if raw.get("version").and_then(serde_json::Value::as_u64) == Some(1) { - let legacy: LegacyVerificationDocument = - serde_json::from_value(raw).map_err(|_| { - WorkspaceCapabilityVerificationError::VerificationAuthorityUnavailable - })?; - if legacy.version != 1 - || legacy.records.len() > MAX_WORKSPACE_ISSUER_TRUST_RECORDS - { - return Err( - WorkspaceCapabilityVerificationError::VerificationAuthorityUnavailable, - ); - } - let migrated = WorkspaceRuntimeVerificationDocument { - version: 2, - records: Vec::new(), - pending: Vec::new(), - archived_legacy_document: Some(String::from_utf8(bytes).map_err(|_| { - WorkspaceCapabilityVerificationError::VerificationAuthorityUnavailable - })?), - }; - self.write(&migrated)?; - return Ok(migrated); - } - let document: WorkspaceRuntimeVerificationDocument = serde_json::from_value(raw) - .map_err(|_| { - WorkspaceCapabilityVerificationError::VerificationAuthorityUnavailable - })?; - if document.version != 2 - || document.records.len() > MAX_WORKSPACE_ISSUER_TRUST_RECORDS - || document.pending.len() > MAX_WORKSPACE_ISSUER_TRUST_RECORDS - { - return Err( - WorkspaceCapabilityVerificationError::VerificationAuthorityUnavailable, - ); - } - let mut identities = HashSet::new(); - for record in &document.records { - validate_verification_record(record)?; - if !identities - .insert((record.workspace_id.as_str(), record.runtime_id.as_str())) - { - return Err( - WorkspaceCapabilityVerificationError::VerificationAuthorityUnavailable, - ); - } - } - let mut pending_identities = HashSet::new(); - for challenge in &document.pending { - validate_verification_challenge(challenge)?; - if !pending_identities.insert((&challenge.workspace_id, &challenge.runtime_id)) - { - return Err( - WorkspaceCapabilityVerificationError::VerificationAuthorityUnavailable, - ); - } - } - Ok(document) - } - Err(error) if error.kind() == std::io::ErrorKind::NotFound => { - Ok(WorkspaceRuntimeVerificationDocument { - version: 2, - records: Vec::new(), - pending: Vec::new(), - archived_legacy_document: None, - }) - } - Err(_) => Err(WorkspaceCapabilityVerificationError::VerificationAuthorityUnavailable), - } - } - - fn write( - &self, - document: &WorkspaceRuntimeVerificationDocument, - ) -> Result<(), WorkspaceCapabilityVerificationError> { - let parent = self.path.parent().unwrap_or_else(|| Path::new(".")); - fs::create_dir_all(parent) - .map_err(|_| WorkspaceCapabilityVerificationError::VerificationAuthorityUnavailable)?; - let temporary = parent.join(format!( - ".workspace-runtime-verification-{}.tmp", - uuid::Uuid::now_v7() - )); - let bytes = serde_json::to_vec(document) - .map_err(|_| WorkspaceCapabilityVerificationError::VerificationAuthorityUnavailable)?; - fs::write(&temporary, bytes) - .map_err(|_| WorkspaceCapabilityVerificationError::VerificationAuthorityUnavailable)?; - #[cfg(unix)] - { - use std::os::unix::fs::PermissionsExt as _; - fs::set_permissions(&temporary, fs::Permissions::from_mode(0o600)).map_err(|_| { - WorkspaceCapabilityVerificationError::VerificationAuthorityUnavailable - })?; - } - fs::rename(&temporary, &self.path).map_err(|_| { - let _ = fs::remove_file(&temporary); - WorkspaceCapabilityVerificationError::VerificationAuthorityUnavailable - })?; - Ok(()) - } -} - -impl WorkspaceRuntimeVerificationAuthority for FileWorkspaceRuntimeVerificationAuthority { - fn begin( - &self, - challenge: WorkspaceRuntimeVerificationChallenge, - ) -> Result<(), WorkspaceCapabilityVerificationError> { - let _guard = self - .lock - .lock() - .map_err(|_| WorkspaceCapabilityVerificationError::VerificationAuthorityUnavailable)?; - let mut document = self.read()?; - begin_verification(&mut document, challenge)?; - self.write(&document) - } - fn acknowledge( - &self, - challenge_id: &str, - record: WorkspaceRuntimeVerificationRecord, - ) -> Result<(), WorkspaceCapabilityVerificationError> { - let _guard = self - .lock - .lock() - .map_err(|_| WorkspaceCapabilityVerificationError::VerificationAuthorityUnavailable)?; - let mut document = self.read()?; - acknowledge_verification(&mut document, challenge_id, record)?; - self.write(&document) - } - - fn get( - &self, - workspace_id: &str, - runtime_id: &str, - ) -> Result, WorkspaceCapabilityVerificationError> - { - let _guard = self - .lock - .lock() - .map_err(|_| WorkspaceCapabilityVerificationError::VerificationAuthorityUnavailable)?; - Ok(self - .read()? - .records - .into_iter() - .find(|record| record.workspace_id == workspace_id && record.runtime_id == runtime_id)) - } - - fn record( - &self, - record: WorkspaceRuntimeVerificationRecord, - ) -> Result<(), WorkspaceCapabilityVerificationError> { - validate_verification_record(&record)?; - let _guard = self - .lock - .lock() - .map_err(|_| WorkspaceCapabilityVerificationError::VerificationAuthorityUnavailable)?; - let mut document = self.read()?; - document.records.retain(|current| { - current.workspace_id != record.workspace_id || current.runtime_id != record.runtime_id - }); - document.records.push(record); - self.write(&document) - } -} - -fn begin_verification( - document: &mut WorkspaceRuntimeVerificationDocument, - challenge: WorkspaceRuntimeVerificationChallenge, -) -> Result<(), WorkspaceCapabilityVerificationError> { - validate_verification_challenge(&challenge)?; - if let Some(pending) = document.pending.iter().find(|pending| { - pending.workspace_id == challenge.workspace_id && pending.runtime_id == challenge.runtime_id - }) { - if pending.challenge_id == challenge.challenge_id && pending != &challenge { - return Err(WorkspaceCapabilityVerificationError::VerificationChallengeMismatch); - } - } - if document.pending.len() >= MAX_WORKSPACE_ISSUER_TRUST_RECORDS { - return Err(WorkspaceCapabilityVerificationError::TrustRecordLimitExceeded); - } - document.pending.retain(|pending| { - pending.workspace_id != challenge.workspace_id || pending.runtime_id != challenge.runtime_id - }); - document.records.retain(|record| { - record.workspace_id != challenge.workspace_id || record.runtime_id != challenge.runtime_id - }); - document.pending.push(challenge); - Ok(()) -} - -fn acknowledge_verification( - document: &mut WorkspaceRuntimeVerificationDocument, - challenge_id: &str, - record: WorkspaceRuntimeVerificationRecord, -) -> Result<(), WorkspaceCapabilityVerificationError> { - validate_verification_record(&record)?; - let index = document - .pending - .iter() - .position(|pending| { - pending.workspace_id == record.workspace_id - && pending.runtime_id == record.runtime_id - && pending.challenge_id == challenge_id - && pending.binding_id == record.binding_id - && pending.workspace_key_id == record.workspace_key_id - && pending.workspace_public_key_fingerprint - == record.workspace_public_key_fingerprint - && pending.workspace_trust_id == record.workspace_trust_id - && pending.runtime_public_key_fingerprint == record.runtime_public_key_fingerprint - && record.verified_at < pending.expires_at - }) - .ok_or(WorkspaceCapabilityVerificationError::VerificationChallengeMismatch)?; - if document.records.len() >= MAX_WORKSPACE_ISSUER_TRUST_RECORDS { - return Err(WorkspaceCapabilityVerificationError::TrustRecordLimitExceeded); - } - document.pending.remove(index); - document.records.retain(|current| { - current.workspace_id != record.workspace_id || current.runtime_id != record.runtime_id - }); - document.records.push(record); - Ok(()) -} - -#[derive(Clone, Debug, PartialEq, Eq, Serialize, Deserialize)] -#[serde(deny_unknown_fields)] -struct RuntimeVerificationResponseClaims { - response: WorkspaceRuntimeVerificationResponseUnsigned, - method: String, - path_and_query: String, - body_digest: String, - iat: i64, - exp: i64, - jti: String, -} - -#[derive(Clone, Debug, PartialEq, Eq, Serialize, Deserialize)] -#[serde(deny_unknown_fields)] -struct WorkspaceRuntimeVerificationResponseUnsigned { - challenge_id: String, - workspace_id: String, - runtime_id: String, - binding_id: String, - workspace_key_id: String, - workspace_public_key_fingerprint: String, - workspace_trust_id: String, - runtime_public_key_fingerprint: String, - workspace_nonce: String, - runtime_nonce: String, - expires_at: i64, -} - -impl WorkspaceRuntimeVerificationResponse { - fn unsigned(&self) -> WorkspaceRuntimeVerificationResponseUnsigned { - WorkspaceRuntimeVerificationResponseUnsigned { - challenge_id: self.challenge_id.clone(), - workspace_id: self.workspace_id.clone(), - runtime_id: self.runtime_id.clone(), - binding_id: self.binding_id.clone(), - workspace_key_id: self.workspace_key_id.clone(), - workspace_public_key_fingerprint: self.workspace_public_key_fingerprint.clone(), - workspace_trust_id: self.workspace_trust_id.clone(), - runtime_public_key_fingerprint: self.runtime_public_key_fingerprint.clone(), - workspace_nonce: self.workspace_nonce.clone(), - runtime_nonce: self.runtime_nonce.clone(), - expires_at: self.expires_at, - } - } -} - -#[derive(Clone)] -pub struct RuntimeVerificationSigner { - runtime_id: String, - public_key: String, - public_key_fingerprint: String, - signing_key: Arc, -} - -impl fmt::Debug for RuntimeVerificationSigner { - fn fmt(&self, formatter: &mut fmt::Formatter<'_>) -> fmt::Result { - formatter - .debug_struct("RuntimeVerificationSigner") - .field("runtime_id", &self.runtime_id) - .field("public_key_fingerprint", &self.public_key_fingerprint) - .finish_non_exhaustive() - } -} - -impl RuntimeVerificationSigner { - pub fn from_identity( - identity: &crate::auth::RuntimeIdentityMaterial, - ) -> Result { - let public_key = crate::auth::decode_public_key(&identity.public_key) - .map_err(|_| WorkspaceCapabilityVerificationError::TrustRecordCorrupt)?; - let fingerprint = format!("sha256:{}", hex_lower(&Sha256::digest(public_key))); - Ok(Self { - runtime_id: identity.identity_id.clone(), - public_key: identity.public_key.clone(), - public_key_fingerprint: fingerprint, - signing_key: Arc::new( - identity - .signing_key() - .map_err(|_| WorkspaceCapabilityVerificationError::TrustRecordCorrupt)?, - ), - }) - } - - pub fn runtime_id(&self) -> &str { - &self.runtime_id - } - - pub fn public_key_fingerprint(&self) -> &str { - &self.public_key_fingerprint - } - - pub fn public_key(&self) -> &str { - &self.public_key - } - - pub fn sign_response( - &self, - challenge: &WorkspaceRuntimeVerificationChallenge, - runtime_nonce: String, - now_unix: i64, - ) -> Result { - validate_verification_challenge(challenge)?; - if challenge.runtime_id != self.runtime_id - || challenge.runtime_public_key_fingerprint != self.public_key_fingerprint - || challenge.expires_at <= now_unix - { - return Err(WorkspaceCapabilityVerificationError::VerificationChallengeMismatch); - } - let unsigned = WorkspaceRuntimeVerificationResponseUnsigned { - challenge_id: challenge.challenge_id.clone(), - workspace_id: challenge.workspace_id.clone(), - runtime_id: challenge.runtime_id.clone(), - binding_id: challenge.binding_id.clone(), - workspace_key_id: challenge.workspace_key_id.clone(), - workspace_public_key_fingerprint: challenge.workspace_public_key_fingerprint.clone(), - workspace_trust_id: challenge.workspace_trust_id.clone(), - runtime_public_key_fingerprint: challenge.runtime_public_key_fingerprint.clone(), - workspace_nonce: challenge.workspace_nonce.clone(), - runtime_nonce, - expires_at: challenge.expires_at, - }; - let claims = RuntimeVerificationResponseClaims { - response: unsigned.clone(), - method: "POST".to_string(), - path_and_query: WORKSPACE_VERIFICATION_CHALLENGE_PATH.to_string(), - body_digest: workspace_request_body_digest( - &serde_json::to_vec(challenge) - .map_err(|_| WorkspaceCapabilityVerificationError::MalformedClaims)?, - ), - iat: now_unix, - exp: challenge.expires_at, - jti: uuid::Uuid::now_v7().to_string(), - }; - let proof = crate::auth::sign_json_token( - RUNTIME_VERIFICATION_TOKEN_PREFIX, - RUNTIME_VERIFICATION_SIGNING_INPUT_PREFIX, - self.signing_key.as_ref(), - &claims, - ) - .map_err(|_| WorkspaceCapabilityVerificationError::MalformedClaims)?; - Ok(WorkspaceRuntimeVerificationResponse { - challenge_id: unsigned.challenge_id, - workspace_id: unsigned.workspace_id, - runtime_id: unsigned.runtime_id, - binding_id: unsigned.binding_id, - workspace_key_id: unsigned.workspace_key_id, - workspace_public_key_fingerprint: unsigned.workspace_public_key_fingerprint, - workspace_trust_id: unsigned.workspace_trust_id, - runtime_public_key_fingerprint: unsigned.runtime_public_key_fingerprint, - workspace_nonce: unsigned.workspace_nonce, - runtime_nonce: unsigned.runtime_nonce, - expires_at: unsigned.expires_at, - response_proof: proof, - }) - } -} - -pub fn verify_runtime_verification_response( - response: &WorkspaceRuntimeVerificationResponse, - challenge: &WorkspaceRuntimeVerificationChallenge, - runtime_public_key: &str, - now_unix: i64, -) -> Result<(), WorkspaceCapabilityVerificationError> { - validate_verification_challenge(challenge)?; - let public_key = crate::auth::decode_public_key(runtime_public_key) - .map_err(|_| WorkspaceCapabilityVerificationError::InvalidSignature)?; - let fingerprint = format!("sha256:{}", hex_lower(&Sha256::digest(&public_key))); - if fingerprint != challenge.runtime_public_key_fingerprint { - return Err(WorkspaceCapabilityVerificationError::VerificationChallengeMismatch); - } - let signed = crate::auth::decode_signed_json_token::( - &response.response_proof, - RUNTIME_VERIFICATION_TOKEN_PREFIX, - ) - .map_err(|_| WorkspaceCapabilityVerificationError::MalformedToken)?; - crate::auth::verify_signed_json_token( - RUNTIME_VERIFICATION_SIGNING_INPUT_PREFIX, - &signed.payload, - &signed.signature, - runtime_public_key, - ) - .map_err(|_| WorkspaceCapabilityVerificationError::InvalidSignature)?; - let expected_body_digest = workspace_request_body_digest( - &serde_json::to_vec(challenge) - .map_err(|_| WorkspaceCapabilityVerificationError::MalformedClaims)?, - ); - if signed.claims.response != response.unsigned() - || signed.claims.method != "POST" - || signed.claims.path_and_query != WORKSPACE_VERIFICATION_CHALLENGE_PATH - || signed.claims.body_digest != expected_body_digest - || signed.claims.exp != response.expires_at - || signed.claims.iat > now_unix.saturating_add(MAX_CLOCK_SKEW_SECONDS) - || signed.claims.exp <= now_unix - || response.challenge_id != challenge.challenge_id - || response.workspace_id != challenge.workspace_id - || response.runtime_id != challenge.runtime_id - || response.binding_id != challenge.binding_id - || response.workspace_key_id != challenge.workspace_key_id - || response.workspace_public_key_fingerprint != challenge.workspace_public_key_fingerprint - || response.workspace_trust_id != challenge.workspace_trust_id - || response.runtime_public_key_fingerprint != challenge.runtime_public_key_fingerprint - || response.workspace_nonce != challenge.workspace_nonce - || response.runtime_nonce.is_empty() - { - return Err(WorkspaceCapabilityVerificationError::VerificationChallengeMismatch); - } - Ok(()) -} - #[derive(Clone, Debug, PartialEq, Eq, Serialize, Deserialize)] #[serde(deny_unknown_fields)] pub struct WorkspaceCapabilityClaims { @@ -965,8 +309,6 @@ pub struct WorkspaceCapabilityClaims { pub issuer_workspace_id: String, pub issuer_key_id: String, pub issuer_public_key_fingerprint: String, - pub trust_id: String, - pub binding_id: String, pub runtime_id: String, #[serde(skip_serializing_if = "Option::is_none")] pub worker_id: Option, @@ -982,7 +324,6 @@ pub struct WorkspaceCapabilityClaims { #[derive(Clone, Debug, PartialEq, Eq)] pub struct WorkspaceCapabilityExpectation<'a> { pub workspace_id: &'a str, - pub binding_id: &'a str, pub runtime_id: &'a str, pub worker_id: Option<&'a str>, pub operation: &'a str, @@ -998,8 +339,6 @@ pub struct VerifiedWorkspaceCapability { pub workspace_id: String, pub issuer_key_id: String, pub issuer_public_key_fingerprint: String, - pub trust_id: String, - pub binding_id: String, pub runtime_id: String, pub worker_id: Option, pub operation: String, @@ -1009,11 +348,16 @@ pub struct VerifiedWorkspaceCapability { pub expires_at: i64, } +/// Replay scope is Runtime-owned Workspace + trusted key material, not sender claims +/// or administrative enrollment IDs. Re-enrollment (including rotate away/back) +/// cannot reset consumed tokens. Fresh unconsumed tokens under the same reaccepted +/// key remain valid until expiry; revoked trust denies all requests. +/// A genuinely different trusted key has an independent token namespace. pub trait WorkspaceClaimReplayProtection: Send + Sync { fn consume_once( &self, workspace_id: &str, - trust_id: &str, + key_fingerprint: &str, token_id: &str, expires_at: i64, now_unix: i64, @@ -1029,7 +373,7 @@ impl WorkspaceClaimReplayProtection for InMemoryWorkspaceClaimReplayProtection { fn consume_once( &self, workspace_id: &str, - trust_id: &str, + key_fingerprint: &str, token_id: &str, expires_at: i64, now_unix: i64, @@ -1041,7 +385,7 @@ impl WorkspaceClaimReplayProtection for InMemoryWorkspaceClaimReplayProtection { consumed.retain(|_, expiry| *expiry > now_unix); let key = ( workspace_id.to_string(), - trust_id.to_string(), + key_fingerprint.to_string(), token_id.to_string(), ); if consumed.contains_key(&key) { @@ -1072,11 +416,14 @@ struct WorkspaceClaimReplayDocument { #[serde(deny_unknown_fields)] struct WorkspaceClaimReplayEntry { workspace_id: String, - trust_id: String, + // None occurs only when migrating old saved replay scopes: conservatively + // fence the token across every key for that Workspace until its expiry. + key_fingerprint: Option, token_id: String, expires_at: i64, } +// Frozen saved-data formats only. Neither format is accepted on the network. #[derive(Deserialize)] #[serde(deny_unknown_fields)] struct LegacyReplayDocument { @@ -1085,13 +432,26 @@ struct LegacyReplayDocument { } #[derive(Deserialize)] #[serde(deny_unknown_fields)] -#[allow(dead_code)] struct LegacyReplayEntry { workspace_id: String, trust_generation: u64, token_id: String, expires_at: i64, } +#[derive(Deserialize)] +#[serde(deny_unknown_fields)] +struct EnrollmentReplayDocument { + version: u32, + entries: Vec, +} +#[derive(Deserialize)] +#[serde(deny_unknown_fields)] +struct EnrollmentReplayEntry { + workspace_id: String, + trust_id: String, + token_id: String, + expires_at: i64, +} impl FileWorkspaceClaimReplayProtection { pub fn new(path: impl Into) -> Self { @@ -1107,19 +467,72 @@ impl FileWorkspaceClaimReplayProtection { let raw: serde_json::Value = serde_json::from_slice(&bytes).map_err(|_| { WorkspaceCapabilityVerificationError::ReplayAuthorityUnavailable })?; - if raw.get("version").and_then(serde_json::Value::as_u64) == Some(1) { - let legacy: LegacyReplayDocument = - serde_json::from_value(raw).map_err(|_| { - WorkspaceCapabilityVerificationError::ReplayAuthorityUnavailable - })?; - if legacy.version != 1 || legacy.entries.len() > MAX_REPLAY_ENTRIES { - return Err( - WorkspaceCapabilityVerificationError::ReplayAuthorityUnavailable, - ); + let version = raw.get("version").and_then(serde_json::Value::as_u64); + let entries = match version { + Some(1) => { + let legacy: LegacyReplayDocument = serde_json::from_value(raw.clone()) + .map_err(|_| { + WorkspaceCapabilityVerificationError::ReplayAuthorityUnavailable + })?; + if legacy.version != 1 + || legacy.entries.len() > MAX_REPLAY_ENTRIES + || legacy + .entries + .iter() + .any(|entry| entry.trust_generation == 0) + { + return Err( + WorkspaceCapabilityVerificationError::ReplayAuthorityUnavailable, + ); + } + Some( + legacy + .entries + .into_iter() + .map(|entry| WorkspaceClaimReplayEntry { + workspace_id: entry.workspace_id, + key_fingerprint: None, + token_id: entry.token_id, + expires_at: entry.expires_at, + }) + .collect(), + ) } + Some(2) => { + let legacy: EnrollmentReplayDocument = serde_json::from_value(raw.clone()) + .map_err(|_| { + WorkspaceCapabilityVerificationError::ReplayAuthorityUnavailable + })?; + if legacy.version != 2 + || legacy.entries.len() > MAX_REPLAY_ENTRIES + || legacy + .entries + .iter() + .any(|entry| validate_id(&entry.trust_id).is_err()) + { + return Err( + WorkspaceCapabilityVerificationError::ReplayAuthorityUnavailable, + ); + } + Some( + legacy + .entries + .into_iter() + .map(|entry| WorkspaceClaimReplayEntry { + workspace_id: entry.workspace_id, + key_fingerprint: None, + token_id: entry.token_id, + expires_at: entry.expires_at, + }) + .collect(), + ) + } + _ => None, + }; + if let Some(entries) = entries { let migrated = WorkspaceClaimReplayDocument { - version: 2, - entries: Vec::new(), + version: 3, + entries, }; self.write(&migrated)?; return Ok(migrated); @@ -1128,14 +541,14 @@ impl FileWorkspaceClaimReplayProtection { serde_json::from_value(raw).map_err(|_| { WorkspaceCapabilityVerificationError::ReplayAuthorityUnavailable })?; - if document.version != 2 || document.entries.len() > MAX_REPLAY_ENTRIES { + if document.version != 3 || document.entries.len() > MAX_REPLAY_ENTRIES { return Err(WorkspaceCapabilityVerificationError::ReplayAuthorityUnavailable); } Ok(document) } Err(error) if error.kind() == std::io::ErrorKind::NotFound => { Ok(WorkspaceClaimReplayDocument { - version: 2, + version: 3, entries: Vec::new(), }) } @@ -1147,7 +560,13 @@ impl FileWorkspaceClaimReplayProtection { &self, document: &WorkspaceClaimReplayDocument, ) -> Result<(), WorkspaceCapabilityVerificationError> { - let parent = self.path.parent().unwrap_or_else(|| Path::new(".")); + use std::io::Write as _; + + let parent = self + .path + .parent() + .filter(|parent| !parent.as_os_str().is_empty()) + .unwrap_or_else(|| Path::new(".")); fs::create_dir_all(parent) .map_err(|_| WorkspaceCapabilityVerificationError::ReplayAuthorityUnavailable)?; let bytes = serde_json::to_vec(document) @@ -1156,19 +575,30 @@ impl FileWorkspaceClaimReplayProtection { ".workspace-claim-replay-{}.tmp", uuid::Uuid::now_v7() )); - fs::write(&temporary, bytes) - .map_err(|_| WorkspaceCapabilityVerificationError::ReplayAuthorityUnavailable)?; - #[cfg(unix)] - { - use std::os::unix::fs::PermissionsExt as _; - fs::set_permissions(&temporary, fs::Permissions::from_mode(0o600)) - .map_err(|_| WorkspaceCapabilityVerificationError::ReplayAuthorityUnavailable)?; - } - fs::rename(&temporary, &self.path).map_err(|_| { + let result = (|| -> std::io::Result<()> { + let mut options = fs::OpenOptions::new(); + options.create_new(true).write(true); + #[cfg(unix)] + { + use std::os::unix::fs::OpenOptionsExt as _; + options.mode(0o600); + } + let mut file = options.open(&temporary)?; + file.write_all(&bytes)?; + // A successful consumption must survive a crash: persist the file + // contents before publication, then persist the atomic rename. + file.sync_all()?; + fs::rename(&temporary, &self.path)?; + fs::File::open(parent)?.sync_all()?; + Ok(()) + })(); + if result.is_err() { + // Before rename, remove the unpublished temporary file. After rename + // (e.g. a directory-sync failure), retain the consumed-token record + // but reject authorization; never roll replay protection back. let _ = fs::remove_file(&temporary); - WorkspaceCapabilityVerificationError::ReplayAuthorityUnavailable - })?; - Ok(()) + } + result.map_err(|_| WorkspaceCapabilityVerificationError::ReplayAuthorityUnavailable) } } @@ -1176,7 +606,7 @@ impl WorkspaceClaimReplayProtection for FileWorkspaceClaimReplayProtection { fn consume_once( &self, workspace_id: &str, - trust_id: &str, + key_fingerprint: &str, token_id: &str, expires_at: i64, now_unix: i64, @@ -1189,7 +619,10 @@ impl WorkspaceClaimReplayProtection for FileWorkspaceClaimReplayProtection { document.entries.retain(|entry| entry.expires_at > now_unix); if document.entries.iter().any(|entry| { entry.workspace_id == workspace_id - && entry.trust_id == trust_id + && entry + .key_fingerprint + .as_deref() + .is_none_or(|key| key == key_fingerprint) && entry.token_id == token_id }) { return Ok(false); @@ -1199,7 +632,7 @@ impl WorkspaceClaimReplayProtection for FileWorkspaceClaimReplayProtection { } document.entries.push(WorkspaceClaimReplayEntry { workspace_id: workspace_id.to_string(), - trust_id: trust_id.to_string(), + key_fingerprint: Some(key_fingerprint.to_string()), token_id: token_id.to_string(), expires_at, }); @@ -1228,9 +661,6 @@ impl WorkspaceCapabilityVerifier { records: Vec, replay: Arc, ) -> Result { - if records.is_empty() { - return Err(WorkspaceCapabilityVerificationError::TrustAuthorityMissing); - } validate_workspace_issuer_trust_records(&records)?; Ok(Self { records: records.into(), @@ -1274,9 +704,6 @@ impl WorkspaceCapabilityVerifier { if record.backend_url != claims.issuer { return Err(WorkspaceCapabilityVerificationError::WrongIssuer); } - if record.trust_id != claims.trust_id { - return Err(WorkspaceCapabilityVerificationError::WrongTrustIdentity); - } if record.key_id != claims.issuer_key_id { return Err(WorkspaceCapabilityVerificationError::WrongIssuerKey); } @@ -1300,9 +727,6 @@ impl WorkspaceCapabilityVerifier { if claims.runtime_id != expected.runtime_id { return Err(WorkspaceCapabilityVerificationError::WrongRuntime); } - if claims.binding_id != expected.binding_id { - return Err(WorkspaceCapabilityVerificationError::WrongBindingIdentity); - } if claims.worker_id.as_deref() != expected.worker_id { return Err(WorkspaceCapabilityVerificationError::WrongWorker); } @@ -1331,7 +755,7 @@ impl WorkspaceCapabilityVerifier { } if !self.replay.consume_once( &claims.issuer_workspace_id, - &claims.trust_id, + &record.public_key_fingerprint, &claims.jti, claims.exp, expected.now_unix, @@ -1344,8 +768,6 @@ impl WorkspaceCapabilityVerifier { workspace_id: claims.issuer_workspace_id, issuer_key_id: claims.issuer_key_id, issuer_public_key_fingerprint: claims.issuer_public_key_fingerprint, - trust_id: claims.trust_id, - binding_id: claims.binding_id, runtime_id: claims.runtime_id, worker_id: claims.worker_id, operation: claims.operation, @@ -1359,8 +781,6 @@ impl WorkspaceCapabilityVerifier { #[derive(Debug, thiserror::Error, PartialEq, Eq)] pub enum WorkspaceCapabilityVerificationError { - #[error("Workspace issuer trust authority is not configured")] - TrustAuthorityMissing, #[error("Workspace issuer trust contains duplicate Workspace identities")] DuplicateWorkspaceTrust, #[error("Workspace issuer trust contains too many records")] @@ -1381,8 +801,6 @@ pub enum WorkspaceCapabilityVerificationError { WrongIssuer, #[error("Workspace capability issuer trust is revoked")] IssuerRevoked, - #[error("Workspace capability trust identity does not match current trust")] - WrongTrustIdentity, #[error("Workspace capability issuer key does not match trust")] WrongIssuerKey, #[error("Workspace capability signature is invalid")] @@ -1391,8 +809,6 @@ pub enum WorkspaceCapabilityVerificationError { WrongWorkspace, #[error("Workspace capability targets another Runtime")] WrongRuntime, - #[error("Workspace capability binding identity does not match the verified binding")] - WrongBindingIdentity, #[error("Workspace capability targets another Worker")] WrongWorker, #[error("Workspace capability does not authorize this operation")] @@ -1413,12 +829,6 @@ pub enum WorkspaceCapabilityVerificationError { Replay, #[error("Workspace capability replay authority is unavailable")] ReplayAuthorityUnavailable, - #[error("Workspace Runtime verification authority is unavailable")] - VerificationAuthorityUnavailable, - #[error( - "Workspace Runtime verification challenge or response does not match current authority" - )] - VerificationChallengeMismatch, } #[derive(Clone, Debug, PartialEq, Eq)] @@ -1500,77 +910,6 @@ fn validate_trust_record( Ok(()) } -fn validate_verification_record( - record: &WorkspaceRuntimeVerificationRecord, -) -> Result<(), WorkspaceCapabilityVerificationError> { - for value in [ - record.workspace_id.as_str(), - record.runtime_id.as_str(), - record.workspace_key_id.as_str(), - record.binding_id.as_str(), - record.workspace_trust_id.as_str(), - ] { - if value.is_empty() - || value.len() > MAX_ID_BYTES - || value.trim() != value - || !value.bytes().all(|byte| { - byte.is_ascii_alphanumeric() || matches!(byte, b'-' | b'_' | b'.' | b':') - }) - { - return Err(WorkspaceCapabilityVerificationError::VerificationAuthorityUnavailable); - } - } - if record.binding_id.is_empty() - || record.workspace_public_key_fingerprint.is_empty() - || record.workspace_trust_id.is_empty() - || !is_key_fingerprint(&record.workspace_public_key_fingerprint) - || !is_key_fingerprint(&record.runtime_public_key_fingerprint) - || record.verified_at <= 0 - || record.runtime_public_key_fingerprint.len() > 128 - || !record.runtime_public_key_fingerprint.starts_with("sha256:") - { - return Err(WorkspaceCapabilityVerificationError::VerificationAuthorityUnavailable); - } - Ok(()) -} - -fn validate_verification_challenge( - challenge: &WorkspaceRuntimeVerificationChallenge, -) -> Result<(), WorkspaceCapabilityVerificationError> { - for value in [ - challenge.challenge_id.as_str(), - challenge.workspace_id.as_str(), - challenge.runtime_id.as_str(), - challenge.workspace_key_id.as_str(), - challenge.binding_id.as_str(), - challenge.workspace_trust_id.as_str(), - challenge.workspace_nonce.as_str(), - ] { - if value.is_empty() - || value.len() > MAX_ID_BYTES - || value.trim() != value - || !value.bytes().all(|byte| { - byte.is_ascii_alphanumeric() || matches!(byte, b'-' | b'_' | b'.' | b':') - }) - { - return Err(WorkspaceCapabilityVerificationError::VerificationChallengeMismatch); - } - } - if challenge.binding_id.is_empty() - || challenge.workspace_public_key_fingerprint.is_empty() - || challenge.workspace_trust_id.is_empty() - || !is_key_fingerprint(&challenge.workspace_public_key_fingerprint) - || !is_key_fingerprint(&challenge.runtime_public_key_fingerprint) - || challenge.runtime_public_key_fingerprint.len() > 128 - || !challenge - .runtime_public_key_fingerprint - .starts_with("sha256:") - { - return Err(WorkspaceCapabilityVerificationError::VerificationChallengeMismatch); - } - Ok(()) -} - fn validate_claim_shape( claims: &WorkspaceCapabilityClaims, ) -> Result<(), WorkspaceCapabilityVerificationError> { @@ -1594,8 +933,6 @@ fn validate_claim_shape( for value in [ claims.issuer_workspace_id.as_str(), claims.issuer_key_id.as_str(), - claims.binding_id.as_str(), - claims.trust_id.as_str(), claims.runtime_id.as_str(), claims.jti.as_str(), ] { @@ -1620,10 +957,7 @@ fn validate_claim_shape( return Err(WorkspaceCapabilityVerificationError::InvalidIdentifier); } } - if !is_key_fingerprint(&claims.issuer_public_key_fingerprint) - || claims.trust_id.is_empty() - || claims.binding_id.is_empty() - { + if !is_key_fingerprint(&claims.issuer_public_key_fingerprint) { return Err(WorkspaceCapabilityVerificationError::MalformedClaims); } if claims.operation.is_empty() @@ -1713,8 +1047,6 @@ mod tests { issuer_workspace_id: record.workspace_id.clone(), issuer_key_id: record.key_id.clone(), issuer_public_key_fingerprint: record.public_key_fingerprint.clone(), - trust_id: record.trust_id.clone(), - binding_id: "binding-test".into(), runtime_id: "runtime-1".to_string(), worker_id: Some("worker-1".to_string()), operation: "worker.submit".to_string(), @@ -1730,7 +1062,6 @@ mod tests { fn expectation<'a>(body_digest: &'a str) -> WorkspaceCapabilityExpectation<'a> { WorkspaceCapabilityExpectation { workspace_id: "workspace-1", - binding_id: "binding-test", runtime_id: "runtime-1", worker_id: Some("worker-1"), operation: "worker.submit", @@ -1782,7 +1113,7 @@ mod tests { } #[test] - fn verifier_binds_workspace_key_trust_identity_runtime_worker_operation_and_body() { + fn verifier_binds_workspace_key_runtime_worker_operation_and_body() { let (key_1, bundle_1) = identity("workspace-1", "WK-1"); let (key_2, bundle_2) = identity("workspace-2", "WK-2"); let record_1 = @@ -1815,7 +1146,6 @@ mod tests { ); let workspace_2_expectation = WorkspaceCapabilityExpectation { workspace_id: "workspace-2", - binding_id: "binding-test".into(), runtime_id: "runtime-1", worker_id: Some("worker-1"), operation: "worker.submit", @@ -1857,13 +1187,6 @@ mod tests { }) as ClaimsMutation, WorkspaceCapabilityVerificationError::WrongWorker, ), - ( - "binding", - (|claims: &mut WorkspaceCapabilityClaims| { - claims.binding_id = "other-binding".into() - }) as ClaimsMutation, - WorkspaceCapabilityVerificationError::WrongBindingIdentity, - ), ( "issuer", (|claims: &mut WorkspaceCapabilityClaims| { @@ -1914,7 +1237,7 @@ mod tests { } #[test] - fn verifier_rejects_replay_revocation_old_trust_expiry_and_future_claims() { + fn verifier_rejects_replay_revocation_expiry_and_future_claims() { let (key, bundle) = identity("workspace-1", "WK-1"); let record = WorkspaceIssuerTrustRecord::from_bundle(bundle, "trust-test".into(), 1).unwrap(); @@ -1928,14 +1251,6 @@ mod tests { WorkspaceCapabilityVerificationError::Replay ); - let mut stale = claims(&record, "stale"); - stale.trust_id = "retired-trust".into(); - let token = issue_workspace_capability_token(&key, &stale).unwrap(); - assert_eq!( - verifier.verify(&token, &expectation(&body)).unwrap_err(), - WorkspaceCapabilityVerificationError::WrongTrustIdentity - ); - let mut expired = claims(&record, "expired"); expired.iat = 900; expired.exp = 950; @@ -1972,96 +1287,323 @@ mod tests { } #[test] - fn reaccepting_the_same_key_does_not_reactivate_old_capabilities() { + fn same_workspace_key_reenrollment_and_rotation_back_do_not_reset_durable_replay() { + let dir = tempfile::tempdir().unwrap(); + let path = dir.path().join("replay.json"); let (key, bundle) = identity("workspace-1", "WK-1"); let mut records = Vec::new(); - let (_, first) = add_workspace_issuer_trust(&mut records, bundle.clone(), 10).unwrap(); - let token = issue_workspace_capability_token(&key, &claims(&first, "old-token")).unwrap(); - revoke_workspace_issuer_trust(&mut records, "workspace-1", 11).unwrap(); - let (_, reaccepted) = replace_workspace_issuer_trust(&mut records, bundle, 12).unwrap(); - assert_ne!(first.trust_id, reaccepted.trust_id); + let (_, first) = add_workspace_issuer_trust(&mut records, bundle.clone(), 1).unwrap(); + let body = workspace_request_body_digest(br#"{"content":"hello"}"#); + let token = issue_workspace_capability_token(&key, &claims(&first, "consumed")).unwrap(); + WorkspaceCapabilityVerifier::new( + records.clone(), + Arc::new(FileWorkspaceClaimReplayProtection::new(&path)), + ) + .unwrap() + .verify(&token, &expectation(&body)) + .unwrap(); + revoke_workspace_issuer_trust(&mut records, "workspace-1", 2).unwrap(); + replace_workspace_issuer_trust(&mut records, bundle.clone(), 3).unwrap(); + for reenroll in [true, false] { + if !reenroll { + let (_, other) = identity("workspace-1", "WK-2"); + replace_workspace_issuer_trust(&mut records, other, 4).unwrap(); + replace_workspace_issuer_trust(&mut records, bundle.clone(), 5).unwrap(); + } + assert_ne!(first.trust_id, records[0].trust_id); + let verifier = WorkspaceCapabilityVerifier::new( + records.clone(), + Arc::new(FileWorkspaceClaimReplayProtection::new(&path)), + ) + .unwrap(); + assert_eq!( + verifier.verify(&token, &expectation(&body)), + Err(WorkspaceCapabilityVerificationError::Replay) + ); + } + // Reaccepting the same actual key authorizes fresh, unconsumed short-lived + // requests; there is no sender enrollment ID fence or handshake status. + let token = issue_workspace_capability_token(&key, &claims(&first, "fresh")).unwrap(); + WorkspaceCapabilityVerifier::new( + records, + Arc::new(FileWorkspaceClaimReplayProtection::new(&path)), + ) + .unwrap() + .verify(&token, &expectation(&body)) + .unwrap(); + } + + #[test] + fn rotated_trust_accepts_only_requests_verified_with_the_current_key() { + let (old_key, old_bundle) = identity("workspace-1", "WK-1"); + let (new_key, new_bundle) = identity("workspace-1", "WK-2"); + let mut records = Vec::new(); + let (_, old) = add_workspace_issuer_trust(&mut records, old_bundle, 1).unwrap(); + let (_, current) = replace_workspace_issuer_trust(&mut records, new_bundle, 2).unwrap(); let verifier = WorkspaceCapabilityVerifier::new( records, Arc::new(InMemoryWorkspaceClaimReplayProtection::default()), ) .unwrap(); let body = workspace_request_body_digest(br#"{"content":"hello"}"#); + let old_request = issue_workspace_capability_token(&old_key, &claims(&old, "old")).unwrap(); assert_eq!( - verifier.verify(&token, &expectation(&body)).unwrap_err(), - WorkspaceCapabilityVerificationError::WrongTrustIdentity + verifier.verify(&old_request, &expectation(&body)), + Err(WorkspaceCapabilityVerificationError::WrongIssuerKey) ); + let forged_current = + issue_workspace_capability_token(&old_key, &claims(¤t, "current")).unwrap(); + assert_eq!( + verifier.verify(&forged_current, &expectation(&body)), + Err(WorkspaceCapabilityVerificationError::InvalidSignature) + ); + // Failed verification does not consume jti; actual current-key verification does. + let current_request = + issue_workspace_capability_token(&new_key, &claims(¤t, "current")).unwrap(); + verifier + .verify(¤t_request, &expectation(&body)) + .unwrap(); } #[test] - fn frozen_enrollment_migration_requires_explicit_reacceptance() { - let (_, bundle) = identity("workspace-1", "WK-1"); - let record = - WorkspaceIssuerTrustRecord::from_bundle(bundle, "trust-test".into(), 10).unwrap(); - for state in ["active", "revoked"] { - let mut legacy = serde_json::to_value(&record).unwrap(); - legacy.as_object_mut().unwrap().remove("trust_id"); - legacy["identity_revision"] = serde_json::json!(1); - legacy["trust_generation"] = serde_json::json!(2); - legacy["state"] = serde_json::json!(state); - assert!(serde_json::from_value::(legacy.clone()).is_err()); - let migrated = migrate_legacy_workspace_issuer_trust_record(legacy).unwrap(); - assert_eq!(migrated.state, WorkspaceIssuerTrustState::Revoked); - assert_eq!(migrated.public_bundle(), record.public_bundle()); + fn saved_replay_migration_preserves_consumed_tokens_across_unknown_key_scopes() { + for version in [1, 2] { + let dir = tempfile::tempdir().unwrap(); + let path = dir.path().join("replay.json"); + let mut entry = serde_json::json!({"workspace_id":"workspace-a", "token_id":"token-a", "expires_at":1000}); + if version == 1 { + entry["trust_generation"] = serde_json::json!(1); + } else { + entry["trust_id"] = serde_json::json!("old-enrollment"); + } + fs::write( + &path, + serde_json::to_vec(&serde_json::json!({"version":version,"entries":[entry]})) + .unwrap(), + ) + .unwrap(); + let replay = FileWorkspaceClaimReplayProtection::new(&path); + assert!( + !replay + .consume_once("workspace-a", "key-a", "token-a", 1000, 100) + .unwrap() + ); + // Reconstruction must also retain the old consumed token, not reset it. + let replay = FileWorkspaceClaimReplayProtection::new(&path); + assert!( + !replay + .consume_once("workspace-a", "key-b", "token-a", 1000, 101) + .unwrap() + ); + let saved: serde_json::Value = + serde_json::from_slice(&fs::read(&path).unwrap()).unwrap(); + assert_eq!(saved["version"], 3); + assert_eq!(saved["entries"][0]["token_id"], "token-a"); + assert!( + replay + .consume_once("workspace-a", "key-a", "fresh", 1000, 101) + .unwrap() + ); } } #[test] - fn frozen_verification_file_is_invalidated_durably_without_losing_archived_evidence() { - let dir = tempfile::tempdir().unwrap(); - let path = dir.path().join("verification.json"); - fs::write(&path, serde_json::to_vec(&serde_json::json!({"version":1,"records":[{ - "workspace_id":"workspace-1","runtime_id":"runtime-1","binding_revision":1, - "workspace_key_id":"WK-1","workspace_identity_revision":1,"workspace_trust_generation":1, - "runtime_public_key_fingerprint":"sha256:old","runtime_identity_revision":1,"verified_at":10 - }]})).unwrap()).unwrap(); - let original_bytes = fs::read(&path).unwrap(); - let authority = FileWorkspaceRuntimeVerificationAuthority::new(&path); - assert!(authority.get("workspace-1", "runtime-1").unwrap().is_none()); - let migrated: serde_json::Value = - serde_json::from_slice(&fs::read(&path).unwrap()).unwrap(); - assert_eq!(migrated["version"], 2); - assert_eq!(migrated["records"], serde_json::json!([])); + fn current_network_claims_reject_handshake_ids_and_self_reported_public_keys() { + let (key, bundle) = identity("workspace-1", "WK-1"); + let record = + WorkspaceIssuerTrustRecord::from_bundle(bundle, "internal-trust".into(), 1).unwrap(); + let wire = serde_json::to_value(claims(&record, "token")).unwrap(); + assert!(wire.get("trust_id").is_none()); + assert!(wire.get("binding_id").is_none()); + for obsolete in ["binding_id", "trust_id", "public_key"] { + let mut changed = wire.clone(); + changed[obsolete] = serde_json::json!("sender-authority"); + let token = crate::auth::sign_json_token( + WORKSPACE_TOKEN_PREFIX, + WORKSPACE_SIGNING_INPUT_PREFIX, + &key, + &changed, + ) + .unwrap(); + assert_eq!( + inspect_workspace_capability_claims(&token), + Err(WorkspaceCapabilityVerificationError::MalformedToken) + ); + let verifier = WorkspaceCapabilityVerifier::new( + vec![record.clone()], + Arc::new(InMemoryWorkspaceClaimReplayProtection::default()), + ) + .unwrap(); + assert_eq!( + verifier.verify(&token, &expectation(&claims(&record, "token").body_digest)), + Err(WorkspaceCapabilityVerificationError::MalformedToken) + ); + } + } + + fn legacy_enrollment(record: &WorkspaceIssuerTrustRecord) -> serde_json::Value { + let mut legacy = serde_json::to_value(&record).unwrap(); + legacy.as_object_mut().unwrap().remove("trust_id"); + legacy["identity_revision"] = serde_json::json!(1); + legacy["trust_generation"] = serde_json::json!(2); + legacy + } + + #[test] + fn legacy_enrollment_migration_preserves_trust_permission_for_actual_signed_ping() { + let (key, bundle) = identity("workspace-1", "WK-1"); + let mut record = + WorkspaceIssuerTrustRecord::from_bundle(bundle, "trust-test".into(), 10).unwrap(); + record.updated_at_unix = 20; + for state in [ + WorkspaceIssuerTrustState::Active, + WorkspaceIssuerTrustState::Revoked, + ] { + record.state = state; + let legacy = legacy_enrollment(&record); + assert!(serde_json::from_value::(legacy.clone()).is_err()); + let migrated = migrate_legacy_workspace_issuer_trust_record(legacy).unwrap(); + assert_eq!(migrated.state, state); + assert_eq!(migrated.public_bundle(), record.public_bundle()); + assert_eq!(migrated.registered_at_unix, record.registered_at_unix); + assert_eq!(migrated.updated_at_unix, record.updated_at_unix); + assert!(!migrated.trust_id.is_empty()); + let verifier = WorkspaceCapabilityVerifier::new( + vec![migrated], + Arc::new(InMemoryWorkspaceClaimReplayProtection::default()), + ) + .unwrap(); + let request = WorkspaceCapabilityClaims { + worker_id: None, + operation: "runtime:ping".into(), + method: "GET".into(), + path_and_query: "/v1/ping".into(), + body_digest: workspace_request_body_digest(&[]), + ..claims(&record, "ordinary-ping") + }; + let token = issue_workspace_capability_token(&key, &request).unwrap(); + let expected = WorkspaceCapabilityExpectation { + worker_id: None, + operation: &request.operation, + method: "GET", + path_and_query: "/v1/ping", + ..expectation(&request.body_digest) + }; + let result = verifier.verify(&token, &expected); + if state == WorkspaceIssuerTrustState::Active { + assert_eq!(result.unwrap().workspace_id, "workspace-1"); + } else { + assert_eq!( + result, + Err(WorkspaceCapabilityVerificationError::IssuerRevoked) + ); + } + } + } + + #[test] + fn malformed_legacy_enrollment_is_rejected_instead_of_becoming_trusted() { + let (_, bundle) = identity("workspace-1", "WK-1"); + let record = + WorkspaceIssuerTrustRecord::from_bundle(bundle, "trust-test".into(), 10).unwrap(); + for (field, value, error) in [ + ( + "workspace_id", + serde_json::json!(""), + WorkspaceIssuerTrustError::InvalidIdentifier, + ), + ( + "backend_url", + serde_json::json!("file:///untrusted"), + WorkspaceIssuerTrustError::InvalidBackendUrl, + ), + ( + "algorithm", + serde_json::json!("none"), + WorkspaceIssuerTrustError::UnsupportedAlgorithm, + ), + ( + "public_key", + serde_json::json!("invalid"), + WorkspaceIssuerTrustError::InvalidPublicKey, + ), + ( + "public_key_fingerprint", + serde_json::json!(format!("sha256:{}", "0".repeat(64))), + WorkspaceIssuerTrustError::PublicKeyFingerprintMismatch, + ), + ( + "state", + serde_json::json!("invalid"), + WorkspaceIssuerTrustError::InvalidIdentifier, + ), + ( + "unexpected", + serde_json::json!(true), + WorkspaceIssuerTrustError::InvalidIdentifier, + ), + ] { + let mut legacy = legacy_enrollment(&record); + legacy[field] = value; + assert_eq!( + migrate_legacy_workspace_issuer_trust_record(legacy), + Err(error), + "{field}" + ); + } + let mut missing = legacy_enrollment(&record); + missing.as_object_mut().unwrap().remove("state"); assert_eq!( - migrated["archived_legacy_document"] - .as_str() - .unwrap() - .as_bytes(), - original_bytes + migrate_legacy_workspace_issuer_trust_record(missing), + Err(WorkspaceIssuerTrustError::InvalidIdentifier) ); - assert!( - FileWorkspaceRuntimeVerificationAuthority::new(&path) - .get("workspace-1", "runtime-1") - .unwrap() - .is_none() - ); - let current = WorkspaceRuntimeVerificationRecord { - workspace_id: "workspace-1".into(), - runtime_id: "runtime-1".into(), - binding_id: "accepted-binding".into(), - workspace_key_id: "WK-1".into(), - workspace_public_key_fingerprint: format!("sha256:{}", "a".repeat(64)), - workspace_trust_id: "accepted-trust".into(), - runtime_public_key_fingerprint: format!("sha256:{}", "b".repeat(64)), - verified_at: 20, - }; - authority.record(current.clone()).unwrap(); - let restored = FileWorkspaceRuntimeVerificationAuthority::new(&path); + } + + #[test] + fn failed_replay_rename_preserves_destination_and_cleans_temporary_file() { + let temporary = tempfile::tempdir().unwrap(); + let path = temporary.path().join("workspace-replay.json"); + // Replacing a directory with the temporary regular file fails on real I/O, + // after the file write/sync. No permission assumptions or fault hooks. + fs::create_dir(&path).unwrap(); + fs::write(path.join("existing"), "preserve me").unwrap(); + let replay = FileWorkspaceClaimReplayProtection::new(&path); assert_eq!( - restored.get("workspace-1", "runtime-1").unwrap(), - Some(current) + replay.write(&WorkspaceClaimReplayDocument { + version: 3, + entries: Vec::new(), + }), + Err(WorkspaceCapabilityVerificationError::ReplayAuthorityUnavailable) ); - let updated: serde_json::Value = serde_json::from_slice(&fs::read(&path).unwrap()).unwrap(); assert_eq!( - updated["archived_legacy_document"] - .as_str() - .unwrap() - .as_bytes(), - original_bytes + fs::read_to_string(path.join("existing")).unwrap(), + "preserve me" + ); + let remaining = fs::read_dir(temporary.path()) + .unwrap() + .map(|entry| entry.unwrap().path()) + .collect::>(); + assert_eq!(remaining, vec![path]); + } + + #[test] + fn unavailable_replay_file_rejects_otherwise_valid_signed_request() { + let temporary = tempfile::tempdir().unwrap(); + let path = temporary.path().join("workspace-replay.json"); + fs::create_dir(&path).unwrap(); + let (key, bundle) = identity("workspace-1", "WK-1"); + let record = + WorkspaceIssuerTrustRecord::from_bundle(bundle, "trust-test".into(), 1).unwrap(); + let request = claims(&record, "token"); + let token = issue_workspace_capability_token(&key, &request).unwrap(); + let verifier = WorkspaceCapabilityVerifier::new( + vec![record], + Arc::new(FileWorkspaceClaimReplayProtection::new(path)), + ) + .unwrap(); + assert_eq!( + verifier.verify(&token, &expectation(&request.body_digest)), + Err(WorkspaceCapabilityVerificationError::ReplayAuthorityUnavailable) ); } @@ -2072,119 +1614,19 @@ mod tests { let first = FileWorkspaceClaimReplayProtection::new(&path); assert!( first - .consume_once("workspace-a", "trust-a", "token-a", 1_000, 100) + .consume_once("workspace-a", "key-a", "token-a", 1_000, 100) .unwrap() ); let restored = FileWorkspaceClaimReplayProtection::new(&path); assert!( !restored - .consume_once("workspace-a", "trust-a", "token-a", 1_000, 101) + .consume_once("workspace-a", "key-a", "token-a", 1_000, 101) .unwrap() ); assert!( restored - .consume_once("workspace-a", "trust-b", "token-a", 1_000, 101) + .consume_once("workspace-a", "key-b", "token-a", 1_000, 101) .unwrap() ); } - - #[test] - fn file_runtime_verification_authority_survives_reconstruction() { - let temporary = tempfile::tempdir().unwrap(); - let path = temporary.path().join("workspace-verifications.json"); - let authority = FileWorkspaceRuntimeVerificationAuthority::new(&path); - let record = WorkspaceRuntimeVerificationRecord { - workspace_id: "workspace-a".to_string(), - runtime_id: "runtime-a".to_string(), - binding_id: "binding-test".into(), - workspace_key_id: "WK-a".to_string(), - workspace_public_key_fingerprint: format!("sha256:{}", "a".repeat(64)), - workspace_trust_id: "trust-test".into(), - runtime_public_key_fingerprint: format!("sha256:{}", "b".repeat(64)), - verified_at: 100, - }; - authority.record(record.clone()).unwrap(); - let restored = FileWorkspaceRuntimeVerificationAuthority::new(&path); - assert_eq!( - restored.get("workspace-a", "runtime-a").unwrap(), - Some(record) - ); - } - - #[test] - fn verification_response_binds_the_exact_challenge_and_runtime_identity() { - let runtime_identity = RuntimeIdentityMaterial::generate("runtime-1").unwrap(); - let signer = RuntimeVerificationSigner::from_identity(&runtime_identity).unwrap(); - let public_key_fingerprint = format!( - "sha256:{}", - hex_lower(&Sha256::digest( - crate::auth::decode_public_key(&runtime_identity.public_key).unwrap() - )) - ); - let challenge = WorkspaceRuntimeVerificationChallenge { - challenge_id: "challenge-1".to_string(), - workspace_id: "workspace-a".to_string(), - runtime_id: "runtime-1".to_string(), - binding_id: "binding-test".into(), - workspace_key_id: "WK-1".to_string(), - workspace_public_key_fingerprint: format!("sha256:{}", "a".repeat(64)), - workspace_trust_id: "trust-test".into(), - runtime_public_key_fingerprint: public_key_fingerprint, - workspace_nonce: "workspace-nonce".to_string(), - expires_at: 1_100, - }; - let response = signer - .sign_response(&challenge, "runtime-nonce".to_string(), 1_000) - .unwrap(); - verify_runtime_verification_response( - &response, - &challenge, - &runtime_identity.public_key, - 1_001, - ) - .unwrap(); - - // A valid signature from a different key must not authenticate the - // fingerprint named by this challenge, even when the response repeats it. - let other_identity = RuntimeIdentityMaterial::generate("runtime-1").unwrap(); - let mut wrong_key = response.clone(); - wrong_key.response_proof = crate::auth::sign_json_token( - RUNTIME_VERIFICATION_TOKEN_PREFIX, - RUNTIME_VERIFICATION_SIGNING_INPUT_PREFIX, - &other_identity.signing_key().unwrap(), - &RuntimeVerificationResponseClaims { - response: response.unsigned(), - method: "POST".into(), - path_and_query: WORKSPACE_VERIFICATION_CHALLENGE_PATH.into(), - body_digest: workspace_request_body_digest( - &serde_json::to_vec(&challenge).unwrap(), - ), - iat: 1_000, - exp: challenge.expires_at, - jti: "different-key-proof".into(), - }, - ) - .unwrap(); - assert_eq!( - verify_runtime_verification_response( - &wrong_key, - &challenge, - &other_identity.public_key, - 1_001, - ), - Err(WorkspaceCapabilityVerificationError::VerificationChallengeMismatch) - ); - - let mut tampered = response.clone(); - tampered.binding_id = "other-binding".into(); - assert_eq!( - verify_runtime_verification_response( - &tampered, - &challenge, - &runtime_identity.public_key, - 1_001, - ), - Err(WorkspaceCapabilityVerificationError::VerificationChallengeMismatch) - ); - } } diff --git a/crates/workspace-server/src/frozen_runtime_binding_tests.rs b/crates/workspace-server/src/frozen_runtime_binding_tests.rs index 3ea64b2ab..fcf7dcc97 100644 --- a/crates/workspace-server/src/frozen_runtime_binding_tests.rs +++ b/crates/workspace-server/src/frozen_runtime_binding_tests.rs @@ -817,17 +817,12 @@ mod current_authority { fn binding_store() -> (SqliteWorkspaceStore, WorkspaceRuntimeBinding) { let store = SqliteWorkspaceStore::in_memory().unwrap(); - let workspace_key = - worker_runtime::auth::RuntimeIdentityMaterial::generate("workspace-key").unwrap(); let runtime_key = worker_runtime::auth::RuntimeIdentityMaterial::generate("runtime-key").unwrap(); - let workspace_fingerprint = normalize_runtime_public_key(&workspace_key.public_key) - .unwrap() - .1; store.with_conn(|conn| { conn.execute_batch("INSERT INTO accounts(account_id,kind,handle,display_name,created_at,updated_at) VALUES('owner','user','owner','Owner','created','updated'); - INSERT INTO workspaces(workspace_id,owner_account_id,display_name,state,created_at,updated_at) VALUES('space','owner','Space','active','created','updated');")?; - conn.execute("INSERT INTO workspace_signing_identities(workspace_id,key_id,algorithm,public_key,public_key_fingerprint,private_material_ref,state,created_at,provisioned_at,updated_at) VALUES('space','workspace-key','ed25519',?1,?2,'material','active','created','provisioned','updated')", params![workspace_key.public_key,workspace_fingerprint])?; + INSERT INTO workspaces(workspace_id,owner_account_id,display_name,state,created_at,updated_at) VALUES('space','owner','Space','active','created','updated'); + INSERT INTO workspace_signing_identities(workspace_id,key_id,algorithm,public_key,public_key_fingerprint,private_material_ref,state,created_at,provisioned_at,updated_at) VALUES('space','workspace-key','ed25519','public','sha256:workspace','material','active','created','provisioned','updated');")?; Ok(()) }).unwrap(); let request = WorkspaceRuntimeBinding { @@ -838,11 +833,7 @@ mod current_authority { public_key: runtime_key.public_key, public_key_fingerprint: String::new(), binding_id: String::new(), - state: WorkspaceRuntimeBindingState::Configured, authentication_mode: WorkspaceRuntimeAuthenticationMode::WorkspaceIdentity, - workspace_key_id: Some("workspace-key".into()), - workspace_public_key_fingerprint: Some(workspace_fingerprint), - workspace_trust_id: Some("runtime-issued-trust".into()), created_at: "created".into(), updated_at: "updated".into(), revoked_at: None, @@ -853,36 +844,10 @@ mod current_authority { (store, binding) } - fn evidence(binding: &WorkspaceRuntimeBinding) -> WorkspaceRuntimeVerificationEvidence { - WorkspaceRuntimeVerificationEvidence { - workspace_id: binding.workspace_id.clone(), - runtime_id: binding.runtime_id.clone(), - binding_id: binding.binding_id.clone(), - workspace_key_id: binding.workspace_key_id.clone().unwrap(), - workspace_public_key_fingerprint: binding - .workspace_public_key_fingerprint - .clone() - .unwrap(), - workspace_trust_id: binding.workspace_trust_id.clone().unwrap(), - runtime_public_key_fingerprint: binding.public_key_fingerprint.clone(), - challenge_id: "challenge".into(), - state: "verified".into(), - last_outcome: "verified".into(), - verified_at: Some("verified-at".into()), - checked_at: "checked-at".into(), - } - } - #[test] - fn same_key_reactivation_fences_old_cas_and_verification_evidence() { + fn same_key_reactivation_fences_old_config_and_admin_mutations() { let (store, binding) = binding_store(); - let old_proof = evidence(&binding); - store - .record_workspace_runtime_verification_attempt(&old_proof) - .unwrap(); - store - .complete_workspace_runtime_verification(&old_proof) - .unwrap(); + assert!(store.workspace_runtime_binding_matches(&binding).unwrap()); let (_, revoked) = store .revoke_workspace_runtime_binding_key( "space", @@ -893,16 +858,22 @@ mod current_authority { ) .unwrap(); assert_ne!(binding.binding_id, revoked.binding_id); - let mut replacement = binding.clone(); - replacement.state = WorkspaceRuntimeBindingState::Configured; + assert!(!store.workspace_runtime_binding_matches(&binding).unwrap()); + assert!(!store.workspace_runtime_binding_matches(&revoked).unwrap()); let (_, reactivated) = store - .put_workspace_runtime_binding_key(replacement, Some(&revoked.binding_id), "owner") + .put_workspace_runtime_binding_key(binding.clone(), Some(&revoked.binding_id), "owner") .unwrap(); assert_eq!( binding.public_key_fingerprint, reactivated.public_key_fingerprint ); assert_ne!(binding.binding_id, reactivated.binding_id); + assert!( + store + .workspace_runtime_binding_matches(&reactivated) + .unwrap() + ); + assert!(!store.workspace_runtime_binding_matches(&binding).unwrap()); assert!(matches!( store.put_workspace_runtime_binding_key( reactivated.clone(), @@ -921,42 +892,27 @@ mod current_authority { ), Err(Error::RuntimeBindingIdConflict { .. }) )); - assert!( - store - .record_workspace_runtime_verification_attempt(&old_proof) - .is_err() - ); - assert!( - store - .complete_workspace_runtime_verification(&old_proof) - .is_err() - ); - assert!( - !store - .workspace_runtime_verification_matches( - &reactivated, - &old_proof.workspace_public_key_fingerprint, - &old_proof.workspace_trust_id - ) - .unwrap() - ); } #[test] - fn endpoint_aba_and_workspace_key_change_cannot_reuse_verification_authority() { + fn endpoint_aba_fences_old_config_without_copying_workspace_signing_identity() { let (store, binding) = binding_store(); - let proof = evidence(&binding); - store - .record_workspace_runtime_verification_attempt(&proof) - .unwrap(); - store - .complete_workspace_runtime_verification(&proof) + let renamed = store + .update_workspace_runtime_binding_metadata( + "space", + "remote", + "Renamed", + &binding.base_url, + "renamed-at", + ) .unwrap(); + assert_eq!(binding.binding_id, renamed.binding_id); + assert!(!store.workspace_runtime_binding_matches(&binding).unwrap()); let moved = store .update_workspace_runtime_binding_metadata( "space", "remote", - "Remote", + "Renamed", "https://other.test", "changed", ) @@ -965,35 +921,166 @@ mod current_authority { .update_workspace_runtime_binding_metadata( "space", "remote", - "Remote", + "Renamed", "https://remote.test", "changed-back", ) .unwrap(); assert_ne!(moved.binding_id, binding.binding_id); assert_ne!(returned.binding_id, binding.binding_id); - assert!( - store - .record_workspace_runtime_verification_attempt(&proof) - .is_err() - ); - let current = evidence(&returned); - store - .record_workspace_runtime_verification_attempt(¤t) - .unwrap(); - store.with_conn(|conn| { conn.execute("UPDATE workspace_signing_identities SET public_key_fingerprint='sha256:replacement' WHERE workspace_id='space'",[])?; Ok(()) }).unwrap(); - assert!( - store - .complete_workspace_runtime_verification(¤t) - .is_err() - ); + assert!(!store.workspace_runtime_binding_matches(&renamed).unwrap()); + store.with_conn(|conn| { conn.execute("UPDATE workspace_signing_identities SET public_key_fingerprint='sha256:replacement' WHERE workspace_id='space'", [])?; Ok(()) }).unwrap(); assert_eq!( store .get_workspace_runtime_binding("space", "remote") + .unwrap(), + Some(returned.clone()) + ); + assert!(store.workspace_runtime_binding_matches(&returned).unwrap()); + } + + fn rows(conn: &Connection, sql: &str) -> Vec> { + let mut stmt = conn.prepare(sql).unwrap(); + let count = stmt.column_count(); + stmt.query_map([], |row| (0..count).map(|i| row.get(i)).collect()) + .unwrap() + .collect::>>() + .unwrap() + } + + fn schema89_bindings() -> Connection { + let conn = legacy_binding_database(); + migrate_runtime_bindings_v87_to_v88(&conn).unwrap(); + conn.execute( + "INSERT INTO __yoi_schema_migrations VALUES(89,'descriptive legacy evidence columns')", + [], + ) + .unwrap(); + conn.execute_batch("INSERT INTO workspace_runtime_bindings VALUES + ('space','configured-trust','Configured','https://configured.test','key-c','fp-c','binding-c','configured','workspace_identity','workspace-key','sha256:workspace','trust-c','created-c','updated-c',NULL), + ('space','verified','Verified','https://verified.test','key-v','fp-v','binding-v','verified','workspace_identity','workspace-key','sha256:workspace','trust-v','created-v','updated-v',NULL), + ('space','revoked-null-trust','Revoked','https://revoked.test','key-r','fp-r','binding-r','revoked','workspace_identity','workspace-key',NULL,NULL,'created-r','updated-r','revoked-r'), + ('space','legacy-verified','Legacy','https://legacy.test','key-l','fp-l','binding-l','verified','legacy_server_issuer',NULL,NULL,NULL,'created-l','updated-l',NULL), + ('space','legacy-revoked','Legacy revoked','https://legacy-r.test','key-lr','fp-lr','binding-lr','revoked','legacy_server_issuer',NULL,NULL,NULL,'created-lr','updated-lr','revoked-lr'); + INSERT INTO workspace_runtime_verifications VALUES + ('space','verified','binding-v','workspace-key','sha256:workspace','trust-v','fp-v','challenge-v','verified','verified','verified-at','checked-v'), + ('space','configured-trust','binding-c','workspace-key','sha256:workspace','trust-c','fp-c','challenge-c','pending','challenge_issued',NULL,'checked-c'), + ('space','revoked-null-trust','binding-r','workspace-key','sha256:workspace','old-trust','fp-r','challenge-r','failed','verification_failed',NULL,'checked-r');").unwrap(); + conn + } + + #[test] + fn schema90_preserves_connections_and_archives_all_auth_metadata_without_reenrollment() { + let conn = schema89_bindings(); + let connections = rows( + &conn, + "SELECT workspace_id,runtime_id,display_name,base_url,public_key,public_key_fingerprint,binding_id,authentication_mode,created_at,updated_at,revoked_at FROM workspace_runtime_bindings ORDER BY runtime_id", + ); + let metadata = rows( + &conn, + "SELECT workspace_id,runtime_id,binding_id,state,workspace_key_id,workspace_public_key_fingerprint,workspace_trust_id,updated_at FROM workspace_runtime_bindings ORDER BY runtime_id", + ); + let proofs = rows( + &conn, + "SELECT * FROM workspace_runtime_verifications ORDER BY runtime_id", + ); + let identities = rows(&conn, "SELECT * FROM workspace_signing_identities"); + let audit = rows(&conn, "SELECT * FROM workspace_runtime_binding_audit"); + let removal = rows(&conn, "SELECT * FROM runtime_removal_operations"); + migrate_runtime_binding_connections_v89_to_v90(&conn).unwrap(); + assert_eq!(current_schema_version(&conn).unwrap(), 90); + assert_eq!( + rows( + &conn, + "SELECT * FROM workspace_runtime_bindings ORDER BY runtime_id" + ), + connections + ); + assert_eq!( + rows( + &conn, + "SELECT * FROM workspace_runtime_binding_verification_archive ORDER BY runtime_id" + ), + metadata + ); + assert_eq!( + rows( + &conn, + "SELECT * FROM workspace_runtime_verification_archive ORDER BY runtime_id" + ), + proofs + ); + assert_eq!( + rows(&conn, "SELECT * FROM workspace_signing_identities"), + identities + ); + assert_eq!( + rows(&conn, "SELECT * FROM workspace_runtime_binding_audit"), + audit + ); + assert_eq!( + rows(&conn, "SELECT * FROM runtime_removal_operations"), + removal + ); + verify_runtime_binding_connection_schema(&conn).unwrap(); + assert!(!table_exists(&conn, "workspace_runtime_verifications").unwrap()); + assert!( + conn.query_row("PRAGMA foreign_keys", [], |r| r.get::<_, bool>(0)) .unwrap() + ); + assert_eq!( + rows(&conn, "SELECT * FROM pragma_foreign_key_check"), + Vec::>::new() + ); + conn.execute_batch("UPDATE workspace_runtime_bindings SET base_url='https://moved.test',binding_id='binding-new' WHERE runtime_id='verified'; + UPDATE workspace_runtime_bindings SET revoked_at='admin-revoked' WHERE runtime_id='configured-trust'; + DELETE FROM workspace_runtime_bindings WHERE runtime_id='legacy-revoked';").unwrap(); + assert_eq!( + rows( + &conn, + "SELECT * FROM workspace_runtime_binding_verification_archive ORDER BY runtime_id" + ), + metadata + ); + assert_eq!( + rows( + &conn, + "SELECT * FROM workspace_runtime_verification_archive ORDER BY runtime_id" + ), + proofs + ); + conn.execute( + "UPDATE runtime_removal_operations SET state='cleanup_pending'", + [], + ) + .unwrap(); + assert!(conn.execute("UPDATE workspace_runtime_bindings SET display_name='Blocked' WHERE runtime_id='remote'", []).unwrap_err().to_string().contains("runtime_removal_in_progress")); + } + + #[test] + fn schema90_failed_rebuild_rolls_back_connections_archives_and_schema_marker() { + let conn = schema89_bindings(); + // An incompatible preexisting archive is not permission to discard historical evidence. + conn.execute_batch("CREATE TABLE workspace_runtime_verification_archive(existing TEXT);") + .unwrap(); + let before = rows( + &conn, + "SELECT * FROM workspace_runtime_bindings ORDER BY runtime_id", + ); + assert!(migrate_runtime_binding_connections_v89_to_v90(&conn).is_err()); + assert_eq!(current_schema_version(&conn).unwrap(), 89); + assert_eq!( + rows( + &conn, + "SELECT * FROM workspace_runtime_bindings ORDER BY runtime_id" + ), + before + ); + assert!(!table_exists(&conn, "workspace_runtime_binding_verification_archive").unwrap()); + assert!(table_exists(&conn, "workspace_runtime_verifications").unwrap()); + assert!( + conn.query_row("PRAGMA foreign_keys", [], |r| r.get::<_, bool>(0)) .unwrap() - .state, - WorkspaceRuntimeBindingState::Configured ); } } diff --git a/crates/workspace-server/src/hosts.rs b/crates/workspace-server/src/hosts.rs index 8a5e3aa2c..cb8dfa169 100644 --- a/crates/workspace-server/src/hosts.rs +++ b/crates/workspace-server/src/hosts.rs @@ -2,9 +2,7 @@ use crate::Error; use crate::resource_broker::BackendResourceBroker; #[cfg(test)] use crate::resource_broker::BackendResourceTarget; -use crate::store::{ - ControlPlaneStore, WorkspaceRuntimeAuthenticationMode, WorkspaceRuntimeBindingState, -}; +use crate::store::{ControlPlaneStore, WorkspaceRuntimeAuthenticationMode}; use crate::workspace_signing_identity::WorkspaceSigningIdentityService; use chrono::Utc; use protocol::Segment; @@ -66,11 +64,7 @@ use worker_runtime::ssh_host_key_probe::{ SSH_HOST_KEY_PROBE_OPERATION, SSH_HOST_KEY_PROBE_PATH, SshHostKeyProbeRequest, SshHostKeyProbeResponse, }; -use worker_runtime::workspace_issuer::{ - WorkspaceCapabilityClaims, WorkspaceRuntimeVerificationAcknowledgement, - WorkspaceRuntimeVerificationChallenge, WorkspaceRuntimeVerificationReceipt, - WorkspaceRuntimeVerificationResponse, workspace_request_body_digest, -}; +use worker_runtime::workspace_issuer::{WorkspaceCapabilityClaims, workspace_request_body_digest}; pub const EMBEDDED_RUNTIME_ID: &str = "embedded-worker-runtime"; const EMBEDDED_HOST_KIND: &str = "embedded-worker-runtime-host"; @@ -925,32 +919,6 @@ pub trait WorkspaceWorkerRuntime: Send + Sync { }) } - fn activate_workspace_authorization(&self, _binding: crate::store::WorkspaceRuntimeBinding) {} - - fn send_workspace_verification_challenge( - &self, - _challenge: &WorkspaceRuntimeVerificationChallenge, - _bearer_token: &str, - ) -> Result { - Err(RuntimePingFailure::new( - RuntimePingFailureKind::Unsupported, - "runtime_workspace_verification_unsupported", - "Workspace Runtime verification is unavailable for this Runtime provider", - )) - } - - fn send_workspace_verification_acknowledgement( - &self, - _acknowledgement: &WorkspaceRuntimeVerificationAcknowledgement, - _bearer_token: &str, - ) -> Result { - Err(RuntimePingFailure::new( - RuntimePingFailureKind::Unsupported, - "runtime_workspace_verification_unsupported", - "Workspace Runtime verification is unavailable for this Runtime provider", - )) - } - fn list_hosts(&self, limit: usize) -> RuntimeList; fn list_workers(&self, limit: usize) -> RuntimeList; @@ -2277,48 +2245,6 @@ impl RuntimeRegistry { runtime.ping() } - pub fn activate_workspace_authorization( - &self, - runtime_id: &str, - binding: crate::store::WorkspaceRuntimeBinding, - ) -> Result<(), RuntimeRegistryError> { - self.runtime(runtime_id)? - .activate_workspace_authorization(binding); - Ok(()) - } - - pub fn send_workspace_verification_challenge( - &self, - runtime_id: &str, - challenge: &WorkspaceRuntimeVerificationChallenge, - bearer_token: &str, - ) -> Result { - let runtime = self.runtime(runtime_id).map_err(|_| { - RuntimePingFailure::new( - RuntimePingFailureKind::Configuration, - "runtime_verification_registration_unavailable", - "Registered Runtime binding is unavailable", - ) - })?; - runtime.send_workspace_verification_challenge(challenge, bearer_token) - } - - pub fn send_workspace_verification_acknowledgement( - &self, - runtime_id: &str, - acknowledgement: &WorkspaceRuntimeVerificationAcknowledgement, - bearer_token: &str, - ) -> Result { - let runtime = self.runtime(runtime_id).map_err(|_| { - RuntimePingFailure::new( - RuntimePingFailureKind::Configuration, - "runtime_verification_registration_unavailable", - "Registered Runtime binding is unavailable", - ) - })?; - runtime.send_workspace_verification_acknowledgement(acknowledgement, bearer_token) - } - fn runtimes_snapshot(&self) -> Vec> { self.runtimes .read() @@ -3518,7 +3444,9 @@ pub struct WorkspaceRuntimeAuthorization { store: Arc, signing_identities: WorkspaceSigningIdentityService, backend_url: String, - binding: Arc>>, + // The connection selected for this client. Compare it with the current + // administrative configuration before sending, never with a past auth result. + binding: crate::store::WorkspaceRuntimeBinding, } impl std::fmt::Debug for WorkspaceRuntimeAuthorization { @@ -3535,19 +3463,13 @@ impl WorkspaceRuntimeAuthorization { store: Arc, signing_identities: WorkspaceSigningIdentityService, backend_url: impl Into, - binding: Option, + binding: crate::store::WorkspaceRuntimeBinding, ) -> Self { Self { store, signing_identities, backend_url: backend_url.into(), - binding: Arc::new(RwLock::new(binding)), - } - } - - fn activate(&self, binding: crate::store::WorkspaceRuntimeBinding) { - if let Ok(mut current) = self.binding.write() { - *current = Some(binding); + binding, } } @@ -3559,30 +3481,14 @@ impl WorkspaceRuntimeAuthorization { worker_id: Option<&str>, body: &[u8], ) -> Result { - let binding = self - .binding - .read() - .map_err(|_| { - diagnostic( - "workspace_runtime_authorization_unavailable", - HostDiagnosticSeverity::Error, - "Workspace Runtime authorization is unavailable".to_string(), - ) - })? - .clone() - .ok_or_else(|| { - diagnostic( - "workspace_runtime_verification_required", - HostDiagnosticSeverity::Error, - "Workspace Runtime binding is not verified".to_string(), - ) - })?; - if binding.state != WorkspaceRuntimeBindingState::Verified - || binding.authentication_mode != WorkspaceRuntimeAuthenticationMode::WorkspaceIdentity + let binding = &self.binding; + // A removed/replaced endpoint must not be used by an old client. This + // compares connection configuration, not cached authentication success. + if binding.authentication_mode != WorkspaceRuntimeAuthenticationMode::WorkspaceIdentity || binding.revoked_at.is_some() || !self .store - .workspace_runtime_binding_matches(&binding) + .workspace_runtime_binding_matches(binding) .map_err(|error| { diagnostic( "workspace_runtime_authorization_unavailable", @@ -3594,7 +3500,7 @@ impl WorkspaceRuntimeAuthorization { return Err(diagnostic( "workspace_runtime_authorization_stale", HostDiagnosticSeverity::Error, - "Workspace Runtime binding changed or was revoked".to_string(), + "Workspace Runtime connection changed or was revoked".to_string(), )); } let identity = self @@ -3607,63 +3513,26 @@ impl WorkspaceRuntimeAuthorization { error.to_string(), ) })?; - let workspace_key_id = binding.workspace_key_id.as_deref().ok_or_else(|| { - diagnostic( - "workspace_runtime_authorization_invalid", - HostDiagnosticSeverity::Error, - "Workspace Runtime binding is missing its Workspace key".to_string(), - ) - })?; - let workspace_public_key_fingerprint = binding - .workspace_public_key_fingerprint - .as_deref() - .ok_or_else(|| { - diagnostic( - "workspace_runtime_authorization_invalid", - HostDiagnosticSeverity::Error, - "Workspace Runtime binding is missing its Workspace public key fingerprint" - .to_string(), - ) - })?; - let workspace_trust_id = binding.workspace_trust_id.as_deref().ok_or_else(|| { - diagnostic( - "workspace_runtime_authorization_invalid", - HostDiagnosticSeverity::Error, - "Workspace Runtime binding is missing its Workspace trust ID".to_string(), - ) - })?; - if identity.state != "active" - || identity.key_id != workspace_key_id - || identity.public_key_fingerprint.as_deref() != Some(workspace_public_key_fingerprint) - || !self - .store - .workspace_runtime_verification_matches( - &binding, - workspace_public_key_fingerprint, - workspace_trust_id, - ) - .map_err(|error| { - diagnostic( - "workspace_runtime_authorization_unavailable", - HostDiagnosticSeverity::Error, - error.to_string(), - ) - })? - { + if identity.state != "active" { return Err(diagnostic( - "workspace_runtime_authorization_stale", + "workspace_signing_identity_unavailable", HostDiagnosticSeverity::Error, - "Workspace signing identity no longer matches the verified binding".to_string(), + "Workspace signing identity is not active".to_string(), )); } + let fingerprint = identity.public_key_fingerprint.ok_or_else(|| { + diagnostic( + "workspace_signing_identity_unavailable", + HostDiagnosticSeverity::Error, + "Workspace signing identity has no public key fingerprint".to_string(), + ) + })?; let now = Utc::now().timestamp(); let claims = WorkspaceCapabilityClaims { issuer: self.backend_url.clone(), issuer_workspace_id: binding.workspace_id.clone(), issuer_key_id: identity.key_id, - issuer_public_key_fingerprint: workspace_public_key_fingerprint.to_string(), - trust_id: workspace_trust_id.to_string(), - binding_id: binding.binding_id.clone(), + issuer_public_key_fingerprint: fingerprint, runtime_id: binding.runtime_id.clone(), worker_id: worker_id.map(str::to_string), operation: operation.to_string(), @@ -3674,6 +3543,8 @@ impl WorkspaceRuntimeAuthorization { exp: now.saturating_add(60), jti: uuid::Uuid::now_v7().to_string(), }; + // The signer validates that these claims still name the selected current + // identity; a concurrent rotation cannot silently sign mismatched claims. self.signing_identities .issue_workspace_capability(&binding.workspace_id, &claims) .map_err(|error| { @@ -4251,49 +4122,6 @@ impl RemoteWorkerRuntime { format!("{base}/v1/workers/{worker_id}/protocol/ws") } - fn post_bearer_json( - &self, - path: &str, - body: &T, - bearer_token: &str, - ) -> Result - where - T: Serialize + ?Sized, - U: DeserializeOwned, - { - let response = self - .http - .post(self.endpoint(path)) - .bearer_auth(bearer_token) - .json(body) - .send() - .map_err(|error| { - RuntimePingFailure::new( - RuntimePingFailureKind::NetworkUnreachable, - "runtime_workspace_verification_unreachable", - format!("Runtime verification request failed: {error}"), - ) - })?; - let status = response.status(); - if !status.is_success() { - return Err(RuntimePingFailure::new( - RuntimePingFailureKind::MalformedResponse, - "runtime_workspace_verification_rejected", - format!( - "Runtime verification request returned HTTP {}", - status.as_u16() - ), - )); - } - response.json::().map_err(|error| { - RuntimePingFailure::new( - RuntimePingFailureKind::MalformedResponse, - "runtime_workspace_verification_invalid_response", - format!("Runtime verification response was invalid: {error}"), - ) - }) - } - fn get_json(&self, path: &str) -> Result where T: DeserializeOwned + Send + 'static, @@ -4702,36 +4530,6 @@ impl WorkspaceWorkerRuntime for RemoteWorkerRuntime { }) } - fn activate_workspace_authorization(&self, binding: crate::store::WorkspaceRuntimeBinding) { - if let Some(authorization) = &self.workspace_authorization { - authorization.activate(binding); - } - } - - fn send_workspace_verification_challenge( - &self, - challenge: &WorkspaceRuntimeVerificationChallenge, - bearer_token: &str, - ) -> Result { - self.post_bearer_json( - worker_runtime::workspace_issuer::WORKSPACE_VERIFICATION_CHALLENGE_PATH, - challenge, - bearer_token, - ) - } - - fn send_workspace_verification_acknowledgement( - &self, - acknowledgement: &WorkspaceRuntimeVerificationAcknowledgement, - bearer_token: &str, - ) -> Result { - self.post_bearer_json( - worker_runtime::workspace_issuer::WORKSPACE_VERIFICATION_ACK_PATH, - acknowledgement, - bearer_token, - ) - } - fn list_hosts(&self, limit: usize) -> RuntimeList { if limit == 0 { return RuntimeList::new(Vec::new(), Vec::new()); @@ -6536,6 +6334,7 @@ pub fn placeholder_spawn_response(host_id: impl Into) -> WorkerSpawnResu #[cfg(test)] mod tests { mod restore_auth_tests; + mod workspace_auth_tests; use super::*; use serde_json::json; diff --git a/crates/workspace-server/src/hosts/tests/restore_auth_tests.rs b/crates/workspace-server/src/hosts/tests/restore_auth_tests.rs index 82a9db0b9..8e77550f5 100644 --- a/crates/workspace-server/src/hosts/tests/restore_auth_tests.rs +++ b/crates/workspace-server/src/hosts/tests/restore_auth_tests.rs @@ -9,36 +9,18 @@ use worker_runtime::{ auth::RuntimeIdentityMaterial, http_server::{WorkspaceRuntimeHttpAuth, runtime_http_router_with_workspace_auth}, workspace_issuer::{ - InMemoryWorkspaceClaimReplayProtection, InMemoryWorkspaceRuntimeVerificationAuthority, - RuntimeVerificationSigner, WorkspaceCapabilityVerifier, WorkspaceIssuerTrustRecord, - WorkspaceIssuerTrustState, WorkspaceRuntimeVerificationAuthority, - WorkspaceRuntimeVerificationRecord, issue_workspace_capability_token, + InMemoryWorkspaceClaimReplayProtection, WorkspaceCapabilityVerifier, + WorkspaceIssuerTrustRecord, WorkspaceIssuerTrustState, issue_workspace_capability_token, }, }; #[tokio::test] async fn restore_capabilities_authorize_runtime_routes_without_relaxing_request_binding() { let workspace_identity = RuntimeIdentityMaterial::generate("workspace-key").unwrap(); - let runtime_identity = RuntimeIdentityMaterial::generate("runtime-test").unwrap(); - let signer = RuntimeVerificationSigner::from_identity(&runtime_identity).unwrap(); let now = Utc::now().timestamp(); let workspace_public_key_fingerprint = crate::workspace_signing_identity::public_key_fingerprint(&workspace_identity.public_key) .unwrap(); - let binding_id = "binding-restore-test".to_string(); - let verifications = Arc::new(InMemoryWorkspaceRuntimeVerificationAuthority::default()); - verifications - .record(WorkspaceRuntimeVerificationRecord { - workspace_id: "workspace-test".into(), - runtime_id: "runtime-test".into(), - binding_id: binding_id.clone(), - workspace_trust_id: "trust-restore-test".to_string(), - workspace_key_id: "workspace-key".into(), - workspace_public_key_fingerprint: workspace_public_key_fingerprint.clone(), - runtime_public_key_fingerprint: signer.public_key_fingerprint().to_owned(), - verified_at: now, - }) - .unwrap(); let public_key = worker_runtime::auth::decode_public_key(&workspace_identity.public_key).unwrap(); let verifier = WorkspaceCapabilityVerifier::new( @@ -69,8 +51,7 @@ async fn restore_capabilities_authorize_runtime_routes_without_relaxing_request_ None, WorkspaceRuntimeHttpAuth { verifier, - signer, - verifications, + runtime_id: "runtime-test".into(), }, ); let worker_id = EmbeddedWorkerId::from_legacy_u64(1).to_string(); @@ -93,8 +74,6 @@ async fn restore_capabilities_authorize_runtime_routes_without_relaxing_request_ "wrong-worker", "wrong-body", "wrong-key-fingerprint", - "wrong-binding-id", - "wrong-trust-id", ] { let claims = WorkspaceCapabilityClaims { issuer: "https://backend.test".into(), @@ -111,16 +90,6 @@ async fn restore_capabilities_authorize_runtime_routes_without_relaxing_request_ } else { workspace_public_key_fingerprint.clone() }, - binding_id: if case == "wrong-binding-id" { - "binding-retired".to_string() - } else { - binding_id.clone() - }, - trust_id: if case == "wrong-trust-id" { - "trust-retired".to_string() - } else { - "trust-restore-test".to_string() - }, runtime_id: "runtime-test".into(), worker_id: if case == "wrong-worker" { Some(EmbeddedWorkerId::from_legacy_u64(2).to_string()) @@ -176,27 +145,16 @@ async fn restore_capabilities_authorize_runtime_routes_without_relaxing_request_ } else { assert_eq!( status, - if case == "wrong-binding-id" { - StatusCode::FORBIDDEN - } else { - StatusCode::UNAUTHORIZED - }, + StatusCode::UNAUTHORIZED, "{path} {case}: {}", error.error.message ); - assert_eq!( - error.error.code, - if case == "wrong-binding-id" { - "workspace_runtime_verification_stale" - } else { - "unauthorized" - } - ); + assert_eq!(error.error.code, "unauthorized"); let expected = match case { "wrong-operation" => Some("does not authorize this operation"), "wrong-worker" => Some("targets another Worker"), "wrong-body" => Some("does not bind this request body"), - "wrong-key-fingerprint" | "wrong-binding-id" | "wrong-trust-id" => None, + "wrong-key-fingerprint" => None, _ => unreachable!(), }; if let Some(expected) = expected { diff --git a/crates/workspace-server/src/hosts/tests/workspace_auth_tests.rs b/crates/workspace-server/src/hosts/tests/workspace_auth_tests.rs new file mode 100644 index 000000000..f68964f35 --- /dev/null +++ b/crates/workspace-server/src/hosts/tests/workspace_auth_tests.rs @@ -0,0 +1,316 @@ +use super::*; +use crate::store::{ + AccountRecord, SqliteWorkspaceStore, WorkspaceRecord, WorkspaceRuntimeBinding, + WorkspaceSigningIdentityRecord, +}; +use crate::workspace_signing_identity::{ + InMemoryWorkspaceSigningMaterialStore, WorkspaceSigningMaterialStore, + WorkspaceSigningPrivateMaterial, public_key_fingerprint, +}; +use worker_runtime::auth::RuntimeIdentityMaterial; +use worker_runtime::workspace_issuer::{ + InMemoryWorkspaceClaimReplayProtection, VerifiedWorkspaceCapability, + WorkspaceCapabilityExpectation, WorkspaceCapabilityVerificationError, + WorkspaceCapabilityVerifier, WorkspaceIssuerTrustRecord, WorkspaceIssuerTrustState, +}; + +const WORKSPACE: &str = "workspace-auth-test"; +const RUNTIME: &str = "runtime-auth-test"; +const BACKEND: &str = "https://backend.test"; +const PATH: &str = "/v1/ping?probe=1"; + +struct Fixture { + _temp: tempfile::TempDir, + store: Arc, + materials: Arc, + service: WorkspaceSigningIdentityService, + authorizer: WorkspaceRuntimeAuthorization, + binding: WorkspaceRuntimeBinding, + identity: WorkspaceSigningIdentityRecord, +} + +impl Fixture { + async fn new() -> Self { + let temp = tempfile::tempdir().unwrap(); + let store = Arc::new(SqliteWorkspaceStore::open(temp.path().join("server.db")).unwrap()); + store + .upsert_account(&AccountRecord { + account_id: "owner".into(), + kind: "user".into(), + handle: "owner".into(), + display_name: "Owner".into(), + created_at: "1".into(), + updated_at: "1".into(), + }) + .unwrap(); + store + .upsert_workspace(&WorkspaceRecord { + workspace_id: WORKSPACE.into(), + owner_account_id: "owner".into(), + display_name: "Workspace".into(), + state: "active".into(), + created_at: "1".into(), + updated_at: "1".into(), + }) + .await + .unwrap(); + let materials = Arc::new(InMemoryWorkspaceSigningMaterialStore::default()); + let service = WorkspaceSigningIdentityService::new(store.clone(), materials.clone()); + let identity = service.provision_existing(WORKSPACE, "owner").unwrap(); + let runtime_identity = RuntimeIdentityMaterial::generate(RUNTIME).unwrap(); + store + .upsert_workspace_runtime_binding( + WorkspaceRuntimeBinding { + workspace_id: WORKSPACE.into(), + runtime_id: RUNTIME.into(), + display_name: "Runtime".into(), + base_url: "https://runtime.test".into(), + public_key_fingerprint: public_key_fingerprint(&runtime_identity.public_key) + .unwrap(), + public_key: runtime_identity.public_key, + binding_id: String::new(), + authentication_mode: WorkspaceRuntimeAuthenticationMode::WorkspaceIdentity, + created_at: "1".into(), + updated_at: "1".into(), + revoked_at: None, + }, + false, + ) + .unwrap(); + let binding = store + .get_workspace_runtime_binding(WORKSPACE, RUNTIME) + .unwrap() + .unwrap(); + let authorizer = WorkspaceRuntimeAuthorization::new( + store.clone(), + service.clone(), + BACKEND, + binding.clone(), + ); + Self { + _temp: temp, + store, + materials, + service, + authorizer, + binding, + identity, + } + } + + fn issue(&self) -> Result { + self.authorizer + .issue("GET", PATH, RUNTIME_PING_PERMISSION, None, b"") + } + + fn verifier(&self) -> WorkspaceCapabilityVerifier { + WorkspaceCapabilityVerifier::new( + vec![WorkspaceIssuerTrustRecord { + workspace_id: WORKSPACE.into(), + backend_url: BACKEND.into(), + key_id: self.identity.key_id.clone(), + algorithm: "ed25519".into(), + public_key: self.identity.public_key.clone().unwrap(), + public_key_fingerprint: self.identity.public_key_fingerprint.clone().unwrap(), + trust_id: uuid::Uuid::now_v7().to_string(), + state: WorkspaceIssuerTrustState::Active, + registered_at_unix: Utc::now().timestamp(), + updated_at_unix: Utc::now().timestamp(), + }], + Arc::new(InMemoryWorkspaceClaimReplayProtection::default()), + ) + .unwrap() + } + + // Fixture-only rotation: deliberately no production rotation API or live database. + fn rotate_identity(&mut self) { + let key_id = "WK-rotated"; + let material_ref = format!("workspace-signing/{WORKSPACE}/{key_id}"); + let material = WorkspaceSigningPrivateMaterial::generate(WORKSPACE, key_id).unwrap(); + let public_key = material.validate_and_public_key(WORKSPACE, key_id).unwrap(); + let fingerprint = public_key_fingerprint(&public_key).unwrap(); + self.materials + .put_if_absent(&material_ref, &material) + .unwrap(); + self.store.with_conn(|conn| { + conn.execute("UPDATE workspace_signing_identities SET key_id = ?1, public_key = ?2, public_key_fingerprint = ?3, private_material_ref = ?4 WHERE workspace_id = ?5", + rusqlite::params![key_id, public_key, fingerprint, material_ref, WORKSPACE])?; + Ok(()) + }).unwrap(); + self.identity = self.service.get_validated(WORKSPACE).unwrap(); + } +} + +fn verify( + verifier: &WorkspaceCapabilityVerifier, + token: &str, +) -> Result { + verifier.verify( + token, + &WorkspaceCapabilityExpectation { + workspace_id: WORKSPACE, + runtime_id: RUNTIME, + worker_id: None, + operation: RUNTIME_PING_PERMISSION, + method: "GET", + path_and_query: PATH, + body_digest: &workspace_request_body_digest(b""), + now_unix: Utc::now().timestamp(), + }, + ) +} + +#[tokio::test] +async fn authorizer_token_authenticates_actual_request_without_prior_verification_record() { + let fixture = Fixture::new().await; + let token = fixture.issue().unwrap(); + let verifier = fixture.verifier(); + let authenticated = verify(&verifier, &token).unwrap(); + assert_eq!(authenticated.workspace_id, WORKSPACE); + assert_eq!(authenticated.issuer_key_id, fixture.identity.key_id); + assert_eq!( + verify(&verifier, &token).unwrap_err(), + WorkspaceCapabilityVerificationError::Replay + ); + let token = fixture.issue().unwrap(); + let wrong_request = WorkspaceCapabilityExpectation { + workspace_id: WORKSPACE, + runtime_id: RUNTIME, + worker_id: None, + operation: RUNTIME_PING_PERMISSION, + method: "GET", + path_and_query: "/v1/ping?probe=2", + body_digest: &workspace_request_body_digest(b""), + now_unix: Utc::now().timestamp(), + }; + assert_eq!( + verifier.verify(&token, &wrong_request).unwrap_err(), + WorkspaceCapabilityVerificationError::WrongPathAndQuery + ); +} + +#[tokio::test] +async fn existing_authorizer_selects_rotated_workspace_key_and_new_runtime_trust() { + let mut fixture = Fixture::new().await; + let old_verifier = fixture.verifier(); + let old_token = fixture.issue().unwrap(); + fixture.rotate_identity(); + let new_token = fixture.issue().unwrap(); + let new_verifier = fixture.verifier(); + assert_eq!( + verify(&new_verifier, &new_token).unwrap().issuer_key_id, + "WK-rotated" + ); + assert_eq!( + verify(&new_verifier, &old_token).unwrap_err(), + WorkspaceCapabilityVerificationError::WrongIssuerKey + ); + assert_eq!( + verify(&old_verifier, &new_token).unwrap_err(), + WorkspaceCapabilityVerificationError::WrongIssuerKey + ); + // Changing Workspace authority did not replace or copy metadata into the connection. + assert_eq!( + fixture + .store + .get_workspace_runtime_binding(WORKSPACE, RUNTIME) + .unwrap() + .unwrap(), + fixture.binding + ); +} + +#[tokio::test] +async fn replaced_or_revoked_connection_rejects_stale_authorizer() { + for revoked in [false, true] { + let fixture = Fixture::new().await; + fixture.issue().unwrap(); + if revoked { + fixture + .store + .revoke_workspace_runtime_binding(WORKSPACE, RUNTIME, "2") + .unwrap(); + } else { + let mut replacement = fixture.binding.clone(); + replacement.base_url = "https://replacement-runtime.test".into(); + fixture + .store + .upsert_workspace_runtime_binding(replacement, true) + .unwrap(); + } + assert_eq!( + fixture.issue().unwrap_err().code, + "workspace_runtime_authorization_stale" + ); + } +} + +#[tokio::test] +async fn missing_workspace_signing_material_rejects_authorization_without_fallback() { + let fixture = Fixture::new().await; + fixture + .materials + .delete(&fixture.identity.private_material_ref) + .unwrap(); + let error = fixture.issue().unwrap_err(); + assert_eq!(error.code, "workspace_runtime_authorization_unavailable"); + assert!( + error.message.contains("private material is missing"), + "{}", + error.message + ); +} + +#[tokio::test] +async fn capability_claims_must_match_current_workspace_identity_before_signing() { + let mut fixture = Fixture::new().await; + let now = Utc::now().timestamp(); + let mut claims = WorkspaceCapabilityClaims { + issuer: BACKEND.into(), + issuer_workspace_id: WORKSPACE.into(), + issuer_key_id: fixture.identity.key_id.clone(), + issuer_public_key_fingerprint: fixture.identity.public_key_fingerprint.clone().unwrap(), + runtime_id: RUNTIME.into(), + worker_id: None, + operation: RUNTIME_PING_PERMISSION.into(), + method: "GET".into(), + path_and_query: PATH.into(), + body_digest: workspace_request_body_digest(b""), + iat: now, + exp: now + 60, + jti: uuid::Uuid::now_v7().to_string(), + }; + let token = fixture + .service + .issue_workspace_capability(WORKSPACE, &claims) + .unwrap(); + verify(&fixture.verifier(), &token).unwrap(); + for field in ["workspace", "key", "fingerprint"] { + let mut mismatched = claims.clone(); + match field { + "workspace" => mismatched.issuer_workspace_id = "another-workspace".into(), + "key" => mismatched.issuer_key_id = "WK-another".into(), + "fingerprint" => { + mismatched.issuer_public_key_fingerprint = format!("sha256:{}", "0".repeat(64)) + } + _ => unreachable!(), + } + assert!( + matches!(fixture.service.issue_workspace_capability(WORKSPACE, &mismatched), + Err(Error::WorkspaceSigningIdentity { ref code, .. }) if code == "workspace_capability_identity_mismatch"), + "{field}" + ); + } + fixture.rotate_identity(); + assert!( + matches!(fixture.service.issue_workspace_capability(WORKSPACE, &claims), + Err(Error::WorkspaceSigningIdentity { ref code, .. }) if code == "workspace_capability_identity_mismatch") + ); + claims.issuer_key_id = fixture.identity.key_id.clone(); + claims.issuer_public_key_fingerprint = fixture.identity.public_key_fingerprint.clone().unwrap(); + let token = fixture + .service + .issue_workspace_capability(WORKSPACE, &claims) + .unwrap(); + verify(&fixture.verifier(), &token).unwrap(); +} diff --git a/crates/workspace-server/src/latest_schema.sql b/crates/workspace-server/src/latest_schema.sql index 341645d6d..1ce68d15e 100644 --- a/crates/workspace-server/src/latest_schema.sql +++ b/crates/workspace-server/src/latest_schema.sql @@ -442,49 +442,43 @@ CREATE TABLE workspace_runtime_bindings ( public_key TEXT NOT NULL, public_key_fingerprint TEXT NOT NULL, binding_id TEXT NOT NULL CHECK (length(binding_id) > 0), - state TEXT NOT NULL CHECK (state IN ('configured', 'verified', 'revoked')), authentication_mode TEXT NOT NULL CHECK (authentication_mode IN ('legacy_server_issuer', 'workspace_identity')), - workspace_key_id TEXT, - workspace_public_key_fingerprint TEXT, - workspace_trust_id TEXT CHECK (length(workspace_trust_id) > 0), created_at TEXT NOT NULL, updated_at TEXT NOT NULL, revoked_at TEXT, PRIMARY KEY (workspace_id, runtime_id), UNIQUE (workspace_id, public_key_fingerprint), - FOREIGN KEY(workspace_id) REFERENCES workspaces(workspace_id) ON DELETE RESTRICT, - CHECK ( - (authentication_mode = 'legacy_server_issuer' AND workspace_key_id IS NULL AND workspace_public_key_fingerprint IS NULL AND workspace_trust_id IS NULL) - OR - (authentication_mode = 'workspace_identity' AND workspace_key_id IS NOT NULL AND (state != 'verified' OR (workspace_trust_id IS NOT NULL AND workspace_public_key_fingerprint IS NOT NULL))) - ), - CHECK ( - (state = 'revoked' AND revoked_at IS NOT NULL) - OR - (state != 'revoked' AND revoked_at IS NULL) - ) + FOREIGN KEY(workspace_id) REFERENCES workspaces(workspace_id) ON DELETE RESTRICT ); -CREATE TABLE workspace_runtime_verifications ( +-- Historical authentication metadata only, never current admission authority. +CREATE TABLE workspace_runtime_binding_verification_archive ( workspace_id TEXT NOT NULL, runtime_id TEXT NOT NULL, - binding_id TEXT NOT NULL CHECK(length(binding_id) > 0), + binding_id TEXT NOT NULL, + state TEXT NOT NULL, + workspace_key_id TEXT, + workspace_public_key_fingerprint TEXT, + workspace_trust_id TEXT, + updated_at TEXT NOT NULL, + PRIMARY KEY (workspace_id, runtime_id, binding_id), + FOREIGN KEY(workspace_id) REFERENCES workspaces(workspace_id) ON DELETE CASCADE +); +CREATE TABLE workspace_runtime_verification_archive ( + workspace_id TEXT NOT NULL, + runtime_id TEXT NOT NULL, + binding_id TEXT NOT NULL, workspace_key_id TEXT NOT NULL, - workspace_public_key_fingerprint TEXT NOT NULL CHECK(length(workspace_public_key_fingerprint) > 0), - workspace_trust_id TEXT NOT NULL CHECK(length(workspace_trust_id) > 0), + workspace_public_key_fingerprint TEXT NOT NULL, + workspace_trust_id TEXT NOT NULL, runtime_public_key_fingerprint TEXT NOT NULL, challenge_id TEXT NOT NULL, - state TEXT NOT NULL CHECK(state IN ('pending', 'verified', 'failed')), + state TEXT NOT NULL, last_outcome TEXT NOT NULL, verified_at TEXT, checked_at TEXT NOT NULL, - PRIMARY KEY(workspace_id, runtime_id), - FOREIGN KEY(workspace_id, runtime_id) - REFERENCES workspace_runtime_bindings(workspace_id, runtime_id) ON DELETE CASCADE, - CHECK((state = 'verified' AND verified_at IS NOT NULL) - OR (state != 'verified' AND verified_at IS NULL)) + PRIMARY KEY (workspace_id, runtime_id), + FOREIGN KEY(workspace_id) REFERENCES workspaces(workspace_id) ON DELETE CASCADE ); -CREATE INDEX workspace_runtime_verifications_state_idx - ON workspace_runtime_verifications(workspace_id, state, checked_at DESC); CREATE TABLE workspace_runtime_binding_audit ( workspace_id TEXT NOT NULL, runtime_id TEXT NOT NULL, diff --git a/crates/workspace-server/src/legacy_evidence_column_tests.rs b/crates/workspace-server/src/legacy_evidence_column_tests.rs index 0aeefd5ef..a59f2a0ca 100644 --- a/crates/workspace-server/src/legacy_evidence_column_tests.rs +++ b/crates/workspace-server/src/legacy_evidence_column_tests.rs @@ -48,7 +48,7 @@ fn evidence_column_cutover_preserves_all_values_and_current_schema_names() { let upgraded = SqliteWorkspaceStore::open(&path).unwrap(); upgraded .with_conn(|conn| { - assert_eq!(current_schema_version(conn)?, 89); + assert_eq!(current_schema_version(conn)?, LATEST_SCHEMA_VERSION); assert_eq!(evidence_rows(conn, "artifacts"), artifacts); assert_eq!( evidence_rows(conn, "repository_secret_legacy_receipts"), diff --git a/crates/workspace-server/src/main.rs b/crates/workspace-server/src/main.rs index 8858a8cec..f68f9f639 100644 --- a/crates/workspace-server/src/main.rs +++ b/crates/workspace-server/src/main.rs @@ -10,7 +10,7 @@ use yoi_workspace_server::hosts::{ }; use yoi_workspace_server::store::{ AccountRecord, ApiTokenRecord, SqliteWorkspaceStore, UserRecord, - WorkspaceRuntimeAuthenticationMode, WorkspaceRuntimeBinding, WorkspaceRuntimeBindingState, + WorkspaceRuntimeAuthenticationMode, WorkspaceRuntimeBinding, }; use yoi_workspace_server::{ ControlPlaneStore, ResolvedWorkspaceBackendConfig, ServerConfig, ServerHostConfigFile, @@ -456,7 +456,7 @@ fn append_workspace_runtime_sources( .into_iter() .filter(|binding| { binding.runtime_id != yoi_workspace_server::hosts::EMBEDDED_RUNTIME_ID - && binding.state == WorkspaceRuntimeBindingState::Verified + && binding.revoked_at.is_none() }) .collect::>() }) @@ -787,11 +787,7 @@ mod tests { public_key: "unused".to_owned(), public_key_fingerprint: "unused".to_owned(), binding_id: "binding-test".to_owned(), - state: WorkspaceRuntimeBindingState::Verified, authentication_mode: WorkspaceRuntimeAuthenticationMode::LegacyServerIssuer, - workspace_key_id: None, - workspace_public_key_fingerprint: None, - workspace_trust_id: None, created_at: "2026-09-01T00:00:00Z".to_owned(), updated_at: "2026-09-01T00:00:00Z".to_owned(), revoked_at: None, @@ -815,11 +811,7 @@ mod tests { public_key: "unused".to_owned(), public_key_fingerprint: "unused".to_owned(), binding_id: "binding-test".to_owned(), - state: WorkspaceRuntimeBindingState::Verified, authentication_mode: WorkspaceRuntimeAuthenticationMode::WorkspaceIdentity, - workspace_key_id: Some("WK-test".to_owned()), - workspace_public_key_fingerprint: Some("sha256:workspace-key".to_owned()), - workspace_trust_id: Some("trust-test".to_owned()), created_at: "2026-09-01T00:00:00Z".to_owned(), updated_at: "2026-09-01T00:00:00Z".to_owned(), revoked_at: None, diff --git a/crates/workspace-server/src/schema_equivalence_tests.rs b/crates/workspace-server/src/schema_equivalence_tests.rs index 5e9618f19..3d98af689 100644 --- a/crates/workspace-server/src/schema_equivalence_tests.rs +++ b/crates/workspace-server/src/schema_equivalence_tests.rs @@ -53,7 +53,8 @@ fn fresh_and_upgraded_state_contract_schemas_enforce_the_same_constraints() { "flow_source_contents", "repositories", "workspace_runtime_bindings", - "workspace_runtime_verifications", + "workspace_runtime_binding_verification_archive", + "workspace_runtime_verification_archive", "workspace_runtime_binding_audit", "workspace_signing_identities", "workspace_signing_identity_provisioning_operations", diff --git a/crates/workspace-server/src/server.rs b/crates/workspace-server/src/server.rs index 339facd0e..f9f417722 100644 --- a/crates/workspace-server/src/server.rs +++ b/crates/workspace-server/src/server.rs @@ -64,9 +64,9 @@ use server_api::{ RepositorySshConnectionTrustState, RepositorySshCredential, RepositorySshHostKeyCandidate, RepositorySshHostTrust, RepositorySshPublicKey, RequestActor, RotateRepositorySshCredentialRequest, RuntimeCleanupExecutionResponse, - RuntimeCleanupExecutionResult, RuntimeCleanupPlanResponse, RuntimeConnectionDisplayState, - RuntimeConnectionTestFailureKind, RuntimeConnectionTestResponse, RuntimeConnectionTestStatus, - RuntimeManagementSummary, RuntimeRemovalOperationResponse, + RuntimeCleanupExecutionResult, RuntimeCleanupPlanResponse, RuntimeConnectionTestFailureKind, + RuntimeConnectionTestResponse, RuntimeConnectionTestStatus, RuntimeManagementSummary, + RuntimeRemovalOperationResponse, RuntimeRemovalOperationState as ApiRuntimeRemovalOperationState, RuntimeTrustAuditAction, RuntimeTrustAuditEntry, RuntimeTrustKeyRevealResponse, RuntimeTrustKeyState, RuntimeTrustKeyStatus, UpdateRemoteRuntimeRequest, UpdateWorkspaceMetadataRequest, @@ -83,11 +83,10 @@ use server_api::{ WorkspaceDeletionPreflightResponse, WorkspaceDeletionRequest, WorkspaceDeletionState, WorkspaceExtensionPointState, WorkspaceExtensionPoints, WorkspaceMetadataMutationResponse, WorkspaceMetadataSettingsResponse, WorkspacePermissionSummary, WorkspacePublicIdentityBundle, - WorkspaceRepositoryRecord, WorkspaceResponse, WorkspaceRuntimeBindingState, - WorkspaceRuntimeBindingSummary, WorkspaceRuntimeDetail, WorkspaceRuntimeResource, - WorkspaceSigningIdentityPublic, WorkspaceSigningIdentityResponse, - WorkspaceSigningIdentityState, WorkspaceSummary, WorkspaceWorkerDiscoveryItem, - WorkspaceWorkerDiscoveryPage, WorkspaceWorkerSubject, + WorkspaceRepositoryRecord, WorkspaceResponse, WorkspaceRuntimeBindingSummary, + WorkspaceRuntimeDetail, WorkspaceRuntimeResource, WorkspaceSigningIdentityPublic, + WorkspaceSigningIdentityResponse, WorkspaceSigningIdentityState, WorkspaceSummary, + WorkspaceWorkerDiscoveryItem, WorkspaceWorkerDiscoveryPage, WorkspaceWorkerSubject, }; use server_api::{ AttachmentUploadCancelResponse, AttachmentUploadGrantResponse, RestoreTicketAssignmentQuery, @@ -141,12 +140,6 @@ use worker_runtime::http_server::{ use worker_runtime::resource::BackendResourceError; use worker_runtime::resource::BackendResourceFetchRequest; use worker_runtime::worker_backend::{ProfileRuntimeWorkerFactory, WorkerRuntimeExecutionBackend}; -use worker_runtime::workspace_issuer::{ - WORKSPACE_VERIFICATION_ACK_PATH, WORKSPACE_VERIFICATION_CHALLENGE_PATH, - WORKSPACE_VERIFICATION_OPERATION, WorkspaceCapabilityClaims, - WorkspaceRuntimeVerificationAcknowledgement, WorkspaceRuntimeVerificationChallenge, - verify_runtime_verification_response, workspace_request_body_digest, -}; use crate::auth::{ SessionCookiePolicy, auth_error, is_expired, mint_secret, new_id, new_user_code, @@ -219,7 +212,6 @@ use crate::store::{ WorkerControlGrantRecord, WorkerRegistryRecord, WorkerWorkdirLinkRecord, WorkspaceRecord, WorkspaceResourceKind, WorkspaceRuntimeAuthenticationMode as StoredRuntimeAuthenticationMode, WorkspaceRuntimeBinding, WorkspaceRuntimeBindingAuditRecord, WorkspaceRuntimeBindingMutation, - WorkspaceRuntimeBindingState as StoredRuntimeBindingState, }; use crate::ticket_item_checker::{self, TicketItemCheckBodyReplacement, TicketItemCheckEdit}; use crate::workdir_removal::{ @@ -3311,13 +3303,7 @@ impl WorkspaceApi { public_key: embedded_identity.public_key.clone(), public_key_fingerprint: String::new(), binding_id: "binding-test".to_string(), - state: StoredRuntimeBindingState::Verified, authentication_mode: StoredRuntimeAuthenticationMode::LegacyServerIssuer, - workspace_key_id: None, - - workspace_public_key_fingerprint: None, - - workspace_trust_id: None, created_at: config.workspace_created_at.clone(), updated_at: config.workspace_created_at.clone(), revoked_at: None, @@ -3361,7 +3347,7 @@ impl WorkspaceApi { .await? .into_iter() .filter(|binding| binding.runtime_id != EMBEDDED_RUNTIME_ID) - .filter(|binding| binding.state != StoredRuntimeBindingState::Revoked) + .filter(|binding| binding.revoked_at.is_none()) .map(|binding| { ( (binding.workspace_id.clone(), binding.runtime_id.clone()), @@ -3416,10 +3402,9 @@ impl WorkspaceApi { binding.authentication_mode == crate::store::WorkspaceRuntimeAuthenticationMode::WorkspaceIdentity }) - .filter(|binding| binding.state != StoredRuntimeBindingState::Revoked) + .filter(|binding| binding.revoked_at.is_none()) { - let activate = binding.state == StoredRuntimeBindingState::Verified; - api.register_workspace_runtime_binding(binding, activate)?; + api.register_workspace_runtime_binding(binding)?; } Ok(api) } @@ -3498,13 +3483,13 @@ impl WorkspaceApi { binding.authentication_mode == StoredRuntimeAuthenticationMode::WorkspaceIdentity && binding.revoked_at.is_none() }) { - let verified_binding = current_binding - .filter(|binding| binding.state == StoredRuntimeBindingState::Verified); + let binding = + current_binding.expect("Workspace identity binding was checked above"); remote_config.workspace_authorization = Some(WorkspaceRuntimeAuthorization::new( store.clone(), signing_identities.clone(), backend_url.clone(), - verified_binding, + binding, )); } let remote_runtime = RemoteWorkerRuntime::new( @@ -4746,11 +4731,7 @@ impl WorkspaceApi { ); } - fn register_workspace_runtime_binding( - &self, - binding: WorkspaceRuntimeBinding, - activate: bool, - ) -> Result<()> { + fn register_workspace_runtime_binding(&self, binding: WorkspaceRuntimeBinding) -> Result<()> { let backend_url = self .config .backend_base_url @@ -4762,7 +4743,7 @@ impl WorkspaceApi { self.store.clone(), self.signing_identities.clone(), backend_url.clone(), - activate.then_some(binding), + binding, )); let remote_runtime = RemoteWorkerRuntime::new( remote_config.clone(), @@ -25588,7 +25569,6 @@ async fn scoped_update_remote_runtime( &Utc::now().to_rfc3339(), ) .await?; - let activate_runtime = binding.state == StoredRuntimeBindingState::Verified; api.runtime_binding_expectations .write() .unwrap_or_else(std::sync::PoisonError::into_inner) @@ -25598,7 +25578,7 @@ async fn scoped_update_remote_runtime( ); api.runtime_subscription_broker .unregister_runtime(&path.runtime_id); - api.register_workspace_runtime_binding(binding, activate_runtime)?; + api.register_workspace_runtime_binding(binding)?; Ok(Json( workspace_runtime_detail(&api, &path.workspace_id, &path.runtime_id).await?, )) @@ -25664,7 +25644,7 @@ async fn scoped_test_runtime_connection( .get_workspace_runtime_binding(&path.workspace_id, &path.runtime_id) .await? .ok_or_else(|| Error::UnknownRuntime(path.runtime_id.clone()))?; - if binding.state == StoredRuntimeBindingState::Revoked || binding.revoked_at.is_some() { + if binding.revoked_at.is_some() { return Err(Error::RuntimeBindingConflict(format!( "Runtime `{}` binding is revoked", path.runtime_id @@ -25679,11 +25659,6 @@ async fn scoped_test_runtime_connection( }, display_name: Some(binding.display_name.clone()), endpoint: binding.base_url.clone(), - workspace_trust_id: binding.workspace_trust_id.clone().ok_or_else(|| { - Error::RuntimeBindingConflict( - "Runtime binding is missing its Workspace trust ID".to_string(), - ) - })?, expected_binding_id: Some(binding.binding_id.clone()), }) .await?; @@ -27027,32 +27002,23 @@ async fn create_remote_runtime( "an active Workspace signing identity is required before registering a Runtime", )); } - let existing = api - .store - .get_workspace_runtime_binding(&api.config.workspace_id, &runtime_id) - .await?; - if existing - .as_ref() - .is_some_and(|binding| binding.state == StoredRuntimeBindingState::Verified) - { - return Err(Error::RuntimeBindingConflict( - "a verified Runtime binding must be revoked before it can be replaced as configured" - .to_string(), - ) - .into()); - } + // Protect observed active Workers without removing the current connection. + // This scan is not proof of idleness: an unavailable/auth-failing endpoint + // must remain repairable by an owner using binding-ID CAS below. Replacement + // neither stops nor deletes Workers; Runtime removal has separate preflight. + // Rejected key/CAS mutations must leave the current registration usable. match api .runtime - .unregister_if_idle(&runtime_id, api.config.max_records.min(200)) - .map_err(|err| err.into_error())? + .list_workers_for_runtime(&runtime_id, api.config.max_records.min(200)) { - RuntimeRegistryUnregisterResult::Removed | RuntimeRegistryUnregisterResult::NotFound => {} - RuntimeRegistryUnregisterResult::BlockedByWorkers { worker_count, .. } => { + Ok(workers) if !workers.items.is_empty() => { return Err(Error::RuntimeBindingConflict(format!( - "Runtime `{runtime_id}` still has {worker_count} active worker(s) and cannot become a configured-only binding" - )) - .into()); + "Runtime `{runtime_id}` still has {} active worker(s) and cannot replace its connection", + workers.items.len() + )).into()); } + Ok(_) | Err(RuntimeRegistryError::UnknownRuntime(_)) => {} + Err(error) => return Err(error.into_error().into()), } let now = Utc::now().to_rfc3339(); let record = WorkspaceRuntimeBinding { @@ -27069,13 +27035,7 @@ async fn create_remote_runtime( public_key: request.public_bundle.public_key.trim().to_string(), public_key_fingerprint: String::new(), binding_id: Uuid::now_v7().to_string(), - state: StoredRuntimeBindingState::Configured, authentication_mode: StoredRuntimeAuthenticationMode::WorkspaceIdentity, - workspace_key_id: Some(identity.key_id.clone()), - - workspace_public_key_fingerprint: identity.public_key_fingerprint.clone(), - - workspace_trust_id: Some(request.workspace_trust_id.clone()), created_at: now.clone(), updated_at: now, revoked_at: None, @@ -27097,7 +27057,7 @@ async fn create_remote_runtime( ); api.runtime_subscription_broker .unregister_runtime(&runtime_id); - api.register_workspace_runtime_binding(binding, false)?; + api.register_workspace_runtime_binding(binding)?; let resource = workspace_runtime_resources_response(&api, &api.config.workspace_id) .await? .items @@ -27266,7 +27226,7 @@ async fn execute_runtime_removal( "runtime_removal_commit_failed", ) .await; - let restore_result = api.register_workspace_runtime_binding(binding, true); + let restore_result = api.register_workspace_runtime_binding(binding); if let Err(restore_error) = restore_result { tracing::warn!( workspace_id = %operation.workspace_id, @@ -27418,219 +27378,6 @@ async fn remove_remote_runtime( Ok(Json(runtime_removal_response(operation))) } -async fn perform_workspace_runtime_verification( - api: &WorkspaceApi, - runtime: Arc, - binding: &WorkspaceRuntimeBinding, -) -> std::result::Result { - if binding.authentication_mode != StoredRuntimeAuthenticationMode::WorkspaceIdentity { - return Ok(binding.clone()); - } - if binding.state == crate::store::WorkspaceRuntimeBindingState::Revoked - || binding.revoked_at.is_some() - { - return Err("Runtime binding is revoked".to_string()); - } - let backend_url = api.config.backend_base_url.as_deref().ok_or_else(|| { - "Workspace identity verification requires configured backend_base_url".to_string() - })?; - let identity = api - .signing_identities - .get_validated(&binding.workspace_id) - .map_err(|error| error.to_string())?; - if identity.state != "active" { - return Err("Workspace signing identity is not active".to_string()); - } - let workspace_key_id = binding - .workspace_key_id - .as_deref() - .ok_or_else(|| "Runtime binding is missing the Workspace key identity".to_string())?; - let workspace_trust_id = binding - .workspace_trust_id - .as_deref() - .ok_or_else(|| "Runtime binding is missing the Workspace trust ID".to_string())?; - let workspace_public_key_fingerprint = binding - .workspace_public_key_fingerprint - .as_deref() - .ok_or_else(|| { - "Runtime binding is missing the Workspace public key fingerprint".to_string() - })?; - if identity.key_id != workspace_key_id - || identity.public_key_fingerprint.as_deref() != Some(workspace_public_key_fingerprint) - { - return Err( - "Runtime binding no longer matches the active Workspace or Runtime identity" - .to_string(), - ); - } - - let now = Utc::now(); - let expires_at = (now + Duration::seconds(60)).timestamp(); - let challenge = WorkspaceRuntimeVerificationChallenge { - challenge_id: Uuid::now_v7().to_string(), - workspace_id: binding.workspace_id.clone(), - runtime_id: binding.runtime_id.clone(), - binding_id: binding.binding_id.clone(), - workspace_key_id: workspace_key_id.to_string(), - workspace_public_key_fingerprint: workspace_public_key_fingerprint.to_string(), - workspace_trust_id: workspace_trust_id.to_string(), - runtime_public_key_fingerprint: binding.public_key_fingerprint.clone(), - workspace_nonce: Uuid::now_v7().to_string(), - expires_at, - }; - let checked_at = now.to_rfc3339_opts(SecondsFormat::Millis, true); - let pending = crate::store::WorkspaceRuntimeVerificationEvidence { - workspace_id: binding.workspace_id.clone(), - runtime_id: binding.runtime_id.clone(), - binding_id: binding.binding_id.clone(), - workspace_key_id: workspace_key_id.to_string(), - workspace_public_key_fingerprint: workspace_public_key_fingerprint.to_string(), - workspace_trust_id: workspace_trust_id.to_string(), - runtime_public_key_fingerprint: binding.public_key_fingerprint.clone(), - challenge_id: challenge.challenge_id.clone(), - state: "pending".to_string(), - last_outcome: "challenge_issued".to_string(), - verified_at: None, - checked_at: checked_at.clone(), - }; - api.store - .record_workspace_runtime_verification_attempt(&pending) - .await - .map_err(|error| error.to_string())?; - - let result = async { - let challenge_body = serde_json::to_vec(&challenge).map_err(|error| error.to_string())?; - let challenge_claims = WorkspaceCapabilityClaims { - issuer: backend_url.to_string(), - issuer_workspace_id: binding.workspace_id.clone(), - issuer_key_id: identity.key_id.clone(), - issuer_public_key_fingerprint: workspace_public_key_fingerprint.to_string(), - trust_id: workspace_trust_id.to_string(), - binding_id: binding.binding_id.clone(), - runtime_id: binding.runtime_id.clone(), - worker_id: None, - operation: WORKSPACE_VERIFICATION_OPERATION.to_string(), - method: "POST".to_string(), - path_and_query: WORKSPACE_VERIFICATION_CHALLENGE_PATH.to_string(), - body_digest: workspace_request_body_digest(&challenge_body), - iat: now.timestamp(), - exp: expires_at, - jti: Uuid::now_v7().to_string(), - }; - let challenge_token = api - .signing_identities - .issue_workspace_capability(&binding.workspace_id, &challenge_claims) - .map_err(|error| error.to_string())?; - let challenge_runtime = runtime.clone(); - let challenge_runtime_id = binding.runtime_id.clone(); - let challenge_request = challenge.clone(); - let response = tokio::task::spawn_blocking(move || { - challenge_runtime.send_workspace_verification_challenge( - &challenge_runtime_id, - &challenge_request, - &challenge_token, - ) - }) - .await - .map_err(|_| "Runtime verification challenge task failed".to_string())? - .map_err(|failure| failure.diagnostic.message)?; - verify_runtime_verification_response( - &response, - &challenge, - &binding.public_key, - Utc::now().timestamp(), - ) - .map_err(|error| error.to_string())?; - - let response_bytes = serde_json::to_vec(&response).map_err(|error| error.to_string())?; - let acknowledgement = WorkspaceRuntimeVerificationAcknowledgement { - challenge_id: response.challenge_id.clone(), - workspace_id: response.workspace_id.clone(), - runtime_id: response.runtime_id.clone(), - binding_id: response.binding_id.clone(), - workspace_key_id: response.workspace_key_id.clone(), - workspace_public_key_fingerprint: response.workspace_public_key_fingerprint.clone(), - workspace_trust_id: response.workspace_trust_id.clone(), - runtime_public_key_fingerprint: response.runtime_public_key_fingerprint.clone(), - workspace_nonce: response.workspace_nonce.clone(), - runtime_nonce: response.runtime_nonce.clone(), - response_digest: workspace_request_body_digest(&response_bytes), - response: response.clone(), - expires_at: response.expires_at, - }; - let acknowledgement_body = - serde_json::to_vec(&acknowledgement).map_err(|error| error.to_string())?; - let acknowledgement_claims = WorkspaceCapabilityClaims { - issuer: backend_url.to_string(), - issuer_workspace_id: binding.workspace_id.clone(), - issuer_key_id: identity.key_id.clone(), - issuer_public_key_fingerprint: workspace_public_key_fingerprint.to_string(), - trust_id: workspace_trust_id.to_string(), - binding_id: binding.binding_id.clone(), - runtime_id: binding.runtime_id.clone(), - worker_id: None, - operation: WORKSPACE_VERIFICATION_OPERATION.to_string(), - method: "POST".to_string(), - path_and_query: WORKSPACE_VERIFICATION_ACK_PATH.to_string(), - body_digest: workspace_request_body_digest(&acknowledgement_body), - iat: Utc::now().timestamp(), - exp: expires_at, - jti: Uuid::now_v7().to_string(), - }; - let acknowledgement_token = api - .signing_identities - .issue_workspace_capability(&binding.workspace_id, &acknowledgement_claims) - .map_err(|error| error.to_string())?; - let acknowledgement_runtime = runtime; - let acknowledgement_runtime_id = binding.runtime_id.clone(); - let acknowledgement_request = acknowledgement.clone(); - let receipt = tokio::task::spawn_blocking(move || { - acknowledgement_runtime.send_workspace_verification_acknowledgement( - &acknowledgement_runtime_id, - &acknowledgement_request, - &acknowledgement_token, - ) - }) - .await - .map_err(|_| "Runtime verification acknowledgement task failed".to_string())? - .map_err(|failure| failure.diagnostic.message)?; - if receipt.challenge_id != challenge.challenge_id - || receipt.workspace_id != binding.workspace_id - || receipt.runtime_id != binding.runtime_id - || receipt.binding_id != binding.binding_id - { - return Err("Runtime verification acknowledgement receipt mismatched".to_string()); - } - - let verified_at = Utc::now().to_rfc3339_opts(SecondsFormat::Millis, true); - let verified = crate::store::WorkspaceRuntimeVerificationEvidence { - state: "verified".to_string(), - last_outcome: "verified".to_string(), - verified_at: Some(verified_at.clone()), - checked_at: verified_at, - ..pending.clone() - }; - api.store - .complete_workspace_runtime_verification(&verified) - .await - .map_err(|error| error.to_string()) - } - .await; - if result.is_err() { - let checked_at = Utc::now().to_rfc3339_opts(SecondsFormat::Millis, true); - let _ = api - .store - .record_workspace_runtime_verification_outcome_if_current( - &pending, - "failed", - "verification_failed", - &checked_at, - ) - .await; - } - result -} - async fn test_runtime_connection( State(api): State, AxumPath(runtime_id): AxumPath, @@ -27650,44 +27397,6 @@ async fn test_runtime_connection( .filter(|binding| binding.revoked_at.is_none()) .ok_or_else(|| Error::UnknownRuntime(runtime_id.clone()))?; - if binding.authentication_mode == StoredRuntimeAuthenticationMode::WorkspaceIdentity { - match perform_workspace_runtime_verification(&api, api.runtime.clone(), &binding).await { - Ok(verified_binding) => { - api.register_workspace_runtime_binding(verified_binding.clone(), true)?; - api.runtime_binding_expectations - .write() - .unwrap_or_else(std::sync::PoisonError::into_inner) - .insert( - ( - verified_binding.workspace_id.clone(), - verified_binding.runtime_id.clone(), - ), - verified_binding.clone(), - ); - api.runtime - .activate_workspace_authorization(&runtime_id, verified_binding.clone()) - .map_err(|error| Error::Store(format!("{error:?}")))?; - } - Err(message) => { - let mut result = runtime_connection_test_failure( - api.workspace_id(), - &runtime_id, - Utc::now().to_rfc3339(), - RuntimeConnectionTestFailureKind::Authentication, - None, - RuntimeDiagnostic::new( - "runtime_workspace_verification_failed", - "error", - message, - ), - ); - result.binding_id = binding.binding_id.clone(); - result.connection_state = RuntimeConnectionDisplayState::Unavailable; - return Ok(Json(result)); - } - } - } - let checked_at = Utc::now().to_rfc3339(); let runtime = api.runtime.clone(); let ping_runtime_id = runtime_id.clone(); @@ -27699,44 +27408,19 @@ async fn test_runtime_connection( message: "Runtime connection test could not be completed".to_string(), })?; - if ping.is_err() - && binding.authentication_mode == StoredRuntimeAuthenticationMode::WorkspaceIdentity - && let Some(evidence) = api - .store - .get_workspace_runtime_verification(api.workspace_id(), &runtime_id) - .await? - { - let checked_at = Utc::now().to_rfc3339(); - api.store - .record_workspace_runtime_verification_outcome_if_current( - &evidence, - "failed", - "connectivity_failed", - &checked_at, - ) - .await?; + // The ping runs outside the store lock. A replaced/revoked connection must + // not be reported as an observation of the earlier binding. + if !api.store.workspace_runtime_binding_matches(&binding)? { + return Err(Error::RuntimeBindingConflict( + "Runtime connection changed while the connection test was running".to_string(), + ) + .into()); } - let current_binding = api - .store - .get_workspace_runtime_binding(api.workspace_id(), &runtime_id) - .await? - .ok_or_else(|| Error::UnknownRuntime(runtime_id.clone()))?; - let verification = api - .store - .get_workspace_runtime_verification(api.workspace_id(), &runtime_id) - .await?; - let summary = runtime_binding_summary(¤t_binding, verification.as_ref()); let mut result = runtime_connection_test_response(api.workspace_id(), &runtime_id, checked_at, ping); - result.binding_id = current_binding.binding_id.clone(); - result.connection_state = if result.status == RuntimeConnectionTestStatus::Compatible { - summary.connection_state - } else if summary.connection_state == RuntimeConnectionDisplayState::Revoked { - RuntimeConnectionDisplayState::Revoked - } else { - RuntimeConnectionDisplayState::Unavailable - }; - result.verification = summary.verification; + // This result describes this attempt, not permission for a future request. + result.binding_id = binding.binding_id; + Ok(Json(result)) } @@ -30180,16 +29864,6 @@ async fn workspace_runtime_resources_response( .store .list_workspace_runtime_bindings(workspace_id, true) .await?; - let mut verifications = HashMap::new(); - for binding in &bindings { - if let Some(verification) = api - .store - .get_workspace_runtime_verification(workspace_id, &binding.runtime_id) - .await? - { - verifications.insert(binding.runtime_id.clone(), verification); - } - } let mut items = runtimes .items .into_iter() @@ -30199,7 +29873,7 @@ async fn workspace_runtime_resources_response( .find(|binding| binding.runtime_id == runtime.runtime_id); let built_in = runtime.runtime_id == EMBEDDED_WORKER_RUNTIME_ID; let mut runtime: server_api::RuntimeSummary = runtime.into(); - if binding.is_some_and(|binding| binding.state != StoredRuntimeBindingState::Verified) { + if binding.is_some_and(|binding| binding.revoked_at.is_some()) { runtime.worker_creation_available = false; } WorkspaceRuntimeResource { @@ -30211,9 +29885,7 @@ async fn workspace_runtime_resources_response( endpoint_configured: binding .is_some_and(|binding| !binding.base_url.trim().is_empty()), token_ref_configured: false, - binding: binding.map(|binding| { - runtime_binding_summary(binding, verifications.get(&binding.runtime_id)) - }), + binding: binding.map(|binding| runtime_binding_summary(binding)), }, http_status: StatusCode::OK.as_u16(), } @@ -30260,10 +29932,7 @@ async fn workspace_runtime_resources_response( removable: true, endpoint_configured: !binding.base_url.trim().is_empty(), token_ref_configured: false, - binding: Some(runtime_binding_summary( - binding, - verifications.get(&binding.runtime_id), - )), + binding: Some(runtime_binding_summary(binding)), }, http_status: StatusCode::OK.as_u16(), }); @@ -30278,75 +29947,13 @@ async fn workspace_runtime_resources_response( }) } -fn runtime_binding_summary( - binding: &WorkspaceRuntimeBinding, - verification: Option<&crate::store::WorkspaceRuntimeVerificationEvidence>, -) -> WorkspaceRuntimeBindingSummary { - let current_verification = verification.filter(|verification| { - verification.binding_id == binding.binding_id - && verification.runtime_public_key_fingerprint == binding.public_key_fingerprint - && verification.workspace_key_id - == binding.workspace_key_id.as_deref().unwrap_or_default() - && Some(verification.workspace_public_key_fingerprint.as_str()) - == binding.workspace_public_key_fingerprint.as_deref() - && Some(verification.workspace_trust_id.as_str()) - == binding.workspace_trust_id.as_deref() - }); - let valid_verification = current_verification.filter(|verification| { - verification.state == "verified" && verification.last_outcome == "verified" - }); - let connection_state = match binding.state { - StoredRuntimeBindingState::Revoked => RuntimeConnectionDisplayState::Revoked, - _ if current_verification - .is_some_and(|verification| verification.last_outcome != "verified") => - { - RuntimeConnectionDisplayState::Unavailable - } - StoredRuntimeBindingState::Verified - if binding.authentication_mode - == StoredRuntimeAuthenticationMode::LegacyServerIssuer - || valid_verification.is_some() => - { - RuntimeConnectionDisplayState::Verified - } - _ => RuntimeConnectionDisplayState::Configured, - }; +fn runtime_binding_summary(binding: &WorkspaceRuntimeBinding) -> WorkspaceRuntimeBindingSummary { WorkspaceRuntimeBindingSummary { - state: match binding.state { - StoredRuntimeBindingState::Configured => WorkspaceRuntimeBindingState::Configured, - StoredRuntimeBindingState::Verified => WorkspaceRuntimeBindingState::Verified, - StoredRuntimeBindingState::Revoked => WorkspaceRuntimeBindingState::Revoked, - }, - connection_state, binding_id: binding.binding_id.clone(), - workspace_key_id: binding.workspace_key_id.clone(), - workspace_trust_id: binding.workspace_trust_id.clone(), - verification: current_verification.and_then(runtime_verification_summary), + revoked_at: binding.revoked_at.clone(), } } -fn runtime_verification_summary( - verification: &crate::store::WorkspaceRuntimeVerificationEvidence, -) -> Option { - let last_outcome = match verification.last_outcome.as_str() { - "verified" => server_api::RuntimeVerificationOutcome::Verified, - "challenge_issued" => server_api::RuntimeVerificationOutcome::ChallengeIssued, - "verification_failed" => server_api::RuntimeVerificationOutcome::VerificationFailed, - "connectivity_failed" => server_api::RuntimeVerificationOutcome::ConnectivityFailed, - _ => return None, - }; - Some(server_api::RuntimeVerificationEvidenceSummary { - verified_at: verification.verified_at.clone(), - last_checked_at: verification.checked_at.clone(), - last_outcome, - binding_id: verification.binding_id.clone(), - workspace_key_id: verification.workspace_key_id.clone(), - workspace_public_key_fingerprint: verification.workspace_public_key_fingerprint.clone(), - workspace_trust_id: verification.workspace_trust_id.clone(), - runtime_public_key_fingerprint: verification.runtime_public_key_fingerprint.clone(), - }) -} - async fn workspace_runtime_detail( api: &WorkspaceApi, workspace_id: &str, @@ -30388,7 +29995,7 @@ async fn workspace_runtime_detail( removable: false, endpoint_configured: !binding.base_url.trim().is_empty(), token_ref_configured: false, - binding: Some(runtime_binding_summary(&binding, None)), + binding: Some(runtime_binding_summary(binding)), }, http_status: StatusCode::OK.as_u16(), }); @@ -30483,18 +30090,6 @@ async fn validate_runtime_connection_request( "Runtime public bundle must contain a public key", )); } - if request.workspace_trust_id.trim().is_empty() - || request.workspace_trust_id.len() > 256 - || request - .workspace_trust_id - .chars() - .any(|ch| ch.is_control() || ch.is_whitespace()) - { - return Err(settings_bad_request( - "invalid_workspace_trust_id", - "workspace_trust_id must be a bounded Runtime-issued trust identity", - )); - } if request .expected_binding_id .as_deref() @@ -30666,8 +30261,7 @@ fn runtime_connection_test_response( workspace_id: workspace_id.to_string(), runtime_id: runtime_id.to_string(), binding_id: String::new(), - connection_state: RuntimeConnectionDisplayState::Configured, - verification: None, + checked_at, status: RuntimeConnectionTestStatus::Compatible, failure_kind: None, @@ -30718,8 +30312,7 @@ fn runtime_connection_test_failure( workspace_id: workspace_id.to_string(), runtime_id: runtime_id.to_string(), binding_id: String::new(), - connection_state: RuntimeConnectionDisplayState::Unavailable, - verification: None, + checked_at, status: RuntimeConnectionTestStatus::Failed, failure_kind: Some(failure_kind), @@ -32822,10 +32415,11 @@ impl ApiError { let message = api_error_response_message(&self.error); server_api::RepositoryApiError::new( status.as_u16(), - if matches!(self.error, Error::RestoreObservationConflict) { - "restore_observation_conflict" - } else { - status.canonical_reason().unwrap_or("error") + match self.error { + Error::RestoreObservationConflict => "restore_observation_conflict", + Error::RuntimeBindingIdConflict { .. } => "stale_binding", + Error::RuntimeBindingFingerprintConflict { .. } => "fingerprint_in_use", + _ => status.canonical_reason().unwrap_or("error"), }, message, working_directory_diagnostics(self.diagnostics), @@ -36522,13 +36116,7 @@ mod tests { public_key: identity.public_key.clone(), public_key_fingerprint: String::new(), binding_id: "binding-test".to_string(), - state: StoredRuntimeBindingState::Verified, authentication_mode: StoredRuntimeAuthenticationMode::WorkspaceIdentity, - workspace_key_id: Some("WK-test".to_owned()), - - workspace_public_key_fingerprint: Some("sha256:workspace-test".to_string()), - - workspace_trust_id: Some("trust-test".to_string()), created_at: "2026-01-01T00:00:00Z".to_owned(), updated_at: "2026-01-01T00:00:00Z".to_owned(), revoked_at: None, @@ -42602,13 +42190,12 @@ mod tests { } #[tokio::test] - async fn verified_workspace_runtime_binding_is_restored_into_live_registry() { + async fn runtime_connection_is_restored_without_prior_authentication() { let dir = tempfile::tempdir().unwrap(); let root = dir.path().to_path_buf(); let api = test_api(&root).await; let actor = test_owner_actor(); - let identity = api - .signing_identities + api.signing_identities .provision_existing(&api.config.workspace_id, &actor.account_id) .unwrap(); let runtime_identity = RuntimeIdentityMaterial::generate("restored-runtime").unwrap(); @@ -42622,13 +42209,7 @@ mod tests { public_key: runtime_identity.public_key, public_key_fingerprint: String::new(), binding_id: "binding-test".to_string(), - state: StoredRuntimeBindingState::Configured, authentication_mode: StoredRuntimeAuthenticationMode::WorkspaceIdentity, - workspace_key_id: Some(identity.key_id.clone()), - - workspace_public_key_fingerprint: identity.public_key_fingerprint.clone(), - - workspace_trust_id: Some("trust-test".to_string()), created_at: "1".to_string(), updated_at: "1".to_string(), revoked_at: None, @@ -42637,38 +42218,12 @@ mod tests { ) .await .unwrap(); - let configured = api - .store - .get_workspace_runtime_binding(&api.config.workspace_id, "restored-runtime") - .await - .unwrap() - .unwrap(); - let evidence = crate::store::WorkspaceRuntimeVerificationEvidence { - workspace_id: configured.workspace_id.clone(), - runtime_id: configured.runtime_id.clone(), - binding_id: configured.binding_id.clone(), - workspace_key_id: identity.key_id, - workspace_public_key_fingerprint: identity.public_key_fingerprint.clone().unwrap(), - workspace_trust_id: "trust-test".to_string(), - runtime_public_key_fingerprint: configured.public_key_fingerprint.clone(), - challenge_id: "restart-challenge".to_string(), - state: "verified".to_string(), - last_outcome: "verified".to_string(), - verified_at: Some("2".to_string()), - checked_at: "2".to_string(), - }; - api.store - .record_workspace_runtime_verification_attempt(&evidence) - .await - .unwrap(); - api.store - .complete_workspace_runtime_verification(&evidence) - .await - .unwrap(); drop(api); - tokio::time::sleep(std::time::Duration::from_millis(50)).await; - - let restored_config = test_server_config(&root); + // The assertion concerns persisted connection configuration, not async + // shutdown of the first embedded execution store. Use an independent + // execution store while reopening the same Workspace database. + let restored_config = test_server_config(&root) + .with_embedded_runtime_store_root(root.join("restored-embedded-runtime")); let restored_store = SqliteWorkspaceStore::open(restored_config.database_path.clone()).unwrap(); let restored = WorkspaceApi::new(restored_config, Arc::new(restored_store)) @@ -42681,7 +42236,7 @@ mod tests { .items .iter() .any(|runtime| runtime.runtime_id == "restored-runtime"), - "verified persisted Runtime binding must return to the live registry" + "persisted Runtime connection must return without a prior authentication result" ); assert!( restored @@ -42697,12 +42252,11 @@ mod tests { } #[tokio::test] - async fn remote_runtime_endpoint_update_preserves_public_key_but_requires_reverification() { + async fn remote_runtime_endpoint_update_preserves_key_and_replaces_connection_identity() { let dir = tempfile::tempdir().unwrap(); let api = test_api(dir.path()).await; let actor = test_owner_actor(); - let workspace_identity = api - .signing_identities + api.signing_identities .provision_existing(&api.config.workspace_id, &actor.account_id) .unwrap(); let runtime_identity = RuntimeIdentityMaterial::generate("verified-runtime").unwrap(); @@ -42716,13 +42270,7 @@ mod tests { public_key: runtime_identity.public_key.clone(), public_key_fingerprint: String::new(), binding_id: "binding-test".to_string(), - state: StoredRuntimeBindingState::Configured, authentication_mode: StoredRuntimeAuthenticationMode::WorkspaceIdentity, - workspace_key_id: Some(workspace_identity.key_id.clone()), - workspace_public_key_fingerprint: workspace_identity - .public_key_fingerprint - .clone(), - workspace_trust_id: Some("trust-test".to_string()), created_at: "1".to_string(), updated_at: "1".to_string(), revoked_at: None, @@ -42737,32 +42285,6 @@ mod tests { .await .unwrap() .unwrap(); - let evidence = crate::store::WorkspaceRuntimeVerificationEvidence { - workspace_id: configured.workspace_id.clone(), - runtime_id: configured.runtime_id.clone(), - binding_id: configured.binding_id.clone(), - workspace_key_id: workspace_identity.key_id, - workspace_public_key_fingerprint: workspace_identity - .public_key_fingerprint - .clone() - .unwrap(), - workspace_trust_id: "trust-test".to_string(), - runtime_public_key_fingerprint: configured.public_key_fingerprint.clone(), - challenge_id: "verified-metadata-update".to_string(), - state: "verified".to_string(), - last_outcome: "verified".to_string(), - verified_at: Some("2".to_string()), - checked_at: "2".to_string(), - }; - api.store - .record_workspace_runtime_verification_attempt(&evidence) - .await - .unwrap(); - api.store - .complete_workspace_runtime_verification(&evidence) - .await - .unwrap(); - let Json(updated) = scoped_update_remote_runtime( State(api.clone()), AxumPath(ScopedRuntimePath { @@ -42781,9 +42303,8 @@ mod tests { assert_eq!(updated.runtime.runtime.label, "New label"); assert_eq!(updated.endpoint.as_deref(), Some("https://8.8.4.4")); let binding = updated.runtime.management.binding.unwrap(); - assert_eq!(binding.state, WorkspaceRuntimeBindingState::Configured); + assert_eq!(binding.revoked_at, None); assert_ne!(binding.binding_id, configured.binding_id); - assert_eq!(binding.verification, None); assert_eq!( updated.trust_key.fingerprint.as_deref(), Some(configured.public_key_fingerprint.as_str()) @@ -42811,7 +42332,7 @@ mod tests { }, display_name: Some("New label".to_string()), endpoint: "https://8.8.4.4".to_string(), - workspace_trust_id: "trust-test".to_string(), + expected_binding_id: Some(configured.binding_id.clone()), }), ) @@ -42821,7 +42342,7 @@ mod tests { } #[tokio::test] - async fn remote_runtime_registration_is_workspace_scoped_binding_fenced_and_configured() { + async fn remote_runtime_registration_is_workspace_scoped_and_binding_fenced() { let dir = tempfile::tempdir().unwrap(); let api = test_api(dir.path()).await; let actor = test_owner_actor(); @@ -42849,7 +42370,7 @@ mod tests { }, display_name: Some("Configured Runtime".to_string()), endpoint: "https://8.8.8.8".to_string(), - workspace_trust_id: "trust-test".to_string(), + expected_binding_id: None, }; let mut non_owner = actor.clone(); @@ -42874,10 +42395,8 @@ mod tests { .unwrap(); assert_eq!(status, StatusCode::CREATED); let binding = created.management.binding.as_ref().unwrap(); - assert_eq!(binding.state, WorkspaceRuntimeBindingState::Configured); + assert_eq!(binding.revoked_at, None); assert!(!binding.binding_id.is_empty()); - assert!(binding.workspace_key_id.is_some()); - assert_eq!(binding.workspace_trust_id.as_deref(), Some("trust-test")); assert!(!created.runtime.worker_creation_available); assert!( api.runtime @@ -42885,7 +42404,7 @@ mod tests { .items .iter() .any(|runtime| runtime.runtime_id == "configured-runtime"), - "configured binding must remain registered so verification can reach it" + "registered connection permits actual authentication attempts" ); assert!( api.runtime_binding_expectations @@ -42897,22 +42416,6 @@ mod tests { )), "configured binding must remain fenced by its current binding ID" ); - let Json(configured_test) = scoped_test_runtime_connection( - State(api.clone()), - AxumPath(ScopedRuntimePath { - workspace_id: api.config.workspace_id.clone(), - runtime_id: "configured-runtime".to_string(), - }), - Extension(actor.clone()), - ) - .await - .unwrap(); - assert_eq!(configured_test.status, RuntimeConnectionTestStatus::Failed); - assert_eq!( - configured_test.connection_state, - RuntimeConnectionDisplayState::Unavailable - ); - let (status, Json(replayed)) = create_remote_runtime( State(api.clone()), Extension(actor.clone()), @@ -42936,6 +42439,30 @@ mod tests { .await .unwrap_err(); assert_eq!(error.into_response().status(), StatusCode::CONFLICT); + assert!( + api.runtime + .list_runtimes(100) + .items + .iter() + .any(|runtime| runtime.runtime_id == "configured-runtime"), + "rejected binding mutation must retain the live registration" + ); + let unchanged = api + .store + .get_workspace_runtime_binding(&api.config.workspace_id, "configured-runtime") + .await + .unwrap() + .unwrap(); + assert_eq!(unchanged.binding_id, binding.binding_id); + let conflict = request_json( + build_inner_router(api.clone()).layer(Extension(actor.clone())), + "POST", + &format!("/api/w/{TEST_WORKSPACE_ID}/runtimes"), + Some(serde_json::to_value(&mismatched).unwrap()), + StatusCode::CONFLICT, + ) + .await; + assert_eq!(conflict["error"], "stale_binding"); mismatched.expected_binding_id = Some(binding.binding_id.clone()); let (status, Json(replaced)) = @@ -42959,7 +42486,7 @@ mod tests { }, display_name: None, endpoint: "https://8.8.4.4".to_string(), - workspace_trust_id: "trust-test".to_string(), + expected_binding_id: Some("binding-missing".to_string()), }), ) @@ -43003,7 +42530,7 @@ mod tests { }, display_name: None, endpoint: "http://127.0.0.1:1".to_string(), - workspace_trust_id: "trust-first-acceptance".to_string(), + expected_binding_id: None, }; let (_, Json(created)) = create_remote_runtime( @@ -43030,7 +42557,6 @@ mod tests { let revoked_id = revoked.trust_key.binding_id.unwrap(); assert_ne!(revoked_id, original.binding_id); let mut reaccepted = request; - reaccepted.workspace_trust_id = "trust-second-acceptance".to_string(); reaccepted.expected_binding_id = Some(revoked_id); let (_, Json(created)) = create_remote_runtime( State(api.clone()), @@ -43041,14 +42567,6 @@ mod tests { .unwrap(); let current = created.management.binding.unwrap(); assert_ne!(current.binding_id, original.binding_id); - assert_eq!( - current.workspace_trust_id.as_deref(), - Some("trust-second-acceptance") - ); - assert_eq!( - current.connection_state, - RuntimeConnectionDisplayState::Configured - ); let stale = ::runtime_trust_key_revoke( &service, actor, @@ -43084,7 +42602,7 @@ mod tests { }, display_name: Some("Team Runtime".to_string()), endpoint: "https://8.8.8.8".to_string(), - workspace_trust_id: "trust-test".to_string(), + expected_binding_id: None, }; assert!(validate_runtime_connection_request(&ok).await.is_ok()); @@ -43114,59 +42632,6 @@ mod tests { } } - #[tokio::test] - async fn runtime_registration_rejects_blank_and_malformed_workspace_trust_ids() { - let temp = tempfile::tempdir().unwrap(); - let api = test_api(temp.path()).await; - let actor = test_owner_actor(); - let identity = RuntimeIdentityMaterial::generate("invalid-trust-runtime").unwrap(); - for trust_id in [ - String::new(), - " ".to_string(), - " trust-valid".to_string(), - "trust-valid ".to_string(), - "trust invalid".to_string(), - "trust\ninvalid".to_string(), - "x".repeat(257), - ] { - let error = create_remote_runtime( - State(api.clone()), - Extension(actor.clone()), - Json(CreateRemoteRuntimeRequest { - public_bundle: server_api::RuntimePublicIdentityBundle { - identity_id: identity.identity_id.clone(), - public_key: identity.public_key.clone(), - }, - display_name: None, - endpoint: "http://127.0.0.1:1".to_string(), - workspace_trust_id: trust_id.clone(), - expected_binding_id: None, - }), - ) - .await - .unwrap_err(); - let response = error.into_response(); - assert_eq!(response.status(), StatusCode::BAD_REQUEST, "{trust_id:?}"); - let body: serde_json::Value = - serde_json::from_slice(&to_bytes(response.into_body(), usize::MAX).await.unwrap()) - .unwrap(); - assert!( - body["message"] - .as_str() - .unwrap() - .contains("invalid_workspace_trust_id"), - "{body}" - ); - assert!( - api.store - .get_workspace_runtime_binding(api.workspace_id(), &identity.identity_id) - .await - .unwrap() - .is_none() - ); - } - } - #[test] fn backend_errors_preserve_operation_details() { let sanitized = sanitize_backend_error("failed to open server database"); @@ -45946,7 +45411,7 @@ mod tests { } #[test] - fn runtime_binding_summary_omits_stale_verification_binding() { + fn runtime_binding_summary_exposes_configuration_and_revocation_without_auth_state() { let binding = WorkspaceRuntimeBinding { workspace_id: "workspace-a".to_string(), runtime_id: "runtime-a".to_string(), @@ -45955,36 +45420,17 @@ mod tests { public_key: "runtime-public-key".to_string(), public_key_fingerprint: "sha256:runtime".to_string(), binding_id: "binding-test".to_string(), - state: crate::store::WorkspaceRuntimeBindingState::Revoked, authentication_mode: crate::store::WorkspaceRuntimeAuthenticationMode::WorkspaceIdentity, - workspace_key_id: Some("WK-a".to_string()), - - workspace_public_key_fingerprint: Some("sha256:workspace-test".to_string()), - - workspace_trust_id: Some("trust-test".to_string()), created_at: "1".to_string(), updated_at: "2".to_string(), revoked_at: Some("2".to_string()), }; - let stale = crate::store::WorkspaceRuntimeVerificationEvidence { - workspace_id: "workspace-a".to_string(), - runtime_id: "runtime-a".to_string(), - binding_id: "binding-retired".to_string(), - workspace_key_id: "WK-a".to_string(), - workspace_public_key_fingerprint: "sha256:workspace-test".to_string(), - workspace_trust_id: "trust-test".to_string(), - runtime_public_key_fingerprint: "sha256:runtime".to_string(), - challenge_id: "challenge-a".to_string(), - state: "failed".to_string(), - last_outcome: "connectivity_failed".to_string(), - verified_at: None, - checked_at: "1".to_string(), - }; - - let summary = runtime_binding_summary(&binding, Some(&stale)); - - assert_eq!(summary.verification, None); + let summary = serde_json::to_value(runtime_binding_summary(&binding)).unwrap(); + assert_eq!( + summary, + serde_json::json!({"binding_id": "binding-test", "revoked_at": "2"}) + ); } #[test] @@ -51944,13 +51390,7 @@ mod tests { public_key: identity.public_key, public_key_fingerprint: String::new(), binding_id: "binding-test".to_string(), - state: StoredRuntimeBindingState::Verified, authentication_mode: StoredRuntimeAuthenticationMode::WorkspaceIdentity, - workspace_key_id: Some("WK-test".to_owned()), - - workspace_public_key_fingerprint: Some("sha256:workspace-test".to_string()), - - workspace_trust_id: Some("trust-test".to_string()), created_at: now.clone(), updated_at: now, revoked_at: None, @@ -52337,13 +51777,7 @@ mod tests { public_key: identity.public_key, public_key_fingerprint: String::new(), binding_id: "binding-test".to_string(), - state: StoredRuntimeBindingState::Verified, authentication_mode: StoredRuntimeAuthenticationMode::LegacyServerIssuer, - workspace_key_id: None, - - workspace_public_key_fingerprint: None, - - workspace_trust_id: None, created_at: "1".to_string(), updated_at: "1".to_string(), revoked_at: None, @@ -52954,13 +52388,7 @@ mod tests { public_key: identity.public_key.clone(), public_key_fingerprint: String::new(), binding_id: "binding-test".to_string(), - state: StoredRuntimeBindingState::Verified, authentication_mode: StoredRuntimeAuthenticationMode::LegacyServerIssuer, - workspace_key_id: None, - - workspace_public_key_fingerprint: None, - - workspace_trust_id: None, created_at: "2026-08-11T00:00:00Z".to_string(), updated_at: "2026-08-11T00:00:00Z".to_string(), revoked_at: None, @@ -53228,7 +52656,6 @@ mod tests { assert_eq!(missing_mutation_response.status(), StatusCode::UNAUTHORIZED); let mut revoked = trust; - revoked.state = StoredRuntimeBindingState::Revoked; revoked.revoked_at = Some("2026-08-11T00:01:00Z".to_string()); let authority = SqliteWorkspaceStore::open(api.config.database_path.clone()).unwrap(); authority @@ -53688,13 +53115,7 @@ mod tests { public_key: provider_identity.public_key, public_key_fingerprint: String::new(), binding_id: "binding-test".to_string(), - state: StoredRuntimeBindingState::Verified, authentication_mode: StoredRuntimeAuthenticationMode::LegacyServerIssuer, - workspace_key_id: None, - - workspace_public_key_fingerprint: None, - - workspace_trust_id: None, created_at: now.clone(), updated_at: now, revoked_at: None, @@ -56487,13 +55908,7 @@ mod tests { .public_key, public_key_fingerprint: String::new(), binding_id: "binding-test".to_string(), - state: StoredRuntimeBindingState::Verified, authentication_mode: StoredRuntimeAuthenticationMode::LegacyServerIssuer, - workspace_key_id: None, - - workspace_public_key_fingerprint: None, - - workspace_trust_id: None, created_at: "1".to_string(), updated_at: "1".to_string(), revoked_at: None, @@ -57789,13 +57204,7 @@ mod tests { public_key: identity.public_key, public_key_fingerprint: String::new(), binding_id: "binding-test".to_string(), - state: StoredRuntimeBindingState::Verified, authentication_mode: StoredRuntimeAuthenticationMode::LegacyServerIssuer, - workspace_key_id: None, - - workspace_public_key_fingerprint: None, - - workspace_trust_id: None, created_at: "1".to_string(), updated_at: "1".to_string(), revoked_at: None, @@ -57861,8 +57270,7 @@ mod tests { "public_key": "" }, "display_name": "Keyless Runtime", - "endpoint": "https://8.8.8.8", - "workspace_trust_id": "trust-keyless-test" + "endpoint": "https://8.8.8.8" })), StatusCode::BAD_REQUEST, ) @@ -58155,13 +57563,7 @@ mod tests { .public_key, public_key_fingerprint: String::new(), binding_id: "binding-test".to_string(), - state: StoredRuntimeBindingState::Verified, authentication_mode: StoredRuntimeAuthenticationMode::LegacyServerIssuer, - workspace_key_id: None, - - workspace_public_key_fingerprint: None, - - workspace_trust_id: None, created_at: "1".to_string(), updated_at: "1".to_string(), revoked_at: None, @@ -58219,6 +57621,67 @@ mod tests { assert!(persisted.revoked_at.is_none()); } + #[tokio::test(flavor = "multi_thread")] + async fn runtime_connection_test_rejects_binding_changed_during_ping() { + let dir = tempfile::tempdir().unwrap(); + let api = test_api(dir.path()).await; + let listener = tokio::net::TcpListener::bind("127.0.0.1:0").await.unwrap(); + let endpoint = format!("http://{}", listener.local_addr().unwrap()); + let binding = WorkspaceRuntimeBinding { + workspace_id: TEST_WORKSPACE_ID.to_string(), + runtime_id: "probe-runtime".to_string(), + display_name: "Probe Runtime".to_string(), + base_url: endpoint.clone(), + public_key: RuntimeIdentityMaterial::generate("probe-runtime") + .unwrap() + .public_key, + public_key_fingerprint: String::new(), + binding_id: "binding-before-ping".to_string(), + authentication_mode: StoredRuntimeAuthenticationMode::LegacyServerIssuer, + created_at: "1".to_string(), + updated_at: "1".to_string(), + revoked_at: None, + }; + api.store + .upsert_workspace_runtime_binding_record(binding.clone(), false) + .await + .unwrap(); + let mut config = RemoteRuntimeConfig::new( + "probe-runtime", + "Probe Runtime", + &endpoint, + Some("test-token".to_string()), + ); + config.strict_public_egress = false; + api.runtime.register_or_replace( + RemoteWorkerRuntime::new( + config, + TEST_WORKSPACE_ID.to_string(), + "http://127.0.0.1:1".to_string(), + ) + .unwrap(), + ); + let store = api.store.clone(); + // Mutate only after the HTTP request reached the selected connection. + let stub = Router::new().route("/v1/ping", axum::routing::get(move || { + let store = store.clone(); let mut replacement = binding.clone(); + async move { + replacement.binding_id = "binding-after-ping".to_string(); + replacement.display_name = "Replacement Runtime".to_string(); + store.upsert_workspace_runtime_binding_record(replacement, true).await.unwrap(); + Json(serde_json::json!({"runtime_id":"probe-runtime", "protocol_version":RUNTIME_HTTP_PROTOCOL_VERSION})) + } + })); + let server = tokio::spawn(async move { + axum::serve(listener, stub).await.unwrap(); + }); + let result = + test_runtime_connection(State(api), AxumPath("probe-runtime".to_string())).await; + server.abort(); + let error = result.unwrap_err(); + assert_eq!(error.into_response().status(), StatusCode::CONFLICT); + } + async fn run_runtime_connection_test( body: serde_json::Value, status: StatusCode, @@ -58269,8 +57732,8 @@ mod tests { .await; assert_eq!(response["status"], "compatible"); - assert_eq!(response["connection_state"], "verified"); - assert_eq!(response["verification"], serde_json::Value::Null); + assert!(response.get("connection_state").is_none()); + assert!(response.get("verification").is_none()); assert_eq!(response["failure_kind"], serde_json::Value::Null); assert_eq!( response["expected_protocol_version"], diff --git a/crates/workspace-server/src/server/tests/drive_web_roundtrip_tests.rs b/crates/workspace-server/src/server/tests/drive_web_roundtrip_tests.rs index e267a77ca..02ab40e35 100644 --- a/crates/workspace-server/src/server/tests/drive_web_roundtrip_tests.rs +++ b/crates/workspace-server/src/server/tests/drive_web_roundtrip_tests.rs @@ -111,11 +111,9 @@ impl RealDriveFixture { public_key: identity.public_key.clone(), public_key_fingerprint: String::new(), binding_id: "binding-drive-test".to_string(), - state: StoredRuntimeBindingState::Verified, + authentication_mode: StoredRuntimeAuthenticationMode::LegacyServerIssuer, - workspace_key_id: None, - workspace_public_key_fingerprint: None, - workspace_trust_id: None, + created_at: "1".into(), updated_at: "1".into(), revoked_at: None, diff --git a/crates/workspace-server/src/server/tests/ticket_worker_launch_tests.rs b/crates/workspace-server/src/server/tests/ticket_worker_launch_tests.rs index 0a267d1b7..f9082a218 100644 --- a/crates/workspace-server/src/server/tests/ticket_worker_launch_tests.rs +++ b/crates/workspace-server/src/server/tests/ticket_worker_launch_tests.rs @@ -160,12 +160,10 @@ impl LaunchFixture { if remote { fixture.runtime_id = "ticket-worker-real-http".to_string(); use worker_runtime::workspace_issuer::{ - InMemoryWorkspaceClaimReplayProtection, - InMemoryWorkspaceRuntimeVerificationAuthority, RuntimeVerificationSigner, - WorkspaceCapabilityVerifier, WorkspaceIssuerTrustRecord, WorkspaceIssuerTrustState, + InMemoryWorkspaceClaimReplayProtection, WorkspaceCapabilityVerifier, + WorkspaceIssuerTrustRecord, WorkspaceIssuerTrustState, }; let runtime_identity = RuntimeIdentityMaterial::generate(&fixture.runtime_id).unwrap(); - let signer = RuntimeVerificationSigner::from_identity(&runtime_identity).unwrap(); let workspace_identity = fixture .api .signing_identities @@ -204,13 +202,11 @@ impl LaunchFixture { display_name: "Ticket Worker real HTTP Runtime".into(), base_url: endpoint.clone(), public_key: runtime_identity.public_key.clone(), - public_key_fingerprint: signer.public_key_fingerprint().to_string(), + public_key_fingerprint: String::new(), binding_id: "binding-launch-test".to_string(), - state: StoredRuntimeBindingState::Configured, + authentication_mode: StoredRuntimeAuthenticationMode::WorkspaceIdentity, - workspace_key_id: Some(workspace_identity.key_id), - workspace_public_key_fingerprint: workspace_identity.public_key_fingerprint.clone(), - workspace_trust_id: Some("trust-launch-test".to_string()), + created_at: TEST_CREATED_AT.into(), updated_at: TEST_CREATED_AT.into(), revoked_at: None, @@ -228,6 +224,7 @@ impl LaunchFixture { .await .unwrap() .unwrap(); + let runtime_id = fixture.runtime_id.clone(); fixture.runtime_server = Some(tokio::spawn(async move { worker_runtime::http_server::serve_runtime_http_with_workspace_auth( runtime, @@ -235,10 +232,7 @@ impl LaunchFixture { None, worker_runtime::http_server::WorkspaceRuntimeHttpAuth { verifier, - signer, - verifications: Arc::new( - InMemoryWorkspaceRuntimeVerificationAuthority::default(), - ), + runtime_id, }, ) .await @@ -258,27 +252,12 @@ impl LaunchFixture { cached_status: "active".into(), timeout: std::time::Duration::from_secs(5), }; - fixture.api.runtime.register_or_replace( - RemoteWorkerRuntime::new( - remote_config.clone(), - TEST_WORKSPACE_ID.to_string(), - backend_url.clone(), - ) - .unwrap(), - ); - let verified = perform_workspace_runtime_verification( - &fixture.api, - fixture.api.runtime.clone(), - &binding, - ) - .await - .unwrap(); remote_config.workspace_authorization = Some(crate::hosts::WorkspaceRuntimeAuthorization::new( fixture.api.store.clone(), fixture.api.signing_identities.clone(), backend_url.clone(), - Some(verified), + binding, )); fixture.api.runtime.register_or_replace( RemoteWorkerRuntime::new(remote_config, TEST_WORKSPACE_ID.to_string(), backend_url) diff --git a/crates/workspace-server/src/server/tests/workspace_admission_tests.rs b/crates/workspace-server/src/server/tests/workspace_admission_tests.rs index d90918a36..9bfbd3364 100644 --- a/crates/workspace-server/src/server/tests/workspace_admission_tests.rs +++ b/crates/workspace-server/src/server/tests/workspace_admission_tests.rs @@ -694,7 +694,6 @@ async fn revoked_runtime_and_persisted_workspace_fence_reject_new_requests() { .unwrap() .unwrap(); binding.revoked_at = Some("revoked".into()); - binding.state = StoredRuntimeBindingState::Revoked; f.api .store .upsert_workspace_runtime_binding_record(binding, true) diff --git a/crates/workspace-server/src/store.rs b/crates/workspace-server/src/store.rs index 3db49cc5d..922884608 100644 --- a/crates/workspace-server/src/store.rs +++ b/crates/workspace-server/src/store.rs @@ -28,7 +28,7 @@ use crate::workspace_deletion::WorkspaceDeletionStore; use crate::{Error, Result}; const OLDEST_SCHEMA_VERSION: i64 = 50; -const LATEST_SCHEMA_VERSION: i64 = 89; +const LATEST_SCHEMA_VERSION: i64 = 90; const WORKSPACE_DRIVE_GRANTS_MIGRATION_NAME: &str = "durable Workspace Drive grants"; const WORKSPACE_CONFIG_GRANTS_MIGRATION_NAME: &str = "Workspace config grants and logical Workdirs"; const SCHEMA_BASELINE_NAME: &str = "workspace schema baseline"; @@ -281,6 +281,11 @@ const MIGRATIONS: &[Migration] = &[ name: "descriptive legacy evidence columns", apply: migrate_evidence_column_names_v88_to_v89, }, + Migration { + version: 90, + name: "Runtime binding connection configuration and historical verification archive", + apply: migrate_runtime_binding_connections_v89_to_v90, + }, ]; #[derive(Clone, Copy)] @@ -441,22 +446,6 @@ pub struct WorkspaceSigningIdentityProvisioningOperation { pub completed_at: Option, } -#[derive(Debug, Clone, Serialize, Deserialize, PartialEq, Eq)] -pub struct WorkspaceRuntimeVerificationEvidence { - pub workspace_id: String, - pub runtime_id: String, - pub binding_id: String, - pub workspace_key_id: String, - pub workspace_public_key_fingerprint: String, - pub workspace_trust_id: String, - pub runtime_public_key_fingerprint: String, - pub challenge_id: String, - pub state: String, - pub last_outcome: String, - pub verified_at: Option, - pub checked_at: String, -} - #[derive(Debug, Clone, Serialize, Deserialize, PartialEq, Eq)] pub struct WorkspaceRuntimeBinding { pub workspace_id: String, @@ -466,11 +455,7 @@ pub struct WorkspaceRuntimeBinding { pub public_key: String, pub public_key_fingerprint: String, pub binding_id: String, - pub state: WorkspaceRuntimeBindingState, pub authentication_mode: WorkspaceRuntimeAuthenticationMode, - pub workspace_key_id: Option, - pub workspace_public_key_fingerprint: Option, - pub workspace_trust_id: Option, pub created_at: String, pub updated_at: String, pub revoked_at: Option, @@ -521,24 +506,6 @@ pub struct RuntimeRemovalReservation { pub replay: bool, } -#[derive(Debug, Clone, Copy, Serialize, Deserialize, PartialEq, Eq)] -#[serde(rename_all = "snake_case")] -pub enum WorkspaceRuntimeBindingState { - Configured, - Verified, - Revoked, -} - -impl WorkspaceRuntimeBindingState { - fn as_str(self) -> &'static str { - match self { - Self::Configured => "configured", - Self::Verified => "verified", - Self::Revoked => "revoked", - } - } -} - #[derive(Debug, Clone, Copy, Serialize, Deserialize, PartialEq, Eq)] #[serde(rename_all = "snake_case")] pub enum WorkspaceRuntimeAuthenticationMode { @@ -1190,32 +1157,6 @@ pub trait ControlPlaneStore: Send + Sync + WorkspaceDeletionStore { ) -> Result; fn workspace_runtime_binding_matches(&self, expected: &WorkspaceRuntimeBinding) -> Result; - fn workspace_runtime_verification_matches( - &self, - binding: &WorkspaceRuntimeBinding, - workspace_public_key_fingerprint: &str, - workspace_trust_id: &str, - ) -> Result; - async fn get_workspace_runtime_verification( - &self, - workspace_id: &str, - runtime_id: &str, - ) -> Result>; - async fn record_workspace_runtime_verification_attempt( - &self, - evidence: &WorkspaceRuntimeVerificationEvidence, - ) -> Result<()>; - async fn record_workspace_runtime_verification_outcome_if_current( - &self, - expected: &WorkspaceRuntimeVerificationEvidence, - state: &str, - outcome: &str, - checked_at: &str, - ) -> Result; - async fn complete_workspace_runtime_verification( - &self, - evidence: &WorkspaceRuntimeVerificationEvidence, - ) -> Result; async fn get_workspace_runtime_binding( &self, workspace_id: &str, @@ -2930,15 +2871,15 @@ impl SqliteWorkspaceStore { self.with_conn(|conn| { let sql = if include_revoked { r#"SELECT workspace_id, runtime_id, display_name, base_url, public_key, - public_key_fingerprint, binding_id, state, authentication_mode, - workspace_key_id, workspace_trust_id, created_at, updated_at, revoked_at, workspace_public_key_fingerprint + public_key_fingerprint, binding_id, authentication_mode, + created_at, updated_at, revoked_at FROM workspace_runtime_bindings WHERE workspace_id = ?1 ORDER BY runtime_id ASC"# } else { r#"SELECT workspace_id, runtime_id, display_name, base_url, public_key, - public_key_fingerprint, binding_id, state, authentication_mode, - workspace_key_id, workspace_trust_id, created_at, updated_at, revoked_at, workspace_public_key_fingerprint + public_key_fingerprint, binding_id, authentication_mode, + created_at, updated_at, revoked_at FROM workspace_runtime_bindings WHERE workspace_id = ?1 AND revoked_at IS NULL ORDER BY runtime_id ASC"# @@ -2960,8 +2901,8 @@ impl SqliteWorkspaceStore { self.with_conn(|conn| { conn.query_row( r#"SELECT workspace_id, runtime_id, display_name, base_url, public_key, - public_key_fingerprint, binding_id, state, authentication_mode, - workspace_key_id, workspace_trust_id, created_at, updated_at, revoked_at, workspace_public_key_fingerprint + public_key_fingerprint, binding_id, authentication_mode, + created_at, updated_at, revoked_at FROM workspace_runtime_bindings WHERE workspace_id = ?1 AND runtime_id = ?2"#, params![workspace_id, runtime_id], @@ -3314,8 +3255,8 @@ impl SqliteWorkspaceStore { let existing = tx .query_row( r#"SELECT workspace_id, runtime_id, display_name, base_url, public_key, - public_key_fingerprint, binding_id, state, authentication_mode, - workspace_key_id, workspace_trust_id, created_at, updated_at, revoked_at, workspace_public_key_fingerprint + public_key_fingerprint, binding_id, authentication_mode, + created_at, updated_at, revoked_at FROM workspace_runtime_bindings WHERE workspace_id = ?1 AND runtime_id = ?2"#, params![record.workspace_id, record.runtime_id], @@ -3329,76 +3270,42 @@ impl SqliteWorkspaceStore { && existing.base_url == record.base_url && existing.public_key == record.public_key && existing.public_key_fingerprint == record.public_key_fingerprint - && existing.state == record.state - && existing.authentication_mode == record.authentication_mode - && existing.workspace_key_id == record.workspace_key_id - && existing.workspace_public_key_fingerprint == record.workspace_public_key_fingerprint - && existing.workspace_trust_id == record.workspace_trust_id; + && existing.authentication_mode == record.authentication_mode; if exact_active_match { tx.commit()?; return Ok(WorkspaceRuntimeBindingUpsert::Unchanged); } if !replace { return Err(Error::RuntimeBindingConflict(format!( - "binding {}/{} already exists with different endpoint, trust, or lifecycle state; retry with explicit replacement", + "binding {}/{} already exists with different endpoint, pinned key, authentication mode, or administrative revocation; retry with explicit replacement", record.workspace_id, record.runtime_id ))); } tx.execute( r#"UPDATE workspace_runtime_bindings SET display_name = ?3, base_url = ?4, public_key = ?5, - public_key_fingerprint = ?6, binding_id = ?13, - state = ?7, authentication_mode = ?8, - workspace_key_id = ?9, workspace_trust_id = ?10, - updated_at = ?11, revoked_at = ?12, workspace_public_key_fingerprint = ?14 + public_key_fingerprint = ?6, binding_id = ?10, + authentication_mode = ?7, updated_at = ?8, revoked_at = ?9 WHERE workspace_id = ?1 AND runtime_id = ?2"#, - params![ - record.workspace_id, - record.runtime_id, - record.display_name, - record.base_url, - record.public_key, - record.public_key_fingerprint, - record.state.as_str(), - record.authentication_mode.as_str(), - record.workspace_key_id, - record.workspace_trust_id, - record.updated_at, - record.revoked_at, - new_runtime_binding_id(), - record.workspace_public_key_fingerprint, ], + params![record.workspace_id, record.runtime_id, record.display_name, + record.base_url, record.public_key, record.public_key_fingerprint, + record.authentication_mode.as_str(), record.updated_at, record.revoked_at, + new_runtime_binding_id()], ) .map_err(map_runtime_binding_write_error)?; - tx.execute( - "DELETE FROM workspace_runtime_verifications WHERE workspace_id = ?1 AND runtime_id = ?2", - params![record.workspace_id, record.runtime_id], - )?; tx.commit()?; return Ok(WorkspaceRuntimeBindingUpsert::Replaced); } tx.execute( r#"INSERT INTO workspace_runtime_bindings ( workspace_id, runtime_id, display_name, base_url, public_key, - public_key_fingerprint, binding_id, state, authentication_mode, - workspace_key_id, workspace_trust_id, created_at, updated_at, revoked_at, workspace_public_key_fingerprint - ) VALUES (?1, ?2, ?3, ?4, ?5, ?6, ?14, ?7, ?8, ?9, ?10, ?11, ?12, ?13, ?15)"#, - params![ - record.workspace_id, - record.runtime_id, - record.display_name, - record.base_url, - record.public_key, - record.public_key_fingerprint, - record.state.as_str(), - record.authentication_mode.as_str(), - record.workspace_key_id, - record.workspace_trust_id, - record.created_at, - record.updated_at, - record.revoked_at, - new_runtime_binding_id(), - record.workspace_public_key_fingerprint, - ], + public_key_fingerprint, binding_id, authentication_mode, + created_at, updated_at, revoked_at + ) VALUES (?1, ?2, ?3, ?4, ?5, ?6, ?7, ?8, ?9, ?10, ?11)"#, + params![record.workspace_id, record.runtime_id, record.display_name, + record.base_url, record.public_key, record.public_key_fingerprint, + new_runtime_binding_id(), record.authentication_mode.as_str(), + record.created_at, record.updated_at, record.revoked_at], ) .map_err(map_runtime_binding_write_error)?; tx.commit()?; @@ -3419,17 +3326,16 @@ impl SqliteWorkspaceStore { let tx = conn.transaction_with_behavior(TransactionBehavior::Immediate)?; let changed = tx.execute( r#"UPDATE workspace_runtime_bindings - SET state = 'revoked', revoked_at = ?3, updated_at = ?3, + SET revoked_at = ?3, updated_at = ?3, binding_id = ?4 WHERE workspace_id = ?1 AND runtime_id = ?2 AND revoked_at IS NULL"#, - params![workspace_id, runtime_id, revoked_at, new_runtime_binding_id()], + params![ + workspace_id, + runtime_id, + revoked_at, + new_runtime_binding_id() + ], )?; - if changed > 0 { - tx.execute( - "DELETE FROM workspace_runtime_verifications WHERE workspace_id = ?1 AND runtime_id = ?2", - params![workspace_id, runtime_id], - )?; - } tx.commit()?; Ok(changed > 0) }) @@ -3453,8 +3359,8 @@ impl SqliteWorkspaceStore { let existing = tx .query_row( r#"SELECT workspace_id, runtime_id, display_name, base_url, public_key, - public_key_fingerprint, binding_id, state, authentication_mode, - workspace_key_id, workspace_trust_id, created_at, updated_at, revoked_at, workspace_public_key_fingerprint + public_key_fingerprint, binding_id, authentication_mode, + created_at, updated_at, revoked_at FROM workspace_runtime_bindings WHERE workspace_id = ?1 AND runtime_id = ?2"#, params![record.workspace_id, record.runtime_id], @@ -3474,11 +3380,7 @@ impl SqliteWorkspaceStore { && existing.base_url == record.base_url && existing.public_key == record.public_key && existing.public_key_fingerprint == record.public_key_fingerprint - && existing.state == record.state && existing.authentication_mode == record.authentication_mode - && existing.workspace_key_id == record.workspace_key_id - && existing.workspace_public_key_fingerprint == record.workspace_public_key_fingerprint - && existing.workspace_trust_id == record.workspace_trust_id { tx.commit()?; return Ok((WorkspaceRuntimeBindingMutation::Unchanged, existing)); @@ -3522,9 +3424,8 @@ impl SqliteWorkspaceStore { r#"UPDATE workspace_runtime_bindings SET display_name = ?3, base_url = ?4, public_key = ?5, public_key_fingerprint = ?6, - state = ?7, authentication_mode = ?8, - workspace_key_id = ?9, workspace_trust_id = ?10, - binding_id = ?11, updated_at = ?12, revoked_at = NULL, workspace_public_key_fingerprint = ?13 + authentication_mode = ?7, binding_id = ?8, + updated_at = ?9, revoked_at = NULL WHERE workspace_id = ?1 AND runtime_id = ?2"#, params![ record.workspace_id, @@ -3533,19 +3434,11 @@ impl SqliteWorkspaceStore { record.base_url, record.public_key, record.public_key_fingerprint, - record.state.as_str(), record.authentication_mode.as_str(), - record.workspace_key_id, - record.workspace_trust_id, &next_binding_id, - record.updated_at, - record.workspace_public_key_fingerprint, + record.updated_at ], )?; - tx.execute( - "DELETE FROM workspace_runtime_verifications WHERE workspace_id = ?1 AND runtime_id = ?2", - params![record.workspace_id, record.runtime_id], - )?; insert_workspace_runtime_binding_audit( &tx, &record.workspace_id, @@ -3559,8 +3452,8 @@ impl SqliteWorkspaceStore { )?; let updated = tx.query_row( r#"SELECT workspace_id, runtime_id, display_name, base_url, public_key, - public_key_fingerprint, binding_id, state, authentication_mode, - workspace_key_id, workspace_trust_id, created_at, updated_at, revoked_at, workspace_public_key_fingerprint + public_key_fingerprint, binding_id, authentication_mode, + created_at, updated_at, revoked_at FROM workspace_runtime_bindings WHERE workspace_id = ?1 AND runtime_id = ?2"#, params![record.workspace_id, record.runtime_id], @@ -3594,9 +3487,9 @@ impl SqliteWorkspaceStore { tx.execute( r#"INSERT INTO workspace_runtime_bindings ( workspace_id, runtime_id, display_name, base_url, public_key, - public_key_fingerprint, binding_id, state, authentication_mode, - workspace_key_id, workspace_trust_id, created_at, updated_at, revoked_at, workspace_public_key_fingerprint - ) VALUES (?1, ?2, ?3, ?4, ?5, ?6, ?13, ?7, ?8, ?9, ?10, ?11, ?12, NULL, ?14)"#, + public_key_fingerprint, binding_id, authentication_mode, + created_at, updated_at, revoked_at + ) VALUES (?1, ?2, ?3, ?4, ?5, ?6, ?7, ?8, ?9, ?10, NULL)"#, params![ record.workspace_id, record.runtime_id, @@ -3604,14 +3497,11 @@ impl SqliteWorkspaceStore { record.base_url, record.public_key, record.public_key_fingerprint, - record.state.as_str(), - record.authentication_mode.as_str(), - record.workspace_key_id, - record.workspace_trust_id, - record.created_at, - record.updated_at, record.binding_id, - record.workspace_public_key_fingerprint, ], + record.authentication_mode.as_str(), + record.created_at, + record.updated_at + ], )?; insert_workspace_runtime_binding_audit( &tx, @@ -3647,8 +3537,8 @@ impl SqliteWorkspaceStore { let existing = tx .query_row( r#"SELECT workspace_id, runtime_id, display_name, base_url, public_key, - public_key_fingerprint, binding_id, state, authentication_mode, - workspace_key_id, workspace_trust_id, created_at, updated_at, revoked_at, workspace_public_key_fingerprint + public_key_fingerprint, binding_id, authentication_mode, + created_at, updated_at, revoked_at FROM workspace_runtime_bindings WHERE workspace_id = ?1 AND runtime_id = ?2"#, params![workspace_id, runtime_id], @@ -3665,21 +3555,25 @@ impl SqliteWorkspaceStore { tx.execute( r#"UPDATE workspace_runtime_bindings SET display_name = ?3, base_url = ?4, updated_at = ?5, - binding_id = ?6, - state = CASE WHEN base_url != ?4 AND state = 'verified' - AND authentication_mode = 'workspace_identity' - THEN 'configured' ELSE state END + binding_id = ?6 WHERE workspace_id = ?1 AND runtime_id = ?2"#, - params![workspace_id, runtime_id, display_name, base_url, updated_at, - if existing.base_url != base_url { new_runtime_binding_id() } else { existing.binding_id.clone() }], + params![ + workspace_id, + runtime_id, + display_name, + base_url, + updated_at, + if existing.base_url != base_url { + new_runtime_binding_id() + } else { + existing.binding_id.clone() + } + ], )?; - if existing.base_url != base_url { - tx.execute("DELETE FROM workspace_runtime_verifications WHERE workspace_id=?1 AND runtime_id=?2", params![workspace_id,runtime_id])?; - } let updated = tx.query_row( r#"SELECT workspace_id, runtime_id, display_name, base_url, public_key, - public_key_fingerprint, binding_id, state, authentication_mode, - workspace_key_id, workspace_trust_id, created_at, updated_at, revoked_at, workspace_public_key_fingerprint + public_key_fingerprint, binding_id, authentication_mode, + created_at, updated_at, revoked_at FROM workspace_runtime_bindings WHERE workspace_id = ?1 AND runtime_id = ?2"#, params![workspace_id, runtime_id], @@ -3707,8 +3601,8 @@ impl SqliteWorkspaceStore { let existing = tx .query_row( r#"SELECT workspace_id, runtime_id, display_name, base_url, public_key, - public_key_fingerprint, binding_id, state, authentication_mode, - workspace_key_id, workspace_trust_id, created_at, updated_at, revoked_at, workspace_public_key_fingerprint + public_key_fingerprint, binding_id, authentication_mode, + created_at, updated_at, revoked_at FROM workspace_runtime_bindings WHERE workspace_id = ?1 AND runtime_id = ?2"#, params![workspace_id, runtime_id], @@ -3731,14 +3625,10 @@ impl SqliteWorkspaceStore { let next_binding_id = new_runtime_binding_id(); tx.execute( r#"UPDATE workspace_runtime_bindings - SET state = 'revoked', revoked_at = ?3, updated_at = ?3, binding_id = ?4 + SET revoked_at = ?3, updated_at = ?3, binding_id = ?4 WHERE workspace_id = ?1 AND runtime_id = ?2"#, params![workspace_id, runtime_id, revoked_at, next_binding_id], )?; - tx.execute( - "DELETE FROM workspace_runtime_verifications WHERE workspace_id = ?1 AND runtime_id = ?2", - params![workspace_id, runtime_id], - )?; insert_workspace_runtime_binding_audit( &tx, workspace_id, @@ -3752,8 +3642,8 @@ impl SqliteWorkspaceStore { )?; let updated = tx.query_row( r#"SELECT workspace_id, runtime_id, display_name, base_url, public_key, - public_key_fingerprint, binding_id, state, authentication_mode, - workspace_key_id, workspace_trust_id, created_at, updated_at, revoked_at, workspace_public_key_fingerprint + public_key_fingerprint, binding_id, authentication_mode, + created_at, updated_at, revoked_at FROM workspace_runtime_bindings WHERE workspace_id = ?1 AND runtime_id = ?2"#, params![workspace_id, runtime_id], @@ -3764,310 +3654,6 @@ impl SqliteWorkspaceStore { }) } - pub fn workspace_runtime_verification_matches( - &self, - binding: &WorkspaceRuntimeBinding, - workspace_public_key_fingerprint: &str, - workspace_trust_id: &str, - ) -> Result { - let Some(evidence) = - self.get_workspace_runtime_verification(&binding.workspace_id, &binding.runtime_id)? - else { - return Ok(false); - }; - Ok(evidence.state == "verified" - && evidence.last_outcome == "verified" - && evidence.verified_at.is_some() - && evidence.binding_id == binding.binding_id - && evidence.workspace_key_id == binding.workspace_key_id.as_deref().unwrap_or_default() - && evidence.workspace_public_key_fingerprint == workspace_public_key_fingerprint - && binding.workspace_public_key_fingerprint.as_deref() - == Some(workspace_public_key_fingerprint) - && evidence.workspace_trust_id == workspace_trust_id - && evidence.runtime_public_key_fingerprint == binding.public_key_fingerprint - && binding.revoked_at.is_none() - && binding.workspace_trust_id.as_deref() == Some(workspace_trust_id)) - } - - pub fn get_workspace_runtime_verification( - &self, - workspace_id: &str, - runtime_id: &str, - ) -> Result> { - validate_identifier("workspace_id", workspace_id)?; - validate_identifier("runtime_id", runtime_id)?; - self.with_conn(|conn| { - let evidence = conn - .query_row( - r#"SELECT workspace_id, runtime_id, binding_id, workspace_key_id, - workspace_public_key_fingerprint, workspace_trust_id, - runtime_public_key_fingerprint, - challenge_id, state, last_outcome, verified_at, checked_at - FROM workspace_runtime_verifications - WHERE workspace_id = ?1 AND runtime_id = ?2"#, - params![workspace_id, runtime_id], - read_workspace_runtime_verification, - ) - .optional()?; - if let Some(evidence) = &evidence { - validate_workspace_runtime_verification(evidence)?; - } - Ok(evidence) - }) - } - - pub fn record_workspace_runtime_verification_attempt( - &self, - evidence: &WorkspaceRuntimeVerificationEvidence, - ) -> Result<()> { - validate_workspace_runtime_verification(evidence)?; - self.with_conn(|conn| { - let changed = conn.execute( - r#"INSERT INTO workspace_runtime_verifications ( - workspace_id, runtime_id, binding_id, workspace_key_id, - workspace_public_key_fingerprint, workspace_trust_id, - runtime_public_key_fingerprint, - challenge_id, state, last_outcome, verified_at, checked_at - ) SELECT ?1, ?2, ?3, ?4, ?5, ?6, ?7, ?8, ?9, ?10, ?11, ?12 - WHERE EXISTS(SELECT 1 FROM workspace_runtime_bindings binding - WHERE binding.workspace_id=?1 AND binding.runtime_id=?2 - AND binding.binding_id=?3 AND binding.workspace_key_id=?4 - AND binding.workspace_public_key_fingerprint=?5 - AND binding.workspace_trust_id=?6 AND binding.public_key_fingerprint=?7 - AND binding.state IN ('configured','verified') AND binding.revoked_at IS NULL) - ON CONFLICT(workspace_id, runtime_id) DO UPDATE SET - binding_id = excluded.binding_id, - workspace_key_id = excluded.workspace_key_id, - workspace_public_key_fingerprint = excluded.workspace_public_key_fingerprint, - workspace_trust_id = excluded.workspace_trust_id, - runtime_public_key_fingerprint = excluded.runtime_public_key_fingerprint, - challenge_id = excluded.challenge_id, - state = CASE - WHEN workspace_runtime_verifications.state = 'verified' - AND workspace_runtime_verifications.binding_id = excluded.binding_id - AND workspace_runtime_verifications.workspace_key_id = excluded.workspace_key_id - AND workspace_runtime_verifications.workspace_public_key_fingerprint = excluded.workspace_public_key_fingerprint - AND workspace_runtime_verifications.workspace_trust_id = excluded.workspace_trust_id - AND workspace_runtime_verifications.runtime_public_key_fingerprint = excluded.runtime_public_key_fingerprint - THEN workspace_runtime_verifications.state - ELSE excluded.state - END, - last_outcome = excluded.last_outcome, - verified_at = CASE - WHEN workspace_runtime_verifications.state = 'verified' - AND workspace_runtime_verifications.binding_id = excluded.binding_id - AND workspace_runtime_verifications.workspace_key_id = excluded.workspace_key_id - AND workspace_runtime_verifications.workspace_public_key_fingerprint = excluded.workspace_public_key_fingerprint - AND workspace_runtime_verifications.workspace_trust_id = excluded.workspace_trust_id - AND workspace_runtime_verifications.runtime_public_key_fingerprint = excluded.runtime_public_key_fingerprint - THEN workspace_runtime_verifications.verified_at - ELSE excluded.verified_at - END, - checked_at = excluded.checked_at"#, - params![ - evidence.workspace_id, - evidence.runtime_id, - evidence.binding_id, - evidence.workspace_key_id, - evidence.workspace_public_key_fingerprint, - evidence.workspace_trust_id, - evidence.runtime_public_key_fingerprint, - - evidence.challenge_id, - evidence.state, - evidence.last_outcome, - evidence.verified_at, - evidence.checked_at, - ], - )?; - if changed != 1 { - return Err(Error::RuntimeBindingConflict("Runtime verification attempt no longer matches binding authority".into())); - } - Ok(()) - }) - } - - pub fn record_workspace_runtime_verification_outcome_if_current( - &self, - expected: &WorkspaceRuntimeVerificationEvidence, - state: &str, - outcome: &str, - checked_at: &str, - ) -> Result { - validate_workspace_runtime_verification(expected)?; - if !matches!(state, "pending" | "verified" | "failed") { - return Err(Error::Store( - "invalid Workspace Runtime verification state".to_string(), - )); - } - if !matches!( - outcome, - "challenge_issued" | "verified" | "verification_failed" | "connectivity_failed" - ) { - return Err(Error::Store( - "invalid Workspace Runtime verification outcome".to_string(), - )); - } - if checked_at.is_empty() || checked_at.len() > 128 { - return Err(Error::Store( - "invalid Workspace Runtime verification checked_at".to_string(), - )); - } - self.with_conn(|conn| { - let changed = conn.execute( - r#"UPDATE workspace_runtime_verifications - SET state = CASE WHEN state = 'verified' THEN state ELSE ?9 END, - last_outcome = ?10, - checked_at = ?11 - WHERE workspace_id = ?1 AND runtime_id = ?2 - AND binding_id = ?3 - AND workspace_key_id = ?4 - AND workspace_public_key_fingerprint = ?5 - AND workspace_trust_id = ?6 - AND runtime_public_key_fingerprint = ?7 - AND challenge_id = ?8"#, - params![ - expected.workspace_id, - expected.runtime_id, - expected.binding_id, - expected.workspace_key_id, - expected.workspace_public_key_fingerprint, - expected.workspace_trust_id, - expected.runtime_public_key_fingerprint, - expected.challenge_id, - state, - outcome, - checked_at, - ], - )?; - Ok(changed == 1) - }) - } - - pub fn complete_workspace_runtime_verification( - &self, - evidence: &WorkspaceRuntimeVerificationEvidence, - ) -> Result { - validate_workspace_runtime_verification(evidence)?; - if evidence.state != "verified" - || evidence.last_outcome != "verified" - || evidence.verified_at.is_none() - { - return Err(Error::Store( - "completed Runtime verification evidence must be verified".to_string(), - )); - } - self.with_conn_mut(|conn| { - let tx = conn.transaction_with_behavior(TransactionBehavior::Immediate)?; - let current_attempt = tx.query_row( - r#"SELECT EXISTS( - SELECT 1 FROM workspace_runtime_verifications - WHERE workspace_id = ?1 AND runtime_id = ?2 - AND binding_id = ?3 - AND workspace_key_id = ?4 - AND workspace_public_key_fingerprint = ?5 - AND workspace_trust_id = ?6 - AND runtime_public_key_fingerprint = ?7 - AND challenge_id = ?8 - )"#, - params![ - evidence.workspace_id, - evidence.runtime_id, - evidence.binding_id, - evidence.workspace_key_id, - evidence.workspace_public_key_fingerprint, - evidence.workspace_trust_id, - evidence.runtime_public_key_fingerprint, - - evidence.challenge_id, - ], - |row| row.get::<_, bool>(0), - )?; - if !current_attempt { - return Err(Error::RuntimeBindingConflict( - "Runtime verification attempt was superseded".to_string(), - )); - } - let changed = tx.execute( - r#"UPDATE workspace_runtime_bindings - SET state = 'verified', updated_at = ?7 - WHERE workspace_id = ?1 AND runtime_id = ?2 - AND binding_id = ?3 - AND state IN ('configured', 'verified') - AND authentication_mode = 'workspace_identity' - AND workspace_key_id = ?4 - AND workspace_trust_id = ?5 - AND workspace_public_key_fingerprint = ?8 - AND public_key_fingerprint = ?6 - AND revoked_at IS NULL - AND EXISTS(SELECT 1 FROM workspace_signing_identities identity - WHERE identity.workspace_id = ?1 AND identity.key_id = ?4 - AND identity.public_key_fingerprint = ?8 AND identity.state = 'active')"#, - params![ - evidence.workspace_id, - evidence.runtime_id, - evidence.binding_id, - evidence.workspace_key_id, - evidence.workspace_trust_id, - evidence.runtime_public_key_fingerprint, - evidence.checked_at, - evidence.workspace_public_key_fingerprint, - ], - )?; - if changed != 1 { - return Err(Error::RuntimeBindingConflict( - "Runtime verification evidence no longer matches the configured binding" - .to_string(), - )); - } - tx.execute( - r#"INSERT INTO workspace_runtime_verifications ( - workspace_id, runtime_id, binding_id, workspace_key_id, - workspace_public_key_fingerprint, workspace_trust_id, - runtime_public_key_fingerprint, - challenge_id, state, last_outcome, verified_at, checked_at - ) VALUES (?1, ?2, ?3, ?4, ?5, ?6, ?7, ?8, ?9, ?10, ?11, ?12) - ON CONFLICT(workspace_id, runtime_id) DO UPDATE SET - binding_id = excluded.binding_id, - workspace_key_id = excluded.workspace_key_id, - workspace_public_key_fingerprint = excluded.workspace_public_key_fingerprint, - workspace_trust_id = excluded.workspace_trust_id, - runtime_public_key_fingerprint = excluded.runtime_public_key_fingerprint, - challenge_id = excluded.challenge_id, - state = excluded.state, - last_outcome = excluded.last_outcome, - verified_at = excluded.verified_at, - checked_at = excluded.checked_at"#, - params![ - evidence.workspace_id, - evidence.runtime_id, - evidence.binding_id, - evidence.workspace_key_id, - evidence.workspace_public_key_fingerprint, - evidence.workspace_trust_id, - evidence.runtime_public_key_fingerprint, - - evidence.challenge_id, - evidence.state, - evidence.last_outcome, - evidence.verified_at, - evidence.checked_at, - ], - )?; - let binding = tx.query_row( - r#"SELECT workspace_id, runtime_id, display_name, base_url, public_key, - public_key_fingerprint, binding_id, state, authentication_mode, - workspace_key_id, workspace_trust_id, created_at, updated_at, revoked_at, workspace_public_key_fingerprint - FROM workspace_runtime_bindings - WHERE workspace_id = ?1 AND runtime_id = ?2"#, - params![evidence.workspace_id, evidence.runtime_id], - read_workspace_runtime_binding, - )?; - tx.commit()?; - Ok(binding) - }) - } - pub fn list_workspace_runtime_binding_audit( &self, workspace_id: &str, @@ -4937,54 +4523,6 @@ impl ControlPlaneStore for SqliteWorkspaceStore { .is_some_and(|binding| binding == *expected && binding.revoked_at.is_none())) } - fn workspace_runtime_verification_matches( - &self, - binding: &WorkspaceRuntimeBinding, - workspace_public_key_fingerprint: &str, - workspace_trust_id: &str, - ) -> Result { - SqliteWorkspaceStore::workspace_runtime_verification_matches( - self, - binding, - workspace_public_key_fingerprint, - workspace_trust_id, - ) - } - - async fn get_workspace_runtime_verification( - &self, - workspace_id: &str, - runtime_id: &str, - ) -> Result> { - SqliteWorkspaceStore::get_workspace_runtime_verification(self, workspace_id, runtime_id) - } - - async fn record_workspace_runtime_verification_attempt( - &self, - evidence: &WorkspaceRuntimeVerificationEvidence, - ) -> Result<()> { - SqliteWorkspaceStore::record_workspace_runtime_verification_attempt(self, evidence) - } - - async fn record_workspace_runtime_verification_outcome_if_current( - &self, - expected: &WorkspaceRuntimeVerificationEvidence, - state: &str, - outcome: &str, - checked_at: &str, - ) -> Result { - SqliteWorkspaceStore::record_workspace_runtime_verification_outcome_if_current( - self, expected, state, outcome, checked_at, - ) - } - - async fn complete_workspace_runtime_verification( - &self, - evidence: &WorkspaceRuntimeVerificationEvidence, - ) -> Result { - SqliteWorkspaceStore::complete_workspace_runtime_verification(self, evidence) - } - async fn get_workspace_runtime_binding( &self, workspace_id: &str, @@ -10765,65 +10303,6 @@ fn account_select_sql(where_clause: &str) -> String { ) } -fn read_workspace_runtime_verification( - row: &rusqlite::Row<'_>, -) -> rusqlite::Result { - Ok(WorkspaceRuntimeVerificationEvidence { - workspace_id: row.get(0)?, - runtime_id: row.get(1)?, - binding_id: row.get(2)?, - workspace_key_id: row.get(3)?, - workspace_public_key_fingerprint: row.get(4)?, - workspace_trust_id: row.get(5)?, - runtime_public_key_fingerprint: row.get(6)?, - challenge_id: row.get(7)?, - state: row.get(8)?, - last_outcome: row.get(9)?, - verified_at: row.get(10)?, - checked_at: row.get(11)?, - }) -} - -fn validate_workspace_runtime_verification( - evidence: &WorkspaceRuntimeVerificationEvidence, -) -> Result<()> { - for (field, value) in [ - ("workspace_id", evidence.workspace_id.as_str()), - ("runtime_id", evidence.runtime_id.as_str()), - ("workspace_key_id", evidence.workspace_key_id.as_str()), - ("challenge_id", evidence.challenge_id.as_str()), - ] { - validate_identifier(field, value)?; - } - validate_identifier("binding_id", &evidence.binding_id)?; - validate_identifier("workspace_trust_id", &evidence.workspace_trust_id)?; - validate_non_empty( - "workspace_public_key_fingerprint", - &evidence.workspace_public_key_fingerprint, - )?; - validate_non_empty( - "runtime_public_key_fingerprint", - &evidence.runtime_public_key_fingerprint, - )?; - if !matches!(evidence.state.as_str(), "pending" | "verified" | "failed") - || (evidence.state == "verified") != evidence.verified_at.is_some() - { - return Err(Error::InvalidInput( - "Runtime verification lifecycle evidence is invalid".into(), - )); - } - if !matches!( - evidence.last_outcome.as_str(), - "verified" | "challenge_issued" | "verification_failed" | "connectivity_failed" - ) { - return Err(Error::InvalidInput( - "Runtime verification outcome is invalid".to_string(), - )); - } - validate_non_empty("checked_at", &evidence.checked_at)?; - Ok(()) -} - fn read_runtime_removal_operation( row: &rusqlite::Row<'_>, ) -> rusqlite::Result { @@ -10902,7 +10381,7 @@ fn runtime_removal_preflight( ( "other_workspace_binding", "SELECT EXISTS(SELECT 1 FROM workspace_runtime_bindings \ - WHERE runtime_id = ?1 AND workspace_id <> ?2 AND state <> 'revoked')", + WHERE runtime_id = ?1 AND workspace_id <> ?2 AND revoked_at IS NULL)", ), ( "active_worker", @@ -10962,24 +10441,12 @@ fn runtime_removal_preflight( fn read_workspace_runtime_binding( row: &rusqlite::Row<'_>, ) -> rusqlite::Result { - let state = match row.get::<_, String>(7)?.as_str() { - "configured" => WorkspaceRuntimeBindingState::Configured, - "verified" => WorkspaceRuntimeBindingState::Verified, - "revoked" => WorkspaceRuntimeBindingState::Revoked, - value => { - return Err(rusqlite::Error::FromSqlConversionFailure( - 7, - rusqlite::types::Type::Text, - format!("unknown Workspace Runtime binding state {value}").into(), - )); - } - }; - let authentication_mode = match row.get::<_, String>(8)?.as_str() { + let authentication_mode = match row.get::<_, String>(7)?.as_str() { "legacy_server_issuer" => WorkspaceRuntimeAuthenticationMode::LegacyServerIssuer, "workspace_identity" => WorkspaceRuntimeAuthenticationMode::WorkspaceIdentity, value => { return Err(rusqlite::Error::FromSqlConversionFailure( - 8, + 7, rusqlite::types::Type::Text, format!("unknown Workspace Runtime authentication mode {value}").into(), )); @@ -10993,14 +10460,10 @@ fn read_workspace_runtime_binding( public_key: row.get(4)?, public_key_fingerprint: row.get(5)?, binding_id: row.get(6)?, - state, authentication_mode, - workspace_key_id: row.get(9)?, - workspace_public_key_fingerprint: row.get(14)?, - workspace_trust_id: row.get(10)?, - created_at: row.get(11)?, - updated_at: row.get(12)?, - revoked_at: row.get(13)?, + created_at: row.get(8)?, + updated_at: row.get(9)?, + revoked_at: row.get(10)?, }) } @@ -11056,56 +10519,17 @@ fn normalize_workspace_runtime_binding_key(record: &mut WorkspaceRuntimeBinding) } record.public_key = canonical; record.public_key_fingerprint = fingerprint; - match record.authentication_mode { - WorkspaceRuntimeAuthenticationMode::LegacyServerIssuer => { - if record.workspace_key_id.is_some() - || record.workspace_trust_id.is_some() - || record.workspace_public_key_fingerprint.is_some() - { - return Err(Error::InvalidInput( - "legacy Runtime bindings must not carry Workspace key metadata".into(), - )); - } - #[cfg(not(test))] - if record.runtime_id != crate::hosts::EMBEDDED_RUNTIME_ID { - return Err(Error::InvalidInput( - "legacy Server issuer authentication is reserved for the embedded Runtime" - .into(), - )); - } - } - WorkspaceRuntimeAuthenticationMode::WorkspaceIdentity => { - let key_id = record.workspace_key_id.as_deref().ok_or_else(|| { - Error::InvalidInput( - "Workspace identity Runtime bindings require workspace_key_id".into(), - ) - })?; - validate_identifier("workspace_key_id", key_id)?; - validate_non_empty( - "workspace_public_key_fingerprint", - record - .workspace_public_key_fingerprint - .as_deref() - .unwrap_or_default(), - )?; - if record - .workspace_trust_id - .as_deref() - .is_none_or(str::is_empty) - { - return Err(Error::InvalidInput( - "Workspace identity Runtime bindings require a non-empty workspace_trust_id" - .into(), - )); - } - } - } - let revoked = record.revoked_at.is_some(); - if (record.state == WorkspaceRuntimeBindingState::Revoked) != revoked { + #[cfg(not(test))] + if record.authentication_mode == WorkspaceRuntimeAuthenticationMode::LegacyServerIssuer + && record.runtime_id != crate::hosts::EMBEDDED_RUNTIME_ID + { return Err(Error::InvalidInput( - "Runtime binding state and revoked_at must agree".into(), + "legacy Server issuer authentication is reserved for the embedded Runtime".into(), )); } + if let Some(revoked_at) = &record.revoked_at { + validate_non_empty("revoked_at", revoked_at)?; + } Ok(()) } @@ -11952,7 +11376,7 @@ pub(crate) fn drive_worker_is_live( }; conn.query_row(&format!("SELECT EXISTS(SELECT 1 FROM {a}worker_registry wr JOIN {a}workspaces ws ON ws.workspace_id=wr.workspace_id WHERE wr.workspace_id=?1 AND wr.runtime_id=?2 AND wr.worker_id=?3 AND ws.state='active' - AND (wr.runtime_id=?4 OR EXISTS(SELECT 1 FROM {a}workspace_runtime_bindings rb WHERE rb.workspace_id=?1 AND rb.runtime_id=?2 AND rb.state='verified' AND rb.revoked_at IS NULL)) + AND (wr.runtime_id=?4 OR EXISTS(SELECT 1 FROM {a}workspace_runtime_bindings rb WHERE rb.workspace_id=?1 AND rb.runtime_id=?2 AND rb.revoked_at IS NULL)) AND NOT EXISTS(SELECT 1 FROM {a}worker_registry_projection_removals rm WHERE rm.workspace_id=?1 AND rm.runtime_id=?2 AND rm.worker_id=?3) AND NOT EXISTS(SELECT 1 FROM {a}worker_tombstones wt WHERE wt.workspace_id=?1 AND wt.runtime_id=?2 AND wt.worker_id=?3) AND NOT EXISTS(SELECT 1 FROM {a}worker_removal_operations ro WHERE ro.workspace_id=?1 AND ro.runtime_id=?2 AND ro.worker_id=?3 AND ro.state IN ('executing','failed','succeeded')))") , @@ -12634,6 +12058,13 @@ fn new_runtime_binding_id() -> String { } include!("frozen_legacy_migrations.rs"); +#[derive(Debug, Clone, Copy, PartialEq, Eq)] +enum FrozenWorkspaceRuntimeBindingState { + Configured, + Verified, + Revoked, +} + #[allow(dead_code)] #[derive(Clone)] struct FrozenWorkspaceRuntimeBinding { @@ -12644,7 +12075,7 @@ struct FrozenWorkspaceRuntimeBinding { public_key: String, public_key_fingerprint: String, legacy_counter: u64, - state: WorkspaceRuntimeBindingState, + state: FrozenWorkspaceRuntimeBindingState, authentication_mode: WorkspaceRuntimeAuthenticationMode, workspace_key_id: Option, workspace_key_generation: Option, @@ -12656,9 +12087,9 @@ fn read_frozen_workspace_runtime_binding( row: &rusqlite::Row<'_>, ) -> rusqlite::Result { let state = match row.get::<_, String>(7)?.as_str() { - "configured" => WorkspaceRuntimeBindingState::Configured, - "verified" => WorkspaceRuntimeBindingState::Verified, - "revoked" => WorkspaceRuntimeBindingState::Revoked, + "configured" => FrozenWorkspaceRuntimeBindingState::Configured, + "verified" => FrozenWorkspaceRuntimeBindingState::Verified, + "revoked" => FrozenWorkspaceRuntimeBindingState::Revoked, _ => return Err(rusqlite::Error::InvalidQuery), }; let authentication_mode = match row.get::<_, String>(8)?.as_str() { @@ -12690,6 +12121,160 @@ include!("frozen_repository_key_migration.rs"); include!("frozen_runtime_binding_migration.rs"); include!("legacy_evidence_column_migration.rs"); +// Schema 90 archives authentication conclusions rather than treating them as current authority. +fn migrate_runtime_binding_connections_v89_to_v90(conn: &Connection) -> Result<()> { + if current_schema_version(conn)? != 89 { + return Err(Error::Store( + "Runtime connection migration requires schema 89".into(), + )); + } + let foreign_keys: bool = conn.query_row("PRAGMA foreign_keys", [], |r| r.get(0))?; + let legacy_alter: bool = conn.query_row("PRAGMA legacy_alter_table", [], |r| r.get(0))?; + conn.execute_batch("PRAGMA foreign_keys=OFF; PRAGMA legacy_alter_table=ON;")?; + let result = (|| -> Result<()> { + let tx = rusqlite::Transaction::new_unchecked(conn, TransactionBehavior::Exclusive)?; + // Preserve cross-domain removal fences while rebuilding their referenced table. + let triggers = { + let mut stmt = tx.prepare("SELECT name,sql FROM sqlite_schema WHERE type='trigger'")?; + stmt.query_map([], |r| Ok((r.get::<_, String>(0)?, r.get::<_, String>(1)?)))? + .collect::>>()? + }; + let indexes = { + let mut stmt = tx.prepare("SELECT sql FROM sqlite_schema WHERE type='index' AND tbl_name='workspace_runtime_bindings' AND sql IS NOT NULL")?; + stmt.query_map([], |r| r.get::<_, String>(0))? + .collect::>>()? + }; + for (name, _) in &triggers { + tx.execute_batch(&format!("DROP TRIGGER \"{}\"", name.replace('"', "\"\"")))?; + } + tx.execute_batch(RUNTIME_BINDING_VERIFICATION_ARCHIVE_SCHEMA)?; + tx.execute_batch("INSERT INTO workspace_runtime_binding_verification_archive + SELECT workspace_id,runtime_id,binding_id,state,workspace_key_id, + workspace_public_key_fingerprint,workspace_trust_id,updated_at + FROM workspace_runtime_bindings; + INSERT INTO workspace_runtime_verification_archive SELECT * FROM workspace_runtime_verifications;")?; + tx.execute_batch(RUNTIME_BINDING_CONNECTION_SCHEMA_V90)?; + tx.execute_batch("INSERT INTO workspace_runtime_bindings_v90 + SELECT workspace_id,runtime_id,display_name,base_url,public_key, + public_key_fingerprint,binding_id,authentication_mode,created_at,updated_at,revoked_at + FROM workspace_runtime_bindings; + DROP TABLE workspace_runtime_verifications; + DROP TABLE workspace_runtime_bindings; + ALTER TABLE workspace_runtime_bindings_v90 RENAME TO workspace_runtime_bindings;")?; + for sql in indexes { + tx.execute_batch(&sql)?; + } + for (_, sql) in triggers { + tx.execute_batch(&sql)?; + } + let violation: bool = tx.query_row( + "SELECT EXISTS(SELECT 1 FROM pragma_foreign_key_check)", + [], + |r| r.get(0), + )?; + if violation { + return Err(Error::Store( + "Runtime connection migration foreign-key violation".into(), + )); + } + verify_runtime_binding_connection_schema(&tx)?; + tx.execute("INSERT INTO __yoi_schema_migrations(version,name) VALUES(90,'Runtime binding connection configuration and historical verification archive')", [])?; + tx.commit()?; + Ok(()) + })(); + let restore = conn + .execute_batch(&format!( + "PRAGMA legacy_alter_table={}; PRAGMA foreign_keys={};", + i32::from(legacy_alter), + i32::from(foreign_keys) + )) + .map_err(Error::from); + result.and(restore) +} + +const RUNTIME_BINDING_CONNECTION_SCHEMA_V90: &str = r#" +CREATE TABLE workspace_runtime_bindings_v90 ( + workspace_id TEXT NOT NULL, + runtime_id TEXT NOT NULL, + display_name TEXT NOT NULL, + base_url TEXT NOT NULL, + public_key TEXT NOT NULL, + public_key_fingerprint TEXT NOT NULL, + binding_id TEXT NOT NULL CHECK (length(binding_id) > 0), + authentication_mode TEXT NOT NULL CHECK (authentication_mode IN ('legacy_server_issuer', 'workspace_identity')), + created_at TEXT NOT NULL, + updated_at TEXT NOT NULL, + revoked_at TEXT, + PRIMARY KEY (workspace_id, runtime_id), + UNIQUE (workspace_id, public_key_fingerprint), + FOREIGN KEY(workspace_id) REFERENCES workspaces(workspace_id) ON DELETE RESTRICT +);"#; + +// No binding foreign key: later mutation, reactivation, and administrative removal +// must neither mutate nor erase migrated historical authentication evidence. +const RUNTIME_BINDING_VERIFICATION_ARCHIVE_SCHEMA: &str = r#" +CREATE TABLE workspace_runtime_binding_verification_archive ( + workspace_id TEXT NOT NULL, + runtime_id TEXT NOT NULL, + binding_id TEXT NOT NULL, + state TEXT NOT NULL, + workspace_key_id TEXT, + workspace_public_key_fingerprint TEXT, + workspace_trust_id TEXT, + updated_at TEXT NOT NULL, + PRIMARY KEY (workspace_id, runtime_id, binding_id), + FOREIGN KEY(workspace_id) REFERENCES workspaces(workspace_id) ON DELETE CASCADE +); +CREATE TABLE workspace_runtime_verification_archive ( + workspace_id TEXT NOT NULL, + runtime_id TEXT NOT NULL, + binding_id TEXT NOT NULL, + workspace_key_id TEXT NOT NULL, + workspace_public_key_fingerprint TEXT NOT NULL, + workspace_trust_id TEXT NOT NULL, + runtime_public_key_fingerprint TEXT NOT NULL, + challenge_id TEXT NOT NULL, + state TEXT NOT NULL, + last_outcome TEXT NOT NULL, + verified_at TEXT, + checked_at TEXT NOT NULL, + PRIMARY KEY (workspace_id, runtime_id), + FOREIGN KEY(workspace_id) REFERENCES workspaces(workspace_id) ON DELETE CASCADE +);"#; + +fn verify_runtime_binding_connection_schema(conn: &Connection) -> Result<()> { + let expected = [ + "workspace_id", + "runtime_id", + "display_name", + "base_url", + "public_key", + "public_key_fingerprint", + "binding_id", + "authentication_mode", + "created_at", + "updated_at", + "revoked_at", + ]; + let actual = table_columns(conn, "workspace_runtime_bindings")?; + if actual != expected || table_exists(conn, "workspace_runtime_verifications")? { + return Err(Error::Store( + "Runtime binding does not match connection-only schema".into(), + )); + } + for table in [ + "workspace_runtime_binding_verification_archive", + "workspace_runtime_verification_archive", + ] { + if !table_exists(conn, table)? { + return Err(Error::Store(format!( + "missing historical Runtime verification archive {table}" + ))); + } + } + Ok(()) +} + fn prepare_connection(conn: &Connection) -> Result<()> { prepare_connection_with_secret_source( conn, @@ -12758,6 +12343,9 @@ struct WorkspaceRuntimeBindingV51 { } fn verify_workspace_runtime_verification_schema(conn: &Connection) -> Result<()> { + if current_schema_version(conn)? >= 90 { + return verify_runtime_binding_connection_schema(conn); + } if column_exists(conn, "workspace_runtime_verifications", "binding_id")? { return verify_runtime_binding_content_schema(conn); } @@ -12791,6 +12379,9 @@ fn verify_workspace_runtime_verification_schema(conn: &Connection) -> Result<()> } fn verify_workspace_signing_identity_schema(conn: &Connection) -> Result<()> { + if current_schema_version(conn)? >= 90 { + return verify_runtime_binding_connection_schema(conn); + } if !column_exists(conn, "workspace_signing_identities", "revision")? { return verify_runtime_binding_content_schema(conn); } @@ -12919,6 +12510,9 @@ fn verify_workspace_deletion_schema(conn: &Connection) -> Result<()> { } fn verify_workspace_runtime_binding_schema(conn: &Connection) -> Result<()> { + if current_schema_version(conn)? >= 90 { + return verify_runtime_binding_connection_schema(conn); + } if column_exists(conn, "workspace_runtime_bindings", "binding_id")? { return verify_runtime_binding_content_schema(conn); } @@ -13081,9 +12675,9 @@ fn verify_workspace_runtime_binding_schema(conn: &Connection) -> Result<()> { public_key_fingerprint: row.get(5)?, legacy_counter: row.get(6)?, state: if row.get::<_, Option>(9)?.is_some() { - WorkspaceRuntimeBindingState::Revoked + FrozenWorkspaceRuntimeBindingState::Revoked } else { - WorkspaceRuntimeBindingState::Verified + FrozenWorkspaceRuntimeBindingState::Verified }, authentication_mode: WorkspaceRuntimeAuthenticationMode::LegacyServerIssuer, workspace_key_id: None, @@ -15361,11 +14955,11 @@ mod tests { VALUES ('workspace-a', 'owner', 'Workspace A', 'active', '1', '1'); \ INSERT INTO workspace_runtime_bindings(\ workspace_id, runtime_id, display_name, base_url, public_key, \ - public_key_fingerprint, binding_id, state, authentication_mode, \ + public_key_fingerprint, binding_id, authentication_mode, \ created_at, updated_at\ ) VALUES (\ 'workspace-a', 'runtime-a', 'Runtime A', 'https://runtime.invalid', \ - 'key-a', 'fingerprint-a', 'binding-live', 'verified', 'legacy_server_issuer', '1', '1'\ + 'key-a', 'fingerprint-a', 'binding-live', 'legacy_server_issuer', '1', '1'\ );", )?; Ok(()) @@ -15383,7 +14977,7 @@ mod tests { .unwrap() .expect("Runtime binding must remain"); assert_eq!(binding.binding_id, "binding-live"); - assert_eq!(binding.state, WorkspaceRuntimeBindingState::Verified); + assert!(binding.revoked_at.is_none()); } @@ -15541,11 +15135,11 @@ mod tests { VALUES ('workspace-b', 'owner-b', 'Workspace B', 'active', '1', '1'); \ INSERT INTO workspace_runtime_bindings(\ workspace_id, runtime_id, display_name, base_url, public_key, \ - public_key_fingerprint, binding_id, state, authentication_mode, \ + public_key_fingerprint, binding_id, authentication_mode, \ created_at, updated_at\ ) VALUES (\ 'workspace-b', 'runtime-a', 'Runtime A', 'https://runtime.invalid', \ - 'key-a', 'fingerprint-b', 'binding-other', 'verified', 'legacy_server_issuer', '1', '1'\ + 'key-a', 'fingerprint-b', 'binding-other', 'legacy_server_issuer', '1', '1'\ );", )?; Ok(()) @@ -15610,11 +15204,11 @@ mod tests { let blocked_insert = competing_conn.execute( "INSERT INTO workspace_runtime_bindings(\ workspace_id, runtime_id, display_name, base_url, public_key, \ - public_key_fingerprint, binding_id, state, authentication_mode, \ + public_key_fingerprint, binding_id, authentication_mode, \ created_at, updated_at\ ) VALUES (\ 'workspace-b', 'runtime-a', 'Runtime A', 'https://runtime.invalid', \ - 'key-a', 'fingerprint-b', 'binding-other', 'verified', 'legacy_server_issuer', '1', '1'\ + 'key-a', 'fingerprint-b', 'binding-other', 'legacy_server_issuer', '1', '1'\ )", [], ); @@ -16650,6 +16244,10 @@ CREATE UNIQUE INDEX ticket_current_worker_role_idx version: 89, name: "descriptive legacy evidence columns".to_string() }, + WorkspaceSchemaMigrationStep { + version: 90, + name: "Runtime binding connection configuration and historical verification archive".to_string() + }, ] ); @@ -16749,6 +16347,7 @@ CREATE UNIQUE INDEX ticket_current_worker_role_idx (87, "Repository SSH key identities and encrypted operation objects".to_string()), (88, "Runtime binding identities and retention content contracts".to_string()), (89, "descriptive legacy evidence columns".to_string()), +(90, "Runtime binding connection configuration and historical verification archive".to_string()), ] ); assert!(!table_exists(conn, "trusted_runtime_records")?); @@ -16762,17 +16361,17 @@ CREATE UNIQUE INDEX ticket_current_worker_role_idx ); assert_eq!( conn.query_row( - "SELECT state || ':' || authentication_mode + "SELECT authentication_mode FROM workspace_runtime_bindings WHERE workspace_id='workspace-a' AND runtime_id='shared'", [], |row| row.get::<_, String>(0), )?, - "configured:workspace_identity" + "workspace_identity" ); assert_eq!( conn.query_row( - "SELECT COUNT(*) FROM workspace_runtime_bindings + "SELECT COUNT(*) FROM workspace_runtime_binding_verification_archive WHERE workspace_id='workspace-a' AND runtime_id='shared' AND workspace_key_id IS NOT NULL AND length(binding_id) > 0 @@ -17550,11 +17149,7 @@ CREATE UNIQUE INDEX ticket_current_worker_role_idx public_key: identity.public_key.clone(), public_key_fingerprint: String::new(), binding_id: String::new(), - state: WorkspaceRuntimeBindingState::Verified, authentication_mode: WorkspaceRuntimeAuthenticationMode::WorkspaceIdentity, - workspace_key_id: Some(format!("WK-{}", &workspace_id["workspace-".len()..])), - workspace_trust_id: Some("trust-fixture".into()), - workspace_public_key_fingerprint: Some("sha256:key".into()), created_at: "1".to_string(), updated_at: "1".to_string(), revoked_at: None, @@ -17672,7 +17267,7 @@ CREATE UNIQUE INDEX ticket_current_worker_role_idx } #[test] - fn workspace_runtime_verification_is_key_bound_and_restart_safe() { + fn workspace_runtime_connection_persists_without_authentication_evidence() { let temp = tempfile::tempdir().unwrap(); let path = temp.path().join("server.db"); let store = SqliteWorkspaceStore::open(&path).unwrap(); @@ -17706,11 +17301,7 @@ CREATE UNIQUE INDEX ticket_current_worker_role_idx public_key: runtime_identity.public_key, public_key_fingerprint: String::new(), binding_id: String::new(), - state: WorkspaceRuntimeBindingState::Configured, authentication_mode: WorkspaceRuntimeAuthenticationMode::WorkspaceIdentity, - workspace_key_id: Some("WK-a".to_string()), - workspace_trust_id: Some("trust-fixture".into()), - workspace_public_key_fingerprint: Some("sha256:key".into()), created_at: "1".to_string(), updated_at: "1".to_string(), revoked_at: None, @@ -17722,106 +17313,14 @@ CREATE UNIQUE INDEX ticket_current_worker_role_idx .get_workspace_runtime_binding("workspace-a", "runtime-a") .unwrap() .unwrap(); - let evidence = WorkspaceRuntimeVerificationEvidence { - workspace_id: "workspace-a".to_string(), - runtime_id: "runtime-a".to_string(), - binding_id: persisted.binding_id.clone(), - workspace_key_id: "WK-a".to_string(), - workspace_public_key_fingerprint: "sha256:key".into(), - workspace_trust_id: "trust-fixture".into(), - runtime_public_key_fingerprint: persisted.public_key_fingerprint.clone(), - - challenge_id: "challenge-a".to_string(), - state: "verified".to_string(), - last_outcome: "verified".to_string(), - verified_at: Some("2".to_string()), - checked_at: "2".to_string(), - }; - store - .record_workspace_runtime_verification_attempt(&evidence) - .unwrap(); - let verified = store - .complete_workspace_runtime_verification(&evidence) - .unwrap(); - assert_eq!(verified.state, WorkspaceRuntimeBindingState::Verified); - let pending_retry = WorkspaceRuntimeVerificationEvidence { - state: "pending".to_string(), - last_outcome: "challenge_issued".to_string(), - verified_at: None, - checked_at: "3".to_string(), - ..evidence.clone() - }; - store - .record_workspace_runtime_verification_attempt(&pending_retry) - .unwrap(); - let retained = store - .get_workspace_runtime_verification("workspace-a", "runtime-a") - .unwrap() - .unwrap(); - assert_eq!(retained.state, "verified"); - assert_eq!(retained.verified_at.as_deref(), Some("2")); - assert_eq!(retained.last_outcome, "challenge_issued"); - assert!( - !store - .workspace_runtime_verification_matches(&verified, "sha256:key", "trust-fixture") - .unwrap() - ); - store - .complete_workspace_runtime_verification(&evidence) - .unwrap(); - let newer_pending = WorkspaceRuntimeVerificationEvidence { - challenge_id: "challenge-b".to_string(), - state: "pending".to_string(), - last_outcome: "challenge_issued".to_string(), - verified_at: None, - checked_at: "4".to_string(), - ..evidence.clone() - }; - store - .record_workspace_runtime_verification_attempt(&newer_pending) - .unwrap(); - let newer_verified = WorkspaceRuntimeVerificationEvidence { - state: "verified".to_string(), - last_outcome: "verified".to_string(), - verified_at: Some("5".to_string()), - checked_at: "5".to_string(), - ..newer_pending - }; - store - .complete_workspace_runtime_verification(&newer_verified) - .unwrap(); - assert!( - !store - .record_workspace_runtime_verification_outcome_if_current( - &pending_retry, - "failed", - "verification_failed", - "6", - ) - .unwrap() - ); - assert_eq!( - store - .get_workspace_runtime_verification("workspace-a", "runtime-a") - .unwrap(), - Some(newer_verified.clone()) - ); + assert!(store.workspace_runtime_binding_matches(&persisted).unwrap()); drop(store); - let reopened = SqliteWorkspaceStore::open(&path).unwrap(); - assert_eq!( - reopened - .get_workspace_runtime_verification("workspace-a", "runtime-a") - .unwrap(), - Some(newer_verified) - ); assert_eq!( reopened .get_workspace_runtime_binding("workspace-a", "runtime-a") - .unwrap() - .unwrap() - .state, - WorkspaceRuntimeBindingState::Verified + .unwrap(), + Some(persisted) ); } @@ -17829,22 +17328,15 @@ CREATE UNIQUE INDEX ticket_current_worker_role_idx fn schema_v55_migrates_runtime_verification_table_atomically() { let temp = tempfile::tempdir().unwrap(); let path = temp.path().join("server.db"); - let store = SqliteWorkspaceStore::open(&path).unwrap(); - store - .with_conn(|conn| { - conn.execute_batch( - "DROP TABLE workspace_runtime_verifications; - DELETE FROM __yoi_schema_migrations; - INSERT INTO __yoi_schema_migrations(version, name) - VALUES (55, 'Workspace Runtime binding state and identity mode');", - )?; - Ok(()) - }) - .unwrap(); - drop(store); - + prepare_schema_v50(&path, Some("workspace-a")); let conn = Connection::open(&path).unwrap(); configure_sqlite(&conn).unwrap(); + for migration in MIGRATIONS + .iter() + .filter(|migration| migration.version > 50 && migration.version <= 55) + { + (migration.apply)(&conn).unwrap(); + } assert_eq!(current_schema_version(&conn).unwrap(), 55); migrate_workspace_runtime_verification_v55_to_v56(&conn).unwrap(); drop(conn); @@ -18101,11 +17593,7 @@ CREATE UNIQUE INDEX ticket_current_worker_role_idx public_key, public_key_fingerprint: String::new(), binding_id: String::new(), - state: WorkspaceRuntimeBindingState::Configured, authentication_mode: WorkspaceRuntimeAuthenticationMode::WorkspaceIdentity, - workspace_key_id: Some("WK-a".to_string()), - workspace_trust_id: Some("trust-fixture".into()), - workspace_public_key_fingerprint: Some("sha256:key".into()), created_at: at.to_string(), updated_at: at.to_string(), revoked_at: None, @@ -18120,19 +17608,11 @@ CREATE UNIQUE INDEX ticket_current_worker_role_idx .unwrap(); assert_eq!(created, WorkspaceRuntimeBindingMutation::Created); assert!(!created_binding.binding_id.is_empty()); - assert_eq!( - created_binding.state, - WorkspaceRuntimeBindingState::Configured - ); + assert!(created_binding.revoked_at.is_none()); assert_eq!( created_binding.authentication_mode, WorkspaceRuntimeAuthenticationMode::WorkspaceIdentity ); - assert_eq!(created_binding.workspace_key_id.as_deref(), Some("WK-a")); - assert_eq!( - created_binding.workspace_trust_id.as_deref(), - Some("trust-fixture") - ); let (replayed, replayed_binding) = store .put_workspace_runtime_binding_key(binding(first.public_key, "2"), None, "owner") .unwrap(); @@ -18171,29 +17651,6 @@ CREATE UNIQUE INDEX ticket_current_worker_role_idx .unwrap(); assert_eq!(replaced, WorkspaceRuntimeBindingMutation::Replaced); assert_ne!(replaced_binding.binding_id, created_binding.binding_id); - store - .record_workspace_runtime_verification_attempt(&WorkspaceRuntimeVerificationEvidence { - workspace_id: "workspace-a".to_string(), - runtime_id: "runtime-a".to_string(), - binding_id: replaced_binding.binding_id.clone(), - workspace_key_id: "WK-a".to_string(), - workspace_public_key_fingerprint: "sha256:key".into(), - workspace_trust_id: "trust-fixture".into(), - runtime_public_key_fingerprint: replaced_binding.public_key_fingerprint.clone(), - - challenge_id: "challenge-a".to_string(), - state: "failed".to_string(), - last_outcome: "connectivity_failed".to_string(), - verified_at: None, - checked_at: "3".to_string(), - }) - .unwrap(); - assert!( - store - .get_workspace_runtime_verification("workspace-a", "runtime-a") - .unwrap() - .is_some() - ); let (revoked, revoked_binding) = store .revoke_workspace_runtime_binding_key( "workspace-a", @@ -18206,12 +17663,15 @@ CREATE UNIQUE INDEX ticket_current_worker_role_idx assert_eq!(revoked, WorkspaceRuntimeBindingMutation::Revoked); assert_ne!(revoked_binding.binding_id, replaced_binding.binding_id); assert_eq!(revoked_binding.revoked_at.as_deref(), Some("4")); - assert_eq!(revoked_binding.state, WorkspaceRuntimeBindingState::Revoked); - assert_eq!( - store - .get_workspace_runtime_verification("workspace-a", "runtime-a") - .unwrap(), - None + assert!( + !store + .workspace_runtime_binding_matches(&replaced_binding) + .unwrap() + ); + assert!( + !store + .workspace_runtime_binding_matches(&revoked_binding) + .unwrap() ); let (reactivated, reactivated_binding) = store .put_workspace_runtime_binding_key( @@ -18222,10 +17682,7 @@ CREATE UNIQUE INDEX ticket_current_worker_role_idx .unwrap(); assert_eq!(reactivated, WorkspaceRuntimeBindingMutation::Reactivated); assert_ne!(reactivated_binding.binding_id, revoked_binding.binding_id); - assert_eq!( - reactivated_binding.state, - WorkspaceRuntimeBindingState::Configured - ); + assert!(reactivated_binding.revoked_at.is_none()); let mut duplicate = binding(second.public_key, "6"); duplicate.runtime_id = "runtime-b".to_string(); @@ -18276,11 +17733,7 @@ CREATE UNIQUE INDEX ticket_current_worker_role_idx public_key, public_key_fingerprint: String::new(), binding_id: String::new(), - state: WorkspaceRuntimeBindingState::Verified, authentication_mode: WorkspaceRuntimeAuthenticationMode::LegacyServerIssuer, - workspace_key_id: None, - workspace_trust_id: None, - workspace_public_key_fingerprint: None, created_at: "1".to_string(), updated_at: "1".to_string(), revoked_at: None, diff --git a/crates/workspace-server/src/workspace_deletion.rs b/crates/workspace-server/src/workspace_deletion.rs index dfc1b70da..5dcf9d009 100644 --- a/crates/workspace-server/src/workspace_deletion.rs +++ b/crates/workspace-server/src/workspace_deletion.rs @@ -99,7 +99,8 @@ const WORKSPACE_DELETION_PURGE_TABLES: &[&str] = &[ "workspace_resource_keys", "workspace_runtime_binding_audit", "workspace_runtime_bindings", - "workspace_runtime_verifications", + "workspace_runtime_binding_verification_archive", + "workspace_runtime_verification_archive", "workspace_signing_identities", "workspace_signing_identity_audit", "workspace_signing_identity_provisioning_operations", diff --git a/crates/workspace-server/src/workspace_signing_identity.rs b/crates/workspace-server/src/workspace_signing_identity.rs index bf58d407d..f724f2ca1 100644 --- a/crates/workspace-server/src/workspace_signing_identity.rs +++ b/crates/workspace-server/src/workspace_signing_identity.rs @@ -228,36 +228,7 @@ impl WorkspaceSigningIdentityService { } pub fn sign(&self, workspace_id: &str, payload: &[u8]) -> Result> { - let identity = self.get_validated(workspace_id)?; - if identity.state != "active" { - return Err(identity_error( - "workspace_signing_identity_not_provisioned", - "Workspace signing identity is not provisioned", - )); - } - let material = self - .materials - .load(&identity.private_material_ref)? - .ok_or_else(|| { - identity_error( - "workspace_signing_identity_material_missing", - "Workspace signing private material is missing", - ) - })?; - let fingerprint = identity.public_key_fingerprint.as_deref().ok_or_else(|| { - identity_error( - "workspace_signing_identity_material_mismatch", - "Workspace signing public key fingerprint is missing", - ) - })?; - let signing_key = material.signing_key(workspace_id, &identity.key_id)?; - let public_key = encode_public_key(signing_key.public_key().as_ref()); - if public_key_fingerprint(&public_key)? != fingerprint { - return Err(identity_error( - "workspace_signing_identity_material_mismatch", - "Workspace signing private material does not match public metadata", - )); - } + let (_, signing_key) = self.select_signing_key(workspace_id)?; Ok(signing_key.sign(payload).as_ref().to_vec()) } @@ -266,9 +237,20 @@ impl WorkspaceSigningIdentityService { workspace_id: &str, claims: &WorkspaceCapabilityClaims, ) -> Result { + let (identity, signing_key) = self.select_signing_key(workspace_id)?; + if claims.issuer_workspace_id != identity.workspace_id + || claims.issuer_key_id != identity.key_id + || Some(claims.issuer_public_key_fingerprint.as_str()) + != identity.public_key_fingerprint.as_deref() + { + return Err(identity_error( + "workspace_capability_identity_mismatch", + "Workspace capability claims do not match the current signing identity", + )); + } let input = workspace_capability_signing_input(claims).map_err(capability_error)?; - let signature = self.sign(workspace_id, input.bytes())?; - assemble_workspace_capability_token(input, &signature).map_err(capability_error) + let signature = signing_key.sign(input.bytes()); + assemble_workspace_capability_token(input, signature.as_ref()).map_err(capability_error) } pub fn delete_material(&self, workspace_id: &str) -> Result<()> { @@ -306,7 +288,40 @@ impl WorkspaceSigningIdentityService { }) } + fn select_signing_key( + &self, + workspace_id: &str, + ) -> Result<( + WorkspaceSigningIdentityRecord, + ring::signature::Ed25519KeyPair, + )> { + let identity = self + .store + .get_workspace_signing_identity(workspace_id)? + .ok_or_else(|| { + identity_error( + "workspace_signing_identity_metadata_missing", + "Workspace signing identity metadata is missing", + ) + })?; + if identity.state != "active" { + return Err(identity_error( + "workspace_signing_identity_not_provisioned", + "Workspace signing identity is not provisioned", + )); + } + let signing_key = self.validated_signing_key(&identity)?; + Ok((identity, signing_key)) + } + fn validate_active_material(&self, identity: &WorkspaceSigningIdentityRecord) -> Result<()> { + self.validated_signing_key(identity).map(|_| ()) + } + + fn validated_signing_key( + &self, + identity: &WorkspaceSigningIdentityRecord, + ) -> Result { let material = self .materials .load(&identity.private_material_ref)? @@ -316,8 +331,8 @@ impl WorkspaceSigningIdentityService { "Workspace signing private material is missing", ) })?; - let public_key = - material.validate_and_public_key(&identity.workspace_id, &identity.key_id)?; + let signing_key = material.signing_key(&identity.workspace_id, &identity.key_id)?; + let public_key = encode_public_key(signing_key.public_key().as_ref()); let fingerprint = public_key_fingerprint(&public_key)?; if identity.public_key.as_deref() != Some(public_key.as_str()) || identity.public_key_fingerprint.as_deref() != Some(fingerprint.as_str()) @@ -327,7 +342,7 @@ impl WorkspaceSigningIdentityService { "Workspace signing private material does not match public metadata", )); } - Ok(()) + Ok(signing_key) } } @@ -843,6 +858,14 @@ mod tests { loads: std::sync::atomic::AtomicUsize::new(0), }), ); + let selected_signature = swapping_service.sign("workspace-1", payload).unwrap(); + ring::signature::UnparsedPublicKey::new( + &ring::signature::ED25519, + worker_runtime::auth::decode_public_key(provisioned.public_key.as_deref().unwrap()) + .unwrap(), + ) + .verify(payload, &selected_signature) + .unwrap(); assert!(matches!( swapping_service.sign("workspace-1", payload), Err(Error::WorkspaceSigningIdentity { ref code, .. }) diff --git a/docs/development/dogfooding.md b/docs/development/dogfooding.md index baec4ea0a..4bdad7c14 100644 --- a/docs/development/dogfooding.md +++ b/docs/development/dogfooding.md @@ -8,11 +8,11 @@ Never use the live dogfood Server or Runtime as the first startup test for a new 1. Build the production entrypoints: `cargo build -p worker-runtime --bin yoi-runtime -p yoi-workspace-server --bin yoi-server`. 2. Run the focused and dependent tests for the changed contracts, followed by workspace-root `cargo check`, `cargo fmt --all -- --check`, and `git diff --check HEAD`. -3. Exercise the provisioning and operational checks in [Workspace ↔ Runtime authentication](server-runtime-auth.md) against isolated Server DB, Runtime data, and ports. The Workspace owner must create the binding, the Runtime operator must install the Workspace issuer bundle, and the challenge proof must become verified. +3. Exercise the provisioning and operational checks in [Workspace ↔ Runtime authentication](server-runtime-auth.md) against isolated Server DB, Runtime data, and ports. The Workspace owner must configure the Runtime endpoint and pinned Runtime key, the Runtime operator must install the Workspace issuer bundle, and actual signed requests must authenticate. A connection-test result describes only that attempt; no stored verification state is a readiness prerequisite. 4. Have an external supervisor or operator restart Server and Runtime from the same validated source commit/build artifacts. A Worker hosted by the target Runtime must never terminate its own Runtime. 5. Verify post-restart readiness through the Workspace Runtime projection, ping, Worker list/create, protocol subscription, and a Runtime-to-Server source-proof operation before treating the environment as healthy. -The former isolated startup shell harness depended on removed Server-global trust commands and is intentionally not a fallback smoke path. New automated startup coverage must provision the same Workspace-scoped binding and challenge authority used by production rather than recreating global trust or seeding private authority directly. +The former isolated startup shell harness depended on removed Server-global trust commands and is intentionally not a fallback smoke path. New automated startup coverage must configure the same Workspace-scoped connection and explicit Runtime-side Workspace trust used by production, then exercise per-request authentication rather than recreating global trust or seeding a past verification result. ## What to record @@ -25,7 +25,7 @@ A report is useful when it explains: - what design boundary was missing - what evidence was observed -For a Remote Runtime rollout, also record the source commit, binary artifact digest, Server schema version, Runtime `binding_id`, Workspace `key_id` and `workspace_trust_id`, and typed HTTP/WebSocket outcomes. A successful document response does not outweigh visible UI, console, or API errors. +For a Remote Runtime rollout, also record the source commit, binary artifact digest, Server schema version, Runtime `binding_id` and pinned key fingerprint, the current Workspace `key_id` and public fingerprint, the Runtime-side trust registration/revocation, and typed HTTP/WebSocket outcomes. Do not copy a Runtime trust ID into the Server connection configuration or treat a successful connection test as persisted authentication. A successful document response does not outweigh visible UI, console, or API errors. ## Runtime command caveat diff --git a/docs/development/server-runtime-auth.md b/docs/development/server-runtime-auth.md index 77719d3c2..6058836eb 100644 --- a/docs/development/server-runtime-auth.md +++ b/docs/development/server-runtime-auth.md @@ -1,28 +1,25 @@ # Workspace ↔ Runtime 認証 Yoi の Remote Runtime 認証は Workspace ごとの署名 identity を authority とする。 +認証の成否は実際の HTTP / WebSocket request ごとに決まり、保存済みの「verified」状態や過去の接続テスト結果を前提にしない。 Server-global な署名鍵や Runtime 側の trusted-Server catalog は使わない。 ## Authority -- Server DB は Workspace ごとの signing identity と Runtime binding を保持する。 -- Runtime は `trust-workspace` で受理した `WorkspacePublicIdentityBundle` を保持する。 +- Server DB は Workspace ごとの signing identity と Runtime 接続設定（binding）を保持する。 +- binding は endpoint、固定した Runtime public key/fingerprint、`binding_id`、authentication mode、管理者による失効を保持する。Workspace key や Runtime 側の trust ID のコピー、認証成功状態は保持しない。 +- Runtime は管理者が `trust-workspace` で受理した `WorkspacePublicIdentityBundle` を保持する。 - bundle は `workspace_id`、Workspace `key_id`、Workspace public key/fingerprint、Backend URL を固定する。 -- Server → Runtime の各 HTTP / WebSocket request は、対象 Workspace の signing identity で短命な capability token を発行する。 -- Runtime は request method、`path_and_query`、body digest、permission、Workspace、Runtime、key ID、現在の trust ID、expiry、JTI を検証する。 -- Runtime → Server の source proof は Runtime identity で署名し、対象 Workspace と bundle の Backend URL を audience に固定する。 +- Server → Runtime の各 HTTP / WebSocket request は、その時点の Workspace signing identity で短命な capability token を発行する。署名時には claims の Workspace・key ID・fingerprint が選択した identity と一致することを検証する。秘密鍵materialがない場合は fallback せず拒否する。 +- Runtime は現在の明示的な Workspace trust、署名、request method、`path_and_query`、body digest、permission、Workspace、Runtime、必要に応じ Worker、key ID/fingerprint、expiry、JTI を検証する。過去の handshake や verification record は不要である。 +- Runtime → Server の source proof は Runtime identity で署名し、対象 Workspace と bundle の Backend URL を audience に固定する。この proof の契約は変わらない。 - Server は現在の Workspace Runtime binding、Runtime public key、Backend public URL、request target、body digest、permission、expiry、replay state を検証する。 旧 Server identity/trust 管理 command と旧 Runtime-side Server trust command、旧 Runtime auth key flags は廃止済みである。これらに相当する Server-global trust を fallback として使ってはならない。 ## Provisioning -The enrollment input includes the exact Runtime-issued `workspace_trust_id` from -`trust-workspace add/show`. Export the current Workspace public bundle first, -accept it on the Runtime, then register the Runtime public bundle, endpoint and -trust ID. Server verification binds that enrollment ID; it must never invent an -ID from the key fingerprint or assume an initial counter value. Re-enrollment -requires registering the newly issued trust ID, even when the key is unchanged. +Runtime 側の Workspace trust 登録と Server 側の Runtime 接続設定は独立した管理操作である。Runtime が発行する `trust_id` を Server にコピーする必要はない。 1. Runtime identity を初期化する。 @@ -31,17 +28,16 @@ requires registering the newly issued trust ID, even when the key is unchanged. yoi-runtime identity show ``` -2. The Workspace owner exports the current Workspace issuer public bundle from Workspace identity settings. -3. The operator accepts that bundle on the Runtime and reads the issued trust ID. +2. Workspace owner が Workspace identity settings から現在の Workspace issuer public bundle を export する。 +3. Runtime 管理者が bundle を確認し、Runtime に明示的に登録する。 ```sh yoi-runtime trust-workspace add --bundle yoi-runtime trust-workspace show --workspace-id ``` -4. The owner registers the Runtime public bundle, endpoint and exact `workspace_trust_id` in Settings > Runtimes. -5. The Server sends a signed challenge bound to that trust ID, the Workspace key fingerprint and the binding ID, then verifies the Runtime proof and acknowledgement. -6. Normal Workspace-signed requests become available only after the verified binding is committed. Revoking and re-enrolling the same key still requires the newly issued Runtime trust ID. +4. Workspace owner が Settings > Runtimes に Runtime public bundle と endpoint を登録する。接続設定には Runtime key を固定するが、Workspace key や Runtime の trust ID は登録しない。 +5. 接続テストまたは通常操作を実行する。実際の Workspace-signed request が Runtime の現在の trust と照合される。接続テスト結果はその試行の診断であり、永続的な認証状態を作らず、以後の request の成功を保証しない。 同じ Runtime identity は異なる Workspace から独立して信頼できる。trust record、replay protection、binding、失効はすべて Workspace scope で評価する。 @@ -55,29 +51,41 @@ requires registering the newly issued trust ID, even when the key is unchanged. - 最大 8 MiB - 最大 4,096 records - exact Workspace / Runtime identity -- `key_id` と public key fingerprint、opaque `trust_id` +- `key_id` と public key fingerprint、管理 record の opaque `trust_id` - normalized Backend URL -- 同じ `key_id` に異なる鍵materialを割り当てないこと。実際の置換・失効は新しい `trust_id` を発行する +- 同じ `key_id` に異なる鍵materialを割り当てないこと。実際の置換・失効は新しい `trust_id` を発行するが、この ID は request token や Server binding に含めない - list は `offset` / `limit` 必須で、1 page 最大 100 records ## Local token `--local-token` は明示的な local Runtime 呼び出し専用であり、Remote Workspace binding の代替ではない。Workspace issuer auth が有効な Remote Runtime request は Workspace capability token を使う。 -## Rotation と失効 +## Key 変更と失効 -Workspace signing key または Runtime key の変更は、現在 binding を置き換える明示的な provisioning 操作として行う。古い key/trust ID、古い Runtime key、revoked binding、失効済み token、replayed JTI は即時拒否する。 +Workspace signing identity は各 request の発行時に現在の authority から選択する。Workspace public key が変更された場合、Runtime 管理者が現在の public bundle を明示的に `trust-workspace replace` で受理する必要がある。Workspace key のコピーを更新するためだけに Server の Runtime 接続設定を置き換える操作は不要である。これは Workspace key rotation の新しい production API を定義するものではない。 -Server の Runtime cache は現在の persisted binding 全体と照合する。endpoint、Runtime public key/fingerprint、`binding_id`、Workspace `key_id`、`workspace_trust_id` の変更を検知した場合、stale client を利用しない。 +Runtime 管理者は `trust-workspace revoke` で Workspace trust を失効できる。Runtime は現在の trust と一致しない key、失効済み trust、expired token、replayed JTI を拒否する。同じ key の再登録も管理者の明示的な操作で行い、request の自己申告による trust 登録は行わない。 + +Runtime key や endpoint の変更は Server の接続設定を明示的に置き換える。Server の Runtime cache と authorizer は現在の接続設定（endpoint、Runtime public key/fingerprint、`binding_id`、authentication mode、管理者による失効）と照合し、replaced/revoked binding を使う stale client を拒否する。接続設定の一致は送信先の選択であって、認証成功の証拠ではない。 + +接続先が到達不能・認証失敗でも、owner は観測した `binding_id` を指定して接続設定を修復できる。取得できた active Worker は置換を遮断するが、Worker 一覧の取得失敗を「idle の証明」とは扱わず、認証成功を修復の前提にもしない。接続設定の置換は Worker の停止・削除ではなく、Runtime 削除の資源確認とは別の操作である。 + +## 保存形式の移行 + +Server schema90 は既存 binding の endpoint、Runtime key、binding ID、明示的失効を保持し、古い認証状態と検証記録を履歴専用 archive に移す。履歴は要求の認可には参照しない。既存 Workspace identity や接続設定の作り直しは不要である。 + +Runtime の旧形式の Workspace trust は、検証済みの公開鍵と管理者の Active/Revoked 設定を保持して読み込む。形式の移行だけでは失効させず、すでに保存されている Revoked 設定を自動で解除することもしない。旧 verification file は認可に使用しない。消費済み token の replay 記録は引き継ぎ、同じ鍵の再登録でリセットしない。 + +API と capability claims の旧 handshake 契約は維持しないため、Backend・Runtime・Web は対応する版へ更新する。接続テストは更新後の実認証を確認するものであり、登録を有効化する手順ではない。 ## 運用確認 -Remote Runtime を有効化した後は次を確認する。 +Remote Runtime を設定した後は次を確認する。 -1. `yoi-runtime trust-workspace show --workspace-id ` が期待する bundle を表示する。 -2. Workspace Settings の Runtime binding が `verified` で、現在の `binding_id`・`workspace_key_id`・`workspace_trust_id` と verification evidence を表示する。 -3. Runtime ping、Worker list/create、`worker.protocol` subscription が Workspace-signed token で成功する。 -4. wrong Workspace、wrong Runtime、wrong target/body、expired token、revoked/replaced binding、replayed JTI が拒否される。 +1. `yoi-runtime trust-workspace show --workspace-id ` が期待する active bundle を表示する。 +2. Workspace Settings の Runtime 接続設定が期待する endpoint と固定 Runtime key を示し、失効していない。 +3. 接続テストの今回の結果を確認し、Runtime ping、Worker list/create、`worker.protocol` subscription が実際の Workspace-signed request で成功することを確認する。保存済み「verified」表示や verification evidence は確認項目ではない。 +4. wrong Workspace、wrong Runtime、wrong target/body、expired token、revoked Runtime trust、replaced/revoked binding、replayed JTI が拒否される。 5. Runtime → Server source proof が configured Backend public URL audience と一致し、spoofed headers だけでは認証されない。 Server / Runtime の再起動は live reload ではない authority 変更を反映するときだけ、通常の運用権限と migration gate に従って行う。実行中プロセスを開発 Worker が無断で停止してはならない。 diff --git a/openapi/server-api.json b/openapi/server-api.json index 53424c5a8..acb68c9ac 100644 --- a/openapi/server-api.json +++ b/openapi/server-api.json @@ -1530,15 +1530,10 @@ }, "public_bundle": { "$ref": "#/components/schemas/RuntimePublicIdentityBundle" - }, - "workspace_trust_id": { - "description": "Exact Runtime-issued enrollment identity from `trust-workspace add/show`.\nA key fingerprint alone cannot distinguish revocation and re-enrollment.", - "type": "string" } }, "required": [ "public_bundle", - "workspace_trust_id", "endpoint" ], "type": "object" @@ -7840,15 +7835,6 @@ ], "type": "object" }, - "RuntimeConnectionDisplayState": { - "enum": [ - "configured", - "verified", - "unavailable", - "revoked" - ], - "type": "string" - }, "RuntimeConnectionTestFailureKind": { "enum": [ "authentication", @@ -7865,6 +7851,7 @@ }, "RuntimeConnectionTestResponse": { "additionalProperties": false, + "description": "Observation of an actual signed ping, not a grant or cached approval.", "properties": { "actual_protocol_version": { "format": "uint32", @@ -7881,9 +7868,6 @@ "checked_at": { "type": "string" }, - "connection_state": { - "$ref": "#/components/schemas/RuntimeConnectionDisplayState" - }, "diagnostics": { "default": [], "items": { @@ -7913,16 +7897,6 @@ "status": { "$ref": "#/components/schemas/RuntimeConnectionTestStatus" }, - "verification": { - "anyOf": [ - { - "$ref": "#/components/schemas/RuntimeVerificationEvidenceSummary" - }, - { - "type": "null" - } - ] - }, "workspace_id": { "type": "string" } @@ -7931,7 +7905,6 @@ "workspace_id", "runtime_id", "binding_id", - "connection_state", "checked_at", "status", "expected_protocol_version" @@ -8385,57 +8358,6 @@ ], "type": "string" }, - "RuntimeVerificationEvidenceSummary": { - "additionalProperties": false, - "properties": { - "binding_id": { - "type": "string" - }, - "last_checked_at": { - "type": "string" - }, - "last_outcome": { - "$ref": "#/components/schemas/RuntimeVerificationOutcome" - }, - "runtime_public_key_fingerprint": { - "type": "string" - }, - "verified_at": { - "type": [ - "string", - "null" - ] - }, - "workspace_key_id": { - "type": "string" - }, - "workspace_public_key_fingerprint": { - "type": "string" - }, - "workspace_trust_id": { - "type": "string" - } - }, - "required": [ - "last_checked_at", - "last_outcome", - "binding_id", - "workspace_key_id", - "workspace_public_key_fingerprint", - "workspace_trust_id", - "runtime_public_key_fingerprint" - ], - "type": "object" - }, - "RuntimeVerificationOutcome": { - "enum": [ - "verified", - "challenge_issued", - "verification_failed", - "connectivity_failed" - ], - "type": "string" - }, "RuntimeWorkerCompletionsRequest": { "additionalProperties": false, "properties": { @@ -15335,43 +15257,14 @@ ], "type": "object" }, - "WorkspaceRuntimeBindingState": { - "enum": [ - "configured", - "verified", - "revoked" - ], - "type": "string" - }, "WorkspaceRuntimeBindingSummary": { "additionalProperties": false, + "description": "Binding identity and explicit revocation only. Each signed request authenticates\nusing the current Workspace signing identity, not a stored approval.", "properties": { "binding_id": { "type": "string" }, - "connection_state": { - "$ref": "#/components/schemas/RuntimeConnectionDisplayState" - }, - "state": { - "$ref": "#/components/schemas/WorkspaceRuntimeBindingState" - }, - "verification": { - "anyOf": [ - { - "$ref": "#/components/schemas/RuntimeVerificationEvidenceSummary" - }, - { - "type": "null" - } - ] - }, - "workspace_key_id": { - "type": [ - "string", - "null" - ] - }, - "workspace_trust_id": { + "revoked_at": { "type": [ "string", "null" @@ -15379,8 +15272,6 @@ } }, "required": [ - "state", - "connection_state", "binding_id" ], "type": "object" @@ -16135,7 +16026,7 @@ "info": { "title": "Yoi Server API", "version": "0.1.0", - "x-yoi-source-digest": "sha256:fc25104c077ab8c5c43f77611f505171585d2d40fd54c6538774ac17aae6cce3" + "x-yoi-source-digest": "sha256:a23a7b9989d37fd08459929a4e726396a33334489207799e02d45302e8d4c238" }, "openapi": "3.1.0", "paths": { diff --git a/tools/state-contract-name-exceptions.json b/tools/state-contract-name-exceptions.json index 53ae6cd1d..d8d6714d4 100644 --- a/tools/state-contract-name-exceptions.json +++ b/tools/state-contract-name-exceptions.json @@ -673,39 +673,15 @@ { "path": "crates/worker-runtime/src/workspace_issuer.rs", "kind": "saved_data", - "reason": "LegacyWorkspaceIssuerTrustRecord/LegacyVerificationDocumentの保存済みenrollment・verification keyとfixture。旧trustをrevokedへ移行し、旧verification bytesをarchiveするがlive recordsを空にするconsumerがある。削除すると既存記録を読めないか、旧capabilityを再承認なしに復活させない移行検証を失う。", + "reason": "LegacyWorkspaceIssuerTrustRecordの保存済みenrollment keyとfixture。旧形式の管理者によるWorkspace信頼登録を読み取る境界だけに綴りを保持する。通常要求の署名検証は現行の信頼済み公開鍵を使い、旧counterや過去のverification記録は認可に使わない。削除すると既存の信頼登録を読めなくなる。", "lines": [ { "sha256": "56e399dc4e828e66677d4ad09b4c66a9c09658322b3886a06b4ae688c63dfc84", "count": 1 }, - { - "sha256": "ea16c387d0048571bd0a5576f1358af32acf63f8ee5ebe08c262da83c5a04d88", - "count": 1 - }, - { - "sha256": "bfd48831eb694b879a2859d42dba00fe74a4a817177acc61a44795cbd19b1056", - "count": 1 - }, - { - "sha256": "94af0abdfb45c712e9cd1949ac6a15d1a21c3b271f004c9282f81b5fd5ae4f5d", - "count": 1 - }, { "sha256": "d099e3a7216349c0ef30026967b8fe344e3b08b58a77bad6437468502b56bf71", "count": 1 - }, - { - "sha256": "bed791d41faa85b38602c238d30460681e6d9ebbe56c4c08b843dd1971a589b2", - "count": 1 - }, - { - "sha256": "ce36822c65f785f330e42b7d37241062d994dc4e1c18abaa327b7996da5b6521", - "count": 1 - }, - { - "sha256": "5f4aed5ed50f2bbf54bbb9293948c3294eb2d30086d0fbe7cb07899efb5c96ba", - "count": 1 } ] }, diff --git a/web/workspace/src/lib/generated/companion-api.ts b/web/workspace/src/lib/generated/companion-api.ts index a03341919..b8b63d7a0 100644 --- a/web/workspace/src/lib/generated/companion-api.ts +++ b/web/workspace/src/lib/generated/companion-api.ts @@ -1,6 +1,6 @@ // @generated by yoi-companion-openapi-typescript@1. Do not edit. // Input: openapi/server-api.json -// Input contract: 0.1.0 (sha256:fc25104c077ab8c5c43f77611f505171585d2d40fd54c6538774ac17aae6cce3) +// Input contract: 0.1.0 (sha256:a23a7b9989d37fd08459929a4e726396a33334489207799e02d45302e8d4c238) // Options: companion-operation-schema-closure,openapi-3.1,strict-integer-ranges,sorted-output // Output: web/workspace/src/lib/generated/companion-api.ts // Regenerate/check: cargo run -q -p server-api --example generate_companion_api_types -- --check diff --git a/web/workspace/src/lib/generated/repository-api.ts b/web/workspace/src/lib/generated/repository-api.ts index 93145145c..f65371b66 100644 --- a/web/workspace/src/lib/generated/repository-api.ts +++ b/web/workspace/src/lib/generated/repository-api.ts @@ -1,6 +1,6 @@ // @generated by yoi-repository-openapi-typescript@1. Do not edit. // Input: openapi/server-api.json -// Input contract: 0.1.0 (sha256:fc25104c077ab8c5c43f77611f505171585d2d40fd54c6538774ac17aae6cce3) +// Input contract: 0.1.0 (sha256:a23a7b9989d37fd08459929a4e726396a33334489207799e02d45302e8d4c238) // Options: repository-operation-schema-closure,openapi-3.1,format:uint64=safe-number,strict-safe-integers,sorted-output // Output: web/workspace/src/lib/generated/repository-api.ts // Regenerate/check: cargo run -q -p server-api --example generate_repository_openapi_types -- --check diff --git a/web/workspace/src/lib/generated/runtime-api.ts b/web/workspace/src/lib/generated/runtime-api.ts index 388f6867b..e6e9e6411 100644 --- a/web/workspace/src/lib/generated/runtime-api.ts +++ b/web/workspace/src/lib/generated/runtime-api.ts @@ -1,6 +1,6 @@ // @generated by yoi-runtime-openapi-typescript@1. Do not edit. // Input: openapi/server-api.json -// Input contract: 0.1.0 (sha256:fc25104c077ab8c5c43f77611f505171585d2d40fd54c6538774ac17aae6cce3) +// Input contract: 0.1.0 (sha256:a23a7b9989d37fd08459929a4e726396a33334489207799e02d45302e8d4c238) // Options: runtime-operation-schema-closure,openapi-3.1,strict-integer-ranges,sorted-output // Output: web/workspace/src/lib/generated/runtime-api.ts // Regenerate/check: cargo run -q -p server-api --example generate_runtime_api_types -- --check @@ -25,7 +25,7 @@ export type CompletionEntry = { description?: string | null; invocation?: Featur export type CompletionKind = "file" | "feature" | "feature_argument"; -export type CreateRemoteRuntimeRequest = { display_name?: string | null; endpoint: string; expected_binding_id?: string | null; public_bundle: RuntimePublicIdentityBundle; workspace_trust_id: string; }; +export type CreateRemoteRuntimeRequest = { display_name?: string | null; endpoint: string; expected_binding_id?: string | null; public_bundle: RuntimePublicIdentityBundle; }; export type Diagnostic = { code: string; message: string; severity: DiagnosticSeverity; }; @@ -77,11 +77,9 @@ export type RuntimeConfigBundleResult = { availability?: unknown; diagnostics?: export type RuntimeConfigBundleSyncRequest = { bundle: unknown; }; -export type RuntimeConnectionDisplayState = "configured" | "verified" | "unavailable" | "revoked"; - export type RuntimeConnectionTestFailureKind = "authentication" | "authorization" | "network_unreachable" | "timeout" | "tls_or_transport" | "malformed_response" | "protocol_version_mismatch" | "runtime_identity_mismatch" | "configuration"; -export type RuntimeConnectionTestResponse = { actual_protocol_version?: number | null; binding_id: string; checked_at: string; connection_state: RuntimeConnectionDisplayState; diagnostics?: Array; expected_protocol_version: number; failure_kind?: RuntimeConnectionTestFailureKind | null; runtime_id: string; status: RuntimeConnectionTestStatus; verification?: RuntimeVerificationEvidenceSummary | null; workspace_id: string; }; +export type RuntimeConnectionTestResponse = { actual_protocol_version?: number | null; binding_id: string; checked_at: string; diagnostics?: Array; expected_protocol_version: number; failure_kind?: RuntimeConnectionTestFailureKind | null; runtime_id: string; status: RuntimeConnectionTestStatus; workspace_id: string; }; export type RuntimeConnectionTestStatus = "compatible" | "failed"; @@ -121,10 +119,6 @@ export type RuntimeTrustKeyState = { binding_id?: string | null; created_at?: st export type RuntimeTrustKeyStatus = "unconfigured" | "active" | "revoked"; -export type RuntimeVerificationEvidenceSummary = { binding_id: string; last_checked_at: string; last_outcome: RuntimeVerificationOutcome; runtime_public_key_fingerprint: string; verified_at?: string | null; workspace_key_id: string; workspace_public_key_fingerprint: string; workspace_trust_id: string; }; - -export type RuntimeVerificationOutcome = "verified" | "challenge_issued" | "verification_failed" | "connectivity_failed"; - export type RuntimeWorkerCompletionsRequest = { context?: CompletionContext | null; kind: CompletionKind; prefix?: string; request_id?: string | null; }; export type RuntimeWorkerCompletionsResult = { context?: CompletionContext | null; diagnostics?: Array; entries: Array; kind: CompletionKind; prefix: string; request_id?: string | null; runtime_id: string; worker_id: string; }; @@ -217,9 +211,7 @@ export type WorkingDirectorySummary = { cleanliness?: string | null; cleanup_tar export type WorkspaceConfigAccess = "read_only" | "read_write"; -export type WorkspaceRuntimeBindingState = "configured" | "verified" | "revoked"; - -export type WorkspaceRuntimeBindingSummary = { binding_id: string; connection_state: RuntimeConnectionDisplayState; state: WorkspaceRuntimeBindingState; verification?: RuntimeVerificationEvidenceSummary | null; workspace_key_id?: string | null; workspace_trust_id?: string | null; }; +export type WorkspaceRuntimeBindingSummary = { binding_id: string; revoked_at?: string | null; }; export type WorkspaceRuntimeDetail = { endpoint?: string | null; recent_audit?: Array; runtime: WorkspaceRuntimeResource; trust_key: RuntimeTrustKeyState; workspace_id: string; }; diff --git a/web/workspace/src/lib/generated/ticket-api.ts b/web/workspace/src/lib/generated/ticket-api.ts index 8818a7ad0..f14cd9755 100644 --- a/web/workspace/src/lib/generated/ticket-api.ts +++ b/web/workspace/src/lib/generated/ticket-api.ts @@ -1,6 +1,6 @@ // @generated by yoi-ticket-browser-openapi-typescript@1. Do not edit. // Input: openapi/server-api.json -// Input contract: 0.1.0 (sha256:fc25104c077ab8c5c43f77611f505171585d2d40fd54c6538774ac17aae6cce3) +// Input contract: 0.1.0 (sha256:a23a7b9989d37fd08459929a4e726396a33334489207799e02d45302e8d4c238) // Options: ticket-objective-merge-request-browser-operation-schema-closure,openapi-3.1,strict-integer-ranges,sorted-output // Output: web/workspace/src/lib/generated/ticket-api.ts // Regenerate/check: cargo run -q -p server-api --example generate_ticket_api_types -- --check diff --git a/web/workspace/src/lib/generated/worker-launch-api.ts b/web/workspace/src/lib/generated/worker-launch-api.ts index 8814b40e8..6b2b5184f 100644 --- a/web/workspace/src/lib/generated/worker-launch-api.ts +++ b/web/workspace/src/lib/generated/worker-launch-api.ts @@ -1,6 +1,6 @@ // @generated by yoi-worker-openapi-typescript@1. Do not edit. // Input: openapi/server-api.json -// Input contract: 0.1.0 (sha256:fc25104c077ab8c5c43f77611f505171585d2d40fd54c6538774ac17aae6cce3) +// Input contract: 0.1.0 (sha256:a23a7b9989d37fd08459929a4e726396a33334489207799e02d45302e8d4c238) // Options: worker-operation-schema-closure,openapi-3.1,strict-integer-ranges,sorted-output // Output: web/workspace/src/lib/generated/worker-launch-api.ts // Regenerate/check: cargo run -q -p server-api --example generate_worker_launch_api_types -- --check diff --git a/web/workspace/src/lib/workspace/api/runtime-connection.ts b/web/workspace/src/lib/workspace/api/runtime-connection.ts index bde3120a8..535129cb9 100644 --- a/web/workspace/src/lib/workspace/api/runtime-connection.ts +++ b/web/workspace/src/lib/workspace/api/runtime-connection.ts @@ -2,15 +2,12 @@ import type { Diagnostic, RuntimeConnectionTestFailureKind, RuntimeConnectionTestResponse, - RuntimeVerificationEvidenceSummary, } from "#lib/generated/runtime-api.ts"; const RESPONSE_KEYS = [ "workspace_id", "runtime_id", "binding_id", - "connection_state", - "verification", "checked_at", "status", "failure_kind", @@ -108,27 +105,11 @@ export function parseRuntimeConnectionTestResponse( return null; } const bindingId = value.binding_id; - const connectionState = parseConnectionState(value.connection_state); - const verification = parseVerificationEvidence(value.verification); - if ( - !isBoundedString(bindingId) || - connectionState === null || - (value.verification !== null && verification === null) || - (verification !== null && - verification.binding_id !== bindingId) || - (connectionState === "verified" && value.status !== "compatible") || - (connectionState === "verified" && verification !== null && - (verification.last_outcome !== "verified" || - verification.verified_at === null)) - ) { - return null; - } + if (!isBoundedString(bindingId, 256)) return null; return { workspace_id: value.workspace_id, runtime_id: value.runtime_id, binding_id: bindingId, - connection_state: connectionState, - verification, checked_at: value.checked_at, status: value.status, failure_kind: failureKind as RuntimeConnectionTestFailureKind | null, @@ -138,60 +119,6 @@ export function parseRuntimeConnectionTestResponse( }; } -function parseConnectionState( - value: unknown, -): "configured" | "verified" | "unavailable" | "revoked" | null { - return value === "configured" || value === "verified" || - value === "unavailable" || value === "revoked" - ? value - : null; -} - -function parseVerificationEvidence( - value: unknown, -): RuntimeVerificationEvidenceSummary | null { - if (value === null) return null; - const keys = [ - "verified_at", - "last_checked_at", - "last_outcome", - "binding_id", - "workspace_key_id", - "workspace_public_key_fingerprint", - "workspace_trust_id", - "runtime_public_key_fingerprint", - ] as const; - if (!isRecord(value) || !hasExactKeys(value, keys)) return null; - if ( - (value.verified_at !== null && - (!isBoundedString(value.verified_at, 128) || - Number.isNaN(Date.parse(value.verified_at)))) || - !isBoundedString(value.last_checked_at, 128) || - Number.isNaN(Date.parse(value.last_checked_at)) || - (value.last_outcome !== "verified" && - value.last_outcome !== "challenge_issued" && - value.last_outcome !== "verification_failed" && - value.last_outcome !== "connectivity_failed") || - !isBoundedString(value.binding_id) || - !isBoundedString(value.workspace_key_id, 128) || - !isBoundedString(value.workspace_public_key_fingerprint) || - !isBoundedString(value.workspace_trust_id) || - !isBoundedString(value.runtime_public_key_fingerprint, 128) - ) { - return null; - } - return { - verified_at: value.verified_at, - last_checked_at: value.last_checked_at, - last_outcome: value.last_outcome, - binding_id: value.binding_id, - workspace_key_id: value.workspace_key_id, - workspace_public_key_fingerprint: value.workspace_public_key_fingerprint, - workspace_trust_id: value.workspace_trust_id, - runtime_public_key_fingerprint: value.runtime_public_key_fingerprint, - }; -} - export async function testRuntimeConnection( workspaceId: string, runtimeId: string, diff --git a/web/workspace/src/lib/workspace/api/runtime-management.ts b/web/workspace/src/lib/workspace/api/runtime-management.ts index fa2809323..a34a83871 100644 --- a/web/workspace/src/lib/workspace/api/runtime-management.ts +++ b/web/workspace/src/lib/workspace/api/runtime-management.ts @@ -3,7 +3,6 @@ import type { Diagnostic, RemoveRuntimeRequest, RevokeRuntimeTrustKeyRequest, - RuntimeConnectionDisplayState, RuntimeIdentityAuthority, RuntimeManagementApiError, RuntimeManagementSummary, @@ -17,14 +16,13 @@ import type { RuntimeTrustKeyRevealResponse, RuntimeTrustKeyState, RuntimeTrustKeyStatus, - RuntimeVerificationEvidenceSummary, UpdateRemoteRuntimeRequest, - WorkspaceRuntimeBindingState, WorkspaceRuntimeBindingSummary, WorkspaceRuntimeDetail, WorkspaceRuntimeListResponse, WorkspaceRuntimeResource, } from "#lib/generated/runtime-api.ts"; +import type { RepositoryApiError } from "#lib/generated/repository-api.ts"; import { workspaceApiPath } from "./http.ts"; export type WorkspaceRuntimeList = @@ -90,17 +88,6 @@ const CONFLICT_KINDS = new Set([ "stale_binding", "fingerprint_in_use", ]); -const BINDING_STATES = new Set([ - "configured", - "verified", - "revoked", -]); -const CONNECTION_STATES = new Set([ - "configured", - "verified", - "unavailable", - "revoked", -]); const REMOVAL_OPERATION_STATES = new Set([ "pending", "cleanup_pending", @@ -326,140 +313,16 @@ function runtimeSource(value: unknown, path: string): RuntimeSourceSummary { }; } -function runtimeVerification( - value: unknown, - path: string, -): RuntimeVerificationEvidenceSummary { - const item = object(value, path); - exactKeys( - item, - [ - "verified_at", - "last_checked_at", - "last_outcome", - "binding_id", - "workspace_key_id", - "workspace_public_key_fingerprint", - "workspace_trust_id", - "runtime_public_key_fingerprint", - ], - [], - path, - ); - const verifiedAt = item.verified_at === null - ? null - : boundedString(item.verified_at, `${path}.verified_at`, 128); - const lastOutcome = enumValue( - item.last_outcome, - `${path}.last_outcome`, - new Set( - [ - "verified", - "challenge_issued", - "verification_failed", - "connectivity_failed", - ] as const, - ), - ); - return { - verified_at: verifiedAt, - last_checked_at: boundedString( - item.last_checked_at, - `${path}.last_checked_at`, - 128, - ), - last_outcome: lastOutcome, - binding_id: boundedIdentity( - item.binding_id, - `${path}.binding_id`, - ), - workspace_key_id: boundedString( - item.workspace_key_id, - `${path}.workspace_key_id`, - LIMITS.idBytes, - ), - workspace_public_key_fingerprint: boundedIdentity( - item.workspace_public_key_fingerprint, - `${path}.workspace_public_key_fingerprint`, - ), - workspace_trust_id: boundedIdentity( - item.workspace_trust_id, - `${path}.workspace_trust_id`, - ), - runtime_public_key_fingerprint: boundedString( - item.runtime_public_key_fingerprint, - `${path}.runtime_public_key_fingerprint`, - LIMITS.fingerprintBytes, - ), - }; -} - function runtimeBinding( value: unknown, path: string, - requiresWorkspaceIdentity: boolean, ): WorkspaceRuntimeBindingSummary { const item = object(value, path); - exactKeys( - item, - ["state", "connection_state", "binding_id"], - ["workspace_key_id", "workspace_trust_id", "verification"], - path, - ); - const workspaceKeyId = optionalNullableString( - item.workspace_key_id, - `${path}.workspace_key_id`, - LIMITS.idBytes, - ); - const workspaceTrustId = optionalNullableIdentity( - item.workspace_trust_id, - `${path}.workspace_trust_id`, - ); - const state = enumValue(item.state, `${path}.state`, BINDING_STATES); - if ( - requiresWorkspaceIdentity && - state !== "revoked" && - (workspaceKeyId == null || workspaceTrustId == null) - ) { - return fail(path, "requires Workspace signing key identity metadata"); - } - const connectionState = enumValue( - item.connection_state, - `${path}.connection_state`, - CONNECTION_STATES, - ); - const binding_id = boundedIdentity(item.binding_id, `${path}.binding_id`); - const verification = item.verification === undefined - ? undefined - : runtimeVerification(item.verification, `${path}.verification`); - if ( - verification !== undefined && verification.binding_id !== binding_id - ) { - return fail(path, "verification must match the current binding binding_id"); - } - if ( - requiresWorkspaceIdentity && - connectionState === "verified" && - (verification === undefined || - verification.verified_at === null || - verification.last_outcome !== "verified") - ) { - return fail( - path, - "verified Workspace identity binding requires verification evidence", - ); - } + exactKeys(item, ["binding_id"], ["revoked_at"], path); return { - state, - connection_state: connectionState, - binding_id, - ...(workspaceKeyId === undefined - ? {} - : { workspace_key_id: workspaceKeyId }), - ...(workspaceTrustId === undefined - ? {} - : { workspace_trust_id: workspaceTrustId }), - ...(verification === undefined ? {} : { verification }), + binding_id: boundedIdentity(item.binding_id, `${path}.binding_id`), + revoked_at: + optionalNullableTimestamp(item.revoked_at, `${path}.revoked_at`) ?? null, }; } @@ -483,7 +346,7 @@ function runtimeManagement( const builtIn = boolean(item.built_in, `${path}.built_in`); const binding = item.binding == null ? undefined - : runtimeBinding(item.binding, `${path}.binding`, !builtIn); + : runtimeBinding(item.binding, `${path}.binding`); return { built_in: builtIn, config_managed: boolean(item.config_managed, `${path}.config_managed`), @@ -763,7 +626,7 @@ export function parseRuntimeTrustConflict( exactKeys( response, ["error", "message"], - ["current_binding_id", "current_fingerprint"], + ["current_binding_id", "current_fingerprint", "diagnostics"], "Runtime trust conflict", ); return { @@ -786,6 +649,15 @@ export function parseRuntimeTrustConflict( "Runtime trust conflict.current_fingerprint", LIMITS.fingerprintBytes, ), + ...(response.diagnostics === undefined ? {} : { + diagnostics: array( + response.diagnostics, + "Runtime trust conflict.diagnostics", + LIMITS.diagnostics, + ).map((entry, index) => + diagnostic(entry, `Runtime trust conflict.diagnostics[${index}]`) + ), + }), }; } @@ -891,6 +763,37 @@ async function readBoundedJson(response: Response): Promise { } } +function repositoryApiError(value: unknown): RepositoryApiError { + const response = object(value, "Runtime trust error"); + exactKeys( + response, + ["error", "message"], + ["diagnostics"], + "Runtime trust error", + ); + return { + error: boundedString( + response.error, + "Runtime trust error.error", + LIMITS.idBytes, + ), + message: boundedString( + response.message, + "Runtime trust error.message", + LIMITS.conflictMessageBytes, + ), + ...(response.diagnostics === undefined ? {} : { + diagnostics: array( + response.diagnostics, + "Runtime trust error.diagnostics", + LIMITS.diagnostics, + ).map((entry, index) => + diagnostic(entry, `Runtime trust error.diagnostics[${index}]`) + ), + }), + }; +} + function requestErrorFrom( value: unknown, status: number, @@ -917,30 +820,13 @@ function requestErrorFrom( ), ); } - exactKeys( - response, - ["error", "message", "diagnostics"], - [], - "Runtime trust error", - ); - const diagnostics = array( - response.diagnostics, - "Runtime trust error.diagnostics", - LIMITS.diagnostics, - ).map((entry, index) => - diagnostic(entry, `Runtime trust error.diagnostics[${index}]`) - ); - const message = boundedString( - response.message, - "Runtime trust error.message", - LIMITS.conflictMessageBytes, - ); - const field = diagnostics.some((entry) => + const error = repositoryApiError(response); + const field = (error.diagnostics ?? []).some((entry) => entry.code.startsWith("runtime_public_key_") ) ? "public_key" : null; - return new RuntimeTrustRequestError(message, field); + return new RuntimeTrustRequestError(error.message, field); } catch { return new RuntimeTrustRequestError( `Runtime trust request failed (${status})`, @@ -960,6 +846,21 @@ function throwRuntimeTrustConflict(payload: unknown): never { throw new RuntimeTrustConflictError(conflict); } +function throwRuntimeCreationConflict(payload: unknown): never { + let error: RepositoryApiError; + try { + error = repositoryApiError(payload); + } catch { + throw new RuntimeTrustRequestError( + "Runtime trust conflict response was invalid", + ); + } + if (CONFLICT_KINDS.has(error.error as RuntimeTrustConflictKind)) { + throwRuntimeTrustConflict(error); + } + throw requestErrorFrom(error, 409); +} + async function finishMutation( response: Response, workspaceId: string, @@ -990,17 +891,16 @@ export async function createRemoteRuntime( request: CreateRemoteRuntimeRequest, fetchImpl: typeof fetch = fetch, ): Promise { - const enrollmentId = boundedIdentity(request.workspace_trust_id, "workspace_trust_id"); const response = await fetchImpl( workspaceApiPath(workspaceId, "/runtimes"), { method: "POST", headers: { "content-type": "application/json" }, - body: JSON.stringify({ ...request, workspace_trust_id: enrollmentId }), + body: JSON.stringify(request), }, ); const payload = await readBoundedJson(response); - if (response.status === 409) throwRuntimeTrustConflict(payload); + if (response.status === 409) throwRuntimeCreationConflict(payload); if (!response.ok) throw requestErrorFrom(payload, response.status); const runtime = runtimeResource(payload, "Runtime create response"); if (runtime.runtime_id !== request.public_bundle.identity_id) { diff --git a/web/workspace/src/lib/workspace/repositories/ssh-connection.ts b/web/workspace/src/lib/workspace/repositories/ssh-connection.ts index 6bf8eb28a..983fd4a1d 100644 --- a/web/workspace/src/lib/workspace/repositories/ssh-connection.ts +++ b/web/workspace/src/lib/workspace/repositories/ssh-connection.ts @@ -68,6 +68,6 @@ export function repositorySshProbeRuntimes( runtime.management.endpoint_configured && runtime.management.binding !== undefined && runtime.management.binding !== null && - runtime.management.binding.state !== "revoked" + runtime.management.binding.revoked_at == null ); } diff --git a/web/workspace/src/routes/w/[workspaceId]/settings/runtimes/+page.svelte b/web/workspace/src/routes/w/[workspaceId]/settings/runtimes/+page.svelte index b1f7d5aab..f9d32bae8 100644 --- a/web/workspace/src/routes/w/[workspaceId]/settings/runtimes/+page.svelte +++ b/web/workspace/src/routes/w/[workspaceId]/settings/runtimes/+page.svelte @@ -19,7 +19,6 @@ let { data }: PageProps = $props(); let runtimePublicBundle = $state(''); - let workspaceTrustId = $state(''); let displayName = $state(''); let endpoint = $state(''); let runtimeFingerprint = $state(null); @@ -36,9 +35,7 @@ function connectionTestSummary(result: RuntimeConnectionTestResponse): string { if (result.status === 'compatible') { - return result.connection_state === 'verified' - ? `Verified · protocol v${result.actual_protocol_version}` - : `Compatible · ${result.connection_state} · protocol v${result.actual_protocol_version}`; + return `Signed ping succeeded · protocol v${result.actual_protocol_version}`; } switch (result.failure_kind) { case 'authentication': return 'Authentication failed'; @@ -145,18 +142,16 @@ } await createRemoteRuntime(data.workspaceId, { public_bundle: publicBundle, - workspace_trust_id: workspaceTrustId, display_name: displayName || null, endpoint, expected_binding_id: null, }); runtimePublicBundle = ''; - workspaceTrustId = ''; runtimeFingerprint = null; displayName = ''; endpoint = ''; showAddRuntime = false; - requestNotice = 'Runtime registered for this Workspace. Run Test to complete authenticated verification.'; + requestNotice = 'Runtime registered for this Workspace. Run Test to observe a signed ping.'; await invalidateAll(); } catch (error) { requestError = error instanceof RuntimeTrustRequestError || error instanceof Error @@ -171,7 +166,8 @@ runtime: WorkspaceRuntimeResource, ): RuntimeConnectionTestResponse | undefined { const result = testResults[runtime.runtime_id]; - return result?.binding_id === runtime.management?.binding?.binding_id + return result?.workspace_id === data.workspaceId && + result?.binding_id === runtime.management?.binding?.binding_id ? result : undefined; } @@ -179,12 +175,16 @@ async function testRuntime(runtime: WorkspaceRuntimeResource): Promise { const bindingId = runtime.management?.binding?.binding_id; if (typeof bindingId !== 'string') return; + const workspaceId = data.workspaceId; const generation = ++connectionTestGeneration; requestError = null; + const { [runtime.runtime_id]: _previousResult, ...remainingResults } = testResults; + testResults = remainingResults; busyRuntimeId = runtime.runtime_id; try { const result = await testRuntimeConnection(data.workspaceId, runtime.runtime_id); if ( + workspaceId !== data.workspaceId || generation !== connectionTestGeneration || result.binding_id !== bindingId ) { @@ -261,15 +261,9 @@ 
yoi-runtime trust-workspace add --bundle {workspaceBundleFilename()}
- Copy the Runtime-issued workspace_trust_id from trust-workspace add/show. - It identifies this enrollment, not the Workspace key fingerprint. Pass the same --fs-root and --fs-runtime-dir options used by the Runtime service. Existing Workspace trust entries are preserved. - {:else}

Loading Workspace public identity…

{/if} @@ -323,14 +317,14 @@

Registration stores this Workspace-scoped binding. After it appears in the list, run - Test to complete authenticated verification. + Test to observe a signed ping. Each request authenticates independently.

{/if} - {#if runtime.management?.binding?.state === 'configured'} - Verification required - {:else if !runtime.management?.config_managed} + {#if !runtime.management?.config_managed} Test unavailable {/if}
diff --git a/web/workspace/src/routes/w/[workspaceId]/settings/runtimes/[runtimeId]/+page.svelte b/web/workspace/src/routes/w/[workspaceId]/settings/runtimes/[runtimeId]/+page.svelte index 770c81310..bde2d1441 100644 --- a/web/workspace/src/routes/w/[workspaceId]/settings/runtimes/[runtimeId]/+page.svelte +++ b/web/workspace/src/routes/w/[workspaceId]/settings/runtimes/[runtimeId]/+page.svelte @@ -2,6 +2,7 @@ import { goto, invalidateAll } from '$app/navigation'; import type { RevokeRuntimeTrustKeyRequest, + RuntimeConnectionTestResponse, RuntimeTrustKeyStatus, } from '#lib/generated/runtime-api.ts'; import { @@ -17,6 +18,7 @@ RuntimeTrustRequestError, type RuntimeTrustRouteOperation, } from '#lib/workspace/api/runtime-management.ts'; + import { testRuntimeConnection } from '#lib/workspace/api/runtime-connection.ts'; import type { PageProps } from './$types'; type TrustAction = 'create' | 'replace' | 'reactivate'; @@ -25,18 +27,19 @@ let showPublicKey = $state(false); let revealedPublicKey = $state(null); let publicKey = $state(''); - let workspaceTrustId = $state(''); let displayName = $state(''); let endpoint = $state(''); let editingMetadata = $state(false); let deleteRuntimeConfirmation = $state(''); - let busyAction = $state<'metadata' | 'trust' | 'revoke' | 'reveal' | 'copy' | 'delete' | null>(null); + let busyAction = $state<'metadata' | 'trust' | 'revoke' | 'reveal' | 'copy' | 'delete' | 'test' | null>(null); let fieldError = $state(null); let deleteRuntimeError = $state(null); let requestError = $state(null); let successMessage = $state(null); let replacementFingerprint = $state(null); let replacementFingerprintError = $state(null); + let connectionResult = $state(null); + let connectionError = $state(null); let fingerprintGeneration = 0; const routeFence = new RuntimeTrustRouteFence(); const runtimeRemovalAttempt = new RuntimeRemovalAttempt(); @@ -47,10 +50,11 @@ if (nextGeneration === routeGeneration) return; routeGeneration = nextGeneration; fingerprintGeneration += 1; + connectionResult = null; + connectionError = null; showPublicKey = false; revealedPublicKey = null; publicKey = ''; - workspaceTrustId = ''; displayName = data.runtimeDetail?.runtime.label ?? ''; endpoint = data.runtimeDetail?.endpoint ?? ''; editingMetadata = false; @@ -109,6 +113,28 @@ return new TextEncoder().encode(value).byteLength; } + async function testConnection(): Promise { + const bindingId = data.runtimeDetail?.runtime.management.binding?.binding_id; + if (busyAction !== null || !bindingId) return; + const operation = routeFence.capture(data.runtimeId); + const workspaceId = data.workspaceId; + busyAction = 'test'; + connectionResult = null; + connectionError = null; + try { + const result = await testRuntimeConnection(workspaceId, operation.runtimeId); + if (!isCurrentRoute(operation) || workspaceId !== data.workspaceId || + result.binding_id !== data.runtimeDetail?.runtime.management.binding?.binding_id) return; + connectionResult = result; + } catch (error) { + if (isCurrentRoute(operation) && workspaceId === data.workspaceId) { + connectionError = error instanceof Error ? error.message : String(error); + } + } finally { + if (isCurrentRoute(operation) && workspaceId === data.workspaceId) busyAction = null; + } + } + async function reloadAuthority(): Promise { await invalidateAll(); } @@ -195,7 +221,7 @@ const binding = data.runtimeDetail.runtime.management.binding; if (!binding || !data.runtimeDetail.endpoint) { throw new RuntimeTrustRequestError( - 'The Workspace identity binding and authoritative Runtime endpoint are required.', + 'The Runtime binding and authoritative endpoint are required.', ); } await createRemoteRuntime(data.workspaceId, { @@ -203,14 +229,12 @@ identity_id: operation.runtimeId, public_key: key, }, - workspace_trust_id: workspaceTrustId, display_name: data.runtimeDetail.runtime.label, endpoint: data.runtimeDetail.endpoint, expected_binding_id: binding.binding_id, }); if (!isCurrentRoute(operation)) return; publicKey = ''; - workspaceTrustId = ''; showPublicKey = false; revealedPublicKey = null; successMessage = action === 'create' @@ -396,8 +420,8 @@ {@const detail = data.runtimeDetail} {@const runtime = detail.runtime} {@const trust = detail.trust_key} - {@const verifiedTrust = runtime.management.binding?.state === 'verified'} {@const currentAction = trustAction(trust.status)} + {@const latestResult = connectionResult?.workspace_id === data.workspaceId && connectionResult?.runtime_id === data.runtimeId && connectionResult?.binding_id === runtime.management.binding?.binding_id ? connectionResult : null}

Identity and binding

@@ -405,17 +429,32 @@
Runtime ID
{runtime.runtime_id}
Kind
{runtime.kind}
Endpoint
{detail.endpoint ?? 'Not configured'}
-
Status
{runtime.status}
-
Connection state
{runtime.management.binding?.connection_state ?? 'Not configured'}
-
Workspace signing key
{runtime.management.binding?.workspace_key_id ?? '—'}
-
Verified
{formatTimestamp(runtime.management.binding?.verification?.verified_at)}
-
Last verification check
{runtime.management.binding?.verification?.last_outcome ?? '—'} · {formatTimestamp(runtime.management.binding?.verification?.last_checked_at)}
+
Connection test
{runtime.management.binding?.revoked_at ? 'Revoked' : latestResult ? latestResult.status === 'compatible' ? 'Signed ping succeeded' : `Signed ping failed · ${latestResult.failure_kind}` : 'Not tested'}
+
Binding ID
{runtime.management.binding?.binding_id ?? '—'}
Runtime key status
{trust.status}
Fingerprint
{trust.fingerprint ?? '—'}
Created
{formatTimestamp(trust.created_at)}
Updated
{formatTimestamp(trust.updated_at)}
Revoked
{formatTimestamp(trust.revoked_at)}
+ {#if runtime.management.config_managed && !runtime.management.binding?.revoked_at} +
+ +
+ {/if} + {#if latestResult} +

+ Checked {formatTimestamp(latestResult.checked_at)} + {#each latestResult.diagnostics ?? [] as diagnostic} + {diagnostic.message} + {/each} +

+ {/if} + {#if connectionError} + + {/if} {#if (runtime.diagnostics ?? []).length > 0}
    {#each runtime.diagnostics ?? [] as diagnostic} @@ -495,25 +534,7 @@ {/if} {/if} - {#if verifiedTrust} -
    - Runtime public key -

    - This verified key is read-only. Revoke Workspace trust and register the Runtime again to use a different public key. -

    - {trust.fingerprint} -
    - {:else} -
    - - -

    Copy the Runtime-issued workspace_trust_id from trust-workspace add/show for the current enrollment. Do not use a key fingerprint or a revoked enrollment ID.

    +