disassembler_win32_fuzzer.cc 1.5 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152
  1. // Copyright 2018 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 <map>
  7. #include <memory>
  8. #include <vector>
  9. #include "base/logging.h"
  10. #include "components/zucchini/buffer_view.h"
  11. #include "components/zucchini/disassembler.h"
  12. #include "components/zucchini/disassembler_win32.h"
  13. #include "components/zucchini/fuzzers/fuzz_utils.h"
  14. namespace {
  15. struct Environment {
  16. Environment() {
  17. logging::SetMinLogLevel(logging::LOG_FATAL); // Disable console spamming.
  18. }
  19. };
  20. } // namespace
  21. // Entry point for LibFuzzer.
  22. extern "C" int LLVMFuzzerTestOneInput(const uint8_t* data, size_t size) {
  23. static Environment env;
  24. if (!size)
  25. return 0;
  26. // Prepare data.
  27. std::vector<uint8_t> mutable_data(data, data + size);
  28. zucchini::ConstBufferView image(mutable_data.data(), mutable_data.size());
  29. // One of x86 or x64 should return a non-nullptr if the data is valid.
  30. auto disassembler_win32x86 =
  31. zucchini::Disassembler::Make<zucchini::DisassemblerWin32X86>(image);
  32. if (disassembler_win32x86) {
  33. zucchini::ReadAndWriteReferences(std::move(disassembler_win32x86),
  34. &mutable_data);
  35. return 0;
  36. }
  37. auto disassembler_win32x64 =
  38. zucchini::Disassembler::Make<zucchini::DisassemblerWin32X64>(image);
  39. if (disassembler_win32x64)
  40. zucchini::ReadAndWriteReferences(std::move(disassembler_win32x64),
  41. &mutable_data);
  42. return 0;
  43. }