espressif/esp-idf
1
0
Fork 1
mirror of https://github.com/espressif/esp-idf.git synced 2025-08-01 19:54:32 +02:00
Code Issues Packages Projects Releases Wiki Activity

esp_https_ota: add support for configurable ota buffer size

Browse Source 
Closes https://github.com/espressif/esp-idf/issues/2998
This commit is contained in:
Mahavir Jain
2019-01-30 14:24:20 +05:30
parent d894e133ad
commit b912d0fa9d
2 changed files with 11 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
components/esp_https_ota/src/esp_https_ota.c
View File

@@ -19,7 +19,7 @@
#include <esp_ota_ops.h> #include <esp_ota_ops.h>
#include <esp_log.h> #include <esp_log.h>
#define OTA_BUF_SIZE 256 #define DEFAULT_OTA_BUF_SIZE 256
static const char *TAG = "esp_https_ota"; static const char *TAG = "esp_https_ota";
static void http_cleanup(esp_http_client_handle_t client) static void http_cleanup(esp_http_client_handle_t client)
@@ -85,16 +85,18 @@ esp_err_t esp_https_ota(const esp_http_client_config_t *config)
ESP_LOGI(TAG, "Please Wait. This may take time"); ESP_LOGI(TAG, "Please Wait. This may take time");
esp_err_t ota_write_err = ESP_OK; esp_err_t ota_write_err = ESP_OK;
char *upgrade_data_buf = (char *)malloc(OTA_BUF_SIZE); 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) { if (!upgrade_data_buf) {
ESP_LOGE(TAG, "Couldn't allocate memory to upgrade data buffer"); ESP_LOGE(TAG, "Couldn't allocate memory to upgrade data buffer");
return ESP_ERR_NO_MEM; return ESP_ERR_NO_MEM;
} }
int binary_file_len = 0; int binary_file_len = 0;
while (1) { while (1) {
int data_read = esp_http_client_read(client, upgrade_data_buf, OTA_BUF_SIZE); int data_read = esp_http_client_read(client, upgrade_data_buf, alloc_size);
if (data_read == 0) { if (data_read == 0) {
ESP_LOGI(TAG, "Connection closed,all data received"); ESP_LOGI(TAG, "Connection closed, all data received");
break; break;
} }
if (data_read < 0) { if (data_read < 0) {
@@ -102,7 +104,7 @@ esp_err_t esp_https_ota(const esp_http_client_config_t *config)
break; break;
} }
if (data_read > 0) { if (data_read > 0) {
ota_write_err = esp_ota_write( update_handle, (const void *)upgrade_data_buf, data_read); ota_write_err = esp_ota_write(update_handle, (const void *) upgrade_data_buf, data_read);
if (ota_write_err != ESP_OK) { if (ota_write_err != ESP_OK) {
break; break;
} }

8
examples/system/ota/simple_ota_example/main/simple_ota_example.c
View File

@@ -96,7 +96,7 @@ static void initialise_wifi(void)
.password = CONFIG_WIFI_PASSWORD, .password = CONFIG_WIFI_PASSWORD,
}, },
}; };
ESP_LOGI(TAG, "Setting WiFi configuration SSID %s...", wifi_config.sta.ssid); ESP_LOGI(TAG, "Setting WiFi configuration SSID %s", wifi_config.sta.ssid);
ESP_ERROR_CHECK( esp_wifi_set_mode(WIFI_MODE_STA) ); ESP_ERROR_CHECK( esp_wifi_set_mode(WIFI_MODE_STA) );
ESP_ERROR_CHECK( esp_wifi_set_config(ESP_IF_WIFI_STA, &wifi_config) ); ESP_ERROR_CHECK( esp_wifi_set_config(ESP_IF_WIFI_STA, &wifi_config) );
ESP_ERROR_CHECK( esp_wifi_start() ); ESP_ERROR_CHECK( esp_wifi_start() );
@@ -104,14 +104,14 @@ static void initialise_wifi(void)
void simple_ota_example_task(void * pvParameter) void simple_ota_example_task(void * pvParameter)
{ {
ESP_LOGI(TAG, "Starting OTA example..."); ESP_LOGI(TAG, "Starting OTA example");
/* Wait for the callback to set the CONNECTED_BIT in the /* Wait for the callback to set the CONNECTED_BIT in the
event group. event group.
*/ */
xEventGroupWaitBits(wifi_event_group, CONNECTED_BIT, xEventGroupWaitBits(wifi_event_group, CONNECTED_BIT,
false, true, portMAX_DELAY); false, true, portMAX_DELAY);
ESP_LOGI(TAG, "Connect to Wifi ! Start to Connect to Server...."); ESP_LOGI(TAG, "Connected to WiFi network! Attempting to connect to server...");
esp_http_client_config_t config = { esp_http_client_config_t config = {
.url = CONFIG_FIRMWARE_UPGRADE_URL, .url = CONFIG_FIRMWARE_UPGRADE_URL,
@@ -122,7 +122,7 @@ void simple_ota_example_task(void * pvParameter)
if (ret == ESP_OK) { if (ret == ESP_OK) {
esp_restart(); esp_restart();
} else { } else {
ESP_LOGE(TAG, "Firmware Upgrades Failed"); ESP_LOGE(TAG, "Firmware upgrade failed");
} }
while (1) { while (1) {
vTaskDelay(1000 / portTICK_PERIOD_MS); vTaskDelay(1000 / portTICK_PERIOD_MS);