Merge tag 'dm-pull-22222' of https://gitlab.denx.de/u-boot/custodians/u-boot-dm
[platform/kernel/u-boot.git] / common / spl / spl_sata.c
1 // SPDX-License-Identifier: GPL-2.0+
2 /*
3  * (C) Copyright 2013
4  * Texas Instruments, <www.ti.com>
5  *
6  * Dan Murphy <dmurphy@ti.com>
7  *
8  * Derived work from spl_usb.c
9  */
10
11 #include <common.h>
12 #include <spl.h>
13 #include <asm/u-boot.h>
14 #include <sata.h>
15 #include <scsi.h>
16 #include <errno.h>
17 #include <fat.h>
18 #include <image.h>
19
20 #ifndef CONFIG_SYS_SATA_FAT_BOOT_PARTITION
21 #define CONFIG_SYS_SATA_FAT_BOOT_PARTITION      1
22 #endif
23
24 #ifndef CONFIG_SPL_FS_LOAD_PAYLOAD_NAME
25 #define CONFIG_SPL_FS_LOAD_PAYLOAD_NAME "u-boot.img"
26 #endif
27
28 #ifndef CONFIG_SPL_SATA_RAW_U_BOOT_SECTOR
29 /* Dummy value to make the compiler happy */
30 #define CONFIG_SPL_SATA_RAW_U_BOOT_SECTOR 0x100
31 #endif
32
33 static int spl_sata_load_image_raw(struct spl_image_info *spl_image,
34                 struct spl_boot_device *bootdev,
35                 struct blk_desc *stor_dev, unsigned long sector)
36 {
37         struct image_header *header;
38         unsigned long count;
39         u32 image_size_sectors;
40         u32 image_offset_sectors;
41         u32 image_offset;
42         int ret;
43
44         header = spl_get_load_buffer(-sizeof(*header), stor_dev->blksz);
45         count = blk_dread(stor_dev, sector, 1, header);
46         if (count == 0)
47                 return -EIO;
48
49         ret = spl_parse_image_header(spl_image, bootdev, header);
50         if (ret)
51                 return ret;
52
53         image_size_sectors = DIV_ROUND_UP(spl_image->size, stor_dev->blksz);
54         image_offset_sectors = spl_image->offset / stor_dev->blksz;
55         image_offset = spl_image->offset % stor_dev->blksz;
56         count = blk_dread(stor_dev, sector + image_offset_sectors,
57                         image_size_sectors,
58                         (void *)spl_image->load_addr);
59         if (count != image_size_sectors)
60                 return -EIO;
61
62         if (image_offset)
63                 memmove((void *)spl_image->load_addr,
64                         (void *)spl_image->load_addr + image_offset,
65                         spl_image->size);
66
67         return 0;
68 }
69
70 static int spl_sata_load_image(struct spl_image_info *spl_image,
71                                struct spl_boot_device *bootdev)
72 {
73         int err = 0;
74         struct blk_desc *stor_dev;
75
76 #if !defined(CONFIG_DM_SCSI) && !defined(CONFIG_AHCI)
77         err = init_sata(CONFIG_SPL_SATA_BOOT_DEVICE);
78 #endif
79         if (err) {
80 #ifdef CONFIG_SPL_LIBCOMMON_SUPPORT
81                 printf("spl: sata init failed: err - %d\n", err);
82 #endif
83                 return err;
84         } else {
85                 /* try to recognize storage devices immediately */
86                 scsi_scan(false);
87                 stor_dev = blk_get_devnum_by_type(IF_TYPE_SCSI, 0);
88                 if (!stor_dev)
89                         return -ENODEV;
90         }
91
92 #if CONFIG_IS_ENABLED(OS_BOOT)
93         if (spl_start_uboot() ||
94             spl_load_image_fat_os(spl_image, bootdev, stor_dev,
95                                   CONFIG_SYS_SATA_FAT_BOOT_PARTITION))
96 #endif
97         {
98                 err = -ENOSYS;
99
100                 if (IS_ENABLED(CONFIG_SPL_FS_FAT)) {
101                         err = spl_load_image_fat(spl_image, bootdev, stor_dev,
102                                         CONFIG_SYS_SATA_FAT_BOOT_PARTITION,
103                                         CONFIG_SPL_FS_LOAD_PAYLOAD_NAME);
104                 } else if (IS_ENABLED(CONFIG_SPL_SATA_RAW_U_BOOT_USE_SECTOR)) {
105                         err = spl_sata_load_image_raw(spl_image, bootdev, stor_dev,
106                                 CONFIG_SPL_SATA_RAW_U_BOOT_SECTOR);
107                 }
108         }
109         if (err) {
110                 puts("Error loading sata device\n");
111                 return err;
112         }
113
114         return 0;
115 }
116 SPL_LOAD_IMAGE_METHOD("SATA", 0, BOOT_DEVICE_SATA, spl_sata_load_image);