Tasmota/tasmota/berry/modules/bootloader.be

98 lines
3.3 KiB
Plaintext
Raw Normal View History

2022-08-28 08:39:24 +01:00
#
# Flash bootloader from URL
#
class bootloader
static var _addr = [0x1000, 0x0000] # possible addresses for bootloader
static var _sign = bytes('E9') # signature of the bootloader
static var _addr_high = 0x8000 # address of next partition after bootloader
# get the bootloader address, 0x1000 for Xtensa based, 0x0000 for RISC-V based (but might have some exception)
# we prefer to probed what's already in place rather than manage a hardcoded list of architectures
# (there is a low risk of collision if the address is 0x0000 and offset 0x1000 is actually E9)
def get_bootloader_address()
import flash
# let's see where we find 0xE9, trying first 0x1000 then 0x0000
for addr : self._addr
if flash.read(addr, size(self._sign)) == self._sign
return addr
end
end
return nil
end
#
# download from URL and store to `bootloader.bin`
#
def download(url)
# address to flash the bootloader
var addr = self.get_bootloader_address()
if addr == nil raise "internal_error", "can't find address for bootloader" end
var cl = webclient()
cl.begin(url)
var r = cl.GET()
if r != 200 raise "network_error", "GET returned "+str(r) end
var bl_size = cl.get_size()
if bl_size <= 8291 raise "internal_error", "wrong bootloader size "+str(bl_size) end
if bl_size > (0x8000 - addr) raise "internal_error", "bootloader is too large "+str(bl_size / 1024)+"kB" end
cl.write_file("bootloader.bin")
cl.close()
end
def flash(url)
if url != nil
self.download(url)
end
# address to flash the bootloader
var addr = self.get_bootloader_address()
if addr == nil raise "internal_error", "can't find address for bootloader" end
var bl = open("bootloader.bin", "r")
if bl.readbytes(size(self._sign)) != self._sign
raise "value_error", "the file does not contain a bootloader signature"
end
bl.seek(0) # reset to start of file
var bl_size = bl.size()
if bl_size <= 8291 raise "internal_error", "wrong bootloader size "+str(bl_size) end
if bl_size > (0x8000 - addr) raise "internal_error", "bootloader is too large "+str(bl_size / 1024)+"kB" end
print("OTA: Flashing bootloader")
# from now on there is no turning back, any failure means a bricked device
import flash
# read current value for bytes 2/3
var cur_config = flash.read(addr, 4)
flash.erase(addr, self._addr_high - addr) # erase the bootloader
var buf = bl.readbytes(0x1000) # read by chunks of 4kb
# put back signature
buf[2] = cur_config[2]
buf[3] = cur_config[3]
while size(buf) > 0
flash.write(addr, buf, true) # set flag to no-erase since we already erased it
addr += size(buf)
buf = bl.readbytes(0x1000) # read next chunk
end
bl.close()
print("OTA: Booloader flashed, please restart")
end
end
return bootloader
#-
### FLASH
import bootloader
bootloader().flash('https://raw.githubusercontent.com/espressif/arduino-esp32/master/tools/sdk/esp32/bin/bootloader_dio_40m.bin')
#bootloader().flash('https://raw.githubusercontent.com/espressif/arduino-esp32/master/tools/sdk/esp32/bin/bootloader_dout_40m.bin')
#### debug only
bl = bootloader()
print(format("0x%04X", bl.get_bootloader_address()))
2022-08-28 08:39:24 +01:00
-#