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.
101 lines
3.0 KiB
Python
101 lines
3.0 KiB
Python
import codecs
|
|
import os
|
|
import re
|
|
import subprocess
|
|
|
|
import gdb
|
|
import pytest
|
|
|
|
import tests
|
|
|
|
from .utils import run_gdb_with_script
|
|
|
|
can_attach = False
|
|
|
|
if os.getuid() == 0:
|
|
can_attach = True
|
|
else:
|
|
# see `man ptrace`
|
|
with open('/proc/sys/kernel/yama/ptrace_scope') as f:
|
|
can_attach = f.read() == '0'
|
|
|
|
REASON_CANNOT_ATTACH = 'Test skipped due to inability to attach (needs sudo or sysctl -w kernel.yama.ptrace_scope=0'
|
|
|
|
|
|
@pytest.fixture
|
|
def launched_bash_binary():
|
|
path = '/tmp/pwndbg_test_bash'
|
|
subprocess.check_output(['cp', '/bin/bash', path])
|
|
|
|
process = subprocess.Popen([path], stdout=subprocess.PIPE, stdin=subprocess.PIPE)
|
|
|
|
yield process.pid, path
|
|
|
|
process.kill()
|
|
|
|
os.remove(path) # Cleanup
|
|
|
|
|
|
@pytest.mark.skipif(can_attach == False, reason=REASON_CANNOT_ATTACH)
|
|
def test_attachp_command_attaches_to_procname(launched_bash_binary):
|
|
pid, binary_path = launched_bash_binary
|
|
|
|
binary_name = binary_path.split('/')[-1]
|
|
result = run_gdb_with_script(pyafter='attachp %s' % binary_name)
|
|
|
|
matches = re.search(r'Attaching to ([0-9]+)', result).groups()
|
|
assert matches == (str(pid),)
|
|
|
|
assert re.search(r'Detaching from program: %s, process %s' % (binary_path, pid), result)
|
|
|
|
|
|
@pytest.mark.skipif(can_attach == False, reason=REASON_CANNOT_ATTACH)
|
|
def test_attachp_command_attaches_to_pid(launched_bash_binary):
|
|
pid, binary_path = launched_bash_binary
|
|
|
|
result = run_gdb_with_script(pyafter='attachp %s' % pid)
|
|
|
|
matches = re.search(r'Attaching to ([0-9]+)', result).groups()
|
|
assert matches == (str(pid),)
|
|
|
|
assert re.search(r'Detaching from program: %s, process %s' % (binary_path, pid), result)
|
|
|
|
|
|
@pytest.mark.skipif(can_attach == False, reason=REASON_CANNOT_ATTACH)
|
|
def test_attachp_command_attaches_to_procname_too_many_pids(launched_bash_binary):
|
|
pid, binary_path = launched_bash_binary
|
|
|
|
process = subprocess.Popen([binary_path], stdout=subprocess.PIPE, stdin=subprocess.PIPE)
|
|
|
|
binary_name = binary_path.split('/')[-1]
|
|
result = run_gdb_with_script(pyafter='attachp %s' % binary_name)
|
|
|
|
process.kill()
|
|
|
|
matches = re.search(r'Found pids: ([0-9]+), ([0-9]+) \(use `attach <pid>`\)', result).groups()
|
|
matches = list(map(int, matches))
|
|
matches.sort()
|
|
|
|
expected_pids = [pid, process.pid]
|
|
expected_pids.sort()
|
|
|
|
assert matches == expected_pids
|
|
|
|
|
|
@pytest.mark.skipif(can_attach == False, reason=REASON_CANNOT_ATTACH)
|
|
def test_attachp_command_nonexistent_procname():
|
|
result = run_gdb_with_script(pyafter='attachp some-nonexistent-process-name') # No chance there is a process name like this
|
|
assert 'Process some-nonexistent-process-name not found' in result
|
|
|
|
|
|
def test_attachp_command_no_pids():
|
|
try:
|
|
# On some machines/GDB versions this halts/waits forever, so we add a timeout here
|
|
result = run_gdb_with_script(pyafter='attachp 99999999', timeout=5) # No chance there is a PID like this
|
|
except subprocess.TimeoutExpired:
|
|
# Assume it works
|
|
return
|
|
|
|
assert 'Error: ptrace: No such process.' in result
|
|
|