def contents(*tags, **location):
"""Iterate through the specified contents `tags` within the database using the cache.
Each iteration yields a tuple of the format `(location, tags)` where
`location` can be either an address or a chunk identifier and offset
depending on whether `location` was specified as true or not.
"""
global export
location = location.get('location', False)
iterable = db.selectcontents(Or=tags) if tags else db.selectcontents()
for F, res in iterable:
for loc, res in export.content(F, *res, location=location):
if res: yield loc, res
continue
return
def contents(*tags, **location):
"""Iterate through the specified contents `tags` within the database using the cache.
Each iteration yields a tuple of the format `(location, tags)` where
`location` can be either an address or a chunk identifier and offset
depending on whether `location` was specified as true or not.
"""
global export
location = location.get('location', False)
iterable = db.selectcontents(Or=tags) if tags else db.selectcontents()
for F, res in iterable:
for loc, res in export.content(F, *res, location=location):
if res: yield loc, res
continue
return
def cached():
'''Return all tags using the database's tag cache as (globals, contents).'''
print >> output, '--> Grabbing globals (cached)...'
g = {ea: t for ea, t in db.select()}
print >> output, '--> Grabbing contents from all functions (cached)...'
res = itertools.starmap(fn.select, db.selectcontents())
f = {ea: t for ea, t in itertools.chain(*res)}
return (g, f)
def breakpoints(f=None, **kwargs):
"""Query the function `f` for the "break" tag, and use it to emit a list of breakpoints for windbg.
If the string `module` is specified, then use it instead of the current filename when emitting the location.
If the string `tagname` is provided, then use it to query instead of "break".
"""
tagname = kwargs.get('tagname', 'break')
# if no function was provided, then recurse into ourself for all of them
if f is None:
for f, _ in db.selectcontents(tagname):
breakpoints(f, **kwargs)
return
#entry, exit = func.top(f), func.bottom(f)
#funcname = func.name(f)
#[(entry,{tagname:'.printf "Entering {:s} %x,%x\\n",poi(@esp),@esp'.format(funcname)})], [(x,{tagname:'.printf "Exiting {:s} %x,%x\\n",poi(@esp),@esp'.format(funcname)}) for x in exit],
# query the given function for the requested tagname
tags, select = {}, itertools.chain(func.select(f, And=(tagname,), Or=('',)))
for ea, t in select:
h = tags.setdefault(ea, {})
for k in t.keys():
if k == tagname:
h.setdefault(k, []).extend(t[k] if isinstance(t[k], builtins.list) else t[k].split(';'))
elif operator.contains(h, k):
logging.warning(u"{:s}.breakpoints({:#x}{:s}) : The specified key \"{:s}\" already exists in dictionary for address {:#x}.".format(__name__, func.addr(f), u", {:s}".format(utils.strings.kwargs(kwargs)) if kwargs else '', utils.string.escape(k, '"'), ea))
else:
h[k] = t[k]
continue
continue
# aggregate all of the discovered tags into a list of breakpoints
for ea, t in tags.items():
ofs, commands = db.offset(ea), []
# create the command that emits the current label
label_t = string.Template(r'.printf "$label -- $note\n"' if operator.contains(t, '') else r'.printf "$label\n"')
commands.append(label_t.safe_substitute(label=label(ea), note=t.get('', '')))
# append the commands to execute when encountering the given breakpoint
res = t.get(tagname, ['g'])
if isinstance(res, builtins.list):
commands.extend(res)
else:
commands.append(res)
# escape all of the commands since we're going to join them together
commands = (escape(cmd) for cmd in commands)
six.print_('bp {:s} "{:s}"'.format(reference(ea, **kwargs), escape(';'.join(commands), depth=1)))
return
def dump_breaks(func=None, tagname='break', stdout=True):
if func is None:
for func, _ in db.selectcontents(tagname):
dump_breaks(func, tagname=tagname)
return
Escape = lambda s: s.replace('"', '\\"')
entry, exit = fn.top(func), fn.bottom(func)
funcname = fn.name(func)
#[(entry,{tagname:'.printf "Entering {:s} %x,%x\\n",poi(@esp),@esp'.format(funcname)})], [(x,{tagname:'.printf "Exiting {:s} %x,%x\\n",poi(@esp),@esp'.format(funcname)}) for x in exit],
select = itertools.chain(fn.select(func, And=(tagname, ), Or=('', )))
res = {}
for ea, t in select:
h = res.setdefault(ea, {})
for k in t.keys():
if k == tagname:
h.setdefault(k, []).extend(t[k].split(';'))
else:
assert k not in h
h[k] = t[k]
continue
output = []
for ea, t in res.iteritems():
ofs = db.offset(ea)
commands = []
label = Template('.printf "$label -- $note\\n"' if t.
has_key('') else '.printf "$label\\n"')
commands.append(
label.safe_substitute(label=eaToLabel(ea), note=t.get('', '')))
commands.extend(t.get(tagname, ['g']))
commands = map(windbgescape, commands)
breakpoint = 'bp {:s} "{:s}"'.format(eaToReference(ea),
Escape(';'.join(commands)))
if stdout:
print(breakpoint)
output.append(breakpoint)
if len(output) == 1:
return output[0] + '\n'
return '\n'.join(output)
def list():
'''Return all the contents tags within the database as a set.'''
return {t for t in itertools.chain(*(t for _, t in db.selectcontents()))}
def contents(*tags, **boolean):
for res in db.selectcontents(*tags, **boolean):
for ea, res in fn.select(*res):
yield ea, res
continue
return
def list():
'''Return the tags for the all of the function contents within the database as a set.'''
return {
res
for res in itertools.chain(*(res for _, res in db.selectcontents()))
}
def list():
'''Return the tags for the all of the function contents within the database as a set.'''
return {res for res in itertools.chain(*(res for _, res in db.selectcontents()))}