Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 1 | /* Generate assembler source containing symbol information |
| 2 | * |
| 3 | * Copyright 2002 by Kai Germaschewski |
| 4 | * |
| 5 | * This software may be used and distributed according to the terms |
| 6 | * of the GNU General Public License, incorporated herein by reference. |
| 7 | * |
| 8 | * Usage: nm -n vmlinux | scripts/kallsyms [--all-symbols] > symbols.S |
| 9 | * |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 10 | * Table compression uses all the unused char codes on the symbols and |
| 11 | * maps these to the most used substrings (tokens). For instance, it might |
| 12 | * map char code 0xF7 to represent "write_" and then in every symbol where |
| 13 | * "write_" appears it can be replaced by 0xF7, saving 5 bytes. |
| 14 | * The used codes themselves are also placed in the table so that the |
| 15 | * decompresion can work without "special cases". |
| 16 | * Applied to kernel symbols, this usually produces a compression ratio |
| 17 | * of about 50%. |
| 18 | * |
| 19 | */ |
| 20 | |
| 21 | #include <stdio.h> |
| 22 | #include <stdlib.h> |
| 23 | #include <string.h> |
| 24 | #include <ctype.h> |
Ard Biesheuvel | 2213e9a | 2016-03-15 14:58:19 -0700 | [diff] [blame] | 25 | #include <limits.h> |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 26 | |
Mike Frysinger | 17b1f0d | 2009-06-08 19:12:13 -0400 | [diff] [blame] | 27 | #ifndef ARRAY_SIZE |
| 28 | #define ARRAY_SIZE(arr) (sizeof(arr) / sizeof(arr[0])) |
| 29 | #endif |
| 30 | |
Tejun Heo | 9281ace | 2007-07-17 04:03:51 -0700 | [diff] [blame] | 31 | #define KSYM_NAME_LEN 128 |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 32 | |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 33 | struct sym_entry { |
| 34 | unsigned long long addr; |
Paulo Marques | b3dbb4e | 2005-09-06 15:16:31 -0700 | [diff] [blame] | 35 | unsigned int len; |
Paulo Marques | f2df3f6 | 2008-02-06 01:37:33 -0800 | [diff] [blame] | 36 | unsigned int start_pos; |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 37 | unsigned char *sym; |
Ard Biesheuvel | 8c99694 | 2016-03-15 14:58:15 -0700 | [diff] [blame] | 38 | unsigned int percpu_absolute; |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 39 | }; |
| 40 | |
Kees Cook | 78eb715 | 2014-03-17 13:18:27 +1030 | [diff] [blame] | 41 | struct addr_range { |
| 42 | const char *start_sym, *end_sym; |
Mike Frysinger | 17b1f0d | 2009-06-08 19:12:13 -0400 | [diff] [blame] | 43 | unsigned long long start, end; |
| 44 | }; |
| 45 | |
| 46 | static unsigned long long _text; |
Ard Biesheuvel | 2213e9a | 2016-03-15 14:58:19 -0700 | [diff] [blame] | 47 | static unsigned long long relative_base; |
Kees Cook | 78eb715 | 2014-03-17 13:18:27 +1030 | [diff] [blame] | 48 | static struct addr_range text_ranges[] = { |
Mike Frysinger | 17b1f0d | 2009-06-08 19:12:13 -0400 | [diff] [blame] | 49 | { "_stext", "_etext" }, |
| 50 | { "_sinittext", "_einittext" }, |
| 51 | { "_stext_l1", "_etext_l1" }, /* Blackfin on-chip L1 inst SRAM */ |
| 52 | { "_stext_l2", "_etext_l2" }, /* Blackfin on-chip L2 SRAM */ |
| 53 | }; |
| 54 | #define text_range_text (&text_ranges[0]) |
| 55 | #define text_range_inittext (&text_ranges[1]) |
| 56 | |
Rusty Russell | c6bda7c | 2014-03-17 14:05:46 +1030 | [diff] [blame] | 57 | static struct addr_range percpu_range = { |
| 58 | "__per_cpu_start", "__per_cpu_end", -1ULL, 0 |
| 59 | }; |
| 60 | |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 61 | static struct sym_entry *table; |
Paulo Marques | b3dbb4e | 2005-09-06 15:16:31 -0700 | [diff] [blame] | 62 | static unsigned int table_size, table_cnt; |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 63 | static int all_symbols = 0; |
Rusty Russell | c6bda7c | 2014-03-17 14:05:46 +1030 | [diff] [blame] | 64 | static int absolute_percpu = 0; |
Yoshinori Sato | 41f11a4 | 2005-05-01 08:59:06 -0700 | [diff] [blame] | 65 | static char symbol_prefix_char = '\0'; |
Ard Biesheuvel | 2213e9a | 2016-03-15 14:58:19 -0700 | [diff] [blame] | 66 | static int base_relative = 0; |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 67 | |
Paulo Marques | b3dbb4e | 2005-09-06 15:16:31 -0700 | [diff] [blame] | 68 | int token_profit[0x10000]; |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 69 | |
| 70 | /* the table that holds the result of the compression */ |
Paulo Marques | b3dbb4e | 2005-09-06 15:16:31 -0700 | [diff] [blame] | 71 | unsigned char best_table[256][2]; |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 72 | unsigned char best_table_len[256]; |
| 73 | |
| 74 | |
Paulo Marques | b3dbb4e | 2005-09-06 15:16:31 -0700 | [diff] [blame] | 75 | static void usage(void) |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 76 | { |
Ming Lei | f6537f2 | 2013-11-02 09:11:33 +1030 | [diff] [blame] | 77 | fprintf(stderr, "Usage: kallsyms [--all-symbols] " |
| 78 | "[--symbol-prefix=<prefix char>] " |
| 79 | "[--page-offset=<CONFIG_PAGE_OFFSET>] " |
Ard Biesheuvel | 2213e9a | 2016-03-15 14:58:19 -0700 | [diff] [blame] | 80 | "[--base-relative] < in.map > out.S\n"); |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 81 | exit(1); |
| 82 | } |
| 83 | |
| 84 | /* |
| 85 | * This ignores the intensely annoying "mapping symbols" found |
| 86 | * in ARM ELF files: $a, $t and $d. |
| 87 | */ |
Paulo Marques | b3dbb4e | 2005-09-06 15:16:31 -0700 | [diff] [blame] | 88 | static inline int is_arm_mapping_symbol(const char *str) |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 89 | { |
Kyle McMartin | 6c34f1f | 2014-09-16 22:37:18 +0100 | [diff] [blame] | 90 | return str[0] == '$' && strchr("axtd", str[1]) |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 91 | && (str[2] == '\0' || str[2] == '.'); |
| 92 | } |
| 93 | |
Kees Cook | 78eb715 | 2014-03-17 13:18:27 +1030 | [diff] [blame] | 94 | static int check_symbol_range(const char *sym, unsigned long long addr, |
| 95 | struct addr_range *ranges, int entries) |
Mike Frysinger | 17b1f0d | 2009-06-08 19:12:13 -0400 | [diff] [blame] | 96 | { |
| 97 | size_t i; |
Kees Cook | 78eb715 | 2014-03-17 13:18:27 +1030 | [diff] [blame] | 98 | struct addr_range *ar; |
Mike Frysinger | 17b1f0d | 2009-06-08 19:12:13 -0400 | [diff] [blame] | 99 | |
Kees Cook | 78eb715 | 2014-03-17 13:18:27 +1030 | [diff] [blame] | 100 | for (i = 0; i < entries; ++i) { |
| 101 | ar = &ranges[i]; |
Mike Frysinger | 17b1f0d | 2009-06-08 19:12:13 -0400 | [diff] [blame] | 102 | |
Kees Cook | 78eb715 | 2014-03-17 13:18:27 +1030 | [diff] [blame] | 103 | if (strcmp(sym, ar->start_sym) == 0) { |
| 104 | ar->start = addr; |
Mike Frysinger | 17b1f0d | 2009-06-08 19:12:13 -0400 | [diff] [blame] | 105 | return 0; |
Kees Cook | 78eb715 | 2014-03-17 13:18:27 +1030 | [diff] [blame] | 106 | } else if (strcmp(sym, ar->end_sym) == 0) { |
| 107 | ar->end = addr; |
Mike Frysinger | 17b1f0d | 2009-06-08 19:12:13 -0400 | [diff] [blame] | 108 | return 0; |
| 109 | } |
| 110 | } |
| 111 | |
| 112 | return 1; |
| 113 | } |
| 114 | |
Paulo Marques | b3dbb4e | 2005-09-06 15:16:31 -0700 | [diff] [blame] | 115 | static int read_symbol(FILE *in, struct sym_entry *s) |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 116 | { |
| 117 | char str[500]; |
Paulo Marques | b3dbb4e | 2005-09-06 15:16:31 -0700 | [diff] [blame] | 118 | char *sym, stype; |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 119 | int rc; |
| 120 | |
Paulo Marques | b3dbb4e | 2005-09-06 15:16:31 -0700 | [diff] [blame] | 121 | rc = fscanf(in, "%llx %c %499s\n", &s->addr, &stype, str); |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 122 | if (rc != 3) { |
Jean Sacren | ef89487 | 2010-09-10 23:13:33 -0600 | [diff] [blame] | 123 | if (rc != EOF && fgets(str, 500, in) == NULL) |
| 124 | fprintf(stderr, "Read error or end of file.\n"); |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 125 | return -1; |
| 126 | } |
Andi Kleen | f3462aa | 2013-10-23 15:07:53 +0200 | [diff] [blame] | 127 | if (strlen(str) > KSYM_NAME_LEN) { |
Fabio Estevam | 6f62259 | 2013-11-08 00:45:01 -0200 | [diff] [blame] | 128 | fprintf(stderr, "Symbol %s too long for kallsyms (%zu vs %d).\n" |
Masahiro Yamada | bb66fc6 | 2014-06-10 19:08:13 +0900 | [diff] [blame] | 129 | "Please increase KSYM_NAME_LEN both in kernel and kallsyms.c\n", |
Andi Kleen | f3462aa | 2013-10-23 15:07:53 +0200 | [diff] [blame] | 130 | str, strlen(str), KSYM_NAME_LEN); |
| 131 | return -1; |
| 132 | } |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 133 | |
Yoshinori Sato | 41f11a4 | 2005-05-01 08:59:06 -0700 | [diff] [blame] | 134 | sym = str; |
| 135 | /* skip prefix char */ |
| 136 | if (symbol_prefix_char && str[0] == symbol_prefix_char) |
| 137 | sym++; |
| 138 | |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 139 | /* Ignore most absolute/undefined (?) symbols. */ |
Eric W. Biederman | fd593d1 | 2006-12-07 02:14:04 +0100 | [diff] [blame] | 140 | if (strcmp(sym, "_text") == 0) |
| 141 | _text = s->addr; |
Kees Cook | 78eb715 | 2014-03-17 13:18:27 +1030 | [diff] [blame] | 142 | else if (check_symbol_range(sym, s->addr, text_ranges, |
| 143 | ARRAY_SIZE(text_ranges)) == 0) |
Mike Frysinger | 17b1f0d | 2009-06-08 19:12:13 -0400 | [diff] [blame] | 144 | /* nothing to do */; |
Paulo Marques | b3dbb4e | 2005-09-06 15:16:31 -0700 | [diff] [blame] | 145 | else if (toupper(stype) == 'A') |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 146 | { |
| 147 | /* Keep these useful absolute symbols */ |
Yoshinori Sato | 41f11a4 | 2005-05-01 08:59:06 -0700 | [diff] [blame] | 148 | if (strcmp(sym, "__kernel_syscall_via_break") && |
| 149 | strcmp(sym, "__kernel_syscall_via_epc") && |
| 150 | strcmp(sym, "__kernel_sigtramp") && |
| 151 | strcmp(sym, "__gp")) |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 152 | return -1; |
| 153 | |
| 154 | } |
Paulo Marques | b3dbb4e | 2005-09-06 15:16:31 -0700 | [diff] [blame] | 155 | else if (toupper(stype) == 'U' || |
Yoshinori Sato | 41f11a4 | 2005-05-01 08:59:06 -0700 | [diff] [blame] | 156 | is_arm_mapping_symbol(sym)) |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 157 | return -1; |
Ralf Baechle | 6f00df2 | 2005-09-06 15:16:41 -0700 | [diff] [blame] | 158 | /* exclude also MIPS ELF local symbols ($L123 instead of .L123) */ |
| 159 | else if (str[0] == '$') |
| 160 | return -1; |
Sam Ravnborg | aab34ac | 2008-05-19 20:07:58 +0200 | [diff] [blame] | 161 | /* exclude debugging symbols */ |
| 162 | else if (stype == 'N') |
| 163 | return -1; |
Vasily Gorbik | c677e7a | 2019-06-28 19:22:47 +0200 | [diff] [blame] | 164 | /* exclude s390 kasan local symbols */ |
| 165 | else if (!strncmp(sym, ".LASANPC", 8)) |
| 166 | return -1; |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 167 | |
| 168 | /* include the type field in the symbol name, so that it gets |
| 169 | * compressed together */ |
| 170 | s->len = strlen(str) + 1; |
Paulo Marques | b3dbb4e | 2005-09-06 15:16:31 -0700 | [diff] [blame] | 171 | s->sym = malloc(s->len + 1); |
Jesper Juhl | f1a136e | 2006-03-25 03:07:46 -0800 | [diff] [blame] | 172 | if (!s->sym) { |
| 173 | fprintf(stderr, "kallsyms failure: " |
| 174 | "unable to allocate required amount of memory\n"); |
| 175 | exit(EXIT_FAILURE); |
| 176 | } |
Paulo Marques | b3dbb4e | 2005-09-06 15:16:31 -0700 | [diff] [blame] | 177 | strcpy((char *)s->sym + 1, str); |
| 178 | s->sym[0] = stype; |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 179 | |
Ard Biesheuvel | 8c99694 | 2016-03-15 14:58:15 -0700 | [diff] [blame] | 180 | s->percpu_absolute = 0; |
| 181 | |
Rusty Russell | c6bda7c | 2014-03-17 14:05:46 +1030 | [diff] [blame] | 182 | /* Record if we've found __per_cpu_start/end. */ |
| 183 | check_symbol_range(sym, s->addr, &percpu_range, 1); |
| 184 | |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 185 | return 0; |
| 186 | } |
| 187 | |
Kees Cook | 78eb715 | 2014-03-17 13:18:27 +1030 | [diff] [blame] | 188 | static int symbol_in_range(struct sym_entry *s, struct addr_range *ranges, |
| 189 | int entries) |
Mike Frysinger | 17b1f0d | 2009-06-08 19:12:13 -0400 | [diff] [blame] | 190 | { |
| 191 | size_t i; |
Kees Cook | 78eb715 | 2014-03-17 13:18:27 +1030 | [diff] [blame] | 192 | struct addr_range *ar; |
Mike Frysinger | 17b1f0d | 2009-06-08 19:12:13 -0400 | [diff] [blame] | 193 | |
Kees Cook | 78eb715 | 2014-03-17 13:18:27 +1030 | [diff] [blame] | 194 | for (i = 0; i < entries; ++i) { |
| 195 | ar = &ranges[i]; |
Mike Frysinger | 17b1f0d | 2009-06-08 19:12:13 -0400 | [diff] [blame] | 196 | |
Kees Cook | 78eb715 | 2014-03-17 13:18:27 +1030 | [diff] [blame] | 197 | if (s->addr >= ar->start && s->addr <= ar->end) |
Mike Frysinger | ac6ca5c | 2009-06-15 07:52:48 -0400 | [diff] [blame] | 198 | return 1; |
Mike Frysinger | 17b1f0d | 2009-06-08 19:12:13 -0400 | [diff] [blame] | 199 | } |
| 200 | |
Mike Frysinger | ac6ca5c | 2009-06-15 07:52:48 -0400 | [diff] [blame] | 201 | return 0; |
Mike Frysinger | 17b1f0d | 2009-06-08 19:12:13 -0400 | [diff] [blame] | 202 | } |
| 203 | |
Paulo Marques | b3dbb4e | 2005-09-06 15:16:31 -0700 | [diff] [blame] | 204 | static int symbol_valid(struct sym_entry *s) |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 205 | { |
| 206 | /* Symbols which vary between passes. Passes 1 and 2 must have |
Sam Ravnborg | 2ea0389 | 2009-01-14 21:38:20 +0100 | [diff] [blame] | 207 | * identical symbol lists. The kallsyms_* symbols below are only added |
| 208 | * after pass 1, they would be included in pass 2 when --all-symbols is |
| 209 | * specified so exclude them to get a stable symbol list. |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 210 | */ |
| 211 | static char *special_symbols[] = { |
Sam Ravnborg | 2ea0389 | 2009-01-14 21:38:20 +0100 | [diff] [blame] | 212 | "kallsyms_addresses", |
Ard Biesheuvel | 2213e9a | 2016-03-15 14:58:19 -0700 | [diff] [blame] | 213 | "kallsyms_offsets", |
| 214 | "kallsyms_relative_base", |
Sam Ravnborg | 2ea0389 | 2009-01-14 21:38:20 +0100 | [diff] [blame] | 215 | "kallsyms_num_syms", |
| 216 | "kallsyms_names", |
| 217 | "kallsyms_markers", |
| 218 | "kallsyms_token_table", |
| 219 | "kallsyms_token_index", |
| 220 | |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 221 | /* Exclude linker generated symbols which vary between passes */ |
| 222 | "_SDA_BASE_", /* ppc */ |
| 223 | "_SDA2_BASE_", /* ppc */ |
| 224 | NULL }; |
Ard Biesheuvel | bd8b22d | 2015-03-30 15:20:31 +0200 | [diff] [blame] | 225 | |
| 226 | static char *special_suffixes[] = { |
Ard Biesheuvel | bd8b22d | 2015-03-30 15:20:31 +0200 | [diff] [blame] | 227 | "_veneer", /* arm */ |
Ard Biesheuvel | b9b74be | 2016-04-01 14:33:48 +0100 | [diff] [blame] | 228 | "_from_arm", /* arm */ |
| 229 | "_from_thumb", /* arm */ |
Ard Biesheuvel | bd8b22d | 2015-03-30 15:20:31 +0200 | [diff] [blame] | 230 | NULL }; |
| 231 | |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 232 | int i; |
Ard Biesheuvel | bd8b22d | 2015-03-30 15:20:31 +0200 | [diff] [blame] | 233 | char *sym_name = (char *)s->sym + 1; |
| 234 | |
Yoshinori Sato | 41f11a4 | 2005-05-01 08:59:06 -0700 | [diff] [blame] | 235 | /* skip prefix char */ |
Ard Biesheuvel | bd8b22d | 2015-03-30 15:20:31 +0200 | [diff] [blame] | 236 | if (symbol_prefix_char && *sym_name == symbol_prefix_char) |
| 237 | sym_name++; |
| 238 | |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 239 | |
| 240 | /* if --all-symbols is not specified, then symbols outside the text |
| 241 | * and inittext sections are discarded */ |
| 242 | if (!all_symbols) { |
Kees Cook | 78eb715 | 2014-03-17 13:18:27 +1030 | [diff] [blame] | 243 | if (symbol_in_range(s, text_ranges, |
| 244 | ARRAY_SIZE(text_ranges)) == 0) |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 245 | return 0; |
| 246 | /* Corner case. Discard any symbols with the same value as |
Robin Getz | a3b8111 | 2008-02-06 01:36:26 -0800 | [diff] [blame] | 247 | * _etext _einittext; they can move between pass 1 and 2 when |
| 248 | * the kallsyms data are added. If these symbols move then |
| 249 | * they may get dropped in pass 2, which breaks the kallsyms |
| 250 | * rules. |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 251 | */ |
Mike Frysinger | 17b1f0d | 2009-06-08 19:12:13 -0400 | [diff] [blame] | 252 | if ((s->addr == text_range_text->end && |
Ard Biesheuvel | bd8b22d | 2015-03-30 15:20:31 +0200 | [diff] [blame] | 253 | strcmp(sym_name, |
Kees Cook | 78eb715 | 2014-03-17 13:18:27 +1030 | [diff] [blame] | 254 | text_range_text->end_sym)) || |
Mike Frysinger | 17b1f0d | 2009-06-08 19:12:13 -0400 | [diff] [blame] | 255 | (s->addr == text_range_inittext->end && |
Ard Biesheuvel | bd8b22d | 2015-03-30 15:20:31 +0200 | [diff] [blame] | 256 | strcmp(sym_name, |
Kees Cook | 78eb715 | 2014-03-17 13:18:27 +1030 | [diff] [blame] | 257 | text_range_inittext->end_sym))) |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 258 | return 0; |
| 259 | } |
| 260 | |
| 261 | /* Exclude symbols which vary between passes. */ |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 262 | for (i = 0; special_symbols[i]; i++) |
Ard Biesheuvel | bd8b22d | 2015-03-30 15:20:31 +0200 | [diff] [blame] | 263 | if (strcmp(sym_name, special_symbols[i]) == 0) |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 264 | return 0; |
| 265 | |
Ard Biesheuvel | bd8b22d | 2015-03-30 15:20:31 +0200 | [diff] [blame] | 266 | for (i = 0; special_suffixes[i]; i++) { |
| 267 | int l = strlen(sym_name) - strlen(special_suffixes[i]); |
| 268 | |
| 269 | if (l >= 0 && strcmp(sym_name + l, special_suffixes[i]) == 0) |
| 270 | return 0; |
| 271 | } |
| 272 | |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 273 | return 1; |
| 274 | } |
| 275 | |
Paulo Marques | b3dbb4e | 2005-09-06 15:16:31 -0700 | [diff] [blame] | 276 | static void read_map(FILE *in) |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 277 | { |
| 278 | while (!feof(in)) { |
Paulo Marques | b3dbb4e | 2005-09-06 15:16:31 -0700 | [diff] [blame] | 279 | if (table_cnt >= table_size) { |
| 280 | table_size += 10000; |
| 281 | table = realloc(table, sizeof(*table) * table_size); |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 282 | if (!table) { |
| 283 | fprintf(stderr, "out of memory\n"); |
| 284 | exit (1); |
| 285 | } |
| 286 | } |
Paulo Marques | f2df3f6 | 2008-02-06 01:37:33 -0800 | [diff] [blame] | 287 | if (read_symbol(in, &table[table_cnt]) == 0) { |
| 288 | table[table_cnt].start_pos = table_cnt; |
Paulo Marques | b3dbb4e | 2005-09-06 15:16:31 -0700 | [diff] [blame] | 289 | table_cnt++; |
Paulo Marques | f2df3f6 | 2008-02-06 01:37:33 -0800 | [diff] [blame] | 290 | } |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 291 | } |
| 292 | } |
| 293 | |
| 294 | static void output_label(char *label) |
| 295 | { |
Yoshinori Sato | 41f11a4 | 2005-05-01 08:59:06 -0700 | [diff] [blame] | 296 | if (symbol_prefix_char) |
| 297 | printf(".globl %c%s\n", symbol_prefix_char, label); |
| 298 | else |
| 299 | printf(".globl %s\n", label); |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 300 | printf("\tALGN\n"); |
Yoshinori Sato | 41f11a4 | 2005-05-01 08:59:06 -0700 | [diff] [blame] | 301 | if (symbol_prefix_char) |
| 302 | printf("%c%s:\n", symbol_prefix_char, label); |
| 303 | else |
| 304 | printf("%s:\n", label); |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 305 | } |
| 306 | |
| 307 | /* uncompress a compressed symbol. When this function is called, the best table |
| 308 | * might still be compressed itself, so the function needs to be recursive */ |
| 309 | static int expand_symbol(unsigned char *data, int len, char *result) |
| 310 | { |
| 311 | int c, rlen, total=0; |
| 312 | |
| 313 | while (len) { |
| 314 | c = *data; |
| 315 | /* if the table holds a single char that is the same as the one |
| 316 | * we are looking for, then end the search */ |
| 317 | if (best_table[c][0]==c && best_table_len[c]==1) { |
| 318 | *result++ = c; |
| 319 | total++; |
| 320 | } else { |
| 321 | /* if not, recurse and expand */ |
| 322 | rlen = expand_symbol(best_table[c], best_table_len[c], result); |
| 323 | total += rlen; |
| 324 | result += rlen; |
| 325 | } |
| 326 | data++; |
| 327 | len--; |
| 328 | } |
| 329 | *result=0; |
| 330 | |
| 331 | return total; |
| 332 | } |
| 333 | |
Kees Cook | 78eb715 | 2014-03-17 13:18:27 +1030 | [diff] [blame] | 334 | static int symbol_absolute(struct sym_entry *s) |
| 335 | { |
Ard Biesheuvel | 8c99694 | 2016-03-15 14:58:15 -0700 | [diff] [blame] | 336 | return s->percpu_absolute; |
Kees Cook | 78eb715 | 2014-03-17 13:18:27 +1030 | [diff] [blame] | 337 | } |
| 338 | |
Paulo Marques | b3dbb4e | 2005-09-06 15:16:31 -0700 | [diff] [blame] | 339 | static void write_src(void) |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 340 | { |
Paulo Marques | b3dbb4e | 2005-09-06 15:16:31 -0700 | [diff] [blame] | 341 | unsigned int i, k, off; |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 342 | unsigned int best_idx[256]; |
| 343 | unsigned int *markers; |
Tejun Heo | 9281ace | 2007-07-17 04:03:51 -0700 | [diff] [blame] | 344 | char buf[KSYM_NAME_LEN]; |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 345 | |
| 346 | printf("#include <asm/types.h>\n"); |
| 347 | printf("#if BITS_PER_LONG == 64\n"); |
| 348 | printf("#define PTR .quad\n"); |
| 349 | printf("#define ALGN .align 8\n"); |
| 350 | printf("#else\n"); |
| 351 | printf("#define PTR .long\n"); |
| 352 | printf("#define ALGN .align 4\n"); |
| 353 | printf("#endif\n"); |
| 354 | |
Jan Beulich | aad0947 | 2006-12-08 02:35:57 -0800 | [diff] [blame] | 355 | printf("\t.section .rodata, \"a\"\n"); |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 356 | |
Ard Biesheuvel | 2213e9a | 2016-03-15 14:58:19 -0700 | [diff] [blame] | 357 | /* Provide proper symbols relocatability by their relativeness |
| 358 | * to a fixed anchor point in the runtime image, either '_text' |
| 359 | * for absolute address tables, in which case the linker will |
| 360 | * emit the final addresses at build time. Otherwise, use the |
| 361 | * offset relative to the lowest value encountered of all relative |
| 362 | * symbols, and emit non-relocatable fixed offsets that will be fixed |
| 363 | * up at runtime. |
| 364 | * |
| 365 | * The symbol names cannot be used to construct normal symbol |
| 366 | * references as the list of symbols contains symbols that are |
| 367 | * declared static and are private to their .o files. This prevents |
| 368 | * .tmp_kallsyms.o or any other object from referencing them. |
Eric W. Biederman | fd593d1 | 2006-12-07 02:14:04 +0100 | [diff] [blame] | 369 | */ |
Ard Biesheuvel | 2213e9a | 2016-03-15 14:58:19 -0700 | [diff] [blame] | 370 | if (!base_relative) |
| 371 | output_label("kallsyms_addresses"); |
| 372 | else |
| 373 | output_label("kallsyms_offsets"); |
| 374 | |
Paulo Marques | b3dbb4e | 2005-09-06 15:16:31 -0700 | [diff] [blame] | 375 | for (i = 0; i < table_cnt; i++) { |
Ard Biesheuvel | 2213e9a | 2016-03-15 14:58:19 -0700 | [diff] [blame] | 376 | if (base_relative) { |
| 377 | long long offset; |
| 378 | int overflow; |
| 379 | |
| 380 | if (!absolute_percpu) { |
| 381 | offset = table[i].addr - relative_base; |
| 382 | overflow = (offset < 0 || offset > UINT_MAX); |
| 383 | } else if (symbol_absolute(&table[i])) { |
| 384 | offset = table[i].addr; |
| 385 | overflow = (offset < 0 || offset > INT_MAX); |
| 386 | } else { |
| 387 | offset = relative_base - table[i].addr - 1; |
| 388 | overflow = (offset < INT_MIN || offset >= 0); |
| 389 | } |
| 390 | if (overflow) { |
| 391 | fprintf(stderr, "kallsyms failure: " |
| 392 | "%s symbol value %#llx out of range in relative mode\n", |
| 393 | symbol_absolute(&table[i]) ? "absolute" : "relative", |
| 394 | table[i].addr); |
| 395 | exit(EXIT_FAILURE); |
| 396 | } |
| 397 | printf("\t.long\t%#x\n", (int)offset); |
| 398 | } else if (!symbol_absolute(&table[i])) { |
Vivek Goyal | 2c22d8b | 2006-12-07 02:14:10 +0100 | [diff] [blame] | 399 | if (_text <= table[i].addr) |
| 400 | printf("\tPTR\t_text + %#llx\n", |
| 401 | table[i].addr - _text); |
| 402 | else |
Andrew Morton | 2930ffc | 2014-03-10 15:49:48 -0700 | [diff] [blame] | 403 | printf("\tPTR\t_text - %#llx\n", |
| 404 | _text - table[i].addr); |
Eric W. Biederman | fd593d1 | 2006-12-07 02:14:04 +0100 | [diff] [blame] | 405 | } else { |
| 406 | printf("\tPTR\t%#llx\n", table[i].addr); |
| 407 | } |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 408 | } |
| 409 | printf("\n"); |
| 410 | |
Ard Biesheuvel | 2213e9a | 2016-03-15 14:58:19 -0700 | [diff] [blame] | 411 | if (base_relative) { |
| 412 | output_label("kallsyms_relative_base"); |
| 413 | printf("\tPTR\t_text - %#llx\n", _text - relative_base); |
| 414 | printf("\n"); |
| 415 | } |
| 416 | |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 417 | output_label("kallsyms_num_syms"); |
Paulo Marques | b3dbb4e | 2005-09-06 15:16:31 -0700 | [diff] [blame] | 418 | printf("\tPTR\t%d\n", table_cnt); |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 419 | printf("\n"); |
| 420 | |
| 421 | /* table of offset markers, that give the offset in the compressed stream |
| 422 | * every 256 symbols */ |
Jesper Juhl | f1a136e | 2006-03-25 03:07:46 -0800 | [diff] [blame] | 423 | markers = malloc(sizeof(unsigned int) * ((table_cnt + 255) / 256)); |
| 424 | if (!markers) { |
| 425 | fprintf(stderr, "kallsyms failure: " |
| 426 | "unable to allocate required memory\n"); |
| 427 | exit(EXIT_FAILURE); |
| 428 | } |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 429 | |
| 430 | output_label("kallsyms_names"); |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 431 | off = 0; |
Paulo Marques | b3dbb4e | 2005-09-06 15:16:31 -0700 | [diff] [blame] | 432 | for (i = 0; i < table_cnt; i++) { |
| 433 | if ((i & 0xFF) == 0) |
| 434 | markers[i >> 8] = off; |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 435 | |
| 436 | printf("\t.byte 0x%02x", table[i].len); |
| 437 | for (k = 0; k < table[i].len; k++) |
| 438 | printf(", 0x%02x", table[i].sym[k]); |
| 439 | printf("\n"); |
| 440 | |
| 441 | off += table[i].len + 1; |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 442 | } |
| 443 | printf("\n"); |
| 444 | |
| 445 | output_label("kallsyms_markers"); |
Paulo Marques | b3dbb4e | 2005-09-06 15:16:31 -0700 | [diff] [blame] | 446 | for (i = 0; i < ((table_cnt + 255) >> 8); i++) |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 447 | printf("\tPTR\t%d\n", markers[i]); |
| 448 | printf("\n"); |
| 449 | |
| 450 | free(markers); |
| 451 | |
| 452 | output_label("kallsyms_token_table"); |
| 453 | off = 0; |
| 454 | for (i = 0; i < 256; i++) { |
| 455 | best_idx[i] = off; |
Paulo Marques | b3dbb4e | 2005-09-06 15:16:31 -0700 | [diff] [blame] | 456 | expand_symbol(best_table[i], best_table_len[i], buf); |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 457 | printf("\t.asciz\t\"%s\"\n", buf); |
| 458 | off += strlen(buf) + 1; |
| 459 | } |
| 460 | printf("\n"); |
| 461 | |
| 462 | output_label("kallsyms_token_index"); |
| 463 | for (i = 0; i < 256; i++) |
| 464 | printf("\t.short\t%d\n", best_idx[i]); |
| 465 | printf("\n"); |
| 466 | } |
| 467 | |
| 468 | |
| 469 | /* table lookup compression functions */ |
| 470 | |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 471 | /* count all the possible tokens in a symbol */ |
| 472 | static void learn_symbol(unsigned char *symbol, int len) |
| 473 | { |
| 474 | int i; |
| 475 | |
| 476 | for (i = 0; i < len - 1; i++) |
Paulo Marques | b3dbb4e | 2005-09-06 15:16:31 -0700 | [diff] [blame] | 477 | token_profit[ symbol[i] + (symbol[i + 1] << 8) ]++; |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 478 | } |
| 479 | |
| 480 | /* decrease the count for all the possible tokens in a symbol */ |
| 481 | static void forget_symbol(unsigned char *symbol, int len) |
| 482 | { |
| 483 | int i; |
| 484 | |
| 485 | for (i = 0; i < len - 1; i++) |
Paulo Marques | b3dbb4e | 2005-09-06 15:16:31 -0700 | [diff] [blame] | 486 | token_profit[ symbol[i] + (symbol[i + 1] << 8) ]--; |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 487 | } |
| 488 | |
Paulo Marques | b3dbb4e | 2005-09-06 15:16:31 -0700 | [diff] [blame] | 489 | /* remove all the invalid symbols from the table and do the initial token count */ |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 490 | static void build_initial_tok_table(void) |
| 491 | { |
Paulo Marques | b3dbb4e | 2005-09-06 15:16:31 -0700 | [diff] [blame] | 492 | unsigned int i, pos; |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 493 | |
Paulo Marques | b3dbb4e | 2005-09-06 15:16:31 -0700 | [diff] [blame] | 494 | pos = 0; |
| 495 | for (i = 0; i < table_cnt; i++) { |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 496 | if ( symbol_valid(&table[i]) ) { |
Paulo Marques | b3dbb4e | 2005-09-06 15:16:31 -0700 | [diff] [blame] | 497 | if (pos != i) |
| 498 | table[pos] = table[i]; |
| 499 | learn_symbol(table[pos].sym, table[pos].len); |
| 500 | pos++; |
Masahiro Yamada | 9abb8f6 | 2019-11-24 01:04:30 +0900 | [diff] [blame] | 501 | } else { |
| 502 | free(table[i].sym); |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 503 | } |
| 504 | } |
Paulo Marques | b3dbb4e | 2005-09-06 15:16:31 -0700 | [diff] [blame] | 505 | table_cnt = pos; |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 506 | } |
| 507 | |
Paulo Marques | 7c5d249 | 2007-06-20 18:09:00 +0100 | [diff] [blame] | 508 | static void *find_token(unsigned char *str, int len, unsigned char *token) |
| 509 | { |
| 510 | int i; |
| 511 | |
| 512 | for (i = 0; i < len - 1; i++) { |
| 513 | if (str[i] == token[0] && str[i+1] == token[1]) |
| 514 | return &str[i]; |
| 515 | } |
| 516 | return NULL; |
| 517 | } |
| 518 | |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 519 | /* replace a given token in all the valid symbols. Use the sampled symbols |
| 520 | * to update the counts */ |
Paulo Marques | b3dbb4e | 2005-09-06 15:16:31 -0700 | [diff] [blame] | 521 | static void compress_symbols(unsigned char *str, int idx) |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 522 | { |
Paulo Marques | b3dbb4e | 2005-09-06 15:16:31 -0700 | [diff] [blame] | 523 | unsigned int i, len, size; |
| 524 | unsigned char *p1, *p2; |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 525 | |
Paulo Marques | b3dbb4e | 2005-09-06 15:16:31 -0700 | [diff] [blame] | 526 | for (i = 0; i < table_cnt; i++) { |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 527 | |
| 528 | len = table[i].len; |
Paulo Marques | b3dbb4e | 2005-09-06 15:16:31 -0700 | [diff] [blame] | 529 | p1 = table[i].sym; |
| 530 | |
| 531 | /* find the token on the symbol */ |
Paulo Marques | 7c5d249 | 2007-06-20 18:09:00 +0100 | [diff] [blame] | 532 | p2 = find_token(p1, len, str); |
Paulo Marques | b3dbb4e | 2005-09-06 15:16:31 -0700 | [diff] [blame] | 533 | if (!p2) continue; |
| 534 | |
| 535 | /* decrease the counts for this symbol's tokens */ |
| 536 | forget_symbol(table[i].sym, len); |
| 537 | |
| 538 | size = len; |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 539 | |
| 540 | do { |
Paulo Marques | b3dbb4e | 2005-09-06 15:16:31 -0700 | [diff] [blame] | 541 | *p2 = idx; |
| 542 | p2++; |
| 543 | size -= (p2 - p1); |
| 544 | memmove(p2, p2 + 1, size); |
| 545 | p1 = p2; |
| 546 | len--; |
| 547 | |
| 548 | if (size < 2) break; |
| 549 | |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 550 | /* find the token on the symbol */ |
Paulo Marques | 7c5d249 | 2007-06-20 18:09:00 +0100 | [diff] [blame] | 551 | p2 = find_token(p1, size, str); |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 552 | |
Paulo Marques | b3dbb4e | 2005-09-06 15:16:31 -0700 | [diff] [blame] | 553 | } while (p2); |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 554 | |
Paulo Marques | b3dbb4e | 2005-09-06 15:16:31 -0700 | [diff] [blame] | 555 | table[i].len = len; |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 556 | |
Paulo Marques | b3dbb4e | 2005-09-06 15:16:31 -0700 | [diff] [blame] | 557 | /* increase the counts for this symbol's new tokens */ |
| 558 | learn_symbol(table[i].sym, len); |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 559 | } |
| 560 | } |
| 561 | |
| 562 | /* search the token with the maximum profit */ |
Paulo Marques | b3dbb4e | 2005-09-06 15:16:31 -0700 | [diff] [blame] | 563 | static int find_best_token(void) |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 564 | { |
Paulo Marques | b3dbb4e | 2005-09-06 15:16:31 -0700 | [diff] [blame] | 565 | int i, best, bestprofit; |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 566 | |
| 567 | bestprofit=-10000; |
Paulo Marques | b3dbb4e | 2005-09-06 15:16:31 -0700 | [diff] [blame] | 568 | best = 0; |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 569 | |
Paulo Marques | b3dbb4e | 2005-09-06 15:16:31 -0700 | [diff] [blame] | 570 | for (i = 0; i < 0x10000; i++) { |
| 571 | if (token_profit[i] > bestprofit) { |
| 572 | best = i; |
| 573 | bestprofit = token_profit[i]; |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 574 | } |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 575 | } |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 576 | return best; |
| 577 | } |
| 578 | |
| 579 | /* this is the core of the algorithm: calculate the "best" table */ |
| 580 | static void optimize_result(void) |
| 581 | { |
Paulo Marques | b3dbb4e | 2005-09-06 15:16:31 -0700 | [diff] [blame] | 582 | int i, best; |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 583 | |
| 584 | /* using the '\0' symbol last allows compress_symbols to use standard |
| 585 | * fast string functions */ |
| 586 | for (i = 255; i >= 0; i--) { |
| 587 | |
| 588 | /* if this table slot is empty (it is not used by an actual |
| 589 | * original char code */ |
| 590 | if (!best_table_len[i]) { |
| 591 | |
| 592 | /* find the token with the breates profit value */ |
| 593 | best = find_best_token(); |
Xiaochen Wang | e0a04b1 | 2011-05-01 11:41:41 +0800 | [diff] [blame] | 594 | if (token_profit[best] == 0) |
| 595 | break; |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 596 | |
| 597 | /* place it in the "best" table */ |
Paulo Marques | b3dbb4e | 2005-09-06 15:16:31 -0700 | [diff] [blame] | 598 | best_table_len[i] = 2; |
| 599 | best_table[i][0] = best & 0xFF; |
| 600 | best_table[i][1] = (best >> 8) & 0xFF; |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 601 | |
| 602 | /* replace this token in all the valid symbols */ |
Paulo Marques | b3dbb4e | 2005-09-06 15:16:31 -0700 | [diff] [blame] | 603 | compress_symbols(best_table[i], i); |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 604 | } |
| 605 | } |
| 606 | } |
| 607 | |
| 608 | /* start by placing the symbols that are actually used on the table */ |
| 609 | static void insert_real_symbols_in_table(void) |
| 610 | { |
Paulo Marques | b3dbb4e | 2005-09-06 15:16:31 -0700 | [diff] [blame] | 611 | unsigned int i, j, c; |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 612 | |
| 613 | memset(best_table, 0, sizeof(best_table)); |
| 614 | memset(best_table_len, 0, sizeof(best_table_len)); |
| 615 | |
Paulo Marques | b3dbb4e | 2005-09-06 15:16:31 -0700 | [diff] [blame] | 616 | for (i = 0; i < table_cnt; i++) { |
| 617 | for (j = 0; j < table[i].len; j++) { |
| 618 | c = table[i].sym[j]; |
| 619 | best_table[c][0]=c; |
| 620 | best_table_len[c]=1; |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 621 | } |
| 622 | } |
| 623 | } |
| 624 | |
| 625 | static void optimize_token_table(void) |
| 626 | { |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 627 | build_initial_tok_table(); |
| 628 | |
| 629 | insert_real_symbols_in_table(); |
| 630 | |
Yoshinori Sato | 41f11a4 | 2005-05-01 08:59:06 -0700 | [diff] [blame] | 631 | /* When valid symbol is not registered, exit to error */ |
Paulo Marques | b3dbb4e | 2005-09-06 15:16:31 -0700 | [diff] [blame] | 632 | if (!table_cnt) { |
Yoshinori Sato | 41f11a4 | 2005-05-01 08:59:06 -0700 | [diff] [blame] | 633 | fprintf(stderr, "No valid symbol.\n"); |
| 634 | exit(1); |
| 635 | } |
| 636 | |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 637 | optimize_result(); |
| 638 | } |
| 639 | |
Lai Jiangshan | b478b78 | 2009-03-13 15:10:26 +0800 | [diff] [blame] | 640 | /* guess for "linker script provide" symbol */ |
| 641 | static int may_be_linker_script_provide_symbol(const struct sym_entry *se) |
| 642 | { |
| 643 | const char *symbol = (char *)se->sym + 1; |
| 644 | int len = se->len - 1; |
| 645 | |
| 646 | if (len < 8) |
| 647 | return 0; |
| 648 | |
| 649 | if (symbol[0] != '_' || symbol[1] != '_') |
| 650 | return 0; |
| 651 | |
| 652 | /* __start_XXXXX */ |
| 653 | if (!memcmp(symbol + 2, "start_", 6)) |
| 654 | return 1; |
| 655 | |
| 656 | /* __stop_XXXXX */ |
| 657 | if (!memcmp(symbol + 2, "stop_", 5)) |
| 658 | return 1; |
| 659 | |
| 660 | /* __end_XXXXX */ |
| 661 | if (!memcmp(symbol + 2, "end_", 4)) |
| 662 | return 1; |
| 663 | |
| 664 | /* __XXXXX_start */ |
| 665 | if (!memcmp(symbol + len - 6, "_start", 6)) |
| 666 | return 1; |
| 667 | |
| 668 | /* __XXXXX_end */ |
| 669 | if (!memcmp(symbol + len - 4, "_end", 4)) |
| 670 | return 1; |
| 671 | |
| 672 | return 0; |
| 673 | } |
| 674 | |
| 675 | static int prefix_underscores_count(const char *str) |
| 676 | { |
| 677 | const char *tail = str; |
| 678 | |
Paul Mundt | a9ece53 | 2009-09-22 16:44:12 -0700 | [diff] [blame] | 679 | while (*tail == '_') |
Lai Jiangshan | b478b78 | 2009-03-13 15:10:26 +0800 | [diff] [blame] | 680 | tail++; |
| 681 | |
| 682 | return tail - str; |
| 683 | } |
| 684 | |
Paulo Marques | f2df3f6 | 2008-02-06 01:37:33 -0800 | [diff] [blame] | 685 | static int compare_symbols(const void *a, const void *b) |
| 686 | { |
| 687 | const struct sym_entry *sa; |
| 688 | const struct sym_entry *sb; |
| 689 | int wa, wb; |
| 690 | |
| 691 | sa = a; |
| 692 | sb = b; |
| 693 | |
| 694 | /* sort by address first */ |
| 695 | if (sa->addr > sb->addr) |
| 696 | return 1; |
| 697 | if (sa->addr < sb->addr) |
| 698 | return -1; |
| 699 | |
| 700 | /* sort by "weakness" type */ |
| 701 | wa = (sa->sym[0] == 'w') || (sa->sym[0] == 'W'); |
| 702 | wb = (sb->sym[0] == 'w') || (sb->sym[0] == 'W'); |
| 703 | if (wa != wb) |
| 704 | return wa - wb; |
| 705 | |
Lai Jiangshan | b478b78 | 2009-03-13 15:10:26 +0800 | [diff] [blame] | 706 | /* sort by "linker script provide" type */ |
| 707 | wa = may_be_linker_script_provide_symbol(sa); |
| 708 | wb = may_be_linker_script_provide_symbol(sb); |
| 709 | if (wa != wb) |
| 710 | return wa - wb; |
| 711 | |
| 712 | /* sort by the number of prefix underscores */ |
| 713 | wa = prefix_underscores_count((const char *)sa->sym + 1); |
| 714 | wb = prefix_underscores_count((const char *)sb->sym + 1); |
| 715 | if (wa != wb) |
| 716 | return wa - wb; |
| 717 | |
Paulo Marques | f2df3f6 | 2008-02-06 01:37:33 -0800 | [diff] [blame] | 718 | /* sort by initial order, so that other symbols are left undisturbed */ |
| 719 | return sa->start_pos - sb->start_pos; |
| 720 | } |
| 721 | |
| 722 | static void sort_symbols(void) |
| 723 | { |
| 724 | qsort(table, table_cnt, sizeof(struct sym_entry), compare_symbols); |
| 725 | } |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 726 | |
Rusty Russell | c6bda7c | 2014-03-17 14:05:46 +1030 | [diff] [blame] | 727 | static void make_percpus_absolute(void) |
| 728 | { |
| 729 | unsigned int i; |
| 730 | |
| 731 | for (i = 0; i < table_cnt; i++) |
Ard Biesheuvel | 8c99694 | 2016-03-15 14:58:15 -0700 | [diff] [blame] | 732 | if (symbol_in_range(&table[i], &percpu_range, 1)) { |
| 733 | /* |
| 734 | * Keep the 'A' override for percpu symbols to |
| 735 | * ensure consistent behavior compared to older |
| 736 | * versions of this tool. |
| 737 | */ |
Rusty Russell | c6bda7c | 2014-03-17 14:05:46 +1030 | [diff] [blame] | 738 | table[i].sym[0] = 'A'; |
Ard Biesheuvel | 8c99694 | 2016-03-15 14:58:15 -0700 | [diff] [blame] | 739 | table[i].percpu_absolute = 1; |
| 740 | } |
Rusty Russell | c6bda7c | 2014-03-17 14:05:46 +1030 | [diff] [blame] | 741 | } |
| 742 | |
Ard Biesheuvel | 2213e9a | 2016-03-15 14:58:19 -0700 | [diff] [blame] | 743 | /* find the minimum non-absolute symbol address */ |
| 744 | static void record_relative_base(void) |
| 745 | { |
| 746 | unsigned int i; |
| 747 | |
| 748 | relative_base = -1ULL; |
| 749 | for (i = 0; i < table_cnt; i++) |
| 750 | if (!symbol_absolute(&table[i]) && |
| 751 | table[i].addr < relative_base) |
| 752 | relative_base = table[i].addr; |
| 753 | } |
| 754 | |
Paulo Marques | b3dbb4e | 2005-09-06 15:16:31 -0700 | [diff] [blame] | 755 | int main(int argc, char **argv) |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 756 | { |
Yoshinori Sato | 41f11a4 | 2005-05-01 08:59:06 -0700 | [diff] [blame] | 757 | if (argc >= 2) { |
| 758 | int i; |
| 759 | for (i = 1; i < argc; i++) { |
| 760 | if(strcmp(argv[i], "--all-symbols") == 0) |
| 761 | all_symbols = 1; |
Rusty Russell | c6bda7c | 2014-03-17 14:05:46 +1030 | [diff] [blame] | 762 | else if (strcmp(argv[i], "--absolute-percpu") == 0) |
| 763 | absolute_percpu = 1; |
Yoshinori Sato | 41f11a4 | 2005-05-01 08:59:06 -0700 | [diff] [blame] | 764 | else if (strncmp(argv[i], "--symbol-prefix=", 16) == 0) { |
| 765 | char *p = &argv[i][16]; |
| 766 | /* skip quote */ |
| 767 | if ((*p == '"' && *(p+2) == '"') || (*p == '\'' && *(p+2) == '\'')) |
| 768 | p++; |
| 769 | symbol_prefix_char = *p; |
Ard Biesheuvel | 2213e9a | 2016-03-15 14:58:19 -0700 | [diff] [blame] | 770 | } else if (strcmp(argv[i], "--base-relative") == 0) |
| 771 | base_relative = 1; |
| 772 | else |
Yoshinori Sato | 41f11a4 | 2005-05-01 08:59:06 -0700 | [diff] [blame] | 773 | usage(); |
| 774 | } |
| 775 | } else if (argc != 1) |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 776 | usage(); |
| 777 | |
| 778 | read_map(stdin); |
Rusty Russell | c6bda7c | 2014-03-17 14:05:46 +1030 | [diff] [blame] | 779 | if (absolute_percpu) |
| 780 | make_percpus_absolute(); |
Ard Biesheuvel | 2213e9a | 2016-03-15 14:58:19 -0700 | [diff] [blame] | 781 | if (base_relative) |
| 782 | record_relative_base(); |
Sam Ravnborg | 2ea0389 | 2009-01-14 21:38:20 +0100 | [diff] [blame] | 783 | sort_symbols(); |
| 784 | optimize_token_table(); |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 785 | write_src(); |
| 786 | |
| 787 | return 0; |
| 788 | } |