context ¤

log = getLogger(__name__)

T = TypeVar('T')

P = ParamSpec('P')

c = ColorConfig(
    "backtrace",
    [
        ColorParamSpec(
            "prefix", "none", "color for prefix of current backtrace label"
        ),
        ColorParamSpec("address", "none", "color for backtrace (address)"),
        ColorParamSpec("symbol", "none", "color for backtrace (symbol)"),
        ColorParamSpec(
            "frame-label", "none", "color for backtrace (frame label)"
        ),
    ],
)

config_reserve_lines = add_param(
    "context-reserve-lines",
    "if-ctx-fits",
    "when to reserve lines after the prompt to reduce context shake",
    help_docstring='\nThe "if-ctx-fits" setting only reserves lines if the whole context would still fit vertically in the current terminal window.\nIt doesn\'t take into account line-wrapping due to insufficient terminal width.\n',
    param_class=PARAM_ENUM,
    enum_sequence=["never", "if-ctx-fits", "always"],
)

config_clear_screen = add_param(
    "context-clear-screen",
    False,
    "whether to clear the screen before printing the context",
)

config_output = add_param(
    "context-output",
    "stdout",
    'where pwndbg should output ("stdout" or file/tty)',
)

config_context_sections = add_param(
    "context-sections",
    "regs disasm code ghidra stack backtrace expressions threads heap_tracker",
    "which context sections are displayed (controls order)",
)

config_max_threads_display = add_param(
    "context-max-threads",
    4,
    "maximum number of threads displayed by the context command",
)

outputs: dict[str, str] = {}

output_settings: DefaultDict[str, dict[str, Any]] = defaultdict(dict)

banner_arg = add_argument(
    "banner",
    type=str,
    nargs="?",
    default="both",
    help="Where a banner should be placed: both, top , bottom, none",
)

context_history: DefaultDict[str, list[list[str]]] = defaultdict(list)

selected_history_index: int | None = None

context_history_size = add_param(
    "context-history-size", 50, "number of context history entries to store"
)

expressions = []

config_context_ghidra = add_param(
    "context-ghidra",
    "never",
    "when to try to decompile the current function with ghidra",
    help_docstring="Doing this is slow and requires radare2/r2pipe or rizin/rzpipe.",
    param_class=PARAM_ENUM,
    enum_sequence=["always", "never", "if-no-source"],
)

parser = ArgumentParser(
    description="Print out all registers and enhance the information."
)

disasm_lines = add_param(
    "context-disasm-lines",
    10,
    "number of additional lines to print in the disasm context",
)

source_disasm_lines = add_param(
    "context-code-lines",
    10,
    "number of source code lines to print by the context command",
)

should_decompile = add_param(
    "context-integration-decompile",
    True,
    "whether context should fall back to decompilation with no source code",
)

stack_lines = add_param(
    "context-stack-lines", 8, "number of lines to print in the stack context"
)

backtrace_lines = add_param(
    "context-backtrace-lines",
    8,
    "number of lines to print in the backtrace context",
)

backtrace_frame_label = add_param(
    "backtrace-frame-label", "", "frame number label for backtrace"
)

last_signal: list[str] = []

thread_status_messages = {
    "running": light_green("running"),
    "stopped": yellow("stopped"),
    "exited": gray("exited "),
}

context_sections = {
    "a": context_args,
    "r": context_regs,
    "d": context_disasm,
    "s": context_stack,
    "b": context_backtrace,
    "c": context_code,
}