]> git.kernelconcepts.de Git - karo-tx-uboot.git/blobdiff - disk/part.c
sandbox: Use os functions to read host device tree
[karo-tx-uboot.git] / disk / part.c
index ed6bbe29bc878bc0073b32e8bd5a007c106748ff..b8c6aac801626f165bab1bb800ccf657c5a944f4 100644 (file)
@@ -2,28 +2,13 @@
  * (C) Copyright 2001
  * Wolfgang Denk, DENX Software Engineering, wd@denx.de.
  *
- * See file CREDITS for list of people who contributed to this
- * project.
- *
- * This program is free software; you can redistribute it and/or
- * modify it under the terms of the GNU General Public License as
- * published by the Free Software Foundation; either version 2 of
- * the License, or (at your option) any later version.
- *
- * This program is distributed in the hope that it will be useful,
- * but WITHOUT ANY WARRANTY; without even the implied warranty of
- * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
- * GNU General Public License for more details.
- *
- * You should have received a copy of the GNU General Public License
- * along with this program; if not, write to the Free Software
- * Foundation, Inc., 59 Temple Place, Suite 330, Boston,
- * MA 02111-1307 USA
+ * SPDX-License-Identifier:    GPL-2.0+
  */
 
 #include <common.h>
 #include <command.h>
 #include <ide.h>
+#include <malloc.h>
 #include <part.h>
 
 #undef PART_DEBUG
 #define PRINTF(fmt,args...)
 #endif
 
-#if (defined(CONFIG_CMD_IDE) || \
-     defined(CONFIG_CMD_SATA) || \
-     defined(CONFIG_CMD_SCSI) || \
-     defined(CONFIG_CMD_USB) || \
-     defined(CONFIG_MMC) || \
-     defined(CONFIG_SYSTEMACE) )
-
 struct block_drvr {
        char *name;
        block_dev_desc_t* (*get_dev)(int dev);
@@ -64,12 +42,16 @@ static const struct block_drvr block_drvr[] = {
 #endif
 #if defined(CONFIG_SYSTEMACE)
        { .name = "ace", .get_dev = systemace_get_dev, },
+#endif
+#if defined(CONFIG_SANDBOX)
+       { .name = "host", .get_dev = host_get_dev, },
 #endif
        { },
 };
 
 DECLARE_GLOBAL_DATA_PTR;
 
+#ifdef HAVE_BLOCK_DEVICE
 block_dev_desc_t *get_dev(const char *ifname, int dev)
 {
        const struct block_drvr *drvr = block_drvr;
@@ -103,12 +85,7 @@ block_dev_desc_t *get_dev(const char *ifname, int dev)
 }
 #endif
 
-#if (defined(CONFIG_CMD_IDE) || \
-     defined(CONFIG_CMD_SATA) || \
-     defined(CONFIG_CMD_SCSI) || \
-     defined(CONFIG_CMD_USB) || \
-     defined(CONFIG_MMC) || \
-     defined(CONFIG_SYSTEMACE) )
+#ifdef HAVE_BLOCK_DEVICE
 
 /* ------------------------------------------------------------------------- */
 /*
@@ -199,7 +176,7 @@ void dev_print (block_dev_desc_t *dev_desc)
                break;
        }
        puts ("\n");
-       if ((dev_desc->lba * dev_desc->blksz)>0L) {
+       if (dev_desc->lba > 0L && dev_desc->blksz > 0L) {
                ulong mb, mb_quot, mb_rem, gb, gb_quot, gb_rem;
                lbaint_t lba;
 
@@ -238,18 +215,7 @@ void dev_print (block_dev_desc_t *dev_desc)
 }
 #endif
 
-#if (defined(CONFIG_CMD_IDE) || \
-     defined(CONFIG_CMD_SATA) || \
-     defined(CONFIG_CMD_SCSI) || \
-     defined(CONFIG_CMD_USB) || \
-     defined(CONFIG_MMC)               || \
-     defined(CONFIG_SYSTEMACE) )
-
-#if defined(CONFIG_MAC_PARTITION) || \
-    defined(CONFIG_DOS_PARTITION) || \
-    defined(CONFIG_ISO_PARTITION) || \
-    defined(CONFIG_AMIGA_PARTITION) || \
-    defined(CONFIG_EFI_PARTITION)
+#ifdef HAVE_BLOCK_DEVICE
 
 void init_part (block_dev_desc_t * dev_desc)
 {
@@ -292,6 +258,12 @@ void init_part (block_dev_desc_t * dev_desc)
 }
 
 
+#if defined(CONFIG_MAC_PARTITION) || \
+       defined(CONFIG_DOS_PARTITION) || \
+       defined(CONFIG_ISO_PARTITION) || \
+       defined(CONFIG_AMIGA_PARTITION) || \
+       defined(CONFIG_EFI_PARTITION)
+
 static void print_part_header (const char *type, block_dev_desc_t * dev_desc)
 {
        puts ("\nPartition Map for ");
@@ -317,6 +289,9 @@ static void print_part_header (const char *type, block_dev_desc_t * dev_desc)
        case IF_TYPE_MMC:
                puts ("MMC");
                break;
+       case IF_TYPE_HOST:
+               puts("HOST");
+               break;
        default:
                puts ("UNKNOWN");
                break;
@@ -325,6 +300,8 @@ static void print_part_header (const char *type, block_dev_desc_t * dev_desc)
                        dev_desc->dev, type);
 }
 
+#endif /* any CONFIG_..._PARTITION */
+
 void print_part (block_dev_desc_t * dev_desc)
 {
 
@@ -371,24 +348,17 @@ void print_part (block_dev_desc_t * dev_desc)
        puts ("## Unknown partition table\n");
 }
 
