File size: 3,751 Bytes
ffb6330 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 |
"""
工具扫描CLI模块
处理工具扫描相关的命令行功能
"""
from deepteam.plugin_system.tool_scanner import ToolScanner
def handle_tool_scanning(args):
"""处理工具扫描相关命令"""
scanner = ToolScanner()
# 添加插件路径
if args.plugins:
for plugin_path in args.plugins:
scanner.add_plugin_path(plugin_path)
# 扫描所有工具
tools_info = scanner.scan_all_tools()
if args.scan_tools:
print("=== 可用工具列表 ===")
# 根据参数选择要显示的工具类型
tool_types = []
if args.scan_tools == "all":
tool_types = ['attack', 'metric', 'vulnerability']
elif args.scan_tools == "techniques":
tool_types = ['attack']
elif args.scan_tools == "metrics":
tool_types = ['metric']
elif args.scan_tools == "scenarios":
tool_types = ['vulnerability']
for tool_type in tool_types:
# 显示用户友好的类型名称
display_name = {
'attack': 'TECHNIQUES (攻击技术)',
'metric': 'METRICS (评估指标)',
'vulnerability': 'SCENARIOS (测试场景)'
}.get(tool_type, tool_type.upper())
print(f"\n## {display_name}:")
tools_of_type = scanner.get_tools_by_type(tool_type)
if tools_of_type:
for tool_name, tool_info in tools_of_type.items():
print(f" - {tool_name}")
if tool_info['parameters']:
for param_name, param_info in tool_info['parameters'].items():
required = "必需" if param_info['required'] else "可选"
default_str = f" (默认: {param_info['default']})" if param_info['default'] is not None else ""
print(f" * {param_name} ({required}){default_str}")
if param_info['description']:
print(f" {param_info['description']}")
else:
print(" (无可用工具)")
# # 显示验证警告
# warnings = scanner.validate_tool_completeness()
# if warnings:
# print("\n=== 验证警告 ===")
# for warning in warnings:
# print(warning)
return True
if args.show_tool_params:
tool_name = args.show_tool_params
tool_info = scanner.get_tool_info(tool_name)
if tool_info:
print(f"=== {tool_name} 详细信息 ===")
print(f"类型: {tool_info['type']}")
print(f"文件: {tool_info['file']}")
if tool_info['description']:
print(f"描述: {tool_info['description']}")
if tool_info['parameters']:
print("\n参数:")
for param_name, param_info in tool_info['parameters'].items():
required = "必需" if param_info['required'] else "可选"
default_str = f" (默认值: {param_info['default']})" if param_info['default'] is not None else ""
print(f" {param_name} ({required}){default_str}")
if param_info['description']:
print(f" 描述: {param_info['description']}")
else:
print("\n参数: (无参数)")
else:
print(f"错误: 找不到工具 '{tool_name}'")
print("可用工具:")
for tool_name in tools_info.keys():
print(f" - {tool_name}")
return True
return False |