feat: Implement web dashboard with system information, reboot, and OTA update functionality.
This commit is contained in:
149
Provider/main/api/ota/firmware.cpp
Normal file
149
Provider/main/api/ota/firmware.cpp
Normal file
@@ -0,0 +1,149 @@
|
||||
// SDK
|
||||
#include "cJSON.h"
|
||||
#include "esp_http_server.h"
|
||||
#include "esp_log.h"
|
||||
#include "esp_ota_ops.h"
|
||||
#include "esp_system.h"
|
||||
#include "esp_timer.h"
|
||||
#include <sys/param.h>
|
||||
|
||||
// Project
|
||||
#include "appstate.hpp"
|
||||
#include "types.hpp"
|
||||
|
||||
#define OTA_FIRMWARE_SCRATCH_BUFSIZE 4096
|
||||
|
||||
internal void firmware_ota_restart_timer_callback(void *arg) { esp_restart(); }
|
||||
|
||||
internal esp_err_t api_ota_firmware_handler(httpd_req_t *req)
|
||||
{
|
||||
httpd_resp_set_hdr(req, "Access-Control-Allow-Origin", "*");
|
||||
|
||||
const esp_partition_t *update_partition =
|
||||
esp_ota_get_next_update_partition(NULL);
|
||||
if (update_partition == NULL)
|
||||
{
|
||||
ESP_LOGE("OTA_FW", "Passive OTA partition not found");
|
||||
httpd_resp_send_err(req, HTTPD_500_INTERNAL_SERVER_ERROR,
|
||||
"OTA partition not found");
|
||||
return ESP_FAIL;
|
||||
}
|
||||
|
||||
ESP_LOGI("OTA_FW", "Writing to partition subtype %d at offset 0x%lx",
|
||||
update_partition->subtype, update_partition->address);
|
||||
|
||||
esp_ota_handle_t update_handle = 0;
|
||||
esp_err_t err =
|
||||
esp_ota_begin(update_partition, OTA_SIZE_UNKNOWN, &update_handle);
|
||||
if (err != ESP_OK)
|
||||
{
|
||||
ESP_LOGE("OTA_FW", "esp_ota_begin failed (%s)", esp_err_to_name(err));
|
||||
httpd_resp_send_err(req, HTTPD_500_INTERNAL_SERVER_ERROR,
|
||||
"OTA begin failed");
|
||||
return ESP_FAIL;
|
||||
}
|
||||
|
||||
char *buf = (char *)malloc(OTA_FIRMWARE_SCRATCH_BUFSIZE);
|
||||
if (!buf)
|
||||
{
|
||||
ESP_LOGE("OTA_FW", "Failed to allocate buffer");
|
||||
httpd_resp_send_err(req, HTTPD_500_INTERNAL_SERVER_ERROR, "Out of memory");
|
||||
return ESP_FAIL;
|
||||
}
|
||||
|
||||
int binary_file_len = 0;
|
||||
int remaining = req->content_len;
|
||||
|
||||
while (remaining > 0)
|
||||
{
|
||||
int recv_len =
|
||||
httpd_req_recv(req, buf, MIN(remaining, OTA_FIRMWARE_SCRATCH_BUFSIZE));
|
||||
if (recv_len <= 0)
|
||||
{
|
||||
if (recv_len == HTTPD_SOCK_ERR_TIMEOUT)
|
||||
{
|
||||
continue;
|
||||
}
|
||||
ESP_LOGE("OTA_FW", "Receive failed");
|
||||
esp_ota_abort(update_handle);
|
||||
free(buf);
|
||||
httpd_resp_send_err(req, HTTPD_500_INTERNAL_SERVER_ERROR,
|
||||
"Receive failed");
|
||||
return ESP_FAIL;
|
||||
}
|
||||
|
||||
err = esp_ota_write(update_handle, (const void *)buf, recv_len);
|
||||
if (err != ESP_OK)
|
||||
{
|
||||
ESP_LOGE("OTA_FW", "esp_ota_write failed (%s)", esp_err_to_name(err));
|
||||
esp_ota_abort(update_handle);
|
||||
free(buf);
|
||||
httpd_resp_send_err(req, HTTPD_500_INTERNAL_SERVER_ERROR,
|
||||
"Flash write failed");
|
||||
return ESP_FAIL;
|
||||
}
|
||||
|
||||
binary_file_len += recv_len;
|
||||
remaining -= recv_len;
|
||||
}
|
||||
|
||||
free(buf);
|
||||
ESP_LOGI("OTA_FW", "Total binary data written: %d", binary_file_len);
|
||||
|
||||
err = esp_ota_end(update_handle);
|
||||
if (err != ESP_OK)
|
||||
{
|
||||
if (err == ESP_ERR_OTA_VALIDATE_FAILED)
|
||||
{
|
||||
ESP_LOGE("OTA_FW", "Image validation failed, image is corrupted");
|
||||
}
|
||||
else
|
||||
{
|
||||
ESP_LOGE("OTA_FW", "esp_ota_end failed (%s)!", esp_err_to_name(err));
|
||||
}
|
||||
httpd_resp_send_err(req, HTTPD_500_INTERNAL_SERVER_ERROR,
|
||||
"OTA validation/end failed");
|
||||
return ESP_FAIL;
|
||||
}
|
||||
|
||||
err = esp_ota_set_boot_partition(update_partition);
|
||||
if (err != ESP_OK)
|
||||
{
|
||||
ESP_LOGE("OTA_FW", "esp_ota_set_boot_partition failed (%s)!",
|
||||
esp_err_to_name(err));
|
||||
httpd_resp_send_err(req, HTTPD_500_INTERNAL_SERVER_ERROR,
|
||||
"Failed to set boot partition");
|
||||
return ESP_FAIL;
|
||||
}
|
||||
|
||||
ESP_LOGI("OTA_FW", "OTA successful, rebooting...");
|
||||
|
||||
httpd_resp_set_type(req, "application/json");
|
||||
cJSON *root = cJSON_CreateObject();
|
||||
cJSON_AddStringToObject(root, "status", "success");
|
||||
cJSON_AddStringToObject(root, "message",
|
||||
"Firmware update successful, rebooting...");
|
||||
const char *response_text = cJSON_Print(root);
|
||||
httpd_resp_sendstr(req, response_text);
|
||||
free((void *)response_text);
|
||||
cJSON_Delete(root);
|
||||
|
||||
// Trigger reboot with 1s delay
|
||||
const esp_timer_create_args_t restart_timer_args = {
|
||||
.callback = &firmware_ota_restart_timer_callback,
|
||||
.arg = (void *)0,
|
||||
.dispatch_method = ESP_TIMER_TASK,
|
||||
.name = "fw_ota_restart_timer",
|
||||
.skip_unhandled_events = false};
|
||||
esp_timer_handle_t restart_timer;
|
||||
esp_timer_create(&restart_timer_args, &restart_timer);
|
||||
esp_timer_start_once(restart_timer, 1'000'000);
|
||||
|
||||
return ESP_OK;
|
||||
}
|
||||
|
||||
internal const httpd_uri_t api_ota_firmware_uri = {.uri = "/api/ota/firmware",
|
||||
.method = HTTP_POST,
|
||||
.handler =
|
||||
api_ota_firmware_handler,
|
||||
.user_ctx = NULL};
|
||||
@@ -4,6 +4,7 @@
|
||||
#include "cJSON.h"
|
||||
#include "esp_http_server.h"
|
||||
#include "esp_littlefs.h"
|
||||
#include "esp_ota_ops.h"
|
||||
#include "esp_partition.h"
|
||||
|
||||
// Project
|
||||
@@ -20,36 +21,53 @@ internal esp_err_t api_ota_status_handler(httpd_req_t *req)
|
||||
|
||||
cJSON_AddNumberToObject(root, "active_slot", g_Active_WWW_Partition);
|
||||
|
||||
constexpr const char *kPartitions[] = {"www_0", "www_1", "ota_0", "ota_1",
|
||||
"factory"};
|
||||
constexpr size_t kPartitionCount = ArrayCount(kPartitions);
|
||||
cJSON *parts_arr = cJSON_AddArrayToObject(root, "partitions");
|
||||
|
||||
for (size_t i = 0; i < kPartitionCount; i++)
|
||||
esp_partition_iterator_t it = esp_partition_find(
|
||||
ESP_PARTITION_TYPE_ANY, ESP_PARTITION_SUBTYPE_ANY, NULL);
|
||||
while (it != NULL)
|
||||
{
|
||||
const esp_partition_t *p = esp_partition_get(it);
|
||||
cJSON *p_obj = cJSON_CreateObject();
|
||||
cJSON_AddStringToObject(p_obj, "label", kPartitions[i]);
|
||||
cJSON_AddStringToObject(p_obj, "label", p->label);
|
||||
cJSON_AddNumberToObject(p_obj, "type", p->type);
|
||||
cJSON_AddNumberToObject(p_obj, "subtype", p->subtype);
|
||||
cJSON_AddNumberToObject(p_obj, "address", p->address);
|
||||
cJSON_AddNumberToObject(p_obj, "size", p->size);
|
||||
|
||||
const esp_partition_t *p = esp_partition_find_first(
|
||||
ESP_PARTITION_TYPE_DATA, ESP_PARTITION_SUBTYPE_ANY, kPartitions[i]);
|
||||
if (p)
|
||||
// Try to get LittleFS info if it's a data partition
|
||||
if (p->type == ESP_PARTITION_TYPE_DATA)
|
||||
{
|
||||
cJSON_AddNumberToObject(p_obj, "size", p->size);
|
||||
|
||||
size_t total = 0, used = 0;
|
||||
if (esp_littlefs_info(kPartitions[i], &total, &used) == ESP_OK)
|
||||
if (esp_littlefs_info(p->label, &total, &used) == ESP_OK)
|
||||
{
|
||||
cJSON_AddNumberToObject(p_obj, "used", used);
|
||||
cJSON_AddNumberToObject(p_obj, "free", total - used);
|
||||
}
|
||||
else
|
||||
{
|
||||
// Not mounted or not LFS
|
||||
cJSON_AddNumberToObject(p_obj, "used", 0);
|
||||
cJSON_AddNumberToObject(p_obj, "free", p->size);
|
||||
// For other data partitions (nvs, phy_init), just show total as used
|
||||
// for now
|
||||
cJSON_AddNumberToObject(p_obj, "used", p->size);
|
||||
cJSON_AddNumberToObject(p_obj, "free", 0);
|
||||
}
|
||||
}
|
||||
// For app partitions, try to find the binary size
|
||||
else if (p->type == ESP_PARTITION_TYPE_APP)
|
||||
{
|
||||
esp_app_desc_t app_desc;
|
||||
if (esp_ota_get_partition_description(p, &app_desc) == ESP_OK)
|
||||
{
|
||||
// This is a bit of a hack as we don't have a direct "binary size" in
|
||||
// the header but we can at least show it's occupied. For simplicity, if
|
||||
// it's a valid app, we'll mark some space as used. Actually, without a
|
||||
// better way to get the exact bin size, we'll just show it's an App.
|
||||
cJSON_AddStringToObject(p_obj, "app_version", app_desc.version);
|
||||
}
|
||||
}
|
||||
|
||||
cJSON_AddItemToArray(parts_arr, p_obj);
|
||||
it = esp_partition_next(it);
|
||||
}
|
||||
|
||||
cJSON_AddStringToObject(root, "active_partition",
|
||||
@@ -57,6 +75,24 @@ internal esp_err_t api_ota_status_handler(httpd_req_t *req)
|
||||
cJSON_AddStringToObject(root, "target_partition",
|
||||
g_Active_WWW_Partition == 0 ? "www_1" : "www_0");
|
||||
|
||||
const esp_partition_t *running = esp_ota_get_running_partition();
|
||||
if (running)
|
||||
{
|
||||
cJSON_AddStringToObject(root, "running_firmware_label", running->label);
|
||||
if (running->subtype >= ESP_PARTITION_SUBTYPE_APP_OTA_MIN &&
|
||||
running->subtype <= ESP_PARTITION_SUBTYPE_APP_OTA_MAX)
|
||||
{
|
||||
cJSON_AddNumberToObject(root, "running_firmware_slot",
|
||||
running->subtype -
|
||||
ESP_PARTITION_SUBTYPE_APP_OTA_MIN);
|
||||
}
|
||||
else
|
||||
{
|
||||
cJSON_AddNumberToObject(root, "running_firmware_slot",
|
||||
-1); // Factory or other
|
||||
}
|
||||
}
|
||||
|
||||
const char *status_info = cJSON_Print(root);
|
||||
httpd_resp_sendstr(req, status_info);
|
||||
|
||||
|
||||
Reference in New Issue
Block a user