xref: /dpdk/examples/vhost/main.c (revision 67b6d3039e9edbc4624c878c6930be5e126e8b58)
1d19533e8SHuawei Xie /*-
2d19533e8SHuawei Xie  *   BSD LICENSE
3d19533e8SHuawei Xie  *
4d19533e8SHuawei Xie  *   Copyright(c) 2010-2014 Intel Corporation. All rights reserved.
5d19533e8SHuawei Xie  *   All rights reserved.
6d19533e8SHuawei Xie  *
7d19533e8SHuawei Xie  *   Redistribution and use in source and binary forms, with or without
8d19533e8SHuawei Xie  *   modification, are permitted provided that the following conditions
9d19533e8SHuawei Xie  *   are met:
10d19533e8SHuawei Xie  *
11d19533e8SHuawei Xie  *     * Redistributions of source code must retain the above copyright
12d19533e8SHuawei Xie  *       notice, this list of conditions and the following disclaimer.
13d19533e8SHuawei Xie  *     * Redistributions in binary form must reproduce the above copyright
14d19533e8SHuawei Xie  *       notice, this list of conditions and the following disclaimer in
15d19533e8SHuawei Xie  *       the documentation and/or other materials provided with the
16d19533e8SHuawei Xie  *       distribution.
17d19533e8SHuawei Xie  *     * Neither the name of Intel Corporation nor the names of its
18d19533e8SHuawei Xie  *       contributors may be used to endorse or promote products derived
19d19533e8SHuawei Xie  *       from this software without specific prior written permission.
20d19533e8SHuawei Xie  *
21d19533e8SHuawei Xie  *   THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
22d19533e8SHuawei Xie  *   "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
23d19533e8SHuawei Xie  *   LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
24d19533e8SHuawei Xie  *   A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
25d19533e8SHuawei Xie  *   OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
26d19533e8SHuawei Xie  *   SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
27d19533e8SHuawei Xie  *   LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
28d19533e8SHuawei Xie  *   DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
29d19533e8SHuawei Xie  *   THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
30d19533e8SHuawei Xie  *   (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
31d19533e8SHuawei Xie  *   OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
32d19533e8SHuawei Xie  */
33d19533e8SHuawei Xie 
34d19533e8SHuawei Xie #include <arpa/inet.h>
35d19533e8SHuawei Xie #include <getopt.h>
36d19533e8SHuawei Xie #include <linux/if_ether.h>
37d19533e8SHuawei Xie #include <linux/if_vlan.h>
38d19533e8SHuawei Xie #include <linux/virtio_net.h>
39d19533e8SHuawei Xie #include <linux/virtio_ring.h>
40d19533e8SHuawei Xie #include <signal.h>
41d19533e8SHuawei Xie #include <stdint.h>
42d19533e8SHuawei Xie #include <sys/eventfd.h>
43d19533e8SHuawei Xie #include <sys/param.h>
44d19533e8SHuawei Xie #include <unistd.h>
45d19533e8SHuawei Xie 
46d19533e8SHuawei Xie #include <rte_atomic.h>
47d19533e8SHuawei Xie #include <rte_cycles.h>
48d19533e8SHuawei Xie #include <rte_ethdev.h>
49d19533e8SHuawei Xie #include <rte_log.h>
50d19533e8SHuawei Xie #include <rte_string_fns.h>
51d19533e8SHuawei Xie #include <rte_malloc.h>
525cf27144SHuawei Xie #include <rte_virtio_net.h>
53d19533e8SHuawei Xie 
54d19533e8SHuawei Xie #include "main.h"
55d19533e8SHuawei Xie 
568bd6c395SHuawei Xie #define MAX_QUEUES 512
57d19533e8SHuawei Xie 
58d19533e8SHuawei Xie /* the maximum number of external ports supported */
59d19533e8SHuawei Xie #define MAX_SUP_PORTS 1
60d19533e8SHuawei Xie 
61d19533e8SHuawei Xie /*
62d19533e8SHuawei Xie  * Calculate the number of buffers needed per port
63d19533e8SHuawei Xie  */
64d19533e8SHuawei Xie #define NUM_MBUFS_PER_PORT ((MAX_QUEUES*RTE_TEST_RX_DESC_DEFAULT) +		\
65d19533e8SHuawei Xie 							(num_switching_cores*MAX_PKT_BURST) +  			\
66d19533e8SHuawei Xie 							(num_switching_cores*RTE_TEST_TX_DESC_DEFAULT) +\
67d19533e8SHuawei Xie 							(num_switching_cores*MBUF_CACHE_SIZE))
68d19533e8SHuawei Xie 
69d19533e8SHuawei Xie #define MBUF_CACHE_SIZE	128
70824cb29cSKonstantin Ananyev #define MBUF_DATA_SIZE	RTE_MBUF_DEFAULT_BUF_SIZE
71d19533e8SHuawei Xie 
72d19533e8SHuawei Xie /*
73d19533e8SHuawei Xie  * No frame data buffer allocated from host are required for zero copy
74d19533e8SHuawei Xie  * implementation, guest will allocate the frame data buffer, and vhost
75d19533e8SHuawei Xie  * directly use it.
76d19533e8SHuawei Xie  */
77824cb29cSKonstantin Ananyev #define VIRTIO_DESCRIPTOR_LEN_ZCP	RTE_MBUF_DEFAULT_DATAROOM
78824cb29cSKonstantin Ananyev #define MBUF_DATA_SIZE_ZCP		RTE_MBUF_DEFAULT_BUF_SIZE
79d19533e8SHuawei Xie #define MBUF_CACHE_SIZE_ZCP 0
80d19533e8SHuawei Xie 
81d19533e8SHuawei Xie #define MAX_PKT_BURST 32		/* Max burst size for RX/TX */
82d19533e8SHuawei Xie #define BURST_TX_DRAIN_US 100	/* TX drain every ~100us */
83d19533e8SHuawei Xie 
84d19533e8SHuawei Xie #define BURST_RX_WAIT_US 15	/* Defines how long we wait between retries on RX */
85d19533e8SHuawei Xie #define BURST_RX_RETRIES 4		/* Number of retries on RX. */
86d19533e8SHuawei Xie 
87d19533e8SHuawei Xie #define JUMBO_FRAME_MAX_SIZE    0x2600
88d19533e8SHuawei Xie 
89d19533e8SHuawei Xie /* State of virtio device. */
90d19533e8SHuawei Xie #define DEVICE_MAC_LEARNING 0
91d19533e8SHuawei Xie #define DEVICE_RX			1
92d19533e8SHuawei Xie #define DEVICE_SAFE_REMOVE	2
93d19533e8SHuawei Xie 
94d19533e8SHuawei Xie /* Config_core_flag status definitions. */
95d19533e8SHuawei Xie #define REQUEST_DEV_REMOVAL 1
96d19533e8SHuawei Xie #define ACK_DEV_REMOVAL 0
97d19533e8SHuawei Xie 
98d19533e8SHuawei Xie /* Configurable number of RX/TX ring descriptors */
99d19533e8SHuawei Xie #define RTE_TEST_RX_DESC_DEFAULT 1024
100d19533e8SHuawei Xie #define RTE_TEST_TX_DESC_DEFAULT 512
101d19533e8SHuawei Xie 
102d19533e8SHuawei Xie /*
103d19533e8SHuawei Xie  * Need refine these 2 macros for legacy and DPDK based front end:
104d19533e8SHuawei Xie  * Max vring avail descriptor/entries from guest - MAX_PKT_BURST
105d19533e8SHuawei Xie  * And then adjust power 2.
106d19533e8SHuawei Xie  */
107d19533e8SHuawei Xie /*
108d19533e8SHuawei Xie  * For legacy front end, 128 descriptors,
109d19533e8SHuawei Xie  * half for virtio header, another half for mbuf.
110d19533e8SHuawei Xie  */
111d19533e8SHuawei Xie #define RTE_TEST_RX_DESC_DEFAULT_ZCP 32   /* legacy: 32, DPDK virt FE: 128. */
112d19533e8SHuawei Xie #define RTE_TEST_TX_DESC_DEFAULT_ZCP 64   /* legacy: 64, DPDK virt FE: 64.  */
113d19533e8SHuawei Xie 
114d19533e8SHuawei Xie /* Get first 4 bytes in mbuf headroom. */
115d19533e8SHuawei Xie #define MBUF_HEADROOM_UINT32(mbuf) (*(uint32_t *)((uint8_t *)(mbuf) \
116d19533e8SHuawei Xie 		+ sizeof(struct rte_mbuf)))
117d19533e8SHuawei Xie 
118d19533e8SHuawei Xie /* true if x is a power of 2 */
119d19533e8SHuawei Xie #define POWEROF2(x) ((((x)-1) & (x)) == 0)
120d19533e8SHuawei Xie 
121d19533e8SHuawei Xie #define INVALID_PORT_ID 0xFF
122d19533e8SHuawei Xie 
123d19533e8SHuawei Xie /* Max number of devices. Limited by vmdq. */
124d19533e8SHuawei Xie #define MAX_DEVICES 64
125d19533e8SHuawei Xie 
126d19533e8SHuawei Xie /* Size of buffers used for snprintfs. */
127d19533e8SHuawei Xie #define MAX_PRINT_BUFF 6072
128d19533e8SHuawei Xie 
129d19533e8SHuawei Xie /* Maximum character device basename size. */
130d19533e8SHuawei Xie #define MAX_BASENAME_SZ 10
131d19533e8SHuawei Xie 
132d19533e8SHuawei Xie /* Maximum long option length for option parsing. */
133d19533e8SHuawei Xie #define MAX_LONG_OPT_SZ 64
134d19533e8SHuawei Xie 
135d19533e8SHuawei Xie /* Used to compare MAC addresses. */
136d19533e8SHuawei Xie #define MAC_ADDR_CMP 0xFFFFFFFFFFFFULL
137d19533e8SHuawei Xie 
138d19533e8SHuawei Xie /* Number of descriptors per cacheline. */
139fdf20fa7SSergio Gonzalez Monroy #define DESC_PER_CACHELINE (RTE_CACHE_LINE_SIZE / sizeof(struct vring_desc))
140d19533e8SHuawei Xie 
141355e6735SOlivier Matz #define MBUF_EXT_MEM(mb)   (rte_mbuf_from_indirect(mb) != (mb))
142e8b9ef87SSergio Gonzalez Monroy 
143d19533e8SHuawei Xie /* mask of enabled ports */
144d19533e8SHuawei Xie static uint32_t enabled_port_mask = 0;
145d19533e8SHuawei Xie 
14690924cafSOuyang Changchun /* Promiscuous mode */
14790924cafSOuyang Changchun static uint32_t promiscuous;
14890924cafSOuyang Changchun 
149d19533e8SHuawei Xie /*Number of switching cores enabled*/
150d19533e8SHuawei Xie static uint32_t num_switching_cores = 0;
151d19533e8SHuawei Xie 
152d19533e8SHuawei Xie /* number of devices/queues to support*/
153d19533e8SHuawei Xie static uint32_t num_queues = 0;
154a981294bSHuawei Xie static uint32_t num_devices;
155d19533e8SHuawei Xie 
156d19533e8SHuawei Xie /*
157d19533e8SHuawei Xie  * Enable zero copy, pkts buffer will directly dma to hw descriptor,
158d19533e8SHuawei Xie  * disabled on default.
159d19533e8SHuawei Xie  */
160d19533e8SHuawei Xie static uint32_t zero_copy;
16128deb020SHuawei Xie static int mergeable;
162d19533e8SHuawei Xie 
163e3d61d16SOuyang Changchun /* Do vlan strip on host, enabled on default */
164e3d61d16SOuyang Changchun static uint32_t vlan_strip = 1;
165e3d61d16SOuyang Changchun 
166d19533e8SHuawei Xie /* number of descriptors to apply*/
167d19533e8SHuawei Xie static uint32_t num_rx_descriptor = RTE_TEST_RX_DESC_DEFAULT_ZCP;
168d19533e8SHuawei Xie static uint32_t num_tx_descriptor = RTE_TEST_TX_DESC_DEFAULT_ZCP;
169d19533e8SHuawei Xie 
170d19533e8SHuawei Xie /* max ring descriptor, ixgbe, i40e, e1000 all are 4096. */
171d19533e8SHuawei Xie #define MAX_RING_DESC 4096
172d19533e8SHuawei Xie 
173d19533e8SHuawei Xie struct vpool {
174d19533e8SHuawei Xie 	struct rte_mempool *pool;
175d19533e8SHuawei Xie 	struct rte_ring *ring;
176d19533e8SHuawei Xie 	uint32_t buf_size;
177d19533e8SHuawei Xie } vpool_array[MAX_QUEUES+MAX_QUEUES];
178d19533e8SHuawei Xie 
179d19533e8SHuawei Xie /* Enable VM2VM communications. If this is disabled then the MAC address compare is skipped. */
180d19533e8SHuawei Xie typedef enum {
181d19533e8SHuawei Xie 	VM2VM_DISABLED = 0,
182d19533e8SHuawei Xie 	VM2VM_SOFTWARE = 1,
183d19533e8SHuawei Xie 	VM2VM_HARDWARE = 2,
184d19533e8SHuawei Xie 	VM2VM_LAST
185d19533e8SHuawei Xie } vm2vm_type;
186d19533e8SHuawei Xie static vm2vm_type vm2vm_mode = VM2VM_SOFTWARE;
187d19533e8SHuawei Xie 
188d19533e8SHuawei Xie /* The type of host physical address translated from guest physical address. */
189d19533e8SHuawei Xie typedef enum {
190d19533e8SHuawei Xie 	PHYS_ADDR_CONTINUOUS = 0,
191d19533e8SHuawei Xie 	PHYS_ADDR_CROSS_SUBREG = 1,
192d19533e8SHuawei Xie 	PHYS_ADDR_INVALID = 2,
193d19533e8SHuawei Xie 	PHYS_ADDR_LAST
194d19533e8SHuawei Xie } hpa_type;
195d19533e8SHuawei Xie 
196d19533e8SHuawei Xie /* Enable stats. */
197d19533e8SHuawei Xie static uint32_t enable_stats = 0;
198d19533e8SHuawei Xie /* Enable retries on RX. */
199d19533e8SHuawei Xie static uint32_t enable_retry = 1;
200d19533e8SHuawei Xie /* Specify timeout (in useconds) between retries on RX. */
201d19533e8SHuawei Xie static uint32_t burst_rx_delay_time = BURST_RX_WAIT_US;
202d19533e8SHuawei Xie /* Specify the number of retries on RX. */
203d19533e8SHuawei Xie static uint32_t burst_rx_retry_num = BURST_RX_RETRIES;
204d19533e8SHuawei Xie 
205d19533e8SHuawei Xie /* Character device basename. Can be set by user. */
206d19533e8SHuawei Xie static char dev_basename[MAX_BASENAME_SZ] = "vhost-net";
207d19533e8SHuawei Xie 
208d19533e8SHuawei Xie /* empty vmdq configuration structure. Filled in programatically */
209d19533e8SHuawei Xie static struct rte_eth_conf vmdq_conf_default = {
210d19533e8SHuawei Xie 	.rxmode = {
211d19533e8SHuawei Xie 		.mq_mode        = ETH_MQ_RX_VMDQ_ONLY,
212d19533e8SHuawei Xie 		.split_hdr_size = 0,
213d19533e8SHuawei Xie 		.header_split   = 0, /**< Header Split disabled */
214d19533e8SHuawei Xie 		.hw_ip_checksum = 0, /**< IP checksum offload disabled */
215d19533e8SHuawei Xie 		.hw_vlan_filter = 0, /**< VLAN filtering disabled */
216d19533e8SHuawei Xie 		/*
217d19533e8SHuawei Xie 		 * It is necessary for 1G NIC such as I350,
218d19533e8SHuawei Xie 		 * this fixes bug of ipv4 forwarding in guest can't
219d19533e8SHuawei Xie 		 * forward pakets from one virtio dev to another virtio dev.
220d19533e8SHuawei Xie 		 */
221d19533e8SHuawei Xie 		.hw_vlan_strip  = 1, /**< VLAN strip enabled. */
222d19533e8SHuawei Xie 		.jumbo_frame    = 0, /**< Jumbo Frame Support disabled */
223d19533e8SHuawei Xie 		.hw_strip_crc   = 0, /**< CRC stripped by hardware */
224d19533e8SHuawei Xie 	},
225d19533e8SHuawei Xie 
226d19533e8SHuawei Xie 	.txmode = {
227d19533e8SHuawei Xie 		.mq_mode = ETH_MQ_TX_NONE,
228d19533e8SHuawei Xie 	},
229d19533e8SHuawei Xie 	.rx_adv_conf = {
230d19533e8SHuawei Xie 		/*
231d19533e8SHuawei Xie 		 * should be overridden separately in code with
232d19533e8SHuawei Xie 		 * appropriate values
233d19533e8SHuawei Xie 		 */
234d19533e8SHuawei Xie 		.vmdq_rx_conf = {
235d19533e8SHuawei Xie 			.nb_queue_pools = ETH_8_POOLS,
236d19533e8SHuawei Xie 			.enable_default_pool = 0,
237d19533e8SHuawei Xie 			.default_pool = 0,
238d19533e8SHuawei Xie 			.nb_pool_maps = 0,
239d19533e8SHuawei Xie 			.pool_map = {{0, 0},},
240d19533e8SHuawei Xie 		},
241d19533e8SHuawei Xie 	},
242d19533e8SHuawei Xie };
243d19533e8SHuawei Xie 
244d19533e8SHuawei Xie static unsigned lcore_ids[RTE_MAX_LCORE];
245d19533e8SHuawei Xie static uint8_t ports[RTE_MAX_ETHPORTS];
246d19533e8SHuawei Xie static unsigned num_ports = 0; /**< The number of ports specified in command line */
24784b02d16SHuawei Xie static uint16_t num_pf_queues, num_vmdq_queues;
24884b02d16SHuawei Xie static uint16_t vmdq_pool_base, vmdq_queue_base;
24984b02d16SHuawei Xie static uint16_t queues_per_pool;
250d19533e8SHuawei Xie 
251d19533e8SHuawei Xie static const uint16_t external_pkt_default_vlan_tag = 2000;
252d19533e8SHuawei Xie const uint16_t vlan_tags[] = {
253d19533e8SHuawei Xie 	1000, 1001, 1002, 1003, 1004, 1005, 1006, 1007,
254d19533e8SHuawei Xie 	1008, 1009, 1010, 1011,	1012, 1013, 1014, 1015,
255d19533e8SHuawei Xie 	1016, 1017, 1018, 1019, 1020, 1021, 1022, 1023,
256d19533e8SHuawei Xie 	1024, 1025, 1026, 1027, 1028, 1029, 1030, 1031,
257d19533e8SHuawei Xie 	1032, 1033, 1034, 1035, 1036, 1037, 1038, 1039,
258d19533e8SHuawei Xie 	1040, 1041, 1042, 1043, 1044, 1045, 1046, 1047,
259d19533e8SHuawei Xie 	1048, 1049, 1050, 1051, 1052, 1053, 1054, 1055,
260d19533e8SHuawei Xie 	1056, 1057, 1058, 1059, 1060, 1061, 1062, 1063,
261d19533e8SHuawei Xie };
262d19533e8SHuawei Xie 
263d19533e8SHuawei Xie /* ethernet addresses of ports */
264d19533e8SHuawei Xie static struct ether_addr vmdq_ports_eth_addr[RTE_MAX_ETHPORTS];
265d19533e8SHuawei Xie 
266d19533e8SHuawei Xie /* heads for the main used and free linked lists for the data path. */
267d19533e8SHuawei Xie static struct virtio_net_data_ll *ll_root_used = NULL;
268d19533e8SHuawei Xie static struct virtio_net_data_ll *ll_root_free = NULL;
269d19533e8SHuawei Xie 
270d19533e8SHuawei Xie /* Array of data core structures containing information on individual core linked lists. */
271d19533e8SHuawei Xie static struct lcore_info lcore_info[RTE_MAX_LCORE];
272d19533e8SHuawei Xie 
273d19533e8SHuawei Xie /* Used for queueing bursts of TX packets. */
274d19533e8SHuawei Xie struct mbuf_table {
275d19533e8SHuawei Xie 	unsigned len;
276d19533e8SHuawei Xie 	unsigned txq_id;
277d19533e8SHuawei Xie 	struct rte_mbuf *m_table[MAX_PKT_BURST];
278d19533e8SHuawei Xie };
279d19533e8SHuawei Xie 
280d19533e8SHuawei Xie /* TX queue for each data core. */
281d19533e8SHuawei Xie struct mbuf_table lcore_tx_queue[RTE_MAX_LCORE];
282d19533e8SHuawei Xie 
283d19533e8SHuawei Xie /* TX queue fori each virtio device for zero copy. */
284d19533e8SHuawei Xie struct mbuf_table tx_queue_zcp[MAX_QUEUES];
285d19533e8SHuawei Xie 
286d19533e8SHuawei Xie /* Vlan header struct used to insert vlan tags on TX. */
287d19533e8SHuawei Xie struct vlan_ethhdr {
288d19533e8SHuawei Xie 	unsigned char   h_dest[ETH_ALEN];
289d19533e8SHuawei Xie 	unsigned char   h_source[ETH_ALEN];
290d19533e8SHuawei Xie 	__be16          h_vlan_proto;
291d19533e8SHuawei Xie 	__be16          h_vlan_TCI;
292d19533e8SHuawei Xie 	__be16          h_vlan_encapsulated_proto;
293d19533e8SHuawei Xie };
294d19533e8SHuawei Xie 
295d19533e8SHuawei Xie /* IPv4 Header */
296d19533e8SHuawei Xie struct ipv4_hdr {
297d19533e8SHuawei Xie 	uint8_t  version_ihl;		/**< version and header length */
298d19533e8SHuawei Xie 	uint8_t  type_of_service;	/**< type of service */
299d19533e8SHuawei Xie 	uint16_t total_length;		/**< length of packet */
300d19533e8SHuawei Xie 	uint16_t packet_id;		/**< packet ID */
301d19533e8SHuawei Xie 	uint16_t fragment_offset;	/**< fragmentation offset */
302d19533e8SHuawei Xie 	uint8_t  time_to_live;		/**< time to live */
303d19533e8SHuawei Xie 	uint8_t  next_proto_id;		/**< protocol ID */
304d19533e8SHuawei Xie 	uint16_t hdr_checksum;		/**< header checksum */
305d19533e8SHuawei Xie 	uint32_t src_addr;		/**< source address */
306d19533e8SHuawei Xie 	uint32_t dst_addr;		/**< destination address */
307d19533e8SHuawei Xie } __attribute__((__packed__));
308d19533e8SHuawei Xie 
309d19533e8SHuawei Xie /* Header lengths. */
310d19533e8SHuawei Xie #define VLAN_HLEN       4
311d19533e8SHuawei Xie #define VLAN_ETH_HLEN   18
312d19533e8SHuawei Xie 
313d19533e8SHuawei Xie /* Per-device statistics struct */
314d19533e8SHuawei Xie struct device_statistics {
315d19533e8SHuawei Xie 	uint64_t tx_total;
316d19533e8SHuawei Xie 	rte_atomic64_t rx_total_atomic;
317d19533e8SHuawei Xie 	uint64_t rx_total;
318d19533e8SHuawei Xie 	uint64_t tx;
319d19533e8SHuawei Xie 	rte_atomic64_t rx_atomic;
320d19533e8SHuawei Xie 	uint64_t rx;
321d19533e8SHuawei Xie } __rte_cache_aligned;
322d19533e8SHuawei Xie struct device_statistics dev_statistics[MAX_DEVICES];
323d19533e8SHuawei Xie 
324d19533e8SHuawei Xie /*
325d19533e8SHuawei Xie  * Builds up the correct configuration for VMDQ VLAN pool map
326d19533e8SHuawei Xie  * according to the pool & queue limits.
327d19533e8SHuawei Xie  */
328d19533e8SHuawei Xie static inline int
329d19533e8SHuawei Xie get_eth_conf(struct rte_eth_conf *eth_conf, uint32_t num_devices)
330d19533e8SHuawei Xie {
331d19533e8SHuawei Xie 	struct rte_eth_vmdq_rx_conf conf;
33290924cafSOuyang Changchun 	struct rte_eth_vmdq_rx_conf *def_conf =
33390924cafSOuyang Changchun 		&vmdq_conf_default.rx_adv_conf.vmdq_rx_conf;
334d19533e8SHuawei Xie 	unsigned i;
335d19533e8SHuawei Xie 
336d19533e8SHuawei Xie 	memset(&conf, 0, sizeof(conf));
337d19533e8SHuawei Xie 	conf.nb_queue_pools = (enum rte_eth_nb_pools)num_devices;
338d19533e8SHuawei Xie 	conf.nb_pool_maps = num_devices;
33990924cafSOuyang Changchun 	conf.enable_loop_back = def_conf->enable_loop_back;
34090924cafSOuyang Changchun 	conf.rx_mode = def_conf->rx_mode;
341d19533e8SHuawei Xie 
342d19533e8SHuawei Xie 	for (i = 0; i < conf.nb_pool_maps; i++) {
343d19533e8SHuawei Xie 		conf.pool_map[i].vlan_id = vlan_tags[ i ];
344d19533e8SHuawei Xie 		conf.pool_map[i].pools = (1UL << i);
345d19533e8SHuawei Xie 	}
346d19533e8SHuawei Xie 
347d19533e8SHuawei Xie 	(void)(rte_memcpy(eth_conf, &vmdq_conf_default, sizeof(*eth_conf)));
348d19533e8SHuawei Xie 	(void)(rte_memcpy(&eth_conf->rx_adv_conf.vmdq_rx_conf, &conf,
349d19533e8SHuawei Xie 		   sizeof(eth_conf->rx_adv_conf.vmdq_rx_conf)));
350d19533e8SHuawei Xie 	return 0;
351d19533e8SHuawei Xie }
352d19533e8SHuawei Xie 
353d19533e8SHuawei Xie /*
354d19533e8SHuawei Xie  * Validate the device number according to the max pool number gotten form
355d19533e8SHuawei Xie  * dev_info. If the device number is invalid, give the error message and
356d19533e8SHuawei Xie  * return -1. Each device must have its own pool.
357d19533e8SHuawei Xie  */
358d19533e8SHuawei Xie static inline int
359d19533e8SHuawei Xie validate_num_devices(uint32_t max_nb_devices)
360d19533e8SHuawei Xie {
361d19533e8SHuawei Xie 	if (num_devices > max_nb_devices) {
362d19533e8SHuawei Xie 		RTE_LOG(ERR, VHOST_PORT, "invalid number of devices\n");
363d19533e8SHuawei Xie 		return -1;
364d19533e8SHuawei Xie 	}
365d19533e8SHuawei Xie 	return 0;
366d19533e8SHuawei Xie }
367d19533e8SHuawei Xie 
368d19533e8SHuawei Xie /*
369d19533e8SHuawei Xie  * Initialises a given port using global settings and with the rx buffers
370d19533e8SHuawei Xie  * coming from the mbuf_pool passed as parameter
371d19533e8SHuawei Xie  */
372d19533e8SHuawei Xie static inline int
373d19533e8SHuawei Xie port_init(uint8_t port)
374d19533e8SHuawei Xie {
375d19533e8SHuawei Xie 	struct rte_eth_dev_info dev_info;
376d19533e8SHuawei Xie 	struct rte_eth_conf port_conf;
377db4014f2SHuawei Xie 	struct rte_eth_rxconf *rxconf;
378db4014f2SHuawei Xie 	struct rte_eth_txconf *txconf;
379db4014f2SHuawei Xie 	int16_t rx_rings, tx_rings;
380d19533e8SHuawei Xie 	uint16_t rx_ring_size, tx_ring_size;
381d19533e8SHuawei Xie 	int retval;
382d19533e8SHuawei Xie 	uint16_t q;
383d19533e8SHuawei Xie 
384d19533e8SHuawei Xie 	/* The max pool number from dev_info will be used to validate the pool number specified in cmd line */
385d19533e8SHuawei Xie 	rte_eth_dev_info_get (port, &dev_info);
386d19533e8SHuawei Xie 
3878bd6c395SHuawei Xie 	if (dev_info.max_rx_queues > MAX_QUEUES) {
3888bd6c395SHuawei Xie 		rte_exit(EXIT_FAILURE,
3898bd6c395SHuawei Xie 			"please define MAX_QUEUES no less than %u in %s\n",
3908bd6c395SHuawei Xie 			dev_info.max_rx_queues, __FILE__);
3918bd6c395SHuawei Xie 	}
3928bd6c395SHuawei Xie 
393db4014f2SHuawei Xie 	rxconf = &dev_info.default_rxconf;
394db4014f2SHuawei Xie 	txconf = &dev_info.default_txconf;
395db4014f2SHuawei Xie 	rxconf->rx_drop_en = 1;
396db4014f2SHuawei Xie 
397f0adccd4SOuyang Changchun 	/* Enable vlan offload */
398f0adccd4SOuyang Changchun 	txconf->txq_flags &= ~ETH_TXQ_FLAGS_NOVLANOFFL;
399f0adccd4SOuyang Changchun 
400db4014f2SHuawei Xie 	/*
401db4014f2SHuawei Xie 	 * Zero copy defers queue RX/TX start to the time when guest
402db4014f2SHuawei Xie 	 * finishes its startup and packet buffers from that guest are
403db4014f2SHuawei Xie 	 * available.
404db4014f2SHuawei Xie 	 */
405db4014f2SHuawei Xie 	if (zero_copy) {
406db4014f2SHuawei Xie 		rxconf->rx_deferred_start = 1;
407db4014f2SHuawei Xie 		rxconf->rx_drop_en = 0;
408db4014f2SHuawei Xie 		txconf->tx_deferred_start = 1;
409db4014f2SHuawei Xie 	}
410db4014f2SHuawei Xie 
411d19533e8SHuawei Xie 	/*configure the number of supported virtio devices based on VMDQ limits */
412d19533e8SHuawei Xie 	num_devices = dev_info.max_vmdq_pools;
413d19533e8SHuawei Xie 
414d19533e8SHuawei Xie 	if (zero_copy) {
415d19533e8SHuawei Xie 		rx_ring_size = num_rx_descriptor;
416d19533e8SHuawei Xie 		tx_ring_size = num_tx_descriptor;
417d19533e8SHuawei Xie 		tx_rings = dev_info.max_tx_queues;
418d19533e8SHuawei Xie 	} else {
419d19533e8SHuawei Xie 		rx_ring_size = RTE_TEST_RX_DESC_DEFAULT;
420d19533e8SHuawei Xie 		tx_ring_size = RTE_TEST_TX_DESC_DEFAULT;
421d19533e8SHuawei Xie 		tx_rings = (uint16_t)rte_lcore_count();
422d19533e8SHuawei Xie 	}
423d19533e8SHuawei Xie 
424d19533e8SHuawei Xie 	retval = validate_num_devices(MAX_DEVICES);
425d19533e8SHuawei Xie 	if (retval < 0)
426d19533e8SHuawei Xie 		return retval;
427d19533e8SHuawei Xie 
428d19533e8SHuawei Xie 	/* Get port configuration. */
429d19533e8SHuawei Xie 	retval = get_eth_conf(&port_conf, num_devices);
430d19533e8SHuawei Xie 	if (retval < 0)
431d19533e8SHuawei Xie 		return retval;
43284b02d16SHuawei Xie 	/* NIC queues are divided into pf queues and vmdq queues.  */
43384b02d16SHuawei Xie 	num_pf_queues = dev_info.max_rx_queues - dev_info.vmdq_queue_num;
43484b02d16SHuawei Xie 	queues_per_pool = dev_info.vmdq_queue_num / dev_info.max_vmdq_pools;
43584b02d16SHuawei Xie 	num_vmdq_queues = num_devices * queues_per_pool;
43684b02d16SHuawei Xie 	num_queues = num_pf_queues + num_vmdq_queues;
43784b02d16SHuawei Xie 	vmdq_queue_base = dev_info.vmdq_queue_base;
43884b02d16SHuawei Xie 	vmdq_pool_base  = dev_info.vmdq_pool_base;
43984b02d16SHuawei Xie 	printf("pf queue num: %u, configured vmdq pool num: %u, each vmdq pool has %u queues\n",
44084b02d16SHuawei Xie 		num_pf_queues, num_devices, queues_per_pool);
441d19533e8SHuawei Xie 
442d19533e8SHuawei Xie 	if (port >= rte_eth_dev_count()) return -1;
443d19533e8SHuawei Xie 
44484b02d16SHuawei Xie 	rx_rings = (uint16_t)dev_info.max_rx_queues;
445d19533e8SHuawei Xie 	/* Configure ethernet device. */
446d19533e8SHuawei Xie 	retval = rte_eth_dev_configure(port, rx_rings, tx_rings, &port_conf);
447d19533e8SHuawei Xie 	if (retval != 0)
448d19533e8SHuawei Xie 		return retval;
449d19533e8SHuawei Xie 
450d19533e8SHuawei Xie 	/* Setup the queues. */
451d19533e8SHuawei Xie 	for (q = 0; q < rx_rings; q ++) {
452d19533e8SHuawei Xie 		retval = rte_eth_rx_queue_setup(port, q, rx_ring_size,
453db4014f2SHuawei Xie 						rte_eth_dev_socket_id(port),
454db4014f2SHuawei Xie 						rxconf,
455d19533e8SHuawei Xie 						vpool_array[q].pool);
456d19533e8SHuawei Xie 		if (retval < 0)
457d19533e8SHuawei Xie 			return retval;
458d19533e8SHuawei Xie 	}
459d19533e8SHuawei Xie 	for (q = 0; q < tx_rings; q ++) {
460d19533e8SHuawei Xie 		retval = rte_eth_tx_queue_setup(port, q, tx_ring_size,
461db4014f2SHuawei Xie 						rte_eth_dev_socket_id(port),
462db4014f2SHuawei Xie 						txconf);
463d19533e8SHuawei Xie 		if (retval < 0)
464d19533e8SHuawei Xie 			return retval;
465d19533e8SHuawei Xie 	}
466d19533e8SHuawei Xie 
467d19533e8SHuawei Xie 	/* Start the device. */
468d19533e8SHuawei Xie 	retval  = rte_eth_dev_start(port);
469d19533e8SHuawei Xie 	if (retval < 0) {
470d19533e8SHuawei Xie 		RTE_LOG(ERR, VHOST_DATA, "Failed to start the device.\n");
471d19533e8SHuawei Xie 		return retval;
472d19533e8SHuawei Xie 	}
473d19533e8SHuawei Xie 
47490924cafSOuyang Changchun 	if (promiscuous)
47590924cafSOuyang Changchun 		rte_eth_promiscuous_enable(port);
47690924cafSOuyang Changchun 
477d19533e8SHuawei Xie 	rte_eth_macaddr_get(port, &vmdq_ports_eth_addr[port]);
478d19533e8SHuawei Xie 	RTE_LOG(INFO, VHOST_PORT, "Max virtio devices supported: %u\n", num_devices);
479d19533e8SHuawei Xie 	RTE_LOG(INFO, VHOST_PORT, "Port %u MAC: %02"PRIx8" %02"PRIx8" %02"PRIx8
480d19533e8SHuawei Xie 			" %02"PRIx8" %02"PRIx8" %02"PRIx8"\n",
481d19533e8SHuawei Xie 			(unsigned)port,
482d19533e8SHuawei Xie 			vmdq_ports_eth_addr[port].addr_bytes[0],
483d19533e8SHuawei Xie 			vmdq_ports_eth_addr[port].addr_bytes[1],
484d19533e8SHuawei Xie 			vmdq_ports_eth_addr[port].addr_bytes[2],
485d19533e8SHuawei Xie 			vmdq_ports_eth_addr[port].addr_bytes[3],
486d19533e8SHuawei Xie 			vmdq_ports_eth_addr[port].addr_bytes[4],
487d19533e8SHuawei Xie 			vmdq_ports_eth_addr[port].addr_bytes[5]);
488d19533e8SHuawei Xie 
489d19533e8SHuawei Xie 	return 0;
490d19533e8SHuawei Xie }
491d19533e8SHuawei Xie 
492d19533e8SHuawei Xie /*
493d19533e8SHuawei Xie  * Set character device basename.
494d19533e8SHuawei Xie  */
495d19533e8SHuawei Xie static int
496d19533e8SHuawei Xie us_vhost_parse_basename(const char *q_arg)
497d19533e8SHuawei Xie {
498d19533e8SHuawei Xie 	/* parse number string */
499d19533e8SHuawei Xie 
500d19533e8SHuawei Xie 	if (strnlen(q_arg, MAX_BASENAME_SZ) > MAX_BASENAME_SZ)
501d19533e8SHuawei Xie 		return -1;
502d19533e8SHuawei Xie 	else
503d19533e8SHuawei Xie 		snprintf((char*)&dev_basename, MAX_BASENAME_SZ, "%s", q_arg);
504d19533e8SHuawei Xie 
505d19533e8SHuawei Xie 	return 0;
506d19533e8SHuawei Xie }
507d19533e8SHuawei Xie 
508d19533e8SHuawei Xie /*
509d19533e8SHuawei Xie  * Parse the portmask provided at run time.
510d19533e8SHuawei Xie  */
511d19533e8SHuawei Xie static int
512d19533e8SHuawei Xie parse_portmask(const char *portmask)
513d19533e8SHuawei Xie {
514d19533e8SHuawei Xie 	char *end = NULL;
515d19533e8SHuawei Xie 	unsigned long pm;
516d19533e8SHuawei Xie 
517d19533e8SHuawei Xie 	errno = 0;
518d19533e8SHuawei Xie 
519d19533e8SHuawei Xie 	/* parse hexadecimal string */
520d19533e8SHuawei Xie 	pm = strtoul(portmask, &end, 16);
521d19533e8SHuawei Xie 	if ((portmask[0] == '\0') || (end == NULL) || (*end != '\0') || (errno != 0))
522d19533e8SHuawei Xie 		return -1;
523d19533e8SHuawei Xie 
524d19533e8SHuawei Xie 	if (pm == 0)
525d19533e8SHuawei Xie 		return -1;
526d19533e8SHuawei Xie 
527d19533e8SHuawei Xie 	return pm;
528d19533e8SHuawei Xie 
529d19533e8SHuawei Xie }
530d19533e8SHuawei Xie 
531d19533e8SHuawei Xie /*
532d19533e8SHuawei Xie  * Parse num options at run time.
533d19533e8SHuawei Xie  */
534d19533e8SHuawei Xie static int
535d19533e8SHuawei Xie parse_num_opt(const char *q_arg, uint32_t max_valid_value)
536d19533e8SHuawei Xie {
537d19533e8SHuawei Xie 	char *end = NULL;
538d19533e8SHuawei Xie 	unsigned long num;
539d19533e8SHuawei Xie 
540d19533e8SHuawei Xie 	errno = 0;
541d19533e8SHuawei Xie 
542d19533e8SHuawei Xie 	/* parse unsigned int string */
543d19533e8SHuawei Xie 	num = strtoul(q_arg, &end, 10);
544d19533e8SHuawei Xie 	if ((q_arg[0] == '\0') || (end == NULL) || (*end != '\0') || (errno != 0))
545d19533e8SHuawei Xie 		return -1;
546d19533e8SHuawei Xie 
547d19533e8SHuawei Xie 	if (num > max_valid_value)
548d19533e8SHuawei Xie 		return -1;
549d19533e8SHuawei Xie 
550d19533e8SHuawei Xie 	return num;
551d19533e8SHuawei Xie 
552d19533e8SHuawei Xie }
553d19533e8SHuawei Xie 
554d19533e8SHuawei Xie /*
555d19533e8SHuawei Xie  * Display usage
556d19533e8SHuawei Xie  */
557d19533e8SHuawei Xie static void
558d19533e8SHuawei Xie us_vhost_usage(const char *prgname)
559d19533e8SHuawei Xie {
560d19533e8SHuawei Xie 	RTE_LOG(INFO, VHOST_CONFIG, "%s [EAL options] -- -p PORTMASK\n"
561d19533e8SHuawei Xie 	"		--vm2vm [0|1|2]\n"
562d19533e8SHuawei Xie 	"		--rx_retry [0|1] --mergeable [0|1] --stats [0-N]\n"
5635cf27144SHuawei Xie 	"		--dev-basename <name>\n"
564d19533e8SHuawei Xie 	"		--nb-devices ND\n"
565d19533e8SHuawei Xie 	"		-p PORTMASK: Set mask for ports to be used by application\n"
566d19533e8SHuawei Xie 	"		--vm2vm [0|1|2]: disable/software(default)/hardware vm2vm comms\n"
567d19533e8SHuawei Xie 	"		--rx-retry [0|1]: disable/enable(default) retries on rx. Enable retry if destintation queue is full\n"
568d19533e8SHuawei Xie 	"		--rx-retry-delay [0-N]: timeout(in usecond) between retries on RX. This makes effect only if retries on rx enabled\n"
569d19533e8SHuawei Xie 	"		--rx-retry-num [0-N]: the number of retries on rx. This makes effect only if retries on rx enabled\n"
570d19533e8SHuawei Xie 	"		--mergeable [0|1]: disable(default)/enable RX mergeable buffers\n"
571e3d61d16SOuyang Changchun 	"		--vlan-strip [0|1]: disable/enable(default) RX VLAN strip on host\n"
572d19533e8SHuawei Xie 	"		--stats [0-N]: 0: Disable stats, N: Time in seconds to print stats\n"
573d19533e8SHuawei Xie 	"		--dev-basename: The basename to be used for the character device.\n"
574d19533e8SHuawei Xie 	"		--zero-copy [0|1]: disable(default)/enable rx/tx "
575d19533e8SHuawei Xie 			"zero copy\n"
576d19533e8SHuawei Xie 	"		--rx-desc-num [0-N]: the number of descriptors on rx, "
577d19533e8SHuawei Xie 			"used only when zero copy is enabled.\n"
578d19533e8SHuawei Xie 	"		--tx-desc-num [0-N]: the number of descriptors on tx, "
579d19533e8SHuawei Xie 			"used only when zero copy is enabled.\n",
580d19533e8SHuawei Xie 	       prgname);
581d19533e8SHuawei Xie }
582d19533e8SHuawei Xie 
583d19533e8SHuawei Xie /*
584d19533e8SHuawei Xie  * Parse the arguments given in the command line of the application.
585d19533e8SHuawei Xie  */
586d19533e8SHuawei Xie static int
587d19533e8SHuawei Xie us_vhost_parse_args(int argc, char **argv)
588d19533e8SHuawei Xie {
589d19533e8SHuawei Xie 	int opt, ret;
590d19533e8SHuawei Xie 	int option_index;
591d19533e8SHuawei Xie 	unsigned i;
592d19533e8SHuawei Xie 	const char *prgname = argv[0];
593d19533e8SHuawei Xie 	static struct option long_option[] = {
594d19533e8SHuawei Xie 		{"vm2vm", required_argument, NULL, 0},
595d19533e8SHuawei Xie 		{"rx-retry", required_argument, NULL, 0},
596d19533e8SHuawei Xie 		{"rx-retry-delay", required_argument, NULL, 0},
597d19533e8SHuawei Xie 		{"rx-retry-num", required_argument, NULL, 0},
598d19533e8SHuawei Xie 		{"mergeable", required_argument, NULL, 0},
599e3d61d16SOuyang Changchun 		{"vlan-strip", required_argument, NULL, 0},
600d19533e8SHuawei Xie 		{"stats", required_argument, NULL, 0},
601d19533e8SHuawei Xie 		{"dev-basename", required_argument, NULL, 0},
602d19533e8SHuawei Xie 		{"zero-copy", required_argument, NULL, 0},
603d19533e8SHuawei Xie 		{"rx-desc-num", required_argument, NULL, 0},
604d19533e8SHuawei Xie 		{"tx-desc-num", required_argument, NULL, 0},
605d19533e8SHuawei Xie 		{NULL, 0, 0, 0},
606d19533e8SHuawei Xie 	};
607d19533e8SHuawei Xie 
608d19533e8SHuawei Xie 	/* Parse command line */
60990924cafSOuyang Changchun 	while ((opt = getopt_long(argc, argv, "p:P",
61090924cafSOuyang Changchun 			long_option, &option_index)) != EOF) {
611d19533e8SHuawei Xie 		switch (opt) {
612d19533e8SHuawei Xie 		/* Portmask */
613d19533e8SHuawei Xie 		case 'p':
614d19533e8SHuawei Xie 			enabled_port_mask = parse_portmask(optarg);
615d19533e8SHuawei Xie 			if (enabled_port_mask == 0) {
616d19533e8SHuawei Xie 				RTE_LOG(INFO, VHOST_CONFIG, "Invalid portmask\n");
617d19533e8SHuawei Xie 				us_vhost_usage(prgname);
618d19533e8SHuawei Xie 				return -1;
619d19533e8SHuawei Xie 			}
620d19533e8SHuawei Xie 			break;
621d19533e8SHuawei Xie 
62290924cafSOuyang Changchun 		case 'P':
62390924cafSOuyang Changchun 			promiscuous = 1;
62490924cafSOuyang Changchun 			vmdq_conf_default.rx_adv_conf.vmdq_rx_conf.rx_mode =
62590924cafSOuyang Changchun 				ETH_VMDQ_ACCEPT_BROADCAST |
62690924cafSOuyang Changchun 				ETH_VMDQ_ACCEPT_MULTICAST;
62790924cafSOuyang Changchun 			rte_vhost_feature_enable(1ULL << VIRTIO_NET_F_CTRL_RX);
62890924cafSOuyang Changchun 
62990924cafSOuyang Changchun 			break;
63090924cafSOuyang Changchun 
631d19533e8SHuawei Xie 		case 0:
632d19533e8SHuawei Xie 			/* Enable/disable vm2vm comms. */
633d19533e8SHuawei Xie 			if (!strncmp(long_option[option_index].name, "vm2vm",
634d19533e8SHuawei Xie 				MAX_LONG_OPT_SZ)) {
635d19533e8SHuawei Xie 				ret = parse_num_opt(optarg, (VM2VM_LAST - 1));
636d19533e8SHuawei Xie 				if (ret == -1) {
637d19533e8SHuawei Xie 					RTE_LOG(INFO, VHOST_CONFIG,
638d19533e8SHuawei Xie 						"Invalid argument for "
639d19533e8SHuawei Xie 						"vm2vm [0|1|2]\n");
640d19533e8SHuawei Xie 					us_vhost_usage(prgname);
641d19533e8SHuawei Xie 					return -1;
642d19533e8SHuawei Xie 				} else {
643d19533e8SHuawei Xie 					vm2vm_mode = (vm2vm_type)ret;
644d19533e8SHuawei Xie 				}
645d19533e8SHuawei Xie 			}
646d19533e8SHuawei Xie 
647d19533e8SHuawei Xie 			/* Enable/disable retries on RX. */
648d19533e8SHuawei Xie 			if (!strncmp(long_option[option_index].name, "rx-retry", MAX_LONG_OPT_SZ)) {
649d19533e8SHuawei Xie 				ret = parse_num_opt(optarg, 1);
650d19533e8SHuawei Xie 				if (ret == -1) {
651d19533e8SHuawei Xie 					RTE_LOG(INFO, VHOST_CONFIG, "Invalid argument for rx-retry [0|1]\n");
652d19533e8SHuawei Xie 					us_vhost_usage(prgname);
653d19533e8SHuawei Xie 					return -1;
654d19533e8SHuawei Xie 				} else {
655d19533e8SHuawei Xie 					enable_retry = ret;
656d19533e8SHuawei Xie 				}
657d19533e8SHuawei Xie 			}
658d19533e8SHuawei Xie 
659d19533e8SHuawei Xie 			/* Specify the retries delay time (in useconds) on RX. */
660d19533e8SHuawei Xie 			if (!strncmp(long_option[option_index].name, "rx-retry-delay", MAX_LONG_OPT_SZ)) {
661d19533e8SHuawei Xie 				ret = parse_num_opt(optarg, INT32_MAX);
662d19533e8SHuawei Xie 				if (ret == -1) {
663d19533e8SHuawei Xie 					RTE_LOG(INFO, VHOST_CONFIG, "Invalid argument for rx-retry-delay [0-N]\n");
664d19533e8SHuawei Xie 					us_vhost_usage(prgname);
665d19533e8SHuawei Xie 					return -1;
666d19533e8SHuawei Xie 				} else {
667d19533e8SHuawei Xie 					burst_rx_delay_time = ret;
668d19533e8SHuawei Xie 				}
669d19533e8SHuawei Xie 			}
670d19533e8SHuawei Xie 
671d19533e8SHuawei Xie 			/* Specify the retries number on RX. */
672d19533e8SHuawei Xie 			if (!strncmp(long_option[option_index].name, "rx-retry-num", MAX_LONG_OPT_SZ)) {
673d19533e8SHuawei Xie 				ret = parse_num_opt(optarg, INT32_MAX);
674d19533e8SHuawei Xie 				if (ret == -1) {
675d19533e8SHuawei Xie 					RTE_LOG(INFO, VHOST_CONFIG, "Invalid argument for rx-retry-num [0-N]\n");
676d19533e8SHuawei Xie 					us_vhost_usage(prgname);
677d19533e8SHuawei Xie 					return -1;
678d19533e8SHuawei Xie 				} else {
679d19533e8SHuawei Xie 					burst_rx_retry_num = ret;
680d19533e8SHuawei Xie 				}
681d19533e8SHuawei Xie 			}
682d19533e8SHuawei Xie 
683d19533e8SHuawei Xie 			/* Enable/disable RX mergeable buffers. */
684d19533e8SHuawei Xie 			if (!strncmp(long_option[option_index].name, "mergeable", MAX_LONG_OPT_SZ)) {
685d19533e8SHuawei Xie 				ret = parse_num_opt(optarg, 1);
686d19533e8SHuawei Xie 				if (ret == -1) {
687d19533e8SHuawei Xie 					RTE_LOG(INFO, VHOST_CONFIG, "Invalid argument for mergeable [0|1]\n");
688d19533e8SHuawei Xie 					us_vhost_usage(prgname);
689d19533e8SHuawei Xie 					return -1;
690d19533e8SHuawei Xie 				} else {
69128deb020SHuawei Xie 					mergeable = !!ret;
692d19533e8SHuawei Xie 					if (ret) {
693d19533e8SHuawei Xie 						vmdq_conf_default.rxmode.jumbo_frame = 1;
694d19533e8SHuawei Xie 						vmdq_conf_default.rxmode.max_rx_pkt_len
695d19533e8SHuawei Xie 							= JUMBO_FRAME_MAX_SIZE;
696d19533e8SHuawei Xie 					}
697d19533e8SHuawei Xie 				}
698d19533e8SHuawei Xie 			}
699d19533e8SHuawei Xie 
700e3d61d16SOuyang Changchun 			/* Enable/disable RX VLAN strip on host. */
701e3d61d16SOuyang Changchun 			if (!strncmp(long_option[option_index].name,
702e3d61d16SOuyang Changchun 				"vlan-strip", MAX_LONG_OPT_SZ)) {
703e3d61d16SOuyang Changchun 				ret = parse_num_opt(optarg, 1);
704e3d61d16SOuyang Changchun 				if (ret == -1) {
705e3d61d16SOuyang Changchun 					RTE_LOG(INFO, VHOST_CONFIG,
706e3d61d16SOuyang Changchun 						"Invalid argument for VLAN strip [0|1]\n");
707e3d61d16SOuyang Changchun 					us_vhost_usage(prgname);
708e3d61d16SOuyang Changchun 					return -1;
709e3d61d16SOuyang Changchun 				} else {
710e3d61d16SOuyang Changchun 					vlan_strip = !!ret;
711e3d61d16SOuyang Changchun 					vmdq_conf_default.rxmode.hw_vlan_strip =
712e3d61d16SOuyang Changchun 						vlan_strip;
713e3d61d16SOuyang Changchun 				}
714e3d61d16SOuyang Changchun 			}
715e3d61d16SOuyang Changchun 
716d19533e8SHuawei Xie 			/* Enable/disable stats. */
717d19533e8SHuawei Xie 			if (!strncmp(long_option[option_index].name, "stats", MAX_LONG_OPT_SZ)) {
718d19533e8SHuawei Xie 				ret = parse_num_opt(optarg, INT32_MAX);
719d19533e8SHuawei Xie 				if (ret == -1) {
720d19533e8SHuawei Xie 					RTE_LOG(INFO, VHOST_CONFIG, "Invalid argument for stats [0..N]\n");
721d19533e8SHuawei Xie 					us_vhost_usage(prgname);
722d19533e8SHuawei Xie 					return -1;
723d19533e8SHuawei Xie 				} else {
724d19533e8SHuawei Xie 					enable_stats = ret;
725d19533e8SHuawei Xie 				}
726d19533e8SHuawei Xie 			}
727d19533e8SHuawei Xie 
728d19533e8SHuawei Xie 			/* Set character device basename. */
729d19533e8SHuawei Xie 			if (!strncmp(long_option[option_index].name, "dev-basename", MAX_LONG_OPT_SZ)) {
730d19533e8SHuawei Xie 				if (us_vhost_parse_basename(optarg) == -1) {
731d19533e8SHuawei Xie 					RTE_LOG(INFO, VHOST_CONFIG, "Invalid argument for character device basename (Max %d characters)\n", MAX_BASENAME_SZ);
732d19533e8SHuawei Xie 					us_vhost_usage(prgname);
733d19533e8SHuawei Xie 					return -1;
734d19533e8SHuawei Xie 				}
735d19533e8SHuawei Xie 			}
736d19533e8SHuawei Xie 
737d19533e8SHuawei Xie 			/* Enable/disable rx/tx zero copy. */
738d19533e8SHuawei Xie 			if (!strncmp(long_option[option_index].name,
739d19533e8SHuawei Xie 				"zero-copy", MAX_LONG_OPT_SZ)) {
740d19533e8SHuawei Xie 				ret = parse_num_opt(optarg, 1);
741d19533e8SHuawei Xie 				if (ret == -1) {
742d19533e8SHuawei Xie 					RTE_LOG(INFO, VHOST_CONFIG,
743d19533e8SHuawei Xie 						"Invalid argument"
744d19533e8SHuawei Xie 						" for zero-copy [0|1]\n");
745d19533e8SHuawei Xie 					us_vhost_usage(prgname);
746d19533e8SHuawei Xie 					return -1;
747d19533e8SHuawei Xie 				} else
748d19533e8SHuawei Xie 					zero_copy = ret;
749d19533e8SHuawei Xie 			}
750d19533e8SHuawei Xie 
751d19533e8SHuawei Xie 			/* Specify the descriptor number on RX. */
752d19533e8SHuawei Xie 			if (!strncmp(long_option[option_index].name,
753d19533e8SHuawei Xie 				"rx-desc-num", MAX_LONG_OPT_SZ)) {
754d19533e8SHuawei Xie 				ret = parse_num_opt(optarg, MAX_RING_DESC);
755d19533e8SHuawei Xie 				if ((ret == -1) || (!POWEROF2(ret))) {
756d19533e8SHuawei Xie 					RTE_LOG(INFO, VHOST_CONFIG,
757d19533e8SHuawei Xie 					"Invalid argument for rx-desc-num[0-N],"
758d19533e8SHuawei Xie 					"power of 2 required.\n");
759d19533e8SHuawei Xie 					us_vhost_usage(prgname);
760d19533e8SHuawei Xie 					return -1;
761d19533e8SHuawei Xie 				} else {
762d19533e8SHuawei Xie 					num_rx_descriptor = ret;
763d19533e8SHuawei Xie 				}
764d19533e8SHuawei Xie 			}
765d19533e8SHuawei Xie 
766d19533e8SHuawei Xie 			/* Specify the descriptor number on TX. */
767d19533e8SHuawei Xie 			if (!strncmp(long_option[option_index].name,
768d19533e8SHuawei Xie 				"tx-desc-num", MAX_LONG_OPT_SZ)) {
769d19533e8SHuawei Xie 				ret = parse_num_opt(optarg, MAX_RING_DESC);
770d19533e8SHuawei Xie 				if ((ret == -1) || (!POWEROF2(ret))) {
771d19533e8SHuawei Xie 					RTE_LOG(INFO, VHOST_CONFIG,
772d19533e8SHuawei Xie 					"Invalid argument for tx-desc-num [0-N],"
773d19533e8SHuawei Xie 					"power of 2 required.\n");
774d19533e8SHuawei Xie 					us_vhost_usage(prgname);
775d19533e8SHuawei Xie 					return -1;
776d19533e8SHuawei Xie 				} else {
777d19533e8SHuawei Xie 					num_tx_descriptor = ret;
778d19533e8SHuawei Xie 				}
779d19533e8SHuawei Xie 			}
780d19533e8SHuawei Xie 
781d19533e8SHuawei Xie 			break;
782d19533e8SHuawei Xie 
783d19533e8SHuawei Xie 			/* Invalid option - print options. */
784d19533e8SHuawei Xie 		default:
785d19533e8SHuawei Xie 			us_vhost_usage(prgname);
786d19533e8SHuawei Xie 			return -1;
787d19533e8SHuawei Xie 		}
788d19533e8SHuawei Xie 	}
789d19533e8SHuawei Xie 
790d19533e8SHuawei Xie 	for (i = 0; i < RTE_MAX_ETHPORTS; i++) {
791d19533e8SHuawei Xie 		if (enabled_port_mask & (1 << i))
792d19533e8SHuawei Xie 			ports[num_ports++] = (uint8_t)i;
793d19533e8SHuawei Xie 	}
794d19533e8SHuawei Xie 
795d19533e8SHuawei Xie 	if ((num_ports ==  0) || (num_ports > MAX_SUP_PORTS)) {
796d19533e8SHuawei Xie 		RTE_LOG(INFO, VHOST_PORT, "Current enabled port number is %u,"
797d19533e8SHuawei Xie 			"but only %u port can be enabled\n",num_ports, MAX_SUP_PORTS);
798d19533e8SHuawei Xie 		return -1;
799d19533e8SHuawei Xie 	}
800d19533e8SHuawei Xie 
801d19533e8SHuawei Xie 	if ((zero_copy == 1) && (vm2vm_mode == VM2VM_SOFTWARE)) {
802d19533e8SHuawei Xie 		RTE_LOG(INFO, VHOST_PORT,
803d19533e8SHuawei Xie 			"Vhost zero copy doesn't support software vm2vm,"
804d19533e8SHuawei Xie 			"please specify 'vm2vm 2' to use hardware vm2vm.\n");
805d19533e8SHuawei Xie 		return -1;
806d19533e8SHuawei Xie 	}
807d19533e8SHuawei Xie 
808d19533e8SHuawei Xie 	if ((zero_copy == 1) && (vmdq_conf_default.rxmode.jumbo_frame == 1)) {
809d19533e8SHuawei Xie 		RTE_LOG(INFO, VHOST_PORT,
810d19533e8SHuawei Xie 			"Vhost zero copy doesn't support jumbo frame,"
811d19533e8SHuawei Xie 			"please specify '--mergeable 0' to disable the "
812d19533e8SHuawei Xie 			"mergeable feature.\n");
813d19533e8SHuawei Xie 		return -1;
814d19533e8SHuawei Xie 	}
815d19533e8SHuawei Xie 
816d19533e8SHuawei Xie 	return 0;
817d19533e8SHuawei Xie }
818d19533e8SHuawei Xie 
819d19533e8SHuawei Xie /*
820d19533e8SHuawei Xie  * Update the global var NUM_PORTS and array PORTS according to system ports number
821d19533e8SHuawei Xie  * and return valid ports number
822d19533e8SHuawei Xie  */
823d19533e8SHuawei Xie static unsigned check_ports_num(unsigned nb_ports)
824d19533e8SHuawei Xie {
825d19533e8SHuawei Xie 	unsigned valid_num_ports = num_ports;
826d19533e8SHuawei Xie 	unsigned portid;
827d19533e8SHuawei Xie 
828d19533e8SHuawei Xie 	if (num_ports > nb_ports) {
829d19533e8SHuawei Xie 		RTE_LOG(INFO, VHOST_PORT, "\nSpecified port number(%u) exceeds total system port number(%u)\n",
830d19533e8SHuawei Xie 			num_ports, nb_ports);
831d19533e8SHuawei Xie 		num_ports = nb_ports;
832d19533e8SHuawei Xie 	}
833d19533e8SHuawei Xie 
834d19533e8SHuawei Xie 	for (portid = 0; portid < num_ports; portid ++) {
835d19533e8SHuawei Xie 		if (ports[portid] >= nb_ports) {
836d19533e8SHuawei Xie 			RTE_LOG(INFO, VHOST_PORT, "\nSpecified port ID(%u) exceeds max system port ID(%u)\n",
837d19533e8SHuawei Xie 				ports[portid], (nb_ports - 1));
838d19533e8SHuawei Xie 			ports[portid] = INVALID_PORT_ID;
839d19533e8SHuawei Xie 			valid_num_ports--;
840d19533e8SHuawei Xie 		}
841d19533e8SHuawei Xie 	}
842d19533e8SHuawei Xie 	return valid_num_ports;
843d19533e8SHuawei Xie }
844d19533e8SHuawei Xie 
845d19533e8SHuawei Xie /*
846d19533e8SHuawei Xie  * Macro to print out packet contents. Wrapped in debug define so that the
847d19533e8SHuawei Xie  * data path is not effected when debug is disabled.
848d19533e8SHuawei Xie  */
849d19533e8SHuawei Xie #ifdef DEBUG
850d19533e8SHuawei Xie #define PRINT_PACKET(device, addr, size, header) do {																\
851d19533e8SHuawei Xie 	char *pkt_addr = (char*)(addr);																					\
852d19533e8SHuawei Xie 	unsigned int index;																								\
853d19533e8SHuawei Xie 	char packet[MAX_PRINT_BUFF];																					\
854d19533e8SHuawei Xie 																													\
855d19533e8SHuawei Xie 	if ((header))																									\
856d19533e8SHuawei Xie 		snprintf(packet, MAX_PRINT_BUFF, "(%"PRIu64") Header size %d: ", (device->device_fh), (size));				\
857d19533e8SHuawei Xie 	else																											\
858d19533e8SHuawei Xie 		snprintf(packet, MAX_PRINT_BUFF, "(%"PRIu64") Packet size %d: ", (device->device_fh), (size));				\
859d19533e8SHuawei Xie 	for (index = 0; index < (size); index++) {																		\
860d19533e8SHuawei Xie 		snprintf(packet + strnlen(packet, MAX_PRINT_BUFF), MAX_PRINT_BUFF - strnlen(packet, MAX_PRINT_BUFF),	\
861d19533e8SHuawei Xie 			"%02hhx ", pkt_addr[index]);																			\
862d19533e8SHuawei Xie 	}																												\
863d19533e8SHuawei Xie 	snprintf(packet + strnlen(packet, MAX_PRINT_BUFF), MAX_PRINT_BUFF - strnlen(packet, MAX_PRINT_BUFF), "\n");	\
864d19533e8SHuawei Xie 																													\
865d19533e8SHuawei Xie 	LOG_DEBUG(VHOST_DATA, "%s", packet);																					\
866d19533e8SHuawei Xie } while(0)
867d19533e8SHuawei Xie #else
868d19533e8SHuawei Xie #define PRINT_PACKET(device, addr, size, header) do{} while(0)
869d19533e8SHuawei Xie #endif
870d19533e8SHuawei Xie 
871d19533e8SHuawei Xie /*
872d19533e8SHuawei Xie  * Function to convert guest physical addresses to vhost physical addresses.
873d19533e8SHuawei Xie  * This is used to convert virtio buffer addresses.
874d19533e8SHuawei Xie  */
875d19533e8SHuawei Xie static inline uint64_t __attribute__((always_inline))
876e571e6b4SHuawei Xie gpa_to_hpa(struct vhost_dev  *vdev, uint64_t guest_pa,
877d19533e8SHuawei Xie 	uint32_t buf_len, hpa_type *addr_type)
878d19533e8SHuawei Xie {
879d19533e8SHuawei Xie 	struct virtio_memory_regions_hpa *region;
880d19533e8SHuawei Xie 	uint32_t regionidx;
881d19533e8SHuawei Xie 	uint64_t vhost_pa = 0;
882d19533e8SHuawei Xie 
883d19533e8SHuawei Xie 	*addr_type = PHYS_ADDR_INVALID;
884d19533e8SHuawei Xie 
885e571e6b4SHuawei Xie 	for (regionidx = 0; regionidx < vdev->nregions_hpa; regionidx++) {
886e571e6b4SHuawei Xie 		region = &vdev->regions_hpa[regionidx];
887d19533e8SHuawei Xie 		if ((guest_pa >= region->guest_phys_address) &&
888d19533e8SHuawei Xie 			(guest_pa <= region->guest_phys_address_end)) {
889d19533e8SHuawei Xie 			vhost_pa = region->host_phys_addr_offset + guest_pa;
890d19533e8SHuawei Xie 			if (likely((guest_pa + buf_len - 1)
891d19533e8SHuawei Xie 				<= region->guest_phys_address_end))
892d19533e8SHuawei Xie 				*addr_type = PHYS_ADDR_CONTINUOUS;
893d19533e8SHuawei Xie 			else
894d19533e8SHuawei Xie 				*addr_type = PHYS_ADDR_CROSS_SUBREG;
895d19533e8SHuawei Xie 			break;
896d19533e8SHuawei Xie 		}
897d19533e8SHuawei Xie 	}
898d19533e8SHuawei Xie 
899d19533e8SHuawei Xie 	LOG_DEBUG(VHOST_DATA, "(%"PRIu64") GPA %p| HPA %p\n",
900e571e6b4SHuawei Xie 		vdev->dev->device_fh, (void *)(uintptr_t)guest_pa,
901d19533e8SHuawei Xie 		(void *)(uintptr_t)vhost_pa);
902d19533e8SHuawei Xie 
903d19533e8SHuawei Xie 	return vhost_pa;
904d19533e8SHuawei Xie }
905d19533e8SHuawei Xie 
906d19533e8SHuawei Xie /*
907d19533e8SHuawei Xie  * Compares a packet destination MAC address to a device MAC address.
908d19533e8SHuawei Xie  */
909d19533e8SHuawei Xie static inline int __attribute__((always_inline))
910d19533e8SHuawei Xie ether_addr_cmp(struct ether_addr *ea, struct ether_addr *eb)
911d19533e8SHuawei Xie {
912d19533e8SHuawei Xie 	return (((*(uint64_t *)ea ^ *(uint64_t *)eb) & MAC_ADDR_CMP) == 0);
913d19533e8SHuawei Xie }
914d19533e8SHuawei Xie 
915d19533e8SHuawei Xie /*
916d19533e8SHuawei Xie  * This function learns the MAC address of the device and registers this along with a
917d19533e8SHuawei Xie  * vlan tag to a VMDQ.
918d19533e8SHuawei Xie  */
919d19533e8SHuawei Xie static int
920e571e6b4SHuawei Xie link_vmdq(struct vhost_dev *vdev, struct rte_mbuf *m)
921d19533e8SHuawei Xie {
922d19533e8SHuawei Xie 	struct ether_hdr *pkt_hdr;
923d19533e8SHuawei Xie 	struct virtio_net_data_ll *dev_ll;
924e571e6b4SHuawei Xie 	struct virtio_net *dev = vdev->dev;
925d19533e8SHuawei Xie 	int i, ret;
926d19533e8SHuawei Xie 
927d19533e8SHuawei Xie 	/* Learn MAC address of guest device from packet */
928d19533e8SHuawei Xie 	pkt_hdr = rte_pktmbuf_mtod(m, struct ether_hdr *);
929d19533e8SHuawei Xie 
930d19533e8SHuawei Xie 	dev_ll = ll_root_used;
931d19533e8SHuawei Xie 
932d19533e8SHuawei Xie 	while (dev_ll != NULL) {
933e571e6b4SHuawei Xie 		if (ether_addr_cmp(&(pkt_hdr->s_addr), &dev_ll->vdev->mac_address)) {
934d19533e8SHuawei Xie 			RTE_LOG(INFO, VHOST_DATA, "(%"PRIu64") WARNING: This device is using an existing MAC address and has not been registered.\n", dev->device_fh);
935d19533e8SHuawei Xie 			return -1;
936d19533e8SHuawei Xie 		}
937d19533e8SHuawei Xie 		dev_ll = dev_ll->next;
938d19533e8SHuawei Xie 	}
939d19533e8SHuawei Xie 
940d19533e8SHuawei Xie 	for (i = 0; i < ETHER_ADDR_LEN; i++)
941e571e6b4SHuawei Xie 		vdev->mac_address.addr_bytes[i] = pkt_hdr->s_addr.addr_bytes[i];
942d19533e8SHuawei Xie 
943d19533e8SHuawei Xie 	/* vlan_tag currently uses the device_id. */
944e571e6b4SHuawei Xie 	vdev->vlan_tag = vlan_tags[dev->device_fh];
945d19533e8SHuawei Xie 
946d19533e8SHuawei Xie 	/* Print out VMDQ registration info. */
947d19533e8SHuawei Xie 	RTE_LOG(INFO, VHOST_DATA, "(%"PRIu64") MAC_ADDRESS %02x:%02x:%02x:%02x:%02x:%02x and VLAN_TAG %d registered\n",
948d19533e8SHuawei Xie 		dev->device_fh,
949e571e6b4SHuawei Xie 		vdev->mac_address.addr_bytes[0], vdev->mac_address.addr_bytes[1],
950e571e6b4SHuawei Xie 		vdev->mac_address.addr_bytes[2], vdev->mac_address.addr_bytes[3],
951e571e6b4SHuawei Xie 		vdev->mac_address.addr_bytes[4], vdev->mac_address.addr_bytes[5],
952e571e6b4SHuawei Xie 		vdev->vlan_tag);
953d19533e8SHuawei Xie 
954d19533e8SHuawei Xie 	/* Register the MAC address. */
95584b02d16SHuawei Xie 	ret = rte_eth_dev_mac_addr_add(ports[0], &vdev->mac_address,
95684b02d16SHuawei Xie 				(uint32_t)dev->device_fh + vmdq_pool_base);
957d19533e8SHuawei Xie 	if (ret)
958d19533e8SHuawei Xie 		RTE_LOG(ERR, VHOST_DATA, "(%"PRIu64") Failed to add device MAC address to VMDQ\n",
959d19533e8SHuawei Xie 					dev->device_fh);
960d19533e8SHuawei Xie 
961d19533e8SHuawei Xie 	/* Enable stripping of the vlan tag as we handle routing. */
962e3d61d16SOuyang Changchun 	if (vlan_strip)
963e3d61d16SOuyang Changchun 		rte_eth_dev_set_vlan_strip_on_queue(ports[0],
964e3d61d16SOuyang Changchun 			(uint16_t)vdev->vmdq_rx_q, 1);
965d19533e8SHuawei Xie 
966d19533e8SHuawei Xie 	/* Set device as ready for RX. */
967e571e6b4SHuawei Xie 	vdev->ready = DEVICE_RX;
968d19533e8SHuawei Xie 
969d19533e8SHuawei Xie 	return 0;
970d19533e8SHuawei Xie }
971d19533e8SHuawei Xie 
972d19533e8SHuawei Xie /*
973d19533e8SHuawei Xie  * Removes MAC address and vlan tag from VMDQ. Ensures that nothing is adding buffers to the RX
974d19533e8SHuawei Xie  * queue before disabling RX on the device.
975d19533e8SHuawei Xie  */
976d19533e8SHuawei Xie static inline void
977e571e6b4SHuawei Xie unlink_vmdq(struct vhost_dev *vdev)
978d19533e8SHuawei Xie {
979d19533e8SHuawei Xie 	unsigned i = 0;
980d19533e8SHuawei Xie 	unsigned rx_count;
981d19533e8SHuawei Xie 	struct rte_mbuf *pkts_burst[MAX_PKT_BURST];
982d19533e8SHuawei Xie 
983e571e6b4SHuawei Xie 	if (vdev->ready == DEVICE_RX) {
984d19533e8SHuawei Xie 		/*clear MAC and VLAN settings*/
985e571e6b4SHuawei Xie 		rte_eth_dev_mac_addr_remove(ports[0], &vdev->mac_address);
986d19533e8SHuawei Xie 		for (i = 0; i < 6; i++)
987e571e6b4SHuawei Xie 			vdev->mac_address.addr_bytes[i] = 0;
988d19533e8SHuawei Xie 
989e571e6b4SHuawei Xie 		vdev->vlan_tag = 0;
990d19533e8SHuawei Xie 
991d19533e8SHuawei Xie 		/*Clear out the receive buffers*/
992d19533e8SHuawei Xie 		rx_count = rte_eth_rx_burst(ports[0],
993e571e6b4SHuawei Xie 					(uint16_t)vdev->vmdq_rx_q, pkts_burst, MAX_PKT_BURST);
994d19533e8SHuawei Xie 
995d19533e8SHuawei Xie 		while (rx_count) {
996d19533e8SHuawei Xie 			for (i = 0; i < rx_count; i++)
997d19533e8SHuawei Xie 				rte_pktmbuf_free(pkts_burst[i]);
998d19533e8SHuawei Xie 
999d19533e8SHuawei Xie 			rx_count = rte_eth_rx_burst(ports[0],
1000e571e6b4SHuawei Xie 					(uint16_t)vdev->vmdq_rx_q, pkts_burst, MAX_PKT_BURST);
1001d19533e8SHuawei Xie 		}
1002d19533e8SHuawei Xie 
1003e571e6b4SHuawei Xie 		vdev->ready = DEVICE_MAC_LEARNING;
1004d19533e8SHuawei Xie 	}
1005d19533e8SHuawei Xie }
1006d19533e8SHuawei Xie 
1007d19533e8SHuawei Xie /*
1008d19533e8SHuawei Xie  * Check if the packet destination MAC address is for a local device. If so then put
1009d19533e8SHuawei Xie  * the packet on that devices RX queue. If not then return.
1010d19533e8SHuawei Xie  */
1011a981294bSHuawei Xie static inline int __attribute__((always_inline))
1012e571e6b4SHuawei Xie virtio_tx_local(struct vhost_dev *vdev, struct rte_mbuf *m)
1013d19533e8SHuawei Xie {
1014d19533e8SHuawei Xie 	struct virtio_net_data_ll *dev_ll;
1015d19533e8SHuawei Xie 	struct ether_hdr *pkt_hdr;
1016d19533e8SHuawei Xie 	uint64_t ret = 0;
1017e571e6b4SHuawei Xie 	struct virtio_net *dev = vdev->dev;
1018e571e6b4SHuawei Xie 	struct virtio_net *tdev; /* destination virito device */
1019d19533e8SHuawei Xie 
1020d19533e8SHuawei Xie 	pkt_hdr = rte_pktmbuf_mtod(m, struct ether_hdr *);
1021d19533e8SHuawei Xie 
1022d19533e8SHuawei Xie 	/*get the used devices list*/
1023d19533e8SHuawei Xie 	dev_ll = ll_root_used;
1024d19533e8SHuawei Xie 
1025d19533e8SHuawei Xie 	while (dev_ll != NULL) {
1026e571e6b4SHuawei Xie 		if ((dev_ll->vdev->ready == DEVICE_RX) && ether_addr_cmp(&(pkt_hdr->d_addr),
1027e571e6b4SHuawei Xie 				          &dev_ll->vdev->mac_address)) {
1028d19533e8SHuawei Xie 
1029d19533e8SHuawei Xie 			/* Drop the packet if the TX packet is destined for the TX device. */
1030e571e6b4SHuawei Xie 			if (dev_ll->vdev->dev->device_fh == dev->device_fh) {
1031d19533e8SHuawei Xie 				LOG_DEBUG(VHOST_DATA, "(%"PRIu64") TX: Source and destination MAC addresses are the same. Dropping packet.\n",
1032e571e6b4SHuawei Xie 							dev->device_fh);
1033d19533e8SHuawei Xie 				return 0;
1034d19533e8SHuawei Xie 			}
1035e571e6b4SHuawei Xie 			tdev = dev_ll->vdev->dev;
1036d19533e8SHuawei Xie 
1037d19533e8SHuawei Xie 
1038e571e6b4SHuawei Xie 			LOG_DEBUG(VHOST_DATA, "(%"PRIu64") TX: MAC address is local\n", tdev->device_fh);
1039d19533e8SHuawei Xie 
1040364dddcdSHuawei Xie 			if (unlikely(dev_ll->vdev->remove)) {
1041d19533e8SHuawei Xie 				/*drop the packet if the device is marked for removal*/
1042e571e6b4SHuawei Xie 				LOG_DEBUG(VHOST_DATA, "(%"PRIu64") Device is marked for removal\n", tdev->device_fh);
1043d19533e8SHuawei Xie 			} else {
1044d19533e8SHuawei Xie 				/*send the packet to the local virtio device*/
1045be800696SHuawei Xie 				ret = rte_vhost_enqueue_burst(tdev, VIRTIO_RXQ, &m, 1);
1046d19533e8SHuawei Xie 				if (enable_stats) {
1047d19533e8SHuawei Xie 					rte_atomic64_add(
1048e571e6b4SHuawei Xie 					&dev_statistics[tdev->device_fh].rx_total_atomic,
1049d19533e8SHuawei Xie 					1);
1050d19533e8SHuawei Xie 					rte_atomic64_add(
1051e571e6b4SHuawei Xie 					&dev_statistics[tdev->device_fh].rx_atomic,
1052d19533e8SHuawei Xie 					ret);
1053e571e6b4SHuawei Xie 					dev_statistics[tdev->device_fh].tx_total++;
1054e571e6b4SHuawei Xie 					dev_statistics[tdev->device_fh].tx += ret;
1055d19533e8SHuawei Xie 				}
1056d19533e8SHuawei Xie 			}
1057d19533e8SHuawei Xie 
1058d19533e8SHuawei Xie 			return 0;
1059d19533e8SHuawei Xie 		}
1060d19533e8SHuawei Xie 		dev_ll = dev_ll->next;
1061d19533e8SHuawei Xie 	}
1062d19533e8SHuawei Xie 
1063d19533e8SHuawei Xie 	return -1;
1064d19533e8SHuawei Xie }
1065d19533e8SHuawei Xie 
1066d19533e8SHuawei Xie /*
106772ec8d77SOuyang Changchun  * Check if the destination MAC of a packet is one local VM,
106872ec8d77SOuyang Changchun  * and get its vlan tag, and offset if it is.
1069d19533e8SHuawei Xie  */
107072ec8d77SOuyang Changchun static inline int __attribute__((always_inline))
107172ec8d77SOuyang Changchun find_local_dest(struct virtio_net *dev, struct rte_mbuf *m,
107272ec8d77SOuyang Changchun 	uint32_t *offset, uint16_t *vlan_tag)
1073d19533e8SHuawei Xie {
1074d19533e8SHuawei Xie 	struct virtio_net_data_ll *dev_ll = ll_root_used;
1075d19533e8SHuawei Xie 	struct ether_hdr *pkt_hdr = rte_pktmbuf_mtod(m, struct ether_hdr *);
1076d19533e8SHuawei Xie 
1077d19533e8SHuawei Xie 	while (dev_ll != NULL) {
1078e571e6b4SHuawei Xie 		if ((dev_ll->vdev->ready == DEVICE_RX)
1079d19533e8SHuawei Xie 			&& ether_addr_cmp(&(pkt_hdr->d_addr),
1080e571e6b4SHuawei Xie 		&dev_ll->vdev->mac_address)) {
1081d19533e8SHuawei Xie 			/*
1082d19533e8SHuawei Xie 			 * Drop the packet if the TX packet is
1083d19533e8SHuawei Xie 			 * destined for the TX device.
1084d19533e8SHuawei Xie 			 */
1085e571e6b4SHuawei Xie 			if (dev_ll->vdev->dev->device_fh == dev->device_fh) {
1086d19533e8SHuawei Xie 				LOG_DEBUG(VHOST_DATA,
1087d19533e8SHuawei Xie 				"(%"PRIu64") TX: Source and destination"
1088d19533e8SHuawei Xie 				" MAC addresses are the same. Dropping "
1089d19533e8SHuawei Xie 				"packet.\n",
10904d50b6acSHuawei Xie 				dev_ll->vdev->dev->device_fh);
109172ec8d77SOuyang Changchun 				return -1;
1092d19533e8SHuawei Xie 			}
1093e44fb8a4SOuyang Changchun 
1094e44fb8a4SOuyang Changchun 			/*
1095e44fb8a4SOuyang Changchun 			 * HW vlan strip will reduce the packet length
1096e44fb8a4SOuyang Changchun 			 * by minus length of vlan tag, so need restore
1097e44fb8a4SOuyang Changchun 			 * the packet length by plus it.
1098e44fb8a4SOuyang Changchun 			 */
109972ec8d77SOuyang Changchun 			*offset = VLAN_HLEN;
110072ec8d77SOuyang Changchun 			*vlan_tag =
1101d19533e8SHuawei Xie 			(uint16_t)
1102e571e6b4SHuawei Xie 			vlan_tags[(uint16_t)dev_ll->vdev->dev->device_fh];
1103d19533e8SHuawei Xie 
1104d19533e8SHuawei Xie 			LOG_DEBUG(VHOST_DATA,
1105d19533e8SHuawei Xie 			"(%"PRIu64") TX: pkt to local VM device id:"
1106d19533e8SHuawei Xie 			"(%"PRIu64") vlan tag: %d.\n",
1107e571e6b4SHuawei Xie 			dev->device_fh, dev_ll->vdev->dev->device_fh,
1108e6eff1baSOuyang Changchun 			(int)*vlan_tag);
1109d19533e8SHuawei Xie 
1110d19533e8SHuawei Xie 			break;
1111d19533e8SHuawei Xie 		}
1112d19533e8SHuawei Xie 		dev_ll = dev_ll->next;
1113d19533e8SHuawei Xie 	}
111472ec8d77SOuyang Changchun 	return 0;
111572ec8d77SOuyang Changchun }
111672ec8d77SOuyang Changchun 
111772ec8d77SOuyang Changchun /*
111872ec8d77SOuyang Changchun  * This function routes the TX packet to the correct interface. This may be a local device
111972ec8d77SOuyang Changchun  * or the physical port.
112072ec8d77SOuyang Changchun  */
112172ec8d77SOuyang Changchun static inline void __attribute__((always_inline))
112272ec8d77SOuyang Changchun virtio_tx_route(struct vhost_dev *vdev, struct rte_mbuf *m, uint16_t vlan_tag)
112372ec8d77SOuyang Changchun {
112472ec8d77SOuyang Changchun 	struct mbuf_table *tx_q;
112572ec8d77SOuyang Changchun 	struct rte_mbuf **m_table;
112672ec8d77SOuyang Changchun 	unsigned len, ret, offset = 0;
112772ec8d77SOuyang Changchun 	const uint16_t lcore_id = rte_lcore_id();
112872ec8d77SOuyang Changchun 	struct virtio_net *dev = vdev->dev;
11298b9bb988SOuyang Changchun 	struct ether_hdr *nh;
113072ec8d77SOuyang Changchun 
113172ec8d77SOuyang Changchun 	/*check if destination is local VM*/
113272ec8d77SOuyang Changchun 	if ((vm2vm_mode == VM2VM_SOFTWARE) && (virtio_tx_local(vdev, m) == 0)) {
113372ec8d77SOuyang Changchun 		rte_pktmbuf_free(m);
113472ec8d77SOuyang Changchun 		return;
113572ec8d77SOuyang Changchun 	}
113672ec8d77SOuyang Changchun 
1137c2ab5162SOuyang Changchun 	if (unlikely(vm2vm_mode == VM2VM_HARDWARE)) {
1138c2ab5162SOuyang Changchun 		if (unlikely(find_local_dest(dev, m, &offset, &vlan_tag) != 0)) {
113972ec8d77SOuyang Changchun 			rte_pktmbuf_free(m);
114072ec8d77SOuyang Changchun 			return;
114172ec8d77SOuyang Changchun 		}
1142d19533e8SHuawei Xie 	}
1143d19533e8SHuawei Xie 
1144d19533e8SHuawei Xie 	LOG_DEBUG(VHOST_DATA, "(%"PRIu64") TX: MAC address is external\n", dev->device_fh);
1145d19533e8SHuawei Xie 
1146d19533e8SHuawei Xie 	/*Add packet to the port tx queue*/
1147d19533e8SHuawei Xie 	tx_q = &lcore_tx_queue[lcore_id];
1148d19533e8SHuawei Xie 	len = tx_q->len;
1149d19533e8SHuawei Xie 
11508b9bb988SOuyang Changchun 	nh = rte_pktmbuf_mtod(m, struct ether_hdr *);
11518b9bb988SOuyang Changchun 	if (unlikely(nh->ether_type == rte_cpu_to_be_16(ETHER_TYPE_VLAN))) {
11528b9bb988SOuyang Changchun 		/* Guest has inserted the vlan tag. */
11538b9bb988SOuyang Changchun 		struct vlan_hdr *vh = (struct vlan_hdr *) (nh + 1);
11548b9bb988SOuyang Changchun 		uint16_t vlan_tag_be = rte_cpu_to_be_16(vlan_tag);
11558b9bb988SOuyang Changchun 		if ((vm2vm_mode == VM2VM_HARDWARE) &&
11568b9bb988SOuyang Changchun 			(vh->vlan_tci != vlan_tag_be))
11578b9bb988SOuyang Changchun 			vh->vlan_tci = vlan_tag_be;
11588b9bb988SOuyang Changchun 	} else {
11594d50b6acSHuawei Xie 		m->ol_flags = PKT_TX_VLAN_PKT;
1160e44fb8a4SOuyang Changchun 
1161c2ab5162SOuyang Changchun 		/*
1162c2ab5162SOuyang Changchun 		 * Find the right seg to adjust the data len when offset is
1163c2ab5162SOuyang Changchun 		 * bigger than tail room size.
1164c2ab5162SOuyang Changchun 		 */
1165c2ab5162SOuyang Changchun 		if (unlikely(vm2vm_mode == VM2VM_HARDWARE)) {
1166c2ab5162SOuyang Changchun 			if (likely(offset <= rte_pktmbuf_tailroom(m)))
11674d50b6acSHuawei Xie 				m->data_len += offset;
1168c2ab5162SOuyang Changchun 			else {
1169c2ab5162SOuyang Changchun 				struct rte_mbuf *seg = m;
1170c2ab5162SOuyang Changchun 
1171c2ab5162SOuyang Changchun 				while ((seg->next != NULL) &&
1172c2ab5162SOuyang Changchun 					(offset > rte_pktmbuf_tailroom(seg)))
1173c2ab5162SOuyang Changchun 					seg = seg->next;
1174c2ab5162SOuyang Changchun 
1175c2ab5162SOuyang Changchun 				seg->data_len += offset;
1176c2ab5162SOuyang Changchun 			}
1177e44fb8a4SOuyang Changchun 			m->pkt_len += offset;
1178c2ab5162SOuyang Changchun 		}
1179e44fb8a4SOuyang Changchun 
11804d50b6acSHuawei Xie 		m->vlan_tci = vlan_tag;
11818b9bb988SOuyang Changchun 	}
1182d19533e8SHuawei Xie 
11834d50b6acSHuawei Xie 	tx_q->m_table[len] = m;
1184d19533e8SHuawei Xie 	len++;
1185d19533e8SHuawei Xie 	if (enable_stats) {
1186d19533e8SHuawei Xie 		dev_statistics[dev->device_fh].tx_total++;
1187d19533e8SHuawei Xie 		dev_statistics[dev->device_fh].tx++;
1188d19533e8SHuawei Xie 	}
1189d19533e8SHuawei Xie 
1190d19533e8SHuawei Xie 	if (unlikely(len == MAX_PKT_BURST)) {
1191d19533e8SHuawei Xie 		m_table = (struct rte_mbuf **)tx_q->m_table;
1192d19533e8SHuawei Xie 		ret = rte_eth_tx_burst(ports[0], (uint16_t)tx_q->txq_id, m_table, (uint16_t) len);
1193d19533e8SHuawei Xie 		/* Free any buffers not handled by TX and update the port stats. */
1194d19533e8SHuawei Xie 		if (unlikely(ret < len)) {
1195d19533e8SHuawei Xie 			do {
1196d19533e8SHuawei Xie 				rte_pktmbuf_free(m_table[ret]);
1197d19533e8SHuawei Xie 			} while (++ret < len);
1198d19533e8SHuawei Xie 		}
1199d19533e8SHuawei Xie 
1200d19533e8SHuawei Xie 		len = 0;
1201d19533e8SHuawei Xie 	}
1202d19533e8SHuawei Xie 
1203d19533e8SHuawei Xie 	tx_q->len = len;
1204d19533e8SHuawei Xie 	return;
1205d19533e8SHuawei Xie }
1206d19533e8SHuawei Xie /*
1207d19533e8SHuawei Xie  * This function is called by each data core. It handles all RX/TX registered with the
1208d19533e8SHuawei Xie  * core. For TX the specific lcore linked list is used. For RX, MAC addresses are compared
1209d19533e8SHuawei Xie  * with all devices in the main linked list.
1210d19533e8SHuawei Xie  */
1211d19533e8SHuawei Xie static int
1212d19533e8SHuawei Xie switch_worker(__attribute__((unused)) void *arg)
1213d19533e8SHuawei Xie {
1214d19533e8SHuawei Xie 	struct rte_mempool *mbuf_pool = arg;
1215d19533e8SHuawei Xie 	struct virtio_net *dev = NULL;
1216e571e6b4SHuawei Xie 	struct vhost_dev *vdev = NULL;
1217d19533e8SHuawei Xie 	struct rte_mbuf *pkts_burst[MAX_PKT_BURST];
1218d19533e8SHuawei Xie 	struct virtio_net_data_ll *dev_ll;
1219d19533e8SHuawei Xie 	struct mbuf_table *tx_q;
1220d19533e8SHuawei Xie 	volatile struct lcore_ll_info *lcore_ll;
1221d19533e8SHuawei Xie 	const uint64_t drain_tsc = (rte_get_tsc_hz() + US_PER_S - 1) / US_PER_S * BURST_TX_DRAIN_US;
1222d19533e8SHuawei Xie 	uint64_t prev_tsc, diff_tsc, cur_tsc, ret_count = 0;
1223d19533e8SHuawei Xie 	unsigned ret, i;
1224d19533e8SHuawei Xie 	const uint16_t lcore_id = rte_lcore_id();
1225d19533e8SHuawei Xie 	const uint16_t num_cores = (uint16_t)rte_lcore_count();
1226d19533e8SHuawei Xie 	uint16_t rx_count = 0;
1227be800696SHuawei Xie 	uint16_t tx_count;
1228be800696SHuawei Xie 	uint32_t retry = 0;
1229d19533e8SHuawei Xie 
1230d19533e8SHuawei Xie 	RTE_LOG(INFO, VHOST_DATA, "Procesing on Core %u started\n", lcore_id);
1231d19533e8SHuawei Xie 	lcore_ll = lcore_info[lcore_id].lcore_ll;
1232d19533e8SHuawei Xie 	prev_tsc = 0;
1233d19533e8SHuawei Xie 
1234d19533e8SHuawei Xie 	tx_q = &lcore_tx_queue[lcore_id];
1235d19533e8SHuawei Xie 	for (i = 0; i < num_cores; i ++) {
1236d19533e8SHuawei Xie 		if (lcore_ids[i] == lcore_id) {
1237d19533e8SHuawei Xie 			tx_q->txq_id = i;
1238d19533e8SHuawei Xie 			break;
1239d19533e8SHuawei Xie 		}
1240d19533e8SHuawei Xie 	}
1241d19533e8SHuawei Xie 
1242d19533e8SHuawei Xie 	while(1) {
1243d19533e8SHuawei Xie 		cur_tsc = rte_rdtsc();
1244d19533e8SHuawei Xie 		/*
1245d19533e8SHuawei Xie 		 * TX burst queue drain
1246d19533e8SHuawei Xie 		 */
1247d19533e8SHuawei Xie 		diff_tsc = cur_tsc - prev_tsc;
1248d19533e8SHuawei Xie 		if (unlikely(diff_tsc > drain_tsc)) {
1249d19533e8SHuawei Xie 
1250d19533e8SHuawei Xie 			if (tx_q->len) {
1251d19533e8SHuawei Xie 				LOG_DEBUG(VHOST_DATA, "TX queue drained after timeout with burst size %u \n", tx_q->len);
1252d19533e8SHuawei Xie 
1253d19533e8SHuawei Xie 				/*Tx any packets in the queue*/
1254d19533e8SHuawei Xie 				ret = rte_eth_tx_burst(ports[0], (uint16_t)tx_q->txq_id,
1255d19533e8SHuawei Xie 									   (struct rte_mbuf **)tx_q->m_table,
1256d19533e8SHuawei Xie 									   (uint16_t)tx_q->len);
1257d19533e8SHuawei Xie 				if (unlikely(ret < tx_q->len)) {
1258d19533e8SHuawei Xie 					do {
1259d19533e8SHuawei Xie 						rte_pktmbuf_free(tx_q->m_table[ret]);
1260d19533e8SHuawei Xie 					} while (++ret < tx_q->len);
1261d19533e8SHuawei Xie 				}
1262d19533e8SHuawei Xie 
1263d19533e8SHuawei Xie 				tx_q->len = 0;
1264d19533e8SHuawei Xie 			}
1265d19533e8SHuawei Xie 
1266d19533e8SHuawei Xie 			prev_tsc = cur_tsc;
1267d19533e8SHuawei Xie 
1268d19533e8SHuawei Xie 		}
1269d19533e8SHuawei Xie 
1270d19533e8SHuawei Xie 		rte_prefetch0(lcore_ll->ll_root_used);
1271d19533e8SHuawei Xie 		/*
1272d19533e8SHuawei Xie 		 * Inform the configuration core that we have exited the linked list and that no devices are
1273d19533e8SHuawei Xie 		 * in use if requested.
1274d19533e8SHuawei Xie 		 */
1275d19533e8SHuawei Xie 		if (lcore_ll->dev_removal_flag == REQUEST_DEV_REMOVAL)
1276d19533e8SHuawei Xie 			lcore_ll->dev_removal_flag = ACK_DEV_REMOVAL;
1277d19533e8SHuawei Xie 
1278d19533e8SHuawei Xie 		/*
1279d19533e8SHuawei Xie 		 * Process devices
1280d19533e8SHuawei Xie 		 */
1281d19533e8SHuawei Xie 		dev_ll = lcore_ll->ll_root_used;
1282d19533e8SHuawei Xie 
1283d19533e8SHuawei Xie 		while (dev_ll != NULL) {
1284d19533e8SHuawei Xie 			/*get virtio device ID*/
1285e571e6b4SHuawei Xie 			vdev = dev_ll->vdev;
1286e571e6b4SHuawei Xie 			dev = vdev->dev;
1287d19533e8SHuawei Xie 
1288364dddcdSHuawei Xie 			if (unlikely(vdev->remove)) {
1289d19533e8SHuawei Xie 				dev_ll = dev_ll->next;
1290e571e6b4SHuawei Xie 				unlink_vmdq(vdev);
1291e571e6b4SHuawei Xie 				vdev->ready = DEVICE_SAFE_REMOVE;
1292d19533e8SHuawei Xie 				continue;
1293d19533e8SHuawei Xie 			}
1294e571e6b4SHuawei Xie 			if (likely(vdev->ready == DEVICE_RX)) {
1295d19533e8SHuawei Xie 				/*Handle guest RX*/
1296d19533e8SHuawei Xie 				rx_count = rte_eth_rx_burst(ports[0],
1297e571e6b4SHuawei Xie 					vdev->vmdq_rx_q, pkts_burst, MAX_PKT_BURST);
1298d19533e8SHuawei Xie 
1299d19533e8SHuawei Xie 				if (rx_count) {
1300be800696SHuawei Xie 					/*
1301be800696SHuawei Xie 					* Retry is enabled and the queue is full then we wait and retry to avoid packet loss
1302be800696SHuawei Xie 					* Here MAX_PKT_BURST must be less than virtio queue size
1303be800696SHuawei Xie 					*/
1304be800696SHuawei Xie 					if (enable_retry && unlikely(rx_count > rte_vring_available_entries(dev, VIRTIO_RXQ))) {
1305be800696SHuawei Xie 						for (retry = 0; retry < burst_rx_retry_num; retry++) {
1306be800696SHuawei Xie 							rte_delay_us(burst_rx_delay_time);
1307be800696SHuawei Xie 							if (rx_count <= rte_vring_available_entries(dev, VIRTIO_RXQ))
1308be800696SHuawei Xie 								break;
1309be800696SHuawei Xie 						}
1310be800696SHuawei Xie 					}
1311be800696SHuawei Xie 					ret_count = rte_vhost_enqueue_burst(dev, VIRTIO_RXQ, pkts_burst, rx_count);
1312d19533e8SHuawei Xie 					if (enable_stats) {
1313d19533e8SHuawei Xie 						rte_atomic64_add(
1314e571e6b4SHuawei Xie 						&dev_statistics[dev_ll->vdev->dev->device_fh].rx_total_atomic,
1315d19533e8SHuawei Xie 						rx_count);
1316d19533e8SHuawei Xie 						rte_atomic64_add(
1317e571e6b4SHuawei Xie 						&dev_statistics[dev_ll->vdev->dev->device_fh].rx_atomic, ret_count);
1318d19533e8SHuawei Xie 					}
1319d19533e8SHuawei Xie 					while (likely(rx_count)) {
1320d19533e8SHuawei Xie 						rx_count--;
1321d19533e8SHuawei Xie 						rte_pktmbuf_free(pkts_burst[rx_count]);
1322d19533e8SHuawei Xie 					}
1323d19533e8SHuawei Xie 
1324d19533e8SHuawei Xie 				}
1325d19533e8SHuawei Xie 			}
1326d19533e8SHuawei Xie 
1327364dddcdSHuawei Xie 			if (likely(!vdev->remove)) {
1328d19533e8SHuawei Xie 				/* Handle guest TX*/
1329be800696SHuawei Xie 				tx_count = rte_vhost_dequeue_burst(dev, VIRTIO_TXQ, mbuf_pool, pkts_burst, MAX_PKT_BURST);
1330be800696SHuawei Xie 				/* If this is the first received packet we need to learn the MAC and setup VMDQ */
1331be800696SHuawei Xie 				if (unlikely(vdev->ready == DEVICE_MAC_LEARNING) && tx_count) {
1332be800696SHuawei Xie 					if (vdev->remove || (link_vmdq(vdev, pkts_burst[0]) == -1)) {
133351ec3ae2SHuawei Xie 						while (tx_count)
133451ec3ae2SHuawei Xie 							rte_pktmbuf_free(pkts_burst[--tx_count]);
1335be800696SHuawei Xie 					}
1336be800696SHuawei Xie 				}
1337be800696SHuawei Xie 				while (tx_count)
13384d50b6acSHuawei Xie 					virtio_tx_route(vdev, pkts_burst[--tx_count], (uint16_t)dev->device_fh);
1339d19533e8SHuawei Xie 			}
1340d19533e8SHuawei Xie 
1341d19533e8SHuawei Xie 			/*move to the next device in the list*/
1342d19533e8SHuawei Xie 			dev_ll = dev_ll->next;
1343d19533e8SHuawei Xie 		}
1344d19533e8SHuawei Xie 	}
1345d19533e8SHuawei Xie 
1346d19533e8SHuawei Xie 	return 0;
1347d19533e8SHuawei Xie }
1348d19533e8SHuawei Xie 
1349d19533e8SHuawei Xie /*
1350d19533e8SHuawei Xie  * This function gets available ring number for zero copy rx.
1351d19533e8SHuawei Xie  * Only one thread will call this funciton for a paticular virtio device,
1352d19533e8SHuawei Xie  * so, it is designed as non-thread-safe function.
1353d19533e8SHuawei Xie  */
1354d19533e8SHuawei Xie static inline uint32_t __attribute__((always_inline))
1355d19533e8SHuawei Xie get_available_ring_num_zcp(struct virtio_net *dev)
1356d19533e8SHuawei Xie {
1357d19533e8SHuawei Xie 	struct vhost_virtqueue *vq = dev->virtqueue[VIRTIO_RXQ];
1358d19533e8SHuawei Xie 	uint16_t avail_idx;
1359d19533e8SHuawei Xie 
1360d19533e8SHuawei Xie 	avail_idx = *((volatile uint16_t *)&vq->avail->idx);
1361d19533e8SHuawei Xie 	return (uint32_t)(avail_idx - vq->last_used_idx_res);
1362d19533e8SHuawei Xie }
1363d19533e8SHuawei Xie 
1364d19533e8SHuawei Xie /*
1365d19533e8SHuawei Xie  * This function gets available ring index for zero copy rx,
1366d19533e8SHuawei Xie  * it will retry 'burst_rx_retry_num' times till it get enough ring index.
1367d19533e8SHuawei Xie  * Only one thread will call this funciton for a paticular virtio device,
1368d19533e8SHuawei Xie  * so, it is designed as non-thread-safe function.
1369d19533e8SHuawei Xie  */
1370d19533e8SHuawei Xie static inline uint32_t __attribute__((always_inline))
1371d19533e8SHuawei Xie get_available_ring_index_zcp(struct virtio_net *dev,
1372d19533e8SHuawei Xie 	uint16_t *res_base_idx, uint32_t count)
1373d19533e8SHuawei Xie {
1374d19533e8SHuawei Xie 	struct vhost_virtqueue *vq = dev->virtqueue[VIRTIO_RXQ];
1375d19533e8SHuawei Xie 	uint16_t avail_idx;
1376d19533e8SHuawei Xie 	uint32_t retry = 0;
1377d19533e8SHuawei Xie 	uint16_t free_entries;
1378d19533e8SHuawei Xie 
1379d19533e8SHuawei Xie 	*res_base_idx = vq->last_used_idx_res;
1380d19533e8SHuawei Xie 	avail_idx = *((volatile uint16_t *)&vq->avail->idx);
1381d19533e8SHuawei Xie 	free_entries = (avail_idx - *res_base_idx);
1382d19533e8SHuawei Xie 
1383d19533e8SHuawei Xie 	LOG_DEBUG(VHOST_DATA, "(%"PRIu64") in get_available_ring_index_zcp: "
1384d19533e8SHuawei Xie 			"avail idx: %d, "
1385d19533e8SHuawei Xie 			"res base idx:%d, free entries:%d\n",
1386d19533e8SHuawei Xie 			dev->device_fh, avail_idx, *res_base_idx,
1387d19533e8SHuawei Xie 			free_entries);
1388d19533e8SHuawei Xie 
1389d19533e8SHuawei Xie 	/*
1390d19533e8SHuawei Xie 	 * If retry is enabled and the queue is full then we wait
1391d19533e8SHuawei Xie 	 * and retry to avoid packet loss.
1392d19533e8SHuawei Xie 	 */
1393d19533e8SHuawei Xie 	if (enable_retry && unlikely(count > free_entries)) {
1394d19533e8SHuawei Xie 		for (retry = 0; retry < burst_rx_retry_num; retry++) {
1395d19533e8SHuawei Xie 			rte_delay_us(burst_rx_delay_time);
1396d19533e8SHuawei Xie 			avail_idx = *((volatile uint16_t *)&vq->avail->idx);
1397d19533e8SHuawei Xie 			free_entries = (avail_idx - *res_base_idx);
1398d19533e8SHuawei Xie 			if (count <= free_entries)
1399d19533e8SHuawei Xie 				break;
1400d19533e8SHuawei Xie 		}
1401d19533e8SHuawei Xie 	}
1402d19533e8SHuawei Xie 
1403d19533e8SHuawei Xie 	/*check that we have enough buffers*/
1404d19533e8SHuawei Xie 	if (unlikely(count > free_entries))
1405d19533e8SHuawei Xie 		count = free_entries;
1406d19533e8SHuawei Xie 
1407d19533e8SHuawei Xie 	if (unlikely(count == 0)) {
1408d19533e8SHuawei Xie 		LOG_DEBUG(VHOST_DATA,
1409d19533e8SHuawei Xie 			"(%"PRIu64") Fail in get_available_ring_index_zcp: "
1410d19533e8SHuawei Xie 			"avail idx: %d, res base idx:%d, free entries:%d\n",
1411d19533e8SHuawei Xie 			dev->device_fh, avail_idx,
1412d19533e8SHuawei Xie 			*res_base_idx, free_entries);
1413d19533e8SHuawei Xie 		return 0;
1414d19533e8SHuawei Xie 	}
1415d19533e8SHuawei Xie 
1416d19533e8SHuawei Xie 	vq->last_used_idx_res = *res_base_idx + count;
1417d19533e8SHuawei Xie 
1418d19533e8SHuawei Xie 	return count;
1419d19533e8SHuawei Xie }
1420d19533e8SHuawei Xie 
1421d19533e8SHuawei Xie /*
1422d19533e8SHuawei Xie  * This function put descriptor back to used list.
1423d19533e8SHuawei Xie  */
1424d19533e8SHuawei Xie static inline void __attribute__((always_inline))
1425d19533e8SHuawei Xie put_desc_to_used_list_zcp(struct vhost_virtqueue *vq, uint16_t desc_idx)
1426d19533e8SHuawei Xie {
1427d19533e8SHuawei Xie 	uint16_t res_cur_idx = vq->last_used_idx;
1428d19533e8SHuawei Xie 	vq->used->ring[res_cur_idx & (vq->size - 1)].id = (uint32_t)desc_idx;
1429d19533e8SHuawei Xie 	vq->used->ring[res_cur_idx & (vq->size - 1)].len = 0;
1430d19533e8SHuawei Xie 	rte_compiler_barrier();
1431d19533e8SHuawei Xie 	*(volatile uint16_t *)&vq->used->idx += 1;
1432d19533e8SHuawei Xie 	vq->last_used_idx += 1;
1433d19533e8SHuawei Xie 
1434d19533e8SHuawei Xie 	/* Kick the guest if necessary. */
1435d19533e8SHuawei Xie 	if (!(vq->avail->flags & VRING_AVAIL_F_NO_INTERRUPT))
14369702b2b5SYuanhan Liu 		eventfd_write(vq->callfd, (eventfd_t)1);
1437d19533e8SHuawei Xie }
1438d19533e8SHuawei Xie 
1439d19533e8SHuawei Xie /*
1440d19533e8SHuawei Xie  * This function get available descriptor from vitio vring and un-attached mbuf
1441d19533e8SHuawei Xie  * from vpool->ring, and then attach them together. It needs adjust the offset
1442d19533e8SHuawei Xie  * for buff_addr and phys_addr accroding to PMD implementation, otherwise the
1443d19533e8SHuawei Xie  * frame data may be put to wrong location in mbuf.
1444d19533e8SHuawei Xie  */
1445d19533e8SHuawei Xie static inline void __attribute__((always_inline))
1446d19533e8SHuawei Xie attach_rxmbuf_zcp(struct virtio_net *dev)
1447d19533e8SHuawei Xie {
1448d19533e8SHuawei Xie 	uint16_t res_base_idx, desc_idx;
1449d19533e8SHuawei Xie 	uint64_t buff_addr, phys_addr;
1450d19533e8SHuawei Xie 	struct vhost_virtqueue *vq;
1451d19533e8SHuawei Xie 	struct vring_desc *desc;
1452d19533e8SHuawei Xie 	struct rte_mbuf *mbuf = NULL;
1453d19533e8SHuawei Xie 	struct vpool *vpool;
1454d19533e8SHuawei Xie 	hpa_type addr_type;
1455e571e6b4SHuawei Xie 	struct vhost_dev *vdev = (struct vhost_dev *)dev->priv;
1456d19533e8SHuawei Xie 
1457e571e6b4SHuawei Xie 	vpool = &vpool_array[vdev->vmdq_rx_q];
1458d19533e8SHuawei Xie 	vq = dev->virtqueue[VIRTIO_RXQ];
1459d19533e8SHuawei Xie 
1460d19533e8SHuawei Xie 	do {
1461e571e6b4SHuawei Xie 		if (unlikely(get_available_ring_index_zcp(vdev->dev, &res_base_idx,
1462d19533e8SHuawei Xie 				1) != 1))
1463d19533e8SHuawei Xie 			return;
1464d19533e8SHuawei Xie 		desc_idx = vq->avail->ring[(res_base_idx) & (vq->size - 1)];
1465d19533e8SHuawei Xie 
1466d19533e8SHuawei Xie 		desc = &vq->desc[desc_idx];
1467d19533e8SHuawei Xie 		if (desc->flags & VRING_DESC_F_NEXT) {
1468d19533e8SHuawei Xie 			desc = &vq->desc[desc->next];
1469d19533e8SHuawei Xie 			buff_addr = gpa_to_vva(dev, desc->addr);
1470e571e6b4SHuawei Xie 			phys_addr = gpa_to_hpa(vdev, desc->addr, desc->len,
1471d19533e8SHuawei Xie 					&addr_type);
1472d19533e8SHuawei Xie 		} else {
1473d19533e8SHuawei Xie 			buff_addr = gpa_to_vva(dev,
1474d19533e8SHuawei Xie 					desc->addr + vq->vhost_hlen);
1475e571e6b4SHuawei Xie 			phys_addr = gpa_to_hpa(vdev,
1476d19533e8SHuawei Xie 					desc->addr + vq->vhost_hlen,
1477d19533e8SHuawei Xie 					desc->len, &addr_type);
1478d19533e8SHuawei Xie 		}
1479d19533e8SHuawei Xie 
1480d19533e8SHuawei Xie 		if (unlikely(addr_type == PHYS_ADDR_INVALID)) {
1481d19533e8SHuawei Xie 			RTE_LOG(ERR, VHOST_DATA, "(%"PRIu64") Invalid frame buffer"
1482d19533e8SHuawei Xie 				" address found when attaching RX frame buffer"
1483d19533e8SHuawei Xie 				" address!\n", dev->device_fh);
1484d19533e8SHuawei Xie 			put_desc_to_used_list_zcp(vq, desc_idx);
1485d19533e8SHuawei Xie 			continue;
1486d19533e8SHuawei Xie 		}
1487d19533e8SHuawei Xie 
1488d19533e8SHuawei Xie 		/*
1489d19533e8SHuawei Xie 		 * Check if the frame buffer address from guest crosses
1490d19533e8SHuawei Xie 		 * sub-region or not.
1491d19533e8SHuawei Xie 		 */
1492d19533e8SHuawei Xie 		if (unlikely(addr_type == PHYS_ADDR_CROSS_SUBREG)) {
1493d19533e8SHuawei Xie 			RTE_LOG(ERR, VHOST_DATA,
1494d19533e8SHuawei Xie 				"(%"PRIu64") Frame buffer address cross "
1495d19533e8SHuawei Xie 				"sub-regioin found when attaching RX frame "
1496d19533e8SHuawei Xie 				"buffer address!\n",
1497d19533e8SHuawei Xie 				dev->device_fh);
1498d19533e8SHuawei Xie 			put_desc_to_used_list_zcp(vq, desc_idx);
1499d19533e8SHuawei Xie 			continue;
1500d19533e8SHuawei Xie 		}
1501d19533e8SHuawei Xie 	} while (unlikely(phys_addr == 0));
1502d19533e8SHuawei Xie 
1503d19533e8SHuawei Xie 	rte_ring_sc_dequeue(vpool->ring, (void **)&mbuf);
1504d19533e8SHuawei Xie 	if (unlikely(mbuf == NULL)) {
1505d19533e8SHuawei Xie 		LOG_DEBUG(VHOST_DATA,
1506d19533e8SHuawei Xie 			"(%"PRIu64") in attach_rxmbuf_zcp: "
1507d19533e8SHuawei Xie 			"ring_sc_dequeue fail.\n",
1508d19533e8SHuawei Xie 			dev->device_fh);
1509d19533e8SHuawei Xie 		put_desc_to_used_list_zcp(vq, desc_idx);
1510d19533e8SHuawei Xie 		return;
1511d19533e8SHuawei Xie 	}
1512d19533e8SHuawei Xie 
1513d19533e8SHuawei Xie 	if (unlikely(vpool->buf_size > desc->len)) {
1514d19533e8SHuawei Xie 		LOG_DEBUG(VHOST_DATA,
1515d19533e8SHuawei Xie 			"(%"PRIu64") in attach_rxmbuf_zcp: frame buffer "
1516d19533e8SHuawei Xie 			"length(%d) of descriptor idx: %d less than room "
1517d19533e8SHuawei Xie 			"size required: %d\n",
1518d19533e8SHuawei Xie 			dev->device_fh, desc->len, desc_idx, vpool->buf_size);
1519d19533e8SHuawei Xie 		put_desc_to_used_list_zcp(vq, desc_idx);
1520d19533e8SHuawei Xie 		rte_ring_sp_enqueue(vpool->ring, (void *)mbuf);
1521d19533e8SHuawei Xie 		return;
1522d19533e8SHuawei Xie 	}
1523d19533e8SHuawei Xie 
1524d19533e8SHuawei Xie 	mbuf->buf_addr = (void *)(uintptr_t)(buff_addr - RTE_PKTMBUF_HEADROOM);
1525d19533e8SHuawei Xie 	mbuf->data_off = RTE_PKTMBUF_HEADROOM;
1526d19533e8SHuawei Xie 	mbuf->buf_physaddr = phys_addr - RTE_PKTMBUF_HEADROOM;
1527d19533e8SHuawei Xie 	mbuf->data_len = desc->len;
1528d19533e8SHuawei Xie 	MBUF_HEADROOM_UINT32(mbuf) = (uint32_t)desc_idx;
1529d19533e8SHuawei Xie 
1530d19533e8SHuawei Xie 	LOG_DEBUG(VHOST_DATA,
1531d19533e8SHuawei Xie 		"(%"PRIu64") in attach_rxmbuf_zcp: res base idx:%d, "
1532d19533e8SHuawei Xie 		"descriptor idx:%d\n",
1533d19533e8SHuawei Xie 		dev->device_fh, res_base_idx, desc_idx);
1534d19533e8SHuawei Xie 
1535d19533e8SHuawei Xie 	__rte_mbuf_raw_free(mbuf);
1536d19533e8SHuawei Xie 
1537d19533e8SHuawei Xie 	return;
1538d19533e8SHuawei Xie }
1539d19533e8SHuawei Xie 
1540d19533e8SHuawei Xie /*
1541d19533e8SHuawei Xie  * Detach an attched packet mbuf -
1542d19533e8SHuawei Xie  *  - restore original mbuf address and length values.
1543d19533e8SHuawei Xie  *  - reset pktmbuf data and data_len to their default values.
1544d19533e8SHuawei Xie  *  All other fields of the given packet mbuf will be left intact.
1545d19533e8SHuawei Xie  *
1546d19533e8SHuawei Xie  * @param m
1547d19533e8SHuawei Xie  *   The attached packet mbuf.
1548d19533e8SHuawei Xie  */
1549d19533e8SHuawei Xie static inline void pktmbuf_detach_zcp(struct rte_mbuf *m)
1550d19533e8SHuawei Xie {
1551d19533e8SHuawei Xie 	const struct rte_mempool *mp = m->pool;
1552355e6735SOlivier Matz 	void *buf = rte_mbuf_to_baddr(m);
1553d19533e8SHuawei Xie 	uint32_t buf_ofs;
1554d19533e8SHuawei Xie 	uint32_t buf_len = mp->elt_size - sizeof(*m);
1555d19533e8SHuawei Xie 	m->buf_physaddr = rte_mempool_virt2phy(mp, m) + sizeof(*m);
1556d19533e8SHuawei Xie 
1557d19533e8SHuawei Xie 	m->buf_addr = buf;
1558d19533e8SHuawei Xie 	m->buf_len = (uint16_t)buf_len;
1559d19533e8SHuawei Xie 
1560d19533e8SHuawei Xie 	buf_ofs = (RTE_PKTMBUF_HEADROOM <= m->buf_len) ?
1561d19533e8SHuawei Xie 			RTE_PKTMBUF_HEADROOM : m->buf_len;
1562d19533e8SHuawei Xie 	m->data_off = buf_ofs;
1563d19533e8SHuawei Xie 
1564d19533e8SHuawei Xie 	m->data_len = 0;
1565d19533e8SHuawei Xie }
1566d19533e8SHuawei Xie 
1567d19533e8SHuawei Xie /*
1568d19533e8SHuawei Xie  * This function is called after packets have been transimited. It fetchs mbuf
1569d19533e8SHuawei Xie  * from vpool->pool, detached it and put into vpool->ring. It also update the
1570d19533e8SHuawei Xie  * used index and kick the guest if necessary.
1571d19533e8SHuawei Xie  */
1572d19533e8SHuawei Xie static inline uint32_t __attribute__((always_inline))
1573d19533e8SHuawei Xie txmbuf_clean_zcp(struct virtio_net *dev, struct vpool *vpool)
1574d19533e8SHuawei Xie {
1575d19533e8SHuawei Xie 	struct rte_mbuf *mbuf;
1576d19533e8SHuawei Xie 	struct vhost_virtqueue *vq = dev->virtqueue[VIRTIO_TXQ];
1577d19533e8SHuawei Xie 	uint32_t used_idx = vq->last_used_idx & (vq->size - 1);
1578d19533e8SHuawei Xie 	uint32_t index = 0;
1579d19533e8SHuawei Xie 	uint32_t mbuf_count = rte_mempool_count(vpool->pool);
1580d19533e8SHuawei Xie 
1581d19533e8SHuawei Xie 	LOG_DEBUG(VHOST_DATA,
1582d19533e8SHuawei Xie 		"(%"PRIu64") in txmbuf_clean_zcp: mbuf count in mempool before "
1583d19533e8SHuawei Xie 		"clean is: %d\n",
1584d19533e8SHuawei Xie 		dev->device_fh, mbuf_count);
1585d19533e8SHuawei Xie 	LOG_DEBUG(VHOST_DATA,
1586d19533e8SHuawei Xie 		"(%"PRIu64") in txmbuf_clean_zcp: mbuf count in  ring before "
1587d19533e8SHuawei Xie 		"clean  is : %d\n",
1588d19533e8SHuawei Xie 		dev->device_fh, rte_ring_count(vpool->ring));
1589d19533e8SHuawei Xie 
1590d19533e8SHuawei Xie 	for (index = 0; index < mbuf_count; index++) {
1591d19533e8SHuawei Xie 		mbuf = __rte_mbuf_raw_alloc(vpool->pool);
1592e8b9ef87SSergio Gonzalez Monroy 		if (likely(MBUF_EXT_MEM(mbuf)))
1593d19533e8SHuawei Xie 			pktmbuf_detach_zcp(mbuf);
1594d19533e8SHuawei Xie 		rte_ring_sp_enqueue(vpool->ring, mbuf);
1595d19533e8SHuawei Xie 
1596d19533e8SHuawei Xie 		/* Update used index buffer information. */
1597d19533e8SHuawei Xie 		vq->used->ring[used_idx].id = MBUF_HEADROOM_UINT32(mbuf);
1598d19533e8SHuawei Xie 		vq->used->ring[used_idx].len = 0;
1599d19533e8SHuawei Xie 
1600d19533e8SHuawei Xie 		used_idx = (used_idx + 1) & (vq->size - 1);
1601d19533e8SHuawei Xie 	}
1602d19533e8SHuawei Xie 
1603d19533e8SHuawei Xie 	LOG_DEBUG(VHOST_DATA,
1604d19533e8SHuawei Xie 		"(%"PRIu64") in txmbuf_clean_zcp: mbuf count in mempool after "
1605d19533e8SHuawei Xie 		"clean is: %d\n",
1606d19533e8SHuawei Xie 		dev->device_fh, rte_mempool_count(vpool->pool));
1607d19533e8SHuawei Xie 	LOG_DEBUG(VHOST_DATA,
1608d19533e8SHuawei Xie 		"(%"PRIu64") in txmbuf_clean_zcp: mbuf count in  ring after "
1609d19533e8SHuawei Xie 		"clean  is : %d\n",
1610d19533e8SHuawei Xie 		dev->device_fh, rte_ring_count(vpool->ring));
1611d19533e8SHuawei Xie 	LOG_DEBUG(VHOST_DATA,
1612d19533e8SHuawei Xie 		"(%"PRIu64") in txmbuf_clean_zcp: before updated "
1613d19533e8SHuawei Xie 		"vq->last_used_idx:%d\n",
1614d19533e8SHuawei Xie 		dev->device_fh, vq->last_used_idx);
1615d19533e8SHuawei Xie 
1616d19533e8SHuawei Xie 	vq->last_used_idx += mbuf_count;
1617d19533e8SHuawei Xie 
1618d19533e8SHuawei Xie 	LOG_DEBUG(VHOST_DATA,
1619d19533e8SHuawei Xie 		"(%"PRIu64") in txmbuf_clean_zcp: after updated "
1620d19533e8SHuawei Xie 		"vq->last_used_idx:%d\n",
1621d19533e8SHuawei Xie 		dev->device_fh, vq->last_used_idx);
1622d19533e8SHuawei Xie 
1623d19533e8SHuawei Xie 	rte_compiler_barrier();
1624d19533e8SHuawei Xie 
1625d19533e8SHuawei Xie 	*(volatile uint16_t *)&vq->used->idx += mbuf_count;
1626d19533e8SHuawei Xie 
1627d19533e8SHuawei Xie 	/* Kick guest if required. */
1628d19533e8SHuawei Xie 	if (!(vq->avail->flags & VRING_AVAIL_F_NO_INTERRUPT))
16299702b2b5SYuanhan Liu 		eventfd_write(vq->callfd, (eventfd_t)1);
1630d19533e8SHuawei Xie 
1631d19533e8SHuawei Xie 	return 0;
1632d19533e8SHuawei Xie }
1633d19533e8SHuawei Xie 
1634d19533e8SHuawei Xie /*
1635d19533e8SHuawei Xie  * This function is called when a virtio device is destroy.
1636d19533e8SHuawei Xie  * It fetchs mbuf from vpool->pool, and detached it, and put into vpool->ring.
1637d19533e8SHuawei Xie  */
1638d19533e8SHuawei Xie static void mbuf_destroy_zcp(struct vpool *vpool)
1639d19533e8SHuawei Xie {
1640d19533e8SHuawei Xie 	struct rte_mbuf *mbuf = NULL;
1641d19533e8SHuawei Xie 	uint32_t index, mbuf_count = rte_mempool_count(vpool->pool);
1642d19533e8SHuawei Xie 
1643d19533e8SHuawei Xie 	LOG_DEBUG(VHOST_CONFIG,
1644d19533e8SHuawei Xie 		"in mbuf_destroy_zcp: mbuf count in mempool before "
1645d19533e8SHuawei Xie 		"mbuf_destroy_zcp is: %d\n",
1646d19533e8SHuawei Xie 		mbuf_count);
1647d19533e8SHuawei Xie 	LOG_DEBUG(VHOST_CONFIG,
1648d19533e8SHuawei Xie 		"in mbuf_destroy_zcp: mbuf count in  ring before "
1649d19533e8SHuawei Xie 		"mbuf_destroy_zcp  is : %d\n",
1650d19533e8SHuawei Xie 		rte_ring_count(vpool->ring));
1651d19533e8SHuawei Xie 
1652d19533e8SHuawei Xie 	for (index = 0; index < mbuf_count; index++) {
1653d19533e8SHuawei Xie 		mbuf = __rte_mbuf_raw_alloc(vpool->pool);
1654d19533e8SHuawei Xie 		if (likely(mbuf != NULL)) {
1655e8b9ef87SSergio Gonzalez Monroy 			if (likely(MBUF_EXT_MEM(mbuf)))
1656d19533e8SHuawei Xie 				pktmbuf_detach_zcp(mbuf);
1657d19533e8SHuawei Xie 			rte_ring_sp_enqueue(vpool->ring, (void *)mbuf);
1658d19533e8SHuawei Xie 		}
1659d19533e8SHuawei Xie 	}
1660d19533e8SHuawei Xie 
1661d19533e8SHuawei Xie 	LOG_DEBUG(VHOST_CONFIG,
1662d19533e8SHuawei Xie 		"in mbuf_destroy_zcp: mbuf count in mempool after "
1663d19533e8SHuawei Xie 		"mbuf_destroy_zcp is: %d\n",
1664d19533e8SHuawei Xie 		rte_mempool_count(vpool->pool));
1665d19533e8SHuawei Xie 	LOG_DEBUG(VHOST_CONFIG,
1666d19533e8SHuawei Xie 		"in mbuf_destroy_zcp: mbuf count in ring after "
1667d19533e8SHuawei Xie 		"mbuf_destroy_zcp is : %d\n",
1668d19533e8SHuawei Xie 		rte_ring_count(vpool->ring));
1669d19533e8SHuawei Xie }
1670d19533e8SHuawei Xie 
1671d19533e8SHuawei Xie /*
1672d19533e8SHuawei Xie  * This function update the use flag and counter.
1673d19533e8SHuawei Xie  */
1674d19533e8SHuawei Xie static inline uint32_t __attribute__((always_inline))
1675d19533e8SHuawei Xie virtio_dev_rx_zcp(struct virtio_net *dev, struct rte_mbuf **pkts,
1676d19533e8SHuawei Xie 	uint32_t count)
1677d19533e8SHuawei Xie {
1678d19533e8SHuawei Xie 	struct vhost_virtqueue *vq;
1679d19533e8SHuawei Xie 	struct vring_desc *desc;
1680d19533e8SHuawei Xie 	struct rte_mbuf *buff;
1681d19533e8SHuawei Xie 	/* The virtio_hdr is initialised to 0. */
1682d19533e8SHuawei Xie 	struct virtio_net_hdr_mrg_rxbuf virtio_hdr
1683d19533e8SHuawei Xie 		= {{0, 0, 0, 0, 0, 0}, 0};
1684d19533e8SHuawei Xie 	uint64_t buff_hdr_addr = 0;
1685d19533e8SHuawei Xie 	uint32_t head[MAX_PKT_BURST], packet_len = 0;
1686d19533e8SHuawei Xie 	uint32_t head_idx, packet_success = 0;
1687d19533e8SHuawei Xie 	uint16_t res_cur_idx;
1688d19533e8SHuawei Xie 
1689d19533e8SHuawei Xie 	LOG_DEBUG(VHOST_DATA, "(%"PRIu64") virtio_dev_rx()\n", dev->device_fh);
1690d19533e8SHuawei Xie 
1691d19533e8SHuawei Xie 	if (count == 0)
1692d19533e8SHuawei Xie 		return 0;
1693d19533e8SHuawei Xie 
1694d19533e8SHuawei Xie 	vq = dev->virtqueue[VIRTIO_RXQ];
1695d19533e8SHuawei Xie 	count = (count > MAX_PKT_BURST) ? MAX_PKT_BURST : count;
1696d19533e8SHuawei Xie 
1697d19533e8SHuawei Xie 	res_cur_idx = vq->last_used_idx;
1698d19533e8SHuawei Xie 	LOG_DEBUG(VHOST_DATA, "(%"PRIu64") Current Index %d| End Index %d\n",
1699d19533e8SHuawei Xie 		dev->device_fh, res_cur_idx, res_cur_idx + count);
1700d19533e8SHuawei Xie 
1701d19533e8SHuawei Xie 	/* Retrieve all of the head indexes first to avoid caching issues. */
1702d19533e8SHuawei Xie 	for (head_idx = 0; head_idx < count; head_idx++)
1703d19533e8SHuawei Xie 		head[head_idx] = MBUF_HEADROOM_UINT32(pkts[head_idx]);
1704d19533e8SHuawei Xie 
1705d19533e8SHuawei Xie 	/*Prefetch descriptor index. */
1706d19533e8SHuawei Xie 	rte_prefetch0(&vq->desc[head[packet_success]]);
1707d19533e8SHuawei Xie 
1708d19533e8SHuawei Xie 	while (packet_success != count) {
1709d19533e8SHuawei Xie 		/* Get descriptor from available ring */
1710d19533e8SHuawei Xie 		desc = &vq->desc[head[packet_success]];
1711d19533e8SHuawei Xie 
1712d19533e8SHuawei Xie 		buff = pkts[packet_success];
1713d19533e8SHuawei Xie 		LOG_DEBUG(VHOST_DATA,
1714d19533e8SHuawei Xie 			"(%"PRIu64") in dev_rx_zcp: update the used idx for "
1715d19533e8SHuawei Xie 			"pkt[%d] descriptor idx: %d\n",
1716d19533e8SHuawei Xie 			dev->device_fh, packet_success,
1717d19533e8SHuawei Xie 			MBUF_HEADROOM_UINT32(buff));
1718d19533e8SHuawei Xie 
1719d19533e8SHuawei Xie 		PRINT_PACKET(dev,
1720d19533e8SHuawei Xie 			(uintptr_t)(((uint64_t)(uintptr_t)buff->buf_addr)
1721d19533e8SHuawei Xie 			+ RTE_PKTMBUF_HEADROOM),
1722d19533e8SHuawei Xie 			rte_pktmbuf_data_len(buff), 0);
1723d19533e8SHuawei Xie 
1724d19533e8SHuawei Xie 		/* Buffer address translation for virtio header. */
1725d19533e8SHuawei Xie 		buff_hdr_addr = gpa_to_vva(dev, desc->addr);
1726d19533e8SHuawei Xie 		packet_len = rte_pktmbuf_data_len(buff) + vq->vhost_hlen;
1727d19533e8SHuawei Xie 
1728d19533e8SHuawei Xie 		/*
1729d19533e8SHuawei Xie 		 * If the descriptors are chained the header and data are
1730d19533e8SHuawei Xie 		 * placed in separate buffers.
1731d19533e8SHuawei Xie 		 */
1732d19533e8SHuawei Xie 		if (desc->flags & VRING_DESC_F_NEXT) {
1733d19533e8SHuawei Xie 			desc->len = vq->vhost_hlen;
1734d19533e8SHuawei Xie 			desc = &vq->desc[desc->next];
1735d19533e8SHuawei Xie 			desc->len = rte_pktmbuf_data_len(buff);
1736d19533e8SHuawei Xie 		} else {
1737d19533e8SHuawei Xie 			desc->len = packet_len;
1738d19533e8SHuawei Xie 		}
1739d19533e8SHuawei Xie 
1740d19533e8SHuawei Xie 		/* Update used ring with desc information */
1741d19533e8SHuawei Xie 		vq->used->ring[res_cur_idx & (vq->size - 1)].id
1742d19533e8SHuawei Xie 			= head[packet_success];
1743d19533e8SHuawei Xie 		vq->used->ring[res_cur_idx & (vq->size - 1)].len
1744d19533e8SHuawei Xie 			= packet_len;
1745d19533e8SHuawei Xie 		res_cur_idx++;
1746d19533e8SHuawei Xie 		packet_success++;
1747d19533e8SHuawei Xie 
1748d19533e8SHuawei Xie 		/* A header is required per buffer. */
1749d19533e8SHuawei Xie 		rte_memcpy((void *)(uintptr_t)buff_hdr_addr,
1750d19533e8SHuawei Xie 			(const void *)&virtio_hdr, vq->vhost_hlen);
1751d19533e8SHuawei Xie 
1752d19533e8SHuawei Xie 		PRINT_PACKET(dev, (uintptr_t)buff_hdr_addr, vq->vhost_hlen, 1);
1753d19533e8SHuawei Xie 
1754d19533e8SHuawei Xie 		if (likely(packet_success < count)) {
1755d19533e8SHuawei Xie 			/* Prefetch descriptor index. */
1756d19533e8SHuawei Xie 			rte_prefetch0(&vq->desc[head[packet_success]]);
1757d19533e8SHuawei Xie 		}
1758d19533e8SHuawei Xie 	}
1759d19533e8SHuawei Xie 
1760d19533e8SHuawei Xie 	rte_compiler_barrier();
1761d19533e8SHuawei Xie 
1762d19533e8SHuawei Xie 	LOG_DEBUG(VHOST_DATA,
1763d19533e8SHuawei Xie 		"(%"PRIu64") in dev_rx_zcp: before update used idx: "
1764d19533e8SHuawei Xie 		"vq.last_used_idx: %d, vq->used->idx: %d\n",
1765d19533e8SHuawei Xie 		dev->device_fh, vq->last_used_idx, vq->used->idx);
1766d19533e8SHuawei Xie 
1767d19533e8SHuawei Xie 	*(volatile uint16_t *)&vq->used->idx += count;
1768d19533e8SHuawei Xie 	vq->last_used_idx += count;
1769d19533e8SHuawei Xie 
1770d19533e8SHuawei Xie 	LOG_DEBUG(VHOST_DATA,
1771d19533e8SHuawei Xie 		"(%"PRIu64") in dev_rx_zcp: after  update used idx: "
1772d19533e8SHuawei Xie 		"vq.last_used_idx: %d, vq->used->idx: %d\n",
1773d19533e8SHuawei Xie 		dev->device_fh, vq->last_used_idx, vq->used->idx);
1774d19533e8SHuawei Xie 
1775d19533e8SHuawei Xie 	/* Kick the guest if necessary. */
1776d19533e8SHuawei Xie 	if (!(vq->avail->flags & VRING_AVAIL_F_NO_INTERRUPT))
17779702b2b5SYuanhan Liu 		eventfd_write(vq->callfd, (eventfd_t)1);
1778d19533e8SHuawei Xie 
1779d19533e8SHuawei Xie 	return count;
1780d19533e8SHuawei Xie }
1781d19533e8SHuawei Xie 
1782d19533e8SHuawei Xie /*
1783d19533e8SHuawei Xie  * This function routes the TX packet to the correct interface.
1784d19533e8SHuawei Xie  * This may be a local device or the physical port.
1785d19533e8SHuawei Xie  */
1786d19533e8SHuawei Xie static inline void __attribute__((always_inline))
1787d19533e8SHuawei Xie virtio_tx_route_zcp(struct virtio_net *dev, struct rte_mbuf *m,
1788d19533e8SHuawei Xie 	uint32_t desc_idx, uint8_t need_copy)
1789d19533e8SHuawei Xie {
1790d19533e8SHuawei Xie 	struct mbuf_table *tx_q;
1791d19533e8SHuawei Xie 	struct rte_mbuf **m_table;
1792d19533e8SHuawei Xie 	struct rte_mbuf *mbuf = NULL;
1793d19533e8SHuawei Xie 	unsigned len, ret, offset = 0;
1794d19533e8SHuawei Xie 	struct vpool *vpool;
1795d19533e8SHuawei Xie 	uint16_t vlan_tag = (uint16_t)vlan_tags[(uint16_t)dev->device_fh];
1796e571e6b4SHuawei Xie 	uint16_t vmdq_rx_q = ((struct vhost_dev *)dev->priv)->vmdq_rx_q;
1797d19533e8SHuawei Xie 
1798d19533e8SHuawei Xie 	/*Add packet to the port tx queue*/
1799e571e6b4SHuawei Xie 	tx_q = &tx_queue_zcp[vmdq_rx_q];
1800d19533e8SHuawei Xie 	len = tx_q->len;
1801d19533e8SHuawei Xie 
1802d19533e8SHuawei Xie 	/* Allocate an mbuf and populate the structure. */
1803e571e6b4SHuawei Xie 	vpool = &vpool_array[MAX_QUEUES + vmdq_rx_q];
1804d19533e8SHuawei Xie 	rte_ring_sc_dequeue(vpool->ring, (void **)&mbuf);
1805d19533e8SHuawei Xie 	if (unlikely(mbuf == NULL)) {
1806d19533e8SHuawei Xie 		struct vhost_virtqueue *vq = dev->virtqueue[VIRTIO_TXQ];
1807d19533e8SHuawei Xie 		RTE_LOG(ERR, VHOST_DATA,
1808d19533e8SHuawei Xie 			"(%"PRIu64") Failed to allocate memory for mbuf.\n",
1809d19533e8SHuawei Xie 			dev->device_fh);
1810d19533e8SHuawei Xie 		put_desc_to_used_list_zcp(vq, desc_idx);
1811d19533e8SHuawei Xie 		return;
1812d19533e8SHuawei Xie 	}
1813d19533e8SHuawei Xie 
1814d19533e8SHuawei Xie 	if (vm2vm_mode == VM2VM_HARDWARE) {
1815d19533e8SHuawei Xie 		/* Avoid using a vlan tag from any vm for external pkt, such as
1816d19533e8SHuawei Xie 		 * vlan_tags[dev->device_fh], oterwise, it conflicts when pool
1817d19533e8SHuawei Xie 		 * selection, MAC address determines it as an external pkt
1818d19533e8SHuawei Xie 		 * which should go to network, while vlan tag determine it as
1819d19533e8SHuawei Xie 		 * a vm2vm pkt should forward to another vm. Hardware confuse
1820d19533e8SHuawei Xie 		 * such a ambiguous situation, so pkt will lost.
1821d19533e8SHuawei Xie 		 */
1822d19533e8SHuawei Xie 		vlan_tag = external_pkt_default_vlan_tag;
182372ec8d77SOuyang Changchun 		if (find_local_dest(dev, m, &offset, &vlan_tag) != 0) {
182472ec8d77SOuyang Changchun 			MBUF_HEADROOM_UINT32(mbuf) = (uint32_t)desc_idx;
1825d19533e8SHuawei Xie 			__rte_mbuf_raw_free(mbuf);
1826d19533e8SHuawei Xie 			return;
1827d19533e8SHuawei Xie 		}
1828d19533e8SHuawei Xie 	}
1829d19533e8SHuawei Xie 
1830d19533e8SHuawei Xie 	mbuf->nb_segs = m->nb_segs;
1831d19533e8SHuawei Xie 	mbuf->next = m->next;
1832d19533e8SHuawei Xie 	mbuf->data_len = m->data_len + offset;
1833d19533e8SHuawei Xie 	mbuf->pkt_len = mbuf->data_len;
1834d19533e8SHuawei Xie 	if (unlikely(need_copy)) {
1835d19533e8SHuawei Xie 		/* Copy the packet contents to the mbuf. */
1836d19533e8SHuawei Xie 		rte_memcpy(rte_pktmbuf_mtod(mbuf, void *),
1837d19533e8SHuawei Xie 			rte_pktmbuf_mtod(m, void *),
1838d19533e8SHuawei Xie 			m->data_len);
1839d19533e8SHuawei Xie 	} else {
1840d19533e8SHuawei Xie 		mbuf->data_off = m->data_off;
1841d19533e8SHuawei Xie 		mbuf->buf_physaddr = m->buf_physaddr;
1842d19533e8SHuawei Xie 		mbuf->buf_addr = m->buf_addr;
1843d19533e8SHuawei Xie 	}
1844d19533e8SHuawei Xie 	mbuf->ol_flags = PKT_TX_VLAN_PKT;
1845d19533e8SHuawei Xie 	mbuf->vlan_tci = vlan_tag;
1846d19533e8SHuawei Xie 	mbuf->l2_len = sizeof(struct ether_hdr);
1847d19533e8SHuawei Xie 	mbuf->l3_len = sizeof(struct ipv4_hdr);
1848d19533e8SHuawei Xie 	MBUF_HEADROOM_UINT32(mbuf) = (uint32_t)desc_idx;
1849d19533e8SHuawei Xie 
1850d19533e8SHuawei Xie 	tx_q->m_table[len] = mbuf;
1851d19533e8SHuawei Xie 	len++;
1852d19533e8SHuawei Xie 
1853d19533e8SHuawei Xie 	LOG_DEBUG(VHOST_DATA,
1854d19533e8SHuawei Xie 		"(%"PRIu64") in tx_route_zcp: pkt: nb_seg: %d, next:%s\n",
1855d19533e8SHuawei Xie 		dev->device_fh,
1856d19533e8SHuawei Xie 		mbuf->nb_segs,
1857d19533e8SHuawei Xie 		(mbuf->next == NULL) ? "null" : "non-null");
1858d19533e8SHuawei Xie 
1859d19533e8SHuawei Xie 	if (enable_stats) {
1860d19533e8SHuawei Xie 		dev_statistics[dev->device_fh].tx_total++;
1861d19533e8SHuawei Xie 		dev_statistics[dev->device_fh].tx++;
1862d19533e8SHuawei Xie 	}
1863d19533e8SHuawei Xie 
1864d19533e8SHuawei Xie 	if (unlikely(len == MAX_PKT_BURST)) {
1865d19533e8SHuawei Xie 		m_table = (struct rte_mbuf **)tx_q->m_table;
1866d19533e8SHuawei Xie 		ret = rte_eth_tx_burst(ports[0],
1867d19533e8SHuawei Xie 			(uint16_t)tx_q->txq_id, m_table, (uint16_t) len);
1868d19533e8SHuawei Xie 
1869d19533e8SHuawei Xie 		/*
1870d19533e8SHuawei Xie 		 * Free any buffers not handled by TX and update
1871d19533e8SHuawei Xie 		 * the port stats.
1872d19533e8SHuawei Xie 		 */
1873d19533e8SHuawei Xie 		if (unlikely(ret < len)) {
1874d19533e8SHuawei Xie 			do {
1875d19533e8SHuawei Xie 				rte_pktmbuf_free(m_table[ret]);
1876d19533e8SHuawei Xie 			} while (++ret < len);
1877d19533e8SHuawei Xie 		}
1878d19533e8SHuawei Xie 
1879d19533e8SHuawei Xie 		len = 0;
1880d19533e8SHuawei Xie 		txmbuf_clean_zcp(dev, vpool);
1881d19533e8SHuawei Xie 	}
1882d19533e8SHuawei Xie 
1883d19533e8SHuawei Xie 	tx_q->len = len;
1884d19533e8SHuawei Xie 
1885d19533e8SHuawei Xie 	return;
1886d19533e8SHuawei Xie }
1887d19533e8SHuawei Xie 
1888d19533e8SHuawei Xie /*
1889d19533e8SHuawei Xie  * This function TX all available packets in virtio TX queue for one
1890d19533e8SHuawei Xie  * virtio-net device. If it is first packet, it learns MAC address and
1891d19533e8SHuawei Xie  * setup VMDQ.
1892d19533e8SHuawei Xie  */
1893d19533e8SHuawei Xie static inline void __attribute__((always_inline))
1894d19533e8SHuawei Xie virtio_dev_tx_zcp(struct virtio_net *dev)
1895d19533e8SHuawei Xie {
1896d19533e8SHuawei Xie 	struct rte_mbuf m;
1897d19533e8SHuawei Xie 	struct vhost_virtqueue *vq;
1898d19533e8SHuawei Xie 	struct vring_desc *desc;
1899d19533e8SHuawei Xie 	uint64_t buff_addr = 0, phys_addr;
1900d19533e8SHuawei Xie 	uint32_t head[MAX_PKT_BURST];
1901d19533e8SHuawei Xie 	uint32_t i;
1902d19533e8SHuawei Xie 	uint16_t free_entries, packet_success = 0;
1903d19533e8SHuawei Xie 	uint16_t avail_idx;
1904d19533e8SHuawei Xie 	uint8_t need_copy = 0;
1905d19533e8SHuawei Xie 	hpa_type addr_type;
1906e571e6b4SHuawei Xie 	struct vhost_dev *vdev = (struct vhost_dev *)dev->priv;
1907d19533e8SHuawei Xie 
1908d19533e8SHuawei Xie 	vq = dev->virtqueue[VIRTIO_TXQ];
1909d19533e8SHuawei Xie 	avail_idx =  *((volatile uint16_t *)&vq->avail->idx);
1910d19533e8SHuawei Xie 
1911d19533e8SHuawei Xie 	/* If there are no available buffers then return. */
1912d19533e8SHuawei Xie 	if (vq->last_used_idx_res == avail_idx)
1913d19533e8SHuawei Xie 		return;
1914d19533e8SHuawei Xie 
1915d19533e8SHuawei Xie 	LOG_DEBUG(VHOST_DATA, "(%"PRIu64") virtio_dev_tx()\n", dev->device_fh);
1916d19533e8SHuawei Xie 
1917d19533e8SHuawei Xie 	/* Prefetch available ring to retrieve head indexes. */
1918d19533e8SHuawei Xie 	rte_prefetch0(&vq->avail->ring[vq->last_used_idx_res & (vq->size - 1)]);
1919d19533e8SHuawei Xie 
1920d19533e8SHuawei Xie 	/* Get the number of free entries in the ring */
1921d19533e8SHuawei Xie 	free_entries = (avail_idx - vq->last_used_idx_res);
1922d19533e8SHuawei Xie 
1923d19533e8SHuawei Xie 	/* Limit to MAX_PKT_BURST. */
1924d19533e8SHuawei Xie 	free_entries
1925d19533e8SHuawei Xie 		= (free_entries > MAX_PKT_BURST) ? MAX_PKT_BURST : free_entries;
1926d19533e8SHuawei Xie 
1927d19533e8SHuawei Xie 	LOG_DEBUG(VHOST_DATA, "(%"PRIu64") Buffers available %d\n",
1928d19533e8SHuawei Xie 		dev->device_fh, free_entries);
1929d19533e8SHuawei Xie 
1930d19533e8SHuawei Xie 	/* Retrieve all of the head indexes first to avoid caching issues. */
1931d19533e8SHuawei Xie 	for (i = 0; i < free_entries; i++)
1932d19533e8SHuawei Xie 		head[i]
1933d19533e8SHuawei Xie 			= vq->avail->ring[(vq->last_used_idx_res + i)
1934d19533e8SHuawei Xie 			& (vq->size - 1)];
1935d19533e8SHuawei Xie 
1936d19533e8SHuawei Xie 	vq->last_used_idx_res += free_entries;
1937d19533e8SHuawei Xie 
1938d19533e8SHuawei Xie 	/* Prefetch descriptor index. */
1939d19533e8SHuawei Xie 	rte_prefetch0(&vq->desc[head[packet_success]]);
1940d19533e8SHuawei Xie 	rte_prefetch0(&vq->used->ring[vq->last_used_idx & (vq->size - 1)]);
1941d19533e8SHuawei Xie 
1942d19533e8SHuawei Xie 	while (packet_success < free_entries) {
1943d19533e8SHuawei Xie 		desc = &vq->desc[head[packet_success]];
1944d19533e8SHuawei Xie 
1945d19533e8SHuawei Xie 		/* Discard first buffer as it is the virtio header */
1946d19533e8SHuawei Xie 		desc = &vq->desc[desc->next];
1947d19533e8SHuawei Xie 
1948d19533e8SHuawei Xie 		/* Buffer address translation. */
1949d19533e8SHuawei Xie 		buff_addr = gpa_to_vva(dev, desc->addr);
19506630bc42SOuyang Changchun 		/* Need check extra VLAN_HLEN size for inserting VLAN tag */
19516630bc42SOuyang Changchun 		phys_addr = gpa_to_hpa(vdev, desc->addr, desc->len + VLAN_HLEN,
19526630bc42SOuyang Changchun 			&addr_type);
1953d19533e8SHuawei Xie 
1954d19533e8SHuawei Xie 		if (likely(packet_success < (free_entries - 1)))
1955d19533e8SHuawei Xie 			/* Prefetch descriptor index. */
1956d19533e8SHuawei Xie 			rte_prefetch0(&vq->desc[head[packet_success + 1]]);
1957d19533e8SHuawei Xie 
1958d19533e8SHuawei Xie 		if (unlikely(addr_type == PHYS_ADDR_INVALID)) {
1959d19533e8SHuawei Xie 			RTE_LOG(ERR, VHOST_DATA,
1960d19533e8SHuawei Xie 				"(%"PRIu64") Invalid frame buffer address found"
1961d19533e8SHuawei Xie 				"when TX packets!\n",
1962d19533e8SHuawei Xie 				dev->device_fh);
1963d19533e8SHuawei Xie 			packet_success++;
1964d19533e8SHuawei Xie 			continue;
1965d19533e8SHuawei Xie 		}
1966d19533e8SHuawei Xie 
1967d19533e8SHuawei Xie 		/* Prefetch buffer address. */
1968d19533e8SHuawei Xie 		rte_prefetch0((void *)(uintptr_t)buff_addr);
1969d19533e8SHuawei Xie 
1970d19533e8SHuawei Xie 		/*
1971d19533e8SHuawei Xie 		 * Setup dummy mbuf. This is copied to a real mbuf if
1972d19533e8SHuawei Xie 		 * transmitted out the physical port.
1973d19533e8SHuawei Xie 		 */
1974d19533e8SHuawei Xie 		m.data_len = desc->len;
1975d19533e8SHuawei Xie 		m.nb_segs = 1;
1976d19533e8SHuawei Xie 		m.next = NULL;
1977d19533e8SHuawei Xie 		m.data_off = 0;
1978d19533e8SHuawei Xie 		m.buf_addr = (void *)(uintptr_t)buff_addr;
1979d19533e8SHuawei Xie 		m.buf_physaddr = phys_addr;
1980d19533e8SHuawei Xie 
1981d19533e8SHuawei Xie 		/*
1982d19533e8SHuawei Xie 		 * Check if the frame buffer address from guest crosses
1983d19533e8SHuawei Xie 		 * sub-region or not.
1984d19533e8SHuawei Xie 		 */
1985d19533e8SHuawei Xie 		if (unlikely(addr_type == PHYS_ADDR_CROSS_SUBREG)) {
1986d19533e8SHuawei Xie 			RTE_LOG(ERR, VHOST_DATA,
1987d19533e8SHuawei Xie 				"(%"PRIu64") Frame buffer address cross "
1988d19533e8SHuawei Xie 				"sub-regioin found when attaching TX frame "
1989d19533e8SHuawei Xie 				"buffer address!\n",
1990d19533e8SHuawei Xie 				dev->device_fh);
1991d19533e8SHuawei Xie 			need_copy = 1;
1992d19533e8SHuawei Xie 		} else
1993d19533e8SHuawei Xie 			need_copy = 0;
1994d19533e8SHuawei Xie 
1995d19533e8SHuawei Xie 		PRINT_PACKET(dev, (uintptr_t)buff_addr, desc->len, 0);
1996d19533e8SHuawei Xie 
1997d19533e8SHuawei Xie 		/*
1998d19533e8SHuawei Xie 		 * If this is the first received packet we need to learn
1999d19533e8SHuawei Xie 		 * the MAC and setup VMDQ
2000d19533e8SHuawei Xie 		 */
2001e571e6b4SHuawei Xie 		if (unlikely(vdev->ready == DEVICE_MAC_LEARNING)) {
2002e571e6b4SHuawei Xie 			if (vdev->remove || (link_vmdq(vdev, &m) == -1)) {
2003d19533e8SHuawei Xie 				/*
2004d19533e8SHuawei Xie 				 * Discard frame if device is scheduled for
2005d19533e8SHuawei Xie 				 * removal or a duplicate MAC address is found.
2006d19533e8SHuawei Xie 				 */
2007d19533e8SHuawei Xie 				packet_success += free_entries;
2008d19533e8SHuawei Xie 				vq->last_used_idx += packet_success;
2009d19533e8SHuawei Xie 				break;
2010d19533e8SHuawei Xie 			}
2011d19533e8SHuawei Xie 		}
2012d19533e8SHuawei Xie 
2013d19533e8SHuawei Xie 		virtio_tx_route_zcp(dev, &m, head[packet_success], need_copy);
2014d19533e8SHuawei Xie 		packet_success++;
2015d19533e8SHuawei Xie 	}
2016d19533e8SHuawei Xie }
2017d19533e8SHuawei Xie 
2018d19533e8SHuawei Xie /*
2019d19533e8SHuawei Xie  * This function is called by each data core. It handles all RX/TX registered
2020d19533e8SHuawei Xie  * with the core. For TX the specific lcore linked list is used. For RX, MAC
2021d19533e8SHuawei Xie  * addresses are compared with all devices in the main linked list.
2022d19533e8SHuawei Xie  */
2023d19533e8SHuawei Xie static int
2024d19533e8SHuawei Xie switch_worker_zcp(__attribute__((unused)) void *arg)
2025d19533e8SHuawei Xie {
2026d19533e8SHuawei Xie 	struct virtio_net *dev = NULL;
2027e571e6b4SHuawei Xie 	struct vhost_dev  *vdev = NULL;
2028d19533e8SHuawei Xie 	struct rte_mbuf *pkts_burst[MAX_PKT_BURST];
2029d19533e8SHuawei Xie 	struct virtio_net_data_ll *dev_ll;
2030d19533e8SHuawei Xie 	struct mbuf_table *tx_q;
2031d19533e8SHuawei Xie 	volatile struct lcore_ll_info *lcore_ll;
2032d19533e8SHuawei Xie 	const uint64_t drain_tsc
2033d19533e8SHuawei Xie 		= (rte_get_tsc_hz() + US_PER_S - 1) / US_PER_S
2034d19533e8SHuawei Xie 		* BURST_TX_DRAIN_US;
2035d19533e8SHuawei Xie 	uint64_t prev_tsc, diff_tsc, cur_tsc, ret_count = 0;
2036d19533e8SHuawei Xie 	unsigned ret;
2037d19533e8SHuawei Xie 	const uint16_t lcore_id = rte_lcore_id();
2038d19533e8SHuawei Xie 	uint16_t count_in_ring, rx_count = 0;
2039d19533e8SHuawei Xie 
2040d19533e8SHuawei Xie 	RTE_LOG(INFO, VHOST_DATA, "Procesing on Core %u started\n", lcore_id);
2041d19533e8SHuawei Xie 
2042d19533e8SHuawei Xie 	lcore_ll = lcore_info[lcore_id].lcore_ll;
2043d19533e8SHuawei Xie 	prev_tsc = 0;
2044d19533e8SHuawei Xie 
2045d19533e8SHuawei Xie 	while (1) {
2046d19533e8SHuawei Xie 		cur_tsc = rte_rdtsc();
2047d19533e8SHuawei Xie 
2048d19533e8SHuawei Xie 		/* TX burst queue drain */
2049d19533e8SHuawei Xie 		diff_tsc = cur_tsc - prev_tsc;
2050d19533e8SHuawei Xie 		if (unlikely(diff_tsc > drain_tsc)) {
2051d19533e8SHuawei Xie 			/*
2052d19533e8SHuawei Xie 			 * Get mbuf from vpool.pool and detach mbuf and
2053d19533e8SHuawei Xie 			 * put back into vpool.ring.
2054d19533e8SHuawei Xie 			 */
2055d19533e8SHuawei Xie 			dev_ll = lcore_ll->ll_root_used;
2056e571e6b4SHuawei Xie 			while ((dev_ll != NULL) && (dev_ll->vdev != NULL)) {
2057d19533e8SHuawei Xie 				/* Get virtio device ID */
2058e571e6b4SHuawei Xie 				vdev = dev_ll->vdev;
2059e571e6b4SHuawei Xie 				dev = vdev->dev;
2060d19533e8SHuawei Xie 
2061e571e6b4SHuawei Xie 				if (likely(!vdev->remove)) {
2062e571e6b4SHuawei Xie 					tx_q = &tx_queue_zcp[(uint16_t)vdev->vmdq_rx_q];
2063d19533e8SHuawei Xie 					if (tx_q->len) {
2064d19533e8SHuawei Xie 						LOG_DEBUG(VHOST_DATA,
2065d19533e8SHuawei Xie 						"TX queue drained after timeout"
2066d19533e8SHuawei Xie 						" with burst size %u\n",
2067d19533e8SHuawei Xie 						tx_q->len);
2068d19533e8SHuawei Xie 
2069d19533e8SHuawei Xie 						/*
2070d19533e8SHuawei Xie 						 * Tx any packets in the queue
2071d19533e8SHuawei Xie 						 */
2072d19533e8SHuawei Xie 						ret = rte_eth_tx_burst(
2073d19533e8SHuawei Xie 							ports[0],
2074d19533e8SHuawei Xie 							(uint16_t)tx_q->txq_id,
2075d19533e8SHuawei Xie 							(struct rte_mbuf **)
2076d19533e8SHuawei Xie 							tx_q->m_table,
2077d19533e8SHuawei Xie 							(uint16_t)tx_q->len);
2078d19533e8SHuawei Xie 						if (unlikely(ret < tx_q->len)) {
2079d19533e8SHuawei Xie 							do {
2080d19533e8SHuawei Xie 								rte_pktmbuf_free(
2081d19533e8SHuawei Xie 									tx_q->m_table[ret]);
2082d19533e8SHuawei Xie 							} while (++ret < tx_q->len);
2083d19533e8SHuawei Xie 						}
2084d19533e8SHuawei Xie 						tx_q->len = 0;
2085d19533e8SHuawei Xie 
2086d19533e8SHuawei Xie 						txmbuf_clean_zcp(dev,
2087e571e6b4SHuawei Xie 							&vpool_array[MAX_QUEUES+vdev->vmdq_rx_q]);
2088d19533e8SHuawei Xie 					}
2089d19533e8SHuawei Xie 				}
2090d19533e8SHuawei Xie 				dev_ll = dev_ll->next;
2091d19533e8SHuawei Xie 			}
2092d19533e8SHuawei Xie 			prev_tsc = cur_tsc;
2093d19533e8SHuawei Xie 		}
2094d19533e8SHuawei Xie 
2095d19533e8SHuawei Xie 		rte_prefetch0(lcore_ll->ll_root_used);
2096d19533e8SHuawei Xie 
2097d19533e8SHuawei Xie 		/*
2098d19533e8SHuawei Xie 		 * Inform the configuration core that we have exited the linked
2099d19533e8SHuawei Xie 		 * list and that no devices are in use if requested.
2100d19533e8SHuawei Xie 		 */
2101d19533e8SHuawei Xie 		if (lcore_ll->dev_removal_flag == REQUEST_DEV_REMOVAL)
2102d19533e8SHuawei Xie 			lcore_ll->dev_removal_flag = ACK_DEV_REMOVAL;
2103d19533e8SHuawei Xie 
2104d19533e8SHuawei Xie 		/* Process devices */
2105d19533e8SHuawei Xie 		dev_ll = lcore_ll->ll_root_used;
2106d19533e8SHuawei Xie 
2107e571e6b4SHuawei Xie 		while ((dev_ll != NULL) && (dev_ll->vdev != NULL)) {
2108e571e6b4SHuawei Xie 			vdev = dev_ll->vdev;
2109e571e6b4SHuawei Xie 			dev  = vdev->dev;
2110e571e6b4SHuawei Xie 			if (unlikely(vdev->remove)) {
2111d19533e8SHuawei Xie 				dev_ll = dev_ll->next;
2112e571e6b4SHuawei Xie 				unlink_vmdq(vdev);
2113e571e6b4SHuawei Xie 				vdev->ready = DEVICE_SAFE_REMOVE;
2114d19533e8SHuawei Xie 				continue;
2115d19533e8SHuawei Xie 			}
2116d19533e8SHuawei Xie 
2117e571e6b4SHuawei Xie 			if (likely(vdev->ready == DEVICE_RX)) {
2118e571e6b4SHuawei Xie 				uint32_t index = vdev->vmdq_rx_q;
2119d19533e8SHuawei Xie 				uint16_t i;
2120d19533e8SHuawei Xie 				count_in_ring
2121d19533e8SHuawei Xie 				= rte_ring_count(vpool_array[index].ring);
2122d19533e8SHuawei Xie 				uint16_t free_entries
2123d19533e8SHuawei Xie 				= (uint16_t)get_available_ring_num_zcp(dev);
2124d19533e8SHuawei Xie 
2125d19533e8SHuawei Xie 				/*
2126d19533e8SHuawei Xie 				 * Attach all mbufs in vpool.ring and put back
2127d19533e8SHuawei Xie 				 * into vpool.pool.
2128d19533e8SHuawei Xie 				 */
2129d19533e8SHuawei Xie 				for (i = 0;
2130d19533e8SHuawei Xie 				i < RTE_MIN(free_entries,
2131d19533e8SHuawei Xie 				RTE_MIN(count_in_ring, MAX_PKT_BURST));
2132d19533e8SHuawei Xie 				i++)
2133d19533e8SHuawei Xie 					attach_rxmbuf_zcp(dev);
2134d19533e8SHuawei Xie 
2135d19533e8SHuawei Xie 				/* Handle guest RX */
2136d19533e8SHuawei Xie 				rx_count = rte_eth_rx_burst(ports[0],
2137e571e6b4SHuawei Xie 					vdev->vmdq_rx_q, pkts_burst,
2138d19533e8SHuawei Xie 					MAX_PKT_BURST);
2139d19533e8SHuawei Xie 
2140d19533e8SHuawei Xie 				if (rx_count) {
2141d19533e8SHuawei Xie 					ret_count = virtio_dev_rx_zcp(dev,
2142d19533e8SHuawei Xie 							pkts_burst, rx_count);
2143d19533e8SHuawei Xie 					if (enable_stats) {
2144d19533e8SHuawei Xie 						dev_statistics[dev->device_fh].rx_total
2145d19533e8SHuawei Xie 							+= rx_count;
2146d19533e8SHuawei Xie 						dev_statistics[dev->device_fh].rx
2147d19533e8SHuawei Xie 							+= ret_count;
2148d19533e8SHuawei Xie 					}
2149d19533e8SHuawei Xie 					while (likely(rx_count)) {
2150d19533e8SHuawei Xie 						rx_count--;
2151d19533e8SHuawei Xie 						pktmbuf_detach_zcp(
2152d19533e8SHuawei Xie 							pkts_burst[rx_count]);
2153d19533e8SHuawei Xie 						rte_ring_sp_enqueue(
2154d19533e8SHuawei Xie 							vpool_array[index].ring,
2155d19533e8SHuawei Xie 							(void *)pkts_burst[rx_count]);
2156d19533e8SHuawei Xie 					}
2157d19533e8SHuawei Xie 				}
2158d19533e8SHuawei Xie 			}
2159d19533e8SHuawei Xie 
2160e571e6b4SHuawei Xie 			if (likely(!vdev->remove))
2161d19533e8SHuawei Xie 				/* Handle guest TX */
2162d19533e8SHuawei Xie 				virtio_dev_tx_zcp(dev);
2163d19533e8SHuawei Xie 
2164d19533e8SHuawei Xie 			/* Move to the next device in the list */
2165d19533e8SHuawei Xie 			dev_ll = dev_ll->next;
2166d19533e8SHuawei Xie 		}
2167d19533e8SHuawei Xie 	}
2168d19533e8SHuawei Xie 
2169d19533e8SHuawei Xie 	return 0;
2170d19533e8SHuawei Xie }
2171d19533e8SHuawei Xie 
2172d19533e8SHuawei Xie 
2173d19533e8SHuawei Xie /*
2174d19533e8SHuawei Xie  * Add an entry to a used linked list. A free entry must first be found
2175d19533e8SHuawei Xie  * in the free linked list using get_data_ll_free_entry();
2176d19533e8SHuawei Xie  */
2177d19533e8SHuawei Xie static void
2178d19533e8SHuawei Xie add_data_ll_entry(struct virtio_net_data_ll **ll_root_addr,
2179d19533e8SHuawei Xie 	struct virtio_net_data_ll *ll_dev)
2180d19533e8SHuawei Xie {
2181d19533e8SHuawei Xie 	struct virtio_net_data_ll *ll = *ll_root_addr;
2182d19533e8SHuawei Xie 
2183d19533e8SHuawei Xie 	/* Set next as NULL and use a compiler barrier to avoid reordering. */
2184d19533e8SHuawei Xie 	ll_dev->next = NULL;
2185d19533e8SHuawei Xie 	rte_compiler_barrier();
2186d19533e8SHuawei Xie 
2187d19533e8SHuawei Xie 	/* If ll == NULL then this is the first device. */
2188d19533e8SHuawei Xie 	if (ll) {
2189d19533e8SHuawei Xie 		/* Increment to the tail of the linked list. */
2190d19533e8SHuawei Xie 		while ((ll->next != NULL) )
2191d19533e8SHuawei Xie 			ll = ll->next;
2192d19533e8SHuawei Xie 
2193d19533e8SHuawei Xie 		ll->next = ll_dev;
2194d19533e8SHuawei Xie 	} else {
2195d19533e8SHuawei Xie 		*ll_root_addr = ll_dev;
2196d19533e8SHuawei Xie 	}
2197d19533e8SHuawei Xie }
2198d19533e8SHuawei Xie 
2199d19533e8SHuawei Xie /*
2200d19533e8SHuawei Xie  * Remove an entry from a used linked list. The entry must then be added to
2201d19533e8SHuawei Xie  * the free linked list using put_data_ll_free_entry().
2202d19533e8SHuawei Xie  */
2203d19533e8SHuawei Xie static void
2204d19533e8SHuawei Xie rm_data_ll_entry(struct virtio_net_data_ll **ll_root_addr,
2205d19533e8SHuawei Xie 	struct virtio_net_data_ll *ll_dev,
2206d19533e8SHuawei Xie 	struct virtio_net_data_ll *ll_dev_last)
2207d19533e8SHuawei Xie {
2208d19533e8SHuawei Xie 	struct virtio_net_data_ll *ll = *ll_root_addr;
2209d19533e8SHuawei Xie 
2210d19533e8SHuawei Xie 	if (unlikely((ll == NULL) || (ll_dev == NULL)))
2211d19533e8SHuawei Xie 		return;
2212d19533e8SHuawei Xie 
2213d19533e8SHuawei Xie 	if (ll_dev == ll)
2214d19533e8SHuawei Xie 		*ll_root_addr = ll_dev->next;
2215d19533e8SHuawei Xie 	else
2216d19533e8SHuawei Xie 		if (likely(ll_dev_last != NULL))
2217d19533e8SHuawei Xie 			ll_dev_last->next = ll_dev->next;
2218d19533e8SHuawei Xie 		else
2219d19533e8SHuawei Xie 			RTE_LOG(ERR, VHOST_CONFIG, "Remove entry form ll failed.\n");
2220d19533e8SHuawei Xie }
2221d19533e8SHuawei Xie 
2222d19533e8SHuawei Xie /*
2223d19533e8SHuawei Xie  * Find and return an entry from the free linked list.
2224d19533e8SHuawei Xie  */
2225d19533e8SHuawei Xie static struct virtio_net_data_ll *
2226d19533e8SHuawei Xie get_data_ll_free_entry(struct virtio_net_data_ll **ll_root_addr)
2227d19533e8SHuawei Xie {
2228d19533e8SHuawei Xie 	struct virtio_net_data_ll *ll_free = *ll_root_addr;
2229d19533e8SHuawei Xie 	struct virtio_net_data_ll *ll_dev;
2230d19533e8SHuawei Xie 
2231d19533e8SHuawei Xie 	if (ll_free == NULL)
2232d19533e8SHuawei Xie 		return NULL;
2233d19533e8SHuawei Xie 
2234d19533e8SHuawei Xie 	ll_dev = ll_free;
2235d19533e8SHuawei Xie 	*ll_root_addr = ll_free->next;
2236d19533e8SHuawei Xie 
2237d19533e8SHuawei Xie 	return ll_dev;
2238d19533e8SHuawei Xie }
2239d19533e8SHuawei Xie 
2240d19533e8SHuawei Xie /*
2241d19533e8SHuawei Xie  * Place an entry back on to the free linked list.
2242d19533e8SHuawei Xie  */
2243d19533e8SHuawei Xie static void
2244d19533e8SHuawei Xie put_data_ll_free_entry(struct virtio_net_data_ll **ll_root_addr,
2245d19533e8SHuawei Xie 	struct virtio_net_data_ll *ll_dev)
2246d19533e8SHuawei Xie {
2247d19533e8SHuawei Xie 	struct virtio_net_data_ll *ll_free = *ll_root_addr;
2248d19533e8SHuawei Xie 
2249d19533e8SHuawei Xie 	if (ll_dev == NULL)
2250d19533e8SHuawei Xie 		return;
2251d19533e8SHuawei Xie 
2252d19533e8SHuawei Xie 	ll_dev->next = ll_free;
2253d19533e8SHuawei Xie 	*ll_root_addr = ll_dev;
2254d19533e8SHuawei Xie }
2255d19533e8SHuawei Xie 
2256d19533e8SHuawei Xie /*
2257d19533e8SHuawei Xie  * Creates a linked list of a given size.
2258d19533e8SHuawei Xie  */
2259d19533e8SHuawei Xie static struct virtio_net_data_ll *
2260d19533e8SHuawei Xie alloc_data_ll(uint32_t size)
2261d19533e8SHuawei Xie {
2262d19533e8SHuawei Xie 	struct virtio_net_data_ll *ll_new;
2263d19533e8SHuawei Xie 	uint32_t i;
2264d19533e8SHuawei Xie 
2265d19533e8SHuawei Xie 	/* Malloc and then chain the linked list. */
2266d19533e8SHuawei Xie 	ll_new = malloc(size * sizeof(struct virtio_net_data_ll));
2267d19533e8SHuawei Xie 	if (ll_new == NULL) {
2268d19533e8SHuawei Xie 		RTE_LOG(ERR, VHOST_CONFIG, "Failed to allocate memory for ll_new.\n");
2269d19533e8SHuawei Xie 		return NULL;
2270d19533e8SHuawei Xie 	}
2271d19533e8SHuawei Xie 
2272d19533e8SHuawei Xie 	for (i = 0; i < size - 1; i++) {
2273e571e6b4SHuawei Xie 		ll_new[i].vdev = NULL;
2274d19533e8SHuawei Xie 		ll_new[i].next = &ll_new[i+1];
2275d19533e8SHuawei Xie 	}
2276d19533e8SHuawei Xie 	ll_new[i].next = NULL;
2277d19533e8SHuawei Xie 
2278d19533e8SHuawei Xie 	return (ll_new);
2279d19533e8SHuawei Xie }
2280d19533e8SHuawei Xie 
2281d19533e8SHuawei Xie /*
2282d19533e8SHuawei Xie  * Create the main linked list along with each individual cores linked list. A used and a free list
2283d19533e8SHuawei Xie  * are created to manage entries.
2284d19533e8SHuawei Xie  */
2285d19533e8SHuawei Xie static int
2286d19533e8SHuawei Xie init_data_ll (void)
2287d19533e8SHuawei Xie {
2288d19533e8SHuawei Xie 	int lcore;
2289d19533e8SHuawei Xie 
2290d19533e8SHuawei Xie 	RTE_LCORE_FOREACH_SLAVE(lcore) {
2291d19533e8SHuawei Xie 		lcore_info[lcore].lcore_ll = malloc(sizeof(struct lcore_ll_info));
2292d19533e8SHuawei Xie 		if (lcore_info[lcore].lcore_ll == NULL) {
2293d19533e8SHuawei Xie 			RTE_LOG(ERR, VHOST_CONFIG, "Failed to allocate memory for lcore_ll.\n");
2294d19533e8SHuawei Xie 			return -1;
2295d19533e8SHuawei Xie 		}
2296d19533e8SHuawei Xie 
2297d19533e8SHuawei Xie 		lcore_info[lcore].lcore_ll->device_num = 0;
2298d19533e8SHuawei Xie 		lcore_info[lcore].lcore_ll->dev_removal_flag = ACK_DEV_REMOVAL;
2299d19533e8SHuawei Xie 		lcore_info[lcore].lcore_ll->ll_root_used = NULL;
2300d19533e8SHuawei Xie 		if (num_devices % num_switching_cores)
2301d19533e8SHuawei Xie 			lcore_info[lcore].lcore_ll->ll_root_free = alloc_data_ll((num_devices / num_switching_cores) + 1);
2302d19533e8SHuawei Xie 		else
2303d19533e8SHuawei Xie 			lcore_info[lcore].lcore_ll->ll_root_free = alloc_data_ll(num_devices / num_switching_cores);
2304d19533e8SHuawei Xie 	}
2305d19533e8SHuawei Xie 
2306d19533e8SHuawei Xie 	/* Allocate devices up to a maximum of MAX_DEVICES. */
2307d19533e8SHuawei Xie 	ll_root_free = alloc_data_ll(MIN((num_devices), MAX_DEVICES));
2308d19533e8SHuawei Xie 
2309d19533e8SHuawei Xie 	return 0;
2310d19533e8SHuawei Xie }
2311d19533e8SHuawei Xie 
2312d19533e8SHuawei Xie /*
2313d19533e8SHuawei Xie  * Remove a device from the specific data core linked list and from the main linked list. Synchonization
2314d19533e8SHuawei Xie  * occurs through the use of the lcore dev_removal_flag. Device is made volatile here to avoid re-ordering
2315d19533e8SHuawei Xie  * of dev->remove=1 which can cause an infinite loop in the rte_pause loop.
2316d19533e8SHuawei Xie  */
2317d19533e8SHuawei Xie static void
2318d19533e8SHuawei Xie destroy_device (volatile struct virtio_net *dev)
2319d19533e8SHuawei Xie {
2320d19533e8SHuawei Xie 	struct virtio_net_data_ll *ll_lcore_dev_cur;
2321d19533e8SHuawei Xie 	struct virtio_net_data_ll *ll_main_dev_cur;
2322d19533e8SHuawei Xie 	struct virtio_net_data_ll *ll_lcore_dev_last = NULL;
2323d19533e8SHuawei Xie 	struct virtio_net_data_ll *ll_main_dev_last = NULL;
2324e571e6b4SHuawei Xie 	struct vhost_dev *vdev;
2325d19533e8SHuawei Xie 	int lcore;
2326d19533e8SHuawei Xie 
2327d19533e8SHuawei Xie 	dev->flags &= ~VIRTIO_DEV_RUNNING;
2328d19533e8SHuawei Xie 
2329e571e6b4SHuawei Xie 	vdev = (struct vhost_dev *)dev->priv;
2330d19533e8SHuawei Xie 	/*set the remove flag. */
2331e571e6b4SHuawei Xie 	vdev->remove = 1;
2332e571e6b4SHuawei Xie 	while(vdev->ready != DEVICE_SAFE_REMOVE) {
2333d19533e8SHuawei Xie 		rte_pause();
2334d19533e8SHuawei Xie 	}
2335d19533e8SHuawei Xie 
2336d19533e8SHuawei Xie 	/* Search for entry to be removed from lcore ll */
2337e571e6b4SHuawei Xie 	ll_lcore_dev_cur = lcore_info[vdev->coreid].lcore_ll->ll_root_used;
2338d19533e8SHuawei Xie 	while (ll_lcore_dev_cur != NULL) {
2339e571e6b4SHuawei Xie 		if (ll_lcore_dev_cur->vdev == vdev) {
2340d19533e8SHuawei Xie 			break;
2341d19533e8SHuawei Xie 		} else {
2342d19533e8SHuawei Xie 			ll_lcore_dev_last = ll_lcore_dev_cur;
2343d19533e8SHuawei Xie 			ll_lcore_dev_cur = ll_lcore_dev_cur->next;
2344d19533e8SHuawei Xie 		}
2345d19533e8SHuawei Xie 	}
2346d19533e8SHuawei Xie 
2347d19533e8SHuawei Xie 	if (ll_lcore_dev_cur == NULL) {
2348d19533e8SHuawei Xie 		RTE_LOG(ERR, VHOST_CONFIG,
2349d19533e8SHuawei Xie 			"(%"PRIu64") Failed to find the dev to be destroy.\n",
2350d19533e8SHuawei Xie 			dev->device_fh);
2351d19533e8SHuawei Xie 		return;
2352d19533e8SHuawei Xie 	}
2353d19533e8SHuawei Xie 
2354d19533e8SHuawei Xie 	/* Search for entry to be removed from main ll */
2355d19533e8SHuawei Xie 	ll_main_dev_cur = ll_root_used;
2356d19533e8SHuawei Xie 	ll_main_dev_last = NULL;
2357d19533e8SHuawei Xie 	while (ll_main_dev_cur != NULL) {
2358e571e6b4SHuawei Xie 		if (ll_main_dev_cur->vdev == vdev) {
2359d19533e8SHuawei Xie 			break;
2360d19533e8SHuawei Xie 		} else {
2361d19533e8SHuawei Xie 			ll_main_dev_last = ll_main_dev_cur;
2362d19533e8SHuawei Xie 			ll_main_dev_cur = ll_main_dev_cur->next;
2363d19533e8SHuawei Xie 		}
2364d19533e8SHuawei Xie 	}
2365d19533e8SHuawei Xie 
2366d19533e8SHuawei Xie 	/* Remove entries from the lcore and main ll. */
2367e571e6b4SHuawei Xie 	rm_data_ll_entry(&lcore_info[vdev->coreid].lcore_ll->ll_root_used, ll_lcore_dev_cur, ll_lcore_dev_last);
2368d19533e8SHuawei Xie 	rm_data_ll_entry(&ll_root_used, ll_main_dev_cur, ll_main_dev_last);
2369d19533e8SHuawei Xie 
2370d19533e8SHuawei Xie 	/* Set the dev_removal_flag on each lcore. */
2371d19533e8SHuawei Xie 	RTE_LCORE_FOREACH_SLAVE(lcore) {
2372d19533e8SHuawei Xie 		lcore_info[lcore].lcore_ll->dev_removal_flag = REQUEST_DEV_REMOVAL;
2373d19533e8SHuawei Xie 	}
2374d19533e8SHuawei Xie 
2375d19533e8SHuawei Xie 	/*
2376d19533e8SHuawei Xie 	 * Once each core has set the dev_removal_flag to ACK_DEV_REMOVAL we can be sure that
2377d19533e8SHuawei Xie 	 * they can no longer access the device removed from the linked lists and that the devices
2378d19533e8SHuawei Xie 	 * are no longer in use.
2379d19533e8SHuawei Xie 	 */
2380d19533e8SHuawei Xie 	RTE_LCORE_FOREACH_SLAVE(lcore) {
2381d19533e8SHuawei Xie 		while (lcore_info[lcore].lcore_ll->dev_removal_flag != ACK_DEV_REMOVAL) {
2382d19533e8SHuawei Xie 			rte_pause();
2383d19533e8SHuawei Xie 		}
2384d19533e8SHuawei Xie 	}
2385d19533e8SHuawei Xie 
2386d19533e8SHuawei Xie 	/* Add the entries back to the lcore and main free ll.*/
2387e571e6b4SHuawei Xie 	put_data_ll_free_entry(&lcore_info[vdev->coreid].lcore_ll->ll_root_free, ll_lcore_dev_cur);
2388d19533e8SHuawei Xie 	put_data_ll_free_entry(&ll_root_free, ll_main_dev_cur);
2389d19533e8SHuawei Xie 
2390d19533e8SHuawei Xie 	/* Decrement number of device on the lcore. */
2391e571e6b4SHuawei Xie 	lcore_info[vdev->coreid].lcore_ll->device_num--;
2392d19533e8SHuawei Xie 
2393d19533e8SHuawei Xie 	RTE_LOG(INFO, VHOST_DATA, "(%"PRIu64") Device has been removed from data core\n", dev->device_fh);
2394d19533e8SHuawei Xie 
2395d19533e8SHuawei Xie 	if (zero_copy) {
2396e571e6b4SHuawei Xie 		struct vpool *vpool = &vpool_array[vdev->vmdq_rx_q];
2397d19533e8SHuawei Xie 
2398d19533e8SHuawei Xie 		/* Stop the RX queue. */
2399e571e6b4SHuawei Xie 		if (rte_eth_dev_rx_queue_stop(ports[0], vdev->vmdq_rx_q) != 0) {
2400d19533e8SHuawei Xie 			LOG_DEBUG(VHOST_CONFIG,
2401d19533e8SHuawei Xie 				"(%"PRIu64") In destroy_device: Failed to stop "
2402d19533e8SHuawei Xie 				"rx queue:%d\n",
2403d19533e8SHuawei Xie 				dev->device_fh,
2404e571e6b4SHuawei Xie 				vdev->vmdq_rx_q);
2405d19533e8SHuawei Xie 		}
2406d19533e8SHuawei Xie 
2407d19533e8SHuawei Xie 		LOG_DEBUG(VHOST_CONFIG,
2408d19533e8SHuawei Xie 			"(%"PRIu64") in destroy_device: Start put mbuf in "
2409d19533e8SHuawei Xie 			"mempool back to ring for RX queue: %d\n",
2410e571e6b4SHuawei Xie 			dev->device_fh, vdev->vmdq_rx_q);
2411d19533e8SHuawei Xie 
2412d19533e8SHuawei Xie 		mbuf_destroy_zcp(vpool);
2413d19533e8SHuawei Xie 
2414d19533e8SHuawei Xie 		/* Stop the TX queue. */
2415e571e6b4SHuawei Xie 		if (rte_eth_dev_tx_queue_stop(ports[0], vdev->vmdq_rx_q) != 0) {
2416d19533e8SHuawei Xie 			LOG_DEBUG(VHOST_CONFIG,
2417d19533e8SHuawei Xie 				"(%"PRIu64") In destroy_device: Failed to "
2418d19533e8SHuawei Xie 				"stop tx queue:%d\n",
2419e571e6b4SHuawei Xie 				dev->device_fh, vdev->vmdq_rx_q);
2420d19533e8SHuawei Xie 		}
2421d19533e8SHuawei Xie 
2422e571e6b4SHuawei Xie 		vpool = &vpool_array[vdev->vmdq_rx_q + MAX_QUEUES];
2423d19533e8SHuawei Xie 
2424d19533e8SHuawei Xie 		LOG_DEBUG(VHOST_CONFIG,
2425d19533e8SHuawei Xie 			"(%"PRIu64") destroy_device: Start put mbuf in mempool "
2426d19533e8SHuawei Xie 			"back to ring for TX queue: %d, dev:(%"PRIu64")\n",
2427e571e6b4SHuawei Xie 			dev->device_fh, (vdev->vmdq_rx_q + MAX_QUEUES),
2428d19533e8SHuawei Xie 			dev->device_fh);
2429d19533e8SHuawei Xie 
2430d19533e8SHuawei Xie 		mbuf_destroy_zcp(vpool);
24319915bb1fSHuawei Xie 		rte_free(vdev->regions_hpa);
2432d19533e8SHuawei Xie 	}
2433e571e6b4SHuawei Xie 	rte_free(vdev);
2434d19533e8SHuawei Xie 
2435d19533e8SHuawei Xie }
2436d19533e8SHuawei Xie 
2437d19533e8SHuawei Xie /*
24389915bb1fSHuawei Xie  * Calculate the region count of physical continous regions for one particular
24399915bb1fSHuawei Xie  * region of whose vhost virtual address is continous. The particular region
24409915bb1fSHuawei Xie  * start from vva_start, with size of 'size' in argument.
24419915bb1fSHuawei Xie  */
24429915bb1fSHuawei Xie static uint32_t
24439915bb1fSHuawei Xie check_hpa_regions(uint64_t vva_start, uint64_t size)
24449915bb1fSHuawei Xie {
24459915bb1fSHuawei Xie 	uint32_t i, nregions = 0, page_size = getpagesize();
24469915bb1fSHuawei Xie 	uint64_t cur_phys_addr = 0, next_phys_addr = 0;
24479915bb1fSHuawei Xie 	if (vva_start % page_size) {
24489915bb1fSHuawei Xie 		LOG_DEBUG(VHOST_CONFIG,
24499915bb1fSHuawei Xie 			"in check_countinous: vva start(%p) mod page_size(%d) "
24509915bb1fSHuawei Xie 			"has remainder\n",
24519915bb1fSHuawei Xie 			(void *)(uintptr_t)vva_start, page_size);
24529915bb1fSHuawei Xie 		return 0;
24539915bb1fSHuawei Xie 	}
24549915bb1fSHuawei Xie 	if (size % page_size) {
24559915bb1fSHuawei Xie 		LOG_DEBUG(VHOST_CONFIG,
24569915bb1fSHuawei Xie 			"in check_countinous: "
24579915bb1fSHuawei Xie 			"size((%"PRIu64")) mod page_size(%d) has remainder\n",
24589915bb1fSHuawei Xie 			size, page_size);
24599915bb1fSHuawei Xie 		return 0;
24609915bb1fSHuawei Xie 	}
24619915bb1fSHuawei Xie 	for (i = 0; i < size - page_size; i = i + page_size) {
24629915bb1fSHuawei Xie 		cur_phys_addr
24639915bb1fSHuawei Xie 			= rte_mem_virt2phy((void *)(uintptr_t)(vva_start + i));
24649915bb1fSHuawei Xie 		next_phys_addr = rte_mem_virt2phy(
24659915bb1fSHuawei Xie 			(void *)(uintptr_t)(vva_start + i + page_size));
24669915bb1fSHuawei Xie 		if ((cur_phys_addr + page_size) != next_phys_addr) {
24679915bb1fSHuawei Xie 			++nregions;
24689915bb1fSHuawei Xie 			LOG_DEBUG(VHOST_CONFIG,
24699915bb1fSHuawei Xie 				"in check_continuous: hva addr:(%p) is not "
24709915bb1fSHuawei Xie 				"continuous with hva addr:(%p), diff:%d\n",
24719915bb1fSHuawei Xie 				(void *)(uintptr_t)(vva_start + (uint64_t)i),
24729915bb1fSHuawei Xie 				(void *)(uintptr_t)(vva_start + (uint64_t)i
24739915bb1fSHuawei Xie 				+ page_size), page_size);
24749915bb1fSHuawei Xie 			LOG_DEBUG(VHOST_CONFIG,
24759915bb1fSHuawei Xie 				"in check_continuous: hpa addr:(%p) is not "
24769915bb1fSHuawei Xie 				"continuous with hpa addr:(%p), "
24779915bb1fSHuawei Xie 				"diff:(%"PRIu64")\n",
24789915bb1fSHuawei Xie 				(void *)(uintptr_t)cur_phys_addr,
24799915bb1fSHuawei Xie 				(void *)(uintptr_t)next_phys_addr,
24809915bb1fSHuawei Xie 				(next_phys_addr-cur_phys_addr));
24819915bb1fSHuawei Xie 		}
24829915bb1fSHuawei Xie 	}
24839915bb1fSHuawei Xie 	return nregions;
24849915bb1fSHuawei Xie }
24859915bb1fSHuawei Xie 
24869915bb1fSHuawei Xie /*
24879915bb1fSHuawei Xie  * Divide each region whose vhost virtual address is continous into a few
24889915bb1fSHuawei Xie  * sub-regions, make sure the physical address within each sub-region are
24899915bb1fSHuawei Xie  * continous. And fill offset(to GPA) and size etc. information of each
24909915bb1fSHuawei Xie  * sub-region into regions_hpa.
24919915bb1fSHuawei Xie  */
24929915bb1fSHuawei Xie static uint32_t
24939915bb1fSHuawei Xie fill_hpa_memory_regions(struct virtio_memory_regions_hpa *mem_region_hpa, struct virtio_memory *virtio_memory)
24949915bb1fSHuawei Xie {
24959915bb1fSHuawei Xie 	uint32_t regionidx, regionidx_hpa = 0, i, k, page_size = getpagesize();
24969915bb1fSHuawei Xie 	uint64_t cur_phys_addr = 0, next_phys_addr = 0, vva_start;
24979915bb1fSHuawei Xie 
24989915bb1fSHuawei Xie 	if (mem_region_hpa == NULL)
24999915bb1fSHuawei Xie 		return 0;
25009915bb1fSHuawei Xie 
25019915bb1fSHuawei Xie 	for (regionidx = 0; regionidx < virtio_memory->nregions; regionidx++) {
25029915bb1fSHuawei Xie 		vva_start = virtio_memory->regions[regionidx].guest_phys_address +
25039915bb1fSHuawei Xie 			virtio_memory->regions[regionidx].address_offset;
25049915bb1fSHuawei Xie 		mem_region_hpa[regionidx_hpa].guest_phys_address
25059915bb1fSHuawei Xie 			= virtio_memory->regions[regionidx].guest_phys_address;
25069915bb1fSHuawei Xie 		mem_region_hpa[regionidx_hpa].host_phys_addr_offset =
25079915bb1fSHuawei Xie 			rte_mem_virt2phy((void *)(uintptr_t)(vva_start)) -
25089915bb1fSHuawei Xie 			mem_region_hpa[regionidx_hpa].guest_phys_address;
25099915bb1fSHuawei Xie 		LOG_DEBUG(VHOST_CONFIG,
25109915bb1fSHuawei Xie 			"in fill_hpa_regions: guest phys addr start[%d]:(%p)\n",
25119915bb1fSHuawei Xie 			regionidx_hpa,
25129915bb1fSHuawei Xie 			(void *)(uintptr_t)
25139915bb1fSHuawei Xie 			(mem_region_hpa[regionidx_hpa].guest_phys_address));
25149915bb1fSHuawei Xie 		LOG_DEBUG(VHOST_CONFIG,
25159915bb1fSHuawei Xie 			"in fill_hpa_regions: host  phys addr start[%d]:(%p)\n",
25169915bb1fSHuawei Xie 			regionidx_hpa,
25179915bb1fSHuawei Xie 			(void *)(uintptr_t)
25189915bb1fSHuawei Xie 			(mem_region_hpa[regionidx_hpa].host_phys_addr_offset));
25199915bb1fSHuawei Xie 		for (i = 0, k = 0;
25209915bb1fSHuawei Xie 			i < virtio_memory->regions[regionidx].memory_size -
25219915bb1fSHuawei Xie 				page_size;
25229915bb1fSHuawei Xie 			i += page_size) {
25239915bb1fSHuawei Xie 			cur_phys_addr = rte_mem_virt2phy(
25249915bb1fSHuawei Xie 					(void *)(uintptr_t)(vva_start + i));
25259915bb1fSHuawei Xie 			next_phys_addr = rte_mem_virt2phy(
25269915bb1fSHuawei Xie 					(void *)(uintptr_t)(vva_start +
25279915bb1fSHuawei Xie 					i + page_size));
25289915bb1fSHuawei Xie 			if ((cur_phys_addr + page_size) != next_phys_addr) {
25299915bb1fSHuawei Xie 				mem_region_hpa[regionidx_hpa].guest_phys_address_end =
25309915bb1fSHuawei Xie 					mem_region_hpa[regionidx_hpa].guest_phys_address +
25319915bb1fSHuawei Xie 					k + page_size;
25329915bb1fSHuawei Xie 				mem_region_hpa[regionidx_hpa].memory_size
25339915bb1fSHuawei Xie 					= k + page_size;
25349915bb1fSHuawei Xie 				LOG_DEBUG(VHOST_CONFIG, "in fill_hpa_regions: guest "
25359915bb1fSHuawei Xie 					"phys addr end  [%d]:(%p)\n",
25369915bb1fSHuawei Xie 					regionidx_hpa,
25379915bb1fSHuawei Xie 					(void *)(uintptr_t)
25389915bb1fSHuawei Xie 					(mem_region_hpa[regionidx_hpa].guest_phys_address_end));
25399915bb1fSHuawei Xie 				LOG_DEBUG(VHOST_CONFIG,
25409915bb1fSHuawei Xie 					"in fill_hpa_regions: guest phys addr "
25419915bb1fSHuawei Xie 					"size [%d]:(%p)\n",
25429915bb1fSHuawei Xie 					regionidx_hpa,
25439915bb1fSHuawei Xie 					(void *)(uintptr_t)
25449915bb1fSHuawei Xie 					(mem_region_hpa[regionidx_hpa].memory_size));
25459915bb1fSHuawei Xie 				mem_region_hpa[regionidx_hpa + 1].guest_phys_address
25469915bb1fSHuawei Xie 					= mem_region_hpa[regionidx_hpa].guest_phys_address_end;
25479915bb1fSHuawei Xie 				++regionidx_hpa;
25489915bb1fSHuawei Xie 				mem_region_hpa[regionidx_hpa].host_phys_addr_offset =
25499915bb1fSHuawei Xie 					next_phys_addr -
25509915bb1fSHuawei Xie 					mem_region_hpa[regionidx_hpa].guest_phys_address;
25519915bb1fSHuawei Xie 				LOG_DEBUG(VHOST_CONFIG, "in fill_hpa_regions: guest"
25529915bb1fSHuawei Xie 					" phys addr start[%d]:(%p)\n",
25539915bb1fSHuawei Xie 					regionidx_hpa,
25549915bb1fSHuawei Xie 					(void *)(uintptr_t)
25559915bb1fSHuawei Xie 					(mem_region_hpa[regionidx_hpa].guest_phys_address));
25569915bb1fSHuawei Xie 				LOG_DEBUG(VHOST_CONFIG,
25579915bb1fSHuawei Xie 					"in fill_hpa_regions: host  phys addr "
25589915bb1fSHuawei Xie 					"start[%d]:(%p)\n",
25599915bb1fSHuawei Xie 					regionidx_hpa,
25609915bb1fSHuawei Xie 					(void *)(uintptr_t)
25619915bb1fSHuawei Xie 					(mem_region_hpa[regionidx_hpa].host_phys_addr_offset));
25629915bb1fSHuawei Xie 				k = 0;
25639915bb1fSHuawei Xie 			} else {
25649915bb1fSHuawei Xie 				k += page_size;
25659915bb1fSHuawei Xie 			}
25669915bb1fSHuawei Xie 		}
25679915bb1fSHuawei Xie 		mem_region_hpa[regionidx_hpa].guest_phys_address_end
25689915bb1fSHuawei Xie 			= mem_region_hpa[regionidx_hpa].guest_phys_address
25699915bb1fSHuawei Xie 			+ k + page_size;
25709915bb1fSHuawei Xie 		mem_region_hpa[regionidx_hpa].memory_size = k + page_size;
25719915bb1fSHuawei Xie 		LOG_DEBUG(VHOST_CONFIG, "in fill_hpa_regions: guest phys addr end  "
25729915bb1fSHuawei Xie 			"[%d]:(%p)\n", regionidx_hpa,
25739915bb1fSHuawei Xie 			(void *)(uintptr_t)
25749915bb1fSHuawei Xie 			(mem_region_hpa[regionidx_hpa].guest_phys_address_end));
25759915bb1fSHuawei Xie 		LOG_DEBUG(VHOST_CONFIG, "in fill_hpa_regions: guest phys addr size "
25769915bb1fSHuawei Xie 			"[%d]:(%p)\n", regionidx_hpa,
25779915bb1fSHuawei Xie 			(void *)(uintptr_t)
25789915bb1fSHuawei Xie 			(mem_region_hpa[regionidx_hpa].memory_size));
25799915bb1fSHuawei Xie 		++regionidx_hpa;
25809915bb1fSHuawei Xie 	}
25819915bb1fSHuawei Xie 	return regionidx_hpa;
25829915bb1fSHuawei Xie }
25839915bb1fSHuawei Xie 
25849915bb1fSHuawei Xie /*
2585d19533e8SHuawei Xie  * A new device is added to a data core. First the device is added to the main linked list
2586d19533e8SHuawei Xie  * and the allocated to a specific data core.
2587d19533e8SHuawei Xie  */
2588d19533e8SHuawei Xie static int
2589d19533e8SHuawei Xie new_device (struct virtio_net *dev)
2590d19533e8SHuawei Xie {
2591d19533e8SHuawei Xie 	struct virtio_net_data_ll *ll_dev;
2592d19533e8SHuawei Xie 	int lcore, core_add = 0;
2593d19533e8SHuawei Xie 	uint32_t device_num_min = num_devices;
2594e571e6b4SHuawei Xie 	struct vhost_dev *vdev;
25959915bb1fSHuawei Xie 	uint32_t regionidx;
2596e571e6b4SHuawei Xie 
2597fdf20fa7SSergio Gonzalez Monroy 	vdev = rte_zmalloc("vhost device", sizeof(*vdev), RTE_CACHE_LINE_SIZE);
2598e571e6b4SHuawei Xie 	if (vdev == NULL) {
2599e571e6b4SHuawei Xie 		RTE_LOG(INFO, VHOST_DATA, "(%"PRIu64") Couldn't allocate memory for vhost dev\n",
2600e571e6b4SHuawei Xie 			dev->device_fh);
2601e571e6b4SHuawei Xie 		return -1;
2602e571e6b4SHuawei Xie 	}
2603e571e6b4SHuawei Xie 	vdev->dev = dev;
2604e571e6b4SHuawei Xie 	dev->priv = vdev;
2605d19533e8SHuawei Xie 
26069915bb1fSHuawei Xie 	if (zero_copy) {
26079915bb1fSHuawei Xie 		vdev->nregions_hpa = dev->mem->nregions;
26089915bb1fSHuawei Xie 		for (regionidx = 0; regionidx < dev->mem->nregions; regionidx++) {
26099915bb1fSHuawei Xie 			vdev->nregions_hpa
26109915bb1fSHuawei Xie 				+= check_hpa_regions(
26119915bb1fSHuawei Xie 					dev->mem->regions[regionidx].guest_phys_address
26129915bb1fSHuawei Xie 					+ dev->mem->regions[regionidx].address_offset,
26139915bb1fSHuawei Xie 					dev->mem->regions[regionidx].memory_size);
26149915bb1fSHuawei Xie 
26159915bb1fSHuawei Xie 		}
26169915bb1fSHuawei Xie 
2617435eb142SStephen Hemminger 		vdev->regions_hpa = rte_calloc("vhost hpa region",
2618435eb142SStephen Hemminger 					       vdev->nregions_hpa,
2619435eb142SStephen Hemminger 					       sizeof(struct virtio_memory_regions_hpa),
2620fdf20fa7SSergio Gonzalez Monroy 					       RTE_CACHE_LINE_SIZE);
26219915bb1fSHuawei Xie 		if (vdev->regions_hpa == NULL) {
26229915bb1fSHuawei Xie 			RTE_LOG(ERR, VHOST_CONFIG, "Cannot allocate memory for hpa region\n");
26239915bb1fSHuawei Xie 			rte_free(vdev);
26249915bb1fSHuawei Xie 			return -1;
26259915bb1fSHuawei Xie 		}
26269915bb1fSHuawei Xie 
26279915bb1fSHuawei Xie 
26289915bb1fSHuawei Xie 		if (fill_hpa_memory_regions(
26299915bb1fSHuawei Xie 			vdev->regions_hpa, dev->mem
26309915bb1fSHuawei Xie 			) != vdev->nregions_hpa) {
26319915bb1fSHuawei Xie 
26329915bb1fSHuawei Xie 			RTE_LOG(ERR, VHOST_CONFIG,
26339915bb1fSHuawei Xie 				"hpa memory regions number mismatch: "
26349915bb1fSHuawei Xie 				"[%d]\n", vdev->nregions_hpa);
26359915bb1fSHuawei Xie 			rte_free(vdev->regions_hpa);
26369915bb1fSHuawei Xie 			rte_free(vdev);
26379915bb1fSHuawei Xie 			return -1;
26389915bb1fSHuawei Xie 		}
26399915bb1fSHuawei Xie 	}
26409915bb1fSHuawei Xie 
26419915bb1fSHuawei Xie 
2642d19533e8SHuawei Xie 	/* Add device to main ll */
2643d19533e8SHuawei Xie 	ll_dev = get_data_ll_free_entry(&ll_root_free);
2644d19533e8SHuawei Xie 	if (ll_dev == NULL) {
2645d19533e8SHuawei Xie 		RTE_LOG(INFO, VHOST_DATA, "(%"PRIu64") No free entry found in linked list. Device limit "
2646d19533e8SHuawei Xie 			"of %d devices per core has been reached\n",
2647d19533e8SHuawei Xie 			dev->device_fh, num_devices);
26489915bb1fSHuawei Xie 		if (vdev->regions_hpa)
26499915bb1fSHuawei Xie 			rte_free(vdev->regions_hpa);
2650e571e6b4SHuawei Xie 		rte_free(vdev);
2651d19533e8SHuawei Xie 		return -1;
2652d19533e8SHuawei Xie 	}
2653e571e6b4SHuawei Xie 	ll_dev->vdev = vdev;
2654d19533e8SHuawei Xie 	add_data_ll_entry(&ll_root_used, ll_dev);
2655e571e6b4SHuawei Xie 	vdev->vmdq_rx_q
265684b02d16SHuawei Xie 		= dev->device_fh * queues_per_pool + vmdq_queue_base;
2657d19533e8SHuawei Xie 
2658d19533e8SHuawei Xie 	if (zero_copy) {
2659e571e6b4SHuawei Xie 		uint32_t index = vdev->vmdq_rx_q;
2660d19533e8SHuawei Xie 		uint32_t count_in_ring, i;
2661d19533e8SHuawei Xie 		struct mbuf_table *tx_q;
2662d19533e8SHuawei Xie 
2663d19533e8SHuawei Xie 		count_in_ring = rte_ring_count(vpool_array[index].ring);
2664d19533e8SHuawei Xie 
2665d19533e8SHuawei Xie 		LOG_DEBUG(VHOST_CONFIG,
2666d19533e8SHuawei Xie 			"(%"PRIu64") in new_device: mbuf count in mempool "
2667d19533e8SHuawei Xie 			"before attach is: %d\n",
2668d19533e8SHuawei Xie 			dev->device_fh,
2669d19533e8SHuawei Xie 			rte_mempool_count(vpool_array[index].pool));
2670d19533e8SHuawei Xie 		LOG_DEBUG(VHOST_CONFIG,
2671d19533e8SHuawei Xie 			"(%"PRIu64") in new_device: mbuf count in  ring "
2672d19533e8SHuawei Xie 			"before attach  is : %d\n",
2673d19533e8SHuawei Xie 			dev->device_fh, count_in_ring);
2674d19533e8SHuawei Xie 
2675d19533e8SHuawei Xie 		/*
2676d19533e8SHuawei Xie 		 * Attach all mbufs in vpool.ring and put back intovpool.pool.
2677d19533e8SHuawei Xie 		 */
2678d19533e8SHuawei Xie 		for (i = 0; i < count_in_ring; i++)
2679d19533e8SHuawei Xie 			attach_rxmbuf_zcp(dev);
2680d19533e8SHuawei Xie 
2681d19533e8SHuawei Xie 		LOG_DEBUG(VHOST_CONFIG, "(%"PRIu64") in new_device: mbuf count in "
2682d19533e8SHuawei Xie 			"mempool after attach is: %d\n",
2683d19533e8SHuawei Xie 			dev->device_fh,
2684d19533e8SHuawei Xie 			rte_mempool_count(vpool_array[index].pool));
2685d19533e8SHuawei Xie 		LOG_DEBUG(VHOST_CONFIG, "(%"PRIu64") in new_device: mbuf count in "
2686d19533e8SHuawei Xie 			"ring after attach  is : %d\n",
2687d19533e8SHuawei Xie 			dev->device_fh,
2688d19533e8SHuawei Xie 			rte_ring_count(vpool_array[index].ring));
2689d19533e8SHuawei Xie 
2690e571e6b4SHuawei Xie 		tx_q = &tx_queue_zcp[(uint16_t)vdev->vmdq_rx_q];
2691e571e6b4SHuawei Xie 		tx_q->txq_id = vdev->vmdq_rx_q;
2692d19533e8SHuawei Xie 
2693e571e6b4SHuawei Xie 		if (rte_eth_dev_tx_queue_start(ports[0], vdev->vmdq_rx_q) != 0) {
2694e571e6b4SHuawei Xie 			struct vpool *vpool = &vpool_array[vdev->vmdq_rx_q];
2695d19533e8SHuawei Xie 
2696d19533e8SHuawei Xie 			LOG_DEBUG(VHOST_CONFIG,
2697d19533e8SHuawei Xie 				"(%"PRIu64") In new_device: Failed to start "
2698d19533e8SHuawei Xie 				"tx queue:%d\n",
2699e571e6b4SHuawei Xie 				dev->device_fh, vdev->vmdq_rx_q);
2700d19533e8SHuawei Xie 
2701d19533e8SHuawei Xie 			mbuf_destroy_zcp(vpool);
27029915bb1fSHuawei Xie 			rte_free(vdev->regions_hpa);
2703e571e6b4SHuawei Xie 			rte_free(vdev);
2704d19533e8SHuawei Xie 			return -1;
2705d19533e8SHuawei Xie 		}
2706d19533e8SHuawei Xie 
2707e571e6b4SHuawei Xie 		if (rte_eth_dev_rx_queue_start(ports[0], vdev->vmdq_rx_q) != 0) {
2708e571e6b4SHuawei Xie 			struct vpool *vpool = &vpool_array[vdev->vmdq_rx_q];
2709d19533e8SHuawei Xie 
2710d19533e8SHuawei Xie 			LOG_DEBUG(VHOST_CONFIG,
2711d19533e8SHuawei Xie 				"(%"PRIu64") In new_device: Failed to start "
2712d19533e8SHuawei Xie 				"rx queue:%d\n",
2713e571e6b4SHuawei Xie 				dev->device_fh, vdev->vmdq_rx_q);
2714d19533e8SHuawei Xie 
2715d19533e8SHuawei Xie 			/* Stop the TX queue. */
2716d19533e8SHuawei Xie 			if (rte_eth_dev_tx_queue_stop(ports[0],
2717e571e6b4SHuawei Xie 				vdev->vmdq_rx_q) != 0) {
2718d19533e8SHuawei Xie 				LOG_DEBUG(VHOST_CONFIG,
2719d19533e8SHuawei Xie 					"(%"PRIu64") In new_device: Failed to "
2720d19533e8SHuawei Xie 					"stop tx queue:%d\n",
2721e571e6b4SHuawei Xie 					dev->device_fh, vdev->vmdq_rx_q);
2722d19533e8SHuawei Xie 			}
2723d19533e8SHuawei Xie 
2724d19533e8SHuawei Xie 			mbuf_destroy_zcp(vpool);
27259915bb1fSHuawei Xie 			rte_free(vdev->regions_hpa);
2726e571e6b4SHuawei Xie 			rte_free(vdev);
2727d19533e8SHuawei Xie 			return -1;
2728d19533e8SHuawei Xie 		}
2729d19533e8SHuawei Xie 
2730d19533e8SHuawei Xie 	}
2731d19533e8SHuawei Xie 
2732d19533e8SHuawei Xie 	/*reset ready flag*/
2733e571e6b4SHuawei Xie 	vdev->ready = DEVICE_MAC_LEARNING;
2734e571e6b4SHuawei Xie 	vdev->remove = 0;
2735d19533e8SHuawei Xie 
2736d19533e8SHuawei Xie 	/* Find a suitable lcore to add the device. */
2737d19533e8SHuawei Xie 	RTE_LCORE_FOREACH_SLAVE(lcore) {
2738d19533e8SHuawei Xie 		if (lcore_info[lcore].lcore_ll->device_num < device_num_min) {
2739d19533e8SHuawei Xie 			device_num_min = lcore_info[lcore].lcore_ll->device_num;
2740d19533e8SHuawei Xie 			core_add = lcore;
2741d19533e8SHuawei Xie 		}
2742d19533e8SHuawei Xie 	}
2743d19533e8SHuawei Xie 	/* Add device to lcore ll */
2744a981294bSHuawei Xie 	ll_dev = get_data_ll_free_entry(&lcore_info[core_add].lcore_ll->ll_root_free);
2745d19533e8SHuawei Xie 	if (ll_dev == NULL) {
2746d19533e8SHuawei Xie 		RTE_LOG(INFO, VHOST_DATA, "(%"PRIu64") Failed to add device to data core\n", dev->device_fh);
2747e571e6b4SHuawei Xie 		vdev->ready = DEVICE_SAFE_REMOVE;
2748d19533e8SHuawei Xie 		destroy_device(dev);
27499915bb1fSHuawei Xie 		rte_free(vdev->regions_hpa);
2750e571e6b4SHuawei Xie 		rte_free(vdev);
2751d19533e8SHuawei Xie 		return -1;
2752d19533e8SHuawei Xie 	}
2753e571e6b4SHuawei Xie 	ll_dev->vdev = vdev;
2754e571e6b4SHuawei Xie 	vdev->coreid = core_add;
2755e571e6b4SHuawei Xie 
2756a981294bSHuawei Xie 	add_data_ll_entry(&lcore_info[vdev->coreid].lcore_ll->ll_root_used, ll_dev);
2757d19533e8SHuawei Xie 
2758d19533e8SHuawei Xie 	/* Initialize device stats */
2759d19533e8SHuawei Xie 	memset(&dev_statistics[dev->device_fh], 0, sizeof(struct device_statistics));
2760d19533e8SHuawei Xie 
2761d19533e8SHuawei Xie 	/* Disable notifications. */
2762b5967c1fSHuawei Xie 	rte_vhost_enable_guest_notification(dev, VIRTIO_RXQ, 0);
2763b5967c1fSHuawei Xie 	rte_vhost_enable_guest_notification(dev, VIRTIO_TXQ, 0);
2764e571e6b4SHuawei Xie 	lcore_info[vdev->coreid].lcore_ll->device_num++;
2765d19533e8SHuawei Xie 	dev->flags |= VIRTIO_DEV_RUNNING;
2766d19533e8SHuawei Xie 
2767e571e6b4SHuawei Xie 	RTE_LOG(INFO, VHOST_DATA, "(%"PRIu64") Device has been added to data core %d\n", dev->device_fh, vdev->coreid);
2768d19533e8SHuawei Xie 
2769d19533e8SHuawei Xie 	return 0;
2770d19533e8SHuawei Xie }
2771d19533e8SHuawei Xie 
2772d19533e8SHuawei Xie /*
2773d19533e8SHuawei Xie  * These callback allow devices to be added to the data core when configuration
2774d19533e8SHuawei Xie  * has been fully complete.
2775d19533e8SHuawei Xie  */
2776d19533e8SHuawei Xie static const struct virtio_net_device_ops virtio_net_device_ops =
2777d19533e8SHuawei Xie {
2778d19533e8SHuawei Xie 	.new_device =  new_device,
2779d19533e8SHuawei Xie 	.destroy_device = destroy_device,
2780d19533e8SHuawei Xie };
2781d19533e8SHuawei Xie 
2782d19533e8SHuawei Xie /*
2783d19533e8SHuawei Xie  * This is a thread will wake up after a period to print stats if the user has
2784d19533e8SHuawei Xie  * enabled them.
2785d19533e8SHuawei Xie  */
2786d19533e8SHuawei Xie static void
2787d19533e8SHuawei Xie print_stats(void)
2788d19533e8SHuawei Xie {
2789d19533e8SHuawei Xie 	struct virtio_net_data_ll *dev_ll;
2790d19533e8SHuawei Xie 	uint64_t tx_dropped, rx_dropped;
2791d19533e8SHuawei Xie 	uint64_t tx, tx_total, rx, rx_total;
2792d19533e8SHuawei Xie 	uint32_t device_fh;
2793d19533e8SHuawei Xie 	const char clr[] = { 27, '[', '2', 'J', '\0' };
2794d19533e8SHuawei Xie 	const char top_left[] = { 27, '[', '1', ';', '1', 'H','\0' };
2795d19533e8SHuawei Xie 
2796d19533e8SHuawei Xie 	while(1) {
2797d19533e8SHuawei Xie 		sleep(enable_stats);
2798d19533e8SHuawei Xie 
2799d19533e8SHuawei Xie 		/* Clear screen and move to top left */
2800d19533e8SHuawei Xie 		printf("%s%s", clr, top_left);
2801d19533e8SHuawei Xie 
2802d19533e8SHuawei Xie 		printf("\nDevice statistics ====================================");
2803d19533e8SHuawei Xie 
2804d19533e8SHuawei Xie 		dev_ll = ll_root_used;
2805d19533e8SHuawei Xie 		while (dev_ll != NULL) {
2806e571e6b4SHuawei Xie 			device_fh = (uint32_t)dev_ll->vdev->dev->device_fh;
2807d19533e8SHuawei Xie 			tx_total = dev_statistics[device_fh].tx_total;
2808d19533e8SHuawei Xie 			tx = dev_statistics[device_fh].tx;
2809d19533e8SHuawei Xie 			tx_dropped = tx_total - tx;
2810d19533e8SHuawei Xie 			if (zero_copy == 0) {
2811d19533e8SHuawei Xie 				rx_total = rte_atomic64_read(
2812d19533e8SHuawei Xie 					&dev_statistics[device_fh].rx_total_atomic);
2813d19533e8SHuawei Xie 				rx = rte_atomic64_read(
2814d19533e8SHuawei Xie 					&dev_statistics[device_fh].rx_atomic);
2815d19533e8SHuawei Xie 			} else {
2816d19533e8SHuawei Xie 				rx_total = dev_statistics[device_fh].rx_total;
2817d19533e8SHuawei Xie 				rx = dev_statistics[device_fh].rx;
2818d19533e8SHuawei Xie 			}
2819d19533e8SHuawei Xie 			rx_dropped = rx_total - rx;
2820d19533e8SHuawei Xie 
2821d19533e8SHuawei Xie 			printf("\nStatistics for device %"PRIu32" ------------------------------"
2822d19533e8SHuawei Xie 					"\nTX total: 		%"PRIu64""
2823d19533e8SHuawei Xie 					"\nTX dropped: 		%"PRIu64""
2824d19533e8SHuawei Xie 					"\nTX successful: 		%"PRIu64""
2825d19533e8SHuawei Xie 					"\nRX total: 		%"PRIu64""
2826d19533e8SHuawei Xie 					"\nRX dropped: 		%"PRIu64""
2827d19533e8SHuawei Xie 					"\nRX successful: 		%"PRIu64"",
2828d19533e8SHuawei Xie 					device_fh,
2829d19533e8SHuawei Xie 					tx_total,
2830d19533e8SHuawei Xie 					tx_dropped,
2831d19533e8SHuawei Xie 					tx,
2832d19533e8SHuawei Xie 					rx_total,
2833d19533e8SHuawei Xie 					rx_dropped,
2834d19533e8SHuawei Xie 					rx);
2835d19533e8SHuawei Xie 
2836d19533e8SHuawei Xie 			dev_ll = dev_ll->next;
2837d19533e8SHuawei Xie 		}
2838d19533e8SHuawei Xie 		printf("\n======================================================\n");
2839d19533e8SHuawei Xie 	}
2840d19533e8SHuawei Xie }
2841d19533e8SHuawei Xie 
2842d19533e8SHuawei Xie static void
2843d19533e8SHuawei Xie setup_mempool_tbl(int socket, uint32_t index, char *pool_name,
2844d19533e8SHuawei Xie 	char *ring_name, uint32_t nb_mbuf)
2845d19533e8SHuawei Xie {
2846ea0c20eaSOlivier Matz 	vpool_array[index].pool	= rte_pktmbuf_pool_create(pool_name, nb_mbuf,
2847ea0c20eaSOlivier Matz 		MBUF_CACHE_SIZE_ZCP, 0, MBUF_DATA_SIZE_ZCP, socket);
2848d19533e8SHuawei Xie 	if (vpool_array[index].pool != NULL) {
2849d19533e8SHuawei Xie 		vpool_array[index].ring
2850d19533e8SHuawei Xie 			= rte_ring_create(ring_name,
2851d19533e8SHuawei Xie 				rte_align32pow2(nb_mbuf + 1),
2852d19533e8SHuawei Xie 				socket, RING_F_SP_ENQ | RING_F_SC_DEQ);
2853d19533e8SHuawei Xie 		if (likely(vpool_array[index].ring != NULL)) {
2854d19533e8SHuawei Xie 			LOG_DEBUG(VHOST_CONFIG,
2855d19533e8SHuawei Xie 				"in setup_mempool_tbl: mbuf count in "
2856d19533e8SHuawei Xie 				"mempool is: %d\n",
2857d19533e8SHuawei Xie 				rte_mempool_count(vpool_array[index].pool));
2858d19533e8SHuawei Xie 			LOG_DEBUG(VHOST_CONFIG,
2859d19533e8SHuawei Xie 				"in setup_mempool_tbl: mbuf count in "
2860d19533e8SHuawei Xie 				"ring   is: %d\n",
2861d19533e8SHuawei Xie 				rte_ring_count(vpool_array[index].ring));
2862d19533e8SHuawei Xie 		} else {
2863d19533e8SHuawei Xie 			rte_exit(EXIT_FAILURE, "ring_create(%s) failed",
2864d19533e8SHuawei Xie 				ring_name);
2865d19533e8SHuawei Xie 		}
2866d19533e8SHuawei Xie 
2867d19533e8SHuawei Xie 		/* Need consider head room. */
28681d493a49SOlivier Matz 		vpool_array[index].buf_size = VIRTIO_DESCRIPTOR_LEN_ZCP;
2869d19533e8SHuawei Xie 	} else {
2870d19533e8SHuawei Xie 		rte_exit(EXIT_FAILURE, "mempool_create(%s) failed", pool_name);
2871d19533e8SHuawei Xie 	}
2872d19533e8SHuawei Xie }
2873d19533e8SHuawei Xie 
2874c83d2d00SOuyang Changchun /* When we receive a INT signal, unregister vhost driver */
2875c83d2d00SOuyang Changchun static void
2876c83d2d00SOuyang Changchun sigint_handler(__rte_unused int signum)
2877c83d2d00SOuyang Changchun {
2878c83d2d00SOuyang Changchun 	/* Unregister vhost driver. */
2879c83d2d00SOuyang Changchun 	int ret = rte_vhost_driver_unregister((char *)&dev_basename);
2880c83d2d00SOuyang Changchun 	if (ret != 0)
2881c83d2d00SOuyang Changchun 		rte_exit(EXIT_FAILURE, "vhost driver unregister failure.\n");
2882c83d2d00SOuyang Changchun 	exit(0);
2883c83d2d00SOuyang Changchun }
2884d19533e8SHuawei Xie 
2885d19533e8SHuawei Xie /*
2886d19533e8SHuawei Xie  * Main function, does initialisation and calls the per-lcore functions. The CUSE
2887d19533e8SHuawei Xie  * device is also registered here to handle the IOCTLs.
2888d19533e8SHuawei Xie  */
2889d19533e8SHuawei Xie int
289098a16481SDavid Marchand main(int argc, char *argv[])
2891d19533e8SHuawei Xie {
2892d19533e8SHuawei Xie 	struct rte_mempool *mbuf_pool = NULL;
2893d19533e8SHuawei Xie 	unsigned lcore_id, core_id = 0;
2894d19533e8SHuawei Xie 	unsigned nb_ports, valid_num_ports;
2895d19533e8SHuawei Xie 	int ret;
289684b02d16SHuawei Xie 	uint8_t portid;
289784b02d16SHuawei Xie 	uint16_t queue_id;
2898d19533e8SHuawei Xie 	static pthread_t tid;
2899*67b6d303SRavi Kerur 	char thread_name[RTE_MAX_THREAD_NAME_LEN];
2900d19533e8SHuawei Xie 
2901c83d2d00SOuyang Changchun 	signal(SIGINT, sigint_handler);
2902c83d2d00SOuyang Changchun 
2903d19533e8SHuawei Xie 	/* init EAL */
2904d19533e8SHuawei Xie 	ret = rte_eal_init(argc, argv);
2905d19533e8SHuawei Xie 	if (ret < 0)
2906d19533e8SHuawei Xie 		rte_exit(EXIT_FAILURE, "Error with EAL initialization\n");
2907d19533e8SHuawei Xie 	argc -= ret;
2908d19533e8SHuawei Xie 	argv += ret;
2909d19533e8SHuawei Xie 
2910d19533e8SHuawei Xie 	/* parse app arguments */
2911d19533e8SHuawei Xie 	ret = us_vhost_parse_args(argc, argv);
2912d19533e8SHuawei Xie 	if (ret < 0)
2913d19533e8SHuawei Xie 		rte_exit(EXIT_FAILURE, "Invalid argument\n");
2914d19533e8SHuawei Xie 
2915d19533e8SHuawei Xie 	for (lcore_id = 0; lcore_id < RTE_MAX_LCORE; lcore_id ++)
2916d19533e8SHuawei Xie 		if (rte_lcore_is_enabled(lcore_id))
2917d19533e8SHuawei Xie 			lcore_ids[core_id ++] = lcore_id;
2918d19533e8SHuawei Xie 
2919d19533e8SHuawei Xie 	if (rte_lcore_count() > RTE_MAX_LCORE)
2920d19533e8SHuawei Xie 		rte_exit(EXIT_FAILURE,"Not enough cores\n");
2921d19533e8SHuawei Xie 
2922d19533e8SHuawei Xie 	/*set the number of swithcing cores available*/
2923d19533e8SHuawei Xie 	num_switching_cores = rte_lcore_count()-1;
2924d19533e8SHuawei Xie 
2925d19533e8SHuawei Xie 	/* Get the number of physical ports. */
2926d19533e8SHuawei Xie 	nb_ports = rte_eth_dev_count();
2927d19533e8SHuawei Xie 	if (nb_ports > RTE_MAX_ETHPORTS)
2928d19533e8SHuawei Xie 		nb_ports = RTE_MAX_ETHPORTS;
2929d19533e8SHuawei Xie 
2930d19533e8SHuawei Xie 	/*
2931d19533e8SHuawei Xie 	 * Update the global var NUM_PORTS and global array PORTS
2932d19533e8SHuawei Xie 	 * and get value of var VALID_NUM_PORTS according to system ports number
2933d19533e8SHuawei Xie 	 */
2934d19533e8SHuawei Xie 	valid_num_ports = check_ports_num(nb_ports);
2935d19533e8SHuawei Xie 
2936d19533e8SHuawei Xie 	if ((valid_num_ports ==  0) || (valid_num_ports > MAX_SUP_PORTS)) {
2937d19533e8SHuawei Xie 		RTE_LOG(INFO, VHOST_PORT, "Current enabled port number is %u,"
2938d19533e8SHuawei Xie 			"but only %u port can be enabled\n",num_ports, MAX_SUP_PORTS);
2939d19533e8SHuawei Xie 		return -1;
2940d19533e8SHuawei Xie 	}
2941d19533e8SHuawei Xie 
2942d19533e8SHuawei Xie 	if (zero_copy == 0) {
2943d19533e8SHuawei Xie 		/* Create the mbuf pool. */
2944ea0c20eaSOlivier Matz 		mbuf_pool = rte_pktmbuf_pool_create("MBUF_POOL",
2945ea0c20eaSOlivier Matz 			NUM_MBUFS_PER_PORT * valid_num_ports, MBUF_CACHE_SIZE,
2946ea0c20eaSOlivier Matz 			0, MBUF_DATA_SIZE, rte_socket_id());
2947d19533e8SHuawei Xie 		if (mbuf_pool == NULL)
2948d19533e8SHuawei Xie 			rte_exit(EXIT_FAILURE, "Cannot create mbuf pool\n");
2949d19533e8SHuawei Xie 
2950d19533e8SHuawei Xie 		for (queue_id = 0; queue_id < MAX_QUEUES + 1; queue_id++)
2951d19533e8SHuawei Xie 			vpool_array[queue_id].pool = mbuf_pool;
2952d19533e8SHuawei Xie 
2953d19533e8SHuawei Xie 		if (vm2vm_mode == VM2VM_HARDWARE) {
2954d19533e8SHuawei Xie 			/* Enable VT loop back to let L2 switch to do it. */
2955d19533e8SHuawei Xie 			vmdq_conf_default.rx_adv_conf.vmdq_rx_conf.enable_loop_back = 1;
2956d19533e8SHuawei Xie 			LOG_DEBUG(VHOST_CONFIG,
2957d19533e8SHuawei Xie 				"Enable loop back for L2 switch in vmdq.\n");
2958d19533e8SHuawei Xie 		}
2959d19533e8SHuawei Xie 	} else {
2960d19533e8SHuawei Xie 		uint32_t nb_mbuf;
2961d19533e8SHuawei Xie 		char pool_name[RTE_MEMPOOL_NAMESIZE];
2962d19533e8SHuawei Xie 		char ring_name[RTE_MEMPOOL_NAMESIZE];
2963d19533e8SHuawei Xie 
2964d19533e8SHuawei Xie 		nb_mbuf = num_rx_descriptor
2965d19533e8SHuawei Xie 			+ num_switching_cores * MBUF_CACHE_SIZE_ZCP
2966d19533e8SHuawei Xie 			+ num_switching_cores * MAX_PKT_BURST;
2967d19533e8SHuawei Xie 
2968d19533e8SHuawei Xie 		for (queue_id = 0; queue_id < MAX_QUEUES; queue_id++) {
2969d19533e8SHuawei Xie 			snprintf(pool_name, sizeof(pool_name),
2970d19533e8SHuawei Xie 				"rxmbuf_pool_%u", queue_id);
2971d19533e8SHuawei Xie 			snprintf(ring_name, sizeof(ring_name),
2972d19533e8SHuawei Xie 				"rxmbuf_ring_%u", queue_id);
2973d19533e8SHuawei Xie 			setup_mempool_tbl(rte_socket_id(), queue_id,
2974d19533e8SHuawei Xie 				pool_name, ring_name, nb_mbuf);
2975d19533e8SHuawei Xie 		}
2976d19533e8SHuawei Xie 
2977d19533e8SHuawei Xie 		nb_mbuf = num_tx_descriptor
2978d19533e8SHuawei Xie 				+ num_switching_cores * MBUF_CACHE_SIZE_ZCP
2979d19533e8SHuawei Xie 				+ num_switching_cores * MAX_PKT_BURST;
2980d19533e8SHuawei Xie 
2981d19533e8SHuawei Xie 		for (queue_id = 0; queue_id < MAX_QUEUES; queue_id++) {
2982d19533e8SHuawei Xie 			snprintf(pool_name, sizeof(pool_name),
2983d19533e8SHuawei Xie 				"txmbuf_pool_%u", queue_id);
2984d19533e8SHuawei Xie 			snprintf(ring_name, sizeof(ring_name),
2985d19533e8SHuawei Xie 				"txmbuf_ring_%u", queue_id);
2986d19533e8SHuawei Xie 			setup_mempool_tbl(rte_socket_id(),
2987d19533e8SHuawei Xie 				(queue_id + MAX_QUEUES),
2988d19533e8SHuawei Xie 				pool_name, ring_name, nb_mbuf);
2989d19533e8SHuawei Xie 		}
2990d19533e8SHuawei Xie 
2991d19533e8SHuawei Xie 		if (vm2vm_mode == VM2VM_HARDWARE) {
2992d19533e8SHuawei Xie 			/* Enable VT loop back to let L2 switch to do it. */
2993d19533e8SHuawei Xie 			vmdq_conf_default.rx_adv_conf.vmdq_rx_conf.enable_loop_back = 1;
2994d19533e8SHuawei Xie 			LOG_DEBUG(VHOST_CONFIG,
2995d19533e8SHuawei Xie 				"Enable loop back for L2 switch in vmdq.\n");
2996d19533e8SHuawei Xie 		}
2997d19533e8SHuawei Xie 	}
2998d19533e8SHuawei Xie 	/* Set log level. */
2999d19533e8SHuawei Xie 	rte_set_log_level(LOG_LEVEL);
3000d19533e8SHuawei Xie 
3001d19533e8SHuawei Xie 	/* initialize all ports */
3002d19533e8SHuawei Xie 	for (portid = 0; portid < nb_ports; portid++) {
3003d19533e8SHuawei Xie 		/* skip ports that are not enabled */
3004d19533e8SHuawei Xie 		if ((enabled_port_mask & (1 << portid)) == 0) {
3005d19533e8SHuawei Xie 			RTE_LOG(INFO, VHOST_PORT,
3006d19533e8SHuawei Xie 				"Skipping disabled port %d\n", portid);
3007d19533e8SHuawei Xie 			continue;
3008d19533e8SHuawei Xie 		}
3009d19533e8SHuawei Xie 		if (port_init(portid) != 0)
3010d19533e8SHuawei Xie 			rte_exit(EXIT_FAILURE,
3011d19533e8SHuawei Xie 				"Cannot initialize network ports\n");
3012d19533e8SHuawei Xie 	}
3013d19533e8SHuawei Xie 
3014d19533e8SHuawei Xie 	/* Initialise all linked lists. */
3015d19533e8SHuawei Xie 	if (init_data_ll() == -1)
3016d19533e8SHuawei Xie 		rte_exit(EXIT_FAILURE, "Failed to initialize linked list\n");
3017d19533e8SHuawei Xie 
3018d19533e8SHuawei Xie 	/* Initialize device stats */
3019d19533e8SHuawei Xie 	memset(&dev_statistics, 0, sizeof(dev_statistics));
3020d19533e8SHuawei Xie 
3021d19533e8SHuawei Xie 	/* Enable stats if the user option is set. */
3022*67b6d303SRavi Kerur 	if (enable_stats) {
3023*67b6d303SRavi Kerur 		ret = pthread_create(&tid, NULL, (void *)print_stats, NULL);
3024*67b6d303SRavi Kerur 		if (ret != 0)
3025*67b6d303SRavi Kerur 			rte_exit(EXIT_FAILURE,
3026*67b6d303SRavi Kerur 				"Cannot create print-stats thread\n");
3027*67b6d303SRavi Kerur 
3028*67b6d303SRavi Kerur 		/* Set thread_name for aid in debugging.  */
3029*67b6d303SRavi Kerur 		snprintf(thread_name, RTE_MAX_THREAD_NAME_LEN, "print-stats");
3030*67b6d303SRavi Kerur 		ret = pthread_setname_np(tid, thread_name);
3031*67b6d303SRavi Kerur 		if (ret != 0)
3032*67b6d303SRavi Kerur 			RTE_LOG(ERR, VHOST_CONFIG,
3033*67b6d303SRavi Kerur 				"Cannot set print-stats name\n");
3034*67b6d303SRavi Kerur 	}
3035d19533e8SHuawei Xie 
3036d19533e8SHuawei Xie 	/* Launch all data cores. */
3037d19533e8SHuawei Xie 	if (zero_copy == 0) {
3038d19533e8SHuawei Xie 		RTE_LCORE_FOREACH_SLAVE(lcore_id) {
3039d19533e8SHuawei Xie 			rte_eal_remote_launch(switch_worker,
3040d19533e8SHuawei Xie 				mbuf_pool, lcore_id);
3041d19533e8SHuawei Xie 		}
3042d19533e8SHuawei Xie 	} else {
3043d19533e8SHuawei Xie 		uint32_t count_in_mempool, index, i;
3044d19533e8SHuawei Xie 		for (index = 0; index < 2*MAX_QUEUES; index++) {
3045d19533e8SHuawei Xie 			/* For all RX and TX queues. */
3046d19533e8SHuawei Xie 			count_in_mempool
3047d19533e8SHuawei Xie 				= rte_mempool_count(vpool_array[index].pool);
3048d19533e8SHuawei Xie 
3049d19533e8SHuawei Xie 			/*
3050d19533e8SHuawei Xie 			 * Transfer all un-attached mbufs from vpool.pool
3051d19533e8SHuawei Xie 			 * to vpoo.ring.
3052d19533e8SHuawei Xie 			 */
3053d19533e8SHuawei Xie 			for (i = 0; i < count_in_mempool; i++) {
3054d19533e8SHuawei Xie 				struct rte_mbuf *mbuf
3055d19533e8SHuawei Xie 					= __rte_mbuf_raw_alloc(
3056d19533e8SHuawei Xie 						vpool_array[index].pool);
3057d19533e8SHuawei Xie 				rte_ring_sp_enqueue(vpool_array[index].ring,
3058d19533e8SHuawei Xie 						(void *)mbuf);
3059d19533e8SHuawei Xie 			}
3060d19533e8SHuawei Xie 
3061d19533e8SHuawei Xie 			LOG_DEBUG(VHOST_CONFIG,
306298a16481SDavid Marchand 				"in main: mbuf count in mempool at initial "
3063d19533e8SHuawei Xie 				"is: %d\n", count_in_mempool);
3064d19533e8SHuawei Xie 			LOG_DEBUG(VHOST_CONFIG,
306598a16481SDavid Marchand 				"in main: mbuf count in  ring at initial  is :"
3066d19533e8SHuawei Xie 				" %d\n",
3067d19533e8SHuawei Xie 				rte_ring_count(vpool_array[index].ring));
3068d19533e8SHuawei Xie 		}
3069d19533e8SHuawei Xie 
3070d19533e8SHuawei Xie 		RTE_LCORE_FOREACH_SLAVE(lcore_id)
3071d19533e8SHuawei Xie 			rte_eal_remote_launch(switch_worker_zcp, NULL,
3072d19533e8SHuawei Xie 				lcore_id);
3073d19533e8SHuawei Xie 	}
3074d19533e8SHuawei Xie 
307528deb020SHuawei Xie 	if (mergeable == 0)
307628deb020SHuawei Xie 		rte_vhost_feature_disable(1ULL << VIRTIO_NET_F_MRG_RXBUF);
307728deb020SHuawei Xie 
307829c0f3c8SOuyang Changchun 	/* Register vhost(cuse or user) driver to handle vhost messages. */
30795cf27144SHuawei Xie 	ret = rte_vhost_driver_register((char *)&dev_basename);
3080d19533e8SHuawei Xie 	if (ret != 0)
308129c0f3c8SOuyang Changchun 		rte_exit(EXIT_FAILURE, "vhost driver register failure.\n");
3082d19533e8SHuawei Xie 
30835cf27144SHuawei Xie 	rte_vhost_driver_callback_register(&virtio_net_device_ops);
3084d19533e8SHuawei Xie 
3085d19533e8SHuawei Xie 	/* Start CUSE session. */
30865cf27144SHuawei Xie 	rte_vhost_driver_session_start();
3087d19533e8SHuawei Xie 	return 0;
3088d19533e8SHuawei Xie 
3089d19533e8SHuawei Xie }
3090