820448bc2503ba871e0c67579c6496256345c122
[deb_dpdk.git] / examples / multi_process / l2fwd_fork / main.c
1 /*-
2  *   BSD LICENSE
3  *
4  *   Copyright(c) 2010-2016 Intel Corporation. All rights reserved.
5  *   All rights reserved.
6  *
7  *   Redistribution and use in source and binary forms, with or without
8  *   modification, are permitted provided that the following conditions
9  *   are met:
10  *
11  *     * Redistributions of source code must retain the above copyright
12  *       notice, this list of conditions and the following disclaimer.
13  *     * Redistributions in binary form must reproduce the above copyright
14  *       notice, this list of conditions and the following disclaimer in
15  *       the documentation and/or other materials provided with the
16  *       distribution.
17  *     * Neither the name of Intel Corporation nor the names of its
18  *       contributors may be used to endorse or promote products derived
19  *       from this software without specific prior written permission.
20  *
21  *   THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
22  *   "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
23  *   LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
24  *   A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
25  *   OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
26  *   SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
27  *   LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
28  *   DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
29  *   THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
30  *   (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
31  *   OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
32  */
33 #define _GNU_SOURCE
34 #include <stdio.h>
35 #include <stdlib.h>
36 #include <unistd.h>
37 #include <string.h>
38 #include <stdint.h>
39 #include <sched.h>
40 #include <inttypes.h>
41 #include <sys/types.h>
42 #include <sys/queue.h>
43 #include <netinet/in.h>
44 #include <setjmp.h>
45 #include <stdarg.h>
46 #include <ctype.h>
47 #include <errno.h>
48 #include <getopt.h>
49
50 #include <rte_common.h>
51 #include <rte_log.h>
52 #include <rte_memory.h>
53 #include <rte_memcpy.h>
54 #include <rte_memzone.h>
55 #include <rte_eal.h>
56 #include <rte_per_lcore.h>
57 #include <rte_launch.h>
58 #include <rte_atomic.h>
59 #include <rte_spinlock.h>
60 #include <rte_cycles.h>
61 #include <rte_prefetch.h>
62 #include <rte_lcore.h>
63 #include <rte_per_lcore.h>
64 #include <rte_branch_prediction.h>
65 #include <rte_interrupts.h>
66 #include <rte_pci.h>
67 #include <rte_random.h>
68 #include <rte_debug.h>
69 #include <rte_ether.h>
70 #include <rte_ethdev.h>
71 #include <rte_ring.h>
72 #include <rte_mempool.h>
73 #include <rte_mbuf.h>
74 #include <rte_malloc.h>
75
76 #include "flib.h"
77
78 #define RTE_LOGTYPE_L2FWD RTE_LOGTYPE_USER1
79 #define MBUF_NAME       "mbuf_pool_%d"
80 #define MBUF_SIZE       \
81 (RTE_MBUF_DEFAULT_DATAROOM + sizeof(struct rte_mbuf) + RTE_PKTMBUF_HEADROOM)
82 #define NB_MBUF   8192
83 #define RING_MASTER_NAME        "l2fwd_ring_m2s_"
84 #define RING_SLAVE_NAME         "l2fwd_ring_s2m_"
85 #define MAX_NAME_LEN    32
86 /* RECREATE flag indicate needs initialize resource and launch slave_core again */
87 #define SLAVE_RECREATE_FLAG 0x1
88 /* RESTART flag indicate needs restart port and send START command again */
89 #define SLAVE_RESTART_FLAG 0x2
90 #define INVALID_MAPPING_ID      ((unsigned)LCORE_ID_ANY)
91 /* Maximum message buffer per slave */
92 #define NB_CORE_MSGBUF  32
93 enum l2fwd_cmd{
94         CMD_START,
95         CMD_STOP,
96 };
97
98 #define MAX_PKT_BURST 32
99 #define BURST_TX_DRAIN_US 100 /* TX drain every ~100us */
100
101 /*
102  * Configurable number of RX/TX ring descriptors
103  */
104 #define RTE_TEST_RX_DESC_DEFAULT 128
105 #define RTE_TEST_TX_DESC_DEFAULT 512
106 static uint16_t nb_rxd = RTE_TEST_RX_DESC_DEFAULT;
107 static uint16_t nb_txd = RTE_TEST_TX_DESC_DEFAULT;
108
109 /* ethernet addresses of ports */
110 static struct ether_addr l2fwd_ports_eth_addr[RTE_MAX_ETHPORTS];
111
112 /* mask of enabled ports */
113 static uint32_t l2fwd_enabled_port_mask = 0;
114
115 /* list of enabled ports */
116 static uint32_t l2fwd_dst_ports[RTE_MAX_ETHPORTS];
117
118 static unsigned int l2fwd_rx_queue_per_lcore = 1;
119
120 struct mbuf_table {
121         unsigned len;
122         struct rte_mbuf *m_table[MAX_PKT_BURST];
123 };
124
125 #define MAX_RX_QUEUE_PER_LCORE 16
126 #define MAX_TX_QUEUE_PER_PORT 16
127 struct lcore_queue_conf {
128         unsigned n_rx_port;
129         unsigned rx_port_list[MAX_RX_QUEUE_PER_LCORE];
130 } __rte_cache_aligned;
131 struct lcore_queue_conf lcore_queue_conf[RTE_MAX_LCORE];
132
133 struct rte_eth_dev_tx_buffer *tx_buffer[RTE_MAX_ETHPORTS];
134
135 struct lcore_resource_struct {
136         int enabled;    /* Only set in case this lcore involved into packet forwarding */
137         int flags;          /* Set only slave need to restart or recreate */
138         unsigned lcore_id;  /*  lcore ID */
139         unsigned pair_id;       /* dependency lcore ID on port */
140         char ring_name[2][MAX_NAME_LEN];
141         /* ring[0] for master send cmd, slave read */
142         /* ring[1] for slave send ack, master read */
143         struct rte_ring *ring[2];
144         int port_num;                                   /* Total port numbers */
145         uint8_t port[RTE_MAX_ETHPORTS]; /* Port id for that lcore to receive packets */
146 }__attribute__((packed)) __rte_cache_aligned;
147
148 static struct lcore_resource_struct lcore_resource[RTE_MAX_LCORE];
149 static struct rte_mempool *message_pool;
150 static rte_spinlock_t res_lock = RTE_SPINLOCK_INITIALIZER;
151 /* use floating processes */
152 static int float_proc = 0;
153 /* Save original cpu affinity */
154 struct cpu_aff_arg{
155         cpu_set_t set;
156         size_t size;
157 }cpu_aff;
158
159 static const struct rte_eth_conf port_conf = {
160         .rxmode = {
161                 .split_hdr_size = 0,
162                 .header_split   = 0, /**< Header Split disabled */
163                 .hw_ip_checksum = 0, /**< IP checksum offload disabled */
164                 .hw_vlan_filter = 0, /**< VLAN filtering disabled */
165                 .jumbo_frame    = 0, /**< Jumbo Frame Support disabled */
166                 .hw_strip_crc   = 1, /**< CRC stripped by hardware */
167         },
168         .txmode = {
169                 .mq_mode = ETH_MQ_TX_NONE,
170         },
171 };
172
173 static struct rte_mempool * l2fwd_pktmbuf_pool[RTE_MAX_ETHPORTS];
174
175 /* Per-port statistics struct */
176 struct l2fwd_port_statistics {
177         uint64_t tx;
178         uint64_t rx;
179         uint64_t dropped;
180 } __rte_cache_aligned;
181 struct l2fwd_port_statistics *port_statistics;
182 /**
183  * pointer to lcore ID mapping array, used to return lcore id in case slave
184  * process exited unexpectedly, use only floating process option applied
185  **/
186 unsigned *mapping_id;
187
188 /* A tsc-based timer responsible for triggering statistics printout */
189 #define TIMER_MILLISECOND 2000000ULL /* around 1ms at 2 Ghz */
190 #define MAX_TIMER_PERIOD 86400 /* 1 day max */
191 static int64_t timer_period = 10 * TIMER_MILLISECOND * 1000; /* default period is 10 seconds */
192
193 static int l2fwd_launch_one_lcore(void *dummy);
194
195 /* Print out statistics on packets dropped */
196 static void
197 print_stats(void)
198 {
199         uint64_t total_packets_dropped, total_packets_tx, total_packets_rx;
200         unsigned portid;
201
202         total_packets_dropped = 0;
203         total_packets_tx = 0;
204         total_packets_rx = 0;
205
206         const char clr[] = { 27, '[', '2', 'J', '\0' };
207         const char topLeft[] = { 27, '[', '1', ';', '1', 'H','\0' };
208
209                 /* Clear screen and move to top left */
210         printf("%s%s", clr, topLeft);
211
212         printf("\nPort statistics ====================================");
213
214         for (portid = 0; portid < RTE_MAX_ETHPORTS; portid++) {
215                 /* skip disabled ports */
216                 if ((l2fwd_enabled_port_mask & (1 << portid)) == 0)
217                         continue;
218                 printf("\nStatistics for port %u ------------------------------"
219                            "\nPackets sent: %24"PRIu64
220                            "\nPackets received: %20"PRIu64
221                            "\nPackets dropped: %21"PRIu64,
222                            portid,
223                            port_statistics[portid].tx,
224                            port_statistics[portid].rx,
225                            port_statistics[portid].dropped);
226
227                 total_packets_dropped += port_statistics[portid].dropped;
228                 total_packets_tx += port_statistics[portid].tx;
229                 total_packets_rx += port_statistics[portid].rx;
230         }
231         printf("\nAggregate statistics ==============================="
232                    "\nTotal packets sent: %18"PRIu64
233                    "\nTotal packets received: %14"PRIu64
234                    "\nTotal packets dropped: %15"PRIu64,
235                    total_packets_tx,
236                    total_packets_rx,
237                    total_packets_dropped);
238         printf("\n====================================================\n");
239 }
240
241 static int
242 clear_cpu_affinity(void)
243 {
244         int s;
245
246         s = sched_setaffinity(0, cpu_aff.size, &cpu_aff.set);
247         if (s != 0) {
248                 printf("sched_setaffinity failed:%s\n", strerror(errno));
249                 return -1;
250         }
251
252         return 0;
253 }
254
255 static int
256 get_cpu_affinity(void)
257 {
258         int s;
259
260         cpu_aff.size = sizeof(cpu_set_t);
261         CPU_ZERO(&cpu_aff.set);
262
263         s = sched_getaffinity(0, cpu_aff.size, &cpu_aff.set);
264         if (s != 0) {
265                 printf("sched_getaffinity failed:%s\n", strerror(errno));
266                 return -1;
267         }
268
269         return 0;
270 }
271
272 /**
273  * This fnciton demonstrates the approach to create ring in first instance
274  * or re-attach an existed ring in later instance.
275  **/
276 static struct rte_ring *
277 create_ring(const char *name, unsigned count,
278                                         int socket_id,unsigned flags)
279 {
280         struct rte_ring *ring;
281
282         if (name == NULL)
283                 return NULL;
284
285         /* If already create, just attached it */
286         if (likely((ring = rte_ring_lookup(name)) != NULL))
287                 return ring;
288
289         /* First call it, create one */
290         return rte_ring_create(name, count, socket_id, flags);
291 }
292
293 /* Malloc with rte_malloc on structures that shared by master and slave */
294 static int
295 l2fwd_malloc_shared_struct(void)
296 {
297         port_statistics = rte_zmalloc("port_stat",
298                                                 sizeof(struct l2fwd_port_statistics) * RTE_MAX_ETHPORTS,
299                                                 0);
300         if (port_statistics == NULL)
301                 return -1;
302
303         /* allocate  mapping_id array */
304         if (float_proc) {
305                 int i;
306                 mapping_id = rte_malloc("mapping_id", sizeof(unsigned) * RTE_MAX_LCORE,
307                                                                 0);
308
309                 if (mapping_id == NULL)
310                         return -1;
311
312                 for (i = 0 ;i < RTE_MAX_LCORE; i++)
313                         mapping_id[i] = INVALID_MAPPING_ID;
314         }
315         return 0;
316 }
317
318 /* Create ring which used for communicate among master and slave */
319 static int
320 create_ms_ring(unsigned slaveid)
321 {
322         unsigned flag = RING_F_SP_ENQ | RING_F_SC_DEQ;
323         struct lcore_resource_struct *res = &lcore_resource[slaveid];
324         unsigned socketid = rte_socket_id();
325
326         /* Always assume create ring on master socket_id */
327         /* Default only create a ring size 32 */
328         snprintf(res->ring_name[0], MAX_NAME_LEN, "%s%u",
329                         RING_MASTER_NAME, slaveid);
330         if ((res->ring[0] = create_ring(res->ring_name[0], NB_CORE_MSGBUF,
331                                 socketid, flag)) == NULL) {
332                 printf("Create m2s ring %s failed\n", res->ring_name[0]);
333                 return -1;
334         }
335
336         snprintf(res->ring_name[1], MAX_NAME_LEN, "%s%u",
337                         RING_SLAVE_NAME, slaveid);
338         if ((res->ring[1] = create_ring(res->ring_name[1], NB_CORE_MSGBUF,
339                 socketid, flag)) == NULL) {
340                 printf("Create s2m ring %s failed\n", res->ring_name[1]);
341                 return -1;
342         }
343
344         return 0;
345 }
346
347 /* send command to pair in paired master and slave ring */
348 static inline int
349 sendcmd(unsigned slaveid, enum l2fwd_cmd cmd, int is_master)
350 {
351         struct lcore_resource_struct *res = &lcore_resource[slaveid];
352         void *msg;
353         int fd = !is_master;
354
355         /* Only check master, it must be enabled and running if it is slave */
356         if (is_master && !res->enabled)
357                 return -1;
358
359         if (res->ring[fd] == NULL)
360                 return -1;
361
362         if (rte_mempool_get(message_pool, &msg) < 0) {
363                 printf("Error to get message buffer\n");
364                 return -1;
365         }
366
367         *(enum l2fwd_cmd *)msg = cmd;
368
369         if (rte_ring_enqueue(res->ring[fd], msg) != 0) {
370                 printf("Enqueue error\n");
371                 rte_mempool_put(message_pool, msg);
372                 return -1;
373         }
374
375         return 0;
376 }
377
378 /* Get command from pair in paired master and slave ring */
379 static inline int
380 getcmd(unsigned slaveid, enum l2fwd_cmd *cmd, int is_master)
381 {
382         struct lcore_resource_struct *res = &lcore_resource[slaveid];
383         void *msg;
384         int fd = !!is_master;
385         int ret;
386         /* Only check master, it must be enabled and running if it is slave */
387         if (is_master && (!res->enabled))
388                 return -1;
389
390         if (res->ring[fd] == NULL)
391                 return -1;
392
393         ret = rte_ring_dequeue(res->ring[fd], &msg);
394
395         if (ret == 0) {
396                 *cmd = *(enum l2fwd_cmd *)msg;
397                 rte_mempool_put(message_pool, msg);
398         }
399         return ret;
400 }
401
402 /* Master send command to slave and wait until ack received or error met */
403 static int
404 master_sendcmd_with_ack(unsigned slaveid, enum l2fwd_cmd cmd)
405 {
406         enum l2fwd_cmd ack_cmd;
407         int ret = -1;
408
409         if (sendcmd(slaveid, cmd, 1) != 0)
410                 rte_exit(EXIT_FAILURE, "Failed to send message\n");
411
412         /* Get ack */
413         while (1) {
414                 ret = getcmd(slaveid, &ack_cmd, 1);
415                 if (ret == 0 && cmd == ack_cmd)
416                         break;
417
418                 /* If slave not running yet, return an error */
419                 if (flib_query_slave_status(slaveid) != ST_RUN) {
420                         ret = -ENOENT;
421                         break;
422                 }
423         }
424
425         return ret;
426 }
427
428 /* restart all port that assigned to that slave lcore */
429 static int
430 reset_slave_all_ports(unsigned slaveid)
431 {
432         struct lcore_resource_struct *slave = &lcore_resource[slaveid];
433         int i, ret = 0;
434
435         /* stop/start port */
436         for (i = 0; i < slave->port_num; i++) {
437                 char buf_name[RTE_MEMPOOL_NAMESIZE];
438                 struct rte_mempool *pool;
439                 printf("Stop port :%d\n", slave->port[i]);
440                 rte_eth_dev_stop(slave->port[i]);
441                 snprintf(buf_name, RTE_MEMPOOL_NAMESIZE, MBUF_NAME, slave->port[i]);
442                 pool = rte_mempool_lookup(buf_name);
443                 if (pool)
444                         printf("Port %d mempool free object is %u(%u)\n", slave->port[i],
445                                 rte_mempool_avail_count(pool),
446                                 (unsigned int)NB_MBUF);
447                 else
448                         printf("Can't find mempool %s\n", buf_name);
449
450                 printf("Start port :%d\n", slave->port[i]);
451                 ret = rte_eth_dev_start(slave->port[i]);
452                 if (ret != 0)
453                         break;
454         }
455         return ret;
456 }
457
458 static int
459 reset_shared_structures(unsigned slaveid)
460 {
461         int ret;
462         /* Only port are shared resource here */
463         ret = reset_slave_all_ports(slaveid);
464
465         return ret;
466 }
467
468 /**
469  * Call this function to re-create resource that needed for slave process that
470  * exited in last instance
471  **/
472 static int
473 init_slave_res(unsigned slaveid)
474 {
475         struct lcore_resource_struct *slave = &lcore_resource[slaveid];
476         enum l2fwd_cmd cmd;
477
478         if (!slave->enabled) {
479                 printf("Something wrong with lcore=%u enabled=%d\n",slaveid,
480                         slave->enabled);
481                 return -1;
482         }
483
484         /* Initialize ring */
485         if (create_ms_ring(slaveid) != 0)
486                 rte_exit(EXIT_FAILURE, "failed to create ring for slave %u\n",
487                                 slaveid);
488
489         /* drain un-read buffer if have */
490         while (getcmd(slaveid, &cmd, 1) == 0);
491         while (getcmd(slaveid, &cmd, 0) == 0);
492
493         return 0;
494 }
495
496 static int
497 recreate_one_slave(unsigned slaveid)
498 {
499         int ret = 0;
500         /* Re-initialize resource for stalled slave */
501         if ((ret = init_slave_res(slaveid)) != 0) {
502                 printf("Init slave=%u failed\n", slaveid);
503                 return ret;
504         }
505
506         if ((ret = flib_remote_launch(l2fwd_launch_one_lcore, NULL, slaveid))
507                 != 0)
508                 printf("Launch slave %u failed\n", slaveid);
509
510         return ret;
511 }
512
513 /**
514  * remapping resource belong to slave_id to new lcore that gets from flib_assign_lcore_id(),
515  * used only floating process option applied.
516  *
517  * @param slaveid
518  *   original lcore_id that apply for remapping
519  */
520 static void
521 remapping_slave_resource(unsigned slaveid, unsigned map_id)
522 {
523
524         /* remapping lcore_resource */
525         memcpy(&lcore_resource[map_id], &lcore_resource[slaveid],
526                         sizeof(struct lcore_resource_struct));
527
528         /* remapping lcore_queue_conf */
529         memcpy(&lcore_queue_conf[map_id], &lcore_queue_conf[slaveid],
530                         sizeof(struct lcore_queue_conf));
531 }
532
533 static int
534 reset_pair(unsigned slaveid, unsigned pairid)
535 {
536         int ret;
537         if ((ret = reset_shared_structures(slaveid)) != 0)
538                 goto back;
539
540         if((ret = reset_shared_structures(pairid)) != 0)
541                 goto back;
542
543         if (float_proc) {
544                 unsigned map_id = mapping_id[slaveid];
545
546                 if (map_id != INVALID_MAPPING_ID) {
547                         printf("%u return mapping id %u\n", slaveid, map_id);
548                         flib_free_lcore_id(map_id);
549                         mapping_id[slaveid] = INVALID_MAPPING_ID;
550                 }
551
552                 map_id = mapping_id[pairid];
553                 if (map_id != INVALID_MAPPING_ID) {
554                         printf("%u return mapping id %u\n", pairid, map_id);
555                         flib_free_lcore_id(map_id);
556                         mapping_id[pairid] = INVALID_MAPPING_ID;
557                 }
558         }
559
560         if((ret = recreate_one_slave(slaveid)) != 0)
561                 goto back;
562
563         ret = recreate_one_slave(pairid);
564
565 back:
566         return ret;
567 }
568
569 static void
570 slave_exit_cb(unsigned slaveid, __attribute__((unused))int stat)
571 {
572         struct lcore_resource_struct *slave = &lcore_resource[slaveid];
573
574         printf("Get slave %u leave info\n", slaveid);
575         if (!slave->enabled) {
576                 printf("Lcore=%u not registered for it's exit\n", slaveid);
577                 return;
578         }
579         rte_spinlock_lock(&res_lock);
580
581         /* Change the state and wait master to start them */
582         slave->flags = SLAVE_RECREATE_FLAG;
583
584         rte_spinlock_unlock(&res_lock);
585 }
586
587 static void
588 l2fwd_simple_forward(struct rte_mbuf *m, unsigned portid)
589 {
590         struct ether_hdr *eth;
591         void *tmp;
592         unsigned dst_port;
593         int sent;
594         struct rte_eth_dev_tx_buffer *buffer;
595
596         dst_port = l2fwd_dst_ports[portid];
597         eth = rte_pktmbuf_mtod(m, struct ether_hdr *);
598
599         /* 02:00:00:00:00:xx */
600         tmp = &eth->d_addr.addr_bytes[0];
601         *((uint64_t *)tmp) = 0x000000000002 + ((uint64_t)dst_port << 40);
602
603         /* src addr */
604         ether_addr_copy(&l2fwd_ports_eth_addr[dst_port], &eth->s_addr);
605
606         buffer = tx_buffer[dst_port];
607         sent = rte_eth_tx_buffer(dst_port, 0, buffer, m);
608         if (sent)
609                 port_statistics[dst_port].tx += sent;
610 }
611
612 /* main processing loop */
613 static void
614 l2fwd_main_loop(void)
615 {
616         struct rte_mbuf *pkts_burst[MAX_PKT_BURST];
617         struct rte_mbuf *m;
618         int sent;
619         unsigned lcore_id;
620         uint64_t prev_tsc, diff_tsc, cur_tsc;
621         unsigned i, j, portid, nb_rx;
622         struct lcore_queue_conf *qconf;
623         const uint64_t drain_tsc = (rte_get_tsc_hz() + US_PER_S - 1) / US_PER_S *
624                         BURST_TX_DRAIN_US;
625         struct rte_eth_dev_tx_buffer *buffer;
626
627         prev_tsc = 0;
628
629         lcore_id = rte_lcore_id();
630
631         qconf = &lcore_queue_conf[lcore_id];
632
633         if (qconf->n_rx_port == 0) {
634                 RTE_LOG(INFO, L2FWD, "lcore %u has nothing to do\n", lcore_id);
635                 return;
636         }
637
638         RTE_LOG(INFO, L2FWD, "entering main loop on lcore %u\n", lcore_id);
639
640         for (i = 0; i < qconf->n_rx_port; i++) {
641                 portid = qconf->rx_port_list[i];
642                 RTE_LOG(INFO, L2FWD, " -- lcoreid=%u portid=%u\n", lcore_id,
643                         portid);
644         }
645
646         while (1) {
647                 enum l2fwd_cmd cmd;
648                 cur_tsc = rte_rdtsc();
649
650                 if (unlikely(getcmd(lcore_id, &cmd, 0) == 0)) {
651                         sendcmd(lcore_id, cmd, 0);
652
653                         /* If get stop command, stop forwarding and exit */
654                         if (cmd == CMD_STOP) {
655                                 return;
656                         }
657                 }
658
659                 /*
660                  * TX burst queue drain
661                  */
662                 diff_tsc = cur_tsc - prev_tsc;
663                 if (unlikely(diff_tsc > drain_tsc)) {
664
665                         for (i = 0; i < qconf->n_rx_port; i++) {
666
667                                 portid = l2fwd_dst_ports[qconf->rx_port_list[i]];
668                                 buffer = tx_buffer[portid];
669
670                                 sent = rte_eth_tx_buffer_flush(portid, 0, buffer);
671                                 if (sent)
672                                         port_statistics[portid].tx += sent;
673
674                         }
675
676                         prev_tsc = cur_tsc;
677                 }
678
679                 /*
680                  * Read packet from RX queues
681                  */
682                 for (i = 0; i < qconf->n_rx_port; i++) {
683
684                         portid = qconf->rx_port_list[i];
685                         nb_rx = rte_eth_rx_burst((uint8_t) portid, 0,
686                                                  pkts_burst, MAX_PKT_BURST);
687
688                         port_statistics[portid].rx += nb_rx;
689
690                         for (j = 0; j < nb_rx; j++) {
691                                 m = pkts_burst[j];
692                                 rte_prefetch0(rte_pktmbuf_mtod(m, void *));
693                                 l2fwd_simple_forward(m, portid);
694                         }
695                 }
696         }
697 }
698
699 static int
700 l2fwd_launch_one_lcore(__attribute__((unused)) void *dummy)
701 {
702         unsigned lcore_id = rte_lcore_id();
703
704         if (float_proc) {
705                 unsigned flcore_id;
706
707                 /* Change it to floating process, also change it's lcore_id */
708                 clear_cpu_affinity();
709                 RTE_PER_LCORE(_lcore_id) = 0;
710                 /* Get a lcore_id */
711                 if (flib_assign_lcore_id() < 0 ) {
712                         printf("flib_assign_lcore_id failed\n");
713                         return -1;
714                 }
715                 flcore_id = rte_lcore_id();
716                 /* Set mapping id, so master can return it after slave exited */
717                 mapping_id[lcore_id] = flcore_id;
718                 printf("Org lcore_id = %u, cur lcore_id = %u\n",
719                                 lcore_id, flcore_id);
720                 remapping_slave_resource(lcore_id, flcore_id);
721         }
722
723         l2fwd_main_loop();
724
725         /* return lcore_id before return */
726         if (float_proc) {
727                 flib_free_lcore_id(rte_lcore_id());
728                 mapping_id[lcore_id] = INVALID_MAPPING_ID;
729         }
730         return 0;
731 }
732
733 /* display usage */
734 static void
735 l2fwd_usage(const char *prgname)
736 {
737         printf("%s [EAL options] -- -p PORTMASK -s COREMASK [-q NQ] -f\n"
738                "  -p PORTMASK: hexadecimal bitmask of ports to configure\n"
739                "  -q NQ: number of queue (=ports) per lcore (default is 1)\n"
740                "  -f use floating process which won't bind to any core to run\n"
741                    "  -T PERIOD: statistics will be refreshed each PERIOD seconds (0 to disable, 10 default, 86400 maximum)\n",
742                prgname);
743 }
744
745 static int
746 l2fwd_parse_portmask(const char *portmask)
747 {
748         char *end = NULL;
749         unsigned long pm;
750
751         /* parse hexadecimal string */
752         pm = strtoul(portmask, &end, 16);
753         if ((portmask[0] == '\0') || (end == NULL) || (*end != '\0'))
754                 return -1;
755
756         if (pm == 0)
757                 return -1;
758
759         return pm;
760 }
761
762 static unsigned int
763 l2fwd_parse_nqueue(const char *q_arg)
764 {
765         char *end = NULL;
766         unsigned long n;
767
768         /* parse hexadecimal string */
769         n = strtoul(q_arg, &end, 10);
770         if ((q_arg[0] == '\0') || (end == NULL) || (*end != '\0'))
771                 return 0;
772         if (n == 0)
773                 return 0;
774         if (n >= MAX_RX_QUEUE_PER_LCORE)
775                 return 0;
776
777         return n;
778 }
779
780 static int
781 l2fwd_parse_timer_period(const char *q_arg)
782 {
783         char *end = NULL;
784         int n;
785
786         /* parse number string */
787         n = strtol(q_arg, &end, 10);
788         if ((q_arg[0] == '\0') || (end == NULL) || (*end != '\0'))
789                 return -1;
790         if (n >= MAX_TIMER_PERIOD)
791                 return -1;
792
793         return n;
794 }
795
796 /* Parse the argument given in the command line of the application */
797 static int
798 l2fwd_parse_args(int argc, char **argv)
799 {
800         int opt, ret;
801         char **argvopt;
802         int option_index;
803         char *prgname = argv[0];
804         static struct option lgopts[] = {
805                 {NULL, 0, 0, 0}
806         };
807         int has_pmask = 0;
808
809         argvopt = argv;
810
811         while ((opt = getopt_long(argc, argvopt, "p:q:T:f",
812                                   lgopts, &option_index)) != EOF) {
813
814                 switch (opt) {
815                 /* portmask */
816                 case 'p':
817                         l2fwd_enabled_port_mask = l2fwd_parse_portmask(optarg);
818                         if (l2fwd_enabled_port_mask == 0) {
819                                 printf("invalid portmask\n");
820                                 l2fwd_usage(prgname);
821                                 return -1;
822                         }
823                         has_pmask = 1;
824                         break;
825
826                 /* nqueue */
827                 case 'q':
828                         l2fwd_rx_queue_per_lcore = l2fwd_parse_nqueue(optarg);
829                         if (l2fwd_rx_queue_per_lcore == 0) {
830                                 printf("invalid queue number\n");
831                                 l2fwd_usage(prgname);
832                                 return -1;
833                         }
834                         break;
835
836                 /* timer period */
837                 case 'T':
838                         timer_period = l2fwd_parse_timer_period(optarg) * 1000 * TIMER_MILLISECOND;
839                         if (timer_period < 0) {
840                                 printf("invalid timer period\n");
841                                 l2fwd_usage(prgname);
842                                 return -1;
843                         }
844                         break;
845
846                 /* use floating process */
847                 case 'f':
848                         float_proc = 1;
849                         break;
850
851                 /* long options */
852                 case 0:
853                         l2fwd_usage(prgname);
854                         return -1;
855
856                 default:
857                         l2fwd_usage(prgname);
858                         return -1;
859                 }
860         }
861
862         if (optind >= 0)
863                 argv[optind-1] = prgname;
864
865         if (!has_pmask) {
866                 l2fwd_usage(prgname);
867                 return -1;
868         }
869         ret = optind-1;
870         optind = 0; /* reset getopt lib */
871         return ret;
872 }
873
874 /* Check the link status of all ports in up to 9s, and print them finally */
875 static void
876 check_all_ports_link_status(uint8_t port_num, uint32_t port_mask)
877 {
878 #define CHECK_INTERVAL 100 /* 100ms */
879 #define MAX_CHECK_TIME 90 /* 9s (90 * 100ms) in total */
880         uint8_t portid, count, all_ports_up, print_flag = 0;
881         struct rte_eth_link link;
882
883         printf("\nChecking link status");
884         fflush(stdout);
885         for (count = 0; count <= MAX_CHECK_TIME; count++) {
886                 all_ports_up = 1;
887                 for (portid = 0; portid < port_num; portid++) {
888                         if ((port_mask & (1 << portid)) == 0)
889                                 continue;
890                         memset(&link, 0, sizeof(link));
891                         rte_eth_link_get_nowait(portid, &link);
892                         /* print link status if flag set */
893                         if (print_flag == 1) {
894                                 if (link.link_status)
895                                         printf("Port %d Link Up - speed %u "
896                                                 "Mbps - %s\n", (uint8_t)portid,
897                                                 (unsigned)link.link_speed,
898                                 (link.link_duplex == ETH_LINK_FULL_DUPLEX) ?
899                                         ("full-duplex") : ("half-duplex\n"));
900                                 else
901                                         printf("Port %d Link Down\n",
902                                                 (uint8_t)portid);
903                                 continue;
904                         }
905                         /* clear all_ports_up flag if any link down */
906                         if (link.link_status == ETH_LINK_DOWN) {
907                                 all_ports_up = 0;
908                                 break;
909                         }
910                 }
911                 /* after finally printing all link status, get out */
912                 if (print_flag == 1)
913                         break;
914
915                 if (all_ports_up == 0) {
916                         printf(".");
917                         fflush(stdout);
918                         rte_delay_ms(CHECK_INTERVAL);
919                 }
920
921                 /* set the print_flag if all ports up or timeout */
922                 if (all_ports_up == 1 || count == (MAX_CHECK_TIME - 1)) {
923                         print_flag = 1;
924                         printf("done\n");
925                 }
926         }
927 }
928
929 int
930 main(int argc, char **argv)
931 {
932         struct lcore_queue_conf *qconf;
933         struct rte_eth_dev_info dev_info;
934         int ret;
935         uint8_t nb_ports;
936         uint8_t nb_ports_available;
937         uint8_t portid, last_port;
938         unsigned rx_lcore_id;
939         unsigned nb_ports_in_mask = 0;
940         unsigned i;
941         int flags = 0;
942         uint64_t prev_tsc, diff_tsc, cur_tsc, timer_tsc;
943
944         /* Save cpu_affinity first, restore it in case it's floating process option */
945         if (get_cpu_affinity() != 0)
946                 rte_exit(EXIT_FAILURE, "get_cpu_affinity error\n");
947
948         /* Also tries to set cpu affinity to detect whether  it will fail in child process */
949         if(clear_cpu_affinity() != 0)
950                 rte_exit(EXIT_FAILURE, "clear_cpu_affinity error\n");
951
952         /* init EAL */
953         ret = rte_eal_init(argc, argv);
954         if (ret < 0)
955                 rte_exit(EXIT_FAILURE, "Invalid EAL arguments\n");
956         argc -= ret;
957         argv += ret;
958
959         /* parse application arguments (after the EAL ones) */
960         ret = l2fwd_parse_args(argc, argv);
961         if (ret < 0)
962                 rte_exit(EXIT_FAILURE, "Invalid L2FWD arguments\n");
963
964         /*flib init */
965         if (flib_init() != 0)
966                 rte_exit(EXIT_FAILURE, "flib init error");
967
968         /**
969           * Allocated structures that slave lcore would change. For those that slaves are
970           * read only, needn't use malloc to share and global or static variables is ok since
971           * slave inherit all the knowledge that master initialized.
972           **/
973         if (l2fwd_malloc_shared_struct() != 0)
974                 rte_exit(EXIT_FAILURE, "malloc mem failed\n");
975
976         /* Initialize lcore_resource structures */
977         memset(lcore_resource, 0, sizeof(lcore_resource));
978         for (i = 0; i < RTE_MAX_LCORE; i++)
979                 lcore_resource[i].lcore_id = i;
980
981         nb_ports = rte_eth_dev_count();
982         if (nb_ports == 0)
983                 rte_exit(EXIT_FAILURE, "No Ethernet ports - bye\n");
984
985         /* create the mbuf pool */
986         for (portid = 0; portid < nb_ports; portid++) {
987                 /* skip ports that are not enabled */
988                 if ((l2fwd_enabled_port_mask & (1 << portid)) == 0)
989                         continue;
990                 char buf_name[RTE_MEMPOOL_NAMESIZE];
991                 flags = MEMPOOL_F_SP_PUT | MEMPOOL_F_SC_GET;
992                 snprintf(buf_name, RTE_MEMPOOL_NAMESIZE, MBUF_NAME, portid);
993                 l2fwd_pktmbuf_pool[portid] =
994                         rte_mempool_create(buf_name, NB_MBUF,
995                                            MBUF_SIZE, 32,
996                                            sizeof(struct rte_pktmbuf_pool_private),
997                                            rte_pktmbuf_pool_init, NULL,
998                                            rte_pktmbuf_init, NULL,
999                                            rte_socket_id(), flags);
1000                 if (l2fwd_pktmbuf_pool[portid] == NULL)
1001                         rte_exit(EXIT_FAILURE, "Cannot init mbuf pool\n");
1002
1003                 printf("Create mbuf %s\n", buf_name);
1004         }
1005
1006         /* reset l2fwd_dst_ports */
1007         for (portid = 0; portid < RTE_MAX_ETHPORTS; portid++)
1008                 l2fwd_dst_ports[portid] = 0;
1009         last_port = 0;
1010
1011         /*
1012          * Each logical core is assigned a dedicated TX queue on each port.
1013          */
1014         for (portid = 0; portid < nb_ports; portid++) {
1015                 /* skip ports that are not enabled */
1016                 if ((l2fwd_enabled_port_mask & (1 << portid)) == 0)
1017                         continue;
1018
1019                 if (nb_ports_in_mask % 2) {
1020                         l2fwd_dst_ports[portid] = last_port;
1021                         l2fwd_dst_ports[last_port] = portid;
1022                 }
1023                 else
1024                         last_port = portid;
1025
1026                 nb_ports_in_mask++;
1027
1028                 rte_eth_dev_info_get(portid, &dev_info);
1029         }
1030         if (nb_ports_in_mask % 2) {
1031                 printf("Notice: odd number of ports in portmask.\n");
1032                 l2fwd_dst_ports[last_port] = last_port;
1033         }
1034
1035         rx_lcore_id = 0;
1036         qconf = NULL;
1037
1038         /* Initialize the port/queue configuration of each logical core */
1039         for (portid = 0; portid < nb_ports; portid++) {
1040                 struct lcore_resource_struct *res;
1041                 /* skip ports that are not enabled */
1042                 if ((l2fwd_enabled_port_mask & (1 << portid)) == 0)
1043                         continue;
1044
1045                 /* get the lcore_id for this port */
1046                 /* skip master lcore */
1047                 while (rte_lcore_is_enabled(rx_lcore_id) == 0 ||
1048                            rte_get_master_lcore() == rx_lcore_id ||
1049                        lcore_queue_conf[rx_lcore_id].n_rx_port ==
1050                        l2fwd_rx_queue_per_lcore) {
1051
1052                         rx_lcore_id++;
1053                         if (rx_lcore_id >= RTE_MAX_LCORE)
1054                                 rte_exit(EXIT_FAILURE, "Not enough cores\n");
1055                 }
1056
1057                 if (qconf != &lcore_queue_conf[rx_lcore_id])
1058                         /* Assigned a new logical core in the loop above. */
1059                         qconf = &lcore_queue_conf[rx_lcore_id];
1060
1061                 qconf->rx_port_list[qconf->n_rx_port] = portid;
1062                 qconf->n_rx_port++;
1063
1064                 /* Save the port resource info into lcore_resource strucutres */
1065                 res = &lcore_resource[rx_lcore_id];
1066                 res->enabled = 1;
1067                 res->port[res->port_num++] = portid;
1068
1069                 printf("Lcore %u: RX port %u\n", rx_lcore_id, (unsigned) portid);
1070         }
1071
1072         nb_ports_available = nb_ports;
1073
1074         /* Initialise each port */
1075         for (portid = 0; portid < nb_ports; portid++) {
1076                 /* skip ports that are not enabled */
1077                 if ((l2fwd_enabled_port_mask & (1 << portid)) == 0) {
1078                         printf("Skipping disabled port %u\n", (unsigned) portid);
1079                         nb_ports_available--;
1080                         continue;
1081                 }
1082                 /* init port */
1083                 printf("Initializing port %u... ", (unsigned) portid);
1084                 fflush(stdout);
1085                 ret = rte_eth_dev_configure(portid, 1, 1, &port_conf);
1086                 if (ret < 0)
1087                         rte_exit(EXIT_FAILURE, "Cannot configure device: err=%d, port=%u\n",
1088                                   ret, (unsigned) portid);
1089
1090                 rte_eth_macaddr_get(portid,&l2fwd_ports_eth_addr[portid]);
1091
1092                 /* init one RX queue */
1093                 fflush(stdout);
1094                 ret = rte_eth_rx_queue_setup(portid, 0, nb_rxd,
1095                                              rte_eth_dev_socket_id(portid),
1096                                              NULL,
1097                                              l2fwd_pktmbuf_pool[portid]);
1098                 if (ret < 0)
1099                         rte_exit(EXIT_FAILURE, "rte_eth_rx_queue_setup:err=%d, port=%u\n",
1100                                   ret, (unsigned) portid);
1101
1102                 /* init one TX queue on each port */
1103                 fflush(stdout);
1104                 ret = rte_eth_tx_queue_setup(portid, 0, nb_txd,
1105                                 rte_eth_dev_socket_id(portid),
1106                                 NULL);
1107                 if (ret < 0)
1108                         rte_exit(EXIT_FAILURE, "rte_eth_tx_queue_setup:err=%d, port=%u\n",
1109                                 ret, (unsigned) portid);
1110
1111                 /* Initialize TX buffers */
1112                 tx_buffer[portid] = rte_zmalloc_socket("tx_buffer",
1113                                 RTE_ETH_TX_BUFFER_SIZE(MAX_PKT_BURST), 0,
1114                                 rte_eth_dev_socket_id(portid));
1115                 if (tx_buffer[portid] == NULL)
1116                         rte_exit(EXIT_FAILURE, "Cannot allocate buffer for tx on port %u\n",
1117                                         (unsigned) portid);
1118
1119                 rte_eth_tx_buffer_init(tx_buffer[portid], MAX_PKT_BURST);
1120
1121                 ret = rte_eth_tx_buffer_set_err_callback(tx_buffer[portid],
1122                                 rte_eth_tx_buffer_count_callback,
1123                                 &port_statistics[portid].dropped);
1124                 if (ret < 0)
1125                                 rte_exit(EXIT_FAILURE, "Cannot set error callback for "
1126                                                 "tx buffer on port %u\n", (unsigned) portid);
1127
1128                 /* Start device */
1129                 ret = rte_eth_dev_start(portid);
1130                 if (ret < 0)
1131                         rte_exit(EXIT_FAILURE, "rte_eth_dev_start:err=%d, port=%u\n",
1132                                   ret, (unsigned) portid);
1133
1134                 printf("done: \n");
1135
1136                 rte_eth_promiscuous_enable(portid);
1137
1138                 printf("Port %u, MAC address: %02X:%02X:%02X:%02X:%02X:%02X\n\n",
1139                                 (unsigned) portid,
1140                                 l2fwd_ports_eth_addr[portid].addr_bytes[0],
1141                                 l2fwd_ports_eth_addr[portid].addr_bytes[1],
1142                                 l2fwd_ports_eth_addr[portid].addr_bytes[2],
1143                                 l2fwd_ports_eth_addr[portid].addr_bytes[3],
1144                                 l2fwd_ports_eth_addr[portid].addr_bytes[4],
1145                                 l2fwd_ports_eth_addr[portid].addr_bytes[5]);
1146
1147                 /* initialize port stats */
1148                 //memset(&port_statistics, 0, sizeof(port_statistics));
1149         }
1150
1151         if (!nb_ports_available) {
1152                 rte_exit(EXIT_FAILURE,
1153                         "All available ports are disabled. Please set portmask.\n");
1154         }
1155
1156         check_all_ports_link_status(nb_ports, l2fwd_enabled_port_mask);
1157
1158         /* Record pair lcore */
1159         /**
1160          * Since l2fwd example would create pair between different neighbour port, that's
1161          * port 0 receive and forward to port 1, the same to port 1, these 2 ports will have
1162          * dependency. If one port stopped working (killed, for example), the port need to
1163          * be stopped/started again. During the time, another port need to wait until stop/start
1164          * procedure completed. So, record the pair relationship for those lcores working
1165          * on ports.
1166          **/
1167         for (portid = 0; portid < nb_ports; portid++) {
1168                 uint32_t pair_port;
1169                 unsigned lcore = 0, pair_lcore = 0;
1170                 unsigned j, find_lcore, find_pair_lcore;
1171                 /* skip ports that are not enabled */
1172                 if ((l2fwd_enabled_port_mask & (1 << portid)) == 0)
1173                         continue;
1174
1175                 /* Find pair ports' lcores */
1176                 find_lcore = find_pair_lcore = 0;
1177                 pair_port = l2fwd_dst_ports[portid];
1178                 for (i = 0; i < RTE_MAX_LCORE; i++) {
1179                         if (!rte_lcore_is_enabled(i))
1180                                 continue;
1181                         for (j = 0; j < lcore_queue_conf[i].n_rx_port;j++) {
1182                                 if (lcore_queue_conf[i].rx_port_list[j] == portid) {
1183                                         lcore = i;
1184                                         find_lcore = 1;
1185                                         break;
1186                                 }
1187                                 if (lcore_queue_conf[i].rx_port_list[j] == pair_port) {
1188                                         pair_lcore = i;
1189                                         find_pair_lcore = 1;
1190                                         break;
1191                                 }
1192                         }
1193                         if (find_lcore && find_pair_lcore)
1194                                 break;
1195                 }
1196                 if (!find_lcore || !find_pair_lcore)
1197                         rte_exit(EXIT_FAILURE, "Not find port=%d pair\n", portid);
1198
1199                 printf("lcore %u and %u paired\n", lcore, pair_lcore);
1200                 lcore_resource[lcore].pair_id = pair_lcore;
1201                 lcore_resource[pair_lcore].pair_id = lcore;
1202         }
1203
1204         /* Create message buffer for all master and slave */
1205         message_pool = rte_mempool_create("ms_msg_pool",
1206                            NB_CORE_MSGBUF * RTE_MAX_LCORE,
1207                            sizeof(enum l2fwd_cmd), NB_CORE_MSGBUF / 2,
1208                            0,
1209                            rte_pktmbuf_pool_init, NULL,
1210                            rte_pktmbuf_init, NULL,
1211                            rte_socket_id(), 0);
1212
1213         if (message_pool == NULL)
1214                 rte_exit(EXIT_FAILURE, "Create msg mempool failed\n");
1215
1216         /* Create ring for each master and slave pair, also register cb when slave leaves */
1217         for (i = 0; i < RTE_MAX_LCORE; i++) {
1218                 /**
1219                  * Only create ring and register slave_exit cb in case that core involved into
1220                  * packet forwarding
1221                  **/
1222                 if (lcore_resource[i].enabled) {
1223                         /* Create ring for master and slave communication */
1224                         ret = create_ms_ring(i);
1225                         if (ret != 0)
1226                                 rte_exit(EXIT_FAILURE, "Create ring for lcore=%u failed",
1227                                 i);
1228
1229                         if (flib_register_slave_exit_notify(i,
1230                                 slave_exit_cb) != 0)
1231                                 rte_exit(EXIT_FAILURE,
1232                                                 "Register master_trace_slave_exit failed");
1233                 }
1234         }
1235
1236         /* launch per-lcore init on every lcore except master */
1237         flib_mp_remote_launch(l2fwd_launch_one_lcore, NULL, SKIP_MASTER);
1238
1239         /* print statistics 10 second */
1240         prev_tsc = cur_tsc = rte_rdtsc();
1241         timer_tsc = 0;
1242         while (1) {
1243                 sleep(1);
1244                 cur_tsc = rte_rdtsc();
1245                 diff_tsc = cur_tsc - prev_tsc;
1246                 /* if timer is enabled */
1247                 if (timer_period > 0) {
1248
1249                         /* advance the timer */
1250                         timer_tsc += diff_tsc;
1251
1252                         /* if timer has reached its timeout */
1253                         if (unlikely(timer_tsc >= (uint64_t) timer_period)) {
1254
1255                                 print_stats();
1256                                 /* reset the timer */
1257                                 timer_tsc = 0;
1258                         }
1259                 }
1260
1261                 prev_tsc = cur_tsc;
1262
1263                 /* Check any slave need restart or recreate */
1264                 rte_spinlock_lock(&res_lock);
1265                 for (i = 0; i < RTE_MAX_LCORE; i++) {
1266                         struct lcore_resource_struct *res  = &lcore_resource[i];
1267                         struct lcore_resource_struct *pair = &lcore_resource[res->pair_id];
1268
1269                         /* If find slave exited, try to reset pair */
1270                         if (res->enabled && res->flags && pair->enabled) {
1271                                 if (!pair->flags) {
1272                                         master_sendcmd_with_ack(pair->lcore_id, CMD_STOP);
1273                                         rte_spinlock_unlock(&res_lock);
1274                                         sleep(1);
1275                                         rte_spinlock_lock(&res_lock);
1276                                         if (pair->flags)
1277                                                 continue;
1278                                 }
1279                                 if (reset_pair(res->lcore_id, pair->lcore_id) != 0)
1280                                         rte_exit(EXIT_FAILURE, "failed to reset slave");
1281                                 res->flags  = 0;
1282                                 pair->flags = 0;
1283                         }
1284                 }
1285                 rte_spinlock_unlock(&res_lock);
1286         }
1287
1288 }