完善后端异常处理，即使cmd运行但是没有数据写入和修改依旧显示未成功。

fuzzbackend/README.md

+## ferofuzz

fuzzbackend/pom.xml

@@ -101,6 +101,14 @@
                     </execution>
                 </executions>
             </plugin>
+            <plugin>
+                <groupId>org.apache.maven.plugins</groupId>
+                <artifactId>maven-compiler-plugin</artifactId>
+                <configuration>
+                    <source>7</source>
+                    <target>7</target>
+                </configuration>
+            </plugin>
 
 
         </plugins>

fuzzbackend/src/main/java/com/example/fuzzControll/conf/kittyProperties.java → fuzzbackend/src/main/java/com/example/fuzzControll/conf/KittyProperties.java

@@ -5,7 +5,7 @@ import org.springframework.stereotype.Component;
 
 @Component("kittyProperties")
 @ConfigurationProperties(prefix = "kitty")
-public class kittyProperties {
+public class KittyProperties {
     String path;
     String venvPath;
     String methodPath;

fuzzbackend/src/main/java/com/example/fuzzControll/conf/seedProperties.java → fuzzbackend/src/main/java/com/example/fuzzControll/conf/SeedProperties.java

@@ -5,7 +5,7 @@ import org.springframework.stereotype.Component;
 
 @Component("seedProperties")
 @ConfigurationProperties(prefix = "filepath")
-public class seedProperties {
+public class SeedProperties {
     String seedPath;
 
     public String getSeedPath() {

fuzzbackend/src/main/java/com/example/fuzzControll/constents/cmdConstent.java → fuzzbackend/src/main/java/com/example/fuzzControll/constents/CmdConstent.java

 package com.example.fuzzControll.constents;
 
 import com.example.fuzzControll.conf.SpringContextUtil;
-import com.example.fuzzControll.conf.kittyProperties;
+import com.example.fuzzControll.conf.KittyProperties;
 
-public class cmdConstent {
-    static kittyProperties kittyProperties = (kittyProperties)SpringContextUtil.getBean("kittyProperties");
+public class CmdConstent {
+    static KittyProperties kittyProperties = (KittyProperties)SpringContextUtil.getBean("kittyProperties");
     public static final String GET_FILE_NAME = "ls -h ";
     public static final String DELETE_FILE = "sudo rm -r ";
+    public static final String COUNT_FILE = "ls -l | grep \"^-\" | wc -l";
+    public static final String COUNT_DIR = "ls -l | grep \"^d\" | wc -l";
     public static final String RUN_AFLNET = "afl-fuzz -d -i "+kittyProperties.getAflnetPath()+"aflnet/tutorials/live555/in-rtsp -o out-live8 " +
             "-x "+kittyProperties.getAflnetPath()+"aflnet/tutorials/live555/rtsp.dict ";
     public static final String RUN_PING = "ping www.baidu.com";

fuzzbackend/src/main/java/com/example/fuzzControll/constents/mutationConstent.java → fuzzbackend/src/main/java/com/example/fuzzControll/constents/MutationConstent.java

 package com.example.fuzzControll.constents;
 
-public class mutationConstent {
+public class MutationConstent {
     public static final String TEST_GRANULARITY_BIT_BYTE = "test_granularity_bit_byte.py ";
     public static final String TEST_MUTATED_LIBS = "test_mutated_libs.py ";
     public static final String TEST_MUTATION_STRATEGY = "test_mutation_strategy.py ";

fuzzbackend/src/main/java/com/example/fuzzControll/constents/protocolConstent.java → fuzzbackend/src/main/java/com/example/fuzzControll/constents/ProtocolConstent.java

 package com.example.fuzzControll.constents;
 
-public class protocolConstent {
+public class ProtocolConstent {
     public static final String ARP = "arp_raw.py ";
     public static final String BGP = "bgp_tcp.py ";
     public static final String DHCP = "dhcp_scapy.py ";

fuzzbackend/src/main/java/com/example/fuzzControll/controller/kittyServerMessageController.java → fuzzbackend/src/main/java/com/example/fuzzControll/controller/KittyServerMessageController.java

 package com.example.fuzzControll.controller;
 
 import com.example.fuzzControll.pojo.vo.AjaxResult;
-import com.example.fuzzControll.service.getServerMessageService;
+import com.example.fuzzControll.service.GetServerMessageService;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.web.bind.annotation.RequestMapping;
 import org.springframework.web.bind.annotation.RequestMethod;
@@ -12,9 +12,9 @@ import org.springframework.web.bind.annotation.RestController;
  */
 @RestController
 @RequestMapping("/kittyServer")
-public class kittyServerMessageController {
+public class KittyServerMessageController {
     @Autowired
-    getServerMessageService getServerMessageService;
+    GetServerMessageService getServerMessageService;
     /**
      * 获取服务器stats信息
      */
@@ -45,7 +45,7 @@ public class kittyServerMessageController {
         try {
             return AjaxResult.success(getServerMessageService.getStages());
         } catch (Exception e) {
-            return AjaxResult.error("stats信息获取失败！");
+            return AjaxResult.error("stages信息获取失败！");
         }
     }
     /**
@@ -56,7 +56,7 @@ public class kittyServerMessageController {
         try {
             return AjaxResult.success(getServerMessageService.getReport());
         } catch (Exception e) {
-            return AjaxResult.error("stats信息获取失败！");
+            return AjaxResult.error("report信息获取失败！");
         }
     }
 }

fuzzbackend/src/main/java/com/example/fuzzControll/controller/seedFileController.java → fuzzbackend/src/main/java/com/example/fuzzControll/controller/SeedFileController.java

 package com.example.fuzzControll.controller;
 
 import com.example.fuzzControll.pojo.vo.AjaxResult;
-import com.example.fuzzControll.service.seedFileService;
+import com.example.fuzzControll.service.SeedFileService;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.web.bind.annotation.PathVariable;
 import org.springframework.web.bind.annotation.RequestMapping;
@@ -16,10 +16,10 @@ import java.util.List;
  */
 @RestController
 @RequestMapping("/seedFile")
-public class seedFileController {
+public class SeedFileController {
 
     @Autowired
-    seedFileService service;
+    SeedFileService service;
 
     /**
      * 种子文件查询接口

fuzzbackend/src/main/java/com/example/fuzzControll/controller/testClassController.java → fuzzbackend/src/main/java/com/example/fuzzControll/controller/TestClassController.java

 package com.example.fuzzControll.controller;
 
 import com.example.fuzzControll.pojo.vo.AjaxResult;
-import com.example.fuzzControll.pojo.vo.testEntity;
-import com.example.fuzzControll.service.generateMethodService;
-import com.example.fuzzControll.service.mutationService;
-import com.example.fuzzControll.service.protocolTemplateService;
-import com.example.fuzzControll.service.vulnerabilityTypeService;
+import com.example.fuzzControll.pojo.vo.TestEntity;
+import com.example.fuzzControll.service.GenerateMethodService;
+import com.example.fuzzControll.service.MutationService;
+import com.example.fuzzControll.service.ProtocolTemplateService;
+import com.example.fuzzControll.service.VulnerabilityTypeService;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.web.bind.annotation.RequestBody;
 import org.springframework.web.bind.annotation.RequestMapping;
@@ -20,20 +20,20 @@ import java.util.Map;
  */
 @RestController
 @RequestMapping("/testClass")
-public class testClassController {
+public class TestClassController {
     @Autowired
-    generateMethodService generateMethodService;
+    GenerateMethodService generateMethodService;
     @Autowired
-    mutationService mutationService;
+    MutationService mutationService;
     @Autowired
-    protocolTemplateService protocolTemplateService;
+    ProtocolTemplateService protocolTemplateService;
     @Autowired
-    vulnerabilityTypeService vulnerabilityTypeService;
+    VulnerabilityTypeService vulnerabilityTypeService;
     /**
      * 模板
      */
     @RequestMapping(value = "/protocolTemplate", method = RequestMethod.POST)
-    public AjaxResult protocolTemplate(@RequestBody testEntity testEntity) {
+    public AjaxResult protocolTemplate(@RequestBody TestEntity testEntity) {
         try {
             Map<String, List<String>> result = protocolTemplateService.generation(testEntity);
             return AjaxResult.success(result==null?"模板文件生成未成功运行":result);
@@ -46,7 +46,7 @@ public class testClassController {
      *生成方法
      */
     @RequestMapping(value = "/generate", method = RequestMethod.POST)
-    public AjaxResult generate(@RequestBody testEntity testEntity) {
+    public AjaxResult generate(@RequestBody TestEntity testEntity) {
         try {
             Map<String, List<String>> result = generateMethodService.generation(testEntity);
             return AjaxResult.success(result==null?"生成方法未成功运行":result);
@@ -59,7 +59,7 @@ public class testClassController {
      *变异方法
      */
     @RequestMapping(value = "/mutation", method = RequestMethod.POST)
-    public AjaxResult mutation(@RequestBody testEntity testEntity) {
+    public AjaxResult mutation(@RequestBody TestEntity testEntity) {
         try {
             Map<String, List<String>> result = mutationService.generation(testEntity);
             return AjaxResult.success(result==null?"mutationTest未成功运行":result);
@@ -72,7 +72,7 @@ public class testClassController {
      *漏洞类型
      */
     @RequestMapping(value = "/vulnerabilityType", method = RequestMethod.POST)
-    public AjaxResult upload(@RequestBody testEntity testEntity) {
+    public AjaxResult upload(@RequestBody TestEntity testEntity) {
         try {
             Map<String, List<String>> result = vulnerabilityTypeService.generation(testEntity);
             return AjaxResult.success(result==null?"漏洞类型未成功运行":result);

fuzzbackend/src/main/java/com/example/fuzzControll/controller/testControler.java → fuzzbackend/src/main/java/com/example/fuzzControll/controller/TestControler.java

 package com.example.fuzzControll.controller;
 
 import com.example.fuzzControll.pojo.vo.AjaxResult;
-import com.example.fuzzControll.pojo.vo.cmdStartParams;
-import com.example.fuzzControll.service.testService;
+import com.example.fuzzControll.pojo.vo.CmdStartParams;
+import com.example.fuzzControll.service.TestService;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.web.bind.annotation.RequestBody;
 import org.springframework.web.bind.annotation.RequestMapping;
@@ -14,15 +14,15 @@ import org.springframework.web.bind.annotation.RestController;
  */
 @RestController
 @RequestMapping("/test")
-public class testControler {
+public class TestControler {
     @Autowired
-    testService service;
+    TestService service;
 
     /**
      * 测试启动
      */
     @RequestMapping(value = "/testStart", method = RequestMethod.POST)
-    public AjaxResult list(@RequestBody final cmdStartParams cmdStartParams) {
+    public AjaxResult list(@RequestBody final CmdStartParams cmdStartParams) {
         try {
             new Thread(new Runnable() {
                 @Override

fuzzbackend/src/main/java/com/example/fuzzControll/exception/AflnetException.java

+package com.example.fuzzControll.exception;
+
+public class AflnetException extends BaseException{
+    private static final long serialVersionUID = 1L;
+
+    public AflnetException(String defaultMessage) {
+        super(defaultMessage, "aflnet");
+    }
+}

fuzzbackend/src/main/java/com/example/fuzzControll/exception/BaseException.java

+package com.example.fuzzControll.exception;
+
+/**
+ * 基础异常
+ */
+public class BaseException extends RuntimeException{
+    private static final long serialVersionUID = 1L;
+
+    /**
+     * 错误消息
+     */
+    private String defaultMessage;
+    /**
+     * 所属模块
+     */
+    private String module;
+
+    public BaseException(String defaultMessage, String module) {
+        this.defaultMessage = defaultMessage;
+        this.module = module;
+    }
+}

fuzzbackend/src/main/java/com/example/fuzzControll/exception/CmdException.java

+package com.example.fuzzControll.exception;
+
+/**
+ * cmd运行异常
+ */
+public class CmdException extends BaseException{
+    private static final long serialVersionUID = 1L;
+    public CmdException(String defaultMessage) {
+        super(defaultMessage, "cmd");
+    }
+}

fuzzbackend/src/main/java/com/example/fuzzControll/exception/FileException.java

+package com.example.fuzzControll.exception;
+
+/**
+ * 文件操作异常
+ */
+public class FileException extends BaseException{
+    private static final long serialVersionUID = 1L;
+    public FileException(String defaultMessage) {
+        super(defaultMessage, "file");
+    }
+}

fuzzbackend/src/main/java/com/example/fuzzControll/exception/FuzzException.java

+package com.example.fuzzControll.exception;
+
+public class FuzzException  extends BaseException{
+    private static final long serialVersionUID = 1L;
+    public FuzzException(String defaultMessage) {
+        super(defaultMessage, "fuzz");
+    }
+}

fuzzbackend/src/main/java/com/example/fuzzControll/exception/ServerException.java

+package com.example.fuzzControll.exception;
+
+public class ServerException extends BaseException{
+    private static final long serialVersionUID = 1L;
+    public ServerException(String defaultMessage) {
+        super(defaultMessage, "server");
+    }
+}

fuzzbackend/src/main/java/com/example/fuzzControll/pojo/vo/cmdStartParams.java → fuzzbackend/src/main/java/com/example/fuzzControll/pojo/vo/CmdStartParams.java

@@ -7,7 +7,7 @@ import lombok.Setter;
 @Data
 @Getter
 @Setter
-public class cmdStartParams {
+public class CmdStartParams {
     String netinfo; //netInfo
     String protopcol; //protocol
     int waiting; //usec

fuzzbackend/src/main/java/com/example/fuzzControll/pojo/vo/testEntity.java → fuzzbackend/src/main/java/com/example/fuzzControll/pojo/vo/TestEntity.java

@@ -7,7 +7,7 @@ import lombok.Setter;
 @Data
 @Getter
 @Setter
-public class testEntity {
+public class TestEntity {
     String testClassName;
     String[] paramJson;
 }

fuzzbackend/src/main/java/com/example/fuzzControll/pojo/vo/testReturnEntity.java → fuzzbackend/src/main/java/com/example/fuzzControll/pojo/vo/TestReturnEntity.java

@@ -3,7 +3,7 @@ package com.example.fuzzControll.pojo.vo;
 import lombok.Data;
 
 @Data
-public class testReturnEntity {
+public class TestReturnEntity {
     String run_time;
     String cycles_done;
     String last_new_path;

fuzzbackend/src/main/java/com/example/fuzzControll/service/generateMethodService.java → fuzzbackend/src/main/java/com/example/fuzzControll/service/GenerateMethodService.java

 package com.example.fuzzControll.service;
 
-import com.example.fuzzControll.pojo.vo.testEntity;
+import com.example.fuzzControll.pojo.vo.TestEntity;
 
 import java.util.List;
 import java.util.Map;
 
-public interface generateMethodService {
-    Map<String,List<String>> generation(testEntity testEntity);
+public interface GenerateMethodService {
+    Map<String,List<String>> generation(TestEntity testEntity);
 }

fuzzbackend/src/main/java/com/example/fuzzControll/service/getServerMessageService.java → fuzzbackend/src/main/java/com/example/fuzzControll/service/GetServerMessageService.java

 package com.example.fuzzControll.service;
 
-import java.util.List;
-import java.util.Map;
-
-public interface getServerMessageService {
+public interface GetServerMessageService {
     String getStats();
 
     String getTemplateInfo();

fuzzbackend/src/main/java/com/example/fuzzControll/service/mutationService.java → fuzzbackend/src/main/java/com/example/fuzzControll/service/MutationService.java

 package com.example.fuzzControll.service;
 
-import com.example.fuzzControll.pojo.vo.testEntity;
+import com.example.fuzzControll.pojo.vo.TestEntity;
 
 import java.util.List;
 import java.util.Map;
 
-public interface mutationService {
-    Map<String, List<String>> generation(testEntity testEntity);
+public interface MutationService {
+    Map<String, List<String>> generation(TestEntity testEntity);
 }

fuzzbackend/src/main/java/com/example/fuzzControll/service/protocolTemplateService.java → fuzzbackend/src/main/java/com/example/fuzzControll/service/ProtocolTemplateService.java

 package com.example.fuzzControll.service;
 
-import com.example.fuzzControll.pojo.vo.testEntity;
+import com.example.fuzzControll.pojo.vo.TestEntity;
 
 import java.util.List;
 import java.util.Map;
 
-public interface protocolTemplateService {
-    Map<String,List<String>> generation(testEntity testEntity);
+public interface ProtocolTemplateService {
+    Map<String,List<String>> generation(TestEntity testEntity);
 
 }

fuzzbackend/src/main/java/com/example/fuzzControll/service/seedFileService.java → fuzzbackend/src/main/java/com/example/fuzzControll/service/SeedFileService.java

@@ -4,10 +4,11 @@ import org.springframework.web.multipart.MultipartFile;
 
 import java.util.List;
 
-public interface seedFileService {
+public interface SeedFileService {
     public List<String> getSeedFiles();
 
     void delFile(String fileName);
 
     void upload(MultipartFile file);
+    int getSeedFileCount();
 }

fuzzbackend/src/main/java/com/example/fuzzControll/service/testService.java → fuzzbackend/src/main/java/com/example/fuzzControll/service/TestService.java

 package com.example.fuzzControll.service;
 
-import com.example.fuzzControll.pojo.vo.cmdStartParams;
+import com.example.fuzzControll.pojo.vo.CmdStartParams;
 
-public interface testService {
+public interface TestService {
 
-    void testStart(cmdStartParams cmdStartParams);
+    void testStart(CmdStartParams cmdStartParams);
 
     void testStop();
 

fuzzbackend/src/main/java/com/example/fuzzControll/service/vulnerabilityTypeService.java → fuzzbackend/src/main/java/com/example/fuzzControll/service/VulnerabilityTypeService.java

 package com.example.fuzzControll.service;
 
-import com.example.fuzzControll.pojo.vo.testEntity;
+import com.example.fuzzControll.pojo.vo.TestEntity;
 
 import java.util.List;
 import java.util.Map;
 
-public interface vulnerabilityTypeService {
-    Map<String, List<String>> generation(testEntity testEntity);
+public interface VulnerabilityTypeService {
+    Map<String, List<String>> generation(TestEntity testEntity);
 }

fuzzbackend/src/main/java/com/example/fuzzControll/service/impl/generateMethodServiceImpl.java → fuzzbackend/src/main/java/com/example/fuzzControll/service/impl/GenerateMethodServiceImpl.java

 package com.example.fuzzControll.service.impl;
 
-import com.example.fuzzControll.conf.kittyProperties;
-import com.example.fuzzControll.pojo.vo.testEntity;
-import com.example.fuzzControll.service.generateMethodService;
-import com.example.fuzzControll.tools.cmdTools;
-import com.example.fuzzControll.tools.testTools;
+import com.example.fuzzControll.conf.KittyProperties;
+import com.example.fuzzControll.exception.FuzzException;
+import com.example.fuzzControll.pojo.vo.TestEntity;
+import com.example.fuzzControll.service.GenerateMethodService;
+import com.example.fuzzControll.tools.CmdTools;
+import com.example.fuzzControll.tools.TestTools;
 import lombok.extern.slf4j.Slf4j;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
@@ -14,13 +15,13 @@ import java.util.Map;
 
 @Service
 @Slf4j
-public class generateMethodServiceImpl implements generateMethodService {
-    cmdTools cmdTools = new cmdTools();
+public class GenerateMethodServiceImpl implements GenerateMethodService {
+    CmdTools cmdTools = new CmdTools();
     @Autowired
-    kittyProperties kitty;
+    KittyProperties kitty;
 
     @Override
-    public Map<String, List<String>> generation(testEntity testEntity) {
+    public Map<String, List<String>> generation(TestEntity testEntity) throws FuzzException {
         String cmd = parseParameters(testEntity);
         if (cmd.isEmpty()) {
             return null;
@@ -28,7 +29,7 @@ public class generateMethodServiceImpl implements generateMethodService {
         return cmdTools.runProgramCmdAndResult(cmd);
     }
 
-    public String parseParameters(testEntity testEntity) {
+    public String parseParameters(TestEntity testEntity) {
         switch (testEntity.getTestClassName().toLowerCase()) {
             case "foreach":
                 return cmd(testEntity, "-f");
@@ -56,8 +57,8 @@ public class generateMethodServiceImpl implements generateMethodService {
         }
     }
 
-    private String cmd(testEntity testEntity, String cmd) {
-        if (!testTools.paramsLenghtTest(testEntity.getParamJson().length, 5, "generationMethod"))
+    private String cmd(TestEntity testEntity, String cmd) {
+        if (!TestTools.paramsLenghtTest(testEntity.getParamJson().length, 5, "generationMethod"))
             return "";
         String target_host = null;
         String target_port = null;

fuzzbackend/src/main/java/com/example/fuzzControll/service/impl/getServerMessageImpl.java → fuzzbackend/src/main/java/com/example/fuzzControll/service/impl/GetServerMessageImpl.java

 package com.example.fuzzControll.service.impl;
 
-import com.example.fuzzControll.conf.kittyProperties;
-import com.example.fuzzControll.service.getServerMessageService;
+import com.example.fuzzControll.conf.KittyProperties;
+import com.example.fuzzControll.exception.ServerException;
+import com.example.fuzzControll.service.GetServerMessageService;
 import lombok.extern.slf4j.Slf4j;
-import org.apache.http.HttpStatus;
 import org.apache.http.client.methods.CloseableHttpResponse;
 import org.apache.http.client.methods.HttpGet;
 import org.apache.http.impl.client.CloseableHttpClient;
@@ -12,18 +12,13 @@ import org.apache.http.util.EntityUtils;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
 
-import java.util.ArrayList;
-import java.util.HashMap;
-import java.util.List;
-import java.util.Map;
-
 @Slf4j
 @Service("getServerMessageService")
-public class getServerMessageImpl implements getServerMessageService {
+public class GetServerMessageImpl implements GetServerMessageService {
     @Autowired
-    kittyProperties kitty;
+    KittyProperties kitty;
 
-    public String getServerMsg(String messageName) {
+    public String getServerMsg(String messageName) throws ServerException{
         switch (messageName) {
             case "templateInfo": {
                 HttpGet httpGetTemplateInfo = new HttpGet(kitty.getTemplateInfoHttp());
@@ -32,6 +27,7 @@ public class getServerMessageImpl implements getServerMessageService {
                     return EntityUtils.toString(templateInfoResponse.getEntity(), "utf-8");
                 } catch (Exception e) {
                     log.error("templateInfo http error!");
+                    throw new ServerException("get server templateInfo error !");
                 }
             }
             case "stats": {
@@ -41,6 +37,7 @@ public class getServerMessageImpl implements getServerMessageService {
                     return EntityUtils.toString(statsResponse.getEntity(), "utf-8");
                 } catch (Exception e) {
                     log.error("stats http error!");
+                    throw new ServerException("get server stats error !");
                 }
             }
             case "report": {
@@ -50,6 +47,8 @@ public class getServerMessageImpl implements getServerMessageService {
                     return EntityUtils.toString(reportResponse.getEntity(), "utf-8");
                 } catch (Exception e) {
                     log.error("report http error!");
+                    throw new ServerException("get server report error !");
+
                 }
             }
             case "stages": {
@@ -59,6 +58,7 @@ public class getServerMessageImpl implements getServerMessageService {
                     return EntityUtils.toString(stagesResponse.getEntity(), "utf-8");
                 } catch (Exception e) {
                     log.error("stages http error!");
+                    throw new ServerException("get server stages error !");
                 }
             }
             default:
@@ -67,22 +67,22 @@ public class getServerMessageImpl implements getServerMessageService {
     }
 
     @Override
-    public String getStats() {
+    public String getStats() throws ServerException {
         return getServerMsg("stats");
     }
 
     @Override
-    public String getTemplateInfo() {
+    public String getTemplateInfo() throws ServerException{
         return getServerMsg("templateInfo");
     }
 
     @Override
-    public String getStages() {
-        return getServerMsg("report");
+    public String getStages() throws ServerException{
+        return getServerMsg("stages");
     }
 
     @Override
-    public String getReport() {
-        return getServerMsg("stages");
+    public String getReport() throws ServerException{
+        return getServerMsg("report");
     }
 }

fuzzbackend/src/main/java/com/example/fuzzControll/service/impl/mutationServiceImpl.java → fuzzbackend/src/main/java/com/example/fuzzControll/service/impl/MutationServiceImpl.java

 package com.example.fuzzControll.service.impl;
 
-import com.example.fuzzControll.conf.kittyProperties;
-import com.example.fuzzControll.constents.mutationConstent;
-import com.example.fuzzControll.pojo.vo.testEntity;
-import com.example.fuzzControll.service.mutationService;
-import com.example.fuzzControll.tools.cmdTools;
-import com.example.fuzzControll.tools.testTools;
+import com.example.fuzzControll.conf.KittyProperties;
+import com.example.fuzzControll.constents.MutationConstent;
+import com.example.fuzzControll.exception.FuzzException;
+import com.example.fuzzControll.pojo.vo.TestEntity;
+import com.example.fuzzControll.service.MutationService;
+import com.example.fuzzControll.tools.CmdTools;
+import com.example.fuzzControll.tools.TestTools;
 import lombok.extern.slf4j.Slf4j;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
@@ -15,14 +16,14 @@ import java.util.Map;
 
 @Service("mutationService")
 @Slf4j
-class mutationServiceImpl implements mutationService {
-    cmdTools cmdTools = new cmdTools();
+class MutationServiceImpl implements MutationService {
+    CmdTools cmdTools = new CmdTools();
     @Autowired
-    kittyProperties kitty;
+    KittyProperties kitty;
 
 
     @Override
-    public Map<String, List<String>> generation(testEntity testEntity) {
+    public Map<String, List<String>> generation(TestEntity testEntity) throws FuzzException {
         String cmd = parseParameters(testEntity);
         if (cmd.isEmpty()) {
             return null;
@@ -30,7 +31,7 @@ class mutationServiceImpl implements mutationService {
         return cmdTools.runProgramCmdAndResult(cmd);
     }
 
-    public String parseParameters(testEntity testEntity) {
+    public String parseParameters(TestEntity testEntity) {
         switch (testEntity.getTestClassName().toLowerCase()) {
             case "bit":
                 return variationGranularityCmd(testEntity, 1);
@@ -96,8 +97,8 @@ class mutationServiceImpl implements mutationService {
         }
     }
 
-    private String distortionLibCmd(testEntity testEntity, int methodNum) {
-        if (!testTools.paramsLenghtTest(testEntity.getParamJson().length, 2, "distortionLib" + methodNum))
+    private String distortionLibCmd(TestEntity testEntity, int methodNum) {
+        if (!TestTools.paramsLenghtTest(testEntity.getParamJson().length, 2, "distortionLib" + methodNum))
             return "";
         String dst_ip = null;
         String dst_port = null;
@@ -107,12 +108,12 @@ class mutationServiceImpl implements mutationService {
         } catch (Exception e) {
             log.error("distortionLib [{}] 参数解析失败！", methodNum);
         }
-        return kitty.getVenvPath() + " " + kitty.getMutationPath() + mutationConstent.TEST_MUTATED_LIBS + " -g " + methodNum + " -d " + dst_ip + " -p " + dst_port;
+        return kitty.getVenvPath() + " " + kitty.getMutationPath() + MutationConstent.TEST_MUTATED_LIBS + " -g " + methodNum + " -d " + dst_ip + " -p " + dst_port;
     }
 
 
-    private String variationGranularityCmd(testEntity testEntity, int methodNum) {
-        if (!testTools.paramsLenghtTest(testEntity.getParamJson().length, 2, "variationGranularity" + methodNum))
+    private String variationGranularityCmd(TestEntity testEntity, int methodNum) {
+        if (!TestTools.paramsLenghtTest(testEntity.getParamJson().length, 2, "variationGranularity" + methodNum))
             return "";
         String dst_ip = null;
         String dst_port = null;
@@ -122,10 +123,10 @@ class mutationServiceImpl implements mutationService {
         } catch (Exception e) {
             log.error("variationGranularity [{}] 参数解析失败！", methodNum);
         }
-        return kitty.getVenvPath() + " " + kitty.getMutationPath() + mutationConstent.TEST_GRANULARITY_BIT_BYTE + " -g " + methodNum + " -d " + dst_ip + " -p " + dst_port;
+        return kitty.getVenvPath() + " " + kitty.getMutationPath() + MutationConstent.TEST_GRANULARITY_BIT_BYTE + " -g " + methodNum + " -d " + dst_ip + " -p " + dst_port;
     }
-    private String mutationStrategyCmd(testEntity testEntity, int methodNum) {
-        if (!testTools.paramsLenghtTest(testEntity.getParamJson().length, 2, "mutationStrategy" + methodNum))
+    private String mutationStrategyCmd(TestEntity testEntity, int methodNum) {
+        if (!TestTools.paramsLenghtTest(testEntity.getParamJson().length, 2, "mutationStrategy" + methodNum))
             return "";
         String dst_ip = null;
         String dst_port = null;
@@ -135,7 +136,7 @@ class mutationServiceImpl implements mutationService {
         } catch (Exception e) {
             log.error("mutationStrategy [{}] 参数解析失败！", methodNum);
         }
-        return kitty.getVenvPath() + " " + kitty.getMutationPath() + mutationConstent.TEST_MUTATION_STRATEGY + " -g " + methodNum + " -d " + dst_ip + " -p " + dst_port;
+        return kitty.getVenvPath() + " " + kitty.getMutationPath() + MutationConstent.TEST_MUTATION_STRATEGY + " -g " + methodNum + " -d " + dst_ip + " -p " + dst_port;
     }
 
 }
\ No newline at end of file

fuzzbackend/src/main/java/com/example/fuzzControll/service/impl/SeedFileServiceImpl.java

+package com.example.fuzzControll.service.impl;
+
+import com.example.fuzzControll.conf.SeedProperties;
+import com.example.fuzzControll.constents.CmdConstent;
+import com.example.fuzzControll.exception.CmdException;
+import com.example.fuzzControll.exception.FileException;
+import com.example.fuzzControll.service.SeedFileService;
+import com.example.fuzzControll.tools.CmdTools;
+import com.example.fuzzControll.tools.FileTools;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.stereotype.Service;
+import org.springframework.web.multipart.MultipartFile;
+
+import java.util.ArrayList;
+import java.util.List;
+
+@Service
+public class SeedFileServiceImpl implements SeedFileService {
+    CmdTools cmdTools = new CmdTools();
+    FileTools fileTools = new FileTools();
+    @Autowired
+    SeedProperties properties;
+
+    @Override
+    public List<String> getSeedFiles() throws CmdException{
+        return cmdTools.runCmd(CmdConstent.GET_FILE_NAME + properties.getSeedPath());
+    }
+
+    //todo 同步修改可能会出现问题
+    @Override
+    public void delFile(String fileName) throws CmdException {
+        int fileCountBefore = getSeedFileCount();
+        cmdTools.runCmd(CmdConstent.DELETE_FILE + properties.getSeedPath() + "/" + fileName);
+        int fileCountAfter = getSeedFileCount();
+        if(fileCountAfter==fileCountBefore){
+            throw new CmdException("delete cmd error !The file has not changed.");
+        }
+    }
+
+    @Override
+    public void upload(MultipartFile file) throws FileException {
+        int fileCountBefore = getSeedFileCount();
+        fileTools.load(file);
+        int fileCountAfter = getSeedFileCount();
+        if(fileCountAfter==fileCountBefore){
+            throw new CmdException("upload file error !The file failed to be submitted.");
+        }
+    }
+
+    /**
+     *
+     * 获取种子文件目录下文件数量
+     */
+    @Override
+    public int getSeedFileCount() throws CmdException {
+        List<String> files = cmdTools.runCmd(CmdConstent.GET_FILE_NAME+ properties.getSeedPath());
+        int count = files.size();
+        return count;
+    }
+}

fuzzbackend/src/main/java/com/example/fuzzControll/service/impl/testServiceImpl.java → fuzzbackend/src/main/java/com/example/fuzzControll/service/impl/TestServiceImpl.java

 package com.example.fuzzControll.service.impl;
 
 import com.example.fuzzControll.conf.SpringContextUtil;
-import com.example.fuzzControll.conf.kittyProperties;
-import com.example.fuzzControll.constents.cmdConstent;
-import com.example.fuzzControll.pojo.vo.cmdStartParams;
-import com.example.fuzzControll.service.testService;
-import com.example.fuzzControll.tools.cmdTools;
-import com.example.fuzzControll.tools.testControlTools;
+import com.example.fuzzControll.conf.KittyProperties;
+import com.example.fuzzControll.constents.CmdConstent;
+import com.example.fuzzControll.pojo.vo.CmdStartParams;
+import com.example.fuzzControll.service.TestService;
+import com.example.fuzzControll.tools.CmdTools;
+import com.example.fuzzControll.tools.TestControlTools;
 import org.springframework.stereotype.Service;
 
 @Service("testService")
-public class testServiceImpl implements testService {
-    kittyProperties kittyProperties = (kittyProperties) SpringContextUtil.getBean("kittyProperties");
+public class TestServiceImpl implements TestService {
+    KittyProperties kittyProperties = (KittyProperties) SpringContextUtil.getBean("kittyProperties");
 
-    cmdTools cmdTools = new cmdTools();
+    CmdTools cmdTools = new CmdTools();
 
     //todo 不同服务不同端口
     @Override
-    public void testStart(cmdStartParams cmdStartParams) {
-        testControlTools.setIsRunning(true);
+    public void testStart(CmdStartParams cmdStartParams) {
+        TestControlTools.setIsRunning(true);
         String cmd = cmdTools.parse(cmdStartParams);
-        String finalCmd = cmdConstent.RUN_AFLNET + cmd + kittyProperties.getAflnetPath()+"live555/testProgs/testOnDemandRTSPServer 8554";
+        String finalCmd = CmdConstent.RUN_AFLNET + cmd + kittyProperties.getAflnetPath()+"live555/testProgs/testOnDemandRTSPServer 8554";
         cmdTools.runProgramCmd(finalCmd);
     }
 
 
     @Override
     public void testStop() {
-        testControlTools.setIsRunning(false);
+        TestControlTools.setIsRunning(false);
     }
 }

fuzzbackend/src/main/java/com/example/fuzzControll/service/impl/vulnerabilityTypeServiceImpl.java → fuzzbackend/src/main/java/com/example/fuzzControll/service/impl/VulnerabilityTypeServiceImpl.java

 package com.example.fuzzControll.service.impl;
 
-import com.example.fuzzControll.conf.kittyProperties;
-import com.example.fuzzControll.pojo.vo.testEntity;
-import com.example.fuzzControll.service.vulnerabilityTypeService;
-import com.example.fuzzControll.tools.cmdTools;
+import com.example.fuzzControll.conf.KittyProperties;
+import com.example.fuzzControll.exception.FuzzException;
+import com.example.fuzzControll.pojo.vo.TestEntity;
+import com.example.fuzzControll.service.VulnerabilityTypeService;
+import com.example.fuzzControll.tools.CmdTools;
 import lombok.extern.slf4j.Slf4j;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
@@ -13,18 +14,18 @@ import java.util.Map;
 
 @Slf4j
 @Service("vulnerabilityTypeService")
-public class vulnerabilityTypeServiceImpl implements vulnerabilityTypeService {
-    cmdTools cmdTools = new cmdTools();
+public class VulnerabilityTypeServiceImpl implements VulnerabilityTypeService {
+    CmdTools cmdTools = new CmdTools();
     @Autowired
-    kittyProperties kitty;
+    KittyProperties kitty;
 
     @Override
-    public Map<String, List<String>> generation(testEntity testEntity) {
+    public Map<String, List<String>> generation(TestEntity testEntity) throws FuzzException {
         String cmd = parseParameters(testEntity);
         return cmdTools.runProgramCmdAndResult(cmd);
     }
 
-    public String parseParameters(testEntity testEntity) {
+    public String parseParameters(TestEntity testEntity) {
         switch (testEntity.getTestClassName().toLowerCase()) {
             case "array_index_out_of_bounds_vulnerabilit"://have error
                 return cmd(testEntity, 0);
@@ -56,7 +57,7 @@ public class vulnerabilityTypeServiceImpl implements vulnerabilityTypeService {
         }
     }
 
-    private String cmd(testEntity testEntity, int kindNum) {
+    private String cmd(TestEntity testEntity, int kindNum) {
         return kitty.getVenvPath() + " " + kitty.getVulnerabilityTypePath() + "vul_types_test.py " + kindNum;
     }
     //todo 还有很多类型要写

fuzzbackend/src/main/java/com/example/fuzzControll/service/impl/seedFileServiceImpl.java

-package com.example.fuzzControll.service.impl;
-
-import com.example.fuzzControll.conf.seedProperties;
-import com.example.fuzzControll.constents.cmdConstent;
-import com.example.fuzzControll.service.seedFileService;
-import com.example.fuzzControll.tools.cmdTools;
-import com.example.fuzzControll.tools.fileTools;
-import org.springframework.beans.factory.annotation.Autowired;
-import org.springframework.stereotype.Service;
-import org.springframework.web.multipart.MultipartFile;
-
-import java.util.List;
-
-@Service
-public class seedFileServiceImpl implements seedFileService {
-    cmdTools cmdTools = new cmdTools();
-    fileTools fileTools = new fileTools();
-    @Autowired
-    seedProperties properties;
-    @Override
-
-    public List<String> getSeedFiles() {
-        return cmdTools.runCmd(cmdConstent.GET_FILE_NAME+properties.getSeedPath());
-    }
-
-    //todo 没有执行结果提示，
-    @Override
-    public void delFile(String fileName) {
-        cmdTools.runCmd(cmdConstent.DELETE_FILE+properties.getSeedPath()+"/"+fileName);
-    }
-
-    @Override
-    public void upload(MultipartFile file) {
-        fileTools.load(file);
-    }
-}

fuzzbackend/src/main/java/com/example/fuzzControll/tools/cmdTools.java → fuzzbackend/src/main/java/com/example/fuzzControll/tools/CmdTools.java

@@ -3,8 +3,10 @@ package com.example.fuzzControll.tools;
 import com.alibaba.fastjson.JSONObject;
 import com.example.fuzzControll.conf.SpringContextUtil;
 import com.example.fuzzControll.controller.WebSocket;
-import com.example.fuzzControll.pojo.vo.cmdStartParams;
-import com.example.fuzzControll.pojo.vo.testReturnEntity;
+import com.example.fuzzControll.exception.CmdException;
+import com.example.fuzzControll.exception.FuzzException;
+import com.example.fuzzControll.pojo.vo.CmdStartParams;
+import com.example.fuzzControll.pojo.vo.TestReturnEntity;
 
 import java.io.*;
 
@@ -12,14 +14,14 @@ import java.util.*;
 
 
 //todo need modify
-public class cmdTools {
+public class CmdTools {
     Boolean send = false;
     WebSocket socket = (WebSocket) SpringContextUtil.getBean("WebSocket");
 
     /**
      * 运行不需要后台运行cmd
      */
-    public List<String> runCmd(String cmd) {
+    public List<String> runCmd(String cmd) throws CmdException {
         List<String> result = new ArrayList<String>();
         try {
             Process process = Runtime.getRuntime().exec(cmd);
@@ -27,7 +29,7 @@ public class cmdTools {
             printMessage(process.getErrorStream(), new ArrayList<String>());
             process.waitFor();
         } catch (Exception e) {
-            e.printStackTrace();
+            throw new CmdException("run delete or search cmd error !");
         }
 
         return result;
@@ -54,7 +56,7 @@ public class cmdTools {
      * 运行需要后台运行cmd
      * 将数据存入文件中
      */
-    public Map<String, List<String>> runProgramCmdAndResult(String cmd) {
+    public Map<String, List<String>> runProgramCmdAndResult(String cmd) throws FuzzException {
         Map<String, List<String>> result = new HashMap();
         List<String> out = Collections.synchronizedList(new ArrayList<String>());
         List<String> error = Collections.synchronizedList(new ArrayList<String>());
@@ -64,7 +66,7 @@ public class cmdTools {
             printMessageByProgramCmd(process.getErrorStream(), error);
             process.waitFor();
         } catch (Exception e) {
-            e.printStackTrace();
+            throw new FuzzException("run fuzz error !");
         }
         result.put("out", out);
         result.put("error", error);
@@ -115,8 +117,8 @@ public class cmdTools {
         Reader reader = new InputStreamReader(input);
         BufferedReader bf = new BufferedReader(reader);
         String line = null;
-        testReturnEntity returnEntity = new testReturnEntity();
-        while ((line = bf.readLine()) != null && testControlTools.getIsRunning()) {
+        TestReturnEntity returnEntity = new TestReturnEntity();
+        while ((line = bf.readLine()) != null && TestControlTools.getIsRunning()) {
             makeReturnEntity(line, returnEntity);
             if (send) {
                 String data = JSONObject.toJSONString(returnEntity);
@@ -125,7 +127,7 @@ public class cmdTools {
         }
     }
 
-    private testReturnEntity makeReturnEntity(String line, testReturnEntity returnEntity) {
+    private TestReturnEntity makeReturnEntity(String line, TestReturnEntity returnEntity) {
         if (line.contains("run time")) {
             send = false;
             int run_time = line.indexOf(":");
@@ -215,7 +217,7 @@ public class cmdTools {
         return returnEntity;
     }
 
-    public String parse(cmdStartParams cmdStartParams) {
+    public String parse(CmdStartParams cmdStartParams) {
         StringBuilder cmd = new StringBuilder();
         if (cmdStartParams.getNetinfo() != "") {
             cmd.append(" -N " + cmdStartParams.getNetinfo());

fuzzbackend/src/main/java/com/example/fuzzControll/tools/fileTools.java → fuzzbackend/src/main/java/com/example/fuzzControll/tools/FileTools.java

 package com.example.fuzzControll.tools;
 
 import com.example.fuzzControll.conf.SpringContextUtil;
-import com.example.fuzzControll.conf.seedProperties;
+import com.example.fuzzControll.conf.SeedProperties;
+import com.example.fuzzControll.exception.FileException;
 import org.springframework.web.multipart.MultipartFile;
 
 import java.io.BufferedInputStream;
@@ -9,10 +10,13 @@ import java.io.BufferedOutputStream;
 import java.io.FileOutputStream;
 import java.io.InputStream;
 
-public class fileTools {
-    seedProperties properties = (seedProperties) SpringContextUtil.getBean("seedProperties");
+public class FileTools {
+    SeedProperties properties = (SeedProperties) SpringContextUtil.getBean("seedProperties");
 
-    public void load(MultipartFile file) {
+    public void load(MultipartFile file) throws FileException {
+        if (file==null){
+            throw new FileException("upload file is null !");
+        }
         try (InputStream inputStream = file.getInputStream();
              FileOutputStream outputStream = new FileOutputStream(properties.getSeedPath() + "/" + file.getOriginalFilename());) {
             BufferedInputStream bufferedInputStream = new BufferedInputStream(inputStream);
@@ -24,7 +28,7 @@ public class fileTools {
                 bufferedOutputStream.write(buffer, 0, bytesRead);
             }
         } catch (Exception e) {
-            e.printStackTrace();
+            throw new FileException("write file error !");
         }
 
     }

fuzzbackend/src/main/java/com/example/fuzzControll/tools/regularTools.java → fuzzbackend/src/main/java/com/example/fuzzControll/tools/RegularTools.java

 package com.example.fuzzControll.tools;
 //todo 对ip等增加正则判断
-public class regularTools {
+public class RegularTools {
 }

fuzzbackend/src/main/java/com/example/fuzzControll/tools/testControlTools.java → fuzzbackend/src/main/java/com/example/fuzzControll/tools/TestControlTools.java

 package com.example.fuzzControll.tools;
 
-public class testControlTools {
+public class TestControlTools {
     private static  Boolean isRunning;
 
     public static Boolean getIsRunning() {
@@ -8,6 +8,6 @@ public class testControlTools {
     }
 
     public static void setIsRunning(Boolean isRunning) {
-        testControlTools.isRunning = isRunning;
+        TestControlTools.isRunning = isRunning;
     }
 }

fuzzbackend/src/main/java/com/example/fuzzControll/tools/testTools.java → fuzzbackend/src/main/java/com/example/fuzzControll/tools/TestTools.java

@@ -3,7 +3,7 @@ package com.example.fuzzControll.tools;
 import lombok.extern.slf4j.Slf4j;
 
 @Slf4j
-public class testTools {
+public class TestTools {
     public static boolean paramsLenghtTest(int paramsLen,int needParamsLen,String name){
        Boolean isOk = paramsLen==needParamsLen;
        if(!isOk){