コード例 #1
0
ファイル: test_arm_arch.py プロジェクト: 13572293130/dissy
 def setUp(self):
     self.arch = ArmArchitecture()
     self.assertNotEqual(self.arch, None)
コード例 #2
0
ファイル: test_arm_arch.py プロジェクト: 13572293130/dissy
class TestArmArch(unittest.TestCase):
    def setUp(self):
        self.arch = ArmArchitecture()
        self.assertNotEqual(self.arch, None)

    def test_arg_parsing(self):
        self.assertEqual(self.arch.parseArguments(instr('mov', 'r8, #6')),
            ([], ['r8'], [6]))
        self.assertEqual(self.arch.parseArguments(instr('mov', 'ip, #0')),
            ([], ['ip'], [0]))
        self.assertEqual(self.arch.parseArguments(instr('movle', 'r0, #1')),
            ([], ['r0'], [1]))
        self.assertEqual(self.arch.parseArguments(instr('cmp', 'r0, #1')),
            (['r0'], [], [1]))
        self.assertEqual(self.arch.parseArguments(instr('cmn', 'r0, #1')),
            (['r0'], [], [1]))
        self.assertEqual(self.arch.parseArguments(instr('tst', 'r0, #32768')),
            (['r0'], [], [32768]))

        #add, sub
        self.assertEqual(self.arch.parseArguments(instr('sub', 'sp, sp, #4')),
            (['sp'], ['sp'], [4]))
        self.assertEqual(self.arch.parseArguments(instr('add', 'sp, sp, #4')),
            (['sp'], ['sp'], [4]))
        self.assertEqual(self.arch.parseArguments(instr('add', 'r2, r2, #1')),
            (['r2'], ['r2'], [1]))
        self.assertEqual(self.arch.parseArguments(instr('addlt', 'r5, r5, r3')),
            (['r5', 'r3'], ['r5'], []))
        self.assertEqual(self.arch.parseArguments(instr('sub', 'r3, r3, r0, lsl #2')),
            (['r3', 'r0'], ['r3'], []))

        #shifts
        self.assertEqual(self.arch.parseArguments(instr('lsl', 'r1, r3, #7')),
            (['r3'], ['r1'], [7]))
        self.assertEqual(self.arch.parseArguments(instr('asr', 'r1, r2, #31')),
            (['r2'], ['r1'], [31]))
        self.assertEqual(self.arch.parseArguments(instr('rsb', 'r0, r1, r0, asr #9')),
            (['r1', 'r0'], ['r0'], []))
        self.assertEqual(self.arch.parseArguments(instr('lsl', 'r1, r3, #7')),
            (['r3'], ['r1'], [7]))

        #branches
        self.assertEqual(self.arch.parseArguments(instr('bl', '0')),
            ([], ['pc', 'lr'], []))
        self.assertEqual(self.arch.parseArguments(instr('bx', 'lr')),
            (['lr'], ['pc'], []))
        self.assertEqual(self.arch.parseArguments(instr('ble', '4c')),
            ([], ['pc'], []))

        #dynamic load
        self.assertEqual(self.arch.parseArguments(instr('ldr', 'r3, [ip]')),
            (['ip'], ['r3'], []))
        self.assertEqual(self.arch.parseArguments(instr('ldr', 'r3, [pc, #8]')),
            (['pc'], ['r3'], [8]))
        self.assertEqual(self.arch.parseArguments(instr('ldr', 'r3, [r2, r0]')),
            (['r2', 'r0'], ['r3'], []))
        self.assertEqual(self.arch.parseArguments(instr('ldrb', 'r3, [lr, #1]')),
            (['lr'], ['r3'], [1]))
        self.assertEqual(self.arch.parseArguments(instr('ldm', 'r0, {r0, r1}')),
            (['r0'], ['r0', 'r1'], []))
        self.assertEqual(self.arch.parseArguments(instr('ldrh', 'r3, [r1, r4]')),
            (['r1', 'r4'], ['r3'], []))

        #dynamic store
        self.assertEqual(self.arch.parseArguments(instr('str', 'r0, [r3]')),
            (['r0', 'r3'], [], []))
        self.assertEqual(self.arch.parseArguments(instr('str', 'r0, [r4, r5]')),
            (['r0', 'r4', 'r5'], [], []))
        self.assertEqual(self.arch.parseArguments(instr('strb', 'r0, [r4, #3]')),
            (['r0', 'r4'], [], []))
        self.assertEqual(self.arch.parseArguments(instr('stm', 'sp, {r2, r3}')),
            (['sp', 'r2', 'r3'], [], []))
        self.assertEqual(self.arch.parseArguments(instr('strh', 'lr, [r2, fp]')),
            (['lr', 'r2', 'fp'], [], []))
        #postindexing, immediate
        self.assertEqual(self.arch.parseArguments(instr('str', 'r0, [r3], #4')),
            (['r0', 'r3'], ['r3'], [4]))
        #postindexing, register
        self.assertEqual(self.arch.parseArguments(instr('str', 'r0, [r3], r2')),
            (['r0', 'r3', 'r2'], ['r3'], []))

        #unimplemented instruction type
        self.assertRaises(ValueError, self.arch.parseArguments, instr('abemad', 'r1, r0'))

        #multiplication
        self.assertEqual(self.arch.parseArguments(instr('smull', 'r1, r0, r3, r2')),
            (['r3', 'r2'], ['r1', 'r0'], []))

        #multiplication
        self.assertEqual(self.arch.parseArguments(instr('mul', 'r0, r3, r2')),
            (['r3', 'r2'], ['r0'], []))
        self.assertEqual(self.arch.parseArguments(instr('mla', 'r1, r2, r3, r1')),
            (['r2', 'r3', 'r1'], ['r1'], []))

        #push, pop
        self.assertEqual(self.arch.parseArguments(instr('push', '{lr}')),
            (['sp', 'lr'], ['sp'], []))
        self.assertEqual(self.arch.parseArguments(instr('push', '{r4, r5, lr}')),
            (['sp', 'r4', 'r5', 'lr'], ['sp'], []))
        self.assertEqual(self.arch.parseArguments(instr('pop', '{r4, r5, lr}')),
            (['sp'], ['sp', 'r4', 'r5', 'lr'], []))
        self.assertEqual(self.arch.parseArguments(instr('pop', '{lr}')),
            (['sp'], ['sp', 'lr'], []))

        #load, store multiple
        self.assertEqual(self.arch.parseArguments(instr('stmdb', 'r6, {r0, lr}')),
            (['r6', 'r0', 'lr'], ['r6'], []))
        self.assertEqual(self.arch.parseArguments(instr('ldmib', 'sp, {r2, r3}')),
            (['sp'], ['sp', 'r2', 'r3'], []))

        #misc
        self.assertEqual(self.arch.parseArguments(instr('and', 'r3, r3, #31')),
            (['r3'], ['r3'], [31]))
        self.assertEqual(self.arch.parseArguments(instr('muls', 'r3, r4, r3')),
            (['r4', 'r3'], ['r3'], []))
        self.assertEqual(self.arch.parseArguments(instr('eor', 'r2, r2, r0')),
            (['r2', 'r0'], ['r2'], []))
        self.assertEqual(self.arch.parseArguments(instr('eors', 'r2, r2, r0')),
            (['r2', 'r0'], ['r2'], []))
        self.assertEqual(self.arch.parseArguments(instr('orr', 'r2, r2, r0')),
            (['r2', 'r0'], ['r2'], []))
        self.assertEqual(self.arch.parseArguments(instr('orrs', 'r2, r2, r0')),
            (['r2', 'r0'], ['r2'], []))
        self.assertEqual(self.arch.parseArguments(instr('teq', 'ip, r0')),
            (['ip', 'r0'], [], []))
