mirror of https://github.com/esphome/esphome.git
109 lines
2.9 KiB
C++
109 lines
2.9 KiB
C++
#pragma once
|
|
|
|
#ifdef USE_ESP32
|
|
#include "esphome/core/component.h"
|
|
#include "esphome/core/helpers.h"
|
|
|
|
#include <queue>
|
|
#include <mutex>
|
|
#include <cstring>
|
|
|
|
#include <esp_gap_ble_api.h>
|
|
#include <esp_gattc_api.h>
|
|
#include <freertos/FreeRTOS.h>
|
|
#include <freertos/semphr.h>
|
|
|
|
/*
|
|
* BLE events come in from a separate Task (thread) in the ESP32 stack. Rather
|
|
* than trying to deal with various locking strategies, all incoming GAP and GATT
|
|
* events will simply be placed on a semaphore guarded queue. The next time the
|
|
* component runs loop(), these events are popped off the queue and handed at
|
|
* this safer time.
|
|
*/
|
|
|
|
namespace esphome {
|
|
namespace esp32_ble_tracker {
|
|
|
|
template<class T> class Queue {
|
|
public:
|
|
Queue() { m_ = xSemaphoreCreateMutex(); }
|
|
|
|
void push(T *element) {
|
|
if (element == nullptr)
|
|
return;
|
|
if (xSemaphoreTake(m_, 5L / portTICK_PERIOD_MS)) {
|
|
q_.push(element);
|
|
xSemaphoreGive(m_);
|
|
}
|
|
}
|
|
|
|
T *pop() {
|
|
T *element = nullptr;
|
|
|
|
if (xSemaphoreTake(m_, 5L / portTICK_PERIOD_MS)) {
|
|
if (!q_.empty()) {
|
|
element = q_.front();
|
|
q_.pop();
|
|
}
|
|
xSemaphoreGive(m_);
|
|
}
|
|
return element;
|
|
}
|
|
|
|
protected:
|
|
std::queue<T *> q_;
|
|
SemaphoreHandle_t m_;
|
|
};
|
|
|
|
// Received GAP and GATTC events are only queued, and get processed in the main loop().
|
|
// This class stores each event in a single type.
|
|
class BLEEvent {
|
|
public:
|
|
BLEEvent(esp_gap_ble_cb_event_t e, esp_ble_gap_cb_param_t *p) {
|
|
this->event_.gap.gap_event = e;
|
|
memcpy(&this->event_.gap.gap_param, p, sizeof(esp_ble_gap_cb_param_t));
|
|
this->type_ = 0;
|
|
};
|
|
|
|
BLEEvent(esp_gattc_cb_event_t e, esp_gatt_if_t i, esp_ble_gattc_cb_param_t *p) {
|
|
this->event_.gattc.gattc_event = e;
|
|
this->event_.gattc.gattc_if = i;
|
|
memcpy(&this->event_.gattc.gattc_param, p, sizeof(esp_ble_gattc_cb_param_t));
|
|
// Need to also make a copy of relevant event data.
|
|
switch (e) {
|
|
case ESP_GATTC_NOTIFY_EVT:
|
|
this->data.assign(p->notify.value, p->notify.value + p->notify.value_len);
|
|
this->event_.gattc.gattc_param.notify.value = this->data.data();
|
|
break;
|
|
case ESP_GATTC_READ_CHAR_EVT:
|
|
case ESP_GATTC_READ_DESCR_EVT:
|
|
this->data.assign(p->read.value, p->read.value + p->read.value_len);
|
|
this->event_.gattc.gattc_param.read.value = this->data.data();
|
|
break;
|
|
default:
|
|
break;
|
|
}
|
|
this->type_ = 1;
|
|
};
|
|
|
|
union {
|
|
struct gap_event { // NOLINT(readability-identifier-naming)
|
|
esp_gap_ble_cb_event_t gap_event;
|
|
esp_ble_gap_cb_param_t gap_param;
|
|
} gap;
|
|
|
|
struct gattc_event { // NOLINT(readability-identifier-naming)
|
|
esp_gattc_cb_event_t gattc_event;
|
|
esp_gatt_if_t gattc_if;
|
|
esp_ble_gattc_cb_param_t gattc_param;
|
|
} gattc;
|
|
} event_;
|
|
std::vector<uint8_t> data{};
|
|
uint8_t type_; // 0=gap 1=gattc
|
|
};
|
|
|
|
} // namespace esp32_ble_tracker
|
|
} // namespace esphome
|
|
|
|
#endif
|