PageRenderTime 86ms CodeModel.GetById 28ms RepoModel.GetById 1ms app.codeStats 0ms

/rpython/jit/backend/x86/assembler.py

https://bitbucket.org/pjenvey/pypy-mq
Python | 2678 lines | 2047 code | 236 blank | 395 comment | 353 complexity | c47643153212b61224566887c76ba55e MD5 | raw file
Possible License(s): Apache-2.0, AGPL-3.0, BSD-3-Clause
  1. import sys
  2. import os
  3. import py
  4. from rpython.jit.backend.llsupport import symbolic, jitframe, rewrite
  5. from rpython.jit.backend.llsupport.assembler import (GuardToken, BaseAssembler, debug_bridge)
  6. from rpython.jit.backend.llsupport.asmmemmgr import MachineDataBlockWrapper
  7. from rpython.jit.backend.llsupport.gcmap import allocate_gcmap
  8. from rpython.jit.metainterp.history import (Const, VOID, ConstInt)
  9. from rpython.jit.metainterp.history import AbstractFailDescr, INT, REF, FLOAT
  10. from rpython.jit.metainterp.compile import ResumeGuardDescr
  11. from rpython.rlib.rjitlog import rjitlog as jl
  12. from rpython.rtyper.lltypesystem import lltype, rffi, rstr, llmemory
  13. from rpython.rtyper.lltypesystem.lloperation import llop
  14. from rpython.rtyper.annlowlevel import cast_instance_to_gcref
  15. from rpython.rtyper import rclass
  16. from rpython.rlib.jit import AsmInfo
  17. from rpython.jit.backend.model import CompiledLoopToken
  18. from rpython.jit.backend.x86.regalloc import (RegAlloc, get_ebp_ofs,
  19. gpr_reg_mgr_cls, xmm_reg_mgr_cls)
  20. from rpython.jit.backend.llsupport.regalloc import (get_scale, valid_addressing_size)
  21. from rpython.jit.backend.x86.arch import (FRAME_FIXED_SIZE, WORD, IS_X86_64,
  22. JITFRAME_FIXED_SIZE, IS_X86_32,
  23. PASS_ON_MY_FRAME, THREADLOCAL_OFS,
  24. DEFAULT_FRAME_BYTES)
  25. from rpython.jit.backend.x86.regloc import (eax, ecx, edx, ebx, esp, ebp, esi,
  26. xmm0, xmm1, xmm2, xmm3, xmm4, xmm5, xmm6, xmm7, r8, r9, r10, r11, edi,
  27. r12, r13, r14, r15, X86_64_SCRATCH_REG, X86_64_XMM_SCRATCH_REG,
  28. RegLoc, FrameLoc, ConstFloatLoc, ImmedLoc, AddressLoc, imm,
  29. imm0, imm1, FloatImmedLoc, RawEbpLoc, RawEspLoc)
  30. from rpython.rlib.objectmodel import we_are_translated
  31. from rpython.jit.backend.x86 import rx86, codebuf, callbuilder
  32. from rpython.jit.backend.x86.vector_ext import VectorAssemblerMixin
  33. from rpython.jit.backend.x86.callbuilder import follow_jump
  34. from rpython.jit.metainterp.resoperation import rop
  35. from rpython.jit.backend.x86 import support
  36. from rpython.rlib.debug import debug_print, debug_start, debug_stop
  37. from rpython.rlib import rgc
  38. from rpython.jit.codewriter.effectinfo import EffectInfo
  39. from rpython.jit.codewriter import longlong
  40. from rpython.rlib.rarithmetic import intmask, r_uint
  41. from rpython.rlib.objectmodel import compute_unique_id
  42. class Assembler386(BaseAssembler, VectorAssemblerMixin):
  43. _regalloc = None
  44. _output_loop_log = None
  45. _second_tmp_reg = ecx
  46. DEBUG_FRAME_DEPTH = False
  47. def __init__(self, cpu, translate_support_code=False):
  48. BaseAssembler.__init__(self, cpu, translate_support_code)
  49. self.verbose = False
  50. self.loop_run_counters = []
  51. self.float_const_neg_addr = 0
  52. self.float_const_abs_addr = 0
  53. self.single_float_const_neg_addr = 0
  54. self.single_float_const_abs_addr = 0
  55. self.expand_byte_mask_addr = 0
  56. self.malloc_slowpath = 0
  57. self.malloc_slowpath_varsize = 0
  58. self.wb_slowpath = [0, 0, 0, 0, 0]
  59. self.setup_failure_recovery()
  60. self.datablockwrapper = None
  61. self.stack_check_slowpath = 0
  62. self.propagate_exception_path = 0
  63. self.teardown()
  64. def setup_once(self):
  65. BaseAssembler.setup_once(self)
  66. if self.cpu.supports_floats:
  67. support.ensure_sse2_floats()
  68. self._build_float_constants()
  69. def setup(self, looptoken):
  70. BaseAssembler.setup(self, looptoken)
  71. assert self.memcpy_addr != 0, "setup_once() not called?"
  72. self.current_clt = looptoken.compiled_loop_token
  73. self.pending_guard_tokens = []
  74. if WORD == 8:
  75. self.pending_memoryerror_trampoline_from = []
  76. self.error_trampoline_64 = 0
  77. self.mc = codebuf.MachineCodeBlockWrapper()
  78. #assert self.datablockwrapper is None --- but obscure case
  79. # possible, e.g. getting MemoryError and continuing
  80. allblocks = self.get_asmmemmgr_blocks(looptoken)
  81. self.datablockwrapper = MachineDataBlockWrapper(self.cpu.asmmemmgr,
  82. allblocks)
  83. self.target_tokens_currently_compiling = {}
  84. self.frame_depth_to_patch = []
  85. def teardown(self):
  86. self.pending_guard_tokens = None
  87. if WORD == 8:
  88. self.pending_memoryerror_trampoline_from = None
  89. self.mc = None
  90. self.current_clt = None
  91. def _build_float_constants(self):
  92. # 0x80000000000000008000000000000000
  93. neg_const = '\x00\x00\x00\x00\x00\x00\x00\x80\x00\x00\x00\x00\x00\x00\x00\x80'
  94. # 0x7FFFFFFFFFFFFFFF7FFFFFFFFFFFFFFF
  95. abs_const = '\xFF\xFF\xFF\xFF\xFF\xFF\xFF\x7F\xFF\xFF\xFF\xFF\xFF\xFF\xFF\x7F'
  96. # 0x7FFFFFFF7FFFFFFF7FFFFFFF7FFFFFFF
  97. single_abs_const = '\xFF\xFF\xFF\x7F\xFF\xFF\xFF\x7F\xFF\xFF\xFF\x7F\xFF\xFF\xFF\x7F'
  98. # 0x80000000800000008000000080000000
  99. single_neg_const = '\x00\x00\x00\x80\x00\x00\x00\x80\x00\x00\x00\x80\x00\x00\x00\x80'
  100. zero_const = '\x00' * 16
  101. #
  102. data = neg_const + abs_const + \
  103. single_neg_const + single_abs_const + \
  104. zero_const
  105. datablockwrapper = MachineDataBlockWrapper(self.cpu.asmmemmgr, [])
  106. float_constants = datablockwrapper.malloc_aligned(len(data), alignment=16)
  107. datablockwrapper.done()
  108. addr = rffi.cast(rffi.CArrayPtr(lltype.Char), float_constants)
  109. for i in range(len(data)):
  110. addr[i] = data[i]
  111. self.float_const_neg_addr = float_constants
  112. self.float_const_abs_addr = float_constants + 16
  113. self.single_float_const_neg_addr = float_constants + 32
  114. self.single_float_const_abs_addr = float_constants + 48
  115. self.expand_byte_mask_addr = float_constants + 64
  116. def set_extra_stack_depth(self, mc, value):
  117. if self._is_asmgcc():
  118. extra_ofs = self.cpu.get_ofs_of_frame_field('jf_extra_stack_depth')
  119. mc.MOV_bi(extra_ofs, value)
  120. def build_frame_realloc_slowpath(self):
  121. mc = codebuf.MachineCodeBlockWrapper()
  122. self._push_all_regs_to_frame(mc, [], self.cpu.supports_floats)
  123. # the caller already did push_gcmap(store=True)
  124. if IS_X86_64:
  125. mc.MOV_rs(esi.value, WORD*2)
  126. # push first arg
  127. mc.MOV_rr(edi.value, ebp.value)
  128. align = callbuilder.align_stack_words(1)
  129. mc.SUB_ri(esp.value, (align - 1) * WORD)
  130. else:
  131. align = callbuilder.align_stack_words(3)
  132. mc.MOV_rs(eax.value, WORD * 2)
  133. mc.SUB_ri(esp.value, (align - 1) * WORD)
  134. mc.MOV_sr(WORD, eax.value)
  135. mc.MOV_sr(0, ebp.value)
  136. # align
  137. self.set_extra_stack_depth(mc, align * WORD)
  138. self._store_and_reset_exception(mc, None, ebx, ecx)
  139. mc.CALL(imm(self.cpu.realloc_frame))
  140. mc.MOV_rr(ebp.value, eax.value)
  141. self._restore_exception(mc, None, ebx, ecx)
  142. mc.ADD_ri(esp.value, (align - 1) * WORD)
  143. self.set_extra_stack_depth(mc, 0)
  144. gcrootmap = self.cpu.gc_ll_descr.gcrootmap
  145. if gcrootmap and gcrootmap.is_shadow_stack:
  146. self._load_shadowstack_top_in_ebx(mc, gcrootmap)
  147. mc.MOV_mr((ebx.value, -WORD), eax.value)
  148. self.pop_gcmap(mc) # cancel the push_gcmap(store=True) in the caller
  149. self._pop_all_regs_from_frame(mc, [], self.cpu.supports_floats)
  150. mc.RET()
  151. self._frame_realloc_slowpath = mc.materialize(self.cpu, [])
  152. def _build_cond_call_slowpath(self, supports_floats, callee_only):
  153. """ This builds a general call slowpath, for whatever call happens to
  154. come.
  155. """
  156. mc = codebuf.MachineCodeBlockWrapper()
  157. # copy registers to the frame, with the exception of the
  158. # 'cond_call_register_arguments' and eax, because these have already
  159. # been saved by the caller. Note that this is not symmetrical:
  160. # these 5 registers are saved by the caller but 4 of them are
  161. # restored here at the end of this function.
  162. self._push_all_regs_to_frame(mc, cond_call_register_arguments + [eax],
  163. supports_floats, callee_only)
  164. # the caller already did push_gcmap(store=True)
  165. if IS_X86_64:
  166. mc.SUB(esp, imm(WORD)) # alignment
  167. self.set_extra_stack_depth(mc, 2 * WORD)
  168. # the arguments are already in the correct registers
  169. else:
  170. # we want space for 4 arguments + call + alignment
  171. mc.SUB(esp, imm(WORD * 7))
  172. self.set_extra_stack_depth(mc, 8 * WORD)
  173. # store the arguments at the correct place in the stack
  174. for i in range(4):
  175. mc.MOV_sr(i * WORD, cond_call_register_arguments[i].value)
  176. mc.CALL(eax)
  177. self._reload_frame_if_necessary(mc)
  178. if IS_X86_64:
  179. mc.ADD(esp, imm(WORD))
  180. else:
  181. mc.ADD(esp, imm(WORD * 7))
  182. self.set_extra_stack_depth(mc, 0)
  183. self.pop_gcmap(mc) # cancel the push_gcmap(store=True) in the caller
  184. self._pop_all_regs_from_frame(mc, [eax], supports_floats, callee_only)
  185. mc.RET()
  186. return mc.materialize(self.cpu, [])
  187. def _build_malloc_slowpath(self, kind):
  188. """ While arriving on slowpath, we have a gcpattern on stack 0.
  189. The arguments are passed in ecx and edx, as follows:
  190. kind == 'fixed': nursery_head in ecx and the size in (edx - ecx).
  191. kind == 'str/unicode': length of the string to allocate in edx.
  192. kind == 'var': length to allocate in edx, tid in ecx,
  193. and itemsize in the stack 1 (position esp+WORD).
  194. This function must preserve all registers apart from ecx and edx.
  195. """
  196. assert kind in ['fixed', 'str', 'unicode', 'var']
  197. mc = codebuf.MachineCodeBlockWrapper()
  198. self._push_all_regs_to_frame(mc, [ecx, edx], self.cpu.supports_floats)
  199. # the caller already did push_gcmap(store=True)
  200. #
  201. if kind == 'fixed':
  202. addr = self.cpu.gc_ll_descr.get_malloc_slowpath_addr()
  203. elif kind == 'str':
  204. addr = self.cpu.gc_ll_descr.get_malloc_fn_addr('malloc_str')
  205. elif kind == 'unicode':
  206. addr = self.cpu.gc_ll_descr.get_malloc_fn_addr('malloc_unicode')
  207. else:
  208. addr = self.cpu.gc_ll_descr.get_malloc_slowpath_array_addr()
  209. mc.SUB_ri(esp.value, 16 - WORD) # restore 16-byte alignment
  210. # magically, the above is enough on X86_32 to reserve 3 stack places
  211. if kind == 'fixed':
  212. mc.SUB_rr(edx.value, ecx.value) # compute the size we want
  213. if IS_X86_32:
  214. mc.MOV_sr(0, edx.value) # store the length
  215. if hasattr(self.cpu.gc_ll_descr, 'passes_frame'):
  216. mc.MOV_sr(WORD, ebp.value) # for tests only
  217. else:
  218. mc.MOV_rr(edi.value, edx.value) # length argument
  219. if hasattr(self.cpu.gc_ll_descr, 'passes_frame'):
  220. mc.MOV_rr(esi.value, ebp.value) # for tests only
  221. elif kind == 'str' or kind == 'unicode':
  222. if IS_X86_32:
  223. # stack layout: [---][---][---][ret].. with 3 free stack places
  224. mc.MOV_sr(0, edx.value) # store the length
  225. elif IS_X86_64:
  226. mc.MOV_rr(edi.value, edx.value) # length argument
  227. else:
  228. if IS_X86_32:
  229. # stack layout: [---][---][---][ret][gcmap][itemsize]...
  230. mc.MOV_sr(WORD * 2, edx.value) # store the length
  231. mc.MOV_sr(WORD * 1, ecx.value) # store the tid
  232. mc.MOV_rs(edx.value, WORD * 5) # load the itemsize
  233. mc.MOV_sr(WORD * 0, edx.value) # store the itemsize
  234. else:
  235. # stack layout: [---][ret][gcmap][itemsize]...
  236. # (already in edx) # length
  237. mc.MOV_rr(esi.value, ecx.value) # tid
  238. mc.MOV_rs(edi.value, WORD * 3) # load the itemsize
  239. self.set_extra_stack_depth(mc, 16)
  240. mc.CALL(imm(follow_jump(addr)))
  241. self._reload_frame_if_necessary(mc)
  242. mc.ADD_ri(esp.value, 16 - WORD)
  243. self.set_extra_stack_depth(mc, 0)
  244. #
  245. mc.TEST_rr(eax.value, eax.value)
  246. mc.J_il(rx86.Conditions['Z'], 0xfffff) # patched later
  247. jz_location = mc.get_relative_pos()
  248. mc.MOV_rr(ecx.value, eax.value)
  249. #
  250. nursery_free_adr = self.cpu.gc_ll_descr.get_nursery_free_addr()
  251. self._pop_all_regs_from_frame(mc, [ecx, edx], self.cpu.supports_floats)
  252. mc.MOV(edx, heap(nursery_free_adr)) # load this in EDX
  253. self.pop_gcmap(mc) # push_gcmap(store=True) done by the caller
  254. mc.RET()
  255. #
  256. # If the slowpath malloc failed, we raise a MemoryError that
  257. # always interrupts the current loop, as a "good enough"
  258. # approximation. We have to adjust the esp a little, to point to
  259. # the correct "ret" arg
  260. offset = mc.get_relative_pos() - jz_location
  261. mc.overwrite32(jz_location-4, offset)
  262. # From now on this function is basically "merged" with
  263. # its caller and so contains DEFAULT_FRAME_BYTES bytes
  264. # plus my own return address, which we'll ignore next
  265. mc.force_frame_size(DEFAULT_FRAME_BYTES + WORD)
  266. mc.ADD_ri(esp.value, WORD)
  267. mc.JMP(imm(self.propagate_exception_path))
  268. #
  269. rawstart = mc.materialize(self.cpu, [])
  270. return rawstart
  271. def _build_propagate_exception_path(self):
  272. self.mc = codebuf.MachineCodeBlockWrapper()
  273. self.mc.force_frame_size(DEFAULT_FRAME_BYTES)
  274. #
  275. # read and reset the current exception
  276. self._store_and_reset_exception(self.mc, eax)
  277. ofs = self.cpu.get_ofs_of_frame_field('jf_guard_exc')
  278. self.mc.MOV_br(ofs, eax.value)
  279. propagate_exception_descr = rffi.cast(lltype.Signed,
  280. cast_instance_to_gcref(self.cpu.propagate_exception_descr))
  281. ofs = self.cpu.get_ofs_of_frame_field('jf_descr')
  282. self.mc.MOV(RawEbpLoc(ofs), imm(propagate_exception_descr))
  283. #
  284. self._call_footer()
  285. rawstart = self.mc.materialize(self.cpu, [])
  286. self.propagate_exception_path = rawstart
  287. self.mc = None
  288. def _build_stack_check_slowpath(self):
  289. _, _, slowpathaddr = self.cpu.insert_stack_check()
  290. if slowpathaddr == 0 or not self.cpu.propagate_exception_descr:
  291. return # no stack check (for tests, or non-translated)
  292. #
  293. # make a regular function that is called from a point near the start
  294. # of an assembler function (after it adjusts the stack and saves
  295. # registers).
  296. mc = codebuf.MachineCodeBlockWrapper()
  297. #
  298. if IS_X86_64:
  299. mc.MOV_rr(edi.value, esp.value)
  300. mc.SUB_ri(esp.value, WORD) # alignment
  301. #
  302. if IS_X86_32:
  303. mc.SUB_ri(esp.value, 2*WORD) # alignment
  304. mc.PUSH_r(esp.value)
  305. #
  306. # esp is now aligned to a multiple of 16 again
  307. mc.CALL(imm(follow_jump(slowpathaddr)))
  308. #
  309. if IS_X86_32:
  310. mc.ADD_ri(esp.value, 3*WORD) # alignment
  311. else:
  312. mc.ADD_ri(esp.value, WORD)
  313. #
  314. mc.MOV(eax, heap(self.cpu.pos_exception()))
  315. mc.TEST_rr(eax.value, eax.value)
  316. mc.J_il8(rx86.Conditions['NZ'], 0)
  317. jnz_location = mc.get_relative_pos()
  318. #
  319. mc.RET()
  320. #
  321. # patch the JNZ above
  322. offset = mc.get_relative_pos() - jnz_location
  323. assert 0 < offset <= 127
  324. mc.overwrite(jnz_location-1, chr(offset))
  325. # From now on this function is basically "merged" with
  326. # its caller and so contains DEFAULT_FRAME_BYTES bytes
  327. # plus my own return address, which we'll ignore next
  328. mc.force_frame_size(DEFAULT_FRAME_BYTES + WORD)
  329. mc.ADD_ri(esp.value, WORD)
  330. mc.JMP(imm(self.propagate_exception_path))
  331. #
  332. rawstart = mc.materialize(self.cpu, [])
  333. self.stack_check_slowpath = rawstart
  334. def _build_wb_slowpath(self, withcards, withfloats=False, for_frame=False):
  335. descr = self.cpu.gc_ll_descr.write_barrier_descr
  336. exc0, exc1 = None, None
  337. if descr is None:
  338. return
  339. if not withcards:
  340. func = descr.get_write_barrier_fn(self.cpu)
  341. else:
  342. if descr.jit_wb_cards_set == 0:
  343. return
  344. func = descr.get_write_barrier_from_array_fn(self.cpu)
  345. if func == 0:
  346. return
  347. #
  348. # This builds a helper function called from the slow path of
  349. # write barriers. It must save all registers, and optionally
  350. # all XMM registers. It takes a single argument just pushed
  351. # on the stack even on X86_64. It must restore stack alignment
  352. # accordingly.
  353. mc = codebuf.MachineCodeBlockWrapper()
  354. #
  355. if not for_frame:
  356. self._push_all_regs_to_frame(mc, [], withfloats, callee_only=True)
  357. if IS_X86_32:
  358. # we have 2 extra words on stack for retval and we pass 1 extra
  359. # arg, so we need to substract 2 words
  360. mc.SUB_ri(esp.value, 2 * WORD)
  361. mc.MOV_rs(eax.value, 3 * WORD) # 2 + 1
  362. mc.MOV_sr(0, eax.value)
  363. else:
  364. mc.MOV_rs(edi.value, WORD)
  365. else:
  366. # NOTE: don't save registers on the jitframe here!
  367. # It might override already-saved values that will be
  368. # restored later...
  369. #
  370. # This 'for_frame' version is called after a CALL. It does not
  371. # need to save many registers: the registers that are anyway
  372. # destroyed by the call can be ignored (volatiles), and the
  373. # non-volatile registers won't be changed here. It only needs
  374. # to save eax, maybe edx, and xmm0 (possible results of the call)
  375. # and two more non-volatile registers (used to store the RPython
  376. # exception that occurred in the CALL, if any).
  377. assert not withcards
  378. # we have one word to align
  379. mc.SUB_ri(esp.value, 7 * WORD) # align and reserve some space
  380. mc.MOV_sr(WORD, eax.value) # save for later
  381. if self.cpu.supports_floats:
  382. mc.MOVSD_sx(2 * WORD, xmm0.value) # 32-bit: also 3 * WORD
  383. if IS_X86_32:
  384. mc.MOV_sr(4 * WORD, edx.value)
  385. mc.MOV_sr(0, ebp.value)
  386. exc0, exc1 = esi, edi
  387. else:
  388. mc.MOV_rr(edi.value, ebp.value)
  389. exc0, exc1 = ebx, r12
  390. mc.MOV(RawEspLoc(WORD * 5, REF), exc0)
  391. mc.MOV(RawEspLoc(WORD * 6, INT), exc1)
  392. # note that it's safe to store the exception in register,
  393. # since the call to write barrier can't collect
  394. # (and this is assumed a bit left and right here, like lack
  395. # of _reload_frame_if_necessary)
  396. self._store_and_reset_exception(mc, exc0, exc1)
  397. mc.CALL(imm(func))
  398. #
  399. if withcards:
  400. # A final TEST8 before the RET, for the caller. Careful to
  401. # not follow this instruction with another one that changes
  402. # the status of the CPU flags!
  403. if IS_X86_32:
  404. mc.MOV_rs(eax.value, 3*WORD)
  405. else:
  406. mc.MOV_rs(eax.value, WORD)
  407. mc.TEST8(addr_add_const(eax, descr.jit_wb_if_flag_byteofs),
  408. imm(-0x80))
  409. #
  410. if not for_frame:
  411. if IS_X86_32:
  412. # ADD touches CPU flags
  413. mc.LEA_rs(esp.value, 2 * WORD)
  414. self._pop_all_regs_from_frame(mc, [], withfloats, callee_only=True)
  415. mc.RET16_i(WORD)
  416. # Note that wb_slowpath[0..3] end with a RET16_i, which must be
  417. # taken care of in the caller by stack_frame_size_delta(-WORD)
  418. else:
  419. if IS_X86_32:
  420. mc.MOV_rs(edx.value, 4 * WORD)
  421. if self.cpu.supports_floats:
  422. mc.MOVSD_xs(xmm0.value, 2 * WORD)
  423. mc.MOV_rs(eax.value, WORD) # restore
  424. self._restore_exception(mc, exc0, exc1)
  425. mc.MOV(exc0, RawEspLoc(WORD * 5, REF))
  426. mc.MOV(exc1, RawEspLoc(WORD * 6, INT))
  427. mc.LEA_rs(esp.value, 7 * WORD)
  428. mc.RET()
  429. rawstart = mc.materialize(self.cpu, [])
  430. if for_frame:
  431. self.wb_slowpath[4] = rawstart
  432. else:
  433. self.wb_slowpath[withcards + 2 * withfloats] = rawstart
  434. @rgc.no_release_gil
  435. def assemble_loop(self, jd_id, unique_id, logger, loopname, inputargs,
  436. operations, looptoken, log):
  437. '''adds the following attributes to looptoken:
  438. _ll_function_addr (address of the generated func, as an int)
  439. _ll_loop_code (debug: addr of the start of the ResOps)
  440. _x86_fullsize (debug: full size including failure)
  441. '''
  442. # XXX this function is too longish and contains some code
  443. # duplication with assemble_bridge(). Also, we should think
  444. # about not storing on 'self' attributes that will live only
  445. # for the duration of compiling one loop or a one bridge.
  446. clt = CompiledLoopToken(self.cpu, looptoken.number)
  447. looptoken.compiled_loop_token = clt
  448. clt._debug_nbargs = len(inputargs)
  449. if not we_are_translated():
  450. # Arguments should be unique
  451. assert len(set(inputargs)) == len(inputargs)
  452. self.setup(looptoken)
  453. if self.cpu.HAS_CODEMAP:
  454. self.codemap_builder.enter_portal_frame(jd_id, unique_id,
  455. self.mc.get_relative_pos())
  456. frame_info = self.datablockwrapper.malloc_aligned(
  457. jitframe.JITFRAMEINFO_SIZE, alignment=WORD)
  458. clt.frame_info = rffi.cast(jitframe.JITFRAMEINFOPTR, frame_info)
  459. clt.frame_info.clear() # for now
  460. if log:
  461. number = looptoken.number
  462. operations = self._inject_debugging_code(looptoken, operations,
  463. 'e', number)
  464. regalloc = RegAlloc(self, self.cpu.translate_support_code)
  465. #
  466. allgcrefs = []
  467. operations = regalloc.prepare_loop(inputargs, operations,
  468. looptoken, allgcrefs)
  469. self.reserve_gcref_table(allgcrefs)
  470. functionpos = self.mc.get_relative_pos()
  471. self._call_header_with_stack_check()
  472. self._check_frame_depth_debug(self.mc)
  473. looppos = self.mc.get_relative_pos()
  474. frame_depth_no_fixed_size = self._assemble(regalloc, inputargs,
  475. operations)
  476. self.update_frame_depth(frame_depth_no_fixed_size + JITFRAME_FIXED_SIZE)
  477. #
  478. size_excluding_failure_stuff = self.mc.get_relative_pos()
  479. self.write_pending_failure_recoveries(regalloc)
  480. full_size = self.mc.get_relative_pos()
  481. #
  482. rawstart = self.materialize_loop(looptoken)
  483. self.patch_gcref_table(looptoken, rawstart)
  484. self.patch_stack_checks(frame_depth_no_fixed_size + JITFRAME_FIXED_SIZE,
  485. rawstart)
  486. looptoken._ll_loop_code = looppos + rawstart
  487. debug_start("jit-backend-addr")
  488. debug_print("Loop %d (%s) has address 0x%x to 0x%x (bootstrap 0x%x)" % (
  489. looptoken.number, loopname,
  490. r_uint(rawstart + looppos),
  491. r_uint(rawstart + size_excluding_failure_stuff),
  492. r_uint(rawstart + functionpos)))
  493. debug_print(" gc table: 0x%x" % r_uint(self.gc_table_addr))
  494. debug_print(" function: 0x%x" % r_uint(rawstart + functionpos))
  495. debug_print(" resops: 0x%x" % r_uint(rawstart + looppos))
  496. debug_print(" failures: 0x%x" % r_uint(rawstart +
  497. size_excluding_failure_stuff))
  498. debug_print(" end: 0x%x" % r_uint(rawstart + full_size))
  499. debug_stop("jit-backend-addr")
  500. self.patch_pending_failure_recoveries(rawstart)
  501. #
  502. ops_offset = self.mc.ops_offset
  503. if not we_are_translated():
  504. # used only by looptoken.dump() -- useful in tests
  505. looptoken._x86_rawstart = rawstart
  506. looptoken._x86_fullsize = full_size
  507. looptoken._x86_ops_offset = ops_offset
  508. looptoken._ll_function_addr = rawstart + functionpos
  509. if logger:
  510. log = logger.log_trace(jl.MARK_TRACE_ASM, None, self.mc)
  511. log.write(inputargs, operations, ops_offset=ops_offset)
  512. # legacy
  513. if logger.logger_ops:
  514. logger.logger_ops.log_loop(inputargs, operations, 0,
  515. "rewritten", name=loopname,
  516. ops_offset=ops_offset)
  517. self.fixup_target_tokens(rawstart)
  518. self.teardown()
  519. # oprofile support
  520. if self.cpu.profile_agent is not None:
  521. name = "Loop # %s: %s" % (looptoken.number, loopname)
  522. self.cpu.profile_agent.native_code_written(name,
  523. rawstart, full_size)
  524. return AsmInfo(ops_offset, rawstart + looppos,
  525. size_excluding_failure_stuff - looppos, rawstart)
  526. @rgc.no_release_gil
  527. def assemble_bridge(self, faildescr, inputargs, operations,
  528. original_loop_token, log, logger):
  529. if not we_are_translated():
  530. # Arguments should be unique
  531. assert len(set(inputargs)) == len(inputargs)
  532. self.setup(original_loop_token)
  533. if self.cpu.HAS_CODEMAP:
  534. self.codemap_builder.inherit_code_from_position(
  535. faildescr.adr_jump_offset)
  536. self.mc.force_frame_size(DEFAULT_FRAME_BYTES)
  537. descr_number = compute_unique_id(faildescr)
  538. if log:
  539. operations = self._inject_debugging_code(faildescr, operations,
  540. 'b', descr_number)
  541. arglocs = self.rebuild_faillocs_from_descr(faildescr, inputargs)
  542. regalloc = RegAlloc(self, self.cpu.translate_support_code)
  543. allgcrefs = []
  544. operations = regalloc.prepare_bridge(inputargs, arglocs,
  545. operations,
  546. allgcrefs,
  547. self.current_clt.frame_info)
  548. self.reserve_gcref_table(allgcrefs)
  549. startpos = self.mc.get_relative_pos()
  550. self._check_frame_depth(self.mc, regalloc.get_gcmap())
  551. bridgestartpos = self.mc.get_relative_pos()
  552. self._update_at_exit(arglocs, inputargs, faildescr, regalloc)
  553. frame_depth_no_fixed_size = self._assemble(regalloc, inputargs, operations)
  554. codeendpos = self.mc.get_relative_pos()
  555. self.write_pending_failure_recoveries(regalloc)
  556. fullsize = self.mc.get_relative_pos()
  557. #
  558. rawstart = self.materialize_loop(original_loop_token)
  559. self.patch_gcref_table(original_loop_token, rawstart)
  560. self.patch_stack_checks(frame_depth_no_fixed_size + JITFRAME_FIXED_SIZE,
  561. rawstart)
  562. debug_start("jit-backend-addr")
  563. debug_print("bridge out of Guard 0x%x has address 0x%x to 0x%x" %
  564. (r_uint(descr_number), r_uint(rawstart + startpos),
  565. r_uint(rawstart + codeendpos)))
  566. debug_print(" gc table: 0x%x" % r_uint(self.gc_table_addr))
  567. debug_print(" jump target: 0x%x" % r_uint(rawstart + startpos))
  568. debug_print(" resops: 0x%x" % r_uint(rawstart + bridgestartpos))
  569. debug_print(" failures: 0x%x" % r_uint(rawstart + codeendpos))
  570. debug_print(" end: 0x%x" % r_uint(rawstart + fullsize))
  571. debug_stop("jit-backend-addr")
  572. self.patch_pending_failure_recoveries(rawstart)
  573. # patch the jump from original guard
  574. self.patch_jump_for_descr(faildescr, rawstart + startpos)
  575. ops_offset = self.mc.ops_offset
  576. frame_depth = max(self.current_clt.frame_info.jfi_frame_depth,
  577. frame_depth_no_fixed_size + JITFRAME_FIXED_SIZE)
  578. if logger:
  579. log = logger.log_trace(jl.MARK_TRACE_ASM, None, self.mc)
  580. log.write(inputargs, operations, ops_offset)
  581. # log that the already written bridge is stitched to a descr!
  582. logger.log_patch_guard(descr_number, rawstart)
  583. # legacy
  584. if logger.logger_ops:
  585. logger.logger_ops.log_bridge(inputargs, operations, "rewritten",
  586. faildescr, ops_offset=ops_offset)
  587. self.fixup_target_tokens(rawstart)
  588. self.update_frame_depth(frame_depth)
  589. self.teardown()
  590. # oprofile support
  591. if self.cpu.profile_agent is not None:
  592. name = "Bridge # %s" % (descr_number,)
  593. self.cpu.profile_agent.native_code_written(name,
  594. rawstart, fullsize)
  595. return AsmInfo(ops_offset, startpos + rawstart, codeendpos - startpos, rawstart+bridgestartpos)
  596. def stitch_bridge(self, faildescr, target):
  597. """ Stitching means that one can enter a bridge with a complete different register
  598. allocation. This needs remapping which is done here for both normal registers
  599. and accumulation registers.
  600. Why? Because this only generates a very small junk of memory, instead of
  601. duplicating the loop assembler for each faildescr!
  602. """
  603. asminfo, bridge_faildescr, version, looptoken = target
  604. assert isinstance(bridge_faildescr, ResumeGuardDescr)
  605. assert isinstance(faildescr, ResumeGuardDescr)
  606. assert asminfo.rawstart != 0
  607. self.mc = codebuf.MachineCodeBlockWrapper()
  608. allblocks = self.get_asmmemmgr_blocks(looptoken)
  609. self.datablockwrapper = MachineDataBlockWrapper(self.cpu.asmmemmgr,
  610. allblocks)
  611. frame_info = self.datablockwrapper.malloc_aligned(
  612. jitframe.JITFRAMEINFO_SIZE, alignment=WORD)
  613. self.mc.force_frame_size(DEFAULT_FRAME_BYTES)
  614. # if accumulation is saved at the guard, we need to update it here!
  615. guard_locs = self.rebuild_faillocs_from_descr(faildescr, version.inputargs)
  616. bridge_locs = self.rebuild_faillocs_from_descr(bridge_faildescr, version.inputargs)
  617. #import pdb; pdb.set_trace()
  618. guard_accum_info = faildescr.rd_vector_info
  619. # O(n^2), but usually you only have at most 1 fail argument
  620. while guard_accum_info:
  621. bridge_accum_info = bridge_faildescr.rd_vector_info
  622. while bridge_accum_info:
  623. if bridge_accum_info.failargs_pos == guard_accum_info.failargs_pos:
  624. # the mapping might be wrong!
  625. if bridge_accum_info.location is not guard_accum_info.location:
  626. self.mov(guard_accum_info.location, bridge_accum_info.location)
  627. bridge_accum_info = bridge_accum_info.next()
  628. guard_accum_info = guard_accum_info.next()
  629. # register mapping is most likely NOT valid, thus remap it in this
  630. # short piece of assembler
  631. assert len(guard_locs) == len(bridge_locs)
  632. for i,gloc in enumerate(guard_locs):
  633. bloc = bridge_locs[i]
  634. bstack = bloc.location_code() == 'b'
  635. gstack = gloc.location_code() == 'b'
  636. if bstack and gstack:
  637. pass
  638. elif gloc is not bloc:
  639. self.mov(gloc, bloc)
  640. offset = self.mc.get_relative_pos()
  641. self.mc.JMP_l(0)
  642. self.mc.writeimm32(0)
  643. self.mc.force_frame_size(DEFAULT_FRAME_BYTES)
  644. rawstart = self.materialize_loop(looptoken)
  645. # update the jump (above) to the real trace
  646. self._patch_jump_to(rawstart + offset, asminfo.rawstart)
  647. # update the guard to jump right to this custom piece of assembler
  648. self.patch_jump_for_descr(faildescr, rawstart)
  649. def _patch_jump_to(self, adr_jump_offset, adr_new_target):
  650. assert adr_jump_offset != 0
  651. offset = adr_new_target - (adr_jump_offset + 5)
  652. mc = codebuf.MachineCodeBlockWrapper()
  653. mc.force_frame_size(DEFAULT_FRAME_BYTES)
  654. if rx86.fits_in_32bits(offset):
  655. mc.JMP_l(offset)
  656. else:
  657. mc.MOV_ri(X86_64_SCRATCH_REG.value, adr_new_target)
  658. mc.JMP_r(X86_64_SCRATCH_REG.value)
  659. mc.copy_to_raw_memory(adr_jump_offset)
  660. def reserve_gcref_table(self, allgcrefs):
  661. gcref_table_size = len(allgcrefs) * WORD
  662. if IS_X86_64:
  663. # align to a multiple of 16 and reserve space at the beginning
  664. # of the machine code for the gc table. This lets us write
  665. # machine code with relative addressing (%rip - constant).
  666. gcref_table_size = (gcref_table_size + 15) & ~15
  667. mc = self.mc
  668. assert mc.get_relative_pos() == 0
  669. for i in range(gcref_table_size):
  670. mc.writechar('\x00')
  671. elif IS_X86_32:
  672. # allocate the gc table right now. This lets us write
  673. # machine code with absolute 32-bit addressing.
  674. self.gc_table_addr = self.datablockwrapper.malloc_aligned(
  675. gcref_table_size, alignment=WORD)
  676. #
  677. self.setup_gcrefs_list(allgcrefs)
  678. def patch_gcref_table(self, looptoken, rawstart):
  679. if IS_X86_64:
  680. # the gc table is at the start of the machine code
  681. self.gc_table_addr = rawstart
  682. elif IS_X86_32:
  683. # the gc table was already allocated by reserve_gcref_table()
  684. rawstart = self.gc_table_addr
  685. #
  686. tracer = self.cpu.gc_ll_descr.make_gcref_tracer(rawstart,
  687. self._allgcrefs)
  688. gcreftracers = self.get_asmmemmgr_gcreftracers(looptoken)
  689. gcreftracers.append(tracer) # keepalive
  690. self.teardown_gcrefs_list()
  691. def write_pending_failure_recoveries(self, regalloc):
  692. # for each pending guard, generate the code of the recovery stub
  693. # at the end of self.mc.
  694. for tok in self.pending_guard_tokens:
  695. descr = tok.faildescr
  696. if descr.loop_version():
  697. startpos = self.mc.get_relative_pos()
  698. self.store_info_on_descr(startpos, tok)
  699. else:
  700. tok.pos_recovery_stub = self.generate_quick_failure(tok, regalloc)
  701. if WORD == 8 and len(self.pending_memoryerror_trampoline_from) > 0:
  702. self.error_trampoline_64 = self.generate_propagate_error_64()
  703. def patch_pending_failure_recoveries(self, rawstart):
  704. # after we wrote the assembler to raw memory, set up
  705. # tok.faildescr.adr_jump_offset to contain the raw address of
  706. # the 4-byte target field in the JMP/Jcond instruction, and patch
  707. # the field in question to point (initially) to the recovery stub
  708. clt = self.current_clt
  709. for tok in self.pending_guard_tokens:
  710. addr = rawstart + tok.pos_jump_offset
  711. tok.faildescr.adr_jump_offset = addr
  712. descr = tok.faildescr
  713. if descr.loop_version():
  714. continue # patch them later
  715. relative_target = tok.pos_recovery_stub - (tok.pos_jump_offset + 4)
  716. assert rx86.fits_in_32bits(relative_target)
  717. #
  718. if not tok.guard_not_invalidated():
  719. mc = codebuf.MachineCodeBlockWrapper()
  720. mc.writeimm32(relative_target)
  721. mc.copy_to_raw_memory(addr)
  722. else:
  723. # GUARD_NOT_INVALIDATED, record an entry in
  724. # clt.invalidate_positions of the form:
  725. # (addr-in-the-code-of-the-not-yet-written-jump-target,
  726. # relative-target-to-use)
  727. relpos = tok.pos_jump_offset
  728. clt.invalidate_positions.append((rawstart + relpos,
  729. relative_target))
  730. # General idea: Although no code was generated by this
  731. # guard, the code might be patched with a "JMP rel32" to
  732. # the guard recovery code. This recovery code is
  733. # already generated, and looks like the recovery code
  734. # for any guard, even if at first it has no jump to it.
  735. # So we may later write 5 bytes overriding the existing
  736. # instructions; this works because a CALL instruction
  737. # would also take at least 5 bytes. If it could take
  738. # less, we would run into the issue that overwriting the
  739. # 5 bytes here might get a few nonsense bytes at the
  740. # return address of the following CALL.
  741. if WORD == 8:
  742. for pos_after_jz in self.pending_memoryerror_trampoline_from:
  743. assert self.error_trampoline_64 != 0 # only if non-empty
  744. mc = codebuf.MachineCodeBlockWrapper()
  745. mc.writeimm32(self.error_trampoline_64 - pos_after_jz)
  746. mc.copy_to_raw_memory(rawstart + pos_after_jz - 4)
  747. def update_frame_depth(self, frame_depth):
  748. baseofs = self.cpu.get_baseofs_of_frame_field()
  749. self.current_clt.frame_info.update_frame_depth(baseofs, frame_depth)
  750. def patch_stack_checks(self, framedepth, rawstart):
  751. for ofs in self.frame_depth_to_patch:
  752. self._patch_frame_depth(ofs + rawstart, framedepth)
  753. def _check_frame_depth(self, mc, gcmap):
  754. """ check if the frame is of enough depth to follow this bridge.
  755. Otherwise reallocate the frame in a helper.
  756. There are other potential solutions
  757. to that, but this one does not sound too bad.
  758. """
  759. descrs = self.cpu.gc_ll_descr.getframedescrs(self.cpu)
  760. ofs = self.cpu.unpack_fielddescr(descrs.arraydescr.lendescr)
  761. mc.CMP_bi(ofs, 0xffffff) # force writing 32 bit
  762. stack_check_cmp_ofs = mc.get_relative_pos() - 4
  763. mc.J_il8(rx86.Conditions['GE'], 0)
  764. jg_location = mc.get_relative_pos()
  765. mc.MOV_si(WORD, 0xffffff) # force writing 32 bit
  766. ofs2 = mc.get_relative_pos() - 4
  767. self.push_gcmap(mc, gcmap, store=True)
  768. mc.CALL(imm(self._frame_realloc_slowpath))
  769. # patch the JG above
  770. offset = mc.get_relative_pos() - jg_location
  771. assert 0 < offset <= 127
  772. mc.overwrite(jg_location-1, chr(offset))
  773. self.frame_depth_to_patch.append(stack_check_cmp_ofs)
  774. self.frame_depth_to_patch.append(ofs2)
  775. def _check_frame_depth_debug(self, mc):
  776. """ double check the depth size. It prints the error (and potentially
  777. segfaults later)
  778. """
  779. if not self.DEBUG_FRAME_DEPTH:
  780. return
  781. descrs = self.cpu.gc_ll_descr.getframedescrs(self.cpu)
  782. ofs = self.cpu.unpack_fielddescr(descrs.arraydescr.lendescr)
  783. mc.CMP_bi(ofs, 0xffffff)
  784. stack_check_cmp_ofs = mc.get_relative_pos() - 4
  785. mc.J_il8(rx86.Conditions['GE'], 0)
  786. jg_location = mc.get_relative_pos()
  787. mc.MOV_rr(edi.value, ebp.value)
  788. mc.MOV_ri(esi.value, 0xffffff)
  789. ofs2 = mc.get_relative_pos() - 4
  790. mc.CALL(imm(self.cpu.realloc_frame_crash))
  791. # patch the JG above
  792. offset = mc.get_relative_pos() - jg_location
  793. assert 0 < offset <= 127
  794. mc.overwrite(jg_location-1, chr(offset))
  795. self.frame_depth_to_patch.append(stack_check_cmp_ofs)
  796. self.frame_depth_to_patch.append(ofs2)
  797. def _patch_frame_depth(self, adr, allocated_depth):
  798. mc = codebuf.MachineCodeBlockWrapper()
  799. mc.writeimm32(allocated_depth)
  800. mc.copy_to_raw_memory(adr)
  801. def materialize_loop(self, looptoken):
  802. self.datablockwrapper.done() # finish using cpu.asmmemmgr
  803. self.datablockwrapper = None
  804. allblocks = self.get_asmmemmgr_blocks(looptoken)
  805. size = self.mc.get_relative_pos()
  806. res = self.mc.materialize(self.cpu, allblocks,
  807. self.cpu.gc_ll_descr.gcrootmap)
  808. if self.cpu.HAS_CODEMAP:
  809. self.cpu.codemap.register_codemap(
  810. self.codemap_builder.get_final_bytecode(res, size))
  811. return res
  812. def patch_jump_for_descr(self, faildescr, adr_new_target):
  813. adr_jump_offset = faildescr.adr_jump_offset
  814. assert adr_jump_offset != 0
  815. offset = adr_new_target - (adr_jump_offset + 4)
  816. # If the new target fits within a rel32 of the jump, just patch
  817. # that. Otherwise, leave the original rel32 to the recovery stub in
  818. # place, but clobber the recovery stub with a jump to the real
  819. # target.
  820. mc = codebuf.MachineCodeBlockWrapper()
  821. mc.force_frame_size(DEFAULT_FRAME_BYTES)
  822. if rx86.fits_in_32bits(offset):
  823. mc.writeimm32(offset)
  824. mc.copy_to_raw_memory(adr_jump_offset)
  825. else:
  826. # "mov r11, addr; jmp r11" is up to 13 bytes, which fits in there
  827. # because we always write "mov r11, imm-as-8-bytes; call *r11" in
  828. # the first place.
  829. mc.MOV_ri(X86_64_SCRATCH_REG.value, adr_new_target)
  830. mc.JMP_r(X86_64_SCRATCH_REG.value)
  831. p = rffi.cast(rffi.INTP, adr_jump_offset)
  832. adr_target = adr_jump_offset + 4 + rffi.cast(lltype.Signed, p[0])
  833. mc.copy_to_raw_memory(adr_target)
  834. faildescr.adr_jump_offset = 0 # means "patched"
  835. def fixup_target_tokens(self, rawstart):
  836. for targettoken in self.target_tokens_currently_compiling:
  837. targettoken._ll_loop_code += rawstart
  838. self.target_tokens_currently_compiling = None
  839. def _assemble(self, regalloc, inputargs, operations):
  840. self._regalloc = regalloc
  841. self.guard_success_cc = rx86.cond_none
  842. regalloc.compute_hint_frame_locations(operations)
  843. regalloc.walk_operations(inputargs, operations)
  844. assert self.guard_success_cc == rx86.cond_none
  845. if we_are_translated() or self.cpu.dont_keepalive_stuff:
  846. self._regalloc = None # else keep it around for debugging
  847. frame_depth = regalloc.get_final_frame_depth()
  848. jump_target_descr = regalloc.jump_target_descr
  849. if jump_target_descr is not None:
  850. tgt_depth = jump_target_descr._x86_clt.frame_info.jfi_frame_depth
  851. target_frame_depth = tgt_depth - JITFRAME_FIXED_SIZE
  852. frame_depth = max(frame_depth, target_frame_depth)
  853. return frame_depth
  854. def _call_header_vmprof(self):
  855. from rpython.rlib.rvmprof.rvmprof import cintf, VMPROF_JITTED_TAG
  856. # tloc = address of pypy_threadlocal_s
  857. if IS_X86_32:
  858. # Can't use esi here, its old value is not saved yet.
  859. # But we can use eax and ecx.
  860. self.mc.MOV_rs(edx.value, THREADLOCAL_OFS)
  861. tloc = edx
  862. old = ecx
  863. else:
  864. # The thread-local value is already in esi.
  865. # We should avoid if possible to use ecx or edx because they
  866. # would be used to pass arguments #3 and #4 (even though, so
  867. # far, the assembler only receives two arguments).
  868. tloc = esi
  869. old = r11
  870. # eax = address in the stack of a 3-words struct vmprof_stack_s
  871. self.mc.LEA_rs(eax.value, (FRAME_FIXED_SIZE - 4) * WORD)
  872. # old = current value of vmprof_tl_stack
  873. offset = cintf.vmprof_tl_stack.getoffset()
  874. self.mc.MOV_rm(old.value, (tloc.value, offset))
  875. # eax->next = old
  876. self.mc.MOV_mr((eax.value, 0), old.value)
  877. # eax->value = my esp
  878. self.mc.MOV_mr((eax.value, WORD), esp.value)
  879. # eax->kind = VMPROF_JITTED_TAG
  880. self.mc.MOV_mi((eax.value, WORD * 2), VMPROF_JITTED_TAG)
  881. # save in vmprof_tl_stack the new eax
  882. self.mc.MOV_mr((tloc.value, offset), eax.value)
  883. def _call_footer_vmprof(self):
  884. from rpython.rlib.rvmprof.rvmprof import cintf
  885. # edx = address of pypy_threadlocal_s
  886. self.mc.MOV_rs(edx.value, THREADLOCAL_OFS)
  887. self.mc.AND_ri(edx.value, ~1)
  888. # eax = (our local vmprof_tl_stack).next
  889. self.mc.MOV_rs(eax.value, (FRAME_FIXED_SIZE - 4 + 0) * WORD)
  890. # save in vmprof_tl_stack the value eax
  891. offset = cintf.vmprof_tl_stack.getoffset()
  892. self.mc.MOV_mr((edx.value, offset), eax.value)
  893. def _call_header(self):
  894. self.mc.SUB_ri(esp.value, FRAME_FIXED_SIZE * WORD)
  895. self.mc.MOV_sr(PASS_ON_MY_FRAME * WORD, ebp.value)
  896. if IS_X86_64:
  897. self.mc.MOV_sr(THREADLOCAL_OFS, esi.value)
  898. if self.cpu.translate_support_code:
  899. self._call_header_vmprof() # on X86_64, this uses esi
  900. if IS_X86_64:
  901. self.mc.MOV_rr(ebp.value, edi.value)
  902. else:
  903. self.mc.MOV_rs(ebp.value, (FRAME_FIXED_SIZE + 1) * WORD)
  904. for i, loc in enumerate(self.cpu.CALLEE_SAVE_REGISTERS):
  905. self.mc.MOV_sr((PASS_ON_MY_FRAME + i + 1) * WORD, loc.value)
  906. gcrootmap = self.cpu.gc_ll_descr.gcrootmap
  907. if gcrootmap and gcrootmap.is_shadow_stack:
  908. self._call_header_shadowstack(gcrootmap)
  909. def _call_header_with_stack_check(self):
  910. self._call_header()
  911. if self.stack_check_slowpath == 0:
  912. pass # no stack check (e.g. not translated)
  913. else:
  914. endaddr, lengthaddr, _ = self.cpu.insert_stack_check()
  915. self.mc.MOV(eax, heap(endaddr)) # MOV eax, [start]
  916. self.mc.SUB(eax, esp) # SUB eax, current
  917. self.mc.CMP(eax, heap(lengthaddr)) # CMP eax, [length]
  918. self.mc.J_il8(rx86.Conditions['BE'], 0) # JBE .skip
  919. jb_location = self.mc.get_relative_pos()
  920. self.mc.CALL(imm(self.stack_check_slowpath))# CALL slowpath
  921. # patch the JB above # .skip:
  922. offset = self.mc.get_relative_pos() - jb_location
  923. assert 0 < offset <= 127
  924. self.mc.overwrite(jb_location-1, chr(offset))
  925. #
  926. def _call_footer(self):
  927. # the return value is the jitframe
  928. if self.cpu.translate_support_code:
  929. self._call_footer_vmprof()
  930. self.mc.MOV_rr(eax.value, ebp.value)
  931. gcrootmap = self.cpu.gc_ll_descr.gcrootmap
  932. if gcrootmap and gcrootmap.is_shadow_stack:
  933. self._call_footer_shadowstack(gcrootmap)
  934. for i in range(len(self.cpu.CALLEE_SAVE_REGISTERS)-1, -1, -1):
  935. self.mc.MOV_rs(self.cpu.CALLEE_SAVE_REGISTERS[i].value,
  936. (i + 1 + PASS_ON_MY_FRAME) * WORD)
  937. self.mc.MOV_rs(ebp.value, PASS_ON_MY_FRAME * WORD)
  938. self.mc.ADD_ri(esp.value, FRAME_FIXED_SIZE * WORD)
  939. self.mc.RET()
  940. def _load_shadowstack_top_in_ebx(self, mc, gcrootmap):
  941. """Loads the shadowstack top in ebx, and returns an integer
  942. that gives the address of the stack top. If this integer doesn't
  943. fit in 32 bits, it will be loaded in r11.
  944. """
  945. rst = gcrootmap.get_root_stack_top_addr()
  946. if rx86.fits_in_32bits(rst):
  947. mc.MOV_rj(ebx.value, rst) # MOV ebx, [rootstacktop]
  948. else:
  949. mc.MOV_ri(X86_64_SCRATCH_REG.value, rst) # MOV r11, rootstacktop
  950. mc.MOV_rm(ebx.value, (X86_64_SCRATCH_REG.value, 0))
  951. # MOV ebx, [r11]
  952. #
  953. return rst
  954. def _call_header_shadowstack(self, gcrootmap):
  955. rst = self._load_shadowstack_top_in_ebx(self.mc, gcrootmap)
  956. self.mc.MOV_mr((ebx.value, 0), ebp.value) # MOV [ebx], ebp
  957. self.mc.ADD_ri(ebx.value, WORD)
  958. if rx86.fits_in_32bits(rst):
  959. self.mc.MOV_jr(rst, ebx.value) # MOV [rootstacktop], ebx
  960. else:
  961. # The integer 'rst' doesn't fit in 32 bits, so we know that
  962. # _load_shadowstack_top_in_ebx() above loaded it in r11.
  963. # Reuse it. Be careful not to overwrite r11 in the middle!
  964. self.mc.MOV_mr((X86_64_SCRATCH_REG.value, 0),
  965. ebx.value) # MOV [r11], ebx
  966. def _call_footer_shadowstack(self, gcrootmap):
  967. rst = gcrootmap.get_root_stack_top_addr()
  968. if rx86.fits_in_32bits(rst):
  969. self.mc.SUB_ji8(rst, WORD) # SUB [rootstacktop], WORD
  970. else:
  971. self.mc.MOV_ri(ebx.value, rst) # MOV ebx, rootstacktop
  972. self.mc.SUB_mi8((ebx.value, 0), WORD) # SUB [ebx], WORD
  973. def redirect_call_assembler(self, oldlooptoken, newlooptoken):
  974. # some minimal sanity checking
  975. old_nbargs = oldlooptoken.compiled_loop_token._debug_nbargs
  976. new_nbargs = newlooptoken.compiled_loop_token._debug_nbargs
  977. assert old_nbargs == new_nbargs
  978. # we overwrite the instructions at the old _ll_function_addr
  979. # to start with a JMP to the new _ll_function_addr.
  980. # Ideally we should rather patch all existing CALLs, but well.
  981. oldadr = oldlooptoken._ll_function_addr
  982. target = newlooptoken._ll_function_addr
  983. # copy frame-info data
  984. baseofs = self.cpu.get_baseofs_of_frame_field()
  985. newlooptoken.compiled_loop_token.update_frame_info(
  986. oldlooptoken.compiled_loop_token, baseofs)
  987. mc = codebuf.MachineCodeBlockWrapper()
  988. mc.JMP(imm(follow_jump(target)))
  989. if WORD == 4: # keep in sync with prepare_loop()
  990. assert mc.get_relative_pos() == 5
  991. else:
  992. assert mc.get_relative_pos() <= 13
  993. mc.copy_to_raw_memory(oldadr)
  994. # log the redirection of the call_assembler_* operation
  995. jl.redirect_assembler(oldlooptoken, newlooptoken, target)
  996. def dump(self, text):
  997. if not self.verbose:
  998. return
  999. pos = self.mc.get_relative_pos()
  1000. print >> sys.stderr, ' 0x%x %s' % (pos, text)
  1001. # ------------------------------------------------------------
  1002. def mov(self, from_loc, to_loc):
  1003. from_xmm = isinstance(from_loc, RegLoc) and from_loc.is_xmm
  1004. to_xmm = isinstance(to_loc, RegLoc) and to_loc.is_xmm
  1005. if from_xmm or to_xmm:
  1006. if from_xmm and to_xmm:
  1007. # copy 128-bit from -> to
  1008. self.mc.MOVAPD(to_loc, from_loc)
  1009. else:
  1010. self.mc.MOVSD(to_loc, from_loc)
  1011. else:
  1012. assert to_loc is not ebp
  1013. self.mc.MOV(to_loc, from_loc)
  1014. regalloc_mov = mov # legacy interface
  1015. def regalloc_push(self, loc):
  1016. if isinstance(loc, RegLoc) and loc.is_xmm:
  1017. self.mc.SUB_ri(esp.value, 8) # = size of doubles
  1018. self.mc.MOVSD_sx(0, loc.value)
  1019. elif WORD == 4 and isinstance(loc, FrameLoc) and loc.get_width() == 8:
  1020. # XXX evil trick
  1021. self.mc.PUSH_b(loc.value + 4)
  1022. self.mc.PUSH_b(loc.value)
  1023. else:
  1024. self.mc.PUSH(loc)
  1025. def regalloc_pop(self, loc):
  1026. if isinstance(loc, RegLoc) and loc.is_xmm:
  1027. self.mc.MOVSD_xs(loc.value, 0)
  1028. self.mc.ADD_ri(esp.value, 8) # = size of doubles
  1029. elif WORD == 4 and isinstance(loc, FrameLoc) and loc.get_width() == 8:
  1030. # XXX evil trick
  1031. self.mc.POP_b(loc.value)
  1032. self.mc.POP_b(loc.value + 4)
  1033. else:
  1034. self.mc.POP(loc)
  1035. def regalloc_immedmem2mem(self, from_loc, to_loc):
  1036. # move a ConstFloatLoc directly to a FrameLoc, as two MOVs
  1037. # (even on x86-64, because the immediates are encoded as 32 bits)
  1038. assert isinstance(from_loc, ConstFloatLoc)
  1039. low_part = rffi.cast(rffi.CArrayPtr(rffi.INT), from_loc.value)[0]
  1040. high_part = rffi.cast(rffi.CArrayPtr(rffi.INT), from_loc.value)[1]
  1041. low_part = intmask(low_part)
  1042. high_part = intmask(high_part)
  1043. if isinstance(to_loc, RawEbpLoc):
  1044. self.mc.MOV32_bi(to_loc.value, low_part)
  1045. self.mc.MOV32_bi(to_loc.value + 4, high_part)
  1046. else:
  1047. assert isinstance(to_loc, RawEspLoc)
  1048. self.mc.MOV32_si(to_loc.value, low_part)
  1049. self.mc.MOV32_si(to_loc.value + 4, high_part)
  1050. def regalloc_perform(self, op, arglocs, resloc):
  1051. genop_list[op.getopnum()](self, op, arglocs, resloc)
  1052. def regalloc_perform_discard(self, op, arglocs):
  1053. genop_discard_list[op.getopnum()](self, op, arglocs)
  1054. def regalloc_perform_llong(self, op, arglocs, resloc):
  1055. effectinfo = op.getdescr().get_extra_info()
  1056. oopspecindex = effectinfo.oopspecindex
  1057. genop_llong_list[oopspecindex](self, op, arglocs, resloc)
  1058. def regalloc_perform_math(self, op, arglocs, resloc):
  1059. effectinfo = op.getdescr().get_extra_info()
  1060. oopspecindex = effectinfo.oopspecindex
  1061. genop_math_list[oopspecindex](self, op, arglocs, resloc)
  1062. def regalloc_perform_guard(self, guard_op, faillocs, arglocs, resloc,
  1063. frame_depth):
  1064. faildescr = guard_op.getdescr()
  1065. assert isinstance(faildescr, AbstractFailDescr)
  1066. failargs = guard_op.getfailargs()
  1067. guard_opnum = guard_op.getopnum()
  1068. guard_token = self.implement_guard_recovery(guard_opnum,
  1069. faildescr, failargs,
  1070. faillocs, frame_depth)
  1071. genop_guard_list[guard_opnum](self, guard_op, guard_token,
  1072. arglocs, resloc)
  1073. if not we_are_translated():
  1074. # must be added by the genop_guard_list[]()
  1075. assert guard_token is self.pending_guard_tokens[-1]
  1076. def load_effective_addr(self, sizereg, baseofs, scale, result, frm=imm0):
  1077. self.mc.LEA(result, addr_add(frm, sizereg, baseofs, scale))
  1078. def _unaryop(asmop):
  1079. def genop_unary(self, op, arglocs, resloc):
  1080. getattr(self.mc, asmop)(arglocs[0])
  1081. return genop_unary
  1082. def _binaryop(asmop):
  1083. def genop_binary(self, op, arglocs, result_loc):
  1084. getattr(self.mc, asmop)(arglocs[0], arglocs[1])
  1085. return genop_binary
  1086. def _binaryop_or_lea(asmop, is_add):
  1087. def genop_binary_or_lea(self, op, arglocs, result_loc):
  1088. # use a regular ADD or SUB if result_loc is arglocs[0],
  1089. # and a LEA only if different.
  1090. if result_loc is arglocs[0]:
  1091. getattr(self.mc, asmop)(arglocs[0], arglocs[1])
  1092. else:
  1093. loc = arglocs[0]
  1094. argloc = arglocs[1]
  1095. assert isinstance(loc, RegLoc)
  1096. assert isinstance(argloc, ImmedLoc)
  1097. assert isinstance(result_loc, RegLoc)
  1098. delta = argloc.value
  1099. if not is_add: # subtraction
  1100. delta = -delta
  1101. self.mc.LEA_rm(result_loc.value, (loc.value, delta))
  1102. return genop_binary_or_lea
  1103. def flush_cc(self, cond, result_loc):
  1104. # After emitting a instruction that leaves a boolean result in
  1105. # a condition code (cc), call this. In the common case, result_loc
  1106. # will be set to ebp by the regalloc, which in this case means
  1107. # "propagate it between this operation and the next guard by keeping
  1108. # it in the cc". In the uncommon case, result_loc is another
  1109. # register, and we emit a load from the cc into this register.
  1110. assert self.guard_success_cc == rx86.cond_none
  1111. if result_loc is ebp:
  1112. self.guard_success_cc = cond
  1113. else:
  1114. self.mc.MOV_ri(result_loc.value, 0)
  1115. rl = result_loc.lowest8bits()
  1116. self.mc.SET_ir(cond, rl.value)
  1117. def _cmpop(cond, rev_cond):
  1118. cond = rx86.Conditions[cond]
  1119. rev_cond = rx86.Conditions[rev_cond]
  1120. #
  1121. def genop_cmp(self, op, arglocs, result_loc):
  1122. if isinstance(op.getarg(0), Const):
  1123. self.mc.CMP(arglocs[1], arglocs[0])
  1124. self.flush_cc(rev_cond, result_loc)
  1125. else:
  1126. self.mc.CMP(arglocs[0], arglocs[1])
  1127. self.flush_cc(cond, result_loc)
  1128. return genop_cmp
  1129. def _if_parity_clear_zero_and_carry(self):
  1130. self.mc.J_il8(rx86.Conditions['NP'], 0)
  1131. jnp_location = self.mc.get_relative_pos()
  1132. # CMP EBP, 0: as EBP cannot be null here, that operation should
  1133. # always clear zero and carry
  1134. self.mc.CMP_ri(ebp.value, 0)
  1135. # patch the JNP above
  1136. offset = self.mc.get_relative_pos() - jnp_location
  1137. assert 0 < offset <= 127
  1138. self.mc.overwrite(jnp_location-1, chr(offset))
  1139. def _cmpop_float(cond, rev_cond):
  1140. is_ne = cond == 'NE'
  1141. need_direct_p = 'A' not in cond
  1142. need_rev_p = 'A' not in rev_cond
  1143. cond_contains_e = ('E' in cond) ^ ('N' in cond)
  1144. cond = rx86.Conditions[cond]
  1145. rev_cond = rx86.Conditions[rev_cond]
  1146. #
  1147. def genop_cmp_float(self, op, arglocs, result_loc):
  1148. if need_direct_p:
  1149. direct_case = not isinstance(arglocs[1], RegLoc)
  1150. else:
  1151. direct_case = isinstance(arglocs[0], RegLoc)
  1152. if direct_case:
  1153. self.mc.UCOMISD(arglocs[0], arglocs[1])
  1154. checkcond = cond
  1155. need_p = need_direct_p
  1156. else:
  1157. self.mc.UCOMISD(arglocs[1], arglocs[0])
  1158. checkcond = rev_cond
  1159. need_p = need_rev_p
  1160. if need_p:
  1161. self._if_parity_clear_zero_and_carry()
  1162. self.flush_cc(checkcond, result_loc)
  1163. return genop_cmp_float
  1164. def simple_call(self, fnloc, arglocs, result_loc=eax):
  1165. if result_loc is xmm0:
  1166. result_type = FLOAT
  1167. result_size = 8
  1168. elif result_loc is None:
  1169. result_type = VOID
  1170. result_size = 0
  1171. else:
  1172. result_type = INT
  1173. result_size = WORD
  1174. cb = callbuilder.CallBuilder(self, fnloc, arglocs,
  1175. result_loc, result_type,
  1176. result_size)
  1177. cb.emit()
  1178. def simple_call_no_collect(self, fnloc, arglocs):
  1179. cb = callbuilder.CallBuilder(self, fnloc, arglocs)
  1180. cb.emit_no_collect()
  1181. def _reload_frame_if_necessary(self, mc, shadowstack_reg=None):
  1182. gcrootmap = self.cpu.gc_ll_descr.gcrootmap
  1183. if gcrootmap:
  1184. if gcrootmap.is_shadow_stack:
  1185. if shadowstack_reg is None:
  1186. rst = gcrootmap.get_root_stack_top_addr()
  1187. mc.MOV(ecx, heap(rst))
  1188. shadowstack_reg = ecx
  1189. mc.MOV(ebp, mem(shadowstack_reg, -WORD))
  1190. wbdescr = self.cpu.gc_ll_descr.write_barrier_descr
  1191. if gcrootmap and wbdescr:
  1192. # frame never uses card marking, so we enforce this is not
  1193. # an array
  1194. self._write_barrier_fastpath(mc, wbdescr, [ebp], array=False,
  1195. is_frame=True)
  1196. genop_int_neg = _unaryop("NEG")
  1197. genop_int_invert = _unaryop("NOT")
  1198. genop_int_add = _binaryop_or_lea("ADD", is_add=True)
  1199. genop_nursery_ptr_increment = _binaryop_or_lea('ADD', is_add=True)
  1200. genop_int_sub = _binaryop_or_lea("SUB", is_add=False)
  1201. genop_int_mul = _binaryop("IMUL")
  1202. genop_int_or = _binaryop("OR")
  1203. genop_int_xor = _binaryop("XOR")
  1204. genop_int_lshift = _binaryop("SHL")
  1205. genop_int_rshift = _binaryop("SAR")
  1206. genop_uint_rshift = _binaryop("SHR")
  1207. genop_float_add = _binaryop("ADDSD")
  1208. genop_float_sub = _binaryop('SUBSD')
  1209. genop_float_mul = _binaryop('MULSD')
  1210. genop_float_truediv = _binaryop('DIVSD')
  1211. def genop_uint_mul_high(self, op, arglocs, result_loc):
  1212. self.mc.MUL(arglocs[0])
  1213. def genop_int_and(self, op, arglocs, result_loc):
  1214. arg1 = arglocs[1]
  1215. if IS_X86_64 and (isinstance(arg1, ImmedLoc) and
  1216. arg1.value == (1 << 32) - 1):
  1217. # special case
  1218. self.mc.MOV32(arglocs[0], arglocs[0])
  1219. else:
  1220. self.mc.AND(arglocs[0], arg1)
  1221. genop_int_lt = _cmpop("L", "G")
  1222. genop_int_le = _cmpop("LE", "GE")
  1223. genop_int_eq = _cmpop("E", "E")
  1224. genop_int_ne = _cmpop("NE", "NE")
  1225. genop_int_gt = _cmpop("G", "L")
  1226. genop_int_ge = _cmpop("GE", "LE")
  1227. genop_ptr_eq = genop_instance_ptr_eq = genop_int_eq
  1228. genop_ptr_ne = genop_instance_ptr_ne = genop_int_ne
  1229. genop_uint_gt = _cmpop("A", "B")
  1230. genop_uint_lt = _cmpop("B", "A")
  1231. genop_uint_le = _cmpop("BE", "AE")
  1232. genop_uint_ge = _cmpop("AE", "BE")
  1233. genop_float_lt = _cmpop_float("B", "A")
  1234. genop_float_le = _cmpop_float("BE","AE")
  1235. genop_float_eq = _cmpop_float("E", "E")
  1236. genop_float_ne = _cmpop_float("NE", "NE")
  1237. genop_float_gt = _cmpop_float("A", "B")
  1238. genop_float_ge = _cmpop_float("AE","BE")
  1239. def genop_math_sqrt(self, op, arglocs, resloc):
  1240. self.mc.SQRTSD(arglocs[0], resloc)
  1241. def genop_int_signext(self, op, arglocs, resloc):
  1242. argloc, numbytesloc = arglocs
  1243. assert isinstance(numbytesloc, ImmedLoc)
  1244. assert isinstance(resloc, RegLoc)
  1245. if numbytesloc.value == 1:
  1246. if isinstance(argloc, RegLoc):
  1247. if WORD == 4 and argloc.value >= 4:
  1248. # meh, can't read the lowest byte of esi or edi on 32-bit
  1249. if resloc is not argloc:
  1250. self.mc.MOV(resloc, argloc)
  1251. argloc = resloc
  1252. if resloc.value >= 4:
  1253. # still annoyed, hack needed
  1254. self.mc.SHL_ri(resloc.value, 24)
  1255. self.mc.SAR_ri(resloc.value, 24)
  1256. return
  1257. argloc = argloc.lowest8bits()
  1258. self.mc.MOVSX8(resloc, argloc)
  1259. elif numbytesloc.value == 2:
  1260. self.mc.MOVSX16(resloc, argloc)
  1261. elif IS_X86_64 and numbytesloc.value == 4:
  1262. self.mc.MOVSX32(resloc, argloc)
  1263. else:
  1264. raise AssertionError("bad number of bytes")
  1265. def genop_float_neg(self, op, arglocs, resloc):
  1266. # Following what gcc does: res = x ^ 0x8000000000000000
  1267. self.mc.XORPD(arglocs[0], heap(self.float_const_neg_addr))
  1268. def genop_float_abs(self, op, arglocs, resloc):
  1269. # Following what gcc does: res = x & 0x7FFFFFFFFFFFFFFF
  1270. self.mc.ANDPD(arglocs[0], heap(self.float_const_abs_addr))
  1271. def genop_cast_float_to_int(self, op, arglocs, resloc):
  1272. self.mc.CVTTSD2SI(resloc, arglocs[0])
  1273. def genop_cast_int_to_float(self, op, arglocs, resloc):
  1274. self.mc.CVTSI2SD(resloc, arglocs[0])
  1275. def genop_cast_float_to_singlefloat(self, op, arglocs, resloc):
  1276. loc0, loctmp = arglocs
  1277. self.mc.CVTSD2SS(loctmp, loc0)
  1278. assert isinstance(resloc, RegLoc)
  1279. assert isinstance(loctmp, RegLoc)
  1280. self.mc.MOVD32_rx(resloc.value, loctmp.value)
  1281. def genop_cast_singlefloat_to_float(self, op, arglocs, resloc):
  1282. loc0, = arglocs
  1283. assert isinstance(resloc, RegLoc)
  1284. assert isinstance(loc0, RegLoc)
  1285. self.mc.MOVD32_xr(resloc.value, loc0.value)
  1286. self.mc.CVTSS2SD_xx(resloc.value, resloc.value)
  1287. def genop_convert_float_bytes_to_longlong(self, op, arglocs, resloc):
  1288. loc0, = arglocs
  1289. if longlong.is_64_bit:
  1290. assert isinstance(resloc, RegLoc)
  1291. assert isinstance(loc0, RegLoc)
  1292. self.mc.MOVDQ(resloc, loc0)
  1293. else:
  1294. self.mov(loc0, resloc)
  1295. def genop_convert_longlong_bytes_to_float(self, op, arglocs, resloc):
  1296. loc0, = arglocs
  1297. if longlong.is_64_bit:
  1298. assert isinstance(resloc, RegLoc)
  1299. assert isinstance(loc0, RegLoc)
  1300. self.mc.MOVDQ(resloc, loc0)
  1301. else:
  1302. self.mov(loc0, resloc)
  1303. def test_location(self, loc):
  1304. assert not isinstance(loc, ImmedLoc)
  1305. if isinstance(loc, RegLoc):
  1306. self.mc.TEST_rr(loc.value, loc.value) # more compact
  1307. else:
  1308. self.mc.CMP(loc, imm0) # works from memory too
  1309. def genop_int_is_true(self, op, arglocs, resloc):
  1310. self.test_location(arglocs[0])
  1311. self.flush_cc(rx86.Conditions['NZ'], resloc)
  1312. def genop_int_is_zero(self, op, arglocs, resloc):
  1313. self.test_location(arglocs[0])
  1314. self.flush_cc(rx86.Conditions['Z'], resloc)
  1315. def _genop_same_as(self, op, arglocs, resloc):
  1316. self.mov(arglocs[0], resloc)
  1317. genop_same_as_i = _genop_same_as
  1318. genop_same_as_r = _genop_same_as
  1319. genop_same_as_f = _genop_same_as
  1320. genop_cast_ptr_to_int = _genop_same_as
  1321. genop_cast_int_to_ptr = _genop_same_as
  1322. def _patch_load_from_gc_table(self, index):
  1323. # must be called immediately after a "p"-mode instruction
  1324. # has been emitted. 64-bit mode only.
  1325. assert IS_X86_64
  1326. address_in_buffer = index * WORD # at the start of the buffer
  1327. p_location = self.mc.get_relative_pos()
  1328. offset = address_in_buffer - p_location
  1329. self.mc.overwrite32(p_location-4, offset)
  1330. def _addr_from_gc_table(self, index):
  1331. # get the address of the gc table entry 'index'. 32-bit mode only.
  1332. assert IS_X86_32
  1333. return self.gc_table_addr + index * WORD
  1334. def genop_load_from_gc_table(self, op, arglocs, resloc):
  1335. index = op.getarg(0).getint()
  1336. assert isinstance(resloc, RegLoc)
  1337. if IS_X86_64:
  1338. self.mc.MOV_rp(resloc.value, 0) # %rip-relative
  1339. self._patch_load_from_gc_table(index)
  1340. elif IS_X86_32:
  1341. self.mc.MOV_rj(resloc.value, self._addr_from_gc_table(index))
  1342. def genop_int_force_ge_zero(self, op, arglocs, resloc):
  1343. self.mc.TEST(arglocs[0], arglocs[0])
  1344. self.mov(imm0, resloc)
  1345. self.mc.CMOVNS(resloc, arglocs[0])
  1346. genop_llong_add = _binaryop("PADDQ")
  1347. genop_llong_sub = _binaryop("PSUBQ")
  1348. genop_llong_and = _binaryop("PAND")
  1349. genop_llong_or = _binaryop("POR")
  1350. genop_llong_xor = _binaryop("PXOR")
  1351. def genop_llong_to_int(self, op, arglocs, resloc):
  1352. loc = arglocs[0]
  1353. assert isinstance(resloc, RegLoc)
  1354. if isinstance(loc, RegLoc):
  1355. self.mc.MOVD32_rx(resloc.value, loc.value)
  1356. elif isinstance(loc, FrameLoc):
  1357. self.mc.MOV_rb(resloc.value, loc.value)
  1358. else:
  1359. not_implemented("llong_to_int: %s" % (loc,))
  1360. def genop_llong_from_int(self, op, arglocs, resloc):
  1361. loc1, loc2 = arglocs
  1362. if isinstance(loc1, ConstFloatLoc):
  1363. assert loc2 is None
  1364. self.mc.MOVSD(resloc, loc1)
  1365. else:
  1366. assert isinstance(loc1, RegLoc)
  1367. assert isinstance(loc2, RegLoc)
  1368. assert isinstance(resloc, RegLoc)
  1369. self.mc.MOVD32_xr(loc2.value, loc1.value)
  1370. self.mc.PSRAD_xi(loc2.value, 31) # -> 0 or -1
  1371. self.mc.MOVD32_xr(resloc.value, loc1.value)
  1372. self.mc.PUNPCKLDQ_xx(resloc.value, loc2.value)
  1373. def genop_llong_from_uint(self, op, arglocs, resloc):
  1374. loc1, = arglocs
  1375. assert isinstance(resloc, RegLoc)
  1376. assert isinstance(loc1, RegLoc)
  1377. self.mc.MOVD32_xr(resloc.value, loc1.value) # zero-extending
  1378. def genop_llong_eq(self, op, arglocs, resloc):
  1379. loc1, loc2, locxtmp = arglocs
  1380. self.mc.MOVSD(locxtmp, loc1)
  1381. self.mc.PCMPEQD(locxtmp, loc2)
  1382. self.mc.PMOVMSKB_rx(resloc.value, locxtmp.value)
  1383. # Now the lower 8 bits of resloc contain 0x00, 0x0F, 0xF0 or 0xFF
  1384. # depending on the result of the comparison of each of the two
  1385. # double-words of loc1 and loc2. The higher 8 bits contain random
  1386. # results. We want to map 0xFF to 1, and 0x00, 0x0F and 0xF0 to 0.
  1387. self.mc.CMP8_ri(resloc.value | rx86.BYTE_REG_FLAG, -1)
  1388. self.mc.SBB_rr(resloc.value, resloc.value)
  1389. self.mc.ADD_ri(resloc.value, 1)
  1390. def genop_llong_ne(self, op, arglocs, resloc):
  1391. loc1, loc2, locxtmp = arglocs
  1392. self.mc.MOVSD(locxtmp, loc1)
  1393. self.mc.PCMPEQD(locxtmp, loc2)
  1394. self.mc.PMOVMSKB_rx(resloc.value, locxtmp.value)
  1395. # Now the lower 8 bits of resloc contain 0x00, 0x0F, 0xF0 or 0xFF
  1396. # depending on the result of the comparison of each of the two
  1397. # double-words of loc1 and loc2. The higher 8 bits contain random
  1398. # results. We want to map 0xFF to 0, and 0x00, 0x0F and 0xF0 to 1.
  1399. self.mc.CMP8_ri(resloc.value | rx86.BYTE_REG_FLAG, -1)
  1400. self.mc.SBB_rr(resloc.value, resloc.value)
  1401. self.mc.NEG_r(resloc.value)
  1402. def genop_llong_lt(self, op, arglocs, resloc):
  1403. # XXX just a special case for now: "x < 0"
  1404. loc1, = arglocs
  1405. self.mc.PMOVMSKB_rx(resloc.value, loc1.value)
  1406. self.mc.SHR_ri(resloc.value, 7)
  1407. self.mc.AND_ri(resloc.value, 1)
  1408. # ----------
  1409. def genop_discard_check_memory_error(self, op, arglocs):
  1410. reg = arglocs[0]
  1411. self.mc.TEST(reg, reg)
  1412. if WORD == 4:
  1413. self.mc.J_il(rx86.Conditions['Z'], self.propagate_exception_path)
  1414. self.mc.add_pending_relocation()
  1415. elif WORD == 8:
  1416. self.mc.J_il(rx86.Conditions['Z'], 0)
  1417. pos = self.mc.get_relative_pos()
  1418. self.pending_memoryerror_trampoline_from.append(pos)
  1419. # ----------
  1420. def load_from_mem(self, resloc, source_addr, size_loc, sign_loc):
  1421. assert isinstance(resloc, RegLoc)
  1422. size = size_loc.value
  1423. sign = sign_loc.value
  1424. if resloc.is_xmm:
  1425. self.mc.MOVSD(resloc, source_addr)
  1426. elif size == WORD:
  1427. self.mc.MOV(resloc, source_addr)
  1428. elif size == 1:
  1429. if sign:
  1430. self.mc.MOVSX8(resloc, source_addr)
  1431. else:
  1432. self.mc.MOVZX8(resloc, source_addr)
  1433. elif size == 2:
  1434. if sign:
  1435. self.mc.MOVSX16(resloc, source_addr)
  1436. else:
  1437. self.mc.MOVZX16(resloc, source_addr)
  1438. elif IS_X86_64 and size == 4:
  1439. if sign:
  1440. self.mc.MOVSX32(resloc, source_addr)
  1441. else:
  1442. self.mc.MOV32(resloc, source_addr) # zero-extending
  1443. else:
  1444. not_implemented("load_from_mem size = %d" % size)
  1445. def save_into_mem(self, dest_addr, value_loc, size_loc):
  1446. size = size_loc.value
  1447. if isinstance(value_loc, RegLoc) and value_loc.is_xmm:
  1448. self.mc.MOVSD(dest_addr, value_loc)
  1449. elif size == 1:
  1450. self.mc.MOV8(dest_addr, value_loc.lowest8bits())
  1451. elif size == 2:
  1452. self.mc.MOV16(dest_addr, value_loc)
  1453. elif size == 4:
  1454. self.mc.MOV32(dest_addr, value_loc)
  1455. elif size == 8:
  1456. if IS_X86_64:
  1457. self.mc.MOV(dest_addr, value_loc)
  1458. else:
  1459. assert isinstance(value_loc, FloatImmedLoc)
  1460. self.mc.MOV(dest_addr, value_loc.low_part_loc())
  1461. self.mc.MOV(dest_addr.add_offset(4), value_loc.high_part_loc())
  1462. else:
  1463. not_implemented("save_into_mem size = %d" % size)
  1464. def _genop_getfield(self, op, arglocs, resloc):
  1465. base_loc, ofs_loc, size_loc, sign_loc = arglocs
  1466. assert isinstance(size_loc, ImmedLoc)
  1467. source_addr = AddressLoc(base_loc, ofs_loc)
  1468. self.load_from_mem(resloc, source_addr, size_loc, sign_loc)
  1469. genop_getfield_gc_i = _genop_getfield
  1470. genop_getfield_gc_r = _genop_getfield
  1471. genop_getfield_gc_f = _genop_getfield
  1472. genop_getfield_raw_i = _genop_getfield
  1473. genop_getfield_raw_f = _genop_getfield
  1474. def _genop_gc_load(self, op, arglocs, resloc):
  1475. base_loc, ofs_loc, size_loc, sign_loc = arglocs
  1476. assert isinstance(size_loc, ImmedLoc)
  1477. src_addr = addr_add(base_loc, ofs_loc, 0, 0)
  1478. self.load_from_mem(resloc, src_addr, size_loc, sign_loc)
  1479. genop_gc_load_i = _genop_gc_load
  1480. genop_gc_load_r = _genop_gc_load
  1481. genop_gc_load_f = _genop_gc_load
  1482. def _genop_gc_load_indexed(self, op, arglocs, resloc):
  1483. base_loc, ofs_loc, scale_loc, offset_loc, size_loc, sign_loc = arglocs
  1484. assert isinstance(scale_loc, ImmedLoc)
  1485. scale = get_scale(scale_loc.value)
  1486. src_addr = addr_add(base_loc, ofs_loc, offset_loc.value, scale)
  1487. self.load_from_mem(resloc, src_addr, size_loc, sign_loc)
  1488. genop_gc_load_indexed_i = _genop_gc_load_indexed
  1489. genop_gc_load_indexed_r = _genop_gc_load_indexed
  1490. genop_gc_load_indexed_f = _genop_gc_load_indexed
  1491. def _imul_const_scaled(self, mc, targetreg, sourcereg, itemsize):
  1492. """Produce one operation to do roughly
  1493. targetreg = sourcereg * itemsize
  1494. except that the targetreg may still need shifting by 0,1,2,3.
  1495. """
  1496. if (itemsize & 7) == 0:
  1497. shift = 3
  1498. elif (itemsize & 3) == 0:
  1499. shift = 2
  1500. elif (itemsize & 1) == 0:
  1501. shift = 1
  1502. else:
  1503. shift = 0
  1504. itemsize >>= shift
  1505. #
  1506. if valid_addressing_size(itemsize - 1):
  1507. mc.LEA_ra(targetreg, (sourcereg, sourcereg,
  1508. get_scale(itemsize - 1), 0))
  1509. elif valid_addressing_size(itemsize):
  1510. mc.LEA_ra(targetreg, (rx86.NO_BASE_REGISTER, sourcereg,
  1511. get_scale(itemsize), 0))
  1512. else:
  1513. mc.IMUL_rri(targetreg, sourcereg, itemsize)
  1514. #
  1515. return shift
  1516. def genop_discard_increment_debug_counter(self, op, arglocs):
  1517. # The argument should be an immediate address. This should
  1518. # generate code equivalent to a GETFIELD_RAW, an ADD(1), and a
  1519. # SETFIELD_RAW. Here we use the direct from-memory-to-memory
  1520. # increment operation of x86.
  1521. base_loc, = arglocs
  1522. self.mc.INC(mem(base_loc, 0))
  1523. def genop_discard_gc_store(self, op, arglocs):
  1524. base_loc, ofs_loc, value_loc, size_loc = arglocs
  1525. assert isinstance(size_loc, ImmedLoc)
  1526. scale = get_scale(size_loc.value)
  1527. dest_addr = AddressLoc(base_loc, ofs_loc, 0, 0)
  1528. self.save_into_mem(dest_addr, value_loc, size_loc)
  1529. def genop_discard_gc_store_indexed(self, op, arglocs):
  1530. base_loc, ofs_loc, value_loc, factor_loc, offset_loc, size_loc = arglocs
  1531. assert isinstance(size_loc, ImmedLoc)
  1532. scale = get_scale(factor_loc.value)
  1533. dest_addr = AddressLoc(base_loc, ofs_loc, scale, offset_loc.value)
  1534. self.save_into_mem(dest_addr, value_loc, size_loc)
  1535. # genop_discard_setfield_raw = genop_discard_setfield_gc
  1536. def genop_math_read_timestamp(self, op, arglocs, resloc):
  1537. self.mc.RDTSC()
  1538. if longlong.is_64_bit:
  1539. self.mc.SHL_ri(edx.value, 32)
  1540. self.mc.OR_rr(edx.value, eax.value)
  1541. else:
  1542. loc1, = arglocs
  1543. self.mc.MOVD32_xr(loc1.value, edx.value)
  1544. self.mc.MOVD32_xr(resloc.value, eax.value)
  1545. self.mc.PUNPCKLDQ_xx(resloc.value, loc1.value)
  1546. def genop_guard_guard_true(self, guard_op, guard_token, locs, resloc):
  1547. self.implement_guard(guard_token)
  1548. genop_guard_guard_nonnull = genop_guard_guard_true
  1549. def genop_guard_guard_false(self, guard_op, guard_token, locs, resloc):
  1550. self.guard_success_cc = rx86.invert_condition(self.guard_success_cc)
  1551. self.implement_guard(guard_token)
  1552. genop_guard_guard_isnull = genop_guard_guard_false
  1553. def genop_guard_guard_no_exception(self, guard_op, guard_token, locs, ign):
  1554. self.mc.CMP(heap(self.cpu.pos_exception()), imm0)
  1555. self.guard_success_cc = rx86.Conditions['Z']
  1556. self.implement_guard(guard_token)
  1557. # If the previous operation was a COND_CALL, overwrite its conditional
  1558. # jump to jump over this GUARD_NO_EXCEPTION as well, if we can
  1559. if self._find_nearby_operation(-1).getopnum() in (
  1560. rop.COND_CALL, rop.COND_CALL_VALUE_I, rop.COND_CALL_VALUE_R):
  1561. jmp_adr = self.previous_cond_call_jcond
  1562. offset = self.mc.get_relative_pos() - jmp_adr
  1563. if offset <= 127:
  1564. self.mc.overwrite(jmp_adr-1, chr(offset))
  1565. def genop_guard_guard_not_invalidated(self, guard_op, guard_token,
  1566. locs, ign):
  1567. pos = self.mc.get_relative_pos() + 1 # after potential jmp
  1568. guard_token.pos_jump_offset = pos
  1569. self.pending_guard_tokens.append(guard_token)
  1570. def genop_guard_guard_exception(self, guard_op, guard_token, locs, resloc):
  1571. loc = locs[0]
  1572. loc1 = locs[1]
  1573. self.mc.MOV(loc1, heap(self.cpu.pos_exception()))
  1574. self.mc.CMP(loc1, loc)
  1575. self.guard_success_cc = rx86.Conditions['E']
  1576. self.implement_guard(guard_token)
  1577. self._store_and_reset_exception(self.mc, resloc)
  1578. def genop_save_exc_class(self, op, arglocs, resloc):
  1579. self.mc.MOV(resloc, heap(self.cpu.pos_exception()))
  1580. def genop_save_exception(self, op, arglocs, resloc):
  1581. self._store_and_reset_exception(self.mc, resloc)
  1582. def genop_discard_restore_exception(self, op, arglocs):
  1583. self._restore_exception(self.mc, arglocs[1], arglocs[0])
  1584. def _store_and_reset_exception(self, mc, excvalloc=None, exctploc=None,
  1585. tmploc=None):
  1586. """ Resest the exception. If excvalloc is None, then store it on the
  1587. frame in jf_guard_exc
  1588. """
  1589. if excvalloc is not None:
  1590. assert excvalloc.is_core_reg()
  1591. mc.MOV(excvalloc, heap(self.cpu.pos_exc_value()))
  1592. elif tmploc is not None: # if both are None, just ignore
  1593. ofs = self.cpu.get_ofs_of_frame_field('jf_guard_exc')
  1594. mc.MOV(tmploc, heap(self.cpu.pos_exc_value()))
  1595. mc.MOV(RawEbpLoc(ofs), tmploc)
  1596. if exctploc is not None:
  1597. assert exctploc.is_core_reg()
  1598. mc.MOV(exctploc, heap(self.cpu.pos_exception()))
  1599. mc.MOV(heap(self.cpu.pos_exception()), imm0)
  1600. mc.MOV(heap(self.cpu.pos_exc_value()), imm0)
  1601. def _restore_exception(self, mc, excvalloc, exctploc, tmploc=None):
  1602. if excvalloc is not None:
  1603. mc.MOV(heap(self.cpu.pos_exc_value()), excvalloc)
  1604. else:
  1605. assert tmploc is not None
  1606. ofs = self.cpu.get_ofs_of_frame_field('jf_guard_exc')
  1607. mc.MOV(tmploc, RawEbpLoc(ofs))
  1608. mc.MOV_bi(ofs, 0)
  1609. mc.MOV(heap(self.cpu.pos_exc_value()), tmploc)
  1610. mc.MOV(heap(self.cpu.pos_exception()), exctploc)
  1611. def genop_int_add_ovf(self, op, arglocs, resloc):
  1612. self.genop_int_add(op, arglocs, resloc)
  1613. self.guard_success_cc = rx86.Conditions['NO']
  1614. def genop_int_sub_ovf(self, op, arglocs, resloc):
  1615. self.genop_int_sub(op, arglocs, resloc)
  1616. self.guard_success_cc = rx86.Conditions['NO']
  1617. def genop_int_mul_ovf(self, op, arglocs, resloc):
  1618. self.genop_int_mul(op, arglocs, resloc)
  1619. self.guard_success_cc = rx86.Conditions['NO']
  1620. genop_guard_guard_no_overflow = genop_guard_guard_true
  1621. genop_guard_guard_overflow = genop_guard_guard_false
  1622. def genop_guard_guard_value(self, guard_op, guard_token, locs, ign):
  1623. if guard_op.getarg(0).type == FLOAT:
  1624. assert guard_op.getarg(1).type == FLOAT
  1625. self.mc.UCOMISD(locs[0], locs[1])
  1626. else:
  1627. self.mc.CMP(locs[0], locs[1])
  1628. self.guard_success_cc = rx86.Conditions['E']
  1629. self.implement_guard(guard_token)
  1630. def _cmp_guard_class(self, locs):
  1631. loc_ptr = locs[0]
  1632. loc_classptr = locs[1]
  1633. offset = self.cpu.vtable_offset
  1634. if offset is not None:
  1635. self.mc.CMP(mem(loc_ptr, offset), loc_classptr)
  1636. else:
  1637. assert isinstance(loc_classptr, ImmedLoc)
  1638. classptr = loc_classptr.value
  1639. expected_typeid = (self.cpu.gc_ll_descr
  1640. .get_typeid_from_classptr_if_gcremovetypeptr(classptr))
  1641. self._cmp_guard_gc_type(loc_ptr, ImmedLoc(expected_typeid))
  1642. def _cmp_guard_gc_type(self, loc_ptr, loc_expected_typeid):
  1643. # Note that the typeid half-word is at offset 0 on a little-endian
  1644. # machine; it would be at offset 2 or 4 on a big-endian machine.
  1645. assert self.cpu.supports_guard_gc_type
  1646. if IS_X86_32:
  1647. self.mc.CMP16(mem(loc_ptr, 0), loc_expected_typeid)
  1648. else:
  1649. assert isinstance(loc_expected_typeid, ImmedLoc)
  1650. self.mc.CMP32_mi((loc_ptr.value, 0), loc_expected_typeid.value)
  1651. def genop_guard_guard_class(self, guard_op, guard_token, locs, ign):
  1652. self._cmp_guard_class(locs)
  1653. self.guard_success_cc = rx86.Conditions['E']
  1654. self.implement_guard(guard_token)
  1655. def genop_guard_guard_nonnull_class(self, guard_op, guard_token, locs, ign):
  1656. self.mc.CMP(locs[0], imm1)
  1657. # Patched below
  1658. self.mc.J_il8(rx86.Conditions['B'], 0)
  1659. jb_location = self.mc.get_relative_pos()
  1660. self._cmp_guard_class(locs)
  1661. # patch the JB above
  1662. offset = self.mc.get_relative_pos() - jb_location
  1663. assert 0 < offset <= 127
  1664. self.mc.overwrite(jb_location-1, chr(offset))
  1665. #
  1666. self.guard_success_cc = rx86.Conditions['E']
  1667. self.implement_guard(guard_token)
  1668. def genop_guard_guard_gc_type(self, guard_op, guard_token, locs, ign):
  1669. self._cmp_guard_gc_type(locs[0], locs[1])
  1670. self.guard_success_cc = rx86.Conditions['E']
  1671. self.implement_guard(guard_token)
  1672. def genop_guard_guard_is_object(self, guard_op, guard_token, locs, ign):
  1673. assert self.cpu.supports_guard_gc_type
  1674. [loc_object, loc_typeid] = locs
  1675. # idea: read the typeid, fetch the field 'infobits' from the big
  1676. # typeinfo table, and check the flag 'T_IS_RPYTHON_INSTANCE'.
  1677. if IS_X86_32:
  1678. self.mc.MOVZX16(loc_typeid, mem(loc_object, 0))
  1679. else:
  1680. self.mc.MOV32(loc_typeid, mem(loc_object, 0))
  1681. #
  1682. base_type_info, shift_by, sizeof_ti = (
  1683. self.cpu.gc_ll_descr.get_translated_info_for_typeinfo())
  1684. infobits_offset, IS_OBJECT_FLAG = (
  1685. self.cpu.gc_ll_descr.get_translated_info_for_guard_is_object())
  1686. loc_infobits = addr_add(imm(base_type_info), loc_typeid,
  1687. scale=shift_by, offset=infobits_offset)
  1688. self.mc.TEST8(loc_infobits, imm(IS_OBJECT_FLAG))
  1689. #
  1690. self.guard_success_cc = rx86.Conditions['NZ']
  1691. self.implement_guard(guard_token)
  1692. def genop_guard_guard_subclass(self, guard_op, guard_token, locs, ign):
  1693. assert self.cpu.supports_guard_gc_type
  1694. [loc_object, loc_check_against_class, loc_tmp] = locs
  1695. assert isinstance(loc_object, RegLoc)
  1696. assert isinstance(loc_tmp, RegLoc)
  1697. offset = self.cpu.vtable_offset
  1698. offset2 = self.cpu.subclassrange_min_offset
  1699. if offset is not None:
  1700. # read this field to get the vtable pointer
  1701. self.mc.MOV_rm(loc_tmp.value, (loc_object.value, offset))
  1702. # read the vtable's subclassrange_min field
  1703. self.mc.MOV_rm(loc_tmp.value, (loc_tmp.value, offset2))
  1704. else:
  1705. # read the typeid
  1706. if IS_X86_32:
  1707. self.mc.MOVZX16(loc_tmp, mem(loc_object, 0))
  1708. else:
  1709. self.mc.MOV32(loc_tmp, mem(loc_object, 0))
  1710. # read the vtable's subclassrange_min field, as a single
  1711. # step with the correct offset
  1712. base_type_info, shift_by, sizeof_ti = (
  1713. self.cpu.gc_ll_descr.get_translated_info_for_typeinfo())
  1714. self.mc.MOV(loc_tmp, addr_add(imm(base_type_info), loc_tmp,
  1715. scale = shift_by,
  1716. offset = sizeof_ti + offset2))
  1717. # get the two bounds to check against
  1718. vtable_ptr = loc_check_against_class.getint()
  1719. vtable_ptr = rffi.cast(rclass.CLASSTYPE, vtable_ptr)
  1720. check_min = vtable_ptr.subclassrange_min
  1721. check_max = vtable_ptr.subclassrange_max
  1722. # check by doing the unsigned comparison (tmp - min) < (max - min)
  1723. self.mc.SUB_ri(loc_tmp.value, check_min)
  1724. self.mc.CMP_ri(loc_tmp.value, check_max - check_min)
  1725. # the guard passes if we get a result of "below"
  1726. self.guard_success_cc = rx86.Conditions['B']
  1727. self.implement_guard(guard_token)
  1728. def implement_guard_recovery(self, guard_opnum, faildescr, failargs,
  1729. fail_locs, frame_depth):
  1730. gcmap = allocate_gcmap(self, frame_depth, JITFRAME_FIXED_SIZE)
  1731. faildescrindex = self.get_gcref_from_faildescr(faildescr)
  1732. return GuardToken(self.cpu, gcmap, faildescr, failargs, fail_locs,
  1733. guard_opnum, frame_depth, faildescrindex)
  1734. def generate_propagate_error_64(self):
  1735. assert WORD == 8
  1736. self.mc.force_frame_size(DEFAULT_FRAME_BYTES)
  1737. startpos = self.mc.get_relative_pos()
  1738. self.mc.JMP(imm(self.propagate_exception_path))
  1739. return startpos
  1740. def generate_quick_failure(self, guardtok, regalloc):
  1741. """ Gather information about failure
  1742. """
  1743. self.mc.force_frame_size(DEFAULT_FRAME_BYTES)
  1744. startpos = self.mc.get_relative_pos()
  1745. #
  1746. self._update_at_exit(guardtok.fail_locs, guardtok.failargs,
  1747. guardtok.faildescr, regalloc)
  1748. #
  1749. faildescrindex, target = self.store_info_on_descr(startpos, guardtok)
  1750. if IS_X86_64:
  1751. self.mc.PUSH_p(0) # %rip-relative
  1752. self._patch_load_from_gc_table(faildescrindex)
  1753. elif IS_X86_32:
  1754. self.mc.PUSH_j(self._addr_from_gc_table(faildescrindex))
  1755. self.push_gcmap(self.mc, guardtok.gcmap, push=True)
  1756. self.mc.JMP(imm(target))
  1757. return startpos
  1758. def push_gcmap(self, mc, gcmap, push=False, store=False):
  1759. if push:
  1760. mc.PUSH(imm(rffi.cast(lltype.Signed, gcmap)))
  1761. else:
  1762. assert store
  1763. ofs = self.cpu.get_ofs_of_frame_field('jf_gcmap')
  1764. mc.MOV(raw_stack(ofs), imm(rffi.cast(lltype.Signed, gcmap)))
  1765. def pop_gcmap(self, mc):
  1766. ofs = self.cpu.get_ofs_of_frame_field('jf_gcmap')
  1767. mc.MOV_bi(ofs, 0)
  1768. def new_stack_loc(self, i, tp):
  1769. base_ofs = self.cpu.get_baseofs_of_frame_field()
  1770. return FrameLoc(i, get_ebp_ofs(base_ofs, i), tp)
  1771. def setup_failure_recovery(self):
  1772. self.failure_recovery_code = [0, 0, 0, 0]
  1773. def _push_all_regs_to_frame(self, mc, ignored_regs, withfloats,
  1774. callee_only=False):
  1775. # Push all general purpose registers
  1776. base_ofs = self.cpu.get_baseofs_of_frame_field()
  1777. if callee_only:
  1778. regs = gpr_reg_mgr_cls.save_around_call_regs
  1779. else:
  1780. regs = gpr_reg_mgr_cls.all_regs
  1781. for gpr in regs:
  1782. if gpr not in ignored_regs:
  1783. v = gpr_reg_mgr_cls.all_reg_indexes[gpr.value]
  1784. mc.MOV_br(v * WORD + base_ofs, gpr.value)
  1785. if withfloats:
  1786. if IS_X86_64:
  1787. coeff = 1
  1788. else:
  1789. coeff = 2
  1790. # Push all XMM regs
  1791. ofs = len(gpr_reg_mgr_cls.all_regs)
  1792. for i in range(len(xmm_reg_mgr_cls.all_regs)):
  1793. mc.MOVSD_bx((ofs + i * coeff) * WORD + base_ofs, i)
  1794. def _pop_all_regs_from_frame(self, mc, ignored_regs, withfloats,
  1795. callee_only=False):
  1796. # Pop all general purpose registers
  1797. base_ofs = self.cpu.get_baseofs_of_frame_field()
  1798. if callee_only:
  1799. regs = gpr_reg_mgr_cls.save_around_call_regs
  1800. else:
  1801. regs = gpr_reg_mgr_cls.all_regs
  1802. for gpr in regs:
  1803. if gpr not in ignored_regs:
  1804. v = gpr_reg_mgr_cls.all_reg_indexes[gpr.value]
  1805. mc.MOV_rb(gpr.value, v * WORD + base_ofs)
  1806. if withfloats:
  1807. # Pop all XMM regs
  1808. if IS_X86_64:
  1809. coeff = 1
  1810. else:
  1811. coeff = 2
  1812. ofs = len(gpr_reg_mgr_cls.all_regs)
  1813. for i in range(len(xmm_reg_mgr_cls.all_regs)):
  1814. mc.MOVSD_xb(i, (ofs + i * coeff) * WORD + base_ofs)
  1815. def _build_failure_recovery(self, exc, withfloats=False):
  1816. mc = codebuf.MachineCodeBlockWrapper()
  1817. # this is jumped to, from a stack that has DEFAULT_FRAME_BYTES
  1818. # followed by 2 extra words just pushed
  1819. mc.force_frame_size(DEFAULT_FRAME_BYTES + 2 * WORD)
  1820. self.mc = mc
  1821. self._push_all_regs_to_frame(mc, [], withfloats)
  1822. if exc:
  1823. # We might have an exception pending. Load it into ebx...
  1824. mc.MOV(ebx, heap(self.cpu.pos_exc_value()))
  1825. mc.MOV(heap(self.cpu.pos_exception()), imm0)
  1826. mc.MOV(heap(self.cpu.pos_exc_value()), imm0)
  1827. # ...and save ebx into 'jf_guard_exc'
  1828. offset = self.cpu.get_ofs_of_frame_field('jf_guard_exc')
  1829. mc.MOV_br(offset, ebx.value)
  1830. # fill in the jf_descr and jf_gcmap fields of the frame according
  1831. # to which failure we are resuming from. These are constants
  1832. # pushed on the stack just before we jump to the current helper,
  1833. # in generate_quick_failure().
  1834. ofs = self.cpu.get_ofs_of_frame_field('jf_descr')
  1835. ofs2 = self.cpu.get_ofs_of_frame_field('jf_gcmap')
  1836. mc.POP_b(ofs2)
  1837. mc.POP_b(ofs)
  1838. # now we return from the complete frame, which starts from
  1839. # _call_header_with_stack_check(). The _call_footer below does it.
  1840. self._call_footer()
  1841. rawstart = mc.materialize(self.cpu, [])
  1842. self.failure_recovery_code[exc + 2 * withfloats] = rawstart
  1843. self.mc = None
  1844. def genop_finish(self, op, arglocs, result_loc):
  1845. base_ofs = self.cpu.get_baseofs_of_frame_field()
  1846. if len(arglocs) > 0:
  1847. [return_val] = arglocs
  1848. if op.getarg(0).type == FLOAT and not IS_X86_64:
  1849. size = WORD * 2
  1850. else:
  1851. size = WORD
  1852. self.save_into_mem(raw_stack(base_ofs), return_val, imm(size))
  1853. ofs = self.cpu.get_ofs_of_frame_field('jf_descr')
  1854. descr = op.getdescr()
  1855. faildescrindex = self.get_gcref_from_faildescr(descr)
  1856. if IS_X86_64:
  1857. self.mc.MOV_rp(eax.value, 0)
  1858. self._patch_load_from_gc_table(faildescrindex)
  1859. elif IS_X86_32:
  1860. self.mc.MOV_rj(eax.value, self._addr_from_gc_table(faildescrindex))
  1861. self.mov(eax, RawEbpLoc(ofs))
  1862. arglist = op.getarglist()
  1863. if arglist and arglist[0].type == REF:
  1864. if self._finish_gcmap:
  1865. # we're returning with a guard_not_forced_2, and
  1866. # additionally we need to say that eax/rax contains
  1867. # a reference too:
  1868. self._finish_gcmap[0] |= r_uint(1)
  1869. gcmap = self._finish_gcmap
  1870. else:
  1871. gcmap = self.gcmap_for_finish
  1872. self.push_gcmap(self.mc, gcmap, store=True)
  1873. elif self._finish_gcmap:
  1874. # we're returning with a guard_not_forced_2
  1875. gcmap = self._finish_gcmap
  1876. self.push_gcmap(self.mc, gcmap, store=True)
  1877. else:
  1878. # note that the 0 here is redundant, but I would rather
  1879. # keep that one and kill all the others
  1880. ofs = self.cpu.get_ofs_of_frame_field('jf_gcmap')
  1881. self.mc.MOV_bi(ofs, 0)
  1882. # exit function
  1883. self._call_footer()
  1884. def implement_guard(self, guard_token):
  1885. # These jumps are patched later.
  1886. assert self.guard_success_cc >= 0
  1887. self.mc.J_il(rx86.invert_condition(self.guard_success_cc), 0)
  1888. self.guard_success_cc = rx86.cond_none
  1889. guard_token.pos_jump_offset = self.mc.get_relative_pos() - 4
  1890. self.pending_guard_tokens.append(guard_token)
  1891. def _genop_real_call(self, op, arglocs, resloc):
  1892. self._genop_call(op, arglocs, resloc)
  1893. genop_call_i = _genop_real_call
  1894. genop_call_r = _genop_real_call
  1895. genop_call_f = _genop_real_call
  1896. genop_call_n = _genop_real_call
  1897. def _genop_call(self, op, arglocs, resloc, is_call_release_gil=False):
  1898. from rpython.jit.backend.llsupport.descr import CallDescr
  1899. func_index = 2 + is_call_release_gil
  1900. cb = callbuilder.CallBuilder(self, arglocs[func_index],
  1901. arglocs[func_index+1:], resloc)
  1902. descr = op.getdescr()
  1903. assert isinstance(descr, CallDescr)
  1904. cb.callconv = descr.get_call_conv()
  1905. cb.argtypes = descr.get_arg_types()
  1906. cb.restype = descr.get_result_type()
  1907. sizeloc = arglocs[0]
  1908. assert isinstance(sizeloc, ImmedLoc)
  1909. cb.ressize = sizeloc.value
  1910. signloc = arglocs[1]
  1911. assert isinstance(signloc, ImmedLoc)
  1912. cb.ressign = signloc.value
  1913. if is_call_release_gil:
  1914. saveerrloc = arglocs[2]
  1915. assert isinstance(saveerrloc, ImmedLoc)
  1916. cb.emit_call_release_gil(saveerrloc.value)
  1917. else:
  1918. effectinfo = descr.get_extra_info()
  1919. if effectinfo is None or effectinfo.check_can_collect():
  1920. cb.emit()
  1921. else:
  1922. cb.emit_no_collect()
  1923. def _store_force_index(self, guard_op):
  1924. assert (guard_op.getopnum() == rop.GUARD_NOT_FORCED or
  1925. guard_op.getopnum() == rop.GUARD_NOT_FORCED_2)
  1926. faildescr = guard_op.getdescr()
  1927. ofs = self.cpu.get_ofs_of_frame_field('jf_force_descr')
  1928. faildescrindex = self.get_gcref_from_faildescr(faildescr)
  1929. if IS_X86_64:
  1930. self.mc.MOV_rp(X86_64_SCRATCH_REG.value, 0)
  1931. self._patch_load_from_gc_table(faildescrindex)
  1932. self.mc.MOV(raw_stack(ofs), X86_64_SCRATCH_REG)
  1933. elif IS_X86_32:
  1934. # XXX need a scratch reg here for efficiency; be more clever
  1935. self.mc.PUSH_j(self._addr_from_gc_table(faildescrindex))
  1936. self.mc.POP(raw_stack(ofs))
  1937. def _find_nearby_operation(self, delta):
  1938. regalloc = self._regalloc
  1939. return regalloc.operations[regalloc.rm.position + delta]
  1940. def genop_guard_guard_not_forced(self, guard_op, guard_token, locs, resloc):
  1941. ofs = self.cpu.get_ofs_of_frame_field('jf_descr')
  1942. self.mc.CMP_bi(ofs, 0)
  1943. self.guard_success_cc = rx86.Conditions['E']
  1944. self.implement_guard(guard_token)
  1945. def _genop_call_may_force(self, op, arglocs, result_loc):
  1946. self._store_force_index(self._find_nearby_operation(+1))
  1947. self._genop_call(op, arglocs, result_loc)
  1948. genop_call_may_force_i = _genop_call_may_force
  1949. genop_call_may_force_r = _genop_call_may_force
  1950. genop_call_may_force_f = _genop_call_may_force
  1951. genop_call_may_force_n = _genop_call_may_force
  1952. def _genop_call_release_gil(self, op, arglocs, result_loc):
  1953. self._store_force_index(self._find_nearby_operation(+1))
  1954. self._genop_call(op, arglocs, result_loc, is_call_release_gil=True)
  1955. genop_call_release_gil_i = _genop_call_release_gil
  1956. genop_call_release_gil_f = _genop_call_release_gil
  1957. genop_call_release_gil_n = _genop_call_release_gil
  1958. def imm(self, v):
  1959. return imm(v)
  1960. # ------------------- CALL ASSEMBLER --------------------------
  1961. def _genop_call_assembler(self, op, arglocs, result_loc):
  1962. if len(arglocs) == 2:
  1963. [argloc, vloc] = arglocs
  1964. else:
  1965. [argloc] = arglocs
  1966. vloc = self.imm(0)
  1967. self._store_force_index(self._find_nearby_operation(+1))
  1968. self.call_assembler(op, argloc, vloc, result_loc, eax)
  1969. genop_call_assembler_i = _genop_call_assembler
  1970. genop_call_assembler_r = _genop_call_assembler
  1971. genop_call_assembler_f = _genop_call_assembler
  1972. genop_call_assembler_n = _genop_call_assembler
  1973. def _call_assembler_emit_call(self, addr, argloc, _):
  1974. threadlocal_loc = RawEspLoc(THREADLOCAL_OFS, INT)
  1975. if self._is_asmgcc():
  1976. # We need to remove the bit "already seen during the
  1977. # previous minor collection" instead of passing this
  1978. # value directly.
  1979. if IS_X86_64:
  1980. tmploc = esi # already the correct place
  1981. if argloc is tmploc:
  1982. # this case is theoretical only so far: in practice,
  1983. # argloc is always eax, never esi
  1984. self.mc.MOV_rr(edi.value, esi.value)
  1985. argloc = edi
  1986. else:
  1987. tmploc = eax
  1988. if tmploc is argloc:
  1989. tmploc = edx
  1990. self.mc.MOV(tmploc, threadlocal_loc)
  1991. self.mc.AND_ri(tmploc.value, ~1)
  1992. threadlocal_loc = tmploc
  1993. #
  1994. self.simple_call(addr, [argloc, threadlocal_loc])
  1995. def _call_assembler_emit_helper_call(self, addr, arglocs, result_loc):
  1996. self.simple_call(addr, arglocs, result_loc)
  1997. def _call_assembler_check_descr(self, value, tmploc):
  1998. ofs = self.cpu.get_ofs_of_frame_field('jf_descr')
  1999. self.mc.CMP(mem(eax, ofs), imm(value))
  2000. # patched later
  2001. self.mc.J_il8(rx86.Conditions['E'], 0) # goto B if we get 'done_with_this_frame'
  2002. return self.mc.get_relative_pos()
  2003. def _call_assembler_patch_je(self, result_loc, je_location):
  2004. if (IS_X86_32 and isinstance(result_loc, FrameLoc) and
  2005. result_loc.type == FLOAT):
  2006. self.mc.FSTPL_b(result_loc.value)
  2007. self.mc.JMP_l8(0) # jump to done, patched later
  2008. jmp_location = self.mc.get_relative_pos()
  2009. #
  2010. offset = jmp_location - je_location
  2011. assert 0 < offset <= 127
  2012. self.mc.overwrite(je_location - 1, chr(offset))
  2013. self.mc.force_frame_size(DEFAULT_FRAME_BYTES)
  2014. #
  2015. return jmp_location
  2016. def _call_assembler_load_result(self, op, result_loc):
  2017. if op.type != 'v':
  2018. # load the return value from the dead frame's value index 0
  2019. kind = op.type
  2020. descr = self.cpu.getarraydescr_for_frame(kind)
  2021. ofs = self.cpu.unpack_arraydescr(descr)
  2022. if kind == FLOAT:
  2023. self.mc.MOVSD_xm(xmm0.value, (eax.value, ofs))
  2024. if result_loc is not xmm0:
  2025. self.mc.MOVSD(result_loc, xmm0)
  2026. else:
  2027. assert result_loc is eax
  2028. self.mc.MOV_rm(eax.value, (eax.value, ofs))
  2029. def _call_assembler_patch_jmp(self, jmp_location):
  2030. offset = self.mc.get_relative_pos() - jmp_location
  2031. assert 0 <= offset <= 127
  2032. self.mc.overwrite(jmp_location - 1, chr(offset))
  2033. # ------------------- END CALL ASSEMBLER -----------------------
  2034. def _write_barrier_fastpath(self, mc, descr, arglocs, array=False,
  2035. is_frame=False):
  2036. # Write code equivalent to write_barrier() in the GC: it checks
  2037. # a flag in the object at arglocs[0], and if set, it calls a
  2038. # helper piece of assembler. The latter saves registers as needed
  2039. # and call the function remember_young_pointer() from the GC.
  2040. if we_are_translated():
  2041. cls = self.cpu.gc_ll_descr.has_write_barrier_class()
  2042. assert cls is not None and isinstance(descr, cls)
  2043. #
  2044. card_marking = False
  2045. mask = descr.jit_wb_if_flag_singlebyte
  2046. if array and descr.jit_wb_cards_set != 0:
  2047. # assumptions the rest of the function depends on:
  2048. assert (descr.jit_wb_cards_set_byteofs ==
  2049. descr.jit_wb_if_flag_byteofs)
  2050. assert descr.jit_wb_cards_set_singlebyte == -0x80
  2051. card_marking = True
  2052. mask = descr.jit_wb_if_flag_singlebyte | -0x80
  2053. #
  2054. loc_base = arglocs[0]
  2055. if is_frame:
  2056. assert loc_base is ebp
  2057. loc = raw_stack(descr.jit_wb_if_flag_byteofs)
  2058. else:
  2059. loc = addr_add_const(loc_base, descr.jit_wb_if_flag_byteofs)
  2060. mc.TEST8(loc, imm(mask))
  2061. mc.J_il8(rx86.Conditions['Z'], 0) # patched later
  2062. jz_location = mc.get_relative_pos()
  2063. # for cond_call_gc_wb_array, also add another fast path:
  2064. # if GCFLAG_CARDS_SET, then we can just set one bit and be done
  2065. if card_marking:
  2066. # GCFLAG_CARDS_SET is in this byte at 0x80, so this fact can
  2067. # been checked by the status flags of the previous TEST8
  2068. mc.J_il8(rx86.Conditions['S'], 0) # patched later
  2069. js_location = mc.get_relative_pos()
  2070. else:
  2071. js_location = 0
  2072. # Write only a CALL to the helper prepared in advance, passing it as
  2073. # argument the address of the structure we are writing into
  2074. # (the first argument to COND_CALL_GC_WB).
  2075. helper_num = card_marking
  2076. if is_frame:
  2077. helper_num = 4
  2078. elif self._regalloc is not None and self._regalloc.xrm.reg_bindings:
  2079. helper_num += 2
  2080. if self.wb_slowpath[helper_num] == 0: # tests only
  2081. assert not we_are_translated()
  2082. self.cpu.gc_ll_descr.write_barrier_descr = descr
  2083. self._build_wb_slowpath(card_marking,
  2084. bool(self._regalloc.xrm.reg_bindings))
  2085. assert self.wb_slowpath[helper_num] != 0
  2086. #
  2087. if not is_frame:
  2088. mc.PUSH(loc_base)
  2089. mc.CALL(imm(self.wb_slowpath[helper_num]))
  2090. if not is_frame:
  2091. mc.stack_frame_size_delta(-WORD)
  2092. if card_marking:
  2093. # The helper ends again with a check of the flag in the object.
  2094. # So here, we can simply write again a 'JNS', which will be
  2095. # taken if GCFLAG_CARDS_SET is still not set.
  2096. mc.J_il8(rx86.Conditions['NS'], 0) # patched later
  2097. jns_location = mc.get_relative_pos()
  2098. #
  2099. # patch the JS above
  2100. offset = mc.get_relative_pos() - js_location
  2101. assert 0 < offset <= 127
  2102. mc.overwrite(js_location-1, chr(offset))
  2103. #
  2104. # case GCFLAG_CARDS_SET: emit a few instructions to do
  2105. # directly the card flag setting
  2106. loc_index = arglocs[1]
  2107. if isinstance(loc_index, RegLoc):
  2108. if IS_X86_64 and isinstance(loc_base, RegLoc):
  2109. # copy loc_index into r11
  2110. tmp1 = X86_64_SCRATCH_REG
  2111. mc.MOV_rr(tmp1.value, loc_index.value)
  2112. final_pop = False
  2113. else:
  2114. # must save the register loc_index before it is mutated
  2115. mc.PUSH_r(loc_index.value)
  2116. tmp1 = loc_index
  2117. final_pop = True
  2118. # SHR tmp, card_page_shift
  2119. mc.SHR_ri(tmp1.value, descr.jit_wb_card_page_shift)
  2120. # XOR tmp, -8
  2121. mc.XOR_ri(tmp1.value, -8)
  2122. # BTS [loc_base], tmp
  2123. mc.BTS(addr_add_const(loc_base, 0), tmp1)
  2124. # done
  2125. if final_pop:
  2126. mc.POP_r(loc_index.value)
  2127. #
  2128. elif isinstance(loc_index, ImmedLoc):
  2129. byte_index = loc_index.value >> descr.jit_wb_card_page_shift
  2130. byte_ofs = ~(byte_index >> 3)
  2131. byte_val = 1 << (byte_index & 7)
  2132. mc.OR8(addr_add_const(loc_base, byte_ofs), imm(byte_val))
  2133. else:
  2134. raise AssertionError("index is neither RegLoc nor ImmedLoc")
  2135. #
  2136. # patch the JNS above
  2137. offset = mc.get_relative_pos() - jns_location
  2138. assert 0 < offset <= 127
  2139. mc.overwrite(jns_location-1, chr(offset))
  2140. # patch the JZ above
  2141. offset = mc.get_relative_pos() - jz_location
  2142. assert 0 < offset <= 127
  2143. mc.overwrite(jz_location-1, chr(offset))
  2144. def genop_discard_cond_call_gc_wb(self, op, arglocs):
  2145. self._write_barrier_fastpath(self.mc, op.getdescr(), arglocs)
  2146. def genop_discard_cond_call_gc_wb_array(self, op, arglocs):
  2147. self._write_barrier_fastpath(self.mc, op.getdescr(), arglocs,
  2148. array=True)
  2149. def not_implemented_op_discard(self, op, arglocs):
  2150. not_implemented("not implemented operation: %s" % op.getopname())
  2151. def not_implemented_op(self, op, arglocs, resloc):
  2152. not_implemented("not implemented operation with res: %s" %
  2153. op.getopname())
  2154. def not_implemented_op_guard(self, guard_op, guard_token, locs, resloc):
  2155. not_implemented("not implemented operation (guard): %s" %
  2156. guard_op.getopname())
  2157. def closing_jump(self, target_token):
  2158. target = target_token._ll_loop_code
  2159. if target_token in self.target_tokens_currently_compiling:
  2160. curpos = self.mc.get_relative_pos() + 5
  2161. self.mc.JMP_l(target - curpos)
  2162. else:
  2163. self.mc.JMP(imm(target))
  2164. def label(self):
  2165. self._check_frame_depth_debug(self.mc)
  2166. def cond_call(self, gcmap, imm_func, arglocs, resloc=None):
  2167. assert self.guard_success_cc >= 0
  2168. self.mc.J_il8(rx86.invert_condition(self.guard_success_cc), 0)
  2169. # patched later
  2170. jmp_adr = self.mc.get_relative_pos()
  2171. self.guard_success_cc = rx86.cond_none
  2172. #
  2173. self.push_gcmap(self.mc, gcmap, store=True)
  2174. #
  2175. # first save away the 4 registers from 'cond_call_register_arguments'
  2176. # plus the register 'eax'
  2177. base_ofs = self.cpu.get_baseofs_of_frame_field()
  2178. should_be_saved = self._regalloc.rm.reg_bindings.values()
  2179. restore_eax = False
  2180. for gpr in cond_call_register_arguments + [eax]:
  2181. if gpr not in should_be_saved or gpr is resloc:
  2182. continue
  2183. v = gpr_reg_mgr_cls.all_reg_indexes[gpr.value]
  2184. self.mc.MOV_br(v * WORD + base_ofs, gpr.value)
  2185. if gpr is eax:
  2186. restore_eax = True
  2187. #
  2188. # load the 0-to-4 arguments into these registers
  2189. from rpython.jit.backend.x86.jump import remap_frame_layout
  2190. remap_frame_layout(self, arglocs,
  2191. cond_call_register_arguments[:len(arglocs)],
  2192. X86_64_SCRATCH_REG if IS_X86_64 else None)
  2193. #
  2194. # load the constant address of the function to call into eax
  2195. self.mc.MOV(eax, imm_func)
  2196. #
  2197. # figure out which variant of cond_call_slowpath to call, and call it
  2198. callee_only = False
  2199. floats = False
  2200. if self._regalloc is not None:
  2201. for reg in self._regalloc.rm.reg_bindings.values():
  2202. if reg not in self._regalloc.rm.save_around_call_regs:
  2203. break
  2204. else:
  2205. callee_only = True
  2206. if self._regalloc.xrm.reg_bindings:
  2207. floats = True
  2208. cond_call_adr = self.cond_call_slowpath[floats * 2 + callee_only]
  2209. self.mc.CALL(imm(follow_jump(cond_call_adr)))
  2210. # if this is a COND_CALL_VALUE, we need to move the result in place
  2211. if resloc is not None and resloc is not eax:
  2212. self.mc.MOV(resloc, eax)
  2213. # restoring the registers saved above, and doing pop_gcmap(), is left
  2214. # to the cond_call_slowpath helper. We must only restore eax, if
  2215. # needed.
  2216. if restore_eax:
  2217. v = gpr_reg_mgr_cls.all_reg_indexes[eax.value]
  2218. self.mc.MOV_rb(eax.value, v * WORD + base_ofs)
  2219. #
  2220. offset = self.mc.get_relative_pos() - jmp_adr
  2221. assert 0 < offset <= 127
  2222. self.mc.overwrite(jmp_adr-1, chr(offset))
  2223. # might be overridden again to skip over the following
  2224. # guard_no_exception too
  2225. self.previous_cond_call_jcond = jmp_adr
  2226. def malloc_cond(self, nursery_free_adr, nursery_top_adr, size, gcmap):
  2227. assert size & (WORD-1) == 0 # must be correctly aligned
  2228. self.mc.MOV(ecx, heap(nursery_free_adr))
  2229. self.mc.LEA_rm(edx.value, (ecx.value, size))
  2230. self.mc.CMP(edx, heap(nursery_top_adr))
  2231. self.mc.J_il8(rx86.Conditions['NA'], 0) # patched later
  2232. jmp_adr = self.mc.get_relative_pos()
  2233. # save the gcmap
  2234. self.push_gcmap(self.mc, gcmap, store=True)
  2235. self.mc.CALL(imm(follow_jump(self.malloc_slowpath)))
  2236. offset = self.mc.get_relative_pos() - jmp_adr
  2237. assert 0 < offset <= 127
  2238. self.mc.overwrite(jmp_adr-1, chr(offset))
  2239. self.mc.MOV(heap(nursery_free_adr), edx)
  2240. def malloc_cond_varsize_frame(self, nursery_free_adr, nursery_top_adr,
  2241. sizeloc, gcmap):
  2242. if sizeloc is ecx:
  2243. self.mc.MOV(edx, sizeloc)
  2244. sizeloc = edx
  2245. self.mc.MOV(ecx, heap(nursery_free_adr))
  2246. if sizeloc is edx:
  2247. self.mc.ADD_rr(edx.value, ecx.value)
  2248. else:
  2249. self.mc.LEA_ra(edx.value, (ecx.value, sizeloc.value, 0, 0))
  2250. self.mc.CMP(edx, heap(nursery_top_adr))
  2251. self.mc.J_il8(rx86.Conditions['NA'], 0) # patched later
  2252. jmp_adr = self.mc.get_relative_pos()
  2253. # save the gcmap
  2254. self.push_gcmap(self.mc, gcmap, store=True)
  2255. self.mc.CALL(imm(follow_jump(self.malloc_slowpath)))
  2256. offset = self.mc.get_relative_pos() - jmp_adr
  2257. assert 0 < offset <= 127
  2258. self.mc.overwrite(jmp_adr-1, chr(offset))
  2259. self.mc.MOV(heap(nursery_free_adr), edx)
  2260. def malloc_cond_varsize(self, kind, nursery_free_adr, nursery_top_adr,
  2261. lengthloc, itemsize, maxlength, gcmap,
  2262. arraydescr):
  2263. from rpython.jit.backend.llsupport.descr import ArrayDescr
  2264. assert isinstance(arraydescr, ArrayDescr)
  2265. # lengthloc is the length of the array, which we must not modify!
  2266. assert lengthloc is not ecx and lengthloc is not edx
  2267. if isinstance(lengthloc, RegLoc):
  2268. varsizeloc = lengthloc
  2269. else:
  2270. self.mc.MOV(edx, lengthloc)
  2271. varsizeloc = edx
  2272. self.mc.CMP(varsizeloc, imm(maxlength))
  2273. self.mc.J_il8(rx86.Conditions['A'], 0) # patched later
  2274. jmp_adr0 = self.mc.get_relative_pos()
  2275. self.mc.MOV(ecx, heap(nursery_free_adr))
  2276. if valid_addressing_size(itemsize):
  2277. shift = get_scale(itemsize)
  2278. else:
  2279. shift = self._imul_const_scaled(self.mc, edx.value,
  2280. varsizeloc.value, itemsize)
  2281. varsizeloc = edx
  2282. # now varsizeloc is a register != ecx. The size of
  2283. # the variable part of the array is (varsizeloc << shift)
  2284. assert arraydescr.basesize >= self.gc_minimal_size_in_nursery
  2285. constsize = arraydescr.basesize + self.gc_size_of_header
  2286. force_realignment = (itemsize % WORD) != 0
  2287. if force_realignment:
  2288. constsize += WORD - 1
  2289. self.mc.LEA_ra(edx.value, (ecx.value, varsizeloc.value, shift,
  2290. constsize))
  2291. if force_realignment:
  2292. self.mc.AND_ri(edx.value, ~(WORD - 1))
  2293. # now edx contains the total size in bytes, rounded up to a multiple
  2294. # of WORD, plus nursery_free_adr
  2295. self.mc.CMP(edx, heap(nursery_top_adr))
  2296. self.mc.J_il8(rx86.Conditions['NA'], 0) # patched later
  2297. jmp_adr1 = self.mc.get_relative_pos()
  2298. #
  2299. offset = self.mc.get_relative_pos() - jmp_adr0
  2300. assert 0 < offset <= 127
  2301. self.mc.overwrite(jmp_adr0-1, chr(offset))
  2302. # save the gcmap
  2303. self.push_gcmap(self.mc, gcmap, store=True)
  2304. if kind == rewrite.FLAG_ARRAY:
  2305. self.mc.MOV_si(WORD, itemsize)
  2306. self.mc.MOV(edx, lengthloc)
  2307. self.mc.MOV_ri(ecx.value, arraydescr.tid)
  2308. addr = self.malloc_slowpath_varsize
  2309. else:
  2310. if kind == rewrite.FLAG_STR:
  2311. addr = self.malloc_slowpath_str
  2312. else:
  2313. assert kind == rewrite.FLAG_UNICODE
  2314. addr = self.malloc_slowpath_unicode
  2315. self.mc.MOV(edx, lengthloc)
  2316. self.mc.CALL(imm(follow_jump(addr)))
  2317. self.mc.JMP_l8(0) # jump to done, patched later
  2318. jmp_location = self.mc.get_relative_pos()
  2319. #
  2320. offset = self.mc.get_relative_pos() - jmp_adr1
  2321. assert 0 < offset <= 127
  2322. self.mc.overwrite(jmp_adr1-1, chr(offset))
  2323. self.mc.force_frame_size(DEFAULT_FRAME_BYTES)
  2324. # write down the tid, but not if it's the result of the CALL
  2325. self.mc.MOV(mem(ecx, 0), imm(arraydescr.tid))
  2326. # while we're at it, this line is not needed if we've done the CALL
  2327. self.mc.MOV(heap(nursery_free_adr), edx)
  2328. #
  2329. offset = self.mc.get_relative_pos() - jmp_location
  2330. assert 0 < offset <= 127
  2331. self.mc.overwrite(jmp_location - 1, chr(offset))
  2332. def store_force_descr(self, op, fail_locs, frame_depth):
  2333. guard_token = self.implement_guard_recovery(op.opnum,
  2334. op.getdescr(),
  2335. op.getfailargs(),
  2336. fail_locs, frame_depth)
  2337. self._finish_gcmap = guard_token.gcmap
  2338. self._store_force_index(op)
  2339. self.store_info_on_descr(0, guard_token)
  2340. def force_token(self, reg):
  2341. # XXX kill me
  2342. assert isinstance(reg, RegLoc)
  2343. self.mc.MOV_rr(reg.value, ebp.value)
  2344. def threadlocalref_get(self, offset, resloc, size, sign):
  2345. # This loads the stack location THREADLOCAL_OFS into a
  2346. # register, and then read the word at the given offset.
  2347. # It is only supported if 'translate_support_code' is
  2348. # true; otherwise, the execute_token() was done with a
  2349. # dummy value for the stack location THREADLOCAL_OFS
  2350. #
  2351. assert self.cpu.translate_support_code
  2352. assert isinstance(resloc, RegLoc)
  2353. self.mc.MOV_rs(resloc.value, THREADLOCAL_OFS)
  2354. if self._is_asmgcc():
  2355. self.mc.AND_ri(resloc.value, ~1)
  2356. self.load_from_mem(resloc, addr_add_const(resloc, offset),
  2357. imm(size), imm(sign))
  2358. def genop_discard_zero_array(self, op, arglocs):
  2359. (base_loc, startindex_loc, bytes_loc,
  2360. itemsize_loc, baseofs_loc, null_loc) = arglocs
  2361. assert isinstance(bytes_loc, ImmedLoc)
  2362. assert isinstance(itemsize_loc, ImmedLoc)
  2363. assert isinstance(baseofs_loc, ImmedLoc)
  2364. assert isinstance(null_loc, RegLoc) and null_loc.is_xmm
  2365. baseofs = baseofs_loc.value
  2366. nbytes = bytes_loc.value
  2367. assert valid_addressing_size(itemsize_loc.value)
  2368. scale = get_scale(itemsize_loc.value)
  2369. null_reg_cleared = False
  2370. i = 0
  2371. while i < nbytes:
  2372. addr = addr_add(base_loc, startindex_loc, baseofs + i, scale)
  2373. current = nbytes - i
  2374. if current >= 16 and self.cpu.supports_floats:
  2375. current = 16
  2376. if not null_reg_cleared:
  2377. self.mc.XORPS_xx(null_loc.value, null_loc.value)
  2378. null_reg_cleared = True
  2379. self.mc.MOVUPS(addr, null_loc)
  2380. else:
  2381. if current >= WORD:
  2382. current = WORD
  2383. elif current >= 4:
  2384. current = 4
  2385. elif current >= 2:
  2386. current = 2
  2387. self.save_into_mem(addr, imm0, imm(current))
  2388. i += current
  2389. genop_discard_list = [Assembler386.not_implemented_op_discard] * rop._LAST
  2390. genop_list = [Assembler386.not_implemented_op] * rop._LAST
  2391. genop_llong_list = {}
  2392. genop_math_list = {}
  2393. genop_tlref_list = {}
  2394. genop_guard_list = [Assembler386.not_implemented_op_guard] * rop._LAST
  2395. import itertools
  2396. iterate = itertools.chain(Assembler386.__dict__.iteritems(),
  2397. VectorAssemblerMixin.__dict__.iteritems())
  2398. for name, value in iterate:
  2399. if name.startswith('genop_discard_'):
  2400. opname = name[len('genop_discard_'):]
  2401. num = getattr(rop, opname.upper())
  2402. genop_discard_list[num] = value
  2403. elif name.startswith('genop_guard_'):
  2404. opname = name[len('genop_guard_'):]
  2405. num = getattr(rop, opname.upper())
  2406. genop_guard_list[num] = value
  2407. elif name.startswith('genop_llong_'):
  2408. opname = name[len('genop_llong_'):]
  2409. num = getattr(EffectInfo, 'OS_LLONG_' + opname.upper())
  2410. genop_llong_list[num] = value
  2411. elif name.startswith('genop_math_'):
  2412. opname = name[len('genop_math_'):]
  2413. num = getattr(EffectInfo, 'OS_MATH_' + opname.upper())
  2414. genop_math_list[num] = value
  2415. elif name.startswith('genop_'):
  2416. opname = name[len('genop_'):]
  2417. num = getattr(rop, opname.upper())
  2418. genop_list[num] = value
  2419. # XXX: ri386 migration shims:
  2420. def addr_add(reg_or_imm1, reg_or_imm2, offset=0, scale=0):
  2421. return AddressLoc(reg_or_imm1, reg_or_imm2, scale, offset)
  2422. def addr_add_const(reg_or_imm1, offset):
  2423. return AddressLoc(reg_or_imm1, imm0, 0, offset)
  2424. def mem(loc, offset):
  2425. return AddressLoc(loc, imm0, 0, offset)
  2426. def raw_stack(offset, type=INT):
  2427. return RawEbpLoc(offset, type)
  2428. def heap(addr):
  2429. return AddressLoc(ImmedLoc(addr), imm0, 0, 0)
  2430. def not_implemented(msg):
  2431. msg = '[x86/asm] %s\n' % msg
  2432. if we_are_translated():
  2433. llop.debug_print(lltype.Void, msg)
  2434. raise NotImplementedError(msg)
  2435. cond_call_register_arguments = [edi, esi, edx, ecx]
  2436. class BridgeAlreadyCompiled(Exception):
  2437. pass