workerclaw 0.2.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/LICENSE +21 -0
- package/README.md +129 -0
- package/dist/active-behavior/behavior-scheduler.d.ts +119 -0
- package/dist/active-behavior/behavior-scheduler.d.ts.map +1 -0
- package/dist/active-behavior/behavior-scheduler.js +297 -0
- package/dist/active-behavior/behavior-scheduler.js.map +1 -0
- package/dist/active-behavior/frequency-control.d.ts +72 -0
- package/dist/active-behavior/frequency-control.d.ts.map +1 -0
- package/dist/active-behavior/frequency-control.js +206 -0
- package/dist/active-behavior/frequency-control.js.map +1 -0
- package/dist/active-behavior/index.d.ts +8 -0
- package/dist/active-behavior/index.d.ts.map +1 -0
- package/dist/active-behavior/index.js +6 -0
- package/dist/active-behavior/index.js.map +1 -0
- package/dist/agent/agent-engine.d.ts +122 -0
- package/dist/agent/agent-engine.d.ts.map +1 -0
- package/dist/agent/agent-engine.js +380 -0
- package/dist/agent/agent-engine.js.map +1 -0
- package/dist/agent/context-window.d.ts +69 -0
- package/dist/agent/context-window.d.ts.map +1 -0
- package/dist/agent/context-window.js +210 -0
- package/dist/agent/context-window.js.map +1 -0
- package/dist/agent/llm-client.d.ts +37 -0
- package/dist/agent/llm-client.d.ts.map +1 -0
- package/dist/agent/llm-client.js +155 -0
- package/dist/agent/llm-client.js.map +1 -0
- package/dist/agent/personality.d.ts +71 -0
- package/dist/agent/personality.d.ts.map +1 -0
- package/dist/agent/personality.js +147 -0
- package/dist/agent/personality.js.map +1 -0
- package/dist/agent/session-manager.d.ts +111 -0
- package/dist/agent/session-manager.d.ts.map +1 -0
- package/dist/agent/session-manager.js +205 -0
- package/dist/agent/session-manager.js.map +1 -0
- package/dist/agent/tool-executor.d.ts +34 -0
- package/dist/agent/tool-executor.d.ts.map +1 -0
- package/dist/agent/tool-executor.js +145 -0
- package/dist/agent/tool-executor.js.map +1 -0
- package/dist/agent/tool-registry.d.ts +67 -0
- package/dist/agent/tool-registry.d.ts.map +1 -0
- package/dist/agent/tool-registry.js +272 -0
- package/dist/agent/tool-registry.js.map +1 -0
- package/dist/cli/configure.d.ts +16 -0
- package/dist/cli/configure.d.ts.map +1 -0
- package/dist/cli/configure.js +212 -0
- package/dist/cli/configure.js.map +1 -0
- package/dist/cli/index.d.ts +17 -0
- package/dist/cli/index.d.ts.map +1 -0
- package/dist/cli/index.js +256 -0
- package/dist/cli/index.js.map +1 -0
- package/dist/cli/prompter.d.ts +59 -0
- package/dist/cli/prompter.d.ts.map +1 -0
- package/dist/cli/prompter.js +120 -0
- package/dist/cli/prompter.js.map +1 -0
- package/dist/cli/sections/index.d.ts +9 -0
- package/dist/cli/sections/index.d.ts.map +1 -0
- package/dist/cli/sections/index.js +9 -0
- package/dist/cli/sections/index.js.map +1 -0
- package/dist/cli/sections/llm.d.ts +14 -0
- package/dist/cli/sections/llm.d.ts.map +1 -0
- package/dist/cli/sections/llm.js +155 -0
- package/dist/cli/sections/llm.js.map +1 -0
- package/dist/cli/sections/personality.d.ts +14 -0
- package/dist/cli/sections/personality.d.ts.map +1 -0
- package/dist/cli/sections/personality.js +90 -0
- package/dist/cli/sections/personality.js.map +1 -0
- package/dist/cli/sections/platform.d.ts +14 -0
- package/dist/cli/sections/platform.d.ts.map +1 -0
- package/dist/cli/sections/platform.js +179 -0
- package/dist/cli/sections/platform.js.map +1 -0
- package/dist/cli/sections/security.d.ts +14 -0
- package/dist/cli/sections/security.d.ts.map +1 -0
- package/dist/cli/sections/security.js +106 -0
- package/dist/cli/sections/security.js.map +1 -0
- package/dist/cli/sections/skills.d.ts +10 -0
- package/dist/cli/sections/skills.d.ts.map +1 -0
- package/dist/cli/sections/skills.js +149 -0
- package/dist/cli/sections/skills.js.map +1 -0
- package/dist/cli.d.ts +8 -0
- package/dist/cli.d.ts.map +1 -0
- package/dist/cli.js +116 -0
- package/dist/cli.js.map +1 -0
- package/dist/core/config.d.ts +175 -0
- package/dist/core/config.d.ts.map +1 -0
- package/dist/core/config.js +109 -0
- package/dist/core/config.js.map +1 -0
- package/dist/core/events.d.ts +211 -0
- package/dist/core/events.d.ts.map +1 -0
- package/dist/core/events.js +152 -0
- package/dist/core/events.js.map +1 -0
- package/dist/core/logger.d.ts +23 -0
- package/dist/core/logger.d.ts.map +1 -0
- package/dist/core/logger.js +53 -0
- package/dist/core/logger.js.map +1 -0
- package/dist/core/workerclaw.d.ts +83 -0
- package/dist/core/workerclaw.d.ts.map +1 -0
- package/dist/core/workerclaw.js +161 -0
- package/dist/core/workerclaw.js.map +1 -0
- package/dist/index.d.ts +27 -0
- package/dist/index.d.ts.map +1 -0
- package/dist/index.js +21 -0
- package/dist/index.js.map +1 -0
- package/dist/ingress/message-parser.d.ts +36 -0
- package/dist/ingress/message-parser.d.ts.map +1 -0
- package/dist/ingress/message-parser.js +107 -0
- package/dist/ingress/message-parser.js.map +1 -0
- package/dist/ingress/miniabc-client.d.ts +109 -0
- package/dist/ingress/miniabc-client.d.ts.map +1 -0
- package/dist/ingress/miniabc-client.js +345 -0
- package/dist/ingress/miniabc-client.js.map +1 -0
- package/dist/ingress/platform-api.d.ts +82 -0
- package/dist/ingress/platform-api.d.ts.map +1 -0
- package/dist/ingress/platform-api.js +209 -0
- package/dist/ingress/platform-api.js.map +1 -0
- package/dist/sandbox/command-sandbox.d.ts +46 -0
- package/dist/sandbox/command-sandbox.d.ts.map +1 -0
- package/dist/sandbox/command-sandbox.js +144 -0
- package/dist/sandbox/command-sandbox.js.map +1 -0
- package/dist/sandbox/fs-sandbox.d.ts +56 -0
- package/dist/sandbox/fs-sandbox.d.ts.map +1 -0
- package/dist/sandbox/fs-sandbox.js +119 -0
- package/dist/sandbox/fs-sandbox.js.map +1 -0
- package/dist/sandbox/network-sandbox.d.ts +41 -0
- package/dist/sandbox/network-sandbox.d.ts.map +1 -0
- package/dist/sandbox/network-sandbox.js +114 -0
- package/dist/sandbox/network-sandbox.js.map +1 -0
- package/dist/security/content-scanner.d.ts +59 -0
- package/dist/security/content-scanner.d.ts.map +1 -0
- package/dist/security/content-scanner.js +289 -0
- package/dist/security/content-scanner.js.map +1 -0
- package/dist/security/gate.d.ts +78 -0
- package/dist/security/gate.d.ts.map +1 -0
- package/dist/security/gate.js +150 -0
- package/dist/security/gate.js.map +1 -0
- package/dist/security/permission-level.d.ts +68 -0
- package/dist/security/permission-level.d.ts.map +1 -0
- package/dist/security/permission-level.js +191 -0
- package/dist/security/permission-level.js.map +1 -0
- package/dist/security/rate-limiter.d.ts +52 -0
- package/dist/security/rate-limiter.d.ts.map +1 -0
- package/dist/security/rate-limiter.js +133 -0
- package/dist/security/rate-limiter.js.map +1 -0
- package/dist/security/source-verifier.d.ts +33 -0
- package/dist/security/source-verifier.d.ts.map +1 -0
- package/dist/security/source-verifier.js +106 -0
- package/dist/security/source-verifier.js.map +1 -0
- package/dist/skills/builtin/code.d.ts +27 -0
- package/dist/skills/builtin/code.d.ts.map +1 -0
- package/dist/skills/builtin/code.js +132 -0
- package/dist/skills/builtin/code.js.map +1 -0
- package/dist/skills/builtin/index.d.ts +12 -0
- package/dist/skills/builtin/index.d.ts.map +1 -0
- package/dist/skills/builtin/index.js +16 -0
- package/dist/skills/builtin/index.js.map +1 -0
- package/dist/skills/builtin/search.d.ts +27 -0
- package/dist/skills/builtin/search.d.ts.map +1 -0
- package/dist/skills/builtin/search.js +105 -0
- package/dist/skills/builtin/search.js.map +1 -0
- package/dist/skills/builtin/writing.d.ts +22 -0
- package/dist/skills/builtin/writing.d.ts.map +1 -0
- package/dist/skills/builtin/writing.js +93 -0
- package/dist/skills/builtin/writing.js.map +1 -0
- package/dist/skills/index.d.ts +11 -0
- package/dist/skills/index.d.ts.map +1 -0
- package/dist/skills/index.js +10 -0
- package/dist/skills/index.js.map +1 -0
- package/dist/skills/pack-loader.d.ts +31 -0
- package/dist/skills/pack-loader.d.ts.map +1 -0
- package/dist/skills/pack-loader.js +167 -0
- package/dist/skills/pack-loader.js.map +1 -0
- package/dist/skills/pack-registry.d.ts +40 -0
- package/dist/skills/pack-registry.d.ts.map +1 -0
- package/dist/skills/pack-registry.js +80 -0
- package/dist/skills/pack-registry.js.map +1 -0
- package/dist/skills/pack-types.d.ts +64 -0
- package/dist/skills/pack-types.d.ts.map +1 -0
- package/dist/skills/pack-types.js +7 -0
- package/dist/skills/pack-types.js.map +1 -0
- package/dist/skills/skill-registry.d.ts +83 -0
- package/dist/skills/skill-registry.d.ts.map +1 -0
- package/dist/skills/skill-registry.js +206 -0
- package/dist/skills/skill-registry.js.map +1 -0
- package/dist/skills/skill-runner.d.ts +49 -0
- package/dist/skills/skill-runner.d.ts.map +1 -0
- package/dist/skills/skill-runner.js +128 -0
- package/dist/skills/skill-runner.js.map +1 -0
- package/dist/skills/types.d.ts +94 -0
- package/dist/skills/types.d.ts.map +1 -0
- package/dist/skills/types.js +7 -0
- package/dist/skills/types.js.map +1 -0
- package/dist/task/concurrency.d.ts +87 -0
- package/dist/task/concurrency.d.ts.map +1 -0
- package/dist/task/concurrency.js +269 -0
- package/dist/task/concurrency.js.map +1 -0
- package/dist/task/task-evaluator.d.ts +52 -0
- package/dist/task/task-evaluator.d.ts.map +1 -0
- package/dist/task/task-evaluator.js +141 -0
- package/dist/task/task-evaluator.js.map +1 -0
- package/dist/task/task-manager.d.ts +112 -0
- package/dist/task/task-manager.d.ts.map +1 -0
- package/dist/task/task-manager.js +382 -0
- package/dist/task/task-manager.js.map +1 -0
- package/dist/task/task-state-machine.d.ts +63 -0
- package/dist/task/task-state-machine.d.ts.map +1 -0
- package/dist/task/task-state-machine.js +149 -0
- package/dist/task/task-state-machine.js.map +1 -0
- package/dist/types/agent.d.ts +79 -0
- package/dist/types/agent.d.ts.map +1 -0
- package/dist/types/agent.js +7 -0
- package/dist/types/agent.js.map +1 -0
- package/dist/types/index.d.ts +7 -0
- package/dist/types/index.d.ts.map +1 -0
- package/dist/types/index.js +5 -0
- package/dist/types/index.js.map +1 -0
- package/dist/types/message.d.ts +87 -0
- package/dist/types/message.d.ts.map +1 -0
- package/dist/types/message.js +30 -0
- package/dist/types/message.js.map +1 -0
- package/dist/types/task.d.ts +176 -0
- package/dist/types/task.d.ts.map +1 -0
- package/dist/types/task.js +5 -0
- package/dist/types/task.js.map +1 -0
- package/package.json +57 -0
- package/workerclaw.config.example.json +78 -0
|
@@ -0,0 +1,150 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* 安全门(Security Gate)
|
|
3
|
+
*
|
|
4
|
+
* 安全审查的总调度器,串联四层安全检查:
|
|
5
|
+
* Layer 1: 速率限制
|
|
6
|
+
* Layer 2: 来源验证
|
|
7
|
+
* Layer 3: 内容安全扫描
|
|
8
|
+
* Layer 4: 权限分级
|
|
9
|
+
*/
|
|
10
|
+
import { createLogger } from '../core/logger.js';
|
|
11
|
+
import { WorkerClawEvent } from '../core/events.js';
|
|
12
|
+
import { RateLimiter } from './rate-limiter.js';
|
|
13
|
+
import { SourceVerifier } from './source-verifier.js';
|
|
14
|
+
import { ContentScanner } from './content-scanner.js';
|
|
15
|
+
import { PermissionGrader } from './permission-level.js';
|
|
16
|
+
export class SecurityGate {
|
|
17
|
+
logger;
|
|
18
|
+
rateLimiter;
|
|
19
|
+
sourceVerifier;
|
|
20
|
+
contentScanner;
|
|
21
|
+
permissionGrader;
|
|
22
|
+
eventBus;
|
|
23
|
+
constructor(config, eventBus) {
|
|
24
|
+
this.rateLimiter = new RateLimiter(config.rateLimit);
|
|
25
|
+
this.sourceVerifier = new SourceVerifier(config.sourceVerify);
|
|
26
|
+
this.permissionGrader = new PermissionGrader({
|
|
27
|
+
highValueThreshold: 100, // 超过 100 元降一级
|
|
28
|
+
});
|
|
29
|
+
this.eventBus = eventBus;
|
|
30
|
+
this.logger = createLogger('SecurityGate');
|
|
31
|
+
// Layer 3: 内容扫描器
|
|
32
|
+
if (config.contentScan?.promptInjection?.enabled || config.contentScan?.maliciousCommands?.enabled) {
|
|
33
|
+
this.contentScanner = new ContentScanner({
|
|
34
|
+
promptInjection: config.contentScan.promptInjection,
|
|
35
|
+
maliciousCommands: config.contentScan.maliciousCommands,
|
|
36
|
+
piiProtection: config.contentScan.piiProtection || { enabled: false },
|
|
37
|
+
});
|
|
38
|
+
}
|
|
39
|
+
else {
|
|
40
|
+
this.contentScanner = null;
|
|
41
|
+
}
|
|
42
|
+
}
|
|
43
|
+
/**
|
|
44
|
+
* 对消息执行完整安全检查(Layer 1-2-3)
|
|
45
|
+
*/
|
|
46
|
+
async check(message) {
|
|
47
|
+
// Layer 1: 速率限制
|
|
48
|
+
const senderId = message.from || 'unknown';
|
|
49
|
+
const rateResult = this.rateLimiter.check(senderId);
|
|
50
|
+
if (!rateResult.allowed) {
|
|
51
|
+
this.eventBus.emit(WorkerClawEvent.SECURITY_BLOCKED, {
|
|
52
|
+
message: `速率限制触发`,
|
|
53
|
+
reason: rateResult.reason || 'rate limited',
|
|
54
|
+
data: { senderId },
|
|
55
|
+
});
|
|
56
|
+
this.logger.warn('消息被速率限制拦截', { senderId, reason: rateResult.reason || 'rate limited' });
|
|
57
|
+
return { passed: false, blockedBy: 'rate_limiter', reason: rateResult.reason || 'rate limited' };
|
|
58
|
+
}
|
|
59
|
+
// Layer 2: 来源验证
|
|
60
|
+
const sourceResult = this.sourceVerifier.verify(message);
|
|
61
|
+
if (!sourceResult.valid) {
|
|
62
|
+
this.eventBus.emit(WorkerClawEvent.SECURITY_BLOCKED, {
|
|
63
|
+
message: `来源验证失败`,
|
|
64
|
+
reason: sourceResult.reason || 'source verification failed',
|
|
65
|
+
data: { msgId: message.msgId },
|
|
66
|
+
});
|
|
67
|
+
this.logger.warn('消息来源验证失败', { reason: sourceResult.reason || 'unknown' });
|
|
68
|
+
return { passed: false, blockedBy: 'source_verifier', reason: sourceResult.reason || 'unknown' };
|
|
69
|
+
}
|
|
70
|
+
// Layer 3: 内容安全扫描(仅对任务推送消息执行)
|
|
71
|
+
if (this.contentScanner && message.data) {
|
|
72
|
+
const contentToScan = typeof message.data === 'string'
|
|
73
|
+
? message.data
|
|
74
|
+
: message.data.description || JSON.stringify(message.data);
|
|
75
|
+
const scanResult = this.contentScanner.scan(contentToScan);
|
|
76
|
+
if (!scanResult.safe) {
|
|
77
|
+
this.eventBus.emit(WorkerClawEvent.SECURITY_BLOCKED, {
|
|
78
|
+
message: `内容安全扫描未通过`,
|
|
79
|
+
reason: scanResult.rejectionReason || 'content scan failed',
|
|
80
|
+
data: { msgId: message.msgId, flags: scanResult.flags },
|
|
81
|
+
});
|
|
82
|
+
this.logger.warn('消息被内容扫描拦截', {
|
|
83
|
+
reason: scanResult.rejectionReason || 'unknown',
|
|
84
|
+
flagCount: scanResult.flags.length,
|
|
85
|
+
});
|
|
86
|
+
return {
|
|
87
|
+
passed: false,
|
|
88
|
+
blockedBy: 'content_scanner',
|
|
89
|
+
reason: scanResult.rejectionReason || 'content scan failed',
|
|
90
|
+
contentFlags: scanResult.flags,
|
|
91
|
+
};
|
|
92
|
+
}
|
|
93
|
+
// 低风险:记录警告但不拦截
|
|
94
|
+
if (scanResult.riskLevel !== 'none' && scanResult.flags.length > 0) {
|
|
95
|
+
this.eventBus.emit(WorkerClawEvent.SECURITY_WARNED, {
|
|
96
|
+
message: `内容扫描发现风险`,
|
|
97
|
+
reason: `${scanResult.riskLevel} 风险,${scanResult.flags.length} 个标记`,
|
|
98
|
+
data: { msgId: message.msgId, flags: scanResult.flags },
|
|
99
|
+
});
|
|
100
|
+
}
|
|
101
|
+
}
|
|
102
|
+
return { passed: true };
|
|
103
|
+
}
|
|
104
|
+
/**
|
|
105
|
+
* Layer 4: 为任务确定权限级别
|
|
106
|
+
*/
|
|
107
|
+
gradePermission(task) {
|
|
108
|
+
return this.permissionGrader.grade(task);
|
|
109
|
+
}
|
|
110
|
+
/**
|
|
111
|
+
* 获取权限分级器(供外部查询权限)
|
|
112
|
+
*/
|
|
113
|
+
getPermissionGrader() {
|
|
114
|
+
return this.permissionGrader;
|
|
115
|
+
}
|
|
116
|
+
/**
|
|
117
|
+
* 扫描 Agent 输出内容的安全性
|
|
118
|
+
*/
|
|
119
|
+
scanOutput(content) {
|
|
120
|
+
if (!this.contentScanner)
|
|
121
|
+
return { safe: true };
|
|
122
|
+
const result = this.contentScanner.scan(content);
|
|
123
|
+
return { safe: result.safe, reason: result.rejectionReason };
|
|
124
|
+
}
|
|
125
|
+
/**
|
|
126
|
+
* 检查任务容量
|
|
127
|
+
*/
|
|
128
|
+
checkTaskCapacity() {
|
|
129
|
+
return this.rateLimiter.checkTaskCapacity().allowed;
|
|
130
|
+
}
|
|
131
|
+
/**
|
|
132
|
+
* 通知任务开始
|
|
133
|
+
*/
|
|
134
|
+
taskStarted() {
|
|
135
|
+
this.rateLimiter.taskStarted();
|
|
136
|
+
}
|
|
137
|
+
/**
|
|
138
|
+
* 通知任务结束
|
|
139
|
+
*/
|
|
140
|
+
taskFinished() {
|
|
141
|
+
this.rateLimiter.taskFinished();
|
|
142
|
+
}
|
|
143
|
+
/**
|
|
144
|
+
* 获取速率限制器状态
|
|
145
|
+
*/
|
|
146
|
+
getRateLimitStatus() {
|
|
147
|
+
return this.rateLimiter.getStatus();
|
|
148
|
+
}
|
|
149
|
+
}
|
|
150
|
+
//# sourceMappingURL=gate.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"gate.js","sourceRoot":"","sources":["../../src/security/gate.ts"],"names":[],"mappings":"AAAA;;;;;;;;GAQG;AAEH,OAAO,EAAE,YAAY,EAAe,MAAM,mBAAmB,CAAC;AAC9D,OAAO,EAAY,eAAe,EAAE,MAAM,mBAAmB,CAAC;AAC9D,OAAO,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAChD,OAAO,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAC;AACtD,OAAO,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAC;AACtD,OAAO,EAAE,gBAAgB,EAAwB,MAAM,uBAAuB,CAAC;AAuB/E,MAAM,OAAO,YAAY;IACf,MAAM,CAAS;IACf,WAAW,CAAc;IACzB,cAAc,CAAiB;IAC/B,cAAc,CAAwB;IACtC,gBAAgB,CAAmB;IACnC,QAAQ,CAAW;IAE3B,YAAY,MAA0B,EAAE,QAAkB;QACxD,IAAI,CAAC,WAAW,GAAG,IAAI,WAAW,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;QACrD,IAAI,CAAC,cAAc,GAAG,IAAI,cAAc,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC;QAC9D,IAAI,CAAC,gBAAgB,GAAG,IAAI,gBAAgB,CAAC;YAC3C,kBAAkB,EAAE,GAAG,EAAE,cAAc;SACxC,CAAC,CAAC;QACH,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;QACzB,IAAI,CAAC,MAAM,GAAG,YAAY,CAAC,cAAc,CAAC,CAAC;QAE3C,iBAAiB;QACjB,IAAI,MAAM,CAAC,WAAW,EAAE,eAAe,EAAE,OAAO,IAAI,MAAM,CAAC,WAAW,EAAE,iBAAiB,EAAE,OAAO,EAAE,CAAC;YACnG,IAAI,CAAC,cAAc,GAAG,IAAI,cAAc,CAAC;gBACvC,eAAe,EAAE,MAAM,CAAC,WAAW,CAAC,eAAe;gBACnD,iBAAiB,EAAE,MAAM,CAAC,WAAW,CAAC,iBAAiB;gBACvD,aAAa,EAAE,MAAM,CAAC,WAAW,CAAC,aAAa,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE;aACtE,CAAC,CAAC;QACL,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC;QAC7B,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,KAAK,CAAC,OAAwB;QAClC,gBAAgB;QAChB,MAAM,QAAQ,GAAG,OAAO,CAAC,IAAI,IAAI,SAAS,CAAC;QAC3C,MAAM,UAAU,GAAG,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;QACpD,IAAI,CAAC,UAAU,CAAC,OAAO,EAAE,CAAC;YACxB,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,eAAe,CAAC,gBAAgB,EAAE;gBACnD,OAAO,EAAE,QAAQ;gBACjB,MAAM,EAAE,UAAU,CAAC,MAAM,IAAI,cAAc;gBAC3C,IAAI,EAAE,EAAE,QAAQ,EAAE;aACnB,CAAC,CAAC;YACH,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,WAAW,EAAE,EAAE,QAAQ,EAAE,MAAM,EAAE,UAAU,CAAC,MAAM,IAAI,cAAc,EAAE,CAAC,CAAC;YACzF,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,SAAS,EAAE,cAAc,EAAE,MAAM,EAAE,UAAU,CAAC,MAAM,IAAI,cAAc,EAAE,CAAC;QACnG,CAAC;QAED,gBAAgB;QAChB,MAAM,YAAY,GAAG,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;QACzD,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,CAAC;YACxB,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,eAAe,CAAC,gBAAgB,EAAE;gBACnD,OAAO,EAAE,QAAQ;gBACjB,MAAM,EAAE,YAAY,CAAC,MAAM,IAAI,4BAA4B;gBAC3D,IAAI,EAAE,EAAE,KAAK,EAAE,OAAO,CAAC,KAAK,EAAE;aAC/B,CAAC,CAAC;YACH,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,UAAU,EAAE,EAAE,MAAM,EAAE,YAAY,CAAC,MAAM,IAAI,SAAS,EAAE,CAAC,CAAC;YAC3E,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,SAAS,EAAE,iBAAiB,EAAE,MAAM,EAAE,YAAY,CAAC,MAAM,IAAI,SAAS,EAAE,CAAC;QACnG,CAAC;QAED,8BAA8B;QAC9B,IAAI,IAAI,CAAC,cAAc,IAAI,OAAO,CAAC,IAAI,EAAE,CAAC;YACxC,MAAM,aAAa,GAAG,OAAO,OAAO,CAAC,IAAI,KAAK,QAAQ;gBACpD,CAAC,CAAC,OAAO,CAAC,IAAI;gBACd,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,WAAW,IAAI,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;YAE7D,MAAM,UAAU,GAAG,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;YAE3D,IAAI,CAAC,UAAU,CAAC,IAAI,EAAE,CAAC;gBACrB,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,eAAe,CAAC,gBAAgB,EAAE;oBACnD,OAAO,EAAE,WAAW;oBACpB,MAAM,EAAE,UAAU,CAAC,eAAe,IAAI,qBAAqB;oBAC3D,IAAI,EAAE,EAAE,KAAK,EAAE,OAAO,CAAC,KAAK,EAAE,KAAK,EAAE,UAAU,CAAC,KAAK,EAAE;iBACxD,CAAC,CAAC;gBACH,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,WAAW,EAAE;oBAC5B,MAAM,EAAE,UAAU,CAAC,eAAe,IAAI,SAAS;oBAC/C,SAAS,EAAE,UAAU,CAAC,KAAK,CAAC,MAAM;iBACnC,CAAC,CAAC;gBACH,OAAO;oBACL,MAAM,EAAE,KAAK;oBACb,SAAS,EAAE,iBAAiB;oBAC5B,MAAM,EAAE,UAAU,CAAC,eAAe,IAAI,qBAAqB;oBAC3D,YAAY,EAAE,UAAU,CAAC,KAAK;iBAC/B,CAAC;YACJ,CAAC;YAED,eAAe;YACf,IAAI,UAAU,CAAC,SAAS,KAAK,MAAM,IAAI,UAAU,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACnE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,eAAe,CAAC,eAAe,EAAE;oBAClD,OAAO,EAAE,UAAU;oBACnB,MAAM,EAAE,GAAG,UAAU,CAAC,SAAS,OAAO,UAAU,CAAC,KAAK,CAAC,MAAM,MAAM;oBACnE,IAAI,EAAE,EAAE,KAAK,EAAE,OAAO,CAAC,KAAK,EAAE,KAAK,EAAE,UAAU,CAAC,KAAK,EAAE;iBACxD,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QAED,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC;IAC1B,CAAC;IAED;;OAEG;IACH,eAAe,CAAC,IAAU;QACxB,OAAO,IAAI,CAAC,gBAAgB,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;IAC3C,CAAC;IAED;;OAEG;IACH,mBAAmB;QACjB,OAAO,IAAI,CAAC,gBAAgB,CAAC;IAC/B,CAAC;IAED;;OAEG;IACH,UAAU,CAAC,OAAe;QACxB,IAAI,CAAC,IAAI,CAAC,cAAc;YAAE,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC;QAChD,MAAM,MAAM,GAAG,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACjD,OAAO,EAAE,IAAI,EAAE,MAAM,CAAC,IAAI,EAAE,MAAM,EAAE,MAAM,CAAC,eAAe,EAAE,CAAC;IAC/D,CAAC;IAED;;OAEG;IACH,iBAAiB;QACf,OAAO,IAAI,CAAC,WAAW,CAAC,iBAAiB,EAAE,CAAC,OAAO,CAAC;IACtD,CAAC;IAED;;OAEG;IACH,WAAW;QACT,IAAI,CAAC,WAAW,CAAC,WAAW,EAAE,CAAC;IACjC,CAAC;IAED;;OAEG;IACH,YAAY;QACV,IAAI,CAAC,WAAW,CAAC,YAAY,EAAE,CAAC;IAClC,CAAC;IAED;;OAEG;IACH,kBAAkB;QAChB,OAAO,IAAI,CAAC,WAAW,CAAC,SAAS,EAAE,CAAC;IACtC,CAAC;CACF"}
|
|
@@ -0,0 +1,68 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* 权限分级系统
|
|
3
|
+
*
|
|
4
|
+
* 根据任务类型自动确定权限级别
|
|
5
|
+
* 四级权限: read_only → limited → standard → elevated
|
|
6
|
+
*/
|
|
7
|
+
import type { Task } from '../types/task.js';
|
|
8
|
+
export type PermissionLevel = 'read_only' | 'limited' | 'standard' | 'elevated';
|
|
9
|
+
export interface PermissionProfile {
|
|
10
|
+
/** 允许的工具 */
|
|
11
|
+
allowedTools: string[];
|
|
12
|
+
/** 允许的命令(支持通配符 *,排除项以 - 开头) */
|
|
13
|
+
allowedCommands: string[];
|
|
14
|
+
/** 允许的网络协议和域名模式 */
|
|
15
|
+
allowedNetwork: string[];
|
|
16
|
+
/** 最大输出 token 数 */
|
|
17
|
+
maxOutputTokens: number;
|
|
18
|
+
/** 最大工具调用次数 */
|
|
19
|
+
maxToolCallsPerTask: number;
|
|
20
|
+
/** 允许的文件操作 */
|
|
21
|
+
allowedFileOps: ('read' | 'write' | 'delete')[];
|
|
22
|
+
}
|
|
23
|
+
export interface PermissionAutoGradeConfig {
|
|
24
|
+
/** 高信誉发单人可以升一级 */
|
|
25
|
+
highReputationBoost?: boolean;
|
|
26
|
+
/** 金额阈值(超过则降一级) */
|
|
27
|
+
highValueThreshold?: number;
|
|
28
|
+
}
|
|
29
|
+
export declare class PermissionGrader {
|
|
30
|
+
private logger;
|
|
31
|
+
private config;
|
|
32
|
+
constructor(config?: PermissionAutoGradeConfig);
|
|
33
|
+
/**
|
|
34
|
+
* 根据任务自动确定权限级别
|
|
35
|
+
*/
|
|
36
|
+
grade(task: Task): PermissionLevel;
|
|
37
|
+
/**
|
|
38
|
+
* 按任务类型分级
|
|
39
|
+
*/
|
|
40
|
+
private gradeByTaskType;
|
|
41
|
+
/**
|
|
42
|
+
* 降一级
|
|
43
|
+
*/
|
|
44
|
+
private demote;
|
|
45
|
+
/**
|
|
46
|
+
* 获取权限配置
|
|
47
|
+
*/
|
|
48
|
+
getProfile(level: PermissionLevel): PermissionProfile;
|
|
49
|
+
/**
|
|
50
|
+
* 检查工具是否允许
|
|
51
|
+
*/
|
|
52
|
+
isToolAllowed(toolName: string, level: PermissionLevel): boolean;
|
|
53
|
+
/**
|
|
54
|
+
* 检查命令是否允许
|
|
55
|
+
*/
|
|
56
|
+
isCommandAllowed(command: string, level: PermissionLevel): boolean;
|
|
57
|
+
/**
|
|
58
|
+
* 检查网络访问是否允许
|
|
59
|
+
*/
|
|
60
|
+
isNetworkAllowed(url: string, level: PermissionLevel): boolean;
|
|
61
|
+
/**
|
|
62
|
+
* 检查文件操作是否允许
|
|
63
|
+
*/
|
|
64
|
+
isFileOpAllowed(op: 'read' | 'write' | 'delete', level: PermissionLevel): boolean;
|
|
65
|
+
}
|
|
66
|
+
/** 全局权限分级器单例 */
|
|
67
|
+
export declare const permissionGrader: PermissionGrader;
|
|
68
|
+
//# sourceMappingURL=permission-level.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"permission-level.d.ts","sourceRoot":"","sources":["../../src/security/permission-level.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAGH,OAAO,KAAK,EAAE,IAAI,EAAY,MAAM,kBAAkB,CAAC;AAIvD,MAAM,MAAM,eAAe,GAAG,WAAW,GAAG,SAAS,GAAG,UAAU,GAAG,UAAU,CAAC;AAIhF,MAAM,WAAW,iBAAiB;IAChC,YAAY;IACZ,YAAY,EAAE,MAAM,EAAE,CAAC;IACvB,+BAA+B;IAC/B,eAAe,EAAE,MAAM,EAAE,CAAC;IAC1B,mBAAmB;IACnB,cAAc,EAAE,MAAM,EAAE,CAAC;IACzB,mBAAmB;IACnB,eAAe,EAAE,MAAM,CAAC;IACxB,eAAe;IACf,mBAAmB,EAAE,MAAM,CAAC;IAC5B,cAAc;IACd,cAAc,EAAE,CAAC,MAAM,GAAG,OAAO,GAAG,QAAQ,CAAC,EAAE,CAAC;CACjD;AAqED,MAAM,WAAW,yBAAyB;IACxC,kBAAkB;IAClB,mBAAmB,CAAC,EAAE,OAAO,CAAC;IAC9B,mBAAmB;IACnB,kBAAkB,CAAC,EAAE,MAAM,CAAC;CAC7B;AAID,qBAAa,gBAAgB;IAC3B,OAAO,CAAC,MAAM,CAAS;IACvB,OAAO,CAAC,MAAM,CAA4B;gBAE9B,MAAM,GAAE,yBAA8B;IAKlD;;OAEG;IACH,KAAK,CAAC,IAAI,EAAE,IAAI,GAAG,eAAe;IAwBlC;;OAEG;IACH,OAAO,CAAC,eAAe;IAcvB;;OAEG;IACH,OAAO,CAAC,MAAM;IAOd;;OAEG;IACH,UAAU,CAAC,KAAK,EAAE,eAAe,GAAG,iBAAiB;IAIrD;;OAEG;IACH,aAAa,CAAC,QAAQ,EAAE,MAAM,EAAE,KAAK,EAAE,eAAe,GAAG,OAAO;IAMhE;;OAEG;IACH,gBAAgB,CAAC,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,eAAe,GAAG,OAAO;IAiBlE;;OAEG;IACH,gBAAgB,CAAC,GAAG,EAAE,MAAM,EAAE,KAAK,EAAE,eAAe,GAAG,OAAO;IAsB9D;;OAEG;IACH,eAAe,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,GAAG,QAAQ,EAAE,KAAK,EAAE,eAAe,GAAG,OAAO;CAIlF;AAED,gBAAgB;AAChB,eAAO,MAAM,gBAAgB,kBAAyB,CAAC"}
|
|
@@ -0,0 +1,191 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* 权限分级系统
|
|
3
|
+
*
|
|
4
|
+
* 根据任务类型自动确定权限级别
|
|
5
|
+
* 四级权限: read_only → limited → standard → elevated
|
|
6
|
+
*/
|
|
7
|
+
import { createLogger } from '../core/logger.js';
|
|
8
|
+
// ==================== 各级别默认权限 ====================
|
|
9
|
+
const PERMISSION_LEVELS = {
|
|
10
|
+
read_only: {
|
|
11
|
+
allowedTools: ['llm_query'],
|
|
12
|
+
allowedCommands: [],
|
|
13
|
+
allowedNetwork: [],
|
|
14
|
+
maxOutputTokens: 2000,
|
|
15
|
+
maxToolCallsPerTask: 3,
|
|
16
|
+
allowedFileOps: ['read'],
|
|
17
|
+
},
|
|
18
|
+
limited: {
|
|
19
|
+
allowedTools: ['llm_query', 'web_search'],
|
|
20
|
+
allowedCommands: ['echo', 'cat', 'ls', 'head', 'tail', 'wc', 'grep', 'find'],
|
|
21
|
+
allowedNetwork: ['https://api.miniabc.top/*'],
|
|
22
|
+
maxOutputTokens: 4000,
|
|
23
|
+
maxToolCallsPerTask: 10,
|
|
24
|
+
allowedFileOps: ['read'],
|
|
25
|
+
},
|
|
26
|
+
standard: {
|
|
27
|
+
allowedTools: ['llm_query', 'web_search', 'image_gen', 'file_read', 'file_write'],
|
|
28
|
+
allowedCommands: ['*'],
|
|
29
|
+
allowedNetwork: ['https://*'],
|
|
30
|
+
maxOutputTokens: 8000,
|
|
31
|
+
maxToolCallsPerTask: 20,
|
|
32
|
+
allowedFileOps: ['read', 'write'],
|
|
33
|
+
},
|
|
34
|
+
elevated: {
|
|
35
|
+
allowedTools: ['*'],
|
|
36
|
+
allowedCommands: ['*'],
|
|
37
|
+
allowedNetwork: ['https://*', 'wss://*'],
|
|
38
|
+
maxOutputTokens: 16000,
|
|
39
|
+
maxToolCallsPerTask: 50,
|
|
40
|
+
allowedFileOps: ['read', 'write', 'delete'],
|
|
41
|
+
},
|
|
42
|
+
};
|
|
43
|
+
// ==================== 任务类型→权限映射 ====================
|
|
44
|
+
const TASK_TYPE_MAPPING = {
|
|
45
|
+
text_reply: 'read_only',
|
|
46
|
+
qa: 'read_only',
|
|
47
|
+
translation: 'limited',
|
|
48
|
+
search_summary: 'limited',
|
|
49
|
+
writing: 'standard',
|
|
50
|
+
image_gen: 'standard',
|
|
51
|
+
data_analysis: 'standard',
|
|
52
|
+
code_dev: 'elevated',
|
|
53
|
+
system_op: 'elevated',
|
|
54
|
+
other: 'standard',
|
|
55
|
+
};
|
|
56
|
+
// 中文任务类型映射
|
|
57
|
+
const CN_TASK_TYPE_MAPPING = {
|
|
58
|
+
'文字回复': 'read_only',
|
|
59
|
+
'问答': 'read_only',
|
|
60
|
+
'搜索整理': 'limited',
|
|
61
|
+
'翻译': 'limited',
|
|
62
|
+
'写文章': 'standard',
|
|
63
|
+
'生成图片': 'standard',
|
|
64
|
+
'数据分析': 'standard',
|
|
65
|
+
'代码开发': 'elevated',
|
|
66
|
+
'系统操作': 'elevated',
|
|
67
|
+
};
|
|
68
|
+
// ==================== 权限分级器 ====================
|
|
69
|
+
export class PermissionGrader {
|
|
70
|
+
logger;
|
|
71
|
+
config;
|
|
72
|
+
constructor(config = {}) {
|
|
73
|
+
this.config = config;
|
|
74
|
+
this.logger = createLogger('PermissionGrader');
|
|
75
|
+
}
|
|
76
|
+
/**
|
|
77
|
+
* 根据任务自动确定权限级别
|
|
78
|
+
*/
|
|
79
|
+
grade(task) {
|
|
80
|
+
// 1. 基础分级(根据任务类型)
|
|
81
|
+
let level = this.gradeByTaskType(task.taskType, task.title);
|
|
82
|
+
// 2. 金额调整(金额越大越谨慎)
|
|
83
|
+
if (this.config.highValueThreshold && task.reward) {
|
|
84
|
+
if (task.reward > this.config.highValueThreshold) {
|
|
85
|
+
const demoted = this.demote(level);
|
|
86
|
+
this.logger.info(`任务金额 ${task.reward} 超过阈值,权限从 ${level} 降为 ${demoted}`);
|
|
87
|
+
level = demoted;
|
|
88
|
+
}
|
|
89
|
+
}
|
|
90
|
+
// 3. 信誉调整(暂无信誉数据,预留接口)
|
|
91
|
+
// TODO: 集成平台信誉系统后实现
|
|
92
|
+
this.logger.debug(`任务 [${task.taskId}] 权限级别: ${level}`, {
|
|
93
|
+
taskType: task.taskType,
|
|
94
|
+
reward: task.reward,
|
|
95
|
+
});
|
|
96
|
+
return level;
|
|
97
|
+
}
|
|
98
|
+
/**
|
|
99
|
+
* 按任务类型分级
|
|
100
|
+
*/
|
|
101
|
+
gradeByTaskType(taskType, title) {
|
|
102
|
+
// 先查英文映射
|
|
103
|
+
if (TASK_TYPE_MAPPING[taskType]) {
|
|
104
|
+
return TASK_TYPE_MAPPING[taskType];
|
|
105
|
+
}
|
|
106
|
+
// 查中文映射
|
|
107
|
+
if (title && CN_TASK_TYPE_MAPPING[title]) {
|
|
108
|
+
return CN_TASK_TYPE_MAPPING[title];
|
|
109
|
+
}
|
|
110
|
+
return 'standard';
|
|
111
|
+
}
|
|
112
|
+
/**
|
|
113
|
+
* 降一级
|
|
114
|
+
*/
|
|
115
|
+
demote(level) {
|
|
116
|
+
const order = ['read_only', 'limited', 'standard', 'elevated'];
|
|
117
|
+
const idx = order.indexOf(level);
|
|
118
|
+
if (idx <= 0)
|
|
119
|
+
return level;
|
|
120
|
+
return order[idx - 1];
|
|
121
|
+
}
|
|
122
|
+
/**
|
|
123
|
+
* 获取权限配置
|
|
124
|
+
*/
|
|
125
|
+
getProfile(level) {
|
|
126
|
+
return PERMISSION_LEVELS[level];
|
|
127
|
+
}
|
|
128
|
+
/**
|
|
129
|
+
* 检查工具是否允许
|
|
130
|
+
*/
|
|
131
|
+
isToolAllowed(toolName, level) {
|
|
132
|
+
const profile = this.getProfile(level);
|
|
133
|
+
if (profile.allowedTools.includes('*'))
|
|
134
|
+
return true;
|
|
135
|
+
return profile.allowedTools.includes(toolName);
|
|
136
|
+
}
|
|
137
|
+
/**
|
|
138
|
+
* 检查命令是否允许
|
|
139
|
+
*/
|
|
140
|
+
isCommandAllowed(command, level) {
|
|
141
|
+
const profile = this.getProfile(level);
|
|
142
|
+
if (profile.allowedCommands.length === 0)
|
|
143
|
+
return false;
|
|
144
|
+
if (profile.allowedCommands.includes('*')) {
|
|
145
|
+
// 检查排除项
|
|
146
|
+
const cmdBase = command.trim().split(/\s+/)[0];
|
|
147
|
+
for (const pattern of profile.allowedCommands) {
|
|
148
|
+
if (pattern.startsWith('-') && command.match(new RegExp(pattern.slice(1).replace(/\*/g, '.*'), 'i'))) {
|
|
149
|
+
return false;
|
|
150
|
+
}
|
|
151
|
+
}
|
|
152
|
+
return true;
|
|
153
|
+
}
|
|
154
|
+
const cmdBase = command.trim().split(/\s+/)[0];
|
|
155
|
+
return profile.allowedCommands.includes(cmdBase);
|
|
156
|
+
}
|
|
157
|
+
/**
|
|
158
|
+
* 检查网络访问是否允许
|
|
159
|
+
*/
|
|
160
|
+
isNetworkAllowed(url, level) {
|
|
161
|
+
const profile = this.getProfile(level);
|
|
162
|
+
if (profile.allowedNetwork.length === 0)
|
|
163
|
+
return false;
|
|
164
|
+
if (profile.allowedNetwork.includes('*'))
|
|
165
|
+
return true;
|
|
166
|
+
try {
|
|
167
|
+
const parsed = new URL(url);
|
|
168
|
+
const urlOrigin = `${parsed.protocol}//${parsed.hostname}`;
|
|
169
|
+
for (const pattern of profile.allowedNetwork) {
|
|
170
|
+
const regex = new RegExp('^' + pattern.replace(/\*/g, '.*') + '$');
|
|
171
|
+
if (regex.test(urlOrigin) || regex.test(`${parsed.protocol}//${parsed.hostname}/*`)) {
|
|
172
|
+
return true;
|
|
173
|
+
}
|
|
174
|
+
}
|
|
175
|
+
return false;
|
|
176
|
+
}
|
|
177
|
+
catch {
|
|
178
|
+
return false;
|
|
179
|
+
}
|
|
180
|
+
}
|
|
181
|
+
/**
|
|
182
|
+
* 检查文件操作是否允许
|
|
183
|
+
*/
|
|
184
|
+
isFileOpAllowed(op, level) {
|
|
185
|
+
const profile = this.getProfile(level);
|
|
186
|
+
return profile.allowedFileOps.includes(op);
|
|
187
|
+
}
|
|
188
|
+
}
|
|
189
|
+
/** 全局权限分级器单例 */
|
|
190
|
+
export const permissionGrader = new PermissionGrader();
|
|
191
|
+
//# sourceMappingURL=permission-level.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"permission-level.js","sourceRoot":"","sources":["../../src/security/permission-level.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,EAAE,YAAY,EAAe,MAAM,mBAAmB,CAAC;AAwB9D,oDAAoD;AAEpD,MAAM,iBAAiB,GAA+C;IACpE,SAAS,EAAE;QACT,YAAY,EAAE,CAAC,WAAW,CAAC;QAC3B,eAAe,EAAE,EAAE;QACnB,cAAc,EAAE,EAAE;QAClB,eAAe,EAAE,IAAI;QACrB,mBAAmB,EAAE,CAAC;QACtB,cAAc,EAAE,CAAC,MAAM,CAAC;KACzB;IACD,OAAO,EAAE;QACP,YAAY,EAAE,CAAC,WAAW,EAAE,YAAY,CAAC;QACzC,eAAe,EAAE,CAAC,MAAM,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,CAAC;QAC5E,cAAc,EAAE,CAAC,2BAA2B,CAAC;QAC7C,eAAe,EAAE,IAAI;QACrB,mBAAmB,EAAE,EAAE;QACvB,cAAc,EAAE,CAAC,MAAM,CAAC;KACzB;IACD,QAAQ,EAAE;QACR,YAAY,EAAE,CAAC,WAAW,EAAE,YAAY,EAAE,WAAW,EAAE,WAAW,EAAE,YAAY,CAAC;QACjF,eAAe,EAAE,CAAC,GAAG,CAAC;QACtB,cAAc,EAAE,CAAC,WAAW,CAAC;QAC7B,eAAe,EAAE,IAAI;QACrB,mBAAmB,EAAE,EAAE;QACvB,cAAc,EAAE,CAAC,MAAM,EAAE,OAAO,CAAC;KAClC;IACD,QAAQ,EAAE;QACR,YAAY,EAAE,CAAC,GAAG,CAAC;QACnB,eAAe,EAAE,CAAC,GAAG,CAAC;QACtB,cAAc,EAAE,CAAC,WAAW,EAAE,SAAS,CAAC;QACxC,eAAe,EAAE,KAAK;QACtB,mBAAmB,EAAE,EAAE;QACvB,cAAc,EAAE,CAAC,MAAM,EAAE,OAAO,EAAE,QAAQ,CAAC;KAC5C;CACF,CAAC;AAEF,sDAAsD;AAEtD,MAAM,iBAAiB,GAAsC;IAC3D,UAAU,EAAE,WAAW;IACvB,EAAE,EAAE,WAAW;IACf,WAAW,EAAE,SAAS;IACtB,cAAc,EAAE,SAAS;IACzB,OAAO,EAAE,UAAU;IACnB,SAAS,EAAE,UAAU;IACrB,aAAa,EAAE,UAAU;IACzB,QAAQ,EAAE,UAAU;IACpB,SAAS,EAAE,UAAU;IACrB,KAAK,EAAE,UAAU;CAClB,CAAC;AAEF,WAAW;AACX,MAAM,oBAAoB,GAAoC;IAC5D,MAAM,EAAE,WAAW;IACnB,IAAI,EAAE,WAAW;IACjB,MAAM,EAAE,SAAS;IACjB,IAAI,EAAE,SAAS;IACf,KAAK,EAAE,UAAU;IACjB,MAAM,EAAE,UAAU;IAClB,MAAM,EAAE,UAAU;IAClB,MAAM,EAAE,UAAU;IAClB,MAAM,EAAE,UAAU;CACnB,CAAC;AAWF,kDAAkD;AAElD,MAAM,OAAO,gBAAgB;IACnB,MAAM,CAAS;IACf,MAAM,CAA4B;IAE1C,YAAY,SAAoC,EAAE;QAChD,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,MAAM,GAAG,YAAY,CAAC,kBAAkB,CAAC,CAAC;IACjD,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,IAAU;QACd,kBAAkB;QAClB,IAAI,KAAK,GAAG,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,QAAQ,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC;QAE5D,mBAAmB;QACnB,IAAI,IAAI,CAAC,MAAM,CAAC,kBAAkB,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YAClD,IAAI,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,kBAAkB,EAAE,CAAC;gBACjD,MAAM,OAAO,GAAG,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;gBACnC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,QAAQ,IAAI,CAAC,MAAM,aAAa,KAAK,OAAO,OAAO,EAAE,CAAC,CAAC;gBACxE,KAAK,GAAG,OAAO,CAAC;YAClB,CAAC;QACH,CAAC;QAED,uBAAuB;QACvB,oBAAoB;QAEpB,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,OAAO,IAAI,CAAC,MAAM,WAAW,KAAK,EAAE,EAAE;YACtD,QAAQ,EAAE,IAAI,CAAC,QAAQ;YACvB,MAAM,EAAE,IAAI,CAAC,MAAM;SACpB,CAAC,CAAC;QAEH,OAAO,KAAK,CAAC;IACf,CAAC;IAED;;OAEG;IACK,eAAe,CAAC,QAAkB,EAAE,KAAc;QACxD,SAAS;QACT,IAAI,iBAAiB,CAAC,QAAQ,CAAC,EAAE,CAAC;YAChC,OAAO,iBAAiB,CAAC,QAAQ,CAAC,CAAC;QACrC,CAAC;QAED,QAAQ;QACR,IAAI,KAAK,IAAI,oBAAoB,CAAC,KAAK,CAAC,EAAE,CAAC;YACzC,OAAO,oBAAoB,CAAC,KAAK,CAAC,CAAC;QACrC,CAAC;QAED,OAAO,UAAU,CAAC;IACpB,CAAC;IAED;;OAEG;IACK,MAAM,CAAC,KAAsB;QACnC,MAAM,KAAK,GAAsB,CAAC,WAAW,EAAE,SAAS,EAAE,UAAU,EAAE,UAAU,CAAC,CAAC;QAClF,MAAM,GAAG,GAAG,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QACjC,IAAI,GAAG,IAAI,CAAC;YAAE,OAAO,KAAK,CAAC;QAC3B,OAAO,KAAK,CAAC,GAAG,GAAG,CAAC,CAAC,CAAC;IACxB,CAAC;IAED;;OAEG;IACH,UAAU,CAAC,KAAsB;QAC/B,OAAO,iBAAiB,CAAC,KAAK,CAAC,CAAC;IAClC,CAAC;IAED;;OAEG;IACH,aAAa,CAAC,QAAgB,EAAE,KAAsB;QACpD,MAAM,OAAO,GAAG,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;QACvC,IAAI,OAAO,CAAC,YAAY,CAAC,QAAQ,CAAC,GAAG,CAAC;YAAE,OAAO,IAAI,CAAC;QACpD,OAAO,OAAO,CAAC,YAAY,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;IACjD,CAAC;IAED;;OAEG;IACH,gBAAgB,CAAC,OAAe,EAAE,KAAsB;QACtD,MAAM,OAAO,GAAG,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;QACvC,IAAI,OAAO,CAAC,eAAe,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,KAAK,CAAC;QACvD,IAAI,OAAO,CAAC,eAAe,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC;YAC1C,QAAQ;YACR,MAAM,OAAO,GAAG,OAAO,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;YAC/C,KAAK,MAAM,OAAO,IAAI,OAAO,CAAC,eAAe,EAAE,CAAC;gBAC9C,IAAI,OAAO,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,OAAO,CAAC,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,KAAK,EAAE,IAAI,CAAC,EAAE,GAAG,CAAC,CAAC,EAAE,CAAC;oBACrG,OAAO,KAAK,CAAC;gBACf,CAAC;YACH,CAAC;YACD,OAAO,IAAI,CAAC;QACd,CAAC;QACD,MAAM,OAAO,GAAG,OAAO,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;QAC/C,OAAO,OAAO,CAAC,eAAe,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;IACnD,CAAC;IAED;;OAEG;IACH,gBAAgB,CAAC,GAAW,EAAE,KAAsB;QAClD,MAAM,OAAO,GAAG,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;QACvC,IAAI,OAAO,CAAC,cAAc,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,KAAK,CAAC;QACtD,IAAI,OAAO,CAAC,cAAc,CAAC,QAAQ,CAAC,GAAG,CAAC;YAAE,OAAO,IAAI,CAAC;QAEtD,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,IAAI,GAAG,CAAC,GAAG,CAAC,CAAC;YAC5B,MAAM,SAAS,GAAG,GAAG,MAAM,CAAC,QAAQ,KAAK,MAAM,CAAC,QAAQ,EAAE,CAAC;YAE3D,KAAK,MAAM,OAAO,IAAI,OAAO,CAAC,cAAc,EAAE,CAAC;gBAC7C,MAAM,KAAK,GAAG,IAAI,MAAM,CAAC,GAAG,GAAG,OAAO,CAAC,OAAO,CAAC,KAAK,EAAE,IAAI,CAAC,GAAG,GAAG,CAAC,CAAC;gBACnE,IAAI,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,KAAK,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC,QAAQ,KAAK,MAAM,CAAC,QAAQ,IAAI,CAAC,EAAE,CAAC;oBACpF,OAAO,IAAI,CAAC;gBACd,CAAC;YACH,CAAC;YAED,OAAO,KAAK,CAAC;QACf,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,KAAK,CAAC;QACf,CAAC;IACH,CAAC;IAED;;OAEG;IACH,eAAe,CAAC,EAA+B,EAAE,KAAsB;QACrE,MAAM,OAAO,GAAG,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;QACvC,OAAO,OAAO,CAAC,cAAc,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;IAC7C,CAAC;CACF;AAED,gBAAgB;AAChB,MAAM,CAAC,MAAM,gBAAgB,GAAG,IAAI,gBAAgB,EAAE,CAAC"}
|
|
@@ -0,0 +1,52 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* 速率限制器
|
|
3
|
+
*
|
|
4
|
+
* 使用滑动窗口计数器实现,防止消息洪泛
|
|
5
|
+
*/
|
|
6
|
+
import type { RateLimitConfig } from '../core/config.js';
|
|
7
|
+
export interface RateLimitResult {
|
|
8
|
+
allowed: boolean;
|
|
9
|
+
reason?: string;
|
|
10
|
+
retryAfterMs?: number;
|
|
11
|
+
}
|
|
12
|
+
export declare class RateLimiter {
|
|
13
|
+
private logger;
|
|
14
|
+
private config;
|
|
15
|
+
private senderCounters;
|
|
16
|
+
private globalCounter;
|
|
17
|
+
private runningTasks;
|
|
18
|
+
constructor(config: RateLimitConfig);
|
|
19
|
+
/**
|
|
20
|
+
* 检查消息是否允许通过
|
|
21
|
+
*/
|
|
22
|
+
check(senderId: string): RateLimitResult;
|
|
23
|
+
/**
|
|
24
|
+
* 检查是否可以接受新任务
|
|
25
|
+
*/
|
|
26
|
+
checkTaskCapacity(): RateLimitResult;
|
|
27
|
+
/**
|
|
28
|
+
* 增加运行中任务数
|
|
29
|
+
*/
|
|
30
|
+
taskStarted(): void;
|
|
31
|
+
/**
|
|
32
|
+
* 减少运行中任务数
|
|
33
|
+
*/
|
|
34
|
+
taskFinished(): void;
|
|
35
|
+
/**
|
|
36
|
+
* 获取当前状态
|
|
37
|
+
*/
|
|
38
|
+
getStatus(): {
|
|
39
|
+
runningTasks: number;
|
|
40
|
+
maxConcurrent: number;
|
|
41
|
+
};
|
|
42
|
+
/**
|
|
43
|
+
* 重置所有计数器
|
|
44
|
+
*/
|
|
45
|
+
reset(): void;
|
|
46
|
+
private checkGlobal;
|
|
47
|
+
private checkSender;
|
|
48
|
+
private recordMessage;
|
|
49
|
+
private cleanWindow;
|
|
50
|
+
private getRetryAfter;
|
|
51
|
+
}
|
|
52
|
+
//# sourceMappingURL=rate-limiter.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"rate-limiter.d.ts","sourceRoot":"","sources":["../../src/security/rate-limiter.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAGH,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,mBAAmB,CAAC;AAEzD,MAAM,WAAW,eAAe;IAC9B,OAAO,EAAE,OAAO,CAAC;IACjB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,YAAY,CAAC,EAAE,MAAM,CAAC;CACvB;AAMD,qBAAa,WAAW;IACtB,OAAO,CAAC,MAAM,CAAS;IACvB,OAAO,CAAC,MAAM,CAAkB;IAGhC,OAAO,CAAC,cAAc,CAA2C;IAEjE,OAAO,CAAC,aAAa,CAA4C;IAEjE,OAAO,CAAC,YAAY,CAAK;gBAEb,MAAM,EAAE,eAAe;IAKnC;;OAEG;IACH,KAAK,CAAC,QAAQ,EAAE,MAAM,GAAG,eAAe;IAgBxC;;OAEG;IACH,iBAAiB,IAAI,eAAe;IAcpC;;OAEG;IACH,WAAW,IAAI,IAAI;IAKnB;;OAEG;IACH,YAAY,IAAI,IAAI;IAKpB;;OAEG;IACH,SAAS,IAAI;QAAE,YAAY,EAAE,MAAM,CAAC;QAAC,aAAa,EAAE,MAAM,CAAA;KAAE;IAO5D;;OAEG;IACH,KAAK,IAAI,IAAI;IAQb,OAAO,CAAC,WAAW;IAcnB,OAAO,CAAC,WAAW;IAoBnB,OAAO,CAAC,aAAa;IAWrB,OAAO,CAAC,WAAW;IAKnB,OAAO,CAAC,aAAa;CAMtB"}
|
|
@@ -0,0 +1,133 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* 速率限制器
|
|
3
|
+
*
|
|
4
|
+
* 使用滑动窗口计数器实现,防止消息洪泛
|
|
5
|
+
*/
|
|
6
|
+
import { createLogger } from '../core/logger.js';
|
|
7
|
+
export class RateLimiter {
|
|
8
|
+
logger;
|
|
9
|
+
config;
|
|
10
|
+
// 每个发送者的计数器
|
|
11
|
+
senderCounters = new Map();
|
|
12
|
+
// 全局计数器
|
|
13
|
+
globalCounter = { timestamps: [] };
|
|
14
|
+
// 当前运行中的任务数
|
|
15
|
+
runningTasks = 0;
|
|
16
|
+
constructor(config) {
|
|
17
|
+
this.config = config;
|
|
18
|
+
this.logger = createLogger('RateLimiter');
|
|
19
|
+
}
|
|
20
|
+
/**
|
|
21
|
+
* 检查消息是否允许通过
|
|
22
|
+
*/
|
|
23
|
+
check(senderId) {
|
|
24
|
+
const now = Date.now();
|
|
25
|
+
// 1. 检查全局速率
|
|
26
|
+
const globalResult = this.checkGlobal(now);
|
|
27
|
+
if (!globalResult.allowed)
|
|
28
|
+
return globalResult;
|
|
29
|
+
// 2. 检查发送者速率
|
|
30
|
+
const senderResult = this.checkSender(senderId, now);
|
|
31
|
+
if (!senderResult.allowed)
|
|
32
|
+
return senderResult;
|
|
33
|
+
// 记录
|
|
34
|
+
this.recordMessage(senderId, now);
|
|
35
|
+
return { allowed: true };
|
|
36
|
+
}
|
|
37
|
+
/**
|
|
38
|
+
* 检查是否可以接受新任务
|
|
39
|
+
*/
|
|
40
|
+
checkTaskCapacity() {
|
|
41
|
+
if (this.runningTasks >= this.config.maxConcurrentTasks) {
|
|
42
|
+
this.logger.warn('达到最大并发任务数', {
|
|
43
|
+
running: this.runningTasks,
|
|
44
|
+
max: this.config.maxConcurrentTasks,
|
|
45
|
+
});
|
|
46
|
+
return {
|
|
47
|
+
allowed: false,
|
|
48
|
+
reason: `并发任务数已达上限 (${this.runningTasks}/${this.config.maxConcurrentTasks})`,
|
|
49
|
+
};
|
|
50
|
+
}
|
|
51
|
+
return { allowed: true };
|
|
52
|
+
}
|
|
53
|
+
/**
|
|
54
|
+
* 增加运行中任务数
|
|
55
|
+
*/
|
|
56
|
+
taskStarted() {
|
|
57
|
+
this.runningTasks++;
|
|
58
|
+
this.logger.debug(`任务开始,当前并发: ${this.runningTasks}`);
|
|
59
|
+
}
|
|
60
|
+
/**
|
|
61
|
+
* 减少运行中任务数
|
|
62
|
+
*/
|
|
63
|
+
taskFinished() {
|
|
64
|
+
this.runningTasks = Math.max(0, this.runningTasks - 1);
|
|
65
|
+
this.logger.debug(`任务结束,当前并发: ${this.runningTasks}`);
|
|
66
|
+
}
|
|
67
|
+
/**
|
|
68
|
+
* 获取当前状态
|
|
69
|
+
*/
|
|
70
|
+
getStatus() {
|
|
71
|
+
return {
|
|
72
|
+
runningTasks: this.runningTasks,
|
|
73
|
+
maxConcurrent: this.config.maxConcurrentTasks,
|
|
74
|
+
};
|
|
75
|
+
}
|
|
76
|
+
/**
|
|
77
|
+
* 重置所有计数器
|
|
78
|
+
*/
|
|
79
|
+
reset() {
|
|
80
|
+
this.senderCounters.clear();
|
|
81
|
+
this.globalCounter = { timestamps: [] };
|
|
82
|
+
this.runningTasks = 0;
|
|
83
|
+
}
|
|
84
|
+
// ==================== 私有方法 ====================
|
|
85
|
+
checkGlobal(now) {
|
|
86
|
+
this.cleanWindow(this.globalCounter, now, 60_000);
|
|
87
|
+
if (this.globalCounter.timestamps.length >= this.config.maxMessagesPerMinute * 2) {
|
|
88
|
+
return {
|
|
89
|
+
allowed: false,
|
|
90
|
+
reason: `全局消息速率超限`,
|
|
91
|
+
retryAfterMs: this.getRetryAfter(this.globalCounter.timestamps, now, 60_000),
|
|
92
|
+
};
|
|
93
|
+
}
|
|
94
|
+
return { allowed: true };
|
|
95
|
+
}
|
|
96
|
+
checkSender(senderId, now) {
|
|
97
|
+
if (!this.senderCounters.has(senderId)) {
|
|
98
|
+
this.senderCounters.set(senderId, { timestamps: [] });
|
|
99
|
+
}
|
|
100
|
+
const counter = this.senderCounters.get(senderId);
|
|
101
|
+
this.cleanWindow(counter, now, 60_000);
|
|
102
|
+
if (counter.timestamps.length >= this.config.maxMessagesPerMinute) {
|
|
103
|
+
this.logger.warn(`发送者 ${senderId} 消息速率超限`);
|
|
104
|
+
return {
|
|
105
|
+
allowed: false,
|
|
106
|
+
reason: `该发送者消息速率超限 (${counter.timestamps.length}/min)`,
|
|
107
|
+
retryAfterMs: this.getRetryAfter(counter.timestamps, now, 60_000),
|
|
108
|
+
};
|
|
109
|
+
}
|
|
110
|
+
return { allowed: true };
|
|
111
|
+
}
|
|
112
|
+
recordMessage(senderId, now) {
|
|
113
|
+
// 全局记录
|
|
114
|
+
this.globalCounter.timestamps.push(now);
|
|
115
|
+
// 发送者记录
|
|
116
|
+
if (!this.senderCounters.has(senderId)) {
|
|
117
|
+
this.senderCounters.set(senderId, { timestamps: [] });
|
|
118
|
+
}
|
|
119
|
+
this.senderCounters.get(senderId).timestamps.push(now);
|
|
120
|
+
}
|
|
121
|
+
cleanWindow(counter, now, windowMs) {
|
|
122
|
+
const cutoff = now - windowMs;
|
|
123
|
+
counter.timestamps = counter.timestamps.filter(ts => ts > cutoff);
|
|
124
|
+
}
|
|
125
|
+
getRetryAfter(timestamps, now, windowMs) {
|
|
126
|
+
if (timestamps.length === 0)
|
|
127
|
+
return 0;
|
|
128
|
+
const oldest = timestamps[0];
|
|
129
|
+
const remaining = windowMs - (now - oldest);
|
|
130
|
+
return Math.max(0, remaining + 100); // 加 100ms 缓冲
|
|
131
|
+
}
|
|
132
|
+
}
|
|
133
|
+
//# sourceMappingURL=rate-limiter.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"rate-limiter.js","sourceRoot":"","sources":["../../src/security/rate-limiter.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,YAAY,EAAe,MAAM,mBAAmB,CAAC;AAa9D,MAAM,OAAO,WAAW;IACd,MAAM,CAAS;IACf,MAAM,CAAkB;IAEhC,YAAY;IACJ,cAAc,GAAG,IAAI,GAAG,EAAgC,CAAC;IACjE,QAAQ;IACA,aAAa,GAAyB,EAAE,UAAU,EAAE,EAAE,EAAE,CAAC;IACjE,YAAY;IACJ,YAAY,GAAG,CAAC,CAAC;IAEzB,YAAY,MAAuB;QACjC,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,MAAM,GAAG,YAAY,CAAC,aAAa,CAAC,CAAC;IAC5C,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,QAAgB;QACpB,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAEvB,YAAY;QACZ,MAAM,YAAY,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC;QAC3C,IAAI,CAAC,YAAY,CAAC,OAAO;YAAE,OAAO,YAAY,CAAC;QAE/C,aAAa;QACb,MAAM,YAAY,GAAG,IAAI,CAAC,WAAW,CAAC,QAAQ,EAAE,GAAG,CAAC,CAAC;QACrD,IAAI,CAAC,YAAY,CAAC,OAAO;YAAE,OAAO,YAAY,CAAC;QAE/C,KAAK;QACL,IAAI,CAAC,aAAa,CAAC,QAAQ,EAAE,GAAG,CAAC,CAAC;QAClC,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;IAC3B,CAAC;IAED;;OAEG;IACH,iBAAiB;QACf,IAAI,IAAI,CAAC,YAAY,IAAI,IAAI,CAAC,MAAM,CAAC,kBAAkB,EAAE,CAAC;YACxD,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,WAAW,EAAE;gBAC5B,OAAO,EAAE,IAAI,CAAC,YAAY;gBAC1B,GAAG,EAAE,IAAI,CAAC,MAAM,CAAC,kBAAkB;aACpC,CAAC,CAAC;YACH,OAAO;gBACL,OAAO,EAAE,KAAK;gBACd,MAAM,EAAE,cAAc,IAAI,CAAC,YAAY,IAAI,IAAI,CAAC,MAAM,CAAC,kBAAkB,GAAG;aAC7E,CAAC;QACJ,CAAC;QACD,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;IAC3B,CAAC;IAED;;OAEG;IACH,WAAW;QACT,IAAI,CAAC,YAAY,EAAE,CAAC;QACpB,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,cAAc,IAAI,CAAC,YAAY,EAAE,CAAC,CAAC;IACvD,CAAC;IAED;;OAEG;IACH,YAAY;QACV,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,YAAY,GAAG,CAAC,CAAC,CAAC;QACvD,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,cAAc,IAAI,CAAC,YAAY,EAAE,CAAC,CAAC;IACvD,CAAC;IAED;;OAEG;IACH,SAAS;QACP,OAAO;YACL,YAAY,EAAE,IAAI,CAAC,YAAY;YAC/B,aAAa,EAAE,IAAI,CAAC,MAAM,CAAC,kBAAkB;SAC9C,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,KAAK;QACH,IAAI,CAAC,cAAc,CAAC,KAAK,EAAE,CAAC;QAC5B,IAAI,CAAC,aAAa,GAAG,EAAE,UAAU,EAAE,EAAE,EAAE,CAAC;QACxC,IAAI,CAAC,YAAY,GAAG,CAAC,CAAC;IACxB,CAAC;IAED,iDAAiD;IAEzC,WAAW,CAAC,GAAW;QAC7B,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,aAAa,EAAE,GAAG,EAAE,MAAM,CAAC,CAAC;QAElD,IAAI,IAAI,CAAC,aAAa,CAAC,UAAU,CAAC,MAAM,IAAI,IAAI,CAAC,MAAM,CAAC,oBAAoB,GAAG,CAAC,EAAE,CAAC;YACjF,OAAO;gBACL,OAAO,EAAE,KAAK;gBACd,MAAM,EAAE,UAAU;gBAClB,YAAY,EAAE,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,aAAa,CAAC,UAAU,EAAE,GAAG,EAAE,MAAM,CAAC;aAC7E,CAAC;QACJ,CAAC;QAED,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;IAC3B,CAAC;IAEO,WAAW,CAAC,QAAgB,EAAE,GAAW;QAC/C,IAAI,CAAC,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC;YACvC,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,QAAQ,EAAE,EAAE,UAAU,EAAE,EAAE,EAAE,CAAC,CAAC;QACxD,CAAC;QACD,MAAM,OAAO,GAAG,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,QAAQ,CAAE,CAAC;QAEnD,IAAI,CAAC,WAAW,CAAC,OAAO,EAAE,GAAG,EAAE,MAAM,CAAC,CAAC;QAEvC,IAAI,OAAO,CAAC,UAAU,CAAC,MAAM,IAAI,IAAI,CAAC,MAAM,CAAC,oBAAoB,EAAE,CAAC;YAClE,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,QAAQ,SAAS,CAAC,CAAC;YAC3C,OAAO;gBACL,OAAO,EAAE,KAAK;gBACd,MAAM,EAAE,eAAe,OAAO,CAAC,UAAU,CAAC,MAAM,OAAO;gBACvD,YAAY,EAAE,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC,UAAU,EAAE,GAAG,EAAE,MAAM,CAAC;aAClE,CAAC;QACJ,CAAC;QAED,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;IAC3B,CAAC;IAEO,aAAa,CAAC,QAAgB,EAAE,GAAW;QACjD,OAAO;QACP,IAAI,CAAC,aAAa,CAAC,UAAU,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAExC,QAAQ;QACR,IAAI,CAAC,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC;YACvC,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,QAAQ,EAAE,EAAE,UAAU,EAAE,EAAE,EAAE,CAAC,CAAC;QACxD,CAAC;QACD,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,QAAQ,CAAE,CAAC,UAAU,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IAC1D,CAAC;IAEO,WAAW,CAAC,OAA6B,EAAE,GAAW,EAAE,QAAgB;QAC9E,MAAM,MAAM,GAAG,GAAG,GAAG,QAAQ,CAAC;QAC9B,OAAO,CAAC,UAAU,GAAG,OAAO,CAAC,UAAU,CAAC,MAAM,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,GAAG,MAAM,CAAC,CAAC;IACpE,CAAC;IAEO,aAAa,CAAC,UAAoB,EAAE,GAAW,EAAE,QAAgB;QACvE,IAAI,UAAU,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,CAAC,CAAC;QACtC,MAAM,MAAM,GAAG,UAAU,CAAC,CAAC,CAAC,CAAC;QAC7B,MAAM,SAAS,GAAG,QAAQ,GAAG,CAAC,GAAG,GAAG,MAAM,CAAC,CAAC;QAC5C,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,SAAS,GAAG,GAAG,CAAC,CAAC,CAAC,aAAa;IACpD,CAAC;CACF"}
|
|
@@ -0,0 +1,33 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* 来源验证器
|
|
3
|
+
*
|
|
4
|
+
* 验证消息是否来自智工坊平台
|
|
5
|
+
*/
|
|
6
|
+
import type { PlatformMessage } from '../types/message.js';
|
|
7
|
+
export interface SourceVerifyResult {
|
|
8
|
+
valid: boolean;
|
|
9
|
+
reason?: string;
|
|
10
|
+
}
|
|
11
|
+
export interface SourceVerifierConfig {
|
|
12
|
+
/** 是否验证消息时间戳 */
|
|
13
|
+
validateTimestamp: boolean;
|
|
14
|
+
/** 最大时间偏移(毫秒) */
|
|
15
|
+
maxTimestampSkewMs: number;
|
|
16
|
+
/** 任务消息是否必须有 taskId */
|
|
17
|
+
requireTaskId: boolean;
|
|
18
|
+
/** 消息是否必须有发送者 */
|
|
19
|
+
requireSenderId: boolean;
|
|
20
|
+
}
|
|
21
|
+
export declare class SourceVerifier {
|
|
22
|
+
private logger;
|
|
23
|
+
private config;
|
|
24
|
+
constructor(config?: Partial<SourceVerifierConfig>);
|
|
25
|
+
/**
|
|
26
|
+
* 验证消息来源
|
|
27
|
+
*/
|
|
28
|
+
verify(message: PlatformMessage): SourceVerifyResult;
|
|
29
|
+
private validateType;
|
|
30
|
+
private validateStructure;
|
|
31
|
+
private validateTimestamp;
|
|
32
|
+
}
|
|
33
|
+
//# sourceMappingURL=source-verifier.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"source-verifier.d.ts","sourceRoot":"","sources":["../../src/security/source-verifier.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAGH,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,qBAAqB,CAAC;AAG3D,MAAM,WAAW,kBAAkB;IACjC,KAAK,EAAE,OAAO,CAAC;IACf,MAAM,CAAC,EAAE,MAAM,CAAC;CACjB;AAED,MAAM,WAAW,oBAAoB;IACnC,gBAAgB;IAChB,iBAAiB,EAAE,OAAO,CAAC;IAC3B,iBAAiB;IACjB,kBAAkB,EAAE,MAAM,CAAC;IAC3B,uBAAuB;IACvB,aAAa,EAAE,OAAO,CAAC;IACvB,iBAAiB;IACjB,eAAe,EAAE,OAAO,CAAC;CAC1B;AASD,qBAAa,cAAc;IACzB,OAAO,CAAC,MAAM,CAAS;IACvB,OAAO,CAAC,MAAM,CAAuB;gBAEzB,MAAM,GAAE,OAAO,CAAC,oBAAoB,CAAM;IAKtD;;OAEG;IACH,MAAM,CAAC,OAAO,EAAE,eAAe,GAAG,kBAAkB;IAoBpD,OAAO,CAAC,YAAY;IAepB,OAAO,CAAC,iBAAiB;IAoCzB,OAAO,CAAC,iBAAiB;CAuB1B"}
|