1affae2bfSwdenk /* 2affae2bfSwdenk * (C) Copyright 2001 3affae2bfSwdenk * Wolfgang Denk, DENX Software Engineering, wd@denx.de. 4affae2bfSwdenk * 51a459660SWolfgang Denk * SPDX-License-Identifier: GPL-2.0+ 6affae2bfSwdenk */ 7affae2bfSwdenk 8affae2bfSwdenk #include <common.h> 9affae2bfSwdenk #include <command.h> 10affae2bfSwdenk #include <ide.h> 1110a37fd7SStephen Warren #include <malloc.h> 12735dd97bSGrant Likely #include <part.h> 13affae2bfSwdenk 14affae2bfSwdenk #undef PART_DEBUG 15affae2bfSwdenk 16affae2bfSwdenk #ifdef PART_DEBUG 17affae2bfSwdenk #define PRINTF(fmt,args...) printf (fmt ,##args) 18affae2bfSwdenk #else 19affae2bfSwdenk #define PRINTF(fmt,args...) 20affae2bfSwdenk #endif 21affae2bfSwdenk 22735dd97bSGrant Likely struct block_drvr { 23735dd97bSGrant Likely char *name; 24735dd97bSGrant Likely block_dev_desc_t* (*get_dev)(int dev); 25336b6f90SStephen Warren int (*select_hwpart)(int dev_num, int hwpart); 26735dd97bSGrant Likely }; 27735dd97bSGrant Likely 28735dd97bSGrant Likely static const struct block_drvr block_drvr[] = { 29cde5c64dSJon Loeliger #if defined(CONFIG_CMD_IDE) 30735dd97bSGrant Likely { .name = "ide", .get_dev = ide_get_dev, }, 31735dd97bSGrant Likely #endif 32c7057b52SDave Liu #if defined(CONFIG_CMD_SATA) 33c7057b52SDave Liu {.name = "sata", .get_dev = sata_get_dev, }, 34c7057b52SDave Liu #endif 35cde5c64dSJon Loeliger #if defined(CONFIG_CMD_SCSI) 36735dd97bSGrant Likely { .name = "scsi", .get_dev = scsi_get_dev, }, 37735dd97bSGrant Likely #endif 38cde5c64dSJon Loeliger #if defined(CONFIG_CMD_USB) && defined(CONFIG_USB_STORAGE) 39735dd97bSGrant Likely { .name = "usb", .get_dev = usb_stor_get_dev, }, 40735dd97bSGrant Likely #endif 41735dd97bSGrant Likely #if defined(CONFIG_MMC) 42d2356284SStephen Warren { 43d2356284SStephen Warren .name = "mmc", 44d2356284SStephen Warren .get_dev = mmc_get_dev, 45d2356284SStephen Warren .select_hwpart = mmc_select_hwpart, 46d2356284SStephen Warren }, 47735dd97bSGrant Likely #endif 48735dd97bSGrant Likely #if defined(CONFIG_SYSTEMACE) 49735dd97bSGrant Likely { .name = "ace", .get_dev = systemace_get_dev, }, 50735dd97bSGrant Likely #endif 51f4d8de48SHenrik Nordström #if defined(CONFIG_SANDBOX) 52f4d8de48SHenrik Nordström { .name = "host", .get_dev = host_get_dev, }, 53f4d8de48SHenrik Nordström #endif 54735dd97bSGrant Likely { }, 55735dd97bSGrant Likely }; 56735dd97bSGrant Likely 57751bb571SStefan Roese DECLARE_GLOBAL_DATA_PTR; 58751bb571SStefan Roese 590c9c8fb5SGabe Black #ifdef HAVE_BLOCK_DEVICE 60336b6f90SStephen Warren block_dev_desc_t *get_dev_hwpart(const char *ifname, int dev, int hwpart) 61735dd97bSGrant Likely { 62735dd97bSGrant Likely const struct block_drvr *drvr = block_drvr; 63751bb571SStefan Roese block_dev_desc_t* (*reloc_get_dev)(int dev); 64336b6f90SStephen Warren int (*select_hwpart)(int dev_num, int hwpart); 6523090dacSHeiko Schocher char *name; 66336b6f90SStephen Warren int ret; 67751bb571SStefan Roese 687e71dc68STim Kientzle if (!ifname) 697e71dc68STim Kientzle return NULL; 707e71dc68STim Kientzle 7123090dacSHeiko Schocher name = drvr->name; 722e5167ccSWolfgang Denk #ifdef CONFIG_NEEDS_MANUAL_RELOC 7323090dacSHeiko Schocher name += gd->reloc_off; 7423090dacSHeiko Schocher #endif 75b16aadf4SLei Wen while (drvr->name) { 7623090dacSHeiko Schocher name = drvr->name; 77521af04dSPeter Tyser reloc_get_dev = drvr->get_dev; 78336b6f90SStephen Warren select_hwpart = drvr->select_hwpart; 792e5167ccSWolfgang Denk #ifdef CONFIG_NEEDS_MANUAL_RELOC 8023090dacSHeiko Schocher name += gd->reloc_off; 81521af04dSPeter Tyser reloc_get_dev += gd->reloc_off; 82336b6f90SStephen Warren if (select_hwpart) 83336b6f90SStephen Warren select_hwpart += gd->reloc_off; 84521af04dSPeter Tyser #endif 85336b6f90SStephen Warren if (strncmp(ifname, name, strlen(name)) == 0) { 86336b6f90SStephen Warren block_dev_desc_t *dev_desc = reloc_get_dev(dev); 87336b6f90SStephen Warren if (!dev_desc) 88336b6f90SStephen Warren return NULL; 89ecdd57e2SStephen Warren if (hwpart == 0 && !select_hwpart) 90336b6f90SStephen Warren return dev_desc; 91336b6f90SStephen Warren if (!select_hwpart) 92336b6f90SStephen Warren return NULL; 93336b6f90SStephen Warren ret = select_hwpart(dev_desc->dev, hwpart); 94336b6f90SStephen Warren if (ret < 0) 95336b6f90SStephen Warren return NULL; 96336b6f90SStephen Warren return dev_desc; 97336b6f90SStephen Warren } 98735dd97bSGrant Likely drvr++; 99735dd97bSGrant Likely } 100735dd97bSGrant Likely return NULL; 101735dd97bSGrant Likely } 102336b6f90SStephen Warren 103336b6f90SStephen Warren block_dev_desc_t *get_dev(const char *ifname, int dev) 104336b6f90SStephen Warren { 105ecdd57e2SStephen Warren return get_dev_hwpart(ifname, dev, 0); 106336b6f90SStephen Warren } 107735dd97bSGrant Likely #else 108336b6f90SStephen Warren block_dev_desc_t *get_dev_hwpart(const char *ifname, int dev, int hwpart) 109336b6f90SStephen Warren { 110336b6f90SStephen Warren return NULL; 111336b6f90SStephen Warren } 112336b6f90SStephen Warren 11399d2c205SRob Herring block_dev_desc_t *get_dev(const char *ifname, int dev) 114735dd97bSGrant Likely { 115735dd97bSGrant Likely return NULL; 116735dd97bSGrant Likely } 117735dd97bSGrant Likely #endif 118735dd97bSGrant Likely 1190c9c8fb5SGabe Black #ifdef HAVE_BLOCK_DEVICE 120735dd97bSGrant Likely 121affae2bfSwdenk /* ------------------------------------------------------------------------- */ 122affae2bfSwdenk /* 123affae2bfSwdenk * reports device info to the user 124affae2bfSwdenk */ 12569a2a4d9SSergei Trofimovich 12669a2a4d9SSergei Trofimovich #ifdef CONFIG_LBA48 12769a2a4d9SSergei Trofimovich typedef uint64_t lba512_t; 12869a2a4d9SSergei Trofimovich #else 12969a2a4d9SSergei Trofimovich typedef lbaint_t lba512_t; 13069a2a4d9SSergei Trofimovich #endif 13169a2a4d9SSergei Trofimovich 13269a2a4d9SSergei Trofimovich /* 13369a2a4d9SSergei Trofimovich * Overflowless variant of (block_count * mul_by / div_by) 13469a2a4d9SSergei Trofimovich * when div_by > mul_by 13569a2a4d9SSergei Trofimovich */ 13669a2a4d9SSergei Trofimovich static lba512_t lba512_muldiv (lba512_t block_count, lba512_t mul_by, lba512_t div_by) 13769a2a4d9SSergei Trofimovich { 13869a2a4d9SSergei Trofimovich lba512_t bc_quot, bc_rem; 13969a2a4d9SSergei Trofimovich 14069a2a4d9SSergei Trofimovich /* x * m / d == x / d * m + (x % d) * m / d */ 14169a2a4d9SSergei Trofimovich bc_quot = block_count / div_by; 14269a2a4d9SSergei Trofimovich bc_rem = block_count - div_by * bc_quot; 14369a2a4d9SSergei Trofimovich return bc_quot * mul_by + (bc_rem * mul_by) / div_by; 14469a2a4d9SSergei Trofimovich } 14569a2a4d9SSergei Trofimovich 146affae2bfSwdenk void dev_print (block_dev_desc_t *dev_desc) 147affae2bfSwdenk { 14869a2a4d9SSergei Trofimovich lba512_t lba512; /* number of blocks if 512bytes block size */ 149affae2bfSwdenk 150af75a45dSWolfgang Denk if (dev_desc->type == DEV_TYPE_UNKNOWN) { 151af75a45dSWolfgang Denk puts ("not available\n"); 152af75a45dSWolfgang Denk return; 153af75a45dSWolfgang Denk } 154af75a45dSWolfgang Denk 1558ec6e332STor Krill switch (dev_desc->if_type) { 156574b3195SDetlev Zundel case IF_TYPE_SCSI: 157574b3195SDetlev Zundel printf ("(%d:%d) Vendor: %s Prod.: %s Rev: %s\n", 158574b3195SDetlev Zundel dev_desc->target,dev_desc->lun, 159affae2bfSwdenk dev_desc->vendor, 160affae2bfSwdenk dev_desc->product, 161affae2bfSwdenk dev_desc->revision); 162574b3195SDetlev Zundel break; 1636e24a1ebSRemy Bohmer case IF_TYPE_ATAPI: 164574b3195SDetlev Zundel case IF_TYPE_IDE: 165574b3195SDetlev Zundel case IF_TYPE_SATA: 166574b3195SDetlev Zundel printf ("Model: %s Firm: %s Ser#: %s\n", 167574b3195SDetlev Zundel dev_desc->vendor, 168574b3195SDetlev Zundel dev_desc->revision, 169574b3195SDetlev Zundel dev_desc->product); 170574b3195SDetlev Zundel break; 1716e24a1ebSRemy Bohmer case IF_TYPE_SD: 1726e24a1ebSRemy Bohmer case IF_TYPE_MMC: 17347bebe34SNícolas Carneiro Lebedenco case IF_TYPE_USB: 17447bebe34SNícolas Carneiro Lebedenco printf ("Vendor: %s Rev: %s Prod: %s\n", 17547bebe34SNícolas Carneiro Lebedenco dev_desc->vendor, 17647bebe34SNícolas Carneiro Lebedenco dev_desc->revision, 17747bebe34SNícolas Carneiro Lebedenco dev_desc->product); 17847bebe34SNícolas Carneiro Lebedenco break; 1796e24a1ebSRemy Bohmer case IF_TYPE_DOC: 1806e24a1ebSRemy Bohmer puts("device type DOC\n"); 1816e24a1ebSRemy Bohmer return; 1828ec6e332STor Krill case IF_TYPE_UNKNOWN: 1836e24a1ebSRemy Bohmer puts("device type unknown\n"); 1846e24a1ebSRemy Bohmer return; 185574b3195SDetlev Zundel default: 1866e24a1ebSRemy Bohmer printf("Unhandled device type: %i\n", dev_desc->if_type); 187574b3195SDetlev Zundel return; 188affae2bfSwdenk } 189affae2bfSwdenk puts (" Type: "); 190affae2bfSwdenk if (dev_desc->removable) 191affae2bfSwdenk puts ("Removable "); 192affae2bfSwdenk switch (dev_desc->type & 0x1F) { 193726c0f1eSDetlev Zundel case DEV_TYPE_HARDDISK: 194726c0f1eSDetlev Zundel puts ("Hard Disk"); 195affae2bfSwdenk break; 196726c0f1eSDetlev Zundel case DEV_TYPE_CDROM: 197726c0f1eSDetlev Zundel puts ("CD ROM"); 198affae2bfSwdenk break; 199726c0f1eSDetlev Zundel case DEV_TYPE_OPDISK: 200726c0f1eSDetlev Zundel puts ("Optical Device"); 201affae2bfSwdenk break; 202726c0f1eSDetlev Zundel case DEV_TYPE_TAPE: 203726c0f1eSDetlev Zundel puts ("Tape"); 204affae2bfSwdenk break; 205726c0f1eSDetlev Zundel default: 206726c0f1eSDetlev Zundel printf ("# %02X #", dev_desc->type & 0x1F); 207affae2bfSwdenk break; 208affae2bfSwdenk } 209affae2bfSwdenk puts ("\n"); 21033699df1SJerry Huang if (dev_desc->lba > 0L && dev_desc->blksz > 0L) { 211affae2bfSwdenk ulong mb, mb_quot, mb_rem, gb, gb_quot, gb_rem; 212c40b2956Swdenk lbaint_t lba; 2136e592385Swdenk 214c40b2956Swdenk lba = dev_desc->lba; 215affae2bfSwdenk 216c40b2956Swdenk lba512 = (lba * (dev_desc->blksz/512)); 217affae2bfSwdenk /* round to 1 digit */ 21869a2a4d9SSergei Trofimovich mb = lba512_muldiv(lba512, 10, 2048); /* 2048 = (1024 * 1024) / 512 MB */ 21969a2a4d9SSergei Trofimovich 220affae2bfSwdenk mb_quot = mb / 10; 221affae2bfSwdenk mb_rem = mb - (10 * mb_quot); 222affae2bfSwdenk 223affae2bfSwdenk gb = mb / 1024; 224affae2bfSwdenk gb_quot = gb / 10; 225affae2bfSwdenk gb_rem = gb - (10 * gb_quot); 22642dfe7a1Swdenk #ifdef CONFIG_LBA48 2276e592385Swdenk if (dev_desc->lba48) 228c40b2956Swdenk printf (" Supports 48-bit addressing\n"); 229c40b2956Swdenk #endif 2304b142febSHeiko Schocher #if defined(CONFIG_SYS_64BIT_LBA) 2316c6166f5SMike Frysinger printf (" Capacity: %ld.%ld MB = %ld.%ld GB (%Ld x %ld)\n", 232c40b2956Swdenk mb_quot, mb_rem, 233c40b2956Swdenk gb_quot, gb_rem, 234c40b2956Swdenk lba, 235c40b2956Swdenk dev_desc->blksz); 236c40b2956Swdenk #else 237affae2bfSwdenk printf (" Capacity: %ld.%ld MB = %ld.%ld GB (%ld x %ld)\n", 238affae2bfSwdenk mb_quot, mb_rem, 239affae2bfSwdenk gb_quot, gb_rem, 240c40b2956Swdenk (ulong)lba, 241affae2bfSwdenk dev_desc->blksz); 242c40b2956Swdenk #endif 243affae2bfSwdenk } else { 244affae2bfSwdenk puts (" Capacity: not available\n"); 245affae2bfSwdenk } 246affae2bfSwdenk } 247b3aff0cbSJon Loeliger #endif 248affae2bfSwdenk 2490c9c8fb5SGabe Black #ifdef HAVE_BLOCK_DEVICE 250affae2bfSwdenk 251affae2bfSwdenk void init_part (block_dev_desc_t * dev_desc) 252affae2bfSwdenk { 253affae2bfSwdenk #ifdef CONFIG_ISO_PARTITION 254affae2bfSwdenk if (test_part_iso(dev_desc) == 0) { 255affae2bfSwdenk dev_desc->part_type = PART_TYPE_ISO; 256affae2bfSwdenk return; 257affae2bfSwdenk } 258affae2bfSwdenk #endif 259affae2bfSwdenk 260affae2bfSwdenk #ifdef CONFIG_MAC_PARTITION 261affae2bfSwdenk if (test_part_mac(dev_desc) == 0) { 262affae2bfSwdenk dev_desc->part_type = PART_TYPE_MAC; 263affae2bfSwdenk return; 264affae2bfSwdenk } 265affae2bfSwdenk #endif 266affae2bfSwdenk 26707f3d789Srichardretanubun /* must be placed before DOS partition detection */ 26807f3d789Srichardretanubun #ifdef CONFIG_EFI_PARTITION 26907f3d789Srichardretanubun if (test_part_efi(dev_desc) == 0) { 27007f3d789Srichardretanubun dev_desc->part_type = PART_TYPE_EFI; 27107f3d789Srichardretanubun return; 27207f3d789Srichardretanubun } 27307f3d789Srichardretanubun #endif 27407f3d789Srichardretanubun 275affae2bfSwdenk #ifdef CONFIG_DOS_PARTITION 276affae2bfSwdenk if (test_part_dos(dev_desc) == 0) { 277affae2bfSwdenk dev_desc->part_type = PART_TYPE_DOS; 278affae2bfSwdenk return; 279affae2bfSwdenk } 280affae2bfSwdenk #endif 281c7de829cSwdenk 282c7de829cSwdenk #ifdef CONFIG_AMIGA_PARTITION 283c7de829cSwdenk if (test_part_amiga(dev_desc) == 0) { 284c7de829cSwdenk dev_desc->part_type = PART_TYPE_AMIGA; 285c7de829cSwdenk return; 286c7de829cSwdenk } 287c7de829cSwdenk #endif 28899d2c205SRob Herring dev_desc->part_type = PART_TYPE_UNKNOWN; 289affae2bfSwdenk } 290affae2bfSwdenk 291affae2bfSwdenk 2920c9c8fb5SGabe Black #if defined(CONFIG_MAC_PARTITION) || \ 2930c9c8fb5SGabe Black defined(CONFIG_DOS_PARTITION) || \ 2940c9c8fb5SGabe Black defined(CONFIG_ISO_PARTITION) || \ 2950c9c8fb5SGabe Black defined(CONFIG_AMIGA_PARTITION) || \ 2960c9c8fb5SGabe Black defined(CONFIG_EFI_PARTITION) 2970c9c8fb5SGabe Black 298affae2bfSwdenk static void print_part_header (const char *type, block_dev_desc_t * dev_desc) 299affae2bfSwdenk { 300affae2bfSwdenk puts ("\nPartition Map for "); 301affae2bfSwdenk switch (dev_desc->if_type) { 302726c0f1eSDetlev Zundel case IF_TYPE_IDE: 303726c0f1eSDetlev Zundel puts ("IDE"); 304affae2bfSwdenk break; 305726c0f1eSDetlev Zundel case IF_TYPE_SATA: 306726c0f1eSDetlev Zundel puts ("SATA"); 307c7057b52SDave Liu break; 308726c0f1eSDetlev Zundel case IF_TYPE_SCSI: 309726c0f1eSDetlev Zundel puts ("SCSI"); 310affae2bfSwdenk break; 311726c0f1eSDetlev Zundel case IF_TYPE_ATAPI: 312726c0f1eSDetlev Zundel puts ("ATAPI"); 313affae2bfSwdenk break; 314726c0f1eSDetlev Zundel case IF_TYPE_USB: 315726c0f1eSDetlev Zundel puts ("USB"); 316affae2bfSwdenk break; 317726c0f1eSDetlev Zundel case IF_TYPE_DOC: 318726c0f1eSDetlev Zundel puts ("DOC"); 319affae2bfSwdenk break; 3208f3b9642SLei Wen case IF_TYPE_MMC: 3218f3b9642SLei Wen puts ("MMC"); 3228f3b9642SLei Wen break; 323f4d8de48SHenrik Nordström case IF_TYPE_HOST: 324f4d8de48SHenrik Nordström puts("HOST"); 325f4d8de48SHenrik Nordström break; 326726c0f1eSDetlev Zundel default: 327726c0f1eSDetlev Zundel puts ("UNKNOWN"); 328affae2bfSwdenk break; 329affae2bfSwdenk } 330affae2bfSwdenk printf (" device %d -- Partition Type: %s\n\n", 331affae2bfSwdenk dev_desc->dev, type); 332affae2bfSwdenk } 333affae2bfSwdenk 3340c9c8fb5SGabe Black #endif /* any CONFIG_..._PARTITION */ 3350c9c8fb5SGabe Black 336affae2bfSwdenk void print_part (block_dev_desc_t * dev_desc) 337affae2bfSwdenk { 338affae2bfSwdenk 339affae2bfSwdenk switch (dev_desc->part_type) { 340affae2bfSwdenk #ifdef CONFIG_MAC_PARTITION 341affae2bfSwdenk case PART_TYPE_MAC: 342affae2bfSwdenk PRINTF ("## Testing for valid MAC partition ##\n"); 343affae2bfSwdenk print_part_header ("MAC", dev_desc); 344affae2bfSwdenk print_part_mac (dev_desc); 345affae2bfSwdenk return; 346affae2bfSwdenk #endif 347affae2bfSwdenk #ifdef CONFIG_DOS_PARTITION 348affae2bfSwdenk case PART_TYPE_DOS: 349affae2bfSwdenk PRINTF ("## Testing for valid DOS partition ##\n"); 350affae2bfSwdenk print_part_header ("DOS", dev_desc); 351affae2bfSwdenk print_part_dos (dev_desc); 352affae2bfSwdenk return; 353affae2bfSwdenk #endif 354affae2bfSwdenk 355affae2bfSwdenk #ifdef CONFIG_ISO_PARTITION 356affae2bfSwdenk case PART_TYPE_ISO: 357affae2bfSwdenk PRINTF ("## Testing for valid ISO Boot partition ##\n"); 358affae2bfSwdenk print_part_header ("ISO", dev_desc); 359affae2bfSwdenk print_part_iso (dev_desc); 360affae2bfSwdenk return; 361affae2bfSwdenk #endif 362c7de829cSwdenk 363c7de829cSwdenk #ifdef CONFIG_AMIGA_PARTITION 364c7de829cSwdenk case PART_TYPE_AMIGA: 365c7de829cSwdenk PRINTF ("## Testing for a valid Amiga partition ##\n"); 366c7de829cSwdenk print_part_header ("AMIGA", dev_desc); 367c7de829cSwdenk print_part_amiga (dev_desc); 368c7de829cSwdenk return; 369c7de829cSwdenk #endif 37007f3d789Srichardretanubun 37107f3d789Srichardretanubun #ifdef CONFIG_EFI_PARTITION 37207f3d789Srichardretanubun case PART_TYPE_EFI: 37307f3d789Srichardretanubun PRINTF ("## Testing for valid EFI partition ##\n"); 37407f3d789Srichardretanubun print_part_header ("EFI", dev_desc); 37507f3d789Srichardretanubun print_part_efi (dev_desc); 37607f3d789Srichardretanubun return; 37707f3d789Srichardretanubun #endif 378affae2bfSwdenk } 379affae2bfSwdenk puts ("## Unknown partition table\n"); 380affae2bfSwdenk } 381affae2bfSwdenk 3820c9c8fb5SGabe Black #endif /* HAVE_BLOCK_DEVICE */ 3832f501646SStephen Warren 3842f501646SStephen Warren int get_partition_info(block_dev_desc_t *dev_desc, int part 3852f501646SStephen Warren , disk_partition_t *info) 3862f501646SStephen Warren { 3870c9c8fb5SGabe Black #ifdef HAVE_BLOCK_DEVICE 3882f501646SStephen Warren 389894bfbbfSStephen Warren #ifdef CONFIG_PARTITION_UUIDS 390894bfbbfSStephen Warren /* The common case is no UUID support */ 391894bfbbfSStephen Warren info->uuid[0] = 0; 392894bfbbfSStephen Warren #endif 393894bfbbfSStephen Warren 3942f501646SStephen Warren switch (dev_desc->part_type) { 3952f501646SStephen Warren #ifdef CONFIG_MAC_PARTITION 3962f501646SStephen Warren case PART_TYPE_MAC: 3972f501646SStephen Warren if (get_partition_info_mac(dev_desc, part, info) == 0) { 3982f501646SStephen Warren PRINTF("## Valid MAC partition found ##\n"); 3992f501646SStephen Warren return 0; 4002f501646SStephen Warren } 4012f501646SStephen Warren break; 4022f501646SStephen Warren #endif 4032f501646SStephen Warren 4042f501646SStephen Warren #ifdef CONFIG_DOS_PARTITION 4052f501646SStephen Warren case PART_TYPE_DOS: 4062f501646SStephen Warren if (get_partition_info_dos(dev_desc, part, info) == 0) { 4072f501646SStephen Warren PRINTF("## Valid DOS partition found ##\n"); 4082f501646SStephen Warren return 0; 4092f501646SStephen Warren } 4102f501646SStephen Warren break; 4112f501646SStephen Warren #endif 4122f501646SStephen Warren 4132f501646SStephen Warren #ifdef CONFIG_ISO_PARTITION 4142f501646SStephen Warren case PART_TYPE_ISO: 4152f501646SStephen Warren if (get_partition_info_iso(dev_desc, part, info) == 0) { 4162f501646SStephen Warren PRINTF("## Valid ISO boot partition found ##\n"); 4172f501646SStephen Warren return 0; 4182f501646SStephen Warren } 4192f501646SStephen Warren break; 4202f501646SStephen Warren #endif 4212f501646SStephen Warren 4222f501646SStephen Warren #ifdef CONFIG_AMIGA_PARTITION 4232f501646SStephen Warren case PART_TYPE_AMIGA: 4242f501646SStephen Warren if (get_partition_info_amiga(dev_desc, part, info) == 0) { 4252f501646SStephen Warren PRINTF("## Valid Amiga partition found ##\n"); 4262f501646SStephen Warren return 0; 4272f501646SStephen Warren } 4282f501646SStephen Warren break; 4292f501646SStephen Warren #endif 4302f501646SStephen Warren 4312f501646SStephen Warren #ifdef CONFIG_EFI_PARTITION 4322f501646SStephen Warren case PART_TYPE_EFI: 4332f501646SStephen Warren if (get_partition_info_efi(dev_desc, part, info) == 0) { 4342f501646SStephen Warren PRINTF("## Valid EFI partition found ##\n"); 4352f501646SStephen Warren return 0; 4362f501646SStephen Warren } 4372f501646SStephen Warren break; 4382f501646SStephen Warren #endif 4392f501646SStephen Warren default: 4402f501646SStephen Warren break; 4412f501646SStephen Warren } 4420c9c8fb5SGabe Black #endif /* HAVE_BLOCK_DEVICE */ 4432f501646SStephen Warren 4442f501646SStephen Warren return -1; 4452f501646SStephen Warren } 44699d2c205SRob Herring 447336b6f90SStephen Warren int get_device(const char *ifname, const char *dev_hwpart_str, 4482023e608SStephen Warren block_dev_desc_t **dev_desc) 4492023e608SStephen Warren { 4502023e608SStephen Warren char *ep; 451336b6f90SStephen Warren char *dup_str = NULL; 452336b6f90SStephen Warren const char *dev_str, *hwpart_str; 453336b6f90SStephen Warren int dev, hwpart; 454336b6f90SStephen Warren 455336b6f90SStephen Warren hwpart_str = strchr(dev_hwpart_str, '.'); 456336b6f90SStephen Warren if (hwpart_str) { 457336b6f90SStephen Warren dup_str = strdup(dev_hwpart_str); 458336b6f90SStephen Warren dup_str[hwpart_str - dev_hwpart_str] = 0; 459336b6f90SStephen Warren dev_str = dup_str; 460336b6f90SStephen Warren hwpart_str++; 461336b6f90SStephen Warren } else { 462336b6f90SStephen Warren dev_str = dev_hwpart_str; 463ecdd57e2SStephen Warren hwpart = 0; 464336b6f90SStephen Warren } 4652023e608SStephen Warren 4662023e608SStephen Warren dev = simple_strtoul(dev_str, &ep, 16); 4672023e608SStephen Warren if (*ep) { 4682023e608SStephen Warren printf("** Bad device specification %s %s **\n", 4692023e608SStephen Warren ifname, dev_str); 470336b6f90SStephen Warren dev = -1; 471336b6f90SStephen Warren goto cleanup; 4722023e608SStephen Warren } 4732023e608SStephen Warren 474336b6f90SStephen Warren if (hwpart_str) { 475336b6f90SStephen Warren hwpart = simple_strtoul(hwpart_str, &ep, 16); 476336b6f90SStephen Warren if (*ep) { 477336b6f90SStephen Warren printf("** Bad HW partition specification %s %s **\n", 478336b6f90SStephen Warren ifname, hwpart_str); 479336b6f90SStephen Warren dev = -1; 480336b6f90SStephen Warren goto cleanup; 481336b6f90SStephen Warren } 482336b6f90SStephen Warren } 483336b6f90SStephen Warren 484336b6f90SStephen Warren *dev_desc = get_dev_hwpart(ifname, dev, hwpart); 4852023e608SStephen Warren if (!(*dev_desc) || ((*dev_desc)->type == DEV_TYPE_UNKNOWN)) { 486336b6f90SStephen Warren printf("** Bad device %s %s **\n", ifname, dev_hwpart_str); 487336b6f90SStephen Warren dev = -1; 488336b6f90SStephen Warren goto cleanup; 4892023e608SStephen Warren } 4902023e608SStephen Warren 491336b6f90SStephen Warren cleanup: 492336b6f90SStephen Warren free(dup_str); 4932023e608SStephen Warren return dev; 4942023e608SStephen Warren } 4952023e608SStephen Warren 49610a37fd7SStephen Warren #define PART_UNSPECIFIED -2 49710a37fd7SStephen Warren #define PART_AUTO -1 49810a37fd7SStephen Warren #define MAX_SEARCH_PARTITIONS 16 49910a37fd7SStephen Warren int get_device_and_partition(const char *ifname, const char *dev_part_str, 50099d2c205SRob Herring block_dev_desc_t **dev_desc, 50110a37fd7SStephen Warren disk_partition_t *info, int allow_whole_dev) 50299d2c205SRob Herring { 50310a37fd7SStephen Warren int ret = -1; 50410a37fd7SStephen Warren const char *part_str; 50510a37fd7SStephen Warren char *dup_str = NULL; 50610a37fd7SStephen Warren const char *dev_str; 50799d2c205SRob Herring int dev; 50810a37fd7SStephen Warren char *ep; 50910a37fd7SStephen Warren int p; 51010a37fd7SStephen Warren int part; 51110a37fd7SStephen Warren disk_partition_t tmpinfo; 51299d2c205SRob Herring 513*4d907025SStephen Warren /* 514*4d907025SStephen Warren * Special-case a psuedo block device "hostfs", to allow access to the 515*4d907025SStephen Warren * host's own filesystem. 516*4d907025SStephen Warren */ 517*4d907025SStephen Warren if (0 == strcmp(ifname, "hostfs")) { 518*4d907025SStephen Warren *dev_desc = NULL; 519*4d907025SStephen Warren info->start = 0; 520*4d907025SStephen Warren info->size = 0; 521*4d907025SStephen Warren info->blksz = 0; 522*4d907025SStephen Warren info->bootable = 0; 523*4d907025SStephen Warren strcpy((char *)info->type, BOOT_PART_TYPE); 524*4d907025SStephen Warren strcpy((char *)info->name, "Sandbox host"); 525*4d907025SStephen Warren #ifdef CONFIG_PARTITION_UUIDS 526*4d907025SStephen Warren info->uuid[0] = 0; 527*4d907025SStephen Warren #endif 528*4d907025SStephen Warren 529*4d907025SStephen Warren return 0; 530*4d907025SStephen Warren } 531*4d907025SStephen Warren 53210a37fd7SStephen Warren /* If no dev_part_str, use bootdevice environment variable */ 533a10973e7SStephen Warren if (!dev_part_str || !strlen(dev_part_str) || 534a10973e7SStephen Warren !strcmp(dev_part_str, "-")) 53510a37fd7SStephen Warren dev_part_str = getenv("bootdevice"); 53699d2c205SRob Herring 53710a37fd7SStephen Warren /* If still no dev_part_str, it's an error */ 53810a37fd7SStephen Warren if (!dev_part_str) { 53910a37fd7SStephen Warren printf("** No device specified **\n"); 54010a37fd7SStephen Warren goto cleanup; 54199d2c205SRob Herring } 54299d2c205SRob Herring 54310a37fd7SStephen Warren /* Separate device and partition ID specification */ 54410a37fd7SStephen Warren part_str = strchr(dev_part_str, ':'); 54510a37fd7SStephen Warren if (part_str) { 54610a37fd7SStephen Warren dup_str = strdup(dev_part_str); 54710a37fd7SStephen Warren dup_str[part_str - dev_part_str] = 0; 54810a37fd7SStephen Warren dev_str = dup_str; 54910a37fd7SStephen Warren part_str++; 55010a37fd7SStephen Warren } else { 55110a37fd7SStephen Warren dev_str = dev_part_str; 55299d2c205SRob Herring } 55310a37fd7SStephen Warren 55410a37fd7SStephen Warren /* Look up the device */ 55510a37fd7SStephen Warren dev = get_device(ifname, dev_str, dev_desc); 55610a37fd7SStephen Warren if (dev < 0) 55710a37fd7SStephen Warren goto cleanup; 55810a37fd7SStephen Warren 55910a37fd7SStephen Warren /* Convert partition ID string to number */ 56010a37fd7SStephen Warren if (!part_str || !*part_str) { 56110a37fd7SStephen Warren part = PART_UNSPECIFIED; 56210a37fd7SStephen Warren } else if (!strcmp(part_str, "auto")) { 56310a37fd7SStephen Warren part = PART_AUTO; 56410a37fd7SStephen Warren } else { 56510a37fd7SStephen Warren /* Something specified -> use exactly that */ 56610a37fd7SStephen Warren part = (int)simple_strtoul(part_str, &ep, 16); 56710a37fd7SStephen Warren /* 56810a37fd7SStephen Warren * Less than whole string converted, 56910a37fd7SStephen Warren * or request for whole device, but caller requires partition. 57010a37fd7SStephen Warren */ 57110a37fd7SStephen Warren if (*ep || (part == 0 && !allow_whole_dev)) { 57210a37fd7SStephen Warren printf("** Bad partition specification %s %s **\n", 57310a37fd7SStephen Warren ifname, dev_part_str); 57410a37fd7SStephen Warren goto cleanup; 57510a37fd7SStephen Warren } 57610a37fd7SStephen Warren } 57710a37fd7SStephen Warren 57810a37fd7SStephen Warren /* 57910a37fd7SStephen Warren * No partition table on device, 58010a37fd7SStephen Warren * or user requested partition 0 (entire device). 58110a37fd7SStephen Warren */ 58210a37fd7SStephen Warren if (((*dev_desc)->part_type == PART_TYPE_UNKNOWN) || 58310a37fd7SStephen Warren (part == 0)) { 58410a37fd7SStephen Warren if (!(*dev_desc)->lba) { 58510a37fd7SStephen Warren printf("** Bad device size - %s %s **\n", ifname, 58610a37fd7SStephen Warren dev_str); 58710a37fd7SStephen Warren goto cleanup; 58810a37fd7SStephen Warren } 58910a37fd7SStephen Warren 59010a37fd7SStephen Warren /* 59110a37fd7SStephen Warren * If user specified a partition ID other than 0, 59210a37fd7SStephen Warren * or the calling command only accepts partitions, 59310a37fd7SStephen Warren * it's an error. 59410a37fd7SStephen Warren */ 59510a37fd7SStephen Warren if ((part > 0) || (!allow_whole_dev)) { 59610a37fd7SStephen Warren printf("** No partition table - %s %s **\n", ifname, 59710a37fd7SStephen Warren dev_str); 59810a37fd7SStephen Warren goto cleanup; 59910a37fd7SStephen Warren } 60010a37fd7SStephen Warren 60150ffc3b6SLan Yixun (dlan) (*dev_desc)->log2blksz = LOG2((*dev_desc)->blksz); 60250ffc3b6SLan Yixun (dlan) 60399d2c205SRob Herring info->start = 0; 60410a37fd7SStephen Warren info->size = (*dev_desc)->lba; 60510a37fd7SStephen Warren info->blksz = (*dev_desc)->blksz; 60610a37fd7SStephen Warren info->bootable = 0; 6076ab6a650SStephen Warren strcpy((char *)info->type, BOOT_PART_TYPE); 6086ab6a650SStephen Warren strcpy((char *)info->name, "Whole Disk"); 60910a37fd7SStephen Warren #ifdef CONFIG_PARTITION_UUIDS 61010a37fd7SStephen Warren info->uuid[0] = 0; 61110a37fd7SStephen Warren #endif 61299d2c205SRob Herring 61310a37fd7SStephen Warren ret = 0; 61410a37fd7SStephen Warren goto cleanup; 61599d2c205SRob Herring } 61699d2c205SRob Herring 61710a37fd7SStephen Warren /* 61810a37fd7SStephen Warren * Now there's known to be a partition table, 61910a37fd7SStephen Warren * not specifying a partition means to pick partition 1. 62010a37fd7SStephen Warren */ 62110a37fd7SStephen Warren if (part == PART_UNSPECIFIED) 62210a37fd7SStephen Warren part = 1; 62399d2c205SRob Herring 62410a37fd7SStephen Warren /* 62510a37fd7SStephen Warren * If user didn't specify a partition number, or did specify something 62610a37fd7SStephen Warren * other than "auto", use that partition number directly. 62710a37fd7SStephen Warren */ 62810a37fd7SStephen Warren if (part != PART_AUTO) { 62910a37fd7SStephen Warren ret = get_partition_info(*dev_desc, part, info); 63099d2c205SRob Herring if (ret) { 63110a37fd7SStephen Warren printf("** Invalid partition %d **\n", part); 63210a37fd7SStephen Warren goto cleanup; 63310a37fd7SStephen Warren } 63410a37fd7SStephen Warren } else { 63510a37fd7SStephen Warren /* 63610a37fd7SStephen Warren * Find the first bootable partition. 63710a37fd7SStephen Warren * If none are bootable, fall back to the first valid partition. 63810a37fd7SStephen Warren */ 63910a37fd7SStephen Warren part = 0; 64010a37fd7SStephen Warren for (p = 1; p <= MAX_SEARCH_PARTITIONS; p++) { 64110a37fd7SStephen Warren ret = get_partition_info(*dev_desc, p, info); 64210a37fd7SStephen Warren if (ret) 64310a37fd7SStephen Warren continue; 64410a37fd7SStephen Warren 64510a37fd7SStephen Warren /* 64610a37fd7SStephen Warren * First valid partition, or new better partition? 64710a37fd7SStephen Warren * If so, save partition ID. 64810a37fd7SStephen Warren */ 64910a37fd7SStephen Warren if (!part || info->bootable) 65010a37fd7SStephen Warren part = p; 65110a37fd7SStephen Warren 65210a37fd7SStephen Warren /* Best possible partition? Stop searching. */ 65310a37fd7SStephen Warren if (info->bootable) 65410a37fd7SStephen Warren break; 65510a37fd7SStephen Warren 65610a37fd7SStephen Warren /* 65710a37fd7SStephen Warren * We now need to search further for best possible. 65810a37fd7SStephen Warren * If we what we just queried was the best so far, 65910a37fd7SStephen Warren * save the info since we over-write it next loop. 66010a37fd7SStephen Warren */ 66110a37fd7SStephen Warren if (part == p) 66210a37fd7SStephen Warren tmpinfo = *info; 66310a37fd7SStephen Warren } 66410a37fd7SStephen Warren /* If we found any acceptable partition */ 66510a37fd7SStephen Warren if (part) { 66610a37fd7SStephen Warren /* 66710a37fd7SStephen Warren * If we searched all possible partition IDs, 66810a37fd7SStephen Warren * return the first valid partition we found. 66910a37fd7SStephen Warren */ 67010a37fd7SStephen Warren if (p == MAX_SEARCH_PARTITIONS + 1) 67110a37fd7SStephen Warren *info = tmpinfo; 67210a37fd7SStephen Warren } else { 67310a37fd7SStephen Warren printf("** No valid partitions found **\n"); 67471bba424SStephen Warren ret = -1; 67510a37fd7SStephen Warren goto cleanup; 67610a37fd7SStephen Warren } 67799d2c205SRob Herring } 67899d2c205SRob Herring if (strncmp((char *)info->type, BOOT_PART_TYPE, sizeof(info->type)) != 0) { 67999d2c205SRob Herring printf("** Invalid partition type \"%.32s\"" 68099d2c205SRob Herring " (expect \"" BOOT_PART_TYPE "\")\n", 68199d2c205SRob Herring info->type); 68210a37fd7SStephen Warren ret = -1; 68310a37fd7SStephen Warren goto cleanup; 68499d2c205SRob Herring } 68599d2c205SRob Herring 68650ffc3b6SLan Yixun (dlan) (*dev_desc)->log2blksz = LOG2((*dev_desc)->blksz); 68750ffc3b6SLan Yixun (dlan) 68810a37fd7SStephen Warren ret = part; 68910a37fd7SStephen Warren goto cleanup; 69099d2c205SRob Herring 69110a37fd7SStephen Warren cleanup: 69210a37fd7SStephen Warren free(dup_str); 69310a37fd7SStephen Warren return ret; 69499d2c205SRob Herring } 695