MyBatis（23）MyBatis 如何实现 SQL 注入的防护

MyBatis 通过预处理语句（Prepared Statements）的使用来防止 SQL 注入攻击。预处理语句不仅能提高数据库操作的性能，还能有效防御 SQL 注入。在这里，我们将深入探讨 MyBatis 如何利用这一机制来预防 SQL 注入，并结合源码分析其实现细节。

预处理语句和 SQL 注入

SQL 注入攻击发生在攻击者将 SQL 代码注入到应用程序的输入参数中，这些参数之后被嵌入到 SQL 语句中并执行。预处理语句（或参数化查询）通过定义所有的 SQL 代码，并在执行前给 SQL 语句的参数赋值来避免这种攻击。这种方法确保了数据被明确地区分为代码和数据，即使数据来自不可信的源也不会被错误地解释为代码。

MyBatis 的实现

MyBatis 在底层使用 JDBC 的预处理语句来执行所有的 SQL 操作。当你使用 MyBatis 的 Mapper 接口或 XML 映射文件定义 SQL 语句时，MyBatis 会在执行这些语句时创建对应的 PreparedStatement 对象。这一过程主要涉及以下几个关键类和接口：

SqlSession：这是 MyBatis 的主要接口，用于发送 SQL 命令给数据库。
Executor：这个接口定义了数据库操作的基本方法，例如 update, query, 和 commit。它的实现类负责管理 PreparedStatement 的创建和执行。
StatementHandler：这个接口及其实现类处理 SQL 语句的预处理和执行。它使用 ParameterHandler 来设置 PreparedStatement 的参数。

代码示例

考虑以下 MyBatis Mapper 接口方法：

@Select("SELECT * FROM users WHERE id = #{id}")
User getUserById(@Param("id") Integer id);

在这个例子中，#{id} 是 MyBatis 的参数占位符。当这个方法被调用时，MyBatis 执行以下操作来预防 SQL 注入：

解析 SQL 语句：首先，MyBatis 解析 Mapper 接口或 XML 映射文件中定义的 SQL 语句，识别出参数占位符。
创建 PreparedStatement：MyBatis 通过 JDBC 创建一个 PreparedStatement 对象，SQL 语句是 "SELECT * FROM users WHERE id = ?"，其中 ? 是一个参数占位符。
设置参数：MyBatis 使用 ParameterHandler 来设置 PreparedStatement 的参数。在本例中，它会将 id 参数的值绑定到 SQL 语句的占位符上。
执行 SQL 语句：最后，执行预处理语句。

源码解析

这一过程主要涉及 DefaultSqlSession, BaseExecutor, SimpleStatementHandler, 和 DefaultParameterHandler 类。

以 BaseExecutor 类中的 query 方法为例，它会创建一个 PreparedStatement 对象，并通过 ParameterHandler 设置参数：

PreparedStatement stmt = prepareStatement(handler, ms.getStatementLog());
handler.getParameterHandler().setParameters(stmt);

prepareStatement 方法会调用 connection.prepareStatement(sql)，其中 sql 是已经解析过的，包含参数占位符的 SQL 字符串。ParameterHandler 的实现类（通常是 DefaultParameterHandler）会负责将方法参数值绑定到 SQL 语句的占位符上。