feat(bridge): BLE 桥接能力——bridge_server.py(Windows原生) + BridgeTransport + --via/PPCLOCK_VIA,149 测试全绿
This commit is contained in:
1 parent
17e2512f6a
commit
49b2010088
7 files changed
+658
-2
No files matched your search
@@ -0,0 +1,236 @@
|
||||
#!/usr/bin/env python3
|
||||
"""ppclock BLE bridge 服务端 —— 在蓝牙物理所在机(host-win)原生运行。
|
||||
|
||||
依赖:pip install bleak(仅此一个,Windows 用 WinRT 后端)。
|
||||
运行:python bridge_server.py [--host 0.0.0.0] [--port 8971]
|
||||
|
||||
协议:JSONL over TCP。每行一个请求 {"id":N,"op":...},
|
||||
响应 {"id":N,"ok":bool,"data":...,"error":...};OTA 期间穿插
|
||||
{"event":"ota_progress",...} 事件。单 BLE 设备会话按连接串行化。
|
||||
|
||||
RPC:
|
||||
ping → {pong, version, platform}
|
||||
scan {timeout} → {devices:[{name,address,rssi}]}
|
||||
connect {address} → {}(连接+服务发现+订阅 RXTX notify)
|
||||
write {char,data,response}→ {}(char: "epd"|"rxtx",data: hex)
|
||||
read {char} → {data: hex}
|
||||
request_device_id {timeout}→ {device_id}(EFEF + 收集 14B notify)
|
||||
ota {image} → {blocks, bytes}(image: hex;完整 SUOTA)
|
||||
close → {}
|
||||
"""
|
||||
from __future__ import annotations
|
||||
|
||||
import argparse
|
||||
import asyncio
|
||||
import json
|
||||
import platform
|
||||
import sys
|
||||
|
||||
PROTOCOL_VERSION = 1
|
||||
|
||||
EPD_SERVICE_UUID = "13187b10-eba9-a3ba-044e-83d3217d9a38"
|
||||
EPD_CHAR_UUID = "4b646063-6264-f3a7-8941-e65356ea82fe"
|
||||
RXTX_SERVICE_UUID = "00001f10-0000-1000-8000-00805f9b34fb"
|
||||
RXTX_CHAR_UUID = "00001f1f-0000-1000-8000-00805f9b34fb"
|
||||
DEVICE_NAME_PREFIX = "NRF-"
|
||||
|
||||
SPOTA_SERVICE_UUID = "0000fef5-0000-1000-8000-00805f9b34fb"
|
||||
SPOTA_SERV_STATUS_UUID = "64b4e8b5-0de5-401b-a21d-acc8db3b913a"
|
||||
SPOTA_MEM_DEV_UUID = "8082caa8-41a6-4021-91c6-56f9b954cc34"
|
||||
SPOTA_GPIO_MAP_UUID = "724249f0-5ec3-4b5f-8804-42345af08651"
|
||||
SPOTA_PATCH_DATA_UUID = "457871e8-d516-4ca1-9116-57d0b17b9cb2"
|
||||
SPOTA_PATCH_LEN_UUID = "9d84b9a3-000c-49d8-9183-855b673fda31"
|
||||
MEM_DEV_SPI_FLASH = bytes([0x00, 0x00, 0x00, 0x13])
|
||||
GPIO_MAP_DEFAULT = bytes([0x00, 0x00, 0x06, 0x05])
|
||||
END_SIGNAL = bytes([0x00, 0x00, 0x00, 0xFE])
|
||||
REBOOT_SIGNAL = bytes([0x00, 0x00, 0x00, 0xFD])
|
||||
BLOCK_SIZE = 240
|
||||
WRITE_CHUNK = 20
|
||||
STATUS_OK = 2
|
||||
|
||||
CHARS = {"epd": EPD_CHAR_UUID, "rxtx": RXTX_CHAR_UUID}
|
||||
|
||||
|
||||
def parse_device_id(data: bytes) -> str:
|
||||
if all(0x20 <= b < 0x7F for b in data):
|
||||
return data.decode("ascii")
|
||||
return data.hex().upper()
|
||||
|
||||
|
||||
class BleSession:
|
||||
"""单设备 BLE 会话(bleak),与 ppclock.transport.BLETransport 同语义。"""
|
||||
|
||||
def __init__(self):
|
||||
self.client = None
|
||||
self._id_buf = bytearray()
|
||||
self._id_event = asyncio.Event()
|
||||
|
||||
async def scan(self, timeout: float):
|
||||
from bleak import BleakScanner
|
||||
found = await BleakScanner.discover(timeout=timeout, return_adv=True)
|
||||
out = []
|
||||
for addr, (dev, adv) in found.items():
|
||||
name = dev.name or adv.local_name or ""
|
||||
if name.startswith(DEVICE_NAME_PREFIX):
|
||||
out.append({"name": name, "address": dev.address, "rssi": adv.rssi})
|
||||
return {"devices": out}
|
||||
|
||||
async def connect(self, address: str, timeout: float = 15.0):
|
||||
from bleak import BleakClient
|
||||
await self.close()
|
||||
self.client = BleakClient(address, timeout=timeout)
|
||||
await self.client.connect()
|
||||
await self.client.start_notify(RXTX_CHAR_UUID, self._on_notify)
|
||||
return {"address": address}
|
||||
|
||||
def _on_notify(self, _sender, data: bytearray):
|
||||
self._id_buf.extend(data)
|
||||
if len(self._id_buf) >= 14:
|
||||
self._id_event.set()
|
||||
|
||||
async def write(self, char: str, data: bytes, response: bool):
|
||||
await self.client.write_gatt_char(CHARS[char], data, response=response)
|
||||
return {"sent": len(data)}
|
||||
|
||||
async def read(self, char: str):
|
||||
return {"data": bytes(await self.client.read_gatt_char(CHARS[char])).hex()}
|
||||
|
||||
async def request_device_id(self, timeout: float = 8.0):
|
||||
self._id_buf.clear()
|
||||
self._id_event.clear()
|
||||
await self.client.write_gatt_char(RXTX_CHAR_UUID, bytes([0xEF, 0xEF]), response=True)
|
||||
await asyncio.wait_for(self._id_event.wait(), timeout)
|
||||
return {"device_id": parse_device_id(bytes(self._id_buf[:14]))}
|
||||
|
||||
async def ota(self, image: bytes, status_timeout: float = 60.0):
|
||||
if SPOTA_SERVICE_UUID not in [str(s.uuid) for s in self.client.services]:
|
||||
raise RuntimeError("设备不存在 fef5 SUOTA 服务")
|
||||
status_q: asyncio.Queue[int] = asyncio.Queue()
|
||||
|
||||
def _on_status(_s, data: bytearray):
|
||||
status_q.put_nowait(int.from_bytes(bytes(data), "little"))
|
||||
|
||||
async def wait_ok(what: str):
|
||||
status = await asyncio.wait_for(status_q.get(), status_timeout)
|
||||
if status != STATUS_OK:
|
||||
raise RuntimeError(f"{what} 失败: 状态码 {status}")
|
||||
|
||||
acc = 0
|
||||
for b in image:
|
||||
acc ^= b
|
||||
payload = image + bytes([acc])
|
||||
blocks = [payload[i:i + BLOCK_SIZE] for i in range(0, len(payload), BLOCK_SIZE)]
|
||||
|
||||
await self.client.start_notify(SPOTA_SERV_STATUS_UUID, _on_status)
|
||||
events = []
|
||||
try:
|
||||
await self.client.write_gatt_char(SPOTA_MEM_DEV_UUID, MEM_DEV_SPI_FLASH, response=True)
|
||||
await self.client.write_gatt_char(SPOTA_GPIO_MAP_UUID, GPIO_MAP_DEFAULT, response=True)
|
||||
for idx, block in enumerate(blocks, 1):
|
||||
await self.client.write_gatt_char(SPOTA_PATCH_LEN_UUID,
|
||||
len(block).to_bytes(2, "little"), response=True)
|
||||
for i in range(0, len(block), WRITE_CHUNK):
|
||||
await self.client.write_gatt_char(SPOTA_PATCH_DATA_UUID,
|
||||
block[i:i + WRITE_CHUNK], response=False)
|
||||
await wait_ok(f"块 {idx}/{len(blocks)}")
|
||||
events.append({"block": idx, "total": len(blocks)})
|
||||
yield {"event": "ota_progress", "block": idx, "total": len(blocks)}
|
||||
await self.client.write_gatt_char(SPOTA_MEM_DEV_UUID, END_SIGNAL, response=True)
|
||||
await wait_ok("镜像校验")
|
||||
await self.client.write_gatt_char(SPOTA_MEM_DEV_UUID, REBOOT_SIGNAL, response=True)
|
||||
finally:
|
||||
try:
|
||||
await self.client.stop_notify(SPOTA_SERV_STATUS_UUID)
|
||||
except Exception:
|
||||
pass
|
||||
|
||||
async def close(self):
|
||||
if self.client is not None:
|
||||
try:
|
||||
if self.client.is_connected:
|
||||
await self.client.disconnect()
|
||||
finally:
|
||||
self.client = None
|
||||
|
||||
|
||||
class BridgeServer:
|
||||
def __init__(self):
|
||||
self.ble = BleSession()
|
||||
self._ble_lock = asyncio.Lock()
|
||||
|
||||
async def handle(self, reader: asyncio.StreamReader, writer: asyncio.StreamWriter):
|
||||
peer = writer.get_extra_info("peername")
|
||||
print(f"[bridge] client {peer} connected", flush=True)
|
||||
|
||||
async def send(obj):
|
||||
writer.write((json.dumps(obj, ensure_ascii=False) + "\n").encode())
|
||||
await writer.drain()
|
||||
|
||||
try:
|
||||
async for line in reader:
|
||||
line = line.strip()
|
||||
if not line:
|
||||
continue
|
||||
try:
|
||||
req = json.loads(line)
|
||||
rid = req.get("id")
|
||||
op = req["op"]
|
||||
except (KeyError, ValueError) as e:
|
||||
await send({"id": None, "ok": False, "error": f"bad request: {e}"})
|
||||
continue
|
||||
try:
|
||||
async with self._ble_lock:
|
||||
if op == "ota":
|
||||
gen = self.ble.ota(bytes.fromhex(req["image"]))
|
||||
async for ev in gen:
|
||||
await send(ev)
|
||||
data = {"done": True}
|
||||
else:
|
||||
data = await self._dispatch(op, req)
|
||||
await send({"id": rid, "ok": True, "data": data})
|
||||
except Exception as e:
|
||||
await send({"id": rid, "ok": False,
|
||||
"error": f"{type(e).__name__}: {e}"})
|
||||
finally:
|
||||
await self.ble.close()
|
||||
writer.close()
|
||||
print(f"[bridge] client {peer} disconnected", flush=True)
|
||||
|
||||
async def _dispatch(self, op: str, req: dict):
|
||||
if op == "ping":
|
||||
return {"pong": True, "version": PROTOCOL_VERSION,
|
||||
"platform": platform.system()}
|
||||
if op == "scan":
|
||||
return await self.ble.scan(float(req.get("timeout", 5.0)))
|
||||
if op == "connect":
|
||||
return await self.ble.connect(req["address"], float(req.get("timeout", 15.0)))
|
||||
if op == "write":
|
||||
return await self.ble.write(req["char"], bytes.fromhex(req["data"]),
|
||||
bool(req.get("response", True)))
|
||||
if op == "read":
|
||||
return await self.ble.read(req["char"])
|
||||
if op == "request_device_id":
|
||||
return await self.ble.request_device_id(float(req.get("timeout", 8.0)))
|
||||
if op == "close":
|
||||
return await self.ble.close() or {}
|
||||
raise ValueError(f"unknown op {op!r}")
|
||||
|
||||
|
||||
async def main():
|
||||
ap = argparse.ArgumentParser(description="ppclock BLE bridge server")
|
||||
ap.add_argument("--host", default="0.0.0.0")
|
||||
ap.add_argument("--port", type=int, default=8971)
|
||||
args = ap.parse_args()
|
||||
server = BridgeServer()
|
||||
srv = await asyncio.start_server(server.handle, args.host, args.port)
|
||||
print(f"[bridge] listening on {args.host}:{args.port} "
|
||||
f"(protocol v{PROTOCOL_VERSION}, platform {platform.system()})", flush=True)
|
||||
async with srv:
|
||||
await srv.serve_forever()
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
try:
|
||||
asyncio.run(main())
|
||||
except KeyboardInterrupt:
|
||||
sys.exit(0)
|
||||
Reference in new issue
Block a user