1 // SPDX-License-Identifier: GPL-2.0+
2 /*
3  * Copyright (C) 2018, Tuomas Tynkkynen <tuomas.tynkkynen@iki.fi>
4  * Copyright (C) 2018, Bin Meng <bmeng.cn@gmail.com>
5  */
6 
7 #include <common.h>
8 #include <blk.h>
9 #include <command.h>
10 #include <dm.h>
11 #include <virtio_types.h>
12 #include <virtio.h>
13 
14 static int virtio_curr_dev;
15 
do_virtio(struct cmd_tbl * cmdtp,int flag,int argc,char * const argv[])16 static int do_virtio(struct cmd_tbl *cmdtp, int flag, int argc,
17 		     char *const argv[])
18 {
19 	if (argc == 2 && !strcmp(argv[1], "scan")) {
20 		/*
21 		 * make sure all virtio devices are enumerated.
22 		 * Do the same as virtio_init(), but also call
23 		 * device_probe() for children (i.e. virtio devices)
24 		 */
25 		struct udevice *bus, *child;
26 
27 		uclass_first_device(UCLASS_VIRTIO, &bus);
28 		if (!bus)
29 			return CMD_RET_FAILURE;
30 
31 		while (bus) {
32 			device_foreach_child_probe(child, bus)
33 				;
34 			uclass_next_device(&bus);
35 		}
36 
37 		return CMD_RET_SUCCESS;
38 	}
39 
40 	return blk_common_cmd(argc, argv, UCLASS_VIRTIO, &virtio_curr_dev);
41 }
42 
43 U_BOOT_CMD(
44 	virtio, 8, 1, do_virtio,
45 	"virtio block devices sub-system",
46 	"scan - initialize virtio bus\n"
47 	"virtio info - show all available virtio block devices\n"
48 	"virtio device [dev] - show or set current virtio block device\n"
49 	"virtio part [dev] - print partition table of one or all virtio block devices\n"
50 	"virtio read addr blk# cnt - read `cnt' blocks starting at block\n"
51 	"     `blk#' to memory address `addr'\n"
52 	"virtio write addr blk# cnt - write `cnt' blocks starting at block\n"
53 	"     `blk#' from memory address `addr'"
54 );
55