1 /* SPDX-License-Identifier: BSD-3-Clause
2 * Copyright(C) 2020 Marvell International Ltd.
3 */
4
5 #include <rte_debug.h>
6 #include <rte_ethdev.h>
7 #include <rte_graph.h>
8 #include <rte_graph_worker.h>
9 #include <rte_mbuf.h>
10
11 #include "ethdev_tx_priv.h"
12
13 static struct ethdev_tx_node_main ethdev_tx_main;
14
15 static uint16_t
ethdev_tx_node_process(struct rte_graph * graph,struct rte_node * node,void ** objs,uint16_t nb_objs)16 ethdev_tx_node_process(struct rte_graph *graph, struct rte_node *node,
17 void **objs, uint16_t nb_objs)
18 {
19 ethdev_tx_node_ctx_t *ctx = (ethdev_tx_node_ctx_t *)node->ctx;
20 uint16_t port, queue;
21 uint16_t count;
22
23 /* Get Tx port id */
24 port = ctx->port;
25 queue = ctx->queue;
26
27 count = rte_eth_tx_burst(port, queue, (struct rte_mbuf **)objs,
28 nb_objs);
29
30 /* Redirect unsent pkts to drop node */
31 if (count != nb_objs) {
32 rte_node_enqueue(graph, node, ETHDEV_TX_NEXT_PKT_DROP,
33 &objs[count], nb_objs - count);
34 }
35
36 return count;
37 }
38
39 static int
ethdev_tx_node_init(const struct rte_graph * graph,struct rte_node * node)40 ethdev_tx_node_init(const struct rte_graph *graph, struct rte_node *node)
41 {
42 ethdev_tx_node_ctx_t *ctx = (ethdev_tx_node_ctx_t *)node->ctx;
43 uint64_t port_id = RTE_MAX_ETHPORTS;
44 int i;
45
46 /* Find our port id */
47 for (i = 0; i < RTE_MAX_ETHPORTS; i++) {
48 if (ethdev_tx_main.nodes[i] == node->id) {
49 port_id = i;
50 break;
51 }
52 }
53 RTE_VERIFY(port_id < RTE_MAX_ETHPORTS);
54
55 /* Update port and queue */
56 ctx->port = port_id;
57 ctx->queue = graph->id;
58
59 return 0;
60 }
61
62 struct ethdev_tx_node_main *
ethdev_tx_node_data_get(void)63 ethdev_tx_node_data_get(void)
64 {
65 return ðdev_tx_main;
66 }
67
68 static struct rte_node_register ethdev_tx_node_base = {
69 .process = ethdev_tx_node_process,
70 .name = "ethdev_tx",
71
72 .init = ethdev_tx_node_init,
73
74 .nb_edges = ETHDEV_TX_NEXT_MAX,
75 .next_nodes = {
76 [ETHDEV_TX_NEXT_PKT_DROP] = "pkt_drop",
77 },
78 };
79
80 struct rte_node_register *
ethdev_tx_node_get(void)81 ethdev_tx_node_get(void)
82 {
83 return ðdev_tx_node_base;
84 }
85
86 RTE_NODE_REGISTER(ethdev_tx_node_base);
87