dragonpilot for 0.8.9-3

2026-08-21 16:23:50 +08:00 · 2021-10-26 22:39:29 -07:00
parent 37192c1aba
commit f3f330886b
347 changed files with 44709 additions and 9016 deletions
@@ -0,0 +1,58 @@
+#!/usr/bin/env python
+import os
+import sys
+from subprocess import Popen, PIPE
+import fcntl
+
+def create_usb_list():
+    device_list = list()
+    try:
+        lsusb_out = Popen('lsusb -v', shell=True, bufsize=64, stdin=PIPE, stdout=PIPE, close_fds=True).stdout.read().strip().decode('utf-8')
+        usb_devices = lsusb_out.split('%s%s' % (os.linesep, os.linesep))
+        for device_categories in usb_devices:
+            if not device_categories:
+                continue
+            categories = device_categories.split(os.linesep)
+            device_stuff = categories[0].strip().split()
+            bus = device_stuff[1]
+            device = device_stuff[3][:-1]
+            device_dict = {'bus': bus, 'device': device}
+            device_info = ' '.join(device_stuff[6:])
+            device_dict['description'] = device_info
+            for category in categories:
+                if not category:
+                    continue
+                categoryinfo = category.strip().split()
+                if categoryinfo[0] == 'iManufacturer':
+                    manufacturer_info = ' '.join(categoryinfo[2:])
+                    device_dict['manufacturer'] = manufacturer_info
+                if categoryinfo[0] == 'iProduct':
+                    device_info = ' '.join(categoryinfo[2:])
+                    device_dict['device'] = device_info
+            path = '/dev/bus/usb/%s/%s' % (bus, device)
+            device_dict['path'] = path
+
+            device_list.append(device_dict)
+    except Exception as ex:
+        print('Failed to list usb devices! Error: %s' % ex)
+        sys.exit(-1)
+    return device_list
+
+
+def kill_usb(dev_path):
+    USBDEVFS_RESET = 21780
+    try:
+        f = open(dev_path, 'w', os.O_WRONLY)
+        fcntl.ioctl(f, USBDEVFS_RESET, 0)
+        print('Successfully reset %s' % dev_path)
+        sys.exit(0)
+    except Exception as ex:
+        print('Failed to reset device! Error: %s' % ex)
+        sys.exit(-1)
+
+if __name__ == "__main__":
+    usb_list = create_usb_list()
+    for device in usb_list:
+        if device['manufacturer'] == 'comma.ai':
+            kill_usb(device['path'])
+    sys.exit(0)