78 lines
2.5 KiB
Python
78 lines
2.5 KiB
Python
import unittest
|
|
|
|
from manticore.core.cpu.arm import Armv7RegisterFile as RF
|
|
from manticore.core.cpu.arm import *
|
|
|
|
from capstone.arm import *
|
|
|
|
class Armv7RF(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(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 )
|
|
|