Ver código fonte

添加 上传前文件名校验

wuweihao 5 anos atrás
pai
commit
acb7fd5cde

+ 5 - 0
src/main/java/com/fd/controller/FdModelController.java

@@ -129,6 +129,11 @@ public class FdModelController {
             return new R(51005, MsgCode.E51005);
         }
 
+        String s = StringUtils.substringAfterLast(fileName, ".");
+        if (!"zip".equals(s)) {
+            return new R(50007, MsgCode.E50007);
+        }
+
         List<FileEntity> list = modelServer.findByFileName(fileName);
         if (list.size() > 0) {
             return new R(51006, MsgCode.E51006);

+ 5 - 0
src/main/java/com/fd/controller/RasterController.java

@@ -87,6 +87,11 @@ public class RasterController {
             return new R(51005, MsgCode.E51005);
         }
 
+        String s = StringUtils.substringAfterLast(fileName, ".");
+        if (!"tif".equals(s)) {
+            return new R(50008,MsgCode.E50008);
+        }
+
         List<FileEntity> list = rasterServer.findByFileName(fileName);
         if (list.size() > 0) {
             return new R(51006, MsgCode.E51006);