create-dwy 0.5.1 → 0.6.0

package/templates/claude-global/skills/dwy-pentest/SKILL.md

@@ -0,0 +1,344 @@
+---
+name: dwy-pentest
+description: "白帽渗透测试：对指定服务器执行全面安全评估。触发条件：用户说'渗透测试'、'pentest'、'安全测试'、'漏洞扫描'、'安全审计' 时。"
+---
+
+# 白帽渗透测试
+
+对指定目标执行全面安全评估，发现漏洞后报警提示，不执行破坏性操作。
+
+**脚本目录：** `~/.claude/skills/dwy-pentest/scripts/`（以下简称 `{scripts}`）
+
+## 流程总览
+
+```
+输入目标 → 环境准备 → 信息收集 → 端口扫描 → 漏洞扫描 → Web 测试 → SSL 审计 → 认证测试 → 漏洞验证 → 报告生成 → 安全告警
+```
+
+**共 10 步，不可跳步，不可调换顺序。**
+
+---
+
+## 第 0 步：获取目标
+
+向用户询问测试目标，支持以下格式：
+- IP 地址：`192.168.1.100`
+- 域名：`example.com`
+- URL：`https://example.com:8443`
+- IP:端口：`10.0.0.1:8080`
+
+```
+请输入要测试的目标（IP/域名/URL）：
+```
+
+收到目标后：
+1. 确认目标格式有效
+2. 如果是私网/本地地址，提示用户确认
+3. 创建输出目录：`{项目根目录}/.pentest-results/{YYYYMMDD-HHmmss}/`
+4. 将输出目录路径记为 `{output}`，后续所有步骤引用此路径
+
+---
+
+## 第 1 步：环境准备
+
+检查并安装所需工具：
+
+```bash
+bash {scripts}/setup_tools.sh
+```
+
+如果脚本 Python 依赖未安装：
+
+```bash
+cd {scripts} && uv sync
+```
+
+**工具清单（必需 + 可选）：**
+
+| 工具 | 必需 | 用途 | 不可用时的降级 |
+|------|------|------|--------------|
+| nmap | 是 | 端口扫描 + NSE 漏洞脚本 | 无法降级，必须安装 |
+| nuclei | 否 | 模板驱动漏洞扫描 | 仅用 nmap NSE |
+| testssl.sh | 否 | SSL/TLS 审计 | 用 Python ssl 模块 |
+| gobuster | 否 | 目录枚举 | 用 ffuf 或内置路径列表 |
+| sqlmap | 否 | SQL 注入检测 | 用基础 HTTP 探针 |
+| hydra | 否 | 凭据暴力测试 | 跳过认证测试 |
+| ffuf | 否 | Web fuzzer（gobuster 备选） | 用内置路径列表 |
+| whatweb | 否 | 技术栈指纹 | 用 HTTP 头分析 |
+
+**nmap 是唯一强依赖**，其余工具脚本内置降级方案。
+
+缺少可选工具时，向用户报告缺失项和影响，然后继续执行。
+
+---
+
+## 第 2 步：信息收集（3 Agent 并行）
+
+**在同一条消息中**派出 3 个 Agent 并行执行：
+
+### Agent-Recon-DNS
+```
+目标：对 {target} 执行 DNS 和 WHOIS 信息收集
+上下文：渗透测试第 2 步，需要获取目标的 IP 地址、DNS 记录、注册信息
+范围：仅执行 DNS 查询和 WHOIS，不发送任何攻击流量
+规范：使用 {scripts} 目录下的脚本
+产出：执行命令并返回结果摘要
+
+执行：
+cd {scripts} && uv run pentest_recon.py --target {target} --mode dns --output-dir {output}
+```
+
+### Agent-Recon-Tech
+```
+目标：识别 {target} 的技术栈和 Web 指纹
+上下文：渗透测试第 2 步，需要知道目标运行什么软件和版本
+范围：HTTP 请求 + whatweb 指纹，不发送攻击 payload
+规范：使用 {scripts} 目录下的脚本
+产出：执行命令并返回结果摘要
+
+执行：
+cd {scripts} && uv run pentest_recon.py --target {target} --mode tech --output-dir {output}
+```
+
+### Agent-Recon-OSINT
+```
+目标：搜索 {target} 相关的已知漏洞和安全事件
+上下文：渗透测试第 2 步，需要了解目标是否有已知的安全问题
+范围：仅使用 WebSearch 搜索公开信息
+产出：返回搜索结果摘要，包含 CVE 编号和关键发现
+
+使用 WebSearch 搜索以下关键词（分多次搜索）：
+1. "{target}" CVE vulnerability
+2. "{target}" security breach
+3. "{target}" 使用的技术栈（从 Agent-Recon-Tech 结果获取）的最新 CVE
+```
+
+**等待所有 Agent 完成后**，汇总信息收集结果，向用户报告发现：
+- 解析到的 IP 地址
+- 技术栈和版本信息
+- 发现的敏感文件
+- 公开的安全信息
+
+---
+
+## 第 3 步：端口扫描
+
+依赖第 2 步的 DNS 结果确认真实 IP。脚本内部 TCP / UDP 并发执行,版本探测与 OS 指纹识别也并发。
+
+```bash
+cd {scripts} && uv run pentest_scan.py --target {target} --output-dir {output}
+```
+
+扫描完成后，提取关键信息供后续步骤使用：
+- `open_ports`：所有开放端口列表
+- `http_ports`：HTTP 服务端口（用于 Web 测试）
+- `ssl_ports`：SSL/TLS 端口（用于 SSL 审计）
+- `ssh_port`：SSH 端口（用于认证测试）
+- `ftp_port`：FTP 端口（用于认证测试）
+
+向用户报告端口扫描结果。
+
+---
+
+## 第 4 步：漏洞扫描 + Web 枚举 + SSL（最多 5 Agent 并行）
+
+根据第 3 步发现的服务，**在同一条消息中**派出 Agent：
+
+### Agent-Vuln-Nuclei（始终执行）
+```
+目标：对 {target} 执行 Nuclei 漏洞模板扫描
+上下文：渗透测试第 4 步，基于端口扫描发现的服务进行漏洞匹配
+范围：使用 nuclei 官方模板库，限速 50 req/s
+规范：使用 {scripts} 目录下的脚本
+产出：执行命令并返回发现的漏洞列表
+
+执行：
+cd {scripts} && uv run pentest_vuln.py --target {target} --mode nuclei --output-dir {output}
+```
+
+### Agent-Vuln-NSE（始终执行）
+```
+目标：对 {target} 的开放端口执行 Nmap 漏洞脚本扫描
+上下文：渗透测试第 4 步，使用 nmap 内置 NSE 漏洞检测脚本
+范围：仅 --script vuln，不执行 exploit 脚本
+产出：执行命令并返回发现的漏洞列表
+
+执行：
+cd {scripts} && uv run pentest_vuln.py --target {target} --mode nse --ports {open_ports} --output-dir {output}
+```
+
+### Agent-Web-Dir（仅 HTTP 端口存在时）
+```
+目标：对 {target} 的 Web 服务执行目录枚举
+上下文：渗透测试第 4 步，发现隐藏路径和敏感文件
+范围：目录枚举，不修改任何数据
+产出：执行命令并返回发现的路径列表
+
+执行：
+cd {scripts} && uv run pentest_web.py --target {web_url} --mode dir --output-dir {output}
+```
+
+### Agent-SSL（仅 SSL/TLS 端口存在时）
+```
+目标：对 {target} 执行 SSL/TLS 安全审计
+上下文：渗透测试第 4 步，检查证书和加密配置
+范围：SSL/TLS 握手分析，不攻击
+产出：执行命令并返回 SSL 审计结果
+
+执行：
+cd {scripts} && uv run pentest_ssl.py --target {target} --port {ssl_port} --output-dir {output}
+```
+
+### Agent-CVE-Research（始终执行）
+```
+目标：搜索第 3 步发现的服务版本对应的 CVE
+上下文：渗透测试第 4 步，端口扫描已发现以下服务和版本：{services_versions}
+范围：WebSearch 搜索公开 CVE 数据库
+产出：返回每个服务的已知 CVE 列表和利用条件
+
+对每个 {service} {version} 组合，使用 WebSearch 搜索：
+1. "{service} {version}" CVE exploit
+2. 在搜索结果中重点关注 CVSS >= 7.0 的高危漏洞
+```
+
+---
+
+## 第 5 步：Web 深度测试（2 Agent 并行）
+
+**仅当第 4 步 Agent-Web-Dir 发现了 Web 路径时执行。**
+
+### Agent-Web-SQLi
+```
+目标：对 {target} 的 Web 应用进行 SQL 注入探测
+上下文：渗透测试第 5 步，目录枚举已发现以下路径：{discovered_paths}
+范围：仅检测注入点，禁止数据提取（sqlmap --technique=BEU, 无 --dump）
+规范：使用 {scripts} 目录下的脚本
+产出：执行命令并返回发现的注入点
+
+执行：
+cd {scripts} && uv run pentest_web.py --target {web_url} --mode sqli --output-dir {output}
+```
+
+### Agent-Web-XSS
+```
+目标：对 {target} 的 Web 应用进行 XSS 反射探测
+上下文：渗透测试第 5 步，使用无害探针检测反射点
+范围：发送无害标记 <dwy-xss-test>，不执行真实 XSS payload
+规范：使用 {scripts} 目录下的脚本
+产出：执行命令并返回发现的 XSS 反射点
+
+执行：
+cd {scripts} && uv run pentest_web.py --target {web_url} --mode xss --output-dir {output}
+```
+
+---
+
+## 第 6 步：认证测试
+
+脚本内部 SSH / FTP / HTTP 三个服务并发执行,FTP 的匿名探测与 hydra 弱密码并发,HTTP 的登录路径探测与默认凭据尝试也都是异步并发。
+
+**执行前向用户确认：**
+
+```
+认证测试将尝试常见弱密码（最多 20 组），可能触发目标的安全防护。
+发现的服务：SSH({ssh_port}), FTP({ftp_port}), HTTP({http_ports})
+是否执行？(y/n)
+```
+
+用户同意后：
+
+```bash
+cd {scripts} && uv run pentest_auth.py --target {target} --output-dir {output} --services ssh,ftp,http
+```
+
+用户拒绝 → 跳过此步。
+
+---
+
+## 第 7 步：漏洞验证与利用评估
+
+读取 `{output}` 下所有 JSON 结果，对 Critical 和 High 级别的漏洞进行验证：
+
+1. 分析每个高危漏洞的详细信息
+2. 使用 WebSearch 搜索该漏洞的公开 PoC 和利用方式
+3. 评估是否可实际利用获取服务器控制权
+4. **仅做到验证漏洞确实存在为止**，不执行实际利用
+
+对每个高危漏洞输出评估结论：
+- **已确认可利用** — 有成熟 PoC，理论上可获取 shell/控制权
+- **可能可利用** — CVE 存在但需要特定条件
+- **信息泄露** — 不直接导致控制权获取但构成风险
+
+**如果发现可获取控制权的漏洞，立即告警：**
+
+```
+=============================================================
+  !! 安全告警 — 可获取服务器控制权 !!
+=============================================================
+
+漏洞: {vuln_title} (CVE-XXXX-XXXX)
+CVSS: {score}
+评估: 存在公开 PoC，攻击者可利用此漏洞获取服务器控制权
+
+!! 此处仅做告警，未执行实际利用 !!
+
+建议立即修复。
+=============================================================
+```
+
+---
+
+## 第 8 步：报告生成
+
+```bash
+cd {scripts} && uv run pentest_report.py --input-dir {output} --output {output}/report.md
+```
+
+报告包含：基本信息、执行摘要、目标概况、漏洞详情、SSL 审计、修复优先级、免责声明。
+
+---
+
+## 第 9 步：安全告警与总结
+
+向用户展示最终结果：
+
+1. **按严重性分级展示所有发现**（Critical → High → Medium → Low）
+2. **给出修复优先级建议**
+3. **告知报告位置**：`{output}/report.md`
+4. **如果发现 Critical 级漏洞**，强烈建议立即修复并重新测试
+
+---
+
+## 安全保障
+
+### 脚本内置的安全限制
+
+- `executor.py` 仅允许执行 8 个预定义工具命令（白名单）
+- `executor.py` 拦截危险参数（`--os-shell`, `--dump-all`, `--file-write` 等）
+- 所有命令 `shell=False`，参数列表传入，防止命令注入
+- 单次扫描最大超时 300 秒
+- 认证测试每服务最多 20 次尝试
+- sqlmap 禁止 Stacked Queries（`--technique=BEU`）
+- hydra 找到第一个弱密码即停止（`-f`）
+
+### 禁止事项
+
+- **禁止**在未经用户同意的目标上执行任何扫描
+- **禁止**执行任何 exploit（Metasploit、手写 exploit、PoC 执行）
+- **禁止**在目标服务器上植入代码、webshell、后门
+- **禁止**提取目标数据库中的实际数据
+- **禁止**进行 DoS/DDoS 测试
+- **禁止**修改目标系统的任何配置或文件
+- **禁止**使用社会工程学手段
+- **禁止**绕过脚本的安全限制（SAFETY_LIMITS）
+- **禁止**将发现的漏洞详情（PoC/exploit 细节）输出给用户以外的人
+
+### 降级策略
+
+工具不可用时，脚本自动降级：
+- nuclei 不可用 → 仅用 nmap NSE 扫描
+- gobuster/ffuf 不可用 → 用内置常见路径列表 + httpx 探测
+- sqlmap 不可用 → 用基础 HTTP SQL 错误探针
+- testssl.sh 不可用 → 用 Python ssl 模块基础检测
+- hydra 不可用 → 跳过暴力认证测试
+- whatweb 不可用 → 用 HTTP 头分析识别技术栈

