3diy
/
klipper
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

update_chitu: Rename script and minor changes 

Signed-off-by: Kevin O'Connor <kevin@koconnor.net>
This commit is contained in:
Kevin O'Connor 2020-08-30 13:11:34 -04:00
parent 1d201c3592
commit 12529ef6cd
2 changed files with 24 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
config/printer-tronxy-x5sa-v6-2019.cfg
View File

@ -7,7 +7,7 @@
# at 0x8008800 (Chitu v6 Bootloader). Uncheck USB, and leave default # at 0x8008800 (Chitu v6 Bootloader). Uncheck USB, and leave default
# serial settings. # serial settings.
# #
# Use "./chitu_crypt.py out/klipper.bin" to generate update.cbz. # Use "./update_chitu.py out/klipper.bin update.cbz" to generate update.cbz.
# Put `update.cbz` onto SD card, and reboot the printer. # Put `update.cbz` onto SD card, and reboot the printer.
# It will be automatically installed, and you will be able to update it this way. # It will be automatically installed, and you will be able to update it this way.

42
scripts/chitu_crypt.py → scripts/update_chitu.py
View File

@ -1,5 +1,5 @@
#!/usr/bin/env python2 #!/usr/bin/env python2
# Encrypts STM32 firmwares to be flashable from SD card by Chitu motherboards. # Encodes STM32 firmwares to be flashable from SD card by Chitu motherboards.
# Relocate firmware to 0x08008800! # Relocate firmware to 0x08008800!
# Copied from Marlin and modified. # Copied from Marlin and modified.
@ -64,7 +64,7 @@ def xor_block(r0, r1, block_number, block_size, file_key):
loop_counter = loop_counter + 1 loop_counter = loop_counter + 1
def encrypt_file(input, output_file, file_length): def encode_file(input, output_file, file_length):
input_file = bytearray(input.read()) input_file = bytearray(input.read())
block_size = 0x800 block_size = 0x800
key_length = 0x18 key_length = 0x18
@ -81,7 +81,7 @@ def encrypt_file(input, output_file, file_length):
# write the file header # write the file header
output_file.write(struct.pack(">I", 0x443D2D3F)) output_file.write(struct.pack(">I", 0x443D2D3F))
# encrypt the contents using a known file header key # encode the contents using a known file header key
# write the file_key # write the file_key
output_file.write(struct.pack("<I", file_key)) output_file.write(struct.pack("<I", file_key))
@ -103,28 +103,32 @@ def encrypt_file(input, output_file, file_length):
# write CRC # write CRC
output_file.write(struct.pack("<I", xor_crc)) output_file.write(struct.pack("<I", xor_crc))
# finally, append the encrypted results. # finally, append the encoded results.
output_file.write(input_file) output_file.write(input_file)
return return
if len(sys.argv) == 1: def main():
print("Usage: chitu_crypt [firmware.bin]") if len(sys.argv) != 3:
exit(1) print("Usage: update_chitu <input_file> <output_file>")
exit(1)
fw = sys.argv[-1] fw, output = sys.argv[1:]
if not os.path.isfile(fw): if not os.path.isfile(fw):
print("Usage: chitu_crypt [firmware.bin]") print("Usage: update_chitu <input_file> <output_file>")
print("Firmware file", fw, "does not exist") print("Firmware file", fw, "does not exist")
exit(1) exit(1)
firmware = open(fw, "rb") firmware = open(fw, "rb")
update = open('./update.cbd', "wb") update = open(output, "wb")
length = os.path.getsize(fw) length = os.path.getsize(fw)
encrypt_file(firmware, update, length) encode_file(firmware, update, length)
firmware.close() firmware.close()
update.close() update.close()
print("Encryption complete. Firmware is written to update.cbd.") print("Encoding complete.")
if __name__ == '__main__':
main()