jeb-samplecode/scripts/JavaASTDemo.py at master · pnfsoftware/jeb-samplecode · GitHub

Repository navigation

24 lines (22 loc) · 765 Bytes

#?description=Show how to navigate and dump Java AST trees

#?shortcut=

from com.pnfsoftware.jeb.client.api import IScript

from com.pnfsoftware.jeb.core.units.code.java import IJavaSourceUnit, IJavaClass

"""

Sample script for JEB Decompiler.

"""

class JavaASTDemo(IScript):

def run(self, ctx):

prj = ctx.getMainProject()

assert prj, 'Need a project'

for unit in prj.findUnits(IJavaSourceUnit):

elt = unit.getASTElement()

if isinstance(elt, IJavaClass):

for m in elt.getMethods():

self.displayTree(m)

def displayTree(self, e, level=0):

print('%s%s @ 0x%X' % (level*' ', e.getElementType(), e.getPhysicalOffset()))

if e:

elts = e.getSubElements()

for e in elts:

self.displayTree(e, level+1)