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
16DECLARE_GLOBAL_DATA_PTR;
17
18/* ASCII version of "RSC\0x5" defined in Linux kernel */
19#define LINUX_RISCV_IMAGE_MAGIC 0x05435352
20
21struct 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
35int 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