Provide a tagged allocation top pointer.

test/cctest/test-macro-assembler-x64.cc

 // Copyright 2009 the V8 project authors. All rights reserved.
 // Redistribution and use in source and binary forms, with or without
 // modification, are permitted provided that the following conditions are
 // met:
 //
 //     * Redistributions of source code must retain the above copyright
 //       notice, this list of conditions and the following disclaimer.
 //     * Redistributions in binary form must reproduce the above
 //       copyright notice, this list of conditions and the following
 //       disclaimer in the documentation and/or other materials provided
@@ skipping 15 matching lines @@
 // OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 
 #include <stdlib.h>
 
 #include "src/v8.h"
 
 #include "src/base/platform/platform.h"
 #include "src/factory.h"
 #include "src/macro-assembler.h"
 #include "test/cctest/cctest.h"
+#include "test/cctest/heap/heap-utils.h"
 
 namespace i = v8::internal;
 using i::Address;
 using i::Assembler;
 using i::CodeDesc;
 using i::Condition;
 using i::FUNCTION_CAST;
 using i::HandleScope;
 using i::Immediate;
 using i::Isolate;
@@ skipping 38 matching lines @@
 // Test the x64 assembler by compiling some simple functions into
 // a buffer and executing them.  These tests do not initialize the
 // V8 library, create a context, or use any V8 objects.
 // The AMD64 calling convention is used, with the first five arguments
 // in RSI, RDI, RDX, RCX, R8, and R9, and floating point arguments in
 // the XMM registers.  The return value is in RAX.
 // This calling convention is used on Linux, with GCC, and on Mac OS,
 // with GCC.  A different convention is used on 64-bit windows.
 
 typedef int (*F0)();
+typedef intptr_t (*F1)();
 
 #define __ masm->
 
 
 static void EntryCode(MacroAssembler* masm) {
   // Smi constant register is callee save.
   __ pushq(i::kRootRegister);
   __ InitializeRootRegister();
 }
 
@@ skipping 2617 matching lines @@
   ExitCode(masm);
   __ ret(0);
 
   CodeDesc desc;
   masm->GetCode(&desc);
   // Call the function from C++.
   int result = FUNCTION_CAST<F0>(buffer)();
   CHECK_EQ(0, result);
 }
 
