ASTCodegen.cpp 163 KB

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