理性看待AI对代码库的改变

  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
 56
 57
 58
 59
 60
 61
 62
 63
 64
 65
 66
 67
 68
 69
 70
 71
 72
 73
 74
 75
 76
 77
 78
 79
 80
 81
 82
 83
 84
 85
 86
 87
 88
 89
 90
 91
 92
 93
 94
 95
 96
 97
 98
 99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133

# 示例1：AI生成代码的安全审查工具
# 用途：自动检测AI生成的代码是否存在常见的安全问题
import re
import subprocess

class AICodeSecurityChecker:
    """检查AI生成代码的安全隐患"""
    
    def __init__(self):
        # 常见的不安全模式
        self.dangerous_patterns = {
            'eval_usage': (r'\beval\s*\(', 'eval() 可能导致代码注入攻击'),
            'exec_usage': (r'\bexec\s*\(', 'exec() 存在严重安全风险'),
            'hardcoded_password': (r'password\s*=\s*["\'][^"\']+["\']', '硬编码密码应使用环境变量'),
            'sql_injection': (r'execute\s*\(\s*f["\']', 'SQL查询存在注入风险'),
            'hardcoded_api_key': (r'api[_-]?key\s*=\s*["\'][A-Za-z0-9]{20,}["\']', 'API密钥不应硬编码'),
            'os_system': (r'os\.system\s*\(', 'os.system() 存在命令注入风险'),
        }
    
    def scan_file(self, file_path):
        """
        扫描指定文件，查找潜在的安全问题
        
        参数:
            file_path: 要扫描的文件路径
        返回:
            发现的问题列表
        """
        issues = []
        
        try:
            with open(file_path, 'r', encoding='utf-8') as f:
                lines = f.readlines()
                
            for line_num, line in enumerate(lines, 1):
                for pattern_name, (pattern, description) in self.dangerous_patterns.items():
                    if re.search(pattern, line, re.IGNORECASE):
                        issues.append({
                            'file': file_path,
                            'line': line_num,
                            'type': pattern_name,
                            'description': description,
                            'code': line.strip()
                        })
                        
        except FileNotFoundError:
            print(f"文件未找到: {file_path}")
            
        return issues
    
    def generate_report(self, issues):
        """生成安全审查报告"""
        if not issues:
            print("[PASS] 未发现安全问题")
            return
            
        print(f"[WARNING] 发现 {len(issues)} 个潜在安全问题:\n")
        for issue in issues:
            print(f"[{issue['type']}] {issue['file']}:{issue['line']}")
            print(f"   描述: {issue['description']}")
            print(f"   代码: {issue['code']}")
            print()
    
    def scan_directory(self, directory_path):
        """
        扫描目录下的所有Python文件
        
        参数:
            directory_path: 要扫描的目录路径
        返回:
            所有发现的问题列表
        """
        all_issues = []
        
        try:
            result = subprocess.run(
                ['find', directory_path, '-name', '*.py', '-type', 'f'],
                capture_output=True,
                text=True
            )
            
            if result.returncode == 0:
                file_list = result.stdout.strip().split('\n')
                for file_path in file_list:
                    if file_path:
                        issues = self.scan_file(file_path)
                        all_issues.extend(issues)
                        
        except Exception as e:
            print(f"扫描目录时出错: {e}")
            
        return all_issues
    
    def auto_fix(self, file_path, dry_run=True):
        """
        自动修复检测到的问题
        
        参数:
            file_path: 要修复的文件路径
            dry_run: 是否仅模拟修复（不实际修改文件）
        """
        issues = self.scan_file(file_path)
        
        if not issues:
            print("[PASS] 文件无安全问题")
            return
            
        print(f"[INFO] 发现 {len(issues)} 个问题待修复")
        
        if dry_run:
            print("[INFO] 模拟模式 - 不会实际修改文件")
            for issue in issues:
                print(f"   行 {issue['line']}: {issue['type']} - {issue['description']}")
        else:
            print("[INFO] 开始修复...")
            # 此处可添加具体的修复逻辑
            print("[DONE] 修复完成")


# 使用示例
if __name__ == "__main__":
    checker = AICodeSecurityChecker()
    
    # 扫描单个文件
    issues = checker.scan_file("example.py")
    checker.generate_report(issues)
    
    # 扫描目录
    # all_issues = checker.scan_directory("./src")
    # checker.generate_report(all_issues)
    
    # 模拟修复
    # checker.auto_fix("example.py", dry_run=True)