如何用 Java 校验 SQL 语句的合法性?

本文涉及的产品
RDS MySQL DuckDB 分析主实例,集群系列 4核8GB
RDS AI 助手,专业版
简介: 本文介绍了五种校验 SQL 语句合法性的方案:1) 使用 JDBC API 的 `execute()` 方法,通过捕获异常判断合法性;2) 使用 JSqlParser 库解析 SQL 语句为 Java 对象;3) 使用正则表达式检查 SQL 语句格式;4) 使用 ANTLR 生成 SQL 解析器;5) 使用 Apache Calcite 解析 SQL。每种方法各有优劣,具体选择取决于需求和个人偏好。需要注意的是,这些方法仅能校验语法合法性,无法保证语义正确性,仍需防范 SQL 注入攻击。

方案一:使用 JDBC API 中提供的 Statement 接口的 execute()方法

要在 Java 中校验 SQL 语句的合法性,可以使用 JDBC API 中提供的 Statement 接口的 execute()方法。这个方法会尝试执行给定的 SQL 语句,如果 SQL 语句不合法,则会抛出一个 SQLException 异常。因此,我们可以利用这个异常来判断 SQL 语句的合法性。

java

代码解读

复制代码

import java.sql.*; public class SQLValidator {     public static boolean validateSQL(String sql) {        try {            Connection conn = DriverManager.getConnection("jdbc:mysql://localhost:3306/mydatabase", "username", "password");            Statement stmt = conn.createStatement();            stmt.execute(sql);            return true;        } catch (SQLException e) {            return false;        }    }     public static void main(String[] args) {        String sql1 = "SELECT * FROM mytable WHERE id = 1";        String sql2 = "SELECT * FROM mytable WHERE id = '1'";        String sql3 = "SELECT * FROM mytable WHERE id = ;DROP TABLE mytable;";         System.out.println(validateSQL(sql1)); // true        System.out.println(validateSQL(sql2)); // false        System.out.println(validateSQL(sql3)); // false    }}

复制代码

在这个示例代码中,validateSQL()方法接受一个 SQL 语句作为参数,然后尝试执行这个 SQL 语句。如果执行成功,返回 true,否则返回 false。在 main()方法中,我们调用了 validateSQL()方法来校验三个 SQL 语句的合法性,并打印了结果。

需要注意的是,这个方法只能判断 SQL 语句的语法是否合法,而无法判断 SQL 语句的语义是否合法。因此,如果应用程序允许用户输入 SQL 语句,一定要进行严格的输入校验和过滤,避免 SQL 注入攻击。

方案二:使用 JSqlParser 这个 Java 库

如果你不希望实际执行 SQL 语句,而只是想校验 SQL 语句的合法性,可以使用 JSqlParser 这个 Java 库。这个库可以将 SQL 语句解析成 Java 对象,然后你可以对这些 Java 对象进行检查,以判断 SQL 语句是否合法。

以下是一个简单的示例代码:

typescript

代码解读

复制代码

import net.sf.jsqlparser.JSQLParserException;import net.sf.jsqlparser.parser.CCJSqlParserUtil;import net.sf.jsqlparser.statement.Statement; public class SQLValidator {     public static boolean validateSQL(String sql) {        try {            Statement stmt = CCJSqlParserUtil.parse(sql);            return true;        } catch (JSQLParserException e) {            return false;        }    }     public static void main(String[] args) {        String sql1 = "SELECT * FROM mytable WHERE id = 1";        String sql2 = "SELECT * FROM mytable WHERE id = '1'";        String sql3 = "SELECT * FROM mytable WHERE id = ;DROP TABLE mytable;";         System.out.println(validateSQL(sql1)); // true        System.out.println(validateSQL(sql2)); // true        System.out.println(validateSQL(sql3)); // false    }}

复制代码

在这个示例代码中,validateSQL()方法使用 JSqlParser 库将 SQL 语句解析成 Java 对象。如果解析成功,返回 true,否则返回 false。在 main()方法中,我们调用了 validateSQL()方法来校验三个 SQL 语句的合法性,并打印了结果。

需要注意的是,JSqlParser 库只能检查 SQL 语句的语法是否合法,而无法检查 SQL 语句的语义是否合法。因此,同样需要进行严格的输入校验和过滤,避免 SQL 注入攻击。

方案三:使用正则表达式检查 SQL 语句的格式是否正确

使用正则表达式检查 SQL 语句的格式是否正确。例如,可以检查 SQL 语句是否以 SELECT、UPDATE、DELETE、INSERT 等关键字开头,是否包含必需的关键字和语法元素等。

vbnet

代码解读

复制代码

import java.util.regex.Pattern; public class SQLValidator {    private static final String SELECT_PATTERN = "^\\s*SELECT.*";    private static final String UPDATE_PATTERN = "^\\s*UPDATE.*";    private static final String DELETE_PATTERN = "^\\s*DELETE.*";    private static final String INSERT_PATTERN = "^\\s*INSERT.*";     public static boolean validateSQL(String sql) {        if (Pattern.matches(SELECT_PATTERN, sql)) {            // 校验SELECT语句的合法性            return true;        } else if (Pattern.matches(UPDATE_PATTERN, sql)) {            // 校验UPDATE语句的合法性            return true;        } else if (Pattern.matches(DELETE_PATTERN, sql)) {            // 校验DELETE语句的合法性            return true;        } else if (Pattern.matches(INSERT_PATTERN, sql)) {            // 校验INSERT语句的合法性            return true;        } else {            // SQL语句格式不正确            return false;        }    }     public static void main(String[] args) {        String sql1 = "SELECT * FROM mytable WHERE id = 1";        String sql2 = "SELECT * FROM mytable WHERE id = '1'";        String sql3 = "SELECT * FROM mytable WHERE id = ;DROP TABLE mytable;";         System.out.println(validateSQL(sql1)); // true        System.out.println(validateSQL(sql2)); // true        System.out.println(validateSQL(sql3)); // false    }}

复制代码

方案四:使用 ANTLR 等工具生成 SQL 语法解析器,然后使用生成的解析器解析 SQL 语句,以判断 SQL 语句的合法性

ANTLR 是一种流行的解析器生成器,可以根据定义的语法规则生成解析器。

以下是一个简单的示例代码:

java

代码解读

复制代码

import org.antlr.v4.runtime.*;import org.antlr.v4.runtime.tree.*; public class SQLValidator {    public static boolean validateSQL(String sql) {        try {            CharStream input = CharStreams.fromString(sql);            SQLLexer lexer = new SQLLexer(input);            CommonTokenStream tokens = new CommonTokenStream(lexer);            SQLParser parser = new SQLParser(tokens);            ParseTree tree = parser.statement();            return true;        } catch (Exception e) {            return false;        }    }     public static void main(String[] args) {        String sql1 = "SELECT * FROM mytable WHERE id = 1";        String sql2 = "SELECT * FROM mytable WHERE id = '1'";        String sql3 = "SELECT * FROM mytable WHERE id = ;DROP TABLE mytable;";         System.out.println(validateSQL(sql1)); // true        System.out.println(validateSQL(sql2)); // true        System.out.println(validateSQL(sql3)); // false    }}

复制代码

在这个示例代码中,我们使用 ANTLR 生成了一个 SQL 语法解析器,并在 validateSQL()方法中使用这个解析器来解析 SQL 语句。如果解析成功,则说明 SQL 语句格式正确,返回 true,否则返回 false。

方案五:使用 Apache Calcite 等 SQL 解析器库来解析 SQL 语句

Apache Calcite 是一个强大的 SQL 解析器和优化器,它支持大多数 SQL 语法,并能够将 SQL 语句解析为抽象语法树(AST)。

arduino

代码解读

复制代码

import org.apache.calcite.sql.SqlNode;import org.apache.calcite.sql.parser.SqlParseException;import org.apache.calcite.sql.parser.SqlParser;import org.apache.calcite.sql.parser.SqlParser.Config;import org.apache.calcite.sql.parser.SqlParserImplFactory; public class SQLValidator {    public static boolean validateSQL(String sql) {        try {            Config config = SqlParser.config();            SqlParserImplFactory factory = config.parserFactory();            SqlParser parser = SqlParser.create(sql, config.withParserFactory(factory));            SqlNode node = parser.parseStmt();            return true;        } catch (SqlParseException e) {            return false;        }    }     public static void main(String[] args) {        String sql1 = "SELECT * FROM mytable WHERE id = 1";        String sql2 = "SELECT * FROM mytable WHERE id = '1'";        String sql3 = "SELECT * FROM mytable WHERE id = ;DROP TABLE mytable;";         System.out.println(validateSQL(sql1)); // true        System.out.println(validateSQL(sql2)); // true        System.out.println(validateSQL(sql3)); // false    }}

复制代码

在这个示例代码中,我们使用 Apache Calcite 库来解析 SQL 语句。validateSQL()方法首先创建一个 SqlParser 对象,并使用它来解析传入的 SQL 语句。如果解析成功,则返回 true,否则返回 false。

总结

总的来说,使用 JDBC API 和 JSqlParser 库、正则表达式、ANTLR 解析器生成器或 Apache Calcite 库都可以实现校验 SQL 语句的合法性。具体使用哪种方法取决于你的需求和个人喜好。

相关内容拓展:(技术前沿)

近 10 年间,甚至连传统企业都开始大面积数字化时,我们发现开发内部工具的过程中,大量的页面、场景、组件等在不断重复,这种重复造轮子的工作,浪费工程师的大量时间。

针对这类问题,低代码把某些重复出现的场景、流程,具象化成一个个组件、api、数据库接口,避免了重复造轮子。极大的提高了程序员的生产效率。

介绍一款程序员都应该知道的软件 JNPF 快速开发平台,基于 Java/.Net 双技术引擎,专注于低代码,采用业内领先的 SpringBoot 微服务架构、支持 SpringCloud 模式,完善了平台的扩增基础,满足了系统快速开发、灵活拓展、无缝集成和高性能应用等综合能力;采用前后端分离模式,前端和后端的开发人员可分工合作负责不同板块,省事又便捷。


转载来源:https://juejin.cn/post/7304268951316791346

相关文章
|
8月前
|
SQL Java 数据库连接
Java中实现SQL分页的方法
无论何种情况,选择适合自己的,理解了背后的工作原理,并能根据实际需求灵活变通的方式才是最重要的。
241 9
|
SQL NoSQL Java
Java使用sql查询mongodb
通过MongoDB Atlas Data Lake或Apache Drill,可以在Java中使用SQL语法查询MongoDB数据。这两种方法都需要适当的配置和依赖库的支持。希望本文提供的示例和说明能够帮助开发者实现这一目标。
550 17
|
SQL Java 数据库连接
如何在 Java 代码中使用 JSqlParser 解析复杂的 SQL 语句?
大家好,我是 V 哥。JSqlParser 是一个用于解析 SQL 语句的 Java 库,可将 SQL 解析为 Java 对象树,支持多种 SQL 类型(如 `SELECT`、`INSERT` 等)。它适用于 SQL 分析、修改、生成和验证等场景。通过 Maven 或 Gradle 安装后,可以方便地在 Java 代码中使用。
3996 11
|
SQL Java 数据库连接
【潜意识Java】MyBatis中的动态SQL灵活、高效的数据库查询以及深度总结
本文详细介绍了MyBatis中的动态SQL功能,涵盖其背景、应用场景及实现方式。
1488 6
|
10月前
|
SQL XML Java
六、MyBatis特殊的SQL:模糊查询、动态设置表名、校验名称唯一性
六、MyBatis特殊的SQL:模糊查询、动态设置表名、校验名称唯一性
307 0
|
SQL NoSQL Java
Java使用sql查询mongodb
通过使用 MongoDB Connector for BI 和 JDBC,开发者可以在 Java 中使用 SQL 语法查询 MongoDB 数据库。这种方法对于熟悉 SQL 的团队非常有帮助,能够快速实现对 MongoDB 数据的操作。同时,也需要注意到这种方法的性能和功能限制,根据具体应用场景进行选择和优化。
509 9
|
SQL Java
使用java在未知表字段情况下通过sql查询信息
使用java在未知表字段情况下通过sql查询信息
174 8
|
SQL 分布式计算 Java
Hadoop-11-MapReduce JOIN 操作的Java实现 Driver Mapper Reducer具体实现逻辑 模拟SQL进行联表操作
Hadoop-11-MapReduce JOIN 操作的Java实现 Driver Mapper Reducer具体实现逻辑 模拟SQL进行联表操作
252 3
|
SQL Java
使用java在未知表字段情况下通过sql查询信息
使用java在未知表字段情况下通过sql查询信息
143 1
|
SQL 存储 数据处理
"SQL触发器实战大揭秘:一键解锁数据自动化校验与更新魔法,让数据库管理从此告别繁琐,精准高效不再是梦!"
【8月更文挑战第31天】在数据库管理中,确保数据准确性和一致性至关重要。SQL触发器能自动执行数据校验与更新,显著提升工作效率。本文通过一个员工信息表的例子,详细介绍了如何利用触发器自动设定和校验薪资,确保其符合业务规则。提供的示例代码展示了在插入新记录时如何自动检查并调整薪资,以满足最低标准。这不仅减轻了数据库管理员的负担,还提高了数据处理的准确性和效率。触发器虽强大,但也需谨慎使用,以避免复杂性和性能问题。
285 1