xref: /dpdk/examples/vhost/main.c (revision e571e6b472a1e52eb39d4ded937ef426f7cd0be2)
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>
52d19533e8SHuawei Xie 
53d19533e8SHuawei Xie #include "main.h"
54d19533e8SHuawei Xie #include "virtio-net.h"
55d19533e8SHuawei Xie #include "vhost-net-cdev.h"
56d19533e8SHuawei Xie 
57d19533e8SHuawei Xie #define MAX_QUEUES 128
58d19533e8SHuawei Xie 
59d19533e8SHuawei Xie /* the maximum number of external ports supported */
60d19533e8SHuawei Xie #define MAX_SUP_PORTS 1
61d19533e8SHuawei Xie 
62d19533e8SHuawei Xie /*
63d19533e8SHuawei Xie  * Calculate the number of buffers needed per port
64d19533e8SHuawei Xie  */
65d19533e8SHuawei Xie #define NUM_MBUFS_PER_PORT ((MAX_QUEUES*RTE_TEST_RX_DESC_DEFAULT) +  		\
66d19533e8SHuawei Xie 							(num_switching_cores*MAX_PKT_BURST) +  			\
67d19533e8SHuawei Xie 							(num_switching_cores*RTE_TEST_TX_DESC_DEFAULT) +\
68d19533e8SHuawei Xie 							(num_switching_cores*MBUF_CACHE_SIZE))
69d19533e8SHuawei Xie 
70d19533e8SHuawei Xie #define MBUF_CACHE_SIZE 128
71d19533e8SHuawei Xie #define MBUF_SIZE (2048 + sizeof(struct rte_mbuf) + RTE_PKTMBUF_HEADROOM)
72d19533e8SHuawei Xie 
73d19533e8SHuawei Xie /*
74d19533e8SHuawei Xie  * No frame data buffer allocated from host are required for zero copy
75d19533e8SHuawei Xie  * implementation, guest will allocate the frame data buffer, and vhost
76d19533e8SHuawei Xie  * directly use it.
77d19533e8SHuawei Xie  */
78d19533e8SHuawei Xie #define VIRTIO_DESCRIPTOR_LEN_ZCP 1518
79d19533e8SHuawei Xie #define MBUF_SIZE_ZCP (VIRTIO_DESCRIPTOR_LEN_ZCP + sizeof(struct rte_mbuf) \
80d19533e8SHuawei Xie 	+ RTE_PKTMBUF_HEADROOM)
81d19533e8SHuawei Xie #define MBUF_CACHE_SIZE_ZCP 0
82d19533e8SHuawei Xie 
83d19533e8SHuawei Xie /*
84d19533e8SHuawei Xie  * RX and TX Prefetch, Host, and Write-back threshold values should be
85d19533e8SHuawei Xie  * carefully set for optimal performance. Consult the network
86d19533e8SHuawei Xie  * controller's datasheet and supporting DPDK documentation for guidance
87d19533e8SHuawei Xie  * on how these parameters should be set.
88d19533e8SHuawei Xie  */
89d19533e8SHuawei Xie #define RX_PTHRESH 8 /* Default values of RX prefetch threshold reg. */
90d19533e8SHuawei Xie #define RX_HTHRESH 8 /* Default values of RX host threshold reg. */
91d19533e8SHuawei Xie #define RX_WTHRESH 4 /* Default values of RX write-back threshold reg. */
92d19533e8SHuawei Xie 
93d19533e8SHuawei Xie /*
94d19533e8SHuawei Xie  * These default values are optimized for use with the Intel(R) 82599 10 GbE
95d19533e8SHuawei Xie  * Controller and the DPDK ixgbe PMD. Consider using other values for other
96d19533e8SHuawei Xie  * network controllers and/or network drivers.
97d19533e8SHuawei Xie  */
98d19533e8SHuawei Xie #define TX_PTHRESH 36 /* Default values of TX prefetch threshold reg. */
99d19533e8SHuawei Xie #define TX_HTHRESH 0  /* Default values of TX host threshold reg. */
100d19533e8SHuawei Xie #define TX_WTHRESH 0  /* Default values of TX write-back threshold reg. */
101d19533e8SHuawei Xie 
102d19533e8SHuawei Xie #define MAX_PKT_BURST 32 		/* Max burst size for RX/TX */
103d19533e8SHuawei Xie #define MAX_MRG_PKT_BURST 16 	/* Max burst for merge buffers. Set to 1 due to performance issue. */
104d19533e8SHuawei Xie #define BURST_TX_DRAIN_US 100 	/* TX drain every ~100us */
105d19533e8SHuawei Xie 
106d19533e8SHuawei Xie #define BURST_RX_WAIT_US 15 	/* Defines how long we wait between retries on RX */
107d19533e8SHuawei Xie #define BURST_RX_RETRIES 4		/* Number of retries on RX. */
108d19533e8SHuawei Xie 
109d19533e8SHuawei Xie #define JUMBO_FRAME_MAX_SIZE    0x2600
110d19533e8SHuawei Xie 
111d19533e8SHuawei Xie /* State of virtio device. */
112d19533e8SHuawei Xie #define DEVICE_MAC_LEARNING 0
113d19533e8SHuawei Xie #define DEVICE_RX			1
114d19533e8SHuawei Xie #define DEVICE_SAFE_REMOVE	2
115d19533e8SHuawei Xie 
116d19533e8SHuawei Xie /* Config_core_flag status definitions. */
117d19533e8SHuawei Xie #define REQUEST_DEV_REMOVAL 1
118d19533e8SHuawei Xie #define ACK_DEV_REMOVAL 0
119d19533e8SHuawei Xie 
120d19533e8SHuawei Xie /* Configurable number of RX/TX ring descriptors */
121d19533e8SHuawei Xie #define RTE_TEST_RX_DESC_DEFAULT 1024
122d19533e8SHuawei Xie #define RTE_TEST_TX_DESC_DEFAULT 512
123d19533e8SHuawei Xie 
124d19533e8SHuawei Xie /*
125d19533e8SHuawei Xie  * Need refine these 2 macros for legacy and DPDK based front end:
126d19533e8SHuawei Xie  * Max vring avail descriptor/entries from guest - MAX_PKT_BURST
127d19533e8SHuawei Xie  * And then adjust power 2.
128d19533e8SHuawei Xie  */
129d19533e8SHuawei Xie /*
130d19533e8SHuawei Xie  * For legacy front end, 128 descriptors,
131d19533e8SHuawei Xie  * half for virtio header, another half for mbuf.
132d19533e8SHuawei Xie  */
133d19533e8SHuawei Xie #define RTE_TEST_RX_DESC_DEFAULT_ZCP 32   /* legacy: 32, DPDK virt FE: 128. */
134d19533e8SHuawei Xie #define RTE_TEST_TX_DESC_DEFAULT_ZCP 64   /* legacy: 64, DPDK virt FE: 64.  */
135d19533e8SHuawei Xie 
136d19533e8SHuawei Xie /* Get first 4 bytes in mbuf headroom. */
137d19533e8SHuawei Xie #define MBUF_HEADROOM_UINT32(mbuf) (*(uint32_t *)((uint8_t *)(mbuf) \
138d19533e8SHuawei Xie 		+ sizeof(struct rte_mbuf)))
139d19533e8SHuawei Xie 
140d19533e8SHuawei Xie /* true if x is a power of 2 */
141d19533e8SHuawei Xie #define POWEROF2(x) ((((x)-1) & (x)) == 0)
142d19533e8SHuawei Xie 
143d19533e8SHuawei Xie #define INVALID_PORT_ID 0xFF
144d19533e8SHuawei Xie 
145d19533e8SHuawei Xie /* Max number of devices. Limited by vmdq. */
146d19533e8SHuawei Xie #define MAX_DEVICES 64
147d19533e8SHuawei Xie 
148d19533e8SHuawei Xie /* Size of buffers used for snprintfs. */
149d19533e8SHuawei Xie #define MAX_PRINT_BUFF 6072
150d19533e8SHuawei Xie 
151d19533e8SHuawei Xie /* Maximum character device basename size. */
152d19533e8SHuawei Xie #define MAX_BASENAME_SZ 10
153d19533e8SHuawei Xie 
154d19533e8SHuawei Xie /* Maximum long option length for option parsing. */
155d19533e8SHuawei Xie #define MAX_LONG_OPT_SZ 64
156d19533e8SHuawei Xie 
157d19533e8SHuawei Xie /* Used to compare MAC addresses. */
158d19533e8SHuawei Xie #define MAC_ADDR_CMP 0xFFFFFFFFFFFFULL
159d19533e8SHuawei Xie 
160d19533e8SHuawei Xie /* Number of descriptors per cacheline. */
161d19533e8SHuawei Xie #define DESC_PER_CACHELINE (CACHE_LINE_SIZE / sizeof(struct vring_desc))
162d19533e8SHuawei Xie 
163d19533e8SHuawei Xie /* mask of enabled ports */
164d19533e8SHuawei Xie static uint32_t enabled_port_mask = 0;
165d19533e8SHuawei Xie 
166d19533e8SHuawei Xie /*Number of switching cores enabled*/
167d19533e8SHuawei Xie static uint32_t num_switching_cores = 0;
168d19533e8SHuawei Xie 
169d19533e8SHuawei Xie /* number of devices/queues to support*/
170d19533e8SHuawei Xie static uint32_t num_queues = 0;
171d19533e8SHuawei Xie uint32_t num_devices = 0;
172d19533e8SHuawei Xie 
173d19533e8SHuawei Xie /*
174d19533e8SHuawei Xie  * Enable zero copy, pkts buffer will directly dma to hw descriptor,
175d19533e8SHuawei Xie  * disabled on default.
176d19533e8SHuawei Xie  */
177d19533e8SHuawei Xie static uint32_t zero_copy;
178d19533e8SHuawei Xie 
179d19533e8SHuawei Xie /* number of descriptors to apply*/
180d19533e8SHuawei Xie static uint32_t num_rx_descriptor = RTE_TEST_RX_DESC_DEFAULT_ZCP;
181d19533e8SHuawei Xie static uint32_t num_tx_descriptor = RTE_TEST_TX_DESC_DEFAULT_ZCP;
182d19533e8SHuawei Xie 
183d19533e8SHuawei Xie /* max ring descriptor, ixgbe, i40e, e1000 all are 4096. */
184d19533e8SHuawei Xie #define MAX_RING_DESC 4096
185d19533e8SHuawei Xie 
186d19533e8SHuawei Xie struct vpool {
187d19533e8SHuawei Xie 	struct rte_mempool *pool;
188d19533e8SHuawei Xie 	struct rte_ring *ring;
189d19533e8SHuawei Xie 	uint32_t buf_size;
190d19533e8SHuawei Xie } vpool_array[MAX_QUEUES+MAX_QUEUES];
191d19533e8SHuawei Xie 
192d19533e8SHuawei Xie /* Enable VM2VM communications. If this is disabled then the MAC address compare is skipped. */
193d19533e8SHuawei Xie typedef enum {
194d19533e8SHuawei Xie 	VM2VM_DISABLED = 0,
195d19533e8SHuawei Xie 	VM2VM_SOFTWARE = 1,
196d19533e8SHuawei Xie 	VM2VM_HARDWARE = 2,
197d19533e8SHuawei Xie 	VM2VM_LAST
198d19533e8SHuawei Xie } vm2vm_type;
199d19533e8SHuawei Xie static vm2vm_type vm2vm_mode = VM2VM_SOFTWARE;
200d19533e8SHuawei Xie 
201d19533e8SHuawei Xie /* The type of host physical address translated from guest physical address. */
202d19533e8SHuawei Xie typedef enum {
203d19533e8SHuawei Xie 	PHYS_ADDR_CONTINUOUS = 0,
204d19533e8SHuawei Xie 	PHYS_ADDR_CROSS_SUBREG = 1,
205d19533e8SHuawei Xie 	PHYS_ADDR_INVALID = 2,
206d19533e8SHuawei Xie 	PHYS_ADDR_LAST
207d19533e8SHuawei Xie } hpa_type;
208d19533e8SHuawei Xie 
209d19533e8SHuawei Xie /* Enable stats. */
210d19533e8SHuawei Xie static uint32_t enable_stats = 0;
211d19533e8SHuawei Xie /* Enable retries on RX. */
212d19533e8SHuawei Xie static uint32_t enable_retry = 1;
213d19533e8SHuawei Xie /* Specify timeout (in useconds) between retries on RX. */
214d19533e8SHuawei Xie static uint32_t burst_rx_delay_time = BURST_RX_WAIT_US;
215d19533e8SHuawei Xie /* Specify the number of retries on RX. */
216d19533e8SHuawei Xie static uint32_t burst_rx_retry_num = BURST_RX_RETRIES;
217d19533e8SHuawei Xie 
218d19533e8SHuawei Xie /* Character device basename. Can be set by user. */
219d19533e8SHuawei Xie static char dev_basename[MAX_BASENAME_SZ] = "vhost-net";
220d19533e8SHuawei Xie 
221d19533e8SHuawei Xie /* Charater device index. Can be set by user. */
222d19533e8SHuawei Xie static uint32_t dev_index = 0;
223d19533e8SHuawei Xie 
224d19533e8SHuawei Xie /* This can be set by the user so it is made available here. */
225d19533e8SHuawei Xie extern uint64_t VHOST_FEATURES;
226d19533e8SHuawei Xie 
227d19533e8SHuawei Xie /* Default configuration for rx and tx thresholds etc. */
228d19533e8SHuawei Xie static struct rte_eth_rxconf rx_conf_default = {
229d19533e8SHuawei Xie 	.rx_thresh = {
230d19533e8SHuawei Xie 		.pthresh = RX_PTHRESH,
231d19533e8SHuawei Xie 		.hthresh = RX_HTHRESH,
232d19533e8SHuawei Xie 		.wthresh = RX_WTHRESH,
233d19533e8SHuawei Xie 	},
234d19533e8SHuawei Xie 	.rx_drop_en = 1,
235d19533e8SHuawei Xie };
236d19533e8SHuawei Xie 
237d19533e8SHuawei Xie /*
238d19533e8SHuawei Xie  * These default values are optimized for use with the Intel(R) 82599 10 GbE
239d19533e8SHuawei Xie  * Controller and the DPDK ixgbe/igb PMD. Consider using other values for other
240d19533e8SHuawei Xie  * network controllers and/or network drivers.
241d19533e8SHuawei Xie  */
242d19533e8SHuawei Xie static struct rte_eth_txconf tx_conf_default = {
243d19533e8SHuawei Xie 	.tx_thresh = {
244d19533e8SHuawei Xie 		.pthresh = TX_PTHRESH,
245d19533e8SHuawei Xie 		.hthresh = TX_HTHRESH,
246d19533e8SHuawei Xie 		.wthresh = TX_WTHRESH,
247d19533e8SHuawei Xie 	},
248d19533e8SHuawei Xie 	.tx_free_thresh = 0, /* Use PMD default values */
249d19533e8SHuawei Xie 	.tx_rs_thresh = 0, /* Use PMD default values */
250d19533e8SHuawei Xie };
251d19533e8SHuawei Xie 
252d19533e8SHuawei Xie /* empty vmdq configuration structure. Filled in programatically */
253d19533e8SHuawei Xie static struct rte_eth_conf vmdq_conf_default = {
254d19533e8SHuawei Xie 	.rxmode = {
255d19533e8SHuawei Xie 		.mq_mode        = ETH_MQ_RX_VMDQ_ONLY,
256d19533e8SHuawei Xie 		.split_hdr_size = 0,
257d19533e8SHuawei Xie 		.header_split   = 0, /**< Header Split disabled */
258d19533e8SHuawei Xie 		.hw_ip_checksum = 0, /**< IP checksum offload disabled */
259d19533e8SHuawei Xie 		.hw_vlan_filter = 0, /**< VLAN filtering disabled */
260d19533e8SHuawei Xie 		/*
261d19533e8SHuawei Xie 		 * It is necessary for 1G NIC such as I350,
262d19533e8SHuawei Xie 		 * this fixes bug of ipv4 forwarding in guest can't
263d19533e8SHuawei Xie 		 * forward pakets from one virtio dev to another virtio dev.
264d19533e8SHuawei Xie 		 */
265d19533e8SHuawei Xie 		.hw_vlan_strip  = 1, /**< VLAN strip enabled. */
266d19533e8SHuawei Xie 		.jumbo_frame    = 0, /**< Jumbo Frame Support disabled */
267d19533e8SHuawei Xie 		.hw_strip_crc   = 0, /**< CRC stripped by hardware */
268d19533e8SHuawei Xie 	},
269d19533e8SHuawei Xie 
270d19533e8SHuawei Xie 	.txmode = {
271d19533e8SHuawei Xie 		.mq_mode = ETH_MQ_TX_NONE,
272d19533e8SHuawei Xie 	},
273d19533e8SHuawei Xie 	.rx_adv_conf = {
274d19533e8SHuawei Xie 		/*
275d19533e8SHuawei Xie 		 * should be overridden separately in code with
276d19533e8SHuawei Xie 		 * appropriate values
277d19533e8SHuawei Xie 		 */
278d19533e8SHuawei Xie 		.vmdq_rx_conf = {
279d19533e8SHuawei Xie 			.nb_queue_pools = ETH_8_POOLS,
280d19533e8SHuawei Xie 			.enable_default_pool = 0,
281d19533e8SHuawei Xie 			.default_pool = 0,
282d19533e8SHuawei Xie 			.nb_pool_maps = 0,
283d19533e8SHuawei Xie 			.pool_map = {{0, 0},},
284d19533e8SHuawei Xie 		},
285d19533e8SHuawei Xie 	},
286d19533e8SHuawei Xie };
287d19533e8SHuawei Xie 
288d19533e8SHuawei Xie static unsigned lcore_ids[RTE_MAX_LCORE];
289d19533e8SHuawei Xie static uint8_t ports[RTE_MAX_ETHPORTS];
290d19533e8SHuawei Xie static unsigned num_ports = 0; /**< The number of ports specified in command line */
291d19533e8SHuawei Xie 
292d19533e8SHuawei Xie static const uint16_t external_pkt_default_vlan_tag = 2000;
293d19533e8SHuawei Xie const uint16_t vlan_tags[] = {
294d19533e8SHuawei Xie 	1000, 1001, 1002, 1003, 1004, 1005, 1006, 1007,
295d19533e8SHuawei Xie 	1008, 1009, 1010, 1011,	1012, 1013, 1014, 1015,
296d19533e8SHuawei Xie 	1016, 1017, 1018, 1019, 1020, 1021, 1022, 1023,
297d19533e8SHuawei Xie 	1024, 1025, 1026, 1027, 1028, 1029, 1030, 1031,
298d19533e8SHuawei Xie 	1032, 1033, 1034, 1035, 1036, 1037, 1038, 1039,
299d19533e8SHuawei Xie 	1040, 1041, 1042, 1043, 1044, 1045, 1046, 1047,
300d19533e8SHuawei Xie 	1048, 1049, 1050, 1051, 1052, 1053, 1054, 1055,
301d19533e8SHuawei Xie 	1056, 1057, 1058, 1059, 1060, 1061, 1062, 1063,
302d19533e8SHuawei Xie };
303d19533e8SHuawei Xie 
304d19533e8SHuawei Xie /* ethernet addresses of ports */
305d19533e8SHuawei Xie static struct ether_addr vmdq_ports_eth_addr[RTE_MAX_ETHPORTS];
306d19533e8SHuawei Xie 
307d19533e8SHuawei Xie /* heads for the main used and free linked lists for the data path. */
308d19533e8SHuawei Xie static struct virtio_net_data_ll *ll_root_used = NULL;
309d19533e8SHuawei Xie static struct virtio_net_data_ll *ll_root_free = NULL;
310d19533e8SHuawei Xie 
311d19533e8SHuawei Xie /* Array of data core structures containing information on individual core linked lists. */
312d19533e8SHuawei Xie static struct lcore_info lcore_info[RTE_MAX_LCORE];
313d19533e8SHuawei Xie 
314d19533e8SHuawei Xie /* Used for queueing bursts of TX packets. */
315d19533e8SHuawei Xie struct mbuf_table {
316d19533e8SHuawei Xie 	unsigned len;
317d19533e8SHuawei Xie 	unsigned txq_id;
318d19533e8SHuawei Xie 	struct rte_mbuf *m_table[MAX_PKT_BURST];
319d19533e8SHuawei Xie };
320d19533e8SHuawei Xie 
321d19533e8SHuawei Xie /* TX queue for each data core. */
322d19533e8SHuawei Xie struct mbuf_table lcore_tx_queue[RTE_MAX_LCORE];
323d19533e8SHuawei Xie 
324d19533e8SHuawei Xie /* TX queue fori each virtio device for zero copy. */
325d19533e8SHuawei Xie struct mbuf_table tx_queue_zcp[MAX_QUEUES];
326d19533e8SHuawei Xie 
327d19533e8SHuawei Xie /* Vlan header struct used to insert vlan tags on TX. */
328d19533e8SHuawei Xie struct vlan_ethhdr {
329d19533e8SHuawei Xie 	unsigned char   h_dest[ETH_ALEN];
330d19533e8SHuawei Xie 	unsigned char   h_source[ETH_ALEN];
331d19533e8SHuawei Xie 	__be16          h_vlan_proto;
332d19533e8SHuawei Xie 	__be16          h_vlan_TCI;
333d19533e8SHuawei Xie 	__be16          h_vlan_encapsulated_proto;
334d19533e8SHuawei Xie };
335d19533e8SHuawei Xie 
336d19533e8SHuawei Xie /* IPv4 Header */
337d19533e8SHuawei Xie struct ipv4_hdr {
338d19533e8SHuawei Xie 	uint8_t  version_ihl;		/**< version and header length */
339d19533e8SHuawei Xie 	uint8_t  type_of_service;	/**< type of service */
340d19533e8SHuawei Xie 	uint16_t total_length;		/**< length of packet */
341d19533e8SHuawei Xie 	uint16_t packet_id;		/**< packet ID */
342d19533e8SHuawei Xie 	uint16_t fragment_offset;	/**< fragmentation offset */
343d19533e8SHuawei Xie 	uint8_t  time_to_live;		/**< time to live */
344d19533e8SHuawei Xie 	uint8_t  next_proto_id;		/**< protocol ID */
345d19533e8SHuawei Xie 	uint16_t hdr_checksum;		/**< header checksum */
346d19533e8SHuawei Xie 	uint32_t src_addr;		/**< source address */
347d19533e8SHuawei Xie 	uint32_t dst_addr;		/**< destination address */
348d19533e8SHuawei Xie } __attribute__((__packed__));
349d19533e8SHuawei Xie 
350d19533e8SHuawei Xie /* Header lengths. */
351d19533e8SHuawei Xie #define VLAN_HLEN       4
352d19533e8SHuawei Xie #define VLAN_ETH_HLEN   18
353d19533e8SHuawei Xie 
354d19533e8SHuawei Xie /* Per-device statistics struct */
355d19533e8SHuawei Xie struct device_statistics {
356d19533e8SHuawei Xie 	uint64_t tx_total;
357d19533e8SHuawei Xie 	rte_atomic64_t rx_total_atomic;
358d19533e8SHuawei Xie 	uint64_t rx_total;
359d19533e8SHuawei Xie 	uint64_t tx;
360d19533e8SHuawei Xie 	rte_atomic64_t rx_atomic;
361d19533e8SHuawei Xie 	uint64_t rx;
362d19533e8SHuawei Xie } __rte_cache_aligned;
363d19533e8SHuawei Xie struct device_statistics dev_statistics[MAX_DEVICES];
364d19533e8SHuawei Xie 
365d19533e8SHuawei Xie /*
366d19533e8SHuawei Xie  * Builds up the correct configuration for VMDQ VLAN pool map
367d19533e8SHuawei Xie  * according to the pool & queue limits.
368d19533e8SHuawei Xie  */
369d19533e8SHuawei Xie static inline int
370d19533e8SHuawei Xie get_eth_conf(struct rte_eth_conf *eth_conf, uint32_t num_devices)
371d19533e8SHuawei Xie {
372d19533e8SHuawei Xie 	struct rte_eth_vmdq_rx_conf conf;
373d19533e8SHuawei Xie 	unsigned i;
374d19533e8SHuawei Xie 
375d19533e8SHuawei Xie 	memset(&conf, 0, sizeof(conf));
376d19533e8SHuawei Xie 	conf.nb_queue_pools = (enum rte_eth_nb_pools)num_devices;
377d19533e8SHuawei Xie 	conf.nb_pool_maps = num_devices;
378d19533e8SHuawei Xie 	conf.enable_loop_back =
379d19533e8SHuawei Xie 		vmdq_conf_default.rx_adv_conf.vmdq_rx_conf.enable_loop_back;
380d19533e8SHuawei Xie 
381d19533e8SHuawei Xie 	for (i = 0; i < conf.nb_pool_maps; i++) {
382d19533e8SHuawei Xie 		conf.pool_map[i].vlan_id = vlan_tags[ i ];
383d19533e8SHuawei Xie 		conf.pool_map[i].pools = (1UL << i);
384d19533e8SHuawei Xie 	}
385d19533e8SHuawei Xie 
386d19533e8SHuawei Xie 	(void)(rte_memcpy(eth_conf, &vmdq_conf_default, sizeof(*eth_conf)));
387d19533e8SHuawei Xie 	(void)(rte_memcpy(&eth_conf->rx_adv_conf.vmdq_rx_conf, &conf,
388d19533e8SHuawei Xie 		   sizeof(eth_conf->rx_adv_conf.vmdq_rx_conf)));
389d19533e8SHuawei Xie 	return 0;
390d19533e8SHuawei Xie }
391d19533e8SHuawei Xie 
392d19533e8SHuawei Xie /*
393d19533e8SHuawei Xie  * Validate the device number according to the max pool number gotten form
394d19533e8SHuawei Xie  * dev_info. If the device number is invalid, give the error message and
395d19533e8SHuawei Xie  * return -1. Each device must have its own pool.
396d19533e8SHuawei Xie  */
397d19533e8SHuawei Xie static inline int
398d19533e8SHuawei Xie validate_num_devices(uint32_t max_nb_devices)
399d19533e8SHuawei Xie {
400d19533e8SHuawei Xie 	if (num_devices > max_nb_devices) {
401d19533e8SHuawei Xie 		RTE_LOG(ERR, VHOST_PORT, "invalid number of devices\n");
402d19533e8SHuawei Xie 		return -1;
403d19533e8SHuawei Xie 	}
404d19533e8SHuawei Xie 	return 0;
405d19533e8SHuawei Xie }
406d19533e8SHuawei Xie 
407d19533e8SHuawei Xie /*
408d19533e8SHuawei Xie  * Initialises a given port using global settings and with the rx buffers
409d19533e8SHuawei Xie  * coming from the mbuf_pool passed as parameter
410d19533e8SHuawei Xie  */
411d19533e8SHuawei Xie static inline int
412d19533e8SHuawei Xie port_init(uint8_t port)
413d19533e8SHuawei Xie {
414d19533e8SHuawei Xie 	struct rte_eth_dev_info dev_info;
415d19533e8SHuawei Xie 	struct rte_eth_conf port_conf;
416d19533e8SHuawei Xie 	uint16_t rx_rings, tx_rings;
417d19533e8SHuawei Xie 	uint16_t rx_ring_size, tx_ring_size;
418d19533e8SHuawei Xie 	int retval;
419d19533e8SHuawei Xie 	uint16_t q;
420d19533e8SHuawei Xie 
421d19533e8SHuawei Xie 	/* The max pool number from dev_info will be used to validate the pool number specified in cmd line */
422d19533e8SHuawei Xie 	rte_eth_dev_info_get (port, &dev_info);
423d19533e8SHuawei Xie 
424d19533e8SHuawei Xie 	/*configure the number of supported virtio devices based on VMDQ limits */
425d19533e8SHuawei Xie 	num_devices = dev_info.max_vmdq_pools;
426d19533e8SHuawei Xie 	num_queues = dev_info.max_rx_queues;
427d19533e8SHuawei Xie 
428d19533e8SHuawei Xie 	if (zero_copy) {
429d19533e8SHuawei Xie 		rx_ring_size = num_rx_descriptor;
430d19533e8SHuawei Xie 		tx_ring_size = num_tx_descriptor;
431d19533e8SHuawei Xie 		tx_rings = dev_info.max_tx_queues;
432d19533e8SHuawei Xie 	} else {
433d19533e8SHuawei Xie 		rx_ring_size = RTE_TEST_RX_DESC_DEFAULT;
434d19533e8SHuawei Xie 		tx_ring_size = RTE_TEST_TX_DESC_DEFAULT;
435d19533e8SHuawei Xie 		tx_rings = (uint16_t)rte_lcore_count();
436d19533e8SHuawei Xie 	}
437d19533e8SHuawei Xie 
438d19533e8SHuawei Xie 	retval = validate_num_devices(MAX_DEVICES);
439d19533e8SHuawei Xie 	if (retval < 0)
440d19533e8SHuawei Xie 		return retval;
441d19533e8SHuawei Xie 
442d19533e8SHuawei Xie 	/* Get port configuration. */
443d19533e8SHuawei Xie 	retval = get_eth_conf(&port_conf, num_devices);
444d19533e8SHuawei Xie 	if (retval < 0)
445d19533e8SHuawei Xie 		return retval;
446d19533e8SHuawei Xie 
447d19533e8SHuawei Xie 	if (port >= rte_eth_dev_count()) return -1;
448d19533e8SHuawei Xie 
449d19533e8SHuawei Xie 	rx_rings = (uint16_t)num_queues,
450d19533e8SHuawei Xie 	/* Configure ethernet device. */
451d19533e8SHuawei Xie 	retval = rte_eth_dev_configure(port, rx_rings, tx_rings, &port_conf);
452d19533e8SHuawei Xie 	if (retval != 0)
453d19533e8SHuawei Xie 		return retval;
454d19533e8SHuawei Xie 
455d19533e8SHuawei Xie 	/* Setup the queues. */
456d19533e8SHuawei Xie 	for (q = 0; q < rx_rings; q ++) {
457d19533e8SHuawei Xie 		retval = rte_eth_rx_queue_setup(port, q, rx_ring_size,
458d19533e8SHuawei Xie 						rte_eth_dev_socket_id(port), &rx_conf_default,
459d19533e8SHuawei Xie 						vpool_array[q].pool);
460d19533e8SHuawei Xie 		if (retval < 0)
461d19533e8SHuawei Xie 			return retval;
462d19533e8SHuawei Xie 	}
463d19533e8SHuawei Xie 	for (q = 0; q < tx_rings; q ++) {
464d19533e8SHuawei Xie 		retval = rte_eth_tx_queue_setup(port, q, tx_ring_size,
465d19533e8SHuawei Xie 						rte_eth_dev_socket_id(port), &tx_conf_default);
466d19533e8SHuawei Xie 		if (retval < 0)
467d19533e8SHuawei Xie 			return retval;
468d19533e8SHuawei Xie 	}
469d19533e8SHuawei Xie 
470d19533e8SHuawei Xie 	/* Start the device. */
471d19533e8SHuawei Xie 	retval  = rte_eth_dev_start(port);
472d19533e8SHuawei Xie 	if (retval < 0) {
473d19533e8SHuawei Xie 		RTE_LOG(ERR, VHOST_DATA, "Failed to start the device.\n");
474d19533e8SHuawei Xie 		return retval;
475d19533e8SHuawei Xie 	}
476d19533e8SHuawei Xie 
477d19533e8SHuawei Xie 	rte_eth_macaddr_get(port, &vmdq_ports_eth_addr[port]);
478d19533e8SHuawei Xie 	RTE_LOG(INFO, VHOST_PORT, "Max virtio devices supported: %u\n", num_devices);
479d19533e8SHuawei Xie 	RTE_LOG(INFO, VHOST_PORT, "Port %u MAC: %02"PRIx8" %02"PRIx8" %02"PRIx8
480d19533e8SHuawei Xie 			" %02"PRIx8" %02"PRIx8" %02"PRIx8"\n",
481d19533e8SHuawei Xie 			(unsigned)port,
482d19533e8SHuawei Xie 			vmdq_ports_eth_addr[port].addr_bytes[0],
483d19533e8SHuawei Xie 			vmdq_ports_eth_addr[port].addr_bytes[1],
484d19533e8SHuawei Xie 			vmdq_ports_eth_addr[port].addr_bytes[2],
485d19533e8SHuawei Xie 			vmdq_ports_eth_addr[port].addr_bytes[3],
486d19533e8SHuawei Xie 			vmdq_ports_eth_addr[port].addr_bytes[4],
487d19533e8SHuawei Xie 			vmdq_ports_eth_addr[port].addr_bytes[5]);
488d19533e8SHuawei Xie 
489d19533e8SHuawei Xie 	return 0;
490d19533e8SHuawei Xie }
491d19533e8SHuawei Xie 
492d19533e8SHuawei Xie /*
493d19533e8SHuawei Xie  * Set character device basename.
494d19533e8SHuawei Xie  */
495d19533e8SHuawei Xie static int
496d19533e8SHuawei Xie us_vhost_parse_basename(const char *q_arg)
497d19533e8SHuawei Xie {
498d19533e8SHuawei Xie 	/* parse number string */
499d19533e8SHuawei Xie 
500d19533e8SHuawei Xie 	if (strnlen(q_arg, MAX_BASENAME_SZ) > MAX_BASENAME_SZ)
501d19533e8SHuawei Xie 		return -1;
502d19533e8SHuawei Xie 	else
503d19533e8SHuawei Xie 		snprintf((char*)&dev_basename, MAX_BASENAME_SZ, "%s", q_arg);
504d19533e8SHuawei Xie 
505d19533e8SHuawei Xie 	return 0;
506d19533e8SHuawei Xie }
507d19533e8SHuawei Xie 
508d19533e8SHuawei Xie /*
509d19533e8SHuawei Xie  * Parse the portmask provided at run time.
510d19533e8SHuawei Xie  */
511d19533e8SHuawei Xie static int
512d19533e8SHuawei Xie parse_portmask(const char *portmask)
513d19533e8SHuawei Xie {
514d19533e8SHuawei Xie 	char *end = NULL;
515d19533e8SHuawei Xie 	unsigned long pm;
516d19533e8SHuawei Xie 
517d19533e8SHuawei Xie 	errno = 0;
518d19533e8SHuawei Xie 
519d19533e8SHuawei Xie 	/* parse hexadecimal string */
520d19533e8SHuawei Xie 	pm = strtoul(portmask, &end, 16);
521d19533e8SHuawei Xie 	if ((portmask[0] == '\0') || (end == NULL) || (*end != '\0') || (errno != 0))
522d19533e8SHuawei Xie 		return -1;
523d19533e8SHuawei Xie 
524d19533e8SHuawei Xie 	if (pm == 0)
525d19533e8SHuawei Xie 		return -1;
526d19533e8SHuawei Xie 
527d19533e8SHuawei Xie 	return pm;
528d19533e8SHuawei Xie 
529d19533e8SHuawei Xie }
530d19533e8SHuawei Xie 
531d19533e8SHuawei Xie /*
532d19533e8SHuawei Xie  * Parse num options at run time.
533d19533e8SHuawei Xie  */
534d19533e8SHuawei Xie static int
535d19533e8SHuawei Xie parse_num_opt(const char *q_arg, uint32_t max_valid_value)
536d19533e8SHuawei Xie {
537d19533e8SHuawei Xie 	char *end = NULL;
538d19533e8SHuawei Xie 	unsigned long num;
539d19533e8SHuawei Xie 
540d19533e8SHuawei Xie 	errno = 0;
541d19533e8SHuawei Xie 
542d19533e8SHuawei Xie 	/* parse unsigned int string */
543d19533e8SHuawei Xie 	num = strtoul(q_arg, &end, 10);
544d19533e8SHuawei Xie 	if ((q_arg[0] == '\0') || (end == NULL) || (*end != '\0') || (errno != 0))
545d19533e8SHuawei Xie 		return -1;
546d19533e8SHuawei Xie 
547d19533e8SHuawei Xie 	if (num > max_valid_value)
548d19533e8SHuawei Xie 		return -1;
549d19533e8SHuawei Xie 
550d19533e8SHuawei Xie 	return num;
551d19533e8SHuawei Xie 
552d19533e8SHuawei Xie }
553d19533e8SHuawei Xie 
554d19533e8SHuawei Xie /*
555d19533e8SHuawei Xie  * Display usage
556d19533e8SHuawei Xie  */
557d19533e8SHuawei Xie static void
558d19533e8SHuawei Xie us_vhost_usage(const char *prgname)
559d19533e8SHuawei Xie {
560d19533e8SHuawei Xie 	RTE_LOG(INFO, VHOST_CONFIG, "%s [EAL options] -- -p PORTMASK\n"
561d19533e8SHuawei Xie 	"		--vm2vm [0|1|2]\n"
562d19533e8SHuawei Xie 	"		--rx_retry [0|1] --mergeable [0|1] --stats [0-N]\n"
563d19533e8SHuawei Xie 	"		--dev-basename <name> --dev-index [0-N]\n"
564d19533e8SHuawei Xie 	"		--nb-devices ND\n"
565d19533e8SHuawei Xie 	"		-p PORTMASK: Set mask for ports to be used by application\n"
566d19533e8SHuawei Xie 	"		--vm2vm [0|1|2]: disable/software(default)/hardware vm2vm comms\n"
567d19533e8SHuawei Xie 	"		--rx-retry [0|1]: disable/enable(default) retries on rx. Enable retry if destintation queue is full\n"
568d19533e8SHuawei Xie 	"		--rx-retry-delay [0-N]: timeout(in usecond) between retries on RX. This makes effect only if retries on rx enabled\n"
569d19533e8SHuawei Xie 	"		--rx-retry-num [0-N]: the number of retries on rx. This makes effect only if retries on rx enabled\n"
570d19533e8SHuawei Xie 	"		--mergeable [0|1]: disable(default)/enable RX mergeable buffers\n"
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 	"		--dev-index [0-N]: Defaults to zero if not used. Index is appended to basename.\n"
574d19533e8SHuawei Xie 	"		--zero-copy [0|1]: disable(default)/enable rx/tx "
575d19533e8SHuawei Xie 			"zero copy\n"
576d19533e8SHuawei Xie 	"		--rx-desc-num [0-N]: the number of descriptors on rx, "
577d19533e8SHuawei Xie 			"used only when zero copy is enabled.\n"
578d19533e8SHuawei Xie 	"		--tx-desc-num [0-N]: the number of descriptors on tx, "
579d19533e8SHuawei Xie 			"used only when zero copy is enabled.\n",
580d19533e8SHuawei Xie 	       prgname);
581d19533e8SHuawei Xie }
582d19533e8SHuawei Xie 
583d19533e8SHuawei Xie /*
584d19533e8SHuawei Xie  * Parse the arguments given in the command line of the application.
585d19533e8SHuawei Xie  */
586d19533e8SHuawei Xie static int
587d19533e8SHuawei Xie us_vhost_parse_args(int argc, char **argv)
588d19533e8SHuawei Xie {
589d19533e8SHuawei Xie 	int opt, ret;
590d19533e8SHuawei Xie 	int option_index;
591d19533e8SHuawei Xie 	unsigned i;
592d19533e8SHuawei Xie 	const char *prgname = argv[0];
593d19533e8SHuawei Xie 	static struct option long_option[] = {
594d19533e8SHuawei Xie 		{"vm2vm", required_argument, NULL, 0},
595d19533e8SHuawei Xie 		{"rx-retry", required_argument, NULL, 0},
596d19533e8SHuawei Xie 		{"rx-retry-delay", required_argument, NULL, 0},
597d19533e8SHuawei Xie 		{"rx-retry-num", required_argument, NULL, 0},
598d19533e8SHuawei Xie 		{"mergeable", required_argument, NULL, 0},
599d19533e8SHuawei Xie 		{"stats", required_argument, NULL, 0},
600d19533e8SHuawei Xie 		{"dev-basename", required_argument, NULL, 0},
601d19533e8SHuawei Xie 		{"dev-index", required_argument, NULL, 0},
602d19533e8SHuawei Xie 		{"zero-copy", required_argument, NULL, 0},
603d19533e8SHuawei Xie 		{"rx-desc-num", required_argument, NULL, 0},
604d19533e8SHuawei Xie 		{"tx-desc-num", required_argument, NULL, 0},
605d19533e8SHuawei Xie 		{NULL, 0, 0, 0},
606d19533e8SHuawei Xie 	};
607d19533e8SHuawei Xie 
608d19533e8SHuawei Xie 	/* Parse command line */
609d19533e8SHuawei Xie 	while ((opt = getopt_long(argc, argv, "p:",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 
621d19533e8SHuawei Xie 		case 0:
622d19533e8SHuawei Xie 			/* Enable/disable vm2vm comms. */
623d19533e8SHuawei Xie 			if (!strncmp(long_option[option_index].name, "vm2vm",
624d19533e8SHuawei Xie 				MAX_LONG_OPT_SZ)) {
625d19533e8SHuawei Xie 				ret = parse_num_opt(optarg, (VM2VM_LAST - 1));
626d19533e8SHuawei Xie 				if (ret == -1) {
627d19533e8SHuawei Xie 					RTE_LOG(INFO, VHOST_CONFIG,
628d19533e8SHuawei Xie 						"Invalid argument for "
629d19533e8SHuawei Xie 						"vm2vm [0|1|2]\n");
630d19533e8SHuawei Xie 					us_vhost_usage(prgname);
631d19533e8SHuawei Xie 					return -1;
632d19533e8SHuawei Xie 				} else {
633d19533e8SHuawei Xie 					vm2vm_mode = (vm2vm_type)ret;
634d19533e8SHuawei Xie 				}
635d19533e8SHuawei Xie 			}
636d19533e8SHuawei Xie 
637d19533e8SHuawei Xie 			/* Enable/disable retries on RX. */
638d19533e8SHuawei Xie 			if (!strncmp(long_option[option_index].name, "rx-retry", MAX_LONG_OPT_SZ)) {
639d19533e8SHuawei Xie 				ret = parse_num_opt(optarg, 1);
640d19533e8SHuawei Xie 				if (ret == -1) {
641d19533e8SHuawei Xie 					RTE_LOG(INFO, VHOST_CONFIG, "Invalid argument for rx-retry [0|1]\n");
642d19533e8SHuawei Xie 					us_vhost_usage(prgname);
643d19533e8SHuawei Xie 					return -1;
644d19533e8SHuawei Xie 				} else {
645d19533e8SHuawei Xie 					enable_retry = ret;
646d19533e8SHuawei Xie 				}
647d19533e8SHuawei Xie 			}
648d19533e8SHuawei Xie 
649d19533e8SHuawei Xie 			/* Specify the retries delay time (in useconds) on RX. */
650d19533e8SHuawei Xie 			if (!strncmp(long_option[option_index].name, "rx-retry-delay", MAX_LONG_OPT_SZ)) {
651d19533e8SHuawei Xie 				ret = parse_num_opt(optarg, INT32_MAX);
652d19533e8SHuawei Xie 				if (ret == -1) {
653d19533e8SHuawei Xie 					RTE_LOG(INFO, VHOST_CONFIG, "Invalid argument for rx-retry-delay [0-N]\n");
654d19533e8SHuawei Xie 					us_vhost_usage(prgname);
655d19533e8SHuawei Xie 					return -1;
656d19533e8SHuawei Xie 				} else {
657d19533e8SHuawei Xie 					burst_rx_delay_time = ret;
658d19533e8SHuawei Xie 				}
659d19533e8SHuawei Xie 			}
660d19533e8SHuawei Xie 
661d19533e8SHuawei Xie 			/* Specify the retries number on RX. */
662d19533e8SHuawei Xie 			if (!strncmp(long_option[option_index].name, "rx-retry-num", MAX_LONG_OPT_SZ)) {
663d19533e8SHuawei Xie 				ret = parse_num_opt(optarg, INT32_MAX);
664d19533e8SHuawei Xie 				if (ret == -1) {
665d19533e8SHuawei Xie 					RTE_LOG(INFO, VHOST_CONFIG, "Invalid argument for rx-retry-num [0-N]\n");
666d19533e8SHuawei Xie 					us_vhost_usage(prgname);
667d19533e8SHuawei Xie 					return -1;
668d19533e8SHuawei Xie 				} else {
669d19533e8SHuawei Xie 					burst_rx_retry_num = ret;
670d19533e8SHuawei Xie 				}
671d19533e8SHuawei Xie 			}
672d19533e8SHuawei Xie 
673d19533e8SHuawei Xie 			/* Enable/disable RX mergeable buffers. */
674d19533e8SHuawei Xie 			if (!strncmp(long_option[option_index].name, "mergeable", MAX_LONG_OPT_SZ)) {
675d19533e8SHuawei Xie 				ret = parse_num_opt(optarg, 1);
676d19533e8SHuawei Xie 				if (ret == -1) {
677d19533e8SHuawei Xie 					RTE_LOG(INFO, VHOST_CONFIG, "Invalid argument for mergeable [0|1]\n");
678d19533e8SHuawei Xie 					us_vhost_usage(prgname);
679d19533e8SHuawei Xie 					return -1;
680d19533e8SHuawei Xie 				} else {
681d19533e8SHuawei Xie 					if (ret) {
682d19533e8SHuawei Xie 						vmdq_conf_default.rxmode.jumbo_frame = 1;
683d19533e8SHuawei Xie 						vmdq_conf_default.rxmode.max_rx_pkt_len
684d19533e8SHuawei Xie 							= JUMBO_FRAME_MAX_SIZE;
685d19533e8SHuawei Xie 						VHOST_FEATURES = (1ULL << VIRTIO_NET_F_MRG_RXBUF);
686d19533e8SHuawei Xie 					}
687d19533e8SHuawei Xie 				}
688d19533e8SHuawei Xie 			}
689d19533e8SHuawei Xie 
690d19533e8SHuawei Xie 			/* Enable/disable stats. */
691d19533e8SHuawei Xie 			if (!strncmp(long_option[option_index].name, "stats", MAX_LONG_OPT_SZ)) {
692d19533e8SHuawei Xie 				ret = parse_num_opt(optarg, INT32_MAX);
693d19533e8SHuawei Xie 				if (ret == -1) {
694d19533e8SHuawei Xie 					RTE_LOG(INFO, VHOST_CONFIG, "Invalid argument for stats [0..N]\n");
695d19533e8SHuawei Xie 					us_vhost_usage(prgname);
696d19533e8SHuawei Xie 					return -1;
697d19533e8SHuawei Xie 				} else {
698d19533e8SHuawei Xie 					enable_stats = ret;
699d19533e8SHuawei Xie 				}
700d19533e8SHuawei Xie 			}
701d19533e8SHuawei Xie 
702d19533e8SHuawei Xie 			/* Set character device basename. */
703d19533e8SHuawei Xie 			if (!strncmp(long_option[option_index].name, "dev-basename", MAX_LONG_OPT_SZ)) {
704d19533e8SHuawei Xie 				if (us_vhost_parse_basename(optarg) == -1) {
705d19533e8SHuawei Xie 					RTE_LOG(INFO, VHOST_CONFIG, "Invalid argument for character device basename (Max %d characters)\n", MAX_BASENAME_SZ);
706d19533e8SHuawei Xie 					us_vhost_usage(prgname);
707d19533e8SHuawei Xie 					return -1;
708d19533e8SHuawei Xie 				}
709d19533e8SHuawei Xie 			}
710d19533e8SHuawei Xie 
711d19533e8SHuawei Xie 			/* Set character device index. */
712d19533e8SHuawei Xie 			if (!strncmp(long_option[option_index].name, "dev-index", MAX_LONG_OPT_SZ)) {
713d19533e8SHuawei Xie 				ret = parse_num_opt(optarg, INT32_MAX);
714d19533e8SHuawei Xie 				if (ret == -1) {
715d19533e8SHuawei Xie 					RTE_LOG(INFO, VHOST_CONFIG, "Invalid argument for character device index [0..N]\n");
716d19533e8SHuawei Xie 					us_vhost_usage(prgname);
717d19533e8SHuawei Xie 					return -1;
718d19533e8SHuawei Xie 				} else
719d19533e8SHuawei Xie 					dev_index = ret;
720d19533e8SHuawei Xie 			}
721d19533e8SHuawei Xie 
722d19533e8SHuawei Xie 			/* Enable/disable rx/tx zero copy. */
723d19533e8SHuawei Xie 			if (!strncmp(long_option[option_index].name,
724d19533e8SHuawei Xie 				"zero-copy", MAX_LONG_OPT_SZ)) {
725d19533e8SHuawei Xie 				ret = parse_num_opt(optarg, 1);
726d19533e8SHuawei Xie 				if (ret == -1) {
727d19533e8SHuawei Xie 					RTE_LOG(INFO, VHOST_CONFIG,
728d19533e8SHuawei Xie 						"Invalid argument"
729d19533e8SHuawei Xie 						" for zero-copy [0|1]\n");
730d19533e8SHuawei Xie 					us_vhost_usage(prgname);
731d19533e8SHuawei Xie 					return -1;
732d19533e8SHuawei Xie 				} else
733d19533e8SHuawei Xie 					zero_copy = ret;
734d19533e8SHuawei Xie 
735d19533e8SHuawei Xie 				if (zero_copy) {
736d19533e8SHuawei Xie #ifdef RTE_MBUF_REFCNT
737d19533e8SHuawei Xie 					RTE_LOG(ERR, VHOST_CONFIG, "Before running "
738d19533e8SHuawei Xie 					"zero copy vhost APP, please "
739d19533e8SHuawei Xie 					"disable RTE_MBUF_REFCNT\n"
740d19533e8SHuawei Xie 					"in config file and then rebuild DPDK "
741d19533e8SHuawei Xie 					"core lib!\n"
742d19533e8SHuawei Xie 					"Otherwise please disable zero copy "
743d19533e8SHuawei Xie 					"flag in command line!\n");
744d19533e8SHuawei Xie 					return -1;
745d19533e8SHuawei Xie #endif
746d19533e8SHuawei Xie 				}
747d19533e8SHuawei Xie 			}
748d19533e8SHuawei Xie 
749d19533e8SHuawei Xie 			/* Specify the descriptor number on RX. */
750d19533e8SHuawei Xie 			if (!strncmp(long_option[option_index].name,
751d19533e8SHuawei Xie 				"rx-desc-num", MAX_LONG_OPT_SZ)) {
752d19533e8SHuawei Xie 				ret = parse_num_opt(optarg, MAX_RING_DESC);
753d19533e8SHuawei Xie 				if ((ret == -1) || (!POWEROF2(ret))) {
754d19533e8SHuawei Xie 					RTE_LOG(INFO, VHOST_CONFIG,
755d19533e8SHuawei Xie 					"Invalid argument for rx-desc-num[0-N],"
756d19533e8SHuawei Xie 					"power of 2 required.\n");
757d19533e8SHuawei Xie 					us_vhost_usage(prgname);
758d19533e8SHuawei Xie 					return -1;
759d19533e8SHuawei Xie 				} else {
760d19533e8SHuawei Xie 					num_rx_descriptor = ret;
761d19533e8SHuawei Xie 				}
762d19533e8SHuawei Xie 			}
763d19533e8SHuawei Xie 
764d19533e8SHuawei Xie 			/* Specify the descriptor number on TX. */
765d19533e8SHuawei Xie 			if (!strncmp(long_option[option_index].name,
766d19533e8SHuawei Xie 				"tx-desc-num", MAX_LONG_OPT_SZ)) {
767d19533e8SHuawei Xie 				ret = parse_num_opt(optarg, MAX_RING_DESC);
768d19533e8SHuawei Xie 				if ((ret == -1) || (!POWEROF2(ret))) {
769d19533e8SHuawei Xie 					RTE_LOG(INFO, VHOST_CONFIG,
770d19533e8SHuawei Xie 					"Invalid argument for tx-desc-num [0-N],"
771d19533e8SHuawei Xie 					"power of 2 required.\n");
772d19533e8SHuawei Xie 					us_vhost_usage(prgname);
773d19533e8SHuawei Xie 					return -1;
774d19533e8SHuawei Xie 				} else {
775d19533e8SHuawei Xie 					num_tx_descriptor = ret;
776d19533e8SHuawei Xie 				}
777d19533e8SHuawei Xie 			}
778d19533e8SHuawei Xie 
779d19533e8SHuawei Xie 			break;
780d19533e8SHuawei Xie 
781d19533e8SHuawei Xie 			/* Invalid option - print options. */
782d19533e8SHuawei Xie 		default:
783d19533e8SHuawei Xie 			us_vhost_usage(prgname);
784d19533e8SHuawei Xie 			return -1;
785d19533e8SHuawei Xie 		}
786d19533e8SHuawei Xie 	}
787d19533e8SHuawei Xie 
788d19533e8SHuawei Xie 	for (i = 0; i < RTE_MAX_ETHPORTS; i++) {
789d19533e8SHuawei Xie 		if (enabled_port_mask & (1 << i))
790d19533e8SHuawei Xie 			ports[num_ports++] = (uint8_t)i;
791d19533e8SHuawei Xie 	}
792d19533e8SHuawei Xie 
793d19533e8SHuawei Xie 	if ((num_ports ==  0) || (num_ports > MAX_SUP_PORTS)) {
794d19533e8SHuawei Xie 		RTE_LOG(INFO, VHOST_PORT, "Current enabled port number is %u,"
795d19533e8SHuawei Xie 			"but only %u port can be enabled\n",num_ports, MAX_SUP_PORTS);
796d19533e8SHuawei Xie 		return -1;
797d19533e8SHuawei Xie 	}
798d19533e8SHuawei Xie 
799d19533e8SHuawei Xie 	if ((zero_copy == 1) && (vm2vm_mode == VM2VM_SOFTWARE)) {
800d19533e8SHuawei Xie 		RTE_LOG(INFO, VHOST_PORT,
801d19533e8SHuawei Xie 			"Vhost zero copy doesn't support software vm2vm,"
802d19533e8SHuawei Xie 			"please specify 'vm2vm 2' to use hardware vm2vm.\n");
803d19533e8SHuawei Xie 		return -1;
804d19533e8SHuawei Xie 	}
805d19533e8SHuawei Xie 
806d19533e8SHuawei Xie 	if ((zero_copy == 1) && (vmdq_conf_default.rxmode.jumbo_frame == 1)) {
807d19533e8SHuawei Xie 		RTE_LOG(INFO, VHOST_PORT,
808d19533e8SHuawei Xie 			"Vhost zero copy doesn't support jumbo frame,"
809d19533e8SHuawei Xie 			"please specify '--mergeable 0' to disable the "
810d19533e8SHuawei Xie 			"mergeable feature.\n");
811d19533e8SHuawei Xie 		return -1;
812d19533e8SHuawei Xie 	}
813d19533e8SHuawei Xie 
814d19533e8SHuawei Xie 	return 0;
815d19533e8SHuawei Xie }
816d19533e8SHuawei Xie 
817d19533e8SHuawei Xie /*
818d19533e8SHuawei Xie  * Update the global var NUM_PORTS and array PORTS according to system ports number
819d19533e8SHuawei Xie  * and return valid ports number
820d19533e8SHuawei Xie  */
821d19533e8SHuawei Xie static unsigned check_ports_num(unsigned nb_ports)
822d19533e8SHuawei Xie {
823d19533e8SHuawei Xie 	unsigned valid_num_ports = num_ports;
824d19533e8SHuawei Xie 	unsigned portid;
825d19533e8SHuawei Xie 
826d19533e8SHuawei Xie 	if (num_ports > nb_ports) {
827d19533e8SHuawei Xie 		RTE_LOG(INFO, VHOST_PORT, "\nSpecified port number(%u) exceeds total system port number(%u)\n",
828d19533e8SHuawei Xie 			num_ports, nb_ports);
829d19533e8SHuawei Xie 		num_ports = nb_ports;
830d19533e8SHuawei Xie 	}
831d19533e8SHuawei Xie 
832d19533e8SHuawei Xie 	for (portid = 0; portid < num_ports; portid ++) {
833d19533e8SHuawei Xie 		if (ports[portid] >= nb_ports) {
834d19533e8SHuawei Xie 			RTE_LOG(INFO, VHOST_PORT, "\nSpecified port ID(%u) exceeds max system port ID(%u)\n",
835d19533e8SHuawei Xie 				ports[portid], (nb_ports - 1));
836d19533e8SHuawei Xie 			ports[portid] = INVALID_PORT_ID;
837d19533e8SHuawei Xie 			valid_num_ports--;
838d19533e8SHuawei Xie 		}
839d19533e8SHuawei Xie 	}
840d19533e8SHuawei Xie 	return valid_num_ports;
841d19533e8SHuawei Xie }
842d19533e8SHuawei Xie 
843d19533e8SHuawei Xie /*
844d19533e8SHuawei Xie  * Macro to print out packet contents. Wrapped in debug define so that the
845d19533e8SHuawei Xie  * data path is not effected when debug is disabled.
846d19533e8SHuawei Xie  */
847d19533e8SHuawei Xie #ifdef DEBUG
848d19533e8SHuawei Xie #define PRINT_PACKET(device, addr, size, header) do {																\
849d19533e8SHuawei Xie 	char *pkt_addr = (char*)(addr);																					\
850d19533e8SHuawei Xie 	unsigned int index;																								\
851d19533e8SHuawei Xie 	char packet[MAX_PRINT_BUFF];																					\
852d19533e8SHuawei Xie 																													\
853d19533e8SHuawei Xie 	if ((header))																									\
854d19533e8SHuawei Xie 		snprintf(packet, MAX_PRINT_BUFF, "(%"PRIu64") Header size %d: ", (device->device_fh), (size));				\
855d19533e8SHuawei Xie 	else																											\
856d19533e8SHuawei Xie 		snprintf(packet, MAX_PRINT_BUFF, "(%"PRIu64") Packet size %d: ", (device->device_fh), (size));				\
857d19533e8SHuawei Xie 	for (index = 0; index < (size); index++) {																		\
858d19533e8SHuawei Xie 		snprintf(packet + strnlen(packet, MAX_PRINT_BUFF), MAX_PRINT_BUFF - strnlen(packet, MAX_PRINT_BUFF),	\
859d19533e8SHuawei Xie 			"%02hhx ", pkt_addr[index]);																			\
860d19533e8SHuawei Xie 	}																												\
861d19533e8SHuawei Xie 	snprintf(packet + strnlen(packet, MAX_PRINT_BUFF), MAX_PRINT_BUFF - strnlen(packet, MAX_PRINT_BUFF), "\n");	\
862d19533e8SHuawei Xie 																													\
863d19533e8SHuawei Xie 	LOG_DEBUG(VHOST_DATA, "%s", packet);																					\
864d19533e8SHuawei Xie } while(0)
865d19533e8SHuawei Xie #else
866d19533e8SHuawei Xie #define PRINT_PACKET(device, addr, size, header) do{} while(0)
867d19533e8SHuawei Xie #endif
868d19533e8SHuawei Xie 
869d19533e8SHuawei Xie /*
870d19533e8SHuawei Xie  * Function to convert guest physical addresses to vhost physical addresses.
871d19533e8SHuawei Xie  * This is used to convert virtio buffer addresses.
872d19533e8SHuawei Xie  */
873d19533e8SHuawei Xie static inline uint64_t __attribute__((always_inline))
874*e571e6b4SHuawei Xie gpa_to_hpa(struct vhost_dev  *vdev, uint64_t guest_pa,
875d19533e8SHuawei Xie 	uint32_t buf_len, hpa_type *addr_type)
876d19533e8SHuawei Xie {
877d19533e8SHuawei Xie 	struct virtio_memory_regions_hpa *region;
878d19533e8SHuawei Xie 	uint32_t regionidx;
879d19533e8SHuawei Xie 	uint64_t vhost_pa = 0;
880d19533e8SHuawei Xie 
881d19533e8SHuawei Xie 	*addr_type = PHYS_ADDR_INVALID;
882d19533e8SHuawei Xie 
883*e571e6b4SHuawei Xie 	for (regionidx = 0; regionidx < vdev->nregions_hpa; regionidx++) {
884*e571e6b4SHuawei Xie 		region = &vdev->regions_hpa[regionidx];
885d19533e8SHuawei Xie 		if ((guest_pa >= region->guest_phys_address) &&
886d19533e8SHuawei Xie 			(guest_pa <= region->guest_phys_address_end)) {
887d19533e8SHuawei Xie 			vhost_pa = region->host_phys_addr_offset + guest_pa;
888d19533e8SHuawei Xie 			if (likely((guest_pa + buf_len - 1)
889d19533e8SHuawei Xie 				<= region->guest_phys_address_end))
890d19533e8SHuawei Xie 				*addr_type = PHYS_ADDR_CONTINUOUS;
891d19533e8SHuawei Xie 			else
892d19533e8SHuawei Xie 				*addr_type = PHYS_ADDR_CROSS_SUBREG;
893d19533e8SHuawei Xie 			break;
894d19533e8SHuawei Xie 		}
895d19533e8SHuawei Xie 	}
896d19533e8SHuawei Xie 
897d19533e8SHuawei Xie 	LOG_DEBUG(VHOST_DATA, "(%"PRIu64") GPA %p| HPA %p\n",
898*e571e6b4SHuawei Xie 		vdev->dev->device_fh, (void *)(uintptr_t)guest_pa,
899d19533e8SHuawei Xie 		(void *)(uintptr_t)vhost_pa);
900d19533e8SHuawei Xie 
901d19533e8SHuawei Xie 	return vhost_pa;
902d19533e8SHuawei Xie }
903d19533e8SHuawei Xie 
904d19533e8SHuawei Xie /*
905d19533e8SHuawei Xie  * Compares a packet destination MAC address to a device MAC address.
906d19533e8SHuawei Xie  */
907d19533e8SHuawei Xie static inline int __attribute__((always_inline))
908d19533e8SHuawei Xie ether_addr_cmp(struct ether_addr *ea, struct ether_addr *eb)
909d19533e8SHuawei Xie {
910d19533e8SHuawei Xie 	return (((*(uint64_t *)ea ^ *(uint64_t *)eb) & MAC_ADDR_CMP) == 0);
911d19533e8SHuawei Xie }
912d19533e8SHuawei Xie 
913d19533e8SHuawei Xie /*
914d19533e8SHuawei Xie  * This function learns the MAC address of the device and registers this along with a
915d19533e8SHuawei Xie  * vlan tag to a VMDQ.
916d19533e8SHuawei Xie  */
917d19533e8SHuawei Xie static int
918*e571e6b4SHuawei Xie link_vmdq(struct vhost_dev *vdev, struct rte_mbuf *m)
919d19533e8SHuawei Xie {
920d19533e8SHuawei Xie 	struct ether_hdr *pkt_hdr;
921d19533e8SHuawei Xie 	struct virtio_net_data_ll *dev_ll;
922*e571e6b4SHuawei Xie 	struct virtio_net *dev = vdev->dev;
923d19533e8SHuawei Xie 	int i, ret;
924d19533e8SHuawei Xie 
925d19533e8SHuawei Xie 	/* Learn MAC address of guest device from packet */
926d19533e8SHuawei Xie 	pkt_hdr = rte_pktmbuf_mtod(m, struct ether_hdr *);
927d19533e8SHuawei Xie 
928d19533e8SHuawei Xie 	dev_ll = ll_root_used;
929d19533e8SHuawei Xie 
930d19533e8SHuawei Xie 	while (dev_ll != NULL) {
931*e571e6b4SHuawei Xie 		if (ether_addr_cmp(&(pkt_hdr->s_addr), &dev_ll->vdev->mac_address)) {
932d19533e8SHuawei 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);
933d19533e8SHuawei Xie 			return -1;
934d19533e8SHuawei Xie 		}
935d19533e8SHuawei Xie 		dev_ll = dev_ll->next;
936d19533e8SHuawei Xie 	}
937d19533e8SHuawei Xie 
938d19533e8SHuawei Xie 	for (i = 0; i < ETHER_ADDR_LEN; i++)
939*e571e6b4SHuawei Xie 		vdev->mac_address.addr_bytes[i] = pkt_hdr->s_addr.addr_bytes[i];
940d19533e8SHuawei Xie 
941d19533e8SHuawei Xie 	/* vlan_tag currently uses the device_id. */
942*e571e6b4SHuawei Xie 	vdev->vlan_tag = vlan_tags[dev->device_fh];
943d19533e8SHuawei Xie 
944d19533e8SHuawei Xie 	/* Print out VMDQ registration info. */
945d19533e8SHuawei Xie 	RTE_LOG(INFO, VHOST_DATA, "(%"PRIu64") MAC_ADDRESS %02x:%02x:%02x:%02x:%02x:%02x and VLAN_TAG %d registered\n",
946d19533e8SHuawei Xie 		dev->device_fh,
947*e571e6b4SHuawei Xie 		vdev->mac_address.addr_bytes[0], vdev->mac_address.addr_bytes[1],
948*e571e6b4SHuawei Xie 		vdev->mac_address.addr_bytes[2], vdev->mac_address.addr_bytes[3],
949*e571e6b4SHuawei Xie 		vdev->mac_address.addr_bytes[4], vdev->mac_address.addr_bytes[5],
950*e571e6b4SHuawei Xie 		vdev->vlan_tag);
951d19533e8SHuawei Xie 
952d19533e8SHuawei Xie 	/* Register the MAC address. */
953*e571e6b4SHuawei Xie 	ret = rte_eth_dev_mac_addr_add(ports[0], &vdev->mac_address, (uint32_t)dev->device_fh);
954d19533e8SHuawei Xie 	if (ret)
955d19533e8SHuawei Xie 		RTE_LOG(ERR, VHOST_DATA, "(%"PRIu64") Failed to add device MAC address to VMDQ\n",
956d19533e8SHuawei Xie 					dev->device_fh);
957d19533e8SHuawei Xie 
958d19533e8SHuawei Xie 	/* Enable stripping of the vlan tag as we handle routing. */
959*e571e6b4SHuawei Xie 	rte_eth_dev_set_vlan_strip_on_queue(ports[0], (uint16_t)vdev->vmdq_rx_q, 1);
960d19533e8SHuawei Xie 
961d19533e8SHuawei Xie 	/* Set device as ready for RX. */
962*e571e6b4SHuawei Xie 	vdev->ready = DEVICE_RX;
963d19533e8SHuawei Xie 
964d19533e8SHuawei Xie 	return 0;
965d19533e8SHuawei Xie }
966d19533e8SHuawei Xie 
967d19533e8SHuawei Xie /*
968d19533e8SHuawei Xie  * Removes MAC address and vlan tag from VMDQ. Ensures that nothing is adding buffers to the RX
969d19533e8SHuawei Xie  * queue before disabling RX on the device.
970d19533e8SHuawei Xie  */
971d19533e8SHuawei Xie static inline void
972*e571e6b4SHuawei Xie unlink_vmdq(struct vhost_dev *vdev)
973d19533e8SHuawei Xie {
974d19533e8SHuawei Xie 	unsigned i = 0;
975d19533e8SHuawei Xie 	unsigned rx_count;
976d19533e8SHuawei Xie 	struct rte_mbuf *pkts_burst[MAX_PKT_BURST];
977d19533e8SHuawei Xie 
978*e571e6b4SHuawei Xie 	if (vdev->ready == DEVICE_RX) {
979d19533e8SHuawei Xie 		/*clear MAC and VLAN settings*/
980*e571e6b4SHuawei Xie 		rte_eth_dev_mac_addr_remove(ports[0], &vdev->mac_address);
981d19533e8SHuawei Xie 		for (i = 0; i < 6; i++)
982*e571e6b4SHuawei Xie 			vdev->mac_address.addr_bytes[i] = 0;
983d19533e8SHuawei Xie 
984*e571e6b4SHuawei Xie 		vdev->vlan_tag = 0;
985d19533e8SHuawei Xie 
986d19533e8SHuawei Xie 		/*Clear out the receive buffers*/
987d19533e8SHuawei Xie 		rx_count = rte_eth_rx_burst(ports[0],
988*e571e6b4SHuawei Xie 					(uint16_t)vdev->vmdq_rx_q, pkts_burst, MAX_PKT_BURST);
989d19533e8SHuawei Xie 
990d19533e8SHuawei Xie 		while (rx_count) {
991d19533e8SHuawei Xie 			for (i = 0; i < rx_count; i++)
992d19533e8SHuawei Xie 				rte_pktmbuf_free(pkts_burst[i]);
993d19533e8SHuawei Xie 
994d19533e8SHuawei Xie 			rx_count = rte_eth_rx_burst(ports[0],
995*e571e6b4SHuawei Xie 					(uint16_t)vdev->vmdq_rx_q, pkts_burst, MAX_PKT_BURST);
996d19533e8SHuawei Xie 		}
997d19533e8SHuawei Xie 
998*e571e6b4SHuawei Xie 		vdev->ready = DEVICE_MAC_LEARNING;
999d19533e8SHuawei Xie 	}
1000d19533e8SHuawei Xie }
1001d19533e8SHuawei Xie 
1002d19533e8SHuawei Xie /*
1003d19533e8SHuawei Xie  * Check if the packet destination MAC address is for a local device. If so then put
1004d19533e8SHuawei Xie  * the packet on that devices RX queue. If not then return.
1005d19533e8SHuawei Xie  */
1006d19533e8SHuawei Xie static inline unsigned __attribute__((always_inline))
1007*e571e6b4SHuawei Xie virtio_tx_local(struct vhost_dev *vdev, struct rte_mbuf *m)
1008d19533e8SHuawei Xie {
1009d19533e8SHuawei Xie 	struct virtio_net_data_ll *dev_ll;
1010d19533e8SHuawei Xie 	struct ether_hdr *pkt_hdr;
1011d19533e8SHuawei Xie 	uint64_t ret = 0;
1012*e571e6b4SHuawei Xie 	struct virtio_net *dev = vdev->dev;
1013*e571e6b4SHuawei Xie 	struct virtio_net *tdev; /* destination virito device */
1014d19533e8SHuawei Xie 
1015d19533e8SHuawei Xie 	pkt_hdr = rte_pktmbuf_mtod(m, struct ether_hdr *);
1016d19533e8SHuawei Xie 
1017d19533e8SHuawei Xie 	/*get the used devices list*/
1018d19533e8SHuawei Xie 	dev_ll = ll_root_used;
1019d19533e8SHuawei Xie 
1020d19533e8SHuawei Xie 	while (dev_ll != NULL) {
1021*e571e6b4SHuawei Xie 		if ((dev_ll->vdev->ready == DEVICE_RX) && ether_addr_cmp(&(pkt_hdr->d_addr),
1022*e571e6b4SHuawei Xie 				          &dev_ll->vdev->mac_address)) {
1023d19533e8SHuawei Xie 
1024d19533e8SHuawei Xie 			/* Drop the packet if the TX packet is destined for the TX device. */
1025*e571e6b4SHuawei Xie 			if (dev_ll->vdev->dev->device_fh == dev->device_fh) {
1026d19533e8SHuawei Xie 				LOG_DEBUG(VHOST_DATA, "(%"PRIu64") TX: Source and destination MAC addresses are the same. Dropping packet.\n",
1027*e571e6b4SHuawei Xie 							dev->device_fh);
1028d19533e8SHuawei Xie 				return 0;
1029d19533e8SHuawei Xie 			}
1030*e571e6b4SHuawei Xie 			tdev = dev_ll->vdev->dev;
1031d19533e8SHuawei Xie 
1032d19533e8SHuawei Xie 
1033*e571e6b4SHuawei Xie 			LOG_DEBUG(VHOST_DATA, "(%"PRIu64") TX: MAC address is local\n", tdev->device_fh);
1034d19533e8SHuawei Xie 
1035*e571e6b4SHuawei Xie 			if (dev_ll->vdev->remove) {
1036d19533e8SHuawei Xie 				/*drop the packet if the device is marked for removal*/
1037*e571e6b4SHuawei Xie 				LOG_DEBUG(VHOST_DATA, "(%"PRIu64") Device is marked for removal\n", tdev->device_fh);
1038d19533e8SHuawei Xie 			} else {
1039d19533e8SHuawei Xie 				uint32_t mergeable =
1040d19533e8SHuawei Xie 					dev_ll->dev->features &
1041d19533e8SHuawei Xie 					(1 << VIRTIO_NET_F_MRG_RXBUF);
1042d19533e8SHuawei Xie 
1043d19533e8SHuawei Xie 				/*send the packet to the local virtio device*/
1044d19533e8SHuawei Xie 				if (likely(mergeable == 0))
1045d19533e8SHuawei Xie 					ret = virtio_dev_rx(dev_ll->dev, &m, 1);
1046d19533e8SHuawei Xie 				else
1047d19533e8SHuawei Xie 					ret = virtio_dev_merge_rx(dev_ll->dev,
1048d19533e8SHuawei Xie 						&m, 1);
1049d19533e8SHuawei Xie 
1050d19533e8SHuawei Xie 				if (enable_stats) {
1051d19533e8SHuawei Xie 					rte_atomic64_add(
1052*e571e6b4SHuawei Xie 					&dev_statistics[tdev->device_fh].rx_total_atomic,
1053d19533e8SHuawei Xie 					1);
1054d19533e8SHuawei Xie 					rte_atomic64_add(
1055*e571e6b4SHuawei Xie 					&dev_statistics[tdev->device_fh].rx_atomic,
1056d19533e8SHuawei Xie 					ret);
1057*e571e6b4SHuawei Xie 					dev_statistics[tdev->device_fh].tx_total++;
1058*e571e6b4SHuawei Xie 					dev_statistics[tdev->device_fh].tx += ret;
1059d19533e8SHuawei Xie 				}
1060d19533e8SHuawei Xie 			}
1061d19533e8SHuawei Xie 
1062d19533e8SHuawei Xie 			return 0;
1063d19533e8SHuawei Xie 		}
1064d19533e8SHuawei Xie 		dev_ll = dev_ll->next;
1065d19533e8SHuawei Xie 	}
1066d19533e8SHuawei Xie 
1067d19533e8SHuawei Xie 	return -1;
1068d19533e8SHuawei Xie }
1069d19533e8SHuawei Xie 
1070d19533e8SHuawei Xie /*
1071d19533e8SHuawei Xie  * This function routes the TX packet to the correct interface. This may be a local device
1072d19533e8SHuawei Xie  * or the physical port.
1073d19533e8SHuawei Xie  */
1074d19533e8SHuawei Xie static inline void __attribute__((always_inline))
1075*e571e6b4SHuawei Xie virtio_tx_route(struct vhost_dev *vdev, struct rte_mbuf *m, struct rte_mempool *mbuf_pool, uint16_t vlan_tag)
1076d19533e8SHuawei Xie {
1077d19533e8SHuawei Xie 	struct mbuf_table *tx_q;
1078d19533e8SHuawei Xie 	struct vlan_ethhdr *vlan_hdr;
1079d19533e8SHuawei Xie 	struct rte_mbuf **m_table;
1080d19533e8SHuawei Xie 	struct rte_mbuf *mbuf, *prev;
1081d19533e8SHuawei Xie 	unsigned len, ret, offset = 0;
1082d19533e8SHuawei Xie 	const uint16_t lcore_id = rte_lcore_id();
1083d19533e8SHuawei Xie 	struct virtio_net_data_ll *dev_ll = ll_root_used;
1084d19533e8SHuawei Xie 	struct ether_hdr *pkt_hdr = rte_pktmbuf_mtod(m, struct ether_hdr *);
1085*e571e6b4SHuawei Xie 	struct virtio_net *dev = vdev->dev;
1086d19533e8SHuawei Xie 
1087d19533e8SHuawei Xie 	/*check if destination is local VM*/
1088*e571e6b4SHuawei Xie 	if ((vm2vm_mode == VM2VM_SOFTWARE) && (virtio_tx_local(vdev, m) == 0))
1089d19533e8SHuawei Xie 		return;
1090d19533e8SHuawei Xie 
1091d19533e8SHuawei Xie 	if (vm2vm_mode == VM2VM_HARDWARE) {
1092d19533e8SHuawei Xie 		while (dev_ll != NULL) {
1093*e571e6b4SHuawei Xie 			if ((dev_ll->vdev->ready == DEVICE_RX)
1094d19533e8SHuawei Xie 				&& ether_addr_cmp(&(pkt_hdr->d_addr),
1095*e571e6b4SHuawei Xie 				&dev_ll->vdev->mac_address)) {
1096d19533e8SHuawei Xie 				/*
1097d19533e8SHuawei Xie 				 * Drop the packet if the TX packet is
1098d19533e8SHuawei Xie 				 * destined for the TX device.
1099d19533e8SHuawei Xie 				 */
1100*e571e6b4SHuawei Xie 				if (dev_ll->vdev->dev->device_fh == dev->device_fh) {
1101d19533e8SHuawei Xie 					LOG_DEBUG(VHOST_DATA,
1102d19533e8SHuawei Xie 					"(%"PRIu64") TX: Source and destination"
1103d19533e8SHuawei Xie 					" MAC addresses are the same. Dropping "
1104d19533e8SHuawei Xie 					"packet.\n",
1105*e571e6b4SHuawei Xie 					dev_ll->vdev->device_fh);
1106d19533e8SHuawei Xie 					return;
1107d19533e8SHuawei Xie 				}
1108d19533e8SHuawei Xie 				offset = 4;
1109d19533e8SHuawei Xie 				vlan_tag =
1110d19533e8SHuawei Xie 				(uint16_t)
1111*e571e6b4SHuawei Xie 				vlan_tags[(uint16_t)dev_ll->vdev->dev->device_fh];
1112d19533e8SHuawei Xie 
1113d19533e8SHuawei Xie 				LOG_DEBUG(VHOST_DATA,
1114d19533e8SHuawei Xie 				"(%"PRIu64") TX: pkt to local VM device id:"
1115d19533e8SHuawei Xie 				"(%"PRIu64") vlan tag: %d.\n",
1116*e571e6b4SHuawei Xie 				dev->device_fh, dev_ll->vdev->dev->device_fh,
1117d19533e8SHuawei Xie 				vlan_tag);
1118d19533e8SHuawei Xie 
1119d19533e8SHuawei Xie 				break;
1120d19533e8SHuawei Xie 			}
1121d19533e8SHuawei Xie 			dev_ll = dev_ll->next;
1122d19533e8SHuawei Xie 		}
1123d19533e8SHuawei Xie 	}
1124d19533e8SHuawei Xie 
1125d19533e8SHuawei Xie 	LOG_DEBUG(VHOST_DATA, "(%"PRIu64") TX: MAC address is external\n", dev->device_fh);
1126d19533e8SHuawei Xie 
1127d19533e8SHuawei Xie 	/*Add packet to the port tx queue*/
1128d19533e8SHuawei Xie 	tx_q = &lcore_tx_queue[lcore_id];
1129d19533e8SHuawei Xie 	len = tx_q->len;
1130d19533e8SHuawei Xie 
1131d19533e8SHuawei Xie 	/* Allocate an mbuf and populate the structure. */
1132d19533e8SHuawei Xie 	mbuf = rte_pktmbuf_alloc(mbuf_pool);
1133d19533e8SHuawei Xie 	if (unlikely(mbuf == NULL)) {
1134d19533e8SHuawei Xie 		RTE_LOG(ERR, VHOST_DATA,
1135d19533e8SHuawei Xie 			"Failed to allocate memory for mbuf.\n");
1136d19533e8SHuawei Xie 		return;
1137d19533e8SHuawei Xie 	}
1138d19533e8SHuawei Xie 
1139d19533e8SHuawei Xie 	mbuf->data_len = m->data_len + VLAN_HLEN + offset;
1140d19533e8SHuawei Xie 	mbuf->pkt_len = m->pkt_len + VLAN_HLEN + offset;
1141d19533e8SHuawei Xie 	mbuf->nb_segs = m->nb_segs;
1142d19533e8SHuawei Xie 
1143d19533e8SHuawei Xie 	/* Copy ethernet header to mbuf. */
1144d19533e8SHuawei Xie 	rte_memcpy(rte_pktmbuf_mtod(mbuf, void *),
1145d19533e8SHuawei Xie 		rte_pktmbuf_mtod(m, const void *),
1146d19533e8SHuawei Xie 		ETH_HLEN);
1147d19533e8SHuawei Xie 
1148d19533e8SHuawei Xie 
1149d19533e8SHuawei Xie 	/* Setup vlan header. Bytes need to be re-ordered for network with htons()*/
1150d19533e8SHuawei Xie 	vlan_hdr = rte_pktmbuf_mtod(mbuf, struct vlan_ethhdr *);
1151d19533e8SHuawei Xie 	vlan_hdr->h_vlan_encapsulated_proto = vlan_hdr->h_vlan_proto;
1152d19533e8SHuawei Xie 	vlan_hdr->h_vlan_proto = htons(ETH_P_8021Q);
1153d19533e8SHuawei Xie 	vlan_hdr->h_vlan_TCI = htons(vlan_tag);
1154d19533e8SHuawei Xie 
1155d19533e8SHuawei Xie 	/* Copy the remaining packet contents to the mbuf. */
1156d19533e8SHuawei Xie 	rte_memcpy((void *)(rte_pktmbuf_mtod(mbuf, uint8_t *) + VLAN_ETH_HLEN),
1157d19533e8SHuawei Xie 		(const void *)(rte_pktmbuf_mtod(m, uint8_t *) + ETH_HLEN),
1158d19533e8SHuawei Xie 		(m->data_len - ETH_HLEN));
1159d19533e8SHuawei Xie 
1160d19533e8SHuawei Xie 	/* Copy the remaining segments for the whole packet. */
1161d19533e8SHuawei Xie 	prev = mbuf;
1162d19533e8SHuawei Xie 	while (m->next) {
1163d19533e8SHuawei Xie 		/* Allocate an mbuf and populate the structure. */
1164d19533e8SHuawei Xie 		struct rte_mbuf *next_mbuf = rte_pktmbuf_alloc(mbuf_pool);
1165d19533e8SHuawei Xie 		if (unlikely(next_mbuf == NULL)) {
1166d19533e8SHuawei Xie 			rte_pktmbuf_free(mbuf);
1167d19533e8SHuawei Xie 			RTE_LOG(ERR, VHOST_DATA,
1168d19533e8SHuawei Xie 				"Failed to allocate memory for mbuf.\n");
1169d19533e8SHuawei Xie 			return;
1170d19533e8SHuawei Xie 		}
1171d19533e8SHuawei Xie 
1172d19533e8SHuawei Xie 		m = m->next;
1173d19533e8SHuawei Xie 		prev->next = next_mbuf;
1174d19533e8SHuawei Xie 		prev = next_mbuf;
1175d19533e8SHuawei Xie 		next_mbuf->data_len = m->data_len;
1176d19533e8SHuawei Xie 
1177d19533e8SHuawei Xie 		/* Copy data to next mbuf. */
1178d19533e8SHuawei Xie 		rte_memcpy(rte_pktmbuf_mtod(next_mbuf, void *),
1179d19533e8SHuawei Xie 			rte_pktmbuf_mtod(m, const void *), m->data_len);
1180d19533e8SHuawei Xie 	}
1181d19533e8SHuawei Xie 
1182d19533e8SHuawei Xie 	tx_q->m_table[len] = mbuf;
1183d19533e8SHuawei Xie 	len++;
1184d19533e8SHuawei Xie 	if (enable_stats) {
1185d19533e8SHuawei Xie 		dev_statistics[dev->device_fh].tx_total++;
1186d19533e8SHuawei Xie 		dev_statistics[dev->device_fh].tx++;
1187d19533e8SHuawei Xie 	}
1188d19533e8SHuawei Xie 
1189d19533e8SHuawei Xie 	if (unlikely(len == MAX_PKT_BURST)) {
1190d19533e8SHuawei Xie 		m_table = (struct rte_mbuf **)tx_q->m_table;
1191d19533e8SHuawei Xie 		ret = rte_eth_tx_burst(ports[0], (uint16_t)tx_q->txq_id, m_table, (uint16_t) len);
1192d19533e8SHuawei Xie 		/* Free any buffers not handled by TX and update the port stats. */
1193d19533e8SHuawei Xie 		if (unlikely(ret < len)) {
1194d19533e8SHuawei Xie 			do {
1195d19533e8SHuawei Xie 				rte_pktmbuf_free(m_table[ret]);
1196d19533e8SHuawei Xie 			} while (++ret < len);
1197d19533e8SHuawei Xie 		}
1198d19533e8SHuawei Xie 
1199d19533e8SHuawei Xie 		len = 0;
1200d19533e8SHuawei Xie 	}
1201d19533e8SHuawei Xie 
1202d19533e8SHuawei Xie 	tx_q->len = len;
1203d19533e8SHuawei Xie 	return;
1204d19533e8SHuawei Xie }
1205d19533e8SHuawei Xie /*
1206d19533e8SHuawei Xie  * This function is called by each data core. It handles all RX/TX registered with the
1207d19533e8SHuawei Xie  * core. For TX the specific lcore linked list is used. For RX, MAC addresses are compared
1208d19533e8SHuawei Xie  * with all devices in the main linked list.
1209d19533e8SHuawei Xie  */
1210d19533e8SHuawei Xie static int
1211d19533e8SHuawei Xie switch_worker(__attribute__((unused)) void *arg)
1212d19533e8SHuawei Xie {
1213d19533e8SHuawei Xie 	struct rte_mempool *mbuf_pool = arg;
1214d19533e8SHuawei Xie 	struct virtio_net *dev = NULL;
1215*e571e6b4SHuawei Xie 	struct vhost_dev *vdev = NULL;
1216d19533e8SHuawei Xie 	struct rte_mbuf *pkts_burst[MAX_PKT_BURST];
1217d19533e8SHuawei Xie 	struct virtio_net_data_ll *dev_ll;
1218d19533e8SHuawei Xie 	struct mbuf_table *tx_q;
1219d19533e8SHuawei Xie 	volatile struct lcore_ll_info *lcore_ll;
1220d19533e8SHuawei Xie 	const uint64_t drain_tsc = (rte_get_tsc_hz() + US_PER_S - 1) / US_PER_S * BURST_TX_DRAIN_US;
1221d19533e8SHuawei Xie 	uint64_t prev_tsc, diff_tsc, cur_tsc, ret_count = 0;
1222d19533e8SHuawei Xie 	unsigned ret, i;
1223d19533e8SHuawei Xie 	const uint16_t lcore_id = rte_lcore_id();
1224d19533e8SHuawei Xie 	const uint16_t num_cores = (uint16_t)rte_lcore_count();
1225d19533e8SHuawei Xie 	uint16_t rx_count = 0;
1226d19533e8SHuawei Xie 	uint32_t mergeable = 0;
1227d19533e8SHuawei Xie 
1228d19533e8SHuawei Xie 	RTE_LOG(INFO, VHOST_DATA, "Procesing on Core %u started\n", lcore_id);
1229d19533e8SHuawei Xie 	lcore_ll = lcore_info[lcore_id].lcore_ll;
1230d19533e8SHuawei Xie 	prev_tsc = 0;
1231d19533e8SHuawei Xie 
1232d19533e8SHuawei Xie 	tx_q = &lcore_tx_queue[lcore_id];
1233d19533e8SHuawei Xie 	for (i = 0; i < num_cores; i ++) {
1234d19533e8SHuawei Xie 		if (lcore_ids[i] == lcore_id) {
1235d19533e8SHuawei Xie 			tx_q->txq_id = i;
1236d19533e8SHuawei Xie 			break;
1237d19533e8SHuawei Xie 		}
1238d19533e8SHuawei Xie 	}
1239d19533e8SHuawei Xie 
1240d19533e8SHuawei Xie 	while(1) {
1241d19533e8SHuawei Xie 		cur_tsc = rte_rdtsc();
1242d19533e8SHuawei Xie 		/*
1243d19533e8SHuawei Xie 		 * TX burst queue drain
1244d19533e8SHuawei Xie 		 */
1245d19533e8SHuawei Xie 		diff_tsc = cur_tsc - prev_tsc;
1246d19533e8SHuawei Xie 		if (unlikely(diff_tsc > drain_tsc)) {
1247d19533e8SHuawei Xie 
1248d19533e8SHuawei Xie 			if (tx_q->len) {
1249d19533e8SHuawei Xie 				LOG_DEBUG(VHOST_DATA, "TX queue drained after timeout with burst size %u \n", tx_q->len);
1250d19533e8SHuawei Xie 
1251d19533e8SHuawei Xie 				/*Tx any packets in the queue*/
1252d19533e8SHuawei Xie 				ret = rte_eth_tx_burst(ports[0], (uint16_t)tx_q->txq_id,
1253d19533e8SHuawei Xie 									   (struct rte_mbuf **)tx_q->m_table,
1254d19533e8SHuawei Xie 									   (uint16_t)tx_q->len);
1255d19533e8SHuawei Xie 				if (unlikely(ret < tx_q->len)) {
1256d19533e8SHuawei Xie 					do {
1257d19533e8SHuawei Xie 						rte_pktmbuf_free(tx_q->m_table[ret]);
1258d19533e8SHuawei Xie 					} while (++ret < tx_q->len);
1259d19533e8SHuawei Xie 				}
1260d19533e8SHuawei Xie 
1261d19533e8SHuawei Xie 				tx_q->len = 0;
1262d19533e8SHuawei Xie 			}
1263d19533e8SHuawei Xie 
1264d19533e8SHuawei Xie 			prev_tsc = cur_tsc;
1265d19533e8SHuawei Xie 
1266d19533e8SHuawei Xie 		}
1267d19533e8SHuawei Xie 
1268d19533e8SHuawei Xie 		rte_prefetch0(lcore_ll->ll_root_used);
1269d19533e8SHuawei Xie 		/*
1270d19533e8SHuawei Xie 		 * Inform the configuration core that we have exited the linked list and that no devices are
1271d19533e8SHuawei Xie 		 * in use if requested.
1272d19533e8SHuawei Xie 		 */
1273d19533e8SHuawei Xie 		if (lcore_ll->dev_removal_flag == REQUEST_DEV_REMOVAL)
1274d19533e8SHuawei Xie 			lcore_ll->dev_removal_flag = ACK_DEV_REMOVAL;
1275d19533e8SHuawei Xie 
1276d19533e8SHuawei Xie 		/*
1277d19533e8SHuawei Xie 		 * Process devices
1278d19533e8SHuawei Xie 		 */
1279d19533e8SHuawei Xie 		dev_ll = lcore_ll->ll_root_used;
1280d19533e8SHuawei Xie 
1281d19533e8SHuawei Xie 		while (dev_ll != NULL) {
1282d19533e8SHuawei Xie 			/*get virtio device ID*/
1283*e571e6b4SHuawei Xie 			vdev = dev_ll->vdev;
1284*e571e6b4SHuawei Xie 			dev = vdev->dev;
1285d19533e8SHuawei Xie 			mergeable =
1286d19533e8SHuawei Xie 				dev->features & (1 << VIRTIO_NET_F_MRG_RXBUF);
1287d19533e8SHuawei Xie 
1288*e571e6b4SHuawei Xie 			if (vdev->remove) {
1289d19533e8SHuawei Xie 				dev_ll = dev_ll->next;
1290*e571e6b4SHuawei Xie 				unlink_vmdq(vdev);
1291*e571e6b4SHuawei Xie 				vdev->ready = DEVICE_SAFE_REMOVE;
1292d19533e8SHuawei Xie 				continue;
1293d19533e8SHuawei Xie 			}
1294*e571e6b4SHuawei Xie 			if (likely(vdev->ready == DEVICE_RX)) {
1295d19533e8SHuawei Xie 				/*Handle guest RX*/
1296d19533e8SHuawei Xie 				rx_count = rte_eth_rx_burst(ports[0],
1297*e571e6b4SHuawei Xie 					vdev->vmdq_rx_q, pkts_burst, MAX_PKT_BURST);
1298d19533e8SHuawei Xie 
1299d19533e8SHuawei Xie 				if (rx_count) {
1300d19533e8SHuawei Xie 					if (likely(mergeable == 0))
1301d19533e8SHuawei Xie 						ret_count =
1302d19533e8SHuawei Xie 							virtio_dev_rx(dev,
1303d19533e8SHuawei Xie 							pkts_burst, rx_count);
1304d19533e8SHuawei Xie 					else
1305d19533e8SHuawei Xie 						ret_count =
1306d19533e8SHuawei Xie 							virtio_dev_merge_rx(dev,
1307d19533e8SHuawei Xie 							pkts_burst, rx_count);
1308d19533e8SHuawei Xie 
1309d19533e8SHuawei Xie 					if (enable_stats) {
1310d19533e8SHuawei Xie 						rte_atomic64_add(
1311*e571e6b4SHuawei Xie 						&dev_statistics[dev_ll->vdev->dev->device_fh].rx_total_atomic,
1312d19533e8SHuawei Xie 						rx_count);
1313d19533e8SHuawei Xie 						rte_atomic64_add(
1314*e571e6b4SHuawei Xie 						&dev_statistics[dev_ll->vdev->dev->device_fh].rx_atomic, ret_count);
1315d19533e8SHuawei Xie 					}
1316d19533e8SHuawei Xie 					while (likely(rx_count)) {
1317d19533e8SHuawei Xie 						rx_count--;
1318d19533e8SHuawei Xie 						rte_pktmbuf_free(pkts_burst[rx_count]);
1319d19533e8SHuawei Xie 					}
1320d19533e8SHuawei Xie 
1321d19533e8SHuawei Xie 				}
1322d19533e8SHuawei Xie 			}
1323d19533e8SHuawei Xie 
1324*e571e6b4SHuawei Xie 			if (!vdev->remove) {
1325d19533e8SHuawei Xie 				/*Handle guest TX*/
1326d19533e8SHuawei Xie 				if (likely(mergeable == 0))
1327d19533e8SHuawei Xie 					virtio_dev_tx(dev, mbuf_pool);
1328d19533e8SHuawei Xie 				else
1329d19533e8SHuawei Xie 					virtio_dev_merge_tx(dev, mbuf_pool);
1330d19533e8SHuawei Xie 			}
1331d19533e8SHuawei Xie 
1332d19533e8SHuawei Xie 			/*move to the next device in the list*/
1333d19533e8SHuawei Xie 			dev_ll = dev_ll->next;
1334d19533e8SHuawei Xie 		}
1335d19533e8SHuawei Xie 	}
1336d19533e8SHuawei Xie 
1337d19533e8SHuawei Xie 	return 0;
1338d19533e8SHuawei Xie }
1339d19533e8SHuawei Xie 
1340d19533e8SHuawei Xie /*
1341d19533e8SHuawei Xie  * This function gets available ring number for zero copy rx.
1342d19533e8SHuawei Xie  * Only one thread will call this funciton for a paticular virtio device,
1343d19533e8SHuawei Xie  * so, it is designed as non-thread-safe function.
1344d19533e8SHuawei Xie  */
1345d19533e8SHuawei Xie static inline uint32_t __attribute__((always_inline))
1346d19533e8SHuawei Xie get_available_ring_num_zcp(struct virtio_net *dev)
1347d19533e8SHuawei Xie {
1348d19533e8SHuawei Xie 	struct vhost_virtqueue *vq = dev->virtqueue[VIRTIO_RXQ];
1349d19533e8SHuawei Xie 	uint16_t avail_idx;
1350d19533e8SHuawei Xie 
1351d19533e8SHuawei Xie 	avail_idx = *((volatile uint16_t *)&vq->avail->idx);
1352d19533e8SHuawei Xie 	return (uint32_t)(avail_idx - vq->last_used_idx_res);
1353d19533e8SHuawei Xie }
1354d19533e8SHuawei Xie 
1355d19533e8SHuawei Xie /*
1356d19533e8SHuawei Xie  * This function gets available ring index for zero copy rx,
1357d19533e8SHuawei Xie  * it will retry 'burst_rx_retry_num' times till it get enough ring index.
1358d19533e8SHuawei Xie  * Only one thread will call this funciton for a paticular virtio device,
1359d19533e8SHuawei Xie  * so, it is designed as non-thread-safe function.
1360d19533e8SHuawei Xie  */
1361d19533e8SHuawei Xie static inline uint32_t __attribute__((always_inline))
1362d19533e8SHuawei Xie get_available_ring_index_zcp(struct virtio_net *dev,
1363d19533e8SHuawei Xie 	uint16_t *res_base_idx, uint32_t count)
1364d19533e8SHuawei Xie {
1365d19533e8SHuawei Xie 	struct vhost_virtqueue *vq = dev->virtqueue[VIRTIO_RXQ];
1366d19533e8SHuawei Xie 	uint16_t avail_idx;
1367d19533e8SHuawei Xie 	uint32_t retry = 0;
1368d19533e8SHuawei Xie 	uint16_t free_entries;
1369d19533e8SHuawei Xie 
1370d19533e8SHuawei Xie 	*res_base_idx = vq->last_used_idx_res;
1371d19533e8SHuawei Xie 	avail_idx = *((volatile uint16_t *)&vq->avail->idx);
1372d19533e8SHuawei Xie 	free_entries = (avail_idx - *res_base_idx);
1373d19533e8SHuawei Xie 
1374d19533e8SHuawei Xie 	LOG_DEBUG(VHOST_DATA, "(%"PRIu64") in get_available_ring_index_zcp: "
1375d19533e8SHuawei Xie 			"avail idx: %d, "
1376d19533e8SHuawei Xie 			"res base idx:%d, free entries:%d\n",
1377d19533e8SHuawei Xie 			dev->device_fh, avail_idx, *res_base_idx,
1378d19533e8SHuawei Xie 			free_entries);
1379d19533e8SHuawei Xie 
1380d19533e8SHuawei Xie 	/*
1381d19533e8SHuawei Xie 	 * If retry is enabled and the queue is full then we wait
1382d19533e8SHuawei Xie 	 * and retry to avoid packet loss.
1383d19533e8SHuawei Xie 	 */
1384d19533e8SHuawei Xie 	if (enable_retry && unlikely(count > free_entries)) {
1385d19533e8SHuawei Xie 		for (retry = 0; retry < burst_rx_retry_num; retry++) {
1386d19533e8SHuawei Xie 			rte_delay_us(burst_rx_delay_time);
1387d19533e8SHuawei Xie 			avail_idx = *((volatile uint16_t *)&vq->avail->idx);
1388d19533e8SHuawei Xie 			free_entries = (avail_idx - *res_base_idx);
1389d19533e8SHuawei Xie 			if (count <= free_entries)
1390d19533e8SHuawei Xie 				break;
1391d19533e8SHuawei Xie 		}
1392d19533e8SHuawei Xie 	}
1393d19533e8SHuawei Xie 
1394d19533e8SHuawei Xie 	/*check that we have enough buffers*/
1395d19533e8SHuawei Xie 	if (unlikely(count > free_entries))
1396d19533e8SHuawei Xie 		count = free_entries;
1397d19533e8SHuawei Xie 
1398d19533e8SHuawei Xie 	if (unlikely(count == 0)) {
1399d19533e8SHuawei Xie 		LOG_DEBUG(VHOST_DATA,
1400d19533e8SHuawei Xie 			"(%"PRIu64") Fail in get_available_ring_index_zcp: "
1401d19533e8SHuawei Xie 			"avail idx: %d, res base idx:%d, free entries:%d\n",
1402d19533e8SHuawei Xie 			dev->device_fh, avail_idx,
1403d19533e8SHuawei Xie 			*res_base_idx, free_entries);
1404d19533e8SHuawei Xie 		return 0;
1405d19533e8SHuawei Xie 	}
1406d19533e8SHuawei Xie 
1407d19533e8SHuawei Xie 	vq->last_used_idx_res = *res_base_idx + count;
1408d19533e8SHuawei Xie 
1409d19533e8SHuawei Xie 	return count;
1410d19533e8SHuawei Xie }
1411d19533e8SHuawei Xie 
1412d19533e8SHuawei Xie /*
1413d19533e8SHuawei Xie  * This function put descriptor back to used list.
1414d19533e8SHuawei Xie  */
1415d19533e8SHuawei Xie static inline void __attribute__((always_inline))
1416d19533e8SHuawei Xie put_desc_to_used_list_zcp(struct vhost_virtqueue *vq, uint16_t desc_idx)
1417d19533e8SHuawei Xie {
1418d19533e8SHuawei Xie 	uint16_t res_cur_idx = vq->last_used_idx;
1419d19533e8SHuawei Xie 	vq->used->ring[res_cur_idx & (vq->size - 1)].id = (uint32_t)desc_idx;
1420d19533e8SHuawei Xie 	vq->used->ring[res_cur_idx & (vq->size - 1)].len = 0;
1421d19533e8SHuawei Xie 	rte_compiler_barrier();
1422d19533e8SHuawei Xie 	*(volatile uint16_t *)&vq->used->idx += 1;
1423d19533e8SHuawei Xie 	vq->last_used_idx += 1;
1424d19533e8SHuawei Xie 
1425d19533e8SHuawei Xie 	/* Kick the guest if necessary. */
1426d19533e8SHuawei Xie 	if (!(vq->avail->flags & VRING_AVAIL_F_NO_INTERRUPT))
1427d19533e8SHuawei Xie 		eventfd_write((int)vq->kickfd, 1);
1428d19533e8SHuawei Xie }
1429d19533e8SHuawei Xie 
1430d19533e8SHuawei Xie /*
1431d19533e8SHuawei Xie  * This function get available descriptor from vitio vring and un-attached mbuf
1432d19533e8SHuawei Xie  * from vpool->ring, and then attach them together. It needs adjust the offset
1433d19533e8SHuawei Xie  * for buff_addr and phys_addr accroding to PMD implementation, otherwise the
1434d19533e8SHuawei Xie  * frame data may be put to wrong location in mbuf.
1435d19533e8SHuawei Xie  */
1436d19533e8SHuawei Xie static inline void __attribute__((always_inline))
1437d19533e8SHuawei Xie attach_rxmbuf_zcp(struct virtio_net *dev)
1438d19533e8SHuawei Xie {
1439d19533e8SHuawei Xie 	uint16_t res_base_idx, desc_idx;
1440d19533e8SHuawei Xie 	uint64_t buff_addr, phys_addr;
1441d19533e8SHuawei Xie 	struct vhost_virtqueue *vq;
1442d19533e8SHuawei Xie 	struct vring_desc *desc;
1443d19533e8SHuawei Xie 	struct rte_mbuf *mbuf = NULL;
1444d19533e8SHuawei Xie 	struct vpool *vpool;
1445d19533e8SHuawei Xie 	hpa_type addr_type;
1446*e571e6b4SHuawei Xie 	struct vhost_dev *vdev = (struct vhost_dev *)dev->priv;
1447d19533e8SHuawei Xie 
1448*e571e6b4SHuawei Xie 	vpool = &vpool_array[vdev->vmdq_rx_q];
1449d19533e8SHuawei Xie 	vq = dev->virtqueue[VIRTIO_RXQ];
1450d19533e8SHuawei Xie 
1451d19533e8SHuawei Xie 	do {
1452*e571e6b4SHuawei Xie 		if (unlikely(get_available_ring_index_zcp(vdev->dev, &res_base_idx,
1453d19533e8SHuawei Xie 				1) != 1))
1454d19533e8SHuawei Xie 			return;
1455d19533e8SHuawei Xie 		desc_idx = vq->avail->ring[(res_base_idx) & (vq->size - 1)];
1456d19533e8SHuawei Xie 
1457d19533e8SHuawei Xie 		desc = &vq->desc[desc_idx];
1458d19533e8SHuawei Xie 		if (desc->flags & VRING_DESC_F_NEXT) {
1459d19533e8SHuawei Xie 			desc = &vq->desc[desc->next];
1460d19533e8SHuawei Xie 			buff_addr = gpa_to_vva(dev, desc->addr);
1461*e571e6b4SHuawei Xie 			phys_addr = gpa_to_hpa(vdev, desc->addr, desc->len,
1462d19533e8SHuawei Xie 					&addr_type);
1463d19533e8SHuawei Xie 		} else {
1464d19533e8SHuawei Xie 			buff_addr = gpa_to_vva(dev,
1465d19533e8SHuawei Xie 					desc->addr + vq->vhost_hlen);
1466*e571e6b4SHuawei Xie 			phys_addr = gpa_to_hpa(vdev,
1467d19533e8SHuawei Xie 					desc->addr + vq->vhost_hlen,
1468d19533e8SHuawei Xie 					desc->len, &addr_type);
1469d19533e8SHuawei Xie 		}
1470d19533e8SHuawei Xie 
1471d19533e8SHuawei Xie 		if (unlikely(addr_type == PHYS_ADDR_INVALID)) {
1472d19533e8SHuawei Xie 			RTE_LOG(ERR, VHOST_DATA, "(%"PRIu64") Invalid frame buffer"
1473d19533e8SHuawei Xie 				" address found when attaching RX frame buffer"
1474d19533e8SHuawei Xie 				" address!\n", dev->device_fh);
1475d19533e8SHuawei Xie 			put_desc_to_used_list_zcp(vq, desc_idx);
1476d19533e8SHuawei Xie 			continue;
1477d19533e8SHuawei Xie 		}
1478d19533e8SHuawei Xie 
1479d19533e8SHuawei Xie 		/*
1480d19533e8SHuawei Xie 		 * Check if the frame buffer address from guest crosses
1481d19533e8SHuawei Xie 		 * sub-region or not.
1482d19533e8SHuawei Xie 		 */
1483d19533e8SHuawei Xie 		if (unlikely(addr_type == PHYS_ADDR_CROSS_SUBREG)) {
1484d19533e8SHuawei Xie 			RTE_LOG(ERR, VHOST_DATA,
1485d19533e8SHuawei Xie 				"(%"PRIu64") Frame buffer address cross "
1486d19533e8SHuawei Xie 				"sub-regioin found when attaching RX frame "
1487d19533e8SHuawei Xie 				"buffer address!\n",
1488d19533e8SHuawei Xie 				dev->device_fh);
1489d19533e8SHuawei Xie 			put_desc_to_used_list_zcp(vq, desc_idx);
1490d19533e8SHuawei Xie 			continue;
1491d19533e8SHuawei Xie 		}
1492d19533e8SHuawei Xie 	} while (unlikely(phys_addr == 0));
1493d19533e8SHuawei Xie 
1494d19533e8SHuawei Xie 	rte_ring_sc_dequeue(vpool->ring, (void **)&mbuf);
1495d19533e8SHuawei Xie 	if (unlikely(mbuf == NULL)) {
1496d19533e8SHuawei Xie 		LOG_DEBUG(VHOST_DATA,
1497d19533e8SHuawei Xie 			"(%"PRIu64") in attach_rxmbuf_zcp: "
1498d19533e8SHuawei Xie 			"ring_sc_dequeue fail.\n",
1499d19533e8SHuawei Xie 			dev->device_fh);
1500d19533e8SHuawei Xie 		put_desc_to_used_list_zcp(vq, desc_idx);
1501d19533e8SHuawei Xie 		return;
1502d19533e8SHuawei Xie 	}
1503d19533e8SHuawei Xie 
1504d19533e8SHuawei Xie 	if (unlikely(vpool->buf_size > desc->len)) {
1505d19533e8SHuawei Xie 		LOG_DEBUG(VHOST_DATA,
1506d19533e8SHuawei Xie 			"(%"PRIu64") in attach_rxmbuf_zcp: frame buffer "
1507d19533e8SHuawei Xie 			"length(%d) of descriptor idx: %d less than room "
1508d19533e8SHuawei Xie 			"size required: %d\n",
1509d19533e8SHuawei Xie 			dev->device_fh, desc->len, desc_idx, vpool->buf_size);
1510d19533e8SHuawei Xie 		put_desc_to_used_list_zcp(vq, desc_idx);
1511d19533e8SHuawei Xie 		rte_ring_sp_enqueue(vpool->ring, (void *)mbuf);
1512d19533e8SHuawei Xie 		return;
1513d19533e8SHuawei Xie 	}
1514d19533e8SHuawei Xie 
1515d19533e8SHuawei Xie 	mbuf->buf_addr = (void *)(uintptr_t)(buff_addr - RTE_PKTMBUF_HEADROOM);
1516d19533e8SHuawei Xie 	mbuf->data_off = RTE_PKTMBUF_HEADROOM;
1517d19533e8SHuawei Xie 	mbuf->buf_physaddr = phys_addr - RTE_PKTMBUF_HEADROOM;
1518d19533e8SHuawei Xie 	mbuf->data_len = desc->len;
1519d19533e8SHuawei Xie 	MBUF_HEADROOM_UINT32(mbuf) = (uint32_t)desc_idx;
1520d19533e8SHuawei Xie 
1521d19533e8SHuawei Xie 	LOG_DEBUG(VHOST_DATA,
1522d19533e8SHuawei Xie 		"(%"PRIu64") in attach_rxmbuf_zcp: res base idx:%d, "
1523d19533e8SHuawei Xie 		"descriptor idx:%d\n",
1524d19533e8SHuawei Xie 		dev->device_fh, res_base_idx, desc_idx);
1525d19533e8SHuawei Xie 
1526d19533e8SHuawei Xie 	__rte_mbuf_raw_free(mbuf);
1527d19533e8SHuawei Xie 
1528d19533e8SHuawei Xie 	return;
1529d19533e8SHuawei Xie }
1530d19533e8SHuawei Xie 
1531d19533e8SHuawei Xie /*
1532d19533e8SHuawei Xie  * Detach an attched packet mbuf -
1533d19533e8SHuawei Xie  *  - restore original mbuf address and length values.
1534d19533e8SHuawei Xie  *  - reset pktmbuf data and data_len to their default values.
1535d19533e8SHuawei Xie  *  All other fields of the given packet mbuf will be left intact.
1536d19533e8SHuawei Xie  *
1537d19533e8SHuawei Xie  * @param m
1538d19533e8SHuawei Xie  *   The attached packet mbuf.
1539d19533e8SHuawei Xie  */
1540d19533e8SHuawei Xie static inline void pktmbuf_detach_zcp(struct rte_mbuf *m)
1541d19533e8SHuawei Xie {
1542d19533e8SHuawei Xie 	const struct rte_mempool *mp = m->pool;
1543d19533e8SHuawei Xie 	void *buf = RTE_MBUF_TO_BADDR(m);
1544d19533e8SHuawei Xie 	uint32_t buf_ofs;
1545d19533e8SHuawei Xie 	uint32_t buf_len = mp->elt_size - sizeof(*m);
1546d19533e8SHuawei Xie 	m->buf_physaddr = rte_mempool_virt2phy(mp, m) + sizeof(*m);
1547d19533e8SHuawei Xie 
1548d19533e8SHuawei Xie 	m->buf_addr = buf;
1549d19533e8SHuawei Xie 	m->buf_len = (uint16_t)buf_len;
1550d19533e8SHuawei Xie 
1551d19533e8SHuawei Xie 	buf_ofs = (RTE_PKTMBUF_HEADROOM <= m->buf_len) ?
1552d19533e8SHuawei Xie 			RTE_PKTMBUF_HEADROOM : m->buf_len;
1553d19533e8SHuawei Xie 	m->data_off = buf_ofs;
1554d19533e8SHuawei Xie 
1555d19533e8SHuawei Xie 	m->data_len = 0;
1556d19533e8SHuawei Xie }
1557d19533e8SHuawei Xie 
1558d19533e8SHuawei Xie /*
1559d19533e8SHuawei Xie  * This function is called after packets have been transimited. It fetchs mbuf
1560d19533e8SHuawei Xie  * from vpool->pool, detached it and put into vpool->ring. It also update the
1561d19533e8SHuawei Xie  * used index and kick the guest if necessary.
1562d19533e8SHuawei Xie  */
1563d19533e8SHuawei Xie static inline uint32_t __attribute__((always_inline))
1564d19533e8SHuawei Xie txmbuf_clean_zcp(struct virtio_net *dev, struct vpool *vpool)
1565d19533e8SHuawei Xie {
1566d19533e8SHuawei Xie 	struct rte_mbuf *mbuf;
1567d19533e8SHuawei Xie 	struct vhost_virtqueue *vq = dev->virtqueue[VIRTIO_TXQ];
1568d19533e8SHuawei Xie 	uint32_t used_idx = vq->last_used_idx & (vq->size - 1);
1569d19533e8SHuawei Xie 	uint32_t index = 0;
1570d19533e8SHuawei Xie 	uint32_t mbuf_count = rte_mempool_count(vpool->pool);
1571d19533e8SHuawei Xie 
1572d19533e8SHuawei Xie 	LOG_DEBUG(VHOST_DATA,
1573d19533e8SHuawei Xie 		"(%"PRIu64") in txmbuf_clean_zcp: mbuf count in mempool before "
1574d19533e8SHuawei Xie 		"clean is: %d\n",
1575d19533e8SHuawei Xie 		dev->device_fh, mbuf_count);
1576d19533e8SHuawei Xie 	LOG_DEBUG(VHOST_DATA,
1577d19533e8SHuawei Xie 		"(%"PRIu64") in txmbuf_clean_zcp: mbuf count in  ring before "
1578d19533e8SHuawei Xie 		"clean  is : %d\n",
1579d19533e8SHuawei Xie 		dev->device_fh, rte_ring_count(vpool->ring));
1580d19533e8SHuawei Xie 
1581d19533e8SHuawei Xie 	for (index = 0; index < mbuf_count; index++) {
1582d19533e8SHuawei Xie 		mbuf = __rte_mbuf_raw_alloc(vpool->pool);
1583d19533e8SHuawei Xie 		if (likely(RTE_MBUF_INDIRECT(mbuf)))
1584d19533e8SHuawei Xie 			pktmbuf_detach_zcp(mbuf);
1585d19533e8SHuawei Xie 		rte_ring_sp_enqueue(vpool->ring, mbuf);
1586d19533e8SHuawei Xie 
1587d19533e8SHuawei Xie 		/* Update used index buffer information. */
1588d19533e8SHuawei Xie 		vq->used->ring[used_idx].id = MBUF_HEADROOM_UINT32(mbuf);
1589d19533e8SHuawei Xie 		vq->used->ring[used_idx].len = 0;
1590d19533e8SHuawei Xie 
1591d19533e8SHuawei Xie 		used_idx = (used_idx + 1) & (vq->size - 1);
1592d19533e8SHuawei Xie 	}
1593d19533e8SHuawei Xie 
1594d19533e8SHuawei Xie 	LOG_DEBUG(VHOST_DATA,
1595d19533e8SHuawei Xie 		"(%"PRIu64") in txmbuf_clean_zcp: mbuf count in mempool after "
1596d19533e8SHuawei Xie 		"clean is: %d\n",
1597d19533e8SHuawei Xie 		dev->device_fh, rte_mempool_count(vpool->pool));
1598d19533e8SHuawei Xie 	LOG_DEBUG(VHOST_DATA,
1599d19533e8SHuawei Xie 		"(%"PRIu64") in txmbuf_clean_zcp: mbuf count in  ring after "
1600d19533e8SHuawei Xie 		"clean  is : %d\n",
1601d19533e8SHuawei Xie 		dev->device_fh, rte_ring_count(vpool->ring));
1602d19533e8SHuawei Xie 	LOG_DEBUG(VHOST_DATA,
1603d19533e8SHuawei Xie 		"(%"PRIu64") in txmbuf_clean_zcp: before updated "
1604d19533e8SHuawei Xie 		"vq->last_used_idx:%d\n",
1605d19533e8SHuawei Xie 		dev->device_fh, vq->last_used_idx);
1606d19533e8SHuawei Xie 
1607d19533e8SHuawei Xie 	vq->last_used_idx += mbuf_count;
1608d19533e8SHuawei Xie 
1609d19533e8SHuawei Xie 	LOG_DEBUG(VHOST_DATA,
1610d19533e8SHuawei Xie 		"(%"PRIu64") in txmbuf_clean_zcp: after updated "
1611d19533e8SHuawei Xie 		"vq->last_used_idx:%d\n",
1612d19533e8SHuawei Xie 		dev->device_fh, vq->last_used_idx);
1613d19533e8SHuawei Xie 
1614d19533e8SHuawei Xie 	rte_compiler_barrier();
1615d19533e8SHuawei Xie 
1616d19533e8SHuawei Xie 	*(volatile uint16_t *)&vq->used->idx += mbuf_count;
1617d19533e8SHuawei Xie 
1618d19533e8SHuawei Xie 	/* Kick guest if required. */
1619d19533e8SHuawei Xie 	if (!(vq->avail->flags & VRING_AVAIL_F_NO_INTERRUPT))
1620d19533e8SHuawei Xie 		eventfd_write((int)vq->kickfd, 1);
1621d19533e8SHuawei Xie 
1622d19533e8SHuawei Xie 	return 0;
1623d19533e8SHuawei Xie }
1624d19533e8SHuawei Xie 
1625d19533e8SHuawei Xie /*
1626d19533e8SHuawei Xie  * This function is called when a virtio device is destroy.
1627d19533e8SHuawei Xie  * It fetchs mbuf from vpool->pool, and detached it, and put into vpool->ring.
1628d19533e8SHuawei Xie  */
1629d19533e8SHuawei Xie static void mbuf_destroy_zcp(struct vpool *vpool)
1630d19533e8SHuawei Xie {
1631d19533e8SHuawei Xie 	struct rte_mbuf *mbuf = NULL;
1632d19533e8SHuawei Xie 	uint32_t index, mbuf_count = rte_mempool_count(vpool->pool);
1633d19533e8SHuawei Xie 
1634d19533e8SHuawei Xie 	LOG_DEBUG(VHOST_CONFIG,
1635d19533e8SHuawei Xie 		"in mbuf_destroy_zcp: mbuf count in mempool before "
1636d19533e8SHuawei Xie 		"mbuf_destroy_zcp is: %d\n",
1637d19533e8SHuawei Xie 		mbuf_count);
1638d19533e8SHuawei Xie 	LOG_DEBUG(VHOST_CONFIG,
1639d19533e8SHuawei Xie 		"in mbuf_destroy_zcp: mbuf count in  ring before "
1640d19533e8SHuawei Xie 		"mbuf_destroy_zcp  is : %d\n",
1641d19533e8SHuawei Xie 		rte_ring_count(vpool->ring));
1642d19533e8SHuawei Xie 
1643d19533e8SHuawei Xie 	for (index = 0; index < mbuf_count; index++) {
1644d19533e8SHuawei Xie 		mbuf = __rte_mbuf_raw_alloc(vpool->pool);
1645d19533e8SHuawei Xie 		if (likely(mbuf != NULL)) {
1646d19533e8SHuawei Xie 			if (likely(RTE_MBUF_INDIRECT(mbuf)))
1647d19533e8SHuawei Xie 				pktmbuf_detach_zcp(mbuf);
1648d19533e8SHuawei Xie 			rte_ring_sp_enqueue(vpool->ring, (void *)mbuf);
1649d19533e8SHuawei Xie 		}
1650d19533e8SHuawei Xie 	}
1651d19533e8SHuawei Xie 
1652d19533e8SHuawei Xie 	LOG_DEBUG(VHOST_CONFIG,
1653d19533e8SHuawei Xie 		"in mbuf_destroy_zcp: mbuf count in mempool after "
1654d19533e8SHuawei Xie 		"mbuf_destroy_zcp is: %d\n",
1655d19533e8SHuawei Xie 		rte_mempool_count(vpool->pool));
1656d19533e8SHuawei Xie 	LOG_DEBUG(VHOST_CONFIG,
1657d19533e8SHuawei Xie 		"in mbuf_destroy_zcp: mbuf count in ring after "
1658d19533e8SHuawei Xie 		"mbuf_destroy_zcp is : %d\n",
1659d19533e8SHuawei Xie 		rte_ring_count(vpool->ring));
1660d19533e8SHuawei Xie }
1661d19533e8SHuawei Xie 
1662d19533e8SHuawei Xie /*
1663d19533e8SHuawei Xie  * This function update the use flag and counter.
1664d19533e8SHuawei Xie  */
1665d19533e8SHuawei Xie static inline uint32_t __attribute__((always_inline))
1666d19533e8SHuawei Xie virtio_dev_rx_zcp(struct virtio_net *dev, struct rte_mbuf **pkts,
1667d19533e8SHuawei Xie 	uint32_t count)
1668d19533e8SHuawei Xie {
1669d19533e8SHuawei Xie 	struct vhost_virtqueue *vq;
1670d19533e8SHuawei Xie 	struct vring_desc *desc;
1671d19533e8SHuawei Xie 	struct rte_mbuf *buff;
1672d19533e8SHuawei Xie 	/* The virtio_hdr is initialised to 0. */
1673d19533e8SHuawei Xie 	struct virtio_net_hdr_mrg_rxbuf virtio_hdr
1674d19533e8SHuawei Xie 		= {{0, 0, 0, 0, 0, 0}, 0};
1675d19533e8SHuawei Xie 	uint64_t buff_hdr_addr = 0;
1676d19533e8SHuawei Xie 	uint32_t head[MAX_PKT_BURST], packet_len = 0;
1677d19533e8SHuawei Xie 	uint32_t head_idx, packet_success = 0;
1678d19533e8SHuawei Xie 	uint16_t res_cur_idx;
1679d19533e8SHuawei Xie 
1680d19533e8SHuawei Xie 	LOG_DEBUG(VHOST_DATA, "(%"PRIu64") virtio_dev_rx()\n", dev->device_fh);
1681d19533e8SHuawei Xie 
1682d19533e8SHuawei Xie 	if (count == 0)
1683d19533e8SHuawei Xie 		return 0;
1684d19533e8SHuawei Xie 
1685d19533e8SHuawei Xie 	vq = dev->virtqueue[VIRTIO_RXQ];
1686d19533e8SHuawei Xie 	count = (count > MAX_PKT_BURST) ? MAX_PKT_BURST : count;
1687d19533e8SHuawei Xie 
1688d19533e8SHuawei Xie 	res_cur_idx = vq->last_used_idx;
1689d19533e8SHuawei Xie 	LOG_DEBUG(VHOST_DATA, "(%"PRIu64") Current Index %d| End Index %d\n",
1690d19533e8SHuawei Xie 		dev->device_fh, res_cur_idx, res_cur_idx + count);
1691d19533e8SHuawei Xie 
1692d19533e8SHuawei Xie 	/* Retrieve all of the head indexes first to avoid caching issues. */
1693d19533e8SHuawei Xie 	for (head_idx = 0; head_idx < count; head_idx++)
1694d19533e8SHuawei Xie 		head[head_idx] = MBUF_HEADROOM_UINT32(pkts[head_idx]);
1695d19533e8SHuawei Xie 
1696d19533e8SHuawei Xie 	/*Prefetch descriptor index. */
1697d19533e8SHuawei Xie 	rte_prefetch0(&vq->desc[head[packet_success]]);
1698d19533e8SHuawei Xie 
1699d19533e8SHuawei Xie 	while (packet_success != count) {
1700d19533e8SHuawei Xie 		/* Get descriptor from available ring */
1701d19533e8SHuawei Xie 		desc = &vq->desc[head[packet_success]];
1702d19533e8SHuawei Xie 
1703d19533e8SHuawei Xie 		buff = pkts[packet_success];
1704d19533e8SHuawei Xie 		LOG_DEBUG(VHOST_DATA,
1705d19533e8SHuawei Xie 			"(%"PRIu64") in dev_rx_zcp: update the used idx for "
1706d19533e8SHuawei Xie 			"pkt[%d] descriptor idx: %d\n",
1707d19533e8SHuawei Xie 			dev->device_fh, packet_success,
1708d19533e8SHuawei Xie 			MBUF_HEADROOM_UINT32(buff));
1709d19533e8SHuawei Xie 
1710d19533e8SHuawei Xie 		PRINT_PACKET(dev,
1711d19533e8SHuawei Xie 			(uintptr_t)(((uint64_t)(uintptr_t)buff->buf_addr)
1712d19533e8SHuawei Xie 			+ RTE_PKTMBUF_HEADROOM),
1713d19533e8SHuawei Xie 			rte_pktmbuf_data_len(buff), 0);
1714d19533e8SHuawei Xie 
1715d19533e8SHuawei Xie 		/* Buffer address translation for virtio header. */
1716d19533e8SHuawei Xie 		buff_hdr_addr = gpa_to_vva(dev, desc->addr);
1717d19533e8SHuawei Xie 		packet_len = rte_pktmbuf_data_len(buff) + vq->vhost_hlen;
1718d19533e8SHuawei Xie 
1719d19533e8SHuawei Xie 		/*
1720d19533e8SHuawei Xie 		 * If the descriptors are chained the header and data are
1721d19533e8SHuawei Xie 		 * placed in separate buffers.
1722d19533e8SHuawei Xie 		 */
1723d19533e8SHuawei Xie 		if (desc->flags & VRING_DESC_F_NEXT) {
1724d19533e8SHuawei Xie 			desc->len = vq->vhost_hlen;
1725d19533e8SHuawei Xie 			desc = &vq->desc[desc->next];
1726d19533e8SHuawei Xie 			desc->len = rte_pktmbuf_data_len(buff);
1727d19533e8SHuawei Xie 		} else {
1728d19533e8SHuawei Xie 			desc->len = packet_len;
1729d19533e8SHuawei Xie 		}
1730d19533e8SHuawei Xie 
1731d19533e8SHuawei Xie 		/* Update used ring with desc information */
1732d19533e8SHuawei Xie 		vq->used->ring[res_cur_idx & (vq->size - 1)].id
1733d19533e8SHuawei Xie 			= head[packet_success];
1734d19533e8SHuawei Xie 		vq->used->ring[res_cur_idx & (vq->size - 1)].len
1735d19533e8SHuawei Xie 			= packet_len;
1736d19533e8SHuawei Xie 		res_cur_idx++;
1737d19533e8SHuawei Xie 		packet_success++;
1738d19533e8SHuawei Xie 
1739d19533e8SHuawei Xie 		/* A header is required per buffer. */
1740d19533e8SHuawei Xie 		rte_memcpy((void *)(uintptr_t)buff_hdr_addr,
1741d19533e8SHuawei Xie 			(const void *)&virtio_hdr, vq->vhost_hlen);
1742d19533e8SHuawei Xie 
1743d19533e8SHuawei Xie 		PRINT_PACKET(dev, (uintptr_t)buff_hdr_addr, vq->vhost_hlen, 1);
1744d19533e8SHuawei Xie 
1745d19533e8SHuawei Xie 		if (likely(packet_success < count)) {
1746d19533e8SHuawei Xie 			/* Prefetch descriptor index. */
1747d19533e8SHuawei Xie 			rte_prefetch0(&vq->desc[head[packet_success]]);
1748d19533e8SHuawei Xie 		}
1749d19533e8SHuawei Xie 	}
1750d19533e8SHuawei Xie 
1751d19533e8SHuawei Xie 	rte_compiler_barrier();
1752d19533e8SHuawei Xie 
1753d19533e8SHuawei Xie 	LOG_DEBUG(VHOST_DATA,
1754d19533e8SHuawei Xie 		"(%"PRIu64") in dev_rx_zcp: before update used idx: "
1755d19533e8SHuawei Xie 		"vq.last_used_idx: %d, vq->used->idx: %d\n",
1756d19533e8SHuawei Xie 		dev->device_fh, vq->last_used_idx, vq->used->idx);
1757d19533e8SHuawei Xie 
1758d19533e8SHuawei Xie 	*(volatile uint16_t *)&vq->used->idx += count;
1759d19533e8SHuawei Xie 	vq->last_used_idx += count;
1760d19533e8SHuawei Xie 
1761d19533e8SHuawei Xie 	LOG_DEBUG(VHOST_DATA,
1762d19533e8SHuawei Xie 		"(%"PRIu64") in dev_rx_zcp: after  update used idx: "
1763d19533e8SHuawei Xie 		"vq.last_used_idx: %d, vq->used->idx: %d\n",
1764d19533e8SHuawei Xie 		dev->device_fh, vq->last_used_idx, vq->used->idx);
1765d19533e8SHuawei Xie 
1766d19533e8SHuawei Xie 	/* Kick the guest if necessary. */
1767d19533e8SHuawei Xie 	if (!(vq->avail->flags & VRING_AVAIL_F_NO_INTERRUPT))
1768d19533e8SHuawei Xie 		eventfd_write((int)vq->kickfd, 1);
1769d19533e8SHuawei Xie 
1770d19533e8SHuawei Xie 	return count;
1771d19533e8SHuawei Xie }
1772d19533e8SHuawei Xie 
1773d19533e8SHuawei Xie /*
1774d19533e8SHuawei Xie  * This function routes the TX packet to the correct interface.
1775d19533e8SHuawei Xie  * This may be a local device or the physical port.
1776d19533e8SHuawei Xie  */
1777d19533e8SHuawei Xie static inline void __attribute__((always_inline))
1778d19533e8SHuawei Xie virtio_tx_route_zcp(struct virtio_net *dev, struct rte_mbuf *m,
1779d19533e8SHuawei Xie 	uint32_t desc_idx, uint8_t need_copy)
1780d19533e8SHuawei Xie {
1781d19533e8SHuawei Xie 	struct mbuf_table *tx_q;
1782d19533e8SHuawei Xie 	struct rte_mbuf **m_table;
1783d19533e8SHuawei Xie 	struct rte_mbuf *mbuf = NULL;
1784d19533e8SHuawei Xie 	unsigned len, ret, offset = 0;
1785d19533e8SHuawei Xie 	struct vpool *vpool;
1786d19533e8SHuawei Xie 	struct virtio_net_data_ll *dev_ll = ll_root_used;
1787d19533e8SHuawei Xie 	struct ether_hdr *pkt_hdr = rte_pktmbuf_mtod(m, struct ether_hdr *);
1788d19533e8SHuawei Xie 	uint16_t vlan_tag = (uint16_t)vlan_tags[(uint16_t)dev->device_fh];
1789*e571e6b4SHuawei Xie 	uint16_t vmdq_rx_q = ((struct vhost_dev *)dev->priv)->vmdq_rx_q;
1790d19533e8SHuawei Xie 
1791d19533e8SHuawei Xie 	/*Add packet to the port tx queue*/
1792*e571e6b4SHuawei Xie 	tx_q = &tx_queue_zcp[vmdq_rx_q];
1793d19533e8SHuawei Xie 	len = tx_q->len;
1794d19533e8SHuawei Xie 
1795d19533e8SHuawei Xie 	/* Allocate an mbuf and populate the structure. */
1796*e571e6b4SHuawei Xie 	vpool = &vpool_array[MAX_QUEUES + vmdq_rx_q];
1797d19533e8SHuawei Xie 	rte_ring_sc_dequeue(vpool->ring, (void **)&mbuf);
1798d19533e8SHuawei Xie 	if (unlikely(mbuf == NULL)) {
1799d19533e8SHuawei Xie 		struct vhost_virtqueue *vq = dev->virtqueue[VIRTIO_TXQ];
1800d19533e8SHuawei Xie 		RTE_LOG(ERR, VHOST_DATA,
1801d19533e8SHuawei Xie 			"(%"PRIu64") Failed to allocate memory for mbuf.\n",
1802d19533e8SHuawei Xie 			dev->device_fh);
1803d19533e8SHuawei Xie 		put_desc_to_used_list_zcp(vq, desc_idx);
1804d19533e8SHuawei Xie 		return;
1805d19533e8SHuawei Xie 	}
1806d19533e8SHuawei Xie 
1807d19533e8SHuawei Xie 	if (vm2vm_mode == VM2VM_HARDWARE) {
1808d19533e8SHuawei Xie 		/* Avoid using a vlan tag from any vm for external pkt, such as
1809d19533e8SHuawei Xie 		 * vlan_tags[dev->device_fh], oterwise, it conflicts when pool
1810d19533e8SHuawei Xie 		 * selection, MAC address determines it as an external pkt
1811d19533e8SHuawei Xie 		 * which should go to network, while vlan tag determine it as
1812d19533e8SHuawei Xie 		 * a vm2vm pkt should forward to another vm. Hardware confuse
1813d19533e8SHuawei Xie 		 * such a ambiguous situation, so pkt will lost.
1814d19533e8SHuawei Xie 		 */
1815d19533e8SHuawei Xie 		vlan_tag = external_pkt_default_vlan_tag;
1816d19533e8SHuawei Xie 		while (dev_ll != NULL) {
1817*e571e6b4SHuawei Xie 			if (likely(dev_ll->vdev->ready == DEVICE_RX) &&
1818d19533e8SHuawei Xie 				ether_addr_cmp(&(pkt_hdr->d_addr),
1819*e571e6b4SHuawei Xie 				&dev_ll->vdev->mac_address)) {
1820d19533e8SHuawei Xie 
1821d19533e8SHuawei Xie 				/*
1822d19533e8SHuawei Xie 				 * Drop the packet if the TX packet is destined
1823d19533e8SHuawei Xie 				 * for the TX device.
1824d19533e8SHuawei Xie 				 */
1825*e571e6b4SHuawei Xie 				if (unlikely(dev_ll->vdev->dev->device_fh
1826d19533e8SHuawei Xie 					== dev->device_fh)) {
1827d19533e8SHuawei Xie 					LOG_DEBUG(VHOST_DATA,
1828d19533e8SHuawei Xie 					"(%"PRIu64") TX: Source and destination"
1829d19533e8SHuawei Xie 					"MAC addresses are the same. Dropping "
1830d19533e8SHuawei Xie 					"packet.\n",
1831*e571e6b4SHuawei Xie 					dev_ll->vdev->dev->device_fh);
1832d19533e8SHuawei Xie 					MBUF_HEADROOM_UINT32(mbuf)
1833d19533e8SHuawei Xie 						= (uint32_t)desc_idx;
1834d19533e8SHuawei Xie 					__rte_mbuf_raw_free(mbuf);
1835d19533e8SHuawei Xie 					return;
1836d19533e8SHuawei Xie 				}
1837d19533e8SHuawei Xie 
1838d19533e8SHuawei Xie 				/*
1839d19533e8SHuawei Xie 				 * Packet length offset 4 bytes for HW vlan
1840d19533e8SHuawei Xie 				 * strip when L2 switch back.
1841d19533e8SHuawei Xie 				 */
1842d19533e8SHuawei Xie 				offset = 4;
1843d19533e8SHuawei Xie 				vlan_tag =
1844d19533e8SHuawei Xie 				(uint16_t)
1845*e571e6b4SHuawei Xie 				vlan_tags[(uint16_t)dev_ll->vdev->dev->device_fh];
1846d19533e8SHuawei Xie 
1847d19533e8SHuawei Xie 				LOG_DEBUG(VHOST_DATA,
1848d19533e8SHuawei Xie 				"(%"PRIu64") TX: pkt to local VM device id:"
1849d19533e8SHuawei Xie 				"(%"PRIu64") vlan tag: %d.\n",
1850*e571e6b4SHuawei Xie 				dev->device_fh, dev_ll->vdev->dev->device_fh,
1851d19533e8SHuawei Xie 				vlan_tag);
1852d19533e8SHuawei Xie 
1853d19533e8SHuawei Xie 				break;
1854d19533e8SHuawei Xie 			}
1855d19533e8SHuawei Xie 			dev_ll = dev_ll->next;
1856d19533e8SHuawei Xie 		}
1857d19533e8SHuawei Xie 	}
1858d19533e8SHuawei Xie 
1859d19533e8SHuawei Xie 	mbuf->nb_segs = m->nb_segs;
1860d19533e8SHuawei Xie 	mbuf->next = m->next;
1861d19533e8SHuawei Xie 	mbuf->data_len = m->data_len + offset;
1862d19533e8SHuawei Xie 	mbuf->pkt_len = mbuf->data_len;
1863d19533e8SHuawei Xie 	if (unlikely(need_copy)) {
1864d19533e8SHuawei Xie 		/* Copy the packet contents to the mbuf. */
1865d19533e8SHuawei Xie 		rte_memcpy(rte_pktmbuf_mtod(mbuf, void *),
1866d19533e8SHuawei Xie 			rte_pktmbuf_mtod(m, void *),
1867d19533e8SHuawei Xie 			m->data_len);
1868d19533e8SHuawei Xie 	} else {
1869d19533e8SHuawei Xie 		mbuf->data_off = m->data_off;
1870d19533e8SHuawei Xie 		mbuf->buf_physaddr = m->buf_physaddr;
1871d19533e8SHuawei Xie 		mbuf->buf_addr = m->buf_addr;
1872d19533e8SHuawei Xie 	}
1873d19533e8SHuawei Xie 	mbuf->ol_flags = PKT_TX_VLAN_PKT;
1874d19533e8SHuawei Xie 	mbuf->vlan_tci = vlan_tag;
1875d19533e8SHuawei Xie 	mbuf->l2_len = sizeof(struct ether_hdr);
1876d19533e8SHuawei Xie 	mbuf->l3_len = sizeof(struct ipv4_hdr);
1877d19533e8SHuawei Xie 	MBUF_HEADROOM_UINT32(mbuf) = (uint32_t)desc_idx;
1878d19533e8SHuawei Xie 
1879d19533e8SHuawei Xie 	tx_q->m_table[len] = mbuf;
1880d19533e8SHuawei Xie 	len++;
1881d19533e8SHuawei Xie 
1882d19533e8SHuawei Xie 	LOG_DEBUG(VHOST_DATA,
1883d19533e8SHuawei Xie 		"(%"PRIu64") in tx_route_zcp: pkt: nb_seg: %d, next:%s\n",
1884d19533e8SHuawei Xie 		dev->device_fh,
1885d19533e8SHuawei Xie 		mbuf->nb_segs,
1886d19533e8SHuawei Xie 		(mbuf->next == NULL) ? "null" : "non-null");
1887d19533e8SHuawei Xie 
1888d19533e8SHuawei Xie 	if (enable_stats) {
1889d19533e8SHuawei Xie 		dev_statistics[dev->device_fh].tx_total++;
1890d19533e8SHuawei Xie 		dev_statistics[dev->device_fh].tx++;
1891d19533e8SHuawei Xie 	}
1892d19533e8SHuawei Xie 
1893d19533e8SHuawei Xie 	if (unlikely(len == MAX_PKT_BURST)) {
1894d19533e8SHuawei Xie 		m_table = (struct rte_mbuf **)tx_q->m_table;
1895d19533e8SHuawei Xie 		ret = rte_eth_tx_burst(ports[0],
1896d19533e8SHuawei Xie 			(uint16_t)tx_q->txq_id, m_table, (uint16_t) len);
1897d19533e8SHuawei Xie 
1898d19533e8SHuawei Xie 		/*
1899d19533e8SHuawei Xie 		 * Free any buffers not handled by TX and update
1900d19533e8SHuawei Xie 		 * the port stats.
1901d19533e8SHuawei Xie 		 */
1902d19533e8SHuawei Xie 		if (unlikely(ret < len)) {
1903d19533e8SHuawei Xie 			do {
1904d19533e8SHuawei Xie 				rte_pktmbuf_free(m_table[ret]);
1905d19533e8SHuawei Xie 			} while (++ret < len);
1906d19533e8SHuawei Xie 		}
1907d19533e8SHuawei Xie 
1908d19533e8SHuawei Xie 		len = 0;
1909d19533e8SHuawei Xie 		txmbuf_clean_zcp(dev, vpool);
1910d19533e8SHuawei Xie 	}
1911d19533e8SHuawei Xie 
1912d19533e8SHuawei Xie 	tx_q->len = len;
1913d19533e8SHuawei Xie 
1914d19533e8SHuawei Xie 	return;
1915d19533e8SHuawei Xie }
1916d19533e8SHuawei Xie 
1917d19533e8SHuawei Xie /*
1918d19533e8SHuawei Xie  * This function TX all available packets in virtio TX queue for one
1919d19533e8SHuawei Xie  * virtio-net device. If it is first packet, it learns MAC address and
1920d19533e8SHuawei Xie  * setup VMDQ.
1921d19533e8SHuawei Xie  */
1922d19533e8SHuawei Xie static inline void __attribute__((always_inline))
1923d19533e8SHuawei Xie virtio_dev_tx_zcp(struct virtio_net *dev)
1924d19533e8SHuawei Xie {
1925d19533e8SHuawei Xie 	struct rte_mbuf m;
1926d19533e8SHuawei Xie 	struct vhost_virtqueue *vq;
1927d19533e8SHuawei Xie 	struct vring_desc *desc;
1928d19533e8SHuawei Xie 	uint64_t buff_addr = 0, phys_addr;
1929d19533e8SHuawei Xie 	uint32_t head[MAX_PKT_BURST];
1930d19533e8SHuawei Xie 	uint32_t i;
1931d19533e8SHuawei Xie 	uint16_t free_entries, packet_success = 0;
1932d19533e8SHuawei Xie 	uint16_t avail_idx;
1933d19533e8SHuawei Xie 	uint8_t need_copy = 0;
1934d19533e8SHuawei Xie 	hpa_type addr_type;
1935*e571e6b4SHuawei Xie 	struct vhost_dev *vdev = (struct vhost_dev *)dev->priv;
1936d19533e8SHuawei Xie 
1937d19533e8SHuawei Xie 	vq = dev->virtqueue[VIRTIO_TXQ];
1938d19533e8SHuawei Xie 	avail_idx =  *((volatile uint16_t *)&vq->avail->idx);
1939d19533e8SHuawei Xie 
1940d19533e8SHuawei Xie 	/* If there are no available buffers then return. */
1941d19533e8SHuawei Xie 	if (vq->last_used_idx_res == avail_idx)
1942d19533e8SHuawei Xie 		return;
1943d19533e8SHuawei Xie 
1944d19533e8SHuawei Xie 	LOG_DEBUG(VHOST_DATA, "(%"PRIu64") virtio_dev_tx()\n", dev->device_fh);
1945d19533e8SHuawei Xie 
1946d19533e8SHuawei Xie 	/* Prefetch available ring to retrieve head indexes. */
1947d19533e8SHuawei Xie 	rte_prefetch0(&vq->avail->ring[vq->last_used_idx_res & (vq->size - 1)]);
1948d19533e8SHuawei Xie 
1949d19533e8SHuawei Xie 	/* Get the number of free entries in the ring */
1950d19533e8SHuawei Xie 	free_entries = (avail_idx - vq->last_used_idx_res);
1951d19533e8SHuawei Xie 
1952d19533e8SHuawei Xie 	/* Limit to MAX_PKT_BURST. */
1953d19533e8SHuawei Xie 	free_entries
1954d19533e8SHuawei Xie 		= (free_entries > MAX_PKT_BURST) ? MAX_PKT_BURST : free_entries;
1955d19533e8SHuawei Xie 
1956d19533e8SHuawei Xie 	LOG_DEBUG(VHOST_DATA, "(%"PRIu64") Buffers available %d\n",
1957d19533e8SHuawei Xie 		dev->device_fh, free_entries);
1958d19533e8SHuawei Xie 
1959d19533e8SHuawei Xie 	/* Retrieve all of the head indexes first to avoid caching issues. */
1960d19533e8SHuawei Xie 	for (i = 0; i < free_entries; i++)
1961d19533e8SHuawei Xie 		head[i]
1962d19533e8SHuawei Xie 			= vq->avail->ring[(vq->last_used_idx_res + i)
1963d19533e8SHuawei Xie 			& (vq->size - 1)];
1964d19533e8SHuawei Xie 
1965d19533e8SHuawei Xie 	vq->last_used_idx_res += free_entries;
1966d19533e8SHuawei Xie 
1967d19533e8SHuawei Xie 	/* Prefetch descriptor index. */
1968d19533e8SHuawei Xie 	rte_prefetch0(&vq->desc[head[packet_success]]);
1969d19533e8SHuawei Xie 	rte_prefetch0(&vq->used->ring[vq->last_used_idx & (vq->size - 1)]);
1970d19533e8SHuawei Xie 
1971d19533e8SHuawei Xie 	while (packet_success < free_entries) {
1972d19533e8SHuawei Xie 		desc = &vq->desc[head[packet_success]];
1973d19533e8SHuawei Xie 
1974d19533e8SHuawei Xie 		/* Discard first buffer as it is the virtio header */
1975d19533e8SHuawei Xie 		desc = &vq->desc[desc->next];
1976d19533e8SHuawei Xie 
1977d19533e8SHuawei Xie 		/* Buffer address translation. */
1978d19533e8SHuawei Xie 		buff_addr = gpa_to_vva(dev, desc->addr);
1979*e571e6b4SHuawei Xie 		phys_addr = gpa_to_hpa(vdev, desc->addr, desc->len, &addr_type);
1980d19533e8SHuawei Xie 
1981d19533e8SHuawei Xie 		if (likely(packet_success < (free_entries - 1)))
1982d19533e8SHuawei Xie 			/* Prefetch descriptor index. */
1983d19533e8SHuawei Xie 			rte_prefetch0(&vq->desc[head[packet_success + 1]]);
1984d19533e8SHuawei Xie 
1985d19533e8SHuawei Xie 		if (unlikely(addr_type == PHYS_ADDR_INVALID)) {
1986d19533e8SHuawei Xie 			RTE_LOG(ERR, VHOST_DATA,
1987d19533e8SHuawei Xie 				"(%"PRIu64") Invalid frame buffer address found"
1988d19533e8SHuawei Xie 				"when TX packets!\n",
1989d19533e8SHuawei Xie 				dev->device_fh);
1990d19533e8SHuawei Xie 			packet_success++;
1991d19533e8SHuawei Xie 			continue;
1992d19533e8SHuawei Xie 		}
1993d19533e8SHuawei Xie 
1994d19533e8SHuawei Xie 		/* Prefetch buffer address. */
1995d19533e8SHuawei Xie 		rte_prefetch0((void *)(uintptr_t)buff_addr);
1996d19533e8SHuawei Xie 
1997d19533e8SHuawei Xie 		/*
1998d19533e8SHuawei Xie 		 * Setup dummy mbuf. This is copied to a real mbuf if
1999d19533e8SHuawei Xie 		 * transmitted out the physical port.
2000d19533e8SHuawei Xie 		 */
2001d19533e8SHuawei Xie 		m.data_len = desc->len;
2002d19533e8SHuawei Xie 		m.nb_segs = 1;
2003d19533e8SHuawei Xie 		m.next = NULL;
2004d19533e8SHuawei Xie 		m.data_off = 0;
2005d19533e8SHuawei Xie 		m.buf_addr = (void *)(uintptr_t)buff_addr;
2006d19533e8SHuawei Xie 		m.buf_physaddr = phys_addr;
2007d19533e8SHuawei Xie 
2008d19533e8SHuawei Xie 		/*
2009d19533e8SHuawei Xie 		 * Check if the frame buffer address from guest crosses
2010d19533e8SHuawei Xie 		 * sub-region or not.
2011d19533e8SHuawei Xie 		 */
2012d19533e8SHuawei Xie 		if (unlikely(addr_type == PHYS_ADDR_CROSS_SUBREG)) {
2013d19533e8SHuawei Xie 			RTE_LOG(ERR, VHOST_DATA,
2014d19533e8SHuawei Xie 				"(%"PRIu64") Frame buffer address cross "
2015d19533e8SHuawei Xie 				"sub-regioin found when attaching TX frame "
2016d19533e8SHuawei Xie 				"buffer address!\n",
2017d19533e8SHuawei Xie 				dev->device_fh);
2018d19533e8SHuawei Xie 			need_copy = 1;
2019d19533e8SHuawei Xie 		} else
2020d19533e8SHuawei Xie 			need_copy = 0;
2021d19533e8SHuawei Xie 
2022d19533e8SHuawei Xie 		PRINT_PACKET(dev, (uintptr_t)buff_addr, desc->len, 0);
2023d19533e8SHuawei Xie 
2024d19533e8SHuawei Xie 		/*
2025d19533e8SHuawei Xie 		 * If this is the first received packet we need to learn
2026d19533e8SHuawei Xie 		 * the MAC and setup VMDQ
2027d19533e8SHuawei Xie 		 */
2028*e571e6b4SHuawei Xie 		if (unlikely(vdev->ready == DEVICE_MAC_LEARNING)) {
2029*e571e6b4SHuawei Xie 			if (vdev->remove || (link_vmdq(vdev, &m) == -1)) {
2030d19533e8SHuawei Xie 				/*
2031d19533e8SHuawei Xie 				 * Discard frame if device is scheduled for
2032d19533e8SHuawei Xie 				 * removal or a duplicate MAC address is found.
2033d19533e8SHuawei Xie 				 */
2034d19533e8SHuawei Xie 				packet_success += free_entries;
2035d19533e8SHuawei Xie 				vq->last_used_idx += packet_success;
2036d19533e8SHuawei Xie 				break;
2037d19533e8SHuawei Xie 			}
2038d19533e8SHuawei Xie 		}
2039d19533e8SHuawei Xie 
2040d19533e8SHuawei Xie 		virtio_tx_route_zcp(dev, &m, head[packet_success], need_copy);
2041d19533e8SHuawei Xie 		packet_success++;
2042d19533e8SHuawei Xie 	}
2043d19533e8SHuawei Xie }
2044d19533e8SHuawei Xie 
2045d19533e8SHuawei Xie /*
2046d19533e8SHuawei Xie  * This function is called by each data core. It handles all RX/TX registered
2047d19533e8SHuawei Xie  * with the core. For TX the specific lcore linked list is used. For RX, MAC
2048d19533e8SHuawei Xie  * addresses are compared with all devices in the main linked list.
2049d19533e8SHuawei Xie  */
2050d19533e8SHuawei Xie static int
2051d19533e8SHuawei Xie switch_worker_zcp(__attribute__((unused)) void *arg)
2052d19533e8SHuawei Xie {
2053d19533e8SHuawei Xie 	struct virtio_net *dev = NULL;
2054*e571e6b4SHuawei Xie 	struct vhost_dev  *vdev = NULL;
2055d19533e8SHuawei Xie 	struct rte_mbuf *pkts_burst[MAX_PKT_BURST];
2056d19533e8SHuawei Xie 	struct virtio_net_data_ll *dev_ll;
2057d19533e8SHuawei Xie 	struct mbuf_table *tx_q;
2058d19533e8SHuawei Xie 	volatile struct lcore_ll_info *lcore_ll;
2059d19533e8SHuawei Xie 	const uint64_t drain_tsc
2060d19533e8SHuawei Xie 		= (rte_get_tsc_hz() + US_PER_S - 1) / US_PER_S
2061d19533e8SHuawei Xie 		* BURST_TX_DRAIN_US;
2062d19533e8SHuawei Xie 	uint64_t prev_tsc, diff_tsc, cur_tsc, ret_count = 0;
2063d19533e8SHuawei Xie 	unsigned ret;
2064d19533e8SHuawei Xie 	const uint16_t lcore_id = rte_lcore_id();
2065d19533e8SHuawei Xie 	uint16_t count_in_ring, rx_count = 0;
2066d19533e8SHuawei Xie 
2067d19533e8SHuawei Xie 	RTE_LOG(INFO, VHOST_DATA, "Procesing on Core %u started\n", lcore_id);
2068d19533e8SHuawei Xie 
2069d19533e8SHuawei Xie 	lcore_ll = lcore_info[lcore_id].lcore_ll;
2070d19533e8SHuawei Xie 	prev_tsc = 0;
2071d19533e8SHuawei Xie 
2072d19533e8SHuawei Xie 	while (1) {
2073d19533e8SHuawei Xie 		cur_tsc = rte_rdtsc();
2074d19533e8SHuawei Xie 
2075d19533e8SHuawei Xie 		/* TX burst queue drain */
2076d19533e8SHuawei Xie 		diff_tsc = cur_tsc - prev_tsc;
2077d19533e8SHuawei Xie 		if (unlikely(diff_tsc > drain_tsc)) {
2078d19533e8SHuawei Xie 			/*
2079d19533e8SHuawei Xie 			 * Get mbuf from vpool.pool and detach mbuf and
2080d19533e8SHuawei Xie 			 * put back into vpool.ring.
2081d19533e8SHuawei Xie 			 */
2082d19533e8SHuawei Xie 			dev_ll = lcore_ll->ll_root_used;
2083*e571e6b4SHuawei Xie 			while ((dev_ll != NULL) && (dev_ll->vdev != NULL)) {
2084d19533e8SHuawei Xie 				/* Get virtio device ID */
2085*e571e6b4SHuawei Xie 				vdev = dev_ll->vdev;
2086*e571e6b4SHuawei Xie 				dev = vdev->dev;
2087d19533e8SHuawei Xie 
2088*e571e6b4SHuawei Xie 				if (likely(!vdev->remove)) {
2089*e571e6b4SHuawei Xie 					tx_q = &tx_queue_zcp[(uint16_t)vdev->vmdq_rx_q];
2090d19533e8SHuawei Xie 					if (tx_q->len) {
2091d19533e8SHuawei Xie 						LOG_DEBUG(VHOST_DATA,
2092d19533e8SHuawei Xie 						"TX queue drained after timeout"
2093d19533e8SHuawei Xie 						" with burst size %u\n",
2094d19533e8SHuawei Xie 						tx_q->len);
2095d19533e8SHuawei Xie 
2096d19533e8SHuawei Xie 						/*
2097d19533e8SHuawei Xie 						 * Tx any packets in the queue
2098d19533e8SHuawei Xie 						 */
2099d19533e8SHuawei Xie 						ret = rte_eth_tx_burst(
2100d19533e8SHuawei Xie 							ports[0],
2101d19533e8SHuawei Xie 							(uint16_t)tx_q->txq_id,
2102d19533e8SHuawei Xie 							(struct rte_mbuf **)
2103d19533e8SHuawei Xie 							tx_q->m_table,
2104d19533e8SHuawei Xie 							(uint16_t)tx_q->len);
2105d19533e8SHuawei Xie 						if (unlikely(ret < tx_q->len)) {
2106d19533e8SHuawei Xie 							do {
2107d19533e8SHuawei Xie 								rte_pktmbuf_free(
2108d19533e8SHuawei Xie 									tx_q->m_table[ret]);
2109d19533e8SHuawei Xie 							} while (++ret < tx_q->len);
2110d19533e8SHuawei Xie 						}
2111d19533e8SHuawei Xie 						tx_q->len = 0;
2112d19533e8SHuawei Xie 
2113d19533e8SHuawei Xie 						txmbuf_clean_zcp(dev,
2114*e571e6b4SHuawei Xie 							&vpool_array[MAX_QUEUES+vdev->vmdq_rx_q]);
2115d19533e8SHuawei Xie 					}
2116d19533e8SHuawei Xie 				}
2117d19533e8SHuawei Xie 				dev_ll = dev_ll->next;
2118d19533e8SHuawei Xie 			}
2119d19533e8SHuawei Xie 			prev_tsc = cur_tsc;
2120d19533e8SHuawei Xie 		}
2121d19533e8SHuawei Xie 
2122d19533e8SHuawei Xie 		rte_prefetch0(lcore_ll->ll_root_used);
2123d19533e8SHuawei Xie 
2124d19533e8SHuawei Xie 		/*
2125d19533e8SHuawei Xie 		 * Inform the configuration core that we have exited the linked
2126d19533e8SHuawei Xie 		 * list and that no devices are in use if requested.
2127d19533e8SHuawei Xie 		 */
2128d19533e8SHuawei Xie 		if (lcore_ll->dev_removal_flag == REQUEST_DEV_REMOVAL)
2129d19533e8SHuawei Xie 			lcore_ll->dev_removal_flag = ACK_DEV_REMOVAL;
2130d19533e8SHuawei Xie 
2131d19533e8SHuawei Xie 		/* Process devices */
2132d19533e8SHuawei Xie 		dev_ll = lcore_ll->ll_root_used;
2133d19533e8SHuawei Xie 
2134*e571e6b4SHuawei Xie 		while ((dev_ll != NULL) && (dev_ll->vdev != NULL)) {
2135*e571e6b4SHuawei Xie 			vdev = dev_ll->vdev;
2136*e571e6b4SHuawei Xie 			dev  = vdev->dev;
2137*e571e6b4SHuawei Xie 			if (unlikely(vdev->remove)) {
2138d19533e8SHuawei Xie 				dev_ll = dev_ll->next;
2139*e571e6b4SHuawei Xie 				unlink_vmdq(vdev);
2140*e571e6b4SHuawei Xie 				vdev->ready = DEVICE_SAFE_REMOVE;
2141d19533e8SHuawei Xie 				continue;
2142d19533e8SHuawei Xie 			}
2143d19533e8SHuawei Xie 
2144*e571e6b4SHuawei Xie 			if (likely(vdev->ready == DEVICE_RX)) {
2145*e571e6b4SHuawei Xie 				uint32_t index = vdev->vmdq_rx_q;
2146d19533e8SHuawei Xie 				uint16_t i;
2147d19533e8SHuawei Xie 				count_in_ring
2148d19533e8SHuawei Xie 				= rte_ring_count(vpool_array[index].ring);
2149d19533e8SHuawei Xie 				uint16_t free_entries
2150d19533e8SHuawei Xie 				= (uint16_t)get_available_ring_num_zcp(dev);
2151d19533e8SHuawei Xie 
2152d19533e8SHuawei Xie 				/*
2153d19533e8SHuawei Xie 				 * Attach all mbufs in vpool.ring and put back
2154d19533e8SHuawei Xie 				 * into vpool.pool.
2155d19533e8SHuawei Xie 				 */
2156d19533e8SHuawei Xie 				for (i = 0;
2157d19533e8SHuawei Xie 				i < RTE_MIN(free_entries,
2158d19533e8SHuawei Xie 				RTE_MIN(count_in_ring, MAX_PKT_BURST));
2159d19533e8SHuawei Xie 				i++)
2160d19533e8SHuawei Xie 					attach_rxmbuf_zcp(dev);
2161d19533e8SHuawei Xie 
2162d19533e8SHuawei Xie 				/* Handle guest RX */
2163d19533e8SHuawei Xie 				rx_count = rte_eth_rx_burst(ports[0],
2164*e571e6b4SHuawei Xie 					vdev->vmdq_rx_q, pkts_burst,
2165d19533e8SHuawei Xie 					MAX_PKT_BURST);
2166d19533e8SHuawei Xie 
2167d19533e8SHuawei Xie 				if (rx_count) {
2168d19533e8SHuawei Xie 					ret_count = virtio_dev_rx_zcp(dev,
2169d19533e8SHuawei Xie 							pkts_burst, rx_count);
2170d19533e8SHuawei Xie 					if (enable_stats) {
2171d19533e8SHuawei Xie 						dev_statistics[dev->device_fh].rx_total
2172d19533e8SHuawei Xie 							+= rx_count;
2173d19533e8SHuawei Xie 						dev_statistics[dev->device_fh].rx
2174d19533e8SHuawei Xie 							+= ret_count;
2175d19533e8SHuawei Xie 					}
2176d19533e8SHuawei Xie 					while (likely(rx_count)) {
2177d19533e8SHuawei Xie 						rx_count--;
2178d19533e8SHuawei Xie 						pktmbuf_detach_zcp(
2179d19533e8SHuawei Xie 							pkts_burst[rx_count]);
2180d19533e8SHuawei Xie 						rte_ring_sp_enqueue(
2181d19533e8SHuawei Xie 							vpool_array[index].ring,
2182d19533e8SHuawei Xie 							(void *)pkts_burst[rx_count]);
2183d19533e8SHuawei Xie 					}
2184d19533e8SHuawei Xie 				}
2185d19533e8SHuawei Xie 			}
2186d19533e8SHuawei Xie 
2187*e571e6b4SHuawei Xie 			if (likely(!vdev->remove))
2188d19533e8SHuawei Xie 				/* Handle guest TX */
2189d19533e8SHuawei Xie 				virtio_dev_tx_zcp(dev);
2190d19533e8SHuawei Xie 
2191d19533e8SHuawei Xie 			/* Move to the next device in the list */
2192d19533e8SHuawei Xie 			dev_ll = dev_ll->next;
2193d19533e8SHuawei Xie 		}
2194d19533e8SHuawei Xie 	}
2195d19533e8SHuawei Xie 
2196d19533e8SHuawei Xie 	return 0;
2197d19533e8SHuawei Xie }
2198d19533e8SHuawei Xie 
2199d19533e8SHuawei Xie 
2200d19533e8SHuawei Xie /*
2201d19533e8SHuawei Xie  * Add an entry to a used linked list. A free entry must first be found
2202d19533e8SHuawei Xie  * in the free linked list using get_data_ll_free_entry();
2203d19533e8SHuawei Xie  */
2204d19533e8SHuawei Xie static void
2205d19533e8SHuawei Xie add_data_ll_entry(struct virtio_net_data_ll **ll_root_addr,
2206d19533e8SHuawei Xie 	struct virtio_net_data_ll *ll_dev)
2207d19533e8SHuawei Xie {
2208d19533e8SHuawei Xie 	struct virtio_net_data_ll *ll = *ll_root_addr;
2209d19533e8SHuawei Xie 
2210d19533e8SHuawei Xie 	/* Set next as NULL and use a compiler barrier to avoid reordering. */
2211d19533e8SHuawei Xie 	ll_dev->next = NULL;
2212d19533e8SHuawei Xie 	rte_compiler_barrier();
2213d19533e8SHuawei Xie 
2214d19533e8SHuawei Xie 	/* If ll == NULL then this is the first device. */
2215d19533e8SHuawei Xie 	if (ll) {
2216d19533e8SHuawei Xie 		/* Increment to the tail of the linked list. */
2217d19533e8SHuawei Xie 		while ((ll->next != NULL) )
2218d19533e8SHuawei Xie 			ll = ll->next;
2219d19533e8SHuawei Xie 
2220d19533e8SHuawei Xie 		ll->next = ll_dev;
2221d19533e8SHuawei Xie 	} else {
2222d19533e8SHuawei Xie 		*ll_root_addr = ll_dev;
2223d19533e8SHuawei Xie 	}
2224d19533e8SHuawei Xie }
2225d19533e8SHuawei Xie 
2226d19533e8SHuawei Xie /*
2227d19533e8SHuawei Xie  * Remove an entry from a used linked list. The entry must then be added to
2228d19533e8SHuawei Xie  * the free linked list using put_data_ll_free_entry().
2229d19533e8SHuawei Xie  */
2230d19533e8SHuawei Xie static void
2231d19533e8SHuawei Xie rm_data_ll_entry(struct virtio_net_data_ll **ll_root_addr,
2232d19533e8SHuawei Xie 	struct virtio_net_data_ll *ll_dev,
2233d19533e8SHuawei Xie 	struct virtio_net_data_ll *ll_dev_last)
2234d19533e8SHuawei Xie {
2235d19533e8SHuawei Xie 	struct virtio_net_data_ll *ll = *ll_root_addr;
2236d19533e8SHuawei Xie 
2237d19533e8SHuawei Xie 	if (unlikely((ll == NULL) || (ll_dev == NULL)))
2238d19533e8SHuawei Xie 		return;
2239d19533e8SHuawei Xie 
2240d19533e8SHuawei Xie 	if (ll_dev == ll)
2241d19533e8SHuawei Xie 		*ll_root_addr = ll_dev->next;
2242d19533e8SHuawei Xie 	else
2243d19533e8SHuawei Xie 		if (likely(ll_dev_last != NULL))
2244d19533e8SHuawei Xie 			ll_dev_last->next = ll_dev->next;
2245d19533e8SHuawei Xie 		else
2246d19533e8SHuawei Xie 			RTE_LOG(ERR, VHOST_CONFIG, "Remove entry form ll failed.\n");
2247d19533e8SHuawei Xie }
2248d19533e8SHuawei Xie 
2249d19533e8SHuawei Xie /*
2250d19533e8SHuawei Xie  * Find and return an entry from the free linked list.
2251d19533e8SHuawei Xie  */
2252d19533e8SHuawei Xie static struct virtio_net_data_ll *
2253d19533e8SHuawei Xie get_data_ll_free_entry(struct virtio_net_data_ll **ll_root_addr)
2254d19533e8SHuawei Xie {
2255d19533e8SHuawei Xie 	struct virtio_net_data_ll *ll_free = *ll_root_addr;
2256d19533e8SHuawei Xie 	struct virtio_net_data_ll *ll_dev;
2257d19533e8SHuawei Xie 
2258d19533e8SHuawei Xie 	if (ll_free == NULL)
2259d19533e8SHuawei Xie 		return NULL;
2260d19533e8SHuawei Xie 
2261d19533e8SHuawei Xie 	ll_dev = ll_free;
2262d19533e8SHuawei Xie 	*ll_root_addr = ll_free->next;
2263d19533e8SHuawei Xie 
2264d19533e8SHuawei Xie 	return ll_dev;
2265d19533e8SHuawei Xie }
2266d19533e8SHuawei Xie 
2267d19533e8SHuawei Xie /*
2268d19533e8SHuawei Xie  * Place an entry back on to the free linked list.
2269d19533e8SHuawei Xie  */
2270d19533e8SHuawei Xie static void
2271d19533e8SHuawei Xie put_data_ll_free_entry(struct virtio_net_data_ll **ll_root_addr,
2272d19533e8SHuawei Xie 	struct virtio_net_data_ll *ll_dev)
2273d19533e8SHuawei Xie {
2274d19533e8SHuawei Xie 	struct virtio_net_data_ll *ll_free = *ll_root_addr;
2275d19533e8SHuawei Xie 
2276d19533e8SHuawei Xie 	if (ll_dev == NULL)
2277d19533e8SHuawei Xie 		return;
2278d19533e8SHuawei Xie 
2279d19533e8SHuawei Xie 	ll_dev->next = ll_free;
2280d19533e8SHuawei Xie 	*ll_root_addr = ll_dev;
2281d19533e8SHuawei Xie }
2282d19533e8SHuawei Xie 
2283d19533e8SHuawei Xie /*
2284d19533e8SHuawei Xie  * Creates a linked list of a given size.
2285d19533e8SHuawei Xie  */
2286d19533e8SHuawei Xie static struct virtio_net_data_ll *
2287d19533e8SHuawei Xie alloc_data_ll(uint32_t size)
2288d19533e8SHuawei Xie {
2289d19533e8SHuawei Xie 	struct virtio_net_data_ll *ll_new;
2290d19533e8SHuawei Xie 	uint32_t i;
2291d19533e8SHuawei Xie 
2292d19533e8SHuawei Xie 	/* Malloc and then chain the linked list. */
2293d19533e8SHuawei Xie 	ll_new = malloc(size * sizeof(struct virtio_net_data_ll));
2294d19533e8SHuawei Xie 	if (ll_new == NULL) {
2295d19533e8SHuawei Xie 		RTE_LOG(ERR, VHOST_CONFIG, "Failed to allocate memory for ll_new.\n");
2296d19533e8SHuawei Xie 		return NULL;
2297d19533e8SHuawei Xie 	}
2298d19533e8SHuawei Xie 
2299d19533e8SHuawei Xie 	for (i = 0; i < size - 1; i++) {
2300*e571e6b4SHuawei Xie 		ll_new[i].vdev = NULL;
2301d19533e8SHuawei Xie 		ll_new[i].next = &ll_new[i+1];
2302d19533e8SHuawei Xie 	}
2303d19533e8SHuawei Xie 	ll_new[i].next = NULL;
2304d19533e8SHuawei Xie 
2305d19533e8SHuawei Xie 	return (ll_new);
2306d19533e8SHuawei Xie }
2307d19533e8SHuawei Xie 
2308d19533e8SHuawei Xie /*
2309d19533e8SHuawei Xie  * Create the main linked list along with each individual cores linked list. A used and a free list
2310d19533e8SHuawei Xie  * are created to manage entries.
2311d19533e8SHuawei Xie  */
2312d19533e8SHuawei Xie static int
2313d19533e8SHuawei Xie init_data_ll (void)
2314d19533e8SHuawei Xie {
2315d19533e8SHuawei Xie 	int lcore;
2316d19533e8SHuawei Xie 
2317d19533e8SHuawei Xie 	RTE_LCORE_FOREACH_SLAVE(lcore) {
2318d19533e8SHuawei Xie 		lcore_info[lcore].lcore_ll = malloc(sizeof(struct lcore_ll_info));
2319d19533e8SHuawei Xie 		if (lcore_info[lcore].lcore_ll == NULL) {
2320d19533e8SHuawei Xie 			RTE_LOG(ERR, VHOST_CONFIG, "Failed to allocate memory for lcore_ll.\n");
2321d19533e8SHuawei Xie 			return -1;
2322d19533e8SHuawei Xie 		}
2323d19533e8SHuawei Xie 
2324d19533e8SHuawei Xie 		lcore_info[lcore].lcore_ll->device_num = 0;
2325d19533e8SHuawei Xie 		lcore_info[lcore].lcore_ll->dev_removal_flag = ACK_DEV_REMOVAL;
2326d19533e8SHuawei Xie 		lcore_info[lcore].lcore_ll->ll_root_used = NULL;
2327d19533e8SHuawei Xie 		if (num_devices % num_switching_cores)
2328d19533e8SHuawei Xie 			lcore_info[lcore].lcore_ll->ll_root_free = alloc_data_ll((num_devices / num_switching_cores) + 1);
2329d19533e8SHuawei Xie 		else
2330d19533e8SHuawei Xie 			lcore_info[lcore].lcore_ll->ll_root_free = alloc_data_ll(num_devices / num_switching_cores);
2331d19533e8SHuawei Xie 	}
2332d19533e8SHuawei Xie 
2333d19533e8SHuawei Xie 	/* Allocate devices up to a maximum of MAX_DEVICES. */
2334d19533e8SHuawei Xie 	ll_root_free = alloc_data_ll(MIN((num_devices), MAX_DEVICES));
2335d19533e8SHuawei Xie 
2336d19533e8SHuawei Xie 	return 0;
2337d19533e8SHuawei Xie }
2338d19533e8SHuawei Xie 
2339d19533e8SHuawei Xie /*
2340d19533e8SHuawei Xie  * Set virtqueue flags so that we do not receive interrupts.
2341d19533e8SHuawei Xie  */
2342d19533e8SHuawei Xie static void
2343d19533e8SHuawei Xie set_irq_status (struct virtio_net *dev)
2344d19533e8SHuawei Xie {
2345d19533e8SHuawei Xie 	dev->virtqueue[VIRTIO_RXQ]->used->flags = VRING_USED_F_NO_NOTIFY;
2346d19533e8SHuawei Xie 	dev->virtqueue[VIRTIO_TXQ]->used->flags = VRING_USED_F_NO_NOTIFY;
2347d19533e8SHuawei Xie }
2348d19533e8SHuawei Xie 
2349d19533e8SHuawei Xie /*
2350d19533e8SHuawei Xie  * Remove a device from the specific data core linked list and from the main linked list. Synchonization
2351d19533e8SHuawei Xie  * occurs through the use of the lcore dev_removal_flag. Device is made volatile here to avoid re-ordering
2352d19533e8SHuawei Xie  * of dev->remove=1 which can cause an infinite loop in the rte_pause loop.
2353d19533e8SHuawei Xie  */
2354d19533e8SHuawei Xie static void
2355d19533e8SHuawei Xie destroy_device (volatile struct virtio_net *dev)
2356d19533e8SHuawei Xie {
2357d19533e8SHuawei Xie 	struct virtio_net_data_ll *ll_lcore_dev_cur;
2358d19533e8SHuawei Xie 	struct virtio_net_data_ll *ll_main_dev_cur;
2359d19533e8SHuawei Xie 	struct virtio_net_data_ll *ll_lcore_dev_last = NULL;
2360d19533e8SHuawei Xie 	struct virtio_net_data_ll *ll_main_dev_last = NULL;
2361*e571e6b4SHuawei Xie 	struct vhost_dev *vdev;
2362d19533e8SHuawei Xie 	int lcore;
2363d19533e8SHuawei Xie 
2364d19533e8SHuawei Xie 	dev->flags &= ~VIRTIO_DEV_RUNNING;
2365d19533e8SHuawei Xie 
2366*e571e6b4SHuawei Xie 	vdev = (struct vhost_dev *)dev->priv;
2367d19533e8SHuawei Xie 	/*set the remove flag. */
2368*e571e6b4SHuawei Xie 	vdev->remove = 1;
2369*e571e6b4SHuawei Xie 	while(vdev->ready != DEVICE_SAFE_REMOVE) {
2370d19533e8SHuawei Xie 		rte_pause();
2371d19533e8SHuawei Xie 	}
2372d19533e8SHuawei Xie 
2373d19533e8SHuawei Xie 	/* Search for entry to be removed from lcore ll */
2374*e571e6b4SHuawei Xie 	ll_lcore_dev_cur = lcore_info[vdev->coreid].lcore_ll->ll_root_used;
2375d19533e8SHuawei Xie 	while (ll_lcore_dev_cur != NULL) {
2376*e571e6b4SHuawei Xie 		if (ll_lcore_dev_cur->vdev == vdev) {
2377d19533e8SHuawei Xie 			break;
2378d19533e8SHuawei Xie 		} else {
2379d19533e8SHuawei Xie 			ll_lcore_dev_last = ll_lcore_dev_cur;
2380d19533e8SHuawei Xie 			ll_lcore_dev_cur = ll_lcore_dev_cur->next;
2381d19533e8SHuawei Xie 		}
2382d19533e8SHuawei Xie 	}
2383d19533e8SHuawei Xie 
2384d19533e8SHuawei Xie 	if (ll_lcore_dev_cur == NULL) {
2385d19533e8SHuawei Xie 		RTE_LOG(ERR, VHOST_CONFIG,
2386d19533e8SHuawei Xie 			"(%"PRIu64") Failed to find the dev to be destroy.\n",
2387d19533e8SHuawei Xie 			dev->device_fh);
2388d19533e8SHuawei Xie 		return;
2389d19533e8SHuawei Xie 	}
2390d19533e8SHuawei Xie 
2391d19533e8SHuawei Xie 	/* Search for entry to be removed from main ll */
2392d19533e8SHuawei Xie 	ll_main_dev_cur = ll_root_used;
2393d19533e8SHuawei Xie 	ll_main_dev_last = NULL;
2394d19533e8SHuawei Xie 	while (ll_main_dev_cur != NULL) {
2395*e571e6b4SHuawei Xie 		if (ll_main_dev_cur->vdev == vdev) {
2396d19533e8SHuawei Xie 			break;
2397d19533e8SHuawei Xie 		} else {
2398d19533e8SHuawei Xie 			ll_main_dev_last = ll_main_dev_cur;
2399d19533e8SHuawei Xie 			ll_main_dev_cur = ll_main_dev_cur->next;
2400d19533e8SHuawei Xie 		}
2401d19533e8SHuawei Xie 	}
2402d19533e8SHuawei Xie 
2403d19533e8SHuawei Xie 	/* Remove entries from the lcore and main ll. */
2404*e571e6b4SHuawei Xie 	rm_data_ll_entry(&lcore_info[vdev->coreid].lcore_ll->ll_root_used, ll_lcore_dev_cur, ll_lcore_dev_last);
2405d19533e8SHuawei Xie 	rm_data_ll_entry(&ll_root_used, ll_main_dev_cur, ll_main_dev_last);
2406d19533e8SHuawei Xie 
2407d19533e8SHuawei Xie 	/* Set the dev_removal_flag on each lcore. */
2408d19533e8SHuawei Xie 	RTE_LCORE_FOREACH_SLAVE(lcore) {
2409d19533e8SHuawei Xie 		lcore_info[lcore].lcore_ll->dev_removal_flag = REQUEST_DEV_REMOVAL;
2410d19533e8SHuawei Xie 	}
2411d19533e8SHuawei Xie 
2412d19533e8SHuawei Xie 	/*
2413d19533e8SHuawei Xie 	 * Once each core has set the dev_removal_flag to ACK_DEV_REMOVAL we can be sure that
2414d19533e8SHuawei Xie 	 * they can no longer access the device removed from the linked lists and that the devices
2415d19533e8SHuawei Xie 	 * are no longer in use.
2416d19533e8SHuawei Xie 	 */
2417d19533e8SHuawei Xie 	RTE_LCORE_FOREACH_SLAVE(lcore) {
2418d19533e8SHuawei Xie 		while (lcore_info[lcore].lcore_ll->dev_removal_flag != ACK_DEV_REMOVAL) {
2419d19533e8SHuawei Xie 			rte_pause();
2420d19533e8SHuawei Xie 		}
2421d19533e8SHuawei Xie 	}
2422d19533e8SHuawei Xie 
2423d19533e8SHuawei Xie 	/* Add the entries back to the lcore and main free ll.*/
2424*e571e6b4SHuawei Xie 	put_data_ll_free_entry(&lcore_info[vdev->coreid].lcore_ll->ll_root_free, ll_lcore_dev_cur);
2425d19533e8SHuawei Xie 	put_data_ll_free_entry(&ll_root_free, ll_main_dev_cur);
2426d19533e8SHuawei Xie 
2427d19533e8SHuawei Xie 	/* Decrement number of device on the lcore. */
2428*e571e6b4SHuawei Xie 	lcore_info[vdev->coreid].lcore_ll->device_num--;
2429d19533e8SHuawei Xie 
2430d19533e8SHuawei Xie 	RTE_LOG(INFO, VHOST_DATA, "(%"PRIu64") Device has been removed from data core\n", dev->device_fh);
2431d19533e8SHuawei Xie 
2432d19533e8SHuawei Xie 	if (zero_copy) {
2433*e571e6b4SHuawei Xie 		struct vpool *vpool = &vpool_array[vdev->vmdq_rx_q];
2434d19533e8SHuawei Xie 
2435d19533e8SHuawei Xie 		/* Stop the RX queue. */
2436*e571e6b4SHuawei Xie 		if (rte_eth_dev_rx_queue_stop(ports[0], vdev->vmdq_rx_q) != 0) {
2437d19533e8SHuawei Xie 			LOG_DEBUG(VHOST_CONFIG,
2438d19533e8SHuawei Xie 				"(%"PRIu64") In destroy_device: Failed to stop "
2439d19533e8SHuawei Xie 				"rx queue:%d\n",
2440d19533e8SHuawei Xie 				dev->device_fh,
2441*e571e6b4SHuawei Xie 				vdev->vmdq_rx_q);
2442d19533e8SHuawei Xie 		}
2443d19533e8SHuawei Xie 
2444d19533e8SHuawei Xie 		LOG_DEBUG(VHOST_CONFIG,
2445d19533e8SHuawei Xie 			"(%"PRIu64") in destroy_device: Start put mbuf in "
2446d19533e8SHuawei Xie 			"mempool back to ring for RX queue: %d\n",
2447*e571e6b4SHuawei Xie 			dev->device_fh, vdev->vmdq_rx_q);
2448d19533e8SHuawei Xie 
2449d19533e8SHuawei Xie 		mbuf_destroy_zcp(vpool);
2450d19533e8SHuawei Xie 
2451d19533e8SHuawei Xie 		/* Stop the TX queue. */
2452*e571e6b4SHuawei Xie 		if (rte_eth_dev_tx_queue_stop(ports[0], vdev->vmdq_rx_q) != 0) {
2453d19533e8SHuawei Xie 			LOG_DEBUG(VHOST_CONFIG,
2454d19533e8SHuawei Xie 				"(%"PRIu64") In destroy_device: Failed to "
2455d19533e8SHuawei Xie 				"stop tx queue:%d\n",
2456*e571e6b4SHuawei Xie 				dev->device_fh, vdev->vmdq_rx_q);
2457d19533e8SHuawei Xie 		}
2458d19533e8SHuawei Xie 
2459*e571e6b4SHuawei Xie 		vpool = &vpool_array[vdev->vmdq_rx_q + MAX_QUEUES];
2460d19533e8SHuawei Xie 
2461d19533e8SHuawei Xie 		LOG_DEBUG(VHOST_CONFIG,
2462d19533e8SHuawei Xie 			"(%"PRIu64") destroy_device: Start put mbuf in mempool "
2463d19533e8SHuawei Xie 			"back to ring for TX queue: %d, dev:(%"PRIu64")\n",
2464*e571e6b4SHuawei Xie 			dev->device_fh, (vdev->vmdq_rx_q + MAX_QUEUES),
2465d19533e8SHuawei Xie 			dev->device_fh);
2466d19533e8SHuawei Xie 
2467d19533e8SHuawei Xie 		mbuf_destroy_zcp(vpool);
2468d19533e8SHuawei Xie 	}
2469*e571e6b4SHuawei Xie 	rte_free(vdev);
2470d19533e8SHuawei Xie 
2471d19533e8SHuawei Xie }
2472d19533e8SHuawei Xie 
2473d19533e8SHuawei Xie /*
2474d19533e8SHuawei Xie  * A new device is added to a data core. First the device is added to the main linked list
2475d19533e8SHuawei Xie  * and the allocated to a specific data core.
2476d19533e8SHuawei Xie  */
2477d19533e8SHuawei Xie static int
2478d19533e8SHuawei Xie new_device (struct virtio_net *dev)
2479d19533e8SHuawei Xie {
2480d19533e8SHuawei Xie 	struct virtio_net_data_ll *ll_dev;
2481d19533e8SHuawei Xie 	int lcore, core_add = 0;
2482d19533e8SHuawei Xie 	uint32_t device_num_min = num_devices;
2483*e571e6b4SHuawei Xie 	struct vhost_dev *vdev;
2484*e571e6b4SHuawei Xie 
2485*e571e6b4SHuawei Xie 	vdev = rte_zmalloc("vhost device", sizeof(*vdev), CACHE_LINE_SIZE);
2486*e571e6b4SHuawei Xie 	if (vdev == NULL) {
2487*e571e6b4SHuawei Xie 		RTE_LOG(INFO, VHOST_DATA, "(%"PRIu64") Couldn't allocate memory for vhost dev\n",
2488*e571e6b4SHuawei Xie 			dev->device_fh);
2489*e571e6b4SHuawei Xie 		return -1;
2490*e571e6b4SHuawei Xie 	}
2491*e571e6b4SHuawei Xie 	vdev->dev = dev;
2492*e571e6b4SHuawei Xie 	dev->priv = vdev;
2493d19533e8SHuawei Xie 
2494d19533e8SHuawei Xie 	/* Add device to main ll */
2495d19533e8SHuawei Xie 	ll_dev = get_data_ll_free_entry(&ll_root_free);
2496d19533e8SHuawei Xie 	if (ll_dev == NULL) {
2497d19533e8SHuawei Xie 		RTE_LOG(INFO, VHOST_DATA, "(%"PRIu64") No free entry found in linked list. Device limit "
2498d19533e8SHuawei Xie 			"of %d devices per core has been reached\n",
2499d19533e8SHuawei Xie 			dev->device_fh, num_devices);
2500*e571e6b4SHuawei Xie 		rte_free(vdev);
2501d19533e8SHuawei Xie 		return -1;
2502d19533e8SHuawei Xie 	}
2503*e571e6b4SHuawei Xie 	ll_dev->vdev = vdev;
2504d19533e8SHuawei Xie 	add_data_ll_entry(&ll_root_used, ll_dev);
2505*e571e6b4SHuawei Xie 	vdev->vmdq_rx_q
2506*e571e6b4SHuawei Xie 		= dev->device_fh * (num_queues / num_devices);
2507d19533e8SHuawei Xie 
2508d19533e8SHuawei Xie 	if (zero_copy) {
2509*e571e6b4SHuawei Xie 		uint32_t index = vdev->vmdq_rx_q;
2510d19533e8SHuawei Xie 		uint32_t count_in_ring, i;
2511d19533e8SHuawei Xie 		struct mbuf_table *tx_q;
2512d19533e8SHuawei Xie 
2513d19533e8SHuawei Xie 		count_in_ring = rte_ring_count(vpool_array[index].ring);
2514d19533e8SHuawei Xie 
2515d19533e8SHuawei Xie 		LOG_DEBUG(VHOST_CONFIG,
2516d19533e8SHuawei Xie 			"(%"PRIu64") in new_device: mbuf count in mempool "
2517d19533e8SHuawei Xie 			"before attach is: %d\n",
2518d19533e8SHuawei Xie 			dev->device_fh,
2519d19533e8SHuawei Xie 			rte_mempool_count(vpool_array[index].pool));
2520d19533e8SHuawei Xie 		LOG_DEBUG(VHOST_CONFIG,
2521d19533e8SHuawei Xie 			"(%"PRIu64") in new_device: mbuf count in  ring "
2522d19533e8SHuawei Xie 			"before attach  is : %d\n",
2523d19533e8SHuawei Xie 			dev->device_fh, count_in_ring);
2524d19533e8SHuawei Xie 
2525d19533e8SHuawei Xie 		/*
2526d19533e8SHuawei Xie 		 * Attach all mbufs in vpool.ring and put back intovpool.pool.
2527d19533e8SHuawei Xie 		 */
2528d19533e8SHuawei Xie 		for (i = 0; i < count_in_ring; i++)
2529d19533e8SHuawei Xie 			attach_rxmbuf_zcp(dev);
2530d19533e8SHuawei Xie 
2531d19533e8SHuawei Xie 		LOG_DEBUG(VHOST_CONFIG, "(%"PRIu64") in new_device: mbuf count in "
2532d19533e8SHuawei Xie 			"mempool after attach is: %d\n",
2533d19533e8SHuawei Xie 			dev->device_fh,
2534d19533e8SHuawei Xie 			rte_mempool_count(vpool_array[index].pool));
2535d19533e8SHuawei Xie 		LOG_DEBUG(VHOST_CONFIG, "(%"PRIu64") in new_device: mbuf count in "
2536d19533e8SHuawei Xie 			"ring after attach  is : %d\n",
2537d19533e8SHuawei Xie 			dev->device_fh,
2538d19533e8SHuawei Xie 			rte_ring_count(vpool_array[index].ring));
2539d19533e8SHuawei Xie 
2540*e571e6b4SHuawei Xie 		tx_q = &tx_queue_zcp[(uint16_t)vdev->vmdq_rx_q];
2541*e571e6b4SHuawei Xie 		tx_q->txq_id = vdev->vmdq_rx_q;
2542d19533e8SHuawei Xie 
2543*e571e6b4SHuawei Xie 		if (rte_eth_dev_tx_queue_start(ports[0], vdev->vmdq_rx_q) != 0) {
2544*e571e6b4SHuawei Xie 			struct vpool *vpool = &vpool_array[vdev->vmdq_rx_q];
2545d19533e8SHuawei Xie 
2546d19533e8SHuawei Xie 			LOG_DEBUG(VHOST_CONFIG,
2547d19533e8SHuawei Xie 				"(%"PRIu64") In new_device: Failed to start "
2548d19533e8SHuawei Xie 				"tx queue:%d\n",
2549*e571e6b4SHuawei Xie 				dev->device_fh, vdev->vmdq_rx_q);
2550d19533e8SHuawei Xie 
2551d19533e8SHuawei Xie 			mbuf_destroy_zcp(vpool);
2552*e571e6b4SHuawei Xie 			rte_free(vdev);
2553d19533e8SHuawei Xie 			return -1;
2554d19533e8SHuawei Xie 		}
2555d19533e8SHuawei Xie 
2556*e571e6b4SHuawei Xie 		if (rte_eth_dev_rx_queue_start(ports[0], vdev->vmdq_rx_q) != 0) {
2557*e571e6b4SHuawei Xie 			struct vpool *vpool = &vpool_array[vdev->vmdq_rx_q];
2558d19533e8SHuawei Xie 
2559d19533e8SHuawei Xie 			LOG_DEBUG(VHOST_CONFIG,
2560d19533e8SHuawei Xie 				"(%"PRIu64") In new_device: Failed to start "
2561d19533e8SHuawei Xie 				"rx queue:%d\n",
2562*e571e6b4SHuawei Xie 				dev->device_fh, vdev->vmdq_rx_q);
2563d19533e8SHuawei Xie 
2564d19533e8SHuawei Xie 			/* Stop the TX queue. */
2565d19533e8SHuawei Xie 			if (rte_eth_dev_tx_queue_stop(ports[0],
2566*e571e6b4SHuawei Xie 				vdev->vmdq_rx_q) != 0) {
2567d19533e8SHuawei Xie 				LOG_DEBUG(VHOST_CONFIG,
2568d19533e8SHuawei Xie 					"(%"PRIu64") In new_device: Failed to "
2569d19533e8SHuawei Xie 					"stop tx queue:%d\n",
2570*e571e6b4SHuawei Xie 					dev->device_fh, vdev->vmdq_rx_q);
2571d19533e8SHuawei Xie 			}
2572d19533e8SHuawei Xie 
2573d19533e8SHuawei Xie 			mbuf_destroy_zcp(vpool);
2574*e571e6b4SHuawei Xie 			rte_free(vdev);
2575d19533e8SHuawei Xie 			return -1;
2576d19533e8SHuawei Xie 		}
2577d19533e8SHuawei Xie 
2578d19533e8SHuawei Xie 	}
2579d19533e8SHuawei Xie 
2580d19533e8SHuawei Xie 	/*reset ready flag*/
2581*e571e6b4SHuawei Xie 	vdev->ready = DEVICE_MAC_LEARNING;
2582*e571e6b4SHuawei Xie 	vdev->remove = 0;
2583d19533e8SHuawei Xie 
2584d19533e8SHuawei Xie 	/* Find a suitable lcore to add the device. */
2585d19533e8SHuawei Xie 	RTE_LCORE_FOREACH_SLAVE(lcore) {
2586d19533e8SHuawei Xie 		if (lcore_info[lcore].lcore_ll->device_num < device_num_min) {
2587d19533e8SHuawei Xie 			device_num_min = lcore_info[lcore].lcore_ll->device_num;
2588d19533e8SHuawei Xie 			core_add = lcore;
2589d19533e8SHuawei Xie 		}
2590d19533e8SHuawei Xie 	}
2591d19533e8SHuawei Xie 	/* Add device to lcore ll */
2592d19533e8SHuawei Xie 	ll_dev->dev->coreid = core_add;
2593d19533e8SHuawei Xie 	ll_dev = get_data_ll_free_entry(&lcore_info[ll_dev->dev->coreid].lcore_ll->ll_root_free);
2594d19533e8SHuawei Xie 	if (ll_dev == NULL) {
2595d19533e8SHuawei Xie 		RTE_LOG(INFO, VHOST_DATA, "(%"PRIu64") Failed to add device to data core\n", dev->device_fh);
2596*e571e6b4SHuawei Xie 		vdev->ready = DEVICE_SAFE_REMOVE;
2597d19533e8SHuawei Xie 		destroy_device(dev);
2598*e571e6b4SHuawei Xie 		rte_free(vdev);
2599d19533e8SHuawei Xie 		return -1;
2600d19533e8SHuawei Xie 	}
2601*e571e6b4SHuawei Xie 	ll_dev->vdev = vdev;
2602*e571e6b4SHuawei Xie 	vdev->coreid = core_add;
2603*e571e6b4SHuawei Xie 
2604d19533e8SHuawei Xie 	add_data_ll_entry(&lcore_info[ll_dev->dev->coreid].lcore_ll->ll_root_used, ll_dev);
2605d19533e8SHuawei Xie 
2606d19533e8SHuawei Xie 	/* Initialize device stats */
2607d19533e8SHuawei Xie 	memset(&dev_statistics[dev->device_fh], 0, sizeof(struct device_statistics));
2608d19533e8SHuawei Xie 
2609d19533e8SHuawei Xie 	/* Disable notifications. */
2610d19533e8SHuawei Xie 	set_irq_status(dev);
2611*e571e6b4SHuawei Xie 	lcore_info[vdev->coreid].lcore_ll->device_num++;
2612d19533e8SHuawei Xie 	dev->flags |= VIRTIO_DEV_RUNNING;
2613d19533e8SHuawei Xie 
2614*e571e6b4SHuawei Xie 	RTE_LOG(INFO, VHOST_DATA, "(%"PRIu64") Device has been added to data core %d\n", dev->device_fh, vdev->coreid);
2615d19533e8SHuawei Xie 
2616d19533e8SHuawei Xie 	return 0;
2617d19533e8SHuawei Xie }
2618d19533e8SHuawei Xie 
2619d19533e8SHuawei Xie /*
2620d19533e8SHuawei Xie  * These callback allow devices to be added to the data core when configuration
2621d19533e8SHuawei Xie  * has been fully complete.
2622d19533e8SHuawei Xie  */
2623d19533e8SHuawei Xie static const struct virtio_net_device_ops virtio_net_device_ops =
2624d19533e8SHuawei Xie {
2625d19533e8SHuawei Xie 	.new_device =  new_device,
2626d19533e8SHuawei Xie 	.destroy_device = destroy_device,
2627d19533e8SHuawei Xie };
2628d19533e8SHuawei Xie 
2629d19533e8SHuawei Xie /*
2630d19533e8SHuawei Xie  * This is a thread will wake up after a period to print stats if the user has
2631d19533e8SHuawei Xie  * enabled them.
2632d19533e8SHuawei Xie  */
2633d19533e8SHuawei Xie static void
2634d19533e8SHuawei Xie print_stats(void)
2635d19533e8SHuawei Xie {
2636d19533e8SHuawei Xie 	struct virtio_net_data_ll *dev_ll;
2637d19533e8SHuawei Xie 	uint64_t tx_dropped, rx_dropped;
2638d19533e8SHuawei Xie 	uint64_t tx, tx_total, rx, rx_total;
2639d19533e8SHuawei Xie 	uint32_t device_fh;
2640d19533e8SHuawei Xie 	const char clr[] = { 27, '[', '2', 'J', '\0' };
2641d19533e8SHuawei Xie 	const char top_left[] = { 27, '[', '1', ';', '1', 'H','\0' };
2642d19533e8SHuawei Xie 
2643d19533e8SHuawei Xie 	while(1) {
2644d19533e8SHuawei Xie 		sleep(enable_stats);
2645d19533e8SHuawei Xie 
2646d19533e8SHuawei Xie 		/* Clear screen and move to top left */
2647d19533e8SHuawei Xie 		printf("%s%s", clr, top_left);
2648d19533e8SHuawei Xie 
2649d19533e8SHuawei Xie 		printf("\nDevice statistics ====================================");
2650d19533e8SHuawei Xie 
2651d19533e8SHuawei Xie 		dev_ll = ll_root_used;
2652d19533e8SHuawei Xie 		while (dev_ll != NULL) {
2653*e571e6b4SHuawei Xie 			device_fh = (uint32_t)dev_ll->vdev->dev->device_fh;
2654d19533e8SHuawei Xie 			tx_total = dev_statistics[device_fh].tx_total;
2655d19533e8SHuawei Xie 			tx = dev_statistics[device_fh].tx;
2656d19533e8SHuawei Xie 			tx_dropped = tx_total - tx;
2657d19533e8SHuawei Xie 			if (zero_copy == 0) {
2658d19533e8SHuawei Xie 				rx_total = rte_atomic64_read(
2659d19533e8SHuawei Xie 					&dev_statistics[device_fh].rx_total_atomic);
2660d19533e8SHuawei Xie 				rx = rte_atomic64_read(
2661d19533e8SHuawei Xie 					&dev_statistics[device_fh].rx_atomic);
2662d19533e8SHuawei Xie 			} else {
2663d19533e8SHuawei Xie 				rx_total = dev_statistics[device_fh].rx_total;
2664d19533e8SHuawei Xie 				rx = dev_statistics[device_fh].rx;
2665d19533e8SHuawei Xie 			}
2666d19533e8SHuawei Xie 			rx_dropped = rx_total - rx;
2667d19533e8SHuawei Xie 
2668d19533e8SHuawei Xie 			printf("\nStatistics for device %"PRIu32" ------------------------------"
2669d19533e8SHuawei Xie 					"\nTX total: 		%"PRIu64""
2670d19533e8SHuawei Xie 					"\nTX dropped: 		%"PRIu64""
2671d19533e8SHuawei Xie 					"\nTX successful: 		%"PRIu64""
2672d19533e8SHuawei Xie 					"\nRX total: 		%"PRIu64""
2673d19533e8SHuawei Xie 					"\nRX dropped: 		%"PRIu64""
2674d19533e8SHuawei Xie 					"\nRX successful: 		%"PRIu64"",
2675d19533e8SHuawei Xie 					device_fh,
2676d19533e8SHuawei Xie 					tx_total,
2677d19533e8SHuawei Xie 					tx_dropped,
2678d19533e8SHuawei Xie 					tx,
2679d19533e8SHuawei Xie 					rx_total,
2680d19533e8SHuawei Xie 					rx_dropped,
2681d19533e8SHuawei Xie 					rx);
2682d19533e8SHuawei Xie 
2683d19533e8SHuawei Xie 			dev_ll = dev_ll->next;
2684d19533e8SHuawei Xie 		}
2685d19533e8SHuawei Xie 		printf("\n======================================================\n");
2686d19533e8SHuawei Xie 	}
2687d19533e8SHuawei Xie }
2688d19533e8SHuawei Xie 
2689d19533e8SHuawei Xie static void
2690d19533e8SHuawei Xie setup_mempool_tbl(int socket, uint32_t index, char *pool_name,
2691d19533e8SHuawei Xie 	char *ring_name, uint32_t nb_mbuf)
2692d19533e8SHuawei Xie {
2693d19533e8SHuawei Xie 	uint16_t roomsize = VIRTIO_DESCRIPTOR_LEN_ZCP + RTE_PKTMBUF_HEADROOM;
2694d19533e8SHuawei Xie 	vpool_array[index].pool
2695d19533e8SHuawei Xie 		= rte_mempool_create(pool_name, nb_mbuf, MBUF_SIZE_ZCP,
2696d19533e8SHuawei Xie 		MBUF_CACHE_SIZE_ZCP, sizeof(struct rte_pktmbuf_pool_private),
2697d19533e8SHuawei Xie 		rte_pktmbuf_pool_init, (void *)(uintptr_t)roomsize,
2698d19533e8SHuawei Xie 		rte_pktmbuf_init, NULL, socket, 0);
2699d19533e8SHuawei Xie 	if (vpool_array[index].pool != NULL) {
2700d19533e8SHuawei Xie 		vpool_array[index].ring
2701d19533e8SHuawei Xie 			= rte_ring_create(ring_name,
2702d19533e8SHuawei Xie 				rte_align32pow2(nb_mbuf + 1),
2703d19533e8SHuawei Xie 				socket, RING_F_SP_ENQ | RING_F_SC_DEQ);
2704d19533e8SHuawei Xie 		if (likely(vpool_array[index].ring != NULL)) {
2705d19533e8SHuawei Xie 			LOG_DEBUG(VHOST_CONFIG,
2706d19533e8SHuawei Xie 				"in setup_mempool_tbl: mbuf count in "
2707d19533e8SHuawei Xie 				"mempool is: %d\n",
2708d19533e8SHuawei Xie 				rte_mempool_count(vpool_array[index].pool));
2709d19533e8SHuawei Xie 			LOG_DEBUG(VHOST_CONFIG,
2710d19533e8SHuawei Xie 				"in setup_mempool_tbl: mbuf count in "
2711d19533e8SHuawei Xie 				"ring   is: %d\n",
2712d19533e8SHuawei Xie 				rte_ring_count(vpool_array[index].ring));
2713d19533e8SHuawei Xie 		} else {
2714d19533e8SHuawei Xie 			rte_exit(EXIT_FAILURE, "ring_create(%s) failed",
2715d19533e8SHuawei Xie 				ring_name);
2716d19533e8SHuawei Xie 		}
2717d19533e8SHuawei Xie 
2718d19533e8SHuawei Xie 		/* Need consider head room. */
2719d19533e8SHuawei Xie 		vpool_array[index].buf_size = roomsize - RTE_PKTMBUF_HEADROOM;
2720d19533e8SHuawei Xie 	} else {
2721d19533e8SHuawei Xie 		rte_exit(EXIT_FAILURE, "mempool_create(%s) failed", pool_name);
2722d19533e8SHuawei Xie 	}
2723d19533e8SHuawei Xie }
2724d19533e8SHuawei Xie 
2725d19533e8SHuawei Xie 
2726d19533e8SHuawei Xie /*
2727d19533e8SHuawei Xie  * Main function, does initialisation and calls the per-lcore functions. The CUSE
2728d19533e8SHuawei Xie  * device is also registered here to handle the IOCTLs.
2729d19533e8SHuawei Xie  */
2730d19533e8SHuawei Xie int
2731d19533e8SHuawei Xie MAIN(int argc, char *argv[])
2732d19533e8SHuawei Xie {
2733d19533e8SHuawei Xie 	struct rte_mempool *mbuf_pool = NULL;
2734d19533e8SHuawei Xie 	unsigned lcore_id, core_id = 0;
2735d19533e8SHuawei Xie 	unsigned nb_ports, valid_num_ports;
2736d19533e8SHuawei Xie 	int ret;
2737d19533e8SHuawei Xie 	uint8_t portid, queue_id = 0;
2738d19533e8SHuawei Xie 	static pthread_t tid;
2739d19533e8SHuawei Xie 
2740d19533e8SHuawei Xie 	/* init EAL */
2741d19533e8SHuawei Xie 	ret = rte_eal_init(argc, argv);
2742d19533e8SHuawei Xie 	if (ret < 0)
2743d19533e8SHuawei Xie 		rte_exit(EXIT_FAILURE, "Error with EAL initialization\n");
2744d19533e8SHuawei Xie 	argc -= ret;
2745d19533e8SHuawei Xie 	argv += ret;
2746d19533e8SHuawei Xie 
2747d19533e8SHuawei Xie 	/* parse app arguments */
2748d19533e8SHuawei Xie 	ret = us_vhost_parse_args(argc, argv);
2749d19533e8SHuawei Xie 	if (ret < 0)
2750d19533e8SHuawei Xie 		rte_exit(EXIT_FAILURE, "Invalid argument\n");
2751d19533e8SHuawei Xie 
2752d19533e8SHuawei Xie 	for (lcore_id = 0; lcore_id < RTE_MAX_LCORE; lcore_id ++)
2753d19533e8SHuawei Xie 		if (rte_lcore_is_enabled(lcore_id))
2754d19533e8SHuawei Xie 			lcore_ids[core_id ++] = lcore_id;
2755d19533e8SHuawei Xie 
2756d19533e8SHuawei Xie 	if (rte_lcore_count() > RTE_MAX_LCORE)
2757d19533e8SHuawei Xie 		rte_exit(EXIT_FAILURE,"Not enough cores\n");
2758d19533e8SHuawei Xie 
2759d19533e8SHuawei Xie 	/*set the number of swithcing cores available*/
2760d19533e8SHuawei Xie 	num_switching_cores = rte_lcore_count()-1;
2761d19533e8SHuawei Xie 
2762d19533e8SHuawei Xie 	/* Get the number of physical ports. */
2763d19533e8SHuawei Xie 	nb_ports = rte_eth_dev_count();
2764d19533e8SHuawei Xie 	if (nb_ports > RTE_MAX_ETHPORTS)
2765d19533e8SHuawei Xie 		nb_ports = RTE_MAX_ETHPORTS;
2766d19533e8SHuawei Xie 
2767d19533e8SHuawei Xie 	/*
2768d19533e8SHuawei Xie 	 * Update the global var NUM_PORTS and global array PORTS
2769d19533e8SHuawei Xie 	 * and get value of var VALID_NUM_PORTS according to system ports number
2770d19533e8SHuawei Xie 	 */
2771d19533e8SHuawei Xie 	valid_num_ports = check_ports_num(nb_ports);
2772d19533e8SHuawei Xie 
2773d19533e8SHuawei Xie 	if ((valid_num_ports ==  0) || (valid_num_ports > MAX_SUP_PORTS)) {
2774d19533e8SHuawei Xie 		RTE_LOG(INFO, VHOST_PORT, "Current enabled port number is %u,"
2775d19533e8SHuawei Xie 			"but only %u port can be enabled\n",num_ports, MAX_SUP_PORTS);
2776d19533e8SHuawei Xie 		return -1;
2777d19533e8SHuawei Xie 	}
2778d19533e8SHuawei Xie 
2779d19533e8SHuawei Xie 	if (zero_copy == 0) {
2780d19533e8SHuawei Xie 		/* Create the mbuf pool. */
2781d19533e8SHuawei Xie 		mbuf_pool = rte_mempool_create(
2782d19533e8SHuawei Xie 				"MBUF_POOL",
2783d19533e8SHuawei Xie 				NUM_MBUFS_PER_PORT
2784d19533e8SHuawei Xie 				* valid_num_ports,
2785d19533e8SHuawei Xie 				MBUF_SIZE, MBUF_CACHE_SIZE,
2786d19533e8SHuawei Xie 				sizeof(struct rte_pktmbuf_pool_private),
2787d19533e8SHuawei Xie 				rte_pktmbuf_pool_init, NULL,
2788d19533e8SHuawei Xie 				rte_pktmbuf_init, NULL,
2789d19533e8SHuawei Xie 				rte_socket_id(), 0);
2790d19533e8SHuawei Xie 		if (mbuf_pool == NULL)
2791d19533e8SHuawei Xie 			rte_exit(EXIT_FAILURE, "Cannot create mbuf pool\n");
2792d19533e8SHuawei Xie 
2793d19533e8SHuawei Xie 		for (queue_id = 0; queue_id < MAX_QUEUES + 1; queue_id++)
2794d19533e8SHuawei Xie 			vpool_array[queue_id].pool = mbuf_pool;
2795d19533e8SHuawei Xie 
2796d19533e8SHuawei Xie 		if (vm2vm_mode == VM2VM_HARDWARE) {
2797d19533e8SHuawei Xie 			/* Enable VT loop back to let L2 switch to do it. */
2798d19533e8SHuawei Xie 			vmdq_conf_default.rx_adv_conf.vmdq_rx_conf.enable_loop_back = 1;
2799d19533e8SHuawei Xie 			LOG_DEBUG(VHOST_CONFIG,
2800d19533e8SHuawei Xie 				"Enable loop back for L2 switch in vmdq.\n");
2801d19533e8SHuawei Xie 		}
2802d19533e8SHuawei Xie 	} else {
2803d19533e8SHuawei Xie 		uint32_t nb_mbuf;
2804d19533e8SHuawei Xie 		char pool_name[RTE_MEMPOOL_NAMESIZE];
2805d19533e8SHuawei Xie 		char ring_name[RTE_MEMPOOL_NAMESIZE];
2806d19533e8SHuawei Xie 
2807d19533e8SHuawei Xie 		/*
2808d19533e8SHuawei Xie 		 * Zero copy defers queue RX/TX start to the time when guest
2809d19533e8SHuawei Xie 		 * finishes its startup and packet buffers from that guest are
2810d19533e8SHuawei Xie 		 * available.
2811d19533e8SHuawei Xie 		 */
2812d19533e8SHuawei Xie 		rx_conf_default.rx_deferred_start = (uint8_t)zero_copy;
2813d19533e8SHuawei Xie 		rx_conf_default.rx_drop_en = 0;
2814d19533e8SHuawei Xie 		tx_conf_default.tx_deferred_start = (uint8_t)zero_copy;
2815d19533e8SHuawei Xie 		nb_mbuf = num_rx_descriptor
2816d19533e8SHuawei Xie 			+ num_switching_cores * MBUF_CACHE_SIZE_ZCP
2817d19533e8SHuawei Xie 			+ num_switching_cores * MAX_PKT_BURST;
2818d19533e8SHuawei Xie 
2819d19533e8SHuawei Xie 		for (queue_id = 0; queue_id < MAX_QUEUES; queue_id++) {
2820d19533e8SHuawei Xie 			snprintf(pool_name, sizeof(pool_name),
2821d19533e8SHuawei Xie 				"rxmbuf_pool_%u", queue_id);
2822d19533e8SHuawei Xie 			snprintf(ring_name, sizeof(ring_name),
2823d19533e8SHuawei Xie 				"rxmbuf_ring_%u", queue_id);
2824d19533e8SHuawei Xie 			setup_mempool_tbl(rte_socket_id(), queue_id,
2825d19533e8SHuawei Xie 				pool_name, ring_name, nb_mbuf);
2826d19533e8SHuawei Xie 		}
2827d19533e8SHuawei Xie 
2828d19533e8SHuawei Xie 		nb_mbuf = num_tx_descriptor
2829d19533e8SHuawei Xie 				+ num_switching_cores * MBUF_CACHE_SIZE_ZCP
2830d19533e8SHuawei Xie 				+ num_switching_cores * MAX_PKT_BURST;
2831d19533e8SHuawei Xie 
2832d19533e8SHuawei Xie 		for (queue_id = 0; queue_id < MAX_QUEUES; queue_id++) {
2833d19533e8SHuawei Xie 			snprintf(pool_name, sizeof(pool_name),
2834d19533e8SHuawei Xie 				"txmbuf_pool_%u", queue_id);
2835d19533e8SHuawei Xie 			snprintf(ring_name, sizeof(ring_name),
2836d19533e8SHuawei Xie 				"txmbuf_ring_%u", queue_id);
2837d19533e8SHuawei Xie 			setup_mempool_tbl(rte_socket_id(),
2838d19533e8SHuawei Xie 				(queue_id + MAX_QUEUES),
2839d19533e8SHuawei Xie 				pool_name, ring_name, nb_mbuf);
2840d19533e8SHuawei Xie 		}
2841d19533e8SHuawei Xie 
2842d19533e8SHuawei Xie 		if (vm2vm_mode == VM2VM_HARDWARE) {
2843d19533e8SHuawei Xie 			/* Enable VT loop back to let L2 switch to do it. */
2844d19533e8SHuawei Xie 			vmdq_conf_default.rx_adv_conf.vmdq_rx_conf.enable_loop_back = 1;
2845d19533e8SHuawei Xie 			LOG_DEBUG(VHOST_CONFIG,
2846d19533e8SHuawei Xie 				"Enable loop back for L2 switch in vmdq.\n");
2847d19533e8SHuawei Xie 		}
2848d19533e8SHuawei Xie 	}
2849d19533e8SHuawei Xie 	/* Set log level. */
2850d19533e8SHuawei Xie 	rte_set_log_level(LOG_LEVEL);
2851d19533e8SHuawei Xie 
2852d19533e8SHuawei Xie 	/* initialize all ports */
2853d19533e8SHuawei Xie 	for (portid = 0; portid < nb_ports; portid++) {
2854d19533e8SHuawei Xie 		/* skip ports that are not enabled */
2855d19533e8SHuawei Xie 		if ((enabled_port_mask & (1 << portid)) == 0) {
2856d19533e8SHuawei Xie 			RTE_LOG(INFO, VHOST_PORT,
2857d19533e8SHuawei Xie 				"Skipping disabled port %d\n", portid);
2858d19533e8SHuawei Xie 			continue;
2859d19533e8SHuawei Xie 		}
2860d19533e8SHuawei Xie 		if (port_init(portid) != 0)
2861d19533e8SHuawei Xie 			rte_exit(EXIT_FAILURE,
2862d19533e8SHuawei Xie 				"Cannot initialize network ports\n");
2863d19533e8SHuawei Xie 	}
2864d19533e8SHuawei Xie 
2865d19533e8SHuawei Xie 	/* Initialise all linked lists. */
2866d19533e8SHuawei Xie 	if (init_data_ll() == -1)
2867d19533e8SHuawei Xie 		rte_exit(EXIT_FAILURE, "Failed to initialize linked list\n");
2868d19533e8SHuawei Xie 
2869d19533e8SHuawei Xie 	/* Initialize device stats */
2870d19533e8SHuawei Xie 	memset(&dev_statistics, 0, sizeof(dev_statistics));
2871d19533e8SHuawei Xie 
2872d19533e8SHuawei Xie 	/* Enable stats if the user option is set. */
2873d19533e8SHuawei Xie 	if (enable_stats)
2874d19533e8SHuawei Xie 		pthread_create(&tid, NULL, (void*)print_stats, NULL );
2875d19533e8SHuawei Xie 
2876d19533e8SHuawei Xie 	/* Launch all data cores. */
2877d19533e8SHuawei Xie 	if (zero_copy == 0) {
2878d19533e8SHuawei Xie 		RTE_LCORE_FOREACH_SLAVE(lcore_id) {
2879d19533e8SHuawei Xie 			rte_eal_remote_launch(switch_worker,
2880d19533e8SHuawei Xie 				mbuf_pool, lcore_id);
2881d19533e8SHuawei Xie 		}
2882d19533e8SHuawei Xie 	} else {
2883d19533e8SHuawei Xie 		uint32_t count_in_mempool, index, i;
2884d19533e8SHuawei Xie 		for (index = 0; index < 2*MAX_QUEUES; index++) {
2885d19533e8SHuawei Xie 			/* For all RX and TX queues. */
2886d19533e8SHuawei Xie 			count_in_mempool
2887d19533e8SHuawei Xie 				= rte_mempool_count(vpool_array[index].pool);
2888d19533e8SHuawei Xie 
2889d19533e8SHuawei Xie 			/*
2890d19533e8SHuawei Xie 			 * Transfer all un-attached mbufs from vpool.pool
2891d19533e8SHuawei Xie 			 * to vpoo.ring.
2892d19533e8SHuawei Xie 			 */
2893d19533e8SHuawei Xie 			for (i = 0; i < count_in_mempool; i++) {
2894d19533e8SHuawei Xie 				struct rte_mbuf *mbuf
2895d19533e8SHuawei Xie 					= __rte_mbuf_raw_alloc(
2896d19533e8SHuawei Xie 						vpool_array[index].pool);
2897d19533e8SHuawei Xie 				rte_ring_sp_enqueue(vpool_array[index].ring,
2898d19533e8SHuawei Xie 						(void *)mbuf);
2899d19533e8SHuawei Xie 			}
2900d19533e8SHuawei Xie 
2901d19533e8SHuawei Xie 			LOG_DEBUG(VHOST_CONFIG,
2902d19533e8SHuawei Xie 				"in MAIN: mbuf count in mempool at initial "
2903d19533e8SHuawei Xie 				"is: %d\n", count_in_mempool);
2904d19533e8SHuawei Xie 			LOG_DEBUG(VHOST_CONFIG,
2905d19533e8SHuawei Xie 				"in MAIN: mbuf count in  ring at initial  is :"
2906d19533e8SHuawei Xie 				" %d\n",
2907d19533e8SHuawei Xie 				rte_ring_count(vpool_array[index].ring));
2908d19533e8SHuawei Xie 		}
2909d19533e8SHuawei Xie 
2910d19533e8SHuawei Xie 		RTE_LCORE_FOREACH_SLAVE(lcore_id)
2911d19533e8SHuawei Xie 			rte_eal_remote_launch(switch_worker_zcp, NULL,
2912d19533e8SHuawei Xie 				lcore_id);
2913d19533e8SHuawei Xie 	}
2914d19533e8SHuawei Xie 
2915d19533e8SHuawei Xie 	/* Register CUSE device to handle IOCTLs. */
2916d19533e8SHuawei Xie 	ret = register_cuse_device((char*)&dev_basename, dev_index, get_virtio_net_callbacks());
2917d19533e8SHuawei Xie 	if (ret != 0)
2918d19533e8SHuawei Xie 		rte_exit(EXIT_FAILURE,"CUSE device setup failure.\n");
2919d19533e8SHuawei Xie 
2920d19533e8SHuawei Xie 	init_virtio_net(&virtio_net_device_ops);
2921d19533e8SHuawei Xie 
2922d19533e8SHuawei Xie 	/* Start CUSE session. */
2923d19533e8SHuawei Xie 	start_cuse_session_loop();
2924d19533e8SHuawei Xie 	return 0;
2925d19533e8SHuawei Xie 
2926d19533e8SHuawei Xie }
2927d19533e8SHuawei Xie 
2928