rvm-cap 0.1.1

//! Capability granting with monotonic attenuation.
//!
//! Implements the grant semantics from ADR-135:
//! - Source must hold GRANT right
//! - Derived rights must be a subset of source rights
//! - Delegation depth is enforced (max 8)

use crate::error::{CapError, CapResult};
use crate::table::CapSlot;
use crate::DEFAULT_MAX_DELEGATION_DEPTH;
use rvm_types::{CapRights, CapToken};

/// Policy configuration for capability grants.
#[derive(Debug, Clone, Copy)]
pub struct GrantPolicy {
    /// Maximum delegation depth allowed.
    pub max_depth: u8,
    /// Whether `GRANT_ONCE` capabilities are allowed.
    pub allow_grant_once: bool,
}

impl GrantPolicy {
    /// Creates a grant policy with default settings.
    #[must_use]
    pub const fn new() -> Self {
        Self {
            max_depth: DEFAULT_MAX_DELEGATION_DEPTH,
            allow_grant_once: true,
        }
    }

    /// Creates a grant policy with a custom depth limit.
    #[must_use]
    pub const fn with_max_depth(max_depth: u8) -> Self {
        Self {
            max_depth,
            allow_grant_once: true,
        }
    }
}

impl Default for GrantPolicy {
    fn default() -> Self {
        Self::new()
    }
}

/// Validates a grant request and produces the derived token.
///
/// If the source has `GRANT_ONCE` (but not `GRANT`), `consume_grant_once`
/// is set to `true` in the return value so the caller can strip the right
/// from the source slot.
///
/// Returns `(derived_token, depth, consume_grant_once)` on success.
pub fn validate_grant(
    source: &CapSlot,
    requested_rights: CapRights,
    new_id: u64,
    badge: u64,
    epoch: u32,
    policy: GrantPolicy,
) -> CapResult<(CapToken, u8, bool)> {
    let source_rights = source.token.rights();

    let has_grant = source_rights.contains(CapRights::GRANT);
    let has_grant_once = policy.allow_grant_once && source_rights.contains(CapRights::GRANT_ONCE);

    // Source must hold GRANT or GRANT_ONCE to delegate.
    if !has_grant && !has_grant_once {
        return Err(CapError::GrantNotPermitted);
    }

    // Monotonic attenuation: requested must be a subset of source.
    if !source_rights.contains(requested_rights) {
        return Err(CapError::RightsEscalation);
    }

    // Delegation depth check with overflow protection.
    let new_depth = source
        .depth
        .checked_add(1)
        .ok_or(CapError::DelegationDepthExceeded)?;
    if new_depth > policy.max_depth {
        return Err(CapError::DelegationDepthExceeded);
    }

    let _ = badge; // Badge is carried by the slot, not the token.

    let derived_token = CapToken::new(new_id, source.token.cap_type(), requested_rights, epoch);

    // Signal that GRANT_ONCE should be consumed if it was the only
    // grant authority (source has GRANT_ONCE but not GRANT).
    let consume = !has_grant && has_grant_once;

    Ok((derived_token, new_depth, consume))
}

#[cfg(test)]
mod tests {
    use super::*;
    use rvm_types::{CapType, PartitionId};

    fn make_source(rights: CapRights, depth: u8) -> CapSlot {
        CapSlot {
            token: CapToken::new(1, CapType::Region, rights, 0),
            generation: 1,
            owner: PartitionId::new(1),
            depth,
            parent_index: u32::MAX,
            badge: 0,
        }
    }

    fn all_rights() -> CapRights {
        CapRights::READ
            .union(CapRights::WRITE)
            .union(CapRights::EXECUTE)
            .union(CapRights::GRANT)
            .union(CapRights::REVOKE)
    }

    #[test]
    fn test_valid_grant() {
        let source = make_source(all_rights(), 0);
        let policy = GrantPolicy::new();
        let (token, depth, consume) =
            validate_grant(&source, CapRights::READ, 10, 42, 0, policy).unwrap();
        assert_eq!(token.rights(), CapRights::READ);
        assert_eq!(depth, 1);
        assert!(!consume); // Source has full GRANT, so GRANT_ONCE is not consumed.
    }

    #[test]
    fn test_grant_without_grant_right() {
        let source = make_source(CapRights::READ, 0);
        let policy = GrantPolicy::new();
        let result = validate_grant(&source, CapRights::READ, 10, 0, 0, policy);
        assert_eq!(result, Err(CapError::GrantNotPermitted));
    }

    #[test]
    fn test_rights_escalation() {
        let source = make_source(CapRights::READ.union(CapRights::GRANT), 0);
        let policy = GrantPolicy::new();
        let result = validate_grant(&source, CapRights::WRITE, 10, 0, 0, policy);
        assert_eq!(result, Err(CapError::RightsEscalation));
    }

    #[test]
    fn test_depth_limit() {
        let source = make_source(all_rights(), 8);
        let policy = GrantPolicy::new();
        let result = validate_grant(&source, CapRights::READ, 10, 0, 0, policy);
        assert_eq!(result, Err(CapError::DelegationDepthExceeded));
    }

    #[test]
    fn test_grant_preserves_type() {
        let source = CapSlot {
            token: CapToken::new(1, CapType::CommEdge, all_rights(), 5),
            generation: 1,
            owner: PartitionId::new(1),
            depth: 0,
            parent_index: u32::MAX,
            badge: 0,
        };
        let policy = GrantPolicy::new();
        let (token, _, _) = validate_grant(&source, CapRights::READ, 10, 0, 5, policy).unwrap();
        assert_eq!(token.cap_type(), CapType::CommEdge);
        assert_eq!(token.epoch(), 5);
    }

    #[test]
    fn test_grant_at_max_minus_one() {
        let source = make_source(all_rights(), 7);
        let policy = GrantPolicy::new();
        let (_, depth, _) = validate_grant(&source, CapRights::READ, 10, 0, 0, policy).unwrap();
        assert_eq!(depth, 8);
    }

    #[test]
    fn test_grant_once_consumed() {
        // Source has GRANT_ONCE but not GRANT.
        let rights = CapRights::READ.union(CapRights::GRANT_ONCE);
        let source = make_source(rights, 0);
        let policy = GrantPolicy::new();
        let (token, depth, consume) =
            validate_grant(&source, CapRights::READ, 10, 0, 0, policy).unwrap();
        assert_eq!(token.rights(), CapRights::READ);
        assert_eq!(depth, 1);
        assert!(consume); // GRANT_ONCE should be consumed.
    }

    #[test]
    fn test_grant_once_not_consumed_when_grant_also_present() {
        // Source has both GRANT and GRANT_ONCE -- GRANT takes precedence.
        let rights = CapRights::READ
            .union(CapRights::GRANT)
            .union(CapRights::GRANT_ONCE);
        let source = make_source(rights, 0);
        let policy = GrantPolicy::new();
        let (_, _, consume) = validate_grant(&source, CapRights::READ, 10, 0, 0, policy).unwrap();
        assert!(!consume);
    }

    #[test]
    fn test_depth_overflow_protection() {
        let source = make_source(all_rights(), u8::MAX);
        let policy = GrantPolicy::with_max_depth(u8::MAX);
        let result = validate_grant(&source, CapRights::READ, 10, 0, 0, policy);
        assert_eq!(result, Err(CapError::DelegationDepthExceeded));
    }
}