aflnet结果读出成功

fuzzIntegration/src/main/java/com/example/fuzzControll/controller/dataController/AflnetDataController.java

 package com.example.fuzzControll.controller.dataController;
 
+import com.example.fuzzControll.pojo.vo.AflnetDataParams;
+import com.example.fuzzControll.pojo.vo.AjaxResult;
+import com.example.fuzzControll.pojo.vo.KittyResult;
 import com.example.fuzzControll.service.AflnetPersistenceService;
 import lombok.extern.slf4j.Slf4j;
 import org.springframework.beans.factory.annotation.Autowired;
-import org.springframework.web.bind.annotation.RequestMapping;
-import org.springframework.web.bind.annotation.RestController;
+import org.springframework.web.bind.annotation.*;
+
+import java.util.List;
 
 @RestController
 @RequestMapping("/aflnet")
 @Slf4j
 public class AflnetDataController {
-//    @Autowired
-//    AflnetPersistenceService aflnetPersistenceService;
+    @Autowired
+    AflnetPersistenceService aflnetPersistenceService;
+
+    /**
+     * 读取数据库文件至指定目录
+     */
+    @RequestMapping(value = "/loadFile", method = RequestMethod.GET)
+    public AjaxResult AflnetResultSelect(@RequestBody AflnetDataParams aflnetDataParams) {
+        try {
+            aflnetPersistenceService.loadInFile(aflnetDataParams.getMissionId(),aflnetDataParams.getFilPath());
+        } catch (Exception e) {
+            e.printStackTrace();
+            return AjaxResult.error("File load failed!");
+        }
+        return AjaxResult.success("File loaded successfully!");
+    }
+
 }

fuzzIntegration/src/main/java/com/example/fuzzControll/controller/dataController/KittyDataController.java

 package com.example.fuzzControll.controller.dataController;
 
-import com.example.fuzzControll.exception.CmdException;
 import com.example.fuzzControll.pojo.vo.AjaxResult;
 import com.example.fuzzControll.pojo.vo.KittyResult;
-import com.example.fuzzControll.service.FuzzPersistenceService;
-import com.github.pagehelper.PageHelper;
-import com.github.pagehelper.PageInfo;
+import com.example.fuzzControll.service.KittyFuzzPersistenceService;
 import lombok.extern.slf4j.Slf4j;
 import org.springframework.beans.factory.annotation.Autowired;
-import org.springframework.web.bind.annotation.PathVariable;
 import org.springframework.web.bind.annotation.RequestMapping;
 import org.springframework.web.bind.annotation.RequestMethod;
 import org.springframework.web.bind.annotation.RestController;
