小龙虾
9ca53ae66e
- 新增ota目录和esp32_ota_core.cpp实现ESP32 OTA核心功能，支持begin/write/finalize操作 - 修改CMakeLists.txt添加ota源文件和包含目录 - 重构ble_ota.cpp将OTA写入/收尾逻辑委托给Esp32OtaCore，移除mbedtls直接引用 - 新增BLE_OTA_CTRL_RESTART_ESP32(0xFA)命令用于重启ESP32设备 - 更新BLE_OTA_CTRL_TRIGGER_ESP32(0xF8)逻辑，要求前端提供SHA-256校验并比对 - 添加SHA-256校验不匹配错误码BLE_OTA_FINISH_ERR_SHA_MISMATCH(0x85) - 优化OTA收尾流程，设备端读取分区计算SHA-256并与前端提供的哈希比对 - 添加BLE_OTA_NOTIFY_SHA256通知类型用于上报设备端计算的SHA-256值 - 更新app_config.h中的BLE OTA命令描述文档 - 移除BleOta类中的OTA状态变量，仅保留BLE传输层相关状态 - 更新版本构建时间和Git哈希信息
83 lines
3.6 KiB
C++
83 lines
3.6 KiB
C++
/*
|
||
* esp32_ota_core.h - ESP32 本机 OTA 核心流程（WiFi / BLE 通道共用）
|
||
*
|
||
* 将 ESP32 自身固件升级的完整生命周期封装为单例：
|
||
* begin → write*(N) → finalize / abort
|
||
*
|
||
* 统一以 WiFi 流程为基准：
|
||
* - esp_ota_begin 第二参数传 total_size
|
||
* - SHA-256 在 finalize 时从 OTA 分区读回计算（不在上传期间增量计算）
|
||
* - finalize 顺序: esp_ota_end → 比对前端 SHA → esp_ota_set_boot_partition → 可选重启
|
||
* - m_ended 标志位防止 esp_ota_end 成功后再次 esp_ota_abort 同一 handle
|
||
*
|
||
* WiFi 与 BLE 仅传输方式不同（HTTP / GATT），OTA 写入与收尾逻辑完全复用本核心。
|
||
*/
|
||
#ifndef ESP32_OTA_CORE_H
|
||
#define ESP32_OTA_CORE_H
|
||
|
||
#include <stddef.h>
|
||
#include "esp_err.h"
|
||
#include "esp_ota_ops.h"
|
||
#include "esp_partition.h"
|
||
|
||
class Esp32OtaCore {
|
||
public:
|
||
static Esp32OtaCore& instance();
|
||
|
||
// ── 阶段 1：开启 OTA 会话 ──
|
||
// total_size: 固件总大小（字节），用于分区大小校验和传给 esp_ota_begin
|
||
// 返回 ESP_OK / ESP_ERR_INVALID_STATE（被其它通道占用）/ ESP_FAIL
|
||
esp_err_t begin(size_t total_size);
|
||
|
||
// ── 阶段 2：写入固件数据（可在 begin 后多次调用）──
|
||
// 纯 esp_ota_write，不做增量 SHA（SHA 在 finalize 时统一从分区读回计算）
|
||
esp_err_t write(const uint8_t *data, size_t len);
|
||
|
||
// ── 阶段 3：收尾（end → 校验 SHA → set_boot → 可选重启）──
|
||
// frontend_sha_hex: 前端提供的 64 字符 hex SHA-256，nullptr 表示跳过校验直接应用
|
||
// reboot: true 则 set_boot 成功后延时 500ms 调 esp_restart()
|
||
// out_device_sha_hex: 输出设备端计算出的 SHA hex（缓冲区至少 65 字节），可为 nullptr
|
||
// 返回:
|
||
// ESP_OK 成功（若 reboot=true，本调用不会返回）
|
||
// ESP_ERR_INVALID_VERSION SHA 不匹配（已 abort 并清理）
|
||
// 其他 end/set_boot 失败（已清理）
|
||
esp_err_t finalize(const char *frontend_sha_hex, bool reboot,
|
||
char *out_device_sha_hex);
|
||
|
||
// ── 仅计算设备端 SHA（不 end、不 set_boot、不重启）──
|
||
// 供前端只读校验场景（如 WiFi verify_only 模式）
|
||
// out_device_sha_hex: 输出缓冲区，至少 65 字节
|
||
// 返回 ESP_OK / 失败
|
||
esp_err_t computeDeviceSha(char *out_device_sha_hex);
|
||
|
||
// ── 中止并清理（abort + 释放 mutex + 状态复位）──
|
||
// 若 esp_ota_end 已成功（m_ended=true），则跳过 esp_ota_abort 仅复位状态
|
||
void abort();
|
||
|
||
// ── 状态查询 ──
|
||
bool isActive() const { return m_active; }
|
||
size_t getReceived() const { return m_received; }
|
||
bool isRebootPending() const { return m_rebootPending; }
|
||
void setRebootPending(bool v) { m_rebootPending = v; }
|
||
|
||
private:
|
||
Esp32OtaCore() = default;
|
||
|
||
// 从 OTA 分区读回计算 SHA-256（WiFi 风格），结果写入 m_sha256，置 m_shaComputed=true
|
||
esp_err_t computeShaFromPartition();
|
||
|
||
// 状态全复位（不调 esp_ota_abort，由调用方根据 m_ended 决定）
|
||
void resetState();
|
||
|
||
bool m_active = false;
|
||
esp_ota_handle_t m_handle = 0;
|
||
const esp_partition_t *m_partition = nullptr;
|
||
size_t m_received = 0;
|
||
uint8_t m_sha256[32] = {0};
|
||
bool m_shaComputed = false;
|
||
bool m_ended = false; // ★ esp_ota_end 成功后置 true，防 abort 误用
|
||
bool m_rebootPending = false; // BLE 收尾任务去重用
|
||
};
|
||
|
||
#endif // ESP32_OTA_CORE_H
|