Compiler.cpp 52 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360
  1. /*
  2. * Copyright (c) 2023, Andreas Kling <kling@serenityos.org>
  3. *
  4. * SPDX-License-Identifier: BSD-2-Clause
  5. */
  6. #include <AK/OwnPtr.h>
  7. #include <AK/Platform.h>
  8. #include <LibJS/Bytecode/CommonImplementations.h>
  9. #include <LibJS/Bytecode/Instruction.h>
  10. #include <LibJS/Bytecode/Interpreter.h>
  11. #include <LibJS/Bytecode/RegexTable.h>
  12. #include <LibJS/JIT/Compiler.h>
  13. #include <LibJS/Runtime/AbstractOperations.h>
  14. #include <LibJS/Runtime/Array.h>
  15. #include <LibJS/Runtime/DeclarativeEnvironment.h>
  16. #include <LibJS/Runtime/ECMAScriptFunctionObject.h>
  17. #include <LibJS/Runtime/VM.h>
  18. #include <LibJS/Runtime/ValueInlines.h>
  19. #include <sys/mman.h>
  20. #include <unistd.h>
  21. #if ARCH(X86_64)
  22. # define LOG_JIT_SUCCESS 1
  23. # define LOG_JIT_FAILURE 1
  24. # define DUMP_JIT_MACHINE_CODE_TO_STDOUT 0
  25. # define DUMP_JIT_DISASSEMBLY 0
  26. # define TRY_OR_SET_EXCEPTION(expression) \
  27. ({ \
  28. /* Ignore -Wshadow to allow nesting the macro. */ \
  29. AK_IGNORE_DIAGNOSTIC("-Wshadow", \
  30. auto&& _temporary_result = (expression)); \
  31. static_assert(!::AK::Detail::IsLvalueReference<decltype(_temporary_result.release_value())>, \
  32. "Do not return a reference from a fallible expression"); \
  33. if (_temporary_result.is_error()) [[unlikely]] { \
  34. vm.bytecode_interpreter().reg(Bytecode::Register::exception()) = _temporary_result.release_error().value().value(); \
  35. return {}; \
  36. } \
  37. _temporary_result.release_value(); \
  38. })
  39. namespace JS::JIT {
  40. void Compiler::store_vm_register(Bytecode::Register dst, Assembler::Reg src)
  41. {
  42. m_assembler.mov(
  43. Assembler::Operand::Mem64BaseAndOffset(REGISTER_ARRAY_BASE, dst.index() * sizeof(Value)),
  44. Assembler::Operand::Register(src));
  45. }
  46. void Compiler::load_vm_register(Assembler::Reg dst, Bytecode::Register src)
  47. {
  48. m_assembler.mov(
  49. Assembler::Operand::Register(dst),
  50. Assembler::Operand::Mem64BaseAndOffset(REGISTER_ARRAY_BASE, src.index() * sizeof(Value)));
  51. }
  52. void Compiler::store_vm_local(size_t dst, Assembler::Reg src)
  53. {
  54. m_assembler.mov(
  55. Assembler::Operand::Mem64BaseAndOffset(LOCALS_ARRAY_BASE, dst * sizeof(Value)),
  56. Assembler::Operand::Register(src));
  57. }
  58. void Compiler::load_vm_local(Assembler::Reg dst, size_t src)
  59. {
  60. m_assembler.mov(
  61. Assembler::Operand::Register(dst),
  62. Assembler::Operand::Mem64BaseAndOffset(LOCALS_ARRAY_BASE, src * sizeof(Value)));
  63. }
  64. void Compiler::compile_load_immediate(Bytecode::Op::LoadImmediate const& op)
  65. {
  66. m_assembler.mov(
  67. Assembler::Operand::Register(GPR0),
  68. Assembler::Operand::Imm(op.value().encoded()));
  69. store_vm_register(Bytecode::Register::accumulator(), GPR0);
  70. }
  71. void Compiler::compile_load(Bytecode::Op::Load const& op)
  72. {
  73. load_vm_register(GPR0, op.src());
  74. store_vm_register(Bytecode::Register::accumulator(), GPR0);
  75. }
  76. void Compiler::compile_store(Bytecode::Op::Store const& op)
  77. {
  78. load_vm_register(GPR0, Bytecode::Register::accumulator());
  79. store_vm_register(op.dst(), GPR0);
  80. }
  81. void Compiler::compile_get_local(Bytecode::Op::GetLocal const& op)
  82. {
  83. load_vm_local(GPR0, op.index());
  84. store_vm_register(Bytecode::Register::accumulator(), GPR0);
  85. }
  86. void Compiler::compile_set_local(Bytecode::Op::SetLocal const& op)
  87. {
  88. load_vm_register(GPR0, Bytecode::Register::accumulator());
  89. store_vm_local(op.index(), GPR0);
  90. }
  91. static Value cxx_typeof_local(VM& vm, Value value)
  92. {
  93. return PrimitiveString::create(vm, value.typeof());
  94. }
  95. void Compiler::compile_typeof_local(Bytecode::Op::TypeofLocal const& op)
  96. {
  97. load_vm_local(ARG1, op.index());
  98. native_call((void*)cxx_typeof_local);
  99. store_vm_register(Bytecode::Register::accumulator(), GPR0);
  100. }
  101. void Compiler::compile_jump(Bytecode::Op::Jump const& op)
  102. {
  103. m_assembler.jump(label_for(op.true_target()->block()));
  104. }
  105. static bool cxx_to_boolean(VM&, Value value)
  106. {
  107. return value.to_boolean();
  108. }
  109. void Compiler::compile_to_boolean(Assembler::Reg dst, Assembler::Reg src)
  110. {
  111. // dst = src;
  112. m_assembler.mov(
  113. Assembler::Operand::Register(dst),
  114. Assembler::Operand::Register(src));
  115. // dst >>= 48;
  116. m_assembler.shift_right(
  117. Assembler::Operand::Register(dst),
  118. Assembler::Operand::Imm(48));
  119. // if (dst != BOOLEAN_TAG) goto slow_case;
  120. Assembler::Label slow_case {};
  121. m_assembler.jump_if_not_equal(
  122. Assembler::Operand::Register(dst),
  123. Assembler::Operand::Imm(BOOLEAN_TAG),
  124. slow_case);
  125. // Fast path for JS::Value booleans.
  126. // dst = src;
  127. m_assembler.mov(
  128. Assembler::Operand::Register(dst),
  129. Assembler::Operand::Register(src));
  130. // goto end;
  131. auto end = m_assembler.jump();
  132. // slow_case: // call C++ helper
  133. slow_case.link(m_assembler);
  134. m_assembler.mov(
  135. Assembler::Operand::Register(ARG1),
  136. Assembler::Operand::Register(src));
  137. native_call((void*)cxx_to_boolean);
  138. m_assembler.mov(
  139. Assembler::Operand::Register(dst),
  140. Assembler::Operand::Register(RET));
  141. // end:
  142. end.link(m_assembler);
  143. // dst &= 1;
  144. m_assembler.bitwise_and(
  145. Assembler::Operand::Register(dst),
  146. Assembler::Operand::Imm(1));
  147. }
  148. void Compiler::compile_jump_conditional(Bytecode::Op::JumpConditional const& op)
  149. {
  150. load_vm_register(GPR1, Bytecode::Register::accumulator());
  151. compile_to_boolean(GPR0, GPR1);
  152. m_assembler.jump_if_zero(
  153. Assembler::Operand::Register(GPR0),
  154. label_for(op.false_target()->block()));
  155. m_assembler.jump(label_for(op.true_target()->block()));
  156. }
  157. void Compiler::compile_jump_nullish(Bytecode::Op::JumpNullish const& op)
  158. {
  159. load_vm_register(GPR0, Bytecode::Register::accumulator());
  160. m_assembler.shift_right(
  161. Assembler::Operand::Register(GPR0),
  162. Assembler::Operand::Imm(48));
  163. m_assembler.bitwise_and(
  164. Assembler::Operand::Register(GPR0),
  165. Assembler::Operand::Imm(IS_NULLISH_EXTRACT_PATTERN));
  166. m_assembler.jump_if_equal(
  167. Assembler::Operand::Register(GPR0),
  168. Assembler::Operand::Imm(IS_NULLISH_PATTERN),
  169. label_for(op.true_target()->block()));
  170. m_assembler.jump(label_for(op.false_target()->block()));
  171. }
  172. [[maybe_unused]] static Value cxx_increment(VM& vm, Value value)
  173. {
  174. auto old_value = TRY_OR_SET_EXCEPTION(value.to_numeric(vm));
  175. if (old_value.is_number())
  176. return Value(old_value.as_double() + 1);
  177. return BigInt::create(vm, old_value.as_bigint().big_integer().plus(Crypto::SignedBigInteger { 1 }));
  178. }
  179. template<typename Codegen>
  180. void Compiler::branch_if_int32(Assembler::Reg reg, Codegen codegen)
  181. {
  182. // GPR0 = reg >> 48;
  183. m_assembler.mov(Assembler::Operand::Register(GPR0), Assembler::Operand::Register(reg));
  184. m_assembler.shift_right(Assembler::Operand::Register(GPR0), Assembler::Operand::Imm(48));
  185. Assembler::Label not_int32_case {};
  186. m_assembler.jump_if_not_equal(
  187. Assembler::Operand::Register(GPR0),
  188. Assembler::Operand::Imm(INT32_TAG),
  189. not_int32_case);
  190. codegen();
  191. not_int32_case.link(m_assembler);
  192. }
  193. template<typename Codegen>
  194. void Compiler::branch_if_both_int32(Assembler::Reg lhs, Assembler::Reg rhs, Codegen codegen)
  195. {
  196. // GPR0 = lhs >> 48;
  197. m_assembler.mov(Assembler::Operand::Register(GPR0), Assembler::Operand::Register(lhs));
  198. m_assembler.shift_right(Assembler::Operand::Register(GPR0), Assembler::Operand::Imm(48));
  199. // GPR1 = rhs >> 48;
  200. m_assembler.mov(Assembler::Operand::Register(GPR1), Assembler::Operand::Register(rhs));
  201. m_assembler.shift_right(Assembler::Operand::Register(GPR1), Assembler::Operand::Imm(48));
  202. Assembler::Label not_int32_case {};
  203. m_assembler.jump_if_not_equal(
  204. Assembler::Operand::Register(GPR0),
  205. Assembler::Operand::Imm(INT32_TAG),
  206. not_int32_case);
  207. m_assembler.jump_if_not_equal(
  208. Assembler::Operand::Register(GPR1),
  209. Assembler::Operand::Imm(INT32_TAG),
  210. not_int32_case);
  211. codegen();
  212. not_int32_case.link(m_assembler);
  213. }
  214. void Compiler::compile_increment(Bytecode::Op::Increment const&)
  215. {
  216. load_vm_register(ARG1, Bytecode::Register::accumulator());
  217. Assembler::Label end {};
  218. Assembler::Label slow_case {};
  219. branch_if_int32(ARG1, [&] {
  220. // GPR0 = ARG1 & 0xffffffff;
  221. m_assembler.mov(
  222. Assembler::Operand::Register(GPR0),
  223. Assembler::Operand::Register(ARG1));
  224. m_assembler.mov(
  225. Assembler::Operand::Register(GPR1),
  226. Assembler::Operand::Imm(0xffffffff));
  227. m_assembler.bitwise_and(
  228. Assembler::Operand::Register(GPR0),
  229. Assembler::Operand::Register(GPR1));
  230. // if (GPR0 == 0x7fffffff) goto slow_case;
  231. m_assembler.jump_if_equal(
  232. Assembler::Operand::Register(GPR0),
  233. Assembler::Operand::Imm(0x7fffffff),
  234. slow_case);
  235. // ARG1 += 1;
  236. m_assembler.add(
  237. Assembler::Operand::Register(ARG1),
  238. Assembler::Operand::Imm(1));
  239. // accumulator = ARG1;
  240. store_vm_register(Bytecode::Register::accumulator(), ARG1);
  241. m_assembler.jump(end);
  242. });
  243. slow_case.link(m_assembler);
  244. native_call((void*)cxx_increment);
  245. store_vm_register(Bytecode::Register::accumulator(), RET);
  246. check_exception();
  247. end.link(m_assembler);
  248. }
  249. static Value cxx_decrement(VM& vm, Value value)
  250. {
  251. auto old_value = TRY_OR_SET_EXCEPTION(value.to_numeric(vm));
  252. if (old_value.is_number())
  253. return Value(old_value.as_double() - 1);
  254. return BigInt::create(vm, old_value.as_bigint().big_integer().minus(Crypto::SignedBigInteger { 1 }));
  255. }
  256. void Compiler::compile_decrement(Bytecode::Op::Decrement const&)
  257. {
  258. load_vm_register(ARG1, Bytecode::Register::accumulator());
  259. native_call((void*)cxx_decrement);
  260. store_vm_register(Bytecode::Register::accumulator(), RET);
  261. check_exception();
  262. }
  263. void Compiler::check_exception()
  264. {
  265. // if (!exception.is_empty()) goto m_exception_handler;
  266. load_vm_register(GPR0, Bytecode::Register::exception());
  267. m_assembler.mov(Assembler::Operand::Register(GPR1), Assembler::Operand::Imm(Value().encoded()));
  268. m_assembler.jump_if_not_equal(
  269. Assembler::Operand::Register(GPR0),
  270. Assembler::Operand::Register(GPR1),
  271. m_exception_handler);
  272. }
  273. void Compiler::handle_exception()
  274. {
  275. // if (!unwind_context.valid) return;
  276. Assembler::Label handle_exception {};
  277. m_assembler.mov(
  278. Assembler::Operand::Register(GPR0),
  279. Assembler::Operand::Mem64BaseAndOffset(UNWIND_CONTEXT_BASE, 0));
  280. m_assembler.jump_if_not_equal(
  281. Assembler::Operand::Register(GPR0),
  282. Assembler::Operand::Imm(0),
  283. handle_exception);
  284. jump_to_exit();
  285. // handle_exception:
  286. handle_exception.link(m_assembler);
  287. // if (unwind_context.handler) {
  288. Assembler::Label no_handler {};
  289. m_assembler.mov(
  290. Assembler::Operand::Register(GPR0),
  291. Assembler::Operand::Mem64BaseAndOffset(UNWIND_CONTEXT_BASE, 8));
  292. m_assembler.jump_if_equal(
  293. Assembler::Operand::Register(GPR0),
  294. Assembler::Operand::Imm(0),
  295. no_handler);
  296. // accumulator = exception;
  297. load_vm_register(GPR1, Bytecode::Register::exception());
  298. store_vm_register(Bytecode::Register::accumulator(), GPR1);
  299. // exception = Value();
  300. m_assembler.mov(
  301. Assembler::Operand::Register(GPR1),
  302. Assembler::Operand::Imm(Value().encoded()));
  303. store_vm_register(Bytecode::Register::exception(), GPR1);
  304. // unwind_context.handler = nullptr;
  305. m_assembler.mov(
  306. Assembler::Operand::Register(GPR1),
  307. Assembler::Operand::Imm(0));
  308. m_assembler.mov(
  309. Assembler::Operand::Mem64BaseAndOffset(UNWIND_CONTEXT_BASE, 8),
  310. Assembler::Operand::Register(GPR1));
  311. // goto handler;
  312. m_assembler.jump(Assembler::Operand::Register(GPR0));
  313. // }
  314. // no_handler:
  315. no_handler.link(m_assembler);
  316. // if (unwind_context.finalizer) goto finalizer;
  317. Assembler::Label no_finalizer {};
  318. m_assembler.mov(
  319. Assembler::Operand::Register(GPR0),
  320. Assembler::Operand::Mem64BaseAndOffset(UNWIND_CONTEXT_BASE, 16));
  321. m_assembler.jump_if_equal(
  322. Assembler::Operand::Register(GPR0),
  323. Assembler::Operand::Imm(0),
  324. no_finalizer);
  325. m_assembler.jump(Assembler::Operand::Register(GPR0));
  326. // no_finalizer:
  327. // NOTE: No catch and no finally!? Crash.
  328. no_finalizer.link(m_assembler);
  329. m_assembler.verify_not_reached();
  330. }
  331. void Compiler::push_unwind_context(bool valid, Optional<Bytecode::Label> const& handler, Optional<Bytecode::Label> const& finalizer)
  332. {
  333. // Put this on the stack, and then point UNWIND_CONTEXT_BASE at it.
  334. // struct {
  335. // u64 valid;
  336. // u64 handler;
  337. // u64 finalizer;
  338. // };
  339. if (finalizer.has_value()) {
  340. // push finalizer (patched later)
  341. m_assembler.mov(
  342. Assembler::Operand::Register(GPR0),
  343. Assembler::Operand::Imm(0),
  344. Assembler::Patchable::Yes);
  345. block_data_for(finalizer.value().block()).absolute_references_to_here.append(m_assembler.m_output.size() - 8);
  346. m_assembler.push(Assembler::Operand::Register(GPR0));
  347. } else {
  348. m_assembler.push(Assembler::Operand::Imm(0));
  349. }
  350. if (handler.has_value()) {
  351. // push handler (patched later)
  352. m_assembler.mov(
  353. Assembler::Operand::Register(GPR0),
  354. Assembler::Operand::Imm(0),
  355. Assembler::Patchable::Yes);
  356. block_data_for(handler.value().block()).absolute_references_to_here.append(m_assembler.m_output.size() - 8);
  357. m_assembler.push(Assembler::Operand::Register(GPR0));
  358. } else {
  359. m_assembler.push(Assembler::Operand::Imm(0));
  360. }
  361. // push valid
  362. m_assembler.push(Assembler::Operand::Imm(valid));
  363. // UNWIND_CONTEXT_BASE = STACK_POINTER
  364. m_assembler.mov(
  365. Assembler::Operand::Register(UNWIND_CONTEXT_BASE),
  366. Assembler::Operand::Register(STACK_POINTER));
  367. // align stack pointer
  368. m_assembler.sub(Assembler::Operand::Register(STACK_POINTER), Assembler::Operand::Imm(8));
  369. }
  370. void Compiler::pop_unwind_context()
  371. {
  372. m_assembler.add(Assembler::Operand::Register(STACK_POINTER), Assembler::Operand::Imm(32));
  373. m_assembler.add(Assembler::Operand::Register(UNWIND_CONTEXT_BASE), Assembler::Operand::Imm(32));
  374. }
  375. void Compiler::compile_enter_unwind_context(Bytecode::Op::EnterUnwindContext const& op)
  376. {
  377. push_unwind_context(true, op.handler_target(), op.finalizer_target());
  378. m_assembler.jump(label_for(op.entry_point().block()));
  379. }
  380. void Compiler::compile_leave_unwind_context(Bytecode::Op::LeaveUnwindContext const&)
  381. {
  382. pop_unwind_context();
  383. }
  384. void Compiler::compile_throw(Bytecode::Op::Throw const&)
  385. {
  386. load_vm_register(GPR0, Bytecode::Register::accumulator());
  387. store_vm_register(Bytecode::Register::exception(), GPR0);
  388. check_exception();
  389. }
  390. static ThrowCompletionOr<Value> abstract_inequals(VM& vm, Value src1, Value src2)
  391. {
  392. return Value(!TRY(is_loosely_equal(vm, src1, src2)));
  393. }
  394. static ThrowCompletionOr<Value> abstract_equals(VM& vm, Value src1, Value src2)
  395. {
  396. return Value(TRY(is_loosely_equal(vm, src1, src2)));
  397. }
  398. static ThrowCompletionOr<Value> typed_inequals(VM&, Value src1, Value src2)
  399. {
  400. return Value(!is_strictly_equal(src1, src2));
  401. }
  402. static ThrowCompletionOr<Value> typed_equals(VM&, Value src1, Value src2)
  403. {
  404. return Value(is_strictly_equal(src1, src2));
  405. }
  406. # define DO_COMPILE_COMMON_BINARY_OP(TitleCaseName, snake_case_name) \
  407. static Value cxx_##snake_case_name(VM& vm, Value lhs, Value rhs) \
  408. { \
  409. return TRY_OR_SET_EXCEPTION(snake_case_name(vm, lhs, rhs)); \
  410. } \
  411. \
  412. void Compiler::compile_##snake_case_name(Bytecode::Op::TitleCaseName const& op) \
  413. { \
  414. load_vm_register(ARG1, op.lhs()); \
  415. load_vm_register(ARG2, Bytecode::Register::accumulator()); \
  416. native_call((void*)cxx_##snake_case_name); \
  417. store_vm_register(Bytecode::Register::accumulator(), RET); \
  418. check_exception(); \
  419. }
  420. JS_ENUMERATE_COMMON_BINARY_OPS_WITHOUT_FAST_PATH(DO_COMPILE_COMMON_BINARY_OP)
  421. # undef DO_COMPILE_COMMON_BINARY_OP
  422. static Value cxx_less_than(VM& vm, Value lhs, Value rhs)
  423. {
  424. return TRY_OR_SET_EXCEPTION(less_than(vm, lhs, rhs));
  425. }
  426. void Compiler::compile_less_than(Bytecode::Op::LessThan const& op)
  427. {
  428. load_vm_register(ARG1, op.lhs());
  429. load_vm_register(ARG2, Bytecode::Register::accumulator());
  430. Assembler::Label end {};
  431. branch_if_both_int32(ARG1, ARG2, [&] {
  432. // if (ARG1 < ARG2) return true;
  433. // else return false;
  434. Assembler::Label true_case {};
  435. m_assembler.sign_extend_32_to_64_bits(ARG1);
  436. m_assembler.sign_extend_32_to_64_bits(ARG2);
  437. m_assembler.jump_if_less_than(
  438. Assembler::Operand::Register(ARG1),
  439. Assembler::Operand::Register(ARG2),
  440. true_case);
  441. m_assembler.mov(
  442. Assembler::Operand::Register(GPR0),
  443. Assembler::Operand::Imm(Value(false).encoded()));
  444. store_vm_register(Bytecode::Register::accumulator(), GPR0);
  445. m_assembler.jump(end);
  446. true_case.link(m_assembler);
  447. m_assembler.mov(
  448. Assembler::Operand::Register(GPR0),
  449. Assembler::Operand::Imm(Value(true).encoded()));
  450. store_vm_register(Bytecode::Register::accumulator(), GPR0);
  451. m_assembler.jump(end);
  452. });
  453. native_call((void*)cxx_less_than);
  454. store_vm_register(Bytecode::Register::accumulator(), RET);
  455. check_exception();
  456. end.link(m_assembler);
  457. }
  458. static ThrowCompletionOr<Value> not_(VM&, Value value)
  459. {
  460. return Value(!value.to_boolean());
  461. }
  462. static ThrowCompletionOr<Value> typeof_(VM& vm, Value value)
  463. {
  464. return PrimitiveString::create(vm, value.typeof());
  465. }
  466. # define DO_COMPILE_COMMON_UNARY_OP(TitleCaseName, snake_case_name) \
  467. static Value cxx_##snake_case_name(VM& vm, Value value) \
  468. { \
  469. return TRY_OR_SET_EXCEPTION(snake_case_name(vm, value)); \
  470. } \
  471. \
  472. void Compiler::compile_##snake_case_name(Bytecode::Op::TitleCaseName const&) \
  473. { \
  474. load_vm_register(ARG1, Bytecode::Register::accumulator()); \
  475. native_call((void*)cxx_##snake_case_name); \
  476. store_vm_register(Bytecode::Register::accumulator(), RET); \
  477. check_exception(); \
  478. }
  479. JS_ENUMERATE_COMMON_UNARY_OPS(DO_COMPILE_COMMON_UNARY_OP)
  480. # undef DO_COMPILE_COMMON_UNARY_OP
  481. void Compiler::compile_return(Bytecode::Op::Return const&)
  482. {
  483. load_vm_register(GPR0, Bytecode::Register::accumulator());
  484. // check for finalizer
  485. // if (!unwind_context.valid) goto normal_return;
  486. Assembler::Label normal_return {};
  487. m_assembler.mov(
  488. Assembler::Operand::Register(GPR1),
  489. Assembler::Operand::Mem64BaseAndOffset(UNWIND_CONTEXT_BASE, 0));
  490. m_assembler.jump_if_equal(
  491. Assembler::Operand::Register(GPR1),
  492. Assembler::Operand::Imm(0),
  493. normal_return);
  494. // if (!unwind_context.finalizer) goto normal_return;
  495. m_assembler.mov(
  496. Assembler::Operand::Register(GPR1),
  497. Assembler::Operand::Mem64BaseAndOffset(UNWIND_CONTEXT_BASE, 16));
  498. m_assembler.jump_if_equal(
  499. Assembler::Operand::Register(GPR1),
  500. Assembler::Operand::Imm(0),
  501. normal_return);
  502. store_vm_register(Bytecode::Register::saved_return_value(), GPR0);
  503. m_assembler.jump(Assembler::Operand::Register(GPR1));
  504. // normal_return:
  505. normal_return.link(m_assembler);
  506. store_vm_register(Bytecode::Register::return_value(), GPR0);
  507. jump_to_exit();
  508. }
  509. static Value cxx_new_string(VM& vm, DeprecatedString const& string)
  510. {
  511. return PrimitiveString::create(vm, string);
  512. }
  513. void Compiler::compile_new_string(Bytecode::Op::NewString const& op)
  514. {
  515. auto const& string = m_bytecode_executable.string_table->get(op.index());
  516. m_assembler.mov(
  517. Assembler::Operand::Register(ARG1),
  518. Assembler::Operand::Imm(bit_cast<u64>(&string)));
  519. native_call((void*)cxx_new_string);
  520. store_vm_register(Bytecode::Register::accumulator(), RET);
  521. }
  522. void Compiler::compile_new_regexp(Bytecode::Op::NewRegExp const& op)
  523. {
  524. auto const& parsed_regex = m_bytecode_executable.regex_table->get(op.regex_index());
  525. auto const& pattern = m_bytecode_executable.string_table->get(op.source_index());
  526. auto const& flags = m_bytecode_executable.string_table->get(op.flags_index());
  527. m_assembler.mov(
  528. Assembler::Operand::Register(ARG1),
  529. Assembler::Operand::Imm(bit_cast<u64>(&parsed_regex)));
  530. m_assembler.mov(
  531. Assembler::Operand::Register(ARG2),
  532. Assembler::Operand::Imm(bit_cast<u64>(&pattern)));
  533. m_assembler.mov(
  534. Assembler::Operand::Register(ARG3),
  535. Assembler::Operand::Imm(bit_cast<u64>(&flags)));
  536. native_call((void*)Bytecode::new_regexp);
  537. store_vm_register(Bytecode::Register::accumulator(), RET);
  538. }
  539. static Value cxx_new_bigint(VM& vm, Crypto::SignedBigInteger const& bigint)
  540. {
  541. return BigInt::create(vm, bigint);
  542. }
  543. void Compiler::compile_new_bigint(Bytecode::Op::NewBigInt const& op)
  544. {
  545. m_assembler.mov(
  546. Assembler::Operand::Register(ARG1),
  547. Assembler::Operand::Imm(bit_cast<u64>(&op.bigint())));
  548. native_call((void*)cxx_new_bigint);
  549. store_vm_register(Bytecode::Register::accumulator(), RET);
  550. }
  551. static Value cxx_new_object(VM& vm)
  552. {
  553. auto& realm = *vm.current_realm();
  554. return Object::create(realm, realm.intrinsics().object_prototype());
  555. }
  556. void Compiler::compile_new_object(Bytecode::Op::NewObject const&)
  557. {
  558. native_call((void*)cxx_new_object);
  559. store_vm_register(Bytecode::Register::accumulator(), RET);
  560. }
  561. static Value cxx_new_array(VM& vm, size_t element_count, u32 first_register_index)
  562. {
  563. auto& realm = *vm.current_realm();
  564. auto array = MUST(Array::create(realm, 0));
  565. for (size_t i = 0; i < element_count; ++i) {
  566. auto& value = vm.bytecode_interpreter().reg(Bytecode::Register(first_register_index + i));
  567. array->indexed_properties().put(i, value, default_attributes);
  568. }
  569. return array;
  570. }
  571. void Compiler::compile_new_array(Bytecode::Op::NewArray const& op)
  572. {
  573. m_assembler.mov(
  574. Assembler::Operand::Register(ARG1),
  575. Assembler::Operand::Imm(op.element_count()));
  576. m_assembler.mov(
  577. Assembler::Operand::Register(ARG2),
  578. Assembler::Operand::Imm(op.element_count() ? op.start().index() : 0));
  579. native_call((void*)cxx_new_array);
  580. store_vm_register(Bytecode::Register::accumulator(), RET);
  581. }
  582. void Compiler::compile_new_function(Bytecode::Op::NewFunction const& op)
  583. {
  584. m_assembler.mov(
  585. Assembler::Operand::Register(ARG1),
  586. Assembler::Operand::Imm(bit_cast<u64>(&op.function_node())));
  587. m_assembler.mov(
  588. Assembler::Operand::Register(ARG2),
  589. Assembler::Operand::Imm(bit_cast<u64>(&op.lhs_name())));
  590. m_assembler.mov(
  591. Assembler::Operand::Register(ARG3),
  592. Assembler::Operand::Imm(bit_cast<u64>(&op.home_object())));
  593. native_call((void*)Bytecode::new_function);
  594. store_vm_register(Bytecode::Register::accumulator(), RET);
  595. }
  596. static Value cxx_new_class(VM& vm, ClassExpression const& class_expression, Optional<Bytecode::IdentifierTableIndex> const& lhs_name)
  597. {
  598. return TRY_OR_SET_EXCEPTION(Bytecode::new_class(vm, class_expression, lhs_name));
  599. }
  600. void Compiler::compile_new_class(Bytecode::Op::NewClass const& op)
  601. {
  602. m_assembler.mov(
  603. Assembler::Operand::Register(ARG1),
  604. Assembler::Operand::Imm(bit_cast<u64>(&op.class_expression())));
  605. m_assembler.mov(
  606. Assembler::Operand::Register(ARG2),
  607. Assembler::Operand::Imm(bit_cast<u64>(&op.lhs_name())));
  608. native_call((void*)cxx_new_class);
  609. store_vm_register(Bytecode::Register::accumulator(), RET);
  610. }
  611. static Value cxx_get_by_id(VM& vm, Value base, Bytecode::IdentifierTableIndex property, u32 cache_index)
  612. {
  613. return TRY_OR_SET_EXCEPTION(Bytecode::get_by_id(vm.bytecode_interpreter(), property, base, base, cache_index));
  614. }
  615. void Compiler::compile_get_by_id(Bytecode::Op::GetById const& op)
  616. {
  617. load_vm_register(ARG1, Bytecode::Register::accumulator());
  618. m_assembler.mov(
  619. Assembler::Operand::Register(ARG2),
  620. Assembler::Operand::Imm(op.property().value()));
  621. m_assembler.mov(
  622. Assembler::Operand::Register(ARG3),
  623. Assembler::Operand::Imm(op.cache_index()));
  624. native_call((void*)cxx_get_by_id);
  625. store_vm_register(Bytecode::Register::accumulator(), RET);
  626. check_exception();
  627. }
  628. static Value cxx_get_by_value(VM& vm, Value base, Value property)
  629. {
  630. return TRY_OR_SET_EXCEPTION(Bytecode::get_by_value(vm.bytecode_interpreter(), base, property));
  631. }
  632. void Compiler::compile_get_by_value(Bytecode::Op::GetByValue const& op)
  633. {
  634. load_vm_register(ARG1, op.base());
  635. load_vm_register(ARG2, Bytecode::Register::accumulator());
  636. native_call((void*)cxx_get_by_value);
  637. store_vm_register(Bytecode::Register::accumulator(), RET);
  638. check_exception();
  639. }
  640. static Value cxx_get_global(VM& vm, Bytecode::IdentifierTableIndex identifier, u32 cache_index)
  641. {
  642. return TRY_OR_SET_EXCEPTION(Bytecode::get_global(vm.bytecode_interpreter(), identifier, cache_index));
  643. }
  644. void Compiler::compile_get_global(Bytecode::Op::GetGlobal const& op)
  645. {
  646. m_assembler.mov(
  647. Assembler::Operand::Register(ARG1),
  648. Assembler::Operand::Imm(op.identifier().value()));
  649. m_assembler.mov(
  650. Assembler::Operand::Register(ARG2),
  651. Assembler::Operand::Imm(op.cache_index()));
  652. native_call((void*)cxx_get_global);
  653. store_vm_register(Bytecode::Register::accumulator(), RET);
  654. check_exception();
  655. }
  656. static Value cxx_get_variable(VM& vm, DeprecatedFlyString const& name, u32 cache_index)
  657. {
  658. return TRY_OR_SET_EXCEPTION(Bytecode::get_variable(vm.bytecode_interpreter(), name, cache_index));
  659. }
  660. void Compiler::compile_get_variable(Bytecode::Op::GetVariable const& op)
  661. {
  662. m_assembler.mov(
  663. Assembler::Operand::Register(ARG1),
  664. Assembler::Operand::Imm(bit_cast<u64>(&m_bytecode_executable.get_identifier(op.identifier()))));
  665. m_assembler.mov(
  666. Assembler::Operand::Register(ARG2),
  667. Assembler::Operand::Imm(op.cache_index()));
  668. native_call((void*)cxx_get_variable);
  669. store_vm_register(Bytecode::Register::accumulator(), RET);
  670. check_exception();
  671. }
  672. static Value cxx_get_callee_and_this_from_environment(VM& vm, DeprecatedFlyString const& name, u32 cache_index, Bytecode::Register callee_reg, Bytecode::Register this_reg)
  673. {
  674. auto& bytecode_interpreter = vm.bytecode_interpreter();
  675. auto callee_and_this = TRY_OR_SET_EXCEPTION(Bytecode::get_callee_and_this_from_environment(
  676. bytecode_interpreter,
  677. name,
  678. cache_index));
  679. bytecode_interpreter.reg(callee_reg) = callee_and_this.callee;
  680. bytecode_interpreter.reg(this_reg) = callee_and_this.this_value;
  681. return {};
  682. }
  683. void Compiler::compile_get_callee_and_this_from_environment(Bytecode::Op::GetCalleeAndThisFromEnvironment const& op)
  684. {
  685. m_assembler.mov(
  686. Assembler::Operand::Register(ARG1),
  687. Assembler::Operand::Imm(bit_cast<u64>(&m_bytecode_executable.get_identifier(op.identifier()))));
  688. m_assembler.mov(
  689. Assembler::Operand::Register(ARG2),
  690. Assembler::Operand::Imm(op.cache_index()));
  691. m_assembler.mov(
  692. Assembler::Operand::Register(ARG3),
  693. Assembler::Operand::Imm(op.callee().index()));
  694. m_assembler.mov(
  695. Assembler::Operand::Register(ARG4),
  696. Assembler::Operand::Imm(op.this_().index()));
  697. native_call((void*)cxx_get_callee_and_this_from_environment);
  698. check_exception();
  699. }
  700. static Value cxx_to_numeric(VM& vm, Value value)
  701. {
  702. return TRY_OR_SET_EXCEPTION(value.to_numeric(vm));
  703. }
  704. void Compiler::compile_to_numeric(Bytecode::Op::ToNumeric const&)
  705. {
  706. load_vm_register(ARG1, Bytecode::Register::accumulator());
  707. native_call((void*)cxx_to_numeric);
  708. store_vm_register(Bytecode::Register::accumulator(), RET);
  709. check_exception();
  710. }
  711. static Value cxx_resolve_this_binding(VM& vm)
  712. {
  713. auto this_value = TRY_OR_SET_EXCEPTION(vm.resolve_this_binding());
  714. vm.bytecode_interpreter().reg(Bytecode::Register::this_value()) = this_value;
  715. return this_value;
  716. }
  717. void Compiler::compile_resolve_this_binding(Bytecode::Op::ResolveThisBinding const&)
  718. {
  719. // OPTIMIZATION: We cache the `this` value in a special VM register.
  720. // So first we check if the cache is non-empty, and if so,
  721. // we can avoid calling out to C++ at all. :^)
  722. load_vm_register(GPR0, Bytecode::Register::this_value());
  723. m_assembler.mov(
  724. Assembler::Operand::Register(GPR1),
  725. Assembler::Operand::Imm(Value().encoded()));
  726. Assembler::Label slow_case {};
  727. m_assembler.jump_if_equal(
  728. Assembler::Operand::Register(GPR0),
  729. Assembler::Operand::Register(GPR1),
  730. slow_case);
  731. // Fast case: We have a cached `this` value!
  732. store_vm_register(Bytecode::Register::accumulator(), GPR0);
  733. auto end = m_assembler.jump();
  734. slow_case.link(m_assembler);
  735. native_call((void*)cxx_resolve_this_binding);
  736. store_vm_register(Bytecode::Register::accumulator(), RET);
  737. check_exception();
  738. end.link(m_assembler);
  739. }
  740. static Value cxx_put_by_id(VM& vm, Value base, Bytecode::IdentifierTableIndex property, Value value, Bytecode::Op::PropertyKind kind)
  741. {
  742. PropertyKey name = vm.bytecode_interpreter().current_executable().get_identifier(property);
  743. TRY_OR_SET_EXCEPTION(Bytecode::put_by_property_key(vm, base, base, value, name, kind));
  744. vm.bytecode_interpreter().accumulator() = value;
  745. return {};
  746. }
  747. void Compiler::compile_put_by_id(Bytecode::Op::PutById const& op)
  748. {
  749. load_vm_register(ARG1, op.base());
  750. m_assembler.mov(
  751. Assembler::Operand::Register(ARG2),
  752. Assembler::Operand::Imm(op.property().value()));
  753. load_vm_register(ARG3, Bytecode::Register::accumulator());
  754. m_assembler.mov(
  755. Assembler::Operand::Register(ARG4),
  756. Assembler::Operand::Imm(to_underlying(op.kind())));
  757. native_call((void*)cxx_put_by_id);
  758. check_exception();
  759. }
  760. static Value cxx_put_by_value(VM& vm, Value base, Value property, Value value, Bytecode::Op::PropertyKind kind)
  761. {
  762. TRY_OR_SET_EXCEPTION(Bytecode::put_by_value(vm, base, property, value, kind));
  763. vm.bytecode_interpreter().accumulator() = value;
  764. return {};
  765. }
  766. void Compiler::compile_put_by_value(Bytecode::Op::PutByValue const& op)
  767. {
  768. load_vm_register(ARG1, op.base());
  769. load_vm_register(ARG2, op.property());
  770. load_vm_register(ARG3, Bytecode::Register::accumulator());
  771. m_assembler.mov(
  772. Assembler::Operand::Register(ARG4),
  773. Assembler::Operand::Imm(to_underlying(op.kind())));
  774. native_call((void*)cxx_put_by_value);
  775. check_exception();
  776. }
  777. static Value cxx_call(VM& vm, Value callee, u32 first_argument_index, u32 argument_count, Value this_value, Bytecode::Op::CallType call_type, Optional<Bytecode::StringTableIndex> const& expression_string)
  778. {
  779. TRY_OR_SET_EXCEPTION(throw_if_needed_for_call(vm.bytecode_interpreter(), callee, call_type, expression_string));
  780. MarkedVector<Value> argument_values(vm.heap());
  781. argument_values.ensure_capacity(argument_count);
  782. for (u32 i = 0; i < argument_count; ++i) {
  783. argument_values.unchecked_append(vm.bytecode_interpreter().reg(Bytecode::Register { first_argument_index + i }));
  784. }
  785. return TRY_OR_SET_EXCEPTION(perform_call(vm.bytecode_interpreter(), this_value, call_type, callee, move(argument_values)));
  786. }
  787. void Compiler::compile_call(Bytecode::Op::Call const& op)
  788. {
  789. load_vm_register(ARG1, op.callee());
  790. m_assembler.mov(
  791. Assembler::Operand::Register(ARG2),
  792. Assembler::Operand::Imm(op.first_argument().index()));
  793. m_assembler.mov(
  794. Assembler::Operand::Register(ARG3),
  795. Assembler::Operand::Imm(op.argument_count()));
  796. load_vm_register(ARG4, op.this_value());
  797. m_assembler.mov(
  798. Assembler::Operand::Register(ARG5),
  799. Assembler::Operand::Imm(to_underlying(op.call_type())));
  800. m_assembler.mov(
  801. Assembler::Operand::Register(GPR0),
  802. Assembler::Operand::Imm(bit_cast<u64>(&op.expression_string())));
  803. native_call((void*)cxx_call, { Assembler::Operand::Register(GPR0) });
  804. store_vm_register(Bytecode::Register::accumulator(), RET);
  805. check_exception();
  806. }
  807. static Value cxx_call_with_argument_array(VM& vm, Value callee, Value this_value, Bytecode::Op::CallType call_type, Optional<Bytecode::StringTableIndex> const& expression_string)
  808. {
  809. TRY_OR_SET_EXCEPTION(throw_if_needed_for_call(vm.bytecode_interpreter(), callee, call_type, expression_string));
  810. auto argument_values = argument_list_evaluation(vm.bytecode_interpreter());
  811. return TRY_OR_SET_EXCEPTION(perform_call(vm.bytecode_interpreter(), this_value, call_type, callee, move(argument_values)));
  812. }
  813. void Compiler::compile_call_with_argument_array(Bytecode::Op::CallWithArgumentArray const& op)
  814. {
  815. load_vm_register(ARG1, op.callee());
  816. load_vm_register(ARG2, op.this_value());
  817. m_assembler.mov(
  818. Assembler::Operand::Register(ARG3),
  819. Assembler::Operand::Imm(to_underlying(op.call_type())));
  820. m_assembler.mov(
  821. Assembler::Operand::Register(ARG4),
  822. Assembler::Operand::Imm(bit_cast<u64>(&op.expression_string())));
  823. native_call((void*)cxx_call_with_argument_array);
  824. store_vm_register(Bytecode::Register::accumulator(), RET);
  825. check_exception();
  826. }
  827. static Value cxx_typeof_variable(VM& vm, DeprecatedFlyString const& identifier)
  828. {
  829. return TRY_OR_SET_EXCEPTION(Bytecode::typeof_variable(vm, identifier));
  830. }
  831. void Compiler::compile_typeof_variable(Bytecode::Op::TypeofVariable const& op)
  832. {
  833. m_assembler.mov(
  834. Assembler::Operand::Register(ARG1),
  835. Assembler::Operand::Imm(bit_cast<u64>(&m_bytecode_executable.get_identifier(op.identifier().value()))));
  836. native_call((void*)cxx_typeof_variable);
  837. store_vm_register(Bytecode::Register::accumulator(), RET);
  838. check_exception();
  839. }
  840. static Value cxx_create_variable(
  841. VM& vm,
  842. DeprecatedFlyString const& name,
  843. Bytecode::Op::EnvironmentMode mode,
  844. bool is_global,
  845. bool is_immutable,
  846. bool is_strict)
  847. {
  848. TRY_OR_SET_EXCEPTION(Bytecode::create_variable(vm, name, mode, is_global, is_immutable, is_strict));
  849. return {};
  850. }
  851. void Compiler::compile_create_variable(Bytecode::Op::CreateVariable const& op)
  852. {
  853. m_assembler.mov(
  854. Assembler::Operand::Register(ARG1),
  855. Assembler::Operand::Imm(bit_cast<u64>(&m_bytecode_executable.get_identifier(op.identifier().value()))));
  856. m_assembler.mov(
  857. Assembler::Operand::Register(ARG2),
  858. Assembler::Operand::Imm(to_underlying(op.mode())));
  859. m_assembler.mov(
  860. Assembler::Operand::Register(ARG3),
  861. Assembler::Operand::Imm(static_cast<u64>(op.is_global())));
  862. m_assembler.mov(
  863. Assembler::Operand::Register(ARG4),
  864. Assembler::Operand::Imm(static_cast<u64>(op.is_immutable())));
  865. m_assembler.mov(
  866. Assembler::Operand::Register(ARG5),
  867. Assembler::Operand::Imm(static_cast<u64>(op.is_strict())));
  868. native_call((void*)cxx_create_variable);
  869. check_exception();
  870. }
  871. static Value cxx_set_variable(
  872. VM& vm,
  873. DeprecatedFlyString const& identifier,
  874. Value value,
  875. Bytecode::Op::EnvironmentMode environment_mode,
  876. Bytecode::Op::SetVariable::InitializationMode initialization_mode)
  877. {
  878. TRY_OR_SET_EXCEPTION(Bytecode::set_variable(vm, identifier, value, environment_mode, initialization_mode));
  879. return {};
  880. }
  881. void Compiler::compile_set_variable(Bytecode::Op::SetVariable const& op)
  882. {
  883. m_assembler.mov(
  884. Assembler::Operand::Register(ARG1),
  885. Assembler::Operand::Imm(bit_cast<u64>(&m_bytecode_executable.get_identifier(op.identifier().value()))));
  886. load_vm_register(ARG2, Bytecode::Register::accumulator());
  887. m_assembler.mov(
  888. Assembler::Operand::Register(ARG3),
  889. Assembler::Operand::Imm(to_underlying(op.mode())));
  890. m_assembler.mov(
  891. Assembler::Operand::Register(ARG4),
  892. Assembler::Operand::Imm(to_underlying(op.initialization_mode())));
  893. native_call((void*)cxx_set_variable);
  894. check_exception();
  895. }
  896. void Compiler::compile_continue_pending_unwind(Bytecode::Op::ContinuePendingUnwind const& op)
  897. {
  898. // re-throw the exception if we reached the end of the finally block and there was no catch block to handle it
  899. check_exception();
  900. // if (!saved_return_value.is_empty()) goto resume_block;
  901. load_vm_register(GPR0, Bytecode::Register::saved_return_value());
  902. m_assembler.mov(Assembler::Operand::Register(GPR1), Assembler::Operand::Imm(Value().encoded()));
  903. m_assembler.jump_if_not_equal(Assembler::Operand::Register(GPR0), Assembler::Operand::Register(GPR1), label_for(op.resume_target().block()));
  904. // finish the pending return from the try block
  905. store_vm_register(Bytecode::Register::return_value(), GPR0);
  906. jump_to_exit();
  907. }
  908. static void cxx_create_lexical_environment(VM& vm)
  909. {
  910. auto make_and_swap_envs = [&](auto& old_environment) {
  911. GCPtr<Environment> environment = new_declarative_environment(*old_environment).ptr();
  912. swap(old_environment, environment);
  913. return environment;
  914. };
  915. vm.bytecode_interpreter().saved_lexical_environment_stack().append(make_and_swap_envs(vm.running_execution_context().lexical_environment));
  916. }
  917. void Compiler::compile_create_lexical_environment(Bytecode::Op::CreateLexicalEnvironment const&)
  918. {
  919. native_call((void*)cxx_create_lexical_environment);
  920. }
  921. static void cxx_leave_lexical_environment(VM& vm)
  922. {
  923. vm.running_execution_context().lexical_environment = vm.bytecode_interpreter().saved_lexical_environment_stack().take_last();
  924. }
  925. void Compiler::compile_leave_lexical_environment(Bytecode::Op::LeaveLexicalEnvironment const&)
  926. {
  927. native_call((void*)cxx_leave_lexical_environment);
  928. }
  929. static Value cxx_concat_string(VM& vm, Value lhs, Value rhs)
  930. {
  931. auto string = TRY_OR_SET_EXCEPTION(rhs.to_primitive_string(vm));
  932. return PrimitiveString::create(vm, lhs.as_string(), string);
  933. }
  934. void Compiler::compile_concat_string(Bytecode::Op::ConcatString const& op)
  935. {
  936. load_vm_register(ARG1, op.lhs());
  937. load_vm_register(ARG2, Bytecode::Register::accumulator());
  938. native_call((void*)cxx_concat_string);
  939. store_vm_register(op.lhs(), RET);
  940. check_exception();
  941. }
  942. void Compiler::jump_to_exit()
  943. {
  944. m_assembler.jump(m_exit_label);
  945. }
  946. void Compiler::native_call(void* function_address, Vector<Assembler::Operand> const& stack_arguments)
  947. {
  948. // Make sure we don't clobber the VM&.
  949. m_assembler.push(Assembler::Operand::Register(ARG0));
  950. // Align the stack pointer.
  951. m_assembler.sub(Assembler::Operand::Register(STACK_POINTER), Assembler::Operand::Imm(8));
  952. // NOTE: We don't preserve caller-saved registers when making a native call.
  953. // This means that they may have changed after we return from the call.
  954. m_assembler.native_call(function_address, stack_arguments);
  955. // Restore the stack pointer.
  956. m_assembler.add(Assembler::Operand::Register(STACK_POINTER), Assembler::Operand::Imm(8));
  957. // Restore our VM&.
  958. m_assembler.pop(Assembler::Operand::Register(ARG0));
  959. }
  960. OwnPtr<NativeExecutable> Compiler::compile(Bytecode::Executable& bytecode_executable)
  961. {
  962. if (!getenv("LIBJS_JIT"))
  963. return nullptr;
  964. Compiler compiler { bytecode_executable };
  965. compiler.m_assembler.enter();
  966. compiler.m_assembler.mov(
  967. Assembler::Operand::Register(REGISTER_ARRAY_BASE),
  968. Assembler::Operand::Register(ARG1));
  969. compiler.m_assembler.mov(
  970. Assembler::Operand::Register(LOCALS_ARRAY_BASE),
  971. Assembler::Operand::Register(ARG2));
  972. compiler.push_unwind_context(false, {}, {});
  973. for (auto& block : bytecode_executable.basic_blocks) {
  974. compiler.block_data_for(*block).start_offset = compiler.m_output.size();
  975. auto it = Bytecode::InstructionStreamIterator(block->instruction_stream());
  976. while (!it.at_end()) {
  977. auto const& op = *it;
  978. switch (op.type()) {
  979. case Bytecode::Instruction::Type::LoadImmediate:
  980. compiler.compile_load_immediate(static_cast<Bytecode::Op::LoadImmediate const&>(op));
  981. break;
  982. case Bytecode::Instruction::Type::Store:
  983. compiler.compile_store(static_cast<Bytecode::Op::Store const&>(op));
  984. break;
  985. case Bytecode::Instruction::Type::Load:
  986. compiler.compile_load(static_cast<Bytecode::Op::Load const&>(op));
  987. break;
  988. case Bytecode::Instruction::Type::GetLocal:
  989. compiler.compile_get_local(static_cast<Bytecode::Op::GetLocal const&>(op));
  990. break;
  991. case Bytecode::Instruction::Type::SetLocal:
  992. compiler.compile_set_local(static_cast<Bytecode::Op::SetLocal const&>(op));
  993. break;
  994. case Bytecode::Instruction::Type::TypeofLocal:
  995. compiler.compile_typeof_local(static_cast<Bytecode::Op::TypeofLocal const&>(op));
  996. break;
  997. case Bytecode::Instruction::Type::Jump:
  998. compiler.compile_jump(static_cast<Bytecode::Op::Jump const&>(op));
  999. break;
  1000. case Bytecode::Instruction::Type::JumpConditional:
  1001. compiler.compile_jump_conditional(static_cast<Bytecode::Op::JumpConditional const&>(op));
  1002. break;
  1003. case Bytecode::Instruction::Type::JumpNullish:
  1004. compiler.compile_jump_nullish(static_cast<Bytecode::Op::JumpNullish const&>(op));
  1005. break;
  1006. case Bytecode::Instruction::Type::Increment:
  1007. compiler.compile_increment(static_cast<Bytecode::Op::Increment const&>(op));
  1008. break;
  1009. case Bytecode::Instruction::Type::Decrement:
  1010. compiler.compile_decrement(static_cast<Bytecode::Op::Decrement const&>(op));
  1011. break;
  1012. case Bytecode::Instruction::Type::EnterUnwindContext:
  1013. compiler.compile_enter_unwind_context(static_cast<Bytecode::Op::EnterUnwindContext const&>(op));
  1014. break;
  1015. case Bytecode::Instruction::Type::LeaveUnwindContext:
  1016. compiler.compile_leave_unwind_context(static_cast<Bytecode::Op::LeaveUnwindContext const&>(op));
  1017. break;
  1018. case Bytecode::Instruction::Type::Throw:
  1019. compiler.compile_throw(static_cast<Bytecode::Op::Throw const&>(op));
  1020. break;
  1021. case Bytecode::Instruction::Type::Return:
  1022. compiler.compile_return(static_cast<Bytecode::Op::Return const&>(op));
  1023. break;
  1024. case Bytecode::Instruction::Type::CreateLexicalEnvironment:
  1025. compiler.compile_create_lexical_environment(static_cast<Bytecode::Op::CreateLexicalEnvironment const&>(op));
  1026. break;
  1027. case Bytecode::Instruction::Type::LeaveLexicalEnvironment:
  1028. compiler.compile_leave_lexical_environment(static_cast<Bytecode::Op::LeaveLexicalEnvironment const&>(op));
  1029. break;
  1030. case Bytecode::Instruction::Type::NewString:
  1031. compiler.compile_new_string(static_cast<Bytecode::Op::NewString const&>(op));
  1032. break;
  1033. case Bytecode::Instruction::Type::NewObject:
  1034. compiler.compile_new_object(static_cast<Bytecode::Op::NewObject const&>(op));
  1035. break;
  1036. case Bytecode::Instruction::Type::NewArray:
  1037. compiler.compile_new_array(static_cast<Bytecode::Op::NewArray const&>(op));
  1038. break;
  1039. case Bytecode::Instruction::Type::NewFunction:
  1040. compiler.compile_new_function(static_cast<Bytecode::Op::NewFunction const&>(op));
  1041. break;
  1042. case Bytecode::Instruction::Type::NewRegExp:
  1043. compiler.compile_new_regexp(static_cast<Bytecode::Op::NewRegExp const&>(op));
  1044. break;
  1045. case Bytecode::Instruction::Type::NewBigInt:
  1046. compiler.compile_new_bigint(static_cast<Bytecode::Op::NewBigInt const&>(op));
  1047. break;
  1048. case Bytecode::Instruction::Type::GetById:
  1049. compiler.compile_get_by_id(static_cast<Bytecode::Op::GetById const&>(op));
  1050. break;
  1051. case Bytecode::Instruction::Type::GetByValue:
  1052. compiler.compile_get_by_value(static_cast<Bytecode::Op::GetByValue const&>(op));
  1053. break;
  1054. case Bytecode::Instruction::Type::GetGlobal:
  1055. compiler.compile_get_global(static_cast<Bytecode::Op::GetGlobal const&>(op));
  1056. break;
  1057. case Bytecode::Instruction::Type::GetVariable:
  1058. compiler.compile_get_variable(static_cast<Bytecode::Op::GetVariable const&>(op));
  1059. break;
  1060. case Bytecode::Instruction::Type::GetCalleeAndThisFromEnvironment:
  1061. compiler.compile_get_callee_and_this_from_environment(static_cast<Bytecode::Op::GetCalleeAndThisFromEnvironment const&>(op));
  1062. break;
  1063. case Bytecode::Instruction::Type::PutById:
  1064. compiler.compile_put_by_id(static_cast<Bytecode::Op::PutById const&>(op));
  1065. break;
  1066. case Bytecode::Instruction::Type::PutByValue:
  1067. compiler.compile_put_by_value(static_cast<Bytecode::Op::PutByValue const&>(op));
  1068. break;
  1069. case Bytecode::Instruction::Type::ToNumeric:
  1070. compiler.compile_to_numeric(static_cast<Bytecode::Op::ToNumeric const&>(op));
  1071. break;
  1072. case Bytecode::Instruction::Type::ResolveThisBinding:
  1073. compiler.compile_resolve_this_binding(static_cast<Bytecode::Op::ResolveThisBinding const&>(op));
  1074. break;
  1075. case Bytecode::Instruction::Type::Call:
  1076. compiler.compile_call(static_cast<Bytecode::Op::Call const&>(op));
  1077. break;
  1078. case Bytecode::Instruction::Type::CallWithArgumentArray:
  1079. compiler.compile_call_with_argument_array(static_cast<Bytecode::Op::CallWithArgumentArray const&>(op));
  1080. break;
  1081. case Bytecode::Instruction::Type::TypeofVariable:
  1082. compiler.compile_typeof_variable(static_cast<Bytecode::Op::TypeofVariable const&>(op));
  1083. break;
  1084. case Bytecode::Instruction::Type::SetVariable:
  1085. compiler.compile_set_variable(static_cast<Bytecode::Op::SetVariable const&>(op));
  1086. break;
  1087. case Bytecode::Instruction::Type::LessThan:
  1088. compiler.compile_less_than(static_cast<Bytecode::Op::LessThan const&>(op));
  1089. break;
  1090. case Bytecode::Instruction::Type::ContinuePendingUnwind:
  1091. compiler.compile_continue_pending_unwind(static_cast<Bytecode::Op::ContinuePendingUnwind const&>(op));
  1092. break;
  1093. case Bytecode::Instruction::Type::ConcatString:
  1094. compiler.compile_concat_string(static_cast<Bytecode::Op::ConcatString const&>(op));
  1095. break;
  1096. case Bytecode::Instruction::Type::CreateVariable:
  1097. compiler.compile_create_variable(static_cast<Bytecode::Op::CreateVariable const&>(op));
  1098. break;
  1099. case Bytecode::Instruction::Type::NewClass:
  1100. compiler.compile_new_class(static_cast<Bytecode::Op::NewClass const&>(op));
  1101. break;
  1102. # define DO_COMPILE_COMMON_BINARY_OP(TitleCaseName, snake_case_name) \
  1103. case Bytecode::Instruction::Type::TitleCaseName: \
  1104. compiler.compile_##snake_case_name(static_cast<Bytecode::Op::TitleCaseName const&>(op)); \
  1105. break;
  1106. JS_ENUMERATE_COMMON_BINARY_OPS_WITHOUT_FAST_PATH(DO_COMPILE_COMMON_BINARY_OP)
  1107. # undef DO_COMPILE_COMMON_BINARY_OP
  1108. # define DO_COMPILE_COMMON_UNARY_OP(TitleCaseName, snake_case_name) \
  1109. case Bytecode::Instruction::Type::TitleCaseName: \
  1110. compiler.compile_##snake_case_name(static_cast<Bytecode::Op::TitleCaseName const&>(op)); \
  1111. break;
  1112. JS_ENUMERATE_COMMON_UNARY_OPS(DO_COMPILE_COMMON_UNARY_OP)
  1113. # undef DO_COMPILE_COMMON_UNARY_OP
  1114. default:
  1115. if constexpr (LOG_JIT_FAILURE) {
  1116. dbgln("\033[31;1mJIT compilation failed\033[0m: {}", bytecode_executable.name);
  1117. dbgln("Unsupported bytecode op: {}", op.to_deprecated_string(bytecode_executable));
  1118. }
  1119. return nullptr;
  1120. }
  1121. ++it;
  1122. }
  1123. if (!block->is_terminated())
  1124. compiler.jump_to_exit();
  1125. }
  1126. compiler.m_exit_label.link(compiler.m_assembler);
  1127. compiler.m_assembler.exit();
  1128. if (!compiler.m_exception_handler.jump_slot_offsets_in_instruction_stream.is_empty()) {
  1129. compiler.m_exception_handler.link(compiler.m_assembler);
  1130. compiler.handle_exception();
  1131. }
  1132. auto* executable_memory = mmap(nullptr, compiler.m_output.size(), PROT_READ | PROT_WRITE, MAP_ANONYMOUS | MAP_PRIVATE, 0, 0);
  1133. if (executable_memory == MAP_FAILED) {
  1134. dbgln("mmap: {}", strerror(errno));
  1135. return nullptr;
  1136. }
  1137. for (auto& block : bytecode_executable.basic_blocks) {
  1138. auto& block_data = compiler.block_data_for(*block);
  1139. block_data.label.link_to(compiler.m_assembler, block_data.start_offset);
  1140. // Patch up all the absolute references
  1141. for (auto& absolute_reference : block_data.absolute_references_to_here) {
  1142. auto offset = bit_cast<u64>(executable_memory) + block_data.start_offset;
  1143. compiler.m_output[absolute_reference + 0] = (offset >> 0) & 0xff;
  1144. compiler.m_output[absolute_reference + 1] = (offset >> 8) & 0xff;
  1145. compiler.m_output[absolute_reference + 2] = (offset >> 16) & 0xff;
  1146. compiler.m_output[absolute_reference + 3] = (offset >> 24) & 0xff;
  1147. compiler.m_output[absolute_reference + 4] = (offset >> 32) & 0xff;
  1148. compiler.m_output[absolute_reference + 5] = (offset >> 40) & 0xff;
  1149. compiler.m_output[absolute_reference + 6] = (offset >> 48) & 0xff;
  1150. compiler.m_output[absolute_reference + 7] = (offset >> 56) & 0xff;
  1151. }
  1152. }
  1153. if constexpr (DUMP_JIT_MACHINE_CODE_TO_STDOUT) {
  1154. (void)write(STDOUT_FILENO, compiler.m_output.data(), compiler.m_output.size());
  1155. }
  1156. memcpy(executable_memory, compiler.m_output.data(), compiler.m_output.size());
  1157. if (mprotect(executable_memory, compiler.m_output.size(), PROT_READ | PROT_EXEC) < 0) {
  1158. dbgln("mprotect: {}", strerror(errno));
  1159. return nullptr;
  1160. }
  1161. if constexpr (LOG_JIT_SUCCESS) {
  1162. dbgln("\033[32;1mJIT compilation succeeded!\033[0m {}", bytecode_executable.name);
  1163. }
  1164. auto executable = make<NativeExecutable>(executable_memory, compiler.m_output.size());
  1165. if constexpr (DUMP_JIT_DISASSEMBLY)
  1166. executable->dump_disassembly();
  1167. return executable;
  1168. }
  1169. }
  1170. #endif