def setUp(self):
self.arch = ArmArchitecture()
self.assertNotEqual(self.arch, None)
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'], [], []))
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'], [], []))
def setUp(self):
self.arch = ArmArchitecture()
self.assertNotEqual(self.arch, None)
def __init__(self):
self.baseAddress = 0
self.arch = ArmArchitecture()