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.
121 lines
3.6 KiB
Python
121 lines
3.6 KiB
Python
#!/usr/bin/env python
|
|
# -*- coding: utf-8 -*-
|
|
|
|
import codecs
|
|
import os
|
|
import re
|
|
import subprocess
|
|
|
|
import pytest
|
|
|
|
import tests
|
|
|
|
|
|
def run_gdb_with_script(binary='', core='', pybefore=None, pyafter=None):
|
|
"""
|
|
Runs GDB with given commands launched before and after loading of gdbinit.py
|
|
Returns GDB output.
|
|
"""
|
|
pybefore = ([pybefore] if isinstance(pybefore, str) else pybefore) or []
|
|
pyafter = ([pyafter] if isinstance(pyafter, str) else pyafter) or []
|
|
|
|
command = ['gdb', '--silent', '--nx', '--nh']
|
|
|
|
for cmd in pybefore:
|
|
command += ['--eval-command', cmd]
|
|
|
|
command += ['--command', 'gdbinit.py']
|
|
|
|
if binary:
|
|
command += [binary]
|
|
|
|
if core:
|
|
command += ['--core', core]
|
|
|
|
for cmd in pyafter:
|
|
command += ['--eval-command', cmd]
|
|
|
|
command += ['--eval-command', 'quit']
|
|
|
|
print("Launching command: %s" % command)
|
|
output = subprocess.check_output(command, stderr=subprocess.STDOUT)
|
|
|
|
# Python 3 returns bytes-like object so lets have it consistent
|
|
output = codecs.decode(output, 'utf8')
|
|
|
|
# The pwndbg banner shows number of loaded commands, it might differ between
|
|
# testing environments, so lets change it to ###
|
|
output = re.sub(r'loaded [0-9]+ commands', r'loaded ### commands', output)
|
|
|
|
return output
|
|
|
|
|
|
HELLO = [
|
|
'pwndbg: loaded ### commands. Type pwndbg [filter] for a list.',
|
|
'pwndbg: created $rebase, $ida gdb functions (can be used with print/break)'
|
|
]
|
|
|
|
BINARY = tests.binaries.div_zero_binary.get('binary')
|
|
CORE = tests.binaries.div_zero_binary.get('core')
|
|
|
|
launched_locally = not (os.environ.get('PWNDBG_TRAVIS_TEST_RUN'))
|
|
|
|
|
|
def test_loads_pure_gdb_without_crashing():
|
|
output = run_gdb_with_script().splitlines()
|
|
assert output == HELLO
|
|
|
|
|
|
@pytest.mark.skipif(launched_locally, reason='This test uses binaries compiled on travis builds.')
|
|
def test_loads_binary_without_crashing():
|
|
output = run_gdb_with_script(binary=BINARY).splitlines()
|
|
|
|
expected = ['Reading symbols from %s...' % BINARY,
|
|
'(No debugging symbols found in %s)' % BINARY]
|
|
expected += HELLO
|
|
|
|
assert all(item in output for item in expected)
|
|
|
|
|
|
@pytest.mark.skipif(launched_locally, reason='This test uses binaries compiled on travis builds.')
|
|
def test_loads_binary_with_core_without_crashing():
|
|
output = run_gdb_with_script(binary=BINARY, core=CORE).splitlines()
|
|
|
|
expected = [
|
|
'Reading symbols from %s...' % BINARY,
|
|
'(No debugging symbols found in %s)' % BINARY,
|
|
'[New LWP 45429]',
|
|
"Core was generated by `./binary'.",
|
|
'Program terminated with signal SIGFPE, Arithmetic exception.',
|
|
'#0 0x0000556e99b7913c in main ()',
|
|
]
|
|
expected += HELLO
|
|
|
|
assert all(item in output for item in expected)
|
|
|
|
|
|
@pytest.mark.skipif(launched_locally, reason='This test uses binaries compiled on travis builds.')
|
|
def test_loads_core_without_crashing():
|
|
output = run_gdb_with_script(core=CORE).splitlines()
|
|
|
|
expected = [
|
|
"[New LWP 45429]",
|
|
"Core was generated by `./binary'.",
|
|
'Program terminated with signal SIGFPE, Arithmetic exception.',
|
|
'#0 0x0000556e99b7913c in ?? ()',
|
|
]
|
|
expected += HELLO
|
|
|
|
assert all(item in output for item in expected)
|
|
|
|
|
|
def test_entry_no_file_loaded():
|
|
# This test is just to demonstrate that if gdb fails, all we have left is its stdout/err
|
|
output = run_gdb_with_script(binary='not_existing_binary', pyafter='entry').splitlines()
|
|
|
|
expected = ['not_existing_binary: No such file or directory.']
|
|
expected += HELLO
|
|
expected += ['entry: There is no file loaded.']
|
|
|
|
assert all(item in output for item in expected)
|