支持鸿蒙OSNEXT_HDC
This commit is contained in:
floatingstarZ
381
phone_agent/hdc/connection.py
Normal file
381
phone_agent/hdc/connection.py
Normal file
@@ -0,0 +1,381 @@
|
||||
"""HDC connection management for HarmonyOS devices."""
|
||||
|
||||
import os
|
||||
import subprocess
|
||||
import time
|
||||
from dataclasses import dataclass
|
||||
from enum import Enum
|
||||
from typing import Optional
|
||||
|
||||
from phone_agent.config.timing import TIMING_CONFIG
|
||||
|
||||
|
||||
# Global flag to control HDC command output
|
||||
_HDC_VERBOSE = os.getenv("HDC_VERBOSE", "false").lower() in ("true", "1", "yes")
|
||||
|
||||
|
||||
def _run_hdc_command(cmd: list, **kwargs) -> subprocess.CompletedProcess:
|
||||
"""
|
||||
Run HDC command with optional verbose output.
|
||||
|
||||
Args:
|
||||
cmd: Command list to execute.
|
||||
**kwargs: Additional arguments for subprocess.run.
|
||||
|
||||
Returns:
|
||||
CompletedProcess result.
|
||||
"""
|
||||
if _HDC_VERBOSE:
|
||||
print(f"[HDC] Running command: {' '.join(cmd)}")
|
||||
|
||||
result = subprocess.run(cmd, **kwargs)
|
||||
|
||||
if _HDC_VERBOSE and result.returncode != 0:
|
||||
print(f"[HDC] Command failed with return code {result.returncode}")
|
||||
if hasattr(result, 'stderr') and result.stderr:
|
||||
print(f"[HDC] Error: {result.stderr}")
|
||||
|
||||
return result
|
||||
|
||||
|
||||
def set_hdc_verbose(verbose: bool):
|
||||
"""Set HDC verbose mode globally."""
|
||||
global _HDC_VERBOSE
|
||||
_HDC_VERBOSE = verbose
|
||||
|
||||
|
||||
class ConnectionType(Enum):
|
||||
"""Type of HDC connection."""
|
||||
|
||||
USB = "usb"
|
||||
WIFI = "wifi"
|
||||
REMOTE = "remote"
|
||||
|
||||
|
||||
@dataclass
|
||||
class DeviceInfo:
|
||||
"""Information about a connected device."""
|
||||
|
||||
device_id: str
|
||||
status: str
|
||||
connection_type: ConnectionType
|
||||
model: str | None = None
|
||||
harmony_version: str | None = None
|
||||
|
||||
|
||||
class HDCConnection:
|
||||
"""
|
||||
Manages HDC connections to HarmonyOS devices.
|
||||
|
||||
Supports USB, WiFi, and remote TCP/IP connections.
|
||||
|
||||
Example:
|
||||
>>> conn = HDCConnection()
|
||||
>>> # Connect to remote device
|
||||
>>> conn.connect("192.168.1.100:5555")
|
||||
>>> # List devices
|
||||
>>> devices = conn.list_devices()
|
||||
>>> # Disconnect
|
||||
>>> conn.disconnect("192.168.1.100:5555")
|
||||
"""
|
||||
|
||||
def __init__(self, hdc_path: str = "hdc"):
|
||||
"""
|
||||
Initialize HDC connection manager.
|
||||
|
||||
Args:
|
||||
hdc_path: Path to HDC executable.
|
||||
"""
|
||||
self.hdc_path = hdc_path
|
||||
|
||||
def connect(self, address: str, timeout: int = 10) -> tuple[bool, str]:
|
||||
"""
|
||||
Connect to a remote device via TCP/IP.
|
||||
|
||||
Args:
|
||||
address: Device address in format "host:port" (e.g., "192.168.1.100:5555").
|
||||
timeout: Connection timeout in seconds.
|
||||
|
||||
Returns:
|
||||
Tuple of (success, message).
|
||||
|
||||
Note:
|
||||
The remote device must have TCP/IP debugging enabled.
|
||||
"""
|
||||
# Validate address format
|
||||
if ":" not in address:
|
||||
address = f"{address}:5555" # Default HDC port
|
||||
|
||||
try:
|
||||
result = _run_hdc_command(
|
||||
[self.hdc_path, "tconn", address],
|
||||
capture_output=True,
|
||||
text=True,
|
||||
timeout=timeout,
|
||||
)
|
||||
|
||||
output = result.stdout + result.stderr
|
||||
|
||||
if "Connect OK" in output or "connected" in output.lower():
|
||||
return True, f"Connected to {address}"
|
||||
elif "already connected" in output.lower():
|
||||
return True, f"Already connected to {address}"
|
||||
else:
|
||||
return False, output.strip()
|
||||
|
||||
except subprocess.TimeoutExpired:
|
||||
return False, f"Connection timeout after {timeout}s"
|
||||
except Exception as e:
|
||||
return False, f"Connection error: {e}"
|
||||
|
||||
def disconnect(self, address: str | None = None) -> tuple[bool, str]:
|
||||
"""
|
||||
Disconnect from a remote device.
|
||||
|
||||
Args:
|
||||
address: Device address to disconnect. If None, disconnects all.
|
||||
|
||||
Returns:
|
||||
Tuple of (success, message).
|
||||
"""
|
||||
try:
|
||||
if address:
|
||||
cmd = [self.hdc_path, "tdisconn", address]
|
||||
else:
|
||||
# HDC doesn't have a "disconnect all" command, so we need to list and disconnect each
|
||||
devices = self.list_devices()
|
||||
for device in devices:
|
||||
if ":" in device.device_id: # Remote device
|
||||
_run_hdc_command(
|
||||
[self.hdc_path, "tdisconn", device.device_id],
|
||||
capture_output=True,
|
||||
text=True,
|
||||
timeout=5
|
||||
)
|
||||
return True, "Disconnected all remote devices"
|
||||
|
||||
result = _run_hdc_command(cmd, capture_output=True, text=True, encoding="utf-8", timeout=5)
|
||||
|
||||
output = result.stdout + result.stderr
|
||||
return True, output.strip() or "Disconnected"
|
||||
|
||||
except Exception as e:
|
||||
return False, f"Disconnect error: {e}"
|
||||
|
||||
def list_devices(self) -> list[DeviceInfo]:
|
||||
"""
|
||||
List all connected devices.
|
||||
|
||||
Returns:
|
||||
List of DeviceInfo objects.
|
||||
"""
|
||||
try:
|
||||
result = _run_hdc_command(
|
||||
[self.hdc_path, "list", "targets"],
|
||||
capture_output=True,
|
||||
text=True,
|
||||
timeout=5,
|
||||
)
|
||||
|
||||
devices = []
|
||||
for line in result.stdout.strip().split("\n"):
|
||||
if not line.strip():
|
||||
continue
|
||||
|
||||
# HDC output format: device_id (status)
|
||||
# Example: "192.168.1.100:5555" or "FMR0223C13000649"
|
||||
device_id = line.strip()
|
||||
|
||||
# Determine connection type
|
||||
if ":" in device_id:
|
||||
conn_type = ConnectionType.REMOTE
|
||||
else:
|
||||
conn_type = ConnectionType.USB
|
||||
|
||||
# HDC doesn't provide detailed status in list command
|
||||
# We assume "Connected" status for devices that appear
|
||||
devices.append(
|
||||
DeviceInfo(
|
||||
device_id=device_id,
|
||||
status="device",
|
||||
connection_type=conn_type,
|
||||
model=None,
|
||||
)
|
||||
)
|
||||
|
||||
return devices
|
||||
|
||||
except Exception as e:
|
||||
print(f"Error listing devices: {e}")
|
||||
return []
|
||||
|
||||
def get_device_info(self, device_id: str | None = None) -> DeviceInfo | None:
|
||||
"""
|
||||
Get detailed information about a device.
|
||||
|
||||
Args:
|
||||
device_id: Device ID. If None, uses first available device.
|
||||
|
||||
Returns:
|
||||
DeviceInfo or None if not found.
|
||||
"""
|
||||
devices = self.list_devices()
|
||||
|
||||
if not devices:
|
||||
return None
|
||||
|
||||
if device_id is None:
|
||||
return devices[0]
|
||||
|
||||
for device in devices:
|
||||
if device.device_id == device_id:
|
||||
return device
|
||||
|
||||
return None
|
||||
|
||||
def is_connected(self, device_id: str | None = None) -> bool:
|
||||
"""
|
||||
Check if a device is connected.
|
||||
|
||||
Args:
|
||||
device_id: Device ID to check. If None, checks if any device is connected.
|
||||
|
||||
Returns:
|
||||
True if connected, False otherwise.
|
||||
"""
|
||||
devices = self.list_devices()
|
||||
|
||||
if not devices:
|
||||
return False
|
||||
|
||||
if device_id is None:
|
||||
return len(devices) > 0
|
||||
|
||||
return any(d.device_id == device_id for d in devices)
|
||||
|
||||
def enable_tcpip(
|
||||
self, port: int = 5555, device_id: str | None = None
|
||||
) -> tuple[bool, str]:
|
||||
"""
|
||||
Enable TCP/IP debugging on a USB-connected device.
|
||||
|
||||
This allows subsequent wireless connections to the device.
|
||||
|
||||
Args:
|
||||
port: TCP port for HDC (default: 5555).
|
||||
device_id: Device ID. If None, uses first available device.
|
||||
|
||||
Returns:
|
||||
Tuple of (success, message).
|
||||
|
||||
Note:
|
||||
The device must be connected via USB first.
|
||||
After this, you can disconnect USB and connect via WiFi.
|
||||
"""
|
||||
try:
|
||||
cmd = [self.hdc_path]
|
||||
if device_id:
|
||||
cmd.extend(["-t", device_id])
|
||||
cmd.extend(["tmode", "port", str(port)])
|
||||
|
||||
result = _run_hdc_command(cmd, capture_output=True, text=True, encoding="utf-8", timeout=10)
|
||||
|
||||
output = result.stdout + result.stderr
|
||||
|
||||
if result.returncode == 0 or "success" in output.lower():
|
||||
time.sleep(TIMING_CONFIG.connection.adb_restart_delay)
|
||||
return True, f"TCP/IP mode enabled on port {port}"
|
||||
else:
|
||||
return False, output.strip()
|
||||
|
||||
except Exception as e:
|
||||
return False, f"Error enabling TCP/IP: {e}"
|
||||
|
||||
def get_device_ip(self, device_id: str | None = None) -> str | None:
|
||||
"""
|
||||
Get the IP address of a connected device.
|
||||
|
||||
Args:
|
||||
device_id: Device ID. If None, uses first available device.
|
||||
|
||||
Returns:
|
||||
IP address string or None if not found.
|
||||
"""
|
||||
try:
|
||||
cmd = [self.hdc_path]
|
||||
if device_id:
|
||||
cmd.extend(["-t", device_id])
|
||||
cmd.extend(["shell", "ifconfig"])
|
||||
|
||||
result = _run_hdc_command(cmd, capture_output=True, text=True, encoding="utf-8", timeout=5)
|
||||
|
||||
# Parse IP from ifconfig output
|
||||
for line in result.stdout.split("\n"):
|
||||
if "inet addr:" in line or "inet " in line:
|
||||
parts = line.strip().split()
|
||||
for i, part in enumerate(parts):
|
||||
if "addr:" in part:
|
||||
ip = part.split(":")[1]
|
||||
# Filter out localhost
|
||||
if not ip.startswith("127."):
|
||||
return ip
|
||||
elif part == "inet" and i + 1 < len(parts):
|
||||
ip = parts[i + 1].split("/")[0]
|
||||
if not ip.startswith("127."):
|
||||
return ip
|
||||
|
||||
return None
|
||||
|
||||
except Exception as e:
|
||||
print(f"Error getting device IP: {e}")
|
||||
return None
|
||||
|
||||
def restart_server(self) -> tuple[bool, str]:
|
||||
"""
|
||||
Restart the HDC server.
|
||||
|
||||
Returns:
|
||||
Tuple of (success, message).
|
||||
"""
|
||||
try:
|
||||
# Kill server
|
||||
_run_hdc_command(
|
||||
[self.hdc_path, "kill"], capture_output=True, timeout=5
|
||||
)
|
||||
|
||||
time.sleep(TIMING_CONFIG.connection.server_restart_delay)
|
||||
|
||||
# Start server (HDC auto-starts when running commands)
|
||||
_run_hdc_command(
|
||||
[self.hdc_path, "start", "-r"], capture_output=True, timeout=5
|
||||
)
|
||||
|
||||
return True, "HDC server restarted"
|
||||
|
||||
except Exception as e:
|
||||
return False, f"Error restarting server: {e}"
|
||||
|
||||
|
||||
def quick_connect(address: str) -> tuple[bool, str]:
|
||||
"""
|
||||
Quick helper to connect to a remote device.
|
||||
|
||||
Args:
|
||||
address: Device address (e.g., "192.168.1.100" or "192.168.1.100:5555").
|
||||
|
||||
Returns:
|
||||
Tuple of (success, message).
|
||||
"""
|
||||
conn = HDCConnection()
|
||||
return conn.connect(address)
|
||||
|
||||
|
||||
def list_devices() -> list[DeviceInfo]:
|
||||
"""
|
||||
Quick helper to list connected devices.
|
||||
|
||||
Returns:
|
||||
List of DeviceInfo objects.
|
||||
"""
|
||||
conn = HDCConnection()
|
||||
return conn.list_devices()
|
||||
Reference in New Issue
Block a user