test-factory.cc 3.48 KB
Newer Older
1 2 3 4
// Copyright 2018 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.

5 6
#include <memory>

7 8
#include "include/v8.h"

9
#include "src/codegen/code-desc.h"
10
#include "src/execution/isolate.h"
11
#include "src/handles/handles-inl.h"
12 13 14 15 16 17
#include "test/cctest/cctest.h"

namespace v8 {
namespace internal {
namespace test_factory {

18 19 20 21 22
namespace {

// This needs to be large enough to create a new nosnap Isolate, but smaller
// than kMaximalCodeRangeSize so we can recover from the OOM.
constexpr int kInstructionSize = 100 * MB;
23 24
STATIC_ASSERT(kInstructionSize < kMaximalCodeRangeSize ||
              !kPlatformRequiresCodeRange);
25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57

size_t NearHeapLimitCallback(void* raw_bool, size_t current_heap_limit,
                             size_t initial_heap_limit) {
  bool* oom_triggered = static_cast<bool*>(raw_bool);
  *oom_triggered = true;
  return kInstructionSize * 2;
}

class SetupIsolateWithSmallHeap {
 public:
  SetupIsolateWithSmallHeap() {
    FLAG_max_old_space_size = kInstructionSize / MB / 2;  // In MB.
    v8::Isolate::CreateParams create_params;
    create_params.array_buffer_allocator = CcTest::array_buffer_allocator();
    isolate_ = reinterpret_cast<Isolate*>(v8::Isolate::New(create_params));
    isolate_->heap()->AddNearHeapLimitCallback(NearHeapLimitCallback,
                                               &oom_triggered_);
  }

  ~SetupIsolateWithSmallHeap() {
    reinterpret_cast<v8::Isolate*>(isolate_)->Dispose();
  }

  Isolate* isolate() { return isolate_; }
  bool oom_triggered() const { return oom_triggered_; }

 private:
  Isolate* isolate_;
  bool oom_triggered_ = false;
};

}  // namespace

58
TEST(Factory_CodeBuilder) {
59 60
  Isolate* isolate = CcTest::i_isolate();
  HandleScope scope(isolate);
61 62 63 64 65 66 67 68 69 70 71 72 73 74 75

  // Create a big function that ends up in CODE_LO_SPACE.
  const int instruction_size = kMaxRegularHeapObjectSize + 1;
  std::unique_ptr<byte[]> instructions(new byte[instruction_size]);

  CodeDesc desc;
  desc.buffer = instructions.get();
  desc.buffer_size = instruction_size;
  desc.instr_size = instruction_size;
  desc.reloc_size = 0;
  desc.constant_pool_size = 0;
  desc.unwinding_info = nullptr;
  desc.unwinding_info_size = 0;
  desc.origin = nullptr;
  Handle<Code> code =
76
      Factory::CodeBuilder(isolate, desc, Code::WASM_FUNCTION).Build();
77

78
  CHECK(isolate->heap()->InSpace(*code, CODE_LO_SPACE));
79
#if VERIFY_HEAP
80
  code->ObjectVerify(isolate);
81 82 83
#endif
}

84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115
UNINITIALIZED_TEST(Factory_CodeBuilder_BuildOOM) {
  SetupIsolateWithSmallHeap isolate_scope;
  HandleScope scope(isolate_scope.isolate());
  std::unique_ptr<byte[]> instructions(new byte[kInstructionSize]);
  CodeDesc desc;
  desc.instr_size = kInstructionSize;
  desc.buffer = instructions.get();

  const Handle<Code> code =
      Factory::CodeBuilder(isolate_scope.isolate(), desc, Code::WASM_FUNCTION)
          .Build();

  CHECK(!code.is_null());
  CHECK(isolate_scope.oom_triggered());
}

UNINITIALIZED_TEST(Factory_CodeBuilder_TryBuildOOM) {
  SetupIsolateWithSmallHeap isolate_scope;
  HandleScope scope(isolate_scope.isolate());
  std::unique_ptr<byte[]> instructions(new byte[kInstructionSize]);
  CodeDesc desc;
  desc.instr_size = kInstructionSize;
  desc.buffer = instructions.get();

  const MaybeHandle<Code> code =
      Factory::CodeBuilder(isolate_scope.isolate(), desc, Code::WASM_FUNCTION)
          .TryBuild();

  CHECK(code.is_null());
  CHECK(!isolate_scope.oom_triggered());
}

116 117 118
}  // namespace test_factory
}  // namespace internal
}  // namespace v8