admin/wisdomisitezw_java
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

文件上传类型限制

Browse Source
This commit is contained in:
pengjie 2023-12-07 16:24:06 +08:00
parent 6d63bc901e
commit d52090d402
1 changed files with 19 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
src/main/java/com/zhgd/xmgl/modules/basicdata/controller/admin/FileController.java
View File

@ -16,6 +16,7 @@ import org.springframework.web.multipart.MultipartFile;
import springfox.documentation.annotations.ApiIgnore; import springfox.documentation.annotations.ApiIgnore;
import javax.servlet.http.HttpServletResponse; import javax.servlet.http.HttpServletResponse;
import java.util.Locale;
import java.util.Map; import java.util.Map;
/** /**
@ -56,6 +57,24 @@ public class FileController {
@ApiOperation(value = "文件上传", notes = "文件上传", httpMethod="POST") @ApiOperation(value = "文件上传", notes = "文件上传", httpMethod="POST")
@PostMapping("/upload") @PostMapping("/upload")
public Result<Map<String, Object>> upload(MultipartFile file) { public Result<Map<String, Object>> upload(MultipartFile file) {
String[] FILE_SUFFIX_SUPPORT ={".jsp",".php",".asp",".aspx",".exe"};
// 校验文件是否为空
if (file == null) {
throw new RuntimeException("文件不能为空！");
}
//得到文件名
String originalFilename = file.getOriginalFilename();
// 校验文件后缀
if (!originalFilename.contains(".")) {
throw new RuntimeException("文件不能没有后缀！");
}
String suffix = originalFilename.substring(originalFilename.lastIndexOf('.'));
for (String s : FILE_SUFFIX_SUPPORT) {
//转换为小写比较，Locale.ROOT为区域转换规则可不写
if (s.equals(suffix.toLowerCase(Locale.ROOT))) {
throw new RuntimeException("请上传正常的文件!");
}
}
return Result.success(fileDetailService.upload(file)); return Result.success(fileDetailService.upload(file));
} }