@aion0/bastion 0.1.14 → 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.
Files changed (46) hide show
  1. package/README.md +1 -1
  2. package/README.zh.md +1 -1
  3. package/config/default.yaml +17 -0
  4. package/dist/cli/commands/plugins.d.ts.map +1 -1
  5. package/dist/cli/commands/plugins.js +13 -7
  6. package/dist/cli/commands/plugins.js.map +1 -1
  7. package/dist/cli/commands/start.d.ts.map +1 -1
  8. package/dist/cli/commands/start.js +4 -0
  9. package/dist/cli/commands/start.js.map +1 -1
  10. package/dist/config/schema.d.ts +21 -1
  11. package/dist/config/schema.d.ts.map +1 -1
  12. package/dist/core/bootstrap.d.ts.map +1 -1
  13. package/dist/core/bootstrap.js +30 -0
  14. package/dist/core/bootstrap.js.map +1 -1
  15. package/dist/dashboard/api-routes.d.ts.map +1 -1
  16. package/dist/dashboard/api-routes.js +338 -8
  17. package/dist/dashboard/api-routes.js.map +1 -1
  18. package/dist/dashboard/page.d.ts.map +1 -1
  19. package/dist/dashboard/page.js +648 -111
  20. package/dist/dashboard/page.js.map +1 -1
  21. package/dist/dlp/actions.d.ts +2 -0
  22. package/dist/dlp/actions.d.ts.map +1 -1
  23. package/dist/dlp/ai-validator.d.ts +15 -1
  24. package/dist/dlp/ai-validator.d.ts.map +1 -1
  25. package/dist/dlp/ai-validator.js +84 -3
  26. package/dist/dlp/ai-validator.js.map +1 -1
  27. package/dist/dlp/engine.d.ts +5 -1
  28. package/dist/dlp/engine.d.ts.map +1 -1
  29. package/dist/dlp/engine.js +25 -7
  30. package/dist/dlp/engine.js.map +1 -1
  31. package/dist/dlp/message-cache.js +2 -2
  32. package/dist/dlp/message-cache.js.map +1 -1
  33. package/dist/plugins/builtin/rate-limiter.d.ts +24 -0
  34. package/dist/plugins/builtin/rate-limiter.d.ts.map +1 -0
  35. package/dist/plugins/builtin/rate-limiter.js +248 -0
  36. package/dist/plugins/builtin/rate-limiter.js.map +1 -0
  37. package/dist/plugins/builtin/threat-scorer.d.ts.map +1 -1
  38. package/dist/plugins/builtin/threat-scorer.js +13 -1
  39. package/dist/plugins/builtin/threat-scorer.js.map +1 -1
  40. package/dist/plugins/builtin/tool-guard.d.ts +18 -0
  41. package/dist/plugins/builtin/tool-guard.d.ts.map +1 -1
  42. package/dist/plugins/builtin/tool-guard.js +148 -8
  43. package/dist/plugins/builtin/tool-guard.js.map +1 -1
  44. package/dist/plugins/types.d.ts +3 -0
  45. package/dist/plugins/types.d.ts.map +1 -1
  46. 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;AA4F3C,wBAAgB,wBAAwB,CACtC,MAAM,EAAE,aAAa,EACrB,EAAE,EAAE,QAAQ,CAAC,QAAQ,EACrB,QAAQ,EAAE,cAAc,GACvB,MAAM,CA4NR"}
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
- addPoints(event.sessionId, tiConfig.scoring.piWeight, 'pi', `pi:detected severity=${event.severity ?? 'unknown'}`);
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;AAQrB,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,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;AAqFD,wBAAgB,qBAAqB,CAAC,EAAE,EAAE,QAAQ,CAAC,QAAQ,EAAE,MAAM,EAAE,eAAe,EAAE,QAAQ,CAAC,EAAE,OAAO,iBAAiB,EAAE,cAAc,GAAG,MAAM,CAiOjJ"}
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
- /** Get effective blockMinSeverity for a request context (uses threat-scorer's _threatLevel) */
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
- return 'low'; // Block all severities
95
- if (threatLevel === 'high')
96
- return 'medium'; // Block medium+
97
- if (threatLevel === 'elevated')
98
- return 'high'; // Block high+
99
- return getBlockMinSeverity();
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
  }