IQ.Pilot Prebuilt Release @ 27f668a

iqpilot/system/hardware/tici/usb_storage.py

@@ -0,0 +1,71 @@
+import os
+import subprocess
+
+from iqpilot.common.params import Params
+
+SCRIPT_PATH = os.path.join(os.path.dirname(os.path.abspath(__file__)), "set_usb_storage.sh")
+
+
+def apply_usb_storage_state(state: bool):
+  Params().put_bool("UsbStorageEnabled", state)
+  try:
+    args = ["sudo", SCRIPT_PATH]
+    if state:
+      args.append("--rebuild")
+    subprocess.Popen(args)
+  except OSError:
+    pass
+
+
+NCM_TRIED_MARKER = "/tmp/.iqemac_ncm_provisioned"
+MAX_NCM_ATTEMPTS = 3
+
+
+def _ncm_attempts() -> int:
+  try:
+    with open(NCM_TRIED_MARKER) as f:
+      return int(f.read().strip() or 0)
+  except (OSError, ValueError):
+    return 0
+
+
+def ensure_ncm_gadget() -> bool:
+  # configfs is RAM backed, so the gadget must be rebuilt every boot; binding it
+  # enumerates on the host, which must not happen mid-drive
+  if os.path.isdir("/sys/class/net/usb0"):
+    return True
+  params = Params()
+  from iqpilot.selfdrive.iqmodeld.egpu_helpers import egpu_selected
+  if not (params.get_bool("IQEmacEnabled") or egpu_selected(params)):
+    return False
+  attempts = _ncm_attempts()
+  if attempts >= MAX_NCM_ATTEMPTS:
+    return False
+  try:
+    # stamped before the run: the gadget build can wedge configfs in
+    # uninterruptible D state, and retrying that forever helps nobody
+    with open(NCM_TRIED_MARKER, "w") as f:
+      f.write(str(attempts + 1))
+    subprocess.run(["sudo", "-n", SCRIPT_PATH], timeout=120, check=False)
+  except (OSError, subprocess.SubprocessError):
+    return False
+  return os.path.isdir("/sys/class/net/usb0")
+
+
+INPUT_SUSPEND = "/sys/class/power_supply/battery/input_suspend"
+
+
+def suspend_usb_input(suspend: bool = True) -> bool:
+  # a host on the data port makes the PMIC sink USB-PD while OBD-C feeds the same
+  # rail; the SOM browns out. This closes the charge path only, data is untouched
+  if not os.path.exists(INPUT_SUSPEND):
+    return False
+  try:
+    with open(INPUT_SUSPEND) as f:
+      if f.read().strip() == ("1" if suspend else "0"):
+        return True
+  except OSError:
+    pass
+  rc = subprocess.run(["sudo", "-n", "sh", "-c", f"echo {int(suspend)} > {INPUT_SUSPEND}"],
+                      check=False, capture_output=True)
+  return rc.returncode == 0