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