Files
Bionic_sphere/Lib/Wireless/Wireless.c
T
Misaki ba5e47bc77 这是一次长久的提交:
1. 应用界面增加了返回主页的按钮
2. 修复了gif渲染内存泄漏的严重bug
3. 将PetDao当中的cJSON API替换为cpp_json,完美通过测试
4. 整合已经实现的各种上层建筑,实现了一个宠物对话基本业务应用,用于样品测试展示用
5. 重构了音频播放类,使其更modern,更加便于移植和拓展
2025-10-16 11:36:45 +08:00

412 lines
14 KiB
C

#include "Wireless.h"
uint16_t BLE_NUM = 0;
uint16_t WIFI_NUM = 0;
bool Scan_finish = 0;
bool WiFi_Scan_Finish = 0;
bool BLE_Scan_Finish = 0;
wifi_ap_info_t *wifi_ap_list = NULL;
uint16_t wifi_ap_count = 0;
ble_device_info_t *ble_device_list = NULL;
esp_err_t Wireless_Init(void)
{
// Initialize NVS.
esp_err_t ret = nvs_flash_init(); // 初始化Flash
if (ret == ESP_ERR_NVS_NO_FREE_PAGES || ret == ESP_ERR_NVS_NEW_VERSION_FOUND) { // 如果Flash有错误,则进行修复
ESP_ERROR_CHECK(nvs_flash_erase()); // 清空Flash
ret = nvs_flash_init(); // 重新初始化Flash
}
ESP_ERROR_CHECK( ret ); // 检查错误
// WiFi 手动调用Wifi初始化
// xTaskCreatePinnedToCore(
// WIFI_Init,
// "WIFI task",
// 4096,
// NULL,
// 1,
// NULL,
// 0);
// BLE 当前工程不启用蓝牙
// xTaskCreatePinnedToCore(
// BLE_Init,
// "BLE task",
// 4096,
// NULL,
// 2,
// NULL,
// 0);
return ret;
}
esp_err_t WIFI_Init(void *arg)
{
esp_netif_init(); // 初始化网络接口
esp_event_loop_create_default(); // 创建事件循环
esp_netif_create_default_wifi_sta(); // 创建默认的WiFi Station
wifi_init_config_t cfg = WIFI_INIT_CONFIG_DEFAULT(); // 创建WiFi初始化配置(使用默认配置)
esp_wifi_init(&cfg); // 初始化WiFi
esp_wifi_set_mode(WIFI_MODE_STA); // 设置WiFi模式为Station
esp_err_t ret = esp_wifi_start(); // 启动WiFi
return ret;
}
uint16_t WIFI_Scan(void)
{
uint16_t ap_count = 0;
esp_wifi_scan_start(NULL, true); // 开始扫描
ESP_ERROR_CHECK(esp_wifi_scan_get_ap_num(&ap_count));
// 分配内存保存AP信息
if (wifi_ap_list) { // 如果已有内存,则释放
free(wifi_ap_list);
}
wifi_ap_list = malloc(ap_count * sizeof(wifi_ap_info_t));
if (!wifi_ap_list) {
printf("Failed to allocate memory for WiFi AP list\n");
return 0;
}
// 获取AP列表
wifi_ap_record_t *ap_records = malloc(ap_count * sizeof(wifi_ap_record_t));
if (!ap_records) {
printf("Failed to allocate memory for AP records\n");
free(wifi_ap_list); // 释放AP列表内存
return 0;
}
ESP_ERROR_CHECK(esp_wifi_scan_get_ap_records(&ap_count, ap_records));
// 保存AP信息
wifi_ap_count = ap_count;
for (int i = 0; i < ap_count; i++) {
memset(wifi_ap_list[i].ssid, 0, sizeof(wifi_ap_list[i].ssid));
strncpy(wifi_ap_list[i].ssid, (char*)ap_records[i].ssid, sizeof(wifi_ap_list[i].ssid) - 1);
memcpy(wifi_ap_list[i].bssid, ap_records[i].bssid, 6);
wifi_ap_list[i].rssi = ap_records[i].rssi;
wifi_ap_list[i].authmode = ap_records[i].authmode;
}
free(ap_records);
esp_wifi_scan_stop();
WiFi_Scan_Finish = 1;
if (BLE_Scan_Finish == 1 || WiFi_Scan_Finish == 1) {
Scan_finish = 1;
}
return ap_count;
}
/*
#define GATTC_TAG "GATTC_TAG"
#define SCAN_DURATION 5
#define MAX_DISCOVERED_DEVICES 100
typedef struct {
uint8_t address[6];
bool is_valid;
} discovered_device_t;
static discovered_device_t discovered_devices[MAX_DISCOVERED_DEVICES];
static size_t num_discovered_devices = 0;
static size_t num_devices_with_name = 0;
static bool is_device_discovered(const uint8_t *addr) {
for (size_t i = 0; i < num_discovered_devices; i++) {
if (memcmp(discovered_devices[i].address, addr, 6) == 0) {
return true;
}
}
return false;
}
static void add_device_to_list(const uint8_t *addr) {
if (num_discovered_devices < MAX_DISCOVERED_DEVICES) {
memcpy(discovered_devices[num_discovered_devices].address, addr, 6);
discovered_devices[num_discovered_devices].is_valid = true;
num_discovered_devices++;
}
}
static bool extract_device_name(const uint8_t *adv_data, uint8_t adv_data_len, char *device_name, size_t max_name_len) {
size_t offset = 0;
while (offset < adv_data_len) {
if (adv_data[offset] == 0) break;
uint8_t length = adv_data[offset];
if (length == 0 || offset + length > adv_data_len) break;
uint8_t type = adv_data[offset + 1];
if (type == ESP_BLE_AD_TYPE_NAME_CMPL || type == ESP_BLE_AD_TYPE_NAME_SHORT) {
if (length > 1 && length - 1 < max_name_len) {
memcpy(device_name, &adv_data[offset + 2], length - 1);
device_name[length - 1] = '\0';
return true;
} else {
return false;
}
}
offset += length + 1;
}
return false;
}
static void esp_gap_cb(esp_gap_ble_cb_event_t event, esp_ble_gap_cb_param_t *param) {
static char device_name[100];
switch (event) {
case ESP_GAP_BLE_SCAN_RESULT_EVT:
if (param->scan_rst.search_evt == ESP_GAP_SEARCH_INQ_RES_EVT) {
if (!is_device_discovered(param->scan_rst.bda)) {
add_device_to_list(param->scan_rst.bda);
BLE_NUM++;
// 分配内存保存BLE设备信息
if (!ble_device_list) {
ble_device_list = malloc(MAX_DISCOVERED_DEVICES * sizeof(ble_device_info_t));
if (!ble_device_list) {
printf("Failed to allocate memory for BLE device list\n");
return;
}
}
// 保存设备信息
int idx = BLE_NUM - 1;
if (idx < MAX_DISCOVERED_DEVICES) {
memset(ble_device_list[idx].name, 0, sizeof(ble_device_list[idx].name));
if (extract_device_name(param->scan_rst.ble_adv,
param->scan_rst.adv_data_len,
device_name,
sizeof(device_name))) {
strncpy(ble_device_list[idx].name, device_name, sizeof(ble_device_list[idx].name) - 1);
num_devices_with_name++;
} else {
strncpy(ble_device_list[idx].name, "Unknown", sizeof(ble_device_list[idx].name) - 1);
}
memcpy(ble_device_list[idx].address, param->scan_rst.bda, 6);
ble_device_list[idx].rssi = param->scan_rst.rssi;
}
}
}
break;
case ESP_GAP_BLE_SCAN_STOP_COMPLETE_EVT:
ESP_LOGI(GATTC_TAG, "Scan complete. Total devices found: %d (with names: %d)", BLE_NUM, num_devices_with_name);
break;
default:
break;
}
}
void BLE_Init(void *arg)
{
ESP_ERROR_CHECK(esp_bt_controller_mem_release(ESP_BT_MODE_CLASSIC_BT));
esp_bt_controller_config_t bt_cfg = BT_CONTROLLER_INIT_CONFIG_DEFAULT();
esp_err_t ret = esp_bt_controller_init(&bt_cfg);
if (ret) {
printf("%s initialize controller failed: %s\n", __func__, esp_err_to_name(ret));
return;}
ret = esp_bt_controller_enable(ESP_BT_MODE_BLE);
if (ret) {
printf("%s enable controller failed: %s\n", __func__, esp_err_to_name(ret));
return;}
ret = esp_bluedroid_init();
if (ret) {
printf("%s init bluetooth failed: %s\n", __func__, esp_err_to_name(ret));
return;}
ret = esp_bluedroid_enable();
if (ret) {
printf("%s enable bluetooth failed: %s\n", __func__, esp_err_to_name(ret));
return;}
//register the callback function to the gap module
ret = esp_ble_gap_register_callback(esp_gap_cb);
if (ret){
printf("%s gap register error, error code = %x\n", __func__, ret);
return;
}
BLE_Scan();
// while(1)
// {
// vTaskDelay(pdMS_TO_TICKS(150));
// }
vTaskDelete(NULL);
}
uint16_t BLE_Scan(void)
{
esp_ble_scan_params_t scan_params = {
.scan_type = BLE_SCAN_TYPE_ACTIVE,
.own_addr_type = BLE_ADDR_TYPE_RPA_PUBLIC,
.scan_filter_policy = BLE_SCAN_FILTER_ALLOW_ALL,
.scan_interval = 0x50,
.scan_window = 0x30,
.scan_duplicate = BLE_SCAN_DUPLICATE_DISABLE
};
ESP_ERROR_CHECK(esp_ble_gap_set_scan_params(&scan_params));
printf("Starting BLE scan...\n");
ESP_ERROR_CHECK(esp_ble_gap_start_scanning(SCAN_DURATION));
// Set scanning duration
vTaskDelay(SCAN_DURATION * 1000 / portTICK_PERIOD_MS);
printf("Stopping BLE scan...\n");
// ESP_ERROR_CHECK(esp_ble_gap_stop_scanning());
ESP_ERROR_CHECK(esp_ble_dtm_stop());
BLE_Scan_Finish = 1;
if(WiFi_Scan_Finish == 1)
Scan_finish = 1;
return BLE_NUM;
}
*/
// 下面是新增的实现
static EventGroupHandle_t s_wifi_event_group;
#define WIFI_CONNECTED_BIT BIT0
#define WIFI_FAIL_BIT BIT1
static const char *TAG = "WiFi_UTIL";
/* --------------------------- 事件回调 --------------------------- */
static void event_handler(void *arg, esp_event_base_t event_base,
int32_t event_id, void *event_data)
{
if (event_base == WIFI_EVENT && event_id == WIFI_EVENT_STA_DISCONNECTED) {
xEventGroupSetBits(s_wifi_event_group, WIFI_FAIL_BIT);
} else if (event_base == IP_EVENT && event_id == IP_EVENT_STA_GOT_IP) {
xEventGroupSetBits(s_wifi_event_group, WIFI_CONNECTED_BIT);
}
}
/* --------------------------- 自动连接 --------------------------- */
bool WiFi_AutoConnect(const char *ssid, const char *password, uint8_t max_retry)
{
if (s_wifi_event_group == NULL) {
s_wifi_event_group = xEventGroupCreate();
}
/* 注册一次性事件监听 */
ESP_ERROR_CHECK(esp_event_handler_instance_register(WIFI_EVENT,
WIFI_EVENT_STA_DISCONNECTED,
&event_handler,
NULL,
NULL));
ESP_ERROR_CHECK(esp_event_handler_instance_register(IP_EVENT,
IP_EVENT_STA_GOT_IP,
&event_handler,
NULL,
NULL));
wifi_config_t wifi_config = { 0 };
strncpy((char *)wifi_config.sta.ssid, ssid, sizeof(wifi_config.sta.ssid));
strncpy((char *)wifi_config.sta.password, password, sizeof(wifi_config.sta.password));
wifi_config.sta.threshold.authmode = WIFI_AUTH_WPA2_PSK;
ESP_ERROR_CHECK(esp_wifi_set_config(WIFI_IF_STA, &wifi_config));
uint8_t retry_cnt = 0;
while (retry_cnt <= max_retry) {
ESP_LOGI(TAG, "WiFi connect attempt %u/%u to \"%s\"", retry_cnt + 1, max_retry + 1, ssid);
xEventGroupClearBits(s_wifi_event_group, WIFI_CONNECTED_BIT | WIFI_FAIL_BIT);
esp_wifi_connect();
EventBits_t bits = xEventGroupWaitBits(s_wifi_event_group,
WIFI_CONNECTED_BIT | WIFI_FAIL_BIT,
pdFALSE, pdFALSE,
pdMS_TO_TICKS(6000)); /* 6 秒超时 */
if (bits & WIFI_CONNECTED_BIT) {
ESP_LOGI(TAG, "WiFi connected");
goto success;
}
/* 失败,准备重试 */
retry_cnt++;
}
ESP_LOGE(TAG, "WiFi connect failed after %u retries", max_retry + 1);
return false;
success:
return true;
}
/* --------------------------- 断开连接 --------------------------- */
esp_err_t WiFi_Disconnect(void)
{
ESP_LOGI(TAG, "Disconnecting WiFi...");
return esp_wifi_disconnect();
}
// 获取WiFi AP列表
uint16_t wireless_get_wifi_ap_list(wifi_ap_info_t *ap_list, uint16_t max_aps) {
if (!wifi_ap_list || wifi_ap_count == 0) {
return 0;
}
uint16_t copy_count = (wifi_ap_count < max_aps) ? wifi_ap_count : max_aps;
memcpy(ap_list, wifi_ap_list, copy_count * sizeof(wifi_ap_info_t));
return copy_count;
}
// 获取BLE设备列表
uint16_t wireless_get_ble_device_list(ble_device_info_t *device_list, uint16_t max_devices) {
if (!ble_device_list || BLE_NUM == 0) {
return 0;
}
uint16_t copy_count = (BLE_NUM < max_devices) ? BLE_NUM : max_devices;
memcpy(device_list, ble_device_list, copy_count * sizeof(ble_device_info_t));
return copy_count;
}
// 打印所有WiFi AP信息
void wireless_print_wifi_aps(void) {
if (!wifi_ap_list || wifi_ap_count == 0) {
printf("No WiFi APs found or scan not completed\n");
return;
}
printf("Found %d WiFi APs:\n", wifi_ap_count);
for (int i = 0; i < wifi_ap_count; i++) {
printf(" %d: SSID: %s, BSSID: %02X:%02X:%02X:%02X:%02X:%02X, RSSI: %d, Auth: %d\n",
i + 1,
wifi_ap_list[i].ssid,
wifi_ap_list[i].bssid[0], wifi_ap_list[i].bssid[1], wifi_ap_list[i].bssid[2],
wifi_ap_list[i].bssid[3], wifi_ap_list[i].bssid[4], wifi_ap_list[i].bssid[5],
wifi_ap_list[i].rssi,
wifi_ap_list[i].authmode);
}
}
// 打印所有BLE设备信息
void wireless_print_ble_devices(void) {
if (!ble_device_list || BLE_NUM == 0) {
printf("No BLE devices found or scan not completed\n");
return;
}
printf("Found %d BLE devices:\n", BLE_NUM);
for (int i = 0; i < BLE_NUM; i++) {
printf(" %d: Name: %s, Address: %02X:%02X:%02X:%02X:%02X:%02X, RSSI: %d\n",
i + 1,
ble_device_list[i].name,
ble_device_list[i].address[0], ble_device_list[i].address[1], ble_device_list[i].address[2],
ble_device_list[i].address[3], ble_device_list[i].address[4], ble_device_list[i].address[5],
ble_device_list[i].rssi);
}
}