agent-threat-rules 0.4.0 → 1.0.0
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 +161 -52
- package/package.json +3 -1
- package/rules/agent-manipulation/{ATR-2026-030-cross-agent-attack.yaml → ATR-2026-00030-cross-agent-attack.yaml} +3 -1
- package/rules/agent-manipulation/{ATR-2026-032-goal-hijacking.yaml → ATR-2026-00032-goal-hijacking.yaml} +3 -1
- package/rules/agent-manipulation/{ATR-2026-074-cross-agent-privilege-escalation.yaml → ATR-2026-00074-cross-agent-privilege-escalation.yaml} +3 -1
- package/rules/agent-manipulation/{ATR-2026-076-inter-agent-message-spoofing.yaml → ATR-2026-00076-inter-agent-message-spoofing.yaml} +3 -1
- package/rules/agent-manipulation/{ATR-2026-077-human-trust-exploitation.yaml → ATR-2026-00077-human-trust-exploitation.yaml} +3 -1
- package/rules/agent-manipulation/{ATR-2026-108-consensus-sybil-attack.yaml → ATR-2026-00108-consensus-sybil-attack.yaml} +3 -1
- package/rules/agent-manipulation/{ATR-2026-116-a2a-message-validation.yaml → ATR-2026-00116-a2a-message-validation.yaml} +4 -2
- package/rules/agent-manipulation/{ATR-2026-117-agent-identity-spoofing.yaml → ATR-2026-00117-agent-identity-spoofing.yaml} +4 -2
- package/rules/agent-manipulation/{ATR-2026-118-approval-fatigue.yaml → ATR-2026-00118-approval-fatigue.yaml} +3 -1
- package/rules/agent-manipulation/{ATR-2026-119-social-engineering-via-agent.yaml → ATR-2026-00119-social-engineering-via-agent.yaml} +3 -1
- package/rules/agent-manipulation/ATR-2026-00132-casual-authority-escalation.yaml +105 -0
- package/rules/agent-manipulation/ATR-2026-00139-casual-authority-redirect.yaml +53 -0
- package/rules/context-exfiltration/{ATR-2026-020-system-prompt-leak.yaml → ATR-2026-00020-system-prompt-leak.yaml} +3 -1
- package/rules/context-exfiltration/{ATR-2026-021-api-key-exposure.yaml → ATR-2026-00021-api-key-exposure.yaml} +3 -1
- package/rules/context-exfiltration/{ATR-2026-075-agent-memory-manipulation.yaml → ATR-2026-00075-agent-memory-manipulation.yaml} +3 -1
- package/rules/context-exfiltration/{ATR-2026-102-disguised-analytics-exfiltration.yaml → ATR-2026-00102-disguised-analytics-exfiltration.yaml} +3 -1
- package/rules/context-exfiltration/{ATR-2026-113-credential-theft.yaml → ATR-2026-00113-credential-theft.yaml} +3 -1
- package/rules/context-exfiltration/{ATR-2026-114-oauth-token-abuse.yaml → ATR-2026-00114-oauth-token-abuse.yaml} +3 -1
- package/rules/context-exfiltration/{ATR-2026-115-env-var-harvesting.yaml → ATR-2026-00115-env-var-harvesting.yaml} +3 -1
- package/rules/context-exfiltration/ATR-2026-00136-tool-response-data-piggyback.yaml +100 -0
- package/rules/context-exfiltration/ATR-2026-00141-example-format-key-leak.yaml +52 -0
- package/rules/context-exfiltration/ATR-2026-00142-piggyback-transition-words.yaml +55 -0
- package/rules/context-exfiltration/ATR-2026-00145-obfuscated-key-disclosure.yaml +49 -0
- package/rules/context-exfiltration/ATR-2026-00146-env-var-existence-probe.yaml +49 -0
- package/rules/data-poisoning/{ATR-2026-070-data-poisoning.yaml → ATR-2026-00070-data-poisoning.yaml} +3 -1
- package/rules/excessive-autonomy/{ATR-2026-050-runaway-agent-loop.yaml → ATR-2026-00050-runaway-agent-loop.yaml} +3 -1
- package/rules/excessive-autonomy/{ATR-2026-051-resource-exhaustion.yaml → ATR-2026-00051-resource-exhaustion.yaml} +3 -1
- package/rules/excessive-autonomy/{ATR-2026-052-cascading-failure.yaml → ATR-2026-00052-cascading-failure.yaml} +3 -1
- package/rules/excessive-autonomy/{ATR-2026-098-unauthorized-financial-action.yaml → ATR-2026-00098-unauthorized-financial-action.yaml} +3 -1
- package/rules/excessive-autonomy/{ATR-2026-099-high-risk-tool-gate.yaml → ATR-2026-00099-high-risk-tool-gate.yaml} +3 -1
- package/rules/model-security/{ATR-2026-072-model-behavior-extraction.yaml → ATR-2026-00072-model-behavior-extraction.yaml} +3 -1
- package/rules/model-security/{ATR-2026-073-malicious-finetuning-data.yaml → ATR-2026-00073-malicious-finetuning-data.yaml} +3 -1
- package/rules/privilege-escalation/{ATR-2026-040-privilege-escalation.yaml → ATR-2026-00040-privilege-escalation.yaml} +3 -1
- package/rules/privilege-escalation/{ATR-2026-041-scope-creep.yaml → ATR-2026-00041-scope-creep.yaml} +3 -1
- package/rules/privilege-escalation/{ATR-2026-107-delayed-execution-bypass.yaml → ATR-2026-00107-delayed-execution-bypass.yaml} +3 -1
- package/rules/privilege-escalation/{ATR-2026-110-eval-injection.yaml → ATR-2026-00110-eval-injection.yaml} +3 -1
- package/rules/privilege-escalation/{ATR-2026-111-shell-escape.yaml → ATR-2026-00111-shell-escape.yaml} +5 -3
- package/rules/privilege-escalation/{ATR-2026-112-dynamic-import-exploitation.yaml → ATR-2026-00112-dynamic-import-exploitation.yaml} +3 -1
- package/rules/privilege-escalation/ATR-2026-00143-casual-privilege-escalation.yaml +53 -0
- package/rules/privilege-escalation/ATR-2026-00144-rationalized-safety-bypass.yaml +49 -0
- package/rules/prompt-injection/{ATR-2026-001-direct-prompt-injection.yaml → ATR-2026-00001-direct-prompt-injection.yaml} +3 -1
- package/rules/prompt-injection/{ATR-2026-002-indirect-prompt-injection.yaml → ATR-2026-00002-indirect-prompt-injection.yaml} +3 -1
- package/rules/prompt-injection/{ATR-2026-003-jailbreak-attempt.yaml → ATR-2026-00003-jailbreak-attempt.yaml} +3 -1
- package/rules/prompt-injection/{ATR-2026-004-system-prompt-override.yaml → ATR-2026-00004-system-prompt-override.yaml} +3 -1
- package/rules/prompt-injection/{ATR-2026-005-multi-turn-injection.yaml → ATR-2026-00005-multi-turn-injection.yaml} +3 -1
- package/rules/prompt-injection/{ATR-2026-080-encoding-evasion.yaml → ATR-2026-00080-encoding-evasion.yaml} +3 -1
- package/rules/prompt-injection/{ATR-2026-081-semantic-multi-turn.yaml → ATR-2026-00081-semantic-multi-turn.yaml} +3 -1
- package/rules/prompt-injection/{ATR-2026-082-fingerprint-evasion.yaml → ATR-2026-00082-fingerprint-evasion.yaml} +3 -1
- package/rules/prompt-injection/{ATR-2026-083-indirect-tool-injection.yaml → ATR-2026-00083-indirect-tool-injection.yaml} +3 -1
- package/rules/prompt-injection/{ATR-2026-084-structured-data-injection.yaml → ATR-2026-00084-structured-data-injection.yaml} +3 -1
- package/rules/prompt-injection/{ATR-2026-085-audit-evasion.yaml → ATR-2026-00085-audit-evasion.yaml} +3 -1
- package/rules/prompt-injection/{ATR-2026-086-visual-spoofing.yaml → ATR-2026-00086-visual-spoofing.yaml} +3 -1
- package/rules/prompt-injection/{ATR-2026-087-rule-probing.yaml → ATR-2026-00087-rule-probing.yaml} +3 -1
- package/rules/prompt-injection/{ATR-2026-088-adaptive-countermeasure.yaml → ATR-2026-00088-adaptive-countermeasure.yaml} +3 -1
- package/rules/prompt-injection/{ATR-2026-089-polymorphic-skill.yaml → ATR-2026-00089-polymorphic-skill.yaml} +3 -1
- package/rules/prompt-injection/{ATR-2026-090-threat-intel-exfil.yaml → ATR-2026-00090-threat-intel-exfil.yaml} +3 -1
- package/rules/prompt-injection/{ATR-2026-091-nested-payload.yaml → ATR-2026-00091-nested-payload.yaml} +3 -1
- package/rules/prompt-injection/{ATR-2026-092-consensus-poisoning.yaml → ATR-2026-00092-consensus-poisoning.yaml} +3 -1
- package/rules/prompt-injection/{ATR-2026-093-gradual-escalation.yaml → ATR-2026-00093-gradual-escalation.yaml} +3 -1
- package/rules/prompt-injection/{ATR-2026-094-audit-bypass.yaml → ATR-2026-00094-audit-bypass.yaml} +3 -1
- package/rules/prompt-injection/{ATR-2026-097-cjk-injection-patterns.yaml → ATR-2026-00097-cjk-injection-patterns.yaml} +3 -1
- package/rules/prompt-injection/{ATR-2026-104-persona-hijacking.yaml → ATR-2026-00104-persona-hijacking.yaml} +3 -1
- package/rules/prompt-injection/ATR-2026-00130-indirect-authority-claim.yaml +103 -0
- package/rules/prompt-injection/ATR-2026-00131-fictional-academic-framing.yaml +99 -0
- package/rules/prompt-injection/ATR-2026-00133-paraphrase-injection.yaml +117 -0
- package/rules/prompt-injection/ATR-2026-00137-authority-claim-injection.yaml +52 -0
- package/rules/prompt-injection/ATR-2026-00138-fictional-framing-bypass.yaml +51 -0
- package/rules/prompt-injection/ATR-2026-00140-indirect-reference-reversal.yaml +52 -0
- package/rules/prompt-injection/ATR-2026-00148-language-switch-injection.yaml +71 -0
- package/rules/skill-compromise/{ATR-2026-060-skill-impersonation.yaml → ATR-2026-00060-skill-impersonation.yaml} +3 -1
- package/rules/skill-compromise/{ATR-2026-061-description-behavior-mismatch.yaml → ATR-2026-00061-description-behavior-mismatch.yaml} +3 -1
- package/rules/skill-compromise/{ATR-2026-062-hidden-capability.yaml → ATR-2026-00062-hidden-capability.yaml} +3 -1
- package/rules/skill-compromise/{ATR-2026-063-skill-chain-attack.yaml → ATR-2026-00063-skill-chain-attack.yaml} +3 -1
- package/rules/skill-compromise/{ATR-2026-064-over-permissioned-skill.yaml → ATR-2026-00064-over-permissioned-skill.yaml} +3 -1
- package/rules/skill-compromise/{ATR-2026-065-skill-update-attack.yaml → ATR-2026-00065-skill-update-attack.yaml} +3 -1
- package/rules/skill-compromise/{ATR-2026-066-parameter-injection.yaml → ATR-2026-00066-parameter-injection.yaml} +3 -1
- package/rules/skill-compromise/ATR-2026-00120-skill-instruction-injection.yaml +121 -0
- package/rules/skill-compromise/ATR-2026-00121-skill-dangerous-script.yaml +165 -0
- package/rules/skill-compromise/ATR-2026-00122-skill-weaponized-instruction.yaml +114 -0
- package/rules/skill-compromise/ATR-2026-00123-skill-overreach-permissions.yaml +118 -0
- package/rules/skill-compromise/ATR-2026-00124-skill-name-squatting.yaml +98 -0
- package/rules/skill-compromise/ATR-2026-00125-context-poisoning-compaction.yaml +93 -0
- package/rules/skill-compromise/ATR-2026-00126-skill-rug-pull-setup.yaml +99 -0
- package/rules/skill-compromise/ATR-2026-00127-subcommand-overflow.yaml +74 -0
- package/rules/skill-compromise/ATR-2026-00128-html-comment-hidden-payload.yaml +79 -0
- package/rules/skill-compromise/ATR-2026-00129-unicode-smuggling.yaml +73 -0
- package/rules/skill-compromise/ATR-2026-00134-fork-claim-impersonation.yaml +93 -0
- package/rules/skill-compromise/ATR-2026-00135-exfil-url-in-instructions.yaml +82 -0
- package/rules/skill-compromise/ATR-2026-00147-fork-impersonation.yaml +48 -0
- package/rules/tool-poisoning/{ATR-2026-010-mcp-malicious-response.yaml → ATR-2026-00010-mcp-malicious-response.yaml} +3 -1
- package/rules/tool-poisoning/{ATR-2026-011-tool-output-injection.yaml → ATR-2026-00011-tool-output-injection.yaml} +3 -1
- package/rules/tool-poisoning/{ATR-2026-012-unauthorized-tool-call.yaml → ATR-2026-00012-unauthorized-tool-call.yaml} +3 -1
- package/rules/tool-poisoning/{ATR-2026-013-tool-ssrf.yaml → ATR-2026-00013-tool-ssrf.yaml} +3 -1
- package/rules/tool-poisoning/{ATR-2026-095-supply-chain-poisoning.yaml → ATR-2026-00095-supply-chain-poisoning.yaml} +3 -1
- package/rules/tool-poisoning/{ATR-2026-096-registry-poisoning.yaml → ATR-2026-00096-registry-poisoning.yaml} +3 -1
- package/rules/tool-poisoning/{ATR-2026-100-consent-bypass-instruction.yaml → ATR-2026-00100-consent-bypass-instruction.yaml} +3 -1
- package/rules/tool-poisoning/{ATR-2026-101-trust-escalation-override.yaml → ATR-2026-00101-trust-escalation-override.yaml} +3 -1
- package/rules/tool-poisoning/{ATR-2026-103-hidden-safety-bypass-instruction.yaml → ATR-2026-00103-hidden-safety-bypass-instruction.yaml} +3 -1
- package/rules/tool-poisoning/{ATR-2026-105-silent-action-concealment.yaml → ATR-2026-00105-silent-action-concealment.yaml} +3 -1
- package/rules/tool-poisoning/{ATR-2026-106-schema-description-contradiction.yaml → ATR-2026-00106-schema-description-contradiction.yaml} +3 -1
- package/spec/atr-schema.yaml +32 -3
- package/dist/action-executor.d.ts +0 -44
- package/dist/action-executor.d.ts.map +0 -1
- package/dist/action-executor.js +0 -130
- package/dist/action-executor.js.map +0 -1
- package/dist/adapters/default-adapter.d.ts +0 -24
- package/dist/adapters/default-adapter.d.ts.map +0 -1
- package/dist/adapters/default-adapter.js +0 -51
- package/dist/adapters/default-adapter.js.map +0 -1
- package/dist/adapters/stdio-adapter.d.ts +0 -30
- package/dist/adapters/stdio-adapter.d.ts.map +0 -1
- package/dist/adapters/stdio-adapter.js +0 -128
- package/dist/adapters/stdio-adapter.js.map +0 -1
- package/dist/badge.d.ts +0 -42
- package/dist/badge.d.ts.map +0 -1
- package/dist/badge.js +0 -158
- package/dist/badge.js.map +0 -1
- package/dist/capability-extractor.d.ts +0 -35
- package/dist/capability-extractor.d.ts.map +0 -1
- package/dist/capability-extractor.js +0 -91
- package/dist/capability-extractor.js.map +0 -1
- package/dist/cli.d.ts +0 -12
- package/dist/cli.d.ts.map +0 -1
- package/dist/cli.js +0 -892
- package/dist/cli.js.map +0 -1
- package/dist/converters/elastic.d.ts +0 -36
- package/dist/converters/elastic.d.ts.map +0 -1
- package/dist/converters/elastic.js +0 -125
- package/dist/converters/elastic.js.map +0 -1
- package/dist/converters/index.d.ts +0 -28
- package/dist/converters/index.d.ts.map +0 -1
- package/dist/converters/index.js +0 -36
- package/dist/converters/index.js.map +0 -1
- package/dist/converters/splunk.d.ts +0 -19
- package/dist/converters/splunk.d.ts.map +0 -1
- package/dist/converters/splunk.js +0 -148
- package/dist/converters/splunk.js.map +0 -1
- package/dist/coverage-analyzer.d.ts +0 -43
- package/dist/coverage-analyzer.d.ts.map +0 -1
- package/dist/coverage-analyzer.js +0 -329
- package/dist/coverage-analyzer.js.map +0 -1
- package/dist/embedding/build-corpus.d.ts +0 -15
- package/dist/embedding/build-corpus.d.ts.map +0 -1
- package/dist/embedding/build-corpus.js +0 -105
- package/dist/embedding/build-corpus.js.map +0 -1
- package/dist/embedding/model-loader.d.ts +0 -41
- package/dist/embedding/model-loader.d.ts.map +0 -1
- package/dist/embedding/model-loader.js +0 -90
- package/dist/embedding/model-loader.js.map +0 -1
- package/dist/embedding/vector-store.d.ts +0 -41
- package/dist/embedding/vector-store.d.ts.map +0 -1
- package/dist/embedding/vector-store.js +0 -70
- package/dist/embedding/vector-store.js.map +0 -1
- package/dist/engine.d.ts +0 -163
- package/dist/engine.d.ts.map +0 -1
- package/dist/engine.js +0 -869
- package/dist/engine.js.map +0 -1
- package/dist/eval/corpus.d.ts +0 -42
- package/dist/eval/corpus.d.ts.map +0 -1
- package/dist/eval/corpus.js +0 -427
- package/dist/eval/corpus.js.map +0 -1
- package/dist/eval/eval-harness.d.ts +0 -44
- package/dist/eval/eval-harness.d.ts.map +0 -1
- package/dist/eval/eval-harness.js +0 -296
- package/dist/eval/eval-harness.js.map +0 -1
- package/dist/eval/index.d.ts +0 -13
- package/dist/eval/index.d.ts.map +0 -1
- package/dist/eval/index.js +0 -9
- package/dist/eval/index.js.map +0 -1
- package/dist/eval/metrics.d.ts +0 -74
- package/dist/eval/metrics.d.ts.map +0 -1
- package/dist/eval/metrics.js +0 -108
- package/dist/eval/metrics.js.map +0 -1
- package/dist/eval/pint-corpus.d.ts +0 -34
- package/dist/eval/pint-corpus.d.ts.map +0 -1
- package/dist/eval/pint-corpus.js +0 -109
- package/dist/eval/pint-corpus.js.map +0 -1
- package/dist/eval/rule-corpus.d.ts +0 -9
- package/dist/eval/rule-corpus.d.ts.map +0 -1
- package/dist/eval/rule-corpus.js +0 -4780
- package/dist/eval/rule-corpus.js.map +0 -1
- package/dist/eval/rule-metrics.d.ts +0 -34
- package/dist/eval/rule-metrics.d.ts.map +0 -1
- package/dist/eval/rule-metrics.js +0 -92
- package/dist/eval/rule-metrics.js.map +0 -1
- package/dist/eval/run-eval.d.ts +0 -7
- package/dist/eval/run-eval.d.ts.map +0 -1
- package/dist/eval/run-eval.js +0 -11
- package/dist/eval/run-eval.js.map +0 -1
- package/dist/eval/run-pint-benchmark.d.ts +0 -18
- package/dist/eval/run-pint-benchmark.d.ts.map +0 -1
- package/dist/eval/run-pint-benchmark.js +0 -159
- package/dist/eval/run-pint-benchmark.js.map +0 -1
- package/dist/flywheel.d.ts +0 -54
- package/dist/flywheel.d.ts.map +0 -1
- package/dist/flywheel.js +0 -121
- package/dist/flywheel.js.map +0 -1
- package/dist/hook-handler.d.ts +0 -61
- package/dist/hook-handler.d.ts.map +0 -1
- package/dist/hook-handler.js +0 -178
- package/dist/hook-handler.js.map +0 -1
- package/dist/index.d.ts +0 -62
- package/dist/index.d.ts.map +0 -1
- package/dist/index.js +0 -54
- package/dist/index.js.map +0 -1
- package/dist/layer-integration.d.ts +0 -55
- package/dist/layer-integration.d.ts.map +0 -1
- package/dist/layer-integration.js +0 -185
- package/dist/layer-integration.js.map +0 -1
- package/dist/loader.d.ts +0 -21
- package/dist/loader.d.ts.map +0 -1
- package/dist/loader.js +0 -124
- package/dist/loader.js.map +0 -1
- package/dist/mcp-server.d.ts +0 -13
- package/dist/mcp-server.d.ts.map +0 -1
- package/dist/mcp-server.js +0 -220
- package/dist/mcp-server.js.map +0 -1
- package/dist/mcp-tools/coverage-gaps.d.ts +0 -13
- package/dist/mcp-tools/coverage-gaps.d.ts.map +0 -1
- package/dist/mcp-tools/coverage-gaps.js +0 -55
- package/dist/mcp-tools/coverage-gaps.js.map +0 -1
- package/dist/mcp-tools/list-rules.d.ts +0 -17
- package/dist/mcp-tools/list-rules.d.ts.map +0 -1
- package/dist/mcp-tools/list-rules.js +0 -45
- package/dist/mcp-tools/list-rules.js.map +0 -1
- package/dist/mcp-tools/scan.d.ts +0 -24
- package/dist/mcp-tools/scan.d.ts.map +0 -1
- package/dist/mcp-tools/scan.js +0 -94
- package/dist/mcp-tools/scan.js.map +0 -1
- package/dist/mcp-tools/submit-proposal.d.ts +0 -12
- package/dist/mcp-tools/submit-proposal.d.ts.map +0 -1
- package/dist/mcp-tools/submit-proposal.js +0 -103
- package/dist/mcp-tools/submit-proposal.js.map +0 -1
- package/dist/mcp-tools/threat-summary.d.ts +0 -12
- package/dist/mcp-tools/threat-summary.d.ts.map +0 -1
- package/dist/mcp-tools/threat-summary.js +0 -74
- package/dist/mcp-tools/threat-summary.js.map +0 -1
- package/dist/mcp-tools/validate.d.ts +0 -15
- package/dist/mcp-tools/validate.d.ts.map +0 -1
- package/dist/mcp-tools/validate.js +0 -45
- package/dist/mcp-tools/validate.js.map +0 -1
- package/dist/modules/embedding.d.ts +0 -71
- package/dist/modules/embedding.d.ts.map +0 -1
- package/dist/modules/embedding.js +0 -141
- package/dist/modules/embedding.js.map +0 -1
- package/dist/modules/index.d.ts +0 -144
- package/dist/modules/index.d.ts.map +0 -1
- package/dist/modules/index.js +0 -82
- package/dist/modules/index.js.map +0 -1
- package/dist/modules/semantic.d.ts +0 -106
- package/dist/modules/semantic.d.ts.map +0 -1
- package/dist/modules/semantic.js +0 -359
- package/dist/modules/semantic.js.map +0 -1
- package/dist/modules/session.d.ts +0 -70
- package/dist/modules/session.d.ts.map +0 -1
- package/dist/modules/session.js +0 -128
- package/dist/modules/session.js.map +0 -1
- package/dist/rule-scaffolder.d.ts +0 -53
- package/dist/rule-scaffolder.d.ts.map +0 -1
- package/dist/rule-scaffolder.js +0 -301
- package/dist/rule-scaffolder.js.map +0 -1
- package/dist/session-tracker.d.ts +0 -58
- package/dist/session-tracker.d.ts.map +0 -1
- package/dist/session-tracker.js +0 -176
- package/dist/session-tracker.js.map +0 -1
- package/dist/shadow-evaluator.d.ts +0 -48
- package/dist/shadow-evaluator.d.ts.map +0 -1
- package/dist/shadow-evaluator.js +0 -128
- package/dist/shadow-evaluator.js.map +0 -1
- package/dist/skill-fingerprint.d.ts +0 -85
- package/dist/skill-fingerprint.d.ts.map +0 -1
- package/dist/skill-fingerprint.js +0 -284
- package/dist/skill-fingerprint.js.map +0 -1
- package/dist/tier0-invariant.d.ts +0 -49
- package/dist/tier0-invariant.d.ts.map +0 -1
- package/dist/tier0-invariant.js +0 -184
- package/dist/tier0-invariant.js.map +0 -1
- package/dist/tier1-blacklist.d.ts +0 -48
- package/dist/tier1-blacklist.d.ts.map +0 -1
- package/dist/tier1-blacklist.js +0 -91
- package/dist/tier1-blacklist.js.map +0 -1
- package/dist/types.d.ts +0 -190
- package/dist/types.d.ts.map +0 -1
- package/dist/types.js +0 -6
- package/dist/types.js.map +0 -1
- package/dist/verdict.d.ts +0 -26
- package/dist/verdict.d.ts.map +0 -1
- package/dist/verdict.js +0 -127
- package/dist/verdict.js.map +0 -1
package/dist/rule-scaffolder.js
DELETED
|
@@ -1,301 +0,0 @@
|
|
|
1
|
-
/**
|
|
2
|
-
* ATR Rule Scaffolder - Generates ATR rule YAML scaffolds from structured input
|
|
3
|
-
* @module agent-threat-rules/rule-scaffolder
|
|
4
|
-
*/
|
|
5
|
-
import yaml from 'js-yaml';
|
|
6
|
-
const CATEGORY_TO_SOURCE_TYPE = {
|
|
7
|
-
'prompt-injection': 'llm_io',
|
|
8
|
-
'tool-poisoning': 'tool_call',
|
|
9
|
-
'context-exfiltration': 'context_window',
|
|
10
|
-
'agent-manipulation': 'multi_agent_comm',
|
|
11
|
-
'privilege-escalation': 'agent_behavior',
|
|
12
|
-
'excessive-autonomy': 'agent_behavior',
|
|
13
|
-
'data-poisoning': 'llm_io',
|
|
14
|
-
'model-abuse': 'llm_io',
|
|
15
|
-
'skill-compromise': 'skill_lifecycle',
|
|
16
|
-
};
|
|
17
|
-
const CATEGORY_TO_FIELD = {
|
|
18
|
-
'prompt-injection': 'user_input',
|
|
19
|
-
'tool-poisoning': 'tool_response',
|
|
20
|
-
'context-exfiltration': 'agent_output',
|
|
21
|
-
'agent-manipulation': 'agent_message',
|
|
22
|
-
'privilege-escalation': 'agent_action',
|
|
23
|
-
'excessive-autonomy': 'agent_action',
|
|
24
|
-
'data-poisoning': 'training_input',
|
|
25
|
-
'model-abuse': 'user_input',
|
|
26
|
-
'skill-compromise': 'skill_manifest',
|
|
27
|
-
};
|
|
28
|
-
const SEVERITY_TO_ACTIONS = {
|
|
29
|
-
critical: ['block_input', 'alert', 'escalate'],
|
|
30
|
-
high: ['block_input', 'alert'],
|
|
31
|
-
medium: ['alert', 'snapshot'],
|
|
32
|
-
low: ['alert'],
|
|
33
|
-
informational: ['alert'],
|
|
34
|
-
};
|
|
35
|
-
const REGEX_SPECIAL_CHARS = /[.*+?^${}()|[\]\\]/g;
|
|
36
|
-
function escapeRegex(str) {
|
|
37
|
-
return str.replace(REGEX_SPECIAL_CHARS, '\\$&');
|
|
38
|
-
}
|
|
39
|
-
/**
|
|
40
|
-
* Attack pattern templates by category — reusable regex building blocks
|
|
41
|
-
* that detect BEHAVIOR, not package names.
|
|
42
|
-
*/
|
|
43
|
-
export const ATTACK_PATTERN_INDICATORS = [
|
|
44
|
-
// Shell execution patterns
|
|
45
|
-
{
|
|
46
|
-
test: /exec(Sync)?|spawn|child_process|shell|subprocess|popen|os\.system/i,
|
|
47
|
-
pattern: '(execSync?|spawn|child_process|shell|subprocess|popen|os\\.system)\\s*\\(',
|
|
48
|
-
description: 'Shell/command execution',
|
|
49
|
-
categories: ['tool-poisoning', 'skill-compromise', 'privilege-escalation'],
|
|
50
|
-
},
|
|
51
|
-
// Dynamic shell with interpolation (RCE)
|
|
52
|
-
{
|
|
53
|
-
test: /exec.*\$\{|spawn.*\$\{|`.*\$\{.*`/i,
|
|
54
|
-
pattern: '(exec|spawn|shell)\\s*\\(.*\\$\\{',
|
|
55
|
-
description: 'Dynamic shell execution with variable interpolation',
|
|
56
|
-
categories: ['tool-poisoning', 'skill-compromise'],
|
|
57
|
-
},
|
|
58
|
-
// Network exfiltration
|
|
59
|
-
{
|
|
60
|
-
test: /fetch|http|request|axios|got|node-fetch|urllib|curl|wget/i,
|
|
61
|
-
pattern: '(fetch|https?://|request|axios|got|node-fetch|urllib|curl|wget)\\s*\\(?',
|
|
62
|
-
description: 'Outbound network request',
|
|
63
|
-
categories: ['context-exfiltration', 'tool-poisoning', 'data-poisoning'],
|
|
64
|
-
},
|
|
65
|
-
// Credential/secret access
|
|
66
|
-
{
|
|
67
|
-
test: /password|secret|token|credential|api[_\s]?key|auth|cookie/i,
|
|
68
|
-
pattern: '(password|secret|token|credential|api[_ ]?key|auth_token|cookie)',
|
|
69
|
-
description: 'Credential/secret access',
|
|
70
|
-
categories: ['context-exfiltration', 'privilege-escalation', 'tool-poisoning'],
|
|
71
|
-
},
|
|
72
|
-
// Environment variable exfiltration
|
|
73
|
-
{
|
|
74
|
-
test: /process\.env|os\.environ|getenv|ENV\[/i,
|
|
75
|
-
pattern: '(process\\.env|os\\.environ|getenv|ENV\\[)',
|
|
76
|
-
description: 'Environment variable access',
|
|
77
|
-
categories: ['context-exfiltration', 'tool-poisoning', 'skill-compromise'],
|
|
78
|
-
},
|
|
79
|
-
// eval / dynamic code execution
|
|
80
|
-
{
|
|
81
|
-
test: /\beval\s*\(|new\s+Function\s*\(|vm\.run/i,
|
|
82
|
-
pattern: '(\\beval\\s*\\(|new\\s+Function\\s*\\(|vm\\.run)',
|
|
83
|
-
description: 'Dynamic code execution',
|
|
84
|
-
categories: ['tool-poisoning', 'skill-compromise'],
|
|
85
|
-
},
|
|
86
|
-
// Instruction override (prompt injection)
|
|
87
|
-
{
|
|
88
|
-
test: /ignore|disregard|forget|override|overwrite/i,
|
|
89
|
-
pattern: '(override|overwrite|ignore|disregard|forget)\\s+(previous|prior|above|existing|all|any)\\s+(instructions?|rules?|constraints?|guidelines?|protocols?)',
|
|
90
|
-
description: 'Instruction override attempt',
|
|
91
|
-
categories: ['prompt-injection', 'agent-manipulation'],
|
|
92
|
-
},
|
|
93
|
-
// Role manipulation
|
|
94
|
-
{
|
|
95
|
-
test: /you are now|act as|pretend|new role|system prompt/i,
|
|
96
|
-
pattern: '(you\\s+are\\s+now|act\\s+as\\s+(a|an|if)|pretend\\s+(to|you)|new\\s+role|system\\s+prompt)',
|
|
97
|
-
description: 'Role/identity manipulation',
|
|
98
|
-
categories: ['prompt-injection', 'agent-manipulation'],
|
|
99
|
-
},
|
|
100
|
-
// File system destructive operations
|
|
101
|
-
{
|
|
102
|
-
test: /rm\s+-rf|rmdir|unlink|deleteFile|fs\.rm/i,
|
|
103
|
-
pattern: '(rm\\s+-rf|rmdir\\s|unlink\\s*\\(|deleteFile|fs\\.rm)',
|
|
104
|
-
description: 'Destructive file system operation',
|
|
105
|
-
categories: ['tool-poisoning', 'excessive-autonomy'],
|
|
106
|
-
},
|
|
107
|
-
// Base64/encoding evasion
|
|
108
|
-
{
|
|
109
|
-
test: /atob|btoa|base64|Buffer\.from.*encoding|fromCharCode/i,
|
|
110
|
-
pattern: '(atob|btoa|base64|Buffer\\.from|fromCharCode)\\s*\\(',
|
|
111
|
-
description: 'Encoding-based payload obfuscation',
|
|
112
|
-
categories: ['tool-poisoning', 'skill-compromise'],
|
|
113
|
-
},
|
|
114
|
-
// Data exfiltration combo (credential + network)
|
|
115
|
-
{
|
|
116
|
-
test: /(password|secret|token|key).*(fetch|http|send|post|upload)/i,
|
|
117
|
-
pattern: '(password|secret|token|api[_ ]?key).*(fetch|https?://|request|send|post|upload)',
|
|
118
|
-
description: 'Credential access combined with network exfiltration',
|
|
119
|
-
categories: ['context-exfiltration', 'tool-poisoning'],
|
|
120
|
-
},
|
|
121
|
-
// Download + execute combo
|
|
122
|
-
{
|
|
123
|
-
test: /(download|fetch|curl|wget).*(exec|eval|spawn)/i,
|
|
124
|
-
pattern: '(download|fetch|curl|wget).*(exec|eval|spawn|child_process)',
|
|
125
|
-
description: 'Download and execute pattern',
|
|
126
|
-
categories: ['tool-poisoning', 'skill-compromise'],
|
|
127
|
-
},
|
|
128
|
-
];
|
|
129
|
-
/**
|
|
130
|
-
* Build detection regex from a payload string, using category-aware
|
|
131
|
-
* attack pattern templates instead of naive keyword extraction.
|
|
132
|
-
*
|
|
133
|
-
* Priority:
|
|
134
|
-
* 1. Match known attack indicators in the payload -> use behavioral regex
|
|
135
|
-
* 2. Combine multiple indicators with alternation for multi-vector attacks
|
|
136
|
-
* 3. Fall back to keyword extraction only for text that has no code patterns
|
|
137
|
-
*/
|
|
138
|
-
function buildRegexPattern(payload, category) {
|
|
139
|
-
const trimmed = payload.trim();
|
|
140
|
-
// Step 1: Find all attack indicators present in this payload
|
|
141
|
-
const matched = ATTACK_PATTERN_INDICATORS.filter((ind) => {
|
|
142
|
-
const matchesPayload = ind.test.test(trimmed);
|
|
143
|
-
const matchesCategory = !category || ind.categories.includes(category);
|
|
144
|
-
return matchesPayload && matchesCategory;
|
|
145
|
-
});
|
|
146
|
-
// Step 2: If we found behavioral patterns, use them
|
|
147
|
-
if (matched.length > 0) {
|
|
148
|
-
if (matched.length === 1) {
|
|
149
|
-
return `(?i)${matched[0].pattern}`;
|
|
150
|
-
}
|
|
151
|
-
// Combine multiple indicators — detect the most specific one
|
|
152
|
-
// Sort by pattern length (longer = more specific) and take top 3
|
|
153
|
-
const sorted = [...matched].sort((a, b) => b.pattern.length - a.pattern.length);
|
|
154
|
-
const top = sorted.slice(0, 3);
|
|
155
|
-
if (top.length === 1) {
|
|
156
|
-
return `(?i)${top[0].pattern}`;
|
|
157
|
-
}
|
|
158
|
-
// Use alternation for multi-vector detection
|
|
159
|
-
return `(?i)(${top.map((t) => t.pattern).join('|')})`;
|
|
160
|
-
}
|
|
161
|
-
// Step 3: Fallback — keyword extraction for text-based payloads
|
|
162
|
-
// (e.g., prompt injection text without code patterns)
|
|
163
|
-
const words = trimmed.split(/\s+/).filter((w) => w.length > 3);
|
|
164
|
-
if (words.length === 0) {
|
|
165
|
-
return `(?i).*${escapeRegex(trimmed)}.*`;
|
|
166
|
-
}
|
|
167
|
-
const keywords = words.slice(0, 4);
|
|
168
|
-
return `(?i)${keywords.map((k) => `(?=.*${escapeRegex(k)})`).join('')}`;
|
|
169
|
-
}
|
|
170
|
-
function generateId(existingIds = new Set()) {
|
|
171
|
-
const year = new Date().getFullYear();
|
|
172
|
-
const maxAttempts = 800;
|
|
173
|
-
for (let attempt = 0; attempt < maxAttempts; attempt++) {
|
|
174
|
-
const seq = String(Math.floor(Math.random() * 900) + 100);
|
|
175
|
-
const id = `ATR-${year}-${seq}`;
|
|
176
|
-
if (!existingIds.has(id)) {
|
|
177
|
-
return id;
|
|
178
|
-
}
|
|
179
|
-
}
|
|
180
|
-
throw new Error('Unable to generate a unique ATR rule ID after maximum attempts');
|
|
181
|
-
}
|
|
182
|
-
function getCurrentDate() {
|
|
183
|
-
const d = new Date();
|
|
184
|
-
const yyyy = d.getFullYear();
|
|
185
|
-
const mm = String(d.getMonth() + 1).padStart(2, '0');
|
|
186
|
-
const dd = String(d.getDate()).padStart(2, '0');
|
|
187
|
-
return `${yyyy}/${mm}/${dd}`;
|
|
188
|
-
}
|
|
189
|
-
export class RuleScaffolder {
|
|
190
|
-
options;
|
|
191
|
-
constructor(options = {}) {
|
|
192
|
-
this.options = {
|
|
193
|
-
author: options.author ?? 'ATR Community (auto-scaffolded)',
|
|
194
|
-
schemaVersion: options.schemaVersion ?? '0.1',
|
|
195
|
-
};
|
|
196
|
-
}
|
|
197
|
-
/**
|
|
198
|
-
* Generate a complete ATR YAML rule from structured input.
|
|
199
|
-
* Returns a ScaffoldResult with the YAML string, generated ID, and any warnings.
|
|
200
|
-
*/
|
|
201
|
-
scaffold(input, existingIds = new Set()) {
|
|
202
|
-
const warnings = this.validateInput(input);
|
|
203
|
-
const severity = input.severity ?? 'medium';
|
|
204
|
-
const sourceType = input.agentSourceType ?? CATEGORY_TO_SOURCE_TYPE[input.category];
|
|
205
|
-
const field = CATEGORY_TO_FIELD[input.category];
|
|
206
|
-
const id = generateId(existingIds);
|
|
207
|
-
const date = getCurrentDate();
|
|
208
|
-
const conditions = input.examplePayloads.map((payload, idx) => ({
|
|
209
|
-
field,
|
|
210
|
-
operator: 'regex',
|
|
211
|
-
value: buildRegexPattern(payload, input.category),
|
|
212
|
-
description: `Pattern ${idx + 1}: detects "${payload.trim().slice(0, 80)}"`,
|
|
213
|
-
}));
|
|
214
|
-
const truePositives = input.examplePayloads.map((payload) => ({
|
|
215
|
-
input: payload.trim(),
|
|
216
|
-
expected: 'trigger',
|
|
217
|
-
}));
|
|
218
|
-
const trueNegatives = [
|
|
219
|
-
{
|
|
220
|
-
input: 'TODO: Add benign input that should not trigger this rule',
|
|
221
|
-
expected: 'no_trigger',
|
|
222
|
-
},
|
|
223
|
-
];
|
|
224
|
-
const references = {};
|
|
225
|
-
if (input.owaspRefs && input.owaspRefs.length > 0) {
|
|
226
|
-
references.owasp_llm = [...input.owaspRefs];
|
|
227
|
-
}
|
|
228
|
-
if (input.mitreRefs && input.mitreRefs.length > 0) {
|
|
229
|
-
references.mitre_atlas = [...input.mitreRefs];
|
|
230
|
-
}
|
|
231
|
-
const conditionExpr = conditions.length > 1 ? 'any' : 'all';
|
|
232
|
-
const rule = {
|
|
233
|
-
title: input.title,
|
|
234
|
-
id,
|
|
235
|
-
schema_version: this.options.schemaVersion,
|
|
236
|
-
status: 'draft',
|
|
237
|
-
description: input.attackDescription,
|
|
238
|
-
author: this.options.author,
|
|
239
|
-
date,
|
|
240
|
-
severity,
|
|
241
|
-
detection_tier: 'pattern',
|
|
242
|
-
maturity: 'draft',
|
|
243
|
-
...(Object.keys(references).length > 0 ? { references } : {}),
|
|
244
|
-
tags: {
|
|
245
|
-
category: input.category,
|
|
246
|
-
confidence: severity === 'critical' || severity === 'high' ? 'high' : 'medium',
|
|
247
|
-
},
|
|
248
|
-
agent_source: {
|
|
249
|
-
type: sourceType,
|
|
250
|
-
},
|
|
251
|
-
detection: {
|
|
252
|
-
conditions,
|
|
253
|
-
condition: conditionExpr,
|
|
254
|
-
false_positives: [
|
|
255
|
-
'TODO: Document known false positive scenarios',
|
|
256
|
-
],
|
|
257
|
-
},
|
|
258
|
-
response: {
|
|
259
|
-
actions: [...SEVERITY_TO_ACTIONS[severity]],
|
|
260
|
-
message_template: `Potential ${input.category} detected: {{matched_patterns}}`,
|
|
261
|
-
},
|
|
262
|
-
test_cases: {
|
|
263
|
-
true_positives: truePositives,
|
|
264
|
-
true_negatives: trueNegatives,
|
|
265
|
-
},
|
|
266
|
-
};
|
|
267
|
-
const yamlStr = yaml.dump(rule, {
|
|
268
|
-
indent: 2,
|
|
269
|
-
lineWidth: 120,
|
|
270
|
-
noRefs: true,
|
|
271
|
-
sortKeys: false,
|
|
272
|
-
quotingType: '"',
|
|
273
|
-
forceQuotes: false,
|
|
274
|
-
});
|
|
275
|
-
return { yaml: yamlStr, id, warnings };
|
|
276
|
-
}
|
|
277
|
-
/**
|
|
278
|
-
* Validate scaffold input, throwing on invalid required fields
|
|
279
|
-
* and returning warnings for non-critical issues.
|
|
280
|
-
*/
|
|
281
|
-
validateInput(input) {
|
|
282
|
-
const warnings = [];
|
|
283
|
-
if (!input.title || input.title.trim().length === 0) {
|
|
284
|
-
throw new Error('ScaffoldInput.title is required and must be non-empty');
|
|
285
|
-
}
|
|
286
|
-
if (!input.category) {
|
|
287
|
-
throw new Error('ScaffoldInput.category is required');
|
|
288
|
-
}
|
|
289
|
-
if (!input.attackDescription || input.attackDescription.trim().length === 0) {
|
|
290
|
-
throw new Error('ScaffoldInput.attackDescription is required and must be non-empty');
|
|
291
|
-
}
|
|
292
|
-
if (!input.examplePayloads || input.examplePayloads.length === 0) {
|
|
293
|
-
throw new Error('ScaffoldInput.examplePayloads must contain at least one payload');
|
|
294
|
-
}
|
|
295
|
-
if (input.examplePayloads.length < 3) {
|
|
296
|
-
warnings.push('Fewer than 3 example payloads - consider adding more for better pattern coverage.');
|
|
297
|
-
}
|
|
298
|
-
return warnings;
|
|
299
|
-
}
|
|
300
|
-
}
|
|
301
|
-
//# sourceMappingURL=rule-scaffolder.js.map
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"rule-scaffolder.js","sourceRoot":"","sources":["../src/rule-scaffolder.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,IAAI,MAAM,SAAS,CAAC;AA+B3B,MAAM,uBAAuB,GAAiD;IAC5E,kBAAkB,EAAE,QAAQ;IAC5B,gBAAgB,EAAE,WAAW;IAC7B,sBAAsB,EAAE,gBAAgB;IACxC,oBAAoB,EAAE,kBAAkB;IACxC,sBAAsB,EAAE,gBAAgB;IACxC,oBAAoB,EAAE,gBAAgB;IACtC,gBAAgB,EAAE,QAAQ;IAC1B,aAAa,EAAE,QAAQ;IACvB,kBAAkB,EAAE,iBAAiB;CACtC,CAAC;AAEF,MAAM,iBAAiB,GAA0C;IAC/D,kBAAkB,EAAE,YAAY;IAChC,gBAAgB,EAAE,eAAe;IACjC,sBAAsB,EAAE,cAAc;IACtC,oBAAoB,EAAE,eAAe;IACrC,sBAAsB,EAAE,cAAc;IACtC,oBAAoB,EAAE,cAAc;IACpC,gBAAgB,EAAE,gBAAgB;IAClC,aAAa,EAAE,YAAY;IAC3B,kBAAkB,EAAE,gBAAgB;CACrC,CAAC;AAEF,MAAM,mBAAmB,GAAwD;IAC/E,QAAQ,EAAE,CAAC,aAAa,EAAE,OAAO,EAAE,UAAU,CAAC;IAC9C,IAAI,EAAE,CAAC,aAAa,EAAE,OAAO,CAAC;IAC9B,MAAM,EAAE,CAAC,OAAO,EAAE,UAAU,CAAC;IAC7B,GAAG,EAAE,CAAC,OAAO,CAAC;IACd,aAAa,EAAE,CAAC,OAAO,CAAC;CACzB,CAAC;AAEF,MAAM,mBAAmB,GAAG,qBAAqB,CAAC;AAElD,SAAS,WAAW,CAAC,GAAW;IAC9B,OAAO,GAAG,CAAC,OAAO,CAAC,mBAAmB,EAAE,MAAM,CAAC,CAAC;AAClD,CAAC;AAED;;;GAGG;AACH,MAAM,CAAC,MAAM,yBAAyB,GASjC;IACH,2BAA2B;IAC3B;QACE,IAAI,EAAE,oEAAoE;QAC1E,OAAO,EAAE,2EAA2E;QACpF,WAAW,EAAE,yBAAyB;QACtC,UAAU,EAAE,CAAC,gBAAgB,EAAE,kBAAkB,EAAE,sBAAsB,CAAC;KAC3E;IACD,yCAAyC;IACzC;QACE,IAAI,EAAE,oCAAoC;QAC1C,OAAO,EAAE,mCAAmC;QAC5C,WAAW,EAAE,qDAAqD;QAClE,UAAU,EAAE,CAAC,gBAAgB,EAAE,kBAAkB,CAAC;KACnD;IACD,uBAAuB;IACvB;QACE,IAAI,EAAE,2DAA2D;QACjE,OAAO,EAAE,yEAAyE;QAClF,WAAW,EAAE,0BAA0B;QACvC,UAAU,EAAE,CAAC,sBAAsB,EAAE,gBAAgB,EAAE,gBAAgB,CAAC;KACzE;IACD,2BAA2B;IAC3B;QACE,IAAI,EAAE,4DAA4D;QAClE,OAAO,EAAE,kEAAkE;QAC3E,WAAW,EAAE,0BAA0B;QACvC,UAAU,EAAE,CAAC,sBAAsB,EAAE,sBAAsB,EAAE,gBAAgB,CAAC;KAC/E;IACD,oCAAoC;IACpC;QACE,IAAI,EAAE,wCAAwC;QAC9C,OAAO,EAAE,4CAA4C;QACrD,WAAW,EAAE,6BAA6B;QAC1C,UAAU,EAAE,CAAC,sBAAsB,EAAE,gBAAgB,EAAE,kBAAkB,CAAC;KAC3E;IACD,gCAAgC;IAChC;QACE,IAAI,EAAE,0CAA0C;QAChD,OAAO,EAAE,kDAAkD;QAC3D,WAAW,EAAE,wBAAwB;QACrC,UAAU,EAAE,CAAC,gBAAgB,EAAE,kBAAkB,CAAC;KACnD;IACD,0CAA0C;IAC1C;QACE,IAAI,EAAE,6CAA6C;QACnD,OAAO,EAAE,uJAAuJ;QAChK,WAAW,EAAE,8BAA8B;QAC3C,UAAU,EAAE,CAAC,kBAAkB,EAAE,oBAAoB,CAAC;KACvD;IACD,oBAAoB;IACpB;QACE,IAAI,EAAE,oDAAoD;QAC1D,OAAO,EAAE,6FAA6F;QACtG,WAAW,EAAE,4BAA4B;QACzC,UAAU,EAAE,CAAC,kBAAkB,EAAE,oBAAoB,CAAC;KACvD;IACD,qCAAqC;IACrC;QACE,IAAI,EAAE,0CAA0C;QAChD,OAAO,EAAE,uDAAuD;QAChE,WAAW,EAAE,mCAAmC;QAChD,UAAU,EAAE,CAAC,gBAAgB,EAAE,oBAAoB,CAAC;KACrD;IACD,0BAA0B;IAC1B;QACE,IAAI,EAAE,uDAAuD;QAC7D,OAAO,EAAE,sDAAsD;QAC/D,WAAW,EAAE,oCAAoC;QACjD,UAAU,EAAE,CAAC,gBAAgB,EAAE,kBAAkB,CAAC;KACnD;IACD,iDAAiD;IACjD;QACE,IAAI,EAAE,6DAA6D;QACnE,OAAO,EAAE,iFAAiF;QAC1F,WAAW,EAAE,sDAAsD;QACnE,UAAU,EAAE,CAAC,sBAAsB,EAAE,gBAAgB,CAAC;KACvD;IACD,2BAA2B;IAC3B;QACE,IAAI,EAAE,gDAAgD;QACtD,OAAO,EAAE,6DAA6D;QACtE,WAAW,EAAE,8BAA8B;QAC3C,UAAU,EAAE,CAAC,gBAAgB,EAAE,kBAAkB,CAAC;KACnD;CACF,CAAC;AAEF;;;;;;;;GAQG;AACH,SAAS,iBAAiB,CAAC,OAAe,EAAE,QAAsB;IAChE,MAAM,OAAO,GAAG,OAAO,CAAC,IAAI,EAAE,CAAC;IAE/B,6DAA6D;IAC7D,MAAM,OAAO,GAAG,yBAAyB,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,EAAE;QACvD,MAAM,cAAc,GAAG,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAC9C,MAAM,eAAe,GAAG,CAAC,QAAQ,IAAI,GAAG,CAAC,UAAU,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;QACvE,OAAO,cAAc,IAAI,eAAe,CAAC;IAC3C,CAAC,CAAC,CAAC;IAEH,oDAAoD;IACpD,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACvB,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACzB,OAAO,OAAO,OAAO,CAAC,CAAC,CAAE,CAAC,OAAO,EAAE,CAAC;QACtC,CAAC;QACD,6DAA6D;QAC7D,iEAAiE;QACjE,MAAM,MAAM,GAAG,CAAC,GAAG,OAAO,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;QAChF,MAAM,GAAG,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;QAC/B,IAAI,GAAG,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACrB,OAAO,OAAO,GAAG,CAAC,CAAC,CAAE,CAAC,OAAO,EAAE,CAAC;QAClC,CAAC;QACD,6CAA6C;QAC7C,OAAO,QAAQ,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC;IACxD,CAAC;IAED,gEAAgE;IAChE,sDAAsD;IACtD,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;IAE/D,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACvB,OAAO,SAAS,WAAW,CAAC,OAAO,CAAC,IAAI,CAAC;IAC3C,CAAC;IAED,MAAM,QAAQ,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;IACnC,OAAO,OAAO,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,QAAQ,WAAW,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC;AAC1E,CAAC;AAED,SAAS,UAAU,CAAC,cAAmC,IAAI,GAAG,EAAE;IAC9D,MAAM,IAAI,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;IACtC,MAAM,WAAW,GAAG,GAAG,CAAC;IACxB,KAAK,IAAI,OAAO,GAAG,CAAC,EAAE,OAAO,GAAG,WAAW,EAAE,OAAO,EAAE,EAAE,CAAC;QACvD,MAAM,GAAG,GAAG,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,EAAE,GAAG,GAAG,CAAC,GAAG,GAAG,CAAC,CAAC;QAC1D,MAAM,EAAE,GAAG,OAAO,IAAI,IAAI,GAAG,EAAE,CAAC;QAChC,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC;YACzB,OAAO,EAAE,CAAC;QACZ,CAAC;IACH,CAAC;IACD,MAAM,IAAI,KAAK,CAAC,gEAAgE,CAAC,CAAC;AACpF,CAAC;AAED,SAAS,cAAc;IACrB,MAAM,CAAC,GAAG,IAAI,IAAI,EAAE,CAAC;IACrB,MAAM,IAAI,GAAG,CAAC,CAAC,WAAW,EAAE,CAAC;IAC7B,MAAM,EAAE,GAAG,MAAM,CAAC,CAAC,CAAC,QAAQ,EAAE,GAAG,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC;IACrD,MAAM,EAAE,GAAG,MAAM,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC;IAChD,OAAO,GAAG,IAAI,IAAI,EAAE,IAAI,EAAE,EAAE,CAAC;AAC/B,CAAC;AAED,MAAM,OAAO,cAAc;IACR,OAAO,CAA4B;IAEpD,YAAY,UAA2B,EAAE;QACvC,IAAI,CAAC,OAAO,GAAG;YACb,MAAM,EAAE,OAAO,CAAC,MAAM,IAAI,iCAAiC;YAC3D,aAAa,EAAE,OAAO,CAAC,aAAa,IAAI,KAAK;SAC9C,CAAC;IACJ,CAAC;IAED;;;OAGG;IACH,QAAQ,CAAC,KAAoB,EAAE,cAAmC,IAAI,GAAG,EAAE;QACzE,MAAM,QAAQ,GAAG,IAAI,CAAC,aAAa,CAAC,KAAK,CAAC,CAAC;QAE3C,MAAM,QAAQ,GAAG,KAAK,CAAC,QAAQ,IAAI,QAAQ,CAAC;QAC5C,MAAM,UAAU,GAAG,KAAK,CAAC,eAAe,IAAI,uBAAuB,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;QACpF,MAAM,KAAK,GAAG,iBAAiB,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;QAChD,MAAM,EAAE,GAAG,UAAU,CAAC,WAAW,CAAC,CAAC;QACnC,MAAM,IAAI,GAAG,cAAc,EAAE,CAAC;QAE9B,MAAM,UAAU,GAAwB,KAAK,CAAC,eAAe,CAAC,GAAG,CAC/D,CAAC,OAAO,EAAE,GAAG,EAAE,EAAE,CAAC,CAAC;YACjB,KAAK;YACL,QAAQ,EAAE,OAAO;YACjB,KAAK,EAAE,iBAAiB,CAAC,OAAO,EAAE,KAAK,CAAC,QAAQ,CAAC;YACjD,WAAW,EAAE,WAAW,GAAG,GAAG,CAAC,cAAc,OAAO,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,GAAG;SAC5E,CAAC,CACH,CAAC;QAEF,MAAM,aAAa,GAAG,KAAK,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;YAC5D,KAAK,EAAE,OAAO,CAAC,IAAI,EAAE;YACrB,QAAQ,EAAE,SAAkB;SAC7B,CAAC,CAAC,CAAC;QAEJ,MAAM,aAAa,GAAG;YACpB;gBACE,KAAK,EAAE,0DAA0D;gBACjE,QAAQ,EAAE,YAAqB;aAChC;SACF,CAAC;QAEF,MAAM,UAAU,GAA6B,EAAE,CAAC;QAChD,IAAI,KAAK,CAAC,SAAS,IAAI,KAAK,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAClD,UAAU,CAAC,SAAS,GAAG,CAAC,GAAG,KAAK,CAAC,SAAS,CAAC,CAAC;QAC9C,CAAC;QACD,IAAI,KAAK,CAAC,SAAS,IAAI,KAAK,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAClD,UAAU,CAAC,WAAW,GAAG,CAAC,GAAG,KAAK,CAAC,SAAS,CAAC,CAAC;QAChD,CAAC;QAED,MAAM,aAAa,GAAG,UAAU,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC;QAE5D,MAAM,IAAI,GAA4B;YACpC,KAAK,EAAE,KAAK,CAAC,KAAK;YAClB,EAAE;YACF,cAAc,EAAE,IAAI,CAAC,OAAO,CAAC,aAAa;YAC1C,MAAM,EAAE,OAAO;YACf,WAAW,EAAE,KAAK,CAAC,iBAAiB;YACpC,MAAM,EAAE,IAAI,CAAC,OAAO,CAAC,MAAM;YAC3B,IAAI;YACJ,QAAQ;YACR,cAAc,EAAE,SAAS;YACzB,QAAQ,EAAE,OAAO;YACjB,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YAC7D,IAAI,EAAE;gBACJ,QAAQ,EAAE,KAAK,CAAC,QAAQ;gBACxB,UAAU,EAAE,QAAQ,KAAK,UAAU,IAAI,QAAQ,KAAK,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,QAAQ;aAC/E;YACD,YAAY,EAAE;gBACZ,IAAI,EAAE,UAAU;aACjB;YACD,SAAS,EAAE;gBACT,UAAU;gBACV,SAAS,EAAE,aAAa;gBACxB,eAAe,EAAE;oBACf,+CAA+C;iBAChD;aACF;YACD,QAAQ,EAAE;gBACR,OAAO,EAAE,CAAC,GAAG,mBAAmB,CAAC,QAAQ,CAAC,CAAC;gBAC3C,gBAAgB,EAAE,aAAa,KAAK,CAAC,QAAQ,iCAAiC;aAC/E;YACD,UAAU,EAAE;gBACV,cAAc,EAAE,aAAa;gBAC7B,cAAc,EAAE,aAAa;aAC9B;SACF,CAAC;QAEF,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE;YAC9B,MAAM,EAAE,CAAC;YACT,SAAS,EAAE,GAAG;YACd,MAAM,EAAE,IAAI;YACZ,QAAQ,EAAE,KAAK;YACf,WAAW,EAAE,GAAG;YAChB,WAAW,EAAE,KAAK;SACnB,CAAC,CAAC;QAEH,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,EAAE,EAAE,QAAQ,EAAE,CAAC;IACzC,CAAC;IAED;;;OAGG;IACK,aAAa,CAAC,KAAoB;QACxC,MAAM,QAAQ,GAAa,EAAE,CAAC;QAE9B,IAAI,CAAC,KAAK,CAAC,KAAK,IAAI,KAAK,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACpD,MAAM,IAAI,KAAK,CAAC,uDAAuD,CAAC,CAAC;QAC3E,CAAC;QACD,IAAI,CAAC,KAAK,CAAC,QAAQ,EAAE,CAAC;YACpB,MAAM,IAAI,KAAK,CAAC,oCAAoC,CAAC,CAAC;QACxD,CAAC;QACD,IAAI,CAAC,KAAK,CAAC,iBAAiB,IAAI,KAAK,CAAC,iBAAiB,CAAC,IAAI,EAAE,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC5E,MAAM,IAAI,KAAK,CAAC,mEAAmE,CAAC,CAAC;QACvF,CAAC;QACD,IAAI,CAAC,KAAK,CAAC,eAAe,IAAI,KAAK,CAAC,eAAe,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACjE,MAAM,IAAI,KAAK,CAAC,iEAAiE,CAAC,CAAC;QACrF,CAAC;QAED,IAAI,KAAK,CAAC,eAAe,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACrC,QAAQ,CAAC,IAAI,CACX,mFAAmF,CACpF,CAAC;QACJ,CAAC;QAED,OAAO,QAAQ,CAAC;IAClB,CAAC;CACF"}
|
|
@@ -1,58 +0,0 @@
|
|
|
1
|
-
/**
|
|
2
|
-
* SessionTracker - Tracks per-session state for behavioral detection operators.
|
|
3
|
-
*
|
|
4
|
-
* Enables multi-turn injection detection, call frequency tracking,
|
|
5
|
-
* and pattern repetition counting. All state is internal; public methods
|
|
6
|
-
* return copies to preserve immutability.
|
|
7
|
-
*
|
|
8
|
-
* @module agent-threat-rules/session-tracker
|
|
9
|
-
*/
|
|
10
|
-
import type { AgentEvent } from './types.js';
|
|
11
|
-
/** Snapshot of session state returned to callers (immutable copy) */
|
|
12
|
-
export interface SessionStateSnapshot {
|
|
13
|
-
readonly sessionId: string;
|
|
14
|
-
readonly eventCount: number;
|
|
15
|
-
readonly oldestEventTimestamp: number | undefined;
|
|
16
|
-
readonly newestEventTimestamp: number | undefined;
|
|
17
|
-
/** Event history for sequence detection (immutable copies) */
|
|
18
|
-
readonly events: readonly Readonly<AgentEvent>[];
|
|
19
|
-
}
|
|
20
|
-
export declare class SessionTracker {
|
|
21
|
-
private readonly sessions;
|
|
22
|
-
/**
|
|
23
|
-
* Record an agent event for the given session.
|
|
24
|
-
* Extracts tool name and patterns from event fields/content.
|
|
25
|
-
*/
|
|
26
|
-
recordEvent(sessionId: string, event: AgentEvent, patterns?: readonly string[]): void;
|
|
27
|
-
/**
|
|
28
|
-
* Get the number of calls to a specific tool within a time window.
|
|
29
|
-
*/
|
|
30
|
-
getCallFrequency(sessionId: string, toolName: string, windowMs: number): number;
|
|
31
|
-
/**
|
|
32
|
-
* Get the number of times a pattern has been observed within a time window.
|
|
33
|
-
*/
|
|
34
|
-
getPatternFrequency(sessionId: string, pattern: string, windowMs: number): number;
|
|
35
|
-
/**
|
|
36
|
-
* Get total event count for a session, optionally within a time window.
|
|
37
|
-
*/
|
|
38
|
-
getEventCount(sessionId: string, windowMs?: number): number;
|
|
39
|
-
/**
|
|
40
|
-
* Get an immutable snapshot of session state. Returns undefined if session does not exist.
|
|
41
|
-
*/
|
|
42
|
-
getSessionSnapshot(sessionId: string): SessionStateSnapshot | undefined;
|
|
43
|
-
/**
|
|
44
|
-
* Evict sessions that have been inactive longer than maxAgeMs.
|
|
45
|
-
* Returns the number of sessions evicted.
|
|
46
|
-
*/
|
|
47
|
-
cleanup(maxAgeMs: number): number;
|
|
48
|
-
/** Get the number of tracked sessions */
|
|
49
|
-
getSessionCount(): number;
|
|
50
|
-
/**
|
|
51
|
-
* Ensure we don't exceed the maximum session count.
|
|
52
|
-
* Evicts the oldest session if at capacity.
|
|
53
|
-
*/
|
|
54
|
-
private ensureCapacity;
|
|
55
|
-
private getOrCreateSession;
|
|
56
|
-
private extractToolName;
|
|
57
|
-
}
|
|
58
|
-
//# sourceMappingURL=session-tracker.d.ts.map
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"session-tracker.d.ts","sourceRoot":"","sources":["../src/session-tracker.ts"],"names":[],"mappings":"AAAA;;;;;;;;GAQG;AAEH,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,YAAY,CAAC;AAgB7C,qEAAqE;AACrE,MAAM,WAAW,oBAAoB;IACnC,QAAQ,CAAC,SAAS,EAAE,MAAM,CAAC;IAC3B,QAAQ,CAAC,UAAU,EAAE,MAAM,CAAC;IAC5B,QAAQ,CAAC,oBAAoB,EAAE,MAAM,GAAG,SAAS,CAAC;IAClD,QAAQ,CAAC,oBAAoB,EAAE,MAAM,GAAG,SAAS,CAAC;IAClD,8DAA8D;IAC9D,QAAQ,CAAC,MAAM,EAAE,SAAS,QAAQ,CAAC,UAAU,CAAC,EAAE,CAAC;CAClD;AAWD,qBAAa,cAAc;IACzB,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAmC;IAE5D;;;OAGG;IACH,WAAW,CAAC,SAAS,EAAE,MAAM,EAAE,KAAK,EAAE,UAAU,EAAE,QAAQ,GAAE,SAAS,MAAM,EAAO,GAAG,IAAI;IAkCzF;;OAEG;IACH,gBAAgB,CAAC,SAAS,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,GAAG,MAAM;IAc/E;;OAEG;IACH,mBAAmB,CAAC,SAAS,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,GAAG,MAAM;IAcjF;;OAEG;IACH,aAAa,CAAC,SAAS,EAAE,MAAM,EAAE,QAAQ,CAAC,EAAE,MAAM,GAAG,MAAM;IAkB3D;;OAEG;IACH,kBAAkB,CAAC,SAAS,EAAE,MAAM,GAAG,oBAAoB,GAAG,SAAS;IAgBvE;;;OAGG;IACH,OAAO,CAAC,QAAQ,EAAE,MAAM,GAAG,MAAM;IAcjC,yCAAyC;IACzC,eAAe,IAAI,MAAM;IAIzB;;;OAGG;IACH,OAAO,CAAC,cAAc;IAkBtB,OAAO,CAAC,kBAAkB;IAgB1B,OAAO,CAAC,eAAe;CAMxB"}
|
package/dist/session-tracker.js
DELETED
|
@@ -1,176 +0,0 @@
|
|
|
1
|
-
/**
|
|
2
|
-
* SessionTracker - Tracks per-session state for behavioral detection operators.
|
|
3
|
-
*
|
|
4
|
-
* Enables multi-turn injection detection, call frequency tracking,
|
|
5
|
-
* and pattern repetition counting. All state is internal; public methods
|
|
6
|
-
* return copies to preserve immutability.
|
|
7
|
-
*
|
|
8
|
-
* @module agent-threat-rules/session-tracker
|
|
9
|
-
*/
|
|
10
|
-
/** Maximum number of events stored per session */
|
|
11
|
-
const MAX_EVENTS_PER_SESSION = 1000;
|
|
12
|
-
/** Maximum number of tracked sessions */
|
|
13
|
-
const MAX_SESSIONS = 10_000;
|
|
14
|
-
export class SessionTracker {
|
|
15
|
-
sessions = new Map();
|
|
16
|
-
/**
|
|
17
|
-
* Record an agent event for the given session.
|
|
18
|
-
* Extracts tool name and patterns from event fields/content.
|
|
19
|
-
*/
|
|
20
|
-
recordEvent(sessionId, event, patterns = []) {
|
|
21
|
-
this.ensureCapacity();
|
|
22
|
-
const state = this.getOrCreateSession(sessionId);
|
|
23
|
-
const toolName = this.extractToolName(event);
|
|
24
|
-
const now = Date.now();
|
|
25
|
-
const tracked = {
|
|
26
|
-
event: Object.freeze({ ...event }),
|
|
27
|
-
recordedAt: now,
|
|
28
|
-
toolName,
|
|
29
|
-
patterns,
|
|
30
|
-
};
|
|
31
|
-
// Evict oldest if at capacity
|
|
32
|
-
if (state.events.length >= MAX_EVENTS_PER_SESSION) {
|
|
33
|
-
state.events = state.events.slice(1);
|
|
34
|
-
}
|
|
35
|
-
state.events = [...state.events, tracked];
|
|
36
|
-
state.lastActivityAt = now;
|
|
37
|
-
// Update call counts
|
|
38
|
-
if (toolName) {
|
|
39
|
-
const prev = state.callCounts.get(toolName) ?? 0;
|
|
40
|
-
state.callCounts.set(toolName, prev + 1);
|
|
41
|
-
}
|
|
42
|
-
// Update pattern counts
|
|
43
|
-
for (const p of patterns) {
|
|
44
|
-
const prev = state.patternCounts.get(p) ?? 0;
|
|
45
|
-
state.patternCounts.set(p, prev + 1);
|
|
46
|
-
}
|
|
47
|
-
}
|
|
48
|
-
/**
|
|
49
|
-
* Get the number of calls to a specific tool within a time window.
|
|
50
|
-
*/
|
|
51
|
-
getCallFrequency(sessionId, toolName, windowMs) {
|
|
52
|
-
const state = this.sessions.get(sessionId);
|
|
53
|
-
if (!state)
|
|
54
|
-
return 0;
|
|
55
|
-
const cutoff = Date.now() - windowMs;
|
|
56
|
-
let count = 0;
|
|
57
|
-
for (const tracked of state.events) {
|
|
58
|
-
if (tracked.recordedAt >= cutoff && tracked.toolName === toolName) {
|
|
59
|
-
count++;
|
|
60
|
-
}
|
|
61
|
-
}
|
|
62
|
-
return count;
|
|
63
|
-
}
|
|
64
|
-
/**
|
|
65
|
-
* Get the number of times a pattern has been observed within a time window.
|
|
66
|
-
*/
|
|
67
|
-
getPatternFrequency(sessionId, pattern, windowMs) {
|
|
68
|
-
const state = this.sessions.get(sessionId);
|
|
69
|
-
if (!state)
|
|
70
|
-
return 0;
|
|
71
|
-
const cutoff = Date.now() - windowMs;
|
|
72
|
-
let count = 0;
|
|
73
|
-
for (const tracked of state.events) {
|
|
74
|
-
if (tracked.recordedAt >= cutoff && tracked.patterns.includes(pattern)) {
|
|
75
|
-
count++;
|
|
76
|
-
}
|
|
77
|
-
}
|
|
78
|
-
return count;
|
|
79
|
-
}
|
|
80
|
-
/**
|
|
81
|
-
* Get total event count for a session, optionally within a time window.
|
|
82
|
-
*/
|
|
83
|
-
getEventCount(sessionId, windowMs) {
|
|
84
|
-
const state = this.sessions.get(sessionId);
|
|
85
|
-
if (!state)
|
|
86
|
-
return 0;
|
|
87
|
-
if (windowMs === undefined) {
|
|
88
|
-
return state.events.length;
|
|
89
|
-
}
|
|
90
|
-
const cutoff = Date.now() - windowMs;
|
|
91
|
-
let count = 0;
|
|
92
|
-
for (const tracked of state.events) {
|
|
93
|
-
if (tracked.recordedAt >= cutoff) {
|
|
94
|
-
count++;
|
|
95
|
-
}
|
|
96
|
-
}
|
|
97
|
-
return count;
|
|
98
|
-
}
|
|
99
|
-
/**
|
|
100
|
-
* Get an immutable snapshot of session state. Returns undefined if session does not exist.
|
|
101
|
-
*/
|
|
102
|
-
getSessionSnapshot(sessionId) {
|
|
103
|
-
const state = this.sessions.get(sessionId);
|
|
104
|
-
if (!state)
|
|
105
|
-
return undefined;
|
|
106
|
-
const oldest = state.events.length > 0 ? state.events[0].recordedAt : undefined;
|
|
107
|
-
const newest = state.events.length > 0 ? state.events[state.events.length - 1].recordedAt : undefined;
|
|
108
|
-
return Object.freeze({
|
|
109
|
-
sessionId,
|
|
110
|
-
eventCount: state.events.length,
|
|
111
|
-
oldestEventTimestamp: oldest,
|
|
112
|
-
newestEventTimestamp: newest,
|
|
113
|
-
events: Object.freeze(state.events.map((e) => e.event)),
|
|
114
|
-
});
|
|
115
|
-
}
|
|
116
|
-
/**
|
|
117
|
-
* Evict sessions that have been inactive longer than maxAgeMs.
|
|
118
|
-
* Returns the number of sessions evicted.
|
|
119
|
-
*/
|
|
120
|
-
cleanup(maxAgeMs) {
|
|
121
|
-
const cutoff = Date.now() - maxAgeMs;
|
|
122
|
-
let evicted = 0;
|
|
123
|
-
for (const [id, state] of this.sessions) {
|
|
124
|
-
if (state.lastActivityAt < cutoff) {
|
|
125
|
-
this.sessions.delete(id);
|
|
126
|
-
evicted++;
|
|
127
|
-
}
|
|
128
|
-
}
|
|
129
|
-
return evicted;
|
|
130
|
-
}
|
|
131
|
-
/** Get the number of tracked sessions */
|
|
132
|
-
getSessionCount() {
|
|
133
|
-
return this.sessions.size;
|
|
134
|
-
}
|
|
135
|
-
/**
|
|
136
|
-
* Ensure we don't exceed the maximum session count.
|
|
137
|
-
* Evicts the oldest session if at capacity.
|
|
138
|
-
*/
|
|
139
|
-
ensureCapacity() {
|
|
140
|
-
if (this.sessions.size < MAX_SESSIONS)
|
|
141
|
-
return;
|
|
142
|
-
let oldestId;
|
|
143
|
-
let oldestTime = Infinity;
|
|
144
|
-
for (const [id, state] of this.sessions) {
|
|
145
|
-
if (state.lastActivityAt < oldestTime) {
|
|
146
|
-
oldestTime = state.lastActivityAt;
|
|
147
|
-
oldestId = id;
|
|
148
|
-
}
|
|
149
|
-
}
|
|
150
|
-
if (oldestId) {
|
|
151
|
-
this.sessions.delete(oldestId);
|
|
152
|
-
}
|
|
153
|
-
}
|
|
154
|
-
getOrCreateSession(sessionId) {
|
|
155
|
-
const existing = this.sessions.get(sessionId);
|
|
156
|
-
if (existing)
|
|
157
|
-
return existing;
|
|
158
|
-
const now = Date.now();
|
|
159
|
-
const state = {
|
|
160
|
-
events: [],
|
|
161
|
-
callCounts: new Map(),
|
|
162
|
-
patternCounts: new Map(),
|
|
163
|
-
createdAt: now,
|
|
164
|
-
lastActivityAt: now,
|
|
165
|
-
};
|
|
166
|
-
this.sessions.set(sessionId, state);
|
|
167
|
-
return state;
|
|
168
|
-
}
|
|
169
|
-
extractToolName(event) {
|
|
170
|
-
if (event.type === 'tool_call' || event.type === 'tool_response') {
|
|
171
|
-
return event.fields?.['tool_name'] ?? event.content;
|
|
172
|
-
}
|
|
173
|
-
return undefined;
|
|
174
|
-
}
|
|
175
|
-
}
|
|
176
|
-
//# sourceMappingURL=session-tracker.js.map
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"session-tracker.js","sourceRoot":"","sources":["../src/session-tracker.ts"],"names":[],"mappings":"AAAA;;;;;;;;GAQG;AAIH,kDAAkD;AAClD,MAAM,sBAAsB,GAAG,IAAI,CAAC;AAEpC,yCAAyC;AACzC,MAAM,YAAY,GAAG,MAAM,CAAC;AA6B5B,MAAM,OAAO,cAAc;IACR,QAAQ,GAAG,IAAI,GAAG,EAAwB,CAAC;IAE5D;;;OAGG;IACH,WAAW,CAAC,SAAiB,EAAE,KAAiB,EAAE,WAA8B,EAAE;QAChF,IAAI,CAAC,cAAc,EAAE,CAAC;QACtB,MAAM,KAAK,GAAG,IAAI,CAAC,kBAAkB,CAAC,SAAS,CAAC,CAAC;QACjD,MAAM,QAAQ,GAAG,IAAI,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;QAC7C,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAEvB,MAAM,OAAO,GAAiB;YAC5B,KAAK,EAAE,MAAM,CAAC,MAAM,CAAC,EAAE,GAAG,KAAK,EAAE,CAAC;YAClC,UAAU,EAAE,GAAG;YACf,QAAQ;YACR,QAAQ;SACT,CAAC;QAEF,8BAA8B;QAC9B,IAAI,KAAK,CAAC,MAAM,CAAC,MAAM,IAAI,sBAAsB,EAAE,CAAC;YAClD,KAAK,CAAC,MAAM,GAAG,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QACvC,CAAC;QAED,KAAK,CAAC,MAAM,GAAG,CAAC,GAAG,KAAK,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;QAC1C,KAAK,CAAC,cAAc,GAAG,GAAG,CAAC;QAE3B,qBAAqB;QACrB,IAAI,QAAQ,EAAE,CAAC;YACb,MAAM,IAAI,GAAG,KAAK,CAAC,UAAU,CAAC,GAAG,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;YACjD,KAAK,CAAC,UAAU,CAAC,GAAG,CAAC,QAAQ,EAAE,IAAI,GAAG,CAAC,CAAC,CAAC;QAC3C,CAAC;QAED,wBAAwB;QACxB,KAAK,MAAM,CAAC,IAAI,QAAQ,EAAE,CAAC;YACzB,MAAM,IAAI,GAAG,KAAK,CAAC,aAAa,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;YAC7C,KAAK,CAAC,aAAa,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,GAAG,CAAC,CAAC,CAAC;QACvC,CAAC;IACH,CAAC;IAED;;OAEG;IACH,gBAAgB,CAAC,SAAiB,EAAE,QAAgB,EAAE,QAAgB;QACpE,MAAM,KAAK,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAC3C,IAAI,CAAC,KAAK;YAAE,OAAO,CAAC,CAAC;QAErB,MAAM,MAAM,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,QAAQ,CAAC;QACrC,IAAI,KAAK,GAAG,CAAC,CAAC;QACd,KAAK,MAAM,OAAO,IAAI,KAAK,CAAC,MAAM,EAAE,CAAC;YACnC,IAAI,OAAO,CAAC,UAAU,IAAI,MAAM,IAAI,OAAO,CAAC,QAAQ,KAAK,QAAQ,EAAE,CAAC;gBAClE,KAAK,EAAE,CAAC;YACV,CAAC;QACH,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAED;;OAEG;IACH,mBAAmB,CAAC,SAAiB,EAAE,OAAe,EAAE,QAAgB;QACtE,MAAM,KAAK,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAC3C,IAAI,CAAC,KAAK;YAAE,OAAO,CAAC,CAAC;QAErB,MAAM,MAAM,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,QAAQ,CAAC;QACrC,IAAI,KAAK,GAAG,CAAC,CAAC;QACd,KAAK,MAAM,OAAO,IAAI,KAAK,CAAC,MAAM,EAAE,CAAC;YACnC,IAAI,OAAO,CAAC,UAAU,IAAI,MAAM,IAAI,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAC,OAAO,CAAC,EAAE,CAAC;gBACvE,KAAK,EAAE,CAAC;YACV,CAAC;QACH,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAED;;OAEG;IACH,aAAa,CAAC,SAAiB,EAAE,QAAiB;QAChD,MAAM,KAAK,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAC3C,IAAI,CAAC,KAAK;YAAE,OAAO,CAAC,CAAC;QAErB,IAAI,QAAQ,KAAK,SAAS,EAAE,CAAC;YAC3B,OAAO,KAAK,CAAC,MAAM,CAAC,MAAM,CAAC;QAC7B,CAAC;QAED,MAAM,MAAM,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,QAAQ,CAAC;QACrC,IAAI,KAAK,GAAG,CAAC,CAAC;QACd,KAAK,MAAM,OAAO,IAAI,KAAK,CAAC,MAAM,EAAE,CAAC;YACnC,IAAI,OAAO,CAAC,UAAU,IAAI,MAAM,EAAE,CAAC;gBACjC,KAAK,EAAE,CAAC;YACV,CAAC;QACH,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAED;;OAEG;IACH,kBAAkB,CAAC,SAAiB;QAClC,MAAM,KAAK,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAC3C,IAAI,CAAC,KAAK;YAAE,OAAO,SAAS,CAAC;QAE7B,MAAM,MAAM,GAAG,KAAK,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAE,CAAC,UAAU,CAAC,CAAC,CAAC,SAAS,CAAC;QACjF,MAAM,MAAM,GAAG,KAAK,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,CAAE,CAAC,UAAU,CAAC,CAAC,CAAC,SAAS,CAAC;QAEvG,OAAO,MAAM,CAAC,MAAM,CAAC;YACnB,SAAS;YACT,UAAU,EAAE,KAAK,CAAC,MAAM,CAAC,MAAM;YAC/B,oBAAoB,EAAE,MAAM;YAC5B,oBAAoB,EAAE,MAAM;YAC5B,MAAM,EAAE,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC;SACxD,CAAC,CAAC;IACL,CAAC;IAED;;;OAGG;IACH,OAAO,CAAC,QAAgB;QACtB,MAAM,MAAM,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,QAAQ,CAAC;QACrC,IAAI,OAAO,GAAG,CAAC,CAAC;QAEhB,KAAK,MAAM,CAAC,EAAE,EAAE,KAAK,CAAC,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;YACxC,IAAI,KAAK,CAAC,cAAc,GAAG,MAAM,EAAE,CAAC;gBAClC,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;gBACzB,OAAO,EAAE,CAAC;YACZ,CAAC;QACH,CAAC;QAED,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,yCAAyC;IACzC,eAAe;QACb,OAAO,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC;IAC5B,CAAC;IAED;;;OAGG;IACK,cAAc;QACpB,IAAI,IAAI,CAAC,QAAQ,CAAC,IAAI,GAAG,YAAY;YAAE,OAAO;QAE9C,IAAI,QAA4B,CAAC;QACjC,IAAI,UAAU,GAAG,QAAQ,CAAC;QAE1B,KAAK,MAAM,CAAC,EAAE,EAAE,KAAK,CAAC,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;YACxC,IAAI,KAAK,CAAC,cAAc,GAAG,UAAU,EAAE,CAAC;gBACtC,UAAU,GAAG,KAAK,CAAC,cAAc,CAAC;gBAClC,QAAQ,GAAG,EAAE,CAAC;YAChB,CAAC;QACH,CAAC;QAED,IAAI,QAAQ,EAAE,CAAC;YACb,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;QACjC,CAAC;IACH,CAAC;IAEO,kBAAkB,CAAC,SAAiB;QAC1C,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAC9C,IAAI,QAAQ;YAAE,OAAO,QAAQ,CAAC;QAE9B,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACvB,MAAM,KAAK,GAAiB;YAC1B,MAAM,EAAE,EAAE;YACV,UAAU,EAAE,IAAI,GAAG,EAAE;YACrB,aAAa,EAAE,IAAI,GAAG,EAAE;YACxB,SAAS,EAAE,GAAG;YACd,cAAc,EAAE,GAAG;SACpB,CAAC;QACF,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC;QACpC,OAAO,KAAK,CAAC;IACf,CAAC;IAEO,eAAe,CAAC,KAAiB;QACvC,IAAI,KAAK,CAAC,IAAI,KAAK,WAAW,IAAI,KAAK,CAAC,IAAI,KAAK,eAAe,EAAE,CAAC;YACjE,OAAO,KAAK,CAAC,MAAM,EAAE,CAAC,WAAW,CAAC,IAAI,KAAK,CAAC,OAAO,CAAC;QACtD,CAAC;QACD,OAAO,SAAS,CAAC;IACnB,CAAC;CACF"}
|
|
@@ -1,48 +0,0 @@
|
|
|
1
|
-
/**
|
|
2
|
-
* Shadow Evaluator -- runs experimental rules without affecting verdict.
|
|
3
|
-
*
|
|
4
|
-
* Experimental rules evaluate against every event but don't influence
|
|
5
|
-
* the real verdict. Their match/no-match results are tracked to measure
|
|
6
|
-
* false positive rates. Rules with FP < threshold get promoted.
|
|
7
|
-
*
|
|
8
|
-
* @module agent-threat-rules/shadow-evaluator
|
|
9
|
-
*/
|
|
10
|
-
import type { ATRRule, ATRMatch, AgentEvent } from './types.js';
|
|
11
|
-
interface ShadowRuleStats {
|
|
12
|
-
readonly ruleId: string;
|
|
13
|
-
totalEvaluations: number;
|
|
14
|
-
totalMatches: number;
|
|
15
|
-
confirmedTruePositives: number;
|
|
16
|
-
confirmedFalsePositives: number;
|
|
17
|
-
firstSeen: number;
|
|
18
|
-
lastSeen: number;
|
|
19
|
-
}
|
|
20
|
-
export interface PromotionCandidate {
|
|
21
|
-
readonly rule: ATRRule;
|
|
22
|
-
readonly stats: Readonly<ShadowRuleStats>;
|
|
23
|
-
readonly fpRate: number;
|
|
24
|
-
}
|
|
25
|
-
export declare class ShadowEvaluator {
|
|
26
|
-
private readonly rules;
|
|
27
|
-
private readonly stats;
|
|
28
|
-
private readonly compiledPatterns;
|
|
29
|
-
/** Add an experimental rule for shadow evaluation */
|
|
30
|
-
addRule(rule: ATRRule): void;
|
|
31
|
-
/** Evaluate all shadow rules against an event (does NOT affect verdict) */
|
|
32
|
-
evaluate(event: AgentEvent): readonly ATRMatch[];
|
|
33
|
-
/** Record user feedback on a shadow match */
|
|
34
|
-
recordFeedback(ruleId: string, isTruePositive: boolean): void;
|
|
35
|
-
/**
|
|
36
|
-
* Get rules ready for promotion.
|
|
37
|
-
* Criteria: FP rate < maxFPRate AND minimum evaluations reached.
|
|
38
|
-
*/
|
|
39
|
-
getPromotionCandidates(maxFPRate?: number, minEvaluations?: number): readonly PromotionCandidate[];
|
|
40
|
-
/** Get stats for a specific rule */
|
|
41
|
-
getStats(ruleId: string): Readonly<ShadowRuleStats> | undefined;
|
|
42
|
-
/** Get all shadow rule stats */
|
|
43
|
-
getAllStats(): ReadonlyMap<string, Readonly<ShadowRuleStats>>;
|
|
44
|
-
/** Number of shadow rules */
|
|
45
|
-
size(): number;
|
|
46
|
-
}
|
|
47
|
-
export {};
|
|
48
|
-
//# sourceMappingURL=shadow-evaluator.d.ts.map
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"shadow-evaluator.d.ts","sourceRoot":"","sources":["../src/shadow-evaluator.ts"],"names":[],"mappings":"AAAA;;;;;;;;GAQG;AAEH,OAAO,KAAK,EAAE,OAAO,EAAE,QAAQ,EAAE,UAAU,EAAE,MAAM,YAAY,CAAC;AAEhE,UAAU,eAAe;IACvB,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAC;IACxB,gBAAgB,EAAE,MAAM,CAAC;IACzB,YAAY,EAAE,MAAM,CAAC;IACrB,sBAAsB,EAAE,MAAM,CAAC;IAC/B,uBAAuB,EAAE,MAAM,CAAC;IAChC,SAAS,EAAE,MAAM,CAAC;IAClB,QAAQ,EAAE,MAAM,CAAC;CAClB;AAED,MAAM,WAAW,kBAAkB;IACjC,QAAQ,CAAC,IAAI,EAAE,OAAO,CAAC;IACvB,QAAQ,CAAC,KAAK,EAAE,QAAQ,CAAC,eAAe,CAAC,CAAC;IAC1C,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAC;CACzB;AAED,qBAAa,eAAe;IAC1B,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAiB;IACvC,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAsC;IAC5D,OAAO,CAAC,QAAQ,CAAC,gBAAgB,CAA6B;IAE9D,qDAAqD;IACrD,OAAO,CAAC,IAAI,EAAE,OAAO,GAAG,IAAI;IAc5B,2EAA2E;IAC3E,QAAQ,CAAC,KAAK,EAAE,UAAU,GAAG,SAAS,QAAQ,EAAE;IAsDhD,6CAA6C;IAC7C,cAAc,CAAC,MAAM,EAAE,MAAM,EAAE,cAAc,EAAE,OAAO,GAAG,IAAI;IAU7D;;;OAGG;IACH,sBAAsB,CACpB,SAAS,GAAE,MAAc,EACzB,cAAc,GAAE,MAAa,GAC5B,SAAS,kBAAkB,EAAE;IAuBhC,oCAAoC;IACpC,QAAQ,CAAC,MAAM,EAAE,MAAM,GAAG,QAAQ,CAAC,eAAe,CAAC,GAAG,SAAS;IAK/D,gCAAgC;IAChC,WAAW,IAAI,WAAW,CAAC,MAAM,EAAE,QAAQ,CAAC,eAAe,CAAC,CAAC;IAI7D,6BAA6B;IAC7B,IAAI,IAAI,MAAM;CAGf"}
|