PentestCode

터미널에서 사용하는 AI 침투 테스트 에이전트.
멀티 에이전트 아키텍처 • 공격 상태 추적 • 20개 이상의 LLM 제공자

PentestCode는 터미널용 자율 침투 테스트 에이전트입니다. 대상을 지정하면 도구를 실행하고, 출력을 읽고, 네트워크에 대한 그림을 업데이트하고, 다음에 무엇을 할지 결정합니다 — 마치 운영자가 하는 것처럼. 코드 편집에 초점을 맞춘 OpenCode(MIT)의 하드 포크로, 공격 보안(offensive security)용으로 재구축되었습니다.

베타 — 실제 침투 테스트와 CTF에서 잘 작동하지만, 거친 부분이 있을 수 있습니다. 문제가 발생하면 이슈를 등록해 주세요. 그래야 이 도구가 더 나아집니다.

기능

지시 하나를 넣으면, 전체 공격 체인이 나옵니다:

you: "pentest 10.10.10.5, goal is domain admin"

사람들이 게을러지기 쉬운 부분에서 이 도구는 체계적입니다. 모든 호스트의 모든 서비스에 모든 자격 증명을 스프레이하고, 확인해야 할 것을 빠뜨리지 않습니다. 학습한 모든 것은 구조화된 상태로 저장되며, 실행 중에도 /status, /vulns, /creds로 조회할 수 있습니다.

설치

npm install -g pentestcode-ai

# via curl
curl -fsSL https://raw.githubusercontent.com/s0ld13rr/pentestcode/main/install.sh | bash

단일 자체 포함 바이너리입니다 — Bun, Node 또는 런타임을 설치할 필요가 없습니다. Linux 및 macOS, x64 및 arm64를 지원합니다.

PENTESTCODE_VERSION=0.1.7 curl -fsSL https://raw.githubusercontent.com/s0ld13rr/pentestcode/main/install.sh | bash

PENTESTCODE_INSTALL=/usr/local/bin curl -fsSL https://raw.githubusercontent.com/s0ld13rr/pentestcode/main/install.sh | bash

bun install
bun run build --single --skip-embed-web-ui
# binary at packages/opencode/dist/pentestcode-<os>-<arch>/bin/pentestcode

빠른 시작

pentestcode auth login          # connect your LLM provider
pentestcode                     # interactive session
pentestcode --prompt "scan 10.10.10.0/24 and enumerate all services"   # one-shot

ai-sdk를 통해 20개 이상의 제공자를 지원합니다 — Anthropic, OpenAI, Google, Azure, AWS Bedrock, Ollama 등.

작동 방식

PentestCode를 채팅 창에 붙여넣은 침투 테스트 프롬프트와 차별화하는 두 가지는 에이전트 팀과 그들이 공유하는 메모리입니다.

독백이 아닌 팀

이 설계는 HPTSA 연구의 전략가-조정자(strategist-coordinator) 모델을 따릅니다 — 단일 에이전트 대비 4.3배의 개선 효과를 냅니다:

                    ┌─────────────┐
                    │   pentest   │  strategist / coordinator
                    │   (lead)    │  plans, dispatches, tracks state
                    └──────┬──────┘
           ┌───────┬───────┼───────┬───────┐
           ▼       ▼       ▼       ▼       ▼
        ┌──────┐┌──────┐┌──────┐┌──────┐┌──────┐
        │recon ││scan- ││explo-││iden- ││post- │
        │      ││ner   ││iter  ││tity  ││explo │  ... + 7 more
        └──────┘└──────┘└──────┘└──────┘└──────┘

리드 에이전트(pentest)는 공격 작업을 여러 태스크로 나누고 전문 서브에이전트를 병렬로 파견합니다 — 각각 고유한 시스템 프롬프트, 도구 권한, 도메인 지식을 가집니다. 총 13개의 에이전트: recon(정찰), scanner(스캐너), enumerator(열거자), exploiter(익스플로이터), identity(AD/Kerberos), infrastructure(SNMP/IPMI/데이터베이스), webapp(OWASP Top 10), post-exploit(사후 익스플로잇), exploit-dev(익스플로잇 개발), critic(오탐지 검사기), reporter(리포터), 그리고 컨텍스트 압축 및 세션 관리를 위한 숨은 에이전트.

모두가 공유하는 메모리

스캐너가 포트를 발견하면 열거자가 즉시 확인합니다 — 모든 에이전트가 하나의 구조화된 **공격 상태(engagement state)**를 읽고 쓰기 때문입니다:

