mirror of
https://github.com/meshcore-dev/MeshCore.git
synced 2026-07-23 23:53:07 +00:00
fix: synchronize BLE receive queue
This commit is contained in:
@@ -0,0 +1,39 @@
|
||||
#pragma once
|
||||
|
||||
#include <stddef.h>
|
||||
|
||||
namespace mesh {
|
||||
|
||||
template <typename T, size_t Capacity>
|
||||
class BoundedQueue {
|
||||
static_assert(Capacity > 0, "BoundedQueue capacity must be greater than zero");
|
||||
|
||||
T _items[Capacity];
|
||||
size_t _head = 0;
|
||||
size_t _size = 0;
|
||||
|
||||
public:
|
||||
bool push(const T& item) {
|
||||
if (_size == Capacity) return false;
|
||||
_items[(_head + _size) % Capacity] = item;
|
||||
_size++;
|
||||
return true;
|
||||
}
|
||||
|
||||
bool pop(T& item) {
|
||||
if (_size == 0) return false;
|
||||
item = _items[_head];
|
||||
_head = (_head + 1) % Capacity;
|
||||
_size--;
|
||||
return true;
|
||||
}
|
||||
|
||||
void clear() {
|
||||
_head = 0;
|
||||
_size = 0;
|
||||
}
|
||||
|
||||
size_t size() const { return _size; }
|
||||
};
|
||||
|
||||
} // namespace mesh
|
||||
@@ -118,17 +118,30 @@ void SerialBLEInterface::onWrite(BLECharacteristic* pCharacteristic, esp_ble_gat
|
||||
|
||||
if (len > MAX_FRAME_SIZE) {
|
||||
BLE_DEBUG_PRINTLN("ERROR: onWrite(), frame too big, len=%d", len);
|
||||
} else if (recv_queue_len >= FRAME_QUEUE_SIZE) {
|
||||
BLE_DEBUG_PRINTLN("ERROR: onWrite(), recv_queue is full!");
|
||||
} else {
|
||||
recv_queue[recv_queue_len].len = len;
|
||||
memcpy(recv_queue[recv_queue_len].buf, rxValue, len);
|
||||
recv_queue_len++;
|
||||
Frame frame = {};
|
||||
frame.len = len;
|
||||
memcpy(frame.buf, rxValue, len);
|
||||
|
||||
portENTER_CRITICAL(&recv_queue_mux);
|
||||
bool queued = recv_queue.push(frame);
|
||||
portEXIT_CRITICAL(&recv_queue_mux);
|
||||
|
||||
if (!queued) {
|
||||
BLE_DEBUG_PRINTLN("ERROR: onWrite(), recv_queue is full!");
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// ---------- public methods
|
||||
|
||||
void SerialBLEInterface::clearBuffers() {
|
||||
portENTER_CRITICAL(&recv_queue_mux);
|
||||
recv_queue.clear();
|
||||
portEXIT_CRITICAL(&recv_queue_mux);
|
||||
send_queue_len = 0;
|
||||
}
|
||||
|
||||
void SerialBLEInterface::enable() {
|
||||
if (_isEnabled) return;
|
||||
|
||||
@@ -202,17 +215,15 @@ size_t SerialBLEInterface::checkRecvFrame(uint8_t dest[]) {
|
||||
}
|
||||
}
|
||||
|
||||
if (recv_queue_len > 0) { // check recv queue
|
||||
size_t len = recv_queue[0].len; // take from top of queue
|
||||
memcpy(dest, recv_queue[0].buf, len);
|
||||
Frame frame = {};
|
||||
portENTER_CRITICAL(&recv_queue_mux);
|
||||
bool received = recv_queue.pop(frame);
|
||||
portEXIT_CRITICAL(&recv_queue_mux);
|
||||
|
||||
BLE_DEBUG_PRINTLN("readBytes: sz=%d, hdr=%d", len, (uint32_t) dest[0]);
|
||||
|
||||
recv_queue_len--;
|
||||
for (int i = 0; i < recv_queue_len; i++) { // delete top item from queue
|
||||
recv_queue[i] = recv_queue[i + 1];
|
||||
}
|
||||
return len;
|
||||
if (received) {
|
||||
memcpy(dest, frame.buf, frame.len);
|
||||
BLE_DEBUG_PRINTLN("readBytes: sz=%d, hdr=%d", (uint32_t) frame.len, (uint32_t) dest[0]);
|
||||
return frame.len;
|
||||
}
|
||||
|
||||
if (pServer->getConnectedCount() == 0) deviceConnected = false;
|
||||
|
||||
@@ -1,10 +1,12 @@
|
||||
#pragma once
|
||||
|
||||
#include "../BaseSerialInterface.h"
|
||||
#include "../BoundedQueue.h"
|
||||
#include <BLEDevice.h>
|
||||
#include <BLEServer.h>
|
||||
#include <BLEUtils.h>
|
||||
#include <BLE2902.h>
|
||||
#include <freertos/FreeRTOS.h>
|
||||
|
||||
class SerialBLEInterface : public BaseSerialInterface, BLESecurityCallbacks, BLEServerCallbacks, BLECharacteristicCallbacks {
|
||||
BLEServer *pServer;
|
||||
@@ -24,12 +26,12 @@ class SerialBLEInterface : public BaseSerialInterface, BLESecurityCallbacks, BLE
|
||||
};
|
||||
|
||||
#define FRAME_QUEUE_SIZE 4
|
||||
int recv_queue_len;
|
||||
Frame recv_queue[FRAME_QUEUE_SIZE];
|
||||
portMUX_TYPE recv_queue_mux = portMUX_INITIALIZER_UNLOCKED;
|
||||
mesh::BoundedQueue<Frame, FRAME_QUEUE_SIZE> recv_queue;
|
||||
int send_queue_len;
|
||||
Frame send_queue[FRAME_QUEUE_SIZE];
|
||||
|
||||
void clearBuffers() { recv_queue_len = 0; send_queue_len = 0; }
|
||||
void clearBuffers();
|
||||
|
||||
protected:
|
||||
// BLESecurityCallbacks methods
|
||||
@@ -58,7 +60,7 @@ public:
|
||||
_isEnabled = false;
|
||||
_last_write = 0;
|
||||
last_conn_id = 0;
|
||||
send_queue_len = recv_queue_len = 0;
|
||||
send_queue_len = 0;
|
||||
}
|
||||
|
||||
/**
|
||||
|
||||
@@ -0,0 +1,58 @@
|
||||
#include <gtest/gtest.h>
|
||||
|
||||
#include <helpers/BoundedQueue.h>
|
||||
|
||||
TEST(BoundedQueue, PreservesOrderAcrossWraparound) {
|
||||
mesh::BoundedQueue<int, 4> queue;
|
||||
int value = 0;
|
||||
|
||||
for (int i = 0; i < 4; i++) EXPECT_TRUE(queue.push(i));
|
||||
EXPECT_FALSE(queue.push(4));
|
||||
|
||||
EXPECT_TRUE(queue.pop(value));
|
||||
EXPECT_EQ(0, value);
|
||||
EXPECT_TRUE(queue.pop(value));
|
||||
EXPECT_EQ(1, value);
|
||||
|
||||
EXPECT_TRUE(queue.push(4));
|
||||
EXPECT_TRUE(queue.push(5));
|
||||
|
||||
for (int expected = 2; expected < 6; expected++) {
|
||||
EXPECT_TRUE(queue.pop(value));
|
||||
EXPECT_EQ(expected, value);
|
||||
}
|
||||
EXPECT_FALSE(queue.pop(value));
|
||||
}
|
||||
|
||||
TEST(BoundedQueue, SurvivesRepeatedFillAndDrainCycles) {
|
||||
mesh::BoundedQueue<int, 4> queue;
|
||||
int value = 0;
|
||||
|
||||
for (int cycle = 0; cycle < 1000; cycle++) {
|
||||
for (int i = 0; i < 4; i++) EXPECT_TRUE(queue.push(cycle * 4 + i));
|
||||
for (int i = 0; i < 4; i++) {
|
||||
EXPECT_TRUE(queue.pop(value));
|
||||
EXPECT_EQ(cycle * 4 + i, value);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
TEST(BoundedQueue, ClearDropsPendingItems) {
|
||||
mesh::BoundedQueue<int, 4> queue;
|
||||
int value = 0;
|
||||
|
||||
EXPECT_TRUE(queue.push(1));
|
||||
EXPECT_TRUE(queue.push(2));
|
||||
queue.clear();
|
||||
|
||||
EXPECT_EQ(0u, queue.size());
|
||||
EXPECT_FALSE(queue.pop(value));
|
||||
EXPECT_TRUE(queue.push(3));
|
||||
EXPECT_TRUE(queue.pop(value));
|
||||
EXPECT_EQ(3, value);
|
||||
}
|
||||
|
||||
int main(int argc, char **argv) {
|
||||
::testing::InitGoogleTest(&argc, argv);
|
||||
return RUN_ALL_TESTS();
|
||||
}
|
||||
Reference in New Issue
Block a user