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
123 行
3.5 KiB
Bash
可执行文件
123 行
3.5 KiB
Bash
可执行文件
#!/usr/bin/env bash
|
|
# PentestGPT Authentication Configuration
|
|
# Interactive setup for Claude Code authentication
|
|
|
|
set -e
|
|
|
|
# Colors
|
|
RED='\033[0;31m'
|
|
GREEN='\033[0;32m'
|
|
BLUE='\033[0;34m'
|
|
PURPLE='\033[0;35m'
|
|
YELLOW='\033[0;33m'
|
|
NC='\033[0m'
|
|
|
|
# Get the directory where this script is located, then go up to project root
|
|
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
|
PROJECT_DIR="$(dirname "$SCRIPT_DIR")"
|
|
ENV_FILE="${PROJECT_DIR}/.env.auth"
|
|
|
|
echo -e "${PURPLE}"
|
|
cat << "EOF"
|
|
____ __ __ __________ ______
|
|
/ __ \___ ____ / /____ _____/ /_/ ____/ __ /_ __/
|
|
/ /_/ / _ \/ __ \/ __/ _ \/ ___/ __/ / __/ /_/ // /
|
|
/ ____/ __/ / / / /_/ __(__ ) /_/ /_/ / ____// /
|
|
/_/ \___/_/ /_/\__/\___/____/\__/\____/_/ /_/
|
|
|
|
Authentication Configuration
|
|
EOF
|
|
echo -e "${NC}"
|
|
|
|
echo -e "${BLUE}Select authentication method:${NC}\n"
|
|
echo -e " ${GREEN}[1]${NC} Claude Code Subscription ${YELLOW}(Recommended)${NC}"
|
|
echo -e " Use your Claude Code subscription via 'claude login'"
|
|
echo ""
|
|
echo -e " ${GREEN}[2]${NC} OpenRouter API Key"
|
|
echo -e " Route requests through OpenRouter to use GPT-5.1, Gemini, etc."
|
|
echo ""
|
|
echo -e " ${GREEN}[3]${NC} Anthropic API Key"
|
|
echo -e " Use Anthropic's Claude directly with your API key"
|
|
echo ""
|
|
|
|
read -p "Enter your choice [1-3] (default: 1): " choice
|
|
choice="${choice:-1}"
|
|
|
|
case $choice in
|
|
1)
|
|
# Save auth mode for Claude Code subscription (manual login)
|
|
cat > "$ENV_FILE" << EOF
|
|
# PentestGPT Authentication Configuration
|
|
# Generated by make config
|
|
PENTESTGPT_AUTH_MODE=manual
|
|
EOF
|
|
|
|
echo -e "${GREEN}Claude Code subscription mode selected.${NC}"
|
|
echo -e "${BLUE}After running 'make connect', execute 'claude login' inside the container.${NC}"
|
|
;;
|
|
|
|
2)
|
|
echo ""
|
|
echo -e "${BLUE}OpenRouter Configuration${NC}"
|
|
echo -e "${YELLOW}Get your API key from: https://openrouter.ai/keys${NC}"
|
|
echo ""
|
|
read -sp "Enter your OpenRouter API key: " openrouter_key
|
|
echo ""
|
|
|
|
if [ -z "$openrouter_key" ]; then
|
|
echo -e "${RED}Error: API key cannot be empty${NC}"
|
|
exit 1
|
|
fi
|
|
|
|
# Save auth mode
|
|
cat > "$ENV_FILE" << EOF
|
|
# PentestGPT Authentication Configuration
|
|
# Generated by make config
|
|
PENTESTGPT_AUTH_MODE=openrouter
|
|
OPENROUTER_API_KEY=${openrouter_key}
|
|
EOF
|
|
|
|
# Set restrictive permissions on the file
|
|
chmod 600 "$ENV_FILE"
|
|
|
|
echo -e "${GREEN}OpenRouter configuration saved!${NC}"
|
|
echo -e "${BLUE}CCR will be configured automatically when you run 'make connect'${NC}"
|
|
;;
|
|
|
|
3)
|
|
echo ""
|
|
echo -e "${BLUE}Anthropic API Key Configuration${NC}"
|
|
echo -e "${YELLOW}Get your API key from: https://console.anthropic.com/settings/keys${NC}"
|
|
echo ""
|
|
read -sp "Enter your Anthropic API key: " anthropic_key
|
|
echo ""
|
|
|
|
if [ -z "$anthropic_key" ]; then
|
|
echo -e "${RED}Error: API key cannot be empty${NC}"
|
|
exit 1
|
|
fi
|
|
|
|
# Save auth mode
|
|
cat > "$ENV_FILE" << EOF
|
|
# PentestGPT Authentication Configuration
|
|
# Generated by make config
|
|
PENTESTGPT_AUTH_MODE=anthropic
|
|
ANTHROPIC_API_KEY=${anthropic_key}
|
|
EOF
|
|
|
|
# Set restrictive permissions on the file
|
|
chmod 600 "$ENV_FILE"
|
|
|
|
echo -e "${GREEN}Anthropic API key saved!${NC}"
|
|
;;
|
|
|
|
*)
|
|
echo -e "${RED}Invalid choice. Exiting.${NC}"
|
|
exit 1
|
|
;;
|
|
esac
|
|
|
|
echo ""
|
|
echo -e "${GREEN}Configuration complete!${NC}"
|
|
echo -e "Run ${PURPLE}make connect${NC} to start PentestGPT."
|