1 // SPDX-License-Identifier: GPL-2.0+
2 /*
3  * Copyright (C) 2019 Western Digital Corporation or its affiliates.
4  * Authors:
5  *	Atish Patra <atish.patra@wdc.com>
6  * Based on arm/lib/image.c
7  */
8 
9 #include <image.h>
10 #include <mapmem.h>
11 #include <errno.h>
12 #include <asm/global_data.h>
13 #include <linux/sizes.h>
14 #include <linux/stddef.h>
15 
16 DECLARE_GLOBAL_DATA_PTR;
17 
18 /* ASCII version of "RSC\0x5" defined in Linux kernel */
19 #define LINUX_RISCV_IMAGE_MAGIC 0x05435352
20 
21 struct linux_image_h {
22 	uint32_t	code0;		/* Executable code */
23 	uint32_t	code1;		/* Executable code */
24 	uint64_t	text_offset;	/* Image load offset */
25 	uint64_t	image_size;	/* Effective Image size */
26 	uint64_t	flags;		/* kernel flags (little endian) */
27 	uint32_t	version;	/* version of the header */
28 	uint32_t	res1;		/* reserved */
29 	uint64_t	res2;		/* reserved */
30 	uint64_t	res3;		/* reserved */
31 	uint32_t	magic;		/* Magic number */
32 	uint32_t	res4;		/* reserved */
33 };
34 
booti_setup(ulong image,ulong * relocated_addr,ulong * size,bool force_reloc)35 int booti_setup(ulong image, ulong *relocated_addr, ulong *size,
36 		bool force_reloc)
37 {
38 	struct linux_image_h *lhdr;
39 
40 	lhdr = (struct linux_image_h *)map_sysmem(image, 0);
41 
42 	if (lhdr->magic != LINUX_RISCV_IMAGE_MAGIC) {
43 		puts("Bad Linux RISCV Image magic!\n");
44 		return -EINVAL;
45 	}
46 
47 	if (lhdr->image_size == 0) {
48 		puts("Image lacks image_size field, error!\n");
49 		return -EINVAL;
50 	}
51 	*size = lhdr->image_size;
52 	if (force_reloc ||
53 	   (gd->ram_base <= image && image < gd->ram_base + gd->ram_size)) {
54 		*relocated_addr = gd->ram_base + lhdr->text_offset;
55 	} else {
56 		*relocated_addr = image;
57 	}
58 
59 	unmap_sysmem(lhdr);
60 
61 	return 0;
62 }
63