drivers/remoteproc/remoteproc_elf_helpers.h

   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 */