1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19#include <common.h>
20#include <config.h>
21#include <command.h>
22#include <part.h>
23#include <vsprintf.h>
24
25#ifndef CONFIG_PARTITION_UUIDS
26#error CONFIG_PARTITION_UUIDS must be enabled for CONFIG_CMD_PART to be enabled
27#endif
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 setenv(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 < 128; 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 setenv(var, str);
107 return 0;
108 }
109
110 part_print(desc);
111
112 return 0;
113}
114
115static int do_part_start(int argc, char * const argv[])
116{
117 struct blk_desc *desc;
118 disk_partition_t info;
119 char buf[512] = { 0 };
120 int part;
121 int err;
122 int ret;
123
124 if (argc < 3)
125 return CMD_RET_USAGE;
126 if (argc > 4)
127 return CMD_RET_USAGE;
128
129 part = simple_strtoul(argv[2], NULL, 0);
130
131 ret = blk_get_device_by_str(argv[0], argv[1], &desc);
132 if (ret < 0)
133 return 1;
134
135 err = part_get_info(desc, part, &info);
136 if (err)
137 return 1;
138
139 snprintf(buf, sizeof(buf), LBAF, info.start);
140
141 if (argc > 3)
142 setenv(argv[3], buf);
143 else
144 printf("%s\n", buf);
145
146 return 0;
147}
148
149static int do_part_size(int argc, char * const argv[])
150{
151 struct blk_desc *desc;
152 disk_partition_t info;
153 char buf[512] = { 0 };
154 int part;
155 int err;
156 int ret;
157
158 if (argc < 3)
159 return CMD_RET_USAGE;
160 if (argc > 4)
161 return CMD_RET_USAGE;
162
163 part = simple_strtoul(argv[2], NULL, 0);
164
165 ret = blk_get_device_by_str(argv[0], argv[1], &desc);
166 if (ret < 0)
167 return 1;
168
169 err = part_get_info(desc, part, &info);
170 if (err)
171 return 1;
172
173 snprintf(buf, sizeof(buf), LBAF, info.size);
174
175 if (argc > 3)
176 setenv(argv[3], buf);
177 else
178 printf("%s\n", buf);
179
180 return 0;
181}
182
183static int do_part(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
184{
185 if (argc < 2)
186 return CMD_RET_USAGE;
187
188 if (!strcmp(argv[1], "uuid"))
189 return do_part_uuid(argc - 2, argv + 2);
190 else if (!strcmp(argv[1], "list"))
191 return do_part_list(argc - 2, argv + 2);
192 else if (!strcmp(argv[1], "start"))
193 return do_part_start(argc - 2, argv + 2);
194 else if (!strcmp(argv[1], "size"))
195 return do_part_size(argc - 2, argv + 2);
196
197 return CMD_RET_USAGE;
198}
199
200U_BOOT_CMD(
201 part, CONFIG_SYS_MAXARGS, 1, do_part,
202 "disk partition related commands",
203 "uuid <interface> <dev>:<part>\n"
204 " - print partition UUID\n"
205 "part uuid <interface> <dev>:<part> <varname>\n"
206 " - set environment variable to partition UUID\n"
207 "part list <interface> <dev>\n"
208 " - print a device's partition table\n"
209 "part list <interface> <dev> [flags] <varname>\n"
210 " - set environment variable to the list of partitions\n"
211 " flags can be -bootable (list only bootable partitions)\n"
212 "part start <interface> <dev> <part> <varname>\n"
213 " - set environment variable to the start of the partition (in blocks)\n"
214 "part size <interface> <dev> <part> <varname>\n"
215 " - set environment variable to the size of the partition (in blocks)"
216);
217
