diff test/testx86asm.py @ 292:534b94b40aa8

Fixup reorganize
author Windel Bouwman
date Wed, 27 Nov 2013 08:06:42 +0100
parents 1c7c1e619be8
children 3b0c495e3008
line wrap: on
line diff
--- a/test/testx86asm.py	Sun Nov 24 11:24:15 2013 +0100
+++ b/test/testx86asm.py	Wed Nov 27 08:06:42 2013 +0100
@@ -1,18 +1,27 @@
 #!/usr/bin/python
 
 import unittest
+from testasm import AsmTestCaseBase
 
 
-class AssemblerTestCase(unittest.TestCase):
-   """ 
+class AssemblerTestCase(AsmTestCaseBase):
+    """ 
     test methods start with 'test*' 
      Checks several assembly constructs agains their bytecodes
-   """
-   def setUp(self):
+    """
+    def setUp(self):
         self.skipTest('not implemented yet')
         self.assembler = Assembler('x86-64')
+        a = Assembler()
 
-   def tstAssembler(self):
+    @unittest.skip 
+    def testX86(self):
+        self.feed('mov rax, rbx')  # ; 0x48, 0x89, 0xd8
+        self.feed('xor rcx, rbx')  # ; 0x48, 0x31, 0xd9
+        self.feed('inc rcx')  # ; 0x48 0xff 0xc1
+        self.check('48 89 d8 48 31 d9 48 ff c1')
+        
+    def tstAssembler(self):
       """ Check all kind of assembler cases """
       assert(assembler.shortjump(5) == [0xeb, 0x5])
       assert(assembler.shortjump(-2) == [0xeb, 0xfc])
@@ -21,35 +30,43 @@
       assert(assembler.nearjump(-2) == [0xe9, 0xf9, 0xff,0xff,0xff])
       assert(assembler.nearjump(10,'LE') == [0x0f, 0x8e, 0xa,0x0,0x0,0x0])
 
-   def testCall(self):
-      assert(assembler.call('r10') == [0x41, 0xff, 0xd2])
-      assert(assembler.call('rcx') == [0xff, 0xd1])
+    def testCall(self):
+        self.feed('call r10')
+        self.check('')
+        self.feed('call rcx')
+        
+        # assert(assembler.call('r10') == [0x41, 0xff, 0xd2])
+        # assert(assembler.call('rcx') == [0xff, 0xd1])
 
-   def testXOR(self):
+    def testXOR(self):
       assert(assembler.xorreg64('rax', 'rax') == [0x48, 0x31, 0xc0])
       assert(assembler.xorreg64('r9', 'r8') == [0x4d, 0x31, 0xc1])
       assert(assembler.xorreg64('rbx', 'r11') == [0x4c, 0x31, 0xdb])
 
-   def testINC(self):
+    def testINC(self):
       assert(assembler.increg64('r11') == [0x49, 0xff, 0xc3])
       assert(assembler.increg64('rcx') == [0x48, 0xff, 0xc1])
 
-   def testPush(self):
+    def testPush(self):
       assert(assembler.push('rbp') == [0x55])
       assert(assembler.push('rbx') == [0x53])
       assert(assembler.push('r12') == [0x41, 0x54])
-   def testPop(self):
-      assert(assembler.pop('rbx') == [0x5b])
-      assert(assembler.pop('rbp') == [0x5d])
-      assert(assembler.pop('r12') == [0x41, 0x5c])
 
-   def testAsmLoads(self):
+    def testPop(self):
+        self.feed('pop rbx')
+        self.feed('pop rbp')
+        self.feed('pop r12')
+        assert(assembler.pop('rbx') == [0x5b])
+        assert(assembler.pop('rbp') == [0x5d])
+        assert(assembler.pop('r12') == [0x41, 0x5c])
+
+    def testAsmLoads(self):
       # TODO constant add testcases
       assert(assembler.mov('rbx', 'r14') == [0x4c, 0x89, 0xf3])
       assert(assembler.mov('r12', 'r8')  == [0x4d, 0x89, 0xc4])
       assert(assembler.mov('rdi', 'rsp') == [0x48, 0x89, 0xe7])
 
-   def testAsmMemLoads(self):
+    def testAsmMemLoads(self):
       assert(assembler.mov('rax', ['r8','r15',0x11]) == [0x4b,0x8b,0x44,0x38,0x11])
       assert(assembler.mov('r13', ['rbp','rcx',0x23]) == [0x4c,0x8b,0x6c,0xd,0x23])
 
