mirror of
https://github.com/MCKero6423/uv-k5-v3-emulator.git
synced 2026-10-02 03:15:36 +00:00
mckero
59ca82403f
Wraps the connect/negotiate/command dance with a lock, since only one client can hold the QMP socket at a time. Events interleave with replies, so command() reads until it sees the matching return rather than trusting the first message. Connection failure carries the reason and the likely cause: key.py already holding the socket.
65 lines
2.3 KiB
Python
65 lines
2.3 KiB
Python
#!/usr/bin/env python3
|
|
"""Minimal QMP client for the UV-K5 emulator.
|
|
|
|
Deliberately not shared with tools/key.py: that script is standalone on purpose,
|
|
so it runs with nothing but the stdlib and no sys.path juggling.
|
|
|
|
Only one client can hold the QMP socket at a time -- the emulator is started with
|
|
server=on,wait=off, which accepts a single connection. A long-lived server owns
|
|
it and callers serialise through the lock.
|
|
"""
|
|
import json
|
|
import socket
|
|
import threading
|
|
|
|
|
|
class QmpClient:
|
|
def __init__(self, path: str, timeout: float = 5.0):
|
|
self._lock = threading.Lock()
|
|
self._sock = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
|
|
self._sock.settimeout(timeout)
|
|
try:
|
|
self._sock.connect(path)
|
|
except OSError as exc:
|
|
raise RuntimeError(
|
|
f"cannot reach the emulator at {path}: {exc}\n"
|
|
"Start it with tools/run.sh first. Note the QMP socket takes a "
|
|
"single client, so tools/key.py cannot be connected at the same "
|
|
"time."
|
|
) from exc
|
|
self._buf = b""
|
|
self._read_json() # greeting
|
|
self.command("qmp_capabilities")
|
|
|
|
def _read_json(self) -> dict:
|
|
while b"\n" not in self._buf:
|
|
chunk = self._sock.recv(65536)
|
|
if not chunk:
|
|
raise RuntimeError("QMP connection closed")
|
|
self._buf += chunk
|
|
line, self._buf = self._buf.split(b"\n", 1)
|
|
return json.loads(line)
|
|
|
|
def command(self, name: str, **args):
|
|
payload = {"execute": name}
|
|
if args:
|
|
payload["arguments"] = args
|
|
with self._lock:
|
|
self._sock.sendall(json.dumps(payload).encode() + b"\n")
|
|
while True:
|
|
msg = self._read_json()
|
|
if "error" in msg:
|
|
raise RuntimeError(
|
|
f"QMP {name} failed: "
|
|
f"{msg['error'].get('desc', msg['error'])}")
|
|
if "return" in msg:
|
|
return msg["return"]
|
|
# Events (STOP, RESET, RESUME, ...) interleave with replies;
|
|
# keep reading until the matching return arrives.
|
|
|
|
def close(self):
|
|
try:
|
|
self._sock.close()
|
|
except OSError:
|
|
pass
|