From 495253345194da937f134cfb2adb9eaf00c413d5 Mon Sep 17 00:00:00 2001 From: 13693261870 <252740454@qq.com> Date: 星期日, 08 十月 2023 14:26:10 +0800 Subject: [PATCH] 修改令牌的IP检查功能 --- src/main/java/com/moon/server/service/sys/ProxyService.java | 12 +++++++++++- 1 files changed, 11 insertions(+), 1 deletions(-) diff --git a/src/main/java/com/moon/server/service/sys/ProxyService.java b/src/main/java/com/moon/server/service/sys/ProxyService.java index 2d097d8..a64b1fa 100644 --- a/src/main/java/com/moon/server/service/sys/ProxyService.java +++ b/src/main/java/com/moon/server/service/sys/ProxyService.java @@ -4,6 +4,7 @@ import com.moon.server.entity.all.*; import com.moon.server.entity.sys.ResEntity; import com.moon.server.entity.sys.ResLogEntity; +import com.moon.server.entity.sys.TokenEntity; import com.moon.server.entity.sys.UserEntity; import com.moon.server.helper.AsyncHelper; import com.moon.server.helper.HttpHelper; @@ -108,7 +109,7 @@ } // 7.妫�鏌ョ櫧鍚嶅崟鍜孖P涓�鑷存�� if (!checkWhiteList(ip, req)) { - if (!ip.equals(sysService.tokenService.getEntityByToken(token).getIp())) { + if (!checkIpSource(ip, token)) { return WebHelper.writeStr2Page(res, AuthInterceptor.ILLEGAL_TOKEN); } } @@ -145,6 +146,15 @@ } /** + * 妫�鏌P涓�鑷存�э細鍥哄畾浠ょ墝涓嶆鏌� + */ + private boolean checkIpSource(String ip, String token) { + TokenEntity te = sysService.tokenService.getEntityByToken(token); + + return StaticData.I1 == te.getType() || te.getIp().equals(ip); + } + + /** * 妫�鏌ヨ祫婧愭潈闄� */ private ResEntity getResEntity(UserEntity ue, int resId) { -- Gitblit v1.9.3