agent-threat-rules 0.2.2 → 0.3.1
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/README.md +152 -642
- package/dist/capability-extractor.d.ts +35 -0
- package/dist/capability-extractor.d.ts.map +1 -0
- package/dist/capability-extractor.js +91 -0
- package/dist/capability-extractor.js.map +1 -0
- package/dist/cli.js +56 -2
- package/dist/cli.js.map +1 -1
- package/dist/converters/elastic.d.ts +36 -0
- package/dist/converters/elastic.d.ts.map +1 -0
- package/dist/converters/elastic.js +125 -0
- package/dist/converters/elastic.js.map +1 -0
- package/dist/converters/index.d.ts +28 -0
- package/dist/converters/index.d.ts.map +1 -0
- package/dist/converters/index.js +36 -0
- package/dist/converters/index.js.map +1 -0
- package/dist/converters/splunk.d.ts +19 -0
- package/dist/converters/splunk.d.ts.map +1 -0
- package/dist/converters/splunk.js +148 -0
- package/dist/converters/splunk.js.map +1 -0
- package/dist/embedding/build-corpus.d.ts +15 -0
- package/dist/embedding/build-corpus.d.ts.map +1 -0
- package/dist/embedding/build-corpus.js +105 -0
- package/dist/embedding/build-corpus.js.map +1 -0
- package/dist/embedding/model-loader.d.ts +41 -0
- package/dist/embedding/model-loader.d.ts.map +1 -0
- package/dist/embedding/model-loader.js +90 -0
- package/dist/embedding/model-loader.js.map +1 -0
- package/dist/embedding/vector-store.d.ts +41 -0
- package/dist/embedding/vector-store.d.ts.map +1 -0
- package/dist/embedding/vector-store.js +70 -0
- package/dist/embedding/vector-store.js.map +1 -0
- package/dist/engine.d.ts +23 -20
- package/dist/engine.d.ts.map +1 -1
- package/dist/engine.js +173 -24
- package/dist/engine.js.map +1 -1
- package/dist/eval/corpus.d.ts +42 -0
- package/dist/eval/corpus.d.ts.map +1 -0
- package/dist/eval/corpus.js +427 -0
- package/dist/eval/corpus.js.map +1 -0
- package/dist/eval/eval-harness.d.ts +44 -0
- package/dist/eval/eval-harness.d.ts.map +1 -0
- package/dist/eval/eval-harness.js +296 -0
- package/dist/eval/eval-harness.js.map +1 -0
- package/dist/eval/index.d.ts +13 -0
- package/dist/eval/index.d.ts.map +1 -0
- package/dist/eval/index.js +9 -0
- package/dist/eval/index.js.map +1 -0
- package/dist/eval/metrics.d.ts +74 -0
- package/dist/eval/metrics.d.ts.map +1 -0
- package/dist/eval/metrics.js +108 -0
- package/dist/eval/metrics.js.map +1 -0
- package/dist/eval/pint-corpus.d.ts +34 -0
- package/dist/eval/pint-corpus.d.ts.map +1 -0
- package/dist/eval/pint-corpus.js +109 -0
- package/dist/eval/pint-corpus.js.map +1 -0
- package/dist/eval/rule-corpus.d.ts +9 -0
- package/dist/eval/rule-corpus.d.ts.map +1 -0
- package/dist/eval/rule-corpus.js +4780 -0
- package/dist/eval/rule-corpus.js.map +1 -0
- package/dist/eval/rule-metrics.d.ts +34 -0
- package/dist/eval/rule-metrics.d.ts.map +1 -0
- package/dist/eval/rule-metrics.js +92 -0
- package/dist/eval/rule-metrics.js.map +1 -0
- package/dist/eval/run-eval.d.ts +7 -0
- package/dist/eval/run-eval.d.ts.map +1 -0
- package/dist/eval/run-eval.js +11 -0
- package/dist/eval/run-eval.js.map +1 -0
- package/dist/eval/run-pint-benchmark.d.ts +18 -0
- package/dist/eval/run-pint-benchmark.d.ts.map +1 -0
- package/dist/eval/run-pint-benchmark.js +157 -0
- package/dist/eval/run-pint-benchmark.js.map +1 -0
- package/dist/flywheel.d.ts +54 -0
- package/dist/flywheel.d.ts.map +1 -0
- package/dist/flywheel.js +121 -0
- package/dist/flywheel.js.map +1 -0
- package/dist/index.d.ts +21 -1
- package/dist/index.d.ts.map +1 -1
- package/dist/index.js +19 -2
- package/dist/index.js.map +1 -1
- package/dist/modules/embedding.d.ts +71 -0
- package/dist/modules/embedding.d.ts.map +1 -0
- package/dist/modules/embedding.js +141 -0
- package/dist/modules/embedding.js.map +1 -0
- package/dist/modules/semantic.d.ts +1 -0
- package/dist/modules/semantic.d.ts.map +1 -1
- package/dist/modules/semantic.js +77 -1
- package/dist/modules/semantic.js.map +1 -1
- package/dist/rule-scaffolder.d.ts +14 -0
- package/dist/rule-scaffolder.d.ts.map +1 -1
- package/dist/rule-scaffolder.js +123 -6
- package/dist/rule-scaffolder.js.map +1 -1
- package/dist/session-tracker.d.ts +2 -0
- package/dist/session-tracker.d.ts.map +1 -1
- package/dist/session-tracker.js +1 -0
- package/dist/session-tracker.js.map +1 -1
- package/dist/shadow-evaluator.d.ts +48 -0
- package/dist/shadow-evaluator.d.ts.map +1 -0
- package/dist/shadow-evaluator.js +128 -0
- package/dist/shadow-evaluator.js.map +1 -0
- package/dist/skill-fingerprint.d.ts.map +1 -1
- package/dist/skill-fingerprint.js +10 -52
- package/dist/skill-fingerprint.js.map +1 -1
- package/dist/tier0-invariant.d.ts +49 -0
- package/dist/tier0-invariant.d.ts.map +1 -0
- package/dist/tier0-invariant.js +184 -0
- package/dist/tier0-invariant.js.map +1 -0
- package/dist/tier1-blacklist.d.ts +48 -0
- package/dist/tier1-blacklist.d.ts.map +1 -0
- package/dist/tier1-blacklist.js +91 -0
- package/dist/tier1-blacklist.js.map +1 -0
- package/package.json +7 -1
- package/rules/agent-manipulation/ATR-2026-108-consensus-sybil-attack.yaml +103 -0
- package/rules/context-exfiltration/ATR-2026-102-disguised-analytics-exfiltration.yaml +69 -0
- package/rules/privilege-escalation/ATR-2026-107-delayed-execution-bypass.yaml +67 -0
- package/rules/prompt-injection/ATR-2026-001-direct-prompt-injection.yaml +181 -94
- package/rules/prompt-injection/ATR-2026-003-jailbreak-attempt.yaml +23 -12
- package/rules/prompt-injection/ATR-2026-004-system-prompt-override.yaml +3 -3
- package/rules/prompt-injection/ATR-2026-081-semantic-multi-turn.yaml +2 -2
- package/rules/prompt-injection/ATR-2026-093-gradual-escalation.yaml +1 -1
- package/rules/prompt-injection/ATR-2026-104-persona-hijacking.yaml +72 -0
- package/rules/tool-poisoning/ATR-2026-100-consent-bypass-instruction.yaml +80 -0
- package/rules/tool-poisoning/ATR-2026-101-trust-escalation-override.yaml +66 -0
- package/rules/tool-poisoning/ATR-2026-103-hidden-safety-bypass-instruction.yaml +71 -0
- package/rules/tool-poisoning/ATR-2026-105-silent-action-concealment.yaml +67 -0
- package/rules/tool-poisoning/ATR-2026-106-schema-description-contradiction.yaml +66 -0
package/dist/flywheel.js
ADDED
|
@@ -0,0 +1,121 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Flywheel Manager -- automates the threat detection → rule generation → promotion cycle.
|
|
3
|
+
*
|
|
4
|
+
* Flow:
|
|
5
|
+
* 1. Tier 4 (LLM) detects novel threat → auto-scaffold rule
|
|
6
|
+
* 2. Rule enters shadow mode → ShadowEvaluator tracks FP rate
|
|
7
|
+
* 3. FP < threshold after N evaluations → auto-promote to stable
|
|
8
|
+
* 4. Promoted rule distributes to all users via Threat Cloud
|
|
9
|
+
*
|
|
10
|
+
* Machine speed, not human speed. No manual proposals or voting required.
|
|
11
|
+
*
|
|
12
|
+
* @module agent-threat-rules/flywheel
|
|
13
|
+
*/
|
|
14
|
+
import { RuleScaffolder } from './rule-scaffolder.js';
|
|
15
|
+
import { ShadowEvaluator } from './shadow-evaluator.js';
|
|
16
|
+
export class FlywheelManager {
|
|
17
|
+
scaffolder;
|
|
18
|
+
shadow;
|
|
19
|
+
config;
|
|
20
|
+
existingIds = new Set();
|
|
21
|
+
constructor(config = {}) {
|
|
22
|
+
this.scaffolder = new RuleScaffolder({ author: 'ATR Flywheel (auto-generated)' });
|
|
23
|
+
this.shadow = new ShadowEvaluator();
|
|
24
|
+
this.config = {
|
|
25
|
+
maxFPRate: config.maxFPRate ?? 0.001,
|
|
26
|
+
minEvaluations: config.minEvaluations ?? 1000,
|
|
27
|
+
onPromote: config.onPromote ?? (() => { }),
|
|
28
|
+
onShadowRule: config.onShadowRule ?? (() => { }),
|
|
29
|
+
};
|
|
30
|
+
}
|
|
31
|
+
/**
|
|
32
|
+
* Called when Tier 4 (LLM semantic) detects a novel threat.
|
|
33
|
+
* Auto-generates a shadow rule from the detection.
|
|
34
|
+
*/
|
|
35
|
+
async onTier4Detection(match, event) {
|
|
36
|
+
// Only generate from high-confidence Tier 4 matches
|
|
37
|
+
if (match.confidence < 0.7)
|
|
38
|
+
return null;
|
|
39
|
+
// Extract category and severity from the match
|
|
40
|
+
const category = match.rule.tags?.category ?? 'prompt-injection';
|
|
41
|
+
const severity = match.rule.severity ?? 'medium';
|
|
42
|
+
// Build example payloads from ATTACK PATTERNS, not just raw content.
|
|
43
|
+
// Priority: matched patterns > event fields > event content
|
|
44
|
+
const payloads = [];
|
|
45
|
+
// 1. Matched patterns from the Tier 4 detection — these ARE the attack signals
|
|
46
|
+
if (match.matchedPatterns.length > 0) {
|
|
47
|
+
payloads.push(...match.matchedPatterns.filter((p) => p.length > 5));
|
|
48
|
+
}
|
|
49
|
+
// 2. Event fields (tool_args, tool_response, etc.) — more specific than content
|
|
50
|
+
if (event.fields) {
|
|
51
|
+
for (const value of Object.values(event.fields)) {
|
|
52
|
+
if (value && value.length > 10) {
|
|
53
|
+
payloads.push(value.slice(0, 500));
|
|
54
|
+
}
|
|
55
|
+
}
|
|
56
|
+
}
|
|
57
|
+
// 3. Event content as fallback — but only if we don't have better signals
|
|
58
|
+
if (payloads.length === 0 && event.content) {
|
|
59
|
+
payloads.push(event.content.slice(0, 500));
|
|
60
|
+
}
|
|
61
|
+
// Ensure at least one payload
|
|
62
|
+
if (payloads.length === 0) {
|
|
63
|
+
payloads.push(match.rule.description ?? match.rule.title);
|
|
64
|
+
}
|
|
65
|
+
const input = {
|
|
66
|
+
title: `Auto: ${match.rule.description?.slice(0, 60) ?? match.rule.title}`,
|
|
67
|
+
category: category,
|
|
68
|
+
severity: severity,
|
|
69
|
+
attackDescription: match.rule.description ?? match.matchedPatterns.join('; '),
|
|
70
|
+
examplePayloads: payloads,
|
|
71
|
+
};
|
|
72
|
+
try {
|
|
73
|
+
const result = this.scaffolder.scaffold(input, this.existingIds);
|
|
74
|
+
const ruleYaml = result.yaml;
|
|
75
|
+
// Parse back to ATRRule object
|
|
76
|
+
const { default: yaml } = await import('js-yaml');
|
|
77
|
+
const rule = yaml.load(ruleYaml);
|
|
78
|
+
rule.status = 'experimental';
|
|
79
|
+
this.existingIds.add(result.id);
|
|
80
|
+
this.shadow.addRule(rule);
|
|
81
|
+
await this.config.onShadowRule(rule);
|
|
82
|
+
return rule;
|
|
83
|
+
}
|
|
84
|
+
catch {
|
|
85
|
+
return null;
|
|
86
|
+
}
|
|
87
|
+
}
|
|
88
|
+
/**
|
|
89
|
+
* Called for every event -- runs shadow evaluation.
|
|
90
|
+
* Returns shadow matches (for logging only, not verdict).
|
|
91
|
+
*/
|
|
92
|
+
evaluateShadow(event) {
|
|
93
|
+
return this.shadow.evaluate(event);
|
|
94
|
+
}
|
|
95
|
+
/** Record user feedback on a shadow match */
|
|
96
|
+
recordFeedback(ruleId, isTruePositive) {
|
|
97
|
+
this.shadow.recordFeedback(ruleId, isTruePositive);
|
|
98
|
+
}
|
|
99
|
+
/**
|
|
100
|
+
* Check for rules ready to promote and execute promotion.
|
|
101
|
+
* Call periodically (e.g., every 15 minutes).
|
|
102
|
+
*/
|
|
103
|
+
async promoteReady() {
|
|
104
|
+
const candidates = this.shadow.getPromotionCandidates(this.config.maxFPRate, this.config.minEvaluations);
|
|
105
|
+
for (const candidate of candidates) {
|
|
106
|
+
// Promote: change status from experimental to stable
|
|
107
|
+
const promoted = { ...candidate.rule, status: 'stable' };
|
|
108
|
+
await this.config.onPromote(promoted, candidate.stats);
|
|
109
|
+
}
|
|
110
|
+
return candidates;
|
|
111
|
+
}
|
|
112
|
+
/** Get shadow evaluator stats */
|
|
113
|
+
getShadowStats() {
|
|
114
|
+
return this.shadow.getAllStats();
|
|
115
|
+
}
|
|
116
|
+
/** Number of rules in shadow mode */
|
|
117
|
+
shadowRuleCount() {
|
|
118
|
+
return this.shadow.size();
|
|
119
|
+
}
|
|
120
|
+
}
|
|
121
|
+
//# sourceMappingURL=flywheel.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"flywheel.js","sourceRoot":"","sources":["../src/flywheel.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;GAYG;AAGH,OAAO,EAAE,cAAc,EAAsB,MAAM,sBAAsB,CAAC;AAC1E,OAAO,EAAE,eAAe,EAA2B,MAAM,uBAAuB,CAAC;AAajF,MAAM,OAAO,eAAe;IACT,UAAU,CAAiB;IAC3B,MAAM,CAAkB;IACxB,MAAM,CAA2B;IACjC,WAAW,GAAG,IAAI,GAAG,EAAU,CAAC;IAEjD,YAAY,SAAyB,EAAE;QACrC,IAAI,CAAC,UAAU,GAAG,IAAI,cAAc,CAAC,EAAE,MAAM,EAAE,+BAA+B,EAAE,CAAC,CAAC;QAClF,IAAI,CAAC,MAAM,GAAG,IAAI,eAAe,EAAE,CAAC;QACpC,IAAI,CAAC,MAAM,GAAG;YACZ,SAAS,EAAE,MAAM,CAAC,SAAS,IAAI,KAAK;YACpC,cAAc,EAAE,MAAM,CAAC,cAAc,IAAI,IAAI;YAC7C,SAAS,EAAE,MAAM,CAAC,SAAS,IAAI,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC;YACzC,YAAY,EAAE,MAAM,CAAC,YAAY,IAAI,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC;SAChD,CAAC;IACJ,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,gBAAgB,CAAC,KAAe,EAAE,KAAiB;QACvD,oDAAoD;QACpD,IAAI,KAAK,CAAC,UAAU,GAAG,GAAG;YAAE,OAAO,IAAI,CAAC;QAExC,+CAA+C;QAC/C,MAAM,QAAQ,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,EAAE,QAAQ,IAAI,kBAAkB,CAAC;QACjE,MAAM,QAAQ,GAAG,KAAK,CAAC,IAAI,CAAC,QAAQ,IAAI,QAAQ,CAAC;QAEjD,qEAAqE;QACrE,4DAA4D;QAC5D,MAAM,QAAQ,GAAa,EAAE,CAAC;QAE9B,+EAA+E;QAC/E,IAAI,KAAK,CAAC,eAAe,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACrC,QAAQ,CAAC,IAAI,CAAC,GAAG,KAAK,CAAC,eAAe,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC;QACtE,CAAC;QAED,gFAAgF;QAChF,IAAI,KAAK,CAAC,MAAM,EAAE,CAAC;YACjB,KAAK,MAAM,KAAK,IAAI,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,EAAE,CAAC;gBAChD,IAAI,KAAK,IAAI,KAAK,CAAC,MAAM,GAAG,EAAE,EAAE,CAAC;oBAC/B,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC,CAAC;gBACrC,CAAC;YACH,CAAC;QACH,CAAC;QAED,0EAA0E;QAC1E,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC,IAAI,KAAK,CAAC,OAAO,EAAE,CAAC;YAC3C,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC,CAAC;QAC7C,CAAC;QAED,8BAA8B;QAC9B,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC1B,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,WAAW,IAAI,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC5D,CAAC;QAED,MAAM,KAAK,GAAkB;YAC3B,KAAK,EAAE,SAAS,KAAK,CAAC,IAAI,CAAC,WAAW,EAAE,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,KAAK,CAAC,IAAI,CAAC,KAAK,EAAE;YAC1E,QAAQ,EAAE,QAAqC;YAC/C,QAAQ,EAAE,QAAqC;YAC/C,iBAAiB,EAAE,KAAK,CAAC,IAAI,CAAC,WAAW,IAAI,KAAK,CAAC,eAAe,CAAC,IAAI,CAAC,IAAI,CAAC;YAC7E,eAAe,EAAE,QAAQ;SAC1B,CAAC;QAEF,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,KAAK,EAAE,IAAI,CAAC,WAAW,CAAC,CAAC;YACjE,MAAM,QAAQ,GAAG,MAAM,CAAC,IAAI,CAAC;YAE7B,+BAA+B;YAC/B,MAAM,EAAE,OAAO,EAAE,IAAI,EAAE,GAAG,MAAM,MAAM,CAAC,SAAS,CAAC,CAAC;YAClD,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAY,CAAC;YAC5C,IAAI,CAAC,MAAM,GAAG,cAAc,CAAC;YAE7B,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;YAChC,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;YAE1B,MAAM,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;YAErC,OAAO,IAAI,CAAC;QACd,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,IAAI,CAAC;QACd,CAAC;IACH,CAAC;IAED;;;OAGG;IACH,cAAc,CAAC,KAAiB;QAC9B,OAAO,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;IACrC,CAAC;IAED,6CAA6C;IAC7C,cAAc,CAAC,MAAc,EAAE,cAAuB;QACpD,IAAI,CAAC,MAAM,CAAC,cAAc,CAAC,MAAM,EAAE,cAAc,CAAC,CAAC;IACrD,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,YAAY;QAChB,MAAM,UAAU,GAAG,IAAI,CAAC,MAAM,CAAC,sBAAsB,CACnD,IAAI,CAAC,MAAM,CAAC,SAAS,EACrB,IAAI,CAAC,MAAM,CAAC,cAAc,CAC3B,CAAC;QAEF,KAAK,MAAM,SAAS,IAAI,UAAU,EAAE,CAAC;YACnC,qDAAqD;YACrD,MAAM,QAAQ,GAAG,EAAE,GAAG,SAAS,CAAC,IAAI,EAAE,MAAM,EAAE,QAAiB,EAAE,CAAC;YAClE,MAAM,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,QAAQ,EAAE,SAAS,CAAC,KAAK,CAAC,CAAC;QACzD,CAAC;QAED,OAAO,UAAU,CAAC;IACpB,CAAC;IAED,iCAAiC;IACjC,cAAc;QACZ,OAAO,IAAI,CAAC,MAAM,CAAC,WAAW,EAAE,CAAC;IACnC,CAAC;IAED,qCAAqC;IACrC,eAAe;QACb,OAAO,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC;IAC5B,CAAC;CACF"}
|
package/dist/index.d.ts
CHANGED
|
@@ -4,7 +4,7 @@
|
|
|
4
4
|
* ATR is the detection layer: it evaluates agent events against rules
|
|
5
5
|
* and returns match results. It does NOT execute response actions,
|
|
6
6
|
* send notifications, or manage dashboards. Those are the responsibility
|
|
7
|
-
* of products built on ATR (e.g.,
|
|
7
|
+
* of products built on ATR (e.g., LlamaFirewall, or your own).
|
|
8
8
|
*
|
|
9
9
|
* ATR 是偵測層:評估 agent 事件、回傳匹配結果。
|
|
10
10
|
* 不執行回應動作、不發通知、不管 dashboard。
|
|
@@ -17,6 +17,18 @@ export type { ATREngineConfig } from './engine.js';
|
|
|
17
17
|
export { loadRuleFile, loadRulesFromDirectory, validateRule } from './loader.js';
|
|
18
18
|
export { SessionTracker } from './session-tracker.js';
|
|
19
19
|
export type { SessionStateSnapshot } from './session-tracker.js';
|
|
20
|
+
export { InvariantChecker } from './tier0-invariant.js';
|
|
21
|
+
export type { SkillManifest, InvariantViolation, InvariantViolationType } from './tier0-invariant.js';
|
|
22
|
+
export { InMemoryBlacklist, buildBlacklistMatch } from './tier1-blacklist.js';
|
|
23
|
+
export type { BlacklistProvider, BlacklistEntry } from './tier1-blacklist.js';
|
|
24
|
+
export { extractCapabilities } from './capability-extractor.js';
|
|
25
|
+
export type { ExtractedCapabilities } from './capability-extractor.js';
|
|
26
|
+
export { EmbeddingModule } from './modules/embedding.js';
|
|
27
|
+
export type { EmbeddingModuleConfig } from './modules/embedding.js';
|
|
28
|
+
export { VectorStore, loadVectorEntries } from './embedding/vector-store.js';
|
|
29
|
+
export type { VectorEntry, SearchResult } from './embedding/vector-store.js';
|
|
30
|
+
export { TransformersJSModel, MockEmbeddingModel } from './embedding/model-loader.js';
|
|
31
|
+
export type { EmbeddingModel } from './embedding/model-loader.js';
|
|
20
32
|
export { ModuleRegistry } from './modules/index.js';
|
|
21
33
|
export type { ATRModule, ModuleCondition, ModuleResult } from './modules/index.js';
|
|
22
34
|
export { SessionModule } from './modules/session.js';
|
|
@@ -31,6 +43,14 @@ export { RuleScaffolder } from './rule-scaffolder.js';
|
|
|
31
43
|
export type { ScaffoldInput, ScaffoldResult, ScaffoldOptions } from './rule-scaffolder.js';
|
|
32
44
|
export { CoverageAnalyzer } from './coverage-analyzer.js';
|
|
33
45
|
export type { CoverageGap, CoverageReport } from './coverage-analyzer.js';
|
|
46
|
+
export { convertRule, convertAllRules } from './converters/index.js';
|
|
47
|
+
export type { ConvertedQuery, SIEMFormat } from './converters/index.js';
|
|
48
|
+
export { ruleToSPL } from './converters/splunk.js';
|
|
49
|
+
export { ruleToElastic } from './converters/elastic.js';
|
|
50
|
+
export { ShadowEvaluator } from './shadow-evaluator.js';
|
|
51
|
+
export type { PromotionCandidate } from './shadow-evaluator.js';
|
|
52
|
+
export { FlywheelManager } from './flywheel.js';
|
|
53
|
+
export type { FlywheelConfig } from './flywheel.js';
|
|
34
54
|
export { computeVerdict, SEVERITY_RANK, isAutoResponseEnabled } from './verdict.js';
|
|
35
55
|
export { ActionExecutor } from './action-executor.js';
|
|
36
56
|
export type { ActionExecutorConfig } from './action-executor.js';
|
package/dist/index.d.ts.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;GAaG;AAGH,OAAO,EAAE,SAAS,EAAE,MAAM,aAAa,CAAC;AACxC,YAAY,EAAE,eAAe,EAAE,MAAM,aAAa,CAAC;AACnD,OAAO,EAAE,YAAY,EAAE,sBAAsB,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AACjF,OAAO,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAC;AACtD,YAAY,EAAE,oBAAoB,EAAE,MAAM,sBAAsB,CAAC;AAGjE,OAAO,EAAE,cAAc,EAAE,MAAM,oBAAoB,CAAC;AACpD,YAAY,EAAE,SAAS,EAAE,eAAe,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AACnF,OAAO,EAAE,aAAa,EAAE,MAAM,sBAAsB,CAAC;AACrD,kDAAkD;AAClD,OAAO,EAAE,cAAc,EAAE,MAAM,uBAAuB,CAAC;AACvD,YAAY,EAAE,oBAAoB,EAAE,MAAM,uBAAuB,CAAC;AAClE,kDAAkD;AAClD,OAAO,EAAE,qBAAqB,EAAE,MAAM,wBAAwB,CAAC;AAC/D,YAAY,EACV,gBAAgB,EAChB,eAAe,EACf,sBAAsB,GACvB,MAAM,wBAAwB,CAAC;AAChC,YAAY,EAAE,mBAAmB,EAAE,MAAM,wBAAwB,CAAC;AAGlE,OAAO,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAC;AACtD,YAAY,EAAE,aAAa,EAAE,cAAc,EAAE,eAAe,EAAE,MAAM,sBAAsB,CAAC;AAC3F,OAAO,EAAE,gBAAgB,EAAE,MAAM,wBAAwB,CAAC;AAC1D,YAAY,EAAE,WAAW,EAAE,cAAc,EAAE,MAAM,wBAAwB,CAAC;
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;GAaG;AAGH,OAAO,EAAE,SAAS,EAAE,MAAM,aAAa,CAAC;AACxC,YAAY,EAAE,eAAe,EAAE,MAAM,aAAa,CAAC;AACnD,OAAO,EAAE,YAAY,EAAE,sBAAsB,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AACjF,OAAO,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAC;AACtD,YAAY,EAAE,oBAAoB,EAAE,MAAM,sBAAsB,CAAC;AAGjE,OAAO,EAAE,gBAAgB,EAAE,MAAM,sBAAsB,CAAC;AACxD,YAAY,EAAE,aAAa,EAAE,kBAAkB,EAAE,sBAAsB,EAAE,MAAM,sBAAsB,CAAC;AAGtG,OAAO,EAAE,iBAAiB,EAAE,mBAAmB,EAAE,MAAM,sBAAsB,CAAC;AAC9E,YAAY,EAAE,iBAAiB,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAC;AAG9E,OAAO,EAAE,mBAAmB,EAAE,MAAM,2BAA2B,CAAC;AAChE,YAAY,EAAE,qBAAqB,EAAE,MAAM,2BAA2B,CAAC;AAGvE,OAAO,EAAE,eAAe,EAAE,MAAM,wBAAwB,CAAC;AACzD,YAAY,EAAE,qBAAqB,EAAE,MAAM,wBAAwB,CAAC;AACpE,OAAO,EAAE,WAAW,EAAE,iBAAiB,EAAE,MAAM,6BAA6B,CAAC;AAC7E,YAAY,EAAE,WAAW,EAAE,YAAY,EAAE,MAAM,6BAA6B,CAAC;AAC7E,OAAO,EAAE,mBAAmB,EAAE,kBAAkB,EAAE,MAAM,6BAA6B,CAAC;AACtF,YAAY,EAAE,cAAc,EAAE,MAAM,6BAA6B,CAAC;AAGlE,OAAO,EAAE,cAAc,EAAE,MAAM,oBAAoB,CAAC;AACpD,YAAY,EAAE,SAAS,EAAE,eAAe,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AACnF,OAAO,EAAE,aAAa,EAAE,MAAM,sBAAsB,CAAC;AACrD,kDAAkD;AAClD,OAAO,EAAE,cAAc,EAAE,MAAM,uBAAuB,CAAC;AACvD,YAAY,EAAE,oBAAoB,EAAE,MAAM,uBAAuB,CAAC;AAClE,kDAAkD;AAClD,OAAO,EAAE,qBAAqB,EAAE,MAAM,wBAAwB,CAAC;AAC/D,YAAY,EACV,gBAAgB,EAChB,eAAe,EACf,sBAAsB,GACvB,MAAM,wBAAwB,CAAC;AAChC,YAAY,EAAE,mBAAmB,EAAE,MAAM,wBAAwB,CAAC;AAGlE,OAAO,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAC;AACtD,YAAY,EAAE,aAAa,EAAE,cAAc,EAAE,eAAe,EAAE,MAAM,sBAAsB,CAAC;AAC3F,OAAO,EAAE,gBAAgB,EAAE,MAAM,wBAAwB,CAAC;AAC1D,YAAY,EAAE,WAAW,EAAE,cAAc,EAAE,MAAM,wBAAwB,CAAC;AAG1E,OAAO,EAAE,WAAW,EAAE,eAAe,EAAE,MAAM,uBAAuB,CAAC;AACrE,YAAY,EAAE,cAAc,EAAE,UAAU,EAAE,MAAM,uBAAuB,CAAC;AACxE,OAAO,EAAE,SAAS,EAAE,MAAM,wBAAwB,CAAC;AACnD,OAAO,EAAE,aAAa,EAAE,MAAM,yBAAyB,CAAC;AAGxD,OAAO,EAAE,eAAe,EAAE,MAAM,uBAAuB,CAAC;AACxD,YAAY,EAAE,kBAAkB,EAAE,MAAM,uBAAuB,CAAC;AAChE,OAAO,EAAE,eAAe,EAAE,MAAM,eAAe,CAAC;AAChD,YAAY,EAAE,cAAc,EAAE,MAAM,eAAe,CAAC;AAKpD,OAAO,EAAE,cAAc,EAAE,aAAa,EAAE,qBAAqB,EAAE,MAAM,cAAc,CAAC;AACpF,OAAO,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAC;AACtD,YAAY,EAAE,oBAAoB,EAAE,MAAM,sBAAsB,CAAC;AACjE,OAAO,EAAE,cAAc,EAAE,MAAM,+BAA+B,CAAC;AAC/D,OAAO,EAAE,YAAY,EAAE,MAAM,6BAA6B,CAAC;AAC3D,OAAO,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAChD,YAAY,EAAE,iBAAiB,EAAE,MAAM,mBAAmB,CAAC;AAC3D,YAAY,EACV,OAAO,EACP,QAAQ,EACR,UAAU,EACV,cAAc,EACd,SAAS,EACT,WAAW,EACX,WAAW,EACX,SAAS,EACT,aAAa,EACb,aAAa,EACb,YAAY,EACZ,WAAW,EACX,aAAa,EACb,OAAO,EACP,cAAc,EACd,YAAY,EACZ,WAAW,EACX,YAAY,EACZ,WAAW,EACX,mBAAmB,EACnB,sBAAsB,EACtB,oBAAoB,EACpB,eAAe,EACf,cAAc,EACd,UAAU,EACV,YAAY,EACZ,gBAAgB,EAChB,eAAe,EACf,SAAS,EACT,UAAU,GACX,MAAM,YAAY,CAAC"}
|
package/dist/index.js
CHANGED
|
@@ -4,7 +4,7 @@
|
|
|
4
4
|
* ATR is the detection layer: it evaluates agent events against rules
|
|
5
5
|
* and returns match results. It does NOT execute response actions,
|
|
6
6
|
* send notifications, or manage dashboards. Those are the responsibility
|
|
7
|
-
* of products built on ATR (e.g.,
|
|
7
|
+
* of products built on ATR (e.g., LlamaFirewall, or your own).
|
|
8
8
|
*
|
|
9
9
|
* ATR 是偵測層:評估 agent 事件、回傳匹配結果。
|
|
10
10
|
* 不執行回應動作、不發通知、不管 dashboard。
|
|
@@ -16,6 +16,16 @@
|
|
|
16
16
|
export { ATREngine } from './engine.js';
|
|
17
17
|
export { loadRuleFile, loadRulesFromDirectory, validateRule } from './loader.js';
|
|
18
18
|
export { SessionTracker } from './session-tracker.js';
|
|
19
|
+
// ── Tier 0: Invariant Enforcement (hard boundaries) ──────────────
|
|
20
|
+
export { InvariantChecker } from './tier0-invariant.js';
|
|
21
|
+
// ── Tier 1: Blacklist Provider (known-bad lookup) ────────────────
|
|
22
|
+
export { InMemoryBlacklist, buildBlacklistMatch } from './tier1-blacklist.js';
|
|
23
|
+
// ── Shared Capability Extraction ─────────────────────────────────
|
|
24
|
+
export { extractCapabilities } from './capability-extractor.js';
|
|
25
|
+
// ── Tier 2.5: Embedding Similarity ───────────────────────────────
|
|
26
|
+
export { EmbeddingModule } from './modules/embedding.js';
|
|
27
|
+
export { VectorStore, loadVectorEntries } from './embedding/vector-store.js';
|
|
28
|
+
export { TransformersJSModel, MockEmbeddingModel } from './embedding/model-loader.js';
|
|
19
29
|
// ── Optional Detection Modules (Layer 2-3, beta) ────────────────
|
|
20
30
|
export { ModuleRegistry } from './modules/index.js';
|
|
21
31
|
export { SessionModule } from './modules/session.js';
|
|
@@ -26,8 +36,15 @@ export { SkillFingerprintStore } from './skill-fingerprint.js';
|
|
|
26
36
|
// ── Tooling (rule authoring and coverage analysis) ──────────────
|
|
27
37
|
export { RuleScaffolder } from './rule-scaffolder.js';
|
|
28
38
|
export { CoverageAnalyzer } from './coverage-analyzer.js';
|
|
39
|
+
// ── SIEM Converters (Splunk SPL, Elasticsearch Query DSL) ────────
|
|
40
|
+
export { convertRule, convertAllRules } from './converters/index.js';
|
|
41
|
+
export { ruleToSPL } from './converters/splunk.js';
|
|
42
|
+
export { ruleToElastic } from './converters/elastic.js';
|
|
43
|
+
// ── Flywheel (auto rule generation + shadow + promotion) ─────────
|
|
44
|
+
export { ShadowEvaluator } from './shadow-evaluator.js';
|
|
45
|
+
export { FlywheelManager } from './flywheel.js';
|
|
29
46
|
// ── Integration Helpers (for products built on ATR) ─────────────
|
|
30
|
-
// These help products like
|
|
47
|
+
// These help products like LlamaFirewall, etc. build
|
|
31
48
|
// protection layers on top of ATR detection results.
|
|
32
49
|
export { computeVerdict, SEVERITY_RANK, isAutoResponseEnabled } from './verdict.js';
|
|
33
50
|
export { ActionExecutor } from './action-executor.js';
|
package/dist/index.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;GAaG;AAEH,mEAAmE;AACnE,OAAO,EAAE,SAAS,EAAE,MAAM,aAAa,CAAC;AAExC,OAAO,EAAE,YAAY,EAAE,sBAAsB,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AACjF,OAAO,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAC;AAGtD,mEAAmE;AACnE,OAAO,EAAE,cAAc,EAAE,MAAM,oBAAoB,CAAC;AAEpD,OAAO,EAAE,aAAa,EAAE,MAAM,sBAAsB,CAAC;AACrD,kDAAkD;AAClD,OAAO,EAAE,cAAc,EAAE,MAAM,uBAAuB,CAAC;AAEvD,kDAAkD;AAClD,OAAO,EAAE,qBAAqB,EAAE,MAAM,wBAAwB,CAAC;AAQ/D,mEAAmE;AACnE,OAAO,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAC;AAEtD,OAAO,EAAE,gBAAgB,EAAE,MAAM,wBAAwB,CAAC;AAG1D,mEAAmE;AACnE
|
|
1
|
+
{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;GAaG;AAEH,mEAAmE;AACnE,OAAO,EAAE,SAAS,EAAE,MAAM,aAAa,CAAC;AAExC,OAAO,EAAE,YAAY,EAAE,sBAAsB,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AACjF,OAAO,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAC;AAGtD,oEAAoE;AACpE,OAAO,EAAE,gBAAgB,EAAE,MAAM,sBAAsB,CAAC;AAGxD,oEAAoE;AACpE,OAAO,EAAE,iBAAiB,EAAE,mBAAmB,EAAE,MAAM,sBAAsB,CAAC;AAG9E,oEAAoE;AACpE,OAAO,EAAE,mBAAmB,EAAE,MAAM,2BAA2B,CAAC;AAGhE,oEAAoE;AACpE,OAAO,EAAE,eAAe,EAAE,MAAM,wBAAwB,CAAC;AAEzD,OAAO,EAAE,WAAW,EAAE,iBAAiB,EAAE,MAAM,6BAA6B,CAAC;AAE7E,OAAO,EAAE,mBAAmB,EAAE,kBAAkB,EAAE,MAAM,6BAA6B,CAAC;AAGtF,mEAAmE;AACnE,OAAO,EAAE,cAAc,EAAE,MAAM,oBAAoB,CAAC;AAEpD,OAAO,EAAE,aAAa,EAAE,MAAM,sBAAsB,CAAC;AACrD,kDAAkD;AAClD,OAAO,EAAE,cAAc,EAAE,MAAM,uBAAuB,CAAC;AAEvD,kDAAkD;AAClD,OAAO,EAAE,qBAAqB,EAAE,MAAM,wBAAwB,CAAC;AAQ/D,mEAAmE;AACnE,OAAO,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAC;AAEtD,OAAO,EAAE,gBAAgB,EAAE,MAAM,wBAAwB,CAAC;AAG1D,oEAAoE;AACpE,OAAO,EAAE,WAAW,EAAE,eAAe,EAAE,MAAM,uBAAuB,CAAC;AAErE,OAAO,EAAE,SAAS,EAAE,MAAM,wBAAwB,CAAC;AACnD,OAAO,EAAE,aAAa,EAAE,MAAM,yBAAyB,CAAC;AAExD,oEAAoE;AACpE,OAAO,EAAE,eAAe,EAAE,MAAM,uBAAuB,CAAC;AAExD,OAAO,EAAE,eAAe,EAAE,MAAM,eAAe,CAAC;AAGhD,mEAAmE;AACnE,qDAAqD;AACrD,qDAAqD;AACrD,OAAO,EAAE,cAAc,EAAE,aAAa,EAAE,qBAAqB,EAAE,MAAM,cAAc,CAAC;AACpF,OAAO,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAC;AAEtD,OAAO,EAAE,cAAc,EAAE,MAAM,+BAA+B,CAAC;AAC/D,OAAO,EAAE,YAAY,EAAE,MAAM,6BAA6B,CAAC;AAC3D,OAAO,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC"}
|
|
@@ -0,0 +1,71 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Embedding Module -- Tier 2.5 semantic similarity detection.
|
|
3
|
+
*
|
|
4
|
+
* Compares incoming text against pre-computed attack embeddings using
|
|
5
|
+
* cosine similarity. Catches paraphrases, multilingual attacks, and
|
|
6
|
+
* semantic variants that regex cannot detect.
|
|
7
|
+
*
|
|
8
|
+
* Uses all-MiniLM-L6-v2 (384 dimensions, ~22MB, runs locally in JS/WASM).
|
|
9
|
+
* No API calls. Optional dependency: @xenova/transformers.
|
|
10
|
+
*
|
|
11
|
+
* @module agent-threat-rules/modules/embedding
|
|
12
|
+
*/
|
|
13
|
+
import type { AgentEvent } from '../types.js';
|
|
14
|
+
import type { ATRModule, ModuleCondition, ModuleResult } from './index.js';
|
|
15
|
+
import { type VectorEntry, type SearchResult } from '../embedding/vector-store.js';
|
|
16
|
+
import type { EmbeddingModel } from '../embedding/model-loader.js';
|
|
17
|
+
export interface EmbeddingModuleConfig {
|
|
18
|
+
/** Pre-loaded attack vector entries */
|
|
19
|
+
readonly attackVectors?: readonly VectorEntry[];
|
|
20
|
+
/** Path to pre-computed attack-embeddings.json file */
|
|
21
|
+
readonly attackVectorsPath?: string;
|
|
22
|
+
/** Raw JSON data (alternative to file path) */
|
|
23
|
+
readonly attackVectorsData?: readonly {
|
|
24
|
+
id: string;
|
|
25
|
+
vector: number[];
|
|
26
|
+
label: string;
|
|
27
|
+
category: string;
|
|
28
|
+
severity: string;
|
|
29
|
+
}[];
|
|
30
|
+
/** Cosine similarity threshold (default: 0.65) */
|
|
31
|
+
readonly similarityThreshold?: number;
|
|
32
|
+
/** Top-K results to consider (default: 3) */
|
|
33
|
+
readonly topK?: number;
|
|
34
|
+
/** Custom embedding model (default: TransformersJSModel) */
|
|
35
|
+
readonly model?: EmbeddingModel;
|
|
36
|
+
}
|
|
37
|
+
export declare class EmbeddingModule implements ATRModule {
|
|
38
|
+
private readonly config;
|
|
39
|
+
readonly name = "embedding";
|
|
40
|
+
readonly description = "Vector similarity detection against known attack embeddings";
|
|
41
|
+
readonly version = "0.1.0";
|
|
42
|
+
readonly functions: {
|
|
43
|
+
name: string;
|
|
44
|
+
description: string;
|
|
45
|
+
args: ({
|
|
46
|
+
name: string;
|
|
47
|
+
type: "string";
|
|
48
|
+
required: boolean;
|
|
49
|
+
description: string;
|
|
50
|
+
} | {
|
|
51
|
+
name: string;
|
|
52
|
+
type: "number";
|
|
53
|
+
required: boolean;
|
|
54
|
+
description: string;
|
|
55
|
+
})[];
|
|
56
|
+
}[];
|
|
57
|
+
private store;
|
|
58
|
+
private model;
|
|
59
|
+
private readonly threshold;
|
|
60
|
+
private readonly topK;
|
|
61
|
+
private initialized;
|
|
62
|
+
constructor(config?: EmbeddingModuleConfig);
|
|
63
|
+
initialize(): Promise<void>;
|
|
64
|
+
evaluate(event: AgentEvent, condition: ModuleCondition): Promise<ModuleResult>;
|
|
65
|
+
/** Get search results with full details (for debugging/testing) */
|
|
66
|
+
searchDetailed(text: string, threshold?: number): Promise<readonly SearchResult[]>;
|
|
67
|
+
destroy(): Promise<void>;
|
|
68
|
+
/** Check if module is operational */
|
|
69
|
+
isAvailable(): boolean;
|
|
70
|
+
}
|
|
71
|
+
//# sourceMappingURL=embedding.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"embedding.d.ts","sourceRoot":"","sources":["../../src/modules/embedding.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;GAWG;AAEH,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAC9C,OAAO,KAAK,EAAE,SAAS,EAAE,eAAe,EAAE,YAAY,EAAE,MAAM,YAAY,CAAC;AAC3E,OAAO,EAGL,KAAK,WAAW,EAChB,KAAK,YAAY,EAClB,MAAM,8BAA8B,CAAC;AACtC,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,8BAA8B,CAAC;AAMnE,MAAM,WAAW,qBAAqB;IACpC,uCAAuC;IACvC,QAAQ,CAAC,aAAa,CAAC,EAAE,SAAS,WAAW,EAAE,CAAC;IAChD,uDAAuD;IACvD,QAAQ,CAAC,iBAAiB,CAAC,EAAE,MAAM,CAAC;IACpC,+CAA+C;IAC/C,QAAQ,CAAC,iBAAiB,CAAC,EAAE,SAAS;QACpC,EAAE,EAAE,MAAM,CAAC;QACX,MAAM,EAAE,MAAM,EAAE,CAAC;QACjB,KAAK,EAAE,MAAM,CAAC;QACd,QAAQ,EAAE,MAAM,CAAC;QACjB,QAAQ,EAAE,MAAM,CAAC;KAClB,EAAE,CAAC;IACJ,kDAAkD;IAClD,QAAQ,CAAC,mBAAmB,CAAC,EAAE,MAAM,CAAC;IACtC,6CAA6C;IAC7C,QAAQ,CAAC,IAAI,CAAC,EAAE,MAAM,CAAC;IACvB,4DAA4D;IAC5D,QAAQ,CAAC,KAAK,CAAC,EAAE,cAAc,CAAC;CACjC;AAMD,qBAAa,eAAgB,YAAW,SAAS;IAgCnC,OAAO,CAAC,QAAQ,CAAC,MAAM;IA/BnC,QAAQ,CAAC,IAAI,eAAe;IAC5B,QAAQ,CAAC,WAAW,iEAAiE;IACrF,QAAQ,CAAC,OAAO,WAAW;IAE3B,QAAQ,CAAC,SAAS;;;;;;;;;;;;;;QAmBhB;IAEF,OAAO,CAAC,KAAK,CAAc;IAC3B,OAAO,CAAC,KAAK,CAAwB;IACrC,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAS;IACnC,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAS;IAC9B,OAAO,CAAC,WAAW,CAAS;gBAEC,MAAM,GAAE,qBAA0B;IAOzD,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC;IAsC3B,QAAQ,CAAC,KAAK,EAAE,UAAU,EAAE,SAAS,EAAE,eAAe,GAAG,OAAO,CAAC,YAAY,CAAC;IA2CpF,mEAAmE;IAC7D,cAAc,CAAC,IAAI,EAAE,MAAM,EAAE,SAAS,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,SAAS,YAAY,EAAE,CAAC;IAMlF,OAAO,IAAI,OAAO,CAAC,IAAI,CAAC;IAK9B,qCAAqC;IACrC,WAAW,IAAI,OAAO;CAGvB"}
|
|
@@ -0,0 +1,141 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Embedding Module -- Tier 2.5 semantic similarity detection.
|
|
3
|
+
*
|
|
4
|
+
* Compares incoming text against pre-computed attack embeddings using
|
|
5
|
+
* cosine similarity. Catches paraphrases, multilingual attacks, and
|
|
6
|
+
* semantic variants that regex cannot detect.
|
|
7
|
+
*
|
|
8
|
+
* Uses all-MiniLM-L6-v2 (384 dimensions, ~22MB, runs locally in JS/WASM).
|
|
9
|
+
* No API calls. Optional dependency: @xenova/transformers.
|
|
10
|
+
*
|
|
11
|
+
* @module agent-threat-rules/modules/embedding
|
|
12
|
+
*/
|
|
13
|
+
import { VectorStore, loadVectorEntries, } from '../embedding/vector-store.js';
|
|
14
|
+
// ---------------------------------------------------------------------------
|
|
15
|
+
// Module
|
|
16
|
+
// ---------------------------------------------------------------------------
|
|
17
|
+
export class EmbeddingModule {
|
|
18
|
+
config;
|
|
19
|
+
name = 'embedding';
|
|
20
|
+
description = 'Vector similarity detection against known attack embeddings';
|
|
21
|
+
version = '0.1.0';
|
|
22
|
+
functions = [
|
|
23
|
+
{
|
|
24
|
+
name: 'similarity_search',
|
|
25
|
+
description: 'Find nearest known attacks by embedding similarity',
|
|
26
|
+
args: [
|
|
27
|
+
{
|
|
28
|
+
name: 'field',
|
|
29
|
+
type: 'string',
|
|
30
|
+
required: false,
|
|
31
|
+
description: 'Event field to embed (default: content)',
|
|
32
|
+
},
|
|
33
|
+
{
|
|
34
|
+
name: 'threshold',
|
|
35
|
+
type: 'number',
|
|
36
|
+
required: false,
|
|
37
|
+
description: 'Similarity threshold override',
|
|
38
|
+
},
|
|
39
|
+
],
|
|
40
|
+
},
|
|
41
|
+
];
|
|
42
|
+
store;
|
|
43
|
+
model;
|
|
44
|
+
threshold;
|
|
45
|
+
topK;
|
|
46
|
+
initialized = false;
|
|
47
|
+
constructor(config = {}) {
|
|
48
|
+
this.config = config;
|
|
49
|
+
this.threshold = config.similarityThreshold ?? 0.65;
|
|
50
|
+
this.topK = config.topK ?? 3;
|
|
51
|
+
this.model = config.model ?? null;
|
|
52
|
+
this.store = new VectorStore(config.attackVectors);
|
|
53
|
+
}
|
|
54
|
+
async initialize() {
|
|
55
|
+
if (this.initialized)
|
|
56
|
+
return;
|
|
57
|
+
// Load attack vectors from data
|
|
58
|
+
if (this.config.attackVectorsData) {
|
|
59
|
+
const entries = loadVectorEntries(this.config.attackVectorsData);
|
|
60
|
+
this.store = this.store.withEntries(entries);
|
|
61
|
+
}
|
|
62
|
+
// Load attack vectors from file
|
|
63
|
+
if (this.config.attackVectorsPath) {
|
|
64
|
+
try {
|
|
65
|
+
const { readFileSync } = await import('node:fs');
|
|
66
|
+
const data = JSON.parse(readFileSync(this.config.attackVectorsPath, 'utf-8'));
|
|
67
|
+
const entries = loadVectorEntries(data);
|
|
68
|
+
this.store = this.store.withEntries(entries);
|
|
69
|
+
}
|
|
70
|
+
catch {
|
|
71
|
+
// File not found = no pre-computed vectors, continue without them
|
|
72
|
+
}
|
|
73
|
+
}
|
|
74
|
+
// Load model if not provided
|
|
75
|
+
if (!this.model) {
|
|
76
|
+
try {
|
|
77
|
+
const { TransformersJSModel } = await import('../embedding/model-loader.js');
|
|
78
|
+
this.model = new TransformersJSModel();
|
|
79
|
+
await this.model.initialize();
|
|
80
|
+
}
|
|
81
|
+
catch (err) {
|
|
82
|
+
// Model not available = module degrades gracefully
|
|
83
|
+
const msg = err instanceof Error ? err.message : String(err);
|
|
84
|
+
console.warn(`[embedding] Model not available: ${msg}. Module disabled.`);
|
|
85
|
+
this.model = null;
|
|
86
|
+
}
|
|
87
|
+
}
|
|
88
|
+
this.initialized = true;
|
|
89
|
+
}
|
|
90
|
+
async evaluate(event, condition) {
|
|
91
|
+
if (!this.model || this.store.size() === 0) {
|
|
92
|
+
return { matched: false, value: 0, description: 'Embedding module not initialized' };
|
|
93
|
+
}
|
|
94
|
+
// Extract text to embed
|
|
95
|
+
const field = condition.args?.field ?? 'content';
|
|
96
|
+
const text = field === 'content'
|
|
97
|
+
? event.content
|
|
98
|
+
: event.fields?.[field] ?? event.content;
|
|
99
|
+
if (!text || text.length < 5) {
|
|
100
|
+
return { matched: false, value: 0, description: 'Input too short for embedding' };
|
|
101
|
+
}
|
|
102
|
+
// Truncate to avoid excessive token usage
|
|
103
|
+
const truncated = text.slice(0, 512);
|
|
104
|
+
try {
|
|
105
|
+
// Encode input
|
|
106
|
+
const queryVec = await this.model.encode(truncated);
|
|
107
|
+
// Search for similar attacks
|
|
108
|
+
const threshold = condition.args?.threshold ?? this.threshold;
|
|
109
|
+
const results = this.store.search(queryVec, this.topK, threshold);
|
|
110
|
+
if (results.length === 0) {
|
|
111
|
+
return { matched: false, value: 0, description: 'No similar attacks found' };
|
|
112
|
+
}
|
|
113
|
+
const top = results[0];
|
|
114
|
+
return {
|
|
115
|
+
matched: true,
|
|
116
|
+
value: top.similarity,
|
|
117
|
+
description: `Similar to known attack: "${top.entry.label}" (${top.entry.category}, similarity: ${top.similarity.toFixed(3)})`,
|
|
118
|
+
};
|
|
119
|
+
}
|
|
120
|
+
catch (err) {
|
|
121
|
+
const msg = err instanceof Error ? err.message : String(err);
|
|
122
|
+
return { matched: false, value: 0, description: `Embedding error: ${msg}` };
|
|
123
|
+
}
|
|
124
|
+
}
|
|
125
|
+
/** Get search results with full details (for debugging/testing) */
|
|
126
|
+
async searchDetailed(text, threshold) {
|
|
127
|
+
if (!this.model || this.store.size() === 0)
|
|
128
|
+
return [];
|
|
129
|
+
const queryVec = await this.model.encode(text.slice(0, 512));
|
|
130
|
+
return this.store.search(queryVec, this.topK, threshold ?? this.threshold);
|
|
131
|
+
}
|
|
132
|
+
async destroy() {
|
|
133
|
+
this.model = null;
|
|
134
|
+
this.initialized = false;
|
|
135
|
+
}
|
|
136
|
+
/** Check if module is operational */
|
|
137
|
+
isAvailable() {
|
|
138
|
+
return this.initialized && this.model !== null && this.store.size() > 0;
|
|
139
|
+
}
|
|
140
|
+
}
|
|
141
|
+
//# sourceMappingURL=embedding.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"embedding.js","sourceRoot":"","sources":["../../src/modules/embedding.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;GAWG;AAIH,OAAO,EACL,WAAW,EACX,iBAAiB,GAGlB,MAAM,8BAA8B,CAAC;AA4BtC,8EAA8E;AAC9E,SAAS;AACT,8EAA8E;AAE9E,MAAM,OAAO,eAAe;IAgCG;IA/BpB,IAAI,GAAG,WAAW,CAAC;IACnB,WAAW,GAAG,6DAA6D,CAAC;IAC5E,OAAO,GAAG,OAAO,CAAC;IAElB,SAAS,GAAG;QACnB;YACE,IAAI,EAAE,mBAAmB;YACzB,WAAW,EAAE,oDAAoD;YACjE,IAAI,EAAE;gBACJ;oBACE,IAAI,EAAE,OAAO;oBACb,IAAI,EAAE,QAAiB;oBACvB,QAAQ,EAAE,KAAK;oBACf,WAAW,EAAE,yCAAyC;iBACvD;gBACD;oBACE,IAAI,EAAE,WAAW;oBACjB,IAAI,EAAE,QAAiB;oBACvB,QAAQ,EAAE,KAAK;oBACf,WAAW,EAAE,+BAA+B;iBAC7C;aACF;SACF;KACF,CAAC;IAEM,KAAK,CAAc;IACnB,KAAK,CAAwB;IACpB,SAAS,CAAS;IAClB,IAAI,CAAS;IACtB,WAAW,GAAG,KAAK,CAAC;IAE5B,YAA6B,SAAgC,EAAE;QAAlC,WAAM,GAAN,MAAM,CAA4B;QAC7D,IAAI,CAAC,SAAS,GAAG,MAAM,CAAC,mBAAmB,IAAI,IAAI,CAAC;QACpD,IAAI,CAAC,IAAI,GAAG,MAAM,CAAC,IAAI,IAAI,CAAC,CAAC;QAC7B,IAAI,CAAC,KAAK,GAAG,MAAM,CAAC,KAAK,IAAI,IAAI,CAAC;QAClC,IAAI,CAAC,KAAK,GAAG,IAAI,WAAW,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC;IACrD,CAAC;IAED,KAAK,CAAC,UAAU;QACd,IAAI,IAAI,CAAC,WAAW;YAAE,OAAO;QAE7B,gCAAgC;QAChC,IAAI,IAAI,CAAC,MAAM,CAAC,iBAAiB,EAAE,CAAC;YAClC,MAAM,OAAO,GAAG,iBAAiB,CAAC,IAAI,CAAC,MAAM,CAAC,iBAAiB,CAAC,CAAC;YACjE,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC;QAC/C,CAAC;QAED,gCAAgC;QAChC,IAAI,IAAI,CAAC,MAAM,CAAC,iBAAiB,EAAE,CAAC;YAClC,IAAI,CAAC;gBACH,MAAM,EAAE,YAAY,EAAE,GAAG,MAAM,MAAM,CAAC,SAAS,CAAC,CAAC;gBACjD,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,IAAI,CAAC,MAAM,CAAC,iBAAiB,EAAE,OAAO,CAAC,CAAC,CAAC;gBAC9E,MAAM,OAAO,GAAG,iBAAiB,CAAC,IAAI,CAAC,CAAC;gBACxC,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC;YAC/C,CAAC;YAAC,MAAM,CAAC;gBACP,kEAAkE;YACpE,CAAC;QACH,CAAC;QAED,6BAA6B;QAC7B,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC;YAChB,IAAI,CAAC;gBACH,MAAM,EAAE,mBAAmB,EAAE,GAAG,MAAM,MAAM,CAAC,8BAA8B,CAAC,CAAC;gBAC7E,IAAI,CAAC,KAAK,GAAG,IAAI,mBAAmB,EAAE,CAAC;gBACvC,MAAM,IAAI,CAAC,KAAK,CAAC,UAAU,EAAE,CAAC;YAChC,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACb,mDAAmD;gBACnD,MAAM,GAAG,GAAG,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;gBAC7D,OAAO,CAAC,IAAI,CAAC,oCAAoC,GAAG,oBAAoB,CAAC,CAAC;gBAC1E,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC;YACpB,CAAC;QACH,CAAC;QAED,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC;IAC1B,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,KAAiB,EAAE,SAA0B;QAC1D,IAAI,CAAC,IAAI,CAAC,KAAK,IAAI,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE,KAAK,CAAC,EAAE,CAAC;YAC3C,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,EAAE,WAAW,EAAE,kCAAkC,EAAE,CAAC;QACvF,CAAC;QAED,wBAAwB;QACxB,MAAM,KAAK,GAAI,SAAS,CAAC,IAAI,EAAE,KAAgB,IAAI,SAAS,CAAC;QAC7D,MAAM,IAAI,GACR,KAAK,KAAK,SAAS;YACjB,CAAC,CAAC,KAAK,CAAC,OAAO;YACf,CAAC,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC,KAAK,CAAC,IAAI,KAAK,CAAC,OAAO,CAAC;QAE7C,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC7B,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,EAAE,WAAW,EAAE,+BAA+B,EAAE,CAAC;QACpF,CAAC;QAED,0CAA0C;QAC1C,MAAM,SAAS,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC;QAErC,IAAI,CAAC;YACH,eAAe;YACf,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;YAEpD,6BAA6B;YAC7B,MAAM,SAAS,GAAI,SAAS,CAAC,IAAI,EAAE,SAAoB,IAAI,IAAI,CAAC,SAAS,CAAC;YAC1E,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,QAAQ,EAAE,IAAI,CAAC,IAAI,EAAE,SAAS,CAAC,CAAC;YAElE,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBACzB,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,EAAE,WAAW,EAAE,0BAA0B,EAAE,CAAC;YAC/E,CAAC;YAED,MAAM,GAAG,GAAG,OAAO,CAAC,CAAC,CAAE,CAAC;YACxB,OAAO;gBACL,OAAO,EAAE,IAAI;gBACb,KAAK,EAAE,GAAG,CAAC,UAAU;gBACrB,WAAW,EAAE,6BAA6B,GAAG,CAAC,KAAK,CAAC,KAAK,MAAM,GAAG,CAAC,KAAK,CAAC,QAAQ,iBAAiB,GAAG,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG;aAC/H,CAAC;QACJ,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,MAAM,GAAG,GAAG,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;YAC7D,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,EAAE,WAAW,EAAE,oBAAoB,GAAG,EAAE,EAAE,CAAC;QAC9E,CAAC;IACH,CAAC;IAED,mEAAmE;IACnE,KAAK,CAAC,cAAc,CAAC,IAAY,EAAE,SAAkB;QACnD,IAAI,CAAC,IAAI,CAAC,KAAK,IAAI,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE,KAAK,CAAC;YAAE,OAAO,EAAE,CAAC;QACtD,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC,CAAC;QAC7D,OAAO,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,QAAQ,EAAE,IAAI,CAAC,IAAI,EAAE,SAAS,IAAI,IAAI,CAAC,SAAS,CAAC,CAAC;IAC7E,CAAC;IAED,KAAK,CAAC,OAAO;QACX,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC;QAClB,IAAI,CAAC,WAAW,GAAG,KAAK,CAAC;IAC3B,CAAC;IAED,qCAAqC;IACrC,WAAW;QACT,OAAO,IAAI,CAAC,WAAW,IAAI,IAAI,CAAC,KAAK,KAAK,IAAI,IAAI,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC;IAC1E,CAAC;CACF"}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"semantic.d.ts","sourceRoot":"","sources":["../../src/modules/semantic.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;GAgBG;AAEH,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAC9C,OAAO,KAAK,EAAE,SAAS,EAAE,eAAe,EAAE,YAAY,EAAE,MAAM,YAAY,CAAC;AAE3E,MAAM,WAAW,oBAAoB;IACnC,qCAAqC;IACrC,MAAM,EAAE,MAAM,CAAC;IACf,cAAc;IACd,MAAM,EAAE,MAAM,CAAC;IACf,8DAA8D;IAC9D,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,6CAA6C;IAC7C,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,iDAAiD;IACjD,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,qCAAqC;IACrC,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,qEAAqE;IACrE,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,wCAAwC;IACxC,YAAY,CAAC,EAAE,MAAM,CAAC;CACvB;
|
|
1
|
+
{"version":3,"file":"semantic.d.ts","sourceRoot":"","sources":["../../src/modules/semantic.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;GAgBG;AAEH,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAC9C,OAAO,KAAK,EAAE,SAAS,EAAE,eAAe,EAAE,YAAY,EAAE,MAAM,YAAY,CAAC;AAE3E,MAAM,WAAW,oBAAoB;IACnC,qCAAqC;IACrC,MAAM,EAAE,MAAM,CAAC;IACf,cAAc;IACd,MAAM,EAAE,MAAM,CAAC;IACf,8DAA8D;IAC9D,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,6CAA6C;IAC7C,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,iDAAiD;IACjD,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,qCAAqC;IACrC,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,qEAAqE;IACrE,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,wCAAwC;IACxC,YAAY,CAAC,EAAE,MAAM,CAAC;CACvB;AA6DD;;;;;;;;;;;;;;;;GAgBG;AACH,qBAAa,cAAe,YAAW,SAAS;IAC9C,QAAQ,CAAC,IAAI,cAAc;IAC3B,QAAQ,CAAC,WAAW,kDAAkD;IACtE,QAAQ,CAAC,OAAO,WAAW;IAE3B,QAAQ,CAAC,SAAS;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;OA2CP;IAEX,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAiC;IACxD,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAiC;gBAE3C,MAAM,EAAE,oBAAoB;IAalC,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC;IAK3B,QAAQ,CAAC,KAAK,EAAE,UAAU,EAAE,SAAS,EAAE,eAAe,GAAG,OAAO,CAAC,YAAY,CAAC;IA8C9E,OAAO,IAAI,OAAO,CAAC,IAAI,CAAC;YAMhB,gBAAgB;YA4ChB,OAAO;YAsDP,WAAW;IAkDzB,OAAO,CAAC,aAAa;IA0BrB,OAAO,CAAC,eAAe;IAOvB,OAAO,CAAC,WAAW;IAUnB,OAAO,CAAC,gBAAgB;CAUzB"}
|
package/dist/modules/semantic.js
CHANGED
|
@@ -38,6 +38,24 @@ Respond with ONLY a JSON object:
|
|
|
38
38
|
Be conservative: legitimate requests should score < 0.3.
|
|
39
39
|
Obvious attacks should score > 0.7.
|
|
40
40
|
Subtle/ambiguous cases should score 0.3-0.7.`;
|
|
41
|
+
const FP_CHECK_PROMPT = `You are a false positive reviewer for an AI agent security system.
|
|
42
|
+
|
|
43
|
+
A previous analysis flagged this text as a potential security threat. Your job is to determine if it is a FALSE POSITIVE -- a legitimate, harmless input that was incorrectly flagged.
|
|
44
|
+
|
|
45
|
+
Common false positives:
|
|
46
|
+
- Legitimate tool descriptions that mention security concepts (e.g., "delete user account")
|
|
47
|
+
- Normal API documentation mentioning credentials or authentication
|
|
48
|
+
- Legitimate admin/management tool descriptions
|
|
49
|
+
- Educational or security research content
|
|
50
|
+
|
|
51
|
+
Respond with ONLY a JSON object:
|
|
52
|
+
{
|
|
53
|
+
"likely_benign": <true or false>,
|
|
54
|
+
"confidence": <0.0 to 1.0>,
|
|
55
|
+
"reasoning": "<1 sentence>"
|
|
56
|
+
}
|
|
57
|
+
|
|
58
|
+
If the text is genuinely suspicious, set likely_benign to false.`;
|
|
41
59
|
/**
|
|
42
60
|
* Semantic detection module using LLM-as-judge.
|
|
43
61
|
*
|
|
@@ -169,7 +187,24 @@ export class SemanticModule {
|
|
|
169
187
|
if (cached && cached.expiresAt > now) {
|
|
170
188
|
return cached.result;
|
|
171
189
|
}
|
|
172
|
-
|
|
190
|
+
let result = await this.callLLM(text);
|
|
191
|
+
// If threat detected with moderate score, run parallel FP check to reduce false positives
|
|
192
|
+
if (result.threatScore >= 0.4 && result.threatScore < 0.85) {
|
|
193
|
+
try {
|
|
194
|
+
const fpResult = await this.callFPCheck(text);
|
|
195
|
+
if (fpResult.likelyBenign && fpResult.confidence >= 0.7) {
|
|
196
|
+
// Reduce threat score -- FP check says it's benign
|
|
197
|
+
result = {
|
|
198
|
+
...result,
|
|
199
|
+
threatScore: result.threatScore * 0.4,
|
|
200
|
+
reasoning: `${result.reasoning} [FP check: likely benign (${fpResult.reasoning})]`,
|
|
201
|
+
};
|
|
202
|
+
}
|
|
203
|
+
}
|
|
204
|
+
catch {
|
|
205
|
+
// FP check failure is non-fatal
|
|
206
|
+
}
|
|
207
|
+
}
|
|
173
208
|
// Evict oldest entries if cache is full
|
|
174
209
|
if (this.cache.size >= this.config.maxCacheSize) {
|
|
175
210
|
const firstKey = this.cache.keys().next().value;
|
|
@@ -228,6 +263,47 @@ export class SemanticModule {
|
|
|
228
263
|
};
|
|
229
264
|
}
|
|
230
265
|
}
|
|
266
|
+
async callFPCheck(text) {
|
|
267
|
+
const truncated = text.length > 1000 ? text.slice(0, 1000) + '...[truncated]' : text;
|
|
268
|
+
const body = {
|
|
269
|
+
model: this.config.model,
|
|
270
|
+
messages: [
|
|
271
|
+
{ role: 'system', content: FP_CHECK_PROMPT },
|
|
272
|
+
{ role: 'user', content: `Is this a false positive?\n\n${truncated}` },
|
|
273
|
+
],
|
|
274
|
+
temperature: 0,
|
|
275
|
+
max_tokens: 256,
|
|
276
|
+
};
|
|
277
|
+
const controller = new AbortController();
|
|
278
|
+
const timeoutId = setTimeout(() => controller.abort(), this.config.timeout);
|
|
279
|
+
const response = await fetch(this.resolveEndpoint(), {
|
|
280
|
+
method: 'POST',
|
|
281
|
+
headers: {
|
|
282
|
+
'Content-Type': 'application/json',
|
|
283
|
+
'Authorization': `Bearer ${this.config.apiKey}`,
|
|
284
|
+
},
|
|
285
|
+
body: JSON.stringify(body),
|
|
286
|
+
signal: controller.signal,
|
|
287
|
+
});
|
|
288
|
+
clearTimeout(timeoutId);
|
|
289
|
+
if (!response.ok) {
|
|
290
|
+
throw new Error(`FP check API error: HTTP ${response.status}`);
|
|
291
|
+
}
|
|
292
|
+
const data = await response.json();
|
|
293
|
+
const content = data.choices?.[0]?.message?.content ?? '';
|
|
294
|
+
try {
|
|
295
|
+
const cleaned = content.replace(/^```(?:json)?\s*\n?/i, '').replace(/\n?```\s*$/, '').trim();
|
|
296
|
+
const parsed = JSON.parse(cleaned);
|
|
297
|
+
return {
|
|
298
|
+
likelyBenign: parsed.likely_benign === true,
|
|
299
|
+
confidence: typeof parsed.confidence === 'number' ? parsed.confidence : 0,
|
|
300
|
+
reasoning: typeof parsed.reasoning === 'string' ? parsed.reasoning : 'unknown',
|
|
301
|
+
};
|
|
302
|
+
}
|
|
303
|
+
catch {
|
|
304
|
+
return { likelyBenign: false, confidence: 0, reasoning: 'Failed to parse FP check response' };
|
|
305
|
+
}
|
|
306
|
+
}
|
|
231
307
|
parseAnalysis(content) {
|
|
232
308
|
try {
|
|
233
309
|
// Strip markdown code blocks if present
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"semantic.js","sourceRoot":"","sources":["../../src/modules/semantic.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;GAgBG;AAwCH,MAAM,mBAAmB,GAAG;;;;;;;;;;;;;;;;;;;;;;6CAsBiB,CAAC;AAE9C;;;;;;;;;;;;;;;;GAgBG;AACH,MAAM,OAAO,cAAc;IAChB,IAAI,GAAG,UAAU,CAAC;IAClB,WAAW,GAAG,8CAA8C,CAAC;IAC7D,OAAO,GAAG,OAAO,CAAC;IAElB,SAAS,GAAG;QACnB;YACE,IAAI,EAAE,gBAAgB;YACtB,WAAW,EAAE,uDAAuD;YACpE,IAAI,EAAE;gBACJ;oBACE,IAAI,EAAE,OAAO;oBACb,IAAI,EAAE,QAAiB;oBACvB,QAAQ,EAAE,KAAK;oBACf,WAAW,EAAE,2CAA2C;iBACzD;aACF;SACF;QACD;YACE,IAAI,EAAE,cAAc;YACpB,WAAW,EAAE,mDAAmD;YAChE,IAAI,EAAE;gBACJ;oBACE,IAAI,EAAE,OAAO;oBACb,IAAI,EAAE,QAAiB;oBACvB,QAAQ,EAAE,KAAK;oBACf,WAAW,EAAE,2CAA2C;iBACzD;aACF;SACF;QACD;YACE,IAAI,EAAE,iBAAiB;YACvB,WAAW,EAAE,2DAA2D;YACxE,IAAI,EAAE;gBACJ;oBACE,IAAI,EAAE,OAAO;oBACb,IAAI,EAAE,QAAiB;oBACvB,QAAQ,EAAE,KAAK;oBACf,WAAW,EAAE,2CAA2C;iBACzD;gBACD;oBACE,IAAI,EAAE,iBAAiB;oBACvB,IAAI,EAAE,QAAiB;oBACvB,QAAQ,EAAE,IAAI;oBACd,WAAW,EAAE,+BAA+B;iBAC7C;aACF;SACF;KACO,CAAC;IAEM,MAAM,CAAiC;IACvC,KAAK,GAAG,IAAI,GAAG,EAAsB,CAAC;IAEvD,YAAY,MAA4B;QACtC,IAAI,CAAC,MAAM,GAAG;YACZ,MAAM,EAAE,MAAM,CAAC,MAAM;YACrB,MAAM,EAAE,MAAM,CAAC,MAAM;YACrB,KAAK,EAAE,MAAM,CAAC,KAAK,IAAI,aAAa;YACpC,SAAS,EAAE,MAAM,CAAC,SAAS,IAAI,GAAG;YAClC,WAAW,EAAE,MAAM,CAAC,WAAW,IAAI,GAAG;YACtC,OAAO,EAAE,MAAM,CAAC,OAAO,IAAI,MAAM;YACjC,UAAU,EAAE,MAAM,CAAC,UAAU,IAAI,OAAO;YACxC,YAAY,EAAE,MAAM,CAAC,YAAY,IAAI,IAAI;SAC1C,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,UAAU;QACd,mDAAmD;QACnD,gEAAgE;IAClE,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,KAAiB,EAAE,SAA0B;QAC1D,MAAM,KAAK,GAAI,SAAS,CAAC,IAAI,CAAC,OAAO,CAAY,IAAI,SAAS,CAAC;QAC/D,MAAM,IAAI,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,KAAK,CAAC,IAAI,KAAK,CAAC,OAAO,CAAC;QAEpD,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC7B,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,EAAE,WAAW,EAAE,uCAAuC,EAAE,CAAC;QAC5F,CAAC;QAED,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC;QAEnD,IAAI,KAAa,CAAC;QAClB,IAAI,WAAmB,CAAC;QAExB,QAAQ,SAAS,CAAC,QAAQ,EAAE,CAAC;YAC3B,KAAK,gBAAgB;gBACnB,KAAK,GAAG,QAAQ,CAAC,WAAW,CAAC;gBAC7B,WAAW,GAAG,QAAQ,CAAC,SAAS,CAAC;gBACjC,MAAM;YAER,KAAK,cAAc,CAAC,CAAC,CAAC;gBACpB,MAAM,WAAW,GAAG,QAAQ,CAAC,QAAQ,KAAK,kBAAkB,IAAI,QAAQ,CAAC,WAAW,IAAI,GAAG,CAAC;gBAC5F,KAAK,GAAG,WAAW,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC;gBAChC,WAAW,GAAG,WAAW;oBACvB,CAAC,CAAC,8BAA8B,QAAQ,CAAC,SAAS,EAAE;oBACpD,CAAC,CAAC,uBAAuB,CAAC;gBAC5B,MAAM;YACR,CAAC;YAED,KAAK,iBAAiB,CAAC,CAAC,CAAC;gBACvB,MAAM,cAAc,GAAG,SAAS,CAAC,IAAI,CAAC,iBAAiB,CAAW,CAAC;gBACnE,MAAM,eAAe,GAAG,QAAQ,CAAC,QAAQ,KAAK,cAAc,CAAC;gBAC7D,KAAK,GAAG,eAAe,CAAC,CAAC,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC,CAAC,GAAG,CAAC;gBACrD,WAAW,GAAG,eAAe;oBAC3B,CAAC,CAAC,WAAW,cAAc,KAAK,QAAQ,CAAC,SAAS,EAAE;oBACpD,CAAC,CAAC,kBAAkB,cAAc,EAAE,CAAC;gBACvC,MAAM;YACR,CAAC;YAED;gBACE,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,EAAE,WAAW,EAAE,qBAAqB,SAAS,CAAC,QAAQ,EAAE,EAAE,CAAC;QAChG,CAAC;QAED,MAAM,OAAO,GAAG,IAAI,CAAC,gBAAgB,CAAC,KAAK,EAAE,SAAS,CAAC,QAAQ,EAAE,SAAS,CAAC,SAAS,CAAC,CAAC;QACtF,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC;IACzC,CAAC;IAED,KAAK,CAAC,OAAO;QACX,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC;IACrB,CAAC;IAED,2BAA2B;IAEnB,KAAK,CAAC,gBAAgB,CAAC,IAAY;QACzC,MAAM,QAAQ,GAAG,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;QACxC,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAEvB,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QACxC,IAAI,MAAM,IAAI,MAAM,CAAC,SAAS,GAAG,GAAG,EAAE,CAAC;YACrC,OAAO,MAAM,CAAC,MAAM,CAAC;QACvB,CAAC;QAED,
|
|
1
|
+
{"version":3,"file":"semantic.js","sourceRoot":"","sources":["../../src/modules/semantic.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;GAgBG;AAwCH,MAAM,mBAAmB,GAAG;;;;;;;;;;;;;;;;;;;;;;6CAsBiB,CAAC;AAE9C,MAAM,eAAe,GAAG;;;;;;;;;;;;;;;;;iEAiByC,CAAC;AAElE;;;;;;;;;;;;;;;;GAgBG;AACH,MAAM,OAAO,cAAc;IAChB,IAAI,GAAG,UAAU,CAAC;IAClB,WAAW,GAAG,8CAA8C,CAAC;IAC7D,OAAO,GAAG,OAAO,CAAC;IAElB,SAAS,GAAG;QACnB;YACE,IAAI,EAAE,gBAAgB;YACtB,WAAW,EAAE,uDAAuD;YACpE,IAAI,EAAE;gBACJ;oBACE,IAAI,EAAE,OAAO;oBACb,IAAI,EAAE,QAAiB;oBACvB,QAAQ,EAAE,KAAK;oBACf,WAAW,EAAE,2CAA2C;iBACzD;aACF;SACF;QACD;YACE,IAAI,EAAE,cAAc;YACpB,WAAW,EAAE,mDAAmD;YAChE,IAAI,EAAE;gBACJ;oBACE,IAAI,EAAE,OAAO;oBACb,IAAI,EAAE,QAAiB;oBACvB,QAAQ,EAAE,KAAK;oBACf,WAAW,EAAE,2CAA2C;iBACzD;aACF;SACF;QACD;YACE,IAAI,EAAE,iBAAiB;YACvB,WAAW,EAAE,2DAA2D;YACxE,IAAI,EAAE;gBACJ;oBACE,IAAI,EAAE,OAAO;oBACb,IAAI,EAAE,QAAiB;oBACvB,QAAQ,EAAE,KAAK;oBACf,WAAW,EAAE,2CAA2C;iBACzD;gBACD;oBACE,IAAI,EAAE,iBAAiB;oBACvB,IAAI,EAAE,QAAiB;oBACvB,QAAQ,EAAE,IAAI;oBACd,WAAW,EAAE,+BAA+B;iBAC7C;aACF;SACF;KACO,CAAC;IAEM,MAAM,CAAiC;IACvC,KAAK,GAAG,IAAI,GAAG,EAAsB,CAAC;IAEvD,YAAY,MAA4B;QACtC,IAAI,CAAC,MAAM,GAAG;YACZ,MAAM,EAAE,MAAM,CAAC,MAAM;YACrB,MAAM,EAAE,MAAM,CAAC,MAAM;YACrB,KAAK,EAAE,MAAM,CAAC,KAAK,IAAI,aAAa;YACpC,SAAS,EAAE,MAAM,CAAC,SAAS,IAAI,GAAG;YAClC,WAAW,EAAE,MAAM,CAAC,WAAW,IAAI,GAAG;YACtC,OAAO,EAAE,MAAM,CAAC,OAAO,IAAI,MAAM;YACjC,UAAU,EAAE,MAAM,CAAC,UAAU,IAAI,OAAO;YACxC,YAAY,EAAE,MAAM,CAAC,YAAY,IAAI,IAAI;SAC1C,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,UAAU;QACd,mDAAmD;QACnD,gEAAgE;IAClE,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,KAAiB,EAAE,SAA0B;QAC1D,MAAM,KAAK,GAAI,SAAS,CAAC,IAAI,CAAC,OAAO,CAAY,IAAI,SAAS,CAAC;QAC/D,MAAM,IAAI,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,KAAK,CAAC,IAAI,KAAK,CAAC,OAAO,CAAC;QAEpD,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC7B,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,EAAE,WAAW,EAAE,uCAAuC,EAAE,CAAC;QAC5F,CAAC;QAED,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC;QAEnD,IAAI,KAAa,CAAC;QAClB,IAAI,WAAmB,CAAC;QAExB,QAAQ,SAAS,CAAC,QAAQ,EAAE,CAAC;YAC3B,KAAK,gBAAgB;gBACnB,KAAK,GAAG,QAAQ,CAAC,WAAW,CAAC;gBAC7B,WAAW,GAAG,QAAQ,CAAC,SAAS,CAAC;gBACjC,MAAM;YAER,KAAK,cAAc,CAAC,CAAC,CAAC;gBACpB,MAAM,WAAW,GAAG,QAAQ,CAAC,QAAQ,KAAK,kBAAkB,IAAI,QAAQ,CAAC,WAAW,IAAI,GAAG,CAAC;gBAC5F,KAAK,GAAG,WAAW,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC;gBAChC,WAAW,GAAG,WAAW;oBACvB,CAAC,CAAC,8BAA8B,QAAQ,CAAC,SAAS,EAAE;oBACpD,CAAC,CAAC,uBAAuB,CAAC;gBAC5B,MAAM;YACR,CAAC;YAED,KAAK,iBAAiB,CAAC,CAAC,CAAC;gBACvB,MAAM,cAAc,GAAG,SAAS,CAAC,IAAI,CAAC,iBAAiB,CAAW,CAAC;gBACnE,MAAM,eAAe,GAAG,QAAQ,CAAC,QAAQ,KAAK,cAAc,CAAC;gBAC7D,KAAK,GAAG,eAAe,CAAC,CAAC,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC,CAAC,GAAG,CAAC;gBACrD,WAAW,GAAG,eAAe;oBAC3B,CAAC,CAAC,WAAW,cAAc,KAAK,QAAQ,CAAC,SAAS,EAAE;oBACpD,CAAC,CAAC,kBAAkB,cAAc,EAAE,CAAC;gBACvC,MAAM;YACR,CAAC;YAED;gBACE,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,EAAE,WAAW,EAAE,qBAAqB,SAAS,CAAC,QAAQ,EAAE,EAAE,CAAC;QAChG,CAAC;QAED,MAAM,OAAO,GAAG,IAAI,CAAC,gBAAgB,CAAC,KAAK,EAAE,SAAS,CAAC,QAAQ,EAAE,SAAS,CAAC,SAAS,CAAC,CAAC;QACtF,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC;IACzC,CAAC;IAED,KAAK,CAAC,OAAO;QACX,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC;IACrB,CAAC;IAED,2BAA2B;IAEnB,KAAK,CAAC,gBAAgB,CAAC,IAAY;QACzC,MAAM,QAAQ,GAAG,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;QACxC,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAEvB,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QACxC,IAAI,MAAM,IAAI,MAAM,CAAC,SAAS,GAAG,GAAG,EAAE,CAAC;YACrC,OAAO,MAAM,CAAC,MAAM,CAAC;QACvB,CAAC;QAED,IAAI,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;QAEtC,0FAA0F;QAC1F,IAAI,MAAM,CAAC,WAAW,IAAI,GAAG,IAAI,MAAM,CAAC,WAAW,GAAG,IAAI,EAAE,CAAC;YAC3D,IAAI,CAAC;gBACH,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;gBAC9C,IAAI,QAAQ,CAAC,YAAY,IAAI,QAAQ,CAAC,UAAU,IAAI,GAAG,EAAE,CAAC;oBACxD,mDAAmD;oBACnD,MAAM,GAAG;wBACP,GAAG,MAAM;wBACT,WAAW,EAAE,MAAM,CAAC,WAAW,GAAG,GAAG;wBACrC,SAAS,EAAE,GAAG,MAAM,CAAC,SAAS,8BAA8B,QAAQ,CAAC,SAAS,IAAI;qBACnF,CAAC;gBACJ,CAAC;YACH,CAAC;YAAC,MAAM,CAAC;gBACP,gCAAgC;YAClC,CAAC;QACH,CAAC;QAED,wCAAwC;QACxC,IAAI,IAAI,CAAC,KAAK,CAAC,IAAI,IAAI,IAAI,CAAC,MAAM,CAAC,YAAY,EAAE,CAAC;YAChD,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC;YAChD,IAAI,QAAQ,KAAK,SAAS,EAAE,CAAC;gBAC3B,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;YAC9B,CAAC;QACH,CAAC;QAED,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,QAAQ,EAAE;YACvB,MAAM;YACN,SAAS,EAAE,GAAG,GAAG,IAAI,CAAC,MAAM,CAAC,UAAU;SACxC,CAAC,CAAC;QAEH,OAAO,MAAM,CAAC;IAChB,CAAC;IAEO,KAAK,CAAC,OAAO,CAAC,IAAY;QAChC,0CAA0C;QAC1C,MAAM,SAAS,GAAG,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC,GAAG,gBAAgB,CAAC,CAAC,CAAC,IAAI,CAAC;QAErF,MAAM,IAAI,GAAG;YACX,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,KAAK;YACxB,QAAQ,EAAE;gBACR,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,mBAAmB,EAAE;gBAChD,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,0BAA0B,SAAS,EAAE,EAAE;aACjE;YACD,WAAW,EAAE,IAAI,CAAC,MAAM,CAAC,WAAW;YACpC,UAAU,EAAE,IAAI,CAAC,MAAM,CAAC,SAAS;SAClC,CAAC;QAEF,IAAI,CAAC;YACH,MAAM,UAAU,GAAG,IAAI,eAAe,EAAE,CAAC;YACzC,MAAM,SAAS,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,UAAU,CAAC,KAAK,EAAE,EAAE,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;YAE5E,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,IAAI,CAAC,eAAe,EAAE,EAAE;gBACnD,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE;oBACP,cAAc,EAAE,kBAAkB;oBAClC,eAAe,EAAE,UAAU,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE;iBAChD;gBACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;gBAC1B,MAAM,EAAE,UAAU,CAAC,MAAM;aAC1B,CAAC,CAAC;YAEH,YAAY,CAAC,SAAS,CAAC,CAAC;YAExB,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;gBACjB,mFAAmF;gBACnF,MAAM,IAAI,KAAK,CAAC,uBAAuB,QAAQ,CAAC,MAAM,EAAE,CAAC,CAAC;YAC5D,CAAC;YAED,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAE/B,CAAC;YAEF,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,OAAO,IAAI,EAAE,CAAC;YAC1D,OAAO,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC;QACrC,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,uDAAuD;YACvD,sEAAsE;YACtE,MAAM,GAAG,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YACnE,OAAO;gBACL,WAAW,EAAE,CAAC;gBACd,QAAQ,EAAE,IAAI;gBACd,SAAS,EAAE,kCAAkC,GAAG,EAAE;gBAClD,cAAc,EAAE,IAAI;aACrB,CAAC;QACJ,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,WAAW,CAAC,IAAY;QACpC,MAAM,SAAS,GAAG,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC,GAAG,gBAAgB,CAAC,CAAC,CAAC,IAAI,CAAC;QAErF,MAAM,IAAI,GAAG;YACX,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,KAAK;YACxB,QAAQ,EAAE;gBACR,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,eAAe,EAAE;gBAC5C,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,gCAAgC,SAAS,EAAE,EAAE;aACvE;YACD,WAAW,EAAE,CAAC;YACd,UAAU,EAAE,GAAG;SAChB,CAAC;QAEF,MAAM,UAAU,GAAG,IAAI,eAAe,EAAE,CAAC;QACzC,MAAM,SAAS,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,UAAU,CAAC,KAAK,EAAE,EAAE,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;QAE5E,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,IAAI,CAAC,eAAe,EAAE,EAAE;YACnD,MAAM,EAAE,MAAM;YACd,OAAO,EAAE;gBACP,cAAc,EAAE,kBAAkB;gBAClC,eAAe,EAAE,UAAU,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE;aAChD;YACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;YAC1B,MAAM,EAAE,UAAU,CAAC,MAAM;SAC1B,CAAC,CAAC;QAEH,YAAY,CAAC,SAAS,CAAC,CAAC;QAExB,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;YACjB,MAAM,IAAI,KAAK,CAAC,4BAA4B,QAAQ,CAAC,MAAM,EAAE,CAAC,CAAC;QACjE,CAAC;QAED,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAE/B,CAAC;QAEF,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,OAAO,IAAI,EAAE,CAAC;QAC1D,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC,sBAAsB,EAAE,EAAE,CAAC,CAAC,OAAO,CAAC,YAAY,EAAE,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;YAC7F,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAyE,CAAC;YAC3G,OAAO;gBACL,YAAY,EAAE,MAAM,CAAC,aAAa,KAAK,IAAI;gBAC3C,UAAU,EAAE,OAAO,MAAM,CAAC,UAAU,KAAK,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC;gBACzE,SAAS,EAAE,OAAO,MAAM,CAAC,SAAS,KAAK,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS;aAC/E,CAAC;QACJ,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,EAAE,YAAY,EAAE,KAAK,EAAE,UAAU,EAAE,CAAC,EAAE,SAAS,EAAE,mCAAmC,EAAE,CAAC;QAChG,CAAC;IACH,CAAC;IAEO,aAAa,CAAC,OAAe;QACnC,IAAI,CAAC;YACH,wCAAwC;YACxC,MAAM,OAAO,GAAG,OAAO;iBACpB,OAAO,CAAC,sBAAsB,EAAE,EAAE,CAAC;iBACnC,OAAO,CAAC,YAAY,EAAE,EAAE,CAAC;iBACzB,IAAI,EAAE,CAAC;YAEV,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAA4B,CAAC;YAE9D,OAAO;gBACL,WAAW,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,MAAM,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;gBAC1E,QAAQ,EAAE,OAAO,MAAM,CAAC,UAAU,CAAC,KAAK,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,IAAI;gBAC5E,SAAS,EAAE,OAAO,MAAM,CAAC,WAAW,CAAC,KAAK,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,uBAAuB;gBAClG,cAAc,EAAE,OAAO,MAAM,CAAC,iBAAiB,CAAC,KAAK,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,iBAAiB,CAAC,CAAC,CAAC,CAAC,IAAI;aACjG,CAAC;QACJ,CAAC;QAAC,MAAM,CAAC;YACP,OAAO;gBACL,WAAW,EAAE,CAAC;gBACd,QAAQ,EAAE,IAAI;gBACd,SAAS,EAAE,8BAA8B;gBACzC,cAAc,EAAE,IAAI;aACrB,CAAC;QACJ,CAAC;IACH,CAAC;IAEO,eAAe;QACrB,MAAM,IAAI,GAAG,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;QACpD,IAAI,IAAI,CAAC,QAAQ,CAAC,mBAAmB,CAAC;YAAE,OAAO,IAAI,CAAC;QACpD,IAAI,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC;YAAE,OAAO,GAAG,IAAI,mBAAmB,CAAC;QAC5D,OAAO,GAAG,IAAI,sBAAsB,CAAC;IACvC,CAAC;IAEO,WAAW,CAAC,IAAY;QAC9B,mCAAmC;QACnC,IAAI,IAAI,GAAG,UAAU,CAAC;QACtB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YACrC,IAAI,IAAI,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC;YAC3B,IAAI,GAAG,CAAC,IAAI,GAAG,UAAU,CAAC,KAAK,CAAC,CAAC;QACnC,CAAC;QACD,OAAO,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;IAC3B,CAAC;IAEO,gBAAgB,CAAC,KAAa,EAAE,QAAgB,EAAE,SAAiB;QACzE,QAAQ,QAAQ,EAAE,CAAC;YACjB,KAAK,IAAI,CAAC,CAAC,OAAO,KAAK,GAAG,SAAS,CAAC;YACpC,KAAK,KAAK,CAAC,CAAC,OAAO,KAAK,IAAI,SAAS,CAAC;YACtC,KAAK,IAAI,CAAC,CAAC,OAAO,KAAK,GAAG,SAAS,CAAC;YACpC,KAAK,KAAK,CAAC,CAAC,OAAO,KAAK,IAAI,SAAS,CAAC;YACtC,KAAK,IAAI,CAAC,CAAC,OAAO,KAAK,KAAK,SAAS,CAAC;YACtC,OAAO,CAAC,CAAC,OAAO,KAAK,IAAI,SAAS,CAAC;QACrC,CAAC;IACH,CAAC;CACF"}
|
|
@@ -22,6 +22,20 @@ export interface ScaffoldOptions {
|
|
|
22
22
|
author?: string;
|
|
23
23
|
schemaVersion?: string;
|
|
24
24
|
}
|
|
25
|
+
/**
|
|
26
|
+
* Attack pattern templates by category — reusable regex building blocks
|
|
27
|
+
* that detect BEHAVIOR, not package names.
|
|
28
|
+
*/
|
|
29
|
+
export declare const ATTACK_PATTERN_INDICATORS: ReadonlyArray<{
|
|
30
|
+
/** Regex to test if the payload contains this attack indicator */
|
|
31
|
+
readonly test: RegExp;
|
|
32
|
+
/** The detection regex to use in the rule */
|
|
33
|
+
readonly pattern: string;
|
|
34
|
+
/** Human-readable description */
|
|
35
|
+
readonly description: string;
|
|
36
|
+
/** Which categories this indicator applies to */
|
|
37
|
+
readonly categories: readonly ATRCategory[];
|
|
38
|
+
}>;
|
|
25
39
|
export declare class RuleScaffolder {
|
|
26
40
|
private readonly options;
|
|
27
41
|
constructor(options?: ScaffoldOptions);
|