b38dea0597
Share Gaia assets and adapters, keep private profiles and development documents local, and update both public READMEs.
45 lines
2 KiB
Python
45 lines
2 KiB
Python
#!/usr/bin/env python3
|
|
# SPDX-License-Identifier: MPL-2.0
|
|
"""Read the Tundra diagnostic protocol on an explicitly selected Linux USB device."""
|
|
import argparse
|
|
import json
|
|
import os
|
|
from pathlib import Path
|
|
import re
|
|
import sys
|
|
from tundra import ROOT, now, save
|
|
|
|
from usb_protocol import exchange
|
|
|
|
|
|
def main():
|
|
parser=argparse.ArgumentParser(description=__doc__)
|
|
parser.add_argument('--device',required=True,help='/dev/bus/usb/BBB/DDD, determined after manual connection')
|
|
parser.add_argument('--stdout',action='store_true',help='Emit JSON only; do not create a report file')
|
|
parser.add_argument('--check-invalid',action='store_true',help='Also verify rejection of an unknown command')
|
|
args=parser.parse_args()
|
|
if not re.fullmatch(r'/dev/bus/usb/[0-9]{3}/[0-9]{3}',args.device):parser.error('Expected an explicit usbfs device path')
|
|
selected=Path(args.device)
|
|
identity=None
|
|
for entry in Path('/sys/bus/usb/devices').iterdir():
|
|
try:
|
|
path=Path('/dev/bus/usb')/f'{int((entry/"busnum").read_text()):03d}'/f'{int((entry/"devnum").read_text()):03d}'
|
|
if path==selected:
|
|
identity={key:(entry/key).read_text().strip() for key in ('idVendor','idProduct','serial')}
|
|
except (FileNotFoundError,NotADirectoryError):continue
|
|
if identity!={'idVendor':'1d6b','idProduct':'0104','serial':'TUNDRA-DIAG'}:
|
|
raise SystemExit('Selected device is not the Tundra development diagnostic; no commands sent.')
|
|
os.umask(0o077)
|
|
results=exchange(selected,('status','version','mounts','invalid') if args.check_invalid else ('status','version','mounts'))
|
|
if args.stdout:
|
|
print(json.dumps(results));return
|
|
path=ROOT/'logs/device'/('usb-'+now().replace(':','-')+'.json')
|
|
save(path,{'recordedAt':now(),'identity':identity,'readOnly':True,'results':results})
|
|
print(json.dumps(results,indent=2));print('Saved:',path)
|
|
|
|
if __name__=='__main__':
|
|
try:
|
|
main()
|
|
except (OSError,ValueError,RuntimeError) as error:
|
|
print('Tundra USB:',error,file=sys.stderr)
|
|
sys.exit(1)
|