diff --git a/esphome/components/bluetooth_proxy/bluetooth_connection.cpp b/esphome/components/bluetooth_proxy/bluetooth_connection.cpp index 72efd17742..abfcf5ec88 100644 --- a/esphome/components/bluetooth_proxy/bluetooth_connection.cpp +++ b/esphome/components/bluetooth_proxy/bluetooth_connection.cpp @@ -93,84 +93,136 @@ void BluetoothConnection::send_service_for_discovery_() { fill_128bit_uuid_array(service_resp.uuid, service_result.uuid); service_resp.handle = service_result.start_handle; + // Process characteristics for this service + if (!this->process_characteristics_for_service_(service_result.start_handle, service_result.end_handle, + service_resp)) { + return; // Error processing characteristics + } + + // Send the message (we already checked api_conn is not null at the beginning) + api_conn->send_message(resp, api::BluetoothGATTGetServicesResponse::MESSAGE_TYPE); +} + +bool BluetoothConnection::process_characteristics_for_service_(uint16_t service_start_handle, + uint16_t service_end_handle, + api::BluetoothGATTService &service_resp) { // Get the number of characteristics directly with one call uint16_t total_char_count = 0; esp_gatt_status_t char_count_status = - esp_ble_gattc_get_attr_count(this->gattc_if_, this->conn_id_, ESP_GATT_DB_CHARACTERISTIC, - service_result.start_handle, service_result.end_handle, 0, &total_char_count); + esp_ble_gattc_get_attr_count(this->gattc_if_, this->conn_id_, ESP_GATT_DB_CHARACTERISTIC, service_start_handle, + service_end_handle, 0, &total_char_count); if (char_count_status != ESP_GATT_OK) { ESP_LOGW(TAG, "[%d] [%s] Error getting characteristic count, status=%d", this->connection_index_, this->address_str().c_str(), char_count_status); - return; + return false; } - if (total_char_count > 0) { - // Reserve space and process characteristics - service_resp.characteristics.reserve(total_char_count); - uint16_t char_offset = 0; - esp_gattc_char_elem_t char_result; - while (true) { // characteristics - uint16_t char_count = 1; - esp_gatt_status_t char_status = - esp_ble_gattc_get_all_char(this->gattc_if_, this->conn_id_, service_result.start_handle, - service_result.end_handle, &char_result, &char_count, char_offset); - if (char_status == ESP_GATT_INVALID_OFFSET || char_status == ESP_GATT_NOT_FOUND || char_count == 0) { - break; - } else if (char_status != ESP_GATT_OK) { - ESP_LOGE(TAG, "[%d] [%s] esp_ble_gattc_get_all_char error, status=%d", this->connection_index_, - this->address_str().c_str(), char_status); - return; - } + if (total_char_count == 0) { + return true; // No characteristics, which is valid + } + // Reserve space and process characteristics + service_resp.characteristics.reserve(total_char_count); + + // Use a reasonable fixed-size buffer on the stack + constexpr uint8_t MAX_CHARACTERISTICS_PER_BATCH = 8; + esp_gattc_char_elem_t char_results[MAX_CHARACTERISTICS_PER_BATCH]; + uint16_t char_offset = 0; + + while (char_offset < total_char_count) { + uint16_t char_count = std::min(static_cast(MAX_CHARACTERISTICS_PER_BATCH), + static_cast(total_char_count - char_offset)); + + esp_gatt_status_t char_status = + esp_ble_gattc_get_all_char(this->gattc_if_, this->conn_id_, service_start_handle, service_end_handle, + char_results, &char_count, char_offset); + + if (char_status != ESP_GATT_OK) { + ESP_LOGE(TAG, "[%d] [%s] esp_ble_gattc_get_all_char error, status=%d", this->connection_index_, + this->address_str().c_str(), char_status); + return false; + } + + if (char_count == 0) { + break; // No more characteristics + } + + // Process this batch of characteristics + for (uint8_t i = 0; i < char_count; i++) { service_resp.characteristics.emplace_back(); auto &characteristic_resp = service_resp.characteristics.back(); - fill_128bit_uuid_array(characteristic_resp.uuid, char_result.uuid); - characteristic_resp.handle = char_result.char_handle; - characteristic_resp.properties = char_result.properties; - char_offset++; + fill_128bit_uuid_array(characteristic_resp.uuid, char_results[i].uuid); + characteristic_resp.handle = char_results[i].char_handle; + characteristic_resp.properties = char_results[i].properties; - // Get the number of descriptors directly with one call - uint16_t total_desc_count = 0; - esp_gatt_status_t desc_count_status = - esp_ble_gattc_get_attr_count(this->gattc_if_, this->conn_id_, ESP_GATT_DB_DESCRIPTOR, char_result.char_handle, - service_result.end_handle, 0, &total_desc_count); - - if (desc_count_status != ESP_GATT_OK) { - ESP_LOGW(TAG, "[%d] [%s] Error getting descriptor count for char handle %d, status=%d", this->connection_index_, - this->address_str().c_str(), char_result.char_handle, desc_count_status); - return; + // Process descriptors for this characteristic + if (!this->process_descriptors_for_characteristic_(char_results[i].char_handle, service_end_handle, + characteristic_resp)) { + return false; // Error processing descriptors } - - if (total_desc_count > 0) { - // Reserve space and process descriptors - characteristic_resp.descriptors.reserve(total_desc_count); - uint16_t desc_offset = 0; - esp_gattc_descr_elem_t desc_result; - while (true) { // descriptors - uint16_t desc_count = 1; - esp_gatt_status_t desc_status = esp_ble_gattc_get_all_descr( - this->gattc_if_, this->conn_id_, char_result.char_handle, &desc_result, &desc_count, desc_offset); - if (desc_status == ESP_GATT_INVALID_OFFSET || desc_status == ESP_GATT_NOT_FOUND || desc_count == 0) { - break; - } else if (desc_status != ESP_GATT_OK) { - ESP_LOGE(TAG, "[%d] [%s] esp_ble_gattc_get_all_descr error, status=%d", this->connection_index_, - this->address_str().c_str(), desc_status); - return; - } - - characteristic_resp.descriptors.emplace_back(); - auto &descriptor_resp = characteristic_resp.descriptors.back(); - fill_128bit_uuid_array(descriptor_resp.uuid, desc_result.uuid); - descriptor_resp.handle = desc_result.handle; - desc_offset++; - } - } // end else if (total_desc_count > 0) } - } // end else if (total_char_count > 0) - // Send the message (we already checked api_conn is not null at the beginning) - api_conn->send_message(resp, api::BluetoothGATTGetServicesResponse::MESSAGE_TYPE); + char_offset += char_count; + } + + return true; +} + +bool BluetoothConnection::process_descriptors_for_characteristic_( + uint16_t char_handle, uint16_t service_end_handle, api::BluetoothGATTCharacteristic &characteristic_resp) { + // Get the number of descriptors directly with one call + uint16_t total_desc_count = 0; + esp_gatt_status_t desc_count_status = esp_ble_gattc_get_attr_count( + this->gattc_if_, this->conn_id_, ESP_GATT_DB_DESCRIPTOR, char_handle, service_end_handle, 0, &total_desc_count); + + if (desc_count_status != ESP_GATT_OK) { + ESP_LOGW(TAG, "[%d] [%s] Error getting descriptor count for char handle %d, status=%d", this->connection_index_, + this->address_str().c_str(), char_handle, desc_count_status); + return false; + } + + if (total_desc_count == 0) { + return true; // No descriptors, which is valid + } + + // Reserve space and process descriptors + characteristic_resp.descriptors.reserve(total_desc_count); + + // Use a reasonable fixed-size buffer on the stack + constexpr uint8_t MAX_DESCRIPTORS_PER_BATCH = 8; + esp_gattc_descr_elem_t desc_results[MAX_DESCRIPTORS_PER_BATCH]; + uint16_t desc_offset = 0; + + while (desc_offset < total_desc_count) { + uint16_t desc_count = std::min(static_cast(MAX_DESCRIPTORS_PER_BATCH), + static_cast(total_desc_count - desc_offset)); + + esp_gatt_status_t desc_status = esp_ble_gattc_get_all_descr(this->gattc_if_, this->conn_id_, char_handle, + desc_results, &desc_count, desc_offset); + + if (desc_status != ESP_GATT_OK) { + ESP_LOGE(TAG, "[%d] [%s] esp_ble_gattc_get_all_descr error, status=%d", this->connection_index_, + this->address_str().c_str(), desc_status); + return false; + } + + if (desc_count == 0) { + break; // No more descriptors + } + + // Process this batch of descriptors + for (uint8_t i = 0; i < desc_count; i++) { + characteristic_resp.descriptors.emplace_back(); + auto &descriptor_resp = characteristic_resp.descriptors.back(); + fill_128bit_uuid_array(descriptor_resp.uuid, desc_results[i].uuid); + descriptor_resp.handle = desc_results[i].handle; + } + + desc_offset += desc_count; + } + + return true; } bool BluetoothConnection::gattc_event_handler(esp_gattc_cb_event_t event, esp_gatt_if_t gattc_if, diff --git a/esphome/components/bluetooth_proxy/bluetooth_connection.h b/esphome/components/bluetooth_proxy/bluetooth_connection.h index 3fed9d531f..6f3bd169a3 100644 --- a/esphome/components/bluetooth_proxy/bluetooth_connection.h +++ b/esphome/components/bluetooth_proxy/bluetooth_connection.h @@ -3,6 +3,7 @@ #ifdef USE_ESP32 #include "esphome/components/esp32_ble_client/ble_client_base.h" +#include "esphome/components/api/api_pb2.h" namespace esphome::bluetooth_proxy { @@ -28,6 +29,10 @@ class BluetoothConnection : public esp32_ble_client::BLEClientBase { friend class BluetoothProxy; void send_service_for_discovery_(); + bool process_characteristics_for_service_(uint16_t service_start_handle, uint16_t service_end_handle, + api::BluetoothGATTService &service_resp); + bool process_descriptors_for_characteristic_(uint16_t char_handle, uint16_t service_end_handle, + api::BluetoothGATTCharacteristic &characteristic_resp); void reset_connection_(esp_err_t reason); // Memory optimized layout for 32-bit systems