greydgl--pentestgpt
gelei
abe3be01b4
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
144 行
3.5 KiB
TOML
144 行
3.5 KiB
TOML
[project]
|
|
name = "pentestgpt"
|
|
version = "1.0.0"
|
|
description = "AI-powered autonomous penetration testing agent with beautiful TUI - Published at USENIX Security 2024"
|
|
authors = [{ name = "Gelei Deng", email = "GELEI.DENG@ntu.edu.sg" }]
|
|
readme = "README.md"
|
|
license = { text = "MIT" }
|
|
requires-python = ">=3.12,<4.0"
|
|
keywords = [
|
|
"penetration-testing",
|
|
"security",
|
|
"ai",
|
|
"claude",
|
|
"pentest",
|
|
"cybersecurity",
|
|
"ctf",
|
|
"hacking",
|
|
]
|
|
classifiers = [
|
|
"Development Status :: 4 - Beta",
|
|
"Environment :: Console",
|
|
"Intended Audience :: Developers",
|
|
"Intended Audience :: Information Technology",
|
|
"Topic :: Security",
|
|
"License :: OSI Approved :: MIT License",
|
|
"Programming Language :: Python :: 3.12",
|
|
]
|
|
dependencies = [
|
|
"textual>=0.47.0",
|
|
"rich>=13.7.0",
|
|
"pydantic>=2.5.0",
|
|
"pydantic-settings>=2.1.0",
|
|
"claude-agent-sdk>=0.1.0",
|
|
"python-dotenv>=1.0.0",
|
|
"anthropic>=0.75.0",
|
|
"langfuse>=2.0.0",
|
|
]
|
|
|
|
[project.scripts]
|
|
pentestgpt = "pentestgpt.interface.main:main"
|
|
pentestgpt-benchmark = "pentestgpt.benchmark.cli:main"
|
|
|
|
[project.urls]
|
|
Homepage = "https://github.com/GreyDGL/PentestGPT"
|
|
Repository = "https://github.com/GreyDGL/PentestGPT"
|
|
|
|
[build-system]
|
|
requires = ["hatchling"]
|
|
build-backend = "hatchling.build"
|
|
|
|
[tool.hatch.build.targets.wheel]
|
|
packages = ["pentestgpt"]
|
|
|
|
# ============================================================================
|
|
# uv Configuration
|
|
# ============================================================================
|
|
|
|
[tool.uv]
|
|
dev-dependencies = [
|
|
"ruff>=0.1.0",
|
|
"mypy>=1.7.0",
|
|
"pytest>=7.4.0",
|
|
"pytest-asyncio>=0.21.0",
|
|
"pytest-cov>=4.1.0",
|
|
]
|
|
|
|
# ============================================================================
|
|
# Ruff Configuration (Linter & Formatter)
|
|
# ============================================================================
|
|
|
|
[tool.ruff]
|
|
target-version = "py312"
|
|
line-length = 100
|
|
extend-exclude = [
|
|
".git",
|
|
".mypy_cache",
|
|
".pytest_cache",
|
|
".ruff_cache",
|
|
"__pycache__",
|
|
"build",
|
|
"dist",
|
|
"legacy",
|
|
"PentestGPTClaude",
|
|
]
|
|
|
|
[tool.ruff.lint]
|
|
select = [
|
|
"E", # pycodestyle errors
|
|
"W", # pycodestyle warnings
|
|
"F", # Pyflakes
|
|
"I", # isort
|
|
"UP", # pyupgrade
|
|
"B", # flake8-bugbear
|
|
"C4", # flake8-comprehensions
|
|
"SIM", # flake8-simplify
|
|
"RUF", # Ruff-specific rules
|
|
]
|
|
|
|
ignore = [
|
|
"E501", # Line too long (handled by formatter)
|
|
]
|
|
|
|
[tool.ruff.format]
|
|
quote-style = "double"
|
|
indent-style = "space"
|
|
|
|
# ============================================================================
|
|
# MyPy Configuration (Type Checking)
|
|
# ============================================================================
|
|
|
|
[tool.mypy]
|
|
python_version = "3.12"
|
|
strict = true
|
|
warn_return_any = true
|
|
warn_unused_configs = true
|
|
disallow_untyped_defs = true
|
|
|
|
[[tool.mypy.overrides]]
|
|
module = [
|
|
"claude_agent_sdk.*",
|
|
"textual.*",
|
|
]
|
|
ignore_missing_imports = true
|
|
|
|
# ============================================================================
|
|
# Pytest Configuration
|
|
# ============================================================================
|
|
|
|
[tool.pytest.ini_options]
|
|
minversion = "7.0"
|
|
addopts = [
|
|
"--strict-markers",
|
|
"--strict-config",
|
|
]
|
|
testpaths = ["tests"]
|
|
pythonpath = ["."]
|
|
asyncio_mode = "auto"
|
|
markers = [
|
|
"unit: Unit tests (fast, no external dependencies)",
|
|
"integration: Integration tests (may use mocks)",
|
|
"docker: Docker tests (requires Docker daemon)",
|
|
"slow: Slow tests (skip with -m 'not slow')",
|
|
]
|