@ai-jshook/mcp 0.1.1
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/.env.example +38 -0
- package/CLAUDE.md +170 -0
- package/README.md +564 -0
- package/bun.lock +1484 -0
- package/dist/index.d.ts +3 -0
- package/dist/index.d.ts.map +1 -0
- package/dist/index.js +57 -0
- package/dist/index.js.map +1 -0
- package/dist/modules/analyzer/AISummarizer.d.ts +41 -0
- package/dist/modules/analyzer/AISummarizer.d.ts.map +1 -0
- package/dist/modules/analyzer/AISummarizer.js +186 -0
- package/dist/modules/analyzer/AISummarizer.js.map +1 -0
- package/dist/modules/analyzer/CodeAnalyzer.d.ts +28 -0
- package/dist/modules/analyzer/CodeAnalyzer.d.ts.map +1 -0
- package/dist/modules/analyzer/CodeAnalyzer.js +1287 -0
- package/dist/modules/analyzer/CodeAnalyzer.js.map +1 -0
- package/dist/modules/analyzer/IntelligentAnalyzer.d.ts +114 -0
- package/dist/modules/analyzer/IntelligentAnalyzer.d.ts.map +1 -0
- package/dist/modules/analyzer/IntelligentAnalyzer.js +1176 -0
- package/dist/modules/analyzer/IntelligentAnalyzer.js.map +1 -0
- package/dist/modules/browser/BrowserModeManager.d.ts +31 -0
- package/dist/modules/browser/BrowserModeManager.d.ts.map +1 -0
- package/dist/modules/browser/BrowserModeManager.js +241 -0
- package/dist/modules/browser/BrowserModeManager.js.map +1 -0
- package/dist/modules/captcha/AICaptchaDetector.d.ts +32 -0
- package/dist/modules/captcha/AICaptchaDetector.d.ts.map +1 -0
- package/dist/modules/captcha/AICaptchaDetector.js +387 -0
- package/dist/modules/captcha/AICaptchaDetector.js.map +1 -0
- package/dist/modules/captcha/CaptchaDetector.d.ts +28 -0
- package/dist/modules/captcha/CaptchaDetector.d.ts.map +1 -0
- package/dist/modules/captcha/CaptchaDetector.js +513 -0
- package/dist/modules/captcha/CaptchaDetector.js.map +1 -0
- package/dist/modules/collector/CodeCache.d.ts +37 -0
- package/dist/modules/collector/CodeCache.d.ts.map +1 -0
- package/dist/modules/collector/CodeCache.js +188 -0
- package/dist/modules/collector/CodeCache.js.map +1 -0
- package/dist/modules/collector/CodeCollector.d.ts +107 -0
- package/dist/modules/collector/CodeCollector.d.ts.map +1 -0
- package/dist/modules/collector/CodeCollector.js +796 -0
- package/dist/modules/collector/CodeCollector.js.map +1 -0
- package/dist/modules/collector/CodeCompressor.d.ts +65 -0
- package/dist/modules/collector/CodeCompressor.d.ts.map +1 -0
- package/dist/modules/collector/CodeCompressor.js +245 -0
- package/dist/modules/collector/CodeCompressor.js.map +1 -0
- package/dist/modules/collector/DOMInspector.d.ts +51 -0
- package/dist/modules/collector/DOMInspector.d.ts.map +1 -0
- package/dist/modules/collector/DOMInspector.js +437 -0
- package/dist/modules/collector/DOMInspector.js.map +1 -0
- package/dist/modules/collector/PageController.d.ts +79 -0
- package/dist/modules/collector/PageController.d.ts.map +1 -0
- package/dist/modules/collector/PageController.js +287 -0
- package/dist/modules/collector/PageController.js.map +1 -0
- package/dist/modules/collector/SmartCodeCollector.d.ts +38 -0
- package/dist/modules/collector/SmartCodeCollector.d.ts.map +1 -0
- package/dist/modules/collector/SmartCodeCollector.js +208 -0
- package/dist/modules/collector/SmartCodeCollector.js.map +1 -0
- package/dist/modules/collector/StreamingCollector.d.ts +46 -0
- package/dist/modules/collector/StreamingCollector.d.ts.map +1 -0
- package/dist/modules/collector/StreamingCollector.js +127 -0
- package/dist/modules/collector/StreamingCollector.js.map +1 -0
- package/dist/modules/crypto/CryptoDetector.d.ts +22 -0
- package/dist/modules/crypto/CryptoDetector.d.ts.map +1 -0
- package/dist/modules/crypto/CryptoDetector.js +168 -0
- package/dist/modules/crypto/CryptoDetector.js.map +1 -0
- package/dist/modules/crypto/CryptoDetectorEnhanced.d.ts +31 -0
- package/dist/modules/crypto/CryptoDetectorEnhanced.d.ts.map +1 -0
- package/dist/modules/crypto/CryptoDetectorEnhanced.js +269 -0
- package/dist/modules/crypto/CryptoDetectorEnhanced.js.map +1 -0
- package/dist/modules/crypto/CryptoRules.d.ts +59 -0
- package/dist/modules/crypto/CryptoRules.d.ts.map +1 -0
- package/dist/modules/crypto/CryptoRules.js +234 -0
- package/dist/modules/crypto/CryptoRules.js.map +1 -0
- package/dist/modules/debugger/BlackboxManager.d.ts +14 -0
- package/dist/modules/debugger/BlackboxManager.d.ts.map +1 -0
- package/dist/modules/debugger/BlackboxManager.js +98 -0
- package/dist/modules/debugger/BlackboxManager.js.map +1 -0
- package/dist/modules/debugger/DebuggerManager.d.ts +138 -0
- package/dist/modules/debugger/DebuggerManager.d.ts.map +1 -0
- package/dist/modules/debugger/DebuggerManager.js +777 -0
- package/dist/modules/debugger/DebuggerManager.js.map +1 -0
- package/dist/modules/debugger/EventBreakpointManager.d.ts +30 -0
- package/dist/modules/debugger/EventBreakpointManager.d.ts.map +1 -0
- package/dist/modules/debugger/EventBreakpointManager.js +125 -0
- package/dist/modules/debugger/EventBreakpointManager.js.map +1 -0
- package/dist/modules/debugger/RuntimeInspector.d.ts +54 -0
- package/dist/modules/debugger/RuntimeInspector.d.ts.map +1 -0
- package/dist/modules/debugger/RuntimeInspector.js +277 -0
- package/dist/modules/debugger/RuntimeInspector.js.map +1 -0
- package/dist/modules/debugger/ScriptManager.d.ts +94 -0
- package/dist/modules/debugger/ScriptManager.d.ts.map +1 -0
- package/dist/modules/debugger/ScriptManager.js +433 -0
- package/dist/modules/debugger/ScriptManager.js.map +1 -0
- package/dist/modules/debugger/WatchExpressionManager.d.ts +52 -0
- package/dist/modules/debugger/WatchExpressionManager.d.ts.map +1 -0
- package/dist/modules/debugger/WatchExpressionManager.js +136 -0
- package/dist/modules/debugger/WatchExpressionManager.js.map +1 -0
- package/dist/modules/debugger/XHRBreakpointManager.d.ts +21 -0
- package/dist/modules/debugger/XHRBreakpointManager.d.ts.map +1 -0
- package/dist/modules/debugger/XHRBreakpointManager.js +81 -0
- package/dist/modules/debugger/XHRBreakpointManager.js.map +1 -0
- package/dist/modules/deobfuscator/ASTOptimizer.d.ts +12 -0
- package/dist/modules/deobfuscator/ASTOptimizer.d.ts.map +1 -0
- package/dist/modules/deobfuscator/ASTOptimizer.js +234 -0
- package/dist/modules/deobfuscator/ASTOptimizer.js.map +1 -0
- package/dist/modules/deobfuscator/AdvancedDeobfuscator.d.ts +52 -0
- package/dist/modules/deobfuscator/AdvancedDeobfuscator.d.ts.map +1 -0
- package/dist/modules/deobfuscator/AdvancedDeobfuscator.js +985 -0
- package/dist/modules/deobfuscator/AdvancedDeobfuscator.js.map +1 -0
- package/dist/modules/deobfuscator/Deobfuscator.d.ts +23 -0
- package/dist/modules/deobfuscator/Deobfuscator.d.ts.map +1 -0
- package/dist/modules/deobfuscator/Deobfuscator.js +487 -0
- package/dist/modules/deobfuscator/Deobfuscator.js.map +1 -0
- package/dist/modules/deobfuscator/JSVMPDeobfuscator.d.ts +19 -0
- package/dist/modules/deobfuscator/JSVMPDeobfuscator.d.ts.map +1 -0
- package/dist/modules/deobfuscator/JSVMPDeobfuscator.js +594 -0
- package/dist/modules/deobfuscator/JSVMPDeobfuscator.js.map +1 -0
- package/dist/modules/deobfuscator/JScramberDeobfuscator.d.ts +28 -0
- package/dist/modules/deobfuscator/JScramberDeobfuscator.d.ts.map +1 -0
- package/dist/modules/deobfuscator/JScramberDeobfuscator.js +239 -0
- package/dist/modules/deobfuscator/JScramberDeobfuscator.js.map +1 -0
- package/dist/modules/deobfuscator/PackerDeobfuscator.d.ts +38 -0
- package/dist/modules/deobfuscator/PackerDeobfuscator.d.ts.map +1 -0
- package/dist/modules/deobfuscator/PackerDeobfuscator.js +191 -0
- package/dist/modules/deobfuscator/PackerDeobfuscator.js.map +1 -0
- package/dist/modules/detector/ObfuscationDetector.d.ts +35 -0
- package/dist/modules/detector/ObfuscationDetector.d.ts.map +1 -0
- package/dist/modules/detector/ObfuscationDetector.js +278 -0
- package/dist/modules/detector/ObfuscationDetector.js.map +1 -0
- package/dist/modules/emulator/AIEnvironmentAnalyzer.d.ts +32 -0
- package/dist/modules/emulator/AIEnvironmentAnalyzer.d.ts.map +1 -0
- package/dist/modules/emulator/AIEnvironmentAnalyzer.js +548 -0
- package/dist/modules/emulator/AIEnvironmentAnalyzer.js.map +1 -0
- package/dist/modules/emulator/BrowserAPIDatabase.d.ts +34 -0
- package/dist/modules/emulator/BrowserAPIDatabase.d.ts.map +1 -0
- package/dist/modules/emulator/BrowserAPIDatabase.js +326 -0
- package/dist/modules/emulator/BrowserAPIDatabase.js.map +1 -0
- package/dist/modules/emulator/BrowserEnvironmentRules.d.ts +47 -0
- package/dist/modules/emulator/BrowserEnvironmentRules.d.ts.map +1 -0
- package/dist/modules/emulator/BrowserEnvironmentRules.js +493 -0
- package/dist/modules/emulator/BrowserEnvironmentRules.js.map +1 -0
- package/dist/modules/emulator/EnvironmentEmulator.d.ts +27 -0
- package/dist/modules/emulator/EnvironmentEmulator.d.ts.map +1 -0
- package/dist/modules/emulator/EnvironmentEmulator.js +1113 -0
- package/dist/modules/emulator/EnvironmentEmulator.js.map +1 -0
- package/dist/modules/emulator/EnvironmentEmulatorEnhanced.d.ts +26 -0
- package/dist/modules/emulator/EnvironmentEmulatorEnhanced.d.ts.map +1 -0
- package/dist/modules/emulator/EnvironmentEmulatorEnhanced.js +493 -0
- package/dist/modules/emulator/EnvironmentEmulatorEnhanced.js.map +1 -0
- package/dist/modules/emulator/templates/chrome-env.d.ts +260 -0
- package/dist/modules/emulator/templates/chrome-env.d.ts.map +1 -0
- package/dist/modules/emulator/templates/chrome-env.js +253 -0
- package/dist/modules/emulator/templates/chrome-env.js.map +1 -0
- package/dist/modules/hook/AIHookGenerator.d.ts +53 -0
- package/dist/modules/hook/AIHookGenerator.d.ts.map +1 -0
- package/dist/modules/hook/AIHookGenerator.js +353 -0
- package/dist/modules/hook/AIHookGenerator.js.map +1 -0
- package/dist/modules/hook/HookManager.d.ts +67 -0
- package/dist/modules/hook/HookManager.d.ts.map +1 -0
- package/dist/modules/hook/HookManager.js +1225 -0
- package/dist/modules/hook/HookManager.js.map +1 -0
- package/dist/modules/monitor/ConsoleMonitor.d.ts +140 -0
- package/dist/modules/monitor/ConsoleMonitor.d.ts.map +1 -0
- package/dist/modules/monitor/ConsoleMonitor.js +834 -0
- package/dist/modules/monitor/ConsoleMonitor.js.map +1 -0
- package/dist/modules/monitor/PerformanceMonitor.d.ts +65 -0
- package/dist/modules/monitor/PerformanceMonitor.d.ts.map +1 -0
- package/dist/modules/monitor/PerformanceMonitor.js +175 -0
- package/dist/modules/monitor/PerformanceMonitor.js.map +1 -0
- package/dist/modules/stealth/StealthScripts2025.d.ts +17 -0
- package/dist/modules/stealth/StealthScripts2025.d.ts.map +1 -0
- package/dist/modules/stealth/StealthScripts2025.js +274 -0
- package/dist/modules/stealth/StealthScripts2025.js.map +1 -0
- package/dist/modules/symbolic/JSVMPSymbolicExecutor.d.ts +69 -0
- package/dist/modules/symbolic/JSVMPSymbolicExecutor.d.ts.map +1 -0
- package/dist/modules/symbolic/JSVMPSymbolicExecutor.js +232 -0
- package/dist/modules/symbolic/JSVMPSymbolicExecutor.js.map +1 -0
- package/dist/modules/symbolic/SymbolicExecutor.d.ts +69 -0
- package/dist/modules/symbolic/SymbolicExecutor.d.ts.map +1 -0
- package/dist/modules/symbolic/SymbolicExecutor.js +346 -0
- package/dist/modules/symbolic/SymbolicExecutor.js.map +1 -0
- package/dist/server/AIHookToolDefinitions.d.ts +3 -0
- package/dist/server/AIHookToolDefinitions.d.ts.map +1 -0
- package/dist/server/AIHookToolDefinitions.js +284 -0
- package/dist/server/AIHookToolDefinitions.js.map +1 -0
- package/dist/server/AIHookToolHandlers.d.ts +50 -0
- package/dist/server/AIHookToolHandlers.d.ts.map +1 -0
- package/dist/server/AIHookToolHandlers.js +311 -0
- package/dist/server/AIHookToolHandlers.js.map +1 -0
- package/dist/server/AdvancedToolDefinitions.d.ts +3 -0
- package/dist/server/AdvancedToolDefinitions.d.ts.map +1 -0
- package/dist/server/AdvancedToolDefinitions.js +218 -0
- package/dist/server/AdvancedToolDefinitions.js.map +1 -0
- package/dist/server/AdvancedToolHandlers.d.ts +85 -0
- package/dist/server/AdvancedToolHandlers.d.ts.map +1 -0
- package/dist/server/AdvancedToolHandlers.js +431 -0
- package/dist/server/AdvancedToolHandlers.js.map +1 -0
- package/dist/server/BrowserToolDefinitions.d.ts +3 -0
- package/dist/server/BrowserToolDefinitions.d.ts.map +1 -0
- package/dist/server/BrowserToolDefinitions.js +841 -0
- package/dist/server/BrowserToolDefinitions.js.map +1 -0
- package/dist/server/BrowserToolHandlers.d.ts +290 -0
- package/dist/server/BrowserToolHandlers.d.ts.map +1 -0
- package/dist/server/BrowserToolHandlers.js +784 -0
- package/dist/server/BrowserToolHandlers.js.map +1 -0
- package/dist/server/CacheToolDefinitions.d.ts +3 -0
- package/dist/server/CacheToolDefinitions.d.ts.map +1 -0
- package/dist/server/CacheToolDefinitions.js +166 -0
- package/dist/server/CacheToolDefinitions.js.map +1 -0
- package/dist/server/DebuggerToolDefinitions.d.ts +3 -0
- package/dist/server/DebuggerToolDefinitions.d.ts.map +1 -0
- package/dist/server/DebuggerToolDefinitions.js +600 -0
- package/dist/server/DebuggerToolDefinitions.js.map +1 -0
- package/dist/server/DebuggerToolHandlers.d.ts +230 -0
- package/dist/server/DebuggerToolHandlers.d.ts.map +1 -0
- package/dist/server/DebuggerToolHandlers.js +935 -0
- package/dist/server/DebuggerToolHandlers.js.map +1 -0
- package/dist/server/MCPServer.d.ts +55 -0
- package/dist/server/MCPServer.d.ts.map +1 -0
- package/dist/server/MCPServer.js +1344 -0
- package/dist/server/MCPServer.js.map +1 -0
- package/dist/server/TokenBudgetToolDefinitions.d.ts +3 -0
- package/dist/server/TokenBudgetToolDefinitions.d.ts.map +1 -0
- package/dist/server/TokenBudgetToolDefinitions.js +114 -0
- package/dist/server/TokenBudgetToolDefinitions.js.map +1 -0
- package/dist/services/LLMService.d.ts +41 -0
- package/dist/services/LLMService.d.ts.map +1 -0
- package/dist/services/LLMService.js +792 -0
- package/dist/services/LLMService.js.map +1 -0
- package/dist/types/index.d.ts +527 -0
- package/dist/types/index.d.ts.map +1 -0
- package/dist/types/index.js +2 -0
- package/dist/types/index.js.map +1 -0
- package/dist/utils/AdaptiveDataSerializer.d.ts +27 -0
- package/dist/utils/AdaptiveDataSerializer.d.ts.map +1 -0
- package/dist/utils/AdaptiveDataSerializer.js +215 -0
- package/dist/utils/AdaptiveDataSerializer.js.map +1 -0
- package/dist/utils/CacheAdapters.d.ts +30 -0
- package/dist/utils/CacheAdapters.d.ts.map +1 -0
- package/dist/utils/CacheAdapters.js +83 -0
- package/dist/utils/CacheAdapters.js.map +1 -0
- package/dist/utils/TokenBudgetManager.d.ts +52 -0
- package/dist/utils/TokenBudgetManager.d.ts.map +1 -0
- package/dist/utils/TokenBudgetManager.js +190 -0
- package/dist/utils/TokenBudgetManager.js.map +1 -0
- package/dist/utils/UnifiedCacheManager.d.ts +55 -0
- package/dist/utils/UnifiedCacheManager.d.ts.map +1 -0
- package/dist/utils/UnifiedCacheManager.js +207 -0
- package/dist/utils/UnifiedCacheManager.js.map +1 -0
- package/dist/utils/cache.d.ts +13 -0
- package/dist/utils/cache.d.ts.map +1 -0
- package/dist/utils/cache.js +92 -0
- package/dist/utils/cache.js.map +1 -0
- package/dist/utils/config.d.ts +7 -0
- package/dist/utils/config.d.ts.map +1 -0
- package/dist/utils/config.js +93 -0
- package/dist/utils/config.js.map +1 -0
- package/dist/utils/detailedDataManager.d.ts +60 -0
- package/dist/utils/detailedDataManager.d.ts.map +1 -0
- package/dist/utils/detailedDataManager.js +204 -0
- package/dist/utils/detailedDataManager.js.map +1 -0
- package/dist/utils/logger.d.ts +16 -0
- package/dist/utils/logger.d.ts.map +1 -0
- package/dist/utils/logger.js +47 -0
- package/dist/utils/logger.js.map +1 -0
- package/dist/utils/parallel.d.ts +40 -0
- package/dist/utils/parallel.d.ts.map +1 -0
- package/dist/utils/parallel.js +148 -0
- package/dist/utils/parallel.js.map +1 -0
- package/package.json +94 -0
- package/server.json +39 -0
- package/tsconfig.dev.json +14 -0
|
@@ -0,0 +1,346 @@
|
|
|
1
|
+
import * as parser from '@babel/parser';
|
|
2
|
+
import traverse from '@babel/traverse';
|
|
3
|
+
import * as t from '@babel/types';
|
|
4
|
+
import { logger } from '../../utils/logger.js';
|
|
5
|
+
export class SymbolicExecutor {
|
|
6
|
+
symbolCounter = 0;
|
|
7
|
+
pathCounter = 0;
|
|
8
|
+
async execute(options) {
|
|
9
|
+
const startTime = Date.now();
|
|
10
|
+
const { code, maxPaths = 100, maxDepth = 50, timeout = 30000, enableConstraintSolving = false, } = options;
|
|
11
|
+
logger.info('🔬 开始符号执行分析...');
|
|
12
|
+
const paths = [];
|
|
13
|
+
const allSymbolicValues = [];
|
|
14
|
+
const allConstraints = [];
|
|
15
|
+
const warnings = [];
|
|
16
|
+
try {
|
|
17
|
+
const ast = parser.parse(code, {
|
|
18
|
+
sourceType: 'unambiguous',
|
|
19
|
+
plugins: ['jsx', 'typescript'],
|
|
20
|
+
errorRecovery: true,
|
|
21
|
+
});
|
|
22
|
+
const initialState = {
|
|
23
|
+
pc: 0,
|
|
24
|
+
stack: [],
|
|
25
|
+
registers: new Map(),
|
|
26
|
+
memory: new Map(),
|
|
27
|
+
pathConstraints: [],
|
|
28
|
+
};
|
|
29
|
+
const worklist = [
|
|
30
|
+
{ state: initialState, depth: 0 },
|
|
31
|
+
];
|
|
32
|
+
while (worklist.length > 0 && paths.length < maxPaths) {
|
|
33
|
+
if (Date.now() - startTime > timeout) {
|
|
34
|
+
warnings.push('符号执行超时');
|
|
35
|
+
break;
|
|
36
|
+
}
|
|
37
|
+
const { state, depth } = worklist.pop();
|
|
38
|
+
if (depth >= maxDepth) {
|
|
39
|
+
warnings.push(`路径深度达到限制: ${maxDepth}`);
|
|
40
|
+
continue;
|
|
41
|
+
}
|
|
42
|
+
const nextStates = this.executeStep(state, ast);
|
|
43
|
+
for (const nextState of nextStates) {
|
|
44
|
+
if (this.isTerminalState(nextState)) {
|
|
45
|
+
const path = this.createPath(nextState);
|
|
46
|
+
paths.push(path);
|
|
47
|
+
this.collectSymbolicValues(nextState, allSymbolicValues);
|
|
48
|
+
this.collectConstraints(nextState, allConstraints);
|
|
49
|
+
}
|
|
50
|
+
else {
|
|
51
|
+
worklist.push({ state: nextState, depth: depth + 1 });
|
|
52
|
+
}
|
|
53
|
+
}
|
|
54
|
+
}
|
|
55
|
+
if (enableConstraintSolving) {
|
|
56
|
+
await this.solveConstraints(paths, warnings);
|
|
57
|
+
}
|
|
58
|
+
const coverage = this.calculateCoverage(paths, ast);
|
|
59
|
+
const executionTime = Date.now() - startTime;
|
|
60
|
+
logger.info(`✅ 符号执行完成,耗时 ${executionTime}ms`);
|
|
61
|
+
logger.info(`📊 生成路径: ${paths.length}`);
|
|
62
|
+
logger.info(`📈 覆盖率: ${(coverage * 100).toFixed(1)}%`);
|
|
63
|
+
return {
|
|
64
|
+
paths,
|
|
65
|
+
coverage,
|
|
66
|
+
symbolicValues: allSymbolicValues,
|
|
67
|
+
constraints: allConstraints,
|
|
68
|
+
warnings,
|
|
69
|
+
stats: {
|
|
70
|
+
totalPaths: paths.length,
|
|
71
|
+
feasiblePaths: paths.filter((p) => p.isFeasible).length,
|
|
72
|
+
infeasiblePaths: paths.filter((p) => !p.isFeasible).length,
|
|
73
|
+
executionTime,
|
|
74
|
+
},
|
|
75
|
+
};
|
|
76
|
+
}
|
|
77
|
+
catch (error) {
|
|
78
|
+
logger.error('符号执行失败', error);
|
|
79
|
+
throw error;
|
|
80
|
+
}
|
|
81
|
+
}
|
|
82
|
+
executeStep(state, ast) {
|
|
83
|
+
const nextStates = [];
|
|
84
|
+
let currentNode = null;
|
|
85
|
+
let nodeIndex = 0;
|
|
86
|
+
traverse(ast, {
|
|
87
|
+
enter(path) {
|
|
88
|
+
if (nodeIndex === state.pc) {
|
|
89
|
+
currentNode = path.node;
|
|
90
|
+
path.stop();
|
|
91
|
+
}
|
|
92
|
+
nodeIndex++;
|
|
93
|
+
},
|
|
94
|
+
});
|
|
95
|
+
if (!currentNode) {
|
|
96
|
+
return [];
|
|
97
|
+
}
|
|
98
|
+
if (t.isVariableDeclaration(currentNode)) {
|
|
99
|
+
const newState = this.cloneState(state);
|
|
100
|
+
const varDecl = currentNode;
|
|
101
|
+
varDecl.declarations.forEach((decl) => {
|
|
102
|
+
if (t.isIdentifier(decl.id)) {
|
|
103
|
+
const varName = decl.id.name;
|
|
104
|
+
const symbolicValue = this.createSymbolicValue('unknown', varName, varName);
|
|
105
|
+
newState.memory.set(varName, symbolicValue);
|
|
106
|
+
}
|
|
107
|
+
});
|
|
108
|
+
newState.pc++;
|
|
109
|
+
nextStates.push(newState);
|
|
110
|
+
}
|
|
111
|
+
else if (t.isIfStatement(currentNode)) {
|
|
112
|
+
const trueState = this.cloneState(state);
|
|
113
|
+
const falseState = this.cloneState(state);
|
|
114
|
+
const ifStmt = currentNode;
|
|
115
|
+
const conditionExpr = this.nodeToString(ifStmt.test);
|
|
116
|
+
trueState.pathConstraints.push({
|
|
117
|
+
type: 'custom',
|
|
118
|
+
expression: conditionExpr,
|
|
119
|
+
description: '条件为真',
|
|
120
|
+
});
|
|
121
|
+
falseState.pathConstraints.push({
|
|
122
|
+
type: 'custom',
|
|
123
|
+
expression: `!(${conditionExpr})`,
|
|
124
|
+
description: '条件为假',
|
|
125
|
+
});
|
|
126
|
+
trueState.pc++;
|
|
127
|
+
falseState.pc++;
|
|
128
|
+
nextStates.push(trueState, falseState);
|
|
129
|
+
}
|
|
130
|
+
else if (t.isWhileStatement(currentNode) || t.isForStatement(currentNode)) {
|
|
131
|
+
const enterState = this.cloneState(state);
|
|
132
|
+
const skipState = this.cloneState(state);
|
|
133
|
+
enterState.pc++;
|
|
134
|
+
skipState.pc += 2;
|
|
135
|
+
nextStates.push(enterState, skipState);
|
|
136
|
+
}
|
|
137
|
+
else if (t.isAssignmentExpression(currentNode)) {
|
|
138
|
+
const newState = this.cloneState(state);
|
|
139
|
+
const assignExpr = currentNode;
|
|
140
|
+
if (t.isIdentifier(assignExpr.left)) {
|
|
141
|
+
const varName = assignExpr.left.name;
|
|
142
|
+
const rightExpr = this.nodeToString(assignExpr.right);
|
|
143
|
+
const symbolicValue = this.createSymbolicValue('unknown', rightExpr, rightExpr);
|
|
144
|
+
newState.memory.set(varName, symbolicValue);
|
|
145
|
+
}
|
|
146
|
+
newState.pc++;
|
|
147
|
+
nextStates.push(newState);
|
|
148
|
+
}
|
|
149
|
+
else {
|
|
150
|
+
const newState = this.cloneState(state);
|
|
151
|
+
newState.pc++;
|
|
152
|
+
nextStates.push(newState);
|
|
153
|
+
}
|
|
154
|
+
return nextStates;
|
|
155
|
+
}
|
|
156
|
+
nodeToString(node) {
|
|
157
|
+
if (t.isIdentifier(node)) {
|
|
158
|
+
return node.name;
|
|
159
|
+
}
|
|
160
|
+
else if (t.isNumericLiteral(node)) {
|
|
161
|
+
return String(node.value);
|
|
162
|
+
}
|
|
163
|
+
else if (t.isStringLiteral(node)) {
|
|
164
|
+
return `"${node.value}"`;
|
|
165
|
+
}
|
|
166
|
+
else if (t.isBinaryExpression(node)) {
|
|
167
|
+
return `${this.nodeToString(node.left)} ${node.operator} ${this.nodeToString(node.right)}`;
|
|
168
|
+
}
|
|
169
|
+
else if (t.isUnaryExpression(node)) {
|
|
170
|
+
return `${node.operator}${this.nodeToString(node.argument)}`;
|
|
171
|
+
}
|
|
172
|
+
else {
|
|
173
|
+
return '[Complex Expression]';
|
|
174
|
+
}
|
|
175
|
+
}
|
|
176
|
+
isTerminalState(state) {
|
|
177
|
+
if (state.pc > 1000) {
|
|
178
|
+
return true;
|
|
179
|
+
}
|
|
180
|
+
if (state.pathConstraints.length > 50) {
|
|
181
|
+
return true;
|
|
182
|
+
}
|
|
183
|
+
if (state.stack.length === 0 && state.memory.size === 0) {
|
|
184
|
+
return true;
|
|
185
|
+
}
|
|
186
|
+
return false;
|
|
187
|
+
}
|
|
188
|
+
createPath(state) {
|
|
189
|
+
const pathId = `path-${this.pathCounter++}`;
|
|
190
|
+
const coverage = this.calculatePathCoverage(state);
|
|
191
|
+
return {
|
|
192
|
+
id: pathId,
|
|
193
|
+
states: [state],
|
|
194
|
+
constraints: [...state.pathConstraints],
|
|
195
|
+
isFeasible: this.checkPathFeasibility(state.pathConstraints),
|
|
196
|
+
coverage,
|
|
197
|
+
};
|
|
198
|
+
}
|
|
199
|
+
calculatePathCoverage(state) {
|
|
200
|
+
return Math.min(state.pc / 100, 1.0);
|
|
201
|
+
}
|
|
202
|
+
checkPathFeasibility(constraints) {
|
|
203
|
+
const expressions = new Set();
|
|
204
|
+
for (const constraint of constraints) {
|
|
205
|
+
const expr = constraint.expression;
|
|
206
|
+
if (expressions.has(`!(${expr})`)) {
|
|
207
|
+
return false;
|
|
208
|
+
}
|
|
209
|
+
expressions.add(expr);
|
|
210
|
+
}
|
|
211
|
+
return true;
|
|
212
|
+
}
|
|
213
|
+
collectSymbolicValues(state, collection) {
|
|
214
|
+
const seen = new Set();
|
|
215
|
+
for (const value of state.stack) {
|
|
216
|
+
if (!seen.has(value.id)) {
|
|
217
|
+
collection.push(value);
|
|
218
|
+
seen.add(value.id);
|
|
219
|
+
}
|
|
220
|
+
}
|
|
221
|
+
for (const value of state.registers.values()) {
|
|
222
|
+
if (!seen.has(value.id)) {
|
|
223
|
+
collection.push(value);
|
|
224
|
+
seen.add(value.id);
|
|
225
|
+
}
|
|
226
|
+
}
|
|
227
|
+
for (const value of state.memory.values()) {
|
|
228
|
+
if (!seen.has(value.id)) {
|
|
229
|
+
collection.push(value);
|
|
230
|
+
seen.add(value.id);
|
|
231
|
+
}
|
|
232
|
+
}
|
|
233
|
+
}
|
|
234
|
+
collectConstraints(state, collection) {
|
|
235
|
+
const seen = new Set();
|
|
236
|
+
for (const constraint of state.pathConstraints) {
|
|
237
|
+
const key = `${constraint.type}:${constraint.expression}`;
|
|
238
|
+
if (!seen.has(key)) {
|
|
239
|
+
collection.push(constraint);
|
|
240
|
+
seen.add(key);
|
|
241
|
+
}
|
|
242
|
+
}
|
|
243
|
+
const allValues = [
|
|
244
|
+
...state.stack,
|
|
245
|
+
...Array.from(state.registers.values()),
|
|
246
|
+
...Array.from(state.memory.values()),
|
|
247
|
+
];
|
|
248
|
+
for (const value of allValues) {
|
|
249
|
+
for (const constraint of value.constraints) {
|
|
250
|
+
const key = `${constraint.type}:${constraint.expression}`;
|
|
251
|
+
if (!seen.has(key)) {
|
|
252
|
+
collection.push(constraint);
|
|
253
|
+
seen.add(key);
|
|
254
|
+
}
|
|
255
|
+
}
|
|
256
|
+
}
|
|
257
|
+
}
|
|
258
|
+
async solveConstraints(paths, warnings) {
|
|
259
|
+
logger.info('🔍 开始约束求解...');
|
|
260
|
+
for (const path of paths) {
|
|
261
|
+
const result = this.simpleSMTSolver(path.constraints);
|
|
262
|
+
if (!result.satisfiable) {
|
|
263
|
+
path.isFeasible = false;
|
|
264
|
+
warnings.push(`路径 ${path.id} 不可行: ${result.reason}`);
|
|
265
|
+
}
|
|
266
|
+
else {
|
|
267
|
+
path.isFeasible = true;
|
|
268
|
+
}
|
|
269
|
+
}
|
|
270
|
+
logger.info(`✅ 约束求解完成,可行路径: ${paths.filter((p) => p.isFeasible).length}/${paths.length}`);
|
|
271
|
+
}
|
|
272
|
+
simpleSMTSolver(constraints) {
|
|
273
|
+
const numericConstraints = constraints.filter((c) => c.type === 'range' || c.type === 'inequality');
|
|
274
|
+
for (let i = 0; i < numericConstraints.length; i++) {
|
|
275
|
+
for (let j = i + 1; j < numericConstraints.length; j++) {
|
|
276
|
+
const c1 = numericConstraints[i];
|
|
277
|
+
const c2 = numericConstraints[j];
|
|
278
|
+
if (!c1 || !c2)
|
|
279
|
+
continue;
|
|
280
|
+
if (this.areContradictory(c1.expression, c2.expression)) {
|
|
281
|
+
return {
|
|
282
|
+
satisfiable: false,
|
|
283
|
+
reason: `约束矛盾: ${c1.expression} 与 ${c2.expression}`,
|
|
284
|
+
};
|
|
285
|
+
}
|
|
286
|
+
}
|
|
287
|
+
}
|
|
288
|
+
return { satisfiable: true };
|
|
289
|
+
}
|
|
290
|
+
areContradictory(expr1, expr2) {
|
|
291
|
+
const pattern1 = /(\w+)\s*>\s*(\d+)/;
|
|
292
|
+
const pattern2 = /(\w+)\s*<\s*(\d+)/;
|
|
293
|
+
const match1 = expr1.match(pattern1);
|
|
294
|
+
const match2 = expr2.match(pattern2);
|
|
295
|
+
if (match1 && match2 && match1[1] === match2[1] && match1[2] && match2[2]) {
|
|
296
|
+
const val1 = parseInt(match1[2], 10);
|
|
297
|
+
const val2 = parseInt(match2[2], 10);
|
|
298
|
+
return val1 >= val2;
|
|
299
|
+
}
|
|
300
|
+
return false;
|
|
301
|
+
}
|
|
302
|
+
calculateCoverage(paths, ast) {
|
|
303
|
+
let totalStatements = 0;
|
|
304
|
+
traverse(ast, {
|
|
305
|
+
Statement() {
|
|
306
|
+
totalStatements++;
|
|
307
|
+
},
|
|
308
|
+
});
|
|
309
|
+
if (totalStatements === 0) {
|
|
310
|
+
return 0;
|
|
311
|
+
}
|
|
312
|
+
const coveredStatements = new Set();
|
|
313
|
+
for (const path of paths) {
|
|
314
|
+
for (const state of path.states) {
|
|
315
|
+
coveredStatements.add(state.pc);
|
|
316
|
+
}
|
|
317
|
+
}
|
|
318
|
+
return coveredStatements.size / totalStatements;
|
|
319
|
+
}
|
|
320
|
+
cloneState(state) {
|
|
321
|
+
return {
|
|
322
|
+
pc: state.pc,
|
|
323
|
+
stack: [...state.stack],
|
|
324
|
+
registers: new Map(state.registers),
|
|
325
|
+
memory: new Map(state.memory),
|
|
326
|
+
pathConstraints: [...state.pathConstraints],
|
|
327
|
+
};
|
|
328
|
+
}
|
|
329
|
+
createSymbolicValue(type, name, source) {
|
|
330
|
+
return {
|
|
331
|
+
id: `sym-${this.symbolCounter++}`,
|
|
332
|
+
type,
|
|
333
|
+
name,
|
|
334
|
+
constraints: [],
|
|
335
|
+
source,
|
|
336
|
+
};
|
|
337
|
+
}
|
|
338
|
+
addConstraint(value, type, expression, description) {
|
|
339
|
+
value.constraints.push({
|
|
340
|
+
type,
|
|
341
|
+
expression,
|
|
342
|
+
description,
|
|
343
|
+
});
|
|
344
|
+
}
|
|
345
|
+
}
|
|
346
|
+
//# sourceMappingURL=SymbolicExecutor.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"SymbolicExecutor.js","sourceRoot":"","sources":["../../../src/modules/symbolic/SymbolicExecutor.ts"],"names":[],"mappings":"AAUA,OAAO,KAAK,MAAM,MAAM,eAAe,CAAC;AACxC,OAAO,QAAQ,MAAM,iBAAiB,CAAC;AACvC,OAAO,KAAK,CAAC,MAAM,cAAc,CAAC;AAClC,OAAO,EAAE,MAAM,EAAE,MAAM,uBAAuB,CAAC;AAkF/C,MAAM,OAAO,gBAAgB;IACnB,aAAa,GAAG,CAAC,CAAC;IAClB,WAAW,GAAG,CAAC,CAAC;IAKxB,KAAK,CAAC,OAAO,CAAC,OAAgC;QAC5C,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC7B,MAAM,EACJ,IAAI,EACJ,QAAQ,GAAG,GAAG,EACd,QAAQ,GAAG,EAAE,EACb,OAAO,GAAG,KAAK,EACf,uBAAuB,GAAG,KAAK,GAChC,GAAG,OAAO,CAAC;QAEZ,MAAM,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;QAE9B,MAAM,KAAK,GAAoB,EAAE,CAAC;QAClC,MAAM,iBAAiB,GAAoB,EAAE,CAAC;QAC9C,MAAM,cAAc,GAAiB,EAAE,CAAC;QACxC,MAAM,QAAQ,GAAa,EAAE,CAAC;QAE9B,IAAI,CAAC;YAEH,MAAM,GAAG,GAAG,MAAM,CAAC,KAAK,CAAC,IAAI,EAAE;gBAC7B,UAAU,EAAE,aAAa;gBACzB,OAAO,EAAE,CAAC,KAAK,EAAE,YAAY,CAAC;gBAC9B,aAAa,EAAE,IAAI;aACpB,CAAC,CAAC;YAGH,MAAM,YAAY,GAAkB;gBAClC,EAAE,EAAE,CAAC;gBACL,KAAK,EAAE,EAAE;gBACT,SAAS,EAAE,IAAI,GAAG,EAAE;gBACpB,MAAM,EAAE,IAAI,GAAG,EAAE;gBACjB,eAAe,EAAE,EAAE;aACpB,CAAC;YAGF,MAAM,QAAQ,GAA8C;gBAC1D,EAAE,KAAK,EAAE,YAAY,EAAE,KAAK,EAAE,CAAC,EAAE;aAClC,CAAC;YAEF,OAAO,QAAQ,CAAC,MAAM,GAAG,CAAC,IAAI,KAAK,CAAC,MAAM,GAAG,QAAQ,EAAE,CAAC;gBAEtD,IAAI,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,GAAG,OAAO,EAAE,CAAC;oBACrC,QAAQ,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;oBACxB,MAAM;gBACR,CAAC;gBAED,MAAM,EAAE,KAAK,EAAE,KAAK,EAAE,GAAG,QAAQ,CAAC,GAAG,EAAG,CAAC;gBAGzC,IAAI,KAAK,IAAI,QAAQ,EAAE,CAAC;oBACtB,QAAQ,CAAC,IAAI,CAAC,aAAa,QAAQ,EAAE,CAAC,CAAC;oBACvC,SAAS;gBACX,CAAC;gBAGD,MAAM,UAAU,GAAG,IAAI,CAAC,WAAW,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC;gBAGhD,KAAK,MAAM,SAAS,IAAI,UAAU,EAAE,CAAC;oBACnC,IAAI,IAAI,CAAC,eAAe,CAAC,SAAS,CAAC,EAAE,CAAC;wBAEpC,MAAM,IAAI,GAAG,IAAI,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC;wBACxC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;wBAGjB,IAAI,CAAC,qBAAqB,CAAC,SAAS,EAAE,iBAAiB,CAAC,CAAC;wBACzD,IAAI,CAAC,kBAAkB,CAAC,SAAS,EAAE,cAAc,CAAC,CAAC;oBACrD,CAAC;yBAAM,CAAC;wBAEN,QAAQ,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,KAAK,EAAE,KAAK,GAAG,CAAC,EAAE,CAAC,CAAC;oBACxD,CAAC;gBACH,CAAC;YACH,CAAC;YAGD,IAAI,uBAAuB,EAAE,CAAC;gBAC5B,MAAM,IAAI,CAAC,gBAAgB,CAAC,KAAK,EAAE,QAAQ,CAAC,CAAC;YAC/C,CAAC;YAGD,MAAM,QAAQ,GAAG,IAAI,CAAC,iBAAiB,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC;YAEpD,MAAM,aAAa,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;YAE7C,MAAM,CAAC,IAAI,CAAC,eAAe,aAAa,IAAI,CAAC,CAAC;YAC9C,MAAM,CAAC,IAAI,CAAC,YAAY,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC;YACxC,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,QAAQ,GAAG,GAAG,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;YAEvD,OAAO;gBACL,KAAK;gBACL,QAAQ;gBACR,cAAc,EAAE,iBAAiB;gBACjC,WAAW,EAAE,cAAc;gBAC3B,QAAQ;gBACR,KAAK,EAAE;oBACL,UAAU,EAAE,KAAK,CAAC,MAAM;oBACxB,aAAa,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,MAAM;oBACvD,eAAe,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,MAAM;oBAC1D,aAAa;iBACd;aACF,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,CAAC,KAAK,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;YAC9B,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IAKO,WAAW,CAAC,KAAoB,EAAE,GAAW;QACnD,MAAM,UAAU,GAAoB,EAAE,CAAC;QACvC,IAAI,WAAW,GAAkB,IAAI,CAAC;QAGtC,IAAI,SAAS,GAAG,CAAC,CAAC;QAClB,QAAQ,CAAC,GAAG,EAAE;YACZ,KAAK,CAAC,IAAI;gBACR,IAAI,SAAS,KAAK,KAAK,CAAC,EAAE,EAAE,CAAC;oBAC3B,WAAW,GAAG,IAAI,CAAC,IAAI,CAAC;oBACxB,IAAI,CAAC,IAAI,EAAE,CAAC;gBACd,CAAC;gBACD,SAAS,EAAE,CAAC;YACd,CAAC;SACF,CAAC,CAAC;QAEH,IAAI,CAAC,WAAW,EAAE,CAAC;YAEjB,OAAO,EAAE,CAAC;QACZ,CAAC;QAGD,IAAI,CAAC,CAAC,qBAAqB,CAAC,WAAW,CAAC,EAAE,CAAC;YAEzC,MAAM,QAAQ,GAAG,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;YACxC,MAAM,OAAO,GAAG,WAAoC,CAAC;YACrD,OAAO,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC,IAA0B,EAAE,EAAE;gBAC1D,IAAI,CAAC,CAAC,YAAY,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC;oBAC5B,MAAM,OAAO,GAAG,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC;oBAC7B,MAAM,aAAa,GAAG,IAAI,CAAC,mBAAmB,CAAC,SAAS,EAAE,OAAO,EAAE,OAAO,CAAC,CAAC;oBAC5E,QAAQ,CAAC,MAAM,CAAC,GAAG,CAAC,OAAO,EAAE,aAAa,CAAC,CAAC;gBAC9C,CAAC;YACH,CAAC,CAAC,CAAC;YACH,QAAQ,CAAC,EAAE,EAAE,CAAC;YACd,UAAU,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC5B,CAAC;aAAM,IAAI,CAAC,CAAC,aAAa,CAAC,WAAW,CAAC,EAAE,CAAC;YAExC,MAAM,SAAS,GAAG,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;YACzC,MAAM,UAAU,GAAG,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;YAG1C,MAAM,MAAM,GAAG,WAA4B,CAAC;YAC5C,MAAM,aAAa,GAAG,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;YACrD,SAAS,CAAC,eAAe,CAAC,IAAI,CAAC;gBAC7B,IAAI,EAAE,QAAQ;gBACd,UAAU,EAAE,aAAa;gBACzB,WAAW,EAAE,MAAM;aACpB,CAAC,CAAC;YACH,UAAU,CAAC,eAAe,CAAC,IAAI,CAAC;gBAC9B,IAAI,EAAE,QAAQ;gBACd,UAAU,EAAE,KAAK,aAAa,GAAG;gBACjC,WAAW,EAAE,MAAM;aACpB,CAAC,CAAC;YAEH,SAAS,CAAC,EAAE,EAAE,CAAC;YACf,UAAU,CAAC,EAAE,EAAE,CAAC;YAChB,UAAU,CAAC,IAAI,CAAC,SAAS,EAAE,UAAU,CAAC,CAAC;QACzC,CAAC;aAAM,IAAI,CAAC,CAAC,gBAAgB,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC,cAAc,CAAC,WAAW,CAAC,EAAE,CAAC;YAE5E,MAAM,UAAU,GAAG,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;YAC1C,MAAM,SAAS,GAAG,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;YAEzC,UAAU,CAAC,EAAE,EAAE,CAAC;YAChB,SAAS,CAAC,EAAE,IAAI,CAAC,CAAC;YAClB,UAAU,CAAC,IAAI,CAAC,UAAU,EAAE,SAAS,CAAC,CAAC;QACzC,CAAC;aAAM,IAAI,CAAC,CAAC,sBAAsB,CAAC,WAAW,CAAC,EAAE,CAAC;YAEjD,MAAM,QAAQ,GAAG,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;YACxC,MAAM,UAAU,GAAG,WAAqC,CAAC;YACzD,IAAI,CAAC,CAAC,YAAY,CAAC,UAAU,CAAC,IAAI,CAAC,EAAE,CAAC;gBACpC,MAAM,OAAO,GAAG,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC;gBACrC,MAAM,SAAS,GAAG,IAAI,CAAC,YAAY,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;gBACtD,MAAM,aAAa,GAAG,IAAI,CAAC,mBAAmB,CAAC,SAAS,EAAE,SAAS,EAAE,SAAS,CAAC,CAAC;gBAChF,QAAQ,CAAC,MAAM,CAAC,GAAG,CAAC,OAAO,EAAE,aAAa,CAAC,CAAC;YAC9C,CAAC;YACD,QAAQ,CAAC,EAAE,EAAE,CAAC;YACd,UAAU,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC5B,CAAC;aAAM,CAAC;YAEN,MAAM,QAAQ,GAAG,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;YACxC,QAAQ,CAAC,EAAE,EAAE,CAAC;YACd,UAAU,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC5B,CAAC;QAED,OAAO,UAAU,CAAC;IACpB,CAAC;IAKO,YAAY,CAAC,IAAY;QAC/B,IAAI,CAAC,CAAC,YAAY,CAAC,IAAI,CAAC,EAAE,CAAC;YACzB,OAAO,IAAI,CAAC,IAAI,CAAC;QACnB,CAAC;aAAM,IAAI,CAAC,CAAC,gBAAgB,CAAC,IAAI,CAAC,EAAE,CAAC;YACpC,OAAO,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC5B,CAAC;aAAM,IAAI,CAAC,CAAC,eAAe,CAAC,IAAI,CAAC,EAAE,CAAC;YACnC,OAAO,IAAI,IAAI,CAAC,KAAK,GAAG,CAAC;QAC3B,CAAC;aAAM,IAAI,CAAC,CAAC,kBAAkB,CAAC,IAAI,CAAC,EAAE,CAAC;YACtC,OAAO,GAAG,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,IAAI,CAAC,QAAQ,IAAI,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC;QAC7F,CAAC;aAAM,IAAI,CAAC,CAAC,iBAAiB,CAAC,IAAI,CAAC,EAAE,CAAC;YACrC,OAAO,GAAG,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC;QAC/D,CAAC;aAAM,CAAC;YACN,OAAO,sBAAsB,CAAC;QAChC,CAAC;IACH,CAAC;IAKO,eAAe,CAAC,KAAoB;QAE1C,IAAI,KAAK,CAAC,EAAE,GAAG,IAAI,EAAE,CAAC;YACpB,OAAO,IAAI,CAAC;QACd,CAAC;QAGD,IAAI,KAAK,CAAC,eAAe,CAAC,MAAM,GAAG,EAAE,EAAE,CAAC;YACtC,OAAO,IAAI,CAAC;QACd,CAAC;QAGD,IAAI,KAAK,CAAC,KAAK,CAAC,MAAM,KAAK,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,EAAE,CAAC;YACxD,OAAO,IAAI,CAAC;QACd,CAAC;QAED,OAAO,KAAK,CAAC;IACf,CAAC;IAKO,UAAU,CAAC,KAAoB;QACrC,MAAM,MAAM,GAAG,QAAQ,IAAI,CAAC,WAAW,EAAE,EAAE,CAAC;QAG5C,MAAM,QAAQ,GAAG,IAAI,CAAC,qBAAqB,CAAC,KAAK,CAAC,CAAC;QAEnD,OAAO;YACL,EAAE,EAAE,MAAM;YACV,MAAM,EAAE,CAAC,KAAK,CAAC;YACf,WAAW,EAAE,CAAC,GAAG,KAAK,CAAC,eAAe,CAAC;YACvC,UAAU,EAAE,IAAI,CAAC,oBAAoB,CAAC,KAAK,CAAC,eAAe,CAAC;YAC5D,QAAQ;SACT,CAAC;IACJ,CAAC;IAKO,qBAAqB,CAAC,KAAoB;QAEhD,OAAO,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,GAAG,GAAG,EAAE,GAAG,CAAC,CAAC;IACvC,CAAC;IAKO,oBAAoB,CAAC,WAAyB;QAEpD,MAAM,WAAW,GAAG,IAAI,GAAG,EAAU,CAAC;QAEtC,KAAK,MAAM,UAAU,IAAI,WAAW,EAAE,CAAC;YACrC,MAAM,IAAI,GAAG,UAAU,CAAC,UAAU,CAAC;YAGnC,IAAI,WAAW,CAAC,GAAG,CAAC,KAAK,IAAI,GAAG,CAAC,EAAE,CAAC;gBAClC,OAAO,KAAK,CAAC;YACf,CAAC;YAED,WAAW,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QACxB,CAAC;QAED,OAAO,IAAI,CAAC;IACd,CAAC;IAKO,qBAAqB,CAAC,KAAoB,EAAE,UAA2B;QAC7E,MAAM,IAAI,GAAG,IAAI,GAAG,EAAU,CAAC;QAG/B,KAAK,MAAM,KAAK,IAAI,KAAK,CAAC,KAAK,EAAE,CAAC;YAChC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC,EAAE,CAAC;gBACxB,UAAU,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;gBACvB,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;YACrB,CAAC;QACH,CAAC;QAGD,KAAK,MAAM,KAAK,IAAI,KAAK,CAAC,SAAS,CAAC,MAAM,EAAE,EAAE,CAAC;YAC7C,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC,EAAE,CAAC;gBACxB,UAAU,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;gBACvB,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;YACrB,CAAC;QACH,CAAC;QAGD,KAAK,MAAM,KAAK,IAAI,KAAK,CAAC,MAAM,CAAC,MAAM,EAAE,EAAE,CAAC;YAC1C,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC,EAAE,CAAC;gBACxB,UAAU,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;gBACvB,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;YACrB,CAAC;QACH,CAAC;IACH,CAAC;IAKO,kBAAkB,CAAC,KAAoB,EAAE,UAAwB;QACvE,MAAM,IAAI,GAAG,IAAI,GAAG,EAAU,CAAC;QAG/B,KAAK,MAAM,UAAU,IAAI,KAAK,CAAC,eAAe,EAAE,CAAC;YAC/C,MAAM,GAAG,GAAG,GAAG,UAAU,CAAC,IAAI,IAAI,UAAU,CAAC,UAAU,EAAE,CAAC;YAC1D,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC;gBACnB,UAAU,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;gBAC5B,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;YAChB,CAAC;QACH,CAAC;QAGD,MAAM,SAAS,GAAG;YAChB,GAAG,KAAK,CAAC,KAAK;YACd,GAAG,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,MAAM,EAAE,CAAC;YACvC,GAAG,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC;SACrC,CAAC;QAEF,KAAK,MAAM,KAAK,IAAI,SAAS,EAAE,CAAC;YAC9B,KAAK,MAAM,UAAU,IAAI,KAAK,CAAC,WAAW,EAAE,CAAC;gBAC3C,MAAM,GAAG,GAAG,GAAG,UAAU,CAAC,IAAI,IAAI,UAAU,CAAC,UAAU,EAAE,CAAC;gBAC1D,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC;oBACnB,UAAU,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;oBAC5B,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;gBAChB,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;IAKO,KAAK,CAAC,gBAAgB,CAAC,KAAsB,EAAE,QAAkB;QACvE,MAAM,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QAE5B,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YAEzB,MAAM,MAAM,GAAG,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;YAEtD,IAAI,CAAC,MAAM,CAAC,WAAW,EAAE,CAAC;gBACxB,IAAI,CAAC,UAAU,GAAG,KAAK,CAAC;gBACxB,QAAQ,CAAC,IAAI,CAAC,MAAM,IAAI,CAAC,EAAE,SAAS,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC;YACvD,CAAC;iBAAM,CAAC;gBACN,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC;YACzB,CAAC;QACH,CAAC;QAED,MAAM,CAAC,IAAI,CAAC,kBAAkB,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,MAAM,IAAI,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC;IAC5F,CAAC;IAKO,eAAe,CAAC,WAAyB;QAE/C,MAAM,kBAAkB,GAAG,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,OAAO,IAAI,CAAC,CAAC,IAAI,KAAK,YAAY,CAAC,CAAC;QAEpG,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,kBAAkB,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YACnD,KAAK,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,kBAAkB,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;gBACvD,MAAM,EAAE,GAAG,kBAAkB,CAAC,CAAC,CAAC,CAAC;gBACjC,MAAM,EAAE,GAAG,kBAAkB,CAAC,CAAC,CAAC,CAAC;gBAEjC,IAAI,CAAC,EAAE,IAAI,CAAC,EAAE;oBAAE,SAAS;gBAGzB,IAAI,IAAI,CAAC,gBAAgB,CAAC,EAAE,CAAC,UAAU,EAAE,EAAE,CAAC,UAAU,CAAC,EAAE,CAAC;oBACxD,OAAO;wBACL,WAAW,EAAE,KAAK;wBAClB,MAAM,EAAE,SAAS,EAAE,CAAC,UAAU,MAAM,EAAE,CAAC,UAAU,EAAE;qBACpD,CAAC;gBACJ,CAAC;YACH,CAAC;QACH,CAAC;QAED,OAAO,EAAE,WAAW,EAAE,IAAI,EAAE,CAAC;IAC/B,CAAC;IAKO,gBAAgB,CAAC,KAAa,EAAE,KAAa;QAGnD,MAAM,QAAQ,GAAG,mBAAmB,CAAC;QACrC,MAAM,QAAQ,GAAG,mBAAmB,CAAC;QAErC,MAAM,MAAM,GAAG,KAAK,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;QACrC,MAAM,MAAM,GAAG,KAAK,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;QAErC,IAAI,MAAM,IAAI,MAAM,IAAI,MAAM,CAAC,CAAC,CAAC,KAAK,MAAM,CAAC,CAAC,CAAC,IAAI,MAAM,CAAC,CAAC,CAAC,IAAI,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC;YAC1E,MAAM,IAAI,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;YACrC,MAAM,IAAI,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;YACrC,OAAO,IAAI,IAAI,IAAI,CAAC;QACtB,CAAC;QAED,OAAO,KAAK,CAAC;IACf,CAAC;IAKO,iBAAiB,CAAC,KAAsB,EAAE,GAAW;QAE3D,IAAI,eAAe,GAAG,CAAC,CAAC;QACxB,QAAQ,CAAC,GAAG,EAAE;YACZ,SAAS;gBACP,eAAe,EAAE,CAAC;YACpB,CAAC;SACF,CAAC,CAAC;QAEH,IAAI,eAAe,KAAK,CAAC,EAAE,CAAC;YAC1B,OAAO,CAAC,CAAC;QACX,CAAC;QAGD,MAAM,iBAAiB,GAAG,IAAI,GAAG,EAAU,CAAC;QAC5C,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,KAAK,MAAM,KAAK,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;gBAChC,iBAAiB,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;YAClC,CAAC;QACH,CAAC;QAED,OAAO,iBAAiB,CAAC,IAAI,GAAG,eAAe,CAAC;IAClD,CAAC;IAKO,UAAU,CAAC,KAAoB;QACrC,OAAO;YACL,EAAE,EAAE,KAAK,CAAC,EAAE;YACZ,KAAK,EAAE,CAAC,GAAG,KAAK,CAAC,KAAK,CAAC;YACvB,SAAS,EAAE,IAAI,GAAG,CAAC,KAAK,CAAC,SAAS,CAAC;YACnC,MAAM,EAAE,IAAI,GAAG,CAAC,KAAK,CAAC,MAAM,CAAC;YAC7B,eAAe,EAAE,CAAC,GAAG,KAAK,CAAC,eAAe,CAAC;SAC5C,CAAC;IACJ,CAAC;IAKD,mBAAmB,CACjB,IAAuB,EACvB,IAAY,EACZ,MAAe;QAEf,OAAO;YACL,EAAE,EAAE,OAAO,IAAI,CAAC,aAAa,EAAE,EAAE;YACjC,IAAI;YACJ,IAAI;YACJ,WAAW,EAAE,EAAE;YACf,MAAM;SACP,CAAC;IACJ,CAAC;IAKD,aAAa,CACX,KAAoB,EACpB,IAAwB,EACxB,UAAkB,EAClB,WAAmB;QAEnB,KAAK,CAAC,WAAW,CAAC,IAAI,CAAC;YACrB,IAAI;YACJ,UAAU;YACV,WAAW;SACZ,CAAC,CAAC;IACL,CAAC;CACF"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"AIHookToolDefinitions.d.ts","sourceRoot":"","sources":["../../src/server/AIHookToolDefinitions.ts"],"names":[],"mappings":"AAMA,OAAO,KAAK,EAAE,IAAI,EAAE,MAAM,oCAAoC,CAAC;AAE/D,eAAO,MAAM,WAAW,EAAE,IAAI,EAgS7B,CAAC"}
|
|
@@ -0,0 +1,284 @@
|
|
|
1
|
+
export const aiHookTools = [
|
|
2
|
+
{
|
|
3
|
+
name: 'ai_hook_generate',
|
|
4
|
+
description: `🤖 AI驱动的Hook代码生成器
|
|
5
|
+
|
|
6
|
+
**功能**:根据自然语言描述自动生成Hook代码
|
|
7
|
+
|
|
8
|
+
**使用场景**:
|
|
9
|
+
1. 分析目标网站后,发现需要Hook某个特定函数
|
|
10
|
+
2. 描述Hook需求(例如:"Hook所有加密相关的函数调用")
|
|
11
|
+
3. 自动生成对应的Hook代码
|
|
12
|
+
4. 使用 ai_hook_inject 注入到浏览器
|
|
13
|
+
|
|
14
|
+
**支持的Hook类型**:
|
|
15
|
+
- function: Hook全局函数(如 window.btoa, window.atob)
|
|
16
|
+
- object-method: Hook对象方法(如 crypto.subtle.encrypt)
|
|
17
|
+
- api: Hook浏览器API(如 fetch, XMLHttpRequest)
|
|
18
|
+
- property: Hook对象属性
|
|
19
|
+
- event: Hook事件监听器
|
|
20
|
+
- custom: 自定义Hook代码
|
|
21
|
+
|
|
22
|
+
**示例**:
|
|
23
|
+
\`\`\`json
|
|
24
|
+
{
|
|
25
|
+
"description": "Hook所有fetch请求,捕获URL和响应",
|
|
26
|
+
"target": {
|
|
27
|
+
"type": "api",
|
|
28
|
+
"name": "fetch"
|
|
29
|
+
},
|
|
30
|
+
"behavior": {
|
|
31
|
+
"captureArgs": true,
|
|
32
|
+
"captureReturn": true,
|
|
33
|
+
"logToConsole": true
|
|
34
|
+
},
|
|
35
|
+
"condition": {
|
|
36
|
+
"urlPattern": "api\\\\.example\\\\.com"
|
|
37
|
+
}
|
|
38
|
+
}
|
|
39
|
+
\`\`\``,
|
|
40
|
+
inputSchema: {
|
|
41
|
+
type: 'object',
|
|
42
|
+
properties: {
|
|
43
|
+
description: {
|
|
44
|
+
type: 'string',
|
|
45
|
+
description: 'Hook的自然语言描述(例如:"Hook所有加密函数")',
|
|
46
|
+
},
|
|
47
|
+
target: {
|
|
48
|
+
type: 'object',
|
|
49
|
+
description: 'Hook目标',
|
|
50
|
+
properties: {
|
|
51
|
+
type: {
|
|
52
|
+
type: 'string',
|
|
53
|
+
enum: ['function', 'object-method', 'api', 'property', 'event', 'custom'],
|
|
54
|
+
description: 'Hook类型',
|
|
55
|
+
},
|
|
56
|
+
name: {
|
|
57
|
+
type: 'string',
|
|
58
|
+
description: '函数名或API名(如 "btoa", "fetch")',
|
|
59
|
+
},
|
|
60
|
+
pattern: {
|
|
61
|
+
type: 'string',
|
|
62
|
+
description: '正则匹配模式(用于匹配多个函数)',
|
|
63
|
+
},
|
|
64
|
+
object: {
|
|
65
|
+
type: 'string',
|
|
66
|
+
description: '对象路径(如 "window.crypto.subtle")',
|
|
67
|
+
},
|
|
68
|
+
property: {
|
|
69
|
+
type: 'string',
|
|
70
|
+
description: '属性名或方法名',
|
|
71
|
+
},
|
|
72
|
+
},
|
|
73
|
+
required: ['type'],
|
|
74
|
+
},
|
|
75
|
+
behavior: {
|
|
76
|
+
type: 'object',
|
|
77
|
+
description: 'Hook行为配置',
|
|
78
|
+
properties: {
|
|
79
|
+
captureArgs: {
|
|
80
|
+
type: 'boolean',
|
|
81
|
+
description: '是否捕获函数参数',
|
|
82
|
+
default: true,
|
|
83
|
+
},
|
|
84
|
+
captureReturn: {
|
|
85
|
+
type: 'boolean',
|
|
86
|
+
description: '是否捕获返回值',
|
|
87
|
+
default: true,
|
|
88
|
+
},
|
|
89
|
+
captureStack: {
|
|
90
|
+
type: 'boolean',
|
|
91
|
+
description: '是否捕获调用栈',
|
|
92
|
+
default: false,
|
|
93
|
+
},
|
|
94
|
+
modifyArgs: {
|
|
95
|
+
type: 'boolean',
|
|
96
|
+
description: '是否修改参数',
|
|
97
|
+
default: false,
|
|
98
|
+
},
|
|
99
|
+
modifyReturn: {
|
|
100
|
+
type: 'boolean',
|
|
101
|
+
description: '是否修改返回值',
|
|
102
|
+
default: false,
|
|
103
|
+
},
|
|
104
|
+
blockExecution: {
|
|
105
|
+
type: 'boolean',
|
|
106
|
+
description: '是否阻止函数执行',
|
|
107
|
+
default: false,
|
|
108
|
+
},
|
|
109
|
+
logToConsole: {
|
|
110
|
+
type: 'boolean',
|
|
111
|
+
description: '是否输出到控制台',
|
|
112
|
+
default: true,
|
|
113
|
+
},
|
|
114
|
+
},
|
|
115
|
+
},
|
|
116
|
+
condition: {
|
|
117
|
+
type: 'object',
|
|
118
|
+
description: '条件过滤',
|
|
119
|
+
properties: {
|
|
120
|
+
argFilter: {
|
|
121
|
+
type: 'string',
|
|
122
|
+
description: '参数过滤条件(JS表达式,如 "args[0].includes(\'password\')")',
|
|
123
|
+
},
|
|
124
|
+
returnFilter: {
|
|
125
|
+
type: 'string',
|
|
126
|
+
description: '返回值过滤条件',
|
|
127
|
+
},
|
|
128
|
+
urlPattern: {
|
|
129
|
+
type: 'string',
|
|
130
|
+
description: 'URL匹配模式(正则表达式字符串)',
|
|
131
|
+
},
|
|
132
|
+
maxCalls: {
|
|
133
|
+
type: 'number',
|
|
134
|
+
description: '最大捕获调用次数',
|
|
135
|
+
},
|
|
136
|
+
},
|
|
137
|
+
},
|
|
138
|
+
customCode: {
|
|
139
|
+
type: 'object',
|
|
140
|
+
description: '自定义代码片段',
|
|
141
|
+
properties: {
|
|
142
|
+
before: {
|
|
143
|
+
type: 'string',
|
|
144
|
+
description: '在函数执行前运行的代码',
|
|
145
|
+
},
|
|
146
|
+
after: {
|
|
147
|
+
type: 'string',
|
|
148
|
+
description: '在函数执行后运行的代码',
|
|
149
|
+
},
|
|
150
|
+
replace: {
|
|
151
|
+
type: 'string',
|
|
152
|
+
description: '完全替换原函数的代码',
|
|
153
|
+
},
|
|
154
|
+
},
|
|
155
|
+
},
|
|
156
|
+
},
|
|
157
|
+
required: ['description', 'target', 'behavior'],
|
|
158
|
+
},
|
|
159
|
+
},
|
|
160
|
+
{
|
|
161
|
+
name: 'ai_hook_inject',
|
|
162
|
+
description: `注入AI生成的Hook代码到浏览器
|
|
163
|
+
|
|
164
|
+
**注入方法**:
|
|
165
|
+
- evaluateOnNewDocument: 在新文档加载前注入(适用于API Hook,如fetch、XHR)
|
|
166
|
+
- evaluate: 在当前页面注入(适用于函数Hook、事件Hook)
|
|
167
|
+
|
|
168
|
+
**注意**:
|
|
169
|
+
- API Hook必须在页面加载前注入才能生效
|
|
170
|
+
- 函数Hook可以在页面加载后注入`,
|
|
171
|
+
inputSchema: {
|
|
172
|
+
type: 'object',
|
|
173
|
+
properties: {
|
|
174
|
+
hookId: {
|
|
175
|
+
type: 'string',
|
|
176
|
+
description: 'Hook ID(从 ai_hook_generate 返回)',
|
|
177
|
+
},
|
|
178
|
+
code: {
|
|
179
|
+
type: 'string',
|
|
180
|
+
description: 'Hook代码(从 ai_hook_generate 返回)',
|
|
181
|
+
},
|
|
182
|
+
method: {
|
|
183
|
+
type: 'string',
|
|
184
|
+
enum: ['evaluateOnNewDocument', 'evaluate'],
|
|
185
|
+
description: '注入方法',
|
|
186
|
+
default: 'evaluate',
|
|
187
|
+
},
|
|
188
|
+
},
|
|
189
|
+
required: ['hookId', 'code'],
|
|
190
|
+
},
|
|
191
|
+
},
|
|
192
|
+
{
|
|
193
|
+
name: 'ai_hook_get_data',
|
|
194
|
+
description: `获取Hook捕获的数据
|
|
195
|
+
|
|
196
|
+
返回指定Hook捕获的所有调用记录,包括:
|
|
197
|
+
- 参数
|
|
198
|
+
- 返回值
|
|
199
|
+
- 调用栈
|
|
200
|
+
- 时间戳
|
|
201
|
+
- 调用次数`,
|
|
202
|
+
inputSchema: {
|
|
203
|
+
type: 'object',
|
|
204
|
+
properties: {
|
|
205
|
+
hookId: {
|
|
206
|
+
type: 'string',
|
|
207
|
+
description: 'Hook ID',
|
|
208
|
+
},
|
|
209
|
+
},
|
|
210
|
+
required: ['hookId'],
|
|
211
|
+
},
|
|
212
|
+
},
|
|
213
|
+
{
|
|
214
|
+
name: 'ai_hook_list',
|
|
215
|
+
description: `列出所有已注入的Hook
|
|
216
|
+
|
|
217
|
+
返回所有活动Hook的列表,包括:
|
|
218
|
+
- Hook ID
|
|
219
|
+
- 创建时间
|
|
220
|
+
- 启用状态
|
|
221
|
+
- 捕获的记录数量`,
|
|
222
|
+
inputSchema: {
|
|
223
|
+
type: 'object',
|
|
224
|
+
properties: {},
|
|
225
|
+
},
|
|
226
|
+
},
|
|
227
|
+
{
|
|
228
|
+
name: 'ai_hook_clear',
|
|
229
|
+
description: `清除Hook捕获的数据
|
|
230
|
+
|
|
231
|
+
可以清除特定Hook的数据,或清除所有Hook的数据`,
|
|
232
|
+
inputSchema: {
|
|
233
|
+
type: 'object',
|
|
234
|
+
properties: {
|
|
235
|
+
hookId: {
|
|
236
|
+
type: 'string',
|
|
237
|
+
description: 'Hook ID(可选,不提供则清除所有Hook数据)',
|
|
238
|
+
},
|
|
239
|
+
},
|
|
240
|
+
},
|
|
241
|
+
},
|
|
242
|
+
{
|
|
243
|
+
name: 'ai_hook_toggle',
|
|
244
|
+
description: `启用或禁用Hook
|
|
245
|
+
|
|
246
|
+
禁用的Hook不会捕获新的调用,但已捕获的数据会保留`,
|
|
247
|
+
inputSchema: {
|
|
248
|
+
type: 'object',
|
|
249
|
+
properties: {
|
|
250
|
+
hookId: {
|
|
251
|
+
type: 'string',
|
|
252
|
+
description: 'Hook ID',
|
|
253
|
+
},
|
|
254
|
+
enabled: {
|
|
255
|
+
type: 'boolean',
|
|
256
|
+
description: '是否启用',
|
|
257
|
+
},
|
|
258
|
+
},
|
|
259
|
+
required: ['hookId', 'enabled'],
|
|
260
|
+
},
|
|
261
|
+
},
|
|
262
|
+
{
|
|
263
|
+
name: 'ai_hook_export',
|
|
264
|
+
description: `导出Hook数据
|
|
265
|
+
|
|
266
|
+
支持导出为JSON或CSV格式`,
|
|
267
|
+
inputSchema: {
|
|
268
|
+
type: 'object',
|
|
269
|
+
properties: {
|
|
270
|
+
hookId: {
|
|
271
|
+
type: 'string',
|
|
272
|
+
description: 'Hook ID(可选,不提供则导出所有Hook)',
|
|
273
|
+
},
|
|
274
|
+
format: {
|
|
275
|
+
type: 'string',
|
|
276
|
+
enum: ['json', 'csv'],
|
|
277
|
+
description: '导出格式',
|
|
278
|
+
default: 'json',
|
|
279
|
+
},
|
|
280
|
+
},
|
|
281
|
+
},
|
|
282
|
+
},
|
|
283
|
+
];
|
|
284
|
+
//# sourceMappingURL=AIHookToolDefinitions.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"AIHookToolDefinitions.js","sourceRoot":"","sources":["../../src/server/AIHookToolDefinitions.ts"],"names":[],"mappings":"AAQA,MAAM,CAAC,MAAM,WAAW,GAAW;IACjC;QACE,IAAI,EAAE,kBAAkB;QACxB,WAAW,EAAE;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;OAmCV;QACH,WAAW,EAAE;YACX,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE;gBACV,WAAW,EAAE;oBACX,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,8BAA8B;iBAC5C;gBACD,MAAM,EAAE;oBACN,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,QAAQ;oBACrB,UAAU,EAAE;wBACV,IAAI,EAAE;4BACJ,IAAI,EAAE,QAAQ;4BACd,IAAI,EAAE,CAAC,UAAU,EAAE,eAAe,EAAE,KAAK,EAAE,UAAU,EAAE,OAAO,EAAE,QAAQ,CAAC;4BACzE,WAAW,EAAE,QAAQ;yBACtB;wBACD,IAAI,EAAE;4BACJ,IAAI,EAAE,QAAQ;4BACd,WAAW,EAAE,6BAA6B;yBAC3C;wBACD,OAAO,EAAE;4BACP,IAAI,EAAE,QAAQ;4BACd,WAAW,EAAE,kBAAkB;yBAChC;wBACD,MAAM,EAAE;4BACN,IAAI,EAAE,QAAQ;4BACd,WAAW,EAAE,gCAAgC;yBAC9C;wBACD,QAAQ,EAAE;4BACR,IAAI,EAAE,QAAQ;4BACd,WAAW,EAAE,SAAS;yBACvB;qBACF;oBACD,QAAQ,EAAE,CAAC,MAAM,CAAC;iBACnB;gBACD,QAAQ,EAAE;oBACR,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,UAAU;oBACvB,UAAU,EAAE;wBACV,WAAW,EAAE;4BACX,IAAI,EAAE,SAAS;4BACf,WAAW,EAAE,UAAU;4BACvB,OAAO,EAAE,IAAI;yBACd;wBACD,aAAa,EAAE;4BACb,IAAI,EAAE,SAAS;4BACf,WAAW,EAAE,SAAS;4BACtB,OAAO,EAAE,IAAI;yBACd;wBACD,YAAY,EAAE;4BACZ,IAAI,EAAE,SAAS;4BACf,WAAW,EAAE,SAAS;4BACtB,OAAO,EAAE,KAAK;yBACf;wBACD,UAAU,EAAE;4BACV,IAAI,EAAE,SAAS;4BACf,WAAW,EAAE,QAAQ;4BACrB,OAAO,EAAE,KAAK;yBACf;wBACD,YAAY,EAAE;4BACZ,IAAI,EAAE,SAAS;4BACf,WAAW,EAAE,SAAS;4BACtB,OAAO,EAAE,KAAK;yBACf;wBACD,cAAc,EAAE;4BACd,IAAI,EAAE,SAAS;4BACf,WAAW,EAAE,UAAU;4BACvB,OAAO,EAAE,KAAK;yBACf;wBACD,YAAY,EAAE;4BACZ,IAAI,EAAE,SAAS;4BACf,WAAW,EAAE,UAAU;4BACvB,OAAO,EAAE,IAAI;yBACd;qBACF;iBACF;gBACD,SAAS,EAAE;oBACT,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,MAAM;oBACnB,UAAU,EAAE;wBACV,SAAS,EAAE;4BACT,IAAI,EAAE,QAAQ;4BACd,WAAW,EAAE,kDAAkD;yBAChE;wBACD,YAAY,EAAE;4BACZ,IAAI,EAAE,QAAQ;4BACd,WAAW,EAAE,SAAS;yBACvB;wBACD,UAAU,EAAE;4BACV,IAAI,EAAE,QAAQ;4BACd,WAAW,EAAE,mBAAmB;yBACjC;wBACD,QAAQ,EAAE;4BACR,IAAI,EAAE,QAAQ;4BACd,WAAW,EAAE,UAAU;yBACxB;qBACF;iBACF;gBACD,UAAU,EAAE;oBACV,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,SAAS;oBACtB,UAAU,EAAE;wBACV,MAAM,EAAE;4BACN,IAAI,EAAE,QAAQ;4BACd,WAAW,EAAE,aAAa;yBAC3B;wBACD,KAAK,EAAE;4BACL,IAAI,EAAE,QAAQ;4BACd,WAAW,EAAE,aAAa;yBAC3B;wBACD,OAAO,EAAE;4BACP,IAAI,EAAE,QAAQ;4BACd,WAAW,EAAE,YAAY;yBAC1B;qBACF;iBACF;aACF;YACD,QAAQ,EAAE,CAAC,aAAa,EAAE,QAAQ,EAAE,UAAU,CAAC;SAChD;KACF;IAED;QACE,IAAI,EAAE,gBAAgB;QACtB,WAAW,EAAE;;;;;;;;mBAQE;QACf,WAAW,EAAE;YACX,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE;gBACV,MAAM,EAAE;oBACN,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,gCAAgC;iBAC9C;gBACD,IAAI,EAAE;oBACJ,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,+BAA+B;iBAC7C;gBACD,MAAM,EAAE;oBACN,IAAI,EAAE,QAAQ;oBACd,IAAI,EAAE,CAAC,uBAAuB,EAAE,UAAU,CAAC;oBAC3C,WAAW,EAAE,MAAM;oBACnB,OAAO,EAAE,UAAU;iBACpB;aACF;YACD,QAAQ,EAAE,CAAC,QAAQ,EAAE,MAAM,CAAC;SAC7B;KACF;IAED;QACE,IAAI,EAAE,kBAAkB;QACxB,WAAW,EAAE;;;;;;;OAOV;QACH,WAAW,EAAE;YACX,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE;gBACV,MAAM,EAAE;oBACN,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,SAAS;iBACvB;aACF;YACD,QAAQ,EAAE,CAAC,QAAQ,CAAC;SACrB;KACF;IAED;QACE,IAAI,EAAE,cAAc;QACpB,WAAW,EAAE;;;;;;UAMP;QACN,WAAW,EAAE;YACX,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE,EAAE;SACf;KACF;IAED;QACE,IAAI,EAAE,eAAe;QACrB,WAAW,EAAE;;2BAEU;QACvB,WAAW,EAAE;YACX,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE;gBACV,MAAM,EAAE;oBACN,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,4BAA4B;iBAC1C;aACF;SACF;KACF;IAED;QACE,IAAI,EAAE,gBAAgB;QACtB,WAAW,EAAE;;2BAEU;QACvB,WAAW,EAAE;YACX,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE;gBACV,MAAM,EAAE;oBACN,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,SAAS;iBACvB;gBACD,OAAO,EAAE;oBACP,IAAI,EAAE,SAAS;oBACf,WAAW,EAAE,MAAM;iBACpB;aACF;YACD,QAAQ,EAAE,CAAC,QAAQ,EAAE,SAAS,CAAC;SAChC;KACF;IAED;QACE,IAAI,EAAE,gBAAgB;QACtB,WAAW,EAAE;;gBAED;QACZ,WAAW,EAAE;YACX,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE;gBACV,MAAM,EAAE;oBACN,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,0BAA0B;iBACxC;gBACD,MAAM,EAAE;oBACN,IAAI,EAAE,QAAQ;oBACd,IAAI,EAAE,CAAC,MAAM,EAAE,KAAK,CAAC;oBACrB,WAAW,EAAE,MAAM;oBACnB,OAAO,EAAE,MAAM;iBAChB;aACF;SACF;KACF;CACF,CAAC"}
|