@@ -61,7 +78,7 @@
 
       assert(assembler.mov('r11', ['RIP', 0xf]) == [0x4c,0x8b,0x1d,0x0f,0x0,0x0,0x0])
 
-   def testAsmMemStores(self):
+    def testAsmMemStores(self):
       assert(assembler.mov(['rbp', 0x13],'rbx') == [0x48,0x89,0x5d,0x13])
       assert(assembler.mov(['r12', 0x12],'r9') == [0x4d,0x89,0x4c,0x24,0x12])
       assert(assembler.mov(['rcx', 0x11],'r14') == [0x4c,0x89,0x71,0x11])
@@ -72,27 +89,28 @@
 
       assert(assembler.mov(['RIP', 0xf], 'r9') == [0x4c,0x89,0x0d,0x0f,0x0,0x0,0x0])
 
-   def testAsmMOV8(self):
+    def testAsmMOV8(self):
       assert(assembler.mov(['rbp', -8], 'al') == [0x88, 0x45, 0xf8])
       assert(assembler.mov(['r11', 9], 'cl') == [0x41, 0x88, 0x4b, 0x09])
 
       assert(assembler.mov(['rbx'], 'al') == [0x88, 0x03])
       assert(assembler.mov(['r11'], 'dl') == [0x41, 0x88, 0x13])
 
-   def testAsmLea(self):
+    def testAsmLea(self):
       assert(assembler.leareg64('r11', ['RIP', 0xf]) == [0x4c,0x8d,0x1d,0x0f,0x0,0x0,0x0])
       assert(assembler.leareg64('rsi', ['RIP', 0x7]) == [0x48,0x8d,0x35,0x07,0x0,0x0,0x0])
 
       assert(assembler.leareg64('rcx', ['rbp', -8]) == [0x48,0x8d,0x4d,0xf8])
 
-   def testAssemblerCMP(self):
+    def testAssemblerCMP(self):
       assert(assembler.cmpreg64('rdi', 'r13') == [0x4c, 0x39, 0xef])
       assert(assembler.cmpreg64('rbx', 'r14') == [0x4c, 0x39, 0xf3])
       assert(assembler.cmpreg64('r12', 'r9')  == [0x4d, 0x39, 0xcc])
 
       assert(assembler.cmpreg64('rdi', 1)  == [0x48, 0x83, 0xff, 0x01])
       assert(assembler.cmpreg64('r11', 2)  == [0x49, 0x83, 0xfb, 0x02])
-   def testAssemblerADD(self):
+
+    def testAssemblerADD(self):
       assert(assembler.addreg64('rbx', 'r13') == [0x4c, 0x01, 0xeb])
       assert(assembler.addreg64('rax', 'rbx') == [0x48, 0x01, 0xd8])
       assert(assembler.addreg64('r12', 'r13') == [0x4d, 0x01, 0xec])
@@ -101,7 +119,7 @@
       assert(assembler.addreg64('r11', 0x1234567) == [0x49, 0x81, 0xc3, 0x67, 0x45,0x23,0x1])
       assert(assembler.addreg64('rsp', 0x33) == [0x48, 0x83, 0xc4, 0x33])
 
-   def testAssemblerSUB(self):
+    def testAssemblerSUB(self):
       assert(assembler.subreg64('rdx', 'r14') == [0x4c, 0x29, 0xf2])
       assert(assembler.subreg64('r15', 'rbx') == [0x49, 0x29, 0xdf])
       assert(assembler.subreg64('r8', 'r9') == [0x4d, 0x29, 0xc8])
@@ -109,12 +127,12 @@
       assert(assembler.subreg64('rsp', 0x123456) == [0x48, 0x81, 0xec, 0x56,0x34,0x12,0x0])
       assert(assembler.subreg64('rsp', 0x12) == [0x48, 0x83, 0xec, 0x12])
 
-   def testAssemblerIDIV(self):
+    def testAssemblerIDIV(self):
       assert(assembler.idivreg64('r11') == [0x49, 0xf7, 0xfb])
       assert(assembler.idivreg64('rcx') == [0x48, 0xf7, 0xf9])
       assert(assembler.idivreg64('rsp') == [0x48, 0xf7, 0xfc])
 
-   def testAssemblerIMUL(self):
+    def testAssemblerIMUL(self):
       assert(assembler.imulreg64_rax('rdi') == [0x48, 0xf7, 0xef])
       assert(assembler.imulreg64_rax('r10') == [0x49, 0xf7, 0xea])
       assert(assembler.imulreg64_rax('rdx') == [0x48, 0xf7, 0xea])