mirror of
https://github.com/microsoft/regorus.git
synced 2026-08-05 02:16:11 +00:00
Compare commits
17 Commits
copilot/ad
...
copilot/ad
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
838046ba60 | ||
|
|
9955662e7c | ||
|
|
c7216d8c66 | ||
|
|
c22f148778 | ||
|
|
ab162fcaec | ||
|
|
ed6ae465b0 | ||
|
|
bd90453dd3 | ||
|
|
11940ddb04 | ||
|
|
5b7010ba16 | ||
|
|
ba7d29b134 | ||
|
|
acf7f7a25e | ||
|
|
86b4a279fa | ||
|
|
5467cd9e69 | ||
|
|
dae3052781 | ||
|
|
3111bf58f2 | ||
|
|
be3fde7706 | ||
|
|
d2c483e93e |
2
.github/copilot-setup-steps.yml
vendored
2
.github/copilot-setup-steps.yml
vendored
@@ -8,3 +8,5 @@ steps:
|
||||
- uses: actions/checkout@de0fac2e4500dabe0009e67214ff5f5447ce83dd # v6.0.2
|
||||
with:
|
||||
fetch-depth: 0 # full history needed for git diff against main
|
||||
- run: git fetch origin main:refs/remotes/origin/main
|
||||
name: Ensure origin/main ref is available for diff computation
|
||||
|
||||
28
.github/skills/code-review/SKILL.md
vendored
28
.github/skills/code-review/SKILL.md
vendored
@@ -25,15 +25,21 @@ Key constraints (details in copilot-instructions.md):
|
||||
## Step 1: Get the Diff
|
||||
|
||||
```bash
|
||||
BASE=$(git merge-base upstream/main HEAD 2>/dev/null \
|
||||
|| git merge-base origin/main HEAD 2>/dev/null)
|
||||
if [ -z "$BASE" ]; then
|
||||
echo "ERROR: Cannot find upstream/main or origin/main. Cannot determine review scope."
|
||||
exit 1
|
||||
# Primary: use gh pr diff (works in cloud agent + any PR context).
|
||||
# Fallback: git merge-base for local non-PR usage.
|
||||
if gh pr diff --name-only >/dev/null 2>&1; then
|
||||
echo "---STAT---"
|
||||
gh pr diff --name-only
|
||||
echo "---DIFF---"
|
||||
gh pr diff
|
||||
else
|
||||
BASE=$(git merge-base upstream/main HEAD 2>/dev/null \
|
||||
|| git merge-base origin/main HEAD 2>/dev/null \
|
||||
|| git merge-base main HEAD 2>/dev/null)
|
||||
echo "Reviewing changes since: $BASE"
|
||||
git diff "$BASE"..HEAD --stat
|
||||
git diff "$BASE"..HEAD
|
||||
fi
|
||||
echo "Reviewing changes since: $BASE"
|
||||
git diff "$BASE"..HEAD --stat
|
||||
git diff "$BASE"..HEAD -- '*.rs' '*.toml' 'examples/'
|
||||
```
|
||||
|
||||
If the diff is empty, stop and report: "No changes found to review."
|
||||
@@ -196,3 +202,9 @@ one pass. If any were skipped, note them and briefly assess.
|
||||
### Summary
|
||||
|
||||
X findings (N critical, N high, N medium, N low). One sentence overall assessment.
|
||||
|
||||
### Output
|
||||
|
||||
After generating the report above, write the COMPLETE report to `/tmp/code-review-report.md`
|
||||
using the `create` tool or shell. This ensures the full report is preserved even if
|
||||
display output is truncated.
|
||||
|
||||
59
.github/skills/deep-review/SKILL.md
vendored
59
.github/skills/deep-review/SKILL.md
vendored
@@ -40,22 +40,25 @@ Use `read_agent` with `wait: true` to wait for each background agent.
|
||||
## Step 1: Get the Diff and Build Inventory
|
||||
|
||||
```bash
|
||||
BASE=$(git merge-base upstream/main HEAD 2>/dev/null \
|
||||
|| git merge-base origin/main HEAD 2>/dev/null)
|
||||
if [ -z "$BASE" ]; then
|
||||
echo "ERROR: Cannot find upstream/main or origin/main."
|
||||
exit 1
|
||||
# Primary: use gh pr diff (works in cloud agent + any PR context).
|
||||
# Fallback: git merge-base for local non-PR usage.
|
||||
if gh pr diff --name-only >/dev/null 2>&1; then
|
||||
echo "---STAT---"
|
||||
gh pr diff --name-only
|
||||
echo "---DIFF---"
|
||||
gh pr diff
|
||||
else
|
||||
BASE=$(git merge-base upstream/main HEAD 2>/dev/null \
|
||||
|| git merge-base origin/main HEAD 2>/dev/null \
|
||||
|| git merge-base main HEAD 2>/dev/null)
|
||||
echo "Reviewing changes since: $BASE"
|
||||
git diff "$BASE"..HEAD --stat
|
||||
git diff "$BASE"..HEAD
|
||||
fi
|
||||
echo "Reviewing changes since: $BASE"
|
||||
git diff "$BASE"..HEAD --stat
|
||||
git diff "$BASE"..HEAD -- '*.rs' '*.toml' 'examples/' | head -2000
|
||||
```
|
||||
|
||||
If the diff is empty, stop and report: "No changes found to review."
|
||||
|
||||
**Scope rule:** Focus on code files (`*.rs`, `*.toml`, examples). Do NOT pass
|
||||
docs/config diffs to agents.
|
||||
|
||||
**Build a risk-classified inventory.** List every changed function, struct,
|
||||
impl, trait, pub item, and significant code block. Number them and tag with
|
||||
risk predicates:
|
||||
@@ -106,8 +109,10 @@ Use `model: "gpt-5.4"` in the task tool call (provides model diversity).
|
||||
> Get the diff:
|
||||
> ```
|
||||
> BASE=$(git merge-base upstream/main HEAD 2>/dev/null \
|
||||
> || git merge-base origin/main HEAD 2>/dev/null)
|
||||
> git diff "$BASE"..HEAD -- '*.rs' '*.toml' 'examples/'
|
||||
> || git merge-base origin/main HEAD 2>/dev/null \
|
||||
> || git merge-base main HEAD 2>/dev/null)
|
||||
> # If no merge-base, use: gh pr diff
|
||||
> git diff "$BASE"..HEAD # or: gh pr diff
|
||||
> ```
|
||||
>
|
||||
> Key regorus constraints:
|
||||
@@ -161,8 +166,10 @@ Use `model: "claude-opus-4.6"` in the task tool call.
|
||||
> Get the diff AND read full source files for context:
|
||||
> ```
|
||||
> BASE=$(git merge-base upstream/main HEAD 2>/dev/null \
|
||||
> || git merge-base origin/main HEAD 2>/dev/null)
|
||||
> git diff "$BASE"..HEAD -- '*.rs' '*.toml' 'examples/'
|
||||
> || git merge-base origin/main HEAD 2>/dev/null \
|
||||
> || git merge-base main HEAD 2>/dev/null)
|
||||
> # If no merge-base, use: gh pr diff
|
||||
> git diff "$BASE"..HEAD # or: gh pr diff
|
||||
> ```
|
||||
> Then use `view` to read the full source files that were changed.
|
||||
>
|
||||
@@ -219,8 +226,10 @@ Use the default model (no `model` parameter).
|
||||
> Get the diff:
|
||||
> ```
|
||||
> BASE=$(git merge-base upstream/main HEAD 2>/dev/null \
|
||||
> || git merge-base origin/main HEAD 2>/dev/null)
|
||||
> git diff "$BASE"..HEAD -- '*.rs' '*.toml' 'examples/'
|
||||
> || git merge-base origin/main HEAD 2>/dev/null \
|
||||
> || git merge-base main HEAD 2>/dev/null)
|
||||
> # If no merge-base, use: gh pr diff
|
||||
> git diff "$BASE"..HEAD # or: gh pr diff
|
||||
> ```
|
||||
> Use `view` to read surrounding context.
|
||||
>
|
||||
@@ -439,8 +448,10 @@ Launch **1 general-purpose agent in background mode**.
|
||||
> Get the diff:
|
||||
> ```
|
||||
> BASE=$(git merge-base upstream/main HEAD 2>/dev/null \
|
||||
> || git merge-base origin/main HEAD 2>/dev/null)
|
||||
> git diff "$BASE"..HEAD -- '*.rs' '*.toml' 'examples/'
|
||||
> || git merge-base origin/main HEAD 2>/dev/null \
|
||||
> || git merge-base main HEAD 2>/dev/null)
|
||||
> # If no merge-base, use: gh pr diff
|
||||
> git diff "$BASE"..HEAD # or: gh pr diff
|
||||
> ```
|
||||
> Use `view` to read full source files.
|
||||
>
|
||||
@@ -481,8 +492,8 @@ Launch **1 general-purpose agent in background mode**.
|
||||
|
||||
## Step 5: Synthesize and Report
|
||||
|
||||
**IMPORTANT:** This is the primary output. Everything above was preparation.
|
||||
Keep the report COMPACT — one finding per block, no filler prose.
|
||||
**CRITICAL:** Write the report to `/tmp/deep-review-report.md` FIRST, then display it.
|
||||
Use a shell command to write the file before any other output in this step.
|
||||
|
||||
Apply verdicts from the adversarial verifier:
|
||||
- **CONFIRMED**: keep at stated severity
|
||||
@@ -522,3 +533,9 @@ would catch it. If not, name the minimal test that should exist.
|
||||
X findings (N critical, N high, N medium, N low). Y "likely" findings.
|
||||
Z dropped (one-line reasons).
|
||||
Risk assessment in one sentence.
|
||||
|
||||
---
|
||||
|
||||
**Remember:** The report above MUST be written to `/tmp/deep-review-report.md` at the
|
||||
START of Step 5 (before displaying it). Use shell: `cat > /tmp/deep-review-report.md << 'REPORT_EOF'`
|
||||
... report content ... `REPORT_EOF`
|
||||
|
||||
15
CHANGELOG.md
15
CHANGELOG.md
@@ -6,6 +6,21 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
|
||||
|
||||
## [Unreleased]
|
||||
|
||||
## [0.10.1](https://github.com/microsoft/regorus/compare/regorus-v0.10.0...regorus-v0.10.1) - 2026-05-22
|
||||
|
||||
### Fixed
|
||||
|
||||
- *(ffi)* eliminate aliasing UB + add Azure Policy JSON compilation FFI ([#727](https://github.com/microsoft/regorus/pull/727))
|
||||
- *(interpreter,rvm)* correct partial object rule iteration and classification ([#718](https://github.com/microsoft/regorus/pull/718))
|
||||
- *(copilot)* robust diff computation for cloud agent environments ([#709](https://github.com/microsoft/regorus/pull/709))
|
||||
|
||||
### Other
|
||||
|
||||
- *(azure_policy)* reduce AliasRegistry allocations via Rc sharing ([#725](https://github.com/microsoft/regorus/pull/725))
|
||||
- *(normalizer)* use Rc<str> interning to reduce alias resolution allocations ([#726](https://github.com/microsoft/regorus/pull/726))
|
||||
- *(deps)* bump the rust-dependencies group across 5 directories with 2 updates ([#724](https://github.com/microsoft/regorus/pull/724))
|
||||
- *(deps)* bump the rust-dependencies group across 5 directories with 4 updates ([#717](https://github.com/microsoft/regorus/pull/717))
|
||||
|
||||
## [0.10.0] - 2026-05-05
|
||||
|
||||
### Added
|
||||
|
||||
106
Cargo.lock
generated
106
Cargo.lock
generated
@@ -119,9 +119,9 @@ checksum = "7f202df86484c868dbad7eaa557ef785d5c66295e41b460ef922eca0723b842c"
|
||||
|
||||
[[package]]
|
||||
name = "autocfg"
|
||||
version = "1.5.0"
|
||||
version = "1.5.1"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "c08606f8c3cbf4ce6ec8e28fb0014a2c086708fe954eaa885384a6165172e7e8"
|
||||
checksum = "f2032f911046de80f0a198e0901378627c33f59ea0ac00e363d481118bd70a53"
|
||||
|
||||
[[package]]
|
||||
name = "bit-set"
|
||||
@@ -162,9 +162,9 @@ dependencies = [
|
||||
|
||||
[[package]]
|
||||
name = "bumpalo"
|
||||
version = "3.20.2"
|
||||
version = "3.20.3"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "5d20789868f4b01b2f2caec9f5c4e0213b41e3e5702a50157d699ae31ced2fcb"
|
||||
checksum = "72f5acc6cb2ba439de613abc23857ec3d78374d8ed5ac84e9d11336e87da8649"
|
||||
|
||||
[[package]]
|
||||
name = "bytecount"
|
||||
@@ -180,9 +180,9 @@ checksum = "37b2a672a2cb129a2e41c10b1224bb368f9f37a2b16b612598138befd7b37eb5"
|
||||
|
||||
[[package]]
|
||||
name = "cc"
|
||||
version = "1.2.61"
|
||||
version = "1.2.62"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "d16d90359e986641506914ba71350897565610e87ce0ad9e6f28569db3dd5c6d"
|
||||
checksum = "a1dce859f0832a7d088c4f1119888ab94ef4b5d6795d1ce05afb7fe159d79f98"
|
||||
dependencies = [
|
||||
"find-msvc-tools",
|
||||
"shlex",
|
||||
@@ -402,9 +402,9 @@ checksum = "460fbee9c2c2f33933d720630a6a0bac33ba7053db5344fac858d4b8952d77d5"
|
||||
|
||||
[[package]]
|
||||
name = "dashmap"
|
||||
version = "6.1.0"
|
||||
version = "6.2.1"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "5041cc499144891f3790297212f32a74fb938e5136a14943f338ef9e0ae276cf"
|
||||
checksum = "e6361d5c062261c78a176addb82d4c821ae42bed6089de0e12603cd25de2059c"
|
||||
dependencies = [
|
||||
"cfg-if",
|
||||
"crossbeam-utils",
|
||||
@@ -422,9 +422,9 @@ checksum = "a4ae5f15dda3c708c0ade84bfee31ccab44a3da4f88015ed22f63732abe300c8"
|
||||
|
||||
[[package]]
|
||||
name = "displaydoc"
|
||||
version = "0.2.5"
|
||||
version = "0.2.6"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "97369cbbc041bc366949bc74d34658d6cda5621039731c6310521892a3a20ae0"
|
||||
checksum = "1ac70aa55017e108007fbaf5aa0f54b021c98f92ff8af59d42eda9da96e3dd4f"
|
||||
dependencies = [
|
||||
"proc-macro2 1.0.106",
|
||||
"quote 1.0.45",
|
||||
@@ -433,9 +433,9 @@ dependencies = [
|
||||
|
||||
[[package]]
|
||||
name = "either"
|
||||
version = "1.15.0"
|
||||
version = "1.16.0"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "48c757948c5ede0e46177b7add2e67155f70e33c07fea8284df6576da70b3719"
|
||||
checksum = "91622ff5e7162018101f2fea40d6ebf4a78bbe5a49736a2020649edf9693679e"
|
||||
|
||||
[[package]]
|
||||
name = "email_address"
|
||||
@@ -650,9 +650,9 @@ dependencies = [
|
||||
|
||||
[[package]]
|
||||
name = "hashbrown"
|
||||
version = "0.17.0"
|
||||
version = "0.17.1"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "4f467dd6dccf739c208452f8014c75c18bb8301b050ad1cfb27153803edb0f51"
|
||||
checksum = "ed5909b6e89a2db4456e54cd5f673791d7eca6732202bbf2a9cc504fe2f9b84a"
|
||||
dependencies = [
|
||||
"foldhash 0.2.0",
|
||||
]
|
||||
@@ -835,7 +835,7 @@ source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "d466e9454f08e4a911e14806c24e16fba1b4c121d1ea474396f396069cf949d9"
|
||||
dependencies = [
|
||||
"equivalent",
|
||||
"hashbrown 0.17.0",
|
||||
"hashbrown 0.17.1",
|
||||
"serde",
|
||||
"serde_core",
|
||||
]
|
||||
@@ -869,9 +869,9 @@ checksum = "8f42a60cbdf9a97f5d2305f08a87dc4e09308d1276d28c869c684d7777685682"
|
||||
|
||||
[[package]]
|
||||
name = "js-sys"
|
||||
version = "0.3.98"
|
||||
version = "0.3.99"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "67df7112613f8bfd9150013a0314e196f4800d3201ae742489d999db2f979f08"
|
||||
checksum = "142bc4740e452c1e57ade0cbc129f139c9093e354346f0872ef985f4f5cf5f11"
|
||||
dependencies = [
|
||||
"cfg-if",
|
||||
"futures-util",
|
||||
@@ -881,9 +881,9 @@ dependencies = [
|
||||
|
||||
[[package]]
|
||||
name = "jsonschema"
|
||||
version = "0.46.4"
|
||||
version = "0.46.5"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "fc59d2432e047d6090ba1d83c782d0128bd6203857978218f5614dbd3287281f"
|
||||
checksum = "6a5fe5206f06e589caf25e79fc05ccdf91fca745685fe9fe1a13bbdfb479a631"
|
||||
dependencies = [
|
||||
"ahash",
|
||||
"bytecount",
|
||||
@@ -944,9 +944,9 @@ dependencies = [
|
||||
|
||||
[[package]]
|
||||
name = "log"
|
||||
version = "0.4.29"
|
||||
version = "0.4.30"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "5e5032e24019045c762d3c0f28f5b6b8bbf38563a65908389bf7978758920897"
|
||||
checksum = "616ec5685824bcc94416c6d4a7a446eea774a31efd7062c8480ba6fd06d7a6e5"
|
||||
|
||||
[[package]]
|
||||
name = "lru"
|
||||
@@ -956,9 +956,9 @@ checksum = "8a860605968fce16869fd239cf4237a82f3ac470723415db603b0e8b6c8d4fb9"
|
||||
|
||||
[[package]]
|
||||
name = "memchr"
|
||||
version = "2.8.0"
|
||||
version = "2.8.1"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "f8ca58f447f06ed17d5fc4043ce1b10dd205e060fb3ce5b979b8ed8e59ff3f79"
|
||||
checksum = "6b947ae49db0d222b1dbc6b113ce7248a3fc3a6ca21b696717bfc000ba4484d8"
|
||||
|
||||
[[package]]
|
||||
name = "micromap"
|
||||
@@ -1353,9 +1353,9 @@ dependencies = [
|
||||
|
||||
[[package]]
|
||||
name = "referencing"
|
||||
version = "0.46.4"
|
||||
version = "0.46.5"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "cb674900ca31acd75c4aaf63f48e43e719631c0539ea5a9e64163d1296bcb730"
|
||||
checksum = "69e4e17ef386c5383591d07623d3de49cbc601156e7582973e6db98d66a57de2"
|
||||
dependencies = [
|
||||
"ahash",
|
||||
"fluent-uri",
|
||||
@@ -1399,7 +1399,7 @@ checksum = "dc897dd8d9e8bd1ed8cdad82b5966c3e0ecae09fb1907d58efaa013543185d0a"
|
||||
|
||||
[[package]]
|
||||
name = "regorus"
|
||||
version = "0.10.0"
|
||||
version = "0.10.1"
|
||||
dependencies = [
|
||||
"anyhow",
|
||||
"cfg-if",
|
||||
@@ -1410,7 +1410,7 @@ dependencies = [
|
||||
"dashmap",
|
||||
"data-encoding",
|
||||
"globset",
|
||||
"hashbrown 0.17.0",
|
||||
"hashbrown 0.17.1",
|
||||
"icu_casemap",
|
||||
"indexmap",
|
||||
"ipnet",
|
||||
@@ -1431,7 +1431,7 @@ dependencies = [
|
||||
"serde",
|
||||
"serde_json",
|
||||
"serde_yaml",
|
||||
"spin 0.10.0",
|
||||
"spin 0.12.0",
|
||||
"test-generator",
|
||||
"thiserror",
|
||||
"url",
|
||||
@@ -1441,7 +1441,7 @@ dependencies = [
|
||||
|
||||
[[package]]
|
||||
name = "regorus-mimalloc"
|
||||
version = "2.2.6"
|
||||
version = "2.2.7"
|
||||
dependencies = [
|
||||
"regorus-mimalloc-sys",
|
||||
]
|
||||
@@ -1518,9 +1518,9 @@ dependencies = [
|
||||
|
||||
[[package]]
|
||||
name = "serde_json"
|
||||
version = "1.0.149"
|
||||
version = "1.0.150"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "83fc039473c5595ace860d8c4fafa220ff474b3fc6bfdb4293327f1a37e94d86"
|
||||
checksum = "e8014e44b4736ed0538adeecded0fce2a272f22dc9578a7eb6b2d9993c74cfb9"
|
||||
dependencies = [
|
||||
"itoa",
|
||||
"memchr",
|
||||
@@ -1580,9 +1580,9 @@ checksum = "6980e8d7511241f8acf4aebddbb1ff938df5eebe98691418c4468d0b72a96a67"
|
||||
|
||||
[[package]]
|
||||
name = "spin"
|
||||
version = "0.10.0"
|
||||
version = "0.12.0"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "d5fe4ccb98d9c292d56fec89a5e07da7fc4cf0dc11e156b41793132775d3e591"
|
||||
checksum = "1527984ca054dfca79333baec451042863f485fbee01b7bf6d911de915cac865"
|
||||
|
||||
[[package]]
|
||||
name = "stable_deref_trait"
|
||||
@@ -1693,9 +1693,9 @@ dependencies = [
|
||||
|
||||
[[package]]
|
||||
name = "toml_edit"
|
||||
version = "0.25.11+spec-1.1.0"
|
||||
version = "0.25.12+spec-1.1.0"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "0b59c4d22ed448339746c59b905d24568fcbb3ab65a500494f7b8c3e97739f2b"
|
||||
checksum = "d2153edc6955a6c354fad8f5efd38b6a8769bdccf9fe50f8e1329f81b0baa5d7"
|
||||
dependencies = [
|
||||
"indexmap",
|
||||
"toml_datetime",
|
||||
@@ -1841,9 +1841,9 @@ dependencies = [
|
||||
|
||||
[[package]]
|
||||
name = "wasm-bindgen"
|
||||
version = "0.2.121"
|
||||
version = "0.2.122"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "49ace1d07c165b0864824eee619580c4689389afa9dc9ed3a4c75040d82e6790"
|
||||
checksum = "3ed04576f974d2b2fba0f38c51dbc5518011e38c36bf1143164be765528fd409"
|
||||
dependencies = [
|
||||
"cfg-if",
|
||||
"once_cell",
|
||||
@@ -1854,9 +1854,9 @@ dependencies = [
|
||||
|
||||
[[package]]
|
||||
name = "wasm-bindgen-macro"
|
||||
version = "0.2.121"
|
||||
version = "0.2.122"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "8e68e6f4afd367a562002c05637acb8578ff2dea1943df76afb9e83d177c8578"
|
||||
checksum = "916151b09da36bd82f6615cbf3a419e2f0ba23a03c6160e8e92eb6bd4aa1dec6"
|
||||
dependencies = [
|
||||
"quote 1.0.45",
|
||||
"wasm-bindgen-macro-support",
|
||||
@@ -1864,9 +1864,9 @@ dependencies = [
|
||||
|
||||
[[package]]
|
||||
name = "wasm-bindgen-macro-support"
|
||||
version = "0.2.121"
|
||||
version = "0.2.122"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "d95a9ec35c64b2a7cb35d3fead40c4238d0940c86d107136999567a4703259f2"
|
||||
checksum = "299047362ccbfce148b67ab7e73349f77748e00c8296f9542adfad2ad82c5c5e"
|
||||
dependencies = [
|
||||
"bumpalo",
|
||||
"proc-macro2 1.0.106",
|
||||
@@ -1877,9 +1877,9 @@ dependencies = [
|
||||
|
||||
[[package]]
|
||||
name = "wasm-bindgen-shared"
|
||||
version = "0.2.121"
|
||||
version = "0.2.122"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "c4e0100b01e9f0d03189a92b96772a1fb998639d981193d7dbab487302513441"
|
||||
checksum = "9a929b2c61f11ba3e9bc35b50c1f25cb38e0e892c0c231ae2b8cf78d5dad4437"
|
||||
dependencies = [
|
||||
"unicode-ident",
|
||||
]
|
||||
@@ -1920,9 +1920,9 @@ dependencies = [
|
||||
|
||||
[[package]]
|
||||
name = "web-sys"
|
||||
version = "0.3.98"
|
||||
version = "0.3.99"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "4b572dff8bcf38bad0fa19729c89bb5748b2b9b1d8be70cf90df697e3a8f32aa"
|
||||
checksum = "6d621441cfc37b84979402712047321980c178f299193a3589d05b99e8763436"
|
||||
dependencies = [
|
||||
"js-sys",
|
||||
"wasm-bindgen",
|
||||
@@ -2029,9 +2029,9 @@ dependencies = [
|
||||
|
||||
[[package]]
|
||||
name = "winnow"
|
||||
version = "1.0.2"
|
||||
version = "1.0.3"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "2ee1708bef14716a11bae175f579062d4554d95be2c6829f518df847b7b3fdd0"
|
||||
checksum = "0592e1c9d151f854e6fd382574c3a0855250e1d9b2f99d9281c6e6391af352f1"
|
||||
dependencies = [
|
||||
"memchr",
|
||||
]
|
||||
@@ -2173,18 +2173,18 @@ dependencies = [
|
||||
|
||||
[[package]]
|
||||
name = "zerocopy"
|
||||
version = "0.8.48"
|
||||
version = "0.8.49"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "eed437bf9d6692032087e337407a86f04cd8d6a16a37199ed57949d415bd68e9"
|
||||
checksum = "bce33a6288fa3f072a8c2c7d0f2fdbb90e28298f0135c1f99b96c3db2efcc60b"
|
||||
dependencies = [
|
||||
"zerocopy-derive",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "zerocopy-derive"
|
||||
version = "0.8.48"
|
||||
version = "0.8.49"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "70e3cd084b1788766f53af483dd21f93881ff30d7320490ec3ef7526d203bad4"
|
||||
checksum = "8fd425244944f4ab65ccff928e7323354c5a018c75838362fdce749dfad2ee1e"
|
||||
dependencies = [
|
||||
"proc-macro2 1.0.106",
|
||||
"quote 1.0.45",
|
||||
@@ -2193,9 +2193,9 @@ dependencies = [
|
||||
|
||||
[[package]]
|
||||
name = "zerofrom"
|
||||
version = "0.1.7"
|
||||
version = "0.1.8"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "69faa1f2a1ea75661980b013019ed6687ed0e83d069bc1114e2cc74c6c04c4df"
|
||||
checksum = "0ec05a11813ea801ff6d75110ad09cd0824ddba17dfe17128ea0d5f68e6c5272"
|
||||
dependencies = [
|
||||
"zerofrom-derive",
|
||||
]
|
||||
|
||||
10
Cargo.toml
10
Cargo.toml
@@ -8,7 +8,7 @@ members = [
|
||||
[package]
|
||||
name = "regorus"
|
||||
description = "A fast, lightweight Rego (OPA policy language) interpreter"
|
||||
version = "0.10.0"
|
||||
version = "0.10.1"
|
||||
edition = "2021"
|
||||
license = "MIT AND Apache-2.0 AND BSD-3-Clause"
|
||||
repository = "https://github.com/microsoft/regorus"
|
||||
@@ -98,7 +98,7 @@ rand = ["dep:rand"]
|
||||
[dependencies]
|
||||
anyhow = { version = "1.0.102", default-features = false }
|
||||
serde = {version = "1.0.150", default-features = false, features = ["derive", "rc", "alloc"] }
|
||||
serde_json = { version = "1.0.89", default-features = false, features = ["alloc"] }
|
||||
serde_json = { version = "1.0.150", default-features = false, features = ["alloc"] }
|
||||
hashbrown = { version = "0.17", default-features = false, features = ["default-hasher"], optional = true }
|
||||
lazy_static = { version = "1.4.0", default-features = false }
|
||||
thiserror = { version = "2.0", default-features = false }
|
||||
@@ -107,14 +107,14 @@ data-encoding = { version = "2.8.0", optional = true, default-features=false, fe
|
||||
num-bigint = { version = "0.4", default-features = false }
|
||||
num-traits = { version = "0.2", default-features = false }
|
||||
parking_lot = { version = "0.12", optional = true }
|
||||
spin = { version = "0.10.0", default-features = false, features = ["mutex", "spin_mutex"] }
|
||||
spin = { version = "0.12.0", default-features = false, features = ["mutex", "spin_mutex"] }
|
||||
|
||||
globset = { version = "0.4.16", features = ["simd-accel"], default-features = false, optional = true }
|
||||
regex = {version = "1.12.3", optional = true, default-features = false }
|
||||
semver = {version = "1.0.28", optional = true, default-features = false }
|
||||
url = { version = "2.5.4", optional = true }
|
||||
uuid = { version = "1.22.0", default-features = false, features = ["v4", "fast-rng"], optional = true }
|
||||
jsonschema = { version = "0.46.4", default-features = false, optional = true }
|
||||
jsonschema = { version = "0.46.5", default-features = false, optional = true }
|
||||
chrono = { version = "0.4.44", optional = true }
|
||||
chrono-tz = { version = "0.10.1", optional = true }
|
||||
ipnet = { version = "2.12.0", optional = true, default-features = false }
|
||||
@@ -128,7 +128,7 @@ rand = { version = "0.10.0", default-features = false, features = ["thread_rng"]
|
||||
msvc_spectre_libs = { version = "0.1", features = ["error"], optional = true }
|
||||
dashmap = { version = "6.1", default-features = false, optional = true }
|
||||
lru = { version = "0.18", default-features = false, optional = true }
|
||||
mimalloc = { package = "regorus-mimalloc", path = "mimalloc", version = "2.2.6", optional = true }
|
||||
mimalloc = { package = "regorus-mimalloc", path = "mimalloc", version = "2.2.7", optional = true }
|
||||
|
||||
# rvm related deps
|
||||
indexmap = { version = "2.13.1", default-features = false, features = ["serde"], optional = true }
|
||||
|
||||
@@ -1,7 +1,7 @@
|
||||
<Project>
|
||||
<PropertyGroup>
|
||||
<ManagePackageVersionsCentrally>true</ManagePackageVersionsCentrally>
|
||||
<RegorusPackageVersion>0.10.0</RegorusPackageVersion>
|
||||
<RegorusPackageVersion>0.10.1</RegorusPackageVersion>
|
||||
<RegorusPackageVersionSuffix Condition="'$(VersionSuffix)' != ''">-$(VersionSuffix)</RegorusPackageVersionSuffix>
|
||||
</PropertyGroup>
|
||||
|
||||
|
||||
@@ -150,3 +150,76 @@ const string ContextJson = """
|
||||
var allowed = RbacEngine.EvaluateCondition(Condition, ContextJson);
|
||||
Console.WriteLine($"RBAC condition allowed: {allowed}");
|
||||
```
|
||||
|
||||
## Azure Policy JSON Evaluation
|
||||
|
||||
Compile and evaluate Azure Policy JSON `policyRule` definitions directly — no Rego translation required.
|
||||
The `AzurePolicyCompiler` compiles JSON policy rules into RVM programs that can be executed with the `Rvm` engine.
|
||||
|
||||
```csharp
|
||||
using Regorus;
|
||||
|
||||
// 1. Load alias definitions for the resource provider
|
||||
const string AliasesJson = """
|
||||
[{
|
||||
"namespace": "Microsoft.Storage",
|
||||
"resourceTypes": [{
|
||||
"resourceType": "storageAccounts",
|
||||
"aliases": [{
|
||||
"name": "Microsoft.Storage/storageAccounts/supportsHttpsTrafficOnly",
|
||||
"defaultPath": "properties.supportsHttpsTrafficOnly",
|
||||
"paths": []
|
||||
}]
|
||||
}]
|
||||
}]
|
||||
""";
|
||||
|
||||
using var registry = AliasRegistry.FromJson(AliasesJson);
|
||||
|
||||
// 2. Compile a JSON policy rule (the native Azure Policy language)
|
||||
const string PolicyRule = """
|
||||
{
|
||||
"if": {
|
||||
"allOf": [
|
||||
{ "field": "type", "equals": "Microsoft.Storage/storageAccounts" },
|
||||
{ "field": "Microsoft.Storage/storageAccounts/supportsHttpsTrafficOnly", "equals": false }
|
||||
]
|
||||
},
|
||||
"then": { "effect": "deny" }
|
||||
}
|
||||
""";
|
||||
|
||||
using var program = AzurePolicyCompiler.CompilePolicyRule(registry, PolicyRule);
|
||||
|
||||
// 3. Normalize an ARM resource and evaluate
|
||||
var armResource = """
|
||||
{
|
||||
"type": "Microsoft.Storage/storageAccounts",
|
||||
"name": "mystorage",
|
||||
"properties": { "supportsHttpsTrafficOnly": false }
|
||||
}
|
||||
""";
|
||||
var envelope = registry.NormalizeAndWrap(armResource);
|
||||
|
||||
using var vm = new Rvm();
|
||||
vm.LoadProgram(program);
|
||||
vm.SetInputJson(envelope!);
|
||||
|
||||
var result = vm.ExecuteEntryPoint("main");
|
||||
// result: {"effect": "deny"} for non-compliant, "<undefined>" for compliant
|
||||
Console.WriteLine($"Policy result: {result}");
|
||||
```
|
||||
|
||||
**Context-dependent policies:** If your policy uses context functions like
|
||||
`subscription()`, `resourceGroup()`, or `requestContext()`, you must also set
|
||||
the VM context separately:
|
||||
|
||||
```csharp
|
||||
// The context JSON from NormalizeAndWrap is in the input envelope,
|
||||
// but must also be provided to the VM's ambient context:
|
||||
vm.SetContextJson(contextJson);
|
||||
```
|
||||
|
||||
You can also compile full policy definitions (with parameters) using
|
||||
`AzurePolicyCompiler.CompilePolicyDefinition()`. See
|
||||
`bindings/csharp/Regorus.Tests/AzurePolicyCompilerTests.cs` for comprehensive examples.
|
||||
|
||||
@@ -43,31 +43,28 @@ public class AliasRegistryTests
|
||||
[TestMethod]
|
||||
public void Create_and_dispose_succeeds()
|
||||
{
|
||||
using var registry = new AliasRegistry();
|
||||
using var registry = AliasRegistry.Empty();
|
||||
Assert.AreEqual(0, registry.Length);
|
||||
}
|
||||
|
||||
[TestMethod]
|
||||
public void LoadJson_populates_registry()
|
||||
{
|
||||
using var registry = new AliasRegistry();
|
||||
registry.LoadJson(AliasesJson);
|
||||
using var registry = AliasRegistry.FromJson(AliasesJson);
|
||||
Assert.AreEqual(1, registry.Length);
|
||||
}
|
||||
|
||||
[TestMethod]
|
||||
public void LoadManifest_populates_registry()
|
||||
{
|
||||
using var registry = new AliasRegistry();
|
||||
registry.LoadManifest(ManifestJson);
|
||||
using var registry = AliasRegistry.FromManifest(ManifestJson);
|
||||
Assert.AreEqual(1, registry.Length);
|
||||
}
|
||||
|
||||
[TestMethod]
|
||||
public void NormalizeAndWrap_produces_envelope()
|
||||
{
|
||||
using var registry = new AliasRegistry();
|
||||
registry.LoadJson(AliasesJson);
|
||||
using var registry = AliasRegistry.FromJson(AliasesJson);
|
||||
|
||||
var resource = @"{
|
||||
""name"": ""acct1"",
|
||||
@@ -93,8 +90,7 @@ public class AliasRegistryTests
|
||||
[TestMethod]
|
||||
public void NormalizeAndWrap_with_context_and_parameters()
|
||||
{
|
||||
using var registry = new AliasRegistry();
|
||||
registry.LoadJson(AliasesJson);
|
||||
using var registry = AliasRegistry.FromJson(AliasesJson);
|
||||
|
||||
var resource = @"{
|
||||
""name"": ""acct1"",
|
||||
@@ -115,8 +111,7 @@ public class AliasRegistryTests
|
||||
[TestMethod]
|
||||
public void Denormalize_restores_properties()
|
||||
{
|
||||
using var registry = new AliasRegistry();
|
||||
registry.LoadJson(AliasesJson);
|
||||
using var registry = AliasRegistry.FromJson(AliasesJson);
|
||||
|
||||
var normalized = @"{
|
||||
""name"": ""acct1"",
|
||||
@@ -137,8 +132,7 @@ public class AliasRegistryTests
|
||||
[TestMethod]
|
||||
public void Round_trip_normalize_then_denormalize()
|
||||
{
|
||||
using var registry = new AliasRegistry();
|
||||
registry.LoadJson(AliasesJson);
|
||||
using var registry = AliasRegistry.FromJson(AliasesJson);
|
||||
|
||||
var resource = @"{
|
||||
""name"": ""acct1"",
|
||||
@@ -166,8 +160,7 @@ public class AliasRegistryTests
|
||||
[TestMethod]
|
||||
public void DataPlane_manifest_normalize()
|
||||
{
|
||||
using var registry = new AliasRegistry();
|
||||
registry.LoadManifest(ManifestJson);
|
||||
using var registry = AliasRegistry.FromManifest(ManifestJson);
|
||||
|
||||
var resource = @"{
|
||||
""type"": ""Microsoft.KeyVault.Data/vaults/certificates"",
|
||||
@@ -185,7 +178,7 @@ public class AliasRegistryTests
|
||||
[ExpectedException(typeof(InvalidOperationException))]
|
||||
public void LoadJson_invalid_throws()
|
||||
{
|
||||
using var registry = new AliasRegistry();
|
||||
registry.LoadJson("not valid json");
|
||||
using var builder = new AliasRegistryBuilder();
|
||||
builder.LoadJson("not valid json");
|
||||
}
|
||||
}
|
||||
|
||||
436
bindings/csharp/Regorus.Tests/AzurePolicyCompilerTests.cs
Normal file
436
bindings/csharp/Regorus.Tests/AzurePolicyCompilerTests.cs
Normal file
@@ -0,0 +1,436 @@
|
||||
// Copyright (c) Microsoft Corporation.
|
||||
// Licensed under the MIT License.
|
||||
|
||||
using System;
|
||||
using System.Text.Json.Nodes;
|
||||
using Microsoft.VisualStudio.TestTools.UnitTesting;
|
||||
using Regorus;
|
||||
|
||||
namespace Regorus.Tests;
|
||||
|
||||
/// <summary>
|
||||
/// Tests for <see cref="AzurePolicyCompiler"/> — compiling Azure Policy JSON
|
||||
/// policyRule and policyDefinition into RVM programs and evaluating them.
|
||||
/// </summary>
|
||||
[TestClass]
|
||||
public class AzurePolicyCompilerTests
|
||||
{
|
||||
// -----------------------------------------------------------------------
|
||||
// Test data
|
||||
// -----------------------------------------------------------------------
|
||||
|
||||
private const string StorageAliasesJson = @"[{
|
||||
""namespace"": ""Microsoft.Storage"",
|
||||
""resourceTypes"": [{
|
||||
""resourceType"": ""storageAccounts"",
|
||||
""capabilities"": ""SupportsTags, SupportsLocation"",
|
||||
""aliases"": [
|
||||
{
|
||||
""name"": ""Microsoft.Storage/storageAccounts/supportsHttpsTrafficOnly"",
|
||||
""defaultPath"": ""properties.supportsHttpsTrafficOnly"",
|
||||
""paths"": []
|
||||
},
|
||||
{
|
||||
""name"": ""Microsoft.Storage/storageAccounts/minimumTlsVersion"",
|
||||
""defaultPath"": ""properties.minimumTlsVersion"",
|
||||
""paths"": []
|
||||
}
|
||||
]
|
||||
}]
|
||||
}]";
|
||||
|
||||
/// <summary>Simple policy rule that checks the resource type.</summary>
|
||||
private const string SimpleAuditRule = @"{
|
||||
""if"": {
|
||||
""field"": ""type"",
|
||||
""equals"": ""Microsoft.Storage/storageAccounts""
|
||||
},
|
||||
""then"": { ""effect"": ""audit"" }
|
||||
}";
|
||||
|
||||
/// <summary>Policy rule that uses an alias to check HTTPS-only.</summary>
|
||||
private const string HttpsDenyRule = @"{
|
||||
""if"": {
|
||||
""allOf"": [
|
||||
{ ""field"": ""type"", ""equals"": ""Microsoft.Storage/storageAccounts"" },
|
||||
{ ""field"": ""Microsoft.Storage/storageAccounts/supportsHttpsTrafficOnly"", ""equals"": false }
|
||||
]
|
||||
},
|
||||
""then"": { ""effect"": ""deny"" }
|
||||
}";
|
||||
|
||||
/// <summary>Full policy definition with parameters.</summary>
|
||||
private const string PolicyDefinitionWithParams = @"{
|
||||
""displayName"": ""Require HTTPS for storage accounts"",
|
||||
""policyType"": ""Custom"",
|
||||
""mode"": ""Indexed"",
|
||||
""parameters"": {
|
||||
""effect"": {
|
||||
""type"": ""String"",
|
||||
""defaultValue"": ""deny""
|
||||
}
|
||||
},
|
||||
""policyRule"": {
|
||||
""if"": {
|
||||
""allOf"": [
|
||||
{ ""field"": ""type"", ""equals"": ""Microsoft.Storage/storageAccounts"" },
|
||||
{ ""field"": ""Microsoft.Storage/storageAccounts/supportsHttpsTrafficOnly"", ""equals"": false }
|
||||
]
|
||||
},
|
||||
""then"": { ""effect"": ""[parameters('effect')]"" }
|
||||
}
|
||||
}";
|
||||
|
||||
// -----------------------------------------------------------------------
|
||||
// Helper
|
||||
// -----------------------------------------------------------------------
|
||||
|
||||
/// <summary>
|
||||
/// Wrap a normalized resource JSON and parameters into the input envelope
|
||||
/// expected by compiled Azure Policy RVM programs.
|
||||
/// </summary>
|
||||
private static string WrapInput(string resourceJson, string parametersJson = "{}")
|
||||
{
|
||||
return $@"{{""resource"": {resourceJson}, ""parameters"": {parametersJson}}}";
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Compile a policy rule, load it into an RVM, set input, and execute.
|
||||
/// Returns the result string from <c>ExecuteEntryPoint("main")</c>.
|
||||
/// </summary>
|
||||
private static string? CompileAndEval(
|
||||
AliasRegistry? registry,
|
||||
string policyRuleJson,
|
||||
string inputJson)
|
||||
{
|
||||
using var program = AzurePolicyCompiler.CompilePolicyRule(registry, policyRuleJson);
|
||||
using var vm = new Rvm();
|
||||
vm.LoadProgram(program);
|
||||
vm.SetInputJson(inputJson);
|
||||
return vm.ExecuteEntryPoint("main");
|
||||
}
|
||||
|
||||
// -----------------------------------------------------------------------
|
||||
// CompilePolicyRule tests
|
||||
// -----------------------------------------------------------------------
|
||||
|
||||
[TestMethod]
|
||||
public void CompilePolicyRule_no_aliases_succeeds()
|
||||
{
|
||||
using var program = AzurePolicyCompiler.CompilePolicyRule(null, SimpleAuditRule);
|
||||
Assert.IsNotNull(program);
|
||||
}
|
||||
|
||||
[TestMethod]
|
||||
public void CompilePolicyRule_with_aliases_succeeds()
|
||||
{
|
||||
using var registry = AliasRegistry.FromJson(StorageAliasesJson);
|
||||
|
||||
using var program = AzurePolicyCompiler.CompilePolicyRule(registry, HttpsDenyRule);
|
||||
Assert.IsNotNull(program);
|
||||
}
|
||||
|
||||
[TestMethod]
|
||||
[ExpectedException(typeof(ArgumentNullException))]
|
||||
public void CompilePolicyRule_null_json_throws()
|
||||
{
|
||||
AzurePolicyCompiler.CompilePolicyRule(null, null!);
|
||||
}
|
||||
|
||||
[TestMethod]
|
||||
[ExpectedException(typeof(InvalidOperationException))]
|
||||
public void CompilePolicyRule_invalid_json_throws()
|
||||
{
|
||||
AzurePolicyCompiler.CompilePolicyRule(null, "not valid json");
|
||||
}
|
||||
|
||||
// -----------------------------------------------------------------------
|
||||
// CompilePolicyDefinition tests
|
||||
// -----------------------------------------------------------------------
|
||||
|
||||
[TestMethod]
|
||||
public void CompilePolicyDefinition_no_aliases_succeeds()
|
||||
{
|
||||
using var program = AzurePolicyCompiler.CompilePolicyDefinition(null, PolicyDefinitionWithParams);
|
||||
Assert.IsNotNull(program);
|
||||
}
|
||||
|
||||
[TestMethod]
|
||||
public void CompilePolicyDefinition_with_aliases_succeeds()
|
||||
{
|
||||
using var registry = AliasRegistry.FromJson(StorageAliasesJson);
|
||||
|
||||
using var program = AzurePolicyCompiler.CompilePolicyDefinition(registry, PolicyDefinitionWithParams);
|
||||
Assert.IsNotNull(program);
|
||||
}
|
||||
|
||||
[TestMethod]
|
||||
[ExpectedException(typeof(ArgumentNullException))]
|
||||
public void CompilePolicyDefinition_null_json_throws()
|
||||
{
|
||||
AzurePolicyCompiler.CompilePolicyDefinition(null, null!);
|
||||
}
|
||||
|
||||
[TestMethod]
|
||||
[ExpectedException(typeof(InvalidOperationException))]
|
||||
public void CompilePolicyDefinition_invalid_json_throws()
|
||||
{
|
||||
AzurePolicyCompiler.CompilePolicyDefinition(null, @"{""not"": ""a definition""}");
|
||||
}
|
||||
|
||||
// -----------------------------------------------------------------------
|
||||
// End-to-end evaluation tests
|
||||
// -----------------------------------------------------------------------
|
||||
|
||||
[TestMethod]
|
||||
public void Eval_simple_rule_matching_resource_returns_effect()
|
||||
{
|
||||
var input = WrapInput(
|
||||
@"{""type"": ""microsoft.storage/storageaccounts""}");
|
||||
|
||||
var result = CompileAndEval(null, SimpleAuditRule, input);
|
||||
Assert.IsNotNull(result, "expected a result for matching resource");
|
||||
|
||||
var doc = JsonNode.Parse(result!)!;
|
||||
Assert.AreEqual("audit", doc["effect"]?.GetValue<string>(),
|
||||
$"expected 'audit' effect, got: {result}");
|
||||
}
|
||||
|
||||
[TestMethod]
|
||||
public void Eval_simple_rule_non_matching_resource_returns_undefined()
|
||||
{
|
||||
var input = WrapInput(
|
||||
@"{""type"": ""microsoft.compute/virtualmachines""}");
|
||||
|
||||
var result = CompileAndEval(null, SimpleAuditRule, input);
|
||||
Assert.IsNotNull(result);
|
||||
StringAssert.Contains(result!, "undefined",
|
||||
"expected undefined for non-matching resource type");
|
||||
}
|
||||
|
||||
[TestMethod]
|
||||
public void Eval_alias_rule_non_compliant_returns_deny()
|
||||
{
|
||||
using var registry = AliasRegistry.FromJson(StorageAliasesJson);
|
||||
|
||||
// Non-compliant: HTTPS not enabled (normalized/lowercased form)
|
||||
var input = WrapInput(
|
||||
@"{""type"": ""microsoft.storage/storageaccounts"", ""supportshttpstrafficonly"": false}");
|
||||
|
||||
using var program = AzurePolicyCompiler.CompilePolicyRule(registry, HttpsDenyRule);
|
||||
using var vm = new Rvm();
|
||||
vm.LoadProgram(program);
|
||||
vm.SetInputJson(input);
|
||||
|
||||
var result = vm.ExecuteEntryPoint("main");
|
||||
Assert.IsNotNull(result);
|
||||
|
||||
var doc = JsonNode.Parse(result!)!;
|
||||
Assert.AreEqual("deny", doc["effect"]?.GetValue<string>(),
|
||||
$"expected 'deny' for non-compliant resource, got: {result}");
|
||||
}
|
||||
|
||||
[TestMethod]
|
||||
public void Eval_alias_rule_compliant_returns_undefined()
|
||||
{
|
||||
using var registry = AliasRegistry.FromJson(StorageAliasesJson);
|
||||
|
||||
// Compliant: HTTPS enabled
|
||||
var input = WrapInput(
|
||||
@"{""type"": ""microsoft.storage/storageaccounts"", ""supportshttpstrafficonly"": true}");
|
||||
|
||||
using var program = AzurePolicyCompiler.CompilePolicyRule(registry, HttpsDenyRule);
|
||||
using var vm = new Rvm();
|
||||
vm.LoadProgram(program);
|
||||
vm.SetInputJson(input);
|
||||
|
||||
var result = vm.ExecuteEntryPoint("main");
|
||||
Assert.IsNotNull(result);
|
||||
StringAssert.Contains(result!, "undefined",
|
||||
"expected undefined for compliant resource");
|
||||
}
|
||||
|
||||
[TestMethod]
|
||||
public void Eval_definition_with_default_parameters()
|
||||
{
|
||||
using var registry = AliasRegistry.FromJson(StorageAliasesJson);
|
||||
|
||||
using var program = AzurePolicyCompiler.CompilePolicyDefinition(
|
||||
registry, PolicyDefinitionWithParams);
|
||||
using var vm = new Rvm();
|
||||
vm.LoadProgram(program);
|
||||
|
||||
// Non-compliant resource
|
||||
var input = WrapInput(
|
||||
@"{""type"": ""microsoft.storage/storageaccounts"", ""supportshttpstrafficonly"": false}");
|
||||
vm.SetInputJson(input);
|
||||
|
||||
var result = vm.ExecuteEntryPoint("main");
|
||||
Assert.IsNotNull(result);
|
||||
|
||||
var doc = JsonNode.Parse(result!)!;
|
||||
// Default parameter value is "deny"
|
||||
Assert.AreEqual("deny", doc["effect"]?.GetValue<string>(),
|
||||
$"expected default 'deny' effect, got: {result}");
|
||||
}
|
||||
|
||||
[TestMethod]
|
||||
public void Eval_with_normalized_arm_resource_end_to_end()
|
||||
{
|
||||
using var registry = AliasRegistry.FromJson(StorageAliasesJson);
|
||||
|
||||
// Simulate the full production flow:
|
||||
// 1. Start with an ARM resource
|
||||
var armResource = @"{
|
||||
""type"": ""Microsoft.Storage/storageAccounts"",
|
||||
""name"": ""mystorage"",
|
||||
""location"": ""eastus"",
|
||||
""properties"": {
|
||||
""supportsHttpsTrafficOnly"": false,
|
||||
""minimumTlsVersion"": ""TLS1_0""
|
||||
}
|
||||
}";
|
||||
|
||||
// 2. Normalize via AliasRegistry
|
||||
var normalizedEnvelope = registry.NormalizeAndWrap(
|
||||
armResource,
|
||||
apiVersion: null,
|
||||
contextJson: "{}",
|
||||
parametersJson: "{}");
|
||||
Assert.IsNotNull(normalizedEnvelope);
|
||||
|
||||
// 3. Compile the policy rule
|
||||
using var program = AzurePolicyCompiler.CompilePolicyRule(registry, HttpsDenyRule);
|
||||
|
||||
// 4. Execute
|
||||
using var vm = new Rvm();
|
||||
vm.LoadProgram(program);
|
||||
vm.SetInputJson(normalizedEnvelope!);
|
||||
|
||||
var result = vm.ExecuteEntryPoint("main");
|
||||
Assert.IsNotNull(result);
|
||||
|
||||
var doc = JsonNode.Parse(result!)!;
|
||||
Assert.AreEqual("deny", doc["effect"]?.GetValue<string>(),
|
||||
$"expected 'deny' for non-HTTPS storage account, got: {result}");
|
||||
}
|
||||
|
||||
[TestMethod]
|
||||
public void Eval_normalized_compliant_resource_end_to_end()
|
||||
{
|
||||
using var registry = AliasRegistry.FromJson(StorageAliasesJson);
|
||||
|
||||
var armResource = @"{
|
||||
""type"": ""Microsoft.Storage/storageAccounts"",
|
||||
""name"": ""secureastorage"",
|
||||
""location"": ""westus"",
|
||||
""properties"": {
|
||||
""supportsHttpsTrafficOnly"": true,
|
||||
""minimumTlsVersion"": ""TLS1_2""
|
||||
}
|
||||
}";
|
||||
|
||||
var normalizedEnvelope = registry.NormalizeAndWrap(
|
||||
armResource,
|
||||
apiVersion: null,
|
||||
contextJson: "{}",
|
||||
parametersJson: "{}");
|
||||
Assert.IsNotNull(normalizedEnvelope);
|
||||
|
||||
using var program = AzurePolicyCompiler.CompilePolicyRule(registry, HttpsDenyRule);
|
||||
using var vm = new Rvm();
|
||||
vm.LoadProgram(program);
|
||||
vm.SetInputJson(normalizedEnvelope!);
|
||||
|
||||
var result = vm.ExecuteEntryPoint("main");
|
||||
Assert.IsNotNull(result);
|
||||
StringAssert.Contains(result!, "undefined",
|
||||
"expected undefined for compliant HTTPS storage account");
|
||||
}
|
||||
|
||||
[TestMethod]
|
||||
public void Program_can_be_serialized_and_reloaded()
|
||||
{
|
||||
using var program = AzurePolicyCompiler.CompilePolicyRule(null, SimpleAuditRule);
|
||||
|
||||
// Serialize to binary
|
||||
var binary = program.SerializeBinary();
|
||||
Assert.IsTrue(binary.Length > 0, "serialized program should not be empty");
|
||||
|
||||
// Deserialize and run
|
||||
using var restored = Program.DeserializeBinary(binary, out var isPartial);
|
||||
Assert.IsFalse(isPartial, "program should not be partial");
|
||||
|
||||
using var vm = new Rvm();
|
||||
vm.LoadProgram(restored);
|
||||
var input = WrapInput(@"{""type"": ""microsoft.storage/storageaccounts""}");
|
||||
vm.SetInputJson(input);
|
||||
|
||||
var result = vm.ExecuteEntryPoint("main");
|
||||
Assert.IsNotNull(result);
|
||||
var doc = JsonNode.Parse(result!)!;
|
||||
Assert.AreEqual("audit", doc["effect"]?.GetValue<string>());
|
||||
}
|
||||
|
||||
[TestMethod]
|
||||
public void Program_generates_listing()
|
||||
{
|
||||
using var program = AzurePolicyCompiler.CompilePolicyRule(null, SimpleAuditRule);
|
||||
var listing = program.GenerateListing();
|
||||
Assert.IsFalse(string.IsNullOrWhiteSpace(listing),
|
||||
"generated listing should not be empty");
|
||||
}
|
||||
|
||||
// -----------------------------------------------------------------------
|
||||
// Context-dependent policy tests
|
||||
// -----------------------------------------------------------------------
|
||||
|
||||
/// Policy rule that uses subscription() context function.
|
||||
private const string ContextPolicyRule = @"{
|
||||
""if"": {
|
||||
""allOf"": [
|
||||
{ ""field"": ""type"", ""equals"": ""Microsoft.Storage/storageAccounts"" },
|
||||
{ ""value"": ""[subscription().subscriptionId]"", ""equals"": ""sub-123"" }
|
||||
]
|
||||
},
|
||||
""then"": { ""effect"": ""deny"" }
|
||||
}";
|
||||
|
||||
[TestMethod]
|
||||
public void Eval_context_policy_with_set_context_returns_effect()
|
||||
{
|
||||
using var program = AzurePolicyCompiler.CompilePolicyRule(null, ContextPolicyRule);
|
||||
using var vm = new Rvm();
|
||||
vm.LoadProgram(program);
|
||||
|
||||
vm.SetContextJson(@"{""subscription"": {""subscriptionId"": ""sub-123""}}");
|
||||
|
||||
var input = WrapInput(
|
||||
@"{""type"": ""microsoft.storage/storageaccounts""}");
|
||||
vm.SetInputJson(input);
|
||||
|
||||
var result = vm.ExecuteEntryPoint("main");
|
||||
Assert.IsNotNull(result);
|
||||
var doc = JsonNode.Parse(result!)!;
|
||||
Assert.AreEqual("deny", doc["effect"]?.GetValue<string>(),
|
||||
$"expected 'deny' with matching context, got: {result}");
|
||||
}
|
||||
|
||||
[TestMethod]
|
||||
public void Eval_context_policy_without_context_returns_undefined()
|
||||
{
|
||||
using var program = AzurePolicyCompiler.CompilePolicyRule(null, ContextPolicyRule);
|
||||
using var vm = new Rvm();
|
||||
vm.LoadProgram(program);
|
||||
|
||||
// No context set — subscription() will be undefined
|
||||
var input = WrapInput(
|
||||
@"{""type"": ""microsoft.storage/storageaccounts""}");
|
||||
vm.SetInputJson(input);
|
||||
|
||||
var result = vm.ExecuteEntryPoint("main");
|
||||
Assert.IsNotNull(result);
|
||||
StringAssert.Contains(result!, "undefined",
|
||||
"expected undefined without context set");
|
||||
}
|
||||
}
|
||||
@@ -62,8 +62,7 @@ public class AzurePolicyTests
|
||||
[TestMethod]
|
||||
public void AliasRegistry_NormalizeAndWrap_produces_input_envelope()
|
||||
{
|
||||
using var registry = new AliasRegistry();
|
||||
registry.LoadJson(StorageAliasesJson);
|
||||
using var registry = AliasRegistry.FromJson(StorageAliasesJson);
|
||||
|
||||
var result = registry.NormalizeAndWrap(
|
||||
StorageResourceJson,
|
||||
@@ -84,8 +83,7 @@ public class AzurePolicyTests
|
||||
[TestMethod]
|
||||
public void AliasRegistry_NormalizeAndWrap_flattens_properties()
|
||||
{
|
||||
using var registry = new AliasRegistry();
|
||||
registry.LoadJson(StorageAliasesJson);
|
||||
using var registry = AliasRegistry.FromJson(StorageAliasesJson);
|
||||
|
||||
var result = registry.NormalizeAndWrap(StorageResourceJson);
|
||||
Assert.IsNotNull(result);
|
||||
@@ -107,8 +105,7 @@ public class AzurePolicyTests
|
||||
[TestMethod]
|
||||
public void AliasRegistry_NormalizeAndWrap_preserves_type_field()
|
||||
{
|
||||
using var registry = new AliasRegistry();
|
||||
registry.LoadJson(StorageAliasesJson);
|
||||
using var registry = AliasRegistry.FromJson(StorageAliasesJson);
|
||||
|
||||
var result = registry.NormalizeAndWrap(StorageResourceJson);
|
||||
var doc = JsonNode.Parse(result!);
|
||||
@@ -125,8 +122,7 @@ public class AzurePolicyTests
|
||||
[TestMethod]
|
||||
public void AliasRegistry_NormalizeAndWrap_includes_parameters()
|
||||
{
|
||||
using var registry = new AliasRegistry();
|
||||
registry.LoadJson(StorageAliasesJson);
|
||||
using var registry = AliasRegistry.FromJson(StorageAliasesJson);
|
||||
|
||||
var parametersJson = @"{ ""effect"": ""Deny"" }";
|
||||
var result = registry.NormalizeAndWrap(
|
||||
@@ -143,8 +139,7 @@ public class AzurePolicyTests
|
||||
[TestMethod]
|
||||
public void AliasRegistry_Denormalize_roundtrips_correctly()
|
||||
{
|
||||
using var registry = new AliasRegistry();
|
||||
registry.LoadJson(StorageAliasesJson);
|
||||
using var registry = AliasRegistry.FromJson(StorageAliasesJson);
|
||||
|
||||
// Normalize the ARM resource.
|
||||
var envelope = registry.NormalizeAndWrap(StorageResourceJson);
|
||||
@@ -177,8 +172,7 @@ public class AzurePolicyTests
|
||||
}
|
||||
|
||||
var aliasesJson = File.ReadAllText(aliasesPath);
|
||||
using var registry = new AliasRegistry();
|
||||
registry.LoadJson(aliasesJson);
|
||||
using var registry = AliasRegistry.FromJson(aliasesJson);
|
||||
|
||||
// The test_aliases.json file contains multiple providers.
|
||||
Assert.IsTrue(registry.Length > 0,
|
||||
|
||||
@@ -8,51 +8,43 @@ using Regorus.Internal;
|
||||
namespace Regorus
|
||||
{
|
||||
/// <summary>
|
||||
/// Manages Azure Policy alias definitions used for resource normalization
|
||||
/// Immutable Azure Policy alias registry used for resource normalization
|
||||
/// and policy compilation.
|
||||
/// </summary>
|
||||
public unsafe sealed class AliasRegistry : SafeHandleWrapper
|
||||
{
|
||||
/// <summary>
|
||||
/// Create an empty alias registry.
|
||||
/// </summary>
|
||||
public AliasRegistry()
|
||||
: base(RegorusAliasRegistryHandle.Create(), nameof(AliasRegistry))
|
||||
internal AliasRegistry(RegorusAliasRegistryHandle handle)
|
||||
: base(handle, nameof(AliasRegistry))
|
||||
{
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Load control-plane alias data (array of ProviderAliases) from a JSON string.
|
||||
/// Create an empty immutable alias registry.
|
||||
/// </summary>
|
||||
/// <param name="json">JSON array of ProviderAliases (e.g. from Get-AzPolicyAlias or ResourceTypesAndAliases.json)</param>
|
||||
public void LoadJson(string json)
|
||||
public static AliasRegistry Empty()
|
||||
{
|
||||
Utf8Marshaller.WithUtf8(json, jsonPtr =>
|
||||
{
|
||||
UseHandle(regPtr =>
|
||||
{
|
||||
CheckAndDropResult(API.regorus_alias_registry_load_json(
|
||||
(RegorusAliasRegistry*)regPtr, (byte*)jsonPtr));
|
||||
return 0;
|
||||
});
|
||||
});
|
||||
using var builder = new AliasRegistryBuilder();
|
||||
return builder.Build();
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Load a data-plane policy manifest from a JSON string.
|
||||
/// Create an immutable alias registry from control-plane alias JSON.
|
||||
/// </summary>
|
||||
/// <param name="json">JSON object containing a DataPolicyManifest</param>
|
||||
public void LoadManifest(string json)
|
||||
public static AliasRegistry FromJson(string json)
|
||||
{
|
||||
Utf8Marshaller.WithUtf8(json, jsonPtr =>
|
||||
{
|
||||
UseHandle(regPtr =>
|
||||
{
|
||||
CheckAndDropResult(API.regorus_alias_registry_load_manifest(
|
||||
(RegorusAliasRegistry*)regPtr, (byte*)jsonPtr));
|
||||
return 0;
|
||||
});
|
||||
});
|
||||
using var builder = new AliasRegistryBuilder();
|
||||
builder.LoadJson(json);
|
||||
return builder.Build();
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Create an immutable alias registry from a data-plane manifest JSON document.
|
||||
/// </summary>
|
||||
public static AliasRegistry FromManifest(string json)
|
||||
{
|
||||
using var builder = new AliasRegistryBuilder();
|
||||
builder.LoadManifest(json);
|
||||
return builder.Build();
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
@@ -74,11 +66,6 @@ namespace Regorus
|
||||
/// Normalize an ARM resource JSON and wrap it into the standard input envelope
|
||||
/// expected by a compiled Azure Policy program.
|
||||
/// </summary>
|
||||
/// <param name="resourceJson">Raw ARM resource JSON</param>
|
||||
/// <param name="apiVersion">API version string (e.g. "2023-01-01"), or null to use default alias paths</param>
|
||||
/// <param name="contextJson">Additional context JSON object (pass "{}" if none)</param>
|
||||
/// <param name="parametersJson">Policy parameter values JSON (pass "{}" if none)</param>
|
||||
/// <returns>JSON string: { "resource": <normalized>, "context": <context>, "parameters": <params> }</returns>
|
||||
public string? NormalizeAndWrap(string resourceJson, string? apiVersion = null, string contextJson = "{}", string parametersJson = "{}")
|
||||
{
|
||||
return Utf8Marshaller.WithUtf8(resourceJson, resPtr =>
|
||||
@@ -96,27 +83,22 @@ namespace Regorus
|
||||
(byte*)ctxPtr, (byte*)paramsPtr));
|
||||
});
|
||||
}
|
||||
else
|
||||
{
|
||||
return Utf8Marshaller.WithUtf8(apiVersion, apiPtr =>
|
||||
UseHandle(regPtr =>
|
||||
{
|
||||
return ResultHelpers.GetStringResult(
|
||||
API.regorus_alias_registry_normalize_and_wrap(
|
||||
(RegorusAliasRegistry*)regPtr,
|
||||
(byte*)resPtr, (byte*)apiPtr,
|
||||
(byte*)ctxPtr, (byte*)paramsPtr));
|
||||
}));
|
||||
}
|
||||
|
||||
return Utf8Marshaller.WithUtf8(apiVersion, apiPtr =>
|
||||
UseHandle(regPtr =>
|
||||
{
|
||||
return ResultHelpers.GetStringResult(
|
||||
API.regorus_alias_registry_normalize_and_wrap(
|
||||
(RegorusAliasRegistry*)regPtr,
|
||||
(byte*)resPtr, (byte*)apiPtr,
|
||||
(byte*)ctxPtr, (byte*)paramsPtr));
|
||||
}));
|
||||
})));
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Denormalize a previously-normalized resource JSON back to ARM format.
|
||||
/// </summary>
|
||||
/// <param name="normalizedJson">The normalized resource JSON</param>
|
||||
/// <param name="apiVersion">API version string, or null to use default alias paths</param>
|
||||
/// <returns>Denormalized ARM JSON string</returns>
|
||||
public string? Denormalize(string normalizedJson, string? apiVersion = null)
|
||||
{
|
||||
return Utf8Marshaller.WithUtf8(normalizedJson, normPtr =>
|
||||
@@ -131,23 +113,16 @@ namespace Regorus
|
||||
(byte*)normPtr, null));
|
||||
});
|
||||
}
|
||||
else
|
||||
{
|
||||
return Utf8Marshaller.WithUtf8(apiVersion, apiPtr =>
|
||||
UseHandle(regPtr =>
|
||||
{
|
||||
return ResultHelpers.GetStringResult(
|
||||
API.regorus_alias_registry_denormalize(
|
||||
(RegorusAliasRegistry*)regPtr,
|
||||
(byte*)normPtr, (byte*)apiPtr));
|
||||
}));
|
||||
}
|
||||
});
|
||||
}
|
||||
|
||||
private static string? CheckAndDropResult(RegorusResult result)
|
||||
{
|
||||
return ResultHelpers.GetStringResult(result);
|
||||
return Utf8Marshaller.WithUtf8(apiVersion, apiPtr =>
|
||||
UseHandle(regPtr =>
|
||||
{
|
||||
return ResultHelpers.GetStringResult(
|
||||
API.regorus_alias_registry_denormalize(
|
||||
(RegorusAliasRegistry*)regPtr,
|
||||
(byte*)normPtr, (byte*)apiPtr));
|
||||
}));
|
||||
});
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
69
bindings/csharp/Regorus/AliasRegistryBuilder.cs
Normal file
69
bindings/csharp/Regorus/AliasRegistryBuilder.cs
Normal file
@@ -0,0 +1,69 @@
|
||||
// Copyright (c) Microsoft Corporation.
|
||||
// Licensed under the MIT License.
|
||||
|
||||
using System;
|
||||
using Regorus.Internal;
|
||||
|
||||
#nullable enable
|
||||
namespace Regorus
|
||||
{
|
||||
/// <summary>
|
||||
/// Mutable, single-threaded builder for <see cref="AliasRegistry"/>.
|
||||
/// Load alias data, then call <see cref="Build"/> to freeze the registry.
|
||||
/// </summary>
|
||||
public unsafe sealed class AliasRegistryBuilder : SafeHandleWrapper
|
||||
{
|
||||
/// <summary>
|
||||
/// Create an empty alias registry builder.
|
||||
/// </summary>
|
||||
public AliasRegistryBuilder()
|
||||
: base(RegorusAliasRegistryBuilderHandle.Create(), nameof(AliasRegistryBuilder))
|
||||
{
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Load control-plane alias data (array of ProviderAliases) from a JSON string.
|
||||
/// </summary>
|
||||
public void LoadJson(string json)
|
||||
{
|
||||
Utf8Marshaller.WithUtf8(json, jsonPtr =>
|
||||
{
|
||||
UseHandle(builderPtr =>
|
||||
{
|
||||
ResultHelpers.GetStringResult(API.regorus_alias_registry_builder_load_json(
|
||||
(RegorusAliasRegistryBuilder*)builderPtr,
|
||||
(byte*)jsonPtr));
|
||||
});
|
||||
});
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Load a data-plane policy manifest from a JSON string.
|
||||
/// </summary>
|
||||
public void LoadManifest(string json)
|
||||
{
|
||||
Utf8Marshaller.WithUtf8(json, jsonPtr =>
|
||||
{
|
||||
UseHandle(builderPtr =>
|
||||
{
|
||||
ResultHelpers.GetStringResult(API.regorus_alias_registry_builder_load_manifest(
|
||||
(RegorusAliasRegistryBuilder*)builderPtr,
|
||||
(byte*)jsonPtr));
|
||||
});
|
||||
});
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Freeze the builder into an immutable, thread-safe alias registry.
|
||||
/// </summary>
|
||||
public AliasRegistry Build()
|
||||
{
|
||||
return UseHandle(builderPtr =>
|
||||
{
|
||||
var registryPtr = ResultHelpers.GetPointerResult(
|
||||
API.regorus_alias_registry_builder_build((RegorusAliasRegistryBuilder*)builderPtr));
|
||||
return new AliasRegistry(RegorusAliasRegistryHandle.FromPointer(registryPtr));
|
||||
});
|
||||
}
|
||||
}
|
||||
}
|
||||
183
bindings/csharp/Regorus/AzurePolicyCompiler.cs
Normal file
183
bindings/csharp/Regorus/AzurePolicyCompiler.cs
Normal file
@@ -0,0 +1,183 @@
|
||||
// Copyright (c) Microsoft Corporation.
|
||||
// Licensed under the MIT License.
|
||||
|
||||
using System;
|
||||
using Regorus.Internal;
|
||||
|
||||
#nullable enable
|
||||
namespace Regorus
|
||||
{
|
||||
/// <summary>
|
||||
/// Provides static methods for compiling Azure Policy JSON definitions
|
||||
/// into RVM programs that can be executed by <see cref="Rvm"/>.
|
||||
/// </summary>
|
||||
/// <remarks>
|
||||
/// <para>
|
||||
/// This class bridges the gap between Azure Policy JSON (the native
|
||||
/// Azure policy language with <c>policyRule</c>, <c>field</c>,
|
||||
/// <c>equals</c>, etc.) and Regorus's RVM execution engine.
|
||||
/// </para>
|
||||
///
|
||||
/// <para>
|
||||
/// <b>Typical workflow:</b>
|
||||
/// </para>
|
||||
/// <list type="number">
|
||||
/// <item>Load alias definitions with <see cref="AliasRegistryBuilder"/> and freeze them into an <see cref="AliasRegistry"/>.</item>
|
||||
/// <item>Normalize the ARM resource via <see cref="AliasRegistry.NormalizeAndWrap"/>.</item>
|
||||
/// <item>Compile the JSON policyRule with <see cref="CompilePolicyRule"/> or the
|
||||
/// full definition with <see cref="CompilePolicyDefinition"/>.</item>
|
||||
/// <item>Execute the resulting <see cref="Program"/> in an <see cref="Rvm"/>
|
||||
/// instance with the normalized input.</item>
|
||||
/// </list>
|
||||
///
|
||||
/// <para>
|
||||
/// <b>Context-dependent policies:</b> Policies that use context functions
|
||||
/// such as <c>subscription()</c>, <c>resourceGroup()</c>, or
|
||||
/// <c>requestContext()</c> require the VM context to be set separately via
|
||||
/// <see cref="Rvm.SetContextJson"/> before execution. The context JSON
|
||||
/// returned by <see cref="AliasRegistry.NormalizeAndWrap"/> is passed as
|
||||
/// <c>input.context</c> but is <b>not</b> automatically wired into the VM's
|
||||
/// ambient context — the caller must do both:
|
||||
/// <c>vm.SetInputJson(envelope)</c> and <c>vm.SetContextJson(contextJson)</c>.
|
||||
/// </para>
|
||||
/// </remarks>
|
||||
public static unsafe class AzurePolicyCompiler
|
||||
{
|
||||
/// <summary>
|
||||
/// Compile an Azure Policy JSON policy rule into an RVM <see cref="Program"/>.
|
||||
/// </summary>
|
||||
/// <param name="aliasRegistry">
|
||||
/// Alias registry for resolving fully-qualified alias names in field
|
||||
/// references. Pass <c>null</c> if no alias resolution is needed.
|
||||
/// <para>
|
||||
/// <b>Warning:</b> When <c>null</c>, alias field references compile as raw
|
||||
/// property paths and will silently produce incorrect evaluation results for
|
||||
/// policies that use aliases. Modify/Append effect policies will also skip
|
||||
/// the compile-time modifiability validation. Only pass <c>null</c> when the
|
||||
/// policy is known to contain no alias references (e.g. simple type/location
|
||||
/// checks or unit-test scenarios).
|
||||
/// </para>
|
||||
/// </param>
|
||||
/// <param name="policyRuleJson">
|
||||
/// JSON string containing the policyRule object, e.g.
|
||||
/// <c>{ "if": { "field": "type", "equals": "..." }, "then": { "effect": "deny" } }</c>
|
||||
/// </param>
|
||||
/// <returns>
|
||||
/// A compiled <see cref="Program"/> ready to be loaded into an
|
||||
/// <see cref="Rvm"/> instance.
|
||||
/// </returns>
|
||||
/// <exception cref="ArgumentNullException">
|
||||
/// Thrown when <paramref name="policyRuleJson"/> is <c>null</c>.
|
||||
/// </exception>
|
||||
/// <exception cref="Exception">
|
||||
/// Thrown when parsing or compilation fails.
|
||||
/// </exception>
|
||||
public static Program CompilePolicyRule(AliasRegistry? aliasRegistry, string policyRuleJson)
|
||||
{
|
||||
if (policyRuleJson is null)
|
||||
{
|
||||
throw new ArgumentNullException(nameof(policyRuleJson));
|
||||
}
|
||||
|
||||
return Utf8Marshaller.WithUtf8(policyRuleJson, rulePtr =>
|
||||
{
|
||||
if (aliasRegistry is null)
|
||||
{
|
||||
var result = API.regorus_compile_azure_policy_rule(
|
||||
null, (byte*)rulePtr);
|
||||
return GetProgramResult(result);
|
||||
}
|
||||
else
|
||||
{
|
||||
return aliasRegistry.UseHandleForInterop(regPtr =>
|
||||
{
|
||||
var result = API.regorus_compile_azure_policy_rule(
|
||||
(RegorusAliasRegistry*)regPtr, (byte*)rulePtr);
|
||||
return GetProgramResult(result);
|
||||
});
|
||||
}
|
||||
});
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Compile a full Azure Policy definition JSON into an RVM <see cref="Program"/>.
|
||||
/// </summary>
|
||||
/// <param name="aliasRegistry">
|
||||
/// Alias registry for resolving fully-qualified alias names in field
|
||||
/// references. Pass <c>null</c> if no alias resolution is needed.
|
||||
/// <para>
|
||||
/// <b>Warning:</b> When <c>null</c>, alias field references compile as raw
|
||||
/// property paths and will silently produce incorrect evaluation results for
|
||||
/// policies that use aliases. Modify/Append effect policies will also skip
|
||||
/// the compile-time modifiability validation. Only pass <c>null</c> when the
|
||||
/// policy is known to contain no alias references (e.g. simple type/location
|
||||
/// checks or unit-test scenarios).
|
||||
/// </para>
|
||||
/// </param>
|
||||
/// <param name="policyDefinitionJson">
|
||||
/// JSON string containing the full policy definition, which includes
|
||||
/// <c>policyRule</c>, <c>parameters</c>, <c>displayName</c>, etc.
|
||||
/// Accepted in both wrapped and unwrapped forms.
|
||||
/// </param>
|
||||
/// <returns>
|
||||
/// A compiled <see cref="Program"/> ready to be loaded into an
|
||||
/// <see cref="Rvm"/> instance.
|
||||
/// </returns>
|
||||
/// <exception cref="ArgumentNullException">
|
||||
/// Thrown when <paramref name="policyDefinitionJson"/> is <c>null</c>.
|
||||
/// </exception>
|
||||
/// <exception cref="Exception">
|
||||
/// Thrown when parsing or compilation fails.
|
||||
/// </exception>
|
||||
public static Program CompilePolicyDefinition(AliasRegistry? aliasRegistry, string policyDefinitionJson)
|
||||
{
|
||||
if (policyDefinitionJson is null)
|
||||
{
|
||||
throw new ArgumentNullException(nameof(policyDefinitionJson));
|
||||
}
|
||||
|
||||
return Utf8Marshaller.WithUtf8(policyDefinitionJson, defnPtr =>
|
||||
{
|
||||
if (aliasRegistry is null)
|
||||
{
|
||||
var result = API.regorus_compile_azure_policy_definition(
|
||||
null, (byte*)defnPtr);
|
||||
return GetProgramResult(result);
|
||||
}
|
||||
else
|
||||
{
|
||||
return aliasRegistry.UseHandleForInterop(regPtr =>
|
||||
{
|
||||
var result = API.regorus_compile_azure_policy_definition(
|
||||
(RegorusAliasRegistry*)regPtr, (byte*)defnPtr);
|
||||
return GetProgramResult(result);
|
||||
});
|
||||
}
|
||||
});
|
||||
}
|
||||
|
||||
private static Program GetProgramResult(RegorusResult result)
|
||||
{
|
||||
try
|
||||
{
|
||||
if (result.status != RegorusStatus.Ok)
|
||||
{
|
||||
var message = Utf8Marshaller.FromUtf8(result.error_message);
|
||||
throw result.status.CreateException(message);
|
||||
}
|
||||
|
||||
if (result.data_type != RegorusDataType.Pointer || result.pointer_value == null)
|
||||
{
|
||||
throw new Exception("Expected program pointer but got different data type");
|
||||
}
|
||||
|
||||
var handle = RegorusProgramHandle.FromPointer((IntPtr)result.pointer_value);
|
||||
return new Program(handle);
|
||||
}
|
||||
finally
|
||||
{
|
||||
API.regorus_result_drop(result);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
@@ -178,6 +178,14 @@ namespace Regorus.Internal
|
||||
[DllImport(LibraryName, EntryPoint = "regorus_rvm_set_input", CallingConvention = CallingConvention.Cdecl, ExactSpelling = true)]
|
||||
internal static extern RegorusResult regorus_rvm_set_input(RegorusRvm* vm, byte* input_json);
|
||||
|
||||
/// <summary>
|
||||
/// Set the context document for the RVM.
|
||||
/// The context provides host-supplied ambient data (e.g. resourceGroup(), subscription())
|
||||
/// that Azure Policy functions can access.
|
||||
/// </summary>
|
||||
[DllImport(LibraryName, EntryPoint = "regorus_rvm_set_context", CallingConvention = CallingConvention.Cdecl, ExactSpelling = true)]
|
||||
internal static extern RegorusResult regorus_rvm_set_context(RegorusRvm* vm, byte* context_json);
|
||||
|
||||
/// <summary>
|
||||
/// Execute the program.
|
||||
/// </summary>
|
||||
@@ -490,6 +498,20 @@ namespace Regorus.Internal
|
||||
[DllImport(LibraryName, EntryPoint = "regorus_compile_policy_for_target", CallingConvention = CallingConvention.Cdecl, ExactSpelling = true)]
|
||||
internal static extern RegorusResult regorus_compile_policy_for_target(byte* data_json, RegorusPolicyModule* modules, UIntPtr modules_len);
|
||||
|
||||
/// <summary>
|
||||
/// Compile an Azure Policy JSON policy rule into an RVM program.
|
||||
/// </summary>
|
||||
[DllImport(LibraryName, EntryPoint = "regorus_compile_azure_policy_rule", CallingConvention = CallingConvention.Cdecl, ExactSpelling = true)]
|
||||
internal static extern RegorusResult regorus_compile_azure_policy_rule(
|
||||
RegorusAliasRegistry* registry, byte* policy_rule_json);
|
||||
|
||||
/// <summary>
|
||||
/// Compile a full Azure Policy definition JSON into an RVM program.
|
||||
/// </summary>
|
||||
[DllImport(LibraryName, EntryPoint = "regorus_compile_azure_policy_definition", CallingConvention = CallingConvention.Cdecl, ExactSpelling = true)]
|
||||
internal static extern RegorusResult regorus_compile_azure_policy_definition(
|
||||
RegorusAliasRegistry* registry, byte* policy_definition_json);
|
||||
|
||||
#endregion
|
||||
|
||||
#region Compiled Policy Methods
|
||||
@@ -673,10 +695,34 @@ namespace Regorus.Internal
|
||||
#region Alias Registry Methods
|
||||
|
||||
/// <summary>
|
||||
/// Create a new, empty AliasRegistry.
|
||||
/// Create a new alias registry builder.
|
||||
/// </summary>
|
||||
[DllImport(LibraryName, EntryPoint = "regorus_alias_registry_new", CallingConvention = CallingConvention.Cdecl, ExactSpelling = true)]
|
||||
internal static extern RegorusAliasRegistry* regorus_alias_registry_new();
|
||||
[DllImport(LibraryName, EntryPoint = "regorus_alias_registry_builder_new", CallingConvention = CallingConvention.Cdecl, ExactSpelling = true)]
|
||||
internal static extern RegorusAliasRegistryBuilder* regorus_alias_registry_builder_new();
|
||||
|
||||
/// <summary>
|
||||
/// Drop an alias registry builder.
|
||||
/// </summary>
|
||||
[DllImport(LibraryName, EntryPoint = "regorus_alias_registry_builder_drop", CallingConvention = CallingConvention.Cdecl, ExactSpelling = true)]
|
||||
internal static extern void regorus_alias_registry_builder_drop(RegorusAliasRegistryBuilder* builder);
|
||||
|
||||
/// <summary>
|
||||
/// Load control-plane alias data into the builder.
|
||||
/// </summary>
|
||||
[DllImport(LibraryName, EntryPoint = "regorus_alias_registry_builder_load_json", CallingConvention = CallingConvention.Cdecl, ExactSpelling = true)]
|
||||
internal static extern RegorusResult regorus_alias_registry_builder_load_json(RegorusAliasRegistryBuilder* builder, byte* json);
|
||||
|
||||
/// <summary>
|
||||
/// Load a data-plane policy manifest into the builder.
|
||||
/// </summary>
|
||||
[DllImport(LibraryName, EntryPoint = "regorus_alias_registry_builder_load_manifest", CallingConvention = CallingConvention.Cdecl, ExactSpelling = true)]
|
||||
internal static extern RegorusResult regorus_alias_registry_builder_load_manifest(RegorusAliasRegistryBuilder* builder, byte* json);
|
||||
|
||||
/// <summary>
|
||||
/// Freeze a builder into an immutable alias registry.
|
||||
/// </summary>
|
||||
[DllImport(LibraryName, EntryPoint = "regorus_alias_registry_builder_build", CallingConvention = CallingConvention.Cdecl, ExactSpelling = true)]
|
||||
internal static extern RegorusResult regorus_alias_registry_builder_build(RegorusAliasRegistryBuilder* builder);
|
||||
|
||||
/// <summary>
|
||||
/// Drop an AliasRegistry.
|
||||
@@ -684,18 +730,6 @@ namespace Regorus.Internal
|
||||
[DllImport(LibraryName, EntryPoint = "regorus_alias_registry_drop", CallingConvention = CallingConvention.Cdecl, ExactSpelling = true)]
|
||||
internal static extern void regorus_alias_registry_drop(RegorusAliasRegistry* registry);
|
||||
|
||||
/// <summary>
|
||||
/// Load control-plane alias data (array of ProviderAliases) into the registry.
|
||||
/// </summary>
|
||||
[DllImport(LibraryName, EntryPoint = "regorus_alias_registry_load_json", CallingConvention = CallingConvention.Cdecl, ExactSpelling = true)]
|
||||
internal static extern RegorusResult regorus_alias_registry_load_json(RegorusAliasRegistry* registry, byte* json);
|
||||
|
||||
/// <summary>
|
||||
/// Load a data-plane policy manifest into the registry.
|
||||
/// </summary>
|
||||
[DllImport(LibraryName, EntryPoint = "regorus_alias_registry_load_manifest", CallingConvention = CallingConvention.Cdecl, ExactSpelling = true)]
|
||||
internal static extern RegorusResult regorus_alias_registry_load_manifest(RegorusAliasRegistry* registry, byte* json);
|
||||
|
||||
/// <summary>
|
||||
/// Return the number of resource types loaded in the alias registry.
|
||||
/// </summary>
|
||||
@@ -923,6 +957,14 @@ namespace Regorus.Internal
|
||||
public byte* content;
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Wrapper for AliasRegistryBuilder.
|
||||
/// </summary>
|
||||
[StructLayout(LayoutKind.Sequential)]
|
||||
internal unsafe partial struct RegorusAliasRegistryBuilder
|
||||
{
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Wrapper for AliasRegistry.
|
||||
/// </summary>
|
||||
|
||||
@@ -15,7 +15,7 @@ namespace Regorus
|
||||
/// </summary>
|
||||
public unsafe sealed class Program : SafeHandleWrapper
|
||||
{
|
||||
private Program(RegorusProgramHandle handle)
|
||||
internal Program(RegorusProgramHandle handle)
|
||||
: base(handle, nameof(Program))
|
||||
{
|
||||
}
|
||||
|
||||
@@ -69,5 +69,29 @@ namespace Regorus.Internal
|
||||
API.regorus_result_drop(result);
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
internal static IntPtr GetPointerResult(RegorusResult result)
|
||||
{
|
||||
try
|
||||
{
|
||||
if (result.status != RegorusStatus.Ok)
|
||||
{
|
||||
var message = Utf8Marshaller.FromUtf8(result.error_message);
|
||||
throw result.status.CreateException(message);
|
||||
}
|
||||
|
||||
if (result.data_type != RegorusDataType.Pointer || result.pointer_value == null)
|
||||
{
|
||||
throw new InvalidOperationException("Expected pointer result.");
|
||||
}
|
||||
|
||||
return (IntPtr)result.pointer_value;
|
||||
}
|
||||
finally
|
||||
{
|
||||
API.regorus_result_drop(result);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
@@ -106,6 +106,24 @@ namespace Regorus
|
||||
});
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Set the context document for the VM.
|
||||
/// The context provides host-supplied ambient data (e.g. resourceGroup(),
|
||||
/// subscription()) that Azure Policy functions can access via LoadContext
|
||||
/// instructions.
|
||||
/// </summary>
|
||||
public void SetContextJson(string contextJson)
|
||||
{
|
||||
Utf8Marshaller.WithUtf8(contextJson, contextPtr =>
|
||||
{
|
||||
UseHandle(vmPtr =>
|
||||
{
|
||||
CheckAndDropResult(API.regorus_rvm_set_context((RegorusRvm*)vmPtr, (byte*)contextPtr));
|
||||
return 0;
|
||||
});
|
||||
});
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Set the execution mode (0 = run-to-completion, 1 = suspendable).
|
||||
/// </summary>
|
||||
|
||||
@@ -184,28 +184,48 @@ namespace Regorus
|
||||
}
|
||||
}
|
||||
|
||||
internal sealed class RegorusAliasRegistryBuilderHandle : SafeHandleZeroOrMinusOneIsInvalid
|
||||
{
|
||||
private RegorusAliasRegistryBuilderHandle() : base(ownsHandle: true)
|
||||
{
|
||||
}
|
||||
|
||||
internal static RegorusAliasRegistryBuilderHandle Create()
|
||||
{
|
||||
unsafe
|
||||
{
|
||||
var raw = Internal.API.regorus_alias_registry_builder_new();
|
||||
if (raw is null)
|
||||
{
|
||||
throw new InvalidOperationException("Failed to create Regorus alias registry builder.");
|
||||
}
|
||||
|
||||
var handle = new RegorusAliasRegistryBuilderHandle();
|
||||
handle.SetHandle((IntPtr)raw);
|
||||
return handle;
|
||||
}
|
||||
}
|
||||
|
||||
protected override bool ReleaseHandle()
|
||||
{
|
||||
if (!IsInvalid)
|
||||
{
|
||||
unsafe
|
||||
{
|
||||
Internal.API.regorus_alias_registry_builder_drop((Internal.RegorusAliasRegistryBuilder*)handle);
|
||||
}
|
||||
SetHandle(IntPtr.Zero);
|
||||
}
|
||||
return true;
|
||||
}
|
||||
}
|
||||
|
||||
internal sealed class RegorusAliasRegistryHandle : SafeHandleZeroOrMinusOneIsInvalid
|
||||
{
|
||||
private RegorusAliasRegistryHandle() : base(ownsHandle: true)
|
||||
{
|
||||
}
|
||||
|
||||
internal static RegorusAliasRegistryHandle Create()
|
||||
{
|
||||
unsafe
|
||||
{
|
||||
var raw = Internal.API.regorus_alias_registry_new();
|
||||
if (raw is null)
|
||||
{
|
||||
throw new InvalidOperationException("Failed to create Regorus alias registry.");
|
||||
}
|
||||
|
||||
var handle = new RegorusAliasRegistryHandle();
|
||||
handle.SetHandle((IntPtr)raw);
|
||||
return handle;
|
||||
}
|
||||
}
|
||||
|
||||
internal static RegorusAliasRegistryHandle FromPointer(IntPtr pointer)
|
||||
{
|
||||
if (pointer == IntPtr.Zero)
|
||||
|
||||
@@ -232,6 +232,9 @@ allow if {
|
||||
|
||||
Console.WriteLine("\n8. RVM host await (suspend/resume):");
|
||||
DemonstrateRvmHostAwait();
|
||||
|
||||
Console.WriteLine("\n9. Azure Policy JSON compilation:");
|
||||
DemonstrateAzurePolicyJsonCompilation();
|
||||
}
|
||||
|
||||
static void DemonstrateConcurrentEvaluation(Regorus.CompiledPolicy compiledPolicy)
|
||||
@@ -492,4 +495,80 @@ allow if {
|
||||
var resumed = vm.Resume("{\"tier\":\"gold\"}");
|
||||
Console.WriteLine($"HostAwait resumed result: {resumed}");
|
||||
}
|
||||
|
||||
// Azure Policy JSON constants
|
||||
private const string STORAGE_ALIASES_JSON = @"[{
|
||||
""namespace"": ""Microsoft.Storage"",
|
||||
""resourceTypes"": [{
|
||||
""resourceType"": ""storageAccounts"",
|
||||
""capabilities"": ""SupportsTags, SupportsLocation"",
|
||||
""aliases"": [
|
||||
{
|
||||
""name"": ""Microsoft.Storage/storageAccounts/supportsHttpsTrafficOnly"",
|
||||
""defaultPath"": ""properties.supportsHttpsTrafficOnly"",
|
||||
""paths"": []
|
||||
}
|
||||
]
|
||||
}]
|
||||
}]";
|
||||
|
||||
private const string HTTPS_DENY_RULE = @"{
|
||||
""if"": {
|
||||
""allOf"": [
|
||||
{ ""field"": ""type"", ""equals"": ""Microsoft.Storage/storageAccounts"" },
|
||||
{ ""field"": ""Microsoft.Storage/storageAccounts/supportsHttpsTrafficOnly"", ""equals"": false }
|
||||
]
|
||||
},
|
||||
""then"": { ""effect"": ""deny"" }
|
||||
}";
|
||||
|
||||
static void DemonstrateAzurePolicyJsonCompilation()
|
||||
{
|
||||
// 1. Set up alias registry
|
||||
using var registry = Regorus.AliasRegistry.FromJson(STORAGE_ALIASES_JSON);
|
||||
Console.WriteLine("Loaded storage account aliases");
|
||||
|
||||
// 2. Compile the JSON policy rule directly (no Rego needed)
|
||||
using var program = Regorus.AzurePolicyCompiler.CompilePolicyRule(registry, HTTPS_DENY_RULE);
|
||||
Console.WriteLine("Compiled Azure Policy JSON rule to RVM program");
|
||||
|
||||
// 3. Normalize an ARM resource
|
||||
var armResource = @"{
|
||||
""type"": ""Microsoft.Storage/storageAccounts"",
|
||||
""name"": ""insecurestorage"",
|
||||
""location"": ""eastus"",
|
||||
""properties"": { ""supportsHttpsTrafficOnly"": false }
|
||||
}";
|
||||
var envelope = registry.NormalizeAndWrap(armResource, apiVersion: null, contextJson: "{}", parametersJson: "{}");
|
||||
Console.WriteLine($"Normalized ARM resource to evaluation envelope");
|
||||
|
||||
// 4. Execute in the RVM
|
||||
// Note: For policies using context functions (subscription(), resourceGroup()),
|
||||
// call vm.SetContextJson(contextJson) before execution. The context from
|
||||
// NormalizeAndWrap is in the envelope but must also be set on the VM separately.
|
||||
using var vm = new Regorus.Rvm();
|
||||
vm.LoadProgram(program);
|
||||
vm.SetInputJson(envelope!);
|
||||
// vm.SetContextJson(contextJson); // ← required for context-dependent policies
|
||||
var result = vm.ExecuteEntryPoint("main");
|
||||
Console.WriteLine($"Evaluation result (non-compliant): {result}");
|
||||
|
||||
// 5. Test with a compliant resource
|
||||
var compliantResource = @"{
|
||||
""type"": ""Microsoft.Storage/storageAccounts"",
|
||||
""name"": ""securestorage"",
|
||||
""location"": ""eastus"",
|
||||
""properties"": { ""supportsHttpsTrafficOnly"": true }
|
||||
}";
|
||||
var compliantEnvelope = registry.NormalizeAndWrap(compliantResource, apiVersion: null, contextJson: "{}", parametersJson: "{}");
|
||||
using var vm2 = new Regorus.Rvm();
|
||||
vm2.LoadProgram(program);
|
||||
vm2.SetInputJson(compliantEnvelope!);
|
||||
var compliantResult = vm2.ExecuteEntryPoint("main");
|
||||
Console.WriteLine($"Evaluation result (compliant): {compliantResult}");
|
||||
|
||||
// 6. Demonstrate program serialization
|
||||
var binary = program.SerializeBinary();
|
||||
Console.WriteLine($"Serialized program size: {binary.Length} bytes");
|
||||
}
|
||||
}
|
||||
|
||||
100
bindings/ffi/Cargo.lock
generated
100
bindings/ffi/Cargo.lock
generated
@@ -98,9 +98,9 @@ checksum = "7f202df86484c868dbad7eaa557ef785d5c66295e41b460ef922eca0723b842c"
|
||||
|
||||
[[package]]
|
||||
name = "autocfg"
|
||||
version = "1.5.0"
|
||||
version = "1.5.1"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "c08606f8c3cbf4ce6ec8e28fb0014a2c086708fe954eaa885384a6165172e7e8"
|
||||
checksum = "f2032f911046de80f0a198e0901378627c33f59ea0ac00e363d481118bd70a53"
|
||||
|
||||
[[package]]
|
||||
name = "bit-set"
|
||||
@@ -141,9 +141,9 @@ dependencies = [
|
||||
|
||||
[[package]]
|
||||
name = "bumpalo"
|
||||
version = "3.20.2"
|
||||
version = "3.20.3"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "5d20789868f4b01b2f2caec9f5c4e0213b41e3e5702a50157d699ae31ced2fcb"
|
||||
checksum = "72f5acc6cb2ba439de613abc23857ec3d78374d8ed5ac84e9d11336e87da8649"
|
||||
|
||||
[[package]]
|
||||
name = "bytecount"
|
||||
@@ -153,9 +153,9 @@ checksum = "175812e0be2bccb6abe50bb8d566126198344f707e304f45c648fd8f2cc0365e"
|
||||
|
||||
[[package]]
|
||||
name = "cbindgen"
|
||||
version = "0.29.2"
|
||||
version = "0.29.3"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "befbfd072a8e81c02f8c507aefce431fe5e7d051f83d48a23ffc9b9fe5a11799"
|
||||
checksum = "c95537b45400390270fae69ac098d057c8f5399001cde9d04f700c105ddfff2d"
|
||||
dependencies = [
|
||||
"clap",
|
||||
"heck",
|
||||
@@ -172,9 +172,9 @@ dependencies = [
|
||||
|
||||
[[package]]
|
||||
name = "cc"
|
||||
version = "1.2.61"
|
||||
version = "1.2.62"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "d16d90359e986641506914ba71350897565610e87ce0ad9e6f28569db3dd5c6d"
|
||||
checksum = "a1dce859f0832a7d088c4f1119888ab94ef4b5d6795d1ce05afb7fe159d79f98"
|
||||
dependencies = [
|
||||
"find-msvc-tools",
|
||||
"shlex",
|
||||
@@ -285,9 +285,9 @@ checksum = "d0a5c400df2834b80a4c3327b3aad3a4c4cd4de0629063962b03235697506a28"
|
||||
|
||||
[[package]]
|
||||
name = "dashmap"
|
||||
version = "6.1.0"
|
||||
version = "6.2.1"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "5041cc499144891f3790297212f32a74fb938e5136a14943f338ef9e0ae276cf"
|
||||
checksum = "e6361d5c062261c78a176addb82d4c821ae42bed6089de0e12603cd25de2059c"
|
||||
dependencies = [
|
||||
"cfg-if",
|
||||
"crossbeam-utils",
|
||||
@@ -305,9 +305,9 @@ checksum = "a4ae5f15dda3c708c0ade84bfee31ccab44a3da4f88015ed22f63732abe300c8"
|
||||
|
||||
[[package]]
|
||||
name = "displaydoc"
|
||||
version = "0.2.5"
|
||||
version = "0.2.6"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "97369cbbc041bc366949bc74d34658d6cda5621039731c6310521892a3a20ae0"
|
||||
checksum = "1ac70aa55017e108007fbaf5aa0f54b021c98f92ff8af59d42eda9da96e3dd4f"
|
||||
dependencies = [
|
||||
"proc-macro2",
|
||||
"quote",
|
||||
@@ -508,9 +508,9 @@ dependencies = [
|
||||
|
||||
[[package]]
|
||||
name = "hashbrown"
|
||||
version = "0.17.0"
|
||||
version = "0.17.1"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "4f467dd6dccf739c208452f8014c75c18bb8301b050ad1cfb27153803edb0f51"
|
||||
checksum = "ed5909b6e89a2db4456e54cd5f673791d7eca6732202bbf2a9cc504fe2f9b84a"
|
||||
dependencies = [
|
||||
"foldhash 0.2.0",
|
||||
]
|
||||
@@ -687,7 +687,7 @@ source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "d466e9454f08e4a911e14806c24e16fba1b4c121d1ea474396f396069cf949d9"
|
||||
dependencies = [
|
||||
"equivalent",
|
||||
"hashbrown 0.17.0",
|
||||
"hashbrown 0.17.1",
|
||||
"serde",
|
||||
"serde_core",
|
||||
]
|
||||
@@ -712,9 +712,9 @@ checksum = "8f42a60cbdf9a97f5d2305f08a87dc4e09308d1276d28c869c684d7777685682"
|
||||
|
||||
[[package]]
|
||||
name = "js-sys"
|
||||
version = "0.3.98"
|
||||
version = "0.3.99"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "67df7112613f8bfd9150013a0314e196f4800d3201ae742489d999db2f979f08"
|
||||
checksum = "142bc4740e452c1e57ade0cbc129f139c9093e354346f0872ef985f4f5cf5f11"
|
||||
dependencies = [
|
||||
"cfg-if",
|
||||
"futures-util",
|
||||
@@ -724,9 +724,9 @@ dependencies = [
|
||||
|
||||
[[package]]
|
||||
name = "jsonschema"
|
||||
version = "0.46.4"
|
||||
version = "0.46.5"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "fc59d2432e047d6090ba1d83c782d0128bd6203857978218f5614dbd3287281f"
|
||||
checksum = "6a5fe5206f06e589caf25e79fc05ccdf91fca745685fe9fe1a13bbdfb479a631"
|
||||
dependencies = [
|
||||
"ahash",
|
||||
"bytecount",
|
||||
@@ -790,9 +790,9 @@ dependencies = [
|
||||
|
||||
[[package]]
|
||||
name = "log"
|
||||
version = "0.4.29"
|
||||
version = "0.4.30"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "5e5032e24019045c762d3c0f28f5b6b8bbf38563a65908389bf7978758920897"
|
||||
checksum = "616ec5685824bcc94416c6d4a7a446eea774a31efd7062c8480ba6fd06d7a6e5"
|
||||
|
||||
[[package]]
|
||||
name = "lru"
|
||||
@@ -802,9 +802,9 @@ checksum = "8a860605968fce16869fd239cf4237a82f3ac470723415db603b0e8b6c8d4fb9"
|
||||
|
||||
[[package]]
|
||||
name = "memchr"
|
||||
version = "2.8.0"
|
||||
version = "2.8.1"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "f8ca58f447f06ed17d5fc4043ce1b10dd205e060fb3ce5b979b8ed8e59ff3f79"
|
||||
checksum = "6b947ae49db0d222b1dbc6b113ce7248a3fc3a6ca21b696717bfc000ba4484d8"
|
||||
|
||||
[[package]]
|
||||
name = "micromap"
|
||||
@@ -1082,9 +1082,9 @@ dependencies = [
|
||||
|
||||
[[package]]
|
||||
name = "referencing"
|
||||
version = "0.46.4"
|
||||
version = "0.46.5"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "cb674900ca31acd75c4aaf63f48e43e719631c0539ea5a9e64163d1296bcb730"
|
||||
checksum = "69e4e17ef386c5383591d07623d3de49cbc601156e7582973e6db98d66a57de2"
|
||||
dependencies = [
|
||||
"ahash",
|
||||
"fluent-uri",
|
||||
@@ -1128,7 +1128,7 @@ checksum = "dc897dd8d9e8bd1ed8cdad82b5966c3e0ecae09fb1907d58efaa013543185d0a"
|
||||
|
||||
[[package]]
|
||||
name = "regorus"
|
||||
version = "0.10.0"
|
||||
version = "0.10.1"
|
||||
dependencies = [
|
||||
"anyhow",
|
||||
"chrono",
|
||||
@@ -1136,7 +1136,7 @@ dependencies = [
|
||||
"dashmap",
|
||||
"data-encoding",
|
||||
"globset",
|
||||
"hashbrown 0.17.0",
|
||||
"hashbrown 0.17.1",
|
||||
"icu_casemap",
|
||||
"indexmap",
|
||||
"ipnet",
|
||||
@@ -1163,7 +1163,7 @@ dependencies = [
|
||||
|
||||
[[package]]
|
||||
name = "regorus-ffi"
|
||||
version = "0.10.0"
|
||||
version = "0.10.1"
|
||||
dependencies = [
|
||||
"anyhow",
|
||||
"cbindgen",
|
||||
@@ -1174,7 +1174,7 @@ dependencies = [
|
||||
|
||||
[[package]]
|
||||
name = "regorus-mimalloc"
|
||||
version = "2.2.6"
|
||||
version = "2.2.7"
|
||||
dependencies = [
|
||||
"regorus-mimalloc-sys",
|
||||
]
|
||||
@@ -1255,9 +1255,9 @@ dependencies = [
|
||||
|
||||
[[package]]
|
||||
name = "serde_json"
|
||||
version = "1.0.149"
|
||||
version = "1.0.150"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "83fc039473c5595ace860d8c4fafa220ff474b3fc6bfdb4293327f1a37e94d86"
|
||||
checksum = "e8014e44b4736ed0538adeecded0fce2a272f22dc9578a7eb6b2d9993c74cfb9"
|
||||
dependencies = [
|
||||
"itoa",
|
||||
"memchr",
|
||||
@@ -1314,9 +1314,9 @@ checksum = "67b1b7a3b5fe4f1376887184045fcf45c69e92af734b7aaddc05fb777b6fbd03"
|
||||
|
||||
[[package]]
|
||||
name = "spin"
|
||||
version = "0.10.0"
|
||||
version = "0.12.0"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "d5fe4ccb98d9c292d56fec89a5e07da7fc4cf0dc11e156b41793132775d3e591"
|
||||
checksum = "1527984ca054dfca79333baec451042863f485fbee01b7bf6d911de915cac865"
|
||||
|
||||
[[package]]
|
||||
name = "stable_deref_trait"
|
||||
@@ -1426,7 +1426,7 @@ version = "1.1.2+spec-1.1.0"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "a2abe9b86193656635d2411dc43050282ca48aa31c2451210f4202550afb7526"
|
||||
dependencies = [
|
||||
"winnow 1.0.2",
|
||||
"winnow 1.0.3",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
@@ -1535,9 +1535,9 @@ dependencies = [
|
||||
|
||||
[[package]]
|
||||
name = "wasm-bindgen"
|
||||
version = "0.2.121"
|
||||
version = "0.2.122"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "49ace1d07c165b0864824eee619580c4689389afa9dc9ed3a4c75040d82e6790"
|
||||
checksum = "3ed04576f974d2b2fba0f38c51dbc5518011e38c36bf1143164be765528fd409"
|
||||
dependencies = [
|
||||
"cfg-if",
|
||||
"once_cell",
|
||||
@@ -1548,9 +1548,9 @@ dependencies = [
|
||||
|
||||
[[package]]
|
||||
name = "wasm-bindgen-macro"
|
||||
version = "0.2.121"
|
||||
version = "0.2.122"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "8e68e6f4afd367a562002c05637acb8578ff2dea1943df76afb9e83d177c8578"
|
||||
checksum = "916151b09da36bd82f6615cbf3a419e2f0ba23a03c6160e8e92eb6bd4aa1dec6"
|
||||
dependencies = [
|
||||
"quote",
|
||||
"wasm-bindgen-macro-support",
|
||||
@@ -1558,9 +1558,9 @@ dependencies = [
|
||||
|
||||
[[package]]
|
||||
name = "wasm-bindgen-macro-support"
|
||||
version = "0.2.121"
|
||||
version = "0.2.122"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "d95a9ec35c64b2a7cb35d3fead40c4238d0940c86d107136999567a4703259f2"
|
||||
checksum = "299047362ccbfce148b67ab7e73349f77748e00c8296f9542adfad2ad82c5c5e"
|
||||
dependencies = [
|
||||
"bumpalo",
|
||||
"proc-macro2",
|
||||
@@ -1571,9 +1571,9 @@ dependencies = [
|
||||
|
||||
[[package]]
|
||||
name = "wasm-bindgen-shared"
|
||||
version = "0.2.121"
|
||||
version = "0.2.122"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "c4e0100b01e9f0d03189a92b96772a1fb998639d981193d7dbab487302513441"
|
||||
checksum = "9a929b2c61f11ba3e9bc35b50c1f25cb38e0e892c0c231ae2b8cf78d5dad4437"
|
||||
dependencies = [
|
||||
"unicode-ident",
|
||||
]
|
||||
@@ -1688,9 +1688,9 @@ checksum = "df79d97927682d2fd8adb29682d1140b343be4ac0f08fd68b7765d9c059d3945"
|
||||
|
||||
[[package]]
|
||||
name = "winnow"
|
||||
version = "1.0.2"
|
||||
version = "1.0.3"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "2ee1708bef14716a11bae175f579062d4554d95be2c6829f518df847b7b3fdd0"
|
||||
checksum = "0592e1c9d151f854e6fd382574c3a0855250e1d9b2f99d9281c6e6391af352f1"
|
||||
|
||||
[[package]]
|
||||
name = "wit-bindgen"
|
||||
@@ -1817,18 +1817,18 @@ dependencies = [
|
||||
|
||||
[[package]]
|
||||
name = "zerocopy"
|
||||
version = "0.8.48"
|
||||
version = "0.8.49"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "eed437bf9d6692032087e337407a86f04cd8d6a16a37199ed57949d415bd68e9"
|
||||
checksum = "bce33a6288fa3f072a8c2c7d0f2fdbb90e28298f0135c1f99b96c3db2efcc60b"
|
||||
dependencies = [
|
||||
"zerocopy-derive",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "zerocopy-derive"
|
||||
version = "0.8.48"
|
||||
version = "0.8.49"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "70e3cd084b1788766f53af483dd21f93881ff30d7320490ec3ef7526d203bad4"
|
||||
checksum = "8fd425244944f4ab65ccff928e7323354c5a018c75838362fdce749dfad2ee1e"
|
||||
dependencies = [
|
||||
"proc-macro2",
|
||||
"quote",
|
||||
@@ -1837,9 +1837,9 @@ dependencies = [
|
||||
|
||||
[[package]]
|
||||
name = "zerofrom"
|
||||
version = "0.1.7"
|
||||
version = "0.1.8"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "69faa1f2a1ea75661980b013019ed6687ed0e83d069bc1114e2cc74c6c04c4df"
|
||||
checksum = "0ec05a11813ea801ff6d75110ad09cd0824ddba17dfe17128ea0d5f68e6c5272"
|
||||
dependencies = [
|
||||
"zerofrom-derive",
|
||||
]
|
||||
|
||||
@@ -2,7 +2,7 @@
|
||||
|
||||
[package]
|
||||
name = "regorus-ffi"
|
||||
version = "0.10.0"
|
||||
version = "0.10.1"
|
||||
edition = "2021"
|
||||
license = "MIT AND Apache-2.0 AND BSD-3-Clause"
|
||||
|
||||
@@ -13,7 +13,7 @@ crate-type = ["cdylib", "staticlib"]
|
||||
[dependencies]
|
||||
anyhow = "1.0"
|
||||
regorus = { path = "../..", default-features = false }
|
||||
serde_json = "1.0.140"
|
||||
serde_json = "1.0.150"
|
||||
parking_lot = { version = "0.12", optional = true }
|
||||
|
||||
[profile.release]
|
||||
|
||||
@@ -5,66 +5,108 @@
|
||||
|
||||
#![cfg(feature = "azure_policy")]
|
||||
|
||||
use crate::common::{from_c_str, to_ref, RegorusResult, RegorusStatus};
|
||||
use crate::common::{from_c_str, to_ref, to_shared_ref, RegorusResult, RegorusStatus};
|
||||
use crate::panic_guard::with_unwind_guard;
|
||||
|
||||
use alloc::boxed::Box;
|
||||
use alloc::format;
|
||||
use alloc::string::String;
|
||||
use anyhow::Result;
|
||||
use core::ffi::c_char;
|
||||
use core::ptr;
|
||||
use alloc::sync::Arc;
|
||||
use anyhow::{anyhow, Result};
|
||||
use core::ffi::{c_char, c_void};
|
||||
use core::{mem, ptr};
|
||||
|
||||
use regorus::languages::azure_policy::aliases::AliasRegistry;
|
||||
|
||||
/// Opaque wrapper for `AliasRegistry`.
|
||||
pub struct RegorusAliasRegistry {
|
||||
registry: AliasRegistry,
|
||||
}
|
||||
|
||||
// ---------------------------------------------------------------------------
|
||||
// Lifecycle
|
||||
// ---------------------------------------------------------------------------
|
||||
|
||||
/// Create a new, empty `AliasRegistry`.
|
||||
/// Mutable builder for `AliasRegistry`.
|
||||
///
|
||||
/// The caller must eventually call `regorus_alias_registry_drop` to free the handle.
|
||||
#[no_mangle]
|
||||
pub extern "C" fn regorus_alias_registry_new() -> *mut RegorusAliasRegistry {
|
||||
let wrapper = RegorusAliasRegistry {
|
||||
registry: AliasRegistry::new(),
|
||||
};
|
||||
Box::into_raw(Box::new(wrapper))
|
||||
/// This handle is intentionally single-threaded and must not be used
|
||||
/// concurrently. Callers should finish loading alias data and then freeze it
|
||||
/// into a `RegorusAliasRegistry` via `regorus_alias_registry_builder_build`.
|
||||
pub struct RegorusAliasRegistryBuilder {
|
||||
registry: AliasRegistry,
|
||||
built: bool,
|
||||
}
|
||||
|
||||
/// Drop a `RegorusAliasRegistry`.
|
||||
impl RegorusAliasRegistryBuilder {
|
||||
fn new() -> Self {
|
||||
Self {
|
||||
registry: AliasRegistry::new(),
|
||||
built: false,
|
||||
}
|
||||
}
|
||||
|
||||
fn registry_mut(&mut self) -> Result<&mut AliasRegistry> {
|
||||
if self.built {
|
||||
return Err(anyhow!("alias registry builder has already been built"));
|
||||
}
|
||||
Ok(&mut self.registry)
|
||||
}
|
||||
|
||||
fn build(&mut self) -> Result<RegorusAliasRegistry> {
|
||||
if self.built {
|
||||
return Err(anyhow!("alias registry builder has already been built"));
|
||||
}
|
||||
|
||||
self.built = true;
|
||||
Ok(RegorusAliasRegistry {
|
||||
registry: Arc::new(mem::replace(&mut self.registry, AliasRegistry::new())),
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
/// Frozen, immutable alias registry.
|
||||
pub struct RegorusAliasRegistry {
|
||||
registry: Arc<AliasRegistry>,
|
||||
}
|
||||
|
||||
impl RegorusAliasRegistry {
|
||||
/// Return a shared reference to the inner registry for use by the compiler.
|
||||
pub(crate) fn inner(&self) -> Arc<AliasRegistry> {
|
||||
Arc::clone(&self.registry)
|
||||
}
|
||||
}
|
||||
|
||||
// ---------------------------------------------------------------------------
|
||||
// Builder lifecycle
|
||||
// ---------------------------------------------------------------------------
|
||||
|
||||
/// Create a new, empty `AliasRegistry` builder.
|
||||
///
|
||||
/// The caller must eventually call `regorus_alias_registry_builder_drop`.
|
||||
#[no_mangle]
|
||||
pub extern "C" fn regorus_alias_registry_drop(registry: *mut RegorusAliasRegistry) {
|
||||
if let Ok(r) = to_ref(registry) {
|
||||
pub extern "C" fn regorus_alias_registry_builder_new() -> *mut RegorusAliasRegistryBuilder {
|
||||
Box::into_raw(Box::new(RegorusAliasRegistryBuilder::new()))
|
||||
}
|
||||
|
||||
/// Drop a `RegorusAliasRegistryBuilder`.
|
||||
#[no_mangle]
|
||||
pub extern "C" fn regorus_alias_registry_builder_drop(builder: *mut RegorusAliasRegistryBuilder) {
|
||||
if let Ok(builder) = to_ref(builder) {
|
||||
unsafe {
|
||||
let _ = Box::from_raw(ptr::from_mut(r));
|
||||
let _ = Box::from_raw(ptr::from_mut(builder));
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// ---------------------------------------------------------------------------
|
||||
// Loading
|
||||
// Builder loading
|
||||
// ---------------------------------------------------------------------------
|
||||
|
||||
/// Load control-plane alias data (array of `ProviderAliases`) into the registry.
|
||||
/// Load control-plane alias data (array of `ProviderAliases`) into the builder.
|
||||
///
|
||||
/// `json` must be a valid null-terminated UTF-8 string containing the JSON
|
||||
/// array returned by `Get-AzPolicyAlias` or the static
|
||||
/// `ResourceTypesAndAliases.json` file.
|
||||
#[no_mangle]
|
||||
pub extern "C" fn regorus_alias_registry_load_json(
|
||||
registry: *mut RegorusAliasRegistry,
|
||||
pub extern "C" fn regorus_alias_registry_builder_load_json(
|
||||
builder: *mut RegorusAliasRegistryBuilder,
|
||||
json: *const c_char,
|
||||
) -> RegorusResult {
|
||||
with_unwind_guard(|| {
|
||||
let output = || -> Result<()> {
|
||||
let json_str = from_c_str(json)?;
|
||||
to_ref(registry)?.registry.load_from_json(&json_str)?;
|
||||
to_ref(builder)?.registry_mut()?.load_from_json(&json_str)?;
|
||||
Ok(())
|
||||
}();
|
||||
|
||||
@@ -78,20 +120,20 @@ pub extern "C" fn regorus_alias_registry_load_json(
|
||||
})
|
||||
}
|
||||
|
||||
/// Load a data-plane policy manifest into the registry.
|
||||
/// Load a data-plane policy manifest into the builder.
|
||||
///
|
||||
/// `json` must be a valid null-terminated UTF-8 string containing a single
|
||||
/// `DataPolicyManifest` JSON object.
|
||||
#[no_mangle]
|
||||
pub extern "C" fn regorus_alias_registry_load_manifest(
|
||||
registry: *mut RegorusAliasRegistry,
|
||||
pub extern "C" fn regorus_alias_registry_builder_load_manifest(
|
||||
builder: *mut RegorusAliasRegistryBuilder,
|
||||
json: *const c_char,
|
||||
) -> RegorusResult {
|
||||
with_unwind_guard(|| {
|
||||
let output = || -> Result<()> {
|
||||
let json_str = from_c_str(json)?;
|
||||
to_ref(registry)?
|
||||
.registry
|
||||
to_ref(builder)?
|
||||
.registry_mut()?
|
||||
.load_data_policy_manifest_json(&json_str)?;
|
||||
Ok(())
|
||||
}();
|
||||
@@ -106,16 +148,52 @@ pub extern "C" fn regorus_alias_registry_load_manifest(
|
||||
})
|
||||
}
|
||||
|
||||
/// Freeze a builder into an immutable `RegorusAliasRegistry`.
|
||||
#[no_mangle]
|
||||
pub extern "C" fn regorus_alias_registry_builder_build(
|
||||
builder: *mut RegorusAliasRegistryBuilder,
|
||||
) -> RegorusResult {
|
||||
with_unwind_guard(|| {
|
||||
let output = || -> Result<*mut RegorusAliasRegistry> {
|
||||
let registry = to_ref(builder)?.build()?;
|
||||
Ok(Box::into_raw(Box::new(registry)))
|
||||
}();
|
||||
|
||||
match output {
|
||||
Ok(registry) => RegorusResult::ok_pointer(registry as *mut c_void),
|
||||
Err(e) => {
|
||||
RegorusResult::err_with_message(RegorusStatus::InvalidArgument, format!("{e}"))
|
||||
}
|
||||
}
|
||||
})
|
||||
}
|
||||
|
||||
// ---------------------------------------------------------------------------
|
||||
// Queries
|
||||
// Frozen registry lifecycle
|
||||
// ---------------------------------------------------------------------------
|
||||
|
||||
/// Drop a `RegorusAliasRegistry`.
|
||||
#[no_mangle]
|
||||
pub extern "C" fn regorus_alias_registry_drop(registry: *mut RegorusAliasRegistry) {
|
||||
if let Ok(registry) = to_ref(registry) {
|
||||
unsafe {
|
||||
let _ = Box::from_raw(ptr::from_mut(registry));
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// ---------------------------------------------------------------------------
|
||||
// Frozen registry queries
|
||||
// ---------------------------------------------------------------------------
|
||||
|
||||
/// Return the number of resource types loaded in the alias registry.
|
||||
#[no_mangle]
|
||||
pub extern "C" fn regorus_alias_registry_len(registry: *mut RegorusAliasRegistry) -> RegorusResult {
|
||||
pub extern "C" fn regorus_alias_registry_len(
|
||||
registry: *const RegorusAliasRegistry,
|
||||
) -> RegorusResult {
|
||||
with_unwind_guard(|| {
|
||||
let output = || -> Result<i64> {
|
||||
let len = to_ref(registry)?.registry.len();
|
||||
let len = to_shared_ref(registry)?.registry.len();
|
||||
Ok(len as i64)
|
||||
}();
|
||||
|
||||
@@ -134,15 +212,9 @@ pub extern "C" fn regorus_alias_registry_len(registry: *mut RegorusAliasRegistry
|
||||
///
|
||||
/// Returns a JSON string:
|
||||
/// `{ "resource": <normalized>, "context": <context>, "parameters": <params> }`.
|
||||
///
|
||||
/// * `resource_json` – raw ARM resource JSON
|
||||
/// * `api_version` – API version string (e.g. `"2023-01-01"`), or null to use
|
||||
/// the default alias paths
|
||||
/// * `context_json` – JSON object for additional context (pass `"{}"` if none)
|
||||
/// * `parameters_json` – JSON object of policy parameter values (pass `"{}"` if none)
|
||||
#[no_mangle]
|
||||
pub extern "C" fn regorus_alias_registry_normalize_and_wrap(
|
||||
registry: *mut RegorusAliasRegistry,
|
||||
registry: *const RegorusAliasRegistry,
|
||||
resource_json: *const c_char,
|
||||
api_version: *const c_char,
|
||||
context_json: *const c_char,
|
||||
@@ -168,7 +240,7 @@ pub extern "C" fn regorus_alias_registry_normalize_and_wrap(
|
||||
let context = regorus::Value::from_json_str(&context_str)?;
|
||||
let params = regorus::Value::from_json_str(¶ms_str)?;
|
||||
|
||||
let wrapped = to_ref(registry)?.registry.normalize_and_wrap(
|
||||
let wrapped = to_shared_ref(registry)?.registry.normalize_and_wrap(
|
||||
&resource,
|
||||
api_ver.as_deref(),
|
||||
Some(context),
|
||||
@@ -185,14 +257,9 @@ pub extern "C" fn regorus_alias_registry_normalize_and_wrap(
|
||||
}
|
||||
|
||||
/// Denormalize a previously-normalized resource JSON back to ARM format.
|
||||
///
|
||||
/// * `normalized_json` – the normalized resource JSON
|
||||
/// * `api_version` – API version string, or null to use the default alias paths
|
||||
///
|
||||
/// Returns the denormalized ARM JSON string.
|
||||
#[no_mangle]
|
||||
pub extern "C" fn regorus_alias_registry_denormalize(
|
||||
registry: *mut RegorusAliasRegistry,
|
||||
registry: *const RegorusAliasRegistry,
|
||||
normalized_json: *const c_char,
|
||||
api_version: *const c_char,
|
||||
) -> RegorusResult {
|
||||
@@ -212,7 +279,7 @@ pub extern "C" fn regorus_alias_registry_denormalize(
|
||||
|
||||
let normalized = regorus::Value::from_json_str(&normalized_str)?;
|
||||
|
||||
let result = to_ref(registry)?
|
||||
let result = to_shared_ref(registry)?
|
||||
.registry
|
||||
.denormalize(&normalized, api_ver.as_deref());
|
||||
result.to_json_str()
|
||||
@@ -232,12 +299,10 @@ mod tests {
|
||||
use core::ffi::CStr;
|
||||
use std::ffi::CString;
|
||||
|
||||
/// Helper: create a C string from a Rust &str.
|
||||
fn c(s: &str) -> CString {
|
||||
CString::new(s).expect("CString::new failed")
|
||||
}
|
||||
|
||||
/// Helper: assert a RegorusResult has Ok status and extract string output.
|
||||
fn assert_ok_string(r: &RegorusResult) -> String {
|
||||
assert_eq!(r.status, RegorusStatus::Ok, "expected Ok status");
|
||||
assert!(!r.output.is_null(), "expected non-null output");
|
||||
@@ -248,12 +313,51 @@ mod tests {
|
||||
s
|
||||
}
|
||||
|
||||
/// Helper: assert a RegorusResult has Ok status with integer output.
|
||||
fn assert_ok_int(r: &RegorusResult) -> i64 {
|
||||
assert_eq!(r.status, RegorusStatus::Ok, "expected Ok status");
|
||||
r.int_value
|
||||
}
|
||||
|
||||
fn assert_ok_pointer(r: &RegorusResult) -> *mut c_void {
|
||||
assert_eq!(r.status, RegorusStatus::Ok, "expected Ok status");
|
||||
assert!(matches!(
|
||||
r.data_type,
|
||||
crate::common::RegorusDataType::Pointer
|
||||
));
|
||||
assert!(!r.pointer_value.is_null());
|
||||
r.pointer_value
|
||||
}
|
||||
|
||||
fn build_registry_with_json(json: &str) -> *mut RegorusAliasRegistry {
|
||||
let builder = regorus_alias_registry_builder_new();
|
||||
let json = c(json);
|
||||
|
||||
let r = regorus_alias_registry_builder_load_json(builder, json.as_ptr());
|
||||
assert_eq!(r.status, RegorusStatus::Ok);
|
||||
regorus_result_drop(r);
|
||||
|
||||
let r = regorus_alias_registry_builder_build(builder);
|
||||
let registry = assert_ok_pointer(&r) as *mut RegorusAliasRegistry;
|
||||
regorus_result_drop(r);
|
||||
regorus_alias_registry_builder_drop(builder);
|
||||
registry
|
||||
}
|
||||
|
||||
fn build_registry_with_manifest(json: &str) -> *mut RegorusAliasRegistry {
|
||||
let builder = regorus_alias_registry_builder_new();
|
||||
let json = c(json);
|
||||
|
||||
let r = regorus_alias_registry_builder_load_manifest(builder, json.as_ptr());
|
||||
assert_eq!(r.status, RegorusStatus::Ok);
|
||||
regorus_result_drop(r);
|
||||
|
||||
let r = regorus_alias_registry_builder_build(builder);
|
||||
let registry = assert_ok_pointer(&r) as *mut RegorusAliasRegistry;
|
||||
regorus_result_drop(r);
|
||||
regorus_alias_registry_builder_drop(builder);
|
||||
registry
|
||||
}
|
||||
|
||||
const ALIASES: &str = r#"[{
|
||||
"namespace": "Microsoft.Storage",
|
||||
"resourceTypes": [{
|
||||
@@ -279,20 +383,21 @@ mod tests {
|
||||
}"#;
|
||||
|
||||
#[test]
|
||||
fn lifecycle_new_and_drop() {
|
||||
let reg = regorus_alias_registry_new();
|
||||
assert!(!reg.is_null());
|
||||
regorus_alias_registry_drop(reg);
|
||||
fn lifecycle_builder_build_and_drop() {
|
||||
let builder = regorus_alias_registry_builder_new();
|
||||
assert!(!builder.is_null());
|
||||
|
||||
let r = regorus_alias_registry_builder_build(builder);
|
||||
let registry = assert_ok_pointer(&r) as *mut RegorusAliasRegistry;
|
||||
regorus_result_drop(r);
|
||||
|
||||
regorus_alias_registry_builder_drop(builder);
|
||||
regorus_alias_registry_drop(registry);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn load_json_and_check_len() {
|
||||
let reg = regorus_alias_registry_new();
|
||||
let json = c(ALIASES);
|
||||
|
||||
let r = regorus_alias_registry_load_json(reg, json.as_ptr());
|
||||
assert_eq!(r.status, RegorusStatus::Ok);
|
||||
regorus_result_drop(r);
|
||||
let reg = build_registry_with_json(ALIASES);
|
||||
|
||||
let r = regorus_alias_registry_len(reg);
|
||||
assert_eq!(assert_ok_int(&r), 1);
|
||||
@@ -303,12 +408,7 @@ mod tests {
|
||||
|
||||
#[test]
|
||||
fn load_manifest_and_check_len() {
|
||||
let reg = regorus_alias_registry_new();
|
||||
let json = c(MANIFEST);
|
||||
|
||||
let r = regorus_alias_registry_load_manifest(reg, json.as_ptr());
|
||||
assert_eq!(r.status, RegorusStatus::Ok);
|
||||
regorus_result_drop(r);
|
||||
let reg = build_registry_with_manifest(MANIFEST);
|
||||
|
||||
let r = regorus_alias_registry_len(reg);
|
||||
assert_eq!(assert_ok_int(&r), 1);
|
||||
@@ -319,23 +419,39 @@ mod tests {
|
||||
|
||||
#[test]
|
||||
fn load_invalid_json_returns_error() {
|
||||
let reg = regorus_alias_registry_new();
|
||||
let builder = regorus_alias_registry_builder_new();
|
||||
let bad = c("not valid json");
|
||||
|
||||
let r = regorus_alias_registry_load_json(reg, bad.as_ptr());
|
||||
let r = regorus_alias_registry_builder_load_json(builder, bad.as_ptr());
|
||||
assert_ne!(r.status, RegorusStatus::Ok);
|
||||
regorus_result_drop(r);
|
||||
|
||||
regorus_alias_registry_drop(reg);
|
||||
regorus_alias_registry_builder_drop(builder);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn builder_cannot_be_reused_after_build() {
|
||||
let builder = regorus_alias_registry_builder_new();
|
||||
let r = regorus_alias_registry_builder_build(builder);
|
||||
let registry = assert_ok_pointer(&r) as *mut RegorusAliasRegistry;
|
||||
regorus_result_drop(r);
|
||||
|
||||
let aliases = c(ALIASES);
|
||||
let r = regorus_alias_registry_builder_load_json(builder, aliases.as_ptr());
|
||||
assert_ne!(r.status, RegorusStatus::Ok);
|
||||
regorus_result_drop(r);
|
||||
|
||||
let r = regorus_alias_registry_builder_build(builder);
|
||||
assert_ne!(r.status, RegorusStatus::Ok);
|
||||
regorus_result_drop(r);
|
||||
|
||||
regorus_alias_registry_builder_drop(builder);
|
||||
regorus_alias_registry_drop(registry);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn normalize_and_wrap_round_trip() {
|
||||
let reg = regorus_alias_registry_new();
|
||||
let aliases = c(ALIASES);
|
||||
let r = regorus_alias_registry_load_json(reg, aliases.as_ptr());
|
||||
assert_eq!(r.status, RegorusStatus::Ok);
|
||||
regorus_result_drop(r);
|
||||
let reg = build_registry_with_json(ALIASES);
|
||||
|
||||
let resource = c(r#"{
|
||||
"name": "acct1",
|
||||
@@ -346,7 +462,6 @@ mod tests {
|
||||
let ctx = c(r#"{"resourceGroup": {"name": "rg1"}}"#);
|
||||
let params = c(r#"{"env": "prod"}"#);
|
||||
|
||||
// Normalize
|
||||
let r = regorus_alias_registry_normalize_and_wrap(
|
||||
reg,
|
||||
resource.as_ptr(),
|
||||
@@ -357,7 +472,6 @@ mod tests {
|
||||
let envelope_json = assert_ok_string(&r);
|
||||
regorus_result_drop(r);
|
||||
|
||||
// Parse and verify structure
|
||||
let envelope: serde_json::Value =
|
||||
serde_json::from_str(&envelope_json).expect("invalid JSON output");
|
||||
assert!(
|
||||
@@ -373,16 +487,13 @@ mod tests {
|
||||
"envelope missing 'context'"
|
||||
);
|
||||
|
||||
// The normalized resource should have lowercased alias fields
|
||||
let res = &envelope["resource"];
|
||||
assert_eq!(res["supportshttpstrafficonly"], true);
|
||||
assert_eq!(res["name"], "acct1");
|
||||
|
||||
// Context and parameters should be passed through
|
||||
assert_eq!(envelope["context"]["resourceGroup"]["name"], "rg1");
|
||||
assert_eq!(envelope["parameters"]["env"], "prod");
|
||||
|
||||
// Denormalize the resource portion
|
||||
let resource_json = serde_json::to_string(&res).expect("serialize resource");
|
||||
let norm_cstr = c(&resource_json);
|
||||
|
||||
@@ -392,7 +503,6 @@ mod tests {
|
||||
|
||||
let denorm: serde_json::Value =
|
||||
serde_json::from_str(&denorm_json).expect("invalid denorm JSON");
|
||||
// Should be back under properties with restored casing
|
||||
assert_eq!(
|
||||
denorm["properties"]["supportsHttpsTrafficOnly"], true,
|
||||
"expected restored casing under properties"
|
||||
@@ -403,11 +513,7 @@ mod tests {
|
||||
|
||||
#[test]
|
||||
fn denormalize_invalid_json_returns_error() {
|
||||
let reg = regorus_alias_registry_new();
|
||||
let aliases = c(ALIASES);
|
||||
let r = regorus_alias_registry_load_json(reg, aliases.as_ptr());
|
||||
assert_eq!(r.status, RegorusStatus::Ok);
|
||||
regorus_result_drop(r);
|
||||
let reg = build_registry_with_json(ALIASES);
|
||||
|
||||
let bad = c("not json");
|
||||
let api = c("2023-01-01");
|
||||
@@ -420,11 +526,7 @@ mod tests {
|
||||
|
||||
#[test]
|
||||
fn normalize_data_plane_manifest() {
|
||||
let reg = regorus_alias_registry_new();
|
||||
let manifest = c(MANIFEST);
|
||||
let r = regorus_alias_registry_load_manifest(reg, manifest.as_ptr());
|
||||
assert_eq!(r.status, RegorusStatus::Ok);
|
||||
regorus_result_drop(r);
|
||||
let reg = build_registry_with_manifest(MANIFEST);
|
||||
|
||||
let resource = c(r#"{
|
||||
"type": "Microsoft.KeyVault.Data/vaults/certificates",
|
||||
@@ -453,7 +555,12 @@ mod tests {
|
||||
|
||||
#[test]
|
||||
fn empty_registry_normalize() {
|
||||
let reg = regorus_alias_registry_new();
|
||||
let builder = regorus_alias_registry_builder_new();
|
||||
let r = regorus_alias_registry_builder_build(builder);
|
||||
let reg = assert_ok_pointer(&r) as *mut RegorusAliasRegistry;
|
||||
regorus_result_drop(r);
|
||||
regorus_alias_registry_builder_drop(builder);
|
||||
|
||||
let resource = c(r#"{"name": "test", "type": "Unknown/type", "properties": {"foo": 1}}"#);
|
||||
let api = c("");
|
||||
let ctx = c("{}");
|
||||
@@ -470,7 +577,6 @@ mod tests {
|
||||
regorus_result_drop(r);
|
||||
|
||||
let envelope: serde_json::Value = serde_json::from_str(&json).expect("invalid JSON");
|
||||
// Without aliases, properties should still be flattened
|
||||
assert_eq!(envelope["resource"]["foo"], 1);
|
||||
assert_eq!(envelope["resource"]["name"], "test");
|
||||
|
||||
|
||||
@@ -236,6 +236,10 @@ pub(crate) fn to_ref<'a, T>(t: *mut T) -> Result<&'a mut T> {
|
||||
unsafe { t.as_mut().ok_or_else(|| anyhow!("null pointer")) }
|
||||
}
|
||||
|
||||
pub(crate) fn to_shared_ref<'a, T>(t: *const T) -> Result<&'a T> {
|
||||
unsafe { t.as_ref().ok_or_else(|| anyhow!("null pointer")) }
|
||||
}
|
||||
|
||||
pub(crate) fn to_regorus_result(r: Result<()>) -> RegorusResult {
|
||||
match r {
|
||||
Ok(()) => RegorusResult::ok_void(),
|
||||
|
||||
@@ -1,6 +1,6 @@
|
||||
// Copyright (c) Microsoft Corporation.
|
||||
// Licensed under the MIT License.
|
||||
use crate::common::{from_c_str, RegorusResult, RegorusStatus};
|
||||
use crate::common::{from_c_str, to_shared_ref, RegorusResult, RegorusStatus};
|
||||
use crate::compiled_policy::RegorusCompiledPolicy;
|
||||
use crate::panic_guard::with_unwind_guard;
|
||||
use alloc::boxed::Box;
|
||||
@@ -208,6 +208,220 @@ fn convert_c_modules_to_rust(
|
||||
Ok(policy_modules)
|
||||
}
|
||||
|
||||
// ---------------------------------------------------------------------------
|
||||
// Azure Policy JSON compilation
|
||||
// ---------------------------------------------------------------------------
|
||||
|
||||
/// Compile an Azure Policy JSON policy rule into an RVM program.
|
||||
///
|
||||
/// Parses the JSON `policyRule` (the `{ "if": ..., "then": ... }` object),
|
||||
/// resolves aliases using the provided registry, and compiles the result
|
||||
/// into an RVM [`Program`] that can be loaded into a [`RegorusRvm`].
|
||||
///
|
||||
/// # Parameters
|
||||
/// * `registry` - Alias registry handle, or null.
|
||||
/// * `policy_rule_json` - JSON string containing the policyRule object
|
||||
///
|
||||
/// # Null registry behavior
|
||||
///
|
||||
/// When `registry` is null, compilation proceeds **without alias resolution**.
|
||||
/// Field references that correspond to Azure resource provider aliases
|
||||
/// (e.g. `Microsoft.Storage/storageAccounts/supportsHttpsTrafficOnly`) will
|
||||
/// be compiled as raw property paths rather than being resolved to their
|
||||
/// short forms. This means:
|
||||
///
|
||||
/// - Policies that rely on aliases will **silently produce incorrect
|
||||
/// evaluation results** because the field paths won't match the
|
||||
/// normalized resource structure.
|
||||
/// - **Modify / Append** effect policies will **skip the modifiability
|
||||
/// validation** that normally rejects writes to non-modifiable aliases
|
||||
/// at compile time.
|
||||
///
|
||||
/// Pass null only when the policy is known to contain no alias references
|
||||
/// (e.g. simple `type` / `location` checks, or in unit-test scenarios).
|
||||
///
|
||||
/// # Returns
|
||||
/// Returns a `RegorusResult` containing a `RegorusProgram` pointer on success.
|
||||
///
|
||||
/// # Safety
|
||||
/// `policy_rule_json` must be a valid null-terminated UTF-8 string.
|
||||
/// If `registry` is non-null it must be a valid `RegorusAliasRegistry` pointer.
|
||||
/// The caller must eventually call `regorus_program_drop` on the returned handle.
|
||||
#[cfg(all(feature = "azure_policy", feature = "rvm"))]
|
||||
#[no_mangle]
|
||||
pub extern "C" fn regorus_compile_azure_policy_rule(
|
||||
registry: *const crate::alias_registry::RegorusAliasRegistry,
|
||||
policy_rule_json: *const c_char,
|
||||
) -> RegorusResult {
|
||||
use crate::alias_registry::RegorusAliasRegistry;
|
||||
use crate::rvm::RegorusProgram;
|
||||
use alloc::sync::Arc;
|
||||
use regorus::languages::azure_policy::{compiler, parser};
|
||||
use regorus::Rc;
|
||||
use regorus::Source;
|
||||
|
||||
with_unwind_guard(|| {
|
||||
let result = || -> Result<RegorusProgram, (RegorusStatus, alloc::string::String)> {
|
||||
let json_str = from_c_str(policy_rule_json).map_err(|e| {
|
||||
(
|
||||
RegorusStatus::InvalidDataFormat,
|
||||
format!("Invalid policy rule JSON string: {e}"),
|
||||
)
|
||||
})?;
|
||||
|
||||
let source = Source::from_contents("policy_rule".into(), json_str).map_err(|e| {
|
||||
(
|
||||
RegorusStatus::InvalidDataFormat,
|
||||
format!("Failed to create source: {e}"),
|
||||
)
|
||||
})?;
|
||||
|
||||
let ast = parser::parse_policy_rule(&source).map_err(|e| {
|
||||
(
|
||||
RegorusStatus::InvalidPolicy,
|
||||
format!("Failed to parse policy rule: {e}"),
|
||||
)
|
||||
})?;
|
||||
|
||||
let program = if registry.is_null() {
|
||||
compiler::compile_policy_rule(&ast)
|
||||
} else {
|
||||
let reg: &RegorusAliasRegistry = to_shared_ref(registry).map_err(|e| {
|
||||
(
|
||||
RegorusStatus::InvalidArgument,
|
||||
format!("Invalid alias registry: {e}"),
|
||||
)
|
||||
})?;
|
||||
compiler::compile_policy_rule_with_aliases(&ast, reg.inner())
|
||||
};
|
||||
|
||||
program
|
||||
.map(|p| RegorusProgram {
|
||||
program: Arc::new(Rc::try_unwrap(p).unwrap_or_else(|rc| (*rc).clone())),
|
||||
})
|
||||
.map_err(|e| {
|
||||
(
|
||||
RegorusStatus::CompilationFailed,
|
||||
format!("Failed to compile policy rule: {e}"),
|
||||
)
|
||||
})
|
||||
}();
|
||||
|
||||
match result {
|
||||
Ok(program) => {
|
||||
RegorusResult::ok_pointer(Box::into_raw(Box::new(program)) as *mut c_void)
|
||||
}
|
||||
Err((status, msg)) => RegorusResult::err_with_message(status, msg),
|
||||
}
|
||||
})
|
||||
}
|
||||
|
||||
/// Compile a full Azure Policy definition JSON into an RVM program.
|
||||
///
|
||||
/// Parses the JSON policy definition (which includes `policyRule`, `parameters`,
|
||||
/// `displayName`, etc.), resolves aliases using the provided registry, and
|
||||
/// compiles the result into an RVM [`Program`].
|
||||
///
|
||||
/// The definition JSON may be in either wrapped or unwrapped form:
|
||||
/// - **Wrapped**: `{ "properties": { "policyRule": ..., "parameters": ... }, "id": ... }`
|
||||
/// - **Unwrapped**: `{ "policyRule": ..., "parameters": ..., "displayName": ... }`
|
||||
///
|
||||
/// # Parameters
|
||||
/// * `registry` - Alias registry handle, or null.
|
||||
/// * `policy_definition_json` - JSON string containing the full policy definition
|
||||
///
|
||||
/// # Null registry behavior
|
||||
///
|
||||
/// When `registry` is null, compilation proceeds **without alias resolution**.
|
||||
/// Field references that correspond to Azure resource provider aliases will
|
||||
/// be compiled as raw property paths rather than being resolved. This means:
|
||||
///
|
||||
/// - Policies that rely on aliases will **silently produce incorrect
|
||||
/// evaluation results**.
|
||||
/// - **Modify / Append** effect policies will **skip the modifiability
|
||||
/// validation** that normally rejects writes to non-modifiable aliases
|
||||
/// at compile time.
|
||||
///
|
||||
/// Pass null only when the policy is known to contain no alias references
|
||||
/// (e.g. simple `type` / `location` checks, or in unit-test scenarios).
|
||||
///
|
||||
/// # Returns
|
||||
/// Returns a `RegorusResult` containing a `RegorusProgram` pointer on success.
|
||||
///
|
||||
/// # Safety
|
||||
/// `policy_definition_json` must be a valid null-terminated UTF-8 string.
|
||||
/// If `registry` is non-null it must be a valid `RegorusAliasRegistry` pointer.
|
||||
/// The caller must eventually call `regorus_program_drop` on the returned handle.
|
||||
#[cfg(all(feature = "azure_policy", feature = "rvm"))]
|
||||
#[no_mangle]
|
||||
pub extern "C" fn regorus_compile_azure_policy_definition(
|
||||
registry: *const crate::alias_registry::RegorusAliasRegistry,
|
||||
policy_definition_json: *const c_char,
|
||||
) -> RegorusResult {
|
||||
use crate::alias_registry::RegorusAliasRegistry;
|
||||
use crate::rvm::RegorusProgram;
|
||||
use alloc::sync::Arc;
|
||||
use regorus::languages::azure_policy::{compiler, parser};
|
||||
use regorus::Rc;
|
||||
use regorus::Source;
|
||||
|
||||
with_unwind_guard(|| {
|
||||
let result = || -> Result<RegorusProgram, (RegorusStatus, alloc::string::String)> {
|
||||
let json_str = from_c_str(policy_definition_json).map_err(|e| {
|
||||
(
|
||||
RegorusStatus::InvalidDataFormat,
|
||||
format!("Invalid policy definition JSON string: {e}"),
|
||||
)
|
||||
})?;
|
||||
|
||||
let source =
|
||||
Source::from_contents("policy_definition".into(), json_str).map_err(|e| {
|
||||
(
|
||||
RegorusStatus::InvalidDataFormat,
|
||||
format!("Failed to create source: {e}"),
|
||||
)
|
||||
})?;
|
||||
|
||||
let defn = parser::parse_policy_definition(&source).map_err(|e| {
|
||||
(
|
||||
RegorusStatus::InvalidPolicy,
|
||||
format!("Failed to parse policy definition: {e}"),
|
||||
)
|
||||
})?;
|
||||
|
||||
let program = if registry.is_null() {
|
||||
compiler::compile_policy_definition(&defn)
|
||||
} else {
|
||||
let reg: &RegorusAliasRegistry = to_shared_ref(registry).map_err(|e| {
|
||||
(
|
||||
RegorusStatus::InvalidArgument,
|
||||
format!("Invalid alias registry: {e}"),
|
||||
)
|
||||
})?;
|
||||
compiler::compile_policy_definition_with_aliases(&defn, reg.inner())
|
||||
};
|
||||
|
||||
program
|
||||
.map(|p| RegorusProgram {
|
||||
program: Arc::new(Rc::try_unwrap(p).unwrap_or_else(|rc| (*rc).clone())),
|
||||
})
|
||||
.map_err(|e| {
|
||||
(
|
||||
RegorusStatus::CompilationFailed,
|
||||
format!("Failed to compile policy definition: {e}"),
|
||||
)
|
||||
})
|
||||
}();
|
||||
|
||||
match result {
|
||||
Ok(program) => {
|
||||
RegorusResult::ok_pointer(Box::into_raw(Box::new(program)) as *mut c_void)
|
||||
}
|
||||
Err((status, msg)) => RegorusResult::err_with_message(status, msg),
|
||||
}
|
||||
})
|
||||
}
|
||||
|
||||
#[cfg(feature = "std")]
|
||||
fn report_module_error(index: usize, kind: &str, err: &anyhow::Error) {
|
||||
eprintln!("Invalid {} at index {}: {}", kind, index, err);
|
||||
@@ -215,3 +429,402 @@ fn report_module_error(index: usize, kind: &str, err: &anyhow::Error) {
|
||||
|
||||
#[cfg(not(feature = "std"))]
|
||||
fn report_module_error(_index: usize, _kind: &str, _err: &anyhow::Error) {}
|
||||
|
||||
// ---------------------------------------------------------------------------
|
||||
// Tests
|
||||
// ---------------------------------------------------------------------------
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
use crate::common::regorus_result_drop;
|
||||
use core::ffi::CStr;
|
||||
use std::ffi::CString;
|
||||
|
||||
fn c(s: &str) -> CString {
|
||||
CString::new(s).expect("CString::new failed")
|
||||
}
|
||||
|
||||
fn assert_ok_pointer(r: &RegorusResult) -> *mut c_void {
|
||||
assert_eq!(
|
||||
r.status,
|
||||
RegorusStatus::Ok,
|
||||
"expected Ok, got {:?}",
|
||||
r.status
|
||||
);
|
||||
assert!(!r.pointer_value.is_null(), "expected non-null pointer");
|
||||
r.pointer_value
|
||||
}
|
||||
|
||||
#[cfg(all(feature = "azure_policy", feature = "rvm"))]
|
||||
mod azure_policy_json {
|
||||
use super::*;
|
||||
use crate::alias_registry::regorus_alias_registry_drop;
|
||||
use crate::rvm::{
|
||||
regorus_program_drop, regorus_rvm_drop, regorus_rvm_execute_entry_point_by_name,
|
||||
regorus_rvm_load_program, regorus_rvm_new, regorus_rvm_set_context,
|
||||
regorus_rvm_set_input, RegorusProgram,
|
||||
};
|
||||
|
||||
const ALIASES: &str = r#"[{
|
||||
"namespace": "Microsoft.Storage",
|
||||
"resourceTypes": [{
|
||||
"resourceType": "storageAccounts",
|
||||
"aliases": [{
|
||||
"name": "Microsoft.Storage/storageAccounts/supportsHttpsTrafficOnly",
|
||||
"defaultPath": "properties.supportsHttpsTrafficOnly",
|
||||
"paths": []
|
||||
}, {
|
||||
"name": "Microsoft.Storage/storageAccounts/minimumTlsVersion",
|
||||
"defaultPath": "properties.minimumTlsVersion",
|
||||
"paths": []
|
||||
}]
|
||||
}]
|
||||
}]"#;
|
||||
|
||||
const SIMPLE_POLICY_RULE: &str = r#"{
|
||||
"if": {
|
||||
"field": "type",
|
||||
"equals": "Microsoft.Storage/storageAccounts"
|
||||
},
|
||||
"then": { "effect": "audit" }
|
||||
}"#;
|
||||
|
||||
const ALIAS_POLICY_RULE: &str = r#"{
|
||||
"if": {
|
||||
"allOf": [
|
||||
{ "field": "type", "equals": "Microsoft.Storage/storageAccounts" },
|
||||
{ "field": "Microsoft.Storage/storageAccounts/supportsHttpsTrafficOnly", "equals": false }
|
||||
]
|
||||
},
|
||||
"then": { "effect": "deny" }
|
||||
}"#;
|
||||
|
||||
const POLICY_DEFINITION: &str = r#"{
|
||||
"displayName": "Require HTTPS for storage accounts",
|
||||
"policyType": "Custom",
|
||||
"mode": "Indexed",
|
||||
"parameters": {
|
||||
"effect": {
|
||||
"type": "String",
|
||||
"defaultValue": "deny"
|
||||
}
|
||||
},
|
||||
"policyRule": {
|
||||
"if": {
|
||||
"allOf": [
|
||||
{ "field": "type", "equals": "Microsoft.Storage/storageAccounts" },
|
||||
{ "field": "Microsoft.Storage/storageAccounts/supportsHttpsTrafficOnly", "equals": false }
|
||||
]
|
||||
},
|
||||
"then": { "effect": "[parameters('effect')]" }
|
||||
}
|
||||
}"#;
|
||||
|
||||
/// Wrap a normalized resource JSON into the input envelope expected by
|
||||
/// the compiled Azure Policy RVM program.
|
||||
fn wrap_input(resource_json: &str, parameters_json: &str) -> String {
|
||||
format!(r#"{{"resource": {resource_json}, "parameters": {parameters_json}}}"#)
|
||||
}
|
||||
|
||||
fn build_registry_with_json(
|
||||
json: &str,
|
||||
) -> *mut crate::alias_registry::RegorusAliasRegistry {
|
||||
let builder = crate::alias_registry::regorus_alias_registry_builder_new();
|
||||
let json_c = c(json);
|
||||
let r = crate::alias_registry::regorus_alias_registry_builder_load_json(
|
||||
builder,
|
||||
json_c.as_ptr(),
|
||||
);
|
||||
assert_eq!(r.status, RegorusStatus::Ok);
|
||||
regorus_result_drop(r);
|
||||
|
||||
let r = crate::alias_registry::regorus_alias_registry_builder_build(builder);
|
||||
let registry =
|
||||
assert_ok_pointer(&r) as *mut crate::alias_registry::RegorusAliasRegistry;
|
||||
regorus_result_drop(r);
|
||||
crate::alias_registry::regorus_alias_registry_builder_drop(builder);
|
||||
registry
|
||||
}
|
||||
|
||||
/// Helper: compile a policy rule, execute it with input, and return the
|
||||
/// result string.
|
||||
unsafe fn compile_and_eval_rule(
|
||||
registry: *const crate::alias_registry::RegorusAliasRegistry,
|
||||
policy_rule: &str,
|
||||
input_json: &str,
|
||||
) -> String {
|
||||
let rule_c = c(policy_rule);
|
||||
let r = regorus_compile_azure_policy_rule(registry, rule_c.as_ptr());
|
||||
let program_ptr = assert_ok_pointer(&r) as *mut RegorusProgram;
|
||||
regorus_result_drop(r);
|
||||
|
||||
let vm = regorus_rvm_new();
|
||||
assert!(!vm.is_null());
|
||||
|
||||
let r = regorus_rvm_load_program(vm, program_ptr);
|
||||
assert_eq!(r.status, RegorusStatus::Ok);
|
||||
regorus_result_drop(r);
|
||||
|
||||
let input_c = c(input_json);
|
||||
let r = regorus_rvm_set_input(vm, input_c.as_ptr());
|
||||
assert_eq!(r.status, RegorusStatus::Ok);
|
||||
regorus_result_drop(r);
|
||||
|
||||
let entry = c("main");
|
||||
let r = regorus_rvm_execute_entry_point_by_name(vm, entry.as_ptr());
|
||||
assert_eq!(r.status, RegorusStatus::Ok, "execute failed");
|
||||
let output = CStr::from_ptr(r.output)
|
||||
.to_str()
|
||||
.expect("invalid UTF-8")
|
||||
.to_string();
|
||||
regorus_result_drop(r);
|
||||
|
||||
regorus_rvm_drop(vm);
|
||||
regorus_program_drop(program_ptr);
|
||||
output
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn compile_simple_rule_no_aliases() {
|
||||
let rule_c = c(SIMPLE_POLICY_RULE);
|
||||
let r = regorus_compile_azure_policy_rule(core::ptr::null_mut(), rule_c.as_ptr());
|
||||
let ptr = assert_ok_pointer(&r);
|
||||
regorus_result_drop(r);
|
||||
regorus_program_drop(ptr as *mut RegorusProgram);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn compile_rule_with_aliases() {
|
||||
let reg = build_registry_with_json(ALIASES);
|
||||
|
||||
let rule_c = c(ALIAS_POLICY_RULE);
|
||||
let r = regorus_compile_azure_policy_rule(reg, rule_c.as_ptr());
|
||||
let ptr = assert_ok_pointer(&r);
|
||||
regorus_result_drop(r);
|
||||
|
||||
regorus_program_drop(ptr as *mut RegorusProgram);
|
||||
regorus_alias_registry_drop(reg);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn compile_and_eval_simple_rule_matching() {
|
||||
let input = wrap_input(r#"{"type":"microsoft.storage/storageaccounts"}"#, "{}");
|
||||
let result =
|
||||
unsafe { compile_and_eval_rule(core::ptr::null_mut(), SIMPLE_POLICY_RULE, &input) };
|
||||
let parsed: serde_json::Value =
|
||||
serde_json::from_str(&result).expect("result should be valid JSON");
|
||||
assert_eq!(
|
||||
parsed["effect"], "audit",
|
||||
"expected audit effect, got: {result}"
|
||||
);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn compile_and_eval_simple_rule_not_matching() {
|
||||
let input = wrap_input(r#"{"type":"microsoft.compute/virtualmachines"}"#, "{}");
|
||||
let result =
|
||||
unsafe { compile_and_eval_rule(core::ptr::null_mut(), SIMPLE_POLICY_RULE, &input) };
|
||||
// When the "if" condition doesn't match, the result should be undefined
|
||||
assert!(
|
||||
result.contains("undefined"),
|
||||
"expected undefined for non-matching input, got: {result}"
|
||||
);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn compile_and_eval_alias_rule_deny() {
|
||||
let reg = build_registry_with_json(ALIASES);
|
||||
|
||||
// Non-compliant resource: HTTPS not enabled (normalized form)
|
||||
let input = wrap_input(
|
||||
r#"{"type": "microsoft.storage/storageaccounts", "supportshttpstrafficonly": false}"#,
|
||||
"{}",
|
||||
);
|
||||
let result = unsafe { compile_and_eval_rule(reg, ALIAS_POLICY_RULE, &input) };
|
||||
let parsed: serde_json::Value = serde_json::from_str(&result).expect("valid JSON");
|
||||
assert_eq!(parsed["effect"], "deny", "expected deny, got: {result}");
|
||||
|
||||
regorus_alias_registry_drop(reg);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn compile_and_eval_alias_rule_compliant() {
|
||||
let reg = build_registry_with_json(ALIASES);
|
||||
|
||||
// Compliant resource: HTTPS enabled (normalized form)
|
||||
let input = wrap_input(
|
||||
r#"{"type": "microsoft.storage/storageaccounts", "supportshttpstrafficonly": true}"#,
|
||||
"{}",
|
||||
);
|
||||
let result = unsafe { compile_and_eval_rule(reg, ALIAS_POLICY_RULE, &input) };
|
||||
assert!(
|
||||
result.contains("undefined"),
|
||||
"expected undefined for compliant resource, got: {result}"
|
||||
);
|
||||
|
||||
regorus_alias_registry_drop(reg);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn compile_definition_no_aliases() {
|
||||
let defn_c = c(POLICY_DEFINITION);
|
||||
let r = regorus_compile_azure_policy_definition(core::ptr::null_mut(), defn_c.as_ptr());
|
||||
let ptr = assert_ok_pointer(&r);
|
||||
regorus_result_drop(r);
|
||||
regorus_program_drop(ptr as *mut RegorusProgram);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn compile_definition_with_aliases_and_eval() {
|
||||
let reg = build_registry_with_json(ALIASES);
|
||||
|
||||
let defn_c = c(POLICY_DEFINITION);
|
||||
let r = regorus_compile_azure_policy_definition(reg, defn_c.as_ptr());
|
||||
let program_ptr = assert_ok_pointer(&r) as *mut RegorusProgram;
|
||||
regorus_result_drop(r);
|
||||
|
||||
// Evaluate with a non-compliant resource (normalized form, wrapped in envelope)
|
||||
unsafe {
|
||||
let vm = regorus_rvm_new();
|
||||
let r = regorus_rvm_load_program(vm, program_ptr);
|
||||
assert_eq!(r.status, RegorusStatus::Ok);
|
||||
regorus_result_drop(r);
|
||||
|
||||
let input_json = wrap_input(
|
||||
r#"{"type": "microsoft.storage/storageaccounts", "supportshttpstrafficonly": false}"#,
|
||||
"{}",
|
||||
);
|
||||
let input = c(&input_json);
|
||||
let r = regorus_rvm_set_input(vm, input.as_ptr());
|
||||
assert_eq!(r.status, RegorusStatus::Ok);
|
||||
regorus_result_drop(r);
|
||||
|
||||
let entry = c("main");
|
||||
let r = regorus_rvm_execute_entry_point_by_name(vm, entry.as_ptr());
|
||||
assert_eq!(r.status, RegorusStatus::Ok);
|
||||
let result = CStr::from_ptr(r.output)
|
||||
.to_str()
|
||||
.expect("UTF-8")
|
||||
.to_string();
|
||||
regorus_result_drop(r);
|
||||
|
||||
let parsed: serde_json::Value = serde_json::from_str(&result).unwrap();
|
||||
// The default parameter value is "deny"
|
||||
assert_eq!(parsed["effect"], "deny", "got: {result}");
|
||||
|
||||
regorus_rvm_drop(vm);
|
||||
regorus_program_drop(program_ptr);
|
||||
}
|
||||
|
||||
regorus_alias_registry_drop(reg);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn invalid_json_returns_error() {
|
||||
let bad = c("not valid json");
|
||||
let r = regorus_compile_azure_policy_rule(core::ptr::null_mut(), bad.as_ptr());
|
||||
assert_ne!(r.status, RegorusStatus::Ok);
|
||||
regorus_result_drop(r);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn invalid_definition_returns_error() {
|
||||
let bad = c(r#"{"not": "a policy definition"}"#);
|
||||
let r = regorus_compile_azure_policy_definition(core::ptr::null_mut(), bad.as_ptr());
|
||||
assert_ne!(r.status, RegorusStatus::Ok);
|
||||
regorus_result_drop(r);
|
||||
}
|
||||
|
||||
/// Policy rule that uses a context function (subscription()).
|
||||
const CONTEXT_POLICY_RULE: &str = r#"{
|
||||
"if": {
|
||||
"allOf": [
|
||||
{ "field": "type", "equals": "Microsoft.Storage/storageAccounts" },
|
||||
{ "value": "[subscription().subscriptionId]", "equals": "sub-123" }
|
||||
]
|
||||
},
|
||||
"then": { "effect": "deny" }
|
||||
}"#;
|
||||
|
||||
#[test]
|
||||
fn context_policy_evaluates_with_set_context() {
|
||||
let rule_c = c(CONTEXT_POLICY_RULE);
|
||||
let r = regorus_compile_azure_policy_rule(core::ptr::null_mut(), rule_c.as_ptr());
|
||||
let program = assert_ok_pointer(&r) as *mut RegorusProgram;
|
||||
regorus_result_drop(r);
|
||||
|
||||
let vm = regorus_rvm_new();
|
||||
assert!(!vm.is_null());
|
||||
|
||||
let r = regorus_rvm_load_program(vm, program);
|
||||
assert_eq!(r.status, RegorusStatus::Ok);
|
||||
regorus_result_drop(r);
|
||||
|
||||
// Set the context with subscription info
|
||||
let context = c(r#"{"subscription": {"subscriptionId": "sub-123"}}"#);
|
||||
let r = regorus_rvm_set_context(vm, context.as_ptr());
|
||||
assert_eq!(r.status, RegorusStatus::Ok);
|
||||
regorus_result_drop(r);
|
||||
|
||||
// Set matching input
|
||||
let input = c(&wrap_input(
|
||||
r#"{"type": "microsoft.storage/storageaccounts"}"#,
|
||||
"{}",
|
||||
));
|
||||
let r = regorus_rvm_set_input(vm, input.as_ptr());
|
||||
assert_eq!(r.status, RegorusStatus::Ok);
|
||||
regorus_result_drop(r);
|
||||
|
||||
let entry = c("main");
|
||||
let r = regorus_rvm_execute_entry_point_by_name(vm, entry.as_ptr());
|
||||
assert_eq!(r.status, RegorusStatus::Ok);
|
||||
let output = unsafe { CStr::from_ptr(r.output) }.to_str().unwrap();
|
||||
assert!(
|
||||
output.contains("deny"),
|
||||
"expected deny effect with matching context, got: {output}"
|
||||
);
|
||||
regorus_result_drop(r);
|
||||
|
||||
regorus_rvm_drop(vm);
|
||||
regorus_program_drop(program);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn context_policy_undefined_without_context() {
|
||||
let rule_c = c(CONTEXT_POLICY_RULE);
|
||||
let r = regorus_compile_azure_policy_rule(core::ptr::null_mut(), rule_c.as_ptr());
|
||||
let program = assert_ok_pointer(&r) as *mut RegorusProgram;
|
||||
regorus_result_drop(r);
|
||||
|
||||
let vm = regorus_rvm_new();
|
||||
assert!(!vm.is_null());
|
||||
|
||||
let r = regorus_rvm_load_program(vm, program);
|
||||
assert_eq!(r.status, RegorusStatus::Ok);
|
||||
regorus_result_drop(r);
|
||||
|
||||
// No context set — subscription() will be undefined
|
||||
let input = c(&wrap_input(
|
||||
r#"{"type": "microsoft.storage/storageaccounts"}"#,
|
||||
"{}",
|
||||
));
|
||||
let r = regorus_rvm_set_input(vm, input.as_ptr());
|
||||
assert_eq!(r.status, RegorusStatus::Ok);
|
||||
regorus_result_drop(r);
|
||||
|
||||
let entry = c("main");
|
||||
let r = regorus_rvm_execute_entry_point_by_name(vm, entry.as_ptr());
|
||||
assert_eq!(r.status, RegorusStatus::Ok);
|
||||
let output = unsafe { CStr::from_ptr(r.output) }.to_str().unwrap();
|
||||
assert!(
|
||||
output.contains("undefined"),
|
||||
"expected undefined without context, got: {output}"
|
||||
);
|
||||
regorus_result_drop(r);
|
||||
|
||||
regorus_rvm_drop(vm);
|
||||
regorus_program_drop(program);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
@@ -39,7 +39,7 @@ pub extern "C" fn regorus_compiled_policy_eval_with_input(
|
||||
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)?
|
||||
let result = to_shared_ref(compiled_policy as *const RegorusCompiledPolicy)?
|
||||
.compiled_policy
|
||||
.eval_with_input(input_value)?;
|
||||
result.to_json_str()
|
||||
@@ -65,7 +65,9 @@ pub extern "C" fn regorus_compiled_policy_get_policy_info(
|
||||
) -> RegorusResult {
|
||||
with_unwind_guard(|| {
|
||||
let output = || -> Result<String> {
|
||||
let info = to_ref(compiled_policy)?.compiled_policy.get_policy_info()?;
|
||||
let info = to_shared_ref(compiled_policy as *const RegorusCompiledPolicy)?
|
||||
.compiled_policy
|
||||
.get_policy_info()?;
|
||||
serde_json::to_string(&info)
|
||||
.map_err(|e| anyhow::anyhow!("Failed to serialize policy info: {}", e))
|
||||
}();
|
||||
|
||||
@@ -2,7 +2,8 @@
|
||||
// Licensed under the MIT License.
|
||||
|
||||
use crate::common::{
|
||||
from_c_str, to_ref, to_regorus_result, to_regorus_string_result, RegorusResult, RegorusStatus,
|
||||
from_c_str, to_ref, to_regorus_result, to_regorus_string_result, to_shared_ref, RegorusResult,
|
||||
RegorusStatus,
|
||||
};
|
||||
use crate::compiled_policy::RegorusCompiledPolicy;
|
||||
use crate::limits::RegorusExecutionTimerConfig;
|
||||
@@ -193,7 +194,7 @@ pub extern "C" fn regorus_engine_new() -> *mut RegorusEngine {
|
||||
///
|
||||
#[no_mangle]
|
||||
pub extern "C" fn regorus_engine_clone(engine: *mut RegorusEngine) -> *mut RegorusEngine {
|
||||
match to_ref(engine) {
|
||||
match to_shared_ref(engine as *const RegorusEngine) {
|
||||
Ok(e) => Box::into_raw(Box::new(e.clone())),
|
||||
_ => ptr::null_mut(),
|
||||
}
|
||||
@@ -223,7 +224,7 @@ pub extern "C" fn regorus_engine_add_policy(
|
||||
) -> RegorusResult {
|
||||
with_unwind_guard(|| {
|
||||
to_regorus_string_result(|| -> Result<String> {
|
||||
let engine = to_ref(engine)?;
|
||||
let engine = to_shared_ref(engine as *const RegorusEngine)?;
|
||||
let mut guard = engine.try_write()?;
|
||||
guard.add_policy(from_c_str(path)?, from_c_str(rego)?)
|
||||
}())
|
||||
@@ -238,7 +239,7 @@ pub extern "C" fn regorus_engine_add_policy_from_file(
|
||||
) -> RegorusResult {
|
||||
with_unwind_guard(|| {
|
||||
to_regorus_string_result(|| -> Result<String> {
|
||||
let engine = to_ref(engine)?;
|
||||
let engine = to_shared_ref(engine as *const RegorusEngine)?;
|
||||
let mut guard = engine.try_write()?;
|
||||
guard.add_policy_from_file(from_c_str(path)?)
|
||||
}())
|
||||
@@ -256,7 +257,7 @@ pub extern "C" fn regorus_engine_add_data_json(
|
||||
) -> RegorusResult {
|
||||
with_unwind_guard(|| {
|
||||
to_regorus_result(|| -> Result<()> {
|
||||
let engine = to_ref(engine)?;
|
||||
let engine = to_shared_ref(engine as *const RegorusEngine)?;
|
||||
let mut guard = engine.try_write()?;
|
||||
guard.add_data(regorus::Value::from_json_str(&from_c_str(data)?)?)
|
||||
}())
|
||||
@@ -270,7 +271,7 @@ pub extern "C" fn regorus_engine_add_data_json(
|
||||
pub extern "C" fn regorus_engine_get_packages(engine: *mut RegorusEngine) -> RegorusResult {
|
||||
with_unwind_guard(|| {
|
||||
to_regorus_string_result(|| -> Result<String> {
|
||||
let engine = to_ref(engine)?;
|
||||
let engine = to_shared_ref(engine as *const RegorusEngine)?;
|
||||
let guard = engine.try_read()?;
|
||||
serde_json::to_string_pretty(&guard.get_packages()?).map_err(anyhow::Error::msg)
|
||||
}())
|
||||
@@ -284,7 +285,7 @@ pub extern "C" fn regorus_engine_get_packages(engine: *mut RegorusEngine) -> Reg
|
||||
pub extern "C" fn regorus_engine_get_policies(engine: *mut RegorusEngine) -> RegorusResult {
|
||||
with_unwind_guard(|| {
|
||||
to_regorus_string_result(|| -> Result<String> {
|
||||
let engine = to_ref(engine)?;
|
||||
let engine = to_shared_ref(engine as *const RegorusEngine)?;
|
||||
let guard = engine.try_read()?;
|
||||
guard.get_policies_as_json()
|
||||
}())
|
||||
@@ -299,7 +300,7 @@ pub extern "C" fn regorus_engine_add_data_from_json_file(
|
||||
) -> RegorusResult {
|
||||
with_unwind_guard(|| {
|
||||
to_regorus_result(|| -> Result<()> {
|
||||
let engine = to_ref(engine)?;
|
||||
let engine = to_shared_ref(engine as *const RegorusEngine)?;
|
||||
let mut guard = engine.try_write()?;
|
||||
guard.add_data(regorus::Value::from_json_file(from_c_str(path)?)?)
|
||||
}())
|
||||
@@ -313,7 +314,7 @@ pub extern "C" fn regorus_engine_add_data_from_json_file(
|
||||
pub extern "C" fn regorus_engine_clear_data(engine: *mut RegorusEngine) -> RegorusResult {
|
||||
with_unwind_guard(|| {
|
||||
to_regorus_result(|| -> Result<()> {
|
||||
let engine = to_ref(engine)?;
|
||||
let engine = to_shared_ref(engine as *const RegorusEngine)?;
|
||||
let mut guard = engine.try_write()?;
|
||||
guard.clear_data();
|
||||
Ok(())
|
||||
@@ -332,7 +333,7 @@ pub extern "C" fn regorus_engine_set_input_json(
|
||||
) -> RegorusResult {
|
||||
with_unwind_guard(|| {
|
||||
to_regorus_result(|| -> Result<()> {
|
||||
let engine = to_ref(engine)?;
|
||||
let engine = to_shared_ref(engine as *const RegorusEngine)?;
|
||||
let mut guard = engine.try_write()?;
|
||||
guard.set_input(regorus::Value::from_json_str(&from_c_str(input)?)?);
|
||||
Ok(())
|
||||
@@ -348,7 +349,7 @@ pub extern "C" fn regorus_engine_set_input_from_json_file(
|
||||
) -> RegorusResult {
|
||||
with_unwind_guard(|| {
|
||||
to_regorus_result(|| -> Result<()> {
|
||||
let engine = to_ref(engine)?;
|
||||
let engine = to_shared_ref(engine as *const RegorusEngine)?;
|
||||
let mut guard = engine.try_write()?;
|
||||
guard.set_input(regorus::Value::from_json_file(from_c_str(path)?)?);
|
||||
Ok(())
|
||||
@@ -367,7 +368,7 @@ pub extern "C" fn regorus_engine_eval_query(
|
||||
) -> RegorusResult {
|
||||
with_unwind_guard(|| {
|
||||
let output = || -> Result<String> {
|
||||
let engine = to_ref(engine)?;
|
||||
let engine = to_shared_ref(engine as *const RegorusEngine)?;
|
||||
let mut guard = engine.try_write()?;
|
||||
let results = guard.eval_query(from_c_str(query)?, false)?;
|
||||
Ok(serde_json::to_string_pretty(&results)?)
|
||||
@@ -390,7 +391,7 @@ pub extern "C" fn regorus_engine_eval_rule(
|
||||
) -> RegorusResult {
|
||||
with_unwind_guard(|| {
|
||||
let output = || -> Result<String> {
|
||||
let engine = to_ref(engine)?;
|
||||
let engine = to_shared_ref(engine as *const RegorusEngine)?;
|
||||
let mut guard = engine.try_write()?;
|
||||
guard.eval_rule(from_c_str(rule)?)?.to_json_str()
|
||||
}();
|
||||
@@ -413,7 +414,7 @@ pub extern "C" fn regorus_engine_set_enable_coverage(
|
||||
) -> RegorusResult {
|
||||
with_unwind_guard(|| {
|
||||
to_regorus_result(|| -> Result<()> {
|
||||
let engine = to_ref(engine)?;
|
||||
let engine = to_shared_ref(engine as *const RegorusEngine)?;
|
||||
let mut guard = engine.try_write()?;
|
||||
guard.set_enable_coverage(enable);
|
||||
Ok(())
|
||||
@@ -429,7 +430,7 @@ pub extern "C" fn regorus_engine_set_enable_coverage(
|
||||
pub extern "C" fn regorus_engine_get_coverage_report(engine: *mut RegorusEngine) -> RegorusResult {
|
||||
with_unwind_guard(|| {
|
||||
let output = || -> Result<String> {
|
||||
let engine = to_ref(engine)?;
|
||||
let engine = to_shared_ref(engine as *const RegorusEngine)?;
|
||||
let guard = engine.try_read()?;
|
||||
Ok(serde_json::to_string_pretty(&guard.get_coverage_report()?)?)
|
||||
}();
|
||||
@@ -451,7 +452,7 @@ pub extern "C" fn regorus_engine_set_strict_builtin_errors(
|
||||
) -> RegorusResult {
|
||||
with_unwind_guard(|| {
|
||||
to_regorus_result(|| -> Result<()> {
|
||||
let engine = to_ref(engine)?;
|
||||
let engine = to_shared_ref(engine as *const RegorusEngine)?;
|
||||
let mut guard = engine.try_write()?;
|
||||
guard.set_strict_builtin_errors(strict);
|
||||
Ok(())
|
||||
@@ -465,18 +466,20 @@ pub extern "C" fn regorus_engine_set_execution_timer_config(
|
||||
engine: *mut RegorusEngine,
|
||||
config: *const RegorusExecutionTimerConfig,
|
||||
) -> RegorusResult {
|
||||
to_regorus_result(|| -> Result<()> {
|
||||
let engine = to_ref(engine)?;
|
||||
let config = unsafe {
|
||||
config
|
||||
.as_ref()
|
||||
.copied()
|
||||
.ok_or_else(|| anyhow!("execution timer config pointer is null"))?
|
||||
};
|
||||
let mut guard = engine.try_write()?;
|
||||
guard.set_execution_timer_config(config.to_execution_timer_config()?);
|
||||
Ok(())
|
||||
}())
|
||||
with_unwind_guard(|| {
|
||||
to_regorus_result(|| -> Result<()> {
|
||||
let engine = to_shared_ref(engine as *const RegorusEngine)?;
|
||||
let config = unsafe {
|
||||
config
|
||||
.as_ref()
|
||||
.copied()
|
||||
.ok_or_else(|| anyhow!("execution timer config pointer is null"))?
|
||||
};
|
||||
let mut guard = engine.try_write()?;
|
||||
guard.set_execution_timer_config(config.to_execution_timer_config()?);
|
||||
Ok(())
|
||||
}())
|
||||
})
|
||||
}
|
||||
|
||||
#[no_mangle]
|
||||
@@ -484,12 +487,14 @@ pub extern "C" fn regorus_engine_set_execution_timer_config(
|
||||
pub extern "C" fn regorus_engine_clear_execution_timer_config(
|
||||
engine: *mut RegorusEngine,
|
||||
) -> RegorusResult {
|
||||
to_regorus_result(|| -> Result<()> {
|
||||
let engine = to_ref(engine)?;
|
||||
let mut guard = engine.try_write()?;
|
||||
guard.clear_execution_timer_config();
|
||||
Ok(())
|
||||
}())
|
||||
with_unwind_guard(|| {
|
||||
to_regorus_result(|| -> Result<()> {
|
||||
let engine = to_shared_ref(engine as *const RegorusEngine)?;
|
||||
let mut guard = engine.try_write()?;
|
||||
guard.clear_execution_timer_config();
|
||||
Ok(())
|
||||
}())
|
||||
})
|
||||
}
|
||||
|
||||
/// Set the policy length limits used when loading policies.
|
||||
@@ -500,7 +505,7 @@ pub extern "C" fn regorus_engine_set_policy_length_config(
|
||||
) -> RegorusResult {
|
||||
with_unwind_guard(|| {
|
||||
to_regorus_result(|| -> Result<()> {
|
||||
let engine = to_ref(engine)?;
|
||||
let engine = to_shared_ref(engine as *const RegorusEngine)?;
|
||||
let mut guard = engine.try_write()?;
|
||||
guard.set_policy_length_config(config.to_policy_length_config()?);
|
||||
Ok(())
|
||||
@@ -515,7 +520,7 @@ pub extern "C" fn regorus_engine_clear_policy_length_config(
|
||||
) -> RegorusResult {
|
||||
with_unwind_guard(|| {
|
||||
to_regorus_result(|| -> Result<()> {
|
||||
let engine = to_ref(engine)?;
|
||||
let engine = to_shared_ref(engine as *const RegorusEngine)?;
|
||||
let mut guard = engine.try_write()?;
|
||||
guard.clear_policy_length_config();
|
||||
Ok(())
|
||||
@@ -533,7 +538,7 @@ pub extern "C" fn regorus_engine_get_coverage_report_pretty(
|
||||
) -> RegorusResult {
|
||||
with_unwind_guard(|| {
|
||||
let output = || -> Result<String> {
|
||||
let engine = to_ref(engine)?;
|
||||
let engine = to_shared_ref(engine as *const RegorusEngine)?;
|
||||
let guard = engine.try_read()?;
|
||||
guard.get_coverage_report()?.to_string_pretty()
|
||||
}();
|
||||
@@ -552,7 +557,7 @@ pub extern "C" fn regorus_engine_get_coverage_report_pretty(
|
||||
pub extern "C" fn regorus_engine_clear_coverage_data(engine: *mut RegorusEngine) -> RegorusResult {
|
||||
with_unwind_guard(|| {
|
||||
to_regorus_result(|| -> Result<()> {
|
||||
let engine = to_ref(engine)?;
|
||||
let engine = to_shared_ref(engine as *const RegorusEngine)?;
|
||||
let mut guard = engine.try_write()?;
|
||||
guard.clear_coverage_data();
|
||||
Ok(())
|
||||
@@ -571,7 +576,7 @@ pub extern "C" fn regorus_engine_set_gather_prints(
|
||||
) -> RegorusResult {
|
||||
with_unwind_guard(|| {
|
||||
to_regorus_result(|| -> Result<()> {
|
||||
let engine = to_ref(engine)?;
|
||||
let engine = to_shared_ref(engine as *const RegorusEngine)?;
|
||||
let mut guard = engine.try_write()?;
|
||||
guard.set_gather_prints(enable);
|
||||
Ok(())
|
||||
@@ -586,7 +591,7 @@ pub extern "C" fn regorus_engine_set_gather_prints(
|
||||
pub extern "C" fn regorus_engine_take_prints(engine: *mut RegorusEngine) -> RegorusResult {
|
||||
with_unwind_guard(|| {
|
||||
let output = || -> Result<String> {
|
||||
let engine = to_ref(engine)?;
|
||||
let engine = to_shared_ref(engine as *const RegorusEngine)?;
|
||||
let mut guard = engine.try_write()?;
|
||||
Ok(serde_json::to_string_pretty(&guard.take_prints()?)?)
|
||||
}();
|
||||
@@ -605,7 +610,7 @@ pub extern "C" fn regorus_engine_take_prints(engine: *mut RegorusEngine) -> Rego
|
||||
pub extern "C" fn regorus_engine_get_ast_as_json(engine: *mut RegorusEngine) -> RegorusResult {
|
||||
with_unwind_guard(|| {
|
||||
let output = || -> Result<String> {
|
||||
let engine = to_ref(engine)?;
|
||||
let engine = to_shared_ref(engine as *const RegorusEngine)?;
|
||||
let guard = engine.try_read()?;
|
||||
guard.get_ast_as_json()
|
||||
}();
|
||||
@@ -626,7 +631,7 @@ pub extern "C" fn regorus_engine_get_policy_package_names(
|
||||
) -> RegorusResult {
|
||||
with_unwind_guard(|| {
|
||||
let output = || -> Result<String> {
|
||||
let engine = to_ref(engine)?;
|
||||
let engine = to_shared_ref(engine as *const RegorusEngine)?;
|
||||
let guard = engine.try_read()?;
|
||||
serde_json::to_string_pretty(&guard.get_policy_package_names()?)
|
||||
.map_err(anyhow::Error::msg)
|
||||
@@ -648,7 +653,7 @@ pub extern "C" fn regorus_engine_get_policy_parameters(
|
||||
) -> RegorusResult {
|
||||
with_unwind_guard(|| {
|
||||
let output = || -> Result<String> {
|
||||
let engine = to_ref(engine)?;
|
||||
let engine = to_shared_ref(engine as *const RegorusEngine)?;
|
||||
let guard = engine.try_read()?;
|
||||
serde_json::to_string_pretty(&guard.get_policy_parameters()?)
|
||||
.map_err(anyhow::Error::msg)
|
||||
@@ -670,7 +675,7 @@ pub extern "C" fn regorus_engine_set_rego_v0(
|
||||
) -> RegorusResult {
|
||||
with_unwind_guard(|| {
|
||||
let output = || -> Result<()> {
|
||||
let engine = to_ref(engine)?;
|
||||
let engine = to_shared_ref(engine as *const RegorusEngine)?;
|
||||
let mut guard = engine.try_write()?;
|
||||
guard.set_rego_v0(enable);
|
||||
Ok(())
|
||||
@@ -692,7 +697,7 @@ pub extern "C" fn regorus_engine_set_rego_v0(
|
||||
#[cfg(feature = "azure_policy")]
|
||||
pub extern "C" fn regorus_engine_compile_for_target(engine: *mut RegorusEngine) -> RegorusResult {
|
||||
with_unwind_guard(|| {
|
||||
let engine = match to_ref(engine) {
|
||||
let engine = match to_shared_ref(engine as *const RegorusEngine) {
|
||||
Ok(engine) => engine,
|
||||
Err(e) => {
|
||||
return RegorusResult::err_with_message(
|
||||
@@ -741,7 +746,7 @@ pub extern "C" fn regorus_engine_compile_with_entrypoint(
|
||||
let result = || -> Result<RegorusCompiledPolicy> {
|
||||
let rule_str = from_c_str(rule)?;
|
||||
let rule_rc: regorus::Rc<str> = rule_str.into();
|
||||
let engine = to_ref(engine)?;
|
||||
let engine = to_shared_ref(engine as *const RegorusEngine)?;
|
||||
let mut guard = engine.try_write()?;
|
||||
let compiled_policy = guard.compile_with_entrypoint(&rule_rc)?;
|
||||
Ok(RegorusCompiledPolicy { compiled_policy })
|
||||
@@ -800,7 +805,7 @@ pub extern "C" fn regorus_engine_compile_program_with_entrypoints(
|
||||
.ok_or_else(|| anyhow!("entry_points must contain at least one entry"))?;
|
||||
let rule_rc: regorus::Rc<str> = (*rule).into();
|
||||
|
||||
let engine = to_ref(engine)?;
|
||||
let engine = to_shared_ref(engine as *const RegorusEngine)?;
|
||||
let mut guard = engine.try_write()?;
|
||||
let compiled_policy = guard.compile_with_entrypoint(&rule_rc)?;
|
||||
|
||||
|
||||
@@ -2,7 +2,8 @@
|
||||
// Licensed under the MIT License.
|
||||
|
||||
use crate::common::{
|
||||
from_c_str, to_ref, to_regorus_result, RegorusBuffer, RegorusResult, RegorusStatus,
|
||||
from_c_str, to_ref, to_regorus_result, to_shared_ref, RegorusBuffer, RegorusResult,
|
||||
RegorusStatus,
|
||||
};
|
||||
use crate::compile::RegorusPolicyModule;
|
||||
use crate::compiled_policy::RegorusCompiledPolicy;
|
||||
@@ -106,7 +107,8 @@ pub extern "C" fn regorus_program_compile_from_policy(
|
||||
|
||||
let entry_points_ref: Vec<&str> = entry_points_vec.iter().map(|s| s.as_str()).collect();
|
||||
|
||||
let compiled_policy = &to_ref(compiled_policy)?.compiled_policy;
|
||||
let compiled_policy =
|
||||
&to_shared_ref(compiled_policy as *const RegorusCompiledPolicy)?.compiled_policy;
|
||||
let program = Compiler::compile_from_policy(compiled_policy, &entry_points_ref)?;
|
||||
Ok(Box::into_raw(Box::new(RegorusProgram { program })))
|
||||
}();
|
||||
@@ -187,7 +189,7 @@ pub extern "C" fn regorus_program_new() -> *mut RegorusProgram {
|
||||
pub extern "C" fn regorus_program_serialize_binary(program: *mut RegorusProgram) -> RegorusResult {
|
||||
with_unwind_guard(|| {
|
||||
let output = || -> Result<*mut RegorusBuffer> {
|
||||
let program = &to_ref(program)?.program;
|
||||
let program = &to_shared_ref(program as *const RegorusProgram)?.program;
|
||||
let bytes = program.serialize_binary().map_err(|e| anyhow!(e))?;
|
||||
Ok(RegorusBuffer::from_vec(bytes))
|
||||
}();
|
||||
@@ -211,7 +213,10 @@ pub extern "C" fn regorus_program_deserialize_binary(
|
||||
) -> RegorusResult {
|
||||
with_unwind_guard(|| {
|
||||
let output = || -> Result<(*mut RegorusProgram, bool)> {
|
||||
if data.is_null() && len > 0 {
|
||||
if data.is_null() {
|
||||
if len > 0 {
|
||||
return Err(anyhow!("null data pointer with non-zero length"));
|
||||
}
|
||||
return Err(anyhow!("null data pointer"));
|
||||
}
|
||||
let data = unsafe { core::slice::from_raw_parts(data, len) };
|
||||
@@ -249,7 +254,7 @@ pub extern "C" fn regorus_program_deserialize_binary(
|
||||
pub extern "C" fn regorus_program_generate_listing(program: *mut RegorusProgram) -> RegorusResult {
|
||||
with_unwind_guard(|| {
|
||||
let output = || -> Result<String> {
|
||||
let program = &to_ref(program)?.program;
|
||||
let program = &to_shared_ref(program as *const RegorusProgram)?.program;
|
||||
Ok(generate_assembly_listing(
|
||||
program,
|
||||
&AssemblyListingConfig::default(),
|
||||
@@ -270,7 +275,7 @@ pub extern "C" fn regorus_program_generate_tabular_listing(
|
||||
) -> RegorusResult {
|
||||
with_unwind_guard(|| {
|
||||
let output = || -> Result<String> {
|
||||
let program = &to_ref(program)?.program;
|
||||
let program = &to_shared_ref(program as *const RegorusProgram)?.program;
|
||||
Ok(generate_tabular_assembly_listing(
|
||||
program,
|
||||
&AssemblyListingConfig::default(),
|
||||
@@ -297,7 +302,9 @@ pub extern "C" fn regorus_rvm_new_with_policy(
|
||||
) -> RegorusResult {
|
||||
with_unwind_guard(|| {
|
||||
let output = || -> Result<*mut RegorusRvm> {
|
||||
let policy = to_ref(compiled_policy)?.compiled_policy.clone();
|
||||
let policy = to_shared_ref(compiled_policy as *const RegorusCompiledPolicy)?
|
||||
.compiled_policy
|
||||
.clone();
|
||||
Ok(Box::into_raw(Box::new(RegorusRvm::new(
|
||||
RegoVM::new_with_policy(policy),
|
||||
))))
|
||||
@@ -318,9 +325,11 @@ pub extern "C" fn regorus_rvm_load_program(
|
||||
) -> RegorusResult {
|
||||
with_unwind_guard(|| {
|
||||
to_regorus_result(|| -> Result<()> {
|
||||
let vm = to_ref(vm)?;
|
||||
let vm = to_shared_ref(vm as *const RegorusRvm)?;
|
||||
let mut guard = vm.try_write()?;
|
||||
let program = to_ref(program)?.program.clone();
|
||||
let program = to_shared_ref(program as *const RegorusProgram)?
|
||||
.program
|
||||
.clone();
|
||||
guard.load_program(program);
|
||||
Ok(())
|
||||
}())
|
||||
@@ -332,7 +341,7 @@ pub extern "C" fn regorus_rvm_load_program(
|
||||
pub extern "C" fn regorus_rvm_set_data(vm: *mut RegorusRvm, data: *const c_char) -> RegorusResult {
|
||||
with_unwind_guard(|| {
|
||||
to_regorus_result(|| -> Result<()> {
|
||||
let vm = to_ref(vm)?;
|
||||
let vm = to_shared_ref(vm as *const RegorusRvm)?;
|
||||
let mut guard = vm.try_write()?;
|
||||
let data_value = Value::from_json_str(&from_c_str(data)?)?;
|
||||
guard.set_data(data_value)?;
|
||||
@@ -349,7 +358,7 @@ pub extern "C" fn regorus_rvm_set_input(
|
||||
) -> RegorusResult {
|
||||
with_unwind_guard(|| {
|
||||
to_regorus_result(|| -> Result<()> {
|
||||
let vm = to_ref(vm)?;
|
||||
let vm = to_shared_ref(vm as *const RegorusRvm)?;
|
||||
let mut guard = vm.try_write()?;
|
||||
let input_value = Value::from_json_str(&from_c_str(input)?)?;
|
||||
guard.set_input(input_value);
|
||||
@@ -358,6 +367,33 @@ pub extern "C" fn regorus_rvm_set_input(
|
||||
})
|
||||
}
|
||||
|
||||
/// Set the VM context document from JSON.
|
||||
///
|
||||
/// The context provides host-supplied ambient data (e.g. `resourceGroup()`,
|
||||
/// `subscription()`) that Azure Policy functions can access via `LoadContext`
|
||||
/// instructions. This must be called before `regorus_rvm_execute` when
|
||||
/// evaluating policies that reference context functions.
|
||||
///
|
||||
/// # Safety
|
||||
/// - `vm` must be a valid pointer to a `RegorusRvm` created by `regorus_rvm_new`.
|
||||
/// - `context_json` must be a valid null-terminated UTF-8 string.
|
||||
#[cfg(feature = "azure_policy")]
|
||||
#[no_mangle]
|
||||
pub extern "C" fn regorus_rvm_set_context(
|
||||
vm: *mut RegorusRvm,
|
||||
context_json: *const c_char,
|
||||
) -> RegorusResult {
|
||||
with_unwind_guard(|| {
|
||||
to_regorus_result(|| -> Result<()> {
|
||||
let vm = to_shared_ref(vm as *const RegorusRvm)?;
|
||||
let mut guard = vm.try_write()?;
|
||||
let context_value = Value::from_json_str(&from_c_str(context_json)?)?;
|
||||
guard.set_context(context_value);
|
||||
Ok(())
|
||||
}())
|
||||
})
|
||||
}
|
||||
|
||||
/// Set the maximum number of instructions that can execute.
|
||||
#[no_mangle]
|
||||
pub extern "C" fn regorus_rvm_set_max_instructions(
|
||||
@@ -366,7 +402,7 @@ pub extern "C" fn regorus_rvm_set_max_instructions(
|
||||
) -> RegorusResult {
|
||||
with_unwind_guard(|| {
|
||||
to_regorus_result(|| -> Result<()> {
|
||||
let vm = to_ref(vm)?;
|
||||
let vm = to_shared_ref(vm as *const RegorusRvm)?;
|
||||
let mut guard = vm.try_write()?;
|
||||
guard.set_max_instructions(max_instructions);
|
||||
Ok(())
|
||||
@@ -382,7 +418,7 @@ pub extern "C" fn regorus_rvm_set_strict_builtin_errors(
|
||||
) -> RegorusResult {
|
||||
with_unwind_guard(|| {
|
||||
to_regorus_result(|| -> Result<()> {
|
||||
let vm = to_ref(vm)?;
|
||||
let vm = to_shared_ref(vm as *const RegorusRvm)?;
|
||||
let mut guard = vm.try_write()?;
|
||||
guard.set_strict_builtin_errors(strict);
|
||||
Ok(())
|
||||
@@ -395,7 +431,7 @@ pub extern "C" fn regorus_rvm_set_strict_builtin_errors(
|
||||
pub extern "C" fn regorus_rvm_set_execution_mode(vm: *mut RegorusRvm, mode: u8) -> RegorusResult {
|
||||
with_unwind_guard(|| {
|
||||
to_regorus_result(|| -> Result<()> {
|
||||
let vm = to_ref(vm)?;
|
||||
let vm = to_shared_ref(vm as *const RegorusRvm)?;
|
||||
let mut guard = vm.try_write()?;
|
||||
let mode = match mode {
|
||||
0 => ExecutionMode::RunToCompletion,
|
||||
@@ -413,7 +449,7 @@ pub extern "C" fn regorus_rvm_set_execution_mode(vm: *mut RegorusRvm, mode: u8)
|
||||
pub extern "C" fn regorus_rvm_set_step_mode(vm: *mut RegorusRvm, enabled: bool) -> RegorusResult {
|
||||
with_unwind_guard(|| {
|
||||
to_regorus_result(|| -> Result<()> {
|
||||
let vm = to_ref(vm)?;
|
||||
let vm = to_shared_ref(vm as *const RegorusRvm)?;
|
||||
let mut guard = vm.try_write()?;
|
||||
guard.set_step_mode(enabled);
|
||||
Ok(())
|
||||
@@ -430,7 +466,7 @@ pub extern "C" fn regorus_rvm_set_execution_timer_config(
|
||||
) -> RegorusResult {
|
||||
with_unwind_guard(|| {
|
||||
to_regorus_result(|| -> Result<()> {
|
||||
let vm = to_ref(vm)?;
|
||||
let vm = to_shared_ref(vm as *const RegorusRvm)?;
|
||||
let mut guard = vm.try_write()?;
|
||||
if has_config {
|
||||
guard.set_execution_timer_config(Some(config.to_execution_timer_config()?));
|
||||
@@ -447,7 +483,7 @@ pub extern "C" fn regorus_rvm_set_execution_timer_config(
|
||||
pub extern "C" fn regorus_rvm_execute(vm: *mut RegorusRvm) -> RegorusResult {
|
||||
with_unwind_guard(|| {
|
||||
let output = || -> Result<String> {
|
||||
let vm = to_ref(vm)?;
|
||||
let vm = to_shared_ref(vm as *const RegorusRvm)?;
|
||||
let mut guard = vm.try_write()?;
|
||||
let result = guard.execute()?;
|
||||
result.to_json_str()
|
||||
@@ -468,7 +504,7 @@ pub extern "C" fn regorus_rvm_execute_entry_point_by_name(
|
||||
) -> RegorusResult {
|
||||
with_unwind_guard(|| {
|
||||
let output = || -> Result<String> {
|
||||
let vm = to_ref(vm)?;
|
||||
let vm = to_shared_ref(vm as *const RegorusRvm)?;
|
||||
let mut guard = vm.try_write()?;
|
||||
let name = from_c_str(entry_point)?;
|
||||
let result = guard.execute_entry_point_by_name(&name)?;
|
||||
@@ -490,7 +526,7 @@ pub extern "C" fn regorus_rvm_execute_entry_point_by_index(
|
||||
) -> RegorusResult {
|
||||
with_unwind_guard(|| {
|
||||
let output = || -> Result<String> {
|
||||
let vm = to_ref(vm)?;
|
||||
let vm = to_shared_ref(vm as *const RegorusRvm)?;
|
||||
let mut guard = vm.try_write()?;
|
||||
let result = guard.execute_entry_point_by_index(index)?;
|
||||
result.to_json_str()
|
||||
@@ -512,7 +548,7 @@ pub extern "C" fn regorus_rvm_resume(
|
||||
) -> RegorusResult {
|
||||
with_unwind_guard(|| {
|
||||
let output = || -> Result<String> {
|
||||
let vm = to_ref(vm)?;
|
||||
let vm = to_shared_ref(vm as *const RegorusRvm)?;
|
||||
let mut guard = vm.try_write()?;
|
||||
let value = if has_value {
|
||||
Some(Value::from_json_str(&from_c_str(resume_value_json)?)?)
|
||||
@@ -535,7 +571,7 @@ pub extern "C" fn regorus_rvm_resume(
|
||||
pub extern "C" fn regorus_rvm_get_execution_state(vm: *mut RegorusRvm) -> RegorusResult {
|
||||
with_unwind_guard(|| {
|
||||
let output = || -> Result<String> {
|
||||
let vm = to_ref(vm)?;
|
||||
let vm = to_shared_ref(vm as *const RegorusRvm)?;
|
||||
let guard = vm.try_read()?;
|
||||
let state: ExecutionState = guard.execution_state().clone();
|
||||
Ok(format!("{:?}", state))
|
||||
|
||||
84
bindings/java/Cargo.lock
generated
84
bindings/java/Cargo.lock
generated
@@ -48,9 +48,9 @@ checksum = "7f202df86484c868dbad7eaa557ef785d5c66295e41b460ef922eca0723b842c"
|
||||
|
||||
[[package]]
|
||||
name = "autocfg"
|
||||
version = "1.5.0"
|
||||
version = "1.5.1"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "c08606f8c3cbf4ce6ec8e28fb0014a2c086708fe954eaa885384a6165172e7e8"
|
||||
checksum = "f2032f911046de80f0a198e0901378627c33f59ea0ac00e363d481118bd70a53"
|
||||
|
||||
[[package]]
|
||||
name = "bit-set"
|
||||
@@ -91,9 +91,9 @@ dependencies = [
|
||||
|
||||
[[package]]
|
||||
name = "bumpalo"
|
||||
version = "3.20.2"
|
||||
version = "3.20.3"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "5d20789868f4b01b2f2caec9f5c4e0213b41e3e5702a50157d699ae31ced2fcb"
|
||||
checksum = "72f5acc6cb2ba439de613abc23857ec3d78374d8ed5ac84e9d11336e87da8649"
|
||||
|
||||
[[package]]
|
||||
name = "bytecount"
|
||||
@@ -109,9 +109,9 @@ checksum = "1e748733b7cbc798e1434b6ac524f0c1ff2ab456fe201501e6497c8417a4fc33"
|
||||
|
||||
[[package]]
|
||||
name = "cc"
|
||||
version = "1.2.61"
|
||||
version = "1.2.62"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "d16d90359e986641506914ba71350897565610e87ce0ad9e6f28569db3dd5c6d"
|
||||
checksum = "a1dce859f0832a7d088c4f1119888ab94ef4b5d6795d1ce05afb7fe159d79f98"
|
||||
dependencies = [
|
||||
"find-msvc-tools",
|
||||
"shlex",
|
||||
@@ -199,9 +199,9 @@ checksum = "a4ae5f15dda3c708c0ade84bfee31ccab44a3da4f88015ed22f63732abe300c8"
|
||||
|
||||
[[package]]
|
||||
name = "displaydoc"
|
||||
version = "0.2.5"
|
||||
version = "0.2.6"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "97369cbbc041bc366949bc74d34658d6cda5621039731c6310521892a3a20ae0"
|
||||
checksum = "1ac70aa55017e108007fbaf5aa0f54b021c98f92ff8af59d42eda9da96e3dd4f"
|
||||
dependencies = [
|
||||
"proc-macro2",
|
||||
"quote",
|
||||
@@ -380,9 +380,9 @@ dependencies = [
|
||||
|
||||
[[package]]
|
||||
name = "hashbrown"
|
||||
version = "0.17.0"
|
||||
version = "0.17.1"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "4f467dd6dccf739c208452f8014c75c18bb8301b050ad1cfb27153803edb0f51"
|
||||
checksum = "ed5909b6e89a2db4456e54cd5f673791d7eca6732202bbf2a9cc504fe2f9b84a"
|
||||
|
||||
[[package]]
|
||||
name = "heck"
|
||||
@@ -530,7 +530,7 @@ source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "d466e9454f08e4a911e14806c24e16fba1b4c121d1ea474396f396069cf949d9"
|
||||
dependencies = [
|
||||
"equivalent",
|
||||
"hashbrown 0.17.0",
|
||||
"hashbrown 0.17.1",
|
||||
"serde",
|
||||
"serde_core",
|
||||
]
|
||||
@@ -598,9 +598,9 @@ dependencies = [
|
||||
|
||||
[[package]]
|
||||
name = "js-sys"
|
||||
version = "0.3.98"
|
||||
version = "0.3.99"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "67df7112613f8bfd9150013a0314e196f4800d3201ae742489d999db2f979f08"
|
||||
checksum = "142bc4740e452c1e57ade0cbc129f139c9093e354346f0872ef985f4f5cf5f11"
|
||||
dependencies = [
|
||||
"cfg-if",
|
||||
"futures-util",
|
||||
@@ -610,9 +610,9 @@ dependencies = [
|
||||
|
||||
[[package]]
|
||||
name = "jsonschema"
|
||||
version = "0.46.4"
|
||||
version = "0.46.5"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "fc59d2432e047d6090ba1d83c782d0128bd6203857978218f5614dbd3287281f"
|
||||
checksum = "6a5fe5206f06e589caf25e79fc05ccdf91fca745685fe9fe1a13bbdfb479a631"
|
||||
dependencies = [
|
||||
"ahash",
|
||||
"bytecount",
|
||||
@@ -670,9 +670,9 @@ dependencies = [
|
||||
|
||||
[[package]]
|
||||
name = "log"
|
||||
version = "0.4.29"
|
||||
version = "0.4.30"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "5e5032e24019045c762d3c0f28f5b6b8bbf38563a65908389bf7978758920897"
|
||||
checksum = "616ec5685824bcc94416c6d4a7a446eea774a31efd7062c8480ba6fd06d7a6e5"
|
||||
|
||||
[[package]]
|
||||
name = "lru"
|
||||
@@ -682,9 +682,9 @@ checksum = "8a860605968fce16869fd239cf4237a82f3ac470723415db603b0e8b6c8d4fb9"
|
||||
|
||||
[[package]]
|
||||
name = "memchr"
|
||||
version = "2.8.0"
|
||||
version = "2.8.1"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "f8ca58f447f06ed17d5fc4043ce1b10dd205e060fb3ce5b979b8ed8e59ff3f79"
|
||||
checksum = "6b947ae49db0d222b1dbc6b113ce7248a3fc3a6ca21b696717bfc000ba4484d8"
|
||||
|
||||
[[package]]
|
||||
name = "micromap"
|
||||
@@ -954,9 +954,9 @@ dependencies = [
|
||||
|
||||
[[package]]
|
||||
name = "referencing"
|
||||
version = "0.46.4"
|
||||
version = "0.46.5"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "cb674900ca31acd75c4aaf63f48e43e719631c0539ea5a9e64163d1296bcb730"
|
||||
checksum = "69e4e17ef386c5383591d07623d3de49cbc601156e7582973e6db98d66a57de2"
|
||||
dependencies = [
|
||||
"ahash",
|
||||
"fluent-uri",
|
||||
@@ -1000,7 +1000,7 @@ checksum = "dc897dd8d9e8bd1ed8cdad82b5966c3e0ecae09fb1907d58efaa013543185d0a"
|
||||
|
||||
[[package]]
|
||||
name = "regorus"
|
||||
version = "0.10.0"
|
||||
version = "0.10.1"
|
||||
dependencies = [
|
||||
"anyhow",
|
||||
"chrono",
|
||||
@@ -1032,7 +1032,7 @@ dependencies = [
|
||||
|
||||
[[package]]
|
||||
name = "regorus-java"
|
||||
version = "0.10.0"
|
||||
version = "0.10.1"
|
||||
dependencies = [
|
||||
"anyhow",
|
||||
"jni",
|
||||
@@ -1042,7 +1042,7 @@ dependencies = [
|
||||
|
||||
[[package]]
|
||||
name = "regorus-mimalloc"
|
||||
version = "2.2.6"
|
||||
version = "2.2.7"
|
||||
dependencies = [
|
||||
"regorus-mimalloc-sys",
|
||||
]
|
||||
@@ -1128,9 +1128,9 @@ dependencies = [
|
||||
|
||||
[[package]]
|
||||
name = "serde_json"
|
||||
version = "1.0.149"
|
||||
version = "1.0.150"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "83fc039473c5595ace860d8c4fafa220ff474b3fc6bfdb4293327f1a37e94d86"
|
||||
checksum = "e8014e44b4736ed0538adeecded0fce2a272f22dc9578a7eb6b2d9993c74cfb9"
|
||||
dependencies = [
|
||||
"itoa",
|
||||
"memchr",
|
||||
@@ -1194,9 +1194,9 @@ checksum = "67b1b7a3b5fe4f1376887184045fcf45c69e92af734b7aaddc05fb777b6fbd03"
|
||||
|
||||
[[package]]
|
||||
name = "spin"
|
||||
version = "0.10.0"
|
||||
version = "0.12.0"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "d5fe4ccb98d9c292d56fec89a5e07da7fc4cf0dc11e156b41793132775d3e591"
|
||||
checksum = "1527984ca054dfca79333baec451042863f485fbee01b7bf6d911de915cac865"
|
||||
|
||||
[[package]]
|
||||
name = "stable_deref_trait"
|
||||
@@ -1360,9 +1360,9 @@ dependencies = [
|
||||
|
||||
[[package]]
|
||||
name = "wasm-bindgen"
|
||||
version = "0.2.121"
|
||||
version = "0.2.122"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "49ace1d07c165b0864824eee619580c4689389afa9dc9ed3a4c75040d82e6790"
|
||||
checksum = "3ed04576f974d2b2fba0f38c51dbc5518011e38c36bf1143164be765528fd409"
|
||||
dependencies = [
|
||||
"cfg-if",
|
||||
"once_cell",
|
||||
@@ -1373,9 +1373,9 @@ dependencies = [
|
||||
|
||||
[[package]]
|
||||
name = "wasm-bindgen-macro"
|
||||
version = "0.2.121"
|
||||
version = "0.2.122"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "8e68e6f4afd367a562002c05637acb8578ff2dea1943df76afb9e83d177c8578"
|
||||
checksum = "916151b09da36bd82f6615cbf3a419e2f0ba23a03c6160e8e92eb6bd4aa1dec6"
|
||||
dependencies = [
|
||||
"quote",
|
||||
"wasm-bindgen-macro-support",
|
||||
@@ -1383,9 +1383,9 @@ dependencies = [
|
||||
|
||||
[[package]]
|
||||
name = "wasm-bindgen-macro-support"
|
||||
version = "0.2.121"
|
||||
version = "0.2.122"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "d95a9ec35c64b2a7cb35d3fead40c4238d0940c86d107136999567a4703259f2"
|
||||
checksum = "299047362ccbfce148b67ab7e73349f77748e00c8296f9542adfad2ad82c5c5e"
|
||||
dependencies = [
|
||||
"bumpalo",
|
||||
"proc-macro2",
|
||||
@@ -1396,9 +1396,9 @@ dependencies = [
|
||||
|
||||
[[package]]
|
||||
name = "wasm-bindgen-shared"
|
||||
version = "0.2.121"
|
||||
version = "0.2.122"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "c4e0100b01e9f0d03189a92b96772a1fb998639d981193d7dbab487302513441"
|
||||
checksum = "9a929b2c61f11ba3e9bc35b50c1f25cb38e0e892c0c231ae2b8cf78d5dad4437"
|
||||
dependencies = [
|
||||
"unicode-ident",
|
||||
]
|
||||
@@ -1639,18 +1639,18 @@ dependencies = [
|
||||
|
||||
[[package]]
|
||||
name = "zerocopy"
|
||||
version = "0.8.48"
|
||||
version = "0.8.49"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "eed437bf9d6692032087e337407a86f04cd8d6a16a37199ed57949d415bd68e9"
|
||||
checksum = "bce33a6288fa3f072a8c2c7d0f2fdbb90e28298f0135c1f99b96c3db2efcc60b"
|
||||
dependencies = [
|
||||
"zerocopy-derive",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "zerocopy-derive"
|
||||
version = "0.8.48"
|
||||
version = "0.8.49"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "70e3cd084b1788766f53af483dd21f93881ff30d7320490ec3ef7526d203bad4"
|
||||
checksum = "8fd425244944f4ab65ccff928e7323354c5a018c75838362fdce749dfad2ee1e"
|
||||
dependencies = [
|
||||
"proc-macro2",
|
||||
"quote",
|
||||
@@ -1659,9 +1659,9 @@ dependencies = [
|
||||
|
||||
[[package]]
|
||||
name = "zerofrom"
|
||||
version = "0.1.7"
|
||||
version = "0.1.8"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "69faa1f2a1ea75661980b013019ed6687ed0e83d069bc1114e2cc74c6c04c4df"
|
||||
checksum = "0ec05a11813ea801ff6d75110ad09cd0824ddba17dfe17128ea0d5f68e6c5272"
|
||||
dependencies = [
|
||||
"zerofrom-derive",
|
||||
]
|
||||
|
||||
@@ -2,7 +2,7 @@
|
||||
|
||||
[package]
|
||||
name = "regorus-java"
|
||||
version = "0.10.0"
|
||||
version = "0.10.1"
|
||||
edition = "2021"
|
||||
repository = "https://github.com/microsoft/regorus/bindings/java"
|
||||
description = "Java bindings for Regorus - a fast, lightweight Rego interpreter written in Rust"
|
||||
@@ -21,6 +21,6 @@ cache = ["regorus/cache"]
|
||||
|
||||
[dependencies]
|
||||
anyhow = "1.0"
|
||||
serde_json = "1.0.112"
|
||||
serde_json = "1.0.150"
|
||||
jni = "0.22.4"
|
||||
regorus = { path = "../..", default-features = false, features = ["arc", "rvm"] }
|
||||
|
||||
@@ -9,7 +9,7 @@
|
||||
|
||||
<groupId>com.microsoft.regorus</groupId>
|
||||
<artifactId>regorus-java</artifactId>
|
||||
<version>0.10.0</version>
|
||||
<version>0.10.1</version>
|
||||
|
||||
<name>Regorus Java</name>
|
||||
<description>Java bindings for Regorus - a fast, lightweight Rego interpreter written in Rust</description>
|
||||
|
||||
@@ -462,7 +462,7 @@ pub extern "system" fn Java_com_microsoft_regorus_Program_nativeCompileFromModul
|
||||
}
|
||||
|
||||
let mut modules = Vec::with_capacity(ids.len());
|
||||
for (id, content) in ids.into_iter().zip(contents.into_iter()) {
|
||||
for (id, content) in ids.into_iter().zip(contents) {
|
||||
modules.push(PolicyModule {
|
||||
id: Rc::from(id.as_str()),
|
||||
content: Rc::from(content.as_str()),
|
||||
|
||||
84
bindings/python/Cargo.lock
generated
84
bindings/python/Cargo.lock
generated
@@ -48,9 +48,9 @@ checksum = "7f202df86484c868dbad7eaa557ef785d5c66295e41b460ef922eca0723b842c"
|
||||
|
||||
[[package]]
|
||||
name = "autocfg"
|
||||
version = "1.5.0"
|
||||
version = "1.5.1"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "c08606f8c3cbf4ce6ec8e28fb0014a2c086708fe954eaa885384a6165172e7e8"
|
||||
checksum = "f2032f911046de80f0a198e0901378627c33f59ea0ac00e363d481118bd70a53"
|
||||
|
||||
[[package]]
|
||||
name = "bit-set"
|
||||
@@ -91,9 +91,9 @@ dependencies = [
|
||||
|
||||
[[package]]
|
||||
name = "bumpalo"
|
||||
version = "3.20.2"
|
||||
version = "3.20.3"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "5d20789868f4b01b2f2caec9f5c4e0213b41e3e5702a50157d699ae31ced2fcb"
|
||||
checksum = "72f5acc6cb2ba439de613abc23857ec3d78374d8ed5ac84e9d11336e87da8649"
|
||||
|
||||
[[package]]
|
||||
name = "bytecount"
|
||||
@@ -103,9 +103,9 @@ checksum = "175812e0be2bccb6abe50bb8d566126198344f707e304f45c648fd8f2cc0365e"
|
||||
|
||||
[[package]]
|
||||
name = "cc"
|
||||
version = "1.2.61"
|
||||
version = "1.2.62"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "d16d90359e986641506914ba71350897565610e87ce0ad9e6f28569db3dd5c6d"
|
||||
checksum = "a1dce859f0832a7d088c4f1119888ab94ef4b5d6795d1ce05afb7fe159d79f98"
|
||||
dependencies = [
|
||||
"find-msvc-tools",
|
||||
"shlex",
|
||||
@@ -183,9 +183,9 @@ checksum = "a4ae5f15dda3c708c0ade84bfee31ccab44a3da4f88015ed22f63732abe300c8"
|
||||
|
||||
[[package]]
|
||||
name = "displaydoc"
|
||||
version = "0.2.5"
|
||||
version = "0.2.6"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "97369cbbc041bc366949bc74d34658d6cda5621039731c6310521892a3a20ae0"
|
||||
checksum = "1ac70aa55017e108007fbaf5aa0f54b021c98f92ff8af59d42eda9da96e3dd4f"
|
||||
dependencies = [
|
||||
"proc-macro2",
|
||||
"quote",
|
||||
@@ -364,9 +364,9 @@ dependencies = [
|
||||
|
||||
[[package]]
|
||||
name = "hashbrown"
|
||||
version = "0.17.0"
|
||||
version = "0.17.1"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "4f467dd6dccf739c208452f8014c75c18bb8301b050ad1cfb27153803edb0f51"
|
||||
checksum = "ed5909b6e89a2db4456e54cd5f673791d7eca6732202bbf2a9cc504fe2f9b84a"
|
||||
|
||||
[[package]]
|
||||
name = "heck"
|
||||
@@ -514,7 +514,7 @@ source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "d466e9454f08e4a911e14806c24e16fba1b4c121d1ea474396f396069cf949d9"
|
||||
dependencies = [
|
||||
"equivalent",
|
||||
"hashbrown 0.17.0",
|
||||
"hashbrown 0.17.1",
|
||||
"serde",
|
||||
"serde_core",
|
||||
]
|
||||
@@ -533,9 +533,9 @@ checksum = "8f42a60cbdf9a97f5d2305f08a87dc4e09308d1276d28c869c684d7777685682"
|
||||
|
||||
[[package]]
|
||||
name = "js-sys"
|
||||
version = "0.3.98"
|
||||
version = "0.3.99"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "67df7112613f8bfd9150013a0314e196f4800d3201ae742489d999db2f979f08"
|
||||
checksum = "142bc4740e452c1e57ade0cbc129f139c9093e354346f0872ef985f4f5cf5f11"
|
||||
dependencies = [
|
||||
"cfg-if",
|
||||
"futures-util",
|
||||
@@ -545,9 +545,9 @@ dependencies = [
|
||||
|
||||
[[package]]
|
||||
name = "jsonschema"
|
||||
version = "0.46.4"
|
||||
version = "0.46.5"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "fc59d2432e047d6090ba1d83c782d0128bd6203857978218f5614dbd3287281f"
|
||||
checksum = "6a5fe5206f06e589caf25e79fc05ccdf91fca745685fe9fe1a13bbdfb479a631"
|
||||
dependencies = [
|
||||
"ahash",
|
||||
"bytecount",
|
||||
@@ -605,9 +605,9 @@ dependencies = [
|
||||
|
||||
[[package]]
|
||||
name = "log"
|
||||
version = "0.4.29"
|
||||
version = "0.4.30"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "5e5032e24019045c762d3c0f28f5b6b8bbf38563a65908389bf7978758920897"
|
||||
checksum = "616ec5685824bcc94416c6d4a7a446eea774a31efd7062c8480ba6fd06d7a6e5"
|
||||
|
||||
[[package]]
|
||||
name = "lru"
|
||||
@@ -617,9 +617,9 @@ checksum = "8a860605968fce16869fd239cf4237a82f3ac470723415db603b0e8b6c8d4fb9"
|
||||
|
||||
[[package]]
|
||||
name = "memchr"
|
||||
version = "2.8.0"
|
||||
version = "2.8.1"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "f8ca58f447f06ed17d5fc4043ce1b10dd205e060fb3ce5b979b8ed8e59ff3f79"
|
||||
checksum = "6b947ae49db0d222b1dbc6b113ce7248a3fc3a6ca21b696717bfc000ba4484d8"
|
||||
|
||||
[[package]]
|
||||
name = "micromap"
|
||||
@@ -963,9 +963,9 @@ dependencies = [
|
||||
|
||||
[[package]]
|
||||
name = "referencing"
|
||||
version = "0.46.4"
|
||||
version = "0.46.5"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "cb674900ca31acd75c4aaf63f48e43e719631c0539ea5a9e64163d1296bcb730"
|
||||
checksum = "69e4e17ef386c5383591d07623d3de49cbc601156e7582973e6db98d66a57de2"
|
||||
dependencies = [
|
||||
"ahash",
|
||||
"fluent-uri",
|
||||
@@ -1009,7 +1009,7 @@ checksum = "dc897dd8d9e8bd1ed8cdad82b5966c3e0ecae09fb1907d58efaa013543185d0a"
|
||||
|
||||
[[package]]
|
||||
name = "regorus"
|
||||
version = "0.10.0"
|
||||
version = "0.10.1"
|
||||
dependencies = [
|
||||
"anyhow",
|
||||
"chrono",
|
||||
@@ -1041,7 +1041,7 @@ dependencies = [
|
||||
|
||||
[[package]]
|
||||
name = "regorus-mimalloc"
|
||||
version = "2.2.6"
|
||||
version = "2.2.7"
|
||||
dependencies = [
|
||||
"regorus-mimalloc-sys",
|
||||
]
|
||||
@@ -1055,7 +1055,7 @@ dependencies = [
|
||||
|
||||
[[package]]
|
||||
name = "regoruspy"
|
||||
version = "0.10.0"
|
||||
version = "0.10.1"
|
||||
dependencies = [
|
||||
"anyhow",
|
||||
"ordered-float",
|
||||
@@ -1120,9 +1120,9 @@ dependencies = [
|
||||
|
||||
[[package]]
|
||||
name = "serde_json"
|
||||
version = "1.0.149"
|
||||
version = "1.0.150"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "83fc039473c5595ace860d8c4fafa220ff474b3fc6bfdb4293327f1a37e94d86"
|
||||
checksum = "e8014e44b4736ed0538adeecded0fce2a272f22dc9578a7eb6b2d9993c74cfb9"
|
||||
dependencies = [
|
||||
"itoa",
|
||||
"memchr",
|
||||
@@ -1170,9 +1170,9 @@ checksum = "67b1b7a3b5fe4f1376887184045fcf45c69e92af734b7aaddc05fb777b6fbd03"
|
||||
|
||||
[[package]]
|
||||
name = "spin"
|
||||
version = "0.10.0"
|
||||
version = "0.12.0"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "d5fe4ccb98d9c292d56fec89a5e07da7fc4cf0dc11e156b41793132775d3e591"
|
||||
checksum = "1527984ca054dfca79333baec451042863f485fbee01b7bf6d911de915cac865"
|
||||
|
||||
[[package]]
|
||||
name = "stable_deref_trait"
|
||||
@@ -1332,9 +1332,9 @@ dependencies = [
|
||||
|
||||
[[package]]
|
||||
name = "wasm-bindgen"
|
||||
version = "0.2.121"
|
||||
version = "0.2.122"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "49ace1d07c165b0864824eee619580c4689389afa9dc9ed3a4c75040d82e6790"
|
||||
checksum = "3ed04576f974d2b2fba0f38c51dbc5518011e38c36bf1143164be765528fd409"
|
||||
dependencies = [
|
||||
"cfg-if",
|
||||
"once_cell",
|
||||
@@ -1345,9 +1345,9 @@ dependencies = [
|
||||
|
||||
[[package]]
|
||||
name = "wasm-bindgen-macro"
|
||||
version = "0.2.121"
|
||||
version = "0.2.122"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "8e68e6f4afd367a562002c05637acb8578ff2dea1943df76afb9e83d177c8578"
|
||||
checksum = "916151b09da36bd82f6615cbf3a419e2f0ba23a03c6160e8e92eb6bd4aa1dec6"
|
||||
dependencies = [
|
||||
"quote",
|
||||
"wasm-bindgen-macro-support",
|
||||
@@ -1355,9 +1355,9 @@ dependencies = [
|
||||
|
||||
[[package]]
|
||||
name = "wasm-bindgen-macro-support"
|
||||
version = "0.2.121"
|
||||
version = "0.2.122"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "d95a9ec35c64b2a7cb35d3fead40c4238d0940c86d107136999567a4703259f2"
|
||||
checksum = "299047362ccbfce148b67ab7e73349f77748e00c8296f9542adfad2ad82c5c5e"
|
||||
dependencies = [
|
||||
"bumpalo",
|
||||
"proc-macro2",
|
||||
@@ -1368,9 +1368,9 @@ dependencies = [
|
||||
|
||||
[[package]]
|
||||
name = "wasm-bindgen-shared"
|
||||
version = "0.2.121"
|
||||
version = "0.2.122"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "c4e0100b01e9f0d03189a92b96772a1fb998639d981193d7dbab487302513441"
|
||||
checksum = "9a929b2c61f11ba3e9bc35b50c1f25cb38e0e892c0c231ae2b8cf78d5dad4437"
|
||||
dependencies = [
|
||||
"unicode-ident",
|
||||
]
|
||||
@@ -1593,18 +1593,18 @@ dependencies = [
|
||||
|
||||
[[package]]
|
||||
name = "zerocopy"
|
||||
version = "0.8.48"
|
||||
version = "0.8.49"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "eed437bf9d6692032087e337407a86f04cd8d6a16a37199ed57949d415bd68e9"
|
||||
checksum = "bce33a6288fa3f072a8c2c7d0f2fdbb90e28298f0135c1f99b96c3db2efcc60b"
|
||||
dependencies = [
|
||||
"zerocopy-derive",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "zerocopy-derive"
|
||||
version = "0.8.48"
|
||||
version = "0.8.49"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "70e3cd084b1788766f53af483dd21f93881ff30d7320490ec3ef7526d203bad4"
|
||||
checksum = "8fd425244944f4ab65ccff928e7323354c5a018c75838362fdce749dfad2ee1e"
|
||||
dependencies = [
|
||||
"proc-macro2",
|
||||
"quote",
|
||||
@@ -1613,9 +1613,9 @@ dependencies = [
|
||||
|
||||
[[package]]
|
||||
name = "zerofrom"
|
||||
version = "0.1.7"
|
||||
version = "0.1.8"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "69faa1f2a1ea75661980b013019ed6687ed0e83d069bc1114e2cc74c6c04c4df"
|
||||
checksum = "0ec05a11813ea801ff6d75110ad09cd0824ddba17dfe17128ea0d5f68e6c5272"
|
||||
dependencies = [
|
||||
"zerofrom-derive",
|
||||
]
|
||||
|
||||
@@ -2,7 +2,7 @@
|
||||
|
||||
[package]
|
||||
name = "regoruspy"
|
||||
version = "0.10.0"
|
||||
version = "0.10.1"
|
||||
edition = "2021"
|
||||
repository = "https://github.com/microsoft/regorus/bindings/python"
|
||||
description = "Python bindings for Regorus - a fast, lightweight Rego interpreter written in Rust"
|
||||
@@ -25,5 +25,5 @@ anyhow = "1.0"
|
||||
ordered-float = "5.3.0"
|
||||
pyo3 = { version = "0.28.3", features = ["abi3-py310", "anyhow", "extension-module"] }
|
||||
regorus = { path = "../..", default-features = false, features = ["arc", "rvm"] }
|
||||
serde_json = "1.0.140"
|
||||
serde_json = "1.0.150"
|
||||
|
||||
|
||||
88
bindings/ruby/Cargo.lock
generated
88
bindings/ruby/Cargo.lock
generated
@@ -48,9 +48,9 @@ checksum = "7f202df86484c868dbad7eaa557ef785d5c66295e41b460ef922eca0723b842c"
|
||||
|
||||
[[package]]
|
||||
name = "autocfg"
|
||||
version = "1.5.0"
|
||||
version = "1.5.1"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "c08606f8c3cbf4ce6ec8e28fb0014a2c086708fe954eaa885384a6165172e7e8"
|
||||
checksum = "f2032f911046de80f0a198e0901378627c33f59ea0ac00e363d481118bd70a53"
|
||||
|
||||
[[package]]
|
||||
name = "bindgen"
|
||||
@@ -109,9 +109,9 @@ dependencies = [
|
||||
|
||||
[[package]]
|
||||
name = "bumpalo"
|
||||
version = "3.20.2"
|
||||
version = "3.20.3"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "5d20789868f4b01b2f2caec9f5c4e0213b41e3e5702a50157d699ae31ced2fcb"
|
||||
checksum = "72f5acc6cb2ba439de613abc23857ec3d78374d8ed5ac84e9d11336e87da8649"
|
||||
|
||||
[[package]]
|
||||
name = "bytecount"
|
||||
@@ -121,9 +121,9 @@ checksum = "175812e0be2bccb6abe50bb8d566126198344f707e304f45c648fd8f2cc0365e"
|
||||
|
||||
[[package]]
|
||||
name = "cc"
|
||||
version = "1.2.61"
|
||||
version = "1.2.62"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "d16d90359e986641506914ba71350897565610e87ce0ad9e6f28569db3dd5c6d"
|
||||
checksum = "a1dce859f0832a7d088c4f1119888ab94ef4b5d6795d1ce05afb7fe159d79f98"
|
||||
dependencies = [
|
||||
"find-msvc-tools",
|
||||
"shlex",
|
||||
@@ -212,9 +212,9 @@ checksum = "a4ae5f15dda3c708c0ade84bfee31ccab44a3da4f88015ed22f63732abe300c8"
|
||||
|
||||
[[package]]
|
||||
name = "displaydoc"
|
||||
version = "0.2.5"
|
||||
version = "0.2.6"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "97369cbbc041bc366949bc74d34658d6cda5621039731c6310521892a3a20ae0"
|
||||
checksum = "1ac70aa55017e108007fbaf5aa0f54b021c98f92ff8af59d42eda9da96e3dd4f"
|
||||
dependencies = [
|
||||
"proc-macro2",
|
||||
"quote",
|
||||
@@ -223,9 +223,9 @@ dependencies = [
|
||||
|
||||
[[package]]
|
||||
name = "either"
|
||||
version = "1.15.0"
|
||||
version = "1.16.0"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "48c757948c5ede0e46177b7add2e67155f70e33c07fea8284df6576da70b3719"
|
||||
checksum = "91622ff5e7162018101f2fea40d6ebf4a78bbe5a49736a2020649edf9693679e"
|
||||
|
||||
[[package]]
|
||||
name = "email_address"
|
||||
@@ -393,9 +393,9 @@ dependencies = [
|
||||
|
||||
[[package]]
|
||||
name = "hashbrown"
|
||||
version = "0.17.0"
|
||||
version = "0.17.1"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "4f467dd6dccf739c208452f8014c75c18bb8301b050ad1cfb27153803edb0f51"
|
||||
checksum = "ed5909b6e89a2db4456e54cd5f673791d7eca6732202bbf2a9cc504fe2f9b84a"
|
||||
|
||||
[[package]]
|
||||
name = "heck"
|
||||
@@ -543,7 +543,7 @@ source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "d466e9454f08e4a911e14806c24e16fba1b4c121d1ea474396f396069cf949d9"
|
||||
dependencies = [
|
||||
"equivalent",
|
||||
"hashbrown 0.17.0",
|
||||
"hashbrown 0.17.1",
|
||||
"serde",
|
||||
"serde_core",
|
||||
]
|
||||
@@ -571,9 +571,9 @@ checksum = "8f42a60cbdf9a97f5d2305f08a87dc4e09308d1276d28c869c684d7777685682"
|
||||
|
||||
[[package]]
|
||||
name = "js-sys"
|
||||
version = "0.3.98"
|
||||
version = "0.3.99"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "67df7112613f8bfd9150013a0314e196f4800d3201ae742489d999db2f979f08"
|
||||
checksum = "142bc4740e452c1e57ade0cbc129f139c9093e354346f0872ef985f4f5cf5f11"
|
||||
dependencies = [
|
||||
"cfg-if",
|
||||
"futures-util",
|
||||
@@ -583,9 +583,9 @@ dependencies = [
|
||||
|
||||
[[package]]
|
||||
name = "jsonschema"
|
||||
version = "0.46.4"
|
||||
version = "0.46.5"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "fc59d2432e047d6090ba1d83c782d0128bd6203857978218f5614dbd3287281f"
|
||||
checksum = "6a5fe5206f06e589caf25e79fc05ccdf91fca745685fe9fe1a13bbdfb479a631"
|
||||
dependencies = [
|
||||
"ahash",
|
||||
"bytecount",
|
||||
@@ -653,9 +653,9 @@ dependencies = [
|
||||
|
||||
[[package]]
|
||||
name = "log"
|
||||
version = "0.4.29"
|
||||
version = "0.4.30"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "5e5032e24019045c762d3c0f28f5b6b8bbf38563a65908389bf7978758920897"
|
||||
checksum = "616ec5685824bcc94416c6d4a7a446eea774a31efd7062c8480ba6fd06d7a6e5"
|
||||
|
||||
[[package]]
|
||||
name = "lru"
|
||||
@@ -688,9 +688,9 @@ dependencies = [
|
||||
|
||||
[[package]]
|
||||
name = "memchr"
|
||||
version = "2.8.0"
|
||||
version = "2.8.1"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "f8ca58f447f06ed17d5fc4043ce1b10dd205e060fb3ce5b979b8ed8e59ff3f79"
|
||||
checksum = "6b947ae49db0d222b1dbc6b113ce7248a3fc3a6ca21b696717bfc000ba4484d8"
|
||||
|
||||
[[package]]
|
||||
name = "micromap"
|
||||
@@ -994,9 +994,9 @@ dependencies = [
|
||||
|
||||
[[package]]
|
||||
name = "referencing"
|
||||
version = "0.46.4"
|
||||
version = "0.46.5"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "cb674900ca31acd75c4aaf63f48e43e719631c0539ea5a9e64163d1296bcb730"
|
||||
checksum = "69e4e17ef386c5383591d07623d3de49cbc601156e7582973e6db98d66a57de2"
|
||||
dependencies = [
|
||||
"ahash",
|
||||
"fluent-uri",
|
||||
@@ -1040,7 +1040,7 @@ checksum = "dc897dd8d9e8bd1ed8cdad82b5966c3e0ecae09fb1907d58efaa013543185d0a"
|
||||
|
||||
[[package]]
|
||||
name = "regorus"
|
||||
version = "0.10.0"
|
||||
version = "0.10.1"
|
||||
dependencies = [
|
||||
"anyhow",
|
||||
"chrono",
|
||||
@@ -1071,7 +1071,7 @@ dependencies = [
|
||||
|
||||
[[package]]
|
||||
name = "regorus-mimalloc"
|
||||
version = "2.2.6"
|
||||
version = "2.2.7"
|
||||
dependencies = [
|
||||
"regorus-mimalloc-sys",
|
||||
]
|
||||
@@ -1085,7 +1085,7 @@ dependencies = [
|
||||
|
||||
[[package]]
|
||||
name = "regorusrb"
|
||||
version = "0.10.0"
|
||||
version = "0.10.1"
|
||||
dependencies = [
|
||||
"magnus",
|
||||
"regorus",
|
||||
@@ -1162,9 +1162,9 @@ dependencies = [
|
||||
|
||||
[[package]]
|
||||
name = "serde_json"
|
||||
version = "1.0.149"
|
||||
version = "1.0.150"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "83fc039473c5595ace860d8c4fafa220ff474b3fc6bfdb4293327f1a37e94d86"
|
||||
checksum = "e8014e44b4736ed0538adeecded0fce2a272f22dc9578a7eb6b2d9993c74cfb9"
|
||||
dependencies = [
|
||||
"itoa",
|
||||
"memchr",
|
||||
@@ -1229,9 +1229,9 @@ checksum = "67b1b7a3b5fe4f1376887184045fcf45c69e92af734b7aaddc05fb777b6fbd03"
|
||||
|
||||
[[package]]
|
||||
name = "spin"
|
||||
version = "0.10.0"
|
||||
version = "0.12.0"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "d5fe4ccb98d9c292d56fec89a5e07da7fc4cf0dc11e156b41793132775d3e591"
|
||||
checksum = "1527984ca054dfca79333baec451042863f485fbee01b7bf6d911de915cac865"
|
||||
|
||||
[[package]]
|
||||
name = "stable_deref_trait"
|
||||
@@ -1391,9 +1391,9 @@ dependencies = [
|
||||
|
||||
[[package]]
|
||||
name = "wasm-bindgen"
|
||||
version = "0.2.121"
|
||||
version = "0.2.122"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "49ace1d07c165b0864824eee619580c4689389afa9dc9ed3a4c75040d82e6790"
|
||||
checksum = "3ed04576f974d2b2fba0f38c51dbc5518011e38c36bf1143164be765528fd409"
|
||||
dependencies = [
|
||||
"cfg-if",
|
||||
"once_cell",
|
||||
@@ -1404,9 +1404,9 @@ dependencies = [
|
||||
|
||||
[[package]]
|
||||
name = "wasm-bindgen-macro"
|
||||
version = "0.2.121"
|
||||
version = "0.2.122"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "8e68e6f4afd367a562002c05637acb8578ff2dea1943df76afb9e83d177c8578"
|
||||
checksum = "916151b09da36bd82f6615cbf3a419e2f0ba23a03c6160e8e92eb6bd4aa1dec6"
|
||||
dependencies = [
|
||||
"quote",
|
||||
"wasm-bindgen-macro-support",
|
||||
@@ -1414,9 +1414,9 @@ dependencies = [
|
||||
|
||||
[[package]]
|
||||
name = "wasm-bindgen-macro-support"
|
||||
version = "0.2.121"
|
||||
version = "0.2.122"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "d95a9ec35c64b2a7cb35d3fead40c4238d0940c86d107136999567a4703259f2"
|
||||
checksum = "299047362ccbfce148b67ab7e73349f77748e00c8296f9542adfad2ad82c5c5e"
|
||||
dependencies = [
|
||||
"bumpalo",
|
||||
"proc-macro2",
|
||||
@@ -1427,9 +1427,9 @@ dependencies = [
|
||||
|
||||
[[package]]
|
||||
name = "wasm-bindgen-shared"
|
||||
version = "0.2.121"
|
||||
version = "0.2.122"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "c4e0100b01e9f0d03189a92b96772a1fb998639d981193d7dbab487302513441"
|
||||
checksum = "9a929b2c61f11ba3e9bc35b50c1f25cb38e0e892c0c231ae2b8cf78d5dad4437"
|
||||
dependencies = [
|
||||
"unicode-ident",
|
||||
]
|
||||
@@ -1652,18 +1652,18 @@ dependencies = [
|
||||
|
||||
[[package]]
|
||||
name = "zerocopy"
|
||||
version = "0.8.48"
|
||||
version = "0.8.49"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "eed437bf9d6692032087e337407a86f04cd8d6a16a37199ed57949d415bd68e9"
|
||||
checksum = "bce33a6288fa3f072a8c2c7d0f2fdbb90e28298f0135c1f99b96c3db2efcc60b"
|
||||
dependencies = [
|
||||
"zerocopy-derive",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "zerocopy-derive"
|
||||
version = "0.8.48"
|
||||
version = "0.8.49"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "70e3cd084b1788766f53af483dd21f93881ff30d7320490ec3ef7526d203bad4"
|
||||
checksum = "8fd425244944f4ab65ccff928e7323354c5a018c75838362fdce749dfad2ee1e"
|
||||
dependencies = [
|
||||
"proc-macro2",
|
||||
"quote",
|
||||
@@ -1672,9 +1672,9 @@ dependencies = [
|
||||
|
||||
[[package]]
|
||||
name = "zerofrom"
|
||||
version = "0.1.7"
|
||||
version = "0.1.8"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "69faa1f2a1ea75661980b013019ed6687ed0e83d069bc1114e2cc74c6c04c4df"
|
||||
checksum = "0ec05a11813ea801ff6d75110ad09cd0824ddba17dfe17128ea0d5f68e6c5272"
|
||||
dependencies = [
|
||||
"zerofrom-derive",
|
||||
]
|
||||
|
||||
@@ -1,6 +1,6 @@
|
||||
[package]
|
||||
name = "regorusrb"
|
||||
version = "0.10.0"
|
||||
version = "0.10.1"
|
||||
edition = "2024"
|
||||
description = "Ruby bindings for Regorus - a fast, lightweight Rego interpreter written in Rust"
|
||||
license = "MIT AND Apache-2.0 AND BSD-3-Clause"
|
||||
|
||||
@@ -1,5 +1,5 @@
|
||||
# frozen_string_literal: true
|
||||
|
||||
module Regorus
|
||||
VERSION = "0.10.0"
|
||||
VERSION = "0.10.1"
|
||||
end
|
||||
|
||||
98
bindings/wasm/Cargo.lock
generated
98
bindings/wasm/Cargo.lock
generated
@@ -59,9 +59,9 @@ dependencies = [
|
||||
|
||||
[[package]]
|
||||
name = "autocfg"
|
||||
version = "1.5.0"
|
||||
version = "1.5.1"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "c08606f8c3cbf4ce6ec8e28fb0014a2c086708fe954eaa885384a6165172e7e8"
|
||||
checksum = "f2032f911046de80f0a198e0901378627c33f59ea0ac00e363d481118bd70a53"
|
||||
|
||||
[[package]]
|
||||
name = "bit-set"
|
||||
@@ -102,9 +102,9 @@ dependencies = [
|
||||
|
||||
[[package]]
|
||||
name = "bumpalo"
|
||||
version = "3.20.2"
|
||||
version = "3.20.3"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "5d20789868f4b01b2f2caec9f5c4e0213b41e3e5702a50157d699ae31ced2fcb"
|
||||
checksum = "72f5acc6cb2ba439de613abc23857ec3d78374d8ed5ac84e9d11336e87da8649"
|
||||
|
||||
[[package]]
|
||||
name = "bytecount"
|
||||
@@ -120,9 +120,9 @@ checksum = "37b2a672a2cb129a2e41c10b1224bb368f9f37a2b16b612598138befd7b37eb5"
|
||||
|
||||
[[package]]
|
||||
name = "cc"
|
||||
version = "1.2.61"
|
||||
version = "1.2.62"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "d16d90359e986641506914ba71350897565610e87ce0ad9e6f28569db3dd5c6d"
|
||||
checksum = "a1dce859f0832a7d088c4f1119888ab94ef4b5d6795d1ce05afb7fe159d79f98"
|
||||
dependencies = [
|
||||
"find-msvc-tools",
|
||||
"shlex",
|
||||
@@ -200,9 +200,9 @@ checksum = "a4ae5f15dda3c708c0ade84bfee31ccab44a3da4f88015ed22f63732abe300c8"
|
||||
|
||||
[[package]]
|
||||
name = "displaydoc"
|
||||
version = "0.2.5"
|
||||
version = "0.2.6"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "97369cbbc041bc366949bc74d34658d6cda5621039731c6310521892a3a20ae0"
|
||||
checksum = "1ac70aa55017e108007fbaf5aa0f54b021c98f92ff8af59d42eda9da96e3dd4f"
|
||||
dependencies = [
|
||||
"proc-macro2",
|
||||
"quote",
|
||||
@@ -396,9 +396,9 @@ dependencies = [
|
||||
|
||||
[[package]]
|
||||
name = "hashbrown"
|
||||
version = "0.17.0"
|
||||
version = "0.17.1"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "4f467dd6dccf739c208452f8014c75c18bb8301b050ad1cfb27153803edb0f51"
|
||||
checksum = "ed5909b6e89a2db4456e54cd5f673791d7eca6732202bbf2a9cc504fe2f9b84a"
|
||||
|
||||
[[package]]
|
||||
name = "heck"
|
||||
@@ -546,7 +546,7 @@ source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "d466e9454f08e4a911e14806c24e16fba1b4c121d1ea474396f396069cf949d9"
|
||||
dependencies = [
|
||||
"equivalent",
|
||||
"hashbrown 0.17.0",
|
||||
"hashbrown 0.17.1",
|
||||
"serde",
|
||||
"serde_core",
|
||||
]
|
||||
@@ -565,9 +565,9 @@ checksum = "8f42a60cbdf9a97f5d2305f08a87dc4e09308d1276d28c869c684d7777685682"
|
||||
|
||||
[[package]]
|
||||
name = "js-sys"
|
||||
version = "0.3.98"
|
||||
version = "0.3.99"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "67df7112613f8bfd9150013a0314e196f4800d3201ae742489d999db2f979f08"
|
||||
checksum = "142bc4740e452c1e57ade0cbc129f139c9093e354346f0872ef985f4f5cf5f11"
|
||||
dependencies = [
|
||||
"cfg-if",
|
||||
"futures-util",
|
||||
@@ -577,9 +577,9 @@ dependencies = [
|
||||
|
||||
[[package]]
|
||||
name = "jsonschema"
|
||||
version = "0.46.4"
|
||||
version = "0.46.5"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "fc59d2432e047d6090ba1d83c782d0128bd6203857978218f5614dbd3287281f"
|
||||
checksum = "6a5fe5206f06e589caf25e79fc05ccdf91fca745685fe9fe1a13bbdfb479a631"
|
||||
dependencies = [
|
||||
"ahash",
|
||||
"bytecount",
|
||||
@@ -643,9 +643,9 @@ dependencies = [
|
||||
|
||||
[[package]]
|
||||
name = "log"
|
||||
version = "0.4.29"
|
||||
version = "0.4.30"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "5e5032e24019045c762d3c0f28f5b6b8bbf38563a65908389bf7978758920897"
|
||||
checksum = "616ec5685824bcc94416c6d4a7a446eea774a31efd7062c8480ba6fd06d7a6e5"
|
||||
|
||||
[[package]]
|
||||
name = "lru"
|
||||
@@ -655,9 +655,9 @@ checksum = "8a860605968fce16869fd239cf4237a82f3ac470723415db603b0e8b6c8d4fb9"
|
||||
|
||||
[[package]]
|
||||
name = "memchr"
|
||||
version = "2.8.0"
|
||||
version = "2.8.1"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "f8ca58f447f06ed17d5fc4043ce1b10dd205e060fb3ce5b979b8ed8e59ff3f79"
|
||||
checksum = "6b947ae49db0d222b1dbc6b113ce7248a3fc3a6ca21b696717bfc000ba4484d8"
|
||||
|
||||
[[package]]
|
||||
name = "micromap"
|
||||
@@ -953,9 +953,9 @@ dependencies = [
|
||||
|
||||
[[package]]
|
||||
name = "referencing"
|
||||
version = "0.46.4"
|
||||
version = "0.46.5"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "cb674900ca31acd75c4aaf63f48e43e719631c0539ea5a9e64163d1296bcb730"
|
||||
checksum = "69e4e17ef386c5383591d07623d3de49cbc601156e7582973e6db98d66a57de2"
|
||||
dependencies = [
|
||||
"ahash",
|
||||
"fluent-uri",
|
||||
@@ -999,7 +999,7 @@ checksum = "dc897dd8d9e8bd1ed8cdad82b5966c3e0ecae09fb1907d58efaa013543185d0a"
|
||||
|
||||
[[package]]
|
||||
name = "regorus"
|
||||
version = "0.10.0"
|
||||
version = "0.10.1"
|
||||
dependencies = [
|
||||
"anyhow",
|
||||
"chrono",
|
||||
@@ -1030,7 +1030,7 @@ dependencies = [
|
||||
|
||||
[[package]]
|
||||
name = "regorusjs"
|
||||
version = "0.10.0"
|
||||
version = "0.10.1"
|
||||
dependencies = [
|
||||
"getrandom 0.2.17",
|
||||
"getrandom 0.3.4",
|
||||
@@ -1120,9 +1120,9 @@ dependencies = [
|
||||
|
||||
[[package]]
|
||||
name = "serde_json"
|
||||
version = "1.0.149"
|
||||
version = "1.0.150"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "83fc039473c5595ace860d8c4fafa220ff474b3fc6bfdb4293327f1a37e94d86"
|
||||
checksum = "e8014e44b4736ed0538adeecded0fce2a272f22dc9578a7eb6b2d9993c74cfb9"
|
||||
dependencies = [
|
||||
"itoa",
|
||||
"memchr",
|
||||
@@ -1170,9 +1170,9 @@ checksum = "67b1b7a3b5fe4f1376887184045fcf45c69e92af734b7aaddc05fb777b6fbd03"
|
||||
|
||||
[[package]]
|
||||
name = "spin"
|
||||
version = "0.10.0"
|
||||
version = "0.12.0"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "d5fe4ccb98d9c292d56fec89a5e07da7fc4cf0dc11e156b41793132775d3e591"
|
||||
checksum = "1527984ca054dfca79333baec451042863f485fbee01b7bf6d911de915cac865"
|
||||
|
||||
[[package]]
|
||||
name = "stable_deref_trait"
|
||||
@@ -1344,9 +1344,9 @@ dependencies = [
|
||||
|
||||
[[package]]
|
||||
name = "wasm-bindgen"
|
||||
version = "0.2.121"
|
||||
version = "0.2.122"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "49ace1d07c165b0864824eee619580c4689389afa9dc9ed3a4c75040d82e6790"
|
||||
checksum = "3ed04576f974d2b2fba0f38c51dbc5518011e38c36bf1143164be765528fd409"
|
||||
dependencies = [
|
||||
"cfg-if",
|
||||
"once_cell",
|
||||
@@ -1357,9 +1357,9 @@ dependencies = [
|
||||
|
||||
[[package]]
|
||||
name = "wasm-bindgen-futures"
|
||||
version = "0.4.71"
|
||||
version = "0.4.72"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "96492d0d3ffba25305a7dc88720d250b1401d7edca02cc3bcd50633b424673b8"
|
||||
checksum = "9473dbd2991ae90b6291c3c32c30c6187ac49aa32f9905d1cce280ec1e110b0f"
|
||||
dependencies = [
|
||||
"js-sys",
|
||||
"wasm-bindgen",
|
||||
@@ -1367,9 +1367,9 @@ dependencies = [
|
||||
|
||||
[[package]]
|
||||
name = "wasm-bindgen-macro"
|
||||
version = "0.2.121"
|
||||
version = "0.2.122"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "8e68e6f4afd367a562002c05637acb8578ff2dea1943df76afb9e83d177c8578"
|
||||
checksum = "916151b09da36bd82f6615cbf3a419e2f0ba23a03c6160e8e92eb6bd4aa1dec6"
|
||||
dependencies = [
|
||||
"quote",
|
||||
"wasm-bindgen-macro-support",
|
||||
@@ -1377,9 +1377,9 @@ dependencies = [
|
||||
|
||||
[[package]]
|
||||
name = "wasm-bindgen-macro-support"
|
||||
version = "0.2.121"
|
||||
version = "0.2.122"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "d95a9ec35c64b2a7cb35d3fead40c4238d0940c86d107136999567a4703259f2"
|
||||
checksum = "299047362ccbfce148b67ab7e73349f77748e00c8296f9542adfad2ad82c5c5e"
|
||||
dependencies = [
|
||||
"bumpalo",
|
||||
"proc-macro2",
|
||||
@@ -1390,18 +1390,18 @@ dependencies = [
|
||||
|
||||
[[package]]
|
||||
name = "wasm-bindgen-shared"
|
||||
version = "0.2.121"
|
||||
version = "0.2.122"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "c4e0100b01e9f0d03189a92b96772a1fb998639d981193d7dbab487302513441"
|
||||
checksum = "9a929b2c61f11ba3e9bc35b50c1f25cb38e0e892c0c231ae2b8cf78d5dad4437"
|
||||
dependencies = [
|
||||
"unicode-ident",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "wasm-bindgen-test"
|
||||
version = "0.3.71"
|
||||
version = "0.3.72"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "af5ec93229ad9ccd0a545a516dec76dc276613f278f6a91aa6b463d5b33d42d0"
|
||||
checksum = "74fde991ccdc895cb7fbaa14b137d62af74d9011be67b71c694bfc40edd3119c"
|
||||
dependencies = [
|
||||
"async-trait",
|
||||
"cast",
|
||||
@@ -1421,9 +1421,9 @@ dependencies = [
|
||||
|
||||
[[package]]
|
||||
name = "wasm-bindgen-test-macro"
|
||||
version = "0.3.71"
|
||||
version = "0.3.72"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "3c81b9fef827e575e0e54431736d1baa0d700315d8c62cfef1f61fa3aad0cbeb"
|
||||
checksum = "e925354648d2a4d1bf205412e36d520a800280622eef4719678d268e5d40e978"
|
||||
dependencies = [
|
||||
"proc-macro2",
|
||||
"quote",
|
||||
@@ -1432,9 +1432,9 @@ dependencies = [
|
||||
|
||||
[[package]]
|
||||
name = "wasm-bindgen-test-shared"
|
||||
version = "0.2.121"
|
||||
version = "0.2.122"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "4f4d8ae7ad5440360e9799dfd42857d126454a88441ddf72d288ef83fa47f527"
|
||||
checksum = "684365b586a9a6256c1cc3544eee8680de48d6041142f581776ec7b139622ae9"
|
||||
|
||||
[[package]]
|
||||
name = "wasm-encoder"
|
||||
@@ -1672,18 +1672,18 @@ dependencies = [
|
||||
|
||||
[[package]]
|
||||
name = "zerocopy"
|
||||
version = "0.8.48"
|
||||
version = "0.8.49"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "eed437bf9d6692032087e337407a86f04cd8d6a16a37199ed57949d415bd68e9"
|
||||
checksum = "bce33a6288fa3f072a8c2c7d0f2fdbb90e28298f0135c1f99b96c3db2efcc60b"
|
||||
dependencies = [
|
||||
"zerocopy-derive",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "zerocopy-derive"
|
||||
version = "0.8.48"
|
||||
version = "0.8.49"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "70e3cd084b1788766f53af483dd21f93881ff30d7320490ec3ef7526d203bad4"
|
||||
checksum = "8fd425244944f4ab65ccff928e7323354c5a018c75838362fdce749dfad2ee1e"
|
||||
dependencies = [
|
||||
"proc-macro2",
|
||||
"quote",
|
||||
@@ -1692,9 +1692,9 @@ dependencies = [
|
||||
|
||||
[[package]]
|
||||
name = "zerofrom"
|
||||
version = "0.1.7"
|
||||
version = "0.1.8"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "69faa1f2a1ea75661980b013019ed6687ed0e83d069bc1114e2cc74c6c04c4df"
|
||||
checksum = "0ec05a11813ea801ff6d75110ad09cd0824ddba17dfe17128ea0d5f68e6c5272"
|
||||
dependencies = [
|
||||
"zerofrom-derive",
|
||||
]
|
||||
|
||||
@@ -2,7 +2,7 @@
|
||||
|
||||
[package]
|
||||
name = "regorusjs"
|
||||
version = "0.10.0"
|
||||
version = "0.10.1"
|
||||
edition = "2021"
|
||||
repository = "https://github.com/microsoft/regorus/bindings/wasm"
|
||||
description = "WASM bindings for Regorus - a fast, lightweight Rego interpreter written in Rust"
|
||||
@@ -42,7 +42,7 @@ coverage = ["regorus/coverage"]
|
||||
[dependencies]
|
||||
regorus = { path = "../..", default-features = false, features = ["arc", "rvm"] }
|
||||
serde = { version = "1.0.219", features = ["derive"] }
|
||||
serde_json = "1.0.140"
|
||||
serde_json = "1.0.150"
|
||||
wasm-bindgen = "0.2.100"
|
||||
serde-wasm-bindgen = "0.6"
|
||||
# Specify uuid as a mandatory dependency so as to enable `js` feature which is now required
|
||||
@@ -55,7 +55,7 @@ getrandom03 = { package = "getrandom", version = "0.3.1", features = ["std", "wa
|
||||
getrandom = { version = "0.4.2", features = ["wasm_js"] }
|
||||
|
||||
[dev-dependencies]
|
||||
wasm-bindgen-test = "0.3.71"
|
||||
wasm-bindgen-test = "0.3.72"
|
||||
|
||||
[lints.rust]
|
||||
unexpected_cfgs = { level = "warn", check-cfg = ['cfg(wasm_bindgen_unstable_test_coverage)'] }
|
||||
|
||||
84
docs/value/object.md
Normal file
84
docs/value/object.md
Normal file
@@ -0,0 +1,84 @@
|
||||
# Object
|
||||
|
||||
Opaque container for `Value::Object`'s key→value storage, enabling
|
||||
alternative backends without call-site changes.
|
||||
|
||||
## Design
|
||||
|
||||
`Object` wraps the storage for a key→value collection of `Value`s and
|
||||
provides a curated set of methods (`get`, `insert`, `remove`, `iter`,
|
||||
`iter_sorted`, `cursor`, serde). The backing store is private; callers
|
||||
never see or pattern-match on it, so the representation can change
|
||||
without rippling through call sites.
|
||||
|
||||
Multiple backends can coexist at runtime. Because the backing store is
|
||||
private, different `Object` instances in the same process can use
|
||||
different implementations — e.g., a lazy DB-backed object for `input`,
|
||||
inline small-map objects for SARIF location records, and a regular
|
||||
sorted map elsewhere — all interoperating through the same opaque
|
||||
type. This is stronger than the typical Cargo-feature-selected backend
|
||||
seen in precedent crates.
|
||||
|
||||
Iteration is split intentionally. `iter()` makes no ordering promise,
|
||||
which lets backends that don't keep entries sorted skip any sort work.
|
||||
`iter_sorted()` returns entries in `Value` order and is what
|
||||
serialization and `Ord` rely on for deterministic output. Cursor types
|
||||
add resumable, incremental traversal for the RVM iteration state
|
||||
without leaking iterator internals.
|
||||
|
||||
`Ord` and `PartialOrd` are defined against `iter_sorted()` rather than
|
||||
derived from the storage. Two `Object`s built on different backends —
|
||||
or with different insertion histories — compare equal whenever their
|
||||
sorted entries match, so changing the backend never changes observable
|
||||
comparison results.
|
||||
|
||||
## Precedents
|
||||
|
||||
Other crates that hide storage behind a stable API so the implementation
|
||||
can change without breaking callers:
|
||||
|
||||
- **`serde_json::Map`** — opaque newtype allowing cargo-feature based
|
||||
swap between `BTreeMap` (canonical order) and `IndexMap` (insertion
|
||||
order).
|
||||
- **`toml::Table`** — opaque newtype allowing cargo-feature based swap
|
||||
between `BTreeMap` and `IndexMap`.
|
||||
- **`simdjson` DOM** — opaque tree that lazily materializes nodes on
|
||||
access instead of parsing the whole document up front.
|
||||
|
||||
## Use cases
|
||||
|
||||
- **SARIF small-object pressure** — SARIF reports contain millions of
|
||||
small objects (location records, rule references, message arguments),
|
||||
most with 2-5 keys. A small-map-optimized backend (inline storage
|
||||
for ≤N entries, heap above) eliminates per-object BTreeMap allocation
|
||||
for the common case.
|
||||
|
||||
- **Kubernetes admission policies** — large, deeply-nested resource
|
||||
objects (Pod specs, CRDs) where policies typically touch a handful
|
||||
of paths. A lazy-materializing backend (`LazyObjectProvider` over
|
||||
the incoming JSON) parses only the accessed subtrees.
|
||||
|
||||
- **Azure Policy aliases** — ARM exposes the same logical property
|
||||
under multiple aliases (e.g. paths like
|
||||
`Microsoft.Compute/virtualMachines/storageProfile.osDisk.managedDisk.id`).
|
||||
An alias-aware backend resolves lookups across canonical and alias
|
||||
forms without rewriting every policy.
|
||||
|
||||
- **Azure Policy case-insensitive compare** — ARM property names are
|
||||
case-preserving but case-insensitive on lookup (`tags.Environment`
|
||||
and `tags.environment` resolve identically). A case-insensitive
|
||||
backend centralizes this once at the storage layer instead of at
|
||||
every comparison site.
|
||||
|
||||
- **External data sources** — `input` or `data` backed by a database
|
||||
query, CBOR slice, REST endpoint, or other streaming source via a
|
||||
`LazyObjectProvider`. Entries materialize on demand; the policy
|
||||
only pays for what it touches.
|
||||
|
||||
- **Eval-time temporaries** — objects constructed during evaluation
|
||||
(comprehensions, intermediate rule results) on a bumpalo arena.
|
||||
The whole arena drops at query end with zero per-entry free cost.
|
||||
|
||||
- **Host-language interop** — Python dicts or JS objects accessed via
|
||||
FFI callbacks from the embedding application, without copying into
|
||||
Rust on every binding boundary.
|
||||
@@ -23,8 +23,7 @@ use regorus::languages::azure_policy::aliases::AliasRegistry;
|
||||
use regorus::languages::azure_policy::compiler;
|
||||
use regorus::languages::azure_policy::parser;
|
||||
use regorus::rvm::RegoVM;
|
||||
use regorus::Source;
|
||||
use regorus::Value;
|
||||
use regorus::{Rc, Source, Value};
|
||||
|
||||
/// Evaluate an Azure Policy definition against a resource.
|
||||
///
|
||||
@@ -60,11 +59,8 @@ pub fn azure_policy_eval(
|
||||
println!("Parsed policy definition from {policy_definition}");
|
||||
|
||||
// 3. Compile to RVM bytecode.
|
||||
let program = compiler::compile_policy_definition_with_aliases(
|
||||
&defn,
|
||||
registry.alias_map(),
|
||||
registry.alias_modifiable_map(),
|
||||
)?;
|
||||
let registry = Rc::new(registry);
|
||||
let program = compiler::compile_policy_definition_with_aliases(&defn, Rc::clone(®istry))?;
|
||||
println!("Compiled policy to RVM bytecode");
|
||||
|
||||
// 4. Build normalized input.
|
||||
@@ -138,7 +134,7 @@ pub fn azure_policy_aliases(aliases: String, resource_type: Option<String>) -> R
|
||||
if let Some(ref rt) = resource_type {
|
||||
let rt_lower = rt.to_lowercase();
|
||||
let mut found = false;
|
||||
for (alias_name, _) in registry.alias_map() {
|
||||
for alias_name in registry.alias_map().keys() {
|
||||
if alias_name.to_lowercase().starts_with(&rt_lower) {
|
||||
println!(" {alias_name}");
|
||||
found = true;
|
||||
@@ -148,7 +144,7 @@ pub fn azure_policy_aliases(aliases: String, resource_type: Option<String>) -> R
|
||||
bail!("no aliases found for resource type '{rt}'");
|
||||
}
|
||||
} else {
|
||||
for (alias_name, _) in registry.alias_map() {
|
||||
for alias_name in registry.alias_map().keys() {
|
||||
println!(" {alias_name}");
|
||||
}
|
||||
}
|
||||
|
||||
@@ -2,7 +2,7 @@
|
||||
name = "regorus-mimalloc"
|
||||
description = "Vendored mimalloc allocator for regorus"
|
||||
edition = "2021"
|
||||
version = "2.2.6"
|
||||
version = "2.2.7"
|
||||
license = "MIT"
|
||||
repository = "https://github.com/microsoft/regorus"
|
||||
|
||||
|
||||
@@ -319,7 +319,7 @@ pub fn resolve_path(root: &Value, path: &str) -> Value {
|
||||
match ¤t {
|
||||
Value::Object(map) => {
|
||||
let mut next = None;
|
||||
for (key, value) in map.iter() {
|
||||
for (key, value) in map.iter_sorted() {
|
||||
if let Value::String(ref key_str) = *key {
|
||||
if strings::keys::eq(key_str, &segment) {
|
||||
next = Some(value.clone());
|
||||
|
||||
@@ -8,10 +8,10 @@
|
||||
use crate::ast::{Expr, Ref};
|
||||
use crate::builtins;
|
||||
use crate::lexer::Span;
|
||||
use crate::value::Object;
|
||||
use crate::value::Value;
|
||||
use crate::Rc;
|
||||
|
||||
use alloc::collections::BTreeMap;
|
||||
use alloc::vec::Vec;
|
||||
use anyhow::Result;
|
||||
|
||||
@@ -72,7 +72,7 @@ fn fn_intersection(
|
||||
// Intersection of objects: keep key-value pairs from the first
|
||||
// object only when the key exists in every other object AND
|
||||
// the value is equal across all of them.
|
||||
let mut result: BTreeMap<Value, Value> = first.as_ref().clone();
|
||||
let mut result: Object = first.as_ref().clone();
|
||||
for arg in rest {
|
||||
let Value::Object(ref other) = *arg else {
|
||||
return Ok(Value::Undefined);
|
||||
@@ -114,7 +114,7 @@ fn fn_union(_span: &Span, _params: &[Ref<Expr>], args: &[Value], _strict: bool)
|
||||
Value::Object(_) => {
|
||||
// Union of objects: recursive merge. Nested objects are merged
|
||||
// recursively; all other types (including arrays) use last-writer-wins.
|
||||
let mut result = BTreeMap::<Value, Value>::new();
|
||||
let mut result = Object::new();
|
||||
for arg in args {
|
||||
let Value::Object(ref obj) = *arg else {
|
||||
return Ok(Value::Undefined);
|
||||
@@ -264,7 +264,7 @@ fn fn_create_object(
|
||||
);
|
||||
}
|
||||
|
||||
let mut map = BTreeMap::<Value, Value>::new();
|
||||
let mut map = Object::new();
|
||||
|
||||
for pair in args.chunks(2) {
|
||||
#[allow(clippy::pattern_type_mismatch)]
|
||||
@@ -280,9 +280,9 @@ fn fn_create_object(
|
||||
|
||||
/// Recursively merge two objects. Nested objects are merged; everything
|
||||
/// else (including arrays) uses the value from `incoming`.
|
||||
fn merge_objects(base: &BTreeMap<Value, Value>, overlay: &BTreeMap<Value, Value>) -> Value {
|
||||
fn merge_objects(base: &Object, overlay: &Object) -> Value {
|
||||
let mut result = base.clone();
|
||||
for (k, v) in overlay {
|
||||
for (k, v) in overlay.iter() {
|
||||
#[allow(clippy::needless_borrowed_reference)]
|
||||
let merged = match (result.get(k), v) {
|
||||
(Some(&Value::Object(ref prev)), &Value::Object(ref next)) => merge_objects(prev, next),
|
||||
|
||||
@@ -8,10 +8,10 @@
|
||||
use crate::ast::{Expr, Ref};
|
||||
use crate::builtins;
|
||||
use crate::lexer::Span;
|
||||
use crate::value::Object;
|
||||
use crate::value::Value;
|
||||
use crate::Rc;
|
||||
|
||||
use alloc::collections::BTreeMap;
|
||||
use alloc::string::{String, ToString as _};
|
||||
use alloc::vec::Vec;
|
||||
use anyhow::Result;
|
||||
@@ -84,8 +84,8 @@ fn fn_items(_span: &Span, _params: &[Ref<Expr>], args: &[Value], _strict: bool)
|
||||
return Ok(Value::Undefined);
|
||||
};
|
||||
let mut result = Vec::with_capacity(obj.len());
|
||||
for (k, v) in obj.as_ref() {
|
||||
let mut entry = BTreeMap::<Value, Value>::new();
|
||||
for (k, v) in obj.iter_sorted() {
|
||||
let mut entry = Object::new();
|
||||
entry.insert(Value::from("key"), k.clone());
|
||||
entry.insert(Value::from("value"), v.clone());
|
||||
result.push(Value::Object(Rc::new(entry)));
|
||||
|
||||
@@ -308,7 +308,7 @@ fn urlquery_encode_object(
|
||||
|
||||
{
|
||||
let mut pairs = url.query_pairs_mut();
|
||||
for (key, value) in obj.iter() {
|
||||
for (key, value) in obj.iter_sorted() {
|
||||
let key = ensure_string(name, ¶ms[0], key)?;
|
||||
match value {
|
||||
Value::String(v) => {
|
||||
|
||||
@@ -7,10 +7,11 @@ use crate::ast::{Expr, Ref};
|
||||
use crate::builtins;
|
||||
use crate::builtins::utils::{enforce_limit, ensure_args_count, ensure_object};
|
||||
use crate::lexer::Span;
|
||||
use crate::value::Object;
|
||||
use crate::value::Value;
|
||||
use crate::*;
|
||||
|
||||
use alloc::collections::{BTreeMap, BTreeSet};
|
||||
use alloc::collections::BTreeSet;
|
||||
|
||||
use anyhow::{bail, Result};
|
||||
|
||||
@@ -80,7 +81,7 @@ fn reachable(span: &Span, params: &[Ref<Expr>], args: &[Value], strict: bool) ->
|
||||
}
|
||||
|
||||
fn visit(
|
||||
graph: &BTreeMap<Value, Value>,
|
||||
graph: &Object,
|
||||
visited: &mut BTreeSet<Value>,
|
||||
node: &Value,
|
||||
path: &mut Vec<Value>,
|
||||
@@ -211,7 +212,7 @@ fn walk_visit(path: &mut Vec<Value>, value: &Value, paths: &mut Vec<Value>) -> R
|
||||
}
|
||||
}
|
||||
Value::Object(obj) => {
|
||||
for (key, value) in obj.iter() {
|
||||
for (key, value) in obj.iter_sorted() {
|
||||
path.push(key.clone());
|
||||
// Guard path stack growth while traversing object entries.
|
||||
enforce_limit()?;
|
||||
|
||||
@@ -205,7 +205,7 @@ fn merge_filters(
|
||||
let vref = match f {
|
||||
Value::Object(obj) => {
|
||||
let obj = Rc::make_mut(obj);
|
||||
let entry = obj.entry(p.clone()).or_insert_with(Value::new_object);
|
||||
let entry = obj.get_or_insert_with(p.clone(), Value::new_object);
|
||||
// Guard filter map growth when creating nested objects.
|
||||
enforce_limit()?;
|
||||
entry
|
||||
|
||||
@@ -207,7 +207,7 @@ fn to_string(v: &Value, unescape: bool) -> String {
|
||||
}
|
||||
Value::Object(o) => {
|
||||
"{".to_owned()
|
||||
+ &o.iter()
|
||||
+ &o.iter_sorted()
|
||||
.map(|(k, v)| to_string(k, true) + ": " + &to_string(v, true))
|
||||
.collect::<Vec<String>>()
|
||||
.join(", ")
|
||||
@@ -568,7 +568,7 @@ fn replace_n(span: &Span, params: &[Ref<Expr>], args: &[Value], _strict: bool) -
|
||||
let mut s = ensure_string(name, ¶ms[1], &args[1])?;
|
||||
|
||||
let span = params[0].span();
|
||||
for item in obj.as_ref().iter() {
|
||||
for item in obj.as_ref().iter_sorted() {
|
||||
match item {
|
||||
(Value::String(k), Value::String(v)) => {
|
||||
s = s.replace(k.as_ref(), v.as_ref()).into();
|
||||
|
||||
@@ -5,11 +5,12 @@
|
||||
use crate::ast::{Expr, Ref};
|
||||
use crate::lexer::Span;
|
||||
use crate::number::Number;
|
||||
use crate::value::Object;
|
||||
use crate::Rc;
|
||||
use crate::Value;
|
||||
use crate::*;
|
||||
|
||||
use alloc::collections::{BTreeMap, BTreeSet};
|
||||
use alloc::collections::BTreeSet;
|
||||
|
||||
use anyhow::{bail, Result};
|
||||
|
||||
@@ -168,7 +169,7 @@ pub fn ensure_set(fcn: &str, arg: &Expr, v: Value) -> Result<Rc<BTreeSet<Value>>
|
||||
})
|
||||
}
|
||||
|
||||
pub fn ensure_object(fcn: &str, arg: &Expr, v: Value) -> Result<Rc<BTreeMap<Value, Value>>> {
|
||||
pub fn ensure_object(fcn: &str, arg: &Expr, v: Value) -> Result<Rc<Object>> {
|
||||
Ok(match v {
|
||||
Value::Object(o) => o,
|
||||
_ => {
|
||||
|
||||
@@ -314,7 +314,7 @@ fn order_element_pairs<T: VariableBindingContext>(
|
||||
|
||||
if ready {
|
||||
let (value_expr, plan, _deps, binds) = remaining.remove(idx);
|
||||
scheduled.extend(binds.into_iter());
|
||||
scheduled.extend(binds);
|
||||
ordered.push((value_expr, plan));
|
||||
progress = true;
|
||||
break;
|
||||
|
||||
@@ -28,7 +28,6 @@ use crate::{Expression, Extension, Location, QueryResult, QueryResults};
|
||||
use crate::query::traversal::traverse;
|
||||
|
||||
use crate::Rc;
|
||||
use alloc::collections::btree_map::Entry as BTreeMapEntry;
|
||||
use alloc::collections::{BTreeMap, BTreeSet};
|
||||
use anyhow::{anyhow, bail, Result};
|
||||
use core::ops::Bound::*;
|
||||
@@ -1312,10 +1311,10 @@ impl Interpreter {
|
||||
*obj = Value::new_object();
|
||||
}
|
||||
|
||||
obj = obj
|
||||
.as_object_mut()?
|
||||
.entry(Value::String(p.to_string().into()))
|
||||
.or_insert(Value::new_object());
|
||||
obj = obj.as_object_mut()?.get_or_insert_with(
|
||||
Value::String(p.to_string().into()),
|
||||
Value::new_object,
|
||||
);
|
||||
}
|
||||
*obj = value;
|
||||
// Mark modified rules as processed.
|
||||
@@ -1682,8 +1681,7 @@ impl Interpreter {
|
||||
let set = obj
|
||||
.as_object_mut()
|
||||
.map_err(|_| anyhow!(span.error("previous value is not an object")))?
|
||||
.entry(p)
|
||||
.or_insert(Value::new_set())
|
||||
.get_or_insert_with(p, Value::new_set)
|
||||
.as_set_mut()
|
||||
.map_err(|_| anyhow!(span.error("previous value is not a set")))?;
|
||||
set.append(value.as_set_mut()?);
|
||||
@@ -1691,20 +1689,13 @@ impl Interpreter {
|
||||
let obj = obj
|
||||
.as_object_mut()
|
||||
.map_err(|_| anyhow!(span.error("previous value is not an object")))?;
|
||||
match obj.entry(p) {
|
||||
BTreeMapEntry::Vacant(v) => {
|
||||
if value != Value::Undefined {
|
||||
v.insert(value);
|
||||
} else {
|
||||
// TODO: clean this assumption between Undefined vs Object.
|
||||
v.insert(Value::new_object());
|
||||
}
|
||||
}
|
||||
BTreeMapEntry::Occupied(o) => {
|
||||
if o.get() != &value && value != Value::Undefined {
|
||||
bail!(span
|
||||
.error("complete rules should not produce multiple outputs"))
|
||||
}
|
||||
if value == Value::Undefined {
|
||||
// TODO: clean this assumption between Undefined vs Object.
|
||||
obj.get_or_insert_with(p, Value::new_object);
|
||||
} else {
|
||||
let existing = obj.get_or_insert_with(p, || value.clone());
|
||||
if *existing != value {
|
||||
bail!(span.error("complete rules should not produce multiple outputs"))
|
||||
}
|
||||
}
|
||||
}
|
||||
@@ -1713,8 +1704,7 @@ impl Interpreter {
|
||||
obj = obj
|
||||
.as_object_mut()
|
||||
.map_err(|_| anyhow!(span.error("previous value is not an object")))?
|
||||
.entry(p)
|
||||
.or_insert(Value::new_object());
|
||||
.get_or_insert_with(p, Value::new_object);
|
||||
}
|
||||
}
|
||||
Ok(())
|
||||
@@ -1782,6 +1772,7 @@ impl Interpreter {
|
||||
|
||||
let mut comps = self.eval_rule_ref(&rule_ref)?;
|
||||
if let Some(ke) = &key_expr {
|
||||
is_const_rule = is_const_rule && Self::is_simple_literal(ke)?;
|
||||
comps.push(self.eval_expr(ke)?);
|
||||
}
|
||||
let output = if let Some(oe) = &output_expr {
|
||||
@@ -1821,8 +1812,7 @@ impl Interpreter {
|
||||
let set = ctx_mut
|
||||
.rule_value
|
||||
.as_object_mut()?
|
||||
.entry(Value::from_array(comps))
|
||||
.or_insert(Value::new_set());
|
||||
.get_or_insert_with(Value::from_array(comps), Value::new_set);
|
||||
if output != Value::Undefined {
|
||||
set.as_set_mut()?.insert(output);
|
||||
return Ok(true);
|
||||
@@ -1831,20 +1821,13 @@ impl Interpreter {
|
||||
}
|
||||
|
||||
// Non-set rule.
|
||||
match ctx_mut
|
||||
.rule_value
|
||||
.as_object_mut()?
|
||||
.entry(Value::from_array(comps))
|
||||
{
|
||||
BTreeMapEntry::Vacant(v) => {
|
||||
v.insert(output);
|
||||
}
|
||||
BTreeMapEntry::Occupied(o) if o.get() != &output => bail!(rule_ref
|
||||
let key = Value::from_array(comps);
|
||||
let obj_mut = ctx_mut.rule_value.as_object_mut()?;
|
||||
let existing = obj_mut.get_or_insert_with(key, || output.clone());
|
||||
if *existing != output {
|
||||
bail!(rule_ref
|
||||
.span()
|
||||
.error("rules must not produce multiple outputs")),
|
||||
_ => {
|
||||
// Rule produced same value.
|
||||
}
|
||||
.error("rules must not produce multiple outputs"));
|
||||
}
|
||||
|
||||
return Ok(true);
|
||||
@@ -2470,7 +2453,7 @@ impl Interpreter {
|
||||
}
|
||||
Value::Object(map) => {
|
||||
s.push('{');
|
||||
for (idx, (k, entry_value)) in map.iter().enumerate() {
|
||||
for (idx, (k, entry_value)) in map.iter_sorted().enumerate() {
|
||||
if idx > 0 {
|
||||
s.push_str(", ");
|
||||
}
|
||||
|
||||
@@ -213,10 +213,10 @@ pub fn denormalize_with_aliases(
|
||||
// Phase 4: Attach properties to result.
|
||||
if !properties.is_empty() {
|
||||
if let Some(Value::Object(existing_rc)) = result.get_mut("properties") {
|
||||
// Merge directly into the BTreeMap, avoiding full ObjMap round-trip.
|
||||
// Merge directly into the Object, avoiding full ObjMap round-trip.
|
||||
let existing = Rc::make_mut(existing_rc);
|
||||
for (k, v) in properties {
|
||||
existing.entry(Value::String(k)).or_insert(v);
|
||||
existing.get_or_insert_with(Value::String(k), || v);
|
||||
}
|
||||
} else {
|
||||
obj_insert(&mut result, "properties", make_value(properties));
|
||||
|
||||
@@ -7,6 +7,7 @@ use alloc::collections::{BTreeMap, BTreeSet};
|
||||
use alloc::string::String;
|
||||
use alloc::vec::Vec;
|
||||
|
||||
use crate::value::Object;
|
||||
use crate::Value;
|
||||
|
||||
use super::super::obj_map::{make_value, new_map, obj_insert, val_str, ObjMap};
|
||||
@@ -141,7 +142,7 @@ fn rewrap_nested_array(
|
||||
/// BTreeMap-native recursion for nested sub-resource array re-wrapping,
|
||||
/// avoiding ObjMap round-trips on each array element.
|
||||
fn rewrap_nested_array_in_btree(
|
||||
btree: &mut alloc::collections::BTreeMap<Value, Value>,
|
||||
btree: &mut Object,
|
||||
parent_parts: &[&str],
|
||||
array_name: &str,
|
||||
envelope_fields: &BTreeSet<String>,
|
||||
@@ -187,10 +188,7 @@ fn rewrap_nested_array_in_btree(
|
||||
}
|
||||
|
||||
/// Find a key in a BTreeMap using case-insensitive comparison.
|
||||
fn find_key_ci_btree(
|
||||
btree: &alloc::collections::BTreeMap<Value, Value>,
|
||||
key: &str,
|
||||
) -> Option<Value> {
|
||||
fn find_key_ci_btree(btree: &Object, key: &str) -> Option<Value> {
|
||||
btree
|
||||
.keys()
|
||||
.find(|k| val_str(k).is_some_and(|s| s.eq_ignore_ascii_case(key)))
|
||||
|
||||
@@ -172,11 +172,31 @@ impl AliasRegistry {
|
||||
let prefix = alloc::format!("{}/", fq_type);
|
||||
|
||||
for alias in aliases {
|
||||
// Skip aliases without a default_path — the normalizer's
|
||||
// resolve_resource_type also skips these, so inserting them into
|
||||
// compiler maps would cause a divergence where the compiler
|
||||
// resolves the alias but normalized input never contains the field.
|
||||
if alias.default_path.is_none() {
|
||||
continue;
|
||||
}
|
||||
|
||||
// Derive the short name by stripping the resource type prefix.
|
||||
let raw_short = if alias.name.len() > prefix.len()
|
||||
&& alias.name[..prefix.len()].eq_ignore_ascii_case(&prefix)
|
||||
&& alias
|
||||
.name
|
||||
.get(..prefix.len())
|
||||
.is_some_and(|s| s.eq_ignore_ascii_case(&prefix))
|
||||
{
|
||||
alias.name[prefix.len()..].to_string()
|
||||
// Both slice boundaries are valid: prefix is ASCII
|
||||
// (resource type + '/'), so if `..prefix.len()` succeeded
|
||||
// above, `prefix.len()..` is guaranteed to be on a char
|
||||
// boundary too. The `unwrap_or` is a defensive fallback
|
||||
// that can never trigger for well-formed Azure alias names.
|
||||
alias
|
||||
.name
|
||||
.get(prefix.len()..)
|
||||
.unwrap_or(&alias.name)
|
||||
.to_string()
|
||||
} else if let Some(rest) = alias
|
||||
.name
|
||||
.rfind('/')
|
||||
@@ -260,20 +280,19 @@ impl AliasRegistry {
|
||||
.map(String::as_str)
|
||||
}
|
||||
|
||||
/// Return a clone of the alias-to-short-name map for use by the compiler.
|
||||
/// Return a reference to the alias-to-short-name map.
|
||||
///
|
||||
/// The compiler stores this map internally so it can resolve fully-qualified
|
||||
/// alias names without holding a reference to the registry.
|
||||
pub fn alias_map(&self) -> BTreeMap<String, String> {
|
||||
self.alias_to_short.clone()
|
||||
/// Keys are lowercase fully-qualified alias names; values are short names.
|
||||
pub const fn alias_map(&self) -> &BTreeMap<String, String> {
|
||||
&self.alias_to_short
|
||||
}
|
||||
|
||||
/// Return a clone of the alias-to-modifiable map for use by the compiler.
|
||||
/// Return a reference to the alias-to-modifiable map.
|
||||
///
|
||||
/// Maps lowercase fully-qualified alias names to `true` when the alias
|
||||
/// has `defaultMetadata.attributes = "Modifiable"`.
|
||||
pub fn alias_modifiable_map(&self) -> BTreeMap<String, bool> {
|
||||
self.alias_modifiable.clone()
|
||||
/// Keys are lowercase fully-qualified alias names; values are `true` when
|
||||
/// the alias has `defaultMetadata.attributes = "Modifiable"`.
|
||||
pub const fn alias_modifiable_map(&self) -> &BTreeMap<String, bool> {
|
||||
&self.alias_modifiable
|
||||
}
|
||||
|
||||
/// Normalize a raw ARM resource and wrap it in the input envelope.
|
||||
|
||||
@@ -4,14 +4,13 @@
|
||||
//! Per-alias path resolution: reads values from versioned ARM paths and places
|
||||
//! them at alias short name paths in the normalized output.
|
||||
|
||||
use alloc::string::String;
|
||||
|
||||
use crate::Rc;
|
||||
use crate::Value;
|
||||
|
||||
use super::super::obj_map::remove_element_field;
|
||||
use super::super::obj_map::{
|
||||
collision_safe_key, is_root_field_collision, obj_contains, obj_insert, obj_remove,
|
||||
set_nested_lowercased, ObjMap,
|
||||
collision_safe_key, is_root_field_collision, obj_contains, obj_insert, obj_insert_rc,
|
||||
obj_remove, set_nested_lowercased, ObjMap,
|
||||
};
|
||||
use super::super::types::ResolvedAliases;
|
||||
use super::element_remap::apply_element_remap_precomputed;
|
||||
@@ -48,12 +47,14 @@ pub fn apply_alias_entries(
|
||||
if let Some(value) = value {
|
||||
let value = normalize_value(&value, &entry.short_name, None);
|
||||
|
||||
let target = if is_root_field_collision(&entry.short_name, &entry.default_path) {
|
||||
collision_safe_key(&entry.short_name)
|
||||
if is_root_field_collision(&entry.short_name, &entry.default_path) {
|
||||
let target = collision_safe_key(&entry.short_name);
|
||||
set_nested_lowercased(result, &target, value);
|
||||
} else if entry.short_name.contains('.') {
|
||||
set_nested_lowercased(result, &entry.short_name, value);
|
||||
} else {
|
||||
entry.short_name.clone()
|
||||
};
|
||||
set_nested_lowercased(result, &target, value);
|
||||
obj_insert_rc(result, Rc::clone(&entry.short_name_lc), value);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
@@ -84,13 +85,13 @@ pub fn apply_alias_entries(
|
||||
}
|
||||
|
||||
/// Navigate an ARM path using precomputed segments (avoids per-call split).
|
||||
fn navigate_arm_path_segments(value: &Value, segments: &[String]) -> Option<Value> {
|
||||
fn navigate_arm_path_segments(value: &Value, segments: &[Rc<str>]) -> Option<Value> {
|
||||
let mut current = value;
|
||||
for segment in segments {
|
||||
current = current
|
||||
.as_object()
|
||||
.ok()?
|
||||
.get(&Value::from(segment.as_str()))?;
|
||||
.get(&Value::String(Rc::clone(segment)))?;
|
||||
}
|
||||
Some(current.clone())
|
||||
}
|
||||
|
||||
@@ -6,11 +6,12 @@
|
||||
use alloc::string::String;
|
||||
use alloc::vec::Vec;
|
||||
|
||||
use crate::value::Object;
|
||||
use crate::Value;
|
||||
|
||||
use super::super::obj_map::{
|
||||
obj_get, obj_get_mut, obj_insert, set_nested_in_btree, set_nested_lowercased,
|
||||
set_nested_verbatim, ObjMap,
|
||||
obj_get, obj_get_mut, obj_insert, set_nested, set_nested_lowercased, set_nested_verbatim,
|
||||
ObjMap,
|
||||
};
|
||||
use super::super::types::PrecomputedRemap;
|
||||
|
||||
@@ -118,7 +119,7 @@ fn apply_remap_at_depth(
|
||||
/// BTreeMap-native recursion for element-level remap, avoiding ObjMap
|
||||
/// round-trips on each array element.
|
||||
fn remap_at_depth_in_btree(
|
||||
btree: &mut alloc::collections::BTreeMap<Value, Value>,
|
||||
btree: &mut Object,
|
||||
array_chain: &[Vec<String>],
|
||||
depth: usize,
|
||||
source_field: &str,
|
||||
@@ -177,12 +178,7 @@ fn remap_at_depth_in_btree(
|
||||
}
|
||||
|
||||
/// Remap a value between dotted paths directly in a BTreeMap.
|
||||
fn remap_deep_field_in_btree(
|
||||
btree: &mut alloc::collections::BTreeMap<Value, Value>,
|
||||
source: &str,
|
||||
target: &str,
|
||||
lowercase: bool,
|
||||
) {
|
||||
fn remap_deep_field_in_btree(btree: &mut Object, source: &str, target: &str, lowercase: bool) {
|
||||
let val = match read_dotted_path_btree(btree, source) {
|
||||
Some(v) => v,
|
||||
None => return,
|
||||
@@ -198,14 +194,11 @@ fn remap_deep_field_in_btree(
|
||||
}
|
||||
return;
|
||||
}
|
||||
set_nested_in_btree(btree, &segments, val, lowercase);
|
||||
set_nested(btree, &segments, val, lowercase);
|
||||
}
|
||||
|
||||
/// Read a value at a dotted path from a BTreeMap.
|
||||
fn read_dotted_path_btree(
|
||||
btree: &alloc::collections::BTreeMap<Value, Value>,
|
||||
path: &str,
|
||||
) -> Option<Value> {
|
||||
fn read_dotted_path_btree(btree: &Object, path: &str) -> Option<Value> {
|
||||
let segments: Vec<&str> = path.split('.').collect();
|
||||
let first = segments.first()?;
|
||||
let mut cur: &Value = btree.get(&Value::from(*first))?;
|
||||
|
||||
@@ -13,6 +13,7 @@ mod flatten;
|
||||
// Re-export items used by the denormalizer.
|
||||
pub(crate) use element_remap::{apply_element_remap, ElementRemap};
|
||||
|
||||
use crate::value::Object;
|
||||
use crate::Value;
|
||||
|
||||
use super::obj_map::{
|
||||
@@ -109,7 +110,7 @@ pub fn normalize_with_aliases(
|
||||
/// Merge `properties` fields into the result map, skipping keys that already
|
||||
/// exist.
|
||||
fn merge_properties(
|
||||
obj: &alloc::collections::BTreeMap<Value, Value>,
|
||||
obj: &Object,
|
||||
result: &mut ObjMap,
|
||||
sub_arrays: Option<&alloc::collections::BTreeSet<alloc::string::String>>,
|
||||
) {
|
||||
|
||||
@@ -4,14 +4,15 @@
|
||||
//! Lightweight string-keyed map used during normalization/denormalization.
|
||||
//!
|
||||
//! Internally uses `hashbrown::HashMap<Rc<str>, Value>` for O(1) lookups,
|
||||
//! then converts to `Value::Object` (a `BTreeMap<Value, Value>`) only at
|
||||
//! then converts to `Value::Object` (an `Object`) only at
|
||||
//! the output boundary via [`make_value`].
|
||||
|
||||
use alloc::string::{String, ToString as _};
|
||||
use alloc::string::String;
|
||||
use alloc::vec::Vec;
|
||||
|
||||
use hashbrown::HashMap;
|
||||
|
||||
use crate::value::Object;
|
||||
use crate::Rc;
|
||||
use crate::Value;
|
||||
|
||||
@@ -41,6 +42,33 @@ pub fn obj_insert(map: &mut ObjMap, key: &str, val: Value) {
|
||||
map.insert(Rc::from(key), val);
|
||||
}
|
||||
|
||||
/// Insert a key-value pair using a pre-allocated `Rc<str>` key.
|
||||
///
|
||||
/// Avoids the `Rc::from(key)` heap allocation that [`obj_insert`] performs.
|
||||
pub fn obj_insert_rc(map: &mut ObjMap, key: Rc<str>, val: Value) {
|
||||
map.insert(key, val);
|
||||
}
|
||||
|
||||
/// Lowercase a string, returning an `Rc<str>`.
|
||||
///
|
||||
/// Both paths allocate an `Rc<str>` (header + string bytes). The fast-path
|
||||
/// avoids creating an intermediate lowercased `String` when the input is
|
||||
/// already all-lowercase ASCII.
|
||||
pub fn rc_lowercase(s: &str) -> Rc<str> {
|
||||
if s.bytes().all(|b| !b.is_ascii_uppercase()) {
|
||||
Rc::from(s)
|
||||
} else {
|
||||
Rc::from(s.to_ascii_lowercase())
|
||||
}
|
||||
}
|
||||
|
||||
/// Insert a key-value pair with the key lowercased, using [`rc_lowercase`]
|
||||
/// for the allocation fast-path.
|
||||
pub fn obj_insert_lc(map: &mut ObjMap, key: &str, val: Value) {
|
||||
let lc = rc_lowercase(key);
|
||||
map.insert(lc, val);
|
||||
}
|
||||
|
||||
/// Check whether a key exists.
|
||||
pub fn obj_contains(map: &ObjMap, key: &str) -> bool {
|
||||
map.contains_key(key)
|
||||
@@ -54,14 +82,13 @@ pub fn obj_remove(map: &mut ObjMap, key: &str) -> Option<Value> {
|
||||
/// Convert an [`ObjMap`] into a [`Value::Object`].
|
||||
///
|
||||
/// Keys are converted from `Rc<str>` to `Value::String` and inserted into
|
||||
/// a `BTreeMap` to match the `Value::Object` representation.
|
||||
/// an `Object` to match the `Value::Object` representation.
|
||||
pub fn make_value(map: ObjMap) -> Value {
|
||||
use alloc::collections::BTreeMap;
|
||||
let mut btree = BTreeMap::new();
|
||||
for (k, v) in map {
|
||||
btree.insert(Value::String(k), v);
|
||||
}
|
||||
Value::Object(Rc::new(btree))
|
||||
let obj: Object = map
|
||||
.into_iter()
|
||||
.map(|(k, v)| (Value::String(k), v))
|
||||
.collect();
|
||||
Value::Object(Rc::new(obj))
|
||||
}
|
||||
|
||||
/// Convert a `Vec<Value>` into a `Value::Array`.
|
||||
@@ -88,14 +115,14 @@ pub fn extract_type_field(resource: &Value) -> Option<&str> {
|
||||
})
|
||||
}
|
||||
|
||||
/// Convert a `Value::Object` (BTreeMap<Value, Value>) into an [`ObjMap`].
|
||||
/// Convert a `Value::Object` (Object) into an [`ObjMap`].
|
||||
///
|
||||
/// Non-string keys are silently skipped.
|
||||
#[allow(dead_code)]
|
||||
pub fn value_to_obj_map(value: &Value) -> Option<ObjMap> {
|
||||
let btree = value.as_object().ok()?;
|
||||
let mut map = ObjMap::with_capacity(btree.len());
|
||||
for (k, v) in btree.iter() {
|
||||
let obj = value.as_object().ok()?;
|
||||
let mut map = ObjMap::with_capacity(obj.len());
|
||||
for (k, v) in obj.iter() {
|
||||
if let Value::String(s) = k {
|
||||
map.insert(Rc::clone(s), v.clone());
|
||||
}
|
||||
@@ -112,7 +139,7 @@ pub fn set_nested_lowercased(result: &mut ObjMap, path: &str, value: Value) {
|
||||
}
|
||||
if segments.len() == 1 {
|
||||
if let Some(&seg) = segments.first() {
|
||||
obj_insert(result, &seg.to_ascii_lowercase(), value);
|
||||
obj_insert_lc(result, seg, value);
|
||||
}
|
||||
return;
|
||||
}
|
||||
@@ -144,30 +171,30 @@ fn set_nested_inner(obj: &mut ObjMap, segments: &[&str], value: Value, lowercase
|
||||
};
|
||||
|
||||
if segments.len() == 1 {
|
||||
let key = if lowercase {
|
||||
first.to_ascii_lowercase()
|
||||
let key: Rc<str> = if lowercase {
|
||||
rc_lowercase(first)
|
||||
} else {
|
||||
first.to_string()
|
||||
Rc::from(first)
|
||||
};
|
||||
obj_insert(obj, &key, value);
|
||||
obj_insert_rc(obj, key, value);
|
||||
return;
|
||||
}
|
||||
|
||||
let seg = if lowercase {
|
||||
first.to_ascii_lowercase()
|
||||
let seg: Rc<str> = if lowercase {
|
||||
rc_lowercase(first)
|
||||
} else {
|
||||
first.to_string()
|
||||
Rc::from(first)
|
||||
};
|
||||
|
||||
// Ensure an intermediate object exists at `seg`.
|
||||
if !obj_contains(obj, &seg) {
|
||||
obj_insert(obj, &seg, make_value(new_map()));
|
||||
if !obj.contains_key(&*seg) {
|
||||
obj_insert_rc(obj, Rc::clone(&seg), make_value(new_map()));
|
||||
}
|
||||
|
||||
// Descend directly into the BTreeMap, avoiding ObjMap round-trip.
|
||||
if let Some(Value::Object(inner_rc)) = obj_get_mut(obj, &seg) {
|
||||
if let Some(Value::Object(inner_rc)) = obj.get_mut(&*seg) {
|
||||
let inner_btree = Rc::make_mut(inner_rc);
|
||||
set_nested_in_btree(
|
||||
set_nested(
|
||||
inner_btree,
|
||||
segments.get(1..).unwrap_or_default(),
|
||||
value,
|
||||
@@ -176,41 +203,36 @@ fn set_nested_inner(obj: &mut ObjMap, segments: &[&str], value: Value, lowercase
|
||||
}
|
||||
}
|
||||
|
||||
/// Set a value at a path directly in a `BTreeMap<Value, Value>`, creating
|
||||
/// Set a value at a path directly in an `Object`, creating
|
||||
/// intermediate `Value::Object` nodes as needed.
|
||||
///
|
||||
/// This avoids the `btree_to_obj_map` / `obj_map_to_btree` round-trip that
|
||||
/// would clone every sibling entry at each nesting level.
|
||||
pub fn set_nested_in_btree(
|
||||
btree: &mut alloc::collections::BTreeMap<Value, Value>,
|
||||
segments: &[&str],
|
||||
value: Value,
|
||||
lowercase: bool,
|
||||
) {
|
||||
pub fn set_nested(obj: &mut Object, segments: &[&str], value: Value, lowercase: bool) {
|
||||
let Some(&first) = segments.first() else {
|
||||
return;
|
||||
};
|
||||
|
||||
let key_str: String = if lowercase {
|
||||
first.to_ascii_lowercase()
|
||||
let key_rc: Rc<str> = if lowercase {
|
||||
rc_lowercase(first)
|
||||
} else {
|
||||
first.to_string()
|
||||
Rc::from(first)
|
||||
};
|
||||
let key_val = Value::String(Rc::from(key_str.as_str()));
|
||||
let key_val = Value::String(Rc::clone(&key_rc));
|
||||
|
||||
if segments.len() == 1 {
|
||||
btree.insert(key_val, value);
|
||||
obj.insert(key_val, value);
|
||||
return;
|
||||
}
|
||||
|
||||
// Ensure an intermediate object exists.
|
||||
if !btree.contains_key(&key_val) {
|
||||
btree.insert(key_val.clone(), make_value(new_map()));
|
||||
if !obj.contains_key(&key_val) {
|
||||
obj.insert(key_val.clone(), make_value(new_map()));
|
||||
}
|
||||
|
||||
if let Some(Value::Object(inner_rc)) = btree.get_mut(&key_val) {
|
||||
if let Some(Value::Object(inner_rc)) = obj.get_mut(&key_val) {
|
||||
let inner = Rc::make_mut(inner_rc);
|
||||
set_nested_in_btree(
|
||||
set_nested(
|
||||
inner,
|
||||
segments.get(1..).unwrap_or_default(),
|
||||
value,
|
||||
@@ -243,13 +265,24 @@ pub const ROOT_FIELDS: &[&str] = &[
|
||||
"extendedLocation",
|
||||
];
|
||||
|
||||
const PROPERTIES_DOT: &[u8] = b"properties.";
|
||||
|
||||
/// Check whether an alias short name collides with a reserved ARM root field
|
||||
/// and needs a collision-safe key.
|
||||
pub fn is_root_field_collision(short_name: &str, default_path: &str) -> bool {
|
||||
ROOT_FIELDS
|
||||
.iter()
|
||||
.any(|f| f.eq_ignore_ascii_case(short_name))
|
||||
&& default_path.to_ascii_lowercase().starts_with("properties.")
|
||||
&& default_path.len() > PROPERTIES_DOT.len()
|
||||
&& default_path
|
||||
.as_bytes()
|
||||
.get(..PROPERTIES_DOT.len())
|
||||
.is_some_and(|prefix| {
|
||||
prefix
|
||||
.iter()
|
||||
.zip(PROPERTIES_DOT)
|
||||
.all(|(a, b)| a.to_ascii_lowercase() == *b)
|
||||
})
|
||||
}
|
||||
|
||||
/// Return a collision-safe key for an alias whose short name collides with a
|
||||
@@ -314,20 +347,15 @@ fn remove_field_at_depth(obj: &mut ObjMap, array_chain: &[Vec<String>], depth: u
|
||||
for elem in inner.iter_mut() {
|
||||
if let Value::Object(obj_rc) = elem {
|
||||
let inner_btree = Rc::make_mut(obj_rc);
|
||||
remove_field_at_depth_in_btree(
|
||||
inner_btree,
|
||||
array_chain,
|
||||
depth.saturating_add(1),
|
||||
field,
|
||||
);
|
||||
remove_field_at_depth_obj(inner_btree, array_chain, depth.saturating_add(1), field);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/// BTreeMap-native recursion for element-level field removal.
|
||||
fn remove_field_at_depth_in_btree(
|
||||
btree: &mut alloc::collections::BTreeMap<Value, Value>,
|
||||
/// Object-native recursion for element-level field removal.
|
||||
fn remove_field_at_depth_obj(
|
||||
obj: &mut Object,
|
||||
array_chain: &[Vec<String>],
|
||||
depth: usize,
|
||||
field: &str,
|
||||
@@ -336,10 +364,10 @@ fn remove_field_at_depth_in_btree(
|
||||
let segments: Vec<&str> = field.split('.').collect();
|
||||
if segments.len() == 1 {
|
||||
if let Some(&seg) = segments.first() {
|
||||
btree.remove(&Value::from(seg));
|
||||
obj.remove(&Value::from(seg));
|
||||
}
|
||||
} else if segments.len() > 1 {
|
||||
remove_at_dotted_path_in_btree(btree, &segments);
|
||||
remove_at_dotted_path_obj(obj, &segments);
|
||||
}
|
||||
return;
|
||||
};
|
||||
@@ -351,12 +379,12 @@ fn remove_field_at_depth_in_btree(
|
||||
|
||||
let key_val = Value::from(first);
|
||||
let arr_val = if nav.len() == 1 {
|
||||
match btree.get_mut(&key_val) {
|
||||
match obj.get_mut(&key_val) {
|
||||
Some(v) => v,
|
||||
None => return,
|
||||
}
|
||||
} else {
|
||||
let mut cur: &mut Value = match btree.get_mut(&key_val) {
|
||||
let mut cur: &mut Value = match obj.get_mut(&key_val) {
|
||||
Some(v) => v,
|
||||
None => return,
|
||||
};
|
||||
@@ -377,27 +405,19 @@ fn remove_field_at_depth_in_btree(
|
||||
for elem in inner.iter_mut() {
|
||||
if let Value::Object(obj_rc) = elem {
|
||||
let inner_btree = Rc::make_mut(obj_rc);
|
||||
remove_field_at_depth_in_btree(
|
||||
inner_btree,
|
||||
array_chain,
|
||||
depth.saturating_add(1),
|
||||
field,
|
||||
);
|
||||
remove_field_at_depth_obj(inner_btree, array_chain, depth.saturating_add(1), field);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/// Remove the leaf segment at a dotted path directly in a BTreeMap.
|
||||
fn remove_at_dotted_path_in_btree(
|
||||
btree: &mut alloc::collections::BTreeMap<Value, Value>,
|
||||
segments: &[&str],
|
||||
) {
|
||||
/// Remove the leaf segment at a dotted path directly in an Object.
|
||||
fn remove_at_dotted_path_obj(obj: &mut Object, segments: &[&str]) {
|
||||
let Some((&leaf, parent_segs)) = segments.split_last() else {
|
||||
return;
|
||||
};
|
||||
if parent_segs.is_empty() {
|
||||
btree.remove(&Value::from(leaf));
|
||||
obj.remove(&Value::from(leaf));
|
||||
return;
|
||||
}
|
||||
|
||||
@@ -405,7 +425,7 @@ fn remove_at_dotted_path_in_btree(
|
||||
return;
|
||||
};
|
||||
let first_key = Value::from(first);
|
||||
let parent_val = match btree.get_mut(&first_key) {
|
||||
let parent_val = match obj.get_mut(&first_key) {
|
||||
Some(v) => v,
|
||||
None => return,
|
||||
};
|
||||
|
||||
@@ -18,6 +18,22 @@ use alloc::vec::Vec;
|
||||
|
||||
use serde::{Deserialize, Deserializer};
|
||||
|
||||
use crate::Rc;
|
||||
|
||||
// ---------------------------------------------------------------------------
|
||||
// Deserialization helpers
|
||||
// ---------------------------------------------------------------------------
|
||||
|
||||
/// Deserialize a `Vec<T>` that tolerates JSON `null` by mapping it to an
|
||||
/// empty vector.
|
||||
fn deserialize_null_as_empty_vec<'de, T, D>(deserializer: D) -> Result<Vec<T>, D::Error>
|
||||
where
|
||||
T: Deserialize<'de>,
|
||||
D: Deserializer<'de>,
|
||||
{
|
||||
Ok(Option::<Vec<T>>::deserialize(deserializer)?.unwrap_or_default())
|
||||
}
|
||||
|
||||
// ─── Top-level response wrappers ────────────────────────────────────────────
|
||||
|
||||
/// ARM API response envelope: `{ "value": [...] }`
|
||||
@@ -98,7 +114,10 @@ pub struct AliasEntry {
|
||||
|
||||
/// Versioned path entries. Empty for the vast majority of aliases that
|
||||
/// have only a `defaultPath`.
|
||||
#[serde(default)]
|
||||
///
|
||||
/// In real Azure catalog data (~97% of aliases), `az provider list` emits
|
||||
/// `"paths": null` rather than an empty array.
|
||||
#[serde(default, deserialize_with = "deserialize_null_as_empty_vec")]
|
||||
pub paths: Vec<AliasPath>,
|
||||
}
|
||||
|
||||
@@ -404,11 +423,13 @@ pub struct ResolvedEntry {
|
||||
// ── Precomputed fields (derived at registry-load time) ──────────────
|
||||
/// Whether `short_name` contains `[*]` (i.e., this is a wildcard/array alias).
|
||||
pub is_wildcard: bool,
|
||||
/// Pre-lowercased short name as `Rc<str>` for allocation-free common-case inserts.
|
||||
pub(crate) short_name_lc: Rc<str>,
|
||||
/// Precomputed `default_path.split('.').collect()` for fast ARM path navigation.
|
||||
pub default_path_segments: Vec<String>,
|
||||
pub(crate) default_path_segments: Vec<Rc<str>>,
|
||||
/// Precomputed path segments for each versioned path, in the same order
|
||||
/// as `versioned_paths`.
|
||||
pub versioned_path_segments: Vec<Vec<String>>,
|
||||
pub(crate) versioned_path_segments: Vec<Vec<Rc<str>>>,
|
||||
}
|
||||
|
||||
impl ResolvedEntry {
|
||||
@@ -420,10 +441,15 @@ impl ResolvedEntry {
|
||||
metadata: Option<AliasPathMetadata>,
|
||||
) -> Self {
|
||||
let is_wildcard = short_name.contains("[*]");
|
||||
let default_path_segments = default_path.split('.').map(String::from).collect();
|
||||
let short_name_lc = if short_name.bytes().all(|b| !b.is_ascii_uppercase()) {
|
||||
Rc::from(short_name.as_str())
|
||||
} else {
|
||||
Rc::from(short_name.to_ascii_lowercase())
|
||||
};
|
||||
let default_path_segments = default_path.split('.').map(Rc::from).collect();
|
||||
let versioned_path_segments = versioned_paths
|
||||
.iter()
|
||||
.map(|(_, p)| p.split('.').map(String::from).collect())
|
||||
.map(|(_, p)| p.split('.').map(Rc::from).collect())
|
||||
.collect();
|
||||
Self {
|
||||
short_name,
|
||||
@@ -431,6 +457,7 @@ impl ResolvedEntry {
|
||||
versioned_paths,
|
||||
metadata,
|
||||
is_wildcard,
|
||||
short_name_lc,
|
||||
default_path_segments,
|
||||
versioned_path_segments,
|
||||
}
|
||||
@@ -456,7 +483,7 @@ impl ResolvedEntry {
|
||||
/// Returns the versioned segments if `api_version` matches, otherwise
|
||||
/// the default segments. This avoids per-call `split('.')` for both
|
||||
/// default and versioned scalar alias navigation.
|
||||
pub fn select_path_segments(&self, api_version: Option<&str>) -> &[String] {
|
||||
pub(crate) fn select_path_segments(&self, api_version: Option<&str>) -> &[Rc<str>] {
|
||||
if let Some(ver) = api_version {
|
||||
for (i, (v, _)) in self.versioned_paths.iter().enumerate() {
|
||||
if v.eq_ignore_ascii_case(ver) {
|
||||
|
||||
@@ -18,6 +18,7 @@ use crate::rvm::program::{Program, SpanInfo};
|
||||
use crate::rvm::Instruction;
|
||||
use crate::{Rc, Value};
|
||||
|
||||
use crate::languages::azure_policy::aliases::AliasRegistry;
|
||||
use crate::languages::azure_policy::ast::PolicyRule;
|
||||
|
||||
// ---------------------------------------------------------------------------
|
||||
@@ -44,10 +45,9 @@ pub(super) struct Compiler {
|
||||
pub(super) cached_input_reg: Option<u8>,
|
||||
/// Cached register for `LoadContext` — allocated once on first use.
|
||||
pub(super) cached_context_reg: Option<u8>,
|
||||
/// Map from lowercase fully-qualified alias name → short name.
|
||||
pub(super) alias_map: BTreeMap<String, String>,
|
||||
/// Map from lowercase fully-qualified alias name → modifiable flag.
|
||||
pub(super) alias_modifiable: BTreeMap<String, bool>,
|
||||
/// Alias registry for resolving fully-qualified alias names.
|
||||
/// Shared via `Rc` to avoid cloning the 73K-entry alias maps.
|
||||
pub(super) alias_registry: Option<Rc<AliasRegistry>>,
|
||||
/// Default values for policy parameters.
|
||||
pub(super) parameter_defaults: Option<Value>,
|
||||
/// Cached literal-table index for `parameter_defaults` (or an empty object
|
||||
@@ -338,8 +338,13 @@ impl Compiler {
|
||||
path: &str,
|
||||
span: &crate::lexer::Span,
|
||||
) -> Result<String> {
|
||||
let alias_map = match &self.alias_registry {
|
||||
Some(reg) => reg.alias_map(),
|
||||
None => return Ok(path.to_string()),
|
||||
};
|
||||
|
||||
let lc = path.to_ascii_lowercase();
|
||||
if let Some(short) = self.alias_map.get(&lc) {
|
||||
if let Some(short) = alias_map.get(&lc) {
|
||||
let resolved = short.clone();
|
||||
let result = Self::strip_fq_prefix(&resolved).to_ascii_lowercase();
|
||||
return Ok(result);
|
||||
@@ -348,7 +353,7 @@ impl Compiler {
|
||||
// Fallback: derive array path from a corresponding `[*]` alias.
|
||||
if !lc.contains("[*]") {
|
||||
let wildcard_key = alloc::format!("{}[*]", lc);
|
||||
if let Some(short) = self.alias_map.get(&wildcard_key) {
|
||||
if let Some(short) = alias_map.get(&wildcard_key) {
|
||||
let resolved = Self::strip_fq_prefix(short).to_ascii_lowercase();
|
||||
if let Some(base) = resolved.strip_suffix("[*]") {
|
||||
return Ok(base.to_string());
|
||||
@@ -356,14 +361,14 @@ impl Compiler {
|
||||
}
|
||||
}
|
||||
|
||||
if !self.alias_map.is_empty() && !self.alias_fallback_to_raw {
|
||||
if !alias_map.is_empty() && !self.alias_fallback_to_raw {
|
||||
bail!(span.error(&alloc::format!(
|
||||
"unknown alias '{}': field references must use fully-qualified alias names when an alias catalog is loaded",
|
||||
path
|
||||
)));
|
||||
}
|
||||
|
||||
if self.alias_map.is_empty() {
|
||||
if alias_map.is_empty() {
|
||||
Ok(path.to_string())
|
||||
} else {
|
||||
let result = Self::strip_fq_prefix(path).to_ascii_lowercase();
|
||||
|
||||
@@ -998,7 +998,13 @@ impl Compiler {
|
||||
return Ok(result);
|
||||
}
|
||||
}
|
||||
Err(e) if !self.alias_map.is_empty() && !self.alias_fallback_to_raw => {
|
||||
Err(e)
|
||||
if self
|
||||
.alias_registry
|
||||
.as_ref()
|
||||
.is_some_and(|r| !r.alias_map().is_empty())
|
||||
&& !self.alias_fallback_to_raw =>
|
||||
{
|
||||
return Err(e);
|
||||
}
|
||||
_ => {}
|
||||
|
||||
@@ -11,7 +11,7 @@
|
||||
//! to fetch a related resource and an optional `existenceCondition` evaluated
|
||||
//! inline.
|
||||
|
||||
use alloc::collections::BTreeMap;
|
||||
use crate::value::Object;
|
||||
use alloc::format;
|
||||
use alloc::string::ToString as _;
|
||||
use alloc::vec::Vec;
|
||||
@@ -692,13 +692,18 @@ impl Compiler {
|
||||
field_path: &str,
|
||||
span: &crate::lexer::Span,
|
||||
) -> Result<()> {
|
||||
if self.alias_modifiable.is_empty() {
|
||||
let modifiable_map = match &self.alias_registry {
|
||||
Some(reg) => reg.alias_modifiable_map(),
|
||||
None => return Ok(()),
|
||||
};
|
||||
|
||||
if modifiable_map.is_empty() {
|
||||
return Ok(());
|
||||
}
|
||||
|
||||
let lc = field_path.to_lowercase();
|
||||
|
||||
if let Some(&modifiable) = self.alias_modifiable.get(&lc) {
|
||||
if let Some(&modifiable) = modifiable_map.get(&lc) {
|
||||
if !modifiable {
|
||||
bail!(span.error(&format!(
|
||||
"alias '{}' is not modifiable (defaultMetadata.attributes != 'Modifiable')",
|
||||
@@ -803,26 +808,41 @@ fn unescape_arm_literal(s: &str) -> alloc::string::String {
|
||||
/// 1. Build a template `BTreeMap` with `Value::Undefined` placeholders.
|
||||
/// 2. Sort keys by their literal value (BTreeMap order).
|
||||
/// 3. Emit `ObjectCreate`.
|
||||
#[allow(clippy::indexing_slicing)]
|
||||
pub(super) fn build_object_from_keys(
|
||||
compiler: &mut Compiler,
|
||||
mut keys: Vec<(u16, u8)>,
|
||||
span: &crate::lexer::Span,
|
||||
) -> Result<u8> {
|
||||
// Build template: object with all keys set to Undefined.
|
||||
let mut template = BTreeMap::new();
|
||||
let mut template = Object::new();
|
||||
for &(key_idx, _) in &keys {
|
||||
// SAFETY: key_idx was just returned by `add_literal_u16`, so the
|
||||
// index is guaranteed to be in bounds.
|
||||
let key_val = compiler.program.literals[usize::from(key_idx)].clone();
|
||||
// key_idx was returned by `add_literal_u16` in the calling code,
|
||||
// so it is always in bounds. We use `.get()` + `?` instead of
|
||||
// direct indexing to satisfy the crate-wide `deny(indexing_slicing)`.
|
||||
let key_val = compiler
|
||||
.program
|
||||
.literals
|
||||
.get(usize::from(key_idx))
|
||||
.ok_or_else(|| {
|
||||
anyhow!(
|
||||
"internal error in build_object_from_keys: \
|
||||
literal index {} out of bounds (literals len = {})",
|
||||
key_idx,
|
||||
compiler.program.literals.len()
|
||||
)
|
||||
})?
|
||||
.clone();
|
||||
template.insert(key_val, Value::Undefined);
|
||||
}
|
||||
let template_idx = compiler.add_literal_u16(Value::Object(crate::Rc::new(template)))?;
|
||||
|
||||
// Sort keys by literal value (BTreeMap order).
|
||||
// Sort keys by literal value (BTreeMap order). All indices were
|
||||
// validated in the loop above (which returns Err for out-of-bounds),
|
||||
// so `.get()` always returns `Some` here — `None` is unreachable.
|
||||
keys.sort_by(|a, b| {
|
||||
compiler.program.literals[usize::from(a.0)]
|
||||
.cmp(&compiler.program.literals[usize::from(b.0)])
|
||||
let a_val = compiler.program.literals.get(usize::from(a.0));
|
||||
let b_val = compiler.program.literals.get(usize::from(b.0));
|
||||
a_val.cmp(&b_val)
|
||||
});
|
||||
|
||||
let dest = compiler.alloc_register()?;
|
||||
|
||||
@@ -272,7 +272,7 @@ impl Compiler {
|
||||
fn insert_string_set_annotation(
|
||||
annot: &mut alloc::collections::BTreeMap<String, Value>,
|
||||
key: &str,
|
||||
observed: &BTreeSet<String>,
|
||||
observed: &alloc::collections::BTreeSet<String>,
|
||||
) {
|
||||
if !observed.is_empty() {
|
||||
let set: BTreeSet<Value> = observed
|
||||
|
||||
@@ -30,11 +30,11 @@ mod metadata;
|
||||
mod template_dispatch;
|
||||
mod utils;
|
||||
|
||||
use alloc::collections::BTreeMap;
|
||||
use alloc::string::{String, ToString as _};
|
||||
use alloc::string::ToString as _;
|
||||
|
||||
use anyhow::Result;
|
||||
|
||||
use crate::languages::azure_policy::aliases::AliasRegistry;
|
||||
use crate::languages::azure_policy::ast::{PolicyDefinition, PolicyRule};
|
||||
use crate::rvm::program::Program;
|
||||
use crate::{Rc, Value};
|
||||
@@ -69,17 +69,14 @@ pub fn compile_policy_rule(rule: &PolicyRule) -> Result<Rc<Program>> {
|
||||
|
||||
/// Compile a parsed Azure Policy rule with alias resolution.
|
||||
///
|
||||
/// The `alias_map` maps lowercase fully-qualified alias names to their short
|
||||
/// names. Obtain it from
|
||||
/// [`AliasRegistry::alias_map()`](crate::languages::azure_policy::aliases::AliasRegistry::alias_map).
|
||||
/// The registry provides alias-to-short-name resolution and modifiability
|
||||
/// data. Pass it as an `Rc` to avoid cloning the internal alias maps.
|
||||
pub fn compile_policy_rule_with_aliases(
|
||||
rule: &PolicyRule,
|
||||
alias_map: BTreeMap<String, String>,
|
||||
alias_modifiable: BTreeMap<String, bool>,
|
||||
registry: Rc<AliasRegistry>,
|
||||
) -> Result<Rc<Program>> {
|
||||
let mut compiler = Compiler::new();
|
||||
compiler.alias_map = alias_map;
|
||||
compiler.alias_modifiable = alias_modifiable;
|
||||
compiler.alias_registry = Some(registry);
|
||||
init_effect_annotation(&mut compiler, rule);
|
||||
compiler.compile(rule)
|
||||
}
|
||||
@@ -100,12 +97,10 @@ pub fn compile_policy_definition(defn: &PolicyDefinition) -> Result<Rc<Program>>
|
||||
/// Compile a parsed Azure Policy definition with alias resolution.
|
||||
pub fn compile_policy_definition_with_aliases(
|
||||
defn: &PolicyDefinition,
|
||||
alias_map: BTreeMap<String, String>,
|
||||
alias_modifiable: BTreeMap<String, bool>,
|
||||
registry: Rc<AliasRegistry>,
|
||||
) -> Result<Rc<Program>> {
|
||||
let mut compiler = Compiler::new();
|
||||
compiler.alias_map = alias_map;
|
||||
compiler.alias_modifiable = alias_modifiable;
|
||||
compiler.alias_registry = Some(registry);
|
||||
compiler.parameter_defaults = Some(build_parameter_defaults(&defn.parameters)?);
|
||||
compiler.populate_definition_metadata(defn);
|
||||
init_effect_annotation(&mut compiler, &defn.policy_rule);
|
||||
@@ -119,13 +114,11 @@ pub fn compile_policy_definition_with_aliases(
|
||||
/// a known alias are silently treated as raw property paths.
|
||||
pub fn compile_policy_definition_with_aliases_opts(
|
||||
defn: &PolicyDefinition,
|
||||
alias_map: BTreeMap<String, String>,
|
||||
alias_modifiable: BTreeMap<String, bool>,
|
||||
registry: Rc<AliasRegistry>,
|
||||
alias_fallback_to_raw: bool,
|
||||
) -> Result<Rc<Program>> {
|
||||
let mut compiler = Compiler::new();
|
||||
compiler.alias_map = alias_map;
|
||||
compiler.alias_modifiable = alias_modifiable;
|
||||
compiler.alias_registry = Some(registry);
|
||||
compiler.alias_fallback_to_raw = alias_fallback_to_raw;
|
||||
compiler.parameter_defaults = Some(build_parameter_defaults(&defn.parameters)?);
|
||||
compiler.populate_definition_metadata(defn);
|
||||
|
||||
@@ -264,18 +264,17 @@ impl<'source> Parser<'source> {
|
||||
"metadata" => {
|
||||
*metadata = Some(self.parse_json_value()?);
|
||||
}
|
||||
"parameters" => {
|
||||
"parameters" if self.token_text() == "{" => {
|
||||
// Parameters must be a JSON object; if not, push to extra.
|
||||
if self.token_text() == "{" {
|
||||
*parameters = self.parse_parameter_definitions()?;
|
||||
} else {
|
||||
let value = self.parse_json_value()?;
|
||||
extra.push(ObjectEntry {
|
||||
key_span,
|
||||
key: key.into(),
|
||||
value,
|
||||
});
|
||||
}
|
||||
*parameters = self.parse_parameter_definitions()?;
|
||||
}
|
||||
"parameters" => {
|
||||
let value = self.parse_json_value()?;
|
||||
extra.push(ObjectEntry {
|
||||
key_span,
|
||||
key: key.into(),
|
||||
value,
|
||||
});
|
||||
}
|
||||
"policyrule" => {
|
||||
// Parse the policyRule directly from the token stream!
|
||||
|
||||
@@ -63,6 +63,14 @@ pub enum CompilerError {
|
||||
#[error("Invalid function expression with package")]
|
||||
InvalidFunctionExpressionWithPackage,
|
||||
|
||||
#[error("partial object rules with constant keys are not yet supported by the RVM compiler")]
|
||||
PartialObjectConstantKeyUnsupported,
|
||||
|
||||
#[error(
|
||||
"partial object rules with nested bracket keys are not yet supported by the RVM compiler"
|
||||
)]
|
||||
PartialObjectNestedKeyUnsupported,
|
||||
|
||||
#[error("Compilation error: {message}")]
|
||||
General { message: String },
|
||||
}
|
||||
|
||||
@@ -11,8 +11,9 @@ use crate::ast::{Expr, ExprRef};
|
||||
use crate::lexer::Span;
|
||||
use crate::rvm::instructions::{ArrayCreateParams, ObjectCreateParams, SetCreateParams};
|
||||
use crate::rvm::Instruction;
|
||||
use crate::value::Object;
|
||||
use crate::{Rc, Value};
|
||||
use alloc::collections::{BTreeMap, BTreeSet};
|
||||
use alloc::collections::BTreeSet;
|
||||
use alloc::vec::Vec;
|
||||
|
||||
/// Try to evaluate an expression as a compile-time constant.
|
||||
@@ -43,7 +44,7 @@ pub(in crate::languages::rego::compiler) fn try_eval_const(expr: &Expr) -> Optio
|
||||
Expr::Object { fields, .. } => fields
|
||||
.iter()
|
||||
.map(|(_, k, v)| Some((try_eval_const(k.as_ref())?, try_eval_const(v.as_ref())?)))
|
||||
.collect::<Option<BTreeMap<_, _>>>()
|
||||
.collect::<Option<Object>>()
|
||||
.map(|m| Value::Object(Rc::new(m))),
|
||||
_ => None,
|
||||
}
|
||||
@@ -117,7 +118,7 @@ impl<'a> Compiler<'a> {
|
||||
fields: &[(crate::lexer::Span, ExprRef, ExprRef)],
|
||||
span: &Span,
|
||||
) -> Result<Register> {
|
||||
let all_const: Option<BTreeMap<_, _>> = fields
|
||||
let all_const: Option<Object> = fields
|
||||
.iter()
|
||||
.map(|(_, k, v)| Some((try_eval_const(k.as_ref())?, try_eval_const(v.as_ref())?)))
|
||||
.collect();
|
||||
@@ -166,7 +167,7 @@ impl<'a> Compiler<'a> {
|
||||
let mut template_keys = literal_keys.clone();
|
||||
template_keys.sort();
|
||||
|
||||
let mut template_obj = BTreeMap::new();
|
||||
let mut template_obj = Object::new();
|
||||
for key in &template_keys {
|
||||
template_obj.insert(key.clone(), Value::Undefined);
|
||||
}
|
||||
|
||||
@@ -59,7 +59,7 @@ impl<'a> Compiler<'a> {
|
||||
crate::ast::Expr::RefBrack { .. } if assign.is_some() => {
|
||||
RuleType::PartialObject
|
||||
}
|
||||
crate::ast::Expr::RefBrack { .. } => RuleType::PartialSet,
|
||||
crate::ast::Expr::RefBrack { .. } => RuleType::PartialObject,
|
||||
_ => RuleType::Complete,
|
||||
},
|
||||
_ => RuleType::Complete,
|
||||
@@ -88,6 +88,54 @@ impl<'a> Compiler<'a> {
|
||||
})
|
||||
}
|
||||
|
||||
fn validate_partial_object_shape(&self, refr: &ExprRef) -> Result<()> {
|
||||
let Expr::RefBrack {
|
||||
refr: prefix,
|
||||
index,
|
||||
..
|
||||
} = refr.as_ref()
|
||||
else {
|
||||
return Ok(());
|
||||
};
|
||||
|
||||
if Self::has_unsupported_bracket_prefix(prefix) {
|
||||
return Err(CompilerError::PartialObjectNestedKeyUnsupported.at(refr.span()));
|
||||
}
|
||||
|
||||
if Self::is_simple_literal(index) {
|
||||
return Err(CompilerError::PartialObjectConstantKeyUnsupported.at(index.span()));
|
||||
}
|
||||
|
||||
Ok(())
|
||||
}
|
||||
|
||||
fn has_unsupported_bracket_prefix(expr: &ExprRef) -> bool {
|
||||
match expr.as_ref() {
|
||||
Expr::RefBrack { refr, index, .. } => {
|
||||
!Self::is_string_literal(index) || Self::has_unsupported_bracket_prefix(refr)
|
||||
}
|
||||
Expr::RefDot { refr, .. } => Self::has_unsupported_bracket_prefix(refr),
|
||||
_ => false,
|
||||
}
|
||||
}
|
||||
|
||||
fn is_string_literal(expr: &ExprRef) -> bool {
|
||||
matches!(expr.as_ref(), Expr::String { .. } | Expr::RawString { .. })
|
||||
}
|
||||
|
||||
fn is_simple_literal(expr: &ExprRef) -> bool {
|
||||
match expr.as_ref() {
|
||||
Expr::String { .. }
|
||||
| Expr::RawString { .. }
|
||||
| Expr::Number { .. }
|
||||
| Expr::Bool { .. }
|
||||
| Expr::Null { .. } => true,
|
||||
// Unary expressions like `-1` are constant literals too.
|
||||
Expr::UnaryExpr { expr, .. } => Self::is_simple_literal(expr),
|
||||
_ => false,
|
||||
}
|
||||
}
|
||||
|
||||
pub(super) fn get_or_assign_rule_index(&mut self, rule_path: &str) -> Result<u16> {
|
||||
if let Some(&index) = self.rule_index_map.get(rule_path) {
|
||||
return Ok(index);
|
||||
@@ -345,6 +393,10 @@ impl<'a> Compiler<'a> {
|
||||
|
||||
let (key_expr, value_expr) = match head {
|
||||
RuleHead::Compr { refr, assign, .. } => {
|
||||
if rule_type == RuleType::PartialObject {
|
||||
self.validate_partial_object_shape(refr)?;
|
||||
}
|
||||
|
||||
self.rule_definition_function_params[rule_index as usize].push(None);
|
||||
self.rule_definition_destructuring_patterns[rule_index as usize]
|
||||
.push(None);
|
||||
|
||||
@@ -155,7 +155,7 @@ pub mod target;
|
||||
#[cfg(any(test, all(feature = "yaml", feature = "std")))]
|
||||
pub mod test_utils;
|
||||
pub mod utils;
|
||||
mod value;
|
||||
pub mod value;
|
||||
|
||||
#[cfg(feature = "azure_policy")]
|
||||
pub use {
|
||||
|
||||
@@ -354,6 +354,33 @@ impl<'source> Parser<'source> {
|
||||
}
|
||||
}
|
||||
|
||||
/// Parse a field name after `.` in a ref expression.
|
||||
///
|
||||
/// Unlike [`Self::parse_var`] and [`Self::parse_ident`], this method accepts **any**
|
||||
/// `TokenKind::Ident` token, including reserved keywords (e.g. `as`, `default`, `else`,
|
||||
/// `false`, `if`, `import`, `in`, `not`, `null`, `package`, `some`, `true`, `with`).
|
||||
///
|
||||
/// The position immediately after `.` is unambiguously a field name, so there is no
|
||||
/// syntactic ambiguity with statement-level keywords. This matches OPA's
|
||||
/// `keywords_in_refs` capability, which is enabled by default in standard OPA builds.
|
||||
///
|
||||
/// # Example
|
||||
/// ```rego
|
||||
/// allow if { input.v0.package.format == "npm" } # `package` is a keyword but valid here
|
||||
/// ```
|
||||
fn parse_ref_field(&mut self) -> Result<Span> {
|
||||
let span = self.tok.1.clone();
|
||||
match self.tok.0 {
|
||||
TokenKind::Ident => {
|
||||
self.next_token()?;
|
||||
Ok(span)
|
||||
}
|
||||
_ => Err(self
|
||||
.source
|
||||
.error(self.tok.1.line, self.tok.1.col, "expecting identifier")),
|
||||
}
|
||||
}
|
||||
|
||||
fn read_number(&mut self, span: Span) -> Result<Expr> {
|
||||
match Number::from_str(span.text()) {
|
||||
Ok(v) => Ok(Expr::Number {
|
||||
@@ -743,9 +770,10 @@ impl<'source> Parser<'source> {
|
||||
);
|
||||
}
|
||||
"." => {
|
||||
// Read identifier.
|
||||
// Read identifier. Keywords are allowed as field names in
|
||||
// dot-notation refs (e.g. `input.package.name`).
|
||||
self.next_token()?;
|
||||
let field = self.parse_var()?;
|
||||
let field = self.parse_ref_field()?;
|
||||
span.end = self.end;
|
||||
|
||||
// Disallow any whitespace between . and identifier.
|
||||
@@ -1418,9 +1446,10 @@ impl<'source> Parser<'source> {
|
||||
);
|
||||
}
|
||||
"." => {
|
||||
// Read identifier.
|
||||
// Read identifier. Keywords are allowed as field names in
|
||||
// dot-notation refs (e.g. `import data.my.package`).
|
||||
self.next_token()?;
|
||||
let field = self.parse_ident()?;
|
||||
let field = self.parse_ref_field()?;
|
||||
span.end = self.end;
|
||||
|
||||
// Disallow any whitespace between . and identifier.
|
||||
@@ -1523,7 +1552,8 @@ impl<'source> Parser<'source> {
|
||||
"." => {
|
||||
let sep_pos = self.tok.1.start;
|
||||
self.next_token()?;
|
||||
let field = self.parse_var()?;
|
||||
// Keywords are allowed as field names in dot-notation refs.
|
||||
let field = self.parse_ref_field()?;
|
||||
span.end = self.end;
|
||||
|
||||
// Disallow any whitespace between . and identifier.
|
||||
|
||||
@@ -11,9 +11,9 @@
|
||||
//! values are converted through [`MetadataValue`] — a postcard/bincode-safe
|
||||
//! enum that avoids `deserialize_any`.
|
||||
|
||||
use crate::value::Object;
|
||||
use crate::Rc;
|
||||
use alloc::collections::BTreeMap;
|
||||
use alloc::collections::BTreeSet;
|
||||
use alloc::collections::{BTreeMap, BTreeSet};
|
||||
use alloc::string::String;
|
||||
use alloc::vec::Vec;
|
||||
use serde::{Deserialize, Serialize};
|
||||
@@ -52,7 +52,7 @@ impl ProgramMetadata {
|
||||
pub fn to_value(&self) -> crate::value::Value {
|
||||
use crate::value::Value;
|
||||
|
||||
let mut obj = BTreeMap::new();
|
||||
let mut obj = Object::new();
|
||||
obj.insert(
|
||||
Value::String("compiler_version".into()),
|
||||
Value::String(self.compiler_version.as_str().into()),
|
||||
@@ -75,7 +75,7 @@ impl ProgramMetadata {
|
||||
);
|
||||
|
||||
if !self.annotations.is_empty() {
|
||||
let mut annotations_obj = BTreeMap::new();
|
||||
let mut annotations_obj = Object::new();
|
||||
for (k, v) in &self.annotations {
|
||||
annotations_obj.insert(Value::String(k.as_str().into()), v.clone());
|
||||
}
|
||||
@@ -198,7 +198,7 @@ impl MetadataValue {
|
||||
match *self {
|
||||
MetadataValue::String(ref s) => Value::String(s.as_str().into()),
|
||||
MetadataValue::StringSet(ref set) => {
|
||||
let mut bset = alloc::collections::BTreeSet::new();
|
||||
let mut bset = BTreeSet::new();
|
||||
for s in set {
|
||||
bset.insert(Value::String(s.as_str().into()));
|
||||
}
|
||||
@@ -211,7 +211,7 @@ impl MetadataValue {
|
||||
Value::Array(Rc::new(values))
|
||||
}
|
||||
MetadataValue::Map(ref map) => {
|
||||
let mut obj = BTreeMap::new();
|
||||
let mut obj = Object::new();
|
||||
for (k, v) in map {
|
||||
obj.insert(Value::String(k.as_str().into()), v.to_value());
|
||||
}
|
||||
@@ -257,7 +257,6 @@ mod metadata_serde {
|
||||
mod tests {
|
||||
use super::*;
|
||||
use crate::value::Value;
|
||||
use alloc::collections::BTreeSet;
|
||||
|
||||
/// Round-trip: Value → MetadataValue → Value must be equivalent for
|
||||
/// all lossless variants (strings, bools, integers, arrays, objects).
|
||||
|
||||
@@ -1,6 +1,6 @@
|
||||
// Copyright (c) Microsoft Corporation.
|
||||
// Licensed under the MIT License.
|
||||
use alloc::collections::{BTreeMap, BTreeSet};
|
||||
use alloc::collections::BTreeSet;
|
||||
use alloc::format;
|
||||
use alloc::string::String;
|
||||
use alloc::vec::Vec;
|
||||
@@ -11,6 +11,7 @@ use serde::ser::{SerializeSeq as _, SerializeTuple as _};
|
||||
use serde::{Deserialize, Serialize};
|
||||
|
||||
use crate::number::Number;
|
||||
use crate::value::Object;
|
||||
use crate::value::Value;
|
||||
|
||||
const VARIANT_NULL: u32 = 0;
|
||||
@@ -132,7 +133,7 @@ impl<'a> Serialize for BinarySetRef<'a> {
|
||||
}
|
||||
}
|
||||
|
||||
struct BinaryObjectRef<'a>(&'a BTreeMap<Value, Value>);
|
||||
struct BinaryObjectRef<'a>(&'a Object);
|
||||
|
||||
impl<'a> Serialize for BinaryObjectRef<'a> {
|
||||
fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
|
||||
@@ -140,7 +141,7 @@ impl<'a> Serialize for BinaryObjectRef<'a> {
|
||||
S: serde::Serializer,
|
||||
{
|
||||
let mut seq = serializer.serialize_seq(Some(self.0.len()))?;
|
||||
for (key, value) in self.0.iter() {
|
||||
for (key, value) in self.0.iter_sorted() {
|
||||
seq.serialize_element(&BinaryEntryRef(key, value))?;
|
||||
}
|
||||
seq.end()
|
||||
@@ -261,11 +262,11 @@ impl<'de> Visitor<'de> for BinaryValueVisitor {
|
||||
}
|
||||
(BinaryVariant::Object, variant) => {
|
||||
let entries: Vec<(BinaryValue, BinaryValue)> = variant.newtype_variant()?;
|
||||
let mut map = BTreeMap::new();
|
||||
let mut map = Object::new();
|
||||
for (key, value) in entries {
|
||||
map.insert(key.into_value(), value.into_value());
|
||||
}
|
||||
Ok(BinaryValue(Value::from(map)))
|
||||
Ok(BinaryValue(Value::Object(crate::Rc::new(map))))
|
||||
}
|
||||
(BinaryVariant::Undefined, variant) => {
|
||||
variant.unit_variant()?;
|
||||
|
||||
@@ -6,8 +6,6 @@
|
||||
// Disable both to keep patterns consistent within this file.
|
||||
#![allow(clippy::pattern_type_mismatch, clippy::needless_borrowed_reference)]
|
||||
|
||||
use alloc::collections::BTreeSet;
|
||||
|
||||
use crate::number::Number;
|
||||
use crate::value::Value;
|
||||
|
||||
@@ -32,8 +30,9 @@ impl RegoVM {
|
||||
match (a, b) {
|
||||
(&Value::Number(ref x), &Value::Number(ref y)) => Ok(Value::from(x.sub(y)?)),
|
||||
(&Value::Set(ref left), &Value::Set(ref right)) => {
|
||||
let diff: BTreeSet<Value> = left.difference(right).cloned().collect();
|
||||
Ok(Value::from_set(diff))
|
||||
let diff: alloc::collections::BTreeSet<Value> =
|
||||
left.difference(right).cloned().collect();
|
||||
Ok(Value::from(diff))
|
||||
}
|
||||
_ => Err(VmError::InvalidSubtraction {
|
||||
left: a.clone(),
|
||||
|
||||
@@ -2,9 +2,9 @@
|
||||
// Licensed under the MIT License.
|
||||
|
||||
use crate::rvm::instructions::{ComprehensionBeginParams, ComprehensionMode};
|
||||
use crate::value::Object;
|
||||
use crate::value::Value;
|
||||
use crate::Rc;
|
||||
use alloc::collections::BTreeMap;
|
||||
use alloc::format;
|
||||
use alloc::sync::Arc;
|
||||
use alloc::vec::Vec;
|
||||
@@ -34,12 +34,12 @@ impl RegoVM {
|
||||
let initial_result = match params.mode {
|
||||
ComprehensionMode::Set => Value::new_set(),
|
||||
ComprehensionMode::Array => Value::new_array(),
|
||||
ComprehensionMode::Object => Value::Object(Rc::new(BTreeMap::new())),
|
||||
ComprehensionMode::Object => Value::Object(Rc::new(Object::new())),
|
||||
};
|
||||
self.set_register(params.result_reg, initial_result.clone())?;
|
||||
|
||||
let auto_iterate = params.collection_reg != params.result_reg;
|
||||
let iteration_state = if auto_iterate {
|
||||
let mut iteration_state = if auto_iterate {
|
||||
let source_value = self.get_register(params.collection_reg)?.clone();
|
||||
match source_value {
|
||||
Value::Array(items) => {
|
||||
@@ -53,11 +53,9 @@ impl RegoVM {
|
||||
if obj.is_empty() {
|
||||
None
|
||||
} else {
|
||||
Some(IterationState::Object {
|
||||
obj,
|
||||
current_key: None,
|
||||
first_iteration: true,
|
||||
})
|
||||
// O(1) cursor over shared Rc<Object>.
|
||||
let cursor = obj.cursor();
|
||||
Some(IterationState::Object { obj, cursor })
|
||||
}
|
||||
}
|
||||
Value::Set(set) => {
|
||||
@@ -79,7 +77,7 @@ impl RegoVM {
|
||||
None
|
||||
};
|
||||
|
||||
let has_iteration = if let Some(state) = iteration_state.as_ref() {
|
||||
let has_iteration = if let Some(state) = iteration_state.as_mut() {
|
||||
self.setup_next_iteration(state, params.key_reg, params.value_reg)?
|
||||
} else {
|
||||
false
|
||||
@@ -123,12 +121,12 @@ impl RegoVM {
|
||||
let initial_result = match params.mode {
|
||||
ComprehensionMode::Set => Value::new_set(),
|
||||
ComprehensionMode::Array => Value::new_array(),
|
||||
ComprehensionMode::Object => Value::Object(Rc::new(BTreeMap::new())),
|
||||
ComprehensionMode::Object => Value::Object(Rc::new(Object::new())),
|
||||
};
|
||||
self.set_register(params.result_reg, initial_result.clone())?;
|
||||
|
||||
let auto_iterate = params.collection_reg != params.result_reg;
|
||||
let iteration_state = if auto_iterate {
|
||||
let mut iteration_state = if auto_iterate {
|
||||
let source_value = self.get_register(params.collection_reg)?.clone();
|
||||
match source_value {
|
||||
Value::Array(items) => {
|
||||
@@ -142,11 +140,8 @@ impl RegoVM {
|
||||
if obj.is_empty() {
|
||||
None
|
||||
} else {
|
||||
Some(IterationState::Object {
|
||||
obj,
|
||||
current_key: None,
|
||||
first_iteration: true,
|
||||
})
|
||||
let cursor = obj.cursor();
|
||||
Some(IterationState::Object { obj, cursor })
|
||||
}
|
||||
}
|
||||
Value::Set(set) => {
|
||||
@@ -168,7 +163,7 @@ impl RegoVM {
|
||||
None
|
||||
};
|
||||
|
||||
let has_iteration = if let Some(state) = iteration_state.as_ref() {
|
||||
let has_iteration = if let Some(state) = iteration_state.as_mut() {
|
||||
self.setup_next_iteration(state, params.key_reg, params.value_reg)?
|
||||
} else {
|
||||
false
|
||||
@@ -255,6 +250,21 @@ impl RegoVM {
|
||||
};
|
||||
|
||||
let result_reg = comprehension_context.result_reg;
|
||||
// Snapshot the iteration value register BEFORE taking the result
|
||||
// register: if the comprehension compiler ever allocates
|
||||
// `result_reg == context.value_reg`, the writeback at the bottom
|
||||
// of this function would clobber the value register, and a
|
||||
// post-writeback read here would feed the wrong value into
|
||||
// `IterationState::Set::current_item`. Only Set needs the snapshot
|
||||
// (Object uses a self-advancing cursor; Array advances by index).
|
||||
let set_resume_snapshot = if matches!(
|
||||
comprehension_context.iteration_state,
|
||||
Some(IterationState::Set { .. })
|
||||
) {
|
||||
Some(self.get_register(comprehension_context.value_reg)?.clone())
|
||||
} else {
|
||||
None
|
||||
};
|
||||
// Take ownership of the result register so Rc refcount stays at 1,
|
||||
// allowing Rc::make_mut to mutate in-place instead of deep-cloning.
|
||||
let mut current_result = self.take_register(result_reg)?;
|
||||
@@ -292,29 +302,16 @@ impl RegoVM {
|
||||
self.set_register(result_reg, current_result)?;
|
||||
|
||||
if let Some(iter_state) = comprehension_context.iteration_state.as_mut() {
|
||||
match *iter_state {
|
||||
IterationState::Object {
|
||||
ref mut current_key,
|
||||
..
|
||||
} => {
|
||||
let tracked_key =
|
||||
if comprehension_context.key_reg != comprehension_context.value_reg {
|
||||
self.get_register(comprehension_context.key_reg)?.clone()
|
||||
} else {
|
||||
self.get_register(comprehension_context.value_reg)?.clone()
|
||||
};
|
||||
*current_key = Some(tracked_key);
|
||||
}
|
||||
IterationState::Set {
|
||||
ref mut current_item,
|
||||
..
|
||||
} => {
|
||||
*current_item =
|
||||
Some(self.get_register(comprehension_context.value_reg)?.clone());
|
||||
}
|
||||
IterationState::Array { .. } | IterationState::Single { .. } => {}
|
||||
// Set's `Bound::Excluded(current_item)` resume scheme needs the
|
||||
// pre-mutation snapshot taken at the top of this function.
|
||||
// Object uses a self-advancing cursor and needs no snapshot.
|
||||
if let IterationState::Set {
|
||||
ref mut current_item,
|
||||
..
|
||||
} = *iter_state
|
||||
{
|
||||
*current_item = set_resume_snapshot;
|
||||
}
|
||||
|
||||
iter_state.advance();
|
||||
let has_next = self.setup_next_iteration(
|
||||
iter_state,
|
||||
@@ -359,8 +356,7 @@ impl RegoVM {
|
||||
result_reg_idx,
|
||||
key_reg_idx,
|
||||
value_reg_idx,
|
||||
iteration_key,
|
||||
iteration_value,
|
||||
iter_is_set,
|
||||
) = {
|
||||
let frame =
|
||||
self.execution_stack
|
||||
@@ -382,8 +378,8 @@ impl RegoVM {
|
||||
|
||||
let result_reg_idx = context.result_reg;
|
||||
let mode = context.mode.clone();
|
||||
let iteration_key = self.get_register(context.key_reg)?.clone();
|
||||
let iteration_value = self.get_register(context.value_reg)?.clone();
|
||||
let iter_is_set =
|
||||
matches!(context.iteration_state, Some(IterationState::Set { .. }));
|
||||
|
||||
(
|
||||
value_to_add,
|
||||
@@ -392,8 +388,7 @@ impl RegoVM {
|
||||
result_reg_idx,
|
||||
context.key_reg,
|
||||
context.value_reg,
|
||||
iteration_key,
|
||||
iteration_value,
|
||||
iter_is_set,
|
||||
)
|
||||
} else {
|
||||
return Err(VmError::InvalidIteration {
|
||||
@@ -403,6 +398,18 @@ impl RegoVM {
|
||||
}
|
||||
};
|
||||
|
||||
// Snapshot the iteration value register BEFORE the result writeback:
|
||||
// if the compiler ever allocates `result_reg == value_reg_idx`, a
|
||||
// post-writeback read would feed the result accumulator into
|
||||
// `IterationState::Set::current_item`, breaking the next iteration.
|
||||
// Only Set needs this (Object cursor self-advances; Array advances
|
||||
// by index).
|
||||
let set_resume_snapshot = if iter_is_set {
|
||||
Some(self.get_register(value_reg_idx)?.clone())
|
||||
} else {
|
||||
None
|
||||
};
|
||||
|
||||
// Take ownership of the result register so Rc refcount stays at 1,
|
||||
// allowing Rc::make_mut to mutate in-place instead of deep-cloning.
|
||||
let mut current_result = self.take_register(result_reg_idx)?;
|
||||
@@ -450,27 +457,13 @@ impl RegoVM {
|
||||
} = &mut frame.kind
|
||||
{
|
||||
if let Some(iter_state) = context.iteration_state.as_mut() {
|
||||
match *iter_state {
|
||||
IterationState::Object {
|
||||
ref mut current_key,
|
||||
..
|
||||
} => {
|
||||
let tracked_key = if context.key_reg != context.value_reg {
|
||||
iteration_key.clone()
|
||||
} else {
|
||||
iteration_value.clone()
|
||||
};
|
||||
*current_key = Some(tracked_key);
|
||||
}
|
||||
IterationState::Set {
|
||||
ref mut current_item,
|
||||
..
|
||||
} => {
|
||||
*current_item = Some(iteration_value.clone());
|
||||
}
|
||||
IterationState::Array { .. } | IterationState::Single { .. } => {}
|
||||
if let IterationState::Set {
|
||||
ref mut current_item,
|
||||
..
|
||||
} = *iter_state
|
||||
{
|
||||
*current_item = set_resume_snapshot;
|
||||
}
|
||||
|
||||
iter_state.advance();
|
||||
}
|
||||
|
||||
@@ -487,8 +480,21 @@ impl RegoVM {
|
||||
}
|
||||
};
|
||||
|
||||
if let Some(state) = iteration_state_snapshot.as_ref() {
|
||||
let has_next = self.setup_next_iteration(state, key_reg_idx, value_reg_idx)?;
|
||||
if let Some(mut state) = iteration_state_snapshot {
|
||||
let has_next = self.setup_next_iteration(&mut state, key_reg_idx, value_reg_idx)?;
|
||||
|
||||
// `setup_next_iteration` advances Object's internal cursor; the
|
||||
// owning frame holds the iteration_state, so we must write the
|
||||
// updated state back. (The Array/Set variants are also unchanged
|
||||
// by copy, so the writeback is uniform.)
|
||||
if let Some(frame) = self.execution_stack.get_mut(comprehension_index) {
|
||||
if let FrameKind::Comprehension {
|
||||
ref mut context, ..
|
||||
} = frame.kind
|
||||
{
|
||||
context.iteration_state = Some(state);
|
||||
}
|
||||
}
|
||||
|
||||
if has_next {
|
||||
if let Some(frame) = self.execution_stack.get_mut(comprehension_index) {
|
||||
@@ -528,7 +534,6 @@ impl RegoVM {
|
||||
Ok(false)
|
||||
}
|
||||
}
|
||||
|
||||
pub(super) fn handle_comprehension_condition_failure_suspendable(&mut self) -> Result<bool> {
|
||||
if let Some(mut frame) = self.execution_stack.pop() {
|
||||
let handled = if let &mut FrameKind::Comprehension {
|
||||
@@ -554,11 +559,16 @@ impl RegoVM {
|
||||
context: &mut ComprehensionContext,
|
||||
) -> Result<()> {
|
||||
if let Some(iter_state) = context.iteration_state.as_mut() {
|
||||
self.capture_comprehension_iteration_position(
|
||||
iter_state,
|
||||
context.key_reg,
|
||||
context.value_reg,
|
||||
)?;
|
||||
// Snapshot the current value into Set's `current_item` so the
|
||||
// next iteration can resume from `Bound::Excluded(current)`.
|
||||
// Object uses a self-advancing cursor and needs no snapshot here.
|
||||
if let IterationState::Set {
|
||||
ref mut current_item,
|
||||
..
|
||||
} = *iter_state
|
||||
{
|
||||
*current_item = Some(self.get_register(context.value_reg)?.clone());
|
||||
}
|
||||
iter_state.advance();
|
||||
let has_next =
|
||||
self.setup_next_iteration(iter_state, context.key_reg, context.value_reg)?;
|
||||
@@ -575,37 +585,10 @@ impl RegoVM {
|
||||
Ok(())
|
||||
}
|
||||
|
||||
fn capture_comprehension_iteration_position(
|
||||
&mut self,
|
||||
iter_state: &mut IterationState,
|
||||
key_reg: u8,
|
||||
value_reg: u8,
|
||||
) -> Result<()> {
|
||||
match *iter_state {
|
||||
IterationState::Object {
|
||||
ref mut current_key,
|
||||
..
|
||||
} => {
|
||||
let tracked_key = if key_reg != value_reg {
|
||||
self.get_register(key_reg)?.clone()
|
||||
} else {
|
||||
self.get_register(value_reg)?.clone()
|
||||
};
|
||||
*current_key = Some(tracked_key);
|
||||
}
|
||||
IterationState::Set {
|
||||
ref mut current_item,
|
||||
..
|
||||
} => {
|
||||
*current_item = Some(self.get_register(value_reg)?.clone());
|
||||
}
|
||||
IterationState::Array { .. } | IterationState::Single { .. } => {}
|
||||
}
|
||||
|
||||
Ok(())
|
||||
}
|
||||
|
||||
fn execute_comprehension_end_run_to_completion(&mut self) -> Result<()> {
|
||||
// `ComprehensionEnd` is reached from a loaded program; an empty stack
|
||||
// here means malformed user-supplied bytecode, which must still surface
|
||||
// as a typed error rather than a panic — including in debug builds.
|
||||
self.comprehension_stack.pop().map_or_else(
|
||||
|| {
|
||||
Err(VmError::InvalidIteration {
|
||||
|
||||
@@ -3,8 +3,9 @@
|
||||
|
||||
use crate::rvm::instructions::{ComprehensionMode, LoopMode};
|
||||
use crate::value::Value;
|
||||
use crate::value::{Object, ObjectCursor};
|
||||
use crate::Rc;
|
||||
use alloc::collections::{BTreeMap, BTreeSet};
|
||||
use alloc::collections::BTreeSet;
|
||||
use alloc::vec::Vec;
|
||||
|
||||
/// Loop execution context for managing iteration state
|
||||
@@ -24,7 +25,18 @@ pub struct LoopContext {
|
||||
pub current_iteration_failed: bool, // Track if current iteration had condition failures
|
||||
}
|
||||
|
||||
/// Iterator state for different collection types
|
||||
/// Iterator state for different collection types.
|
||||
///
|
||||
/// Snapshot independence for `Object` is provided by the shared
|
||||
/// `Rc<Object>` — `Rc::make_mut` on an aliased Rc allocates a new
|
||||
/// collection, leaving the iterator's Rc pointing at the original
|
||||
/// pre-mutation state. The `ObjectCursor` is opaque and resumes in
|
||||
/// O(log n) for the BTree backend.
|
||||
///
|
||||
/// `Set` continues to use the pre-existing snapshot-by-cloned-key
|
||||
/// approach (`current_item` + `first_iteration`); migration of `Set`
|
||||
/// to a cursor-based iterator ships with the `Set` storage abstraction
|
||||
/// in a follow-up PR.
|
||||
#[derive(Debug, Clone)]
|
||||
pub enum IterationState {
|
||||
Array {
|
||||
@@ -32,9 +44,8 @@ pub enum IterationState {
|
||||
index: usize,
|
||||
},
|
||||
Object {
|
||||
obj: Rc<BTreeMap<Value, Value>>,
|
||||
current_key: Option<Value>,
|
||||
first_iteration: bool,
|
||||
obj: Rc<Object>,
|
||||
cursor: ObjectCursor,
|
||||
},
|
||||
Set {
|
||||
items: Rc<BTreeSet<Value>>,
|
||||
@@ -54,13 +65,21 @@ impl IterationState {
|
||||
pub(super) const fn advance(&mut self) {
|
||||
match *self {
|
||||
Self::Array { ref mut index, .. } => {
|
||||
// Array iteration uses `usize` as the cursor and advances via
|
||||
// `saturating_add(1)`. A cursor already at `usize::MAX` here
|
||||
// means a stuck (non-progressing) iteration was emitted by
|
||||
// malformed bytecode; assert in debug to surface it loudly.
|
||||
debug_assert!(
|
||||
*index < usize::MAX,
|
||||
"IterationState::Array index already at usize::MAX on advance"
|
||||
);
|
||||
*index = index.saturating_add(1);
|
||||
}
|
||||
Self::Object {
|
||||
ref mut first_iteration,
|
||||
..
|
||||
}
|
||||
| Self::Set {
|
||||
// For Object the cursor advances inside `setup_next_iteration`
|
||||
// when it pulls the next item via `Object::next`, so `advance`
|
||||
// is a no-op for the cursor-backed Object variant.
|
||||
Self::Object { .. } => {}
|
||||
Self::Set {
|
||||
ref mut first_iteration,
|
||||
..
|
||||
} => {
|
||||
@@ -69,6 +88,12 @@ impl IterationState {
|
||||
Self::Single {
|
||||
ref mut consumed, ..
|
||||
} => {
|
||||
// `Single` yields exactly once; advancing a consumed Single
|
||||
// means the compiler emitted a redundant LoopNext.
|
||||
debug_assert!(
|
||||
!*consumed,
|
||||
"IterationState::Single advanced after consumption"
|
||||
);
|
||||
*consumed = true;
|
||||
}
|
||||
}
|
||||
@@ -107,3 +132,71 @@ pub(super) struct ComprehensionContext {
|
||||
/// Resume location for the parent frame once this comprehension completes
|
||||
pub(super) resume_pc: usize,
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
#[allow(
|
||||
clippy::expect_used,
|
||||
clippy::unwrap_used,
|
||||
clippy::unreachable,
|
||||
clippy::pattern_type_mismatch,
|
||||
clippy::shadow_unrelated,
|
||||
clippy::panic
|
||||
)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
use crate::value::Object;
|
||||
|
||||
/// IterationState::Object holds an `Rc<Object>` plus an opaque cursor.
|
||||
/// Mutating an aliased Rc via `Rc::make_mut` allocates a new collection
|
||||
/// (CoW) so the in-flight iterator's source is unaffected.
|
||||
#[test]
|
||||
fn iteration_state_object_is_snapshot_independent_of_source() {
|
||||
let mut obj = Object::new();
|
||||
obj.insert(Value::from("a"), Value::from(1));
|
||||
obj.insert(Value::from("b"), Value::from(2));
|
||||
obj.insert(Value::from("c"), Value::from(3));
|
||||
|
||||
let source = Value::Object(Rc::new(obj));
|
||||
|
||||
let snapshot_obj = match &source {
|
||||
Value::Object(o) => Rc::clone(o),
|
||||
_ => unreachable!(),
|
||||
};
|
||||
let state = IterationState::Object {
|
||||
obj: Rc::clone(&snapshot_obj),
|
||||
cursor: snapshot_obj.cursor(),
|
||||
};
|
||||
|
||||
// Mutate a clone of the source mid-iteration.
|
||||
let mut alias = source.clone();
|
||||
let inner = alias.as_object_mut().expect("object");
|
||||
inner.insert(Value::from("a"), Value::from(999));
|
||||
inner.insert(Value::from("d"), Value::from(4));
|
||||
inner.remove(&Value::from("b"));
|
||||
|
||||
// Drain the snapshot via the cursor — must still report the original
|
||||
// 3 entries with original values.
|
||||
let mut collected: Vec<(Value, Value)> = Vec::new();
|
||||
if let IterationState::Object {
|
||||
ref obj,
|
||||
mut cursor,
|
||||
} = state
|
||||
{
|
||||
while let Some((k, v)) = obj.next(&mut cursor) {
|
||||
collected.push((k.clone(), v.clone()));
|
||||
}
|
||||
} else {
|
||||
unreachable!();
|
||||
}
|
||||
assert_eq!(collected.len(), 3);
|
||||
assert!(collected.contains(&(Value::from("a"), Value::from(1))));
|
||||
assert!(collected.contains(&(Value::from("b"), Value::from(2))));
|
||||
assert!(collected.contains(&(Value::from("c"), Value::from(3))));
|
||||
assert!(!collected.iter().any(|kv| kv.0 == Value::from("d")));
|
||||
|
||||
// The original source Value (untouched) is also unchanged.
|
||||
let src_obj = source.as_object().expect("object");
|
||||
assert_eq!(src_obj.len(), 3);
|
||||
assert_eq!(src_obj.get(&Value::from("a")), Some(&Value::from(1)));
|
||||
}
|
||||
}
|
||||
|
||||
@@ -4,7 +4,6 @@
|
||||
use crate::rvm::instructions::{GuardMode, Instruction, LiteralOrRegister};
|
||||
use crate::rvm::program::Program;
|
||||
use crate::value::Value;
|
||||
use alloc::collections::BTreeSet;
|
||||
use alloc::vec::Vec;
|
||||
use core::mem;
|
||||
|
||||
@@ -670,7 +669,7 @@ impl RegoVM {
|
||||
}
|
||||
}
|
||||
SetNew { dest } => {
|
||||
let empty_set = Value::Set(crate::Rc::new(BTreeSet::new()));
|
||||
let empty_set = Value::new_set();
|
||||
self.set_register(dest, empty_set)?;
|
||||
Ok(InstructionOutcome::Continue)
|
||||
}
|
||||
@@ -707,7 +706,7 @@ impl RegoVM {
|
||||
if any_undefined {
|
||||
self.set_register(params.dest, Value::Undefined)?;
|
||||
} else {
|
||||
let mut set = BTreeSet::new();
|
||||
let mut set = alloc::collections::BTreeSet::new();
|
||||
for ® in params.element_registers() {
|
||||
set.insert(self.get_register(reg)?.clone());
|
||||
}
|
||||
|
||||
@@ -295,6 +295,13 @@ pub enum VmError {
|
||||
#[error("Call rule stack underflow during rule finalization (pc={pc})")]
|
||||
CallRuleStackUnderflow { pc: usize },
|
||||
|
||||
#[error("Call rule stack mismatch during rule finalization: expected rule_index {expected}, popped {actual} (pc={pc})")]
|
||||
CallRuleStackMismatch {
|
||||
expected: u16,
|
||||
actual: u16,
|
||||
pc: usize,
|
||||
},
|
||||
|
||||
#[error("Internal VM error: {message} (pc={pc})")]
|
||||
Internal { message: String, pc: usize },
|
||||
}
|
||||
|
||||
@@ -117,6 +117,10 @@ impl RegoVM {
|
||||
let target = self.convert_pc(target, "jump target")?;
|
||||
self.pc = target;
|
||||
while self.pc < program.instructions.len() {
|
||||
// Per-instruction sanity check: every iteration of the dispatch
|
||||
// loop must re-enter with the VM in a Running/Ready state and the
|
||||
// working data structures coherent.
|
||||
self.assert_vm_invariants();
|
||||
self.memory_check()?;
|
||||
if self.executed_instructions >= self.max_instructions {
|
||||
return Err(VmError::InstructionLimitExceeded {
|
||||
@@ -189,6 +193,9 @@ impl RegoVM {
|
||||
}
|
||||
|
||||
fn execute_suspendable_entry(&mut self, entry_point_pc: usize) -> Result<Value> {
|
||||
// Precondition: callers (execute_entry_point_by_{index,name}) reset the
|
||||
// VM before invoking this method, so the VM must be in a clean state.
|
||||
self.debug_assert_state_is_clean();
|
||||
self.execution_state = ExecutionState::Running;
|
||||
self.reset_execution_timer_state();
|
||||
match self.run_stackless_from(entry_point_pc) {
|
||||
@@ -201,6 +208,10 @@ impl RegoVM {
|
||||
}
|
||||
|
||||
pub fn resume(&mut self, resume_value: Option<Value>) -> Result<Value> {
|
||||
// Precondition is enforced below by returning `VmError::InvalidResumeState`
|
||||
// for any non-`Suspended` state. A `debug_assert!` here would diverge
|
||||
// debug vs release behavior and, when invoked via FFI, would trip the
|
||||
// unwind guard and poison the engine on a recoverable misuse.
|
||||
let (reason, mut last_result) = match self.execution_state.clone() {
|
||||
ExecutionState::Suspended {
|
||||
reason,
|
||||
@@ -289,6 +300,9 @@ impl RegoVM {
|
||||
|
||||
fn run_stackless_loop(&mut self, program: &Program, last_result: &mut Value) -> Result<()> {
|
||||
while !self.execution_stack.is_empty() {
|
||||
// Per-instruction sanity check: see `assert_vm_invariants` for the
|
||||
// exact contract. Compiled out in release.
|
||||
self.assert_vm_invariants();
|
||||
self.memory_check()?;
|
||||
self.frame_pc_overridden = false;
|
||||
let should_finalize_rule = self.execution_stack.last().is_some_and(|frame| {
|
||||
|
||||
@@ -3,6 +3,7 @@
|
||||
|
||||
use crate::rvm::instructions::LoopMode;
|
||||
use crate::value::Value;
|
||||
use crate::Rc;
|
||||
|
||||
use super::context::{IterationState, LoopContext};
|
||||
use super::errors::{Result, VmError};
|
||||
@@ -89,13 +90,13 @@ impl RegoVM {
|
||||
) -> Result<()> {
|
||||
self.set_register(params.result_reg, Value::Bool(false))?;
|
||||
|
||||
let iteration_state = match self.resolve_iteration_state(mode, ¶ms)? {
|
||||
let mut iteration_state = match self.resolve_iteration_state(mode, ¶ms)? {
|
||||
Some(state) => state,
|
||||
None => return Ok(()),
|
||||
};
|
||||
|
||||
let has_next =
|
||||
self.setup_next_iteration(&iteration_state, params.key_reg, params.value_reg)?;
|
||||
self.setup_next_iteration(&mut iteration_state, params.key_reg, params.value_reg)?;
|
||||
if !has_next {
|
||||
self.pc = usize::from(params.loop_end);
|
||||
return Ok(());
|
||||
@@ -155,15 +156,10 @@ impl RegoVM {
|
||||
LoopAction::Continue => {}
|
||||
}
|
||||
|
||||
if let &mut IterationState::Object {
|
||||
ref mut current_key,
|
||||
..
|
||||
} = &mut loop_ctx.iteration_state
|
||||
{
|
||||
if loop_ctx.key_reg != loop_ctx.value_reg {
|
||||
*current_key = Some(self.get_register(loop_ctx.key_reg)?.clone());
|
||||
}
|
||||
} else if let &mut IterationState::Set {
|
||||
// Snapshot the current value for Set so its next iteration can resume
|
||||
// from `Bound::Excluded(current)`. Object uses a cursor and advances
|
||||
// inside `setup_next_iteration` itself.
|
||||
if let &mut IterationState::Set {
|
||||
ref mut current_item,
|
||||
..
|
||||
} = &mut loop_ctx.iteration_state
|
||||
@@ -173,7 +169,7 @@ impl RegoVM {
|
||||
|
||||
loop_ctx.iteration_state.advance();
|
||||
let has_next = self.setup_next_iteration(
|
||||
&loop_ctx.iteration_state,
|
||||
&mut loop_ctx.iteration_state,
|
||||
loop_ctx.key_reg,
|
||||
loop_ctx.value_reg,
|
||||
)?;
|
||||
@@ -211,13 +207,13 @@ impl RegoVM {
|
||||
) -> Result<()> {
|
||||
self.set_register(params.result_reg, Value::Bool(false))?;
|
||||
|
||||
let iteration_state = match self.resolve_iteration_state(mode, ¶ms)? {
|
||||
let mut iteration_state = match self.resolve_iteration_state(mode, ¶ms)? {
|
||||
Some(state) => state,
|
||||
None => return Ok(()),
|
||||
};
|
||||
|
||||
let has_next =
|
||||
self.setup_next_iteration(&iteration_state, params.key_reg, params.value_reg)?;
|
||||
self.setup_next_iteration(&mut iteration_state, params.key_reg, params.value_reg)?;
|
||||
if !has_next {
|
||||
self.pc = usize::from(params.loop_end);
|
||||
return Ok(());
|
||||
@@ -316,7 +312,14 @@ impl RegoVM {
|
||||
Ok(())
|
||||
}
|
||||
LoopAction::Continue => {
|
||||
let (mode, success_count, total_iterations, key_reg, value_reg, iteration_state) = {
|
||||
let (
|
||||
mode,
|
||||
success_count,
|
||||
total_iterations,
|
||||
key_reg,
|
||||
value_reg,
|
||||
mut iteration_state,
|
||||
) = {
|
||||
let (mode, success_count, total_iterations, key_reg, value_reg) = {
|
||||
let frame = self
|
||||
.execution_stack
|
||||
@@ -334,11 +337,6 @@ impl RegoVM {
|
||||
}
|
||||
};
|
||||
|
||||
let key_value = if key_reg != value_reg {
|
||||
Some(self.get_register(key_reg)?.clone())
|
||||
} else {
|
||||
None
|
||||
};
|
||||
let value_value = self.get_register(value_reg)?.clone();
|
||||
|
||||
let frame = self
|
||||
@@ -349,20 +347,16 @@ impl RegoVM {
|
||||
&mut FrameKind::Loop {
|
||||
ref mut context, ..
|
||||
} => {
|
||||
if let &mut IterationState::Object {
|
||||
ref mut current_key,
|
||||
..
|
||||
} = &mut context.iteration_state
|
||||
{
|
||||
if context.key_reg != context.value_reg {
|
||||
*current_key = key_value;
|
||||
}
|
||||
} else if let &mut IterationState::Set {
|
||||
// Snapshot the current value for Set so its next
|
||||
// iteration can resume from `Bound::Excluded(current)`.
|
||||
// Object uses a cursor and advances inside
|
||||
// `setup_next_iteration` itself.
|
||||
if let &mut IterationState::Set {
|
||||
ref mut current_item,
|
||||
..
|
||||
} = &mut context.iteration_state
|
||||
{
|
||||
*current_item = Some(value_value.clone());
|
||||
*current_item = Some(value_value);
|
||||
}
|
||||
|
||||
context.iteration_state.advance();
|
||||
@@ -381,7 +375,21 @@ impl RegoVM {
|
||||
}
|
||||
};
|
||||
|
||||
let has_next = self.setup_next_iteration(&iteration_state, key_reg, value_reg)?;
|
||||
let has_next =
|
||||
self.setup_next_iteration(&mut iteration_state, key_reg, value_reg)?;
|
||||
|
||||
// `setup_next_iteration` advances Object's internal cursor;
|
||||
// the owning frame holds the iteration_state, so we must
|
||||
// write the updated state back. (Array/Set are unchanged by
|
||||
// the call, so the writeback is uniform.)
|
||||
if let Some(frame) = self.execution_stack.last_mut() {
|
||||
if let FrameKind::Loop {
|
||||
ref mut context, ..
|
||||
} = frame.kind
|
||||
{
|
||||
context.iteration_state = iteration_state;
|
||||
}
|
||||
}
|
||||
|
||||
if has_next {
|
||||
if let Some(frame) = self.execution_stack.last_mut() {
|
||||
@@ -459,10 +467,14 @@ impl RegoVM {
|
||||
self.handle_empty_collection(mode, params.result_reg, params.loop_end)?;
|
||||
return Ok(None);
|
||||
}
|
||||
// O(1) resumable cursor over the shared Rc<Object>.
|
||||
// No eager pair snapshot: avoids O(N) setup, O(N) memory
|
||||
// floor, and O(N) memory-limit checks. Snapshot
|
||||
// independence is via the shared Rc (CoW).
|
||||
let cursor = obj.cursor();
|
||||
Ok(Some(IterationState::Object {
|
||||
obj: obj.clone(),
|
||||
current_key: None,
|
||||
first_iteration: true,
|
||||
obj: Rc::clone(obj),
|
||||
cursor,
|
||||
}))
|
||||
}
|
||||
}
|
||||
@@ -512,7 +524,7 @@ impl RegoVM {
|
||||
|
||||
pub(super) fn setup_next_iteration(
|
||||
&mut self,
|
||||
state: &IterationState,
|
||||
state: &mut IterationState,
|
||||
key_reg: u8,
|
||||
value_reg: u8,
|
||||
) -> Result<bool> {
|
||||
@@ -538,33 +550,19 @@ impl RegoVM {
|
||||
}
|
||||
IterationState::Object {
|
||||
ref obj,
|
||||
ref current_key,
|
||||
ref first_iteration,
|
||||
ref mut cursor,
|
||||
} => {
|
||||
if *first_iteration {
|
||||
if let Some((key, value)) = obj.iter().next() {
|
||||
if key_reg != value_reg {
|
||||
self.set_register(key_reg, key.clone())?;
|
||||
}
|
||||
self.set_register(value_reg, value.clone())?;
|
||||
Ok(true)
|
||||
} else {
|
||||
Ok(false)
|
||||
}
|
||||
} else if let Some(ref current) = *current_key {
|
||||
let mut range_iter = obj.range((
|
||||
core::ops::Bound::Excluded(current),
|
||||
core::ops::Bound::Unbounded,
|
||||
));
|
||||
if let Some((key, value)) = range_iter.next() {
|
||||
if key_reg != value_reg {
|
||||
self.set_register(key_reg, key.clone())?;
|
||||
}
|
||||
self.set_register(value_reg, value.clone())?;
|
||||
Ok(true)
|
||||
} else {
|
||||
Ok(false)
|
||||
// Object iterates via a resumable cursor on the shared
|
||||
// `Rc<Object>`; `next` both yields the current entry and
|
||||
// advances the cursor. No explicit `current_key` snapshot is
|
||||
// needed — see the doc on `IterationState`.
|
||||
if let Some((key, value)) = obj.next(cursor) {
|
||||
let value = value.clone();
|
||||
if key_reg != value_reg {
|
||||
self.set_register(key_reg, key.clone())?;
|
||||
}
|
||||
self.set_register(value_reg, value)?;
|
||||
Ok(true)
|
||||
} else {
|
||||
Ok(false)
|
||||
}
|
||||
|
||||
@@ -277,6 +277,19 @@ impl RegoVM {
|
||||
.call_rule_stack
|
||||
.pop()
|
||||
.ok_or(VmError::CallRuleStackUnderflow { pc: self.pc })?;
|
||||
// Stack discipline: the context we just popped must belong to the
|
||||
// rule we are finalizing. A mismatch indicates a missing push or an
|
||||
// extra pop somewhere in this rule's execution and would otherwise
|
||||
// silently restore the wrong return_pc / rule_type. Surface as a
|
||||
// typed VmError so the contract holds the same in debug and release
|
||||
// builds (avoiding FFI poisoning via a debug-only panic).
|
||||
if rule_index != call_context.rule_index {
|
||||
return Err(VmError::CallRuleStackMismatch {
|
||||
expected: rule_index,
|
||||
actual: call_context.rule_index,
|
||||
pc: self.pc,
|
||||
});
|
||||
}
|
||||
self.pc = call_context.return_pc;
|
||||
|
||||
let result_from_rule = if !rule_failed_due_to_inconsistency {
|
||||
@@ -831,6 +844,9 @@ impl RegoVM {
|
||||
|
||||
self.registers = parent_registers;
|
||||
|
||||
// Underflow here means malformed/poisoned program state; surface as a
|
||||
// typed error rather than a debug-only panic so the public load_program
|
||||
// contract holds the same in debug and release.
|
||||
if self.call_rule_stack.pop().is_none() {
|
||||
return Err(VmError::CallRuleStackUnderflow { pc: self.pc });
|
||||
}
|
||||
|
||||
@@ -34,6 +34,139 @@ impl RegoVM {
|
||||
|
||||
// Builtin cache entries only live for a single execution
|
||||
self.builtins_cache.clear();
|
||||
|
||||
// Postcondition: every stack/cache that `reset_execution_state` touches
|
||||
// must be in its documented "clean" shape. This catches accidental
|
||||
// omissions in future edits to this function.
|
||||
self.debug_assert_state_is_clean();
|
||||
}
|
||||
|
||||
/// Debug-only postcondition for `reset_execution_state`.
|
||||
///
|
||||
/// Asserts the invariants every caller of `reset_execution_state` relies on
|
||||
/// before starting a fresh execution. The body is fully gated by
|
||||
/// `#[cfg(debug_assertions)]` so this is a zero-cost no-op in release.
|
||||
#[inline]
|
||||
pub(super) fn debug_assert_state_is_clean(&self) {
|
||||
#[cfg(debug_assertions)]
|
||||
{
|
||||
// --- Stacks: every per-execution stack must be drained. ---
|
||||
debug_assert!(
|
||||
self.execution_stack.is_empty(),
|
||||
"reset_execution_state postcondition: execution_stack must be empty"
|
||||
);
|
||||
debug_assert!(
|
||||
self.loop_stack.is_empty(),
|
||||
"reset_execution_state postcondition: loop_stack must be empty"
|
||||
);
|
||||
debug_assert!(
|
||||
self.comprehension_stack.is_empty(),
|
||||
"reset_execution_state postcondition: comprehension_stack must be empty"
|
||||
);
|
||||
debug_assert!(
|
||||
self.call_rule_stack.is_empty(),
|
||||
"reset_execution_state postcondition: call_rule_stack must be empty"
|
||||
);
|
||||
debug_assert!(
|
||||
self.register_stack.is_empty(),
|
||||
"reset_execution_state postcondition: register_stack must be empty"
|
||||
);
|
||||
|
||||
// --- Caches: cleared so a new program/input cannot read stale entries. ---
|
||||
debug_assert!(
|
||||
self.builtins_cache.is_empty(),
|
||||
"reset_execution_state postcondition: builtins_cache must be empty"
|
||||
);
|
||||
|
||||
// --- Registers: window resized to the program's base count and zeroed. ---
|
||||
debug_assert_eq!(
|
||||
self.registers.len(),
|
||||
self.base_register_count,
|
||||
"reset_execution_state postcondition: registers must be sized to base_register_count"
|
||||
);
|
||||
debug_assert!(
|
||||
self.registers.iter().all(|v| matches!(v, Value::Undefined)),
|
||||
"reset_execution_state postcondition: all registers must be Undefined"
|
||||
);
|
||||
|
||||
// --- Rule cache: sized to the current program and marked uncomputed. ---
|
||||
debug_assert_eq!(
|
||||
self.rule_cache.len(),
|
||||
self.program.rule_infos.len(),
|
||||
"reset_execution_state postcondition: rule_cache size must match program rule_infos"
|
||||
);
|
||||
debug_assert!(
|
||||
self.rule_cache.iter().all(|entry| !entry.0),
|
||||
"reset_execution_state postcondition: rule_cache entries must be uncomputed"
|
||||
);
|
||||
|
||||
// --- Counters and execution-state machine: zeroed and back to Ready. ---
|
||||
debug_assert_eq!(
|
||||
self.pc, 0,
|
||||
"reset_execution_state postcondition: pc must be 0"
|
||||
);
|
||||
debug_assert_eq!(
|
||||
self.executed_instructions, 0,
|
||||
"reset_execution_state postcondition: executed_instructions must be 0"
|
||||
);
|
||||
debug_assert!(
|
||||
matches!(self.execution_state, ExecutionState::Ready),
|
||||
"reset_execution_state postcondition: execution_state must be Ready"
|
||||
);
|
||||
}
|
||||
}
|
||||
|
||||
/// Per-opcode VM invariants checked from the inner dispatch loop.
|
||||
///
|
||||
/// These hold every time control re-enters the dispatch loop with another
|
||||
/// instruction to execute. Only conditions that are *purely VM-internal*
|
||||
/// (i.e. cannot be made false by any host-supplied program or out-of-order
|
||||
/// API call) are asserted here — anything reachable from `load_program`
|
||||
/// input must surface as a typed `VmError` instead, to avoid panicking in
|
||||
/// debug builds and poisoning the engine across FFI.
|
||||
///
|
||||
/// Fully `#[cfg(debug_assertions)]`-gated so the method body compiles out
|
||||
/// in release.
|
||||
#[inline]
|
||||
pub(super) fn assert_vm_invariants(&self) {
|
||||
#[cfg(debug_assertions)]
|
||||
{
|
||||
// The dispatch loop only runs while execution is live. Once the VM
|
||||
// has transitioned to a terminal state (Suspended/Completed/Error)
|
||||
// the loop must have exited. Note `Ready` is also valid here because
|
||||
// some entry points (e.g. `execute_entry_point_by_index` in
|
||||
// RunToCompletion mode) drive `jump_to` without flipping the state.
|
||||
// `execution_state` is mutated only inside the VM and is not
|
||||
// host-controllable.
|
||||
debug_assert!(
|
||||
matches!(
|
||||
self.execution_state,
|
||||
ExecutionState::Ready | ExecutionState::Running
|
||||
),
|
||||
"vm invariant: execution_state must be Ready or Running inside the dispatch loop, was {:?}",
|
||||
self.execution_state
|
||||
);
|
||||
|
||||
// Rule cache is sized once at reset (against the currently loaded
|
||||
// program) and the VM does not resize it mid-execution. Any
|
||||
// mismatch here would indicate an internal accounting bug rather
|
||||
// than malformed input.
|
||||
debug_assert_eq!(
|
||||
self.rule_cache.len(),
|
||||
self.program.rule_infos.len(),
|
||||
"vm invariant: rule_cache size must equal program.rule_infos size"
|
||||
);
|
||||
|
||||
// NOTE: `!registers.is_empty()` and an `execution_stack` depth
|
||||
// ceiling were intentionally *not* asserted here: both can be
|
||||
// triggered by a host-loaded program (registers via
|
||||
// `RuleInfo::num_registers == 0`; stack depth via deeply nested
|
||||
// rules/loops/comprehensions) and would therefore panic in debug
|
||||
// and poison the engine across FFI. Register access is already
|
||||
// guarded by `VmError::RegisterIndexOutOfBounds`; runaway recursion
|
||||
// is bounded in production by `set_max_instructions` and
|
||||
// `memory_check`.
|
||||
}
|
||||
}
|
||||
|
||||
/// Return all active objects to their respective pools for reuse
|
||||
|
||||
@@ -569,7 +569,7 @@ impl Analyzer {
|
||||
}
|
||||
Ok(false)
|
||||
}
|
||||
Array { .. } | Object { .. } => Ok(true),
|
||||
Expr::Array { .. } | Expr::Object { .. } => Ok(true),
|
||||
_ => Ok(false),
|
||||
})?;
|
||||
Ok(true)
|
||||
@@ -666,7 +666,7 @@ impl Analyzer {
|
||||
Ok(false)
|
||||
}
|
||||
// TODO: key vs value for object binding
|
||||
Array { .. } | Object { .. } => Ok(true),
|
||||
Expr::Array { .. } | Expr::Object { .. } => Ok(true),
|
||||
_ => Ok(false),
|
||||
})?;
|
||||
Ok(vars)
|
||||
@@ -853,7 +853,7 @@ impl Analyzer {
|
||||
Ok(false)
|
||||
}
|
||||
// TODO: Object key/value
|
||||
Array { .. } | Object { .. } => Ok(true),
|
||||
Expr::Array { .. } | Expr::Object { .. } => Ok(true),
|
||||
_ => {
|
||||
non_vars.push(e.clone());
|
||||
Ok(false)
|
||||
|
||||
@@ -1498,7 +1498,7 @@ fn test_deserialize_object_default_empty_object() {
|
||||
let s = Schema::from_serde_json_value(schema).unwrap();
|
||||
match s.as_type() {
|
||||
Type::Object { default, .. } => {
|
||||
assert_eq!(default, &Some(Value::Object(Rc::new(BTreeMap::new()))));
|
||||
assert_eq!(default, &Some(Value::new_object()));
|
||||
}
|
||||
_ => panic!("Expected Type::Object"),
|
||||
}
|
||||
@@ -1778,8 +1778,11 @@ fn test_deserialize_enum_values_with_object_non_string_keys() {
|
||||
match s.as_type() {
|
||||
Type::Enum { values, .. } => match &values[0] {
|
||||
Value::Object(obj) => {
|
||||
assert_eq!(obj[&Value::from("1")], Value::from("one"));
|
||||
assert_eq!(obj[&Value::from("true")], Value::from("bool"));
|
||||
assert_eq!(*obj.get(&Value::from("1")).expect("1"), Value::from("one"));
|
||||
assert_eq!(
|
||||
*obj.get(&Value::from("true")).expect("true"),
|
||||
Value::from("bool")
|
||||
);
|
||||
}
|
||||
_ => panic!("Expected object in enum values"),
|
||||
},
|
||||
@@ -1802,18 +1805,21 @@ fn test_deserialize_enum_values_with_deeply_nested_structures() {
|
||||
match s.as_type() {
|
||||
Type::Enum { values, .. } => match &values[0] {
|
||||
Value::Object(obj) => {
|
||||
let a = &obj[&Value::from("a")];
|
||||
let a = obj.get(&Value::from("a")).expect("a");
|
||||
match a {
|
||||
Value::Array(arr) => match &arr[0] {
|
||||
Value::Object(inner) => {
|
||||
let b = &inner[&Value::from("b")];
|
||||
let b = inner.get(&Value::from("b")).expect("b");
|
||||
match b {
|
||||
Value::Array(barr) => {
|
||||
assert_eq!(barr[0], Value::from(1));
|
||||
assert_eq!(barr[1], Value::from(2));
|
||||
match &barr[2] {
|
||||
Value::Object(cobj) => {
|
||||
assert_eq!(cobj[&Value::from("c")], Value::Null);
|
||||
assert_eq!(
|
||||
*cobj.get(&Value::from("c")).expect("c"),
|
||||
Value::Null
|
||||
);
|
||||
}
|
||||
_ => panic!("Expected object for 'c'"),
|
||||
}
|
||||
@@ -1886,8 +1892,11 @@ fn test_deserialize_const_value_object() {
|
||||
match s.as_type() {
|
||||
Type::Const { value, .. } => match value {
|
||||
Value::Object(ref obj) => {
|
||||
assert_eq!(obj[&Value::from("foo")], Value::from("bar"));
|
||||
assert_eq!(obj[&Value::from("baz")], Value::from(1));
|
||||
assert_eq!(
|
||||
*obj.get(&Value::from("foo")).expect("foo"),
|
||||
Value::from("bar")
|
||||
);
|
||||
assert_eq!(*obj.get(&Value::from("baz")).expect("baz"), Value::from(1));
|
||||
}
|
||||
_ => panic!("Expected object for const value"),
|
||||
},
|
||||
@@ -1940,13 +1949,13 @@ fn test_deserialize_const_value_deeply_nested() {
|
||||
match s.as_type() {
|
||||
Type::Const { value, .. } => match value {
|
||||
Value::Object(ref obj) => {
|
||||
let a = &obj[&Value::from("a")];
|
||||
let a = obj.get(&Value::from("a")).expect("a");
|
||||
match a {
|
||||
Value::Array(arr) => {
|
||||
assert_eq!(arr[0], Value::from(1));
|
||||
match &arr[1] {
|
||||
Value::Object(inner) => {
|
||||
let b = &inner[&Value::from("b")];
|
||||
let b = inner.get(&Value::from("b")).expect("b");
|
||||
match b {
|
||||
Value::Array(barr) => {
|
||||
assert_eq!(barr[0], Value::Null);
|
||||
|
||||
@@ -7,6 +7,7 @@
|
||||
|
||||
use crate::{
|
||||
schema::{error::ValidationError, Schema, Type},
|
||||
value::Object,
|
||||
*,
|
||||
};
|
||||
use alloc::collections::BTreeMap;
|
||||
@@ -537,7 +538,7 @@ impl SchemaValidator {
|
||||
}
|
||||
|
||||
fn validate_discriminated_subobject_with_base(
|
||||
object_value: &BTreeMap<Value, Value>,
|
||||
object_value: &Object,
|
||||
discriminated_subobject: &crate::schema::DiscriminatedSubobject,
|
||||
base_properties: &BTreeMap<String, Schema>,
|
||||
base_additional_properties: Option<&Schema>,
|
||||
@@ -653,7 +654,7 @@ impl SchemaValidator {
|
||||
}
|
||||
|
||||
fn validate_subobject(
|
||||
object_value: &BTreeMap<Value, Value>,
|
||||
object_value: &Object,
|
||||
subobject: &crate::schema::Subobject,
|
||||
path: &str,
|
||||
) -> Result<(), ValidationError> {
|
||||
|
||||
@@ -88,7 +88,7 @@ fn analyze_file(regos: &[String], expected_scopes: &[Scope]) -> Result<()> {
|
||||
}
|
||||
}
|
||||
|
||||
scopes.sort_by(|a, b| a.0.span.line.cmp(&b.0.span.line));
|
||||
scopes.sort_by_key(|a| a.0.span.line);
|
||||
for (idx, (_, scope)) in scopes.iter().enumerate() {
|
||||
if idx > expected_scopes.len() {
|
||||
bail!("extra scope generated.")
|
||||
|
||||
@@ -11,6 +11,18 @@
|
||||
clippy::as_conversions
|
||||
)] // value helpers index paths directly for performance
|
||||
|
||||
mod object;
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests;
|
||||
|
||||
#[allow(unused_imports)] // surface for downstream PRs
|
||||
pub use object::{IntoIter, Iter, IterMut, Object};
|
||||
|
||||
#[cfg(feature = "rvm")]
|
||||
#[allow(unused_imports)] // surface for downstream PRs
|
||||
pub use object::ObjectCursor;
|
||||
|
||||
use crate::number::Number;
|
||||
|
||||
use alloc::collections::{BTreeMap, BTreeSet};
|
||||
@@ -23,7 +35,7 @@ use core::str::FromStr;
|
||||
|
||||
use anyhow::{anyhow, bail, Result};
|
||||
use serde::de::{self, Deserializer, Error as DeError, MapAccess, SeqAccess, Visitor};
|
||||
use serde::ser::{SerializeMap, Serializer};
|
||||
use serde::ser::Serializer;
|
||||
use serde::{Deserialize, Serialize};
|
||||
|
||||
use crate::*;
|
||||
@@ -63,7 +75,7 @@ pub enum Value {
|
||||
|
||||
/// An object.
|
||||
/// Unlike JSON, keys can be any value, not just string.
|
||||
Object(Rc<BTreeMap<Value, Value>>),
|
||||
Object(Rc<Object>),
|
||||
|
||||
/// Undefined value.
|
||||
/// Used to indicate the absence of a value.
|
||||
@@ -86,26 +98,15 @@ impl Serialize for Value {
|
||||
where
|
||||
S: Serializer,
|
||||
{
|
||||
use serde::ser::Error;
|
||||
match self {
|
||||
Value::Null => serializer.serialize_unit(),
|
||||
Value::Bool(b) => serializer.serialize_bool(*b),
|
||||
Value::String(s) => serializer.serialize_str(s.as_ref()),
|
||||
Value::Number(n) => n.serialize(serializer),
|
||||
Value::Array(a) => a.serialize(serializer),
|
||||
Value::Object(fields) => {
|
||||
let mut map = serializer.serialize_map(Some(fields.len()))?;
|
||||
for (k, v) in fields.iter() {
|
||||
match k {
|
||||
Value::String(_) => map.serialize_entry(k, v)?,
|
||||
_ => {
|
||||
let key_str = serde_json::to_string(k).map_err(Error::custom)?;
|
||||
map.serialize_entry(&key_str, v)?
|
||||
}
|
||||
}
|
||||
}
|
||||
map.end()
|
||||
}
|
||||
// Delegate to the Object/Set serializers — single canonical path,
|
||||
// handles non-string-key stringification internally.
|
||||
Value::Object(fields) => fields.serialize(serializer),
|
||||
|
||||
// display set as an array
|
||||
Value::Set(s) => s.serialize(serializer),
|
||||
@@ -345,7 +346,7 @@ impl Value {
|
||||
/// assert_eq!(array[4], Value::from(12345u64));
|
||||
/// let obj = array[5].as_object().expect("not an object");
|
||||
/// assert_eq!(obj.len(), 1);
|
||||
/// assert_eq!(obj[&Value::from("name")], Value::from("regorus"));
|
||||
/// assert_eq!(obj.get(&Value::from("name")).expect("missing name"), &Value::from("regorus"));
|
||||
/// # Ok(())
|
||||
/// # }
|
||||
/// ```
|
||||
@@ -800,7 +801,7 @@ impl From<BTreeMap<Value, Value>> for Value {
|
||||
/// # Ok(())
|
||||
/// # }
|
||||
fn from(s: BTreeMap<Value, Value>) -> Self {
|
||||
Value::Object(Rc::new(s))
|
||||
Value::Object(Rc::new(Object::from(s)))
|
||||
}
|
||||
}
|
||||
|
||||
@@ -1279,16 +1280,16 @@ impl Value {
|
||||
}
|
||||
}
|
||||
|
||||
/// Cast value to [`& BTreeMap<Value, Value>`] if [`Value::Object`].
|
||||
/// Cast value to [`&Object`] if [`Value::Object`].
|
||||
/// ```
|
||||
/// # use regorus::*;
|
||||
/// # use std::collections::BTreeMap;
|
||||
/// # use regorus::value::Object;
|
||||
/// # fn main() -> anyhow::Result<()> {
|
||||
/// let v = Value::from(
|
||||
/// [(Value::from("Hello"), Value::from("World"))]
|
||||
/// .iter()
|
||||
/// .cloned()
|
||||
/// .collect::<BTreeMap<Value, Value>>(),
|
||||
/// .collect::<Object>(),
|
||||
/// );
|
||||
/// assert_eq!(
|
||||
/// v.as_object()?.iter().next(),
|
||||
@@ -1296,28 +1297,28 @@ impl Value {
|
||||
/// );
|
||||
/// # Ok(())
|
||||
/// # }
|
||||
pub fn as_object(&self) -> Result<&BTreeMap<Value, Value>> {
|
||||
pub fn as_object(&self) -> Result<&Object> {
|
||||
match self {
|
||||
Value::Object(m) => Ok(m),
|
||||
_ => Err(anyhow!("not an object")),
|
||||
}
|
||||
}
|
||||
|
||||
/// Cast value to [`&mut BTreeMap<Value, Value>`] if [`Value::Object`].
|
||||
/// Cast value to [`&mut Object`] if [`Value::Object`].
|
||||
/// ```
|
||||
/// # use regorus::*;
|
||||
/// # use std::collections::BTreeMap;
|
||||
/// # use regorus::value::Object;
|
||||
/// # fn main() -> anyhow::Result<()> {
|
||||
/// let mut v = Value::from(
|
||||
/// [(Value::from("Hello"), Value::from("World"))]
|
||||
/// .iter()
|
||||
/// .cloned()
|
||||
/// .collect::<BTreeMap<Value, Value>>(),
|
||||
/// .collect::<Object>(),
|
||||
/// );
|
||||
/// v.as_object_mut()?.insert(Value::from("Good"), Value::from("Bye"));
|
||||
/// # Ok(())
|
||||
/// # }
|
||||
pub fn as_object_mut(&mut self) -> Result<&mut BTreeMap<Value, Value>> {
|
||||
pub fn as_object_mut(&mut self) -> Result<&mut Object> {
|
||||
match self {
|
||||
Value::Object(m) => Ok(Rc::make_mut(m)),
|
||||
_ => Err(anyhow!("not an object")),
|
||||
148
src/value/object/iter.rs
Normal file
148
src/value/object/iter.rs
Normal file
@@ -0,0 +1,148 @@
|
||||
// Copyright (c) Microsoft Corporation.
|
||||
// Licensed under the MIT License.
|
||||
|
||||
//! Opaque iterator types for [`Object`].
|
||||
//!
|
||||
//! These newtypes wrap the storage backend's iterators so the backend can be
|
||||
//! swapped without changing any iterator type signatures observed by callers.
|
||||
|
||||
use alloc::collections::btree_map;
|
||||
use core::iter::FusedIterator;
|
||||
|
||||
use super::Object;
|
||||
use crate::value::Value;
|
||||
|
||||
/// Owned iterator over `(Value, Value)` entries.
|
||||
#[derive(Debug)]
|
||||
pub struct IntoIter {
|
||||
pub(super) inner: btree_map::IntoIter<Value, Value>,
|
||||
}
|
||||
|
||||
impl Iterator for IntoIter {
|
||||
type Item = (Value, Value);
|
||||
#[inline]
|
||||
fn next(&mut self) -> Option<Self::Item> {
|
||||
self.inner.next()
|
||||
}
|
||||
#[inline]
|
||||
fn size_hint(&self) -> (usize, Option<usize>) {
|
||||
self.inner.size_hint()
|
||||
}
|
||||
}
|
||||
|
||||
impl DoubleEndedIterator for IntoIter {
|
||||
#[inline]
|
||||
fn next_back(&mut self) -> Option<Self::Item> {
|
||||
self.inner.next_back()
|
||||
}
|
||||
}
|
||||
|
||||
impl ExactSizeIterator for IntoIter {
|
||||
#[inline]
|
||||
fn len(&self) -> usize {
|
||||
self.inner.len()
|
||||
}
|
||||
}
|
||||
|
||||
impl FusedIterator for IntoIter {}
|
||||
|
||||
/// Borrowed iterator over `(&Value, &Value)` entries.
|
||||
#[derive(Debug, Clone)]
|
||||
pub struct Iter<'a> {
|
||||
pub(super) inner: btree_map::Iter<'a, Value, Value>,
|
||||
}
|
||||
|
||||
impl<'a> Iterator for Iter<'a> {
|
||||
type Item = (&'a Value, &'a Value);
|
||||
#[inline]
|
||||
fn next(&mut self) -> Option<Self::Item> {
|
||||
self.inner.next()
|
||||
}
|
||||
#[inline]
|
||||
fn size_hint(&self) -> (usize, Option<usize>) {
|
||||
self.inner.size_hint()
|
||||
}
|
||||
}
|
||||
|
||||
impl<'a> DoubleEndedIterator for Iter<'a> {
|
||||
#[inline]
|
||||
fn next_back(&mut self) -> Option<Self::Item> {
|
||||
self.inner.next_back()
|
||||
}
|
||||
}
|
||||
|
||||
impl<'a> ExactSizeIterator for Iter<'a> {
|
||||
#[inline]
|
||||
fn len(&self) -> usize {
|
||||
self.inner.len()
|
||||
}
|
||||
}
|
||||
|
||||
impl<'a> FusedIterator for Iter<'a> {}
|
||||
|
||||
/// Borrowed iterator over `(&Value, &mut Value)` entries.
|
||||
#[derive(Debug)]
|
||||
pub struct IterMut<'a> {
|
||||
pub(super) inner: btree_map::IterMut<'a, Value, Value>,
|
||||
}
|
||||
|
||||
impl<'a> Iterator for IterMut<'a> {
|
||||
type Item = (&'a Value, &'a mut Value);
|
||||
#[inline]
|
||||
fn next(&mut self) -> Option<Self::Item> {
|
||||
self.inner.next()
|
||||
}
|
||||
#[inline]
|
||||
fn size_hint(&self) -> (usize, Option<usize>) {
|
||||
self.inner.size_hint()
|
||||
}
|
||||
}
|
||||
|
||||
impl<'a> DoubleEndedIterator for IterMut<'a> {
|
||||
#[inline]
|
||||
fn next_back(&mut self) -> Option<Self::Item> {
|
||||
self.inner.next_back()
|
||||
}
|
||||
}
|
||||
|
||||
impl<'a> ExactSizeIterator for IterMut<'a> {
|
||||
#[inline]
|
||||
fn len(&self) -> usize {
|
||||
self.inner.len()
|
||||
}
|
||||
}
|
||||
|
||||
impl<'a> FusedIterator for IterMut<'a> {}
|
||||
|
||||
impl IntoIterator for Object {
|
||||
type Item = (Value, Value);
|
||||
type IntoIter = IntoIter;
|
||||
#[inline]
|
||||
fn into_iter(self) -> Self::IntoIter {
|
||||
IntoIter {
|
||||
inner: self.inner.into_iter(),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
impl<'a> IntoIterator for &'a Object {
|
||||
type Item = (&'a Value, &'a Value);
|
||||
type IntoIter = Iter<'a>;
|
||||
#[inline]
|
||||
fn into_iter(self) -> Self::IntoIter {
|
||||
Iter {
|
||||
inner: self.inner.iter(),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
impl<'a> IntoIterator for &'a mut Object {
|
||||
type Item = (&'a Value, &'a mut Value);
|
||||
type IntoIter = IterMut<'a>;
|
||||
#[inline]
|
||||
fn into_iter(self) -> Self::IntoIter {
|
||||
IterMut {
|
||||
inner: self.inner.iter_mut(),
|
||||
}
|
||||
}
|
||||
}
|
||||
265
src/value/object/mod.rs
Normal file
265
src/value/object/mod.rs
Normal file
@@ -0,0 +1,265 @@
|
||||
// Copyright (c) Microsoft Corporation.
|
||||
// Licensed under the MIT License.
|
||||
|
||||
//! See [`Object`].
|
||||
|
||||
mod iter;
|
||||
mod serde;
|
||||
|
||||
use alloc::collections::BTreeMap;
|
||||
use core::cmp::Ordering;
|
||||
use core::fmt;
|
||||
use core::ops::Bound;
|
||||
|
||||
use crate::value::Value;
|
||||
|
||||
pub use iter::{IntoIter, Iter, IterMut};
|
||||
|
||||
/// Opaque, ordered key-value map keyed by [`Value`].
|
||||
///
|
||||
/// The current backing storage is `BTreeMap<Value, Value>`. The inner field
|
||||
/// is private so the representation can change (two-tier inline+hash, lazy,
|
||||
/// schema-shared) without touching call sites.
|
||||
///
|
||||
/// # Iteration
|
||||
///
|
||||
/// - [`Object::iter`] — implementation-defined order; non-resumable.
|
||||
/// - [`Object::iter_sorted`] — sorted by `Value::Ord`; non-resumable.
|
||||
/// - [`Object::cursor`] / [`Object::next`] — implementation-defined order,
|
||||
/// resumable; cheapest per-step cost. Used by interpreter/RVM when iteration
|
||||
/// must yield mid-flight.
|
||||
#[derive(Default, Clone, Eq, PartialEq)]
|
||||
pub struct Object {
|
||||
inner: BTreeMap<Value, Value>,
|
||||
}
|
||||
|
||||
impl Object {
|
||||
/// Create an empty `Object`.
|
||||
#[inline]
|
||||
pub const fn new() -> Self {
|
||||
Self {
|
||||
inner: BTreeMap::new(),
|
||||
}
|
||||
}
|
||||
|
||||
#[inline]
|
||||
pub fn len(&self) -> usize {
|
||||
self.inner.len()
|
||||
}
|
||||
|
||||
#[inline]
|
||||
pub fn is_empty(&self) -> bool {
|
||||
self.inner.is_empty()
|
||||
}
|
||||
|
||||
#[inline]
|
||||
pub fn get(&self, key: &Value) -> Option<&Value> {
|
||||
self.inner.get(key)
|
||||
}
|
||||
|
||||
#[inline]
|
||||
pub fn contains_key(&self, key: &Value) -> bool {
|
||||
self.inner.contains_key(key)
|
||||
}
|
||||
|
||||
#[inline]
|
||||
pub fn get_mut(&mut self, key: &Value) -> Option<&mut Value> {
|
||||
self.inner.get_mut(key)
|
||||
}
|
||||
|
||||
/// Iteration in implementation-defined order. Non-resumable.
|
||||
///
|
||||
/// For the current BTree-backed storage this happens to be sorted, but
|
||||
/// callers MUST NOT depend on that. Use [`Object::iter_sorted`] when
|
||||
/// deterministic order is required, or [`Object::cursor`] when iteration
|
||||
/// must yield and resume.
|
||||
#[inline]
|
||||
pub fn iter(&self) -> impl Iterator<Item = (&Value, &Value)> + '_ {
|
||||
self.inner.iter()
|
||||
}
|
||||
|
||||
/// Iteration in sorted key order (by `Value::Ord`). Non-resumable.
|
||||
///
|
||||
/// Use this for serialization, snapshots, hashing, `Debug`, the
|
||||
/// `object.keys` builtin, etc.
|
||||
#[inline]
|
||||
pub fn iter_sorted(&self) -> Iter<'_> {
|
||||
// BTree backend iterates sorted natively.
|
||||
Iter {
|
||||
inner: self.inner.iter(),
|
||||
}
|
||||
}
|
||||
|
||||
#[inline]
|
||||
pub fn keys(&self) -> impl Iterator<Item = &Value> + '_ {
|
||||
self.inner.keys()
|
||||
}
|
||||
|
||||
/// Keys in sorted order (by `Value::Ord`). Symmetric with
|
||||
/// [`Object::iter_sorted`].
|
||||
#[inline]
|
||||
pub fn keys_sorted(&self) -> impl Iterator<Item = &Value> + '_ {
|
||||
self.iter_sorted().map(|(k, _)| k)
|
||||
}
|
||||
|
||||
#[inline]
|
||||
pub fn values(&self) -> impl Iterator<Item = &Value> + '_ {
|
||||
self.inner.values()
|
||||
}
|
||||
|
||||
#[inline]
|
||||
pub fn iter_mut(&mut self) -> IterMut<'_> {
|
||||
IterMut {
|
||||
inner: self.inner.iter_mut(),
|
||||
}
|
||||
}
|
||||
|
||||
/// Insert a key-value pair. Returns the previous value if any.
|
||||
#[inline]
|
||||
pub fn insert(&mut self, key: Value, value: Value) -> Option<Value> {
|
||||
self.inner.insert(key, value)
|
||||
}
|
||||
|
||||
#[inline]
|
||||
pub fn remove(&mut self, key: &Value) -> Option<Value> {
|
||||
self.inner.remove(key)
|
||||
}
|
||||
|
||||
#[inline]
|
||||
pub fn retain<F>(&mut self, f: F)
|
||||
where
|
||||
F: FnMut(&Value, &mut Value) -> bool,
|
||||
{
|
||||
self.inner.retain(f);
|
||||
}
|
||||
|
||||
#[inline]
|
||||
pub fn clear(&mut self) {
|
||||
self.inner.clear();
|
||||
}
|
||||
|
||||
#[inline]
|
||||
pub fn append(&mut self, other: &mut Object) {
|
||||
self.inner.append(&mut other.inner);
|
||||
}
|
||||
|
||||
/// Gets a mutable reference to the value associated with `key`, inserting
|
||||
/// the result of `default()` if absent. Single O(log n) probe.
|
||||
pub fn get_or_insert_with<F: FnOnce() -> Value>(
|
||||
&mut self,
|
||||
key: Value,
|
||||
default: F,
|
||||
) -> &mut Value {
|
||||
self.inner.entry(key).or_insert_with(default)
|
||||
}
|
||||
|
||||
/// Wrap into a `Value::Object`.
|
||||
#[inline]
|
||||
pub fn into_value(self) -> Value {
|
||||
Value::Object(crate::Rc::new(self))
|
||||
}
|
||||
|
||||
/// Create a resumable cursor over entries in implementation-defined
|
||||
/// order. Stable for the lifetime of `&self`. O(1).
|
||||
///
|
||||
/// The cursor is fully self-owned (it stores a clone of the last-seen
|
||||
/// key, not a reference) so it can be stored as a field of a
|
||||
/// long-lived state struct — e.g. an RVM iteration frame that persists
|
||||
/// across instruction dispatches. As a consequence, mutating the
|
||||
/// `Object` between `next()` calls is not rejected by the borrow
|
||||
/// checker; the resulting iteration order in that case is unspecified.
|
||||
#[inline]
|
||||
pub const fn cursor(&self) -> ObjectCursor {
|
||||
ObjectCursor {
|
||||
inner: ObjectCursorInner::BTree(None),
|
||||
}
|
||||
}
|
||||
|
||||
/// Advance `cursor` and yield the next entry. O(log n) for the BTree
|
||||
/// backend (range probe); future hash/inline variants may be O(1).
|
||||
pub fn next<'a>(&'a self, cursor: &mut ObjectCursor) -> Option<(&'a Value, &'a Value)> {
|
||||
let ObjectCursorInner::BTree(ref mut last) = cursor.inner;
|
||||
let next = last.as_ref().map_or_else(
|
||||
|| self.inner.iter().next(),
|
||||
|prev| {
|
||||
self.inner
|
||||
.range((Bound::Excluded(prev.clone()), Bound::Unbounded))
|
||||
.next()
|
||||
},
|
||||
);
|
||||
let (k, v) = next?;
|
||||
*last = Some(k.clone());
|
||||
Some((k, v))
|
||||
}
|
||||
}
|
||||
|
||||
/// Opaque resumable cursor over an [`Object`]'s entries in
|
||||
/// implementation-defined order.
|
||||
///
|
||||
/// Self-owned: holds no borrow on the `Object`, so it can be stored as a
|
||||
/// field of a long-lived state struct (e.g. an RVM iteration frame).
|
||||
#[derive(Debug, Clone)]
|
||||
pub struct ObjectCursor {
|
||||
inner: ObjectCursorInner,
|
||||
}
|
||||
|
||||
#[derive(Debug, Clone)]
|
||||
enum ObjectCursorInner {
|
||||
/// BTree backend cursor: tracks last-seen key. `None` means "before start".
|
||||
BTree(Option<Value>),
|
||||
}
|
||||
|
||||
// ---- Hand-written Ord/PartialOrd ----------------------------------------
|
||||
//
|
||||
// Implemented in terms of `iter_sorted()` so ordering is consistent with the
|
||||
// canonical (sorted) view of the entries and is therefore independent of
|
||||
// the storage variant.
|
||||
|
||||
impl Ord for Object {
|
||||
fn cmp(&self, other: &Self) -> Ordering {
|
||||
self.iter_sorted().cmp(other.iter_sorted())
|
||||
}
|
||||
}
|
||||
|
||||
impl PartialOrd for Object {
|
||||
#[inline]
|
||||
fn partial_cmp(&self, other: &Self) -> Option<Ordering> {
|
||||
Some(self.cmp(other))
|
||||
}
|
||||
}
|
||||
|
||||
impl fmt::Debug for Object {
|
||||
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
|
||||
// Use sorted iteration so Debug output is stable across storage
|
||||
// variants.
|
||||
f.debug_map().entries(self.iter_sorted()).finish()
|
||||
}
|
||||
}
|
||||
|
||||
impl Extend<(Value, Value)> for Object {
|
||||
fn extend<I: IntoIterator<Item = (Value, Value)>>(&mut self, iter: I) {
|
||||
self.inner.extend(iter);
|
||||
}
|
||||
}
|
||||
|
||||
impl FromIterator<(Value, Value)> for Object {
|
||||
fn from_iter<I: IntoIterator<Item = (Value, Value)>>(iter: I) -> Self {
|
||||
Self {
|
||||
inner: BTreeMap::from_iter(iter),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
impl From<BTreeMap<Value, Value>> for Object {
|
||||
#[inline]
|
||||
fn from(map: BTreeMap<Value, Value>) -> Self {
|
||||
Self { inner: map }
|
||||
}
|
||||
}
|
||||
|
||||
impl From<Object> for Value {
|
||||
#[inline]
|
||||
fn from(o: Object) -> Self {
|
||||
o.into_value()
|
||||
}
|
||||
}
|
||||
59
src/value/object/serde.rs
Normal file
59
src/value/object/serde.rs
Normal file
@@ -0,0 +1,59 @@
|
||||
// Copyright (c) Microsoft Corporation.
|
||||
// Licensed under the MIT License.
|
||||
|
||||
//! Serde `Serialize`/`Deserialize` impls for [`Object`].
|
||||
|
||||
use alloc::string::ToString as _;
|
||||
use core::fmt;
|
||||
|
||||
use serde::de::{Deserialize, Deserializer, Error as _, MapAccess, Visitor};
|
||||
use serde::ser::{Serialize, SerializeMap as _, Serializer};
|
||||
|
||||
use super::Object;
|
||||
use crate::value::Value;
|
||||
|
||||
impl Serialize for Object {
|
||||
fn serialize<S: Serializer>(&self, serializer: S) -> Result<S::Ok, S::Error> {
|
||||
use serde::ser::Error;
|
||||
let mut map = serializer.serialize_map(Some(self.len()))?;
|
||||
// Sorted iteration: canonical JSON.
|
||||
for (k, v) in self.iter_sorted() {
|
||||
match *k {
|
||||
Value::String(_) => map.serialize_entry(k, v)?,
|
||||
_ => {
|
||||
// Non-string keys are stringified via serde_json::to_string
|
||||
// so the resulting JSON has valid string keys.
|
||||
let key_str = serde_json::to_string(k).map_err(Error::custom)?;
|
||||
map.serialize_entry(&key_str, v)?;
|
||||
}
|
||||
}
|
||||
}
|
||||
map.end()
|
||||
}
|
||||
}
|
||||
|
||||
struct ObjectVisitor;
|
||||
|
||||
impl<'de> Visitor<'de> for ObjectVisitor {
|
||||
type Value = Object;
|
||||
|
||||
fn expecting(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
|
||||
f.write_str("a map of Value to Value")
|
||||
}
|
||||
|
||||
fn visit_map<A: MapAccess<'de>>(self, mut access: A) -> Result<Self::Value, A::Error> {
|
||||
let mut obj = Object::new();
|
||||
while let Some((k, v)) = access.next_entry::<Value, Value>()? {
|
||||
obj.insert(k, v);
|
||||
crate::utils::limits::check_memory_limit_if_needed()
|
||||
.map_err(|err| A::Error::custom(err.to_string()))?;
|
||||
}
|
||||
Ok(obj)
|
||||
}
|
||||
}
|
||||
|
||||
impl<'de> Deserialize<'de> for Object {
|
||||
fn deserialize<D: Deserializer<'de>>(deserializer: D) -> Result<Self, D::Error> {
|
||||
deserializer.deserialize_map(ObjectVisitor)
|
||||
}
|
||||
}
|
||||
564
src/value/tests.rs
Normal file
564
src/value/tests.rs
Normal file
@@ -0,0 +1,564 @@
|
||||
// Copyright (c) Microsoft Corporation.
|
||||
// Licensed under the MIT License.
|
||||
|
||||
#![allow(
|
||||
clippy::expect_used,
|
||||
clippy::unwrap_used,
|
||||
clippy::indexing_slicing,
|
||||
clippy::as_conversions,
|
||||
clippy::arithmetic_side_effects,
|
||||
clippy::unseparated_literal_suffix,
|
||||
clippy::map_unwrap_or,
|
||||
clippy::option_if_let_else,
|
||||
clippy::pattern_type_mismatch
|
||||
)]
|
||||
|
||||
use alloc::collections::BTreeMap;
|
||||
use alloc::format;
|
||||
use alloc::vec::Vec;
|
||||
|
||||
use super::Object;
|
||||
use crate::value::Value;
|
||||
|
||||
fn val(i: u64) -> Value {
|
||||
Value::from(i)
|
||||
}
|
||||
|
||||
fn make_pairs(n: u64) -> Vec<(Value, Value)> {
|
||||
(0..n).map(|i| (val(i), val(i.saturating_mul(2)))).collect()
|
||||
}
|
||||
|
||||
const SIZES: &[u64] = &[0, 1, 2, 4, 8, 64, 256, 1024];
|
||||
|
||||
/// `iter_sorted` must yield entries in the same order as a `BTreeMap` oracle.
|
||||
#[test]
|
||||
fn object_iter_sorted_matches_btreemap_oracle() {
|
||||
for &n in SIZES {
|
||||
let pairs = make_pairs(n);
|
||||
let oracle: BTreeMap<Value, Value> = pairs.iter().cloned().collect();
|
||||
let obj: Object = pairs.into_iter().collect();
|
||||
let actual: Vec<(&Value, &Value)> = obj.iter_sorted().collect();
|
||||
let expected: Vec<(&Value, &Value)> = oracle.iter().collect();
|
||||
assert_eq!(actual, expected, "size {n}");
|
||||
}
|
||||
}
|
||||
|
||||
/// `iter` may be in any order, but as a multiset must equal the oracle's entries.
|
||||
#[test]
|
||||
fn object_iter_multiset_equality_with_oracle() {
|
||||
for &n in SIZES {
|
||||
let pairs = make_pairs(n);
|
||||
let oracle: BTreeMap<Value, Value> = pairs.iter().cloned().collect();
|
||||
let obj: Object = pairs.into_iter().collect();
|
||||
assert_eq!(obj.len(), oracle.len(), "size {n}");
|
||||
let mut a: Vec<(Value, Value)> = obj.iter().map(|(k, v)| (k.clone(), v.clone())).collect();
|
||||
let mut b: Vec<(Value, Value)> =
|
||||
oracle.iter().map(|(k, v)| (k.clone(), v.clone())).collect();
|
||||
a.sort();
|
||||
b.sort();
|
||||
assert_eq!(a, b);
|
||||
}
|
||||
}
|
||||
|
||||
/// Serialize-then-deserialize must round-trip through JSON without loss.
|
||||
#[test]
|
||||
fn object_serde_roundtrip() {
|
||||
for &n in &[0_u64, 1, 8, 64] {
|
||||
let pairs: Vec<(Value, Value)> = (0..n)
|
||||
.map(|i| (Value::String(format!("k{i}").into()), val(i)))
|
||||
.collect();
|
||||
let obj: Object = pairs.into_iter().collect();
|
||||
let json = serde_json::to_string(&obj).expect("ser");
|
||||
let back: Object = serde_json::from_str(&json).expect("de");
|
||||
assert_eq!(obj, back, "size {n}");
|
||||
}
|
||||
}
|
||||
|
||||
/// Equality depends only on contents, not the order keys were inserted.
|
||||
#[test]
|
||||
fn object_eq_invariant_to_insertion_order() {
|
||||
let mut a = Object::new();
|
||||
let mut b = Object::new();
|
||||
for i in 0..32_u64 {
|
||||
a.insert(val(i), val(i.saturating_add(1)));
|
||||
}
|
||||
for i in (0..32_u64).rev() {
|
||||
b.insert(val(i), val(i.saturating_add(1)));
|
||||
}
|
||||
assert_eq!(a, b);
|
||||
}
|
||||
|
||||
/// `remove` returns the prior value (or `None`) and `retain` keeps only matching entries.
|
||||
#[test]
|
||||
fn object_remove_and_retain() {
|
||||
let mut obj: Object = make_pairs(16).into_iter().collect();
|
||||
assert_eq!(obj.remove(&val(0)), Some(val(0)));
|
||||
assert!(obj.remove(&val(100)).is_none());
|
||||
obj.retain(|_, v| {
|
||||
if let Value::Number(ref n) = *v {
|
||||
n.as_u64().is_some_and(|x| x % 4 == 0)
|
||||
} else {
|
||||
false
|
||||
}
|
||||
});
|
||||
for (_, v) in obj.iter_sorted() {
|
||||
if let Value::Number(ref n) = *v {
|
||||
assert_eq!(n.as_u64().expect("u64") % 4, 0);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/// `IntoIterator` for `Object` (by value) yields every entry exactly once.
|
||||
#[test]
|
||||
fn object_into_iterator_owned() {
|
||||
let obj: Object = make_pairs(8).into_iter().collect();
|
||||
let collected: Vec<(Value, Value)> = obj.into_iter().collect();
|
||||
assert_eq!(collected.len(), 8);
|
||||
}
|
||||
|
||||
// ---- Duplicate-key semantics --------------------------------------------
|
||||
|
||||
/// `FromIterator` keeps the last value when the same key appears multiple times.
|
||||
#[test]
|
||||
fn object_from_iter_last_wins_on_duplicate_keys() {
|
||||
let obj = Object::from_iter([(val(0), val(1)), (val(0), val(2))]);
|
||||
assert_eq!(obj.get(&val(0)), Some(&val(2)));
|
||||
assert_eq!(obj.len(), 1);
|
||||
}
|
||||
|
||||
/// `From<BTreeMap>` adopts `BTreeMap`'s own last-write-wins semantics for duplicates.
|
||||
#[test]
|
||||
fn object_from_btreemap_last_wins_on_duplicate_keys() {
|
||||
let mut bm: BTreeMap<Value, Value> = BTreeMap::new();
|
||||
bm.insert(val(0), val(1));
|
||||
bm.insert(val(0), val(2));
|
||||
let obj: Object = bm.into();
|
||||
assert_eq!(obj.get(&val(0)), Some(&val(2)));
|
||||
assert_eq!(obj.len(), 1);
|
||||
}
|
||||
|
||||
// ---- get_or_insert_with --------------------------------------------------
|
||||
|
||||
/// `get_or_insert_with` inserts the default when the key is absent and returns a mutable ref to it.
|
||||
#[test]
|
||||
fn object_get_or_insert_with_inserts_when_absent() {
|
||||
let mut obj = Object::new();
|
||||
let v = obj.get_or_insert_with(val(7), || val(42));
|
||||
assert_eq!(*v, val(42));
|
||||
*v = val(43);
|
||||
assert_eq!(obj.get(&val(7)), Some(&val(43)));
|
||||
}
|
||||
|
||||
/// `get_or_insert_with` returns the existing value and never invokes the default closure.
|
||||
#[test]
|
||||
fn object_get_or_insert_with_returns_existing_when_present() {
|
||||
let mut obj = Object::new();
|
||||
obj.insert(val(7), val(1));
|
||||
let mut closure_called = false;
|
||||
let v = obj.get_or_insert_with(val(7), || {
|
||||
closure_called = true;
|
||||
val(999)
|
||||
});
|
||||
assert_eq!(*v, val(1));
|
||||
assert!(!closure_called, "default closure must not run when present");
|
||||
}
|
||||
|
||||
// ---- Accessor coverage ---------------------------------------------------
|
||||
|
||||
/// Smoke-test every accessor: `contains_key`/`get`/`get_mut`/`keys`/`values`/`iter`/`iter_mut`/`append`/`clear`.
|
||||
#[test]
|
||||
fn object_accessor_coverage() {
|
||||
let mut obj: Object = make_pairs(4).into_iter().collect();
|
||||
|
||||
assert!(obj.contains_key(&val(0)));
|
||||
assert!(!obj.contains_key(&val(100)));
|
||||
|
||||
assert_eq!(obj.get(&val(2)), Some(&val(4)));
|
||||
|
||||
if let Some(v) = obj.get_mut(&val(1)) {
|
||||
*v = val(999);
|
||||
}
|
||||
assert_eq!(obj.get(&val(1)), Some(&val(999)));
|
||||
|
||||
let keys: Vec<&Value> = obj.keys().collect();
|
||||
assert_eq!(keys.len(), 4);
|
||||
let values: Vec<&Value> = obj.values().collect();
|
||||
assert_eq!(values.len(), 4);
|
||||
|
||||
for (_, v) in obj.iter_mut() {
|
||||
*v = val(0);
|
||||
}
|
||||
for (_, v) in obj.iter() {
|
||||
assert_eq!(*v, val(0));
|
||||
}
|
||||
|
||||
let mut other = Object::new();
|
||||
other.insert(val(100), val(200));
|
||||
obj.append(&mut other);
|
||||
assert!(other.is_empty());
|
||||
assert!(obj.contains_key(&val(100)));
|
||||
|
||||
obj.clear();
|
||||
assert!(obj.is_empty());
|
||||
}
|
||||
|
||||
// ---- IntoIterator for references -----------------------------------------
|
||||
|
||||
/// `IntoIterator` for `&Object` yields shared refs to every entry.
|
||||
#[test]
|
||||
fn object_into_iterator_ref() {
|
||||
let obj: Object = make_pairs(4).into_iter().collect();
|
||||
let mut count = 0;
|
||||
for (_k, _v) in &obj {
|
||||
count += 1;
|
||||
}
|
||||
assert_eq!(count, 4);
|
||||
}
|
||||
|
||||
/// `IntoIterator` for `&mut Object` exposes mutable refs to values; mutations persist.
|
||||
#[test]
|
||||
fn object_into_iterator_ref_mut() {
|
||||
let mut obj: Object = make_pairs(4).into_iter().collect();
|
||||
for (_k, v) in &mut obj {
|
||||
*v = val(0);
|
||||
}
|
||||
for (_, v) in obj.iter() {
|
||||
assert_eq!(*v, val(0));
|
||||
}
|
||||
}
|
||||
|
||||
// ---- Cursor tests --------------------------------------------------------
|
||||
|
||||
/// Driving `cursor`+`next` to completion visits each entry exactly once.
|
||||
#[test]
|
||||
fn object_cursor_yields_every_entry_once() {
|
||||
for &n in SIZES {
|
||||
let pairs = make_pairs(n);
|
||||
let obj: Object = pairs.clone().into_iter().collect();
|
||||
let mut cursor = obj.cursor();
|
||||
let mut collected: Vec<(Value, Value)> = Vec::new();
|
||||
while let Some((k, v)) = obj.next(&mut cursor) {
|
||||
collected.push((k.clone(), v.clone()));
|
||||
}
|
||||
let mut a = collected;
|
||||
a.sort();
|
||||
let mut b = pairs;
|
||||
b.sort();
|
||||
assert_eq!(a, b, "size {n}");
|
||||
}
|
||||
}
|
||||
|
||||
/// A freshly-constructed cursor restarts from the beginning, independent of any prior cursor's state.
|
||||
#[test]
|
||||
fn object_cursor_resumable_fresh_cursor_restarts() {
|
||||
let obj: Object = make_pairs(8).into_iter().collect();
|
||||
let mut c1 = obj.cursor();
|
||||
let _ = obj.next(&mut c1);
|
||||
let _ = obj.next(&mut c1);
|
||||
let mut c2 = obj.cursor();
|
||||
let first_again = obj.next(&mut c2);
|
||||
let first_original = obj.iter().next();
|
||||
assert_eq!(
|
||||
first_again.map(|(k, v)| (k.clone(), v.clone())),
|
||||
first_original.map(|(k, v)| (k.clone(), v.clone()))
|
||||
);
|
||||
}
|
||||
|
||||
/// When `Object` is shared via `Rc`, `Rc::make_mut` clones — leaving an in-flight cursor on the original snapshot unaffected.
|
||||
#[test]
|
||||
fn object_cursor_snapshot_independence_via_rc() {
|
||||
use crate::Rc;
|
||||
let mut obj = Object::new();
|
||||
obj.insert(Value::from("a"), Value::from(1));
|
||||
obj.insert(Value::from("b"), Value::from(2));
|
||||
obj.insert(Value::from("c"), Value::from(3));
|
||||
let rc_obj = Rc::new(obj);
|
||||
|
||||
let alias = Rc::clone(&rc_obj);
|
||||
let mut cursor = rc_obj.cursor();
|
||||
let _ = rc_obj.next(&mut cursor);
|
||||
|
||||
let mut alias_for_mut = alias;
|
||||
Rc::make_mut(&mut alias_for_mut).insert(Value::from("d"), Value::from(4));
|
||||
Rc::make_mut(&mut alias_for_mut).remove(&Value::from("a"));
|
||||
|
||||
assert_eq!(rc_obj.len(), 3);
|
||||
let mut remaining = 0;
|
||||
while rc_obj.next(&mut cursor).is_some() {
|
||||
remaining += 1;
|
||||
}
|
||||
assert_eq!(remaining, 2);
|
||||
}
|
||||
|
||||
/// A cursor over an empty `Object` returns `None` on the first call.
|
||||
#[test]
|
||||
fn object_cursor_empty_returns_none_immediately() {
|
||||
let obj = Object::new();
|
||||
let mut cursor = obj.cursor();
|
||||
assert!(obj.next(&mut cursor).is_none());
|
||||
}
|
||||
|
||||
/// Mutating an `Object` between `next()` calls is well-defined: the cursor
|
||||
/// must not panic and must terminate. The visit order, and whether
|
||||
/// inserted/removed keys appear, is intentionally unspecified — this test
|
||||
/// only pins the safety + termination guarantees that callers (e.g. a
|
||||
/// future RVM iteration frame) may rely on. It must NOT assert any
|
||||
/// particular order or count, or future backend swaps will be forced to
|
||||
/// honor an accidental contract.
|
||||
#[test]
|
||||
fn object_cursor_mutation_between_steps_is_safe_and_terminates() {
|
||||
let mut obj: Object = make_pairs(16).into_iter().collect();
|
||||
let mut cursor = obj.cursor();
|
||||
|
||||
// Yield a few entries before mutating.
|
||||
for _ in 0..3 {
|
||||
let _ = obj.next(&mut cursor);
|
||||
}
|
||||
|
||||
// Interleave mutations and steps. Each yielded entry must, at the
|
||||
// moment of yield, be a real entry in the map.
|
||||
obj.insert(val(100), val(100));
|
||||
if let Some((k, v)) = obj.next(&mut cursor) {
|
||||
assert_eq!(obj.get(k), Some(v));
|
||||
}
|
||||
obj.remove(&val(2));
|
||||
if let Some((k, v)) = obj.next(&mut cursor) {
|
||||
assert_eq!(obj.get(k), Some(v));
|
||||
}
|
||||
obj.clear();
|
||||
// After clear(), draining the cursor must terminate (not panic, not
|
||||
// loop) within a bounded number of calls.
|
||||
let mut terminated = false;
|
||||
for _ in 0..32 {
|
||||
if obj.next(&mut cursor).is_none() {
|
||||
terminated = true;
|
||||
break;
|
||||
}
|
||||
}
|
||||
assert!(terminated, "cursor failed to terminate after clear()");
|
||||
}
|
||||
|
||||
// ---- Hand-written Ord consistency ---------------------------------------
|
||||
|
||||
/// `Ord` (built atop `iter_sorted`) is invariant to insertion order.
|
||||
#[test]
|
||||
fn object_ord_invariant_to_insertion_order() {
|
||||
let mut a = Object::new();
|
||||
let mut b = Object::new();
|
||||
for i in 0..16_u64 {
|
||||
a.insert(val(i), val(i.saturating_add(1)));
|
||||
}
|
||||
for i in (0..16_u64).rev() {
|
||||
b.insert(val(i), val(i.saturating_add(1)));
|
||||
}
|
||||
use core::cmp::Ordering;
|
||||
assert_eq!(a.cmp(&b), Ordering::Equal);
|
||||
}
|
||||
|
||||
/// `Ord` agrees with lexicographic comparison of the sorted-entries view.
|
||||
#[test]
|
||||
fn object_ord_lexicographic_on_sorted_entries() {
|
||||
let a: Object = [(val(0), val(0)), (val(1), val(1))].into_iter().collect();
|
||||
let b: Object = [(val(0), val(0)), (val(2), val(2))].into_iter().collect();
|
||||
assert!(a < b);
|
||||
}
|
||||
|
||||
/// `empty < non_empty` and a shorter prefix compares less than its extension.
|
||||
#[test]
|
||||
fn object_ord_empty_and_prefix() {
|
||||
use core::cmp::Ordering;
|
||||
let empty = Object::new();
|
||||
let one: Object = [(val(0), val(0))].into_iter().collect();
|
||||
let two: Object = [(val(0), val(0)), (val(1), val(1))].into_iter().collect();
|
||||
assert_eq!(empty.cmp(&one), Ordering::Less);
|
||||
assert_eq!(one.cmp(&two), Ordering::Less);
|
||||
assert_eq!(two.cmp(&empty), Ordering::Greater);
|
||||
}
|
||||
|
||||
/// When keys match, `Ord` falls through to comparing values.
|
||||
#[test]
|
||||
fn object_ord_breaks_ties_on_values() {
|
||||
use core::cmp::Ordering;
|
||||
let a: Object = [(val(0), val(1))].into_iter().collect();
|
||||
let b: Object = [(val(0), val(2))].into_iter().collect();
|
||||
assert_eq!(a.cmp(&b), Ordering::Less);
|
||||
}
|
||||
|
||||
/// `PartialOrd` must agree with `Ord` for every input pair.
|
||||
#[test]
|
||||
fn object_partial_cmp_matches_cmp() {
|
||||
let a: Object = [(val(0), val(0)), (val(1), val(1))].into_iter().collect();
|
||||
let b: Object = [(val(0), val(0)), (val(2), val(2))].into_iter().collect();
|
||||
assert_eq!(a.partial_cmp(&b), Some(a.cmp(&b)));
|
||||
assert_eq!(b.partial_cmp(&a), Some(b.cmp(&a)));
|
||||
assert_eq!(a.partial_cmp(&a), Some(core::cmp::Ordering::Equal));
|
||||
}
|
||||
|
||||
// ---- Debug / keys_sorted determinism ------------------------------------
|
||||
|
||||
/// `Debug` output is byte-identical for equal Objects regardless of insertion order.
|
||||
#[test]
|
||||
fn object_debug_invariant_to_insertion_order() {
|
||||
let mut a = Object::new();
|
||||
let mut b = Object::new();
|
||||
for i in 0..8_u64 {
|
||||
a.insert(val(i), val(i));
|
||||
}
|
||||
for i in (0..8_u64).rev() {
|
||||
b.insert(val(i), val(i));
|
||||
}
|
||||
assert_eq!(format!("{a:?}"), format!("{b:?}"));
|
||||
}
|
||||
|
||||
/// `keys_sorted` yields exactly `iter_sorted().map(|(k,_)| k)`.
|
||||
#[test]
|
||||
fn object_keys_sorted_matches_iter_sorted_keys() {
|
||||
let obj: Object = make_pairs(16).into_iter().collect();
|
||||
let from_keys: Vec<&Value> = obj.keys_sorted().collect();
|
||||
let from_iter: Vec<&Value> = obj.iter_sorted().map(|(k, _)| k).collect();
|
||||
assert_eq!(from_keys, from_iter);
|
||||
}
|
||||
|
||||
// ---- Serde: non-string keys & determinism --------------------------------
|
||||
|
||||
/// `Serialize` stringifies non-string keys, and equal Objects produce identical JSON
|
||||
/// regardless of insertion order.
|
||||
#[test]
|
||||
fn object_serialize_non_string_keys_and_deterministic() {
|
||||
let pairs = [
|
||||
(Value::from("alpha"), val(1)),
|
||||
(Value::Bool(true), val(2)),
|
||||
(val(7), val(3)),
|
||||
];
|
||||
let a: Object = pairs.iter().cloned().collect();
|
||||
let mut b = Object::new();
|
||||
for (k, v) in pairs.iter().rev().cloned() {
|
||||
b.insert(k, v);
|
||||
}
|
||||
let ja = serde_json::to_string(&a).expect("ser a");
|
||||
let jb = serde_json::to_string(&b).expect("ser b");
|
||||
assert_eq!(ja, jb, "serialization must be deterministic");
|
||||
|
||||
// Non-string keys appear as quoted strings in the resulting JSON.
|
||||
let v: serde_json::Value = serde_json::from_str(&ja).expect("parse");
|
||||
let obj = v.as_object().expect("json object");
|
||||
assert!(
|
||||
obj.contains_key("true"),
|
||||
"bool key was not stringified: {ja}"
|
||||
);
|
||||
assert!(
|
||||
obj.contains_key("7"),
|
||||
"number key was not stringified: {ja}"
|
||||
);
|
||||
assert!(obj.contains_key("alpha"));
|
||||
}
|
||||
|
||||
// ---- Extend / append duplicate-key semantics -----------------------------
|
||||
|
||||
/// `extend` overwrites existing entries (last-write-wins) and preserves length when
|
||||
/// only existing keys are touched.
|
||||
#[test]
|
||||
fn object_extend_last_wins_and_empty_noop() {
|
||||
let mut obj: Object = [(val(0), val(0)), (val(1), val(1))].into_iter().collect();
|
||||
obj.extend([(val(0), val(99))]);
|
||||
assert_eq!(obj.get(&val(0)), Some(&val(99)));
|
||||
assert_eq!(obj.len(), 2);
|
||||
|
||||
let before = obj.len();
|
||||
obj.extend(core::iter::empty::<(Value, Value)>());
|
||||
assert_eq!(obj.len(), before, "empty extend is a no-op");
|
||||
}
|
||||
|
||||
/// `append` drains `other` into `self`, overwriting on overlapping keys.
|
||||
#[test]
|
||||
fn object_append_overlapping_keys_drain_and_overwrite() {
|
||||
let mut a: Object = [(val(0), val(0)), (val(1), val(1))].into_iter().collect();
|
||||
let mut b: Object = [(val(1), val(99)), (val(2), val(2))].into_iter().collect();
|
||||
a.append(&mut b);
|
||||
assert!(b.is_empty(), "append must drain `other`");
|
||||
assert_eq!(a.len(), 3);
|
||||
assert_eq!(a.get(&val(1)), Some(&val(99)));
|
||||
assert_eq!(a.get(&val(2)), Some(&val(2)));
|
||||
}
|
||||
|
||||
// ---- Iterator trait surface ---------------------------------------------
|
||||
|
||||
/// `DoubleEndedIterator`/`ExactSizeIterator`/`FusedIterator` and `size_hint` all
|
||||
/// behave correctly across partial consumption from both ends.
|
||||
#[test]
|
||||
fn object_iter_sorted_double_ended_and_exact_size() {
|
||||
let obj: Object = make_pairs(4).into_iter().collect();
|
||||
let mut it = obj.iter_sorted();
|
||||
assert_eq!(it.len(), 4);
|
||||
assert_eq!(it.size_hint(), (4, Some(4)));
|
||||
|
||||
let first = it.next().expect("front");
|
||||
let last = it.next_back().expect("back");
|
||||
assert_eq!(it.len(), 2);
|
||||
assert_eq!(it.size_hint(), (2, Some(2)));
|
||||
assert_ne!(first.0, last.0, "front and back must differ for n=4");
|
||||
|
||||
// Drain remaining.
|
||||
while it.next().is_some() {}
|
||||
assert_eq!(it.len(), 0);
|
||||
// FusedIterator: stays None after exhaustion.
|
||||
assert!(it.next().is_none());
|
||||
assert!(it.next().is_none());
|
||||
assert!(it.next_back().is_none());
|
||||
}
|
||||
|
||||
/// `IntoIter` also honors `DoubleEndedIterator` and `ExactSizeIterator`.
|
||||
#[test]
|
||||
fn object_into_iter_double_ended_and_exact_size() {
|
||||
let obj: Object = make_pairs(4).into_iter().collect();
|
||||
let mut it = obj.into_iter();
|
||||
assert_eq!(it.len(), 4);
|
||||
let _ = it.next().expect("front");
|
||||
let _ = it.next_back().expect("back");
|
||||
assert_eq!(it.len(), 2);
|
||||
let collected: Vec<_> = it.collect();
|
||||
assert_eq!(collected.len(), 2);
|
||||
}
|
||||
|
||||
/// `IterMut` decrements its `len()` after consuming from the front.
|
||||
#[test]
|
||||
fn object_iter_mut_exact_size() {
|
||||
let mut obj: Object = make_pairs(3).into_iter().collect();
|
||||
let mut it = obj.iter_mut();
|
||||
assert_eq!(it.len(), 3);
|
||||
let _ = it.next().expect("front");
|
||||
assert_eq!(it.len(), 2);
|
||||
}
|
||||
|
||||
/// `Iter` is `Clone`; the clone iterates independently from the same point.
|
||||
#[test]
|
||||
fn object_iter_sorted_clone_is_independent() {
|
||||
let obj: Object = make_pairs(4).into_iter().collect();
|
||||
let mut a = obj.iter_sorted();
|
||||
let _ = a.next();
|
||||
let b = a.clone();
|
||||
let rest_a: Vec<_> = a.collect();
|
||||
let rest_b: Vec<_> = b.collect();
|
||||
assert_eq!(rest_a, rest_b);
|
||||
}
|
||||
|
||||
// ---- default / insert ---------------------------------------------------
|
||||
|
||||
/// `Object::default()` and `Object::new()` produce equal, empty Objects.
|
||||
#[test]
|
||||
fn object_default_equals_new_and_is_empty() {
|
||||
let a = Object::default();
|
||||
let b = Object::new();
|
||||
assert_eq!(a, b);
|
||||
assert!(a.is_empty());
|
||||
assert_eq!(a.len(), 0);
|
||||
}
|
||||
|
||||
/// `insert` returns `None` for a fresh key and `Some(old)` when overwriting.
|
||||
#[test]
|
||||
fn object_insert_returns_previous_value() {
|
||||
let mut obj = Object::new();
|
||||
assert_eq!(obj.insert(val(0), val(1)), None);
|
||||
assert_eq!(obj.insert(val(0), val(2)), Some(val(1)));
|
||||
assert_eq!(obj.get(&val(0)), Some(&val(2)));
|
||||
}
|
||||
@@ -23,8 +23,7 @@ use regorus::languages::azure_policy::aliases::AliasRegistry;
|
||||
use regorus::languages::azure_policy::compiler;
|
||||
use regorus::languages::azure_policy::parser;
|
||||
use regorus::rvm::RegoVM;
|
||||
use regorus::Source;
|
||||
use regorus::Value;
|
||||
use regorus::{Rc, Source, Value};
|
||||
use serde::{Deserialize, Serialize};
|
||||
use std::collections::BTreeMap;
|
||||
use std::fs;
|
||||
@@ -194,7 +193,7 @@ fn yaml_test_impl(file: &str) -> Result<()> {
|
||||
let test: YamlTest = serde_yaml::from_str(&yaml_str)?;
|
||||
|
||||
// Load alias registry if an aliases file is specified.
|
||||
let alias_registry = if let Some(ref aliases_file) = test.aliases {
|
||||
let alias_registry: Option<Rc<AliasRegistry>> = if let Some(ref aliases_file) = test.aliases {
|
||||
let aliases_dir = Path::new(file)
|
||||
.parent()
|
||||
.unwrap_or_else(|| Path::new("."))
|
||||
@@ -209,7 +208,7 @@ fn yaml_test_impl(file: &str) -> Result<()> {
|
||||
})?;
|
||||
let mut registry = AliasRegistry::new();
|
||||
registry.load_from_json(&aliases_json)?;
|
||||
Some(registry)
|
||||
Some(Rc::new(registry))
|
||||
} else {
|
||||
None
|
||||
};
|
||||
@@ -281,11 +280,7 @@ fn yaml_test_impl(file: &str) -> Result<()> {
|
||||
);
|
||||
}
|
||||
if let Some(ref registry) = alias_registry {
|
||||
compiler::compile_policy_definition_with_aliases(
|
||||
&defn,
|
||||
registry.alias_map(),
|
||||
registry.alias_modifiable_map(),
|
||||
)
|
||||
compiler::compile_policy_definition_with_aliases(&defn, Rc::clone(registry))
|
||||
} else {
|
||||
compiler::compile_policy_definition(&defn)
|
||||
}
|
||||
@@ -308,11 +303,7 @@ fn yaml_test_impl(file: &str) -> Result<()> {
|
||||
);
|
||||
}
|
||||
if let Some(ref registry) = alias_registry {
|
||||
compiler::compile_policy_rule_with_aliases(
|
||||
&ast,
|
||||
registry.alias_map(),
|
||||
registry.alias_modifiable_map(),
|
||||
)
|
||||
compiler::compile_policy_rule_with_aliases(&ast, Rc::clone(registry))
|
||||
} else {
|
||||
compiler::compile_policy_rule(&ast)
|
||||
}
|
||||
@@ -366,7 +357,7 @@ fn yaml_test_impl(file: &str) -> Result<()> {
|
||||
|
||||
let mut vm = RegoVM::new();
|
||||
vm.load_program(program);
|
||||
vm.set_input(make_input(case, alias_registry.as_ref())?);
|
||||
vm.set_input(make_input(case, alias_registry.as_deref())?);
|
||||
vm.set_context(make_context(case)?);
|
||||
|
||||
// Load host-await responses (for auditIfNotExists / deployIfNotExists policies).
|
||||
@@ -391,7 +382,11 @@ fn yaml_test_impl(file: &str) -> Result<()> {
|
||||
if let Some(rt) = effective_type {
|
||||
inject_type_field(&mut raw, rt);
|
||||
}
|
||||
normalizer::normalize(&raw, Some(registry), case.api_version.as_deref())
|
||||
normalizer::normalize(
|
||||
&raw,
|
||||
Some(registry.as_ref()),
|
||||
case.api_version.as_deref(),
|
||||
)
|
||||
} else {
|
||||
raw
|
||||
}
|
||||
@@ -549,16 +544,14 @@ fn make_context(case: &TestCase) -> Result<Value> {
|
||||
let map = ctx.as_object_mut()?;
|
||||
// Only inject if the caller didn't already provide requestContext
|
||||
// in the context object, to avoid clobbering custom test setups.
|
||||
map.entry(Value::from("requestContext")).or_insert(rc_val);
|
||||
map.get_or_insert_with(Value::from("requestContext"), || rc_val);
|
||||
} else if let Some(ref api_ver) = case.api_version {
|
||||
let map = ctx.as_object_mut()?;
|
||||
if let std::collections::btree_map::Entry::Vacant(e) =
|
||||
map.entry(Value::from("requestContext"))
|
||||
{
|
||||
if !map.contains_key(&Value::from("requestContext")) {
|
||||
let mut req_ctx = Value::new_object();
|
||||
let rc_map = req_ctx.as_object_mut()?;
|
||||
rc_map.insert(Value::from("apiVersion"), Value::from(api_ver.clone()));
|
||||
e.insert(req_ctx);
|
||||
map.insert(Value::from("requestContext"), req_ctx);
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
268
tests/interpreter/cases/refr/keywords_in_refs.yaml
Normal file
268
tests/interpreter/cases/refr/keywords_in_refs.yaml
Normal file
@@ -0,0 +1,268 @@
|
||||
# Copyright (c) Microsoft Corporation.
|
||||
# Licensed under the MIT License.
|
||||
#
|
||||
# Tests for keywords-as-field-names in dot-notation refs.
|
||||
# Matches OPA's `keywords_in_refs` behavior, enabled by default.
|
||||
cases:
|
||||
- note: keywords_in_refs/package field
|
||||
modules:
|
||||
- |
|
||||
package test
|
||||
allow if {
|
||||
input.v0.package.format == "npm"
|
||||
}
|
||||
input:
|
||||
v0:
|
||||
package:
|
||||
format: npm
|
||||
query: data.test.allow
|
||||
want_result: true
|
||||
|
||||
- note: keywords_in_refs/as field
|
||||
modules:
|
||||
- |
|
||||
package test
|
||||
x = input.as.type
|
||||
input:
|
||||
as:
|
||||
type: string
|
||||
query: data.test.x
|
||||
want_result: string
|
||||
|
||||
- note: keywords_in_refs/default field
|
||||
modules:
|
||||
- |
|
||||
package test
|
||||
x = input.default.value
|
||||
input:
|
||||
default:
|
||||
value: 42
|
||||
query: data.test.x
|
||||
want_result: 42
|
||||
|
||||
- note: keywords_in_refs/else field
|
||||
modules:
|
||||
- |
|
||||
package test
|
||||
x = input.else.value
|
||||
input:
|
||||
else:
|
||||
value: hello
|
||||
query: data.test.x
|
||||
want_result: hello
|
||||
|
||||
- note: keywords_in_refs/import field
|
||||
modules:
|
||||
- |
|
||||
package test
|
||||
x = input.import.name
|
||||
input:
|
||||
import:
|
||||
name: foo
|
||||
query: data.test.x
|
||||
want_result: foo
|
||||
|
||||
- note: keywords_in_refs/not field
|
||||
modules:
|
||||
- |
|
||||
package test
|
||||
x = input.not.allowed
|
||||
input:
|
||||
not:
|
||||
allowed: false
|
||||
query: data.test.x
|
||||
want_result: false
|
||||
|
||||
- note: keywords_in_refs/null field
|
||||
modules:
|
||||
- |
|
||||
package test
|
||||
x = input.null.value
|
||||
input:
|
||||
"null":
|
||||
value: 1
|
||||
query: data.test.x
|
||||
want_result: 1
|
||||
|
||||
- note: keywords_in_refs/some field
|
||||
modules:
|
||||
- |
|
||||
package test
|
||||
x = input.some.field
|
||||
input:
|
||||
some:
|
||||
field: bar
|
||||
query: data.test.x
|
||||
want_result: bar
|
||||
|
||||
- note: keywords_in_refs/true field
|
||||
modules:
|
||||
- |
|
||||
package test
|
||||
x = input.true.x
|
||||
input:
|
||||
"true":
|
||||
x: 2
|
||||
query: data.test.x
|
||||
want_result: 2
|
||||
|
||||
- note: keywords_in_refs/false field
|
||||
modules:
|
||||
- |
|
||||
package test
|
||||
x = input.false.x
|
||||
input:
|
||||
"false":
|
||||
x: 3
|
||||
query: data.test.x
|
||||
want_result: 3
|
||||
|
||||
- note: keywords_in_refs/with field
|
||||
modules:
|
||||
- |
|
||||
package test
|
||||
x = input.with.config
|
||||
input:
|
||||
with:
|
||||
config: test
|
||||
query: data.test.x
|
||||
want_result: test
|
||||
|
||||
- note: keywords_in_refs/future keywords (if, in, every, contains)
|
||||
modules:
|
||||
- |
|
||||
package test
|
||||
import future.keywords
|
||||
x if {
|
||||
input.if.condition == true
|
||||
input.in.set == "member"
|
||||
input.every.item == "x"
|
||||
input.contains.key == "val"
|
||||
}
|
||||
input:
|
||||
if:
|
||||
condition: true
|
||||
in:
|
||||
set: member
|
||||
every:
|
||||
item: x
|
||||
contains:
|
||||
key: val
|
||||
query: data.test.x
|
||||
want_result: true
|
||||
|
||||
- note: keywords_in_refs/chained keywords
|
||||
modules:
|
||||
- |
|
||||
package test
|
||||
x = input.package.import.default
|
||||
input:
|
||||
package:
|
||||
import:
|
||||
default: chained
|
||||
query: data.test.x
|
||||
want_result: chained
|
||||
|
||||
- note: keywords_in_refs/data path with keyword
|
||||
modules:
|
||||
- |
|
||||
package test
|
||||
x = data.mydata.package.name
|
||||
data:
|
||||
mydata:
|
||||
package:
|
||||
name: mypackage
|
||||
query: data.test.x
|
||||
want_result: mypackage
|
||||
|
||||
- note: keywords_in_refs/rego v1 all keywords
|
||||
modules:
|
||||
- |
|
||||
package test
|
||||
import rego.v1
|
||||
allow if {
|
||||
input.package.format == "npm"
|
||||
input.default.value == 1
|
||||
input.if.enabled == true
|
||||
input.in.set == "member"
|
||||
input.not.flag == false
|
||||
input.with.config == "ok"
|
||||
}
|
||||
input:
|
||||
package:
|
||||
format: npm
|
||||
default:
|
||||
value: 1
|
||||
if:
|
||||
enabled: true
|
||||
in:
|
||||
set: member
|
||||
not:
|
||||
flag: false
|
||||
with:
|
||||
config: ok
|
||||
query: data.test.allow
|
||||
want_result: true
|
||||
|
||||
- note: keywords_in_refs/future keywords without import
|
||||
modules:
|
||||
- |
|
||||
package test
|
||||
x = [input.if.flag, input.in.value, input.every.item, input.contains.key]
|
||||
input:
|
||||
if:
|
||||
flag: true
|
||||
in:
|
||||
value: member
|
||||
every:
|
||||
item: each
|
||||
contains:
|
||||
key: present
|
||||
query: data.test.x
|
||||
want_result: [true, "member", "each", "present"]
|
||||
|
||||
- note: keywords_in_refs/package path keywords
|
||||
modules:
|
||||
- |
|
||||
package words.if.default
|
||||
value = 7
|
||||
- |
|
||||
package test
|
||||
x = data.words.if.default.value
|
||||
query: data.test.x
|
||||
want_result: 7
|
||||
|
||||
- note: keywords_in_refs/import path keywords
|
||||
modules:
|
||||
- |
|
||||
package test
|
||||
import data.catalog.if.default as kw
|
||||
x = kw.value
|
||||
data:
|
||||
catalog:
|
||||
if:
|
||||
default:
|
||||
value: 99
|
||||
query: data.test.x
|
||||
want_result: 99
|
||||
|
||||
- note: keywords_in_refs/rule head keyword path
|
||||
modules:
|
||||
- |
|
||||
package test
|
||||
policy.default.level := 3
|
||||
query: data.test.policy.default.level
|
||||
want_result: 3
|
||||
|
||||
- note: keywords_in_refs/mixed dot keyword and dynamic bracket
|
||||
modules:
|
||||
- |
|
||||
package test
|
||||
x = input.package[segment].value
|
||||
segment = "import"
|
||||
input:
|
||||
package:
|
||||
import:
|
||||
value: from_dynamic
|
||||
query: data.test.x
|
||||
want_result: from_dynamic
|
||||
150
tests/interpreter/cases/rule/partial_object_v1.yaml
Normal file
150
tests/interpreter/cases/rule/partial_object_v1.yaml
Normal file
@@ -0,0 +1,150 @@
|
||||
# Copyright (c) Microsoft Corporation.
|
||||
# Licensed under the MIT License.
|
||||
|
||||
cases:
|
||||
- note: constant_key_partial_object_v1
|
||||
data: {}
|
||||
input:
|
||||
enabled: true
|
||||
modules:
|
||||
- |
|
||||
package test
|
||||
import rego.v1
|
||||
|
||||
p["fixed"] if {
|
||||
input.enabled
|
||||
}
|
||||
query: data.test
|
||||
want_result:
|
||||
p:
|
||||
fixed: true
|
||||
|
||||
- note: multilevel_partial_object_v1
|
||||
data: {}
|
||||
input:
|
||||
nested:
|
||||
app:
|
||||
read: 1
|
||||
write: 2
|
||||
ops:
|
||||
deploy: 3
|
||||
modules:
|
||||
- |
|
||||
package test
|
||||
import rego.v1
|
||||
|
||||
p[a][b] if {
|
||||
some a, obj in input.nested
|
||||
some b, _ in obj
|
||||
}
|
||||
query: data.test
|
||||
want_result:
|
||||
p:
|
||||
app:
|
||||
read: true
|
||||
write: true
|
||||
ops:
|
||||
deploy: true
|
||||
|
||||
- note: constant_key_partial_object_explicit_value_v1
|
||||
data: {}
|
||||
input:
|
||||
enabled: true
|
||||
modules:
|
||||
- |
|
||||
package test
|
||||
import rego.v1
|
||||
|
||||
p["fixed"] := 7 if {
|
||||
input.enabled
|
||||
}
|
||||
query: data.test
|
||||
want_result:
|
||||
p:
|
||||
fixed: 7
|
||||
|
||||
- note: multilevel_partial_object_explicit_value_v1
|
||||
data: {}
|
||||
input:
|
||||
nested:
|
||||
app:
|
||||
read: 1
|
||||
write: 2
|
||||
ops:
|
||||
deploy: 3
|
||||
modules:
|
||||
- |
|
||||
package test
|
||||
import rego.v1
|
||||
|
||||
p[a][b] := v if {
|
||||
some a, obj in input.nested
|
||||
some b, v in obj
|
||||
}
|
||||
query: data.test
|
||||
want_result:
|
||||
p:
|
||||
app:
|
||||
read: 1
|
||||
write: 2
|
||||
ops:
|
||||
deploy: 3
|
||||
|
||||
- note: issue_712_reproducer_v0_partial_set
|
||||
data: {}
|
||||
input:
|
||||
servers:
|
||||
FOO: 1
|
||||
BAR: 2
|
||||
BAZ: 3
|
||||
modules:
|
||||
- |
|
||||
package test
|
||||
import future.keywords.in
|
||||
|
||||
violations[k] {
|
||||
some k, _ in input.servers
|
||||
}
|
||||
query: data.test.violations
|
||||
want_result:
|
||||
set!: ["BAR", "BAZ", "FOO"]
|
||||
|
||||
- note: issue_712_reproducer_v1_partial_object
|
||||
data: {}
|
||||
input:
|
||||
servers:
|
||||
FOO: 1
|
||||
BAR: 2
|
||||
BAZ: 3
|
||||
modules:
|
||||
- |
|
||||
package test
|
||||
import rego.v1
|
||||
|
||||
violations[k] if {
|
||||
some k, _ in input.servers
|
||||
}
|
||||
query: data.test.violations
|
||||
want_result:
|
||||
BAR: true
|
||||
BAZ: true
|
||||
FOO: true
|
||||
|
||||
- note: issue_712_reproducer_v1_contains_partial_set
|
||||
data: {}
|
||||
input:
|
||||
servers:
|
||||
FOO: 1
|
||||
BAR: 2
|
||||
BAZ: 3
|
||||
modules:
|
||||
- |
|
||||
package test
|
||||
import rego.v1
|
||||
|
||||
violations contains k if {
|
||||
some k, _ in input.servers
|
||||
}
|
||||
query: data.test.violations
|
||||
want_result:
|
||||
set!: ["BAR", "BAZ", "FOO"]
|
||||
22
tests/opa.rs
22
tests/opa.rs
@@ -275,6 +275,14 @@ fn is_with_keyword_unsupported_error(err: &anyhow::Error) -> bool {
|
||||
})
|
||||
}
|
||||
|
||||
fn is_partial_object_unsupported_error(err: &anyhow::Error) -> bool {
|
||||
err.chain().any(|cause| {
|
||||
let msg = cause.to_string();
|
||||
msg.contains("partial object rules with constant keys are not yet supported")
|
||||
|| msg.contains("partial object rules with nested bracket keys are not yet supported")
|
||||
})
|
||||
}
|
||||
|
||||
fn maybe_verify_rvm_case(case: &TestCase, is_rego_v0_test: bool, actual: &Value) -> Result<()> {
|
||||
if case.note == "defaultkeyword/function with var arg, ref head query" {
|
||||
println!(
|
||||
@@ -308,6 +316,14 @@ fn maybe_verify_rvm_case(case: &TestCase, is_rego_v0_test: bool, actual: &Value)
|
||||
return Ok(());
|
||||
}
|
||||
|
||||
if is_partial_object_unsupported_error(&err) {
|
||||
println!(
|
||||
" skipping RVM check for '{}' (partial object pattern unsupported)",
|
||||
case.note
|
||||
);
|
||||
return Ok(());
|
||||
}
|
||||
|
||||
return Err(err);
|
||||
}
|
||||
};
|
||||
@@ -364,7 +380,11 @@ fn run_opa_tests(opa_tests_dir: String, folders: &[String]) -> Result<()> {
|
||||
}
|
||||
let path = Path::new(&path_str);
|
||||
let path_dir = path.strip_prefix(tests_path)?.parent().unwrap();
|
||||
let path_dir_str = path_dir.to_string_lossy().to_string();
|
||||
// Normalize to forward slashes so the folder filter at the
|
||||
// `folders.iter().any(|f| &path_dir_str == f)` check below matches
|
||||
// CLI arguments like `v0/aggregates` on Windows, where
|
||||
// `to_string_lossy` yields backslash separators by default.
|
||||
let path_dir_str = path_dir.to_string_lossy().replace('\\', "/");
|
||||
let folder_name = folder_name_from_path(path_dir);
|
||||
let skip_rvm_for_folder = folder_name
|
||||
.as_deref()
|
||||
|
||||
@@ -315,3 +315,109 @@ cases:
|
||||
}
|
||||
query: data.test.main
|
||||
want_result: "/api/v1/users"
|
||||
|
||||
- note: keywords_in_refs/package_field
|
||||
data: {}
|
||||
input:
|
||||
v0:
|
||||
package:
|
||||
format: npm
|
||||
modules:
|
||||
- |
|
||||
package test
|
||||
allow := true if {
|
||||
input.v0.package.format == "npm"
|
||||
}
|
||||
query: data.test.allow
|
||||
want_result: true
|
||||
|
||||
- note: keywords_in_refs/multiple_keywords
|
||||
data: {}
|
||||
input:
|
||||
default:
|
||||
value: 42
|
||||
import:
|
||||
name: foo
|
||||
not:
|
||||
allowed: false
|
||||
with:
|
||||
config: ok
|
||||
modules:
|
||||
- |
|
||||
package test
|
||||
import rego.v1
|
||||
result if {
|
||||
input.default.value == 42
|
||||
input.import.name == "foo"
|
||||
input.not.allowed == false
|
||||
input.with.config == "ok"
|
||||
}
|
||||
query: data.test.result
|
||||
want_result: true
|
||||
|
||||
- note: keywords_in_refs/future_keywords_without_import
|
||||
data: {}
|
||||
input:
|
||||
if:
|
||||
flag: true
|
||||
in:
|
||||
value: member
|
||||
every:
|
||||
item: each
|
||||
contains:
|
||||
key: present
|
||||
modules:
|
||||
- |
|
||||
package test
|
||||
result := [input.if.flag, input.in.value, input.every.item, input.contains.key]
|
||||
query: data.test.result
|
||||
want_result: [true, "member", "each", "present"]
|
||||
|
||||
- note: keywords_in_refs/package_path_keywords
|
||||
data: {}
|
||||
modules:
|
||||
- |
|
||||
package words.if.default
|
||||
value := 7
|
||||
- |
|
||||
package test
|
||||
result := data.words.if.default.value
|
||||
query: data.test.result
|
||||
want_result: 7
|
||||
|
||||
- note: keywords_in_refs/import_path_keywords
|
||||
data:
|
||||
catalog:
|
||||
if:
|
||||
default:
|
||||
value: 99
|
||||
modules:
|
||||
- |
|
||||
package test
|
||||
import data.catalog.if.default as kw
|
||||
result := kw.value
|
||||
query: data.test.result
|
||||
want_result: 99
|
||||
|
||||
- note: keywords_in_refs/rule_head_keyword_path
|
||||
data: {}
|
||||
modules:
|
||||
- |
|
||||
package test
|
||||
policy.default.level := 3
|
||||
query: data.test.policy.default.level
|
||||
want_result: 3
|
||||
|
||||
- note: keywords_in_refs/mixed_dot_keyword_and_dynamic_bracket
|
||||
data: {}
|
||||
input:
|
||||
package:
|
||||
import:
|
||||
value: from_dynamic
|
||||
modules:
|
||||
- |
|
||||
package test
|
||||
segment := "import"
|
||||
result := input.package[segment].value
|
||||
query: data.test.result
|
||||
want_result: "from_dynamic"
|
||||
|
||||
939
tests/rvm/rego/cases/partial_object_rules.yaml
Normal file
939
tests/rvm/rego/cases/partial_object_rules.yaml
Normal file
@@ -0,0 +1,939 @@
|
||||
# Copyright (c) Microsoft Corporation.
|
||||
# Licensed under the MIT License.
|
||||
|
||||
cases:
|
||||
- note: partial_object_variable_key_collects_all_bindings
|
||||
data: {}
|
||||
input:
|
||||
items:
|
||||
FOO: 1
|
||||
BAR: 2
|
||||
BAZ: 3
|
||||
modules:
|
||||
- |
|
||||
package test
|
||||
|
||||
p[k] if {
|
||||
some k, _ in input.items
|
||||
}
|
||||
query: data.test.p
|
||||
want_result:
|
||||
BAR: true
|
||||
BAZ: true
|
||||
FOO: true
|
||||
|
||||
- note: partial_object_explicit_value_collects_all_bindings
|
||||
data: {}
|
||||
input:
|
||||
items:
|
||||
FOO: 1
|
||||
BAR: 2
|
||||
BAZ: 3
|
||||
modules:
|
||||
- |
|
||||
package test
|
||||
|
||||
p[k] := v if {
|
||||
some k, v in input.items
|
||||
}
|
||||
query: data.test.p
|
||||
want_result:
|
||||
BAR: 2
|
||||
BAZ: 3
|
||||
FOO: 1
|
||||
|
||||
- note: partial_object_dynamic_expression_key_collects_all_bindings
|
||||
data: {}
|
||||
input:
|
||||
items:
|
||||
FOO: 1
|
||||
BAR: 2
|
||||
BAZ: 3
|
||||
aliases:
|
||||
FOO: alias-foo
|
||||
BAR: alias-bar
|
||||
BAZ: alias-baz
|
||||
modules:
|
||||
- |
|
||||
package test
|
||||
|
||||
p[input.aliases[k]] := v if {
|
||||
some k, v in input.items
|
||||
}
|
||||
query: data.test.p
|
||||
want_result:
|
||||
alias-bar: 2
|
||||
alias-baz: 3
|
||||
alias-foo: 1
|
||||
|
||||
- note: partial_object_undefined_key_skipped
|
||||
# TODO(#719): RVM incorrectly materializes undefined keys instead of
|
||||
# skipping iterations where the key is undefined.
|
||||
skip: true
|
||||
data: {}
|
||||
input:
|
||||
items:
|
||||
FOO: 1
|
||||
BAR: 2
|
||||
aliases:
|
||||
FOO: alias-foo
|
||||
modules:
|
||||
- |
|
||||
package test
|
||||
|
||||
p[input.aliases[k]] := v if {
|
||||
some k, v in input.items
|
||||
}
|
||||
query: data.test.p
|
||||
want_result:
|
||||
alias-foo: 1
|
||||
|
||||
- note: partial_object_duplicate_key_last_wins
|
||||
# TODO(#719): regorus silently overwrites conflicting keys instead of
|
||||
# erroring when the same key is produced with different values.
|
||||
skip: true
|
||||
data: {}
|
||||
input: {}
|
||||
modules:
|
||||
- |
|
||||
package test
|
||||
|
||||
p[k] := v if {
|
||||
some k, v in {"a": 1}
|
||||
}
|
||||
|
||||
p[k] := v if {
|
||||
some k, v in {"a": 2}
|
||||
}
|
||||
query: data.test.p
|
||||
want_error: "conflict"
|
||||
|
||||
- note: partial_object_duplicate_key_same_value_ok
|
||||
data: {}
|
||||
input: {}
|
||||
modules:
|
||||
- |
|
||||
package test
|
||||
|
||||
p[k] if {
|
||||
some k, _ in {"a": 1}
|
||||
}
|
||||
|
||||
p[k] if {
|
||||
some k, _ in {"a": 2}
|
||||
}
|
||||
query: data.test.p
|
||||
want_result:
|
||||
a: true
|
||||
|
||||
- note: partial_object_single_element_input
|
||||
data: {}
|
||||
input:
|
||||
items:
|
||||
ONLY: 1
|
||||
modules:
|
||||
- |
|
||||
package test
|
||||
|
||||
p[k] if {
|
||||
some k, _ in input.items
|
||||
}
|
||||
query: data.test.p
|
||||
want_result:
|
||||
ONLY: true
|
||||
|
||||
- note: partial_object_static_bracket_prefix_collects_all_bindings
|
||||
data: {}
|
||||
input:
|
||||
items:
|
||||
FOO: 1
|
||||
BAR: 2
|
||||
BAZ: 3
|
||||
modules:
|
||||
- |
|
||||
package test
|
||||
|
||||
p["a"][k] := v if {
|
||||
some k, v in input.items
|
||||
}
|
||||
query: data.test.p.a
|
||||
want_result:
|
||||
BAR: 2
|
||||
BAZ: 3
|
||||
FOO: 1
|
||||
|
||||
- note: partial_object_constant_key_unsupported_in_rvm
|
||||
data: {}
|
||||
input:
|
||||
enabled: true
|
||||
modules:
|
||||
- |
|
||||
package test
|
||||
|
||||
p["fixed"] if {
|
||||
input.enabled
|
||||
}
|
||||
query: data.test.p.fixed
|
||||
want_error: "partial object rules with constant keys are not yet supported by the RVM compiler"
|
||||
allow_interpreter_success: true
|
||||
|
||||
- note: partial_object_constant_key_explicit_value_unsupported_in_rvm
|
||||
data: {}
|
||||
input:
|
||||
enabled: true
|
||||
modules:
|
||||
- |
|
||||
package test
|
||||
|
||||
p["fixed"] := 7 if {
|
||||
input.enabled
|
||||
}
|
||||
query: data.test.p.fixed
|
||||
want_error: "partial object rules with constant keys are not yet supported by the RVM compiler"
|
||||
allow_interpreter_success: true
|
||||
|
||||
- note: partial_object_multiple_bodies_collects_all_bindings
|
||||
data: {}
|
||||
input:
|
||||
items:
|
||||
FOO: 1
|
||||
BAR: 2
|
||||
BAZ: 3
|
||||
modules:
|
||||
- |
|
||||
package test
|
||||
|
||||
p[k] if {
|
||||
some k, _ in input.items
|
||||
k in {"FOO", "BAR"}
|
||||
}
|
||||
|
||||
p[k] if {
|
||||
some k, _ in input.items
|
||||
k == "BAZ"
|
||||
}
|
||||
query: data.test.p
|
||||
want_result:
|
||||
BAR: true
|
||||
BAZ: true
|
||||
FOO: true
|
||||
|
||||
- note: partial_set_contains_collects_all_bindings
|
||||
data: {}
|
||||
input:
|
||||
items:
|
||||
FOO: 1
|
||||
BAR: 2
|
||||
BAZ: 3
|
||||
modules:
|
||||
- |
|
||||
package test
|
||||
|
||||
p contains k if {
|
||||
some k, _ in input.items
|
||||
}
|
||||
query: data.test.p
|
||||
want_result:
|
||||
set!: ["BAR", "BAZ", "FOO"]
|
||||
|
||||
- note: issue_712_reproducer_v1_partial_object
|
||||
data: {}
|
||||
input:
|
||||
servers:
|
||||
FOO: 1
|
||||
BAR: 2
|
||||
BAZ: 3
|
||||
modules:
|
||||
- |
|
||||
package test
|
||||
import rego.v1
|
||||
|
||||
violations[k] if {
|
||||
some k, _ in input.servers
|
||||
}
|
||||
query: data.test.violations
|
||||
want_result:
|
||||
BAR: true
|
||||
BAZ: true
|
||||
FOO: true
|
||||
|
||||
- note: issue_712_reproducer_v1_contains_partial_set
|
||||
data: {}
|
||||
input:
|
||||
servers:
|
||||
FOO: 1
|
||||
BAR: 2
|
||||
BAZ: 3
|
||||
modules:
|
||||
- |
|
||||
package test
|
||||
import rego.v1
|
||||
|
||||
violations contains k if {
|
||||
some k, _ in input.servers
|
||||
}
|
||||
query: data.test.violations
|
||||
want_result:
|
||||
set!: ["BAR", "BAZ", "FOO"]
|
||||
|
||||
- note: partial_object_multilevel_key_unsupported_in_rvm
|
||||
data: {}
|
||||
input:
|
||||
nested:
|
||||
app:
|
||||
read: 1
|
||||
write: 2
|
||||
ops:
|
||||
deploy: 3
|
||||
modules:
|
||||
- |
|
||||
package test
|
||||
|
||||
p[a][b] if {
|
||||
some a, obj in input.nested
|
||||
some b, _ in obj
|
||||
}
|
||||
|
||||
main := p
|
||||
query: data.test.main
|
||||
want_error: "partial object rules with nested bracket keys are not yet supported by the RVM compiler"
|
||||
allow_interpreter_success: true
|
||||
|
||||
- note: partial_object_multilevel_key_explicit_value_unsupported_in_rvm
|
||||
data: {}
|
||||
input:
|
||||
nested:
|
||||
app:
|
||||
read: 1
|
||||
write: 2
|
||||
ops:
|
||||
deploy: 3
|
||||
modules:
|
||||
- |
|
||||
package test
|
||||
|
||||
p[a][b] := v if {
|
||||
some a, obj in input.nested
|
||||
some b, v in obj
|
||||
}
|
||||
|
||||
main := p
|
||||
query: data.test.main
|
||||
want_error: "partial object rules with nested bracket keys are not yet supported by the RVM compiler"
|
||||
allow_interpreter_success: true
|
||||
|
||||
- note: partial_object_hidden_dynamic_prefix_unsupported_in_rvm
|
||||
data: {}
|
||||
input:
|
||||
nested:
|
||||
app:
|
||||
q:
|
||||
read: 1
|
||||
write: 2
|
||||
ops:
|
||||
q:
|
||||
deploy: 3
|
||||
modules:
|
||||
- |
|
||||
package test
|
||||
|
||||
p[a].q[b] if {
|
||||
some a, obj in input.nested
|
||||
some b, _ in obj.q
|
||||
}
|
||||
|
||||
main := p
|
||||
query: data.test.main
|
||||
want_error: "partial object rules with nested bracket keys are not yet supported by the RVM compiler"
|
||||
allow_interpreter_success: true
|
||||
|
||||
- note: partial_object_array_iteration_collects_all_bindings
|
||||
data: {}
|
||||
input:
|
||||
items: ["FOO", "BAR", "BAZ"]
|
||||
modules:
|
||||
- |
|
||||
package test
|
||||
|
||||
p[v] if {
|
||||
some _, v in input.items
|
||||
}
|
||||
query: data.test.p
|
||||
want_result:
|
||||
BAR: true
|
||||
BAZ: true
|
||||
FOO: true
|
||||
|
||||
- note: partial_object_empty_input_is_empty_object
|
||||
data: {}
|
||||
input:
|
||||
items: {}
|
||||
modules:
|
||||
- |
|
||||
package test
|
||||
|
||||
p[k] if {
|
||||
some k, _ in input.items
|
||||
}
|
||||
query: data.test.p
|
||||
want_result: {}
|
||||
|
||||
- note: partial_object_duplicate_paths_same_key_same_value_deduplicates
|
||||
data: {}
|
||||
input:
|
||||
pairs:
|
||||
- alias: shared
|
||||
value: 1
|
||||
- alias: alpha
|
||||
value: 10
|
||||
- alias: shared
|
||||
value: 1
|
||||
modules:
|
||||
- |
|
||||
package test
|
||||
|
||||
p[entry.alias] := entry.value if {
|
||||
some entry in input.pairs
|
||||
}
|
||||
query: data.test.p
|
||||
want_result:
|
||||
alpha: 10
|
||||
shared: 1
|
||||
|
||||
- note: partial_object_duplicate_paths_same_key_different_values_conflict
|
||||
# TODO(#719): regorus silently overwrites conflicting keys instead of
|
||||
# erroring when the same key is produced with different values.
|
||||
skip: true
|
||||
data: {}
|
||||
input:
|
||||
pairs:
|
||||
- alias: shared
|
||||
value: 1
|
||||
- alias: shared
|
||||
value: 2
|
||||
modules:
|
||||
- |
|
||||
package test
|
||||
|
||||
p[entry.alias] := entry.value if {
|
||||
some entry in input.pairs
|
||||
}
|
||||
query: data.test.p
|
||||
want_error: "conflict"
|
||||
|
||||
- note: partial_object_undefined_key_skips_iteration
|
||||
# TODO(#719): RVM incorrectly materializes undefined keys instead of
|
||||
# skipping iterations where the key is undefined.
|
||||
skip: true
|
||||
data: {}
|
||||
input:
|
||||
items:
|
||||
FOO: 1
|
||||
BAR: 2
|
||||
BAZ: 3
|
||||
aliases:
|
||||
FOO: alias-foo
|
||||
BAZ: alias-baz
|
||||
modules:
|
||||
- |
|
||||
package test
|
||||
|
||||
p[input.aliases[k]] := v if {
|
||||
some k, v in input.items
|
||||
}
|
||||
query: data.test.p
|
||||
want_result:
|
||||
alias-baz: 3
|
||||
alias-foo: 1
|
||||
|
||||
- note: partial_object_undefined_value_skips_iteration
|
||||
# TODO(#719): RVM incorrectly materializes undefined values instead of
|
||||
# skipping iterations where the value is undefined.
|
||||
skip: true
|
||||
data: {}
|
||||
input:
|
||||
keys: ["FOO", "BAR", "BAZ"]
|
||||
values:
|
||||
FOO: 1
|
||||
BAZ: 3
|
||||
modules:
|
||||
- |
|
||||
package test
|
||||
|
||||
p[k] := input.values[k] if {
|
||||
some _, k in input.keys
|
||||
}
|
||||
query: data.test.p
|
||||
want_result:
|
||||
BAZ: 3
|
||||
FOO: 1
|
||||
|
||||
- note: partial_object_mixed_undefined_key_value_cases_skip_bad_iterations
|
||||
# TODO(#719): RVM incorrectly materializes undefined keys/values instead of
|
||||
# skipping iterations where the key or value is undefined.
|
||||
skip: true
|
||||
data: {}
|
||||
input:
|
||||
rows:
|
||||
- src: keep
|
||||
- src: missing_alias
|
||||
- src: missing_value
|
||||
- src: missing_both
|
||||
aliases:
|
||||
keep: alias-keep
|
||||
missing_value: alias-no-value
|
||||
values:
|
||||
keep: 1
|
||||
missing_alias: 2
|
||||
modules:
|
||||
- |
|
||||
package test
|
||||
|
||||
p[input.aliases[row.src]] := input.values[row.src] if {
|
||||
some row in input.rows
|
||||
}
|
||||
query: data.test.p
|
||||
want_result:
|
||||
alias-keep: 1
|
||||
|
||||
- note: partial_object_and_partial_set_same_name_conflict
|
||||
data: {}
|
||||
input:
|
||||
items:
|
||||
FOO: 1
|
||||
modules:
|
||||
- |
|
||||
package test
|
||||
|
||||
p[k] if {
|
||||
some k, _ in input.items
|
||||
}
|
||||
|
||||
p contains "shadow" if {
|
||||
true
|
||||
}
|
||||
query: data.test.p
|
||||
want_error: "has multiple types"
|
||||
|
||||
- note: partial_object_complete_rule_conflicts_with_partial_object
|
||||
data: {}
|
||||
input:
|
||||
items:
|
||||
a: 1
|
||||
modules:
|
||||
- |
|
||||
package test
|
||||
|
||||
p := {"fixed": 1}
|
||||
|
||||
p[k] := v if {
|
||||
some k, v in input.items
|
||||
}
|
||||
query: data.test.p
|
||||
want_error: "multiple types"
|
||||
|
||||
- note: partial_object_partial_set_conflicts_with_partial_object
|
||||
data: {}
|
||||
input:
|
||||
items:
|
||||
a: 1
|
||||
modules:
|
||||
- |
|
||||
package test
|
||||
|
||||
p contains k if {
|
||||
some k, _ in input.items
|
||||
}
|
||||
|
||||
p[k] := 1 if {
|
||||
some k, _ in input.items
|
||||
}
|
||||
query: data.test.p
|
||||
want_error: "multiple types"
|
||||
|
||||
- note: partial_object_large_range_counts_all_entries
|
||||
data: {}
|
||||
input: {}
|
||||
modules:
|
||||
- |
|
||||
package test
|
||||
|
||||
p[key] := n if {
|
||||
n := numbers.range(0, 255)[_]
|
||||
key := sprintf("k-%d", [n])
|
||||
}
|
||||
|
||||
main := count(p)
|
||||
query: data.test.main
|
||||
want_result: 256
|
||||
|
||||
- note: partial_object_rbac_duplicate_actions_deduplicate
|
||||
data:
|
||||
role_permissions:
|
||||
reader: ["read", "list"]
|
||||
writer: ["read", "write"]
|
||||
auditor: ["read", "list"]
|
||||
input:
|
||||
user_roles: ["reader", "writer", "auditor"]
|
||||
modules:
|
||||
- |
|
||||
package test
|
||||
|
||||
allowed_actions[action] if {
|
||||
some role in input.user_roles
|
||||
some action in data.role_permissions[role]
|
||||
}
|
||||
query: data.test.allowed_actions
|
||||
want_result:
|
||||
list: true
|
||||
read: true
|
||||
write: true
|
||||
|
||||
- note: partial_object_violations_real_world_pattern
|
||||
data: {}
|
||||
input:
|
||||
spec:
|
||||
containers:
|
||||
- name: api
|
||||
securityContext:
|
||||
readOnlyRootFilesystem: false
|
||||
- name: worker
|
||||
securityContext:
|
||||
readOnlyRootFilesystem: true
|
||||
- name: sidecar
|
||||
modules:
|
||||
- |
|
||||
package test
|
||||
|
||||
violations[msg] if {
|
||||
some container in input.spec.containers
|
||||
not container.securityContext.readOnlyRootFilesystem
|
||||
msg := sprintf("Container %s must use readOnlyRootFilesystem", [container.name])
|
||||
}
|
||||
query: data.test.violations
|
||||
want_result:
|
||||
Container api must use readOnlyRootFilesystem: true
|
||||
Container sidecar must use readOnlyRootFilesystem: true
|
||||
|
||||
- note: partial_object_resource_mapping_filters_valid_resources
|
||||
data: {}
|
||||
input:
|
||||
resources:
|
||||
svc-api:
|
||||
cpu: 1
|
||||
job-cleanup:
|
||||
cpu: 2
|
||||
svc-worker:
|
||||
cpu: 4
|
||||
modules:
|
||||
- |
|
||||
package test
|
||||
|
||||
valid_resource(name) if {
|
||||
startswith(name, "svc-")
|
||||
}
|
||||
|
||||
resources[name] := config if {
|
||||
some name, config in input.resources
|
||||
valid_resource(name)
|
||||
}
|
||||
query: data.test.resources
|
||||
want_result:
|
||||
svc-api:
|
||||
cpu: 1
|
||||
svc-worker:
|
||||
cpu: 4
|
||||
|
||||
- note: partial_object_computed_concat_key_constant_body
|
||||
data: {}
|
||||
modules:
|
||||
- |
|
||||
package test
|
||||
|
||||
p[concat("", ["edge", "-", "key"])] if {
|
||||
true
|
||||
}
|
||||
query: data.test.p
|
||||
want_result:
|
||||
edge-key: true
|
||||
|
||||
- note: partial_object_duplicate_computed_key_same_value_merges
|
||||
data: {}
|
||||
input:
|
||||
items:
|
||||
A: 0
|
||||
a: 0
|
||||
modules:
|
||||
- |
|
||||
package test
|
||||
|
||||
p[lower(k)] := 1 if {
|
||||
some k, _ in input.items
|
||||
}
|
||||
query: data.test.p
|
||||
want_result:
|
||||
a: 1
|
||||
|
||||
- note: partial_object_function_key_and_object_value
|
||||
data: {}
|
||||
input:
|
||||
items:
|
||||
a: 1
|
||||
b: 2
|
||||
modules:
|
||||
- |
|
||||
package test
|
||||
|
||||
f(x) := concat(":", [x, "suffix"])
|
||||
|
||||
p[f(k)] := {"nested": v + 1} if {
|
||||
some k, v in input.items
|
||||
}
|
||||
query: data.test.p
|
||||
want_result:
|
||||
"a:suffix":
|
||||
nested: 2
|
||||
"b:suffix":
|
||||
nested: 3
|
||||
|
||||
- note: partial_object_array_index_key_uses_selected_elements
|
||||
data: {}
|
||||
input:
|
||||
keys: ["alpha", "beta"]
|
||||
values: [10, 20]
|
||||
modules:
|
||||
- |
|
||||
package test
|
||||
|
||||
p[input.keys[i]] := v if {
|
||||
some i, v in input.values
|
||||
}
|
||||
query: data.test.p
|
||||
want_result:
|
||||
alpha: 10
|
||||
beta: 20
|
||||
|
||||
- note: partial_object_computed_empty_and_special_string_keys
|
||||
data: {}
|
||||
modules:
|
||||
- |
|
||||
package test
|
||||
|
||||
p[concat("", [""])] := "empty" if {
|
||||
true
|
||||
}
|
||||
|
||||
p[concat("", ["a/b?c#d"])] := "special" if {
|
||||
true
|
||||
}
|
||||
query: data.test.p
|
||||
want_result:
|
||||
"": "empty"
|
||||
a/b?c#d: "special"
|
||||
|
||||
- note: partial_object_not_filters_blocked_entries
|
||||
data: {}
|
||||
input:
|
||||
items:
|
||||
allowed: true
|
||||
blocked: true
|
||||
blocked:
|
||||
blocked: true
|
||||
modules:
|
||||
- |
|
||||
package test
|
||||
|
||||
p[k] if {
|
||||
some k, _ in input.items
|
||||
not input.blocked[k]
|
||||
}
|
||||
query: data.test.p
|
||||
want_result:
|
||||
allowed: true
|
||||
|
||||
- note: partial_object_dot_bracket_object_value_collects_all_bindings
|
||||
data: {}
|
||||
input:
|
||||
items:
|
||||
a: 1
|
||||
b: 2
|
||||
modules:
|
||||
- |
|
||||
package test
|
||||
|
||||
p.config[k] := {"nested": v} if {
|
||||
some k, v in input.items
|
||||
}
|
||||
query: data.test.p.config
|
||||
want_result:
|
||||
a:
|
||||
nested: 1
|
||||
b:
|
||||
nested: 2
|
||||
|
||||
- note: partial_object_dynamic_prefix_static_suffix_unsupported_in_rvm
|
||||
data: {}
|
||||
input:
|
||||
items:
|
||||
a: 1
|
||||
b: 2
|
||||
modules:
|
||||
- |
|
||||
package test
|
||||
|
||||
p[k]["fixed"] := upper(k) if {
|
||||
some k, _ in input.items
|
||||
}
|
||||
|
||||
main := p
|
||||
query: data.test.main
|
||||
want_error: "partial object rules with nested bracket keys are not yet supported by the RVM compiler"
|
||||
allow_interpreter_success: true
|
||||
|
||||
- note: partial_object_literal_prefix_nested_unsupported_in_rvm
|
||||
data: {}
|
||||
input:
|
||||
items:
|
||||
a: 1
|
||||
b: 2
|
||||
modules:
|
||||
- |
|
||||
package test
|
||||
|
||||
p[1][k] := v if {
|
||||
some k, v in input.items
|
||||
}
|
||||
|
||||
main := p
|
||||
query: data.test.main
|
||||
want_error: "partial object rules with nested bracket keys are not yet supported by the RVM compiler"
|
||||
allow_interpreter_success: true
|
||||
|
||||
- note: partial_object_three_level_nested_dynamic_unsupported_in_rvm
|
||||
data: {}
|
||||
input:
|
||||
nested:
|
||||
app:
|
||||
read: 1
|
||||
write: 2
|
||||
ops:
|
||||
deploy: 3
|
||||
modules:
|
||||
- |
|
||||
package test
|
||||
|
||||
p["root"][a][b] := v if {
|
||||
some a, obj in input.nested
|
||||
some b, v in obj
|
||||
}
|
||||
|
||||
main := p
|
||||
query: data.test.main
|
||||
want_error: "partial object rules with nested bracket keys are not yet supported by the RVM compiler"
|
||||
allow_interpreter_success: true
|
||||
|
||||
- note: partial_object_with_body_unsupported_in_rvm
|
||||
data: {}
|
||||
input:
|
||||
enabled: false
|
||||
items:
|
||||
a: 1
|
||||
modules:
|
||||
- |
|
||||
package test
|
||||
|
||||
gate if {
|
||||
input.enabled
|
||||
}
|
||||
|
||||
p[k] := v if {
|
||||
some k, v in input.items
|
||||
data.test.gate with input as {"enabled": true}
|
||||
}
|
||||
query: data.test.p
|
||||
want_error: "the `with` keyword is not supported by the compiler yet"
|
||||
allow_interpreter_success: true
|
||||
|
||||
- note: partial_object_every_vacuous_truth_collects_empty_arrays
|
||||
# TODO(#719): RVM currently includes the failing `bad` group here, while the
|
||||
# interpreter returns only `empty` and `ok` (expected per vacuous truth
|
||||
# semantics).
|
||||
skip: true
|
||||
data: {}
|
||||
input:
|
||||
groups:
|
||||
ok: [1, 2]
|
||||
bad: [1, 0]
|
||||
empty: []
|
||||
modules:
|
||||
- |
|
||||
package test
|
||||
|
||||
p[k] if {
|
||||
some k, arr in input.groups
|
||||
every v in arr {
|
||||
v > 0
|
||||
}
|
||||
}
|
||||
query: data.test.p
|
||||
want_result:
|
||||
empty: true
|
||||
ok: true
|
||||
|
||||
- note: partial_object_join_var_multiple_bindings
|
||||
data:
|
||||
a: ["1", "2", "3", "4"]
|
||||
g:
|
||||
a: ["1", "0", "0", "0"]
|
||||
b: ["0", "2", "0", "0"]
|
||||
c: ["0", "0", "0", "4"]
|
||||
modules:
|
||||
- |
|
||||
package test
|
||||
|
||||
p[k] := v if {
|
||||
data.a[i] = v
|
||||
data.g[k][i] = v
|
||||
}
|
||||
query: data.test.p
|
||||
want_result:
|
||||
a: "1"
|
||||
b: "2"
|
||||
c: "4"
|
||||
|
||||
- note: partial_object_composite_value
|
||||
data:
|
||||
g:
|
||||
a: [1, 0, 0, 0]
|
||||
b: [0, 2, 0, 0]
|
||||
c: [0, 0, 0, 4]
|
||||
modules:
|
||||
- |
|
||||
package test
|
||||
|
||||
p[k] := [i, {"v2": v}] if {
|
||||
data.g[k] = x
|
||||
x[i] = v
|
||||
v != 0
|
||||
}
|
||||
query: data.test.p
|
||||
want_result:
|
||||
a: [0, {v2: 1}]
|
||||
b: [1, {v2: 2}]
|
||||
c: [3, {v2: 4}]
|
||||
|
||||
- note: partial_object_true_semantics_dedupes_duplicate_keys
|
||||
data: {}
|
||||
modules:
|
||||
- |
|
||||
package test
|
||||
|
||||
p[k] if {
|
||||
ks := ["a", "b", "c", "a"]
|
||||
ks[_] = k
|
||||
}
|
||||
query: data.test.p
|
||||
want_result:
|
||||
a: true
|
||||
b: true
|
||||
c: true
|
||||
@@ -137,3 +137,45 @@ cases:
|
||||
- "Return { value: 7 }" # Return result set
|
||||
want_result:
|
||||
set!: [1, null, 2]
|
||||
|
||||
# Set iteration resumes from `Bound::Excluded(current_item)`, so the
|
||||
# ComprehensionAdd path must snapshot the current value into
|
||||
# `IterationState::Set.current_item` before advancing — otherwise
|
||||
# iteration stops after the first element. This case exercises
|
||||
# Set-source comprehension end-to-end to lock that requirement in.
|
||||
- note: set_source_visits_all_elements
|
||||
description: Comprehension over a Set source must yield every element, not just the first
|
||||
example_rego: |
|
||||
src := {1, 2, 3, 4}
|
||||
{x | x := src[_]} # {1, 2, 3, 4}
|
||||
literals:
|
||||
- 1
|
||||
- 2
|
||||
- 3
|
||||
- 4
|
||||
instruction_params:
|
||||
comprehension_start_params:
|
||||
- mode: "Set"
|
||||
collection_reg: 0
|
||||
key_reg: 4
|
||||
value_reg: 5
|
||||
result_reg: 7
|
||||
body_start: 11
|
||||
comprehension_end: 13
|
||||
instructions:
|
||||
- "SetNew { dest: 0 }" # Build source set {1,2,3,4} in register 0
|
||||
- "Load { dest: 1, literal_idx: 0 }"
|
||||
- "SetAdd { set: 0, value: 1 }"
|
||||
- "Load { dest: 2, literal_idx: 1 }"
|
||||
- "SetAdd { set: 0, value: 2 }"
|
||||
- "Load { dest: 3, literal_idx: 2 }"
|
||||
- "SetAdd { set: 0, value: 3 }"
|
||||
- "Load { dest: 6, literal_idx: 3 }"
|
||||
- "SetAdd { set: 0, value: 6 }"
|
||||
- "SetNew { dest: 7 }" # Initialize result set
|
||||
- "ComprehensionStart { params_index: 0 }" # Iterate over Set source
|
||||
- "ComprehensionAdd { value_reg: 5 }" # Add current value to result
|
||||
- "Halt"
|
||||
- "Return { value: 7 }"
|
||||
want_result:
|
||||
set!: [1, 2, 3, 4]
|
||||
|
||||
Reference in New Issue
Block a user