mirror of https://github.com/python/cpython.git
3606 lines
129 KiB
Python
3606 lines
129 KiB
Python
"""
|
|
The Python Debugger Pdb
|
|
=======================
|
|
|
|
To use the debugger in its simplest form:
|
|
|
|
>>> import pdb
|
|
>>> pdb.run('<a statement>')
|
|
|
|
The debugger's prompt is '(Pdb) '. This will stop in the first
|
|
function call in <a statement>.
|
|
|
|
Alternatively, if a statement terminated with an unhandled exception,
|
|
you can use pdb's post-mortem facility to inspect the contents of the
|
|
traceback:
|
|
|
|
>>> <a statement>
|
|
<exception traceback>
|
|
>>> import pdb
|
|
>>> pdb.pm()
|
|
|
|
The commands recognized by the debugger are listed in the next
|
|
section. Most can be abbreviated as indicated; e.g., h(elp) means
|
|
that 'help' can be typed as 'h' or 'help' (but not as 'he' or 'hel',
|
|
nor as 'H' or 'Help' or 'HELP'). Optional arguments are enclosed in
|
|
square brackets. Alternatives in the command syntax are separated
|
|
by a vertical bar (|).
|
|
|
|
A blank line repeats the previous command literally, except for
|
|
'list', where it lists the next 11 lines.
|
|
|
|
Commands that the debugger doesn't recognize are assumed to be Python
|
|
statements and are executed in the context of the program being
|
|
debugged. Python statements can also be prefixed with an exclamation
|
|
point ('!'). This is a powerful way to inspect the program being
|
|
debugged; it is even possible to change variables or call functions.
|
|
When an exception occurs in such a statement, the exception name is
|
|
printed but the debugger's state is not changed.
|
|
|
|
The debugger supports aliases, which can save typing. And aliases can
|
|
have parameters (see the alias help entry) which allows one a certain
|
|
level of adaptability to the context under examination.
|
|
|
|
Multiple commands may be entered on a single line, separated by the
|
|
pair ';;'. No intelligence is applied to separating the commands; the
|
|
input is split at the first ';;', even if it is in the middle of a
|
|
quoted string.
|
|
|
|
If a file ".pdbrc" exists in your home directory or in the current
|
|
directory, it is read in and executed as if it had been typed at the
|
|
debugger prompt. This is particularly useful for aliases. If both
|
|
files exist, the one in the home directory is read first and aliases
|
|
defined there can be overridden by the local file. This behavior can be
|
|
disabled by passing the "readrc=False" argument to the Pdb constructor.
|
|
|
|
Aside from aliases, the debugger is not directly programmable; but it
|
|
is implemented as a class from which you can derive your own debugger
|
|
class, which you can make as fancy as you like.
|
|
|
|
|
|
Debugger commands
|
|
=================
|
|
|
|
"""
|
|
# NOTE: the actual command documentation is collected from docstrings of the
|
|
# commands and is appended to __doc__ after the class has been defined.
|
|
|
|
import os
|
|
import io
|
|
import re
|
|
import sys
|
|
import cmd
|
|
import bdb
|
|
import dis
|
|
import code
|
|
import glob
|
|
import json
|
|
import token
|
|
import types
|
|
import atexit
|
|
import codeop
|
|
import pprint
|
|
import signal
|
|
import socket
|
|
import typing
|
|
import asyncio
|
|
import inspect
|
|
import weakref
|
|
import builtins
|
|
import tempfile
|
|
import textwrap
|
|
import tokenize
|
|
import itertools
|
|
import traceback
|
|
import linecache
|
|
import selectors
|
|
import threading
|
|
import _colorize
|
|
import _pyrepl.utils
|
|
|
|
from contextlib import ExitStack, closing, contextmanager
|
|
from rlcompleter import Completer
|
|
from types import CodeType
|
|
from warnings import deprecated
|
|
|
|
|
|
class Restart(Exception):
|
|
"""Causes a debugger to be restarted for the debugged python program."""
|
|
pass
|
|
|
|
__all__ = ["run", "pm", "Pdb", "runeval", "runctx", "runcall", "set_trace",
|
|
"post_mortem", "set_default_backend", "get_default_backend", "help"]
|
|
|
|
|
|
def find_first_executable_line(code):
|
|
""" Try to find the first executable line of the code object.
|
|
|
|
Equivalently, find the line number of the instruction that's
|
|
after RESUME
|
|
|
|
Return code.co_firstlineno if no executable line is found.
|
|
"""
|
|
prev = None
|
|
for instr in dis.get_instructions(code):
|
|
if prev is not None and prev.opname == 'RESUME':
|
|
if instr.positions.lineno is not None:
|
|
return instr.positions.lineno
|
|
return code.co_firstlineno
|
|
prev = instr
|
|
return code.co_firstlineno
|
|
|
|
def find_function(funcname, filename):
|
|
cre = re.compile(r'def\s+%s(\s*\[.+\])?\s*[(]' % re.escape(funcname))
|
|
try:
|
|
fp = tokenize.open(filename)
|
|
except OSError:
|
|
lines = linecache.getlines(filename)
|
|
if not lines:
|
|
return None
|
|
fp = io.StringIO(''.join(lines))
|
|
funcdef = ""
|
|
funcstart = 0
|
|
# consumer of this info expects the first line to be 1
|
|
with fp:
|
|
for lineno, line in enumerate(fp, start=1):
|
|
if cre.match(line):
|
|
funcstart, funcdef = lineno, line
|
|
elif funcdef:
|
|
funcdef += line
|
|
|
|
if funcdef:
|
|
try:
|
|
code = compile(funcdef, filename, 'exec')
|
|
except SyntaxError:
|
|
continue
|
|
# We should always be able to find the code object here
|
|
funccode = next(c for c in code.co_consts if
|
|
isinstance(c, CodeType) and c.co_name == funcname)
|
|
lineno_offset = find_first_executable_line(funccode)
|
|
return funcname, filename, funcstart + lineno_offset - 1
|
|
return None
|
|
|
|
def lasti2lineno(code, lasti):
|
|
linestarts = list(dis.findlinestarts(code))
|
|
linestarts.reverse()
|
|
for i, lineno in linestarts:
|
|
if lasti >= i:
|
|
return lineno
|
|
return 0
|
|
|
|
|
|
class _rstr(str):
|
|
"""String that doesn't quote its repr."""
|
|
def __repr__(self):
|
|
return self
|
|
|
|
|
|
class _ExecutableTarget:
|
|
filename: str
|
|
code: CodeType | str
|
|
namespace: dict
|
|
|
|
|
|
class _ScriptTarget(_ExecutableTarget):
|
|
def __init__(self, target):
|
|
self._target = os.path.realpath(target)
|
|
|
|
if not os.path.exists(self._target):
|
|
print(f'Error: {target} does not exist')
|
|
sys.exit(1)
|
|
if os.path.isdir(self._target):
|
|
print(f'Error: {target} is a directory')
|
|
sys.exit(1)
|
|
|
|
# If safe_path(-P) is not set, sys.path[0] is the directory
|
|
# of pdb, and we should replace it with the directory of the script
|
|
if not sys.flags.safe_path:
|
|
sys.path[0] = os.path.dirname(self._target)
|
|
|
|
def __repr__(self):
|
|
return self._target
|
|
|
|
@property
|
|
def filename(self):
|
|
return self._target
|
|
|
|
@property
|
|
def code(self):
|
|
# Open the file each time because the file may be modified
|
|
with io.open_code(self._target) as fp:
|
|
return f"exec(compile({fp.read()!r}, {self._target!r}, 'exec'))"
|
|
|
|
@property
|
|
def namespace(self):
|
|
return dict(
|
|
__name__='__main__',
|
|
__file__=self._target,
|
|
__builtins__=__builtins__,
|
|
__spec__=None,
|
|
)
|
|
|
|
|
|
class _ModuleTarget(_ExecutableTarget):
|
|
def __init__(self, target):
|
|
self._target = target
|
|
|
|
import runpy
|
|
try:
|
|
_, self._spec, self._code = runpy._get_module_details(self._target)
|
|
except ImportError as e:
|
|
print(f"ImportError: {e}")
|
|
sys.exit(1)
|
|
except Exception:
|
|
traceback.print_exc()
|
|
sys.exit(1)
|
|
|
|
def __repr__(self):
|
|
return self._target
|
|
|
|
@property
|
|
def filename(self):
|
|
return self._code.co_filename
|
|
|
|
@property
|
|
def code(self):
|
|
return self._code
|
|
|
|
@property
|
|
def namespace(self):
|
|
return dict(
|
|
__name__='__main__',
|
|
__file__=os.path.normcase(os.path.abspath(self.filename)),
|
|
__package__=self._spec.parent,
|
|
__loader__=self._spec.loader,
|
|
__spec__=self._spec,
|
|
__builtins__=__builtins__,
|
|
)
|
|
|
|
|
|
class _ZipTarget(_ExecutableTarget):
|
|
def __init__(self, target):
|
|
import runpy
|
|
|
|
self._target = os.path.realpath(target)
|
|
sys.path.insert(0, self._target)
|
|
try:
|
|
_, self._spec, self._code = runpy._get_main_module_details()
|
|
except ImportError as e:
|
|
print(f"ImportError: {e}")
|
|
sys.exit(1)
|
|
except Exception:
|
|
traceback.print_exc()
|
|
sys.exit(1)
|
|
|
|
def __repr__(self):
|
|
return self._target
|
|
|
|
@property
|
|
def filename(self):
|
|
return self._code.co_filename
|
|
|
|
@property
|
|
def code(self):
|
|
return self._code
|
|
|
|
@property
|
|
def namespace(self):
|
|
return dict(
|
|
__name__='__main__',
|
|
__file__=os.path.normcase(os.path.abspath(self.filename)),
|
|
__package__=self._spec.parent,
|
|
__loader__=self._spec.loader,
|
|
__spec__=self._spec,
|
|
__builtins__=__builtins__,
|
|
)
|
|
|
|
|
|
class _PdbInteractiveConsole(code.InteractiveConsole):
|
|
def __init__(self, ns, message):
|
|
self._message = message
|
|
super().__init__(locals=ns, local_exit=True)
|
|
|
|
def write(self, data):
|
|
self._message(data, end='')
|
|
|
|
|
|
# Interaction prompt line will separate file and call info from code
|
|
# text using value of line_prefix string. A newline and arrow may
|
|
# be to your liking. You can set it once pdb is imported using the
|
|
# command "pdb.line_prefix = '\n% '".
|
|
# line_prefix = ': ' # Use this to get the old situation back
|
|
line_prefix = '\n-> ' # Probably a better default
|
|
|
|
|
|
# The default backend to use for Pdb instances if not specified
|
|
# Should be either 'settrace' or 'monitoring'
|
|
_default_backend = 'settrace'
|
|
|
|
|
|
def set_default_backend(backend):
|
|
"""Set the default backend to use for Pdb instances."""
|
|
global _default_backend
|
|
if backend not in ('settrace', 'monitoring'):
|
|
raise ValueError("Invalid backend: %s" % backend)
|
|
_default_backend = backend
|
|
|
|
|
|
def get_default_backend():
|
|
"""Get the default backend to use for Pdb instances."""
|
|
return _default_backend
|
|
|
|
|
|
class Pdb(bdb.Bdb, cmd.Cmd):
|
|
_previous_sigint_handler = None
|
|
|
|
# Limit the maximum depth of chained exceptions, we should be handling cycles,
|
|
# but in case there are recursions, we stop at 999.
|
|
MAX_CHAINED_EXCEPTION_DEPTH = 999
|
|
|
|
_file_mtime_table = {}
|
|
|
|
_last_pdb_instance = None
|
|
|
|
def __init__(self, completekey='tab', stdin=None, stdout=None, skip=None,
|
|
nosigint=False, readrc=True, mode=None, backend=None, colorize=False):
|
|
bdb.Bdb.__init__(self, skip=skip, backend=backend if backend else get_default_backend())
|
|
cmd.Cmd.__init__(self, completekey, stdin, stdout)
|
|
sys.audit("pdb.Pdb")
|
|
if stdout:
|
|
self.use_rawinput = 0
|
|
self.prompt = '(Pdb) '
|
|
self.aliases = {}
|
|
self.displaying = {}
|
|
self.mainpyfile = ''
|
|
self._wait_for_mainpyfile = False
|
|
self.tb_lineno = {}
|
|
self.mode = mode
|
|
self.colorize = colorize and _colorize.can_colorize(file=stdout or sys.stdout)
|
|
# Try to load readline if it exists
|
|
try:
|
|
import readline
|
|
# remove some common file name delimiters
|
|
readline.set_completer_delims(' \t\n`@#%^&*()=+[{]}\\|;:\'",<>?')
|
|
except ImportError:
|
|
pass
|
|
self.allow_kbdint = False
|
|
self.nosigint = nosigint
|
|
# Consider these characters as part of the command so when the users type
|
|
# c.a or c['a'], it won't be recognized as a c(ontinue) command
|
|
self.identchars = cmd.Cmd.identchars + '=.[](),"\'+-*/%@&|<>~^'
|
|
|
|
# Read ~/.pdbrc and ./.pdbrc
|
|
self.rcLines = []
|
|
if readrc:
|
|
try:
|
|
with open(os.path.expanduser('~/.pdbrc'), encoding='utf-8') as rcFile:
|
|
self.rcLines.extend(rcFile)
|
|
except OSError:
|
|
pass
|
|
try:
|
|
with open(".pdbrc", encoding='utf-8') as rcFile:
|
|
self.rcLines.extend(rcFile)
|
|
except OSError:
|
|
pass
|
|
|
|
self.commands = {} # associates a command list to breakpoint numbers
|
|
self.commands_defining = False # True while in the process of defining
|
|
# a command list
|
|
self.commands_bnum = None # The breakpoint number for which we are
|
|
# defining a list
|
|
|
|
self.async_shim_frame = None
|
|
self.async_awaitable = None
|
|
|
|
self._chained_exceptions = tuple()
|
|
self._chained_exception_index = 0
|
|
|
|
self._current_task = None
|
|
|
|
def set_trace(self, frame=None, *, commands=None):
|
|
Pdb._last_pdb_instance = self
|
|
if frame is None:
|
|
frame = sys._getframe().f_back
|
|
|
|
if commands is not None:
|
|
self.rcLines.extend(commands)
|
|
|
|
super().set_trace(frame)
|
|
|
|
async def set_trace_async(self, frame=None, *, commands=None):
|
|
if self.async_awaitable is not None:
|
|
# We are already in a set_trace_async call, do not mess with it
|
|
return
|
|
|
|
if frame is None:
|
|
frame = sys._getframe().f_back
|
|
|
|
# We need set_trace to set up the basics, however, this will call
|
|
# set_stepinstr() will we need to compensate for, because we don't
|
|
# want to trigger on calls
|
|
self.set_trace(frame, commands=commands)
|
|
# Changing the stopframe will disable trace dispatch on calls
|
|
self.stopframe = frame
|
|
# We need to stop tracing because we don't have the privilege to avoid
|
|
# triggering tracing functions as normal, as we are not already in
|
|
# tracing functions
|
|
self.stop_trace()
|
|
|
|
self.async_shim_frame = sys._getframe()
|
|
self.async_awaitable = None
|
|
|
|
while True:
|
|
self.async_awaitable = None
|
|
# Simulate a trace event
|
|
# This should bring up pdb and make pdb believe it's debugging the
|
|
# caller frame
|
|
self.trace_dispatch(frame, "opcode", None)
|
|
if self.async_awaitable is not None:
|
|
try:
|
|
if self.breaks:
|
|
with self.set_enterframe(frame):
|
|
# set_continue requires enterframe to work
|
|
self.set_continue()
|
|
self.start_trace()
|
|
await self.async_awaitable
|
|
except Exception:
|
|
self._error_exc()
|
|
else:
|
|
break
|
|
|
|
self.async_shim_frame = None
|
|
|
|
# start the trace (the actual command is already set by set_* calls)
|
|
if self.returnframe is None and self.stoplineno == -1 and not self.breaks:
|
|
# This means we did a continue without any breakpoints, we should not
|
|
# start the trace
|
|
return
|
|
|
|
self.start_trace()
|
|
|
|
def sigint_handler(self, signum, frame):
|
|
if self.allow_kbdint:
|
|
raise KeyboardInterrupt
|
|
self.message("\nProgram interrupted. (Use 'cont' to resume).")
|
|
self.set_step()
|
|
self.set_trace(frame)
|
|
|
|
def reset(self):
|
|
bdb.Bdb.reset(self)
|
|
self.forget()
|
|
|
|
def forget(self):
|
|
self.lineno = None
|
|
self.stack = []
|
|
self.curindex = 0
|
|
if hasattr(self, 'curframe') and self.curframe:
|
|
self.curframe.f_globals.pop('__pdb_convenience_variables', None)
|
|
self.curframe = None
|
|
self.tb_lineno.clear()
|
|
|
|
def setup(self, f, tb):
|
|
self.forget()
|
|
self.stack, self.curindex = self.get_stack(f, tb)
|
|
while tb:
|
|
# when setting up post-mortem debugging with a traceback, save all
|
|
# the original line numbers to be displayed along the current line
|
|
# numbers (which can be different, e.g. due to finally clauses)
|
|
lineno = lasti2lineno(tb.tb_frame.f_code, tb.tb_lasti)
|
|
self.tb_lineno[tb.tb_frame] = lineno
|
|
tb = tb.tb_next
|
|
self.curframe = self.stack[self.curindex][0]
|
|
self.set_convenience_variable(self.curframe, '_frame', self.curframe)
|
|
if self._current_task:
|
|
self.set_convenience_variable(self.curframe, '_asynctask', self._current_task)
|
|
self._save_initial_file_mtime(self.curframe)
|
|
|
|
if self._chained_exceptions:
|
|
self.set_convenience_variable(
|
|
self.curframe,
|
|
'_exception',
|
|
self._chained_exceptions[self._chained_exception_index],
|
|
)
|
|
|
|
if self.rcLines:
|
|
self.cmdqueue = [
|
|
line for line in self.rcLines
|
|
if line.strip() and not line.strip().startswith("#")
|
|
]
|
|
self.rcLines = []
|
|
|
|
@property
|
|
@deprecated("The frame locals reference is no longer cached. Use 'curframe.f_locals' instead.")
|
|
def curframe_locals(self):
|
|
return self.curframe.f_locals
|
|
|
|
@curframe_locals.setter
|
|
@deprecated("Setting 'curframe_locals' no longer has any effect. Update the contents of 'curframe.f_locals' instead.")
|
|
def curframe_locals(self, value):
|
|
pass
|
|
|
|
# Override Bdb methods
|
|
|
|
def user_call(self, frame, argument_list):
|
|
"""This method is called when there is the remote possibility
|
|
that we ever need to stop in this function."""
|
|
if self._wait_for_mainpyfile:
|
|
return
|
|
if self.stop_here(frame):
|
|
self.message('--Call--')
|
|
self.interaction(frame, None)
|
|
|
|
def user_line(self, frame):
|
|
"""This function is called when we stop or break at this line."""
|
|
if self._wait_for_mainpyfile:
|
|
if (self.mainpyfile != self.canonic(frame.f_code.co_filename)):
|
|
return
|
|
self._wait_for_mainpyfile = False
|
|
if self.trace_opcodes:
|
|
# GH-127321
|
|
# We want to avoid stopping at an opcode that does not have
|
|
# an associated line number because pdb does not like it
|
|
if frame.f_lineno is None:
|
|
self.set_stepinstr()
|
|
return
|
|
self.bp_commands(frame)
|
|
self.interaction(frame, None)
|
|
|
|
user_opcode = user_line
|
|
|
|
def bp_commands(self, frame):
|
|
"""Call every command that was set for the current active breakpoint
|
|
(if there is one).
|
|
|
|
Returns True if the normal interaction function must be called,
|
|
False otherwise."""
|
|
# self.currentbp is set in bdb in Bdb.break_here if a breakpoint was hit
|
|
if getattr(self, "currentbp", False) and \
|
|
self.currentbp in self.commands:
|
|
currentbp = self.currentbp
|
|
self.currentbp = 0
|
|
for line in self.commands[currentbp]:
|
|
self.cmdqueue.append(line)
|
|
self.cmdqueue.append(f'_pdbcmd_restore_lastcmd {self.lastcmd}')
|
|
|
|
def user_return(self, frame, return_value):
|
|
"""This function is called when a return trap is set here."""
|
|
if self._wait_for_mainpyfile:
|
|
return
|
|
frame.f_locals['__return__'] = return_value
|
|
self.set_convenience_variable(frame, '_retval', return_value)
|
|
self.message('--Return--')
|
|
self.interaction(frame, None)
|
|
|
|
def user_exception(self, frame, exc_info):
|
|
"""This function is called if an exception occurs,
|
|
but only if we are to stop at or just below this level."""
|
|
if self._wait_for_mainpyfile:
|
|
return
|
|
exc_type, exc_value, exc_traceback = exc_info
|
|
frame.f_locals['__exception__'] = exc_type, exc_value
|
|
self.set_convenience_variable(frame, '_exception', exc_value)
|
|
|
|
# An 'Internal StopIteration' exception is an exception debug event
|
|
# issued by the interpreter when handling a subgenerator run with
|
|
# 'yield from' or a generator controlled by a for loop. No exception has
|
|
# actually occurred in this case. The debugger uses this debug event to
|
|
# stop when the debuggee is returning from such generators.
|
|
prefix = 'Internal ' if (not exc_traceback
|
|
and exc_type is StopIteration) else ''
|
|
self.message('%s%s' % (prefix, self._format_exc(exc_value)))
|
|
self.interaction(frame, exc_traceback)
|
|
|
|
# General interaction function
|
|
def _cmdloop(self):
|
|
while True:
|
|
try:
|
|
# keyboard interrupts allow for an easy way to cancel
|
|
# the current command, so allow them during interactive input
|
|
self.allow_kbdint = True
|
|
self.cmdloop()
|
|
self.allow_kbdint = False
|
|
break
|
|
except KeyboardInterrupt:
|
|
self.message('--KeyboardInterrupt--')
|
|
|
|
def _save_initial_file_mtime(self, frame):
|
|
"""save the mtime of the all the files in the frame stack in the file mtime table
|
|
if they haven't been saved yet."""
|
|
while frame:
|
|
filename = frame.f_code.co_filename
|
|
if filename not in self._file_mtime_table:
|
|
try:
|
|
self._file_mtime_table[filename] = os.path.getmtime(filename)
|
|
except Exception:
|
|
pass
|
|
frame = frame.f_back
|
|
|
|
def _validate_file_mtime(self):
|
|
"""Check if the source file of the current frame has been modified.
|
|
If so, give a warning and reset the modify time to current."""
|
|
try:
|
|
filename = self.curframe.f_code.co_filename
|
|
mtime = os.path.getmtime(filename)
|
|
except Exception:
|
|
return
|
|
if (filename in self._file_mtime_table and
|
|
mtime != self._file_mtime_table[filename]):
|
|
self.message(f"*** WARNING: file '{filename}' was edited, "
|
|
"running stale code until the program is rerun")
|
|
self._file_mtime_table[filename] = mtime
|
|
|
|
# Called before loop, handles display expressions
|
|
# Set up convenience variable containers
|
|
def _show_display(self):
|
|
displaying = self.displaying.get(self.curframe)
|
|
if displaying:
|
|
for expr, oldvalue in displaying.items():
|
|
newvalue = self._getval_except(expr)
|
|
# check for identity first; this prevents custom __eq__ to
|
|
# be called at every loop, and also prevents instances whose
|
|
# fields are changed to be displayed
|
|
if newvalue is not oldvalue and newvalue != oldvalue:
|
|
displaying[expr] = newvalue
|
|
self.message('display %s: %s [old: %s]' %
|
|
(expr, self._safe_repr(newvalue, expr),
|
|
self._safe_repr(oldvalue, expr)))
|
|
|
|
def _get_tb_and_exceptions(self, tb_or_exc):
|
|
"""
|
|
Given a tracecack or an exception, return a tuple of chained exceptions
|
|
and current traceback to inspect.
|
|
|
|
This will deal with selecting the right ``__cause__`` or ``__context__``
|
|
as well as handling cycles, and return a flattened list of exceptions we
|
|
can jump to with do_exceptions.
|
|
|
|
"""
|
|
_exceptions = []
|
|
if isinstance(tb_or_exc, BaseException):
|
|
traceback, current = tb_or_exc.__traceback__, tb_or_exc
|
|
|
|
while current is not None:
|
|
if current in _exceptions:
|
|
break
|
|
_exceptions.append(current)
|
|
if current.__cause__ is not None:
|
|
current = current.__cause__
|
|
elif (
|
|
current.__context__ is not None and not current.__suppress_context__
|
|
):
|
|
current = current.__context__
|
|
|
|
if len(_exceptions) >= self.MAX_CHAINED_EXCEPTION_DEPTH:
|
|
self.message(
|
|
f"More than {self.MAX_CHAINED_EXCEPTION_DEPTH}"
|
|
" chained exceptions found, not all exceptions"
|
|
"will be browsable with `exceptions`."
|
|
)
|
|
break
|
|
else:
|
|
traceback = tb_or_exc
|
|
return tuple(reversed(_exceptions)), traceback
|
|
|
|
@contextmanager
|
|
def _hold_exceptions(self, exceptions):
|
|
"""
|
|
Context manager to ensure proper cleaning of exceptions references
|
|
|
|
When given a chained exception instead of a traceback,
|
|
pdb may hold references to many objects which may leak memory.
|
|
|
|
We use this context manager to make sure everything is properly cleaned
|
|
|
|
"""
|
|
try:
|
|
self._chained_exceptions = exceptions
|
|
self._chained_exception_index = len(exceptions) - 1
|
|
yield
|
|
finally:
|
|
# we can't put those in forget as otherwise they would
|
|
# be cleared on exception change
|
|
self._chained_exceptions = tuple()
|
|
self._chained_exception_index = 0
|
|
|
|
def _get_asyncio_task(self):
|
|
try:
|
|
task = asyncio.current_task()
|
|
except RuntimeError:
|
|
task = None
|
|
return task
|
|
|
|
def interaction(self, frame, tb_or_exc):
|
|
# Restore the previous signal handler at the Pdb prompt.
|
|
if Pdb._previous_sigint_handler:
|
|
try:
|
|
signal.signal(signal.SIGINT, Pdb._previous_sigint_handler)
|
|
except ValueError: # ValueError: signal only works in main thread
|
|
pass
|
|
else:
|
|
Pdb._previous_sigint_handler = None
|
|
|
|
self._current_task = self._get_asyncio_task()
|
|
|
|
_chained_exceptions, tb = self._get_tb_and_exceptions(tb_or_exc)
|
|
if isinstance(tb_or_exc, BaseException):
|
|
assert tb is not None, "main exception must have a traceback"
|
|
with self._hold_exceptions(_chained_exceptions):
|
|
self.setup(frame, tb)
|
|
# We should print the stack entry if and only if the user input
|
|
# is expected, and we should print it right before the user input.
|
|
# We achieve this by appending _pdbcmd_print_frame_status to the
|
|
# command queue. If cmdqueue is not exhausted, the user input is
|
|
# not expected and we will not print the stack entry.
|
|
self.cmdqueue.append('_pdbcmd_print_frame_status')
|
|
self._cmdloop()
|
|
# If _pdbcmd_print_frame_status is not used, pop it out
|
|
if self.cmdqueue and self.cmdqueue[-1] == '_pdbcmd_print_frame_status':
|
|
self.cmdqueue.pop()
|
|
self.forget()
|
|
|
|
def displayhook(self, obj):
|
|
"""Custom displayhook for the exec in default(), which prevents
|
|
assignment of the _ variable in the builtins.
|
|
"""
|
|
# reproduce the behavior of the standard displayhook, not printing None
|
|
if obj is not None:
|
|
self.message(repr(obj))
|
|
|
|
@contextmanager
|
|
def _enable_multiline_input(self):
|
|
try:
|
|
import readline
|
|
except ImportError:
|
|
yield
|
|
return
|
|
|
|
def input_auto_indent():
|
|
last_index = readline.get_current_history_length()
|
|
last_line = readline.get_history_item(last_index)
|
|
if last_line:
|
|
if last_line.isspace():
|
|
# If the last line is empty, we don't need to indent
|
|
return
|
|
|
|
last_line = last_line.rstrip('\r\n')
|
|
indent = len(last_line) - len(last_line.lstrip())
|
|
if last_line.endswith(":"):
|
|
indent += 4
|
|
readline.insert_text(' ' * indent)
|
|
|
|
completenames = self.completenames
|
|
try:
|
|
self.completenames = self.complete_multiline_names
|
|
readline.set_startup_hook(input_auto_indent)
|
|
yield
|
|
finally:
|
|
readline.set_startup_hook()
|
|
self.completenames = completenames
|
|
return
|
|
|
|
def _exec_in_closure(self, source, globals, locals):
|
|
""" Run source code in closure so code object created within source
|
|
can find variables in locals correctly
|
|
|
|
returns True if the source is executed, False otherwise
|
|
"""
|
|
|
|
# Determine if the source should be executed in closure. Only when the
|
|
# source compiled to multiple code objects, we should use this feature.
|
|
# Otherwise, we can just raise an exception and normal exec will be used.
|
|
|
|
code = compile(source, "<string>", "exec")
|
|
if not any(isinstance(const, CodeType) for const in code.co_consts):
|
|
return False
|
|
|
|
# locals could be a proxy which does not support pop
|
|
# copy it first to avoid modifying the original locals
|
|
locals_copy = dict(locals)
|
|
|
|
locals_copy["__pdb_eval__"] = {
|
|
"result": None,
|
|
"write_back": {}
|
|
}
|
|
|
|
# If the source is an expression, we need to print its value
|
|
try:
|
|
compile(source, "<string>", "eval")
|
|
except SyntaxError:
|
|
pass
|
|
else:
|
|
source = "__pdb_eval__['result'] = " + source
|
|
|
|
# Add write-back to update the locals
|
|
source = ("try:\n" +
|
|
textwrap.indent(source, " ") + "\n" +
|
|
"finally:\n" +
|
|
" __pdb_eval__['write_back'] = locals()")
|
|
|
|
# Build a closure source code with freevars from locals like:
|
|
# def __pdb_outer():
|
|
# var = None
|
|
# def __pdb_scope(): # This is the code object we want to execute
|
|
# nonlocal var
|
|
# <source>
|
|
# return __pdb_scope.__code__
|
|
source_with_closure = ("def __pdb_outer():\n" +
|
|
"\n".join(f" {var} = None" for var in locals_copy) + "\n" +
|
|
" def __pdb_scope():\n" +
|
|
"\n".join(f" nonlocal {var}" for var in locals_copy) + "\n" +
|
|
textwrap.indent(source, " ") + "\n" +
|
|
" return __pdb_scope.__code__"
|
|
)
|
|
|
|
# Get the code object of __pdb_scope()
|
|
# The exec fills locals_copy with the __pdb_outer() function and we can call
|
|
# that to get the code object of __pdb_scope()
|
|
ns = {}
|
|
try:
|
|
exec(source_with_closure, {}, ns)
|
|
except Exception:
|
|
return False
|
|
code = ns["__pdb_outer"]()
|
|
|
|
cells = tuple(types.CellType(locals_copy.get(var)) for var in code.co_freevars)
|
|
|
|
try:
|
|
exec(code, globals, locals_copy, closure=cells)
|
|
except Exception:
|
|
return False
|
|
|
|
# get the data we need from the statement
|
|
pdb_eval = locals_copy["__pdb_eval__"]
|
|
|
|
# __pdb_eval__ should not be updated back to locals
|
|
pdb_eval["write_back"].pop("__pdb_eval__")
|
|
|
|
# Write all local variables back to locals
|
|
locals.update(pdb_eval["write_back"])
|
|
eval_result = pdb_eval["result"]
|
|
if eval_result is not None:
|
|
print(repr(eval_result))
|
|
|
|
return True
|
|
|
|
def _exec_await(self, source, globals, locals):
|
|
""" Run source code that contains await by playing with async shim frame"""
|
|
# Put the source in an async function
|
|
source_async = (
|
|
"async def __pdb_await():\n" +
|
|
textwrap.indent(source, " ") + '\n' +
|
|
" __pdb_locals.update(locals())"
|
|
)
|
|
ns = globals | locals
|
|
# We use __pdb_locals to do write back
|
|
ns["__pdb_locals"] = locals
|
|
exec(source_async, ns)
|
|
self.async_awaitable = ns["__pdb_await"]()
|
|
|
|
def _read_code(self, line):
|
|
buffer = line
|
|
is_await_code = False
|
|
code = None
|
|
try:
|
|
if (code := codeop.compile_command(line + '\n', '<stdin>', 'single')) is None:
|
|
# Multi-line mode
|
|
with self._enable_multiline_input():
|
|
buffer = line
|
|
continue_prompt = "... "
|
|
while (code := codeop.compile_command(buffer, '<stdin>', 'single')) is None:
|
|
if self.use_rawinput:
|
|
try:
|
|
line = input(continue_prompt)
|
|
except (EOFError, KeyboardInterrupt):
|
|
self.lastcmd = ""
|
|
print('\n')
|
|
return None, None, False
|
|
else:
|
|
self.stdout.write(continue_prompt)
|
|
self.stdout.flush()
|
|
line = self.stdin.readline()
|
|
if not len(line):
|
|
self.lastcmd = ""
|
|
self.stdout.write('\n')
|
|
self.stdout.flush()
|
|
return None, None, False
|
|
else:
|
|
line = line.rstrip('\r\n')
|
|
if line.isspace():
|
|
# empty line, just continue
|
|
buffer += '\n'
|
|
else:
|
|
buffer += '\n' + line
|
|
self.lastcmd = buffer
|
|
except SyntaxError as e:
|
|
# Maybe it's an await expression/statement
|
|
if (
|
|
self.async_shim_frame is not None
|
|
and e.msg == "'await' outside function"
|
|
):
|
|
is_await_code = True
|
|
else:
|
|
raise
|
|
|
|
return code, buffer, is_await_code
|
|
|
|
def default(self, line):
|
|
if line[:1] == '!': line = line[1:].strip()
|
|
locals = self.curframe.f_locals
|
|
globals = self.curframe.f_globals
|
|
try:
|
|
code, buffer, is_await_code = self._read_code(line)
|
|
if buffer is None:
|
|
return
|
|
save_stdout = sys.stdout
|
|
save_stdin = sys.stdin
|
|
save_displayhook = sys.displayhook
|
|
try:
|
|
sys.stdin = self.stdin
|
|
sys.stdout = self.stdout
|
|
sys.displayhook = self.displayhook
|
|
if is_await_code:
|
|
self._exec_await(buffer, globals, locals)
|
|
return True
|
|
else:
|
|
if not self._exec_in_closure(buffer, globals, locals):
|
|
exec(code, globals, locals)
|
|
finally:
|
|
sys.stdout = save_stdout
|
|
sys.stdin = save_stdin
|
|
sys.displayhook = save_displayhook
|
|
except:
|
|
self._error_exc()
|
|
|
|
def _replace_convenience_variables(self, line):
|
|
"""Replace the convenience variables in 'line' with their values.
|
|
e.g. $foo is replaced by __pdb_convenience_variables["foo"].
|
|
Note: such pattern in string literals will be skipped"""
|
|
|
|
if "$" not in line:
|
|
return line
|
|
|
|
dollar_start = dollar_end = (-1, -1)
|
|
replace_variables = []
|
|
try:
|
|
for t in tokenize.generate_tokens(io.StringIO(line).readline):
|
|
token_type, token_string, start, end, _ = t
|
|
if token_type == token.OP and token_string == '$':
|
|
dollar_start, dollar_end = start, end
|
|
elif start == dollar_end and token_type == token.NAME:
|
|
# line is a one-line command so we only care about column
|
|
replace_variables.append((dollar_start[1], end[1], token_string))
|
|
except tokenize.TokenError:
|
|
return line
|
|
|
|
if not replace_variables:
|
|
return line
|
|
|
|
last_end = 0
|
|
line_pieces = []
|
|
for start, end, name in replace_variables:
|
|
line_pieces.append(line[last_end:start] + f'__pdb_convenience_variables["{name}"]')
|
|
last_end = end
|
|
line_pieces.append(line[last_end:])
|
|
|
|
return ''.join(line_pieces)
|
|
|
|
def precmd(self, line):
|
|
"""Handle alias expansion and ';;' separator."""
|
|
if not line.strip():
|
|
return line
|
|
args = line.split()
|
|
while args[0] in self.aliases:
|
|
line = self.aliases[args[0]]
|
|
for idx in range(1, 10):
|
|
if f'%{idx}' in line:
|
|
if idx >= len(args):
|
|
self.error(f"Not enough arguments for alias '{args[0]}'")
|
|
# This is a no-op
|
|
return "!"
|
|
line = line.replace(f'%{idx}', args[idx])
|
|
elif '%*' not in line:
|
|
if idx < len(args):
|
|
self.error(f"Too many arguments for alias '{args[0]}'")
|
|
# This is a no-op
|
|
return "!"
|
|
break
|
|
|
|
line = line.replace("%*", ' '.join(args[1:]))
|
|
args = line.split()
|
|
# split into ';;' separated commands
|
|
# unless it's an alias command
|
|
if args[0] != 'alias':
|
|
marker = line.find(';;')
|
|
if marker >= 0:
|
|
# queue up everything after marker
|
|
next = line[marker+2:].lstrip()
|
|
self.cmdqueue.insert(0, next)
|
|
line = line[:marker].rstrip()
|
|
|
|
# Replace all the convenience variables
|
|
line = self._replace_convenience_variables(line)
|
|
|
|
return line
|
|
|
|
def onecmd(self, line):
|
|
"""Interpret the argument as though it had been typed in response
|
|
to the prompt.
|
|
|
|
Checks whether this line is typed at the normal prompt or in
|
|
a breakpoint command list definition.
|
|
"""
|
|
if not self.commands_defining:
|
|
if line.startswith('_pdbcmd'):
|
|
command, arg, line = self.parseline(line)
|
|
if hasattr(self, command):
|
|
return getattr(self, command)(arg)
|
|
return cmd.Cmd.onecmd(self, line)
|
|
else:
|
|
return self.handle_command_def(line)
|
|
|
|
def handle_command_def(self, line):
|
|
"""Handles one command line during command list definition."""
|
|
cmd, arg, line = self.parseline(line)
|
|
if not cmd:
|
|
return False
|
|
if cmd == 'end':
|
|
return True # end of cmd list
|
|
elif cmd == 'EOF':
|
|
self.message('')
|
|
return True # end of cmd list
|
|
cmdlist = self.commands[self.commands_bnum]
|
|
if cmd == 'silent':
|
|
cmdlist.append('_pdbcmd_silence_frame_status')
|
|
return False # continue to handle other cmd def in the cmd list
|
|
if arg:
|
|
cmdlist.append(cmd+' '+arg)
|
|
else:
|
|
cmdlist.append(cmd)
|
|
# Determine if we must stop
|
|
try:
|
|
func = getattr(self, 'do_' + cmd)
|
|
except AttributeError:
|
|
func = self.default
|
|
# one of the resuming commands
|
|
if func.__name__ in self.commands_resuming:
|
|
return True
|
|
return False
|
|
|
|
def _colorize_code(self, code):
|
|
if self.colorize:
|
|
colors = list(_pyrepl.utils.gen_colors(code))
|
|
chars, _ = _pyrepl.utils.disp_str(code, colors=colors, force_color=True)
|
|
code = "".join(chars)
|
|
return code
|
|
|
|
# interface abstraction functions
|
|
|
|
def message(self, msg, end='\n'):
|
|
print(msg, end=end, file=self.stdout)
|
|
|
|
def error(self, msg):
|
|
print('***', msg, file=self.stdout)
|
|
|
|
# convenience variables
|
|
|
|
def set_convenience_variable(self, frame, name, value):
|
|
if '__pdb_convenience_variables' not in frame.f_globals:
|
|
frame.f_globals['__pdb_convenience_variables'] = {}
|
|
frame.f_globals['__pdb_convenience_variables'][name] = value
|
|
|
|
# Generic completion functions. Individual complete_foo methods can be
|
|
# assigned below to one of these functions.
|
|
|
|
def completenames(self, text, line, begidx, endidx):
|
|
# Overwrite completenames() of cmd so for the command completion,
|
|
# if no current command matches, check for expressions as well
|
|
commands = super().completenames(text, line, begidx, endidx)
|
|
for alias in self.aliases:
|
|
if alias.startswith(text):
|
|
commands.append(alias)
|
|
if commands:
|
|
return commands
|
|
else:
|
|
expressions = self._complete_expression(text, line, begidx, endidx)
|
|
if expressions:
|
|
return expressions
|
|
return self.completedefault(text, line, begidx, endidx)
|
|
|
|
def _complete_location(self, text, line, begidx, endidx):
|
|
# Complete a file/module/function location for break/tbreak/clear.
|
|
if line.strip().endswith((':', ',')):
|
|
# Here comes a line number or a condition which we can't complete.
|
|
return []
|
|
# First, try to find matching functions (i.e. expressions).
|
|
try:
|
|
ret = self._complete_expression(text, line, begidx, endidx)
|
|
except Exception:
|
|
ret = []
|
|
# Then, try to complete file names as well.
|
|
globs = glob.glob(glob.escape(text) + '*')
|
|
for fn in globs:
|
|
if os.path.isdir(fn):
|
|
ret.append(fn + '/')
|
|
elif os.path.isfile(fn) and fn.lower().endswith(('.py', '.pyw')):
|
|
ret.append(fn + ':')
|
|
return ret
|
|
|
|
def _complete_bpnumber(self, text, line, begidx, endidx):
|
|
# Complete a breakpoint number. (This would be more helpful if we could
|
|
# display additional info along with the completions, such as file/line
|
|
# of the breakpoint.)
|
|
return [str(i) for i, bp in enumerate(bdb.Breakpoint.bpbynumber)
|
|
if bp is not None and str(i).startswith(text)]
|
|
|
|
def _complete_expression(self, text, line, begidx, endidx):
|
|
# Complete an arbitrary expression.
|
|
if not self.curframe:
|
|
return []
|
|
# Collect globals and locals. It is usually not really sensible to also
|
|
# complete builtins, and they clutter the namespace quite heavily, so we
|
|
# leave them out.
|
|
ns = {**self.curframe.f_globals, **self.curframe.f_locals}
|
|
if '.' in text:
|
|
# Walk an attribute chain up to the last part, similar to what
|
|
# rlcompleter does. This will bail if any of the parts are not
|
|
# simple attribute access, which is what we want.
|
|
dotted = text.split('.')
|
|
try:
|
|
if dotted[0].startswith('$'):
|
|
obj = self.curframe.f_globals['__pdb_convenience_variables'][dotted[0][1:]]
|
|
else:
|
|
obj = ns[dotted[0]]
|
|
for part in dotted[1:-1]:
|
|
obj = getattr(obj, part)
|
|
except (KeyError, AttributeError):
|
|
return []
|
|
prefix = '.'.join(dotted[:-1]) + '.'
|
|
return [prefix + n for n in dir(obj) if n.startswith(dotted[-1])]
|
|
else:
|
|
if text.startswith("$"):
|
|
# Complete convenience variables
|
|
conv_vars = self.curframe.f_globals.get('__pdb_convenience_variables', {})
|
|
return [f"${name}" for name in conv_vars if name.startswith(text[1:])]
|
|
# Complete a simple name.
|
|
return [n for n in ns.keys() if n.startswith(text)]
|
|
|
|
def _complete_indentation(self, text, line, begidx, endidx):
|
|
try:
|
|
import readline
|
|
except ImportError:
|
|
return []
|
|
# Fill in spaces to form a 4-space indent
|
|
return [' ' * (4 - readline.get_begidx() % 4)]
|
|
|
|
def complete_multiline_names(self, text, line, begidx, endidx):
|
|
# If text is space-only, the user entered <tab> before any text.
|
|
# That normally means they want to indent the current line.
|
|
if not text.strip():
|
|
return self._complete_indentation(text, line, begidx, endidx)
|
|
return self.completedefault(text, line, begidx, endidx)
|
|
|
|
def completedefault(self, text, line, begidx, endidx):
|
|
if text.startswith("$"):
|
|
# Complete convenience variables
|
|
conv_vars = self.curframe.f_globals.get('__pdb_convenience_variables', {})
|
|
return [f"${name}" for name in conv_vars if name.startswith(text[1:])]
|
|
|
|
# Use rlcompleter to do the completion
|
|
state = 0
|
|
matches = []
|
|
completer = Completer(self.curframe.f_globals | self.curframe.f_locals)
|
|
while (match := completer.complete(text, state)) is not None:
|
|
matches.append(match)
|
|
state += 1
|
|
return matches
|
|
|
|
@contextmanager
|
|
def _enable_rlcompleter(self, ns):
|
|
try:
|
|
import readline
|
|
except ImportError:
|
|
yield
|
|
return
|
|
|
|
try:
|
|
old_completer = readline.get_completer()
|
|
completer = Completer(ns)
|
|
readline.set_completer(completer.complete)
|
|
yield
|
|
finally:
|
|
readline.set_completer(old_completer)
|
|
|
|
# Pdb meta commands, only intended to be used internally by pdb
|
|
|
|
def _pdbcmd_print_frame_status(self, arg):
|
|
self.print_stack_trace(0)
|
|
self._validate_file_mtime()
|
|
self._show_display()
|
|
|
|
def _pdbcmd_silence_frame_status(self, arg):
|
|
if self.cmdqueue and self.cmdqueue[-1] == '_pdbcmd_print_frame_status':
|
|
self.cmdqueue.pop()
|
|
|
|
def _pdbcmd_restore_lastcmd(self, arg):
|
|
self.lastcmd = arg
|
|
|
|
# Command definitions, called by cmdloop()
|
|
# The argument is the remaining string on the command line
|
|
# Return true to exit from the command loop
|
|
|
|
def do_commands(self, arg):
|
|
"""(Pdb) commands [bpnumber]
|
|
(com) ...
|
|
(com) end
|
|
(Pdb)
|
|
|
|
Specify a list of commands for breakpoint number bpnumber.
|
|
The commands themselves are entered on the following lines.
|
|
Type a line containing just 'end' to terminate the commands.
|
|
The commands are executed when the breakpoint is hit.
|
|
|
|
To remove all commands from a breakpoint, type commands and
|
|
follow it immediately with end; that is, give no commands.
|
|
|
|
With no bpnumber argument, commands refers to the last
|
|
breakpoint set.
|
|
|
|
You can use breakpoint commands to start your program up
|
|
again. Simply use the continue command, or step, or any other
|
|
command that resumes execution.
|
|
|
|
Specifying any command resuming execution (currently continue,
|
|
step, next, return, jump, quit and their abbreviations)
|
|
terminates the command list (as if that command was
|
|
immediately followed by end). This is because any time you
|
|
resume execution (even with a simple next or step), you may
|
|
encounter another breakpoint -- which could have its own
|
|
command list, leading to ambiguities about which list to
|
|
execute.
|
|
|
|
If you use the 'silent' command in the command list, the usual
|
|
message about stopping at a breakpoint is not printed. This
|
|
may be desirable for breakpoints that are to print a specific
|
|
message and then continue. If none of the other commands
|
|
print anything, you will see no sign that the breakpoint was
|
|
reached.
|
|
"""
|
|
if not arg:
|
|
bnum = len(bdb.Breakpoint.bpbynumber) - 1
|
|
else:
|
|
try:
|
|
bnum = int(arg)
|
|
except:
|
|
self._print_invalid_arg(arg)
|
|
return
|
|
try:
|
|
self.get_bpbynumber(bnum)
|
|
except ValueError as err:
|
|
self.error('cannot set commands: %s' % err)
|
|
return
|
|
|
|
self.commands_bnum = bnum
|
|
# Save old definitions for the case of a keyboard interrupt.
|
|
if bnum in self.commands:
|
|
old_commands = self.commands[bnum]
|
|
else:
|
|
old_commands = None
|
|
self.commands[bnum] = []
|
|
|
|
prompt_back = self.prompt
|
|
self.prompt = '(com) '
|
|
self.commands_defining = True
|
|
try:
|
|
self.cmdloop()
|
|
except KeyboardInterrupt:
|
|
# Restore old definitions.
|
|
if old_commands:
|
|
self.commands[bnum] = old_commands
|
|
else:
|
|
del self.commands[bnum]
|
|
self.error('command definition aborted, old commands restored')
|
|
finally:
|
|
self.commands_defining = False
|
|
self.prompt = prompt_back
|
|
|
|
complete_commands = _complete_bpnumber
|
|
|
|
def do_break(self, arg, temporary=False):
|
|
"""b(reak) [ ([filename:]lineno | function) [, condition] ]
|
|
|
|
Without argument, list all breaks.
|
|
|
|
With a line number argument, set a break at this line in the
|
|
current file. With a function name, set a break at the first
|
|
executable line of that function. If a second argument is
|
|
present, it is a string specifying an expression which must
|
|
evaluate to true before the breakpoint is honored.
|
|
|
|
The line number may be prefixed with a filename and a colon,
|
|
to specify a breakpoint in another file (probably one that
|
|
hasn't been loaded yet). The file is searched for on
|
|
sys.path; the .py suffix may be omitted.
|
|
"""
|
|
if not arg:
|
|
if self.breaks: # There's at least one
|
|
self.message("Num Type Disp Enb Where")
|
|
for bp in bdb.Breakpoint.bpbynumber:
|
|
if bp:
|
|
self.message(bp.bpformat())
|
|
return
|
|
# parse arguments; comma has lowest precedence
|
|
# and cannot occur in filename
|
|
filename = None
|
|
lineno = None
|
|
cond = None
|
|
module_globals = None
|
|
comma = arg.find(',')
|
|
if comma > 0:
|
|
# parse stuff after comma: "condition"
|
|
cond = arg[comma+1:].lstrip()
|
|
if err := self._compile_error_message(cond):
|
|
self.error('Invalid condition %s: %r' % (cond, err))
|
|
return
|
|
arg = arg[:comma].rstrip()
|
|
# parse stuff before comma: [filename:]lineno | function
|
|
colon = arg.rfind(':')
|
|
funcname = None
|
|
if colon >= 0:
|
|
filename = arg[:colon].rstrip()
|
|
f = self.lookupmodule(filename)
|
|
if not f:
|
|
self.error('%r not found from sys.path' % filename)
|
|
return
|
|
else:
|
|
filename = f
|
|
arg = arg[colon+1:].lstrip()
|
|
try:
|
|
lineno = int(arg)
|
|
except ValueError:
|
|
self.error('Bad lineno: %s' % arg)
|
|
return
|
|
else:
|
|
# no colon; can be lineno or function
|
|
try:
|
|
lineno = int(arg)
|
|
except ValueError:
|
|
try:
|
|
func = eval(arg,
|
|
self.curframe.f_globals,
|
|
self.curframe.f_locals)
|
|
except:
|
|
func = arg
|
|
try:
|
|
if hasattr(func, '__func__'):
|
|
func = func.__func__
|
|
code = func.__code__
|
|
#use co_name to identify the bkpt (function names
|
|
#could be aliased, but co_name is invariant)
|
|
funcname = code.co_name
|
|
lineno = find_first_executable_line(code)
|
|
filename = code.co_filename
|
|
module_globals = func.__globals__
|
|
except:
|
|
# last thing to try
|
|
(ok, filename, ln) = self.lineinfo(arg)
|
|
if not ok:
|
|
self.error('The specified object %r is not a function '
|
|
'or was not found along sys.path.' % arg)
|
|
return
|
|
funcname = ok # ok contains a function name
|
|
lineno = int(ln)
|
|
if not filename:
|
|
filename = self.defaultFile()
|
|
filename = self.canonic(filename)
|
|
# Check for reasonable breakpoint
|
|
line = self.checkline(filename, lineno, module_globals)
|
|
if line:
|
|
# now set the break point
|
|
err = self.set_break(filename, line, temporary, cond, funcname)
|
|
if err:
|
|
self.error(err)
|
|
else:
|
|
bp = self.get_breaks(filename, line)[-1]
|
|
self.message("Breakpoint %d at %s:%d" %
|
|
(bp.number, bp.file, bp.line))
|
|
|
|
# To be overridden in derived debuggers
|
|
def defaultFile(self):
|
|
"""Produce a reasonable default."""
|
|
filename = self.curframe.f_code.co_filename
|
|
if filename == '<string>' and self.mainpyfile:
|
|
filename = self.mainpyfile
|
|
return filename
|
|
|
|
do_b = do_break
|
|
|
|
complete_break = _complete_location
|
|
complete_b = _complete_location
|
|
|
|
def do_tbreak(self, arg):
|
|
"""tbreak [ ([filename:]lineno | function) [, condition] ]
|
|
|
|
Same arguments as break, but sets a temporary breakpoint: it
|
|
is automatically deleted when first hit.
|
|
"""
|
|
self.do_break(arg, True)
|
|
|
|
complete_tbreak = _complete_location
|
|
|
|
def lineinfo(self, identifier):
|
|
failed = (None, None, None)
|
|
# Input is identifier, may be in single quotes
|
|
idstring = identifier.split("'")
|
|
if len(idstring) == 1:
|
|
# not in single quotes
|
|
id = idstring[0].strip()
|
|
elif len(idstring) == 3:
|
|
# quoted
|
|
id = idstring[1].strip()
|
|
else:
|
|
return failed
|
|
if id == '': return failed
|
|
parts = id.split('.')
|
|
# Protection for derived debuggers
|
|
if parts[0] == 'self':
|
|
del parts[0]
|
|
if len(parts) == 0:
|
|
return failed
|
|
# Best first guess at file to look at
|
|
fname = self.defaultFile()
|
|
if len(parts) == 1:
|
|
item = parts[0]
|
|
else:
|
|
# More than one part.
|
|
# First is module, second is method/class
|
|
f = self.lookupmodule(parts[0])
|
|
if f:
|
|
fname = f
|
|
item = parts[1]
|
|
answer = find_function(item, self.canonic(fname))
|
|
return answer or failed
|
|
|
|
def checkline(self, filename, lineno, module_globals=None):
|
|
"""Check whether specified line seems to be executable.
|
|
|
|
Return `lineno` if it is, 0 if not (e.g. a docstring, comment, blank
|
|
line or EOF). Warning: testing is not comprehensive.
|
|
"""
|
|
# this method should be callable before starting debugging, so default
|
|
# to "no globals" if there is no current frame
|
|
frame = getattr(self, 'curframe', None)
|
|
if module_globals is None:
|
|
module_globals = frame.f_globals if frame else None
|
|
line = linecache.getline(filename, lineno, module_globals)
|
|
if not line:
|
|
self.message('End of file')
|
|
return 0
|
|
line = line.strip()
|
|
# Don't allow setting breakpoint at a blank line
|
|
if (not line or (line[0] == '#') or
|
|
(line[:3] == '"""') or line[:3] == "'''"):
|
|
self.error('Blank or comment')
|
|
return 0
|
|
return lineno
|
|
|
|
def do_enable(self, arg):
|
|
"""enable bpnumber [bpnumber ...]
|
|
|
|
Enables the breakpoints given as a space separated list of
|
|
breakpoint numbers.
|
|
"""
|
|
if not arg:
|
|
self._print_invalid_arg(arg)
|
|
return
|
|
args = arg.split()
|
|
for i in args:
|
|
try:
|
|
bp = self.get_bpbynumber(i)
|
|
except ValueError as err:
|
|
self.error(err)
|
|
else:
|
|
bp.enable()
|
|
self.message('Enabled %s' % bp)
|
|
|
|
complete_enable = _complete_bpnumber
|
|
|
|
def do_disable(self, arg):
|
|
"""disable bpnumber [bpnumber ...]
|
|
|
|
Disables the breakpoints given as a space separated list of
|
|
breakpoint numbers. Disabling a breakpoint means it cannot
|
|
cause the program to stop execution, but unlike clearing a
|
|
breakpoint, it remains in the list of breakpoints and can be
|
|
(re-)enabled.
|
|
"""
|
|
if not arg:
|
|
self._print_invalid_arg(arg)
|
|
return
|
|
args = arg.split()
|
|
for i in args:
|
|
try:
|
|
bp = self.get_bpbynumber(i)
|
|
except ValueError as err:
|
|
self.error(err)
|
|
else:
|
|
bp.disable()
|
|
self.message('Disabled %s' % bp)
|
|
|
|
complete_disable = _complete_bpnumber
|
|
|
|
def do_condition(self, arg):
|
|
"""condition bpnumber [condition]
|
|
|
|
Set a new condition for the breakpoint, an expression which
|
|
must evaluate to true before the breakpoint is honored. If
|
|
condition is absent, any existing condition is removed; i.e.,
|
|
the breakpoint is made unconditional.
|
|
"""
|
|
if not arg:
|
|
self._print_invalid_arg(arg)
|
|
return
|
|
args = arg.split(' ', 1)
|
|
try:
|
|
cond = args[1]
|
|
if err := self._compile_error_message(cond):
|
|
self.error('Invalid condition %s: %r' % (cond, err))
|
|
return
|
|
except IndexError:
|
|
cond = None
|
|
try:
|
|
bp = self.get_bpbynumber(args[0].strip())
|
|
except IndexError:
|
|
self.error('Breakpoint number expected')
|
|
except ValueError as err:
|
|
self.error(err)
|
|
else:
|
|
bp.cond = cond
|
|
if not cond:
|
|
self.message('Breakpoint %d is now unconditional.' % bp.number)
|
|
else:
|
|
self.message('New condition set for breakpoint %d.' % bp.number)
|
|
|
|
complete_condition = _complete_bpnumber
|
|
|
|
def do_ignore(self, arg):
|
|
"""ignore bpnumber [count]
|
|
|
|
Set the ignore count for the given breakpoint number. If
|
|
count is omitted, the ignore count is set to 0. A breakpoint
|
|
becomes active when the ignore count is zero. When non-zero,
|
|
the count is decremented each time the breakpoint is reached
|
|
and the breakpoint is not disabled and any associated
|
|
condition evaluates to true.
|
|
"""
|
|
if not arg:
|
|
self._print_invalid_arg(arg)
|
|
return
|
|
args = arg.split()
|
|
if not args:
|
|
self.error('Breakpoint number expected')
|
|
return
|
|
if len(args) == 1:
|
|
count = 0
|
|
elif len(args) == 2:
|
|
try:
|
|
count = int(args[1])
|
|
except ValueError:
|
|
self._print_invalid_arg(arg)
|
|
return
|
|
else:
|
|
self._print_invalid_arg(arg)
|
|
return
|
|
try:
|
|
bp = self.get_bpbynumber(args[0].strip())
|
|
except ValueError as err:
|
|
self.error(err)
|
|
else:
|
|
bp.ignore = count
|
|
if count > 0:
|
|
if count > 1:
|
|
countstr = '%d crossings' % count
|
|
else:
|
|
countstr = '1 crossing'
|
|
self.message('Will ignore next %s of breakpoint %d.' %
|
|
(countstr, bp.number))
|
|
else:
|
|
self.message('Will stop next time breakpoint %d is reached.'
|
|
% bp.number)
|
|
|
|
complete_ignore = _complete_bpnumber
|
|
|
|
def _prompt_for_confirmation(self, prompt, default):
|
|
try:
|
|
reply = input(prompt)
|
|
except EOFError:
|
|
reply = default
|
|
return reply.strip().lower()
|
|
|
|
def do_clear(self, arg):
|
|
"""cl(ear) [filename:lineno | bpnumber ...]
|
|
|
|
With a space separated list of breakpoint numbers, clear
|
|
those breakpoints. Without argument, clear all breaks (but
|
|
first ask confirmation). With a filename:lineno argument,
|
|
clear all breaks at that line in that file.
|
|
"""
|
|
if not arg:
|
|
reply = self._prompt_for_confirmation(
|
|
'Clear all breaks? ',
|
|
default='no',
|
|
)
|
|
if reply in ('y', 'yes'):
|
|
bplist = [bp for bp in bdb.Breakpoint.bpbynumber if bp]
|
|
self.clear_all_breaks()
|
|
for bp in bplist:
|
|
self.message('Deleted %s' % bp)
|
|
return
|
|
if ':' in arg:
|
|
# Make sure it works for "clear C:\foo\bar.py:12"
|
|
i = arg.rfind(':')
|
|
filename = arg[:i]
|
|
arg = arg[i+1:]
|
|
try:
|
|
lineno = int(arg)
|
|
except ValueError:
|
|
err = "Invalid line number (%s)" % arg
|
|
else:
|
|
bplist = self.get_breaks(filename, lineno)[:]
|
|
err = self.clear_break(filename, lineno)
|
|
if err:
|
|
self.error(err)
|
|
else:
|
|
for bp in bplist:
|
|
self.message('Deleted %s' % bp)
|
|
return
|
|
numberlist = arg.split()
|
|
for i in numberlist:
|
|
try:
|
|
bp = self.get_bpbynumber(i)
|
|
except ValueError as err:
|
|
self.error(err)
|
|
else:
|
|
self.clear_bpbynumber(i)
|
|
self.message('Deleted %s' % bp)
|
|
do_cl = do_clear # 'c' is already an abbreviation for 'continue'
|
|
|
|
complete_clear = _complete_location
|
|
complete_cl = _complete_location
|
|
|
|
def do_where(self, arg):
|
|
"""w(here) [count]
|
|
|
|
Print a stack trace. If count is not specified, print the full stack.
|
|
If count is 0, print the current frame entry. If count is positive,
|
|
print count entries from the most recent frame. If count is negative,
|
|
print -count entries from the least recent frame.
|
|
An arrow indicates the "current frame", which determines the
|
|
context of most commands. 'bt' is an alias for this command.
|
|
"""
|
|
if not arg:
|
|
count = None
|
|
else:
|
|
try:
|
|
count = int(arg)
|
|
except ValueError:
|
|
self.error('Invalid count (%s)' % arg)
|
|
return
|
|
self.print_stack_trace(count)
|
|
do_w = do_where
|
|
do_bt = do_where
|
|
|
|
def _select_frame(self, number):
|
|
assert 0 <= number < len(self.stack)
|
|
self.curindex = number
|
|
self.curframe = self.stack[self.curindex][0]
|
|
self.set_convenience_variable(self.curframe, '_frame', self.curframe)
|
|
self.print_stack_entry(self.stack[self.curindex])
|
|
self.lineno = None
|
|
|
|
def do_exceptions(self, arg):
|
|
"""exceptions [number]
|
|
|
|
List or change current exception in an exception chain.
|
|
|
|
Without arguments, list all the current exception in the exception
|
|
chain. Exceptions will be numbered, with the current exception indicated
|
|
with an arrow.
|
|
|
|
If given an integer as argument, switch to the exception at that index.
|
|
"""
|
|
if not self._chained_exceptions:
|
|
self.message(
|
|
"Did not find chained exceptions. To move between"
|
|
" exceptions, pdb/post_mortem must be given an exception"
|
|
" object rather than a traceback."
|
|
)
|
|
return
|
|
if not arg:
|
|
for ix, exc in enumerate(self._chained_exceptions):
|
|
prompt = ">" if ix == self._chained_exception_index else " "
|
|
rep = repr(exc)
|
|
if len(rep) > 80:
|
|
rep = rep[:77] + "..."
|
|
indicator = (
|
|
" -"
|
|
if self._chained_exceptions[ix].__traceback__ is None
|
|
else f"{ix:>3}"
|
|
)
|
|
self.message(f"{prompt} {indicator} {rep}")
|
|
else:
|
|
try:
|
|
number = int(arg)
|
|
except ValueError:
|
|
self.error("Argument must be an integer")
|
|
return
|
|
if 0 <= number < len(self._chained_exceptions):
|
|
if self._chained_exceptions[number].__traceback__ is None:
|
|
self.error("This exception does not have a traceback, cannot jump to it")
|
|
return
|
|
|
|
self._chained_exception_index = number
|
|
self.setup(None, self._chained_exceptions[number].__traceback__)
|
|
self.print_stack_entry(self.stack[self.curindex])
|
|
else:
|
|
self.error("No exception with that number")
|
|
|
|
def do_up(self, arg):
|
|
"""u(p) [count]
|
|
|
|
Move the current frame count (default one) levels up in the
|
|
stack trace (to an older frame).
|
|
"""
|
|
if self.curindex == 0:
|
|
self.error('Oldest frame')
|
|
return
|
|
try:
|
|
count = int(arg or 1)
|
|
except ValueError:
|
|
self.error('Invalid frame count (%s)' % arg)
|
|
return
|
|
if count < 0:
|
|
newframe = 0
|
|
else:
|
|
newframe = max(0, self.curindex - count)
|
|
self._select_frame(newframe)
|
|
do_u = do_up
|
|
|
|
def do_down(self, arg):
|
|
"""d(own) [count]
|
|
|
|
Move the current frame count (default one) levels down in the
|
|
stack trace (to a newer frame).
|
|
"""
|
|
if self.curindex + 1 == len(self.stack):
|
|
self.error('Newest frame')
|
|
return
|
|
try:
|
|
count = int(arg or 1)
|
|
except ValueError:
|
|
self.error('Invalid frame count (%s)' % arg)
|
|
return
|
|
if count < 0:
|
|
newframe = len(self.stack) - 1
|
|
else:
|
|
newframe = min(len(self.stack) - 1, self.curindex + count)
|
|
self._select_frame(newframe)
|
|
do_d = do_down
|
|
|
|
def do_until(self, arg):
|
|
"""unt(il) [lineno]
|
|
|
|
Without argument, continue execution until the line with a
|
|
number greater than the current one is reached. With a line
|
|
number, continue execution until a line with a number greater
|
|
or equal to that is reached. In both cases, also stop when
|
|
the current frame returns.
|
|
"""
|
|
if arg:
|
|
try:
|
|
lineno = int(arg)
|
|
except ValueError:
|
|
self.error('Error in argument: %r' % arg)
|
|
return
|
|
if lineno <= self.curframe.f_lineno:
|
|
self.error('"until" line number is smaller than current '
|
|
'line number')
|
|
return
|
|
else:
|
|
lineno = None
|
|
self.set_until(self.curframe, lineno)
|
|
return 1
|
|
do_unt = do_until
|
|
|
|
def do_step(self, arg):
|
|
"""s(tep)
|
|
|
|
Execute the current line, stop at the first possible occasion
|
|
(either in a function that is called or in the current
|
|
function).
|
|
"""
|
|
if arg:
|
|
self._print_invalid_arg(arg)
|
|
return
|
|
self.set_step()
|
|
return 1
|
|
do_s = do_step
|
|
|
|
def do_next(self, arg):
|
|
"""n(ext)
|
|
|
|
Continue execution until the next line in the current function
|
|
is reached or it returns.
|
|
"""
|
|
if arg:
|
|
self._print_invalid_arg(arg)
|
|
return
|
|
self.set_next(self.curframe)
|
|
return 1
|
|
do_n = do_next
|
|
|
|
def do_run(self, arg):
|
|
"""run [args...]
|
|
|
|
Restart the debugged python program. If a string is supplied
|
|
it is split with "shlex", and the result is used as the new
|
|
sys.argv. History, breakpoints, actions and debugger options
|
|
are preserved. "restart" is an alias for "run".
|
|
"""
|
|
if self.mode == 'inline':
|
|
self.error('run/restart command is disabled when pdb is running in inline mode.\n'
|
|
'Use the command line interface to enable restarting your program\n'
|
|
'e.g. "python -m pdb myscript.py"')
|
|
return
|
|
if arg:
|
|
import shlex
|
|
argv0 = sys.argv[0:1]
|
|
try:
|
|
sys.argv = shlex.split(arg)
|
|
except ValueError as e:
|
|
self.error('Cannot run %s: %s' % (arg, e))
|
|
return
|
|
sys.argv[:0] = argv0
|
|
# this is caught in the main debugger loop
|
|
raise Restart
|
|
|
|
do_restart = do_run
|
|
|
|
def do_return(self, arg):
|
|
"""r(eturn)
|
|
|
|
Continue execution until the current function returns.
|
|
"""
|
|
if arg:
|
|
self._print_invalid_arg(arg)
|
|
return
|
|
self.set_return(self.curframe)
|
|
return 1
|
|
do_r = do_return
|
|
|
|
def do_continue(self, arg):
|
|
"""c(ont(inue))
|
|
|
|
Continue execution, only stop when a breakpoint is encountered.
|
|
"""
|
|
if arg:
|
|
self._print_invalid_arg(arg)
|
|
return
|
|
if not self.nosigint:
|
|
try:
|
|
Pdb._previous_sigint_handler = \
|
|
signal.signal(signal.SIGINT, self.sigint_handler)
|
|
except ValueError:
|
|
# ValueError happens when do_continue() is invoked from
|
|
# a non-main thread in which case we just continue without
|
|
# SIGINT set. Would printing a message here (once) make
|
|
# sense?
|
|
pass
|
|
self.set_continue()
|
|
return 1
|
|
do_c = do_cont = do_continue
|
|
|
|
def do_jump(self, arg):
|
|
"""j(ump) lineno
|
|
|
|
Set the next line that will be executed. Only available in
|
|
the bottom-most frame. This lets you jump back and execute
|
|
code again, or jump forward to skip code that you don't want
|
|
to run.
|
|
|
|
It should be noted that not all jumps are allowed -- for
|
|
instance it is not possible to jump into the middle of a
|
|
for loop or out of a finally clause.
|
|
"""
|
|
if not arg:
|
|
self._print_invalid_arg(arg)
|
|
return
|
|
if self.curindex + 1 != len(self.stack):
|
|
self.error('You can only jump within the bottom frame')
|
|
return
|
|
try:
|
|
arg = int(arg)
|
|
except ValueError:
|
|
self.error("The 'jump' command requires a line number")
|
|
else:
|
|
try:
|
|
# Do the jump, fix up our copy of the stack, and display the
|
|
# new position
|
|
self.curframe.f_lineno = arg
|
|
self.stack[self.curindex] = self.stack[self.curindex][0], arg
|
|
self.print_stack_entry(self.stack[self.curindex])
|
|
except ValueError as e:
|
|
self.error('Jump failed: %s' % e)
|
|
do_j = do_jump
|
|
|
|
def _create_recursive_debugger(self):
|
|
return Pdb(self.completekey, self.stdin, self.stdout)
|
|
|
|
def do_debug(self, arg):
|
|
"""debug code
|
|
|
|
Enter a recursive debugger that steps through the code
|
|
argument (which is an arbitrary expression or statement to be
|
|
executed in the current environment).
|
|
"""
|
|
if not arg:
|
|
self._print_invalid_arg(arg)
|
|
return
|
|
self.stop_trace()
|
|
globals = self.curframe.f_globals
|
|
locals = self.curframe.f_locals
|
|
p = self._create_recursive_debugger()
|
|
p.prompt = "(%s) " % self.prompt.strip()
|
|
self.message("ENTERING RECURSIVE DEBUGGER")
|
|
try:
|
|
sys.call_tracing(p.run, (arg, globals, locals))
|
|
except Exception:
|
|
self._error_exc()
|
|
self.message("LEAVING RECURSIVE DEBUGGER")
|
|
self.start_trace()
|
|
self.lastcmd = p.lastcmd
|
|
|
|
complete_debug = _complete_expression
|
|
|
|
def do_quit(self, arg):
|
|
"""q(uit) | exit
|
|
|
|
Quit from the debugger. The program being executed is aborted.
|
|
"""
|
|
# Show prompt to kill process when in 'inline' mode and if pdb was not
|
|
# started from an interactive console. The attribute sys.ps1 is only
|
|
# defined if the interpreter is in interactive mode.
|
|
if self.mode == 'inline' and not hasattr(sys, 'ps1'):
|
|
while True:
|
|
try:
|
|
reply = input('Quitting pdb will kill the process. Quit anyway? [y/n] ')
|
|
reply = reply.lower().strip()
|
|
except EOFError:
|
|
reply = 'y'
|
|
self.message('')
|
|
if reply == 'y' or reply == '':
|
|
sys.exit(1)
|
|
elif reply.lower() == 'n':
|
|
return
|
|
|
|
self._user_requested_quit = True
|
|
self.set_quit()
|
|
return 1
|
|
|
|
do_q = do_quit
|
|
do_exit = do_quit
|
|
|
|
def do_EOF(self, arg):
|
|
"""EOF
|
|
|
|
Handles the receipt of EOF as a command.
|
|
"""
|
|
self.message('')
|
|
return self.do_quit(arg)
|
|
|
|
def do_args(self, arg):
|
|
"""a(rgs)
|
|
|
|
Print the argument list of the current function.
|
|
"""
|
|
if arg:
|
|
self._print_invalid_arg(arg)
|
|
return
|
|
co = self.curframe.f_code
|
|
dict = self.curframe.f_locals
|
|
n = co.co_argcount + co.co_kwonlyargcount
|
|
if co.co_flags & inspect.CO_VARARGS: n = n+1
|
|
if co.co_flags & inspect.CO_VARKEYWORDS: n = n+1
|
|
for i in range(n):
|
|
name = co.co_varnames[i]
|
|
if name in dict:
|
|
self.message('%s = %s' % (name, self._safe_repr(dict[name], name)))
|
|
else:
|
|
self.message('%s = *** undefined ***' % (name,))
|
|
do_a = do_args
|
|
|
|
def do_retval(self, arg):
|
|
"""retval
|
|
|
|
Print the return value for the last return of a function.
|
|
"""
|
|
if arg:
|
|
self._print_invalid_arg(arg)
|
|
return
|
|
if '__return__' in self.curframe.f_locals:
|
|
self.message(self._safe_repr(self.curframe.f_locals['__return__'], "retval"))
|
|
else:
|
|
self.error('Not yet returned!')
|
|
do_rv = do_retval
|
|
|
|
def _getval(self, arg):
|
|
try:
|
|
return eval(arg, self.curframe.f_globals, self.curframe.f_locals)
|
|
except:
|
|
self._error_exc()
|
|
raise
|
|
|
|
def _getval_except(self, arg, frame=None):
|
|
try:
|
|
if frame is None:
|
|
return eval(arg, self.curframe.f_globals, self.curframe.f_locals)
|
|
else:
|
|
return eval(arg, frame.f_globals, frame.f_locals)
|
|
except BaseException as exc:
|
|
return _rstr('** raised %s **' % self._format_exc(exc))
|
|
|
|
def _error_exc(self):
|
|
exc = sys.exception()
|
|
self.error(self._format_exc(exc))
|
|
|
|
def _msg_val_func(self, arg, func):
|
|
try:
|
|
val = self._getval(arg)
|
|
except:
|
|
return # _getval() has displayed the error
|
|
try:
|
|
self.message(func(val))
|
|
except:
|
|
self._error_exc()
|
|
|
|
def _safe_repr(self, obj, expr):
|
|
try:
|
|
return repr(obj)
|
|
except Exception as e:
|
|
return _rstr(f"*** repr({expr}) failed: {self._format_exc(e)} ***")
|
|
|
|
def do_p(self, arg):
|
|
"""p expression
|
|
|
|
Print the value of the expression.
|
|
"""
|
|
if not arg:
|
|
self._print_invalid_arg(arg)
|
|
return
|
|
self._msg_val_func(arg, repr)
|
|
|
|
def do_pp(self, arg):
|
|
"""pp expression
|
|
|
|
Pretty-print the value of the expression.
|
|
"""
|
|
if not arg:
|
|
self._print_invalid_arg(arg)
|
|
return
|
|
self._msg_val_func(arg, pprint.pformat)
|
|
|
|
complete_print = _complete_expression
|
|
complete_p = _complete_expression
|
|
complete_pp = _complete_expression
|
|
|
|
def do_list(self, arg):
|
|
"""l(ist) [first[, last] | .]
|
|
|
|
List source code for the current file. Without arguments,
|
|
list 11 lines around the current line or continue the previous
|
|
listing. With . as argument, list 11 lines around the current
|
|
line. With one argument, list 11 lines starting at that line.
|
|
With two arguments, list the given range; if the second
|
|
argument is less than the first, it is a count.
|
|
|
|
The current line in the current frame is indicated by "->".
|
|
If an exception is being debugged, the line where the
|
|
exception was originally raised or propagated is indicated by
|
|
">>", if it differs from the current line.
|
|
"""
|
|
self.lastcmd = 'list'
|
|
last = None
|
|
if arg and arg != '.':
|
|
try:
|
|
if ',' in arg:
|
|
first, last = arg.split(',')
|
|
first = int(first.strip())
|
|
last = int(last.strip())
|
|
if last < first:
|
|
# assume it's a count
|
|
last = first + last
|
|
else:
|
|
first = int(arg.strip())
|
|
first = max(1, first - 5)
|
|
except ValueError:
|
|
self.error('Error in argument: %r' % arg)
|
|
return
|
|
elif self.lineno is None or arg == '.':
|
|
first = max(1, self.curframe.f_lineno - 5)
|
|
else:
|
|
first = self.lineno + 1
|
|
if last is None:
|
|
last = first + 10
|
|
filename = self.curframe.f_code.co_filename
|
|
breaklist = self.get_file_breaks(filename)
|
|
try:
|
|
lines = linecache.getlines(filename, self.curframe.f_globals)
|
|
self._print_lines(lines[first-1:last], first, breaklist,
|
|
self.curframe)
|
|
self.lineno = min(last, len(lines))
|
|
if len(lines) < last:
|
|
self.message('[EOF]')
|
|
except KeyboardInterrupt:
|
|
pass
|
|
self._validate_file_mtime()
|
|
do_l = do_list
|
|
|
|
def do_longlist(self, arg):
|
|
"""ll | longlist
|
|
|
|
List the whole source code for the current function or frame.
|
|
"""
|
|
if arg:
|
|
self._print_invalid_arg(arg)
|
|
return
|
|
filename = self.curframe.f_code.co_filename
|
|
breaklist = self.get_file_breaks(filename)
|
|
try:
|
|
lines, lineno = self._getsourcelines(self.curframe)
|
|
except OSError as err:
|
|
self.error(err)
|
|
return
|
|
self._print_lines(lines, lineno, breaklist, self.curframe)
|
|
self._validate_file_mtime()
|
|
do_ll = do_longlist
|
|
|
|
def do_source(self, arg):
|
|
"""source expression
|
|
|
|
Try to get source code for the given object and display it.
|
|
"""
|
|
if not arg:
|
|
self._print_invalid_arg(arg)
|
|
return
|
|
try:
|
|
obj = self._getval(arg)
|
|
except:
|
|
return
|
|
try:
|
|
lines, lineno = self._getsourcelines(obj)
|
|
except (OSError, TypeError) as err:
|
|
self.error(err)
|
|
return
|
|
self._print_lines(lines, lineno)
|
|
|
|
complete_source = _complete_expression
|
|
|
|
def _print_lines(self, lines, start, breaks=(), frame=None):
|
|
"""Print a range of lines."""
|
|
if frame:
|
|
current_lineno = frame.f_lineno
|
|
exc_lineno = self.tb_lineno.get(frame, -1)
|
|
else:
|
|
current_lineno = exc_lineno = -1
|
|
for lineno, line in enumerate(lines, start):
|
|
s = str(lineno).rjust(3)
|
|
if len(s) < 4:
|
|
s += ' '
|
|
if lineno in breaks:
|
|
s += 'B'
|
|
else:
|
|
s += ' '
|
|
if lineno == current_lineno:
|
|
s += '->'
|
|
elif lineno == exc_lineno:
|
|
s += '>>'
|
|
if self.colorize:
|
|
line = self._colorize_code(line)
|
|
self.message(s + '\t' + line.rstrip())
|
|
|
|
def do_whatis(self, arg):
|
|
"""whatis expression
|
|
|
|
Print the type of the argument.
|
|
"""
|
|
if not arg:
|
|
self._print_invalid_arg(arg)
|
|
return
|
|
try:
|
|
value = self._getval(arg)
|
|
except:
|
|
# _getval() already printed the error
|
|
return
|
|
code = None
|
|
# Is it an instance method?
|
|
try:
|
|
code = value.__func__.__code__
|
|
except Exception:
|
|
pass
|
|
if code:
|
|
self.message('Method %s' % code.co_name)
|
|
return
|
|
# Is it a function?
|
|
try:
|
|
code = value.__code__
|
|
except Exception:
|
|
pass
|
|
if code:
|
|
self.message('Function %s' % code.co_name)
|
|
return
|
|
# Is it a class?
|
|
if value.__class__ is type:
|
|
self.message('Class %s.%s' % (value.__module__, value.__qualname__))
|
|
return
|
|
# None of the above...
|
|
self.message(type(value))
|
|
|
|
complete_whatis = _complete_expression
|
|
|
|
def do_display(self, arg):
|
|
"""display [expression]
|
|
|
|
Display the value of the expression if it changed, each time execution
|
|
stops in the current frame.
|
|
|
|
Without expression, list all display expressions for the current frame.
|
|
"""
|
|
if not arg:
|
|
if self.displaying:
|
|
self.message('Currently displaying:')
|
|
for key, val in self.displaying.get(self.curframe, {}).items():
|
|
self.message('%s: %s' % (key, self._safe_repr(val, key)))
|
|
else:
|
|
self.message('No expression is being displayed')
|
|
else:
|
|
if err := self._compile_error_message(arg):
|
|
self.error('Unable to display %s: %r' % (arg, err))
|
|
else:
|
|
val = self._getval_except(arg)
|
|
self.displaying.setdefault(self.curframe, {})[arg] = val
|
|
self.message('display %s: %s' % (arg, self._safe_repr(val, arg)))
|
|
|
|
complete_display = _complete_expression
|
|
|
|
def do_undisplay(self, arg):
|
|
"""undisplay [expression]
|
|
|
|
Do not display the expression any more in the current frame.
|
|
|
|
Without expression, clear all display expressions for the current frame.
|
|
"""
|
|
if arg:
|
|
try:
|
|
del self.displaying.get(self.curframe, {})[arg]
|
|
except KeyError:
|
|
self.error('not displaying %s' % arg)
|
|
else:
|
|
self.displaying.pop(self.curframe, None)
|
|
|
|
def complete_undisplay(self, text, line, begidx, endidx):
|
|
return [e for e in self.displaying.get(self.curframe, {})
|
|
if e.startswith(text)]
|
|
|
|
def do_interact(self, arg):
|
|
"""interact
|
|
|
|
Start an interactive interpreter whose global namespace
|
|
contains all the (global and local) names found in the current scope.
|
|
"""
|
|
ns = {**self.curframe.f_globals, **self.curframe.f_locals}
|
|
with self._enable_rlcompleter(ns):
|
|
console = _PdbInteractiveConsole(ns, message=self.message)
|
|
console.interact(banner="*pdb interact start*",
|
|
exitmsg="*exit from pdb interact command*")
|
|
|
|
def do_alias(self, arg):
|
|
"""alias [name [command]]
|
|
|
|
Create an alias called 'name' that executes 'command'. The
|
|
command must *not* be enclosed in quotes. Replaceable
|
|
parameters can be indicated by %1, %2, and so on, while %* is
|
|
replaced by all the parameters. If no command is given, the
|
|
current alias for name is shown. If no name is given, all
|
|
aliases are listed.
|
|
|
|
Aliases may be nested and can contain anything that can be
|
|
legally typed at the pdb prompt. Note! You *can* override
|
|
internal pdb commands with aliases! Those internal commands
|
|
are then hidden until the alias is removed. Aliasing is
|
|
recursively applied to the first word of the command line; all
|
|
other words in the line are left alone.
|
|
|
|
As an example, here are two useful aliases (especially when
|
|
placed in the .pdbrc file):
|
|
|
|
# Print instance variables (usage "pi classInst")
|
|
alias pi for k in %1.__dict__.keys(): print("%1.",k,"=",%1.__dict__[k])
|
|
# Print instance variables in self
|
|
alias ps pi self
|
|
"""
|
|
args = arg.split()
|
|
if len(args) == 0:
|
|
keys = sorted(self.aliases.keys())
|
|
for alias in keys:
|
|
self.message("%s = %s" % (alias, self.aliases[alias]))
|
|
return
|
|
if len(args) == 1:
|
|
if args[0] in self.aliases:
|
|
self.message("%s = %s" % (args[0], self.aliases[args[0]]))
|
|
else:
|
|
self.error(f"Unknown alias '{args[0]}'")
|
|
else:
|
|
# Do a validation check to make sure no replaceable parameters
|
|
# are skipped if %* is not used.
|
|
alias = ' '.join(args[1:])
|
|
if '%*' not in alias:
|
|
consecutive = True
|
|
for idx in range(1, 10):
|
|
if f'%{idx}' not in alias:
|
|
consecutive = False
|
|
if f'%{idx}' in alias and not consecutive:
|
|
self.error("Replaceable parameters must be consecutive")
|
|
return
|
|
self.aliases[args[0]] = alias
|
|
|
|
def do_unalias(self, arg):
|
|
"""unalias name
|
|
|
|
Delete the specified alias.
|
|
"""
|
|
args = arg.split()
|
|
if len(args) == 0:
|
|
self._print_invalid_arg(arg)
|
|
return
|
|
if args[0] in self.aliases:
|
|
del self.aliases[args[0]]
|
|
|
|
def complete_unalias(self, text, line, begidx, endidx):
|
|
return [a for a in self.aliases if a.startswith(text)]
|
|
|
|
# List of all the commands making the program resume execution.
|
|
commands_resuming = ['do_continue', 'do_step', 'do_next', 'do_return',
|
|
'do_until', 'do_quit', 'do_jump']
|
|
|
|
# Print a traceback starting at the top stack frame.
|
|
# The most recently entered frame is printed last;
|
|
# this is different from dbx and gdb, but consistent with
|
|
# the Python interpreter's stack trace.
|
|
# It is also consistent with the up/down commands (which are
|
|
# compatible with dbx and gdb: up moves towards 'main()'
|
|
# and down moves towards the most recent stack frame).
|
|
# * if count is None, prints the full stack
|
|
# * if count = 0, prints the current frame entry
|
|
# * if count < 0, prints -count least recent frame entries
|
|
# * if count > 0, prints count most recent frame entries
|
|
|
|
def print_stack_trace(self, count=None):
|
|
if count is None:
|
|
stack_to_print = self.stack
|
|
elif count == 0:
|
|
stack_to_print = [self.stack[self.curindex]]
|
|
elif count < 0:
|
|
stack_to_print = self.stack[:-count]
|
|
else:
|
|
stack_to_print = self.stack[-count:]
|
|
try:
|
|
for frame_lineno in stack_to_print:
|
|
self.print_stack_entry(frame_lineno)
|
|
except KeyboardInterrupt:
|
|
pass
|
|
|
|
def print_stack_entry(self, frame_lineno, prompt_prefix=line_prefix):
|
|
frame, lineno = frame_lineno
|
|
if frame is self.curframe:
|
|
prefix = '> '
|
|
else:
|
|
prefix = ' '
|
|
stack_entry = self.format_stack_entry(frame_lineno, prompt_prefix)
|
|
if self.colorize:
|
|
lines = stack_entry.split(prompt_prefix, 1)
|
|
if len(lines) > 1:
|
|
# We have some code to display
|
|
lines[1] = self._colorize_code(lines[1])
|
|
stack_entry = prompt_prefix.join(lines)
|
|
self.message(prefix + stack_entry)
|
|
|
|
# Provide help
|
|
|
|
def do_help(self, arg):
|
|
"""h(elp)
|
|
|
|
Without argument, print the list of available commands.
|
|
With a command name as argument, print help about that command.
|
|
"help pdb" shows the full pdb documentation.
|
|
"help exec" gives help on the ! command.
|
|
"""
|
|
if not arg:
|
|
return cmd.Cmd.do_help(self, arg)
|
|
try:
|
|
try:
|
|
topic = getattr(self, 'help_' + arg)
|
|
return topic()
|
|
except AttributeError:
|
|
command = getattr(self, 'do_' + arg)
|
|
except AttributeError:
|
|
self.error('No help for %r' % arg)
|
|
else:
|
|
if sys.flags.optimize >= 2:
|
|
self.error('No help for %r; please do not run Python with -OO '
|
|
'if you need command help' % arg)
|
|
return
|
|
if command.__doc__ is None:
|
|
self.error('No help for %r; __doc__ string missing' % arg)
|
|
return
|
|
self.message(self._help_message_from_doc(command.__doc__))
|
|
|
|
do_h = do_help
|
|
|
|
def help_exec(self):
|
|
"""(!) statement
|
|
|
|
Execute the (one-line) statement in the context of the current
|
|
stack frame. The exclamation point can be omitted unless the
|
|
first word of the statement resembles a debugger command, e.g.:
|
|
(Pdb) ! n=42
|
|
(Pdb)
|
|
|
|
To assign to a global variable you must always prefix the command with
|
|
a 'global' command, e.g.:
|
|
(Pdb) global list_options; list_options = ['-l']
|
|
(Pdb)
|
|
"""
|
|
self.message((self.help_exec.__doc__ or '').strip())
|
|
|
|
def help_pdb(self):
|
|
help()
|
|
|
|
# other helper functions
|
|
|
|
def lookupmodule(self, filename):
|
|
"""Helper function for break/clear parsing -- may be overridden.
|
|
|
|
lookupmodule() translates (possibly incomplete) file or module name
|
|
into an absolute file name.
|
|
|
|
filename could be in format of:
|
|
* an absolute path like '/path/to/file.py'
|
|
* a relative path like 'file.py' or 'dir/file.py'
|
|
* a module name like 'module' or 'package.module'
|
|
|
|
files and modules will be searched in sys.path.
|
|
"""
|
|
if not filename.endswith('.py'):
|
|
# A module is passed in so convert it to equivalent file
|
|
filename = filename.replace('.', os.sep) + '.py'
|
|
|
|
if os.path.isabs(filename):
|
|
if os.path.exists(filename):
|
|
return filename
|
|
return None
|
|
|
|
for dirname in sys.path:
|
|
while os.path.islink(dirname):
|
|
dirname = os.readlink(dirname)
|
|
fullname = os.path.join(dirname, filename)
|
|
if os.path.exists(fullname):
|
|
return fullname
|
|
return None
|
|
|
|
def _run(self, target: _ExecutableTarget):
|
|
# When bdb sets tracing, a number of call and line events happen
|
|
# BEFORE debugger even reaches user's code (and the exact sequence of
|
|
# events depends on python version). Take special measures to
|
|
# avoid stopping before reaching the main script (see user_line and
|
|
# user_call for details).
|
|
self._wait_for_mainpyfile = True
|
|
self._user_requested_quit = False
|
|
|
|
self.mainpyfile = self.canonic(target.filename)
|
|
|
|
# The target has to run in __main__ namespace (or imports from
|
|
# __main__ will break). Clear __main__ and replace with
|
|
# the target namespace.
|
|
import __main__
|
|
__main__.__dict__.clear()
|
|
__main__.__dict__.update(target.namespace)
|
|
|
|
# Clear the mtime table for program reruns, assume all the files
|
|
# are up to date.
|
|
self._file_mtime_table.clear()
|
|
|
|
self.run(target.code)
|
|
|
|
def _format_exc(self, exc: BaseException):
|
|
return traceback.format_exception_only(exc)[-1].strip()
|
|
|
|
def _compile_error_message(self, expr):
|
|
"""Return the error message as string if compiling `expr` fails."""
|
|
try:
|
|
compile(expr, "<stdin>", "eval")
|
|
except SyntaxError as exc:
|
|
return _rstr(self._format_exc(exc))
|
|
return ""
|
|
|
|
def _getsourcelines(self, obj):
|
|
# GH-103319
|
|
# inspect.getsourcelines() returns lineno = 0 for
|
|
# module-level frame which breaks our code print line number
|
|
# This method should be replaced by inspect.getsourcelines(obj)
|
|
# once this bug is fixed in inspect
|
|
lines, lineno = inspect.getsourcelines(obj)
|
|
lineno = max(1, lineno)
|
|
return lines, lineno
|
|
|
|
def _help_message_from_doc(self, doc, usage_only=False):
|
|
lines = [line.strip() for line in doc.rstrip().splitlines()]
|
|
if not lines:
|
|
return "No help message found."
|
|
if "" in lines:
|
|
usage_end = lines.index("")
|
|
else:
|
|
usage_end = 1
|
|
formatted = []
|
|
indent = " " * len(self.prompt)
|
|
for i, line in enumerate(lines):
|
|
if i == 0:
|
|
prefix = "Usage: "
|
|
elif i < usage_end:
|
|
prefix = " "
|
|
else:
|
|
if usage_only:
|
|
break
|
|
prefix = ""
|
|
formatted.append(indent + prefix + line)
|
|
return "\n".join(formatted)
|
|
|
|
def _print_invalid_arg(self, arg):
|
|
"""Return the usage string for a function."""
|
|
|
|
if not arg:
|
|
self.error("Argument is required for this command")
|
|
else:
|
|
self.error(f"Invalid argument: {arg}")
|
|
|
|
# Yes it's a bit hacky. Get the caller name, get the method based on
|
|
# that name, and get the docstring from that method.
|
|
# This should NOT fail if the caller is a method of this class.
|
|
doc = inspect.getdoc(getattr(self, sys._getframe(1).f_code.co_name))
|
|
if doc is not None:
|
|
self.message(self._help_message_from_doc(doc, usage_only=True))
|
|
|
|
# Collect all command help into docstring, if not run with -OO
|
|
|
|
if __doc__ is not None:
|
|
# unfortunately we can't guess this order from the class definition
|
|
_help_order = [
|
|
'help', 'where', 'down', 'up', 'break', 'tbreak', 'clear', 'disable',
|
|
'enable', 'ignore', 'condition', 'commands', 'step', 'next', 'until',
|
|
'jump', 'return', 'retval', 'run', 'continue', 'list', 'longlist',
|
|
'args', 'p', 'pp', 'whatis', 'source', 'display', 'undisplay',
|
|
'interact', 'alias', 'unalias', 'debug', 'quit',
|
|
]
|
|
|
|
for _command in _help_order:
|
|
__doc__ += getattr(Pdb, 'do_' + _command).__doc__.strip() + '\n\n'
|
|
__doc__ += Pdb.help_exec.__doc__
|
|
|
|
del _help_order, _command
|
|
|
|
|
|
# Simplified interface
|
|
|
|
def run(statement, globals=None, locals=None):
|
|
"""Execute the *statement* (given as a string or a code object)
|
|
under debugger control.
|
|
|
|
The debugger prompt appears before any code is executed; you can set
|
|
breakpoints and type continue, or you can step through the statement
|
|
using step or next.
|
|
|
|
The optional *globals* and *locals* arguments specify the
|
|
environment in which the code is executed; by default the
|
|
dictionary of the module __main__ is used (see the explanation of
|
|
the built-in exec() or eval() functions.).
|
|
"""
|
|
Pdb().run(statement, globals, locals)
|
|
|
|
def runeval(expression, globals=None, locals=None):
|
|
"""Evaluate the *expression* (given as a string or a code object)
|
|
under debugger control.
|
|
|
|
When runeval() returns, it returns the value of the expression.
|
|
Otherwise this function is similar to run().
|
|
"""
|
|
return Pdb().runeval(expression, globals, locals)
|
|
|
|
def runctx(statement, globals, locals):
|
|
# B/W compatibility
|
|
run(statement, globals, locals)
|
|
|
|
def runcall(*args, **kwds):
|
|
"""Call the function (a function or method object, not a string)
|
|
with the given arguments.
|
|
|
|
When runcall() returns, it returns whatever the function call
|
|
returned. The debugger prompt appears as soon as the function is
|
|
entered.
|
|
"""
|
|
return Pdb().runcall(*args, **kwds)
|
|
|
|
def set_trace(*, header=None, commands=None):
|
|
"""Enter the debugger at the calling stack frame.
|
|
|
|
This is useful to hard-code a breakpoint at a given point in a
|
|
program, even if the code is not otherwise being debugged (e.g. when
|
|
an assertion fails). If given, *header* is printed to the console
|
|
just before debugging begins. *commands* is an optional list of
|
|
pdb commands to run when the debugger starts.
|
|
"""
|
|
if Pdb._last_pdb_instance is not None:
|
|
pdb = Pdb._last_pdb_instance
|
|
else:
|
|
pdb = Pdb(mode='inline', backend='monitoring', colorize=True)
|
|
if header is not None:
|
|
pdb.message(header)
|
|
pdb.set_trace(sys._getframe().f_back, commands=commands)
|
|
|
|
async def set_trace_async(*, header=None, commands=None):
|
|
"""Enter the debugger at the calling stack frame, but in async mode.
|
|
|
|
This should be used as await pdb.set_trace_async(). Users can do await
|
|
if they enter the debugger with this function. Otherwise it's the same
|
|
as set_trace().
|
|
"""
|
|
if Pdb._last_pdb_instance is not None:
|
|
pdb = Pdb._last_pdb_instance
|
|
else:
|
|
pdb = Pdb(mode='inline', backend='monitoring', colorize=True)
|
|
if header is not None:
|
|
pdb.message(header)
|
|
await pdb.set_trace_async(sys._getframe().f_back, commands=commands)
|
|
|
|
# Remote PDB
|
|
|
|
class _PdbServer(Pdb):
|
|
def __init__(
|
|
self,
|
|
sockfile,
|
|
signal_server=None,
|
|
owns_sockfile=True,
|
|
colorize=False,
|
|
**kwargs,
|
|
):
|
|
self._owns_sockfile = owns_sockfile
|
|
self._interact_state = None
|
|
self._sockfile = sockfile
|
|
self._command_name_cache = []
|
|
self._write_failed = False
|
|
if signal_server:
|
|
# Only started by the top level _PdbServer, not recursive ones.
|
|
self._start_signal_listener(signal_server)
|
|
# Override the `colorize` attribute set by the parent constructor,
|
|
# because it checks the server's stdout, rather than the client's.
|
|
super().__init__(colorize=False, **kwargs)
|
|
self.colorize = colorize
|
|
|
|
@staticmethod
|
|
def protocol_version():
|
|
# By default, assume a client and server are compatible if they run
|
|
# the same Python major.minor version. We'll try to keep backwards
|
|
# compatibility between patch versions of a minor version if possible.
|
|
# If we do need to change the protocol in a patch version, we'll change
|
|
# `revision` to the patch version where the protocol changed.
|
|
# We can ignore compatibility for pre-release versions; sys.remote_exec
|
|
# can't attach to a pre-release version except from that same version.
|
|
v = sys.version_info
|
|
revision = 0
|
|
return int(f"{v.major:02X}{v.minor:02X}{revision:02X}F0", 16)
|
|
|
|
def _ensure_valid_message(self, msg):
|
|
# Ensure the message conforms to our protocol.
|
|
# If anything needs to be changed here for a patch release of Python,
|
|
# the 'revision' in protocol_version() should be updated.
|
|
match msg:
|
|
case {"message": str(), "type": str()}:
|
|
# Have the client show a message. The client chooses how to
|
|
# format the message based on its type. The currently defined
|
|
# types are "info" and "error". If a message has a type the
|
|
# client doesn't recognize, it must be treated as "info".
|
|
pass
|
|
case {"help": str()}:
|
|
# Have the client show the help for a given argument.
|
|
pass
|
|
case {"prompt": str(), "state": str()}:
|
|
# Have the client display the given prompt and wait for a reply
|
|
# from the user. If the client recognizes the state it may
|
|
# enable mode-specific features like multi-line editing.
|
|
# If it doesn't recognize the state it must prompt for a single
|
|
# line only and send it directly to the server. A server won't
|
|
# progress until it gets a "reply" or "signal" message, but can
|
|
# process "complete" requests while waiting for the reply.
|
|
pass
|
|
case {
|
|
"completions": list(completions)
|
|
} if all(isinstance(c, str) for c in completions):
|
|
# Return valid completions for a client's "complete" request.
|
|
pass
|
|
case {
|
|
"command_list": list(command_list)
|
|
} if all(isinstance(c, str) for c in command_list):
|
|
# Report the list of legal PDB commands to the client.
|
|
# Due to aliases this list is not static, but the client
|
|
# needs to know it for multi-line editing.
|
|
pass
|
|
case _:
|
|
raise AssertionError(
|
|
f"PDB message doesn't follow the schema! {msg}"
|
|
)
|
|
|
|
@classmethod
|
|
def _start_signal_listener(cls, address):
|
|
def listener(sock):
|
|
with closing(sock):
|
|
# Check if the interpreter is finalizing every quarter of a second.
|
|
# Clean up and exit if so.
|
|
sock.settimeout(0.25)
|
|
sock.shutdown(socket.SHUT_WR)
|
|
while not shut_down.is_set():
|
|
try:
|
|
data = sock.recv(1024)
|
|
except socket.timeout:
|
|
continue
|
|
if data == b"":
|
|
return # EOF
|
|
signal.raise_signal(signal.SIGINT)
|
|
|
|
def stop_thread():
|
|
shut_down.set()
|
|
thread.join()
|
|
|
|
# Use a daemon thread so that we don't detach until after all non-daemon
|
|
# threads are done. Use an atexit handler to stop gracefully at that point,
|
|
# so that our thread is stopped before the interpreter is torn down.
|
|
shut_down = threading.Event()
|
|
thread = threading.Thread(
|
|
target=listener,
|
|
args=[socket.create_connection(address, timeout=5)],
|
|
daemon=True,
|
|
)
|
|
atexit.register(stop_thread)
|
|
thread.start()
|
|
|
|
def _send(self, **kwargs):
|
|
self._ensure_valid_message(kwargs)
|
|
json_payload = json.dumps(kwargs)
|
|
try:
|
|
self._sockfile.write(json_payload.encode() + b"\n")
|
|
self._sockfile.flush()
|
|
except (OSError, ValueError):
|
|
# We get an OSError if the network connection has dropped, and a
|
|
# ValueError if detach() if the sockfile has been closed. We'll
|
|
# handle this the next time we try to read from the client instead
|
|
# of trying to handle it from everywhere _send() may be called.
|
|
# Track this with a flag rather than assuming readline() will ever
|
|
# return an empty string because the socket may be half-closed.
|
|
self._write_failed = True
|
|
|
|
@typing.override
|
|
def message(self, msg, end="\n"):
|
|
self._send(message=str(msg) + end, type="info")
|
|
|
|
@typing.override
|
|
def error(self, msg):
|
|
self._send(message=str(msg), type="error")
|
|
|
|
def _get_input(self, prompt, state) -> str:
|
|
# Before displaying a (Pdb) prompt, send the list of PDB commands
|
|
# unless we've already sent an up-to-date list.
|
|
if state == "pdb" and not self._command_name_cache:
|
|
self._command_name_cache = self.completenames("", "", 0, 0)
|
|
self._send(command_list=self._command_name_cache)
|
|
self._send(prompt=prompt, state=state)
|
|
return self._read_reply()
|
|
|
|
def _read_reply(self):
|
|
# Loop until we get a 'reply' or 'signal' from the client,
|
|
# processing out-of-band 'complete' requests as they arrive.
|
|
while True:
|
|
if self._write_failed:
|
|
raise EOFError
|
|
|
|
msg = self._sockfile.readline()
|
|
if not msg:
|
|
raise EOFError
|
|
|
|
try:
|
|
payload = json.loads(msg)
|
|
except json.JSONDecodeError:
|
|
self.error(f"Disconnecting: client sent invalid JSON {msg!r}")
|
|
raise EOFError
|
|
|
|
match payload:
|
|
case {"reply": str(reply)}:
|
|
return reply
|
|
case {"signal": str(signal)}:
|
|
if signal == "INT":
|
|
raise KeyboardInterrupt
|
|
elif signal == "EOF":
|
|
raise EOFError
|
|
else:
|
|
self.error(
|
|
f"Received unrecognized signal: {signal}"
|
|
)
|
|
# Our best hope of recovering is to pretend we
|
|
# got an EOF to exit whatever mode we're in.
|
|
raise EOFError
|
|
case {
|
|
"complete": {
|
|
"text": str(text),
|
|
"line": str(line),
|
|
"begidx": int(begidx),
|
|
"endidx": int(endidx),
|
|
}
|
|
}:
|
|
items = self._complete_any(text, line, begidx, endidx)
|
|
self._send(completions=items)
|
|
continue
|
|
# Valid JSON, but doesn't meet the schema.
|
|
self.error(f"Ignoring invalid message from client: {msg}")
|
|
|
|
def _complete_any(self, text, line, begidx, endidx):
|
|
# If we're in 'interact' mode, we need to use the default completer
|
|
if self._interact_state:
|
|
compfunc = self.completedefault
|
|
else:
|
|
if begidx == 0:
|
|
return self.completenames(text, line, begidx, endidx)
|
|
|
|
cmd = self.parseline(line)[0]
|
|
if cmd:
|
|
compfunc = getattr(self, "complete_" + cmd, self.completedefault)
|
|
else:
|
|
compfunc = self.completedefault
|
|
return compfunc(text, line, begidx, endidx)
|
|
|
|
def cmdloop(self, intro=None):
|
|
self.preloop()
|
|
if intro is not None:
|
|
self.intro = intro
|
|
if self.intro:
|
|
self.message(str(self.intro))
|
|
stop = None
|
|
while not stop:
|
|
if self._interact_state is not None:
|
|
try:
|
|
reply = self._get_input(prompt=">>> ", state="interact")
|
|
except KeyboardInterrupt:
|
|
# Match how KeyboardInterrupt is handled in a REPL
|
|
self.message("\nKeyboardInterrupt")
|
|
except EOFError:
|
|
self.message("\n*exit from pdb interact command*")
|
|
self._interact_state = None
|
|
else:
|
|
self._run_in_python_repl(reply)
|
|
continue
|
|
|
|
if not self.cmdqueue:
|
|
try:
|
|
state = "commands" if self.commands_defining else "pdb"
|
|
reply = self._get_input(prompt=self.prompt, state=state)
|
|
except EOFError:
|
|
reply = "EOF"
|
|
|
|
self.cmdqueue.append(reply)
|
|
|
|
line = self.cmdqueue.pop(0)
|
|
line = self.precmd(line)
|
|
stop = self.onecmd(line)
|
|
stop = self.postcmd(stop, line)
|
|
self.postloop()
|
|
|
|
def postloop(self):
|
|
super().postloop()
|
|
if self.quitting:
|
|
self.detach()
|
|
|
|
def detach(self):
|
|
# Detach the debugger and close the socket without raising BdbQuit
|
|
self.quitting = False
|
|
if self._owns_sockfile:
|
|
# Don't try to reuse this instance, it's not valid anymore.
|
|
Pdb._last_pdb_instance = None
|
|
try:
|
|
self._sockfile.close()
|
|
except OSError:
|
|
# close() can fail if the connection was broken unexpectedly.
|
|
pass
|
|
|
|
def do_debug(self, arg):
|
|
# Clear our cached list of valid commands; the recursive debugger might
|
|
# send its own differing list, and so ours needs to be re-sent.
|
|
self._command_name_cache = []
|
|
return super().do_debug(arg)
|
|
|
|
def do_alias(self, arg):
|
|
# Clear our cached list of valid commands; one might be added.
|
|
self._command_name_cache = []
|
|
return super().do_alias(arg)
|
|
|
|
def do_unalias(self, arg):
|
|
# Clear our cached list of valid commands; one might be removed.
|
|
self._command_name_cache = []
|
|
return super().do_unalias(arg)
|
|
|
|
def do_help(self, arg):
|
|
# Tell the client to render the help, since it might need a pager.
|
|
self._send(help=arg)
|
|
|
|
do_h = do_help
|
|
|
|
def _interact_displayhook(self, obj):
|
|
# Like the default `sys.displayhook` except sending a socket message.
|
|
if obj is not None:
|
|
self.message(repr(obj))
|
|
builtins._ = obj
|
|
|
|
def _run_in_python_repl(self, lines):
|
|
# Run one 'interact' mode code block against an existing namespace.
|
|
assert self._interact_state
|
|
save_displayhook = sys.displayhook
|
|
try:
|
|
sys.displayhook = self._interact_displayhook
|
|
code_obj = self._interact_state["compiler"](lines + "\n")
|
|
if code_obj is None:
|
|
raise SyntaxError("Incomplete command")
|
|
exec(code_obj, self._interact_state["ns"])
|
|
except:
|
|
self._error_exc()
|
|
finally:
|
|
sys.displayhook = save_displayhook
|
|
|
|
def do_interact(self, arg):
|
|
# Prepare to run 'interact' mode code blocks, and trigger the client
|
|
# to start treating all input as Python commands, not PDB ones.
|
|
self.message("*pdb interact start*")
|
|
self._interact_state = dict(
|
|
compiler=codeop.CommandCompiler(),
|
|
ns={**self.curframe.f_globals, **self.curframe.f_locals},
|
|
)
|
|
|
|
@typing.override
|
|
def _create_recursive_debugger(self):
|
|
return _PdbServer(
|
|
self._sockfile,
|
|
owns_sockfile=False,
|
|
colorize=self.colorize,
|
|
)
|
|
|
|
@typing.override
|
|
def _prompt_for_confirmation(self, prompt, default):
|
|
try:
|
|
return self._get_input(prompt=prompt, state="confirm")
|
|
except (EOFError, KeyboardInterrupt):
|
|
return default
|
|
|
|
def do_run(self, arg):
|
|
self.error("remote PDB cannot restart the program")
|
|
|
|
do_restart = do_run
|
|
|
|
def _error_exc(self):
|
|
if self._interact_state and isinstance(sys.exception(), SystemExit):
|
|
# If we get a SystemExit in 'interact' mode, exit the REPL.
|
|
self._interact_state = None
|
|
ret = super()._error_exc()
|
|
self.message("*exit from pdb interact command*")
|
|
return ret
|
|
else:
|
|
return super()._error_exc()
|
|
|
|
def default(self, line):
|
|
# Unlike Pdb, don't prompt for more lines of a multi-line command.
|
|
# The remote needs to send us the whole block in one go.
|
|
try:
|
|
candidate = line.removeprefix("!") + "\n"
|
|
if codeop.compile_command(candidate, "<stdin>", "single") is None:
|
|
raise SyntaxError("Incomplete command")
|
|
return super().default(candidate)
|
|
except:
|
|
self._error_exc()
|
|
|
|
|
|
class _PdbClient:
|
|
def __init__(self, pid, server_socket, interrupt_sock):
|
|
self.pid = pid
|
|
self.read_buf = b""
|
|
self.signal_read = None
|
|
self.signal_write = None
|
|
self.sigint_received = False
|
|
self.raise_on_sigint = False
|
|
self.server_socket = server_socket
|
|
self.interrupt_sock = interrupt_sock
|
|
self.pdb_instance = Pdb()
|
|
self.pdb_commands = set()
|
|
self.completion_matches = []
|
|
self.state = "dumb"
|
|
self.write_failed = False
|
|
self.multiline_block = False
|
|
|
|
def _ensure_valid_message(self, msg):
|
|
# Ensure the message conforms to our protocol.
|
|
# If anything needs to be changed here for a patch release of Python,
|
|
# the 'revision' in protocol_version() should be updated.
|
|
match msg:
|
|
case {"reply": str()}:
|
|
# Send input typed by a user at a prompt to the remote PDB.
|
|
pass
|
|
case {"signal": "EOF"}:
|
|
# Tell the remote PDB that the user pressed ^D at a prompt.
|
|
pass
|
|
case {"signal": "INT"}:
|
|
# Tell the remote PDB that the user pressed ^C at a prompt.
|
|
pass
|
|
case {
|
|
"complete": {
|
|
"text": str(),
|
|
"line": str(),
|
|
"begidx": int(),
|
|
"endidx": int(),
|
|
}
|
|
}:
|
|
# Ask the remote PDB what completions are valid for the given
|
|
# parameters, using readline's completion protocol.
|
|
pass
|
|
case _:
|
|
raise AssertionError(
|
|
f"PDB message doesn't follow the schema! {msg}"
|
|
)
|
|
|
|
def _send(self, **kwargs):
|
|
self._ensure_valid_message(kwargs)
|
|
json_payload = json.dumps(kwargs)
|
|
try:
|
|
self.server_socket.sendall(json_payload.encode() + b"\n")
|
|
except OSError:
|
|
# This means that the client has abruptly disconnected, but we'll
|
|
# handle that the next time we try to read from the client instead
|
|
# of trying to handle it from everywhere _send() may be called.
|
|
# Track this with a flag rather than assuming readline() will ever
|
|
# return an empty string because the socket may be half-closed.
|
|
self.write_failed = True
|
|
|
|
def _readline(self):
|
|
if self.sigint_received:
|
|
# There's a pending unhandled SIGINT. Handle it now.
|
|
self.sigint_received = False
|
|
raise KeyboardInterrupt
|
|
|
|
# Wait for either a SIGINT or a line or EOF from the PDB server.
|
|
selector = selectors.DefaultSelector()
|
|
selector.register(self.signal_read, selectors.EVENT_READ)
|
|
selector.register(self.server_socket, selectors.EVENT_READ)
|
|
|
|
while b"\n" not in self.read_buf:
|
|
for key, _ in selector.select():
|
|
if key.fileobj == self.signal_read:
|
|
self.signal_read.recv(1024)
|
|
if self.sigint_received:
|
|
# If not, we're reading wakeup events for sigints that
|
|
# we've previously handled, and can ignore them.
|
|
self.sigint_received = False
|
|
raise KeyboardInterrupt
|
|
elif key.fileobj == self.server_socket:
|
|
data = self.server_socket.recv(16 * 1024)
|
|
self.read_buf += data
|
|
if not data and b"\n" not in self.read_buf:
|
|
# EOF without a full final line. Drop the partial line.
|
|
self.read_buf = b""
|
|
return b""
|
|
|
|
ret, sep, self.read_buf = self.read_buf.partition(b"\n")
|
|
return ret + sep
|
|
|
|
def read_input(self, prompt, multiline_block):
|
|
self.multiline_block = multiline_block
|
|
with self._sigint_raises_keyboard_interrupt():
|
|
return input(prompt)
|
|
|
|
def read_command(self, prompt):
|
|
reply = self.read_input(prompt, multiline_block=False)
|
|
if self.state == "dumb":
|
|
# No logic applied whatsoever, just pass the raw reply back.
|
|
return reply
|
|
|
|
prefix = ""
|
|
if self.state == "pdb":
|
|
# PDB command entry mode
|
|
cmd = self.pdb_instance.parseline(reply)[0]
|
|
if cmd in self.pdb_commands or reply.strip() == "":
|
|
# Recognized PDB command, or blank line repeating last command
|
|
return reply
|
|
|
|
# Otherwise, explicit or implicit exec command
|
|
if reply.startswith("!"):
|
|
prefix = "!"
|
|
reply = reply.removeprefix(prefix).lstrip()
|
|
|
|
if codeop.compile_command(reply + "\n", "<stdin>", "single") is not None:
|
|
# Valid single-line statement
|
|
return prefix + reply
|
|
|
|
# Otherwise, valid first line of a multi-line statement
|
|
more_prompt = "...".ljust(len(prompt))
|
|
while codeop.compile_command(reply, "<stdin>", "single") is None:
|
|
reply += "\n" + self.read_input(more_prompt, multiline_block=True)
|
|
|
|
return prefix + reply
|
|
|
|
@contextmanager
|
|
def readline_completion(self, completer):
|
|
try:
|
|
import readline
|
|
except ImportError:
|
|
yield
|
|
return
|
|
|
|
old_completer = readline.get_completer()
|
|
try:
|
|
readline.set_completer(completer)
|
|
if readline.backend == "editline":
|
|
# libedit uses "^I" instead of "tab"
|
|
command_string = "bind ^I rl_complete"
|
|
else:
|
|
command_string = "tab: complete"
|
|
readline.parse_and_bind(command_string)
|
|
yield
|
|
finally:
|
|
readline.set_completer(old_completer)
|
|
|
|
@contextmanager
|
|
def _sigint_handler(self):
|
|
# Signal handling strategy:
|
|
# - When we call input() we want a SIGINT to raise KeyboardInterrupt
|
|
# - Otherwise we want to write to the wakeup FD and set a flag.
|
|
# We'll break out of select() when the wakeup FD is written to,
|
|
# and we'll check the flag whenever we're about to accept input.
|
|
def handler(signum, frame):
|
|
self.sigint_received = True
|
|
if self.raise_on_sigint:
|
|
# One-shot; don't raise again until the flag is set again.
|
|
self.raise_on_sigint = False
|
|
self.sigint_received = False
|
|
raise KeyboardInterrupt
|
|
|
|
sentinel = object()
|
|
old_handler = sentinel
|
|
old_wakeup_fd = sentinel
|
|
|
|
self.signal_read, self.signal_write = socket.socketpair()
|
|
with (closing(self.signal_read), closing(self.signal_write)):
|
|
self.signal_read.setblocking(False)
|
|
self.signal_write.setblocking(False)
|
|
|
|
try:
|
|
old_handler = signal.signal(signal.SIGINT, handler)
|
|
|
|
try:
|
|
old_wakeup_fd = signal.set_wakeup_fd(
|
|
self.signal_write.fileno(),
|
|
warn_on_full_buffer=False,
|
|
)
|
|
yield
|
|
finally:
|
|
# Restore the old wakeup fd if we installed a new one
|
|
if old_wakeup_fd is not sentinel:
|
|
signal.set_wakeup_fd(old_wakeup_fd)
|
|
finally:
|
|
self.signal_read = self.signal_write = None
|
|
if old_handler is not sentinel:
|
|
# Restore the old handler if we installed a new one
|
|
signal.signal(signal.SIGINT, old_handler)
|
|
|
|
@contextmanager
|
|
def _sigint_raises_keyboard_interrupt(self):
|
|
if self.sigint_received:
|
|
# There's a pending unhandled SIGINT. Handle it now.
|
|
self.sigint_received = False
|
|
raise KeyboardInterrupt
|
|
|
|
try:
|
|
self.raise_on_sigint = True
|
|
yield
|
|
finally:
|
|
self.raise_on_sigint = False
|
|
|
|
def cmdloop(self):
|
|
with (
|
|
self._sigint_handler(),
|
|
self.readline_completion(self.complete),
|
|
):
|
|
while not self.write_failed:
|
|
try:
|
|
if not (payload_bytes := self._readline()):
|
|
break
|
|
except KeyboardInterrupt:
|
|
self.send_interrupt()
|
|
continue
|
|
|
|
try:
|
|
payload = json.loads(payload_bytes)
|
|
except json.JSONDecodeError:
|
|
print(
|
|
f"*** Invalid JSON from remote: {payload_bytes!r}",
|
|
flush=True,
|
|
)
|
|
continue
|
|
|
|
self.process_payload(payload)
|
|
|
|
def send_interrupt(self):
|
|
if self.interrupt_sock is not None:
|
|
# Write to a socket that the PDB server listens on. This triggers
|
|
# the remote to raise a SIGINT for itself. We do this because
|
|
# Windows doesn't allow triggering SIGINT remotely.
|
|
# See https://stackoverflow.com/a/35792192 for many more details.
|
|
self.interrupt_sock.sendall(signal.SIGINT.to_bytes())
|
|
else:
|
|
# On Unix we can just send a SIGINT to the remote process.
|
|
# This is preferable to using the signal thread approach that we
|
|
# use on Windows because it can interrupt IO in the main thread.
|
|
os.kill(self.pid, signal.SIGINT)
|
|
|
|
def process_payload(self, payload):
|
|
match payload:
|
|
case {
|
|
"command_list": command_list
|
|
} if all(isinstance(c, str) for c in command_list):
|
|
self.pdb_commands = set(command_list)
|
|
case {"message": str(msg), "type": str(msg_type)}:
|
|
if msg_type == "error":
|
|
print("***", msg, flush=True)
|
|
else:
|
|
print(msg, end="", flush=True)
|
|
case {"help": str(arg)}:
|
|
self.pdb_instance.do_help(arg)
|
|
case {"prompt": str(prompt), "state": str(state)}:
|
|
if state not in ("pdb", "interact"):
|
|
state = "dumb"
|
|
self.state = state
|
|
self.prompt_for_reply(prompt)
|
|
case _:
|
|
raise RuntimeError(f"Unrecognized payload {payload}")
|
|
|
|
def prompt_for_reply(self, prompt):
|
|
while True:
|
|
try:
|
|
payload = {"reply": self.read_command(prompt)}
|
|
except EOFError:
|
|
payload = {"signal": "EOF"}
|
|
except KeyboardInterrupt:
|
|
payload = {"signal": "INT"}
|
|
except Exception as exc:
|
|
msg = traceback.format_exception_only(exc)[-1].strip()
|
|
print("***", msg, flush=True)
|
|
continue
|
|
|
|
self._send(**payload)
|
|
return
|
|
|
|
def complete(self, text, state):
|
|
import readline
|
|
|
|
if state == 0:
|
|
self.completion_matches = []
|
|
if self.state not in ("pdb", "interact"):
|
|
return None
|
|
|
|
origline = readline.get_line_buffer()
|
|
line = origline.lstrip()
|
|
if self.multiline_block:
|
|
# We're completing a line contained in a multi-line block.
|
|
# Force the remote to treat it as a Python expression.
|
|
line = "! " + line
|
|
offset = len(origline) - len(line)
|
|
begidx = readline.get_begidx() - offset
|
|
endidx = readline.get_endidx() - offset
|
|
|
|
msg = {
|
|
"complete": {
|
|
"text": text,
|
|
"line": line,
|
|
"begidx": begidx,
|
|
"endidx": endidx,
|
|
}
|
|
}
|
|
|
|
self._send(**msg)
|
|
if self.write_failed:
|
|
return None
|
|
|
|
payload = self._readline()
|
|
if not payload:
|
|
return None
|
|
|
|
payload = json.loads(payload)
|
|
if "completions" not in payload:
|
|
raise RuntimeError(
|
|
f"Failed to get valid completions. Got: {payload}"
|
|
)
|
|
|
|
self.completion_matches = payload["completions"]
|
|
try:
|
|
return self.completion_matches[state]
|
|
except IndexError:
|
|
return None
|
|
|
|
|
|
def _connect(
|
|
*,
|
|
host,
|
|
port,
|
|
frame,
|
|
commands,
|
|
version,
|
|
signal_raising_thread,
|
|
colorize,
|
|
):
|
|
with closing(socket.create_connection((host, port))) as conn:
|
|
sockfile = conn.makefile("rwb")
|
|
|
|
# The client requests this thread on Windows but not on Unix.
|
|
# Most tests don't request this thread, to keep them simpler.
|
|
if signal_raising_thread:
|
|
signal_server = (host, port)
|
|
else:
|
|
signal_server = None
|
|
|
|
remote_pdb = _PdbServer(
|
|
sockfile,
|
|
signal_server=signal_server,
|
|
colorize=colorize,
|
|
)
|
|
weakref.finalize(remote_pdb, sockfile.close)
|
|
|
|
if Pdb._last_pdb_instance is not None:
|
|
remote_pdb.error("Another PDB instance is already attached.")
|
|
elif version != remote_pdb.protocol_version():
|
|
target_ver = f"0x{remote_pdb.protocol_version():08X}"
|
|
attach_ver = f"0x{version:08X}"
|
|
remote_pdb.error(
|
|
f"The target process is running a Python version that is"
|
|
f" incompatible with this PDB module."
|
|
f"\nTarget process pdb protocol version: {target_ver}"
|
|
f"\nLocal pdb module's protocol version: {attach_ver}"
|
|
)
|
|
else:
|
|
remote_pdb.set_trace(frame=frame, commands=commands.splitlines())
|
|
|
|
|
|
def attach(pid, commands=()):
|
|
"""Attach to a running process with the given PID."""
|
|
with ExitStack() as stack:
|
|
server = stack.enter_context(
|
|
closing(socket.create_server(("localhost", 0)))
|
|
)
|
|
port = server.getsockname()[1]
|
|
|
|
connect_script = stack.enter_context(
|
|
tempfile.NamedTemporaryFile("w", delete_on_close=False)
|
|
)
|
|
|
|
use_signal_thread = sys.platform == "win32"
|
|
colorize = _colorize.can_colorize()
|
|
|
|
connect_script.write(
|
|
textwrap.dedent(
|
|
f"""
|
|
import pdb, sys
|
|
pdb._connect(
|
|
host="localhost",
|
|
port={port},
|
|
frame=sys._getframe(1),
|
|
commands={json.dumps("\n".join(commands))},
|
|
version={_PdbServer.protocol_version()},
|
|
signal_raising_thread={use_signal_thread!r},
|
|
colorize={colorize!r},
|
|
)
|
|
"""
|
|
)
|
|
)
|
|
connect_script.close()
|
|
sys.remote_exec(pid, connect_script.name)
|
|
|
|
# TODO Add a timeout? Or don't bother since the user can ^C?
|
|
client_sock, _ = server.accept()
|
|
stack.enter_context(closing(client_sock))
|
|
|
|
if use_signal_thread:
|
|
interrupt_sock, _ = server.accept()
|
|
stack.enter_context(closing(interrupt_sock))
|
|
interrupt_sock.setblocking(False)
|
|
else:
|
|
interrupt_sock = None
|
|
|
|
_PdbClient(pid, client_sock, interrupt_sock).cmdloop()
|
|
|
|
|
|
# Post-Mortem interface
|
|
|
|
def post_mortem(t=None):
|
|
"""Enter post-mortem debugging of the given *traceback*, or *exception*
|
|
object.
|
|
|
|
If no traceback is given, it uses the one of the exception that is
|
|
currently being handled (an exception must be being handled if the
|
|
default is to be used).
|
|
|
|
If `t` is an exception object, the `exceptions` command makes it possible to
|
|
list and inspect its chained exceptions (if any).
|
|
"""
|
|
return _post_mortem(t, Pdb())
|
|
|
|
|
|
def _post_mortem(t, pdb_instance):
|
|
"""
|
|
Private version of post_mortem, which allow to pass a pdb instance
|
|
for testing purposes.
|
|
"""
|
|
# handling the default
|
|
if t is None:
|
|
exc = sys.exception()
|
|
if exc is not None:
|
|
t = exc.__traceback__
|
|
|
|
if t is None or (isinstance(t, BaseException) and t.__traceback__ is None):
|
|
raise ValueError("A valid traceback must be passed if no "
|
|
"exception is being handled")
|
|
|
|
pdb_instance.reset()
|
|
pdb_instance.interaction(None, t)
|
|
|
|
|
|
def pm():
|
|
"""Enter post-mortem debugging of the traceback found in sys.last_exc."""
|
|
post_mortem(sys.last_exc)
|
|
|
|
|
|
# Main program for testing
|
|
|
|
TESTCMD = 'import x; x.main()'
|
|
|
|
def test():
|
|
run(TESTCMD)
|
|
|
|
# print help
|
|
def help():
|
|
import pydoc
|
|
pydoc.pager(__doc__)
|
|
|
|
_usage = """\
|
|
Debug the Python program given by pyfile. Alternatively,
|
|
an executable module or package to debug can be specified using
|
|
the -m switch.
|
|
|
|
Initial commands are read from .pdbrc files in your home directory
|
|
and in the current directory, if they exist. Commands supplied with
|
|
-c are executed after commands from .pdbrc files.
|
|
|
|
To let the script run until an exception occurs, use "-c continue".
|
|
To let the script run up to a given line X in the debugged file, use
|
|
"-c 'until X'"."""
|
|
|
|
|
|
def main():
|
|
import argparse
|
|
|
|
parser = argparse.ArgumentParser(
|
|
usage="%(prog)s [-h] [-c command] (-m module | -p pid | pyfile) [args ...]",
|
|
description=_usage,
|
|
formatter_class=argparse.RawDescriptionHelpFormatter,
|
|
allow_abbrev=False,
|
|
color=True,
|
|
)
|
|
|
|
# We need to maunally get the script from args, because the first positional
|
|
# arguments could be either the script we need to debug, or the argument
|
|
# to the -m module
|
|
parser.add_argument('-c', '--command', action='append', default=[], metavar='command', dest='commands',
|
|
help='pdb commands to execute as if given in a .pdbrc file')
|
|
parser.add_argument('-m', metavar='module', dest='module')
|
|
parser.add_argument('-p', '--pid', type=int, help="attach to the specified PID", default=None)
|
|
|
|
if len(sys.argv) == 1:
|
|
# If no arguments were given (python -m pdb), print the whole help message.
|
|
# Without this check, argparse would only complain about missing required arguments.
|
|
parser.print_help()
|
|
sys.exit(2)
|
|
|
|
opts, args = parser.parse_known_args()
|
|
|
|
if opts.pid:
|
|
# If attaching to a remote pid, unrecognized arguments are not allowed.
|
|
# This will raise an error if there are extra unrecognized arguments.
|
|
opts = parser.parse_args()
|
|
if opts.module:
|
|
parser.error("argument -m: not allowed with argument --pid")
|
|
attach(opts.pid, opts.commands)
|
|
return
|
|
elif opts.module:
|
|
# If a module is being debugged, we consider the arguments after "-m module" to
|
|
# be potential arguments to the module itself. We need to parse the arguments
|
|
# before "-m" to check if there is any invalid argument.
|
|
# e.g. "python -m pdb -m foo --spam" means passing "--spam" to "foo"
|
|
# "python -m pdb --spam -m foo" means passing "--spam" to "pdb" and is invalid
|
|
idx = sys.argv.index('-m')
|
|
args_to_pdb = sys.argv[1:idx]
|
|
# This will raise an error if there are invalid arguments
|
|
parser.parse_args(args_to_pdb)
|
|
else:
|
|
# If a script is being debugged, then pdb expects the script name as the first argument.
|
|
# Anything before the script is considered an argument to pdb itself, which would
|
|
# be invalid because it's not parsed by argparse.
|
|
invalid_args = list(itertools.takewhile(lambda a: a.startswith('-'), args))
|
|
if invalid_args:
|
|
parser.error(f"unrecognized arguments: {' '.join(invalid_args)}")
|
|
sys.exit(2)
|
|
|
|
if opts.module:
|
|
file = opts.module
|
|
target = _ModuleTarget(file)
|
|
else:
|
|
if not args:
|
|
parser.error("no module or script to run")
|
|
file = args.pop(0)
|
|
if file.endswith('.pyz'):
|
|
target = _ZipTarget(file)
|
|
else:
|
|
target = _ScriptTarget(file)
|
|
|
|
sys.argv[:] = [file] + args # Hide "pdb.py" and pdb options from argument list
|
|
|
|
# Note on saving/restoring sys.argv: it's a good idea when sys.argv was
|
|
# modified by the script being debugged. It's a bad idea when it was
|
|
# changed by the user from the command line. There is a "restart" command
|
|
# which allows explicit specification of command line arguments.
|
|
pdb = Pdb(mode='cli', backend='monitoring', colorize=True)
|
|
pdb.rcLines.extend(opts.commands)
|
|
while True:
|
|
try:
|
|
pdb._run(target)
|
|
except Restart:
|
|
print("Restarting", target, "with arguments:")
|
|
print("\t" + " ".join(sys.argv[1:]))
|
|
except SystemExit as e:
|
|
# In most cases SystemExit does not warrant a post-mortem session.
|
|
print("The program exited via sys.exit(). Exit status:", end=' ')
|
|
print(e)
|
|
except BaseException as e:
|
|
traceback.print_exception(e, colorize=_colorize.can_colorize())
|
|
print("Uncaught exception. Entering post mortem debugging")
|
|
print("Running 'cont' or 'step' will restart the program")
|
|
try:
|
|
pdb.interaction(None, e)
|
|
except Restart:
|
|
print("Restarting", target, "with arguments:")
|
|
print("\t" + " ".join(sys.argv[1:]))
|
|
continue
|
|
if pdb._user_requested_quit:
|
|
break
|
|
print("The program finished and will be restarted")
|
|
|
|
|
|
# When invoked as main program, invoke the debugger on a script
|
|
if __name__ == '__main__':
|
|
import pdb
|
|
pdb.main()
|