package/templates/claude-global/skills/dwy-pentest/scripts/lib/__init__.py

@@ -0,0 +1 @@
+# dwy-pentest lib

package/templates/claude-global/skills/dwy-pentest/scripts/lib/config.py

@@ -0,0 +1,109 @@
+"""全局配置：安全限制、工具白名单、危险参数黑名单。"""
+
+from pathlib import Path
+
+# 扫描参数
+SCAN_TIMEOUT = 300  # 单次扫描最大时长（秒）
+SCAN_RATE_LIMIT = 50  # 每秒最大请求数
+NMAP_TIMING = "T3"  # 扫描速度（T1-T5）
+MAX_AUTH_ATTEMPTS = 20  # 认证测试最大尝试次数
+
+# 工具命令白名单 — executor 仅允许执行这些命令
+ALLOWED_COMMANDS = frozenset({
+    "nmap",
+    "nuclei",
+    "gobuster",
+    "ffuf",
+    "sqlmap",
+    "hydra",
+    "testssl",
+    "testssl.sh",
+    "whatweb",
+    "curl",
+    "dig",
+    "whois",
+})
+
+# 危险参数黑名单 — executor 拦截包含这些参数的调用
+BLOCKED_ARGS = frozenset({
+    # sqlmap 危险参数
+    "--os-shell",
+    "--os-cmd",
+    "--os-pwn",
+    "--dump-all",
+    "--file-read",
+    "--file-write",
+    "--file-dest",
+    "--priv-esc",
+    "--reg-read",
+    "--reg-add",
+    "--reg-del",
+    # nmap exploit 脚本
+    "--script=exploit",
+    # hydra 无限制
+    "-x",  # 密码生成器（可生成无限组合）
+    # 通用危险
+    "--reverse-shell",
+    "--backdoor",
+})
+
+# 硬编码安全限制
+SAFETY_LIMITS = {
+    "no_exploit_execution": True,
+    "no_reverse_shell": True,
+    "no_payload_upload": True,
+    "no_data_exfiltration": True,
+    "sqlmap_no_stacked": True,
+    "sqlmap_no_dump": True,
+    "hydra_stop_on_first": True,
+}
+
+# 内置 Top 20 弱密码列表（认证测试用）
+WEAK_PASSWORDS = [
+    "123456",
+    "password",
+    "admin",
+    "12345678",
+    "123456789",
+    "1234567890",
+    "root",
+    "toor",
+    "admin123",
+    "password123",
+    "P@ssw0rd",
+    "qwerty",
+    "abc123",
+    "111111",
+    "letmein",
+    "welcome",
+    "monkey",
+    "dragon",
+    "master",
+    "login",
+]
+
+# 常见默认凭据组合
+DEFAULT_CREDENTIALS = [
+    ("admin", "admin"),
+    ("admin", "password"),
+    ("admin", "admin123"),
+    ("admin", "123456"),
+    ("root", "root"),
+    ("root", "toor"),
+    ("root", "password"),
+    ("root", "123456"),
+    ("test", "test"),
+    ("user", "user"),
+    ("guest", "guest"),
+    ("administrator", "administrator"),
+    ("postgres", "postgres"),
+    ("mysql", "mysql"),
+    ("ftp", "ftp"),
+]
+
+# 输出目录
+DEFAULT_OUTPUT_DIR = Path(".pentest-results")
+
+# nuclei 扫描模板标签
+NUCLEI_TAGS = "cve,kev,misconfig,exposure"
+NUCLEI_SEVERITY = "critical,high,medium"

