Move type hints to __init__.py

This commit is contained in:
Duncan Ogilvie
2024-09-22 21:37:21 +02:00
parent 7ab27ed890
commit b376658d59
3 changed files with 137 additions and 135 deletions
+137 -1
View File
@@ -1,4 +1,137 @@
from .icicle import *
from typing import List, Dict, Tuple
from enum import Enum
class MemoryProtection(Enum):
NoAccess = ...
ReadOnly = ...
ReadWrite = ...
ExecuteOnly = ...
ExecuteRead = ...
ExecuteReadWrite = ...
class MemoryExceptionCode(Enum):
Unallocated = ...
Unmapped = ...
UnmappedRegisters = ...
Uninitialized = ...
ReadViolation = ...
WriteViolation = ...
ExecViolation = ...
ReadWatch = ...
WriteWatch = ...
Unaligned = ...
OutOfMemory = ...
SelfModifyingCode = ...
AddressOverflow = ...
Unknown = ...
class RunStatus(Enum):
Running = ...
InstructionLimit = ...
Breakpoint = ...
Interrupted = ...
Halt = ...
Killed = ...
Deadlock = ...
OutOfMemory = ...
Unimplemented = ...
UnhandledException = ...
class ExceptionCode(Enum):
NoException = ...
InstructionLimit = ...
Halt = ...
Sleep = ...
Syscall = ...
CpuStateChanged = ...
DivisionException = ...
ReadUnmapped = ...
ReadPerm = ...
ReadUnaligned = ...
ReadWatch = ...
ReadUninitialized = ...
WriteUnmapped = ...
WritePerm = ...
WriteWatch = ...
WriteUnaligned = ...
ExecViolation = ...
SelfModifyingCode = ...
OutOfMemory = ...
AddressOverflow = ...
InvalidInstruction = ...
UnknownInterrupt = ...
UnknownCpuID = ...
InvalidOpSize = ...
InvalidFloatSize = ...
CodeNotTranslated = ...
ShadowStackOverflow = ...
ShadowStackInvalid = ...
InvalidTarget = ...
UnimplementedOp = ...
ExternalAddr = ...
Environment = ...
JitError = ...
InternalError = ...
UnmappedRegister = ...
UnknownError = ...
class Icicle:
def __init__(self, architecture: str, *,
jit = True,
jit_mem = True,
shadow_stack = True,
recompilation = True,
track_uninitialized = False,
optimize_instructions = True,
optimize_block = True,
tracing = False,
) -> None: ...
@property
def exception_code(self) -> ExceptionCode: ...
@property
def exception_value(self) -> int: ...
icount: int
icount_limit: int
# TODO: API to get memory information?
def mem_map(self, address: int, size: int, protection: MemoryProtection): ...
def mem_unmap(self, address: int, size: int): ...
def mem_protect(self, address: int, size: int, protection: MemoryProtection): ...
def mem_read(self, address: int, size: int) -> bytes: ...
def mem_write(self, address: int, data: bytes) -> None: ...
def reg_list(self) -> Dict[str, Tuple[int, int]]: ...
def reg_offset(self, name: str) -> int: ...
def reg_size(self, name: str) -> int: ...
def reg_read(self, name: str) -> int: ...
def reg_write(self, name: str, value: int) -> None: ...
def reset(self): ...
def run(self) -> RunStatus: ...
def run_until(self, address: int) -> RunStatus: ...
def step(self, count: int) -> RunStatus: ...
def add_breakpoint(self, address: int) -> bool: ...
def remove_breakpoint(self, address: int) -> bool: ...
def architectures() -> List[str]: ...
class MemoryException(Exception):
def __init__(self, message: str, code: MemoryExceptionCode):
@@ -19,3 +152,6 @@ def __ghidra_init():
raise FileNotFoundError("Ghidra processor definitions not found")
__ghidra_init()
# NOTE: This overrides the stubs at runtime with the actual implementation
from .icicle import *
-134
View File
@@ -1,134 +0,0 @@
from typing import List, Dict, Tuple
from enum import Enum
class MemoryProtection(Enum):
NoAccess = ...
ReadOnly = ...
ReadWrite = ...
ExecuteOnly = ...
ExecuteRead = ...
ExecuteReadWrite = ...
class MemoryExceptionCode(Enum):
Unallocated = ...
Unmapped = ...
UnmappedRegisters = ...
Uninitialized = ...
ReadViolation = ...
WriteViolation = ...
ExecViolation = ...
ReadWatch = ...
WriteWatch = ...
Unaligned = ...
OutOfMemory = ...
SelfModifyingCode = ...
AddressOverflow = ...
Unknown = ...
class RunStatus(Enum):
Running = ...
InstructionLimit = ...
Breakpoint = ...
Interrupted = ...
Halt = ...
Killed = ...
Deadlock = ...
OutOfMemory = ...
Unimplemented = ...
UnhandledException = ...
class ExceptionCode(Enum):
NoException = ...
InstructionLimit = ...
Halt = ...
Sleep = ...
Syscall = ...
CpuStateChanged = ...
DivisionException = ...
ReadUnmapped = ...
ReadPerm = ...
ReadUnaligned = ...
ReadWatch = ...
ReadUninitialized = ...
WriteUnmapped = ...
WritePerm = ...
WriteWatch = ...
WriteUnaligned = ...
ExecViolation = ...
SelfModifyingCode = ...
OutOfMemory = ...
AddressOverflow = ...
InvalidInstruction = ...
UnknownInterrupt = ...
UnknownCpuID = ...
InvalidOpSize = ...
InvalidFloatSize = ...
CodeNotTranslated = ...
ShadowStackOverflow = ...
ShadowStackInvalid = ...
InvalidTarget = ...
UnimplementedOp = ...
ExternalAddr = ...
Environment = ...
JitError = ...
InternalError = ...
UnmappedRegister = ...
UnknownError = ...
class Icicle:
def __init__(self, architecture: str, *,
jit = True,
jit_mem = True,
shadow_stack = True,
recompilation = True,
track_uninitialized = False,
optimize_instructions = True,
optimize_block = True,
tracing = False,
) -> None: ...
@property
def exception_code(self) -> ExceptionCode: ...
@property
def exception_value(self) -> int: ...
icount: int
icount_limit: int
# TODO: API to get memory information?
def mem_map(self, address: int, size: int, protection: MemoryProtection): ...
def mem_unmap(self, address: int, size: int): ...
def mem_protect(self, address: int, size: int, protection: MemoryProtection): ...
def mem_read(self, address: int, size: int) -> bytes: ...
def mem_write(self, address: int, data: bytes) -> None: ...
def reg_list(self) -> Dict[str, Tuple[int, int]]: ...
def reg_offset(self, name: str) -> int: ...
def reg_size(self, name: str) -> int: ...
def reg_read(self, name: str) -> int: ...
def reg_write(self, name: str, value: int) -> None: ...
def reset(self): ...
def run(self) -> RunStatus: ...
def run_until(self, address: int) -> RunStatus: ...
def step(self, count: int) -> RunStatus: ...
def add_breakpoint(self, address: int) -> bool: ...
def remove_breakpoint(self, address: int) -> bool: ...
def architectures() -> List[str]: ...
View File