From 50d3d5e3e3282f757ea639f9ca1939d429c6fd5d Mon Sep 17 00:00:00 2001
From: Helius <wangdoubleone@gmail.com>
Date: Mon, 30 Nov 2020 11:02:07 +0800
Subject: [PATCH] modify

---
 src/main/java/com/xcong/excoin/configurations/security/TokenFilter.java |   82 +++++++++++++++++++++++++++++++++++-----
 1 files changed, 71 insertions(+), 11 deletions(-)

diff --git a/src/main/java/com/xcong/excoin/configurations/security/TokenFilter.java b/src/main/java/com/xcong/excoin/configurations/security/TokenFilter.java
index 6ebf4d8..e98bac2 100644
--- a/src/main/java/com/xcong/excoin/configurations/security/TokenFilter.java
+++ b/src/main/java/com/xcong/excoin/configurations/security/TokenFilter.java
@@ -4,10 +4,15 @@
 import cn.hutool.crypto.asymmetric.KeyType;
 import cn.hutool.crypto.asymmetric.RSA;
 import com.alibaba.fastjson.JSONObject;
+import com.fasterxml.jackson.databind.ObjectMapper;
+import com.xcong.excoin.common.LoginUserUtils;
 import com.xcong.excoin.common.contants.AppContants;
+import com.xcong.excoin.common.exception.GlobalException;
+import com.xcong.excoin.common.response.Result;
 import com.xcong.excoin.common.system.bean.LoginUserBean;
 import com.xcong.excoin.configurations.properties.ApplicationProperties;
 import com.xcong.excoin.configurations.properties.SecurityProperties;
+import com.xcong.excoin.modules.member.entity.MemberEntity;
 import com.xcong.excoin.utils.RedisUtils;
 import com.xcong.excoin.utils.SpringContextHolder;
 import lombok.extern.slf4j.Slf4j;
@@ -22,6 +27,7 @@
 import javax.servlet.ServletRequest;
 import javax.servlet.ServletResponse;
 import javax.servlet.http.HttpServletRequest;
+import javax.servlet.http.HttpServletResponse;
 import java.io.IOException;
 import java.util.ArrayList;
 
@@ -41,22 +47,72 @@
     @Override
     public void doFilter(ServletRequest servletRequest, ServletResponse servletResponse, FilterChain filterChain) throws IOException, ServletException {
         HttpServletRequest request = (HttpServletRequest) servletRequest;
-        String token = resolveToken(request);
-        if (StrUtil.isNotBlank(token)) {
-            String loginStr = (String) redisUtils.get(AppContants.APP_LOGIN_PREFIX + token);
-            if (StrUtil.isNotBlank(loginStr)) {
-                LoginUserBean loginUser = JSONObject.parseObject(loginStr, LoginUserBean.class);
-                Authentication authentication = new UsernamePasswordAuthenticationToken(loginUser.getMemberEntity(), token, new ArrayList<>());
-                SecurityContextHolder.getContext().setAuthentication(authentication);
-                redisUtils.expire(AppContants.APP_LOGIN_PREFIX + token, 300000);
+        HttpServletResponse response = (HttpServletResponse) servletResponse;
+        String token = resolveTokenOutCenter(request);
+
+        if (!AppContants.TIME_OUT.equals(token)) {
+            if (StrUtil.isNotBlank(token)) {
+                String redisKey = "";
+                // 根据user-agent判断pc端还是app端
+                if (LoginUserUtils.isBrowser(request)) {
+                    redisKey = token;;
+                } else {
+                    redisKey = token;;
+                }
+
+                String loginStr = (String) redisUtils.get(redisKey);
+                if (StrUtil.isNotBlank(loginStr)) {
+                    MemberEntity loginUser = JSONObject.parseObject(loginStr, MemberEntity.class);
+                    Authentication authentication = new UsernamePasswordAuthenticationToken(loginUser, token, new ArrayList<>());
+                    SecurityContextHolder.getContext().setAuthentication(authentication);
+                } else {
+                    log.info("token无法查询:{}", token);
+                    SecurityContextHolder.clearContext();
+                }
             } else {
+//                log.info("token为空:{}", request.getRequestURI());
                 SecurityContextHolder.clearContext();
             }
         } else {
+            response.setHeader("TimeOut", AppContants.TIME_OUT);
             SecurityContextHolder.clearContext();
         }
-
         filterChain.doFilter(servletRequest, servletResponse);
+    }
+    
+    /**
+     * 解析前端传来的token,md5加密后的地址_设备iD_/api
+     *
+     * @param request
+     * @return
+     */
+    private String resolveTokenOutCenter(HttpServletRequest request) {
+        try {
+            String bearerToken = request.getHeader(AppContants.TOKEN_HEADER);
+            //获取请求的完整路径
+            StringBuffer requestURL = request.getRequestURL();
+            
+            if (StringUtils.hasText(bearerToken) && bearerToken.startsWith(AppContants.TOKEN_START_WITH)) {
+                // 去掉令牌前缀
+                String rsaToken = bearerToken.replace(AppContants.TOKEN_START_WITH, "");
+                RSA rsa = new RSA(securityProperties.getPrivateKey(), null);
+                String decryptStr = rsa.decryptStr(rsaToken, KeyType.PrivateKey);
+                String[] tokens = StrUtil.split(decryptStr, "_");
+
+                if (StrUtil.isNotEmpty(requestURL) && requestURL.toString().contains(tokens[2])) {
+                    return tokens[0];
+                } else {
+//                    log.info("前面token为{}", tokens[0]);
+//                    log.info("时间为:{}, 当前时间为:{}", tokens[1], System.currentTimeMillis());
+                    return AppContants.TIME_OUT;
+                }
+            }
+//            log.info("bearerToken---->{}", bearerToken);
+        } catch (Exception e) {
+            log.error("#解析token异常#", e);
+            return null;
+        }
+        return null;
     }
 
     /**
@@ -76,9 +132,13 @@
 
                 if (verifyTokenExpired(Long.parseLong(tokens[1]))) {
                     return tokens[0];
+                } else {
+//                    log.info("前面token为{}", tokens[0]);
+//                    log.info("时间为:{}, 当前时间为:{}", tokens[1], System.currentTimeMillis());
+                    return AppContants.TIME_OUT;
                 }
-                return null;
             }
+//            log.info("bearerToken---->{}", bearerToken);
         } catch (Exception e) {
             log.error("#解析token异常#", e);
             return null;
@@ -90,7 +150,7 @@
         boolean isDebug = applicationProperties.isDebug();
         if (!isDebug) {
             long currentTime = System.currentTimeMillis();
-            return currentTime - time <= 5000;
+            return currentTime - time <= 30000;
         }
         return true;
     }

--
Gitblit v1.9.1