use affinidi_did_resolver_cache_sdk::DIDCacheClient;
use tracing::info;
use crate::audit::{self, audit};
use vta_sdk::protocols::acl_management::{
create::{CreateAclResponseBody, CreateAclResultBody},
delete::DeleteAclResultBody,
get::GetAclResultBody,
list::ListAclResultBody,
swap::AclSwapPresentation,
};
use crate::acl::{
AclEntry, ApproveScope, ContextDirection, Role, acl_entry_matches_context, delete_acl_entry,
get_acl_entry, is_acl_entry_auditable, is_acl_entry_visible, list_acl_entries, store_acl_entry,
update_acl_entry_versioned, validate_acl_modification, validate_approve_scope_grant,
validate_role_assignment,
};
use crate::auth::AuthClaims;
use crate::auth::session::now_epoch;
use crate::contexts::get_context;
use crate::error::AppError;
use crate::store::KeyspaceHandle;
use vti_common::auth::step_up::StepUpMode;
pub struct UpdateAclParams {
pub role: Option<Role>,
pub label: Option<String>,
pub allowed_contexts: Option<Vec<String>>,
pub step_up_approver: Option<String>,
pub step_up_require: Option<String>,
pub approve_scope: Option<ApproveScope>,
pub allowed_keys: Option<Option<std::collections::BTreeSet<String>>>,
}
pub fn parse_step_up_require(s: Option<&str>) -> Result<Option<StepUpMode>, AppError> {
match s.map(str::trim) {
None | Some("") => Ok(None),
Some("self") => Ok(Some(StepUpMode::SelfApprove)),
Some("delegated") => Ok(Some(StepUpMode::Delegated)),
Some(other) => Err(AppError::Validation(format!(
"invalid stepUp.require '{other}': must be 'self' or 'delegated'"
))),
}
}
pub(crate) fn acl_entry_can_confer(entry: &AclEntry, ctx: &str) -> bool {
if entry.approve_scope.covers(ctx) {
return true;
}
let claims = AuthClaims {
did: entry.did.clone(),
role: entry.role.clone(),
allowed_contexts: entry.allowed_contexts.clone(),
..Default::default()
};
claims.role == Role::Admin && claims.has_context_access(ctx)
}
pub fn approve_scope_from_wire(all: bool, contexts: Vec<String>) -> ApproveScope {
if all {
ApproveScope::All
} else if !contexts.is_empty() {
ApproveScope::Contexts(contexts)
} else {
ApproveScope::None
}
}
fn step_up_require_to_wire(m: Option<StepUpMode>) -> Option<String> {
m.map(|m| {
match m {
StepUpMode::SelfApprove => "self",
StepUpMode::Delegated => "delegated",
StepUpMode::DelegatedAny => "delegated-any",
StepUpMode::None => "none",
}
.to_string()
})
}
fn symmetric_difference_contexts(old: &[String], new: &[String]) -> Vec<String> {
use std::collections::HashSet;
let old_set: HashSet<&str> = old.iter().map(String::as_str).collect();
let new_set: HashSet<&str> = new.iter().map(String::as_str).collect();
old_set
.symmetric_difference(&new_set)
.map(|s| (*s).to_string())
.collect()
}
async fn require_contexts_exist(
contexts_ks: &KeyspaceHandle,
contexts: &[String],
) -> Result<(), AppError> {
for ctx in contexts {
if get_context(contexts_ks, ctx).await?.is_none() {
return Err(AppError::NotFound(format!(
"context '{ctx}' is not registered on this VTA — create it first via \
'vta contexts create --id {ctx}' (offline) or 'pnm contexts create' (online)"
)));
}
}
Ok(())
}
fn not_manageable(auth: &AuthClaims, entry: &AclEntry, did: &str, verb: &str) -> AppError {
if is_acl_entry_auditable(auth, entry) {
AppError::Forbidden(format!(
"{did} is visible to you because it may confer in a context you administer, \
but it acts outside your contexts — only an admin of the contexts it acts in \
can {verb} it"
))
} else {
AppError::NotFound(format!("ACL entry not found for DID: {did}"))
}
}
fn to_result_body(e: &AclEntry) -> CreateAclResultBody {
let (approve_all_contexts, approve_contexts) = match &e.approve_scope {
ApproveScope::All => (true, Vec::new()),
ApproveScope::Contexts(cs) => (false, cs.clone()),
ApproveScope::None => (false, Vec::new()),
};
CreateAclResultBody {
did: e.did.clone(),
role: e.role.to_string(),
label: e.label.clone(),
allowed_contexts: e.allowed_contexts.clone(),
created_at: e.created_at,
created_by: e.created_by.clone(),
expires_at: e.expires_at,
step_up_approver: e.step_up_approver.clone(),
step_up_require: step_up_require_to_wire(e.step_up_require),
approve_all_contexts,
approve_contexts,
allowed_keys: e
.allowed_keys
.as_ref()
.map(|keys| keys.iter().cloned().collect()),
}
}
fn validate_allowed_keys(
allowed_keys: Option<&std::collections::BTreeSet<String>>,
) -> Result<(), AppError> {
if let Some(keys) = allowed_keys {
for key in keys {
if key.trim().is_empty() {
return Err(AppError::Validation(
"allowedKeys must not contain an empty key id — pass an empty \
list to authorize no keys, or omit the member for no filter"
.into(),
));
}
}
}
Ok(())
}
#[allow(clippy::too_many_arguments)]
pub async fn create_acl(
acl_ks: &KeyspaceHandle,
audit_ks: &KeyspaceHandle,
contexts_ks: &KeyspaceHandle,
auth: &AuthClaims,
did: &str,
role: Role,
label: Option<String>,
allowed_contexts: Vec<String>,
expires_at: Option<u64>,
step_up_approver: Option<String>,
step_up_require: Option<String>,
approve_scope: ApproveScope,
allowed_keys: Option<std::collections::BTreeSet<String>>,
channel: &str,
) -> Result<CreateAclResultBody, AppError> {
auth.require_manage()?;
validate_role_assignment(auth, &role)?;
validate_acl_modification(auth, &role, &allowed_contexts)?;
validate_allowed_keys(allowed_keys.as_ref())?;
validate_approve_scope_grant(auth, &approve_scope)?;
require_contexts_exist(contexts_ks, &allowed_contexts).await?;
if let ApproveScope::Contexts(cs) = &approve_scope {
require_contexts_exist(contexts_ks, cs).await?;
}
let step_up_require = parse_step_up_require(step_up_require.as_deref())?;
if get_acl_entry(acl_ks, did).await?.is_some() {
return Err(AppError::Conflict(format!(
"ACL entry already exists for DID: {did}"
)));
}
let entry = AclEntry::new(did, role, auth.did.clone())
.with_label(label)
.with_contexts(allowed_contexts)
.with_expires_at(expires_at)
.with_step_up_approver(step_up_approver)
.with_step_up_require(step_up_require)
.with_approve_scope(approve_scope)
.with_allowed_keys(allowed_keys);
store_acl_entry(acl_ks, &entry).await?;
info!(channel, caller = %auth.did, did = %entry.did, role = %entry.role, "ACL entry created");
audit!(
"acl.create",
actor = &auth.did,
resource = did,
outcome = "success"
);
let _ = audit::record(
audit_ks,
"acl.create",
&auth.did,
Some(did),
"success",
Some(channel),
None,
)
.await;
Ok(to_result_body(&entry))
}
pub async fn get_acl(
acl_ks: &KeyspaceHandle,
auth: &AuthClaims,
did: &str,
channel: &str,
) -> Result<CreateAclResultBody, AppError> {
auth.require_manage()?;
let entry = get_acl_entry(acl_ks, did)
.await?
.ok_or_else(|| AppError::NotFound(format!("ACL entry not found for DID: {did}")))?;
if !is_acl_entry_auditable(auth, &entry) {
return Err(AppError::NotFound(format!(
"ACL entry not found for DID: {did}"
)));
}
info!(channel, did = %did, "ACL entry retrieved");
Ok(to_result_body(&entry))
}
pub async fn list_acl(
acl_ks: &KeyspaceHandle,
auth: &AuthClaims,
context_filter: Option<&str>,
direction: ContextDirection,
channel: &str,
) -> Result<Vec<CreateAclResultBody>, AppError> {
auth.require_manage()?;
if context_filter.is_none() && direction != ContextDirection::default() {
return Err(AppError::Validation(format!(
"`direction={direction}` filters a context and there is no context to filter — \
pass a context as well, or drop the direction to list every visible entry"
)));
}
let all_entries = list_acl_entries(acl_ks).await?;
let entries: Vec<CreateAclResultBody> = all_entries
.iter()
.filter(|e| is_acl_entry_auditable(auth, e))
.filter(|e| match context_filter {
Some(ctx) => acl_entry_matches_context(e, ctx, direction),
None => true,
})
.map(to_result_body)
.collect();
info!(
channel,
caller = %auth.did,
context = context_filter.unwrap_or("-"),
direction = %direction,
count = entries.len(),
"ACL listed"
);
Ok(entries)
}
pub async fn update_acl(
acl_ks: &KeyspaceHandle,
audit_ks: &KeyspaceHandle,
contexts_ks: &KeyspaceHandle,
auth: &AuthClaims,
did: &str,
params: UpdateAclParams,
channel: &str,
) -> Result<CreateAclResultBody, AppError> {
auth.require_admin()?;
let mut entry = get_acl_entry(acl_ks, did)
.await?
.ok_or_else(|| AppError::NotFound(format!("ACL entry not found for DID: {did}")))?;
if !is_acl_entry_visible(auth, &entry) {
return Err(not_manageable(auth, &entry, did, "update"));
}
if let Some(ref role) = params.role {
validate_role_assignment(auth, role)?;
entry.role = role.clone();
}
if let Some(label) = params.label {
entry.label = Some(label);
}
if let Some(approver) = params.step_up_approver {
entry.step_up_approver = Some(approver);
}
if let Some(require) = params.step_up_require {
entry.step_up_require = if require.trim().is_empty() {
None
} else {
parse_step_up_require(Some(&require))?
};
}
if let Some(allowed_contexts) = params.allowed_contexts {
let changes = symmetric_difference_contexts(&entry.allowed_contexts, &allowed_contexts);
if !changes.is_empty() {
if !auth.is_super_admin() {
for ctx in &changes {
auth.require_context(ctx)?;
}
}
}
validate_acl_modification(auth, &entry.role, &allowed_contexts)?;
let old_set: std::collections::HashSet<&str> =
entry.allowed_contexts.iter().map(String::as_str).collect();
let added: Vec<String> = allowed_contexts
.iter()
.filter(|c| !old_set.contains(c.as_str()))
.cloned()
.collect();
require_contexts_exist(contexts_ks, &added).await?;
entry.allowed_contexts = allowed_contexts;
}
if let Some(replacement) = params.allowed_keys {
validate_allowed_keys(replacement.as_ref())?;
entry.allowed_keys = replacement;
}
if let Some(scope) = params.approve_scope {
validate_approve_scope_grant(auth, &scope)?;
if let ApproveScope::Contexts(ref cs) = scope {
require_contexts_exist(contexts_ks, cs).await?;
}
entry.approve_scope = scope;
}
store_acl_entry(acl_ks, &entry).await?;
info!(channel, did = %did, "ACL entry updated");
audit!(
"acl.update",
actor = &auth.did,
resource = did,
outcome = "success"
);
let _ = audit::record(
audit_ks,
"acl.update",
&auth.did,
Some(did),
"success",
Some(channel),
None,
)
.await;
Ok(to_result_body(&entry))
}
#[allow(clippy::too_many_arguments)]
pub async fn change_role(
acl_ks: &KeyspaceHandle,
audit_ks: &KeyspaceHandle,
auth: &AuthClaims,
subject: &str,
from_role: &str,
to_role: &str,
reason: Option<&str>,
channel: &str,
) -> Result<CreateAclResultBody, AppError> {
auth.require_admin()?;
let mut entry = get_acl_entry(acl_ks, subject)
.await?
.ok_or_else(|| AppError::NotFound(format!("ACL entry not found for DID: {subject}")))?;
if !is_acl_entry_visible(auth, &entry) {
return Err(not_manageable(auth, &entry, subject, "change-role"));
}
let parse = |raw: &str| -> Result<Role, AppError> {
Role::parse(raw).map_err(|_| {
AppError::Validation(format!(
"role not recognized: {raw}; expected one of admin, initiator, \
application, reader, monitor"
))
})
};
let from = parse(from_role)?;
let to = parse(to_role)?;
if entry.role != from {
return Err(AppError::Conflict(format!(
"role of {subject} is {}, not {from} — the change was based on stale state. \
Re-read the entry and retry with the current role",
entry.role,
)));
}
validate_role_assignment(auth, &to)?;
validate_acl_modification(auth, &to, &entry.allowed_contexts)?;
let expected_version = entry.version;
entry.role = to;
update_acl_entry_versioned(acl_ks, entry.clone(), expected_version).await?;
info!(channel, did = %subject, from = %from_role, to = %to_role, "ACL role changed");
audit!(
"acl.change_role",
actor = &auth.did,
resource = subject,
outcome = "success"
);
let _ = audit::record_with_detail(
audit_ks,
"acl.change_role",
&auth.did,
Some(subject),
"success",
Some(channel),
None,
reason,
)
.await;
Ok(to_result_body(&entry))
}
pub async fn delete_acl(
acl_ks: &KeyspaceHandle,
audit_ks: &KeyspaceHandle,
auth: &AuthClaims,
did: &str,
channel: &str,
) -> Result<(), AppError> {
auth.require_manage()?;
if auth.did == did {
return Err(AppError::Conflict(
"cannot delete your own ACL entry".into(),
));
}
let entry = get_acl_entry(acl_ks, did)
.await?
.ok_or_else(|| AppError::NotFound(format!("ACL entry not found for DID: {did}")))?;
if !is_acl_entry_visible(auth, &entry) {
return Err(not_manageable(auth, &entry, did, "delete"));
}
validate_role_assignment(auth, &entry.role)?;
delete_acl_entry(acl_ks, did).await?;
info!(channel, caller = %auth.did, did = %did, "ACL entry deleted");
audit!(
"acl.delete",
actor = &auth.did,
resource = did,
outcome = "success"
);
let _ = audit::record(
audit_ks,
"acl.delete",
&auth.did,
Some(did),
"success",
Some(channel),
None,
)
.await;
Ok(())
}
#[allow(clippy::too_many_arguments)]
pub async fn swap_acl(
acl_ks: &KeyspaceHandle,
audit_ks: &KeyspaceHandle,
auth: &AuthClaims,
presentation: &str,
did_resolver: &DIDCacheClient,
vta_did: &str,
channel: &str,
) -> Result<CreateAclResultBody, AppError> {
let pres = AclSwapPresentation::new(presentation);
let claimed = pres
.peek_holder()
.map_err(|e| AppError::Authentication(format!("swap presentation: {e}")))?;
let resolved = did_resolver
.resolve(&claimed)
.await
.map_err(|e| AppError::Validation(format!("resolve new DID {claimed}: {e}")))?;
let doc = serde_json::to_value(&resolved.doc)?;
let now = now_epoch();
let verified = pres
.verify(&doc, vta_did, now)
.map_err(|e| AppError::Authentication(format!("swap presentation: {e}")))?;
let new_did = verified.holder().to_string();
if new_did == auth.did {
return Err(AppError::Conflict(
"new DID equals current DID; nothing to swap".into(),
));
}
let old = get_acl_entry(acl_ks, &auth.did)
.await?
.ok_or_else(|| AppError::NotFound(format!("no ACL entry for caller: {}", auth.did)))?;
if get_acl_entry(acl_ks, &new_did).await?.is_some() {
return Err(AppError::Conflict(format!(
"ACL entry already exists for DID: {new_did}"
)));
}
let entry = AclEntry::new(new_did.clone(), old.role.clone(), auth.did.clone())
.with_label(old.label.clone())
.with_contexts(old.allowed_contexts.clone())
.with_created_at(now)
.with_kind(old.kind.clone())
.with_capabilities(old.capabilities.clone())
.with_device(old.device.clone());
store_acl_entry(acl_ks, &entry).await?;
delete_acl_entry(acl_ks, &auth.did).await?;
info!(
channel,
old = %auth.did,
new = %new_did,
role = %entry.role,
old_expires_at = ?old.expires_at,
new_expires_at = ?entry.expires_at,
"ACL entry swapped; long-term entry is permanent (ephemeral TTL not inherited)"
);
audit!(
"acl.swap",
actor = &auth.did,
resource = &new_did,
outcome = "success"
);
let _ = audit::record(
audit_ks,
"acl.swap",
&auth.did,
Some(&new_did),
"success",
Some(channel),
None,
)
.await;
Ok(to_result_body(&entry))
}
use vta_sdk::protocols::acl_management::entry::{AclEntry as WireAclEntry, to_epoch};
pub async fn grant_from_entry(
acl_ks: &KeyspaceHandle,
audit_ks: &KeyspaceHandle,
contexts_ks: &KeyspaceHandle,
auth: &AuthClaims,
entry: WireAclEntry,
channel: &str,
) -> Result<CreateAclResponseBody, AppError> {
let role = Role::parse(&entry.role)
.map_err(|_| AppError::Validation(format!("invalid role: {}", entry.role)))?;
let stored = create_acl(
acl_ks,
audit_ks,
contexts_ks,
auth,
&entry.subject,
role,
entry.label.clone(),
entry.scopes.clone(),
entry.expires_at.map(to_epoch),
entry.step_up_approver(),
entry.step_up_require(),
entry.approve_scope(),
entry
.allowed_keys
.clone()
.map(|keys| keys.into_iter().collect()),
channel,
)
.await?;
Ok(CreateAclResponseBody {
entry: WireAclEntry::from_result(&stored),
})
}
pub async fn show_by_subject(
acl_ks: &KeyspaceHandle,
auth: &AuthClaims,
subject: &str,
channel: &str,
) -> Result<GetAclResultBody, AppError> {
let stored = get_acl(acl_ks, auth, subject, channel).await?;
Ok(GetAclResultBody {
entry: WireAclEntry::from_result(&stored),
redacted_fields: Vec::new(),
})
}
pub async fn list_entries(
acl_ks: &KeyspaceHandle,
auth: &AuthClaims,
scope: Option<&str>,
direction: ContextDirection,
channel: &str,
) -> Result<ListAclResultBody, AppError> {
let stored = list_acl(acl_ks, auth, scope, direction, channel).await?;
Ok(ListAclResultBody {
entries: stored.iter().map(WireAclEntry::from_result).collect(),
truncated: false,
cursor: None,
redacted_fields: Vec::new(),
})
}
pub async fn update_from_params(
acl_ks: &KeyspaceHandle,
audit_ks: &KeyspaceHandle,
contexts_ks: &KeyspaceHandle,
auth: &AuthClaims,
subject: &str,
params: UpdateAclParams,
channel: &str,
) -> Result<CreateAclResponseBody, AppError> {
let stored = update_acl(
acl_ks,
audit_ks,
contexts_ks,
auth,
subject,
params,
channel,
)
.await?;
Ok(CreateAclResponseBody {
entry: WireAclEntry::from_result(&stored),
})
}
pub async fn revoke_by_subject(
acl_ks: &KeyspaceHandle,
audit_ks: &KeyspaceHandle,
auth: &AuthClaims,
subject: &str,
scopes: Option<Vec<String>>,
channel: &str,
) -> Result<DeleteAclResultBody, AppError> {
if let Some(scopes) = scopes.filter(|s| !s.is_empty()) {
return Err(AppError::Validation(format!(
"scope reduction is not supported by this maintainer — `scopes` named {} entr{}, \
but only full revocation is implemented. Omit `scopes` to remove the entry.",
scopes.len(),
if scopes.len() == 1 { "y" } else { "ies" }
)));
}
let stored = get_acl(acl_ks, auth, subject, channel).await?;
delete_acl(acl_ks, audit_ks, auth, subject, channel).await?;
Ok(DeleteAclResultBody {
entry: WireAclEntry::from_result(&stored),
})
}
#[cfg(test)]
mod tests {
use super::*;
use crate::acl::{AclEntry, store_acl_entry};
use crate::store::Store;
use vti_common::config::StoreConfig;
async fn fresh_store() -> (
Store,
KeyspaceHandle,
KeyspaceHandle,
KeyspaceHandle,
tempfile::TempDir,
) {
let dir = tempfile::tempdir().unwrap();
let store = Store::open(&StoreConfig {
data_dir: dir.path().into(),
})
.unwrap();
let acl_ks = store.keyspace(crate::keyspaces::ACL).unwrap();
let audit_ks = store.keyspace(crate::keyspaces::AUDIT).unwrap();
let contexts_ks = store.keyspace(crate::keyspaces::CONTEXTS).unwrap();
(store, acl_ks, audit_ks, contexts_ks, dir)
}
async fn seed_contexts(contexts_ks: &KeyspaceHandle, ids: &[&str]) {
use crate::contexts::{ContextRecord, store_context};
use chrono::Utc;
for (i, id) in ids.iter().enumerate() {
let now = Utc::now();
store_context(
contexts_ks,
&ContextRecord {
id: (*id).into(),
name: (*id).into(),
did: None,
description: None,
parent: None,
base_path: format!("m/26'/2'/{i}'"),
index: i as u32,
created_at: now,
updated_at: now,
context_policy: None,
},
)
.await
.unwrap();
}
}
fn ctx_admin(did: &str, contexts: &[&str]) -> AuthClaims {
AuthClaims {
did: did.into(),
role: Role::Admin,
allowed_contexts: contexts.iter().map(|s| s.to_string()).collect(),
session_id: "test-session".into(),
access_expires_at: 0,
amr: Vec::new(),
acr: String::new(),
}
}
fn super_admin(did: &str) -> AuthClaims {
ctx_admin(did, &[])
}
async fn seed_target(acl_ks: &KeyspaceHandle, did: &str, contexts: &[&str]) {
store_acl_entry(
acl_ks,
&AclEntry::new(did, Role::Admin, "seed")
.with_contexts(contexts.iter().map(|s| s.to_string()).collect()),
)
.await
.unwrap();
}
async fn seed_foreign_admin_conferring_into(
acl_ks: &KeyspaceHandle,
did: &str,
admins: &[&str],
confers: &[&str],
) {
store_acl_entry(
acl_ks,
&AclEntry::new(did, Role::Admin, "seed")
.with_contexts(admins.iter().map(|s| s.to_string()).collect())
.with_approve_scope(ApproveScope::Contexts(
confers.iter().map(|s| s.to_string()).collect(),
)),
)
.await
.unwrap();
}
#[tokio::test]
async fn get_acl_surfaces_an_approver_conferring_into_callers_context() {
let (_store, acl_ks, audit_ks, contexts_ks, _dir) = fresh_store().await;
seed_contexts(&contexts_ks, &["ctx-a"]).await;
let target = "did:key:zApproverRead";
store_acl_entry(
&acl_ks,
&AclEntry::new(target, Role::Reader, "seed")
.with_approve_scope(ApproveScope::Contexts(vec!["ctx-a".into()])),
)
.await
.unwrap();
let caller = ctx_admin("did:key:zCtxAdminA", &["ctx-a"]);
let body = get_acl(&acl_ks, &caller, target, "test")
.await
.expect("an approver in my context must be auditable");
assert_eq!(body.did, target);
let _ = (audit_ks, contexts_ks);
}
#[tokio::test]
async fn list_acl_includes_an_approver_conferring_into_callers_context() {
let (_store, acl_ks, _audit_ks, contexts_ks, _dir) = fresh_store().await;
seed_contexts(&contexts_ks, &["ctx-a"]).await;
store_acl_entry(
&acl_ks,
&AclEntry::new("did:key:zApproverList", Role::Reader, "seed")
.with_approve_scope(ApproveScope::All),
)
.await
.unwrap();
let caller = ctx_admin("did:key:zCtxAdminA", &["ctx-a"]);
let body = list_acl(&acl_ks, &caller, None, ContextDirection::default(), "test")
.await
.unwrap();
assert!(
body.iter().any(|e| e.did == "did:key:zApproverList"),
"an approve-all holder must be auditable by every context admin"
);
}
async fn seed_tenant_subtree(acl_ks: &KeyspaceHandle) {
seed_target(acl_ks, "did:key:zUnitAdmin", &["acme/eng"]).await;
seed_target(acl_ks, "did:key:zTenantAdmin", &["acme"]).await;
seed_target(acl_ks, "did:key:zGateway", &["acme/eng/attestation"]).await;
seed_target(acl_ks, "did:key:zSigner", &["acme/eng/signing"]).await;
seed_target(acl_ks, "did:key:zOps", &["acme/ops/keys"]).await;
}
async fn dids_for(
acl_ks: &KeyspaceHandle,
caller: &AuthClaims,
ctx: &str,
direction: ContextDirection,
) -> Vec<String> {
let body = list_acl(acl_ks, caller, Some(ctx), direction, "test")
.await
.unwrap();
let mut dids: Vec<String> = body.into_iter().map(|e| e.did).collect();
dids.sort();
dids
}
#[tokio::test]
async fn a_subtree_listing_returns_the_leaves_an_act_in_listing_omits() {
let (_store, acl_ks, _audit_ks, _contexts_ks, _dir) = fresh_store().await;
seed_tenant_subtree(&acl_ks).await;
let caller = super_admin("did:key:zRoot");
assert_eq!(
dids_for(&acl_ks, &caller, "acme/eng", ContextDirection::ActingIn).await,
["did:key:zTenantAdmin", "did:key:zUnitAdmin"],
"act-in: authority over the unit, which is not what a sweep revokes"
);
assert_eq!(
dids_for(&acl_ks, &caller, "acme/eng", ContextDirection::Subtree).await,
["did:key:zGateway", "did:key:zSigner", "did:key:zUnitAdmin"],
"subtree: every grant inside the unit, and nothing from acme/ops"
);
assert_eq!(
dids_for(&acl_ks, &caller, "acme/eng", ContextDirection::Any).await,
[
"did:key:zGateway",
"did:key:zSigner",
"did:key:zTenantAdmin",
"did:key:zUnitAdmin"
],
"any: the union — the auditor's question"
);
}
#[tokio::test]
async fn the_default_direction_is_the_historical_filter() {
let (_store, acl_ks, _audit_ks, _contexts_ks, _dir) = fresh_store().await;
seed_tenant_subtree(&acl_ks).await;
let caller = super_admin("did:key:zRoot");
assert_eq!(
dids_for(&acl_ks, &caller, "acme/eng", ContextDirection::default()).await,
dids_for(&acl_ks, &caller, "acme/eng", ContextDirection::ActingIn).await,
);
}
#[tokio::test]
async fn a_subtree_listing_does_not_widen_what_a_context_admin_can_see() {
let (_store, acl_ks, _audit_ks, _contexts_ks, _dir) = fresh_store().await;
seed_tenant_subtree(&acl_ks).await;
let caller = ctx_admin("did:key:zOpsAdmin", &["acme/ops"]);
for direction in ContextDirection::ALL {
let dids = dids_for(&acl_ks, &caller, "acme/eng", direction).await;
assert!(
dids.is_empty(),
"{direction} leaked acme/eng entries to an acme/ops admin: {dids:?}"
);
}
}
#[tokio::test]
async fn a_direction_without_a_context_is_refused() {
let (_store, acl_ks, _audit_ks, _contexts_ks, _dir) = fresh_store().await;
seed_tenant_subtree(&acl_ks).await;
let caller = super_admin("did:key:zRoot");
let err = list_acl(&acl_ks, &caller, None, ContextDirection::Subtree, "test")
.await
.expect_err("a direction needs a context");
assert!(matches!(err, AppError::Validation(_)), "got {err:?}");
list_acl(&acl_ks, &caller, None, ContextDirection::default(), "test")
.await
.expect("the default direction with no context is just an unfiltered list");
}
#[tokio::test]
async fn delete_acl_refuses_an_entry_that_acts_outside_callers_contexts() {
let (_store, acl_ks, audit_ks, contexts_ks, _dir) = fresh_store().await;
seed_contexts(&contexts_ks, &["ctx-a", "ctx-b"]).await;
let target = "did:key:zForeignAdmin";
seed_foreign_admin_conferring_into(&acl_ks, target, &["ctx-b"], &["ctx-a"]).await;
let caller = ctx_admin("did:key:zCtxAdminA", &["ctx-a"]);
get_acl(&acl_ks, &caller, target, "test")
.await
.expect("confers into ctx-a, so ctx-a's admin may read it");
let err = delete_acl(&acl_ks, &audit_ks, &caller, target, "test")
.await
.expect_err("ctx-a admin must not delete an entry that administers ctx-b");
assert!(
matches!(err, AppError::Forbidden(_)),
"expected Forbidden (the row is already readable), got {err:?}"
);
}
#[tokio::test]
async fn delete_acl_still_conflates_a_wholly_invisible_entry_to_not_found() {
let (_store, acl_ks, audit_ks, contexts_ks, _dir) = fresh_store().await;
seed_contexts(&contexts_ks, &["ctx-a", "ctx-b"]).await;
let target = "did:key:zInvisible";
seed_target(&acl_ks, target, &["ctx-b"]).await;
let caller = ctx_admin("did:key:zCtxAdminA", &["ctx-a"]);
let err = delete_acl(&acl_ks, &audit_ks, &caller, target, "test")
.await
.expect_err("not visible, not auditable");
assert!(
matches!(err, AppError::NotFound(_)),
"expected NotFound so existence is not disclosed, got {err:?}"
);
}
#[test]
fn parse_step_up_require_accepts_self_and_delegated_only() {
assert_eq!(parse_step_up_require(None).unwrap(), None);
assert_eq!(parse_step_up_require(Some("")).unwrap(), None);
assert_eq!(parse_step_up_require(Some(" ")).unwrap(), None);
assert_eq!(
parse_step_up_require(Some("self")).unwrap(),
Some(StepUpMode::SelfApprove)
);
assert_eq!(
parse_step_up_require(Some("delegated")).unwrap(),
Some(StepUpMode::Delegated)
);
assert!(parse_step_up_require(Some("delegated-any")).is_err());
assert!(parse_step_up_require(Some("none")).is_err());
assert!(parse_step_up_require(Some("nope")).is_err());
}
#[test]
fn step_up_require_round_trips_to_wire() {
assert_eq!(step_up_require_to_wire(None), None);
assert_eq!(
step_up_require_to_wire(Some(StepUpMode::SelfApprove)).as_deref(),
Some("self")
);
assert_eq!(
step_up_require_to_wire(Some(StepUpMode::Delegated)).as_deref(),
Some("delegated")
);
}
#[test]
fn symmetric_difference_handles_typical_cases() {
let s = symmetric_difference_contexts(&["a".into(), "b".into()], &["a".into(), "c".into()]);
let mut s = s;
s.sort();
assert_eq!(s, vec!["b".to_string(), "c".to_string()]);
assert!(
symmetric_difference_contexts(&["a".into(), "b".into()], &["b".into(), "a".into()])
.is_empty()
);
let s = symmetric_difference_contexts(&[], &["x".into()]);
assert_eq!(s, vec!["x".to_string()]);
let s = symmetric_difference_contexts(&["x".into()], &[]);
assert_eq!(s, vec!["x".to_string()]);
}
#[test]
fn to_result_body_echoes_approve_scope() {
let base = AclEntry::new("did:key:zA", Role::Reader, "did:key:zC");
let scoped = base
.clone()
.with_approve_scope(ApproveScope::Contexts(vec!["openvtc".into()]));
let body = to_result_body(&scoped);
assert!(!body.approve_all_contexts);
assert_eq!(body.approve_contexts, vec!["openvtc"]);
let all = base.clone().with_approve_scope(ApproveScope::All);
let body = to_result_body(&all);
assert!(body.approve_all_contexts);
assert!(body.approve_contexts.is_empty());
let body = to_result_body(&base);
assert!(!body.approve_all_contexts);
assert!(body.approve_contexts.is_empty());
}
#[test]
fn acl_entry_can_confer_matches_scope_and_admin_but_not_a_sibling_context() {
let scoped = AclEntry::new("did:key:zApprover", Role::Reader, "seed")
.with_approve_scope(ApproveScope::Contexts(vec!["openvtc".into()]));
assert!(acl_entry_can_confer(&scoped, "openvtc"));
assert!(
!acl_entry_can_confer(&scoped, "openvtc-glenn"),
"approve-scope `openvtc` must NOT confer the distinct context `openvtc-glenn`"
);
let all = AclEntry::new("did:key:zAll", Role::Reader, "seed")
.with_approve_scope(ApproveScope::All);
assert!(acl_entry_can_confer(&all, "openvtc-glenn"));
let ctx_admin = AclEntry::new("did:key:zAdmin", Role::Admin, "seed")
.with_contexts(vec!["openvtc-glenn".into()]);
assert!(acl_entry_can_confer(&ctx_admin, "openvtc-glenn"));
assert!(!acl_entry_can_confer(&ctx_admin, "some-other-context"));
let super_admin = AclEntry::new("did:key:zSuper", Role::Admin, "seed");
assert!(acl_entry_can_confer(&super_admin, "openvtc-glenn"));
let reader = AclEntry::new("did:key:zReader", Role::Reader, "seed");
assert!(!acl_entry_can_confer(&reader, "openvtc-glenn"));
}
#[tokio::test]
async fn update_acl_sets_and_revokes_approve_scope() {
let (_store, acl_ks, audit_ks, contexts_ks, _dir) = fresh_store().await;
seed_contexts(&contexts_ks, &["ctx-a", "ctx-b"]).await;
let target = "did:key:zApprover";
seed_target(&acl_ks, target, &["ctx-a"]).await;
let admin = super_admin("did:key:zRoot");
let set = |scope| UpdateAclParams {
allowed_keys: None,
role: None,
label: None,
allowed_contexts: None,
step_up_approver: None,
step_up_require: None,
approve_scope: Some(scope),
};
update_acl(
&acl_ks,
&audit_ks,
&contexts_ks,
&admin,
target,
set(ApproveScope::All),
"test",
)
.await
.expect("super admin may grant approve-all");
assert_eq!(
get_acl_entry(&acl_ks, target)
.await
.unwrap()
.unwrap()
.approve_scope,
ApproveScope::All
);
update_acl(
&acl_ks,
&audit_ks,
&contexts_ks,
&admin,
target,
set(ApproveScope::Contexts(vec!["ctx-b".into()])),
"test",
)
.await
.expect("narrowing to a scoped grant");
assert_eq!(
get_acl_entry(&acl_ks, target)
.await
.unwrap()
.unwrap()
.approve_scope,
ApproveScope::Contexts(vec!["ctx-b".into()])
);
update_acl(
&acl_ks,
&audit_ks,
&contexts_ks,
&admin,
target,
set(ApproveScope::None),
"test",
)
.await
.expect("revoking confers nothing");
assert_eq!(
get_acl_entry(&acl_ks, target)
.await
.unwrap()
.unwrap()
.approve_scope,
ApproveScope::None
);
}
#[tokio::test]
async fn update_acl_sets_empties_and_clears_the_key_filter() {
use std::collections::BTreeSet;
let (_store, acl_ks, audit_ks, contexts_ks, _dir) = fresh_store().await;
seed_contexts(&contexts_ks, &["ctx-a"]).await;
let target = "did:key:zSigner";
seed_target(&acl_ks, target, &["ctx-a"]).await;
let admin = super_admin("did:key:zRoot");
let set = |replacement| UpdateAclParams {
role: None,
label: None,
allowed_contexts: None,
step_up_approver: None,
step_up_require: None,
approve_scope: None,
allowed_keys: replacement,
};
let stored_filter = |acl_ks: &KeyspaceHandle| {
let acl_ks = acl_ks.clone();
async move {
get_acl_entry(&acl_ks, target)
.await
.unwrap()
.unwrap()
.allowed_keys
}
};
let keys: BTreeSet<String> = ["key-1".to_string()].into_iter().collect();
update_acl(
&acl_ks,
&audit_ks,
&contexts_ks,
&admin,
target,
set(Some(Some(keys.clone()))),
"test",
)
.await
.expect("setting the filter");
assert_eq!(stored_filter(&acl_ks).await, Some(keys.clone()));
update_acl(
&acl_ks,
&audit_ks,
&contexts_ks,
&admin,
target,
UpdateAclParams {
label: Some("relabelled".into()),
..set(None)
},
"test",
)
.await
.unwrap();
assert_eq!(
stored_filter(&acl_ks).await,
Some(keys),
"omitting allowedKeys must leave the filter unchanged"
);
update_acl(
&acl_ks,
&audit_ks,
&contexts_ks,
&admin,
target,
set(Some(Some(BTreeSet::new()))),
"test",
)
.await
.expect("narrowing to no keys");
assert_eq!(stored_filter(&acl_ks).await, Some(BTreeSet::new()));
update_acl(
&acl_ks,
&audit_ks,
&contexts_ks,
&admin,
target,
set(Some(None)),
"test",
)
.await
.expect("clearing the filter");
assert_eq!(stored_filter(&acl_ks).await, None);
let err = update_acl(
&acl_ks,
&audit_ks,
&contexts_ks,
&admin,
target,
set(Some(Some(["".to_string()].into_iter().collect()))),
"test",
)
.await
.expect_err("an empty key id must be refused");
assert!(matches!(err, AppError::Validation(_)), "got {err:?}");
}
#[tokio::test]
async fn update_acl_leaves_approve_scope_alone_when_absent() {
let (_store, acl_ks, audit_ks, contexts_ks, _dir) = fresh_store().await;
seed_contexts(&contexts_ks, &["ctx-a"]).await;
let target = "did:key:zApprover";
seed_target(&acl_ks, target, &["ctx-a"]).await;
let admin = super_admin("did:key:zRoot");
update_acl(
&acl_ks,
&audit_ks,
&contexts_ks,
&admin,
target,
UpdateAclParams {
allowed_keys: None,
role: None,
label: None,
allowed_contexts: None,
step_up_approver: None,
step_up_require: None,
approve_scope: Some(ApproveScope::All),
},
"test",
)
.await
.unwrap();
update_acl(
&acl_ks,
&audit_ks,
&contexts_ks,
&admin,
target,
UpdateAclParams {
allowed_keys: None,
role: None,
label: Some("renamed".into()),
allowed_contexts: None,
step_up_approver: None,
step_up_require: None,
approve_scope: None,
},
"test",
)
.await
.unwrap();
let entry = get_acl_entry(&acl_ks, target).await.unwrap().unwrap();
assert_eq!(entry.label.as_deref(), Some("renamed"));
assert_eq!(
entry.approve_scope,
ApproveScope::All,
"scope must survive an unrelated edit"
);
}
#[tokio::test]
async fn update_acl_applies_the_same_approve_grant_check_as_create() {
let (_store, acl_ks, audit_ks, contexts_ks, _dir) = fresh_store().await;
seed_contexts(&contexts_ks, &["ctx-a", "ctx-b"]).await;
let target = "did:key:zApprover";
seed_target(&acl_ks, target, &["ctx-a"]).await;
let ctx_admin_a = ctx_admin("did:key:zCallerA", &["ctx-a"]);
let err = update_acl(
&acl_ks,
&audit_ks,
&contexts_ks,
&ctx_admin_a,
target,
UpdateAclParams {
allowed_keys: None,
role: None,
label: None,
allowed_contexts: None,
step_up_approver: None,
step_up_require: None,
approve_scope: Some(ApproveScope::All),
},
"test",
)
.await
.unwrap_err();
assert!(
matches!(err, AppError::Forbidden(_)),
"approve-all is super-admin only: {err:?}"
);
let err = update_acl(
&acl_ks,
&audit_ks,
&contexts_ks,
&ctx_admin_a,
target,
UpdateAclParams {
allowed_keys: None,
role: None,
label: None,
allowed_contexts: None,
step_up_approver: None,
step_up_require: None,
approve_scope: Some(ApproveScope::Contexts(vec!["ctx-b".into()])),
},
"test",
)
.await
.unwrap_err();
assert!(
matches!(err, AppError::Forbidden(_)),
"cannot confer a context it lacks: {err:?}"
);
}
#[tokio::test]
async fn update_acl_rejects_shrink_across_caller_scope() {
let (_store, acl_ks, audit_ks, contexts_ks, _dir) = fresh_store().await;
seed_contexts(&contexts_ks, &["ctx-a", "ctx-b"]).await;
let target = "did:key:zTarget";
seed_target(&acl_ks, target, &["ctx-a", "ctx-b"]).await;
let caller = ctx_admin("did:key:zCallerA", &["ctx-a"]);
let err = update_acl(
&acl_ks,
&audit_ks,
&contexts_ks,
&caller,
target,
UpdateAclParams {
allowed_keys: None,
role: None,
label: None,
step_up_approver: None,
step_up_require: None,
allowed_contexts: Some(vec!["ctx-a".into()]),
approve_scope: None,
},
"test",
)
.await
.unwrap_err();
assert!(matches!(err, AppError::Forbidden(_)), "got {err:?}");
}
#[tokio::test]
async fn update_acl_allows_remove_within_caller_scope() {
let (_store, acl_ks, audit_ks, contexts_ks, _dir) = fresh_store().await;
seed_contexts(&contexts_ks, &["ctx-a", "ctx-b"]).await;
let target = "did:key:zTarget2";
seed_target(&acl_ks, target, &["ctx-a", "ctx-b"]).await;
let caller = ctx_admin("did:key:zCallerAB", &["ctx-a", "ctx-b"]);
let body = update_acl(
&acl_ks,
&audit_ks,
&contexts_ks,
&caller,
target,
UpdateAclParams {
allowed_keys: None,
role: None,
label: None,
step_up_approver: None,
step_up_require: None,
allowed_contexts: Some(vec!["ctx-a".into()]),
approve_scope: None,
},
"test",
)
.await
.unwrap();
assert_eq!(body.allowed_contexts, vec!["ctx-a".to_string()]);
}
#[tokio::test]
async fn update_acl_rejects_add_outside_caller_scope() {
let (_store, acl_ks, audit_ks, contexts_ks, _dir) = fresh_store().await;
seed_contexts(&contexts_ks, &["ctx-a", "ctx-b"]).await;
let target = "did:key:zTarget3";
seed_target(&acl_ks, target, &["ctx-a"]).await;
let caller = ctx_admin("did:key:zCallerA", &["ctx-a"]);
let err = update_acl(
&acl_ks,
&audit_ks,
&contexts_ks,
&caller,
target,
UpdateAclParams {
allowed_keys: None,
role: None,
label: None,
step_up_approver: None,
step_up_require: None,
allowed_contexts: Some(vec!["ctx-a".into(), "ctx-b".into()]),
approve_scope: None,
},
"test",
)
.await
.unwrap_err();
assert!(matches!(err, AppError::Forbidden(_)), "got {err:?}");
}
#[tokio::test]
async fn create_acl_rejects_unknown_context() {
let (_store, acl_ks, audit_ks, contexts_ks, _dir) = fresh_store().await;
seed_contexts(&contexts_ks, &["ctx-real"]).await;
let caller = AuthClaims {
did: "did:key:zSuper".into(),
role: Role::Admin,
allowed_contexts: Vec::new(),
session_id: "test-session".into(),
access_expires_at: 0,
amr: Vec::new(),
acr: String::new(),
};
let err = create_acl(
&acl_ks,
&audit_ks,
&contexts_ks,
&caller,
"did:key:zNewAdmin",
Role::Admin,
None,
vec!["ctx-typo".into()],
None,
None,
None,
ApproveScope::None,
None,
"test",
)
.await
.unwrap_err();
assert!(matches!(err, AppError::NotFound(_)), "got {err:?}");
}
#[tokio::test]
async fn create_acl_accepts_known_context() {
let (_store, acl_ks, audit_ks, contexts_ks, _dir) = fresh_store().await;
seed_contexts(&contexts_ks, &["ctx-real"]).await;
let caller = AuthClaims {
did: "did:key:zSuper".into(),
role: Role::Admin,
allowed_contexts: Vec::new(),
session_id: "test-session".into(),
access_expires_at: 0,
amr: Vec::new(),
acr: String::new(),
};
let body = create_acl(
&acl_ks,
&audit_ks,
&contexts_ks,
&caller,
"did:key:zNewAdmin",
Role::Admin,
None,
vec!["ctx-real".into()],
None,
None,
None,
ApproveScope::None,
None,
"test",
)
.await
.unwrap();
assert_eq!(body.allowed_contexts, vec!["ctx-real".to_string()]);
}
#[tokio::test]
async fn context_admin_can_mint_a_least_privilege_approver() {
let (_store, acl_ks, audit_ks, contexts_ks, _dir) = fresh_store().await;
seed_contexts(&contexts_ks, &["ctx-a"]).await;
let caller = ctx_admin("did:key:zCtxAdminA", &["ctx-a"]);
let body = create_acl(
&acl_ks,
&audit_ks,
&contexts_ks,
&caller,
"did:key:zApprover",
Role::Reader,
None,
Vec::new(), None,
None,
None,
ApproveScope::Contexts(vec!["ctx-a".into()]),
None,
"test",
)
.await
.expect("a context admin may create a least-privilege approver for its own context");
assert!(
body.allowed_contexts.is_empty(),
"the approver acts nowhere"
);
assert!(body.approve_contexts.contains(&"ctx-a".to_string()));
}
#[tokio::test]
async fn context_admin_cannot_confer_a_foreign_context_via_the_approver() {
let (_store, acl_ks, audit_ks, contexts_ks, _dir) = fresh_store().await;
seed_contexts(&contexts_ks, &["ctx-a", "ctx-b"]).await;
let caller = ctx_admin("did:key:zCtxAdminA", &["ctx-a"]);
let err = create_acl(
&acl_ks,
&audit_ks,
&contexts_ks,
&caller,
"did:key:zApprover",
Role::Reader,
None,
Vec::new(),
None,
None,
None,
ApproveScope::Contexts(vec!["ctx-b".into()]),
None,
"test",
)
.await
.expect_err("conferring ctx-b requires administering it");
assert!(matches!(err, AppError::Forbidden(_)), "got {err:?}");
}
#[tokio::test]
async fn context_admin_still_cannot_mint_a_super_admin() {
let (_store, acl_ks, audit_ks, contexts_ks, _dir) = fresh_store().await;
seed_contexts(&contexts_ks, &["ctx-a"]).await;
let caller = ctx_admin("did:key:zCtxAdminA", &["ctx-a"]);
let err = create_acl(
&acl_ks,
&audit_ks,
&contexts_ks,
&caller,
"did:key:zWouldBeSuper",
Role::Admin,
None,
Vec::new(), None,
None,
None,
ApproveScope::None,
None,
"test",
)
.await
.expect_err("a context admin must not mint a super-admin");
assert!(matches!(err, AppError::Forbidden(_)), "got {err:?}");
}
#[tokio::test]
async fn delete_acl_rejects_initiator_deleting_admin() {
let (_store, acl_ks, audit_ks, contexts_ks, _dir) = fresh_store().await;
seed_contexts(&contexts_ks, &["ctx-shared"]).await;
let admin_target = "did:key:zAdminTarget";
seed_target(&acl_ks, admin_target, &["ctx-shared"]).await;
let caller = AuthClaims {
did: "did:key:zInitiator".into(),
role: Role::Initiator,
allowed_contexts: vec!["ctx-shared".into()],
session_id: "test-session".into(),
access_expires_at: 0,
amr: Vec::new(),
acr: String::new(),
};
let err = delete_acl(&acl_ks, &audit_ks, &caller, admin_target, "test")
.await
.unwrap_err();
assert!(
matches!(err, AppError::Forbidden(_)),
"expected Forbidden, got {err:?}"
);
}
#[tokio::test]
async fn delete_acl_admin_can_delete_admin_entry() {
let (_store, acl_ks, audit_ks, contexts_ks, _dir) = fresh_store().await;
seed_contexts(&contexts_ks, &["ctx-shared"]).await;
let admin_target = "did:key:zAdminTarget2";
seed_target(&acl_ks, admin_target, &["ctx-shared"]).await;
let caller = ctx_admin("did:key:zCallerAdmin", &["ctx-shared"]);
delete_acl(&acl_ks, &audit_ks, &caller, admin_target, "test")
.await
.expect("admin-on-admin delete succeeds");
assert!(
get_acl_entry(&acl_ks, admin_target)
.await
.unwrap()
.is_none(),
"the entry is gone after a delete"
);
}
#[tokio::test]
async fn update_acl_rejects_adding_unknown_context() {
let (_store, acl_ks, audit_ks, contexts_ks, _dir) = fresh_store().await;
seed_contexts(&contexts_ks, &["ctx-a"]).await;
let target = "did:key:zTargetUnknown";
seed_target(&acl_ks, target, &["ctx-a"]).await;
let caller = AuthClaims {
did: "did:key:zSuper".into(),
role: Role::Admin,
allowed_contexts: Vec::new(),
session_id: "test-session".into(),
access_expires_at: 0,
amr: Vec::new(),
acr: String::new(),
};
let err = update_acl(
&acl_ks,
&audit_ks,
&contexts_ks,
&caller,
target,
UpdateAclParams {
allowed_keys: None,
role: None,
label: None,
step_up_approver: None,
step_up_require: None,
allowed_contexts: Some(vec!["ctx-a".into(), "ctx-ghost".into()]),
approve_scope: None,
},
"test",
)
.await
.unwrap_err();
assert!(matches!(err, AppError::NotFound(_)), "got {err:?}");
}
}