1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
|
#!/usr/bin/env python3
# vim: set ts=4 sw=4 et sts=4 ai:
"""
Functions needed by hdmi2usb-mode-switch implemented using lsusb and other
Linux command line tools.
This will only run on Linux.
"""
import logging
import os
import re
import subprocess
from .base import *
# Try and find unbind-helper
def find_unbind_helper():
callpaths = [
os.path.join(os.path.dirname(__file__), "..",
"..", "bin", "unbind-helper"),
"unbind-helper",
]
for path in callpaths:
pathret = subprocess.call(
path,
shell=True,
stdout=subprocess.DEVNULL,
stderr=subprocess.DEVNULL)
if pathret == 255:
return path
logging.warning("unbind-helper not found, will have to run as root!")
return None
unbind_helper = None
SYS_ROOT = '/sys/bus/usb/devices'
def get_path_from_sysdir(dirpath):
buspath = os.path.join(dirpath, 'busnum')
devpath = os.path.join(dirpath, 'devnum')
if not os.path.exists(buspath):
logging.info("Skipping %s (no busnum)", buspath)
return None
if not os.path.exists(devpath):
logging.info("Skipping %s (no devnum)", devpath)
return None
busnum = int(open(buspath, 'r').read().strip())
devnum = int(open(devpath, 'r').read().strip())
return Path(bus=busnum, address=devnum)
def create_sys_mapping():
# 1-1.3.1 --> (Device) bus-port.port.port
# 1-0:1.0 --> (Interface) bus-port.port.port:config.interface
# usb1 --> bus<number>
devices = {}
interfaces = {}
dirs = list(sorted(os.listdir(SYS_ROOT)))
for dirname in dirs:
dirpath = os.path.join(SYS_ROOT, dirname)
if ":" in dirname:
continue
path = get_path_from_sysdir(dirpath)
assert path
devices[dirpath] = path
assert path not in interfaces
interfaces[path] = [dirpath]
for dirname in dirs:
dirpath = os.path.join(SYS_ROOT, dirname)
if ":" not in dirname:
continue
device, interface = dirname.split(':')
if device.endswith('-0'):
device = "usb%s" % (device[:-2])
devpath = os.path.join(SYS_ROOT, device)
assert os.path.exists(devpath)
assert devpath in devices
interfaces[devices[devpath]].append(dirpath)
return interfaces
FIND_SYS_CACHE = {}
def find_sys(path, mapping=FIND_SYS_CACHE):
if not mapping:
mapping.update(create_sys_mapping())
return mapping[path]
class LsusbDevice(DeviceBase):
def __new__(cls, *args, **kw):
syspaths = sorted(find_sys(kw['path']))
# Get the did/serialno number from sysfs
did = None
serialno = None
for syspath in syspaths:
didpath = os.path.join(syspath, "bcdDevice")
if os.path.exists(didpath):
newdid = open(didpath, "r").read().strip()
assert did is None or did == newdid, (did, newdid)
did = newdid
serialnopath = os.path.join(syspath, "serial")
if os.path.exists(serialnopath):
newserialno = open(serialnopath, "r").read().strip()
assert serialno is None or serialno == newserialno, (
serialno, newserialno)
serialno = newserialno
d = DeviceBase.__new__(cls, *args, did=did, serialno=serialno, **kw)
d.syspaths = syspaths
return d
def inuse(self):
return bool(self.drivers())
def drivers(self):
drivers = {}
for path in self.syspaths[1:]:
driver_path = os.path.join(path, "driver")
if os.path.exists(driver_path):
drivers[path] = os.readlink(driver_path)
return tuple(set(d.split('/')[-1] for d in drivers.values()))
def detach(self):
for path in self.syspaths[1:]:
driver_path = os.path.join(path, "driver")
if os.path.exists(driver_path):
unbind_path = os.path.join(driver_path, "unbind")
assert os.path.exists(unbind_path), unbind_path
interface = os.path.split(path)[-1]
try:
open(unbind_path, "w").write(interface)
except PermissionError:
if not unbind_helper:
raise
subprocess.check_call("%s '%s' '%s'" % (
unbind_helper, unbind_path, interface), shell=True)
def tty(self):
ttys = []
for path in self.syspaths:
tty_path = os.path.join(path, "tty")
if os.path.exists(tty_path):
names = list(os.listdir(tty_path))
assert len(names) == 1
ttys.append('/dev/' + names[0])
return ttys
Device = LsusbDevice
def find_usb_devices():
FIND_SYS_CACHE.clear()
# 'Bus 002 Device 002: ID 8087:0024 Intel Corp. Integrated Rate Matching Hub' # noqa
lsusb_device_regex = re.compile(
"Bus (?P<bus>[0-9]+) Device (?P<address>[0-9]+):"
" ID (?P<vid>[0-9a-f]+):(?P<pid>[0-9a-f]+)")
devobjs = []
output = subprocess.check_output('lsusb')
for line in output.splitlines():
line = line.decode('utf-8')
bits = lsusb_device_regex.match(line)
assert bits, repr(line)
vid = int(bits.group('vid'), base=16)
pid = int(bits.group('pid'), base=16)
bus = int(bits.group('bus'), base=10)
address = int(bits.group('address'), base=10)
devobjs.append(LsusbDevice(
vid=vid, pid=pid, path=Path(bus=bus, address=address),
))
return devobjs
|
