CodeStubAssembler can generate code for builtins

test/cctest/compiler/test-code-stub-assembler.cc

 // Copyright 2015 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 "src/interface-descriptors.h"
 #include "src/isolate.h"
 #include "test/cctest/compiler/function-tester.h"
 
 namespace v8 {
 namespace internal {
 namespace compiler {
 
 
 class CodeStubAssemblerTester : public CodeStubAssembler {
  public:
+  // Test generating code for a stub.
   CodeStubAssemblerTester(Isolate* isolate,
                           const CallInterfaceDescriptor& descriptor)
       : CodeStubAssembler(isolate, isolate->runtime_zone(), descriptor,
                           Code::ComputeFlags(Code::STUB), "test"),
         scope_(isolate) {}
 
+  // Test generating code for a JS function (e.g. builtins).
+  CodeStubAssemblerTester(Isolate* isolate, int parameter_count)
+      : CodeStubAssembler(isolate, isolate->runtime_zone(), parameter_count,
+                          Code::ComputeFlags(Code::FUNCTION), "test"),
+        scope_(isolate) {}
+
  private:
   HandleScope scope_;
   LocalContext context_;
 };
 
 
 TEST(SimpleSmiReturn) {
   Isolate* isolate(CcTest::InitIsolateOnce());
   VoidDescriptor descriptor(isolate);
   CodeStubAssemblerTester m(isolate, descriptor);
@@ skipping 208 matching lines @@
   Handle<FixedArray> array = isolate->factory()->NewFixedArray(5);
   array->set(4, Smi::FromInt(733));
   m.Return(m.LoadFixedArrayElementSmiIndex(m.HeapConstant(array),
                                            m.SmiTag(m.Int32Constant(4))));
   Handle<Code> code = m.GenerateCode();
   FunctionTester ft(descriptor, code);
   MaybeHandle<Object> result = ft.Call();
   CHECK_EQ(733, Handle<Smi>::cast(result.ToHandleChecked())->value());
 }
 
+TEST(JSFunction) {
+  const int kNumParams = 3;  // Receiver, left, right.
+  Isolate* isolate(CcTest::InitIsolateOnce());
+  CodeStubAssemblerTester m(isolate, kNumParams);
+  m.Return(m.SmiTag(
+      m.Int32Add(m.SmiUntag(m.Parameter(1)), m.SmiUntag(m.Parameter(2)))));
+  Handle<Code> code = m.GenerateCode();
+  FunctionTester ft(kNumParams - 1, code);  // Implicit undefined receiver.

[Jarin, 2016/02/18 09:04:38]

Apparently, there is a bug in FunctionTester::BuildFunction - it adds an extra
argument to the function. Sorry about that.

Fix here: https://codereview.chromium.org/1711513003/ (will try to land ASAP).

[binji, 2016/02/18 19:35:16]

Acknowledged.

+  Handle<Object> left = Handle<Smi>(Smi::FromInt(23), isolate);
+  Handle<Object> right = Handle<Smi>(Smi::FromInt(34), isolate);
+  MaybeHandle<Object> result = ft.Call(left, right);
+  CHECK_EQ(57, Handle<Smi>::cast(result.ToHandleChecked())->value());
+}

[Jarin, 2016/02/18 12:32:15]

Actually, FunctionTester as a whole seems to be a terrible hack (this was
pointed out to me by Benedikt, I agree). How about this:

namespace {

Handle<JSFunction> CreateFunctionFromCode(int parameter_count_with_receiver,
                                          Handle<Code> code) {
  Isolate* isolate = code->GetIsolate();
  Handle<String> name = isolate->factory()->InternalizeUtf8String("test");
  Handle<JSFunction> function =
      isolate->factory()->NewFunctionWithoutPrototype(name, code);
  function->shared()->set_internal_formal_parameter_count(
      parameter_count_with_receiver - 1);  // Implicit undefined receiver.
  return function;
}

}  // namespace

TEST(JSFunction) {
  const int kNumParams = 3;  // Receiver, left, right.
  Isolate* isolate(CcTest::InitIsolateOnce());
  CodeStubAssemblerTester m(isolate, kNumParams);
  m.Return(m.SmiTag(
      m.Int32Add(m.SmiUntag(m.Parameter(1)), m.SmiUntag(m.Parameter(2)))));
  Handle<Code> code = m.GenerateCode();
  Handle<JSFunction> function = CreateFunctionFromCode(kNumParams, code);

  // Run the function.
  Handle<Object> args[] = {Handle<Smi>(Smi::FromInt(23), isolate),
                           Handle<Smi>(Smi::FromInt(34), isolate)};
  MaybeHandle<Object> result =
      Execution::Call(isolate, function, isolate->factory()->undefined_value(),
                      arraysize(args), args);
  CHECK_EQ(57, Handle<Smi>::cast(result.ToHandleChecked())->value());
}

[binji, 2016/02/18 19:35:17]

Done.

+
 }  // namespace compiler
 }  // namespace internal
 }  // namespace v8