| | |
|---|
| | | |
|---|
| | | import com.ruoyi.common.exception.UtilException; |
|---|
| | | import com.ruoyi.common.utils.StringUtils; |
|---|
| | | import lombok.AccessLevel; |
|---|
| | | import lombok.NoArgsConstructor; |
|---|
| | | |
|---|
| | | /** |
|---|
| | | * sql操作工具类 |
|---|
| | | * |
|---|
| | | * @author ruoyi |
|---|
| | | */ |
|---|
| | | @NoArgsConstructor(access = AccessLevel.PRIVATE) |
|---|
| | | 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_\\ \\,\\.]+"; |
|---|
| | | public static final String SQL_PATTERN = "[a-zA-Z0-9_\\ \\,\\.]+"; |
|---|
| | | |
|---|
| | | /** |
|---|
| | | * 检查字符，防止注入绕过 |
|---|
| | |
|---|
| | | 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 (StringUtils.indexOfIgnoreCase(value, sqlKeyword) > -1) { |
|---|
| | | throw new UtilException("参数存在SQL注入风险"); |
|---|
| | | } |
|---|
| | | } |
|---|
| | | } |
|---|
| | | } |
|---|
