From 121031371e3556d9c2e6536ca4fea83f1c09f2f0 Mon Sep 17 00:00:00 2001
From: 李宇 <986321569@qq.com>
Date: 星期三, 03 二月 2021 09:21:41 +0800
Subject: [PATCH] 修改refer限制

---
 src/main/java/org/springframework/web/filter/DelegatingFilterProxy.java |    6 ++++--
 1 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/src/main/java/org/springframework/web/filter/DelegatingFilterProxy.java b/src/main/java/org/springframework/web/filter/DelegatingFilterProxy.java
index 6c74c45..c5a8733 100644
--- a/src/main/java/org/springframework/web/filter/DelegatingFilterProxy.java
+++ b/src/main/java/org/springframework/web/filter/DelegatingFilterProxy.java
@@ -241,9 +241,11 @@
             throws ServletException, IOException {
         String referer = ((HttpServletRequest)request).getHeader("Referer");
         StringBuffer stringBuffer = new StringBuffer();
-        stringBuffer.append(request.getScheme()).append("://").append(request.getServerName());
+        stringBuffer.append("http://").append(request.getServerName());
+        StringBuffer stringBuffer1 = new StringBuffer();
+        stringBuffer1.append("https://").append(request.getServerName());
         if(referer != null && !referer.equals("") ){
-            if(referer.lastIndexOf(String.valueOf(stringBuffer)) != 0){
+            if(referer.lastIndexOf(String.valueOf(stringBuffer)) == 0 && referer.lastIndexOf(String.valueOf(stringBuffer1)) == 0){
                 HttpServletResponse httpResp1 = (HttpServletResponse)response;
                 httpResp1.setStatus(500);
                 httpResp1.setContentType("text/html; charset=utf-8");

--
Gitblit v1.9.2