Source code for libdebug.architectures.ptrace_hardware_breakpoint_provider

#
# This file is part of libdebug Python library (https://github.com/libdebug/libdebug).
# Copyright (c) 2023-2024 Roberto Alessandro Bertolini. All rights reserved.
# Licensed under the MIT license. See LICENSE file in the project root for details.
#

from collections.abc import Callable

from libdebug.architectures.amd64.amd64_ptrace_hw_bp_helper import (
    Amd64PtraceHardwareBreakpointManager,
)
from libdebug.architectures.ptrace_hardware_breakpoint_manager import (
    PtraceHardwareBreakpointManager,
)
from libdebug.state.thread_context import ThreadContext
from libdebug.utils.libcontext import libcontext




def ptrace_hardware_breakpoint_manager_provider(
    thread: ThreadContext,
    peek_user: Callable[[int, int], int],
    poke_user: Callable[[int, int, int], None],
) -> PtraceHardwareBreakpointManager:
    """Returns an instance of the hardware breakpoint manager to be used by the `_InternalDebugger` class."""
    architecture = libcontext.arch

    match architecture:
        case "amd64":
            return Amd64PtraceHardwareBreakpointManager(thread, peek_user, poke_user)
        case _:
            raise NotImplementedError(f"Architecture {architecture} not available.")