https://bugs.webkit.org/show_bug.cgi?id=63561
[WebKit-https.git] / Source / JavaScriptCore / dfg / DFGSpeculativeJIT.cpp
1 /*
2  * Copyright (C) 2011 Apple Inc. All rights reserved.
3  *
4  * Redistribution and use in source and binary forms, with or without
5  * modification, are permitted provided that the following conditions
6  * are met:
7  * 1. Redistributions of source code must retain the above copyright
8  *    notice, this list of conditions and the following disclaimer.
9  * 2. Redistributions in binary form must reproduce the above copyright
10  *    notice, this list of conditions and the following disclaimer in the
11  *    documentation and/or other materials provided with the distribution.
12  *
13  * THIS SOFTWARE IS PROVIDED BY APPLE INC. ``AS IS'' AND ANY
14  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
15  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
16  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL APPLE INC. OR
17  * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
18  * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
19  * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
20  * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
21  * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
22  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
23  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 
24  */
25
26 #include "config.h"
27 #include "DFGSpeculativeJIT.h"
28
29 #if ENABLE(DFG_JIT)
30
31 namespace JSC { namespace DFG {
32
33 template<bool strict>
34 GPRReg SpeculativeJIT::fillSpeculateIntInternal(NodeIndex nodeIndex, DataFormat& returnFormat)
35 {
36     Node& node = m_jit.graph()[nodeIndex];
37     VirtualRegister virtualRegister = node.virtualRegister();
38     GenerationInfo& info = m_generationInfo[virtualRegister];
39
40     switch (info.registerFormat()) {
41     case DataFormatNone: {
42         GPRReg gpr = allocate();
43
44         if (node.isConstant()) {
45             m_gprs.retain(gpr, virtualRegister, SpillOrderConstant);
46             if (isInt32Constant(nodeIndex)) {
47                 m_jit.move(MacroAssembler::Imm32(valueOfInt32Constant(nodeIndex)), gpr);
48                 info.fillInteger(gpr);
49                 returnFormat = DataFormatInteger;
50                 return gpr;
51             }
52             m_jit.move(valueOfJSConstantAsImmPtr(nodeIndex), gpr);
53         } else {
54             DataFormat spillFormat = info.spillFormat();
55             ASSERT(spillFormat & DataFormatJS);
56
57             m_gprs.retain(gpr, virtualRegister, SpillOrderSpilled);
58
59             if (spillFormat == DataFormatJSInteger) {
60                 // If we know this was spilled as an integer we can fill without checking.
61                 if (strict) {
62                     m_jit.load32(JITCompiler::addressFor(virtualRegister), gpr);
63                     info.fillInteger(gpr);
64                     returnFormat = DataFormatInteger;
65                     return gpr;
66                 }
67                 m_jit.loadPtr(JITCompiler::addressFor(virtualRegister), gpr);
68                 info.fillJSValue(gpr, DataFormatJSInteger);
69                 returnFormat = DataFormatJSInteger;
70                 return gpr;
71             }
72             m_jit.loadPtr(JITCompiler::addressFor(virtualRegister), gpr);
73         }
74
75         // Fill as JSValue, and fall through.
76         info.fillJSValue(gpr, DataFormatJSInteger);
77         m_gprs.unlock(gpr);
78     }
79
80     case DataFormatJS: {
81         // Check the value is an integer.
82         GPRReg gpr = info.gpr();
83         m_gprs.lock(gpr);
84         speculationCheck(m_jit.branchPtr(MacroAssembler::Below, gpr, GPRInfo::tagTypeNumberRegister));
85         info.fillJSValue(gpr, DataFormatJSInteger);
86         // If !strict we're done, return.
87         if (!strict) {
88             returnFormat = DataFormatJSInteger;
89             return gpr;
90         }
91         // else fall through & handle as DataFormatJSInteger.
92         m_gprs.unlock(gpr);
93     }
94
95     case DataFormatJSInteger: {
96         // In a strict fill we need to strip off the value tag.
97         if (strict) {
98             GPRReg gpr = info.gpr();
99             GPRReg result;
100             // If the register has already been locked we need to take a copy.
101             // If not, we'll zero extend in place, so mark on the info that this is now type DataFormatInteger, not DataFormatJSInteger.
102             if (m_gprs.isLocked(gpr))
103                 result = allocate();
104             else {
105                 m_gprs.lock(gpr);
106                 info.fillInteger(gpr);
107                 result = gpr;
108             }
109             m_jit.zeroExtend32ToPtr(gpr, result);
110             returnFormat = DataFormatInteger;
111             return result;
112         }
113
114         GPRReg gpr = info.gpr();
115         m_gprs.lock(gpr);
116         returnFormat = DataFormatJSInteger;
117         return gpr;
118     }
119
120     case DataFormatInteger: {
121         GPRReg gpr = info.gpr();
122         m_gprs.lock(gpr);
123         returnFormat = DataFormatInteger;
124         return gpr;
125     }
126
127     case DataFormatDouble:
128     case DataFormatCell:
129     case DataFormatJSDouble:
130     case DataFormatJSCell: {
131         terminateSpeculativeExecution();
132         returnFormat = DataFormatInteger;
133         return allocate();
134     }
135     }
136
137     ASSERT_NOT_REACHED();
138     return InvalidGPRReg;
139 }
140
141 SpeculationCheck::SpeculationCheck(MacroAssembler::Jump check, SpeculativeJIT* jit, unsigned recoveryIndex)
142     : m_check(check)
143     , m_nodeIndex(jit->m_compileIndex)
144     , m_recoveryIndex(recoveryIndex)
145 {
146     for (gpr_iterator iter = jit->m_gprs.begin(); iter != jit->m_gprs.end(); ++iter) {
147         if (iter.name() != InvalidVirtualRegister) {
148             GenerationInfo& info =  jit->m_generationInfo[iter.name()];
149             m_gprInfo[iter.index()].nodeIndex = info.nodeIndex();
150             m_gprInfo[iter.index()].format = info.registerFormat();
151         } else
152             m_gprInfo[iter.index()].nodeIndex = NoNode;
153     }
154     for (fpr_iterator iter = jit->m_fprs.begin(); iter != jit->m_fprs.end(); ++iter) {
155         if (iter.name() != InvalidVirtualRegister) {
156             GenerationInfo& info =  jit->m_generationInfo[iter.name()];
157             ASSERT(info.registerFormat() == DataFormatDouble);
158             m_fprInfo[iter.index()] = info.nodeIndex();
159         } else
160             m_fprInfo[iter.index()] = NoNode;
161     }
162 }
163
164 GPRReg SpeculativeJIT::fillSpeculateInt(NodeIndex nodeIndex, DataFormat& returnFormat)
165 {
166     return fillSpeculateIntInternal<false>(nodeIndex, returnFormat);
167 }
168
169 GPRReg SpeculativeJIT::fillSpeculateIntStrict(NodeIndex nodeIndex)
170 {
171     DataFormat mustBeDataFormatInteger;
172     GPRReg result = fillSpeculateIntInternal<true>(nodeIndex, mustBeDataFormatInteger);
173     ASSERT(mustBeDataFormatInteger == DataFormatInteger);
174     return result;
175 }
176
177 GPRReg SpeculativeJIT::fillSpeculateCell(NodeIndex nodeIndex)
178 {
179     Node& node = m_jit.graph()[nodeIndex];
180     VirtualRegister virtualRegister = node.virtualRegister();
181     GenerationInfo& info = m_generationInfo[virtualRegister];
182
183     switch (info.registerFormat()) {
184     case DataFormatNone: {
185         GPRReg gpr = allocate();
186
187         if (node.isConstant()) {
188             JSValue jsValue = valueOfJSConstant(nodeIndex);
189             if (jsValue.isCell()) {
190                 m_gprs.retain(gpr, virtualRegister, SpillOrderConstant);
191                 m_jit.move(MacroAssembler::TrustedImmPtr(jsValue.asCell()), gpr);
192                 info.fillJSValue(gpr, DataFormatJSCell);
193                 return gpr;
194             }
195             terminateSpeculativeExecution();
196             return gpr;
197         }
198         ASSERT(info.spillFormat() & DataFormatJS);
199         m_gprs.retain(gpr, virtualRegister, SpillOrderSpilled);
200         m_jit.loadPtr(JITCompiler::addressFor(virtualRegister), gpr);
201
202         if (info.spillFormat() != DataFormatJSCell)
203             speculationCheck(m_jit.branchTestPtr(MacroAssembler::NonZero, gpr, GPRInfo::tagMaskRegister));
204         info.fillJSValue(gpr, DataFormatJSCell);
205         return gpr;
206     }
207
208     case DataFormatCell:
209     case DataFormatJSCell: {
210         GPRReg gpr = info.gpr();
211         m_gprs.lock(gpr);
212         return gpr;
213     }
214
215     case DataFormatJS: {
216         GPRReg gpr = info.gpr();
217         m_gprs.lock(gpr);
218         speculationCheck(m_jit.branchTestPtr(MacroAssembler::NonZero, gpr, GPRInfo::tagMaskRegister));
219         info.fillJSValue(gpr, DataFormatJSCell);
220         return gpr;
221     }
222
223     case DataFormatJSInteger:
224     case DataFormatInteger:
225     case DataFormatJSDouble:
226     case DataFormatDouble: {
227         terminateSpeculativeExecution();
228         return allocate();
229     }
230     }
231
232     ASSERT_NOT_REACHED();
233     return InvalidGPRReg;
234 }
235
236 void SpeculativeJIT::compilePeepHoleIntegerBranch(Node& node, NodeIndex branchNodeIndex, JITCompiler::RelationalCondition condition)
237 {
238     Node& branchNode = m_jit.graph()[branchNodeIndex];
239     BlockIndex taken = m_jit.graph().blockIndexForBytecodeOffset(branchNode.takenBytecodeOffset());
240     BlockIndex notTaken = m_jit.graph().blockIndexForBytecodeOffset(branchNode.notTakenBytecodeOffset());
241
242     // The branch instruction will branch to the taken block.
243     // If taken is next, switch taken with notTaken & invert the branch condition so we can fall through.
244     if (taken == (m_block + 1)) {
245         condition = JITCompiler::invert(condition);
246         BlockIndex tmp = taken;
247         taken = notTaken;
248         notTaken = tmp;
249     }
250
251     if (isInt32Constant(node.child1)) {
252         int32_t imm = valueOfInt32Constant(node.child1);
253         SpeculateIntegerOperand op2(this, node.child2);
254         addBranch(m_jit.branch32(condition, JITCompiler::Imm32(imm), op2.gpr()), taken);
255     } else if (isInt32Constant(node.child2)) {
256         SpeculateIntegerOperand op1(this, node.child1);
257         int32_t imm = valueOfInt32Constant(node.child2);
258         addBranch(m_jit.branch32(condition, op1.gpr(), JITCompiler::Imm32(imm)), taken);
259     } else {
260         SpeculateIntegerOperand op1(this, node.child1);
261         SpeculateIntegerOperand op2(this, node.child2);
262         addBranch(m_jit.branch32(condition, op1.gpr(), op2.gpr()), taken);
263     }
264
265     // Check for fall through, otherwise we need to jump.
266     if (notTaken != (m_block + 1))
267         addBranch(m_jit.jump(), notTaken);
268 }
269
270 void SpeculativeJIT::compilePeepHoleCall(Node& node, NodeIndex branchNodeIndex, Z_DFGOperation_EJJ operation)
271 {
272     Node& branchNode = m_jit.graph()[branchNodeIndex];
273     BlockIndex taken = m_jit.graph().blockIndexForBytecodeOffset(branchNode.takenBytecodeOffset());
274     BlockIndex notTaken = m_jit.graph().blockIndexForBytecodeOffset(branchNode.notTakenBytecodeOffset());
275
276     // The branch instruction will branch to the taken block.
277     // If taken is next, switch taken with notTaken & invert the branch condition so we can fall through.
278     JITCompiler::ResultCondition condition = JITCompiler::NonZero;
279     if (taken == (m_block + 1)) {
280         condition = JITCompiler::Zero;
281         BlockIndex tmp = taken;
282         taken = notTaken;
283         notTaken = tmp;
284     }
285
286     JSValueOperand op1(this, node.child1);
287     JSValueOperand op2(this, node.child2);
288     GPRReg op1GPR = op1.gpr();
289     GPRReg op2GPR = op2.gpr();
290     flushRegisters();
291
292     GPRResult result(this);
293     callOperation(operation, result.gpr(), op1GPR, op2GPR);
294     addBranch(m_jit.branchTest8(condition, result.gpr()), taken);
295
296     // Check for fall through, otherwise we need to jump.
297     if (notTaken != (m_block + 1))
298         addBranch(m_jit.jump(), notTaken);
299 }
300
301 void SpeculativeJIT::compile(Node& node)
302 {
303     NodeType op = node.op;
304
305     switch (op) {
306     case JSConstant:
307         initConstantInfo(m_compileIndex);
308         break;
309
310     case GetLocal: {
311         GPRTemporary result(this);
312         PredictedType prediction = m_jit.graph().getPrediction(node.local());
313         if (prediction == PredictInt32) {
314             m_jit.load32(JITCompiler::payloadFor(node.local()), result.gpr());
315
316             // Like integerResult, but don't useChildren - our children are phi nodes,
317             // and don't represent values within this dataflow with virtual registers.
318             VirtualRegister virtualRegister = node.virtualRegister();
319             m_gprs.retain(result.gpr(), virtualRegister, SpillOrderInteger);
320             m_generationInfo[virtualRegister].initInteger(m_compileIndex, node.refCount(), result.gpr());
321         } else {
322             m_jit.loadPtr(JITCompiler::addressFor(node.local()), result.gpr());
323
324             // Like jsValueResult, but don't useChildren - our children are phi nodes,
325             // and don't represent values within this dataflow with virtual registers.
326             VirtualRegister virtualRegister = node.virtualRegister();
327             m_gprs.retain(result.gpr(), virtualRegister, SpillOrderJS);
328             m_generationInfo[virtualRegister].initJSValue(m_compileIndex, node.refCount(), result.gpr(), (prediction == PredictArray) ? DataFormatJSCell : DataFormatJS);
329         }
330         break;
331     }
332
333     case SetLocal: {
334         switch (m_jit.graph().getPrediction(node.local())) {
335         case PredictInt32: {
336             SpeculateIntegerOperand value(this, node.child1);
337             m_jit.store32(value.gpr(), JITCompiler::payloadFor(node.local()));
338             noResult(m_compileIndex);
339             break;
340         }
341         case PredictArray: {
342             SpeculateCellOperand cell(this, node.child1);
343             m_jit.storePtr(cell.gpr(), JITCompiler::addressFor(node.local()));
344             noResult(m_compileIndex);
345             break;
346         }
347
348         default: {
349             JSValueOperand value(this, node.child1);
350             m_jit.storePtr(value.gpr(), JITCompiler::addressFor(node.local()));
351             noResult(m_compileIndex);
352             break;
353         }
354         }
355         break;
356     }
357
358     case BitAnd:
359     case BitOr:
360     case BitXor:
361         if (isInt32Constant(node.child1)) {
362             SpeculateIntegerOperand op2(this, node.child2);
363             GPRTemporary result(this, op2);
364
365             bitOp(op, valueOfInt32Constant(node.child1), op2.gpr(), result.gpr());
366
367             integerResult(result.gpr(), m_compileIndex);
368         } else if (isInt32Constant(node.child2)) {
369             SpeculateIntegerOperand op1(this, node.child1);
370             GPRTemporary result(this, op1);
371
372             bitOp(op, valueOfInt32Constant(node.child2), op1.gpr(), result.gpr());
373
374             integerResult(result.gpr(), m_compileIndex);
375         } else {
376             SpeculateIntegerOperand op1(this, node.child1);
377             SpeculateIntegerOperand op2(this, node.child2);
378             GPRTemporary result(this, op1, op2);
379
380             GPRReg reg1 = op1.gpr();
381             GPRReg reg2 = op2.gpr();
382             bitOp(op, reg1, reg2, result.gpr());
383
384             integerResult(result.gpr(), m_compileIndex);
385         }
386         break;
387
388     case BitRShift:
389     case BitLShift:
390     case BitURShift:
391         if (isInt32Constant(node.child2)) {
392             SpeculateIntegerOperand op1(this, node.child1);
393             GPRTemporary result(this, op1);
394
395             shiftOp(op, op1.gpr(), valueOfInt32Constant(node.child2) & 0x1f, result.gpr());
396
397             integerResult(result.gpr(), m_compileIndex);
398         } else {
399             // Do not allow shift amount to be used as the result, MacroAssembler does not permit this.
400             SpeculateIntegerOperand op1(this, node.child1);
401             SpeculateIntegerOperand op2(this, node.child2);
402             GPRTemporary result(this, op1);
403
404             GPRReg reg1 = op1.gpr();
405             GPRReg reg2 = op2.gpr();
406             shiftOp(op, reg1, reg2, result.gpr());
407
408             integerResult(result.gpr(), m_compileIndex);
409         }
410         break;
411
412     case UInt32ToNumber: {
413         IntegerOperand op1(this, node.child1);
414         GPRTemporary result(this, op1);
415
416         // Test the operand is positive.
417         speculationCheck(m_jit.branch32(MacroAssembler::LessThan, op1.gpr(), TrustedImm32(0)));
418
419         m_jit.move(op1.gpr(), result.gpr());
420         integerResult(result.gpr(), m_compileIndex, op1.format());
421         break;
422     }
423
424     case ValueToInt32: {
425         SpeculateIntegerOperand op1(this, node.child1);
426         GPRTemporary result(this, op1);
427         m_jit.move(op1.gpr(), result.gpr());
428         integerResult(result.gpr(), m_compileIndex, op1.format());
429         break;
430     }
431
432     case ValueToNumber: {
433         SpeculateIntegerOperand op1(this, node.child1);
434         GPRTemporary result(this, op1);
435         m_jit.move(op1.gpr(), result.gpr());
436         integerResult(result.gpr(), m_compileIndex, op1.format());
437         break;
438     }
439
440     case ValueAdd:
441     case ArithAdd: {
442         if (isInt32Constant(node.child1)) {
443             int32_t imm1 = valueOfInt32Constant(node.child1);
444             SpeculateIntegerOperand op2(this, node.child2);
445             GPRTemporary result(this);
446
447             speculationCheck(m_jit.branchAdd32(MacroAssembler::Overflow, op2.gpr(), Imm32(imm1), result.gpr()));
448
449             integerResult(result.gpr(), m_compileIndex);
450             break;
451         }
452             
453         if (isInt32Constant(node.child2)) {
454             SpeculateIntegerOperand op1(this, node.child1);
455             int32_t imm2 = valueOfInt32Constant(node.child2);
456             GPRTemporary result(this);
457
458             speculationCheck(m_jit.branchAdd32(MacroAssembler::Overflow, op1.gpr(), Imm32(imm2), result.gpr()));
459
460             integerResult(result.gpr(), m_compileIndex);
461             break;
462         }
463             
464         SpeculateIntegerOperand op1(this, node.child1);
465         SpeculateIntegerOperand op2(this, node.child2);
466         GPRTemporary result(this, op1, op2);
467
468         GPRReg gpr1 = op1.gpr();
469         GPRReg gpr2 = op2.gpr();
470         GPRReg gprResult = result.gpr();
471         MacroAssembler::Jump check = m_jit.branchAdd32(MacroAssembler::Overflow, gpr1, gpr2, gprResult);
472
473         if (gpr1 == gprResult)
474             speculationCheck(check, SpeculationRecovery(SpeculativeAdd, gprResult, gpr2));
475         else if (gpr2 == gprResult)
476             speculationCheck(check, SpeculationRecovery(SpeculativeAdd, gprResult, gpr1));
477         else
478             speculationCheck(check);
479
480         integerResult(gprResult, m_compileIndex);
481         break;
482     }
483
484     case ArithSub: {
485         if (isInt32Constant(node.child2)) {
486             SpeculateIntegerOperand op1(this, node.child1);
487             int32_t imm2 = valueOfInt32Constant(node.child2);
488             GPRTemporary result(this);
489
490             speculationCheck(m_jit.branchSub32(MacroAssembler::Overflow, op1.gpr(), Imm32(imm2), result.gpr()));
491
492             integerResult(result.gpr(), m_compileIndex);
493             break;
494         }
495             
496         SpeculateIntegerOperand op1(this, node.child1);
497         SpeculateIntegerOperand op2(this, node.child2);
498         GPRTemporary result(this);
499
500         speculationCheck(m_jit.branchSub32(MacroAssembler::Overflow, op1.gpr(), op2.gpr(), result.gpr()));
501
502         integerResult(result.gpr(), m_compileIndex);
503         break;
504     }
505
506     case ArithMul: {
507         SpeculateIntegerOperand op1(this, node.child1);
508         SpeculateIntegerOperand op2(this, node.child2);
509         GPRTemporary result(this);
510
511         GPRReg reg1 = op1.gpr();
512         GPRReg reg2 = op2.gpr();
513         speculationCheck(m_jit.branchMul32(MacroAssembler::Overflow, reg1, reg2, result.gpr()));
514
515         MacroAssembler::Jump resultNonZero = m_jit.branchTest32(MacroAssembler::NonZero, result.gpr());
516         speculationCheck(m_jit.branch32(MacroAssembler::LessThan, reg1, TrustedImm32(0)));
517         speculationCheck(m_jit.branch32(MacroAssembler::LessThan, reg2, TrustedImm32(0)));
518         resultNonZero.link(&m_jit);
519
520         integerResult(result.gpr(), m_compileIndex);
521         break;
522     }
523
524     case ArithDiv: {
525         SpeculateIntegerOperand op1(this, node.child1);
526         SpeculateIntegerOperand op2(this, node.child2);
527         GPRTemporary result(this, op1, op2);
528
529         op1.gpr();
530         op2.gpr();
531         terminateSpeculativeExecution();
532
533         integerResult(result.gpr(), m_compileIndex);
534         break;
535     }
536
537     case ArithMod: {
538         SpeculateIntegerOperand op1(this, node.child1);
539         SpeculateIntegerOperand op2(this, node.child2);
540         GPRTemporary result(this, op1, op2);
541
542         op1.gpr();
543         op2.gpr();
544         terminateSpeculativeExecution();
545
546         integerResult(result.gpr(), m_compileIndex);
547         break;
548     }
549
550     case LogicalNot: {
551         JSValueOperand value(this, node.child1);
552         GPRTemporary result(this); // FIXME: We could reuse, but on speculation fail would need recovery to restore tag (akin to add).
553
554         m_jit.move(value.gpr(), result.gpr());
555         m_jit.xorPtr(TrustedImm32(static_cast<int32_t>(ValueFalse)), result.gpr());
556         speculationCheck(m_jit.branchTestPtr(JITCompiler::NonZero, result.gpr(), TrustedImm32(static_cast<int32_t>(~1))));
557         m_jit.xorPtr(TrustedImm32(static_cast<int32_t>(ValueTrue)), result.gpr());
558
559         // If we add a DataFormatBool, we should use it here.
560         jsValueResult(result.gpr(), m_compileIndex);
561         break;
562     }
563
564     case CompareLess: {
565         // Fused compare & branch.
566         NodeIndex branchNodeIndex = detectPeepHoleBranch();
567         if (branchNodeIndex != NoNode) {
568             // detectPeepHoleBranch currently only permits the branch to be the very next node,
569             // so can be no intervening nodes to also reference the compare. 
570             ASSERT(node.adjustedRefCount() == 1);
571
572             if (compareIsInteger(node.child1, node.child2))
573                 compilePeepHoleIntegerBranch(node, branchNodeIndex, JITCompiler::LessThan);
574             else
575                 compilePeepHoleCall(node, branchNodeIndex, operationCompareLess);
576
577             use(node.child1);
578             use(node.child2);
579             m_compileIndex = branchNodeIndex;
580             return;
581         }
582
583         // Normal case, not fused to branch.
584         SpeculateIntegerOperand op1(this, node.child1);
585         SpeculateIntegerOperand op2(this, node.child2);
586         GPRTemporary result(this, op1, op2);
587
588         m_jit.compare32(JITCompiler::LessThan, op1.gpr(), op2.gpr(), result.gpr());
589
590         // If we add a DataFormatBool, we should use it here.
591         m_jit.or32(TrustedImm32(ValueFalse), result.gpr());
592         jsValueResult(result.gpr(), m_compileIndex);
593         break;
594     }
595
596     case CompareLessEq: {
597         // Fused compare & branch.
598         NodeIndex branchNodeIndex = detectPeepHoleBranch();
599         if (branchNodeIndex != NoNode) {
600             // detectPeepHoleBranch currently only permits the branch to be the very next node,
601             // so can be no intervening nodes to also reference the compare. 
602             ASSERT(node.adjustedRefCount() == 1);
603
604             if (compareIsInteger(node.child1, node.child2))
605                 compilePeepHoleIntegerBranch(node, branchNodeIndex, JITCompiler::LessThanOrEqual);
606             else
607                 compilePeepHoleCall(node, branchNodeIndex, operationCompareLessEq);
608
609             use(node.child1);
610             use(node.child2);
611             m_compileIndex = branchNodeIndex;
612             return;
613         }
614
615         // Normal case, not fused to branch.
616         SpeculateIntegerOperand op1(this, node.child1);
617         SpeculateIntegerOperand op2(this, node.child2);
618         GPRTemporary result(this, op1, op2);
619
620         m_jit.compare32(JITCompiler::LessThanOrEqual, op1.gpr(), op2.gpr(), result.gpr());
621
622         // If we add a DataFormatBool, we should use it here.
623         m_jit.or32(TrustedImm32(ValueFalse), result.gpr());
624         jsValueResult(result.gpr(), m_compileIndex);
625         break;
626     }
627
628     case CompareEq: {
629         // Fused compare & branch.
630         NodeIndex branchNodeIndex = detectPeepHoleBranch();
631         if (branchNodeIndex != NoNode) {
632             // detectPeepHoleBranch currently only permits the branch to be the very next node,
633             // so can be no intervening nodes to also reference the compare. 
634             ASSERT(node.adjustedRefCount() == 1);
635
636             if (compareIsInteger(node.child1, node.child2))
637                 compilePeepHoleIntegerBranch(node, branchNodeIndex, JITCompiler::Equal);
638             else
639                 compilePeepHoleCall(node, branchNodeIndex, operationCompareEq);
640
641             use(node.child1);
642             use(node.child2);
643             m_compileIndex = branchNodeIndex;
644             return;
645         }
646
647         SpeculateIntegerOperand op1(this, node.child1);
648         SpeculateIntegerOperand op2(this, node.child2);
649         GPRTemporary result(this, op1, op2);
650
651         m_jit.compare32(JITCompiler::Equal, op1.gpr(), op2.gpr(), result.gpr());
652
653         // If we add a DataFormatBool, we should use it here.
654         m_jit.or32(TrustedImm32(ValueFalse), result.gpr());
655         jsValueResult(result.gpr(), m_compileIndex);
656         break;
657     }
658
659     case CompareStrictEq: {
660         SpeculateIntegerOperand op1(this, node.child1);
661         SpeculateIntegerOperand op2(this, node.child2);
662         GPRTemporary result(this, op1, op2);
663
664         m_jit.compare32(JITCompiler::Equal, op1.gpr(), op2.gpr(), result.gpr());
665
666         // If we add a DataFormatBool, we should use it here.
667         m_jit.or32(TrustedImm32(ValueFalse), result.gpr());
668         jsValueResult(result.gpr(), m_compileIndex);
669         break;
670     }
671
672     case GetByVal: {
673         NodeIndex alias = node.child3;
674         if (alias != NoNode) {
675             // FIXME: result should be able to reuse child1, child2. Should have an 'UnusedOperand' type.
676             JSValueOperand aliasedValue(this, node.child3);
677             GPRTemporary result(this, aliasedValue);
678             m_jit.move(aliasedValue.gpr(), result.gpr());
679             jsValueResult(result.gpr(), m_compileIndex);
680             break;
681         }
682
683         SpeculateCellOperand base(this, node.child1);
684         SpeculateStrictInt32Operand property(this, node.child2);
685         GPRTemporary storage(this);
686
687         GPRReg baseReg = base.gpr();
688         GPRReg propertyReg = property.gpr();
689         GPRReg storageReg = storage.gpr();
690
691         // Get the array storage. We haven't yet checked this is a JSArray, so this is only safe if
692         // an access with offset JSArray::storageOffset() is valid for all JSCells!
693         m_jit.loadPtr(MacroAssembler::Address(baseReg, JSArray::storageOffset()), storageReg);
694
695         // Check that base is an array, and that property is contained within m_vector (< m_vectorLength).
696         // If we have predicted the base to be type array, we can skip the check.
697         Node& baseNode = m_jit.graph()[node.child1];
698         if (baseNode.op != GetLocal || m_jit.graph().getPrediction(baseNode.local()) != PredictArray)
699             speculationCheck(m_jit.branchPtr(MacroAssembler::NotEqual, MacroAssembler::Address(baseReg), MacroAssembler::TrustedImmPtr(m_jit.globalData()->jsArrayVPtr)));
700         speculationCheck(m_jit.branch32(MacroAssembler::AboveOrEqual, propertyReg, MacroAssembler::Address(baseReg, JSArray::vectorLengthOffset())));
701
702         // FIXME: In cases where there are subsequent by_val accesses to the same base it might help to cache
703         // the storage pointer - especially if there happens to be another register free right now. If we do so,
704         // then we'll need to allocate a new temporary for result.
705         GPRTemporary& result = storage;
706         m_jit.loadPtr(MacroAssembler::BaseIndex(storageReg, propertyReg, MacroAssembler::ScalePtr, OBJECT_OFFSETOF(ArrayStorage, m_vector[0])), result.gpr());
707         speculationCheck(m_jit.branchTestPtr(MacroAssembler::Zero, result.gpr()));
708
709         jsValueResult(result.gpr(), m_compileIndex);
710         break;
711     }
712
713     case PutByVal: {
714         SpeculateCellOperand base(this, node.child1);
715         SpeculateStrictInt32Operand property(this, node.child2);
716         JSValueOperand value(this, node.child3);
717         GPRTemporary storage(this);
718
719         // Map base, property & value into registers, allocate a register for storage.
720         GPRReg baseReg = base.gpr();
721         GPRReg propertyReg = property.gpr();
722         GPRReg valueReg = value.gpr();
723         GPRReg storageReg = storage.gpr();
724
725         // Check that base is an array, and that property is contained within m_vector (< m_vectorLength).
726         // If we have predicted the base to be type array, we can skip the check.
727         Node& baseNode = m_jit.graph()[node.child1];
728         if (baseNode.op != GetLocal || m_jit.graph().getPrediction(baseNode.local()) != PredictArray)
729             speculationCheck(m_jit.branchPtr(MacroAssembler::NotEqual, MacroAssembler::Address(baseReg), MacroAssembler::TrustedImmPtr(m_jit.globalData()->jsArrayVPtr)));
730         MacroAssembler::Jump withinArrayBounds = m_jit.branch32(MacroAssembler::Below, propertyReg, MacroAssembler::Address(baseReg, JSArray::vectorLengthOffset()));
731
732         // Code to handle put beyond array bounds.
733         silentSpillAllRegisters(storageReg, baseReg, propertyReg, valueReg);
734         setupStubArguments(baseReg, propertyReg, valueReg);
735         m_jit.move(GPRInfo::callFrameRegister, GPRInfo::argumentGPR0);
736         JITCompiler::Call functionCall = appendCallWithExceptionCheck(operationPutByValBeyondArrayBounds);
737         silentFillAllRegisters(storageReg);
738         JITCompiler::Jump wasBeyondArrayBounds = m_jit.jump();
739
740         withinArrayBounds.link(&m_jit);
741
742         // Get the array storage.
743         m_jit.loadPtr(MacroAssembler::Address(baseReg, JSArray::storageOffset()), storageReg);
744
745         // Check if we're writing to a hole; if so increment m_numValuesInVector.
746         MacroAssembler::Jump notHoleValue = m_jit.branchTestPtr(MacroAssembler::NonZero, MacroAssembler::BaseIndex(storageReg, propertyReg, MacroAssembler::ScalePtr, OBJECT_OFFSETOF(ArrayStorage, m_vector[0])));
747         m_jit.add32(TrustedImm32(1), MacroAssembler::Address(storageReg, OBJECT_OFFSETOF(ArrayStorage, m_numValuesInVector)));
748
749         // If we're writing to a hole we might be growing the array; 
750         MacroAssembler::Jump lengthDoesNotNeedUpdate = m_jit.branch32(MacroAssembler::Below, propertyReg, MacroAssembler::Address(storageReg, OBJECT_OFFSETOF(ArrayStorage, m_length)));
751         m_jit.add32(TrustedImm32(1), propertyReg);
752         m_jit.store32(propertyReg, MacroAssembler::Address(storageReg, OBJECT_OFFSETOF(ArrayStorage, m_length)));
753         m_jit.sub32(TrustedImm32(1), propertyReg);
754
755         lengthDoesNotNeedUpdate.link(&m_jit);
756         notHoleValue.link(&m_jit);
757
758         // Store the value to the array.
759         m_jit.storePtr(valueReg, MacroAssembler::BaseIndex(storageReg, propertyReg, MacroAssembler::ScalePtr, OBJECT_OFFSETOF(ArrayStorage, m_vector[0])));
760
761         wasBeyondArrayBounds.link(&m_jit);
762
763         noResult(m_compileIndex);
764         break;
765     }
766
767     case PutByValAlias: {
768         SpeculateCellOperand base(this, node.child1);
769         SpeculateStrictInt32Operand property(this, node.child2);
770         JSValueOperand value(this, node.child3);
771         GPRTemporary storage(this, base); // storage may overwrite base.
772
773         // Get the array storage.
774         GPRReg storageReg = storage.gpr();
775         m_jit.loadPtr(MacroAssembler::Address(base.gpr(), JSArray::storageOffset()), storageReg);
776
777         // Map property & value into registers.
778         GPRReg propertyReg = property.gpr();
779         GPRReg valueReg = value.gpr();
780
781         // Store the value to the array.
782         m_jit.storePtr(valueReg, MacroAssembler::BaseIndex(storageReg, propertyReg, MacroAssembler::ScalePtr, OBJECT_OFFSETOF(ArrayStorage, m_vector[0])));
783
784         noResult(m_compileIndex);
785         break;
786     }
787
788     case DFG::Jump: {
789         BlockIndex taken = m_jit.graph().blockIndexForBytecodeOffset(node.takenBytecodeOffset());
790         if (taken != (m_block + 1))
791             addBranch(m_jit.jump(), taken);
792         noResult(m_compileIndex);
793         break;
794     }
795
796     case Branch: {
797         JSValueOperand value(this, node.child1);
798         GPRReg valueReg = value.gpr();
799
800         BlockIndex taken = m_jit.graph().blockIndexForBytecodeOffset(node.takenBytecodeOffset());
801         BlockIndex notTaken = m_jit.graph().blockIndexForBytecodeOffset(node.notTakenBytecodeOffset());
802
803         // Integers
804         addBranch(m_jit.branchPtr(MacroAssembler::Equal, valueReg, MacroAssembler::ImmPtr(JSValue::encode(jsNumber(0)))), notTaken);
805         MacroAssembler::Jump isNonZeroInteger = m_jit.branchPtr(MacroAssembler::AboveOrEqual, valueReg, GPRInfo::tagTypeNumberRegister);
806
807         // Booleans
808         addBranch(m_jit.branchPtr(MacroAssembler::Equal, valueReg, MacroAssembler::ImmPtr(JSValue::encode(jsBoolean(false)))), notTaken);
809         speculationCheck(m_jit.branchPtr(MacroAssembler::NotEqual, valueReg, MacroAssembler::ImmPtr(JSValue::encode(jsBoolean(true)))));
810
811         if (taken == (m_block + 1))
812             isNonZeroInteger.link(&m_jit);
813         else {
814             addBranch(isNonZeroInteger, taken);
815             addBranch(m_jit.jump(), taken);
816         }
817
818         noResult(m_compileIndex);
819         break;
820     }
821
822     case Return: {
823         ASSERT(GPRInfo::callFrameRegister != GPRInfo::regT1);
824         ASSERT(GPRInfo::regT1 != GPRInfo::returnValueGPR);
825         ASSERT(GPRInfo::returnValueGPR != GPRInfo::callFrameRegister);
826
827 #if DFG_SUCCESS_STATS
828         static SamplingCounter counter("SpeculativeJIT");
829         m_jit.emitCount(counter);
830 #endif
831
832         // Return the result in returnValueGPR.
833         JSValueOperand op1(this, node.child1);
834         m_jit.move(op1.gpr(), GPRInfo::returnValueGPR);
835
836         // Grab the return address.
837         m_jit.emitGetFromCallFrameHeaderPtr(RegisterFile::ReturnPC, GPRInfo::regT1);
838         // Restore our caller's "r".
839         m_jit.emitGetFromCallFrameHeaderPtr(RegisterFile::CallerFrame, GPRInfo::callFrameRegister);
840         // Return.
841         m_jit.restoreReturnAddressBeforeReturn(GPRInfo::regT1);
842         m_jit.ret();
843         
844         noResult(m_compileIndex);
845         break;
846     }
847
848     case ConvertThis: {
849         SpeculateCellOperand thisValue(this, node.child1);
850         GPRTemporary temp(this);
851
852         m_jit.loadPtr(JITCompiler::Address(thisValue.gpr(), JSCell::structureOffset()), temp.gpr());
853         speculationCheck(m_jit.branchTest8(JITCompiler::NonZero, JITCompiler::Address(temp.gpr(), Structure::typeInfoFlagsOffset()), JITCompiler::TrustedImm32(NeedsThisConversion)));
854
855         cellResult(thisValue.gpr(), m_compileIndex);
856         break;
857     }
858
859     case GetById: {
860         SpeculateCellOperand base(this, node.child1);
861         GPRTemporary result(this, base);
862
863         GPRReg resultGPR = result.gpr();
864
865         cachedGetById(base.gpr(), resultGPR, node.identifierNumber());
866
867         jsValueResult(resultGPR, m_compileIndex);
868         break;
869     }
870
871     case PutById: {
872         SpeculateCellOperand base(this, node.child1);
873         JSValueOperand value(this, node.child2);
874         GPRTemporary scratch(this, base);
875
876         cachedPutById(base.gpr(), value.gpr(), scratch.gpr(), node.identifierNumber(), NotDirect);
877         
878         noResult(m_compileIndex);
879         break;
880     }
881
882     case PutByIdDirect: {
883         SpeculateCellOperand base(this, node.child1);
884         JSValueOperand value(this, node.child2);
885         GPRTemporary scratch(this, base);
886
887         cachedPutById(base.gpr(), value.gpr(), scratch.gpr(), node.identifierNumber(), Direct);
888
889         noResult(m_compileIndex);
890         break;
891     }
892
893     case GetGlobalVar: {
894         GPRTemporary result(this);
895
896         JSVariableObject* globalObject = m_jit.codeBlock()->globalObject();
897         m_jit.loadPtr(globalObject->addressOfRegisters(), result.gpr());
898         m_jit.loadPtr(JITCompiler::addressForGlobalVar(result.gpr(), node.varNumber()), result.gpr());
899
900         jsValueResult(result.gpr(), m_compileIndex);
901         break;
902     }
903
904     case PutGlobalVar: {
905         JSValueOperand value(this, node.child1);
906         GPRTemporary temp(this);
907
908         JSVariableObject* globalObject = m_jit.codeBlock()->globalObject();
909         m_jit.loadPtr(globalObject->addressOfRegisters(), temp.gpr());
910         m_jit.storePtr(value.gpr(), JITCompiler::addressForGlobalVar(temp.gpr(), node.varNumber()));
911
912         noResult(m_compileIndex);
913         break;
914     }
915
916     case Phi:
917         ASSERT_NOT_REACHED();
918
919     case Breakpoint:
920 #if ENABLE(DEBUG_WITH_BREAKPOINT)
921         m_jit.breakpoint();
922 #else
923         ASSERT_NOT_REACHED();
924 #endif
925     }
926
927     if (node.hasResult() && node.mustGenerate())
928         use(m_compileIndex);
929 }
930
931 void SpeculativeJIT::compile(BasicBlock& block)
932 {
933     ASSERT(m_compileIndex == block.begin);
934     m_blockHeads[m_block] = m_jit.label();
935 #if DFG_JIT_BREAK_ON_EVERY_BLOCK
936     m_jit.breakpoint();
937 #endif
938
939     for (; m_compileIndex < block.end; ++m_compileIndex) {
940         Node& node = m_jit.graph()[m_compileIndex];
941         if (!node.shouldGenerate())
942             continue;
943
944 #if DFG_DEBUG_VERBOSE
945         fprintf(stderr, "SpeculativeJIT generating Node @%d at JIT offset 0x%x\n", (int)m_compileIndex, m_jit.debugOffset());
946 #endif
947 #if DFG_JIT_BREAK_ON_EVERY_NODE
948         m_jit.breakpoint();
949 #endif
950         checkConsistency();
951         compile(node);
952         if (!m_compileOkay)
953             return;
954         checkConsistency();
955     }
956 }
957
958 // If we are making type predictions about our arguments then
959 // we need to check that they are correct on function entry.
960 void SpeculativeJIT::checkArgumentTypes()
961 {
962     ASSERT(!m_compileIndex);
963     for (int i = 0; i < m_jit.codeBlock()->m_numParameters; ++i) {
964         VirtualRegister virtualRegister = (VirtualRegister)(m_jit.codeBlock()->thisRegister() + i);
965         switch (m_jit.graph().getPrediction(virtualRegister)) {
966         case PredictInt32:
967             speculationCheck(m_jit.branchPtr(MacroAssembler::Below, JITCompiler::addressFor(virtualRegister), GPRInfo::tagTypeNumberRegister));
968             break;
969
970         case PredictArray: {
971             GPRTemporary temp(this);
972             m_jit.loadPtr(JITCompiler::addressFor(virtualRegister), temp.gpr());
973             speculationCheck(m_jit.branchTestPtr(MacroAssembler::NonZero, temp.gpr(), GPRInfo::tagMaskRegister));
974             speculationCheck(m_jit.branchPtr(MacroAssembler::NotEqual, MacroAssembler::Address(temp.gpr()), MacroAssembler::TrustedImmPtr(m_jit.globalData()->jsArrayVPtr)));
975             break;
976         }
977
978         default:
979             break;
980         }
981     }
982 }
983
984 // For any vars that we will be treating as numeric, write 0 to
985 // the var on entry. Throughout the block we will only read/write
986 // to the payload, by writing the tag now we prevent the GC from
987 // misinterpreting values as pointers.
988 void SpeculativeJIT::initializeVariableTypes()
989 {
990     ASSERT(!m_compileIndex);
991     for (int var = 0; var < m_jit.codeBlock()->m_numVars; ++var) {
992         if (m_jit.graph().getPrediction(var) == PredictInt32)
993             m_jit.storePtr(GPRInfo::tagTypeNumberRegister, JITCompiler::addressFor((VirtualRegister)var));
994     }
995 }
996
997 bool SpeculativeJIT::compile()
998 {
999     checkArgumentTypes();
1000     initializeVariableTypes();
1001
1002     ASSERT(!m_compileIndex);
1003     for (m_block = 0; m_block < m_jit.graph().m_blocks.size(); ++m_block) {
1004         compile(*m_jit.graph().m_blocks[m_block]);
1005         if (!m_compileOkay)
1006             return false;
1007     }
1008     linkBranches();
1009     return true;
1010 }
1011
1012 } } // namespace JSC::DFG
1013
1014 #endif