コード例 #3
0
class TestArmArch(unittest.TestCase):
    def setUp(self):
        self.arch = ArmArchitecture()
        self.assertNotEqual(self.arch, None)

    def test_arg_parsing(self):
        self.assertEqual(self.arch.parseArguments(instr('mov', 'r8, #6')),
                         ([], ['r8'], [6]))
        self.assertEqual(self.arch.parseArguments(instr('mov', 'ip, #0')),
                         ([], ['ip'], [0]))
        self.assertEqual(self.arch.parseArguments(instr('movle', 'r0, #1')),
                         ([], ['r0'], [1]))
        self.assertEqual(self.arch.parseArguments(instr('cmp', 'r0, #1')),
                         (['r0'], [], [1]))
        self.assertEqual(self.arch.parseArguments(instr('cmn', 'r0, #1')),
                         (['r0'], [], [1]))
        self.assertEqual(self.arch.parseArguments(instr('tst', 'r0, #32768')),
                         (['r0'], [], [32768]))

        #add, sub
        self.assertEqual(self.arch.parseArguments(instr('sub', 'sp, sp, #4')),
                         (['sp'], ['sp'], [4]))
        self.assertEqual(self.arch.parseArguments(instr('add', 'sp, sp, #4')),
                         (['sp'], ['sp'], [4]))
        self.assertEqual(self.arch.parseArguments(instr('add', 'r2, r2, #1')),
                         (['r2'], ['r2'], [1]))
        self.assertEqual(
            self.arch.parseArguments(instr('addlt', 'r5, r5, r3')),
            (['r5', 'r3'], ['r5'], []))
        self.assertEqual(
            self.arch.parseArguments(instr('sub', 'r3, r3, r0, lsl #2')),
            (['r3', 'r0'], ['r3'], []))

        #shifts
        self.assertEqual(self.arch.parseArguments(instr('lsl', 'r1, r3, #7')),
                         (['r3'], ['r1'], [7]))
        self.assertEqual(self.arch.parseArguments(instr('asr', 'r1, r2, #31')),
                         (['r2'], ['r1'], [31]))
        self.assertEqual(
            self.arch.parseArguments(instr('rsb', 'r0, r1, r0, asr #9')),
            (['r1', 'r0'], ['r0'], []))
        self.assertEqual(self.arch.parseArguments(instr('lsl', 'r1, r3, #7')),
                         (['r3'], ['r1'], [7]))

        #branches
        self.assertEqual(self.arch.parseArguments(instr('bl', '0')),
                         ([], ['pc', 'lr'], []))
        self.assertEqual(self.arch.parseArguments(instr('bx', 'lr')),
                         (['lr'], ['pc'], []))
        self.assertEqual(self.arch.parseArguments(instr('ble', '4c')),
                         ([], ['pc'], []))

        #dynamic load
        self.assertEqual(self.arch.parseArguments(instr('ldr', 'r3, [ip]')),
                         (['ip'], ['r3'], []))
        self.assertEqual(
            self.arch.parseArguments(instr('ldr', 'r3, [pc, #8]')),
            (['pc'], ['r3'], [8]))
        self.assertEqual(
            self.arch.parseArguments(instr('ldr', 'r3, [r2, r0]')),
            (['r2', 'r0'], ['r3'], []))
        self.assertEqual(
            self.arch.parseArguments(instr('ldrb', 'r3, [lr, #1]')),
            (['lr'], ['r3'], [1]))
        self.assertEqual(
            self.arch.parseArguments(instr('ldm', 'r0, {r0, r1}')),
            (['r0'], ['r0', 'r1'], []))
        self.assertEqual(
            self.arch.parseArguments(instr('ldrh', 'r3, [r1, r4]')),
            (['r1', 'r4'], ['r3'], []))

        #dynamic store
        self.assertEqual(self.arch.parseArguments(instr('str', 'r0, [r3]')),
                         (['r0', 'r3'], [], []))
        self.assertEqual(
            self.arch.parseArguments(instr('str', 'r0, [r4, r5]')),
            (['r0', 'r4', 'r5'], [], []))
        self.assertEqual(
            self.arch.parseArguments(instr('strb', 'r0, [r4, #3]')),
            (['r0', 'r4'], [], []))
        self.assertEqual(
            self.arch.parseArguments(instr('stm', 'sp, {r2, r3}')),
            (['sp', 'r2', 'r3'], [], []))
        self.assertEqual(
            self.arch.parseArguments(instr('strh', 'lr, [r2, fp]')),
            (['lr', 'r2', 'fp'], [], []))
        #postindexing, immediate
        self.assertEqual(
            self.arch.parseArguments(instr('str', 'r0, [r3], #4')),
            (['r0', 'r3'], ['r3'], [4]))
        #postindexing, register
        self.assertEqual(
            self.arch.parseArguments(instr('str', 'r0, [r3], r2')),
            (['r0', 'r3', 'r2'], ['r3'], []))

        #unimplemented instruction type
        self.assertRaises(ValueError, self.arch.parseArguments,
                          instr('abemad', 'r1, r0'))

        #multiplication
        self.assertEqual(
            self.arch.parseArguments(instr('smull', 'r1, r0, r3, r2')),
            (['r3', 'r2'], ['r1', 'r0'], []))

        #multiplication
        self.assertEqual(self.arch.parseArguments(instr('mul', 'r0, r3, r2')),
                         (['r3', 'r2'], ['r0'], []))
        self.assertEqual(
            self.arch.parseArguments(instr('mla', 'r1, r2, r3, r1')),
            (['r2', 'r3', 'r1'], ['r1'], []))

        #push, pop
        self.assertEqual(self.arch.parseArguments(instr('push', '{lr}')),
                         (['sp', 'lr'], ['sp'], []))
        self.assertEqual(
            self.arch.parseArguments(instr('push', '{r4, r5, lr}')),
            (['sp', 'r4', 'r5', 'lr'], ['sp'], []))
        self.assertEqual(
            self.arch.parseArguments(instr('pop', '{r4, r5, lr}')),
            (['sp'], ['sp', 'r4', 'r5', 'lr'], []))
        self.assertEqual(self.arch.parseArguments(instr('pop', '{lr}')),
                         (['sp'], ['sp', 'lr'], []))

        #load, store multiple
        self.assertEqual(
            self.arch.parseArguments(instr('stmdb', 'r6, {r0, lr}')),
            (['r6', 'r0', 'lr'], ['r6'], []))
        self.assertEqual(
            self.arch.parseArguments(instr('ldmib', 'sp, {r2, r3}')),
            (['sp'], ['sp', 'r2', 'r3'], []))

        #misc
        self.assertEqual(self.arch.parseArguments(instr('and', 'r3, r3, #31')),
                         (['r3'], ['r3'], [31]))
        self.assertEqual(self.arch.parseArguments(instr('muls', 'r3, r4, r3')),
                         (['r4', 'r3'], ['r3'], []))
        self.assertEqual(self.arch.parseArguments(instr('eor', 'r2, r2, r0')),
                         (['r2', 'r0'], ['r2'], []))
        self.assertEqual(self.arch.parseArguments(instr('eors', 'r2, r2, r0')),
                         (['r2', 'r0'], ['r2'], []))
        self.assertEqual(self.arch.parseArguments(instr('orr', 'r2, r2, r0')),
                         (['r2', 'r0'], ['r2'], []))
        self.assertEqual(self.arch.parseArguments(instr('orrs', 'r2, r2, r0')),
                         (['r2', 'r0'], ['r2'], []))
        self.assertEqual(self.arch.parseArguments(instr('teq', 'ip, r0')),
                         (['ip', 'r0'], [], []))
コード例 #4
0
 def setUp(self):
     self.arch = ArmArchitecture()
     self.assertNotEqual(self.arch, None)
コード例 #5
0
 def __init__(self):
     self.baseAddress = 0
     self.arch = ArmArchitecture()