[pypy-svn] r79605 - pypy/branch/jit-free-asm/pypy/jit/backend/x86

[arigo at codespeak.net]

Author: arigo
Date: Sun Nov 28 12:34:02 2010
New Revision: 79605

Added:
   pypy/branch/jit-free-asm/pypy/jit/backend/x86/codebuf.py
Log:
Forgot to add this.


Added: pypy/branch/jit-free-asm/pypy/jit/backend/x86/codebuf.py
==============================================================================
--- (empty file)
+++ pypy/branch/jit-free-asm/pypy/jit/backend/x86/codebuf.py	Sun Nov 28 12:34:02 2010
@@ -0,0 +1,35 @@
+from pypy.rpython.lltypesystem import lltype, rffi
+from pypy.jit.backend.llsupport.asmmemmgr import BlockBuilderMixin
+from pypy.jit.backend.x86.rx86 import X86_32_CodeBuilder, X86_64_CodeBuilder
+from pypy.jit.backend.x86.regloc import LocationCodeBuilder
+from pypy.jit.backend.x86.arch import IS_X86_32, IS_X86_64, WORD
+from pypy.jit.backend.x86 import valgrind
+
+# XXX: Seems nasty to change the superclass of MachineCodeBlockWrapper
+# like this
+if IS_X86_32:
+    codebuilder_cls = X86_32_CodeBuilder
+elif IS_X86_64:
+    codebuilder_cls = X86_64_CodeBuilder
+
+
+class MachineCodeBlockWrapper(BlockBuilderMixin,
+                              codebuilder_cls,
+                              LocationCodeBuilder):
+    def __init__(self):
+        self.init_block_builder()
+        # a list of relative positions; for each position p, the bytes
+        # at [p-4:p] encode an absolute address that will need to be
+        # made relative.
+        self.relocations = []
+
+    def add_pending_relocation(self):
+        self.relocations.append(self.get_relative_pos())
+
+    def copy_to_raw_memory(self, addr):
+        self._copy_to_raw_memory(addr)
+        for reloc in self.relocations:
+            p = addr + reloc
+            adr = rffi.cast(rffi.LONGP, p - WORD)
+            adr[0] -= p
+        valgrind.discard_translations(addr, self.get_relative_pos())