initial commit: @arbiter/core authorization engine with js-rigor hardening
Zanzibar-style authorization graph engine (direct/chain/TTU/defeasible/ binary modes, condensed snapshots, value relations) with 39 rigor test campaigns. Includes fixes for snapshot binary writer/reader format mismatch (snapshot-of-snapshot corruption), possibility write-boundary validation, empty-graph snapshot serialization, relation lookup cache direction collision, config-redefinition cache invalidation, binary threshold semantics, defeasible compiled routing, and comparator reason whitelisting.
This commit is contained in:
John Dvorak
@@ -0,0 +1,207 @@
|
||||
import { Arbiter } from '../src/core/Arbiter.js';
|
||||
|
||||
function parseArgs(argv) {
|
||||
const args = new Map();
|
||||
for (let i = 2; i < argv.length; i++) {
|
||||
const value = argv[i];
|
||||
if (!value.startsWith('--')) continue;
|
||||
const [key, inline] = value.slice(2).split('=');
|
||||
if (inline !== undefined) {
|
||||
args.set(key, inline);
|
||||
continue;
|
||||
}
|
||||
const next = argv[i + 1];
|
||||
if (next && !next.startsWith('--')) {
|
||||
args.set(key, next);
|
||||
i++;
|
||||
} else {
|
||||
args.set(key, true);
|
||||
}
|
||||
}
|
||||
return args;
|
||||
}
|
||||
|
||||
function createRng(seed) {
|
||||
let state = seed >>> 0;
|
||||
return () => {
|
||||
state = (1664525 * state + 1013904223) >>> 0;
|
||||
return state / 0x100000000;
|
||||
};
|
||||
}
|
||||
|
||||
function randInt(rng, max) {
|
||||
return Math.floor(rng() * max);
|
||||
}
|
||||
|
||||
function buildUsers(arbiter, count, prefix) {
|
||||
for (let i = 0; i < count; i++) {
|
||||
arbiter.addNode(`${prefix}:${i}`, prefix);
|
||||
}
|
||||
}
|
||||
|
||||
function buildComparatorScenario(arbiter, size) {
|
||||
const trueIds = [];
|
||||
const falseIds = [];
|
||||
buildUsers(arbiter, size, 'user');
|
||||
buildUsers(arbiter, size, 'resource');
|
||||
for (let i = 0; i < size; i++) {
|
||||
const isSafe = i % 2 === 0;
|
||||
const score = isSafe ? 20 : 80;
|
||||
const bonus = isSafe ? 5 : 15;
|
||||
const noise = isSafe ? 0 : 10;
|
||||
arbiter.addRelation(`user:${i}`, 'risk_score', `resource:${i}`, 1.0, { value: score });
|
||||
arbiter.addRelation(`user:${i}`, 'risk_bonus', `resource:${i}`, 1.0, { value: bonus });
|
||||
arbiter.addRelation(`user:${i}`, 'risk_noise', `resource:${i}`, 1.0, { value: noise });
|
||||
arbiter.addRelation(`resource:${i}`, 'risk_limit', `resource:${i}`, 1.0, { value: 40 });
|
||||
arbiter.addRelation(`resource:${i}`, 'risk_cap', `resource:${i}`, 1.0, { value: 45 });
|
||||
if (isSafe) trueIds.push(i);
|
||||
else falseIds.push(i);
|
||||
}
|
||||
['risk_score', 'risk_bonus', 'risk_noise', 'risk_limit', 'risk_cap'].forEach(relation => {
|
||||
arbiter.setRelationConfig(relation, { type: 'direct' });
|
||||
});
|
||||
arbiter.setRelationConfig('risk_ok_owa', {
|
||||
type: 'relational_comparator',
|
||||
comparator: '<=',
|
||||
fallbackBehavior: 'deny',
|
||||
left: {
|
||||
rule: {
|
||||
union: {
|
||||
rules: [
|
||||
{ type: 'direct', relation: 'risk_score' },
|
||||
{ type: 'direct', relation: 'risk_bonus' },
|
||||
{ type: 'direct', relation: 'risk_noise' }
|
||||
],
|
||||
aggregator: 'owa',
|
||||
owaWeights: [0.5, 0.3, 0.2]
|
||||
}
|
||||
},
|
||||
extractValue: true,
|
||||
valueRelation: 'risk_score',
|
||||
aggregator: 'owa',
|
||||
owaWeights: [0.5, 0.3, 0.2]
|
||||
},
|
||||
right: {
|
||||
rule: {
|
||||
union: {
|
||||
rules: [
|
||||
{ type: 'direct', relation: 'risk_limit' },
|
||||
{ type: 'direct', relation: 'risk_cap' }
|
||||
],
|
||||
aggregator: 'owa',
|
||||
owaWeights: [0.6, 0.4]
|
||||
}
|
||||
},
|
||||
extractValue: true,
|
||||
valueRelation: 'risk_limit',
|
||||
evaluateFrom: 'object',
|
||||
aggregator: 'owa',
|
||||
owaWeights: [0.6, 0.4]
|
||||
}
|
||||
});
|
||||
return { trueIds, falseIds };
|
||||
}
|
||||
|
||||
function buildFeatureFlagScenario(arbiter, size) {
|
||||
const flagCount = Math.max(1, Math.floor(size / 10));
|
||||
buildUsers(arbiter, size, 'user');
|
||||
buildUsers(arbiter, size, 'resource');
|
||||
buildUsers(arbiter, flagCount, 'flag');
|
||||
for (let i = 0; i < size; i++) {
|
||||
const flagId = i % flagCount;
|
||||
arbiter.addRelation(`user:${i}`, 'has_flag', `flag:${flagId}`);
|
||||
if (i % 2 === 0) {
|
||||
arbiter.addRelation(`resource:${i}`, 'feature_flag', `flag:${flagId}`);
|
||||
}
|
||||
}
|
||||
arbiter.setRelationConfig('has_flag', { type: 'direct' });
|
||||
arbiter.setRelationConfig('feature_flag', { type: 'direct' });
|
||||
arbiter.setRelationConfig('can_feature', {
|
||||
type: 'tuple_to_userset',
|
||||
tuplesetRelation: 'feature_flag',
|
||||
tuplesetDirection: 'out',
|
||||
computedRelation: 'has_flag'
|
||||
});
|
||||
}
|
||||
|
||||
function buildComparatorQueries(size, rng, samples, trueIds, falseIds) {
|
||||
const queries = [];
|
||||
const half = Math.floor(samples / 2);
|
||||
for (let i = 0; i < half; i++) {
|
||||
const userId = trueIds.length ? trueIds[randInt(rng, trueIds.length)] : randInt(rng, size);
|
||||
queries.push({ userKey: `user:${userId}`, objectKey: `resource:${userId}` });
|
||||
}
|
||||
for (let i = 0; i < samples - half; i++) {
|
||||
const userId = falseIds.length ? falseIds[randInt(rng, falseIds.length)] : randInt(rng, size);
|
||||
queries.push({ userKey: `user:${userId}`, objectKey: `resource:${userId}` });
|
||||
}
|
||||
return queries;
|
||||
}
|
||||
|
||||
function buildFeatureFlagQueries(size, rng, samples) {
|
||||
const queries = [];
|
||||
const half = Math.floor(samples / 2);
|
||||
const flagCount = Math.max(1, Math.floor(size / 10));
|
||||
for (let i = 0; i < half; i++) {
|
||||
const userId = randInt(rng, Math.floor(size / 2)) * 2;
|
||||
queries.push({ userKey: `user:${userId}`, objectKey: `resource:${userId}`, flagId: userId % flagCount });
|
||||
}
|
||||
for (let i = 0; i < samples - half; i++) {
|
||||
const userId = randInt(rng, Math.floor(size / 2)) * 2 + 1;
|
||||
queries.push({ userKey: `user:${userId}`, objectKey: `resource:${userId}`, flagId: userId % flagCount });
|
||||
}
|
||||
return queries;
|
||||
}
|
||||
|
||||
const args = parseArgs(process.argv);
|
||||
const scenario = String(args.get('scenario') || 'comparator');
|
||||
const size = Number(args.get('size') || 5000);
|
||||
const samples = Number(args.get('samples') || 2000);
|
||||
const iterations = Number(args.get('iterations') || 20000);
|
||||
const seed = Number(args.get('seed') || 42);
|
||||
const includeMeta = !args.has('no-meta');
|
||||
const collectValues = args.has('values-agg') ? true : !args.has('no-values');
|
||||
|
||||
const arbiter = new Arbiter();
|
||||
const rng = createRng(seed);
|
||||
const options = { fastPath: true, includeMeta, collectValues };
|
||||
|
||||
let queries = [];
|
||||
if (scenario === 'feature-flag') {
|
||||
buildFeatureFlagScenario(arbiter, size);
|
||||
queries = buildFeatureFlagQueries(size, rng, samples);
|
||||
} else {
|
||||
const { trueIds, falseIds } = buildComparatorScenario(arbiter, size);
|
||||
queries = buildComparatorQueries(size, rng, samples, trueIds, falseIds);
|
||||
}
|
||||
|
||||
for (let i = 0; i < Math.min(200, samples); i++) {
|
||||
const q = queries[i];
|
||||
if (scenario === 'feature-flag') {
|
||||
arbiter.check(q.userKey, 'can_feature', q.objectKey, options);
|
||||
} else {
|
||||
arbiter.check(q.userKey, 'risk_ok_owa', q.objectKey, options);
|
||||
}
|
||||
}
|
||||
|
||||
const start = process.hrtime.bigint();
|
||||
for (let i = 0; i < iterations; i++) {
|
||||
const q = queries[randInt(rng, queries.length)];
|
||||
if (scenario === 'feature-flag') {
|
||||
arbiter.check(q.userKey, 'can_feature', q.objectKey, options);
|
||||
} else {
|
||||
arbiter.check(q.userKey, 'risk_ok_owa', q.objectKey, options);
|
||||
}
|
||||
}
|
||||
const elapsedMs = Number(process.hrtime.bigint() - start) / 1e6;
|
||||
const qps = elapsedMs > 0 ? Math.round((iterations / elapsedMs) * 1000) : 0;
|
||||
|
||||
console.log('profile_targeted_bench');
|
||||
console.log(`scenario=${scenario}`);
|
||||
console.log(`size=${size}`);
|
||||
console.log(`samples=${samples}`);
|
||||
console.log(`iterations=${iterations}`);
|
||||
console.log(`meta=${includeMeta ? 'on' : 'off'}`);
|
||||
console.log(`values=${collectValues ? 'on' : 'off'}`);
|
||||
console.log(`elapsed_ms=${Math.round(elapsedMs)}`);
|
||||
console.log(`qps=${qps}`);
|
||||
Reference in New Issue
Block a user