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