cmd/disk.c

   1 // SPDX-License-Identifier: GPL-2.0+
   2 /*
   3  * (C) Copyright 2000-2011
   4  * Wolfgang Denk, DENX Software Engineering, [email protected].
   5  */
   6 #include <bootstage.h>
   7 #include <command.h>
   8 #include <cpu_func.h>
   9 #include <image.h>
  10 #include <log.h>
  11 #include <part.h>
  12
  13 int common_diskboot(struct cmd_tbl *cmdtp, const char *intf, int argc,
  14                     char *const argv[])
  15 {
  16         __maybe_unused int dev;
  17         int part;
  18         ulong addr = CONFIG_SYS_LOAD_ADDR;
  19         ulong cnt;
  20         struct disk_partition info;
  21 #if defined(CONFIG_LEGACY_IMAGE_FORMAT)
  22         struct legacy_img_hdr *hdr;
  23 #endif
  24         struct blk_desc *dev_desc;
  25
  26 #if CONFIG_IS_ENABLED(FIT)
  27         const void *fit_hdr = NULL;
  28 #endif
  29
  30         bootstage_mark(BOOTSTAGE_ID_IDE_START);
  31         if (argc > 3) {
  32                 bootstage_error(BOOTSTAGE_ID_IDE_ADDR);
  33                 return CMD_RET_USAGE;
  34         }
  35         bootstage_mark(BOOTSTAGE_ID_IDE_ADDR);
  36
  37         if (argc > 1)
  38                 addr = hextoul(argv[1], NULL);
  39
  40         bootstage_mark(BOOTSTAGE_ID_IDE_BOOT_DEVICE);
  41
  42         part = blk_get_device_part_str(intf, cmd_arg2(argc, argv),
  43                                        &dev_desc, &info, 1);
  44         if (part < 0) {
  45                 bootstage_error(BOOTSTAGE_ID_IDE_TYPE);
  46                 return 1;
  47         }
  48
  49         dev = dev_desc->devnum;
  50         bootstage_mark(BOOTSTAGE_ID_IDE_TYPE);
  51
  52         printf("\nLoading from %s device %d, partition %d: "
  53                "Name: %.32s  Type: %.32s\n", intf, dev, part, info.name,
  54                info.type);
  55
  56         debug("First Block: " LBAFU ",  # of blocks: " LBAFU
  57               ", Block Size: %ld\n",
  58               info.start, info.size, info.blksz);
  59
  60         if (blk_dread(dev_desc, info.start, 1, (ulong *)addr) != 1) {
  61                 printf("** Read error on %d:%d\n", dev, part);
  62                 bootstage_error(BOOTSTAGE_ID_IDE_PART_READ);
  63                 return 1;
  64         }
  65         bootstage_mark(BOOTSTAGE_ID_IDE_PART_READ);
  66
  67         switch (genimg_get_format((void *) addr)) {
  68 #if defined(CONFIG_LEGACY_IMAGE_FORMAT)
  69         case IMAGE_FORMAT_LEGACY:
  70                 hdr = (struct legacy_img_hdr *)addr;
  71
  72                 bootstage_mark(BOOTSTAGE_ID_IDE_FORMAT);
  73
  74                 if (!image_check_hcrc(hdr)) {
  75                         puts("\n** Bad Header Checksum **\n");
  76                         bootstage_error(BOOTSTAGE_ID_IDE_CHECKSUM);
  77                         return 1;
  78                 }
  79                 bootstage_mark(BOOTSTAGE_ID_IDE_CHECKSUM);
  80
  81                 image_print_contents(hdr);
  82
  83                 cnt = image_get_image_size(hdr);
  84                 break;
  85 #endif
  86 #if CONFIG_IS_ENABLED(FIT)
  87         case IMAGE_FORMAT_FIT:
  88                 fit_hdr = (const void *) addr;
  89                 puts("Fit image detected...\n");
  90
  91                 cnt = fit_get_size(fit_hdr);
  92                 break;
  93 #endif
  94         default:
  95                 bootstage_error(BOOTSTAGE_ID_IDE_FORMAT);
  96                 puts("** Unknown image type\n");
  97                 return 1;
  98         }
  99
 100         cnt += info.blksz - 1;
 101         cnt /= info.blksz;
 102         cnt -= 1;
 103
 104         if (blk_dread(dev_desc, info.start + 1, cnt,
 105                       (ulong *)(addr + info.blksz)) != cnt) {
 106                 printf("** Read error on %d:%d\n", dev, part);
 107                 bootstage_error(BOOTSTAGE_ID_IDE_READ);
 108                 return 1;
 109         }
 110         bootstage_mark(BOOTSTAGE_ID_IDE_READ);
 111
 112 #if CONFIG_IS_ENABLED(FIT)
 113         /* This cannot be done earlier,
 114          * we need complete FIT image in RAM first */
 115         if (genimg_get_format((void *) addr) == IMAGE_FORMAT_FIT) {
 116                 if (fit_check_format(fit_hdr, IMAGE_SIZE_INVAL)) {
 117                         bootstage_error(BOOTSTAGE_ID_IDE_FIT_READ);
 118                         puts("** Bad FIT image format\n");
 119                         return 1;
 120                 }
 121                 bootstage_mark(BOOTSTAGE_ID_IDE_FIT_READ_OK);
 122         }
 123 #endif
 124
 125         flush_cache(addr, (cnt+1)*info.blksz);
 126
 127         /* Loading ok, update default load address */
 128         image_load_addr = addr;
 129
 130         return bootm_maybe_autostart(cmdtp, argv[0]);
 131 }