def act_cloudflash(args):
args = loadconfigs(args, "cloudflash")
imgl = args["image"].split(":", 1)
if len(imgl) != 2:
print "bad image descriptor, expected ns:name"
ithen = time.time()
if args["verbose"]:
print "Downloading image ",
sys.stdout.flush()
r = requests.get("http://cloud.storm.rocks/r/"+imgl[0]+"/"+imgl[1])
if args["verbose"]:
print "done (%.3f seconds)" %(time.time() - ithen)
if len(r.content) == 0:
print "Could not download image"
sys.exit(1)
with tempfile.NamedTemporaryFile() as ntf:
try:
ntf.write(r.content)
ntf.flush()
sl = sl_api.StormLoader(args.get("tty", None))
sl.enter_bootload_mode()
then = time.time()
try:
cell = SLCell.load(ntf.name)
except:
print "SDB image is corrupt"
sys.exit(1)
img = cell.get_raw_image()
sl.write_extended_irange(cell.FLASH_BASE, img)
now = time.time()
if args["verbose"]:
print "Wrote %d bytes in %.3f seconds" %(len(img), now-then)
expected_crc = sl.crc32(img)
written_crc = sl.c_crcif(cell.FLASH_BASE, len(img))
if expected_crc != written_crc:
print "CRC failure: expected 0x%04x, got 0x%04x" % (expected_crc, written_crc)
sys.exit(1)
elif args["verbose"]:
print "CRC pass"
sl.enter_payload_mode()
except sl_api.StormloaderException as e:
print "Fatal error:", e
sys.exit(1)
def act_list_assets(args):
args = loadconfigs(args, "assetlist")
cell = SLCell.load(args["sdb"])
iassets = cell.list_iassets()
eassets = cell.list_eassets()
def pr_assets_human(assets):
print " # Address Length Name"
for (i, a) in assets:
print "{:>2d} 0x{:07x} 0x{:05x} {}".format(i, a["address"], a["length"], a["name"])
def pr_assets_header(assets):
s1s = []
s2s = []
s3s = []
for (i, a) in assets:
s1s.append(("#define ASSET_{}_NAME".format(a["name"].upper()),
"\"{}\"".format(a["name"])))
s2s.append(("#define ASSET_{}_ADDR".format(a["name"].upper()),
"0x{:07x}".format(a["address"])))
s3s.append(("#define ASSET_{}_LEN".format(a["name"].upper()),
"0x{:05x}".format(a["length"])))
longest = max(len(i[0]) for i in (s1s+s2s+s3s))
for i in s1s+s2s+s3s:
print ("{:<%ds} {}" % longest).format(*i)
if len(iassets) > 0:
print "Internal flash assets"
if not args["header"]:
pr_assets_human(iassets)
else:
pr_assets_header(iassets)
else:
print "No internal flash assets"
print ""
if len(eassets) > 0:
print "External flash assets"
if not args["header"]:
pr_assets_human(eassets)
else:
pr_assets_header(eassets)
else:
print "No external flash assets"
def act_flash_assets(args):
try:
args = loadconfigs(args, "assetflash")
sl = sl_api.StormLoader(args.get("tty", None))
sl.enter_bootload_mode()
then = time.time()
cell = SLCell.load(args["sdb"])
for i, a in cell.list_iassets():
body = a["contents"]
if args["verbose"]:
print "writing %s to iflash @ 0x%x (%d : 0x%x bytes)... " %\
(a["name"],a["address"],a["length"],a["length"]),
sys.stdout.flush()
sl.write_extended_irange(a["address"], body)
if args["verbose"]:
print "ok"
for i, a in cell.list_eassets():
body = a["contents"]
if args["verbose"]:
print "writing %s to xflash @ 0x%x (%d : 0x%x bytes)... " %\
(a["name"],a["address"],a["length"],a["length"]),
sys.stdout.flush()
sl.write_extended_xrange(a["address"], body)
ecrc = sl.crc32(body)
rcrc = sl.c_crcef(a["address"], a["length"])
if ecrc != rcrc:
print "CRC FAIL! (expected 0x%08x, got 0x%08x" % (ecrc, rcrc)
sys.exit(1)
if args["verbose"]:
print "ok"
now = time.time()
if args["verbose"]:
print "%.2f seconds" % (now-then)
sl.enter_payload_mode()
except sl_api.StormloaderException as e:
print "Fatal error:", e
sys.exit(1)
def act_flash(args, ftdi_device_id=None):
try:
args = loadconfigs(args, "flash")
sl = sl_api.StormLoader(args.get("tty", None), device_id=ftdi_device_id)
sl.enter_bootload_mode()
then = time.time()
cell = SLCell.load(args["sdb"])
img = cell.get_raw_image()
sl.write_extended_irange(cell.FLASH_BASE, img)
now = time.time()
if args["verbose"]:
print "Wrote %d bytes in %.3f seconds" %(len(img), now-then)
expected_crc = sl.crc32(img)
written_crc = sl.c_crcif(cell.FLASH_BASE, len(img))
if expected_crc != written_crc:
print "CRC failure: expected 0x%04x, got 0x%04x" % (expected_crc, written_crc)
sys.exit(1)
elif args["verbose"]:
print "CRC pass"
sl.enter_payload_mode()
except sl_api.StormloaderException as e:
print "Fatal error:", e
sys.exit(1)
def act_hexdump(args):
args = loadconfigs(args, "hexdump")
cell = SLCell.load(args["sdb"])
print cell.get_hex_image()
def act_add_asset(args):
args = loadconfigs(args, "assetadd")
cell = SLCell.load(args["sdb"])
_add_asset(cell, args)
cell.save(args["sdb"])
def act_delta(args, ftdi_device_id=None):
args = loadconfigs(args, "flashdelta")
sl = sl_api.StormLoader(args.get("tty", None), device_id=ftdi_device_id)
sl.enter_bootload_mode()
newcell = SLCell.load(args["newimg"])
newimg = newcell.get_raw_image()
def full_flash():
then = time.time()
#write the whole img
sl.write_extended_irange(SLCell.FLASH_BASE, newimg)
expected_crc = sl.crc32(newimg)
written_crc = sl.c_crcif(SLCell.FLASH_BASE, len(newimg))
if expected_crc != written_crc:
print "CRC failure: expected 0x%04x, got 0x%04x" % (expected_crc, written_crc)
sys.exit(1)
elif args["verbose"]:
print "CRC pass"
sl.enter_payload_mode()
if args["verbose"]:
print "Written and verified in %.2f seconds" % (time.time() - then)
return
#We also permit an empty file to be specified as the old image
#this represents null
if not (os.path.exists(args["oldimg"]) and os.path.isfile(args["oldimg"])) or os.stat(args["oldimg"]).st_size == 0:
if args["verbose"]:
print "Old image is empty, doing full flash"
full_flash()
return
oldcell = SLCell.load(args["oldimg"])
oldimg = oldcell.get_raw_image()
oldimgcrc = sl.crc32(oldimg)
actualcrc = sl.c_crcif(sl.FLASH_FLOOR, len(oldimg))
if oldimgcrc != actualcrc:
print "Actual contents do not match expected image, this will take longer"
full_flash()
return
if len(newimg) % 512 != 0:
newimg += "\xFF"*(512 - (len(newimg)%512))
if len(oldimg) < len(newimg):
old_end = len(oldimg) &~ 511
start_address = (sl.FLASH_FLOOR + old_end)
sl.write_extended_irange(start_address, newimg[old_end:])
oldimg = oldimg[:old_end]
oldimg += newimg[old_end:]
assert len(oldimg) >= len(newimg)
def fix_difference():
for idx in xrange(len(newimg)):
if oldimg[idx] != newimg[idx]:
page_address = idx &~511
real_address = page_address + oldcell.FLASH_BASE
if args["verbose"]:
print "Changed page at 0x%08x" % real_address
sl.c_wpage(real_address, newimg[page_address:page_address+512])
# oldimg[page_address:page_address+512] = newimg[page_address:page_address+512]
newoldimg = oldimg[:page_address] + newimg[page_address:page_address+512] + \
oldimg[page_address + 512:]
return (True, newoldimg)
return (False, oldimg)
then = time.time()
for i in xrange(len(oldimg)+1):
changes, oldimg = fix_difference()
if not changes:
expected_crc = sl.crc32(newimg)
written_crc = sl.c_crcif(oldcell.FLASH_BASE, len(newimg))
if expected_crc != written_crc:
print "CRC failure: expected 0x%04x, got 0x%04x" % (expected_crc, written_crc)
sys.exit(1)
elif args["verbose"]:
print "CRC pass"
print "Written and verified in %.2f seconds" % (time.time() - then)
break
else:
raise sl.StormloaderException("Delta algorithm bug detected")
sl.enter_payload_mode()
