@webdecoy/ai-protection 0.1.0-alpha.4 → 0.1.0-alpha.5
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/MCP.md +7 -4
- package/README.md +6 -0
- package/WORK.md +123 -0
- package/action-runtime.mjs +11 -5
- package/actions.d.mts +18 -0
- package/actions.mjs +16 -9
- package/package.json +4 -2
- package/work.mjs +53 -0
package/MCP.md
CHANGED
|
@@ -8,10 +8,10 @@ resource authorization.
|
|
|
8
8
|
|
|
9
9
|
## Install
|
|
10
10
|
|
|
11
|
-
Available in `0.1.0-alpha.
|
|
11
|
+
Available in `0.1.0-alpha.5` and later compatible alpha releases:
|
|
12
12
|
|
|
13
13
|
```sh
|
|
14
|
-
npm install @webdecoy/ai-protection@0.1.0-alpha.
|
|
14
|
+
npm install @webdecoy/ai-protection@0.1.0-alpha.5 @modelcontextprotocol/sdk@1.31.0
|
|
15
15
|
```
|
|
16
16
|
|
|
17
17
|
Requires Node 22.22.3+ and MCP SDK **1.31.0**. The MCP SDK is an optional peer, so
|
|
@@ -111,9 +111,12 @@ have unknown work. Never retry writes without application/provider idempotency.
|
|
|
111
111
|
405. Reconnects do not grant fresh identity allowances.
|
|
112
112
|
- Resources, prompts, tasks, sampling, elicitation, other routes and pre-existing
|
|
113
113
|
MCP handlers are not wrapped. Unregistered methods/tools do not dispatch.
|
|
114
|
-
Weighted
|
|
115
|
-
|
|
114
|
+
Weighted tool work is available in the alpha API; see [bounded work](WORK.md)
|
|
115
|
+
for its runtime prerequisite and application-enforced bounds. Full product
|
|
116
|
+
acceptance remains separate from this adapter's tested contract.
|
|
116
117
|
|
|
117
118
|
This is an explicit tools integration, not transparent protection of an existing
|
|
118
119
|
whole MCP server. See the [MCP transport specification](https://modelcontextprotocol.io/specification/2025-11-25/basic/transports)
|
|
119
120
|
and [authorization specification](https://modelcontextprotocol.io/specification/2025-11-25/basic/authorization).
|
|
121
|
+
|
|
122
|
+
For weighted search/export limits and tenant concurrency, see [bounded tool work](WORK.md).
|
package/README.md
CHANGED
|
@@ -328,3 +328,9 @@ and maps tenant membership. The [MCP adapter](MCP.md) provides a separate `/mcp`
|
|
|
328
328
|
HTTP tool dispatch starting in `0.1.0-alpha.4`.
|
|
329
329
|
It requires the optional, pinned MCP SDK peer. Optional shared quotas, concurrency and hosted
|
|
330
330
|
action events are documented in the action guide and use the hosted AI Protection runtime and dashboard.
|
|
331
|
+
|
|
332
|
+
## Weighted tool work (Alpha)
|
|
333
|
+
|
|
334
|
+
Node alpha.5 adds weighted tool-work reservations and tenant concurrency. See
|
|
335
|
+
[bounded tool work](WORK.md) for installation, enforced application bounds and
|
|
336
|
+
retry/unknown-outcome semantics. These units are separate from model usage and billing.
|
package/WORK.md
ADDED
|
@@ -0,0 +1,123 @@
|
|
|
1
|
+
# Bounded tool work (Alpha)
|
|
2
|
+
|
|
3
|
+
The Node action and MCP APIs support optional weighted tool-work reservations.
|
|
4
|
+
Available in `@webdecoy/ai-protection@0.1.0-alpha.5`. Requires the
|
|
5
|
+
`/api/v1/sdk/ai-abuse/work` contract enabled on the hosted runtime. Go/Python model budgets
|
|
6
|
+
and invocation controls remain available, but do not expose this new weighted
|
|
7
|
+
operation API. The first integration is the TypeScript MCP tools adapter.
|
|
8
|
+
|
|
9
|
+
## Configure at the action, before execution
|
|
10
|
+
|
|
11
|
+
```js
|
|
12
|
+
limits: {
|
|
13
|
+
work: {
|
|
14
|
+
ruleId: 'search_work_v1',
|
|
15
|
+
windowSeconds: 60,
|
|
16
|
+
maxUnits: 11, // one fixed unit plus at most five rows at two units each
|
|
17
|
+
limits: {caller: 22, tenant: 44, tool: 88},
|
|
18
|
+
mode: 'enforce',
|
|
19
|
+
failureMode: 'closed',
|
|
20
|
+
measure: result => result.workUnits, // server-computed, synchronous, confirmed
|
|
21
|
+
},
|
|
22
|
+
concurrency: {
|
|
23
|
+
ruleId: 'search_parallel_v1', accountLimit: 1, featureLimit: 10,
|
|
24
|
+
mode: 'enforce', failureMode: 'closed',
|
|
25
|
+
},
|
|
26
|
+
tenantConcurrency: {
|
|
27
|
+
ruleId: 'tenant_search_parallel_v1', accountLimit: 2, featureLimit: 10,
|
|
28
|
+
mode: 'enforce', failureMode: 'closed',
|
|
29
|
+
},
|
|
30
|
+
}
|
|
31
|
+
```
|
|
32
|
+
|
|
33
|
+
Use `sharedRuntime` on the action/MCP handler. No browser credentials, prompts,
|
|
34
|
+
arguments or results are sent to the work service. Verified caller identity and
|
|
35
|
+
tenant are hashed; an HMAC binds the action, policy version and canonical arguments
|
|
36
|
+
to the operation. Use a stable server-only subject secret. Changing that secret or
|
|
37
|
+
rule IDs creates new accounting identities; rotate/drain deliberately.
|
|
38
|
+
|
|
39
|
+
All three unit allowances are evaluated atomically across runtime replicas before
|
|
40
|
+
callback dispatch. `caller` is scoped to issuer/tenant/subject, `tenant` is scoped
|
|
41
|
+
to issuer/tenant, and `tool` is shared within this property's rule. Zero/omitted
|
|
42
|
+
limits disable that dimension; at least one positive limit is required. Distinct
|
|
43
|
+
tools need distinct rule IDs. Reconnecting or creating a new MCP session does not
|
|
44
|
+
reset the identity allowance. Invocation quotas count attempts separately; model
|
|
45
|
+
budgets count tokens separately. These units never become a WebDecoy billing meter.
|
|
46
|
+
|
|
47
|
+
Existing `concurrency.accountLimit` is per caller; `tenantConcurrency.accountLimit`
|
|
48
|
+
is per tenant. Each `featureLimit` is property/rule-wide. Admission denial releases
|
|
49
|
+
already acquired slots; unconfirmed work keeps its slots until lease expiry.
|
|
50
|
+
Lease expiry requests cooperative cancellation, not proof a remote operation stopped.
|
|
51
|
+
|
|
52
|
+
## Enforce resource bounds in the application
|
|
53
|
+
|
|
54
|
+
`maxUnits` must be a conservative upper bound you actually enforce. The SDK provides
|
|
55
|
+
`context.work.maxUnits` to the callback, but cannot constrain arbitrary database or
|
|
56
|
+
storage work. Validate requested bounds before dispatch, apply tenant predicates and
|
|
57
|
+
row limits at the data source, and check byte lengths **before** appending/sending
|
|
58
|
+
export payload. Do not fetch an unbounded result and trim it afterward.
|
|
59
|
+
|
|
60
|
+
The [bounded search/export example](examples/mcp/work-tools.mjs) implements:
|
|
61
|
+
|
|
62
|
+
- Search: at most five rows, weight `1 + 2 × returned rows`, reserve 11 units.
|
|
63
|
+
- Export: at most 256 UTF-8 payload bytes, weight `16 + payload bytes`, reserve 272.
|
|
64
|
+
Transport framing/JSON encoding are not included in that payload-byte allowance.
|
|
65
|
+
- Forbidden admin: application authorization denies before any callback.
|
|
66
|
+
|
|
67
|
+
The synthetic source contains fixed small records. A real data source must enforce
|
|
68
|
+
its own database scan, CPU/time and storage-read bounds too; a row limit is not a
|
|
69
|
+
claim that a query scans only those rows. The separate weights use separate rules,
|
|
70
|
+
not interchangeable token or monetary estimates.
|
|
71
|
+
|
|
72
|
+
Run `node examples/mcp/start-work.mjs` after configuring the existing Auth0 example
|
|
73
|
+
variables and `WEBDECOY_URL`, `WEBDECOY_KEY`, `WEBDECOY_PROPERTY_ID`, and
|
|
74
|
+
`WEBDECOY_SUBJECT_SECRET`. The fixture binds to loopback and makes no model calls.
|
|
75
|
+
It requires an updated runtime. Replace its single-subject allowlist with current
|
|
76
|
+
application membership for an actual customer deployment.
|
|
77
|
+
|
|
78
|
+
## Settlement, failures and retries
|
|
79
|
+
|
|
80
|
+
After a completed callback, a synchronous `measure(result, context)` may report
|
|
81
|
+
confirmed integer usage between zero and `maxUnits`. Without it, the fixed maximum
|
|
82
|
+
weight is charged. Confirmed lower usage releases the difference in the original
|
|
83
|
+
admission window. Never measure from untrusted request claims. Errors, cancellation,
|
|
84
|
+
crashes, asynchronous/invalid measurement, excess usage and missing/failed settlement
|
|
85
|
+
keep the conservative maximum charged. Successful results survive reporting or
|
|
86
|
+
settlement failure and expose unknown accounting evidence.
|
|
87
|
+
|
|
88
|
+
Unknown work is never refunded just because a lease or settlement deadline expires.
|
|
89
|
+
A hard ceiling depends on application-enforced maximum work, enforce/closed settings,
|
|
90
|
+
and bounded completion. Observe/open can execute without a confirmed reservation and
|
|
91
|
+
must not claim a ceiling. Detector fail-open is separate from work-state failure.
|
|
92
|
+
|
|
93
|
+
A reservation has a server-generated timestamp/UUID operation ID. For recovery across
|
|
94
|
+
application retries, persist an ID from `createQuotaOperationId` (exported from
|
|
95
|
+
`@webdecoy/ai-protection/fetch`) in trusted application state, and provide it through
|
|
96
|
+
`work.operationId(context)`. Do not accept an arbitrary client ID or use MCP message,
|
|
97
|
+
agent or session IDs as this key. Reuse requires the same caller/tenant/tool, bounds,
|
|
98
|
+
policy and argument binding. Conflicts deny even in open mode.
|
|
99
|
+
|
|
100
|
+
The SDK makes no automatic work-RPC or callback retries. An identical reserve replay
|
|
101
|
+
returns accounting evidence but **never grants another dispatch**; the SDK returns
|
|
102
|
+
`work_replay` (409). Lost admission replies may therefore consume capacity without
|
|
103
|
+
executing work. A changed retry returns `work_conflict` (409). Return stored business
|
|
104
|
+
results through your own idempotency layer when appropriate. Never mint a new ID to
|
|
105
|
+
retry an unknown write blindly. Application/provider idempotency remains necessary;
|
|
106
|
+
this is not an exactly-once side-effect guarantee. Identical settlement is idempotent;
|
|
107
|
+
conflicting or over-bound settlement is rejected without lowering the charge.
|
|
108
|
+
|
|
109
|
+
## Windows and evidence
|
|
110
|
+
|
|
111
|
+
- Fixed UTC-aligned windows: 1–86,400 seconds. Boundary bursts are possible; use
|
|
112
|
+
concurrency controls separately. Limits/maxima: integer units up to 10^12.
|
|
113
|
+
- IDs: ten-minute admission lifetime, at most 30 seconds future skew; expired IDs
|
|
114
|
+
remain invalid even after receipts are purged. Settlement deadline: one hour.
|
|
115
|
+
- Receipts retained through the window end plus 24 hours. Unknown charge persists
|
|
116
|
+
for its admission window; this is not a lifetime allowance or infinite ledger.
|
|
117
|
+
- At most 32 work rules and 10,000 retained operations per property, including denials.
|
|
118
|
+
Capacity failures follow state-failure policy. Existing replay works at capacity.
|
|
119
|
+
These are operational bounds, not commercial plan changes.
|
|
120
|
+
- Reports show reserved/charged units, remaining allowance at admission and
|
|
121
|
+
reserved/settled/unknown/unavailable/denied/replay status. Remaining is a snapshot,
|
|
122
|
+
not current balance. Delivery is best effort; missing reports do not imply no work.
|
|
123
|
+
Local callback completion is not independent proof of a remote side effect.
|
package/action-runtime.mjs
CHANGED
|
@@ -1,3 +1,4 @@
|
|
|
1
|
+
import {prepareWork} from './work.mjs';
|
|
1
2
|
import {prepareQuota,quotaHash} from './quota.mjs';
|
|
2
3
|
import {prepareConcurrency} from './concurrency.mjs';
|
|
3
4
|
import {validPropertyID} from './account.mjs';
|
|
@@ -20,10 +21,15 @@ export function prepareActionRuntime(options, definitions) {
|
|
|
20
21
|
const gate=prepareQuota({...c,accountQuota:{...q,idempotency:false,operationId:undefined,sessionLimit:0,subject:ctx=>subject(ctx,tenant)}});
|
|
21
22
|
gates.push(async(ctx,signal)=>{const r=await gate(ctx,signal);if(r.check)r.check.id=tenant?'tenant_quota':'caller_quota';return r;});
|
|
22
23
|
}
|
|
23
|
-
|
|
24
|
-
|
|
25
|
-
|
|
26
|
-
|
|
24
|
+
const concurrencies=[];
|
|
25
|
+
for(const [key,tenant] of [['concurrency',false],['tenantConcurrency',true]])if(l[key]){
|
|
26
|
+
const option=l[key];if(ruleIDs.has(option.ruleId))throw Error('Action limits require distinct rule IDs');ruleIDs.add(option.ruleId);
|
|
27
|
+
const gate=prepareConcurrency({...c,concurrency:{...option,subject:ctx=>subject(ctx,tenant)}});
|
|
28
|
+
concurrencies.push(async(ctx,signal)=>{const r=await gate(ctx,signal);if(tenant)r.check.id='tenant_concurrency';return r;});
|
|
29
|
+
}
|
|
30
|
+
let work=null;
|
|
31
|
+
if(l.work){if(ruleIDs.has(l.work.ruleId))throw Error('Action limits require distinct rule IDs');ruleIDs.add(l.work.ruleId);work=prepareWork(c,l.work,name,options.policyVersion);}
|
|
32
|
+
limits.set(name,{gates,concurrencies,work});
|
|
27
33
|
}
|
|
28
34
|
const reporter=createReporter({reportingTimeoutMs:c.reportingTimeoutMs??1000,maxPendingReports:c.maxPendingReports??100,
|
|
29
35
|
onObservation:async(event,{signal})=>{
|
|
@@ -32,7 +38,7 @@ export function prepareActionRuntime(options, definitions) {
|
|
|
32
38
|
const payload={schema:2,request_id:event.eventId,timestamp:event.timestamp,decision:event.decision,reason:event.reason,
|
|
33
39
|
degraded:event.checks.some(c=>c.decision==='unavailable'),checks,handler_attempted:event.attempted,
|
|
34
40
|
action:event.decision==='deny'?'denied':event.outcome==='unknown'?'handler_error':'forwarded',
|
|
35
|
-
tool_action:{action_id:event.actionId,name:event.action,policy_version:event.policyVersion,outcome:event.outcome}};
|
|
41
|
+
tool_action:{action_id:event.actionId,name:event.action,policy_version:event.policyVersion,outcome:event.outcome,...(event.work?{work:event.work}:{})}};
|
|
36
42
|
const response=await fetch(new URL('/api/v1/sdk/ai-abuse/reports',url),{method:'POST',redirect:'error',signal,
|
|
37
43
|
headers:{Authorization:`Bearer ${c.webdecoyKey}`,'X-WebDecoy-Property-ID':c.propertyId,'Content-Type':'application/json'},body:JSON.stringify(payload)});
|
|
38
44
|
await response.body?.cancel();if(!response.ok)throw Error('Action reporting unavailable');
|
package/actions.d.mts
CHANGED
|
@@ -13,11 +13,27 @@ export interface TrustedCaller {
|
|
|
13
13
|
}
|
|
14
14
|
export type ActionInput = null | boolean | number | string | readonly ActionInput[] | {readonly [key:string]: ActionInput};
|
|
15
15
|
export interface ActionContext {
|
|
16
|
+
readonly work?: {readonly maxUnits:number};
|
|
16
17
|
readonly caller: TrustedCaller;
|
|
17
18
|
readonly args: ActionInput;
|
|
18
19
|
readonly signal?: AbortSignal;
|
|
19
20
|
}
|
|
21
|
+
export interface ActionWorkEvidence {
|
|
22
|
+
readonly rule_id:string;readonly mode:'observe'|'enforce';
|
|
23
|
+
readonly status:'reserved'|'settled'|'unknown'|'unavailable'|'denied'|'replay';
|
|
24
|
+
readonly reserved_units:number;readonly charged_units?:number;readonly remaining_units?:number;
|
|
25
|
+
}
|
|
26
|
+
export interface ActionWork {
|
|
27
|
+
ruleId:string;windowSeconds:number;maxUnits:number;
|
|
28
|
+
limits:{caller?:number;tenant?:number;tool?:number};
|
|
29
|
+
mode?:'observe'|'enforce';failureMode?:'open'|'closed';timeoutMs?:number;
|
|
30
|
+
/** Trusted server-generated persisted ID. Never use MCP request/session IDs or raw user input. */
|
|
31
|
+
operationId?(context:ActionContext):string;
|
|
32
|
+
/** Confirmed work after completed execution. No async/detached measurement; excess/unknown retains maximum. */
|
|
33
|
+
measure?(result:unknown,context:ActionContext):number;
|
|
34
|
+
}
|
|
20
35
|
export interface ActionEvent {
|
|
36
|
+
readonly work?:ActionWorkEvidence;
|
|
21
37
|
readonly schema: 1;
|
|
22
38
|
readonly eventId: string;
|
|
23
39
|
readonly timestamp: string;
|
|
@@ -33,6 +49,8 @@ export interface ActionEvent {
|
|
|
33
49
|
}
|
|
34
50
|
export interface ActionQuota {ruleId:string;limit:number;windowSeconds:number;mode?:'observe'|'enforce';failureMode?:'open'|'closed';timeoutMs?:number}
|
|
35
51
|
export interface ActionLimits {
|
|
52
|
+
tenantConcurrency?: ActionLimits['concurrency'];
|
|
53
|
+
work?:ActionWork;
|
|
36
54
|
callerQuota?: ActionQuota;
|
|
37
55
|
tenantQuota?: ActionQuota;
|
|
38
56
|
concurrency?: {ruleId:string;accountLimit:number;featureLimit:number;mode?:'observe'|'enforce';failureMode?:'open'|'closed';ttlSeconds?:number;maxSeconds?:number;timeoutMs?:number};
|
package/actions.mjs
CHANGED
|
@@ -59,7 +59,7 @@ function inputSnapshot(input) {
|
|
|
59
59
|
return result;
|
|
60
60
|
}
|
|
61
61
|
|
|
62
|
-
/**
|
|
62
|
+
/** Protected dispatch with local permissions and optional shared controls. Never retries execution. */
|
|
63
63
|
export function createActionProtection(options) {
|
|
64
64
|
if (!options || typeof options.authenticate !== 'function' || (typeof options.policyVersion !== 'string' || !token.test(options.policyVersion))) throw Error('Invalid action configuration');
|
|
65
65
|
const authenticate = options.authenticate, policyVersion = options.policyVersion, sink = options.onEvent;
|
|
@@ -81,7 +81,8 @@ export function createActionProtection(options) {
|
|
|
81
81
|
const actionId = randomUUID();
|
|
82
82
|
// Unknown caller-controlled action strings are never placed in evidence.
|
|
83
83
|
const action = actions.get(name), eventAction = action ? name : 'unregistered';
|
|
84
|
-
let attempted = false,completed=false,lease;
|
|
84
|
+
let attempted = false,completed=false,lease,work;
|
|
85
|
+
const leases=[];
|
|
85
86
|
const checks=[];
|
|
86
87
|
const deadline = new AbortController();
|
|
87
88
|
const admissionSignal = signal ? AbortSignal.any([signal, deadline.signal]) : deadline.signal;
|
|
@@ -90,7 +91,7 @@ export function createActionProtection(options) {
|
|
|
90
91
|
function emit(decision, reason, outcome) {
|
|
91
92
|
|
|
92
93
|
const event = Object.freeze({schema:1, eventId:randomUUID(), timestamp:new Date().toISOString(), actionId, action:eventAction, policyVersion,
|
|
93
|
-
evaluation:'local', decision, reason, attempted, outcome,checks:Object.freeze(checks.map(c=>Object.freeze({...c})))});
|
|
94
|
+
evaluation:'local', decision, reason, attempted, outcome,...(work?{work:Object.freeze({...work.evidence})}:{}),checks:Object.freeze(checks.map(c=>Object.freeze({...c})))});
|
|
94
95
|
if(runtime)void runtime.report(event);
|
|
95
96
|
if(!sink||pendingEvents>=100)return;
|
|
96
97
|
pendingEvents++;
|
|
@@ -131,38 +132,44 @@ export function createActionProtection(options) {
|
|
|
131
132
|
const r=await gate(context,admissionSignal);checks.push(r.check);
|
|
132
133
|
if(r.denial)deny(r.denial.reason,r.denial.status,r.denial.retryAfterSeconds);
|
|
133
134
|
}
|
|
134
|
-
|
|
135
|
-
lease=await
|
|
135
|
+
for(const gate of controls?.concurrencies??[]){
|
|
136
|
+
lease=await gate(context,lease?.signal??admissionSignal);leases.push(lease);checks.push(lease.check);
|
|
136
137
|
if(lease.denial)deny(lease.denial.reason,lease.denial.status,lease.denial.retryAfterSeconds);
|
|
137
138
|
lease.signal.throwIfAborted();
|
|
138
139
|
}
|
|
140
|
+
if(controls?.work){
|
|
141
|
+
work=await controls.work(context,lease?.signal??admissionSignal);checks.push(work.check);
|
|
142
|
+
if(work.denial)deny(work.denial.reason,work.denial.status,work.denial.retryAfterSeconds);
|
|
143
|
+
}
|
|
139
144
|
// Authentication may expire while ownership/policy checks are running.
|
|
140
145
|
if (caller.expiresAt <= Date.now()) deny('authentication_expired',401);
|
|
141
146
|
cancelled();
|
|
142
147
|
clearTimeout(timer);
|
|
143
148
|
attempted = true;
|
|
144
149
|
emit('allow','authorized','attempted');
|
|
145
|
-
const execution=Object.freeze({...context,signal:lease?.signal??context.signal});
|
|
150
|
+
const execution=Object.freeze({...context,...(work?{work:work.bound}:{}),signal:lease?.signal??context.signal});
|
|
146
151
|
const result = await action.execute(execution);
|
|
147
152
|
execution.signal.throwIfAborted();
|
|
148
153
|
// Resolution is application completion, not proof of an external side effect.
|
|
149
154
|
cancelled();
|
|
150
155
|
completed=true;
|
|
151
|
-
if(
|
|
156
|
+
if(work){const check=await work.finish(result,true);if(check)checks.push(check);}
|
|
157
|
+
for(const lease of [...leases].reverse())if(!lease.denial){
|
|
152
158
|
const began=performance.now();
|
|
153
|
-
try{await lease.finish(true);}catch{checks.push({id:'concurrency_release',source:'shared',mode:lease.check.mode,decision:'unavailable',reason:'concurrency_release_unavailable',durationMs:performance.now()-began});}
|
|
159
|
+
try{await lease.finish(true);}catch{checks.push({id:lease.check.id==='tenant_concurrency'?'tenant_concurrency_release':'concurrency_release',source:'shared',mode:lease.check.mode,decision:'unavailable',reason:'concurrency_release_unavailable',durationMs:performance.now()-began});}
|
|
154
160
|
}
|
|
155
161
|
emit('allow','authorized','completed');
|
|
156
162
|
return result;
|
|
157
163
|
} catch (error) {
|
|
158
164
|
if (!attempted && deadline.signal.aborted && !signal?.aborted) deny('admission_timeout',503);
|
|
165
|
+
if(work&&!completed)await work.finish(undefined,false);
|
|
159
166
|
if (attempted) emit('allow',signal?.aborted ? 'execution_cancelled' : 'execution_failed','unknown');
|
|
160
167
|
else if (!(error instanceof ActionDenied)) emit('deny','admission_cancelled','not_attempted');
|
|
161
168
|
throw error;
|
|
162
169
|
} finally {
|
|
163
170
|
clearTimeout(timer);
|
|
164
171
|
// Only confirmed completion (or no callback) releases; uncertain work holds.
|
|
165
|
-
if(lease
|
|
172
|
+
for(const lease of [...leases].reverse())if(!lease.denial)await lease.finish(!attempted||completed).catch(()=>{});
|
|
166
173
|
}
|
|
167
174
|
}});
|
|
168
175
|
}
|
package/package.json
CHANGED
|
@@ -10,7 +10,7 @@
|
|
|
10
10
|
"check:package": "node scripts/check-package.mjs",
|
|
11
11
|
"test:types": "tsc --strict --noEmit --module nodenext --target es2022 test/types.mts test/mcp-types.mts"
|
|
12
12
|
},
|
|
13
|
-
"version": "0.1.0-alpha.
|
|
13
|
+
"version": "0.1.0-alpha.5",
|
|
14
14
|
"exports": {
|
|
15
15
|
"./fetch": {
|
|
16
16
|
"types": "./fetch.d.mts",
|
|
@@ -60,7 +60,9 @@
|
|
|
60
60
|
"action-runtime.mjs",
|
|
61
61
|
"mcp.mjs",
|
|
62
62
|
"mcp.d.mts",
|
|
63
|
-
"MCP.md"
|
|
63
|
+
"MCP.md",
|
|
64
|
+
"work.mjs",
|
|
65
|
+
"WORK.md"
|
|
64
66
|
],
|
|
65
67
|
"description": "Bot and abuse protection for AI-powered applications. Server-side request admission for Node.js.",
|
|
66
68
|
"license": "Apache-2.0",
|
package/work.mjs
ADDED
|
@@ -0,0 +1,53 @@
|
|
|
1
|
+
import {createQuotaOperationId,quotaHash} from './quota.mjs';
|
|
2
|
+
import {readJSON} from './transport.mjs';
|
|
3
|
+
const code=/^[a-z][a-z0-9_]{0,63}$/;
|
|
4
|
+
const integer=n=>Number.isSafeInteger(n)&&n>=0&&n<=1e12;
|
|
5
|
+
const canonical=value=>JSON.stringify(value===null||typeof value!=='object'?value:Array.isArray(value)?value.map(v=>JSON.parse(canonical(v))):Object.fromEntries(Object.keys(value).sort().map(k=>[k,JSON.parse(canonical(value[k]))])));
|
|
6
|
+
export function prepareWork(config,definition,name,policyVersion){
|
|
7
|
+
const w={mode:'observe',failureMode:'open',timeoutMs:1000,...definition};
|
|
8
|
+
const limits=Object.fromEntries(['caller','tenant','tool'].map(k=>[k,w.limits?.[k]??0]));
|
|
9
|
+
if(!code.test(w.ruleId??'')||!integer(w.maxUnits)||w.maxUnits<1||!integer(w.windowSeconds)||w.windowSeconds<1||w.windowSeconds>86400||!integer(w.timeoutMs)||w.timeoutMs<1||w.timeoutMs>10000||!['observe','enforce'].includes(w.mode)||!['open','closed'].includes(w.failureMode)||!Object.values(limits).every(integer)||!Object.values(limits).some(n=>n>0)||(w.measure!==undefined&&typeof w.measure!=='function')||(w.operationId!==undefined&&typeof w.operationId!=='function'))throw Error('Invalid tool work policy');
|
|
10
|
+
async function rpc(body,signal){
|
|
11
|
+
const response=await fetch(new URL('/api/v1/sdk/ai-abuse/work',config.webdecoyUrl),{method:'POST',redirect:'error',signal:AbortSignal.any([signal??new AbortController().signal,AbortSignal.timeout(w.timeoutMs)]),headers:{Authorization:`Bearer ${config.webdecoyKey}`,'X-WebDecoy-Property-ID':config.propertyId,'Content-Type':'application/json'},body:JSON.stringify(body)});
|
|
12
|
+
if(!response.ok){await response.body?.cancel();const e=Error('Work unavailable');e.status=response.status;throw e;}
|
|
13
|
+
const r=await readJSON(response,2048);
|
|
14
|
+
if(r?.schema!==1||!['allowed','granted','replay','settled'].every(k=>typeof r[k]==='boolean')||!['reserved_units','charged_units','remaining_units','retry_after_seconds'].every(k=>integer(r[k]))||r.retry_after_seconds>86400||!code.test(r.reason??''))throw Error('Invalid work response');
|
|
15
|
+
return r;
|
|
16
|
+
}
|
|
17
|
+
return async(ctx,signal)=>{
|
|
18
|
+
let operationId=w.operationId?w.operationId(ctx):createQuotaOperationId();
|
|
19
|
+
if(operationId&&typeof operationId.then==='function')Promise.resolve(operationId).catch(()=>{});
|
|
20
|
+
if(typeof operationId!=='string'||! /^[1-9][0-9]{9}\.[a-f0-9]{8}-[a-f0-9]{4}-4[a-f0-9]{3}-[89ab][a-f0-9]{3}-[a-f0-9]{12}$/.test(operationId))throw Error('Invalid trusted work operation ID');
|
|
21
|
+
const body={schema:1,operation:'reserve',operation_id:operationId,rule_id:w.ruleId,mode:w.mode,window_seconds:w.windowSeconds,limits,units:w.maxUnits,
|
|
22
|
+
subject:quotaHash(config.subjectSecret,'webdecoy.work.caller.v1',ctx.caller.issuer,ctx.caller.tenant,ctx.caller.subject),tenant:quotaHash(config.subjectSecret,'webdecoy.work.tenant.v1',ctx.caller.issuer,ctx.caller.tenant),binding:quotaHash(config.subjectSecret,'webdecoy.work.arguments.v1',name,policyVersion,canonical(ctx.args))};
|
|
23
|
+
const evidence={rule_id:w.ruleId,mode:w.mode,status:'unknown',reserved_units:w.maxUnits};
|
|
24
|
+
const check={id:'tool_work',source:'shared',mode:w.mode,decision:'allow',reason:'work_allowed',durationMs:0};
|
|
25
|
+
const began=performance.now();let grant,denial;
|
|
26
|
+
try{
|
|
27
|
+
grant=await rpc(body,signal);
|
|
28
|
+
if(grant.replay){denial={reason:'work_replay',status:409};evidence.status='replay';}
|
|
29
|
+
else if(!grant.granted){if(grant.reason!=='work_exceeded')throw Error('Invalid work denial');denial={reason:'work_exceeded',status:429,retryAfterSeconds:grant.retry_after_seconds};evidence.status='denied';}
|
|
30
|
+
else {if(!['work_allowed','work_exceeded'].includes(grant.reason)||(!grant.allowed&&w.mode==='enforce')||grant.reserved_units!==w.maxUnits||grant.charged_units!==w.maxUnits)throw Error('Invalid work grant');evidence.status='reserved';}
|
|
31
|
+
evidence.reserved_units=grant.reserved_units;evidence.charged_units=grant.charged_units;evidence.remaining_units=grant.remaining_units;
|
|
32
|
+
check.reason=grant.reason;check.decision=grant.allowed&&!grant.replay?'allow':'deny';
|
|
33
|
+
}catch(e){
|
|
34
|
+
grant=undefined;
|
|
35
|
+
signal?.throwIfAborted();check.decision='unavailable';check.reason='work_outcome_unknown';evidence.status='unavailable';
|
|
36
|
+
if([400,403,409,410].includes(e.status)){denial={reason:e.status===409?'work_conflict':'work_operation_rejected',status:e.status};check.decision='deny';check.reason=denial.reason;}
|
|
37
|
+
else if(w.mode==='enforce'&&w.failureMode==='closed')denial={reason:'work_outcome_unknown',status:503};
|
|
38
|
+
}
|
|
39
|
+
check.durationMs=performance.now()-began;
|
|
40
|
+
return {check,denial,evidence,bound:Object.freeze({maxUnits:w.maxUnits}),async finish(result,completed){
|
|
41
|
+
if(!grant?.granted||grant.replay)return;
|
|
42
|
+
if(!completed){evidence.status='unknown';return;}
|
|
43
|
+
try{
|
|
44
|
+
const units=w.measure?w.measure(result,ctx):w.maxUnits;
|
|
45
|
+
if(units&&typeof units.then==='function')Promise.resolve(units).catch(()=>{});
|
|
46
|
+
if(!integer(units)||units>w.maxUnits)throw Error('Unconfirmed or excess work');
|
|
47
|
+
const settled=await rpc({...body,operation:'settle',units});
|
|
48
|
+
if(!settled.settled||settled.reason!=='work_settled'||settled.charged_units!==units)throw Error('Work settlement unavailable');
|
|
49
|
+
evidence.status='settled';evidence.charged_units=units;
|
|
50
|
+
}catch{evidence.status='unknown';return {id:'tool_work_settlement',source:'shared',mode:w.mode,decision:'unavailable',reason:'work_usage_unknown',durationMs:0};}
|
|
51
|
+
}};
|
|
52
|
+
};
|
|
53
|
+
}
|