113 lines
4.5 KiB
C++
113 lines
4.5 KiB
C++
#include "Radio.h"
|
|
#include "Config.h"
|
|
#include "Log.h"
|
|
#include <WiFi.h>
|
|
#include <esp_wifi.h>
|
|
#include <string.h>
|
|
|
|
Radio *Radio::instance_ = nullptr;
|
|
static const uint8_t BROADCAST_MAC[6] = {0xff, 0xff, 0xff, 0xff, 0xff, 0xff};
|
|
|
|
bool Radio::begin() {
|
|
if (active_) { Log::event("ESP-NOW", "already active"); return true; }
|
|
// Keep both devices on a dedicated, fixed STA channel. Stored access-point
|
|
// credentials must not reconnect in the background and move ESP-NOW to the
|
|
// AP's channel.
|
|
WiFi.persistent(false);
|
|
WiFi.mode(WIFI_STA);
|
|
WiFi.setAutoReconnect(false);
|
|
WiFi.disconnect(false, false);
|
|
WiFi.setSleep(false);
|
|
const bool txPowerOk = WiFi.setTxPower(WIFI_POWER_8_5dBm);
|
|
delay(10);
|
|
if (esp_wifi_set_channel(ESPNOW_WIFI_CHANNEL, WIFI_SECOND_CHAN_NONE) != ESP_OK) {
|
|
Log::event("ESP-NOW", "Wi-Fi channel setup FAILED"); return false;
|
|
}
|
|
queue_ = xQueueCreate(8, sizeof(ReceivedPacket));
|
|
if (!queue_) { Log::event("ESP-NOW", "receive queue creation FAILED"); return false; }
|
|
if (esp_now_init() != ESP_OK) {
|
|
vQueueDelete(queue_); queue_ = nullptr;
|
|
Log::event("ESP-NOW", "initialization FAILED"); return false;
|
|
}
|
|
const esp_err_t rateResult = esp_wifi_config_espnow_rate(WIFI_IF_STA, WIFI_PHY_RATE_1M_L);
|
|
instance_ = this;
|
|
if (esp_now_register_recv_cb(onReceive) != ESP_OK) { end(); return false; }
|
|
active_ = true;
|
|
const bool ok = ensurePeer(BROADCAST_MAC);
|
|
uint8_t primaryChannel = 0;
|
|
wifi_second_chan_t secondaryChannel = WIFI_SECOND_CHAN_NONE;
|
|
esp_wifi_get_channel(&primaryChannel, &secondaryChannel);
|
|
Log::printf("ESP-NOW", "started channel=%u expected=%u rate=1M%s tx-power=max%s broadcast-peer=%s",
|
|
primaryChannel, ESPNOW_WIFI_CHANNEL, rateResult == ESP_OK ? "" : "-FAILED",
|
|
txPowerOk ? "" : "-FAILED", ok ? "OK" : "FAILED");
|
|
return ok;
|
|
}
|
|
|
|
void Radio::end() {
|
|
if (active_) { esp_now_unregister_recv_cb(); esp_now_deinit(); }
|
|
if (queue_) { vQueueDelete(queue_); queue_ = nullptr; }
|
|
active_ = false; if (instance_ == this) instance_ = nullptr;
|
|
Log::event("ESP-NOW", "stopped");
|
|
}
|
|
|
|
bool Radio::ensurePeer(const uint8_t mac[6]) {
|
|
if (esp_now_is_peer_exist(mac)) return true;
|
|
esp_now_peer_info_t peer = {};
|
|
memcpy(peer.peer_addr, mac, 6);
|
|
peer.channel = ESPNOW_WIFI_CHANNEL;
|
|
peer.ifidx = WIFI_IF_STA;
|
|
peer.encrypt = false;
|
|
return esp_now_add_peer(&peer) == ESP_OK;
|
|
}
|
|
|
|
bool Radio::sendBroadcast(ProtocolPacket p) { return sendTo(BROADCAST_MAC, p); }
|
|
|
|
bool Radio::sendTo(const uint8_t mac[6], ProtocolPacket p) {
|
|
maintainChannel();
|
|
char peer[20]; macText(mac, peer, sizeof(peer));
|
|
if (!active_ || !ensurePeer(mac)) {
|
|
Log::printf("ESP-NOW", "TX %s to %s FAILED: inactive/peer", messageName(static_cast<MessageType>(p.type)), peer);
|
|
return false;
|
|
}
|
|
finalizePacket(p);
|
|
const bool ok = esp_now_send(mac, reinterpret_cast<const uint8_t *>(&p), sizeof(p)) == ESP_OK;
|
|
const MessageType type = static_cast<MessageType>(p.type);
|
|
if (type != MessageType::HEARTBEAT && type != MessageType::HEARTBEAT_ACK)
|
|
Log::printf("ESP-NOW", "TX %s to %s session=%08lX stage=%u seq=%u %s",
|
|
messageName(type), peer, p.session, p.stage, p.sequence, ok ? "QUEUED" : "FAILED");
|
|
return ok;
|
|
}
|
|
|
|
bool Radio::receive(ReceivedPacket &r) {
|
|
maintainChannel();
|
|
return queue_ && xQueueReceive(queue_, &r, 0) == pdTRUE;
|
|
}
|
|
|
|
void Radio::maintainChannel() {
|
|
if (!active_) return;
|
|
const uint32_t now = millis();
|
|
if (now - lastChannelCheckMs_ < 250) return;
|
|
lastChannelCheckMs_ = now;
|
|
uint8_t primaryChannel = 0;
|
|
wifi_second_chan_t secondaryChannel = WIFI_SECOND_CHAN_NONE;
|
|
if (esp_wifi_get_channel(&primaryChannel, &secondaryChannel) != ESP_OK ||
|
|
primaryChannel == ESPNOW_WIFI_CHANNEL) return;
|
|
const bool restored = esp_wifi_set_channel(ESPNOW_WIFI_CHANNEL, WIFI_SECOND_CHAN_NONE) == ESP_OK;
|
|
Log::printf("ESP-NOW", "channel drift %u->%u %s", primaryChannel, ESPNOW_WIFI_CHANNEL,
|
|
restored ? "RESTORED" : "FAILED");
|
|
}
|
|
|
|
void Radio::flush() { if (queue_) xQueueReset(queue_); }
|
|
|
|
void Radio::onReceive(const esp_now_recv_info_t *info, const uint8_t *data, int length) {
|
|
if (!instance_ || !instance_->queue_ || !info || length != sizeof(ProtocolPacket)) return;
|
|
ReceivedPacket item;
|
|
memcpy(item.mac, info->src_addr, 6); memcpy(&item.packet, data, sizeof(item.packet));
|
|
if (!validPacket(item.packet)) return;
|
|
xQueueSend(instance_->queue_, &item, 0); // Wi-Fi task callback: copy only, never block
|
|
}
|
|
|
|
void Radio::macText(const uint8_t mac[6], char *out, size_t n) {
|
|
snprintf(out, n, "%02X:%02X:%02X:%02X:%02X:%02X", mac[0], mac[1], mac[2], mac[3], mac[4], mac[5]);
|
|
}
|