def iterate(fn):
'''Iterate through all the instructions in each chunk of the specified function'''
for start,end in chunks(fn):
for ea in database.iterate(start, end):
yield ea
continue
return
def iterate(fn):
'''Iterate through all the instructions in each chunk of the specified function'''
for start, end in chunks(fn):
for ea in database.iterate(start, end):
yield ea
continue
return
def iterate(key=None):
'''Iterate through all the instructions in each chunk of the specified function'''
for start,end in chunks(key):
for ea in itertools.ifilter(database.type.isCode, database.iterate(start, end)):
yield ea
continue
return
def __process_functions():
p = ui.progress()
globals = internal.comment.globals.address()
funcs = list(database.functions())
p.update(current=0, max=len(funcs), title="Pre-building tagcache...")
p.open()
for i, fn in enumerate(funcs):
chunks = list(function.chunks(fn))
text = functools.partial("{:x} : Processing function {:d} of {:d} : ({:d} chunk{:s})".format, fn, i, len(funcs))
p.update(current=i)
contents = set(internal.comment.contents.address(fn))
for ci, (l, r) in enumerate(chunks):
p.update(text=text(len(chunks), 's' if len(chunks) != 1 else ''), tooltip="Chunk #{:d} : {:x} - {:x}".format(ci, l, r))
for ea in database.iterate(l, r):
# FIXME: no need to iterate really since we should have
# all of the addresses
for k, v in database.tag(ea).iteritems():
if ea in globals: internal.comment.globals.dec(ea, k)
if ea not in contents: internal.comment.contents.inc(ea, k, target=fn)
continue
continue
continue
p.close()
def stackdelta(left, right):
'''return the minimum,maximum delta of the range of instructions'''
min,max = 0,0
for ea in database.iterate(left,right):
sp = getSpDelta(ea)
min = sp if sp < min else min
max = max if sp < max else sp
return min,max
def stackdelta(left, right):
'''return the minimum,maximum delta of the range of instructions'''
min, max = 0, 0
for ea in database.iterate(left, right):
sp = getSpDelta(ea)
min = sp if sp < min else min
max = max if sp < max else sp
return min, max
def __select(fn, q):
for start,end in chunks(fn):
for ea in database.iterate(start, end):
d = database.tag(ea) # FIXME: bmn noticed .select yielding empty records
if d and q.has(d):
yield ea
continue
continue
return
def __select(fn, q):
for start,end in chunks(fn):
for ea in database.iterate(start, end):
d = database.tag(ea) # FIXME: bmn noticed .select yielding empty records
if d and q.has(d):
yield ea
continue
continue
return
def removing_func_tail(pfn, tail):
global State
if State != state.ready: return
# tail = area_t
for ea in database.iterate(tail.startEA, tail.endEA):
for k in database.tag(ea):
internal.comment.contents.dec(ea, k, target=pfn.startEA)
internal.comment.globals.inc(ea, k)
continue
return
def customnames():
'''Add all custom names defined in the database to the tagcache as "__name__"'''
# FIXME: first delete all the custom names '__name__' tag
left, right = db.range()
for ea in db.iterate(left, right - 1):
fn = idaapi.get_func(ea)
ctx = internal.comment.globals if not fn or fn.startEA == ea else internal.comment.contents
if db.type.has_customname(ea):
ctx.inc(ea, '__name__')
continue
return
def add_func(pfn):
global State
if State != state.ready: return
# convert all globals into contents
for (l,r) in function.chunks(pfn):
for ea in database.iterate(l, r):
for k in database.tag(ea):
internal.comment.globals.dec(ea, k)
internal.comment.contents.inc(ea, k, target=pfn.startEA)
continue
continue
return
def set_func_end(pfn, new_end):
global State
if State != state.ready: return
# new_end has added addresses to function
# replace globals with contents
if new_end > pfn.endEA:
for ea in database.iterate(pfn.endEA, new_end):
for k in database.tag(ea):
internal.comment.globals.dec(ea, k)
internal.comment.contents.inc(ea, k, target=pfn.startEA)
continue
return
# new_end has removed addresses from function
# replace contents with globals
elif new_end < pfn.endEA:
for ea in database.iterate(new_end, pfn.endEA):
for k in database.tag(ea):
internal.comment.contents.dec(ea, k, target=pfn.startEA)
internal.comment.globals.inc(ea, k)
continue
return
return
def do_data():
addr, tags = {}, {}
left, right = db.range()
print 'fetching global tags'
for ea in db.iterate(left, right - 1):
f = idaapi.get_func(ea)
if f is not None: continue
res = db.tag(ea)
#res.pop('name', None)
for k, v in res.iteritems():
addr[ea] = addr.get(ea, 0) + 1
tags[k] = tags.get(k, 0) + 1
continue
return addr, tags
def extracomments():
'''Add all extra cmts in the database to the tagcache as "__name__"'''
# FIXME: first delete all the custom names '__name__' tag
left, right = db.range()
for ea in db.iterate(left, right - 1):
fn = idaapi.get_func(ea)
ctx = internal.comment.contents if fn else internal.comment.globals
count = db.extra.count(ea, idaapi.E_PREV)
if count: [ctx.inc(ea, '__extra_prefix__') for i in xrange(count)]
count = db.extra.count(ea, idaapi.E_NEXT)
if count: [ctx.inc(ea, '__extra_suffix__') for i in xrange(count)]
return
def codeRefs(ea):
fn = top(ea)
resultData,resultCode = [],[]
for l,r in chunks(fn):
for ea in database.iterate(l,r):
if len(database.down(ea)) == 0:
insn = idaapi.ua_mnem(ea)
if insn and insn.startswith('call'):
resultCode.append((ea, 0))
continue
resultData.extend( (ea,x) for x in database.dxdown(ea) )
resultCode.extend( (ea,x) for x in database.cxdown(ea) if not contains(fn,x) )
continue
return resultData,resultCode
def codeRefs(ea):
fn = top(ea)
resultData,resultCode = [],[]
for l,r in chunks(fn):
for ea in database.iterate(l,r):
if len(database.down(ea)) == 0:
insn = idaapi.ua_mnem(ea)
if insn and insn.startswith('call'):
resultCode.append((ea, 0))
continue
resultData.extend( (ea,x) for x in database.dxdown(ea) )
resultCode.extend( (ea,x) for x in database.cxdown(ea) if not contains(fn,x) )
continue
return resultData,resultCode
def del_func(pfn):
global State
if State != state.ready: return
# convert all contents into globals
for (l,r) in function.chunks(pfn):
for ea in database.iterate(l, r):
for k in database.tag(ea):
internal.comment.contents.dec(ea, k, target=pfn.startEA)
internal.comment.globals.inc(ea, k)
continue
continue
# remove all function tags
for k in function.tag(pfn.startEA):
internal.comment.globals.dec(pfn.startEA, k)
return
SetDelayed = pyml.function("SetDelayed")
Set = pyml.function("Set")
List = pyml.function("List")
Hold = pyml.function("Hold")
RuleDelayed = pyml.function("RuleDelayed")
Rule = pyml.function("Rule")
call = pyml.function("call")
symcall = pyml.function("symcall")
rules = []
max = len(database.functions())
for i,ea in enumerate(database.functions()):
result = []
for l,r in function.chunks(ea):
for x in database.iterate(l,r):
address,size,insn = x,idc.ItemSize(x),idc.GetDisasm(x)
insn = re.sub(" +", " ", insn)
insn = insn.replace("short", "")
insn = insn.replace("dword", "")
insn = insn.replace("offset", "")
insn = insn.replace("large", "")
insn = insn.replace("ptr", "")
insn = insn.replace("[", "")
insn = insn.replace("]", "")
if " " in insn:
insn = insn.replace(" ", "|", 1)
insn = insn.replace(",", "|")
insn = insn.replace(" ", "")
fn = insn.split("|")
name = fn.pop(0)
