deps/libchdr/deps/zstd-1.5.6/tests/fuzz/regression_driver.c

   1 /*
   2  * Copyright (c) Meta Platforms, Inc. and affiliates.
   3  * All rights reserved.
   4  *
   5  * This source code is licensed under both the BSD-style license (found in the
   6  * LICENSE file in the root directory of this source tree) and the GPLv2 (found
   7  * in the COPYING file in the root directory of this source tree).
   8  * You may select, at your option, one of the above-listed licenses.
   9  */
  10
  11 #include "fuzz.h"
  12 #include "fuzz_helpers.h"
  13 #include "util.h"
  14 #include <stddef.h>
  15 #include <stdint.h>
  16 #include <stdio.h>
  17 #include <stdlib.h>
  18
  19 int main(int argc, char const **argv) {
  20   size_t const kMaxFileSize = (size_t)1 << 27;
  21   int const kFollowLinks = 1;
  22   FileNamesTable* files;
  23   const char** const fnTable = argv + 1;
  24   uint8_t *buffer = NULL;
  25   size_t bufferSize = 0;
  26   unsigned i;
  27   unsigned numFilesTested = 0;
  28   int ret = 0;
  29
  30   {
  31     unsigned const numFiles = (unsigned)(argc - 1);
  32 #ifdef UTIL_HAS_CREATEFILELIST
  33     files = UTIL_createExpandedFNT(fnTable, numFiles, kFollowLinks);
  34 #else
  35     files = UTIL_createFNT_fromROTable(fnTable, numFiles);
  36     assert(numFiles == files->tableSize);
  37 #endif
  38   }
  39   if (!files) {
  40     fprintf(stderr, "ERROR: Failed to create file names table\n");
  41     return 1;
  42   }
  43   if (files->tableSize == 0)
  44     fprintf(stderr, "WARNING: No files passed to %s\n", argv[0]);
  45   for (i = 0; i < files->tableSize; ++i) {
  46     char const *fileName = files->fileNames[i];
  47     size_t const fileSize = UTIL_getFileSize(fileName);
  48     size_t readSize;
  49     FILE *file;
  50
  51     DEBUGLOG(3, "Running %s", fileName);
  52
  53     /* Check that it is a regular file, and that the fileSize is valid.
  54      * If it is not a regular file, then it may have been deleted since we
  55      * constructed the list, so just skip it, but return an error exit code.
  56      */
  57     if (!UTIL_isRegularFile(fileName)) {
  58       ret = 1;
  59       continue;
  60     }
  61     FUZZ_ASSERT_MSG(fileSize <= kMaxFileSize, fileName);
  62     /* Ensure we have a large enough buffer allocated */
  63     if (fileSize > bufferSize) {
  64       free(buffer);
  65       buffer = (uint8_t *)malloc(fileSize);
  66       FUZZ_ASSERT_MSG(buffer, fileName);
  67       bufferSize = fileSize;
  68     }
  69     /* Open the file */
  70     file = fopen(fileName, "rb");
  71     FUZZ_ASSERT_MSG(file, fileName);
  72     /* Read the file */
  73     readSize = fread(buffer, 1, fileSize, file);
  74     FUZZ_ASSERT_MSG(readSize == fileSize, fileName);
  75     /* Close the file */
  76     fclose(file);
  77     /* Run the fuzz target */
  78     LLVMFuzzerTestOneInput(buffer, fileSize);
  79     ++numFilesTested;
  80   }
  81   fprintf(stderr, "Tested %u files: ", numFilesTested);
  82   if (ret == 0) {
  83     fprintf(stderr, "Success!\n");
  84   } else {
  85     fprintf(stderr, "Failure!\n");
  86   }
  87   free(buffer);
  88   UTIL_freeFileNamesTable(files);
  89   return ret;
  90 }