SQL工具类新增检查关键字方法

RuoYi

2021-12-21 ec0df2c107fc94a6d40939a1990c82892965541a

SQL工具类新增检查关键字方法

 src/main/java/com/ruoyi/common/utils/sql/SqlUtil.java

@@ -11,6 +11,11 @@
public class SqlUtil

{

    /**

     * 定义常用的 sql关键字

     */

    public static String SQL_REGEX = "select |insert |delete |update |drop |count |exec |chr |mid |master |truncate |char |and |declare ";



    /**

     * 仅支持字母、数字、下划线、空格、逗号、小数点（支持多个字段排序）

     */

    public static String SQL_PATTERN = "[a-zA-Z0-9_\\ \\,\\.]+";

@@ -34,4 +39,23 @@
    {

        return value.matches(SQL_PATTERN);

    }



    /**

     * SQL关键字检查

     */

    public static void filterKeyword(String value)

    {

        if (StringUtils.isEmpty(value))

        {

            return;

        }

        String[] sqlKeywords = StringUtils.split(SQL_REGEX, "\\|");

        for (int i = 0; i < sqlKeywords.length; i++)

        {

            if (StringUtils.indexOfIgnoreCase(value, sqlKeywords[i]) > -1)

            {

                throw new UtilException("参数存在SQL注入风险");

            }

        }

    }

}