circuitpython/ports/esp32s2/common-hal/ota/__init__.c

144 lines
5.3 KiB
C
Raw Normal View History

2020-12-08 01:00:00 -05:00
/*
* This file is part of the MicroPython project, http://micropython.org/
*
* The MIT License (MIT)
*
* Copyright (c) 2020 microDev
*
* Permission is hereby granted, free of charge, to any person obtaining a copy
* of this software and associated documentation files (the "Software"), to deal
* in the Software without restriction, including without limitation the rights
* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
* copies of the Software, and to permit persons to whom the Software is
* furnished to do so, subject to the following conditions:
*
* The above copyright notice and this permission notice shall be included in
* all copies or substantial portions of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
* THE SOFTWARE.
*/
#include "shared-bindings/ota/__init__.h"
2020-12-08 01:15:00 -05:00
#include <string.h>
#include "esp_log.h"
#include "esp_ota_ops.h"
static const esp_partition_t *update_partition = NULL;
static esp_ota_handle_t update_handle = 0;
2020-12-10 03:34:24 -05:00
static bool ota_inited = false;
2020-12-08 01:15:00 -05:00
static const char *TAG = "OTA";
static void ota_reset(void) {
update_handle = 0;
update_partition = NULL;
ota_inited = false;
}
2020-12-08 01:15:00 -05:00
static void __attribute__((noreturn)) task_fatal_error(void) {
ESP_LOGE(TAG, "Exiting task due to fatal error...");
mp_raise_RuntimeError(translate("OTA Update Failed"));
}
void common_hal_ota_flash(const void *buf, const size_t len, const int32_t offset) {
2020-12-08 01:15:00 -05:00
esp_err_t err;
2020-12-10 02:04:56 -05:00
2020-12-08 01:15:00 -05:00
const esp_partition_t *running = esp_ota_get_running_partition();
2020-12-10 02:04:56 -05:00
const esp_partition_t *last_invalid = esp_ota_get_last_invalid_partition();
2020-12-08 01:15:00 -05:00
if (update_partition == NULL) {
update_partition = esp_ota_get_next_update_partition(NULL);
2020-12-08 01:15:00 -05:00
ESP_LOGI(TAG, "Running partition type %d subtype %d (offset 0x%08x)",
running->type, running->subtype, running->address);
2020-12-08 01:15:00 -05:00
ESP_LOGI(TAG, "Writing partition type %d subtype %d (offset 0x%08x)\n",
update_partition->type, update_partition->subtype, update_partition->address);
2020-12-08 01:15:00 -05:00
assert(update_partition != NULL);
}
2020-12-08 01:15:00 -05:00
if (!ota_inited) {
if (len > sizeof(esp_image_header_t) + sizeof(esp_image_segment_header_t) + sizeof(esp_app_desc_t)) {
esp_app_desc_t new_app_info;
memcpy(&new_app_info, &((char *)buf)[sizeof(esp_image_header_t) + sizeof(esp_image_segment_header_t)], sizeof(esp_app_desc_t));
ESP_LOGI(TAG, "New firmware version: %s", new_app_info.version);
2020-12-08 01:15:00 -05:00
esp_app_desc_t running_app_info;
if (esp_ota_get_partition_description(running, &running_app_info) == ESP_OK) {
ESP_LOGI(TAG, "Running firmware version: %s", running_app_info.version);
}
2020-12-08 01:15:00 -05:00
esp_app_desc_t invalid_app_info;
if (esp_ota_get_partition_description(last_invalid, &invalid_app_info) == ESP_OK) {
ESP_LOGI(TAG, "Last invalid firmware version: %s", invalid_app_info.version);
}
2020-12-10 02:04:56 -05:00
// check new version with running version
if (memcmp(new_app_info.version, running_app_info.version, sizeof(new_app_info.version)) == 0) {
ESP_LOGW(TAG, "New version is the same as running version.");
2020-12-08 01:15:00 -05:00
task_fatal_error();
}
// check new version with last invalid partition
if (last_invalid != NULL) {
if (memcmp(new_app_info.version, invalid_app_info.version, sizeof(new_app_info.version)) == 0) {
ESP_LOGW(TAG, "New version is the same as invalid version.");
task_fatal_error();
}
}
err = esp_ota_begin(update_partition, OTA_WITH_SEQUENTIAL_WRITES, &update_handle);
if (err != ESP_OK) {
ESP_LOGE(TAG, "esp_ota_begin failed (%s)", esp_err_to_name(err));
task_fatal_error();
}
ota_inited = true;
} else {
ESP_LOGE(TAG, "received package is not fit len");
2020-12-08 01:15:00 -05:00
task_fatal_error();
}
}
if (offset == -1) {
err = esp_ota_write(update_handle, buf, len);
} else {
err = esp_ota_write_with_offset(update_handle, buf, len, offset);
}
2020-12-08 01:15:00 -05:00
if (err != ESP_OK) {
ESP_LOGE(TAG, "esp_ota_write failed (%s)", esp_err_to_name(err));
task_fatal_error();
}
2020-12-10 03:34:24 -05:00
}
void common_hal_ota_finish(void) {
if (ota_inited) {
esp_err_t err;
2020-12-08 01:15:00 -05:00
err = esp_ota_end(update_handle);
if (err != ESP_OK) {
if (err == ESP_ERR_OTA_VALIDATE_FAILED) {
ESP_LOGE(TAG, "Image validation failed, image is corrupted");
}
ESP_LOGE(TAG, "esp_ota_end failed (%s)!", esp_err_to_name(err));
task_fatal_error();
2020-12-08 01:15:00 -05:00
}
2020-12-08 01:00:00 -05:00
err = esp_ota_set_boot_partition(update_partition);
if (err != ESP_OK) {
ESP_LOGE(TAG, "esp_ota_set_boot_partition failed (%s)!", esp_err_to_name(err));
task_fatal_error();
}
ota_reset();
2020-12-08 01:15:00 -05:00
}
2020-12-08 01:00:00 -05:00
}