Pentest Manager A Simple Vulnerability Management System

🛡️ Pentest Manager

Faraday에서 영감을 받은 간단한 취약점 관리 시스템 (Vulnerability Management System)

• Antigravity를 통해 작성

Pentest Manager는 보안 테스트 중 발견된 취약점을 효율적으로 관리하고 추적할 수 있는 웹 기반 플랫폼입니다.

✨ 주요 기능

• 🗂️ 워크스페이스 관리 - 프로젝트별로 작업 그룹화
• 💻 호스트 관리 - 스캔 대상 시스템 추적
• 🔍 취약점 관리 - 심각도별 취약점 분류 및 관리
• 📊 대시보드 - 실시간 통계 및 시각화
• 📝 자동 문서화 - Swagger UI를 통한 API 문서

🏗️ 기술 스택

백엔드

• FastAPI - 고성능 Python 웹 프레임워크
• SQLAlchemy - ORM (Object-Relational Mapping)
• SQLite - 경량 데이터베이스
• Pydantic - 데이터 검증

프론트엔드

• React 18 - UI 라이브러리
• TypeScript - 타입 안정성
• Vite - 빠른 빌드 도구
• Axios - HTTP 클라이언트

📁 프로젝트 구조

Dashboard/
├── backend/              # FastAPI 백엔드
│   ├── api/             # API 라우터
│   │   ├── dashboard.py
│   │   ├── hosts.py
│   │   ├── vulnerabilities.py
│   │   └── workspaces.py
│   ├── main.py          # 앱 진입점
│   ├── database.py      # DB 설정
│   ├── models.py        # ORM 모델
│   ├── schemas.py       # Pydantic 스키마
│   └── requirements.txt
│
└── frontend/            # React 프론트엔드
    ├── src/
    │   ├── components/  # React 컴포넌트
    │   ├── services/    # API 클라이언트
    │   ├── App.tsx
    │   └── main.tsx
    ├── package.json
    └── vite.config.ts

🚀 시작하기

사전 요구사항

• Python 3.9+
• Node.js 16+
• npm 또는 yarn

백엔드 설정

# 백엔드 디렉토리로 이동
cd backend

# 가상환경 생성 (옵션)
python3 -m venv venv
source venv/bin/activate  # Windows: venv\Scripts\activate

# 의존성 설치
pip3 install -r requirements.txt

# 서버 실행
python3 -m uvicorn main:app --reload --port 8000

백엔드 서버가 http://127.0.0.1:8000 에서 실행됩니다.

📖 API 문서: http://127.0.0.1:8000/docs

프론트엔드 설정

새 터미널 창에서:

# 프론트엔드 디렉토리로 이동
cd frontend

# 의존성 설치
npm install

# 개발 서버 실행
npm run dev

프론트엔드가 http://localhost:5173 에서 실행됩니다.

📖 API 엔드포인트

Workspaces

• GET /api/workspaces - 워크스페이스 목록 조회
• POST /api/workspaces - 새 워크스페이스 생성
• GET /api/workspaces/{id} - 워크스페이스 상세 조회
• PUT /api/workspaces/{id} - 워크스페이스 수정
• DELETE /api/workspaces/{id} - 워크스페이스 삭제

Hosts

• GET /api/hosts - 호스트 목록 조회
• POST /api/hosts - 새 호스트 추가
• GET /api/hosts/{id} - 호스트 상세 조회
• PUT /api/hosts/{id} - 호스트 정보 수정
• DELETE /api/hosts/{id} - 호스트 삭제

Vulnerabilities

• GET /api/vulnerabilities - 취약점 목록 조회
• POST /api/vulnerabilities - 새 취약점 등록
• GET /api/vulnerabilities/{id} - 취약점 상세 조회
• PUT /api/vulnerabilities/{id} - 취약점 정보 수정
• DELETE /api/vulnerabilities/{id} - 취약점 삭제

Dashboard

• GET /api/dashboard/stats - 대시보드 통계 조회

🎨 주요 특징

백엔드

• ✅ RESTful API 설계
• ✅ 자동 데이터 검증 (Pydantic)
• ✅ Swagger/OpenAPI 자동 문서화
• ✅ CORS 지원
• ✅ SQLAlchemy ORM

프론트엔드

• ✅ 다크 테마 UI
• ✅ 반응형 디자인
• ✅ 부드러운 애니메이션
• ✅ 심각도별 색상 코드 (Critical, High, Medium, Low, Info)
• ✅ TypeScript 타입 안정성

📸 스크린샷

대시보드

시스템 전체 통계를 한눈에 확인

워크스페이스

프로젝트별 작업 공간 관리

호스트

스캔 대상 시스템 등록 및 관리

취약점

발견된 보안 취약점 추적 및 관리

🛠️ 사용 예시

1. 워크스페이스 생성

curl -X POST "http://localhost:8000/api/workspaces/" \
  -H "Content-Type: application/json" \
  -d '{
    "name": "Web Application Security Test",
    "description": "Q1 2024 Security Assessment"
  }'

2. 호스트 추가

curl -X POST "http://localhost:8000/api/hosts/" \
  -H "Content-Type: application/json" \
  -d '{
    "workspace_id": 1,
    "ip": "192.168.1.100",
    "hostname": "webserver",
    "os": "Ubuntu 22.04"
  }'

3. 취약점 등록

curl -X POST "http://localhost:8000/api/vulnerabilities/" \
  -H "Content-Type: application/json" \
  -d '{
    "workspace_id": 1,
    "host_id": 1,
    "name": "SQL Injection",
    "severity": "high",
    "description": "SQL injection vulnerability found in login form"
  }'

🔮 향후 개선 계획

• 사용자 인증 및 권한 관리
• 플러그인 시스템 (Nmap, Burp Suite 통합)
• PDF 보고서 자동 생성
• 검색 및 필터링 고도화
• 페이지네이션
• WebSocket 실시간 업데이트
• 데이터 내보내기 (CSV, JSON, XML)
• 취약점 템플릿 시스템
• 협업 기능 (댓글, 태그)

🤝 기여하기

기여는 언제나 환영합니다! 다음 단계를 따라주세요:

1. 이 저장소를 Fork
2. Feature 브랜치 생성 (git checkout -b feature/AmazingFeature)
3. 변경사항 커밋 (git commit -m 'Add some AmazingFeature')
4. 브랜치에 Push (git push origin feature/AmazingFeature)
5. Pull Request 생성

📝 라이선스

이 프로젝트는 MIT 라이선스 하에 있습니다.

🙏 감사의 말

• Faraday - 프로젝트 영감
• FastAPI - 훌륭한 웹 프레임워크
• React - UI 라이브러리

---

**Made with ❤️ for Security Professionals