You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
pwndbg/pwndbg/commands/hexdump.py

83 lines
3.0 KiB
Python

#!/usr/bin/env python
# -*- coding: utf-8 -*-
import argparse
import gdb
import pwndbg.arch
import pwndbg.commands
import pwndbg.config
import pwndbg.hexdump
import pwndbg.memory
import pwndbg.regs
pwndbg.config.Parameter('hexdump-width',
16,
'line width of hexdump command')
pwndbg.config.Parameter('hexdump-bytes',
64,
'number of bytes printed by hexdump command')
pwndbg.config.Parameter('hexdump-group-width',
4,
"number of bytes grouped in hexdump command (If -1, the architecture's pointer size is used)")
pwndbg.config.Parameter('hexdump-group-use-big-endian',
False,
'Use big-endian within each group of bytes. Only applies to raw bytes, not the ASCII part. '
'See also hexdump-highlight-group-lsb.')
def address_or_module_name(s):
gdbval_or_str = pwndbg.commands.sloppy_gdb_parse(s)
if isinstance(gdbval_or_str, str):
module_name = gdbval_or_str
pages = list(filter(lambda page: module_name in page.objfile, pwndbg.vmmap.get()))
if pages:
return pages[0].vaddr
else:
raise argparse.ArgumentError('Could not find pages for module %s' % module_name)
elif isinstance(gdbval_or_str, (int, gdb.Value)):
addr = gdbval_or_str
return addr
else:
raise argparse.ArgumentTypeError('Unknown hexdump argument type.')
parser = argparse.ArgumentParser(description='Hexdumps data at the specified address or module name (or at $sp)')
parser.add_argument('address', type=address_or_module_name, nargs='?', default='$sp',
help='Address or module name to dump')
parser.add_argument('count', nargs='?', default=pwndbg.config.hexdump_bytes,
help='Number of bytes to dump')
@pwndbg.commands.ArgparsedCommand(parser)
@pwndbg.commands.OnlyWhenRunning
def hexdump(address=None, count=pwndbg.config.hexdump_bytes):
if hexdump.repeat:
address = hexdump.last_address
hexdump.offset += 1
else:
hexdump.offset = 0
address = int(address)
address &= pwndbg.arch.ptrmask
count = max(int(count), 0)
width = int(pwndbg.config.hexdump_width)
group_width = int(pwndbg.config.hexdump_group_width)
group_width = pwndbg.typeinfo.ptrsize if group_width == -1 else group_width
flip_group_endianess = pwndbg.config.hexdump_group_use_big_endian and pwndbg.arch.endian == 'little'
if count > address > 0x10000:
count -= address
try:
data = pwndbg.memory.read(address, count, partial=True)
hexdump.last_address = (address + count)
except gdb.error as e:
print(e)
return
for i, line in enumerate(pwndbg.hexdump.hexdump(data, address=address, width=width, group_width=group_width, flip_group_endianess=flip_group_endianess, offset=hexdump.offset)):
print(line)
hexdump.offset += i
hexdump.last_address = 0
hexdump.offset = 0