1
2
3
4
5
6
7
8
9

graph TD
    A[Model Release] --> B[System Card Publishing]
    B --> C[Red Teaming Results]
    C --> D[Refusal Rate Analysis]
    B --> E[Responsible Disclosure Policy]
    D --> F[Security Assessment]
    C --> F
    E --> G[Coordinated Vulnerability Disclosure]
    F --> G

1
2
3
4
5
6
7

graph LR
    A[보안 과제 데이터셋] --> B[Human Red Team]
    A --> C[Automated Evaluation]
    B --> D[Refusal Rate 측정]
    C --> D
    D --> E[성공률 분석]
    E --> F[위험도 등급 산출]

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44

#!/usr/bin/env python3
"""
Vulnerability Pattern Scanner - Educational Purpose Only
Claude API를 활용한 코드 리뷰 어시스턴트 개념 예시
"""

import os
import json
from dataclasses import dataclass
from typing import List, Optional

@dataclass
class VulnerabilityFinding:
    """취약점 발견 결과 데이터 구조"""
    file_path: str
    line_number: int
    severity: str  # CRITICAL, HIGH, MEDIUM, LOW, INFO
    category: str  # OWASP Top 10 카테고리
    description: str
    cwe_id: Optional[str] = None
    remediation: Optional[str] = None

class CodeSecurityScanner:
    """
    AI 보조 코드 보안 스캐너
    
    주의: 이 도구는 자동화된 정적 분석의 보조 수단입니다.
    모든 발견 사항은 인간 보안 전문가의 검토가 필요합니다.
    """
    
    def __init__(self, api_endpoint: str, model_version: str):
        self.api_endpoint = api_endpoint
        self.model_version = model_version
        self.findings: List[VulnerabilityFinding] = []
        
    def create_analysis_prompt(self, code_snippet: str, context: str) -> str:
        """보안 분석용 프롬프트 구성"""
        prompt = f"""
You are a security code reviewer. Analyze the following code for security vulnerabilities.
Focus on OWASP Top 10 categories and provide actionable remediation advice.

Context: {context}

Code:

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55

Provide your analysis in JSON format with the following structure:
{{
    "findings": [
        {{
            "line_number": int,
            "severity": "CRITICAL|HIGH|MEDIUM|LOW|INFO",
            "category": "OWASP category",
            "description": "detailed description",
            "cwe_id": "CWE-XXX",
            "remediation": "specific fix recommendation"
        }}
    ]
}}
"""
        return prompt.strip()
    
    def scan_file(self, file_path: str, file_content: str) -> List[VulnerabilityFinding]:
        """단일 파일 스캔 수행"""
        
        # 파일 확장자에 따른 컨텍스트 설정
        ext = os.path.splitext(file_path)[1].lower()
        context_map = {
            '.py': 'Python web application - focus on injection, deserialization',
            '.js': 'JavaScript/Node.js - focus on XSS, prototype pollution',
            '.java': 'Java enterprise - focus on injection, insecure deserialization',
            '.php': 'PHP application - focus on SQL injection, file inclusion',
            '.go': 'Go service - focus on concurrency issues, command injection',
        }
        context = context_map.get(ext, 'General application code')
        
        # 실제 API 호출은 구현 필요
        prompt = self.create_analysis_prompt(file_content, context)
        
        # 개념적 반환 (실제로는 API 응답 파싱)
        return []
    
    def generate_report(self, output_format: str = 'markdown') -> str:
        """분석 결과 보고서 생성"""
        if output_format == 'markdown':
            report = "# Security Code Review Report

"
            report += "## Summary

"
            report += f"- Total Findings: {len(self.findings)}
"
            
            severity_counts = {}
            for f in self.findings:
                severity_counts[f.severity] = severity_counts.get(f.severity, 0) + 1
            
            report += "
## Findings by Severity

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24

"
            for sev in ['CRITICAL', 'HIGH', 'MEDIUM', 'LOW', 'INFO']:
                if sev in severity_counts:
                    report += f"- {sev}: {severity_counts[sev]}
"
            
            return report
        
        return json.dumps([f.__dict__ for f in self.findings], indent=2)

# 사용 예시 (개념적)
if __name__ == "__main__":
    print("""
    ============================================
    AI-Assisted Code Security Scanner
    Educational & Defensive Purpose Only
    ============================================
    
    이 도구는 개념 증명용입니다.
    실제 사용 시:
    1. 적절한 API 인증 설정
    2. 스캔 대상에 대한 명시적 승인 획득
    3. 모든 결과를 인간 전문가가 검토
    """)

1
2
3
4
5
6
7
8

graph TD
    A[User Query] --> B{Intent Classification}
    B --> C[Benevolent/Neutral]
    B --> D[Malign/Unsure]
    C --> E[Normal Response]
    D --> F[Refusal with Explanation]
    E --> G[Helpful Security Guidance]
    F --> H[Educational Redirect]

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16

# 프롬프트 템플릿 예시

다음 코드를 OWASP Top 10 관점에서 보안 리뷰해주세요:
- 입력 검증 취약점
- 인증/인가 결함
- 암호화 문제
- 보안 설정 오류

코드:
[코드 스니펫]

각 발견 사항에 대해:
1. 심각도 (Critical/High/Medium/Low)
2. 영향 범위
3. 구체적 완화 방법
4. 관련 CWE/OWASP 참조