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 41 using namespace v8::internal; 42 43 static v8::Persistent<v8::Context> env; 44 45 // --- P r i n t E x t e n s i o n --- 46 47 class 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 58 const char* PrintExtension::kSource = "native function print();"; 59 60 61 v8::Handle<v8::FunctionTemplate> PrintExtension::GetNativeFunction( 62 v8::Handle<v8::String> str) { 63 return v8::FunctionTemplate::New(PrintExtension::Print); 64 } 65 66 67 v8::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 86 static PrintExtension kPrintExtension; 87 v8::DeclareExtension kPrintExtensionDeclaration(&kPrintExtension); 88 89 90 static 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 102 static MaybeObject* GetGlobalProperty(const char* name) { 103 Handle<String> symbol = FACTORY->LookupAsciiSymbol(name); 104 return Isolate::Current()->context()->global()->GetProperty(*symbol); 105 } 106 107 108 static 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 116 static 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 132 static 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 148 TEST(Inc) { 149 InitializeVM(); 150 v8::HandleScope scope; 151 CHECK_EQ(4.0, Inc(3)); 152 } 153 154 155 static 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 169 TEST(Add) { 170 InitializeVM(); 171 v8::HandleScope scope; 172 CHECK_EQ(5.0, Add(2, 3)); 173 } 174 175 176 static 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 189 TEST(Abs) { 190 InitializeVM(); 191 v8::HandleScope scope; 192 CHECK_EQ(3.0, Abs(-3)); 193 } 194 195 196 static 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 210 TEST(Sum) { 211 InitializeVM(); 212 v8::HandleScope scope; 213 CHECK_EQ(5050.0, Sum(100)); 214 } 215 216 217 TEST(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 232 TEST(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 264 TEST(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 | 286 TEST(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. 319 TEST(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 331 TEST(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 356 static 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 364 static 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 392 TEST(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