scripts: zephyr_flash_debug: flash like esp32.sh
Add support for flashing targets compatible with esp32.sh.
Only tested by comparing commands that would be run.
Signed-off-by: Marti Bolivar <marti.bolivar@linaro.org>
diff --git a/scripts/support/zephyr_flash_debug.py b/scripts/support/zephyr_flash_debug.py
index f21fdf2..19692f4 100755
--- a/scripts/support/zephyr_flash_debug.py
+++ b/scripts/support/zephyr_flash_debug.py
@@ -191,6 +191,80 @@
print('Now reset your board again to switch back to runtime mode.')
+class Esp32BinaryFlasher(ZephyrBinaryFlasher):
+ '''Flasher front-end for espidf.'''
+
+ def __init__(self, elf, device, baud=921600, flash_size='detect',
+ flash_freq='40m', flash_mode='dio', espdif='espidf',
+ debug=False):
+ super(Esp32BinaryFlasher, self).__init__(debug=debug)
+ self.elf = elf
+ self.device = device
+ self.baud = baud
+ self.flash_size = flash_size
+ self.flash_freq = flash_freq
+ self.flash_mode = flash_mode
+ self.espdif = espdif
+
+ def replaces_shell_script(shell_script):
+ return shell_script == 'esp32.sh'
+
+ def create_from_env(debug):
+ '''Create flasher from environment.
+
+ Required:
+
+ - O: build output directory
+ - KERNEL_ELF_NAME: name of kernel binary in ELF format
+
+ Optional:
+
+ - ESP_DEVICE: serial port to flash, default /dev/ttyUSB0
+ - ESP_BAUD_RATE: serial baud rate, default 921600
+ - ESP_FLASH_SIZE: flash size, default 'detect'
+ - ESP_FLASH_FREQ: flash frequency, default '40m'
+ - ESP_FLASH_MODE: flash mode, default 'dio'
+ - ESP_TOOL: complete path to espdif, or set to 'espidf' to look for it
+ in $ESP_IDF_PATH/components/esptool_py/esptool/esptool.py
+ '''
+ elf = path.join(get_env_or_bail('O'),
+ get_env_or_bail('KERNEL_ELF_NAME'))
+
+ # TODO add sane device defaults on other platforms than Linux.
+ device = os.environ.get('ESP_DEVICE', '/dev/ttyUSB0')
+ baud = os.environ.get('ESP_BAUD_RATE', '921600')
+ flash_size = os.environ.get('ESP_FLASH_SIZE', 'detect')
+ flash_freq = os.environ.get('ESP_FLASH_FREQ', '40m')
+ flash_mode = os.environ.get('ESP_FLASH_MODE', 'dio')
+ espdif = os.environ.get('ESP_TOOL', 'espidf')
+
+ if espdif == 'espdif':
+ idf_path = get_env_or_bail('ESP_IDF_PATH')
+ espdif = path.join(idf_path, 'components', 'esptool_py', 'esptool',
+ 'esptool.py')
+
+ return Esp32BinaryFlasher(elf, device, baud=baud,
+ flash_size=flash_size, flash_freq=flash_freq,
+ flash_mode=flash_mode, espdif=espdif,
+ debug=debug)
+
+ def flash(self, **kwargs):
+ bin_name = path.splitext(self.elf)[0] + path.extsep + 'bin'
+ cmd_convert = [self.espdif, '--chip', 'esp32', 'elf2image', self.elf]
+ cmd_flash = [self.espdif, '--chip', 'esp32', '--port', self.device,
+ '--baud', self.baud, '--before', 'default_reset',
+ '--after', 'hard_reset', 'write_flash', '-u',
+ '--flash_mode', self.flash_mode,
+ '--flash_freq', self.flash_freq,
+ '--flash_size', self.flash_size,
+ '0x1000', bin_name]
+
+ print("Converting ELF to BIN")
+ check_call(cmd_convert, self.debug)
+
+ print("Flashing ESP32 on {} ({}bps)".format(self.device, self.baud))
+ check_call(cmd_flash, self.debug)
+
class PyOcdBinaryFlasher(ZephyrBinaryFlasher):
'''Flasher front-end for pyocd-flashtool.'''
