1/* SPDX-License-Identifier: GPL-2.0+ */
2/*
3 * crt0-efi-riscv.S - PE/COFF header for RISC-V EFI applications
4 *
5 * Copright (C) 2014 Linaro Ltd. <ard.biesheuvel@linaro.org>
6 * Copright (C) 2018 Alexander Graf <agraf@suse.de>
7 *
8 * This file is inspired by arch/arm/lib/crt0_aarch64_efi.S
9 */
10
11#include <asm-generic/pe.h>
12
13#if __riscv_xlen == 64
14#define SIZE_LONG	8
15#define SAVE_LONG(reg, idx)	sd	reg, (idx*SIZE_LONG)(sp)
16#define LOAD_LONG(reg, idx)	ld	reg, (idx*SIZE_LONG)(sp)
17#define PE_MACHINE	IMAGE_FILE_MACHINE_RISCV64
18#define PE_MAGIC    IMAGE_NT_OPTIONAL_HDR64_MAGIC
19#define IMG_CHARACTERISTICS \
20	(IMAGE_FILE_EXECUTABLE_IMAGE | \
21	 IMAGE_FILE_LINE_NUMS_STRIPPED | \
22	 IMAGE_FILE_LOCAL_SYMS_STRIPPED | \
23	 IMAGE_FILE_LARGE_ADDRESS_AWARE | \
24	 IMAGE_FILE_DEBUG_STRIPPED)
25#else
26#define SIZE_LONG	4
27#define SAVE_LONG(reg, idx)	sw	reg, (idx*SIZE_LONG)(sp)
28#define LOAD_LONG(reg, idx)	lw	reg, (idx*SIZE_LONG)(sp)
29#define PE_MACHINE	IMAGE_FILE_MACHINE_RISCV32
30#define PE_MAGIC    IMAGE_NT_OPTIONAL_HDR32_MAGIC
31#define IMG_CHARACTERISTICS \
32	(IMAGE_FILE_EXECUTABLE_IMAGE | \
33	 IMAGE_FILE_LINE_NUMS_STRIPPED | \
34	 IMAGE_FILE_LOCAL_SYMS_STRIPPED | \
35	 IMAGE_FILE_DEBUG_STRIPPED)
36#endif
37
38
39	.section	.text.head
40
41	/*
42	 * Magic "MZ" signature for PE/COFF
43	 */
44	.globl	ImageBase
45ImageBase:
46	.short	IMAGE_DOS_SIGNATURE		/* 'MZ' */
47	.skip	46				/* 'MZ' + pad + offset == 64 */
48	.long	0x43534952			/* Linux magic "RISCV */
49	.long	0x00000056
50	.long	0x05435352			/* Linux magic2 "RSC\x05*/
51	.long	pe_header - ImageBase		/* Offset to the PE header */
52pe_header:
53	.long	IMAGE_NT_SIGNATURE		/* 'PE' */
54coff_header:
55	.short	PE_MACHINE			/* RISC-V 64/32-bit */
56	.short	2				/* nr_sections */
57	.long	0				/* TimeDateStamp */
58	.long	0				/* PointerToSymbolTable */
59	.long	0				/* NumberOfSymbols */
60	.short	section_table - optional_header	/* SizeOfOptionalHeader */
61	.short	IMG_CHARACTERISTICS		/* Characteristics */
62optional_header:
63	.short	PE_MAGIC			/* PE32(+) format */
64	.byte	0x02				/* MajorLinkerVersion */
65	.byte	0x14				/* MinorLinkerVersion */
66	.long	_edata - _start			/* SizeOfCode */
67	.long	0				/* SizeOfInitializedData */
68	.long	0				/* SizeOfUninitializedData */
69	.long	_start - ImageBase		/* AddressOfEntryPoint */
70	.long	_start - ImageBase		/* BaseOfCode */
71#if __riscv_xlen == 32
72	.long	0				/* BaseOfData */
73#endif
74
75extra_header_fields:
76#if __riscv_xlen == 32
77	.long	0				/* ImageBase */
78#else
79	.quad	0				/* ImageBase */
80#endif
81	.long	0x200				/* SectionAlignment */
82	.long	0x200				/* FileAlignment */
83	.short	0				/* MajorOperatingSystemVersion */
84	.short	0				/* MinorOperatingSystemVersion */
85	.short	1				/* MajorImageVersion */
86	.short	0				/* MinorImageVersion */
87	.short	0				/* MajorSubsystemVersion */
88	.short	0				/* MinorSubsystemVersion */
89	.long	0				/* Win32VersionValue */
90
91	.long	_edata - ImageBase		/* SizeOfImage */
92
93	/*
94	 * Everything before the kernel image is considered part of the header
95	 */
96	.long	_start - ImageBase		/* SizeOfHeaders */
97	.long	0				/* CheckSum */
98	.short	IMAGE_SUBSYSTEM_EFI_APPLICATION /* Subsystem */
99	.short	0				/* DllCharacteristics */
100#if __riscv_xlen == 32
101	.long	0				/* SizeOfStackReserve */
102	.long	0				/* SizeOfStackCommit */
103	.long	0				/* SizeOfHeapReserve */
104	.long	0				/* SizeOfHeapCommit */
105#else
106	.quad	0				/* SizeOfStackReserve */
107	.quad	0				/* SizeOfStackCommit */
108	.quad	0				/* SizeOfHeapReserve */
109	.quad	0				/* SizeOfHeapCommit */
110#endif
111	.long	0				/* LoaderFlags */
112	.long	0x6				/* NumberOfRvaAndSizes */
113
114	.quad	0				/* ExportTable */
115	.quad	0				/* ImportTable */
116	.quad	0				/* ResourceTable */
117	.quad	0				/* ExceptionTable */
118	.quad	0				/* CertificationTable */
119	.quad	0				/* BaseRelocationTable */
120
121	/* Section table */
122section_table:
123
124	/*
125	 * The EFI application loader requires a relocation section
126	 * because EFI applications must be relocatable.  This is a
127	 * dummy section as far as we are concerned.
128	 */
129	.ascii	".reloc"
130	.byte	0
131	.byte	0			/* end of 0 padding of section name */
132	.long	0
133	.long	0
134	.long	0			/* SizeOfRawData */
135	.long	0			/* PointerToRawData */
136	.long	0			/* PointerToRelocations */
137	.long	0			/* PointerToLineNumbers */
138	.short	0			/* NumberOfRelocations */
139	.short	0			/* NumberOfLineNumbers */
140	.long	0x42100040		/* Characteristics (section flags) */
141
142
143	.ascii	".text"
144	.byte	0
145	.byte	0
146	.byte	0			/* end of 0 padding of section name */
147	.long	_edata - _start		/* VirtualSize */
148	.long	_start - ImageBase	/* VirtualAddress */
149	.long	_edata - _start		/* SizeOfRawData */
150	.long	_start - ImageBase	/* PointerToRawData */
151
152	.long	0		/* PointerToRelocations (0 for executables) */
153	.long	0		/* PointerToLineNumbers (0 for executables) */
154	.short	0		/* NumberOfRelocations  (0 for executables) */
155	.short	0		/* NumberOfLineNumbers  (0 for executables) */
156	.long	0xe0500020	/* Characteristics (section flags) */
157
158	.align	9
159_start:
160	addi		sp, sp, -(SIZE_LONG * 3)
161	SAVE_LONG(a0, 0)
162	SAVE_LONG(a1, 1)
163	SAVE_LONG(ra, 2)
164
165	lla		a0, ImageBase
166	lla		a1, _DYNAMIC
167	call		_relocate
168	bne		a0, zero, 0f
169
170	LOAD_LONG(a1, 1)
171	LOAD_LONG(a0, 0)
172	call		efi_main
173
174	LOAD_LONG(ra, 2)
175
1760:	addi		sp, sp, (SIZE_LONG * 3)
177	ret
178