From: Stephen Seo <seo.disparate@gmail.com>
Date: Tue, 8 Oct 2024 06:31:40 +0000 (+0900)
Subject: Add validation of files/symlinks count from files
X-Git-Tag: 1.1~6
X-Git-Url: https://git.seodisparate.com/tbm-server-edit-full.png?a=commitdiff_plain;h=78ba5acd70d8818b69c16235103f223576fed9fe;p=SimpleArchiver

Add validation of files/symlinks count from files
---

diff --git a/src/archiver.c b/src/archiver.c
index b3634ab..52db646 100644
--- a/src/archiver.c
+++ b/src/archiver.c
@@ -1861,6 +1861,13 @@ int simple_archiver_write_v1(FILE *out_f, SDArchiverState *state,
     simple_archiver_priority_heap_free(&files_pheap);
   }
 
+  if (symlinks_list->count + files_list->count != filenames->count) {
+    fprintf(stderr,
+            "ERROR: Count mismatch between files and symlinks and files from "
+            "parser!\n");
+    return SDAS_INTERNAL_ERROR;
+  }
+
   if (fwrite("SIMPLE_ARCHIVE_VER", 1, 18, out_f) != 18) {
     return SDAS_FAILED_TO_WRITE;
   }