forked from xiaozhi/xiaozhi-esp32
122 lines
3.3 KiB
C++
122 lines
3.3 KiB
C++
#pragma once
|
|
|
|
#include <aes/esp_aes.h>
|
|
#include "mbedtls/dhm.h"
|
|
#include "mbedtls/aes.h"
|
|
#include "esp_err.h"
|
|
#include "esp_blufi_api.h"
|
|
#include "esp_wifi_types.h"
|
|
|
|
|
|
class Blufi {
|
|
public:
|
|
/**
|
|
* @brief Get the singleton instance of the Blufi class.
|
|
*/
|
|
static Blufi &GetInstance();
|
|
|
|
/**
|
|
* @brief Initializes the Bluetooth controller, host, and Blufi profile.
|
|
* This is the main entry point to start the Blufi process.
|
|
* @return ESP_OK on success, otherwise an error code.
|
|
*/
|
|
esp_err_t init();
|
|
|
|
/**
|
|
* @brief Deinitializes Blufi and the Bluetooth stack.
|
|
* @return ESP_OK on success, otherwise an error code.
|
|
*/
|
|
esp_err_t deinit();
|
|
|
|
// Delete copy constructor and assignment operator for singleton
|
|
Blufi(const Blufi &) = delete;
|
|
|
|
Blufi &operator=(const Blufi &) = delete;
|
|
|
|
private:
|
|
Blufi();
|
|
|
|
~Blufi();
|
|
|
|
|
|
// Initialization logic
|
|
static esp_err_t _controller_init();
|
|
|
|
static esp_err_t _controller_deinit();
|
|
|
|
static esp_err_t _host_init();
|
|
|
|
static esp_err_t _host_deinit();
|
|
|
|
static esp_err_t _gap_register_callback();
|
|
|
|
static esp_err_t _host_and_cb_init();
|
|
|
|
void _security_init();
|
|
|
|
void _security_deinit();
|
|
|
|
void _dh_negotiate_data_handler(uint8_t *data, int len, uint8_t **output_data, int *output_len, bool *need_free);
|
|
|
|
int _aes_encrypt(uint8_t iv8, uint8_t *crypt_data, int crypt_len);
|
|
|
|
int _aes_decrypt(uint8_t iv8, uint8_t *crypt_data, int crypt_len);
|
|
|
|
static uint16_t _crc_checksum(uint8_t iv8, uint8_t *data, int len);
|
|
|
|
void _handle_event(esp_blufi_cb_event_t event, esp_blufi_cb_param_t *param);
|
|
|
|
static int _get_softap_conn_num();
|
|
|
|
// These C-style functions are registered with ESP-IDF and call the corresponding instance methods.
|
|
|
|
static void _event_callback_trampoline(esp_blufi_cb_event_t event, esp_blufi_cb_param_t *param);
|
|
|
|
static void _negotiate_data_handler_trampoline(uint8_t *data, int len, uint8_t **output_data, int *output_len,
|
|
bool *need_free);
|
|
|
|
static int _encrypt_func_trampoline(uint8_t iv8, uint8_t *crypt_data, int crypt_len);
|
|
|
|
static int _decrypt_func_trampoline(uint8_t iv8, uint8_t *crypt_data, int crypt_len);
|
|
|
|
static uint16_t _checksum_func_trampoline(uint8_t iv8, uint8_t *data, int len);
|
|
|
|
#ifdef CONFIG_BT_NIMBLE_ENABLED
|
|
static void _nimble_on_reset(int reason);
|
|
static void _nimble_on_sync();
|
|
static void _nimble_host_task(void *param);
|
|
#endif
|
|
|
|
// Security context, formerly blufi_sec struct
|
|
struct BlufiSecurity {
|
|
#define DH_SELF_PUB_KEY_LEN 128
|
|
uint8_t self_public_key[DH_SELF_PUB_KEY_LEN];
|
|
#define SHARE_KEY_LEN 128
|
|
uint8_t share_key[SHARE_KEY_LEN];
|
|
size_t share_len;
|
|
#define PSK_LEN 16
|
|
uint8_t psk[PSK_LEN];
|
|
uint8_t *dh_param;
|
|
int dh_param_len;
|
|
uint8_t iv[16];
|
|
mbedtls_dhm_context *dhm;
|
|
esp_aes_context *aes;
|
|
};
|
|
|
|
BlufiSecurity *m_sec;
|
|
|
|
// State variables
|
|
wifi_config_t m_sta_config{};
|
|
wifi_config_t m_ap_config{};
|
|
bool m_ble_is_connected;
|
|
bool m_sta_connected;
|
|
bool m_sta_got_ip;
|
|
bool m_provisioned;
|
|
bool m_deinited;
|
|
uint8_t m_sta_bssid[6]{};
|
|
uint8_t m_sta_ssid[32]{};
|
|
int m_sta_ssid_len;
|
|
bool m_sta_is_connecting;
|
|
esp_blufi_extra_info_t m_sta_conn_info{};
|
|
};
|