1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17#include <common.h>
18#include <command.h>
19#include <ide.h>
20#include <memalign.h>
21#include "part_dos.h"
22
23#ifdef HAVE_BLOCK_DEVICE
24
25#define DOS_PART_DEFAULT_SECTOR 512
26
27
28
29static inline unsigned int le32_to_int(unsigned char *le32)
30{
31 return ((le32[3] << 24) +
32 (le32[2] << 16) +
33 (le32[1] << 8) +
34 le32[0]
35 );
36}
37
38static inline int is_extended(int part_type)
39{
40 return (part_type == 0x5 ||
41 part_type == 0xf ||
42 part_type == 0x85);
43}
44
45static inline int is_bootable(dos_partition_t *p)
46{
47 return (p->sys_ind == 0xef) || (p->boot_ind == 0x80);
48}
49
50static void print_one_part(dos_partition_t *p, lbaint_t ext_part_sector,
51 int part_num, unsigned int disksig)
52{
53 lbaint_t lba_start = ext_part_sector + le32_to_int (p->start4);
54 lbaint_t lba_size = le32_to_int (p->size4);
55
56 printf("%3d\t%-10" LBAFlength "u\t%-10" LBAFlength
57 "u\t%08x-%02x\t%02x%s%s\n",
58 part_num, lba_start, lba_size, disksig, part_num, p->sys_ind,
59 (is_extended(p->sys_ind) ? " Extd" : ""),
60 (is_bootable(p) ? " Boot" : ""));
61}
62
63static int test_block_type(unsigned char *buffer)
64{
65 int slot;
66 struct dos_partition *p;
67
68 if((buffer[DOS_PART_MAGIC_OFFSET + 0] != 0x55) ||
69 (buffer[DOS_PART_MAGIC_OFFSET + 1] != 0xaa) ) {
70 return (-1);
71 }
72 p = (struct dos_partition *)&buffer[DOS_PART_TBL_OFFSET];
73 for (slot = 0; slot < 3; slot++) {
74 if (p->boot_ind != 0 && p->boot_ind != 0x80) {
75 if (!slot &&
76 (strncmp((char *)&buffer[DOS_PBR_FSTYPE_OFFSET],
77 "FAT", 3) == 0 ||
78 strncmp((char *)&buffer[DOS_PBR32_FSTYPE_OFFSET],
79 "FAT32", 5) == 0)) {
80 return DOS_PBR;
81 } else {
82 return -1;
83 }
84 }
85 }
86 return DOS_MBR;
87}
88
89
90static int part_test_dos(struct blk_desc *dev_desc)
91{
92#ifndef CONFIG_SPL_BUILD
93 ALLOC_CACHE_ALIGN_BUFFER(legacy_mbr, mbr, dev_desc->blksz);
94
95 if (blk_dread(dev_desc, 0, 1, (ulong *)mbr) != 1)
96 return -1;
97
98 if (test_block_type((unsigned char *)mbr) != DOS_MBR)
99 return -1;
100
101 if (dev_desc->sig_type == SIG_TYPE_NONE &&
102 mbr->unique_mbr_signature != 0) {
103 dev_desc->sig_type = SIG_TYPE_MBR;
104 dev_desc->mbr_sig = mbr->unique_mbr_signature;
105 }
106#else
107 ALLOC_CACHE_ALIGN_BUFFER(unsigned char, buffer, dev_desc->blksz);
108
109 if (blk_dread(dev_desc, 0, 1, (ulong *)buffer) != 1)
110 return -1;
111
112 if (test_block_type(buffer) != DOS_MBR)
113 return -1;
114#endif
115
116 return 0;
117}
118
119
120
121static void print_partition_extended(struct blk_desc *dev_desc,
122 lbaint_t ext_part_sector,
123 lbaint_t relative,
124 int part_num, unsigned int disksig)
125{
126 ALLOC_CACHE_ALIGN_BUFFER(unsigned char, buffer, dev_desc->blksz);
127 dos_partition_t *pt;
128 int i;
129
130 if (blk_dread(dev_desc, ext_part_sector, 1, (ulong *)buffer) != 1) {
131 printf ("** Can't read partition table on %d:" LBAFU " **\n",
132 dev_desc->devnum, ext_part_sector);
133 return;
134 }
135 i=test_block_type(buffer);
136 if (i != DOS_MBR) {
137 printf ("bad MBR sector signature 0x%02x%02x\n",
138 buffer[DOS_PART_MAGIC_OFFSET],
139 buffer[DOS_PART_MAGIC_OFFSET + 1]);
140 return;
141 }
142
143 if (!ext_part_sector)
144 disksig = le32_to_int(&buffer[DOS_PART_DISKSIG_OFFSET]);
145
146
147 pt = (dos_partition_t *) (buffer + DOS_PART_TBL_OFFSET);
148 for (i = 0; i < 4; i++, pt++) {
149
150
151
152
153
154 if ((pt->sys_ind != 0) &&
155 (ext_part_sector == 0 || !is_extended (pt->sys_ind)) ) {
156 print_one_part(pt, ext_part_sector, part_num, disksig);
157 }
158
159
160 if ((ext_part_sector == 0) ||
161 (pt->sys_ind != 0 && !is_extended (pt->sys_ind)) ) {
162 part_num++;
163 }
164 }
165
166
167 pt = (dos_partition_t *) (buffer + DOS_PART_TBL_OFFSET);
168 for (i = 0; i < 4; i++, pt++) {
169 if (is_extended (pt->sys_ind)) {
170 lbaint_t lba_start
171 = le32_to_int (pt->start4) + relative;
172
173 print_partition_extended(dev_desc, lba_start,
174 ext_part_sector == 0 ? lba_start : relative,
175 part_num, disksig);
176 }
177 }
178
179 return;
180}
181
182
183
184
185static int part_get_info_extended(struct blk_desc *dev_desc,
186 lbaint_t ext_part_sector, lbaint_t relative,
187 int part_num, int which_part,
188 disk_partition_t *info, unsigned int disksig)
189{
190 ALLOC_CACHE_ALIGN_BUFFER(unsigned char, buffer, dev_desc->blksz);
191 dos_partition_t *pt;
192 int i;
193 int dos_type;
194
195 if (blk_dread(dev_desc, ext_part_sector, 1, (ulong *)buffer) != 1) {
196 printf ("** Can't read partition table on %d:" LBAFU " **\n",
197 dev_desc->devnum, ext_part_sector);
198 return -1;
199 }
200 if (buffer[DOS_PART_MAGIC_OFFSET] != 0x55 ||
201 buffer[DOS_PART_MAGIC_OFFSET + 1] != 0xaa) {
202 printf ("bad MBR sector signature 0x%02x%02x\n",
203 buffer[DOS_PART_MAGIC_OFFSET],
204 buffer[DOS_PART_MAGIC_OFFSET + 1]);
205 return -1;
206 }
207
208#if CONFIG_IS_ENABLED(PARTITION_UUIDS)
209 if (!ext_part_sector)
210 disksig = le32_to_int(&buffer[DOS_PART_DISKSIG_OFFSET]);
211#endif
212
213
214 pt = (dos_partition_t *) (buffer + DOS_PART_TBL_OFFSET);
215 for (i = 0; i < 4; i++, pt++) {
216
217
218
219
220 if (((pt->boot_ind & ~0x80) == 0) &&
221 (pt->sys_ind != 0) &&
222 (part_num == which_part) &&
223 (ext_part_sector == 0 || is_extended(pt->sys_ind) == 0)) {
224 info->blksz = DOS_PART_DEFAULT_SECTOR;
225 info->start = (lbaint_t)(ext_part_sector +
226 le32_to_int(pt->start4));
227 info->size = (lbaint_t)le32_to_int(pt->size4);
228 part_set_generic_name(dev_desc, part_num,
229 (char *)info->name);
230
231 strcpy((char *)info->type, "U-Boot");
232 info->bootable = is_bootable(pt);
233#if CONFIG_IS_ENABLED(PARTITION_UUIDS)
234 sprintf(info->uuid, "%08x-%02x", disksig, part_num);
235#endif
236 info->sys_ind = pt->sys_ind;
237 return 0;
238 }
239
240
241 if ((ext_part_sector == 0) ||
242 (pt->sys_ind != 0 && !is_extended (pt->sys_ind)) ) {
243 part_num++;
244 }
245 }
246
247
248 pt = (dos_partition_t *) (buffer + DOS_PART_TBL_OFFSET);
249 for (i = 0; i < 4; i++, pt++) {
250 if (is_extended (pt->sys_ind)) {
251 lbaint_t lba_start
252 = le32_to_int (pt->start4) + relative;
253
254 return part_get_info_extended(dev_desc, lba_start,
255 ext_part_sector == 0 ? lba_start : relative,
256 part_num, which_part, info, disksig);
257 }
258 }
259
260
261 dos_type = test_block_type(buffer);
262
263 if (dos_type == DOS_PBR) {
264 info->start = 0;
265 info->size = dev_desc->lba;
266 info->blksz = DOS_PART_DEFAULT_SECTOR;
267 info->bootable = 0;
268 strcpy((char *)info->type, "U-Boot");
269#if CONFIG_IS_ENABLED(PARTITION_UUIDS)
270 info->uuid[0] = 0;
271#endif
272 return 0;
273 }
274
275 return -1;
276}
277
278void part_print_dos(struct blk_desc *dev_desc)
279{
280 printf("Part\tStart Sector\tNum Sectors\tUUID\t\tType\n");
281 print_partition_extended(dev_desc, 0, 0, 1, 0);
282}
283
284int part_get_info_dos(struct blk_desc *dev_desc, int part,
285 disk_partition_t *info)
286{
287 return part_get_info_extended(dev_desc, 0, 0, 1, part, info, 0);
288}
289
290int is_valid_dos_buf(void *buf)
291{
292 return test_block_type(buf) == DOS_MBR ? 0 : -1;
293}
294
295int write_mbr_partition(struct blk_desc *dev_desc, void *buf)
296{
297 if (is_valid_dos_buf(buf))
298 return -1;
299
300
301 if (blk_dwrite(dev_desc, 0, 1, buf) != 1) {
302 printf("%s: failed writing '%s' (1 blks at 0x0)\n",
303 __func__, "MBR");
304 return 1;
305 }
306
307 return 0;
308}
309
310U_BOOT_PART_TYPE(dos) = {
311 .name = "DOS",
312 .part_type = PART_TYPE_DOS,
313 .max_entries = DOS_ENTRY_NUMBERS,
314 .get_info = part_get_info_ptr(part_get_info_dos),
315 .print = part_print_ptr(part_print_dos),
316 .test = part_test_dos,
317};
318
319#endif
320
