347 lines
10 KiB
Python
347 lines
10 KiB
Python
"""
|
|
Multi-core debugging for da peeps!
|
|
"""
|
|
import bdb
|
|
import sys
|
|
from functools import partial
|
|
from contextlib import asynccontextmanager
|
|
from typing import Awaitable, Tuple, Optional, Callable, AsyncIterator
|
|
|
|
from async_generator import aclosing
|
|
import tractor
|
|
import trio
|
|
|
|
from .log import get_logger
|
|
from . import _state
|
|
from ._discovery import get_root
|
|
from ._state import is_root_process
|
|
from ._exceptions import is_multi_cancelled
|
|
|
|
try:
|
|
# wtf: only exported when installed in dev mode?
|
|
import pdbpp
|
|
except ImportError:
|
|
# pdbpp is installed in regular mode...it monkey patches stuff
|
|
import pdb
|
|
assert pdb.xpm, "pdbpp is not installed?" # type: ignore
|
|
pdbpp = pdb
|
|
|
|
log = get_logger(__name__)
|
|
|
|
|
|
__all__ = ['breakpoint', 'post_mortem']
|
|
|
|
# placeholder for function to set a ``trio.Event`` on debugger exit
|
|
_pdb_release_hook: Optional[Callable] = None
|
|
|
|
# actor-wide variable pointing to current task name using debugger
|
|
_in_debug = False
|
|
|
|
# lock in root actor preventing multi-access to local tty
|
|
_debug_lock = trio.StrictFIFOLock()
|
|
|
|
# XXX: set by the current task waiting on the root tty lock
|
|
# and must be cancelled if this actor is cancelled via message
|
|
# otherwise deadlocks with the parent actor may ensure
|
|
_debugger_request_cs: Optional[trio.CancelScope] = None
|
|
|
|
|
|
class TractorConfig(pdbpp.DefaultConfig):
|
|
"""Custom ``pdbpp`` goodness.
|
|
"""
|
|
# sticky_by_default = True
|
|
|
|
|
|
class PdbwTeardown(pdbpp.Pdb):
|
|
"""Add teardown hooks to the regular ``pdbpp.Pdb``.
|
|
"""
|
|
# override the pdbpp config with our coolio one
|
|
DefaultConfig = TractorConfig
|
|
|
|
# TODO: figure out how to dissallow recursive .set_trace() entry
|
|
# since that'll cause deadlock for us.
|
|
def set_continue(self):
|
|
global _in_debug
|
|
try:
|
|
super().set_continue()
|
|
finally:
|
|
_in_debug = False
|
|
_pdb_release_hook()
|
|
|
|
def set_quit(self):
|
|
global _in_debug
|
|
try:
|
|
super().set_quit()
|
|
finally:
|
|
_in_debug = False
|
|
_pdb_release_hook()
|
|
|
|
|
|
# TODO: will be needed whenever we get to true remote debugging.
|
|
# XXX see https://github.com/goodboy/tractor/issues/130
|
|
|
|
# # TODO: is there some way to determine this programatically?
|
|
# _pdb_exit_patterns = tuple(
|
|
# str.encode(patt + "\n") for patt in (
|
|
# 'c', 'cont', 'continue', 'q', 'quit')
|
|
# )
|
|
|
|
# def subactoruid2proc(
|
|
# actor: 'Actor', # noqa
|
|
# uid: Tuple[str, str]
|
|
# ) -> trio.Process:
|
|
# n = actor._actoruid2nursery[uid]
|
|
# _, proc, _ = n._children[uid]
|
|
# return proc
|
|
|
|
# async def hijack_stdin():
|
|
# log.info(f"Hijacking stdin from {actor.uid}")
|
|
|
|
# trap std in and relay to subproc
|
|
# async_stdin = trio.wrap_file(sys.stdin)
|
|
|
|
# async with aclosing(async_stdin):
|
|
# async for msg in async_stdin:
|
|
# log.trace(f"Stdin input:\n{msg}")
|
|
# # encode to bytes
|
|
# bmsg = str.encode(msg)
|
|
|
|
# # relay bytes to subproc over pipe
|
|
# # await proc.stdin.send_all(bmsg)
|
|
|
|
# if bmsg in _pdb_exit_patterns:
|
|
# log.info("Closing stdin hijack")
|
|
# break
|
|
|
|
|
|
@asynccontextmanager
|
|
async def _acquire_debug_lock(uid: Tuple[str, str]) -> AsyncIterator[None]:
|
|
"""Acquire a actor local FIFO lock meant to mutex entry to a local
|
|
debugger entry point to avoid tty clobbering by multiple processes.
|
|
"""
|
|
task_name = trio.lowlevel.current_task().name
|
|
try:
|
|
log.debug(
|
|
f"Attempting to acquire TTY lock, remote task: {task_name}:{uid}")
|
|
await _debug_lock.acquire()
|
|
|
|
log.debug(f"TTY lock acquired, remote task: {task_name}:{uid}")
|
|
yield
|
|
|
|
finally:
|
|
_debug_lock.release()
|
|
log.debug(f"TTY lock released, remote task: {task_name}:{uid}")
|
|
|
|
|
|
# @contextmanager
|
|
# def _disable_sigint():
|
|
# try:
|
|
# # disable sigint handling while in debug
|
|
# prior_handler = signal.signal(signal.SIGINT, handler)
|
|
# yield
|
|
# finally:
|
|
# # restore SIGINT handling
|
|
# signal.signal(signal.SIGINT, prior_handler)
|
|
|
|
|
|
async def _hijack_stdin_relay_to_child(
|
|
subactor_uid: Tuple[str, str]
|
|
) -> AsyncIterator[str]:
|
|
# TODO: when we get to true remote debugging
|
|
# this will deliver stdin data
|
|
log.warning(f"Actor {subactor_uid} is WAITING on stdin hijack lock")
|
|
async with _acquire_debug_lock(subactor_uid):
|
|
log.warning(f"Actor {subactor_uid} ACQUIRED stdin hijack lock")
|
|
|
|
# with _disable_sigint():
|
|
|
|
# indicate to child that we've locked stdio
|
|
yield 'Locked'
|
|
|
|
# wait for cancellation of stream by child
|
|
# indicating debugger is dis-engaged
|
|
await trio.sleep_forever()
|
|
|
|
log.debug(f"Actor {subactor_uid} RELEASED stdin hijack lock")
|
|
|
|
|
|
# XXX: We only make this sync in case someone wants to
|
|
# overload the ``breakpoint()`` built-in.
|
|
def _breakpoint(debug_func) -> Awaitable[None]:
|
|
"""``tractor`` breakpoint entry for engaging pdb machinery
|
|
in subactors.
|
|
"""
|
|
actor = tractor.current_actor()
|
|
do_unlock = trio.Event()
|
|
|
|
async def wait_for_parent_stdin_hijack(
|
|
task_status=trio.TASK_STATUS_IGNORED
|
|
):
|
|
global _debugger_request_cs
|
|
with trio.CancelScope() as cs:
|
|
_debugger_request_cs = cs
|
|
try:
|
|
async with get_root() as portal:
|
|
async with portal.open_stream_from(
|
|
tractor._debug._hijack_stdin_relay_to_child,
|
|
subactor_uid=actor.uid,
|
|
) as stream:
|
|
|
|
# block until first yield above
|
|
async for val in stream:
|
|
|
|
assert val == 'Locked'
|
|
task_status.started()
|
|
|
|
# with trio.CancelScope(shield=True):
|
|
await do_unlock.wait()
|
|
|
|
# trigger cancellation of remote stream
|
|
break
|
|
finally:
|
|
log.debug(f"Exiting debugger for actor {actor}")
|
|
global _in_debug
|
|
_in_debug = False
|
|
log.debug(f"Child {actor} released parent stdio lock")
|
|
|
|
async def _bp():
|
|
"""Async breakpoint which schedules a parent stdio lock, and once complete
|
|
enters the ``pdbpp`` debugging console.
|
|
"""
|
|
task_name = trio.lowlevel.current_task().name
|
|
|
|
global _in_debug
|
|
|
|
# TODO: need a more robust check for the "root" actor
|
|
if actor._parent_chan and not is_root_process():
|
|
if _in_debug:
|
|
if _in_debug == task_name:
|
|
# this task already has the lock and is
|
|
# likely recurrently entering a breakpoint
|
|
return
|
|
|
|
# if **this** actor is already in debug mode block here
|
|
# waiting for the control to be released - this allows
|
|
# support for recursive entries to `tractor.breakpoint()`
|
|
log.warning(
|
|
f"Actor {actor.uid} already has a debug lock, waiting...")
|
|
await do_unlock.wait()
|
|
await trio.sleep(0.1)
|
|
|
|
# assign unlock callback for debugger teardown hooks
|
|
global _pdb_release_hook
|
|
_pdb_release_hook = do_unlock.set
|
|
|
|
# mark local actor as "in debug mode" to avoid recurrent
|
|
# entries/requests to the root process
|
|
_in_debug = task_name
|
|
|
|
# this **must** be awaited by the caller and is done using the
|
|
# root nursery so that the debugger can continue to run without
|
|
# being restricted by the scope of a new task nursery.
|
|
await actor._service_n.start(wait_for_parent_stdin_hijack)
|
|
|
|
elif is_root_process():
|
|
# we also wait in the root-parent for any child that
|
|
# may have the tty locked prior
|
|
if _debug_lock.locked(): # root process already has it; ignore
|
|
return
|
|
await _debug_lock.acquire()
|
|
_pdb_release_hook = _debug_lock.release
|
|
|
|
# block here one (at the appropriate frame *up* where
|
|
# ``breakpoint()`` was awaited and begin handling stdio
|
|
log.debug("Entering the synchronous world of pdb")
|
|
debug_func(actor)
|
|
|
|
# user code **must** await this!
|
|
return _bp()
|
|
|
|
|
|
def _mk_pdb():
|
|
# XXX: setting these flags on the pdb instance are absolutely
|
|
# critical to having ctrl-c work in the ``trio`` standard way!
|
|
# The stdlib's pdb supports entering the current sync frame
|
|
# on a SIGINT, with ``trio`` we pretty much never want this
|
|
# and we did we can handle it in the ``tractor`` task runtime.
|
|
|
|
pdb = PdbwTeardown()
|
|
pdb.allow_kbdint = True
|
|
pdb.nosigint = True
|
|
|
|
return pdb
|
|
|
|
|
|
def _set_trace(actor=None):
|
|
pdb = _mk_pdb()
|
|
|
|
if actor is not None:
|
|
log.runtime(f"\nAttaching pdb to actor: {actor.uid}\n")
|
|
|
|
pdb.set_trace(
|
|
# start 2 levels up in user code
|
|
frame=sys._getframe().f_back.f_back,
|
|
)
|
|
|
|
else:
|
|
# we entered the global ``breakpoint()`` built-in from sync code
|
|
global _in_debug, _pdb_release_hook
|
|
_in_debug = 'sync'
|
|
|
|
def nuttin():
|
|
pass
|
|
|
|
_pdb_release_hook = nuttin
|
|
|
|
pdb.set_trace(
|
|
# start 2 levels up in user code
|
|
frame=sys._getframe().f_back,
|
|
)
|
|
|
|
|
|
breakpoint = partial(
|
|
_breakpoint,
|
|
_set_trace,
|
|
)
|
|
|
|
|
|
def _post_mortem(actor):
|
|
log.runtime(f"\nAttaching to pdb in crashed actor: {actor.uid}\n")
|
|
pdb = _mk_pdb()
|
|
|
|
# custom Pdb post-mortem entry
|
|
pdbpp.xpm(Pdb=lambda: pdb)
|
|
|
|
|
|
post_mortem = partial(
|
|
_breakpoint,
|
|
_post_mortem,
|
|
)
|
|
|
|
|
|
async def _maybe_enter_pm(err):
|
|
if (
|
|
_state.debug_mode()
|
|
|
|
# NOTE: don't enter debug mode recursively after quitting pdb
|
|
# Iow, don't re-enter the repl if the `quit` command was issued
|
|
# by the user.
|
|
and not isinstance(err, bdb.BdbQuit)
|
|
|
|
# XXX: if the error is the likely result of runtime-wide
|
|
# cancellation, we don't want to enter the debugger since
|
|
# there's races between when the parent actor has killed all
|
|
# comms and when the child tries to contact said parent to
|
|
# acquire the tty lock.
|
|
|
|
# Really we just want to mostly avoid catching KBIs here so there
|
|
# might be a simpler check we can do?
|
|
and not is_multi_cancelled(err)
|
|
):
|
|
log.debug("Actor crashed, entering debug mode")
|
|
await post_mortem()
|
|
return True
|
|
|
|
else:
|
|
return False
|