代码审计模式 Ph03n1xRt/dsh-codeaudit

dsh-codeaudit

DSH（DeepSeek Harness）的代码审计模式插件：一个白盒源码审计工作流智能体。
指挥官智能体测绘代码资产、拆解审计意图、并发委派子代理追踪数据流，所有证据/资产/漏洞沉淀为可回放的记录，实时在 Web「代码审计」页可视化（头部统计卡片＋严重级分布条的作战大屏风格），最终产出带证据链、门禁校验与修复建议的中文审计报告。

审计任务 (engagement)
└─ spawns ─> 审计意图 (intent)「追踪 /api/order 的查询参数」 ├─ yields ─> 代码证据 (evidence) [entry] OrderController.java:42 + 代码片段 ├─ yields ─> 代码证据 (evidence) [sink] OrderDao.java:87 + 代码片段 └─ proves ─> 漏洞发现 (finding) [high|confirmed] SQL 注入 ↑ supports ────┘（证据链，逐条挂接）

特性

审计工作流

- 跨会话成果库：全局 SQLite（$DSH_HOME/codeaudit-library.db）随每条 finding/engagement 写透沉淀，会话重置同步镜像；Web「成果库」页跨会话聚合——统计卡片、严重级分布条、任务花名册、按目标分组的漏洞台账（行展开看描述/修复/冻结 POC，一键复制）。
- 报告门禁：codeaudit_report 自带结构化门禁校验表（修复建议齐全 / 已确认项验证留痕 / 意图收敛 / 存疑清零）——存在 WARN 必须补齐或向用户明确说明后才可定稿，"看起来完成"藏不住缺口。
- 动态指挥官信封：systemPrompt.context 逐轮装配注入实时战况（进度计数、结构缺口、下一步建议）， <codeaudit-state rev> 标签包裹抗压缩，变化才投递、稳态零成本。
- 经验沉淀（跨会话）：codeaudit_lessons_add / codeaudit_lessons 把可复用的误报教训、有效手法、排查口径写入全局经验库（读取热度 × 30 天半衰期排序），规划阶段先查、指挥官信封自动注入命中技术栈的前三条；Web 成果库页「经验沉淀」区可视化（类别徽章 + 标签 + 引用次数）。
新库出厂自带三条判界种子经验（持密码清除登录锁定不立项 / 验证码链路六条知码路径判界 / 仅密码解绑 MFA 才是单因子降级），从第一次审计起生效，与真实经验同场竞争排序。
- 双链分档验证：critical/high 双链一致（工人链＋独立追踪链三要素吻合才 confirmed，不一致退回重算、仍不一致标存疑）；medium 单链＋抽检；low/info 登记即收。报告逐条标注验证档位。

- 审计链路记录：engagement → intent → evidence → derived intent → finding 六表存储域，每条关系都是显式边， <kind>-<n> 确定性 id，会话投影可从日志纯重放（存储与视图严守同一 id 不变量，失败写入连计数器一起回滚）。
- 证据链与代码片段：finding 必须携带 file:line 位置与至少一条 supports 证据；evidence / finding 可附 snippet（提交时截断到 4000 字符并冻结——代码之后改动，报告仍可复现）。
- 可重放 POC：finding 可附 poc 字段——复现该漏洞的完整 HTTP raw（上限 16KB，动态验证后由子代理写入）， UI 一键复制/下载，报告中原样输出；poc 为空即表示仅静态结论。
- 误报控制：sanitizer 证据类型专门记录防护点；finding 分 confirmed（已排查防护）/ suspected（存疑待人工复核）两级状态。
- 指挥官/子代理分工：决策智能体测绘资产、按「模块 × 漏洞类」拆意图、并发委派深挖；子代理只经 codeaudit_submit 单通道回写父意图（服务端解析父会话、拒绝占位符 id），提交即返回本批分配的真实 id，模型不需要也不允许臆测编号。
- 只读纪律：协议约定目标仓库只读（读取/搜索/静态分析），动态验证限隔离目录。

Web 可视化（会话旁「代码审计」标签页）

