From 15ad172a847fa667c57a4594ef4158405db9a984 Mon Sep 17 00:00:00 2001
From: Angelo Mantellini <angelo.mantellini@cisco.com>
Date: Tue, 31 Mar 2020 17:50:43 +0200
Subject: [HICN-554] hicn-light refactoring
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Change-Id: I36f2d393741d4502ce14d3791158e43e3e9cd4cf
Signed-off-by: Jordan Augé <jordan.auge+fdio@cisco.com>
---
 hicn-light/src/hicn/base/pool.c | 88 +++++++++++++++++++++++++++++++++++++++++
 1 file changed, 88 insertions(+)
 create mode 100644 hicn-light/src/hicn/base/pool.c

(limited to 'hicn-light/src/hicn/base/pool.c')

diff --git a/hicn-light/src/hicn/base/pool.c b/hicn-light/src/hicn/base/pool.c
new file mode 100644
index 000000000..abca86422
--- /dev/null
+++ b/hicn-light/src/hicn/base/pool.c
@@ -0,0 +1,88 @@
+/*
+ * Copyright (c) 2020 Cisco and/or its affiliates.
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at:
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+/**
+ * \file pool.c
+ * \brief Implementation of fixed-size pool allocator
+ */
+
+#include <stdlib.h> // calloc
+
+#include "common.h"
+#include "pool.h"
+
+
+/**
+ * \brief Initialize the pool data structure
+ * \param [in,out] pool - Pointer to the pool structure storage
+ * \param [in] elt_size - Size of elements in vector
+ * \param [in] max_elts - Maximum size
+ *
+ * Note that an empty pool might be equal to NULL
+ */
+void
+_pool_init(void ** pool_ptr, size_t elt_size, size_t max_elts)
+{
+    pool_hdr_t * ph = calloc(POOL_HDRLEN + elt_size * max_elts, 1);
+    if (!ph)
+        abort();
+
+    /* Free indices */
+    off_t * free_indices;
+    vector_init(free_indices, max_elts);
+
+    uint_fast32_t * fb = ph->free_bitmap;
+    bitmap_init(fb, max_elts);
+    bitmap_set_to(fb, max_elts);
+
+    for(unsigned i = 0; i < max_elts; i++)
+        free_indices[i] = (max_elts - 1) - i;
+    ph->free_indices = free_indices;
+
+    *pool_ptr = (uint8_t*)ph + POOL_HDRLEN;
+}
+
+void
+_pool_free(void ** pool_ptr)
+{
+    free(pool_hdr(*pool_ptr));
+    *pool_ptr = NULL;
+}
+
+void
+_pool_resize(void ** pool_ptr, size_t elt_size)
+{
+    pool_hdr_t * ph = pool_hdr(*pool_ptr);
+    size_t old_elts = ph->max_elts;
+    size_t new_elts = old_elts * 2;
+
+    /* Double pool storage */
+    ph = realloc(ph, POOL_HDRLEN + new_elts * elt_size);
+    if (!ph)
+        abort();
+    ph->max_elts = new_elts;
+
+    /*
+     * After resize, the pool will have old_elts free indices, ranging from
+     * old_elts to (new_elts - 1)
+     */
+    off_t * free_indices = ph->free_indices;
+    vector_ensure_pos(free_indices, old_elts);
+    for (unsigned i = 0; i < old_elts; i++)
+        free_indices[i] = new_elts - 1 - i;
+
+    /* Reassign pool pointer */
+    *pool_ptr = (uint8_t*)ph + POOL_HDRLEN;
+}
-- 
cgit 1.2.3-korg