mirror of https://github.com/pwndbg/pwndbg.git
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.
41 lines
1.3 KiB
Python
41 lines
1.3 KiB
Python
import gdb
|
|
|
|
import pwndbg.gdblib.regs
|
|
import tests
|
|
|
|
REFERENCE_BINARY = tests.binaries.get("reference-binary.out")
|
|
|
|
|
|
def test_command_nextproginstr_binary_not_running():
|
|
out = gdb.execute("nextproginstr", to_string=True)
|
|
assert out == "nextproginstr: The program is not being run.\n"
|
|
|
|
|
|
def test_command_nextproginstr(start_binary):
|
|
start_binary(REFERENCE_BINARY)
|
|
|
|
gdb.execute("break main")
|
|
gdb.execute("continue")
|
|
|
|
out = gdb.execute("nextproginstr", to_string=True)
|
|
assert out == "The pc is already at the binary objfile code. Not stepping.\n"
|
|
|
|
# Sanity check
|
|
exec_bin_pages = [p for p in pwndbg.vmmap.get() if p.objfile == pwndbg.proc.exe and p.execute]
|
|
assert any(pwndbg.gdblib.regs.pc in p for p in exec_bin_pages)
|
|
main_page = pwndbg.vmmap.find(pwndbg.gdblib.regs.pc)
|
|
|
|
gdb.execute("break puts")
|
|
gdb.execute("continue")
|
|
|
|
# Sanity check that we are in libc
|
|
assert "libc" in pwndbg.vmmap.find(pwndbg.gdblib.regs.rip).objfile
|
|
|
|
# Execute nextproginstr and see if we came back to the same vmmap page
|
|
gdb.execute("nextproginstr")
|
|
assert pwndbg.gdblib.regs.pc in main_page
|
|
|
|
# Ensure that nextproginstr won't jump now
|
|
out = gdb.execute("nextproginstr", to_string=True)
|
|
assert out == "The pc is already at the binary objfile code. Not stepping.\n"
|