#include "wifi_ota.h" #include "freertos/FreeRTOS.h" #include "freertos/task.h" #include "esp_system.h" #include "esp_event.h" #include "esp_log.h" #include "esp_ota_ops.h" #include "esp_http_client.h" #include "esp_https_ota.h" #include "string.h" #include "esp_crt_bundle.h" #include "nvs.h" #include "nvs_flash.h" #include #include "esp_wifi.h" #include "bootscreen.h" #define HASH_LEN 32 static const char *TAG = "OTA update"; extern const uint8_t server_cert_pem_start[] asm("_binary_isrgrootx1_pem_start"); extern const uint8_t server_cert_pem_end[] asm("_binary_isrgrootx1_pem_end"); esp_err_t _http_event_handler(esp_http_client_event_t *evt) { switch (evt->event_id) { case HTTP_EVENT_ERROR: ESP_LOGD(TAG, "HTTP_EVENT_ERROR"); break; case HTTP_EVENT_ON_CONNECTED: ESP_LOGD(TAG, "HTTP_EVENT_ON_CONNECTED"); break; case HTTP_EVENT_HEADERS_SENT: ESP_LOGD(TAG, "HTTP_EVENT_HEADERS_SENT"); break; case HTTP_EVENT_ON_HEADER: ESP_LOGD(TAG, "HTTP_EVENT_ON_HEADER, key=%s, value=%s", evt->header_key, evt->header_value); break; case HTTP_EVENT_ON_DATA: ESP_LOGD(TAG, "HTTP_EVENT_ON_DATA, len=%d", evt->data_len); break; case HTTP_EVENT_ON_FINISH: ESP_LOGD(TAG, "HTTP_EVENT_ON_FINISH"); break; case HTTP_EVENT_DISCONNECTED: ESP_LOGD(TAG, "HTTP_EVENT_DISCONNECTED"); break; } return ESP_OK; } static esp_err_t validate_image_header(esp_app_desc_t *new_app_info) { if (new_app_info == NULL) { return ESP_ERR_INVALID_ARG; } const esp_partition_t *running = esp_ota_get_running_partition(); 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); } /* if (memcmp(new_app_info->version, running_app_info.version, sizeof(new_app_info->version)) == 0) { ESP_LOGW(TAG, "Current running version is the same as a new. We will not continue the update."); return ESP_FAIL; } */ return ESP_OK; } static esp_err_t _http_client_init_cb(esp_http_client_handle_t http_client) { esp_err_t err = ESP_OK; /* Uncomment to add custom headers to HTTP request */ // err = esp_http_client_set_header(http_client, "Custom-Header", "Value"); return err; } static void print_sha256(const uint8_t *image_hash, const char *label) { char hash_print[HASH_LEN * 2 + 1]; hash_print[HASH_LEN * 2] = 0; for (int i = 0; i < HASH_LEN; ++i) { sprintf(&hash_print[i * 2], "%02x", image_hash[i]); } ESP_LOGI(TAG, "%s %s", label, hash_print); } static void get_sha256_of_partitions(void) { uint8_t sha_256[HASH_LEN] = { 0 }; esp_partition_t partition; // get sha256 digest for bootloader partition.address = ESP_BOOTLOADER_OFFSET; partition.size = ESP_PARTITION_TABLE_OFFSET; partition.type = ESP_PARTITION_TYPE_APP; esp_partition_get_sha256(&partition, sha_256); print_sha256(sha_256, "SHA-256 for bootloader: "); // get sha256 digest for running partition esp_partition_get_sha256(esp_ota_get_running_partition(), sha_256); print_sha256(sha_256, "SHA-256 for current firmware: "); } void display_ota_state(pax_buf_t* pax_buffer, ILI9341* ili9341, const char* text) { pax_noclip(pax_buffer); const pax_font_t* font = pax_get_font("sky mono"); pax_background(pax_buffer, 0xFFFFFF); pax_vec1_t title_size = pax_text_size(font, 18, "Firmware update"); pax_draw_text(pax_buffer, 0xFF000000, font, 18, (320 / 2) - (title_size.x / 2), 120 - 30, "Firmware update"); pax_vec1_t size = pax_text_size(font, 18, text); pax_draw_text(pax_buffer, 0xFF000000, font, 18, (320 / 2) - (size.x / 2), 120 + 10, text); ili9341_write(ili9341, pax_buffer->buf); } void ota_update(pax_buf_t* pax_buffer, ILI9341* ili9341) { esp_wifi_set_ps(WIFI_PS_NONE); // Disable any WiFi power save mode ESP_LOGI(TAG, "Starting OTA update"); esp_http_client_config_t config = { .url = "https://ota.bodge.team/mch2022.bin", .crt_bundle_attach = esp_crt_bundle_attach, .cert_pem = (char *)server_cert_pem_start, .event_handler = _http_event_handler, .keep_alive_enable = true }; esp_https_ota_config_t ota_config = { .http_config = &config, .http_client_init_cb = _http_client_init_cb, // Register a callback to be invoked after esp_http_client is initialized #ifdef CONFIG_EXAMPLE_ENABLE_PARTIAL_HTTP_DOWNLOAD .partial_http_download = true, .max_http_request_size = CONFIG_EXAMPLE_HTTP_REQUEST_SIZE, #endif }; //config.skip_cert_common_name_check = true; ESP_LOGI(TAG, "Attempting to download update from %s", config.url); display_ota_state(pax_buffer, ili9341, "Starting download..."); esp_https_ota_handle_t https_ota_handle = NULL; esp_err_t err = esp_https_ota_begin(&ota_config, &https_ota_handle); if (err != ESP_OK) { ESP_LOGE(TAG, "ESP HTTPS OTA Begin failed"); display_ota_state(pax_buffer, ili9341, "Failed to start download"); vTaskDelay(5000 / portTICK_PERIOD_MS); esp_restart(); } esp_app_desc_t app_desc; err = esp_https_ota_get_img_desc(https_ota_handle, &app_desc); if (err != ESP_OK) { ESP_LOGE(TAG, "esp_https_ota_read_img_desc failed"); esp_https_ota_abort(https_ota_handle); display_ota_state(pax_buffer, ili9341, "Failed to read image desc"); vTaskDelay(5000 / portTICK_PERIOD_MS); esp_restart(); } err = validate_image_header(&app_desc); if (err != ESP_OK) { ESP_LOGE(TAG, "image header verification failed"); esp_https_ota_abort(https_ota_handle); display_ota_state(pax_buffer, ili9341, "Image header verification failed"); vTaskDelay(5000 / portTICK_PERIOD_MS); esp_restart(); } esp_err_t ota_finish_err = ESP_OK; int percent_shown = -1; while (1) { err = esp_https_ota_perform(https_ota_handle); if (err != ESP_ERR_HTTPS_OTA_IN_PROGRESS) { break; } int len_total = esp_https_ota_get_image_size(https_ota_handle); int len_read = esp_https_ota_get_image_len_read(https_ota_handle); int percent = (len_read * 100) / len_total; if (percent != percent_shown) { ESP_LOGI(TAG, "Downloading %d / %d (%d%%)", len_read, len_total, percent); percent_shown = percent; char buffer[128]; snprintf(buffer, sizeof(buffer), "Updating... %d%%", percent); display_ota_state(pax_buffer, ili9341, buffer); } } if (esp_https_ota_is_complete_data_received(https_ota_handle) != true) { // the OTA image was not completely received and user can customise the response to this situation. ESP_LOGE(TAG, "Complete data was not received."); display_ota_state(pax_buffer, ili9341, "Download failed"); vTaskDelay(5000 / portTICK_PERIOD_MS); esp_restart(); } else { ota_finish_err = esp_https_ota_finish(https_ota_handle); if ((err == ESP_OK) && (ota_finish_err == ESP_OK)) { ESP_LOGI(TAG, "ESP_HTTPS_OTA upgrade successful. Rebooting ..."); display_ota_state(pax_buffer, ili9341, "Update installed"); vTaskDelay(1000 / portTICK_PERIOD_MS); esp_restart(); } else { if (ota_finish_err == ESP_ERR_OTA_VALIDATE_FAILED) { ESP_LOGE(TAG, "Image validation failed, image is corrupted"); display_ota_state(pax_buffer, ili9341, "Image validation failed"); } else { display_ota_state(pax_buffer, ili9341, "Update failed"); } ESP_LOGE(TAG, "ESP_HTTPS_OTA upgrade failed 0x%x", ota_finish_err); vTaskDelay(5000 / portTICK_PERIOD_MS); esp_restart(); } } esp_https_ota_abort(https_ota_handle); esp_restart(); while (1) { vTaskDelay(1000 / portTICK_PERIOD_MS); } }