Line data    Source code

       1             : // Copyright 2017 the V8 project 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             : 
       5             : #include <limits.h>
       6             : #include <stddef.h>
       7             : #include <stdint.h>
       8             : 
       9             : #include "include/v8.h"
      10             : #include "src/api.h"
      11             : #include "src/heap/factory.h"
      12             : #include "src/isolate-inl.h"
      13             : #include "src/objects-inl.h"
      14             : #include "src/wasm/wasm-engine.h"
      15             : #include "src/wasm/wasm-module.h"
      16             : #include "test/common/wasm/flag-utils.h"
      17             : #include "test/common/wasm/wasm-module-runner.h"
      18             : #include "test/fuzzer/fuzzer-support.h"
      19             : #include "test/fuzzer/wasm-fuzzer-common.h"
      20             : 
      21             : namespace v8 {
      22             : namespace internal {
      23             : class WasmModuleObject;
      24             : 
      25             : namespace wasm {
      26             : namespace fuzzer {
      27             : 
      28           6 : class AsyncFuzzerResolver : public i::wasm::CompilationResultResolver {
      29             :  public:
      30             :   AsyncFuzzerResolver(i::Isolate* isolate, bool* done)
      31           3 :       : isolate_(isolate), done_(done) {}
      32             : 
      33           2 :   void OnCompilationSucceeded(i::Handle<i::WasmModuleObject> module) override {
      34           2 :     *done_ = true;
      35           2 :     InterpretAndExecuteModule(isolate_, module);
      36           2 :   }
      37             : 
      38           1 :   void OnCompilationFailed(i::Handle<i::Object> error_reason) override {
      39           1 :     *done_ = true;
      40           1 :   }
      41             : 
      42             :  private:
      43             :   i::Isolate* isolate_;
      44             :   bool* done_;
      45             : };
      46             : 
      47           3 : extern "C" int LLVMFuzzerTestOneInput(const uint8_t* data, size_t size) {
      48             :   FlagScope<bool> turn_on_async_compile(
      49             :       &v8::internal::FLAG_wasm_async_compilation, true);
      50             :   FlagScope<uint32_t> max_mem_flag_scope(&v8::internal::FLAG_wasm_max_mem_pages,
      51             :                                          32);
      52             :   FlagScope<uint32_t> max_table_size_scope(
      53             :       &v8::internal::FLAG_wasm_max_table_size, 100);
      54           3 :   v8_fuzzer::FuzzerSupport* support = v8_fuzzer::FuzzerSupport::Get();
      55             :   v8::Isolate* isolate = support->GetIsolate();
      56           3 :   i::Isolate* i_isolate = reinterpret_cast<v8::internal::Isolate*>(isolate);
      57             : 
      58             :   // Clear any pending exceptions from a prior run.
      59           3 :   if (i_isolate->has_pending_exception()) {
      60             :     i_isolate->clear_pending_exception();
      61             :   }
      62             : 
      63             :   v8::Isolate::Scope isolate_scope(isolate);
      64           6 :   v8::HandleScope handle_scope(isolate);
      65           3 :   i::HandleScope internal_scope(i_isolate);
      66           3 :   v8::Context::Scope context_scope(support->GetContext());
      67           6 :   TryCatch try_catch(isolate);
      68           3 :   testing::SetupIsolateForWasmModule(i_isolate);
      69             : 
      70           3 :   bool done = false;
      71           3 :   auto enabled_features = i::wasm::WasmFeaturesFromIsolate(i_isolate);
      72           9 :   i_isolate->wasm_engine()->AsyncCompile(
      73             :       i_isolate, enabled_features,
      74           6 :       std::make_shared<AsyncFuzzerResolver>(i_isolate, &done),
      75           3 :       ModuleWireBytes(data, data + size), false);
      76             : 
      77             :   // Wait for the promise to resolve.
      78          13 :   while (!done) {
      79           5 :     support->PumpMessageLoop(platform::MessageLoopBehavior::kWaitForWork);
      80           5 :     isolate->RunMicrotasks();
      81             :   }
      82           3 :   return 0;
      83             : }
      84             : 
      85             : }  // namespace fuzzer
      86             : }  // namespace wasm
      87             : }  // namespace internal
      88           6 : }  // namespace v8