From 278e3205503af70148a989e9a38bde38836389d3 Mon Sep 17 00:00:00 2001
From: zhaojun <873019219@qq.com>
Date: Sat, 26 Nov 2022 17:54:43 +0800
Subject: [PATCH] =?UTF-8?q?:bug:=20=E4=BF=AE=E5=A4=8D=20Referer=20?=
 =?UTF-8?q?=E9=98=B2=E7=9B=97=E9=93=BE=E4=B8=8D=E7=94=9F=E6=95=88=E7=9A=84?=
 =?UTF-8?q?=20bug?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../im/zhaojun/zfile/module/link/service/ShortLinkService.java  | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/src/main/java/im/zhaojun/zfile/module/link/service/ShortLinkService.java b/src/main/java/im/zhaojun/zfile/module/link/service/ShortLinkService.java
index f9301e6..cb68501 100644
--- a/src/main/java/im/zhaojun/zfile/module/link/service/ShortLinkService.java
+++ b/src/main/java/im/zhaojun/zfile/module/link/service/ShortLinkService.java
@@ -8,6 +8,7 @@ import cn.hutool.core.util.RandomUtil;
 import cn.hutool.core.util.StrUtil;
 import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
 import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
+import im.zhaojun.zfile.module.storage.annotation.RefererCheck;
 import im.zhaojun.zfile.module.storage.context.StorageSourceContext;
 import im.zhaojun.zfile.core.exception.file.InvalidStorageSourceException;
 import im.zhaojun.zfile.core.exception.file.operator.StorageSourceFileOperatorException;
@@ -146,6 +147,7 @@ public class ShortLinkService {
      *
      * @throws  IOException 可能抛出的 IO 异常
      */
+    @RefererCheck
     public void handlerDownload(String storageKey, String filePath, String shortKey) throws IOException {
         HttpServletRequest request = RequestHolder.getRequest();
         HttpServletResponse response = RequestHolder.getResponse();