41 lines
1.1 KiB
Python
Executable File
41 lines
1.1 KiB
Python
Executable File
#!/usr/bin/python
|
|
# By Ryan Hileman, issue #9
|
|
|
|
# this prints out 2 lines and the contents must be the same
|
|
|
|
from unicorn import *
|
|
import regress
|
|
|
|
class MemMap(regress.RegressTest):
|
|
|
|
def test_mmap_write(self):
|
|
uc = Uc(UC_ARCH_X86, UC_MODE_64)
|
|
|
|
uc.mem_map(0x8048000, 0x2000)
|
|
uc.mem_write(0x8048000, 'test')
|
|
s1 = str(uc.mem_read(0x8048000, 4)).encode('hex')
|
|
|
|
self.assertEqual('test'.encode('hex'), s1)
|
|
|
|
uc.mem_map(0x804a000, 0x8000)
|
|
s2 = str(uc.mem_read(0x8048000, 4)).encode('hex')
|
|
self.assertEqual(s1, s2)
|
|
|
|
def test_mmap_invalid(self):
|
|
u = unicorn.Uc(unicorn.UC_ARCH_X86, unicorn.UC_MODE_32)
|
|
with self.assertRaises(UcError):
|
|
u.mem_map(0x2000, 0)
|
|
with self.assertRaises(UcError):
|
|
u.mem_map(0x4000, 1)
|
|
|
|
def test_mmap_weird(self):
|
|
u = unicorn.Uc(unicorn.UC_ARCH_X86, unicorn.UC_MODE_32)
|
|
|
|
for i in xrange(20):
|
|
with self.assertRaises(UcError):
|
|
u.mem_map(i*0x1000, 5)
|
|
u.mem_read(i*0x1000+6, 1)
|
|
|
|
if __name__ == '__main__':
|
|
regress.main()
|