1 // SPDX-License-Identifier: GPL-2.0+
2 /*
3  * Handling of common block commands
4  *
5  * Copyright (c) 2017 Google, Inc
6  *
7  * (C) Copyright 2000-2011
8  * Wolfgang Denk, DENX Software Engineering, wd@denx.de.
9  */
10 
11 #include <common.h>
12 #include <blk.h>
13 #include <command.h>
14 #include <mapmem.h>
15 
blk_common_cmd(int argc,char * const argv[],enum uclass_id uclass_id,int * cur_devnump)16 int blk_common_cmd(int argc, char *const argv[], enum uclass_id uclass_id,
17 		   int *cur_devnump)
18 {
19 	const char *if_name = blk_get_uclass_name(uclass_id);
20 
21 	switch (argc) {
22 	case 0:
23 	case 1:
24 		return CMD_RET_USAGE;
25 	case 2:
26 		if (strncmp(argv[1], "inf", 3) == 0) {
27 			blk_list_devices(uclass_id);
28 			return 0;
29 		} else if (strncmp(argv[1], "dev", 3) == 0) {
30 			if (blk_print_device_num(uclass_id, *cur_devnump)) {
31 				printf("\nno %s devices available\n", if_name);
32 				return CMD_RET_FAILURE;
33 			}
34 			return 0;
35 		} else if (strncmp(argv[1], "part", 4) == 0) {
36 			if (blk_list_part(uclass_id))
37 				printf("\nno %s partition table available\n",
38 				       if_name);
39 			return 0;
40 		}
41 		return CMD_RET_USAGE;
42 	case 3:
43 		if (strncmp(argv[1], "dev", 3) == 0) {
44 			int dev = (int)dectoul(argv[2], NULL);
45 
46 			if (!blk_show_device(uclass_id, dev)) {
47 				*cur_devnump = dev;
48 				printf("... is now current device\n");
49 			} else {
50 				return CMD_RET_FAILURE;
51 			}
52 			return 0;
53 		} else if (strncmp(argv[1], "part", 4) == 0) {
54 			int dev = (int)dectoul(argv[2], NULL);
55 
56 			if (blk_print_part_devnum(uclass_id, dev)) {
57 				printf("\n%s device %d not available\n",
58 				       if_name, dev);
59 				return CMD_RET_FAILURE;
60 			}
61 			return 0;
62 		}
63 		return CMD_RET_USAGE;
64 
65 	default: /* at least 4 args */
66 		if (strcmp(argv[1], "read") == 0) {
67 			phys_addr_t paddr = hextoul(argv[2], NULL);
68 			lbaint_t blk = hextoul(argv[3], NULL);
69 			ulong cnt = hextoul(argv[4], NULL);
70 			void *vaddr;
71 			ulong n;
72 
73 			printf("\n%s read: device %d block # "LBAFU", count %lu ... ",
74 			       if_name, *cur_devnump, blk, cnt);
75 
76 			vaddr = map_sysmem(paddr, 512 * cnt);
77 			n = blk_read_devnum(uclass_id, *cur_devnump, blk, cnt,
78 					    vaddr);
79 			unmap_sysmem(vaddr);
80 
81 			printf("%ld blocks read: %s\n", n,
82 			       n == cnt ? "OK" : "ERROR");
83 			return n == cnt ? 0 : 1;
84 		} else if (strcmp(argv[1], "write") == 0) {
85 			phys_addr_t paddr = hextoul(argv[2], NULL);
86 			lbaint_t blk = hextoul(argv[3], NULL);
87 			ulong cnt = hextoul(argv[4], NULL);
88 			void *vaddr;
89 			ulong n;
90 
91 			printf("\n%s write: device %d block # "LBAFU", count %lu ... ",
92 			       if_name, *cur_devnump, blk, cnt);
93 
94 			vaddr = map_sysmem(paddr, 512 * cnt);
95 			n = blk_write_devnum(uclass_id, *cur_devnump, blk, cnt,
96 					     vaddr);
97 			unmap_sysmem(vaddr);
98 
99 			printf("%ld blocks written: %s\n", n,
100 			       n == cnt ? "OK" : "ERROR");
101 			return n == cnt ? 0 : 1;
102 		} else {
103 			return CMD_RET_USAGE;
104 		}
105 	}
106 }
107