1 // SPDX-License-Identifier: GPL-2.0+ OR BSD-3-Clause
3 * Copyright (C) 2019, STMicroelectronics - All Rights Reserved
9 #include <remoteproc.h>
11 /* Basic function to verify ELF32 image format */
12 int rproc_elf32_sanity_check(ulong addr, ulong size)
18 pr_debug("Invalid fw address?\n");
22 if (size < sizeof(Elf32_Ehdr)) {
23 pr_debug("Image is too small\n");
27 ehdr = (Elf32_Ehdr *)addr;
28 class = ehdr->e_ident[EI_CLASS];
30 if (!IS_ELF(*ehdr) || ehdr->e_type != ET_EXEC || class != ELFCLASS32) {
31 pr_debug("Not an executable ELF32 image\n");
32 return -EPROTONOSUPPORT;
35 /* We assume the firmware has the same endianness as the host */
36 # ifdef __LITTLE_ENDIAN
37 if (ehdr->e_ident[EI_DATA] != ELFDATA2LSB) {
38 # else /* BIG ENDIAN */
39 if (ehdr->e_ident[EI_DATA] != ELFDATA2MSB) {
41 pr_debug("Unsupported firmware endianness\n");
45 if (size < ehdr->e_shoff + sizeof(Elf32_Shdr)) {
46 pr_debug("Image is too small\n");
50 if (memcmp(ehdr->e_ident, ELFMAG, SELFMAG)) {
51 pr_debug("Image is corrupted (bad magic)\n");
55 if (ehdr->e_phnum == 0) {
56 pr_debug("No loadable segments\n");
60 if (ehdr->e_phoff > size) {
61 pr_debug("Firmware size is too small\n");
68 /* Basic function to verify ELF64 image format */
69 int rproc_elf64_sanity_check(ulong addr, ulong size)
71 Elf64_Ehdr *ehdr = (Elf64_Ehdr *)addr;
75 pr_debug("Invalid fw address?\n");
79 if (size < sizeof(Elf64_Ehdr)) {
80 pr_debug("Image is too small\n");
84 class = ehdr->e_ident[EI_CLASS];
86 if (!IS_ELF(*ehdr) || ehdr->e_type != ET_EXEC || class != ELFCLASS64) {
87 pr_debug("Not an executable ELF64 image\n");
88 return -EPROTONOSUPPORT;
91 /* We assume the firmware has the same endianness as the host */
92 # ifdef __LITTLE_ENDIAN
93 if (ehdr->e_ident[EI_DATA] != ELFDATA2LSB) {
94 # else /* BIG ENDIAN */
95 if (ehdr->e_ident[EI_DATA] != ELFDATA2MSB) {
97 pr_debug("Unsupported firmware endianness\n");
101 if (size < ehdr->e_shoff + sizeof(Elf64_Shdr)) {
102 pr_debug("Image is too small\n");
106 if (memcmp(ehdr->e_ident, ELFMAG, SELFMAG)) {
107 pr_debug("Image is corrupted (bad magic)\n");
111 if (ehdr->e_phnum == 0) {
112 pr_debug("No loadable segments\n");
116 if (ehdr->e_phoff > size) {
117 pr_debug("Firmware size is too small\n");
124 /* Basic function to verify ELF image format */
125 int rproc_elf_sanity_check(ulong addr, ulong size)
127 Elf32_Ehdr *ehdr = (Elf32_Ehdr *)addr;
130 dev_err(dev, "Invalid firmware address\n");
134 if (ehdr->e_ident[EI_CLASS] == ELFCLASS64)
135 return rproc_elf64_sanity_check(addr, size);
137 return rproc_elf32_sanity_check(addr, size);
140 int rproc_elf32_load_image(struct udevice *dev, unsigned long addr, ulong size)
142 Elf32_Ehdr *ehdr; /* Elf header structure pointer */
143 Elf32_Phdr *phdr; /* Program header structure pointer */
144 const struct dm_rproc_ops *ops;
147 ret = rproc_elf32_sanity_check(addr, size);
149 dev_err(dev, "Invalid ELF32 Image %d\n", ret);
153 ehdr = (Elf32_Ehdr *)addr;
154 phdr = (Elf32_Phdr *)(addr + ehdr->e_phoff);
156 ops = rproc_get_ops(dev);
158 /* Load each program header */
159 for (i = 0; i < ehdr->e_phnum; ++i) {
160 void *dst = (void *)(uintptr_t)phdr->p_paddr;
161 void *src = (void *)addr + phdr->p_offset;
163 if (phdr->p_type != PT_LOAD)
166 if (ops->device_to_virt)
167 dst = ops->device_to_virt(dev, (ulong)dst,
170 dev_dbg(dev, "Loading phdr %i to 0x%p (%i bytes)\n",
171 i, dst, phdr->p_filesz);
173 memcpy(dst, src, phdr->p_filesz);
174 if (phdr->p_filesz != phdr->p_memsz)
175 memset(dst + phdr->p_filesz, 0x00,
176 phdr->p_memsz - phdr->p_filesz);
177 flush_cache(rounddown((unsigned long)dst, ARCH_DMA_MINALIGN),
178 roundup((unsigned long)dst + phdr->p_filesz,
180 rounddown((unsigned long)dst, ARCH_DMA_MINALIGN));
187 int rproc_elf64_load_image(struct udevice *dev, ulong addr, ulong size)
189 const struct dm_rproc_ops *ops = rproc_get_ops(dev);
190 u64 da, memsz, filesz, offset;
196 dev_dbg(dev, "%s: addr = 0x%lx size = 0x%lx\n", __func__, addr, size);
198 if (rproc_elf64_sanity_check(addr, size))
201 ehdr = (Elf64_Ehdr *)addr;
202 phdr = (Elf64_Phdr *)(addr + (ulong)ehdr->e_phoff);
204 /* go through the available ELF segments */
205 for (i = 0; i < ehdr->e_phnum; i++, phdr++) {
207 memsz = phdr->p_memsz;
208 filesz = phdr->p_filesz;
209 offset = phdr->p_offset;
211 if (phdr->p_type != PT_LOAD)
214 dev_dbg(dev, "%s:phdr: type %d da 0x%llx memsz 0x%llx filesz 0x%llx\n",
215 __func__, phdr->p_type, da, memsz, filesz);
217 ptr = (void *)(uintptr_t)da;
218 if (ops->device_to_virt) {
219 ptr = ops->device_to_virt(dev, da, phdr->p_memsz);
221 dev_err(dev, "bad da 0x%llx mem 0x%llx\n", da,
229 memcpy(ptr, (void *)addr + offset, filesz);
231 memset(ptr + filesz, 0x00, memsz - filesz);
233 flush_cache(rounddown((ulong)ptr, ARCH_DMA_MINALIGN),
234 roundup((ulong)ptr + filesz, ARCH_DMA_MINALIGN) -
235 rounddown((ulong)ptr, ARCH_DMA_MINALIGN));
241 int rproc_elf_load_image(struct udevice *dev, ulong addr, ulong size)
243 Elf32_Ehdr *ehdr = (Elf32_Ehdr *)addr;
246 dev_err(dev, "Invalid firmware address\n");
250 if (ehdr->e_ident[EI_CLASS] == ELFCLASS64)
251 return rproc_elf64_load_image(dev, addr, size);
253 return rproc_elf32_load_image(dev, addr, size);
256 static ulong rproc_elf32_get_boot_addr(ulong addr)
258 Elf32_Ehdr *ehdr = (Elf32_Ehdr *)addr;
260 return ehdr->e_entry;
263 static ulong rproc_elf64_get_boot_addr(ulong addr)
265 Elf64_Ehdr *ehdr = (Elf64_Ehdr *)addr;
267 return ehdr->e_entry;
270 ulong rproc_elf_get_boot_addr(struct udevice *dev, ulong addr)
272 Elf32_Ehdr *ehdr = (Elf32_Ehdr *)addr;
274 if (ehdr->e_ident[EI_CLASS] == ELFCLASS64)
275 return rproc_elf64_get_boot_addr(addr);
277 return rproc_elf32_get_boot_addr(addr);