disassembler_elf_fuzzer.cc 1.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445
  1. // Copyright 2019 The Chromium Authors. All rights reserved.
  2. // Use of this source code is governed by a BSD-style license that can be
  3. // found in the LICENSE file.
  4. #include <stddef.h>
  5. #include <stdint.h>
  6. #include "base/logging.h"
  7. #include "components/zucchini/buffer_view.h"
  8. #include "components/zucchini/disassembler.h"
  9. #include "components/zucchini/disassembler_elf.h"
  10. #include "components/zucchini/fuzzers/fuzz_utils.h"
  11. namespace {
  12. struct Environment {
  13. Environment() { logging::SetMinLogLevel(logging::LOG_FATAL); }
  14. };
  15. } // namespace
  16. extern "C" int LLVMFuzzerTestOneInput(const uint8_t* data, size_t size) {
  17. static Environment env;
  18. if (!size)
  19. return 0;
  20. // Prepare data.
  21. std::vector<uint8_t> mutable_data(data, data + size);
  22. zucchini::ConstBufferView image(mutable_data.data(), mutable_data.size());
  23. // Create disassembler. Early exit on failure.
  24. auto disassembler_elf_x64 =
  25. zucchini::Disassembler::Make<zucchini::DisassemblerElfX64>(image);
  26. if (disassembler_elf_x64) {
  27. zucchini::ReadAndWriteReferences(std::move(disassembler_elf_x64),
  28. &mutable_data);
  29. return 0;
  30. }
  31. auto disassembler_elf_x86 =
  32. zucchini::Disassembler::Make<zucchini::DisassemblerElfX86>(image);
  33. if (disassembler_elf_x86)
  34. zucchini::ReadAndWriteReferences(std::move(disassembler_elf_x86),
  35. &mutable_data);
  36. return 0;
  37. }