ASTCodegen.cpp 163 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810181118121813181418151816181718181819182018211822182318241825182618271828182918301831183218331834183518361837183818391840184118421843184418451846184718481849185018511852185318541855185618571858185918601861186218631864186518661867186818691870187118721873187418751876187718781879188018811882188318841885188618871888188918901891189218931894189518961897189818991900190119021903190419051906190719081909191019111912191319141915191619171918191919201921192219231924192519261927192819291930193119321933193419351936193719381939194019411942194319441945194619471948194919501951195219531954195519561957195819591960196119621963196419651966196719681969197019711972197319741975197619771978197919801981198219831984198519861987198819891990199119921993199419951996199719981999200020012002200320042005200620072008200920102011201220132014201520162017201820192020202120222023202420252026202720282029203020312032203320342035203620372038203920402041204220432044204520462047204820492050205120522053205420552056205720582059206020612062206320642065206620672068206920702071207220732074207520762077207820792080208120822083208420852086208720882089209020912092209320942095209620972098209921002101210221032104210521062107210821092110211121122113211421152116211721182119212021212122212321242125212621272128212921302131213221332134213521362137213821392140214121422143214421452146214721482149215021512152215321542155215621572158215921602161216221632164216521662167216821692170217121722173217421752176217721782179218021812182218321842185218621872188218921902191219221932194219521962197219821992200220122022203220422052206220722082209221022112212221322142215221622172218221922202221222222232224222522262227222822292230223122322233223422352236223722382239224022412242224322442245224622472248224922502251225222532254225522562257225822592260226122622263226422652266226722682269227022712272227322742275227622772278227922802281228222832284228522862287228822892290229122922293229422952296229722982299230023012302230323042305230623072308230923102311231223132314231523162317231823192320232123222323232423252326232723282329233023312332233323342335233623372338233923402341234223432344234523462347234823492350235123522353235423552356235723582359236023612362236323642365236623672368236923702371237223732374237523762377237823792380238123822383238423852386238723882389239023912392239323942395239623972398239924002401240224032404240524062407240824092410241124122413241424152416241724182419242024212422242324242425242624272428242924302431243224332434243524362437243824392440244124422443244424452446244724482449245024512452245324542455245624572458245924602461246224632464246524662467246824692470247124722473247424752476247724782479248024812482248324842485248624872488248924902491249224932494249524962497249824992500250125022503250425052506250725082509251025112512251325142515251625172518251925202521252225232524252525262527252825292530253125322533253425352536253725382539254025412542254325442545254625472548254925502551255225532554255525562557255825592560256125622563256425652566256725682569257025712572257325742575257625772578257925802581258225832584258525862587258825892590259125922593259425952596259725982599260026012602260326042605260626072608260926102611261226132614261526162617261826192620262126222623262426252626262726282629263026312632263326342635263626372638263926402641264226432644264526462647264826492650265126522653265426552656265726582659266026612662266326642665266626672668266926702671267226732674267526762677267826792680268126822683268426852686268726882689269026912692269326942695269626972698269927002701270227032704270527062707270827092710271127122713271427152716271727182719272027212722272327242725272627272728272927302731273227332734273527362737273827392740274127422743274427452746274727482749275027512752275327542755275627572758275927602761276227632764276527662767276827692770277127722773277427752776277727782779278027812782278327842785278627872788278927902791279227932794279527962797279827992800280128022803280428052806280728082809281028112812281328142815281628172818281928202821282228232824282528262827282828292830283128322833283428352836283728382839284028412842284328442845284628472848284928502851285228532854285528562857285828592860286128622863286428652866286728682869287028712872287328742875287628772878287928802881288228832884288528862887288828892890289128922893289428952896289728982899290029012902290329042905290629072908290929102911291229132914291529162917291829192920292129222923292429252926292729282929293029312932293329342935293629372938293929402941294229432944294529462947294829492950295129522953295429552956295729582959296029612962296329642965296629672968296929702971297229732974297529762977297829792980298129822983298429852986298729882989299029912992299329942995299629972998299930003001300230033004300530063007300830093010301130123013301430153016301730183019302030213022302330243025302630273028302930303031303230333034303530363037303830393040304130423043304430453046304730483049305030513052305330543055305630573058305930603061306230633064306530663067306830693070307130723073307430753076307730783079308030813082308330843085308630873088308930903091309230933094309530963097309830993100310131023103310431053106310731083109311031113112311331143115311631173118311931203121312231233124312531263127312831293130313131323133313431353136313731383139314031413142314331443145314631473148314931503151315231533154315531563157315831593160316131623163316431653166316731683169317031713172317331743175317631773178317931803181318231833184318531863187318831893190319131923193319431953196319731983199320032013202320332043205320632073208320932103211321232133214321532163217321832193220322132223223322432253226322732283229323032313232323332343235323632373238323932403241324232433244324532463247324832493250325132523253325432553256325732583259326032613262326332643265326632673268326932703271327232733274327532763277327832793280328132823283328432853286328732883289329032913292329332943295329632973298329933003301330233033304330533063307330833093310331133123313331433153316331733183319332033213322332333243325332633273328332933303331333233333334333533363337333833393340334133423343334433453346334733483349335033513352335333543355335633573358335933603361336233633364336533663367336833693370337133723373337433753376337733783379338033813382338333843385338633873388338933903391339233933394339533963397339833993400340134023403340434053406340734083409341034113412341334143415341634173418341934203421342234233424342534263427342834293430343134323433343434353436343734383439344034413442344334443445344634473448344934503451345234533454345534563457345834593460346134623463
  1. /*
  2. * Copyright (c) 2021-2024, 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 <AK/Queue.h>
  11. #include <LibJS/AST.h>
  12. #include <LibJS/Bytecode/Generator.h>
  13. #include <LibJS/Bytecode/Instruction.h>
  14. #include <LibJS/Bytecode/Op.h>
  15. #include <LibJS/Bytecode/Register.h>
  16. #include <LibJS/Bytecode/StringTable.h>
  17. #include <LibJS/Runtime/Environment.h>
  18. #include <LibJS/Runtime/ErrorTypes.h>
  19. namespace JS {
  20. using namespace JS::Bytecode;
  21. static ScopedOperand choose_dst(Bytecode::Generator& generator, Optional<ScopedOperand> const& preferred_dst)
  22. {
  23. if (preferred_dst.has_value())
  24. return preferred_dst.value();
  25. return generator.allocate_register();
  26. }
  27. Bytecode::CodeGenerationErrorOr<Optional<ScopedOperand>> ASTNode::generate_bytecode(Bytecode::Generator&, [[maybe_unused]] Optional<ScopedOperand> preferred_dst) const
  28. {
  29. return Bytecode::CodeGenerationError {
  30. this,
  31. "Missing generate_bytecode()"sv,
  32. };
  33. }
  34. Bytecode::CodeGenerationErrorOr<Optional<ScopedOperand>> ScopeNode::generate_bytecode(Bytecode::Generator& generator, [[maybe_unused]] Optional<ScopedOperand> preferred_dst) const
  35. {
  36. Bytecode::Generator::SourceLocationScope scope(generator, *this);
  37. bool did_create_lexical_environment = false;
  38. if (is<BlockStatement>(*this)) {
  39. if (has_lexical_declarations()) {
  40. did_create_lexical_environment = generator.emit_block_declaration_instantiation(*this);
  41. }
  42. } else if (is<Program>(*this)) {
  43. // GlobalDeclarationInstantiation is handled by the C++ AO.
  44. } else {
  45. // FunctionDeclarationInstantiation is handled by the C++ AO.
  46. }
  47. Optional<ScopedOperand> last_result;
  48. for (auto& child : children()) {
  49. auto result = TRY(child->generate_bytecode(generator));
  50. if (result.has_value())
  51. last_result = result;
  52. if (generator.is_current_block_terminated())
  53. break;
  54. }
  55. if (did_create_lexical_environment)
  56. generator.end_variable_scope();
  57. return last_result;
  58. }
  59. Bytecode::CodeGenerationErrorOr<Optional<ScopedOperand>> EmptyStatement::generate_bytecode(Bytecode::Generator&, [[maybe_unused]] Optional<ScopedOperand> preferred_dst) const
  60. {
  61. return Optional<ScopedOperand> {};
  62. }
  63. Bytecode::CodeGenerationErrorOr<Optional<ScopedOperand>> ExpressionStatement::generate_bytecode(Bytecode::Generator& generator, [[maybe_unused]] Optional<ScopedOperand> preferred_dst) const
  64. {
  65. Bytecode::Generator::SourceLocationScope scope(generator, *this);
  66. return m_expression->generate_bytecode(generator);
  67. }
  68. Bytecode::CodeGenerationErrorOr<Optional<ScopedOperand>> BinaryExpression::generate_bytecode(Bytecode::Generator& generator, Optional<ScopedOperand> preferred_dst) const
  69. {
  70. Bytecode::Generator::SourceLocationScope scope(generator, *this);
  71. if (m_op == BinaryOp::In && is<PrivateIdentifier>(*m_lhs)) {
  72. auto const& private_identifier = static_cast<PrivateIdentifier const&>(*m_lhs).string();
  73. auto base = TRY(m_rhs->generate_bytecode(generator)).value();
  74. auto dst = choose_dst(generator, preferred_dst);
  75. generator.emit<Bytecode::Op::HasPrivateId>(dst, base, generator.intern_identifier(private_identifier));
  76. return dst;
  77. }
  78. // OPTIMIZATION: If LHS and/or RHS are numeric literals, we make sure they are converted to i32/u32
  79. // as appropriate, to avoid having to perform these conversions at runtime.
  80. auto get_left_side = [&](Expression const& side) -> CodeGenerationErrorOr<Optional<ScopedOperand>> {
  81. switch (m_op) {
  82. case BinaryOp::BitwiseAnd:
  83. case BinaryOp::BitwiseOr:
  84. case BinaryOp::BitwiseXor:
  85. case BinaryOp::LeftShift:
  86. case BinaryOp::RightShift:
  87. case BinaryOp::UnsignedRightShift:
  88. // LHS will always be converted to i32 for these ops.
  89. if (side.is_numeric_literal()) {
  90. auto value = MUST(static_cast<NumericLiteral const&>(side).value().to_i32(generator.vm()));
  91. return generator.add_constant(Value(value));
  92. }
  93. break;
  94. default:
  95. break;
  96. }
  97. return side.generate_bytecode(generator);
  98. };
  99. auto get_right_side = [&](Expression const& side) -> CodeGenerationErrorOr<Optional<ScopedOperand>> {
  100. switch (m_op) {
  101. case BinaryOp::BitwiseAnd:
  102. case BinaryOp::BitwiseOr:
  103. case BinaryOp::BitwiseXor:
  104. // RHS will always be converted to i32 for these ops.
  105. if (side.is_numeric_literal()) {
  106. auto value = MUST(static_cast<NumericLiteral const&>(side).value().to_i32(generator.vm()));
  107. return generator.add_constant(Value(value));
  108. }
  109. break;
  110. case BinaryOp::LeftShift:
  111. case BinaryOp::RightShift:
  112. case BinaryOp::UnsignedRightShift:
  113. // RHS will always be converted to u32 for these ops.
  114. if (side.is_numeric_literal()) {
  115. auto value = MUST(static_cast<NumericLiteral const&>(side).value().to_u32(generator.vm()));
  116. return generator.add_constant(Value(value));
  117. }
  118. break;
  119. default:
  120. break;
  121. }
  122. return side.generate_bytecode(generator);
  123. };
  124. auto lhs = TRY(get_left_side(*m_lhs)).value();
  125. auto rhs = TRY(get_right_side(*m_rhs)).value();
  126. auto dst = choose_dst(generator, preferred_dst);
  127. switch (m_op) {
  128. case BinaryOp::Addition:
  129. generator.emit<Bytecode::Op::Add>(dst, lhs, rhs);
  130. break;
  131. case BinaryOp::Subtraction:
  132. generator.emit<Bytecode::Op::Sub>(dst, lhs, rhs);
  133. break;
  134. case BinaryOp::Multiplication:
  135. generator.emit<Bytecode::Op::Mul>(dst, lhs, rhs);
  136. break;
  137. case BinaryOp::Division:
  138. generator.emit<Bytecode::Op::Div>(dst, lhs, rhs);
  139. break;
  140. case BinaryOp::Modulo:
  141. generator.emit<Bytecode::Op::Mod>(dst, lhs, rhs);
  142. break;
  143. case BinaryOp::Exponentiation:
  144. generator.emit<Bytecode::Op::Exp>(dst, lhs, rhs);
  145. break;
  146. case BinaryOp::GreaterThan:
  147. generator.emit<Bytecode::Op::GreaterThan>(dst, lhs, rhs);
  148. break;
  149. case BinaryOp::GreaterThanEquals:
  150. generator.emit<Bytecode::Op::GreaterThanEquals>(dst, lhs, rhs);
  151. break;
  152. case BinaryOp::LessThan:
  153. generator.emit<Bytecode::Op::LessThan>(dst, lhs, rhs);
  154. break;
  155. case BinaryOp::LessThanEquals:
  156. generator.emit<Bytecode::Op::LessThanEquals>(dst, lhs, rhs);
  157. break;
  158. case BinaryOp::LooselyInequals:
  159. generator.emit<Bytecode::Op::LooselyInequals>(dst, lhs, rhs);
  160. break;
  161. case BinaryOp::LooselyEquals:
  162. generator.emit<Bytecode::Op::LooselyEquals>(dst, lhs, rhs);
  163. break;
  164. case BinaryOp::StrictlyInequals:
  165. generator.emit<Bytecode::Op::StrictlyInequals>(dst, lhs, rhs);
  166. break;
  167. case BinaryOp::StrictlyEquals:
  168. generator.emit<Bytecode::Op::StrictlyEquals>(dst, lhs, rhs);
  169. break;
  170. case BinaryOp::BitwiseAnd:
  171. generator.emit<Bytecode::Op::BitwiseAnd>(dst, lhs, rhs);
  172. break;
  173. case BinaryOp::BitwiseOr:
  174. generator.emit<Bytecode::Op::BitwiseOr>(dst, lhs, rhs);
  175. break;
  176. case BinaryOp::BitwiseXor:
  177. generator.emit<Bytecode::Op::BitwiseXor>(dst, lhs, rhs);
  178. break;
  179. case BinaryOp::LeftShift:
  180. generator.emit<Bytecode::Op::LeftShift>(dst, lhs, rhs);
  181. break;
  182. case BinaryOp::RightShift:
  183. generator.emit<Bytecode::Op::RightShift>(dst, lhs, rhs);
  184. break;
  185. case BinaryOp::UnsignedRightShift:
  186. generator.emit<Bytecode::Op::UnsignedRightShift>(dst, lhs, rhs);
  187. break;
  188. case BinaryOp::In:
  189. generator.emit<Bytecode::Op::In>(dst, lhs, rhs);
  190. break;
  191. case BinaryOp::InstanceOf:
  192. generator.emit<Bytecode::Op::InstanceOf>(dst, lhs, rhs);
  193. break;
  194. default:
  195. VERIFY_NOT_REACHED();
  196. }
  197. return dst;
  198. }
  199. Bytecode::CodeGenerationErrorOr<Optional<ScopedOperand>> LogicalExpression::generate_bytecode(Bytecode::Generator& generator, Optional<ScopedOperand> preferred_dst) const
  200. {
  201. Bytecode::Generator::SourceLocationScope scope(generator, *this);
  202. auto dst = choose_dst(generator, preferred_dst);
  203. auto lhs = TRY(m_lhs->generate_bytecode(generator, preferred_dst)).value();
  204. // FIXME: Only mov lhs into dst in case lhs is the value taken.
  205. generator.emit<Bytecode::Op::Mov>(dst, lhs);
  206. // lhs
  207. // jump op (true) end (false) rhs
  208. // rhs
  209. // jump always (true) end
  210. // end
  211. auto& rhs_block = generator.make_block();
  212. auto& end_block = generator.make_block();
  213. switch (m_op) {
  214. case LogicalOp::And:
  215. generator.emit_jump_if(
  216. lhs,
  217. Bytecode::Label { rhs_block },
  218. Bytecode::Label { end_block });
  219. break;
  220. case LogicalOp::Or:
  221. generator.emit_jump_if(
  222. lhs,
  223. Bytecode::Label { end_block },
  224. Bytecode::Label { rhs_block });
  225. break;
  226. case LogicalOp::NullishCoalescing:
  227. generator.emit<Bytecode::Op::JumpNullish>(
  228. lhs,
  229. Bytecode::Label { rhs_block },
  230. Bytecode::Label { end_block });
  231. break;
  232. default:
  233. VERIFY_NOT_REACHED();
  234. }
  235. generator.switch_to_basic_block(rhs_block);
  236. auto rhs = TRY(m_rhs->generate_bytecode(generator)).value();
  237. generator.emit<Bytecode::Op::Mov>(dst, rhs);
  238. generator.emit<Bytecode::Op::Jump>(Bytecode::Label { end_block });
  239. generator.switch_to_basic_block(end_block);
  240. return dst;
  241. }
  242. Bytecode::CodeGenerationErrorOr<Optional<ScopedOperand>> UnaryExpression::generate_bytecode(Bytecode::Generator& generator, Optional<ScopedOperand> preferred_dst) const
  243. {
  244. Bytecode::Generator::SourceLocationScope scope(generator, *this);
  245. // OPTIMIZATION: Turn expressions like `-1` into a constant.
  246. if (m_op == UnaryOp::Minus && is<NumericLiteral>(*m_lhs)) {
  247. auto& numeric_literal = static_cast<NumericLiteral const&>(*m_lhs);
  248. auto value = numeric_literal.value();
  249. return generator.add_constant(Value(-value.as_double()));
  250. }
  251. if (m_op == UnaryOp::Delete)
  252. return generator.emit_delete_reference(m_lhs);
  253. Optional<ScopedOperand> src;
  254. // Typeof needs some special handling for when the LHS is an Identifier. Namely, it shouldn't throw on unresolvable references, but instead return "undefined".
  255. if (m_op != UnaryOp::Typeof)
  256. src = TRY(m_lhs->generate_bytecode(generator)).value();
  257. auto dst = choose_dst(generator, preferred_dst);
  258. switch (m_op) {
  259. case UnaryOp::BitwiseNot:
  260. generator.emit<Bytecode::Op::BitwiseNot>(dst, *src);
  261. break;
  262. case UnaryOp::Not:
  263. generator.emit<Bytecode::Op::Not>(dst, *src);
  264. break;
  265. case UnaryOp::Plus:
  266. generator.emit<Bytecode::Op::UnaryPlus>(dst, *src);
  267. break;
  268. case UnaryOp::Minus:
  269. generator.emit<Bytecode::Op::UnaryMinus>(dst, *src);
  270. break;
  271. case UnaryOp::Typeof:
  272. if (is<Identifier>(*m_lhs)) {
  273. auto& identifier = static_cast<Identifier const&>(*m_lhs);
  274. if (!identifier.is_local()) {
  275. generator.emit<Bytecode::Op::TypeofVariable>(dst, generator.intern_identifier(identifier.string()));
  276. break;
  277. }
  278. }
  279. src = TRY(m_lhs->generate_bytecode(generator)).value();
  280. generator.emit<Bytecode::Op::Typeof>(dst, *src);
  281. break;
  282. case UnaryOp::Void:
  283. return generator.add_constant(js_undefined());
  284. case UnaryOp::Delete: // Delete is implemented above.
  285. default:
  286. VERIFY_NOT_REACHED();
  287. }
  288. return dst;
  289. }
  290. Bytecode::CodeGenerationErrorOr<Optional<ScopedOperand>> NumericLiteral::generate_bytecode(Bytecode::Generator& generator, [[maybe_unused]] Optional<ScopedOperand> preferred_dst) const
  291. {
  292. Bytecode::Generator::SourceLocationScope scope(generator, *this);
  293. return generator.add_constant(Value(m_value), Bytecode::Generator::DeduplicateConstant::No);
  294. }
  295. Bytecode::CodeGenerationErrorOr<Optional<ScopedOperand>> BooleanLiteral::generate_bytecode(Bytecode::Generator& generator, [[maybe_unused]] Optional<ScopedOperand> preferred_dst) const
  296. {
  297. Bytecode::Generator::SourceLocationScope scope(generator, *this);
  298. return generator.add_constant(Value(m_value));
  299. }
  300. Bytecode::CodeGenerationErrorOr<Optional<ScopedOperand>> NullLiteral::generate_bytecode(Bytecode::Generator& generator, [[maybe_unused]] Optional<ScopedOperand> preferred_dst) const
  301. {
  302. Bytecode::Generator::SourceLocationScope scope(generator, *this);
  303. return generator.add_constant(js_null());
  304. }
  305. Bytecode::CodeGenerationErrorOr<Optional<ScopedOperand>> BigIntLiteral::generate_bytecode(Bytecode::Generator& generator, [[maybe_unused]] Optional<ScopedOperand> preferred_dst) const
  306. {
  307. Bytecode::Generator::SourceLocationScope scope(generator, *this);
  308. // 1. Return the NumericValue of NumericLiteral as defined in 12.8.3.
  309. auto integer = [&] {
  310. if (m_value[0] == '0' && m_value.length() >= 3)
  311. if (m_value[1] == 'x' || m_value[1] == 'X')
  312. return MUST(Crypto::SignedBigInteger::from_base(16, m_value.substring(2, m_value.length() - 3)));
  313. if (m_value[1] == 'o' || m_value[1] == 'O')
  314. return MUST(Crypto::SignedBigInteger::from_base(8, m_value.substring(2, m_value.length() - 3)));
  315. if (m_value[1] == 'b' || m_value[1] == 'B')
  316. return MUST(Crypto::SignedBigInteger::from_base(2, m_value.substring(2, m_value.length() - 3)));
  317. return MUST(Crypto::SignedBigInteger::from_base(10, m_value.substring(0, m_value.length() - 1)));
  318. }();
  319. return generator.add_constant(BigInt::create(generator.vm(), move(integer)), Bytecode::Generator::DeduplicateConstant::No);
  320. }
  321. Bytecode::CodeGenerationErrorOr<Optional<ScopedOperand>> StringLiteral::generate_bytecode(Bytecode::Generator& generator, [[maybe_unused]] Optional<ScopedOperand> preferred_dst) const
  322. {
  323. Bytecode::Generator::SourceLocationScope scope(generator, *this);
  324. return generator.add_constant(PrimitiveString::create(generator.vm(), m_value), Bytecode::Generator::DeduplicateConstant::No);
  325. }
  326. Bytecode::CodeGenerationErrorOr<Optional<ScopedOperand>> RegExpLiteral::generate_bytecode(Bytecode::Generator& generator, Optional<ScopedOperand> preferred_dst) const
  327. {
  328. Bytecode::Generator::SourceLocationScope scope(generator, *this);
  329. auto source_index = generator.intern_string(m_pattern);
  330. auto flags_index = generator.intern_string(m_flags);
  331. auto regex_index = generator.intern_regex(Bytecode::ParsedRegex {
  332. .regex = m_parsed_regex,
  333. .pattern = m_parsed_pattern,
  334. .flags = m_parsed_flags,
  335. });
  336. auto dst = choose_dst(generator, preferred_dst);
  337. generator.emit<Bytecode::Op::NewRegExp>(dst, source_index, flags_index, regex_index);
  338. return dst;
  339. }
  340. Bytecode::CodeGenerationErrorOr<Optional<ScopedOperand>> Identifier::generate_bytecode(Bytecode::Generator& generator, Optional<ScopedOperand> preferred_dst) const
  341. {
  342. Bytecode::Generator::SourceLocationScope scope(generator, *this);
  343. if (is_local()) {
  344. auto local = generator.local(local_variable_index());
  345. if (!generator.is_local_initialized(local_variable_index())) {
  346. generator.emit<Bytecode::Op::ThrowIfTDZ>(local);
  347. }
  348. return local;
  349. }
  350. if (is_global() && m_string == "undefined"sv) {
  351. return generator.add_constant(js_undefined());
  352. }
  353. auto dst = choose_dst(generator, preferred_dst);
  354. if (is_global()) {
  355. generator.emit<Bytecode::Op::GetGlobal>(dst, generator.intern_identifier(m_string), generator.next_global_variable_cache());
  356. } else {
  357. generator.emit<Bytecode::Op::GetVariable>(dst, generator.intern_identifier(m_string), generator.next_environment_variable_cache());
  358. }
  359. return dst;
  360. }
  361. static Bytecode::CodeGenerationErrorOr<Optional<ScopedOperand>> arguments_to_array_for_call(Bytecode::Generator& generator, ReadonlySpan<CallExpression::Argument> arguments)
  362. {
  363. auto dst = generator.allocate_register();
  364. if (arguments.is_empty()) {
  365. generator.emit<Bytecode::Op::NewArray>(dst);
  366. return dst;
  367. }
  368. auto first_spread = find_if(arguments.begin(), arguments.end(), [](auto el) { return el.is_spread; });
  369. Vector<ScopedOperand> args;
  370. args.ensure_capacity(first_spread.index());
  371. for (auto it = arguments.begin(); it != first_spread; ++it) {
  372. VERIFY(!it->is_spread);
  373. auto reg = generator.allocate_register();
  374. auto value = TRY(it->value->generate_bytecode(generator)).value();
  375. generator.emit<Bytecode::Op::Mov>(reg, value);
  376. args.append(move(reg));
  377. }
  378. if (first_spread.index() != 0)
  379. generator.emit_with_extra_operand_slots<Bytecode::Op::NewArray>(args.size(), dst, args);
  380. else
  381. generator.emit<Bytecode::Op::NewArray>(dst);
  382. if (first_spread != arguments.end()) {
  383. for (auto it = first_spread; it != arguments.end(); ++it) {
  384. auto value = TRY(it->value->generate_bytecode(generator)).value();
  385. generator.emit<Bytecode::Op::ArrayAppend>(dst, value, it->is_spread);
  386. }
  387. }
  388. return dst;
  389. }
  390. Bytecode::CodeGenerationErrorOr<Optional<ScopedOperand>> SuperCall::generate_bytecode(Bytecode::Generator& generator, Optional<ScopedOperand> preferred_dst) const
  391. {
  392. Bytecode::Generator::SourceLocationScope scope(generator, *this);
  393. Optional<ScopedOperand> arguments;
  394. if (m_is_synthetic == IsPartOfSyntheticConstructor::Yes) {
  395. // NOTE: This is the case where we have a fake constructor(...args) { super(...args); } which
  396. // shouldn't call @@iterator of %Array.prototype%.
  397. VERIFY(m_arguments.size() == 1);
  398. VERIFY(m_arguments[0].is_spread);
  399. auto const& argument = m_arguments[0];
  400. // This generates a single argument.
  401. arguments = MUST(argument.value->generate_bytecode(generator));
  402. } else {
  403. arguments = TRY(arguments_to_array_for_call(generator, m_arguments)).value();
  404. }
  405. auto dst = choose_dst(generator, preferred_dst);
  406. generator.emit<Bytecode::Op::SuperCallWithArgumentArray>(dst, *arguments, m_is_synthetic == IsPartOfSyntheticConstructor::Yes);
  407. return dst;
  408. }
  409. Bytecode::CodeGenerationErrorOr<Optional<ScopedOperand>> AssignmentExpression::generate_bytecode(Bytecode::Generator& generator, Optional<ScopedOperand> preferred_dst) const
  410. {
  411. Bytecode::Generator::SourceLocationScope scope(generator, *this);
  412. if (m_op == AssignmentOp::Assignment) {
  413. // AssignmentExpression : LeftHandSideExpression = AssignmentExpression
  414. return m_lhs.visit(
  415. // 1. If LeftHandSideExpression is neither an ObjectLiteral nor an ArrayLiteral, then
  416. [&](NonnullRefPtr<Expression const> const& lhs) -> Bytecode::CodeGenerationErrorOr<Optional<ScopedOperand>> {
  417. // a. Let lref be the result of evaluating LeftHandSideExpression.
  418. // b. ReturnIfAbrupt(lref).
  419. Optional<ScopedOperand> base;
  420. Optional<ScopedOperand> computed_property;
  421. Optional<ScopedOperand> this_value;
  422. bool lhs_is_super_expression = false;
  423. if (is<MemberExpression>(*lhs)) {
  424. auto& expression = static_cast<MemberExpression const&>(*lhs);
  425. lhs_is_super_expression = is<SuperExpression>(expression.object());
  426. if (!lhs_is_super_expression) {
  427. base = TRY(expression.object().generate_bytecode(generator)).value();
  428. } else {
  429. // https://tc39.es/ecma262/#sec-super-keyword-runtime-semantics-evaluation
  430. // 1. Let env be GetThisEnvironment().
  431. // 2. Let actualThis be ? env.GetThisBinding().
  432. this_value = generator.get_this();
  433. // SuperProperty : super [ Expression ]
  434. // 3. Let propertyNameReference be ? Evaluation of Expression.
  435. // 4. Let propertyNameValue be ? GetValue(propertyNameReference).
  436. }
  437. if (expression.is_computed()) {
  438. auto property = TRY(expression.property().generate_bytecode(generator)).value();
  439. computed_property = generator.allocate_register();
  440. generator.emit<Bytecode::Op::Mov>(*computed_property, property);
  441. // To be continued later with PutByValue.
  442. } else if (expression.property().is_identifier()) {
  443. // Do nothing, this will be handled by PutById later.
  444. } else if (expression.property().is_private_identifier()) {
  445. // Do nothing, this will be handled by PutPrivateById later.
  446. } else {
  447. return Bytecode::CodeGenerationError {
  448. &expression,
  449. "Unimplemented non-computed member expression"sv
  450. };
  451. }
  452. if (lhs_is_super_expression) {
  453. // 5/7. Return ? MakeSuperPropertyReference(actualThis, propertyKey, strict).
  454. // https://tc39.es/ecma262/#sec-makesuperpropertyreference
  455. // 1. Let env be GetThisEnvironment().
  456. // 2. Assert: env.HasSuperBinding() is true.
  457. // 3. Let baseValue be ? env.GetSuperBase().
  458. // 4. Return the Reference Record { [[Base]]: baseValue, [[ReferencedName]]: propertyKey, [[Strict]]: strict, [[ThisValue]]: actualThis }.
  459. base = generator.allocate_register();
  460. generator.emit<Bytecode::Op::ResolveSuperBase>(*base);
  461. }
  462. } else if (is<Identifier>(*lhs)) {
  463. // NOTE: For Identifiers, we cannot perform GetVariable and then write into the reference it retrieves, only SetVariable can do this.
  464. // 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.
  465. } else {
  466. (void)TRY(lhs->generate_bytecode(generator));
  467. }
  468. // FIXME: c. If IsAnonymousFunctionDefinition(AssignmentExpression) and IsIdentifierRef of LeftHandSideExpression are both true, then
  469. // i. Let rval be ? NamedEvaluation of AssignmentExpression with argument lref.[[ReferencedName]].
  470. // d. Else,
  471. // i. Let rref be the result of evaluating AssignmentExpression.
  472. // ii. Let rval be ? GetValue(rref).
  473. auto rval = TRY([&]() -> Bytecode::CodeGenerationErrorOr<ScopedOperand> {
  474. if (lhs->is_identifier()) {
  475. return TRY(generator.emit_named_evaluation_if_anonymous_function(*m_rhs, generator.intern_identifier(static_cast<Identifier const&>(*lhs).string()))).value();
  476. } else {
  477. return TRY(m_rhs->generate_bytecode(generator)).value();
  478. }
  479. }());
  480. // e. Perform ? PutValue(lref, rval).
  481. if (is<Identifier>(*lhs)) {
  482. auto& identifier = static_cast<Identifier const&>(*lhs);
  483. generator.emit_set_variable(identifier, rval);
  484. } else if (is<MemberExpression>(*lhs)) {
  485. auto& expression = static_cast<MemberExpression const&>(*lhs);
  486. auto base_identifier = generator.intern_identifier_for_expression(expression.object());
  487. if (expression.is_computed()) {
  488. if (!lhs_is_super_expression)
  489. generator.emit<Bytecode::Op::PutByValue>(*base, *computed_property, rval, Bytecode::Op::PropertyKind::KeyValue, move(base_identifier));
  490. else
  491. generator.emit<Bytecode::Op::PutByValueWithThis>(*base, *computed_property, *this_value, rval);
  492. } else if (expression.property().is_identifier()) {
  493. auto identifier_table_ref = generator.intern_identifier(verify_cast<Identifier>(expression.property()).string());
  494. if (!lhs_is_super_expression)
  495. generator.emit<Bytecode::Op::PutById>(*base, identifier_table_ref, rval, Bytecode::Op::PropertyKind::KeyValue, generator.next_property_lookup_cache(), move(base_identifier));
  496. else
  497. generator.emit<Bytecode::Op::PutByIdWithThis>(*base, *this_value, identifier_table_ref, rval, Bytecode::Op::PropertyKind::KeyValue, generator.next_property_lookup_cache());
  498. } else if (expression.property().is_private_identifier()) {
  499. auto identifier_table_ref = generator.intern_identifier(verify_cast<PrivateIdentifier>(expression.property()).string());
  500. generator.emit<Bytecode::Op::PutPrivateById>(*base, identifier_table_ref, rval);
  501. } else {
  502. return Bytecode::CodeGenerationError {
  503. &expression,
  504. "Unimplemented non-computed member expression"sv
  505. };
  506. }
  507. } else {
  508. return Bytecode::CodeGenerationError {
  509. lhs,
  510. "Unimplemented/invalid node used a reference"sv
  511. };
  512. }
  513. // f. Return rval.
  514. return rval;
  515. },
  516. // 2. Let assignmentPattern be the AssignmentPattern that is covered by LeftHandSideExpression.
  517. [&](NonnullRefPtr<BindingPattern const> const& pattern) -> Bytecode::CodeGenerationErrorOr<Optional<ScopedOperand>> {
  518. // 3. Let rref be the result of evaluating AssignmentExpression.
  519. // 4. Let rval be ? GetValue(rref).
  520. auto rval = TRY(m_rhs->generate_bytecode(generator)).value();
  521. // 5. Perform ? DestructuringAssignmentEvaluation of assignmentPattern with argument rval.
  522. TRY(pattern->generate_bytecode(generator, Bytecode::Op::SetVariable::InitializationMode::Set, rval, false));
  523. // 6. Return rval.
  524. return rval;
  525. });
  526. }
  527. VERIFY(m_lhs.has<NonnullRefPtr<Expression const>>());
  528. auto& lhs_expression = m_lhs.get<NonnullRefPtr<Expression const>>();
  529. auto reference_operands = TRY(generator.emit_load_from_reference(lhs_expression));
  530. auto lhs = reference_operands.loaded_value.value();
  531. Bytecode::BasicBlock* rhs_block_ptr { nullptr };
  532. Bytecode::BasicBlock* lhs_block_ptr { nullptr };
  533. Bytecode::BasicBlock* end_block_ptr { nullptr };
  534. // Logical assignments short circuit.
  535. if (m_op == AssignmentOp::AndAssignment) { // &&=
  536. rhs_block_ptr = &generator.make_block();
  537. lhs_block_ptr = &generator.make_block();
  538. end_block_ptr = &generator.make_block();
  539. generator.emit_jump_if(
  540. lhs,
  541. Bytecode::Label { *rhs_block_ptr },
  542. Bytecode::Label { *lhs_block_ptr });
  543. } else if (m_op == AssignmentOp::OrAssignment) { // ||=
  544. rhs_block_ptr = &generator.make_block();
  545. lhs_block_ptr = &generator.make_block();
  546. end_block_ptr = &generator.make_block();
  547. generator.emit_jump_if(
  548. lhs,
  549. Bytecode::Label { *lhs_block_ptr },
  550. Bytecode::Label { *rhs_block_ptr });
  551. } else if (m_op == AssignmentOp::NullishAssignment) { // ??=
  552. rhs_block_ptr = &generator.make_block();
  553. lhs_block_ptr = &generator.make_block();
  554. end_block_ptr = &generator.make_block();
  555. generator.emit<Bytecode::Op::JumpNullish>(
  556. lhs,
  557. Bytecode::Label { *rhs_block_ptr },
  558. Bytecode::Label { *lhs_block_ptr });
  559. }
  560. if (rhs_block_ptr)
  561. generator.switch_to_basic_block(*rhs_block_ptr);
  562. auto rhs = TRY([&]() -> Bytecode::CodeGenerationErrorOr<ScopedOperand> {
  563. if (lhs_expression->is_identifier()) {
  564. return TRY(generator.emit_named_evaluation_if_anonymous_function(*m_rhs, generator.intern_identifier(static_cast<Identifier const&>(*lhs_expression).string()))).value();
  565. }
  566. return TRY(m_rhs->generate_bytecode(generator)).value();
  567. }());
  568. auto dst = choose_dst(generator, preferred_dst);
  569. switch (m_op) {
  570. case AssignmentOp::AdditionAssignment:
  571. generator.emit<Bytecode::Op::Add>(dst, lhs, rhs);
  572. break;
  573. case AssignmentOp::SubtractionAssignment:
  574. generator.emit<Bytecode::Op::Sub>(dst, lhs, rhs);
  575. break;
  576. case AssignmentOp::MultiplicationAssignment:
  577. generator.emit<Bytecode::Op::Mul>(dst, lhs, rhs);
  578. break;
  579. case AssignmentOp::DivisionAssignment:
  580. generator.emit<Bytecode::Op::Div>(dst, lhs, rhs);
  581. break;
  582. case AssignmentOp::ModuloAssignment:
  583. generator.emit<Bytecode::Op::Mod>(dst, lhs, rhs);
  584. break;
  585. case AssignmentOp::ExponentiationAssignment:
  586. generator.emit<Bytecode::Op::Exp>(dst, lhs, rhs);
  587. break;
  588. case AssignmentOp::BitwiseAndAssignment:
  589. generator.emit<Bytecode::Op::BitwiseAnd>(dst, lhs, rhs);
  590. break;
  591. case AssignmentOp::BitwiseOrAssignment:
  592. generator.emit<Bytecode::Op::BitwiseOr>(dst, lhs, rhs);
  593. break;
  594. case AssignmentOp::BitwiseXorAssignment:
  595. generator.emit<Bytecode::Op::BitwiseXor>(dst, lhs, rhs);
  596. break;
  597. case AssignmentOp::LeftShiftAssignment:
  598. generator.emit<Bytecode::Op::LeftShift>(dst, lhs, rhs);
  599. break;
  600. case AssignmentOp::RightShiftAssignment:
  601. generator.emit<Bytecode::Op::RightShift>(dst, lhs, rhs);
  602. break;
  603. case AssignmentOp::UnsignedRightShiftAssignment:
  604. generator.emit<Bytecode::Op::UnsignedRightShift>(dst, lhs, rhs);
  605. break;
  606. case AssignmentOp::AndAssignment:
  607. case AssignmentOp::OrAssignment:
  608. case AssignmentOp::NullishAssignment:
  609. generator.emit<Bytecode::Op::Mov>(dst, rhs);
  610. break;
  611. default:
  612. return Bytecode::CodeGenerationError {
  613. this,
  614. "Unimplemented operation"sv,
  615. };
  616. }
  617. if (lhs_expression->is_identifier())
  618. generator.emit_set_variable(static_cast<Identifier const&>(*lhs_expression), dst);
  619. else
  620. (void)TRY(generator.emit_store_to_reference(reference_operands, dst));
  621. if (rhs_block_ptr) {
  622. generator.emit<Bytecode::Op::Jump>(Bytecode::Label { *end_block_ptr });
  623. }
  624. if (lhs_block_ptr) {
  625. generator.switch_to_basic_block(*lhs_block_ptr);
  626. generator.emit<Bytecode::Op::Mov>(dst, lhs);
  627. generator.emit<Bytecode::Op::Jump>(Bytecode::Label { *end_block_ptr });
  628. }
  629. if (end_block_ptr) {
  630. generator.switch_to_basic_block(*end_block_ptr);
  631. }
  632. return dst;
  633. }
  634. // 14.13.3 Runtime Semantics: Evaluation, https://tc39.es/ecma262/#sec-labelled-statements-runtime-semantics-evaluation
  635. // LabelledStatement : LabelIdentifier : LabelledItem
  636. Bytecode::CodeGenerationErrorOr<Optional<ScopedOperand>> LabelledStatement::generate_bytecode(Bytecode::Generator& generator, [[maybe_unused]] Optional<ScopedOperand> preferred_dst) const
  637. {
  638. Bytecode::Generator::SourceLocationScope scope(generator, *this);
  639. // Return ? LabelledEvaluation of this LabelledStatement with argument « ».
  640. return generate_labelled_evaluation(generator, {});
  641. }
  642. // 14.13.4 Runtime Semantics: LabelledEvaluation, https://tc39.es/ecma262/#sec-runtime-semantics-labelledevaluation
  643. // LabelledStatement : LabelIdentifier : LabelledItem
  644. Bytecode::CodeGenerationErrorOr<Optional<ScopedOperand>> LabelledStatement::generate_labelled_evaluation(Bytecode::Generator& generator, Vector<DeprecatedFlyString> const& label_set, [[maybe_unused]] Optional<ScopedOperand> preferred_dst) const
  645. {
  646. Bytecode::Generator::SourceLocationScope scope(generator, *this);
  647. // 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.
  648. auto const& labelled_item = *m_labelled_item;
  649. // 1. Let label be the StringValue of LabelIdentifier.
  650. // NOTE: Not necessary, this is m_label.
  651. // 2. Let newLabelSet be the list-concatenation of labelSet and « label ».
  652. // FIXME: Avoid copy here.
  653. auto new_label_set = label_set;
  654. new_label_set.append(m_label);
  655. // 3. Let stmtResult be LabelledEvaluation of LabelledItem with argument newLabelSet.
  656. Optional<ScopedOperand> stmt_result;
  657. if (is<IterationStatement>(labelled_item)) {
  658. auto const& iteration_statement = static_cast<IterationStatement const&>(labelled_item);
  659. stmt_result = TRY(iteration_statement.generate_labelled_evaluation(generator, new_label_set));
  660. } else if (is<SwitchStatement>(labelled_item)) {
  661. auto const& switch_statement = static_cast<SwitchStatement const&>(labelled_item);
  662. stmt_result = TRY(switch_statement.generate_labelled_evaluation(generator, new_label_set));
  663. } else if (is<LabelledStatement>(labelled_item)) {
  664. auto const& labelled_statement = static_cast<LabelledStatement const&>(labelled_item);
  665. stmt_result = TRY(labelled_statement.generate_labelled_evaluation(generator, new_label_set));
  666. } else {
  667. auto& labelled_break_block = generator.make_block();
  668. // NOTE: We do not need a continuable scope as `continue;` is not allowed outside of iteration statements, throwing a SyntaxError in the parser.
  669. generator.begin_breakable_scope(Bytecode::Label { labelled_break_block }, new_label_set);
  670. stmt_result = TRY(labelled_item.generate_bytecode(generator));
  671. generator.end_breakable_scope();
  672. if (!generator.is_current_block_terminated()) {
  673. generator.emit<Bytecode::Op::Jump>(Bytecode::Label { labelled_break_block });
  674. }
  675. generator.switch_to_basic_block(labelled_break_block);
  676. }
  677. // 4. If stmtResult.[[Type]] is break and SameValue(stmtResult.[[Target]], label) is true, then
  678. // a. Set stmtResult to NormalCompletion(stmtResult.[[Value]]).
  679. // 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.
  680. // 5. Return Completion(stmtResult).
  681. return stmt_result;
  682. }
  683. Bytecode::CodeGenerationErrorOr<Optional<ScopedOperand>> IterationStatement::generate_labelled_evaluation(Bytecode::Generator&, Vector<DeprecatedFlyString> const&, [[maybe_unused]] Optional<ScopedOperand> preferred_dst) const
  684. {
  685. return Bytecode::CodeGenerationError {
  686. this,
  687. "Missing generate_labelled_evaluation()"sv,
  688. };
  689. }
  690. Bytecode::CodeGenerationErrorOr<Optional<ScopedOperand>> WhileStatement::generate_bytecode(Bytecode::Generator& generator, [[maybe_unused]] Optional<ScopedOperand> preferred_dst) const
  691. {
  692. Bytecode::Generator::SourceLocationScope scope(generator, *this);
  693. return generate_labelled_evaluation(generator, {});
  694. }
  695. Bytecode::CodeGenerationErrorOr<Optional<ScopedOperand>> WhileStatement::generate_labelled_evaluation(Bytecode::Generator& generator, Vector<DeprecatedFlyString> const& label_set, [[maybe_unused]] Optional<ScopedOperand> preferred_dst) const
  696. {
  697. Bytecode::Generator::SourceLocationScope scope(generator, *this);
  698. // test
  699. // jump if_false (true) end (false) body
  700. // body
  701. // jump always (true) test
  702. // end
  703. auto& test_block = generator.make_block();
  704. auto& body_block = generator.make_block();
  705. auto& end_block = generator.make_block();
  706. auto result = generator.allocate_register();
  707. generator.emit<Bytecode::Op::Mov>(result, generator.add_constant(js_undefined()));
  708. generator.emit<Bytecode::Op::Jump>(Bytecode::Label { test_block });
  709. generator.switch_to_basic_block(test_block);
  710. auto test = TRY(m_test->generate_bytecode(generator)).value();
  711. generator.emit_jump_if(
  712. test,
  713. Bytecode::Label { body_block },
  714. Bytecode::Label { end_block });
  715. generator.switch_to_basic_block(body_block);
  716. generator.begin_continuable_scope(Bytecode::Label { test_block }, label_set);
  717. generator.begin_breakable_scope(Bytecode::Label { end_block }, label_set);
  718. auto body = TRY(m_body->generate_bytecode(generator));
  719. generator.end_breakable_scope();
  720. generator.end_continuable_scope();
  721. if (!generator.is_current_block_terminated()) {
  722. if (body.has_value())
  723. generator.emit<Bytecode::Op::Mov>(result, body.value());
  724. generator.emit<Bytecode::Op::Jump>(Bytecode::Label { test_block });
  725. }
  726. generator.switch_to_basic_block(end_block);
  727. return result;
  728. }
  729. Bytecode::CodeGenerationErrorOr<Optional<ScopedOperand>> DoWhileStatement::generate_bytecode(Bytecode::Generator& generator, [[maybe_unused]] Optional<ScopedOperand> preferred_dst) const
  730. {
  731. Bytecode::Generator::SourceLocationScope scope(generator, *this);
  732. return generate_labelled_evaluation(generator, {});
  733. }
  734. Bytecode::CodeGenerationErrorOr<Optional<ScopedOperand>> DoWhileStatement::generate_labelled_evaluation(Bytecode::Generator& generator, Vector<DeprecatedFlyString> const& label_set, [[maybe_unused]] Optional<ScopedOperand> preferred_dst) const
  735. {
  736. Bytecode::Generator::SourceLocationScope scope(generator, *this);
  737. // jump always (true) body
  738. // test
  739. // jump if_false (true) end (false) body
  740. // body
  741. // jump always (true) test
  742. // end
  743. auto& body_block = generator.make_block();
  744. auto& test_block = generator.make_block();
  745. auto& load_result_and_jump_to_end_block = generator.make_block();
  746. auto& end_block = generator.make_block();
  747. auto completion_value = generator.allocate_register();
  748. generator.emit<Bytecode::Op::Mov>(completion_value, generator.add_constant(js_undefined()));
  749. // jump to the body block
  750. generator.emit<Bytecode::Op::Jump>(Bytecode::Label { body_block });
  751. generator.switch_to_basic_block(test_block);
  752. auto test = TRY(m_test->generate_bytecode(generator)).value();
  753. generator.emit_jump_if(
  754. test,
  755. Bytecode::Label { body_block },
  756. Bytecode::Label { load_result_and_jump_to_end_block });
  757. generator.switch_to_basic_block(body_block);
  758. generator.begin_continuable_scope(Bytecode::Label { test_block }, label_set);
  759. generator.begin_breakable_scope(Bytecode::Label { end_block }, label_set);
  760. auto body_result = TRY(m_body->generate_bytecode(generator));
  761. generator.end_breakable_scope();
  762. generator.end_continuable_scope();
  763. if (!generator.is_current_block_terminated()) {
  764. if (body_result.has_value())
  765. generator.emit<Bytecode::Op::Mov>(completion_value, body_result.value());
  766. generator.emit<Bytecode::Op::Jump>(Bytecode::Label { test_block });
  767. }
  768. generator.switch_to_basic_block(load_result_and_jump_to_end_block);
  769. generator.emit<Bytecode::Op::Jump>(Bytecode::Label { end_block });
  770. generator.switch_to_basic_block(end_block);
  771. return completion_value;
  772. }
  773. Bytecode::CodeGenerationErrorOr<Optional<ScopedOperand>> ForStatement::generate_bytecode(Bytecode::Generator& generator, [[maybe_unused]] Optional<ScopedOperand> preferred_dst) const
  774. {
  775. Bytecode::Generator::SourceLocationScope scope(generator, *this);
  776. return generate_labelled_evaluation(generator, {});
  777. }
  778. Bytecode::CodeGenerationErrorOr<Optional<ScopedOperand>> ForStatement::generate_labelled_evaluation(Bytecode::Generator& generator, Vector<DeprecatedFlyString> const& label_set, [[maybe_unused]] Optional<ScopedOperand> preferred_dst) const
  779. {
  780. Bytecode::Generator::SourceLocationScope scope(generator, *this);
  781. // init
  782. // jump always (true) test
  783. // test
  784. // jump if_true (true) body (false) end
  785. // body
  786. // jump always (true) update
  787. // update
  788. // jump always (true) test
  789. // end
  790. // If 'test' is missing, fuse the 'test' and 'body' basic blocks
  791. // If 'update' is missing, fuse the 'body' and 'update' basic blocks
  792. Bytecode::BasicBlock* test_block_ptr { nullptr };
  793. Bytecode::BasicBlock* body_block_ptr { nullptr };
  794. Bytecode::BasicBlock* update_block_ptr { nullptr };
  795. bool has_lexical_environment = false;
  796. if (m_init) {
  797. if (m_init->is_variable_declaration()) {
  798. auto& variable_declaration = verify_cast<VariableDeclaration>(*m_init);
  799. auto has_non_local_variables = false;
  800. MUST(variable_declaration.for_each_bound_identifier([&](auto const& identifier) {
  801. if (!identifier.is_local())
  802. has_non_local_variables = true;
  803. }));
  804. if (variable_declaration.is_lexical_declaration() && has_non_local_variables) {
  805. has_lexical_environment = true;
  806. // FIXME: Is Block correct?
  807. generator.begin_variable_scope();
  808. bool is_const = variable_declaration.is_constant_declaration();
  809. // NOTE: Nothing in the callback throws an exception.
  810. MUST(variable_declaration.for_each_bound_identifier([&](auto const& identifier) {
  811. if (identifier.is_local())
  812. return;
  813. auto index = generator.intern_identifier(identifier.string());
  814. generator.emit<Bytecode::Op::CreateVariable>(index, Bytecode::Op::EnvironmentMode::Lexical, is_const);
  815. }));
  816. }
  817. }
  818. (void)TRY(m_init->generate_bytecode(generator));
  819. }
  820. body_block_ptr = &generator.make_block();
  821. if (m_update)
  822. update_block_ptr = &generator.make_block();
  823. else
  824. update_block_ptr = body_block_ptr;
  825. if (m_test)
  826. test_block_ptr = &generator.make_block();
  827. else
  828. test_block_ptr = body_block_ptr;
  829. auto& end_block = generator.make_block();
  830. generator.emit<Bytecode::Op::Jump>(Bytecode::Label { *test_block_ptr });
  831. if (m_test) {
  832. generator.switch_to_basic_block(*test_block_ptr);
  833. auto test = TRY(m_test->generate_bytecode(generator)).value();
  834. generator.emit_jump_if(test, Bytecode::Label { *body_block_ptr }, Bytecode::Label { end_block });
  835. }
  836. if (m_update) {
  837. generator.switch_to_basic_block(*update_block_ptr);
  838. (void)TRY(m_update->generate_bytecode(generator));
  839. generator.emit<Bytecode::Op::Jump>(Bytecode::Label { *test_block_ptr });
  840. }
  841. generator.switch_to_basic_block(*body_block_ptr);
  842. generator.begin_continuable_scope(Bytecode::Label { m_update ? *update_block_ptr : *test_block_ptr }, label_set);
  843. generator.begin_breakable_scope(Bytecode::Label { end_block }, label_set);
  844. auto body_result = TRY(m_body->generate_bytecode(generator));
  845. generator.end_breakable_scope();
  846. generator.end_continuable_scope();
  847. if (!generator.is_current_block_terminated()) {
  848. if (m_update) {
  849. generator.emit<Bytecode::Op::Jump>(Bytecode::Label { *update_block_ptr });
  850. } else {
  851. generator.emit<Bytecode::Op::Jump>(Bytecode::Label { *test_block_ptr });
  852. }
  853. }
  854. generator.switch_to_basic_block(end_block);
  855. if (has_lexical_environment)
  856. generator.end_variable_scope();
  857. return body_result;
  858. }
  859. Bytecode::CodeGenerationErrorOr<Optional<ScopedOperand>> ObjectExpression::generate_bytecode(Bytecode::Generator& generator, [[maybe_unused]] Optional<ScopedOperand> preferred_dst) const
  860. {
  861. Bytecode::Generator::SourceLocationScope scope(generator, *this);
  862. auto object = generator.allocate_register();
  863. generator.emit<Bytecode::Op::NewObject>(object);
  864. if (m_properties.is_empty())
  865. return object;
  866. generator.push_home_object(object);
  867. for (auto& property : m_properties) {
  868. Bytecode::Op::PropertyKind property_kind;
  869. switch (property->type()) {
  870. case ObjectProperty::Type::KeyValue:
  871. property_kind = Bytecode::Op::PropertyKind::DirectKeyValue;
  872. break;
  873. case ObjectProperty::Type::Getter:
  874. property_kind = Bytecode::Op::PropertyKind::Getter;
  875. break;
  876. case ObjectProperty::Type::Setter:
  877. property_kind = Bytecode::Op::PropertyKind::Setter;
  878. break;
  879. case ObjectProperty::Type::Spread:
  880. property_kind = Bytecode::Op::PropertyKind::Spread;
  881. break;
  882. case ObjectProperty::Type::ProtoSetter:
  883. property_kind = Bytecode::Op::PropertyKind::ProtoSetter;
  884. break;
  885. }
  886. if (is<StringLiteral>(property->key())) {
  887. auto& string_literal = static_cast<StringLiteral const&>(property->key());
  888. Bytecode::IdentifierTableIndex key_name = generator.intern_identifier(string_literal.value());
  889. Optional<ScopedOperand> value;
  890. if (property_kind == Bytecode::Op::PropertyKind::ProtoSetter) {
  891. value = TRY(property->value().generate_bytecode(generator)).value();
  892. } else if (property_kind != Bytecode::Op::PropertyKind::Spread) {
  893. ByteString identifier = string_literal.value();
  894. if (property_kind == Bytecode::Op::PropertyKind::Getter)
  895. identifier = ByteString::formatted("get {}", identifier);
  896. else if (property_kind == Bytecode::Op::PropertyKind::Setter)
  897. identifier = ByteString::formatted("set {}", identifier);
  898. auto name = generator.intern_identifier(identifier);
  899. value = TRY(generator.emit_named_evaluation_if_anonymous_function(property->value(), name)).value();
  900. } else {
  901. // Spread the key.
  902. value = TRY(property->key().generate_bytecode(generator)).value();
  903. }
  904. generator.emit<Bytecode::Op::PutById>(object, key_name, *value, property_kind, generator.next_property_lookup_cache());
  905. } else {
  906. auto property_name = TRY(property->key().generate_bytecode(generator)).value();
  907. Optional<ScopedOperand> value;
  908. if (property_kind != Bytecode::Op::PropertyKind::Spread)
  909. value = TRY(property->value().generate_bytecode(generator)).value();
  910. else
  911. value = property_name;
  912. generator.emit<Bytecode::Op::PutByValue>(object, property_name, *value, property_kind);
  913. }
  914. }
  915. generator.pop_home_object();
  916. return object;
  917. }
  918. Bytecode::CodeGenerationErrorOr<Optional<ScopedOperand>> ArrayExpression::generate_bytecode(Bytecode::Generator& generator, Optional<ScopedOperand> preferred_dst) const
  919. {
  920. Bytecode::Generator::SourceLocationScope scope(generator, *this);
  921. if (m_elements.is_empty()) {
  922. auto dst = choose_dst(generator, preferred_dst);
  923. generator.emit<Bytecode::Op::NewArray>(dst);
  924. return dst;
  925. }
  926. if (all_of(m_elements, [](auto element) { return !element || is<PrimitiveLiteral>(*element); })) {
  927. // If all elements are constant primitives, we can just emit a single instruction to initialize the array,
  928. // instead of emitting instructions to manually evaluate them one-by-one
  929. Vector<Value> values;
  930. values.resize(m_elements.size());
  931. for (auto i = 0u; i < m_elements.size(); ++i) {
  932. if (!m_elements[i])
  933. continue;
  934. values[i] = static_cast<PrimitiveLiteral const&>(*m_elements[i]).value();
  935. }
  936. auto dst = choose_dst(generator, preferred_dst);
  937. generator.emit_with_extra_value_slots<Bytecode::Op::NewPrimitiveArray>(values.size(), dst, values);
  938. return dst;
  939. }
  940. auto first_spread = find_if(m_elements.begin(), m_elements.end(), [](auto el) { return el && is<SpreadExpression>(*el); });
  941. Vector<ScopedOperand> args;
  942. args.ensure_capacity(m_elements.size());
  943. for (auto it = m_elements.begin(); it != first_spread; ++it) {
  944. if (*it) {
  945. auto value = TRY((*it)->generate_bytecode(generator)).value();
  946. // NOTE: We don't need to protect temporary registers or constants from being clobbered.
  947. if (value.operand().is_register() || value.operand().is_constant()) {
  948. args.append(move(value));
  949. } else {
  950. auto reg = generator.allocate_register();
  951. generator.emit<Bytecode::Op::Mov>(Bytecode::Operand(reg), value);
  952. args.append(move(reg));
  953. }
  954. } else {
  955. args.append(generator.add_constant(Value()));
  956. }
  957. }
  958. auto dst = choose_dst(generator, preferred_dst);
  959. if (first_spread.index() != 0) {
  960. generator.emit_with_extra_operand_slots<Bytecode::Op::NewArray>(args.size(), dst, args);
  961. } else {
  962. generator.emit<Bytecode::Op::NewArray>(dst);
  963. }
  964. if (first_spread != m_elements.end()) {
  965. for (auto it = first_spread; it != m_elements.end(); ++it) {
  966. if (!*it) {
  967. generator.emit<Bytecode::Op::ArrayAppend>(dst, generator.add_constant(Value()), false);
  968. } else {
  969. auto value = TRY((*it)->generate_bytecode(generator)).value();
  970. generator.emit<Bytecode::Op::ArrayAppend>(dst, value, *it && is<SpreadExpression>(**it));
  971. }
  972. }
  973. }
  974. return dst;
  975. }
  976. Bytecode::CodeGenerationErrorOr<Optional<ScopedOperand>> MemberExpression::generate_bytecode(Bytecode::Generator& generator, Optional<ScopedOperand> preferred_dst) const
  977. {
  978. Bytecode::Generator::SourceLocationScope scope(generator, *this);
  979. auto reference = TRY(generator.emit_load_from_reference(*this, preferred_dst));
  980. return reference.loaded_value;
  981. }
  982. Bytecode::CodeGenerationErrorOr<Optional<ScopedOperand>> FunctionDeclaration::generate_bytecode(Bytecode::Generator& generator, [[maybe_unused]] Optional<ScopedOperand> preferred_dst) const
  983. {
  984. if (m_is_hoisted) {
  985. Bytecode::Generator::SourceLocationScope scope(generator, *this);
  986. auto index = generator.intern_identifier(name());
  987. auto value = generator.allocate_register();
  988. generator.emit<Bytecode::Op::GetVariable>(value, index, generator.next_environment_variable_cache());
  989. generator.emit<Bytecode::Op::SetVariable>(index, value, generator.next_environment_variable_cache(), Bytecode::Op::SetVariable::InitializationMode::Set, Bytecode::Op::EnvironmentMode::Var);
  990. }
  991. return Optional<ScopedOperand> {};
  992. }
  993. Bytecode::CodeGenerationErrorOr<Optional<ScopedOperand>> FunctionExpression::generate_bytecode_with_lhs_name(Bytecode::Generator& generator, Optional<Bytecode::IdentifierTableIndex> lhs_name, Optional<ScopedOperand> preferred_dst) const
  994. {
  995. Bytecode::Generator::SourceLocationScope scope(generator, *this);
  996. bool has_name = !name().is_empty();
  997. Optional<Bytecode::IdentifierTableIndex> name_identifier;
  998. if (has_name) {
  999. generator.begin_variable_scope();
  1000. name_identifier = generator.intern_identifier(name());
  1001. generator.emit<Bytecode::Op::CreateVariable>(*name_identifier, Bytecode::Op::EnvironmentMode::Lexical, true);
  1002. }
  1003. auto new_function = choose_dst(generator, preferred_dst);
  1004. generator.emit_new_function(new_function, *this, lhs_name);
  1005. if (has_name) {
  1006. generator.emit<Bytecode::Op::SetVariable>(*name_identifier, new_function, generator.next_environment_variable_cache(), Bytecode::Op::SetVariable::InitializationMode::Initialize, Bytecode::Op::EnvironmentMode::Lexical);
  1007. generator.end_variable_scope();
  1008. }
  1009. return new_function;
  1010. }
  1011. Bytecode::CodeGenerationErrorOr<Optional<ScopedOperand>> FunctionExpression::generate_bytecode(Bytecode::Generator& generator, Optional<ScopedOperand> preferred_dst) const
  1012. {
  1013. Bytecode::Generator::SourceLocationScope scope(generator, *this);
  1014. return generate_bytecode_with_lhs_name(generator, {}, preferred_dst);
  1015. }
  1016. static Bytecode::CodeGenerationErrorOr<void> generate_object_binding_pattern_bytecode(Bytecode::Generator& generator, BindingPattern const& pattern, Bytecode::Op::SetVariable::InitializationMode initialization_mode, ScopedOperand const& object, bool create_variables)
  1017. {
  1018. generator.emit<Bytecode::Op::ThrowIfNullish>(object);
  1019. Vector<ScopedOperand> excluded_property_names;
  1020. auto has_rest = false;
  1021. if (pattern.entries.size() > 0)
  1022. has_rest = pattern.entries[pattern.entries.size() - 1].is_rest;
  1023. for (auto& [name, alias, initializer, is_rest] : pattern.entries) {
  1024. if (is_rest) {
  1025. VERIFY(!initializer);
  1026. if (name.has<NonnullRefPtr<Identifier const>>()) {
  1027. auto identifier = name.get<NonnullRefPtr<Identifier const>>();
  1028. auto interned_identifier = generator.intern_identifier(identifier->string());
  1029. auto copy = generator.allocate_register();
  1030. generator.emit_with_extra_operand_slots<Bytecode::Op::CopyObjectExcludingProperties>(
  1031. excluded_property_names.size(), copy, object, excluded_property_names);
  1032. if (create_variables) {
  1033. VERIFY(!identifier->is_local());
  1034. generator.emit<Bytecode::Op::CreateVariable>(interned_identifier, Bytecode::Op::EnvironmentMode::Lexical, false);
  1035. }
  1036. generator.emit_set_variable(*identifier, copy, initialization_mode);
  1037. return {};
  1038. }
  1039. if (alias.has<NonnullRefPtr<MemberExpression const>>()) {
  1040. auto copy = generator.allocate_register();
  1041. generator.emit_with_extra_operand_slots<Bytecode::Op::CopyObjectExcludingProperties>(
  1042. excluded_property_names.size(), copy, object, excluded_property_names);
  1043. (void)TRY(generator.emit_store_to_reference(alias.get<NonnullRefPtr<MemberExpression const>>(), object));
  1044. return {};
  1045. }
  1046. VERIFY_NOT_REACHED();
  1047. }
  1048. auto value = generator.allocate_register();
  1049. if (name.has<NonnullRefPtr<Identifier const>>()) {
  1050. auto const& identifier = name.get<NonnullRefPtr<Identifier const>>()->string();
  1051. if (has_rest) {
  1052. excluded_property_names.append(generator.add_constant(PrimitiveString::create(generator.vm(), identifier), Bytecode::Generator::DeduplicateConstant::No));
  1053. }
  1054. generator.emit_get_by_id(value, object, generator.intern_identifier(identifier));
  1055. } else {
  1056. auto expression = name.get<NonnullRefPtr<Expression const>>();
  1057. auto property_name = TRY(expression->generate_bytecode(generator)).value();
  1058. if (has_rest) {
  1059. auto excluded_name = generator.allocate_register();
  1060. excluded_property_names.append(excluded_name);
  1061. generator.emit<Bytecode::Op::Mov>(excluded_name, property_name);
  1062. }
  1063. generator.emit<Bytecode::Op::GetByValue>(value, object, property_name);
  1064. }
  1065. if (initializer) {
  1066. auto& if_undefined_block = generator.make_block();
  1067. auto& if_not_undefined_block = generator.make_block();
  1068. generator.emit<Bytecode::Op::JumpUndefined>(
  1069. value,
  1070. Bytecode::Label { if_undefined_block },
  1071. Bytecode::Label { if_not_undefined_block });
  1072. generator.switch_to_basic_block(if_undefined_block);
  1073. Optional<ScopedOperand> default_value;
  1074. if (auto const* alias_identifier = alias.get_pointer<NonnullRefPtr<Identifier const>>()) {
  1075. default_value = TRY(generator.emit_named_evaluation_if_anonymous_function(*initializer, generator.intern_identifier((*alias_identifier)->string()))).value();
  1076. } else if (auto const* lhs = name.get_pointer<NonnullRefPtr<Identifier const>>()) {
  1077. default_value = TRY(generator.emit_named_evaluation_if_anonymous_function(*initializer, generator.intern_identifier((*lhs)->string()))).value();
  1078. } else {
  1079. default_value = TRY(initializer->generate_bytecode(generator)).value();
  1080. }
  1081. generator.emit<Bytecode::Op::Mov>(value, *default_value);
  1082. generator.emit<Bytecode::Op::Jump>(Bytecode::Label { if_not_undefined_block });
  1083. generator.switch_to_basic_block(if_not_undefined_block);
  1084. }
  1085. if (alias.has<NonnullRefPtr<BindingPattern const>>()) {
  1086. auto& binding_pattern = *alias.get<NonnullRefPtr<BindingPattern const>>();
  1087. auto nested_value = generator.allocate_register();
  1088. generator.emit<Bytecode::Op::Mov>(nested_value, value);
  1089. TRY(binding_pattern.generate_bytecode(generator, initialization_mode, nested_value, create_variables));
  1090. } else if (alias.has<Empty>()) {
  1091. if (name.has<NonnullRefPtr<Expression const>>()) {
  1092. // This needs some sort of SetVariableByValue opcode, as it's a runtime binding
  1093. return Bytecode::CodeGenerationError {
  1094. name.get<NonnullRefPtr<Expression const>>().ptr(),
  1095. "Unimplemented name/alias pair: Empty/Expression"sv,
  1096. };
  1097. }
  1098. auto const& identifier = *name.get<NonnullRefPtr<Identifier const>>();
  1099. auto identifier_ref = generator.intern_identifier(identifier.string());
  1100. if (create_variables)
  1101. generator.emit<Bytecode::Op::CreateVariable>(identifier_ref, Bytecode::Op::EnvironmentMode::Lexical, false);
  1102. generator.emit_set_variable(identifier, value, initialization_mode);
  1103. } else if (alias.has<NonnullRefPtr<MemberExpression const>>()) {
  1104. TRY(generator.emit_store_to_reference(alias.get<NonnullRefPtr<MemberExpression const>>(), value));
  1105. } else {
  1106. auto const& identifier = *alias.get<NonnullRefPtr<Identifier const>>();
  1107. auto identifier_ref = generator.intern_identifier(identifier.string());
  1108. if (create_variables)
  1109. generator.emit<Bytecode::Op::CreateVariable>(identifier_ref, Bytecode::Op::EnvironmentMode::Lexical, false);
  1110. generator.emit_set_variable(identifier, value, initialization_mode);
  1111. }
  1112. }
  1113. return {};
  1114. }
  1115. static Bytecode::CodeGenerationErrorOr<void> generate_array_binding_pattern_bytecode(Bytecode::Generator& generator, BindingPattern const& pattern, Bytecode::Op::SetVariable::InitializationMode initialization_mode, ScopedOperand const& input_array, bool create_variables, [[maybe_unused]] Optional<ScopedOperand> preferred_dst = {})
  1116. {
  1117. /*
  1118. * Consider the following destructuring assignment:
  1119. *
  1120. * let [a, b, c, d, e] = o;
  1121. *
  1122. * It would be fairly trivial to just loop through this iterator, getting the value
  1123. * at each step and assigning them to the binding sequentially. However, this is not
  1124. * correct: once an iterator is exhausted, it must not be called again. This complicates
  1125. * the bytecode. In order to accomplish this, we do the following:
  1126. *
  1127. * - Reserve a special boolean register which holds 'true' if the iterator is exhausted,
  1128. * and false otherwise
  1129. * - When we are retrieving the value which should be bound, we first check this register.
  1130. * If it is 'true', we load undefined. Otherwise, we grab the next value from the iterator.
  1131. *
  1132. * Note that the is_exhausted register does not need to be loaded with false because the
  1133. * first IteratorNext bytecode is _not_ proceeded by an exhausted check, as it is
  1134. * unnecessary.
  1135. */
  1136. auto is_iterator_exhausted = generator.allocate_register();
  1137. generator.emit<Bytecode::Op::Mov>(is_iterator_exhausted, generator.add_constant(Value(false)));
  1138. auto iterator = generator.allocate_register();
  1139. generator.emit<Bytecode::Op::GetIterator>(iterator, input_array);
  1140. bool first = true;
  1141. auto assign_value_to_alias = [&](auto& alias, ScopedOperand value) {
  1142. return alias.visit(
  1143. [&](Empty) -> Bytecode::CodeGenerationErrorOr<void> {
  1144. // This element is an elision
  1145. return {};
  1146. },
  1147. [&](NonnullRefPtr<Identifier const> const& identifier) -> Bytecode::CodeGenerationErrorOr<void> {
  1148. auto interned_index = generator.intern_identifier(identifier->string());
  1149. if (create_variables)
  1150. generator.emit<Bytecode::Op::CreateVariable>(interned_index, Bytecode::Op::EnvironmentMode::Lexical, false);
  1151. generator.emit_set_variable(*identifier, value, initialization_mode);
  1152. return {};
  1153. },
  1154. [&](NonnullRefPtr<BindingPattern const> const& pattern) -> Bytecode::CodeGenerationErrorOr<void> {
  1155. return pattern->generate_bytecode(generator, initialization_mode, value, create_variables);
  1156. },
  1157. [&](NonnullRefPtr<MemberExpression const> const& expr) -> Bytecode::CodeGenerationErrorOr<void> {
  1158. (void)generator.emit_store_to_reference(*expr, value);
  1159. return {};
  1160. });
  1161. };
  1162. auto temp_iterator_result = generator.allocate_register();
  1163. for (auto& [name, alias, initializer, is_rest] : pattern.entries) {
  1164. VERIFY(name.has<Empty>());
  1165. if (is_rest) {
  1166. VERIFY(!initializer);
  1167. auto value = generator.allocate_register();
  1168. if (first) {
  1169. // The iterator has not been called, and is thus known to be not exhausted
  1170. generator.emit<Bytecode::Op::IteratorToArray>(value, iterator);
  1171. } else {
  1172. auto& if_exhausted_block = generator.make_block();
  1173. auto& if_not_exhausted_block = generator.make_block();
  1174. auto& continuation_block = generator.make_block();
  1175. generator.emit_jump_if(
  1176. is_iterator_exhausted,
  1177. Bytecode::Label { if_exhausted_block },
  1178. Bytecode::Label { if_not_exhausted_block });
  1179. value = generator.allocate_register();
  1180. generator.switch_to_basic_block(if_exhausted_block);
  1181. generator.emit<Bytecode::Op::NewArray>(value);
  1182. generator.emit<Bytecode::Op::Jump>(Bytecode::Label { continuation_block });
  1183. generator.switch_to_basic_block(if_not_exhausted_block);
  1184. generator.emit<Bytecode::Op::IteratorToArray>(value, iterator);
  1185. generator.emit<Bytecode::Op::Jump>(Bytecode::Label { continuation_block });
  1186. generator.switch_to_basic_block(continuation_block);
  1187. }
  1188. return assign_value_to_alias(alias, value);
  1189. }
  1190. auto& iterator_is_exhausted_block = generator.make_block();
  1191. if (!first) {
  1192. auto& iterator_is_not_exhausted_block = generator.make_block();
  1193. generator.emit_jump_if(
  1194. is_iterator_exhausted,
  1195. Bytecode::Label { iterator_is_exhausted_block },
  1196. Bytecode::Label { iterator_is_not_exhausted_block });
  1197. generator.switch_to_basic_block(iterator_is_not_exhausted_block);
  1198. }
  1199. generator.emit<Bytecode::Op::IteratorNext>(temp_iterator_result, iterator);
  1200. generator.emit_iterator_complete(is_iterator_exhausted, temp_iterator_result);
  1201. // We still have to check for exhaustion here. If the iterator is exhausted,
  1202. // we need to bail before trying to get the value
  1203. auto& no_bail_block = generator.make_block();
  1204. generator.emit_jump_if(
  1205. is_iterator_exhausted,
  1206. Bytecode::Label { iterator_is_exhausted_block },
  1207. Bytecode::Label { no_bail_block });
  1208. generator.switch_to_basic_block(no_bail_block);
  1209. // Get the next value in the iterator
  1210. auto value = generator.allocate_register();
  1211. generator.emit_iterator_value(value, temp_iterator_result);
  1212. auto& create_binding_block = generator.make_block();
  1213. generator.emit<Bytecode::Op::Jump>(Bytecode::Label { create_binding_block });
  1214. // The iterator is exhausted, so we just load undefined and continue binding
  1215. generator.switch_to_basic_block(iterator_is_exhausted_block);
  1216. generator.emit<Bytecode::Op::Mov>(value, generator.add_constant(js_undefined()));
  1217. generator.emit<Bytecode::Op::Jump>(Bytecode::Label { create_binding_block });
  1218. generator.switch_to_basic_block(create_binding_block);
  1219. if (initializer) {
  1220. auto& value_is_undefined_block = generator.make_block();
  1221. auto& value_is_not_undefined_block = generator.make_block();
  1222. generator.emit<Bytecode::Op::JumpUndefined>(
  1223. value,
  1224. Bytecode::Label { value_is_undefined_block },
  1225. Bytecode::Label { value_is_not_undefined_block });
  1226. generator.switch_to_basic_block(value_is_undefined_block);
  1227. Optional<ScopedOperand> default_value;
  1228. if (auto const* alias_identifier = alias.get_pointer<NonnullRefPtr<Identifier const>>()) {
  1229. default_value = TRY(generator.emit_named_evaluation_if_anonymous_function(*initializer, generator.intern_identifier((*alias_identifier)->string()))).value();
  1230. } else if (auto const* name_identifier = name.get_pointer<NonnullRefPtr<Identifier const>>()) {
  1231. default_value = TRY(generator.emit_named_evaluation_if_anonymous_function(*initializer, generator.intern_identifier((*name_identifier)->string()))).value();
  1232. } else {
  1233. default_value = TRY(initializer->generate_bytecode(generator)).value();
  1234. }
  1235. generator.emit<Bytecode::Op::Mov>(value, *default_value);
  1236. generator.emit<Bytecode::Op::Jump>(Bytecode::Label { value_is_not_undefined_block });
  1237. generator.switch_to_basic_block(value_is_not_undefined_block);
  1238. }
  1239. TRY(assign_value_to_alias(alias, value));
  1240. first = false;
  1241. }
  1242. auto& done_block = generator.make_block();
  1243. auto& not_done_block = generator.make_block();
  1244. generator.emit_jump_if(
  1245. is_iterator_exhausted,
  1246. Bytecode::Label { done_block },
  1247. Bytecode::Label { not_done_block });
  1248. generator.switch_to_basic_block(not_done_block);
  1249. generator.emit<Bytecode::Op::IteratorClose>(iterator, Completion::Type::Normal, Optional<Value> {});
  1250. generator.emit<Bytecode::Op::Jump>(Bytecode::Label { done_block });
  1251. generator.switch_to_basic_block(done_block);
  1252. return {};
  1253. }
  1254. Bytecode::CodeGenerationErrorOr<void> BindingPattern::generate_bytecode(Bytecode::Generator& generator, Bytecode::Op::SetVariable::InitializationMode initialization_mode, ScopedOperand const& input_value, bool create_variables) const
  1255. {
  1256. if (kind == Kind::Object)
  1257. return generate_object_binding_pattern_bytecode(generator, *this, initialization_mode, input_value, create_variables);
  1258. return generate_array_binding_pattern_bytecode(generator, *this, initialization_mode, input_value, create_variables);
  1259. }
  1260. static Bytecode::CodeGenerationErrorOr<void> assign_value_to_variable_declarator(Bytecode::Generator& generator, VariableDeclarator const& declarator, VariableDeclaration const& declaration, ScopedOperand value)
  1261. {
  1262. auto initialization_mode = declaration.is_lexical_declaration() ? Bytecode::Op::SetVariable::InitializationMode::Initialize : Bytecode::Op::SetVariable::InitializationMode::Set;
  1263. return declarator.target().visit(
  1264. [&](NonnullRefPtr<Identifier const> const& id) -> Bytecode::CodeGenerationErrorOr<void> {
  1265. generator.emit_set_variable(*id, value, initialization_mode);
  1266. return {};
  1267. },
  1268. [&](NonnullRefPtr<BindingPattern const> const& pattern) -> Bytecode::CodeGenerationErrorOr<void> {
  1269. return pattern->generate_bytecode(generator, initialization_mode, value, false);
  1270. });
  1271. }
  1272. Bytecode::CodeGenerationErrorOr<Optional<ScopedOperand>> VariableDeclaration::generate_bytecode(Bytecode::Generator& generator, [[maybe_unused]] Optional<ScopedOperand> preferred_dst) const
  1273. {
  1274. Bytecode::Generator::SourceLocationScope scope(generator, *this);
  1275. for (auto& declarator : m_declarations) {
  1276. // NOTE: `var` declarations can have duplicates, but duplicate `let` or `const` bindings are a syntax error.
  1277. // Because of this, we can sink `let` and `const` directly into the preferred_dst if available.
  1278. // This is not safe for `var` since the preferred_dst may be used in the initializer.
  1279. Optional<ScopedOperand> init_dst;
  1280. if (declaration_kind() != DeclarationKind::Var) {
  1281. if (auto const* identifier = declarator->target().get_pointer<NonnullRefPtr<Identifier const>>()) {
  1282. if ((*identifier)->is_local()) {
  1283. init_dst = generator.local((*identifier)->local_variable_index());
  1284. }
  1285. }
  1286. }
  1287. if (declarator->init()) {
  1288. auto value = TRY([&]() -> Bytecode::CodeGenerationErrorOr<ScopedOperand> {
  1289. if (auto const* lhs = declarator->target().get_pointer<NonnullRefPtr<Identifier const>>()) {
  1290. return TRY(generator.emit_named_evaluation_if_anonymous_function(*declarator->init(), generator.intern_identifier((*lhs)->string()), init_dst)).value();
  1291. } else {
  1292. return TRY(declarator->init()->generate_bytecode(generator, init_dst)).value();
  1293. }
  1294. }());
  1295. (void)TRY(assign_value_to_variable_declarator(generator, declarator, *this, value));
  1296. } else if (m_declaration_kind != DeclarationKind::Var) {
  1297. (void)TRY(assign_value_to_variable_declarator(generator, declarator, *this, generator.add_constant(js_undefined())));
  1298. }
  1299. }
  1300. for (auto& declarator : m_declarations) {
  1301. if (auto const* identifier = declarator->target().get_pointer<NonnullRefPtr<Identifier const>>()) {
  1302. if ((*identifier)->is_local()) {
  1303. generator.set_local_initialized((*identifier)->local_variable_index());
  1304. }
  1305. }
  1306. }
  1307. // NOTE: VariableDeclaration doesn't return a completion value.
  1308. return Optional<ScopedOperand> {};
  1309. }
  1310. struct BaseAndValue {
  1311. ScopedOperand base;
  1312. ScopedOperand value;
  1313. };
  1314. static Bytecode::CodeGenerationErrorOr<BaseAndValue> get_base_and_value_from_member_expression(Bytecode::Generator& generator, MemberExpression const& member_expression)
  1315. {
  1316. // https://tc39.es/ecma262/#sec-super-keyword-runtime-semantics-evaluation
  1317. if (is<SuperExpression>(member_expression.object())) {
  1318. // 1. Let env be GetThisEnvironment().
  1319. // 2. Let actualThis be ? env.GetThisBinding().
  1320. auto this_value = generator.get_this();
  1321. Optional<ScopedOperand> computed_property;
  1322. if (member_expression.is_computed()) {
  1323. // SuperProperty : super [ Expression ]
  1324. // 3. Let propertyNameReference be ? Evaluation of Expression.
  1325. // 4. Let propertyNameValue be ? GetValue(propertyNameReference).
  1326. computed_property = TRY(member_expression.property().generate_bytecode(generator));
  1327. }
  1328. // 5/7. Return ? MakeSuperPropertyReference(actualThis, propertyKey, strict).
  1329. // https://tc39.es/ecma262/#sec-makesuperpropertyreference
  1330. // 1. Let env be GetThisEnvironment().
  1331. // 2. Assert: env.HasSuperBinding() is true.
  1332. // 3. Let baseValue be ? env.GetSuperBase().
  1333. auto super_base = generator.allocate_register();
  1334. generator.emit<Bytecode::Op::ResolveSuperBase>(super_base);
  1335. auto value = generator.allocate_register();
  1336. // 4. Return the Reference Record { [[Base]]: baseValue, [[ReferencedName]]: propertyKey, [[Strict]]: strict, [[ThisValue]]: actualThis }.
  1337. if (computed_property.has_value()) {
  1338. // 5. Let propertyKey be ? ToPropertyKey(propertyNameValue).
  1339. // FIXME: This does ToPropertyKey out of order, which is observable by Symbol.toPrimitive!
  1340. generator.emit<Bytecode::Op::GetByValueWithThis>(value, super_base, *computed_property, this_value);
  1341. } else {
  1342. // 3. Let propertyKey be StringValue of IdentifierName.
  1343. auto identifier_table_ref = generator.intern_identifier(verify_cast<Identifier>(member_expression.property()).string());
  1344. generator.emit_get_by_id_with_this(value, super_base, identifier_table_ref, this_value);
  1345. }
  1346. return BaseAndValue { this_value, value };
  1347. }
  1348. auto base = TRY(member_expression.object().generate_bytecode(generator)).value();
  1349. auto value = generator.allocate_register();
  1350. if (member_expression.is_computed()) {
  1351. auto property = TRY(member_expression.property().generate_bytecode(generator)).value();
  1352. generator.emit<Bytecode::Op::GetByValue>(value, base, property);
  1353. } else if (is<PrivateIdentifier>(member_expression.property())) {
  1354. generator.emit<Bytecode::Op::GetPrivateById>(
  1355. value,
  1356. base,
  1357. generator.intern_identifier(verify_cast<PrivateIdentifier>(member_expression.property()).string()));
  1358. } else {
  1359. auto base_identifier = generator.intern_identifier_for_expression(member_expression.object());
  1360. generator.emit_get_by_id(value, base, generator.intern_identifier(verify_cast<Identifier>(member_expression.property()).string()), move(base_identifier));
  1361. }
  1362. return BaseAndValue { base, value };
  1363. }
  1364. static Bytecode::CodeGenerationErrorOr<void> generate_optional_chain(Bytecode::Generator& generator, OptionalChain const& optional_chain, ScopedOperand current_value, ScopedOperand current_base, [[maybe_unused]] Optional<ScopedOperand> preferred_dst = {});
  1365. Bytecode::CodeGenerationErrorOr<Optional<ScopedOperand>> CallExpression::generate_bytecode(Bytecode::Generator& generator, Optional<ScopedOperand> preferred_dst) const
  1366. {
  1367. Bytecode::Generator::SourceLocationScope scope(generator, *this);
  1368. Optional<Bytecode::Builtin> builtin;
  1369. Optional<ScopedOperand> original_callee;
  1370. auto this_value = generator.add_constant(js_undefined());
  1371. if (is<NewExpression>(this)) {
  1372. original_callee = TRY(m_callee->generate_bytecode(generator)).value();
  1373. } else if (is<MemberExpression>(*m_callee)) {
  1374. auto& member_expression = static_cast<MemberExpression const&>(*m_callee);
  1375. auto base_and_value = TRY(get_base_and_value_from_member_expression(generator, member_expression));
  1376. original_callee = base_and_value.value;
  1377. this_value = base_and_value.base;
  1378. builtin = Bytecode::get_builtin(member_expression);
  1379. } else if (is<OptionalChain>(*m_callee)) {
  1380. auto& optional_chain = static_cast<OptionalChain const&>(*m_callee);
  1381. original_callee = generator.allocate_register();
  1382. this_value = generator.allocate_register();
  1383. TRY(generate_optional_chain(generator, optional_chain, *original_callee, this_value));
  1384. } else if (is<Identifier>(*m_callee)) {
  1385. // If the original_callee is an identifier, we may need to extract a `this` value.
  1386. // This is important when we're inside a `with` statement and calling a method on
  1387. // the environment's binding object.
  1388. // NOTE: If the identifier refers to a known "local" or "global", we know it can't be
  1389. // a `with` binding, so we can skip this.
  1390. auto& identifier = static_cast<Identifier const&>(*m_callee);
  1391. if (identifier.is_local()) {
  1392. auto local = generator.local(identifier.local_variable_index());
  1393. if (!generator.is_local_initialized(local.operand().index())) {
  1394. generator.emit<Bytecode::Op::ThrowIfTDZ>(local);
  1395. }
  1396. original_callee = local;
  1397. } else if (identifier.is_global()) {
  1398. original_callee = m_callee->generate_bytecode(generator).value();
  1399. } else {
  1400. original_callee = generator.allocate_register();
  1401. this_value = generator.allocate_register();
  1402. generator.emit<Bytecode::Op::GetCalleeAndThisFromEnvironment>(
  1403. *original_callee,
  1404. this_value,
  1405. generator.intern_identifier(identifier.string()),
  1406. generator.next_environment_variable_cache());
  1407. }
  1408. } else {
  1409. // FIXME: this = global object in sloppy mode.
  1410. original_callee = TRY(m_callee->generate_bytecode(generator)).value();
  1411. }
  1412. // NOTE: If the callee isn't already a temporary, we copy it to a new register
  1413. // to avoid overwriting it while evaluating arguments.
  1414. auto callee = [&]() -> ScopedOperand {
  1415. if (!original_callee->operand().is_register()) {
  1416. auto callee = generator.allocate_register();
  1417. generator.emit<Bytecode::Op::Mov>(callee, *original_callee);
  1418. return callee;
  1419. }
  1420. return *original_callee;
  1421. }();
  1422. Bytecode::Op::CallType call_type;
  1423. if (is<NewExpression>(*this)) {
  1424. call_type = Bytecode::Op::CallType::Construct;
  1425. } else if (m_callee->is_identifier() && static_cast<Identifier const&>(*m_callee).string() == "eval"sv) {
  1426. call_type = Bytecode::Op::CallType::DirectEval;
  1427. } else {
  1428. call_type = Bytecode::Op::CallType::Call;
  1429. }
  1430. Optional<Bytecode::StringTableIndex> expression_string_index;
  1431. if (auto expression_string = this->expression_string(); expression_string.has_value())
  1432. expression_string_index = generator.intern_string(expression_string.release_value());
  1433. bool has_spread = any_of(arguments(), [](auto& argument) { return argument.is_spread; });
  1434. auto dst = choose_dst(generator, preferred_dst);
  1435. if (has_spread) {
  1436. auto arguments = TRY(arguments_to_array_for_call(generator, this->arguments())).value();
  1437. generator.emit<Bytecode::Op::CallWithArgumentArray>(call_type, dst, callee, this_value, arguments, expression_string_index);
  1438. } else {
  1439. Vector<ScopedOperand> argument_operands;
  1440. argument_operands.ensure_capacity(arguments().size());
  1441. for (auto const& argument : arguments()) {
  1442. auto argument_value = TRY(argument.value->generate_bytecode(generator)).value();
  1443. // NOTE: We don't need to protect temporary registers or constants from being clobbered.
  1444. if (argument_value.operand().is_register() || argument_value.operand().is_constant()) {
  1445. argument_operands.append(argument_value);
  1446. } else {
  1447. auto temporary = generator.allocate_register();
  1448. generator.emit<Bytecode::Op::Mov>(temporary, argument_value);
  1449. argument_operands.append(temporary);
  1450. }
  1451. }
  1452. generator.emit_with_extra_operand_slots<Bytecode::Op::Call>(
  1453. argument_operands.size(),
  1454. call_type,
  1455. dst,
  1456. callee,
  1457. this_value,
  1458. argument_operands,
  1459. expression_string_index,
  1460. builtin);
  1461. }
  1462. return dst;
  1463. }
  1464. static ScopedOperand generate_await(
  1465. Bytecode::Generator& generator,
  1466. ScopedOperand argument,
  1467. ScopedOperand received_completion,
  1468. ScopedOperand received_completion_type,
  1469. ScopedOperand received_completion_value,
  1470. Bytecode::IdentifierTableIndex type_identifier,
  1471. Bytecode::IdentifierTableIndex value_identifier);
  1472. // https://tc39.es/ecma262/#sec-return-statement-runtime-semantics-evaluation
  1473. Bytecode::CodeGenerationErrorOr<Optional<ScopedOperand>> ReturnStatement::generate_bytecode(Bytecode::Generator& generator, Optional<ScopedOperand>) const
  1474. {
  1475. Bytecode::Generator::SourceLocationScope scope(generator, *this);
  1476. Optional<ScopedOperand> return_value;
  1477. if (m_argument) {
  1478. // ReturnStatement : return Expression ;
  1479. // 1. Let exprRef be ? Evaluation of Expression.
  1480. // 2. Let exprValue be ? GetValue(exprRef).
  1481. return_value = TRY(m_argument->generate_bytecode(generator)).value();
  1482. // 3. If GetGeneratorKind() is async, set exprValue to ? Await(exprValue).
  1483. // Spec Issue?: The spec doesn't seem to do implicit await on explicit return for async functions, but does for
  1484. // async generators. However, the major engines do so, and this is observable via constructor lookups
  1485. // on Promise objects and custom thenables.
  1486. // See: https://tc39.es/ecma262/#sec-asyncblockstart
  1487. // c. Assert: If we return here, the async function either threw an exception or performed an implicit or explicit return; all awaiting is done.
  1488. if (generator.is_in_async_function()) {
  1489. auto received_completion = generator.allocate_register();
  1490. auto received_completion_type = generator.allocate_register();
  1491. auto received_completion_value = generator.allocate_register();
  1492. auto type_identifier = generator.intern_identifier("type");
  1493. auto value_identifier = generator.intern_identifier("value");
  1494. return_value = generate_await(generator, *return_value, received_completion, received_completion_type, received_completion_value, type_identifier, value_identifier);
  1495. }
  1496. // 4. Return Completion Record { [[Type]]: return, [[Value]]: exprValue, [[Target]]: empty }.
  1497. } else {
  1498. // ReturnStatement : return ;
  1499. // 1. Return Completion Record { [[Type]]: return, [[Value]]: undefined, [[Target]]: empty }.
  1500. return_value = generator.add_constant(js_undefined());
  1501. }
  1502. if (generator.is_in_generator_or_async_function()) {
  1503. generator.perform_needed_unwinds<Bytecode::Op::Yield>();
  1504. generator.emit<Bytecode::Op::Yield>(nullptr, *return_value);
  1505. } else {
  1506. generator.perform_needed_unwinds<Bytecode::Op::Return>();
  1507. generator.emit<Bytecode::Op::Return>(return_value.has_value() ? return_value->operand() : Optional<Operand> {});
  1508. }
  1509. return return_value;
  1510. }
  1511. static void get_received_completion_type_and_value(
  1512. Bytecode::Generator& generator,
  1513. ScopedOperand received_completion,
  1514. ScopedOperand received_completion_type,
  1515. ScopedOperand received_completion_value,
  1516. Bytecode::IdentifierTableIndex type_identifier,
  1517. Bytecode::IdentifierTableIndex value_identifier)
  1518. {
  1519. generator.emit_get_by_id(received_completion_type, received_completion, type_identifier);
  1520. generator.emit_get_by_id(received_completion_value, received_completion, value_identifier);
  1521. }
  1522. enum class AwaitBeforeYield {
  1523. No,
  1524. Yes,
  1525. };
  1526. static void generate_yield(Bytecode::Generator& generator,
  1527. Bytecode::Label continuation_label,
  1528. ScopedOperand argument,
  1529. ScopedOperand received_completion,
  1530. ScopedOperand received_completion_type,
  1531. ScopedOperand received_completion_value,
  1532. Bytecode::IdentifierTableIndex type_identifier,
  1533. Bytecode::IdentifierTableIndex value_identifier,
  1534. AwaitBeforeYield await_before_yield)
  1535. {
  1536. if (!generator.is_in_async_generator_function()) {
  1537. generator.emit<Bytecode::Op::Yield>(Bytecode::Label { continuation_label }, argument);
  1538. return;
  1539. }
  1540. if (await_before_yield == AwaitBeforeYield::Yes)
  1541. argument = generate_await(generator, argument, received_completion, received_completion_type, received_completion_value, type_identifier, value_identifier);
  1542. auto& unwrap_yield_resumption_block = generator.make_block();
  1543. generator.emit<Bytecode::Op::Yield>(Bytecode::Label { unwrap_yield_resumption_block }, argument);
  1544. generator.switch_to_basic_block(unwrap_yield_resumption_block);
  1545. generator.emit<Bytecode::Op::Mov>(received_completion, generator.accumulator());
  1546. get_received_completion_type_and_value(generator, received_completion, received_completion_type, received_completion_value, type_identifier, value_identifier);
  1547. // 27.6.3.7 AsyncGeneratorUnwrapYieldResumption ( resumptionValue ), https://tc39.es/ecma262/#sec-asyncgeneratorunwrapyieldresumption
  1548. // 1. If resumptionValue.[[Type]] is not return, return ? resumptionValue.
  1549. auto& resumption_value_type_is_return_block = generator.make_block();
  1550. auto resumption_value_type_is_not_return_result = generator.allocate_register();
  1551. generator.emit<Bytecode::Op::StrictlyInequals>(
  1552. resumption_value_type_is_not_return_result,
  1553. received_completion_type,
  1554. generator.add_constant(Value(to_underlying(Completion::Type::Return))));
  1555. generator.emit_jump_if(
  1556. resumption_value_type_is_not_return_result,
  1557. Bytecode::Label { continuation_label },
  1558. Bytecode::Label { resumption_value_type_is_return_block });
  1559. generator.switch_to_basic_block(resumption_value_type_is_return_block);
  1560. // 2. Let awaited be Completion(Await(resumptionValue.[[Value]])).
  1561. generate_await(generator, received_completion_value, received_completion, received_completion_type, received_completion_value, type_identifier, value_identifier);
  1562. // 3. If awaited.[[Type]] is throw, return ? awaited.
  1563. auto& awaited_type_is_normal_block = generator.make_block();
  1564. auto awaited_type_is_throw_result = generator.allocate_register();
  1565. generator.emit<Bytecode::Op::StrictlyEquals>(
  1566. awaited_type_is_throw_result,
  1567. received_completion_type,
  1568. generator.add_constant(Value(to_underlying(Completion::Type::Throw))));
  1569. generator.emit_jump_if(
  1570. awaited_type_is_throw_result,
  1571. Bytecode::Label { continuation_label },
  1572. Bytecode::Label { awaited_type_is_normal_block });
  1573. // 4. Assert: awaited.[[Type]] is normal.
  1574. generator.switch_to_basic_block(awaited_type_is_normal_block);
  1575. // 5. Return Completion Record { [[Type]]: return, [[Value]]: awaited.[[Value]], [[Target]]: empty }.
  1576. generator.emit<Bytecode::Op::PutById>(
  1577. received_completion,
  1578. type_identifier,
  1579. generator.add_constant(Value(to_underlying(Completion::Type::Return))),
  1580. Bytecode::Op::PropertyKind::KeyValue,
  1581. generator.next_property_lookup_cache());
  1582. generator.emit<Bytecode::Op::Jump>(continuation_label);
  1583. }
  1584. Bytecode::CodeGenerationErrorOr<Optional<ScopedOperand>> YieldExpression::generate_bytecode(Bytecode::Generator& generator, [[maybe_unused]] Optional<ScopedOperand> preferred_dst) const
  1585. {
  1586. // Note: We need to catch any scheduled exceptions and reschedule them on re-entry
  1587. // as the act of yielding would otherwise clear them out
  1588. // This only applies when we are in a finalizer
  1589. bool is_in_finalizer = generator.is_in_finalizer();
  1590. Optional<ScopedOperand> saved_exception;
  1591. Bytecode::Generator::SourceLocationScope scope(generator, *this);
  1592. VERIFY(generator.is_in_generator_function());
  1593. auto received_completion = generator.allocate_register();
  1594. auto received_completion_type = generator.allocate_register();
  1595. auto received_completion_value = generator.allocate_register();
  1596. auto type_identifier = generator.intern_identifier("type");
  1597. auto value_identifier = generator.intern_identifier("value");
  1598. if (m_is_yield_from) {
  1599. // 15.5.5 Runtime Semantics: Evaluation, https://tc39.es/ecma262/#sec-generator-function-definitions-runtime-semantics-evaluation
  1600. // 1. Let generatorKind be GetGeneratorKind().
  1601. // NOTE: is_in_async_generator_function differentiates the generator kind.
  1602. // 2. Let exprRef be ? Evaluation of AssignmentExpression.
  1603. // 3. Let value be ? GetValue(exprRef).
  1604. VERIFY(m_argument);
  1605. auto value = TRY(m_argument->generate_bytecode(generator)).value();
  1606. // 4. Let iteratorRecord be ? GetIterator(value, generatorKind).
  1607. auto iterator_record = generator.allocate_register();
  1608. auto iterator_hint = generator.is_in_async_generator_function() ? IteratorHint::Async : IteratorHint::Sync;
  1609. generator.emit<Bytecode::Op::GetIterator>(iterator_record, value, iterator_hint);
  1610. // 5. Let iterator be iteratorRecord.[[Iterator]].
  1611. auto iterator = generator.allocate_register();
  1612. generator.emit<Bytecode::Op::GetObjectFromIteratorRecord>(iterator, iterator_record);
  1613. // Cache iteratorRecord.[[NextMethod]] for use in step 7.a.i.
  1614. auto next_method = generator.allocate_register();
  1615. generator.emit<Bytecode::Op::GetNextMethodFromIteratorRecord>(next_method, iterator_record);
  1616. // 6. Let received be NormalCompletion(undefined).
  1617. // See get_received_completion_type_and_value above.
  1618. generator.emit<Bytecode::Op::Mov>(received_completion_type, generator.add_constant(Value(to_underlying(Completion::Type::Normal))));
  1619. generator.emit<Bytecode::Op::Mov>(received_completion_value, generator.add_constant(js_undefined()));
  1620. // 7. Repeat,
  1621. auto& loop_block = generator.make_block();
  1622. auto& continuation_block = generator.make_block();
  1623. auto& loop_end_block = generator.make_block();
  1624. generator.emit<Bytecode::Op::Jump>(Bytecode::Label { loop_block });
  1625. generator.switch_to_basic_block(loop_block);
  1626. // a. If received.[[Type]] is normal, then
  1627. auto& type_is_normal_block = generator.make_block();
  1628. auto& is_type_throw_block = generator.make_block();
  1629. auto received_completion_type_register_is_normal = generator.allocate_register();
  1630. generator.emit<Bytecode::Op::StrictlyEquals>(
  1631. received_completion_type_register_is_normal,
  1632. received_completion_type,
  1633. generator.add_constant(Value(to_underlying(Completion::Type::Normal))));
  1634. generator.emit_jump_if(
  1635. received_completion_type_register_is_normal,
  1636. Bytecode::Label { type_is_normal_block },
  1637. Bytecode::Label { is_type_throw_block });
  1638. generator.switch_to_basic_block(type_is_normal_block);
  1639. // i. Let innerResult be ? Call(iteratorRecord.[[NextMethod]], iteratorRecord.[[Iterator]], « received.[[Value]] »).
  1640. auto array = generator.allocate_register();
  1641. generator.emit_with_extra_operand_slots<Bytecode::Op::NewArray>(1, array, ReadonlySpan<ScopedOperand> { &received_completion_value, 1 });
  1642. auto inner_result = generator.allocate_register();
  1643. generator.emit<Bytecode::Op::CallWithArgumentArray>(Bytecode::Op::CallType::Call, inner_result, next_method, iterator, array);
  1644. // ii. If generatorKind is async, set innerResult to ? Await(innerResult).
  1645. if (generator.is_in_async_generator_function()) {
  1646. auto new_inner_result = generate_await(generator, inner_result, received_completion, received_completion_type, received_completion_value, type_identifier, value_identifier);
  1647. generator.emit<Bytecode::Op::Mov>(inner_result, new_inner_result);
  1648. }
  1649. // iii. If innerResult is not an Object, throw a TypeError exception.
  1650. generator.emit<Bytecode::Op::ThrowIfNotObject>(inner_result);
  1651. // iv. Let done be ? IteratorComplete(innerResult).
  1652. auto done = generator.allocate_register();
  1653. generator.emit_iterator_complete(done, inner_result);
  1654. // v. If done is true, then
  1655. auto& type_is_normal_done_block = generator.make_block();
  1656. auto& type_is_normal_not_done_block = generator.make_block();
  1657. generator.emit_jump_if(
  1658. done,
  1659. Bytecode::Label { type_is_normal_done_block },
  1660. Bytecode::Label { type_is_normal_not_done_block });
  1661. generator.switch_to_basic_block(type_is_normal_done_block);
  1662. // 1. Return ? IteratorValue(innerResult).
  1663. auto return_value = generator.allocate_register();
  1664. generator.emit_iterator_value(return_value, inner_result);
  1665. generator.emit<Bytecode::Op::Jump>(Bytecode::Label { loop_end_block });
  1666. generator.switch_to_basic_block(type_is_normal_not_done_block);
  1667. // vi. If generatorKind is async, set received to Completion(AsyncGeneratorYield(? IteratorValue(innerResult))).
  1668. // vii. Else, set received to Completion(GeneratorYield(innerResult)).
  1669. {
  1670. // 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.
  1671. // This only matters for non-async generators.
  1672. auto current_value = generator.allocate_register();
  1673. generator.emit_iterator_value(current_value, inner_result);
  1674. if (is_in_finalizer) {
  1675. saved_exception = generator.allocate_register();
  1676. generator.emit<Bytecode::Op::Mov>(Bytecode::Operand(*saved_exception), Bytecode::Operand(Bytecode::Register::exception()));
  1677. }
  1678. generate_yield(generator,
  1679. Bytecode::Label { continuation_block },
  1680. current_value,
  1681. received_completion,
  1682. received_completion_type,
  1683. received_completion_value,
  1684. type_identifier,
  1685. value_identifier,
  1686. AwaitBeforeYield::No);
  1687. }
  1688. // b. Else if received.[[Type]] is throw, then
  1689. generator.switch_to_basic_block(is_type_throw_block);
  1690. auto& type_is_throw_block = generator.make_block();
  1691. auto& type_is_return_block = generator.make_block();
  1692. auto received_completion_type_register_is_throw = generator.allocate_register();
  1693. generator.emit<Bytecode::Op::StrictlyEquals>(
  1694. received_completion_type_register_is_throw,
  1695. received_completion_type,
  1696. generator.add_constant(Value(to_underlying(Completion::Type::Throw))));
  1697. generator.emit_jump_if(
  1698. received_completion_type_register_is_throw,
  1699. Bytecode::Label { type_is_throw_block },
  1700. Bytecode::Label { type_is_return_block });
  1701. generator.switch_to_basic_block(type_is_throw_block);
  1702. // i. Let throw be ? GetMethod(iterator, "throw").
  1703. auto throw_method = generator.allocate_register();
  1704. generator.emit<Bytecode::Op::GetMethod>(throw_method, iterator, generator.intern_identifier("throw"));
  1705. // ii. If throw is not undefined, then
  1706. auto& throw_method_is_defined_block = generator.make_block();
  1707. auto& throw_method_is_undefined_block = generator.make_block();
  1708. generator.emit<Bytecode::Op::JumpUndefined>(
  1709. throw_method,
  1710. Bytecode::Label { throw_method_is_undefined_block },
  1711. Bytecode::Label { throw_method_is_defined_block });
  1712. generator.switch_to_basic_block(throw_method_is_defined_block);
  1713. // 1. Let innerResult be ? Call(throw, iterator, « received.[[Value]] »).
  1714. auto received_value_array = generator.allocate_register();
  1715. generator.emit_with_extra_operand_slots<Bytecode::Op::NewArray>(1, received_value_array, ReadonlySpan<ScopedOperand> { &received_completion_value, 1 });
  1716. generator.emit<Bytecode::Op::CallWithArgumentArray>(Bytecode::Op::CallType::Call, inner_result, throw_method, iterator, received_value_array);
  1717. // 2. If generatorKind is async, set innerResult to ? Await(innerResult).
  1718. if (generator.is_in_async_generator_function()) {
  1719. auto new_result = generate_await(generator, inner_result, received_completion, received_completion_type, received_completion_value, type_identifier, value_identifier);
  1720. generator.emit<Bytecode::Op::Mov>(inner_result, new_result);
  1721. }
  1722. // 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.
  1723. // 4. If innerResult is not an Object, throw a TypeError exception.
  1724. generator.emit<Bytecode::Op::ThrowIfNotObject>(inner_result);
  1725. // 5. Let done be ? IteratorComplete(innerResult).
  1726. generator.emit_iterator_complete(done, inner_result);
  1727. // 6. If done is true, then
  1728. auto& type_is_throw_done_block = generator.make_block();
  1729. auto& type_is_throw_not_done_block = generator.make_block();
  1730. generator.emit_jump_if(
  1731. done,
  1732. Bytecode::Label { type_is_throw_done_block },
  1733. Bytecode::Label { type_is_throw_not_done_block });
  1734. generator.switch_to_basic_block(type_is_throw_done_block);
  1735. // a. Return ? IteratorValue(innerResult).
  1736. generator.emit_iterator_value(return_value, inner_result);
  1737. generator.emit<Bytecode::Op::Jump>(Bytecode::Label { loop_end_block });
  1738. generator.switch_to_basic_block(type_is_throw_not_done_block);
  1739. {
  1740. // 7. If generatorKind is async, set received to Completion(AsyncGeneratorYield(? IteratorValue(innerResult))).
  1741. // 8. Else, set received to Completion(GeneratorYield(innerResult)).
  1742. // 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.
  1743. // This only matters for non-async generators.
  1744. auto yield_value = generator.allocate_register();
  1745. generator.emit_iterator_value(yield_value, inner_result);
  1746. generate_yield(generator, Bytecode::Label { continuation_block }, yield_value, received_completion, received_completion_type, received_completion_value, type_identifier, value_identifier, AwaitBeforeYield::No);
  1747. }
  1748. generator.switch_to_basic_block(throw_method_is_undefined_block);
  1749. // 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.
  1750. // 2. Let closeCompletion be Completion Record { [[Type]]: normal, [[Value]]: empty, [[Target]]: empty }.
  1751. // 3. If generatorKind is async, perform ? AsyncIteratorClose(iteratorRecord, closeCompletion).
  1752. if (generator.is_in_async_generator_function()) {
  1753. // FIXME: This performs `await` outside of the generator!
  1754. generator.emit<Bytecode::Op::AsyncIteratorClose>(iterator_record, Completion::Type::Normal, Optional<Value> {});
  1755. }
  1756. // 4. Else, perform ? IteratorClose(iteratorRecord, closeCompletion).
  1757. else {
  1758. generator.emit<Bytecode::Op::IteratorClose>(iterator_record, Completion::Type::Normal, Optional<Value> {});
  1759. }
  1760. // 5. NOTE: The next step throws a TypeError to indicate that there was a yield* protocol violation: iterator does not have a throw method.
  1761. // 6. Throw a TypeError exception.
  1762. auto exception = generator.allocate_register();
  1763. generator.emit<Bytecode::Op::NewTypeError>(exception, generator.intern_string(ErrorType::YieldFromIteratorMissingThrowMethod.message()));
  1764. generator.perform_needed_unwinds<Bytecode::Op::Throw>();
  1765. generator.emit<Bytecode::Op::Throw>(exception);
  1766. // c. Else,
  1767. // i. Assert: received.[[Type]] is return.
  1768. generator.switch_to_basic_block(type_is_return_block);
  1769. // ii. Let return be ? GetMethod(iterator, "return").
  1770. auto return_method = generator.allocate_register();
  1771. generator.emit<Bytecode::Op::GetMethod>(return_method, iterator, generator.intern_identifier("return"));
  1772. // iii. If return is undefined, then
  1773. auto& return_is_undefined_block = generator.make_block();
  1774. auto& return_is_defined_block = generator.make_block();
  1775. generator.emit<Bytecode::Op::JumpUndefined>(
  1776. return_method,
  1777. Bytecode::Label { return_is_undefined_block },
  1778. Bytecode::Label { return_is_defined_block });
  1779. generator.switch_to_basic_block(return_is_undefined_block);
  1780. // 1. If generatorKind is async, set received.[[Value]] to ? Await(received.[[Value]]).
  1781. if (generator.is_in_async_generator_function()) {
  1782. generate_await(generator, received_completion_value, received_completion, received_completion_type, received_completion_value, type_identifier, value_identifier);
  1783. }
  1784. // 2. Return ? received.
  1785. // NOTE: This will always be a return completion.
  1786. generator.perform_needed_unwinds<Bytecode::Op::Yield>();
  1787. generator.emit<Bytecode::Op::Yield>(nullptr, received_completion_value);
  1788. generator.switch_to_basic_block(return_is_defined_block);
  1789. // iv. Let innerReturnResult be ? Call(return, iterator, « received.[[Value]] »).
  1790. auto call_array = generator.allocate_register();
  1791. generator.emit_with_extra_operand_slots<Bytecode::Op::NewArray>(1, call_array, ReadonlySpan<ScopedOperand> { &received_completion_value, 1 });
  1792. auto inner_return_result = generator.allocate_register();
  1793. generator.emit<Bytecode::Op::CallWithArgumentArray>(Bytecode::Op::CallType::Call, inner_return_result, return_method, iterator, call_array);
  1794. // v. If generatorKind is async, set innerReturnResult to ? Await(innerReturnResult).
  1795. if (generator.is_in_async_generator_function()) {
  1796. auto new_value = generate_await(generator, inner_return_result, received_completion, received_completion_type, received_completion_value, type_identifier, value_identifier);
  1797. generator.emit<Bytecode::Op::Mov>(inner_return_result, new_value);
  1798. }
  1799. // vi. If innerReturnResult is not an Object, throw a TypeError exception.
  1800. generator.emit<Bytecode::Op::ThrowIfNotObject>(inner_return_result);
  1801. // vii. Let done be ? IteratorComplete(innerReturnResult).
  1802. generator.emit_iterator_complete(done, inner_return_result);
  1803. // viii. If done is true, then
  1804. auto& type_is_return_done_block = generator.make_block();
  1805. auto& type_is_return_not_done_block = generator.make_block();
  1806. generator.emit_jump_if(
  1807. done,
  1808. Bytecode::Label { type_is_return_done_block },
  1809. Bytecode::Label { type_is_return_not_done_block });
  1810. generator.switch_to_basic_block(type_is_return_done_block);
  1811. // 1. Let value be ? IteratorValue(innerReturnResult).
  1812. auto inner_return_result_value = generator.allocate_register();
  1813. generator.emit_iterator_value(inner_return_result_value, inner_return_result);
  1814. // 2. Return Completion Record { [[Type]]: return, [[Value]]: value, [[Target]]: empty }.
  1815. generator.perform_needed_unwinds<Bytecode::Op::Yield>();
  1816. generator.emit<Bytecode::Op::Yield>(nullptr, inner_return_result_value);
  1817. generator.switch_to_basic_block(type_is_return_not_done_block);
  1818. // ix. If generatorKind is async, set received to Completion(AsyncGeneratorYield(? IteratorValue(innerReturnResult))).
  1819. // x. Else, set received to Completion(GeneratorYield(innerReturnResult)).
  1820. // 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.
  1821. // This only matters for non-async generators.
  1822. auto received = generator.allocate_register();
  1823. generator.emit_iterator_value(received, inner_return_result);
  1824. generate_yield(generator, Bytecode::Label { continuation_block }, received, received_completion, received_completion_type, received_completion_value, type_identifier, value_identifier, AwaitBeforeYield::No);
  1825. generator.switch_to_basic_block(continuation_block);
  1826. if (is_in_finalizer)
  1827. generator.emit<Bytecode::Op::Mov>(Bytecode::Operand(Bytecode::Register::exception()), Bytecode::Operand(*saved_exception));
  1828. generator.emit<Bytecode::Op::Mov>(received_completion, generator.accumulator());
  1829. get_received_completion_type_and_value(generator, received_completion, received_completion_type, received_completion_value, type_identifier, value_identifier);
  1830. generator.emit<Bytecode::Op::Jump>(Bytecode::Label { loop_block });
  1831. generator.switch_to_basic_block(loop_end_block);
  1832. return return_value;
  1833. }
  1834. Optional<ScopedOperand> argument;
  1835. if (m_argument)
  1836. argument = TRY(m_argument->generate_bytecode(generator)).value();
  1837. else
  1838. argument = generator.add_constant(js_undefined());
  1839. auto& continuation_block = generator.make_block();
  1840. if (is_in_finalizer) {
  1841. saved_exception = generator.allocate_register();
  1842. generator.emit<Bytecode::Op::Mov>(Bytecode::Operand(*saved_exception), Bytecode::Operand(Bytecode::Register::exception()));
  1843. }
  1844. generate_yield(generator, Bytecode::Label { continuation_block }, *argument, received_completion, received_completion_type, received_completion_value, type_identifier, value_identifier, AwaitBeforeYield::Yes);
  1845. generator.switch_to_basic_block(continuation_block);
  1846. if (is_in_finalizer)
  1847. generator.emit<Bytecode::Op::Mov>(Bytecode::Operand(Bytecode::Register::exception()), Bytecode::Operand(*saved_exception));
  1848. generator.emit<Bytecode::Op::Mov>(received_completion, generator.accumulator());
  1849. get_received_completion_type_and_value(generator, received_completion, received_completion_type, received_completion_value, type_identifier, value_identifier);
  1850. auto& normal_completion_continuation_block = generator.make_block();
  1851. auto& throw_completion_continuation_block = generator.make_block();
  1852. auto received_completion_type_is_normal = generator.allocate_register();
  1853. generator.emit<Bytecode::Op::StrictlyEquals>(
  1854. received_completion_type_is_normal,
  1855. received_completion_type,
  1856. generator.add_constant(Value(to_underlying(Completion::Type::Normal))));
  1857. generator.emit_jump_if(
  1858. received_completion_type_is_normal,
  1859. Bytecode::Label { normal_completion_continuation_block },
  1860. Bytecode::Label { throw_completion_continuation_block });
  1861. auto& throw_value_block = generator.make_block();
  1862. auto& return_value_block = generator.make_block();
  1863. generator.switch_to_basic_block(throw_completion_continuation_block);
  1864. auto received_completion_type_is_throw = generator.allocate_register();
  1865. generator.emit<Bytecode::Op::StrictlyEquals>(
  1866. received_completion_type_is_throw,
  1867. received_completion_type,
  1868. generator.add_constant(Value(to_underlying(Completion::Type::Throw))));
  1869. // If type is not equal to "throw" or "normal", assume it's "return".
  1870. generator.emit_jump_if(
  1871. received_completion_type_is_throw,
  1872. Bytecode::Label { throw_value_block },
  1873. Bytecode::Label { return_value_block });
  1874. generator.switch_to_basic_block(throw_value_block);
  1875. generator.perform_needed_unwinds<Bytecode::Op::Throw>();
  1876. generator.emit<Bytecode::Op::Throw>(received_completion_value);
  1877. generator.switch_to_basic_block(return_value_block);
  1878. generator.perform_needed_unwinds<Bytecode::Op::Yield>();
  1879. generator.emit<Bytecode::Op::Yield>(nullptr, received_completion_value);
  1880. generator.switch_to_basic_block(normal_completion_continuation_block);
  1881. return received_completion_value;
  1882. }
  1883. Bytecode::CodeGenerationErrorOr<Optional<ScopedOperand>> IfStatement::generate_bytecode(Bytecode::Generator& generator, Optional<ScopedOperand> preferred_dst) const
  1884. {
  1885. Bytecode::Generator::SourceLocationScope scope(generator, *this);
  1886. // test
  1887. // jump if_true (true) true (false) false
  1888. // true
  1889. // jump always (true) end
  1890. // false
  1891. // jump always (true) end
  1892. // end
  1893. auto& true_block = generator.make_block();
  1894. auto& false_block = generator.make_block();
  1895. auto& end_block = generator.make_block();
  1896. auto dst = choose_dst(generator, preferred_dst);
  1897. generator.emit<Bytecode::Op::Mov>(dst, generator.add_constant(js_undefined()));
  1898. auto predicate = TRY(m_predicate->generate_bytecode(generator)).value();
  1899. generator.emit_jump_if(
  1900. predicate,
  1901. Bytecode::Label { true_block },
  1902. Bytecode::Label { false_block });
  1903. generator.switch_to_basic_block(true_block);
  1904. auto consequent = TRY(m_consequent->generate_bytecode(generator, dst));
  1905. if (!generator.is_current_block_terminated()) {
  1906. if (consequent.has_value())
  1907. generator.emit<Bytecode::Op::Mov>(dst, *consequent);
  1908. generator.emit<Bytecode::Op::Jump>(Bytecode::Label { end_block });
  1909. }
  1910. generator.switch_to_basic_block(false_block);
  1911. Optional<ScopedOperand> alternate;
  1912. if (m_alternate) {
  1913. alternate = TRY(m_alternate->generate_bytecode(generator, dst));
  1914. }
  1915. if (!generator.is_current_block_terminated()) {
  1916. if (alternate.has_value())
  1917. generator.emit<Bytecode::Op::Mov>(dst, *alternate);
  1918. generator.emit<Bytecode::Op::Jump>(Bytecode::Label { end_block });
  1919. }
  1920. generator.switch_to_basic_block(end_block);
  1921. return dst;
  1922. }
  1923. Bytecode::CodeGenerationErrorOr<Optional<ScopedOperand>> ContinueStatement::generate_bytecode(Bytecode::Generator& generator, [[maybe_unused]] Optional<ScopedOperand> preferred_dst) const
  1924. {
  1925. Bytecode::Generator::SourceLocationScope scope(generator, *this);
  1926. if (!m_target_label.has_value()) {
  1927. generator.generate_continue();
  1928. return Optional<ScopedOperand> {};
  1929. }
  1930. generator.generate_continue(m_target_label.value());
  1931. return Optional<ScopedOperand> {};
  1932. }
  1933. Bytecode::CodeGenerationErrorOr<Optional<ScopedOperand>> DebuggerStatement::generate_bytecode(Bytecode::Generator&, [[maybe_unused]] Optional<ScopedOperand> preferred_dst) const
  1934. {
  1935. return Optional<ScopedOperand> {};
  1936. }
  1937. Bytecode::CodeGenerationErrorOr<Optional<ScopedOperand>> ConditionalExpression::generate_bytecode(Bytecode::Generator& generator, Optional<ScopedOperand> preferred_dst) const
  1938. {
  1939. Bytecode::Generator::SourceLocationScope scope(generator, *this);
  1940. // test
  1941. // jump if_true (true) true (false) false
  1942. // true
  1943. // jump always (true) end
  1944. // false
  1945. // jump always (true) end
  1946. // end
  1947. auto& true_block = generator.make_block();
  1948. auto& false_block = generator.make_block();
  1949. auto& end_block = generator.make_block();
  1950. auto test = TRY(m_test->generate_bytecode(generator)).value();
  1951. generator.emit_jump_if(
  1952. test,
  1953. Bytecode::Label { true_block },
  1954. Bytecode::Label { false_block });
  1955. auto dst = choose_dst(generator, preferred_dst);
  1956. generator.switch_to_basic_block(true_block);
  1957. auto consequent = TRY(m_consequent->generate_bytecode(generator)).value();
  1958. generator.emit<Bytecode::Op::Mov>(dst, consequent);
  1959. generator.emit<Bytecode::Op::Jump>(Bytecode::Label { end_block });
  1960. generator.switch_to_basic_block(false_block);
  1961. auto alternate = TRY(m_alternate->generate_bytecode(generator)).value();
  1962. generator.emit<Bytecode::Op::Mov>(dst, alternate);
  1963. generator.emit<Bytecode::Op::Jump>(Bytecode::Label { end_block });
  1964. generator.switch_to_basic_block(end_block);
  1965. return dst;
  1966. }
  1967. Bytecode::CodeGenerationErrorOr<Optional<ScopedOperand>> SequenceExpression::generate_bytecode(Bytecode::Generator& generator, [[maybe_unused]] Optional<ScopedOperand> preferred_dst) const
  1968. {
  1969. Bytecode::Generator::SourceLocationScope scope(generator, *this);
  1970. Optional<ScopedOperand> last_value;
  1971. for (auto& expression : m_expressions) {
  1972. last_value = TRY(expression->generate_bytecode(generator));
  1973. }
  1974. return last_value;
  1975. }
  1976. Bytecode::CodeGenerationErrorOr<Optional<ScopedOperand>> TemplateLiteral::generate_bytecode(Bytecode::Generator& generator, Optional<ScopedOperand> preferred_dst) const
  1977. {
  1978. Bytecode::Generator::SourceLocationScope scope(generator, *this);
  1979. auto dst = choose_dst(generator, preferred_dst);
  1980. for (size_t i = 0; i < m_expressions.size(); i++) {
  1981. auto value = TRY(m_expressions[i]->generate_bytecode(generator)).value();
  1982. if (i == 0) {
  1983. generator.emit<Bytecode::Op::Mov>(dst, value);
  1984. } else {
  1985. generator.emit<Bytecode::Op::ConcatString>(dst, value);
  1986. }
  1987. }
  1988. return dst;
  1989. }
  1990. Bytecode::CodeGenerationErrorOr<Optional<ScopedOperand>> TaggedTemplateLiteral::generate_bytecode(Bytecode::Generator& generator, Optional<ScopedOperand> preferred_dst) const
  1991. {
  1992. Bytecode::Generator::SourceLocationScope scope(generator, *this);
  1993. auto tag = TRY(m_tag->generate_bytecode(generator)).value();
  1994. // FIXME: Follow
  1995. // 13.2.8.3 GetTemplateObject ( templateLiteral ), https://tc39.es/ecma262/#sec-gettemplateobject
  1996. // more closely, namely:
  1997. // * cache this somehow
  1998. // * add a raw object accessor
  1999. // * freeze array and raw member
  2000. Vector<ScopedOperand> string_regs;
  2001. auto& expressions = m_template_literal->expressions();
  2002. for (size_t i = 0; i < expressions.size(); ++i) {
  2003. if (i % 2 != 0)
  2004. continue;
  2005. // NOTE: If the string contains invalid escapes we get a null expression here,
  2006. // which we then convert to the expected `undefined` TV. See
  2007. // 12.9.6.1 Static Semantics: TV, https://tc39.es/ecma262/#sec-static-semantics-tv
  2008. auto string_reg = generator.allocate_register();
  2009. if (is<NullLiteral>(expressions[i])) {
  2010. generator.emit<Bytecode::Op::Mov>(string_reg, generator.add_constant(js_undefined()));
  2011. } else {
  2012. auto value = TRY(expressions[i]->generate_bytecode(generator)).value();
  2013. generator.emit<Bytecode::Op::Mov>(string_reg, value);
  2014. }
  2015. string_regs.append(move(string_reg));
  2016. }
  2017. auto strings_array = generator.allocate_register();
  2018. if (string_regs.is_empty()) {
  2019. generator.emit<Bytecode::Op::NewArray>(strings_array);
  2020. } else {
  2021. generator.emit_with_extra_operand_slots<Bytecode::Op::NewArray>(string_regs.size(), strings_array, string_regs);
  2022. }
  2023. Vector<ScopedOperand> argument_regs;
  2024. argument_regs.append(strings_array);
  2025. for (size_t i = 1; i < expressions.size(); i += 2) {
  2026. auto string_reg = generator.allocate_register();
  2027. auto string = TRY(expressions[i]->generate_bytecode(generator)).value();
  2028. generator.emit<Bytecode::Op::Mov>(string_reg, string);
  2029. argument_regs.append(move(string_reg));
  2030. }
  2031. Vector<ScopedOperand> raw_string_regs;
  2032. raw_string_regs.ensure_capacity(m_template_literal->raw_strings().size());
  2033. for (auto& raw_string : m_template_literal->raw_strings()) {
  2034. auto value = TRY(raw_string->generate_bytecode(generator)).value();
  2035. auto raw_string_reg = generator.allocate_register();
  2036. generator.emit<Bytecode::Op::Mov>(raw_string_reg, value);
  2037. raw_string_regs.append(move(raw_string_reg));
  2038. }
  2039. auto raw_strings_array = generator.allocate_register();
  2040. if (raw_string_regs.is_empty()) {
  2041. generator.emit<Bytecode::Op::NewArray>(raw_strings_array);
  2042. } else {
  2043. generator.emit_with_extra_operand_slots<Bytecode::Op::NewArray>(raw_string_regs.size(), raw_strings_array, raw_string_regs);
  2044. }
  2045. generator.emit<Bytecode::Op::PutById>(strings_array, generator.intern_identifier("raw"), raw_strings_array, Bytecode::Op::PropertyKind::KeyValue, generator.next_property_lookup_cache());
  2046. auto arguments = generator.allocate_register();
  2047. if (!argument_regs.is_empty())
  2048. generator.emit_with_extra_operand_slots<Bytecode::Op::NewArray>(argument_regs.size(), arguments, argument_regs);
  2049. else
  2050. generator.emit<Bytecode::Op::NewArray>(arguments);
  2051. auto dst = choose_dst(generator, preferred_dst);
  2052. generator.emit<Bytecode::Op::CallWithArgumentArray>(Bytecode::Op::CallType::Call, dst, tag, generator.add_constant(js_undefined()), arguments);
  2053. return dst;
  2054. }
  2055. Bytecode::CodeGenerationErrorOr<Optional<ScopedOperand>> UpdateExpression::generate_bytecode(Bytecode::Generator& generator, Optional<ScopedOperand>) const
  2056. {
  2057. Bytecode::Generator::SourceLocationScope scope(generator, *this);
  2058. auto reference = TRY(generator.emit_load_from_reference(*m_argument));
  2059. Optional<ScopedOperand> previous_value_for_postfix;
  2060. if (m_op == UpdateOp::Increment) {
  2061. if (m_prefixed) {
  2062. generator.emit<Bytecode::Op::Increment>(*reference.loaded_value);
  2063. } else {
  2064. previous_value_for_postfix = generator.allocate_register();
  2065. generator.emit<Bytecode::Op::PostfixIncrement>(*previous_value_for_postfix, *reference.loaded_value);
  2066. }
  2067. } else {
  2068. if (m_prefixed) {
  2069. generator.emit<Bytecode::Op::Decrement>(*reference.loaded_value);
  2070. } else {
  2071. previous_value_for_postfix = generator.allocate_register();
  2072. generator.emit<Bytecode::Op::PostfixDecrement>(*previous_value_for_postfix, *reference.loaded_value);
  2073. }
  2074. }
  2075. if (is<Identifier>(*m_argument))
  2076. (void)TRY(generator.emit_store_to_reference(static_cast<Identifier const&>(*m_argument), *reference.loaded_value));
  2077. else
  2078. (void)TRY(generator.emit_store_to_reference(reference, *reference.loaded_value));
  2079. if (!m_prefixed)
  2080. return *previous_value_for_postfix;
  2081. return *reference.loaded_value;
  2082. }
  2083. Bytecode::CodeGenerationErrorOr<Optional<ScopedOperand>> ThrowStatement::generate_bytecode(Bytecode::Generator& generator, [[maybe_unused]] Optional<ScopedOperand> preferred_dst) const
  2084. {
  2085. Bytecode::Generator::SourceLocationScope scope(generator, *this);
  2086. auto argument = TRY(m_argument->generate_bytecode(generator)).value();
  2087. generator.perform_needed_unwinds<Bytecode::Op::Throw>();
  2088. generator.emit<Bytecode::Op::Throw>(argument);
  2089. return Optional<ScopedOperand> {};
  2090. }
  2091. Bytecode::CodeGenerationErrorOr<Optional<ScopedOperand>> BreakStatement::generate_bytecode(Bytecode::Generator& generator, [[maybe_unused]] Optional<ScopedOperand> preferred_dst) const
  2092. {
  2093. Bytecode::Generator::SourceLocationScope scope(generator, *this);
  2094. // FIXME: Handle finally blocks in a graceful manner
  2095. // We need to execute the finally block, but tell it to resume
  2096. // execution at the designated block
  2097. if (!m_target_label.has_value()) {
  2098. generator.generate_break();
  2099. return Optional<ScopedOperand> {};
  2100. }
  2101. generator.generate_break(m_target_label.value());
  2102. return Optional<ScopedOperand> {};
  2103. }
  2104. Bytecode::CodeGenerationErrorOr<Optional<ScopedOperand>> TryStatement::generate_bytecode(Bytecode::Generator& generator, [[maybe_unused]] Optional<ScopedOperand> preferred_dst) const
  2105. {
  2106. Bytecode::Generator::SourceLocationScope scope(generator, *this);
  2107. auto& saved_block = generator.current_block();
  2108. Optional<Bytecode::Label> handler_target;
  2109. Optional<Bytecode::Label> finalizer_target;
  2110. Optional<Bytecode::Generator::UnwindContext> unwind_context;
  2111. Bytecode::BasicBlock* next_block { nullptr };
  2112. Optional<ScopedOperand> dst;
  2113. if (m_finalizer) {
  2114. // FIXME: See notes in Op.h->ScheduleJump
  2115. auto& finalizer_block = generator.make_block();
  2116. generator.switch_to_basic_block(finalizer_block);
  2117. generator.emit<Bytecode::Op::LeaveUnwindContext>();
  2118. generator.start_boundary(Bytecode::Generator::BlockBoundaryType::LeaveFinally);
  2119. (void)TRY(m_finalizer->generate_bytecode(generator));
  2120. generator.end_boundary(Bytecode::Generator::BlockBoundaryType::LeaveFinally);
  2121. if (!generator.is_current_block_terminated()) {
  2122. next_block = &generator.make_block();
  2123. auto next_target = Bytecode::Label { *next_block };
  2124. generator.emit<Bytecode::Op::ContinuePendingUnwind>(next_target);
  2125. }
  2126. finalizer_target = Bytecode::Label { finalizer_block };
  2127. generator.start_boundary(Bytecode::Generator::BlockBoundaryType::ReturnToFinally);
  2128. unwind_context.emplace(generator, finalizer_target);
  2129. }
  2130. if (m_handler) {
  2131. auto& handler_block = generator.make_block();
  2132. generator.switch_to_basic_block(handler_block);
  2133. auto caught_value = generator.allocate_register();
  2134. generator.emit<Bytecode::Op::Catch>(caught_value);
  2135. if (!m_finalizer) {
  2136. generator.emit<Bytecode::Op::LeaveUnwindContext>();
  2137. generator.emit<Bytecode::Op::RestoreScheduledJump>();
  2138. }
  2139. // OPTIMIZATION: We avoid creating a lexical environment if the catch clause has no parameter.
  2140. bool did_create_variable_scope_for_catch_clause = false;
  2141. TRY(m_handler->parameter().visit(
  2142. [&](DeprecatedFlyString const& parameter) -> Bytecode::CodeGenerationErrorOr<void> {
  2143. if (!parameter.is_empty()) {
  2144. generator.begin_variable_scope();
  2145. did_create_variable_scope_for_catch_clause = true;
  2146. auto parameter_identifier = generator.intern_identifier(parameter);
  2147. generator.emit<Bytecode::Op::CreateVariable>(parameter_identifier, Bytecode::Op::EnvironmentMode::Lexical, false);
  2148. generator.emit<Bytecode::Op::SetVariable>(parameter_identifier, caught_value, generator.next_environment_variable_cache(), Bytecode::Op::SetVariable::InitializationMode::Initialize);
  2149. }
  2150. return {};
  2151. },
  2152. [&](NonnullRefPtr<BindingPattern const> const& binding_pattern) -> Bytecode::CodeGenerationErrorOr<void> {
  2153. generator.begin_variable_scope();
  2154. did_create_variable_scope_for_catch_clause = true;
  2155. TRY(binding_pattern->generate_bytecode(generator, Bytecode::Op::SetVariable::InitializationMode::Initialize, caught_value, true));
  2156. return {};
  2157. }));
  2158. auto handler_result = TRY(m_handler->body().generate_bytecode(generator));
  2159. if (handler_result.has_value() && !generator.is_current_block_terminated()) {
  2160. dst = generator.allocate_register();
  2161. generator.emit<Bytecode::Op::Mov>(*dst, *handler_result);
  2162. }
  2163. handler_target = Bytecode::Label { handler_block };
  2164. if (did_create_variable_scope_for_catch_clause)
  2165. generator.end_variable_scope();
  2166. if (!generator.is_current_block_terminated()) {
  2167. if (m_finalizer) {
  2168. generator.emit<Bytecode::Op::Jump>(*finalizer_target);
  2169. } else {
  2170. VERIFY(!next_block);
  2171. VERIFY(!unwind_context.has_value());
  2172. next_block = &generator.make_block();
  2173. auto next_target = Bytecode::Label { *next_block };
  2174. generator.emit<Bytecode::Op::Jump>(next_target);
  2175. }
  2176. }
  2177. }
  2178. if (m_finalizer)
  2179. generator.end_boundary(Bytecode::Generator::BlockBoundaryType::ReturnToFinally);
  2180. if (m_handler) {
  2181. if (!m_finalizer) {
  2182. auto const* parent_unwind_context = generator.current_unwind_context();
  2183. if (parent_unwind_context)
  2184. unwind_context.emplace(generator, parent_unwind_context->finalizer());
  2185. else
  2186. unwind_context.emplace(generator, OptionalNone());
  2187. }
  2188. unwind_context->set_handler(handler_target.value());
  2189. }
  2190. auto& target_block = generator.make_block();
  2191. generator.switch_to_basic_block(saved_block);
  2192. generator.emit<Bytecode::Op::EnterUnwindContext>(Bytecode::Label { target_block });
  2193. generator.start_boundary(Bytecode::Generator::BlockBoundaryType::Unwind);
  2194. if (m_finalizer)
  2195. generator.start_boundary(Bytecode::Generator::BlockBoundaryType::ReturnToFinally);
  2196. generator.switch_to_basic_block(target_block);
  2197. auto block_result = TRY(m_block->generate_bytecode(generator));
  2198. if (!generator.is_current_block_terminated()) {
  2199. if (block_result.has_value()) {
  2200. dst = generator.allocate_register();
  2201. generator.emit<Bytecode::Op::Mov>(*dst, *block_result);
  2202. }
  2203. if (m_finalizer) {
  2204. generator.emit<Bytecode::Op::Jump>(*finalizer_target);
  2205. } else {
  2206. VERIFY(unwind_context.has_value());
  2207. unwind_context.clear();
  2208. if (!next_block)
  2209. next_block = &generator.make_block();
  2210. generator.emit<Bytecode::Op::LeaveUnwindContext>();
  2211. generator.emit<Bytecode::Op::Jump>(Bytecode::Label { *next_block });
  2212. }
  2213. }
  2214. if (m_finalizer)
  2215. generator.end_boundary(Bytecode::Generator::BlockBoundaryType::ReturnToFinally);
  2216. generator.end_boundary(Bytecode::Generator::BlockBoundaryType::Unwind);
  2217. generator.switch_to_basic_block(next_block ? *next_block : saved_block);
  2218. if (!dst.has_value())
  2219. return generator.add_constant(js_undefined());
  2220. return dst;
  2221. }
  2222. Bytecode::CodeGenerationErrorOr<Optional<ScopedOperand>> SwitchStatement::generate_bytecode(Bytecode::Generator& generator, [[maybe_unused]] Optional<ScopedOperand> preferred_dst) const
  2223. {
  2224. Bytecode::Generator::SourceLocationScope scope(generator, *this);
  2225. return generate_labelled_evaluation(generator, {});
  2226. }
  2227. Bytecode::CodeGenerationErrorOr<Optional<ScopedOperand>> SwitchStatement::generate_labelled_evaluation(Bytecode::Generator& generator, Vector<DeprecatedFlyString> const& label_set, [[maybe_unused]] Optional<ScopedOperand> preferred_dst) const
  2228. {
  2229. Bytecode::Generator::SourceLocationScope scope(generator, *this);
  2230. auto dst = generator.allocate_register();
  2231. generator.emit<Bytecode::Op::Mov>(dst, generator.add_constant(js_undefined()));
  2232. auto discriminant = TRY(m_discriminant->generate_bytecode(generator)).value();
  2233. Vector<Bytecode::BasicBlock&> case_blocks;
  2234. Bytecode::BasicBlock* entry_block_for_default { nullptr };
  2235. Bytecode::BasicBlock* next_test_block = &generator.make_block();
  2236. bool did_create_lexical_environment = false;
  2237. if (has_lexical_declarations())
  2238. did_create_lexical_environment = generator.emit_block_declaration_instantiation(*this);
  2239. generator.emit<Bytecode::Op::Jump>(Bytecode::Label { *next_test_block });
  2240. Queue<Bytecode::BasicBlock*> test_blocks;
  2241. for (auto& switch_case : m_cases) {
  2242. if (switch_case->test())
  2243. test_blocks.enqueue(&generator.make_block());
  2244. }
  2245. for (auto& switch_case : m_cases) {
  2246. auto& case_block = generator.make_block();
  2247. if (switch_case->test()) {
  2248. generator.switch_to_basic_block(*next_test_block);
  2249. auto test_value = TRY(switch_case->test()->generate_bytecode(generator)).value();
  2250. auto result = generator.allocate_register();
  2251. generator.emit<Bytecode::Op::StrictlyEquals>(result, test_value, discriminant);
  2252. next_test_block = test_blocks.dequeue();
  2253. generator.emit_jump_if(
  2254. result,
  2255. Bytecode::Label { case_block },
  2256. Bytecode::Label { *next_test_block });
  2257. } else {
  2258. entry_block_for_default = &case_block;
  2259. }
  2260. case_blocks.append(case_block);
  2261. }
  2262. generator.switch_to_basic_block(*next_test_block);
  2263. auto& end_block = generator.make_block();
  2264. if (entry_block_for_default != nullptr) {
  2265. generator.emit<Bytecode::Op::Jump>(Bytecode::Label { *entry_block_for_default });
  2266. } else {
  2267. generator.emit<Bytecode::Op::Jump>(Bytecode::Label { end_block });
  2268. }
  2269. auto current_block = case_blocks.begin();
  2270. generator.begin_breakable_scope(Bytecode::Label { end_block }, label_set);
  2271. for (auto& switch_case : m_cases) {
  2272. generator.switch_to_basic_block(*current_block);
  2273. for (auto& statement : switch_case->children()) {
  2274. auto result = TRY(statement->generate_bytecode(generator));
  2275. if (generator.is_current_block_terminated())
  2276. break;
  2277. if (result.has_value())
  2278. generator.emit<Bytecode::Op::Mov>(dst, *result);
  2279. else
  2280. generator.emit<Bytecode::Op::Mov>(dst, generator.add_constant(js_undefined()));
  2281. }
  2282. if (!generator.is_current_block_terminated()) {
  2283. auto next_block = current_block;
  2284. next_block++;
  2285. if (next_block.is_end()) {
  2286. generator.emit<Bytecode::Op::Jump>(Bytecode::Label { end_block });
  2287. } else {
  2288. generator.emit<Bytecode::Op::Jump>(Bytecode::Label { *next_block });
  2289. }
  2290. }
  2291. current_block++;
  2292. }
  2293. generator.end_breakable_scope();
  2294. generator.switch_to_basic_block(end_block);
  2295. if (did_create_lexical_environment)
  2296. generator.end_variable_scope();
  2297. return dst;
  2298. }
  2299. Bytecode::CodeGenerationErrorOr<Optional<ScopedOperand>> SuperExpression::generate_bytecode(Bytecode::Generator&, [[maybe_unused]] Optional<ScopedOperand> preferred_dst) const
  2300. {
  2301. // The semantics for SuperExpression are handled in CallExpression and SuperCall.
  2302. VERIFY_NOT_REACHED();
  2303. }
  2304. Bytecode::CodeGenerationErrorOr<Optional<ScopedOperand>> ClassDeclaration::generate_bytecode(Bytecode::Generator& generator, [[maybe_unused]] Optional<ScopedOperand> preferred_dst) const
  2305. {
  2306. Bytecode::Generator::SourceLocationScope scope(generator, *this);
  2307. auto value = TRY(m_class_expression->generate_bytecode(generator)).value();
  2308. generator.emit_set_variable(*m_class_expression.ptr()->m_name, value, Bytecode::Op::SetVariable::InitializationMode::Initialize);
  2309. // NOTE: ClassDeclaration does not produce a value.
  2310. return Optional<ScopedOperand> {};
  2311. }
  2312. // 15.7.14 Runtime Semantics: ClassDefinitionEvaluation, https://tc39.es/ecma262/#sec-runtime-semantics-classdefinitionevaluation
  2313. Bytecode::CodeGenerationErrorOr<Optional<ScopedOperand>> ClassExpression::generate_bytecode_with_lhs_name(Bytecode::Generator& generator, Optional<Bytecode::IdentifierTableIndex> lhs_name, Optional<ScopedOperand> preferred_dst) const
  2314. {
  2315. // NOTE: Step 2 is not a part of NewClass instruction because it is assumed to be done before super class expression evaluation
  2316. generator.emit<Bytecode::Op::CreateLexicalEnvironment>();
  2317. if (has_name() || !lhs_name.has_value()) {
  2318. // NOTE: Step 3.a is not a part of NewClass instruction because it is assumed to be done before super class expression evaluation
  2319. auto interned_index = generator.intern_identifier(name());
  2320. generator.emit<Bytecode::Op::CreateVariable>(interned_index, Bytecode::Op::EnvironmentMode::Lexical, true);
  2321. }
  2322. Optional<ScopedOperand> super_class;
  2323. if (m_super_class)
  2324. super_class = TRY(m_super_class->generate_bytecode(generator)).value();
  2325. generator.emit<Op::CreatePrivateEnvironment>();
  2326. for (auto const& element : m_elements) {
  2327. auto opt_private_name = element->private_bound_identifier();
  2328. if (opt_private_name.has_value()) {
  2329. generator.emit<Op::AddPrivateName>(generator.intern_identifier(*opt_private_name));
  2330. }
  2331. }
  2332. Vector<Optional<ScopedOperand>> elements;
  2333. for (auto const& element : m_elements) {
  2334. Optional<ScopedOperand> key;
  2335. if (is<ClassMethod>(*element)) {
  2336. auto const& class_method = static_cast<ClassMethod const&>(*element);
  2337. if (!is<PrivateIdentifier>(class_method.key()))
  2338. key = TRY(class_method.key().generate_bytecode(generator));
  2339. } else if (is<ClassField>(*element)) {
  2340. auto const& class_field = static_cast<ClassField const&>(*element);
  2341. if (!is<PrivateIdentifier>(class_field.key()))
  2342. key = TRY(class_field.key().generate_bytecode(generator));
  2343. }
  2344. elements.append({ key });
  2345. }
  2346. auto dst = choose_dst(generator, preferred_dst);
  2347. generator.emit_with_extra_slots<Op::NewClass, Optional<Operand>>(elements.size(), dst, super_class.has_value() ? super_class->operand() : Optional<Operand> {}, *this, lhs_name, elements);
  2348. generator.emit<Op::LeavePrivateEnvironment>();
  2349. return dst;
  2350. }
  2351. Bytecode::CodeGenerationErrorOr<Optional<ScopedOperand>> ClassExpression::generate_bytecode(Bytecode::Generator& generator, Optional<ScopedOperand> preferred_dst) const
  2352. {
  2353. Bytecode::Generator::SourceLocationScope scope(generator, *this);
  2354. return generate_bytecode_with_lhs_name(generator, {}, preferred_dst);
  2355. }
  2356. Bytecode::CodeGenerationErrorOr<Optional<ScopedOperand>> SpreadExpression::generate_bytecode(Bytecode::Generator& generator, [[maybe_unused]] Optional<ScopedOperand> preferred_dst) const
  2357. {
  2358. // NOTE: All users of this should handle the behaviour of this on their own,
  2359. // assuming it returns an Array-like object
  2360. return m_target->generate_bytecode(generator);
  2361. }
  2362. Bytecode::CodeGenerationErrorOr<Optional<ScopedOperand>> ThisExpression::generate_bytecode(Bytecode::Generator& generator, Optional<ScopedOperand> preferred_dst) const
  2363. {
  2364. Bytecode::Generator::SourceLocationScope scope(generator, *this);
  2365. return generator.get_this(preferred_dst);
  2366. }
  2367. static ScopedOperand generate_await(
  2368. Bytecode::Generator& generator,
  2369. ScopedOperand argument,
  2370. ScopedOperand received_completion,
  2371. ScopedOperand received_completion_type,
  2372. ScopedOperand received_completion_value,
  2373. Bytecode::IdentifierTableIndex type_identifier,
  2374. Bytecode::IdentifierTableIndex value_identifier)
  2375. {
  2376. VERIFY(generator.is_in_async_function());
  2377. auto& continuation_block = generator.make_block();
  2378. generator.emit<Bytecode::Op::Await>(Bytecode::Label { continuation_block }, argument);
  2379. generator.switch_to_basic_block(continuation_block);
  2380. // FIXME: It's really magical that we can just assume that the completion value is in register 0.
  2381. // It ends up there because we "return" from the Await instruction above via the synthetic
  2382. // generator function that actually drives async execution.
  2383. generator.emit<Bytecode::Op::Mov>(received_completion, generator.accumulator());
  2384. get_received_completion_type_and_value(generator, received_completion, received_completion_type, received_completion_value, type_identifier, value_identifier);
  2385. auto& normal_completion_continuation_block = generator.make_block();
  2386. auto& throw_value_block = generator.make_block();
  2387. auto received_completion_type_is_normal = generator.allocate_register();
  2388. generator.emit<Bytecode::Op::StrictlyEquals>(
  2389. received_completion_type_is_normal,
  2390. received_completion_type,
  2391. generator.add_constant(Value(to_underlying(Completion::Type::Normal))));
  2392. generator.emit_jump_if(
  2393. received_completion_type_is_normal,
  2394. Bytecode::Label { normal_completion_continuation_block },
  2395. Bytecode::Label { throw_value_block });
  2396. // Simplification: The only abrupt completion we receive from AsyncFunctionDriverWrapper or AsyncGenerator is Type::Throw
  2397. // So we do not need to account for the Type::Return path
  2398. generator.switch_to_basic_block(throw_value_block);
  2399. generator.perform_needed_unwinds<Bytecode::Op::Throw>();
  2400. generator.emit<Bytecode::Op::Throw>(received_completion_value);
  2401. generator.switch_to_basic_block(normal_completion_continuation_block);
  2402. return received_completion_value;
  2403. }
  2404. Bytecode::CodeGenerationErrorOr<Optional<ScopedOperand>> AwaitExpression::generate_bytecode(Bytecode::Generator& generator, [[maybe_unused]] Optional<ScopedOperand> preferred_dst) const
  2405. {
  2406. Bytecode::Generator::SourceLocationScope scope(generator, *this);
  2407. auto argument = TRY(m_argument->generate_bytecode(generator)).value();
  2408. auto received_completion = generator.allocate_register();
  2409. auto received_completion_type = generator.allocate_register();
  2410. auto received_completion_value = generator.allocate_register();
  2411. generator.emit<Bytecode::Op::Mov>(received_completion, generator.accumulator());
  2412. auto type_identifier = generator.intern_identifier("type");
  2413. auto value_identifier = generator.intern_identifier("value");
  2414. return generate_await(generator, argument, received_completion, received_completion_type, received_completion_value, type_identifier, value_identifier);
  2415. }
  2416. Bytecode::CodeGenerationErrorOr<Optional<ScopedOperand>> WithStatement::generate_bytecode(Bytecode::Generator& generator, [[maybe_unused]] Optional<ScopedOperand> preferred_dst) const
  2417. {
  2418. Bytecode::Generator::SourceLocationScope scope(generator, *this);
  2419. auto object = TRY(m_object->generate_bytecode(generator)).value();
  2420. generator.emit<Bytecode::Op::EnterObjectEnvironment>(object);
  2421. // EnterObjectEnvironment sets the running execution context's lexical_environment to a new Object Environment.
  2422. generator.start_boundary(Bytecode::Generator::BlockBoundaryType::LeaveLexicalEnvironment);
  2423. auto body_result = TRY(m_body->generate_bytecode(generator));
  2424. if (!body_result.has_value())
  2425. body_result = generator.add_constant(js_undefined());
  2426. generator.end_boundary(Bytecode::Generator::BlockBoundaryType::LeaveLexicalEnvironment);
  2427. if (!generator.is_current_block_terminated())
  2428. generator.emit<Bytecode::Op::LeaveLexicalEnvironment>();
  2429. return body_result;
  2430. }
  2431. enum class LHSKind {
  2432. Assignment,
  2433. VarBinding,
  2434. LexicalBinding,
  2435. };
  2436. enum class IterationKind {
  2437. Enumerate,
  2438. Iterate,
  2439. AsyncIterate,
  2440. };
  2441. // 14.7.5.6 ForIn/OfHeadEvaluation ( uninitializedBoundNames, expr, iterationKind ), https://tc39.es/ecma262/#sec-runtime-semantics-forinofheadevaluation
  2442. struct ForInOfHeadEvaluationResult {
  2443. bool is_destructuring { false };
  2444. LHSKind lhs_kind { LHSKind::Assignment };
  2445. Optional<ScopedOperand> iterator;
  2446. };
  2447. 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)
  2448. {
  2449. ForInOfHeadEvaluationResult result {};
  2450. bool entered_lexical_scope = false;
  2451. if (auto* ast_ptr = lhs.get_pointer<NonnullRefPtr<ASTNode const>>(); ast_ptr && is<VariableDeclaration>(**ast_ptr)) {
  2452. // Runtime Semantics: ForInOfLoopEvaluation, for any of:
  2453. // ForInOfStatement : for ( var ForBinding in Expression ) Statement
  2454. // ForInOfStatement : for ( ForDeclaration in Expression ) Statement
  2455. // ForInOfStatement : for ( var ForBinding of AssignmentExpression ) Statement
  2456. // ForInOfStatement : for ( ForDeclaration of AssignmentExpression ) Statement
  2457. auto& variable_declaration = static_cast<VariableDeclaration const&>(**ast_ptr);
  2458. result.is_destructuring = variable_declaration.declarations().first()->target().has<NonnullRefPtr<BindingPattern const>>();
  2459. result.lhs_kind = variable_declaration.is_lexical_declaration() ? LHSKind::LexicalBinding : LHSKind::VarBinding;
  2460. if (variable_declaration.declaration_kind() == DeclarationKind::Var) {
  2461. // B.3.5 Initializers in ForIn Statement Heads, https://tc39.es/ecma262/#sec-initializers-in-forin-statement-heads
  2462. auto& variable = variable_declaration.declarations().first();
  2463. if (variable->init()) {
  2464. VERIFY(variable->target().has<NonnullRefPtr<Identifier const>>());
  2465. auto identifier = variable->target().get<NonnullRefPtr<Identifier const>>();
  2466. auto identifier_table_ref = generator.intern_identifier(identifier->string());
  2467. auto value = TRY(generator.emit_named_evaluation_if_anonymous_function(*variable->init(), identifier_table_ref)).value();
  2468. generator.emit_set_variable(*identifier, value);
  2469. }
  2470. } else {
  2471. auto has_non_local_variables = false;
  2472. MUST(variable_declaration.for_each_bound_identifier([&](auto const& identifier) {
  2473. if (!identifier.is_local())
  2474. has_non_local_variables = true;
  2475. }));
  2476. if (has_non_local_variables) {
  2477. // 1. Let oldEnv be the running execution context's LexicalEnvironment.
  2478. // NOTE: 'uninitializedBoundNames' refers to the lexical bindings (i.e. Const/Let) present in the second and last form.
  2479. // 2. If uninitializedBoundNames is not an empty List, then
  2480. entered_lexical_scope = true;
  2481. // a. Assert: uninitializedBoundNames has no duplicate entries.
  2482. // b. Let newEnv be NewDeclarativeEnvironment(oldEnv).
  2483. generator.begin_variable_scope();
  2484. // c. For each String name of uninitializedBoundNames, do
  2485. // NOTE: Nothing in the callback throws an exception.
  2486. MUST(variable_declaration.for_each_bound_identifier([&](auto const& identifier) {
  2487. if (identifier.is_local())
  2488. return;
  2489. // i. Perform ! newEnv.CreateMutableBinding(name, false).
  2490. auto interned_identifier = generator.intern_identifier(identifier.string());
  2491. generator.emit<Bytecode::Op::CreateVariable>(interned_identifier, Bytecode::Op::EnvironmentMode::Lexical, false);
  2492. }));
  2493. // d. Set the running execution context's LexicalEnvironment to newEnv.
  2494. // NOTE: Done by CreateLexicalEnvironment.
  2495. }
  2496. }
  2497. } else {
  2498. // Runtime Semantics: ForInOfLoopEvaluation, for any of:
  2499. // ForInOfStatement : for ( LeftHandSideExpression in Expression ) Statement
  2500. // ForInOfStatement : for ( LeftHandSideExpression of AssignmentExpression ) Statement
  2501. result.lhs_kind = LHSKind::Assignment;
  2502. }
  2503. // 3. Let exprRef be the result of evaluating expr.
  2504. auto object = TRY(rhs->generate_bytecode(generator)).value();
  2505. // 4. Set the running execution context's LexicalEnvironment to oldEnv.
  2506. if (entered_lexical_scope)
  2507. generator.end_variable_scope();
  2508. // 5. Let exprValue be ? GetValue(exprRef).
  2509. // NOTE: No need to store this anywhere.
  2510. auto iterator = generator.allocate_register();
  2511. // 6. If iterationKind is enumerate, then
  2512. if (iteration_kind == IterationKind::Enumerate) {
  2513. // a. If exprValue is undefined or null, then
  2514. auto& nullish_block = generator.make_block();
  2515. auto& continuation_block = generator.make_block();
  2516. generator.emit<Bytecode::Op::JumpNullish>(
  2517. object,
  2518. Bytecode::Label { nullish_block },
  2519. Bytecode::Label { continuation_block });
  2520. // i. Return Completion Record { [[Type]]: break, [[Value]]: empty, [[Target]]: empty }.
  2521. generator.switch_to_basic_block(nullish_block);
  2522. generator.generate_break();
  2523. generator.switch_to_basic_block(continuation_block);
  2524. // b. Let obj be ! ToObject(exprValue).
  2525. // NOTE: GetObjectPropertyIterator does this.
  2526. // c. Let iterator be EnumerateObjectProperties(obj).
  2527. // d. Let nextMethod be ! GetV(iterator, "next").
  2528. // e. Return the Iterator Record { [[Iterator]]: iterator, [[NextMethod]]: nextMethod, [[Done]]: false }.
  2529. generator.emit<Bytecode::Op::GetObjectPropertyIterator>(iterator, object);
  2530. }
  2531. // 7. Else,
  2532. else {
  2533. // a. Assert: iterationKind is iterate or async-iterate.
  2534. // b. If iterationKind is async-iterate, let iteratorKind be async.
  2535. // c. Else, let iteratorKind be sync.
  2536. auto iterator_kind = iteration_kind == IterationKind::AsyncIterate ? IteratorHint::Async : IteratorHint::Sync;
  2537. // d. Return ? GetIterator(exprValue, iteratorKind).
  2538. generator.emit<Bytecode::Op::GetIterator>(iterator, object, iterator_kind);
  2539. }
  2540. result.iterator = iterator;
  2541. return result;
  2542. }
  2543. // 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
  2544. static Bytecode::CodeGenerationErrorOr<Optional<ScopedOperand>> 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, IteratorHint iterator_kind = IteratorHint::Sync, [[maybe_unused]] Optional<ScopedOperand> preferred_dst = {})
  2545. {
  2546. // 1. If iteratorKind is not present, set iteratorKind to sync.
  2547. // 2. Let oldEnv be the running execution context's LexicalEnvironment.
  2548. bool has_lexical_binding = false;
  2549. // 3. Let V be undefined.
  2550. auto completion_value = generator.allocate_register();
  2551. generator.emit<Bytecode::Op::Mov>(completion_value, generator.add_constant(js_undefined()));
  2552. // 4. Let destructuring be IsDestructuring of lhs.
  2553. auto destructuring = head_result.is_destructuring;
  2554. // 5. If destructuring is true and if lhsKind is assignment, then
  2555. if (destructuring && head_result.lhs_kind == LHSKind::Assignment) {
  2556. // a. Assert: lhs is a LeftHandSideExpression.
  2557. // b. Let assignmentPattern be the AssignmentPattern that is covered by lhs.
  2558. // FIXME: Implement this.
  2559. return Bytecode::CodeGenerationError {
  2560. &node,
  2561. "Unimplemented: assignment destructuring in for/of"sv,
  2562. };
  2563. }
  2564. // 6. Repeat,
  2565. generator.emit<Bytecode::Op::Jump>(Bytecode::Label { loop_update });
  2566. generator.switch_to_basic_block(loop_update);
  2567. generator.begin_continuable_scope(Bytecode::Label { loop_update }, label_set);
  2568. // a. Let nextResult be ? Call(iteratorRecord.[[NextMethod]], iteratorRecord.[[Iterator]]).
  2569. auto next_result = generator.allocate_register();
  2570. generator.emit<Bytecode::Op::IteratorNext>(next_result, *head_result.iterator);
  2571. // b. If iteratorKind is async, set nextResult to ? Await(nextResult).
  2572. if (iterator_kind == IteratorHint::Async) {
  2573. auto received_completion = generator.allocate_register();
  2574. auto received_completion_type = generator.allocate_register();
  2575. auto received_completion_value = generator.allocate_register();
  2576. auto type_identifier = generator.intern_identifier("type");
  2577. auto value_identifier = generator.intern_identifier("value");
  2578. generator.emit<Bytecode::Op::Mov>(received_completion, generator.accumulator());
  2579. auto new_result = generate_await(generator, next_result, received_completion, received_completion_type, received_completion_value, type_identifier, value_identifier);
  2580. generator.emit<Bytecode::Op::Mov>(next_result, new_result);
  2581. }
  2582. // c. If Type(nextResult) is not Object, throw a TypeError exception.
  2583. generator.emit<Bytecode::Op::ThrowIfNotObject>(next_result);
  2584. // d. Let done be ? IteratorComplete(nextResult).
  2585. auto done = generator.allocate_register();
  2586. generator.emit_iterator_complete(done, next_result);
  2587. // e. If done is true, return V.
  2588. auto& loop_continue = generator.make_block();
  2589. generator.emit_jump_if(
  2590. done,
  2591. Bytecode::Label { loop_end },
  2592. Bytecode::Label { loop_continue });
  2593. generator.switch_to_basic_block(loop_continue);
  2594. // f. Let nextValue be ? IteratorValue(nextResult).
  2595. auto next_value = generator.allocate_register();
  2596. generator.emit_iterator_value(next_value, next_result);
  2597. // g. If lhsKind is either assignment or varBinding, then
  2598. if (head_result.lhs_kind != LHSKind::LexicalBinding) {
  2599. // i. If destructuring is false, then
  2600. if (!destructuring) {
  2601. // 1. Let lhsRef be the result of evaluating lhs. (It may be evaluated repeatedly.)
  2602. // NOTE: We're skipping all the completion stuff that the spec does, as the unwinding mechanism will take case of doing that.
  2603. if (head_result.lhs_kind == LHSKind::VarBinding) {
  2604. auto& declaration = static_cast<VariableDeclaration const&>(*lhs.get<NonnullRefPtr<ASTNode const>>());
  2605. VERIFY(declaration.declarations().size() == 1);
  2606. TRY(assign_value_to_variable_declarator(generator, declaration.declarations().first(), declaration, next_value));
  2607. } else {
  2608. if (auto ptr = lhs.get_pointer<NonnullRefPtr<ASTNode const>>()) {
  2609. TRY(generator.emit_store_to_reference(**ptr, next_value));
  2610. } else {
  2611. auto& binding_pattern = lhs.get<NonnullRefPtr<BindingPattern const>>();
  2612. TRY(binding_pattern->generate_bytecode(generator, Bytecode::Op::SetVariable::InitializationMode::Set, next_value, false));
  2613. }
  2614. }
  2615. }
  2616. }
  2617. // h. Else,
  2618. else {
  2619. // i. Assert: lhsKind is lexicalBinding.
  2620. // ii. Assert: lhs is a ForDeclaration.
  2621. // iii. Let iterationEnv be NewDeclarativeEnvironment(oldEnv).
  2622. // iv. Perform ForDeclarationBindingInstantiation of lhs with argument iterationEnv.
  2623. // v. Set the running execution context's LexicalEnvironment to iterationEnv.
  2624. // 14.7.5.4 Runtime Semantics: ForDeclarationBindingInstantiation, https://tc39.es/ecma262/#sec-runtime-semantics-fordeclarationbindinginstantiation
  2625. // 1. Assert: environment is a declarative Environment Record.
  2626. // NOTE: We just made it.
  2627. auto& variable_declaration = static_cast<VariableDeclaration const&>(*lhs.get<NonnullRefPtr<ASTNode const>>());
  2628. // 2. For each element name of the BoundNames of ForBinding, do
  2629. // NOTE: Nothing in the callback throws an exception.
  2630. auto has_non_local_variables = false;
  2631. MUST(variable_declaration.for_each_bound_identifier([&](auto const& identifier) {
  2632. if (!identifier.is_local())
  2633. has_non_local_variables = true;
  2634. }));
  2635. if (has_non_local_variables) {
  2636. generator.begin_variable_scope();
  2637. has_lexical_binding = true;
  2638. MUST(variable_declaration.for_each_bound_identifier([&](auto const& identifier) {
  2639. if (identifier.is_local())
  2640. return;
  2641. auto interned_identifier = generator.intern_identifier(identifier.string());
  2642. // a. If IsConstantDeclaration of LetOrConst is true, then
  2643. if (variable_declaration.is_constant_declaration()) {
  2644. // i. Perform ! environment.CreateImmutableBinding(name, true).
  2645. generator.emit<Bytecode::Op::CreateVariable>(interned_identifier, Bytecode::Op::EnvironmentMode::Lexical, true, false, true);
  2646. }
  2647. // b. Else,
  2648. else {
  2649. // i. Perform ! environment.CreateMutableBinding(name, false).
  2650. generator.emit<Bytecode::Op::CreateVariable>(interned_identifier, Bytecode::Op::EnvironmentMode::Lexical, false);
  2651. }
  2652. }));
  2653. // 3. Return unused.
  2654. // NOTE: No need to do that as we've inlined this.
  2655. }
  2656. // vi. If destructuring is false, then
  2657. if (!destructuring) {
  2658. // 1. Assert: lhs binds a single name.
  2659. // 2. Let lhsName be the sole element of BoundNames of lhs.
  2660. auto lhs_name = variable_declaration.declarations().first()->target().get<NonnullRefPtr<Identifier const>>();
  2661. // 3. Let lhsRef be ! ResolveBinding(lhsName).
  2662. // NOTE: We're skipping all the completion stuff that the spec does, as the unwinding mechanism will take case of doing that.
  2663. generator.emit_set_variable(*lhs_name, next_value, Bytecode::Op::SetVariable::InitializationMode::Initialize, Bytecode::Op::EnvironmentMode::Lexical);
  2664. }
  2665. }
  2666. // i. If destructuring is false, then
  2667. if (!destructuring) {
  2668. // i. If lhsRef is an abrupt completion, then
  2669. // 1. Let status be lhsRef.
  2670. // ii. Else if lhsKind is lexicalBinding, then
  2671. // 1. Let status be Completion(InitializeReferencedBinding(lhsRef, nextValue)).
  2672. // iii. Else,
  2673. // 1. Let status be Completion(PutValue(lhsRef, nextValue)).
  2674. // NOTE: This is performed above.
  2675. }
  2676. // j. Else,
  2677. else {
  2678. // FIXME: i. If lhsKind is assignment, then
  2679. // 1. Let status be Completion(DestructuringAssignmentEvaluation of assignmentPattern with argument nextValue).
  2680. // ii. Else if lhsKind is varBinding, then
  2681. // 1. Assert: lhs is a ForBinding.
  2682. // 2. Let status be Completion(BindingInitialization of lhs with arguments nextValue and undefined).
  2683. // iii. Else,
  2684. // 1. Assert: lhsKind is lexicalBinding.
  2685. // 2. Assert: lhs is a ForDeclaration.
  2686. // 3. Let status be Completion(ForDeclarationBindingInitialization of lhs with arguments nextValue and iterationEnv).
  2687. if (head_result.lhs_kind == LHSKind::VarBinding || head_result.lhs_kind == LHSKind::LexicalBinding) {
  2688. auto& declaration = static_cast<VariableDeclaration const&>(*lhs.get<NonnullRefPtr<ASTNode const>>());
  2689. VERIFY(declaration.declarations().size() == 1);
  2690. auto& binding_pattern = declaration.declarations().first()->target().get<NonnullRefPtr<BindingPattern const>>();
  2691. (void)TRY(binding_pattern->generate_bytecode(
  2692. generator,
  2693. head_result.lhs_kind == LHSKind::VarBinding ? Bytecode::Op::SetVariable::InitializationMode::Set : Bytecode::Op::SetVariable::InitializationMode::Initialize,
  2694. next_value,
  2695. false));
  2696. } else {
  2697. return Bytecode::CodeGenerationError {
  2698. &node,
  2699. "Unimplemented: assignment destructuring in for/of"sv,
  2700. };
  2701. }
  2702. }
  2703. // FIXME: Implement iteration closure.
  2704. // k. If status is an abrupt completion, then
  2705. // i. Set the running execution context's LexicalEnvironment to oldEnv.
  2706. // ii. If iteratorKind is async, return ? AsyncIteratorClose(iteratorRecord, status).
  2707. // iii. If iterationKind is enumerate, then
  2708. // 1. Return ? status.
  2709. // iv. Else,
  2710. // 1. Assert: iterationKind is iterate.
  2711. // 2. Return ? IteratorClose(iteratorRecord, status).
  2712. // l. Let result be the result of evaluating stmt.
  2713. auto result = TRY(body.generate_bytecode(generator));
  2714. // m. Set the running execution context's LexicalEnvironment to oldEnv.
  2715. if (has_lexical_binding)
  2716. generator.end_variable_scope();
  2717. generator.end_continuable_scope();
  2718. generator.end_breakable_scope();
  2719. // NOTE: If we're here, then the loop definitely continues.
  2720. // n. If LoopContinues(result, labelSet) is false, then
  2721. // i. If iterationKind is enumerate, then
  2722. // 1. Return ? UpdateEmpty(result, V).
  2723. // ii. Else,
  2724. // 1. Assert: iterationKind is iterate.
  2725. // 2. Set status to Completion(UpdateEmpty(result, V)).
  2726. // 3. If iteratorKind is async, return ? AsyncIteratorClose(iteratorRecord, status).
  2727. // 4. Return ? IteratorClose(iteratorRecord, status).
  2728. // o. If result.[[Value]] is not empty, set V to result.[[Value]].
  2729. // The body can contain an unconditional block terminator (e.g. return, throw), so we have to check for that before generating the Jump.
  2730. if (!generator.is_current_block_terminated()) {
  2731. if (result.has_value())
  2732. generator.emit<Bytecode::Op::Mov>(completion_value, *result);
  2733. generator.emit<Bytecode::Op::Jump>(Bytecode::Label { loop_update });
  2734. }
  2735. generator.switch_to_basic_block(loop_end);
  2736. return completion_value;
  2737. }
  2738. Bytecode::CodeGenerationErrorOr<Optional<ScopedOperand>> ForInStatement::generate_bytecode(Bytecode::Generator& generator, [[maybe_unused]] Optional<ScopedOperand> preferred_dst) const
  2739. {
  2740. Bytecode::Generator::SourceLocationScope scope(generator, *this);
  2741. return generate_labelled_evaluation(generator, {});
  2742. }
  2743. // 14.7.5.5 Runtime Semantics: ForInOfLoopEvaluation, https://tc39.es/ecma262/#sec-runtime-semantics-forinofloopevaluation
  2744. Bytecode::CodeGenerationErrorOr<Optional<ScopedOperand>> ForInStatement::generate_labelled_evaluation(Bytecode::Generator& generator, Vector<DeprecatedFlyString> const& label_set, [[maybe_unused]] Optional<ScopedOperand> preferred_dst) const
  2745. {
  2746. auto& loop_end = generator.make_block();
  2747. auto& loop_update = generator.make_block();
  2748. generator.begin_breakable_scope(Bytecode::Label { loop_end }, label_set);
  2749. auto head_result = TRY(for_in_of_head_evaluation(generator, IterationKind::Enumerate, m_lhs, m_rhs));
  2750. return for_in_of_body_evaluation(generator, *this, m_lhs, body(), head_result, label_set, loop_end, loop_update);
  2751. }
  2752. Bytecode::CodeGenerationErrorOr<Optional<ScopedOperand>> ForOfStatement::generate_bytecode(Bytecode::Generator& generator, [[maybe_unused]] Optional<ScopedOperand> preferred_dst) const
  2753. {
  2754. Bytecode::Generator::SourceLocationScope scope(generator, *this);
  2755. return generate_labelled_evaluation(generator, {});
  2756. }
  2757. Bytecode::CodeGenerationErrorOr<Optional<ScopedOperand>> ForOfStatement::generate_labelled_evaluation(Bytecode::Generator& generator, Vector<DeprecatedFlyString> const& label_set, [[maybe_unused]] Optional<ScopedOperand> preferred_dst) const
  2758. {
  2759. auto& loop_end = generator.make_block();
  2760. auto& loop_update = generator.make_block();
  2761. generator.begin_breakable_scope(Bytecode::Label { loop_end }, label_set);
  2762. auto head_result = TRY(for_in_of_head_evaluation(generator, IterationKind::Iterate, m_lhs, m_rhs));
  2763. return for_in_of_body_evaluation(generator, *this, m_lhs, body(), head_result, label_set, loop_end, loop_update);
  2764. }
  2765. Bytecode::CodeGenerationErrorOr<Optional<ScopedOperand>> ForAwaitOfStatement::generate_bytecode(Bytecode::Generator& generator, [[maybe_unused]] Optional<ScopedOperand> preferred_dst) const
  2766. {
  2767. Bytecode::Generator::SourceLocationScope scope(generator, *this);
  2768. return generate_labelled_evaluation(generator, {});
  2769. }
  2770. Bytecode::CodeGenerationErrorOr<Optional<ScopedOperand>> ForAwaitOfStatement::generate_labelled_evaluation(Bytecode::Generator& generator, Vector<DeprecatedFlyString> const& label_set, [[maybe_unused]] Optional<ScopedOperand> preferred_dst) const
  2771. {
  2772. auto& loop_end = generator.make_block();
  2773. auto& loop_update = generator.make_block();
  2774. generator.begin_breakable_scope(Bytecode::Label { loop_end }, label_set);
  2775. auto head_result = TRY(for_in_of_head_evaluation(generator, IterationKind::AsyncIterate, m_lhs, m_rhs));
  2776. return for_in_of_body_evaluation(generator, *this, m_lhs, m_body, head_result, label_set, loop_end, loop_update, IteratorHint::Async);
  2777. }
  2778. // 13.3.12.1 Runtime Semantics: Evaluation, https://tc39.es/ecma262/#sec-meta-properties-runtime-semantics-evaluation
  2779. Bytecode::CodeGenerationErrorOr<Optional<ScopedOperand>> MetaProperty::generate_bytecode(Bytecode::Generator& generator, Optional<ScopedOperand> preferred_dst) const
  2780. {
  2781. Bytecode::Generator::SourceLocationScope scope(generator, *this);
  2782. // NewTarget : new . target
  2783. if (m_type == MetaProperty::Type::NewTarget) {
  2784. // 1. Return GetNewTarget().
  2785. auto dst = choose_dst(generator, preferred_dst);
  2786. generator.emit<Bytecode::Op::GetNewTarget>(dst);
  2787. return dst;
  2788. }
  2789. // ImportMeta : import . meta
  2790. if (m_type == MetaProperty::Type::ImportMeta) {
  2791. auto dst = choose_dst(generator, preferred_dst);
  2792. generator.emit<Bytecode::Op::GetImportMeta>(dst);
  2793. return dst;
  2794. }
  2795. VERIFY_NOT_REACHED();
  2796. }
  2797. Bytecode::CodeGenerationErrorOr<Optional<ScopedOperand>> ClassFieldInitializerStatement::generate_bytecode(Bytecode::Generator& generator, Optional<ScopedOperand> preferred_dst) const
  2798. {
  2799. Bytecode::Generator::SourceLocationScope scope(generator, *this);
  2800. auto value = TRY(generator.emit_named_evaluation_if_anonymous_function(*m_expression, generator.intern_identifier(m_class_field_identifier_name), preferred_dst));
  2801. generator.perform_needed_unwinds<Bytecode::Op::Return>();
  2802. generator.emit<Bytecode::Op::Return>(value.has_value() ? value->operand() : Optional<Operand> {});
  2803. return value;
  2804. }
  2805. static Bytecode::CodeGenerationErrorOr<void> generate_optional_chain(Bytecode::Generator& generator, OptionalChain const& optional_chain, ScopedOperand current_value, ScopedOperand current_base, [[maybe_unused]] Optional<ScopedOperand> preferred_dst)
  2806. {
  2807. Optional<ScopedOperand> new_current_value;
  2808. if (is<MemberExpression>(optional_chain.base())) {
  2809. auto& member_expression = static_cast<MemberExpression const&>(optional_chain.base());
  2810. auto base_and_value = TRY(get_base_and_value_from_member_expression(generator, member_expression));
  2811. new_current_value = base_and_value.value;
  2812. generator.emit<Bytecode::Op::Mov>(current_base, base_and_value.base);
  2813. } else if (is<OptionalChain>(optional_chain.base())) {
  2814. auto& sub_optional_chain = static_cast<OptionalChain const&>(optional_chain.base());
  2815. TRY(generate_optional_chain(generator, sub_optional_chain, current_value, current_base));
  2816. new_current_value = current_value;
  2817. } else {
  2818. new_current_value = TRY(optional_chain.base().generate_bytecode(generator)).value();
  2819. }
  2820. generator.emit<Bytecode::Op::Mov>(current_value, *new_current_value);
  2821. auto& load_undefined_and_jump_to_end_block = generator.make_block();
  2822. auto& end_block = generator.make_block();
  2823. for (auto& reference : optional_chain.references()) {
  2824. auto is_optional = reference.visit([](auto& ref) { return ref.mode; }) == OptionalChain::Mode::Optional;
  2825. if (is_optional) {
  2826. auto& not_nullish_block = generator.make_block();
  2827. generator.emit<Bytecode::Op::JumpNullish>(
  2828. current_value,
  2829. Bytecode::Label { load_undefined_and_jump_to_end_block },
  2830. Bytecode::Label { not_nullish_block });
  2831. generator.switch_to_basic_block(not_nullish_block);
  2832. }
  2833. TRY(reference.visit(
  2834. [&](OptionalChain::Call const& call) -> Bytecode::CodeGenerationErrorOr<void> {
  2835. auto arguments = TRY(arguments_to_array_for_call(generator, call.arguments)).value();
  2836. generator.emit<Bytecode::Op::CallWithArgumentArray>(Bytecode::Op::CallType::Call, current_value, current_value, current_base, arguments);
  2837. generator.emit<Bytecode::Op::Mov>(current_base, generator.add_constant(js_undefined()));
  2838. return {};
  2839. },
  2840. [&](OptionalChain::ComputedReference const& ref) -> Bytecode::CodeGenerationErrorOr<void> {
  2841. generator.emit<Bytecode::Op::Mov>(current_base, current_value);
  2842. auto property = TRY(ref.expression->generate_bytecode(generator)).value();
  2843. generator.emit<Bytecode::Op::GetByValue>(current_value, current_value, property);
  2844. return {};
  2845. },
  2846. [&](OptionalChain::MemberReference const& ref) -> Bytecode::CodeGenerationErrorOr<void> {
  2847. generator.emit<Bytecode::Op::Mov>(current_base, current_value);
  2848. generator.emit_get_by_id(current_value, current_value, generator.intern_identifier(ref.identifier->string()));
  2849. return {};
  2850. },
  2851. [&](OptionalChain::PrivateMemberReference const& ref) -> Bytecode::CodeGenerationErrorOr<void> {
  2852. generator.emit<Bytecode::Op::Mov>(current_base, current_value);
  2853. generator.emit<Bytecode::Op::GetPrivateById>(current_value, current_value, generator.intern_identifier(ref.private_identifier->string()));
  2854. return {};
  2855. }));
  2856. }
  2857. generator.emit<Bytecode::Op::Jump>(Bytecode::Label { end_block });
  2858. generator.switch_to_basic_block(load_undefined_and_jump_to_end_block);
  2859. generator.emit<Bytecode::Op::Mov>(current_value, generator.add_constant(js_undefined()));
  2860. generator.emit<Bytecode::Op::Jump>(Bytecode::Label { end_block });
  2861. generator.switch_to_basic_block(end_block);
  2862. return {};
  2863. }
  2864. Bytecode::CodeGenerationErrorOr<Optional<ScopedOperand>> OptionalChain::generate_bytecode(Bytecode::Generator& generator, [[maybe_unused]] Optional<ScopedOperand> preferred_dst) const
  2865. {
  2866. Bytecode::Generator::SourceLocationScope scope(generator, *this);
  2867. auto current_base = generator.allocate_register();
  2868. auto current_value = choose_dst(generator, preferred_dst);
  2869. generator.emit<Bytecode::Op::Mov>(current_base, generator.add_constant(js_undefined()));
  2870. TRY(generate_optional_chain(generator, *this, current_value, current_base));
  2871. return current_value;
  2872. }
  2873. Bytecode::CodeGenerationErrorOr<Optional<ScopedOperand>> ImportCall::generate_bytecode(Bytecode::Generator& generator, Optional<ScopedOperand> preferred_dst) const
  2874. {
  2875. Bytecode::Generator::SourceLocationScope scope(generator, *this);
  2876. auto specifier = TRY(m_specifier->generate_bytecode(generator)).value();
  2877. Optional<ScopedOperand> options;
  2878. if (m_options) {
  2879. options = TRY(m_options->generate_bytecode(generator)).value();
  2880. } else {
  2881. options = generator.add_constant(js_undefined());
  2882. }
  2883. auto dst = choose_dst(generator, preferred_dst);
  2884. generator.emit<Bytecode::Op::ImportCall>(dst, specifier, *options);
  2885. return dst;
  2886. }
  2887. Bytecode::CodeGenerationErrorOr<Optional<ScopedOperand>> ExportStatement::generate_bytecode(Bytecode::Generator& generator, [[maybe_unused]] Optional<ScopedOperand> preferred_dst) const
  2888. {
  2889. Bytecode::Generator::SourceLocationScope scope(generator, *this);
  2890. if (!is_default_export()) {
  2891. if (m_statement) {
  2892. return m_statement->generate_bytecode(generator);
  2893. }
  2894. return Optional<ScopedOperand> {};
  2895. }
  2896. VERIFY(m_statement);
  2897. if (is<FunctionDeclaration>(*m_statement) || is<ClassDeclaration>(*m_statement)) {
  2898. return m_statement->generate_bytecode(generator);
  2899. }
  2900. if (is<ClassExpression>(*m_statement)) {
  2901. auto value = TRY(generator.emit_named_evaluation_if_anonymous_function(static_cast<ClassExpression const&>(*m_statement), generator.intern_identifier("default"sv))).value();
  2902. if (!static_cast<ClassExpression const&>(*m_statement).has_name()) {
  2903. generator.emit<Bytecode::Op::SetVariable>(
  2904. generator.intern_identifier(ExportStatement::local_name_for_default),
  2905. value,
  2906. generator.next_environment_variable_cache(),
  2907. Bytecode::Op::SetVariable::InitializationMode::Initialize);
  2908. }
  2909. return value;
  2910. }
  2911. // ExportDeclaration : export default AssignmentExpression ;
  2912. VERIFY(is<Expression>(*m_statement));
  2913. auto value = TRY(generator.emit_named_evaluation_if_anonymous_function(static_cast<Expression const&>(*m_statement), generator.intern_identifier("default"sv))).value();
  2914. generator.emit<Bytecode::Op::SetVariable>(
  2915. generator.intern_identifier(ExportStatement::local_name_for_default),
  2916. value,
  2917. generator.next_environment_variable_cache(),
  2918. Bytecode::Op::SetVariable::InitializationMode::Initialize);
  2919. return value;
  2920. }
  2921. Bytecode::CodeGenerationErrorOr<Optional<ScopedOperand>> ImportStatement::generate_bytecode(Bytecode::Generator&, [[maybe_unused]] Optional<ScopedOperand> preferred_dst) const
  2922. {
  2923. return Optional<ScopedOperand> {};
  2924. }
  2925. }