# Python 3.7+ 표준 라이브러리만 필요, 추가 의존성 없음
python3 --version  # Python 버전 확인

# 단계 1: 테스트 실행
python test_demo.py

# 단계 2: 단일 APK 탐지
python unity_vulnerability_checker.py your_game.apk

# 단계 3: 일괄 탐지
python unity_vulnerability_checker.py --batch /path/to/apks

# 단계 4: 사용 예시 확인
python usage_examples.py

from unity_vulnerability_checker import check_unity_vulnerability

# 단일 APK 파일 탐지
result = check_unity_vulnerability("your_game.apk")

# 방법 1: 편리한 속성 사용
if result.is_vulnerable:
    print(f"⚠️  앱이 영향을 받습니다! ({result.version})")
    print(f"권장 사항: {result.recommendation}")
elif result.is_safe:
    print(f"✅ 앱이 안전합니다")
    print(f"이유: {result.message}")
else:
    print(f"❓ 수동 검토 필요")
    print(f"이유: {result.message}")

# 방법 2: status 필드 사용
if result.status == "positive":
    print(f"영향 받음: {result.version}")

from unity_vulnerability_checker import batch_check

# 디렉토리 일괄 스캔
results = batch_check("/path/to/apk/folder")

# 영향을 받는 앱 필터링
vulnerable_apps = [
    (name, result) 
    for name, result in results.items() 
    if result.is_vulnerable
]
print(f"{len(vulnerable_apps)}개의 영향을 받는 앱 발견")
for name, result in vulnerable_apps:
    print(f"  - {name}: {result.version}")

from unity_vulnerability_checker import UnityVulnerabilityChecker

class MyPlatform:
    """당신의 게임 플랫폼 시스템"""

    def __init__(self):
        self.checker = UnityVulnerabilityChecker()

    def check_new_upload(self, apk_path: str) -> bool:
        """새로 업로드된 APK를 검사하고 출시 가능 여부 반환"""
        result = self.checker.check_apk(apk_path)

        if result.is_vulnerable:
            # 영향 받음 - 출시 거부
            self.notify_developer(
                f"귀하의 앱이 Unity 취약점의 영향을 받습니다\n"
                f"버전: {result.version}\n"
                f"권장 사항: {result.recommendation}"
            )
            return False
        elif result.is_safe:
            # 안전 - 출시 승인
            return True
        else:
            # 판단 불가 - 수동 검토
            self.queue_manual_review(apk_path, result.message)
            return False

    def notify_developer(self, message: str):
        """개발자에게 알림 (알림 로직 구현)"""
        pass

    def queue_manual_review(self, apk_path: str):
        """수동 검토 대기열에 추가 (로직 구현)"""
        pass

# 영향을 받는 앱
⚠️  이 앱은 CVE-2025-59489의 영향을 받습니다 (Unity 2019.2.6f1)

# 안전한 앱 (3가지 경우)
✅ 이 앱은 안전합니다 - Unity 앱이 아님
✅ 이 앱은 안전합니다 - Unity patcher 도구로 이미 수정됨
✅ 이 앱은 안전합니다 - 수정된 버전 (2019.4.41f1)

# 판단 불가
❓ 판단 불가 - Unity 앱이지만 버전을 추출할 수 없음, 패킹 또는 난독화되었을 수 있으므로 수동 검토 권장
⚠️  파일이 너무 커서 처리할 수 없음 (1500.0MB, 제한 1024MB) - 코드 15행의 MAX_APK_SIZE_MB를 더 큰 값으로 수정

.
├── unity_vulnerability_checker.py  # 핵심 모듈
│   ├─ UnityVulnerabilityChecker 클래스 - 취약점 탐지 핵심 엔진
│   ├─ check_unity_vulnerability() - 간소화된 탐지 함수
│   ├─ batch_check() - 일괄 탐지 기능
│   └─ 완전한 버전 파싱 및 판정 로직
│
├── usage_examples.py               # 사용 예시
│   ├─ 기본 사용 예시
│   ├─ 일괄 탐지 예시
│   ├─ 플랫폼 통합 예시
│   ├─ 자동 대응 예시
│   └─ Flask API 통합 예시
│
├── test_demo.py                    # 테스트 데모
│   ├─ 버전 번호 파싱 테스트
│   ├─ 버전 추출 테스트
│   ├─ 수정 판정 테스트
│   ├─ APK 스캔 시나리오 시뮬레이션
│   └─ 성능 벤치마크 테스트
│
├── test_version_detection.py       # 단위 테스트 (33개 테스트 케이스)
│
├── README.md                       # 전체 문서 (빠른 시작 가이드 포함)
│
└── CLAUDE.md                       # 개발 규칙

# 단일 파일 탐지
python unity_vulnerability_checker.py game.apk

# 상세 로그 모드
python unity_vulnerability_checker.py game.apk -v

