nes-proj/tools/jn516x/motelist_lib/linux_motelist_impl.py

221 lines
7.9 KiB
Python
Raw Normal View History

2015-09-21 08:59:15 +00:00
# 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()