@@ -20,7 +16,7 @@ import java.util.List;
 @Slf4j
 public class KittyDataController {
     @Autowired
-    FuzzPersistenceService fuzzPersistenceService;
+    KittyFuzzPersistenceService fuzzPersistenceService;
     /**
      * 查询kitty结果
      */

fuzzIntegration/src/main/java/com/example/fuzzControll/exception/MysqlException.java

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

fuzzIntegration/src/main/java/com/example/fuzzControll/mapper/AflnetMapper.java

@@ -14,8 +14,11 @@ public interface AflnetMapper {
      * 查询kitty模糊测试结果
      */
 //    List<KittyResult> getKittyResults();
+
     /**
      * 存储alfnet测试结果
      */
     int aflnetOutputBackup(AflnetResult aflnetResult);
+
+    AflnetResult selectResultById(int missionId);
 }

fuzzIntegration/src/main/java/com/example/fuzzControll/mapper/KittyMapper.java

@@ -18,6 +18,7 @@ public interface KittyMapper {
 
     void kittyResultsBackup(KittyResult kittyResult);
 
+
     /**
      * 查询kitty日志
      */

fuzzIntegration/src/main/java/com/example/fuzzControll/pojo/vo/AflnetDataParams.java

+package com.example.fuzzControll.pojo.vo;
+
+import lombok.Data;
+
+@Data
+public class AflnetDataParams {
+    private String filPath;
+    private int missionId;
+
+    public AflnetDataParams() {
+    }
+
+    public AflnetDataParams(String filPath, int missionId) {
+        this.filPath = filPath;
+        this.missionId = missionId;
+    }
+}

fuzzIntegration/src/main/java/com/example/fuzzControll/service/AflnetPersistenceService.java

@@ -2,4 +2,5 @@ package com.example.fuzzControll.service;
 
 public interface AflnetPersistenceService {
     public int aflnetResultBackup(String filename);
+    public void loadInFile(int missionId,String filePath);
 }

fuzzIntegration/src/main/java/com/example/fuzzControll/service/FuzzPersistenceService.java → fuzzIntegration/src/main/java/com/example/fuzzControll/service/KittyFuzzPersistenceService.java

@@ -4,7 +4,8 @@ import com.example.fuzzControll.pojo.vo.KittyResult;
 
 import java.util.List;
 
-public interface FuzzPersistenceService {
+public interface KittyFuzzPersistenceService {
     List<KittyResult> getKittyResults();
+//    void loadInFile(int missionId);
 
 }

fuzzIntegration/src/main/java/com/example/fuzzControll/service/impl/AflnetPersistenceServiceImpl.java

 package com.example.fuzzControll.service.impl;
 
+import com.example.fuzzControll.exception.MysqlException;
 import com.example.fuzzControll.mapper.AflnetMapper;
 import com.example.fuzzControll.pojo.vo.AflnetResult;
+import com.example.fuzzControll.pojo.vo.KittyResult;
 import com.example.fuzzControll.service.AflnetPersistenceService;
 import com.example.fuzzControll.tools.FileTools;
 import lombok.extern.slf4j.Slf4j;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
+
 @Slf4j
 @Service("AflnetPersistenceService")
 public class AflnetPersistenceServiceImpl implements AflnetPersistenceService {
     @Autowired
     AflnetMapper AflnetMapper;
     FileTools fileTools = new FileTools();
+
     @Override
     public int aflnetResultBackup(String filename) {
         //todo 需要传入任务id
-        AflnetResult result = new AflnetResult(1,fileTools.fileReadAndBackup(filename),filename);
+        AflnetResult result = new AflnetResult(1, fileTools.fileReadAndBackup(filename), filename);
         return AflnetMapper.aflnetOutputBackup(result);
     }
+
+    @Override
+    public void loadInFile(int missionId, String filePath) {
+        AflnetResult aflnetResult = null;
+        try {
+            aflnetResult = AflnetMapper.selectResultById(missionId);
+        } catch (MysqlException e) {
+            e.printStackTrace();
+            throw new MysqlException("AflnetMapper.selectResultById() Error!");
+        }
+        fileTools.loadFileInLocal(aflnetResult, filePath);
+    }
+
 }

fuzzIntegration/src/main/java/com/example/fuzzControll/service/impl/FuzzPersistenceServiceImpl.java → fuzzIntegration/src/main/java/com/example/fuzzControll/service/impl/KittyFuzzPersistenceServiceImpl.java

@@ -2,7 +2,8 @@ package com.example.fuzzControll.service.impl;
 
 import com.example.fuzzControll.mapper.KittyMapper;
 import com.example.fuzzControll.pojo.vo.KittyResult;
-import com.example.fuzzControll.service.FuzzPersistenceService;
+import com.example.fuzzControll.service.AflnetPersistenceService;
+import com.example.fuzzControll.service.KittyFuzzPersistenceService;
 import lombok.extern.slf4j.Slf4j;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
@@ -11,11 +12,15 @@ import java.util.List;
 
 @Service
 @Slf4j
-public class FuzzPersistenceServiceImpl implements FuzzPersistenceService {
+public class KittyFuzzPersistenceServiceImpl implements KittyFuzzPersistenceService {
     @Autowired
     KittyMapper kittyMapper;
+    @Autowired
+    AflnetPersistenceService aflnetPersistenceService;
     @Override
     public List<KittyResult> getKittyResults() {
         return kittyMapper.getKittyResults();
     }
+
+
 }

fuzzIntegration/src/main/java/com/example/fuzzControll/tools/FileTools.java

@@ -3,6 +3,7 @@ package com.example.fuzzControll.tools;
 import com.example.fuzzControll.conf.AflnetProperties;
 import com.example.fuzzControll.conf.SpringContextUtil;
 import com.example.fuzzControll.exception.FileException;
+import com.example.fuzzControll.pojo.vo.AflnetResult;
 import lombok.extern.slf4j.Slf4j;
 import org.springframework.web.multipart.MultipartFile;
 
@@ -32,6 +33,12 @@ public class FileTools {
 
     }
 
+    /**
+     * 文件备份至数据库
+     *
+     * @param filename
+     * @return
+     */
     public byte[] fileReadAndBackup(String filename) {
         File file = new File(properties.getOutputPath() + "/" + filename);
         try (
@@ -52,4 +59,25 @@ public class FileTools {
             throw new RuntimeException("File is to large " + file.getName());
         }
     }
+
+    /**
+     * 读取数据库文件至本地
+     */
+    public void loadFileInLocal(AflnetResult aflnetResult, String fileLoadPath) {
+        File file = new File(fileLoadPath+"/"+aflnetResult.getFileName());
+        try (FileOutputStream fileOutputStream = new FileOutputStream(file);
+             BufferedOutputStream bufferedOutputStream = new BufferedOutputStream(fileOutputStream);
+             ByteArrayInputStream byteArrayInputStream = new ByteArrayInputStream(aflnetResult.getFile());
+             BufferedInputStream bufferedInputStream = new BufferedInputStream(byteArrayInputStream);) {
+            byte[] buffer = new byte[1024];
+            int bytesRead;
+            while ((bytesRead = bufferedInputStream.read(buffer)) != -1) {
+                bufferedOutputStream.write(buffer, 0, bytesRead);
+            }
+        } catch (FileNotFoundException e) {
+            throw new FileException("Cannot find path:" + fileLoadPath+"! Maybe there is no permission for this file!");
+        } catch (IOException e) {
+            throw new FileException("Load file failed!");
+        }
+    }
 }
\ No newline at end of file

fuzzIntegration/src/main/resources/application-dev.yml

@@ -31,7 +31,7 @@ kitty:
   vulnerabilityTypePath: "/usr/fuzzenv/fuzzenv/fuzz50/kitty/2020test/"#kitty下漏洞类型python路径
   mutationPath: "/usr/fuzzenv/fuzzenv/fuzz50/kitty/2020test/"
   templateInfoHttp: "http://127.0.0.1:26001/api/template_info.json"#模板信息请求链接
-  statsHttp: "http://127.0.0.1:26001/api/stats.json"#运行时数据
+  statsHttp: "http://127.0.0.1:26001"#运行时数据
   stagesHttp: "http://127.0.0.1:26001/api/stages.json"#
   reportHttp: "http://127.0.0.1:26001/api/report"#
 spring:

fuzzIntegration/src/main/resources/mapper/AflnetMapper.xml

@@ -11,12 +11,15 @@
         <result property="fileName"    column="fileName"    />
     </resultMap>
 
-    <sql id="selectKittyResult">
-        select id, missionId,file, fileName from kittyResult;
+    <sql id="selectAlfnetResult">
+        select id, missionId,file, fileName from alfnetResult
     </sql>
     <insert id="aflnetOutputBackup">
         insert into alfnetResult(id,missionId, file, fileName) values(#{id}, #{missionId},#{file}, #{fileName})
     </insert>
-
+    <select id="selectResultById" resultMap="AflnetResult">
+        <include refid="selectAlfnetResult"/>
+        where missionId = #{missionId};
+    </select>
 
 </mapper>
\ No newline at end of file