@aion0/bastion 0.1.15 → 0.1.16
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/config/default.yaml +17 -0
- package/dist/cli/commands/start.d.ts.map +1 -1
- package/dist/cli/commands/start.js +4 -0
- package/dist/cli/commands/start.js.map +1 -1
- package/dist/config/schema.d.ts +21 -1
- package/dist/config/schema.d.ts.map +1 -1
- package/dist/core/bootstrap.d.ts.map +1 -1
- package/dist/core/bootstrap.js +30 -0
- package/dist/core/bootstrap.js.map +1 -1
- package/dist/dashboard/api-routes.d.ts.map +1 -1
- package/dist/dashboard/api-routes.js +338 -8
- package/dist/dashboard/api-routes.js.map +1 -1
- package/dist/dashboard/page.d.ts.map +1 -1
- package/dist/dashboard/page.js +646 -109
- package/dist/dashboard/page.js.map +1 -1
- package/dist/dlp/actions.d.ts +2 -0
- package/dist/dlp/actions.d.ts.map +1 -1
- package/dist/dlp/ai-validator.d.ts +15 -1
- package/dist/dlp/ai-validator.d.ts.map +1 -1
- package/dist/dlp/ai-validator.js +84 -3
- package/dist/dlp/ai-validator.js.map +1 -1
- package/dist/dlp/engine.d.ts +5 -1
- package/dist/dlp/engine.d.ts.map +1 -1
- package/dist/dlp/engine.js +25 -7
- package/dist/dlp/engine.js.map +1 -1
- package/dist/dlp/message-cache.js +2 -2
- package/dist/dlp/message-cache.js.map +1 -1
- package/dist/plugins/builtin/rate-limiter.d.ts +24 -0
- package/dist/plugins/builtin/rate-limiter.d.ts.map +1 -0
- package/dist/plugins/builtin/rate-limiter.js +248 -0
- package/dist/plugins/builtin/rate-limiter.js.map +1 -0
- package/dist/plugins/builtin/threat-scorer.d.ts.map +1 -1
- package/dist/plugins/builtin/threat-scorer.js +13 -1
- package/dist/plugins/builtin/threat-scorer.js.map +1 -1
- package/dist/plugins/builtin/tool-guard.d.ts +18 -0
- package/dist/plugins/builtin/tool-guard.d.ts.map +1 -1
- package/dist/plugins/builtin/tool-guard.js +148 -8
- package/dist/plugins/builtin/tool-guard.js.map +1 -1
- package/dist/plugins/types.d.ts +3 -0
- package/dist/plugins/types.d.ts.map +1 -1
- package/package.json +1 -1
|
@@ -0,0 +1,248 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.createRateLimiterPlugin = createRateLimiterPlugin;
|
|
4
|
+
const collector_js_1 = require("../../metrics/collector.js");
|
|
5
|
+
const logger_js_1 = require("../../utils/logger.js");
|
|
6
|
+
const log = (0, logger_js_1.createLogger)('rate-limiter');
|
|
7
|
+
function getHourStart(now) {
|
|
8
|
+
const d = new Date(now);
|
|
9
|
+
d.setMinutes(0, 0, 0);
|
|
10
|
+
return d.getTime();
|
|
11
|
+
}
|
|
12
|
+
function getDayStart(now) {
|
|
13
|
+
const d = new Date(now);
|
|
14
|
+
d.setHours(0, 0, 0, 0);
|
|
15
|
+
return d.getTime();
|
|
16
|
+
}
|
|
17
|
+
function getMonthStart(now) {
|
|
18
|
+
const d = new Date(now);
|
|
19
|
+
d.setDate(1);
|
|
20
|
+
d.setHours(0, 0, 0, 0);
|
|
21
|
+
return d.getTime();
|
|
22
|
+
}
|
|
23
|
+
function createRateLimiterPlugin(db, getLiveConfig, eventBus) {
|
|
24
|
+
const now = Date.now();
|
|
25
|
+
const state = {
|
|
26
|
+
requestTimestamps: [],
|
|
27
|
+
costHour: 0,
|
|
28
|
+
tokensHour: 0,
|
|
29
|
+
hourStart: getHourStart(now),
|
|
30
|
+
costDay: 0,
|
|
31
|
+
dayStart: getDayStart(now),
|
|
32
|
+
costMonth: 0,
|
|
33
|
+
monthStart: getMonthStart(now),
|
|
34
|
+
recentBlocks: 0,
|
|
35
|
+
};
|
|
36
|
+
// Recover counters from DB on startup
|
|
37
|
+
try {
|
|
38
|
+
const hourAgo = new Date(state.hourStart).toISOString();
|
|
39
|
+
const dayAgo = new Date(state.dayStart).toISOString();
|
|
40
|
+
const monthAgo = new Date(state.monthStart).toISOString();
|
|
41
|
+
const hourRow = db.prepare(`SELECT COALESCE(SUM(cost_usd),0) as cost, COALESCE(SUM(input_tokens + output_tokens),0) as tokens FROM requests WHERE created_at >= ?`).get(hourAgo);
|
|
42
|
+
const dayRow = db.prepare(`SELECT COALESCE(SUM(cost_usd),0) as cost FROM requests WHERE created_at >= ?`).get(dayAgo);
|
|
43
|
+
const monthRow = db.prepare(`SELECT COALESCE(SUM(cost_usd),0) as cost FROM requests WHERE created_at >= ?`).get(monthAgo);
|
|
44
|
+
state.costHour = hourRow.cost;
|
|
45
|
+
state.tokensHour = hourRow.tokens;
|
|
46
|
+
state.costDay = dayRow.cost;
|
|
47
|
+
state.costMonth = monthRow.cost;
|
|
48
|
+
log.info('Recovered rate-limiter counters from DB', {
|
|
49
|
+
costHour: state.costHour.toFixed(4),
|
|
50
|
+
tokensHour: state.tokensHour,
|
|
51
|
+
costDay: state.costDay.toFixed(4),
|
|
52
|
+
costMonth: state.costMonth.toFixed(4),
|
|
53
|
+
});
|
|
54
|
+
}
|
|
55
|
+
catch (err) {
|
|
56
|
+
log.warn('Failed to recover rate-limiter counters', { error: err.message });
|
|
57
|
+
}
|
|
58
|
+
function rollPeriods() {
|
|
59
|
+
const now = Date.now();
|
|
60
|
+
const newHourStart = getHourStart(now);
|
|
61
|
+
if (newHourStart > state.hourStart) {
|
|
62
|
+
state.costHour = 0;
|
|
63
|
+
state.tokensHour = 0;
|
|
64
|
+
state.hourStart = newHourStart;
|
|
65
|
+
}
|
|
66
|
+
const newDayStart = getDayStart(now);
|
|
67
|
+
if (newDayStart > state.dayStart) {
|
|
68
|
+
state.costDay = 0;
|
|
69
|
+
state.dayStart = newDayStart;
|
|
70
|
+
}
|
|
71
|
+
const newMonthStart = getMonthStart(now);
|
|
72
|
+
if (newMonthStart > state.monthStart) {
|
|
73
|
+
state.costMonth = 0;
|
|
74
|
+
state.monthStart = newMonthStart;
|
|
75
|
+
}
|
|
76
|
+
}
|
|
77
|
+
function cleanRpmWindow() {
|
|
78
|
+
const cutoff = Date.now() - 60_000;
|
|
79
|
+
state.requestTimestamps = state.requestTimestamps.filter(t => t > cutoff);
|
|
80
|
+
return state.requestTimestamps.length;
|
|
81
|
+
}
|
|
82
|
+
function checkWarning(cfg, type, current, limit) {
|
|
83
|
+
if (limit <= 0)
|
|
84
|
+
return;
|
|
85
|
+
const pct = current / limit;
|
|
86
|
+
if (pct >= cfg.warningThreshold && pct < 1.0) {
|
|
87
|
+
eventBus.emit('rate-limit:warning', { type, current, limit, percentage: pct });
|
|
88
|
+
log.warn('Rate limit warning', { type, current, limit, percentage: Math.round(pct * 100) + '%' });
|
|
89
|
+
}
|
|
90
|
+
}
|
|
91
|
+
// Public state accessor for dashboard API
|
|
92
|
+
function getState() {
|
|
93
|
+
rollPeriods();
|
|
94
|
+
const cfg = getLiveConfig();
|
|
95
|
+
const currentRpm = cleanRpmWindow();
|
|
96
|
+
const limits = {};
|
|
97
|
+
if (cfg.requestsPerMinute > 0) {
|
|
98
|
+
limits.requestsPerMinute = {
|
|
99
|
+
limit: cfg.requestsPerMinute,
|
|
100
|
+
current: currentRpm,
|
|
101
|
+
percentage: currentRpm / cfg.requestsPerMinute,
|
|
102
|
+
};
|
|
103
|
+
}
|
|
104
|
+
if (cfg.tokensPerHour > 0) {
|
|
105
|
+
limits.tokensPerHour = {
|
|
106
|
+
limit: cfg.tokensPerHour,
|
|
107
|
+
current: state.tokensHour,
|
|
108
|
+
percentage: state.tokensHour / cfg.tokensPerHour,
|
|
109
|
+
};
|
|
110
|
+
}
|
|
111
|
+
if (cfg.maxCostPerHour > 0) {
|
|
112
|
+
limits.maxCostPerHour = {
|
|
113
|
+
limit: cfg.maxCostPerHour,
|
|
114
|
+
current: state.costHour,
|
|
115
|
+
percentage: state.costHour / cfg.maxCostPerHour,
|
|
116
|
+
};
|
|
117
|
+
}
|
|
118
|
+
if (cfg.maxCostPerDay > 0) {
|
|
119
|
+
limits.maxCostPerDay = {
|
|
120
|
+
limit: cfg.maxCostPerDay,
|
|
121
|
+
current: state.costDay,
|
|
122
|
+
percentage: state.costDay / cfg.maxCostPerDay,
|
|
123
|
+
};
|
|
124
|
+
}
|
|
125
|
+
if (cfg.maxCostPerMonth > 0) {
|
|
126
|
+
limits.maxCostPerMonth = {
|
|
127
|
+
limit: cfg.maxCostPerMonth,
|
|
128
|
+
current: state.costMonth,
|
|
129
|
+
percentage: state.costMonth / cfg.maxCostPerMonth,
|
|
130
|
+
};
|
|
131
|
+
}
|
|
132
|
+
return { limits, action: cfg.action, recentBlocks: state.recentBlocks };
|
|
133
|
+
}
|
|
134
|
+
// Expose state via a property on the plugin object
|
|
135
|
+
const plugin = {
|
|
136
|
+
name: 'rate-limiter',
|
|
137
|
+
priority: 2, // After metrics-collector (1), before DLP (3) and threat-scorer (4)
|
|
138
|
+
version: '1.0.0',
|
|
139
|
+
apiVersion: 2,
|
|
140
|
+
source: 'builtin',
|
|
141
|
+
getState,
|
|
142
|
+
async onRequest(_context) {
|
|
143
|
+
rollPeriods();
|
|
144
|
+
const cfg = getLiveConfig();
|
|
145
|
+
// 1. Check RPM
|
|
146
|
+
const currentRpm = cleanRpmWindow();
|
|
147
|
+
if (cfg.requestsPerMinute > 0) {
|
|
148
|
+
checkWarning(cfg, 'requestsPerMinute', currentRpm, cfg.requestsPerMinute);
|
|
149
|
+
if (currentRpm >= cfg.requestsPerMinute) {
|
|
150
|
+
const reason = `Rate limit exceeded: ${currentRpm}/${cfg.requestsPerMinute} requests per minute`;
|
|
151
|
+
state.recentBlocks++;
|
|
152
|
+
eventBus.emit('rate-limit:exceeded', {
|
|
153
|
+
type: 'requestsPerMinute', current: currentRpm, limit: cfg.requestsPerMinute, action: cfg.action,
|
|
154
|
+
});
|
|
155
|
+
if (cfg.action === 'block') {
|
|
156
|
+
log.warn(reason);
|
|
157
|
+
return { blocked: { reason } };
|
|
158
|
+
}
|
|
159
|
+
log.warn('Rate limit exceeded (warn mode)', { type: 'requestsPerMinute' });
|
|
160
|
+
}
|
|
161
|
+
}
|
|
162
|
+
// 2. Check tokens per hour
|
|
163
|
+
if (cfg.tokensPerHour > 0) {
|
|
164
|
+
checkWarning(cfg, 'tokensPerHour', state.tokensHour, cfg.tokensPerHour);
|
|
165
|
+
if (state.tokensHour >= cfg.tokensPerHour) {
|
|
166
|
+
const reason = `Rate limit exceeded: ${state.tokensHour}/${cfg.tokensPerHour} tokens per hour`;
|
|
167
|
+
state.recentBlocks++;
|
|
168
|
+
eventBus.emit('rate-limit:exceeded', {
|
|
169
|
+
type: 'tokensPerHour', current: state.tokensHour, limit: cfg.tokensPerHour, action: cfg.action,
|
|
170
|
+
});
|
|
171
|
+
if (cfg.action === 'block') {
|
|
172
|
+
log.warn(reason);
|
|
173
|
+
return { blocked: { reason } };
|
|
174
|
+
}
|
|
175
|
+
log.warn('Rate limit exceeded (warn mode)', { type: 'tokensPerHour' });
|
|
176
|
+
}
|
|
177
|
+
}
|
|
178
|
+
// 3. Check cost per hour
|
|
179
|
+
if (cfg.maxCostPerHour > 0) {
|
|
180
|
+
checkWarning(cfg, 'maxCostPerHour', state.costHour, cfg.maxCostPerHour);
|
|
181
|
+
if (state.costHour >= cfg.maxCostPerHour) {
|
|
182
|
+
const reason = `Budget exceeded: $${state.costHour.toFixed(4)}/$${cfg.maxCostPerHour} per hour`;
|
|
183
|
+
state.recentBlocks++;
|
|
184
|
+
eventBus.emit('rate-limit:exceeded', {
|
|
185
|
+
type: 'maxCostPerHour', current: state.costHour, limit: cfg.maxCostPerHour, action: cfg.action,
|
|
186
|
+
});
|
|
187
|
+
if (cfg.action === 'block') {
|
|
188
|
+
log.warn(reason);
|
|
189
|
+
return { blocked: { reason } };
|
|
190
|
+
}
|
|
191
|
+
log.warn('Budget exceeded (warn mode)', { type: 'maxCostPerHour' });
|
|
192
|
+
}
|
|
193
|
+
}
|
|
194
|
+
// 4. Check cost per day
|
|
195
|
+
if (cfg.maxCostPerDay > 0) {
|
|
196
|
+
checkWarning(cfg, 'maxCostPerDay', state.costDay, cfg.maxCostPerDay);
|
|
197
|
+
if (state.costDay >= cfg.maxCostPerDay) {
|
|
198
|
+
const reason = `Budget exceeded: $${state.costDay.toFixed(4)}/$${cfg.maxCostPerDay} per day`;
|
|
199
|
+
state.recentBlocks++;
|
|
200
|
+
eventBus.emit('rate-limit:exceeded', {
|
|
201
|
+
type: 'maxCostPerDay', current: state.costDay, limit: cfg.maxCostPerDay, action: cfg.action,
|
|
202
|
+
});
|
|
203
|
+
if (cfg.action === 'block') {
|
|
204
|
+
log.warn(reason);
|
|
205
|
+
return { blocked: { reason } };
|
|
206
|
+
}
|
|
207
|
+
log.warn('Budget exceeded (warn mode)', { type: 'maxCostPerDay' });
|
|
208
|
+
}
|
|
209
|
+
}
|
|
210
|
+
// 5. Check cost per month
|
|
211
|
+
if (cfg.maxCostPerMonth > 0) {
|
|
212
|
+
checkWarning(cfg, 'maxCostPerMonth', state.costMonth, cfg.maxCostPerMonth);
|
|
213
|
+
if (state.costMonth >= cfg.maxCostPerMonth) {
|
|
214
|
+
const reason = `Budget exceeded: $${state.costMonth.toFixed(4)}/$${cfg.maxCostPerMonth} per month`;
|
|
215
|
+
state.recentBlocks++;
|
|
216
|
+
eventBus.emit('rate-limit:exceeded', {
|
|
217
|
+
type: 'maxCostPerMonth', current: state.costMonth, limit: cfg.maxCostPerMonth, action: cfg.action,
|
|
218
|
+
});
|
|
219
|
+
if (cfg.action === 'block') {
|
|
220
|
+
log.warn(reason);
|
|
221
|
+
return { blocked: { reason } };
|
|
222
|
+
}
|
|
223
|
+
log.warn('Budget exceeded (warn mode)', { type: 'maxCostPerMonth' });
|
|
224
|
+
}
|
|
225
|
+
}
|
|
226
|
+
// Record timestamp for RPM window
|
|
227
|
+
state.requestTimestamps.push(Date.now());
|
|
228
|
+
},
|
|
229
|
+
async onResponseComplete(context) {
|
|
230
|
+
rollPeriods();
|
|
231
|
+
const metrics = (0, collector_js_1.extractMetrics)(context);
|
|
232
|
+
const totalTokens = metrics.inputTokens + metrics.outputTokens;
|
|
233
|
+
state.tokensHour += totalTokens;
|
|
234
|
+
state.costHour += metrics.costUsd;
|
|
235
|
+
state.costDay += metrics.costUsd;
|
|
236
|
+
state.costMonth += metrics.costUsd;
|
|
237
|
+
log.debug('Rate-limiter counters updated', {
|
|
238
|
+
cost: metrics.costUsd.toFixed(6),
|
|
239
|
+
tokens: totalTokens,
|
|
240
|
+
costHour: state.costHour.toFixed(4),
|
|
241
|
+
costDay: state.costDay.toFixed(4),
|
|
242
|
+
costMonth: state.costMonth.toFixed(4),
|
|
243
|
+
});
|
|
244
|
+
},
|
|
245
|
+
};
|
|
246
|
+
return plugin;
|
|
247
|
+
}
|
|
248
|
+
//# sourceMappingURL=rate-limiter.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"rate-limiter.js","sourceRoot":"","sources":["../../../src/plugins/builtin/rate-limiter.ts"],"names":[],"mappings":";;AA6DA,0DAkQC;AA7TD,6DAA4D;AAC5D,qDAAqD;AAGrD,MAAM,GAAG,GAAG,IAAA,wBAAY,EAAC,cAAc,CAAC,CAAC;AA8BzC,SAAS,YAAY,CAAC,GAAW;IAC/B,MAAM,CAAC,GAAG,IAAI,IAAI,CAAC,GAAG,CAAC,CAAC;IACxB,CAAC,CAAC,UAAU,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;IACtB,OAAO,CAAC,CAAC,OAAO,EAAE,CAAC;AACrB,CAAC;AAED,SAAS,WAAW,CAAC,GAAW;IAC9B,MAAM,CAAC,GAAG,IAAI,IAAI,CAAC,GAAG,CAAC,CAAC;IACxB,CAAC,CAAC,QAAQ,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;IACvB,OAAO,CAAC,CAAC,OAAO,EAAE,CAAC;AACrB,CAAC;AAED,SAAS,aAAa,CAAC,GAAW;IAChC,MAAM,CAAC,GAAG,IAAI,IAAI,CAAC,GAAG,CAAC,CAAC;IACxB,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;IACb,CAAC,CAAC,QAAQ,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;IACvB,OAAO,CAAC,CAAC,OAAO,EAAE,CAAC;AACrB,CAAC;AAQD,SAAgB,uBAAuB,CACrC,EAAqB,EACrB,aAAsC,EACtC,QAAwB;IAExB,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IACvB,MAAM,KAAK,GAAe;QACxB,iBAAiB,EAAE,EAAE;QACrB,QAAQ,EAAE,CAAC;QACX,UAAU,EAAE,CAAC;QACb,SAAS,EAAE,YAAY,CAAC,GAAG,CAAC;QAC5B,OAAO,EAAE,CAAC;QACV,QAAQ,EAAE,WAAW,CAAC,GAAG,CAAC;QAC1B,SAAS,EAAE,CAAC;QACZ,UAAU,EAAE,aAAa,CAAC,GAAG,CAAC;QAC9B,YAAY,EAAE,CAAC;KAChB,CAAC;IAEF,sCAAsC;IACtC,IAAI,CAAC;QACH,MAAM,OAAO,GAAG,IAAI,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,WAAW,EAAE,CAAC;QACxD,MAAM,MAAM,GAAG,IAAI,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,WAAW,EAAE,CAAC;QACtD,MAAM,QAAQ,GAAG,IAAI,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC,WAAW,EAAE,CAAC;QAE1D,MAAM,OAAO,GAAG,EAAE,CAAC,OAAO,CACxB,uIAAuI,CACxI,CAAC,GAAG,CAAC,OAAO,CAAqC,CAAC;QAEnD,MAAM,MAAM,GAAG,EAAE,CAAC,OAAO,CACvB,8EAA8E,CAC/E,CAAC,GAAG,CAAC,MAAM,CAAqB,CAAC;QAElC,MAAM,QAAQ,GAAG,EAAE,CAAC,OAAO,CACzB,8EAA8E,CAC/E,CAAC,GAAG,CAAC,QAAQ,CAAqB,CAAC;QAEpC,KAAK,CAAC,QAAQ,GAAG,OAAO,CAAC,IAAI,CAAC;QAC9B,KAAK,CAAC,UAAU,GAAG,OAAO,CAAC,MAAM,CAAC;QAClC,KAAK,CAAC,OAAO,GAAG,MAAM,CAAC,IAAI,CAAC;QAC5B,KAAK,CAAC,SAAS,GAAG,QAAQ,CAAC,IAAI,CAAC;QAEhC,GAAG,CAAC,IAAI,CAAC,yCAAyC,EAAE;YAClD,QAAQ,EAAE,KAAK,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC;YACnC,UAAU,EAAE,KAAK,CAAC,UAAU;YAC5B,OAAO,EAAE,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC;YACjC,SAAS,EAAE,KAAK,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,CAAC;SACtC,CAAC,CAAC;IACL,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,GAAG,CAAC,IAAI,CAAC,yCAAyC,EAAE,EAAE,KAAK,EAAG,GAAa,CAAC,OAAO,EAAE,CAAC,CAAC;IACzF,CAAC;IAED,SAAS,WAAW;QAClB,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACvB,MAAM,YAAY,GAAG,YAAY,CAAC,GAAG,CAAC,CAAC;QACvC,IAAI,YAAY,GAAG,KAAK,CAAC,SAAS,EAAE,CAAC;YACnC,KAAK,CAAC,QAAQ,GAAG,CAAC,CAAC;YACnB,KAAK,CAAC,UAAU,GAAG,CAAC,CAAC;YACrB,KAAK,CAAC,SAAS,GAAG,YAAY,CAAC;QACjC,CAAC;QACD,MAAM,WAAW,GAAG,WAAW,CAAC,GAAG,CAAC,CAAC;QACrC,IAAI,WAAW,GAAG,KAAK,CAAC,QAAQ,EAAE,CAAC;YACjC,KAAK,CAAC,OAAO,GAAG,CAAC,CAAC;YAClB,KAAK,CAAC,QAAQ,GAAG,WAAW,CAAC;QAC/B,CAAC;QACD,MAAM,aAAa,GAAG,aAAa,CAAC,GAAG,CAAC,CAAC;QACzC,IAAI,aAAa,GAAG,KAAK,CAAC,UAAU,EAAE,CAAC;YACrC,KAAK,CAAC,SAAS,GAAG,CAAC,CAAC;YACpB,KAAK,CAAC,UAAU,GAAG,aAAa,CAAC;QACnC,CAAC;IACH,CAAC;IAED,SAAS,cAAc;QACrB,MAAM,MAAM,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,MAAM,CAAC;QACnC,KAAK,CAAC,iBAAiB,GAAG,KAAK,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,GAAG,MAAM,CAAC,CAAC;QAC1E,OAAO,KAAK,CAAC,iBAAiB,CAAC,MAAM,CAAC;IACxC,CAAC;IAED,SAAS,YAAY,CAAC,GAAsB,EAAE,IAAY,EAAE,OAAe,EAAE,KAAa;QACxF,IAAI,KAAK,IAAI,CAAC;YAAE,OAAO;QACvB,MAAM,GAAG,GAAG,OAAO,GAAG,KAAK,CAAC;QAC5B,IAAI,GAAG,IAAI,GAAG,CAAC,gBAAgB,IAAI,GAAG,GAAG,GAAG,EAAE,CAAC;YAC7C,QAAQ,CAAC,IAAI,CAAC,oBAAoB,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,UAAU,EAAE,GAAG,EAAE,CAAC,CAAC;YAC/E,GAAG,CAAC,IAAI,CAAC,oBAAoB,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,UAAU,EAAE,IAAI,CAAC,KAAK,CAAC,GAAG,GAAG,GAAG,CAAC,GAAG,GAAG,EAAE,CAAC,CAAC;QACpG,CAAC;IACH,CAAC;IAED,0CAA0C;IAC1C,SAAS,QAAQ;QACf,WAAW,EAAE,CAAC;QACd,MAAM,GAAG,GAAG,aAAa,EAAE,CAAC;QAC5B,MAAM,UAAU,GAAG,cAAc,EAAE,CAAC;QACpC,MAAM,MAAM,GAA2E,EAAE,CAAC;QAE1F,IAAI,GAAG,CAAC,iBAAiB,GAAG,CAAC,EAAE,CAAC;YAC9B,MAAM,CAAC,iBAAiB,GAAG;gBACzB,KAAK,EAAE,GAAG,CAAC,iBAAiB;gBAC5B,OAAO,EAAE,UAAU;gBACnB,UAAU,EAAE,UAAU,GAAG,GAAG,CAAC,iBAAiB;aAC/C,CAAC;QACJ,CAAC;QACD,IAAI,GAAG,CAAC,aAAa,GAAG,CAAC,EAAE,CAAC;YAC1B,MAAM,CAAC,aAAa,GAAG;gBACrB,KAAK,EAAE,GAAG,CAAC,aAAa;gBACxB,OAAO,EAAE,KAAK,CAAC,UAAU;gBACzB,UAAU,EAAE,KAAK,CAAC,UAAU,GAAG,GAAG,CAAC,aAAa;aACjD,CAAC;QACJ,CAAC;QACD,IAAI,GAAG,CAAC,cAAc,GAAG,CAAC,EAAE,CAAC;YAC3B,MAAM,CAAC,cAAc,GAAG;gBACtB,KAAK,EAAE,GAAG,CAAC,cAAc;gBACzB,OAAO,EAAE,KAAK,CAAC,QAAQ;gBACvB,UAAU,EAAE,KAAK,CAAC,QAAQ,GAAG,GAAG,CAAC,cAAc;aAChD,CAAC;QACJ,CAAC;QACD,IAAI,GAAG,CAAC,aAAa,GAAG,CAAC,EAAE,CAAC;YAC1B,MAAM,CAAC,aAAa,GAAG;gBACrB,KAAK,EAAE,GAAG,CAAC,aAAa;gBACxB,OAAO,EAAE,KAAK,CAAC,OAAO;gBACtB,UAAU,EAAE,KAAK,CAAC,OAAO,GAAG,GAAG,CAAC,aAAa;aAC9C,CAAC;QACJ,CAAC;QACD,IAAI,GAAG,CAAC,eAAe,GAAG,CAAC,EAAE,CAAC;YAC5B,MAAM,CAAC,eAAe,GAAG;gBACvB,KAAK,EAAE,GAAG,CAAC,eAAe;gBAC1B,OAAO,EAAE,KAAK,CAAC,SAAS;gBACxB,UAAU,EAAE,KAAK,CAAC,SAAS,GAAG,GAAG,CAAC,eAAe;aAClD,CAAC;QACJ,CAAC;QAED,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,GAAG,CAAC,MAAM,EAAE,YAAY,EAAE,KAAK,CAAC,YAAY,EAAE,CAAC;IAC1E,CAAC;IAED,mDAAmD;IACnD,MAAM,MAAM,GAAkD;QAC5D,IAAI,EAAE,cAAc;QACpB,QAAQ,EAAE,CAAC,EAAE,oEAAoE;QACjF,OAAO,EAAE,OAAO;QAChB,UAAU,EAAE,CAAC;QACb,MAAM,EAAE,SAAS;QAEjB,QAAQ;QAER,KAAK,CAAC,SAAS,CAAC,QAAwB;YACtC,WAAW,EAAE,CAAC;YACd,MAAM,GAAG,GAAG,aAAa,EAAE,CAAC;YAE5B,eAAe;YACf,MAAM,UAAU,GAAG,cAAc,EAAE,CAAC;YACpC,IAAI,GAAG,CAAC,iBAAiB,GAAG,CAAC,EAAE,CAAC;gBAC9B,YAAY,CAAC,GAAG,EAAE,mBAAmB,EAAE,UAAU,EAAE,GAAG,CAAC,iBAAiB,CAAC,CAAC;gBAC1E,IAAI,UAAU,IAAI,GAAG,CAAC,iBAAiB,EAAE,CAAC;oBACxC,MAAM,MAAM,GAAG,wBAAwB,UAAU,IAAI,GAAG,CAAC,iBAAiB,sBAAsB,CAAC;oBACjG,KAAK,CAAC,YAAY,EAAE,CAAC;oBACrB,QAAQ,CAAC,IAAI,CAAC,qBAAqB,EAAE;wBACnC,IAAI,EAAE,mBAAmB,EAAE,OAAO,EAAE,UAAU,EAAE,KAAK,EAAE,GAAG,CAAC,iBAAiB,EAAE,MAAM,EAAE,GAAG,CAAC,MAAM;qBACjG,CAAC,CAAC;oBACH,IAAI,GAAG,CAAC,MAAM,KAAK,OAAO,EAAE,CAAC;wBAC3B,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;wBACjB,OAAO,EAAE,OAAO,EAAE,EAAE,MAAM,EAAE,EAAE,CAAC;oBACjC,CAAC;oBACD,GAAG,CAAC,IAAI,CAAC,iCAAiC,EAAE,EAAE,IAAI,EAAE,mBAAmB,EAAE,CAAC,CAAC;gBAC7E,CAAC;YACH,CAAC;YAED,2BAA2B;YAC3B,IAAI,GAAG,CAAC,aAAa,GAAG,CAAC,EAAE,CAAC;gBAC1B,YAAY,CAAC,GAAG,EAAE,eAAe,EAAE,KAAK,CAAC,UAAU,EAAE,GAAG,CAAC,aAAa,CAAC,CAAC;gBACxE,IAAI,KAAK,CAAC,UAAU,IAAI,GAAG,CAAC,aAAa,EAAE,CAAC;oBAC1C,MAAM,MAAM,GAAG,wBAAwB,KAAK,CAAC,UAAU,IAAI,GAAG,CAAC,aAAa,kBAAkB,CAAC;oBAC/F,KAAK,CAAC,YAAY,EAAE,CAAC;oBACrB,QAAQ,CAAC,IAAI,CAAC,qBAAqB,EAAE;wBACnC,IAAI,EAAE,eAAe,EAAE,OAAO,EAAE,KAAK,CAAC,UAAU,EAAE,KAAK,EAAE,GAAG,CAAC,aAAa,EAAE,MAAM,EAAE,GAAG,CAAC,MAAM;qBAC/F,CAAC,CAAC;oBACH,IAAI,GAAG,CAAC,MAAM,KAAK,OAAO,EAAE,CAAC;wBAC3B,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;wBACjB,OAAO,EAAE,OAAO,EAAE,EAAE,MAAM,EAAE,EAAE,CAAC;oBACjC,CAAC;oBACD,GAAG,CAAC,IAAI,CAAC,iCAAiC,EAAE,EAAE,IAAI,EAAE,eAAe,EAAE,CAAC,CAAC;gBACzE,CAAC;YACH,CAAC;YAED,yBAAyB;YACzB,IAAI,GAAG,CAAC,cAAc,GAAG,CAAC,EAAE,CAAC;gBAC3B,YAAY,CAAC,GAAG,EAAE,gBAAgB,EAAE,KAAK,CAAC,QAAQ,EAAE,GAAG,CAAC,cAAc,CAAC,CAAC;gBACxE,IAAI,KAAK,CAAC,QAAQ,IAAI,GAAG,CAAC,cAAc,EAAE,CAAC;oBACzC,MAAM,MAAM,GAAG,qBAAqB,KAAK,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,KAAK,GAAG,CAAC,cAAc,WAAW,CAAC;oBAChG,KAAK,CAAC,YAAY,EAAE,CAAC;oBACrB,QAAQ,CAAC,IAAI,CAAC,qBAAqB,EAAE;wBACnC,IAAI,EAAE,gBAAgB,EAAE,OAAO,EAAE,KAAK,CAAC,QAAQ,EAAE,KAAK,EAAE,GAAG,CAAC,cAAc,EAAE,MAAM,EAAE,GAAG,CAAC,MAAM;qBAC/F,CAAC,CAAC;oBACH,IAAI,GAAG,CAAC,MAAM,KAAK,OAAO,EAAE,CAAC;wBAC3B,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;wBACjB,OAAO,EAAE,OAAO,EAAE,EAAE,MAAM,EAAE,EAAE,CAAC;oBACjC,CAAC;oBACD,GAAG,CAAC,IAAI,CAAC,6BAA6B,EAAE,EAAE,IAAI,EAAE,gBAAgB,EAAE,CAAC,CAAC;gBACtE,CAAC;YACH,CAAC;YAED,wBAAwB;YACxB,IAAI,GAAG,CAAC,aAAa,GAAG,CAAC,EAAE,CAAC;gBAC1B,YAAY,CAAC,GAAG,EAAE,eAAe,EAAE,KAAK,CAAC,OAAO,EAAE,GAAG,CAAC,aAAa,CAAC,CAAC;gBACrE,IAAI,KAAK,CAAC,OAAO,IAAI,GAAG,CAAC,aAAa,EAAE,CAAC;oBACvC,MAAM,MAAM,GAAG,qBAAqB,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,KAAK,GAAG,CAAC,aAAa,UAAU,CAAC;oBAC7F,KAAK,CAAC,YAAY,EAAE,CAAC;oBACrB,QAAQ,CAAC,IAAI,CAAC,qBAAqB,EAAE;wBACnC,IAAI,EAAE,eAAe,EAAE,OAAO,EAAE,KAAK,CAAC,OAAO,EAAE,KAAK,EAAE,GAAG,CAAC,aAAa,EAAE,MAAM,EAAE,GAAG,CAAC,MAAM;qBAC5F,CAAC,CAAC;oBACH,IAAI,GAAG,CAAC,MAAM,KAAK,OAAO,EAAE,CAAC;wBAC3B,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;wBACjB,OAAO,EAAE,OAAO,EAAE,EAAE,MAAM,EAAE,EAAE,CAAC;oBACjC,CAAC;oBACD,GAAG,CAAC,IAAI,CAAC,6BAA6B,EAAE,EAAE,IAAI,EAAE,eAAe,EAAE,CAAC,CAAC;gBACrE,CAAC;YACH,CAAC;YAED,0BAA0B;YAC1B,IAAI,GAAG,CAAC,eAAe,GAAG,CAAC,EAAE,CAAC;gBAC5B,YAAY,CAAC,GAAG,EAAE,iBAAiB,EAAE,KAAK,CAAC,SAAS,EAAE,GAAG,CAAC,eAAe,CAAC,CAAC;gBAC3E,IAAI,KAAK,CAAC,SAAS,IAAI,GAAG,CAAC,eAAe,EAAE,CAAC;oBAC3C,MAAM,MAAM,GAAG,qBAAqB,KAAK,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,CAAC,KAAK,GAAG,CAAC,eAAe,YAAY,CAAC;oBACnG,KAAK,CAAC,YAAY,EAAE,CAAC;oBACrB,QAAQ,CAAC,IAAI,CAAC,qBAAqB,EAAE;wBACnC,IAAI,EAAE,iBAAiB,EAAE,OAAO,EAAE,KAAK,CAAC,SAAS,EAAE,KAAK,EAAE,GAAG,CAAC,eAAe,EAAE,MAAM,EAAE,GAAG,CAAC,MAAM;qBAClG,CAAC,CAAC;oBACH,IAAI,GAAG,CAAC,MAAM,KAAK,OAAO,EAAE,CAAC;wBAC3B,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;wBACjB,OAAO,EAAE,OAAO,EAAE,EAAE,MAAM,EAAE,EAAE,CAAC;oBACjC,CAAC;oBACD,GAAG,CAAC,IAAI,CAAC,6BAA6B,EAAE,EAAE,IAAI,EAAE,iBAAiB,EAAE,CAAC,CAAC;gBACvE,CAAC;YACH,CAAC;YAED,kCAAkC;YAClC,KAAK,CAAC,iBAAiB,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC;QAC3C,CAAC;QAED,KAAK,CAAC,kBAAkB,CAAC,OAAgC;YACvD,WAAW,EAAE,CAAC;YAEd,MAAM,OAAO,GAAG,IAAA,6BAAc,EAAC,OAAO,CAAC,CAAC;YACxC,MAAM,WAAW,GAAG,OAAO,CAAC,WAAW,GAAG,OAAO,CAAC,YAAY,CAAC;YAE/D,KAAK,CAAC,UAAU,IAAI,WAAW,CAAC;YAChC,KAAK,CAAC,QAAQ,IAAI,OAAO,CAAC,OAAO,CAAC;YAClC,KAAK,CAAC,OAAO,IAAI,OAAO,CAAC,OAAO,CAAC;YACjC,KAAK,CAAC,SAAS,IAAI,OAAO,CAAC,OAAO,CAAC;YAEnC,GAAG,CAAC,KAAK,CAAC,+BAA+B,EAAE;gBACzC,IAAI,EAAE,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC;gBAChC,MAAM,EAAE,WAAW;gBACnB,QAAQ,EAAE,KAAK,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC;gBACnC,OAAO,EAAE,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC;gBACjC,SAAS,EAAE,KAAK,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,CAAC;aACtC,CAAC,CAAC;QACL,CAAC;KACF,CAAC;IAEF,OAAO,MAAM,CAAC;AAChB,CAAC"}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"threat-scorer.d.ts","sourceRoot":"","sources":["../../../src/plugins/builtin/threat-scorer.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,MAAM,EAAuC,MAAM,aAAa,CAAC;AAC/E,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,iBAAiB,CAAC;AACtD,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,wBAAwB,CAAC;AAS5D,OAAO,KAAK,QAAQ,MAAM,gBAAgB,CAAC;
|
|
1
|
+
{"version":3,"file":"threat-scorer.d.ts","sourceRoot":"","sources":["../../../src/plugins/builtin/threat-scorer.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,MAAM,EAAuC,MAAM,aAAa,CAAC;AAC/E,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,iBAAiB,CAAC;AACtD,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,wBAAwB,CAAC;AAS5D,OAAO,KAAK,QAAQ,MAAM,gBAAgB,CAAC;AA+F3C,wBAAgB,wBAAwB,CACtC,MAAM,EAAE,aAAa,EACrB,EAAE,EAAE,QAAQ,CAAC,QAAQ,EACrB,QAAQ,EAAE,cAAc,GACvB,MAAM,CAuOR"}
|
|
@@ -15,6 +15,7 @@ function getDefaults() {
|
|
|
15
15
|
enabled: true,
|
|
16
16
|
scoring: {
|
|
17
17
|
piWeight: 30,
|
|
18
|
+
indirectPiWeight: 40,
|
|
18
19
|
dlpWeight: 10,
|
|
19
20
|
toolGuardWeights: { critical: 25, high: 15, medium: 5, low: 2 },
|
|
20
21
|
toolChainWeight: 40,
|
|
@@ -34,6 +35,7 @@ function mergeConfig(config) {
|
|
|
34
35
|
enabled: ti.enabled ?? defaults.enabled,
|
|
35
36
|
scoring: {
|
|
36
37
|
piWeight: ti.scoring?.piWeight ?? defaults.scoring.piWeight,
|
|
38
|
+
indirectPiWeight: ti.scoring?.indirectPiWeight ?? defaults.scoring.indirectPiWeight,
|
|
37
39
|
dlpWeight: ti.scoring?.dlpWeight ?? defaults.scoring.dlpWeight,
|
|
38
40
|
toolGuardWeights: {
|
|
39
41
|
critical: ti.scoring?.toolGuardWeights?.critical ?? defaults.scoring.toolGuardWeights.critical,
|
|
@@ -160,7 +162,17 @@ function createThreatScorerPlugin(config, db, eventBus) {
|
|
|
160
162
|
const event = data;
|
|
161
163
|
if (!event?.sessionId)
|
|
162
164
|
return;
|
|
163
|
-
|
|
165
|
+
// Only score direct (user-input) detections; indirect detections scored via pi:indirect-injection
|
|
166
|
+
const directCount = event.directCount ?? event.detections ?? 1;
|
|
167
|
+
if (directCount > 0) {
|
|
168
|
+
addPoints(event.sessionId, tiConfig.scoring.piWeight, 'pi', `pi:detected severity=${event.severity ?? 'unknown'}`);
|
|
169
|
+
}
|
|
170
|
+
});
|
|
171
|
+
eventBus.on('pi:indirect-injection', (data) => {
|
|
172
|
+
const event = data;
|
|
173
|
+
if (!event?.sessionId)
|
|
174
|
+
return;
|
|
175
|
+
addPoints(event.sessionId, tiConfig.scoring.indirectPiWeight, 'pi-indirect', `pi:indirect-injection detections=${event.detections ?? 1}`);
|
|
164
176
|
});
|
|
165
177
|
eventBus.on('dlp:finding', (data) => {
|
|
166
178
|
const event = data;
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"threat-scorer.js","sourceRoot":"","sources":["../../../src/plugins/builtin/threat-scorer.ts"],"names":[],"mappings":";;AAuGA,4DAgOC;AApUD,kFAAqF;AACrF,8FAAgG;AAChG,kGAAoG;AACpG,8EAAiF;AACjF,0EAAmE;AACnE,wEAAiE;AACjE,oEAA0F;AAC1F,qDAAqD;AAGrD,MAAM,GAAG,GAAG,IAAA,wBAAY,EAAC,eAAe,CAAC,CAAC;AAiC1C,SAAS,WAAW;IAClB,OAAO;QACL,OAAO,EAAE,IAAI;QACb,OAAO,EAAE;YACP,QAAQ,EAAE,EAAE;YACZ,SAAS,EAAE,EAAE;YACb,gBAAgB,EAAE,EAAE,QAAQ,EAAE,EAAE,EAAE,IAAI,EAAE,EAAE,EAAE,MAAM,EAAE,CAAC,EAAE,GAAG,EAAE,CAAC,EAAE;YAC/D,eAAe,EAAE,EAAE;YACnB,cAAc,EAAE,GAAG;SACpB;QACD,UAAU,EAAE,EAAE,QAAQ,EAAE,EAAE,EAAE,IAAI,EAAE,EAAE,EAAE,QAAQ,EAAE,EAAE,EAAE;QACpD,SAAS,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,aAAa,EAAE,EAAE,EAAE;QAC/C,aAAa,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,UAAU,EAAE,EAAE,EAAE;KACjD,CAAC;AACJ,CAAC;AAED,SAAS,WAAW,CAAC,MAAqB;IACxC,MAAM,QAAQ,GAAG,WAAW,EAAE,CAAC;IAC/B,MAAM,EAAE,GAAG,MAAM,CAAC,OAAO,EAAE,kBAAkB,CAAC;IAC9C,IAAI,CAAC,EAAE;QAAE,OAAO,QAAQ,CAAC;IACzB,OAAO;QACL,OAAO,EAAE,EAAE,CAAC,OAAO,IAAI,QAAQ,CAAC,OAAO;QACvC,OAAO,EAAE;YACP,QAAQ,EAAE,EAAE,CAAC,OAAO,EAAE,QAAQ,IAAI,QAAQ,CAAC,OAAO,CAAC,QAAQ;YAC3D,SAAS,EAAE,EAAE,CAAC,OAAO,EAAE,SAAS,IAAI,QAAQ,CAAC,OAAO,CAAC,SAAS;YAC9D,gBAAgB,EAAE;gBAChB,QAAQ,EAAE,EAAE,CAAC,OAAO,EAAE,gBAAgB,EAAE,QAAQ,IAAI,QAAQ,CAAC,OAAO,CAAC,gBAAgB,CAAC,QAAQ;gBAC9F,IAAI,EAAE,EAAE,CAAC,OAAO,EAAE,gBAAgB,EAAE,IAAI,IAAI,QAAQ,CAAC,OAAO,CAAC,gBAAgB,CAAC,IAAI;gBAClF,MAAM,EAAE,EAAE,CAAC,OAAO,EAAE,gBAAgB,EAAE,MAAM,IAAI,QAAQ,CAAC,OAAO,CAAC,gBAAgB,CAAC,MAAM;gBACxF,GAAG,EAAE,EAAE,CAAC,OAAO,EAAE,gBAAgB,EAAE,GAAG,IAAI,QAAQ,CAAC,OAAO,CAAC,gBAAgB,CAAC,GAAG;aAChF;YACD,eAAe,EAAE,EAAE,CAAC,OAAO,EAAE,eAAe,IAAI,QAAQ,CAAC,OAAO,CAAC,eAAe;YAChF,cAAc,EAAE,EAAE,CAAC,OAAO,EAAE,cAAc,IAAI,QAAQ,CAAC,OAAO,CAAC,cAAc;SAC9E;QACD,UAAU,EAAE;YACV,QAAQ,EAAE,EAAE,CAAC,UAAU,EAAE,QAAQ,IAAI,QAAQ,CAAC,UAAU,CAAC,QAAQ;YACjE,IAAI,EAAE,EAAE,CAAC,UAAU,EAAE,IAAI,IAAI,QAAQ,CAAC,UAAU,CAAC,IAAI;YACrD,QAAQ,EAAE,EAAE,CAAC,UAAU,EAAE,QAAQ,IAAI,QAAQ,CAAC,UAAU,CAAC,QAAQ;SAClE;QACD,SAAS,EAAE;YACT,OAAO,EAAE,EAAE,CAAC,SAAS,EAAE,OAAO,IAAI,QAAQ,CAAC,SAAS,CAAC,OAAO;YAC5D,aAAa,EAAE,EAAE,CAAC,SAAS,EAAE,aAAa,IAAI,QAAQ,CAAC,SAAS,CAAC,aAAa;SAC/E;QACD,aAAa,EAAE;YACb,OAAO,EAAE,EAAE,CAAC,aAAa,EAAE,OAAO,IAAI,QAAQ,CAAC,aAAa,CAAC,OAAO;YACpE,UAAU,EAAE,EAAE,CAAC,aAAa,EAAE,UAAU,IAAI,QAAQ,CAAC,aAAa,CAAC,UAAU;SAC9E;KACF,CAAC;AACJ,CAAC;AAED,SAAS,YAAY,CAAC,KAAa,EAAE,UAA2B;IAC9D,IAAI,KAAK,IAAI,UAAU,CAAC,QAAQ;QAAE,OAAO,UAAU,CAAC;IACpD,IAAI,KAAK,IAAI,UAAU,CAAC,IAAI;QAAE,OAAO,MAAM,CAAC;IAC5C,IAAI,KAAK,IAAI,UAAU,CAAC,QAAQ;QAAE,OAAO,UAAU,CAAC;IACpD,OAAO,QAAQ,CAAC;AAClB,CAAC;AAED,SAAgB,wBAAwB,CACtC,MAAqB,EACrB,EAAqB,EACrB,QAAwB;IAExB,MAAM,QAAQ,GAAG,WAAW,CAAC,MAAM,CAAC,CAAC;IACrC,MAAM,UAAU,GAAG,IAAI,yCAAsB,CAAC,EAAE,CAAC,CAAC;IAClD,MAAM,UAAU,GAAG,IAAI,oDAA2B,CAAC,EAAE,CAAC,CAAC;IACvD,MAAM,mBAAmB,GAAG,IAAI,wDAA6B,CAAC,EAAE,CAAC,CAAC;IAClE,MAAM,cAAc,GAAG,IAAI,qCAAoB,CAAC,EAAE,CAAC,CAAC;IAEpD,MAAM,aAAa,GAAG,IAAI,iCAAa,CAAC,QAAQ,CAAC,SAAS,CAAC,aAAa,CAAC,CAAC;IAC1E,MAAM,YAAY,GAAG,IAAI,+BAAY,CAAC,QAAQ,CAAC,aAAa,CAAC,UAAU,CAAC,CAAC;IAEzE,qCAAqC;IACrC,MAAM,QAAQ,GAAG,IAAI,GAAG,EAAuB,CAAC;IAEhD,MAAM,UAAU,GAAoB,CAAC,GAAG,oCAAmB,CAAC,CAAC;IAE7D,SAAS,gBAAgB,CAAC,SAAiB;QACzC,IAAI,KAAK,GAAG,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QACpC,IAAI,CAAC,KAAK,EAAE,CAAC;YACX,sBAAsB;YACtB,MAAM,MAAM,GAAG,UAAU,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;YACzC,IAAI,MAAM,EAAE,CAAC;gBACX,KAAK,GAAG;oBACN,KAAK,EAAE,MAAM,CAAC,KAAK;oBACnB,KAAK,EAAE,MAAM,CAAC,KAAoB;oBAClC,UAAU,EAAE,MAAM,CAAC,WAAW;oBAC9B,WAAW,EAAE,MAAM,CAAC,aAAa,CAAC,CAAC,CAAC,IAAI,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,CAAC;iBACjF,CAAC;YACJ,CAAC;iBAAM,CAAC;gBACN,KAAK,GAAG,EAAE,KAAK,EAAE,CAAC,EAAE,KAAK,EAAE,QAAQ,EAAE,UAAU,EAAE,CAAC,EAAE,WAAW,EAAE,CAAC,EAAE,CAAC;YACvE,CAAC;YACD,QAAQ,CAAC,GAAG,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC;QACjC,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAED,SAAS,UAAU,CAAC,KAAkB;QACpC,IAAI,KAAK,CAAC,KAAK,IAAI,CAAC,IAAI,KAAK,CAAC,WAAW,KAAK,CAAC;YAAE,OAAO;QACxD,MAAM,cAAc,GAAG,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,KAAK,CAAC,WAAW,CAAC,GAAG,KAAK,CAAC;QAChE,IAAI,cAAc,IAAI,CAAC;YAAE,OAAO;QAChC,wDAAwD;QACxD,KAAK,CAAC,KAAK,IAAI,IAAI,CAAC,GAAG,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,cAAc,GAAG,cAAc,CAAC,CAAC;QAC3E,IAAI,KAAK,CAAC,KAAK,GAAG,GAAG;YAAE,KAAK,CAAC,KAAK,GAAG,CAAC,CAAC;IACzC,CAAC;IAED,SAAS,SAAS,CAAC,SAAiB,EAAE,MAAc,EAAE,SAAiB,EAAE,WAAmB;QAC1F,MAAM,KAAK,GAAG,gBAAgB,CAAC,SAAS,CAAC,CAAC;QAC1C,UAAU,CAAC,KAAK,CAAC,CAAC;QAElB,KAAK,CAAC,KAAK,IAAI,MAAM,CAAC;QACtB,KAAK,CAAC,UAAU,EAAE,CAAC;QACnB,KAAK,CAAC,WAAW,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC/B,KAAK,CAAC,KAAK,GAAG,YAAY,CAAC,KAAK,CAAC,KAAK,EAAE,QAAQ,CAAC,UAAU,CAAC,CAAC;QAE7D,QAAQ,CAAC,GAAG,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC;QAE/B,gBAAgB;QAChB,IAAI,CAAC;YACH,UAAU,CAAC,MAAM,CAAC;gBAChB,UAAU,EAAE,SAAS;gBACrB,KAAK,EAAE,KAAK,CAAC,KAAK;gBAClB,KAAK,EAAE,KAAK,CAAC,KAAK;gBAClB,WAAW,EAAE,KAAK,CAAC,UAAU;gBAC7B,aAAa,EAAE,IAAI,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC,WAAW,EAAE;aACzD,CAAC,CAAC;YAEH,UAAU,CAAC,MAAM,CAAC;gBAChB,EAAE,EAAE,MAAM,CAAC,UAAU,EAAE;gBACvB,UAAU,EAAE,SAAS;gBACrB,UAAU,EAAE,SAAS;gBACrB,YAAY,EAAE,WAAW;gBACzB,MAAM;gBACN,WAAW,EAAE,KAAK,CAAC,KAAK;gBACxB,WAAW,EAAE,KAAK,CAAC,KAAK;aACzB,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,GAAG,CAAC,IAAI,CAAC,gCAAgC,EAAE,EAAE,KAAK,EAAG,GAAa,CAAC,OAAO,EAAE,CAAC,CAAC;QAChF,CAAC;QAED,IAAI,KAAK,CAAC,KAAK,KAAK,QAAQ,EAAE,CAAC;YAC7B,GAAG,CAAC,IAAI,CAAC,8BAA8B,EAAE;gBACvC,SAAS;gBACT,KAAK,EAAE,KAAK,CAAC,KAAK;gBAClB,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,KAAK,GAAG,GAAG,CAAC,GAAG,GAAG;gBAC1C,SAAS;aACV,CAAC,CAAC;YAEH,QAAQ,CAAC,IAAI,CAAC,qBAAqB,EAAE;gBACnC,SAAS;gBACT,KAAK,EAAE,KAAK,CAAC,KAAK;gBAClB,KAAK,EAAE,KAAK,CAAC,KAAK;gBAClB,SAAS;aACV,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAED,wBAAwB;IAExB,QAAQ,CAAC,EAAE,CAAC,aAAa,EAAE,CAAC,IAAa,EAAE,EAAE;QAC3C,MAAM,KAAK,GAAG,IAA6D,CAAC;QAC5E,IAAI,CAAC,KAAK,EAAE,SAAS;YAAE,OAAO;QAC9B,SAAS,CAAC,KAAK,CAAC,SAAS,EAAE,QAAQ,CAAC,OAAO,CAAC,QAAQ,EAAE,IAAI,EAAE,wBAAwB,KAAK,CAAC,QAAQ,IAAI,SAAS,EAAE,CAAC,CAAC;IACrH,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,EAAE,CAAC,aAAa,EAAE,CAAC,IAAa,EAAE,EAAE;QAC3C,MAAM,KAAK,GAAG,IAA6H,CAAC;QAC5I,IAAI,CAAC,KAAK,EAAE,SAAS;YAAE,OAAO;QAC9B,SAAS,CAAC,KAAK,CAAC,SAAS,EAAE,QAAQ,CAAC,OAAO,CAAC,SAAS,EAAE,KAAK,EAAE,uBAAuB,KAAK,CAAC,WAAW,IAAI,SAAS,EAAE,CAAC,CAAC;QAEvH,gEAAgE;QAChE,IAAI,QAAQ,CAAC,aAAa,CAAC,OAAO,IAAI,KAAK,CAAC,WAAW,IAAI,KAAK,CAAC,SAAS,EAAE,CAAC;YAC3E,MAAM,WAAW,GAAG,YAAY,CAAC,SAAS,CACxC,KAAK,CAAC,SAAS,EAAE,KAAK,CAAC,SAAS,EAAE,KAAK,CAAC,WAAW,EACnD,GAAG,KAAK,CAAC,WAAW,IAAI,KAAK,CAAC,SAAS,EAAE,CAC1C,CAAC;YACF,IAAI,CAAC;gBACH,cAAc,CAAC,MAAM,CAAC;oBACpB,EAAE,EAAE,MAAM,CAAC,UAAU,EAAE;oBACvB,UAAU,EAAE,KAAK,CAAC,SAAS;oBAC3B,UAAU,EAAE,KAAK,CAAC,SAAS;oBAC3B,YAAY,EAAE,KAAK,CAAC,WAAW;oBAC/B,SAAS,EAAE,KAAK,CAAC,SAAS,IAAI,SAAS;oBACvC,WAAW;iBACZ,CAAC,CAAC;YACL,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACb,GAAG,CAAC,IAAI,CAAC,8BAA8B,EAAE,EAAE,KAAK,EAAG,GAAa,CAAC,OAAO,EAAE,CAAC,CAAC;YAC9E,CAAC;QACH,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,EAAE,CAAC,iBAAiB,EAAE,CAAC,IAAa,EAAE,EAAE;QAC/C,MAAM,KAAK,GAAG,IAID,CAAC;QACd,IAAI,CAAC,KAAK,EAAE,SAAS;YAAE,OAAO;QAE9B,MAAM,QAAQ,GAAG,KAAK,CAAC,QAAQ,IAAI,QAAQ,CAAC;QAC5C,MAAM,OAAO,GAAG,QAAQ,CAAC,OAAO,CAAC,gBAAgB,CAAC;QAClD,MAAM,MAAM,GAAI,OAAkC,CAAC,QAAQ,CAAC,IAAI,OAAO,CAAC,MAAM,CAAC;QAC/E,SAAS,CAAC,KAAK,CAAC,SAAS,EAAE,MAAM,EAAE,WAAW,EAAE,wBAAwB,KAAK,CAAC,QAAQ,IAAI,SAAS,aAAa,QAAQ,EAAE,CAAC,CAAC;QAE5H,6DAA6D;QAC7D,IAAI,QAAQ,CAAC,SAAS,CAAC,OAAO,IAAI,KAAK,CAAC,QAAQ,EAAE,CAAC;YACjD,aAAa,CAAC,cAAc,CAAC,KAAK,CAAC,SAAS,EAAE,KAAK,CAAC,QAAQ,CAAC,CAAC;YAC9D,MAAM,KAAK,GAAG,aAAa,CAAC,WAAW,CAAC,KAAK,CAAC,SAAS,EAAE,UAAU,CAAC,CAAC;YACrE,IAAI,KAAK,EAAE,CAAC;gBACV,SAAS,CAAC,KAAK,CAAC,SAAS,EAAE,QAAQ,CAAC,OAAO,CAAC,eAAe,EAAE,WAAW,EACtE,SAAS,KAAK,CAAC,IAAI,CAAC,EAAE,aAAa,KAAK,CAAC,eAAe,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;gBAExE,IAAI,CAAC;oBACH,mBAAmB,CAAC,MAAM,CAAC;wBACzB,EAAE,EAAE,MAAM,CAAC,UAAU,EAAE;wBACvB,UAAU,EAAE,KAAK,CAAC,SAAS;wBAC3B,OAAO,EAAE,KAAK,CAAC,IAAI,CAAC,EAAE;wBACtB,gBAAgB,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,eAAe,CAAC;wBACvD,MAAM,EAAE,KAAK,CAAC,IAAI,CAAC,MAAM;qBAC1B,CAAC,CAAC;gBACL,CAAC;gBAAC,OAAO,GAAG,EAAE,CAAC;oBACb,GAAG,CAAC,IAAI,CAAC,mCAAmC,EAAE,EAAE,KAAK,EAAG,GAAa,CAAC,OAAO,EAAE,CAAC,CAAC;gBACnF,CAAC;gBAED,QAAQ,CAAC,IAAI,CAAC,oBAAoB,EAAE;oBAClC,SAAS,EAAE,KAAK,CAAC,SAAS;oBAC1B,MAAM,EAAE,KAAK,CAAC,IAAI,CAAC,EAAE;oBACrB,QAAQ,EAAE,KAAK,CAAC,IAAI,CAAC,IAAI;oBACzB,QAAQ,EAAE,KAAK,CAAC,eAAe;oBAC/B,MAAM,EAAE,KAAK,CAAC,IAAI,CAAC,MAAM;iBAC1B,CAAC,CAAC;YACL,CAAC;QACH,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,OAAO;QACL,IAAI,EAAE,eAAe;QACrB,QAAQ,EAAE,CAAC,EAAE,yDAAyD;QACtE,OAAO,EAAE,OAAO;QAChB,UAAU,EAAE,CAAC;QACb,MAAM,EAAE,SAAS;QAEjB,KAAK,CAAC,SAAS,CAAC,OAAuB;YACrC,IAAI,CAAC,OAAO,CAAC,SAAS;gBAAE,OAAO;YAE/B,MAAM,KAAK,GAAG,gBAAgB,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;YAClD,UAAU,CAAC,KAAK,CAAC,CAAC;YAElB,2BAA2B;YAC3B,KAAK,CAAC,KAAK,GAAG,YAAY,CAAC,KAAK,CAAC,KAAK,EAAE,QAAQ,CAAC,UAAU,CAAC,CAAC;YAC7D,QAAQ,CAAC,GAAG,CAAC,OAAO,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC;YAEvC,6EAA6E;YAC7E,OAAO,CAAC,YAAY,GAAG,KAAK,CAAC,KAAK,CAAC;YACnC,OAAO,CAAC,YAAY,GAAG,KAAK,CAAC,KAAK,CAAC;YAEnC,IAAI,KAAK,CAAC,KAAK,KAAK,QAAQ,EAAE,CAAC;gBAC7B,GAAG,CAAC,KAAK,CAAC,wBAAwB,EAAE;oBAClC,SAAS,EAAE,OAAO,CAAC,SAAS;oBAC5B,KAAK,EAAE,KAAK,CAAC,KAAK;oBAClB,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,KAAK,GAAG,GAAG,CAAC,GAAG,GAAG;iBAC3C,CAAC,CAAC;YACL,CAAC;YAED,oEAAoE;YACpE,IAAI,QAAQ,CAAC,aAAa,CAAC,OAAO,EAAE,CAAC;gBACnC,MAAM,MAAM,GAAG,YAAY,CAAC,eAAe,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;gBAC/D,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC,IAAI,OAAO,CAAC,IAAI,EAAE,CAAC;oBACtC,MAAM,KAAK,GAAG,YAAY,CAAC,cAAc,CAAC,OAAO,CAAC,SAAS,EAAE,OAAO,CAAC,IAAI,CAAC,CAAC;oBAC3E,IAAI,KAAK,EAAE,CAAC;wBACV,SAAS,CAAC,OAAO,CAAC,SAAS,EAAE,QAAQ,CAAC,OAAO,CAAC,SAAS,EAAE,OAAO,EAC9D,0BAA0B,KAAK,CAAC,WAAW,gBAAgB,KAAK,CAAC,WAAW,EAAE,CAAC,CAAC;wBAClF,GAAG,CAAC,IAAI,CAAC,qCAAqC,EAAE;4BAC9C,SAAS,EAAE,OAAO,CAAC,SAAS;4BAC5B,SAAS,EAAE,OAAO,CAAC,EAAE;4BACrB,WAAW,EAAE,KAAK,CAAC,WAAW;yBAC/B,CAAC,CAAC;oBACL,CAAC;gBACH,CAAC;YACH,CAAC;QACH,CAAC;KACF,CAAC;AACJ,CAAC"}
|
|
1
|
+
{"version":3,"file":"threat-scorer.js","sourceRoot":"","sources":["../../../src/plugins/builtin/threat-scorer.ts"],"names":[],"mappings":";;AA0GA,4DA2OC;AAlVD,kFAAqF;AACrF,8FAAgG;AAChG,kGAAoG;AACpG,8EAAiF;AACjF,0EAAmE;AACnE,wEAAiE;AACjE,oEAA0F;AAC1F,qDAAqD;AAGrD,MAAM,GAAG,GAAG,IAAA,wBAAY,EAAC,eAAe,CAAC,CAAC;AAkC1C,SAAS,WAAW;IAClB,OAAO;QACL,OAAO,EAAE,IAAI;QACb,OAAO,EAAE;YACP,QAAQ,EAAE,EAAE;YACZ,gBAAgB,EAAE,EAAE;YACpB,SAAS,EAAE,EAAE;YACb,gBAAgB,EAAE,EAAE,QAAQ,EAAE,EAAE,EAAE,IAAI,EAAE,EAAE,EAAE,MAAM,EAAE,CAAC,EAAE,GAAG,EAAE,CAAC,EAAE;YAC/D,eAAe,EAAE,EAAE;YACnB,cAAc,EAAE,GAAG;SACpB;QACD,UAAU,EAAE,EAAE,QAAQ,EAAE,EAAE,EAAE,IAAI,EAAE,EAAE,EAAE,QAAQ,EAAE,EAAE,EAAE;QACpD,SAAS,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,aAAa,EAAE,EAAE,EAAE;QAC/C,aAAa,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,UAAU,EAAE,EAAE,EAAE;KACjD,CAAC;AACJ,CAAC;AAED,SAAS,WAAW,CAAC,MAAqB;IACxC,MAAM,QAAQ,GAAG,WAAW,EAAE,CAAC;IAC/B,MAAM,EAAE,GAAG,MAAM,CAAC,OAAO,EAAE,kBAAkB,CAAC;IAC9C,IAAI,CAAC,EAAE;QAAE,OAAO,QAAQ,CAAC;IACzB,OAAO;QACL,OAAO,EAAE,EAAE,CAAC,OAAO,IAAI,QAAQ,CAAC,OAAO;QACvC,OAAO,EAAE;YACP,QAAQ,EAAE,EAAE,CAAC,OAAO,EAAE,QAAQ,IAAI,QAAQ,CAAC,OAAO,CAAC,QAAQ;YAC3D,gBAAgB,EAAE,EAAE,CAAC,OAAO,EAAE,gBAAgB,IAAI,QAAQ,CAAC,OAAO,CAAC,gBAAgB;YACnF,SAAS,EAAE,EAAE,CAAC,OAAO,EAAE,SAAS,IAAI,QAAQ,CAAC,OAAO,CAAC,SAAS;YAC9D,gBAAgB,EAAE;gBAChB,QAAQ,EAAE,EAAE,CAAC,OAAO,EAAE,gBAAgB,EAAE,QAAQ,IAAI,QAAQ,CAAC,OAAO,CAAC,gBAAgB,CAAC,QAAQ;gBAC9F,IAAI,EAAE,EAAE,CAAC,OAAO,EAAE,gBAAgB,EAAE,IAAI,IAAI,QAAQ,CAAC,OAAO,CAAC,gBAAgB,CAAC,IAAI;gBAClF,MAAM,EAAE,EAAE,CAAC,OAAO,EAAE,gBAAgB,EAAE,MAAM,IAAI,QAAQ,CAAC,OAAO,CAAC,gBAAgB,CAAC,MAAM;gBACxF,GAAG,EAAE,EAAE,CAAC,OAAO,EAAE,gBAAgB,EAAE,GAAG,IAAI,QAAQ,CAAC,OAAO,CAAC,gBAAgB,CAAC,GAAG;aAChF;YACD,eAAe,EAAE,EAAE,CAAC,OAAO,EAAE,eAAe,IAAI,QAAQ,CAAC,OAAO,CAAC,eAAe;YAChF,cAAc,EAAE,EAAE,CAAC,OAAO,EAAE,cAAc,IAAI,QAAQ,CAAC,OAAO,CAAC,cAAc;SAC9E;QACD,UAAU,EAAE;YACV,QAAQ,EAAE,EAAE,CAAC,UAAU,EAAE,QAAQ,IAAI,QAAQ,CAAC,UAAU,CAAC,QAAQ;YACjE,IAAI,EAAE,EAAE,CAAC,UAAU,EAAE,IAAI,IAAI,QAAQ,CAAC,UAAU,CAAC,IAAI;YACrD,QAAQ,EAAE,EAAE,CAAC,UAAU,EAAE,QAAQ,IAAI,QAAQ,CAAC,UAAU,CAAC,QAAQ;SAClE;QACD,SAAS,EAAE;YACT,OAAO,EAAE,EAAE,CAAC,SAAS,EAAE,OAAO,IAAI,QAAQ,CAAC,SAAS,CAAC,OAAO;YAC5D,aAAa,EAAE,EAAE,CAAC,SAAS,EAAE,aAAa,IAAI,QAAQ,CAAC,SAAS,CAAC,aAAa;SAC/E;QACD,aAAa,EAAE;YACb,OAAO,EAAE,EAAE,CAAC,aAAa,EAAE,OAAO,IAAI,QAAQ,CAAC,aAAa,CAAC,OAAO;YACpE,UAAU,EAAE,EAAE,CAAC,aAAa,EAAE,UAAU,IAAI,QAAQ,CAAC,aAAa,CAAC,UAAU;SAC9E;KACF,CAAC;AACJ,CAAC;AAED,SAAS,YAAY,CAAC,KAAa,EAAE,UAA2B;IAC9D,IAAI,KAAK,IAAI,UAAU,CAAC,QAAQ;QAAE,OAAO,UAAU,CAAC;IACpD,IAAI,KAAK,IAAI,UAAU,CAAC,IAAI;QAAE,OAAO,MAAM,CAAC;IAC5C,IAAI,KAAK,IAAI,UAAU,CAAC,QAAQ;QAAE,OAAO,UAAU,CAAC;IACpD,OAAO,QAAQ,CAAC;AAClB,CAAC;AAED,SAAgB,wBAAwB,CACtC,MAAqB,EACrB,EAAqB,EACrB,QAAwB;IAExB,MAAM,QAAQ,GAAG,WAAW,CAAC,MAAM,CAAC,CAAC;IACrC,MAAM,UAAU,GAAG,IAAI,yCAAsB,CAAC,EAAE,CAAC,CAAC;IAClD,MAAM,UAAU,GAAG,IAAI,oDAA2B,CAAC,EAAE,CAAC,CAAC;IACvD,MAAM,mBAAmB,GAAG,IAAI,wDAA6B,CAAC,EAAE,CAAC,CAAC;IAClE,MAAM,cAAc,GAAG,IAAI,qCAAoB,CAAC,EAAE,CAAC,CAAC;IAEpD,MAAM,aAAa,GAAG,IAAI,iCAAa,CAAC,QAAQ,CAAC,SAAS,CAAC,aAAa,CAAC,CAAC;IAC1E,MAAM,YAAY,GAAG,IAAI,+BAAY,CAAC,QAAQ,CAAC,aAAa,CAAC,UAAU,CAAC,CAAC;IAEzE,qCAAqC;IACrC,MAAM,QAAQ,GAAG,IAAI,GAAG,EAAuB,CAAC;IAEhD,MAAM,UAAU,GAAoB,CAAC,GAAG,oCAAmB,CAAC,CAAC;IAE7D,SAAS,gBAAgB,CAAC,SAAiB;QACzC,IAAI,KAAK,GAAG,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QACpC,IAAI,CAAC,KAAK,EAAE,CAAC;YACX,sBAAsB;YACtB,MAAM,MAAM,GAAG,UAAU,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;YACzC,IAAI,MAAM,EAAE,CAAC;gBACX,KAAK,GAAG;oBACN,KAAK,EAAE,MAAM,CAAC,KAAK;oBACnB,KAAK,EAAE,MAAM,CAAC,KAAoB;oBAClC,UAAU,EAAE,MAAM,CAAC,WAAW;oBAC9B,WAAW,EAAE,MAAM,CAAC,aAAa,CAAC,CAAC,CAAC,IAAI,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,CAAC;iBACjF,CAAC;YACJ,CAAC;iBAAM,CAAC;gBACN,KAAK,GAAG,EAAE,KAAK,EAAE,CAAC,EAAE,KAAK,EAAE,QAAQ,EAAE,UAAU,EAAE,CAAC,EAAE,WAAW,EAAE,CAAC,EAAE,CAAC;YACvE,CAAC;YACD,QAAQ,CAAC,GAAG,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC;QACjC,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAED,SAAS,UAAU,CAAC,KAAkB;QACpC,IAAI,KAAK,CAAC,KAAK,IAAI,CAAC,IAAI,KAAK,CAAC,WAAW,KAAK,CAAC;YAAE,OAAO;QACxD,MAAM,cAAc,GAAG,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,KAAK,CAAC,WAAW,CAAC,GAAG,KAAK,CAAC;QAChE,IAAI,cAAc,IAAI,CAAC;YAAE,OAAO;QAChC,wDAAwD;QACxD,KAAK,CAAC,KAAK,IAAI,IAAI,CAAC,GAAG,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,cAAc,GAAG,cAAc,CAAC,CAAC;QAC3E,IAAI,KAAK,CAAC,KAAK,GAAG,GAAG;YAAE,KAAK,CAAC,KAAK,GAAG,CAAC,CAAC;IACzC,CAAC;IAED,SAAS,SAAS,CAAC,SAAiB,EAAE,MAAc,EAAE,SAAiB,EAAE,WAAmB;QAC1F,MAAM,KAAK,GAAG,gBAAgB,CAAC,SAAS,CAAC,CAAC;QAC1C,UAAU,CAAC,KAAK,CAAC,CAAC;QAElB,KAAK,CAAC,KAAK,IAAI,MAAM,CAAC;QACtB,KAAK,CAAC,UAAU,EAAE,CAAC;QACnB,KAAK,CAAC,WAAW,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC/B,KAAK,CAAC,KAAK,GAAG,YAAY,CAAC,KAAK,CAAC,KAAK,EAAE,QAAQ,CAAC,UAAU,CAAC,CAAC;QAE7D,QAAQ,CAAC,GAAG,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC;QAE/B,gBAAgB;QAChB,IAAI,CAAC;YACH,UAAU,CAAC,MAAM,CAAC;gBAChB,UAAU,EAAE,SAAS;gBACrB,KAAK,EAAE,KAAK,CAAC,KAAK;gBAClB,KAAK,EAAE,KAAK,CAAC,KAAK;gBAClB,WAAW,EAAE,KAAK,CAAC,UAAU;gBAC7B,aAAa,EAAE,IAAI,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC,WAAW,EAAE;aACzD,CAAC,CAAC;YAEH,UAAU,CAAC,MAAM,CAAC;gBAChB,EAAE,EAAE,MAAM,CAAC,UAAU,EAAE;gBACvB,UAAU,EAAE,SAAS;gBACrB,UAAU,EAAE,SAAS;gBACrB,YAAY,EAAE,WAAW;gBACzB,MAAM;gBACN,WAAW,EAAE,KAAK,CAAC,KAAK;gBACxB,WAAW,EAAE,KAAK,CAAC,KAAK;aACzB,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,GAAG,CAAC,IAAI,CAAC,gCAAgC,EAAE,EAAE,KAAK,EAAG,GAAa,CAAC,OAAO,EAAE,CAAC,CAAC;QAChF,CAAC;QAED,IAAI,KAAK,CAAC,KAAK,KAAK,QAAQ,EAAE,CAAC;YAC7B,GAAG,CAAC,IAAI,CAAC,8BAA8B,EAAE;gBACvC,SAAS;gBACT,KAAK,EAAE,KAAK,CAAC,KAAK;gBAClB,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,KAAK,GAAG,GAAG,CAAC,GAAG,GAAG;gBAC1C,SAAS;aACV,CAAC,CAAC;YAEH,QAAQ,CAAC,IAAI,CAAC,qBAAqB,EAAE;gBACnC,SAAS;gBACT,KAAK,EAAE,KAAK,CAAC,KAAK;gBAClB,KAAK,EAAE,KAAK,CAAC,KAAK;gBAClB,SAAS;aACV,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAED,wBAAwB;IAExB,QAAQ,CAAC,EAAE,CAAC,aAAa,EAAE,CAAC,IAAa,EAAE,EAAE;QAC3C,MAAM,KAAK,GAAG,IAAwG,CAAC;QACvH,IAAI,CAAC,KAAK,EAAE,SAAS;YAAE,OAAO;QAC9B,kGAAkG;QAClG,MAAM,WAAW,GAAG,KAAK,CAAC,WAAW,IAAI,KAAK,CAAC,UAAU,IAAI,CAAC,CAAC;QAC/D,IAAI,WAAW,GAAG,CAAC,EAAE,CAAC;YACpB,SAAS,CAAC,KAAK,CAAC,SAAS,EAAE,QAAQ,CAAC,OAAO,CAAC,QAAQ,EAAE,IAAI,EAAE,wBAAwB,KAAK,CAAC,QAAQ,IAAI,SAAS,EAAE,CAAC,CAAC;QACrH,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,EAAE,CAAC,uBAAuB,EAAE,CAAC,IAAa,EAAE,EAAE;QACrD,MAAM,KAAK,GAAG,IAAkF,CAAC;QACjG,IAAI,CAAC,KAAK,EAAE,SAAS;YAAE,OAAO;QAC9B,SAAS,CAAC,KAAK,CAAC,SAAS,EAAE,QAAQ,CAAC,OAAO,CAAC,gBAAgB,EAAE,aAAa,EACzE,oCAAoC,KAAK,CAAC,UAAU,IAAI,CAAC,EAAE,CAAC,CAAC;IACjE,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,EAAE,CAAC,aAAa,EAAE,CAAC,IAAa,EAAE,EAAE;QAC3C,MAAM,KAAK,GAAG,IAA6H,CAAC;QAC5I,IAAI,CAAC,KAAK,EAAE,SAAS;YAAE,OAAO;QAC9B,SAAS,CAAC,KAAK,CAAC,SAAS,EAAE,QAAQ,CAAC,OAAO,CAAC,SAAS,EAAE,KAAK,EAAE,uBAAuB,KAAK,CAAC,WAAW,IAAI,SAAS,EAAE,CAAC,CAAC;QAEvH,gEAAgE;QAChE,IAAI,QAAQ,CAAC,aAAa,CAAC,OAAO,IAAI,KAAK,CAAC,WAAW,IAAI,KAAK,CAAC,SAAS,EAAE,CAAC;YAC3E,MAAM,WAAW,GAAG,YAAY,CAAC,SAAS,CACxC,KAAK,CAAC,SAAS,EAAE,KAAK,CAAC,SAAS,EAAE,KAAK,CAAC,WAAW,EACnD,GAAG,KAAK,CAAC,WAAW,IAAI,KAAK,CAAC,SAAS,EAAE,CAC1C,CAAC;YACF,IAAI,CAAC;gBACH,cAAc,CAAC,MAAM,CAAC;oBACpB,EAAE,EAAE,MAAM,CAAC,UAAU,EAAE;oBACvB,UAAU,EAAE,KAAK,CAAC,SAAS;oBAC3B,UAAU,EAAE,KAAK,CAAC,SAAS;oBAC3B,YAAY,EAAE,KAAK,CAAC,WAAW;oBAC/B,SAAS,EAAE,KAAK,CAAC,SAAS,IAAI,SAAS;oBACvC,WAAW;iBACZ,CAAC,CAAC;YACL,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACb,GAAG,CAAC,IAAI,CAAC,8BAA8B,EAAE,EAAE,KAAK,EAAG,GAAa,CAAC,OAAO,EAAE,CAAC,CAAC;YAC9E,CAAC;QACH,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,EAAE,CAAC,iBAAiB,EAAE,CAAC,IAAa,EAAE,EAAE;QAC/C,MAAM,KAAK,GAAG,IAID,CAAC;QACd,IAAI,CAAC,KAAK,EAAE,SAAS;YAAE,OAAO;QAE9B,MAAM,QAAQ,GAAG,KAAK,CAAC,QAAQ,IAAI,QAAQ,CAAC;QAC5C,MAAM,OAAO,GAAG,QAAQ,CAAC,OAAO,CAAC,gBAAgB,CAAC;QAClD,MAAM,MAAM,GAAI,OAAkC,CAAC,QAAQ,CAAC,IAAI,OAAO,CAAC,MAAM,CAAC;QAC/E,SAAS,CAAC,KAAK,CAAC,SAAS,EAAE,MAAM,EAAE,WAAW,EAAE,wBAAwB,KAAK,CAAC,QAAQ,IAAI,SAAS,aAAa,QAAQ,EAAE,CAAC,CAAC;QAE5H,6DAA6D;QAC7D,IAAI,QAAQ,CAAC,SAAS,CAAC,OAAO,IAAI,KAAK,CAAC,QAAQ,EAAE,CAAC;YACjD,aAAa,CAAC,cAAc,CAAC,KAAK,CAAC,SAAS,EAAE,KAAK,CAAC,QAAQ,CAAC,CAAC;YAC9D,MAAM,KAAK,GAAG,aAAa,CAAC,WAAW,CAAC,KAAK,CAAC,SAAS,EAAE,UAAU,CAAC,CAAC;YACrE,IAAI,KAAK,EAAE,CAAC;gBACV,SAAS,CAAC,KAAK,CAAC,SAAS,EAAE,QAAQ,CAAC,OAAO,CAAC,eAAe,EAAE,WAAW,EACtE,SAAS,KAAK,CAAC,IAAI,CAAC,EAAE,aAAa,KAAK,CAAC,eAAe,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;gBAExE,IAAI,CAAC;oBACH,mBAAmB,CAAC,MAAM,CAAC;wBACzB,EAAE,EAAE,MAAM,CAAC,UAAU,EAAE;wBACvB,UAAU,EAAE,KAAK,CAAC,SAAS;wBAC3B,OAAO,EAAE,KAAK,CAAC,IAAI,CAAC,EAAE;wBACtB,gBAAgB,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,eAAe,CAAC;wBACvD,MAAM,EAAE,KAAK,CAAC,IAAI,CAAC,MAAM;qBAC1B,CAAC,CAAC;gBACL,CAAC;gBAAC,OAAO,GAAG,EAAE,CAAC;oBACb,GAAG,CAAC,IAAI,CAAC,mCAAmC,EAAE,EAAE,KAAK,EAAG,GAAa,CAAC,OAAO,EAAE,CAAC,CAAC;gBACnF,CAAC;gBAED,QAAQ,CAAC,IAAI,CAAC,oBAAoB,EAAE;oBAClC,SAAS,EAAE,KAAK,CAAC,SAAS;oBAC1B,MAAM,EAAE,KAAK,CAAC,IAAI,CAAC,EAAE;oBACrB,QAAQ,EAAE,KAAK,CAAC,IAAI,CAAC,IAAI;oBACzB,QAAQ,EAAE,KAAK,CAAC,eAAe;oBAC/B,MAAM,EAAE,KAAK,CAAC,IAAI,CAAC,MAAM;iBAC1B,CAAC,CAAC;YACL,CAAC;QACH,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,OAAO;QACL,IAAI,EAAE,eAAe;QACrB,QAAQ,EAAE,CAAC,EAAE,yDAAyD;QACtE,OAAO,EAAE,OAAO;QAChB,UAAU,EAAE,CAAC;QACb,MAAM,EAAE,SAAS;QAEjB,KAAK,CAAC,SAAS,CAAC,OAAuB;YACrC,IAAI,CAAC,OAAO,CAAC,SAAS;gBAAE,OAAO;YAE/B,MAAM,KAAK,GAAG,gBAAgB,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;YAClD,UAAU,CAAC,KAAK,CAAC,CAAC;YAElB,2BAA2B;YAC3B,KAAK,CAAC,KAAK,GAAG,YAAY,CAAC,KAAK,CAAC,KAAK,EAAE,QAAQ,CAAC,UAAU,CAAC,CAAC;YAC7D,QAAQ,CAAC,GAAG,CAAC,OAAO,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC;YAEvC,6EAA6E;YAC7E,OAAO,CAAC,YAAY,GAAG,KAAK,CAAC,KAAK,CAAC;YACnC,OAAO,CAAC,YAAY,GAAG,KAAK,CAAC,KAAK,CAAC;YAEnC,IAAI,KAAK,CAAC,KAAK,KAAK,QAAQ,EAAE,CAAC;gBAC7B,GAAG,CAAC,KAAK,CAAC,wBAAwB,EAAE;oBAClC,SAAS,EAAE,OAAO,CAAC,SAAS;oBAC5B,KAAK,EAAE,KAAK,CAAC,KAAK;oBAClB,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,KAAK,GAAG,GAAG,CAAC,GAAG,GAAG;iBAC3C,CAAC,CAAC;YACL,CAAC;YAED,oEAAoE;YACpE,IAAI,QAAQ,CAAC,aAAa,CAAC,OAAO,EAAE,CAAC;gBACnC,MAAM,MAAM,GAAG,YAAY,CAAC,eAAe,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;gBAC/D,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC,IAAI,OAAO,CAAC,IAAI,EAAE,CAAC;oBACtC,MAAM,KAAK,GAAG,YAAY,CAAC,cAAc,CAAC,OAAO,CAAC,SAAS,EAAE,OAAO,CAAC,IAAI,CAAC,CAAC;oBAC3E,IAAI,KAAK,EAAE,CAAC;wBACV,SAAS,CAAC,OAAO,CAAC,SAAS,EAAE,QAAQ,CAAC,OAAO,CAAC,SAAS,EAAE,OAAO,EAC9D,0BAA0B,KAAK,CAAC,WAAW,gBAAgB,KAAK,CAAC,WAAW,EAAE,CAAC,CAAC;wBAClF,GAAG,CAAC,IAAI,CAAC,qCAAqC,EAAE;4BAC9C,SAAS,EAAE,OAAO,CAAC,SAAS;4BAC5B,SAAS,EAAE,OAAO,CAAC,EAAE;4BACrB,WAAW,EAAE,KAAK,CAAC,WAAW;yBAC/B,CAAC,CAAC;oBACL,CAAC;gBACH,CAAC;YACH,CAAC;QACH,CAAC;KACF,CAAC;AACJ,CAAC"}
|
|
@@ -8,6 +8,13 @@ export interface ToolGuardConfig {
|
|
|
8
8
|
alertMinSeverity: string;
|
|
9
9
|
alertDesktop: boolean;
|
|
10
10
|
alertWebhookUrl: string;
|
|
11
|
+
piEscalation?: {
|
|
12
|
+
enabled: boolean;
|
|
13
|
+
scoreThreshold: number;
|
|
14
|
+
overrideSeverity: string;
|
|
15
|
+
scope: 'session' | 'request';
|
|
16
|
+
ttlMinutes: number;
|
|
17
|
+
};
|
|
11
18
|
/** Live getter — when provided, overrides static fields for hot-reload */
|
|
12
19
|
getLiveConfig?: () => {
|
|
13
20
|
action: string;
|
|
@@ -16,5 +23,16 @@ export interface ToolGuardConfig {
|
|
|
16
23
|
alertMinSeverity: string;
|
|
17
24
|
};
|
|
18
25
|
}
|
|
26
|
+
export interface PiEscalationEntry {
|
|
27
|
+
sessionId: string;
|
|
28
|
+
score: number;
|
|
29
|
+
label: string;
|
|
30
|
+
overrideSeverity: string;
|
|
31
|
+
escalatedAt: number;
|
|
32
|
+
}
|
|
33
|
+
export declare function getPiEscalations(): PiEscalationEntry[];
|
|
34
|
+
export declare function resetPiEscalation(sessionId: string): boolean;
|
|
35
|
+
export declare function resetAllPiEscalations(): number;
|
|
36
|
+
export declare function getPiEscalationCount(): number;
|
|
19
37
|
export declare function createToolGuardPlugin(db: Database.Database, config: ToolGuardConfig, eventBus?: import('../event-bus.js').PluginEventBus): Plugin;
|
|
20
38
|
//# sourceMappingURL=tool-guard.d.ts.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"tool-guard.d.ts","sourceRoot":"","sources":["../../../src/plugins/builtin/tool-guard.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,MAAM,EAMP,MAAM,aAAa,CAAC;
|
|
1
|
+
{"version":3,"file":"tool-guard.d.ts","sourceRoot":"","sources":["../../../src/plugins/builtin/tool-guard.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,MAAM,EAMP,MAAM,aAAa,CAAC;AASrB,OAAO,KAAK,QAAQ,MAAM,gBAAgB,CAAC;AAI3C,MAAM,WAAW,eAAe;IAC9B,OAAO,EAAE,OAAO,CAAC;IACjB,MAAM,EAAE,OAAO,GAAG,OAAO,CAAC;IAC1B,SAAS,EAAE,OAAO,CAAC;IACnB,gBAAgB,EAAE,MAAM,CAAC;IACzB,gBAAgB,EAAE,MAAM,CAAC;IACzB,YAAY,EAAE,OAAO,CAAC;IACtB,eAAe,EAAE,MAAM,CAAC;IACxB,YAAY,CAAC,EAAE;QACb,OAAO,EAAE,OAAO,CAAC;QACjB,cAAc,EAAE,MAAM,CAAC;QACvB,gBAAgB,EAAE,MAAM,CAAC;QACzB,KAAK,EAAE,SAAS,GAAG,SAAS,CAAC;QAC7B,UAAU,EAAE,MAAM,CAAC;KACpB,CAAC;IACF,0EAA0E;IAC1E,aAAa,CAAC,EAAE,MAAM;QAAE,MAAM,EAAE,MAAM,CAAC;QAAC,SAAS,EAAE,OAAO,CAAC;QAAC,gBAAgB,EAAE,MAAM,CAAC;QAAC,gBAAgB,EAAE,MAAM,CAAA;KAAE,CAAC;CAClH;AAID,MAAM,WAAW,iBAAiB;IAChC,SAAS,EAAE,MAAM,CAAC;IAClB,KAAK,EAAE,MAAM,CAAC;IACd,KAAK,EAAE,MAAM,CAAC;IACd,gBAAgB,EAAE,MAAM,CAAC;IACzB,WAAW,EAAE,MAAM,CAAC;CACrB;AAeD,wBAAgB,gBAAgB,IAAI,iBAAiB,EAAE,CAGtD;AAED,wBAAgB,iBAAiB,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAE5D;AAED,wBAAgB,qBAAqB,IAAI,MAAM,CAI9C;AAED,wBAAgB,oBAAoB,IAAI,MAAM,CAG7C;AAqFD,wBAAgB,qBAAqB,CAAC,EAAE,EAAE,QAAQ,CAAC,QAAQ,EAAE,MAAM,EAAE,eAAe,EAAE,QAAQ,CAAC,EAAE,OAAO,iBAAiB,EAAE,cAAc,GAAG,MAAM,CAmVjJ"}
|
|
@@ -1,5 +1,9 @@
|
|
|
1
1
|
"use strict";
|
|
2
2
|
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.getPiEscalations = getPiEscalations;
|
|
4
|
+
exports.resetPiEscalation = resetPiEscalation;
|
|
5
|
+
exports.resetAllPiEscalations = resetAllPiEscalations;
|
|
6
|
+
exports.getPiEscalationCount = getPiEscalationCount;
|
|
3
7
|
exports.createToolGuardPlugin = createToolGuardPlugin;
|
|
4
8
|
const tool_calls_js_1 = require("../../storage/repositories/tool-calls.js");
|
|
5
9
|
const tool_guard_rules_js_1 = require("../../storage/repositories/tool-guard-rules.js");
|
|
@@ -7,8 +11,37 @@ const audit_log_js_1 = require("../../storage/repositories/audit-log.js");
|
|
|
7
11
|
const extractor_js_1 = require("../../tool-guard/extractor.js");
|
|
8
12
|
const rules_js_1 = require("../../tool-guard/rules.js");
|
|
9
13
|
const alert_js_1 = require("../../tool-guard/alert.js");
|
|
14
|
+
const plugin_events_js_1 = require("../../storage/repositories/plugin-events.js");
|
|
10
15
|
const logger_js_1 = require("../../utils/logger.js");
|
|
11
16
|
const log = (0, logger_js_1.createLogger)('tool-guard');
|
|
17
|
+
const piEscalationMap = new Map();
|
|
18
|
+
let piEscalationTtlMs = 30 * 60000;
|
|
19
|
+
function cleanExpiredPiEscalations() {
|
|
20
|
+
if (piEscalationTtlMs <= 0)
|
|
21
|
+
return;
|
|
22
|
+
const now = Date.now();
|
|
23
|
+
for (const [key, entry] of piEscalationMap) {
|
|
24
|
+
if (now - entry.escalatedAt > piEscalationTtlMs) {
|
|
25
|
+
piEscalationMap.delete(key);
|
|
26
|
+
}
|
|
27
|
+
}
|
|
28
|
+
}
|
|
29
|
+
function getPiEscalations() {
|
|
30
|
+
cleanExpiredPiEscalations();
|
|
31
|
+
return Array.from(piEscalationMap.values());
|
|
32
|
+
}
|
|
33
|
+
function resetPiEscalation(sessionId) {
|
|
34
|
+
return piEscalationMap.delete(sessionId);
|
|
35
|
+
}
|
|
36
|
+
function resetAllPiEscalations() {
|
|
37
|
+
const count = piEscalationMap.size;
|
|
38
|
+
piEscalationMap.clear();
|
|
39
|
+
return count;
|
|
40
|
+
}
|
|
41
|
+
function getPiEscalationCount() {
|
|
42
|
+
cleanExpiredPiEscalations();
|
|
43
|
+
return piEscalationMap.size;
|
|
44
|
+
}
|
|
12
45
|
function analyzeToolCalls(body, isStreaming, rules) {
|
|
13
46
|
const toolCalls = (0, extractor_js_1.extractToolCalls)(body, isStreaming);
|
|
14
47
|
return toolCalls.map(tc => ({
|
|
@@ -80,23 +113,124 @@ function createToolGuardPlugin(db, config, eventBus) {
|
|
|
80
113
|
const repo = new tool_calls_js_1.ToolCallsRepository(db);
|
|
81
114
|
const rulesRepo = new tool_guard_rules_js_1.ToolGuardRulesRepository(db);
|
|
82
115
|
const auditRepo = new audit_log_js_1.AuditLogRepository(db);
|
|
116
|
+
const pluginEventsRepo = new plugin_events_js_1.PluginEventsRepository(db);
|
|
83
117
|
// Seed built-in rules on first init (INSERT OR IGNORE preserves user toggles)
|
|
84
118
|
rulesRepo.seedBuiltins(rules_js_1.BUILTIN_RULES);
|
|
119
|
+
// ── PI Escalation setup ──
|
|
120
|
+
const piCfg = config.piEscalation;
|
|
121
|
+
if (piCfg) {
|
|
122
|
+
piEscalationTtlMs = (piCfg.ttlMinutes ?? 30) * 60000;
|
|
123
|
+
}
|
|
124
|
+
if (piCfg?.enabled && eventBus) {
|
|
125
|
+
eventBus.on('pi:detected', (data) => {
|
|
126
|
+
const ev = data;
|
|
127
|
+
if (!ev.sessionId)
|
|
128
|
+
return;
|
|
129
|
+
if ((ev.score ?? 0) < (piCfg.scoreThreshold ?? 0.8))
|
|
130
|
+
return;
|
|
131
|
+
const entry = {
|
|
132
|
+
sessionId: ev.sessionId,
|
|
133
|
+
score: ev.score ?? 0,
|
|
134
|
+
label: ev.label ?? 'injection',
|
|
135
|
+
overrideSeverity: piCfg.overrideSeverity ?? 'medium',
|
|
136
|
+
escalatedAt: Date.now(),
|
|
137
|
+
};
|
|
138
|
+
piEscalationMap.set(ev.sessionId, entry);
|
|
139
|
+
log.warn('PI escalation triggered', { sessionId: ev.sessionId, score: ev.score, override: entry.overrideSeverity });
|
|
140
|
+
// Audit log to plugin_events table
|
|
141
|
+
try {
|
|
142
|
+
pluginEventsRepo.insertEvent('tool-guard', ev.requestId ?? null, {
|
|
143
|
+
type: 'pi-escalation',
|
|
144
|
+
severity: entry.overrideSeverity,
|
|
145
|
+
rule: 'pi-escalation',
|
|
146
|
+
detail: `PI score ${ev.score?.toFixed(2)} >= ${piCfg.scoreThreshold} → blockMinSeverity override to ${entry.overrideSeverity} (scope=${piCfg.scope})`,
|
|
147
|
+
});
|
|
148
|
+
}
|
|
149
|
+
catch (err) {
|
|
150
|
+
log.warn('Failed to write PI escalation audit', { error: err.message });
|
|
151
|
+
}
|
|
152
|
+
eventBus.emit('toolguard:pi-escalation', {
|
|
153
|
+
sessionId: ev.sessionId,
|
|
154
|
+
score: ev.score,
|
|
155
|
+
label: ev.label,
|
|
156
|
+
overrideSeverity: entry.overrideSeverity,
|
|
157
|
+
scope: piCfg.scope,
|
|
158
|
+
});
|
|
159
|
+
});
|
|
160
|
+
// Indirect injection always escalates (external data injection is inherently high risk)
|
|
161
|
+
eventBus.on('pi:indirect-injection', (data) => {
|
|
162
|
+
const ev = data;
|
|
163
|
+
if (!ev.sessionId)
|
|
164
|
+
return;
|
|
165
|
+
const entry = {
|
|
166
|
+
sessionId: ev.sessionId,
|
|
167
|
+
score: ev.maxScore ?? 0,
|
|
168
|
+
label: 'indirect-injection',
|
|
169
|
+
overrideSeverity: piCfg.overrideSeverity ?? 'medium',
|
|
170
|
+
escalatedAt: Date.now(),
|
|
171
|
+
};
|
|
172
|
+
piEscalationMap.set(ev.sessionId, entry);
|
|
173
|
+
log.warn('PI indirect-injection escalation triggered', { sessionId: ev.sessionId, score: ev.maxScore, detections: ev.detections });
|
|
174
|
+
try {
|
|
175
|
+
pluginEventsRepo.insertEvent('tool-guard', null, {
|
|
176
|
+
type: 'pi-indirect-escalation',
|
|
177
|
+
severity: entry.overrideSeverity,
|
|
178
|
+
rule: 'pi-indirect-escalation',
|
|
179
|
+
detail: `Indirect injection detected in tool_result (score ${(ev.maxScore ?? 0).toFixed(2)}, ${ev.detections ?? 1} detections) → blockMinSeverity override to ${entry.overrideSeverity}`,
|
|
180
|
+
});
|
|
181
|
+
}
|
|
182
|
+
catch (err) {
|
|
183
|
+
log.warn('Failed to write PI indirect escalation audit', { error: err.message });
|
|
184
|
+
}
|
|
185
|
+
eventBus.emit('toolguard:pi-escalation', {
|
|
186
|
+
sessionId: ev.sessionId,
|
|
187
|
+
score: ev.maxScore,
|
|
188
|
+
label: 'indirect-injection',
|
|
189
|
+
overrideSeverity: entry.overrideSeverity,
|
|
190
|
+
scope: piCfg.scope,
|
|
191
|
+
});
|
|
192
|
+
});
|
|
193
|
+
}
|
|
85
194
|
// Live config readers — support hot-reload from Dashboard
|
|
86
195
|
const getAction = () => config.getLiveConfig ? config.getLiveConfig().action : config.action;
|
|
87
196
|
const getRecordAll = () => config.getLiveConfig ? config.getLiveConfig().recordAll : config.recordAll;
|
|
88
197
|
const getBlockMinSeverity = () => config.getLiveConfig ? config.getLiveConfig().blockMinSeverity : config.blockMinSeverity;
|
|
89
198
|
const getAlertMinSeverity = () => config.getLiveConfig ? config.getLiveConfig().alertMinSeverity : config.alertMinSeverity;
|
|
90
|
-
|
|
199
|
+
// Severity rank for comparison (lower rank = more strict)
|
|
200
|
+
const SEVERITY_RANK = { critical: 4, high: 3, medium: 2, low: 1 };
|
|
201
|
+
/** Get effective blockMinSeverity for a request context.
|
|
202
|
+
* Combines threat-scorer path and PI escalation path, taking the stricter (lower rank) of the two. */
|
|
91
203
|
function getEffectiveBlockMinSeverity(context) {
|
|
204
|
+
// Path 1: threat-level based escalation
|
|
205
|
+
let threatSeverity;
|
|
92
206
|
const threatLevel = context._threatLevel;
|
|
93
207
|
if (threatLevel === 'critical')
|
|
94
|
-
|
|
95
|
-
if (threatLevel === 'high')
|
|
96
|
-
|
|
97
|
-
if (threatLevel === 'elevated')
|
|
98
|
-
|
|
99
|
-
|
|
208
|
+
threatSeverity = 'low';
|
|
209
|
+
else if (threatLevel === 'high')
|
|
210
|
+
threatSeverity = 'medium';
|
|
211
|
+
else if (threatLevel === 'elevated')
|
|
212
|
+
threatSeverity = 'high';
|
|
213
|
+
else
|
|
214
|
+
threatSeverity = getBlockMinSeverity();
|
|
215
|
+
// Path 2: PI escalation
|
|
216
|
+
if (piCfg?.enabled && context.sessionId) {
|
|
217
|
+
cleanExpiredPiEscalations();
|
|
218
|
+
const piEntry = piEscalationMap.get(context.sessionId);
|
|
219
|
+
if (piEntry) {
|
|
220
|
+
const piSeverity = piEntry.overrideSeverity;
|
|
221
|
+
const piRank = SEVERITY_RANK[piSeverity] ?? 99;
|
|
222
|
+
const threatRank = SEVERITY_RANK[threatSeverity] ?? 99;
|
|
223
|
+
if (piRank < threatRank) {
|
|
224
|
+
context._piEscalated = true;
|
|
225
|
+
context._piEscalationOverride = piSeverity;
|
|
226
|
+
return piSeverity;
|
|
227
|
+
}
|
|
228
|
+
// Even if threat path is stricter, mark PI escalation as active
|
|
229
|
+
context._piEscalated = true;
|
|
230
|
+
context._piEscalationOverride = piSeverity;
|
|
231
|
+
}
|
|
232
|
+
}
|
|
233
|
+
return threatSeverity;
|
|
100
234
|
}
|
|
101
235
|
function getAlertConfig() {
|
|
102
236
|
return {
|
|
@@ -179,7 +313,7 @@ function createToolGuardPlugin(db, config, eventBus) {
|
|
|
179
313
|
const rules = rulesRepo.getEnabled();
|
|
180
314
|
context._toolGuardRules = rules;
|
|
181
315
|
const effectiveBlockMin = getEffectiveBlockMinSeverity(context);
|
|
182
|
-
log.debug('onRequest', { action: getAction(), recordAll: getRecordAll(), isStreaming: context.isStreaming, effectiveBlockMin, threatLevel: context._threatLevel });
|
|
316
|
+
log.debug('onRequest', { action: getAction(), recordAll: getRecordAll(), isStreaming: context.isStreaming, effectiveBlockMin, threatLevel: context._threatLevel, piEscalated: context._piEscalated });
|
|
183
317
|
if (getAction() === 'block' && context.isStreaming) {
|
|
184
318
|
context._toolGuardStreamBlock = effectiveBlockMin;
|
|
185
319
|
}
|
|
@@ -276,6 +410,12 @@ function createToolGuardPlugin(db, config, eventBus) {
|
|
|
276
410
|
catch (err) {
|
|
277
411
|
log.warn('Tool guard processing failed', { error: err.message });
|
|
278
412
|
}
|
|
413
|
+
finally {
|
|
414
|
+
// Request-scope PI escalation: remove after this request completes
|
|
415
|
+
if (piCfg?.enabled && piCfg.scope === 'request' && context.request.sessionId && context.request._piEscalated) {
|
|
416
|
+
piEscalationMap.delete(context.request.sessionId);
|
|
417
|
+
}
|
|
418
|
+
}
|
|
279
419
|
},
|
|
280
420
|
};
|
|
281
421
|
}
|