Initial commit of tldk code.

[tldk.git] / lib / libtle_udp / buf_cage.c

diff --git a/lib/libtle_udp/buf_cage.c b/lib/libtle_udp/buf_cage.c
new file mode 100644 (file)
index 0000000..0ae21b0
--- /dev/null
+++ b/lib/libtle_udp/buf_cage.c
@@ -0,0 +1,81 @@
+/*
+ * Copyright (c) 2016  Intel Corporation.
+ * 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.
+ */
+
+#include <rte_errno.h>
+#include <rte_malloc.h>
+#include <rte_log.h>
+
+#include "buf_cage.h"
+#include "osdep.h"
+
+struct bcg_store *
+bcg_create(const struct bcg_store_prm *prm)
+{
+       struct buf_cage *bc;
+       struct bcg_store *st;
+       uintptr_t end, p;
+       size_t sz, tsz;
+       uint32_t n;
+
+       if (prm == NULL || (prm->cage_align != 0 &&
+                       rte_is_power_of_2(prm->cage_align) == 0)) {
+               rte_errno = EINVAL;
+               return NULL;
+       }
+
+       /* number of cages required. */
+       n = (prm->max_bufs + prm->cage_bufs - 1) / prm->cage_bufs;
+       n = RTE_MAX(n, prm->min_cages);
+
+       /* size of each cage. */
+       sz = prm->cage_bufs * sizeof(bc->bufs[0]) + sizeof(*bc);
+       sz = RTE_ALIGN_CEIL(sz, prm->cage_align);
+
+       /* total number of bytes required. */
+       tsz = n * sz + RTE_ALIGN_CEIL(sizeof(*st), prm->cage_align);
+
+       st = rte_zmalloc_socket(NULL, tsz, RTE_CACHE_LINE_SIZE, prm->socket_id);
+       if (st == NULL) {
+               UDP_LOG(ERR, "%s: allocation of %zu bytes on "
+                       "socket %d failed\n",
+                       __func__, tsz, prm->socket_id);
+               return NULL;
+       }
+
+       st->prm = prm[0];
+       bcg_queue_reset(&st->free);
+
+       p = (uintptr_t)RTE_PTR_ALIGN_CEIL((st + 1), prm->cage_align);
+       end = p + n * sz;
+
+       for (; p != end; p += sz) {
+               bc = (struct buf_cage *)p;
+               bc->st = st;
+               bc->num = prm->cage_bufs;
+               STAILQ_INSERT_TAIL(&st->free.queue, bc, ql);
+       }
+
+       st->free.num = n;
+       st->nb_cages = n;
+       st->cage_sz = sz;
+       st->total_sz = tsz;
+       return st;
+}
+
+void
+bcg_destroy(struct bcg_store *st)
+{
+       rte_free(st);
+}