mirror of
https://github.com/facebook/zstd.git
synced 2025-11-04 00:02:59 -05:00
b77ad810c9
The `numFiles` variable wasn't updated, so the fuzzer didn't do anything. I did two things to fix this: 1. Remove the `numFiles` variable entirely. 2. Error if we can't open a file and print the number of files tested.
89 lines
2.6 KiB
C
89 lines
2.6 KiB
C
/*
|
|
* Copyright (c) 2016-present, Facebook, Inc.
|
|
* All rights reserved.
|
|
*
|
|
* This source code is licensed under both the BSD-style license (found in the
|
|
* LICENSE file in the root directory of this source tree) and the GPLv2 (found
|
|
* in the COPYING file in the root directory of this source tree).
|
|
*/
|
|
|
|
#include "fuzz.h"
|
|
#include "fuzz_helpers.h"
|
|
#include "util.h"
|
|
#include <stddef.h>
|
|
#include <stdint.h>
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
|
|
int main(int argc, char const **argv) {
|
|
size_t const kMaxFileSize = (size_t)1 << 27;
|
|
int const kFollowLinks = 1;
|
|
FileNamesTable* files;
|
|
const char** const fnTable = argv + 1;
|
|
uint8_t *buffer = NULL;
|
|
size_t bufferSize = 0;
|
|
unsigned i;
|
|
unsigned numFilesTested = 0;
|
|
int ret = 0;
|
|
|
|
{
|
|
unsigned const numFiles = (unsigned)(argc - 1);
|
|
#ifdef UTIL_HAS_CREATEFILELIST
|
|
files = UTIL_createExpandedFNT(fnTable, numFiles, kFollowLinks);
|
|
#else
|
|
files = UTIL_createFNT_fromROTable(fnTable, numFiles);
|
|
assert(numFiles == files->tableSize);
|
|
#endif
|
|
}
|
|
if (!files) {
|
|
fprintf(stderr, "ERROR: Failed to create file names table\n");
|
|
return 1;
|
|
}
|
|
if (files->tableSize == 0)
|
|
fprintf(stderr, "WARNING: No files passed to %s\n", argv[0]);
|
|
for (i = 0; i < files->tableSize; ++i) {
|
|
char const *fileName = files->fileNames[i];
|
|
DEBUGLOG(3, "Running %s", fileName);
|
|
size_t const fileSize = UTIL_getFileSize(fileName);
|
|
size_t readSize;
|
|
FILE *file;
|
|
|
|
/* Check that it is a regular file, and that the fileSize is valid.
|
|
* If it is not a regular file, then it may have been deleted since we
|
|
* constructed the list, so just skip it, but return an error exit code.
|
|
*/
|
|
if (!UTIL_isRegularFile(fileName)) {
|
|
ret = 1;
|
|
continue;
|
|
}
|
|
FUZZ_ASSERT_MSG(fileSize <= kMaxFileSize, fileName);
|
|
/* Ensure we have a large enough buffer allocated */
|
|
if (fileSize > bufferSize) {
|
|
free(buffer);
|
|
buffer = (uint8_t *)malloc(fileSize);
|
|
FUZZ_ASSERT_MSG(buffer, fileName);
|
|
bufferSize = fileSize;
|
|
}
|
|
/* Open the file */
|
|
file = fopen(fileName, "rb");
|
|
FUZZ_ASSERT_MSG(file, fileName);
|
|
/* Read the file */
|
|
readSize = fread(buffer, 1, fileSize, file);
|
|
FUZZ_ASSERT_MSG(readSize == fileSize, fileName);
|
|
/* Close the file */
|
|
fclose(file);
|
|
/* Run the fuzz target */
|
|
LLVMFuzzerTestOneInput(buffer, fileSize);
|
|
++numFilesTested;
|
|
}
|
|
fprintf(stderr, "Tested %u files: ", numFilesTested);
|
|
if (ret == 0) {
|
|
fprintf(stderr, "Success!\n");
|
|
} else {
|
|
fprintf(stderr, "Failure!\n");
|
|
}
|
|
free(buffer);
|
|
UTIL_freeFileNamesTable(files);
|
|
return ret;
|
|
}
|