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#else
20#define SIZE_LONG	4
21#define SAVE_LONG(reg, idx)	sw	reg, (idx*SIZE_LONG)(sp)
22#define LOAD_LONG(reg, idx)	lw	reg, (idx*SIZE_LONG)(sp)
23#define PE_MACHINE	IMAGE_FILE_MACHINE_RISCV32
24#define PE_MAGIC    IMAGE_NT_OPTIONAL_HDR32_MAGIC
25#endif
26
27
28	.section	.text.head
29
30	/*
31	 * Magic "MZ" signature for PE/COFF
32	 */
33	.globl	ImageBase
34ImageBase:
35	.short	IMAGE_DOS_SIGNATURE		/* 'MZ' */
36	.skip	58				/* 'MZ' + pad + offset == 64 */
37	.long	pe_header - ImageBase		/* Offset to the PE header */
38pe_header:
39	.long	IMAGE_NT_SIGNATURE		/* 'PE' */
40coff_header:
41	.short	PE_MACHINE			/* RISC-V 64/32-bit */
42	.short	2				/* nr_sections */
43	.long	0				/* TimeDateStamp */
44	.long	0				/* PointerToSymbolTable */
45	.long	0				/* NumberOfSymbols */
46	.short	section_table - optional_header	/* SizeOfOptionalHeader */
47	/* Characteristics */
48	.short	(IMAGE_FILE_EXECUTABLE_IMAGE | \
49		 IMAGE_FILE_LINE_NUMS_STRIPPED | \
50		 IMAGE_FILE_LOCAL_SYMS_STRIPPED | \
51		 IMAGE_FILE_DEBUG_STRIPPED)
52optional_header:
53	.short	PE_MAGIC			/* PE32(+) format */
54	.byte	0x02				/* MajorLinkerVersion */
55	.byte	0x14				/* MinorLinkerVersion */
56	.long	_edata - _start			/* SizeOfCode */
57	.long	0				/* SizeOfInitializedData */
58	.long	0				/* SizeOfUninitializedData */
59	.long	_start - ImageBase		/* AddressOfEntryPoint */
60	.long	_start - ImageBase		/* BaseOfCode */
61#if __riscv_xlen == 32
62	.long	0				/* BaseOfData */
63#endif
64
65extra_header_fields:
66#if __riscv_xlen == 32
67	.long	0				/* ImageBase */
68#else
69	.quad	0				/* ImageBase */
70#endif
71	.long	0x20				/* SectionAlignment */
72	.long	0x8				/* FileAlignment */
73	.short	0				/* MajorOperatingSystemVersion */
74	.short	0				/* MinorOperatingSystemVersion */
75	.short	0				/* MajorImageVersion */
76	.short	0				/* MinorImageVersion */
77	.short	0				/* MajorSubsystemVersion */
78	.short	0				/* MinorSubsystemVersion */
79	.long	0				/* Win32VersionValue */
80
81	.long	_edata - ImageBase		/* SizeOfImage */
82
83	/*
84	 * Everything before the kernel image is considered part of the header
85	 */
86	.long	_start - ImageBase		/* SizeOfHeaders */
87	.long	0				/* CheckSum */
88	.short	IMAGE_SUBSYSTEM_EFI_APPLICATION /* Subsystem */
89	.short	0				/* DllCharacteristics */
90#if __riscv_xlen == 32
91	.long	0				/* SizeOfStackReserve */
92	.long	0				/* SizeOfStackCommit */
93	.long	0				/* SizeOfHeapReserve */
94	.long	0				/* SizeOfHeapCommit */
95#else
96	.quad	0				/* SizeOfStackReserve */
97	.quad	0				/* SizeOfStackCommit */
98	.quad	0				/* SizeOfHeapReserve */
99	.quad	0				/* SizeOfHeapCommit */
100#endif
101	.long	0				/* LoaderFlags */
102	.long	0x6				/* NumberOfRvaAndSizes */
103
104	.quad	0				/* ExportTable */
105	.quad	0				/* ImportTable */
106	.quad	0				/* ResourceTable */
107	.quad	0				/* ExceptionTable */
108	.quad	0				/* CertificationTable */
109	.quad	0				/* BaseRelocationTable */
110	.quad	0				/* Debug */
111	.quad	0				/* Architecture */
112	.quad	0				/* Global Ptr */
113	.quad	0				/* TLS Table */
114	.quad	0				/* Load Config Table */
115	.quad	0				/* Bound Import */
116	.quad	0				/* IAT */
117	.quad	0				/* Delay Import Descriptor */
118	.quad	0				/* CLR Runtime Header */
119	.quad	0				/* Reserved */
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_start:
159	addi		sp, sp, -(SIZE_LONG * 3)
160	SAVE_LONG(a0, 0)
161	SAVE_LONG(a1, 1)
162	SAVE_LONG(ra, 2)
163
164	lla		a0, ImageBase
165	lla		a1, _DYNAMIC
166	call		_relocate
167	bne		a0, zero, 0f
168
169	LOAD_LONG(a1, 1)
170	LOAD_LONG(a0, 0)
171	call		efi_main
172
173	LOAD_LONG(ra, 2)
174
1750:	addi		sp, sp, (SIZE_LONG * 3)
176	ret
177