1// Copyright 2012 the V8 project authors. All rights reserved.
2// Redistribution and use in source and binary forms, with or without
3// modification, are permitted provided that the following conditions are
4// met:
5//
6//     * Redistributions of source code must retain the above copyright
7//       notice, this list of conditions and the following disclaimer.
8//     * Redistributions in binary form must reproduce the above
9//       copyright notice, this list of conditions and the following
10//       disclaimer in the documentation and/or other materials provided
11//       with the distribution.
12//     * Neither the name of Google Inc. nor the names of its
13//       contributors may be used to endorse or promote products derived
14//       from this software without specific prior written permission.
15//
16// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
17// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
18// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
19// A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
20// OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
21// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
22// LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
23// DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
24// THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
25// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
26// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
27
28#include <stdlib.h>
29#include <wchar.h>
30
31#include "v8.h"
32
33#include "compiler.h"
34#include "disasm.h"
35#include "disassembler.h"
36#include "execution.h"
37#include "factory.h"
38#include "platform.h"
39#include "cctest.h"
40
41using namespace v8::internal;
42
43static v8::Persistent<v8::Context> env;
44
45// --- P r i n t   E x t e n s i o n ---
46
47class PrintExtension : public v8::Extension {
48 public:
49  PrintExtension() : v8::Extension("v8/print", kSource) { }
50  virtual v8::Handle<v8::FunctionTemplate> GetNativeFunction(
51      v8::Handle<v8::String> name);
52  static v8::Handle<v8::Value> Print(const v8::Arguments& args);
53 private:
54  static const char* kSource;
55};
56
57
58const char* PrintExtension::kSource = "native function print();";
59
60
61v8::Handle<v8::FunctionTemplate> PrintExtension::GetNativeFunction(
62    v8::Handle<v8::String> str) {
63  return v8::FunctionTemplate::New(PrintExtension::Print);
64}
65
66
67v8::Handle<v8::Value> PrintExtension::Print(const v8::Arguments& args) {
68  for (int i = 0; i < args.Length(); i++) {
69    if (i != 0) printf(" ");
70    v8::HandleScope scope;
71    v8::Handle<v8::Value> arg = args[i];
72    v8::Handle<v8::String> string_obj = arg->ToString();
73    if (string_obj.IsEmpty()) return string_obj;
74    int length = string_obj->Length();
75    uint16_t* string = NewArray<uint16_t>(length + 1);
76    string_obj->Write(string);
77    for (int j = 0; j < length; j++)
78      printf("%lc", static_cast<wchar_t>(string[j]));
79    DeleteArray(string);
80  }
81  printf("\n");
82  return v8::Undefined();
83}
84
85
86static PrintExtension kPrintExtension;
87v8::DeclareExtension kPrintExtensionDeclaration(&kPrintExtension);
88
89
90static void InitializeVM() {
91  if (env.IsEmpty()) {
92    v8::HandleScope scope;
93    const char* extensions[] = { "v8/print", "v8/gc" };
94    v8::ExtensionConfiguration config(2, extensions);
95    env = v8::Context::New(&config);
96  }
97  v8::HandleScope scope;
98  env->Enter();
99}
100
101
102static MaybeObject* GetGlobalProperty(const char* name) {
103  Handle<String> symbol = FACTORY->LookupAsciiSymbol(name);
104  return Isolate::Current()->context()->global()->GetProperty(*symbol);
105}
106
107
108static void SetGlobalProperty(const char* name, Object* value) {
109  Handle<Object> object(value);
110  Handle<String> symbol = FACTORY->LookupAsciiSymbol(name);
111  Handle<JSObject> global(Isolate::Current()->context()->global());
112  SetProperty(global, symbol, object, NONE, kNonStrictMode);
113}
114
115
116static Handle<JSFunction> Compile(const char* source) {
117  Handle<String> source_code(FACTORY->NewStringFromUtf8(CStrVector(source)));
118  Handle<SharedFunctionInfo> shared_function =
119      Compiler::Compile(source_code,
120                        Handle<String>(),
121                        0,
122                        0,
123                        NULL,
124                        NULL,
125                        Handle<String>::null(),
126                        NOT_NATIVES_CODE);
127  return FACTORY->NewFunctionFromSharedFunctionInfo(shared_function,
128      Isolate::Current()->global_context());
129}
130
131
132static double Inc(int x) {
133  const char* source = "result = %d + 1;";
134  EmbeddedVector<char, 512> buffer;
135  OS::SNPrintF(buffer, source, x);
136
137  Handle<JSFunction> fun = Compile(buffer.start());
138  if (fun.is_null()) return -1;
139
140  bool has_pending_exception;
141  Handle<JSObject> global(Isolate::Current()->context()->global());
142  Execution::Call(fun, global, 0, NULL, &has_pending_exception);
143  CHECK(!has_pending_exception);
144  return GetGlobalProperty("result")->ToObjectChecked()->Number();
145}
146
147
148TEST(Inc) {
149  InitializeVM();
150  v8::HandleScope scope;
151  CHECK_EQ(4.0, Inc(3));
152}
153
154
155static double Add(int x, int y) {
156  Handle<JSFunction> fun = Compile("result = x + y;");
157  if (fun.is_null()) return -1;
158
159  SetGlobalProperty("x", Smi::FromInt(x));
160  SetGlobalProperty("y", Smi::FromInt(y));
161  bool has_pending_exception;
162  Handle<JSObject> global(Isolate::Current()->context()->global());
163  Execution::Call(fun, global, 0, NULL, &has_pending_exception);
164  CHECK(!has_pending_exception);
165  return GetGlobalProperty("result")->ToObjectChecked()->Number();
166}
167
168
169TEST(Add) {
170  InitializeVM();
171  v8::HandleScope scope;
172  CHECK_EQ(5.0, Add(2, 3));
173}
174
175
176static double Abs(int x) {
177  Handle<JSFunction> fun = Compile("if (x < 0) result = -x; else result = x;");
178  if (fun.is_null()) return -1;
179
180  SetGlobalProperty("x", Smi::FromInt(x));
181  bool has_pending_exception;
182  Handle<JSObject> global(Isolate::Current()->context()->global());
183  Execution::Call(fun, global, 0, NULL, &has_pending_exception);
184  CHECK(!has_pending_exception);
185  return GetGlobalProperty("result")->ToObjectChecked()->Number();
186}
187
188
189TEST(Abs) {
190  InitializeVM();
191  v8::HandleScope scope;
192  CHECK_EQ(3.0, Abs(-3));
193}
194
195
196static double Sum(int n) {
197  Handle<JSFunction> fun =
198      Compile("s = 0; while (n > 0) { s += n; n -= 1; }; result = s;");
199  if (fun.is_null()) return -1;
200
201  SetGlobalProperty("n", Smi::FromInt(n));
202  bool has_pending_exception;
203  Handle<JSObject> global(Isolate::Current()->context()->global());
204  Execution::Call(fun, global, 0, NULL, &has_pending_exception);
205  CHECK(!has_pending_exception);
206  return GetGlobalProperty("result")->ToObjectChecked()->Number();
207}
208
209
210TEST(Sum) {
211  InitializeVM();
212  v8::HandleScope scope;
213  CHECK_EQ(5050.0, Sum(100));
214}
215
216
217TEST(Print) {
218  InitializeVM();
219  v8::HandleScope scope;
220  const char* source = "for (n = 0; n < 100; ++n) print(n, 1, 2);";
221  Handle<JSFunction> fun = Compile(source);
222  if (fun.is_null()) return;
223  bool has_pending_exception;
224  Handle<JSObject> global(Isolate::Current()->context()->global());
225  Execution::Call(fun, global, 0, NULL, &has_pending_exception);
226  CHECK(!has_pending_exception);
227}
228
229
230// The following test method stems from my coding efforts today. It
231// tests all the functionality I have added to the compiler today
232TEST(Stuff) {
233  InitializeVM();
234  v8::HandleScope scope;
235  const char* source =
236    "r = 0;\n"
237    "a = new Object;\n"
238    "if (a == a) r+=1;\n"  // 1
239    "if (a != new Object()) r+=2;\n"  // 2
240    "a.x = 42;\n"
241    "if (a.x == 42) r+=4;\n"  // 4
242    "function foo() { var x = 87; return x; }\n"
243    "if (foo() == 87) r+=8;\n"  // 8
244    "function bar() { var x; x = 99; return x; }\n"
245    "if (bar() == 99) r+=16;\n"  // 16
246    "function baz() { var x = 1, y, z = 2; y = 3; return x + y + z; }\n"
247    "if (baz() == 6) r+=32;\n"  // 32
248    "function Cons0() { this.x = 42; this.y = 87; }\n"
249    "if (new Cons0().x == 42) r+=64;\n"  // 64
250    "if (new Cons0().y == 87) r+=128;\n"  // 128
251    "function Cons2(x, y) { this.sum = x + y; }\n"
252    "if (new Cons2(3,4).sum == 7) r+=256;";  // 256
253
254  Handle<JSFunction> fun = Compile(source);
255  CHECK(!fun.is_null());
256  bool has_pending_exception;
257  Handle<JSObject> global(Isolate::Current()->context()->global());
258  Execution::Call(fun, global, 0, NULL, &has_pending_exception);
259  CHECK(!has_pending_exception);
260  CHECK_EQ(511.0, GetGlobalProperty("r")->ToObjectChecked()->Number());
261}
262
263
264TEST(UncaughtThrow) {
265  InitializeVM();
266  v8::HandleScope scope;
267
268  const char* source = "throw 42;";
269  Handle<JSFunction> fun = Compile(source);
270  CHECK(!fun.is_null());
271  bool has_pending_exception;
272  Handle<JSObject> global(Isolate::Current()->context()->global());
273  Execution::Call(fun, global, 0, NULL, &has_pending_exception);
274  CHECK(has_pending_exception);
275  CHECK_EQ(42.0, Isolate::Current()->pending_exception()->
276           ToObjectChecked()->Number());
277}
278
279
280// Tests calling a builtin function from C/C++ code, and the builtin function
281// performs GC. It creates a stack frame looks like following:
282//   | C (PerformGC) |
283//   |   JS-to-C     |
284//   |      JS       |
285//   |   C-to-JS     |
286TEST(C2JSFrames) {
287  InitializeVM();
288  v8::HandleScope scope;
289
290  const char* source = "function foo(a) { gc(), print(a); }";
291
292  Handle<JSFunction> fun0 = Compile(source);
293  CHECK(!fun0.is_null());
294
295  // Run the generated code to populate the global object with 'foo'.
296  bool has_pending_exception;
297  Handle<JSObject> global(Isolate::Current()->context()->global());
298  Execution::Call(fun0, global, 0, NULL, &has_pending_exception);
299  CHECK(!has_pending_exception);
300
301  Object* foo_symbol = FACTORY->LookupAsciiSymbol("foo")->ToObjectChecked();
302  MaybeObject* fun1_object = Isolate::Current()->context()->global()->
303      GetProperty(String::cast(foo_symbol));
304  Handle<Object> fun1(fun1_object->ToObjectChecked());
305  CHECK(fun1->IsJSFunction());
306
307  Handle<Object> argv[] = { FACTORY->LookupAsciiSymbol("hello") };
308  Execution::Call(Handle<JSFunction>::cast(fun1),
309                  global,
310                  ARRAY_SIZE(argv),
311                  argv,
312                  &has_pending_exception);
313  CHECK(!has_pending_exception);
314}
315
316
317// Regression 236. Calling InitLineEnds on a Script with undefined
318// source resulted in crash.
319TEST(Regression236) {
320  InitializeVM();
321  v8::HandleScope scope;
322
323  Handle<Script> script = FACTORY->NewScript(FACTORY->empty_string());
324  script->set_source(HEAP->undefined_value());
325  CHECK_EQ(-1, GetScriptLineNumber(script, 0));
326  CHECK_EQ(-1, GetScriptLineNumber(script, 100));
327  CHECK_EQ(-1, GetScriptLineNumber(script, -1));
328}
329
330
331TEST(GetScriptLineNumber) {
332  LocalContext env;
333  v8::HandleScope scope;
334  v8::ScriptOrigin origin = v8::ScriptOrigin(v8::String::New("test"));
335  const char function_f[] = "function f() {}";
336  const int max_rows = 1000;
337  const int buffer_size = max_rows + sizeof(function_f);
338  ScopedVector<char> buffer(buffer_size);
339  memset(buffer.start(), '\n', buffer_size - 1);
340  buffer[buffer_size - 1] = '\0';
341
342  for (int i = 0; i < max_rows; ++i) {
343    if (i > 0)
344      buffer[i - 1] = '\n';
345    memcpy(&buffer[i], function_f, sizeof(function_f) - 1);
346    v8::Handle<v8::String> script_body = v8::String::New(buffer.start());
347    v8::Script::Compile(script_body, &origin)->Run();
348    v8::Local<v8::Function> f = v8::Local<v8::Function>::Cast(
349        env->Global()->Get(v8::String::New("f")));
350    CHECK_EQ(i, f->GetScriptLineNumber());
351  }
352}
353
354
355#ifdef ENABLE_DISASSEMBLER
356static Handle<JSFunction> GetJSFunction(v8::Handle<v8::Object> obj,
357                                 const char* property_name) {
358  v8::Local<v8::Function> fun =
359      v8::Local<v8::Function>::Cast(obj->Get(v8_str(property_name)));
360  return v8::Utils::OpenHandle(*fun);
361}
362
363
364static void CheckCodeForUnsafeLiteral(Handle<JSFunction> f) {
365  // Create a disassembler with default name lookup.
366  disasm::NameConverter name_converter;
367  disasm::Disassembler d(name_converter);
368
369  if (f->code()->kind() == Code::FUNCTION) {
370    Address pc = f->code()->instruction_start();
371    int decode_size =
372        Min(f->code()->instruction_size(),
373            static_cast<int>(f->code()->stack_check_table_offset()));
374    Address end = pc + decode_size;
375
376    v8::internal::EmbeddedVector<char, 128> decode_buffer;
377    while (pc < end) {
378      int num_const = d.ConstantPoolSizeAt(pc);
379      if (num_const >= 0) {
380        pc += (num_const + 1) * kPointerSize;
381      } else {
382        pc += d.InstructionDecode(decode_buffer, pc);
383        CHECK(strstr(decode_buffer.start(), "mov eax,0x178c29c") == NULL);
384        CHECK(strstr(decode_buffer.start(), "push 0x178c29c") == NULL);
385        CHECK(strstr(decode_buffer.start(), "0x178c29c") == NULL);
386      }
387    }
388  }
389}
390
391
392TEST(SplitConstantsInFullCompiler) {
393  v8::HandleScope scope;
394  LocalContext env;
395
396  CompileRun("function f() { a = 12345678 }; f();");
397  CheckCodeForUnsafeLiteral(GetJSFunction(env->Global(), "f"));
398  CompileRun("function f(x) { a = 12345678 + x}; f(1);");
399  CheckCodeForUnsafeLiteral(GetJSFunction(env->Global(), "f"));
400  CompileRun("function f(x) { var arguments = 1; x += 12345678}; f(1);");
401  CheckCodeForUnsafeLiteral(GetJSFunction(env->Global(), "f"));
402  CompileRun("function f(x) { var arguments = 1; x = 12345678}; f(1);");
403  CheckCodeForUnsafeLiteral(GetJSFunction(env->Global(), "f"));
404}
405#endif
406