-
-#else  /* neither MAC nor DOS nor ISO nor AMIGA nor EFI partition configured */
-# error neither CONFIG_MAC_PARTITION nor CONFIG_DOS_PARTITION
-# error nor CONFIG_ISO_PARTITION nor CONFIG_AMIGA_PARTITION
-# error nor CONFIG_EFI_PARTITION configured!
-#endif
-
-#endif
+#endif /* HAVE_BLOCK_DEVICE */
 
 int get_partition_info(block_dev_desc_t *dev_desc, int part
                                        , disk_partition_t *info)
 {
-#if defined(CONFIG_CMD_IDE) || \
-       defined(CONFIG_CMD_SATA) || \
-       defined(CONFIG_CMD_SCSI) || \
-       defined(CONFIG_CMD_USB) || \
-       defined(CONFIG_MMC) || \
-       defined(CONFIG_SYSTEMACE)
+#ifdef HAVE_BLOCK_DEVICE
+
+#ifdef CONFIG_PARTITION_UUIDS
+       /* The common case is no UUID support */
+       info->uuid[0] = 0;
+#endif
 
        switch (dev_desc->part_type) {
 #ifdef CONFIG_MAC_PARTITION
@@ -438,7 +408,7 @@ int get_partition_info(block_dev_desc_t *dev_desc, int part
        default:
                break;
        }
-#endif
+#endif /* HAVE_BLOCK_DEVICE */
 
        return -1;
 }
@@ -465,66 +435,183 @@ int get_device(const char *ifname, const char *dev_str,
        return dev;
 }
 
-int get_device_and_partition(const char *ifname, const char *dev_str,
+#define PART_UNSPECIFIED -2
+#define PART_AUTO -1
+#define MAX_SEARCH_PARTITIONS 16
+int get_device_and_partition(const char *ifname, const char *dev_part_str,
                             block_dev_desc_t **dev_desc,
-                            disk_partition_t *info)
+                            disk_partition_t *info, int allow_whole_dev)
 {
-       int ret;
-       char *ep;
+       int ret = -1;
+       const char *part_str;
+       char *dup_str = NULL;
+       const char *dev_str;
        int dev;
-       block_dev_desc_t *desc;
-       int part = 0;
-       char *part_str;
-
-       if (dev_str)
-               dev = simple_strtoul(dev_str, &ep, 16);
-
-       if (!dev_str || (dev_str == ep)) {
-               dev_str = getenv("bootdevice");
-               if (dev_str)
-                       dev = simple_strtoul(dev_str, &ep, 16);
-               if (!dev_str || (dev_str == ep))
-                       goto err;
+       char *ep;
+       int p;
+       int part;
+       disk_partition_t tmpinfo;
+
+       /* If no dev_part_str, use bootdevice environment variable */
+       if (!dev_part_str || !strlen(dev_part_str) ||
+           !strcmp(dev_part_str, "-"))
+               dev_part_str = getenv("bootdevice");
+
+       /* If still no dev_part_str, it's an error */
+       if (!dev_part_str) {
+               printf("** No device specified **\n");
+               goto cleanup;
        }
 
-       desc = get_dev(ifname, dev);
-       if (!desc || (desc->type == DEV_TYPE_UNKNOWN))
-               goto err;
+       /* Separate device and partition ID specification */
+       part_str = strchr(dev_part_str, ':');
+       if (part_str) {
+               dup_str = strdup(dev_part_str);
+               dup_str[part_str - dev_part_str] = 0;
+               dev_str = dup_str;
+               part_str++;
+       } else {
+               dev_str = dev_part_str;
+       }
+
+       /* Look up the device */
+       dev = get_device(ifname, dev_str, dev_desc);
+       if (dev < 0)
+               goto cleanup;
+
+       /* Convert partition ID string to number */
+       if (!part_str || !*part_str) {
+               part = PART_UNSPECIFIED;
+       } else if (!strcmp(part_str, "auto")) {
+               part = PART_AUTO;
+       } else {
+               /* Something specified -> use exactly that */
+               part = (int)simple_strtoul(part_str, &ep, 16);
+               /*
+                * Less than whole string converted,
+                * or request for whole device, but caller requires partition.
+                */
+               if (*ep || (part == 0 && !allow_whole_dev)) {
+                       printf("** Bad partition specification %s %s **\n",
+                           ifname, dev_part_str);
+                       goto cleanup;
+               }
+       }
+
+       /*
+        * No partition table on device,
+        * or user requested partition 0 (entire device).
+        */
+       if (((*dev_desc)->part_type == PART_TYPE_UNKNOWN) ||
+           (part == 0)) {
+               if (!(*dev_desc)->lba) {
+                       printf("** Bad device size - %s %s **\n", ifname,
+                              dev_str);
+                       goto cleanup;
+               }
 
-       if (desc->part_type == PART_TYPE_UNKNOWN) {
-               /* disk doesn't use partition table */
-               if (!desc->lba) {
-                       printf("**Bad disk size - %s %d:0 **\n", ifname, dev);
-                       return -1;
+               /*
+                * If user specified a partition ID other than 0,
+                * or the calling command only accepts partitions,
+                * it's an error.
+                */
+               if ((part > 0) || (!allow_whole_dev)) {
+                       printf("** No partition table - %s %s **\n", ifname,
+                              dev_str);
+                       goto cleanup;
                }
+
+               (*dev_desc)->log2blksz = LOG2((*dev_desc)->blksz);
+
                info->start = 0;
-               info->size = desc->lba;
-               info->blksz = desc->blksz;
+               info->size = (*dev_desc)->lba;
+               info->blksz = (*dev_desc)->blksz;
+               info->bootable = 0;
+               strcpy((char *)info->type, BOOT_PART_TYPE);
+               strcpy((char *)info->name, "Whole Disk");
+#ifdef CONFIG_PARTITION_UUIDS
+               info->uuid[0] = 0;
+#endif
 
-               *dev_desc = desc;
-               return 0;
+               ret = 0;
+               goto cleanup;
        }
 
-       part_str = strchr(dev_str, ':');
-       if (part_str)
-               part = (int)simple_strtoul(++part_str, NULL, 16);
-
-       ret = get_partition_info(desc, part, info);
-       if (ret) {
-               printf("** Invalid partition %d, use `dev[:part]' **\n", part);
-               return -1;
+       /*
+        * Now there's known to be a partition table,
+        * not specifying a partition means to pick partition 1.
+        */
+       if (part == PART_UNSPECIFIED)
+               part = 1;
+
+       /*
+        * If user didn't specify a partition number, or did specify something
+        * other than "auto", use that partition number directly.
+        */
+       if (part != PART_AUTO) {
+               ret = get_partition_info(*dev_desc, part, info);
+               if (ret) {
+                       printf("** Invalid partition %d **\n", part);
+                       goto cleanup;
+               }
+       } else {
+               /*
+                * Find the first bootable partition.
+                * If none are bootable, fall back to the first valid partition.
+                */
+               part = 0;
+               for (p = 1; p <= MAX_SEARCH_PARTITIONS; p++) {
+                       ret = get_partition_info(*dev_desc, p, info);
+                       if (ret)
+                               continue;
+
+                       /*
+                        * First valid partition, or new better partition?
+                        * If so, save partition ID.
+                        */
+                       if (!part || info->bootable)
+                               part = p;
+
+                       /* Best possible partition? Stop searching. */
+                       if (info->bootable)
+                               break;
+
+                       /*
+                        * We now need to search further for best possible.
+                        * If we what we just queried was the best so far,
+                        * save the info since we over-write it next loop.
+                        */
+                       if (part == p)
+                               tmpinfo = *info;
+               }
+               /* If we found any acceptable partition */
+               if (part) {
+                       /*
+                        * If we searched all possible partition IDs,
+                        * return the first valid partition we found.
+                        */
+                       if (p == MAX_SEARCH_PARTITIONS + 1)
+                               *info = tmpinfo;
+               } else {
+                       printf("** No valid partitions found **\n");
+                       ret = -1;
+                       goto cleanup;
+               }
        }
        if (strncmp((char *)info->type, BOOT_PART_TYPE, sizeof(info->type)) != 0) {
                printf("** Invalid partition type \"%.32s\""
                        " (expect \"" BOOT_PART_TYPE "\")\n",
                        info->type);
-               return -1;
+               ret  = -1;
+               goto cleanup;
        }
 
-       *dev_desc = desc;
-       return part;
+       (*dev_desc)->log2blksz = LOG2((*dev_desc)->blksz);
 
-err:
-       puts("** Invalid boot device, use `dev[:part]' **\n");
-       return -1;
+       ret = part;
+       goto cleanup;
+
+cleanup:
+       free(dup_str);
+       return ret;
 }