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

本文涉及的产品
云数据库 Tair(兼容Redis),内存型 2GB
云数据库 RDS MySQL,集群系列 2核4GB
推荐场景:
搭建个人博客
Redis 开源版,标准版 2GB
推荐场景:
搭建游戏排行榜
简介: 本文介绍了五种校验 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

相关文章
|
10天前
|
SQL NoSQL Java
Java使用sql查询mongodb
通过MongoDB Atlas Data Lake或Apache Drill,可以在Java中使用SQL语法查询MongoDB数据。这两种方法都需要适当的配置和依赖库的支持。希望本文提供的示例和说明能够帮助开发者实现这一目标。
36 17
|
12天前
|
SQL Java 数据库连接
【潜意识Java】MyBatis中的动态SQL灵活、高效的数据库查询以及深度总结
本文详细介绍了MyBatis中的动态SQL功能,涵盖其背景、应用场景及实现方式。
62 6
|
18天前
|
SQL Java 数据库连接
如何在 Java 代码中使用 JSqlParser 解析复杂的 SQL 语句?
大家好,我是 V 哥。JSqlParser 是一个用于解析 SQL 语句的 Java 库,可将 SQL 解析为 Java 对象树,支持多种 SQL 类型(如 `SELECT`、`INSERT` 等)。它适用于 SQL 分析、修改、生成和验证等场景。通过 Maven 或 Gradle 安装后,可以方便地在 Java 代码中使用。
138 11
|
2月前
|
SQL NoSQL Java
Java使用sql查询mongodb
通过使用 MongoDB Connector for BI 和 JDBC,开发者可以在 Java 中使用 SQL 语法查询 MongoDB 数据库。这种方法对于熟悉 SQL 的团队非常有帮助,能够快速实现对 MongoDB 数据的操作。同时,也需要注意到这种方法的性能和功能限制,根据具体应用场景进行选择和优化。
93 9
|
3月前
|
SQL Java
使用java在未知表字段情况下通过sql查询信息
使用java在未知表字段情况下通过sql查询信息
48 8
|
4月前
|
SQL 分布式计算 Java
Hadoop-11-MapReduce JOIN 操作的Java实现 Driver Mapper Reducer具体实现逻辑 模拟SQL进行联表操作
Hadoop-11-MapReduce JOIN 操作的Java实现 Driver Mapper Reducer具体实现逻辑 模拟SQL进行联表操作
65 3
|
5月前
|
SQL Java
使用java在未知表字段情况下通过sql查询信息
使用java在未知表字段情况下通过sql查询信息
45 1
|
6月前
|
SQL Java 数据库连接
【Azure 应用服务】Java ODBC代码中,启用 Managed Identity 登录 SQL Server 报错 Managed Identity authentication is not available
【Azure 应用服务】Java ODBC代码中,启用 Managed Identity 登录 SQL Server 报错 Managed Identity authentication is not available
|
8月前
|
SQL 安全 Java
java的SQL注入与XSS攻击
java的SQL注入与XSS攻击
156 2
|
8月前
|
Java 数据安全/隐私保护
java密码强度校验
java密码强度校验
163 1