1 // SPDX-License-Identifier: GPL-2.0-or-later
3 * elf.c - ELF access library
5 * Adapted from kpatch (https://github.com/dynup/kpatch):
6 * Copyright (C) 2013-2015 Josh Poimboeuf <jpoimboe@redhat.com>
7 * Copyright (C) 2014 Seth Jennings <sjenning@redhat.com>
10 #include <sys/types.h>
19 #include <objtool/builtin.h>
21 #include <objtool/elf.h>
22 #include <objtool/warn.h>
24 #define MAX_NAME_LEN 128
26 static inline u32 str_hash(const char *str)
28 return jhash(str, strlen(str), 0);
31 #define __elf_table(name) (elf->name##_hash)
32 #define __elf_bits(name) (elf->name##_bits)
34 #define elf_hash_add(name, node, key) \
35 hlist_add_head(node, &__elf_table(name)[hash_min(key, __elf_bits(name))])
37 #define elf_hash_for_each_possible(name, obj, member, key) \
38 hlist_for_each_entry(obj, &__elf_table(name)[hash_min(key, __elf_bits(name))], member)
40 #define elf_alloc_hash(name, size) \
42 __elf_bits(name) = max(10, ilog2(size)); \
43 __elf_table(name) = mmap(NULL, sizeof(struct hlist_head) << __elf_bits(name), \
44 PROT_READ|PROT_WRITE, \
45 MAP_PRIVATE|MAP_ANON, -1, 0); \
46 if (__elf_table(name) == (void *)-1L) { \
47 WARN("mmap fail " #name); \
48 __elf_table(name) = NULL; \
53 static bool symbol_to_offset(struct rb_node *a, const struct rb_node *b)
55 struct symbol *sa = rb_entry(a, struct symbol, node);
56 struct symbol *sb = rb_entry(b, struct symbol, node);
58 if (sa->offset < sb->offset)
60 if (sa->offset > sb->offset)
63 if (sa->len < sb->len)
65 if (sa->len > sb->len)
73 static int symbol_by_offset(const void *key, const struct rb_node *node)
75 const struct symbol *s = rb_entry(node, struct symbol, node);
76 const unsigned long *o = key;
80 if (*o >= s->offset + s->len)
88 const struct symbol *sym;
92 * Find !section symbol where @offset is after it.
94 static int symbol_hole_by_offset(const void *key, const struct rb_node *node)
96 const struct symbol *s = rb_entry(node, struct symbol, node);
97 struct symbol_hole *sh = (void *)key;
99 if (sh->key < s->offset)
102 if (sh->key >= s->offset + s->len) {
103 if (s->type != STT_SECTION)
111 struct section *find_section_by_name(const struct elf *elf, const char *name)
115 elf_hash_for_each_possible(section_name, sec, name_hash, str_hash(name)) {
116 if (!strcmp(sec->name, name))
123 static struct section *find_section_by_index(struct elf *elf,
128 elf_hash_for_each_possible(section, sec, hash, idx) {
136 static struct symbol *find_symbol_by_index(struct elf *elf, unsigned int idx)
140 elf_hash_for_each_possible(symbol, sym, hash, idx) {
148 struct symbol *find_symbol_by_offset(struct section *sec, unsigned long offset)
150 struct rb_node *node;
152 rb_for_each(node, &offset, &sec->symbol_tree, symbol_by_offset) {
153 struct symbol *s = rb_entry(node, struct symbol, node);
155 if (s->offset == offset && s->type != STT_SECTION)
162 struct symbol *find_func_by_offset(struct section *sec, unsigned long offset)
164 struct rb_node *node;
166 rb_for_each(node, &offset, &sec->symbol_tree, symbol_by_offset) {
167 struct symbol *s = rb_entry(node, struct symbol, node);
169 if (s->offset == offset && s->type == STT_FUNC)
176 struct symbol *find_symbol_containing(const struct section *sec, unsigned long offset)
178 struct rb_node *node;
180 rb_for_each(node, &offset, &sec->symbol_tree, symbol_by_offset) {
181 struct symbol *s = rb_entry(node, struct symbol, node);
183 if (s->type != STT_SECTION)
191 * Returns size of hole starting at @offset.
193 int find_symbol_hole_containing(const struct section *sec, unsigned long offset)
195 struct symbol_hole hole = {
203 * Find the rightmost symbol for which @offset is after it.
205 n = rb_find(&hole, &sec->symbol_tree, symbol_hole_by_offset);
207 /* found a symbol that contains @offset */
209 return 0; /* not a hole */
211 /* didn't find a symbol for which @offset is after it */
213 return 0; /* not a hole */
215 /* @offset >= sym->offset + sym->len, find symbol after it */
216 n = rb_next(&hole.sym->node);
218 return -1; /* until end of address space */
220 /* hole until start of next symbol */
221 s = rb_entry(n, struct symbol, node);
222 return s->offset - offset;
225 struct symbol *find_func_containing(struct section *sec, unsigned long offset)
227 struct rb_node *node;
229 rb_for_each(node, &offset, &sec->symbol_tree, symbol_by_offset) {
230 struct symbol *s = rb_entry(node, struct symbol, node);
232 if (s->type == STT_FUNC)
239 struct symbol *find_symbol_by_name(const struct elf *elf, const char *name)
243 elf_hash_for_each_possible(symbol_name, sym, name_hash, str_hash(name)) {
244 if (!strcmp(sym->name, name))
251 struct reloc *find_reloc_by_dest_range(const struct elf *elf, struct section *sec,
252 unsigned long offset, unsigned int len)
254 struct reloc *reloc, *r = NULL;
262 for_offset_range(o, offset, offset + len) {
263 elf_hash_for_each_possible(reloc, reloc, hash,
264 sec_offset_hash(sec, o)) {
265 if (reloc->sec != sec)
268 if (reloc->offset >= offset && reloc->offset < offset + len) {
269 if (!r || reloc->offset < r->offset)
280 struct reloc *find_reloc_by_dest(const struct elf *elf, struct section *sec, unsigned long offset)
282 return find_reloc_by_dest_range(elf, sec, offset, 1);
285 static int read_sections(struct elf *elf)
289 size_t shstrndx, sections_nr;
292 if (elf_getshdrnum(elf->elf, §ions_nr)) {
293 WARN_ELF("elf_getshdrnum");
297 if (elf_getshdrstrndx(elf->elf, &shstrndx)) {
298 WARN_ELF("elf_getshdrstrndx");
302 if (!elf_alloc_hash(section, sections_nr) ||
303 !elf_alloc_hash(section_name, sections_nr))
306 for (i = 0; i < sections_nr; i++) {
307 sec = malloc(sizeof(*sec));
312 memset(sec, 0, sizeof(*sec));
314 INIT_LIST_HEAD(&sec->symbol_list);
315 INIT_LIST_HEAD(&sec->reloc_list);
317 s = elf_getscn(elf->elf, i);
319 WARN_ELF("elf_getscn");
323 sec->idx = elf_ndxscn(s);
325 if (!gelf_getshdr(s, &sec->sh)) {
326 WARN_ELF("gelf_getshdr");
330 sec->name = elf_strptr(elf->elf, shstrndx, sec->sh.sh_name);
332 WARN_ELF("elf_strptr");
336 if (sec->sh.sh_size != 0) {
337 sec->data = elf_getdata(s, NULL);
339 WARN_ELF("elf_getdata");
342 if (sec->data->d_off != 0 ||
343 sec->data->d_size != sec->sh.sh_size) {
344 WARN("unexpected data attributes for %s",
350 if (sec->sh.sh_flags & SHF_EXECINSTR)
351 elf->text_size += sec->sh.sh_size;
353 list_add_tail(&sec->list, &elf->sections);
354 elf_hash_add(section, &sec->hash, sec->idx);
355 elf_hash_add(section_name, &sec->name_hash, str_hash(sec->name));
359 printf("nr_sections: %lu\n", (unsigned long)sections_nr);
360 printf("section_bits: %d\n", elf->section_bits);
363 /* sanity check, one more call to elf_nextscn() should return NULL */
364 if (elf_nextscn(elf->elf, s)) {
365 WARN("section entry mismatch");
372 static void elf_add_symbol(struct elf *elf, struct symbol *sym)
374 struct list_head *entry;
375 struct rb_node *pnode;
377 sym->type = GELF_ST_TYPE(sym->sym.st_info);
378 sym->bind = GELF_ST_BIND(sym->sym.st_info);
380 sym->offset = sym->sym.st_value;
381 sym->len = sym->sym.st_size;
383 rb_add(&sym->node, &sym->sec->symbol_tree, symbol_to_offset);
384 pnode = rb_prev(&sym->node);
386 entry = &rb_entry(pnode, struct symbol, node)->list;
388 entry = &sym->sec->symbol_list;
389 list_add(&sym->list, entry);
390 elf_hash_add(symbol, &sym->hash, sym->idx);
391 elf_hash_add(symbol_name, &sym->name_hash, str_hash(sym->name));
394 * Don't store empty STT_NOTYPE symbols in the rbtree. They
395 * can exist within a function, confusing the sorting.
398 rb_erase(&sym->node, &sym->sec->symbol_tree);
401 static int read_symbols(struct elf *elf)
403 struct section *symtab, *symtab_shndx, *sec;
404 struct symbol *sym, *pfunc;
407 Elf_Data *shndx_data = NULL;
410 symtab = find_section_by_name(elf, ".symtab");
412 symtab_shndx = find_section_by_name(elf, ".symtab_shndx");
414 shndx_data = symtab_shndx->data;
416 symbols_nr = symtab->sh.sh_size / symtab->sh.sh_entsize;
419 * A missing symbol table is actually possible if it's an empty
420 * .o file. This can happen for thunk_64.o. Make sure to at
421 * least allocate the symbol hash tables so we can do symbol
422 * lookups without crashing.
427 if (!elf_alloc_hash(symbol, symbols_nr) ||
428 !elf_alloc_hash(symbol_name, symbols_nr))
431 for (i = 0; i < symbols_nr; i++) {
432 sym = malloc(sizeof(*sym));
437 memset(sym, 0, sizeof(*sym));
438 INIT_LIST_HEAD(&sym->pv_target);
443 if (!gelf_getsymshndx(symtab->data, shndx_data, i, &sym->sym,
445 WARN_ELF("gelf_getsymshndx");
449 sym->name = elf_strptr(elf->elf, symtab->sh.sh_link,
452 WARN_ELF("elf_strptr");
456 if ((sym->sym.st_shndx > SHN_UNDEF &&
457 sym->sym.st_shndx < SHN_LORESERVE) ||
458 (shndx_data && sym->sym.st_shndx == SHN_XINDEX)) {
459 if (sym->sym.st_shndx != SHN_XINDEX)
460 shndx = sym->sym.st_shndx;
462 sym->sec = find_section_by_index(elf, shndx);
464 WARN("couldn't find section for symbol %s",
468 if (GELF_ST_TYPE(sym->sym.st_info) == STT_SECTION) {
469 sym->name = sym->sec->name;
473 sym->sec = find_section_by_index(elf, 0);
475 elf_add_symbol(elf, sym);
479 printf("nr_symbols: %lu\n", (unsigned long)symbols_nr);
480 printf("symbol_bits: %d\n", elf->symbol_bits);
483 /* Create parent/child links for any cold subfunctions */
484 list_for_each_entry(sec, &elf->sections, list) {
485 list_for_each_entry(sym, &sec->symbol_list, list) {
486 char pname[MAX_NAME_LEN + 1];
488 if (sym->type != STT_FUNC)
491 if (sym->pfunc == NULL)
494 if (sym->cfunc == NULL)
497 coldstr = strstr(sym->name, ".cold");
501 pnamelen = coldstr - sym->name;
502 if (pnamelen > MAX_NAME_LEN) {
503 WARN("%s(): parent function name exceeds maximum length of %d characters",
504 sym->name, MAX_NAME_LEN);
508 strncpy(pname, sym->name, pnamelen);
509 pname[pnamelen] = '\0';
510 pfunc = find_symbol_by_name(elf, pname);
513 WARN("%s(): can't find parent function",
522 * Unfortunately, -fnoreorder-functions puts the child
523 * inside the parent. Remove the overlap so we can
524 * have sane assumptions.
526 * Note that pfunc->len now no longer matches
527 * pfunc->sym.st_size.
529 if (sym->sec == pfunc->sec &&
530 sym->offset >= pfunc->offset &&
531 sym->offset + sym->len == pfunc->offset + pfunc->len) {
532 pfunc->len -= sym->len;
544 static struct section *elf_create_reloc_section(struct elf *elf,
545 struct section *base,
548 int elf_add_reloc(struct elf *elf, struct section *sec, unsigned long offset,
549 unsigned int type, struct symbol *sym, long addend)
553 if (!sec->reloc && !elf_create_reloc_section(elf, sec, SHT_RELA))
556 reloc = malloc(sizeof(*reloc));
561 memset(reloc, 0, sizeof(*reloc));
563 reloc->sec = sec->reloc;
564 reloc->offset = offset;
567 reloc->addend = addend;
569 list_add_tail(&reloc->list, &sec->reloc->reloc_list);
570 elf_hash_add(reloc, &reloc->hash, reloc_hash(reloc));
572 sec->reloc->sh.sh_size += sec->reloc->sh.sh_entsize;
573 sec->reloc->changed = true;
579 * Ensure that any reloc section containing references to @sym is marked
580 * changed such that it will get re-generated in elf_rebuild_reloc_sections()
581 * with the new symbol index.
583 static void elf_dirty_reloc_sym(struct elf *elf, struct symbol *sym)
587 list_for_each_entry(sec, &elf->sections, list) {
593 list_for_each_entry(reloc, &sec->reloc_list, list) {
594 if (reloc->sym == sym) {
603 * Move the first global symbol, as per sh_info, into a new, higher symbol
604 * index. This fees up the shndx for a new local symbol.
606 static int elf_move_global_symbol(struct elf *elf, struct section *symtab,
607 struct section *symtab_shndx)
609 Elf_Data *data, *shndx_data = NULL;
610 Elf32_Word first_non_local;
614 first_non_local = symtab->sh.sh_info;
616 sym = find_symbol_by_index(elf, first_non_local);
618 WARN("no non-local symbols !?");
619 return first_non_local;
622 s = elf_getscn(elf->elf, symtab->idx);
624 WARN_ELF("elf_getscn");
628 data = elf_newdata(s);
630 WARN_ELF("elf_newdata");
634 data->d_buf = &sym->sym;
635 data->d_size = sizeof(sym->sym);
637 data->d_type = ELF_T_SYM;
639 sym->idx = symtab->sh.sh_size / sizeof(sym->sym);
640 elf_dirty_reloc_sym(elf, sym);
642 symtab->sh.sh_info += 1;
643 symtab->sh.sh_size += data->d_size;
644 symtab->changed = true;
647 s = elf_getscn(elf->elf, symtab_shndx->idx);
649 WARN_ELF("elf_getscn");
653 shndx_data = elf_newdata(s);
655 WARN_ELF("elf_newshndx_data");
659 shndx_data->d_buf = &sym->sec->idx;
660 shndx_data->d_size = sizeof(Elf32_Word);
661 shndx_data->d_align = 4;
662 shndx_data->d_type = ELF_T_WORD;
664 symtab_shndx->sh.sh_size += 4;
665 symtab_shndx->changed = true;
668 return first_non_local;
671 static struct symbol *
672 elf_create_section_symbol(struct elf *elf, struct section *sec)
674 struct section *symtab, *symtab_shndx;
675 Elf_Data *shndx_data = NULL;
679 symtab = find_section_by_name(elf, ".symtab");
681 symtab_shndx = find_section_by_name(elf, ".symtab_shndx");
683 shndx_data = symtab_shndx->data;
689 sym = malloc(sizeof(*sym));
694 memset(sym, 0, sizeof(*sym));
696 sym->idx = elf_move_global_symbol(elf, symtab, symtab_shndx);
698 WARN("elf_move_global_symbol");
702 sym->name = sec->name;
706 sym->sym.st_info = GELF_ST_INFO(STB_LOCAL, STT_SECTION);
711 if (shndx >= SHN_UNDEF && shndx < SHN_LORESERVE) {
712 sym->sym.st_shndx = shndx;
716 sym->sym.st_shndx = SHN_XINDEX;
718 WARN("no .symtab_shndx");
723 if (!gelf_update_symshndx(symtab->data, shndx_data, sym->idx, &sym->sym, shndx)) {
724 WARN_ELF("gelf_update_symshndx");
728 elf_add_symbol(elf, sym);
733 int elf_add_reloc_to_insn(struct elf *elf, struct section *sec,
734 unsigned long offset, unsigned int type,
735 struct section *insn_sec, unsigned long insn_off)
737 struct symbol *sym = insn_sec->sym;
738 int addend = insn_off;
742 * Due to how weak functions work, we must use section based
743 * relocations. Symbol based relocations would result in the
744 * weak and non-weak function annotations being overlaid on the
745 * non-weak function after linking.
747 sym = elf_create_section_symbol(elf, insn_sec);
754 return elf_add_reloc(elf, sec, offset, type, sym, addend);
757 static int read_rel_reloc(struct section *sec, int i, struct reloc *reloc, unsigned int *symndx)
759 if (!gelf_getrel(sec->data, i, &reloc->rel)) {
760 WARN_ELF("gelf_getrel");
763 reloc->type = GELF_R_TYPE(reloc->rel.r_info);
765 reloc->offset = reloc->rel.r_offset;
766 *symndx = GELF_R_SYM(reloc->rel.r_info);
770 static int read_rela_reloc(struct section *sec, int i, struct reloc *reloc, unsigned int *symndx)
772 if (!gelf_getrela(sec->data, i, &reloc->rela)) {
773 WARN_ELF("gelf_getrela");
776 reloc->type = GELF_R_TYPE(reloc->rela.r_info);
777 reloc->addend = reloc->rela.r_addend;
778 reloc->offset = reloc->rela.r_offset;
779 *symndx = GELF_R_SYM(reloc->rela.r_info);
783 static int read_relocs(struct elf *elf)
789 unsigned long nr_reloc, max_reloc = 0, tot_reloc = 0;
791 if (!elf_alloc_hash(reloc, elf->text_size / 16))
794 list_for_each_entry(sec, &elf->sections, list) {
795 if ((sec->sh.sh_type != SHT_RELA) &&
796 (sec->sh.sh_type != SHT_REL))
799 sec->base = find_section_by_index(elf, sec->sh.sh_info);
801 WARN("can't find base section for reloc section %s",
806 sec->base->reloc = sec;
809 for (i = 0; i < sec->sh.sh_size / sec->sh.sh_entsize; i++) {
810 reloc = malloc(sizeof(*reloc));
815 memset(reloc, 0, sizeof(*reloc));
816 switch (sec->sh.sh_type) {
818 if (read_rel_reloc(sec, i, reloc, &symndx))
822 if (read_rela_reloc(sec, i, reloc, &symndx))
830 reloc->sym = find_symbol_by_index(elf, symndx);
832 WARN("can't find reloc entry symbol %d for %s",
837 list_add_tail(&reloc->list, &sec->reloc_list);
838 elf_hash_add(reloc, &reloc->hash, reloc_hash(reloc));
842 max_reloc = max(max_reloc, nr_reloc);
843 tot_reloc += nr_reloc;
847 printf("max_reloc: %lu\n", max_reloc);
848 printf("tot_reloc: %lu\n", tot_reloc);
849 printf("reloc_bits: %d\n", elf->reloc_bits);
855 struct elf *elf_open_read(const char *name, int flags)
860 elf_version(EV_CURRENT);
862 elf = malloc(sizeof(*elf));
867 memset(elf, 0, offsetof(struct elf, sections));
869 INIT_LIST_HEAD(&elf->sections);
871 elf->fd = open(name, flags);
873 fprintf(stderr, "objtool: Can't open '%s': %s\n",
874 name, strerror(errno));
878 if ((flags & O_ACCMODE) == O_RDONLY)
879 cmd = ELF_C_READ_MMAP;
880 else if ((flags & O_ACCMODE) == O_RDWR)
885 elf->elf = elf_begin(elf->fd, cmd, NULL);
887 WARN_ELF("elf_begin");
891 if (!gelf_getehdr(elf->elf, &elf->ehdr)) {
892 WARN_ELF("gelf_getehdr");
896 if (read_sections(elf))
899 if (read_symbols(elf))
902 if (read_relocs(elf))
912 static int elf_add_string(struct elf *elf, struct section *strtab, char *str)
919 strtab = find_section_by_name(elf, ".strtab");
921 WARN("can't find .strtab section");
925 s = elf_getscn(elf->elf, strtab->idx);
927 WARN_ELF("elf_getscn");
931 data = elf_newdata(s);
933 WARN_ELF("elf_newdata");
938 data->d_size = strlen(str) + 1;
941 len = strtab->sh.sh_size;
942 strtab->sh.sh_size += data->d_size;
943 strtab->changed = true;
948 struct section *elf_create_section(struct elf *elf, const char *name,
949 unsigned int sh_flags, size_t entsize, int nr)
951 struct section *sec, *shstrtab;
952 size_t size = entsize * nr;
955 sec = malloc(sizeof(*sec));
960 memset(sec, 0, sizeof(*sec));
962 INIT_LIST_HEAD(&sec->symbol_list);
963 INIT_LIST_HEAD(&sec->reloc_list);
965 s = elf_newscn(elf->elf);
967 WARN_ELF("elf_newscn");
971 sec->name = strdup(name);
977 sec->idx = elf_ndxscn(s);
980 sec->data = elf_newdata(s);
982 WARN_ELF("elf_newdata");
986 sec->data->d_size = size;
987 sec->data->d_align = 1;
990 sec->data->d_buf = malloc(size);
991 if (!sec->data->d_buf) {
995 memset(sec->data->d_buf, 0, size);
998 if (!gelf_getshdr(s, &sec->sh)) {
999 WARN_ELF("gelf_getshdr");
1003 sec->sh.sh_size = size;
1004 sec->sh.sh_entsize = entsize;
1005 sec->sh.sh_type = SHT_PROGBITS;
1006 sec->sh.sh_addralign = 1;
1007 sec->sh.sh_flags = SHF_ALLOC | sh_flags;
1009 /* Add section name to .shstrtab (or .strtab for Clang) */
1010 shstrtab = find_section_by_name(elf, ".shstrtab");
1012 shstrtab = find_section_by_name(elf, ".strtab");
1014 WARN("can't find .shstrtab or .strtab section");
1017 sec->sh.sh_name = elf_add_string(elf, shstrtab, sec->name);
1018 if (sec->sh.sh_name == -1)
1021 list_add_tail(&sec->list, &elf->sections);
1022 elf_hash_add(section, &sec->hash, sec->idx);
1023 elf_hash_add(section_name, &sec->name_hash, str_hash(sec->name));
1025 elf->changed = true;
1030 static struct section *elf_create_rel_reloc_section(struct elf *elf, struct section *base)
1033 struct section *sec;
1035 relocname = malloc(strlen(base->name) + strlen(".rel") + 1);
1040 strcpy(relocname, ".rel");
1041 strcat(relocname, base->name);
1043 sec = elf_create_section(elf, relocname, 0, sizeof(GElf_Rel), 0);
1051 sec->sh.sh_type = SHT_REL;
1052 sec->sh.sh_addralign = 8;
1053 sec->sh.sh_link = find_section_by_name(elf, ".symtab")->idx;
1054 sec->sh.sh_info = base->idx;
1055 sec->sh.sh_flags = SHF_INFO_LINK;
1060 static struct section *elf_create_rela_reloc_section(struct elf *elf, struct section *base)
1063 struct section *sec;
1065 relocname = malloc(strlen(base->name) + strlen(".rela") + 1);
1070 strcpy(relocname, ".rela");
1071 strcat(relocname, base->name);
1073 sec = elf_create_section(elf, relocname, 0, sizeof(GElf_Rela), 0);
1081 sec->sh.sh_type = SHT_RELA;
1082 sec->sh.sh_addralign = 8;
1083 sec->sh.sh_link = find_section_by_name(elf, ".symtab")->idx;
1084 sec->sh.sh_info = base->idx;
1085 sec->sh.sh_flags = SHF_INFO_LINK;
1090 static struct section *elf_create_reloc_section(struct elf *elf,
1091 struct section *base,
1095 case SHT_REL: return elf_create_rel_reloc_section(elf, base);
1096 case SHT_RELA: return elf_create_rela_reloc_section(elf, base);
1097 default: return NULL;
1101 static int elf_rebuild_rel_reloc_section(struct section *sec)
1103 struct reloc *reloc;
1107 /* Allocate a buffer for relocations */
1108 buf = malloc(sec->sh.sh_size);
1114 sec->data->d_buf = buf;
1115 sec->data->d_size = sec->sh.sh_size;
1116 sec->data->d_type = ELF_T_REL;
1119 list_for_each_entry(reloc, &sec->reloc_list, list) {
1120 reloc->rel.r_offset = reloc->offset;
1121 reloc->rel.r_info = GELF_R_INFO(reloc->sym->idx, reloc->type);
1122 if (!gelf_update_rel(sec->data, idx, &reloc->rel)) {
1123 WARN_ELF("gelf_update_rel");
1132 static int elf_rebuild_rela_reloc_section(struct section *sec)
1134 struct reloc *reloc;
1138 /* Allocate a buffer for relocations with addends */
1139 buf = malloc(sec->sh.sh_size);
1145 sec->data->d_buf = buf;
1146 sec->data->d_size = sec->sh.sh_size;
1147 sec->data->d_type = ELF_T_RELA;
1150 list_for_each_entry(reloc, &sec->reloc_list, list) {
1151 reloc->rela.r_offset = reloc->offset;
1152 reloc->rela.r_addend = reloc->addend;
1153 reloc->rela.r_info = GELF_R_INFO(reloc->sym->idx, reloc->type);
1154 if (!gelf_update_rela(sec->data, idx, &reloc->rela)) {
1155 WARN_ELF("gelf_update_rela");
1164 static int elf_rebuild_reloc_section(struct elf *elf, struct section *sec)
1166 switch (sec->sh.sh_type) {
1167 case SHT_REL: return elf_rebuild_rel_reloc_section(sec);
1168 case SHT_RELA: return elf_rebuild_rela_reloc_section(sec);
1173 int elf_write_insn(struct elf *elf, struct section *sec,
1174 unsigned long offset, unsigned int len,
1177 Elf_Data *data = sec->data;
1179 if (data->d_type != ELF_T_BYTE || data->d_off) {
1180 WARN("write to unexpected data for section: %s", sec->name);
1184 memcpy(data->d_buf + offset, insn, len);
1185 elf_flagdata(data, ELF_C_SET, ELF_F_DIRTY);
1187 elf->changed = true;
1192 int elf_write_reloc(struct elf *elf, struct reloc *reloc)
1194 struct section *sec = reloc->sec;
1196 if (sec->sh.sh_type == SHT_REL) {
1197 reloc->rel.r_info = GELF_R_INFO(reloc->sym->idx, reloc->type);
1198 reloc->rel.r_offset = reloc->offset;
1200 if (!gelf_update_rel(sec->data, reloc->idx, &reloc->rel)) {
1201 WARN_ELF("gelf_update_rel");
1205 reloc->rela.r_info = GELF_R_INFO(reloc->sym->idx, reloc->type);
1206 reloc->rela.r_addend = reloc->addend;
1207 reloc->rela.r_offset = reloc->offset;
1209 if (!gelf_update_rela(sec->data, reloc->idx, &reloc->rela)) {
1210 WARN_ELF("gelf_update_rela");
1215 elf->changed = true;
1220 int elf_write(struct elf *elf)
1222 struct section *sec;
1228 /* Update changed relocation sections and section headers: */
1229 list_for_each_entry(sec, &elf->sections, list) {
1231 s = elf_getscn(elf->elf, sec->idx);
1233 WARN_ELF("elf_getscn");
1236 if (!gelf_update_shdr(s, &sec->sh)) {
1237 WARN_ELF("gelf_update_shdr");
1242 elf_rebuild_reloc_section(elf, sec)) {
1243 WARN("elf_rebuild_reloc_section");
1247 sec->changed = false;
1248 elf->changed = true;
1252 /* Make sure the new section header entries get updated properly. */
1253 elf_flagelf(elf->elf, ELF_C_SET, ELF_F_DIRTY);
1255 /* Write all changes to the file. */
1256 if (elf_update(elf->elf, ELF_C_WRITE) < 0) {
1257 WARN_ELF("elf_update");
1261 elf->changed = false;
1266 void elf_close(struct elf *elf)
1268 struct section *sec, *tmpsec;
1269 struct symbol *sym, *tmpsym;
1270 struct reloc *reloc, *tmpreloc;
1278 list_for_each_entry_safe(sec, tmpsec, &elf->sections, list) {
1279 list_for_each_entry_safe(sym, tmpsym, &sec->symbol_list, list) {
1280 list_del(&sym->list);
1281 hash_del(&sym->hash);
1284 list_for_each_entry_safe(reloc, tmpreloc, &sec->reloc_list, list) {
1285 list_del(&reloc->list);
1286 hash_del(&reloc->hash);
1289 list_del(&sec->list);