| 子视图 | 内容 |
| 探索链路（暂时隐藏，重设计中） | 整张审计图（React Flow），两种布局一键切换：泳道布局——任务卡片在顶部居中，每个意图独占一条竖直泳道，它的证据、漏洞依次挂在它自己正下方，不存在全局类型列堆积；力导向图布局（Neo4j 风格无向网络）——节点互斥、边作弹簧的确定性物理模拟：连通的节点自然聚簇，共享证据的意图彼此拉近，死胡同探索漂向边缘；无固定读向、节点可拖拽微调，跨意图的交叉关系一目了然。为大规模链路另有三重压缩：意图聚合折叠——默认只显示 任务→意图→漏洞 骨架，每个意图卡片带「证据 N · 漏洞 M」计数徽章，点徽章按需展开该意图的证据（折叠时证据的 supports/derived_from 边自动重挂到所属意图，高层语义不丢；小图 ≤15 节点自动全展开）；「仅漏洞链路」过滤隐去无结论的死胡同探索；缩放 0.05–4 倍 + 右下角可拖拽 MiniMap；纵向/横向 生长方向切换。点节点看详情，含代码位置与冻结代码片段，位置行可直接打开源文件或一键复制路径 |
| 漏洞发现 | 按严重级排序的卡片列表，位置可点击打开源文件；每条两个下钻：完整漏洞链路抽屉——顶部是 利用链图（entry → dataflow → sink → 触发点 一字排开，sanitizer 防护点虚线挂在链下方、config/dependency 旁证挂上方，点击节点定位到对应证据卡），下方是 任务 → 意图 → 逐条证据（kind 徽章 + 位置 + 代码片段）→ 漏洞结论 的完整阶梯；yakit POC 抽屉——冻结在漏洞记录里的完整 HTTP raw（请求行+头+体），可一键复制或下载 .txt，直接粘贴 Yakit/Burp 重放（仅静态结论时显示空态提示） |
| 代码资产 | repo / module / file / class / function / endpoint / config / dependency 八类资产，列表分组与父子树图两种视图 |
| 成果库 | 跨会话聚合大屏：统计卡片（任务/漏洞/严重+高危/已确认）、严重级分布条、审计任务花名册（按严重级 chip 与确认率）、漏洞台账按目标分组排序，行展开看详情与冻结 POC（一键复制） |
| 报告 | 与 codeaudit_report 工具同构的中文 Markdown（执行摘要 / 探索链路 / 漏洞+证据链+修复建议 / 资产清单），可复制、可下载 codeaudit-report-<target>.md |

安装

前置要求：dsh 0.1.1-rc.2 或更新，Node ≥ 22.19（node:sqlite 与 dsh 0.1.1 引擎要求）。

从 GitHub Release 安装（推荐）
dsh plugin --profile web add https://github.com/Ph03n1xRt/dsh-codeaudit/releases/latest/download/dsh-codeaudit.tar.gz

或从本地 tarball
dsh plugin --profile web add file:/绝对路径/dsh-codeaudit.tar.gz

重启 dsh，新建会话时预设选择器会出现「代码审计模式」。数据落在 $DSH_HOME/storages/codeaudit-sessions.db（官方 @deepseek-ai/dsh-storage-sqlite 后端）。

装完可用 dsh --profile web --dump-config 核对 patch 层四行（ui-codeaudit / storage-sqlite / storage-domain 路由 / codeaudit-preset-root）；核心 codeaudit 行只随预设出现，不在 patch 里——这是设计如此（防止选中预设时工具目录重复）。

使用

选中预设后，直接发一条包含目标 + 审计目标的消息，无需任何特殊指令：

审计 /home/kali/targets/shop-backend（Java/Spring 项目）。
审计目标：重点排查 SQL 注入、越权访问和硬编码密钥，其次看文件上传和 SSRF。
范围：只审 src/main/java，排除 test 和前端目录。

| 你说的 | 存入字段 | 作用 |
| 代码库路径/名称 | target | 审计对象（建议绝对路径） |
| 重点排查什么 | objective | 指挥官建「模块 × 漏洞类」覆盖矩阵、拆意图的依据 |
| 只审/排除哪些目录 | scope | 越界意图会被协议拒绝 |
| 技术栈 | stack | 帮模型快速定位框架特性；经验库按它匹配注入 |

流程：指挥官登记 engagement → 测绘资产 → 拆分意图 → 并发委派子代理追踪 source→sink 数据流 → 验证定级（排查 sanitizer/框架防护后定 confirmed/suspected）→ 终止时调用 codeaudit_report 出报告。
过程中你只需要回答它的 ask_user 提问；想中途出报告就说一句「出最终报告」。

