1affae2bfSwdenk /* 2affae2bfSwdenk * (C) Copyright 2001 3affae2bfSwdenk * Wolfgang Denk, DENX Software Engineering, wd@denx.de. 4affae2bfSwdenk * 5*1a459660SWolfgang 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); 25735dd97bSGrant Likely }; 26735dd97bSGrant Likely 27735dd97bSGrant Likely static const struct block_drvr block_drvr[] = { 28cde5c64dSJon Loeliger #if defined(CONFIG_CMD_IDE) 29735dd97bSGrant Likely { .name = "ide", .get_dev = ide_get_dev, }, 30735dd97bSGrant Likely #endif 31c7057b52SDave Liu #if defined(CONFIG_CMD_SATA) 32c7057b52SDave Liu {.name = "sata", .get_dev = sata_get_dev, }, 33c7057b52SDave Liu #endif 34cde5c64dSJon Loeliger #if defined(CONFIG_CMD_SCSI) 35735dd97bSGrant Likely { .name = "scsi", .get_dev = scsi_get_dev, }, 36735dd97bSGrant Likely #endif 37cde5c64dSJon Loeliger #if defined(CONFIG_CMD_USB) && defined(CONFIG_USB_STORAGE) 38735dd97bSGrant Likely { .name = "usb", .get_dev = usb_stor_get_dev, }, 39735dd97bSGrant Likely #endif 40735dd97bSGrant Likely #if defined(CONFIG_MMC) 41735dd97bSGrant Likely { .name = "mmc", .get_dev = mmc_get_dev, }, 42735dd97bSGrant Likely #endif 43735dd97bSGrant Likely #if defined(CONFIG_SYSTEMACE) 44735dd97bSGrant Likely { .name = "ace", .get_dev = systemace_get_dev, }, 45735dd97bSGrant Likely #endif 46735dd97bSGrant Likely { }, 47735dd97bSGrant Likely }; 48735dd97bSGrant Likely 49751bb571SStefan Roese DECLARE_GLOBAL_DATA_PTR; 50751bb571SStefan Roese 510c9c8fb5SGabe Black #ifdef HAVE_BLOCK_DEVICE 5299d2c205SRob Herring block_dev_desc_t *get_dev(const char *ifname, int dev) 53735dd97bSGrant Likely { 54735dd97bSGrant Likely const struct block_drvr *drvr = block_drvr; 55751bb571SStefan Roese block_dev_desc_t* (*reloc_get_dev)(int dev); 5623090dacSHeiko Schocher char *name; 57751bb571SStefan Roese 587e71dc68STim Kientzle if (!ifname) 597e71dc68STim Kientzle return NULL; 607e71dc68STim Kientzle 6123090dacSHeiko Schocher name = drvr->name; 622e5167ccSWolfgang Denk #ifdef CONFIG_NEEDS_MANUAL_RELOC 6323090dacSHeiko Schocher name += gd->reloc_off; 6423090dacSHeiko Schocher #endif 65b16aadf4SLei Wen while (drvr->name) { 6623090dacSHeiko Schocher name = drvr->name; 67521af04dSPeter Tyser reloc_get_dev = drvr->get_dev; 682e5167ccSWolfgang Denk #ifdef CONFIG_NEEDS_MANUAL_RELOC 6923090dacSHeiko Schocher name += gd->reloc_off; 70521af04dSPeter Tyser reloc_get_dev += gd->reloc_off; 71521af04dSPeter Tyser #endif 7223090dacSHeiko Schocher if (strncmp(ifname, name, strlen(name)) == 0) 73751bb571SStefan Roese return reloc_get_dev(dev); 74735dd97bSGrant Likely drvr++; 75735dd97bSGrant Likely } 76735dd97bSGrant Likely return NULL; 77735dd97bSGrant Likely } 78735dd97bSGrant Likely #else 7999d2c205SRob Herring block_dev_desc_t *get_dev(const char *ifname, int dev) 80735dd97bSGrant Likely { 81735dd97bSGrant Likely return NULL; 82735dd97bSGrant Likely } 83735dd97bSGrant Likely #endif 84735dd97bSGrant Likely 850c9c8fb5SGabe Black #ifdef HAVE_BLOCK_DEVICE 86735dd97bSGrant Likely 87affae2bfSwdenk /* ------------------------------------------------------------------------- */ 88affae2bfSwdenk /* 89affae2bfSwdenk * reports device info to the user 90affae2bfSwdenk */ 9169a2a4d9SSergei Trofimovich 9269a2a4d9SSergei Trofimovich #ifdef CONFIG_LBA48 9369a2a4d9SSergei Trofimovich typedef uint64_t lba512_t; 9469a2a4d9SSergei Trofimovich #else 9569a2a4d9SSergei Trofimovich typedef lbaint_t lba512_t; 9669a2a4d9SSergei Trofimovich #endif 9769a2a4d9SSergei Trofimovich 9869a2a4d9SSergei Trofimovich /* 9969a2a4d9SSergei Trofimovich * Overflowless variant of (block_count * mul_by / div_by) 10069a2a4d9SSergei Trofimovich * when div_by > mul_by 10169a2a4d9SSergei Trofimovich */ 10269a2a4d9SSergei Trofimovich static lba512_t lba512_muldiv (lba512_t block_count, lba512_t mul_by, lba512_t div_by) 10369a2a4d9SSergei Trofimovich { 10469a2a4d9SSergei Trofimovich lba512_t bc_quot, bc_rem; 10569a2a4d9SSergei Trofimovich 10669a2a4d9SSergei Trofimovich /* x * m / d == x / d * m + (x % d) * m / d */ 10769a2a4d9SSergei Trofimovich bc_quot = block_count / div_by; 10869a2a4d9SSergei Trofimovich bc_rem = block_count - div_by * bc_quot; 10969a2a4d9SSergei Trofimovich return bc_quot * mul_by + (bc_rem * mul_by) / div_by; 11069a2a4d9SSergei Trofimovich } 11169a2a4d9SSergei Trofimovich 112affae2bfSwdenk void dev_print (block_dev_desc_t *dev_desc) 113affae2bfSwdenk { 11469a2a4d9SSergei Trofimovich lba512_t lba512; /* number of blocks if 512bytes block size */ 115affae2bfSwdenk 116af75a45dSWolfgang Denk if (dev_desc->type == DEV_TYPE_UNKNOWN) { 117af75a45dSWolfgang Denk puts ("not available\n"); 118af75a45dSWolfgang Denk return; 119af75a45dSWolfgang Denk } 120af75a45dSWolfgang Denk 1218ec6e332STor Krill switch (dev_desc->if_type) { 122574b3195SDetlev Zundel case IF_TYPE_SCSI: 123574b3195SDetlev Zundel printf ("(%d:%d) Vendor: %s Prod.: %s Rev: %s\n", 124574b3195SDetlev Zundel dev_desc->target,dev_desc->lun, 125affae2bfSwdenk dev_desc->vendor, 126affae2bfSwdenk dev_desc->product, 127affae2bfSwdenk dev_desc->revision); 128574b3195SDetlev Zundel break; 1296e24a1ebSRemy Bohmer case IF_TYPE_ATAPI: 130574b3195SDetlev Zundel case IF_TYPE_IDE: 131574b3195SDetlev Zundel case IF_TYPE_SATA: 132574b3195SDetlev Zundel printf ("Model: %s Firm: %s Ser#: %s\n", 133574b3195SDetlev Zundel dev_desc->vendor, 134574b3195SDetlev Zundel dev_desc->revision, 135574b3195SDetlev Zundel dev_desc->product); 136574b3195SDetlev Zundel break; 1376e24a1ebSRemy Bohmer case IF_TYPE_SD: 1386e24a1ebSRemy Bohmer case IF_TYPE_MMC: 13947bebe34SNícolas Carneiro Lebedenco case IF_TYPE_USB: 14047bebe34SNícolas Carneiro Lebedenco printf ("Vendor: %s Rev: %s Prod: %s\n", 14147bebe34SNícolas Carneiro Lebedenco dev_desc->vendor, 14247bebe34SNícolas Carneiro Lebedenco dev_desc->revision, 14347bebe34SNícolas Carneiro Lebedenco dev_desc->product); 14447bebe34SNícolas Carneiro Lebedenco break; 1456e24a1ebSRemy Bohmer case IF_TYPE_DOC: 1466e24a1ebSRemy Bohmer puts("device type DOC\n"); 1476e24a1ebSRemy Bohmer return; 1488ec6e332STor Krill case IF_TYPE_UNKNOWN: 1496e24a1ebSRemy Bohmer puts("device type unknown\n"); 1506e24a1ebSRemy Bohmer return; 151574b3195SDetlev Zundel default: 1526e24a1ebSRemy Bohmer printf("Unhandled device type: %i\n", dev_desc->if_type); 153574b3195SDetlev Zundel return; 154affae2bfSwdenk } 155affae2bfSwdenk puts (" Type: "); 156affae2bfSwdenk if (dev_desc->removable) 157affae2bfSwdenk puts ("Removable "); 158affae2bfSwdenk switch (dev_desc->type & 0x1F) { 159726c0f1eSDetlev Zundel case DEV_TYPE_HARDDISK: 160726c0f1eSDetlev Zundel puts ("Hard Disk"); 161affae2bfSwdenk break; 162726c0f1eSDetlev Zundel case DEV_TYPE_CDROM: 163726c0f1eSDetlev Zundel puts ("CD ROM"); 164affae2bfSwdenk break; 165726c0f1eSDetlev Zundel case DEV_TYPE_OPDISK: 166726c0f1eSDetlev Zundel puts ("Optical Device"); 167affae2bfSwdenk break; 168726c0f1eSDetlev Zundel case DEV_TYPE_TAPE: 169726c0f1eSDetlev Zundel puts ("Tape"); 170affae2bfSwdenk break; 171726c0f1eSDetlev Zundel default: 172726c0f1eSDetlev Zundel printf ("# %02X #", dev_desc->type & 0x1F); 173affae2bfSwdenk break; 174affae2bfSwdenk } 175affae2bfSwdenk puts ("\n"); 17633699df1SJerry Huang if (dev_desc->lba > 0L && dev_desc->blksz > 0L) { 177affae2bfSwdenk ulong mb, mb_quot, mb_rem, gb, gb_quot, gb_rem; 178c40b2956Swdenk lbaint_t lba; 1796e592385Swdenk 180c40b2956Swdenk lba = dev_desc->lba; 181affae2bfSwdenk 182c40b2956Swdenk lba512 = (lba * (dev_desc->blksz/512)); 183affae2bfSwdenk /* round to 1 digit */ 18469a2a4d9SSergei Trofimovich mb = lba512_muldiv(lba512, 10, 2048); /* 2048 = (1024 * 1024) / 512 MB */ 18569a2a4d9SSergei Trofimovich 186affae2bfSwdenk mb_quot = mb / 10; 187affae2bfSwdenk mb_rem = mb - (10 * mb_quot); 188affae2bfSwdenk 189affae2bfSwdenk gb = mb / 1024; 190affae2bfSwdenk gb_quot = gb / 10; 191affae2bfSwdenk gb_rem = gb - (10 * gb_quot); 19242dfe7a1Swdenk #ifdef CONFIG_LBA48 1936e592385Swdenk if (dev_desc->lba48) 194c40b2956Swdenk printf (" Supports 48-bit addressing\n"); 195c40b2956Swdenk #endif 1964b142febSHeiko Schocher #if defined(CONFIG_SYS_64BIT_LBA) 1976c6166f5SMike Frysinger printf (" Capacity: %ld.%ld MB = %ld.%ld GB (%Ld x %ld)\n", 198c40b2956Swdenk mb_quot, mb_rem, 199c40b2956Swdenk gb_quot, gb_rem, 200c40b2956Swdenk lba, 201c40b2956Swdenk dev_desc->blksz); 202c40b2956Swdenk #else 203affae2bfSwdenk printf (" Capacity: %ld.%ld MB = %ld.%ld GB (%ld x %ld)\n", 204affae2bfSwdenk mb_quot, mb_rem, 205affae2bfSwdenk gb_quot, gb_rem, 206c40b2956Swdenk (ulong)lba, 207affae2bfSwdenk dev_desc->blksz); 208c40b2956Swdenk #endif 209affae2bfSwdenk } else { 210affae2bfSwdenk puts (" Capacity: not available\n"); 211affae2bfSwdenk } 212affae2bfSwdenk } 213b3aff0cbSJon Loeliger #endif 214affae2bfSwdenk 2150c9c8fb5SGabe Black #ifdef HAVE_BLOCK_DEVICE 216affae2bfSwdenk 217affae2bfSwdenk void init_part (block_dev_desc_t * dev_desc) 218affae2bfSwdenk { 219affae2bfSwdenk #ifdef CONFIG_ISO_PARTITION 220affae2bfSwdenk if (test_part_iso(dev_desc) == 0) { 221affae2bfSwdenk dev_desc->part_type = PART_TYPE_ISO; 222affae2bfSwdenk return; 223affae2bfSwdenk } 224affae2bfSwdenk #endif 225affae2bfSwdenk 226affae2bfSwdenk #ifdef CONFIG_MAC_PARTITION 227affae2bfSwdenk if (test_part_mac(dev_desc) == 0) { 228affae2bfSwdenk dev_desc->part_type = PART_TYPE_MAC; 229affae2bfSwdenk return; 230affae2bfSwdenk } 231affae2bfSwdenk #endif 232affae2bfSwdenk 23307f3d789Srichardretanubun /* must be placed before DOS partition detection */ 23407f3d789Srichardretanubun #ifdef CONFIG_EFI_PARTITION 23507f3d789Srichardretanubun if (test_part_efi(dev_desc) == 0) { 23607f3d789Srichardretanubun dev_desc->part_type = PART_TYPE_EFI; 23707f3d789Srichardretanubun return; 23807f3d789Srichardretanubun } 23907f3d789Srichardretanubun #endif 24007f3d789Srichardretanubun 241affae2bfSwdenk #ifdef CONFIG_DOS_PARTITION 242affae2bfSwdenk if (test_part_dos(dev_desc) == 0) { 243affae2bfSwdenk dev_desc->part_type = PART_TYPE_DOS; 244affae2bfSwdenk return; 245affae2bfSwdenk } 246affae2bfSwdenk #endif 247c7de829cSwdenk 248c7de829cSwdenk #ifdef CONFIG_AMIGA_PARTITION 249c7de829cSwdenk if (test_part_amiga(dev_desc) == 0) { 250c7de829cSwdenk dev_desc->part_type = PART_TYPE_AMIGA; 251c7de829cSwdenk return; 252c7de829cSwdenk } 253c7de829cSwdenk #endif 25499d2c205SRob Herring dev_desc->part_type = PART_TYPE_UNKNOWN; 255affae2bfSwdenk } 256affae2bfSwdenk 257affae2bfSwdenk 2580c9c8fb5SGabe Black #if defined(CONFIG_MAC_PARTITION) || \ 2590c9c8fb5SGabe Black defined(CONFIG_DOS_PARTITION) || \ 2600c9c8fb5SGabe Black defined(CONFIG_ISO_PARTITION) || \ 2610c9c8fb5SGabe Black defined(CONFIG_AMIGA_PARTITION) || \ 2620c9c8fb5SGabe Black defined(CONFIG_EFI_PARTITION) 2630c9c8fb5SGabe Black 264affae2bfSwdenk static void print_part_header (const char *type, block_dev_desc_t * dev_desc) 265affae2bfSwdenk { 266affae2bfSwdenk puts ("\nPartition Map for "); 267affae2bfSwdenk switch (dev_desc->if_type) { 268726c0f1eSDetlev Zundel case IF_TYPE_IDE: 269726c0f1eSDetlev Zundel puts ("IDE"); 270affae2bfSwdenk break; 271726c0f1eSDetlev Zundel case IF_TYPE_SATA: 272726c0f1eSDetlev Zundel puts ("SATA"); 273c7057b52SDave Liu break; 274726c0f1eSDetlev Zundel case IF_TYPE_SCSI: 275726c0f1eSDetlev Zundel puts ("SCSI"); 276affae2bfSwdenk break; 277726c0f1eSDetlev Zundel case IF_TYPE_ATAPI: 278726c0f1eSDetlev Zundel puts ("ATAPI"); 279affae2bfSwdenk break; 280726c0f1eSDetlev Zundel case IF_TYPE_USB: 281726c0f1eSDetlev Zundel puts ("USB"); 282affae2bfSwdenk break; 283726c0f1eSDetlev Zundel case IF_TYPE_DOC: 284726c0f1eSDetlev Zundel puts ("DOC"); 285affae2bfSwdenk break; 2868f3b9642SLei Wen case IF_TYPE_MMC: 2878f3b9642SLei Wen puts ("MMC"); 2888f3b9642SLei Wen break; 289726c0f1eSDetlev Zundel default: 290726c0f1eSDetlev Zundel puts ("UNKNOWN"); 291affae2bfSwdenk break; 292affae2bfSwdenk } 293affae2bfSwdenk printf (" device %d -- Partition Type: %s\n\n", 294affae2bfSwdenk dev_desc->dev, type); 295affae2bfSwdenk } 296affae2bfSwdenk 2970c9c8fb5SGabe Black #endif /* any CONFIG_..._PARTITION */ 2980c9c8fb5SGabe Black 299affae2bfSwdenk void print_part (block_dev_desc_t * dev_desc) 300affae2bfSwdenk { 301affae2bfSwdenk 302affae2bfSwdenk switch (dev_desc->part_type) { 303affae2bfSwdenk #ifdef CONFIG_MAC_PARTITION 304affae2bfSwdenk case PART_TYPE_MAC: 305affae2bfSwdenk PRINTF ("## Testing for valid MAC partition ##\n"); 306affae2bfSwdenk print_part_header ("MAC", dev_desc); 307affae2bfSwdenk print_part_mac (dev_desc); 308affae2bfSwdenk return; 309affae2bfSwdenk #endif 310affae2bfSwdenk #ifdef CONFIG_DOS_PARTITION 311affae2bfSwdenk case PART_TYPE_DOS: 312affae2bfSwdenk PRINTF ("## Testing for valid DOS partition ##\n"); 313affae2bfSwdenk print_part_header ("DOS", dev_desc); 314affae2bfSwdenk print_part_dos (dev_desc); 315affae2bfSwdenk return; 316affae2bfSwdenk #endif 317affae2bfSwdenk 318affae2bfSwdenk #ifdef CONFIG_ISO_PARTITION 319affae2bfSwdenk case PART_TYPE_ISO: 320affae2bfSwdenk PRINTF ("## Testing for valid ISO Boot partition ##\n"); 321affae2bfSwdenk print_part_header ("ISO", dev_desc); 322affae2bfSwdenk print_part_iso (dev_desc); 323affae2bfSwdenk return; 324affae2bfSwdenk #endif 325c7de829cSwdenk 326c7de829cSwdenk #ifdef CONFIG_AMIGA_PARTITION 327c7de829cSwdenk case PART_TYPE_AMIGA: 328c7de829cSwdenk PRINTF ("## Testing for a valid Amiga partition ##\n"); 329c7de829cSwdenk print_part_header ("AMIGA", dev_desc); 330c7de829cSwdenk print_part_amiga (dev_desc); 331c7de829cSwdenk return; 332c7de829cSwdenk #endif 33307f3d789Srichardretanubun 33407f3d789Srichardretanubun #ifdef CONFIG_EFI_PARTITION 33507f3d789Srichardretanubun case PART_TYPE_EFI: 33607f3d789Srichardretanubun PRINTF ("## Testing for valid EFI partition ##\n"); 33707f3d789Srichardretanubun print_part_header ("EFI", dev_desc); 33807f3d789Srichardretanubun print_part_efi (dev_desc); 33907f3d789Srichardretanubun return; 34007f3d789Srichardretanubun #endif 341affae2bfSwdenk } 342affae2bfSwdenk puts ("## Unknown partition table\n"); 343affae2bfSwdenk } 344affae2bfSwdenk 3450c9c8fb5SGabe Black #endif /* HAVE_BLOCK_DEVICE */ 3462f501646SStephen Warren 3472f501646SStephen Warren int get_partition_info(block_dev_desc_t *dev_desc, int part 3482f501646SStephen Warren , disk_partition_t *info) 3492f501646SStephen Warren { 3500c9c8fb5SGabe Black #ifdef HAVE_BLOCK_DEVICE 3512f501646SStephen Warren 352894bfbbfSStephen Warren #ifdef CONFIG_PARTITION_UUIDS 353894bfbbfSStephen Warren /* The common case is no UUID support */ 354894bfbbfSStephen Warren info->uuid[0] = 0; 355894bfbbfSStephen Warren #endif 356894bfbbfSStephen Warren 3572f501646SStephen Warren switch (dev_desc->part_type) { 3582f501646SStephen Warren #ifdef CONFIG_MAC_PARTITION 3592f501646SStephen Warren case PART_TYPE_MAC: 3602f501646SStephen Warren if (get_partition_info_mac(dev_desc, part, info) == 0) { 3612f501646SStephen Warren PRINTF("## Valid MAC partition found ##\n"); 3622f501646SStephen Warren return 0; 3632f501646SStephen Warren } 3642f501646SStephen Warren break; 3652f501646SStephen Warren #endif 3662f501646SStephen Warren 3672f501646SStephen Warren #ifdef CONFIG_DOS_PARTITION 3682f501646SStephen Warren case PART_TYPE_DOS: 3692f501646SStephen Warren if (get_partition_info_dos(dev_desc, part, info) == 0) { 3702f501646SStephen Warren PRINTF("## Valid DOS partition found ##\n"); 3712f501646SStephen Warren return 0; 3722f501646SStephen Warren } 3732f501646SStephen Warren break; 3742f501646SStephen Warren #endif 3752f501646SStephen Warren 3762f501646SStephen Warren #ifdef CONFIG_ISO_PARTITION 3772f501646SStephen Warren case PART_TYPE_ISO: 3782f501646SStephen Warren if (get_partition_info_iso(dev_desc, part, info) == 0) { 3792f501646SStephen Warren PRINTF("## Valid ISO boot partition found ##\n"); 3802f501646SStephen Warren return 0; 3812f501646SStephen Warren } 3822f501646SStephen Warren break; 3832f501646SStephen Warren #endif 3842f501646SStephen Warren 3852f501646SStephen Warren #ifdef CONFIG_AMIGA_PARTITION 3862f501646SStephen Warren case PART_TYPE_AMIGA: 3872f501646SStephen Warren if (get_partition_info_amiga(dev_desc, part, info) == 0) { 3882f501646SStephen Warren PRINTF("## Valid Amiga partition found ##\n"); 3892f501646SStephen Warren return 0; 3902f501646SStephen Warren } 3912f501646SStephen Warren break; 3922f501646SStephen Warren #endif 3932f501646SStephen Warren 3942f501646SStephen Warren #ifdef CONFIG_EFI_PARTITION 3952f501646SStephen Warren case PART_TYPE_EFI: 3962f501646SStephen Warren if (get_partition_info_efi(dev_desc, part, info) == 0) { 3972f501646SStephen Warren PRINTF("## Valid EFI partition found ##\n"); 3982f501646SStephen Warren return 0; 3992f501646SStephen Warren } 4002f501646SStephen Warren break; 4012f501646SStephen Warren #endif 4022f501646SStephen Warren default: 4032f501646SStephen Warren break; 4042f501646SStephen Warren } 4050c9c8fb5SGabe Black #endif /* HAVE_BLOCK_DEVICE */ 4062f501646SStephen Warren 4072f501646SStephen Warren return -1; 4082f501646SStephen Warren } 40999d2c205SRob Herring 4102023e608SStephen Warren int get_device(const char *ifname, const char *dev_str, 4112023e608SStephen Warren block_dev_desc_t **dev_desc) 4122023e608SStephen Warren { 4132023e608SStephen Warren char *ep; 4142023e608SStephen Warren int dev; 4152023e608SStephen Warren 4162023e608SStephen Warren dev = simple_strtoul(dev_str, &ep, 16); 4172023e608SStephen Warren if (*ep) { 4182023e608SStephen Warren printf("** Bad device specification %s %s **\n", 4192023e608SStephen Warren ifname, dev_str); 4202023e608SStephen Warren return -1; 4212023e608SStephen Warren } 4222023e608SStephen Warren 4232023e608SStephen Warren *dev_desc = get_dev(ifname, dev); 4242023e608SStephen Warren if (!(*dev_desc) || ((*dev_desc)->type == DEV_TYPE_UNKNOWN)) { 4252023e608SStephen Warren printf("** Bad device %s %s **\n", ifname, dev_str); 4262023e608SStephen Warren return -1; 4272023e608SStephen Warren } 4282023e608SStephen Warren 4292023e608SStephen Warren return dev; 4302023e608SStephen Warren } 4312023e608SStephen Warren 43210a37fd7SStephen Warren #define PART_UNSPECIFIED -2 43310a37fd7SStephen Warren #define PART_AUTO -1 43410a37fd7SStephen Warren #define MAX_SEARCH_PARTITIONS 16 43510a37fd7SStephen Warren int get_device_and_partition(const char *ifname, const char *dev_part_str, 43699d2c205SRob Herring block_dev_desc_t **dev_desc, 43710a37fd7SStephen Warren disk_partition_t *info, int allow_whole_dev) 43899d2c205SRob Herring { 43910a37fd7SStephen Warren int ret = -1; 44010a37fd7SStephen Warren const char *part_str; 44110a37fd7SStephen Warren char *dup_str = NULL; 44210a37fd7SStephen Warren const char *dev_str; 44399d2c205SRob Herring int dev; 44410a37fd7SStephen Warren char *ep; 44510a37fd7SStephen Warren int p; 44610a37fd7SStephen Warren int part; 44710a37fd7SStephen Warren disk_partition_t tmpinfo; 44899d2c205SRob Herring 44992ccc96bSSimon Glass /* 45092ccc96bSSimon Glass * For now, we have a special case for sandbox, since there is no 45192ccc96bSSimon Glass * real block device support. 45292ccc96bSSimon Glass */ 45392ccc96bSSimon Glass if (0 == strcmp(ifname, "host")) { 45492ccc96bSSimon Glass *dev_desc = NULL; 45592ccc96bSSimon Glass info->start = info->size = info->blksz = 0; 45692ccc96bSSimon Glass info->bootable = 0; 45792ccc96bSSimon Glass strcpy((char *)info->type, BOOT_PART_TYPE); 45892ccc96bSSimon Glass strcpy((char *)info->name, "Sandbox host"); 45992ccc96bSSimon Glass #ifdef CONFIG_PARTITION_UUIDS 46092ccc96bSSimon Glass info->uuid[0] = 0; 46192ccc96bSSimon Glass #endif 46292ccc96bSSimon Glass 46392ccc96bSSimon Glass return 0; 46492ccc96bSSimon Glass } 46592ccc96bSSimon Glass 46610a37fd7SStephen Warren /* If no dev_part_str, use bootdevice environment variable */ 467a10973e7SStephen Warren if (!dev_part_str || !strlen(dev_part_str) || 468a10973e7SStephen Warren !strcmp(dev_part_str, "-")) 46910a37fd7SStephen Warren dev_part_str = getenv("bootdevice"); 47099d2c205SRob Herring 47110a37fd7SStephen Warren /* If still no dev_part_str, it's an error */ 47210a37fd7SStephen Warren if (!dev_part_str) { 47310a37fd7SStephen Warren printf("** No device specified **\n"); 47410a37fd7SStephen Warren goto cleanup; 47599d2c205SRob Herring } 47699d2c205SRob Herring 47710a37fd7SStephen Warren /* Separate device and partition ID specification */ 47810a37fd7SStephen Warren part_str = strchr(dev_part_str, ':'); 47910a37fd7SStephen Warren if (part_str) { 48010a37fd7SStephen Warren dup_str = strdup(dev_part_str); 48110a37fd7SStephen Warren dup_str[part_str - dev_part_str] = 0; 48210a37fd7SStephen Warren dev_str = dup_str; 48310a37fd7SStephen Warren part_str++; 48410a37fd7SStephen Warren } else { 48510a37fd7SStephen Warren dev_str = dev_part_str; 48699d2c205SRob Herring } 48710a37fd7SStephen Warren 48810a37fd7SStephen Warren /* Look up the device */ 48910a37fd7SStephen Warren dev = get_device(ifname, dev_str, dev_desc); 49010a37fd7SStephen Warren if (dev < 0) 49110a37fd7SStephen Warren goto cleanup; 49210a37fd7SStephen Warren 49310a37fd7SStephen Warren /* Convert partition ID string to number */ 49410a37fd7SStephen Warren if (!part_str || !*part_str) { 49510a37fd7SStephen Warren part = PART_UNSPECIFIED; 49610a37fd7SStephen Warren } else if (!strcmp(part_str, "auto")) { 49710a37fd7SStephen Warren part = PART_AUTO; 49810a37fd7SStephen Warren } else { 49910a37fd7SStephen Warren /* Something specified -> use exactly that */ 50010a37fd7SStephen Warren part = (int)simple_strtoul(part_str, &ep, 16); 50110a37fd7SStephen Warren /* 50210a37fd7SStephen Warren * Less than whole string converted, 50310a37fd7SStephen Warren * or request for whole device, but caller requires partition. 50410a37fd7SStephen Warren */ 50510a37fd7SStephen Warren if (*ep || (part == 0 && !allow_whole_dev)) { 50610a37fd7SStephen Warren printf("** Bad partition specification %s %s **\n", 50710a37fd7SStephen Warren ifname, dev_part_str); 50810a37fd7SStephen Warren goto cleanup; 50910a37fd7SStephen Warren } 51010a37fd7SStephen Warren } 51110a37fd7SStephen Warren 51210a37fd7SStephen Warren /* 51310a37fd7SStephen Warren * No partition table on device, 51410a37fd7SStephen Warren * or user requested partition 0 (entire device). 51510a37fd7SStephen Warren */ 51610a37fd7SStephen Warren if (((*dev_desc)->part_type == PART_TYPE_UNKNOWN) || 51710a37fd7SStephen Warren (part == 0)) { 51810a37fd7SStephen Warren if (!(*dev_desc)->lba) { 51910a37fd7SStephen Warren printf("** Bad device size - %s %s **\n", ifname, 52010a37fd7SStephen Warren dev_str); 52110a37fd7SStephen Warren goto cleanup; 52210a37fd7SStephen Warren } 52310a37fd7SStephen Warren 52410a37fd7SStephen Warren /* 52510a37fd7SStephen Warren * If user specified a partition ID other than 0, 52610a37fd7SStephen Warren * or the calling command only accepts partitions, 52710a37fd7SStephen Warren * it's an error. 52810a37fd7SStephen Warren */ 52910a37fd7SStephen Warren if ((part > 0) || (!allow_whole_dev)) { 53010a37fd7SStephen Warren printf("** No partition table - %s %s **\n", ifname, 53110a37fd7SStephen Warren dev_str); 53210a37fd7SStephen Warren goto cleanup; 53310a37fd7SStephen Warren } 53410a37fd7SStephen Warren 53550ffc3b6SLan Yixun (dlan) (*dev_desc)->log2blksz = LOG2((*dev_desc)->blksz); 53650ffc3b6SLan Yixun (dlan) 53799d2c205SRob Herring info->start = 0; 53810a37fd7SStephen Warren info->size = (*dev_desc)->lba; 53910a37fd7SStephen Warren info->blksz = (*dev_desc)->blksz; 54010a37fd7SStephen Warren info->bootable = 0; 5416ab6a650SStephen Warren strcpy((char *)info->type, BOOT_PART_TYPE); 5426ab6a650SStephen Warren strcpy((char *)info->name, "Whole Disk"); 54310a37fd7SStephen Warren #ifdef CONFIG_PARTITION_UUIDS 54410a37fd7SStephen Warren info->uuid[0] = 0; 54510a37fd7SStephen Warren #endif 54699d2c205SRob Herring 54710a37fd7SStephen Warren ret = 0; 54810a37fd7SStephen Warren goto cleanup; 54999d2c205SRob Herring } 55099d2c205SRob Herring 55110a37fd7SStephen Warren /* 55210a37fd7SStephen Warren * Now there's known to be a partition table, 55310a37fd7SStephen Warren * not specifying a partition means to pick partition 1. 55410a37fd7SStephen Warren */ 55510a37fd7SStephen Warren if (part == PART_UNSPECIFIED) 55610a37fd7SStephen Warren part = 1; 55799d2c205SRob Herring 55810a37fd7SStephen Warren /* 55910a37fd7SStephen Warren * If user didn't specify a partition number, or did specify something 56010a37fd7SStephen Warren * other than "auto", use that partition number directly. 56110a37fd7SStephen Warren */ 56210a37fd7SStephen Warren if (part != PART_AUTO) { 56310a37fd7SStephen Warren ret = get_partition_info(*dev_desc, part, info); 56499d2c205SRob Herring if (ret) { 56510a37fd7SStephen Warren printf("** Invalid partition %d **\n", part); 56610a37fd7SStephen Warren goto cleanup; 56710a37fd7SStephen Warren } 56810a37fd7SStephen Warren } else { 56910a37fd7SStephen Warren /* 57010a37fd7SStephen Warren * Find the first bootable partition. 57110a37fd7SStephen Warren * If none are bootable, fall back to the first valid partition. 57210a37fd7SStephen Warren */ 57310a37fd7SStephen Warren part = 0; 57410a37fd7SStephen Warren for (p = 1; p <= MAX_SEARCH_PARTITIONS; p++) { 57510a37fd7SStephen Warren ret = get_partition_info(*dev_desc, p, info); 57610a37fd7SStephen Warren if (ret) 57710a37fd7SStephen Warren continue; 57810a37fd7SStephen Warren 57910a37fd7SStephen Warren /* 58010a37fd7SStephen Warren * First valid partition, or new better partition? 58110a37fd7SStephen Warren * If so, save partition ID. 58210a37fd7SStephen Warren */ 58310a37fd7SStephen Warren if (!part || info->bootable) 58410a37fd7SStephen Warren part = p; 58510a37fd7SStephen Warren 58610a37fd7SStephen Warren /* Best possible partition? Stop searching. */ 58710a37fd7SStephen Warren if (info->bootable) 58810a37fd7SStephen Warren break; 58910a37fd7SStephen Warren 59010a37fd7SStephen Warren /* 59110a37fd7SStephen Warren * We now need to search further for best possible. 59210a37fd7SStephen Warren * If we what we just queried was the best so far, 59310a37fd7SStephen Warren * save the info since we over-write it next loop. 59410a37fd7SStephen Warren */ 59510a37fd7SStephen Warren if (part == p) 59610a37fd7SStephen Warren tmpinfo = *info; 59710a37fd7SStephen Warren } 59810a37fd7SStephen Warren /* If we found any acceptable partition */ 59910a37fd7SStephen Warren if (part) { 60010a37fd7SStephen Warren /* 60110a37fd7SStephen Warren * If we searched all possible partition IDs, 60210a37fd7SStephen Warren * return the first valid partition we found. 60310a37fd7SStephen Warren */ 60410a37fd7SStephen Warren if (p == MAX_SEARCH_PARTITIONS + 1) 60510a37fd7SStephen Warren *info = tmpinfo; 60610a37fd7SStephen Warren } else { 60710a37fd7SStephen Warren printf("** No valid partitions found **\n"); 60871bba424SStephen Warren ret = -1; 60910a37fd7SStephen Warren goto cleanup; 61010a37fd7SStephen Warren } 61199d2c205SRob Herring } 61299d2c205SRob Herring if (strncmp((char *)info->type, BOOT_PART_TYPE, sizeof(info->type)) != 0) { 61399d2c205SRob Herring printf("** Invalid partition type \"%.32s\"" 61499d2c205SRob Herring " (expect \"" BOOT_PART_TYPE "\")\n", 61599d2c205SRob Herring info->type); 61610a37fd7SStephen Warren ret = -1; 61710a37fd7SStephen Warren goto cleanup; 61899d2c205SRob Herring } 61999d2c205SRob Herring 62050ffc3b6SLan Yixun (dlan) (*dev_desc)->log2blksz = LOG2((*dev_desc)->blksz); 62150ffc3b6SLan Yixun (dlan) 62210a37fd7SStephen Warren ret = part; 62310a37fd7SStephen Warren goto cleanup; 62499d2c205SRob Herring 62510a37fd7SStephen Warren cleanup: 62610a37fd7SStephen Warren free(dup_str); 62710a37fd7SStephen Warren return ret; 62899d2c205SRob Herring } 629