xref: /dpdk/examples/vdpa/main.c (revision 1b2798ee320657d7b6a2ba8ec1e1e949df62a18d)
1edbed86dSXiaolong Ye /* SPDX-License-Identifier: BSD-3-Clause
2edbed86dSXiaolong Ye  * Copyright(c) 2018 Intel Corporation
3edbed86dSXiaolong Ye  */
4edbed86dSXiaolong Ye 
5edbed86dSXiaolong Ye #include <getopt.h>
6edbed86dSXiaolong Ye #include <signal.h>
7edbed86dSXiaolong Ye #include <stdint.h>
872b452c5SDmitry Kozlyuk #include <stdlib.h>
9edbed86dSXiaolong Ye #include <string.h>
10edbed86dSXiaolong Ye #include <unistd.h>
11edbed86dSXiaolong Ye 
12edbed86dSXiaolong Ye #include <rte_ethdev.h>
13edbed86dSXiaolong Ye #include <rte_malloc.h>
14edbed86dSXiaolong Ye #include <rte_vhost.h>
15edbed86dSXiaolong Ye #include <rte_vdpa.h>
16edbed86dSXiaolong Ye #include <rte_pci.h>
17edbed86dSXiaolong Ye #include <rte_string_fns.h>
18edbed86dSXiaolong Ye 
19edbed86dSXiaolong Ye #include <cmdline_socket.h>
20*1b2798eeSBruce Richardson #include "commands.h"  /* auto-generated file from commands.list */
21d5499149SAndy Pei #include "vdpa_blk_compact.h"
22edbed86dSXiaolong Ye 
23edbed86dSXiaolong Ye #define MAX_PATH_LEN 128
24edbed86dSXiaolong Ye #define MAX_VDPA_SAMPLE_PORTS 1024
25edbed86dSXiaolong Ye #define RTE_LOGTYPE_VDPA RTE_LOGTYPE_USER1
26edbed86dSXiaolong Ye 
27edbed86dSXiaolong Ye struct vdpa_port {
28edbed86dSXiaolong Ye 	char ifname[MAX_PATH_LEN];
290f700f90SMaxime Coquelin 	struct rte_vdpa_device *dev;
30edbed86dSXiaolong Ye 	int vid;
31edbed86dSXiaolong Ye 	uint64_t flags;
326505865aSMatan Azrad 	int stats_n;
336505865aSMatan Azrad 	struct rte_vdpa_stat_name *stats_names;
346505865aSMatan Azrad 	struct rte_vdpa_stat *stats;
35edbed86dSXiaolong Ye };
36edbed86dSXiaolong Ye 
37edbed86dSXiaolong Ye static struct vdpa_port vports[MAX_VDPA_SAMPLE_PORTS];
38edbed86dSXiaolong Ye 
39edbed86dSXiaolong Ye static char iface[MAX_PATH_LEN];
40edbed86dSXiaolong Ye static int devcnt;
41edbed86dSXiaolong Ye static int interactive;
42edbed86dSXiaolong Ye static int client_mode;
43edbed86dSXiaolong Ye 
44edbed86dSXiaolong Ye /* display usage */
45edbed86dSXiaolong Ye static void
vdpa_usage(const char * prgname)46edbed86dSXiaolong Ye vdpa_usage(const char *prgname)
47edbed86dSXiaolong Ye {
48edbed86dSXiaolong Ye 	printf("Usage: %s [EAL options] -- "
49edbed86dSXiaolong Ye 				 "	--interactive|-i: run in interactive mode.\n"
50edbed86dSXiaolong Ye 				 "	--iface <path>: specify the path prefix of the socket files, e.g. /tmp/vhost-user-.\n"
51edbed86dSXiaolong Ye 				 "	--client: register a vhost-user socket as client mode.\n",
52edbed86dSXiaolong Ye 				 prgname);
53edbed86dSXiaolong Ye }
54edbed86dSXiaolong Ye 
55edbed86dSXiaolong Ye static int
parse_args(int argc,char ** argv)56edbed86dSXiaolong Ye parse_args(int argc, char **argv)
57edbed86dSXiaolong Ye {
58edbed86dSXiaolong Ye 	static const char *short_option = "i";
59edbed86dSXiaolong Ye 	static struct option long_option[] = {
60edbed86dSXiaolong Ye 		{"iface", required_argument, NULL, 0},
61edbed86dSXiaolong Ye 		{"interactive", no_argument, &interactive, 1},
62edbed86dSXiaolong Ye 		{"client", no_argument, &client_mode, 1},
63edbed86dSXiaolong Ye 		{NULL, 0, 0, 0},
64edbed86dSXiaolong Ye 	};
65edbed86dSXiaolong Ye 	int opt, idx;
66edbed86dSXiaolong Ye 	char *prgname = argv[0];
67edbed86dSXiaolong Ye 
68edbed86dSXiaolong Ye 	while ((opt = getopt_long(argc, argv, short_option, long_option, &idx))
69edbed86dSXiaolong Ye 			!= EOF) {
70edbed86dSXiaolong Ye 		switch (opt) {
71edbed86dSXiaolong Ye 		case 'i':
72edbed86dSXiaolong Ye 			printf("Interactive-mode selected\n");
73edbed86dSXiaolong Ye 			interactive = 1;
74edbed86dSXiaolong Ye 			break;
75edbed86dSXiaolong Ye 		/* long options */
76edbed86dSXiaolong Ye 		case 0:
77edbed86dSXiaolong Ye 			if (strncmp(long_option[idx].name, "iface",
78edbed86dSXiaolong Ye 						MAX_PATH_LEN) == 0) {
79edbed86dSXiaolong Ye 				rte_strscpy(iface, optarg, MAX_PATH_LEN);
80edbed86dSXiaolong Ye 				printf("iface %s\n", iface);
81edbed86dSXiaolong Ye 			}
82edbed86dSXiaolong Ye 			if (!strcmp(long_option[idx].name, "interactive")) {
83edbed86dSXiaolong Ye 				printf("Interactive-mode selected\n");
84edbed86dSXiaolong Ye 				interactive = 1;
85edbed86dSXiaolong Ye 			}
86edbed86dSXiaolong Ye 			break;
87edbed86dSXiaolong Ye 
88edbed86dSXiaolong Ye 		default:
89edbed86dSXiaolong Ye 			vdpa_usage(prgname);
90edbed86dSXiaolong Ye 			return -1;
91edbed86dSXiaolong Ye 		}
92edbed86dSXiaolong Ye 	}
93edbed86dSXiaolong Ye 
94edbed86dSXiaolong Ye 	if (iface[0] == '\0' && interactive == 0) {
95edbed86dSXiaolong Ye 		vdpa_usage(prgname);
96edbed86dSXiaolong Ye 		return -1;
97edbed86dSXiaolong Ye 	}
98edbed86dSXiaolong Ye 
99edbed86dSXiaolong Ye 	return 0;
100edbed86dSXiaolong Ye }
101edbed86dSXiaolong Ye 
102edbed86dSXiaolong Ye static int
new_device(int vid)103edbed86dSXiaolong Ye new_device(int vid)
104edbed86dSXiaolong Ye {
105edbed86dSXiaolong Ye 	char ifname[MAX_PATH_LEN];
1060f700f90SMaxime Coquelin 	struct rte_device *dev;
107edbed86dSXiaolong Ye 	int i;
108edbed86dSXiaolong Ye 
109edbed86dSXiaolong Ye 	rte_vhost_get_ifname(vid, ifname, sizeof(ifname));
110edbed86dSXiaolong Ye 	for (i = 0; i < MAX_VDPA_SAMPLE_PORTS; i++) {
1110f700f90SMaxime Coquelin 		if (strncmp(ifname, vports[i].ifname, MAX_PATH_LEN))
1120f700f90SMaxime Coquelin 			continue;
1130f700f90SMaxime Coquelin 
1140f700f90SMaxime Coquelin 		dev = rte_vdpa_get_rte_device(vports[i].dev);
1150f700f90SMaxime Coquelin 		if (!dev) {
1160f700f90SMaxime Coquelin 			RTE_LOG(ERR, VDPA,
1170f700f90SMaxime Coquelin 				"Failed to get generic device for port %d\n", i);
1180f700f90SMaxime Coquelin 			continue;
1190f700f90SMaxime Coquelin 		}
120ec5ecd7eSDavid Marchand 		printf("\nnew port %s, device : %s\n", ifname, rte_dev_name(dev));
121edbed86dSXiaolong Ye 		vports[i].vid = vid;
122edbed86dSXiaolong Ye 		break;
123edbed86dSXiaolong Ye 	}
124edbed86dSXiaolong Ye 
125edbed86dSXiaolong Ye 	if (i >= MAX_VDPA_SAMPLE_PORTS)
126edbed86dSXiaolong Ye 		return -1;
127edbed86dSXiaolong Ye 
128edbed86dSXiaolong Ye 	return 0;
129edbed86dSXiaolong Ye }
130edbed86dSXiaolong Ye 
131edbed86dSXiaolong Ye static void
destroy_device(int vid)132edbed86dSXiaolong Ye destroy_device(int vid)
133edbed86dSXiaolong Ye {
1340f700f90SMaxime Coquelin 	struct rte_device *dev;
135edbed86dSXiaolong Ye 	char ifname[MAX_PATH_LEN];
136edbed86dSXiaolong Ye 	int i;
137edbed86dSXiaolong Ye 
138edbed86dSXiaolong Ye 	rte_vhost_get_ifname(vid, ifname, sizeof(ifname));
139edbed86dSXiaolong Ye 	for (i = 0; i < MAX_VDPA_SAMPLE_PORTS; i++) {
1400f700f90SMaxime Coquelin 		if (strncmp(ifname, vports[i].ifname, MAX_PATH_LEN))
1410f700f90SMaxime Coquelin 			continue;
1420f700f90SMaxime Coquelin 
1430f700f90SMaxime Coquelin 		dev = rte_vdpa_get_rte_device(vports[i].dev);
1440f700f90SMaxime Coquelin 		if (!dev) {
1450f700f90SMaxime Coquelin 			RTE_LOG(ERR, VDPA,
1460f700f90SMaxime Coquelin 				"Failed to get generic device for port %d\n", i);
1470f700f90SMaxime Coquelin 			continue;
148edbed86dSXiaolong Ye 		}
1490f700f90SMaxime Coquelin 
150ec5ecd7eSDavid Marchand 		printf("\ndestroy port %s, device: %s\n", ifname, rte_dev_name(dev));
1510f700f90SMaxime Coquelin 		break;
152edbed86dSXiaolong Ye 	}
153edbed86dSXiaolong Ye }
154edbed86dSXiaolong Ye 
155ab4bb424SMaxime Coquelin static const struct rte_vhost_device_ops vdpa_sample_devops = {
156edbed86dSXiaolong Ye 	.new_device = new_device,
157edbed86dSXiaolong Ye 	.destroy_device = destroy_device,
158edbed86dSXiaolong Ye };
159edbed86dSXiaolong Ye 
160edbed86dSXiaolong Ye static int
vdpa_blk_device_set_features_and_protocol(const char * path)161d5499149SAndy Pei vdpa_blk_device_set_features_and_protocol(const char *path)
162d5499149SAndy Pei {
163d5499149SAndy Pei 	uint64_t protocol_features = 0;
164d5499149SAndy Pei 	int ret;
165d5499149SAndy Pei 
166d5499149SAndy Pei 	ret = rte_vhost_driver_set_features(path, VHOST_BLK_FEATURES);
167d5499149SAndy Pei 	if (ret != 0) {
168d5499149SAndy Pei 		RTE_LOG(ERR, VDPA,
169d5499149SAndy Pei 			"rte_vhost_driver_set_features for %s failed.\n",
170d5499149SAndy Pei 			path);
171d5499149SAndy Pei 		goto out;
172d5499149SAndy Pei 	}
173d5499149SAndy Pei 
174d5499149SAndy Pei 	ret = rte_vhost_driver_disable_features(path,
175d5499149SAndy Pei 		VHOST_BLK_DISABLED_FEATURES);
176d5499149SAndy Pei 	if (ret != 0) {
177d5499149SAndy Pei 		RTE_LOG(ERR, VDPA,
178d5499149SAndy Pei 			"rte_vhost_driver_disable_features for %s failed.\n",
179d5499149SAndy Pei 			path);
180d5499149SAndy Pei 		goto out;
181d5499149SAndy Pei 	}
182d5499149SAndy Pei 
183d5499149SAndy Pei 	ret = rte_vhost_driver_get_protocol_features(path, &protocol_features);
184d5499149SAndy Pei 	if (ret != 0) {
185d5499149SAndy Pei 		RTE_LOG(ERR, VDPA,
186d5499149SAndy Pei 			"rte_vhost_driver_get_protocol_features for %s failed.\n",
187d5499149SAndy Pei 			path);
188d5499149SAndy Pei 		goto out;
189d5499149SAndy Pei 	}
190d5499149SAndy Pei 
191d5499149SAndy Pei 	protocol_features |= VHOST_BLK_PROTOCOL_FEATURES;
192d5499149SAndy Pei 
193d5499149SAndy Pei 	ret = rte_vhost_driver_set_protocol_features(path, protocol_features);
194d5499149SAndy Pei 	if (ret != 0) {
195d5499149SAndy Pei 		RTE_LOG(ERR, VDPA,
196d5499149SAndy Pei 			"rte_vhost_driver_set_protocol_features for %s failed.\n",
197d5499149SAndy Pei 			path);
198d5499149SAndy Pei 	}
199d5499149SAndy Pei 
200d5499149SAndy Pei out:
201d5499149SAndy Pei 	return ret;
202d5499149SAndy Pei }
203d5499149SAndy Pei 
204d5499149SAndy Pei static int
start_vdpa(struct vdpa_port * vport)205edbed86dSXiaolong Ye start_vdpa(struct vdpa_port *vport)
206edbed86dSXiaolong Ye {
207d5499149SAndy Pei 	uint32_t device_type = 0;
208edbed86dSXiaolong Ye 	int ret;
209edbed86dSXiaolong Ye 	char *socket_path = vport->ifname;
210edbed86dSXiaolong Ye 
211edbed86dSXiaolong Ye 	if (client_mode)
212edbed86dSXiaolong Ye 		vport->flags |= RTE_VHOST_USER_CLIENT;
213edbed86dSXiaolong Ye 
21434924a89SHao Chen 	vport->flags |= RTE_VHOST_USER_IOMMU_SUPPORT;
21534924a89SHao Chen 
216edbed86dSXiaolong Ye 	if (access(socket_path, F_OK) != -1 && !client_mode) {
217edbed86dSXiaolong Ye 		RTE_LOG(ERR, VDPA,
218edbed86dSXiaolong Ye 			"%s exists, please remove it or specify another file and try again.\n",
219edbed86dSXiaolong Ye 			socket_path);
220edbed86dSXiaolong Ye 		return -1;
221edbed86dSXiaolong Ye 	}
222edbed86dSXiaolong Ye 	ret = rte_vhost_driver_register(socket_path, vport->flags);
223edbed86dSXiaolong Ye 	if (ret != 0)
224edbed86dSXiaolong Ye 		rte_exit(EXIT_FAILURE,
225edbed86dSXiaolong Ye 			"register driver failed: %s\n",
226edbed86dSXiaolong Ye 			socket_path);
227edbed86dSXiaolong Ye 
228edbed86dSXiaolong Ye 	ret = rte_vhost_driver_callback_register(socket_path,
229edbed86dSXiaolong Ye 			&vdpa_sample_devops);
230edbed86dSXiaolong Ye 	if (ret != 0)
231edbed86dSXiaolong Ye 		rte_exit(EXIT_FAILURE,
232edbed86dSXiaolong Ye 			"register driver ops failed: %s\n",
233edbed86dSXiaolong Ye 			socket_path);
234edbed86dSXiaolong Ye 
2350f700f90SMaxime Coquelin 	ret = rte_vhost_driver_attach_vdpa_device(socket_path, vport->dev);
236edbed86dSXiaolong Ye 	if (ret != 0)
237edbed86dSXiaolong Ye 		rte_exit(EXIT_FAILURE,
238edbed86dSXiaolong Ye 			"attach vdpa device failed: %s\n",
239edbed86dSXiaolong Ye 			socket_path);
240edbed86dSXiaolong Ye 
241d5499149SAndy Pei 	ret = rte_vhost_driver_get_vdpa_dev_type(socket_path, &device_type);
242d5499149SAndy Pei 	if (ret == 0 && device_type == RTE_VHOST_VDPA_DEVICE_TYPE_BLK) {
243d5499149SAndy Pei 		RTE_LOG(NOTICE, VDPA, "%s is a blk device\n", socket_path);
244d5499149SAndy Pei 		ret = vdpa_blk_device_set_features_and_protocol(socket_path);
245d5499149SAndy Pei 		if (ret != 0)
246d5499149SAndy Pei 			rte_exit(EXIT_FAILURE,
247d5499149SAndy Pei 				"set vhost blk driver features and protocol features failed: %s\n",
248d5499149SAndy Pei 				socket_path);
249d5499149SAndy Pei 	}
250d5499149SAndy Pei 
251edbed86dSXiaolong Ye 	if (rte_vhost_driver_start(socket_path) < 0)
252edbed86dSXiaolong Ye 		rte_exit(EXIT_FAILURE,
253edbed86dSXiaolong Ye 			"start vhost driver failed: %s\n",
254edbed86dSXiaolong Ye 			socket_path);
255edbed86dSXiaolong Ye 	return 0;
256edbed86dSXiaolong Ye }
257edbed86dSXiaolong Ye 
258edbed86dSXiaolong Ye static void
close_vdpa(struct vdpa_port * vport)259edbed86dSXiaolong Ye close_vdpa(struct vdpa_port *vport)
260edbed86dSXiaolong Ye {
261edbed86dSXiaolong Ye 	int ret;
262edbed86dSXiaolong Ye 	char *socket_path = vport->ifname;
263edbed86dSXiaolong Ye 
264edbed86dSXiaolong Ye 	ret = rte_vhost_driver_detach_vdpa_device(socket_path);
265edbed86dSXiaolong Ye 	if (ret != 0)
266edbed86dSXiaolong Ye 		RTE_LOG(ERR, VDPA,
267edbed86dSXiaolong Ye 				"detach vdpa device failed: %s\n",
268edbed86dSXiaolong Ye 				socket_path);
269edbed86dSXiaolong Ye 
270edbed86dSXiaolong Ye 	ret = rte_vhost_driver_unregister(socket_path);
271edbed86dSXiaolong Ye 	if (ret != 0)
272edbed86dSXiaolong Ye 		RTE_LOG(ERR, VDPA,
273edbed86dSXiaolong Ye 				"Fail to unregister vhost driver for %s.\n",
274edbed86dSXiaolong Ye 				socket_path);
2756505865aSMatan Azrad 	if (vport->stats_names) {
2766505865aSMatan Azrad 		rte_free(vport->stats_names);
2776505865aSMatan Azrad 		vport->stats_names = NULL;
2786505865aSMatan Azrad 	}
279edbed86dSXiaolong Ye }
280edbed86dSXiaolong Ye 
281edbed86dSXiaolong Ye static void
vdpa_sample_quit(void)282edbed86dSXiaolong Ye vdpa_sample_quit(void)
283edbed86dSXiaolong Ye {
284edbed86dSXiaolong Ye 	int i;
285fbe6be6eSMaxime Coquelin 	for (i = 0; i < RTE_MIN(MAX_VDPA_SAMPLE_PORTS, devcnt); i++) {
286edbed86dSXiaolong Ye 		if (vports[i].ifname[0] != '\0')
287edbed86dSXiaolong Ye 			close_vdpa(&vports[i]);
288edbed86dSXiaolong Ye 	}
289edbed86dSXiaolong Ye }
290edbed86dSXiaolong Ye 
291edbed86dSXiaolong Ye static void
signal_handler(int signum)292edbed86dSXiaolong Ye signal_handler(int signum)
293edbed86dSXiaolong Ye {
294edbed86dSXiaolong Ye 	if (signum == SIGINT || signum == SIGTERM) {
295edbed86dSXiaolong Ye 		printf("\nSignal %d received, preparing to exit...\n", signum);
296edbed86dSXiaolong Ye 		vdpa_sample_quit();
297edbed86dSXiaolong Ye 		exit(0);
298edbed86dSXiaolong Ye 	}
299edbed86dSXiaolong Ye }
300edbed86dSXiaolong Ye 
301*1b2798eeSBruce Richardson /* interactive cmd functions */
302edbed86dSXiaolong Ye 
cmd_help_parsed(__rte_unused void * parsed_result,struct cmdline * cl,__rte_unused void * data)303*1b2798eeSBruce Richardson void cmd_help_parsed(__rte_unused void *parsed_result,
304edbed86dSXiaolong Ye 		struct cmdline *cl,
305f2fc83b4SThomas Monjalon 		__rte_unused void *data)
306edbed86dSXiaolong Ye {
307edbed86dSXiaolong Ye 	cmdline_printf(
308edbed86dSXiaolong Ye 		cl,
309edbed86dSXiaolong Ye 		"\n"
310edbed86dSXiaolong Ye 		"The following commands are currently available:\n\n"
311edbed86dSXiaolong Ye 		"Control:\n"
312edbed86dSXiaolong Ye 		"    help                                      : Show interactive instructions.\n"
313edbed86dSXiaolong Ye 		"    list                                      : list all available vdpa devices.\n"
314edbed86dSXiaolong Ye 		"    create <socket file> <vdev addr>          : create a new vdpa port.\n"
3156505865aSMatan Azrad 		"    stats <device ID> <virtio queue ID>       : show statistics of virtio queue, 0xffff for all.\n"
316edbed86dSXiaolong Ye 		"    quit                                      : exit vdpa sample app.\n"
317edbed86dSXiaolong Ye 	);
318edbed86dSXiaolong Ye }
319edbed86dSXiaolong Ye 
cmd_list_parsed(__rte_unused void * parsed_result,struct cmdline * cl,__rte_unused void * data)320*1b2798eeSBruce Richardson void cmd_list_parsed(
321f2fc83b4SThomas Monjalon 		__rte_unused void *parsed_result,
322edbed86dSXiaolong Ye 		struct cmdline *cl,
323f2fc83b4SThomas Monjalon 		__rte_unused void *data)
324edbed86dSXiaolong Ye {
325edbed86dSXiaolong Ye 	uint32_t queue_num;
326edbed86dSXiaolong Ye 	uint64_t features;
327edbed86dSXiaolong Ye 	struct rte_vdpa_device *vdev;
32838f8ab0bSMaxime Coquelin 	struct rte_device *dev;
32938f8ab0bSMaxime Coquelin 	struct rte_dev_iterator dev_iter;
330edbed86dSXiaolong Ye 
3310f700f90SMaxime Coquelin 	cmdline_printf(cl, "device name\tqueue num\tsupported features\n");
33238f8ab0bSMaxime Coquelin 	RTE_DEV_FOREACH(dev, "class=vdpa", &dev_iter) {
333ec5ecd7eSDavid Marchand 		vdev = rte_vdpa_find_device_by_name(rte_dev_name(dev));
334edbed86dSXiaolong Ye 		if (!vdev)
335edbed86dSXiaolong Ye 			continue;
33671296e7bSMaxime Coquelin 		if (rte_vdpa_get_queue_num(vdev, &queue_num) < 0) {
337edbed86dSXiaolong Ye 			RTE_LOG(ERR, VDPA,
338edbed86dSXiaolong Ye 				"failed to get vdpa queue number "
339ec5ecd7eSDavid Marchand 				"for device %s.\n", rte_dev_name(dev));
340edbed86dSXiaolong Ye 			continue;
341edbed86dSXiaolong Ye 		}
34271296e7bSMaxime Coquelin 		if (rte_vdpa_get_features(vdev, &features) < 0) {
343edbed86dSXiaolong Ye 			RTE_LOG(ERR, VDPA,
344edbed86dSXiaolong Ye 				"failed to get vdpa features "
345ec5ecd7eSDavid Marchand 				"for device %s.\n", rte_dev_name(dev));
346edbed86dSXiaolong Ye 			continue;
347edbed86dSXiaolong Ye 		}
3480f700f90SMaxime Coquelin 		cmdline_printf(cl, "%s\t\t%" PRIu32 "\t\t0x%" PRIx64 "\n",
349ec5ecd7eSDavid Marchand 			rte_dev_name(dev), queue_num, features);
350edbed86dSXiaolong Ye 	}
351edbed86dSXiaolong Ye }
352edbed86dSXiaolong Ye 
cmd_create_parsed(void * parsed_result,struct cmdline * cl,__rte_unused void * data)353*1b2798eeSBruce Richardson void cmd_create_parsed(void *parsed_result,
354edbed86dSXiaolong Ye 		struct cmdline *cl,
355f2fc83b4SThomas Monjalon 		__rte_unused void *data)
356edbed86dSXiaolong Ye {
3570f700f90SMaxime Coquelin 	struct rte_vdpa_device *dev;
358edbed86dSXiaolong Ye 	struct cmd_create_result *res = parsed_result;
359edbed86dSXiaolong Ye 
360edbed86dSXiaolong Ye 	rte_strscpy(vports[devcnt].ifname, res->socket_path, MAX_PATH_LEN);
3610f700f90SMaxime Coquelin 	dev = rte_vdpa_find_device_by_name(res->bdf);
3620f700f90SMaxime Coquelin 	if (dev == NULL) {
36338f8ab0bSMaxime Coquelin 		cmdline_printf(cl, "Unable to find vdpa device id for %s.\n",
36438f8ab0bSMaxime Coquelin 				res->bdf);
365edbed86dSXiaolong Ye 		return;
366edbed86dSXiaolong Ye 	}
367edbed86dSXiaolong Ye 
3680f700f90SMaxime Coquelin 	vports[devcnt].dev = dev;
369edbed86dSXiaolong Ye 
370edbed86dSXiaolong Ye 	if (start_vdpa(&vports[devcnt]) == 0)
371edbed86dSXiaolong Ye 		devcnt++;
372edbed86dSXiaolong Ye }
373edbed86dSXiaolong Ye 
cmd_stats_parsed(void * parsed_result,struct cmdline * cl,__rte_unused void * data)374*1b2798eeSBruce Richardson void cmd_stats_parsed(void *parsed_result, struct cmdline *cl,
3756505865aSMatan Azrad 				    __rte_unused void *data)
3766505865aSMatan Azrad {
3776505865aSMatan Azrad 	struct cmd_stats_result *res = parsed_result;
3780f700f90SMaxime Coquelin 	struct rte_vdpa_device *vdev = rte_vdpa_find_device_by_name(res->bdf);
3796505865aSMatan Azrad 	struct vdpa_port *vport = NULL;
3806505865aSMatan Azrad 	uint32_t first, last;
3816505865aSMatan Azrad 	int i;
3826505865aSMatan Azrad 
3836505865aSMatan Azrad 	if (!vdev) {
3840f700f90SMaxime Coquelin 		RTE_LOG(ERR, VDPA, "Invalid device: %s.\n",
3850f700f90SMaxime Coquelin 			res->bdf);
3866505865aSMatan Azrad 		return;
3876505865aSMatan Azrad 	}
388fbe6be6eSMaxime Coquelin 	for (i = 0; i < RTE_MIN(MAX_VDPA_SAMPLE_PORTS, devcnt); i++) {
3890f700f90SMaxime Coquelin 		if (vports[i].dev == vdev) {
3906505865aSMatan Azrad 			vport = &vports[i];
3916505865aSMatan Azrad 			break;
3926505865aSMatan Azrad 		}
3936505865aSMatan Azrad 	}
3946505865aSMatan Azrad 	if (!vport) {
3950f700f90SMaxime Coquelin 		RTE_LOG(ERR, VDPA, "Device %s was not created.\n", res->bdf);
3966505865aSMatan Azrad 		return;
3976505865aSMatan Azrad 	}
3986505865aSMatan Azrad 	if (res->qid == 0xFFFF) {
3996505865aSMatan Azrad 		first = 0;
4006505865aSMatan Azrad 		last = rte_vhost_get_vring_num(vport->vid);
4016505865aSMatan Azrad 		if (last == 0) {
4026505865aSMatan Azrad 			RTE_LOG(ERR, VDPA, "Failed to get num of actual virtqs"
4030f700f90SMaxime Coquelin 				" for device %s.\n", res->bdf);
4046505865aSMatan Azrad 			return;
4056505865aSMatan Azrad 		}
4066505865aSMatan Azrad 		last--;
4076505865aSMatan Azrad 	} else {
4086505865aSMatan Azrad 		first = res->qid;
4096505865aSMatan Azrad 		last = res->qid;
4106505865aSMatan Azrad 	}
4116505865aSMatan Azrad 	if (!vport->stats_names) {
4120f700f90SMaxime Coquelin 		vport->stats_n = rte_vdpa_get_stats_names(vport->dev, NULL, 0);
4136505865aSMatan Azrad 		if (vport->stats_n <= 0) {
4146505865aSMatan Azrad 			RTE_LOG(ERR, VDPA, "Failed to get names number of "
4150f700f90SMaxime Coquelin 				"device %s stats.\n", res->bdf);
4166505865aSMatan Azrad 			return;
4176505865aSMatan Azrad 		}
4186505865aSMatan Azrad 		vport->stats_names = rte_zmalloc(NULL,
4196505865aSMatan Azrad 			(sizeof(*vport->stats_names) + sizeof(*vport->stats)) *
4206505865aSMatan Azrad 							vport->stats_n, 0);
4216505865aSMatan Azrad 		if (!vport->stats_names) {
4226505865aSMatan Azrad 			RTE_LOG(ERR, VDPA, "Failed to allocate memory for stat"
4230f700f90SMaxime Coquelin 				" names of device %s.\n", res->bdf);
4246505865aSMatan Azrad 			return;
4256505865aSMatan Azrad 		}
4260f700f90SMaxime Coquelin 		i = rte_vdpa_get_stats_names(vport->dev, vport->stats_names,
4276505865aSMatan Azrad 						vport->stats_n);
4286505865aSMatan Azrad 		if (vport->stats_n != i) {
4290f700f90SMaxime Coquelin 			RTE_LOG(ERR, VDPA, "Failed to get names of device %s "
4300f700f90SMaxime Coquelin 				"stats.\n", res->bdf);
4316505865aSMatan Azrad 			return;
4326505865aSMatan Azrad 		}
4336505865aSMatan Azrad 		vport->stats = (struct rte_vdpa_stat *)
4346505865aSMatan Azrad 					(vport->stats_names + vport->stats_n);
4356505865aSMatan Azrad 	}
4360f700f90SMaxime Coquelin 	cmdline_printf(cl, "\nDevice %s:\n", res->bdf);
4376505865aSMatan Azrad 	for (; first <= last; first++) {
4386505865aSMatan Azrad 		memset(vport->stats, 0, sizeof(*vport->stats) * vport->stats_n);
4390f700f90SMaxime Coquelin 		if (rte_vdpa_get_stats(vport->dev, (int)first, vport->stats,
4406505865aSMatan Azrad 					vport->stats_n) <= 0) {
4416505865aSMatan Azrad 			RTE_LOG(ERR, VDPA, "Failed to get vdpa queue statistics"
4420f700f90SMaxime Coquelin 				" for device %s qid %d.\n", res->bdf,
4436505865aSMatan Azrad 				(int)first);
4446505865aSMatan Azrad 			return;
4456505865aSMatan Azrad 		}
4466505865aSMatan Azrad 		cmdline_printf(cl, "\tVirtq %" PRIu32 ":\n", first);
4476505865aSMatan Azrad 		for (i = 0; i < vport->stats_n; ++i) {
4486505865aSMatan Azrad 			cmdline_printf(cl, "\t\t%-*s %-16" PRIu64 "\n",
4496505865aSMatan Azrad 				RTE_VDPA_STATS_NAME_SIZE,
4506505865aSMatan Azrad 				vport->stats_names[vport->stats[i].id].name,
4516505865aSMatan Azrad 				vport->stats[i].value);
4526505865aSMatan Azrad 		}
4536505865aSMatan Azrad 	}
4546505865aSMatan Azrad }
4556505865aSMatan Azrad 
cmd_quit_parsed(__rte_unused void * parsed_result,struct cmdline * cl,__rte_unused void * data)456*1b2798eeSBruce Richardson void cmd_quit_parsed(__rte_unused void *parsed_result,
457edbed86dSXiaolong Ye 		struct cmdline *cl,
458f2fc83b4SThomas Monjalon 		__rte_unused void *data)
459edbed86dSXiaolong Ye {
460edbed86dSXiaolong Ye 	vdpa_sample_quit();
461edbed86dSXiaolong Ye 	cmdline_quit(cl);
462edbed86dSXiaolong Ye }
463edbed86dSXiaolong Ye 
464edbed86dSXiaolong Ye int
main(int argc,char * argv[])465edbed86dSXiaolong Ye main(int argc, char *argv[])
466edbed86dSXiaolong Ye {
467edbed86dSXiaolong Ye 	char ch;
468edbed86dSXiaolong Ye 	int ret;
469edbed86dSXiaolong Ye 	struct cmdline *cl;
4700f700f90SMaxime Coquelin 	struct rte_vdpa_device *vdev;
47138f8ab0bSMaxime Coquelin 	struct rte_device *dev;
47238f8ab0bSMaxime Coquelin 	struct rte_dev_iterator dev_iter;
473edbed86dSXiaolong Ye 
474edbed86dSXiaolong Ye 	ret = rte_eal_init(argc, argv);
475edbed86dSXiaolong Ye 	if (ret < 0)
476edbed86dSXiaolong Ye 		rte_exit(EXIT_FAILURE, "eal init failed\n");
477edbed86dSXiaolong Ye 	argc -= ret;
478edbed86dSXiaolong Ye 	argv += ret;
479edbed86dSXiaolong Ye 
480edbed86dSXiaolong Ye 	signal(SIGINT, signal_handler);
481edbed86dSXiaolong Ye 	signal(SIGTERM, signal_handler);
482edbed86dSXiaolong Ye 
483edbed86dSXiaolong Ye 	ret = parse_args(argc, argv);
484edbed86dSXiaolong Ye 	if (ret < 0)
485edbed86dSXiaolong Ye 		rte_exit(EXIT_FAILURE, "invalid argument\n");
486edbed86dSXiaolong Ye 
487edbed86dSXiaolong Ye 	if (interactive == 1) {
488edbed86dSXiaolong Ye 		cl = cmdline_stdin_new(main_ctx, "vdpa> ");
489edbed86dSXiaolong Ye 		if (cl == NULL)
490edbed86dSXiaolong Ye 			rte_panic("Cannot create cmdline instance\n");
491edbed86dSXiaolong Ye 		cmdline_interact(cl);
492edbed86dSXiaolong Ye 		cmdline_stdin_exit(cl);
493edbed86dSXiaolong Ye 	} else {
49438f8ab0bSMaxime Coquelin 		RTE_DEV_FOREACH(dev, "class=vdpa", &dev_iter) {
495ec5ecd7eSDavid Marchand 			vdev = rte_vdpa_find_device_by_name(rte_dev_name(dev));
4960f700f90SMaxime Coquelin 			if (vdev == NULL) {
4970f700f90SMaxime Coquelin 				rte_panic("Failed to find vDPA dev for %s\n",
498ec5ecd7eSDavid Marchand 						rte_dev_name(dev));
49938f8ab0bSMaxime Coquelin 			}
5000f700f90SMaxime Coquelin 			vports[devcnt].dev = vdev;
50138f8ab0bSMaxime Coquelin 			snprintf(vports[devcnt].ifname, MAX_PATH_LEN, "%s%d",
50238f8ab0bSMaxime Coquelin 					iface, devcnt);
503edbed86dSXiaolong Ye 
50438f8ab0bSMaxime Coquelin 			start_vdpa(&vports[devcnt]);
50538f8ab0bSMaxime Coquelin 			devcnt++;
506edbed86dSXiaolong Ye 		}
507edbed86dSXiaolong Ye 
508edbed86dSXiaolong Ye 		printf("enter \'q\' to quit\n");
509edbed86dSXiaolong Ye 		while (scanf("%c", &ch)) {
510edbed86dSXiaolong Ye 			if (ch == 'q')
511edbed86dSXiaolong Ye 				break;
512edbed86dSXiaolong Ye 			while (ch != '\n') {
513edbed86dSXiaolong Ye 				if (scanf("%c", &ch))
514edbed86dSXiaolong Ye 					printf("%c", ch);
515edbed86dSXiaolong Ye 			}
516edbed86dSXiaolong Ye 			printf("enter \'q\' to quit\n");
517edbed86dSXiaolong Ye 		}
518edbed86dSXiaolong Ye 		vdpa_sample_quit();
519edbed86dSXiaolong Ye 	}
520edbed86dSXiaolong Ye 
52110aa3757SChengchang Tang 	/* clean up the EAL */
52210aa3757SChengchang Tang 	rte_eal_cleanup();
52310aa3757SChengchang Tang 
524edbed86dSXiaolong Ye 	return 0;
525edbed86dSXiaolong Ye }
526