通用提示词（复制即用）

每次审计只需粘贴下面这段，改两处：目标路径、范围行。这段提示词只提供只有你能给的信息（目标/范围/优先级）与关键纪律强调——详细工作协议由插件在会话内自动注入，不必重复。

审计【/目标/代码库/路径】（技术栈自行识别并写入 engagement.stack）。

审计目标：全量安全审计，以「取得数据或系统权限」为结果导向。按「模块 × 漏洞类」覆盖矩阵拆分意图并发委派子代理：

- 技术类：SQL/命令/表达式注入、反序列化、鉴权与越权（水平/垂直）、文件上传下载与路径穿越、 SSRF、XSS、CSRF、密钥硬编码与弱加密、危险配置、依赖组件已知漏洞（对照依赖清单逐一核查）
- 逻辑与组合类（按链路立项）：多步流程逻辑冲突（密码重置/换绑/审批流的校验错位、步骤可跳、 token 作用域过宽）、未授权接口组合链（匿名读 userlist/rolist ＋ 匿名写 adduser 拼出提权）、集合级越权与批量拉取（list/导出/批量接口缺归属过滤）、参数信任与重放（金额数量服务端未重算）

立项判界（牢记）：安全漏洞 = 攻击者越过认证/授权/保密边界、取得此前没有的数据或能力。
只是更快地做已有凭证本来就能做的事，不立项——持密码清除登录锁定不是漏洞；可给任意用户发验证码＋知码可重置是标准流程，只有能未经授权拿到验证码（响应回显/无次数限制可爆破/ 可预测/信道劫持/长效重放/A 的码重置 B）才是漏洞；纯代码缺陷（边界条件、异常处理、精度）同样不立项。

执行要求：
1. 开局先查经验库（codeaudit_lessons，按技术栈关键词），参照 refs/coverage-matrix.md 建覆盖矩阵；先测绘（入口点、路由表、接口权限矩阵、依赖、配置全部登记为资产）再拆意图；
2. 验证分档：critical/high 双链一致——另派独立追踪子代理从入口重追 entry→sink（不得引用工人链结论），入口＋路径＋sink 三要素吻合才 confirmed，不一致退回重算、仍不一致标 suspected； medium 单链＋抽检；low/info 登记即收。已排查 sanitizer/框架防护才能 confirmed，存疑写明未验证点；
3. 每条漏洞必须带 file:line 位置、关键代码片段、至少一条证据（组合链的 entry/dataflow/sink 对应链路各跳接口）；能从代码推导出完整请求的都要构造 POC，遵循 Yakit Web Fuzzer 的 HTTP 语法：纯报文、可用 fuzztag，加密/签名/时间戳用 {{yak(...)}} 占位并把热加载函数写入 pocScript；组合链 POC 按攻击顺序分步标注（每段一个报文）；无环境发送时静态构造并在 pocNote 注明「静态构造，未经发送验证」，有环境则发送后注明「已动态验证」；
4. POC 的参数生成规则、占位符含义、前置条件写 pocNote，绝不混入报文；构造前先用 skill 工具加载 codeaudit-methodology（构造规范与判界手册）与 yak 技能（若已拉取）作参考；
5. 经验沉淀——四类时机立即调 codeaudit_lessons_add：误报排除且理由可复用、手法/指纹在某栈确证有效、我纠正了你的判断、明确排查口径；
6. 终止时调用 codeaudit_report 出报告：门禁校验表每条 WARN（缺修复建议、已确认项无 POC 且无 pocNote）必须补齐或向我明确说明后才定稿；漏洞按严重级排序，逐条带修复建议与验证档位。

范围：【只审 src/main，排除 test/、前端/、node_modules/】

可选加强（往执行要求里追加）：

- 聚焦逻辑与组合类：优先深挖认证/重置流程与未授权接口组合链，技术类按矩阵快速过；
- 已知账号体系：系统存在 user/role/admin 三级账号，重点验证跨级组合与集合级越权；
- 有测试环境：目标已在 http://x.x.x.x:8080 部署可用，动态验证在该地址进行，验证流量控制在只读操作；
- 带既有经验：开局先按 "关键词" 查经验库并应用命中条目。

