cd RAG_runner
pip install -r requirements.txt

python3 build_database.py

python3 -m server.http_server

# 상태 확인
curl http://localhost:5004/health

# SQL 인젝션 테스트 방법론 검색
curl -X POST http://localhost:5004/search \
  -H "Content-Type: application/json" \
  -d '{"query": "SQL injection testing methodology"}'

# 특정 WSTG 테스트 케이스 조회
curl http://localhost:5004/wstg/WSTG-INPV-05

{
  "query": "SQL injection testing",
  "n_results": 5,
  "category": "input_validation",
  "wstg_id": "WSTG-INPV-05"
}

{
  "mcpServers": {
    "owasp-wstg-rag": {
      "command": "python3",
      "args": ["/path/to/OWASP_WSTG_Rag/RAG_runner/server/mcp_client.py"],
      "env": {
        "WSTG_RAG_URL": "http://localhost:5004"
      }
    }
  }
}

# SQL 인젝션 테스트 방법론 검색
search_wstg("SQL injection testing methodology")

# 특정 테스트 케이스 조회
get_wstg_test_case("WSTG-INPV-05")

# 카테고리 내에서 검색
search_wstg("authentication bypass", category_filter="authentication")

# IDOR에 대한 테스트 목표 검색
search_test_objectives("IDOR insecure direct object reference")

OWASP_WSTG_Rag/
├── README.md
├── CLAUDE.md                    # Claude Code 프로젝트 가이드
├── raw_data/                    # OWASP WSTG HTML 소스 파일
│   ├── 01-Information_Gathering/
│   ├── 02-Configuration_and_Deployment_Management_Testing/
│   ├── 03-Identity_Management_Testing/
│   ├── 04-Authentication_Testing/
│   ├── 05-Authorization_Testing/
│   ├── 06-Session_Management_Testing/
│   ├── 07-Input_Validation_Testing/
│   ├── 08-Testing_for_Error_Handling/
│   ├── 09-Testing_for_Weak_Cryptography/
│   ├── 10-Business_Logic_Testing/
│   ├── 11-Client-side_Testing/
│   └── 12-API_Testing/
└── RAG_runner/
    ├── build_database.py        # 주요 구축 파이프라인
    ├── requirements.txt
    ├── parsers/
    │   └── wstg_parser.py       # WSTG용 HTML 파서
    ├── chunking/
    │   └── chunker.py           # 의미 기반 청크 분할
    ├── server/
    │   ├── vector_store.py      # ChromaDB 래퍼
    │   ├── http_server.py       # REST API 서버
    │   └── mcp_client.py        # Claude Code용 MCP 도구
    └── data/
        ├── processed/           # 중간 JSON 파일
        └── chroma_db/           # 벡터 데이터베이스

┌─────────────────────────────────────────────────────────────────┐
│                    OWASP WSTG HTML 파일                         │
│                      (raw_data/*.html)                          │
└────────────────────────────┬────────────────────────────────────┘
                             │
                             ▼
┌─────────────────────────────────────────────────────────────────┐
│                     wstg_parser.py                              │
│              HTML 파싱 → 구조화된 JSON                          │
└────────────────────────────┬────────────────────────────────────┘
                             │
                             ▼
┌─────────────────────────────────────────────────────────────────┐
│                       chunker.py                                │
│              RAG를 위한 의미 기반 청크 생성                     │
└────────────────────────────┬────────────────────────────────────┘
                             │
                             ▼
┌─────────────────────────────────────────────────────────────────┐
│                   ChromaDB 벡터 저장소                          │
│                 (data/chroma_db/)                               │
└────────────────────────────┬────────────────────────────────────┘
                             │
              ┌──────────────┴──────────────┐
              ▼                             ▼
┌──────────────────────────┐   ┌──────────────────────────┐
│    http_server.py        │   │    mcp_client.py         │
│    REST API :5004        │   │    Claude Code용 MCP     │
│                          │   │                          │
│  GET  /health            │   │  search_wstg()           │
│  GET  /info              │   │  get_wstg_test_case()    │
│  GET  /wstg/{id}         │   │  search_test_methodology │
│  POST /search            │   │  list_wstg_categories()  │
└──────────────────────────┘   └──────────────────────────┘

사용자: "SQL 인젝션 테스트는 어떻게 하나요?"

Claude: [WSTG RAG 질의]
→ WSTG-INPV-05 방법론 반환:
  - 테스트 목표
  - 단계별 테스트 절차
  - 예시 페이로드
  - 사용할 도구

import requests

# 현재 테스트에 대한 방법론 조회
response = requests.post('http://localhost:5004/search', json={
    'query': 'session fixation testing',
    'n_results': 3
})
methodology = response.json()['results']