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
