SQL注入攻击作为Web应用最常见的安全威胁之一，长期以来一直困扰着开发者。传统的防御手段如参数化查询、输入验证虽然有效，但依赖开发者的经验和严谨性，难免会有疏漏。本文将介绍如何利用C# 9.0引入的源生成器(Source Generator)技术，在编译期彻底封死SQL注入漏洞，让黑客无懈可击。
SQL注入的传统防御方案及其局限性
在探讨新技术之前，我们先回顾一下传统的SQL注入防御方案：
1. 参数化查询
using (var connection = new SqlConnection(connectionString))
{
var query = "SELECT * FROM Users WHERE Username = @Username AND Password = @Password";
using (var command = new SqlCommand(query, connection))
{
command.Parameters.AddWithValue("@Username", username);
command.Parameters.AddWithValue("@Password", password);
// 执行查询
}
}
2. 输入验证与白名单
public bool IsValidInput(string input)
{
// 只允许字母和数字
return Regex.IsMatch(input, @"^[a-zA-Z0-9]+$");
}
3. ORM框架
var user = dbContext.Users
.Where(u => u.Username == username && u.Password == password)
.FirstOrDefault();
这些方法虽然有效，但存在以下问题：
源生成器：编译期防御SQL注入的黑科技
源生成器是C# 9.0引入的一项强大功能，允许开发者在编译期分析代码并生成额外的源代码。利用这一技术，我们可以创建一个SQL注入防护系统，在编译期检测并阻止不安全的SQL操作。
基本原理
我们的解决方案基于以下思路：
实现自定义SQL查询属性
首先，我们定义一个用于标记SQL查询方法的属性：
[AttributeUsage(AttributeTargets.Method, AllowMultiple = false)]
public class SafeSqlQueryAttribute : Attribute
{
public string SqlTemplate { get; }
public SafeSqlQueryAttribute(string sqlTemplate)
{
SqlTemplate = sqlTemplate;
}
}
创建源生成器
下面是核心的源生成器实现：
[Generator]
public class SafeSqlGenerator : IIncrementalGenerator
{
public void Initialize(IncrementalGeneratorInitializationContext context)
{
// 查找所有使用了SafeSqlQueryAttribute的方法
var methodsWithAttribute = context.SyntaxProvider
.CreateSyntaxProvider(
predicate: (s, _) => s is MethodDeclarationSyntax method && method.AttributeLists.Count > 0,
transform: (ctx, _) => GetMethodWithAttribute(ctx))
.Where(m => m != null)!;
// 生成代码
context.RegisterSourceOutput(methodsWithAttribute, (spc, method) =>
{
GenerateSafeSqlMethod(spc, method!);
});
}
private MethodDeclarationSyntax? GetMethodWithAttribute(GeneratorSyntaxContext context)
{
var methodDeclaration = (MethodDeclarationSyntax)context.Node;
// 检查是否有SafeSqlQueryAttribute
foreach (var attributeList in methodDeclaration.AttributeLists)
{
foreach (var attribute in attributeList.Attributes)
{
if (attribute.Name.ToString() == "SafeSqlQuery")
{
return methodDeclaration;
}
}
}
return null;
}
private void GenerateSafeSqlMethod(SourceProductionContext context, MethodDeclarationSyntax method)
{
// 解析方法参数和SQL模板
var methodSymbol = context.Compilation.GetSemanticModel(method.SyntaxTree)
.GetDeclaredSymbol(method)! as IMethodSymbol;
var sqlTemplate = GetSqlTemplate(method);
if (string.IsNullOrEmpty(sqlTemplate))
return;
// 生成安全的SQL执行代码
var sourceCode = GenerateSafeSqlSourceCode(methodSymbol, sqlTemplate);
// 添加生成的代码
context.AddSource($"{methodSymbol.Name}.g.cs", sourceCode);
}
private string GetSqlTemplate(MethodDeclarationSyntax method)
{
// 从属性中提取SQL模板
// ...
}
private string GenerateSafeSqlSourceCode(IMethodSymbol method, string sqlTemplate)
{
// 生成安全的SQL执行代码
// ...
}
}
使用源生成器的安全SQL查询
下面是一个使用我们自定义属性和源生成器的示例：
public class UserRepository
{
private readonly string _connectionString;
public UserRepository(string connectionString)
{
_connectionString = connectionString;
}
[SafeSqlQuery("SELECT * FROM Users WHERE Username = @Username AND IsActive = @IsActive")]
public IEnumerable<User> GetActiveUsers(string username, bool isActive)
{
// 这个方法体将由源生成器替换
throw new NotImplementedException("This method is implemented by source generator");
}
}
源生成器会为这个方法生成类似以下的代码：
public partial class UserRepository
{
public IEnumerable<User> GetActiveUsers(string username, bool isActive)
{
using (var connection = new SqlConnection(_connectionString))
{
connection.Open();
var query = "SELECT * FROM Users WHERE Username = @Username AND IsActive = @IsActive";
using (var command = new SqlCommand(query, connection))
{
// 自动参数化所有输入
command.Parameters.AddWithValue("@Username", username);
command.Parameters.AddWithValue("@IsActive", isActive);
using (var reader = command.ExecuteReader())
{
while (reader.Read())
{
yield return new User
{
Id = reader.GetInt32(0),
Username = reader.GetString(1),
// 其他属性映射
};
}
}
}
}
}
}
强化安全：禁用不安全的SQL构造方式
为了彻底封死SQL注入漏洞，我们还可以通过源生成器检测并报告不安全的SQL构造方式：
[Generator]
public class SqlInjectionDetector : IIncrementalGenerator
{
public void Initialize(IncrementalGeneratorInitializationContext context)
{
// 查找所有字符串拼接的SQL构造
var unsafeSqlNodes = context.SyntaxProvider
.CreateSyntaxProvider(
predicate: (s, _) => IsUnsafeSqlNode(s),
transform: (ctx, _) => ctx.Node)
.Where(n => n != null)!;
// 报告诊断信息
context.RegisterSourceOutput(unsafeSqlNodes, (spc, node) =>
{
spc.ReportDiagnostic(Diagnostic.Create(
new DiagnosticDescriptor(
"SQL001",
"不安全的SQL构造",
"发现可能存在SQL注入风险的字符串拼接SQL，请使用参数化查询或SafeSqlQueryAttribute",
"Security",
DiagnosticSeverity.Error,
isEnabledByDefault: true),
node.GetLocation()));
});
}
private bool IsUnsafeSqlNode(SyntaxNode node)
{
// 检测是否有字符串拼接的SQL构造
// ...
}
}
实际应用案例
让我们看一个具体的例子，比较传统方式和使用源生成器的方式：
传统易受攻击的代码
public IEnumerable<User> GetUsers(string searchTerm)
{
// 这个查询存在SQL注入风险
var sql = $"SELECT * FROM Users WHERE Username LIKE '%{searchTerm}%'";
using (var connection = new SqlConnection(connectionString))
{
using (var command = new SqlCommand(sql, connection))
{
// 执行查询
}
}
}
使用源生成器的安全代码
public partial class UserRepository
{
[SafeSqlQuery("SELECT * FROM Users WHERE Username LIKE @SearchTerm")]
public IEnumerable<User> GetUsers(string searchTerm)
{
// 由源生成器实现
throw new NotImplementedException();
}
}
当我们尝试编译不安全的代码时，源生成器会报错：
错误 SQL001: 发现可能存在SQL注入风险的字符串拼接SQL，请使用参数化查询或SafeSqlQueryAttribute
高级功能：动态SQL模板验证
我们还可以增强源生成器，使其能够验证SQL模板的语法正确性：
private void ValidateSqlTemplate(string sqlTemplate)
{
// 使用SQL解析器验证模板语法
try
{
var parser = new SqlParser();
parser.Parse(sqlTemplate);
}
catch (SqlParseException ex)
{
// 报告SQL语法错误
}
}
性能优势
除了安全性，源生成器方案还有以下性能优势：
- 预编译SQL语句：生成的代码可以使用预编译的SQL语句
- 优化查询执行：源生成器可以生成更高效的查询执行代码
部署与集成
将此安全机制集成到项目中非常简单：
- 在需要执行SQL查询的方法上添加
[SafeSqlQuery]属性
总结
通过利用C#源生成器技术，我们创建了一个在编译期就能够防止SQL注入的安全系统。这种方法具有以下优势：
- 彻底性：从根本上杜绝SQL注入漏洞，而不是依赖开发者的警惕性
- 自动化：安全检查和代码生成完全自动化，减少人工错误
- 高性能：编译期生成的代码通常比运行时动态生成的代码更高效
这种方法不仅让黑客无从下手，也让开发者能够更专注于业务逻辑，而不必担心SQL注入带来的安全风险。
400 186 1886
