数据库代理层是位于应用服务和数据库服务之间的中间组件，所有发往数据库的SQL请求都会先经过代理层处理，再转发到目标数据库。利用这一特性，我们可以在代理层对SQL语句进行解析、校验和改写，从流量入口拦截SQL注入攻击，其中中间件层的语句重写是核心实现手段。
数据库代理层拦截SQL注入的核心原理
SQL注入的本质是攻击者通过在输入参数中插入恶意SQL片段，改变原语句的逻辑。数据库代理层拦截的核心逻辑分为三步：首先是捕获所有经过代理的SQL请求，然后对SQL语句进行语法解析，提取出语句结构、参数内容、操作类型等信息，最后根据预设的安全规则判断是否存在注入风险，对风险语句进行重写或直接拦截。
相比在应用层做防御，代理层拦截的优势在于覆盖范围广，不需要逐个修改业务系统的代码，只要是经过代理的数据库请求都能被统一处理，尤其适合多应用共享数据库、遗留系统难以改造的场景。
中间件层语句重写的实现步骤
1. SQL语句解析
首先需要实现对SQL语句的解析，提取出语句的关键元素。我们可以借助开源的SQL解析库来完成，避免手动解析的复杂度和错误。以下是使用Java的Apache Calcite解析SQL的示例代码：
import org.apache.calcite.sql.SqlNode;
import org.apache.calcite.sql.parser.SqlParser;
import org.apache.calcite.sql.parser.SqlParseException;
public class SqlParseUtil {
public static SqlNode parseSql(String sql) throws SqlParseException {
// 创建SQL解析器配置
SqlParser.Config config = SqlParser.config().withCaseSensitive(false);
SqlParser parser = SqlParser.create(sql, config);
// 解析SQL得到语法树节点
return parser.parseQuery();
}
public static void main(String[] args) {
String testSql = "SELECT * FROM user WHERE id = 1";
try {
SqlNode sqlNode = parseSql(testSql);
System.out.println("解析后的SQL节点类型：" + sqlNode.getKind());
} catch (SqlParseException e) {
System.out.println("SQL解析失败：" + e.getMessage());
}
}
}
2. 注入风险识别
解析完成后，需要识别语句中是否存在注入风险。常见的风险特征包括：参数中包含SQL注释符（如--、/* */）、出现了原语句不存在的SQL关键字（如UNION、DROP、EXEC等）、参数类型与预期不符（比如数值型参数出现了字符串内容）。我们可以维护一个风险特征规则库，对解析后的参数内容进行匹配校验。
3. 语句重写逻辑实现
针对识别出的风险语句，核心的重写策略有以下几种：
- 参数化替换：将拼接在SQL中的用户输入参数替换为占位符，把参数值单独传递到数据库，避免恶意内容被当作SQL指令执行。比如原语句是
SELECT * FROM user WHERE name = '${input}'，重写后变为SELECT * FROM user WHERE name = ?，参数值单独传递。 - 危险操作过滤：如果语句中包含非业务允许的操作类型，比如删表、执行系统命令等，直接拦截请求，返回错误提示。
- 合法化改写：对参数中的特殊字符进行转义，比如把单引号转义为两个单引号，避免闭合原语句的引号结构。
以下是参数化重写的简化示例代码：
import java.util.regex.Matcher;
import java.util.regex.Pattern;
public class SqlRewriteUtil {
// 匹配SQL中字符串拼接参数的正则，这里简化匹配单引号包裹的参数
private static final Pattern PARAM_PATTERN = Pattern.compile("'([^']*)'");
public static String rewriteToParamSql(String originalSql) {
Matcher matcher = PARAM_PATTERN.matcher(originalSql);
StringBuffer result = new StringBuffer();
int paramIndex = 0;
// 逐个替换字符串参数为占位符
while (matcher.find()) {
paramIndex++;
matcher.appendReplacement(result, "?");
// 实际场景中需要把匹配到的参数值保存到参数列表中，单独传递给数据库
}
matcher.appendTail(result);
return result.toString();
}
public static void main(String[] args) {
String riskSql = "SELECT * FROM user WHERE name = 'admin' OR '1'='1'";
String safeSql = rewriteToParamSql(riskSql);
System.out.println("重写后的SQL：" + safeSql);
}
}
实践中的注意事项
首先要注意性能影响，SQL解析和重写会增加请求的处理耗时，代理层需要做好缓存机制，对重复且安全的SQL语句可以缓存解析结果，减少重复计算。其次要避免过度拦截，需要结合业务场景调整风险规则，比如某些业务场景确实需要用到UNION查询，就需要把这类合法场景加入白名单。最后要做好日志记录，所有被拦截或重写的请求都要记录详细日志，方便后续排查问题和优化规则。
总结
利用数据库代理层在中间件层实现语句重写，是一种高效且低侵入的SQL注入防御方案，能够在不改动业务代码的前提下，从流量入口统一管控SQL安全风险。实际落地时需要根据自身业务特点选择合适的解析组件，制定合理的风险规则，平衡安全性和系统性能，才能最大化发挥代理层防御的价值。