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