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