This commit shows the steps necessary to enable traffic generation
using the gen library

Signed-off-by: Ronan Randles <ronan.rand...@intel.com>
---
 examples/generator/main.c      | 175 ++++++++++++++++++++++-----------
 examples/generator/meson.build |   1 +
 examples/meson.build           |   1 +
 3 files changed, 120 insertions(+), 57 deletions(-)

diff --git a/examples/generator/main.c b/examples/generator/main.c
index 0082f588b4..1ddf4c1603 100644
--- a/examples/generator/main.c
+++ b/examples/generator/main.c
@@ -1,14 +1,18 @@
 /* SPDX-License-Identifier: BSD-3-Clause
- * Copyright(c) 2010-2015 Intel Corporation
+ * Copyright(c) 2021 Intel Corporation
  */
 
 #include <stdint.h>
 #include <inttypes.h>
+#include <unistd.h>
+#include <signal.h>
+
 #include <rte_eal.h>
 #include <rte_ethdev.h>
 #include <rte_cycles.h>
 #include <rte_lcore.h>
 #include <rte_mbuf.h>
+#include <rte_gen.h>
 
 #define RX_RING_SIZE 1024
 #define TX_RING_SIZE 1024
@@ -16,23 +20,23 @@
 #define NUM_MBUFS 8191
 #define MBUF_CACHE_SIZE 256
 #define BURST_SIZE 32
+#define MIN_THREADS 3
 
-/* Configuration of ethernet ports. 8<  */
 static const struct rte_eth_conf port_conf_default = {
        .rxmode = {
                .max_lro_pkt_size = RTE_ETHER_MAX_LEN,
        },
 };
-/* >8 End of configuration of ethernet ports. */
 
-/* basicfwd.c: Basic DPDK skeleton forwarding example. */
+static volatile int done;
+static struct rte_mempool *mbuf_pool;
+struct rte_gen *gen;
+
+static void handle_sigint(int sig);
 
-/*
- * Initializes a given port using global settings and with the RX buffers
+/* Initializes a given port using global settings and with the RX buffers
  * coming from the mbuf_pool passed as a parameter.
  */