package/templates/claude-global/skills/dwy-pentest/scripts/lib/executor.py

@@ -0,0 +1,157 @@
+"""安全命令执行器：shell=False + 命令白名单 + 参数黑名单 + 超时 + 审计日志。"""
+
+import json
+import subprocess
+import time
+from dataclasses import asdict, dataclass, field
+from pathlib import Path
+
+from .config import ALLOWED_COMMANDS, BLOCKED_ARGS, SCAN_TIMEOUT
+
+
+@dataclass
+class ExecResult:
+    """命令执行结果。"""
+
+    command: list[str]
+    returncode: int
+    stdout: str
+    stderr: str
+    duration: float
+    blocked: bool = False
+    block_reason: str = ""
+
+
+@dataclass
+class AuditLog:
+    """审计日志管理器。"""
+
+    log_file: Path
+    entries: list[dict] = field(default_factory=list)
+
+    def __post_init__(self):
+        self.log_file.parent.mkdir(parents=True, exist_ok=True)
+
+    def record(self, result: ExecResult):
+        entry = {
+            "timestamp": time.strftime("%Y-%m-%d %H:%M:%S"),
+            "command": " ".join(result.command),
+            "returncode": result.returncode,
+            "duration": round(result.duration, 2),
+            "blocked": result.blocked,
+            "block_reason": result.block_reason,
+        }
+        self.entries.append(entry)
+        with open(self.log_file, "a") as f:
+            f.write(json.dumps(entry, ensure_ascii=False) + "\n")
+
+
+def validate_command(cmd: list[str]) -> tuple[bool, str]:
+    """校验命令是否在白名单内、参数是否安全。
+
+    Returns:
+        (is_safe, reason) — is_safe=False 时 reason 说明拒绝原因。
+    """
+    if not cmd:
+        return False, "空命令"
+
+    binary = Path(cmd[0]).name
+    if binary not in ALLOWED_COMMANDS:
+        return False, f"命令 '{binary}' 不在白名单中，允许的命令: {', '.join(sorted(ALLOWED_COMMANDS))}"
+
+    for arg in cmd[1:]:
+        normalized = arg.lower().strip()
+        for blocked in BLOCKED_ARGS:
+            if normalized == blocked or normalized.startswith(f"{blocked}="):
+                return False, f"危险参数 '{arg}' 已被安全策略拦截"
+
+    # sqlmap 额外检查：禁止 stacked queries
+    if binary == "sqlmap":
+        for i, arg in enumerate(cmd):
+            if arg == "--technique" and i + 1 < len(cmd):
+                technique = cmd[i + 1].upper()
+                if "S" in technique:
+                    return False, "sqlmap --technique 禁止使用 S(tacked queries)"
+            if arg.startswith("--technique="):
+                technique = arg.split("=", 1)[1].upper()
+                if "S" in technique:
+                    return False, "sqlmap --technique 禁止使用 S(tacked queries)"
+
+    return True, ""
+
+
+def run_command(
+    cmd: list[str],
+    timeout: int = SCAN_TIMEOUT,
+    audit: AuditLog | None = None,
+    cwd: str | Path | None = None,
+) -> ExecResult:
+    """安全执行外部命令。
+
+    - shell=False，参数列表传入
+    - 命令白名单 + 参数黑名单校验
+    - 强制超时
+    - 可选审计日志记录
+    """
+    is_safe, reason = validate_command(cmd)
+    if not is_safe:
+        result = ExecResult(
+            command=cmd,
+            returncode=-1,
+            stdout="",
+            stderr=reason,
+            duration=0.0,
+            blocked=True,
+            block_reason=reason,
+        )
+        if audit:
+            audit.record(result)
+        return result
+
+    start = time.time()
+    try:
+        proc = subprocess.run(
+            cmd,
+            capture_output=True,
+            text=True,
+            timeout=timeout,
+            shell=False,
+            cwd=cwd,
+        )
+        duration = time.time() - start
+        result = ExecResult(
+            command=cmd,
+            returncode=proc.returncode,
+            stdout=proc.stdout,
+            stderr=proc.stderr,
+            duration=duration,
+        )
+    except subprocess.TimeoutExpired:
+        duration = time.time() - start
+        result = ExecResult(
+            command=cmd,
+            returncode=-2,
+            stdout="",
+            stderr=f"命令超时（{timeout}s）",
+            duration=duration,
+        )
+    except FileNotFoundError:
+        duration = time.time() - start
+        result = ExecResult(
+            command=cmd,
+            returncode=-3,
+            stdout="",
+            stderr=f"命令未找到: {cmd[0]}（请先运行 setup_tools.sh 安装）",
+            duration=duration,
+        )
+
+    if audit:
+        audit.record(result)
+    return result
+
+
+def save_json_result(data: dict, output_path: Path):
+    """将扫描结果保存为 JSON。"""
+    output_path.parent.mkdir(parents=True, exist_ok=True)
+    with open(output_path, "w") as f:
+        json.dump(data, f, ensure_ascii=False, indent=2)