mirror of
https://github.com/ivellioscolin/pykd.git
synced 2025-04-20 19:53:22 +08:00
53 lines
1.6 KiB
Python
53 lines
1.6 KiB
Python
"""Debug events handler: test [un-]load modules notification"""
|
|
|
|
import unittest
|
|
import target
|
|
import pykd
|
|
import fnmatch
|
|
import testutils
|
|
|
|
class ModuleLoadHandler(pykd.eventHandler):
|
|
"""Track load/unload module implementation"""
|
|
def __init__(self, moduleMask):
|
|
pykd.eventHandler.__init__(self)
|
|
|
|
self.moduleMask = moduleMask.lower()
|
|
|
|
self.wasLoad = 0
|
|
self.wasUnload = False
|
|
|
|
def onLoadModule(self, module):
|
|
"""Load module handler"""
|
|
|
|
if ( fnmatch.fnmatch(module.name().lower(), self.moduleMask) ):
|
|
self.wasLoad = module.begin()
|
|
|
|
return pykd.DEBUG_STATUS_NO_CHANGE
|
|
|
|
def onUnloadModule(self, modBase):
|
|
"""Unload module handler"""
|
|
|
|
if ( self.wasLoad and (self.wasLoad == modBase) ):
|
|
self.wasUnload = True
|
|
|
|
return pykd.DEBUG_STATUS_NO_CHANGE
|
|
|
|
class EhLoadTest(unittest.TestCase):
|
|
"""Unit tests of [un-]load modules notification"""
|
|
|
|
def testLoadUnload(self):
|
|
"""Start new process and track loading and unloading modules"""
|
|
pykd.startProcess(target.appPath + " -testLoadUnload")
|
|
with testutils.ContextCallIt( pykd.killProcess ) as contextCallIt:
|
|
modLoadHandler = ModuleLoadHandler( "*Iphlpapi*" )
|
|
with testutils.ContextCallIt( getattr(modLoadHandler, "reset") ) as resetEventHandler:
|
|
try:
|
|
while True:
|
|
pykd.go()
|
|
except pykd.WaitEventException:
|
|
pass
|
|
|
|
self.assertTrue(modLoadHandler.wasLoad)
|
|
self.assertTrue(modLoadHandler.wasUnload)
|
|
|