From ab849f796bdc17236a95ea5fe5c166fb8f24a75c Mon Sep 17 00:00:00 2001
From: sws <15810472099@163.com>
Date: 星期六, 26 十一月 2022 16:12:02 +0800
Subject: [PATCH] 1
---
src/main/java/com/lf/server/interceptor/AuthInterceptor.java | 172 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++
1 files changed, 172 insertions(+), 0 deletions(-)
diff --git a/src/main/java/com/lf/server/interceptor/AuthInterceptor.java b/src/main/java/com/lf/server/interceptor/AuthInterceptor.java
new file mode 100644
index 0000000..6fc779c
--- /dev/null
+++ b/src/main/java/com/lf/server/interceptor/AuthInterceptor.java
@@ -0,0 +1,172 @@
+package com.lf.server.interceptor;
+
+import com.alibaba.fastjson.JSON;
+import com.lf.server.entity.all.*;
+import com.lf.server.entity.sys.TokenEntity;
+import com.lf.server.entity.sys.UserEntity;
+import com.lf.server.helper.StringHelper;
+import com.lf.server.helper.WebHelper;
+import com.lf.server.service.all.SysService;
+import org.apache.commons.logging.Log;
+import org.apache.commons.logging.LogFactory;
+import org.springframework.context.annotation.Configuration;
+import org.springframework.web.method.HandlerMethod;
+import org.springframework.web.servlet.HandlerInterceptor;
+
+import javax.servlet.http.HttpServletRequest;
+import javax.servlet.http.HttpServletResponse;
+import java.util.List;
+
+/**
+ * 韬唤璁よ瘉鎷︽埅鍣�
+ * @author WWW
+ */
+@Configuration
+public class AuthInterceptor implements HandlerInterceptor {
+ private SysService sysService;
+
+ private static final Log log = LogFactory.getLog(AuthInterceptor.class);
+
+ private static final String NO_TOKEN = JSON.toJSONString(new ResponseMsg<String>(HttpStatus.TOKEN_ERROR, "鎵句笉鍒颁护鐗�"));
+
+ private static final String NO_LOGIN = JSON.toJSONString(new ResponseMsg<String>(HttpStatus.NO_LOGIN_ERROR, "鐢ㄦ埛鏈櫥褰�"));
+
+ private static final String USER_LOCK = JSON.toJSONString(new ResponseMsg<String>(HttpStatus.USER_LOCK_ERROR, "鐢ㄦ埛ID宸茬鐢�"));
+
+ private static final String NO_AUTH = JSON.toJSONString(new ResponseMsg<String>(HttpStatus.NO_AUTH_ERROR, "鏃犳潈闄愯闂�"));
+
+ private static final String IP_NULL = JSON.toJSONString(new ResponseMsg<String>(HttpStatus.UNAUTHORIZED, "IP鍦板潃涓虹┖"));
+
+ private static final String BLACK_LIST = JSON.toJSONString(new ResponseMsg<String>(HttpStatus.UNAUTHORIZED, "璇P鍏ラ粦鍚嶅崟"));
+
+ private static final String ILLEGAL_TOKEN = JSON.toJSONString(new ResponseMsg<String>(HttpStatus.UNAUTHORIZED, "浠ょ墝鏉ユ簮闈炴硶"));
+
+ public AuthInterceptor(SysService sysService) {
+ this.sysService = sysService;
+ }
+
+ /**
+ * Controller鎵ц涔嬪墠鎵ц锛屽鏋滆繑鍥炲�兼槸true鍒欎唬琛ㄦ斁琛岋紝杩斿洖false鍒欐嫤鎴�
+ */
+ @Override
+ public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) {
+ // noinspection AlibabaRemoveCommentedCode
+ try {
+ // 闈炴柟娉� 鎴� 鏃犻渶鎺堟潈锛屾棤闇�鎷︽埅
+ if (!(handler instanceof HandlerMethod) || noNeedAuth(request)) {
+ return true;
+ }
+
+ // 鑾峰彇浠ょ墝
+ String token = WebHelper.getToken(request);
+ if (StringHelper.isNull(token)) {
+ return WebHelper.write2Page(response, NO_TOKEN);
+ }
+
+ // 鑾峰彇IP锛屾鏌ラ粦鍚嶅崟锛屾鏌ョ櫧鍚嶅崟锛屾鏌P涓�鑷存��
+ String ip = WebHelper.getIpAddress(request);
+ if (StringHelper.isEmpty(ip)) {
+ return WebHelper.write2Page(response, IP_NULL);
+ }
+ if (!checkBlackList(ip, request)) {
+ return WebHelper.write2Page(response, BLACK_LIST);
+ }
+ if (!checkWhiteList(ip, request)) {
+ if (!checkIpSource(ip, token)) {
+ return WebHelper.write2Page(response, ILLEGAL_TOKEN);
+ }
+ }
+
+ // 鑾峰彇鐢ㄦ埛浠ょ墝
+ UserEntity ue = sysService.tokenService.getCurrentUser(request);
+ if (ue == null) {
+ return WebHelper.write2Page(response, NO_LOGIN);
+ }
+
+ // 妫�鏌ョ敤鎴稩D鏄惁绂佺敤
+ if (sysService.tokenService.isUidDisable(ue)) {
+ return WebHelper.write2Page(response, USER_LOCK);
+ }
+
+ // 鏉冮檺鏍￠獙
+ if (!checkPerms(ue, request)) {
+ System.out.println("鏃犳潈闄愯闂細" + request.getRequestURI().toLowerCase());
+ return WebHelper.write2Page(response, NO_AUTH);
+ }
+
+ return true;
+ } catch (Exception ex) {
+ log.error(ex.getMessage(), ex);
+ return false;
+ }
+ }
+
+ /**
+ * 鏃犻渶鎺堟潈
+ */
+ private static boolean noNeedAuth(HttpServletRequest request) {
+ String uri = request.getRequestURI().toLowerCase();
+ for (String page : StaticData.EXCLUDE_PATH) {
+ if (uri.contains(page)) {
+ return true;
+ }
+ }
+
+ return false;
+ }
+
+ /**
+ * 妫�鏌ユ潈闄�
+ */
+ private boolean checkPerms(UserEntity ue, HttpServletRequest request) {
+ List<String> list = sysService.permsService.selectPerms(ue.getUid());
+ if (list == null || list.size() == 0) {
+ return false;
+ }
+
+ String url = request.getRequestURI();
+ for (String perm : list) {
+ if (url.startsWith(perm)) {
+ return true;
+ }
+ }
+
+ return false;
+ }
+
+ /**
+ * 妫�鏌ラ粦鍚嶅崟
+ */
+ private boolean checkBlackList(String ip, HttpServletRequest request) {
+ List<String> blackList = sysService.blacklistService.selectIpList(1);
+ if (blackList == null || blackList.isEmpty()) {
+ return true;
+ }
+ if (blackList.contains(ip)) {
+ return false;
+ }
+
+ return true;
+ }
+
+ /**
+ * 妫�鏌ョ櫧鍚嶅崟
+ */
+ private boolean checkWhiteList(String ip, HttpServletRequest request) {
+ List<String> whiteList = sysService.blacklistService.selectIpList(2);
+ if (whiteList == null || whiteList.isEmpty()) {
+ return true;
+ }
+
+ return whiteList.contains(ip);
+ }
+
+ /**
+ * 妫�鏌P涓�鑷存��
+ */
+ private boolean checkIpSource(String ip, String token) {
+ TokenEntity te = sysService.tokenService.getEntityByToken(token);
+
+ return te.getIp().equals(ip);
+ }
+}
--
Gitblit v1.9.3