2.7.18

shikeying

2024-01-11 3b67e947e36133e2a40eb2737b15ea375e157ea0

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
package com.walker.jdbc.util;
 
import com.walker.infrastructure.ApplicationRuntimeException;
 
/**
 * sql操作工具类
 *
 * @author ruoyi
 */
public class SqlUtils {
    /**
     * 定义常用的 sql关键字
     */
    public static final String SQL_REGEX = "select |insert |delete |update |drop |count |exec |chr |mid |master |truncate |char |and |declare ";
 
    /**
     * 仅支持字母、数字、下划线、空格、逗号、小数点（支持多个字段排序）
     */
    public static final String SQL_PATTERN = "[a-zA-Z0-9_\\ \\,\\.]+";
 
    /**
     * 把原始sql变成oracle分页语句。
     * @param sql
     * @return
     * @date 2023-03-03
     */
    public static final String getOraclePageSql(String sql){
        return new StringBuilder(PAGE_PREFIX_ORACLE).append(sql).append(PAGE_SUFFIX_ORACLE).toString();
    }
 
    public static final String PAGE_PREFIX_ORACLE = "select * from (select row_.*, rownum rownum_ from (";
    public static final String PAGE_SUFFIX_ORACLE = ") row_ where rownum <= ?) where rownum_ >= ?";
 
    /**
     * 检查字符，防止注入绕过
     */
    public static String escapeOrderBySql(String value) {
        if (StringUtils.isNotEmpty(value) && !isValidOrderBySql(value)) {
            throw new ApplicationRuntimeException("参数不符合规范，不能进行查询");
        }
        return value;
    }
 
    /**
     * 验证 order by 语法是否符合规范
     */
    public static boolean isValidOrderBySql(String value)
    {
        return value.matches(SQL_PATTERN);
    }
 
    /**
     * SQL关键字检查
     */
    public static void filterKeyword(String value) {
        if (StringUtils.isEmpty(value)) {
            return;
        }
        String[] sqlKeywords = StringUtils.split(SQL_REGEX, "\\|");
        for (String sqlKeyword : sqlKeywords) {
            if (org.apache.commons.lang3.StringUtils.indexOfIgnoreCase(value, sqlKeyword) > -1) {
                throw new ApplicationRuntimeException("参数存在SQL注入风险");
            }
        }
    }
}