examples/vhost/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
  34 #include <arpa/inet.h>
  35 #include <getopt.h>
  36 #include <linux/if_ether.h>
  37 #include <linux/if_vlan.h>
  38 #include <linux/virtio_net.h>
  39 #include <linux/virtio_ring.h>
  40 #include <signal.h>
  41 #include <stdint.h>
  42 #include <sys/eventfd.h>
  43 #include <sys/param.h>
  44 #include <unistd.h>
  45
  46 #include <rte_atomic.h>
  47 #include <rte_cycles.h>
  48 #include <rte_ethdev.h>
  49 #include <rte_log.h>
  50 #include <rte_string_fns.h>
  51 #include <rte_malloc.h>
  52 #include <rte_virtio_net.h>
  53 #include <rte_ip.h>
  54 #include <rte_tcp.h>
  55
  56 #include "main.h"
  57
  58 #ifndef MAX_QUEUES
  59 #define MAX_QUEUES 128
  60 #endif
  61
  62 /* the maximum number of external ports supported */
  63 #define MAX_SUP_PORTS 1
  64
  65 /*
  66  * Calculate the number of buffers needed per port
  67  */
  68 #define NUM_MBUFS_PER_PORT ((MAX_QUEUES*RTE_TEST_RX_DESC_DEFAULT) +             \
  69                                                         (num_switching_cores*MAX_PKT_BURST) +                   \
  70                                                         (num_switching_cores*RTE_TEST_TX_DESC_DEFAULT) +\
  71                                                         ((num_switching_cores+1)*MBUF_CACHE_SIZE))
  72
  73 #define MBUF_CACHE_SIZE 128
  74 #define MBUF_DATA_SIZE  RTE_MBUF_DEFAULT_BUF_SIZE
  75
  76 #define MAX_PKT_BURST 32                /* Max burst size for RX/TX */
  77 #define BURST_TX_DRAIN_US 100   /* TX drain every ~100us */
  78
  79 #define BURST_RX_WAIT_US 15     /* Defines how long we wait between retries on RX */
  80 #define BURST_RX_RETRIES 4              /* Number of retries on RX. */
  81
  82 #define JUMBO_FRAME_MAX_SIZE    0x2600
  83
  84 /* State of virtio device. */
  85 #define DEVICE_MAC_LEARNING 0
  86 #define DEVICE_RX                       1
  87 #define DEVICE_SAFE_REMOVE      2
  88
  89 /* Configurable number of RX/TX ring descriptors */
  90 #define RTE_TEST_RX_DESC_DEFAULT 1024
  91 #define RTE_TEST_TX_DESC_DEFAULT 512
  92
  93 #define INVALID_PORT_ID 0xFF
  94
  95 /* Max number of devices. Limited by vmdq. */
  96 #define MAX_DEVICES 64
  97
  98 /* Size of buffers used for snprintfs. */
  99 #define MAX_PRINT_BUFF 6072
 100
 101 /* Maximum character device basename size. */
 102 #define MAX_BASENAME_SZ 10
 103
 104 /* Maximum long option length for option parsing. */
 105 #define MAX_LONG_OPT_SZ 64
 106
 107 /* mask of enabled ports */
 108 static uint32_t enabled_port_mask = 0;
 109
 110 /* Promiscuous mode */
 111 static uint32_t promiscuous;
 112
 113 /*Number of switching cores enabled*/
 114 static uint32_t num_switching_cores = 0;
 115
 116 /* number of devices/queues to support*/
 117 static uint32_t num_queues = 0;
 118 static uint32_t num_devices;
 119
 120 static struct rte_mempool *mbuf_pool;
 121 static int mergeable;
 122
 123 /* Do vlan strip on host, enabled on default */
 124 static uint32_t vlan_strip = 1;
 125
 126 /* Enable VM2VM communications. If this is disabled then the MAC address compare is skipped. */
 127 typedef enum {
 128         VM2VM_DISABLED = 0,
 129         VM2VM_SOFTWARE = 1,
 130         VM2VM_HARDWARE = 2,
 131         VM2VM_LAST
 132 } vm2vm_type;
 133 static vm2vm_type vm2vm_mode = VM2VM_SOFTWARE;
 134
 135 /* Enable stats. */
 136 static uint32_t enable_stats = 0;
 137 /* Enable retries on RX. */
 138 static uint32_t enable_retry = 1;
 139
 140 /* Disable TX checksum offload */
 141 static uint32_t enable_tx_csum;
 142
 143 /* Disable TSO offload */
 144 static uint32_t enable_tso;
 145
 146 /* Specify timeout (in useconds) between retries on RX. */
 147 static uint32_t burst_rx_delay_time = BURST_RX_WAIT_US;
 148 /* Specify the number of retries on RX. */
 149 static uint32_t burst_rx_retry_num = BURST_RX_RETRIES;
 150
 151 /* Character device basename. Can be set by user. */
 152 static char dev_basename[MAX_BASENAME_SZ] = "vhost-net";
 153
 154 /* empty vmdq configuration structure. Filled in programatically */
 155 static struct rte_eth_conf vmdq_conf_default = {
 156         .rxmode = {
 157                 .mq_mode        = ETH_MQ_RX_VMDQ_ONLY,
 158                 .split_hdr_size = 0,
 159                 .header_split   = 0, /**< Header Split disabled */
 160                 .hw_ip_checksum = 0, /**< IP checksum offload disabled */
 161                 .hw_vlan_filter = 0, /**< VLAN filtering disabled */
 162                 /*
 163                  * It is necessary for 1G NIC such as I350,
 164                  * this fixes bug of ipv4 forwarding in guest can't
 165                  * forward pakets from one virtio dev to another virtio dev.
 166                  */
 167                 .hw_vlan_strip  = 1, /**< VLAN strip enabled. */
 168                 .jumbo_frame    = 0, /**< Jumbo Frame Support disabled */
 169                 .hw_strip_crc   = 0, /**< CRC stripped by hardware */
 170         },
 171
 172         .txmode = {
 173                 .mq_mode = ETH_MQ_TX_NONE,
 174         },
 175         .rx_adv_conf = {
 176                 /*
 177                  * should be overridden separately in code with
 178                  * appropriate values
 179                  */
 180                 .vmdq_rx_conf = {
 181                         .nb_queue_pools = ETH_8_POOLS,
 182                         .enable_default_pool = 0,
 183                         .default_pool = 0,
 184                         .nb_pool_maps = 0,
 185                         .pool_map = {{0, 0},},
 186                 },
 187         },
 188 };
 189
 190 static unsigned lcore_ids[RTE_MAX_LCORE];
 191 static uint8_t ports[RTE_MAX_ETHPORTS];
 192 static unsigned num_ports = 0; /**< The number of ports specified in command line */
 193 static uint16_t num_pf_queues, num_vmdq_queues;
 194 static uint16_t vmdq_pool_base, vmdq_queue_base;
 195 static uint16_t queues_per_pool;
 196
 197 const uint16_t vlan_tags[] = {
 198         1000, 1001, 1002, 1003, 1004, 1005, 1006, 1007,
 199         1008, 1009, 1010, 1011, 1012, 1013, 1014, 1015,
 200         1016, 1017, 1018, 1019, 1020, 1021, 1022, 1023,
 201         1024, 1025, 1026, 1027, 1028, 1029, 1030, 1031,
 202         1032, 1033, 1034, 1035, 1036, 1037, 1038, 1039,
 203         1040, 1041, 1042, 1043, 1044, 1045, 1046, 1047,
 204         1048, 1049, 1050, 1051, 1052, 1053, 1054, 1055,
 205         1056, 1057, 1058, 1059, 1060, 1061, 1062, 1063,
 206 };
 207
 208 /* ethernet addresses of ports */
 209 static struct ether_addr vmdq_ports_eth_addr[RTE_MAX_ETHPORTS];
 210
 211 static struct vhost_dev_tailq_list vhost_dev_list =
 212         TAILQ_HEAD_INITIALIZER(vhost_dev_list);
 213
 214 static struct lcore_info lcore_info[RTE_MAX_LCORE];
 215
 216 /* Used for queueing bursts of TX packets. */
 217 struct mbuf_table {
 218         unsigned len;
 219         unsigned txq_id;
 220         struct rte_mbuf *m_table[MAX_PKT_BURST];
 221 };
 222
 223 /* TX queue for each data core. */
 224 struct mbuf_table lcore_tx_queue[RTE_MAX_LCORE];
 225
 226 #define VLAN_HLEN       4
 227
 228 /* Per-device statistics struct */
 229 struct device_statistics {
 230         uint64_t tx_total;
 231         rte_atomic64_t rx_total_atomic;
 232         uint64_t tx;
 233         rte_atomic64_t rx_atomic;
 234 } __rte_cache_aligned;
 235 struct device_statistics dev_statistics[MAX_DEVICES];
 236
 237 /*
 238  * Builds up the correct configuration for VMDQ VLAN pool map
 239  * according to the pool & queue limits.
 240  */
 241 static inline int
 242 get_eth_conf(struct rte_eth_conf *eth_conf, uint32_t num_devices)
 243 {
 244         struct rte_eth_vmdq_rx_conf conf;
 245         struct rte_eth_vmdq_rx_conf *def_conf =
 246                 &vmdq_conf_default.rx_adv_conf.vmdq_rx_conf;
 247         unsigned i;
 248
 249         memset(&conf, 0, sizeof(conf));
 250         conf.nb_queue_pools = (enum rte_eth_nb_pools)num_devices;
 251         conf.nb_pool_maps = num_devices;
 252         conf.enable_loop_back = def_conf->enable_loop_back;
 253         conf.rx_mode = def_conf->rx_mode;
 254
 255         for (i = 0; i < conf.nb_pool_maps; i++) {
 256                 conf.pool_map[i].vlan_id = vlan_tags[ i ];
 257                 conf.pool_map[i].pools = (1UL << i);
 258         }
 259
 260         (void)(rte_memcpy(eth_conf, &vmdq_conf_default, sizeof(*eth_conf)));
 261         (void)(rte_memcpy(&eth_conf->rx_adv_conf.vmdq_rx_conf, &conf,
 262                    sizeof(eth_conf->rx_adv_conf.vmdq_rx_conf)));
 263         return 0;
 264 }
 265
 266 /*
 267  * Validate the device number according to the max pool number gotten form
 268  * dev_info. If the device number is invalid, give the error message and
 269  * return -1. Each device must have its own pool.
 270  */
 271 static inline int
 272 validate_num_devices(uint32_t max_nb_devices)
 273 {
 274         if (num_devices > max_nb_devices) {
 275                 RTE_LOG(ERR, VHOST_PORT, "invalid number of devices\n");
 276                 return -1;
 277         }
 278         return 0;
 279 }
 280
 281 /*
 282  * Initialises a given port using global settings and with the rx buffers
 283  * coming from the mbuf_pool passed as parameter
 284  */
 285 static inline int
 286 port_init(uint8_t port)
 287 {
 288         struct rte_eth_dev_info dev_info;
 289         struct rte_eth_conf port_conf;
 290         struct rte_eth_rxconf *rxconf;
 291         struct rte_eth_txconf *txconf;
 292         int16_t rx_rings, tx_rings;
 293         uint16_t rx_ring_size, tx_ring_size;
 294         int retval;
 295         uint16_t q;
 296
 297         /* The max pool number from dev_info will be used to validate the pool number specified in cmd line */
 298         rte_eth_dev_info_get (port, &dev_info);
 299
 300         if (dev_info.max_rx_queues > MAX_QUEUES) {
 301                 rte_exit(EXIT_FAILURE,
 302                         "please define MAX_QUEUES no less than %u in %s\n",
 303                         dev_info.max_rx_queues, __FILE__);
 304         }
 305
 306         rxconf = &dev_info.default_rxconf;
 307         txconf = &dev_info.default_txconf;
 308         rxconf->rx_drop_en = 1;
 309
 310         /* Enable vlan offload */
 311         txconf->txq_flags &= ~ETH_TXQ_FLAGS_NOVLANOFFL;
 312
 313         /*configure the number of supported virtio devices based on VMDQ limits */
 314         num_devices = dev_info.max_vmdq_pools;
 315
 316         rx_ring_size = RTE_TEST_RX_DESC_DEFAULT;
 317         tx_ring_size = RTE_TEST_TX_DESC_DEFAULT;
 318         tx_rings = (uint16_t)rte_lcore_count();
 319
 320         retval = validate_num_devices(MAX_DEVICES);
 321         if (retval < 0)
 322                 return retval;
 323
 324         /* Get port configuration. */
 325         retval = get_eth_conf(&port_conf, num_devices);
 326         if (retval < 0)
 327                 return retval;
 328         /* NIC queues are divided into pf queues and vmdq queues.  */
 329         num_pf_queues = dev_info.max_rx_queues - dev_info.vmdq_queue_num;
 330         queues_per_pool = dev_info.vmdq_queue_num / dev_info.max_vmdq_pools;
 331         num_vmdq_queues = num_devices * queues_per_pool;
 332         num_queues = num_pf_queues + num_vmdq_queues;
 333         vmdq_queue_base = dev_info.vmdq_queue_base;
 334         vmdq_pool_base  = dev_info.vmdq_pool_base;
 335         printf("pf queue num: %u, configured vmdq pool num: %u, each vmdq pool has %u queues\n",
 336                 num_pf_queues, num_devices, queues_per_pool);
 337
 338         if (port >= rte_eth_dev_count()) return -1;
 339
 340         if (enable_tx_csum == 0)
 341                 rte_vhost_feature_disable(1ULL << VIRTIO_NET_F_CSUM);
 342
 343         if (enable_tso == 0) {
 344                 rte_vhost_feature_disable(1ULL << VIRTIO_NET_F_HOST_TSO4);
 345                 rte_vhost_feature_disable(1ULL << VIRTIO_NET_F_HOST_TSO6);
 346         }
 347
 348         rx_rings = (uint16_t)dev_info.max_rx_queues;
 349         /* Configure ethernet device. */
 350         retval = rte_eth_dev_configure(port, rx_rings, tx_rings, &port_conf);
 351         if (retval != 0)
 352                 return retval;
 353
 354         /* Setup the queues. */
 355         for (q = 0; q < rx_rings; q ++) {
 356                 retval = rte_eth_rx_queue_setup(port, q, rx_ring_size,
 357                                                 rte_eth_dev_socket_id(port),
 358                                                 rxconf,
 359                                                 mbuf_pool);
 360                 if (retval < 0)
 361                         return retval;
 362         }
 363         for (q = 0; q < tx_rings; q ++) {
 364                 retval = rte_eth_tx_queue_setup(port, q, tx_ring_size,
 365                                                 rte_eth_dev_socket_id(port),
 366                                                 txconf);
 367                 if (retval < 0)
 368                         return retval;
 369         }
 370
 371         /* Start the device. */
 372         retval  = rte_eth_dev_start(port);
 373         if (retval < 0) {
 374                 RTE_LOG(ERR, VHOST_DATA, "Failed to start the device.\n");
 375                 return retval;
 376         }
 377
 378         if (promiscuous)
 379                 rte_eth_promiscuous_enable(port);
 380
 381         rte_eth_macaddr_get(port, &vmdq_ports_eth_addr[port]);
 382         RTE_LOG(INFO, VHOST_PORT, "Max virtio devices supported: %u\n", num_devices);
 383         RTE_LOG(INFO, VHOST_PORT, "Port %u MAC: %02"PRIx8" %02"PRIx8" %02"PRIx8
 384                         " %02"PRIx8" %02"PRIx8" %02"PRIx8"\n",
 385                         (unsigned)port,
 386                         vmdq_ports_eth_addr[port].addr_bytes[0],
 387                         vmdq_ports_eth_addr[port].addr_bytes[1],
 388                         vmdq_ports_eth_addr[port].addr_bytes[2],
 389                         vmdq_ports_eth_addr[port].addr_bytes[3],
 390                         vmdq_ports_eth_addr[port].addr_bytes[4],
 391                         vmdq_ports_eth_addr[port].addr_bytes[5]);
 392
 393         return 0;
 394 }
 395
 396 /*
 397  * Set character device basename.
 398  */
 399 static int
 400 us_vhost_parse_basename(const char *q_arg)
 401 {
 402         /* parse number string */
 403
 404         if (strnlen(q_arg, MAX_BASENAME_SZ) > MAX_BASENAME_SZ)
 405                 return -1;
 406         else
 407                 snprintf((char*)&dev_basename, MAX_BASENAME_SZ, "%s", q_arg);
 408
 409         return 0;
 410 }
 411
 412 /*
 413  * Parse the portmask provided at run time.
 414  */
 415 static int
 416 parse_portmask(const char *portmask)
 417 {
 418         char *end = NULL;
 419         unsigned long pm;
 420
 421         errno = 0;
 422
 423         /* parse hexadecimal string */
 424         pm = strtoul(portmask, &end, 16);
 425         if ((portmask[0] == '\0') || (end == NULL) || (*end != '\0') || (errno != 0))
 426                 return -1;
 427
 428         if (pm == 0)
 429                 return -1;
 430
 431         return pm;
 432
 433 }
 434
 435 /*
 436  * Parse num options at run time.
 437  */
 438 static int
 439 parse_num_opt(const char *q_arg, uint32_t max_valid_value)
 440 {
 441         char *end = NULL;
 442         unsigned long num;
 443
 444         errno = 0;
 445
 446         /* parse unsigned int string */
 447         num = strtoul(q_arg, &end, 10);
 448         if ((q_arg[0] == '\0') || (end == NULL) || (*end != '\0') || (errno != 0))
 449                 return -1;
 450
 451         if (num > max_valid_value)
 452                 return -1;
 453
 454         return num;
 455
 456 }
 457
 458 /*
 459  * Display usage
 460  */
 461 static void
 462 us_vhost_usage(const char *prgname)
 463 {
 464         RTE_LOG(INFO, VHOST_CONFIG, "%s [EAL options] -- -p PORTMASK\n"
 465         "               --vm2vm [0|1|2]\n"
 466         "               --rx_retry [0|1] --mergeable [0|1] --stats [0-N]\n"
 467         "               --dev-basename <name>\n"
 468         "               --nb-devices ND\n"
 469         "               -p PORTMASK: Set mask for ports to be used by application\n"
 470         "               --vm2vm [0|1|2]: disable/software(default)/hardware vm2vm comms\n"
 471         "               --rx-retry [0|1]: disable/enable(default) retries on rx. Enable retry if destintation queue is full\n"
 472         "               --rx-retry-delay [0-N]: timeout(in usecond) between retries on RX. This makes effect only if retries on rx enabled\n"
 473         "               --rx-retry-num [0-N]: the number of retries on rx. This makes effect only if retries on rx enabled\n"
 474         "               --mergeable [0|1]: disable(default)/enable RX mergeable buffers\n"
 475         "               --vlan-strip [0|1]: disable/enable(default) RX VLAN strip on host\n"
 476         "               --stats [0-N]: 0: Disable stats, N: Time in seconds to print stats\n"
 477         "               --dev-basename: The basename to be used for the character device.\n"
 478         "               --tx-csum [0|1] disable/enable TX checksum offload.\n"
 479         "               --tso [0|1] disable/enable TCP segment offload.\n",
 480                prgname);
 481 }
 482
 483 /*
 484  * Parse the arguments given in the command line of the application.
 485  */
 486 static int
 487 us_vhost_parse_args(int argc, char **argv)
 488 {
 489         int opt, ret;
 490         int option_index;
 491         unsigned i;
 492         const char *prgname = argv[0];
 493         static struct option long_option[] = {
 494                 {"vm2vm", required_argument, NULL, 0},
 495                 {"rx-retry", required_argument, NULL, 0},
 496                 {"rx-retry-delay", required_argument, NULL, 0},
 497                 {"rx-retry-num", required_argument, NULL, 0},
 498                 {"mergeable", required_argument, NULL, 0},
 499                 {"vlan-strip", required_argument, NULL, 0},
 500                 {"stats", required_argument, NULL, 0},
 501                 {"dev-basename", required_argument, NULL, 0},
 502                 {"tx-csum", required_argument, NULL, 0},
 503                 {"tso", required_argument, NULL, 0},
 504                 {NULL, 0, 0, 0},
 505         };
 506
 507         /* Parse command line */
 508         while ((opt = getopt_long(argc, argv, "p:P",
 509                         long_option, &option_index)) != EOF) {
 510                 switch (opt) {
 511                 /* Portmask */
 512                 case 'p':
 513                         enabled_port_mask = parse_portmask(optarg);
 514                         if (enabled_port_mask == 0) {
 515                                 RTE_LOG(INFO, VHOST_CONFIG, "Invalid portmask\n");
 516                                 us_vhost_usage(prgname);
 517                                 return -1;
 518                         }
 519                         break;
 520
 521                 case 'P':
 522                         promiscuous = 1;
 523                         vmdq_conf_default.rx_adv_conf.vmdq_rx_conf.rx_mode =
 524                                 ETH_VMDQ_ACCEPT_BROADCAST |
 525                                 ETH_VMDQ_ACCEPT_MULTICAST;
 526                         rte_vhost_feature_enable(1ULL << VIRTIO_NET_F_CTRL_RX);
 527
 528                         break;
 529
 530                 case 0:
 531                         /* Enable/disable vm2vm comms. */
 532                         if (!strncmp(long_option[option_index].name, "vm2vm",
 533                                 MAX_LONG_OPT_SZ)) {
 534                                 ret = parse_num_opt(optarg, (VM2VM_LAST - 1));
 535                                 if (ret == -1) {
 536                                         RTE_LOG(INFO, VHOST_CONFIG,
 537                                                 "Invalid argument for "
 538                                                 "vm2vm [0|1|2]\n");
 539                                         us_vhost_usage(prgname);
 540                                         return -1;
 541                                 } else {
 542                                         vm2vm_mode = (vm2vm_type)ret;
 543                                 }
 544                         }
 545
 546                         /* Enable/disable retries on RX. */
 547                         if (!strncmp(long_option[option_index].name, "rx-retry", MAX_LONG_OPT_SZ)) {
 548                                 ret = parse_num_opt(optarg, 1);
 549                                 if (ret == -1) {
 550                                         RTE_LOG(INFO, VHOST_CONFIG, "Invalid argument for rx-retry [0|1]\n");
 551                                         us_vhost_usage(prgname);
 552                                         return -1;
 553                                 } else {
 554                                         enable_retry = ret;
 555                                 }
 556                         }
 557
 558                         /* Enable/disable TX checksum offload. */
 559                         if (!strncmp(long_option[option_index].name, "tx-csum", MAX_LONG_OPT_SZ)) {
 560                                 ret = parse_num_opt(optarg, 1);
 561                                 if (ret == -1) {
 562                                         RTE_LOG(INFO, VHOST_CONFIG, "Invalid argument for tx-csum [0|1]\n");
 563                                         us_vhost_usage(prgname);
 564                                         return -1;
 565                                 } else
 566                                         enable_tx_csum = ret;
 567                         }
 568
 569                         /* Enable/disable TSO offload. */
 570                         if (!strncmp(long_option[option_index].name, "tso", MAX_LONG_OPT_SZ)) {
 571                                 ret = parse_num_opt(optarg, 1);
 572                                 if (ret == -1) {
 573                                         RTE_LOG(INFO, VHOST_CONFIG, "Invalid argument for tso [0|1]\n");
 574                                         us_vhost_usage(prgname);
 575                                         return -1;
 576                                 } else
 577                                         enable_tso = ret;
 578                         }
 579
 580                         /* Specify the retries delay time (in useconds) on RX. */
 581                         if (!strncmp(long_option[option_index].name, "rx-retry-delay", MAX_LONG_OPT_SZ)) {
 582                                 ret = parse_num_opt(optarg, INT32_MAX);
 583                                 if (ret == -1) {
 584                                         RTE_LOG(INFO, VHOST_CONFIG, "Invalid argument for rx-retry-delay [0-N]\n");
 585                                         us_vhost_usage(prgname);
 586                                         return -1;
 587                                 } else {
 588                                         burst_rx_delay_time = ret;
 589                                 }
 590                         }
 591
 592                         /* Specify the retries number on RX. */
 593                         if (!strncmp(long_option[option_index].name, "rx-retry-num", MAX_LONG_OPT_SZ)) {
 594                                 ret = parse_num_opt(optarg, INT32_MAX);
 595                                 if (ret == -1) {
 596                                         RTE_LOG(INFO, VHOST_CONFIG, "Invalid argument for rx-retry-num [0-N]\n");
 597                                         us_vhost_usage(prgname);
 598                                         return -1;
 599                                 } else {
 600                                         burst_rx_retry_num = ret;
 601                                 }
 602                         }
 603
 604                         /* Enable/disable RX mergeable buffers. */
 605                         if (!strncmp(long_option[option_index].name, "mergeable", MAX_LONG_OPT_SZ)) {
 606                                 ret = parse_num_opt(optarg, 1);
 607                                 if (ret == -1) {
 608                                         RTE_LOG(INFO, VHOST_CONFIG, "Invalid argument for mergeable [0|1]\n");
 609                                         us_vhost_usage(prgname);
 610                                         return -1;
 611                                 } else {
 612                                         mergeable = !!ret;
 613                                         if (ret) {
 614                                                 vmdq_conf_default.rxmode.jumbo_frame = 1;
 615                                                 vmdq_conf_default.rxmode.max_rx_pkt_len
 616                                                         = JUMBO_FRAME_MAX_SIZE;
 617                                         }
 618                                 }
 619                         }
 620
 621                         /* Enable/disable RX VLAN strip on host. */
 622                         if (!strncmp(long_option[option_index].name,
 623                                 "vlan-strip", MAX_LONG_OPT_SZ)) {
 624                                 ret = parse_num_opt(optarg, 1);
 625                                 if (ret == -1) {
 626                                         RTE_LOG(INFO, VHOST_CONFIG,
 627                                                 "Invalid argument for VLAN strip [0|1]\n");
 628                                         us_vhost_usage(prgname);
 629                                         return -1;
 630                                 } else {
 631                                         vlan_strip = !!ret;
 632                                         vmdq_conf_default.rxmode.hw_vlan_strip =
 633                                                 vlan_strip;
 634                                 }
 635                         }
 636
 637                         /* Enable/disable stats. */
 638                         if (!strncmp(long_option[option_index].name, "stats", MAX_LONG_OPT_SZ)) {
 639                                 ret = parse_num_opt(optarg, INT32_MAX);
 640                                 if (ret == -1) {
 641                                         RTE_LOG(INFO, VHOST_CONFIG, "Invalid argument for stats [0..N]\n");
 642                                         us_vhost_usage(prgname);
 643                                         return -1;
 644                                 } else {
 645                                         enable_stats = ret;
 646                                 }
 647                         }
 648
 649                         /* Set character device basename. */
 650                         if (!strncmp(long_option[option_index].name, "dev-basename", MAX_LONG_OPT_SZ)) {
 651                                 if (us_vhost_parse_basename(optarg) == -1) {
 652                                         RTE_LOG(INFO, VHOST_CONFIG, "Invalid argument for character device basename (Max %d characters)\n", MAX_BASENAME_SZ);
 653                                         us_vhost_usage(prgname);
 654                                         return -1;
 655                                 }
 656                         }
 657
 658                         break;
 659
 660                         /* Invalid option - print options. */
 661                 default:
 662                         us_vhost_usage(prgname);
 663                         return -1;
 664                 }
 665         }
 666
 667         for (i = 0; i < RTE_MAX_ETHPORTS; i++) {
 668                 if (enabled_port_mask & (1 << i))
 669                         ports[num_ports++] = (uint8_t)i;
 670         }
 671
 672         if ((num_ports ==  0) || (num_ports > MAX_SUP_PORTS)) {
 673                 RTE_LOG(INFO, VHOST_PORT, "Current enabled port number is %u,"
 674                         "but only %u port can be enabled\n",num_ports, MAX_SUP_PORTS);
 675                 return -1;
 676         }
 677
 678         return 0;
 679 }
 680
 681 /*
 682  * Update the global var NUM_PORTS and array PORTS according to system ports number
 683  * and return valid ports number
 684  */
 685 static unsigned check_ports_num(unsigned nb_ports)
 686 {
 687         unsigned valid_num_ports = num_ports;
 688         unsigned portid;
 689
 690         if (num_ports > nb_ports) {
 691                 RTE_LOG(INFO, VHOST_PORT, "\nSpecified port number(%u) exceeds total system port number(%u)\n",
 692                         num_ports, nb_ports);
 693                 num_ports = nb_ports;
 694         }
 695
 696         for (portid = 0; portid < num_ports; portid ++) {
 697                 if (ports[portid] >= nb_ports) {
 698                         RTE_LOG(INFO, VHOST_PORT, "\nSpecified port ID(%u) exceeds max system port ID(%u)\n",
 699                                 ports[portid], (nb_ports - 1));
 700                         ports[portid] = INVALID_PORT_ID;
 701                         valid_num_ports--;
 702                 }
 703         }
 704         return valid_num_ports;
 705 }
 706
 707 static inline struct vhost_dev *__attribute__((always_inline))
 708 find_vhost_dev(struct ether_addr *mac)
 709 {
 710         struct vhost_dev *vdev;
 711
 712         TAILQ_FOREACH(vdev, &vhost_dev_list, next) {
 713                 if (vdev->ready == DEVICE_RX &&
 714                     is_same_ether_addr(mac, &vdev->mac_address))
 715                         return vdev;
 716         }
 717
 718         return NULL;
 719 }
 720
 721 /*
 722  * This function learns the MAC address of the device and registers this along with a
 723  * vlan tag to a VMDQ.
 724  */
 725 static int
 726 link_vmdq(struct vhost_dev *vdev, struct rte_mbuf *m)
 727 {
 728         struct ether_hdr *pkt_hdr;
 729         struct virtio_net *dev = vdev->dev;
 730         int i, ret;
 731
 732         /* Learn MAC address of guest device from packet */
 733         pkt_hdr = rte_pktmbuf_mtod(m, struct ether_hdr *);
 734
 735         if (find_vhost_dev(&pkt_hdr->s_addr)) {
 736                 RTE_LOG(ERR, VHOST_DATA,
 737                         "Device (%" PRIu64 ") is using a registered MAC!\n",
 738                         dev->device_fh);
 739                 return -1;
 740         }
 741
 742         for (i = 0; i < ETHER_ADDR_LEN; i++)
 743                 vdev->mac_address.addr_bytes[i] = pkt_hdr->s_addr.addr_bytes[i];
 744
 745         /* vlan_tag currently uses the device_id. */
 746         vdev->vlan_tag = vlan_tags[dev->device_fh];
 747
 748         /* Print out VMDQ registration info. */
 749         RTE_LOG(INFO, VHOST_DATA, "(%"PRIu64") MAC_ADDRESS %02x:%02x:%02x:%02x:%02x:%02x and VLAN_TAG %d registered\n",
 750                 dev->device_fh,
 751                 vdev->mac_address.addr_bytes[0], vdev->mac_address.addr_bytes[1],
 752                 vdev->mac_address.addr_bytes[2], vdev->mac_address.addr_bytes[3],
 753                 vdev->mac_address.addr_bytes[4], vdev->mac_address.addr_bytes[5],
 754                 vdev->vlan_tag);
 755
 756         /* Register the MAC address. */
 757         ret = rte_eth_dev_mac_addr_add(ports[0], &vdev->mac_address,
 758                                 (uint32_t)dev->device_fh + vmdq_pool_base);
 759         if (ret)
 760                 RTE_LOG(ERR, VHOST_DATA, "(%"PRIu64") Failed to add device MAC address to VMDQ\n",
 761                                         dev->device_fh);
 762
 763         /* Enable stripping of the vlan tag as we handle routing. */
 764         if (vlan_strip)
 765                 rte_eth_dev_set_vlan_strip_on_queue(ports[0],
 766                         (uint16_t)vdev->vmdq_rx_q, 1);
 767
 768         /* Set device as ready for RX. */
 769         vdev->ready = DEVICE_RX;
 770
 771         return 0;
 772 }
 773
 774 /*
 775  * Removes MAC address and vlan tag from VMDQ. Ensures that nothing is adding buffers to the RX
 776  * queue before disabling RX on the device.
 777  */
 778 static inline void
 779 unlink_vmdq(struct vhost_dev *vdev)
 780 {
 781         unsigned i = 0;
 782         unsigned rx_count;
 783         struct rte_mbuf *pkts_burst[MAX_PKT_BURST];
 784
 785         if (vdev->ready == DEVICE_RX) {
 786                 /*clear MAC and VLAN settings*/
 787                 rte_eth_dev_mac_addr_remove(ports[0], &vdev->mac_address);
 788                 for (i = 0; i < 6; i++)
 789                         vdev->mac_address.addr_bytes[i] = 0;
 790
 791                 vdev->vlan_tag = 0;
 792
 793                 /*Clear out the receive buffers*/
 794                 rx_count = rte_eth_rx_burst(ports[0],
 795                                         (uint16_t)vdev->vmdq_rx_q, pkts_burst, MAX_PKT_BURST);
 796
 797                 while (rx_count) {
 798                         for (i = 0; i < rx_count; i++)
 799                                 rte_pktmbuf_free(pkts_burst[i]);
 800
 801                         rx_count = rte_eth_rx_burst(ports[0],
 802                                         (uint16_t)vdev->vmdq_rx_q, pkts_burst, MAX_PKT_BURST);
 803                 }
 804
 805                 vdev->ready = DEVICE_MAC_LEARNING;
 806         }
 807 }
 808
 809 static inline void __attribute__((always_inline))
 810 virtio_xmit(struct virtio_net *dst_dev, struct virtio_net *src_dev,
 811             struct rte_mbuf *m)
 812 {
 813         uint16_t ret;
 814
 815         ret = rte_vhost_enqueue_burst(dst_dev, VIRTIO_RXQ, &m, 1);
 816         if (enable_stats) {
 817                 rte_atomic64_inc(&dev_statistics[dst_dev->device_fh].rx_total_atomic);
 818                 rte_atomic64_add(&dev_statistics[dst_dev->device_fh].rx_atomic, ret);
 819                 dev_statistics[src_dev->device_fh].tx_total++;
 820                 dev_statistics[src_dev->device_fh].tx += ret;
 821         }
 822 }
 823
 824 /*
 825  * Check if the packet destination MAC address is for a local device. If so then put
 826  * the packet on that devices RX queue. If not then return.
 827  */
 828 static inline int __attribute__((always_inline))
 829 virtio_tx_local(struct vhost_dev *vdev, struct rte_mbuf *m)
 830 {
 831         struct ether_hdr *pkt_hdr;
 832         struct vhost_dev *dst_vdev;
 833         uint64_t fh;
 834
 835         pkt_hdr = rte_pktmbuf_mtod(m, struct ether_hdr *);
 836
 837         dst_vdev = find_vhost_dev(&pkt_hdr->d_addr);
 838         if (!dst_vdev)
 839                 return -1;
 840
 841         fh = dst_vdev->dev->device_fh;
 842         if (fh == vdev->dev->device_fh) {
 843                 RTE_LOG(DEBUG, VHOST_DATA,
 844                         "(%" PRIu64 ") TX: src and dst MAC is same. "
 845                         "Dropping packet.\n", fh);
 846                 return 0;
 847         }
 848
 849         RTE_LOG(DEBUG, VHOST_DATA,
 850                 "(%" PRIu64 ") TX: MAC address is local\n", fh);
 851
 852         if (unlikely(dst_vdev->remove)) {
 853                 RTE_LOG(DEBUG, VHOST_DATA, "(%" PRIu64 ") "
 854                         "Device is marked for removal\n", fh);
 855                 return 0;
 856         }
 857
 858         virtio_xmit(dst_vdev->dev, vdev->dev, m);
 859         return 0;
 860 }
 861
 862 /*
 863  * Check if the destination MAC of a packet is one local VM,
 864  * and get its vlan tag, and offset if it is.
 865  */
 866 static inline int __attribute__((always_inline))
 867 find_local_dest(struct virtio_net *dev, struct rte_mbuf *m,
 868         uint32_t *offset, uint16_t *vlan_tag)
 869 {
 870         struct vhost_dev *dst_vdev;
 871         struct ether_hdr *pkt_hdr = rte_pktmbuf_mtod(m, struct ether_hdr *);
 872
 873         dst_vdev = find_vhost_dev(&pkt_hdr->d_addr);
 874         if (!dst_vdev)
 875                 return 0;
 876
 877         if (dst_vdev->dev->device_fh == dev->device_fh) {
 878                 RTE_LOG(DEBUG, VHOST_DATA,
 879                         "(%" PRIu64 ") TX: src and dst MAC is same. "
 880                         " Dropping packet.\n", dst_vdev->dev->device_fh);
 881                 return -1;
 882         }
 883
 884         /*
 885          * HW vlan strip will reduce the packet length
 886          * by minus length of vlan tag, so need restore
 887          * the packet length by plus it.
 888          */
 889         *offset  = VLAN_HLEN;
 890         *vlan_tag = vlan_tags[(uint16_t)dst_vdev->dev->device_fh];
 891
 892         RTE_LOG(DEBUG, VHOST_DATA,
 893                 "(%" PRIu64 ") TX: pkt to local VM device id: (%" PRIu64 ") "
 894                 "vlan tag: %u.\n",
 895                 dev->device_fh, dst_vdev->dev->device_fh, *vlan_tag);
 896
 897         return 0;
 898 }
 899
 900 static uint16_t
 901 get_psd_sum(void *l3_hdr, uint64_t ol_flags)
 902 {
 903         if (ol_flags & PKT_TX_IPV4)
 904                 return rte_ipv4_phdr_cksum(l3_hdr, ol_flags);
 905         else /* assume ethertype == ETHER_TYPE_IPv6 */
 906                 return rte_ipv6_phdr_cksum(l3_hdr, ol_flags);
 907 }
 908
 909 static void virtio_tx_offload(struct rte_mbuf *m)
 910 {
 911         void *l3_hdr;
 912         struct ipv4_hdr *ipv4_hdr = NULL;
 913         struct tcp_hdr *tcp_hdr = NULL;
 914         struct ether_hdr *eth_hdr = rte_pktmbuf_mtod(m, struct ether_hdr *);
 915
 916         l3_hdr = (char *)eth_hdr + m->l2_len;
 917
 918         if (m->ol_flags & PKT_TX_IPV4) {
 919                 ipv4_hdr = l3_hdr;
 920                 ipv4_hdr->hdr_checksum = 0;
 921                 m->ol_flags |= PKT_TX_IP_CKSUM;
 922         }
 923
 924         tcp_hdr = (struct tcp_hdr *)((char *)l3_hdr + m->l3_len);
 925         tcp_hdr->cksum = get_psd_sum(l3_hdr, m->ol_flags);
 926 }
 927
 928 /*
 929  * This function routes the TX packet to the correct interface. This may be a local device
 930  * or the physical port.
 931  */
 932 static inline void __attribute__((always_inline))
 933 virtio_tx_route(struct vhost_dev *vdev, struct rte_mbuf *m, uint16_t vlan_tag)
 934 {
 935         struct mbuf_table *tx_q;
 936         struct rte_mbuf **m_table;
 937         unsigned len, ret, offset = 0;
 938         const uint16_t lcore_id = rte_lcore_id();
 939         struct virtio_net *dev = vdev->dev;
 940         struct ether_hdr *nh;
 941
 942
 943         nh = rte_pktmbuf_mtod(m, struct ether_hdr *);
 944         if (unlikely(is_broadcast_ether_addr(&nh->d_addr))) {
 945                 struct vhost_dev *vdev2;
 946
 947                 TAILQ_FOREACH(vdev2, &vhost_dev_list, next) {
 948                         virtio_xmit(vdev2->dev, vdev->dev, m);
 949                 }
 950                 goto queue2nic;
 951         }
 952
 953         /*check if destination is local VM*/
 954         if ((vm2vm_mode == VM2VM_SOFTWARE) && (virtio_tx_local(vdev, m) == 0)) {
 955                 rte_pktmbuf_free(m);
 956                 return;
 957         }
 958
 959         if (unlikely(vm2vm_mode == VM2VM_HARDWARE)) {
 960                 if (unlikely(find_local_dest(dev, m, &offset, &vlan_tag) != 0)) {
 961                         rte_pktmbuf_free(m);
 962                         return;
 963                 }
 964         }
 965
 966         RTE_LOG(DEBUG, VHOST_DATA, "(%" PRIu64 ") TX: "
 967                 "MAC address is external\n", dev->device_fh);
 968
 969 queue2nic:
 970
 971         /*Add packet to the port tx queue*/
 972         tx_q = &lcore_tx_queue[lcore_id];
 973         len = tx_q->len;
 974
 975         nh = rte_pktmbuf_mtod(m, struct ether_hdr *);
 976         if (unlikely(nh->ether_type == rte_cpu_to_be_16(ETHER_TYPE_VLAN))) {
 977                 /* Guest has inserted the vlan tag. */
 978                 struct vlan_hdr *vh = (struct vlan_hdr *) (nh + 1);
 979                 uint16_t vlan_tag_be = rte_cpu_to_be_16(vlan_tag);
 980                 if ((vm2vm_mode == VM2VM_HARDWARE) &&
 981                         (vh->vlan_tci != vlan_tag_be))
 982                         vh->vlan_tci = vlan_tag_be;
 983         } else {
 984                 m->ol_flags |= PKT_TX_VLAN_PKT;
 985
 986                 /*
 987                  * Find the right seg to adjust the data len when offset is
 988                  * bigger than tail room size.
 989                  */
 990                 if (unlikely(vm2vm_mode == VM2VM_HARDWARE)) {
 991                         if (likely(offset <= rte_pktmbuf_tailroom(m)))
 992                                 m->data_len += offset;
 993                         else {
 994                                 struct rte_mbuf *seg = m;
 995
 996                                 while ((seg->next != NULL) &&
 997                                         (offset > rte_pktmbuf_tailroom(seg)))
 998                                         seg = seg->next;
 999
1000                                 seg->data_len += offset;
1001                         }
1002                         m->pkt_len += offset;
1003                 }
1004
1005                 m->vlan_tci = vlan_tag;
1006         }
1007
1008         if (m->ol_flags & PKT_TX_TCP_SEG)
1009                 virtio_tx_offload(m);
1010
1011         tx_q->m_table[len] = m;
1012         len++;
1013         if (enable_stats) {
1014                 dev_statistics[dev->device_fh].tx_total++;
1015                 dev_statistics[dev->device_fh].tx++;
1016         }
1017
1018         if (unlikely(len == MAX_PKT_BURST)) {
1019                 m_table = (struct rte_mbuf **)tx_q->m_table;
1020                 ret = rte_eth_tx_burst(ports[0], (uint16_t)tx_q->txq_id, m_table, (uint16_t) len);
1021                 /* Free any buffers not handled by TX and update the port stats. */
1022                 if (unlikely(ret < len)) {
1023                         do {
1024                                 rte_pktmbuf_free(m_table[ret]);
1025                         } while (++ret < len);
1026                 }
1027
1028                 len = 0;
1029         }
1030
1031         tx_q->len = len;
1032         return;
1033 }
1034 /*
1035  * This function is called by each data core. It handles all RX/TX registered with the
1036  * core. For TX the specific lcore linked list is used. For RX, MAC addresses are compared
1037  * with all devices in the main linked list.
1038  */
1039 static int
1040 switch_worker(__attribute__((unused)) void *arg)
1041 {
1042         struct virtio_net *dev = NULL;
1043         struct vhost_dev *vdev = NULL;
1044         struct rte_mbuf *pkts_burst[MAX_PKT_BURST];
1045         struct mbuf_table *tx_q;
1046         const uint64_t drain_tsc = (rte_get_tsc_hz() + US_PER_S - 1) / US_PER_S * BURST_TX_DRAIN_US;
1047         uint64_t prev_tsc, diff_tsc, cur_tsc, ret_count = 0;
1048         unsigned ret, i;
1049         const uint16_t lcore_id = rte_lcore_id();
1050         const uint16_t num_cores = (uint16_t)rte_lcore_count();
1051         uint16_t rx_count = 0;
1052         uint16_t tx_count;
1053         uint32_t retry = 0;
1054
1055         RTE_LOG(INFO, VHOST_DATA, "Procesing on Core %u started\n", lcore_id);
1056         prev_tsc = 0;
1057
1058         tx_q = &lcore_tx_queue[lcore_id];
1059         for (i = 0; i < num_cores; i ++) {
1060                 if (lcore_ids[i] == lcore_id) {
1061                         tx_q->txq_id = i;
1062                         break;
1063                 }
1064         }
1065
1066         while(1) {
1067                 cur_tsc = rte_rdtsc();
1068                 /*
1069                  * TX burst queue drain
1070                  */
1071                 diff_tsc = cur_tsc - prev_tsc;
1072                 if (unlikely(diff_tsc > drain_tsc)) {
1073
1074                         if (tx_q->len) {
1075                                 RTE_LOG(DEBUG, VHOST_DATA,
1076                                         "TX queue drained after timeout with burst size %u\n",
1077                                         tx_q->len);
1078
1079                                 /*Tx any packets in the queue*/
1080                                 ret = rte_eth_tx_burst(ports[0], (uint16_t)tx_q->txq_id,
1081                                                                            (struct rte_mbuf **)tx_q->m_table,
1082                                                                            (uint16_t)tx_q->len);
1083                                 if (unlikely(ret < tx_q->len)) {
1084                                         do {
1085                                                 rte_pktmbuf_free(tx_q->m_table[ret]);
1086                                         } while (++ret < tx_q->len);
1087                                 }
1088
1089                                 tx_q->len = 0;
1090                         }
1091
1092                         prev_tsc = cur_tsc;
1093
1094                 }
1095
1096                 /*
1097                  * Inform the configuration core that we have exited the
1098                  * linked list and that no devices are in use if requested.
1099                  */
1100                 if (lcore_info[lcore_id].dev_removal_flag == REQUEST_DEV_REMOVAL)
1101                         lcore_info[lcore_id].dev_removal_flag = ACK_DEV_REMOVAL;
1102
1103                 /*
1104                  * Process devices
1105                  */
1106                 TAILQ_FOREACH(vdev, &lcore_info[lcore_id].vdev_list, next) {
1107                         uint64_t fh;
1108
1109                         dev = vdev->dev;
1110                         fh  = dev->device_fh;
1111
1112                         if (unlikely(vdev->remove)) {
1113                                 unlink_vmdq(vdev);
1114                                 vdev->ready = DEVICE_SAFE_REMOVE;
1115                                 continue;
1116                         }
1117
1118                         if (likely(vdev->ready == DEVICE_RX)) {
1119                                 /*Handle guest RX*/
1120                                 rx_count = rte_eth_rx_burst(ports[0],
1121                                         vdev->vmdq_rx_q, pkts_burst, MAX_PKT_BURST);
1122
1123                                 if (rx_count) {
1124                                         /*
1125                                         * Retry is enabled and the queue is full then we wait and retry to avoid packet loss
1126                                         * Here MAX_PKT_BURST must be less than virtio queue size
1127                                         */
1128                                         if (enable_retry && unlikely(rx_count > rte_vring_available_entries(dev, VIRTIO_RXQ))) {
1129                                                 for (retry = 0; retry < burst_rx_retry_num; retry++) {
1130                                                         rte_delay_us(burst_rx_delay_time);
1131                                                         if (rx_count <= rte_vring_available_entries(dev, VIRTIO_RXQ))
1132                                                                 break;
1133                                                 }
1134                                         }
1135                                         ret_count = rte_vhost_enqueue_burst(dev, VIRTIO_RXQ, pkts_burst, rx_count);
1136                                         if (enable_stats) {
1137                                                 rte_atomic64_add(
1138                                                         &dev_statistics[fh].rx_total_atomic,
1139                                                         rx_count);
1140                                                 rte_atomic64_add(
1141                                                         &dev_statistics[fh].rx_atomic,
1142                                                         ret_count);
1143                                         }
1144                                         while (likely(rx_count)) {
1145                                                 rx_count--;
1146                                                 rte_pktmbuf_free(pkts_burst[rx_count]);
1147                                         }
1148
1149                                 }
1150                         }
1151
1152                         if (likely(!vdev->remove)) {
1153                                 /* Handle guest TX*/
1154                                 tx_count = rte_vhost_dequeue_burst(dev, VIRTIO_TXQ, mbuf_pool, pkts_burst, MAX_PKT_BURST);
1155                                 /* If this is the first received packet we need to learn the MAC and setup VMDQ */
1156                                 if (unlikely(vdev->ready == DEVICE_MAC_LEARNING) && tx_count) {
1157                                         if (vdev->remove || (link_vmdq(vdev, pkts_burst[0]) == -1)) {
1158                                                 while (tx_count)
1159                                                         rte_pktmbuf_free(pkts_burst[--tx_count]);
1160                                         }
1161                                 }
1162                                 for (i = 0; i < tx_count; ++i) {
1163                                         virtio_tx_route(vdev, pkts_burst[i],
1164                                                 vlan_tags[(uint16_t)dev->device_fh]);
1165                                 }
1166                         }
1167                 }
1168         }
1169
1170         return 0;
1171 }
1172
1173 /*
1174  * Remove a device from the specific data core linked list and from the
1175  * main linked list. Synchonization  occurs through the use of the
1176  * lcore dev_removal_flag. Device is made volatile here to avoid re-ordering
1177  * of dev->remove=1 which can cause an infinite loop in the rte_pause loop.
1178  */
1179 static void
1180 destroy_device (volatile struct virtio_net *dev)
1181 {
1182         struct vhost_dev *vdev;
1183         int lcore;
1184
1185         dev->flags &= ~VIRTIO_DEV_RUNNING;
1186
1187         vdev = (struct vhost_dev *)dev->priv;
1188         /*set the remove flag. */
1189         vdev->remove = 1;
1190         while(vdev->ready != DEVICE_SAFE_REMOVE) {
1191                 rte_pause();
1192         }
1193
1194         TAILQ_REMOVE(&lcore_info[vdev->coreid].vdev_list, vdev, next);
1195         TAILQ_REMOVE(&vhost_dev_list, vdev, next);
1196
1197         /* Set the dev_removal_flag on each lcore. */
1198         RTE_LCORE_FOREACH_SLAVE(lcore)
1199                 lcore_info[lcore].dev_removal_flag = REQUEST_DEV_REMOVAL;
1200
1201         /*
1202          * Once each core has set the dev_removal_flag to ACK_DEV_REMOVAL
1203          * we can be sure that they can no longer access the device removed
1204          * from the linked lists and that the devices are no longer in use.
1205          */
1206         RTE_LCORE_FOREACH_SLAVE(lcore) {
1207                 while (lcore_info[lcore].dev_removal_flag != ACK_DEV_REMOVAL)
1208                         rte_pause();
1209         }
1210
1211         lcore_info[vdev->coreid].device_num--;
1212
1213         RTE_LOG(INFO, VHOST_DATA,
1214                 "(%" PRIu64 ") Device has been removed from data core\n",
1215                 dev->device_fh);
1216
1217         rte_free(vdev);
1218 }
1219
1220 /*
1221  * A new device is added to a data core. First the device is added to the main linked list
1222  * and the allocated to a specific data core.
1223  */
1224 static int
1225 new_device (struct virtio_net *dev)
1226 {
1227         int lcore, core_add = 0;
1228         uint32_t device_num_min = num_devices;
1229         struct vhost_dev *vdev;
1230
1231         vdev = rte_zmalloc("vhost device", sizeof(*vdev), RTE_CACHE_LINE_SIZE);
1232         if (vdev == NULL) {
1233                 RTE_LOG(INFO, VHOST_DATA, "(%"PRIu64") Couldn't allocate memory for vhost dev\n",
1234                         dev->device_fh);
1235                 return -1;
1236         }
1237         vdev->dev = dev;
1238         dev->priv = vdev;
1239
1240         TAILQ_INSERT_TAIL(&vhost_dev_list, vdev, next);
1241         vdev->vmdq_rx_q
1242                 = dev->device_fh * queues_per_pool + vmdq_queue_base;
1243
1244         /*reset ready flag*/
1245         vdev->ready = DEVICE_MAC_LEARNING;
1246         vdev->remove = 0;
1247
1248         /* Find a suitable lcore to add the device. */
1249         RTE_LCORE_FOREACH_SLAVE(lcore) {
1250                 if (lcore_info[lcore].device_num < device_num_min) {
1251                         device_num_min = lcore_info[lcore].device_num;
1252                         core_add = lcore;
1253                 }
1254         }
1255         vdev->coreid = core_add;
1256
1257         TAILQ_INSERT_TAIL(&lcore_info[vdev->coreid].vdev_list, vdev, next);
1258         lcore_info[vdev->coreid].device_num++;
1259
1260         /* Initialize device stats */
1261         memset(&dev_statistics[dev->device_fh], 0, sizeof(struct device_statistics));
1262
1263         /* Disable notifications. */
1264         rte_vhost_enable_guest_notification(dev, VIRTIO_RXQ, 0);
1265         rte_vhost_enable_guest_notification(dev, VIRTIO_TXQ, 0);
1266         dev->flags |= VIRTIO_DEV_RUNNING;
1267
1268         RTE_LOG(INFO, VHOST_DATA, "(%"PRIu64") Device has been added to data core %d\n", dev->device_fh, vdev->coreid);
1269
1270         return 0;
1271 }
1272
1273 /*
1274  * These callback allow devices to be added to the data core when configuration
1275  * has been fully complete.
1276  */
1277 static const struct virtio_net_device_ops virtio_net_device_ops =
1278 {
1279         .new_device =  new_device,
1280         .destroy_device = destroy_device,
1281 };
1282
1283 /*
1284  * This is a thread will wake up after a period to print stats if the user has
1285  * enabled them.
1286  */
1287 static void
1288 print_stats(void)
1289 {
1290         struct vhost_dev *vdev;
1291         uint64_t tx_dropped, rx_dropped;
1292         uint64_t tx, tx_total, rx, rx_total;
1293         uint32_t device_fh;
1294         const char clr[] = { 27, '[', '2', 'J', '\0' };
1295         const char top_left[] = { 27, '[', '1', ';', '1', 'H','\0' };
1296
1297         while(1) {
1298                 sleep(enable_stats);
1299
1300                 /* Clear screen and move to top left */
1301                 printf("%s%s", clr, top_left);
1302
1303                 printf("\nDevice statistics ====================================");
1304
1305                 TAILQ_FOREACH(vdev, &vhost_dev_list, next) {
1306                         device_fh = vdev->dev->device_fh;
1307                         tx_total = dev_statistics[device_fh].tx_total;
1308                         tx = dev_statistics[device_fh].tx;
1309                         tx_dropped = tx_total - tx;
1310                         rx_total = rte_atomic64_read(
1311                                 &dev_statistics[device_fh].rx_total_atomic);
1312                         rx = rte_atomic64_read(
1313                                 &dev_statistics[device_fh].rx_atomic);
1314                         rx_dropped = rx_total - rx;
1315
1316                         printf("\nStatistics for device %"PRIu32" ------------------------------"
1317                                         "\nTX total:            %"PRIu64""
1318                                         "\nTX dropped:          %"PRIu64""
1319                                         "\nTX successful:               %"PRIu64""
1320                                         "\nRX total:            %"PRIu64""
1321                                         "\nRX dropped:          %"PRIu64""
1322                                         "\nRX successful:               %"PRIu64"",
1323                                         device_fh,
1324                                         tx_total,
1325                                         tx_dropped,
1326                                         tx,
1327                                         rx_total,
1328                                         rx_dropped,
1329                                         rx);
1330                 }
1331                 printf("\n======================================================\n");
1332         }
1333 }
1334
1335 /* When we receive a INT signal, unregister vhost driver */
1336 static void
1337 sigint_handler(__rte_unused int signum)
1338 {
1339         /* Unregister vhost driver. */
1340         int ret = rte_vhost_driver_unregister((char *)&dev_basename);
1341         if (ret != 0)
1342                 rte_exit(EXIT_FAILURE, "vhost driver unregister failure.\n");
1343         exit(0);
1344 }
1345
1346 /*
1347  * Main function, does initialisation and calls the per-lcore functions. The CUSE
1348  * device is also registered here to handle the IOCTLs.
1349  */
1350 int
1351 main(int argc, char *argv[])
1352 {
1353         unsigned lcore_id, core_id = 0;
1354         unsigned nb_ports, valid_num_ports;
1355         int ret;
1356         uint8_t portid;
1357         static pthread_t tid;
1358         char thread_name[RTE_MAX_THREAD_NAME_LEN];
1359
1360         signal(SIGINT, sigint_handler);
1361
1362         /* init EAL */
1363         ret = rte_eal_init(argc, argv);
1364         if (ret < 0)
1365                 rte_exit(EXIT_FAILURE, "Error with EAL initialization\n");
1366         argc -= ret;
1367         argv += ret;
1368
1369         /* parse app arguments */
1370         ret = us_vhost_parse_args(argc, argv);
1371         if (ret < 0)
1372                 rte_exit(EXIT_FAILURE, "Invalid argument\n");
1373
1374         for (lcore_id = 0; lcore_id < RTE_MAX_LCORE; lcore_id ++)
1375                 TAILQ_INIT(&lcore_info[lcore_id].vdev_list);
1376
1377                 if (rte_lcore_is_enabled(lcore_id))
1378                         lcore_ids[core_id ++] = lcore_id;
1379
1380         if (rte_lcore_count() > RTE_MAX_LCORE)
1381                 rte_exit(EXIT_FAILURE,"Not enough cores\n");
1382
1383         /*set the number of swithcing cores available*/
1384         num_switching_cores = rte_lcore_count()-1;
1385
1386         /* Get the number of physical ports. */
1387         nb_ports = rte_eth_dev_count();
1388         if (nb_ports > RTE_MAX_ETHPORTS)
1389                 nb_ports = RTE_MAX_ETHPORTS;
1390
1391         /*
1392          * Update the global var NUM_PORTS and global array PORTS
1393          * and get value of var VALID_NUM_PORTS according to system ports number
1394          */
1395         valid_num_ports = check_ports_num(nb_ports);
1396
1397         if ((valid_num_ports ==  0) || (valid_num_ports > MAX_SUP_PORTS)) {
1398                 RTE_LOG(INFO, VHOST_PORT, "Current enabled port number is %u,"
1399                         "but only %u port can be enabled\n",num_ports, MAX_SUP_PORTS);
1400                 return -1;
1401         }
1402
1403         /* Create the mbuf pool. */
1404         mbuf_pool = rte_pktmbuf_pool_create("MBUF_POOL",
1405                 NUM_MBUFS_PER_PORT * valid_num_ports, MBUF_CACHE_SIZE,
1406                 0, MBUF_DATA_SIZE, rte_socket_id());
1407         if (mbuf_pool == NULL)
1408                 rte_exit(EXIT_FAILURE, "Cannot create mbuf pool\n");
1409
1410         if (vm2vm_mode == VM2VM_HARDWARE) {
1411                 /* Enable VT loop back to let L2 switch to do it. */
1412                 vmdq_conf_default.rx_adv_conf.vmdq_rx_conf.enable_loop_back = 1;
1413                 RTE_LOG(DEBUG, VHOST_CONFIG,
1414                         "Enable loop back for L2 switch in vmdq.\n");
1415         }
1416
1417         /* initialize all ports */
1418         for (portid = 0; portid < nb_ports; portid++) {
1419                 /* skip ports that are not enabled */
1420                 if ((enabled_port_mask & (1 << portid)) == 0) {
1421                         RTE_LOG(INFO, VHOST_PORT,
1422                                 "Skipping disabled port %d\n", portid);
1423                         continue;
1424                 }
1425                 if (port_init(portid) != 0)
1426                         rte_exit(EXIT_FAILURE,
1427                                 "Cannot initialize network ports\n");
1428         }
1429
1430         /* Initialize device stats */
1431         memset(&dev_statistics, 0, sizeof(dev_statistics));
1432
1433         /* Enable stats if the user option is set. */
1434         if (enable_stats) {
1435                 ret = pthread_create(&tid, NULL, (void *)print_stats, NULL);
1436                 if (ret != 0)
1437                         rte_exit(EXIT_FAILURE,
1438                                 "Cannot create print-stats thread\n");
1439
1440                 /* Set thread_name for aid in debugging.  */
1441                 snprintf(thread_name, RTE_MAX_THREAD_NAME_LEN, "print-stats");
1442                 ret = rte_thread_setname(tid, thread_name);
1443                 if (ret != 0)
1444                         RTE_LOG(ERR, VHOST_CONFIG,
1445                                 "Cannot set print-stats name\n");
1446         }
1447
1448         /* Launch all data cores. */
1449         RTE_LCORE_FOREACH_SLAVE(lcore_id)
1450                 rte_eal_remote_launch(switch_worker, NULL, lcore_id);
1451
1452         if (mergeable == 0)
1453                 rte_vhost_feature_disable(1ULL << VIRTIO_NET_F_MRG_RXBUF);
1454
1455         /* Register vhost(cuse or user) driver to handle vhost messages. */
1456         ret = rte_vhost_driver_register((char *)&dev_basename);
1457         if (ret != 0)
1458                 rte_exit(EXIT_FAILURE, "vhost driver register failure.\n");
1459
1460         rte_vhost_driver_callback_register(&virtio_net_device_ops);
1461
1462         /* Start CUSE session. */
1463         rte_vhost_driver_session_start();
1464         return 0;
1465
1466 }