1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18#include <common.h>
19#include <config.h>
20#include <command.h>
21#include <part.h>
22#include <vsprintf.h>
23
24enum cmd_part_info {
25 CMD_PART_INFO_START = 0,
26 CMD_PART_INFO_SIZE,
27};
28
29static int do_part_uuid(int argc, char * const argv[])
30{
31 int part;
32 struct blk_desc *dev_desc;
33 disk_partition_t info;
34
35 if (argc < 2)
36 return CMD_RET_USAGE;
37 if (argc > 3)
38 return CMD_RET_USAGE;
39
40 part = blk_get_device_part_str(argv[0], argv[1], &dev_desc, &info, 0);
41 if (part < 0)
42 return 1;
43
44 if (argc > 2)
45 env_set(argv[2], info.uuid);
46 else
47 printf("%s\n", info.uuid);
48
49 return 0;
50}
51
52static int do_part_list(int argc, char * const argv[])
53{
54 int ret;
55 struct blk_desc *desc;
56 char *var = NULL;
57 bool bootable = false;
58 int i;
59
60 if (argc < 2)
61 return CMD_RET_USAGE;
62
63 if (argc > 2) {
64 for (i = 2; i < argc ; i++) {
65 if (argv[i][0] == '-') {
66 if (!strcmp(argv[i], "-bootable")) {
67 bootable = true;
68 } else {
69 printf("Unknown option %s\n", argv[i]);
70 return CMD_RET_USAGE;
71 }
72 } else {
73 var = argv[i];
74 break;
75 }
76 }
77
78
79
80 if (argc != i + 1)
81 return CMD_RET_USAGE;
82 }
83
84 ret = blk_get_device_by_str(argv[0], argv[1], &desc);
85 if (ret < 0)
86 return 1;
87
88 if (var != NULL) {
89 int p;
90 char str[512] = { '\0', };
91 disk_partition_t info;
92
93 for (p = 1; p < MAX_SEARCH_PARTITIONS; p++) {
94 char t[5];
95 int r = part_get_info(desc, p, &info);
96
97 if (r != 0)
98 continue;
99
100 if (bootable && !info.bootable)
101 continue;
102
103 sprintf(t, "%s%x", str[0] ? " " : "", p);
104 strcat(str, t);
105 }
106 env_set(var, str);
107 return 0;
108 }
109
110 part_print(desc);
111
112 return 0;
113}
114
115static int do_part_info(int argc, char * const argv[], enum cmd_part_info param)
116{
117 struct blk_desc *desc;
118 disk_partition_t info;
119 char buf[512] = { 0 };
120 char *endp;
121 int part;
122 int err;
123 int ret;
124
125 if (argc < 3)
126 return CMD_RET_USAGE;
127 if (argc > 4)
128 return CMD_RET_USAGE;
129
130 ret = blk_get_device_by_str(argv[0], argv[1], &desc);
131 if (ret < 0)
132 return 1;
133
134 part = simple_strtoul(argv[2], &endp, 0);
135 if (*endp == '\0') {
136 err = part_get_info(desc, part, &info);
137 if (err)
138 return 1;
139 } else {
140 part = part_get_info_by_name(desc, argv[2], &info);
141 if (part == -1)
142 return 1;
143 }
144
145 switch (param) {
146 case CMD_PART_INFO_START:
147 snprintf(buf, sizeof(buf), LBAF, info.start);
148 break;
149 case CMD_PART_INFO_SIZE:
150 snprintf(buf, sizeof(buf), LBAF, info.size);
151 break;
152 default:
153 printf("** Unknown cmd_part_info value: %d\n", param);
154 return 1;
155 }
156
157 if (argc > 3)
158 env_set(argv[3], buf);
159 else
160 printf("%s\n", buf);
161
162 return 0;
163}
164
165static int do_part_start(int argc, char * const argv[])
166{
167 return do_part_info(argc, argv, CMD_PART_INFO_START);
168}
169
170static int do_part_size(int argc, char * const argv[])
171{
172 return do_part_info(argc, argv, CMD_PART_INFO_SIZE);
173}
174
175static int do_part(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
176{
177 if (argc < 2)
178 return CMD_RET_USAGE;
179
180 if (!strcmp(argv[1], "uuid"))
181 return do_part_uuid(argc - 2, argv + 2);
182 else if (!strcmp(argv[1], "list"))
183 return do_part_list(argc - 2, argv + 2);
184 else if (!strcmp(argv[1], "start"))
185 return do_part_start(argc - 2, argv + 2);
186 else if (!strcmp(argv[1], "size"))
187 return do_part_size(argc - 2, argv + 2);
188
189 return CMD_RET_USAGE;
190}
191
192U_BOOT_CMD(
193 part, CONFIG_SYS_MAXARGS, 1, do_part,
194 "disk partition related commands",
195 "uuid <interface> <dev>:<part>\n"
196 " - print partition UUID\n"
197 "part uuid <interface> <dev>:<part> <varname>\n"
198 " - set environment variable to partition UUID\n"
199 "part list <interface> <dev>\n"
200 " - print a device's partition table\n"
201 "part list <interface> <dev> [flags] <varname>\n"
202 " - set environment variable to the list of partitions\n"
203 " flags can be -bootable (list only bootable partitions)\n"
204 "part start <interface> <dev> <part> <varname>\n"
205 " - set environment variable to the start of the partition (in blocks)\n"
206 " part can be either partition number or partition name\n"
207 "part size <interface> <dev> <part> <varname>\n"
208 " - set environment variable to the size of the partition (in blocks)\n"
209 " part can be either partition number or partition name"
210);
211
