From 7c729a72e24227708a99c24aabf7c93429482685 Mon Sep 17 00:00:00 2001 From: zhaojun1998 Date: Tue, 7 Jan 2020 22:57:59 +0800 Subject: [PATCH] =?UTF-8?q?:bug:=20=E6=9C=AC=E5=9C=B0=E5=AD=98=E5=82=A8,?= =?UTF-8?q?=20=E8=8E=B7=E5=8F=96=E6=96=87=E4=BB=B6=E4=B8=8D=E5=AD=98?= =?UTF-8?q?=E5=9C=A8=E8=BF=94=E5=9B=9E=E7=8A=B6=E6=80=81=E7=A0=81=20404?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../java/im/zhaojun/local/controller/LocalController.java | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/src/main/java/im/zhaojun/local/controller/LocalController.java b/src/main/java/im/zhaojun/local/controller/LocalController.java index 49eed33..ffda9b0 100644 --- a/src/main/java/im/zhaojun/local/controller/LocalController.java +++ b/src/main/java/im/zhaojun/local/controller/LocalController.java @@ -32,7 +32,7 @@ public class LocalController { public ResponseEntity downAttachment(final HttpServletRequest request) { String path = (String) request.getAttribute( HandlerMapping.PATH_WITHIN_HANDLER_MAPPING_ATTRIBUTE); - String bestMatchPattern = (String ) request.getAttribute(HandlerMapping.BEST_MATCHING_PATTERN_ATTRIBUTE); + String bestMatchPattern = (String) request.getAttribute(HandlerMapping.BEST_MATCHING_PATTERN_ATTRIBUTE); AntPathMatcher apm = new AntPathMatcher(); String filePath = apm.extractPathWithinPattern(bestMatchPattern, path); @@ -41,6 +41,11 @@ public class LocalController { private ResponseEntity export(File file) { + if (!file.exists()) { + return ResponseEntity.notFound().build(); + } + + MediaType mediaType = MediaType.APPLICATION_OCTET_STREAM; HttpHeaders headers = new HttpHeaders();