• 호스트 및 서비스 — IP, 호스트명, OS, 포트, 서비스 버전, 배너
• 취약점 — 심각도, 상태(의심됨 / 확인됨 / 악용됨), 증거 체인, 신뢰도 점수
• 자격 증명 — 사용자 이름, 해시/비밀번호, 유형, 도메인, 접근 가능한 대상
• 접근 권한 — 어떤 호스트에서 누가 셸/RDP/DB를 보유하고 있는지, 어떤 권한 수준인지
• 관계 — 엔티티 그래프 (EXPLOITED_VIA, CREDENTIAL_FROM, ADMIN_OF, PIVOT_TO, …)
• AD 도메인 모델 — 도메인 컨트롤러, 트러스트, 관리자, 암호 정책, GPO
• 네트워크 세그먼트 — VLAN, 도달 가능한 네트워크, 피벗 호스트
• 공격 경로 — 관계 그래프를 통한 비용 기반 다익스트라(Dijkstra) + Yen의 K-최단 경로

상태는 세션을 넘어 유지됩니다. 터미널을 닫았다가 다음 날 돌아오면 에이전트는 중단했던 지점부터 다시 시작합니다. 그와 함께 사람이 읽을 수 있는 findings.md가 모든 취약점, 자격 증명, 획득한 접근 권한을 타임스탬프와 함께 기록합니다 — tail -f로 공격이 전개되는 과정을 지켜볼 수 있습니다.

도구

bash 외에도 18개의 내장 침투 테스트 도구가 있습니다. 파서 도구는 필수입니다. nmap 실행 후 에이전트는 XML을 수동으로 grep하는 대신 nmap_parse에 출력을 파이프해야 하며, 그래야 모든 발견 사항이 공격 상태에 반영됩니다.

스킬

19개의 선별된 지식 팩. 관련이 있을 때만 컨텍스트를 소모하도록 요청 시 로드됩니다:

• 단계별 체크리스트 (6) — 각 침투 테스트 단계에서 수행할 작업
• 서비스 지식 (9) — SMB, SSH, FTP, DNS, 데이터베이스, 웹 서버, 메일, Docker/K8s, CI/CD
• 플레이북 (4) — 인프라, Active Directory, 웹 애플리케이션, 클라우드

스킬은 일반 마크다운입니다. skills 디렉터리에 SKILL.md를 넣기만 하면 나만의 스킬을 추가할 수 있습니다 — 코드 변경이 필요 없습니다.

명령어 및 모드

슬래시 명령어로 라이브 세션을 제어하세요:

그리고 에이전트에 부여할 자율성의 정도를 설정할 수 있습니다:

• auto — 침투 테스트 단계를 자율적으로 진행하며 필요에 따라 서브에이전트 생성
• free — 단계 구조 없이 요청에 직접 응답 (범위 검사 우회)
• guided — 단계별로 진행하며 각 작업을 제안하고 승인을 기다림

모드와 일시 중지 동작을 결합할 수 있습니다 — auto + pause always는 모든 발견 사항에서 멈춰 검토하게 하는 자율 실행을 제공합니다.

사용 사례

공격 보안 전반을 아우르는 하나의 툴킷:

• 침투 테스트 — 정찰부터 보고까지의 전체 방법론
• CTF 대회 — 플래그 추적, 목표 관리, 다중 대상 조정
• 버그 바운티 — 웹 애플리케이션 테스트, API 보안, 정찰 자동화
• 취약점 연구 — 체계적인 열거 및 검증
• 인프라 보안 — 네트워크 서비스 감사, 기본 자격 증명 확인

설정

설정 파일은 .pentestcode/pentestcode.jsonc에 있습니다:

{
  "provider": {
    "anthropic": {
      "model": "claude-sonnet-4-20250514"
    }
  }
}

제공자: Anthropic, OpenAI, Google, Azure, AWS Bedrock, Ollama, Together, Groq, Fireworks, DeepSeek, Mistral 및 ai-sdk를 통한 더 많은 제공자.

기여

실제 사용에서 나온 버그 보고가 가장 가치 있는 기여입니다. CTF 박스, HTB 머신, 또는 승인된 침투 테스트에서 PentestCode를 실행하고, 문제가 발생하면 — 무한 루프에 빠지거나, 명백한 경로를 놓치거나, 도구 출력에서 멈추거나, 토큰을 낭비하는 경우 — 다음을 포함해 이슈를 등록해 주세요:

1. 테스트 대상 (민감한 세부 정보가 아닌 대상 유형)
2. 무엇이 잘못되었는지
3. findings.md 및/또는 관련 세션 출력

기능 요청과 PR도 환영합니다. 코드베이스는 Effect 라이브러리 기반의 TypeScript입니다 — 아키텍처는 CLAUDE.md를 참조하세요.

라이선스

MIT — LICENSE 참조.

OpenCode의 하드 포크 • 공격 보안용으로 제작 • 자체 호스팅 및 오픈 소스