AutoPentester: 자동화된 침투 테스트를 위한 LLM 에이전트 기반 프레임워크

참고:

• 본 작업을 학술 목적으로 사용하는 경우, 저희 논문을 인용해 주시기 바랍니다.
• 잘못된 / 부적절한 / 오래된 내용을 발견하시면, 이슈 또는 PR을 열어 주시기 바랍니다.

업데이트

• AutoPentester는 이제 claude를 지원합니다.
• 본 작업은 IEEE TrustCom 2025에 채택되었습니다.

설치

1. 가상 환경을 생성합니다. (python3 -m venv myenv, source myenv/bin/activate)
2. 프로젝트를 복제하고 요구 사항을 설치합니다.
   • git clone <repo_url>
   • cd AutoPentester
   • Python pip3 3.12.3으로 가상 환경을 생성합니다. 그런 다음 그 안에 requirements.txt를 설치합니다.
   • pip3 install -r requirements.txt
   • pip3 install -e .
3. OpenAI API를 사용하려면
   • OpenAI 계정에 결제 수단이 연결되어 있는지 확인하십시오.
   • export OPENAI_API_KEY="<your key here>"로 API 키를 내보냅니다.
   • 선택 사항: export OPENAI_BASE_URL="https://your-host/v1"로 API 엔드포인트를 재정의합니다.
4. Anthropic Claude 모델을 사용하려면

• export ANTHROPIC_API_KEY="<your key here>"로 API 키를 내보냅니다.

5. Google Gemini 모델을 사용하려면
   • export GEMINI_API_KEY="<your key here>" 또는 export GOOGLE_API_KEY="<your key here>"로 API 키를 내보냅니다.
   • Gemini는 pentestgpt/config/agent_config.py에서 에이전트별로 선택됩니다.
6. pentestgpt/config/agent_config.py에서 에이전트 모델을 구성합니다.

• 각 에이전트 이름은 공급자와 모델에 매핑됩니다.
• 지원되는 공급자는 openai, anthropic, gemini, azure, gpt4all, titan, rag입니다.
• 선호도에 따라 에이전트별로 OpenAI와 Claude를 혼합하도록 파일을 업데이트합니다.

7. 프레임워크를 실행하려면 autopentester --loggin을 입력합니다.
8. 구성된 공급자의 키를 묻고, 그다음 IP 주소를 묻습니다.
9. 이전 세션에서 계속하시겠습니까? (y/n) -> n을 누릅니다.
10. 침투 테스트 작업을 제공합니다. "I want to test the machine with the IP (targe_IP)"와 같은 프롬프트를 사용할 수 있습니다.
11. 첫 실행 시에만 시작 부분에서 RAG 모듈의 벡터베이스를 구축하는 데 10분이 소요됩니다. 프로세스가 시작될 때까지 기다려 주십시오.

데모

시연 영상은 다음 링크에서 확인하실 수 있습니다.

로그 파일

처리된 로그 파일은 processed_log_files 디렉터리에 있습니다. 정량적 결과는 이 로그 파일들을 기반으로 계산되었습니다.

설문 분석

설문 분석은 Survey_analysis 디렉터리에 있습니다. 그래프를 그리려면 analysis.py를 실행하십시오.

인용

본 작업을 학술 목적으로 사용하는 경우, 저희 논문을 인용해 주시기 바랍니다.

@inproceedings{ginige2025autopentester,
  title={Autopentester: An llm agent-based framework for automated pentesting},
  author={Ginige, Yasod and Niroshan, Akila and Jain, Sajal and Seneviratne, Suranga},
  booktitle={2025 IEEE 24th International Conference on Trust, Security and Privacy in Computing and Communications (TrustCom)},
  pages={163--174},
  year={2025},
  organization={IEEE}
}