manticore/tests/test_armv7rf.py
Disconnect3d 9c19014e4b Fix duplicated armv7 test name (#833)
* Fix duplicate armv7rf test name

* Small refactor of armv7rf

* Proper test classes names
2018-03-21 15:34:19 -04:00

76 lines
2.5 KiB
Python

import unittest
from manticore.core.cpu.arm import Armv7RegisterFile as RF
class Armv7RFTest(unittest.TestCase):
_multiprocess_can_split_ = True
def setUp(self):
self.r = RF()
def test_init_state(self):
self.assertEqual(self.r.read('R0'), 0)
def test_write_read(self):
self.r.write('R0', 1)
self.assertEqual(self.r.read('R0'), 1)
def test_write_read_sp(self):
self.r.write('SP', 1)
self.assertEqual(self.r.read('SP'), 1)
def test_flag_wr(self):
self.r.write('APSR_Z', True)
self.assertEqual(self.r.read('APSR_Z'), True)
def test_flag_wr_f(self):
self.r.write('APSR_Z', False)
self.assertEqual(self.r.read('APSR_Z'), False)
def test_bad_reg_name(self):
with self.assertRaises(AssertionError):
nonexistant_reg = "Pc"
self.r.read(nonexistant_reg)
def test_flag_wr_aspr(self):
self.r.write('APSR', 0xffffffff)
self.assertEqual(self.r.read('APSR'), 0xf0000000) # 4 more significant bits used
self.assertEqual(self.r.read('APSR_V'), True)
self.assertEqual(self.r.read('APSR_C'), True)
self.assertEqual(self.r.read('APSR_Z'), True)
self.assertEqual(self.r.read('APSR_N'), True)
self.r.write('APSR_N', False)
self.assertEqual(self.r.read('APSR'), 0x70000000)
self.r.write('APSR_Z', False)
self.assertEqual(self.r.read('APSR'), 0x30000000)
self.r.write('APSR_C', False)
self.assertEqual(self.r.read('APSR'), 0x10000000)
self.r.write('APSR_V', False)
self.assertEqual(self.r.read('APSR'), 0x00000000)
def test_register_independence_wr(self):
regs = ('R0', 'R1', 'R2', 'R3', 'R4', 'R5', 'R6', 'R7', 'R8',
'R9', 'R10', 'R11', 'R12', 'R13', 'R14', 'R15')
aliases = {'SB': 'R9', 'SL': 'R10', 'FP': 'R11', 'IP': 'R12', 'STACK': 'R13', 'SP': 'R13', 'LR': 'R14',
'PC': 'R15'}
for j in xrange(16):
for i in xrange(16):
if i == j:
self.r.write(regs[i], 0x41424344)
else:
self.r.write(regs[i], 0)
for a, b in aliases.items():
self.assertEqual(self.r.read(a), self.r.read(b))
for i in xrange(16):
if i == j:
self.assertEqual(self.r.read(regs[i]), 0x41424344)
else:
self.assertEqual(self.r.read(regs[i]), 0x00000000)