wasm-async.cc 3.98 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.h"
11 12 13 14 15
#include "src/factory.h"
#include "src/isolate-inl.h"
#include "src/isolate.h"
#include "src/objects-inl.h"
#include "src/objects.h"
16
#include "src/wasm/module-compiler.h"
17
#include "src/wasm/wasm-api.h"
18 19 20 21
#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"
22
#include "test/fuzzer/wasm-fuzzer-common.h"
23

24 25 26
namespace v8 {
namespace internal {
class WasmModuleObject;
27

28 29
namespace wasm {
namespace fuzzer {
30

31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50
#define ASSIGN(type, var, expr)                      \
  v8::Local<type> var;                               \
  do {                                               \
    if (!expr.ToLocal(&var)) {                       \
      DCHECK(i_isolate->has_scheduled_exception());  \
      return 0;                                      \
    } else {                                         \
      DCHECK(!i_isolate->has_scheduled_exception()); \
    }                                                \
  } while (false)

namespace {
// We need this helper function because we cannot use
// Handle<WasmModuleObject>::cast here. To use this function we would have to
// mark it with V8_EXPORT_PRIVATE, which is quite ugly in this case.
Handle<WasmModuleObject> ToWasmModuleObjectUnchecked(Handle<Object> that) {
  return handle(reinterpret_cast<WasmModuleObject*>(*that));
}
}

51
void InstantiateCallback(const FunctionCallbackInfo<Value>& args) {
52 53
  DCHECK_GE(args.Length(), 1);
  v8::Isolate* isolate = args.GetIsolate();
54
  MicrotasksScope does_not_run_microtasks(
55 56 57 58
      isolate, v8::MicrotasksScope::kDoNotRunMicrotasks);

  v8::HandleScope scope(isolate);

59
  Local<v8::Value> module = args[0];
60 61 62

  i::Isolate* i_isolate = reinterpret_cast<i::Isolate*>(isolate);

63 64
  Handle<WasmModuleObject> module_obj =
      ToWasmModuleObjectUnchecked(Utils::OpenHandle(v8::Object::Cast(*module)));
65
  InterpretAndExecuteModule(i_isolate, module_obj);
66 67 68
}

extern "C" int LLVMFuzzerTestOneInput(const uint8_t* data, size_t size) {
69
  FlagScope<bool> turn_on_async_compile(
70
      &v8::internal::FLAG_wasm_async_compilation, true);
71 72 73
  FlagScope<uint32_t> max_mem_flag_scope(&v8::internal::FLAG_wasm_max_mem_pages,
                                         32);
  FlagScope<uint32_t> max_table_size_scope(
74 75 76
      &v8::internal::FLAG_wasm_max_table_size, 100);
  v8_fuzzer::FuzzerSupport* support = v8_fuzzer::FuzzerSupport::Get();
  v8::Isolate* isolate = support->GetIsolate();
77
  i::Isolate* i_isolate = reinterpret_cast<v8::internal::Isolate*>(isolate);
78 79 80 81 82 83 84 85

  // 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);
86
  i::HandleScope internal_scope(i_isolate);
87
  v8::Context::Scope context_scope(support->GetContext());
88 89
  TryCatch try_catch(isolate);
  testing::SetupIsolateForWasmModule(i_isolate);
90 91

  // Get the promise for async compilation.
92 93 94
  ASSIGN(Promise::Resolver, resolver,
         Promise::Resolver::New(support->GetContext()));
  Local<Promise> promise = resolver->GetPromise();
95

96
  AsyncCompile(i_isolate, Utils::OpenHandle(*promise),
97 98
               ModuleWireBytes(data, data + size));

99 100 101
  ASSIGN(Function, instantiate_impl,
         Function::New(support->GetContext(), &InstantiateCallback,
                       Undefined(isolate)));
102

103
  ASSIGN(Promise, result,
104 105 106
         promise->Then(support->GetContext(), instantiate_impl));

  // Wait for the promise to resolve.
107 108
  while (result->State() == Promise::kPending) {
    support->PumpMessageLoop(platform::MessageLoopBehavior::kWaitForWork);
109 110 111 112
    isolate->RunMicrotasks();
  }
  return 0;
}
113

114 115
#undef ASSIGN

116 117 118 119
}  // namespace fuzzer
}  // namespace wasm
}  // namespace internal
}  // namespace v8