You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
pwndbg/tests/unit-tests/mocks/gdblib.py

37 lines
976 B
Python

import sys
import types
from unittest.mock import MagicMock
from mocks.arch import Amd64Arch
from mocks.config import Config
from mocks.typeinfo import Amd64TypeInfo
class GdbLib(types.ModuleType):
def __init__(self, module_name):
super().__init__(module_name)
self.config_mod = Config(module_name + ".config")
self.arch = Amd64Arch(module_name + ".arch")
self.typeinfo = Amd64TypeInfo(module_name + ".typeinfo")
self.regs = MagicMock(__name__=module_name + ".regs")
self.prompt = MagicMock()
sys.modules[self.config_mod.__name__] = self.config_mod
sys.modules[self.arch.__name__] = self.arch
sys.modules[self.typeinfo.__name__] = self.typeinfo
sys.modules[self.regs.__name__] = self.regs
def load_gdblib(self):
pass
module_name = "pwndbg.gdblib"
module = GdbLib(module_name)
sys.modules[module_name] = module
import pwndbg
pwndbg.gdblib = sys.modules["pwndbg.gdblib"]