mirror of
https://github.com/commaai/agnos-builder.git
synced 2026-07-20 06:12:05 +08:00
Adeeb Shihadeh
98 lines
2.7 KiB
Python
Executable File
98 lines
2.7 KiB
Python
Executable File
#!/usr/bin/env -S uv run --script
|
|
# /// script
|
|
# dependencies = ["pyusb"]
|
|
# ///
|
|
import argparse
|
|
import os
|
|
import subprocess
|
|
import sys
|
|
import time
|
|
|
|
import usb.core
|
|
|
|
|
|
class Pins:
|
|
HFC_VID = 0x0424
|
|
HFC_PID = 0x704C
|
|
USB7002_VID = 0x0424
|
|
USB7002_PID = 0x7002
|
|
USB4002_VID = 0x0424
|
|
USB4002_PID = 0x4002
|
|
PIO96_OEN = 0xBF800908
|
|
PIO96_OUT = 0xBF800928
|
|
VIN_EN = 1 << (92 - 64)
|
|
|
|
|
|
class Mdma:
|
|
def __init__(self):
|
|
self.aux_ports = [
|
|
(self.hub_location(Pins.USB7002_VID, Pins.USB7002_PID), "1"),
|
|
(self.hub_location(Pins.USB4002_VID, Pins.USB4002_PID), "1"),
|
|
]
|
|
self.dev = None
|
|
|
|
def hub_location(self, vid, pid):
|
|
needle = f"[{vid:04x}:{pid:04x} "
|
|
for line in subprocess.check_output(["uhubctl", "-S"], text=True).splitlines():
|
|
if line.startswith("Current status for hub ") and needle in line:
|
|
return line.split()[4]
|
|
raise SystemExit(f"could not find hub {vid:04x}:{pid:04x}")
|
|
|
|
def _dev(self):
|
|
self.dev = self.dev or usb.core.find(idVendor=Pins.HFC_VID, idProduct=Pins.HFC_PID)
|
|
return self.dev
|
|
|
|
def reg(self, addr, value=None):
|
|
dev = self._dev()
|
|
if value is None:
|
|
return int.from_bytes(bytes(dev.ctrl_transfer(0xC0, 0x04, addr & 0xFFFF, addr >> 16, 4)), "little")
|
|
dev.ctrl_transfer(0x40, 0x03, addr & 0xFFFF, addr >> 16, value.to_bytes(4, "little"))
|
|
|
|
def gpio(self, bit, on):
|
|
if on:
|
|
self.reg(Pins.PIO96_OEN, self.reg(Pins.PIO96_OEN) & ~bit)
|
|
else:
|
|
self.reg(Pins.PIO96_OUT, self.reg(Pins.PIO96_OUT) & ~bit)
|
|
self.reg(Pins.PIO96_OEN, self.reg(Pins.PIO96_OEN) | bit)
|
|
|
|
def aux(self, action):
|
|
for hub, port in self.aux_ports:
|
|
subprocess.run(["uhubctl", "-S", "-e", "-l", hub, "-p", port, "-a", action], check=True)
|
|
|
|
def reboot(self):
|
|
self.aux("off")
|
|
self.gpio(Pins.VIN_EN, False)
|
|
time.sleep(0.1)
|
|
self.gpio(Pins.VIN_EN, True)
|
|
time.sleep(0.1)
|
|
self.aux("on")
|
|
|
|
def reboot_qdl(self):
|
|
self.aux("on")
|
|
self.gpio(Pins.VIN_EN, False)
|
|
time.sleep(0.1)
|
|
self.gpio(Pins.VIN_EN, True)
|
|
|
|
def serial(self):
|
|
os.execvp("screen", ["screen", "/dev/serial/by-id/usb-Microchip_Tech_USB2_Controller_Hub-if01", "115200"])
|
|
|
|
|
|
if __name__ == "__main__":
|
|
parser = argparse.ArgumentParser()
|
|
subparsers = parser.add_subparsers(dest="command", required=True)
|
|
subparsers.add_parser("reboot", help="power cycle into normal boot")
|
|
subparsers.add_parser("reboot-qdl", help="power cycle with AUX present to enter QDL")
|
|
subparsers.add_parser("serial", help="open the MSM UART console with screen")
|
|
if len(sys.argv) == 1:
|
|
parser.print_help()
|
|
raise SystemExit(0)
|
|
args = parser.parse_args()
|
|
|
|
mdma = Mdma()
|
|
if args.command == "reboot":
|
|
mdma.reboot()
|
|
elif args.command == "reboot-qdl":
|
|
mdma.reboot_qdl()
|
|
elif args.command == "serial":
|
|
mdma.serial()
|