tasks: add ota update functions

Author: redchenjs

main/inc/tasks/ota_update.h

@@ -0,0 +1,18 @@
+/*
+ * ota_update.h
+ *
+ *  Created on: 2018-04-06 15:12
+ *      Author: Jack Chen <[email protected]>
+ */
+
+#ifndef INC_TASKS_OTA_UPDATE_H_
+#define INC_TASKS_OTA_UPDATE_H_
+
+#include "tasks/http2_client.h"
+
+extern int ota_update_parse_data(struct http2c_handle *handle, const char *data, size_t len, int flags);
+extern int ota_update_prepare_data(struct http2c_handle *handle, char *buf, size_t length, uint32_t *data_flags);
+
+extern void ota_update(void);
+
+#endif /* INC_TASKS_OTA_UPDATE_H_ */

main/src/tasks/ota_update.c

@@ -0,0 +1,116 @@
+/*
+ * ota_update.c
+ *
+ *  Created on: 2018-04-06 15:12
+ *      Author: Jack Chen <[email protected]>
+ */
+
+#include <string.h>
+
+#include "cJSON.h"
+#include "esp_log.h"
+#include "esp_ota_ops.h"
+
+#include "device/wifi.h"
+#include "system/event.h"
+#include "system/firmware.h"
+#include "tasks/gui_daemon.h"
+#include "tasks/nfc_daemon.h"
+#include "tasks/led_daemon.h"
+#include "tasks/audio_daemon.h"
+#include "tasks/http2_client.h"
+
+#define TAG "ota"
+
+int ota_update_parse_data(struct http2c_handle *handle, const char *data, size_t len, int flags)
+{
+    static const esp_partition_t *update_partition = NULL;
+    static esp_ota_handle_t update_handle = 0;
+    static long binary_file_length = 0;
+
+    if (len) {
+        EventBits_t uxBits = xEventGroupGetBits(daemon_event_group);
+        if (!(uxBits & HTTP2_DAEMON_OTA_RUN_BIT)) {
+            xEventGroupSetBits(daemon_event_group, HTTP2_DAEMON_OTA_RUN_BIT);
+
+            led_set_mode(3);
+            gui_show_image(8);
+
+            update_partition = esp_ota_get_next_update_partition(NULL);
+            ESP_LOGI(TAG, "writing to partition subtype %d at offset 0x%x",
+                        update_partition->subtype, update_partition->address);
+            assert(update_partition != NULL);
+
+            esp_err_t err = esp_ota_begin(update_partition, OTA_SIZE_UNKNOWN, &update_handle);
+            if (err != ESP_OK) {
+                ESP_LOGE(TAG, "esp_ota_begin failed (%s)", esp_err_to_name(err));
+                goto exit;
+            }
+
+            binary_file_length = 0;
+        }
+        esp_err_t err = esp_ota_write(update_handle, (const void *)data, len);
+        if (err != ESP_OK) {
+            ESP_LOGE(TAG, "esp_ota_write failed (%s)!", esp_err_to_name(err));
+            xEventGroupSetBits(daemon_event_group, HTTP2_DAEMON_OTA_FAILED_BIT);
+            goto exit;
+        }
+        binary_file_length += len;
+        ESP_LOGI(TAG, "have written image length %ld", binary_file_length);
+    }
+
+    if (flags == DATA_RECV_RST_STREAM) {
+        EventBits_t uxBits = xEventGroupGetBits(daemon_event_group);
+        if (uxBits & HTTP2_DAEMON_OTA_FAILED_BIT) {
+            ESP_LOGE(TAG, "ota update failed");
+        } else if (binary_file_length != 0) {
+            if (esp_ota_end(update_handle) != ESP_OK) {
+                ESP_LOGE(TAG, "esp_ota_end failed!");
+                goto exit;
+            }
+            esp_err_t 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));
+                goto exit;
+            }
+            ESP_LOGW(TAG, "prepare to restart system!");
+            gui_show_image(4);
+            vTaskDelay(2000 / portTICK_RATE_MS);
+            esp_restart();
+        } else {
+            ESP_LOGI(TAG, "no update found");
+        }
+exit:
+        xEventGroupClearBits(daemon_event_group, HTTP2_DAEMON_OTA_READY_BIT);
+    }
+
+    return 0;
+}
+
+int ota_update_prepare_data(struct http2c_handle *handle, char *buf, size_t length, uint32_t *data_flags)
+{
+    cJSON *root = NULL;
+    root = cJSON_CreateObject();
+    cJSON_AddNumberToObject(root, "request", 105);
+    cJSON_AddStringToObject(root, "version", firmware_get_version());
+    cJSON_AddStringToObject(root, "mac", wifi0_mac_str);
+    cJSON_PrintPreallocated(root, buf, length, 0);
+    cJSON_Delete(root);
+    (*data_flags) |= NGHTTP2_DATA_FLAG_EOF;
+    return strlen(buf);
+}
+
+void ota_update(void)
+{
+#if defined(CONFIG_ENABLE_OTA)
+    EventBits_t uxBits = xEventGroupSync(
+        daemon_event_group,
+        HTTP2_DAEMON_OTA_READY_BIT,
+        HTTP2_DAEMON_OTA_FINISH_BIT,
+        300000 / portTICK_RATE_MS
+    );
+    if ((uxBits & HTTP2_DAEMON_OTA_FINISH_BIT) == 0) {
+        xEventGroupClearBits(daemon_event_group, HTTP2_DAEMON_OTA_READY_BIT);
+    }
+#endif
+}