ESP32: Add support for flashing using .flash.py

project-chip · Jun 29, 2021 · 2606fc6 · 2606fc6
1 parent f014dd5
commit 2606fc6
Show file tree

Hide file tree

Showing 5 changed files with 152 additions and 1 deletion.
diff --git a/examples/all-clusters-app/esp32/CMakeLists.txt b/examples/all-clusters-app/esp32/CMakeLists.txt
@@ -35,3 +35,38 @@ endif()
 project(chip-all-clusters-app)
 idf_build_set_property(CXX_COMPILE_OPTIONS "-std=gnu++14;-Os;-DLWIP_IPV6_SCOPES=0;-DCHIP_HAVE_CONFIG_H" APPEND)
 idf_build_set_property(C_COMPILE_OPTIONS "-Os;-DLWIP_IPV6_SCOPES=0" APPEND)
+
+idf_build_get_property(build_dir BUILD_DIR)
+idf_build_get_property(project_path PROJECT_DIR)
+idf_build_get_property(sdkconfig SDKCONFIG)
+idf_build_get_property(idf_path IDF_PATH)
+
+add_custom_command(OUTPUT "${build_dir}/firmware_utils.py"
+    COMMAND ${CMAKE_COMMAND} ARGS -E copy "${project_path}/third_party/connectedhomeip/scripts/flashing/firmware_utils.py" "${build_dir}/"
+    WORKING_DIRECTORY ${build_dir}
+    VERBATIM)
+
+add_custom_command(OUTPUT "${build_dir}/esp32_firmware_utils.py"
+    COMMAND ${CMAKE_COMMAND} ARGS -E copy "${project_path}/third_party/connectedhomeip/scripts/flashing/esp32_firmware_utils.py" "${build_dir}/"
+    WORKING_DIRECTORY ${build_dir}
+    VERBATIM)
+
+add_custom_command(OUTPUT "${build_dir}/${CMAKE_PROJECT_NAME}.flash.py"
+    COMMAND ${python}
+            "${project_path}/../../../scripts/flashing/gen_flashing_script.py" esp32
+            --output "${build_dir}/${CMAKE_PROJECT_NAME}.flash.py"
+            --port "$ENV{ESPPORT}"
+            --baud "$ENV{ESPBAUD}"
+            --before ${CONFIG_ESPTOOLPY_BEFORE}
+            --after ${CONFIG_ESPTOOLPY_AFTER}
+            --application "${CMAKE_PROJECT_NAME}.bin"
+            --bootloader "bootloader/bootloader.bin"
+            --partition "partition_table/partition-table.bin"
+            --use-partition-file "${build_dir}/partition_table/partition-table.bin"
+            --use-parttool ${idf_path}/components/partition_table/parttool.py
+            --use-sdkconfig ${project_path}/sdkconfig
+    WORKING_DIRECTORY ${build_dir}
+    COMMENT "To flash ${build_dir}/${CMAKE_PROJECT_NAME}.bin run ./build/${CMAKE_PROJECT_NAME}.flash.py"
+    VERBATIM)
+
+add_custom_target(flashing_script DEPENDS "${build_dir}/${CMAKE_PROJECT_NAME}.bin" "${build_dir}/${CMAKE_PROJECT_NAME}.flash.py" "${build_dir}/esp32_firmware_utils.py" "${build_dir}/firmware_utils.py")
diff --git a/examples/persistent-storage/esp32/CMakeLists.txt b/examples/persistent-storage/esp32/CMakeLists.txt
@@ -25,3 +25,38 @@ set(EXTRA_COMPONENT_DIRS
 project(persistent-storage)
 idf_build_set_property(CXX_COMPILE_OPTIONS "-std=gnu++14;-Os;-DLWIP_IPV6_SCOPES=0;-DCHIP_HAVE_CONFIG_H" APPEND)
 idf_build_set_property(C_COMPILE_OPTIONS "-Os;-DLWIP_IPV6_SCOPES=0" APPEND)
+
+idf_build_get_property(build_dir BUILD_DIR)
+idf_build_get_property(project_path PROJECT_DIR)
+idf_build_get_property(sdkconfig SDKCONFIG)
+idf_build_get_property(idf_path IDF_PATH)
+
+add_custom_command(OUTPUT "${build_dir}/firmware_utils.py"
+    COMMAND ${CMAKE_COMMAND} ARGS -E copy "${project_path}/third_party/connectedhomeip/scripts/flashing/firmware_utils.py" "${build_dir}/"
+    WORKING_DIRECTORY ${build_dir}
+    VERBATIM)
+
+add_custom_command(OUTPUT "${build_dir}/esp32_firmware_utils.py"
+    COMMAND ${CMAKE_COMMAND} ARGS -E copy "${project_path}/third_party/connectedhomeip/scripts/flashing/esp32_firmware_utils.py" "${build_dir}/"
+    WORKING_DIRECTORY ${build_dir}
+    VERBATIM)
+
+add_custom_command(OUTPUT "${build_dir}/${CMAKE_PROJECT_NAME}.flash.py"
+    COMMAND ${python}
+            "${project_path}/../../../scripts/flashing/gen_flashing_script.py" esp32
+            --output "${build_dir}/${CMAKE_PROJECT_NAME}.flash.py"
+            --port "$ENV{ESPPORT}"
+            --baud "$ENV{ESPBAUD}"
+            --before ${CONFIG_ESPTOOLPY_BEFORE}
+            --after ${CONFIG_ESPTOOLPY_AFTER}
+            --application "${CMAKE_PROJECT_NAME}.bin"
+            --bootloader "bootloader/bootloader.bin"
+            --partition "partition_table/partition-table.bin"
+            --use-partition-file "${build_dir}/partition_table/partition-table.bin"
+            --use-parttool ${idf_path}/components/partition_table/parttool.py
+            --use-sdkconfig ${project_path}/sdkconfig
+    WORKING_DIRECTORY ${build_dir}
+    COMMENT "To flash ${build_dir}/${CMAKE_PROJECT_NAME}.bin run ./build/${CMAKE_PROJECT_NAME}.flash.py"
+    VERBATIM)
+
+add_custom_target(flashing_script DEPENDS "${build_dir}/${CMAKE_PROJECT_NAME}.bin" "${build_dir}/${CMAKE_PROJECT_NAME}.flash.py" "${build_dir}/esp32_firmware_utils.py" "${build_dir}/firmware_utils.py")
diff --git a/examples/pigweed-app/esp32/CMakeLists.txt b/examples/pigweed-app/esp32/CMakeLists.txt
@@ -26,3 +26,48 @@ set(EXTRA_COMPONENT_DIRS
 project(chip-pigweed-app)
 idf_build_set_property(CXX_COMPILE_OPTIONS "-std=gnu++17;-Os;-DLWIP_IPV6_SCOPES=0;-DCHIP_HAVE_CONFIG_H" APPEND)
 idf_build_set_property(C_COMPILE_OPTIONS "-Os;-DLWIP_IPV6_SCOPES=0" APPEND)
+
+idf_build_get_property(build_dir BUILD_DIR)
+idf_build_get_property(project_path PROJECT_DIR)
+idf_build_get_property(sdkconfig SDKCONFIG)
+idf_build_get_property(idf_path IDF_PATH)
+
+add_custom_command(OUTPUT "${build_dir}/firmware_utils.py"
+    COMMAND ${CMAKE_COMMAND} ARGS -E copy "${project_path}/third_party/connectedhomeip/scripts/flashing/firmware_utils.py" "${build_dir}/"
+    WORKING_DIRECTORY ${build_dir}
+    VERBATIM)
+
+add_custom_command(OUTPUT "${build_dir}/esp32_firmware_utils.py"
+    COMMAND ${CMAKE_COMMAND} ARGS -E copy "${project_path}/third_party/connectedhomeip/scripts/flashing/esp32_firmware_utils.py" "${build_dir}/"
+    WORKING_DIRECTORY ${build_dir}
+    VERBATIM)
+
+add_custom_command(OUTPUT "${build_dir}/echo_test.py"
+    COMMAND ${CMAKE_COMMAND} ARGS -E copy "${project_path}/../mobly_tests/echo_test.py" "${build_dir}/"
+    WORKING_DIRECTORY ${build_dir}
+    VERBATIM)
+
+add_custom_command(OUTPUT "${build_dir}/echo_test_config.yml"
+    COMMAND ${CMAKE_COMMAND} ARGS -E copy "${project_path}/echo_test_config.yml" "${build_dir}/"
+    WORKING_DIRECTORY ${build_dir}
+    VERBATIM)
+
+add_custom_command(OUTPUT "${build_dir}/${CMAKE_PROJECT_NAME}.flash.py"
+    COMMAND ${python}
+            "${project_path}/../../../scripts/flashing/gen_flashing_script.py" esp32
+            --output "${build_dir}/${CMAKE_PROJECT_NAME}.flash.py"
+            --port "$ENV{ESPPORT}"
+            --baud "$ENV{ESPBAUD}"
+            --before ${CONFIG_ESPTOOLPY_BEFORE}
+            --after ${CONFIG_ESPTOOLPY_AFTER}
+            --application "${CMAKE_PROJECT_NAME}.bin"
+            --bootloader "bootloader/bootloader.bin"
+            --partition "partition_table/partition-table.bin"
+            --use-partition-file "${build_dir}/partition_table/partition-table.bin"
+            --use-parttool ${idf_path}/components/partition_table/parttool.py
+            --use-sdkconfig ${project_path}/sdkconfig
+    WORKING_DIRECTORY ${build_dir}
+    COMMENT "To flash ${build_dir}/${CMAKE_PROJECT_NAME}.bin run ./build/${CMAKE_PROJECT_NAME}.flash.py"
+    VERBATIM)
+
+add_custom_target(flashing_script DEPENDS "${build_dir}/${CMAKE_PROJECT_NAME}.bin" "${build_dir}/${CMAKE_PROJECT_NAME}.flash.py" "${build_dir}/esp32_firmware_utils.py" "${build_dir}/firmware_utils.py" "${build_dir}/echo_test.py" "${build_dir}/echo_test_config.yml")
diff --git a/examples/temperature-measurement-app/esp32/CMakeLists.txt b/examples/temperature-measurement-app/esp32/CMakeLists.txt
@@ -36,3 +36,38 @@ set(EXTRA_COMPONENT_DIRS
 project(chip-temperature-measurement-app)
 idf_build_set_property(CXX_COMPILE_OPTIONS "-std=gnu++14;-Os;-DLWIP_IPV6_SCOPES=0;-DCHIP_HAVE_CONFIG_H" APPEND)
 idf_build_set_property(C_COMPILE_OPTIONS "-Os;-DLWIP_IPV6_SCOPES=0" APPEND)
+
+idf_build_get_property(build_dir BUILD_DIR)
+idf_build_get_property(project_path PROJECT_DIR)
+idf_build_get_property(sdkconfig SDKCONFIG)
+idf_build_get_property(idf_path IDF_PATH)
+
+add_custom_command(OUTPUT "${build_dir}/firmware_utils.py"
+    COMMAND ${CMAKE_COMMAND} ARGS -E copy "${project_path}/third_party/connectedhomeip/scripts/flashing/firmware_utils.py" "${build_dir}/"
+    WORKING_DIRECTORY ${build_dir}
+    VERBATIM)
+
+add_custom_command(OUTPUT "${build_dir}/esp32_firmware_utils.py"
+    COMMAND ${CMAKE_COMMAND} ARGS -E copy "${project_path}/third_party/connectedhomeip/scripts/flashing/esp32_firmware_utils.py" "${build_dir}/"
+    WORKING_DIRECTORY ${build_dir}
+    VERBATIM)
+
+add_custom_command(OUTPUT "${build_dir}/${CMAKE_PROJECT_NAME}.flash.py"
+    COMMAND ${python}
+            "${project_path}/../../../scripts/flashing/gen_flashing_script.py" esp32
+            --output "${build_dir}/${CMAKE_PROJECT_NAME}.flash.py"
+            --port "$ENV{ESPPORT}"
+            --baud "$ENV{ESPBAUD}"
+            --before ${CONFIG_ESPTOOLPY_BEFORE}
+            --after ${CONFIG_ESPTOOLPY_AFTER}
+            --application "${CMAKE_PROJECT_NAME}.bin"
+            --bootloader "bootloader/bootloader.bin"
+            --partition "partition_table/partition-table.bin"
+            --use-partition-file "${build_dir}/partition_table/partition-table.bin"
+            --use-parttool ${idf_path}/components/partition_table/parttool.py
+            --use-sdkconfig ${project_path}/sdkconfig
+    WORKING_DIRECTORY ${build_dir}
+    COMMENT "To flash ${build_dir}/${CMAKE_PROJECT_NAME}.bin run ./build/${CMAKE_PROJECT_NAME}.flash.py"
+    VERBATIM)
+
+add_custom_target(flashing_script DEPENDS "${build_dir}/${CMAKE_PROJECT_NAME}.bin" "${build_dir}/${CMAKE_PROJECT_NAME}.flash.py" "${build_dir}/esp32_firmware_utils.py" "${build_dir}/firmware_utils.py")
diff --git a/scripts/flashing/esp32_firmware_utils.py b/scripts/flashing/esp32_firmware_utils.py
@@ -227,10 +227,11 @@
         },
         'bootloader_offset': {
             'help': 'Bootloader offset',
-            'default': '0x1000',
+            'default': None,
             'argparse': {
                 'metavar': 'OFFSET'
             },
+            'sdkconfig': 'CONFIG_BOOTLOADER_OFFSET_IN_FLASH',
         },
         'partition': {
             'help': 'Partition table image',