136 lines
4.6 KiB
C
136 lines
4.6 KiB
C
// Copyright 2017-2018 Espressif Systems (Shanghai) PTE LTD
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <esp_https_ota.h>
|
|
#include <esp_ota_ops.h>
|
|
#include <esp_log.h>
|
|
|
|
#define DEFAULT_OTA_BUF_SIZE 256
|
|
static const char *TAG = "esp_https_ota";
|
|
|
|
static void http_cleanup(esp_http_client_handle_t client)
|
|
{
|
|
esp_http_client_close(client);
|
|
esp_http_client_cleanup(client);
|
|
}
|
|
|
|
esp_err_t esp_https_ota(const esp_http_client_config_t *config)
|
|
{
|
|
if (!config) {
|
|
ESP_LOGE(TAG, "esp_http_client config not found");
|
|
return ESP_ERR_INVALID_ARG;
|
|
}
|
|
|
|
#if !CONFIG_OTA_ALLOW_HTTP
|
|
if (!config->cert_pem && !config->use_global_ca_store) {
|
|
ESP_LOGE(TAG, "Server certificate not found, either through configuration or global CA store");
|
|
return ESP_ERR_INVALID_ARG;
|
|
}
|
|
#endif
|
|
|
|
esp_http_client_handle_t client = esp_http_client_init(config);
|
|
if (client == NULL) {
|
|
ESP_LOGE(TAG, "Failed to initialise HTTP connection");
|
|
return ESP_FAIL;
|
|
}
|
|
|
|
#if !CONFIG_OTA_ALLOW_HTTP
|
|
if (esp_http_client_get_transport_type(client) != HTTP_TRANSPORT_OVER_SSL) {
|
|
ESP_LOGE(TAG, "Transport is not over HTTPS");
|
|
return ESP_FAIL;
|
|
}
|
|
#endif
|
|
|
|
esp_err_t err = esp_http_client_open(client, 0);
|
|
if (err != ESP_OK) {
|
|
esp_http_client_cleanup(client);
|
|
ESP_LOGE(TAG, "Failed to open HTTP connection: %s", esp_err_to_name(err));
|
|
return err;
|
|
}
|
|
esp_http_client_fetch_headers(client);
|
|
|
|
esp_ota_handle_t update_handle = 0;
|
|
const esp_partition_t *update_partition = NULL;
|
|
ESP_LOGI(TAG, "Starting OTA...");
|
|
update_partition = esp_ota_get_next_update_partition(NULL);
|
|
if (update_partition == NULL) {
|
|
ESP_LOGE(TAG, "Passive OTA partition not found");
|
|
http_cleanup(client);
|
|
return ESP_FAIL;
|
|
}
|
|
ESP_LOGI(TAG, "Writing to partition subtype %d at offset 0x%x",
|
|
update_partition->subtype, update_partition->address);
|
|
|
|
err = esp_ota_begin(update_partition, OTA_SIZE_UNKNOWN, &update_handle);
|
|
if (err != ESP_OK) {
|
|
ESP_LOGE(TAG, "esp_ota_begin failed, error=%d", err);
|
|
http_cleanup(client);
|
|
return err;
|
|
}
|
|
ESP_LOGI(TAG, "esp_ota_begin succeeded");
|
|
ESP_LOGI(TAG, "Please Wait. This may take time");
|
|
|
|
esp_err_t ota_write_err = ESP_OK;
|
|
const int alloc_size = (config->buffer_size > 0) ? config->buffer_size : DEFAULT_OTA_BUF_SIZE;
|
|
char *upgrade_data_buf = (char *)malloc(alloc_size);
|
|
if (!upgrade_data_buf) {
|
|
ESP_LOGE(TAG, "Couldn't allocate memory to upgrade data buffer");
|
|
return ESP_ERR_NO_MEM;
|
|
}
|
|
|
|
int binary_file_len = 0;
|
|
while (1) {
|
|
int data_read = esp_http_client_read(client, upgrade_data_buf, alloc_size);
|
|
if (data_read == 0) {
|
|
ESP_LOGI(TAG, "Connection closed, all data received");
|
|
break;
|
|
}
|
|
if (data_read < 0) {
|
|
ESP_LOGE(TAG, "Error: SSL data read error");
|
|
break;
|
|
}
|
|
if (data_read > 0) {
|
|
ota_write_err = esp_ota_write(update_handle, (const void *) upgrade_data_buf, data_read);
|
|
if (ota_write_err != ESP_OK) {
|
|
break;
|
|
}
|
|
binary_file_len += data_read;
|
|
ESP_LOGD(TAG, "Written image length %d", binary_file_len);
|
|
}
|
|
}
|
|
free(upgrade_data_buf);
|
|
http_cleanup(client);
|
|
ESP_LOGD(TAG, "Total binary data length writen: %d", binary_file_len);
|
|
|
|
esp_err_t ota_end_err = esp_ota_end(update_handle);
|
|
if (ota_write_err != ESP_OK) {
|
|
ESP_LOGE(TAG, "Error: esp_ota_write failed! err=0x%d", err);
|
|
return ota_write_err;
|
|
} else if (ota_end_err != ESP_OK) {
|
|
ESP_LOGE(TAG, "Error: esp_ota_end failed! err=0x%d. Image is invalid", ota_end_err);
|
|
return ota_end_err;
|
|
}
|
|
|
|
err = esp_ota_set_boot_partition(update_partition);
|
|
if (err != ESP_OK) {
|
|
ESP_LOGE(TAG, "esp_ota_set_boot_partition failed! err=0x%d", err);
|
|
return err;
|
|
}
|
|
ESP_LOGI(TAG, "esp_ota_set_boot_partition succeeded");
|
|
|
|
return ESP_OK;
|
|
}
|