# 디렉토리 일괄 탐지
python unity_vulnerability_checker.py --batch /path/to/apk/folder

# 일괄 탐지 (상세 모드)
python unity_vulnerability_checker.py --batch /path/to/apk/folder -v

from unity_vulnerability_checker import batch_check

results = batch_check("/path/to/apk/directory", verbose=True)

# 영향을 받는 앱 필터링
vulnerable = {name: result for name, result in results.items()
              if result.is_vulnerable}

print(f"{len(vulnerable)}개의 영향을 받는 앱 발견")

from unity_vulnerability_checker import UnityVulnerabilityChecker

class AppReviewSystem:
    def __init__(self):
        self.checker = UnityVulnerabilityChecker(verbose=False)

    def review_upload(self, apk_path: str) -> str:
        result = self.checker.check_apk(apk_path)

        if result.is_vulnerable:
            return "REJECT"  # 출시 거부
        elif result.is_safe:
            return "APPROVE"  # 승인
        else:
            return "MANUAL_REVIEW"  # 수동 검토

reviewer = AppReviewSystem()
decision = reviewer.review_upload("new_game.apk")

from flask import Flask, request, jsonify
from unity_vulnerability_checker import check_unity_vulnerability

app = Flask(__name__)

@app.route('/api/scan', methods=['POST'])
def scan_apk():
    file = request.files['file']
    temp_path = f"/tmp/{file.filename}"
    file.save(temp_path)

    result = check_unity_vulnerability(temp_path)

    # to_dict() 메서드를 사용하여 JSON으로 직접 변환
    response = result.to_dict()
    response["filename"] = file.filename
    response["vulnerable"] = result.is_vulnerable
    
    return jsonify(response)

          ┌─────────────────┐
          │   APK 파일 읽기   │
          └────────┬─────────┘
                   │
                   ▼
          ┌─────────────────┐
          │ Unity 앱인가?    │──NO──> negative
          └────────┬─────────┘
                   │ YES
                   ▼
          ┌─────────────────────┐
          │ patcher 패치 감지?   │──YES──> negative
          └────────┬─────────────┘
                   │ NO
                   ▼
          ┌─────────────────┐
          │ Unity 버전 추출  │──실패──> inconclusive
          └────────┬─────────┘
                   │ 성공
                   ▼
          ┌─────────────────┐
          │ 버전 < 2017.1 ?  │──YES──> negative
          └────────┬─────────┘
                   │ NO
                   ▼
          ┌─────────────────┐
          │  버전이 수정됨?   │──YES──> negative
          └────────┬─────────┘
                   │ NO
                   ▼
               positive

from unity_vulnerability_checker import check_unity_vulnerability

def pre_publish_check(apk_path):
    result = check_unity_vulnerability(apk_path)
    if result.is_vulnerable:
        return {
            "approved": False, 
            "reason": f"CVE-2025-59489 vulnerability detected ({result.version})",
            "recommendation": result.recommendation
        }
    return {"approved": True, "message": result.message}

python unity_vulnerability_checker.py --batch /data/published_apps > scan_report.txt

import schedule
from unity_vulnerability_checker import batch_check

def daily_scan():
    results = batch_check("/data/new_uploads")
    vulnerable = [(name, result) for name, result in results.items() if result.is_vulnerable]
    if vulnerable:
        # 상세 정보가 포함된 알림 전송
        for name, result in vulnerable:
            send_notification(name, result.version, result.message)

schedule.every().day.at("02:00").do(daily_scan)

from unity_vulnerability_checker import batch_check
import json

results = batch_check("/data/all_apps")
report = {
    "total": len(results),
    "vulnerable": sum(1 for r in results.values() if r.is_vulnerable),
    "safe": sum(1 for r in results.values() if r.is_safe),
    "unclear": sum(1 for r in results.values() if r.is_uncertain),
    # 상세 목록
    "vulnerable_apps": [
        {"name": name, "version": r.version, "message": r.message}
        for name, r in results.items() if r.is_vulnerable
    ]
}

with open("security_report.json", "w") as f:
    json.dump(report, f, indent=2)

#!/bin/bash
# CI pipeline에서 사용

python unity_vulnerability_checker.py build/output.apk
result=$(python unity_vulnerability_checker.py build/output.apk | grep "탐지 결과")

if echo "$result" | grep -q "positive"; then
    echo "Security check failed: CVE-2025-59489 detected"
    exit 1
fi

echo "Security check passed"

# 의존성 설치
pip install ruff pre-commit

# pre-commit hooks 설치
pre-commit install

# 모든 검사 수동 실행
pre-commit run --all-files

# pre-commit이 실패하면 일부 문제를 자동 수정합니다
# 수정 후 git add를 다시 실행하고 커밋해야 합니다
git add .
git commit -m "your message"

# Python 파일만 검사
ruff check *.py

# 문제 자동 수정
ruff check --fix *.py

# 코드 포맷팅
ruff format *.py