注意：静态构造的 POC 未经发送验证，重放失败时以说明卡（pocNote）里的规则为准手动修正。

注意：一次会话一个审计任务——登记新 engagement 会清空本会话旧图（刻意的重置语义），换目标直接再发一条即可。审计成果与经验跨会话沉淀在「成果库」，新任务自动继承。

工作原理

数据模型（六表 + 显式边）

| 表 | 记录 | 关键字段 |
| engagements | 审计任务（每会话一个） | target, objective, scope, stack |
| intents | 审计意图 | title, detail（锚点：engagementId 或 derivedFromEvidenceId 恰一） |
| evidences | 代码证据 | kind: entry/sink/dataflow/sanitizer/config/dependency/info, location, detail, snippet, confidence |
| findings | 漏洞发现 | title, severity, status: confirmed/suspected, cwe, location 必填, snippet, fix |
| assets | 代码资产 | type: 八类, value, meta |
| edges | 语义边 | spawns / yields / derived_from / proves / supports / parent |

边语义：spawns（任务→意图）、yields（意图→证据）、derived_from（证据→新意图）、proves（意图→漏洞）、 supports（证据→漏洞，构成证据链，finding 至少一条）、parent（资产→资产）。

模型工具（9 个 codeaudit_）

- 指挥官专用：codeaudit_set_engagement（重置整图）、codeaudit_add_intent、codeaudit_add_evidence、 codeaudit_add_finding、codeaudit_add_asset、codeaudit_state（单行摘要防上下文膨胀）、 codeaudit_graph、codeaudit_report
- 子代理专用：codeaudit_submit（服务端解析父会话；占位符 id 直接拒绝；返回本批真实 id；事务全或无回滚）

源文件查看（打开文件）

证据/漏洞的 file:line 位置在 Web 端可直接点开：宿主侧 /codeaudit/file 路由从该会话的 engagement target 解析出绝对路径并强制包含校验（拒绝任何逃逸目标仓库的路径，穿越与绝对路径逃逸一律 403），读取结果（1MB 封顶）在浏览器内以行号视图呈现——目标行高亮并自动滚动定位，按扩展名做语法高亮，超长文件只渲染目标行 ±200 行；「在本机打开」按钮经 /codeaudit/file/open 调用 dsh 所在主机的系统打开器（open/xdg-open/start），同样受包含校验约束。远程 SSH 场景下浏览器内查看器即主用方式。

安全与质量护栏

写入边界引用校验（同会话同表、恰一锚点、位置必填、证据链必挂）→ 域 zod schema 在持久化边界二次校验 → invariant 伴生在 domain/changed 三次校验边两端表 → 投影镜像拒绝 + CAP 200 防上下文膨胀；子代理 toolFilter.deny 全部写/读工具只留 submit；scope 为协议层留痕约束，实际文件访问由部署的沙箱与审批层强制；全部自注册 HTTP 路由带同源栅栏（Origin 必须回显 Host，含端口）＋状态变更 POST 的进程级 CSRF 令牌。

项目结构

dsh-codeaudit/
├── package.json # 单包 bundle：exports 子路径 = Cordis 插件行；dsh.bundle.patch / dsh.client 声明 ├── cordis.patch.yml # 宿主 profile 层：UI 行、官方 sqlite 后端行、codeaudit 域路由、预设根行 ├── preset/codeaudit/ # agent 预设：standard 副本 + 审计指挥官 persona + 仅 submit 的子代理 + 核心插件行 ├── src/ │ ├── index.ts # bundle 宿主入口（patch-only 惰性席位） │ ├── invariant.ts # 不变量伴生（引用纪律三道防线之一） │ ├── preset-root.ts # 注册包内只读预设根（profile 启动覆盖 roots 的对冲，rc.6–0.1.1 均存在） │ ├── dsh-codeaudit/ # 核心插件：spec / store / tools / instructions / projection / types / library（跨会话库）/ envelope（动态信封）+ tests │ └── dsh-client-ui-codeaudit/ # 浏览器半：视图、布局纯函数、i18n、链路抽屉、成果库页 + webGuard / fileRoute + tests ├── scripts/build.mjs # esbuild 构建管线（宿主半 ×5 + 浏览器 client 打包，含 CSS Modules 内联） ├── tests/bundle.spec.ts # 发布清单契约测试 └── .github/workflows/release.yml # tag 推送 → 测试 → 构建 → npm pack → GitHub Release

