1 /* 2 * (C) Copyright 2001 3 * Wolfgang Denk, DENX Software Engineering, wd@denx.de. 4 * 5 * See file CREDITS for list of people who contributed to this 6 * project. 7 * 8 * This program is free software; you can redistribute it and/or 9 * modify it under the terms of the GNU General Public License as 10 * published by the Free Software Foundation; either version 2 of 11 * the License, or (at your option) any later version. 12 * 13 * This program is distributed in the hope that it will be useful, 14 * but WITHOUT ANY WARRANTY; without even the implied warranty of 15 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 16 * GNU General Public License for more details. 17 * 18 * You should have received a copy of the GNU General Public License 19 * along with this program; if not, write to the Free Software 20 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, 21 * MA 02111-1307 USA 22 */ 23 24 #include <common.h> 25 #include <command.h> 26 #include <ide.h> 27 #include <malloc.h> 28 #include <part.h> 29 30 #undef PART_DEBUG 31 32 #ifdef PART_DEBUG 33 #define PRINTF(fmt,args...) printf (fmt ,##args) 34 #else 35 #define PRINTF(fmt,args...) 36 #endif 37 38 #if (defined(CONFIG_CMD_IDE) || \ 39 defined(CONFIG_CMD_SATA) || \ 40 defined(CONFIG_CMD_SCSI) || \ 41 defined(CONFIG_CMD_USB) || \ 42 defined(CONFIG_MMC) || \ 43 defined(CONFIG_SYSTEMACE) ) 44 45 struct block_drvr { 46 char *name; 47 block_dev_desc_t* (*get_dev)(int dev); 48 }; 49 50 static const struct block_drvr block_drvr[] = { 51 #if defined(CONFIG_CMD_IDE) 52 { .name = "ide", .get_dev = ide_get_dev, }, 53 #endif 54 #if defined(CONFIG_CMD_SATA) 55 {.name = "sata", .get_dev = sata_get_dev, }, 56 #endif 57 #if defined(CONFIG_CMD_SCSI) 58 { .name = "scsi", .get_dev = scsi_get_dev, }, 59 #endif 60 #if defined(CONFIG_CMD_USB) && defined(CONFIG_USB_STORAGE) 61 { .name = "usb", .get_dev = usb_stor_get_dev, }, 62 #endif 63 #if defined(CONFIG_MMC) 64 { .name = "mmc", .get_dev = mmc_get_dev, }, 65 #endif 66 #if defined(CONFIG_SYSTEMACE) 67 { .name = "ace", .get_dev = systemace_get_dev, }, 68 #endif 69 { }, 70 }; 71 72 DECLARE_GLOBAL_DATA_PTR; 73 74 block_dev_desc_t *get_dev(const char *ifname, int dev) 75 { 76 const struct block_drvr *drvr = block_drvr; 77 block_dev_desc_t* (*reloc_get_dev)(int dev); 78 char *name; 79 80 if (!ifname) 81 return NULL; 82 83 name = drvr->name; 84 #ifdef CONFIG_NEEDS_MANUAL_RELOC 85 name += gd->reloc_off; 86 #endif 87 while (drvr->name) { 88 name = drvr->name; 89 reloc_get_dev = drvr->get_dev; 90 #ifdef CONFIG_NEEDS_MANUAL_RELOC 91 name += gd->reloc_off; 92 reloc_get_dev += gd->reloc_off; 93 #endif 94 if (strncmp(ifname, name, strlen(name)) == 0) 95 return reloc_get_dev(dev); 96 drvr++; 97 } 98 return NULL; 99 } 100 #else 101 block_dev_desc_t *get_dev(const char *ifname, int dev) 102 { 103 return NULL; 104 } 105 #endif 106 107 #if (defined(CONFIG_CMD_IDE) || \ 108 defined(CONFIG_CMD_SATA) || \ 109 defined(CONFIG_CMD_SCSI) || \ 110 defined(CONFIG_CMD_USB) || \ 111 defined(CONFIG_MMC) || \ 112 defined(CONFIG_SYSTEMACE) ) 113 114 /* ------------------------------------------------------------------------- */ 115 /* 116 * reports device info to the user 117 */ 118 119 #ifdef CONFIG_LBA48 120 typedef uint64_t lba512_t; 121 #else 122 typedef lbaint_t lba512_t; 123 #endif 124 125 /* 126 * Overflowless variant of (block_count * mul_by / div_by) 127 * when div_by > mul_by 128 */ 129 static lba512_t lba512_muldiv (lba512_t block_count, lba512_t mul_by, lba512_t div_by) 130 { 131 lba512_t bc_quot, bc_rem; 132 133 /* x * m / d == x / d * m + (x % d) * m / d */ 134 bc_quot = block_count / div_by; 135 bc_rem = block_count - div_by * bc_quot; 136 return bc_quot * mul_by + (bc_rem * mul_by) / div_by; 137 } 138 139 void dev_print (block_dev_desc_t *dev_desc) 140 { 141 lba512_t lba512; /* number of blocks if 512bytes block size */ 142 143 if (dev_desc->type == DEV_TYPE_UNKNOWN) { 144 puts ("not available\n"); 145 return; 146 } 147 148 switch (dev_desc->if_type) { 149 case IF_TYPE_SCSI: 150 printf ("(%d:%d) Vendor: %s Prod.: %s Rev: %s\n", 151 dev_desc->target,dev_desc->lun, 152 dev_desc->vendor, 153 dev_desc->product, 154 dev_desc->revision); 155 break; 156 case IF_TYPE_ATAPI: 157 case IF_TYPE_IDE: 158 case IF_TYPE_SATA: 159 printf ("Model: %s Firm: %s Ser#: %s\n", 160 dev_desc->vendor, 161 dev_desc->revision, 162 dev_desc->product); 163 break; 164 case IF_TYPE_SD: 165 case IF_TYPE_MMC: 166 case IF_TYPE_USB: 167 printf ("Vendor: %s Rev: %s Prod: %s\n", 168 dev_desc->vendor, 169 dev_desc->revision, 170 dev_desc->product); 171 break; 172 case IF_TYPE_DOC: 173 puts("device type DOC\n"); 174 return; 175 case IF_TYPE_UNKNOWN: 176 puts("device type unknown\n"); 177 return; 178 default: 179 printf("Unhandled device type: %i\n", dev_desc->if_type); 180 return; 181 } 182 puts (" Type: "); 183 if (dev_desc->removable) 184 puts ("Removable "); 185 switch (dev_desc->type & 0x1F) { 186 case DEV_TYPE_HARDDISK: 187 puts ("Hard Disk"); 188 break; 189 case DEV_TYPE_CDROM: 190 puts ("CD ROM"); 191 break; 192 case DEV_TYPE_OPDISK: 193 puts ("Optical Device"); 194 break; 195 case DEV_TYPE_TAPE: 196 puts ("Tape"); 197 break; 198 default: 199 printf ("# %02X #", dev_desc->type & 0x1F); 200 break; 201 } 202 puts ("\n"); 203 if ((dev_desc->lba * dev_desc->blksz)>0L) { 204 ulong mb, mb_quot, mb_rem, gb, gb_quot, gb_rem; 205 lbaint_t lba; 206 207 lba = dev_desc->lba; 208 209 lba512 = (lba * (dev_desc->blksz/512)); 210 /* round to 1 digit */ 211 mb = lba512_muldiv(lba512, 10, 2048); /* 2048 = (1024 * 1024) / 512 MB */ 212 213 mb_quot = mb / 10; 214 mb_rem = mb - (10 * mb_quot); 215 216 gb = mb / 1024; 217 gb_quot = gb / 10; 218 gb_rem = gb - (10 * gb_quot); 219 #ifdef CONFIG_LBA48 220 if (dev_desc->lba48) 221 printf (" Supports 48-bit addressing\n"); 222 #endif 223 #if defined(CONFIG_SYS_64BIT_LBA) 224 printf (" Capacity: %ld.%ld MB = %ld.%ld GB (%Ld x %ld)\n", 225 mb_quot, mb_rem, 226 gb_quot, gb_rem, 227 lba, 228 dev_desc->blksz); 229 #else 230 printf (" Capacity: %ld.%ld MB = %ld.%ld GB (%ld x %ld)\n", 231 mb_quot, mb_rem, 232 gb_quot, gb_rem, 233 (ulong)lba, 234 dev_desc->blksz); 235 #endif 236 } else { 237 puts (" Capacity: not available\n"); 238 } 239 } 240 #endif 241 242 #if (defined(CONFIG_CMD_IDE) || \ 243 defined(CONFIG_CMD_SATA) || \ 244 defined(CONFIG_CMD_SCSI) || \ 245 defined(CONFIG_CMD_USB) || \ 246 defined(CONFIG_MMC) || \ 247 defined(CONFIG_SYSTEMACE) ) 248 249 #if defined(CONFIG_MAC_PARTITION) || \ 250 defined(CONFIG_DOS_PARTITION) || \ 251 defined(CONFIG_ISO_PARTITION) || \ 252 defined(CONFIG_AMIGA_PARTITION) || \ 253 defined(CONFIG_EFI_PARTITION) 254 255 void init_part (block_dev_desc_t * dev_desc) 256 { 257 #ifdef CONFIG_ISO_PARTITION 258 if (test_part_iso(dev_desc) == 0) { 259 dev_desc->part_type = PART_TYPE_ISO; 260 return; 261 } 262 #endif 263 264 #ifdef CONFIG_MAC_PARTITION 265 if (test_part_mac(dev_desc) == 0) { 266 dev_desc->part_type = PART_TYPE_MAC; 267 return; 268 } 269 #endif 270 271 /* must be placed before DOS partition detection */ 272 #ifdef CONFIG_EFI_PARTITION 273 if (test_part_efi(dev_desc) == 0) { 274 dev_desc->part_type = PART_TYPE_EFI; 275 return; 276 } 277 #endif 278 279 #ifdef CONFIG_DOS_PARTITION 280 if (test_part_dos(dev_desc) == 0) { 281 dev_desc->part_type = PART_TYPE_DOS; 282 return; 283 } 284 #endif 285 286 #ifdef CONFIG_AMIGA_PARTITION 287 if (test_part_amiga(dev_desc) == 0) { 288 dev_desc->part_type = PART_TYPE_AMIGA; 289 return; 290 } 291 #endif 292 dev_desc->part_type = PART_TYPE_UNKNOWN; 293 } 294 295 296 static void print_part_header (const char *type, block_dev_desc_t * dev_desc) 297 { 298 puts ("\nPartition Map for "); 299 switch (dev_desc->if_type) { 300 case IF_TYPE_IDE: 301 puts ("IDE"); 302 break; 303 case IF_TYPE_SATA: 304 puts ("SATA"); 305 break; 306 case IF_TYPE_SCSI: 307 puts ("SCSI"); 308 break; 309 case IF_TYPE_ATAPI: 310 puts ("ATAPI"); 311 break; 312 case IF_TYPE_USB: 313 puts ("USB"); 314 break; 315 case IF_TYPE_DOC: 316 puts ("DOC"); 317 break; 318 case IF_TYPE_MMC: 319 puts ("MMC"); 320 break; 321 default: 322 puts ("UNKNOWN"); 323 break; 324 } 325 printf (" device %d -- Partition Type: %s\n\n", 326 dev_desc->dev, type); 327 } 328 329 void print_part (block_dev_desc_t * dev_desc) 330 { 331 332 switch (dev_desc->part_type) { 333 #ifdef CONFIG_MAC_PARTITION 334 case PART_TYPE_MAC: 335 PRINTF ("## Testing for valid MAC partition ##\n"); 336 print_part_header ("MAC", dev_desc); 337 print_part_mac (dev_desc); 338 return; 339 #endif 340 #ifdef CONFIG_DOS_PARTITION 341 case PART_TYPE_DOS: 342 PRINTF ("## Testing for valid DOS partition ##\n"); 343 print_part_header ("DOS", dev_desc); 344 print_part_dos (dev_desc); 345 return; 346 #endif 347 348 #ifdef CONFIG_ISO_PARTITION 349 case PART_TYPE_ISO: 350 PRINTF ("## Testing for valid ISO Boot partition ##\n"); 351 print_part_header ("ISO", dev_desc); 352 print_part_iso (dev_desc); 353 return; 354 #endif 355 356 #ifdef CONFIG_AMIGA_PARTITION 357 case PART_TYPE_AMIGA: 358 PRINTF ("## Testing for a valid Amiga partition ##\n"); 359 print_part_header ("AMIGA", dev_desc); 360 print_part_amiga (dev_desc); 361 return; 362 #endif 363 364 #ifdef CONFIG_EFI_PARTITION 365 case PART_TYPE_EFI: 366 PRINTF ("## Testing for valid EFI partition ##\n"); 367 print_part_header ("EFI", dev_desc); 368 print_part_efi (dev_desc); 369 return; 370 #endif 371 } 372 puts ("## Unknown partition table\n"); 373 } 374 375 376 #else /* neither MAC nor DOS nor ISO nor AMIGA nor EFI partition configured */ 377 # error neither CONFIG_MAC_PARTITION nor CONFIG_DOS_PARTITION 378 # error nor CONFIG_ISO_PARTITION nor CONFIG_AMIGA_PARTITION 379 # error nor CONFIG_EFI_PARTITION configured! 380 #endif 381 382 #endif 383 384 int get_partition_info(block_dev_desc_t *dev_desc, int part 385 , disk_partition_t *info) 386 { 387 #if defined(CONFIG_CMD_IDE) || \ 388 defined(CONFIG_CMD_SATA) || \ 389 defined(CONFIG_CMD_SCSI) || \ 390 defined(CONFIG_CMD_USB) || \ 391 defined(CONFIG_MMC) || \ 392 defined(CONFIG_SYSTEMACE) 393 394 #ifdef CONFIG_PARTITION_UUIDS 395 /* The common case is no UUID support */ 396 info->uuid[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 448 449 return -1; 450 } 451 452 int get_device(const char *ifname, const char *dev_str, 453 block_dev_desc_t **dev_desc) 454 { 455 char *ep; 456 int dev; 457 458 dev = simple_strtoul(dev_str, &ep, 16); 459 if (*ep) { 460 printf("** Bad device specification %s %s **\n", 461 ifname, dev_str); 462 return -1; 463 } 464 465 *dev_desc = get_dev(ifname, dev); 466 if (!(*dev_desc) || ((*dev_desc)->type == DEV_TYPE_UNKNOWN)) { 467 printf("** Bad device %s %s **\n", ifname, dev_str); 468 return -1; 469 } 470 471 return dev; 472 } 473 474 #define PART_UNSPECIFIED -2 475 #define PART_AUTO -1 476 #define MAX_SEARCH_PARTITIONS 16 477 int get_device_and_partition(const char *ifname, const char *dev_part_str, 478 block_dev_desc_t **dev_desc, 479 disk_partition_t *info, int allow_whole_dev) 480 { 481 int ret = -1; 482 const char *part_str; 483 char *dup_str = NULL; 484 const char *dev_str; 485 int dev; 486 char *ep; 487 int p; 488 int part; 489 disk_partition_t tmpinfo; 490 491 /* If no dev_part_str, use bootdevice environment variable */ 492 if (!dev_part_str || !strlen(dev_part_str) || 493 !strcmp(dev_part_str, "-")) 494 dev_part_str = getenv("bootdevice"); 495 496 /* If still no dev_part_str, it's an error */ 497 if (!dev_part_str) { 498 printf("** No device specified **\n"); 499 goto cleanup; 500 } 501 502 /* Separate device and partition ID specification */ 503 part_str = strchr(dev_part_str, ':'); 504 if (part_str) { 505 dup_str = strdup(dev_part_str); 506 dup_str[part_str - dev_part_str] = 0; 507 dev_str = dup_str; 508 part_str++; 509 } else { 510 dev_str = dev_part_str; 511 } 512 513 /* Look up the device */ 514 dev = get_device(ifname, dev_str, dev_desc); 515 if (dev < 0) 516 goto cleanup; 517 518 /* Convert partition ID string to number */ 519 if (!part_str || !*part_str) { 520 part = PART_UNSPECIFIED; 521 } else if (!strcmp(part_str, "auto")) { 522 part = PART_AUTO; 523 } else { 524 /* Something specified -> use exactly that */ 525 part = (int)simple_strtoul(part_str, &ep, 16); 526 /* 527 * Less than whole string converted, 528 * or request for whole device, but caller requires partition. 529 */ 530 if (*ep || (part == 0 && !allow_whole_dev)) { 531 printf("** Bad partition specification %s %s **\n", 532 ifname, dev_part_str); 533 goto cleanup; 534 } 535 } 536 537 /* 538 * No partition table on device, 539 * or user requested partition 0 (entire device). 540 */ 541 if (((*dev_desc)->part_type == PART_TYPE_UNKNOWN) || 542 (part == 0)) { 543 if (!(*dev_desc)->lba) { 544 printf("** Bad device size - %s %s **\n", ifname, 545 dev_str); 546 goto cleanup; 547 } 548 549 /* 550 * If user specified a partition ID other than 0, 551 * or the calling command only accepts partitions, 552 * it's an error. 553 */ 554 if ((part > 0) || (!allow_whole_dev)) { 555 printf("** No partition table - %s %s **\n", ifname, 556 dev_str); 557 goto cleanup; 558 } 559 560 info->start = 0; 561 info->size = (*dev_desc)->lba; 562 info->blksz = (*dev_desc)->blksz; 563 info->bootable = 0; 564 #ifdef CONFIG_PARTITION_UUIDS 565 info->uuid[0] = 0; 566 #endif 567 568 ret = 0; 569 goto cleanup; 570 } 571 572 /* 573 * Now there's known to be a partition table, 574 * not specifying a partition means to pick partition 1. 575 */ 576 if (part == PART_UNSPECIFIED) 577 part = 1; 578 579 /* 580 * If user didn't specify a partition number, or did specify something 581 * other than "auto", use that partition number directly. 582 */ 583 if (part != PART_AUTO) { 584 ret = get_partition_info(*dev_desc, part, info); 585 if (ret) { 586 printf("** Invalid partition %d **\n", part); 587 goto cleanup; 588 } 589 } else { 590 /* 591 * Find the first bootable partition. 592 * If none are bootable, fall back to the first valid partition. 593 */ 594 part = 0; 595 for (p = 1; p <= MAX_SEARCH_PARTITIONS; p++) { 596 ret = get_partition_info(*dev_desc, p, info); 597 if (ret) 598 continue; 599 600 /* 601 * First valid partition, or new better partition? 602 * If so, save partition ID. 603 */ 604 if (!part || info->bootable) 605 part = p; 606 607 /* Best possible partition? Stop searching. */ 608 if (info->bootable) 609 break; 610 611 /* 612 * We now need to search further for best possible. 613 * If we what we just queried was the best so far, 614 * save the info since we over-write it next loop. 615 */ 616 if (part == p) 617 tmpinfo = *info; 618 } 619 /* If we found any acceptable partition */ 620 if (part) { 621 /* 622 * If we searched all possible partition IDs, 623 * return the first valid partition we found. 624 */ 625 if (p == MAX_SEARCH_PARTITIONS + 1) 626 *info = tmpinfo; 627 ret = 0; 628 } else { 629 printf("** No valid partitions found **\n"); 630 goto cleanup; 631 } 632 } 633 if (strncmp((char *)info->type, BOOT_PART_TYPE, sizeof(info->type)) != 0) { 634 printf("** Invalid partition type \"%.32s\"" 635 " (expect \"" BOOT_PART_TYPE "\")\n", 636 info->type); 637 ret = -1; 638 goto cleanup; 639 } 640 641 ret = part; 642 goto cleanup; 643 644 cleanup: 645 free(dup_str); 646 return ret; 647 } 648