220 lines
7.9 KiB
Python
220 lines
7.9 KiB
Python
# Copyright (c) 2015, SICS Swedish ICT
|
|
# All rights reserved.
|
|
#
|
|
# Redistribution and use in source and binary forms, with or without
|
|
# modification, are permitted provided that the following conditions
|
|
# are met:
|
|
# 1. Redistributions of source code must retain the above copyright
|
|
# notice, this list of conditions and the following disclaimer.
|
|
# 2. Redistributions in binary form must reproduce the above copyright
|
|
# notice, this list of conditions and the following disclaimer in the
|
|
# documentation and/or other materials provided with the distribution.
|
|
# 3. Neither the name of the Institute nor the names of its contributors
|
|
# may be used to endorse or promote products derived from this software
|
|
# without specific prior written permission.
|
|
#
|
|
# THIS SOFTWARE IS PROVIDED BY THE INSTITUTE AND CONTRIBUTORS ``AS IS'' AND
|
|
# ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
|
# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
|
|
# ARE DISCLAIMED. IN NO EVENT SHALL THE INSTITUTE OR CONTRIBUTORS BE LIABLE
|
|
# FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
|
|
# DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
|
|
# OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
|
|
# HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
|
|
# LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
|
|
# OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
|
|
# SUCH DAMAGE.
|
|
#
|
|
# This file is part of the Contiki operating system.
|
|
#
|
|
# Author(s):
|
|
# Janis Judvaitis
|
|
# Atis Elsts <atis.elsts@sics.se>
|
|
|
|
import os, glob, re
|
|
import multiprocessing, subprocess
|
|
|
|
FTDI_VENDOR_ID = "0403"
|
|
FTDI_PRODUCT_ID = "6001"
|
|
|
|
doPrintVendorID = False
|
|
|
|
def read_line(filename):
|
|
"""helper function to read a single line from a file"""
|
|
line = "Unknown"
|
|
try:
|
|
with open(filename) as f:
|
|
line = f.readline().strip()
|
|
finally:
|
|
return line
|
|
|
|
# try to extract descriptions from sysfs. this was done by experimenting,
|
|
# no guarantee that it works for all devices or in the future...
|
|
|
|
def usb_sysfs_hw_string(sysfs_path):
|
|
"""given a path to a usb device in sysfs, return a string describing it"""
|
|
snr = read_line(sysfs_path + '/serial')
|
|
if snr:
|
|
snr_txt = '%s' % (snr,)
|
|
else:
|
|
snr_txt = ''
|
|
if doPrintVendorID:
|
|
return 'USB VID:PID=%s:%s SNR=%s' % (
|
|
read_line(sysfs_path + '/idVendor'),
|
|
read_line(sysfs_path + '/idProduct'),
|
|
snr_txt
|
|
)
|
|
else:
|
|
return snr_txt
|
|
|
|
def usb_string(sysfs_path):
|
|
# Get dir name in /sys/bus/usb/drivers/usb for current usb dev
|
|
dev = os.path.basename(os.path.realpath(sysfs_path))
|
|
dev_dir = os.path.join("/sys/bus/usb/drivers/usb", dev)
|
|
|
|
try:
|
|
# Go to usb dev directory
|
|
product = read_line(os.path.join(dev_dir, "product"))
|
|
manufacturer = read_line(os.path.join(dev_dir, "manufacturer"))
|
|
result = product + " by " + manufacturer
|
|
except:
|
|
result = "Unknown device"
|
|
|
|
return result
|
|
|
|
def describe(device):
|
|
"""
|
|
Get a human readable description.
|
|
For USB-Serial devices try to run lsusb to get a human readable description.
|
|
For USB-CDC devices read the description from sysfs.
|
|
"""
|
|
base = os.path.basename(device)
|
|
# USB-Serial devices
|
|
sys_dev_path = '/sys/class/tty/%s/device/driver/%s' % (base, base)
|
|
if os.path.exists(sys_dev_path):
|
|
sys_usb = os.path.dirname(os.path.dirname(os.path.realpath(sys_dev_path)))
|
|
return usb_string(sys_usb)
|
|
|
|
# Arduino wants special handling
|
|
sys_dev_path = '/sys/class/tty/%s/device/driver/' % (base)
|
|
for x in os.listdir(sys_dev_path):
|
|
# Driver directory's name contains device ID in /sys/bus/usb/drivers/usb
|
|
temp = x.split(":")
|
|
if len(temp) == 2:
|
|
# No Arduino adds, need to save space!
|
|
return usb_string(temp[0]).replace("(www.arduino.cc)", "").strip()
|
|
|
|
# USB-CDC devices
|
|
sys_dev_path = '/sys/class/tty/%s/device/interface' % (base,)
|
|
if os.path.exists(sys_dev_path):
|
|
return read_line(sys_dev_path)
|
|
|
|
return base
|
|
|
|
def hwinfo(device):
|
|
"""Try to get a HW identification using sysfs"""
|
|
base = os.path.basename(device)
|
|
if os.path.exists('/sys/class/tty/%s/device' % (base,)):
|
|
# PCI based devices
|
|
sys_id_path = '/sys/class/tty/%s/device/id' % (base,)
|
|
if os.path.exists(sys_id_path):
|
|
return read_line(sys_id_path)
|
|
# USB-Serial devices
|
|
sys_dev_path = '/sys/class/tty/%s/device/driver/%s' % (base, base)
|
|
if os.path.exists(sys_dev_path):
|
|
sys_usb = os.path.dirname(os.path.dirname(os.path.realpath(sys_dev_path)))
|
|
return usb_sysfs_hw_string(sys_usb)
|
|
# USB-CDC devices
|
|
if base.startswith('ttyACM'):
|
|
sys_dev_path = '/sys/class/tty/%s/device' % (base,)
|
|
if os.path.exists(sys_dev_path):
|
|
return usb_sysfs_hw_string(sys_dev_path + '/..')
|
|
return 'n/a' # XXX directly remove these from the list?
|
|
|
|
#######################################
|
|
|
|
def is_nxp_mote(device):
|
|
base = os.path.basename(device)
|
|
# USB-Serial device?
|
|
sys_dev_path = '/sys/class/tty/%s/device/driver/%s' % (base, base)
|
|
if not os.path.exists(sys_dev_path):
|
|
return False
|
|
|
|
path_usb = os.path.dirname(os.path.dirname(os.path.realpath(sys_dev_path)))
|
|
|
|
dev = os.path.basename(os.path.realpath(path_usb))
|
|
dev_dir = os.path.join("/sys/bus/usb/drivers/usb", dev)
|
|
|
|
try:
|
|
idProduct = read_line(os.path.join(dev_dir, "idProduct"))
|
|
idVendor = read_line(os.path.join(dev_dir, "idVendor"))
|
|
if idVendor != FTDI_VENDOR_ID or idProduct != FTDI_PRODUCT_ID:
|
|
return False
|
|
product = read_line(os.path.join(dev_dir, "product"))
|
|
manufacturer = read_line(os.path.join(dev_dir, "manufacturer"))
|
|
if manufacturer != "NXP":
|
|
return False
|
|
except:
|
|
return False
|
|
return True
|
|
|
|
|
|
def list_motes(flash_programmer):
|
|
devices = glob.glob('/dev/ttyUSB*')# + glob.glob('/dev/ttyACM*')
|
|
return [d for d in devices if is_nxp_mote(d)]
|
|
|
|
|
|
def extract_information(port, stdout_value):
|
|
mac_str='Unknown' # not supported on Linux
|
|
info='' # not properly supported on Linux
|
|
is_program_success=''
|
|
|
|
info = describe(port) + ", SerialID: " + hwinfo(port)
|
|
|
|
res = re.compile('(Success)').search(stdout_value)
|
|
if res:
|
|
is_program_success = str(res.group(1))
|
|
else:
|
|
res = re.compile('(Error .*)\n').search(stdout_value)
|
|
if res:
|
|
is_program_success = str(res.group(1))
|
|
|
|
return [mac_str, info, is_program_success]
|
|
|
|
|
|
def program_motes(flash_programmer, motes, firmware_file):
|
|
for m in motes:
|
|
cmd = [flash_programmer, '-v', '-s', m, '-I', '38400', '-P', '1000000', '-f', firmware_file]
|
|
cmd = " ".join(cmd)
|
|
proc = subprocess.Popen(cmd, shell=True, stdin=subprocess.PIPE, stdout=subprocess.PIPE, stderr=subprocess.PIPE,)
|
|
stdout_value, stderr_value = proc.communicate('through stdin to stdout')
|
|
[mac_str, info, is_program_success] = extract_information(m, stdout_value)
|
|
print m, is_program_success
|
|
|
|
errors = (stderr_value)
|
|
if errors != '':
|
|
print 'Errors:', errors
|
|
|
|
|
|
def print_info(flash_programmer, motes, do_mac_only):
|
|
if do_mac_only:
|
|
print "Listing Mac addresses (not supported on Linux):"
|
|
else:
|
|
print "Listing mote info:"
|
|
|
|
for m in motes:
|
|
[mac_str, info, is_program_success] = extract_information(m, '')
|
|
if do_mac_only:
|
|
print m, mac_str
|
|
else:
|
|
print m, '\n', info, '\n'
|
|
|
|
def serialdump(args):
|
|
port_name = args[0]
|
|
serial_dumper = args[1]
|
|
rv = subprocess.call(serial_dumper + ' -b1000000 ' + port_name, shell=True)
|
|
|
|
def serialdump_ports(flash_programmer, serial_dumper, ports):
|
|
p = multiprocessing.Pool()
|
|
p.map(serialdump, zip(ports, [serial_dumper] * len(ports)))
|
|
p.close()
|