DFG should be able to set watchpoints on global variables
[WebKit-https.git] / Source / JavaScriptCore / jit / JIT.cpp
1 /*
2  * Copyright (C) 2008, 2009 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
28 #if ENABLE(JIT)
29 #include "JIT.h"
30
31 // This probably does not belong here; adding here for now as a quick Windows build fix.
32 #if ENABLE(ASSEMBLER) && CPU(X86) && !OS(MAC_OS_X)
33 #include "MacroAssembler.h"
34 JSC::MacroAssemblerX86Common::SSE2CheckState JSC::MacroAssemblerX86Common::s_sse2CheckState = NotCheckedSSE2;
35 #endif
36
37 #include "CodeBlock.h"
38 #include <wtf/CryptographicallyRandomNumber.h>
39 #include "DFGNode.h" // for DFG_SUCCESS_STATS
40 #include "Interpreter.h"
41 #include "JITInlineMethods.h"
42 #include "JITStubCall.h"
43 #include "JSArray.h"
44 #include "JSFunction.h"
45 #include "LinkBuffer.h"
46 #include "RepatchBuffer.h"
47 #include "ResultType.h"
48 #include "SamplingTool.h"
49
50 using namespace std;
51
52 namespace JSC {
53
54 void ctiPatchNearCallByReturnAddress(CodeBlock* codeblock, ReturnAddressPtr returnAddress, MacroAssemblerCodePtr newCalleeFunction)
55 {
56     RepatchBuffer repatchBuffer(codeblock);
57     repatchBuffer.relinkNearCallerToTrampoline(returnAddress, newCalleeFunction);
58 }
59
60 void ctiPatchCallByReturnAddress(CodeBlock* codeblock, ReturnAddressPtr returnAddress, MacroAssemblerCodePtr newCalleeFunction)
61 {
62     RepatchBuffer repatchBuffer(codeblock);
63     repatchBuffer.relinkCallerToTrampoline(returnAddress, newCalleeFunction);
64 }
65
66 void ctiPatchCallByReturnAddress(CodeBlock* codeblock, ReturnAddressPtr returnAddress, FunctionPtr newCalleeFunction)
67 {
68     RepatchBuffer repatchBuffer(codeblock);
69     repatchBuffer.relinkCallerToFunction(returnAddress, newCalleeFunction);
70 }
71
72 JIT::JIT(JSGlobalData* globalData, CodeBlock* codeBlock)
73     : m_interpreter(globalData->interpreter)
74     , m_globalData(globalData)
75     , m_codeBlock(codeBlock)
76     , m_labels(codeBlock ? codeBlock->numberOfInstructions() : 0)
77     , m_bytecodeOffset((unsigned)-1)
78 #if USE(JSVALUE32_64)
79     , m_jumpTargetIndex(0)
80     , m_mappedBytecodeOffset((unsigned)-1)
81     , m_mappedVirtualRegisterIndex(RegisterFile::ReturnPC)
82     , m_mappedTag((RegisterID)-1)
83     , m_mappedPayload((RegisterID)-1)
84 #else
85     , m_lastResultBytecodeRegister(std::numeric_limits<int>::max())
86     , m_jumpTargetsPosition(0)
87 #endif
88 #if USE(OS_RANDOMNESS)
89     , m_randomGenerator(cryptographicallyRandomNumber())
90 #else
91     , m_randomGenerator(static_cast<unsigned>(randomNumber() * 0xFFFFFFF))
92 #endif
93 {
94 }
95
96 #if ENABLE(DFG_JIT)
97 void JIT::emitOptimizationCheck(OptimizationCheckKind kind)
98 {
99     if (!canBeOptimized())
100         return;
101     
102     Jump skipOptimize = branchAdd32(Signed, TrustedImm32(kind == LoopOptimizationCheck ? Options::executionCounterIncrementForLoop : Options::executionCounterIncrementForReturn), AbsoluteAddress(m_codeBlock->addressOfJITExecuteCounter()));
103     JITStubCall stubCall(this, kind == LoopOptimizationCheck ? cti_optimize_from_loop : cti_optimize_from_ret);
104     if (kind == LoopOptimizationCheck)
105         stubCall.addArgument(TrustedImm32(m_bytecodeOffset));
106     stubCall.call();
107     skipOptimize.link(this);
108 }
109 #endif
110
111 #if CPU(X86)
112 void JIT::emitTimeoutCheck()
113 {
114     Jump skipTimeout = branchSub32(NonZero, TrustedImm32(1), AbsoluteAddress(&m_globalData->m_timeoutCount));
115     JITStubCall stubCall(this, cti_timeout_check);
116     stubCall.addArgument(regT1, regT0); // save last result registers.
117     stubCall.call(regT0);
118     store32(regT0, &m_globalData->m_timeoutCount);
119     stubCall.getArgument(0, regT1, regT0); // reload last result registers.
120     skipTimeout.link(this);
121 }
122 #elif USE(JSVALUE32_64)
123 void JIT::emitTimeoutCheck()
124 {
125     Jump skipTimeout = branchSub32(NonZero, TrustedImm32(1), timeoutCheckRegister);
126     JITStubCall stubCall(this, cti_timeout_check);
127     stubCall.addArgument(regT1, regT0); // save last result registers.
128     stubCall.call(timeoutCheckRegister);
129     stubCall.getArgument(0, regT1, regT0); // reload last result registers.
130     skipTimeout.link(this);
131 }
132 #else
133 void JIT::emitTimeoutCheck()
134 {
135     Jump skipTimeout = branchSub32(NonZero, TrustedImm32(1), timeoutCheckRegister);
136     JITStubCall(this, cti_timeout_check).call(timeoutCheckRegister);
137     skipTimeout.link(this);
138
139     killLastResultRegister();
140 }
141 #endif
142
143 #define NEXT_OPCODE(name) \
144     m_bytecodeOffset += OPCODE_LENGTH(name); \
145     break;
146
147 #if USE(JSVALUE32_64)
148 #define DEFINE_BINARY_OP(name) \
149     case name: { \
150         JITStubCall stubCall(this, cti_##name); \
151         stubCall.addArgument(currentInstruction[2].u.operand); \
152         stubCall.addArgument(currentInstruction[3].u.operand); \
153         stubCall.call(currentInstruction[1].u.operand); \
154         NEXT_OPCODE(name); \
155     }
156
157 #define DEFINE_UNARY_OP(name) \
158     case name: { \
159         JITStubCall stubCall(this, cti_##name); \
160         stubCall.addArgument(currentInstruction[2].u.operand); \
161         stubCall.call(currentInstruction[1].u.operand); \
162         NEXT_OPCODE(name); \
163     }
164
165 #else // USE(JSVALUE32_64)
166
167 #define DEFINE_BINARY_OP(name) \
168     case name: { \
169         JITStubCall stubCall(this, cti_##name); \
170         stubCall.addArgument(currentInstruction[2].u.operand, regT2); \
171         stubCall.addArgument(currentInstruction[3].u.operand, regT2); \
172         stubCall.call(currentInstruction[1].u.operand); \
173         NEXT_OPCODE(name); \
174     }
175
176 #define DEFINE_UNARY_OP(name) \
177     case name: { \
178         JITStubCall stubCall(this, cti_##name); \
179         stubCall.addArgument(currentInstruction[2].u.operand, regT2); \
180         stubCall.call(currentInstruction[1].u.operand); \
181         NEXT_OPCODE(name); \
182     }
183 #endif // USE(JSVALUE32_64)
184
185 #define DEFINE_OP(name) \
186     case name: { \
187         emit_##name(currentInstruction); \
188         NEXT_OPCODE(name); \
189     }
190
191 #define DEFINE_SLOWCASE_OP(name) \
192     case name: { \
193         emitSlow_##name(currentInstruction, iter); \
194         NEXT_OPCODE(name); \
195     }
196
197 void JIT::privateCompileMainPass()
198 {
199     Instruction* instructionsBegin = m_codeBlock->instructions().begin();
200     unsigned instructionCount = m_codeBlock->instructions().size();
201
202     m_globalResolveInfoIndex = 0;
203     m_callLinkInfoIndex = 0;
204
205     for (m_bytecodeOffset = 0; m_bytecodeOffset < instructionCount; ) {
206         Instruction* currentInstruction = instructionsBegin + m_bytecodeOffset;
207         ASSERT_WITH_MESSAGE(m_interpreter->isOpcode(currentInstruction->u.opcode), "privateCompileMainPass gone bad @ %d", m_bytecodeOffset);
208
209 #if ENABLE(OPCODE_SAMPLING)
210         if (m_bytecodeOffset > 0) // Avoid the overhead of sampling op_enter twice.
211             sampleInstruction(currentInstruction);
212 #endif
213
214 #if USE(JSVALUE64)
215         if (atJumpTarget())
216             killLastResultRegister();
217 #endif
218
219         m_labels[m_bytecodeOffset] = label();
220
221 #if ENABLE(JIT_VERBOSE)
222         dataLog("Old JIT emitting code for bc#%u at offset 0x%lx.\n", m_bytecodeOffset, (long)debugOffset());
223 #endif
224
225         switch (m_interpreter->getOpcodeID(currentInstruction->u.opcode)) {
226         DEFINE_BINARY_OP(op_del_by_val)
227         DEFINE_BINARY_OP(op_in)
228         DEFINE_BINARY_OP(op_less)
229         DEFINE_BINARY_OP(op_lesseq)
230         DEFINE_BINARY_OP(op_greater)
231         DEFINE_BINARY_OP(op_greatereq)
232         DEFINE_UNARY_OP(op_is_function)
233         DEFINE_UNARY_OP(op_is_object)
234         DEFINE_UNARY_OP(op_typeof)
235
236         DEFINE_OP(op_add)
237         DEFINE_OP(op_bitand)
238         DEFINE_OP(op_bitor)
239         DEFINE_OP(op_bitxor)
240         DEFINE_OP(op_call)
241         DEFINE_OP(op_call_eval)
242         DEFINE_OP(op_call_varargs)
243         DEFINE_OP(op_catch)
244         DEFINE_OP(op_construct)
245         DEFINE_OP(op_create_this)
246         DEFINE_OP(op_convert_this)
247         DEFINE_OP(op_init_lazy_reg)
248         DEFINE_OP(op_create_arguments)
249         DEFINE_OP(op_debug)
250         DEFINE_OP(op_del_by_id)
251         DEFINE_OP(op_div)
252         DEFINE_OP(op_end)
253         DEFINE_OP(op_enter)
254         DEFINE_OP(op_create_activation)
255         DEFINE_OP(op_eq)
256         DEFINE_OP(op_eq_null)
257         DEFINE_OP(op_get_by_id)
258         DEFINE_OP(op_get_arguments_length)
259         DEFINE_OP(op_get_by_val)
260         DEFINE_OP(op_get_argument_by_val)
261         DEFINE_OP(op_get_by_pname)
262         DEFINE_OP(op_get_global_var_watchable)
263         DEFINE_OP(op_get_global_var)
264         DEFINE_OP(op_get_pnames)
265         DEFINE_OP(op_get_scoped_var)
266         DEFINE_OP(op_check_has_instance)
267         DEFINE_OP(op_instanceof)
268         DEFINE_OP(op_is_undefined)
269         DEFINE_OP(op_is_boolean)
270         DEFINE_OP(op_is_number)
271         DEFINE_OP(op_is_string)
272         DEFINE_OP(op_jeq_null)
273         DEFINE_OP(op_jfalse)
274         DEFINE_OP(op_jmp)
275         DEFINE_OP(op_jmp_scopes)
276         DEFINE_OP(op_jneq_null)
277         DEFINE_OP(op_jneq_ptr)
278         DEFINE_OP(op_jless)
279         DEFINE_OP(op_jlesseq)
280         DEFINE_OP(op_jgreater)
281         DEFINE_OP(op_jgreatereq)
282         DEFINE_OP(op_jnless)
283         DEFINE_OP(op_jnlesseq)
284         DEFINE_OP(op_jngreater)
285         DEFINE_OP(op_jngreatereq)
286         DEFINE_OP(op_jtrue)
287         DEFINE_OP(op_loop)
288         DEFINE_OP(op_loop_hint)
289         DEFINE_OP(op_loop_if_less)
290         DEFINE_OP(op_loop_if_lesseq)
291         DEFINE_OP(op_loop_if_greater)
292         DEFINE_OP(op_loop_if_greatereq)
293         DEFINE_OP(op_loop_if_true)
294         DEFINE_OP(op_loop_if_false)
295         DEFINE_OP(op_lshift)
296         DEFINE_OP(op_method_check)
297         DEFINE_OP(op_mod)
298         DEFINE_OP(op_mov)
299         DEFINE_OP(op_mul)
300         DEFINE_OP(op_negate)
301         DEFINE_OP(op_neq)
302         DEFINE_OP(op_neq_null)
303         DEFINE_OP(op_new_array)
304         DEFINE_OP(op_new_array_buffer)
305         DEFINE_OP(op_new_func)
306         DEFINE_OP(op_new_func_exp)
307         DEFINE_OP(op_new_object)
308         DEFINE_OP(op_new_regexp)
309         DEFINE_OP(op_next_pname)
310         DEFINE_OP(op_not)
311         DEFINE_OP(op_nstricteq)
312         DEFINE_OP(op_pop_scope)
313         DEFINE_OP(op_post_dec)
314         DEFINE_OP(op_post_inc)
315         DEFINE_OP(op_pre_dec)
316         DEFINE_OP(op_pre_inc)
317         DEFINE_OP(op_profile_did_call)
318         DEFINE_OP(op_profile_will_call)
319         DEFINE_OP(op_push_new_scope)
320         DEFINE_OP(op_push_scope)
321         case op_put_by_id_transition_direct:
322         case op_put_by_id_transition_normal:
323         DEFINE_OP(op_put_by_id)
324         DEFINE_OP(op_put_by_index)
325         DEFINE_OP(op_put_by_val)
326         DEFINE_OP(op_put_getter_setter)
327         DEFINE_OP(op_put_global_var)
328         DEFINE_OP(op_put_global_var_check)
329         DEFINE_OP(op_put_scoped_var)
330         DEFINE_OP(op_resolve)
331         DEFINE_OP(op_resolve_base)
332         DEFINE_OP(op_ensure_property_exists)
333         DEFINE_OP(op_resolve_global)
334         DEFINE_OP(op_resolve_global_dynamic)
335         DEFINE_OP(op_resolve_skip)
336         DEFINE_OP(op_resolve_with_base)
337         DEFINE_OP(op_resolve_with_this)
338         DEFINE_OP(op_ret)
339         DEFINE_OP(op_call_put_result)
340         DEFINE_OP(op_ret_object_or_this)
341         DEFINE_OP(op_rshift)
342         DEFINE_OP(op_urshift)
343         DEFINE_OP(op_strcat)
344         DEFINE_OP(op_stricteq)
345         DEFINE_OP(op_sub)
346         DEFINE_OP(op_switch_char)
347         DEFINE_OP(op_switch_imm)
348         DEFINE_OP(op_switch_string)
349         DEFINE_OP(op_tear_off_activation)
350         DEFINE_OP(op_tear_off_arguments)
351         DEFINE_OP(op_throw)
352         DEFINE_OP(op_throw_reference_error)
353         DEFINE_OP(op_to_jsnumber)
354         DEFINE_OP(op_to_primitive)
355
356         case op_get_array_length:
357         case op_get_by_id_chain:
358         case op_get_by_id_generic:
359         case op_get_by_id_proto:
360         case op_get_by_id_self:
361         case op_get_by_id_getter_chain:
362         case op_get_by_id_getter_proto:
363         case op_get_by_id_getter_self:
364         case op_get_by_id_custom_chain:
365         case op_get_by_id_custom_proto:
366         case op_get_by_id_custom_self:
367         case op_get_string_length:
368         case op_put_by_id_generic:
369         case op_put_by_id_replace:
370         case op_put_by_id_transition:
371             ASSERT_NOT_REACHED();
372         }
373     }
374
375     ASSERT(m_callLinkInfoIndex == m_callStructureStubCompilationInfo.size());
376
377 #ifndef NDEBUG
378     // Reset this, in order to guard its use with ASSERTs.
379     m_bytecodeOffset = (unsigned)-1;
380 #endif
381 }
382
383 void JIT::privateCompileLinkPass()
384 {
385     unsigned jmpTableCount = m_jmpTable.size();
386     for (unsigned i = 0; i < jmpTableCount; ++i)
387         m_jmpTable[i].from.linkTo(m_labels[m_jmpTable[i].toBytecodeOffset], this);
388     m_jmpTable.clear();
389 }
390
391 void JIT::privateCompileSlowCases()
392 {
393     Instruction* instructionsBegin = m_codeBlock->instructions().begin();
394
395     m_propertyAccessInstructionIndex = 0;
396     m_globalResolveInfoIndex = 0;
397     m_callLinkInfoIndex = 0;
398     
399 #if !ASSERT_DISABLED && ENABLE(VALUE_PROFILER)
400     // Use this to assert that slow-path code associates new profiling sites with existing
401     // ValueProfiles rather than creating new ones. This ensures that for a given instruction
402     // (say, get_by_id) we get combined statistics for both the fast-path executions of that
403     // instructions and the slow-path executions. Furthermore, if the slow-path code created
404     // new ValueProfiles then the ValueProfiles would no longer be sorted by bytecode offset,
405     // which would break the invariant necessary to use CodeBlock::valueProfileForBytecodeOffset().
406     unsigned numberOfValueProfiles = m_codeBlock->numberOfValueProfiles();
407 #endif
408
409     for (Vector<SlowCaseEntry>::iterator iter = m_slowCases.begin(); iter != m_slowCases.end();) {
410 #if USE(JSVALUE64)
411         killLastResultRegister();
412 #endif
413
414         m_bytecodeOffset = iter->to;
415 #ifndef NDEBUG
416         unsigned firstTo = m_bytecodeOffset;
417 #endif
418         Instruction* currentInstruction = instructionsBegin + m_bytecodeOffset;
419         
420 #if ENABLE(VALUE_PROFILER)
421         RareCaseProfile* rareCaseProfile = 0;
422         if (shouldEmitProfiling())
423             rareCaseProfile = m_codeBlock->addRareCaseProfile(m_bytecodeOffset);
424 #endif
425
426 #if ENABLE(JIT_VERBOSE)
427         dataLog("Old JIT emitting slow code for bc#%u at offset 0x%lx.\n", m_bytecodeOffset, (long)debugOffset());
428 #endif
429
430         switch (m_interpreter->getOpcodeID(currentInstruction->u.opcode)) {
431         DEFINE_SLOWCASE_OP(op_add)
432         DEFINE_SLOWCASE_OP(op_bitand)
433         DEFINE_SLOWCASE_OP(op_bitor)
434         DEFINE_SLOWCASE_OP(op_bitxor)
435         DEFINE_SLOWCASE_OP(op_call)
436         DEFINE_SLOWCASE_OP(op_call_eval)
437         DEFINE_SLOWCASE_OP(op_call_varargs)
438         DEFINE_SLOWCASE_OP(op_construct)
439         DEFINE_SLOWCASE_OP(op_convert_this)
440         DEFINE_SLOWCASE_OP(op_create_this)
441         DEFINE_SLOWCASE_OP(op_div)
442         DEFINE_SLOWCASE_OP(op_eq)
443         DEFINE_SLOWCASE_OP(op_get_by_id)
444         DEFINE_SLOWCASE_OP(op_get_arguments_length)
445         DEFINE_SLOWCASE_OP(op_get_by_val)
446         DEFINE_SLOWCASE_OP(op_get_argument_by_val)
447         DEFINE_SLOWCASE_OP(op_get_by_pname)
448         DEFINE_SLOWCASE_OP(op_check_has_instance)
449         DEFINE_SLOWCASE_OP(op_instanceof)
450         DEFINE_SLOWCASE_OP(op_jfalse)
451         DEFINE_SLOWCASE_OP(op_jless)
452         DEFINE_SLOWCASE_OP(op_jlesseq)
453         DEFINE_SLOWCASE_OP(op_jgreater)
454         DEFINE_SLOWCASE_OP(op_jgreatereq)
455         DEFINE_SLOWCASE_OP(op_jnless)
456         DEFINE_SLOWCASE_OP(op_jnlesseq)
457         DEFINE_SLOWCASE_OP(op_jngreater)
458         DEFINE_SLOWCASE_OP(op_jngreatereq)
459         DEFINE_SLOWCASE_OP(op_jtrue)
460         DEFINE_SLOWCASE_OP(op_loop_if_less)
461         DEFINE_SLOWCASE_OP(op_loop_if_lesseq)
462         DEFINE_SLOWCASE_OP(op_loop_if_greater)
463         DEFINE_SLOWCASE_OP(op_loop_if_greatereq)
464         DEFINE_SLOWCASE_OP(op_loop_if_true)
465         DEFINE_SLOWCASE_OP(op_loop_if_false)
466         DEFINE_SLOWCASE_OP(op_lshift)
467         DEFINE_SLOWCASE_OP(op_method_check)
468         DEFINE_SLOWCASE_OP(op_mod)
469         DEFINE_SLOWCASE_OP(op_mul)
470         DEFINE_SLOWCASE_OP(op_negate)
471         DEFINE_SLOWCASE_OP(op_neq)
472         DEFINE_SLOWCASE_OP(op_new_array)
473         DEFINE_SLOWCASE_OP(op_new_object)
474         DEFINE_SLOWCASE_OP(op_new_func)
475         DEFINE_SLOWCASE_OP(op_new_func_exp)
476         DEFINE_SLOWCASE_OP(op_not)
477         DEFINE_SLOWCASE_OP(op_nstricteq)
478         DEFINE_SLOWCASE_OP(op_post_dec)
479         DEFINE_SLOWCASE_OP(op_post_inc)
480         DEFINE_SLOWCASE_OP(op_pre_dec)
481         DEFINE_SLOWCASE_OP(op_pre_inc)
482         case op_put_by_id_transition_direct:
483         case op_put_by_id_transition_normal:
484         DEFINE_SLOWCASE_OP(op_put_by_id)
485         DEFINE_SLOWCASE_OP(op_put_by_val)
486         DEFINE_SLOWCASE_OP(op_put_global_var_check);
487         DEFINE_SLOWCASE_OP(op_resolve_global)
488         DEFINE_SLOWCASE_OP(op_resolve_global_dynamic)
489         DEFINE_SLOWCASE_OP(op_rshift)
490         DEFINE_SLOWCASE_OP(op_urshift)
491         DEFINE_SLOWCASE_OP(op_stricteq)
492         DEFINE_SLOWCASE_OP(op_sub)
493         DEFINE_SLOWCASE_OP(op_to_jsnumber)
494         DEFINE_SLOWCASE_OP(op_to_primitive)
495         default:
496             ASSERT_NOT_REACHED();
497         }
498
499         ASSERT_WITH_MESSAGE(iter == m_slowCases.end() || firstTo != iter->to,"Not enough jumps linked in slow case codegen.");
500         ASSERT_WITH_MESSAGE(firstTo == (iter - 1)->to, "Too many jumps linked in slow case codegen.");
501         
502 #if ENABLE(VALUE_PROFILER)
503         if (shouldEmitProfiling())
504             add32(TrustedImm32(1), AbsoluteAddress(&rareCaseProfile->m_counter));
505 #endif
506
507         emitJumpSlowToHot(jump(), 0);
508     }
509
510     ASSERT(m_propertyAccessInstructionIndex == m_propertyAccessCompilationInfo.size());
511     ASSERT(m_callLinkInfoIndex == m_callStructureStubCompilationInfo.size());
512 #if ENABLE(VALUE_PROFILER)
513     ASSERT(numberOfValueProfiles == m_codeBlock->numberOfValueProfiles());
514 #endif
515
516 #ifndef NDEBUG
517     // Reset this, in order to guard its use with ASSERTs.
518     m_bytecodeOffset = (unsigned)-1;
519 #endif
520 }
521
522 ALWAYS_INLINE void PropertyStubCompilationInfo::copyToStubInfo(StructureStubInfo& info, LinkBuffer &linkBuffer)
523 {
524     ASSERT(bytecodeIndex != std::numeric_limits<unsigned>::max());
525     info.bytecodeIndex = bytecodeIndex;
526     info.callReturnLocation = linkBuffer.locationOf(callReturnLocation);
527     info.hotPathBegin = linkBuffer.locationOf(hotPathBegin);
528
529     switch (m_type) {
530     case MethodCheck: {
531         CodeLocationDataLabelPtr structureToCompareLocation = linkBuffer.locationOf(methodCheckStructureToCompare);
532         info.patch.baseline.methodCheckProtoObj = MacroAssembler::differenceBetweenCodePtr(structureToCompareLocation, linkBuffer.locationOf(methodCheckProtoObj));
533         info.patch.baseline.methodCheckProtoStructureToCompare = MacroAssembler::differenceBetweenCodePtr(structureToCompareLocation, linkBuffer.locationOf(methodCheckProtoStructureToCompare));
534         info.patch.baseline.methodCheckPutFunction = MacroAssembler::differenceBetweenCodePtr(structureToCompareLocation, linkBuffer.locationOf(methodCheckPutFunction));
535         // No break - fall through to GetById.
536     }
537     case GetById: {
538         CodeLocationLabel hotPathBeginLocation = linkBuffer.locationOf(hotPathBegin);
539         info.patch.baseline.u.get.structureToCompare = MacroAssembler::differenceBetweenCodePtr(hotPathBeginLocation, linkBuffer.locationOf(getStructureToCompare));
540         info.patch.baseline.u.get.structureCheck = MacroAssembler::differenceBetweenCodePtr(hotPathBeginLocation, linkBuffer.locationOf(getStructureCheck));
541 #if USE(JSVALUE64)
542         info.patch.baseline.u.get.displacementLabel = MacroAssembler::differenceBetweenCodePtr(hotPathBeginLocation, linkBuffer.locationOf(getDisplacementLabel));
543 #else
544         info.patch.baseline.u.get.displacementLabel1 = MacroAssembler::differenceBetweenCodePtr(hotPathBeginLocation, linkBuffer.locationOf(getDisplacementLabel1));
545         info.patch.baseline.u.get.displacementLabel2 = MacroAssembler::differenceBetweenCodePtr(hotPathBeginLocation, linkBuffer.locationOf(getDisplacementLabel2));
546 #endif
547         info.patch.baseline.u.get.putResult = MacroAssembler::differenceBetweenCodePtr(hotPathBeginLocation, linkBuffer.locationOf(getPutResult));
548         info.patch.baseline.u.get.coldPathBegin = MacroAssembler::differenceBetweenCodePtr(linkBuffer.locationOf(getColdPathBegin), linkBuffer.locationOf(callReturnLocation));
549         break;
550     }
551     case PutById:
552         CodeLocationLabel hotPathBeginLocation = linkBuffer.locationOf(hotPathBegin);
553         info.patch.baseline.u.put.structureToCompare = MacroAssembler::differenceBetweenCodePtr(hotPathBeginLocation, linkBuffer.locationOf(putStructureToCompare));
554 #if USE(JSVALUE64)
555         info.patch.baseline.u.put.displacementLabel = MacroAssembler::differenceBetweenCodePtr(hotPathBeginLocation, linkBuffer.locationOf(putDisplacementLabel));
556 #else
557         info.patch.baseline.u.put.displacementLabel1 = MacroAssembler::differenceBetweenCodePtr(hotPathBeginLocation, linkBuffer.locationOf(putDisplacementLabel1));
558         info.patch.baseline.u.put.displacementLabel2 = MacroAssembler::differenceBetweenCodePtr(hotPathBeginLocation, linkBuffer.locationOf(putDisplacementLabel2));
559 #endif
560         break;
561     }
562 }
563
564 JITCode JIT::privateCompile(CodePtr* functionEntryArityCheck, JITCompilationEffort effort)
565 {
566 #if ENABLE(JIT_VERBOSE_OSR)
567     printf("Compiling JIT code!\n");
568 #endif
569     
570 #if ENABLE(VALUE_PROFILER)
571     DFG::CapabilityLevel level = m_codeBlock->canCompileWithDFG();
572     switch (level) {
573     case DFG::CannotCompile:
574         m_canBeOptimized = false;
575         m_shouldEmitProfiling = false;
576         break;
577     case DFG::ShouldProfile:
578         m_canBeOptimized = false;
579         m_shouldEmitProfiling = true;
580         break;
581     case DFG::CanCompile:
582         m_canBeOptimized = true;
583         m_shouldEmitProfiling = true;
584         break;
585     default:
586         ASSERT_NOT_REACHED();
587         break;
588     }
589 #endif
590
591     // Just add a little bit of randomness to the codegen
592     if (m_randomGenerator.getUint32() & 1)
593         nop();
594
595     preserveReturnAddressAfterCall(regT2);
596     emitPutToCallFrameHeader(regT2, RegisterFile::ReturnPC);
597     emitPutImmediateToCallFrameHeader(m_codeBlock, RegisterFile::CodeBlock);
598
599     Label beginLabel(this);
600
601     sampleCodeBlock(m_codeBlock);
602 #if ENABLE(OPCODE_SAMPLING)
603     sampleInstruction(m_codeBlock->instructions().begin());
604 #endif
605
606     Jump registerFileCheck;
607     if (m_codeBlock->codeType() == FunctionCode) {
608 #if ENABLE(DFG_JIT)
609 #if DFG_ENABLE(SUCCESS_STATS)
610         static SamplingCounter counter("orignalJIT");
611         emitCount(counter);
612 #endif
613 #endif
614
615 #if ENABLE(VALUE_PROFILER)
616         ASSERT(m_bytecodeOffset == (unsigned)-1);
617         if (shouldEmitProfiling()) {
618             for (int argument = 0; argument < m_codeBlock->numParameters(); ++argument) {
619                 // If this is a constructor, then we want to put in a dummy profiling site (to
620                 // keep things consistent) but we don't actually want to record the dummy value.
621                 if (m_codeBlock->m_isConstructor && !argument)
622                     continue;
623                 int offset = CallFrame::argumentOffsetIncludingThis(argument) * static_cast<int>(sizeof(Register));
624 #if USE(JSVALUE64)
625                 loadPtr(Address(callFrameRegister, offset), regT0);
626 #elif USE(JSVALUE32_64)
627                 load32(Address(callFrameRegister, offset + OBJECT_OFFSETOF(JSValue, u.asBits.payload)), regT0);
628                 load32(Address(callFrameRegister, offset + OBJECT_OFFSETOF(JSValue, u.asBits.tag)), regT1);
629 #endif
630                 emitValueProfilingSite(m_codeBlock->valueProfileForArgument(argument));
631             }
632         }
633 #endif
634
635         addPtr(TrustedImm32(m_codeBlock->m_numCalleeRegisters * sizeof(Register)), callFrameRegister, regT1);
636         registerFileCheck = branchPtr(Below, AbsoluteAddress(m_globalData->interpreter->registerFile().addressOfEnd()), regT1);
637     }
638
639     Label functionBody = label();
640     
641 #if ENABLE(VALUE_PROFILER)
642     if (canBeOptimized())
643         add32(TrustedImm32(1), AbsoluteAddress(&m_codeBlock->m_executionEntryCount));
644 #endif
645
646     privateCompileMainPass();
647     privateCompileLinkPass();
648     privateCompileSlowCases();
649
650     Label arityCheck;
651     if (m_codeBlock->codeType() == FunctionCode) {
652         registerFileCheck.link(this);
653         m_bytecodeOffset = 0;
654         JITStubCall(this, cti_register_file_check).call();
655 #ifndef NDEBUG
656         m_bytecodeOffset = (unsigned)-1; // Reset this, in order to guard its use with ASSERTs.
657 #endif
658         jump(functionBody);
659
660         arityCheck = label();
661         preserveReturnAddressAfterCall(regT2);
662         emitPutToCallFrameHeader(regT2, RegisterFile::ReturnPC);
663         emitPutImmediateToCallFrameHeader(m_codeBlock, RegisterFile::CodeBlock);
664
665         load32(payloadFor(RegisterFile::ArgumentCount), regT1);
666         branch32(AboveOrEqual, regT1, TrustedImm32(m_codeBlock->m_numParameters)).linkTo(beginLabel, this);
667
668         m_bytecodeOffset = 0;
669         JITStubCall(this, m_codeBlock->m_isConstructor ? cti_op_construct_arityCheck : cti_op_call_arityCheck).call(callFrameRegister);
670 #if !ASSERT_DISABLED
671         m_bytecodeOffset = (unsigned)-1; // Reset this, in order to guard its use with ASSERTs.
672 #endif
673
674         jump(beginLabel);
675     }
676
677     ASSERT(m_jmpTable.isEmpty());
678
679     LinkBuffer patchBuffer(*m_globalData, this, m_codeBlock, effort);
680     if (patchBuffer.didFailToAllocate())
681         return JITCode();
682
683     // Translate vPC offsets into addresses in JIT generated code, for switch tables.
684     for (unsigned i = 0; i < m_switches.size(); ++i) {
685         SwitchRecord record = m_switches[i];
686         unsigned bytecodeOffset = record.bytecodeOffset;
687
688         if (record.type != SwitchRecord::String) {
689             ASSERT(record.type == SwitchRecord::Immediate || record.type == SwitchRecord::Character); 
690             ASSERT(record.jumpTable.simpleJumpTable->branchOffsets.size() == record.jumpTable.simpleJumpTable->ctiOffsets.size());
691
692             record.jumpTable.simpleJumpTable->ctiDefault = patchBuffer.locationOf(m_labels[bytecodeOffset + record.defaultOffset]);
693
694             for (unsigned j = 0; j < record.jumpTable.simpleJumpTable->branchOffsets.size(); ++j) {
695                 unsigned offset = record.jumpTable.simpleJumpTable->branchOffsets[j];
696                 record.jumpTable.simpleJumpTable->ctiOffsets[j] = offset ? patchBuffer.locationOf(m_labels[bytecodeOffset + offset]) : record.jumpTable.simpleJumpTable->ctiDefault;
697             }
698         } else {
699             ASSERT(record.type == SwitchRecord::String);
700
701             record.jumpTable.stringJumpTable->ctiDefault = patchBuffer.locationOf(m_labels[bytecodeOffset + record.defaultOffset]);
702
703             StringJumpTable::StringOffsetTable::iterator end = record.jumpTable.stringJumpTable->offsetTable.end();            
704             for (StringJumpTable::StringOffsetTable::iterator it = record.jumpTable.stringJumpTable->offsetTable.begin(); it != end; ++it) {
705                 unsigned offset = it->second.branchOffset;
706                 it->second.ctiOffset = offset ? patchBuffer.locationOf(m_labels[bytecodeOffset + offset]) : record.jumpTable.stringJumpTable->ctiDefault;
707             }
708         }
709     }
710
711     for (size_t i = 0; i < m_codeBlock->numberOfExceptionHandlers(); ++i) {
712         HandlerInfo& handler = m_codeBlock->exceptionHandler(i);
713         handler.nativeCode = patchBuffer.locationOf(m_labels[handler.target]);
714     }
715
716     for (Vector<CallRecord>::iterator iter = m_calls.begin(); iter != m_calls.end(); ++iter) {
717         if (iter->to)
718             patchBuffer.link(iter->from, FunctionPtr(iter->to));
719     }
720
721     if (m_codeBlock->needsCallReturnIndices()) {
722         m_codeBlock->callReturnIndexVector().reserveCapacity(m_calls.size());
723         for (Vector<CallRecord>::iterator iter = m_calls.begin(); iter != m_calls.end(); ++iter)
724             m_codeBlock->callReturnIndexVector().append(CallReturnOffsetToBytecodeOffset(patchBuffer.returnAddressOffset(iter->from), iter->bytecodeOffset));
725     }
726
727     m_codeBlock->setNumberOfStructureStubInfos(m_propertyAccessCompilationInfo.size());
728     for (unsigned i = 0; i < m_propertyAccessCompilationInfo.size(); ++i)
729         m_propertyAccessCompilationInfo[i].copyToStubInfo(m_codeBlock->structureStubInfo(i), patchBuffer);
730     m_codeBlock->setNumberOfCallLinkInfos(m_callStructureStubCompilationInfo.size());
731     for (unsigned i = 0; i < m_codeBlock->numberOfCallLinkInfos(); ++i) {
732         CallLinkInfo& info = m_codeBlock->callLinkInfo(i);
733         info.callType = m_callStructureStubCompilationInfo[i].callType;
734         info.bytecodeIndex = m_callStructureStubCompilationInfo[i].bytecodeIndex;
735         info.callReturnLocation = CodeLocationLabel(patchBuffer.locationOfNearCall(m_callStructureStubCompilationInfo[i].callReturnLocation));
736         info.hotPathBegin = patchBuffer.locationOf(m_callStructureStubCompilationInfo[i].hotPathBegin);
737         info.hotPathOther = patchBuffer.locationOfNearCall(m_callStructureStubCompilationInfo[i].hotPathOther);
738     }
739     unsigned methodCallCount = m_methodCallCompilationInfo.size();
740     m_codeBlock->addMethodCallLinkInfos(methodCallCount);
741     for (unsigned i = 0; i < methodCallCount; ++i) {
742         MethodCallLinkInfo& info = m_codeBlock->methodCallLinkInfo(i);
743         info.bytecodeIndex = m_methodCallCompilationInfo[i].bytecodeIndex;
744         info.cachedStructure.setLocation(patchBuffer.locationOf(m_methodCallCompilationInfo[i].structureToCompare));
745         info.callReturnLocation = m_codeBlock->structureStubInfo(m_methodCallCompilationInfo[i].propertyAccessIndex).callReturnLocation;
746     }
747
748 #if ENABLE(DFG_JIT) || ENABLE(LLINT)
749     if (canBeOptimized()
750 #if ENABLE(LLINT)
751         || true
752 #endif
753         ) {
754         CompactJITCodeMap::Encoder jitCodeMapEncoder;
755         for (unsigned bytecodeOffset = 0; bytecodeOffset < m_labels.size(); ++bytecodeOffset) {
756             if (m_labels[bytecodeOffset].isSet())
757                 jitCodeMapEncoder.append(bytecodeOffset, patchBuffer.offsetOf(m_labels[bytecodeOffset]));
758         }
759         m_codeBlock->setJITCodeMap(jitCodeMapEncoder.finish());
760     }
761 #endif
762
763     if (m_codeBlock->codeType() == FunctionCode && functionEntryArityCheck)
764         *functionEntryArityCheck = patchBuffer.locationOf(arityCheck);
765     
766     CodeRef result = patchBuffer.finalizeCode();
767     
768     m_globalData->machineCodeBytesPerBytecodeWordForBaselineJIT.add(
769         static_cast<double>(result.size()) /
770         static_cast<double>(m_codeBlock->instructions().size()));
771     
772     m_codeBlock->shrinkToFit(CodeBlock::LateShrink);
773     
774 #if ENABLE(JIT_VERBOSE)
775     dataLog("JIT generated code for %p at [%p, %p).\n", m_codeBlock, result.executableMemory()->start(), result.executableMemory()->end());
776 #endif
777     
778     return JITCode(result, JITCode::BaselineJIT);
779 }
780
781 void JIT::linkFor(JSFunction* callee, CodeBlock* callerCodeBlock, CodeBlock* calleeCodeBlock, JIT::CodePtr code, CallLinkInfo* callLinkInfo, JSGlobalData* globalData, CodeSpecializationKind kind)
782 {
783     RepatchBuffer repatchBuffer(callerCodeBlock);
784
785     ASSERT(!callLinkInfo->isLinked());
786     callLinkInfo->callee.set(*globalData, callLinkInfo->hotPathBegin, callerCodeBlock->ownerExecutable(), callee);
787     callLinkInfo->lastSeenCallee.set(*globalData, callerCodeBlock->ownerExecutable(), callee);
788     repatchBuffer.relink(callLinkInfo->hotPathOther, code);
789
790     if (calleeCodeBlock)
791         calleeCodeBlock->linkIncomingCall(callLinkInfo);
792
793     // Patch the slow patch so we do not continue to try to link.
794     if (kind == CodeForCall) {
795         repatchBuffer.relink(CodeLocationNearCall(callLinkInfo->callReturnLocation), globalData->jitStubs->ctiVirtualCall());
796         return;
797     }
798
799     ASSERT(kind == CodeForConstruct);
800     repatchBuffer.relink(CodeLocationNearCall(callLinkInfo->callReturnLocation), globalData->jitStubs->ctiVirtualConstruct());
801 }
802
803 } // namespace JSC
804
805 #endif // ENABLE(JIT)