#!/usr/bin/env python
#
# Copyright (c) 2016, NORDUnet A/S All rights reserved.
#
# Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions are
# met:
# - Redistributions of source code must retain the above copyright notice,
# this list of conditions and the following disclaimer.
#
# - 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.
#
# - Neither the name of the NORDUnet 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 COPYRIGHT HOLDERS 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 COPYRIGHT
# HOLDER 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.
"""
Utility to upload new a firmware image or FPGA bitstream
"""
import os
import sys
import time
import struct
import serial
import argparse
import getpass
from binascii import crc32
FIRMWARE_CHUNK_SIZE = 4096
FPGA_CHUNK_SIZE = 4096
def parse_args():
"""
Parse the command line arguments
"""
parser = argparse.ArgumentParser(description = "File uploader",
add_help = True,
formatter_class = argparse.ArgumentDefaultsHelpFormatter,
)
parser.add_argument('-d', '--device',
dest='device',
default=os.getenv('CRYPTECH_CTY_CLIENT_SERIAL_DEVICE', '/dev/ttyUSB0'),
help='Name of management port USB serial device',
)
parser.add_argument('--fpga',
dest='fpga',
action='store_true', default=False,
help='Upload FPGA bitstream',
)
parser.add_argument('--firmware',
dest='firmware',
action='store_true', default=False,
help='Upload firmware image',
)
parser.add_argument('--bootloader',
dest='bootloader',
action='store_true', default=False,
help='Upload bootloader image',
)
# positional argument(s)
parser.add_argument('filename')
parser.add_argument('signature')
return parser.parse_args()
def _write(dst, data):
dst.write(data)
if len(data) == 4:
print("Wrote 0x{!s}".format(data.encode('hex')))
else:
print("Wrote {!r}".format(data))
def _read(dst):
res = ''
x = dst.read(1)
while not x:
x = dst.read(1)
while x:
res += x
x = dst.read(1)
print ("Read {!r}".format(res))
return res
pin = None
def _execute(dst, cmd):
global pin
_write(dst, '\r')
prompt = _read(dst)
if prompt.endswith('Username: '):
_write(dst, 'so\r')
prompt = _read(dst)
if prompt.endswith('Password: '):
if not pin:
pin = getpass.getpass('SO PIN: ')
_write(dst, pin + '\r')
prompt = _read(dst)
if not prompt.endswith(('> ', '# ')):
print('Device does not seem to be ready for a file transfer (got {!r})'.format(prompt))
return prompt
_write(dst, cmd + '\r')
response = _read(dst)
return response
def send_file(filename, signature, args, dst):
def fsize(fn):
try:
s = os.stat(fn)
except OSError as e:
print e
exit(1)
return s.st_size
size = fsize(filename)
src = open(filename, 'rb')
if args.fpga:
chunk_size = FPGA_CHUNK_SIZE
response = _execute(dst, 'fpga bitstream upload')
elif args.firmware:
chunk_size = FIRMWARE_CHUNK_SIZE
response = _execute(dst, 'firmware upload')
if 'Rebooting' in response:
response = _execute(dst, 'firmware upload')
elif args.bootloader:
chunk_size = FIRMWARE_CHUNK_SIZE
response = _execute(dst, 'bootloader upload')
if 'Access denied' in response:
print 'Access denied'
return False
if not 'OK' in response:
print('Device did not accept the upload command (got {!r})'.format(response))
return False
crc = 0
counter = 0
# 1. Write size of file (4 bytes)
_write(dst, struct.pack('<I', size))
response = _read(dst)
if not response.startswith('Send '):
print response
return False
# 2. Write file contents while calculating CRC-32
while True:
data = src.read(chunk_size)
if not data:
break
dst.write(data)
print("Wrote {!s} bytes (chunk {!s}/{!s})".format(len(data), counter, int(size / chunk_size)))
# read ACK (a counter of number of 4k chunks received)
while True:
ack_bytes = dst.read(4)
if len(ack_bytes) == 4:
break
print('ERROR: Did not receive an ACK, got {!r}'.format(ack_bytes))
dst.write('\r') # eventually get back to the CLI prompt
ack = struct.unpack('<I', ack_bytes)[0]
if ack != counter + 1:
print('ERROR: Did not receive the expected counter as ACK (got {!r}/{!r}, not {!r})'.format(ack, ack_bytes, counter))
flush = dst.read(100)
print('FLUSH data: {!r}'.format(flush))
return False
counter += 1
crc = crc32(data, crc) & 0xffffffff
src.close()
response = _read(dst)
if response.startswith('Send CRC-32'):
# 3a. Write CRC-32 (4 bytes)
_write(dst, struct.pack('<I', crc))
response = _read(dst)
print response
elif response.startswith('Send signature'):
# 3b. Write ECDSA signature
# write signature size
_write(dst, struct.pack('<I', fsize(signature)))
response = _read(dst)
if not response.startswith('Send '):
print response
return False
_write(dst, open(signature, 'rb').read())
response = _read(dst)
print response
if args.fpga:
# tell the fpga to read its new configuration
_execute(dst, 'fpga reset')
if args.fpga or args.bootloader:
# log out of the CLI
# firmware upgrade reboots, doesn't need an exit
_execute(dst, 'exit')
return True
def main(args):
dst = serial.Serial(args.device, 921600, timeout=2)
send_file(args.filename, args.signature, args, dst)
dst.close()
return True
if __name__ == '__main__':
try:
args = parse_args()
if main(args):
sys.exit(0)
sys.exit(1)
except KeyboardInterrupt:
pass
