1 /* SPDX-License-Identifier: GPL-2.0 */ 2 /* 3 * Remote processor elf helpers defines 4 * 5 * Copyright (C) 2020 Kalray, Inc. 6 */ 7 8 #ifndef REMOTEPROC_ELF_LOADER_H 9 #define REMOTEPROC_ELF_LOADER_H 10 11 #include <linux/elf.h> 12 #include <linux/types.h> 13 14 /** 15 * fw_elf_get_class - Get elf class 16 * @fw: the ELF firmware image 17 * 18 * Note that we use and elf32_hdr to access the class since the start of the 19 * struct is the same for both elf class 20 * 21 * Return: elf class of the firmware 22 */ 23 static inline u8 fw_elf_get_class(const struct firmware *fw) 24 { 25 struct elf32_hdr *ehdr = (struct elf32_hdr *)fw->data; 26 27 return ehdr->e_ident[EI_CLASS]; 28 } 29 30 static inline void elf_hdr_init_ident(struct elf32_hdr *hdr, u8 class) 31 { 32 memcpy(hdr->e_ident, ELFMAG, SELFMAG); 33 hdr->e_ident[EI_CLASS] = class; 34 hdr->e_ident[EI_DATA] = ELFDATA2LSB; 35 hdr->e_ident[EI_VERSION] = EV_CURRENT; 36 hdr->e_ident[EI_OSABI] = ELFOSABI_NONE; 37 } 38 39 /* Generate getter and setter for a specific elf struct/field */ 40 #define ELF_GEN_FIELD_GET_SET(__s, __field, __type) \ 41 static inline __type elf_##__s##_get_##__field(u8 class, const void *arg) \ 42 { \ 43 if (class == ELFCLASS32) \ 44 return (__type) ((const struct elf32_##__s *) arg)->__field; \ 45 else \ 46 return (__type) ((const struct elf64_##__s *) arg)->__field; \ 47 } \ 48 static inline void elf_##__s##_set_##__field(u8 class, void *arg, \ 49 __type value) \ 50 { \ 51 if (class == ELFCLASS32) \ 52 ((struct elf32_##__s *) arg)->__field = (__type) value; \ 53 else \ 54 ((struct elf64_##__s *) arg)->__field = (__type) value; \ 55 } 56 57 ELF_GEN_FIELD_GET_SET(hdr, e_entry, u64) 58 ELF_GEN_FIELD_GET_SET(hdr, e_phnum, u16) 59 ELF_GEN_FIELD_GET_SET(hdr, e_shnum, u16) 60 ELF_GEN_FIELD_GET_SET(hdr, e_phoff, u64) 61 ELF_GEN_FIELD_GET_SET(hdr, e_shoff, u64) 62 ELF_GEN_FIELD_GET_SET(hdr, e_shstrndx, u16) 63 ELF_GEN_FIELD_GET_SET(hdr, e_machine, u16) 64 ELF_GEN_FIELD_GET_SET(hdr, e_type, u16) 65 ELF_GEN_FIELD_GET_SET(hdr, e_version, u32) 66 ELF_GEN_FIELD_GET_SET(hdr, e_ehsize, u32) 67 ELF_GEN_FIELD_GET_SET(hdr, e_phentsize, u16) 68 69 ELF_GEN_FIELD_GET_SET(phdr, p_paddr, u64) 70 ELF_GEN_FIELD_GET_SET(phdr, p_vaddr, u64) 71 ELF_GEN_FIELD_GET_SET(phdr, p_filesz, u64) 72 ELF_GEN_FIELD_GET_SET(phdr, p_memsz, u64) 73 ELF_GEN_FIELD_GET_SET(phdr, p_type, u32) 74 ELF_GEN_FIELD_GET_SET(phdr, p_offset, u64) 75 ELF_GEN_FIELD_GET_SET(phdr, p_flags, u32) 76 ELF_GEN_FIELD_GET_SET(phdr, p_align, u64) 77 78 ELF_GEN_FIELD_GET_SET(shdr, sh_size, u64) 79 ELF_GEN_FIELD_GET_SET(shdr, sh_offset, u64) 80 ELF_GEN_FIELD_GET_SET(shdr, sh_name, u32) 81 ELF_GEN_FIELD_GET_SET(shdr, sh_addr, u64) 82 83 #define ELF_STRUCT_SIZE(__s) \ 84 static inline unsigned long elf_size_of_##__s(u8 class) \ 85 { \ 86 if (class == ELFCLASS32)\ 87 return sizeof(struct elf32_##__s); \ 88 else \ 89 return sizeof(struct elf64_##__s); \ 90 } 91 92 ELF_STRUCT_SIZE(shdr) 93 ELF_STRUCT_SIZE(phdr) 94 ELF_STRUCT_SIZE(hdr) 95 96 #endif /* REMOTEPROC_ELF_LOADER_H */ 97