+// Helper for the tests below.
+template <typename Fun>
+void AssembleFunction(Isolate* isolate, size_t actual_size, byte* buffer,
+                      Fun f) {
+  MacroAssembler assembler(isolate, buffer, static_cast<int>(actual_size),
+                           v8::internal::CodeObjectRequired::kYes);
+  MacroAssembler* masm = &assembler;
+  Label exit;
+  EntryCode(masm);
+  f(masm);
+  ExitCode(masm);
+  __ ret(0);
+
+  CodeDesc desc;
+  masm->GetCode(&desc);
+}
+
+TEST(AllocateMacrosNoGCRequired) {
+  // Allocate an executable page of memory.
+  size_t actual_size;
+  byte* buffer = static_cast<byte*>(v8::base::OS::Allocate(
+      Assembler::kMinimalBufferSize, &actual_size, true));
+  CHECK(buffer);
+  Isolate* isolate = CcTest::i_isolate();
+  HandleScope handles(isolate);
+
+  AllocationFlags const kDoubleAligned =
+      static_cast<AllocationFlags>(DOUBLE_ALIGNMENT);
+  AllocationFlags const kNoAllocationFlags =
+      static_cast<AllocationFlags>(NO_ALLOCATION_FLAGS);
+
+#define CHECK_TAGGED(result) CHECK_EQ(result& i::kHeapObjectTag, 1);
+#define CHECK_DOUBLE_ALIGNED(result)                     \
+  do {                                                   \
+    CHECK_TAGGED((result));                              \
+    CHECK_EQ((result)&i::kDoubleAlignmentMaskTagged, 0); \
+  } while (false)
+
+  i::heap::GcAndSweep(isolate->heap(), i::AllocationSpace::NEW_SPACE);
+  {
+    AssembleFunction(isolate, actual_size, buffer, [](MacroAssembler* masm) {
+      Label gc_required, success;
+      __ Allocate(kPointerSize, rax, rcx, rdx, &gc_required,
+                  kNoAllocationFlags);
+      __ jmp(&success);
+      __ bind(&gc_required);
+      __ Abort(i::kNoReason);
+      __ bind(&success);
+    });
+    intptr_t test_result = FUNCTION_CAST<F1>(buffer)();
+    CHECK_TAGGED(test_result);
+  }
+
+  {
+    AssembleFunction(isolate, actual_size, buffer, [](MacroAssembler* masm) {
+      Label gc_required, success;
+      __ Allocate(i::kDoubleSize, rax, rcx, rdx, &gc_required, kDoubleAligned);
+      __ jmp(&success);
+      __ bind(&gc_required);
+      __ Abort(i::kNoReason);
+      __ bind(&success);
+    });
+    i::heap::MakeSureNewSpaceTopIsNotDoubleAligned(isolate->heap());
+    intptr_t test_result = FUNCTION_CAST<F1>(buffer)();
+    CHECK_DOUBLE_ALIGNED(test_result);
+  }
+  {
+    AssembleFunction(isolate, actual_size, buffer, [](MacroAssembler* masm) {
+      Label gc_required, success;
+      __ movl(rbx, Immediate(kPointerSize));
+      __ Allocate(kPointerSize, rax, rcx, rdx, &gc_required,
+                  kNoAllocationFlags);
+      __ jmp(&success);
+      __ bind(&gc_required);
+      __ Abort(i::kNoReason);
+      __ bind(&success);
+    });
+    intptr_t test_result = FUNCTION_CAST<F1>(buffer)();
+    CHECK_TAGGED(test_result);
+  }
+
+  {
+    AssembleFunction(isolate, actual_size, buffer, [](MacroAssembler* masm) {
+      Label gc_required, success;
+      __ movl(rbx, Immediate(i::kDoubleSize));
+      __ Allocate(rbx, rax, rcx, rdx, &gc_required, kDoubleAligned);
+      __ jmp(&success);
+      __ bind(&gc_required);
+      __ Abort(i::kNoReason);
+      __ bind(&success);
+    });
+    i::heap::MakeSureNewSpaceTopIsNotDoubleAligned(isolate->heap());
+    intptr_t test_result = FUNCTION_CAST<F1>(buffer)();
+    CHECK_DOUBLE_ALIGNED(test_result);
+  }
+
+  {
+    AssembleFunction(isolate, actual_size, buffer, [](MacroAssembler* masm) {
+      Label gc_required, success;
+      __ movl(rbx, Immediate(2));
+      __ Allocate(0, i::times_4, rbx, rax, rcx, rdx, &gc_required,
+                  kNoAllocationFlags);
+      __ jmp(&success);
+      __ bind(&gc_required);
+      __ Abort(i::kNoReason);
+      __ bind(&success);
+    });
+    intptr_t test_result = FUNCTION_CAST<F1>(buffer)();
+    CHECK_TAGGED(test_result);
+  }
+
+  {
+    AssembleFunction(isolate, actual_size, buffer, [](MacroAssembler* masm) {
+      Label gc_required, success;
+      __ movl(rbx, Immediate(i::kDoubleSize));
+      __ Allocate(i::FixedDoubleArray::kHeaderSize, i::times_8, rbx, rax, rcx,
+                  rdx, &gc_required, kDoubleAligned);
+      __ jmp(&success);
+      __ bind(&gc_required);
+      __ Abort(i::kNoReason);
+      __ bind(&success);
+    });
+    i::heap::MakeSureNewSpaceTopIsNotDoubleAligned(isolate->heap());
+    intptr_t test_result = FUNCTION_CAST<F1>(buffer)();
+    CHECK_DOUBLE_ALIGNED(test_result);
+  }
+#undef CHECK_TAGGED
+#undef CHECK_DOUBLE_ALIGNED
+}
+
+TEST(AllocateMacrosGCRequired) {
+  // Allocate an executable page of memory.
+  size_t actual_size;
+  byte* buffer = static_cast<byte*>(v8::base::OS::Allocate(
+      Assembler::kMinimalBufferSize, &actual_size, true));
+  CHECK(buffer);
+  Isolate* isolate = CcTest::i_isolate();
+  HandleScope handles(isolate);
+
+  AllocationFlags const kDoubleAligned =
+      static_cast<AllocationFlags>(DOUBLE_ALIGNMENT);
+
+  // Test that the macro jumps to the gc_required label if there's not enough
+  // space to allocate with double alignment.
+  {
+    AssembleFunction(isolate, actual_size, buffer, [](MacroAssembler* masm) {
+      Label gc_required, end;
+      __ Allocate(kPointerSize, rax, rcx, rdx, &gc_required, kDoubleAligned);
+      __ xorl(rax, rax);
+      __ jmp(&end);
+      __ bind(&gc_required);
+      __ movl(rax, Immediate(1));
+      __ bind(&end);
+    });
+    i::heap::AllocateAllButNBytes(isolate->heap()->new_space(), 4);
+    intptr_t test_result = FUNCTION_CAST<F1>(buffer)();
+    CHECK_EQ(test_result, 1);
+  }
+
+  {
+    AssembleFunction(isolate, actual_size, buffer, [](MacroAssembler* masm) {
+      Label gc_required, end;
+      __ movl(rbx, Immediate(kPointerSize));
+      __ Allocate(rbx, rax, rcx, rdx, &gc_required, kDoubleAligned);
+      __ xorl(rax, rax);
+      __ jmp(&end);
+      __ bind(&gc_required);
+      __ movl(rax, Immediate(1));
+      __ bind(&end);
+    });
+    i::heap::AllocateAllButNBytes(isolate->heap()->new_space(), 4);
+    intptr_t test_result = FUNCTION_CAST<F1>(buffer)();
+    CHECK_EQ(test_result, 1);
+  }
+
+  {
+    AssembleFunction(isolate, actual_size, buffer, [](MacroAssembler* masm) {
+      Label gc_required, end;
+      __ movl(rbx, Immediate(kPointerSize));
+      __ Allocate(0, i::times_8, rbx, rax, rcx, rdx, &gc_required,
+                  kDoubleAligned);
+      __ xorl(rax, rax);
+      __ jmp(&end);
+      __ bind(&gc_required);
+      __ movl(rax, Immediate(1));
+      __ bind(&end);
+    });
+    i::heap::AllocateAllButNBytes(isolate->heap()->new_space(), 4);
+    intptr_t test_result = FUNCTION_CAST<F1>(buffer)();
+    CHECK_EQ(test_result, 1);
+  }
+}
 
 #undef __