spl: Make use of CONFIG_IS_ENABLED(OS_BOOT) in SPL/TPL common code paths
[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 blk_desc *stor_dev, unsigned long sector)
35 {
36         struct image_header *header;
37         unsigned long count;
38         u32 image_size_sectors;
39         u32 image_offset_sectors;
40         u32 image_offset;
41         int ret;
42
43         header = spl_get_load_buffer(-sizeof(*header), stor_dev->blksz);
44         count = blk_dread(stor_dev, sector, 1, header);
45         if (count == 0)
46                 return -EIO;
47
48         ret = spl_parse_image_header(spl_image, header);
49         if (ret)
50                 return ret;
51
52         image_size_sectors = DIV_ROUND_UP(spl_image->size, stor_dev->blksz);
53         image_offset_sectors = spl_image->offset / stor_dev->blksz;
54         image_offset = spl_image->offset % stor_dev->blksz;
55         count = blk_dread(stor_dev, sector + image_offset_sectors,
56                         image_size_sectors,
57                         (void *)spl_image->load_addr);
58         if (count != image_size_sectors)
59                 return -EIO;
60
61         if (image_offset)
62                 memmove((void *)spl_image->load_addr,
63                         (void *)spl_image->load_addr + image_offset,
64                         spl_image->size);
65
66         return 0;
67 }
68
69 static int spl_sata_load_image(struct spl_image_info *spl_image,
70                                struct spl_boot_device *bootdev)
71 {
72         int err = 0;
73         struct blk_desc *stor_dev;
74
75 #if !defined(CONFIG_DM_SCSI) && !defined(CONFIG_AHCI)
76         err = init_sata(CONFIG_SPL_SATA_BOOT_DEVICE);
77 #endif
78         if (err) {
79 #ifdef CONFIG_SPL_LIBCOMMON_SUPPORT
80                 printf("spl: sata init failed: err - %d\n", err);
81 #endif
82                 return err;
83         } else {
84                 /* try to recognize storage devices immediately */
85                 scsi_scan(false);
86                 stor_dev = blk_get_devnum_by_type(IF_TYPE_SCSI, 0);
87                 if (!stor_dev)
88                         return -ENODEV;
89         }
90
91 #if CONFIG_IS_ENABLED(OS_BOOT)
92         if (spl_start_uboot() ||
93             spl_load_image_fat_os(spl_image, stor_dev,
94                                   CONFIG_SYS_SATA_FAT_BOOT_PARTITION))
95 #endif
96         {
97                 err = -ENOSYS;
98
99                 if (IS_ENABLED(CONFIG_SPL_FS_FAT)) {
100                         err = spl_load_image_fat(spl_image, stor_dev,
101                                         CONFIG_SYS_SATA_FAT_BOOT_PARTITION,
102                                         CONFIG_SPL_FS_LOAD_PAYLOAD_NAME);
103                 } else if (IS_ENABLED(CONFIG_SPL_SATA_RAW_U_BOOT_USE_SECTOR)) {
104                         err = spl_sata_load_image_raw(spl_image, stor_dev,
105                                 CONFIG_SPL_SATA_RAW_U_BOOT_SECTOR);
106                 }
107         }
108         if (err) {
109                 puts("Error loading sata device\n");
110                 return err;
111         }
112
113         return 0;
114 }
115 SPL_LOAD_IMAGE_METHOD("SATA", 0, BOOT_DEVICE_SATA, spl_sata_load_image);