1 /* SPDX-License-Identifier: BSD-3-Clause
2 * Copyright(C) 2020 Marvell International Ltd.
9 #include <rte_common.h>
10 #include <rte_debug.h>
11 #include <rte_errno.h>
12 #include <rte_string_fns.h>
14 #include "graph_private.h"
16 static struct node_head node_list = STAILQ_HEAD_INITIALIZER(node_list);
17 static rte_node_t node_id;
19 #define NODE_ID_CHECK(id) ID_CHECK(id, node_id)
21 /* Private functions */
23 node_list_head_get(void)
29 node_from_name(const char *name)
33 STAILQ_FOREACH(node, &node_list, next)
34 if (strncmp(node->name, name, RTE_NODE_NAMESIZE) == 0)
41 node_has_duplicate_entry(const char *name)
45 /* Is duplicate name registered */
46 STAILQ_FOREACH(node, &node_list, next) {
47 if (strncmp(node->name, name, RTE_NODE_NAMESIZE) == 0) {
55 /* Public functions */
57 __rte_node_register(const struct rte_node_register *reg)
63 /* Limit Node specific metadata to one cacheline on 64B CL machine */
64 RTE_BUILD_BUG_ON((offsetof(struct rte_node, nodes) -
65 offsetof(struct rte_node, ctx)) !=
66 RTE_CACHE_LINE_MIN_SIZE);
68 graph_spinlock_lock();
71 if (reg == NULL || reg->process == NULL) {
76 /* Check for duplicate name */
77 if (node_has_duplicate_entry(reg->name))
80 sz = sizeof(struct node) + (reg->nb_edges * RTE_NODE_NAMESIZE);
87 /* Initialize the node */
88 if (rte_strscpy(node->name, reg->name, RTE_NODE_NAMESIZE) < 0)
90 node->flags = reg->flags;
91 node->process = reg->process;
92 node->init = reg->init;
93 node->fini = reg->fini;
94 node->nb_edges = reg->nb_edges;
95 node->parent_id = reg->parent_id;
96 for (i = 0; i < reg->nb_edges; i++) {
97 if (rte_strscpy(node->next_nodes[i], reg->next_nodes[i],
98 RTE_NODE_NAMESIZE) < 0)
102 node->id = node_id++;
104 /* Add the node at tail */
105 STAILQ_INSERT_TAIL(&node_list, node, next);
106 graph_spinlock_unlock();
112 graph_spinlock_unlock();
113 return RTE_NODE_ID_INVALID;
117 clone_name(struct rte_node_register *reg, struct node *node, const char *name)
121 #define SZ RTE_NODE_NAMESIZE
122 rc = rte_strscpy(reg->name, node->name, SZ);
126 rc = rte_strscpy(reg->name + sz, "-", RTE_MAX((int16_t)(SZ - sz), 0));
130 sz = rte_strscpy(reg->name + sz, name, RTE_MAX((int16_t)(SZ - sz), 0));
141 node_clone(struct node *node, const char *name)
143 rte_node_t rc = RTE_NODE_ID_INVALID;
144 struct rte_node_register *reg;
147 /* Don't allow to clone a node from a cloned node */
148 if (node->parent_id != RTE_NODE_ID_INVALID) {
153 reg = calloc(1, sizeof(*reg) + (sizeof(char *) * node->nb_edges));
159 /* Clone the source node */
160 reg->flags = node->flags;
161 reg->process = node->process;
162 reg->init = node->init;
163 reg->fini = node->fini;
164 reg->nb_edges = node->nb_edges;
165 reg->parent_id = node->id;
167 for (i = 0; i < node->nb_edges; i++)
168 reg->next_nodes[i] = node->next_nodes[i];
170 /* Naming ceremony of the new node. name is node->name + "-" + name */
171 if (clone_name(reg, node, name))
174 rc = __rte_node_register(reg);
182 rte_node_clone(rte_node_t id, const char *name)
187 STAILQ_FOREACH(node, &node_list, next)
189 return node_clone(node, name);
192 return RTE_NODE_ID_INVALID;
196 rte_node_from_name(const char *name)
200 STAILQ_FOREACH(node, &node_list, next)
201 if (strncmp(node->name, name, RTE_NODE_NAMESIZE) == 0)
204 return RTE_NODE_ID_INVALID;
208 rte_node_id_to_name(rte_node_t id)
213 STAILQ_FOREACH(node, &node_list, next)
222 rte_node_edge_count(rte_node_t id)
227 STAILQ_FOREACH(node, &node_list, next)
229 return node->nb_edges;
231 return RTE_EDGE_ID_INVALID;
235 edge_update(struct node *node, struct node *prev, rte_edge_t from,
236 const char **next_nodes, rte_edge_t nb_edges)
238 rte_edge_t i, max_edges, count = 0;
239 struct node *new_node;
243 if (from == RTE_EDGE_ID_INVALID)
244 from = node->nb_edges;
246 /* Don't create hole in next_nodes[] list */
247 if (from > node->nb_edges) {
252 /* Remove me from list */
253 STAILQ_REMOVE(&node_list, node, node, next);
255 /* Allocate the storage space for new node if required */
256 max_edges = from + nb_edges;
257 need_realloc = max_edges > node->nb_edges;
259 sz = sizeof(struct node) + (max_edges * RTE_NODE_NAMESIZE);
260 new_node = realloc(node, sz);
261 if (new_node == NULL) {
269 /* Update the new nodes name */
270 for (i = from; i < max_edges; i++, count++) {
271 if (rte_strscpy(node->next_nodes[i], next_nodes[count],
272 RTE_NODE_NAMESIZE) < 0)
276 /* Update the linked list to point new node address in prev node */
278 STAILQ_INSERT_AFTER(&node_list, prev, node, next);
280 STAILQ_INSERT_HEAD(&node_list, node, next);
283 node->nb_edges = max_edges;
290 rte_node_edge_shrink(rte_node_t id, rte_edge_t size)
292 rte_edge_t rc = RTE_EDGE_ID_INVALID;
296 graph_spinlock_lock();
298 STAILQ_FOREACH(node, &node_list, next) {
299 if (node->id == id) {
300 if (node->nb_edges < size) {
304 node->nb_edges = size;
311 graph_spinlock_unlock();
316 rte_node_edge_update(rte_node_t id, rte_edge_t from, const char **next_nodes,
319 rte_edge_t rc = RTE_EDGE_ID_INVALID;
320 struct node *n, *prev;
323 graph_spinlock_lock();
326 STAILQ_FOREACH(n, &node_list, next) {
328 rc = edge_update(n, prev, from, next_nodes, nb_edges);
334 graph_spinlock_unlock();
340 node_copy_edges(struct node *node, char *next_nodes[])
344 for (i = 0; i < node->nb_edges; i++)
345 next_nodes[i] = node->next_nodes[i];
351 rte_node_edge_get(rte_node_t id, char *next_nodes[])
353 rte_node_t rc = RTE_NODE_ID_INVALID;
357 graph_spinlock_lock();
359 STAILQ_FOREACH(node, &node_list, next) {
360 if (node->id == id) {
361 if (next_nodes == NULL)
362 rc = sizeof(char *) * node->nb_edges;
364 rc = node_copy_edges(node, next_nodes);
369 graph_spinlock_unlock();
375 node_scan_dump(FILE *f, rte_node_t id, bool all)
379 RTE_ASSERT(f != NULL);
382 STAILQ_FOREACH(node, &node_list, next) {
385 } else if (node->id == id) {
395 rte_node_dump(FILE *f, rte_node_t id)
397 node_scan_dump(f, id, false);
401 rte_node_list_dump(FILE *f)
403 node_scan_dump(f, 0, true);
407 rte_node_max_count(void)