ble_device.h

Go to the documentation of this file.

// ble_device.h
//
// Platform-neutral BLE advertisement types — the generic base every BLE consumer
// (sensor components, bluetooth_proxy, automation triggers) builds against:
//   ESPBTUUID / ServiceData / ESPBLEiBeacon / ESPBTDevice / ESPBTDeviceListener
//
// These types are owned here on EVERY platform, with no chip-SDK types in their
// public surface. Platform trackers produce them:
//   - esp32_ble_tracker adapts ESP-IDF scan results into ESPBTDevice and
//     re-exports these names (esp32 only) for backward compatibility;
//   - the LibreTiny trackers (bk72xx / ln882h) feed from_scan_result() directly.
 
#pragma once
 
#include "esphome/core/defines.h"
#include "esphome/core/helpers.h"
 
#include <cstdint>
#include <cstring>
#include <initializer_list>
#include <string>
#include <vector>
 
#if defined(__cpp_lib_span)
#include <span>
#endif
 
#ifdef USE_ESP32
// Historical esp32_ble API surface (below, under the same define) uses the
// ESP-IDF UUID/address/scan-result types directly; never referenced off-esp32.
#include "esphome/components/esp32_ble/ble_scan_result.h"
#include <esp_bt_defs.h>
#endif
 
namespace esphome::ble_device_base {
 
using adv_data_t = std::vector<uint8_t>;
 
// Bluetooth Core address types (spec values; matches ESP-IDF's esp_ble_addr_type_t).
static constexpr uint8_t BLE_ADDR_TYPE_PUBLIC = 0;
static constexpr uint8_t BLE_ADDR_TYPE_RANDOM = 1;
static constexpr uint8_t BLE_ADDR_TYPE_RPA_PUBLIC = 2;
static constexpr uint8_t BLE_ADDR_TYPE_RPA_RANDOM = 3;
 
static constexpr size_t UUID_STR_LEN = 37;
 
// ---------------------------------------------------------------------------
// ESPBTUUID — 16/32/128-bit Bluetooth UUID value type.
// API-compatible with the historical esp32_ble::ESPBTUUID; the esp_bt_uuid_t
// conversions live in esp32_ble (esp32-only adapters), not here.
// ---------------------------------------------------------------------------
 
class ESPBTUUID {
 public:
  ESPBTUUID() = default;
 
  static ESPBTUUID from_uint16(uint16_t uuid);
  static ESPBTUUID from_uint32(uint32_t uuid);
  static ESPBTUUID from_raw(const uint8_t *data);
  static ESPBTUUID from_raw_reversed(const uint8_t *data);
  static ESPBTUUID from_raw(const char *data, size_t length);
  static ESPBTUUID from_raw(const char *data) { return from_raw(data, strlen(data)); }
  static ESPBTUUID from_raw(const std::string &data) { return from_raw(data.c_str(), data.length()); }
  static ESPBTUUID from_raw(std::initializer_list<uint8_t> data) {
    return from_raw(reinterpret_cast<const char *>(data.begin()), data.size());
  }
 
#ifdef USE_ESP32
  static ESPBTUUID from_uuid(esp_bt_uuid_t uuid);
  esp_bt_uuid_t get_uuid() const;
#endif
 
  ESPBTUUID as_128bit() const;
 
  bool contains(uint8_t data1, uint8_t data2) const;
 
  bool operator==(const ESPBTUUID &other) const;
  bool operator!=(const ESPBTUUID &other) const { return !(*this == other); }
 
  const char *to_str(char *buf) const;
#if defined(__cpp_lib_span)
  const char *to_str(std::span<char, UUID_STR_LEN> output) const { return this->to_str(output.data()); }
#endif
  enum class Type : uint8_t { UUID16, UUID32, UUID128 };
  Type type() const { return this->type_; }
  uint16_t uuid16() const { return this->uuid_.uuid16; }
  uint32_t uuid32() const { return this->uuid_.uuid32; }
  const uint8_t *uuid128() const { return this->uuid_.uuid128; }
 
 protected:
  // Expand to the 128-bit Bluetooth Base UUID byte form (out is 16 bytes, little-endian).
  void to_128bit_(uint8_t out[16]) const;
 
