ASTCodegen.cpp 51 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372
  1. /*
  2. * Copyright (c) 2021, Andreas Kling <kling@serenityos.org>
  3. * Copyright (c) 2021, Linus Groh <linusg@serenityos.org>
  4. * Copyright (c) 2021, Gunnar Beutner <gbeutner@serenityos.org>
  5. * Copyright (c) 2021, Marcin Gasperowicz <xnooga@gmail.com>
  6. *
  7. * SPDX-License-Identifier: BSD-2-Clause
  8. */
  9. #include <AK/Format.h>
  10. #include <LibJS/AST.h>
  11. #include <LibJS/Bytecode/Generator.h>
  12. #include <LibJS/Bytecode/Instruction.h>
  13. #include <LibJS/Bytecode/Op.h>
  14. #include <LibJS/Bytecode/Register.h>
  15. #include <LibJS/Bytecode/StringTable.h>
  16. #include <LibJS/Runtime/Environment.h>
  17. namespace JS {
  18. Bytecode::CodeGenerationErrorOr<void> ASTNode::generate_bytecode(Bytecode::Generator&) const
  19. {
  20. return Bytecode::CodeGenerationError {
  21. this,
  22. "Missing generate_bytecode()"sv,
  23. };
  24. }
  25. Bytecode::CodeGenerationErrorOr<void> ScopeNode::generate_bytecode(Bytecode::Generator& generator) const
  26. {
  27. // FIXME: This is an ad-hoc fix but should be done as the spec says in
  28. // {Global, Block, Function, Eval}DeclarationInstantiation.
  29. for (auto& function : m_functions_hoistable_with_annexB_extension) {
  30. generator.emit<Bytecode::Op::NewFunction>(function);
  31. generator.emit<Bytecode::Op::SetVariable>(generator.intern_identifier(function.name()));
  32. }
  33. HashTable<FlyString> functions_initialized;
  34. for_each_var_function_declaration_in_reverse_order([&](FunctionDeclaration const& function) {
  35. if (functions_initialized.set(function.name()) != AK::HashSetResult::InsertedNewEntry)
  36. return;
  37. generator.emit<Bytecode::Op::NewFunction>(function);
  38. generator.emit<Bytecode::Op::SetVariable>(generator.intern_identifier(function.name()));
  39. });
  40. // FIXME: Register lexical and variable scope declarations
  41. for (auto& child : children()) {
  42. TRY(child.generate_bytecode(generator));
  43. if (generator.is_current_block_terminated())
  44. break;
  45. }
  46. return {};
  47. }
  48. Bytecode::CodeGenerationErrorOr<void> EmptyStatement::generate_bytecode(Bytecode::Generator&) const
  49. {
  50. return {};
  51. }
  52. Bytecode::CodeGenerationErrorOr<void> ExpressionStatement::generate_bytecode(Bytecode::Generator& generator) const
  53. {
  54. return m_expression->generate_bytecode(generator);
  55. }
  56. Bytecode::CodeGenerationErrorOr<void> BinaryExpression::generate_bytecode(Bytecode::Generator& generator) const
  57. {
  58. TRY(m_lhs->generate_bytecode(generator));
  59. auto lhs_reg = generator.allocate_register();
  60. generator.emit<Bytecode::Op::Store>(lhs_reg);
  61. TRY(m_rhs->generate_bytecode(generator));
  62. switch (m_op) {
  63. case BinaryOp::Addition:
  64. generator.emit<Bytecode::Op::Add>(lhs_reg);
  65. break;
  66. case BinaryOp::Subtraction:
  67. generator.emit<Bytecode::Op::Sub>(lhs_reg);
  68. break;
  69. case BinaryOp::Multiplication:
  70. generator.emit<Bytecode::Op::Mul>(lhs_reg);
  71. break;
  72. case BinaryOp::Division:
  73. generator.emit<Bytecode::Op::Div>(lhs_reg);
  74. break;
  75. case BinaryOp::Modulo:
  76. generator.emit<Bytecode::Op::Mod>(lhs_reg);
  77. break;
  78. case BinaryOp::Exponentiation:
  79. generator.emit<Bytecode::Op::Exp>(lhs_reg);
  80. break;
  81. case BinaryOp::GreaterThan:
  82. generator.emit<Bytecode::Op::GreaterThan>(lhs_reg);
  83. break;
  84. case BinaryOp::GreaterThanEquals:
  85. generator.emit<Bytecode::Op::GreaterThanEquals>(lhs_reg);
  86. break;
  87. case BinaryOp::LessThan:
  88. generator.emit<Bytecode::Op::LessThan>(lhs_reg);
  89. break;
  90. case BinaryOp::LessThanEquals:
  91. generator.emit<Bytecode::Op::LessThanEquals>(lhs_reg);
  92. break;
  93. case BinaryOp::LooselyInequals:
  94. generator.emit<Bytecode::Op::LooselyInequals>(lhs_reg);
  95. break;
  96. case BinaryOp::LooselyEquals:
  97. generator.emit<Bytecode::Op::LooselyEquals>(lhs_reg);
  98. break;
  99. case BinaryOp::StrictlyInequals:
  100. generator.emit<Bytecode::Op::StrictlyInequals>(lhs_reg);
  101. break;
  102. case BinaryOp::StrictlyEquals:
  103. generator.emit<Bytecode::Op::StrictlyEquals>(lhs_reg);
  104. break;
  105. case BinaryOp::BitwiseAnd:
  106. generator.emit<Bytecode::Op::BitwiseAnd>(lhs_reg);
  107. break;
  108. case BinaryOp::BitwiseOr:
  109. generator.emit<Bytecode::Op::BitwiseOr>(lhs_reg);
  110. break;
  111. case BinaryOp::BitwiseXor:
  112. generator.emit<Bytecode::Op::BitwiseXor>(lhs_reg);
  113. break;
  114. case BinaryOp::LeftShift:
  115. generator.emit<Bytecode::Op::LeftShift>(lhs_reg);
  116. break;
  117. case BinaryOp::RightShift:
  118. generator.emit<Bytecode::Op::RightShift>(lhs_reg);
  119. break;
  120. case BinaryOp::UnsignedRightShift:
  121. generator.emit<Bytecode::Op::UnsignedRightShift>(lhs_reg);
  122. break;
  123. case BinaryOp::In:
  124. generator.emit<Bytecode::Op::In>(lhs_reg);
  125. break;
  126. case BinaryOp::InstanceOf:
  127. generator.emit<Bytecode::Op::InstanceOf>(lhs_reg);
  128. break;
  129. default:
  130. VERIFY_NOT_REACHED();
  131. }
  132. return {};
  133. }
  134. Bytecode::CodeGenerationErrorOr<void> LogicalExpression::generate_bytecode(Bytecode::Generator& generator) const
  135. {
  136. TRY(m_lhs->generate_bytecode(generator));
  137. // lhs
  138. // jump op (true) end (false) rhs
  139. // rhs
  140. // jump always (true) end
  141. // end
  142. auto& rhs_block = generator.make_block();
  143. auto& end_block = generator.make_block();
  144. switch (m_op) {
  145. case LogicalOp::And:
  146. generator.emit<Bytecode::Op::JumpConditional>().set_targets(
  147. Bytecode::Label { rhs_block },
  148. Bytecode::Label { end_block });
  149. break;
  150. case LogicalOp::Or:
  151. generator.emit<Bytecode::Op::JumpConditional>().set_targets(
  152. Bytecode::Label { end_block },
  153. Bytecode::Label { rhs_block });
  154. break;
  155. case LogicalOp::NullishCoalescing:
  156. generator.emit<Bytecode::Op::JumpNullish>().set_targets(
  157. Bytecode::Label { rhs_block },
  158. Bytecode::Label { end_block });
  159. break;
  160. default:
  161. VERIFY_NOT_REACHED();
  162. }
  163. generator.switch_to_basic_block(rhs_block);
  164. TRY(m_rhs->generate_bytecode(generator));
  165. generator.emit<Bytecode::Op::Jump>().set_targets(
  166. Bytecode::Label { end_block },
  167. {});
  168. generator.switch_to_basic_block(end_block);
  169. return {};
  170. }
  171. Bytecode::CodeGenerationErrorOr<void> UnaryExpression::generate_bytecode(Bytecode::Generator& generator) const
  172. {
  173. TRY(m_lhs->generate_bytecode(generator));
  174. switch (m_op) {
  175. case UnaryOp::BitwiseNot:
  176. generator.emit<Bytecode::Op::BitwiseNot>();
  177. break;
  178. case UnaryOp::Not:
  179. generator.emit<Bytecode::Op::Not>();
  180. break;
  181. case UnaryOp::Plus:
  182. generator.emit<Bytecode::Op::UnaryPlus>();
  183. break;
  184. case UnaryOp::Minus:
  185. generator.emit<Bytecode::Op::UnaryMinus>();
  186. break;
  187. case UnaryOp::Typeof:
  188. generator.emit<Bytecode::Op::Typeof>();
  189. break;
  190. case UnaryOp::Void:
  191. generator.emit<Bytecode::Op::LoadImmediate>(js_undefined());
  192. break;
  193. default:
  194. return Bytecode::CodeGenerationError {
  195. this,
  196. "Unimplemented operation"sv,
  197. };
  198. }
  199. return {};
  200. }
  201. Bytecode::CodeGenerationErrorOr<void> NumericLiteral::generate_bytecode(Bytecode::Generator& generator) const
  202. {
  203. generator.emit<Bytecode::Op::LoadImmediate>(m_value);
  204. return {};
  205. }
  206. Bytecode::CodeGenerationErrorOr<void> BooleanLiteral::generate_bytecode(Bytecode::Generator& generator) const
  207. {
  208. generator.emit<Bytecode::Op::LoadImmediate>(Value(m_value));
  209. return {};
  210. }
  211. Bytecode::CodeGenerationErrorOr<void> NullLiteral::generate_bytecode(Bytecode::Generator& generator) const
  212. {
  213. generator.emit<Bytecode::Op::LoadImmediate>(js_null());
  214. return {};
  215. }
  216. Bytecode::CodeGenerationErrorOr<void> BigIntLiteral::generate_bytecode(Bytecode::Generator& generator) const
  217. {
  218. generator.emit<Bytecode::Op::NewBigInt>(Crypto::SignedBigInteger::from_base(10, m_value.substring(0, m_value.length() - 1)));
  219. return {};
  220. }
  221. Bytecode::CodeGenerationErrorOr<void> StringLiteral::generate_bytecode(Bytecode::Generator& generator) const
  222. {
  223. generator.emit<Bytecode::Op::NewString>(generator.intern_string(m_value));
  224. return {};
  225. }
  226. Bytecode::CodeGenerationErrorOr<void> RegExpLiteral::generate_bytecode(Bytecode::Generator& generator) const
  227. {
  228. auto source_index = generator.intern_string(m_pattern);
  229. auto flags_index = generator.intern_string(m_flags);
  230. generator.emit<Bytecode::Op::NewRegExp>(source_index, flags_index);
  231. return {};
  232. }
  233. Bytecode::CodeGenerationErrorOr<void> Identifier::generate_bytecode(Bytecode::Generator& generator) const
  234. {
  235. generator.emit<Bytecode::Op::GetVariable>(generator.intern_identifier(m_string));
  236. return {};
  237. }
  238. Bytecode::CodeGenerationErrorOr<void> AssignmentExpression::generate_bytecode(Bytecode::Generator& generator) const
  239. {
  240. // FIXME: Implement this for BindingPatterns too.
  241. auto& lhs = m_lhs.get<NonnullRefPtr<Expression>>();
  242. if (m_op == AssignmentOp::Assignment) {
  243. TRY(m_rhs->generate_bytecode(generator));
  244. return generator.emit_store_to_reference(lhs);
  245. }
  246. TRY(generator.emit_load_from_reference(lhs));
  247. Bytecode::BasicBlock* rhs_block_ptr { nullptr };
  248. Bytecode::BasicBlock* end_block_ptr { nullptr };
  249. // Logical assignments short circuit.
  250. if (m_op == AssignmentOp::AndAssignment) { // &&=
  251. rhs_block_ptr = &generator.make_block();
  252. end_block_ptr = &generator.make_block();
  253. generator.emit<Bytecode::Op::JumpConditional>().set_targets(
  254. Bytecode::Label { *rhs_block_ptr },
  255. Bytecode::Label { *end_block_ptr });
  256. } else if (m_op == AssignmentOp::OrAssignment) { // ||=
  257. rhs_block_ptr = &generator.make_block();
  258. end_block_ptr = &generator.make_block();
  259. generator.emit<Bytecode::Op::JumpConditional>().set_targets(
  260. Bytecode::Label { *end_block_ptr },
  261. Bytecode::Label { *rhs_block_ptr });
  262. } else if (m_op == AssignmentOp::NullishAssignment) { // ??=
  263. rhs_block_ptr = &generator.make_block();
  264. end_block_ptr = &generator.make_block();
  265. generator.emit<Bytecode::Op::JumpNullish>().set_targets(
  266. Bytecode::Label { *rhs_block_ptr },
  267. Bytecode::Label { *end_block_ptr });
  268. }
  269. if (rhs_block_ptr)
  270. generator.switch_to_basic_block(*rhs_block_ptr);
  271. // lhs_reg is a part of the rhs_block because the store isn't necessary
  272. // if the logical assignment condition fails.
  273. auto lhs_reg = generator.allocate_register();
  274. generator.emit<Bytecode::Op::Store>(lhs_reg);
  275. TRY(m_rhs->generate_bytecode(generator));
  276. switch (m_op) {
  277. case AssignmentOp::AdditionAssignment:
  278. generator.emit<Bytecode::Op::Add>(lhs_reg);
  279. break;
  280. case AssignmentOp::SubtractionAssignment:
  281. generator.emit<Bytecode::Op::Sub>(lhs_reg);
  282. break;
  283. case AssignmentOp::MultiplicationAssignment:
  284. generator.emit<Bytecode::Op::Mul>(lhs_reg);
  285. break;
  286. case AssignmentOp::DivisionAssignment:
  287. generator.emit<Bytecode::Op::Div>(lhs_reg);
  288. break;
  289. case AssignmentOp::ModuloAssignment:
  290. generator.emit<Bytecode::Op::Mod>(lhs_reg);
  291. break;
  292. case AssignmentOp::ExponentiationAssignment:
  293. generator.emit<Bytecode::Op::Exp>(lhs_reg);
  294. break;
  295. case AssignmentOp::BitwiseAndAssignment:
  296. generator.emit<Bytecode::Op::BitwiseAnd>(lhs_reg);
  297. break;
  298. case AssignmentOp::BitwiseOrAssignment:
  299. generator.emit<Bytecode::Op::BitwiseOr>(lhs_reg);
  300. break;
  301. case AssignmentOp::BitwiseXorAssignment:
  302. generator.emit<Bytecode::Op::BitwiseXor>(lhs_reg);
  303. break;
  304. case AssignmentOp::LeftShiftAssignment:
  305. generator.emit<Bytecode::Op::LeftShift>(lhs_reg);
  306. break;
  307. case AssignmentOp::RightShiftAssignment:
  308. generator.emit<Bytecode::Op::RightShift>(lhs_reg);
  309. break;
  310. case AssignmentOp::UnsignedRightShiftAssignment:
  311. generator.emit<Bytecode::Op::UnsignedRightShift>(lhs_reg);
  312. break;
  313. case AssignmentOp::AndAssignment:
  314. case AssignmentOp::OrAssignment:
  315. case AssignmentOp::NullishAssignment:
  316. break; // These are handled above.
  317. default:
  318. return Bytecode::CodeGenerationError {
  319. this,
  320. "Unimplemented operation"sv,
  321. };
  322. }
  323. TRY(generator.emit_store_to_reference(lhs));
  324. if (end_block_ptr) {
  325. generator.emit<Bytecode::Op::Jump>().set_targets(
  326. Bytecode::Label { *end_block_ptr },
  327. {});
  328. generator.switch_to_basic_block(*end_block_ptr);
  329. }
  330. return {};
  331. }
  332. Bytecode::CodeGenerationErrorOr<void> WhileStatement::generate_bytecode(Bytecode::Generator& generator) const
  333. {
  334. // test
  335. // jump if_false (true) end (false) body
  336. // body
  337. // jump always (true) test
  338. // end
  339. auto& test_block = generator.make_block();
  340. auto& body_block = generator.make_block();
  341. auto& end_block = generator.make_block();
  342. // Init result register
  343. generator.emit<Bytecode::Op::LoadImmediate>(js_undefined());
  344. auto result_reg = generator.allocate_register();
  345. generator.emit<Bytecode::Op::Store>(result_reg);
  346. // jump to the test block
  347. generator.emit<Bytecode::Op::Jump>().set_targets(
  348. Bytecode::Label { test_block },
  349. {});
  350. generator.switch_to_basic_block(test_block);
  351. TRY(m_test->generate_bytecode(generator));
  352. generator.emit<Bytecode::Op::JumpConditional>().set_targets(
  353. Bytecode::Label { body_block },
  354. Bytecode::Label { end_block });
  355. generator.switch_to_basic_block(body_block);
  356. generator.begin_continuable_scope(Bytecode::Label { test_block });
  357. generator.begin_breakable_scope(Bytecode::Label { end_block });
  358. TRY(m_body->generate_bytecode(generator));
  359. if (!generator.is_current_block_terminated()) {
  360. generator.emit<Bytecode::Op::Jump>().set_targets(
  361. Bytecode::Label { test_block },
  362. {});
  363. generator.end_continuable_scope();
  364. generator.end_breakable_scope();
  365. generator.switch_to_basic_block(end_block);
  366. generator.emit<Bytecode::Op::Load>(result_reg);
  367. }
  368. return {};
  369. }
  370. Bytecode::CodeGenerationErrorOr<void> DoWhileStatement::generate_bytecode(Bytecode::Generator& generator) const
  371. {
  372. // jump always (true) body
  373. // test
  374. // jump if_false (true) end (false) body
  375. // body
  376. // jump always (true) test
  377. // end
  378. auto& test_block = generator.make_block();
  379. auto& body_block = generator.make_block();
  380. auto& end_block = generator.make_block();
  381. // Init result register
  382. generator.emit<Bytecode::Op::LoadImmediate>(js_undefined());
  383. auto result_reg = generator.allocate_register();
  384. generator.emit<Bytecode::Op::Store>(result_reg);
  385. // jump to the body block
  386. generator.emit<Bytecode::Op::Jump>().set_targets(
  387. Bytecode::Label { body_block },
  388. {});
  389. generator.switch_to_basic_block(test_block);
  390. TRY(m_test->generate_bytecode(generator));
  391. generator.emit<Bytecode::Op::JumpConditional>().set_targets(
  392. Bytecode::Label { body_block },
  393. Bytecode::Label { end_block });
  394. generator.switch_to_basic_block(body_block);
  395. generator.begin_continuable_scope(Bytecode::Label { test_block });
  396. generator.begin_breakable_scope(Bytecode::Label { end_block });
  397. TRY(m_body->generate_bytecode(generator));
  398. if (!generator.is_current_block_terminated()) {
  399. generator.emit<Bytecode::Op::Jump>().set_targets(
  400. Bytecode::Label { test_block },
  401. {});
  402. generator.end_continuable_scope();
  403. generator.end_breakable_scope();
  404. generator.switch_to_basic_block(end_block);
  405. generator.emit<Bytecode::Op::Load>(result_reg);
  406. }
  407. return {};
  408. }
  409. Bytecode::CodeGenerationErrorOr<void> ForStatement::generate_bytecode(Bytecode::Generator& generator) const
  410. {
  411. // init
  412. // jump always (true) test
  413. // test
  414. // jump if_true (true) body (false) end
  415. // body
  416. // jump always (true) update
  417. // update
  418. // jump always (true) test
  419. // end
  420. // If 'test' is missing, fuse the 'test' and 'body' basic blocks
  421. // If 'update' is missing, fuse the 'body' and 'update' basic blocks
  422. Bytecode::BasicBlock* test_block_ptr { nullptr };
  423. Bytecode::BasicBlock* body_block_ptr { nullptr };
  424. Bytecode::BasicBlock* update_block_ptr { nullptr };
  425. auto& end_block = generator.make_block();
  426. if (m_init)
  427. TRY(m_init->generate_bytecode(generator));
  428. body_block_ptr = &generator.make_block();
  429. if (m_test)
  430. test_block_ptr = &generator.make_block();
  431. else
  432. test_block_ptr = body_block_ptr;
  433. if (m_update)
  434. update_block_ptr = &generator.make_block();
  435. else
  436. update_block_ptr = body_block_ptr;
  437. generator.emit<Bytecode::Op::LoadImmediate>(js_undefined());
  438. auto result_reg = generator.allocate_register();
  439. generator.emit<Bytecode::Op::Store>(result_reg);
  440. generator.emit<Bytecode::Op::Jump>().set_targets(
  441. Bytecode::Label { *test_block_ptr },
  442. {});
  443. if (m_test) {
  444. generator.switch_to_basic_block(*test_block_ptr);
  445. TRY(m_test->generate_bytecode(generator));
  446. generator.emit<Bytecode::Op::JumpConditional>().set_targets(
  447. Bytecode::Label { *body_block_ptr },
  448. Bytecode::Label { end_block });
  449. }
  450. generator.switch_to_basic_block(*body_block_ptr);
  451. generator.begin_continuable_scope(Bytecode::Label { *update_block_ptr });
  452. generator.begin_breakable_scope(Bytecode::Label { end_block });
  453. TRY(m_body->generate_bytecode(generator));
  454. generator.end_continuable_scope();
  455. if (!generator.is_current_block_terminated()) {
  456. if (m_update) {
  457. generator.emit<Bytecode::Op::Jump>().set_targets(
  458. Bytecode::Label { *update_block_ptr },
  459. {});
  460. generator.switch_to_basic_block(*update_block_ptr);
  461. TRY(m_update->generate_bytecode(generator));
  462. }
  463. generator.emit<Bytecode::Op::Jump>().set_targets(
  464. Bytecode::Label { *test_block_ptr },
  465. {});
  466. generator.end_breakable_scope();
  467. generator.switch_to_basic_block(end_block);
  468. generator.emit<Bytecode::Op::Load>(result_reg);
  469. }
  470. return {};
  471. }
  472. Bytecode::CodeGenerationErrorOr<void> ObjectExpression::generate_bytecode(Bytecode::Generator& generator) const
  473. {
  474. generator.emit<Bytecode::Op::NewObject>();
  475. if (m_properties.is_empty())
  476. return {};
  477. auto object_reg = generator.allocate_register();
  478. generator.emit<Bytecode::Op::Store>(object_reg);
  479. for (auto& property : m_properties) {
  480. if (property.type() != ObjectProperty::Type::KeyValue)
  481. return Bytecode::CodeGenerationError {
  482. this,
  483. "Unimplemented property kind"sv,
  484. };
  485. if (is<StringLiteral>(property.key())) {
  486. auto& string_literal = static_cast<StringLiteral const&>(property.key());
  487. Bytecode::IdentifierTableIndex key_name = generator.intern_identifier(string_literal.value());
  488. TRY(property.value().generate_bytecode(generator));
  489. generator.emit<Bytecode::Op::PutById>(object_reg, key_name);
  490. } else {
  491. TRY(property.key().generate_bytecode(generator));
  492. auto property_reg = generator.allocate_register();
  493. generator.emit<Bytecode::Op::Store>(property_reg);
  494. TRY(property.value().generate_bytecode(generator));
  495. generator.emit<Bytecode::Op::PutByValue>(object_reg, property_reg);
  496. }
  497. }
  498. generator.emit<Bytecode::Op::Load>(object_reg);
  499. return {};
  500. }
  501. Bytecode::CodeGenerationErrorOr<void> ArrayExpression::generate_bytecode(Bytecode::Generator& generator) const
  502. {
  503. Vector<Bytecode::Register> element_regs;
  504. for (auto& element : m_elements) {
  505. if (element) {
  506. TRY(element->generate_bytecode(generator));
  507. if (is<SpreadExpression>(*element)) {
  508. return Bytecode::CodeGenerationError {
  509. this,
  510. "Unimplemented element kind: SpreadExpression"sv,
  511. };
  512. }
  513. } else {
  514. generator.emit<Bytecode::Op::LoadImmediate>(Value {});
  515. }
  516. auto element_reg = generator.allocate_register();
  517. generator.emit<Bytecode::Op::Store>(element_reg);
  518. element_regs.append(element_reg);
  519. }
  520. generator.emit_with_extra_register_slots<Bytecode::Op::NewArray>(element_regs.size(), element_regs);
  521. return {};
  522. }
  523. Bytecode::CodeGenerationErrorOr<void> MemberExpression::generate_bytecode(Bytecode::Generator& generator) const
  524. {
  525. return generator.emit_load_from_reference(*this);
  526. }
  527. Bytecode::CodeGenerationErrorOr<void> FunctionDeclaration::generate_bytecode(Bytecode::Generator&) const
  528. {
  529. return {};
  530. }
  531. Bytecode::CodeGenerationErrorOr<void> FunctionExpression::generate_bytecode(Bytecode::Generator& generator) const
  532. {
  533. generator.emit<Bytecode::Op::NewFunction>(*this);
  534. return {};
  535. }
  536. static Bytecode::CodeGenerationErrorOr<void> generate_binding_pattern_bytecode(Bytecode::Generator& generator, BindingPattern const& pattern, Bytecode::Op::SetVariable::InitializationMode, Bytecode::Register const& value_reg);
  537. static Bytecode::CodeGenerationErrorOr<void> generate_object_binding_pattern_bytecode(Bytecode::Generator& generator, BindingPattern const& pattern, Bytecode::Op::SetVariable::InitializationMode initialization_mode, Bytecode::Register const& value_reg)
  538. {
  539. Vector<Bytecode::Register> excluded_property_names;
  540. auto has_rest = false;
  541. if (pattern.entries.size() > 0)
  542. has_rest = pattern.entries[pattern.entries.size() - 1].is_rest;
  543. for (auto& [name, alias, initializer, is_rest] : pattern.entries) {
  544. if (is_rest) {
  545. VERIFY(name.has<NonnullRefPtr<Identifier>>());
  546. VERIFY(alias.has<Empty>());
  547. VERIFY(!initializer);
  548. auto identifier = name.get<NonnullRefPtr<Identifier>>()->string();
  549. auto interned_identifier = generator.intern_identifier(identifier);
  550. generator.emit_with_extra_register_slots<Bytecode::Op::CopyObjectExcludingProperties>(excluded_property_names.size(), value_reg, excluded_property_names);
  551. generator.emit<Bytecode::Op::SetVariable>(interned_identifier, initialization_mode);
  552. return {};
  553. }
  554. Bytecode::StringTableIndex name_index;
  555. if (name.has<NonnullRefPtr<Identifier>>()) {
  556. auto identifier = name.get<NonnullRefPtr<Identifier>>()->string();
  557. name_index = generator.intern_string(identifier);
  558. if (has_rest) {
  559. auto excluded_name_reg = generator.allocate_register();
  560. excluded_property_names.append(excluded_name_reg);
  561. generator.emit<Bytecode::Op::NewString>(name_index);
  562. generator.emit<Bytecode::Op::Store>(excluded_name_reg);
  563. }
  564. generator.emit<Bytecode::Op::Load>(value_reg);
  565. generator.emit<Bytecode::Op::GetById>(generator.intern_identifier(identifier));
  566. } else {
  567. auto expression = name.get<NonnullRefPtr<Expression>>();
  568. TRY(expression->generate_bytecode(generator));
  569. if (has_rest) {
  570. auto excluded_name_reg = generator.allocate_register();
  571. excluded_property_names.append(excluded_name_reg);
  572. generator.emit<Bytecode::Op::Store>(excluded_name_reg);
  573. }
  574. generator.emit<Bytecode::Op::GetByValue>(value_reg);
  575. }
  576. if (initializer) {
  577. auto& if_undefined_block = generator.make_block();
  578. auto& if_not_undefined_block = generator.make_block();
  579. generator.emit<Bytecode::Op::JumpUndefined>().set_targets(
  580. Bytecode::Label { if_undefined_block },
  581. Bytecode::Label { if_not_undefined_block });
  582. generator.switch_to_basic_block(if_undefined_block);
  583. TRY(initializer->generate_bytecode(generator));
  584. generator.emit<Bytecode::Op::Jump>().set_targets(
  585. Bytecode::Label { if_not_undefined_block },
  586. {});
  587. generator.switch_to_basic_block(if_not_undefined_block);
  588. }
  589. if (alias.has<NonnullRefPtr<BindingPattern>>()) {
  590. auto& binding_pattern = *alias.get<NonnullRefPtr<BindingPattern>>();
  591. auto nested_value_reg = generator.allocate_register();
  592. generator.emit<Bytecode::Op::Store>(nested_value_reg);
  593. TRY(generate_binding_pattern_bytecode(generator, binding_pattern, initialization_mode, nested_value_reg));
  594. } else if (alias.has<Empty>()) {
  595. if (name.has<NonnullRefPtr<Expression>>()) {
  596. // This needs some sort of SetVariableByValue opcode, as it's a runtime binding
  597. return Bytecode::CodeGenerationError {
  598. name.get<NonnullRefPtr<Expression>>().ptr(),
  599. "Unimplemented name/alias pair: Empty/Expression"sv,
  600. };
  601. }
  602. auto& identifier = alias.get<NonnullRefPtr<Identifier>>()->string();
  603. generator.emit<Bytecode::Op::SetVariable>(generator.intern_identifier(identifier), initialization_mode);
  604. } else {
  605. auto& identifier = alias.get<NonnullRefPtr<Identifier>>()->string();
  606. generator.emit<Bytecode::Op::SetVariable>(generator.intern_identifier(identifier), initialization_mode);
  607. }
  608. }
  609. return {};
  610. }
  611. static Bytecode::CodeGenerationErrorOr<void> generate_array_binding_pattern_bytecode(Bytecode::Generator& generator, BindingPattern const& pattern, Bytecode::Op::SetVariable::InitializationMode initialization_mode, Bytecode::Register const& value_reg)
  612. {
  613. /*
  614. * Consider the following destructuring assignment:
  615. *
  616. * let [a, b, c, d, e] = o;
  617. *
  618. * It would be fairly trivial to just loop through this iterator, getting the value
  619. * at each step and assigning them to the binding sequentially. However, this is not
  620. * correct: once an iterator is exhausted, it must not be called again. This complicates
  621. * the bytecode. In order to accomplish this, we do the following:
  622. *
  623. * - Reserve a special boolean register which holds 'true' if the iterator is exhausted,
  624. * and false otherwise
  625. * - When we are retrieving the value which should be bound, we first check this register.
  626. * If it is 'true', we load undefined into the accumulator. Otherwise, we grab the next
  627. * value from the iterator and store it into the accumulator.
  628. *
  629. * Note that the is_exhausted register does not need to be loaded with false because the
  630. * first IteratorNext bytecode is _not_ proceeded by an exhausted check, as it is
  631. * unnecessary.
  632. */
  633. auto is_iterator_exhausted_register = generator.allocate_register();
  634. auto iterator_reg = generator.allocate_register();
  635. generator.emit<Bytecode::Op::Load>(value_reg);
  636. generator.emit<Bytecode::Op::GetIterator>();
  637. generator.emit<Bytecode::Op::Store>(iterator_reg);
  638. bool first = true;
  639. auto temp_iterator_result_reg = generator.allocate_register();
  640. auto assign_accumulator_to_alias = [&](auto& alias) {
  641. return alias.visit(
  642. [&](Empty) -> Bytecode::CodeGenerationErrorOr<void> {
  643. // This element is an elision
  644. return {};
  645. },
  646. [&](NonnullRefPtr<Identifier> const& identifier) -> Bytecode::CodeGenerationErrorOr<void> {
  647. auto interned_index = generator.intern_identifier(identifier->string());
  648. generator.emit<Bytecode::Op::SetVariable>(interned_index, initialization_mode);
  649. return {};
  650. },
  651. [&](NonnullRefPtr<BindingPattern> const& pattern) -> Bytecode::CodeGenerationErrorOr<void> {
  652. // Store the accumulator value in a permanent register
  653. auto target_reg = generator.allocate_register();
  654. generator.emit<Bytecode::Op::Store>(target_reg);
  655. return generate_binding_pattern_bytecode(generator, pattern, initialization_mode, target_reg);
  656. },
  657. [&](NonnullRefPtr<MemberExpression> const& expr) -> Bytecode::CodeGenerationErrorOr<void> {
  658. return Bytecode::CodeGenerationError {
  659. expr.ptr(),
  660. "Unimplemented alias mode: MemberExpression"sv,
  661. };
  662. });
  663. };
  664. for (auto& [name, alias, initializer, is_rest] : pattern.entries) {
  665. VERIFY(name.has<Empty>());
  666. if (is_rest) {
  667. if (first) {
  668. // The iterator has not been called, and is thus known to be not exhausted
  669. generator.emit<Bytecode::Op::Load>(iterator_reg);
  670. generator.emit<Bytecode::Op::IteratorToArray>();
  671. } else {
  672. auto& if_exhausted_block = generator.make_block();
  673. auto& if_not_exhausted_block = generator.make_block();
  674. auto& continuation_block = generator.make_block();
  675. generator.emit<Bytecode::Op::Load>(is_iterator_exhausted_register);
  676. generator.emit<Bytecode::Op::JumpConditional>().set_targets(
  677. Bytecode::Label { if_exhausted_block },
  678. Bytecode::Label { if_not_exhausted_block });
  679. generator.switch_to_basic_block(if_exhausted_block);
  680. generator.emit<Bytecode::Op::NewArray>();
  681. generator.emit<Bytecode::Op::Jump>().set_targets(
  682. Bytecode::Label { continuation_block },
  683. {});
  684. generator.switch_to_basic_block(if_not_exhausted_block);
  685. generator.emit<Bytecode::Op::Load>(iterator_reg);
  686. generator.emit<Bytecode::Op::IteratorToArray>();
  687. generator.emit<Bytecode::Op::Jump>().set_targets(
  688. Bytecode::Label { continuation_block },
  689. {});
  690. generator.switch_to_basic_block(continuation_block);
  691. }
  692. return assign_accumulator_to_alias(alias);
  693. }
  694. // In the first iteration of the loop, a few things are true which can save
  695. // us some bytecode:
  696. // - the iterator result is still in the accumulator, so we can avoid a load
  697. // - the iterator is not yet exhausted, which can save us a jump and some
  698. // creation
  699. auto& iterator_is_exhausted_block = generator.make_block();
  700. if (!first) {
  701. auto& iterator_is_not_exhausted_block = generator.make_block();
  702. generator.emit<Bytecode::Op::Load>(is_iterator_exhausted_register);
  703. generator.emit<Bytecode::Op::JumpConditional>().set_targets(
  704. Bytecode::Label { iterator_is_exhausted_block },
  705. Bytecode::Label { iterator_is_not_exhausted_block });
  706. generator.switch_to_basic_block(iterator_is_not_exhausted_block);
  707. generator.emit<Bytecode::Op::Load>(iterator_reg);
  708. }
  709. generator.emit<Bytecode::Op::IteratorNext>();
  710. generator.emit<Bytecode::Op::Store>(temp_iterator_result_reg);
  711. generator.emit<Bytecode::Op::IteratorResultDone>();
  712. generator.emit<Bytecode::Op::Store>(is_iterator_exhausted_register);
  713. // We still have to check for exhaustion here. If the iterator is exhausted,
  714. // we need to bail before trying to get the value
  715. auto& no_bail_block = generator.make_block();
  716. generator.emit<Bytecode::Op::JumpConditional>().set_targets(
  717. Bytecode::Label { iterator_is_exhausted_block },
  718. Bytecode::Label { no_bail_block });
  719. generator.switch_to_basic_block(no_bail_block);
  720. // Get the next value in the iterator
  721. generator.emit<Bytecode::Op::Load>(temp_iterator_result_reg);
  722. generator.emit<Bytecode::Op::IteratorResultValue>();
  723. auto& create_binding_block = generator.make_block();
  724. generator.emit<Bytecode::Op::Jump>().set_targets(
  725. Bytecode::Label { create_binding_block },
  726. {});
  727. // The iterator is exhausted, so we just load undefined and continue binding
  728. generator.switch_to_basic_block(iterator_is_exhausted_block);
  729. generator.emit<Bytecode::Op::LoadImmediate>(js_undefined());
  730. generator.emit<Bytecode::Op::Jump>().set_targets(
  731. Bytecode::Label { create_binding_block },
  732. {});
  733. // Create the actual binding. The value which this entry must bind is now in the
  734. // accumulator. We can proceed, processing the alias as a nested destructuring
  735. // pattern if necessary.
  736. generator.switch_to_basic_block(create_binding_block);
  737. TRY(assign_accumulator_to_alias(alias));
  738. first = false;
  739. }
  740. return {};
  741. }
  742. static Bytecode::CodeGenerationErrorOr<void> generate_binding_pattern_bytecode(Bytecode::Generator& generator, BindingPattern const& pattern, Bytecode::Op::SetVariable::InitializationMode initialization_mode, Bytecode::Register const& value_reg)
  743. {
  744. if (pattern.kind == BindingPattern::Kind::Object)
  745. return generate_object_binding_pattern_bytecode(generator, pattern, initialization_mode, value_reg);
  746. return generate_array_binding_pattern_bytecode(generator, pattern, initialization_mode, value_reg);
  747. }
  748. Bytecode::CodeGenerationErrorOr<void> VariableDeclaration::generate_bytecode(Bytecode::Generator& generator) const
  749. {
  750. for (auto& declarator : m_declarations) {
  751. if (declarator.init())
  752. TRY(declarator.init()->generate_bytecode(generator));
  753. else
  754. generator.emit<Bytecode::Op::LoadImmediate>(js_undefined());
  755. auto initialization_mode = is_lexical_declaration() ? Bytecode::Op::SetVariable::InitializationMode::Initialize : Bytecode::Op::SetVariable::InitializationMode::Set;
  756. TRY(declarator.target().visit(
  757. [&](NonnullRefPtr<Identifier> const& id) -> Bytecode::CodeGenerationErrorOr<void> {
  758. generator.emit<Bytecode::Op::SetVariable>(generator.intern_identifier(id->string()), initialization_mode);
  759. return {};
  760. },
  761. [&](NonnullRefPtr<BindingPattern> const& pattern) -> Bytecode::CodeGenerationErrorOr<void> {
  762. auto value_register = generator.allocate_register();
  763. generator.emit<Bytecode::Op::Store>(value_register);
  764. return generate_binding_pattern_bytecode(generator, pattern, initialization_mode, value_register);
  765. }));
  766. }
  767. return {};
  768. }
  769. Bytecode::CodeGenerationErrorOr<void> CallExpression::generate_bytecode(Bytecode::Generator& generator) const
  770. {
  771. auto callee_reg = generator.allocate_register();
  772. auto this_reg = generator.allocate_register();
  773. generator.emit<Bytecode::Op::LoadImmediate>(js_undefined());
  774. generator.emit<Bytecode::Op::Store>(this_reg);
  775. if (is<NewExpression>(this)) {
  776. TRY(m_callee->generate_bytecode(generator));
  777. generator.emit<Bytecode::Op::Store>(callee_reg);
  778. } else if (is<SuperExpression>(*m_callee)) {
  779. return Bytecode::CodeGenerationError {
  780. this,
  781. "Unimplemented callee kind: SuperExpression"sv,
  782. };
  783. } else if (is<MemberExpression>(*m_callee)) {
  784. auto& member_expression = static_cast<const MemberExpression&>(*m_callee);
  785. if (is<SuperExpression>(member_expression.object())) {
  786. return Bytecode::CodeGenerationError {
  787. this,
  788. "Unimplemented callee kind: MemberExpression on SuperExpression"sv,
  789. };
  790. }
  791. TRY(member_expression.object().generate_bytecode(generator));
  792. generator.emit<Bytecode::Op::Store>(this_reg);
  793. if (member_expression.is_computed()) {
  794. TRY(member_expression.property().generate_bytecode(generator));
  795. generator.emit<Bytecode::Op::GetByValue>(this_reg);
  796. } else {
  797. auto identifier_table_ref = generator.intern_identifier(verify_cast<Identifier>(member_expression.property()).string());
  798. generator.emit<Bytecode::Op::GetById>(identifier_table_ref);
  799. }
  800. generator.emit<Bytecode::Op::Store>(callee_reg);
  801. } else {
  802. // FIXME: this = global object in sloppy mode.
  803. TRY(m_callee->generate_bytecode(generator));
  804. generator.emit<Bytecode::Op::Store>(callee_reg);
  805. }
  806. Vector<Bytecode::Register> argument_registers;
  807. for (auto& arg : m_arguments) {
  808. TRY(arg.value->generate_bytecode(generator));
  809. auto arg_reg = generator.allocate_register();
  810. generator.emit<Bytecode::Op::Store>(arg_reg);
  811. argument_registers.append(arg_reg);
  812. }
  813. Bytecode::Op::Call::CallType call_type;
  814. if (is<NewExpression>(*this)) {
  815. call_type = Bytecode::Op::Call::CallType::Construct;
  816. } else {
  817. call_type = Bytecode::Op::Call::CallType::Call;
  818. }
  819. generator.emit_with_extra_register_slots<Bytecode::Op::Call>(argument_registers.size(), call_type, callee_reg, this_reg, argument_registers);
  820. return {};
  821. }
  822. Bytecode::CodeGenerationErrorOr<void> ReturnStatement::generate_bytecode(Bytecode::Generator& generator) const
  823. {
  824. if (m_argument)
  825. TRY(m_argument->generate_bytecode(generator));
  826. if (generator.is_in_generator_or_async_function())
  827. generator.emit<Bytecode::Op::Yield>(nullptr);
  828. else
  829. generator.emit<Bytecode::Op::Return>();
  830. return {};
  831. }
  832. Bytecode::CodeGenerationErrorOr<void> YieldExpression::generate_bytecode(Bytecode::Generator& generator) const
  833. {
  834. VERIFY(generator.is_in_generator_function());
  835. if (m_is_yield_from) {
  836. return Bytecode::CodeGenerationError {
  837. this,
  838. "Unimplemented form: `yield*`"sv,
  839. };
  840. }
  841. if (m_argument)
  842. TRY(m_argument->generate_bytecode(generator));
  843. auto& continuation_block = generator.make_block();
  844. generator.emit<Bytecode::Op::Yield>(Bytecode::Label { continuation_block });
  845. generator.switch_to_basic_block(continuation_block);
  846. return {};
  847. }
  848. Bytecode::CodeGenerationErrorOr<void> IfStatement::generate_bytecode(Bytecode::Generator& generator) const
  849. {
  850. // test
  851. // jump if_true (true) true (false) false
  852. // true
  853. // jump always (true) end
  854. // false
  855. // jump always (true) end
  856. // end
  857. auto& true_block = generator.make_block();
  858. auto& false_block = generator.make_block();
  859. TRY(m_predicate->generate_bytecode(generator));
  860. generator.emit<Bytecode::Op::JumpConditional>().set_targets(
  861. Bytecode::Label { true_block },
  862. Bytecode::Label { false_block });
  863. Bytecode::Op::Jump* true_block_jump { nullptr };
  864. generator.switch_to_basic_block(true_block);
  865. generator.emit<Bytecode::Op::LoadImmediate>(js_undefined());
  866. TRY(m_consequent->generate_bytecode(generator));
  867. if (!generator.is_current_block_terminated())
  868. true_block_jump = &generator.emit<Bytecode::Op::Jump>();
  869. generator.switch_to_basic_block(false_block);
  870. auto& end_block = generator.make_block();
  871. generator.emit<Bytecode::Op::LoadImmediate>(js_undefined());
  872. if (m_alternate)
  873. TRY(m_alternate->generate_bytecode(generator));
  874. if (!generator.is_current_block_terminated())
  875. generator.emit<Bytecode::Op::Jump>().set_targets(Bytecode::Label { end_block }, {});
  876. if (true_block_jump)
  877. true_block_jump->set_targets(Bytecode::Label { end_block }, {});
  878. generator.switch_to_basic_block(end_block);
  879. return {};
  880. }
  881. Bytecode::CodeGenerationErrorOr<void> ContinueStatement::generate_bytecode(Bytecode::Generator& generator) const
  882. {
  883. generator.emit<Bytecode::Op::Jump>().set_targets(
  884. generator.nearest_continuable_scope(),
  885. {});
  886. return {};
  887. }
  888. Bytecode::CodeGenerationErrorOr<void> DebuggerStatement::generate_bytecode(Bytecode::Generator&) const
  889. {
  890. return {};
  891. }
  892. Bytecode::CodeGenerationErrorOr<void> ConditionalExpression::generate_bytecode(Bytecode::Generator& generator) const
  893. {
  894. // test
  895. // jump if_true (true) true (false) false
  896. // true
  897. // jump always (true) end
  898. // false
  899. // jump always (true) end
  900. // end
  901. auto& true_block = generator.make_block();
  902. auto& false_block = generator.make_block();
  903. auto& end_block = generator.make_block();
  904. TRY(m_test->generate_bytecode(generator));
  905. generator.emit<Bytecode::Op::JumpConditional>().set_targets(
  906. Bytecode::Label { true_block },
  907. Bytecode::Label { false_block });
  908. generator.switch_to_basic_block(true_block);
  909. TRY(m_consequent->generate_bytecode(generator));
  910. generator.emit<Bytecode::Op::Jump>().set_targets(
  911. Bytecode::Label { end_block },
  912. {});
  913. generator.switch_to_basic_block(false_block);
  914. TRY(m_alternate->generate_bytecode(generator));
  915. generator.emit<Bytecode::Op::Jump>().set_targets(
  916. Bytecode::Label { end_block },
  917. {});
  918. generator.switch_to_basic_block(end_block);
  919. return {};
  920. }
  921. Bytecode::CodeGenerationErrorOr<void> SequenceExpression::generate_bytecode(Bytecode::Generator& generator) const
  922. {
  923. for (auto& expression : m_expressions)
  924. TRY(expression.generate_bytecode(generator));
  925. return {};
  926. }
  927. Bytecode::CodeGenerationErrorOr<void> TemplateLiteral::generate_bytecode(Bytecode::Generator& generator) const
  928. {
  929. auto string_reg = generator.allocate_register();
  930. for (size_t i = 0; i < m_expressions.size(); i++) {
  931. TRY(m_expressions[i].generate_bytecode(generator));
  932. if (i == 0) {
  933. generator.emit<Bytecode::Op::Store>(string_reg);
  934. } else {
  935. generator.emit<Bytecode::Op::ConcatString>(string_reg);
  936. }
  937. }
  938. generator.emit<Bytecode::Op::Load>(string_reg);
  939. return {};
  940. }
  941. Bytecode::CodeGenerationErrorOr<void> TaggedTemplateLiteral::generate_bytecode(Bytecode::Generator& generator) const
  942. {
  943. TRY(m_tag->generate_bytecode(generator));
  944. auto tag_reg = generator.allocate_register();
  945. generator.emit<Bytecode::Op::Store>(tag_reg);
  946. Vector<Bytecode::Register> string_regs;
  947. auto& expressions = m_template_literal->expressions();
  948. for (size_t i = 0; i < expressions.size(); ++i) {
  949. if (i % 2 != 0)
  950. continue;
  951. TRY(expressions[i].generate_bytecode(generator));
  952. auto string_reg = generator.allocate_register();
  953. generator.emit<Bytecode::Op::Store>(string_reg);
  954. string_regs.append(string_reg);
  955. }
  956. generator.emit_with_extra_register_slots<Bytecode::Op::NewArray>(string_regs.size(), string_regs);
  957. auto strings_reg = generator.allocate_register();
  958. generator.emit<Bytecode::Op::Store>(strings_reg);
  959. Vector<Bytecode::Register> argument_regs;
  960. argument_regs.append(strings_reg);
  961. for (size_t i = 0; i < expressions.size(); ++i) {
  962. if (i % 2 == 0)
  963. continue;
  964. TRY(expressions[i].generate_bytecode(generator));
  965. auto string_reg = generator.allocate_register();
  966. generator.emit<Bytecode::Op::Store>(string_reg);
  967. argument_regs.append(string_reg);
  968. }
  969. Vector<Bytecode::Register> raw_string_regs;
  970. for (auto& raw_string : m_template_literal->raw_strings()) {
  971. TRY(raw_string.generate_bytecode(generator));
  972. auto raw_string_reg = generator.allocate_register();
  973. generator.emit<Bytecode::Op::Store>(raw_string_reg);
  974. raw_string_regs.append(raw_string_reg);
  975. }
  976. generator.emit_with_extra_register_slots<Bytecode::Op::NewArray>(raw_string_regs.size(), raw_string_regs);
  977. auto raw_strings_reg = generator.allocate_register();
  978. generator.emit<Bytecode::Op::Store>(raw_strings_reg);
  979. generator.emit<Bytecode::Op::Load>(strings_reg);
  980. generator.emit<Bytecode::Op::PutById>(raw_strings_reg, generator.intern_identifier("raw"));
  981. generator.emit<Bytecode::Op::LoadImmediate>(js_undefined());
  982. auto this_reg = generator.allocate_register();
  983. generator.emit<Bytecode::Op::Store>(this_reg);
  984. generator.emit_with_extra_register_slots<Bytecode::Op::Call>(argument_regs.size(), Bytecode::Op::Call::CallType::Call, tag_reg, this_reg, move(argument_regs));
  985. return {};
  986. }
  987. Bytecode::CodeGenerationErrorOr<void> UpdateExpression::generate_bytecode(Bytecode::Generator& generator) const
  988. {
  989. TRY(generator.emit_load_from_reference(*m_argument));
  990. Optional<Bytecode::Register> previous_value_for_postfix_reg;
  991. if (!m_prefixed) {
  992. previous_value_for_postfix_reg = generator.allocate_register();
  993. generator.emit<Bytecode::Op::Store>(*previous_value_for_postfix_reg);
  994. }
  995. if (m_op == UpdateOp::Increment)
  996. generator.emit<Bytecode::Op::Increment>();
  997. else
  998. generator.emit<Bytecode::Op::Decrement>();
  999. TRY(generator.emit_store_to_reference(*m_argument));
  1000. if (!m_prefixed)
  1001. generator.emit<Bytecode::Op::Load>(*previous_value_for_postfix_reg);
  1002. return {};
  1003. }
  1004. Bytecode::CodeGenerationErrorOr<void> ThrowStatement::generate_bytecode(Bytecode::Generator& generator) const
  1005. {
  1006. TRY(m_argument->generate_bytecode(generator));
  1007. generator.emit<Bytecode::Op::Throw>();
  1008. return {};
  1009. }
  1010. Bytecode::CodeGenerationErrorOr<void> BreakStatement::generate_bytecode(Bytecode::Generator& generator) const
  1011. {
  1012. generator.emit<Bytecode::Op::Jump>().set_targets(
  1013. generator.nearest_breakable_scope(),
  1014. {});
  1015. return {};
  1016. }
  1017. Bytecode::CodeGenerationErrorOr<void> TryStatement::generate_bytecode(Bytecode::Generator& generator) const
  1018. {
  1019. auto& saved_block = generator.current_block();
  1020. Optional<Bytecode::Label> handler_target;
  1021. Optional<Bytecode::Label> finalizer_target;
  1022. Bytecode::BasicBlock* next_block { nullptr };
  1023. if (m_finalizer) {
  1024. auto& finalizer_block = generator.make_block();
  1025. generator.switch_to_basic_block(finalizer_block);
  1026. TRY(m_finalizer->generate_bytecode(generator));
  1027. if (!generator.is_current_block_terminated()) {
  1028. next_block = &generator.make_block();
  1029. auto next_target = Bytecode::Label { *next_block };
  1030. generator.emit<Bytecode::Op::ContinuePendingUnwind>(next_target);
  1031. }
  1032. finalizer_target = Bytecode::Label { finalizer_block };
  1033. }
  1034. if (m_handler) {
  1035. auto& handler_block = generator.make_block();
  1036. generator.switch_to_basic_block(handler_block);
  1037. TRY(m_handler->parameter().visit(
  1038. [&](FlyString const& parameter) -> Bytecode::CodeGenerationErrorOr<void> {
  1039. if (!parameter.is_empty()) {
  1040. // FIXME: We need a separate DeclarativeEnvironment here
  1041. generator.emit<Bytecode::Op::SetVariable>(generator.intern_identifier(parameter));
  1042. }
  1043. return {};
  1044. },
  1045. [&](NonnullRefPtr<BindingPattern> const&) -> Bytecode::CodeGenerationErrorOr<void> {
  1046. // FIXME: Implement this path when the above DeclarativeEnvironment issue is dealt with.
  1047. return Bytecode::CodeGenerationError {
  1048. this,
  1049. "Unimplemented catch argument: BindingPattern"sv,
  1050. };
  1051. }));
  1052. TRY(m_handler->body().generate_bytecode(generator));
  1053. handler_target = Bytecode::Label { handler_block };
  1054. if (!generator.is_current_block_terminated()) {
  1055. if (m_finalizer) {
  1056. generator.emit<Bytecode::Op::LeaveUnwindContext>();
  1057. generator.emit<Bytecode::Op::Jump>(finalizer_target);
  1058. } else {
  1059. VERIFY(!next_block);
  1060. next_block = &generator.make_block();
  1061. auto next_target = Bytecode::Label { *next_block };
  1062. generator.emit<Bytecode::Op::Jump>(next_target);
  1063. }
  1064. }
  1065. }
  1066. auto& target_block = generator.make_block();
  1067. generator.switch_to_basic_block(saved_block);
  1068. generator.emit<Bytecode::Op::EnterUnwindContext>(Bytecode::Label { target_block }, handler_target, finalizer_target);
  1069. generator.switch_to_basic_block(target_block);
  1070. TRY(m_block->generate_bytecode(generator));
  1071. if (!generator.is_current_block_terminated()) {
  1072. if (m_finalizer) {
  1073. generator.emit<Bytecode::Op::Jump>(finalizer_target);
  1074. } else {
  1075. auto& block = generator.make_block();
  1076. generator.emit<Bytecode::Op::FinishUnwind>(Bytecode::Label { block });
  1077. next_block = &block;
  1078. }
  1079. }
  1080. generator.switch_to_basic_block(next_block ? *next_block : saved_block);
  1081. return {};
  1082. }
  1083. Bytecode::CodeGenerationErrorOr<void> SwitchStatement::generate_bytecode(Bytecode::Generator& generator) const
  1084. {
  1085. auto discriminant_reg = generator.allocate_register();
  1086. TRY(m_discriminant->generate_bytecode(generator));
  1087. generator.emit<Bytecode::Op::Store>(discriminant_reg);
  1088. Vector<Bytecode::BasicBlock&> case_blocks;
  1089. Bytecode::BasicBlock* default_block { nullptr };
  1090. Bytecode::BasicBlock* next_test_block = &generator.make_block();
  1091. generator.emit<Bytecode::Op::Jump>().set_targets(Bytecode::Label { *next_test_block }, {});
  1092. for (auto& switch_case : m_cases) {
  1093. auto& case_block = generator.make_block();
  1094. if (switch_case.test()) {
  1095. generator.switch_to_basic_block(*next_test_block);
  1096. TRY(switch_case.test()->generate_bytecode(generator));
  1097. generator.emit<Bytecode::Op::StrictlyEquals>(discriminant_reg);
  1098. next_test_block = &generator.make_block();
  1099. generator.emit<Bytecode::Op::JumpConditional>().set_targets(Bytecode::Label { case_block }, Bytecode::Label { *next_test_block });
  1100. } else {
  1101. default_block = &case_block;
  1102. }
  1103. case_blocks.append(case_block);
  1104. }
  1105. generator.switch_to_basic_block(*next_test_block);
  1106. auto& end_block = generator.make_block();
  1107. if (default_block != nullptr) {
  1108. generator.emit<Bytecode::Op::Jump>().set_targets(Bytecode::Label { *default_block }, {});
  1109. } else {
  1110. generator.emit<Bytecode::Op::LoadImmediate>(js_undefined());
  1111. generator.emit<Bytecode::Op::Jump>().set_targets(Bytecode::Label { end_block }, {});
  1112. }
  1113. auto current_block = case_blocks.begin();
  1114. generator.begin_breakable_scope(Bytecode::Label { end_block });
  1115. for (auto& switch_case : m_cases) {
  1116. generator.switch_to_basic_block(*current_block);
  1117. generator.emit<Bytecode::Op::LoadImmediate>(js_undefined());
  1118. for (auto& statement : switch_case.children()) {
  1119. TRY(statement.generate_bytecode(generator));
  1120. }
  1121. if (!generator.is_current_block_terminated()) {
  1122. auto next_block = current_block;
  1123. next_block++;
  1124. if (next_block.is_end()) {
  1125. generator.emit<Bytecode::Op::Jump>().set_targets(Bytecode::Label { end_block }, {});
  1126. } else {
  1127. generator.emit<Bytecode::Op::Jump>().set_targets(Bytecode::Label { *next_block }, {});
  1128. }
  1129. }
  1130. current_block++;
  1131. }
  1132. generator.end_breakable_scope();
  1133. generator.switch_to_basic_block(end_block);
  1134. return {};
  1135. }
  1136. Bytecode::CodeGenerationErrorOr<void> ClassDeclaration::generate_bytecode(Bytecode::Generator& generator) const
  1137. {
  1138. TRY(m_class_expression->generate_bytecode(generator));
  1139. generator.emit<Bytecode::Op::SetVariable>(generator.intern_identifier(m_class_expression.ptr()->name()), Bytecode::Op::SetVariable::InitializationMode::Initialize);
  1140. return {};
  1141. }
  1142. Bytecode::CodeGenerationErrorOr<void> ClassExpression::generate_bytecode(Bytecode::Generator& generator) const
  1143. {
  1144. generator.emit<Bytecode::Op::NewClass>(*this);
  1145. return {};
  1146. }
  1147. Bytecode::CodeGenerationErrorOr<void> ThisExpression::generate_bytecode(Bytecode::Generator& generator) const
  1148. {
  1149. generator.emit<Bytecode::Op::ResolveThisBinding>();
  1150. return {};
  1151. }
  1152. Bytecode::CodeGenerationErrorOr<void> AwaitExpression::generate_bytecode(Bytecode::Generator& generator) const
  1153. {
  1154. VERIFY(generator.is_in_async_function());
  1155. // Transform `await expr` to `yield expr`
  1156. TRY(m_argument->generate_bytecode(generator));
  1157. auto& continuation_block = generator.make_block();
  1158. generator.emit<Bytecode::Op::Yield>(Bytecode::Label { continuation_block });
  1159. generator.switch_to_basic_block(continuation_block);
  1160. return {};
  1161. }
  1162. }