wasm-async.cc 3.19 KB
Newer Older
1 2 3 4 5 6 7 8 9
// Copyright 2017 the V8 project authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include <limits.h>
#include <stddef.h>
#include <stdint.h>

#include "include/v8.h"
10
#include "src/api/api.h"
11
#include "src/execution/isolate-inl.h"
12
#include "src/heap/factory.h"
13
#include "src/objects/objects-inl.h"
14
#include "src/wasm/wasm-engine.h"
15 16 17 18
#include "src/wasm/wasm-module.h"
#include "test/common/wasm/flag-utils.h"
#include "test/common/wasm/wasm-module-runner.h"
#include "test/fuzzer/fuzzer-support.h"
19
#include "test/fuzzer/wasm-fuzzer-common.h"
20

21 22 23
namespace v8 {
namespace internal {
class WasmModuleObject;
24

25 26
namespace wasm {
namespace fuzzer {
27

28 29 30 31 32 33 34 35 36
class AsyncFuzzerResolver : public i::wasm::CompilationResultResolver {
 public:
  AsyncFuzzerResolver(i::Isolate* isolate, bool* done)
      : isolate_(isolate), done_(done) {}

  void OnCompilationSucceeded(i::Handle<i::WasmModuleObject> module) override {
    *done_ = true;
    InterpretAndExecuteModule(isolate_, module);
  }
37

38 39 40
  void OnCompilationFailed(i::Handle<i::Object> error_reason) override {
    *done_ = true;
  }
41

42 43 44 45
 private:
  i::Isolate* isolate_;
  bool* done_;
};
46 47

extern "C" int LLVMFuzzerTestOneInput(const uint8_t* data, size_t size) {
48 49 50 51 52 53 54 55
  // We explicitly enable staged WebAssembly features here to increase fuzzer
  // coverage. For libfuzzer fuzzers it is not possible that the fuzzer enables
  // the flag by itself.
#define ENABLE_STAGED_FEATURES(feat, desc, val) \
  i::FlagScope<bool> enable_##feat(&i::FLAG_experimental_wasm_##feat, true);
  FOREACH_WASM_STAGING_FEATURE_FLAG(ENABLE_STAGED_FEATURES)
#undef ENABLE_STAGED_FEATURES

56
  FlagScope<bool> turn_on_async_compile(
57
      &v8::internal::FLAG_wasm_async_compilation, true);
58 59 60
  FlagScope<uint32_t> max_mem_flag_scope(&v8::internal::FLAG_wasm_max_mem_pages,
                                         32);
  FlagScope<uint32_t> max_table_size_scope(
61 62 63
      &v8::internal::FLAG_wasm_max_table_size, 100);
  v8_fuzzer::FuzzerSupport* support = v8_fuzzer::FuzzerSupport::Get();
  v8::Isolate* isolate = support->GetIsolate();
64
  i::Isolate* i_isolate = reinterpret_cast<v8::internal::Isolate*>(isolate);
65 66 67 68 69 70 71 72

  // Clear any pending exceptions from a prior run.
  if (i_isolate->has_pending_exception()) {
    i_isolate->clear_pending_exception();
  }

  v8::Isolate::Scope isolate_scope(isolate);
  v8::HandleScope handle_scope(isolate);
73
  i::HandleScope internal_scope(i_isolate);
74
  v8::Context::Scope context_scope(support->GetContext());
75 76
  TryCatch try_catch(isolate);
  testing::SetupIsolateForWasmModule(i_isolate);
77

78
  bool done = false;
79
  auto enabled_features = i::wasm::WasmFeatures::FromIsolate(i_isolate);
80
  constexpr const char* kAPIMethodName = "WasmAsyncFuzzer.compile";
81
  i_isolate->wasm_engine()->AsyncCompile(
82
      i_isolate, enabled_features,
83
      std::make_shared<AsyncFuzzerResolver>(i_isolate, &done),
84
      ModuleWireBytes(data, data + size), false, kAPIMethodName);
85 86

  // Wait for the promise to resolve.
87
  while (!done) {
88
    support->PumpMessageLoop(platform::MessageLoopBehavior::kWaitForWork);
89
    isolate->PerformMicrotaskCheckpoint();
90 91 92
  }
  return 0;
}
93 94 95 96 97

}  // namespace fuzzer
}  // namespace wasm
}  // namespace internal
}  // namespace v8