mirror of
https://github.com/ivellioscolin/pykd.git
synced 2025-04-19 19:13:22 +08:00
113 lines
3.5 KiB
Python
113 lines
3.5 KiB
Python
""" breakpoints """
|
|
|
|
import unittest
|
|
import pykd
|
|
import target
|
|
import testutils
|
|
|
|
class callCounter:
|
|
def __init__(self,func):
|
|
self.count = 0
|
|
self.func = func
|
|
def __call__(self,val):
|
|
self.count = self.count+1
|
|
return self.func(val)
|
|
|
|
def stopOnBreak(id):
|
|
return pykd.Break
|
|
|
|
def continueOnBreak(id):
|
|
return pykd.NoChange
|
|
|
|
|
|
class BreakpointTest( unittest.TestCase ):
|
|
|
|
def testNoBreakpoint(self):
|
|
processId = pykd.startProcess( target.appPath + " breakhandlertest" )
|
|
with testutils.ContextCallIt( testutils.KillProcess(processId) ) as killStartedProcess :
|
|
pykd.go()
|
|
self.assertEqual( pykd.NoDebuggee, pykd.go() )
|
|
|
|
def testSetBp(self):
|
|
processId = pykd.startProcess( target.appPath + " breakhandlertest" )
|
|
targetModule = pykd.module( target.moduleName )
|
|
targetModule.reload()
|
|
with testutils.ContextCallIt( testutils.KillProcess(processId) ) as killStartedProcess :
|
|
pykd.go()
|
|
pykd.setBp( targetModule.CdeclFunc )
|
|
self.assertEqual( pykd.Break, pykd.go() )
|
|
|
|
|
|
def testRemoveBp(self):
|
|
processId = pykd.startProcess( target.appPath + " breakhandlertest" )
|
|
targetModule = pykd.module( target.moduleName )
|
|
targetModule.reload()
|
|
with testutils.ContextCallIt( testutils.KillProcess(processId) ) as killStartedProcess :
|
|
pykd.go()
|
|
bpid = pykd.setBp( targetModule.CdeclFunc )
|
|
pykd.removeBp( bpid )
|
|
self.assertEqual( pykd.NoDebuggee, pykd.go() )
|
|
|
|
def testBreakCallback(self):
|
|
processId = pykd.startProcess( target.appPath + " breakhandlertest" )
|
|
targetModule = pykd.module( target.moduleName )
|
|
targetModule.reload()
|
|
with testutils.ContextCallIt( testutils.KillProcess(processId) ) as killStartedProcess :
|
|
pykd.go()
|
|
|
|
breakCount = callCounter(stopOnBreak)
|
|
|
|
bp = pykd.breakpoint( targetModule.CdeclFunc, breakCount )
|
|
|
|
self.assertEqual( pykd.Break, pykd.go() )
|
|
|
|
self.assertEqual( 1, breakCount.count )
|
|
|
|
def testNoBreakCallback(self):
|
|
processId = pykd.startProcess( target.appPath + " breakhandlertest" )
|
|
targetModule = pykd.module( target.moduleName )
|
|
targetModule.reload()
|
|
with testutils.ContextCallIt( testutils.KillProcess(processId) ) as killStartedProcess :
|
|
pykd.go()
|
|
|
|
breakCount = callCounter(continueOnBreak)
|
|
|
|
bp = pykd.breakpoint( targetModule.CdeclFunc, breakCount )
|
|
|
|
self.assertEqual( pykd.NoDebuggee, pykd.go() )
|
|
|
|
self.assertEqual( 1, breakCount.count )
|
|
|
|
def testBreakpointHandler(self):
|
|
|
|
class BreakpointHandler( pykd.eventHandler ):
|
|
|
|
def __init__(self):
|
|
super(BreakpointHandler, self).__init__()
|
|
self.count = 0
|
|
|
|
def onBreakpoint( self, bpid_):
|
|
self.count = self.count + 1
|
|
|
|
processId = pykd.startProcess( target.appPath + " breakhandlertest" )
|
|
targetModule = pykd.module( target.moduleName )
|
|
targetModule.reload()
|
|
with testutils.ContextCallIt( testutils.KillProcess(processId) ) as killStartedProcess :
|
|
|
|
pykd.go()
|
|
|
|
handler = BreakpointHandler()
|
|
|
|
pykd.setBp( targetModule.CdeclFunc )
|
|
|
|
self.assertEqual( pykd.Break, pykd.go() )
|
|
|
|
self.assertEqual( 1, handler.count )
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|