1affae2bfSwdenk /* 2affae2bfSwdenk * (C) Copyright 2001 3affae2bfSwdenk * Wolfgang Denk, DENX Software Engineering, wd@denx.de. 4affae2bfSwdenk * 5affae2bfSwdenk * See file CREDITS for list of people who contributed to this 6affae2bfSwdenk * project. 7affae2bfSwdenk * 8affae2bfSwdenk * This program is free software; you can redistribute it and/or 9affae2bfSwdenk * modify it under the terms of the GNU General Public License as 10affae2bfSwdenk * published by the Free Software Foundation; either version 2 of 11affae2bfSwdenk * the License, or (at your option) any later version. 12affae2bfSwdenk * 13affae2bfSwdenk * This program is distributed in the hope that it will be useful, 14affae2bfSwdenk * but WITHOUT ANY WARRANTY; without even the implied warranty of 15affae2bfSwdenk * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 16affae2bfSwdenk * GNU General Public License for more details. 17affae2bfSwdenk * 18affae2bfSwdenk * You should have received a copy of the GNU General Public License 19affae2bfSwdenk * along with this program; if not, write to the Free Software 20affae2bfSwdenk * Foundation, Inc., 59 Temple Place, Suite 330, Boston, 21affae2bfSwdenk * MA 02111-1307 USA 22affae2bfSwdenk */ 23affae2bfSwdenk 24affae2bfSwdenk #include <common.h> 25affae2bfSwdenk #include <command.h> 26affae2bfSwdenk #include <ide.h> 2710a37fd7SStephen Warren #include <malloc.h> 28735dd97bSGrant Likely #include <part.h> 29affae2bfSwdenk 30affae2bfSwdenk #undef PART_DEBUG 31affae2bfSwdenk 32affae2bfSwdenk #ifdef PART_DEBUG 33affae2bfSwdenk #define PRINTF(fmt,args...) printf (fmt ,##args) 34affae2bfSwdenk #else 35affae2bfSwdenk #define PRINTF(fmt,args...) 36affae2bfSwdenk #endif 37affae2bfSwdenk 38cde5c64dSJon Loeliger #if (defined(CONFIG_CMD_IDE) || \ 39c7057b52SDave Liu defined(CONFIG_CMD_SATA) || \ 40cde5c64dSJon Loeliger defined(CONFIG_CMD_SCSI) || \ 41cde5c64dSJon Loeliger defined(CONFIG_CMD_USB) || \ 421968e615Swdenk defined(CONFIG_MMC) || \ 431968e615Swdenk defined(CONFIG_SYSTEMACE) ) 44affae2bfSwdenk 45735dd97bSGrant Likely struct block_drvr { 46735dd97bSGrant Likely char *name; 47735dd97bSGrant Likely block_dev_desc_t* (*get_dev)(int dev); 48735dd97bSGrant Likely }; 49735dd97bSGrant Likely 50735dd97bSGrant Likely static const struct block_drvr block_drvr[] = { 51cde5c64dSJon Loeliger #if defined(CONFIG_CMD_IDE) 52735dd97bSGrant Likely { .name = "ide", .get_dev = ide_get_dev, }, 53735dd97bSGrant Likely #endif 54c7057b52SDave Liu #if defined(CONFIG_CMD_SATA) 55c7057b52SDave Liu {.name = "sata", .get_dev = sata_get_dev, }, 56c7057b52SDave Liu #endif 57cde5c64dSJon Loeliger #if defined(CONFIG_CMD_SCSI) 58735dd97bSGrant Likely { .name = "scsi", .get_dev = scsi_get_dev, }, 59735dd97bSGrant Likely #endif 60cde5c64dSJon Loeliger #if defined(CONFIG_CMD_USB) && defined(CONFIG_USB_STORAGE) 61735dd97bSGrant Likely { .name = "usb", .get_dev = usb_stor_get_dev, }, 62735dd97bSGrant Likely #endif 63735dd97bSGrant Likely #if defined(CONFIG_MMC) 64735dd97bSGrant Likely { .name = "mmc", .get_dev = mmc_get_dev, }, 65735dd97bSGrant Likely #endif 66735dd97bSGrant Likely #if defined(CONFIG_SYSTEMACE) 67735dd97bSGrant Likely { .name = "ace", .get_dev = systemace_get_dev, }, 68735dd97bSGrant Likely #endif 69735dd97bSGrant Likely { }, 70735dd97bSGrant Likely }; 71735dd97bSGrant Likely 72751bb571SStefan Roese DECLARE_GLOBAL_DATA_PTR; 73751bb571SStefan Roese 7499d2c205SRob Herring block_dev_desc_t *get_dev(const char *ifname, int dev) 75735dd97bSGrant Likely { 76735dd97bSGrant Likely const struct block_drvr *drvr = block_drvr; 77751bb571SStefan Roese block_dev_desc_t* (*reloc_get_dev)(int dev); 7823090dacSHeiko Schocher char *name; 79751bb571SStefan Roese 807e71dc68STim Kientzle if (!ifname) 817e71dc68STim Kientzle return NULL; 827e71dc68STim Kientzle 8323090dacSHeiko Schocher name = drvr->name; 842e5167ccSWolfgang Denk #ifdef CONFIG_NEEDS_MANUAL_RELOC 8523090dacSHeiko Schocher name += gd->reloc_off; 8623090dacSHeiko Schocher #endif 87b16aadf4SLei Wen while (drvr->name) { 8823090dacSHeiko Schocher name = drvr->name; 89521af04dSPeter Tyser reloc_get_dev = drvr->get_dev; 902e5167ccSWolfgang Denk #ifdef CONFIG_NEEDS_MANUAL_RELOC 9123090dacSHeiko Schocher name += gd->reloc_off; 92521af04dSPeter Tyser reloc_get_dev += gd->reloc_off; 93521af04dSPeter Tyser #endif 9423090dacSHeiko Schocher if (strncmp(ifname, name, strlen(name)) == 0) 95751bb571SStefan Roese return reloc_get_dev(dev); 96735dd97bSGrant Likely drvr++; 97735dd97bSGrant Likely } 98735dd97bSGrant Likely return NULL; 99735dd97bSGrant Likely } 100735dd97bSGrant Likely #else 10199d2c205SRob Herring block_dev_desc_t *get_dev(const char *ifname, int dev) 102735dd97bSGrant Likely { 103735dd97bSGrant Likely return NULL; 104735dd97bSGrant Likely } 105735dd97bSGrant Likely #endif 106735dd97bSGrant Likely 107cde5c64dSJon Loeliger #if (defined(CONFIG_CMD_IDE) || \ 108c7057b52SDave Liu defined(CONFIG_CMD_SATA) || \ 109cde5c64dSJon Loeliger defined(CONFIG_CMD_SCSI) || \ 110cde5c64dSJon Loeliger defined(CONFIG_CMD_USB) || \ 111735dd97bSGrant Likely defined(CONFIG_MMC) || \ 112735dd97bSGrant Likely defined(CONFIG_SYSTEMACE) ) 113735dd97bSGrant Likely 114affae2bfSwdenk /* ------------------------------------------------------------------------- */ 115affae2bfSwdenk /* 116affae2bfSwdenk * reports device info to the user 117affae2bfSwdenk */ 11869a2a4d9SSergei Trofimovich 11969a2a4d9SSergei Trofimovich #ifdef CONFIG_LBA48 12069a2a4d9SSergei Trofimovich typedef uint64_t lba512_t; 12169a2a4d9SSergei Trofimovich #else 12269a2a4d9SSergei Trofimovich typedef lbaint_t lba512_t; 12369a2a4d9SSergei Trofimovich #endif 12469a2a4d9SSergei Trofimovich 12569a2a4d9SSergei Trofimovich /* 12669a2a4d9SSergei Trofimovich * Overflowless variant of (block_count * mul_by / div_by) 12769a2a4d9SSergei Trofimovich * when div_by > mul_by 12869a2a4d9SSergei Trofimovich */ 12969a2a4d9SSergei Trofimovich static lba512_t lba512_muldiv (lba512_t block_count, lba512_t mul_by, lba512_t div_by) 13069a2a4d9SSergei Trofimovich { 13169a2a4d9SSergei Trofimovich lba512_t bc_quot, bc_rem; 13269a2a4d9SSergei Trofimovich 13369a2a4d9SSergei Trofimovich /* x * m / d == x / d * m + (x % d) * m / d */ 13469a2a4d9SSergei Trofimovich bc_quot = block_count / div_by; 13569a2a4d9SSergei Trofimovich bc_rem = block_count - div_by * bc_quot; 13669a2a4d9SSergei Trofimovich return bc_quot * mul_by + (bc_rem * mul_by) / div_by; 13769a2a4d9SSergei Trofimovich } 13869a2a4d9SSergei Trofimovich 139affae2bfSwdenk void dev_print (block_dev_desc_t *dev_desc) 140affae2bfSwdenk { 14169a2a4d9SSergei Trofimovich lba512_t lba512; /* number of blocks if 512bytes block size */ 142affae2bfSwdenk 143af75a45dSWolfgang Denk if (dev_desc->type == DEV_TYPE_UNKNOWN) { 144af75a45dSWolfgang Denk puts ("not available\n"); 145af75a45dSWolfgang Denk return; 146af75a45dSWolfgang Denk } 147af75a45dSWolfgang Denk 1488ec6e332STor Krill switch (dev_desc->if_type) { 149574b3195SDetlev Zundel case IF_TYPE_SCSI: 150574b3195SDetlev Zundel printf ("(%d:%d) Vendor: %s Prod.: %s Rev: %s\n", 151574b3195SDetlev Zundel dev_desc->target,dev_desc->lun, 152affae2bfSwdenk dev_desc->vendor, 153affae2bfSwdenk dev_desc->product, 154affae2bfSwdenk dev_desc->revision); 155574b3195SDetlev Zundel break; 1566e24a1ebSRemy Bohmer case IF_TYPE_ATAPI: 157574b3195SDetlev Zundel case IF_TYPE_IDE: 158574b3195SDetlev Zundel case IF_TYPE_SATA: 159574b3195SDetlev Zundel printf ("Model: %s Firm: %s Ser#: %s\n", 160574b3195SDetlev Zundel dev_desc->vendor, 161574b3195SDetlev Zundel dev_desc->revision, 162574b3195SDetlev Zundel dev_desc->product); 163574b3195SDetlev Zundel break; 1646e24a1ebSRemy Bohmer case IF_TYPE_SD: 1656e24a1ebSRemy Bohmer case IF_TYPE_MMC: 16647bebe34SNícolas Carneiro Lebedenco case IF_TYPE_USB: 16747bebe34SNícolas Carneiro Lebedenco printf ("Vendor: %s Rev: %s Prod: %s\n", 16847bebe34SNícolas Carneiro Lebedenco dev_desc->vendor, 16947bebe34SNícolas Carneiro Lebedenco dev_desc->revision, 17047bebe34SNícolas Carneiro Lebedenco dev_desc->product); 17147bebe34SNícolas Carneiro Lebedenco break; 1726e24a1ebSRemy Bohmer case IF_TYPE_DOC: 1736e24a1ebSRemy Bohmer puts("device type DOC\n"); 1746e24a1ebSRemy Bohmer return; 1758ec6e332STor Krill case IF_TYPE_UNKNOWN: 1766e24a1ebSRemy Bohmer puts("device type unknown\n"); 1776e24a1ebSRemy Bohmer return; 178574b3195SDetlev Zundel default: 1796e24a1ebSRemy Bohmer printf("Unhandled device type: %i\n", dev_desc->if_type); 180574b3195SDetlev Zundel return; 181affae2bfSwdenk } 182affae2bfSwdenk puts (" Type: "); 183affae2bfSwdenk if (dev_desc->removable) 184affae2bfSwdenk puts ("Removable "); 185affae2bfSwdenk switch (dev_desc->type & 0x1F) { 186726c0f1eSDetlev Zundel case DEV_TYPE_HARDDISK: 187726c0f1eSDetlev Zundel puts ("Hard Disk"); 188affae2bfSwdenk break; 189726c0f1eSDetlev Zundel case DEV_TYPE_CDROM: 190726c0f1eSDetlev Zundel puts ("CD ROM"); 191affae2bfSwdenk break; 192726c0f1eSDetlev Zundel case DEV_TYPE_OPDISK: 193726c0f1eSDetlev Zundel puts ("Optical Device"); 194affae2bfSwdenk break; 195726c0f1eSDetlev Zundel case DEV_TYPE_TAPE: 196726c0f1eSDetlev Zundel puts ("Tape"); 197affae2bfSwdenk break; 198726c0f1eSDetlev Zundel default: 199726c0f1eSDetlev Zundel printf ("# %02X #", dev_desc->type & 0x1F); 200affae2bfSwdenk break; 201affae2bfSwdenk } 202affae2bfSwdenk puts ("\n"); 203affae2bfSwdenk if ((dev_desc->lba * dev_desc->blksz)>0L) { 204affae2bfSwdenk ulong mb, mb_quot, mb_rem, gb, gb_quot, gb_rem; 205c40b2956Swdenk lbaint_t lba; 2066e592385Swdenk 207c40b2956Swdenk lba = dev_desc->lba; 208affae2bfSwdenk 209c40b2956Swdenk lba512 = (lba * (dev_desc->blksz/512)); 210affae2bfSwdenk /* round to 1 digit */ 21169a2a4d9SSergei Trofimovich mb = lba512_muldiv(lba512, 10, 2048); /* 2048 = (1024 * 1024) / 512 MB */ 21269a2a4d9SSergei Trofimovich 213affae2bfSwdenk mb_quot = mb / 10; 214affae2bfSwdenk mb_rem = mb - (10 * mb_quot); 215affae2bfSwdenk 216affae2bfSwdenk gb = mb / 1024; 217affae2bfSwdenk gb_quot = gb / 10; 218affae2bfSwdenk gb_rem = gb - (10 * gb_quot); 21942dfe7a1Swdenk #ifdef CONFIG_LBA48 2206e592385Swdenk if (dev_desc->lba48) 221c40b2956Swdenk printf (" Supports 48-bit addressing\n"); 222c40b2956Swdenk #endif 2234b142febSHeiko Schocher #if defined(CONFIG_SYS_64BIT_LBA) 2246c6166f5SMike Frysinger printf (" Capacity: %ld.%ld MB = %ld.%ld GB (%Ld x %ld)\n", 225c40b2956Swdenk mb_quot, mb_rem, 226c40b2956Swdenk gb_quot, gb_rem, 227c40b2956Swdenk lba, 228c40b2956Swdenk dev_desc->blksz); 229c40b2956Swdenk #else 230affae2bfSwdenk printf (" Capacity: %ld.%ld MB = %ld.%ld GB (%ld x %ld)\n", 231affae2bfSwdenk mb_quot, mb_rem, 232affae2bfSwdenk gb_quot, gb_rem, 233c40b2956Swdenk (ulong)lba, 234affae2bfSwdenk dev_desc->blksz); 235c40b2956Swdenk #endif 236affae2bfSwdenk } else { 237affae2bfSwdenk puts (" Capacity: not available\n"); 238affae2bfSwdenk } 239affae2bfSwdenk } 240b3aff0cbSJon Loeliger #endif 241affae2bfSwdenk 242cde5c64dSJon Loeliger #if (defined(CONFIG_CMD_IDE) || \ 243c7057b52SDave Liu defined(CONFIG_CMD_SATA) || \ 244cde5c64dSJon Loeliger defined(CONFIG_CMD_SCSI) || \ 245cde5c64dSJon Loeliger defined(CONFIG_CMD_USB) || \ 2467fac3f69SHaavard Skinnemoen defined(CONFIG_MMC) || \ 2473f85ce27Swdenk defined(CONFIG_SYSTEMACE) ) 248affae2bfSwdenk 249affae2bfSwdenk #if defined(CONFIG_MAC_PARTITION) || \ 250affae2bfSwdenk defined(CONFIG_DOS_PARTITION) || \ 251c7de829cSwdenk defined(CONFIG_ISO_PARTITION) || \ 25207f3d789Srichardretanubun defined(CONFIG_AMIGA_PARTITION) || \ 25307f3d789Srichardretanubun defined(CONFIG_EFI_PARTITION) 254affae2bfSwdenk 255affae2bfSwdenk void init_part (block_dev_desc_t * dev_desc) 256affae2bfSwdenk { 257affae2bfSwdenk #ifdef CONFIG_ISO_PARTITION 258affae2bfSwdenk if (test_part_iso(dev_desc) == 0) { 259affae2bfSwdenk dev_desc->part_type = PART_TYPE_ISO; 260affae2bfSwdenk return; 261affae2bfSwdenk } 262affae2bfSwdenk #endif 263affae2bfSwdenk 264affae2bfSwdenk #ifdef CONFIG_MAC_PARTITION 265affae2bfSwdenk if (test_part_mac(dev_desc) == 0) { 266affae2bfSwdenk dev_desc->part_type = PART_TYPE_MAC; 267affae2bfSwdenk return; 268affae2bfSwdenk } 269affae2bfSwdenk #endif 270affae2bfSwdenk 27107f3d789Srichardretanubun /* must be placed before DOS partition detection */ 27207f3d789Srichardretanubun #ifdef CONFIG_EFI_PARTITION 27307f3d789Srichardretanubun if (test_part_efi(dev_desc) == 0) { 27407f3d789Srichardretanubun dev_desc->part_type = PART_TYPE_EFI; 27507f3d789Srichardretanubun return; 27607f3d789Srichardretanubun } 27707f3d789Srichardretanubun #endif 27807f3d789Srichardretanubun 279affae2bfSwdenk #ifdef CONFIG_DOS_PARTITION 280affae2bfSwdenk if (test_part_dos(dev_desc) == 0) { 281affae2bfSwdenk dev_desc->part_type = PART_TYPE_DOS; 282affae2bfSwdenk return; 283affae2bfSwdenk } 284affae2bfSwdenk #endif 285c7de829cSwdenk 286c7de829cSwdenk #ifdef CONFIG_AMIGA_PARTITION 287c7de829cSwdenk if (test_part_amiga(dev_desc) == 0) { 288c7de829cSwdenk dev_desc->part_type = PART_TYPE_AMIGA; 289c7de829cSwdenk return; 290c7de829cSwdenk } 291c7de829cSwdenk #endif 29299d2c205SRob Herring dev_desc->part_type = PART_TYPE_UNKNOWN; 293affae2bfSwdenk } 294affae2bfSwdenk 295affae2bfSwdenk 296affae2bfSwdenk static void print_part_header (const char *type, block_dev_desc_t * dev_desc) 297affae2bfSwdenk { 298affae2bfSwdenk puts ("\nPartition Map for "); 299affae2bfSwdenk switch (dev_desc->if_type) { 300726c0f1eSDetlev Zundel case IF_TYPE_IDE: 301726c0f1eSDetlev Zundel puts ("IDE"); 302affae2bfSwdenk break; 303726c0f1eSDetlev Zundel case IF_TYPE_SATA: 304726c0f1eSDetlev Zundel puts ("SATA"); 305c7057b52SDave Liu break; 306726c0f1eSDetlev Zundel case IF_TYPE_SCSI: 307726c0f1eSDetlev Zundel puts ("SCSI"); 308affae2bfSwdenk break; 309726c0f1eSDetlev Zundel case IF_TYPE_ATAPI: 310726c0f1eSDetlev Zundel puts ("ATAPI"); 311affae2bfSwdenk break; 312726c0f1eSDetlev Zundel case IF_TYPE_USB: 313726c0f1eSDetlev Zundel puts ("USB"); 314affae2bfSwdenk break; 315726c0f1eSDetlev Zundel case IF_TYPE_DOC: 316726c0f1eSDetlev Zundel puts ("DOC"); 317affae2bfSwdenk break; 3188f3b9642SLei Wen case IF_TYPE_MMC: 3198f3b9642SLei Wen puts ("MMC"); 3208f3b9642SLei Wen break; 321726c0f1eSDetlev Zundel default: 322726c0f1eSDetlev Zundel puts ("UNKNOWN"); 323affae2bfSwdenk break; 324affae2bfSwdenk } 325affae2bfSwdenk printf (" device %d -- Partition Type: %s\n\n", 326affae2bfSwdenk dev_desc->dev, type); 327affae2bfSwdenk } 328affae2bfSwdenk 329affae2bfSwdenk void print_part (block_dev_desc_t * dev_desc) 330affae2bfSwdenk { 331affae2bfSwdenk 332affae2bfSwdenk switch (dev_desc->part_type) { 333affae2bfSwdenk #ifdef CONFIG_MAC_PARTITION 334affae2bfSwdenk case PART_TYPE_MAC: 335affae2bfSwdenk PRINTF ("## Testing for valid MAC partition ##\n"); 336affae2bfSwdenk print_part_header ("MAC", dev_desc); 337affae2bfSwdenk print_part_mac (dev_desc); 338affae2bfSwdenk return; 339affae2bfSwdenk #endif 340affae2bfSwdenk #ifdef CONFIG_DOS_PARTITION 341affae2bfSwdenk case PART_TYPE_DOS: 342affae2bfSwdenk PRINTF ("## Testing for valid DOS partition ##\n"); 343affae2bfSwdenk print_part_header ("DOS", dev_desc); 344affae2bfSwdenk print_part_dos (dev_desc); 345affae2bfSwdenk return; 346affae2bfSwdenk #endif 347affae2bfSwdenk 348affae2bfSwdenk #ifdef CONFIG_ISO_PARTITION 349affae2bfSwdenk case PART_TYPE_ISO: 350affae2bfSwdenk PRINTF ("## Testing for valid ISO Boot partition ##\n"); 351affae2bfSwdenk print_part_header ("ISO", dev_desc); 352affae2bfSwdenk print_part_iso (dev_desc); 353affae2bfSwdenk return; 354affae2bfSwdenk #endif 355c7de829cSwdenk 356c7de829cSwdenk #ifdef CONFIG_AMIGA_PARTITION 357c7de829cSwdenk case PART_TYPE_AMIGA: 358c7de829cSwdenk PRINTF ("## Testing for a valid Amiga partition ##\n"); 359c7de829cSwdenk print_part_header ("AMIGA", dev_desc); 360c7de829cSwdenk print_part_amiga (dev_desc); 361c7de829cSwdenk return; 362c7de829cSwdenk #endif 36307f3d789Srichardretanubun 36407f3d789Srichardretanubun #ifdef CONFIG_EFI_PARTITION 36507f3d789Srichardretanubun case PART_TYPE_EFI: 36607f3d789Srichardretanubun PRINTF ("## Testing for valid EFI partition ##\n"); 36707f3d789Srichardretanubun print_part_header ("EFI", dev_desc); 36807f3d789Srichardretanubun print_part_efi (dev_desc); 36907f3d789Srichardretanubun return; 37007f3d789Srichardretanubun #endif 371affae2bfSwdenk } 372affae2bfSwdenk puts ("## Unknown partition table\n"); 373affae2bfSwdenk } 374affae2bfSwdenk 375affae2bfSwdenk 37607f3d789Srichardretanubun #else /* neither MAC nor DOS nor ISO nor AMIGA nor EFI partition configured */ 3773e3b9569SPeter Pearse # error neither CONFIG_MAC_PARTITION nor CONFIG_DOS_PARTITION 37807f3d789Srichardretanubun # error nor CONFIG_ISO_PARTITION nor CONFIG_AMIGA_PARTITION 37907f3d789Srichardretanubun # error nor CONFIG_EFI_PARTITION configured! 380affae2bfSwdenk #endif 381affae2bfSwdenk 382cde5c64dSJon Loeliger #endif 3832f501646SStephen Warren 3842f501646SStephen Warren int get_partition_info(block_dev_desc_t *dev_desc, int part 3852f501646SStephen Warren , disk_partition_t *info) 3862f501646SStephen Warren { 3872f501646SStephen Warren #if defined(CONFIG_CMD_IDE) || \ 3882f501646SStephen Warren defined(CONFIG_CMD_SATA) || \ 3892f501646SStephen Warren defined(CONFIG_CMD_SCSI) || \ 3902f501646SStephen Warren defined(CONFIG_CMD_USB) || \ 3912f501646SStephen Warren defined(CONFIG_MMC) || \ 3922f501646SStephen Warren defined(CONFIG_SYSTEMACE) 3932f501646SStephen Warren 394894bfbbfSStephen Warren #ifdef CONFIG_PARTITION_UUIDS 395894bfbbfSStephen Warren /* The common case is no UUID support */ 396894bfbbfSStephen Warren info->uuid[0] = 0; 397894bfbbfSStephen Warren #endif 398894bfbbfSStephen Warren 3992f501646SStephen Warren switch (dev_desc->part_type) { 4002f501646SStephen Warren #ifdef CONFIG_MAC_PARTITION 4012f501646SStephen Warren case PART_TYPE_MAC: 4022f501646SStephen Warren if (get_partition_info_mac(dev_desc, part, info) == 0) { 4032f501646SStephen Warren PRINTF("## Valid MAC partition found ##\n"); 4042f501646SStephen Warren return 0; 4052f501646SStephen Warren } 4062f501646SStephen Warren break; 4072f501646SStephen Warren #endif 4082f501646SStephen Warren 4092f501646SStephen Warren #ifdef CONFIG_DOS_PARTITION 4102f501646SStephen Warren case PART_TYPE_DOS: 4112f501646SStephen Warren if (get_partition_info_dos(dev_desc, part, info) == 0) { 4122f501646SStephen Warren PRINTF("## Valid DOS partition found ##\n"); 4132f501646SStephen Warren return 0; 4142f501646SStephen Warren } 4152f501646SStephen Warren break; 4162f501646SStephen Warren #endif 4172f501646SStephen Warren 4182f501646SStephen Warren #ifdef CONFIG_ISO_PARTITION 4192f501646SStephen Warren case PART_TYPE_ISO: 4202f501646SStephen Warren if (get_partition_info_iso(dev_desc, part, info) == 0) { 4212f501646SStephen Warren PRINTF("## Valid ISO boot partition found ##\n"); 4222f501646SStephen Warren return 0; 4232f501646SStephen Warren } 4242f501646SStephen Warren break; 4252f501646SStephen Warren #endif 4262f501646SStephen Warren 4272f501646SStephen Warren #ifdef CONFIG_AMIGA_PARTITION 4282f501646SStephen Warren case PART_TYPE_AMIGA: 4292f501646SStephen Warren if (get_partition_info_amiga(dev_desc, part, info) == 0) { 4302f501646SStephen Warren PRINTF("## Valid Amiga partition found ##\n"); 4312f501646SStephen Warren return 0; 4322f501646SStephen Warren } 4332f501646SStephen Warren break; 4342f501646SStephen Warren #endif 4352f501646SStephen Warren 4362f501646SStephen Warren #ifdef CONFIG_EFI_PARTITION 4372f501646SStephen Warren case PART_TYPE_EFI: 4382f501646SStephen Warren if (get_partition_info_efi(dev_desc, part, info) == 0) { 4392f501646SStephen Warren PRINTF("## Valid EFI partition found ##\n"); 4402f501646SStephen Warren return 0; 4412f501646SStephen Warren } 4422f501646SStephen Warren break; 4432f501646SStephen Warren #endif 4442f501646SStephen Warren default: 4452f501646SStephen Warren break; 4462f501646SStephen Warren } 4472f501646SStephen Warren #endif 4482f501646SStephen Warren 4492f501646SStephen Warren return -1; 4502f501646SStephen Warren } 45199d2c205SRob Herring 4522023e608SStephen Warren int get_device(const char *ifname, const char *dev_str, 4532023e608SStephen Warren block_dev_desc_t **dev_desc) 4542023e608SStephen Warren { 4552023e608SStephen Warren char *ep; 4562023e608SStephen Warren int dev; 4572023e608SStephen Warren 4582023e608SStephen Warren dev = simple_strtoul(dev_str, &ep, 16); 4592023e608SStephen Warren if (*ep) { 4602023e608SStephen Warren printf("** Bad device specification %s %s **\n", 4612023e608SStephen Warren ifname, dev_str); 4622023e608SStephen Warren return -1; 4632023e608SStephen Warren } 4642023e608SStephen Warren 4652023e608SStephen Warren *dev_desc = get_dev(ifname, dev); 4662023e608SStephen Warren if (!(*dev_desc) || ((*dev_desc)->type == DEV_TYPE_UNKNOWN)) { 4672023e608SStephen Warren printf("** Bad device %s %s **\n", ifname, dev_str); 4682023e608SStephen Warren return -1; 4692023e608SStephen Warren } 4702023e608SStephen Warren 4712023e608SStephen Warren return dev; 4722023e608SStephen Warren } 4732023e608SStephen Warren 47410a37fd7SStephen Warren #define PART_UNSPECIFIED -2 47510a37fd7SStephen Warren #define PART_AUTO -1 47610a37fd7SStephen Warren #define MAX_SEARCH_PARTITIONS 16 47710a37fd7SStephen Warren int get_device_and_partition(const char *ifname, const char *dev_part_str, 47899d2c205SRob Herring block_dev_desc_t **dev_desc, 47910a37fd7SStephen Warren disk_partition_t *info, int allow_whole_dev) 48099d2c205SRob Herring { 48110a37fd7SStephen Warren int ret = -1; 48210a37fd7SStephen Warren const char *part_str; 48310a37fd7SStephen Warren char *dup_str = NULL; 48410a37fd7SStephen Warren const char *dev_str; 48599d2c205SRob Herring int dev; 48610a37fd7SStephen Warren char *ep; 48710a37fd7SStephen Warren int p; 48810a37fd7SStephen Warren int part; 48910a37fd7SStephen Warren disk_partition_t tmpinfo; 49099d2c205SRob Herring 49110a37fd7SStephen Warren /* If no dev_part_str, use bootdevice environment variable */ 492*a10973e7SStephen Warren if (!dev_part_str || !strlen(dev_part_str) || 493*a10973e7SStephen Warren !strcmp(dev_part_str, "-")) 49410a37fd7SStephen Warren dev_part_str = getenv("bootdevice"); 49599d2c205SRob Herring 49610a37fd7SStephen Warren /* If still no dev_part_str, it's an error */ 49710a37fd7SStephen Warren if (!dev_part_str) { 49810a37fd7SStephen Warren printf("** No device specified **\n"); 49910a37fd7SStephen Warren goto cleanup; 50099d2c205SRob Herring } 50199d2c205SRob Herring 50210a37fd7SStephen Warren /* Separate device and partition ID specification */ 50310a37fd7SStephen Warren part_str = strchr(dev_part_str, ':'); 50410a37fd7SStephen Warren if (part_str) { 50510a37fd7SStephen Warren dup_str = strdup(dev_part_str); 50610a37fd7SStephen Warren dup_str[part_str - dev_part_str] = 0; 50710a37fd7SStephen Warren dev_str = dup_str; 50810a37fd7SStephen Warren part_str++; 50910a37fd7SStephen Warren } else { 51010a37fd7SStephen Warren dev_str = dev_part_str; 51199d2c205SRob Herring } 51210a37fd7SStephen Warren 51310a37fd7SStephen Warren /* Look up the device */ 51410a37fd7SStephen Warren dev = get_device(ifname, dev_str, dev_desc); 51510a37fd7SStephen Warren if (dev < 0) 51610a37fd7SStephen Warren goto cleanup; 51710a37fd7SStephen Warren 51810a37fd7SStephen Warren /* Convert partition ID string to number */ 51910a37fd7SStephen Warren if (!part_str || !*part_str) { 52010a37fd7SStephen Warren part = PART_UNSPECIFIED; 52110a37fd7SStephen Warren } else if (!strcmp(part_str, "auto")) { 52210a37fd7SStephen Warren part = PART_AUTO; 52310a37fd7SStephen Warren } else { 52410a37fd7SStephen Warren /* Something specified -> use exactly that */ 52510a37fd7SStephen Warren part = (int)simple_strtoul(part_str, &ep, 16); 52610a37fd7SStephen Warren /* 52710a37fd7SStephen Warren * Less than whole string converted, 52810a37fd7SStephen Warren * or request for whole device, but caller requires partition. 52910a37fd7SStephen Warren */ 53010a37fd7SStephen Warren if (*ep || (part == 0 && !allow_whole_dev)) { 53110a37fd7SStephen Warren printf("** Bad partition specification %s %s **\n", 53210a37fd7SStephen Warren ifname, dev_part_str); 53310a37fd7SStephen Warren goto cleanup; 53410a37fd7SStephen Warren } 53510a37fd7SStephen Warren } 53610a37fd7SStephen Warren 53710a37fd7SStephen Warren /* 53810a37fd7SStephen Warren * No partition table on device, 53910a37fd7SStephen Warren * or user requested partition 0 (entire device). 54010a37fd7SStephen Warren */ 54110a37fd7SStephen Warren if (((*dev_desc)->part_type == PART_TYPE_UNKNOWN) || 54210a37fd7SStephen Warren (part == 0)) { 54310a37fd7SStephen Warren if (!(*dev_desc)->lba) { 54410a37fd7SStephen Warren printf("** Bad device size - %s %s **\n", ifname, 54510a37fd7SStephen Warren dev_str); 54610a37fd7SStephen Warren goto cleanup; 54710a37fd7SStephen Warren } 54810a37fd7SStephen Warren 54910a37fd7SStephen Warren /* 55010a37fd7SStephen Warren * If user specified a partition ID other than 0, 55110a37fd7SStephen Warren * or the calling command only accepts partitions, 55210a37fd7SStephen Warren * it's an error. 55310a37fd7SStephen Warren */ 55410a37fd7SStephen Warren if ((part > 0) || (!allow_whole_dev)) { 55510a37fd7SStephen Warren printf("** No partition table - %s %s **\n", ifname, 55610a37fd7SStephen Warren dev_str); 55710a37fd7SStephen Warren goto cleanup; 55810a37fd7SStephen Warren } 55910a37fd7SStephen Warren 56099d2c205SRob Herring info->start = 0; 56110a37fd7SStephen Warren info->size = (*dev_desc)->lba; 56210a37fd7SStephen Warren info->blksz = (*dev_desc)->blksz; 56310a37fd7SStephen Warren info->bootable = 0; 56410a37fd7SStephen Warren #ifdef CONFIG_PARTITION_UUIDS 56510a37fd7SStephen Warren info->uuid[0] = 0; 56610a37fd7SStephen Warren #endif 56799d2c205SRob Herring 56810a37fd7SStephen Warren ret = 0; 56910a37fd7SStephen Warren goto cleanup; 57099d2c205SRob Herring } 57199d2c205SRob Herring 57210a37fd7SStephen Warren /* 57310a37fd7SStephen Warren * Now there's known to be a partition table, 57410a37fd7SStephen Warren * not specifying a partition means to pick partition 1. 57510a37fd7SStephen Warren */ 57610a37fd7SStephen Warren if (part == PART_UNSPECIFIED) 57710a37fd7SStephen Warren part = 1; 57899d2c205SRob Herring 57910a37fd7SStephen Warren /* 58010a37fd7SStephen Warren * If user didn't specify a partition number, or did specify something 58110a37fd7SStephen Warren * other than "auto", use that partition number directly. 58210a37fd7SStephen Warren */ 58310a37fd7SStephen Warren if (part != PART_AUTO) { 58410a37fd7SStephen Warren ret = get_partition_info(*dev_desc, part, info); 58599d2c205SRob Herring if (ret) { 58610a37fd7SStephen Warren printf("** Invalid partition %d **\n", part); 58710a37fd7SStephen Warren goto cleanup; 58810a37fd7SStephen Warren } 58910a37fd7SStephen Warren } else { 59010a37fd7SStephen Warren /* 59110a37fd7SStephen Warren * Find the first bootable partition. 59210a37fd7SStephen Warren * If none are bootable, fall back to the first valid partition. 59310a37fd7SStephen Warren */ 59410a37fd7SStephen Warren part = 0; 59510a37fd7SStephen Warren for (p = 1; p <= MAX_SEARCH_PARTITIONS; p++) { 59610a37fd7SStephen Warren ret = get_partition_info(*dev_desc, p, info); 59710a37fd7SStephen Warren if (ret) 59810a37fd7SStephen Warren continue; 59910a37fd7SStephen Warren 60010a37fd7SStephen Warren /* 60110a37fd7SStephen Warren * First valid partition, or new better partition? 60210a37fd7SStephen Warren * If so, save partition ID. 60310a37fd7SStephen Warren */ 60410a37fd7SStephen Warren if (!part || info->bootable) 60510a37fd7SStephen Warren part = p; 60610a37fd7SStephen Warren 60710a37fd7SStephen Warren /* Best possible partition? Stop searching. */ 60810a37fd7SStephen Warren if (info->bootable) 60910a37fd7SStephen Warren break; 61010a37fd7SStephen Warren 61110a37fd7SStephen Warren /* 61210a37fd7SStephen Warren * We now need to search further for best possible. 61310a37fd7SStephen Warren * If we what we just queried was the best so far, 61410a37fd7SStephen Warren * save the info since we over-write it next loop. 61510a37fd7SStephen Warren */ 61610a37fd7SStephen Warren if (part == p) 61710a37fd7SStephen Warren tmpinfo = *info; 61810a37fd7SStephen Warren } 61910a37fd7SStephen Warren /* If we found any acceptable partition */ 62010a37fd7SStephen Warren if (part) { 62110a37fd7SStephen Warren /* 62210a37fd7SStephen Warren * If we searched all possible partition IDs, 62310a37fd7SStephen Warren * return the first valid partition we found. 62410a37fd7SStephen Warren */ 62510a37fd7SStephen Warren if (p == MAX_SEARCH_PARTITIONS + 1) 62610a37fd7SStephen Warren *info = tmpinfo; 62710a37fd7SStephen Warren ret = 0; 62810a37fd7SStephen Warren } else { 62910a37fd7SStephen Warren printf("** No valid partitions found **\n"); 63010a37fd7SStephen Warren goto cleanup; 63110a37fd7SStephen Warren } 63299d2c205SRob Herring } 63399d2c205SRob Herring if (strncmp((char *)info->type, BOOT_PART_TYPE, sizeof(info->type)) != 0) { 63499d2c205SRob Herring printf("** Invalid partition type \"%.32s\"" 63599d2c205SRob Herring " (expect \"" BOOT_PART_TYPE "\")\n", 63699d2c205SRob Herring info->type); 63710a37fd7SStephen Warren ret = -1; 63810a37fd7SStephen Warren goto cleanup; 63999d2c205SRob Herring } 64099d2c205SRob Herring 64110a37fd7SStephen Warren ret = part; 64210a37fd7SStephen Warren goto cleanup; 64399d2c205SRob Herring 64410a37fd7SStephen Warren cleanup: 64510a37fd7SStephen Warren free(dup_str); 64610a37fd7SStephen Warren return ret; 64799d2c205SRob Herring } 648