1 files changed, 96 insertions, 0 deletions

diff --git a/VNFs/DPPD-PROX/stats_mempool.c b/VNFs/DPPD-PROX/stats_mempool.c
new file mode 100644
index 00000000..c5861eb5
--- /dev/null
+++ b/VNFs/DPPD-PROX/stats_mempool.c
@@ -0,0 +1,96 @@
+/*
+// Copyright (c) 2010-2017 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_mempool.h>
+#include <rte_version.h>
+#include <inttypes.h>
+
+#include "prox_malloc.h"
+#include "prox_port_cfg.h"
+#include "stats_mempool.h"
+
+struct stats_mempool_manager {
+	uint32_t n_mempools;
+	struct mempool_stats mempool_stats[0];
+};
+
+static struct stats_mempool_manager *smm;
+
+struct mempool_stats *stats_get_mempool_stats(uint32_t i)
+{
+	return &smm->mempool_stats[i];
+}
+
+int stats_get_n_mempools(void)
+{
+	return smm->n_mempools;
+}
+
+static struct stats_mempool_manager *alloc_stats_mempool_manager(void)
+{
+	const uint32_t socket_id = rte_lcore_to_socket_id(rte_lcore_id());
+	uint32_t n_max_mempools = sizeof(prox_port_cfg[0].pool)/sizeof(prox_port_cfg[0].pool[0]);
+	uint32_t n_mempools = 0;
+	size_t mem_size = sizeof(struct stats_mempool_manager);
+
+	for (uint8_t i = 0; i < PROX_MAX_PORTS; ++i) {
+		if (!prox_port_cfg[i].active)
+			continue;
+
+		for (uint8_t j = 0; j < n_max_mempools; ++j) {
+			if (prox_port_cfg[i].pool[j] && prox_port_cfg[i].pool_size[j]) {
+				mem_size += sizeof(struct mempool_stats);
+			}
+		}
+	}
+
+	return prox_zmalloc(mem_size, socket_id);
+}
+
+void stats_mempool_init(void)
+{
+	uint32_t n_max_mempools = sizeof(prox_port_cfg[0].pool)/sizeof(prox_port_cfg[0].pool[0]);
+
+	smm = alloc_stats_mempool_manager();
+	for (uint8_t i = 0; i < PROX_MAX_PORTS; ++i) {
+		if (!prox_port_cfg[i].active)
+			continue;
+
+		for (uint8_t j = 0; j < n_max_mempools; ++j) {
+			if (prox_port_cfg[i].pool[j] && prox_port_cfg[i].pool_size[j]) {
+				struct mempool_stats *ms = &smm->mempool_stats[smm->n_mempools];
+
+				ms->pool = prox_port_cfg[i].pool[j];
+				ms->port = i;
+				ms->queue = j;
+				ms->size = prox_port_cfg[i].pool_size[j];
+				smm->n_mempools++;
+			}
+		}
+	}
+}
+
+void stats_mempool_update(void)
+{
+	for (uint8_t mp_id = 0; mp_id < smm->n_mempools; ++mp_id) {
+		/* Note: The function free_count returns the number of used entries. */
+#if RTE_VERSION >= RTE_VERSION_NUM(17,5,0,0)
+		smm->mempool_stats[mp_id].free = rte_mempool_avail_count(smm->mempool_stats[mp_id].pool);
+#else
+		smm->mempool_stats[mp_id].free = rte_mempool_count(smm->mempool_stats[mp_id].pool);
+#endif
+	}
+}