文件历史

提交图

97 次代码提交

作者 SHA1 备注 提交日期
gelei 952a12439f docs: mark XBOW as reference-only 2026-07-14 20:53:35 +08:00
Gelei Deng ab5fbb4d90 feat: ship the durable multi-model autonomous PentestGPT runtime (#493)
* first refactor

* feat: dockerized tool with persistent Claude+Codex login + multi-model benchmark

Run the autonomous CTF/pentest tool in Docker with a one-time, persistent login for
BOTH Claude Code and Codex, and add a multi-model benchmark harness.

Backend (multi-model):
- Add `--backend {claude,codex}` to the CTF pipeline. CodexBackend (pentestgpt/core/
  backend.py) wraps unified_agent's Codex backend and translates its events into
  AgentMessages, so the same pipeline runs on Claude (opus/sonnet) or Codex
  (gpt-5.5/gpt-5.4-mini). Wired through config.backend, pipeline stage construction,
  and the CLI (+ PENTESTGPT_CODEX_EFFORT; greppable [CODEX_USAGE] under PENTESTGPT_BENCH=1).

Docker tool (tool-only image; the benchmark stays OUTSIDE the image):
- Extend Dockerfile: Codex CLI (@openai/codex) + openai_codex SDK + unified_agent/
  pentestgpt_agent/pentestgpt_legacy packages + gobuster/dirb + socat. Add .dockerignore
  (keeps creds/benchmark/workspace out of the build context).
- Persistent dual login (the hard part) — asymmetric by token model:
  * Claude: `setup-token` -> token stored in the pentestgpt-claude volume; entrypoint
    exports CLAUDE_CODE_OAUTH_TOKEN (setup-token does not write .credentials.json; macOS
    host creds live in the Keychain and can't be copied).
  * Codex: the container does its OWN `codex login` (NOT seeding -- ChatGPT refresh tokens
    are single-use, so a shared/copied login 401s on first refresh). The 127.0.0.1:1455
    OAuth callback is forwarded into the container via a socat hop (-p 1455:8455).
  * scripts/docker-login.sh is idempotent: checks logins live, logs in only the missing one(s).
- docker-compose codex-config volume (+ pinned names); entrypoint token-export + non-blocking
  preflight; scripts/docker-auth-status.sh; Make targets (docker-build/login/auth-status/
  run/shell/down/nuke).
- Verified end-to-end: one `make docker-login` -> a fresh container reports claude+codex
  logged in with live round-trips; the CTF pipeline (Codex) captured a flag against an
  isolated fixture and the pentest pipeline ran cleanly; persists across recreation, no re-login.

Benchmark (multi-model, host-side):
- benchmark/pilot/ harness (run_pilot.py + report.py): builds each xbow challenge, discovers
  the loopback port, runs the pipeline across the 4 model combos, judges by the baked
  FLAG{sha256(UPPER-dir)}, and renders REPORT.md (infra failures excluded from solve rates).
  Includes the partial pilot's results (results.jsonl + REPORT.md).

Docs: docs/docker-dev-plan.md (full plan + implementation status); CLAUDE.md and README
docker quickstart; benchmark/pilot/README.md; design-doc roadmap (docs/redesign).

Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>

* fix: fail controller on backend error messages

* fix: allow listing sessions without target

* docs: add docker xbow benchmark report

* fix: infer concrete backend constructor type

* docs: refresh docker benchmark documentation

* feat(benchmark): add pure single-agent baseline + pipeline comparison

Add a "pure single agent" benchmark variant -- one bare `claude -p` /
`codex exec` call per target (no pipeline) -- to quantify what the 3-stage
PentestGPT pipeline buys over an un-orchestrated agent on the xbow targets.

- pentestgpt/prompts/stages.py: ctf_single_agent_{system,task}_prompt -- the
  pipeline's shared fragments collapsed into ONE turn, so prompt content is
  held constant and the only variable is the multi-stage decomposition.
- benchmark/pilot/run_docker_bench.py: docker-network runner
  (--variant single|pipeline). Brings the target up, discovers the container's
  internal IP+network (skips DB side-cars/ports), docker-runs the tool image on
  that network, and scores the ground-truth flag against the agent's *assistant
  text* only (parity with the pipeline's raw streaming). Reads stdout in chunks
  to handle >64KB JSON lines. Resumable; --dry-run supported.
- benchmark/pilot/report_comparison.py -> DOCKER_COMPARISON.md: head-to-head
  pipeline-vs-single per model on the common non-infra set.
- tests/unit/test_single_agent_prompt.py: prompt-builder coverage.
- docs: README, CLAUDE.md, benchmark README, DOCKER_REPORT updated.

Recorded result (10 medium/hard targets x 4 models, container-to-container,
same baseline image digest 0c4c0f3e..., commit dca0019 image):

  Model               Pipeline   Single
  Claude Opus           5/10      7/10   (single +2)
  Claude Sonnet         6/10      4/10   (pipeline +2)
  Codex gpt-5.5         7/10      7/10   (tie)
  Codex gpt-5.4-mini    3/10      4/10   (single +1)
  TOTAL                21/40     22/40

Single agent matches the pipeline on solve rate (55% vs 52%) while using
~40% fewer Codex tokens (13.0M vs 21.8M) and solving faster. The pipeline
only clearly helps Claude Sonnet (which times out solo); Opus is better solo.
Full per-challenge grid in DOCKER_COMPARISON.md; raw records in
docker_single_results.jsonl.

Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>

* feat(benchmark): add pentestgpt_agent docker harness

* bench: refresh pentestgpt_agent smoke result

* fix(benchmark): make repeat rows variant-aware

* fix(agent): fall back for semantic executor labels

* fix(agent): tolerate executor prose evidence

* fix(benchmark): score accepted framework findings

* bench: append partial framework repeat results

* bench: complete framework repeat sweep

* bench: expose framework executor concurrency

* bench: add extended parallel framework sweep

* checkpoint: preserve working agent and benchmark state

* feat: harden durable agent loop and xbow qualification

* fix: reserve an exploit result turn

* docs: record clean xbow qualification

* build: consume unified-agent from the git wrapper repo

Repoint pentestgpt_agent_new's unified-agent dependency from the local
editable path (../../UnifiedAgentPoC, now renamed and gone) to the pinned
git source PentestGPT-Project/UnifedAgentWrapper@d05d21f. Regenerate uv.lock
and update test_dependency.py to assert the external package is installed
from that VCS URL (not the repo-root vendored copy) at version 0.2.0.

Co-Authored-By: Claude Fable 5 <noreply@anthropic.com>

* refactor: make pentestgpt_agent_new the sole framework

Remove the retired ledger-based pentestgpt_agent package (instructor/executor/
judge) and its orphaned unit + smoke tests. The nested pentestgpt_agent_new
project (Supervisor/Executor over a durable SQLite loop, consuming unified-agent
from the git wrapper) is now the single maintained framework.

Repoint the top-level tooling to it:
- pyproject: drop the pentestgpt-agent console script and pentestgpt_agent from
  the wheel packages.
- Makefile: lint/format target parent code only; typecheck/check/ci now run the
  nested framework's own gate (ruff, format, mypy, pytest) via test-agent-new /
  check-agent-new, so `make check` finally covers it; `make run` delegates to the
  pentestgpt-agent-new CLI.
- Dockerfile: stop copying the removed package (kept the build working); note the
  framework is not baked into the image yet.
- docker container-health test: import the substrate packages that actually ship.
- CLAUDE.md / AGENT.md: describe the new framework, the git-sourced wrapper, and
  the deprioritized benchmark/Docker rewire.

The XBOW `--variant framework` path and docker-bench Makefile targets still point
at the old in-image framework and are left as a pending rewire (benchmarks
deprioritized); the naive `--variant single` path is unaffected.

Co-Authored-By: Claude Fable 5 <noreply@anthropic.com>

* refactor: rename pentestgpt_agent_new -> pentestgpt_agent

The framework reclaims the clean name now that the old ledger-based package is
gone. Rename the nested project folder, its src package, the distribution
(pentestgpt-agent-new -> pentestgpt-agent) and CLI, and every import/reference in
the package, the umbrella Makefile, the Dockerfile, the docker health test, and
CLAUDE.md / AGENT.md. Regenerate uv.lock. The audit CLI stays pentestgpt-agent-audit;
the git-sourced unified-agent dependency is unchanged. `make check` is green
(108 nested tests). The two historical *_REPORT.md files keep the old name as
dated records.

Co-Authored-By: Claude Fable 5 <noreply@anthropic.com>

* chore: extract benchmark harness to sibling xbow-benchmark repo

Move PentestGPT/benchmark/ out to ../xbow-benchmark (its own repo) to keep this
project clean. The harness was decoupled from the framework code (it scores
container output, never imports pentestgpt_agent/unified_agent), so only
operational ties remain and they now live in the sibling repo.

- Remove benchmark/ and the 4 harness unit tests (relocated + repointed there).
- Strip the docker-bench-*/bench-* targets and their config vars from the
  Makefile; keep the tool-image lifecycle (docker-build/login/run/...) and add a
  help pointer to `make -C ../xbow-benchmark help`.

The sibling repo mounts this checkout read-only (--source-root ../PentestGPT) and
runs the pentestgpt:latest image built here.

Co-Authored-By: Claude Fable 5 <noreply@anthropic.com>

* feat: harden autonomous framework and runtime integration

---------

Co-authored-by: Claude Opus 4.8 <noreply@anthropic.com>
2026-07-13 16:49:08 +08:00
Gelei Deng b9869307d0 Legacy multi llm base (#470)
* fix: 🐛 minor typo and build process

* feat: 🎸 [WIP] Pentest mode

* feat: 🎸 code abstraction

* feat: modernize legacy PentestGPT with native multi-LLM support (#469)

Rebuild the classic USENIX-2024 interactive PentestGPT (reasoning / generation /
parsing sessions + Pentesting Task Tree + REPL) as a standalone
`pentestgpt_legacy` package on a native per-provider LLM layer that supports the
latest 2026 models.

- llm/: BaseProvider + OpenAI-compatible / Anthropic / Gemini connectors, a
  web-verified model registry (OpenAI, Anthropic, Gemini, DeepSeek, xAI, Qwen,
  Moonshot, local Ollama), a factory, and an LLMClient bridging async providers
  to the core's synchronous send_new_message/send_message session API.
- CLI `pentestgpt-legacy`: --list-models and --smoke-test (live per-model
  round-trip matrix), plus --reasoning-model / --parsing-model / --base-url.
- Tests: 25 unit tests (mocked, no network). Live smoke test verified 22/22
  models with a configured key respond.

Co-authored-by: Claude Opus 4.8 <noreply@anthropic.com>

* fix(backend): address review on ClaudeCodeBackend subprocess handling

- _build_env: pop ANTHROPIC_API_KEY instead of setting it to "", so an empty
  value can't shadow the CLI's own auth fallback (e.g. subscription login).
- _kill_process: reap the force-killed process with os.waitpid(.., WNOHANG)
  instead of calling the proc.wait() coroutine without awaiting it (removes the
  "coroutine was never awaited" warning).
- query/_drain_stderr: drain subprocess stderr in a background task so its pipe
  buffer can't fill and deadlock the child.

Also reformats backend.py, fixing the failing Lint (ruff format) check.

Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>

* fix(docker-test): assert uv instead of Poetry in container health check

The project migrated from Poetry to uv (the Dockerfile installs uv to
/home/pentester/.local/bin, which is on PATH), so test_poetry_installed failed
with exit 127. Replace it with test_uv_installed checking `uv --version`.

Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>

---------

Co-authored-by: Claude Opus 4.8 <noreply@anthropic.com>
2026-06-07 15:25:45 +08:00
Gelei Deng d59da099d3 docs: ✏️ update documentation for benchmark (#378)
* docs: ✏️ update documentation for benchmark

* add USENIX benchmark
2025-12-29 01:18:57 +08:00
Yuekang Li 0c095b6053 feat: standalone benchmarking scripts for xbow-validation-benchmark (#374)
* feat: standalone benchmarking scripts for xbow-validation-benchmark

* feat: 🎸 update benchmark

---------

Co-authored-by: gelei <gelei@quantstamp.com>
2025-12-25 01:13:16 +08:00
Yi Liu 0f65f0f415 Add local llm docs in README.md (#332)
* docs: ✏️ add step by step docs for setup local llms

* docs: ✏️ refactor README.md
2025-12-16 20:52:59 +07:00
Gelei Deng aeb3eb6b88 feat: 🎸 local model support (#330) 2025-12-15 14:34:57 +08:00
Yi Liu 96316c70db docs: ✏️ add step by step docs for setup local llms (#331) 2025-12-15 14:34:37 +08:00
Gelei Deng e976ffba5c Update demo (#329)
* feat: 🎸 version 1.0 agentic workflow

Major rewrite of PentestGPT to use an agentic pipeline architecture:
Core Changes: - New event-driven architecture with EventBus for
TUI-agent decoupling - Implemented AgentController with 5-state
lifecycle (IDLE->RUNNING->PAUSED->COMPLETED->ERROR) - Added AgentBackend
interface with ClaudeCodeBackend implementation - Session management
with file-based persistence for resumable pentests - Langfuse
integration for observability and tracing Interface: - New Textual-based
TUI with real-time activity feed - Keyboard shortcuts: F1 help, Ctrl+P
pause, Ctrl+Q quit - Enhanced CLI with --target, --instruction,
--non-interactive, --debug flags Project Structure: - Moved legacy
multi-LLM version (v0.15) to legacy/ directory - New pentestgpt/core/
for agent, controller, events, session modules - New
pentestgpt/interface/ for TUI and CLI components - New
pentestgpt/benchmark/ for xbow benchmark integration - Comprehensive
test suite in tests/ with unit and integration tests DevOps: - Docker
support with Ubuntu 24.04 container - GitHub Actions CI/CD pipeline -
Makefile with dev commands (test, lint, format, typecheck) - Added
xbow-validation-benchmarks as submodule

* style: format code with Black

This commit fixes the style issues introduced in abe3be0 according to the output
from Black.

Details: https://github.com/GreyDGL/PentestGPT/pull/325

* fix: 🐛 fix test pipeline

* feat: 🎸 update format

* feat: 🎸 update

* docs: ✏️ readme and demo video udpate

---------

Co-authored-by: deepsource-autofix[bot] <62050782+deepsource-autofix[bot]@users.noreply.github.com>
2025-12-14 14:17:55 +08:00
Gelei Deng e238d701f2 feat: 🎸 version 1.0 agentic workflow (#325)
* feat: 🎸 version 1.0 agentic workflow

Major rewrite of PentestGPT to use an agentic pipeline architecture:
Core Changes: - New event-driven architecture with EventBus for
TUI-agent decoupling - Implemented AgentController with 5-state
lifecycle (IDLE->RUNNING->PAUSED->COMPLETED->ERROR) - Added AgentBackend
interface with ClaudeCodeBackend implementation - Session management
with file-based persistence for resumable pentests - Langfuse
integration for observability and tracing Interface: - New Textual-based
TUI with real-time activity feed - Keyboard shortcuts: F1 help, Ctrl+P
pause, Ctrl+Q quit - Enhanced CLI with --target, --instruction,
--non-interactive, --debug flags Project Structure: - Moved legacy
multi-LLM version (v0.15) to legacy/ directory - New pentestgpt/core/
for agent, controller, events, session modules - New
pentestgpt/interface/ for TUI and CLI components - New
pentestgpt/benchmark/ for xbow benchmark integration - Comprehensive
test suite in tests/ with unit and integration tests DevOps: - Docker
support with Ubuntu 24.04 container - GitHub Actions CI/CD pipeline -
Makefile with dev commands (test, lint, format, typecheck) - Added
xbow-validation-benchmarks as submodule

* style: format code with Black

This commit fixes the style issues introduced in abe3be0 according to the output
from Black.

Details: https://github.com/GreyDGL/PentestGPT/pull/325

* fix: 🐛 fix test pipeline

* feat: 🎸 update format

* feat: 🎸 update

---------

Co-authored-by: deepsource-autofix[bot] <62050782+deepsource-autofix[bot]@users.noreply.github.com>
2025-12-13 01:57:24 +08:00
Gelei Deng 0f8e73a68e 286 is there any chance of using ollama with local llm model (#294)
* feat: 🎸 add ollama

* docs: ✏️ add documentation for ollama
2025-07-29 15:39:08 +08:00
Gelei Deng c89f452789 Benchmark (#288)
* feat: 🎸 new benchmark design

* feat: 🎸 evaluator update

* docs: ✏️ update README

* docs: ✏️ readme update
2025-07-07 22:43:50 +08:00
Víctor Mayoral Vilches bb6741c12e Add pointers to CAI and notes on scams and copycats (#284)
* Add a note about CAI

Signed-off-by: Víctor Mayoral Vilches <v.mayoralv@gmail.com>

* Add a note about copycats and scams

Signed-off-by: Víctor Mayoral Vilches <v.mayoralv@gmail.com>

* Improve aesthetics

Signed-off-by: Víctor Mayoral Vilches <v.mayoralv@gmail.com>

* Add example

Signed-off-by: Víctor Mayoral Vilches <v.mayoralv@gmail.com>

* Add an update

Signed-off-by: Víctor Mayoral Vilches <v.mayoralv@gmail.com>

* Update general udpate note

Signed-off-by: Víctor Mayoral Vilches <v.mayoralv@gmail.com>

---------

Signed-off-by: Víctor Mayoral Vilches <v.mayoralv@gmail.com>
2025-06-17 00:10:38 +08:00
Gelei Deng 6a767aae02 Model support (#274)
* feat: 🎸 add new models

* minor readme update

* style: format code with Black

This commit fixes the style issues introduced in dafa18b according to the output
from Black.

Details: https://github.com/GreyDGL/PentestGPT/pull/274

---------

Co-authored-by: deepsource-autofix[bot] <62050782+deepsource-autofix[bot]@users.noreply.github.com>
2025-05-01 18:02:42 +08:00
Gelei Deng 1038c90b91 Version update (#247)
* minor changes on README and test connection.

* feat: 🎸 format and install dependency updates
2024-11-01 15:19:54 +08:00
Gelei Deng edf64255f4 Update README.md 2024-10-25 02:04:17 +08:00
Gelei Deng f072f90293 Vision (#230)
* fix: 🐛 fix OPENAI key setting issue and update readme

* feat: 🎸 update gpt4o

* style: format code with Black

This commit fixes the style issues introduced in 99581a8 according to the output
from Black.

Details: https://github.com/GreyDGL/PentestGPT/pull/229

* fix: 🐛 fix OPENAI_KEY typo

* style: format code with Black

This commit fixes the style issues introduced in 8f9091c according to the output
from Black.

Details: https://github.com/GreyDGL/PentestGPT/pull/230

---------

Co-authored-by: deepsource-autofix[bot] <62050782+deepsource-autofix[bot]@users.noreply.github.com>
2024-05-14 17:58:18 +08:00
Gelei Deng df4d330ce3 fix: 🐛 fix OPENAI key setting issue and update readme (#228) 2024-05-14 17:42:02 +08:00
Gelei Deng 43f6e803e0 Gpt4all Dev (#217)
* fix: 🐛 fix default models used

 Closes: #204

* feat: 🎸 support local LLMs with GPT4ALL

* fix lint issue

* style: format code with Black

This commit fixes the style issues introduced in 5eee6a0 according to the output
from Black.

Details: https://github.com/GreyDGL/PentestGPT/pull/217

---------

Co-authored-by: deepsource-autofix[bot] <62050782+deepsource-autofix[bot]@users.noreply.github.com>
2024-04-12 17:10:22 +08:00
Wang Yile ae84989c36 Add support for customizing API Base URL using environment variables (#207)
* Update chatgpt_config.py

Add base api environment variables

* Update README.md

Add guidance for API base changes

* Fix the problem that test-connection cannot run

* Update test_connection.py

Class variable read modification
2024-03-31 16:12:13 +08:00
Gelei Deng 3295953495 Poetry Upgrade (#201)
* chore: 🤖 update to poetry

* chore: 🤖 major poetry update and lint fix

 Closes: #200
2024-03-25 01:00:17 +08:00
Gelei bcf52fc8ac update readme and fix for key binding 2023-11-29 04:12:46 -05:00
Grey_D 1d2b6e9bf7 docs: ✏️ update license 2023-11-19 11:20:17 +08:00
Grey_D e2f46984c1 docs: ✏️ update readme 2023-11-17 20:31:42 +08:00
Grey_D 760a6a7531 feat: 🎸 Add link for GPTs 2023-11-17 00:40:00 +08:00
Grey_D cf53f54bec minor doc update 2023-11-07 11:22:05 +01:00
Grey_D 63efa9df62 minor doc update 2023-11-07 11:21:32 +01:00
Grey_D 7a88ac001f docs: ✏️ README update 2023-11-07 11:19:50 +01:00
Grey_D 6a49a603f2 feat: 🎸 add gpt-4-turbo 2023-11-07 01:54:51 +01:00
Grey_D 60f12cc04e feat: 🎸 support for vectorDB 2023-10-30 20:25:05 +08:00
Robb 6885ff0559 Typo fix 2023-10-17 15:01:57 +02:00
Grey_D 2bf88bac9c update README 2023-10-17 00:43:25 +08:00
Robb 6bdd69eea1 Add description for selection options in README 2023-10-16 16:55:51 +02:00
Robb 30ddc65070 Info Fix
With `--reasoning_model=gpt-3.5-turbo` the tool does not work, the tool works only with `--reasoning_model=gpt-3.5-turbo-16k`.

Added `--help`argument
2023-10-09 10:16:45 +02:00
Grey_D ab6c840be6 feat: 🎸 add aws titan support 2023-08-24 21:47:43 +08:00
Anth0rx ecd7d811fd Fix typo 2023-08-24 14:19:32 +02:00
Grey_D c271bc54a9 feat: 🎸 Local LLM 2023-07-20 23:58:43 +08:00
jiayuqi7813 62c7d279a5 User-defined log directory 2023-06-28 13:19:38 +00:00
Joe 6ff7e106c3 Update README.md 2023-06-22 20:16:11 +01:00
Joe 3bc4180bc2 Update README.md 2023-06-22 20:13:54 +01:00
Grey_D 22484f94ec docs: ✏️ add installation video 2023-06-13 10:13:58 +08:00
Grey_D 60da315867 docs: ✏️ Update readme and test connection logic 2023-05-30 17:03:16 +08:00
Grey_D bc430c40c3 refactor: 💡 Update README and config for new version 2023-05-30 16:57:36 +08:00
Víctor Mayoral Vilches d8e89f64a3 Make PentestGPT a Python module and easily installable
Signed-off-by: Víctor Mayoral Vilches <v.mayoralv@gmail.com>
2023-05-29 09:45:56 +02:00
Grey_D f64d8f9f90 docs: ✏️ minor doc update on Discord icon 2023-05-26 15:15:32 +08:00
Grey_D ae4c94c9f6 fix: 🐛 Cookie bypass with curl 2023-05-20 12:34:28 +08:00
Grey_D ee978561bc docs: ✏️ Update README and Demo Video 2023-05-13 12:14:52 +08:00
Grey_D 343af68158 update discord link 2023-05-10 14:21:39 +08:00
Grey_D 130a04e127 docs: ✏️ Use new readme template 2023-05-10 14:19:21 +08:00
Grey_D 52d7a2b9c3 docs: ✏️ Update installation guide 2023-05-07 13:07:22 +08:00