include/lttng/ust-elf.h

   1 #ifndef _LTTNG_UST_ELF_H
   2 #define _LTTNG_UST_ELF_H
   3 /*
   4  * Copyright (C) 2015  Antoine Busque <abusque@efficios.com>
   5  *
   6  * This library is free software; you can redistribute it and/or
   7  * modify it under the terms of the GNU Lesser General Public
   8  * License as published by the Free Software Foundation; either
   9  * version 2.1 of the License, or (at your option) any later version.
  10  *
  11  * This library is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14  * Lesser General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU Lesser General Public
  17  * License along with this library; if not, write to the Free Software
  18  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301 USA
  19  */
  20
  21 #include <stdlib.h>
  22 #include <stdint.h>
  23 #include <stdio.h>
  24 #include <elf.h>
  25 #include <lttng/ust-endian.h>
  26
  27 /*
  28  * Determine native endianness in order to convert when reading an ELF
  29  * file if there is a mismatch.
  30  */
  31 #if BYTE_ORDER == LITTLE_ENDIAN
  32 #define NATIVE_ELF_ENDIANNESS ELFDATA2LSB
  33 #else
  34 #define NATIVE_ELF_ENDIANNESS ELFDATA2MSB
  35 #endif
  36
  37 /*
  38  * The size in bytes of the debug link CRC as contained in an ELF
  39  * section.
  40  */
  41 #define ELF_CRC_SIZE            4
  42 /*
  43  * ELF notes are aligned on 4 bytes. ref: ELF specification version
  44  * 1.1 p. 2-5.
  45  */
  46 #define ELF_NOTE_ENTRY_ALIGN    4
  47 /*
  48  * Within an ELF note, the `desc` field is also aligned on 4
  49  * bytes. ref: ELF specification version 1.1 p. 2-5.
  50  */
  51 #define ELF_NOTE_DESC_ALIGN     4
  52
  53 #define bswap(x)                                \
  54         do {                                    \
  55                 switch (sizeof(x)) {            \
  56                 case 8:                         \
  57                         x = bswap_64(x);        \
  58                         break;                  \
  59                 case 4:                         \
  60                         x = bswap_32(x);        \
  61                         break;                  \
  62                 case 2:                         \
  63                         x = bswap_16(x);        \
  64                         break;                  \
  65                 case 1:                         \
  66                         break;                  \
  67                 default:                        \
  68                         abort();                \
  69                 }                               \
  70         } while (0)
  71
  72 #define bswap_phdr(phdr)                \
  73         do {                            \
  74                 bswap((phdr).p_type);   \
  75                 bswap((phdr).p_offset); \
  76                 bswap((phdr).p_filesz); \
  77                 bswap((phdr).p_memsz);  \
  78                 bswap((phdr).p_align);  \
  79                 bswap((phdr).p_vaddr);  \
  80         } while (0)
  81
  82 #define bswap_shdr(shdr)                    \
  83         do {                                \
  84                 bswap((shdr).sh_name);      \
  85                 bswap((shdr).sh_type);      \
  86                 bswap((shdr).sh_flags);     \
  87                 bswap((shdr).sh_addr);      \
  88                 bswap((shdr).sh_offset);    \
  89                 bswap((shdr).sh_size);      \
  90                 bswap((shdr).sh_link);      \
  91                 bswap((shdr).sh_info);      \
  92                 bswap((shdr).sh_addralign); \
  93                 bswap((shdr).sh_entsize);   \
  94         } while (0)
  95
  96 #define bswap_ehdr(ehdr)                                \
  97         do {                                            \
  98                 bswap((ehdr).e_type);                   \
  99                 bswap((ehdr).e_machine);                \
 100                 bswap((ehdr).e_version);                \
 101                 bswap((ehdr).e_entry);                  \
 102                 bswap((ehdr).e_phoff);                  \
 103                 bswap((ehdr).e_shoff);                  \
 104                 bswap((ehdr).e_flags);                  \
 105                 bswap((ehdr).e_ehsize);                 \
 106                 bswap((ehdr).e_phentsize);              \
 107                 bswap((ehdr).e_phnum);                  \
 108                 bswap((ehdr).e_shentsize);              \
 109                 bswap((ehdr).e_shnum);                  \
 110                 bswap((ehdr).e_shstrndx);               \
 111         } while (0)
 112
 113 #define copy_phdr(src_phdr, dst_phdr)                           \
 114         do {                                                    \
 115                 (dst_phdr).p_type = (src_phdr).p_type;          \
 116                 (dst_phdr).p_offset = (src_phdr).p_offset;      \
 117                 (dst_phdr).p_filesz = (src_phdr).p_filesz;      \
 118                 (dst_phdr).p_memsz = (src_phdr).p_memsz;        \
 119                 (dst_phdr).p_align = (src_phdr).p_align;        \
 120                 (dst_phdr).p_vaddr = (src_phdr).p_vaddr;        \
 121         } while (0)
 122
 123 #define copy_shdr(src_shdr, dst_shdr)                                   \
 124         do {                                                            \
 125                 (dst_shdr).sh_name = (src_shdr).sh_name;                \
 126                 (dst_shdr).sh_type = (src_shdr).sh_type;                \
 127                 (dst_shdr).sh_flags = (src_shdr).sh_flags;              \
 128                 (dst_shdr).sh_addr = (src_shdr).sh_addr;                \
 129                 (dst_shdr).sh_offset = (src_shdr).sh_offset;            \
 130                 (dst_shdr).sh_size = (src_shdr).sh_size;                \
 131                 (dst_shdr).sh_link = (src_shdr).sh_link;                \
 132                 (dst_shdr).sh_info = (src_shdr).sh_info;                \
 133                 (dst_shdr).sh_addralign = (src_shdr).sh_addralign;      \
 134                 (dst_shdr).sh_entsize = (src_shdr).sh_entsize;          \
 135         } while (0)
 136
 137 #define copy_ehdr(src_ehdr, dst_ehdr)                                   \
 138         do {                                                            \
 139                 (dst_ehdr).e_type = (src_ehdr).e_type;                  \
 140                 (dst_ehdr).e_machine = (src_ehdr).e_machine;            \
 141                 (dst_ehdr).e_version = (src_ehdr).e_version;            \
 142                 (dst_ehdr).e_entry = (src_ehdr).e_entry;                \
 143                 (dst_ehdr).e_phoff = (src_ehdr).e_phoff;                \
 144                 (dst_ehdr).e_shoff = (src_ehdr).e_shoff;                \
 145                 (dst_ehdr).e_flags = (src_ehdr).e_flags;                \
 146                 (dst_ehdr).e_ehsize = (src_ehdr).e_ehsize;              \
 147                 (dst_ehdr).e_phentsize = (src_ehdr).e_phentsize;        \
 148                 (dst_ehdr).e_phnum = (src_ehdr).e_phnum;                \
 149                 (dst_ehdr).e_shentsize = (src_ehdr).e_shentsize;        \
 150                 (dst_ehdr).e_shnum = (src_ehdr).e_shnum;                \
 151                 (dst_ehdr).e_shstrndx = (src_ehdr).e_shstrndx;          \
 152         } while (0)
 153
 154 struct lttng_ust_elf_ehdr {
 155         uint16_t e_type;
 156         uint16_t e_machine;
 157         uint32_t e_version;
 158         uint64_t e_entry;
 159         uint64_t e_phoff;
 160         uint64_t e_shoff;
 161         uint32_t e_flags;
 162         uint16_t e_ehsize;
 163         uint16_t e_phentsize;
 164         uint16_t e_phnum;
 165         uint16_t e_shentsize;
 166         uint16_t e_shnum;
 167         uint16_t e_shstrndx;
 168 };
 169
 170 struct lttng_ust_elf_phdr {
 171         uint32_t p_type;
 172         uint64_t p_offset;
 173         uint64_t p_filesz;
 174         uint64_t p_memsz;
 175         uint64_t p_align;
 176         uint64_t p_vaddr;
 177 };
 178
 179 struct lttng_ust_elf_shdr {
 180         uint32_t sh_name;
 181         uint32_t sh_type;
 182         uint64_t sh_flags;
 183         uint64_t sh_addr;
 184         uint64_t sh_offset;
 185         uint64_t sh_size;
 186         uint32_t sh_link;
 187         uint32_t sh_info;
 188         uint64_t sh_addralign;
 189         uint64_t sh_entsize;
 190 };
 191
 192 struct lttng_ust_elf_nhdr {
 193         uint32_t n_namesz;
 194         uint32_t n_descsz;
 195         uint32_t n_type;
 196 };
 197
 198 struct lttng_ust_elf {
 199         /* Offset in bytes to start of section names string table. */
 200         off_t section_names_offset;
 201         /* Size in bytes of section names string table. */
 202         size_t section_names_size;
 203         char *path;
 204         int fd;
 205         struct lttng_ust_elf_ehdr *ehdr;
 206         uint8_t bitness;
 207         uint8_t endianness;
 208 };
 209
 210 static inline
 211 int is_elf_32_bit(struct lttng_ust_elf *elf)
 212 {
 213         return elf->bitness == ELFCLASS32;
 214 }
 215
 216 static inline
 217 int is_elf_native_endian(struct lttng_ust_elf *elf)
 218 {
 219         return elf->endianness == NATIVE_ELF_ENDIANNESS;
 220 }
 221
 222 struct lttng_ust_elf *lttng_ust_elf_create(const char *path);
 223 void lttng_ust_elf_destroy(struct lttng_ust_elf *elf);
 224 uint8_t lttng_ust_elf_is_pic(struct lttng_ust_elf *elf);
 225 int lttng_ust_elf_get_memsz(struct lttng_ust_elf *elf, uint64_t *memsz);
 226 int lttng_ust_elf_get_build_id(struct lttng_ust_elf *elf, uint8_t **build_id,
 227                         size_t *length, int *found);
 228 int lttng_ust_elf_get_debug_link(struct lttng_ust_elf *elf, char **filename,
 229                         uint32_t *crc, int *found);
 230
 231 #endif  /* _LTTNG_UST_ELF_H */