ASTCodegen.cpp 143 KB

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