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 <spl.h>
12 #include <sata.h>
13 #include <scsi.h>
14 #include <errno.h>
15 #include <fat.h>
16 #include <image.h>
17 
spl_sata_load_image_raw(struct spl_image_info * spl_image,struct spl_boot_device * bootdev,struct blk_desc * stor_dev,unsigned long sector)18 static int spl_sata_load_image_raw(struct spl_image_info *spl_image,
19 		struct spl_boot_device *bootdev,
20 		struct blk_desc *stor_dev, unsigned long sector)
21 {
22 	struct legacy_img_hdr *header;
23 	unsigned long count;
24 	u32 image_size_sectors;
25 	u32 image_offset_sectors;
26 	u32 image_offset;
27 	int ret;
28 
29 	header = spl_get_load_buffer(-sizeof(*header), stor_dev->blksz);
30 	count = blk_dread(stor_dev, sector, 1, header);
31 	if (count == 0)
32 		return -EIO;
33 
34 	ret = spl_parse_image_header(spl_image, bootdev, header);
35 	if (ret)
36 		return ret;
37 
38 	image_size_sectors = DIV_ROUND_UP(spl_image->size, stor_dev->blksz);
39 	image_offset_sectors = spl_image->offset / stor_dev->blksz;
40 	image_offset = spl_image->offset % stor_dev->blksz;
41 	count = blk_dread(stor_dev, sector + image_offset_sectors,
42 			image_size_sectors,
43 			(void *)spl_image->load_addr);
44 	if (count != image_size_sectors)
45 		return -EIO;
46 
47 	if (image_offset)
48 		memmove((void *)spl_image->load_addr,
49 			(void *)spl_image->load_addr + image_offset,
50 			spl_image->size);
51 
52 	return 0;
53 }
54 
spl_sata_load_image(struct spl_image_info * spl_image,struct spl_boot_device * bootdev)55 static int spl_sata_load_image(struct spl_image_info *spl_image,
56 			       struct spl_boot_device *bootdev)
57 {
58 	int err = -ENOSYS;
59 	struct blk_desc *stor_dev;
60 
61 	/* try to recognize storage devices immediately */
62 	scsi_scan(false);
63 	stor_dev = blk_get_devnum_by_uclass_id(UCLASS_SCSI, 0);
64 	if (!stor_dev)
65 		return -ENODEV;
66 
67 #if CONFIG_IS_ENABLED(OS_BOOT)
68 	if (spl_start_uboot() ||
69 	    spl_load_image_fat_os(spl_image, bootdev, stor_dev,
70 				  CONFIG_SYS_SATA_FAT_BOOT_PARTITION))
71 #endif
72 	{
73 #ifdef CONFIG_SPL_FS_FAT
74 		err = spl_load_image_fat(spl_image, bootdev, stor_dev,
75 				CONFIG_SYS_SATA_FAT_BOOT_PARTITION,
76 				CONFIG_SPL_FS_LOAD_PAYLOAD_NAME);
77 #elif defined(CONFIG_SPL_SATA_RAW_U_BOOT_USE_SECTOR)
78 		err = spl_sata_load_image_raw(spl_image, bootdev, stor_dev,
79 			CONFIG_SPL_SATA_RAW_U_BOOT_SECTOR);
80 #endif
81 	}
82 	if (err) {
83 		puts("Error loading sata device\n");
84 		return err;
85 	}
86 
87 	return 0;
88 }
89 SPL_LOAD_IMAGE_METHOD("SATA", 0, BOOT_DEVICE_SATA, spl_sata_load_image);
90