init

1 files changed, 50 insertions, 0 deletions

diff --git a/src/write_eeprom.py b/src/write_eeprom.py
new file mode 100755
index 0000000..9b76e51
--- /dev/null
+++ b/src/write_eeprom.py
@@ -0,0 +1,50 @@
+#!/usr/bin/env python3
+import serial
+import argparse
+import os
+
+#we send syn, recv ack
+SYN_MAGIC = 0xdeafbeef 
+ACK_MAGIC = 0xf00dd00d
+
+EEPROM_PAGESIZE = 64
+EEPROM_FILE = "./compiled_eeprom"
+
+
+parser = argparse.ArgumentParser(description="sends binary file over serial so the avr can write it to eeprom")
+parser.add_argument('port', type=str)
+parser.add_argument('baud', type=int)
+args = parser.parse_args()
+
+eeprom_file = open("./compiled_eeprom", "rb")
+eeprom_filesize = os.fstat(eeprom_file.fileno()).st_size
+
+s = serial.Serial(args.port, args.baud, timeout = 0)
+s.write(SYN_MAGIC)
+
+def handle_writer_error():
+    if write_status == 1: # verification failed
+        print("Verification failed on page {}!".format(page))
+    elif write_status == 0xff:
+        print("Unknown error writing page {}!".format(page))
+
+
+while not s.read(4) == ACK_MAGIC:
+    pass
+
+print("Device detected, writing eeprom...")
+for page in range(0, int(eeprom_filesize / EEPROM_PAGESIZE)):
+    s.write(eeprom_file.read(EEPROM_PAGESIZE))
+    write_status = s.read(1)
+    if not write_status == 0:
+        handle_writer_error(write_status)
+
+if eeprom_file.tell() < eeprom_filesize:
+    s.write(eeprom_file.read())
+    [s.write(0) for b in range(0, eeprom_filesize % EEPROM_PAGESIZE)]
+    write_status = s.read(1)
+    if not write_status == 0:
+        handle_writer_error(write_status)
+
+
+print("Done writing to EEPROM! You are still alone!")