Merge branch 'master' of git://git.denx.de/u-boot-i2c; branch 'master' of git://git...
[platform/kernel/u-boot.git] / disk / part.c
1 /*
2  * (C) Copyright 2001
3  * Wolfgang Denk, DENX Software Engineering, wd@denx.de.
4  *
5  * SPDX-License-Identifier:     GPL-2.0+
6  */
7
8 #include <common.h>
9 #include <command.h>
10 #include <ide.h>
11 #include <malloc.h>
12 #include <part.h>
13 #include <ubifs_uboot.h>
14
15 #undef  PART_DEBUG
16
17 #ifdef  PART_DEBUG
18 #define PRINTF(fmt,args...)     printf (fmt ,##args)
19 #else
20 #define PRINTF(fmt,args...)
21 #endif
22
23 struct block_drvr {
24         char *name;
25         block_dev_desc_t* (*get_dev)(int dev);
26         int (*select_hwpart)(int dev_num, int hwpart);
27 };
28
29 static const struct block_drvr block_drvr[] = {
30 #if defined(CONFIG_CMD_IDE)
31         { .name = "ide", .get_dev = ide_get_dev, },
32 #endif
33 #if defined(CONFIG_CMD_SATA)
34         {.name = "sata", .get_dev = sata_get_dev, },
35 #endif
36 #if defined(CONFIG_CMD_SCSI)
37         { .name = "scsi", .get_dev = scsi_get_dev, },
38 #endif
39 #if defined(CONFIG_CMD_USB) && defined(CONFIG_USB_STORAGE)
40         { .name = "usb", .get_dev = usb_stor_get_dev, },
41 #endif
42 #if defined(CONFIG_MMC)
43         {
44                 .name = "mmc",
45                 .get_dev = mmc_get_dev,
46                 .select_hwpart = mmc_select_hwpart,
47         },
48 #endif
49 #if defined(CONFIG_SYSTEMACE)
50         { .name = "ace", .get_dev = systemace_get_dev, },
51 #endif
52 #if defined(CONFIG_SANDBOX)
53         { .name = "host", .get_dev = host_get_dev, },
54 #endif
55         { },
56 };
57
58 DECLARE_GLOBAL_DATA_PTR;
59
60 #ifdef HAVE_BLOCK_DEVICE
61 static block_dev_desc_t *get_dev_hwpart(const char *ifname, int dev, int hwpart)
62 {
63         const struct block_drvr *drvr = block_drvr;
64         block_dev_desc_t* (*reloc_get_dev)(int dev);
65         int (*select_hwpart)(int dev_num, int hwpart);
66         char *name;
67         int ret;
68
69         if (!ifname)
70                 return NULL;
71
72         name = drvr->name;
73 #ifdef CONFIG_NEEDS_MANUAL_RELOC
74         name += gd->reloc_off;
75 #endif
76         while (drvr->name) {
77                 name = drvr->name;
78                 reloc_get_dev = drvr->get_dev;
79                 select_hwpart = drvr->select_hwpart;
80 #ifdef CONFIG_NEEDS_MANUAL_RELOC
81                 name += gd->reloc_off;
82                 reloc_get_dev += gd->reloc_off;
83                 if (select_hwpart)
84                         select_hwpart += gd->reloc_off;
85 #endif
86                 if (strncmp(ifname, name, strlen(name)) == 0) {
87                         block_dev_desc_t *dev_desc = reloc_get_dev(dev);
88                         if (!dev_desc)
89                                 return NULL;
90                         if (hwpart == 0 && !select_hwpart)
91                                 return dev_desc;
92                         if (!select_hwpart)
93                                 return NULL;
94                         ret = select_hwpart(dev_desc->dev, hwpart);
95                         if (ret < 0)
96                                 return NULL;
97                         return dev_desc;
98                 }
99                 drvr++;
100         }
101         return NULL;
102 }
103
104 block_dev_desc_t *get_dev(const char *ifname, int dev)
105 {
106         return get_dev_hwpart(ifname, dev, 0);
107 }
108 #else
109 block_dev_desc_t *get_dev_hwpart(const char *ifname, int dev, int hwpart)
110 {
111         return NULL;
112 }
113
114 block_dev_desc_t *get_dev(const char *ifname, int dev)
115 {
116         return NULL;
117 }
118 #endif
119
120 #ifdef HAVE_BLOCK_DEVICE
121
122 /* ------------------------------------------------------------------------- */
123 /*
124  * reports device info to the user
125  */
126
127 #ifdef CONFIG_LBA48
128 typedef uint64_t lba512_t;
129 #else
130 typedef lbaint_t lba512_t;
131 #endif
132
133 /*
134  * Overflowless variant of (block_count * mul_by / div_by)
135  * when div_by > mul_by
136  */
137 static lba512_t lba512_muldiv(lba512_t block_count, lba512_t mul_by, lba512_t div_by)
138 {
139         lba512_t bc_quot, bc_rem;
140
141         /* x * m / d == x / d * m + (x % d) * m / d */
142         bc_quot = block_count / div_by;
143         bc_rem  = block_count - div_by * bc_quot;
144         return bc_quot * mul_by + (bc_rem * mul_by) / div_by;
145 }
146
147 void dev_print (block_dev_desc_t *dev_desc)
148 {
149         lba512_t lba512; /* number of blocks if 512bytes block size */
150
151         if (dev_desc->type == DEV_TYPE_UNKNOWN) {
152                 puts ("not available\n");
153                 return;
154         }
155
156         switch (dev_desc->if_type) {
157         case IF_TYPE_SCSI:
158                 printf ("(%d:%d) Vendor: %s Prod.: %s Rev: %s\n",
159                         dev_desc->target,dev_desc->lun,
160                         dev_desc->vendor,
161                         dev_desc->product,
162                         dev_desc->revision);
163                 break;
164         case IF_TYPE_ATAPI:
165         case IF_TYPE_IDE:
166         case IF_TYPE_SATA:
167                 printf ("Model: %s Firm: %s Ser#: %s\n",
168                         dev_desc->vendor,
169                         dev_desc->revision,
170                         dev_desc->product);
171                 break;
172         case IF_TYPE_SD:
173         case IF_TYPE_MMC:
174         case IF_TYPE_USB:
175                 printf ("Vendor: %s Rev: %s Prod: %s\n",
176                         dev_desc->vendor,
177                         dev_desc->revision,
178                         dev_desc->product);
179                 break;
180         case IF_TYPE_DOC:
181                 puts("device type DOC\n");
182                 return;
183         case IF_TYPE_UNKNOWN:
184                 puts("device type unknown\n");
185                 return;
186         default:
187                 printf("Unhandled device type: %i\n", dev_desc->if_type);
188                 return;
189         }
190         puts ("            Type: ");
191         if (dev_desc->removable)
192                 puts ("Removable ");
193         switch (dev_desc->type & 0x1F) {
194         case DEV_TYPE_HARDDISK:
195                 puts ("Hard Disk");
196                 break;
197         case DEV_TYPE_CDROM:
198                 puts ("CD ROM");
199                 break;
200         case DEV_TYPE_OPDISK:
201                 puts ("Optical Device");
202                 break;
203         case DEV_TYPE_TAPE:
204                 puts ("Tape");
205                 break;
206         default:
207                 printf ("# %02X #", dev_desc->type & 0x1F);
208                 break;
209         }
210         puts ("\n");
211         if (dev_desc->lba > 0L && dev_desc->blksz > 0L) {
212                 ulong mb, mb_quot, mb_rem, gb, gb_quot, gb_rem;
213                 lbaint_t lba;
214
215                 lba = dev_desc->lba;
216
217                 lba512 = (lba * (dev_desc->blksz/512));
218                 /* round to 1 digit */
219                 /* 2048 = (1024 * 1024) / 512 MB */
220                 mb = lba512_muldiv(lba512, 10, 2048);
221
222                 mb_quot = mb / 10;
223                 mb_rem  = mb - (10 * mb_quot);
224
225                 gb = mb / 1024;
226                 gb_quot = gb / 10;
227                 gb_rem  = gb - (10 * gb_quot);
228 #ifdef CONFIG_LBA48
229                 if (dev_desc->lba48)
230                         printf ("            Supports 48-bit addressing\n");
231 #endif
232 #if defined(CONFIG_SYS_64BIT_LBA)
233                 printf ("            Capacity: %ld.%ld MB = %ld.%ld GB (%Ld x %ld)\n",
234                         mb_quot, mb_rem,
235                         gb_quot, gb_rem,
236                         lba,
237                         dev_desc->blksz);
238 #else
239                 printf ("            Capacity: %ld.%ld MB = %ld.%ld GB (%ld x %ld)\n",
240                         mb_quot, mb_rem,
241                         gb_quot, gb_rem,
242                         (ulong)lba,
243                         dev_desc->blksz);
244 #endif
245         } else {
246                 puts ("            Capacity: not available\n");
247         }
248 }
249 #endif
250
251 #ifdef HAVE_BLOCK_DEVICE
252
253 void init_part(block_dev_desc_t *dev_desc)
254 {
255 #ifdef CONFIG_ISO_PARTITION
256         if (test_part_iso(dev_desc) == 0) {
257                 dev_desc->part_type = PART_TYPE_ISO;
258                 return;
259         }
260 #endif
261
262 #ifdef CONFIG_MAC_PARTITION
263         if (test_part_mac(dev_desc) == 0) {
264                 dev_desc->part_type = PART_TYPE_MAC;
265                 return;
266         }
267 #endif
268
269 /* must be placed before DOS partition detection */
270 #ifdef CONFIG_EFI_PARTITION
271         if (test_part_efi(dev_desc) == 0) {
272                 dev_desc->part_type = PART_TYPE_EFI;
273                 return;
274         }
275 #endif
276
277 #ifdef CONFIG_DOS_PARTITION
278         if (test_part_dos(dev_desc) == 0) {
279                 dev_desc->part_type = PART_TYPE_DOS;
280                 return;
281         }
282 #endif
283
284 #ifdef CONFIG_AMIGA_PARTITION
285         if (test_part_amiga(dev_desc) == 0) {
286             dev_desc->part_type = PART_TYPE_AMIGA;
287             return;
288         }
289 #endif
290         dev_desc->part_type = PART_TYPE_UNKNOWN;
291 }
292
293
294 #if defined(CONFIG_MAC_PARTITION) || \
295         defined(CONFIG_DOS_PARTITION) || \
296         defined(CONFIG_ISO_PARTITION) || \
297         defined(CONFIG_AMIGA_PARTITION) || \
298         defined(CONFIG_EFI_PARTITION)
299
300 static void print_part_header(const char *type, block_dev_desc_t *dev_desc)
301 {
302         puts ("\nPartition Map for ");
303         switch (dev_desc->if_type) {
304         case IF_TYPE_IDE:
305                 puts ("IDE");
306                 break;
307         case IF_TYPE_SATA:
308                 puts ("SATA");
309                 break;
310         case IF_TYPE_SCSI:
311                 puts ("SCSI");
312                 break;
313         case IF_TYPE_ATAPI:
314                 puts ("ATAPI");
315                 break;
316         case IF_TYPE_USB:
317                 puts ("USB");
318                 break;
319         case IF_TYPE_DOC:
320                 puts ("DOC");
321                 break;
322         case IF_TYPE_MMC:
323                 puts ("MMC");
324                 break;
325         case IF_TYPE_HOST:
326                 puts("HOST");
327                 break;
328         default:
329                 puts ("UNKNOWN");
330                 break;
331         }
332         printf (" device %d  --   Partition Type: %s\n\n",
333                         dev_desc->dev, type);
334 }
335
336 #endif /* any CONFIG_..._PARTITION */
337
338 void print_part(block_dev_desc_t * dev_desc)
339 {
340
341                 switch (dev_desc->part_type) {
342 #ifdef CONFIG_MAC_PARTITION
343         case PART_TYPE_MAC:
344                 PRINTF ("## Testing for valid MAC partition ##\n");
345                 print_part_header ("MAC", dev_desc);
346                 print_part_mac (dev_desc);
347                 return;
348 #endif
349 #ifdef CONFIG_DOS_PARTITION
350         case PART_TYPE_DOS:
351                 PRINTF ("## Testing for valid DOS partition ##\n");
352                 print_part_header ("DOS", dev_desc);
353                 print_part_dos (dev_desc);
354                 return;
355 #endif
356
357 #ifdef CONFIG_ISO_PARTITION
358         case PART_TYPE_ISO:
359                 PRINTF ("## Testing for valid ISO Boot partition ##\n");
360                 print_part_header ("ISO", dev_desc);
361                 print_part_iso (dev_desc);
362                 return;
363 #endif
364
365 #ifdef CONFIG_AMIGA_PARTITION
366         case PART_TYPE_AMIGA:
367             PRINTF ("## Testing for a valid Amiga partition ##\n");
368             print_part_header ("AMIGA", dev_desc);
369             print_part_amiga (dev_desc);
370             return;
371 #endif
372
373 #ifdef CONFIG_EFI_PARTITION
374         case PART_TYPE_EFI:
375                 PRINTF ("## Testing for valid EFI partition ##\n");
376                 print_part_header ("EFI", dev_desc);
377                 print_part_efi (dev_desc);
378                 return;
379 #endif
380         }
381         puts ("## Unknown partition table\n");
382 }
383
384 #endif /* HAVE_BLOCK_DEVICE */
385
386 int get_partition_info(block_dev_desc_t *dev_desc, int part,
387                        disk_partition_t *info)
388 {
389 #ifdef HAVE_BLOCK_DEVICE
390
391 #ifdef CONFIG_PARTITION_UUIDS
392         /* The common case is no UUID support */
393         info->uuid[0] = 0;
394 #endif
395 #ifdef CONFIG_PARTITION_TYPE_GUID
396         info->type_guid[0] = 0;
397 #endif
398
399         switch (dev_desc->part_type) {
400 #ifdef CONFIG_MAC_PARTITION
401         case PART_TYPE_MAC:
402                 if (get_partition_info_mac(dev_desc, part, info) == 0) {
403                         PRINTF("## Valid MAC partition found ##\n");
404                         return 0;
405                 }
406                 break;
407 #endif
408
409 #ifdef CONFIG_DOS_PARTITION
410         case PART_TYPE_DOS:
411                 if (get_partition_info_dos(dev_desc, part, info) == 0) {
412                         PRINTF("## Valid DOS partition found ##\n");
413                         return 0;
414                 }
415                 break;
416 #endif
417
418 #ifdef CONFIG_ISO_PARTITION
419         case PART_TYPE_ISO:
420                 if (get_partition_info_iso(dev_desc, part, info) == 0) {
421                         PRINTF("## Valid ISO boot partition found ##\n");
422                         return 0;
423                 }
424                 break;
425 #endif
426
427 #ifdef CONFIG_AMIGA_PARTITION
428         case PART_TYPE_AMIGA:
429                 if (get_partition_info_amiga(dev_desc, part, info) == 0) {
430                         PRINTF("## Valid Amiga partition found ##\n");
431                         return 0;
432                 }
433                 break;
434 #endif
435
436 #ifdef CONFIG_EFI_PARTITION
437         case PART_TYPE_EFI:
438                 if (get_partition_info_efi(dev_desc, part, info) == 0) {
439                         PRINTF("## Valid EFI partition found ##\n");
440                         return 0;
441                 }
442                 break;
443 #endif
444         default:
445                 break;
446         }
447 #endif /* HAVE_BLOCK_DEVICE */
448
449         return -1;
450 }
451
452 int get_device(const char *ifname, const char *dev_hwpart_str,
453                block_dev_desc_t **dev_desc)
454 {
455         char *ep;
456         char *dup_str = NULL;
457         const char *dev_str, *hwpart_str;
458         int dev, hwpart;
459
460         hwpart_str = strchr(dev_hwpart_str, '.');
461         if (hwpart_str) {
462                 dup_str = strdup(dev_hwpart_str);
463                 dup_str[hwpart_str - dev_hwpart_str] = 0;
464                 dev_str = dup_str;
465                 hwpart_str++;
466         } else {
467                 dev_str = dev_hwpart_str;
468                 hwpart = 0;
469         }
470
471         dev = simple_strtoul(dev_str, &ep, 16);
472         if (*ep) {
473                 printf("** Bad device specification %s %s **\n",
474                        ifname, dev_str);
475                 dev = -1;
476                 goto cleanup;
477         }
478
479         if (hwpart_str) {
480                 hwpart = simple_strtoul(hwpart_str, &ep, 16);
481                 if (*ep) {
482                         printf("** Bad HW partition specification %s %s **\n",
483                             ifname, hwpart_str);
484                         dev = -1;
485                         goto cleanup;
486                 }
487         }
488
489         *dev_desc = get_dev_hwpart(ifname, dev, hwpart);
490         if (!(*dev_desc) || ((*dev_desc)->type == DEV_TYPE_UNKNOWN)) {
491                 printf("** Bad device %s %s **\n", ifname, dev_hwpart_str);
492                 dev = -1;
493                 goto cleanup;
494         }
495
496 #ifdef HAVE_BLOCK_DEVICE
497         /*
498          * Updates the partition table for the specified hw partition.
499          * Does not need to be done for hwpart 0 since it is default and
500          * already loaded.
501          */
502         if(hwpart != 0)
503                 init_part(*dev_desc);
504 #endif
505
506 cleanup:
507         free(dup_str);
508         return dev;
509 }
510
511 #define PART_UNSPECIFIED -2
512 #define PART_AUTO -1
513 #define MAX_SEARCH_PARTITIONS 16
514 int get_device_and_partition(const char *ifname, const char *dev_part_str,
515                              block_dev_desc_t **dev_desc,
516                              disk_partition_t *info, int allow_whole_dev)
517 {
518         int ret = -1;
519         const char *part_str;
520         char *dup_str = NULL;
521         const char *dev_str;
522         int dev;
523         char *ep;
524         int p;
525         int part;
526         disk_partition_t tmpinfo;
527
528 #if defined CONFIG_SANDBOX && defined CONFIG_CMD_UBIFS
529 #error Only one of CONFIG_SANDBOX and CONFIG_CMD_UBIFS may be selected
530 #endif
531
532 #ifdef CONFIG_SANDBOX
533         /*
534          * Special-case a pseudo block device "hostfs", to allow access to the
535          * host's own filesystem.
536          */
537         if (0 == strcmp(ifname, "hostfs")) {
538                 *dev_desc = NULL;
539                 info->start = 0;
540                 info->size = 0;
541                 info->blksz = 0;
542                 info->bootable = 0;
543                 strcpy((char *)info->type, BOOT_PART_TYPE);
544                 strcpy((char *)info->name, "Sandbox host");
545 #ifdef CONFIG_PARTITION_UUIDS
546                 info->uuid[0] = 0;
547 #endif
548 #ifdef CONFIG_PARTITION_TYPE_GUID
549                 info->type_guid[0] = 0;
550 #endif
551
552                 return 0;
553         }
554 #endif
555
556 #ifdef CONFIG_CMD_UBIFS
557         /*
558          * Special-case ubi, ubi goes through a mtd, rathen then through
559          * a regular block device.
560          */
561         if (0 == strcmp(ifname, "ubi")) {
562                 if (!ubifs_is_mounted()) {
563                         printf("UBIFS not mounted, use ubifsmount to mount volume first!\n");
564                         return -1;
565                 }
566
567                 *dev_desc = NULL;
568                 memset(info, 0, sizeof(*info));
569                 strcpy((char *)info->type, BOOT_PART_TYPE);
570                 strcpy((char *)info->name, "UBI");
571 #ifdef CONFIG_PARTITION_UUIDS
572                 info->uuid[0] = 0;
573 #endif
574                 return 0;
575         }
576 #endif
577
578         /* If no dev_part_str, use bootdevice environment variable */
579         if (!dev_part_str || !strlen(dev_part_str) ||
580             !strcmp(dev_part_str, "-"))
581                 dev_part_str = getenv("bootdevice");
582
583         /* If still no dev_part_str, it's an error */
584         if (!dev_part_str) {
585                 printf("** No device specified **\n");
586                 goto cleanup;
587         }
588
589         /* Separate device and partition ID specification */
590         part_str = strchr(dev_part_str, ':');
591         if (part_str) {
592                 dup_str = strdup(dev_part_str);
593                 dup_str[part_str - dev_part_str] = 0;
594                 dev_str = dup_str;
595                 part_str++;
596         } else {
597                 dev_str = dev_part_str;
598         }
599
600         /* Look up the device */
601         dev = get_device(ifname, dev_str, dev_desc);
602         if (dev < 0)
603                 goto cleanup;
604
605         /* Convert partition ID string to number */
606         if (!part_str || !*part_str) {
607                 part = PART_UNSPECIFIED;
608         } else if (!strcmp(part_str, "auto")) {
609                 part = PART_AUTO;
610         } else {
611                 /* Something specified -> use exactly that */
612                 part = (int)simple_strtoul(part_str, &ep, 16);
613                 /*
614                  * Less than whole string converted,
615                  * or request for whole device, but caller requires partition.
616                  */
617                 if (*ep || (part == 0 && !allow_whole_dev)) {
618                         printf("** Bad partition specification %s %s **\n",
619                             ifname, dev_part_str);
620                         goto cleanup;
621                 }
622         }
623
624         /*
625          * No partition table on device,
626          * or user requested partition 0 (entire device).
627          */
628         if (((*dev_desc)->part_type == PART_TYPE_UNKNOWN) ||
629             (part == 0)) {
630                 if (!(*dev_desc)->lba) {
631                         printf("** Bad device size - %s %s **\n", ifname,
632                                dev_str);
633                         goto cleanup;
634                 }
635
636                 /*
637                  * If user specified a partition ID other than 0,
638                  * or the calling command only accepts partitions,
639                  * it's an error.
640                  */
641                 if ((part > 0) || (!allow_whole_dev)) {
642                         printf("** No partition table - %s %s **\n", ifname,
643                                dev_str);
644                         goto cleanup;
645                 }
646
647                 (*dev_desc)->log2blksz = LOG2((*dev_desc)->blksz);
648
649                 info->start = 0;
650                 info->size = (*dev_desc)->lba;
651                 info->blksz = (*dev_desc)->blksz;
652                 info->bootable = 0;
653                 strcpy((char *)info->type, BOOT_PART_TYPE);
654                 strcpy((char *)info->name, "Whole Disk");
655 #ifdef CONFIG_PARTITION_UUIDS
656                 info->uuid[0] = 0;
657 #endif
658 #ifdef CONFIG_PARTITION_TYPE_GUID
659                 info->type_guid[0] = 0;
660 #endif
661
662                 ret = 0;
663                 goto cleanup;
664         }
665
666         /*
667          * Now there's known to be a partition table,
668          * not specifying a partition means to pick partition 1.
669          */
670         if (part == PART_UNSPECIFIED)
671                 part = 1;
672
673         /*
674          * If user didn't specify a partition number, or did specify something
675          * other than "auto", use that partition number directly.
676          */
677         if (part != PART_AUTO) {
678                 ret = get_partition_info(*dev_desc, part, info);
679                 if (ret) {
680                         printf("** Invalid partition %d **\n", part);
681                         goto cleanup;
682                 }
683         } else {
684                 /*
685                  * Find the first bootable partition.
686                  * If none are bootable, fall back to the first valid partition.
687                  */
688                 part = 0;
689                 for (p = 1; p <= MAX_SEARCH_PARTITIONS; p++) {
690                         ret = get_partition_info(*dev_desc, p, info);
691                         if (ret)
692                                 continue;
693
694                         /*
695                          * First valid partition, or new better partition?
696                          * If so, save partition ID.
697                          */
698                         if (!part || info->bootable)
699                                 part = p;
700
701                         /* Best possible partition? Stop searching. */
702                         if (info->bootable)
703                                 break;
704
705                         /*
706                          * We now need to search further for best possible.
707                          * If we what we just queried was the best so far,
708                          * save the info since we over-write it next loop.
709                          */
710                         if (part == p)
711                                 tmpinfo = *info;
712                 }
713                 /* If we found any acceptable partition */
714                 if (part) {
715                         /*
716                          * If we searched all possible partition IDs,
717                          * return the first valid partition we found.
718                          */
719                         if (p == MAX_SEARCH_PARTITIONS + 1)
720                                 *info = tmpinfo;
721                 } else {
722                         printf("** No valid partitions found **\n");
723                         ret = -1;
724                         goto cleanup;
725                 }
726         }
727         if (strncmp((char *)info->type, BOOT_PART_TYPE, sizeof(info->type)) != 0) {
728                 printf("** Invalid partition type \"%.32s\""
729                         " (expect \"" BOOT_PART_TYPE "\")\n",
730                         info->type);
731                 ret  = -1;
732                 goto cleanup;
733         }
734
735         (*dev_desc)->log2blksz = LOG2((*dev_desc)->blksz);
736
737         ret = part;
738         goto cleanup;
739
740 cleanup:
741         free(dup_str);
742         return ret;
743 }