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()