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