mirror of
https://github.com/vivisect/vivisect
synced 2026-06-08 18:04:23 +00:00
invisigoth
64 lines
2.0 KiB
Python
64 lines
2.0 KiB
Python
import os
|
|
import unittest
|
|
import vtrace.tests as vt_tests
|
|
|
|
class VtraceBasicTest(vt_tests.VtraceProcessTest):
|
|
|
|
breakpoints = {
|
|
'windows':'ntdll.NtTerminateProcess',
|
|
'linux':'libc.exit',
|
|
'freebsd':'libc.exit',
|
|
}
|
|
|
|
def test_vtrace_getregisters(self):
|
|
self.assertTrue( self.trace.getProgramCounter() )
|
|
self.assertTrue( self.trace.getStackCounter() )
|
|
self.assertTrue( self.trace.getRegisters() )
|
|
self.runUntilExit()
|
|
|
|
def test_vtrace_setregisters(self):
|
|
pc = self.trace.getProgramCounter()
|
|
sp = self.trace.getStackCounter()
|
|
regs = self.trace.getRegisters()
|
|
self.trace.setStackCounter(sp)
|
|
self.trace.setProgramCounter(pc)
|
|
self.trace.setRegisters(regs)
|
|
self.runUntilExit()
|
|
|
|
def test_vtrace_breakpoint(self):
|
|
plat = self.trace.getMeta('Platform')
|
|
symname = self.breakpoints.get(plat)
|
|
if symname == None:
|
|
raise unittest.SkipTest('no platform breakpoint: %s' % plat)
|
|
|
|
pycode = 'trace.setMeta("testbphit", 1)'
|
|
bpid = self.trace.addBreakByExpr( symname )
|
|
self.trace.setBreakpointCode(bpid, pycode)
|
|
self.runUntilExit()
|
|
|
|
bp = self.trace.getBreakpoint(bpid)
|
|
|
|
self.assertTrue(bp.address != None)
|
|
self.assertTrue(self.trace.getMeta('testbphit'))
|
|
|
|
def test_vtrace_exename(self):
|
|
exename = self.trace.getExe()
|
|
self.assertTrue( os.path.isfile( exename ) )
|
|
self.assertTrue( exename.lower().find('python') != -1)
|
|
|
|
def test_vtrace_mmaps(self):
|
|
pymapfound = False
|
|
for va,size,perms,fname in self.trace.getMemoryMaps():
|
|
if fname.lower().find('python') != -1:
|
|
pymapfound = True
|
|
break
|
|
self.assertTrue( pymapfound )
|
|
|
|
# All of the above "simple" tests should also work in the "exec" case
|
|
class VtraceBasicExecTest(VtraceBasicTest, vt_tests.VtraceExecTest):
|
|
breakpoints = {
|
|
'windows':'ntdll.NtTerminateProcess',
|
|
'linux':'ld.malloc',
|
|
'freebsd':'ld._rtld_thread_init',
|
|
}
|