  Type type_{Type::UUID16};
  union {
    uint16_t uuid16;
    uint32_t uuid32;
    uint8_t uuid128[16];
  } uuid_{};
};
 
// ---------------------------------------------------------------------------
// ServiceData — UUID-tagged advertisement payload (0x16 / 0xFF AD types)
// ---------------------------------------------------------------------------
 
struct ServiceData {
  ESPBTUUID uuid;
  adv_data_t data;
};
 
// ---------------------------------------------------------------------------
// ESPBLEiBeacon
// ---------------------------------------------------------------------------
 
class ESPBLEiBeacon {
 public:
  ESPBLEiBeacon() { memset(&this->beacon_data_, 0, sizeof(this->beacon_data_)); }
  explicit ESPBLEiBeacon(const uint8_t *data);
  static optional<ESPBLEiBeacon> from_manufacturer_data(const ServiceData &data);
 
  uint16_t get_major() const { return byteswap(this->beacon_data_.major); }
  uint16_t get_minor() const { return byteswap(this->beacon_data_.minor); }
  int8_t get_signal_power() const { return this->beacon_data_.signal_power; }
  ESPBTUUID get_uuid() const { return ESPBTUUID::from_raw_reversed(this->beacon_data_.proximity_uuid); }
 
 protected:
  struct PACKED BeaconData {
    uint8_t sub_type;
    uint8_t length;
    uint8_t proximity_uuid[16];
    uint16_t major;
    uint16_t minor;
    int8_t signal_power;
  } beacon_data_;
};
 
// ---------------------------------------------------------------------------
// ESPBTDevice — parsed BLE advertisement
// ---------------------------------------------------------------------------
 
class ESPBTDevice {
 public:
  void from_scan_result(const uint8_t *mac, int rssi, uint8_t addr_type, const uint8_t *data, uint16_t data_len);
 
  // Alias the core constant so the two cannot drift apart.
  static constexpr size_t MAC_ADDRESS_PRETTY_BUFFER_SIZE = esphome::MAC_ADDRESS_PRETTY_BUFFER_SIZE;
 
  std::string address_str() const;
  const char *address_str_to(char *buf) const;
#if defined(__cpp_lib_span)
  const char *address_str_to(std::span<char, MAC_ADDRESS_PRETTY_BUFFER_SIZE> buf) const {
    return this->address_str_to(buf.data());
  }
#endif
  uint64_t address_uint64() const;
  const uint8_t *address() const { return address_; }
#ifdef USE_ESP32
  // Historical esp32 signature: consumers assign the result to esp_ble_addr_type_t.
  esp_ble_addr_type_t get_address_type() const { return static_cast<esp_ble_addr_type_t>(this->address_type_); }
  void parse_scan_rst(const esp32_ble::BLEScanResult &scan_result);
  // Exposed through a function for use in lambdas
  const esp32_ble::BLEScanResult &get_scan_result() const { return *scan_result_; }
#else
  uint8_t get_address_type() const { return this->address_type_; }
#endif
 
  int get_rssi() const { return rssi_; }
  const std::string &get_name() const { return name_; }
 
  const std::vector<ESPBTUUID> &get_service_uuids() const { return service_uuids_; }
  const std::vector<ServiceData> &get_manufacturer_datas() const { return manufacturer_datas_; }
  const std::vector<ServiceData> &get_service_datas() const { return service_datas_; }
  const std::vector<int8_t> &get_tx_powers() const { return tx_powers_; }
  const optional<uint16_t> &get_appearance() const { return appearance_; }
  const optional<uint8_t> &get_ad_flag() const { return ad_flag_; }
 
  bool resolve_irk(const uint8_t *irk) const;
 
  optional<ESPBLEiBeacon> get_ibeacon() const {
    for (const auto &it : this->manufacturer_datas_) {
      auto res = ESPBLEiBeacon::from_manufacturer_data(it);
      if (res.has_value())
        return res;
    }
    return {};
  }
 
 protected:
  void parse_adv_(const uint8_t *payload, uint16_t len);
 
  uint8_t address_[6]{0};
  uint8_t address_type_{0};
  int rssi_{0};
  std::string name_{};
  std::vector<ESPBTUUID> service_uuids_{};
  std::vector<ServiceData> manufacturer_datas_{};
  std::vector<ServiceData> service_datas_{};
#ifdef USE_ESP32
  const esp32_ble::BLEScanResult *scan_result_{nullptr};
#endif
  std::vector<int8_t> tx_powers_{};
  optional<uint16_t> appearance_{};
  optional<uint8_t> ad_flag_{};
};
 
// ---------------------------------------------------------------------------
// ESPBTDeviceListener — base class for BLE consumers (sensors, proxy, triggers)
// ---------------------------------------------------------------------------
 
class ESPBTDeviceListener {
 public:
  virtual ~ESPBTDeviceListener() = default;
  virtual void on_scan_end() {}
  virtual bool parse_device(const ESPBTDevice &device) = 0;
};
 
}  // namespace esphome::ble_device_base