C# 实现：Codex 输出漏洞检测与防御策略

示例展示了如何检测和防御 Codex 生成的代码中常见的漏洞（如 SQL 注入、XSS、硬编码凭证等），并提供了对应的修复策略：

using System;
using System.Data.SqlClient;
using System.Text.RegularExpressions;
using System.Web;

public class CodexVulnerabilityScanner
{
    // 检测 SQL 注入漏洞
    public static bool DetectSqlInjection(string codeSnippet)
    {
        string pattern = @"SqlCommand.*?\+.*?[\""\'].*?[\""\']";
        return Regex.IsMatch(codeSnippet, pattern, RegexOptions.IgnoreCase);
    }

    // SQL 注入修复策略
    public static string FixSqlInjection(string rawQuery, params object[] parameters)
    {
        using (var connection = new SqlConnection("Your_Connection_String"))
        {
            var command = new SqlCommand(rawQuery, connection);
            for (int i = 0; i < parameters.Length; i++)
            {
                command.Parameters.AddWithValue($"@param{i}", parameters[i]);
            }
            return command.CommandText;
        }
    }

    // 检测 XSS 漏洞
    public static bool DetectXss(string codeSnippet)
    {
        string pattern = @"Response\.Write.*?\+.*?Request\.(QueryString|Form)";
        return Regex.IsMatch(codeSnippet, pattern, RegexOptions.IgnoreCase);
    }

    // XSS 修复策略
    public static string SanitizeInput(string input)
    {
        return HttpUtility.HtmlEncode(input);
    }

    // 检测硬编码凭证
    public static bool DetectHardcodedCredentials(string codeSnippet)
    {
        string pattern = @"(password|pwd|connectionString)\s*=\s*[\""\'][^\""\']+[\""\']";
        return Regex.IsMatch(codeSnippet, pattern, RegexOptions.IgnoreCase);
    }

    // 硬编码凭证修复策略
    public static string GetSecureConnectionString()
    {
        return ConfigurationManager.AppSettings["SecureDbConnection"];
    }
}

// 使用示例
public class Program
{
    public static void Main()
    {
        string dangerousCode = "SELECT * FROM Users WHERE Name = '" + userInput + "'";
        
        if (CodexVulnerabilityScanner.DetectSqlInjection(dangerousCode))
        {
            string safeQuery = CodexVulnerabilityScanner.FixSqlInjection(
                "SELECT * FROM Users WHERE Name = @param0", userInput);
            Console.WriteLine($"修复后的查询: {safeQuery}");
        }
    }
}

常见漏洞类型与对应策略

SQL 注入漏洞
检测方法：检查字符串拼接的 SQL 查询语句
修复方案：使用参数化查询（SqlParameter）替代字符串拼接

跨站脚本（XSS）
检测方法：检查未经验证直接输出的用户输入
修复方案：对所有输出到 HTML 的内容进行编码（HtmlEncode）

硬编码敏感信息
检测方法：检查代码中直接包含密码、API 密钥等
修复方案：将敏感信息移至配置文件或密钥管理服务

不安全的反序列化
检测方法：检查 BinaryFormatter 等高风险反序列化操作
修复方案：使用 JSON 等安全序列化格式，并验证数据签名

路径遍历漏洞
检测方法：检查未净化的文件路径拼接操作
修复方案：使用 Path.Combine 并限制访问目录范围

增强安全性的实践建议

• 实施输入验证白名单机制
• 使用安全的 API（如 Entity Framework 代替原始 SQL）
• 定期依赖项扫描（如 NuGet 漏洞检查）
• 集成静态代码分析工具（如 SonarQube）
• 遵循最小权限原则配置数据库账户

注意：实际应用中应结合具体业务场景调整检测规则，并考虑使用专业的安全扫描工具进行深度检测。