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

pull/129/head
RuoYi 3 years ago
parent 1797831afc
commit 054fd7546f

@ -10,6 +10,11 @@ import com.ruoyi.common.core.utils.StringUtils;
*/ */
public class SqlUtil public class SqlUtil
{ {
/**
* sql
*/
public static String SQL_REGEX = "select |insert |delete |update |drop |count |exec |chr |mid |master |truncate |char |and |declare ";
/** /**
* 线 * 线
*/ */
@ -34,4 +39,23 @@ public class SqlUtil
{ {
return value.matches(SQL_PATTERN); 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注入风险");
}
}
}
} }

Loading…
Cancel
Save