From a621333a4c4fcc161f076980c9147faa555de5de Mon Sep 17 00:00:00 2001 From: Paul Sokolovsky Date: Thu, 4 Aug 2016 00:19:09 +0300 Subject: [PATCH] esp8266/makeimg.py: Store firmware size as last 4 bytes of padding area. --- esp8266/makeimg.py | 16 ++++++++++------ 1 file changed, 10 insertions(+), 6 deletions(-) diff --git a/esp8266/makeimg.py b/esp8266/makeimg.py index e63f956bdd..834f778c84 100644 --- a/esp8266/makeimg.py +++ b/esp8266/makeimg.py @@ -1,4 +1,5 @@ import sys +import struct SEGS_MAX_SIZE = 0x9000 @@ -11,13 +12,16 @@ with open(sys.argv[3], 'wb') as fout: fout.write(data_flash) print('flash ', len(data_flash)) - pad = b'\xff' * (SEGS_MAX_SIZE - len(data_flash)) - fout.write(pad) - print('padding ', len(pad)) - with open(sys.argv[2], 'rb') as f: data_rom = f.read() - fout.write(data_rom) - print('irom0text', len(data_rom)) + + pad = b'\xff' * (SEGS_MAX_SIZE - len(data_flash)) + assert len(pad) >= 4 + fout.write(pad[:-4]) + fout.write(struct.pack("I", SEGS_MAX_SIZE + len(data_rom))) + print('padding ', len(pad)) + + fout.write(data_rom) + print('irom0text', len(data_rom)) print('total ', SEGS_MAX_SIZE + len(data_rom))