libFuzzer_dummy_main.cpp

Go to the documentation of this file.

/*
    RawSpeed - RAW file decoder.
 
    Copyright (C) 2017 Roman Lebedev
 
    This library is free software; you can redistribute it and/or
    modify it under the terms of the GNU Lesser General Public
    License as published by the Free Software Foundation; either
    version 2 of the License, or (at your option) any later version.
 
    This library is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
    Lesser General Public License for more details.
 
    You should have received a copy of the GNU Lesser General Public
    License along with this library; if not, write to the Free Software
    Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
*/
 
#include "rawspeedconfig.h"
#include "adt/AlignedAllocator.h"
#include "adt/Array1DRef.h"
#include "adt/DefaultInitAllocatorAdaptor.h"
#include "io/Buffer.h"
#include "io/FileIOException.h"
#include "io/FileReader.h"
#include <algorithm>
#include <cstdint>
#include <cstdlib>
#include <iostream>
#include <memory>
#include <string>
#include <tuple>
#include <vector>
 
#ifdef HAVE_OPENMP
#include <omp.h>
#endif
 
extern "C" int LLVMFuzzerTestOneInput(const uint8_t* Data, size_t Size);
 
namespace {
 
int usage() {
  std::cout << "This is just a placeholder.\nFor fuzzers to actually function, "
               "you need to build rawspeed with clang compiler, with FUZZ "
               "build type.\n";
 
  return EXIT_SUCCESS;
}
 
void process(const char* filename) noexcept {
  rawspeed::FileReader reader(filename);
  std::unique_ptr<std::vector<
      uint8_t, rawspeed::DefaultInitAllocatorAdaptor<
                   uint8_t, rawspeed::AlignedAllocator<uint8_t, 16>>>>
      storage;
  rawspeed::Buffer buf;
 
  try {
    std::tie(storage, buf) = reader.readFile();
  } catch (const rawspeed::FileIOException&) {
    // failed to read the file for some reason.
    // just ignore it.
    return;
  }
 
  LLVMFuzzerTestOneInput(buf.begin(), buf.getSize());
}
 
} // namespace
 
int main(int argc_, char** argv_) {
  auto argv = rawspeed::Array1DRef(argv_, argc_);
 
  if (1 == argv.size() ||
      (2 == argv.size() && std::string("-help=1") == argv(1)))
    return usage();
 
#ifdef HAVE_OPENMP
  const auto corpusCount = argv.size() - 1;
  auto chunkSize = (corpusCount / (10 * omp_get_num_threads()));
  chunkSize = std::max(chunkSize, 1);
#pragma omp parallel for default(none) firstprivate(argv, chunkSize)           \
    schedule(dynamic, chunkSize)
#endif
  for (int i = 1; i < argv.size(); ++i)
    process(argv(i));
 
  return EXIT_SUCCESS;
}