From f59681e9cf7faedb6c28e23920c58ecb919be547 Mon Sep 17 00:00:00 2001
From: ZY <zc857179121@qq.com>
Date: 星期一, 28 十月 2024 10:14:29 +0800
Subject: [PATCH] sql注入漏洞
---
src/main/java/com/zy/common/config/CoolExceptionHandler.java | 8 +++++++-
1 files changed, 7 insertions(+), 1 deletions(-)
diff --git a/src/main/java/com/zy/common/config/CoolExceptionHandler.java b/src/main/java/com/zy/common/config/CoolExceptionHandler.java
index 9a6fe2d..b3968a3 100644
--- a/src/main/java/com/zy/common/config/CoolExceptionHandler.java
+++ b/src/main/java/com/zy/common/config/CoolExceptionHandler.java
@@ -26,7 +26,13 @@
@ExceptionHandler(CoolException.class)
public R handleRRException(CoolException e) {
- return R.parse(e.getMessage());
+ String[] split = e.getMessage().split("-");
+ if (split.length == 2) {
+ if (String.valueOf(split[0]).length() < 3) {
+ return R.parse(e.getMessage());
+ }
+ }
+ return R.error(e.getMessage());
}
}
--
Gitblit v1.9.1