blob: 9b76e5141fc679b18adac6e757be2117afef4445 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
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!")
|
