1 // SPDX-License-Identifier: GPL-2.0+
4 * Raymond Lo, lo@routefree.com
5 * Wolfgang Denk, DENX Software Engineering, wd@denx.de.
9 * Support for harddisk partitions.
11 * To be compatible with LinuxPPC and Apple we use the standard Apple
12 * SCSI disk partitioning scheme. For more information see:
13 * http://developer.apple.com/techpubs/mac/Devices/Devices-126.html#MARKER-14-92
21 #include <asm/unaligned.h>
22 #include <linux/compiler.h>
26 #define DOS_PART_DEFAULT_SECTOR 512
28 /* should this be configurable? It looks like it's not very common at all
29 * to use large numbers of partitions */
30 #define MAX_EXT_PARTS 256
32 static inline int is_extended(int part_type
)
34 return (part_type
== DOS_PART_TYPE_EXTENDED
||
35 part_type
== DOS_PART_TYPE_EXTENDED_LBA
||
36 part_type
== DOS_PART_TYPE_EXTENDED_LINUX
);
39 static int get_bootable(dos_partition_t
*p
)
43 if (p
->sys_ind
== 0xef)
44 ret
|= PART_EFI_SYSTEM_PARTITION
;
45 if (p
->boot_ind
== 0x80)
50 static void print_one_part(dos_partition_t
*p
, lbaint_t ext_part_sector
,
51 int part_num
, unsigned int disksig
)
53 lbaint_t lba_start
= ext_part_sector
+ get_unaligned_le32(p
->start4
);
54 lbaint_t lba_size
= get_unaligned_le32(p
->size4
);
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 (get_bootable(p
) ? " Boot" : ""));
63 static int test_block_type(unsigned char *buffer
)
66 struct dos_partition
*p
;
69 if((buffer
[DOS_PART_MAGIC_OFFSET
+ 0] != 0x55) ||
70 (buffer
[DOS_PART_MAGIC_OFFSET
+ 1] != 0xaa) ) {
72 } /* no DOS Signature at all */
73 p
= (struct dos_partition
*)&buffer
[DOS_PART_TBL_OFFSET
];
75 /* Check that the boot indicators are valid and count the partitions. */
76 for (slot
= 0; slot
< 4; ++slot
, ++p
) {
77 if (p
->boot_ind
!= 0 && p
->boot_ind
!= 0x80)
84 * If the partition table is invalid or empty,
85 * check if this is a DOS PBR
87 if (slot
!= 4 || !part_count
) {
88 if (!strncmp((char *)&buffer
[DOS_PBR_FSTYPE_OFFSET
],
90 !strncmp((char *)&buffer
[DOS_PBR32_FSTYPE_OFFSET
],
92 return DOS_PBR
; /* This is a DOS PBR and not an MBR */
95 return DOS_MBR
; /* This is an DOS MBR */
97 /* This is neither a DOS MBR nor a DOS PBR */
101 static int part_test_dos(struct blk_desc
*desc
)
103 #ifndef CONFIG_SPL_BUILD
104 ALLOC_CACHE_ALIGN_BUFFER(legacy_mbr
, mbr
,
105 DIV_ROUND_UP(desc
->blksz
, sizeof(legacy_mbr
)));
107 if (blk_dread(desc
, 0, 1, (ulong
*)mbr
) != 1)
110 if (test_block_type((unsigned char *)mbr
) != DOS_MBR
)
113 if (desc
->sig_type
== SIG_TYPE_NONE
&& mbr
->unique_mbr_signature
) {
114 desc
->sig_type
= SIG_TYPE_MBR
;
115 desc
->mbr_sig
= mbr
->unique_mbr_signature
;
118 ALLOC_CACHE_ALIGN_BUFFER(unsigned char, buffer
, desc
->blksz
);
120 if (blk_dread(desc
, 0, 1, (ulong
*)buffer
) != 1)
123 if (test_block_type(buffer
) != DOS_MBR
)
130 /* Print a partition that is relative to its Extended partition table
132 static void print_partition_extended(struct blk_desc
*desc
,
133 lbaint_t ext_part_sector
,
135 int part_num
, unsigned int disksig
)
137 ALLOC_CACHE_ALIGN_BUFFER(unsigned char, buffer
, desc
->blksz
);
141 /* set a maximum recursion level */
142 if (part_num
> MAX_EXT_PARTS
)
144 printf("** Nested DOS partitions detected, stopping **\n");
148 if (blk_dread(desc
, ext_part_sector
, 1, (ulong
*)buffer
) != 1) {
149 printf ("** Can't read partition table on %d:" LBAFU
" **\n",
150 desc
->devnum
, ext_part_sector
);
153 i
=test_block_type(buffer
);
155 printf ("bad MBR sector signature 0x%02x%02x\n",
156 buffer
[DOS_PART_MAGIC_OFFSET
],
157 buffer
[DOS_PART_MAGIC_OFFSET
+ 1]);
161 if (!ext_part_sector
)
162 disksig
= get_unaligned_le32(&buffer
[DOS_PART_DISKSIG_OFFSET
]);
164 /* Print all primary/logical partitions */
165 pt
= (dos_partition_t
*) (buffer
+ DOS_PART_TBL_OFFSET
);
166 for (i
= 0; i
< 4; i
++, pt
++) {
168 * fdisk does not show the extended partitions that
172 if ((pt
->sys_ind
!= 0) &&
173 (ext_part_sector
== 0 || !is_extended (pt
->sys_ind
)) ) {
174 print_one_part(pt
, ext_part_sector
, part_num
, disksig
);
177 /* Reverse engr the fdisk part# assignment rule! */
178 if ((ext_part_sector
== 0) ||
179 (pt
->sys_ind
!= 0 && !is_extended (pt
->sys_ind
)) ) {
184 /* Follows the extended partitions */
185 pt
= (dos_partition_t
*) (buffer
+ DOS_PART_TBL_OFFSET
);
186 for (i
= 0; i
< 4; i
++, pt
++) {
187 if (is_extended (pt
->sys_ind
)) {
189 = get_unaligned_le32 (pt
->start4
) + relative
;
191 print_partition_extended(desc
, lba_start
,
192 !ext_part_sector
? lba_start
:
193 relative
, part_num
, disksig
);
201 /* Print a partition that is relative to its Extended partition table
203 static int part_get_info_extended(struct blk_desc
*desc
,
204 lbaint_t ext_part_sector
, lbaint_t relative
,
205 int part_num
, int which_part
,
206 struct disk_partition
*info
, uint disksig
)
208 ALLOC_CACHE_ALIGN_BUFFER(unsigned char, buffer
, desc
->blksz
);
209 struct disk_partition wdinfo
= { 0 };
214 /* set a maximum recursion level */
215 if (part_num
> MAX_EXT_PARTS
)
217 printf("** Nested DOS partitions detected, stopping **\n");
221 if (blk_dread(desc
, ext_part_sector
, 1, (ulong
*)buffer
) != 1) {
222 printf ("** Can't read partition table on %d:" LBAFU
" **\n",
223 desc
->devnum
, ext_part_sector
);
226 if (buffer
[DOS_PART_MAGIC_OFFSET
] != 0x55 ||
227 buffer
[DOS_PART_MAGIC_OFFSET
+ 1] != 0xaa) {
228 printf ("bad MBR sector signature 0x%02x%02x\n",
229 buffer
[DOS_PART_MAGIC_OFFSET
],
230 buffer
[DOS_PART_MAGIC_OFFSET
+ 1]);
234 #if CONFIG_IS_ENABLED(PARTITION_UUIDS)
235 if (!ext_part_sector
)
236 disksig
= get_unaligned_le32(&buffer
[DOS_PART_DISKSIG_OFFSET
]);
239 ret
= part_get_info_whole_disk(desc
, &wdinfo
);
243 /* Print all primary/logical partitions */
244 pt
= (dos_partition_t
*) (buffer
+ DOS_PART_TBL_OFFSET
);
245 for (i
= 0; i
< 4; i
++, pt
++) {
247 * fdisk does not show the extended partitions that
250 if (((pt
->boot_ind
& ~0x80) == 0) &&
251 (pt
->sys_ind
!= 0) &&
252 (part_num
== which_part
) &&
253 (ext_part_sector
== 0 || is_extended(pt
->sys_ind
) == 0)) {
254 if (wdinfo
.blksz
> DOS_PART_DEFAULT_SECTOR
)
255 info
->blksz
= wdinfo
.blksz
;
257 info
->blksz
= DOS_PART_DEFAULT_SECTOR
;
258 info
->start
= (lbaint_t
)(ext_part_sector
+
259 get_unaligned_le32(pt
->start4
));
260 info
->size
= (lbaint_t
)get_unaligned_le32(pt
->size4
);
261 part_set_generic_name(desc
, part_num
,
263 /* sprintf(info->type, "%d, pt->sys_ind); */
264 strcpy((char *)info
->type
, "U-Boot");
265 info
->bootable
= get_bootable(pt
);
266 #if CONFIG_IS_ENABLED(PARTITION_UUIDS)
267 sprintf(info
->uuid
, "%08x-%02x", disksig
, part_num
);
269 info
->sys_ind
= pt
->sys_ind
;
273 /* Reverse engr the fdisk part# assignment rule! */
274 if ((ext_part_sector
== 0) ||
275 (pt
->sys_ind
!= 0 && !is_extended (pt
->sys_ind
)) ) {
280 /* Follows the extended partitions */
281 pt
= (dos_partition_t
*) (buffer
+ DOS_PART_TBL_OFFSET
);
282 for (i
= 0; i
< 4; i
++, pt
++) {
283 if (is_extended (pt
->sys_ind
)) {
285 = get_unaligned_le32 (pt
->start4
) + relative
;
287 return part_get_info_extended(desc
, lba_start
,
288 ext_part_sector
== 0 ? lba_start
: relative
,
289 part_num
, which_part
, info
, disksig
);
293 /* Check for DOS PBR if no partition is found */
294 dos_type
= test_block_type(buffer
);
296 if (dos_type
== DOS_PBR
) {
298 info
->size
= desc
->lba
;
299 if (wdinfo
.blksz
> DOS_PART_DEFAULT_SECTOR
)
300 info
->blksz
= wdinfo
.blksz
;
302 info
->blksz
= DOS_PART_DEFAULT_SECTOR
;
304 strcpy((char *)info
->type
, "U-Boot");
305 #if CONFIG_IS_ENABLED(PARTITION_UUIDS)
314 static void __maybe_unused
part_print_dos(struct blk_desc
*desc
)
316 printf("Part\tStart Sector\tNum Sectors\tUUID\t\tType\n");
317 print_partition_extended(desc
, 0, 0, 1, 0);
320 static int __maybe_unused
part_get_info_dos(struct blk_desc
*desc
, int part
,
321 struct disk_partition
*info
)
323 return part_get_info_extended(desc
, 0, 0, 1, part
, info
, 0);
326 int is_valid_dos_buf(void *buf
)
328 return test_block_type(buf
) == DOS_MBR
? 0 : -1;
331 #if IS_ENABLED(CONFIG_CMD_MBR)
332 static void lba_to_chs(lbaint_t lba
, unsigned char *rc
, unsigned char *rh
,
335 unsigned int c
, h
, s
;
336 /* use fixed CHS geometry */
337 unsigned int sectpertrack
= 63;
338 unsigned int heads
= 255;
340 c
= (lba
+ 1) / sectpertrack
/ heads
;
341 h
= (lba
+ 1) / sectpertrack
- c
* heads
;
342 s
= (lba
+ 1) - (c
* heads
+ h
) * sectpertrack
;
352 *rs
= s
+ ((c
& 0x300) >> 2);
355 static void mbr_fill_pt_entry(dos_partition_t
*pt
, lbaint_t start
,
356 lbaint_t relative
, lbaint_t size
, uchar sys_ind
, bool bootable
)
358 pt
->boot_ind
= bootable
? 0x80 : 0x00;
359 pt
->sys_ind
= sys_ind
;
360 lba_to_chs(start
, &pt
->cyl
, &pt
->head
, &pt
->sector
);
361 lba_to_chs(start
+ size
- 1, &pt
->end_cyl
, &pt
->end_head
, &pt
->end_sector
);
362 put_unaligned_le32(relative
, &pt
->start4
);
363 put_unaligned_le32(size
, &pt
->size4
);
366 int write_mbr_partitions(struct blk_desc
*dev
,
367 struct disk_partition
*p
, int count
, unsigned int disksig
)
369 ALLOC_CACHE_ALIGN_BUFFER(unsigned char, buffer
, dev
->blksz
);
370 lbaint_t ext_part_start
= 0, ext_part_size
= 0, ext_part_sect
= 0;
374 memset(buffer
, 0, dev
->blksz
);
375 buffer
[DOS_PART_MAGIC_OFFSET
] = 0x55;
376 buffer
[DOS_PART_MAGIC_OFFSET
+ 1] = 0xaa;
377 put_unaligned_le32(disksig
, &buffer
[DOS_PART_DISKSIG_OFFSET
]);
378 pt
= (dos_partition_t
*) (buffer
+ DOS_PART_TBL_OFFSET
);
380 /* create all primary partitions */
381 for (i
= 0; i
< 4 && i
< count
; i
++, pt
++) {
382 mbr_fill_pt_entry(pt
, p
[i
].start
, p
[i
].start
, p
[i
].size
,
383 p
[i
].sys_ind
, p
[i
].bootable
);
384 if (is_extended(p
[i
].sys_ind
)) {
385 ext_part_start
= p
[i
].start
;
386 ext_part_size
= p
[i
].size
;
387 ext_part_sect
= p
[i
].start
;
391 if (i
< count
&& !ext_part_start
) {
392 printf("%s: extended partition is needed for more than 4 partitions\n",
398 if (blk_dwrite(dev
, 0, 1, buffer
) != 1) {
399 printf("%s: failed writing 'MBR' (1 blks at 0x0)\n",
404 /* create extended volumes */
405 for (; i
< count
; i
++) {
406 lbaint_t next_ebr
= 0;
408 memset(buffer
, 0, dev
->blksz
);
409 buffer
[DOS_PART_MAGIC_OFFSET
] = 0x55;
410 buffer
[DOS_PART_MAGIC_OFFSET
+ 1] = 0xaa;
411 pt
= (dos_partition_t
*) (buffer
+ DOS_PART_TBL_OFFSET
);
413 mbr_fill_pt_entry(pt
, p
[i
].start
, p
[i
].start
- ext_part_sect
,
414 p
[i
].size
, p
[i
].sys_ind
, p
[i
].bootable
);
418 next_ebr
= p
[i
].start
+ p
[i
].size
;
419 mbr_fill_pt_entry(pt
, next_ebr
,
420 next_ebr
- ext_part_start
,
421 p
[i
+1].start
+ p
[i
+1].size
- next_ebr
,
422 DOS_PART_TYPE_EXTENDED
, 0);
426 if (blk_dwrite(dev
, ext_part_sect
, 1, buffer
) != 1) {
427 printf("%s: failed writing 'EBR' (1 blks at 0x%lx)\n",
428 __func__
, ext_part_sect
);
431 ext_part_sect
= next_ebr
;
434 /* Update the partition table entries*/
440 int layout_mbr_partitions(struct disk_partition
*p
, int count
,
441 lbaint_t total_sectors
)
443 struct disk_partition
*ext
= NULL
;
445 lbaint_t ext_vol_start
;
447 /* calculate primary partitions start and size if needed */
449 p
[0].start
= DOS_PART_DEFAULT_GAP
;
450 for (i
= 0; i
< 4 && i
< count
; i
++) {
452 p
[i
].start
= p
[i
- 1].start
+ p
[i
- 1].size
;
454 lbaint_t end
= total_sectors
;
455 lbaint_t allocated
= 0;
457 for (j
= i
+ 1; j
< 4 && j
< count
; j
++) {
462 allocated
+= p
[j
].size
;
464 p
[i
].size
= end
- allocated
- p
[i
].start
;
466 if (p
[i
].sys_ind
== 0x05)
474 log_err("extended partition is needed for more than 4 partitions\n");
478 /* calculate extended volumes start and size if needed */
479 ext_vol_start
= ext
->start
;
480 for (i
= 4; i
< count
; i
++) {
482 p
[i
].start
= ext_vol_start
+ DOS_PART_DEFAULT_GAP
;
484 lbaint_t end
= ext
->start
+ ext
->size
;
485 lbaint_t allocated
= 0;
487 for (j
= i
+ 1; j
< count
; j
++) {
489 end
= p
[j
].start
- DOS_PART_DEFAULT_GAP
;
492 allocated
+= p
[j
].size
+ DOS_PART_DEFAULT_GAP
;
494 p
[i
].size
= end
- allocated
- p
[i
].start
;
496 ext_vol_start
= p
[i
].start
+ p
[i
].size
;
503 int write_mbr_sector(struct blk_desc
*desc
, void *buf
)
505 if (is_valid_dos_buf(buf
))
509 if (blk_dwrite(desc
, 0, 1, buf
) != 1) {
510 printf("%s: failed writing '%s' (1 blks at 0x0)\n",
515 /* Update the partition table entries*/
521 U_BOOT_PART_TYPE(dos
) = {
523 .part_type
= PART_TYPE_DOS
,
524 .max_entries
= DOS_ENTRY_NUMBERS
,
525 .get_info
= part_get_info_ptr(part_get_info_dos
),
526 .print
= part_print_ptr(part_print_dos
),
527 .test
= part_test_dos
,