mirror of
https://github.com/ivellioscolin/pykd.git
synced 2025-04-21 21:03:23 +08:00
101 lines
2.9 KiB
Python
101 lines
2.9 KiB
Python
|
|
from pykd import *
|
|
import sys
|
|
|
|
def printThreadInfo():
|
|
|
|
nt = module("nt")
|
|
|
|
thread = nt.typedVar( "_KTHREAD", getImplicitThread() )
|
|
|
|
stackPointer = addr64( reg("rsp") if is64bitSystem() else reg("esp") )
|
|
|
|
dprintln("")
|
|
|
|
dprintln( "Stack Base: %x Limit: %x Current: %x Used: %x Unused: %x" %
|
|
( thread.InitialStack, thread.StackLimit, stackPointer, thread.InitialStack - stackPointer, stackPointer - thread.StackLimit ) )
|
|
|
|
|
|
def printDeltaStat():
|
|
|
|
dprintln("")
|
|
|
|
printThreadInfo()
|
|
|
|
stk = getStack()
|
|
|
|
moduleLst = {}
|
|
funcLst = {}
|
|
|
|
for i in range( 0, len(stk) -1 ):
|
|
|
|
try:
|
|
mod = module( stk[i].returnOffset )
|
|
except BaseException:
|
|
continue
|
|
|
|
delta = stk[i+1].frameOffset - stk[i].frameOffset
|
|
if delta > 0:
|
|
|
|
moduleName = mod.name()
|
|
|
|
if moduleName in moduleLst:
|
|
moduleLst[moduleName] = moduleLst[moduleName] + delta
|
|
else:
|
|
moduleLst[moduleName] = delta
|
|
|
|
func = moduleName + "!" + mod.findSymbol( stk[i].returnOffset, showDisplacement = False )
|
|
|
|
if func in funcLst:
|
|
funcLst[func] = funcLst[func] + delta
|
|
else:
|
|
funcLst[func] = delta
|
|
|
|
nt = module("nt")
|
|
|
|
thread = nt.typedVar( "_KTHREAD", getImplicitThread() )
|
|
|
|
stackSize = thread.InitialStack - thread.StackLimit
|
|
|
|
dprintln("")
|
|
dprintln( "%12s\t%s" % ("Stack usage:", "Module:" ) )
|
|
dprintln( "="*30 )
|
|
|
|
for mod,delta in sorted( moduleLst.iteritems(), key=lambda x: x[1], reverse=True ):
|
|
dprintln( "%7d%5s\t%s" % (delta, "(%%%d)" % (delta*100/stackSize), mod ) )
|
|
|
|
dprintln("")
|
|
dprintln( "%12s\t%s" % ("Stack usage:", "Function" ) )
|
|
dprintln( "="*30 )
|
|
|
|
for func,delta in sorted( funcLst.iteritems(), key=lambda x: x[1], reverse=True ):
|
|
dprintln( "%7d%5s\t%s" % (delta, "(%%%d)" % (delta*100/stackSize), func ) )
|
|
|
|
def printDeltaStack():
|
|
|
|
printThreadInfo()
|
|
|
|
stk = getStack()
|
|
|
|
dprintln( "Stack Delta:\tFunction:")
|
|
|
|
for i in range( 0, len(stk) -1 ):
|
|
dprint( "%12s\t" % long( stk[i+1].frameOffset - stk[i].frameOffset) )
|
|
try:
|
|
mod = module( stk[i].returnOffset )
|
|
dprintln( "%s!%s"% ( mod.name(), mod.findSymbol( stk[i].returnOffset, showDisplacement = False ) ) )
|
|
except BaseException:
|
|
dprintln( findSymbol( stk[i].returnOffset ) )
|
|
|
|
def main():
|
|
|
|
if len(sys.argv) > 1:
|
|
if sys.argv[1] == "stat":
|
|
printDeltaStat()
|
|
return
|
|
|
|
printDeltaStack()
|
|
|
|
if __name__ == "__main__":
|
|
main()
|