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.8 KiB
Python
121 lines
3.8 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)'
|
|
]
|
|
|
|
BASH_BIN = tests.binaries.old_bash.get('binary')
|
|
BASH_CORE = tests.binaries.old_bash.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=BASH_BIN).splitlines()
|
|
|
|
expected = ['Reading symbols from %s...(no debugging symbols found)...done.' % BASH_BIN]
|
|
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=BASH_BIN, core=BASH_CORE).splitlines()
|
|
|
|
expected = [
|
|
'Reading symbols from %s...(no debugging symbols found)...done.' % BASH_BIN,
|
|
'''[New LWP 13562]''',
|
|
'''Core was generated by `/home/user/pwndbg/tests/corefiles/bash/binary'.''',
|
|
'''Program terminated with signal SIGINT, Interrupt.''',
|
|
'''#0 __addmntent (stream=0x7ffff79a7640 <_nl_global_locale+192>, ''',
|
|
''' mnt=<optimized out>) at mntent_r.c:257''',
|
|
'''257\tmntent_r.c: No such file or directory.'''
|
|
]
|
|
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=BASH_CORE).splitlines()
|
|
|
|
expected = [
|
|
'''[New LWP 13562]''',
|
|
'''Core was generated by `/home/user/pwndbg/tests/corefiles/bash/binary'.''',
|
|
'''Program terminated with signal SIGINT, Interrupt.''',
|
|
'''#0 0x00007ffff76d36b0 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)
|