Add hexdump test

pull/1134/head
Gulshan Singh 3 years ago
parent b218f34a36
commit fd5e722cbc

@ -0,0 +1,58 @@
import gdb
from pwnlib.util.cyclic import cyclic
import pwndbg.config
import pwndbg.gdblib.memory
import pwndbg.gdblib.regs
import pwndbg.vmmap
import tests
BINARY = tests.binaries.get("reference-binary.out")
def run_tests(stack, use_big_endian, expected):
pwndbg.config.hexdump_group_use_big_endian = use_big_endian
# Put some data onto the stack
pwndbg.gdblib.memory.write(stack, cyclic(0x100))
# Test empty hexdump
result = gdb.execute("hexdump 0", to_string=True)
assert result == "0x000000 \n"
results = []
# TODO: Repetition is not working in tests
results.append(gdb.execute(f"hexdump {stack} 64", to_string=True))
results.append(gdb.execute(f"hexdump {stack} 3", to_string=True))
assert len(results) == len(expected)
for i, result in enumerate(results):
expected_result = expected[i]
assert result == expected_result
def test_hexdump(start_binary):
start_binary(BINARY)
pwndbg.config.hexdump_group_width = -1
# TODO: Setting theme options with Python isn't working
gdb.execute("set hexdump-byte-separator")
stack_addr = pwndbg.gdblib.regs.rsp - 0x100
expected = [
f"""+0000 0x{stack_addr:x} 6161616261616161 6161616461616163 │aaaabaaa│caaadaaa│
+0010 0x{stack_addr+0x10:x} 6161616661616165 6161616861616167 eaaafaaagaaahaaa
+0020 0x{stack_addr+0x20:x} 6161616a61616169 6161616c6161616b iaaajaaakaaalaaa
+0030 0x{stack_addr+0x30:x} 6161616e6161616d 616161706161616f maaanaaaoaaapaaa\n""",
f"""+0000 0x{stack_addr:x} 616161 │aaa │ │\n""",
]
run_tests(stack_addr, True, expected)
expected = [
f"""+0000 0x{stack_addr:x} 6161616162616161 6361616164616161 │aaaabaaa│caaadaaa│
+0010 0x{stack_addr+0x10:x} 6561616166616161 6761616168616161 eaaafaaagaaahaaa
+0020 0x{stack_addr+0x20:x} 696161616a616161 6b6161616c616161 iaaajaaakaaalaaa
+0030 0x{stack_addr+0x30:x} 6d6161616e616161 6f61616170616161 maaanaaaoaaapaaa\n""",
f"""+0000 0x{stack_addr:x} 616161 │aaa │ │\n""",
]
run_tests(stack_addr, False, expected)
Loading…
Cancel
Save