mirror of
https://github.com/microsoft/regorus.git
synced 2026-08-05 02:16:11 +00:00
394625d4bc
- Introduce ExecutionTimer/ExecutionTimerConfig to allow limiting evaluating time. - To amortize time checking costs, checking interval can be configured via the notion of work units - A global fallback time limit can be set to universally limit all evaluation in addition to engine level limit setting. - Implement limnits in interpreter and RVM. In RVM, also handle suspend/resume so that time during pause is not counted. - Add engine-level APIs to set/clear per-engine timer configuration and apply global fallback defaults. - Surface execution-time limits through FFI and C# bindings - Add C# tests and example usage to validate engine overrides, global fallback behavior, and compiled policy enforcement. - Expand docs for execution-time limit - Add interpreter YAML cases and VM unit tests for time-limit behavior and deterministic time sources. Signed-off-by: Anand Krishnamoorthi <anakrish@microsoft.com>
79 lines
2.5 KiB
Rust
79 lines
2.5 KiB
Rust
// Copyright (c) Microsoft Corporation.
|
|
// Licensed under the MIT License.
|
|
|
|
use crate::common::*;
|
|
use crate::panic_guard::with_unwind_guard;
|
|
use alloc::boxed::Box;
|
|
use alloc::string::String;
|
|
use anyhow::Result;
|
|
use core::ffi::c_char;
|
|
use core::ptr;
|
|
|
|
/// Wrapper for `regorus::CompiledPolicy`.
|
|
#[derive(Clone)]
|
|
pub struct RegorusCompiledPolicy {
|
|
pub(crate) compiled_policy: regorus::CompiledPolicy,
|
|
}
|
|
|
|
/// Drop a `RegorusCompiledPolicy`.
|
|
#[no_mangle]
|
|
pub extern "C" fn regorus_compiled_policy_drop(compiled_policy: *mut RegorusCompiledPolicy) {
|
|
if let Ok(cp) = to_ref(compiled_policy) {
|
|
unsafe {
|
|
let _ = Box::from_raw(ptr::from_mut(cp));
|
|
}
|
|
}
|
|
}
|
|
|
|
/// Evaluate the compiled policy with the given input.
|
|
///
|
|
/// For target policies, evaluates the target's effect rule.
|
|
/// For regular policies, evaluates the originally compiled rule.
|
|
///
|
|
/// * `input`: JSON encoded input data (resource) to validate against the policy.
|
|
#[no_mangle]
|
|
pub extern "C" fn regorus_compiled_policy_eval_with_input(
|
|
compiled_policy: *mut RegorusCompiledPolicy,
|
|
input: *const c_char,
|
|
) -> RegorusResult {
|
|
with_unwind_guard(|| {
|
|
let output = || -> Result<String> {
|
|
let input_value = regorus::Value::from_json_str(&from_c_str(input)?)?;
|
|
let result = to_ref(compiled_policy)?
|
|
.compiled_policy
|
|
.eval_with_input(input_value)?;
|
|
result.to_json_str()
|
|
}();
|
|
|
|
match output {
|
|
Ok(out) => RegorusResult::ok_string(out),
|
|
Err(e) => to_regorus_result(Err(e)),
|
|
}
|
|
})
|
|
}
|
|
|
|
/// Configure the execution timer for evaluations of this compiled policy.
|
|
/// Get information about the compiled policy including metadata about modules,
|
|
/// target configuration, and resource types.
|
|
///
|
|
/// Returns a JSON-encoded `PolicyInfo` struct containing comprehensive
|
|
/// information about the compiled policy such as module IDs, target name,
|
|
/// applicable resource types, entry point rule, and parameters.
|
|
#[no_mangle]
|
|
pub extern "C" fn regorus_compiled_policy_get_policy_info(
|
|
compiled_policy: *mut RegorusCompiledPolicy,
|
|
) -> RegorusResult {
|
|
with_unwind_guard(|| {
|
|
let output = || -> Result<String> {
|
|
let info = to_ref(compiled_policy)?.compiled_policy.get_policy_info()?;
|
|
serde_json::to_string(&info)
|
|
.map_err(|e| anyhow::anyhow!("Failed to serialize policy info: {}", e))
|
|
}();
|
|
|
|
match output {
|
|
Ok(out) => RegorusResult::ok_string(out),
|
|
Err(e) => to_regorus_result(Err(e)),
|
|
}
|
|
})
|
|
}
|