ASTCodegen.cpp 112 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810181118121813181418151816181718181819182018211822182318241825182618271828182918301831183218331834183518361837183818391840184118421843184418451846184718481849185018511852185318541855185618571858185918601861186218631864186518661867186818691870187118721873187418751876187718781879188018811882188318841885188618871888188918901891189218931894189518961897189818991900190119021903190419051906190719081909191019111912191319141915191619171918191919201921192219231924192519261927192819291930193119321933193419351936193719381939194019411942194319441945194619471948194919501951195219531954195519561957195819591960196119621963196419651966196719681969197019711972197319741975197619771978197919801981198219831984198519861987198819891990199119921993199419951996199719981999200020012002200320042005200620072008200920102011201220132014201520162017201820192020202120222023202420252026202720282029203020312032203320342035203620372038203920402041204220432044204520462047204820492050205120522053205420552056205720582059206020612062206320642065206620672068206920702071207220732074207520762077207820792080208120822083208420852086208720882089209020912092209320942095209620972098209921002101210221032104210521062107210821092110211121122113211421152116211721182119212021212122212321242125212621272128212921302131213221332134213521362137213821392140214121422143214421452146214721482149215021512152215321542155215621572158215921602161216221632164216521662167216821692170217121722173217421752176217721782179218021812182218321842185218621872188218921902191219221932194219521962197219821992200220122022203220422052206220722082209221022112212221322142215221622172218221922202221222222232224222522262227222822292230223122322233223422352236223722382239224022412242224322442245224622472248224922502251225222532254225522562257225822592260226122622263226422652266226722682269227022712272227322742275227622772278227922802281228222832284228522862287228822892290229122922293229422952296229722982299230023012302230323042305230623072308230923102311231223132314231523162317231823192320232123222323232423252326232723282329233023312332233323342335233623372338233923402341234223432344234523462347234823492350235123522353235423552356235723582359236023612362236323642365236623672368236923702371237223732374237523762377237823792380238123822383238423852386238723882389239023912392239323942395239623972398239924002401240224032404240524062407240824092410241124122413241424152416241724182419242024212422242324242425242624272428242924302431243224332434243524362437243824392440244124422443244424452446244724482449245024512452245324542455245624572458245924602461246224632464246524662467246824692470247124722473247424752476247724782479248024812482248324842485248624872488248924902491249224932494249524962497249824992500250125022503250425052506250725082509251025112512251325142515251625172518251925202521252225232524252525262527252825292530253125322533253425352536253725382539254025412542254325442545254625472548254925502551255225532554255525562557255825592560256125622563256425652566256725682569257025712572257325742575257625772578257925802581258225832584258525862587258825892590259125922593259425952596
  1. /*
  2. * Copyright (c) 2021-2023, 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/Find.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. #include <LibJS/Runtime/ErrorTypes.h>
  18. namespace JS {
  19. Bytecode::CodeGenerationErrorOr<void> ASTNode::generate_bytecode(Bytecode::Generator&) const
  20. {
  21. return Bytecode::CodeGenerationError {
  22. this,
  23. "Missing generate_bytecode()"sv,
  24. };
  25. }
  26. Bytecode::CodeGenerationErrorOr<void> ScopeNode::generate_bytecode(Bytecode::Generator& generator) const
  27. {
  28. // Note: SwitchStatement has its own codegen, but still calls into this function to handle the scoping of the switch body.
  29. auto is_switch_statement = is<SwitchStatement>(*this);
  30. bool did_create_lexical_environment = false;
  31. if (is<BlockStatement>(*this) || is_switch_statement) {
  32. if (has_lexical_declarations()) {
  33. generator.block_declaration_instantiation(*this);
  34. did_create_lexical_environment = true;
  35. }
  36. if (is_switch_statement)
  37. return {};
  38. } else if (is<Program>(*this)) {
  39. // GlobalDeclarationInstantiation is handled by the C++ AO.
  40. } else {
  41. // FunctionDeclarationInstantiation is handled by the C++ AO.
  42. }
  43. for (auto& child : children()) {
  44. TRY(child->generate_bytecode(generator));
  45. if (generator.is_current_block_terminated())
  46. break;
  47. }
  48. if (did_create_lexical_environment)
  49. generator.end_variable_scope();
  50. return {};
  51. }
  52. Bytecode::CodeGenerationErrorOr<void> EmptyStatement::generate_bytecode(Bytecode::Generator&) const
  53. {
  54. return {};
  55. }
  56. Bytecode::CodeGenerationErrorOr<void> ExpressionStatement::generate_bytecode(Bytecode::Generator& generator) const
  57. {
  58. return m_expression->generate_bytecode(generator);
  59. }
  60. Bytecode::CodeGenerationErrorOr<void> BinaryExpression::generate_bytecode(Bytecode::Generator& generator) const
  61. {
  62. TRY(m_lhs->generate_bytecode(generator));
  63. auto lhs_reg = generator.allocate_register();
  64. generator.emit<Bytecode::Op::Store>(lhs_reg);
  65. TRY(m_rhs->generate_bytecode(generator));
  66. switch (m_op) {
  67. case BinaryOp::Addition:
  68. generator.emit<Bytecode::Op::Add>(lhs_reg);
  69. break;
  70. case BinaryOp::Subtraction:
  71. generator.emit<Bytecode::Op::Sub>(lhs_reg);
  72. break;
  73. case BinaryOp::Multiplication:
  74. generator.emit<Bytecode::Op::Mul>(lhs_reg);
  75. break;
  76. case BinaryOp::Division:
  77. generator.emit<Bytecode::Op::Div>(lhs_reg);
  78. break;
  79. case BinaryOp::Modulo:
  80. generator.emit<Bytecode::Op::Mod>(lhs_reg);
  81. break;
  82. case BinaryOp::Exponentiation:
  83. generator.emit<Bytecode::Op::Exp>(lhs_reg);
  84. break;
  85. case BinaryOp::GreaterThan:
  86. generator.emit<Bytecode::Op::GreaterThan>(lhs_reg);
  87. break;
  88. case BinaryOp::GreaterThanEquals:
  89. generator.emit<Bytecode::Op::GreaterThanEquals>(lhs_reg);
  90. break;
  91. case BinaryOp::LessThan:
  92. generator.emit<Bytecode::Op::LessThan>(lhs_reg);
  93. break;
  94. case BinaryOp::LessThanEquals:
  95. generator.emit<Bytecode::Op::LessThanEquals>(lhs_reg);
  96. break;
  97. case BinaryOp::LooselyInequals:
  98. generator.emit<Bytecode::Op::LooselyInequals>(lhs_reg);
  99. break;
  100. case BinaryOp::LooselyEquals:
  101. generator.emit<Bytecode::Op::LooselyEquals>(lhs_reg);
  102. break;
  103. case BinaryOp::StrictlyInequals:
  104. generator.emit<Bytecode::Op::StrictlyInequals>(lhs_reg);
  105. break;
  106. case BinaryOp::StrictlyEquals:
  107. generator.emit<Bytecode::Op::StrictlyEquals>(lhs_reg);
  108. break;
  109. case BinaryOp::BitwiseAnd:
  110. generator.emit<Bytecode::Op::BitwiseAnd>(lhs_reg);
  111. break;
  112. case BinaryOp::BitwiseOr:
  113. generator.emit<Bytecode::Op::BitwiseOr>(lhs_reg);
  114. break;
  115. case BinaryOp::BitwiseXor:
  116. generator.emit<Bytecode::Op::BitwiseXor>(lhs_reg);
  117. break;
  118. case BinaryOp::LeftShift:
  119. generator.emit<Bytecode::Op::LeftShift>(lhs_reg);
  120. break;
  121. case BinaryOp::RightShift:
  122. generator.emit<Bytecode::Op::RightShift>(lhs_reg);
  123. break;
  124. case BinaryOp::UnsignedRightShift:
  125. generator.emit<Bytecode::Op::UnsignedRightShift>(lhs_reg);
  126. break;
  127. case BinaryOp::In:
  128. generator.emit<Bytecode::Op::In>(lhs_reg);
  129. break;
  130. case BinaryOp::InstanceOf:
  131. generator.emit<Bytecode::Op::InstanceOf>(lhs_reg);
  132. break;
  133. default:
  134. VERIFY_NOT_REACHED();
  135. }
  136. return {};
  137. }
  138. Bytecode::CodeGenerationErrorOr<void> LogicalExpression::generate_bytecode(Bytecode::Generator& generator) const
  139. {
  140. TRY(m_lhs->generate_bytecode(generator));
  141. // lhs
  142. // jump op (true) end (false) rhs
  143. // rhs
  144. // jump always (true) end
  145. // end
  146. auto& rhs_block = generator.make_block();
  147. auto& end_block = generator.make_block();
  148. switch (m_op) {
  149. case LogicalOp::And:
  150. generator.emit<Bytecode::Op::JumpConditional>().set_targets(
  151. Bytecode::Label { rhs_block },
  152. Bytecode::Label { end_block });
  153. break;
  154. case LogicalOp::Or:
  155. generator.emit<Bytecode::Op::JumpConditional>().set_targets(
  156. Bytecode::Label { end_block },
  157. Bytecode::Label { rhs_block });
  158. break;
  159. case LogicalOp::NullishCoalescing:
  160. generator.emit<Bytecode::Op::JumpNullish>().set_targets(
  161. Bytecode::Label { rhs_block },
  162. Bytecode::Label { end_block });
  163. break;
  164. default:
  165. VERIFY_NOT_REACHED();
  166. }
  167. generator.switch_to_basic_block(rhs_block);
  168. TRY(m_rhs->generate_bytecode(generator));
  169. generator.emit<Bytecode::Op::Jump>().set_targets(
  170. Bytecode::Label { end_block },
  171. {});
  172. generator.switch_to_basic_block(end_block);
  173. return {};
  174. }
  175. Bytecode::CodeGenerationErrorOr<void> UnaryExpression::generate_bytecode(Bytecode::Generator& generator) const
  176. {
  177. if (m_op == UnaryOp::Delete)
  178. return generator.emit_delete_reference(m_lhs);
  179. // Typeof needs some special handling for when the LHS is an Identifier. Namely, it shouldn't throw on unresolvable references, but instead return "undefined".
  180. if (m_op != UnaryOp::Typeof)
  181. TRY(m_lhs->generate_bytecode(generator));
  182. switch (m_op) {
  183. case UnaryOp::BitwiseNot:
  184. generator.emit<Bytecode::Op::BitwiseNot>();
  185. break;
  186. case UnaryOp::Not:
  187. generator.emit<Bytecode::Op::Not>();
  188. break;
  189. case UnaryOp::Plus:
  190. generator.emit<Bytecode::Op::UnaryPlus>();
  191. break;
  192. case UnaryOp::Minus:
  193. generator.emit<Bytecode::Op::UnaryMinus>();
  194. break;
  195. case UnaryOp::Typeof:
  196. if (is<Identifier>(*m_lhs)) {
  197. auto& identifier = static_cast<Identifier const&>(*m_lhs);
  198. generator.emit<Bytecode::Op::TypeofVariable>(generator.intern_identifier(identifier.string()));
  199. break;
  200. }
  201. TRY(m_lhs->generate_bytecode(generator));
  202. generator.emit<Bytecode::Op::Typeof>();
  203. break;
  204. case UnaryOp::Void:
  205. generator.emit<Bytecode::Op::LoadImmediate>(js_undefined());
  206. break;
  207. case UnaryOp::Delete: // Delete is implemented above.
  208. default:
  209. VERIFY_NOT_REACHED();
  210. }
  211. return {};
  212. }
  213. Bytecode::CodeGenerationErrorOr<void> NumericLiteral::generate_bytecode(Bytecode::Generator& generator) const
  214. {
  215. generator.emit<Bytecode::Op::LoadImmediate>(m_value);
  216. return {};
  217. }
  218. Bytecode::CodeGenerationErrorOr<void> BooleanLiteral::generate_bytecode(Bytecode::Generator& generator) const
  219. {
  220. generator.emit<Bytecode::Op::LoadImmediate>(Value(m_value));
  221. return {};
  222. }
  223. Bytecode::CodeGenerationErrorOr<void> NullLiteral::generate_bytecode(Bytecode::Generator& generator) const
  224. {
  225. generator.emit<Bytecode::Op::LoadImmediate>(js_null());
  226. return {};
  227. }
  228. Bytecode::CodeGenerationErrorOr<void> BigIntLiteral::generate_bytecode(Bytecode::Generator& generator) const
  229. {
  230. // 1. Return the NumericValue of NumericLiteral as defined in 12.8.3.
  231. auto integer = [&] {
  232. if (m_value[0] == '0' && m_value.length() >= 3)
  233. if (m_value[1] == 'x' || m_value[1] == 'X')
  234. return Crypto::SignedBigInteger::from_base(16, m_value.substring(2, m_value.length() - 3));
  235. if (m_value[1] == 'o' || m_value[1] == 'O')
  236. return Crypto::SignedBigInteger::from_base(8, m_value.substring(2, m_value.length() - 3));
  237. if (m_value[1] == 'b' || m_value[1] == 'B')
  238. return Crypto::SignedBigInteger::from_base(2, m_value.substring(2, m_value.length() - 3));
  239. return Crypto::SignedBigInteger::from_base(10, m_value.substring(0, m_value.length() - 1));
  240. }();
  241. generator.emit<Bytecode::Op::NewBigInt>(integer);
  242. return {};
  243. }
  244. Bytecode::CodeGenerationErrorOr<void> StringLiteral::generate_bytecode(Bytecode::Generator& generator) const
  245. {
  246. generator.emit<Bytecode::Op::NewString>(generator.intern_string(m_value));
  247. return {};
  248. }
  249. Bytecode::CodeGenerationErrorOr<void> RegExpLiteral::generate_bytecode(Bytecode::Generator& generator) const
  250. {
  251. auto source_index = generator.intern_string(m_pattern);
  252. auto flags_index = generator.intern_string(m_flags);
  253. generator.emit<Bytecode::Op::NewRegExp>(source_index, flags_index);
  254. return {};
  255. }
  256. Bytecode::CodeGenerationErrorOr<void> Identifier::generate_bytecode(Bytecode::Generator& generator) const
  257. {
  258. generator.emit<Bytecode::Op::GetVariable>(generator.intern_identifier(m_string));
  259. return {};
  260. }
  261. static Bytecode::CodeGenerationErrorOr<void> arguments_to_array_for_call(Bytecode::Generator& generator, ReadonlySpan<CallExpression::Argument> arguments)
  262. {
  263. if (arguments.is_empty()) {
  264. generator.emit<Bytecode::Op::NewArray>();
  265. return {};
  266. }
  267. auto first_spread = find_if(arguments.begin(), arguments.end(), [](auto el) { return el.is_spread; });
  268. Bytecode::Register args_start_reg { 0 };
  269. for (auto it = arguments.begin(); it != first_spread; ++it) {
  270. auto reg = generator.allocate_register();
  271. if (args_start_reg.index() == 0)
  272. args_start_reg = reg;
  273. }
  274. u32 i = 0;
  275. for (auto it = arguments.begin(); it != first_spread; ++it, ++i) {
  276. VERIFY(it->is_spread == false);
  277. Bytecode::Register reg { args_start_reg.index() + i };
  278. TRY(it->value->generate_bytecode(generator));
  279. generator.emit<Bytecode::Op::Store>(reg);
  280. }
  281. if (first_spread.index() != 0)
  282. generator.emit_with_extra_register_slots<Bytecode::Op::NewArray>(2u, AK::Array { args_start_reg, Bytecode::Register { args_start_reg.index() + static_cast<u32>(first_spread.index() - 1) } });
  283. else
  284. generator.emit<Bytecode::Op::NewArray>();
  285. if (first_spread != arguments.end()) {
  286. auto array_reg = generator.allocate_register();
  287. generator.emit<Bytecode::Op::Store>(array_reg);
  288. for (auto it = first_spread; it != arguments.end(); ++it) {
  289. TRY(it->value->generate_bytecode(generator));
  290. generator.emit<Bytecode::Op::Append>(array_reg, it->is_spread);
  291. }
  292. generator.emit<Bytecode::Op::Load>(array_reg);
  293. }
  294. return {};
  295. }
  296. Bytecode::CodeGenerationErrorOr<void> SuperCall::generate_bytecode(Bytecode::Generator& generator) const
  297. {
  298. if (m_is_synthetic == IsPartOfSyntheticConstructor::Yes) {
  299. // NOTE: This is the case where we have a fake constructor(...args) { super(...args); } which
  300. // shouldn't call @@iterator of %Array.prototype%.
  301. VERIFY(m_arguments.size() == 1);
  302. VERIFY(m_arguments[0].is_spread);
  303. auto const& argument = m_arguments[0];
  304. // This generates a single argument, which will be implicitly passed in accumulator
  305. MUST(argument.value->generate_bytecode(generator));
  306. } else {
  307. TRY(arguments_to_array_for_call(generator, m_arguments));
  308. }
  309. generator.emit<Bytecode::Op::SuperCall>(m_is_synthetic == IsPartOfSyntheticConstructor::Yes);
  310. return {};
  311. }
  312. static Bytecode::CodeGenerationErrorOr<void> generate_binding_pattern_bytecode(Bytecode::Generator& generator, BindingPattern const& pattern, Bytecode::Op::SetVariable::InitializationMode, Bytecode::Register const& value_reg);
  313. Bytecode::CodeGenerationErrorOr<void> AssignmentExpression::generate_bytecode(Bytecode::Generator& generator) const
  314. {
  315. if (m_op == AssignmentOp::Assignment) {
  316. // AssignmentExpression : LeftHandSideExpression = AssignmentExpression
  317. return m_lhs.visit(
  318. // 1. If LeftHandSideExpression is neither an ObjectLiteral nor an ArrayLiteral, then
  319. [&](NonnullRefPtr<Expression const> const& lhs) -> Bytecode::CodeGenerationErrorOr<void> {
  320. // a. Let lref be the result of evaluating LeftHandSideExpression.
  321. // b. ReturnIfAbrupt(lref).
  322. Optional<Bytecode::Register> base_object_register;
  323. Optional<Bytecode::Register> computed_property_register;
  324. if (is<MemberExpression>(*lhs)) {
  325. auto& expression = static_cast<MemberExpression const&>(*lhs);
  326. TRY(expression.object().generate_bytecode(generator));
  327. base_object_register = generator.allocate_register();
  328. generator.emit<Bytecode::Op::Store>(*base_object_register);
  329. if (expression.is_computed()) {
  330. TRY(expression.property().generate_bytecode(generator));
  331. computed_property_register = generator.allocate_register();
  332. generator.emit<Bytecode::Op::Store>(*computed_property_register);
  333. // To be continued later with PutByValue.
  334. } else if (expression.property().is_identifier()) {
  335. // Do nothing, this will be handled by PutById later.
  336. } else {
  337. return Bytecode::CodeGenerationError {
  338. &expression,
  339. "Unimplemented non-computed member expression"sv
  340. };
  341. }
  342. } else if (is<Identifier>(*lhs)) {
  343. // NOTE: For Identifiers, we cannot perform GetVariable and then write into the reference it retrieves, only SetVariable can do this.
  344. // FIXME: However, this breaks spec as we are doing variable lookup after evaluating the RHS. This is observable in an object environment, where we visibly perform HasOwnProperty and Get(@@unscopables) on the binded object.
  345. } else {
  346. TRY(lhs->generate_bytecode(generator));
  347. }
  348. // FIXME: c. If IsAnonymousFunctionDefinition(AssignmentExpression) and IsIdentifierRef of LeftHandSideExpression are both true, then
  349. // i. Let rval be ? NamedEvaluation of AssignmentExpression with argument lref.[[ReferencedName]].
  350. // d. Else,
  351. // i. Let rref be the result of evaluating AssignmentExpression.
  352. // ii. Let rval be ? GetValue(rref).
  353. TRY(m_rhs->generate_bytecode(generator));
  354. // e. Perform ? PutValue(lref, rval).
  355. if (is<Identifier>(*lhs)) {
  356. auto& identifier = static_cast<Identifier const&>(*lhs);
  357. generator.emit<Bytecode::Op::SetVariable>(generator.intern_identifier(identifier.string()));
  358. } else if (is<MemberExpression>(*lhs)) {
  359. auto& expression = static_cast<MemberExpression const&>(*lhs);
  360. if (expression.is_computed()) {
  361. generator.emit<Bytecode::Op::PutByValue>(*base_object_register, *computed_property_register);
  362. } else if (expression.property().is_identifier()) {
  363. auto identifier_table_ref = generator.intern_identifier(verify_cast<Identifier>(expression.property()).string());
  364. generator.emit<Bytecode::Op::PutById>(*base_object_register, identifier_table_ref);
  365. } else {
  366. return Bytecode::CodeGenerationError {
  367. &expression,
  368. "Unimplemented non-computed member expression"sv
  369. };
  370. }
  371. } else {
  372. return Bytecode::CodeGenerationError {
  373. lhs,
  374. "Unimplemented/invalid node used a reference"sv
  375. };
  376. }
  377. // f. Return rval.
  378. // NOTE: This is already in the accumulator.
  379. return {};
  380. },
  381. // 2. Let assignmentPattern be the AssignmentPattern that is covered by LeftHandSideExpression.
  382. [&](NonnullRefPtr<BindingPattern const> const& pattern) -> Bytecode::CodeGenerationErrorOr<void> {
  383. // 3. Let rref be the result of evaluating AssignmentExpression.
  384. // 4. Let rval be ? GetValue(rref).
  385. TRY(m_rhs->generate_bytecode(generator));
  386. auto value_register = generator.allocate_register();
  387. generator.emit<Bytecode::Op::Store>(value_register);
  388. // 5. Perform ? DestructuringAssignmentEvaluation of assignmentPattern with argument rval.
  389. TRY(generate_binding_pattern_bytecode(generator, pattern, Bytecode::Op::SetVariable::InitializationMode::Set, value_register));
  390. // 6. Return rval.
  391. generator.emit<Bytecode::Op::Load>(value_register);
  392. return {};
  393. });
  394. }
  395. VERIFY(m_lhs.has<NonnullRefPtr<Expression const>>());
  396. auto& lhs = m_lhs.get<NonnullRefPtr<Expression const>>();
  397. TRY(generator.emit_load_from_reference(lhs));
  398. Bytecode::BasicBlock* rhs_block_ptr { nullptr };
  399. Bytecode::BasicBlock* end_block_ptr { nullptr };
  400. // Logical assignments short circuit.
  401. if (m_op == AssignmentOp::AndAssignment) { // &&=
  402. rhs_block_ptr = &generator.make_block();
  403. end_block_ptr = &generator.make_block();
  404. generator.emit<Bytecode::Op::JumpConditional>().set_targets(
  405. Bytecode::Label { *rhs_block_ptr },
  406. Bytecode::Label { *end_block_ptr });
  407. } else if (m_op == AssignmentOp::OrAssignment) { // ||=
  408. rhs_block_ptr = &generator.make_block();
  409. end_block_ptr = &generator.make_block();
  410. generator.emit<Bytecode::Op::JumpConditional>().set_targets(
  411. Bytecode::Label { *end_block_ptr },
  412. Bytecode::Label { *rhs_block_ptr });
  413. } else if (m_op == AssignmentOp::NullishAssignment) { // ??=
  414. rhs_block_ptr = &generator.make_block();
  415. end_block_ptr = &generator.make_block();
  416. generator.emit<Bytecode::Op::JumpNullish>().set_targets(
  417. Bytecode::Label { *rhs_block_ptr },
  418. Bytecode::Label { *end_block_ptr });
  419. }
  420. if (rhs_block_ptr)
  421. generator.switch_to_basic_block(*rhs_block_ptr);
  422. // lhs_reg is a part of the rhs_block because the store isn't necessary
  423. // if the logical assignment condition fails.
  424. auto lhs_reg = generator.allocate_register();
  425. generator.emit<Bytecode::Op::Store>(lhs_reg);
  426. TRY(m_rhs->generate_bytecode(generator));
  427. switch (m_op) {
  428. case AssignmentOp::AdditionAssignment:
  429. generator.emit<Bytecode::Op::Add>(lhs_reg);
  430. break;
  431. case AssignmentOp::SubtractionAssignment:
  432. generator.emit<Bytecode::Op::Sub>(lhs_reg);
  433. break;
  434. case AssignmentOp::MultiplicationAssignment:
  435. generator.emit<Bytecode::Op::Mul>(lhs_reg);
  436. break;
  437. case AssignmentOp::DivisionAssignment:
  438. generator.emit<Bytecode::Op::Div>(lhs_reg);
  439. break;
  440. case AssignmentOp::ModuloAssignment:
  441. generator.emit<Bytecode::Op::Mod>(lhs_reg);
  442. break;
  443. case AssignmentOp::ExponentiationAssignment:
  444. generator.emit<Bytecode::Op::Exp>(lhs_reg);
  445. break;
  446. case AssignmentOp::BitwiseAndAssignment:
  447. generator.emit<Bytecode::Op::BitwiseAnd>(lhs_reg);
  448. break;
  449. case AssignmentOp::BitwiseOrAssignment:
  450. generator.emit<Bytecode::Op::BitwiseOr>(lhs_reg);
  451. break;
  452. case AssignmentOp::BitwiseXorAssignment:
  453. generator.emit<Bytecode::Op::BitwiseXor>(lhs_reg);
  454. break;
  455. case AssignmentOp::LeftShiftAssignment:
  456. generator.emit<Bytecode::Op::LeftShift>(lhs_reg);
  457. break;
  458. case AssignmentOp::RightShiftAssignment:
  459. generator.emit<Bytecode::Op::RightShift>(lhs_reg);
  460. break;
  461. case AssignmentOp::UnsignedRightShiftAssignment:
  462. generator.emit<Bytecode::Op::UnsignedRightShift>(lhs_reg);
  463. break;
  464. case AssignmentOp::AndAssignment:
  465. case AssignmentOp::OrAssignment:
  466. case AssignmentOp::NullishAssignment:
  467. break; // These are handled above.
  468. default:
  469. return Bytecode::CodeGenerationError {
  470. this,
  471. "Unimplemented operation"sv,
  472. };
  473. }
  474. TRY(generator.emit_store_to_reference(lhs));
  475. if (end_block_ptr) {
  476. generator.emit<Bytecode::Op::Jump>().set_targets(
  477. Bytecode::Label { *end_block_ptr },
  478. {});
  479. generator.switch_to_basic_block(*end_block_ptr);
  480. }
  481. return {};
  482. }
  483. // 14.13.3 Runtime Semantics: Evaluation, https://tc39.es/ecma262/#sec-labelled-statements-runtime-semantics-evaluation
  484. // LabelledStatement : LabelIdentifier : LabelledItem
  485. Bytecode::CodeGenerationErrorOr<void> LabelledStatement::generate_bytecode(Bytecode::Generator& generator) const
  486. {
  487. // Return ? LabelledEvaluation of this LabelledStatement with argument « ».
  488. return generate_labelled_evaluation(generator, {});
  489. }
  490. // 14.13.4 Runtime Semantics: LabelledEvaluation, https://tc39.es/ecma262/#sec-runtime-semantics-labelledevaluation
  491. // LabelledStatement : LabelIdentifier : LabelledItem
  492. Bytecode::CodeGenerationErrorOr<void> LabelledStatement::generate_labelled_evaluation(Bytecode::Generator& generator, Vector<DeprecatedFlyString> const& label_set) const
  493. {
  494. // Convert the m_labelled_item NNRP to a reference early so we don't have to do it every single time we want to use it.
  495. auto const& labelled_item = *m_labelled_item;
  496. // 1. Let label be the StringValue of LabelIdentifier.
  497. // NOTE: Not necessary, this is m_label.
  498. // 2. Let newLabelSet be the list-concatenation of labelSet and « label ».
  499. // FIXME: Avoid copy here.
  500. auto new_label_set = label_set;
  501. new_label_set.append(m_label);
  502. // 3. Let stmtResult be LabelledEvaluation of LabelledItem with argument newLabelSet.
  503. // NOTE: stmtResult will be in the accumulator after running the generated bytecode.
  504. if (is<IterationStatement>(labelled_item)) {
  505. auto const& iteration_statement = static_cast<IterationStatement const&>(labelled_item);
  506. TRY(iteration_statement.generate_labelled_evaluation(generator, new_label_set));
  507. } else if (is<SwitchStatement>(labelled_item)) {
  508. auto const& switch_statement = static_cast<SwitchStatement const&>(labelled_item);
  509. TRY(switch_statement.generate_labelled_evaluation(generator, new_label_set));
  510. } else if (is<LabelledStatement>(labelled_item)) {
  511. auto const& labelled_statement = static_cast<LabelledStatement const&>(labelled_item);
  512. TRY(labelled_statement.generate_labelled_evaluation(generator, new_label_set));
  513. } else {
  514. auto& labelled_break_block = generator.make_block();
  515. // NOTE: We do not need a continuable scope as `continue;` is not allowed outside of iteration statements, throwing a SyntaxError in the parser.
  516. generator.begin_breakable_scope(Bytecode::Label { labelled_break_block }, new_label_set);
  517. TRY(labelled_item.generate_bytecode(generator));
  518. generator.end_breakable_scope();
  519. if (!generator.is_current_block_terminated()) {
  520. generator.emit<Bytecode::Op::Jump>().set_targets(
  521. Bytecode::Label { labelled_break_block },
  522. {});
  523. }
  524. generator.switch_to_basic_block(labelled_break_block);
  525. }
  526. // 4. If stmtResult.[[Type]] is break and SameValue(stmtResult.[[Target]], label) is true, then
  527. // a. Set stmtResult to NormalCompletion(stmtResult.[[Value]]).
  528. // NOTE: These steps are performed by making labelled break jump straight to the appropriate break block, which preserves the statement result's value in the accumulator.
  529. // 5. Return Completion(stmtResult).
  530. // NOTE: This is in the accumulator.
  531. return {};
  532. }
  533. Bytecode::CodeGenerationErrorOr<void> IterationStatement::generate_labelled_evaluation(Bytecode::Generator&, Vector<DeprecatedFlyString> const&) const
  534. {
  535. return Bytecode::CodeGenerationError {
  536. this,
  537. "Missing generate_labelled_evaluation()"sv,
  538. };
  539. }
  540. Bytecode::CodeGenerationErrorOr<void> WhileStatement::generate_bytecode(Bytecode::Generator& generator) const
  541. {
  542. return generate_labelled_evaluation(generator, {});
  543. }
  544. Bytecode::CodeGenerationErrorOr<void> WhileStatement::generate_labelled_evaluation(Bytecode::Generator& generator, Vector<DeprecatedFlyString> const& label_set) const
  545. {
  546. // test
  547. // jump if_false (true) end (false) body
  548. // body
  549. // jump always (true) test
  550. // end
  551. auto& test_block = generator.make_block();
  552. auto& body_block = generator.make_block();
  553. auto& end_block = generator.make_block();
  554. // Init result register
  555. generator.emit<Bytecode::Op::LoadImmediate>(js_undefined());
  556. auto result_reg = generator.allocate_register();
  557. generator.emit<Bytecode::Op::Store>(result_reg);
  558. // jump to the test block
  559. generator.emit<Bytecode::Op::Jump>().set_targets(
  560. Bytecode::Label { test_block },
  561. {});
  562. generator.switch_to_basic_block(test_block);
  563. TRY(m_test->generate_bytecode(generator));
  564. generator.emit<Bytecode::Op::JumpConditional>().set_targets(
  565. Bytecode::Label { body_block },
  566. Bytecode::Label { end_block });
  567. generator.switch_to_basic_block(body_block);
  568. generator.begin_continuable_scope(Bytecode::Label { test_block }, label_set);
  569. generator.begin_breakable_scope(Bytecode::Label { end_block }, label_set);
  570. TRY(m_body->generate_bytecode(generator));
  571. generator.end_breakable_scope();
  572. generator.end_continuable_scope();
  573. if (!generator.is_current_block_terminated()) {
  574. generator.emit<Bytecode::Op::Jump>().set_targets(
  575. Bytecode::Label { test_block },
  576. {});
  577. }
  578. generator.switch_to_basic_block(end_block);
  579. generator.emit<Bytecode::Op::Load>(result_reg);
  580. return {};
  581. }
  582. Bytecode::CodeGenerationErrorOr<void> DoWhileStatement::generate_bytecode(Bytecode::Generator& generator) const
  583. {
  584. return generate_labelled_evaluation(generator, {});
  585. }
  586. Bytecode::CodeGenerationErrorOr<void> DoWhileStatement::generate_labelled_evaluation(Bytecode::Generator& generator, Vector<DeprecatedFlyString> const& label_set) const
  587. {
  588. // jump always (true) body
  589. // test
  590. // jump if_false (true) end (false) body
  591. // body
  592. // jump always (true) test
  593. // end
  594. auto& test_block = generator.make_block();
  595. auto& body_block = generator.make_block();
  596. auto& end_block = generator.make_block();
  597. // Init result register
  598. generator.emit<Bytecode::Op::LoadImmediate>(js_undefined());
  599. auto result_reg = generator.allocate_register();
  600. generator.emit<Bytecode::Op::Store>(result_reg);
  601. // jump to the body block
  602. generator.emit<Bytecode::Op::Jump>().set_targets(
  603. Bytecode::Label { body_block },
  604. {});
  605. generator.switch_to_basic_block(test_block);
  606. TRY(m_test->generate_bytecode(generator));
  607. generator.emit<Bytecode::Op::JumpConditional>().set_targets(
  608. Bytecode::Label { body_block },
  609. Bytecode::Label { end_block });
  610. generator.switch_to_basic_block(body_block);
  611. generator.begin_continuable_scope(Bytecode::Label { test_block }, label_set);
  612. generator.begin_breakable_scope(Bytecode::Label { end_block }, label_set);
  613. TRY(m_body->generate_bytecode(generator));
  614. generator.end_breakable_scope();
  615. generator.end_continuable_scope();
  616. if (!generator.is_current_block_terminated()) {
  617. generator.emit<Bytecode::Op::Jump>().set_targets(
  618. Bytecode::Label { test_block },
  619. {});
  620. }
  621. generator.switch_to_basic_block(end_block);
  622. generator.emit<Bytecode::Op::Load>(result_reg);
  623. return {};
  624. }
  625. Bytecode::CodeGenerationErrorOr<void> ForStatement::generate_bytecode(Bytecode::Generator& generator) const
  626. {
  627. return generate_labelled_evaluation(generator, {});
  628. }
  629. Bytecode::CodeGenerationErrorOr<void> ForStatement::generate_labelled_evaluation(Bytecode::Generator& generator, Vector<DeprecatedFlyString> const& label_set) const
  630. {
  631. // init
  632. // jump always (true) test
  633. // test
  634. // jump if_true (true) body (false) end
  635. // body
  636. // jump always (true) update
  637. // update
  638. // jump always (true) test
  639. // end
  640. // If 'test' is missing, fuse the 'test' and 'body' basic blocks
  641. // If 'update' is missing, fuse the 'body' and 'update' basic blocks
  642. Bytecode::BasicBlock* test_block_ptr { nullptr };
  643. Bytecode::BasicBlock* body_block_ptr { nullptr };
  644. Bytecode::BasicBlock* update_block_ptr { nullptr };
  645. auto& end_block = generator.make_block();
  646. bool has_lexical_environment = false;
  647. if (m_init) {
  648. if (m_init->is_variable_declaration()) {
  649. auto& variable_declaration = verify_cast<VariableDeclaration>(*m_init);
  650. if (variable_declaration.is_lexical_declaration()) {
  651. has_lexical_environment = true;
  652. // FIXME: Is Block correct?
  653. generator.begin_variable_scope();
  654. bool is_const = variable_declaration.is_constant_declaration();
  655. // NOTE: Nothing in the callback throws an exception.
  656. MUST(variable_declaration.for_each_bound_name([&](auto const& name) {
  657. auto index = generator.intern_identifier(name);
  658. generator.emit<Bytecode::Op::CreateVariable>(index, Bytecode::Op::EnvironmentMode::Lexical, is_const);
  659. }));
  660. }
  661. }
  662. TRY(m_init->generate_bytecode(generator));
  663. }
  664. body_block_ptr = &generator.make_block();
  665. if (m_test)
  666. test_block_ptr = &generator.make_block();
  667. else
  668. test_block_ptr = body_block_ptr;
  669. if (m_update)
  670. update_block_ptr = &generator.make_block();
  671. else
  672. update_block_ptr = body_block_ptr;
  673. generator.emit<Bytecode::Op::LoadImmediate>(js_undefined());
  674. auto result_reg = generator.allocate_register();
  675. generator.emit<Bytecode::Op::Store>(result_reg);
  676. generator.emit<Bytecode::Op::Jump>().set_targets(
  677. Bytecode::Label { *test_block_ptr },
  678. {});
  679. if (m_test) {
  680. generator.switch_to_basic_block(*test_block_ptr);
  681. TRY(m_test->generate_bytecode(generator));
  682. generator.emit<Bytecode::Op::JumpConditional>().set_targets(
  683. Bytecode::Label { *body_block_ptr },
  684. Bytecode::Label { end_block });
  685. }
  686. if (m_update) {
  687. generator.switch_to_basic_block(*update_block_ptr);
  688. TRY(m_update->generate_bytecode(generator));
  689. generator.emit<Bytecode::Op::Jump>().set_targets(
  690. Bytecode::Label { *test_block_ptr },
  691. {});
  692. }
  693. generator.switch_to_basic_block(*body_block_ptr);
  694. generator.begin_continuable_scope(Bytecode::Label { m_update ? *update_block_ptr : *test_block_ptr }, label_set);
  695. generator.begin_breakable_scope(Bytecode::Label { end_block }, label_set);
  696. TRY(m_body->generate_bytecode(generator));
  697. generator.end_breakable_scope();
  698. generator.end_continuable_scope();
  699. if (!generator.is_current_block_terminated()) {
  700. if (m_update) {
  701. generator.emit<Bytecode::Op::Jump>().set_targets(
  702. Bytecode::Label { *update_block_ptr },
  703. {});
  704. } else {
  705. generator.emit<Bytecode::Op::Jump>().set_targets(
  706. Bytecode::Label { *test_block_ptr },
  707. {});
  708. }
  709. }
  710. generator.switch_to_basic_block(end_block);
  711. generator.emit<Bytecode::Op::Load>(result_reg);
  712. if (has_lexical_environment)
  713. generator.end_variable_scope();
  714. return {};
  715. }
  716. Bytecode::CodeGenerationErrorOr<void> ObjectExpression::generate_bytecode(Bytecode::Generator& generator) const
  717. {
  718. generator.emit<Bytecode::Op::NewObject>();
  719. if (m_properties.is_empty())
  720. return {};
  721. auto object_reg = generator.allocate_register();
  722. generator.emit<Bytecode::Op::Store>(object_reg);
  723. generator.push_home_object(object_reg);
  724. for (auto& property : m_properties) {
  725. Bytecode::Op::PropertyKind property_kind;
  726. switch (property->type()) {
  727. case ObjectProperty::Type::KeyValue:
  728. property_kind = Bytecode::Op::PropertyKind::KeyValue;
  729. break;
  730. case ObjectProperty::Type::Getter:
  731. property_kind = Bytecode::Op::PropertyKind::Getter;
  732. break;
  733. case ObjectProperty::Type::Setter:
  734. property_kind = Bytecode::Op::PropertyKind::Setter;
  735. break;
  736. case ObjectProperty::Type::Spread:
  737. property_kind = Bytecode::Op::PropertyKind::Spread;
  738. break;
  739. case ObjectProperty::Type::ProtoSetter:
  740. property_kind = Bytecode::Op::PropertyKind::ProtoSetter;
  741. break;
  742. }
  743. if (is<StringLiteral>(property->key())) {
  744. auto& string_literal = static_cast<StringLiteral const&>(property->key());
  745. Bytecode::IdentifierTableIndex key_name = generator.intern_identifier(string_literal.value());
  746. if (property_kind != Bytecode::Op::PropertyKind::Spread)
  747. TRY(property->value().generate_bytecode(generator));
  748. generator.emit<Bytecode::Op::PutById>(object_reg, key_name, property_kind);
  749. } else {
  750. TRY(property->key().generate_bytecode(generator));
  751. auto property_reg = generator.allocate_register();
  752. generator.emit<Bytecode::Op::Store>(property_reg);
  753. if (property_kind != Bytecode::Op::PropertyKind::Spread)
  754. TRY(property->value().generate_bytecode(generator));
  755. generator.emit<Bytecode::Op::PutByValue>(object_reg, property_reg, property_kind);
  756. }
  757. }
  758. generator.emit<Bytecode::Op::Load>(object_reg);
  759. generator.pop_home_object();
  760. return {};
  761. }
  762. Bytecode::CodeGenerationErrorOr<void> ArrayExpression::generate_bytecode(Bytecode::Generator& generator) const
  763. {
  764. if (m_elements.is_empty()) {
  765. generator.emit<Bytecode::Op::NewArray>();
  766. return {};
  767. }
  768. auto first_spread = find_if(m_elements.begin(), m_elements.end(), [](auto el) { return el && is<SpreadExpression>(*el); });
  769. Bytecode::Register args_start_reg { 0 };
  770. for (auto it = m_elements.begin(); it != first_spread; ++it) {
  771. auto reg = generator.allocate_register();
  772. if (args_start_reg.index() == 0)
  773. args_start_reg = reg;
  774. }
  775. u32 i = 0;
  776. for (auto it = m_elements.begin(); it != first_spread; ++it, ++i) {
  777. Bytecode::Register reg { args_start_reg.index() + i };
  778. if (!*it)
  779. generator.emit<Bytecode::Op::LoadImmediate>(Value {});
  780. else {
  781. TRY((*it)->generate_bytecode(generator));
  782. }
  783. generator.emit<Bytecode::Op::Store>(reg);
  784. }
  785. if (first_spread.index() != 0)
  786. generator.emit_with_extra_register_slots<Bytecode::Op::NewArray>(2u, AK::Array { args_start_reg, Bytecode::Register { args_start_reg.index() + static_cast<u32>(first_spread.index() - 1) } });
  787. else
  788. generator.emit<Bytecode::Op::NewArray>();
  789. if (first_spread != m_elements.end()) {
  790. auto array_reg = generator.allocate_register();
  791. generator.emit<Bytecode::Op::Store>(array_reg);
  792. for (auto it = first_spread; it != m_elements.end(); ++it) {
  793. if (!*it) {
  794. generator.emit<Bytecode::Op::LoadImmediate>(Value {});
  795. generator.emit<Bytecode::Op::Append>(array_reg, false);
  796. } else {
  797. TRY((*it)->generate_bytecode(generator));
  798. generator.emit<Bytecode::Op::Append>(array_reg, *it && is<SpreadExpression>(**it));
  799. }
  800. }
  801. generator.emit<Bytecode::Op::Load>(array_reg);
  802. }
  803. return {};
  804. }
  805. Bytecode::CodeGenerationErrorOr<void> MemberExpression::generate_bytecode(Bytecode::Generator& generator) const
  806. {
  807. return generator.emit_load_from_reference(*this);
  808. }
  809. Bytecode::CodeGenerationErrorOr<void> FunctionDeclaration::generate_bytecode(Bytecode::Generator& generator) const
  810. {
  811. if (m_is_hoisted) {
  812. auto index = generator.intern_identifier(name());
  813. generator.emit<Bytecode::Op::GetVariable>(index);
  814. generator.emit<Bytecode::Op::SetVariable>(index, Bytecode::Op::SetVariable::InitializationMode::Set, Bytecode::Op::EnvironmentMode::Var);
  815. }
  816. return {};
  817. }
  818. Bytecode::CodeGenerationErrorOr<void> FunctionExpression::generate_bytecode(Bytecode::Generator& generator) const
  819. {
  820. bool has_name = !name().is_empty();
  821. Optional<Bytecode::IdentifierTableIndex> name_identifier;
  822. if (has_name) {
  823. generator.begin_variable_scope();
  824. name_identifier = generator.intern_identifier(name());
  825. generator.emit<Bytecode::Op::CreateVariable>(*name_identifier, Bytecode::Op::EnvironmentMode::Lexical, true);
  826. }
  827. generator.emit_new_function(*this);
  828. if (has_name) {
  829. generator.emit<Bytecode::Op::SetVariable>(*name_identifier, Bytecode::Op::SetVariable::InitializationMode::Initialize, Bytecode::Op::EnvironmentMode::Lexical);
  830. generator.end_variable_scope();
  831. }
  832. return {};
  833. }
  834. 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)
  835. {
  836. Vector<Bytecode::Register> excluded_property_names;
  837. auto has_rest = false;
  838. if (pattern.entries.size() > 0)
  839. has_rest = pattern.entries[pattern.entries.size() - 1].is_rest;
  840. for (auto& [name, alias, initializer, is_rest] : pattern.entries) {
  841. if (is_rest) {
  842. VERIFY(name.has<NonnullRefPtr<Identifier const>>());
  843. VERIFY(alias.has<Empty>());
  844. VERIFY(!initializer);
  845. auto identifier = name.get<NonnullRefPtr<Identifier const>>()->string();
  846. auto interned_identifier = generator.intern_identifier(identifier);
  847. generator.emit_with_extra_register_slots<Bytecode::Op::CopyObjectExcludingProperties>(excluded_property_names.size(), value_reg, excluded_property_names);
  848. generator.emit<Bytecode::Op::SetVariable>(interned_identifier, initialization_mode);
  849. return {};
  850. }
  851. Bytecode::StringTableIndex name_index;
  852. if (name.has<NonnullRefPtr<Identifier const>>()) {
  853. auto identifier = name.get<NonnullRefPtr<Identifier const>>()->string();
  854. name_index = generator.intern_string(identifier);
  855. if (has_rest) {
  856. auto excluded_name_reg = generator.allocate_register();
  857. excluded_property_names.append(excluded_name_reg);
  858. generator.emit<Bytecode::Op::NewString>(name_index);
  859. generator.emit<Bytecode::Op::Store>(excluded_name_reg);
  860. }
  861. generator.emit<Bytecode::Op::Load>(value_reg);
  862. generator.emit<Bytecode::Op::GetById>(generator.intern_identifier(identifier));
  863. } else {
  864. auto expression = name.get<NonnullRefPtr<Expression const>>();
  865. TRY(expression->generate_bytecode(generator));
  866. if (has_rest) {
  867. auto excluded_name_reg = generator.allocate_register();
  868. excluded_property_names.append(excluded_name_reg);
  869. generator.emit<Bytecode::Op::Store>(excluded_name_reg);
  870. }
  871. generator.emit<Bytecode::Op::GetByValue>(value_reg);
  872. }
  873. if (initializer) {
  874. auto& if_undefined_block = generator.make_block();
  875. auto& if_not_undefined_block = generator.make_block();
  876. generator.emit<Bytecode::Op::JumpUndefined>().set_targets(
  877. Bytecode::Label { if_undefined_block },
  878. Bytecode::Label { if_not_undefined_block });
  879. generator.switch_to_basic_block(if_undefined_block);
  880. TRY(initializer->generate_bytecode(generator));
  881. generator.emit<Bytecode::Op::Jump>().set_targets(
  882. Bytecode::Label { if_not_undefined_block },
  883. {});
  884. generator.switch_to_basic_block(if_not_undefined_block);
  885. }
  886. if (alias.has<NonnullRefPtr<BindingPattern const>>()) {
  887. auto& binding_pattern = *alias.get<NonnullRefPtr<BindingPattern const>>();
  888. auto nested_value_reg = generator.allocate_register();
  889. generator.emit<Bytecode::Op::Store>(nested_value_reg);
  890. TRY(generate_binding_pattern_bytecode(generator, binding_pattern, initialization_mode, nested_value_reg));
  891. } else if (alias.has<Empty>()) {
  892. if (name.has<NonnullRefPtr<Expression const>>()) {
  893. // This needs some sort of SetVariableByValue opcode, as it's a runtime binding
  894. return Bytecode::CodeGenerationError {
  895. name.get<NonnullRefPtr<Expression const>>().ptr(),
  896. "Unimplemented name/alias pair: Empty/Expression"sv,
  897. };
  898. }
  899. auto& identifier = name.get<NonnullRefPtr<Identifier const>>()->string();
  900. generator.emit<Bytecode::Op::SetVariable>(generator.intern_identifier(identifier), initialization_mode);
  901. } else {
  902. auto& identifier = alias.get<NonnullRefPtr<Identifier const>>()->string();
  903. generator.emit<Bytecode::Op::SetVariable>(generator.intern_identifier(identifier), initialization_mode);
  904. }
  905. }
  906. return {};
  907. }
  908. 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)
  909. {
  910. /*
  911. * Consider the following destructuring assignment:
  912. *
  913. * let [a, b, c, d, e] = o;
  914. *
  915. * It would be fairly trivial to just loop through this iterator, getting the value
  916. * at each step and assigning them to the binding sequentially. However, this is not
  917. * correct: once an iterator is exhausted, it must not be called again. This complicates
  918. * the bytecode. In order to accomplish this, we do the following:
  919. *
  920. * - Reserve a special boolean register which holds 'true' if the iterator is exhausted,
  921. * and false otherwise
  922. * - When we are retrieving the value which should be bound, we first check this register.
  923. * If it is 'true', we load undefined into the accumulator. Otherwise, we grab the next
  924. * value from the iterator and store it into the accumulator.
  925. *
  926. * Note that the is_exhausted register does not need to be loaded with false because the
  927. * first IteratorNext bytecode is _not_ proceeded by an exhausted check, as it is
  928. * unnecessary.
  929. */
  930. auto is_iterator_exhausted_register = generator.allocate_register();
  931. auto iterator_reg = generator.allocate_register();
  932. generator.emit<Bytecode::Op::Load>(value_reg);
  933. generator.emit<Bytecode::Op::GetIterator>();
  934. generator.emit<Bytecode::Op::Store>(iterator_reg);
  935. bool first = true;
  936. auto temp_iterator_result_reg = generator.allocate_register();
  937. auto assign_accumulator_to_alias = [&](auto& alias) {
  938. return alias.visit(
  939. [&](Empty) -> Bytecode::CodeGenerationErrorOr<void> {
  940. // This element is an elision
  941. return {};
  942. },
  943. [&](NonnullRefPtr<Identifier const> const& identifier) -> Bytecode::CodeGenerationErrorOr<void> {
  944. auto interned_index = generator.intern_identifier(identifier->string());
  945. generator.emit<Bytecode::Op::SetVariable>(interned_index, initialization_mode);
  946. return {};
  947. },
  948. [&](NonnullRefPtr<BindingPattern const> const& pattern) -> Bytecode::CodeGenerationErrorOr<void> {
  949. // Store the accumulator value in a permanent register
  950. auto target_reg = generator.allocate_register();
  951. generator.emit<Bytecode::Op::Store>(target_reg);
  952. return generate_binding_pattern_bytecode(generator, pattern, initialization_mode, target_reg);
  953. },
  954. [&](NonnullRefPtr<MemberExpression const> const& expr) -> Bytecode::CodeGenerationErrorOr<void> {
  955. return generator.emit_store_to_reference(*expr);
  956. });
  957. };
  958. for (auto& [name, alias, initializer, is_rest] : pattern.entries) {
  959. VERIFY(name.has<Empty>());
  960. if (is_rest) {
  961. VERIFY(!initializer);
  962. if (first) {
  963. // The iterator has not been called, and is thus known to be not exhausted
  964. generator.emit<Bytecode::Op::Load>(iterator_reg);
  965. generator.emit<Bytecode::Op::IteratorToArray>();
  966. } else {
  967. auto& if_exhausted_block = generator.make_block();
  968. auto& if_not_exhausted_block = generator.make_block();
  969. auto& continuation_block = generator.make_block();
  970. generator.emit<Bytecode::Op::Load>(is_iterator_exhausted_register);
  971. generator.emit<Bytecode::Op::JumpConditional>().set_targets(
  972. Bytecode::Label { if_exhausted_block },
  973. Bytecode::Label { if_not_exhausted_block });
  974. generator.switch_to_basic_block(if_exhausted_block);
  975. generator.emit<Bytecode::Op::NewArray>();
  976. generator.emit<Bytecode::Op::Jump>().set_targets(
  977. Bytecode::Label { continuation_block },
  978. {});
  979. generator.switch_to_basic_block(if_not_exhausted_block);
  980. generator.emit<Bytecode::Op::Load>(iterator_reg);
  981. generator.emit<Bytecode::Op::IteratorToArray>();
  982. generator.emit<Bytecode::Op::Jump>().set_targets(
  983. Bytecode::Label { continuation_block },
  984. {});
  985. generator.switch_to_basic_block(continuation_block);
  986. }
  987. return assign_accumulator_to_alias(alias);
  988. }
  989. // In the first iteration of the loop, a few things are true which can save
  990. // us some bytecode:
  991. // - the iterator result is still in the accumulator, so we can avoid a load
  992. // - the iterator is not yet exhausted, which can save us a jump and some
  993. // creation
  994. auto& iterator_is_exhausted_block = generator.make_block();
  995. if (!first) {
  996. auto& iterator_is_not_exhausted_block = generator.make_block();
  997. generator.emit<Bytecode::Op::Load>(is_iterator_exhausted_register);
  998. generator.emit<Bytecode::Op::JumpConditional>().set_targets(
  999. Bytecode::Label { iterator_is_exhausted_block },
  1000. Bytecode::Label { iterator_is_not_exhausted_block });
  1001. generator.switch_to_basic_block(iterator_is_not_exhausted_block);
  1002. generator.emit<Bytecode::Op::Load>(iterator_reg);
  1003. }
  1004. generator.emit<Bytecode::Op::IteratorNext>();
  1005. generator.emit<Bytecode::Op::Store>(temp_iterator_result_reg);
  1006. generator.emit<Bytecode::Op::IteratorResultDone>();
  1007. generator.emit<Bytecode::Op::Store>(is_iterator_exhausted_register);
  1008. // We still have to check for exhaustion here. If the iterator is exhausted,
  1009. // we need to bail before trying to get the value
  1010. auto& no_bail_block = generator.make_block();
  1011. generator.emit<Bytecode::Op::JumpConditional>().set_targets(
  1012. Bytecode::Label { iterator_is_exhausted_block },
  1013. Bytecode::Label { no_bail_block });
  1014. generator.switch_to_basic_block(no_bail_block);
  1015. // Get the next value in the iterator
  1016. generator.emit<Bytecode::Op::Load>(temp_iterator_result_reg);
  1017. generator.emit<Bytecode::Op::IteratorResultValue>();
  1018. auto& create_binding_block = generator.make_block();
  1019. generator.emit<Bytecode::Op::Jump>().set_targets(
  1020. Bytecode::Label { create_binding_block },
  1021. {});
  1022. // The iterator is exhausted, so we just load undefined and continue binding
  1023. generator.switch_to_basic_block(iterator_is_exhausted_block);
  1024. generator.emit<Bytecode::Op::LoadImmediate>(js_undefined());
  1025. generator.emit<Bytecode::Op::Jump>().set_targets(
  1026. Bytecode::Label { create_binding_block },
  1027. {});
  1028. // Create the actual binding. The value which this entry must bind is now in the
  1029. // accumulator. We can proceed, processing the alias as a nested destructuring
  1030. // pattern if necessary.
  1031. generator.switch_to_basic_block(create_binding_block);
  1032. if (initializer) {
  1033. auto& value_is_undefined_block = generator.make_block();
  1034. auto& value_is_not_undefined_block = generator.make_block();
  1035. generator.emit<Bytecode::Op::JumpUndefined>().set_targets(
  1036. Bytecode::Label { value_is_undefined_block },
  1037. Bytecode::Label { value_is_not_undefined_block });
  1038. generator.switch_to_basic_block(value_is_undefined_block);
  1039. TRY(initializer->generate_bytecode(generator));
  1040. generator.emit<Bytecode::Op::Jump>(Bytecode::Label { value_is_not_undefined_block });
  1041. generator.switch_to_basic_block(value_is_not_undefined_block);
  1042. }
  1043. TRY(assign_accumulator_to_alias(alias));
  1044. first = false;
  1045. }
  1046. return {};
  1047. }
  1048. 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)
  1049. {
  1050. if (pattern.kind == BindingPattern::Kind::Object)
  1051. return generate_object_binding_pattern_bytecode(generator, pattern, initialization_mode, value_reg);
  1052. return generate_array_binding_pattern_bytecode(generator, pattern, initialization_mode, value_reg);
  1053. }
  1054. static Bytecode::CodeGenerationErrorOr<void> assign_accumulator_to_variable_declarator(Bytecode::Generator& generator, VariableDeclarator const& declarator, VariableDeclaration const& declaration)
  1055. {
  1056. auto initialization_mode = declaration.is_lexical_declaration() ? Bytecode::Op::SetVariable::InitializationMode::Initialize : Bytecode::Op::SetVariable::InitializationMode::Set;
  1057. return declarator.target().visit(
  1058. [&](NonnullRefPtr<Identifier const> const& id) -> Bytecode::CodeGenerationErrorOr<void> {
  1059. generator.emit<Bytecode::Op::SetVariable>(generator.intern_identifier(id->string()), initialization_mode);
  1060. return {};
  1061. },
  1062. [&](NonnullRefPtr<BindingPattern const> const& pattern) -> Bytecode::CodeGenerationErrorOr<void> {
  1063. auto value_register = generator.allocate_register();
  1064. generator.emit<Bytecode::Op::Store>(value_register);
  1065. return generate_binding_pattern_bytecode(generator, pattern, initialization_mode, value_register);
  1066. });
  1067. }
  1068. Bytecode::CodeGenerationErrorOr<void> VariableDeclaration::generate_bytecode(Bytecode::Generator& generator) const
  1069. {
  1070. for (auto& declarator : m_declarations) {
  1071. if (declarator->init())
  1072. TRY(declarator->init()->generate_bytecode(generator));
  1073. else
  1074. generator.emit<Bytecode::Op::LoadImmediate>(js_undefined());
  1075. TRY(assign_accumulator_to_variable_declarator(generator, declarator, *this));
  1076. }
  1077. return {};
  1078. }
  1079. Bytecode::CodeGenerationErrorOr<void> CallExpression::generate_bytecode(Bytecode::Generator& generator) const
  1080. {
  1081. auto callee_reg = generator.allocate_register();
  1082. auto this_reg = generator.allocate_register();
  1083. generator.emit<Bytecode::Op::LoadImmediate>(js_undefined());
  1084. generator.emit<Bytecode::Op::Store>(this_reg);
  1085. if (is<NewExpression>(this)) {
  1086. TRY(m_callee->generate_bytecode(generator));
  1087. generator.emit<Bytecode::Op::Store>(callee_reg);
  1088. } else if (is<MemberExpression>(*m_callee)) {
  1089. auto& member_expression = static_cast<MemberExpression const&>(*m_callee);
  1090. // https://tc39.es/ecma262/#sec-super-keyword-runtime-semantics-evaluation
  1091. if (is<SuperExpression>(member_expression.object())) {
  1092. // 1. Let env be GetThisEnvironment().
  1093. // 2. Let actualThis be ? env.GetThisBinding().
  1094. generator.emit<Bytecode::Op::ResolveThisBinding>();
  1095. generator.emit<Bytecode::Op::Store>(this_reg);
  1096. Optional<Bytecode::Register> computed_property_value_register;
  1097. if (member_expression.is_computed()) {
  1098. // SuperProperty : super [ Expression ]
  1099. // 3. Let propertyNameReference be ? Evaluation of Expression.
  1100. // 4. Let propertyNameValue be ? GetValue(propertyNameReference).
  1101. TRY(member_expression.property().generate_bytecode(generator));
  1102. computed_property_value_register = generator.allocate_register();
  1103. generator.emit<Bytecode::Op::Store>(*computed_property_value_register);
  1104. }
  1105. // 5/7. Return ? MakeSuperPropertyReference(actualThis, propertyKey, strict).
  1106. // https://tc39.es/ecma262/#sec-makesuperpropertyreference
  1107. // 1. Let env be GetThisEnvironment().
  1108. // 2. Assert: env.HasSuperBinding() is true.
  1109. // 3. Let baseValue be ? env.GetSuperBase().
  1110. generator.emit<Bytecode::Op::ResolveSuperBase>();
  1111. // 4. Return the Reference Record { [[Base]]: baseValue, [[ReferencedName]]: propertyKey, [[Strict]]: strict, [[ThisValue]]: actualThis }.
  1112. if (computed_property_value_register.has_value()) {
  1113. // 5. Let propertyKey be ? ToPropertyKey(propertyNameValue).
  1114. // FIXME: This does ToPropertyKey out of order, which is observable by Symbol.toPrimitive!
  1115. generator.emit<Bytecode::Op::GetByValue>(*computed_property_value_register);
  1116. } else {
  1117. // 3. Let propertyKey be StringValue of IdentifierName.
  1118. auto identifier_table_ref = generator.intern_identifier(verify_cast<Identifier>(member_expression.property()).string());
  1119. generator.emit<Bytecode::Op::GetById>(identifier_table_ref);
  1120. }
  1121. } else {
  1122. TRY(member_expression.object().generate_bytecode(generator));
  1123. generator.emit<Bytecode::Op::Store>(this_reg);
  1124. if (member_expression.is_computed()) {
  1125. TRY(member_expression.property().generate_bytecode(generator));
  1126. generator.emit<Bytecode::Op::GetByValue>(this_reg);
  1127. } else {
  1128. auto identifier_table_ref = [&] {
  1129. if (is<PrivateIdentifier>(member_expression.property()))
  1130. return generator.intern_identifier(verify_cast<PrivateIdentifier>(member_expression.property()).string());
  1131. return generator.intern_identifier(verify_cast<Identifier>(member_expression.property()).string());
  1132. }();
  1133. generator.emit<Bytecode::Op::GetById>(identifier_table_ref);
  1134. }
  1135. }
  1136. generator.emit<Bytecode::Op::Store>(callee_reg);
  1137. } else {
  1138. // FIXME: this = global object in sloppy mode.
  1139. TRY(m_callee->generate_bytecode(generator));
  1140. generator.emit<Bytecode::Op::Store>(callee_reg);
  1141. }
  1142. TRY(arguments_to_array_for_call(generator, arguments()));
  1143. Bytecode::Op::Call::CallType call_type;
  1144. if (is<NewExpression>(*this)) {
  1145. call_type = Bytecode::Op::Call::CallType::Construct;
  1146. } else if (m_callee->is_identifier() && static_cast<Identifier const&>(*m_callee).string() == "eval"sv) {
  1147. call_type = Bytecode::Op::Call::CallType::DirectEval;
  1148. } else {
  1149. call_type = Bytecode::Op::Call::CallType::Call;
  1150. }
  1151. Optional<Bytecode::StringTableIndex> expression_string_index;
  1152. if (auto expression_string = this->expression_string(); expression_string.has_value())
  1153. expression_string_index = generator.intern_string(expression_string.release_value());
  1154. generator.emit<Bytecode::Op::Call>(call_type, callee_reg, this_reg, expression_string_index);
  1155. return {};
  1156. }
  1157. Bytecode::CodeGenerationErrorOr<void> ReturnStatement::generate_bytecode(Bytecode::Generator& generator) const
  1158. {
  1159. if (m_argument)
  1160. TRY(m_argument->generate_bytecode(generator));
  1161. else
  1162. generator.emit<Bytecode::Op::LoadImmediate>(js_undefined());
  1163. if (generator.is_in_generator_or_async_function()) {
  1164. generator.perform_needed_unwinds<Bytecode::Op::Yield>();
  1165. generator.emit<Bytecode::Op::Yield>(nullptr);
  1166. } else {
  1167. generator.perform_needed_unwinds<Bytecode::Op::Return>();
  1168. generator.emit<Bytecode::Op::Return>();
  1169. }
  1170. return {};
  1171. }
  1172. Bytecode::CodeGenerationErrorOr<void> YieldExpression::generate_bytecode(Bytecode::Generator& generator) const
  1173. {
  1174. VERIFY(generator.is_in_generator_function());
  1175. auto received_completion_register = generator.allocate_register();
  1176. auto received_completion_type_register = generator.allocate_register();
  1177. auto received_completion_value_register = generator.allocate_register();
  1178. auto type_identifier = generator.intern_identifier("type");
  1179. auto value_identifier = generator.intern_identifier("value");
  1180. auto get_received_completion_type_and_value = [&]() {
  1181. // The accumulator is set to an object, for example: { "type": 1 (normal), value: 1337 }
  1182. generator.emit<Bytecode::Op::Store>(received_completion_register);
  1183. generator.emit<Bytecode::Op::GetById>(type_identifier);
  1184. generator.emit<Bytecode::Op::Store>(received_completion_type_register);
  1185. generator.emit<Bytecode::Op::Load>(received_completion_register);
  1186. generator.emit<Bytecode::Op::GetById>(value_identifier);
  1187. generator.emit<Bytecode::Op::Store>(received_completion_value_register);
  1188. };
  1189. if (m_is_yield_from) {
  1190. // 15.5.5 Runtime Semantics: Evaluation, https://tc39.es/ecma262/#sec-generator-function-definitions-runtime-semantics-evaluation
  1191. // FIXME: 1. Let generatorKind be GetGeneratorKind().
  1192. // 2. Let exprRef be ? Evaluation of AssignmentExpression.
  1193. // 3. Let value be ? GetValue(exprRef).
  1194. VERIFY(m_argument);
  1195. TRY(m_argument->generate_bytecode(generator));
  1196. // 4. Let iteratorRecord be ? GetIterator(value, generatorKind).
  1197. // FIXME: Consider generatorKind.
  1198. auto iterator_record_register = generator.allocate_register();
  1199. generator.emit<Bytecode::Op::GetIterator>();
  1200. generator.emit<Bytecode::Op::Store>(iterator_record_register);
  1201. // 5. Let iterator be iteratorRecord.[[Iterator]].
  1202. auto iterator_register = generator.allocate_register();
  1203. auto iterator_identifier = generator.intern_identifier("iterator");
  1204. generator.emit<Bytecode::Op::GetById>(iterator_identifier);
  1205. generator.emit<Bytecode::Op::Store>(iterator_register);
  1206. // Cache iteratorRecord.[[NextMethod]] for use in step 7.a.i.
  1207. auto next_method_register = generator.allocate_register();
  1208. auto next_method_identifier = generator.intern_identifier("next");
  1209. generator.emit<Bytecode::Op::Load>(iterator_record_register);
  1210. generator.emit<Bytecode::Op::GetById>(next_method_identifier);
  1211. generator.emit<Bytecode::Op::Store>(next_method_register);
  1212. // 6. Let received be NormalCompletion(undefined).
  1213. // See get_received_completion_type_and_value above.
  1214. generator.emit<Bytecode::Op::LoadImmediate>(Value(to_underlying(Completion::Type::Normal)));
  1215. generator.emit<Bytecode::Op::Store>(received_completion_type_register);
  1216. generator.emit<Bytecode::Op::LoadImmediate>(js_undefined());
  1217. generator.emit<Bytecode::Op::Store>(received_completion_value_register);
  1218. // 7. Repeat,
  1219. auto& loop_block = generator.make_block();
  1220. auto& continuation_block = generator.make_block();
  1221. auto& loop_end_block = generator.make_block();
  1222. generator.emit<Bytecode::Op::Jump>(Bytecode::Label { loop_block });
  1223. generator.switch_to_basic_block(loop_block);
  1224. // a. If received.[[Type]] is normal, then
  1225. auto& type_is_normal_block = generator.make_block();
  1226. auto& is_type_throw_block = generator.make_block();
  1227. generator.emit<Bytecode::Op::LoadImmediate>(Value(to_underlying(Completion::Type::Normal)));
  1228. generator.emit<Bytecode::Op::StrictlyEquals>(received_completion_type_register);
  1229. generator.emit<Bytecode::Op::JumpConditional>(
  1230. Bytecode::Label { type_is_normal_block },
  1231. Bytecode::Label { is_type_throw_block });
  1232. generator.switch_to_basic_block(type_is_normal_block);
  1233. // i. Let innerResult be ? Call(iteratorRecord.[[NextMethod]], iteratorRecord.[[Iterator]], « received.[[Value]] »).
  1234. generator.emit_with_extra_register_slots<Bytecode::Op::NewArray>(2, AK::Array { received_completion_value_register, received_completion_value_register });
  1235. generator.emit<Bytecode::Op::Call>(Bytecode::Op::Call::CallType::Call, next_method_register, iterator_register);
  1236. // FIXME: ii. If generatorKind is async, set innerResult to ? Await(innerResult).
  1237. // iii. If innerResult is not an Object, throw a TypeError exception.
  1238. generator.emit<Bytecode::Op::ThrowIfNotObject>();
  1239. auto inner_result_register = generator.allocate_register();
  1240. generator.emit<Bytecode::Op::Store>(inner_result_register);
  1241. // iv. Let done be ? IteratorComplete(innerResult).
  1242. generator.emit<Bytecode::Op::IteratorResultDone>();
  1243. // v. If done is true, then
  1244. auto& type_is_normal_done_block = generator.make_block();
  1245. auto& type_is_normal_not_done_block = generator.make_block();
  1246. generator.emit<Bytecode::Op::JumpConditional>(
  1247. Bytecode::Label { type_is_normal_done_block },
  1248. Bytecode::Label { type_is_normal_not_done_block });
  1249. generator.switch_to_basic_block(type_is_normal_done_block);
  1250. // 1. Return ? IteratorValue(innerResult).
  1251. generator.emit<Bytecode::Op::Load>(inner_result_register);
  1252. generator.emit<Bytecode::Op::IteratorResultValue>();
  1253. generator.emit<Bytecode::Op::Jump>(Bytecode::Label { loop_end_block });
  1254. generator.switch_to_basic_block(type_is_normal_not_done_block);
  1255. // FIXME: vi. If generatorKind is async, set received to Completion(AsyncGeneratorYield(? IteratorValue(innerResult))).
  1256. // vii. Else, set received to Completion(GeneratorYield(innerResult)).
  1257. // FIXME: Else,
  1258. generator.emit<Bytecode::Op::Load>(inner_result_register);
  1259. // FIXME: Yield currently only accepts a Value, not an object conforming to the IteratorResult interface, so we have to do an observable lookup of `value` here.
  1260. generator.emit<Bytecode::Op::IteratorResultValue>();
  1261. generator.emit<Bytecode::Op::Yield>(Bytecode::Label { continuation_block });
  1262. // b. Else if received.[[Type]] is throw, then
  1263. generator.switch_to_basic_block(is_type_throw_block);
  1264. auto& type_is_throw_block = generator.make_block();
  1265. auto& type_is_return_block = generator.make_block();
  1266. generator.emit<Bytecode::Op::LoadImmediate>(Value(to_underlying(Completion::Type::Throw)));
  1267. generator.emit<Bytecode::Op::StrictlyEquals>(received_completion_type_register);
  1268. generator.emit<Bytecode::Op::JumpConditional>(
  1269. Bytecode::Label { type_is_throw_block },
  1270. Bytecode::Label { type_is_return_block });
  1271. generator.switch_to_basic_block(type_is_throw_block);
  1272. // i. Let throw be ? GetMethod(iterator, "throw").
  1273. auto throw_method_register = generator.allocate_register();
  1274. auto throw_identifier = generator.intern_identifier("throw");
  1275. generator.emit<Bytecode::Op::Load>(iterator_register);
  1276. generator.emit<Bytecode::Op::GetMethod>(throw_identifier);
  1277. generator.emit<Bytecode::Op::Store>(throw_method_register);
  1278. // ii. If throw is not undefined, then
  1279. auto& throw_method_is_defined_block = generator.make_block();
  1280. auto& throw_method_is_undefined_block = generator.make_block();
  1281. generator.emit<Bytecode::Op::LoadImmediate>(js_undefined());
  1282. generator.emit<Bytecode::Op::StrictlyInequals>(throw_method_register);
  1283. generator.emit<Bytecode::Op::JumpConditional>(
  1284. Bytecode::Label { throw_method_is_defined_block },
  1285. Bytecode::Label { throw_method_is_undefined_block });
  1286. generator.switch_to_basic_block(throw_method_is_defined_block);
  1287. // 1. Let innerResult be ? Call(throw, iterator, « received.[[Value]] »).
  1288. generator.emit_with_extra_register_slots<Bytecode::Op::NewArray>(2, AK::Array { received_completion_value_register, received_completion_value_register });
  1289. generator.emit<Bytecode::Op::Call>(Bytecode::Op::Call::CallType::Call, throw_method_register, iterator_register);
  1290. // FIXME: 2. If generatorKind is async, set innerResult to ? Await(innerResult).
  1291. // 3. NOTE: Exceptions from the inner iterator throw method are propagated. Normal completions from an inner throw method are processed similarly to an inner next.
  1292. // 4. If innerResult is not an Object, throw a TypeError exception.
  1293. generator.emit<Bytecode::Op::ThrowIfNotObject>();
  1294. generator.emit<Bytecode::Op::Store>(inner_result_register);
  1295. // 5. Let done be ? IteratorComplete(innerResult).
  1296. generator.emit<Bytecode::Op::IteratorResultDone>();
  1297. // 6. If done is true, then
  1298. auto& type_is_throw_done_block = generator.make_block();
  1299. auto& type_is_throw_not_done_block = generator.make_block();
  1300. generator.emit<Bytecode::Op::JumpConditional>(
  1301. Bytecode::Label { type_is_throw_done_block },
  1302. Bytecode::Label { type_is_throw_not_done_block });
  1303. generator.switch_to_basic_block(type_is_throw_done_block);
  1304. // a. Return ? IteratorValue(innerResult).
  1305. generator.emit<Bytecode::Op::Load>(inner_result_register);
  1306. generator.emit<Bytecode::Op::IteratorResultValue>();
  1307. generator.emit<Bytecode::Op::Jump>(Bytecode::Label { loop_end_block });
  1308. generator.switch_to_basic_block(type_is_throw_not_done_block);
  1309. // FIXME: 7. If generatorKind is async, set received to Completion(AsyncGeneratorYield(? IteratorValue(innerResult))).
  1310. // 8. Else, set received to Completion(GeneratorYield(innerResult)).
  1311. // FIXME: Else,
  1312. generator.emit<Bytecode::Op::Load>(inner_result_register);
  1313. // FIXME: Yield currently only accepts a Value, not an object conforming to the IteratorResult interface, so we have to do an observable lookup of `value` here.
  1314. generator.emit<Bytecode::Op::IteratorResultValue>();
  1315. generator.emit<Bytecode::Op::Yield>(Bytecode::Label { continuation_block });
  1316. generator.switch_to_basic_block(throw_method_is_undefined_block);
  1317. // 1. NOTE: If iterator does not have a throw method, this throw is going to terminate the yield* loop. But first we need to give iterator a chance to clean up.
  1318. // 2. Let closeCompletion be Completion Record { [[Type]]: normal, [[Value]]: empty, [[Target]]: empty }.
  1319. // FIXME: 3. If generatorKind is async, perform ? AsyncIteratorClose(iteratorRecord, closeCompletion).
  1320. // 4. Else, perform ? IteratorClose(iteratorRecord, closeCompletion).
  1321. // FIXME: Else,
  1322. generator.emit<Bytecode::Op::Load>(iterator_record_register);
  1323. generator.emit<Bytecode::Op::IteratorClose>(Completion::Type::Normal, Optional<Value> {});
  1324. // 5. NOTE: The next step throws a TypeError to indicate that there was a yield* protocol violation: iterator does not have a throw method.
  1325. // 6. Throw a TypeError exception.
  1326. generator.emit<Bytecode::Op::NewTypeError>(generator.intern_string(ErrorType::YieldFromIteratorMissingThrowMethod.message()));
  1327. generator.perform_needed_unwinds<Bytecode::Op::Throw>();
  1328. generator.emit<Bytecode::Op::Throw>();
  1329. // c. Else,
  1330. // i. Assert: received.[[Type]] is return.
  1331. generator.switch_to_basic_block(type_is_return_block);
  1332. // ii. Let return be ? GetMethod(iterator, "return").
  1333. auto return_method_register = generator.allocate_register();
  1334. auto return_identifier = generator.intern_identifier("return");
  1335. generator.emit<Bytecode::Op::Load>(iterator_register);
  1336. generator.emit<Bytecode::Op::GetMethod>(return_identifier);
  1337. generator.emit<Bytecode::Op::Store>(return_method_register);
  1338. // iii. If return is undefined, then
  1339. auto& return_is_undefined_block = generator.make_block();
  1340. auto& return_is_defined_block = generator.make_block();
  1341. generator.emit<Bytecode::Op::LoadImmediate>(js_undefined());
  1342. generator.emit<Bytecode::Op::StrictlyEquals>(return_method_register);
  1343. generator.emit<Bytecode::Op::JumpConditional>(
  1344. Bytecode::Label { return_is_undefined_block },
  1345. Bytecode::Label { return_is_defined_block });
  1346. generator.switch_to_basic_block(return_is_undefined_block);
  1347. // FIXME: 1. If generatorKind is async, set received.[[Value]] to ? Await(received.[[Value]]).
  1348. // 2. Return ? received.
  1349. // NOTE: This will always be a return completion.
  1350. generator.emit<Bytecode::Op::Load>(received_completion_value_register);
  1351. generator.perform_needed_unwinds<Bytecode::Op::Yield>();
  1352. generator.emit<Bytecode::Op::Yield>(nullptr);
  1353. generator.switch_to_basic_block(return_is_defined_block);
  1354. // iv. Let innerReturnResult be ? Call(return, iterator, « received.[[Value]] »).
  1355. generator.emit_with_extra_register_slots<Bytecode::Op::NewArray>(2, AK::Array { received_completion_value_register, received_completion_value_register });
  1356. generator.emit<Bytecode::Op::Call>(Bytecode::Op::Call::CallType::Call, return_method_register, iterator_register);
  1357. // FIXME: v. If generatorKind is async, set innerReturnResult to ? Await(innerReturnResult).
  1358. // vi. If innerReturnResult is not an Object, throw a TypeError exception.
  1359. generator.emit<Bytecode::Op::ThrowIfNotObject>();
  1360. auto inner_return_result_register = generator.allocate_register();
  1361. generator.emit<Bytecode::Op::Store>(inner_return_result_register);
  1362. // vii. Let done be ? IteratorComplete(innerReturnResult).
  1363. generator.emit<Bytecode::Op::IteratorResultDone>();
  1364. // viii. If done is true, then
  1365. auto& type_is_return_done_block = generator.make_block();
  1366. auto& type_is_return_not_done_block = generator.make_block();
  1367. generator.emit<Bytecode::Op::JumpConditional>(
  1368. Bytecode::Label { type_is_return_done_block },
  1369. Bytecode::Label { type_is_return_not_done_block });
  1370. generator.switch_to_basic_block(type_is_return_done_block);
  1371. // 1. Let value be ? IteratorValue(innerReturnResult).
  1372. generator.emit<Bytecode::Op::Load>(inner_result_register);
  1373. generator.emit<Bytecode::Op::IteratorResultValue>();
  1374. // 2. Return Completion Record { [[Type]]: return, [[Value]]: value, [[Target]]: empty }.
  1375. generator.perform_needed_unwinds<Bytecode::Op::Yield>();
  1376. generator.emit<Bytecode::Op::Yield>(nullptr);
  1377. generator.switch_to_basic_block(type_is_return_not_done_block);
  1378. // FIXME: ix. If generatorKind is async, set received to Completion(AsyncGeneratorYield(? IteratorValue(innerReturnResult))).
  1379. // x. Else, set received to Completion(GeneratorYield(innerReturnResult)).
  1380. // FIXME: Else,
  1381. generator.emit<Bytecode::Op::Load>(inner_return_result_register);
  1382. // FIXME: Yield currently only accepts a Value, not an object conforming to the IteratorResult interface, so we have to do an observable lookup of `value` here.
  1383. generator.emit<Bytecode::Op::IteratorResultValue>();
  1384. generator.emit<Bytecode::Op::Yield>(Bytecode::Label { continuation_block });
  1385. generator.switch_to_basic_block(continuation_block);
  1386. get_received_completion_type_and_value();
  1387. generator.emit<Bytecode::Op::Jump>(Bytecode::Label { loop_block });
  1388. generator.switch_to_basic_block(loop_end_block);
  1389. return {};
  1390. }
  1391. if (m_argument)
  1392. TRY(m_argument->generate_bytecode(generator));
  1393. else
  1394. generator.emit<Bytecode::Op::LoadImmediate>(js_undefined());
  1395. auto& continuation_block = generator.make_block();
  1396. generator.emit<Bytecode::Op::Yield>(Bytecode::Label { continuation_block });
  1397. generator.switch_to_basic_block(continuation_block);
  1398. get_received_completion_type_and_value();
  1399. auto& normal_completion_continuation_block = generator.make_block();
  1400. auto& throw_completion_continuation_block = generator.make_block();
  1401. generator.emit<Bytecode::Op::LoadImmediate>(Value(to_underlying(Completion::Type::Normal)));
  1402. generator.emit<Bytecode::Op::StrictlyEquals>(received_completion_type_register);
  1403. generator.emit<Bytecode::Op::JumpConditional>(
  1404. Bytecode::Label { normal_completion_continuation_block },
  1405. Bytecode::Label { throw_completion_continuation_block });
  1406. auto& throw_value_block = generator.make_block();
  1407. auto& return_value_block = generator.make_block();
  1408. generator.switch_to_basic_block(throw_completion_continuation_block);
  1409. generator.emit<Bytecode::Op::LoadImmediate>(Value(to_underlying(Completion::Type::Throw)));
  1410. generator.emit<Bytecode::Op::StrictlyEquals>(received_completion_type_register);
  1411. // If type is not equal to "throw" or "normal", assume it's "return".
  1412. generator.emit<Bytecode::Op::JumpConditional>(
  1413. Bytecode::Label { throw_value_block },
  1414. Bytecode::Label { return_value_block });
  1415. generator.switch_to_basic_block(throw_value_block);
  1416. generator.emit<Bytecode::Op::Load>(received_completion_value_register);
  1417. generator.perform_needed_unwinds<Bytecode::Op::Throw>();
  1418. generator.emit<Bytecode::Op::Throw>();
  1419. generator.switch_to_basic_block(return_value_block);
  1420. generator.emit<Bytecode::Op::Load>(received_completion_value_register);
  1421. generator.perform_needed_unwinds<Bytecode::Op::Yield>();
  1422. generator.emit<Bytecode::Op::Yield>(nullptr);
  1423. generator.switch_to_basic_block(normal_completion_continuation_block);
  1424. generator.emit<Bytecode::Op::Load>(received_completion_value_register);
  1425. return {};
  1426. }
  1427. Bytecode::CodeGenerationErrorOr<void> IfStatement::generate_bytecode(Bytecode::Generator& generator) const
  1428. {
  1429. // test
  1430. // jump if_true (true) true (false) false
  1431. // true
  1432. // jump always (true) end
  1433. // false
  1434. // jump always (true) end
  1435. // end
  1436. auto& true_block = generator.make_block();
  1437. auto& false_block = generator.make_block();
  1438. TRY(m_predicate->generate_bytecode(generator));
  1439. generator.emit<Bytecode::Op::JumpConditional>().set_targets(
  1440. Bytecode::Label { true_block },
  1441. Bytecode::Label { false_block });
  1442. Bytecode::Op::Jump* true_block_jump { nullptr };
  1443. generator.switch_to_basic_block(true_block);
  1444. generator.emit<Bytecode::Op::LoadImmediate>(js_undefined());
  1445. TRY(m_consequent->generate_bytecode(generator));
  1446. if (!generator.is_current_block_terminated())
  1447. true_block_jump = &generator.emit<Bytecode::Op::Jump>();
  1448. generator.switch_to_basic_block(false_block);
  1449. auto& end_block = generator.make_block();
  1450. generator.emit<Bytecode::Op::LoadImmediate>(js_undefined());
  1451. if (m_alternate)
  1452. TRY(m_alternate->generate_bytecode(generator));
  1453. if (!generator.is_current_block_terminated())
  1454. generator.emit<Bytecode::Op::Jump>().set_targets(Bytecode::Label { end_block }, {});
  1455. if (true_block_jump)
  1456. true_block_jump->set_targets(Bytecode::Label { end_block }, {});
  1457. generator.switch_to_basic_block(end_block);
  1458. return {};
  1459. }
  1460. Bytecode::CodeGenerationErrorOr<void> ContinueStatement::generate_bytecode(Bytecode::Generator& generator) const
  1461. {
  1462. // FIXME: Handle finally blocks in a graceful manner
  1463. // We need to execute the finally block, but tell it to resume
  1464. // execution at the designated block
  1465. if (m_target_label.is_null()) {
  1466. generator.generate_continue();
  1467. return {};
  1468. }
  1469. generator.generate_continue(m_target_label);
  1470. return {};
  1471. }
  1472. Bytecode::CodeGenerationErrorOr<void> DebuggerStatement::generate_bytecode(Bytecode::Generator&) const
  1473. {
  1474. return {};
  1475. }
  1476. Bytecode::CodeGenerationErrorOr<void> ConditionalExpression::generate_bytecode(Bytecode::Generator& generator) const
  1477. {
  1478. // test
  1479. // jump if_true (true) true (false) false
  1480. // true
  1481. // jump always (true) end
  1482. // false
  1483. // jump always (true) end
  1484. // end
  1485. auto& true_block = generator.make_block();
  1486. auto& false_block = generator.make_block();
  1487. auto& end_block = generator.make_block();
  1488. TRY(m_test->generate_bytecode(generator));
  1489. generator.emit<Bytecode::Op::JumpConditional>().set_targets(
  1490. Bytecode::Label { true_block },
  1491. Bytecode::Label { false_block });
  1492. generator.switch_to_basic_block(true_block);
  1493. TRY(m_consequent->generate_bytecode(generator));
  1494. generator.emit<Bytecode::Op::Jump>().set_targets(
  1495. Bytecode::Label { end_block },
  1496. {});
  1497. generator.switch_to_basic_block(false_block);
  1498. TRY(m_alternate->generate_bytecode(generator));
  1499. generator.emit<Bytecode::Op::Jump>().set_targets(
  1500. Bytecode::Label { end_block },
  1501. {});
  1502. generator.switch_to_basic_block(end_block);
  1503. return {};
  1504. }
  1505. Bytecode::CodeGenerationErrorOr<void> SequenceExpression::generate_bytecode(Bytecode::Generator& generator) const
  1506. {
  1507. for (auto& expression : m_expressions)
  1508. TRY(expression->generate_bytecode(generator));
  1509. return {};
  1510. }
  1511. Bytecode::CodeGenerationErrorOr<void> TemplateLiteral::generate_bytecode(Bytecode::Generator& generator) const
  1512. {
  1513. auto string_reg = generator.allocate_register();
  1514. for (size_t i = 0; i < m_expressions.size(); i++) {
  1515. TRY(m_expressions[i]->generate_bytecode(generator));
  1516. if (i == 0) {
  1517. generator.emit<Bytecode::Op::Store>(string_reg);
  1518. } else {
  1519. generator.emit<Bytecode::Op::ConcatString>(string_reg);
  1520. }
  1521. }
  1522. generator.emit<Bytecode::Op::Load>(string_reg);
  1523. return {};
  1524. }
  1525. Bytecode::CodeGenerationErrorOr<void> TaggedTemplateLiteral::generate_bytecode(Bytecode::Generator& generator) const
  1526. {
  1527. TRY(m_tag->generate_bytecode(generator));
  1528. auto tag_reg = generator.allocate_register();
  1529. generator.emit<Bytecode::Op::Store>(tag_reg);
  1530. // FIXME: We only need to record the first and last register,
  1531. // due to packing everything in an array, same goes for argument_regs
  1532. Vector<Bytecode::Register> string_regs;
  1533. auto& expressions = m_template_literal->expressions();
  1534. for (size_t i = 0; i < expressions.size(); ++i) {
  1535. if (i % 2 != 0)
  1536. continue;
  1537. string_regs.append(generator.allocate_register());
  1538. }
  1539. size_t reg_index = 0;
  1540. for (size_t i = 0; i < expressions.size(); ++i) {
  1541. if (i % 2 != 0)
  1542. continue;
  1543. TRY(expressions[i]->generate_bytecode(generator));
  1544. auto string_reg = string_regs[reg_index++];
  1545. generator.emit<Bytecode::Op::Store>(string_reg);
  1546. }
  1547. if (string_regs.is_empty()) {
  1548. generator.emit<Bytecode::Op::NewArray>();
  1549. } else {
  1550. generator.emit_with_extra_register_slots<Bytecode::Op::NewArray>(2u, AK::Array { string_regs.first(), string_regs.last() });
  1551. }
  1552. auto strings_reg = generator.allocate_register();
  1553. generator.emit<Bytecode::Op::Store>(strings_reg);
  1554. Vector<Bytecode::Register> argument_regs;
  1555. argument_regs.append(strings_reg);
  1556. for (size_t i = 1; i < expressions.size(); i += 2)
  1557. argument_regs.append(generator.allocate_register());
  1558. for (size_t i = 1; i < expressions.size(); i += 2) {
  1559. auto string_reg = argument_regs[1 + i / 2];
  1560. TRY(expressions[i]->generate_bytecode(generator));
  1561. generator.emit<Bytecode::Op::Store>(string_reg);
  1562. }
  1563. Vector<Bytecode::Register> raw_string_regs;
  1564. for ([[maybe_unused]] auto& raw_string : m_template_literal->raw_strings())
  1565. string_regs.append(generator.allocate_register());
  1566. reg_index = 0;
  1567. for (auto& raw_string : m_template_literal->raw_strings()) {
  1568. TRY(raw_string->generate_bytecode(generator));
  1569. auto raw_string_reg = string_regs[reg_index++];
  1570. generator.emit<Bytecode::Op::Store>(raw_string_reg);
  1571. raw_string_regs.append(raw_string_reg);
  1572. }
  1573. if (raw_string_regs.is_empty()) {
  1574. generator.emit<Bytecode::Op::NewArray>();
  1575. } else {
  1576. generator.emit_with_extra_register_slots<Bytecode::Op::NewArray>(2u, AK::Array { raw_string_regs.first(), raw_string_regs.last() });
  1577. }
  1578. auto raw_strings_reg = generator.allocate_register();
  1579. generator.emit<Bytecode::Op::Store>(raw_strings_reg);
  1580. generator.emit<Bytecode::Op::PutById>(strings_reg, generator.intern_identifier("raw"));
  1581. generator.emit<Bytecode::Op::LoadImmediate>(js_undefined());
  1582. auto this_reg = generator.allocate_register();
  1583. generator.emit<Bytecode::Op::Store>(this_reg);
  1584. if (!argument_regs.is_empty())
  1585. generator.emit_with_extra_register_slots<Bytecode::Op::NewArray>(2, AK::Array { argument_regs.first(), argument_regs.last() });
  1586. else
  1587. generator.emit<Bytecode::Op::NewArray>();
  1588. generator.emit<Bytecode::Op::Call>(Bytecode::Op::Call::CallType::Call, tag_reg, this_reg);
  1589. return {};
  1590. }
  1591. Bytecode::CodeGenerationErrorOr<void> UpdateExpression::generate_bytecode(Bytecode::Generator& generator) const
  1592. {
  1593. TRY(generator.emit_load_from_reference(*m_argument));
  1594. Optional<Bytecode::Register> previous_value_for_postfix_reg;
  1595. if (!m_prefixed) {
  1596. previous_value_for_postfix_reg = generator.allocate_register();
  1597. generator.emit<Bytecode::Op::ToNumeric>();
  1598. generator.emit<Bytecode::Op::Store>(*previous_value_for_postfix_reg);
  1599. }
  1600. if (m_op == UpdateOp::Increment)
  1601. generator.emit<Bytecode::Op::Increment>();
  1602. else
  1603. generator.emit<Bytecode::Op::Decrement>();
  1604. TRY(generator.emit_store_to_reference(*m_argument));
  1605. if (!m_prefixed)
  1606. generator.emit<Bytecode::Op::Load>(*previous_value_for_postfix_reg);
  1607. return {};
  1608. }
  1609. Bytecode::CodeGenerationErrorOr<void> ThrowStatement::generate_bytecode(Bytecode::Generator& generator) const
  1610. {
  1611. TRY(m_argument->generate_bytecode(generator));
  1612. generator.perform_needed_unwinds<Bytecode::Op::Throw>();
  1613. generator.emit<Bytecode::Op::Throw>();
  1614. return {};
  1615. }
  1616. Bytecode::CodeGenerationErrorOr<void> BreakStatement::generate_bytecode(Bytecode::Generator& generator) const
  1617. {
  1618. // FIXME: Handle finally blocks in a graceful manner
  1619. // We need to execute the finally block, but tell it to resume
  1620. // execution at the designated block
  1621. if (m_target_label.is_null()) {
  1622. generator.generate_break();
  1623. return {};
  1624. }
  1625. generator.generate_break(m_target_label);
  1626. return {};
  1627. }
  1628. Bytecode::CodeGenerationErrorOr<void> TryStatement::generate_bytecode(Bytecode::Generator& generator) const
  1629. {
  1630. auto& saved_block = generator.current_block();
  1631. Optional<Bytecode::Label> handler_target;
  1632. Optional<Bytecode::Label> finalizer_target;
  1633. Bytecode::BasicBlock* next_block { nullptr };
  1634. if (m_finalizer) {
  1635. // FIXME: See notes in Op.h->ScheduleJump
  1636. auto& finalizer_block = generator.make_block();
  1637. generator.switch_to_basic_block(finalizer_block);
  1638. generator.emit<Bytecode::Op::LeaveUnwindContext>();
  1639. TRY(m_finalizer->generate_bytecode(generator));
  1640. if (!generator.is_current_block_terminated()) {
  1641. next_block = &generator.make_block();
  1642. auto next_target = Bytecode::Label { *next_block };
  1643. generator.emit<Bytecode::Op::ContinuePendingUnwind>(next_target);
  1644. }
  1645. finalizer_target = Bytecode::Label { finalizer_block };
  1646. }
  1647. if (m_finalizer)
  1648. generator.start_boundary(Bytecode::Generator::BlockBoundaryType::ReturnToFinally);
  1649. if (m_handler) {
  1650. auto& handler_block = generator.make_block();
  1651. generator.switch_to_basic_block(handler_block);
  1652. if (!m_finalizer)
  1653. generator.emit<Bytecode::Op::LeaveUnwindContext>();
  1654. generator.begin_variable_scope();
  1655. TRY(m_handler->parameter().visit(
  1656. [&](DeprecatedFlyString const& parameter) -> Bytecode::CodeGenerationErrorOr<void> {
  1657. if (!parameter.is_empty()) {
  1658. auto parameter_identifier = generator.intern_identifier(parameter);
  1659. generator.emit<Bytecode::Op::CreateVariable>(parameter_identifier, Bytecode::Op::EnvironmentMode::Lexical, false);
  1660. generator.emit<Bytecode::Op::SetVariable>(parameter_identifier, Bytecode::Op::SetVariable::InitializationMode::Initialize);
  1661. }
  1662. return {};
  1663. },
  1664. [&](NonnullRefPtr<BindingPattern const> const&) -> Bytecode::CodeGenerationErrorOr<void> {
  1665. // FIXME: Implement this path when the above DeclarativeEnvironment issue is dealt with.
  1666. return Bytecode::CodeGenerationError {
  1667. this,
  1668. "Unimplemented catch argument: BindingPattern"sv,
  1669. };
  1670. }));
  1671. TRY(m_handler->body().generate_bytecode(generator));
  1672. handler_target = Bytecode::Label { handler_block };
  1673. generator.end_variable_scope();
  1674. if (!generator.is_current_block_terminated()) {
  1675. if (m_finalizer) {
  1676. generator.emit<Bytecode::Op::Jump>(finalizer_target);
  1677. } else {
  1678. VERIFY(!next_block);
  1679. next_block = &generator.make_block();
  1680. auto next_target = Bytecode::Label { *next_block };
  1681. generator.emit<Bytecode::Op::Jump>(next_target);
  1682. }
  1683. }
  1684. }
  1685. if (m_finalizer)
  1686. generator.end_boundary(Bytecode::Generator::BlockBoundaryType::ReturnToFinally);
  1687. auto& target_block = generator.make_block();
  1688. generator.switch_to_basic_block(saved_block);
  1689. generator.emit<Bytecode::Op::EnterUnwindContext>(Bytecode::Label { target_block }, handler_target, finalizer_target);
  1690. generator.start_boundary(Bytecode::Generator::BlockBoundaryType::Unwind);
  1691. if (m_finalizer)
  1692. generator.start_boundary(Bytecode::Generator::BlockBoundaryType::ReturnToFinally);
  1693. generator.switch_to_basic_block(target_block);
  1694. TRY(m_block->generate_bytecode(generator));
  1695. if (!generator.is_current_block_terminated()) {
  1696. if (m_finalizer) {
  1697. generator.emit<Bytecode::Op::Jump>(finalizer_target);
  1698. } else {
  1699. if (!next_block)
  1700. next_block = &generator.make_block();
  1701. generator.emit<Bytecode::Op::LeaveUnwindContext>();
  1702. generator.emit<Bytecode::Op::Jump>(Bytecode::Label { *next_block });
  1703. }
  1704. }
  1705. if (m_finalizer)
  1706. generator.end_boundary(Bytecode::Generator::BlockBoundaryType::ReturnToFinally);
  1707. generator.end_boundary(Bytecode::Generator::BlockBoundaryType::Unwind);
  1708. generator.switch_to_basic_block(next_block ? *next_block : saved_block);
  1709. return {};
  1710. }
  1711. Bytecode::CodeGenerationErrorOr<void> SwitchStatement::generate_bytecode(Bytecode::Generator& generator) const
  1712. {
  1713. return generate_labelled_evaluation(generator, {});
  1714. }
  1715. Bytecode::CodeGenerationErrorOr<void> SwitchStatement::generate_labelled_evaluation(Bytecode::Generator& generator, Vector<DeprecatedFlyString> const& label_set) const
  1716. {
  1717. auto discriminant_reg = generator.allocate_register();
  1718. TRY(m_discriminant->generate_bytecode(generator));
  1719. generator.emit<Bytecode::Op::Store>(discriminant_reg);
  1720. Vector<Bytecode::BasicBlock&> case_blocks;
  1721. Bytecode::BasicBlock* default_block { nullptr };
  1722. Bytecode::BasicBlock* next_test_block = &generator.make_block();
  1723. auto has_lexical_block = has_lexical_declarations();
  1724. // Note: This call ends up calling begin_variable_scope() if has_lexical_block is true, so we need to clean up after it at the end.
  1725. TRY(ScopeNode::generate_bytecode(generator));
  1726. generator.emit<Bytecode::Op::Jump>().set_targets(Bytecode::Label { *next_test_block }, {});
  1727. for (auto& switch_case : m_cases) {
  1728. auto& case_block = generator.make_block();
  1729. if (switch_case->test()) {
  1730. generator.switch_to_basic_block(*next_test_block);
  1731. TRY(switch_case->test()->generate_bytecode(generator));
  1732. generator.emit<Bytecode::Op::StrictlyEquals>(discriminant_reg);
  1733. next_test_block = &generator.make_block();
  1734. generator.emit<Bytecode::Op::JumpConditional>().set_targets(Bytecode::Label { case_block }, Bytecode::Label { *next_test_block });
  1735. } else {
  1736. default_block = &case_block;
  1737. }
  1738. case_blocks.append(case_block);
  1739. }
  1740. generator.switch_to_basic_block(*next_test_block);
  1741. auto& end_block = generator.make_block();
  1742. if (default_block != nullptr) {
  1743. generator.emit<Bytecode::Op::Jump>().set_targets(Bytecode::Label { *default_block }, {});
  1744. } else {
  1745. generator.emit<Bytecode::Op::LoadImmediate>(js_undefined());
  1746. generator.emit<Bytecode::Op::Jump>().set_targets(Bytecode::Label { end_block }, {});
  1747. }
  1748. auto current_block = case_blocks.begin();
  1749. generator.begin_breakable_scope(Bytecode::Label { end_block }, label_set);
  1750. for (auto& switch_case : m_cases) {
  1751. generator.switch_to_basic_block(*current_block);
  1752. generator.emit<Bytecode::Op::LoadImmediate>(js_undefined());
  1753. for (auto& statement : switch_case->children()) {
  1754. TRY(statement->generate_bytecode(generator));
  1755. if (generator.is_current_block_terminated())
  1756. break;
  1757. }
  1758. if (!generator.is_current_block_terminated()) {
  1759. auto next_block = current_block;
  1760. next_block++;
  1761. if (next_block.is_end()) {
  1762. generator.emit<Bytecode::Op::Jump>().set_targets(Bytecode::Label { end_block }, {});
  1763. } else {
  1764. generator.emit<Bytecode::Op::Jump>().set_targets(Bytecode::Label { *next_block }, {});
  1765. }
  1766. }
  1767. current_block++;
  1768. }
  1769. generator.end_breakable_scope();
  1770. if (has_lexical_block)
  1771. generator.end_variable_scope();
  1772. generator.switch_to_basic_block(end_block);
  1773. return {};
  1774. }
  1775. Bytecode::CodeGenerationErrorOr<void> ClassDeclaration::generate_bytecode(Bytecode::Generator& generator) const
  1776. {
  1777. TRY(m_class_expression->generate_bytecode(generator));
  1778. generator.emit<Bytecode::Op::SetVariable>(generator.intern_identifier(m_class_expression.ptr()->name()), Bytecode::Op::SetVariable::InitializationMode::Initialize);
  1779. return {};
  1780. }
  1781. Bytecode::CodeGenerationErrorOr<void> ClassExpression::generate_bytecode(Bytecode::Generator& generator) const
  1782. {
  1783. generator.emit<Bytecode::Op::NewClass>(*this);
  1784. return {};
  1785. }
  1786. Bytecode::CodeGenerationErrorOr<void> SpreadExpression::generate_bytecode(Bytecode::Generator& generator) const
  1787. {
  1788. // NOTE: All users of this should handle the behaviour of this on their own,
  1789. // assuming it returns an Array-like object
  1790. return m_target->generate_bytecode(generator);
  1791. }
  1792. Bytecode::CodeGenerationErrorOr<void> ThisExpression::generate_bytecode(Bytecode::Generator& generator) const
  1793. {
  1794. generator.emit<Bytecode::Op::ResolveThisBinding>();
  1795. return {};
  1796. }
  1797. Bytecode::CodeGenerationErrorOr<void> AwaitExpression::generate_bytecode(Bytecode::Generator& generator) const
  1798. {
  1799. VERIFY(generator.is_in_async_function());
  1800. // Transform `await expr` to `yield expr`, see AsyncFunctionDriverWrapper
  1801. // For that we just need to copy most of the code from YieldExpression
  1802. auto received_completion_register = generator.allocate_register();
  1803. auto received_completion_type_register = generator.allocate_register();
  1804. auto received_completion_value_register = generator.allocate_register();
  1805. auto type_identifier = generator.intern_identifier("type");
  1806. auto value_identifier = generator.intern_identifier("value");
  1807. TRY(m_argument->generate_bytecode(generator));
  1808. auto& continuation_block = generator.make_block();
  1809. generator.emit<Bytecode::Op::Yield>(Bytecode::Label { continuation_block });
  1810. generator.switch_to_basic_block(continuation_block);
  1811. // The accumulator is set to an object, for example: { "type": 1 (normal), value: 1337 }
  1812. generator.emit<Bytecode::Op::Store>(received_completion_register);
  1813. generator.emit<Bytecode::Op::GetById>(type_identifier);
  1814. generator.emit<Bytecode::Op::Store>(received_completion_type_register);
  1815. generator.emit<Bytecode::Op::Load>(received_completion_register);
  1816. generator.emit<Bytecode::Op::GetById>(value_identifier);
  1817. generator.emit<Bytecode::Op::Store>(received_completion_value_register);
  1818. auto& normal_completion_continuation_block = generator.make_block();
  1819. auto& throw_value_block = generator.make_block();
  1820. generator.emit<Bytecode::Op::LoadImmediate>(Value(to_underlying(Completion::Type::Normal)));
  1821. generator.emit<Bytecode::Op::StrictlyEquals>(received_completion_type_register);
  1822. generator.emit<Bytecode::Op::JumpConditional>(
  1823. Bytecode::Label { normal_completion_continuation_block },
  1824. Bytecode::Label { throw_value_block });
  1825. // Simplification: The only abrupt completion we receive from AsyncFunctionDriverWrapper is Type::Throw
  1826. // So we do not need to account for the Type::Return path
  1827. generator.switch_to_basic_block(throw_value_block);
  1828. generator.emit<Bytecode::Op::Load>(received_completion_value_register);
  1829. generator.perform_needed_unwinds<Bytecode::Op::Throw>();
  1830. generator.emit<Bytecode::Op::Throw>();
  1831. generator.switch_to_basic_block(normal_completion_continuation_block);
  1832. generator.emit<Bytecode::Op::Load>(received_completion_value_register);
  1833. return {};
  1834. }
  1835. Bytecode::CodeGenerationErrorOr<void> WithStatement::generate_bytecode(Bytecode::Generator& generator) const
  1836. {
  1837. TRY(m_object->generate_bytecode(generator));
  1838. generator.emit<Bytecode::Op::EnterObjectEnvironment>();
  1839. // EnterObjectEnvironment sets the running execution context's lexical_environment to a new Object Environment.
  1840. generator.start_boundary(Bytecode::Generator::BlockBoundaryType::LeaveLexicalEnvironment);
  1841. TRY(m_body->generate_bytecode(generator));
  1842. generator.end_boundary(Bytecode::Generator::BlockBoundaryType::LeaveLexicalEnvironment);
  1843. if (!generator.is_current_block_terminated())
  1844. generator.emit<Bytecode::Op::LeaveLexicalEnvironment>();
  1845. return {};
  1846. }
  1847. enum class LHSKind {
  1848. Assignment,
  1849. VarBinding,
  1850. LexicalBinding,
  1851. };
  1852. enum class IterationKind {
  1853. Enumerate,
  1854. Iterate,
  1855. AsyncIterate,
  1856. };
  1857. // 14.7.5.6 ForIn/OfHeadEvaluation ( uninitializedBoundNames, expr, iterationKind ), https://tc39.es/ecma262/#sec-runtime-semantics-forinofheadevaluation
  1858. struct ForInOfHeadEvaluationResult {
  1859. bool is_destructuring { false };
  1860. LHSKind lhs_kind { LHSKind::Assignment };
  1861. };
  1862. static Bytecode::CodeGenerationErrorOr<ForInOfHeadEvaluationResult> for_in_of_head_evaluation(Bytecode::Generator& generator, IterationKind iteration_kind, Variant<NonnullRefPtr<ASTNode const>, NonnullRefPtr<BindingPattern const>> const& lhs, NonnullRefPtr<ASTNode const> const& rhs)
  1863. {
  1864. ForInOfHeadEvaluationResult result {};
  1865. bool entered_lexical_scope = false;
  1866. if (auto* ast_ptr = lhs.get_pointer<NonnullRefPtr<ASTNode const>>(); ast_ptr && is<VariableDeclaration>(**ast_ptr)) {
  1867. // Runtime Semantics: ForInOfLoopEvaluation, for any of:
  1868. // ForInOfStatement : for ( var ForBinding in Expression ) Statement
  1869. // ForInOfStatement : for ( ForDeclaration in Expression ) Statement
  1870. // ForInOfStatement : for ( var ForBinding of AssignmentExpression ) Statement
  1871. // ForInOfStatement : for ( ForDeclaration of AssignmentExpression ) Statement
  1872. auto& variable_declaration = static_cast<VariableDeclaration const&>(**ast_ptr);
  1873. result.is_destructuring = variable_declaration.declarations().first()->target().has<NonnullRefPtr<BindingPattern const>>();
  1874. result.lhs_kind = variable_declaration.is_lexical_declaration() ? LHSKind::LexicalBinding : LHSKind::VarBinding;
  1875. // 1. Let oldEnv be the running execution context's LexicalEnvironment.
  1876. // NOTE: 'uninitializedBoundNames' refers to the lexical bindings (i.e. Const/Let) present in the second and last form.
  1877. // 2. If uninitializedBoundNames is not an empty List, then
  1878. if (variable_declaration.declaration_kind() != DeclarationKind::Var) {
  1879. entered_lexical_scope = true;
  1880. // a. Assert: uninitializedBoundNames has no duplicate entries.
  1881. // b. Let newEnv be NewDeclarativeEnvironment(oldEnv).
  1882. generator.begin_variable_scope();
  1883. // c. For each String name of uninitializedBoundNames, do
  1884. // NOTE: Nothing in the callback throws an exception.
  1885. MUST(variable_declaration.for_each_bound_name([&](auto const& name) {
  1886. // i. Perform ! newEnv.CreateMutableBinding(name, false).
  1887. auto identifier = generator.intern_identifier(name);
  1888. generator.emit<Bytecode::Op::CreateVariable>(identifier, Bytecode::Op::EnvironmentMode::Lexical, false);
  1889. }));
  1890. // d. Set the running execution context's LexicalEnvironment to newEnv.
  1891. // NOTE: Done by CreateLexicalEnvironment.
  1892. }
  1893. } else {
  1894. // Runtime Semantics: ForInOfLoopEvaluation, for any of:
  1895. // ForInOfStatement : for ( LeftHandSideExpression in Expression ) Statement
  1896. // ForInOfStatement : for ( LeftHandSideExpression of AssignmentExpression ) Statement
  1897. result.lhs_kind = LHSKind::Assignment;
  1898. }
  1899. // 3. Let exprRef be the result of evaluating expr.
  1900. TRY(rhs->generate_bytecode(generator));
  1901. // 4. Set the running execution context's LexicalEnvironment to oldEnv.
  1902. if (entered_lexical_scope)
  1903. generator.end_variable_scope();
  1904. // 5. Let exprValue be ? GetValue(exprRef).
  1905. // NOTE: No need to store this anywhere.
  1906. // 6. If iterationKind is enumerate, then
  1907. if (iteration_kind == IterationKind::Enumerate) {
  1908. // a. If exprValue is undefined or null, then
  1909. auto& nullish_block = generator.make_block();
  1910. auto& continuation_block = generator.make_block();
  1911. auto& jump = generator.emit<Bytecode::Op::JumpNullish>();
  1912. jump.set_targets(Bytecode::Label { nullish_block }, Bytecode::Label { continuation_block });
  1913. // i. Return Completion Record { [[Type]]: break, [[Value]]: empty, [[Target]]: empty }.
  1914. generator.switch_to_basic_block(nullish_block);
  1915. generator.generate_break();
  1916. generator.switch_to_basic_block(continuation_block);
  1917. // b. Let obj be ! ToObject(exprValue).
  1918. // NOTE: GetObjectPropertyIterator does this.
  1919. // c. Let iterator be EnumerateObjectProperties(obj).
  1920. // d. Let nextMethod be ! GetV(iterator, "next").
  1921. // e. Return the Iterator Record { [[Iterator]]: iterator, [[NextMethod]]: nextMethod, [[Done]]: false }.
  1922. generator.emit<Bytecode::Op::GetObjectPropertyIterator>();
  1923. }
  1924. // 7. Else,
  1925. else {
  1926. // a. Assert: iterationKind is iterate or async-iterate.
  1927. // b. If iterationKind is async-iterate, let iteratorHint be async.
  1928. if (iteration_kind == IterationKind::AsyncIterate) {
  1929. return Bytecode::CodeGenerationError {
  1930. rhs.ptr(),
  1931. "Unimplemented iteration mode: AsyncIterate"sv,
  1932. };
  1933. }
  1934. // c. Else, let iteratorHint be sync.
  1935. // d. Return ? GetIterator(exprValue, iteratorHint).
  1936. generator.emit<Bytecode::Op::GetIterator>();
  1937. }
  1938. return result;
  1939. }
  1940. // 14.7.5.7 ForIn/OfBodyEvaluation ( lhs, stmt, iteratorRecord, iterationKind, lhsKind, labelSet [ , iteratorKind ] ), https://tc39.es/ecma262/#sec-runtime-semantics-forin-div-ofbodyevaluation-lhs-stmt-iterator-lhskind-labelset
  1941. static Bytecode::CodeGenerationErrorOr<void> for_in_of_body_evaluation(Bytecode::Generator& generator, ASTNode const& node, Variant<NonnullRefPtr<ASTNode const>, NonnullRefPtr<BindingPattern const>> const& lhs, ASTNode const& body, ForInOfHeadEvaluationResult const& head_result, Vector<DeprecatedFlyString> const& label_set, Bytecode::BasicBlock& loop_end, Bytecode::BasicBlock& loop_update)
  1942. {
  1943. auto iterator_register = generator.allocate_register();
  1944. generator.emit<Bytecode::Op::Store>(iterator_register);
  1945. // FIXME: Implement this
  1946. // 1. If iteratorKind is not present, set iteratorKind to sync.
  1947. // 2. Let oldEnv be the running execution context's LexicalEnvironment.
  1948. bool has_lexical_binding = false;
  1949. // 3. Let V be undefined.
  1950. // NOTE: We don't need 'V' as the resulting value will naturally flow through via the accumulator register.
  1951. // 4. Let destructuring be IsDestructuring of lhs.
  1952. auto destructuring = head_result.is_destructuring;
  1953. // 5. If destructuring is true and if lhsKind is assignment, then
  1954. if (destructuring && head_result.lhs_kind == LHSKind::Assignment) {
  1955. // a. Assert: lhs is a LeftHandSideExpression.
  1956. // b. Let assignmentPattern be the AssignmentPattern that is covered by lhs.
  1957. // FIXME: Implement this.
  1958. return Bytecode::CodeGenerationError {
  1959. &node,
  1960. "Unimplemented: assignment destructuring in for/of"sv,
  1961. };
  1962. }
  1963. // 6. Repeat,
  1964. generator.emit<Bytecode::Op::Jump>(Bytecode::Label { loop_update });
  1965. generator.switch_to_basic_block(loop_update);
  1966. generator.begin_continuable_scope(Bytecode::Label { loop_update }, label_set);
  1967. // a. Let nextResult be ? Call(iteratorRecord.[[NextMethod]], iteratorRecord.[[Iterator]]).
  1968. generator.emit<Bytecode::Op::Load>(iterator_register);
  1969. generator.emit<Bytecode::Op::IteratorNext>();
  1970. // FIXME: Implement this:
  1971. // b. If iteratorKind is async, set nextResult to ? Await(nextResult).
  1972. // c. If Type(nextResult) is not Object, throw a TypeError exception.
  1973. // NOTE: IteratorComplete already does this.
  1974. // d. Let done be ? IteratorComplete(nextResult).
  1975. auto iterator_result_register = generator.allocate_register();
  1976. generator.emit<Bytecode::Op::Store>(iterator_result_register);
  1977. generator.emit<Bytecode::Op::IteratorResultDone>();
  1978. // e. If done is true, return V.
  1979. auto& loop_continue = generator.make_block();
  1980. generator.emit<Bytecode::Op::JumpConditional>().set_targets(Bytecode::Label { loop_end }, Bytecode::Label { loop_continue });
  1981. generator.switch_to_basic_block(loop_continue);
  1982. // f. Let nextValue be ? IteratorValue(nextResult).
  1983. generator.emit<Bytecode::Op::Load>(iterator_result_register);
  1984. generator.emit<Bytecode::Op::IteratorResultValue>();
  1985. // g. If lhsKind is either assignment or varBinding, then
  1986. if (head_result.lhs_kind != LHSKind::LexicalBinding) {
  1987. // i. If destructuring is false, then
  1988. if (!destructuring) {
  1989. // 1. Let lhsRef be the result of evaluating lhs. (It may be evaluated repeatedly.)
  1990. // NOTE: We're skipping all the completion stuff that the spec does, as the unwinding mechanism will take case of doing that.
  1991. if (head_result.lhs_kind == LHSKind::VarBinding) {
  1992. auto& declaration = static_cast<VariableDeclaration const&>(*lhs.get<NonnullRefPtr<ASTNode const>>());
  1993. VERIFY(declaration.declarations().size() == 1);
  1994. TRY(assign_accumulator_to_variable_declarator(generator, declaration.declarations().first(), declaration));
  1995. } else {
  1996. if (auto ptr = lhs.get_pointer<NonnullRefPtr<ASTNode const>>()) {
  1997. TRY(generator.emit_store_to_reference(**ptr));
  1998. } else {
  1999. auto& binding_pattern = lhs.get<NonnullRefPtr<BindingPattern const>>();
  2000. auto value_register = generator.allocate_register();
  2001. generator.emit<Bytecode::Op::Store>(value_register);
  2002. TRY(generate_binding_pattern_bytecode(generator, *binding_pattern, Bytecode::Op::SetVariable::InitializationMode::Set, value_register));
  2003. }
  2004. }
  2005. }
  2006. }
  2007. // h. Else,
  2008. else {
  2009. // i. Assert: lhsKind is lexicalBinding.
  2010. // ii. Assert: lhs is a ForDeclaration.
  2011. // iii. Let iterationEnv be NewDeclarativeEnvironment(oldEnv).
  2012. // iv. Perform ForDeclarationBindingInstantiation of lhs with argument iterationEnv.
  2013. // v. Set the running execution context's LexicalEnvironment to iterationEnv.
  2014. generator.begin_variable_scope();
  2015. has_lexical_binding = true;
  2016. // 14.7.5.4 Runtime Semantics: ForDeclarationBindingInstantiation, https://tc39.es/ecma262/#sec-runtime-semantics-fordeclarationbindinginstantiation
  2017. // 1. Assert: environment is a declarative Environment Record.
  2018. // NOTE: We just made it.
  2019. auto& variable_declaration = static_cast<VariableDeclaration const&>(*lhs.get<NonnullRefPtr<ASTNode const>>());
  2020. // 2. For each element name of the BoundNames of ForBinding, do
  2021. // NOTE: Nothing in the callback throws an exception.
  2022. MUST(variable_declaration.for_each_bound_name([&](auto const& name) {
  2023. auto identifier = generator.intern_identifier(name);
  2024. // a. If IsConstantDeclaration of LetOrConst is true, then
  2025. if (variable_declaration.is_constant_declaration()) {
  2026. // i. Perform ! environment.CreateImmutableBinding(name, true).
  2027. generator.emit<Bytecode::Op::CreateVariable>(identifier, Bytecode::Op::EnvironmentMode::Lexical, true);
  2028. }
  2029. // b. Else,
  2030. else {
  2031. // i. Perform ! environment.CreateMutableBinding(name, false).
  2032. generator.emit<Bytecode::Op::CreateVariable>(identifier, Bytecode::Op::EnvironmentMode::Lexical, false);
  2033. }
  2034. }));
  2035. // 3. Return unused.
  2036. // NOTE: No need to do that as we've inlined this.
  2037. // vi. If destructuring is false, then
  2038. if (!destructuring) {
  2039. // 1. Assert: lhs binds a single name.
  2040. // 2. Let lhsName be the sole element of BoundNames of lhs.
  2041. auto lhs_name = variable_declaration.declarations().first()->target().get<NonnullRefPtr<Identifier const>>()->string();
  2042. // 3. Let lhsRef be ! ResolveBinding(lhsName).
  2043. // NOTE: We're skipping all the completion stuff that the spec does, as the unwinding mechanism will take case of doing that.
  2044. auto identifier = generator.intern_identifier(lhs_name);
  2045. generator.emit<Bytecode::Op::SetVariable>(identifier, Bytecode::Op::SetVariable::InitializationMode::Initialize, Bytecode::Op::EnvironmentMode::Lexical);
  2046. }
  2047. }
  2048. // i. If destructuring is false, then
  2049. if (!destructuring) {
  2050. // i. If lhsRef is an abrupt completion, then
  2051. // 1. Let status be lhsRef.
  2052. // ii. Else if lhsKind is lexicalBinding, then
  2053. // 1. Let status be Completion(InitializeReferencedBinding(lhsRef, nextValue)).
  2054. // iii. Else,
  2055. // 1. Let status be Completion(PutValue(lhsRef, nextValue)).
  2056. // NOTE: This is performed above.
  2057. }
  2058. // j. Else,
  2059. else {
  2060. // FIXME: i. If lhsKind is assignment, then
  2061. // 1. Let status be Completion(DestructuringAssignmentEvaluation of assignmentPattern with argument nextValue).
  2062. // ii. Else if lhsKind is varBinding, then
  2063. // 1. Assert: lhs is a ForBinding.
  2064. // 2. Let status be Completion(BindingInitialization of lhs with arguments nextValue and undefined).
  2065. // iii. Else,
  2066. // 1. Assert: lhsKind is lexicalBinding.
  2067. // 2. Assert: lhs is a ForDeclaration.
  2068. // 3. Let status be Completion(ForDeclarationBindingInitialization of lhs with arguments nextValue and iterationEnv).
  2069. if (head_result.lhs_kind == LHSKind::VarBinding || head_result.lhs_kind == LHSKind::LexicalBinding) {
  2070. auto& declaration = static_cast<VariableDeclaration const&>(*lhs.get<NonnullRefPtr<ASTNode const>>());
  2071. VERIFY(declaration.declarations().size() == 1);
  2072. auto& binding_pattern = declaration.declarations().first()->target().get<NonnullRefPtr<BindingPattern const>>();
  2073. auto value_register = generator.allocate_register();
  2074. generator.emit<Bytecode::Op::Store>(value_register);
  2075. TRY(generate_binding_pattern_bytecode(generator, *binding_pattern, head_result.lhs_kind == LHSKind::VarBinding ? Bytecode::Op::SetVariable::InitializationMode::Set : Bytecode::Op::SetVariable::InitializationMode::Initialize, value_register));
  2076. } else {
  2077. return Bytecode::CodeGenerationError {
  2078. &node,
  2079. "Unimplemented: assignment destructuring in for/of"sv,
  2080. };
  2081. }
  2082. }
  2083. // FIXME: Implement iteration closure.
  2084. // k. If status is an abrupt completion, then
  2085. // i. Set the running execution context's LexicalEnvironment to oldEnv.
  2086. // ii. If iteratorKind is async, return ? AsyncIteratorClose(iteratorRecord, status).
  2087. // iii. If iterationKind is enumerate, then
  2088. // 1. Return ? status.
  2089. // iv. Else,
  2090. // 1. Assert: iterationKind is iterate.
  2091. // 2. Return ? IteratorClose(iteratorRecord, status).
  2092. // l. Let result be the result of evaluating stmt.
  2093. TRY(body.generate_bytecode(generator));
  2094. // m. Set the running execution context's LexicalEnvironment to oldEnv.
  2095. if (has_lexical_binding)
  2096. generator.end_variable_scope();
  2097. generator.end_continuable_scope();
  2098. generator.end_breakable_scope();
  2099. // NOTE: If we're here, then the loop definitely continues.
  2100. // n. If LoopContinues(result, labelSet) is false, then
  2101. // i. If iterationKind is enumerate, then
  2102. // 1. Return ? UpdateEmpty(result, V).
  2103. // ii. Else,
  2104. // 1. Assert: iterationKind is iterate.
  2105. // 2. Set status to Completion(UpdateEmpty(result, V)).
  2106. // 3. If iteratorKind is async, return ? AsyncIteratorClose(iteratorRecord, status).
  2107. // 4. Return ? IteratorClose(iteratorRecord, status).
  2108. // o. If result.[[Value]] is not empty, set V to result.[[Value]].
  2109. // The body can contain an unconditional block terminator (e.g. return, throw), so we have to check for that before generating the Jump.
  2110. if (!generator.is_current_block_terminated())
  2111. generator.emit<Bytecode::Op::Jump>().set_targets(Bytecode::Label { loop_update }, {});
  2112. generator.switch_to_basic_block(loop_end);
  2113. return {};
  2114. }
  2115. Bytecode::CodeGenerationErrorOr<void> ForInStatement::generate_bytecode(Bytecode::Generator& generator) const
  2116. {
  2117. return generate_labelled_evaluation(generator, {});
  2118. }
  2119. // 14.7.5.5 Runtime Semantics: ForInOfLoopEvaluation, https://tc39.es/ecma262/#sec-runtime-semantics-forinofloopevaluation
  2120. Bytecode::CodeGenerationErrorOr<void> ForInStatement::generate_labelled_evaluation(Bytecode::Generator& generator, Vector<DeprecatedFlyString> const& label_set) const
  2121. {
  2122. auto& loop_end = generator.make_block();
  2123. auto& loop_update = generator.make_block();
  2124. generator.begin_breakable_scope(Bytecode::Label { loop_end }, label_set);
  2125. auto head_result = TRY(for_in_of_head_evaluation(generator, IterationKind::Enumerate, m_lhs, m_rhs));
  2126. // Now perform the rest of ForInOfLoopEvaluation, given that the accumulator holds the iterator we're supposed to iterate over.
  2127. return for_in_of_body_evaluation(generator, *this, m_lhs, body(), head_result, label_set, loop_end, loop_update);
  2128. }
  2129. Bytecode::CodeGenerationErrorOr<void> ForOfStatement::generate_bytecode(Bytecode::Generator& generator) const
  2130. {
  2131. return generate_labelled_evaluation(generator, {});
  2132. }
  2133. Bytecode::CodeGenerationErrorOr<void> ForOfStatement::generate_labelled_evaluation(Bytecode::Generator& generator, Vector<DeprecatedFlyString> const& label_set) const
  2134. {
  2135. auto& loop_end = generator.make_block();
  2136. auto& loop_update = generator.make_block();
  2137. generator.begin_breakable_scope(Bytecode::Label { loop_end }, label_set);
  2138. auto head_result = TRY(for_in_of_head_evaluation(generator, IterationKind::Iterate, m_lhs, m_rhs));
  2139. // Now perform the rest of ForInOfLoopEvaluation, given that the accumulator holds the iterator we're supposed to iterate over.
  2140. return for_in_of_body_evaluation(generator, *this, m_lhs, body(), head_result, label_set, loop_end, loop_update);
  2141. }
  2142. // 13.3.12.1 Runtime Semantics: Evaluation, https://tc39.es/ecma262/#sec-meta-properties-runtime-semantics-evaluation
  2143. Bytecode::CodeGenerationErrorOr<void> MetaProperty::generate_bytecode(Bytecode::Generator& generator) const
  2144. {
  2145. // NewTarget : new . target
  2146. if (m_type == MetaProperty::Type::NewTarget) {
  2147. // 1. Return GetNewTarget().
  2148. generator.emit<Bytecode::Op::GetNewTarget>();
  2149. return {};
  2150. }
  2151. // ImportMeta : import . meta
  2152. if (m_type == MetaProperty::Type::ImportMeta) {
  2153. return Bytecode::CodeGenerationError {
  2154. this,
  2155. "Unimplemented meta property: import.meta"sv,
  2156. };
  2157. }
  2158. VERIFY_NOT_REACHED();
  2159. }
  2160. }