feat: 支持漏洞关联多条流量证据

server/task_archives.go

@@ -178,6 +178,13 @@ func (s *Server) archiveTask(job *pgdb.TaskArchive) (runErr error) {
 		snapshot.DataCounts["traffic"] = trafficCount
 	}
 	archivePath := taskArchivePath(s.m.dir, job.ID, taskID)
+	evidenceSnapshots, err := pgdb.ArchiveEvidenceSnapshots(snapshot)
+	if err != nil {
+		return err
+	}
+	if err = s.evidenceStore().CopySnapshots(s.ctx, evidenceSnapshots, filepath.Join(fileStage.payload, "evidence")); err != nil {
+		return err
+	}
 	_ = s.m.pg.UpdateTaskArchiveProgress(job.ID, "compress", 50)
 	originalSize, compressedSize, checksum, err := writeTaskArchivePackage(archivePath, fileStage.payload, snapshot)
 	if err != nil {
@@ -270,6 +277,17 @@ func (s *Server) restoreTaskArchive(job *pgdb.TaskArchive) (runErr error) {
 	if snapshot.TaskID != job.TaskID || !pgdb.IsTaskArchiveFormatSupported(snapshot.FormatVersion) {
 		return pgdb.ErrTaskArchiveFormatMismatch
 	}
+	evidenceSnapshots, err := pgdb.ArchiveEvidenceSnapshots(&snapshot)
+	if err != nil {
+		return err
+	}
+	return s.evidenceStore().WithInstalledSnapshots(s.ctx, evidenceSnapshots, filepath.Join(extracted, "evidence"), func() error {
+		return s.restoreTaskArchivePayload(job, &snapshot, extracted)
+	})
+}
+
+func (s *Server) restoreTaskArchivePayload(job *pgdb.TaskArchive, snapshot *pgdb.TaskArchiveSnapshot, extracted string) (runErr error) {
+	var err error
 	var llmRecordsFile *os.File
 	relative, hasStreamedLLMRecords := snapshot.StreamedTables["llm_records"]
 	if len(snapshot.StreamedTables) > 0 &&
@@ -309,13 +327,13 @@ func (s *Server) restoreTaskArchive(job *pgdb.TaskArchive) (runErr error) {
 	var warnings []string
 	if llmRecordsFile != nil {
 		warnings, err = s.m.pg.RestoreTaskArchiveWithLLMRecords(
-			job.ID, &snapshot, job.RemainingTimeoutSeconds, llmRecordsFile,
+			job.ID, snapshot, job.RemainingTimeoutSeconds, llmRecordsFile,
 		)
 		closeErr := llmRecordsFile.Close()
 		llmRecordsFile = nil
 		err = errors.Join(err, closeErr)
 	} else {
-		warnings, err = s.m.pg.RestoreTaskArchive(job.ID, &snapshot, job.RemainingTimeoutSeconds)
+		warnings, err = s.m.pg.RestoreTaskArchive(job.ID, snapshot, job.RemainingTimeoutSeconds)
 	}
 	if err != nil {
 		return err