开发

npm install # .npmrc 已带 legacy-peer-deps=true（dsh 预发布版 peer 冲突，与 dsh-pentest CI 做法一致）；本机需 Node ≥ 22.19（dsh 0.1.1 要求）
npm test # 156 个用例：host 工具/存储/投影/不变量 + client 布局/视图/注册门控/文件路由 + bundle 契约
npm run check # tsc --noEmit
npm run build # 产出 lib/（七个构建产物）
npm run pack:local # 本地安装包：版本自动打 beta.N 后缀并自增 → 构建 → dist/.tgz（dsh 按版本缓存插件，版本不变会命中旧包）
npm pack # 正式发布用：干净版本号的自包含 tarball

浏览器端打包契约：lib/ui-codeaudit.client.js 以 window.__ModuleLoader__.load({ id, factory }) 握手交付，仅 react / react-dom / react/jsx-runtime 走 loader 模块表，其余（含 @xyflow/react）全部内联； CSS Modules 哈希后经插件自有的 标签注入。

发布

npm version 1.x.0 # 或手动改 package.json
git push && git push --tags # 推送 v 标签即触发 Actions：测试 → 构建 → 打包 → 发布 Release

Release 产物名固定为 dsh-codeaudit.tar.gz，即安装命令里的 URL。

已知边界

- scope 是协议层留痕而非门禁：文件访问的实际强制在部署的沙箱与审批层。
- 会话投影是最近 200 节点的窗口视图；完整记录以 codeaudit_state / codeaudit_graph / 报告为准。
- 一次会话一个 engagement；重置语义见上文。

内置技能（skill）与自定义扩展

插件随包携带一个专属技能，安装即生效（只对「代码审计模式」的会话可见，不影响其他会话）：

- codeaudit-methodology — 审计方法论与 POC 构造规范：模块 × 漏洞类意图拆解矩阵、验证定级三步法（向上追数据流 / 查防护记 sanitizer / 再定 confirmed-suspected）、POC 纯报文与 pocNote 说明的分工示例。模型在规划意图、验证发现、写 POC 时会自动按需加载。
- webfuzzer-hotpatch / yaklang-syntax（随包内置）— Yakit 官方技能（yaklang/yak-skills）中与审计最相关的两个：前者是 HTTP POC 语法、fuzztag、请求/响应热加载钩子（beforeRequest/afterResponse 等）；后者是 Yaklang DSL 语法与加解密函数写法。打包进 tarball，安装即可用，全程无网络依赖。更新版本：仓库里跑 npm run skills:refresh（官方源 → jsDelivr → gitmirror 三镜像）重新拉取并提交。若文件被删除，审计会话开始前指挥官会征询同意后通过 codeaudit_fetch_yak_skill 在线补装，Web 面板头部也会显示未安装徽章与拉取按钮。

预设还内置 refs 参考手册库（preset/codeaudit/refs/，随 tarball 分发、零本机路径）：Java 危险 sink 大表（注入/反序列化/文件/SSRF/JNDI/密钥/越权按类给 sink API、入口特征与防护排查要点）、双链分档验证规范（三要素判据、退回与未决处理、抽检口径）、逻辑与组合漏洞手册（流程冲突/未授权组合链/集合越权/ 参数重放的高发模式与多请求 POC 构造）、模块 × 漏洞类覆盖矩阵模板（终态口径与收敛纪律）， methodology 技能按需路由到对应手册。

加自己的技能无需改插件——dsh 的技能注册表会合并预设层与通用根目录，往任意一处放 <name>/SKILL.md（目录包，frontmatter 写 description）即可：

mkdir -p ~/.dsh/skills/my-framework-checklist
cat > ~/.dsh/skills/my-framework-checklist/SKILL.md << 'EOF'

name: my-framework-checklist
description: 某框架的安全审计 checklist（何时用、查什么）

（正文：逐项检查点……）
EOF

项目级技能放被审仓库的 .dsh/skills/ 或 .agents/skills/（优先级高于用户级）。技能是 Markdown 指令，模型经 skill 工具按目录自动发现、按需加载，改动即时热更新。

致谢

架构大量参考了 dsh-pentest（同为 DSH 插件形式的成熟实践），站在 DeepSeek Harness 的插件机制之上。

License

MIT