Blame view
scripts/sortextable.h
5.48 KB
a79f248b9 scripts: Add sort... |
1 2 3 |
/* * sortextable.h * |
d59a16836 scripts/sortextab... |
4 |
* Copyright 2011 - 2012 Cavium, Inc. |
a79f248b9 scripts: Add sort... |
5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 |
* * Some of this code was taken out of recordmcount.h written by: * * Copyright 2009 John F. Reiser <jreiser@BitWagon.com>. All rights reserved. * Copyright 2010 Steven Rostedt <srostedt@redhat.com>, Red Hat Inc. * * * Licensed under the GNU General Public License, version 2 (GPLv2). */ #undef extable_ent_size #undef compare_extable #undef do_func #undef Elf_Addr #undef Elf_Ehdr #undef Elf_Shdr #undef Elf_Rel #undef Elf_Rela #undef Elf_Sym #undef ELF_R_SYM #undef Elf_r_sym #undef ELF_R_INFO #undef Elf_r_info #undef ELF_ST_BIND #undef ELF_ST_TYPE #undef fn_ELF_R_SYM #undef fn_ELF_R_INFO #undef uint_t |
d59a16836 scripts/sortextab... |
33 |
#undef _r |
a79f248b9 scripts: Add sort... |
34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 |
#undef _w #ifdef SORTEXTABLE_64 # define extable_ent_size 16 # define compare_extable compare_extable_64 # define do_func do64 # define Elf_Addr Elf64_Addr # define Elf_Ehdr Elf64_Ehdr # define Elf_Shdr Elf64_Shdr # define Elf_Rel Elf64_Rel # define Elf_Rela Elf64_Rela # define Elf_Sym Elf64_Sym # define ELF_R_SYM ELF64_R_SYM # define Elf_r_sym Elf64_r_sym # define ELF_R_INFO ELF64_R_INFO # define Elf_r_info Elf64_r_info # define ELF_ST_BIND ELF64_ST_BIND # define ELF_ST_TYPE ELF64_ST_TYPE # define fn_ELF_R_SYM fn_ELF64_R_SYM # define fn_ELF_R_INFO fn_ELF64_R_INFO # define uint_t uint64_t |
d59a16836 scripts/sortextab... |
55 |
# define _r r8 |
a79f248b9 scripts: Add sort... |
56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 |
# define _w w8 #else # define extable_ent_size 8 # define compare_extable compare_extable_32 # define do_func do32 # define Elf_Addr Elf32_Addr # define Elf_Ehdr Elf32_Ehdr # define Elf_Shdr Elf32_Shdr # define Elf_Rel Elf32_Rel # define Elf_Rela Elf32_Rela # define Elf_Sym Elf32_Sym # define ELF_R_SYM ELF32_R_SYM # define Elf_r_sym Elf32_r_sym # define ELF_R_INFO ELF32_R_INFO # define Elf_r_info Elf32_r_info # define ELF_ST_BIND ELF32_ST_BIND # define ELF_ST_TYPE ELF32_ST_TYPE # define fn_ELF_R_SYM fn_ELF32_R_SYM # define fn_ELF_R_INFO fn_ELF32_R_INFO # define uint_t uint32_t |
d59a16836 scripts/sortextab... |
76 |
# define _r r |
a79f248b9 scripts: Add sort... |
77 78 79 80 81 |
# define _w w #endif static int compare_extable(const void *a, const void *b) { |
d59a16836 scripts/sortextab... |
82 83 |
Elf_Addr av = _r(a); Elf_Addr bv = _r(b); |
a79f248b9 scripts: Add sort... |
84 |
|
d59a16836 scripts/sortextab... |
85 |
if (av < bv) |
a79f248b9 scripts: Add sort... |
86 |
return -1; |
d59a16836 scripts/sortextab... |
87 |
if (av > bv) |
a79f248b9 scripts: Add sort... |
88 89 90 91 92 |
return 1; return 0; } static void |
d59a16836 scripts/sortextab... |
93 |
do_func(Elf_Ehdr *ehdr, char const *const fname, table_sort_t custom_sort) |
a79f248b9 scripts: Add sort... |
94 95 96 97 98 99 100 |
{ Elf_Shdr *shdr; Elf_Shdr *shstrtab_sec; Elf_Shdr *strtab_sec = NULL; Elf_Shdr *symtab_sec = NULL; Elf_Shdr *extab_sec = NULL; Elf_Sym *sym; |
59c36455d scripts/sortextab... |
101 102 |
const Elf_Sym *symtab; Elf32_Word *symtab_shndx_start = NULL; |
a79f248b9 scripts: Add sort... |
103 104 |
Elf_Sym *sort_needed_sym; Elf_Shdr *sort_needed_sec; |
d59a16836 scripts/sortextab... |
105 |
Elf_Rel *relocs = NULL; |
7cbc0ea79 scripts/sortextab... |
106 |
int relocs_size = 0; |
a79f248b9 scripts: Add sort... |
107 108 109 |
uint32_t *sort_done_location; const char *secstrtab; const char *strtab; |
d59a16836 scripts/sortextab... |
110 111 |
char *extab_image; int extab_index = 0; |
a79f248b9 scripts: Add sort... |
112 113 |
int i; int idx; |
59c36455d scripts/sortextab... |
114 115 |
unsigned int num_sections; unsigned int secindex_strings; |
a79f248b9 scripts: Add sort... |
116 |
|
d59a16836 scripts/sortextab... |
117 |
shdr = (Elf_Shdr *)((char *)ehdr + _r(&ehdr->e_shoff)); |
59c36455d scripts/sortextab... |
118 119 120 121 122 123 124 125 126 127 |
num_sections = r2(&ehdr->e_shnum); if (num_sections == SHN_UNDEF) num_sections = _r(&shdr[0].sh_size); secindex_strings = r2(&ehdr->e_shstrndx); if (secindex_strings == SHN_XINDEX) secindex_strings = r(&shdr[0].sh_link); shstrtab_sec = shdr + secindex_strings; |
d59a16836 scripts/sortextab... |
128 |
secstrtab = (const char *)ehdr + _r(&shstrtab_sec->sh_offset); |
59c36455d scripts/sortextab... |
129 |
for (i = 0; i < num_sections; i++) { |
d59a16836 scripts/sortextab... |
130 131 |
idx = r(&shdr[i].sh_name); if (strcmp(secstrtab + idx, "__ex_table") == 0) { |
a79f248b9 scripts: Add sort... |
132 |
extab_sec = shdr + i; |
d59a16836 scripts/sortextab... |
133 134 135 136 137 138 139 140 |
extab_index = i; } if ((r(&shdr[i].sh_type) == SHT_REL || r(&shdr[i].sh_type) == SHT_RELA) && r(&shdr[i].sh_info) == extab_index) { relocs = (void *)ehdr + _r(&shdr[i].sh_offset); relocs_size = _r(&shdr[i].sh_size); } |
a79f248b9 scripts: Add sort... |
141 142 143 144 |
if (strcmp(secstrtab + idx, ".symtab") == 0) symtab_sec = shdr + i; if (strcmp(secstrtab + idx, ".strtab") == 0) strtab_sec = shdr + i; |
59c36455d scripts/sortextab... |
145 146 147 |
if (r(&shdr[i].sh_type) == SHT_SYMTAB_SHNDX) symtab_shndx_start = (Elf32_Word *)( (const char *)ehdr + _r(&shdr[i].sh_offset)); |
a79f248b9 scripts: Add sort... |
148 149 150 151 152 153 154 155 156 157 158 |
} if (strtab_sec == NULL) { fprintf(stderr, "no .strtab in file: %s ", fname); fail_file(); } if (symtab_sec == NULL) { fprintf(stderr, "no .symtab in file: %s ", fname); fail_file(); } |
59c36455d scripts/sortextab... |
159 160 |
symtab = (const Elf_Sym *)((const char *)ehdr + _r(&symtab_sec->sh_offset)); |
a79f248b9 scripts: Add sort... |
161 162 163 164 165 |
if (extab_sec == NULL) { fprintf(stderr, "no __ex_table in file: %s ", fname); fail_file(); } |
d59a16836 scripts/sortextab... |
166 |
strtab = (const char *)ehdr + _r(&strtab_sec->sh_offset); |
a79f248b9 scripts: Add sort... |
167 |
|
d59a16836 scripts/sortextab... |
168 169 170 171 172 173 174 175 176 177 178 179 |
extab_image = (void *)ehdr + _r(&extab_sec->sh_offset); if (custom_sort) { custom_sort(extab_image, _r(&extab_sec->sh_size)); } else { int num_entries = _r(&extab_sec->sh_size) / extable_ent_size; qsort(extab_image, num_entries, extable_ent_size, compare_extable); } /* If there were relocations, we no longer need them. */ if (relocs) memset(relocs, 0, relocs_size); |
a79f248b9 scripts: Add sort... |
180 181 182 |
/* find main_extable_sort_needed */ sort_needed_sym = NULL; |
d59a16836 scripts/sortextab... |
183 184 |
for (i = 0; i < _r(&symtab_sec->sh_size) / sizeof(Elf_Sym); i++) { sym = (void *)ehdr + _r(&symtab_sec->sh_offset); |
a79f248b9 scripts: Add sort... |
185 186 187 |
sym += i; if (ELF_ST_TYPE(sym->st_info) != STT_OBJECT) continue; |
d59a16836 scripts/sortextab... |
188 |
idx = r(&sym->st_name); |
a79f248b9 scripts: Add sort... |
189 190 191 192 193 194 195 196 197 198 199 200 |
if (strcmp(strtab + idx, "main_extable_sort_needed") == 0) { sort_needed_sym = sym; break; } } if (sort_needed_sym == NULL) { fprintf(stderr, "no main_extable_sort_needed symbol in file: %s ", fname); fail_file(); } |
59c36455d scripts/sortextab... |
201 202 203 |
sort_needed_sec = &shdr[get_secindex(r2(&sym->st_shndx), sort_needed_sym - symtab, symtab_shndx_start)]; |
a79f248b9 scripts: Add sort... |
204 |
sort_done_location = (void *)ehdr + |
d59a16836 scripts/sortextab... |
205 206 207 |
_r(&sort_needed_sec->sh_offset) + _r(&sort_needed_sym->st_value) - _r(&sort_needed_sec->sh_addr); |
a79f248b9 scripts: Add sort... |
208 |
|
8c0c774dd scripts/sortextab... |
209 |
#if 0 |
a79f248b9 scripts: Add sort... |
210 211 |
printf("sort done marker at %lx ", |
d59a16836 scripts/sortextab... |
212 213 |
(unsigned long)((char *)sort_done_location - (char *)ehdr)); #endif |
a79f248b9 scripts: Add sort... |
214 |
/* We sorted it, clear the flag. */ |
d59a16836 scripts/sortextab... |
215 |
w(0, sort_done_location); |
a79f248b9 scripts: Add sort... |
216 |
} |