oscura 0.5.0__py3-none-any.whl → 0.6.0__py3-none-any.whl

oscura/{acquisition → hardware/acquisition}/visa.py

@@ -9,7 +9,7 @@ waveform data directly from the oscilloscope, returning WaveformTrace format
 for analysis.
 
 Example:
-    >>> from oscura.acquisition import HardwareSource
+    >>> from oscura.hardware.acquisition import HardwareSource
     >>>
     >>> # USB oscilloscope
     >>> with HardwareSource.visa("USB0::0x0699::0x0401::INSTR") as scope:
@@ -44,16 +44,39 @@ References:
 
 from __future__ import annotations
 
+import time
 from collections.abc import Iterator
 from datetime import datetime
-from typing import TYPE_CHECKING, Any
+from typing import TYPE_CHECKING, Any, Protocol, cast
+
+# Optional dependency - import at module level for mocking in tests
+try:
+    import pyvisa
+except ImportError:
+    pyvisa = None  # type: ignore[assignment]
 
 if TYPE_CHECKING:
-    from pyvisa import Resource, ResourceManager
+    from pyvisa import ResourceManager
 
     from oscura.core.types import Trace
 
 
+class VisaResource(Protocol):
+    """Protocol for PyVISA Resource interface."""
+
+    def query(self, message: str) -> str:
+        """Query the instrument."""
+        ...
+
+    def write(self, message: str) -> int:
+        """Write to the instrument."""
+        ...
+
+    def close(self) -> None:
+        """Close the resource."""
+        ...
+
+
 class VISASource:
     """PyVISA instrument acquisition source.
 
@@ -95,7 +118,7 @@ class VISASource:
         self.resource = resource
         self.kwargs = kwargs
         self.rm: ResourceManager | None = None
-        self.instrument: Resource | None = None
+        self.instrument: VisaResource | None = None
         self._closed = False
 
         # Configuration
@@ -114,12 +137,10 @@ class VISASource:
         if self.instrument is not None:
             return
 
-        try:
-            import pyvisa
-        except ImportError as e:
+        if pyvisa is None:
             raise ImportError(
                 "VISA source requires pyvisa library. Install with: pip install pyvisa pyvisa-py"
-            ) from e
+            )
 
         try:
             self.rm = pyvisa.ResourceManager()
@@ -131,7 +152,10 @@ class VISASource:
                     raise RuntimeError("No VISA instruments found")
                 self.resource = resources[0]
 
-            self.instrument = self.rm.open_resource(self.resource, **self.kwargs)
+            # Cast: pyvisa's Resource implements our VisaResource Protocol
+            self.instrument = cast(
+                "VisaResource", self.rm.open_resource(self.resource, **self.kwargs)
+            )
 
             # Query instrument identity
             try:
@@ -319,8 +343,6 @@ class VISASource:
         if self._closed:
             raise ValueError("Cannot stream from closed source")
 
-        import time
-
         start_time = time.time()
 
         while time.time() - start_time < duration:

oscura/hardware/firmware/__init__.py

@@ -0,0 +1,29 @@
+"""Firmware analysis and pattern recognition.
+
+This module provides firmware reverse engineering capabilities including:
+- Pattern-based function detection
+- Architecture fingerprinting
+- String and data region identification
+- Interrupt vector table extraction
+- Compiler signature detection
+"""
+
+from oscura.hardware.firmware.pattern_recognition import (
+    Architecture,
+    CompilerSignature,
+    FirmwareAnalysisResult,
+    FirmwarePatternRecognizer,
+    Function,
+    InterruptVector,
+    StringTable,
+)
+
+__all__ = [
+    "Architecture",
+    "CompilerSignature",
+    "FirmwareAnalysisResult",
+    "FirmwarePatternRecognizer",
+    "Function",
+    "InterruptVector",
+    "StringTable",
+]