Impl. being able to set dir for temporary files

src/archiver.c

@@ -39,7 +39,7 @@
 
 #include "helpers.h"
 
-#define TEMP_FILENAME_CMP "simple_archiver_compressed_%u.tmp"
+#define TEMP_FILENAME_CMP "%s%ssimple_archiver_compressed_%u.tmp"
 
 #if SIMPLE_ARCHIVER_PLATFORM == SIMPLE_ARCHIVER_PLATFORM_COSMOPOLITAN || \
     SIMPLE_ARCHIVER_PLATFORM == SIMPLE_ARCHIVER_PLATFORM_MAC ||          \
@@ -168,15 +168,21 @@ int write_files_fn(void *data, void *ud) {
     SIMPLE_ARCHIVER_PLATFORM == SIMPLE_ARCHIVER_PLATFORM_MAC ||          \
     SIMPLE_ARCHIVER_PLATFORM == SIMPLE_ARCHIVER_PLATFORM_LINUX
       // Use temp file to store compressed data.
-      char temp_filename[128];
+      char temp_filename[512];
       unsigned int idx = 0;
-      snprintf(temp_filename, 128, TEMP_FILENAME_CMP, idx);
+      unsigned int temp_dir_end = strlen(state->parsed->temp_dir);
+      snprintf(temp_filename, 512, TEMP_FILENAME_CMP, state->parsed->temp_dir,
+               state->parsed->temp_dir[temp_dir_end - 1] == '/' ? "" : "/",
+               idx);
       do {
         FILE *test_fd = fopen(temp_filename, "rb");
         if (test_fd) {
           // File exists.
           fclose(test_fd);
-          snprintf(temp_filename, 128, TEMP_FILENAME_CMP, ++idx);
+          snprintf(temp_filename, 512, TEMP_FILENAME_CMP,
+                   state->parsed->temp_dir,
+                   state->parsed->temp_dir[temp_dir_end - 1] == '/' ? "" : "/",
+                   ++idx);
         } else if (idx > 0xFFFF) {
           // Sanity check.
           return 1;
@@ -192,14 +198,14 @@ int write_files_fn(void *data, void *ud) {
       }
       __attribute__((cleanup(free_FILE_helper))) FILE *tmp_fd =
           fopen(temp_filename, "wb");
+      if (!tmp_fd) {
+        fprintf(stderr, "ERROR: Unable to create temp file for compressing!\n");
+        return 1;
+      }
       __attribute__((cleanup(cleanup_temp_filename_delete))) void **ptrs_array =
           malloc(sizeof(void *) * 2);
       ptrs_array[0] = temp_filename;
       ptrs_array[1] = &tmp_fd;
-      if (!tmp_fd) {
-        // Unable to create temp file.
-        return 1;
-      }
 
       // Handle SIGPIPE.
       signal(SIGPIPE, handle_sig_pipe);

src/main.c

@@ -44,7 +44,10 @@ int main(int argc, const char **argv) {
       cleanup(simple_archiver_free_parsed))) SDArchiverParsed parsed =
       simple_archiver_create_parsed();
 
-  simple_archiver_parse_args(argc, argv, &parsed);
+  if (simple_archiver_parse_args(argc, argv, &parsed)) {
+    fprintf(stderr, "Failed to parse args.\n");
+    return 7;
+  }
 
   if (!parsed.filename && (parsed.flags & 0x10) == 0) {
     fprintf(stderr, "ERROR: Filename not specified!\n");

src/parser.c

@@ -153,6 +153,9 @@ void simple_archiver_print_usage(void) {
   fprintf(stderr, "--overwrite-extract : allows overwriting when extracting\n");
   fprintf(stderr,
           "--no-abs-symlink : do not store absolute paths for symlinks\n");
+  fprintf(stderr,
+          "--temp-files-dir <dir> : where to store temporary files created "
+          "when compressing (defaults to current working directory)\n");
   fprintf(stderr,
           "-- : specifies remaining arguments are files to archive/extract\n");
   fprintf(
@@ -171,6 +174,7 @@ SDArchiverParsed simple_archiver_create_parsed(void) {
   parsed.compressor = NULL;
   parsed.decompressor = NULL;
   parsed.working_files = NULL;
+  parsed.temp_dir = NULL;
 
   return parsed;
 }
@@ -199,6 +203,7 @@ int simple_archiver_parse_args(int argc, const char **argv,
   while (argc > 0) {
     if (!is_remaining_args) {
       if (strcmp(argv[0], "-h") == 0 || strcmp(argv[0], "--help") == 0) {
+        simple_archiver_free_parsed(out);
         simple_archiver_print_usage();
         exit(0);
       } else if (strcmp(argv[0], "-c") == 0) {
@@ -214,7 +219,12 @@ int simple_archiver_parse_args(int argc, const char **argv,
         out->flags &= 0xFFFFFFFC;
         // set second bit.
         out->flags |= 0x2;
-      } else if (strcmp(argv[0], "-f") == 0 && argc > 1) {
+      } else if (strcmp(argv[0], "-f") == 0) {
+        if (argc < 2) {
+          fprintf(stderr, "ERROR: -f specified but missing argument!\n");
+          simple_archiver_print_usage();
+          return 1;
+        }
         if (strcmp(argv[1], "-") == 0) {
           out->flags |= 0x10;
           if (out->filename) {
@@ -229,13 +239,23 @@ int simple_archiver_parse_args(int argc, const char **argv,
         }
         --argc;
         ++argv;
-      } else if (strcmp(argv[0], "--compressor") == 0 && argc > 1) {
+      } else if (strcmp(argv[0], "--compressor") == 0) {
+        if (argc < 2) {
+          fprintf(stderr, "--compressor specfied but missing argument!\n");
+          simple_archiver_print_usage();
+          return 1;
+        }
         int size = strlen(argv[1]) + 1;
         out->compressor = malloc(size);
         strncpy(out->compressor, argv[1], size);
         --argc;
         ++argv;
-      } else if (strcmp(argv[0], "--decompressor") == 0 && argc > 1) {
+      } else if (strcmp(argv[0], "--decompressor") == 0) {
+        if (argc < 2) {
+          fprintf(stderr, "--decompressor specfied but missing argument!\n");
+          simple_archiver_print_usage();
+          return 1;
+        }
         int size = strlen(argv[1]) + 1;
         out->decompressor = malloc(size);
         strncpy(out->decompressor, argv[1], size);
@@ -247,6 +267,15 @@ int simple_archiver_parse_args(int argc, const char **argv,
         out->flags |= 0x8;
       } else if (strcmp(argv[0], "--no-abs-symlink") == 0) {
         out->flags |= 0x20;
+      } else if (strcmp(argv[0], "--temp-files-dir") == 0) {
+        if (argc < 2) {
+          fprintf(stderr, "ERROR: --temp-files-dir is missing an argument!\n");
+          simple_archiver_print_usage();
+          return 1;
+        }
+        out->temp_dir = argv[1];
+        --argc;
+        ++argv;
       } else if (argv[0][0] == '-' && argv[0][1] == '-' && argv[0][2] == 0) {
         is_remaining_args = 1;
       } else if (argv[0][0] != '-') {
@@ -292,6 +321,10 @@ int simple_archiver_parse_args(int argc, const char **argv,
     ++argv;
   }
 
+  if (!out->temp_dir) {
+    out->temp_dir = "./";
+  }
+
   return 0;
 }
 

src/parser.h

@@ -42,6 +42,9 @@ typedef struct SDArchiverParsed {
   /// Last entry should be NULL.
   /// Determines a "white-list" of files to extract when extracting.
   char **working_files;
+  /// Determines where to place temporary files. If NULL, temporary files are
+  /// created in the current working directory.
+  const char *temp_dir;
 } SDArchiverParsed;
 
 typedef struct SDArchiverFileInfo {
@@ -57,6 +60,7 @@ SDArchiverParsed simple_archiver_create_parsed(void);
 /// Expects the user to pass a pointer to an SDArchiverParsed.
 /// This means the user should have a SDArchiverParsed variable
 /// and it should be passed with e.g. "&var".
+/// Returns 0 on success.
 int simple_archiver_parse_args(int argc, const char **argv,
                                SDArchiverParsed *out);