mirror of
https://github.com/bytedance/deer-flow.git
synced 2026-06-13 10:55:59 +00:00
feat(skill): add blocking-io-guard — SOP skill for blocking-IO triage and runtime anchors (#3503)
* feat(blocking-io): add changed-lines blocking-IO scanner (L1) * feat(blocking-io): add scan-changed CLI wrapper * feat(skill): add blocking-io-guard developer SOP skill * docs(blocking-io): point contributors at the blocking-io-guard skill * style(blocking-io): apply ruff format to scanner and tests * docs(backend): document changed-lines blocking-IO scanner in CLAUDE.md * feat(skill): add post-fix re-scan check and PR batching policy * refactor(skill): fix SOP step ordering, align template with repo conventions - Move re-scan into an explicit 'apply the fix' step (was wedged after anchor generation while telling you to go back before the anchor) - Renumber steps 0-6; drop undefined 'L1' jargon - Mode A: document that the diff is <base>...HEAD (commit first) - Mode B: prefer make detect-blocking-io + findings JSON file - anchor template: module-level pytestmark per tests/blocking_io convention - CLAUDE.md: fix 'git diff --base' phrasing * fix(skill): catch findings introduced without touching the blocking line Review follow-up: changed-line intersection alone misses the case where a new async caller exposes an old sync helper — the static finding sits on the untouched blocking line, so Mode A returned empty and the SOP stopped on a false 'no blocking-IO surface'. Selection is now a union over the changed files: - findings on added lines of git diff <base>...HEAD (kept: a second identical symbol in an already-flagged function collides on the stable key and only this selection sees it); - findings new versus the merge base, matched by (path, function, symbol) — never line numbers. Base sources are materialized via git show <merge-base>:<path>; files absent at base count every head finding as new. SKILL.md now states the residual same-file-only blind spot (cross-file async callers) instead of treating an empty list as proof of zero exposure, and only requires reading sop-skeleton.md when generalizing to another detector domain. * docs(skill): examples teach test-writing, the teeth check defines the rule All examples in the references/template are filesystem-flavored; make explicit that they are instances, not the SOP's boundary — the same rules apply to every detector category (FILE_IO, HTTP, SUBPROCESS, SLEEP) and acceptance is always red/green teeth, never similarity to an example. Neutralize the template's arrange comment accordingly. * fix(blocking-io): harden changed-lines scanner per review - Dedup the union selection by the stable key (path, function, symbol) instead of dict identity, so a future selector returning copied dicts cannot silently empty the result. - parse_changed_lines now handles any unified diff: context lines advance the new-file counter, \-markers and deletions do not, and the counter resets at each +++ header. Previously correct only for --unified=0. - Add blocking_io_static.scan_source (in-memory scan); base-version comparison no longer round-trips through temp files. - Empty Mode A report now prints the same-file-only reachability caveat at the point of use instead of relying on the SOP text alone. * docs(skill): bound best-effort cleanup when the offload sits in finally Lesson from the #3505 review: the SOP routinely drives 'offload the cleanup branch' transformations, and an awaited cleanup in finally can mask or stall the primary exception. One sentence in Step 2 closes that gap at the point where the fix is written.
This commit is contained in:
@@ -0,0 +1,141 @@
|
|||||||
|
---
|
||||||
|
name: blocking-io-guard
|
||||||
|
description: Ensure async-path backend code that could block the asyncio event loop is protected by a teeth-verified runtime anchor in tests/blocking_io/. Use when changing backend Python under app/, packages/harness/deerflow/, or scripts/, when running a blocking-IO triage round over the whole repo, or when a reviewer/CI asks for blocking-IO coverage. Runs a deterministic scan (changed-lines or full-repo), routes each candidate, drafts/extends an anchor, and proves it fails when the blocking IO regresses.
|
||||||
|
---
|
||||||
|
|
||||||
|
# Blocking-IO Guard Skill
|
||||||
|
|
||||||
|
Help a contributor ship backend async changes together with the runtime anchor
|
||||||
|
that lets DeerFlow's blocking-IO CI gate actually see the new code. The dynamic
|
||||||
|
detector only catches blocking IO on paths a test executes — this skill closes
|
||||||
|
that gap, either for your own diff or for a repo-wide triage round.
|
||||||
|
|
||||||
|
Read `references/good-anchor-rules.md` before writing any anchor.
|
||||||
|
Only read `references/sop-skeleton.md` when generalizing this SOP to another
|
||||||
|
detector domain — it is not needed to execute the steps below.
|
||||||
|
|
||||||
|
## When to use
|
||||||
|
|
||||||
|
- Your change touches Python under `backend/app/`,
|
||||||
|
`backend/packages/harness/deerflow/`, or `backend/scripts/` and may run on
|
||||||
|
the async event loop (Mode A). If unsure, run Step 0 — it answers
|
||||||
|
deterministically.
|
||||||
|
- You are doing a maintenance triage round over the existing codebase
|
||||||
|
(Mode B).
|
||||||
|
|
||||||
|
## SOP (router)
|
||||||
|
|
||||||
|
### Step 0 — Scope (deterministic)
|
||||||
|
|
||||||
|
**Mode A — your own diff** (default, pre-PR). From repo root:
|
||||||
|
|
||||||
|
```bash
|
||||||
|
uv run --project backend python scripts/scan_changed_blocking_io.py --base origin/main
|
||||||
|
```
|
||||||
|
|
||||||
|
Lists blocking-IO candidates your change introduces: findings on lines the
|
||||||
|
diff added, **plus** findings that are new versus the merge base — the latter
|
||||||
|
catches a new async caller exposing an old sync helper whose blocking line is
|
||||||
|
not in the diff. The diff is `<base>...HEAD`, so **commit your work first** —
|
||||||
|
uncommitted lines are not selected.
|
||||||
|
|
||||||
|
If the list is empty, this change introduces no blocking-IO surface *that the
|
||||||
|
static detector can see in the changed files*. One residual blind spot
|
||||||
|
remains: reachability is same-file only, so a new async caller of a sync
|
||||||
|
helper **defined in another file** is invisible to both selections. If your
|
||||||
|
diff adds an async call into a helper that lives elsewhere, check that helper
|
||||||
|
manually (codegraph or `git grep`) before stopping.
|
||||||
|
|
||||||
|
**Mode B — full-repo triage round.** From repo root:
|
||||||
|
|
||||||
|
```bash
|
||||||
|
make detect-blocking-io
|
||||||
|
```
|
||||||
|
|
||||||
|
Prints a summary and writes the complete structured finding list to
|
||||||
|
`.deer-flow/blocking-io-findings.json`. Work HIGH priority first; do not start
|
||||||
|
MEDIUM until every HIGH is dispositioned (fixed, guarded, or recorded
|
||||||
|
NO-ACTION).
|
||||||
|
|
||||||
|
**Batching policy (PR sizing).** One **fix unit** per PR while any HIGH
|
||||||
|
remains: a fix unit is one root cause — usually a single HIGH, but two HIGHs
|
||||||
|
resolved by the same one-place fix belong together. Once no HIGH remains,
|
||||||
|
MEDIUM/LOW may be batched (about five per round, grouped by module or by
|
||||||
|
disposition) so each PR stays reviewable. A new Blockbuster rule is never
|
||||||
|
batched with anything — it always ships alone (see Step 5).
|
||||||
|
|
||||||
|
Both modes emit the same JSON shape per finding: `priority`, `location`
|
||||||
|
(path/line/function), `blocking_call` (category/operation/symbol),
|
||||||
|
`event_loop_exposure`, `reason`, `code`. Priority is a deterministic review
|
||||||
|
ordering, not proof of a bug — Step 1 makes the actual call.
|
||||||
|
|
||||||
|
### Step 1 — Judge each candidate (router)
|
||||||
|
|
||||||
|
Read the code around each candidate and route it:
|
||||||
|
|
||||||
|
- **Already offloaded** (`asyncio.to_thread`, `run_in_executor`, async client) →
|
||||||
|
**GUARD**: add/extend an anchor that locks the offload so a future edit cannot
|
||||||
|
move it back onto the loop.
|
||||||
|
- **On the loop, not offloaded** → **FIX+ANCHOR**: offload the production code
|
||||||
|
(your fix), then add an anchor that guards it.
|
||||||
|
- **Not actually exposed / acceptable** (rare: scanner false positive,
|
||||||
|
startup-only code) → **NO-ACTION**: record one line of why.
|
||||||
|
- **Cross-file caveat**: the scanner's async reachability is same-file only
|
||||||
|
(`ASYNC_REACHABLE_SAME_FILE`). If the candidate is a *sync helper*, check for
|
||||||
|
async callers in other files (codegraph or `git grep`) before deciding
|
||||||
|
NO-ACTION.
|
||||||
|
|
||||||
|
### Step 2 — Apply the fix, then re-scan (FIX+ANCHOR only)
|
||||||
|
|
||||||
|
Offload the blocking call in production code, then re-run the Step 0 scan and
|
||||||
|
confirm the candidate no longer appears. If the offloaded call sits in a
|
||||||
|
`finally` / cleanup path, keep it best-effort and bounded (swallow-and-log,
|
||||||
|
`asyncio.wait_for`) so a failing or hung cleanup cannot mask the primary
|
||||||
|
exception. Match by the stable key
|
||||||
|
**(path, function, symbol)** — line numbers shift after edits, so never
|
||||||
|
compare by line.
|
||||||
|
|
||||||
|
- The finding must disappear. If it still shows, the fix did not remove the
|
||||||
|
blocking pattern (e.g. the call is still a direct call, not offloaded) —
|
||||||
|
go back before touching any test.
|
||||||
|
- GUARD / NO-ACTION routes skip this step: a residual finding there is
|
||||||
|
*expected* (the raw call still exists inside a sync helper with the offload
|
||||||
|
at the caller, or the exposure was judged acceptable).
|
||||||
|
|
||||||
|
This is pattern-level feedback in seconds; it complements but never replaces
|
||||||
|
Step 5 — only the runtime gate proves the event loop is actually protected.
|
||||||
|
|
||||||
|
### Step 3 — Check existing anchors
|
||||||
|
|
||||||
|
Look in `backend/tests/blocking_io/` for a test that drives the production async
|
||||||
|
entry point reaching this candidate's branch.
|
||||||
|
|
||||||
|
- Covers this branch already → go to Step 5 (re-verify teeth).
|
||||||
|
- Covers the entry point but not this branch (e.g. happy path covered,
|
||||||
|
cleanup/404/409 not) → **extend** that anchor.
|
||||||
|
- None → create one from `templates/anchor.template.py`.
|
||||||
|
|
||||||
|
### Step 4 — Generate / extend the anchor
|
||||||
|
|
||||||
|
Follow `references/good-anchor-rules.md`. Drive the *specific* branch (e.g. force
|
||||||
|
the create failure that hits the cleanup `shutil.rmtree`). Never bypass the
|
||||||
|
blocking surface with a test-only `asyncio.to_thread` wrapper.
|
||||||
|
|
||||||
|
### Step 5 — Verify teeth (mandatory; also the anchor-vs-rule discriminator)
|
||||||
|
|
||||||
|
1. Reintroduce the block (GUARD: temporarily revert the offload; FIX+ANCHOR: run
|
||||||
|
against the pre-fix code).
|
||||||
|
2. Run `cd backend && make test-blocking-io` (or target the one test). It **must
|
||||||
|
go RED**.
|
||||||
|
3. Restore the fix. It **must go GREEN**.
|
||||||
|
|
||||||
|
A real block that stays GREEN means Blockbuster has no rule for that
|
||||||
|
primitive — that is the **RULE** route; see `references/good-anchor-rules.md`
|
||||||
|
for the admission criteria before adding one.
|
||||||
|
|
||||||
|
### Step 6 — Deliver
|
||||||
|
|
||||||
|
Commit the anchor(s) with your change; `make test-blocking-io` green. In the PR,
|
||||||
|
note: candidates found, each disposition, the re-scan result (Step 2), and
|
||||||
|
the teeth evidence (red→green). Include the reason for any NO-ACTION. A new
|
||||||
|
Blockbuster rule, if any, goes in its own commit with the evidence from Step 5.
|
||||||
@@ -0,0 +1,65 @@
|
|||||||
|
# Good anchor rules + teeth (blocking-IO fill)
|
||||||
|
|
||||||
|
Distilled from `backend/docs/BLOCKING_IO_DETECTION.md`. An anchor lives in
|
||||||
|
`backend/tests/blocking_io/`; the suite's conftest runs each test under the
|
||||||
|
strict Blockbuster gate scoped to `app.*` / `deerflow.*`.
|
||||||
|
|
||||||
|
The examples in this file and in `templates/` are all filesystem-flavored.
|
||||||
|
They demonstrate how to *write* the test, not what the SOP covers: the same
|
||||||
|
rules apply to every category the detector reports (FILE_IO, HTTP,
|
||||||
|
SUBPROCESS, SLEEP), and the acceptance criterion is always the teeth check
|
||||||
|
below — never similarity to an example.
|
||||||
|
|
||||||
|
## A good anchor
|
||||||
|
|
||||||
|
- Calls the **real production async entry point** — not a low-level helper,
|
||||||
|
unless that helper *is* the entry point production executes.
|
||||||
|
- Does **not** bypass the blocking surface with a test-only
|
||||||
|
`asyncio.to_thread` / `run_in_executor` wrapper.
|
||||||
|
- Uses **real local filesystem** inputs when the bug shape is filesystem IO.
|
||||||
|
- Mocks **only** the external dependency boundary (network service, third-party
|
||||||
|
saver), never the offload being guarded.
|
||||||
|
- Drives the **specific branch** you are protecting (error / cleanup / 404 /
|
||||||
|
409), not just the happy path.
|
||||||
|
|
||||||
|
## Teeth (the acceptance test)
|
||||||
|
|
||||||
|
An anchor only counts if the gate actually fires when the code blocks:
|
||||||
|
|
||||||
|
1. Reintroduce the block (revert the offload, or run pre-fix code).
|
||||||
|
2. `cd backend && make test-blocking-io` → the anchor **must fail** (RED).
|
||||||
|
3. Restore the fix → the anchor **must pass** (GREEN).
|
||||||
|
|
||||||
|
A green-on-happy-path anchor with no proven red is fake coverage. Don't ship it.
|
||||||
|
|
||||||
|
## The RULE route (rare; strict admission criteria)
|
||||||
|
|
||||||
|
Blockbuster's built-in rules cover the common blocking primitives well. The
|
||||||
|
two deliberate openings in this SOP are:
|
||||||
|
|
||||||
|
1. **Coverage opening** (the normal case): the rules already see the
|
||||||
|
primitive — you only need an anchor so runtime detection executes the real
|
||||||
|
business path and CI prevents regression.
|
||||||
|
2. **Rule opening** (rare): you reintroduced a *real* block and the gate
|
||||||
|
stayed GREEN — Blockbuster has no rule for that primitive.
|
||||||
|
|
||||||
|
A project rule lives in `_PROJECT_BLOCKING_RULES` inside
|
||||||
|
`backend/tests/support/detectors/blocking_io_runtime.py` and changes detection
|
||||||
|
for the **entire** blocking-IO suite — global blast radius. Admission criteria
|
||||||
|
for adding one:
|
||||||
|
|
||||||
|
- You have the **fails-to-fail anchor** as evidence: a good anchor (per the
|
||||||
|
rules above) that drives a genuinely blocking path and stays green. No
|
||||||
|
evidence, no rule.
|
||||||
|
- The primitive is a real blocking call (verified against its implementation
|
||||||
|
or docs), not a false positive of the static detector.
|
||||||
|
- The rule ships in its **own commit**, naming the primitive, the anchor that
|
||||||
|
exposed the gap, and the suite-wide impact. Run the full
|
||||||
|
`make test-blocking-io` suite after adding it — a new rule can turn other
|
||||||
|
previously-green tests red, and each such red is either a real latent bug
|
||||||
|
(fix it) or rule overreach (narrow the rule).
|
||||||
|
- If you are not in a position to own that blast radius (e.g. external
|
||||||
|
contributor), escalate to a maintainer with the evidence instead.
|
||||||
|
|
||||||
|
**Never add a runtime rule just because a path is untested** — that case needs
|
||||||
|
an anchor, not a rule.
|
||||||
@@ -0,0 +1,34 @@
|
|||||||
|
# SOP skeleton (generic shape — extraction seam)
|
||||||
|
|
||||||
|
This is the domain-agnostic shape the blocking-IO skill instantiates. It exists
|
||||||
|
so a second detector/gate domain can reuse the flow without copying it. Do not
|
||||||
|
add machinery for that until a second domain actually appears (YAGNI).
|
||||||
|
|
||||||
|
A domain provides:
|
||||||
|
- a **static detector** that can scan a diff (or the whole tree) and emit
|
||||||
|
located candidates,
|
||||||
|
- a **CI gate** that fails when the bad pattern executes,
|
||||||
|
- a **test location** for guard tests,
|
||||||
|
- **good-test rules** for that gate,
|
||||||
|
- a **teeth definition** (how to make the gate fire on purpose).
|
||||||
|
|
||||||
|
Steps:
|
||||||
|
1. **Scope (deterministic):** intersect the diff's added lines with the
|
||||||
|
detector's findings → candidates this change introduced/touched. (Or, in
|
||||||
|
triage mode, take the full finding list ordered by priority.)
|
||||||
|
2. **Judge (router):** per candidate — guard existing fix / fix + guard /
|
||||||
|
no-action / rule (the gate cannot see the primitive).
|
||||||
|
3. **Fix + re-scope (fixes only):** apply the fix, re-run the detector; the
|
||||||
|
fixed candidate must vanish from the findings (match by a stable key, not
|
||||||
|
line numbers). Pattern-level feedback in seconds — complements, never
|
||||||
|
replaces, step 5.
|
||||||
|
4. **Generate:** draft or extend a guard test per the good-test rules, driving
|
||||||
|
the specific branch.
|
||||||
|
5. **Verify teeth:** make the bad pattern happen → gate must fail; restore →
|
||||||
|
gate must pass. A pattern that stays green while genuinely bad is the
|
||||||
|
"rule" signal, not a coverage success.
|
||||||
|
6. **Deliver:** commit the verified guard test; any gate-rule change ships in
|
||||||
|
its own commit with the fails-to-fail evidence attached.
|
||||||
|
|
||||||
|
To add a domain: supply a new fill doc (like `good-anchor-rules.md`) + detector,
|
||||||
|
and promote this file into a parent skill the instances point at.
|
||||||
@@ -0,0 +1,32 @@
|
|||||||
|
"""Template: a tests/blocking_io/ runtime anchor.
|
||||||
|
|
||||||
|
Copy into backend/tests/blocking_io/test_<area>.py and adapt. The suite's
|
||||||
|
conftest already wraps every test here in the strict Blockbuster gate, so you do
|
||||||
|
NOT import or activate the detector — just drive the real async entry point.
|
||||||
|
|
||||||
|
Teeth check before you commit (see references/good-anchor-rules.md):
|
||||||
|
1. reintroduce the block -> `cd backend && make test-blocking-io` must FAIL
|
||||||
|
2. restore the fix -> it must PASS
|
||||||
|
"""
|
||||||
|
|
||||||
|
from __future__ import annotations
|
||||||
|
|
||||||
|
from pathlib import Path
|
||||||
|
|
||||||
|
import pytest
|
||||||
|
|
||||||
|
# from app.<module> import <real_async_entry_point>
|
||||||
|
|
||||||
|
pytestmark = pytest.mark.asyncio
|
||||||
|
|
||||||
|
|
||||||
|
async def test_<entry_point>_offloads_blocking_io_on_<branch>(tmp_path: Path) -> None:
|
||||||
|
# Arrange: real inputs at the boundary the code blocks on (FS -> tmp_path;
|
||||||
|
# HTTP/subprocess -> stub the external service). Mock ONLY the external
|
||||||
|
# boundary, never the offload under test.
|
||||||
|
|
||||||
|
# Act + Assert: call the REAL production async entry point and drive the
|
||||||
|
# specific branch you are guarding (e.g. force a failure to hit the cleanup
|
||||||
|
# path). If the entry point performs blocking IO on the loop, the gate fails.
|
||||||
|
# await <real_async_entry_point>(...)
|
||||||
|
raise NotImplementedError("Replace with the real async entry point call.")
|
||||||
@@ -112,6 +112,14 @@ calls are resolved by function name, so duplicate helper names in one file can
|
|||||||
conservatively over-report async reachability. It is intentionally
|
conservatively over-report async reachability. It is intentionally
|
||||||
informational and is not run from CI in this round.
|
informational and is not run from CI in this round.
|
||||||
|
|
||||||
|
For a diff-scoped view of the same findings, `scripts/scan_changed_blocking_io.py`
|
||||||
|
(repo root) reports findings on the added lines of `git diff <base>...HEAD`
|
||||||
|
plus findings new versus the merge base (so a new async caller exposing an
|
||||||
|
untouched sync helper in the same file is still reported) — used by the
|
||||||
|
`blocking-io-guard` skill (`.agent/skills/blocking-io-guard/`) as the
|
||||||
|
deterministic scope step before routing each candidate to a fix and/or a
|
||||||
|
`tests/blocking_io/` runtime anchor.
|
||||||
|
|
||||||
Regression tests related to Docker/provisioner behavior:
|
Regression tests related to Docker/provisioner behavior:
|
||||||
- `tests/test_docker_sandbox_mode_detection.py` (mode detection from `config.yaml`)
|
- `tests/test_docker_sandbox_mode_detection.py` (mode detection from `config.yaml`)
|
||||||
- `tests/test_provisioner_kubeconfig.py` (kubeconfig file/directory handling)
|
- `tests/test_provisioner_kubeconfig.py` (kubeconfig file/directory handling)
|
||||||
|
|||||||
@@ -67,6 +67,11 @@ The normal workflow is:
|
|||||||
3. Add or update a focused runtime anchor in `backend/tests/blocking_io/`.
|
3. Add or update a focused runtime anchor in `backend/tests/blocking_io/`.
|
||||||
4. Let CI prevent that path from regressing.
|
4. Let CI prevent that path from regressing.
|
||||||
|
|
||||||
|
Contributors changing backend async code can run the `blocking-io-guard` skill
|
||||||
|
(`.agent/skills/blocking-io-guard/`) to execute steps 1–3 for their own diff: it
|
||||||
|
scans the change for blocking-IO candidates, drafts or extends a runtime anchor,
|
||||||
|
and verifies the anchor fails when the blocking IO regresses.
|
||||||
|
|
||||||
Runtime detection has two maintenance paths.
|
Runtime detection has two maintenance paths.
|
||||||
|
|
||||||
### Add a runtime rule
|
### Add a runtime rule
|
||||||
|
|||||||
@@ -0,0 +1,213 @@
|
|||||||
|
"""Intersect a git diff with static blocking-IO findings.
|
||||||
|
|
||||||
|
Wraps the static detector (`blocking_io_static`) to answer a narrower question:
|
||||||
|
which blocking-IO candidates does THIS change introduce? A candidate qualifies
|
||||||
|
when its blocking line is on an added line of the diff, or when the finding is
|
||||||
|
new versus the merge base — the latter catches exposure created without
|
||||||
|
touching the blocking line itself (a new async caller making an old sync
|
||||||
|
helper async-reachable). Used by the `blocking-io-guard` skill as the
|
||||||
|
deterministic scope step.
|
||||||
|
"""
|
||||||
|
|
||||||
|
from __future__ import annotations
|
||||||
|
|
||||||
|
import argparse
|
||||||
|
import json
|
||||||
|
import re
|
||||||
|
import subprocess
|
||||||
|
import sys
|
||||||
|
from collections import defaultdict
|
||||||
|
from collections.abc import Sequence
|
||||||
|
from pathlib import Path
|
||||||
|
|
||||||
|
from support.detectors import blocking_io_static as static
|
||||||
|
|
||||||
|
REPO_ROOT = Path(__file__).resolve().parents[4]
|
||||||
|
SCAN_ROOTS = (
|
||||||
|
"backend/app",
|
||||||
|
"backend/packages/harness/deerflow",
|
||||||
|
"backend/scripts",
|
||||||
|
)
|
||||||
|
|
||||||
|
_HUNK_RE = re.compile(r"^@@ -\d+(?:,\d+)? \+(\d+)(?:,\d+)? @@")
|
||||||
|
|
||||||
|
|
||||||
|
def parse_changed_lines(diff_text: str) -> dict[str, set[int]]:
|
||||||
|
"""Map repo-relative path -> set of added line numbers in the new file.
|
||||||
|
|
||||||
|
Accepts any unified diff (with or without `--unified=0`): context lines
|
||||||
|
advance the new-file counter, deletions (`-`) and `\\ No newline` markers
|
||||||
|
do not. Records only added lines (`+`, not the `+++` header), numbered
|
||||||
|
from each hunk's new-file start line; deleted files (`+++ /dev/null`) are
|
||||||
|
skipped.
|
||||||
|
"""
|
||||||
|
changed: dict[str, set[int]] = defaultdict(set)
|
||||||
|
current_path: str | None = None
|
||||||
|
next_line = 0
|
||||||
|
for raw in diff_text.splitlines():
|
||||||
|
if raw.startswith("+++ "):
|
||||||
|
target = raw[4:].strip()
|
||||||
|
if target == "/dev/null":
|
||||||
|
current_path = None
|
||||||
|
else:
|
||||||
|
current_path = target[2:] if target.startswith("b/") else target
|
||||||
|
next_line = 0
|
||||||
|
continue
|
||||||
|
match = _HUNK_RE.match(raw)
|
||||||
|
if match:
|
||||||
|
next_line = int(match.group(1))
|
||||||
|
continue
|
||||||
|
if not current_path:
|
||||||
|
continue
|
||||||
|
if raw.startswith("+"):
|
||||||
|
changed[current_path].add(next_line)
|
||||||
|
next_line += 1
|
||||||
|
elif raw.startswith(" ") or raw == "":
|
||||||
|
next_line += 1
|
||||||
|
return dict(changed)
|
||||||
|
|
||||||
|
|
||||||
|
def changed_python_lines(base: str, repo_root: Path = REPO_ROOT) -> dict[str, set[int]]:
|
||||||
|
"""Diff `base...HEAD` over scan roots and return added .py lines."""
|
||||||
|
cmd = [
|
||||||
|
"git",
|
||||||
|
"-C",
|
||||||
|
str(repo_root),
|
||||||
|
"diff",
|
||||||
|
"--unified=0",
|
||||||
|
"--no-color",
|
||||||
|
f"{base}...HEAD",
|
||||||
|
"--",
|
||||||
|
*SCAN_ROOTS,
|
||||||
|
]
|
||||||
|
diff_text = subprocess.run(cmd, capture_output=True, text=True, check=True).stdout
|
||||||
|
return {path: lines for path, lines in parse_changed_lines(diff_text).items() if path.endswith(".py")}
|
||||||
|
|
||||||
|
|
||||||
|
def select_findings_on_changed_lines(
|
||||||
|
findings: Sequence[dict[str, object]],
|
||||||
|
changed_lines: dict[str, set[int]],
|
||||||
|
) -> list[dict[str, object]]:
|
||||||
|
"""Keep findings whose (path, line) falls on a changed line."""
|
||||||
|
selected: list[dict[str, object]] = []
|
||||||
|
for finding in findings:
|
||||||
|
location = finding["location"] # type: ignore[index]
|
||||||
|
path = location["path"] # type: ignore[index]
|
||||||
|
line = location["line"] # type: ignore[index]
|
||||||
|
if line in changed_lines.get(path, set()):
|
||||||
|
selected.append(finding)
|
||||||
|
return selected
|
||||||
|
|
||||||
|
|
||||||
|
def base_python_contents(base: str, paths: Sequence[str], repo_root: Path = REPO_ROOT) -> dict[str, str]:
|
||||||
|
"""Return each path's content at the merge base of `base` and HEAD.
|
||||||
|
|
||||||
|
Files absent at the merge base (newly added) are omitted, so every head
|
||||||
|
finding in them counts as new.
|
||||||
|
"""
|
||||||
|
merge_base = subprocess.run(
|
||||||
|
["git", "-C", str(repo_root), "merge-base", base, "HEAD"],
|
||||||
|
capture_output=True,
|
||||||
|
text=True,
|
||||||
|
check=True,
|
||||||
|
).stdout.strip()
|
||||||
|
contents: dict[str, str] = {}
|
||||||
|
for path in paths:
|
||||||
|
shown = subprocess.run(
|
||||||
|
["git", "-C", str(repo_root), "show", f"{merge_base}:{path}"],
|
||||||
|
capture_output=True,
|
||||||
|
text=True,
|
||||||
|
)
|
||||||
|
if shown.returncode == 0:
|
||||||
|
contents[path] = shown.stdout
|
||||||
|
return contents
|
||||||
|
|
||||||
|
|
||||||
|
def scan_python_contents(contents: dict[str, str]) -> list[dict[str, object]]:
|
||||||
|
"""Run the static detector over in-memory sources (repo-relative path -> code)."""
|
||||||
|
findings: list[dict[str, object]] = []
|
||||||
|
for rel_path in sorted(contents):
|
||||||
|
findings.extend(finding.to_dict() for finding in static.scan_source(contents[rel_path], rel_path))
|
||||||
|
return findings
|
||||||
|
|
||||||
|
|
||||||
|
def _stable_key(finding: dict[str, object]) -> tuple[str, str, str]:
|
||||||
|
location = finding["location"] # type: ignore[index]
|
||||||
|
call = finding["blocking_call"] # type: ignore[index]
|
||||||
|
return (location["path"], location["function"], call["symbol"]) # type: ignore[index]
|
||||||
|
|
||||||
|
|
||||||
|
def select_findings_new_vs_base(
|
||||||
|
head_findings: Sequence[dict[str, object]],
|
||||||
|
base_findings: Sequence[dict[str, object]],
|
||||||
|
) -> list[dict[str, object]]:
|
||||||
|
"""Keep head findings whose stable key (path, function, symbol) is absent at base.
|
||||||
|
|
||||||
|
Line numbers shift between revisions, so matching is by stable key only.
|
||||||
|
A second identical symbol added inside a function that already had a
|
||||||
|
finding collides on the key and is NOT reported here — that case is
|
||||||
|
covered by the changed-line selection instead.
|
||||||
|
"""
|
||||||
|
base_keys = {_stable_key(finding) for finding in base_findings}
|
||||||
|
return [finding for finding in head_findings if _stable_key(finding) not in base_keys]
|
||||||
|
|
||||||
|
|
||||||
|
def find_changed_blocking_io(base: str, repo_root: Path = REPO_ROOT) -> list[dict[str, object]]:
|
||||||
|
"""Return static findings this change introduces or touches.
|
||||||
|
|
||||||
|
Union over the changed files of:
|
||||||
|
- findings whose blocking line is on an added line of the diff;
|
||||||
|
- findings new versus the merge base (a new async caller can expose an
|
||||||
|
untouched sync helper — the blocking line itself is not in the diff).
|
||||||
|
"""
|
||||||
|
changed_lines = changed_python_lines(base, repo_root)
|
||||||
|
if not changed_lines:
|
||||||
|
return []
|
||||||
|
files = [repo_root / path for path in changed_lines]
|
||||||
|
head_findings = [finding.to_dict() for finding in static.scan_paths(files, repo_root=repo_root)]
|
||||||
|
on_changed_lines = select_findings_on_changed_lines(head_findings, changed_lines)
|
||||||
|
base_findings = scan_python_contents(base_python_contents(base, sorted(changed_lines), repo_root))
|
||||||
|
new_vs_base = select_findings_new_vs_base(head_findings, base_findings)
|
||||||
|
selected_keys = {_stable_key(finding) for finding in (*on_changed_lines, *new_vs_base)}
|
||||||
|
return [finding for finding in head_findings if _stable_key(finding) in selected_keys]
|
||||||
|
|
||||||
|
|
||||||
|
def format_report(findings: Sequence[dict[str, object]], base: str) -> str:
|
||||||
|
if not findings:
|
||||||
|
return (
|
||||||
|
f"No blocking-IO candidates introduced by this change (base: {base}).\n"
|
||||||
|
"Note: async reachability is resolved within each file only. If this change\n"
|
||||||
|
"adds an async call into a sync helper defined in another file, check that\n"
|
||||||
|
"helper manually (codegraph or git grep) before relying on this empty result."
|
||||||
|
)
|
||||||
|
lines = [
|
||||||
|
f"Blocking-IO candidates introduced/touched by this change (base: {base}): {len(findings)}",
|
||||||
|
"",
|
||||||
|
]
|
||||||
|
order = {"HIGH": 0, "MEDIUM": 1, "LOW": 2}
|
||||||
|
for finding in sorted(findings, key=lambda f: order.get(str(f["priority"]), 9)):
|
||||||
|
location = finding["location"] # type: ignore[index]
|
||||||
|
call = finding["blocking_call"] # type: ignore[index]
|
||||||
|
lines.append(f"{finding['priority']} {call['category']}/{call['operation']} {location['path']}:{location['line']} in {location['function']} exposure={finding['event_loop_exposure']}")
|
||||||
|
lines.append(f" symbol: {call['symbol']}")
|
||||||
|
if finding.get("code"):
|
||||||
|
lines.append(f" code: {finding['code']}")
|
||||||
|
return "\n".join(lines)
|
||||||
|
|
||||||
|
|
||||||
|
def main(argv: Sequence[str] | None = None) -> int:
|
||||||
|
parser = argparse.ArgumentParser(description="List blocking-IO candidates this change introduces: findings on added lines plus findings new versus the merge base (diff against --base).")
|
||||||
|
parser.add_argument("--base", default="origin/main", help="Base ref to diff against (default: origin/main).")
|
||||||
|
parser.add_argument("--format", choices=("text", "json"), default="text", help="Output format.")
|
||||||
|
args = parser.parse_args(argv)
|
||||||
|
|
||||||
|
findings = find_changed_blocking_io(args.base)
|
||||||
|
if args.format == "json":
|
||||||
|
print(json.dumps(findings, indent=2))
|
||||||
|
else:
|
||||||
|
print(format_report(findings, args.base))
|
||||||
|
return 0
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
sys.exit(main())
|
||||||
@@ -717,12 +717,11 @@ def _finalize_findings(visitor: BlockingIOStaticVisitor) -> list[BlockingIOStati
|
|||||||
return findings
|
return findings
|
||||||
|
|
||||||
|
|
||||||
def scan_file(path: Path, *, repo_root: Path = REPO_ROOT) -> list[BlockingIOStaticFinding]:
|
def scan_source(source: str, relative_path: str) -> list[BlockingIOStaticFinding]:
|
||||||
source = path.read_text(encoding="utf-8")
|
"""Scan one in-memory Python source; `relative_path` is reported verbatim in findings."""
|
||||||
source_lines = source.splitlines()
|
source_lines = source.splitlines()
|
||||||
relative_path = relative_to_repo(path, repo_root)
|
|
||||||
try:
|
try:
|
||||||
tree = ast.parse(source, filename=str(path))
|
tree = ast.parse(source, filename=relative_path)
|
||||||
except SyntaxError as exc:
|
except SyntaxError as exc:
|
||||||
line = exc.lineno or 0
|
line = exc.lineno or 0
|
||||||
code = _source_snippet(source_lines, line)
|
code = _source_snippet(source_lines, line)
|
||||||
@@ -746,6 +745,10 @@ def scan_file(path: Path, *, repo_root: Path = REPO_ROOT) -> list[BlockingIOStat
|
|||||||
return sorted(_finalize_findings(visitor), key=lambda finding: (finding.path, finding.line, finding.column, finding.category))
|
return sorted(_finalize_findings(visitor), key=lambda finding: (finding.path, finding.line, finding.column, finding.category))
|
||||||
|
|
||||||
|
|
||||||
|
def scan_file(path: Path, *, repo_root: Path = REPO_ROOT) -> list[BlockingIOStaticFinding]:
|
||||||
|
return scan_source(path.read_text(encoding="utf-8"), relative_to_repo(path, repo_root))
|
||||||
|
|
||||||
|
|
||||||
def is_ignored_path(path: Path) -> bool:
|
def is_ignored_path(path: Path) -> bool:
|
||||||
return any(part in IGNORED_DIR_NAMES for part in path.parts)
|
return any(part in IGNORED_DIR_NAMES for part in path.parts)
|
||||||
|
|
||||||
|
|||||||
@@ -0,0 +1,175 @@
|
|||||||
|
from __future__ import annotations
|
||||||
|
|
||||||
|
import textwrap
|
||||||
|
from pathlib import Path
|
||||||
|
|
||||||
|
from support.detectors import blocking_io_changed as changed
|
||||||
|
from support.detectors import blocking_io_static as static
|
||||||
|
|
||||||
|
|
||||||
|
def _write_python(path: Path, source: str) -> Path:
|
||||||
|
path.write_text(textwrap.dedent(source).strip() + "\n", encoding="utf-8")
|
||||||
|
return path
|
||||||
|
|
||||||
|
|
||||||
|
_CLEANUP_BRANCH_SOURCE = """
|
||||||
|
import shutil
|
||||||
|
from pathlib import Path
|
||||||
|
|
||||||
|
async def create_agent(path: Path) -> None:
|
||||||
|
path.mkdir()
|
||||||
|
try:
|
||||||
|
await _save(path)
|
||||||
|
except Exception:
|
||||||
|
shutil.rmtree(path)
|
||||||
|
raise
|
||||||
|
"""
|
||||||
|
|
||||||
|
|
||||||
|
def test_parse_changed_lines_records_added_lines_only() -> None:
|
||||||
|
diff = textwrap.dedent(
|
||||||
|
"""\
|
||||||
|
diff --git a/backend/app/x.py b/backend/app/x.py
|
||||||
|
--- a/backend/app/x.py
|
||||||
|
+++ b/backend/app/x.py
|
||||||
|
@@ -10,0 +11,2 @@ def f():
|
||||||
|
+ a = 1
|
||||||
|
+ b = 2
|
||||||
|
@@ -20 +22,0 @@ def g():
|
||||||
|
- gone = 1
|
||||||
|
"""
|
||||||
|
)
|
||||||
|
assert changed.parse_changed_lines(diff) == {"backend/app/x.py": {11, 12}}
|
||||||
|
|
||||||
|
|
||||||
|
def test_parse_changed_lines_handles_context_diffs() -> None:
|
||||||
|
diff = textwrap.dedent(
|
||||||
|
"""\
|
||||||
|
diff --git a/backend/app/x.py b/backend/app/x.py
|
||||||
|
--- a/backend/app/x.py
|
||||||
|
+++ b/backend/app/x.py
|
||||||
|
@@ -8,7 +8,8 @@ def f():
|
||||||
|
ctx1
|
||||||
|
ctx2
|
||||||
|
- removed
|
||||||
|
+ added_one
|
||||||
|
ctx3
|
||||||
|
+ added_two
|
||||||
|
ctx4
|
||||||
|
\\ No newline at end of file
|
||||||
|
"""
|
||||||
|
)
|
||||||
|
assert changed.parse_changed_lines(diff) == {"backend/app/x.py": {10, 12}}
|
||||||
|
|
||||||
|
|
||||||
|
def test_parse_changed_lines_ignores_deleted_files() -> None:
|
||||||
|
diff = textwrap.dedent(
|
||||||
|
"""\
|
||||||
|
diff --git a/x.py b/x.py
|
||||||
|
+++ /dev/null
|
||||||
|
@@ -1,2 +0,0 @@
|
||||||
|
-gone
|
||||||
|
"""
|
||||||
|
)
|
||||||
|
assert changed.parse_changed_lines(diff) == {}
|
||||||
|
|
||||||
|
|
||||||
|
def test_select_findings_keeps_only_touched_candidates(tmp_path: Path) -> None:
|
||||||
|
src = _write_python(tmp_path / "agents.py", _CLEANUP_BRANCH_SOURCE)
|
||||||
|
findings = [f.to_dict() for f in static.scan_file(src, repo_root=tmp_path)]
|
||||||
|
rmtree = next(f for f in findings if f["blocking_call"]["symbol"] == "shutil.rmtree")
|
||||||
|
other = next(f for f in findings if f["blocking_call"]["symbol"] != "shutil.rmtree")
|
||||||
|
|
||||||
|
changed_lines = {"agents.py": {rmtree["location"]["line"]}}
|
||||||
|
selected = changed.select_findings_on_changed_lines(findings, changed_lines)
|
||||||
|
|
||||||
|
assert [f["blocking_call"]["symbol"] for f in selected] == ["shutil.rmtree"]
|
||||||
|
assert other not in selected
|
||||||
|
|
||||||
|
|
||||||
|
def test_find_changed_blocking_io_surfaces_only_changed_candidate(tmp_path: Path, monkeypatch) -> None:
|
||||||
|
src = _write_python(tmp_path / "agents.py", _CLEANUP_BRANCH_SOURCE)
|
||||||
|
all_findings = [f.to_dict() for f in static.scan_file(src, repo_root=tmp_path)]
|
||||||
|
rmtree_line = next(f["location"]["line"] for f in all_findings if f["blocking_call"]["symbol"] == "shutil.rmtree")
|
||||||
|
|
||||||
|
# Stub only the git boundary; the static scan runs for real against tmp_path.
|
||||||
|
monkeypatch.setattr(
|
||||||
|
changed,
|
||||||
|
"changed_python_lines",
|
||||||
|
lambda base, repo_root: {"agents.py": {rmtree_line}},
|
||||||
|
)
|
||||||
|
# Base content identical to head: every finding already existed, so only
|
||||||
|
# the changed-line selection contributes (and the union must not double).
|
||||||
|
monkeypatch.setattr(
|
||||||
|
changed,
|
||||||
|
"base_python_contents",
|
||||||
|
lambda base, paths, repo_root: {"agents.py": src.read_text(encoding="utf-8")},
|
||||||
|
)
|
||||||
|
|
||||||
|
result = changed.find_changed_blocking_io("origin/main", repo_root=tmp_path)
|
||||||
|
|
||||||
|
assert [f["blocking_call"]["symbol"] for f in result] == ["shutil.rmtree"]
|
||||||
|
|
||||||
|
|
||||||
|
_SYNC_HELPER_BASE = """
|
||||||
|
from pathlib import Path
|
||||||
|
|
||||||
|
def load(path: Path) -> str:
|
||||||
|
return path.read_text()
|
||||||
|
"""
|
||||||
|
|
||||||
|
_SYNC_HELPER_HEAD = """
|
||||||
|
from pathlib import Path
|
||||||
|
|
||||||
|
def load(path: Path) -> str:
|
||||||
|
return path.read_text()
|
||||||
|
|
||||||
|
async def route(path: Path) -> str:
|
||||||
|
return load(path)
|
||||||
|
"""
|
||||||
|
|
||||||
|
|
||||||
|
def test_new_async_caller_exposing_old_sync_helper_is_reported(tmp_path: Path, monkeypatch) -> None:
|
||||||
|
"""The blocking line is NOT in the diff — only the new async caller is.
|
||||||
|
|
||||||
|
The finding sits on the untouched `read_text` line, so changed-line
|
||||||
|
selection alone would return empty; the new-vs-base comparison must
|
||||||
|
surface it.
|
||||||
|
"""
|
||||||
|
src = _write_python(tmp_path / "mod.py", _SYNC_HELPER_HEAD)
|
||||||
|
head_findings = [f.to_dict() for f in static.scan_file(src, repo_root=tmp_path)]
|
||||||
|
read_text_line = next(f["location"]["line"] for f in head_findings if f["blocking_call"]["symbol"] == "path.read_text")
|
||||||
|
added_lines = {line for line in range(1, len(src.read_text().splitlines()) + 1) if line > read_text_line}
|
||||||
|
|
||||||
|
monkeypatch.setattr(changed, "changed_python_lines", lambda base, repo_root: {"mod.py": added_lines})
|
||||||
|
monkeypatch.setattr(
|
||||||
|
changed,
|
||||||
|
"base_python_contents",
|
||||||
|
lambda base, paths, repo_root: {"mod.py": textwrap.dedent(_SYNC_HELPER_BASE).strip() + "\n"},
|
||||||
|
)
|
||||||
|
|
||||||
|
result = changed.find_changed_blocking_io("origin/main", repo_root=tmp_path)
|
||||||
|
|
||||||
|
assert len(result) == 1
|
||||||
|
assert result[0]["blocking_call"]["symbol"] == "path.read_text"
|
||||||
|
assert result[0]["event_loop_exposure"] == "ASYNC_REACHABLE_SAME_FILE"
|
||||||
|
|
||||||
|
|
||||||
|
def test_select_findings_new_vs_base_matches_by_stable_key(tmp_path: Path) -> None:
|
||||||
|
head = _write_python(tmp_path / "mod.py", _SYNC_HELPER_HEAD)
|
||||||
|
head_findings = [f.to_dict() for f in static.scan_file(head, repo_root=tmp_path)]
|
||||||
|
|
||||||
|
base_findings = changed.scan_python_contents({"mod.py": textwrap.dedent(_SYNC_HELPER_BASE).strip() + "\n"})
|
||||||
|
assert base_findings == [] # no async exposure at base -> detector is silent
|
||||||
|
|
||||||
|
new = changed.select_findings_new_vs_base(head_findings, base_findings)
|
||||||
|
assert [f["blocking_call"]["symbol"] for f in new] == ["path.read_text"]
|
||||||
|
|
||||||
|
# Same content at base and head -> nothing is new, regardless of line drift.
|
||||||
|
assert changed.select_findings_new_vs_base(head_findings, head_findings) == []
|
||||||
|
|
||||||
|
|
||||||
|
def test_format_report_empty_warns_about_cross_file_blind_spot() -> None:
|
||||||
|
report = changed.format_report([], base="origin/main")
|
||||||
|
assert "No blocking-IO candidates" in report
|
||||||
|
assert "defined in another file" in report
|
||||||
@@ -0,0 +1,23 @@
|
|||||||
|
#!/usr/bin/env python3
|
||||||
|
"""CLI wrapper for the changed-lines blocking IO scanner."""
|
||||||
|
|
||||||
|
from __future__ import annotations
|
||||||
|
|
||||||
|
import sys
|
||||||
|
from collections.abc import Sequence
|
||||||
|
from pathlib import Path
|
||||||
|
|
||||||
|
REPO_ROOT = Path(__file__).resolve().parents[1]
|
||||||
|
TEST_SUPPORT_PATH = REPO_ROOT / "backend" / "tests"
|
||||||
|
if str(TEST_SUPPORT_PATH) not in sys.path:
|
||||||
|
sys.path.insert(0, str(TEST_SUPPORT_PATH))
|
||||||
|
|
||||||
|
|
||||||
|
def main(argv: Sequence[str] | None = None) -> int:
|
||||||
|
from support.detectors.blocking_io_changed import main as scanner_main
|
||||||
|
|
||||||
|
return scanner_main(argv)
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
sys.exit(main())
|
||||||
Reference in New Issue
Block a user