From 9a82127c476bf99556bbc69e181025afcba19d26 Mon Sep 17 00:00:00 2001 From: nodist Date: Thu, 24 Sep 2026 14:36:15 -0400 Subject: ring buffer tests --- source/queue.c | 57 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 57 insertions(+) (limited to 'source/queue.c') diff --git a/source/queue.c b/source/queue.c index 519aa1f..6f36ce8 100644 --- a/source/queue.c +++ b/source/queue.c @@ -40,3 +40,60 @@ kpl_task *kpl_task_queue_next(kpl_task_queue *task_queue) { __atomic_store_n(&task->next, NULL, __ATOMIC_RELAXED); return task; } + +bool kpl_ring_queue_init(kpl_ring_queue *ring_queue, int8_t size_pow2) { + if (size_pow2 < KPL_RING_QUEUE_MIN_POW2 || size_pow2 > KPL_RING_QUEUE_MAX_POW2) + return false; + size_t size = 1L << size_pow2; + ring_queue->items = calloc(1, sizeof(kpl_ring_queue_item) * size); + ring_queue->items_mask = size - 1; + __atomic_store_n(&ring_queue->head_index, 0, __ATOMIC_RELAXED); + __atomic_store_n(&ring_queue->tail_index, 0, __ATOMIC_RELAXED); + for (size_t index = 0; index < size; index++) + __atomic_store_n(&ring_queue->items[index].index, index, __ATOMIC_RELAXED); + return true; +} + +void kpl_ring_queue_free(kpl_ring_queue *ring_queue) { + free(ring_queue->items); +} + +bool kpl_ring_queue_add(kpl_ring_queue *ring_queue, void *data) { + for (;;) { + ssize_t tail_index = __atomic_load_n(&ring_queue->tail_index, __ATOMIC_RELAXED); + kpl_ring_queue_item *item = &ring_queue->items[tail_index & ring_queue->items_mask]; + ssize_t item_index = __atomic_load_n(&item->index, __ATOMIC_ACQUIRE), diff = item_index - tail_index; + if (diff > 0) + continue; + if (diff < 0) + break; + if (!__atomic_compare_exchange_n(&ring_queue->tail_index, &tail_index, tail_index + 1, true, + __ATOMIC_RELAXED, __ATOMIC_RELAXED)) + continue; + item->data = data; + __atomic_store_n(&item->index, tail_index + 1, __ATOMIC_RELEASE); + return true; + } + return false; +} + +void *kpl_ring_queue_next(kpl_ring_queue *ring_queue) { + for (;;) { + ssize_t head_index = __atomic_load_n(&ring_queue->head_index, __ATOMIC_RELAXED); + kpl_ring_queue_item *item = &ring_queue->items[head_index & ring_queue->items_mask]; + ssize_t item_index = __atomic_load_n(&item->index, __ATOMIC_ACQUIRE), + diff = item_index - (head_index + 1); + if (diff > 0) + continue; + if (diff < 0) + break; + if (!__atomic_compare_exchange_n(&ring_queue->head_index, &head_index, head_index + 1, true, + __ATOMIC_RELAXED, __ATOMIC_RELAXED)) + continue; + void *data = item->data; + item->data = NULL; + __atomic_store_n(&item->index, head_index + ring_queue->items_mask + 1, __ATOMIC_RELEASE); + return data; + } + return NULL; +} -- cgit v1.2.3