usbtest.py

#!/usr/bin/env python2.7
#
# Original version supplied to me (Kano/kanoi) by xiangfu
#
# Modified to allow supplying the data to send
#
# Linux usAge: ./ubstest.py /dev/ttyUSB0 0xhexcodes|string|icarus
#  OR       python ubstest.py /dev/ttyUSB0 0xhexcodes|string|icarus
#
# Windows usAge: ./ubstest.py COM1 0xhexcodes|string|icarus
#
#   sends the data sepcified to the USB device and waits
#   for a reply then displays it
#
#   the data can be:
#   0xhexcodes: e.g. 0x68656c6c6f20776f726c640a
#           would send "hello world\n"
#
#   string: e.g. sendsometext
#
#   icarus: sends 2 known block payloads for an icarus device
#       and shows the expected and actual answers if it's
#       a working V3 icarus

import sys
from serial import Serial
import binascii

if len(sys.argv) < 2:
    sys.stderr.write("usAge: " + sys.argv[0] + " device strings...\n")
    sys.stderr.write(" where device is either like /dev/ttyUSB0 or COM1\n")
    sys.stderr.write(" and strings are either '0xXXXX' or 'text'\n")
    sys.stderr.write(" if the first string is 'icarus' the rest are ignored\n")
    sys.stderr.write("  and 2 valid icarus test payloads are sent with results displayed\n")
    sys.stderr.write("\nAfter any command is sent it waits up to 30 seconds for a reply\n");
    sys.exit("Aborting")

# Open with a 30 second timeout - just to be sure
ser = Serial(sys.argv[1], 115200, 8, timeout=30)

if sys.argv[2] == "icarus":

    # This show how Icarus use the block and midstate data
    # This will produce nonce 063c5e01
#   block = "0000000120c8222d0497a7ab44a1a2c7bf39de941c9970b1dc7cdc400000079700000000e88aabe1f353238c668d8a4df9318e614c10c474f8cdf8bc5f6397b946c33d7c4e7242c31a098ea500000000000000800000000000000000000000000000000000000000000000000000000000000000000000000000000080020000"
#   midstate = "33c5bf5751ec7f7e056443b5aee3800331432c83f404d9de38b94ecbf907b92d"
    block = "00000001430afcd1af8d5628c53c9b67bc191f8af5854b77bf8af4080000021d000000000918ac11f800eb36152384ca9af8b03661f3b7d0a11271a00b27d807d33df6fd4f55ba0b1a0b350c00000000000000800000000000000000000000000000000000000000000000000000000000000000000000000000000080020000"
    midstate = "5d9f088fa64d2036bb92508f59917fe427977487bc211e100067d6d856f36348"

    rdata2  = block.decode('hex')[95:63:-1]
    rmid    = midstate.decode('hex')[::-1]
    payload = rmid + rdata2

    print("Push payload to icarus: " + binascii.hexlify(payload))
    ser.write(payload)

    b=ser.read(4)
    print("Result:(should be: 063c5e01): " + binascii.hexlify(b))

    # Just another test
    payload2 = "ce92099c5a80bb81c52990d5c0924c625fd25a535640607d5a4bdf8174e2c8d500000000000000000000000080000000000000000b290c1a42313b4f21b5bcb8"
    print("Push payload to icarus: " + payload2)
    ser.write(payload2.decode('hex'))

    b=ser.read(4)
    print("Result:(should be: 8e0b31c5): " + binascii.hexlify(b))
else:
    data = ""
    for arg in sys.argv[2::]:
        if arg[0:2:] == '0x':
            data += arg[2::].decode('hex')
        else:
            data += arg

    print("Sending: 0x" + binascii.hexlify(data))
    ser.write(data)

    # If you're expecting more than one linefeed terminated reply,
    # you'll only see the first one
    # AND with no linefeed, this will wait the full 30 seconds before returning
    print("Waiting up to 30 seconds ...")
    b=ser.readline()
    print("Result: hex 0x" + binascii.hexlify(b))

    # This could mess up the display - do it last
    print("Result: asc '" + b + "'")

ser.close()