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