ruff fmt

Author: toolCHAINZ

crackers_python/crackers/config/library.py

@@ -1,6 +1,21 @@
+from typing import List
+
 from pydantic import BaseModel
 
 
+class LoadedLibraryConfig(BaseModel):
+    """
+    Represents an additional library to load alongside the main library.
+
+    Attributes:
+        path (str): Filesystem path of the target binary.
+        base_address (int | None): Optional base address for loading the library (may be adjusted/aligned on the Rust side).
+    """
+
+    path: str
+    base_address: int | None
+
+
 class LibraryConfig(BaseModel):
     """
     Configuration for a binary library used in analysis or exploitation.
@@ -10,9 +25,11 @@ class LibraryConfig(BaseModel):
         path (str): Filesystem path of the target binary.
         sample_size (int | None): Maximum number of gadgets to randomly sample (None to use all gadgets).
         base_address (int | None): Base address for loading the library, or None if not specified.
+        loaded_libraries (list[LoadedLibraryConfig] | None): Optional additional libraries to load alongside the primary one.
     """
 
     max_gadget_length: int
     path: str
     sample_size: int | None
     base_address: int | None
+    loaded_libraries: list[LoadedLibraryConfig] | None

crackers_python/crackers/crackers.pyi

@@ -1,4 +1,4 @@
-from typing import Callable, Iterable, Optional, Union
+from typing import Callable, Dict, Iterable, List, Optional, Tuple, Union
 
 from z3 import z3  # type: ignore
 
@@ -65,11 +65,16 @@ class CrackersLogLevel:
     Trace: int
     Warn: int
 
+class LoadedLibraryConfig:
+    path: str
+    base_address: Optional[int]
+
 class GadgetLibraryConfig:
     max_gadget_length: int
     path: str
     sample_size: Optional[int]
     base_address: Optional[int]
+    loaded_libraries: Optional[List[LoadedLibraryConfig]]
 
 class MemoryEqualityConstraint:
     space: str
@@ -91,7 +96,7 @@ class PointerRangeConstraints:
 class SleighConfig:
     ghidra_path: str
 
-# New: represent the two possible shapes of the specification
+# Represent the two possible shapes of the specification
 class BinaryFileSpecification:
     """
     Represents the binary-file variant of the specification.
@@ -110,7 +115,7 @@ class RawPcodeSpecification:
 
     raw_pcode: str
 
-# SpecificationConfig is now a discriminated union of the two variants above.
+# SpecificationConfig is a discriminated union of the two variants above.
 SpecificationConfig = Union[BinaryFileSpecification, RawPcodeSpecification]
 
 class StateEqualityConstraint:
@@ -137,7 +142,7 @@ class SelectionFailure:
 
 class PythonDecisionResult_Unsat(DecisionResult):
     _0: SelectionFailure
-    pass
+    __match_args__ = ("_0",)
 
 class DecisionResult:
     AssignmentFound: PythonDecisionResult_AssignmentFound
@@ -151,8 +156,7 @@ StateConstraintGenerator = Callable[[State, int], z3.BoolRef]
 TransitionConstraintGenerator = Callable[[ModeledBlock], z3.BoolRef]
 
 class SynthesisParams:
-    def run(self) -> "DecisionResultType": ...
-    def add_precondition(self, fn: StateConstraintGenerator): ...
-    def add_postcondition(self, fn: StateConstraintGenerator): ...
-    def add_transition_constraint(self, fn: TransitionConstraintGenerator): ...
-    pass
+    def run(self) -> DecisionResultType: ...
+    def add_precondition(self, fn: StateConstraintGenerator) -> None: ...
+    def add_postcondition(self, fn: StateConstraintGenerator) -> None: ...
+    def add_transition_constraint(self, fn: TransitionConstraintGenerator) -> None: ...

crackers_python/gh_actions_setup.py

@@ -69,7 +69,10 @@ def install_z3_glibc(target_platform):
     github_token = os.environ.get("READ_ONLY_GITHUB_TOKEN")
     if github_token:
         request.add_header("Authorization", f"Bearer {github_token}")
-        print("Using GitHub PAT from READ_ONLY_GITHUB_TOKEN for API authentication.", file=sys.stderr)
+        print(
+            "Using GitHub PAT from READ_ONLY_GITHUB_TOKEN for API authentication.",
+            file=sys.stderr,
+        )
 
     with urllib.request.urlopen(request) as response:
         data = json.loads(response.read().decode())