Udp

common/network/webudp/WuPool.cpp

@@ -0,0 +1,60 @@
+#include "WuPool.h"
+#include <assert.h>
+#include <stdlib.h>
+
+struct BlockHeader {
+  int32_t index;
+};
+
+struct WuPool {
+  int32_t slotSize;
+  int32_t numBytes;
+  int32_t numBlocks;
+  uint8_t* memory;
+  int32_t freeIndicesCount;
+  int32_t* freeIndices;
+};
+
+WuPool* WuPoolCreate(int32_t blockSize, int32_t numBlocks) {
+  WuPool* pool = (WuPool*)calloc(1, sizeof(WuPool));
+
+  pool->slotSize = blockSize + sizeof(BlockHeader);
+  pool->numBytes = pool->slotSize * numBlocks;
+  pool->numBlocks = numBlocks;
+  pool->memory = (uint8_t*)calloc(pool->numBytes, 1);
+  pool->freeIndicesCount = numBlocks;
+  pool->freeIndices = (int32_t*)calloc(numBlocks, sizeof(int32_t));
+
+  for (int32_t i = 0; i < numBlocks; i++) {
+    pool->freeIndices[i] = numBlocks - i - 1;
+  }
+
+  return pool;
+}
+
+void WuPoolDestroy(WuPool* pool) {
+  free(pool->memory);
+  free(pool->freeIndices);
+  free(pool);
+}
+
+void* WuPoolAcquire(WuPool* pool) {
+  if (pool->freeIndicesCount == 0) return NULL;
+
+  const int32_t index = pool->freeIndices[pool->freeIndicesCount - 1];
+  pool->freeIndicesCount--;
+  const int32_t offset = index * pool->slotSize;
+
+  uint8_t* block = pool->memory + offset;
+  BlockHeader* header = (BlockHeader*)block;
+  header->index = index;
+
+  uint8_t* userMem = block + sizeof(BlockHeader);
+  return userMem;
+}
+
+void WuPoolRelease(WuPool* pool, void* ptr) {
+  uint8_t* mem = (uint8_t*)ptr - sizeof(BlockHeader);
+  BlockHeader* header = (BlockHeader*)mem;
+  pool->freeIndices[pool->freeIndicesCount++] = header->index;
+}