168 lines
3.8 KiB
Python
168 lines
3.8 KiB
Python
"""
|
||
Agent 工具集
|
||
|
||
提供 Agent 使用的各种工具，包括：
|
||
- 基础工具（文件操作、代码搜索）
|
||
- 分析工具（模式匹配、数据流分析）
|
||
- 外部安全工具（Semgrep、Bandit等）
|
||
- 协作工具（Think、Agent通信）
|
||
- 报告工具（漏洞报告）
|
||
- 🔥 智能扫描工具（批量扫描、快速审计）
|
||
"""
|
||
|
||
from .base import AgentTool, ToolResult
|
||
from .rag_tool import RAGQueryTool, SecurityCodeSearchTool, FunctionContextTool
|
||
from .pattern_tool import PatternMatchTool
|
||
from .code_analysis_tool import CodeAnalysisTool, DataFlowAnalysisTool, VulnerabilityValidationTool
|
||
from .file_tool import FileReadTool, FileSearchTool, ListFilesTool
|
||
from .sandbox_tool import (
|
||
SandboxTool,
|
||
SandboxHttpTool,
|
||
VulnerabilityVerifyTool,
|
||
SandboxManager,
|
||
)
|
||
|
||
# 🔥 多语言代码测试工具
|
||
from .sandbox_language import (
|
||
PhpTestTool,
|
||
PythonTestTool,
|
||
JavaScriptTestTool,
|
||
JavaTestTool,
|
||
GoTestTool,
|
||
RubyTestTool,
|
||
ShellTestTool,
|
||
UniversalCodeTestTool,
|
||
)
|
||
|
||
# 🔥 漏洞验证专用工具
|
||
from .sandbox_vuln import (
|
||
CommandInjectionTestTool,
|
||
SqlInjectionTestTool,
|
||
XssTestTool,
|
||
PathTraversalTestTool,
|
||
SstiTestTool,
|
||
DeserializationTestTool,
|
||
UniversalVulnTestTool,
|
||
)
|
||
|
||
# 外部安全工具
|
||
from .external_tools import (
|
||
SemgrepTool,
|
||
BanditTool,
|
||
GitleaksTool,
|
||
NpmAuditTool,
|
||
SafetyTool,
|
||
TruffleHogTool,
|
||
OSVScannerTool,
|
||
)
|
||
|
||
# 🔥 新增：思考和推理工具
|
||
from .thinking_tool import ThinkTool, ReflectTool
|
||
|
||
# 🔥 新增：漏洞报告工具
|
||
from .reporting_tool import CreateVulnerabilityReportTool
|
||
|
||
# 🔥 新增：扫描完成工具
|
||
from .finish_tool import FinishScanTool
|
||
|
||
# 🔥 新增：Agent协作工具
|
||
from .agent_tools import (
|
||
CreateSubAgentTool,
|
||
SendMessageTool,
|
||
ViewAgentGraphTool,
|
||
WaitForMessageTool,
|
||
AgentFinishTool,
|
||
RunSubAgentsTool,
|
||
CollectSubAgentResultsTool,
|
||
)
|
||
|
||
# 🔥 新增：智能扫描工具
|
||
from .smart_scan_tool import SmartScanTool, QuickAuditTool
|
||
|
||
# 🔥 新增：Kunlun-M 静态代码分析工具 (MIT License)
|
||
from .kunlun_tool import KunlunMTool, KunlunRuleListTool, KunlunPluginTool
|
||
|
||
__all__ = [
|
||
# 基础
|
||
"AgentTool",
|
||
"ToolResult",
|
||
|
||
# RAG 工具
|
||
"RAGQueryTool",
|
||
"SecurityCodeSearchTool",
|
||
"FunctionContextTool",
|
||
|
||
# 代码分析
|
||
"PatternMatchTool",
|
||
"CodeAnalysisTool",
|
||
"DataFlowAnalysisTool",
|
||
"VulnerabilityValidationTool",
|
||
|
||
# 文件操作
|
||
"FileReadTool",
|
||
"FileSearchTool",
|
||
"ListFilesTool",
|
||
|
||
# 沙箱
|
||
"SandboxTool",
|
||
"SandboxHttpTool",
|
||
"VulnerabilityVerifyTool",
|
||
"SandboxManager",
|
||
|
||
# 🔥 多语言代码测试工具
|
||
"PhpTestTool",
|
||
"PythonTestTool",
|
||
"JavaScriptTestTool",
|
||
"JavaTestTool",
|
||
"GoTestTool",
|
||
"RubyTestTool",
|
||
"ShellTestTool",
|
||
"UniversalCodeTestTool",
|
||
|
||
# 🔥 漏洞验证专用工具
|
||
"CommandInjectionTestTool",
|
||
"SqlInjectionTestTool",
|
||
"XssTestTool",
|
||
"PathTraversalTestTool",
|
||
"SstiTestTool",
|
||
"DeserializationTestTool",
|
||
"UniversalVulnTestTool",
|
||
|
||
# 外部安全工具
|
||
"SemgrepTool",
|
||
"BanditTool",
|
||
"GitleaksTool",
|
||
"NpmAuditTool",
|
||
"SafetyTool",
|
||
"TruffleHogTool",
|
||
"OSVScannerTool",
|
||
|
||
# 🔥 思考和推理工具
|
||
"ThinkTool",
|
||
"ReflectTool",
|
||
|
||
# 🔥 漏洞报告工具
|
||
"CreateVulnerabilityReportTool",
|
||
|
||
# 🔥 扫描完成工具
|
||
"FinishScanTool",
|
||
|
||
# 🔥 Agent协作工具
|
||
"CreateSubAgentTool",
|
||
"SendMessageTool",
|
||
"ViewAgentGraphTool",
|
||
"WaitForMessageTool",
|
||
"AgentFinishTool",
|
||
"RunSubAgentsTool",
|
||
"CollectSubAgentResultsTool",
|
||
|
||
# 🔥 智能扫描工具
|
||
"SmartScanTool",
|
||
"QuickAuditTool",
|
||
|
||
# 🔥 Kunlun-M 工具 (MIT License - https://github.com/LoRexxar/Kunlun-M)
|
||
"KunlunMTool",
|
||
"KunlunRuleListTool",
|
||
"KunlunPluginTool",
|
||
]
|