小龙虾
2caf10b9de
- 启用CONFIG_BOOTLOADER_APP_ROLLBACK_ENABLE和CONFIG_APP_ROLLBACK_ENABLE配置选项 - 在main.cpp中添加esp_ota_mark_app_valid_cancel_rollback()调用，防止意外回滚 - 实现SHA-256固件完整性校验，包括计算、存储和验证功能 - 添加ota_mutex.h头文件用于ESP32 OTA全局互斥锁管理 feat(health_monitor): 实现线程安全的健康状态管理 - 将HealthMonitor类的健康状态改为私有成员并通过mutex保护 - 添加getState()和setState()方法提供线程安全的状态访问 - 修复多任务并发访问健康状态时的竞争条件问题 feat(ota_relay): 增强中继模块的线程安全性和可靠性 - 为OtaRelay类添加静态mutex保护状态变量访问 - 实现StateLockGuard RAII锁管理器确保异常安全 - 添加全局超时机制防止OTA过程无限等待 - 改进flash写入逻辑支持非4字节对齐数据处理 refactor(app_config): 更新版本信息和配置常量 - 更新BUILD_TIME至最新编译时间 - 更新GIT_HASH至最新提交 - 添加BLE_OTA_CTRL_VERIFY_SHA256命令常量定义
89 lines
3.4 KiB
C++
89 lines
3.4 KiB
C++
/*
|
|
* wifi_ota.h - WiFi OTA 服务器
|
|
* T4.1 - T4.5
|
|
*
|
|
* WifiOta 类封装 WiFi 初始化、HTTP OTA 服务器和网络状态查询。
|
|
*/
|
|
#ifndef WIFI_OTA_H
|
|
#define WIFI_OTA_H
|
|
|
|
#include "ota_relay.h"
|
|
#include "esp_ota_ops.h"
|
|
#include "esp_http_server.h"
|
|
#include "esp_event.h"
|
|
#include <atomic>
|
|
|
|
// SHA-256 验证所需类型前向声明
|
|
typedef enum {
|
|
OTA_WIFI_MODE_AP,
|
|
OTA_WIFI_MODE_STA
|
|
} ota_wifi_mode_t;
|
|
|
|
class WifiOta {
|
|
public:
|
|
static WifiOta& instance();
|
|
|
|
bool earlyInit();
|
|
static void taskEntry(void *pvParameters);
|
|
void signalRelay();
|
|
|
|
static ota_wifi_mode_t getMode();
|
|
static int8_t getRssi();
|
|
static bool isConnected();
|
|
|
|
// ── WiFi 连接状态（WiFi 事件回调 / HTTP / reconnect 任务共享，需 mutex 保护）──
|
|
bool getStaConnected() const { return m_staConnected.load(); }
|
|
void setStaConnected(bool v) { m_staConnected.store(v); }
|
|
int getStaRetryCount() const { return m_staRetryCount.load(); }
|
|
void setStaRetryCount(int v) { m_staRetryCount.store(v); }
|
|
bool getStaRetryExhausted() const { return m_staRetryExhausted.load(); }
|
|
void setStaRetryExhausted(bool v) { m_staRetryExhausted.store(v); }
|
|
bool getGd32StorageDone() const { return m_gd32StorageDone.load(); }
|
|
void setGd32StorageDone(bool v) { m_gd32StorageDone.store(v); }
|
|
bool getRelayTriggered() const { return m_relayTriggered.load(); }
|
|
void setRelayTriggered(bool v) { m_relayTriggered.store(v); }
|
|
|
|
// ── 状态互斥锁（保护复合读-改-写操作）──
|
|
SemaphoreHandle_t getStateMutex() const { return m_stateMutex; }
|
|
void stateLock() { if (m_stateMutex) xSemaphoreTake(m_stateMutex, portMAX_DELAY); }
|
|
void stateUnlock() { if (m_stateMutex) xSemaphoreGive(m_stateMutex); }
|
|
|
|
// ── 初始化/运行时状态（仅 OTA 任务访问，无需额外锁）──
|
|
ota_wifi_mode_t m_wifiMode = OTA_WIFI_MODE_AP;
|
|
bool m_coldBoot = false;
|
|
SemaphoreHandle_t m_staConnectedSem = NULL;
|
|
SemaphoreHandle_t m_staGotIpSem = NULL;
|
|
TaskHandle_t m_reconnectTask = nullptr;
|
|
|
|
bool m_esp32OtaActive = false;
|
|
esp_ota_handle_t m_esp32OtaHandle = 0;
|
|
const esp_partition_t *m_esp32UpdatePartition = NULL;
|
|
size_t m_esp32Received = 0;
|
|
httpd_handle_t m_httpServer = NULL;
|
|
|
|
// ── SHA-256 哈希校验 ──
|
|
uint8_t m_esp32Sha256[32] = {0}; // 重启前验证计算的SHA-256哈希值
|
|
bool m_esp32HashComputed = false; // 是否已计算哈希
|
|
char m_frontendSha256[65] = {0}; // 前端提供的SHA-256（上传时存储）
|
|
bool m_hasFrontendSha256 = false; // 前端是否提供了SHA-256
|
|
|
|
private:
|
|
// ── 跨任务共享状态（atomic + m_stateMutex 保护）──
|
|
std::atomic<bool> m_staConnected{false};
|
|
std::atomic<int> m_staRetryCount{0};
|
|
std::atomic<bool> m_staRetryExhausted{false};
|
|
std::atomic<bool> m_gd32StorageDone{false};
|
|
std::atomic<bool> m_relayTriggered{false};
|
|
SemaphoreHandle_t m_stateMutex = NULL;
|
|
|
|
void initAp();
|
|
esp_err_t initSta();
|
|
void setStaticIp();
|
|
static void wifiEventHandler(void *arg, esp_event_base_t event_base,
|
|
int32_t event_id, void *event_data);
|
|
static void reconnectTask(void *pvParameters);
|
|
esp_err_t httpServerStart();
|
|
esp_err_t waitStaConnected(char *ip_str, size_t ip_len);
|
|
};
|
|
|
|
#endif // WIFI_OTA_H
|