-
-/* Main functional part of port initialization. 8< */
 static inline int
 port_init(uint16_t port, struct rte_mempool *mbuf_pool)
 {
@@ -68,6 +72,12 @@ port_init(uint16_t port, struct rte_mempool *mbuf_pool)
        if (retval != 0)
                return retval;
 
+       int lcore_available_count = rte_lcore_count();
+       if (lcore_available_count < MIN_THREADS) {
+               printf("Not enough threads available\n");
+               return -1;
+       }
+
        /* Allocate and set up 1 RX queue per Ethernet port. */
        for (q = 0; q < rx_rings; q++) {
                retval = rte_eth_rx_queue_setup(port, q, nb_rxd,
@@ -86,9 +96,8 @@ port_init(uint16_t port, struct rte_mempool *mbuf_pool)
                        return retval;
        }
 
-       /* Starting Ethernet port. 8< */
+       /* Start the Ethernet port. */
        retval = rte_eth_dev_start(port);
-       /* >8 End of starting of ethernet port. */
        if (retval < 0)
                return retval;
 
@@ -104,27 +113,67 @@ port_init(uint16_t port, struct rte_mempool *mbuf_pool)
 
        /* Enable RX in promiscuous mode for the Ethernet device. */
        retval = rte_eth_promiscuous_enable(port);
-       /* End of setting RX port in promiscuous mode. */
        if (retval != 0)
                return retval;
 
        return 0;
 }
-/* >8 End of main functional part of port initialization. */
 
-/*
- * The lcore main. This is the main thread that does the work, reading from
+/* The lcore main. This is the main thread that does the work, reading from
  * an input port and writing to an output port.
  */
+static int
+lcore_producer(__rte_unused void *arg)
+{
+       uint16_t port;
+
+       /* Check that the port is on the same NUMA node as the polling thread
+        * for best performance.
+        */
+       RTE_ETH_FOREACH_DEV(port)
+               if (rte_eth_dev_socket_id(port) >= 0 &&
+                               rte_eth_dev_socket_id(port) !=
+                                               (int)rte_socket_id())
+                       printf("WARNING, port %u is on remote NUMA node to "
+                                       "polling thread.\n\tPerformance will "
+                                       "not be optimal.\n", port);
 
- /* Basic forwarding application lcore. 8< */
-static __rte_noreturn void
-lcore_main(void)
+       /* Run until the application is quit or killed. */
+       while (!done) {
+               struct rte_mbuf *bufs[BURST_SIZE];
+               int i;
+               /* Receive packets from gen and then tx them over port */
+               RTE_ETH_FOREACH_DEV(port) {
+                       int nb_recieved = rte_gen_rx_burst(gen, bufs,
+                                                       BURST_SIZE);
+                       for (i = 0; i < nb_recieved; i++) {
+                               bufs[i]->pkt_len = 64;
+                               bufs[i]->data_len = 64;
+                       }
+
+                       uint16_t nb_tx = rte_eth_tx_burst(port, 0, bufs,
+                                                       nb_recieved);
+                       if (nb_tx != nb_recieved)
+                               rte_pktmbuf_free_bulk(&bufs[nb_tx],
+                                                       (nb_recieved - nb_tx));
+
+                       if (unlikely(nb_tx == 0))
+                               continue;
+
+               }
+       }
+       return 0;
+}
+
+/* The lcore main. This is the main thread that does the work, reading from
+ * an input port and writing to an output port.
+ */
+static int
+lcore_consumer(__rte_unused void *arg)
 {
        uint16_t port;
 
-       /*
-        * Check that the port is on the same NUMA node as the polling thread
+       /* Check that the port is on the same NUMA node as the polling thread
         * for best performance.
         */
        RTE_ETH_FOREACH_DEV(port)
@@ -135,57 +184,53 @@ lcore_main(void)
                                        "polling thread.\n\tPerformance will "
                                        "not be optimal.\n", port);
 
-       printf("\nCore %u forwarding packets. [Ctrl+C to quit]\n",
-                       rte_lcore_id());
+       /* Run until the application is quit or killed. */
+       while (!done) {
+               struct rte_mbuf *bufs[BURST_SIZE];
 
-       /* Main work of application loop. 8< */
-       for (;;) {
-               /*
-                * Receive packets on a port and forward them on the paired
-                * port. The mapping is 0 -> 1, 1 -> 0, 2 -> 3, 3 -> 2, etc.
+               /* Receive packets over port and then tx them to gen library
+                * for stats
                 */
                RTE_ETH_FOREACH_DEV(port) {
+                       uint64_t latency[BURST_SIZE];
+                       uint16_t nb_rx = rte_eth_rx_burst(port, 0, bufs,
+                                                       BURST_SIZE);
+                       rte_gen_tx_burst(gen, bufs, latency, nb_rx);
 
-                       /* Get burst of RX packets, from first port of pair. */
-                       struct rte_mbuf *bufs[BURST_SIZE];
-                       const uint16_t nb_rx = rte_eth_rx_burst(port, 0,
-                                       bufs, BURST_SIZE);
+                       int nb_sent = rte_gen_tx_burst(gen, bufs,
+                                                       latency, nb_rx);
+                       if (nb_sent != nb_rx)
+                               rte_panic("invalid tx quantity\n");
 
                        if (unlikely(nb_rx == 0))
                                continue;
 
-                       /* Send burst of TX packets, to second port of pair. */
-                       const uint16_t nb_tx = rte_eth_tx_burst(port ^ 1, 0,
-                                       bufs, nb_rx);
-
-                       /* Free any unsent packets. */
-                       if (unlikely(nb_tx < nb_rx)) {
-                               uint16_t buf;
-                               for (buf = nb_tx; buf < nb_rx; buf++)
-                                       rte_pktmbuf_free(bufs[buf]);
-                       }
                }
        }
-       /* >8 End of loop. */
+       return 0;
 }
-/* >8 End Basic forwarding application lcore. */
 
-/*
- * The main function, which does initialization and calls the per-lcore
+void handle_sigint(int sig)
+{
+       RTE_SET_USED(sig);
+       printf("\nExiting...\n");
+       done = 1;
+}
+
+/* The main function, which does initialization and calls the per-lcore
  * functions.
  */
 int
 main(int argc, char *argv[])
 {
-       struct rte_mempool *mbuf_pool;
+       signal(SIGINT, handle_sigint);
        unsigned int nb_ports;
        uint16_t portid;
 
-       /* Initializion the Environment Abstraction Layer (EAL). 8< */
+       /* Initialize the Environment Abstraction Layer (EAL). */
        int ret = rte_eal_init(argc, argv);
        if (ret < 0)
                rte_exit(EXIT_FAILURE, "Error with EAL initialization\n");
-       /* >8 End of initializion the Environment Abstraction Layer (EAL). */
 
        argc -= ret;
        argv += ret;
@@ -196,28 +241,44 @@ main(int argc, char *argv[])
                rte_exit(EXIT_FAILURE, "Error: number of ports must be even\n");
 
        /* Creates a new mempool in memory to hold the mbufs. */
-
-       /* Allocates mempool to hold the mbufs. 8< */
        mbuf_pool = rte_pktmbuf_pool_create("MBUF_POOL", NUM_MBUFS * nb_ports,
                MBUF_CACHE_SIZE, 0, RTE_MBUF_DEFAULT_BUF_SIZE, rte_socket_id());
-       /* >8 End of allocating mempool to hold mbuf. */
 
        if (mbuf_pool == NULL)
                rte_exit(EXIT_FAILURE, "Cannot create mbuf pool\n");
 
-       /* Initializing all ports. 8< */
+       /* Initialize all ports. */
        RTE_ETH_FOREACH_DEV(portid)
                if (port_init(portid, mbuf_pool) != 0)
                        rte_exit(EXIT_FAILURE, "Cannot init port %"PRIu16 "\n",
                                        portid);
-       /* >8 End of initializing all ports. */
 
-       if (rte_lcore_count() > 1)
-               printf("\nWARNING: Too many lcores enabled. Only 1 used.\n");
+       gen = rte_gen_create(mbuf_pool);
+       if (!gen)
+               rte_panic("Gen failed to initialize\n");
+
+       int err = rte_gen_packet_parse_string(gen, "Ether()/IP()", NULL);
+       if (err)
+               rte_panic("Failed to parse input args");
+
+       /* launch lcore functions */
+       uint32_t lcore_count = 0;
+       uint32_t lcore_id = 0;
+       RTE_LCORE_FOREACH_WORKER(lcore_id) {
+               if (lcore_count == 0)
+                       rte_eal_remote_launch(lcore_producer, NULL, lcore_id);
+               else if (lcore_count == 1)
+                       rte_eal_remote_launch(lcore_consumer, NULL, lcore_id);
+               else
+                       break;
+
+               lcore_count++;
+       }
+       /* Stall the main thread until all other threads have returned. */
+       rte_eal_mp_wait_lcore();
 
-       /* Call lcore_main on the main core only. Called on single lcore. 8< */
-       lcore_main();
-       /* >8 End of called on single lcore. */
+       /* All threads returned, safe to destroy gen instance */
+       rte_gen_destroy(gen);
 
        /* clean up the EAL */
        rte_eal_cleanup();
diff --git a/examples/generator/meson.build b/examples/generator/meson.build
index 441678bbe5..15d84674a5 100644
--- a/examples/generator/meson.build
+++ b/examples/generator/meson.build
@@ -10,3 +10,4 @@ allow_experimental_apis = true
 sources = files(
         'main.c',
 )
+deps += 'gen'
\ No newline at end of file
diff --git a/examples/meson.build b/examples/meson.build
index bac9b76007..79ce36d02b 100644
--- a/examples/meson.build
+++ b/examples/meson.build
@@ -18,6 +18,7 @@ all_examples = [
         'fips_validation',
         'flow_classify',
         'flow_filtering',
+        'generator',
         'helloworld',
         'ip_fragmentation',
         'ip_pipeline',
-- 
2.25.1

Reply via email to