mirror of
https://github.com/Telecominfraproject/OpenCellular.git
synced 2026-01-03 05:29:29 +00:00
395fce39e2
This tool allows an easy commandline interface to set the USB-C mux position, as well as init and reboot. BRANCH=None BUG=b:35849284 TEST=flash, control tigertail successfully Change-Id: I8d60c215fee04de158c22edca5377c3c6cd48cf0 Reviewed-on: https://chromium-review.googlesource.com/493617 Commit-Ready: Nick Sanders <nsanders@chromium.org> Tested-by: Nick Sanders <nsanders@chromium.org> Reviewed-by: Aseda Aboagye <aaboagye@chromium.org>
146 lines
3.7 KiB
Python
146 lines
3.7 KiB
Python
# Copyright 2017 The Chromium OS Authors. All rights reserved.
|
|
# Use of this source code is governed by a BSD-style license that can be
|
|
# found in the LICENSE file.
|
|
|
|
"""Utilities for using lightweight console functions."""
|
|
|
|
import datetime
|
|
import errno
|
|
import os
|
|
import re
|
|
import subprocess
|
|
import sys
|
|
import time
|
|
|
|
import pty_driver
|
|
import stm32uart
|
|
|
|
|
|
class TinyServoError(Exception):
|
|
"""Exceptions."""
|
|
|
|
|
|
def log(output):
|
|
"""Print output to console, logfiles can be added here.
|
|
|
|
Args:
|
|
output: string to output.
|
|
"""
|
|
sys.stdout.write(output)
|
|
sys.stdout.write('\n')
|
|
sys.stdout.flush()
|
|
|
|
def check_usb(vidpid):
|
|
"""Check if |vidpid| is present on the system's USB.
|
|
|
|
Args:
|
|
vidpid: string representation of the usb vid:pid, eg. '18d1:2001'
|
|
|
|
Returns: True if found, Flase, otherwise.
|
|
"""
|
|
if subprocess.call(['lsusb', '-d', vidpid], stdout=open('/dev/null', 'w')):
|
|
return False
|
|
return True
|
|
|
|
def check_usb_sn(vidpid):
|
|
"""Return the serial number
|
|
|
|
Return the serial number of the first USB device with VID:PID vidpid,
|
|
or None if no device is found. This will not work well with two of
|
|
the same device attached.
|
|
|
|
Args:
|
|
vidpid: string representation of the usb vid:pid, eg. '18d1:2001'
|
|
|
|
Returns: string serial number if found, None otherwise.
|
|
"""
|
|
output = subprocess.check_output(['lsusb', '-v', '-d', vidpid])
|
|
m = re.search(r'^\s*iSerial\s+(.*)$', output, flags=re.M)
|
|
if m:
|
|
return m.group(1)
|
|
|
|
return None
|
|
|
|
def wait_for_usb_remove(vidpid, timeout=None):
|
|
"""Wait for USB device with vidpid to be removed.
|
|
|
|
Wrapper for wait_for_usb below
|
|
"""
|
|
wait_for_usb(vidpid, timeout=timeout, desiredpresence=False)
|
|
|
|
def wait_for_usb(vidpid, timeout=None, desiredpresence=True):
|
|
"""Wait for usb device with vidpid to be present/absent.
|
|
|
|
Args:
|
|
vidpid: string representation of the usb vid:pid, eg. '18d1:2001'
|
|
timeout: timeout in seconds, None for no timeout.
|
|
desiredpresence: True for present, False for not present.
|
|
|
|
Raises:
|
|
TinyServoError: on timeout.
|
|
"""
|
|
if timeout:
|
|
finish = datetime.datetime.now() + datetime.timedelta(seconds=timeout)
|
|
while check_usb(vidpid) != desiredpresence:
|
|
time.sleep(.1)
|
|
if timeout:
|
|
if datetime.datetime.now() > finish:
|
|
raise TinyServoError('Timeout', 'Timeout waiting for USB %s' % vidpid)
|
|
|
|
def do_serialno(serialno, pty):
|
|
"""Set serialnumber 'serialno' via ec console 'pty'.
|
|
|
|
Commands are:
|
|
# > serialno set 1234
|
|
# Saving serial number
|
|
# Serial number: 1234
|
|
|
|
Args:
|
|
serialno: string serial number to set.
|
|
pty: tinyservo console to send commands.
|
|
|
|
Raises:
|
|
TinyServoError: on failure to set.
|
|
ptyError: on command interface error.
|
|
"""
|
|
cmd = 'serialno set %s' % serialno
|
|
regex = 'Serial number: (.*)$'
|
|
|
|
results = pty._issue_cmd_get_results(cmd, [regex])[0]
|
|
sn = results[1].strip().strip('\n\r')
|
|
|
|
if sn == serialno:
|
|
log('Success !')
|
|
log('Serial set to %s' % sn)
|
|
else:
|
|
log('Serial number set to %s but saved as %s.' % (serialno, sn))
|
|
raise TinyServoError(
|
|
'Serial Number',
|
|
'Serial number set to %s but saved as %s.' % (serialno, sn))
|
|
|
|
def setup_tinyservod(vidpid, interface, serialno=None):
|
|
"""Set up a pty
|
|
|
|
Set up a pty to the ec console in order
|
|
to send commands. Returns a pty_driver object.
|
|
|
|
Args:
|
|
vidpid: string vidpid of device to access.
|
|
interface: not used.
|
|
serialno: string serial no of device requested, optional.
|
|
|
|
Returns: pty object
|
|
|
|
Raises:
|
|
UsbError, SusbError: on device not found
|
|
"""
|
|
vidstr, pidstr = vidpid.split(':')
|
|
vid = int(vidstr, 16)
|
|
pid = int(pidstr, 16)
|
|
suart = stm32uart.Suart(vendor=vid, product=pid,
|
|
interface=interface, serialname=serialno)
|
|
suart.run()
|
|
pty = pty_driver.ptyDriver(suart, [])
|
|
|
|
return pty
|