mirror of
https://github.com/microsoft/regorus.git
synced 2026-08-05 02:16:11 +00:00
898643129e
- Add PolicyLengthConfig struct with max_col, max_file_bytes, and max_lines fields, replacing hardcoded constants in the lexer. - Add Engine::set_policy_length_config and clear_policy_length_config to allow callers to override the default limits. - Add Source::from_contents_with_limits and from_file_with_limits for direct Source construction with custom limits; existing from_contents and from_file signatures are preserved using defaults. - Add tests for default rejection, custom limits, and engine plumbing. - Add bindings for C, C++, Python, WASM/JS, Java, Ruby, C#, Go
86 lines
2.2 KiB
C#
86 lines
2.2 KiB
C#
// Copyright (c) Microsoft Corporation.
|
|
// Licensed under the MIT License.
|
|
|
|
using System.Diagnostics;
|
|
|
|
|
|
long nanosecPerTick = (1000L * 1000L * 1000L) / Stopwatch.Frequency;
|
|
var w = new Stopwatch();
|
|
|
|
|
|
// Force load of modules.
|
|
{
|
|
var _e = new Regorus.Engine();
|
|
#if NET8_0_OR_GREATER
|
|
var _j = System.Text.Json.JsonDocument.Parse("{}");
|
|
#endif
|
|
}
|
|
|
|
w.Restart();
|
|
|
|
var engine = new Regorus.Engine();
|
|
engine.SetRegoV0(true);
|
|
// Raise the default col limit to 2000
|
|
engine.SetPolicyLengthConfig(new Regorus.PolicyLengthConfig(maxCol: 2000, maxFileBytes: 1048576, maxLines: 20000));
|
|
|
|
w.Stop();
|
|
var newEngineTicks = w.ElapsedTicks;
|
|
|
|
|
|
w.Restart();
|
|
|
|
// Load policies and data.
|
|
engine.AddPolicyFromFile("../../../tests/aci/framework.rego");
|
|
engine.AddPolicyFromFile("../../../tests/aci/api.rego");
|
|
engine.AddPolicyFromFile("../../../tests/aci/policy.rego");
|
|
engine.AddDataFromJsonFile("../../../tests/aci/data.json");
|
|
|
|
|
|
w.Stop();
|
|
var loadPoliciesTicks = w.ElapsedTicks;
|
|
|
|
|
|
w.Restart();
|
|
|
|
// Set input and eval rule.
|
|
engine.SetInputFromJsonFile("../../../tests/aci/input.json");
|
|
var value = engine.EvalRule("data.framework.mount_overlay")
|
|
?? throw new System.InvalidOperationException("Expected EvalRule to return a JSON value.");
|
|
|
|
#if NET8_0_OR_GREATER
|
|
var valueDoc = System.Text.Json.JsonDocument.Parse(value);
|
|
|
|
w.Stop();
|
|
var evalTicks = w.ElapsedTicks;
|
|
|
|
Console.WriteLine("{0}", valueDoc);
|
|
#else
|
|
w.Stop();
|
|
var evalTicks = w.ElapsedTicks;
|
|
#endif
|
|
|
|
|
|
Console.WriteLine("Engine creation took {0} msecs", (newEngineTicks * nanosecPerTick) / (1000.0 * 1000.0));
|
|
Console.WriteLine("Load policies and data took {0} msecs", (loadPoliciesTicks * nanosecPerTick) / (1000.0 * 1000.0));
|
|
Console.WriteLine("EvalRule took {0} msecs", (evalTicks * nanosecPerTick) / (1000.0 * 1000.0));
|
|
|
|
engine = new Regorus.Engine();
|
|
engine.AddPolicy(
|
|
"test.rego",
|
|
"package test\nx = 1\nmessage = `Hello`");
|
|
|
|
engine.SetEnableCoverage(true);
|
|
Console.WriteLine("data.test.message: {0}", engine.EvalRule("data.test.message"));
|
|
Console.WriteLine("Coverage Report:\n{0}", engine.GetCoverageReportPretty());
|
|
|
|
if (engine.EvalRule("data.test.message") != "\"Hello\"")
|
|
{
|
|
Console.WriteLine("Failure.");
|
|
System.Environment.Exit(1);
|
|
}
|
|
else
|
|
{
|
|
Console.WriteLine("Success.");
|
|
}
|
|
|