1dd590d4dSJosh Poimboeuf // SPDX-License-Identifier: GPL-2.0-or-later
2dd590d4dSJosh Poimboeuf #define _GNU_SOURCE /* memmem() */
3dd590d4dSJosh Poimboeuf #include <subcmd/parse-options.h>
4dd590d4dSJosh Poimboeuf #include <stdlib.h>
5dd590d4dSJosh Poimboeuf #include <string.h>
6dd590d4dSJosh Poimboeuf #include <libgen.h>
7dd590d4dSJosh Poimboeuf #include <stdio.h>
8dd590d4dSJosh Poimboeuf #include <ctype.h>
9dd590d4dSJosh Poimboeuf
10dd590d4dSJosh Poimboeuf #include <objtool/objtool.h>
11dd590d4dSJosh Poimboeuf #include <objtool/warn.h>
12dd590d4dSJosh Poimboeuf #include <objtool/arch.h>
13dd590d4dSJosh Poimboeuf #include <objtool/klp.h>
14dd590d4dSJosh Poimboeuf #include <objtool/util.h>
15dd590d4dSJosh Poimboeuf #include <arch/special.h>
16dd590d4dSJosh Poimboeuf
17*2f2600deSJoe Lawrence #include <linux/align.h>
18dd590d4dSJosh Poimboeuf #include <linux/objtool_types.h>
19dd590d4dSJosh Poimboeuf #include <linux/livepatch_external.h>
20dd590d4dSJosh Poimboeuf #include <linux/stringify.h>
21dd590d4dSJosh Poimboeuf #include <linux/string.h>
22dd590d4dSJosh Poimboeuf #include <linux/jhash.h>
23dd590d4dSJosh Poimboeuf
24dd590d4dSJosh Poimboeuf #define sizeof_field(TYPE, MEMBER) sizeof((((TYPE *)0)->MEMBER))
25dd590d4dSJosh Poimboeuf
26dd590d4dSJosh Poimboeuf struct elfs {
27dd590d4dSJosh Poimboeuf struct elf *orig, *patched, *out;
28dd590d4dSJosh Poimboeuf const char *modname;
29dd590d4dSJosh Poimboeuf };
30dd590d4dSJosh Poimboeuf
31dd590d4dSJosh Poimboeuf struct export {
32dd590d4dSJosh Poimboeuf struct hlist_node hash;
33dd590d4dSJosh Poimboeuf char *mod, *sym;
34dd590d4dSJosh Poimboeuf };
35dd590d4dSJosh Poimboeuf
36dd590d4dSJosh Poimboeuf static const char * const klp_diff_usage[] = {
37dd590d4dSJosh Poimboeuf "objtool klp diff [<options>] <in1.o> <in2.o> <out.o>",
38dd590d4dSJosh Poimboeuf NULL,
39dd590d4dSJosh Poimboeuf };
40dd590d4dSJosh Poimboeuf
41dd590d4dSJosh Poimboeuf static const struct option klp_diff_options[] = {
427c2575a6SJosh Poimboeuf OPT_GROUP("Options:"),
437c2575a6SJosh Poimboeuf OPT_BOOLEAN('d', "debug", &debug, "enable debug output"),
44dd590d4dSJosh Poimboeuf OPT_END(),
45dd590d4dSJosh Poimboeuf };
46dd590d4dSJosh Poimboeuf
47dd590d4dSJosh Poimboeuf static DEFINE_HASHTABLE(exports, 15);
48dd590d4dSJosh Poimboeuf
str_hash(const char * str)49dd590d4dSJosh Poimboeuf static inline u32 str_hash(const char *str)
50dd590d4dSJosh Poimboeuf {
51dd590d4dSJosh Poimboeuf return jhash(str, strlen(str), 0);
52dd590d4dSJosh Poimboeuf }
53dd590d4dSJosh Poimboeuf
escape_str(const char * orig)547c2575a6SJosh Poimboeuf static char *escape_str(const char *orig)
557c2575a6SJosh Poimboeuf {
567c2575a6SJosh Poimboeuf size_t len = 0;
577c2575a6SJosh Poimboeuf const char *a;
587c2575a6SJosh Poimboeuf char *b, *new;
597c2575a6SJosh Poimboeuf
607c2575a6SJosh Poimboeuf for (a = orig; *a; a++) {
617c2575a6SJosh Poimboeuf switch (*a) {
627c2575a6SJosh Poimboeuf case '\001': len += 5; break;
637c2575a6SJosh Poimboeuf case '\n':
647c2575a6SJosh Poimboeuf case '\t': len += 2; break;
657c2575a6SJosh Poimboeuf default: len++;
667c2575a6SJosh Poimboeuf }
677c2575a6SJosh Poimboeuf }
687c2575a6SJosh Poimboeuf
697c2575a6SJosh Poimboeuf new = malloc(len + 1);
707c2575a6SJosh Poimboeuf if (!new)
717c2575a6SJosh Poimboeuf return NULL;
727c2575a6SJosh Poimboeuf
737c2575a6SJosh Poimboeuf for (a = orig, b = new; *a; a++) {
747c2575a6SJosh Poimboeuf switch (*a) {
757c2575a6SJosh Poimboeuf case '\001': memcpy(b, "<SOH>", 5); b += 5; break;
767c2575a6SJosh Poimboeuf case '\n': *b++ = '\\'; *b++ = 'n'; break;
777c2575a6SJosh Poimboeuf case '\t': *b++ = '\\'; *b++ = 't'; break;
787c2575a6SJosh Poimboeuf default: *b++ = *a;
797c2575a6SJosh Poimboeuf }
807c2575a6SJosh Poimboeuf }
817c2575a6SJosh Poimboeuf
827c2575a6SJosh Poimboeuf *b = '\0';
837c2575a6SJosh Poimboeuf return new;
847c2575a6SJosh Poimboeuf }
857c2575a6SJosh Poimboeuf
read_exports(void)86dd590d4dSJosh Poimboeuf static int read_exports(void)
87dd590d4dSJosh Poimboeuf {
88dd590d4dSJosh Poimboeuf const char *symvers = "Module.symvers";
89dd590d4dSJosh Poimboeuf char line[1024], *path = NULL;
90dd590d4dSJosh Poimboeuf unsigned int line_num = 1;
91dd590d4dSJosh Poimboeuf FILE *file;
92dd590d4dSJosh Poimboeuf
93dd590d4dSJosh Poimboeuf file = fopen(symvers, "r");
94dd590d4dSJosh Poimboeuf if (!file) {
95dd590d4dSJosh Poimboeuf path = top_level_dir(symvers);
96dd590d4dSJosh Poimboeuf if (!path) {
97dd590d4dSJosh Poimboeuf ERROR("can't open '%s', \"objtool diff\" should be run from the kernel tree", symvers);
98dd590d4dSJosh Poimboeuf return -1;
99dd590d4dSJosh Poimboeuf }
100dd590d4dSJosh Poimboeuf
101dd590d4dSJosh Poimboeuf file = fopen(path, "r");
102dd590d4dSJosh Poimboeuf if (!file) {
103dd590d4dSJosh Poimboeuf ERROR_GLIBC("fopen");
104dd590d4dSJosh Poimboeuf return -1;
105dd590d4dSJosh Poimboeuf }
106dd590d4dSJosh Poimboeuf }
107dd590d4dSJosh Poimboeuf
108dd590d4dSJosh Poimboeuf while (fgets(line, 1024, file)) {
109dd590d4dSJosh Poimboeuf char *sym, *mod, *type;
110dd590d4dSJosh Poimboeuf struct export *export;
111dd590d4dSJosh Poimboeuf
112dd590d4dSJosh Poimboeuf sym = strchr(line, '\t');
113dd590d4dSJosh Poimboeuf if (!sym) {
114dd590d4dSJosh Poimboeuf ERROR("malformed Module.symvers (sym) at line %d", line_num);
115dd590d4dSJosh Poimboeuf return -1;
116dd590d4dSJosh Poimboeuf }
117dd590d4dSJosh Poimboeuf
118dd590d4dSJosh Poimboeuf *sym++ = '\0';
119dd590d4dSJosh Poimboeuf
120dd590d4dSJosh Poimboeuf mod = strchr(sym, '\t');
121dd590d4dSJosh Poimboeuf if (!mod) {
122dd590d4dSJosh Poimboeuf ERROR("malformed Module.symvers (mod) at line %d", line_num);
123dd590d4dSJosh Poimboeuf return -1;
124dd590d4dSJosh Poimboeuf }
125dd590d4dSJosh Poimboeuf
126dd590d4dSJosh Poimboeuf *mod++ = '\0';
127dd590d4dSJosh Poimboeuf
128dd590d4dSJosh Poimboeuf type = strchr(mod, '\t');
129dd590d4dSJosh Poimboeuf if (!type) {
130dd590d4dSJosh Poimboeuf ERROR("malformed Module.symvers (type) at line %d", line_num);
131dd590d4dSJosh Poimboeuf return -1;
132dd590d4dSJosh Poimboeuf }
133dd590d4dSJosh Poimboeuf
134dd590d4dSJosh Poimboeuf *type++ = '\0';
135dd590d4dSJosh Poimboeuf
136dd590d4dSJosh Poimboeuf if (*sym == '\0' || *mod == '\0') {
137dd590d4dSJosh Poimboeuf ERROR("malformed Module.symvers at line %d", line_num);
138dd590d4dSJosh Poimboeuf return -1;
139dd590d4dSJosh Poimboeuf }
140dd590d4dSJosh Poimboeuf
141dd590d4dSJosh Poimboeuf export = calloc(1, sizeof(*export));
142dd590d4dSJosh Poimboeuf if (!export) {
143dd590d4dSJosh Poimboeuf ERROR_GLIBC("calloc");
144dd590d4dSJosh Poimboeuf return -1;
145dd590d4dSJosh Poimboeuf }
146dd590d4dSJosh Poimboeuf
147dd590d4dSJosh Poimboeuf export->mod = strdup(mod);
148dd590d4dSJosh Poimboeuf if (!export->mod) {
149dd590d4dSJosh Poimboeuf ERROR_GLIBC("strdup");
150dd590d4dSJosh Poimboeuf return -1;
151dd590d4dSJosh Poimboeuf }
152dd590d4dSJosh Poimboeuf
153dd590d4dSJosh Poimboeuf export->sym = strdup(sym);
154dd590d4dSJosh Poimboeuf if (!export->sym) {
155dd590d4dSJosh Poimboeuf ERROR_GLIBC("strdup");
156dd590d4dSJosh Poimboeuf return -1;
157dd590d4dSJosh Poimboeuf }
158dd590d4dSJosh Poimboeuf
159dd590d4dSJosh Poimboeuf hash_add(exports, &export->hash, str_hash(sym));
160dd590d4dSJosh Poimboeuf }
161dd590d4dSJosh Poimboeuf
162dd590d4dSJosh Poimboeuf free(path);
163dd590d4dSJosh Poimboeuf fclose(file);
164dd590d4dSJosh Poimboeuf
165dd590d4dSJosh Poimboeuf return 0;
166dd590d4dSJosh Poimboeuf }
167dd590d4dSJosh Poimboeuf
read_sym_checksums(struct elf * elf)168dd590d4dSJosh Poimboeuf static int read_sym_checksums(struct elf *elf)
169dd590d4dSJosh Poimboeuf {
170dd590d4dSJosh Poimboeuf struct section *sec;
171dd590d4dSJosh Poimboeuf
172dd590d4dSJosh Poimboeuf sec = find_section_by_name(elf, ".discard.sym_checksum");
173dd590d4dSJosh Poimboeuf if (!sec) {
174dd590d4dSJosh Poimboeuf ERROR("'%s' missing .discard.sym_checksum section, file not processed by 'objtool --checksum'?",
175dd590d4dSJosh Poimboeuf elf->name);
176dd590d4dSJosh Poimboeuf return -1;
177dd590d4dSJosh Poimboeuf }
178dd590d4dSJosh Poimboeuf
179dd590d4dSJosh Poimboeuf if (!sec->rsec) {
180dd590d4dSJosh Poimboeuf ERROR("missing reloc section for .discard.sym_checksum");
181dd590d4dSJosh Poimboeuf return -1;
182dd590d4dSJosh Poimboeuf }
183dd590d4dSJosh Poimboeuf
184dd590d4dSJosh Poimboeuf if (sec_size(sec) % sizeof(struct sym_checksum)) {
185dd590d4dSJosh Poimboeuf ERROR("struct sym_checksum size mismatch");
186dd590d4dSJosh Poimboeuf return -1;
187dd590d4dSJosh Poimboeuf }
188dd590d4dSJosh Poimboeuf
189dd590d4dSJosh Poimboeuf for (int i = 0; i < sec_size(sec) / sizeof(struct sym_checksum); i++) {
190dd590d4dSJosh Poimboeuf struct sym_checksum *sym_checksum;
191dd590d4dSJosh Poimboeuf struct reloc *reloc;
192dd590d4dSJosh Poimboeuf struct symbol *sym;
193dd590d4dSJosh Poimboeuf
194dd590d4dSJosh Poimboeuf sym_checksum = (struct sym_checksum *)sec->data->d_buf + i;
195dd590d4dSJosh Poimboeuf
196dd590d4dSJosh Poimboeuf reloc = find_reloc_by_dest(elf, sec, i * sizeof(*sym_checksum));
197dd590d4dSJosh Poimboeuf if (!reloc) {
198dd590d4dSJosh Poimboeuf ERROR("can't find reloc for sym_checksum[%d]", i);
199dd590d4dSJosh Poimboeuf return -1;
200dd590d4dSJosh Poimboeuf }
201dd590d4dSJosh Poimboeuf
202dd590d4dSJosh Poimboeuf sym = reloc->sym;
203dd590d4dSJosh Poimboeuf
204dd590d4dSJosh Poimboeuf if (is_sec_sym(sym)) {
205dd590d4dSJosh Poimboeuf ERROR("not sure how to handle section %s", sym->name);
206dd590d4dSJosh Poimboeuf return -1;
207dd590d4dSJosh Poimboeuf }
208dd590d4dSJosh Poimboeuf
209dd590d4dSJosh Poimboeuf if (is_func_sym(sym))
210dd590d4dSJosh Poimboeuf sym->csum.checksum = sym_checksum->checksum;
211dd590d4dSJosh Poimboeuf }
212dd590d4dSJosh Poimboeuf
213dd590d4dSJosh Poimboeuf return 0;
214dd590d4dSJosh Poimboeuf }
215dd590d4dSJosh Poimboeuf
first_file_symbol(struct elf * elf)216dd590d4dSJosh Poimboeuf static struct symbol *first_file_symbol(struct elf *elf)
217dd590d4dSJosh Poimboeuf {
218dd590d4dSJosh Poimboeuf struct symbol *sym;
219dd590d4dSJosh Poimboeuf
220dd590d4dSJosh Poimboeuf for_each_sym(elf, sym) {
221dd590d4dSJosh Poimboeuf if (is_file_sym(sym))
222dd590d4dSJosh Poimboeuf return sym;
223dd590d4dSJosh Poimboeuf }
224dd590d4dSJosh Poimboeuf
225dd590d4dSJosh Poimboeuf return NULL;
226dd590d4dSJosh Poimboeuf }
227dd590d4dSJosh Poimboeuf
next_file_symbol(struct elf * elf,struct symbol * sym)228dd590d4dSJosh Poimboeuf static struct symbol *next_file_symbol(struct elf *elf, struct symbol *sym)
229dd590d4dSJosh Poimboeuf {
230dd590d4dSJosh Poimboeuf for_each_sym_continue(elf, sym) {
231dd590d4dSJosh Poimboeuf if (is_file_sym(sym))
232dd590d4dSJosh Poimboeuf return sym;
233dd590d4dSJosh Poimboeuf }
234dd590d4dSJosh Poimboeuf
235dd590d4dSJosh Poimboeuf return NULL;
236dd590d4dSJosh Poimboeuf }
237dd590d4dSJosh Poimboeuf
238dd590d4dSJosh Poimboeuf /*
239dd590d4dSJosh Poimboeuf * Certain static local variables should never be correlated. They will be
240dd590d4dSJosh Poimboeuf * used in place rather than referencing the originals.
241dd590d4dSJosh Poimboeuf */
is_uncorrelated_static_local(struct symbol * sym)242dd590d4dSJosh Poimboeuf static bool is_uncorrelated_static_local(struct symbol *sym)
243dd590d4dSJosh Poimboeuf {
244dd590d4dSJosh Poimboeuf static const char * const vars[] = {
245dd590d4dSJosh Poimboeuf "__already_done.",
246dd590d4dSJosh Poimboeuf "__func__.",
24724ebfcd6SJosh Poimboeuf "__key.",
24824ebfcd6SJosh Poimboeuf "__warned.",
24924ebfcd6SJosh Poimboeuf "_entry.",
25024ebfcd6SJosh Poimboeuf "_entry_ptr.",
251dd590d4dSJosh Poimboeuf "_rs.",
252dd590d4dSJosh Poimboeuf "descriptor.",
253dd590d4dSJosh Poimboeuf "CSWTCH.",
254dd590d4dSJosh Poimboeuf };
255dd590d4dSJosh Poimboeuf
256dd590d4dSJosh Poimboeuf if (!is_object_sym(sym) || !is_local_sym(sym))
257dd590d4dSJosh Poimboeuf return false;
258dd590d4dSJosh Poimboeuf
259dd590d4dSJosh Poimboeuf if (!strcmp(sym->sec->name, ".data.once"))
260dd590d4dSJosh Poimboeuf return true;
261dd590d4dSJosh Poimboeuf
262dd590d4dSJosh Poimboeuf for (int i = 0; i < ARRAY_SIZE(vars); i++) {
263dd590d4dSJosh Poimboeuf if (strstarts(sym->name, vars[i]))
264dd590d4dSJosh Poimboeuf return true;
265dd590d4dSJosh Poimboeuf }
266dd590d4dSJosh Poimboeuf
267dd590d4dSJosh Poimboeuf return false;
268dd590d4dSJosh Poimboeuf }
269dd590d4dSJosh Poimboeuf
270dd590d4dSJosh Poimboeuf /*
271dd590d4dSJosh Poimboeuf * Clang emits several useless .Ltmp_* code labels.
272dd590d4dSJosh Poimboeuf */
is_clang_tmp_label(struct symbol * sym)273dd590d4dSJosh Poimboeuf static bool is_clang_tmp_label(struct symbol *sym)
274dd590d4dSJosh Poimboeuf {
275dd590d4dSJosh Poimboeuf return sym->type == STT_NOTYPE &&
276dd590d4dSJosh Poimboeuf is_text_sec(sym->sec) &&
277dd590d4dSJosh Poimboeuf strstarts(sym->name, ".Ltmp") &&
278dd590d4dSJosh Poimboeuf isdigit(sym->name[5]);
279dd590d4dSJosh Poimboeuf }
280dd590d4dSJosh Poimboeuf
is_special_section(struct section * sec)281dd590d4dSJosh Poimboeuf static bool is_special_section(struct section *sec)
282dd590d4dSJosh Poimboeuf {
283dd590d4dSJosh Poimboeuf static const char * const specials[] = {
284dd590d4dSJosh Poimboeuf ".altinstructions",
285dd590d4dSJosh Poimboeuf ".smp_locks",
286dd590d4dSJosh Poimboeuf "__bug_table",
287dd590d4dSJosh Poimboeuf "__ex_table",
288dd590d4dSJosh Poimboeuf "__jump_table",
289dd590d4dSJosh Poimboeuf "__mcount_loc",
290dd590d4dSJosh Poimboeuf
291dd590d4dSJosh Poimboeuf /*
292dd590d4dSJosh Poimboeuf * Extract .static_call_sites here to inherit non-module
293dd590d4dSJosh Poimboeuf * preferential treatment. The later static call processing
294dd590d4dSJosh Poimboeuf * during klp module build will be skipped when it sees this
295dd590d4dSJosh Poimboeuf * section already exists.
296dd590d4dSJosh Poimboeuf */
297dd590d4dSJosh Poimboeuf ".static_call_sites",
298dd590d4dSJosh Poimboeuf };
299dd590d4dSJosh Poimboeuf
300dd590d4dSJosh Poimboeuf static const char * const non_special_discards[] = {
301dd590d4dSJosh Poimboeuf ".discard.addressable",
302dd590d4dSJosh Poimboeuf ".discard.sym_checksum",
303dd590d4dSJosh Poimboeuf };
304dd590d4dSJosh Poimboeuf
305dd590d4dSJosh Poimboeuf if (is_text_sec(sec))
306dd590d4dSJosh Poimboeuf return false;
307dd590d4dSJosh Poimboeuf
308dd590d4dSJosh Poimboeuf for (int i = 0; i < ARRAY_SIZE(specials); i++) {
309dd590d4dSJosh Poimboeuf if (!strcmp(sec->name, specials[i]))
310dd590d4dSJosh Poimboeuf return true;
311dd590d4dSJosh Poimboeuf }
312dd590d4dSJosh Poimboeuf
313dd590d4dSJosh Poimboeuf /* Most .discard data sections are special */
314dd590d4dSJosh Poimboeuf for (int i = 0; i < ARRAY_SIZE(non_special_discards); i++) {
315dd590d4dSJosh Poimboeuf if (!strcmp(sec->name, non_special_discards[i]))
316dd590d4dSJosh Poimboeuf return false;
317dd590d4dSJosh Poimboeuf }
318dd590d4dSJosh Poimboeuf
319dd590d4dSJosh Poimboeuf return strstarts(sec->name, ".discard.");
320dd590d4dSJosh Poimboeuf }
321dd590d4dSJosh Poimboeuf
322dd590d4dSJosh Poimboeuf /*
323dd590d4dSJosh Poimboeuf * These sections are referenced by special sections but aren't considered
324dd590d4dSJosh Poimboeuf * special sections themselves.
325dd590d4dSJosh Poimboeuf */
is_special_section_aux(struct section * sec)326dd590d4dSJosh Poimboeuf static bool is_special_section_aux(struct section *sec)
327dd590d4dSJosh Poimboeuf {
328dd590d4dSJosh Poimboeuf static const char * const specials_aux[] = {
329dd590d4dSJosh Poimboeuf ".altinstr_replacement",
330dd590d4dSJosh Poimboeuf ".altinstr_aux",
331dd590d4dSJosh Poimboeuf };
332dd590d4dSJosh Poimboeuf
333dd590d4dSJosh Poimboeuf for (int i = 0; i < ARRAY_SIZE(specials_aux); i++) {
334dd590d4dSJosh Poimboeuf if (!strcmp(sec->name, specials_aux[i]))
335dd590d4dSJosh Poimboeuf return true;
336dd590d4dSJosh Poimboeuf }
337dd590d4dSJosh Poimboeuf
338dd590d4dSJosh Poimboeuf return false;
339dd590d4dSJosh Poimboeuf }
340dd590d4dSJosh Poimboeuf
341dd590d4dSJosh Poimboeuf /*
342dd590d4dSJosh Poimboeuf * These symbols should never be correlated, so their local patched versions
343dd590d4dSJosh Poimboeuf * are used instead of linking to the originals.
344dd590d4dSJosh Poimboeuf */
dont_correlate(struct symbol * sym)345dd590d4dSJosh Poimboeuf static bool dont_correlate(struct symbol *sym)
346dd590d4dSJosh Poimboeuf {
347dd590d4dSJosh Poimboeuf return is_file_sym(sym) ||
348dd590d4dSJosh Poimboeuf is_null_sym(sym) ||
349dd590d4dSJosh Poimboeuf is_sec_sym(sym) ||
350dd590d4dSJosh Poimboeuf is_prefix_func(sym) ||
351dd590d4dSJosh Poimboeuf is_uncorrelated_static_local(sym) ||
352dd590d4dSJosh Poimboeuf is_clang_tmp_label(sym) ||
353dd590d4dSJosh Poimboeuf is_string_sec(sym->sec) ||
354dd590d4dSJosh Poimboeuf is_special_section(sym->sec) ||
355dd590d4dSJosh Poimboeuf is_special_section_aux(sym->sec) ||
356dd590d4dSJosh Poimboeuf strstarts(sym->name, "__initcall__");
357dd590d4dSJosh Poimboeuf }
358dd590d4dSJosh Poimboeuf
359dd590d4dSJosh Poimboeuf /*
360dd590d4dSJosh Poimboeuf * For each symbol in the original kernel, find its corresponding "twin" in the
361dd590d4dSJosh Poimboeuf * patched kernel.
362dd590d4dSJosh Poimboeuf */
correlate_symbols(struct elfs * e)363dd590d4dSJosh Poimboeuf static int correlate_symbols(struct elfs *e)
364dd590d4dSJosh Poimboeuf {
365dd590d4dSJosh Poimboeuf struct symbol *file1_sym, *file2_sym;
366dd590d4dSJosh Poimboeuf struct symbol *sym1, *sym2;
367dd590d4dSJosh Poimboeuf
36818328546SJosh Poimboeuf file1_sym = first_file_symbol(e->orig);
36918328546SJosh Poimboeuf file2_sym = first_file_symbol(e->patched);
37018328546SJosh Poimboeuf
37118328546SJosh Poimboeuf /*
37218328546SJosh Poimboeuf * Correlate any locals before the first FILE symbol. This has been
37318328546SJosh Poimboeuf * seen when LTO inexplicably strips the initramfs_data.o FILE symbol
37418328546SJosh Poimboeuf * due to the file only containing data and no code.
37518328546SJosh Poimboeuf */
37618328546SJosh Poimboeuf for_each_sym(e->orig, sym1) {
37718328546SJosh Poimboeuf if (sym1 == file1_sym || !is_local_sym(sym1))
37818328546SJosh Poimboeuf break;
37918328546SJosh Poimboeuf
38018328546SJosh Poimboeuf if (dont_correlate(sym1))
38118328546SJosh Poimboeuf continue;
38218328546SJosh Poimboeuf
38318328546SJosh Poimboeuf for_each_sym(e->patched, sym2) {
38418328546SJosh Poimboeuf if (sym2 == file2_sym || !is_local_sym(sym2))
38518328546SJosh Poimboeuf break;
38618328546SJosh Poimboeuf
38718328546SJosh Poimboeuf if (sym2->twin || dont_correlate(sym2))
38818328546SJosh Poimboeuf continue;
38918328546SJosh Poimboeuf
39018328546SJosh Poimboeuf if (strcmp(sym1->demangled_name, sym2->demangled_name))
39118328546SJosh Poimboeuf continue;
39218328546SJosh Poimboeuf
39318328546SJosh Poimboeuf sym1->twin = sym2;
39418328546SJosh Poimboeuf sym2->twin = sym1;
39518328546SJosh Poimboeuf break;
39618328546SJosh Poimboeuf }
39718328546SJosh Poimboeuf }
39818328546SJosh Poimboeuf
39918328546SJosh Poimboeuf /* Correlate locals after the first FILE symbol */
40018328546SJosh Poimboeuf for (; ; file1_sym = next_file_symbol(e->orig, file1_sym),
401dd590d4dSJosh Poimboeuf file2_sym = next_file_symbol(e->patched, file2_sym)) {
402dd590d4dSJosh Poimboeuf
403dd590d4dSJosh Poimboeuf if (!file1_sym && file2_sym) {
404dd590d4dSJosh Poimboeuf ERROR("FILE symbol mismatch: NULL != %s", file2_sym->name);
405dd590d4dSJosh Poimboeuf return -1;
406dd590d4dSJosh Poimboeuf }
407dd590d4dSJosh Poimboeuf
408dd590d4dSJosh Poimboeuf if (file1_sym && !file2_sym) {
409dd590d4dSJosh Poimboeuf ERROR("FILE symbol mismatch: %s != NULL", file1_sym->name);
410dd590d4dSJosh Poimboeuf return -1;
411dd590d4dSJosh Poimboeuf }
412dd590d4dSJosh Poimboeuf
413dd590d4dSJosh Poimboeuf if (!file1_sym)
414dd590d4dSJosh Poimboeuf break;
415dd590d4dSJosh Poimboeuf
416dd590d4dSJosh Poimboeuf if (strcmp(file1_sym->name, file2_sym->name)) {
417dd590d4dSJosh Poimboeuf ERROR("FILE symbol mismatch: %s != %s", file1_sym->name, file2_sym->name);
418dd590d4dSJosh Poimboeuf return -1;
419dd590d4dSJosh Poimboeuf }
420dd590d4dSJosh Poimboeuf
421dd590d4dSJosh Poimboeuf file1_sym->twin = file2_sym;
422dd590d4dSJosh Poimboeuf file2_sym->twin = file1_sym;
423dd590d4dSJosh Poimboeuf
424dd590d4dSJosh Poimboeuf sym1 = file1_sym;
425dd590d4dSJosh Poimboeuf
426dd590d4dSJosh Poimboeuf for_each_sym_continue(e->orig, sym1) {
427dd590d4dSJosh Poimboeuf if (is_file_sym(sym1) || !is_local_sym(sym1))
428dd590d4dSJosh Poimboeuf break;
429dd590d4dSJosh Poimboeuf
430dd590d4dSJosh Poimboeuf if (dont_correlate(sym1))
431dd590d4dSJosh Poimboeuf continue;
432dd590d4dSJosh Poimboeuf
433dd590d4dSJosh Poimboeuf sym2 = file2_sym;
434dd590d4dSJosh Poimboeuf for_each_sym_continue(e->patched, sym2) {
435dd590d4dSJosh Poimboeuf if (is_file_sym(sym2) || !is_local_sym(sym2))
436dd590d4dSJosh Poimboeuf break;
437dd590d4dSJosh Poimboeuf
438dd590d4dSJosh Poimboeuf if (sym2->twin || dont_correlate(sym2))
439dd590d4dSJosh Poimboeuf continue;
440dd590d4dSJosh Poimboeuf
441dd590d4dSJosh Poimboeuf if (strcmp(sym1->demangled_name, sym2->demangled_name))
442dd590d4dSJosh Poimboeuf continue;
443dd590d4dSJosh Poimboeuf
444dd590d4dSJosh Poimboeuf sym1->twin = sym2;
445dd590d4dSJosh Poimboeuf sym2->twin = sym1;
446dd590d4dSJosh Poimboeuf break;
447dd590d4dSJosh Poimboeuf }
448dd590d4dSJosh Poimboeuf }
449dd590d4dSJosh Poimboeuf }
450dd590d4dSJosh Poimboeuf
451dd590d4dSJosh Poimboeuf /* Correlate globals */
452dd590d4dSJosh Poimboeuf for_each_sym(e->orig, sym1) {
453dd590d4dSJosh Poimboeuf if (sym1->bind == STB_LOCAL)
454dd590d4dSJosh Poimboeuf continue;
455dd590d4dSJosh Poimboeuf
456dd590d4dSJosh Poimboeuf sym2 = find_global_symbol_by_name(e->patched, sym1->name);
457dd590d4dSJosh Poimboeuf
458dd590d4dSJosh Poimboeuf if (sym2 && !sym2->twin && !strcmp(sym1->name, sym2->name)) {
459dd590d4dSJosh Poimboeuf sym1->twin = sym2;
460dd590d4dSJosh Poimboeuf sym2->twin = sym1;
461dd590d4dSJosh Poimboeuf }
462dd590d4dSJosh Poimboeuf }
463dd590d4dSJosh Poimboeuf
464dd590d4dSJosh Poimboeuf for_each_sym(e->orig, sym1) {
465dd590d4dSJosh Poimboeuf if (sym1->twin || dont_correlate(sym1))
466dd590d4dSJosh Poimboeuf continue;
467dd590d4dSJosh Poimboeuf WARN("no correlation: %s", sym1->name);
468dd590d4dSJosh Poimboeuf }
469dd590d4dSJosh Poimboeuf
470dd590d4dSJosh Poimboeuf return 0;
471dd590d4dSJosh Poimboeuf }
472dd590d4dSJosh Poimboeuf
473dd590d4dSJosh Poimboeuf /* "sympos" is used by livepatch to disambiguate duplicate symbol names */
find_sympos(struct elf * elf,struct symbol * sym)474dd590d4dSJosh Poimboeuf static unsigned long find_sympos(struct elf *elf, struct symbol *sym)
475dd590d4dSJosh Poimboeuf {
476dd590d4dSJosh Poimboeuf bool vmlinux = str_ends_with(objname, "vmlinux.o");
477dd590d4dSJosh Poimboeuf unsigned long sympos = 0, nr_matches = 0;
478dd590d4dSJosh Poimboeuf bool has_dup = false;
479dd590d4dSJosh Poimboeuf struct symbol *s;
480dd590d4dSJosh Poimboeuf
481dd590d4dSJosh Poimboeuf if (sym->bind != STB_LOCAL)
482dd590d4dSJosh Poimboeuf return 0;
483dd590d4dSJosh Poimboeuf
484dd590d4dSJosh Poimboeuf if (vmlinux && sym->type == STT_FUNC) {
485dd590d4dSJosh Poimboeuf /*
486dd590d4dSJosh Poimboeuf * HACK: Unfortunately, symbol ordering can differ between
487dd590d4dSJosh Poimboeuf * vmlinux.o and vmlinux due to the linker script emitting
488dd590d4dSJosh Poimboeuf * .text.unlikely* before .text*. Count .text.unlikely* first.
489dd590d4dSJosh Poimboeuf *
490dd590d4dSJosh Poimboeuf * TODO: Disambiguate symbols more reliably (checksums?)
491dd590d4dSJosh Poimboeuf */
492dd590d4dSJosh Poimboeuf for_each_sym(elf, s) {
493dd590d4dSJosh Poimboeuf if (strstarts(s->sec->name, ".text.unlikely") &&
494dd590d4dSJosh Poimboeuf !strcmp(s->name, sym->name)) {
495dd590d4dSJosh Poimboeuf nr_matches++;
496dd590d4dSJosh Poimboeuf if (s == sym)
497dd590d4dSJosh Poimboeuf sympos = nr_matches;
498dd590d4dSJosh Poimboeuf else
499dd590d4dSJosh Poimboeuf has_dup = true;
500dd590d4dSJosh Poimboeuf }
501dd590d4dSJosh Poimboeuf }
502dd590d4dSJosh Poimboeuf for_each_sym(elf, s) {
503dd590d4dSJosh Poimboeuf if (!strstarts(s->sec->name, ".text.unlikely") &&
504dd590d4dSJosh Poimboeuf !strcmp(s->name, sym->name)) {
505dd590d4dSJosh Poimboeuf nr_matches++;
506dd590d4dSJosh Poimboeuf if (s == sym)
507dd590d4dSJosh Poimboeuf sympos = nr_matches;
508dd590d4dSJosh Poimboeuf else
509dd590d4dSJosh Poimboeuf has_dup = true;
510dd590d4dSJosh Poimboeuf }
511dd590d4dSJosh Poimboeuf }
512dd590d4dSJosh Poimboeuf } else {
513dd590d4dSJosh Poimboeuf for_each_sym(elf, s) {
514dd590d4dSJosh Poimboeuf if (!strcmp(s->name, sym->name)) {
515dd590d4dSJosh Poimboeuf nr_matches++;
516dd590d4dSJosh Poimboeuf if (s == sym)
517dd590d4dSJosh Poimboeuf sympos = nr_matches;
518dd590d4dSJosh Poimboeuf else
519dd590d4dSJosh Poimboeuf has_dup = true;
520dd590d4dSJosh Poimboeuf }
521dd590d4dSJosh Poimboeuf }
522dd590d4dSJosh Poimboeuf }
523dd590d4dSJosh Poimboeuf
524dd590d4dSJosh Poimboeuf if (!sympos) {
525dd590d4dSJosh Poimboeuf ERROR("can't find sympos for %s", sym->name);
526dd590d4dSJosh Poimboeuf return ULONG_MAX;
527dd590d4dSJosh Poimboeuf }
528dd590d4dSJosh Poimboeuf
529dd590d4dSJosh Poimboeuf return has_dup ? sympos : 0;
530dd590d4dSJosh Poimboeuf }
531dd590d4dSJosh Poimboeuf
532dd590d4dSJosh Poimboeuf static int clone_sym_relocs(struct elfs *e, struct symbol *patched_sym);
533dd590d4dSJosh Poimboeuf
__clone_symbol(struct elf * elf,struct symbol * patched_sym,bool data_too)534dd590d4dSJosh Poimboeuf static struct symbol *__clone_symbol(struct elf *elf, struct symbol *patched_sym,
535dd590d4dSJosh Poimboeuf bool data_too)
536dd590d4dSJosh Poimboeuf {
537dd590d4dSJosh Poimboeuf struct section *out_sec = NULL;
538dd590d4dSJosh Poimboeuf unsigned long offset = 0;
539dd590d4dSJosh Poimboeuf struct symbol *out_sym;
540dd590d4dSJosh Poimboeuf
541dd590d4dSJosh Poimboeuf if (data_too && !is_undef_sym(patched_sym)) {
542dd590d4dSJosh Poimboeuf struct section *patched_sec = patched_sym->sec;
543dd590d4dSJosh Poimboeuf
544dd590d4dSJosh Poimboeuf out_sec = find_section_by_name(elf, patched_sec->name);
545dd590d4dSJosh Poimboeuf if (!out_sec) {
546dd590d4dSJosh Poimboeuf out_sec = elf_create_section(elf, patched_sec->name, 0,
547dd590d4dSJosh Poimboeuf patched_sec->sh.sh_entsize,
548dd590d4dSJosh Poimboeuf patched_sec->sh.sh_type,
549dd590d4dSJosh Poimboeuf patched_sec->sh.sh_addralign,
550dd590d4dSJosh Poimboeuf patched_sec->sh.sh_flags);
551dd590d4dSJosh Poimboeuf if (!out_sec)
552dd590d4dSJosh Poimboeuf return NULL;
553dd590d4dSJosh Poimboeuf }
554dd590d4dSJosh Poimboeuf
555dd590d4dSJosh Poimboeuf if (is_string_sec(patched_sym->sec)) {
556dd590d4dSJosh Poimboeuf out_sym = elf_create_section_symbol(elf, out_sec);
557dd590d4dSJosh Poimboeuf if (!out_sym)
558dd590d4dSJosh Poimboeuf return NULL;
559dd590d4dSJosh Poimboeuf
560dd590d4dSJosh Poimboeuf goto sym_created;
561dd590d4dSJosh Poimboeuf }
562dd590d4dSJosh Poimboeuf
563dd590d4dSJosh Poimboeuf if (!is_sec_sym(patched_sym))
564*2f2600deSJoe Lawrence offset = ALIGN(sec_size(out_sec), out_sec->sh.sh_addralign);
565dd590d4dSJosh Poimboeuf
566dd590d4dSJosh Poimboeuf if (patched_sym->len || is_sec_sym(patched_sym)) {
567dd590d4dSJosh Poimboeuf void *data = NULL;
568dd590d4dSJosh Poimboeuf size_t size;
569dd590d4dSJosh Poimboeuf
570dd590d4dSJosh Poimboeuf /* bss doesn't have data */
571dd590d4dSJosh Poimboeuf if (patched_sym->sec->data->d_buf)
572dd590d4dSJosh Poimboeuf data = patched_sym->sec->data->d_buf + patched_sym->offset;
573dd590d4dSJosh Poimboeuf
574dd590d4dSJosh Poimboeuf if (is_sec_sym(patched_sym))
575dd590d4dSJosh Poimboeuf size = sec_size(patched_sym->sec);
576dd590d4dSJosh Poimboeuf else
577dd590d4dSJosh Poimboeuf size = patched_sym->len;
578dd590d4dSJosh Poimboeuf
579dd590d4dSJosh Poimboeuf if (!elf_add_data(elf, out_sec, data, size))
580dd590d4dSJosh Poimboeuf return NULL;
581dd590d4dSJosh Poimboeuf }
582dd590d4dSJosh Poimboeuf }
583dd590d4dSJosh Poimboeuf
584dd590d4dSJosh Poimboeuf out_sym = elf_create_symbol(elf, patched_sym->name, out_sec,
585dd590d4dSJosh Poimboeuf patched_sym->bind, patched_sym->type,
586dd590d4dSJosh Poimboeuf offset, patched_sym->len);
587dd590d4dSJosh Poimboeuf if (!out_sym)
588dd590d4dSJosh Poimboeuf return NULL;
589dd590d4dSJosh Poimboeuf
590dd590d4dSJosh Poimboeuf sym_created:
591dd590d4dSJosh Poimboeuf patched_sym->clone = out_sym;
592dd590d4dSJosh Poimboeuf out_sym->clone = patched_sym;
593dd590d4dSJosh Poimboeuf
594dd590d4dSJosh Poimboeuf return out_sym;
595dd590d4dSJosh Poimboeuf }
596dd590d4dSJosh Poimboeuf
sym_type(struct symbol * sym)5977c2575a6SJosh Poimboeuf static const char *sym_type(struct symbol *sym)
5987c2575a6SJosh Poimboeuf {
5997c2575a6SJosh Poimboeuf switch (sym->type) {
6007c2575a6SJosh Poimboeuf case STT_NOTYPE: return "NOTYPE";
6017c2575a6SJosh Poimboeuf case STT_OBJECT: return "OBJECT";
6027c2575a6SJosh Poimboeuf case STT_FUNC: return "FUNC";
6037c2575a6SJosh Poimboeuf case STT_SECTION: return "SECTION";
6047c2575a6SJosh Poimboeuf case STT_FILE: return "FILE";
6057c2575a6SJosh Poimboeuf default: return "UNKNOWN";
6067c2575a6SJosh Poimboeuf }
6077c2575a6SJosh Poimboeuf }
6087c2575a6SJosh Poimboeuf
sym_bind(struct symbol * sym)6097c2575a6SJosh Poimboeuf static const char *sym_bind(struct symbol *sym)
6107c2575a6SJosh Poimboeuf {
6117c2575a6SJosh Poimboeuf switch (sym->bind) {
6127c2575a6SJosh Poimboeuf case STB_LOCAL: return "LOCAL";
6137c2575a6SJosh Poimboeuf case STB_GLOBAL: return "GLOBAL";
6147c2575a6SJosh Poimboeuf case STB_WEAK: return "WEAK";
6157c2575a6SJosh Poimboeuf default: return "UNKNOWN";
6167c2575a6SJosh Poimboeuf }
6177c2575a6SJosh Poimboeuf }
6187c2575a6SJosh Poimboeuf
619dd590d4dSJosh Poimboeuf /*
620dd590d4dSJosh Poimboeuf * Copy a symbol to the output object, optionally including its data and
621dd590d4dSJosh Poimboeuf * relocations.
622dd590d4dSJosh Poimboeuf */
clone_symbol(struct elfs * e,struct symbol * patched_sym,bool data_too)623dd590d4dSJosh Poimboeuf static struct symbol *clone_symbol(struct elfs *e, struct symbol *patched_sym,
624dd590d4dSJosh Poimboeuf bool data_too)
625dd590d4dSJosh Poimboeuf {
626dd590d4dSJosh Poimboeuf struct symbol *pfx;
627dd590d4dSJosh Poimboeuf
628dd590d4dSJosh Poimboeuf if (patched_sym->clone)
629dd590d4dSJosh Poimboeuf return patched_sym->clone;
630dd590d4dSJosh Poimboeuf
6317c2575a6SJosh Poimboeuf dbg_indent("%s%s", patched_sym->name, data_too ? " [+DATA]" : "");
6327c2575a6SJosh Poimboeuf
633dd590d4dSJosh Poimboeuf /* Make sure the prefix gets cloned first */
634dd590d4dSJosh Poimboeuf if (is_func_sym(patched_sym) && data_too) {
635dd590d4dSJosh Poimboeuf pfx = get_func_prefix(patched_sym);
636dd590d4dSJosh Poimboeuf if (pfx)
637dd590d4dSJosh Poimboeuf clone_symbol(e, pfx, true);
638dd590d4dSJosh Poimboeuf }
639dd590d4dSJosh Poimboeuf
640dd590d4dSJosh Poimboeuf if (!__clone_symbol(e->out, patched_sym, data_too))
641dd590d4dSJosh Poimboeuf return NULL;
642dd590d4dSJosh Poimboeuf
643dd590d4dSJosh Poimboeuf if (data_too && clone_sym_relocs(e, patched_sym))
644dd590d4dSJosh Poimboeuf return NULL;
645dd590d4dSJosh Poimboeuf
646dd590d4dSJosh Poimboeuf return patched_sym->clone;
647dd590d4dSJosh Poimboeuf }
648dd590d4dSJosh Poimboeuf
mark_included_function(struct symbol * func)649dd590d4dSJosh Poimboeuf static void mark_included_function(struct symbol *func)
650dd590d4dSJosh Poimboeuf {
651dd590d4dSJosh Poimboeuf struct symbol *pfx;
652dd590d4dSJosh Poimboeuf
653dd590d4dSJosh Poimboeuf func->included = 1;
654dd590d4dSJosh Poimboeuf
655dd590d4dSJosh Poimboeuf /* Include prefix function */
656dd590d4dSJosh Poimboeuf pfx = get_func_prefix(func);
657dd590d4dSJosh Poimboeuf if (pfx)
658dd590d4dSJosh Poimboeuf pfx->included = 1;
659dd590d4dSJosh Poimboeuf
660dd590d4dSJosh Poimboeuf /* Make sure .cold parent+child always stay together */
661dd590d4dSJosh Poimboeuf if (func->cfunc && func->cfunc != func)
662dd590d4dSJosh Poimboeuf func->cfunc->included = 1;
663dd590d4dSJosh Poimboeuf if (func->pfunc && func->pfunc != func)
664dd590d4dSJosh Poimboeuf func->pfunc->included = 1;
665dd590d4dSJosh Poimboeuf }
666dd590d4dSJosh Poimboeuf
667dd590d4dSJosh Poimboeuf /*
668dd590d4dSJosh Poimboeuf * Copy all changed functions (and their dependencies) from the patched object
669dd590d4dSJosh Poimboeuf * to the output object.
670dd590d4dSJosh Poimboeuf */
mark_changed_functions(struct elfs * e)671dd590d4dSJosh Poimboeuf static int mark_changed_functions(struct elfs *e)
672dd590d4dSJosh Poimboeuf {
673dd590d4dSJosh Poimboeuf struct symbol *sym_orig, *patched_sym;
674dd590d4dSJosh Poimboeuf bool changed = false;
675dd590d4dSJosh Poimboeuf
676dd590d4dSJosh Poimboeuf /* Find changed functions */
677dd590d4dSJosh Poimboeuf for_each_sym(e->orig, sym_orig) {
678dd590d4dSJosh Poimboeuf if (!is_func_sym(sym_orig) || is_prefix_func(sym_orig))
679dd590d4dSJosh Poimboeuf continue;
680dd590d4dSJosh Poimboeuf
681dd590d4dSJosh Poimboeuf patched_sym = sym_orig->twin;
682dd590d4dSJosh Poimboeuf if (!patched_sym)
683dd590d4dSJosh Poimboeuf continue;
684dd590d4dSJosh Poimboeuf
685dd590d4dSJosh Poimboeuf if (sym_orig->csum.checksum != patched_sym->csum.checksum) {
686dd590d4dSJosh Poimboeuf patched_sym->changed = 1;
687dd590d4dSJosh Poimboeuf mark_included_function(patched_sym);
688dd590d4dSJosh Poimboeuf changed = true;
689dd590d4dSJosh Poimboeuf }
690dd590d4dSJosh Poimboeuf }
691dd590d4dSJosh Poimboeuf
692dd590d4dSJosh Poimboeuf /* Find added functions and print them */
693dd590d4dSJosh Poimboeuf for_each_sym(e->patched, patched_sym) {
694dd590d4dSJosh Poimboeuf if (!is_func_sym(patched_sym) || is_prefix_func(patched_sym))
695dd590d4dSJosh Poimboeuf continue;
696dd590d4dSJosh Poimboeuf
697dd590d4dSJosh Poimboeuf if (!patched_sym->twin) {
698dd590d4dSJosh Poimboeuf printf("%s: new function: %s\n", objname, patched_sym->name);
699dd590d4dSJosh Poimboeuf mark_included_function(patched_sym);
700dd590d4dSJosh Poimboeuf changed = true;
701dd590d4dSJosh Poimboeuf }
702dd590d4dSJosh Poimboeuf }
703dd590d4dSJosh Poimboeuf
704dd590d4dSJosh Poimboeuf /* Print changed functions */
705dd590d4dSJosh Poimboeuf for_each_sym(e->patched, patched_sym) {
706dd590d4dSJosh Poimboeuf if (patched_sym->changed)
707dd590d4dSJosh Poimboeuf printf("%s: changed function: %s\n", objname, patched_sym->name);
708dd590d4dSJosh Poimboeuf }
709dd590d4dSJosh Poimboeuf
710dd590d4dSJosh Poimboeuf return !changed ? -1 : 0;
711dd590d4dSJosh Poimboeuf }
712dd590d4dSJosh Poimboeuf
clone_included_functions(struct elfs * e)713dd590d4dSJosh Poimboeuf static int clone_included_functions(struct elfs *e)
714dd590d4dSJosh Poimboeuf {
715dd590d4dSJosh Poimboeuf struct symbol *patched_sym;
716dd590d4dSJosh Poimboeuf
717dd590d4dSJosh Poimboeuf for_each_sym(e->patched, patched_sym) {
718dd590d4dSJosh Poimboeuf if (patched_sym->included) {
719dd590d4dSJosh Poimboeuf if (!clone_symbol(e, patched_sym, true))
720dd590d4dSJosh Poimboeuf return -1;
721dd590d4dSJosh Poimboeuf }
722dd590d4dSJosh Poimboeuf }
723dd590d4dSJosh Poimboeuf
724dd590d4dSJosh Poimboeuf return 0;
725dd590d4dSJosh Poimboeuf }
726dd590d4dSJosh Poimboeuf
727dd590d4dSJosh Poimboeuf /*
728dd590d4dSJosh Poimboeuf * Determine whether a relocation should reference the section rather than the
729dd590d4dSJosh Poimboeuf * underlying symbol.
730dd590d4dSJosh Poimboeuf */
section_reference_needed(struct section * sec)731dd590d4dSJosh Poimboeuf static bool section_reference_needed(struct section *sec)
732dd590d4dSJosh Poimboeuf {
733dd590d4dSJosh Poimboeuf /*
734dd590d4dSJosh Poimboeuf * String symbols are zero-length and uncorrelated. It's easier to
735dd590d4dSJosh Poimboeuf * deal with them as section symbols.
736dd590d4dSJosh Poimboeuf */
737dd590d4dSJosh Poimboeuf if (is_string_sec(sec))
738dd590d4dSJosh Poimboeuf return true;
739dd590d4dSJosh Poimboeuf
740dd590d4dSJosh Poimboeuf /*
741dd590d4dSJosh Poimboeuf * .rodata has mostly anonymous data so there's no way to determine the
742dd590d4dSJosh Poimboeuf * length of a needed reference. just copy the whole section if needed.
743dd590d4dSJosh Poimboeuf */
744dd590d4dSJosh Poimboeuf if (strstarts(sec->name, ".rodata"))
745dd590d4dSJosh Poimboeuf return true;
746dd590d4dSJosh Poimboeuf
747dd590d4dSJosh Poimboeuf /* UBSAN anonymous data */
748dd590d4dSJosh Poimboeuf if (strstarts(sec->name, ".data..Lubsan") || /* GCC */
749dd590d4dSJosh Poimboeuf strstarts(sec->name, ".data..L__unnamed_")) /* Clang */
750dd590d4dSJosh Poimboeuf return true;
751dd590d4dSJosh Poimboeuf
752dd590d4dSJosh Poimboeuf return false;
753dd590d4dSJosh Poimboeuf }
754dd590d4dSJosh Poimboeuf
is_reloc_allowed(struct reloc * reloc)755dd590d4dSJosh Poimboeuf static bool is_reloc_allowed(struct reloc *reloc)
756dd590d4dSJosh Poimboeuf {
757dd590d4dSJosh Poimboeuf return section_reference_needed(reloc->sym->sec) == is_sec_sym(reloc->sym);
758dd590d4dSJosh Poimboeuf }
759dd590d4dSJosh Poimboeuf
find_export(struct symbol * sym)760dd590d4dSJosh Poimboeuf static struct export *find_export(struct symbol *sym)
761dd590d4dSJosh Poimboeuf {
762dd590d4dSJosh Poimboeuf struct export *export;
763dd590d4dSJosh Poimboeuf
764dd590d4dSJosh Poimboeuf hash_for_each_possible(exports, export, hash, str_hash(sym->name)) {
765dd590d4dSJosh Poimboeuf if (!strcmp(export->sym, sym->name))
766dd590d4dSJosh Poimboeuf return export;
767dd590d4dSJosh Poimboeuf }
768dd590d4dSJosh Poimboeuf
769dd590d4dSJosh Poimboeuf return NULL;
770dd590d4dSJosh Poimboeuf }
771dd590d4dSJosh Poimboeuf
__find_modname(struct elfs * e)772dd590d4dSJosh Poimboeuf static const char *__find_modname(struct elfs *e)
773dd590d4dSJosh Poimboeuf {
774dd590d4dSJosh Poimboeuf struct section *sec;
775dd590d4dSJosh Poimboeuf char *name;
776dd590d4dSJosh Poimboeuf
777dd590d4dSJosh Poimboeuf sec = find_section_by_name(e->orig, ".modinfo");
778dd590d4dSJosh Poimboeuf if (!sec) {
779dd590d4dSJosh Poimboeuf ERROR("missing .modinfo section");
780dd590d4dSJosh Poimboeuf return NULL;
781dd590d4dSJosh Poimboeuf }
782dd590d4dSJosh Poimboeuf
783dd590d4dSJosh Poimboeuf name = memmem(sec->data->d_buf, sec_size(sec), "\0name=", 6);
784dd590d4dSJosh Poimboeuf if (name)
785dd590d4dSJosh Poimboeuf return name + 6;
786dd590d4dSJosh Poimboeuf
787dd590d4dSJosh Poimboeuf name = strdup(e->orig->name);
788dd590d4dSJosh Poimboeuf if (!name) {
789dd590d4dSJosh Poimboeuf ERROR_GLIBC("strdup");
790dd590d4dSJosh Poimboeuf return NULL;
791dd590d4dSJosh Poimboeuf }
792dd590d4dSJosh Poimboeuf
793dd590d4dSJosh Poimboeuf for (char *c = name; *c; c++) {
794dd590d4dSJosh Poimboeuf if (*c == '/')
795dd590d4dSJosh Poimboeuf name = c + 1;
796dd590d4dSJosh Poimboeuf else if (*c == '-')
797dd590d4dSJosh Poimboeuf *c = '_';
798dd590d4dSJosh Poimboeuf else if (*c == '.') {
799dd590d4dSJosh Poimboeuf *c = '\0';
800dd590d4dSJosh Poimboeuf break;
801dd590d4dSJosh Poimboeuf }
802dd590d4dSJosh Poimboeuf }
803dd590d4dSJosh Poimboeuf
804dd590d4dSJosh Poimboeuf return name;
805dd590d4dSJosh Poimboeuf }
806dd590d4dSJosh Poimboeuf
807dd590d4dSJosh Poimboeuf /* Get the object's module name as defined by the kernel (and klp_object) */
find_modname(struct elfs * e)808dd590d4dSJosh Poimboeuf static const char *find_modname(struct elfs *e)
809dd590d4dSJosh Poimboeuf {
810dd590d4dSJosh Poimboeuf const char *modname;
811dd590d4dSJosh Poimboeuf
812dd590d4dSJosh Poimboeuf if (e->modname)
813dd590d4dSJosh Poimboeuf return e->modname;
814dd590d4dSJosh Poimboeuf
815dd590d4dSJosh Poimboeuf modname = __find_modname(e);
816dd590d4dSJosh Poimboeuf e->modname = modname;
817dd590d4dSJosh Poimboeuf return modname;
818dd590d4dSJosh Poimboeuf }
819dd590d4dSJosh Poimboeuf
820dd590d4dSJosh Poimboeuf /*
821dd590d4dSJosh Poimboeuf * Copying a function from its native compiled environment to a kernel module
822dd590d4dSJosh Poimboeuf * removes its natural access to local functions/variables and unexported
823dd590d4dSJosh Poimboeuf * globals. References to such symbols need to be converted to KLP relocs so
824dd590d4dSJosh Poimboeuf * the kernel arch relocation code knows to apply them and where to find the
825dd590d4dSJosh Poimboeuf * symbols. Particularly, duplicate static symbols need to be disambiguated.
826dd590d4dSJosh Poimboeuf */
klp_reloc_needed(struct reloc * patched_reloc)827dd590d4dSJosh Poimboeuf static bool klp_reloc_needed(struct reloc *patched_reloc)
828dd590d4dSJosh Poimboeuf {
829dd590d4dSJosh Poimboeuf struct symbol *patched_sym = patched_reloc->sym;
830dd590d4dSJosh Poimboeuf struct export *export;
831dd590d4dSJosh Poimboeuf
832dd590d4dSJosh Poimboeuf /* no external symbol to reference */
833dd590d4dSJosh Poimboeuf if (dont_correlate(patched_sym))
834dd590d4dSJosh Poimboeuf return false;
835dd590d4dSJosh Poimboeuf
836dd590d4dSJosh Poimboeuf /* For included functions, a regular reloc will do. */
837dd590d4dSJosh Poimboeuf if (patched_sym->included)
838dd590d4dSJosh Poimboeuf return false;
839dd590d4dSJosh Poimboeuf
840dd590d4dSJosh Poimboeuf /*
841dd590d4dSJosh Poimboeuf * If exported by a module, it has to be a klp reloc. Thanks to the
842dd590d4dSJosh Poimboeuf * clusterfunk that is late module patching, the patch module is
843dd590d4dSJosh Poimboeuf * allowed to be loaded before any modules it depends on.
844dd590d4dSJosh Poimboeuf *
845dd590d4dSJosh Poimboeuf * If exported by vmlinux, a normal reloc will do.
846dd590d4dSJosh Poimboeuf */
847dd590d4dSJosh Poimboeuf export = find_export(patched_sym);
848dd590d4dSJosh Poimboeuf if (export)
849dd590d4dSJosh Poimboeuf return strcmp(export->mod, "vmlinux");
850dd590d4dSJosh Poimboeuf
851dd590d4dSJosh Poimboeuf if (!patched_sym->twin) {
852dd590d4dSJosh Poimboeuf /*
853dd590d4dSJosh Poimboeuf * Presumably the symbol and its reference were added by the
854dd590d4dSJosh Poimboeuf * patch. The symbol could be defined in this .o or in another
855dd590d4dSJosh Poimboeuf * .o in the patch module.
856dd590d4dSJosh Poimboeuf *
857dd590d4dSJosh Poimboeuf * This check needs to be *after* the export check due to the
858dd590d4dSJosh Poimboeuf * possibility of the patch adding a new UNDEF reference to an
859dd590d4dSJosh Poimboeuf * exported symbol.
860dd590d4dSJosh Poimboeuf */
861dd590d4dSJosh Poimboeuf return false;
862dd590d4dSJosh Poimboeuf }
863dd590d4dSJosh Poimboeuf
864dd590d4dSJosh Poimboeuf /* Unexported symbol which lives in the original vmlinux or module. */
865dd590d4dSJosh Poimboeuf return true;
866dd590d4dSJosh Poimboeuf }
867dd590d4dSJosh Poimboeuf
convert_reloc_sym_to_secsym(struct elf * elf,struct reloc * reloc)868dd590d4dSJosh Poimboeuf static int convert_reloc_sym_to_secsym(struct elf *elf, struct reloc *reloc)
869dd590d4dSJosh Poimboeuf {
870dd590d4dSJosh Poimboeuf struct symbol *sym = reloc->sym;
871dd590d4dSJosh Poimboeuf struct section *sec = sym->sec;
872dd590d4dSJosh Poimboeuf
873dd590d4dSJosh Poimboeuf if (!sec->sym && !elf_create_section_symbol(elf, sec))
874dd590d4dSJosh Poimboeuf return -1;
875dd590d4dSJosh Poimboeuf
876dd590d4dSJosh Poimboeuf reloc->sym = sec->sym;
877dd590d4dSJosh Poimboeuf set_reloc_sym(elf, reloc, sym->idx);
878dd590d4dSJosh Poimboeuf set_reloc_addend(elf, reloc, sym->offset + reloc_addend(reloc));
879dd590d4dSJosh Poimboeuf return 0;
880dd590d4dSJosh Poimboeuf }
881dd590d4dSJosh Poimboeuf
convert_reloc_secsym_to_sym(struct elf * elf,struct reloc * reloc)882dd590d4dSJosh Poimboeuf static int convert_reloc_secsym_to_sym(struct elf *elf, struct reloc *reloc)
883dd590d4dSJosh Poimboeuf {
884dd590d4dSJosh Poimboeuf struct symbol *sym = reloc->sym;
885dd590d4dSJosh Poimboeuf struct section *sec = sym->sec;
886dd590d4dSJosh Poimboeuf
887dd590d4dSJosh Poimboeuf /* If the symbol has a dedicated section, it's easy to find */
888dd590d4dSJosh Poimboeuf sym = find_symbol_by_offset(sec, 0);
889dd590d4dSJosh Poimboeuf if (sym && sym->len == sec_size(sec))
890dd590d4dSJosh Poimboeuf goto found_sym;
891dd590d4dSJosh Poimboeuf
892dd590d4dSJosh Poimboeuf /* No dedicated section; find the symbol manually */
893dd590d4dSJosh Poimboeuf sym = find_symbol_containing(sec, arch_adjusted_addend(reloc));
894dd590d4dSJosh Poimboeuf if (!sym) {
895dd590d4dSJosh Poimboeuf /*
896dd590d4dSJosh Poimboeuf * This can happen for special section references to weak code
897dd590d4dSJosh Poimboeuf * whose symbol has been stripped by the linker.
898dd590d4dSJosh Poimboeuf */
899dd590d4dSJosh Poimboeuf return -1;
900dd590d4dSJosh Poimboeuf }
901dd590d4dSJosh Poimboeuf
902dd590d4dSJosh Poimboeuf found_sym:
903dd590d4dSJosh Poimboeuf reloc->sym = sym;
904dd590d4dSJosh Poimboeuf set_reloc_sym(elf, reloc, sym->idx);
905dd590d4dSJosh Poimboeuf set_reloc_addend(elf, reloc, reloc_addend(reloc) - sym->offset);
906dd590d4dSJosh Poimboeuf return 0;
907dd590d4dSJosh Poimboeuf }
908dd590d4dSJosh Poimboeuf
909dd590d4dSJosh Poimboeuf /*
910dd590d4dSJosh Poimboeuf * Convert a relocation symbol reference to the needed format: either a section
911dd590d4dSJosh Poimboeuf * symbol or the underlying symbol itself.
912dd590d4dSJosh Poimboeuf */
convert_reloc_sym(struct elf * elf,struct reloc * reloc)913dd590d4dSJosh Poimboeuf static int convert_reloc_sym(struct elf *elf, struct reloc *reloc)
914dd590d4dSJosh Poimboeuf {
915dd590d4dSJosh Poimboeuf if (is_reloc_allowed(reloc))
916dd590d4dSJosh Poimboeuf return 0;
917dd590d4dSJosh Poimboeuf
918dd590d4dSJosh Poimboeuf if (section_reference_needed(reloc->sym->sec))
919dd590d4dSJosh Poimboeuf return convert_reloc_sym_to_secsym(elf, reloc);
920dd590d4dSJosh Poimboeuf else
921dd590d4dSJosh Poimboeuf return convert_reloc_secsym_to_sym(elf, reloc);
922dd590d4dSJosh Poimboeuf }
923dd590d4dSJosh Poimboeuf
924dd590d4dSJosh Poimboeuf /*
925dd590d4dSJosh Poimboeuf * Convert a regular relocation to a klp relocation (sort of).
926dd590d4dSJosh Poimboeuf */
clone_reloc_klp(struct elfs * e,struct reloc * patched_reloc,struct section * sec,unsigned long offset,struct export * export)927dd590d4dSJosh Poimboeuf static int clone_reloc_klp(struct elfs *e, struct reloc *patched_reloc,
928dd590d4dSJosh Poimboeuf struct section *sec, unsigned long offset,
929dd590d4dSJosh Poimboeuf struct export *export)
930dd590d4dSJosh Poimboeuf {
931dd590d4dSJosh Poimboeuf struct symbol *patched_sym = patched_reloc->sym;
932dd590d4dSJosh Poimboeuf s64 addend = reloc_addend(patched_reloc);
933dd590d4dSJosh Poimboeuf const char *sym_modname, *sym_orig_name;
934dd590d4dSJosh Poimboeuf static struct section *klp_relocs;
935dd590d4dSJosh Poimboeuf struct symbol *sym, *klp_sym;
936dd590d4dSJosh Poimboeuf unsigned long klp_reloc_off;
937dd590d4dSJosh Poimboeuf char sym_name[SYM_NAME_LEN];
938dd590d4dSJosh Poimboeuf struct klp_reloc klp_reloc;
939dd590d4dSJosh Poimboeuf unsigned long sympos;
940dd590d4dSJosh Poimboeuf
941dd590d4dSJosh Poimboeuf if (!patched_sym->twin) {
942dd590d4dSJosh Poimboeuf ERROR("unexpected klp reloc for new symbol %s", patched_sym->name);
943dd590d4dSJosh Poimboeuf return -1;
944dd590d4dSJosh Poimboeuf }
945dd590d4dSJosh Poimboeuf
946dd590d4dSJosh Poimboeuf /*
947dd590d4dSJosh Poimboeuf * Keep the original reloc intact for now to avoid breaking objtool run
948dd590d4dSJosh Poimboeuf * which relies on proper relocations for many of its features. This
949dd590d4dSJosh Poimboeuf * will be disabled later by "objtool klp post-link".
950dd590d4dSJosh Poimboeuf *
951dd590d4dSJosh Poimboeuf * Convert it to UNDEF (and WEAK to avoid modpost warnings).
952dd590d4dSJosh Poimboeuf */
953dd590d4dSJosh Poimboeuf
954dd590d4dSJosh Poimboeuf sym = patched_sym->clone;
955dd590d4dSJosh Poimboeuf if (!sym) {
956dd590d4dSJosh Poimboeuf /* STB_WEAK: avoid modpost undefined symbol warnings */
957dd590d4dSJosh Poimboeuf sym = elf_create_symbol(e->out, patched_sym->name, NULL,
958dd590d4dSJosh Poimboeuf STB_WEAK, patched_sym->type, 0, 0);
959dd590d4dSJosh Poimboeuf if (!sym)
960dd590d4dSJosh Poimboeuf return -1;
961dd590d4dSJosh Poimboeuf
962dd590d4dSJosh Poimboeuf patched_sym->clone = sym;
963dd590d4dSJosh Poimboeuf sym->clone = patched_sym;
964dd590d4dSJosh Poimboeuf }
965dd590d4dSJosh Poimboeuf
966dd590d4dSJosh Poimboeuf if (!elf_create_reloc(e->out, sec, offset, sym, addend, reloc_type(patched_reloc)))
967dd590d4dSJosh Poimboeuf return -1;
968dd590d4dSJosh Poimboeuf
969dd590d4dSJosh Poimboeuf /*
970dd590d4dSJosh Poimboeuf * Create the KLP symbol.
971dd590d4dSJosh Poimboeuf */
972dd590d4dSJosh Poimboeuf
973dd590d4dSJosh Poimboeuf if (export) {
974dd590d4dSJosh Poimboeuf sym_modname = export->mod;
975dd590d4dSJosh Poimboeuf sym_orig_name = export->sym;
976dd590d4dSJosh Poimboeuf sympos = 0;
977dd590d4dSJosh Poimboeuf } else {
978dd590d4dSJosh Poimboeuf sym_modname = find_modname(e);
979dd590d4dSJosh Poimboeuf if (!sym_modname)
980dd590d4dSJosh Poimboeuf return -1;
981dd590d4dSJosh Poimboeuf
982dd590d4dSJosh Poimboeuf sym_orig_name = patched_sym->twin->name;
983dd590d4dSJosh Poimboeuf sympos = find_sympos(e->orig, patched_sym->twin);
984dd590d4dSJosh Poimboeuf if (sympos == ULONG_MAX)
985dd590d4dSJosh Poimboeuf return -1;
986dd590d4dSJosh Poimboeuf }
987dd590d4dSJosh Poimboeuf
988dd590d4dSJosh Poimboeuf /* symbol format: .klp.sym.modname.sym_name,sympos */
989dd590d4dSJosh Poimboeuf if (snprintf_check(sym_name, SYM_NAME_LEN, KLP_SYM_PREFIX "%s.%s,%ld",
990dd590d4dSJosh Poimboeuf sym_modname, sym_orig_name, sympos))
991dd590d4dSJosh Poimboeuf return -1;
992dd590d4dSJosh Poimboeuf
993dd590d4dSJosh Poimboeuf klp_sym = find_symbol_by_name(e->out, sym_name);
994dd590d4dSJosh Poimboeuf if (!klp_sym) {
9957c2575a6SJosh Poimboeuf __dbg_indent("%s", sym_name);
9967c2575a6SJosh Poimboeuf
997dd590d4dSJosh Poimboeuf /* STB_WEAK: avoid modpost undefined symbol warnings */
998dd590d4dSJosh Poimboeuf klp_sym = elf_create_symbol(e->out, sym_name, NULL,
999dd590d4dSJosh Poimboeuf STB_WEAK, patched_sym->type, 0, 0);
1000dd590d4dSJosh Poimboeuf if (!klp_sym)
1001dd590d4dSJosh Poimboeuf return -1;
1002dd590d4dSJosh Poimboeuf }
1003dd590d4dSJosh Poimboeuf
1004dd590d4dSJosh Poimboeuf /*
1005dd590d4dSJosh Poimboeuf * Create the __klp_relocs entry. This will be converted to an actual
1006dd590d4dSJosh Poimboeuf * KLP rela by "objtool klp post-link".
1007dd590d4dSJosh Poimboeuf *
1008dd590d4dSJosh Poimboeuf * This intermediate step is necessary to prevent corruption by the
1009dd590d4dSJosh Poimboeuf * linker, which doesn't know how to properly handle two rela sections
1010dd590d4dSJosh Poimboeuf * applying to the same base section.
1011dd590d4dSJosh Poimboeuf */
1012dd590d4dSJosh Poimboeuf
1013dd590d4dSJosh Poimboeuf if (!klp_relocs) {
1014dd590d4dSJosh Poimboeuf klp_relocs = elf_create_section(e->out, KLP_RELOCS_SEC, 0,
1015dd590d4dSJosh Poimboeuf 0, SHT_PROGBITS, 8, SHF_ALLOC);
1016dd590d4dSJosh Poimboeuf if (!klp_relocs)
1017dd590d4dSJosh Poimboeuf return -1;
1018dd590d4dSJosh Poimboeuf }
1019dd590d4dSJosh Poimboeuf
1020dd590d4dSJosh Poimboeuf klp_reloc_off = sec_size(klp_relocs);
1021dd590d4dSJosh Poimboeuf memset(&klp_reloc, 0, sizeof(klp_reloc));
1022dd590d4dSJosh Poimboeuf
1023dd590d4dSJosh Poimboeuf klp_reloc.type = reloc_type(patched_reloc);
1024dd590d4dSJosh Poimboeuf if (!elf_add_data(e->out, klp_relocs, &klp_reloc, sizeof(klp_reloc)))
1025dd590d4dSJosh Poimboeuf return -1;
1026dd590d4dSJosh Poimboeuf
1027dd590d4dSJosh Poimboeuf /* klp_reloc.offset */
1028dd590d4dSJosh Poimboeuf if (!sec->sym && !elf_create_section_symbol(e->out, sec))
1029dd590d4dSJosh Poimboeuf return -1;
1030dd590d4dSJosh Poimboeuf
1031dd590d4dSJosh Poimboeuf if (!elf_create_reloc(e->out, klp_relocs,
1032dd590d4dSJosh Poimboeuf klp_reloc_off + offsetof(struct klp_reloc, offset),
1033dd590d4dSJosh Poimboeuf sec->sym, offset, R_ABS64))
1034dd590d4dSJosh Poimboeuf return -1;
1035dd590d4dSJosh Poimboeuf
1036dd590d4dSJosh Poimboeuf /* klp_reloc.sym */
1037dd590d4dSJosh Poimboeuf if (!elf_create_reloc(e->out, klp_relocs,
1038dd590d4dSJosh Poimboeuf klp_reloc_off + offsetof(struct klp_reloc, sym),
1039dd590d4dSJosh Poimboeuf klp_sym, addend, R_ABS64))
1040dd590d4dSJosh Poimboeuf return -1;
1041dd590d4dSJosh Poimboeuf
1042dd590d4dSJosh Poimboeuf return 0;
1043dd590d4dSJosh Poimboeuf }
1044dd590d4dSJosh Poimboeuf
10457c2575a6SJosh Poimboeuf #define dbg_clone_reloc(sec, offset, patched_sym, addend, export, klp) \
10467c2575a6SJosh Poimboeuf dbg_indent("%s+0x%lx: %s%s0x%lx [%s%s%s%s%s%s]", \
10477c2575a6SJosh Poimboeuf sec->name, offset, patched_sym->name, \
10487c2575a6SJosh Poimboeuf addend >= 0 ? "+" : "-", labs(addend), \
10497c2575a6SJosh Poimboeuf sym_type(patched_sym), \
10507c2575a6SJosh Poimboeuf patched_sym->type == STT_SECTION ? "" : " ", \
10517c2575a6SJosh Poimboeuf patched_sym->type == STT_SECTION ? "" : sym_bind(patched_sym), \
10527c2575a6SJosh Poimboeuf is_undef_sym(patched_sym) ? " UNDEF" : "", \
10537c2575a6SJosh Poimboeuf export ? " EXPORTED" : "", \
10547c2575a6SJosh Poimboeuf klp ? " KLP" : "")
10557c2575a6SJosh Poimboeuf
1056dd590d4dSJosh Poimboeuf /* Copy a reloc and its symbol to the output object */
clone_reloc(struct elfs * e,struct reloc * patched_reloc,struct section * sec,unsigned long offset)1057dd590d4dSJosh Poimboeuf static int clone_reloc(struct elfs *e, struct reloc *patched_reloc,
1058dd590d4dSJosh Poimboeuf struct section *sec, unsigned long offset)
1059dd590d4dSJosh Poimboeuf {
1060dd590d4dSJosh Poimboeuf struct symbol *patched_sym = patched_reloc->sym;
1061dd590d4dSJosh Poimboeuf struct export *export = find_export(patched_sym);
1062dd590d4dSJosh Poimboeuf long addend = reloc_addend(patched_reloc);
1063dd590d4dSJosh Poimboeuf struct symbol *out_sym;
1064dd590d4dSJosh Poimboeuf bool klp;
1065dd590d4dSJosh Poimboeuf
1066dd590d4dSJosh Poimboeuf if (!is_reloc_allowed(patched_reloc)) {
1067dd590d4dSJosh Poimboeuf ERROR_FUNC(patched_reloc->sec->base, reloc_offset(patched_reloc),
1068dd590d4dSJosh Poimboeuf "missing symbol for reference to %s+%ld",
1069dd590d4dSJosh Poimboeuf patched_sym->name, addend);
1070dd590d4dSJosh Poimboeuf return -1;
1071dd590d4dSJosh Poimboeuf }
1072dd590d4dSJosh Poimboeuf
1073dd590d4dSJosh Poimboeuf klp = klp_reloc_needed(patched_reloc);
1074dd590d4dSJosh Poimboeuf
10757c2575a6SJosh Poimboeuf dbg_clone_reloc(sec, offset, patched_sym, addend, export, klp);
10767c2575a6SJosh Poimboeuf
1077dd590d4dSJosh Poimboeuf if (klp) {
1078dd590d4dSJosh Poimboeuf if (clone_reloc_klp(e, patched_reloc, sec, offset, export))
1079dd590d4dSJosh Poimboeuf return -1;
1080dd590d4dSJosh Poimboeuf
1081dd590d4dSJosh Poimboeuf return 0;
1082dd590d4dSJosh Poimboeuf }
1083dd590d4dSJosh Poimboeuf
1084dd590d4dSJosh Poimboeuf /*
1085dd590d4dSJosh Poimboeuf * Why !export sets 'data_too':
1086dd590d4dSJosh Poimboeuf *
1087dd590d4dSJosh Poimboeuf * Unexported non-klp symbols need to live in the patch module,
1088dd590d4dSJosh Poimboeuf * otherwise there will be unresolved symbols. Notably, this includes:
1089dd590d4dSJosh Poimboeuf *
1090dd590d4dSJosh Poimboeuf * - New functions/data
1091dd590d4dSJosh Poimboeuf * - String sections
1092dd590d4dSJosh Poimboeuf * - Special section entries
1093dd590d4dSJosh Poimboeuf * - Uncorrelated static local variables
1094dd590d4dSJosh Poimboeuf * - UBSAN sections
1095dd590d4dSJosh Poimboeuf */
1096dd590d4dSJosh Poimboeuf out_sym = clone_symbol(e, patched_sym, patched_sym->included || !export);
1097dd590d4dSJosh Poimboeuf if (!out_sym)
1098dd590d4dSJosh Poimboeuf return -1;
1099dd590d4dSJosh Poimboeuf
1100dd590d4dSJosh Poimboeuf /*
1101dd590d4dSJosh Poimboeuf * For strings, all references use section symbols, thanks to
1102dd590d4dSJosh Poimboeuf * section_reference_needed(). clone_symbol() has cloned an empty
1103dd590d4dSJosh Poimboeuf * version of the string section. Now copy the string itself.
1104dd590d4dSJosh Poimboeuf */
1105dd590d4dSJosh Poimboeuf if (is_string_sec(patched_sym->sec)) {
1106dd590d4dSJosh Poimboeuf const char *str = patched_sym->sec->data->d_buf + addend;
1107dd590d4dSJosh Poimboeuf
11087c2575a6SJosh Poimboeuf __dbg_indent("\"%s\"", escape_str(str));
11097c2575a6SJosh Poimboeuf
1110dd590d4dSJosh Poimboeuf addend = elf_add_string(e->out, out_sym->sec, str);
1111dd590d4dSJosh Poimboeuf if (addend == -1)
1112dd590d4dSJosh Poimboeuf return -1;
1113dd590d4dSJosh Poimboeuf }
1114dd590d4dSJosh Poimboeuf
1115dd590d4dSJosh Poimboeuf if (!elf_create_reloc(e->out, sec, offset, out_sym, addend,
1116dd590d4dSJosh Poimboeuf reloc_type(patched_reloc)))
1117dd590d4dSJosh Poimboeuf return -1;
1118dd590d4dSJosh Poimboeuf
1119dd590d4dSJosh Poimboeuf return 0;
1120dd590d4dSJosh Poimboeuf }
1121dd590d4dSJosh Poimboeuf
1122dd590d4dSJosh Poimboeuf /* Copy all relocs needed for a symbol's contents */
clone_sym_relocs(struct elfs * e,struct symbol * patched_sym)1123dd590d4dSJosh Poimboeuf static int clone_sym_relocs(struct elfs *e, struct symbol *patched_sym)
1124dd590d4dSJosh Poimboeuf {
1125dd590d4dSJosh Poimboeuf struct section *patched_rsec = patched_sym->sec->rsec;
1126dd590d4dSJosh Poimboeuf struct reloc *patched_reloc;
1127dd590d4dSJosh Poimboeuf unsigned long start, end;
1128dd590d4dSJosh Poimboeuf struct symbol *out_sym;
1129dd590d4dSJosh Poimboeuf
1130dd590d4dSJosh Poimboeuf out_sym = patched_sym->clone;
1131dd590d4dSJosh Poimboeuf if (!out_sym) {
1132dd590d4dSJosh Poimboeuf ERROR("no clone for %s", patched_sym->name);
1133dd590d4dSJosh Poimboeuf return -1;
1134dd590d4dSJosh Poimboeuf }
1135dd590d4dSJosh Poimboeuf
1136dd590d4dSJosh Poimboeuf if (!patched_rsec)
1137dd590d4dSJosh Poimboeuf return 0;
1138dd590d4dSJosh Poimboeuf
1139dd590d4dSJosh Poimboeuf if (!is_sec_sym(patched_sym) && !patched_sym->len)
1140dd590d4dSJosh Poimboeuf return 0;
1141dd590d4dSJosh Poimboeuf
1142dd590d4dSJosh Poimboeuf if (is_string_sec(patched_sym->sec))
1143dd590d4dSJosh Poimboeuf return 0;
1144dd590d4dSJosh Poimboeuf
1145dd590d4dSJosh Poimboeuf if (is_sec_sym(patched_sym)) {
1146dd590d4dSJosh Poimboeuf start = 0;
1147dd590d4dSJosh Poimboeuf end = sec_size(patched_sym->sec);
1148dd590d4dSJosh Poimboeuf } else {
1149dd590d4dSJosh Poimboeuf start = patched_sym->offset;
1150dd590d4dSJosh Poimboeuf end = start + patched_sym->len;
1151dd590d4dSJosh Poimboeuf }
1152dd590d4dSJosh Poimboeuf
1153dd590d4dSJosh Poimboeuf for_each_reloc(patched_rsec, patched_reloc) {
1154dd590d4dSJosh Poimboeuf unsigned long offset;
1155dd590d4dSJosh Poimboeuf
1156dd590d4dSJosh Poimboeuf if (reloc_offset(patched_reloc) < start ||
1157dd590d4dSJosh Poimboeuf reloc_offset(patched_reloc) >= end)
1158dd590d4dSJosh Poimboeuf continue;
1159dd590d4dSJosh Poimboeuf
1160dd590d4dSJosh Poimboeuf /*
1161dd590d4dSJosh Poimboeuf * Skip any reloc referencing .altinstr_aux. Its code is
1162dd590d4dSJosh Poimboeuf * always patched by alternatives. See ALTERNATIVE_TERNARY().
1163dd590d4dSJosh Poimboeuf */
1164dd590d4dSJosh Poimboeuf if (patched_reloc->sym->sec &&
1165dd590d4dSJosh Poimboeuf !strcmp(patched_reloc->sym->sec->name, ".altinstr_aux"))
1166dd590d4dSJosh Poimboeuf continue;
1167dd590d4dSJosh Poimboeuf
1168dd590d4dSJosh Poimboeuf if (convert_reloc_sym(e->patched, patched_reloc)) {
1169dd590d4dSJosh Poimboeuf ERROR_FUNC(patched_rsec->base, reloc_offset(patched_reloc),
1170dd590d4dSJosh Poimboeuf "failed to convert reloc sym '%s' to its proper format",
1171dd590d4dSJosh Poimboeuf patched_reloc->sym->name);
1172dd590d4dSJosh Poimboeuf return -1;
1173dd590d4dSJosh Poimboeuf }
1174dd590d4dSJosh Poimboeuf
1175dd590d4dSJosh Poimboeuf offset = out_sym->offset + (reloc_offset(patched_reloc) - patched_sym->offset);
1176dd590d4dSJosh Poimboeuf
1177dd590d4dSJosh Poimboeuf if (clone_reloc(e, patched_reloc, out_sym->sec, offset))
1178dd590d4dSJosh Poimboeuf return -1;
1179dd590d4dSJosh Poimboeuf }
1180dd590d4dSJosh Poimboeuf return 0;
1181dd590d4dSJosh Poimboeuf
1182dd590d4dSJosh Poimboeuf }
1183dd590d4dSJosh Poimboeuf
create_fake_symbol(struct elf * elf,struct section * sec,unsigned long offset,size_t size)1184dd590d4dSJosh Poimboeuf static int create_fake_symbol(struct elf *elf, struct section *sec,
1185dd590d4dSJosh Poimboeuf unsigned long offset, size_t size)
1186dd590d4dSJosh Poimboeuf {
1187dd590d4dSJosh Poimboeuf char name[SYM_NAME_LEN];
1188dd590d4dSJosh Poimboeuf unsigned int type;
1189dd590d4dSJosh Poimboeuf static int ctr;
1190dd590d4dSJosh Poimboeuf char *c;
1191dd590d4dSJosh Poimboeuf
1192dd590d4dSJosh Poimboeuf if (snprintf_check(name, SYM_NAME_LEN, "%s_%d", sec->name, ctr++))
1193dd590d4dSJosh Poimboeuf return -1;
1194dd590d4dSJosh Poimboeuf
1195dd590d4dSJosh Poimboeuf for (c = name; *c; c++)
1196dd590d4dSJosh Poimboeuf if (*c == '.')
1197dd590d4dSJosh Poimboeuf *c = '_';
1198dd590d4dSJosh Poimboeuf
1199dd590d4dSJosh Poimboeuf /*
1200dd590d4dSJosh Poimboeuf * STT_NOTYPE: Prevent objtool from validating .altinstr_replacement
1201dd590d4dSJosh Poimboeuf * while still allowing objdump to disassemble it.
1202dd590d4dSJosh Poimboeuf */
1203dd590d4dSJosh Poimboeuf type = is_text_sec(sec) ? STT_NOTYPE : STT_OBJECT;
1204dd590d4dSJosh Poimboeuf return elf_create_symbol(elf, name, sec, STB_LOCAL, type, offset, size) ? 0 : -1;
1205dd590d4dSJosh Poimboeuf }
1206dd590d4dSJosh Poimboeuf
1207dd590d4dSJosh Poimboeuf /*
1208dd590d4dSJosh Poimboeuf * Special sections (alternatives, etc) are basically arrays of structs.
1209dd590d4dSJosh Poimboeuf * For all the special sections, create a symbol for each struct entry. This
1210dd590d4dSJosh Poimboeuf * is a bit cumbersome, but it makes the extracting of the individual entries
1211dd590d4dSJosh Poimboeuf * much more straightforward.
1212dd590d4dSJosh Poimboeuf *
1213dd590d4dSJosh Poimboeuf * There are three ways to identify the entry sizes for a special section:
1214dd590d4dSJosh Poimboeuf *
1215dd590d4dSJosh Poimboeuf * 1) ELF section header sh_entsize: Ideally this would be used almost
1216dd590d4dSJosh Poimboeuf * everywhere. But unfortunately the toolchains make it difficult. The
1217dd590d4dSJosh Poimboeuf * assembler .[push]section directive syntax only takes entsize when
1218dd590d4dSJosh Poimboeuf * combined with SHF_MERGE. But Clang disallows combining SHF_MERGE with
1219dd590d4dSJosh Poimboeuf * SHF_WRITE. And some special sections do need to be writable.
1220dd590d4dSJosh Poimboeuf *
1221dd590d4dSJosh Poimboeuf * Another place this wouldn't work is .altinstr_replacement, whose entries
1222dd590d4dSJosh Poimboeuf * don't have a fixed size.
1223dd590d4dSJosh Poimboeuf *
1224dd590d4dSJosh Poimboeuf * 2) ANNOTATE_DATA_SPECIAL: This is a lightweight objtool annotation which
1225dd590d4dSJosh Poimboeuf * points to the beginning of each entry. The size of the entry is then
1226dd590d4dSJosh Poimboeuf * inferred by the location of the subsequent annotation (or end of
1227dd590d4dSJosh Poimboeuf * section).
1228dd590d4dSJosh Poimboeuf *
1229dd590d4dSJosh Poimboeuf * 3) Simple array of pointers: If the special section is just a basic array of
1230dd590d4dSJosh Poimboeuf * pointers, the entry size can be inferred by the number of relocations.
1231dd590d4dSJosh Poimboeuf * No annotations needed.
1232dd590d4dSJosh Poimboeuf *
1233dd590d4dSJosh Poimboeuf * Note I also tried to create per-entry symbols at the time of creation, in
1234dd590d4dSJosh Poimboeuf * the original [inline] asm. Unfortunately, creating uniquely named symbols
1235dd590d4dSJosh Poimboeuf * is trickier than one might think, especially with Clang inline asm. I
1236dd590d4dSJosh Poimboeuf * eventually just gave up trying to make that work, in favor of using
1237dd590d4dSJosh Poimboeuf * ANNOTATE_DATA_SPECIAL and creating the symbols here after the fact.
1238dd590d4dSJosh Poimboeuf */
create_fake_symbols(struct elf * elf)1239dd590d4dSJosh Poimboeuf static int create_fake_symbols(struct elf *elf)
1240dd590d4dSJosh Poimboeuf {
1241dd590d4dSJosh Poimboeuf struct section *sec;
1242dd590d4dSJosh Poimboeuf struct reloc *reloc;
1243dd590d4dSJosh Poimboeuf
1244dd590d4dSJosh Poimboeuf /*
1245dd590d4dSJosh Poimboeuf * 1) Make symbols for all the ANNOTATE_DATA_SPECIAL entries:
1246dd590d4dSJosh Poimboeuf */
1247dd590d4dSJosh Poimboeuf
1248dd590d4dSJosh Poimboeuf sec = find_section_by_name(elf, ".discard.annotate_data");
1249dd590d4dSJosh Poimboeuf if (!sec || !sec->rsec)
1250dd590d4dSJosh Poimboeuf return 0;
1251dd590d4dSJosh Poimboeuf
1252dd590d4dSJosh Poimboeuf for_each_reloc(sec->rsec, reloc) {
1253dd590d4dSJosh Poimboeuf unsigned long offset, size;
1254dd590d4dSJosh Poimboeuf struct reloc *next_reloc;
1255dd590d4dSJosh Poimboeuf
1256dd590d4dSJosh Poimboeuf if (annotype(elf, sec, reloc) != ANNOTYPE_DATA_SPECIAL)
1257dd590d4dSJosh Poimboeuf continue;
1258dd590d4dSJosh Poimboeuf
1259dd590d4dSJosh Poimboeuf offset = reloc_addend(reloc);
1260dd590d4dSJosh Poimboeuf
1261dd590d4dSJosh Poimboeuf size = 0;
1262dd590d4dSJosh Poimboeuf next_reloc = reloc;
1263dd590d4dSJosh Poimboeuf for_each_reloc_continue(sec->rsec, next_reloc) {
1264dd590d4dSJosh Poimboeuf if (annotype(elf, sec, next_reloc) != ANNOTYPE_DATA_SPECIAL ||
1265dd590d4dSJosh Poimboeuf next_reloc->sym->sec != reloc->sym->sec)
1266dd590d4dSJosh Poimboeuf continue;
1267dd590d4dSJosh Poimboeuf
1268dd590d4dSJosh Poimboeuf size = reloc_addend(next_reloc) - offset;
1269dd590d4dSJosh Poimboeuf break;
1270dd590d4dSJosh Poimboeuf }
1271dd590d4dSJosh Poimboeuf
1272dd590d4dSJosh Poimboeuf if (!size)
1273dd590d4dSJosh Poimboeuf size = sec_size(reloc->sym->sec) - offset;
1274dd590d4dSJosh Poimboeuf
1275dd590d4dSJosh Poimboeuf if (create_fake_symbol(elf, reloc->sym->sec, offset, size))
1276dd590d4dSJosh Poimboeuf return -1;
1277dd590d4dSJosh Poimboeuf }
1278dd590d4dSJosh Poimboeuf
1279dd590d4dSJosh Poimboeuf /*
1280dd590d4dSJosh Poimboeuf * 2) Make symbols for sh_entsize, and simple arrays of pointers:
1281dd590d4dSJosh Poimboeuf */
1282dd590d4dSJosh Poimboeuf
1283dd590d4dSJosh Poimboeuf for_each_sec(elf, sec) {
1284dd590d4dSJosh Poimboeuf unsigned int entry_size;
1285dd590d4dSJosh Poimboeuf unsigned long offset;
1286dd590d4dSJosh Poimboeuf
1287dd590d4dSJosh Poimboeuf if (!is_special_section(sec) || find_symbol_by_offset(sec, 0))
1288dd590d4dSJosh Poimboeuf continue;
1289dd590d4dSJosh Poimboeuf
1290dd590d4dSJosh Poimboeuf if (!sec->rsec) {
1291dd590d4dSJosh Poimboeuf ERROR("%s: missing special section relocations", sec->name);
1292dd590d4dSJosh Poimboeuf return -1;
1293dd590d4dSJosh Poimboeuf }
1294dd590d4dSJosh Poimboeuf
1295dd590d4dSJosh Poimboeuf entry_size = sec->sh.sh_entsize;
1296dd590d4dSJosh Poimboeuf if (!entry_size) {
1297dd590d4dSJosh Poimboeuf entry_size = arch_reloc_size(sec->rsec->relocs);
1298dd590d4dSJosh Poimboeuf if (sec_size(sec) != entry_size * sec_num_entries(sec->rsec)) {
1299dd590d4dSJosh Poimboeuf ERROR("%s: missing special section entsize or annotations", sec->name);
1300dd590d4dSJosh Poimboeuf return -1;
1301dd590d4dSJosh Poimboeuf }
1302dd590d4dSJosh Poimboeuf }
1303dd590d4dSJosh Poimboeuf
1304dd590d4dSJosh Poimboeuf for (offset = 0; offset < sec_size(sec); offset += entry_size) {
1305dd590d4dSJosh Poimboeuf if (create_fake_symbol(elf, sec, offset, entry_size))
1306dd590d4dSJosh Poimboeuf return -1;
1307dd590d4dSJosh Poimboeuf }
1308dd590d4dSJosh Poimboeuf }
1309dd590d4dSJosh Poimboeuf
1310dd590d4dSJosh Poimboeuf return 0;
1311dd590d4dSJosh Poimboeuf }
1312dd590d4dSJosh Poimboeuf
1313dd590d4dSJosh Poimboeuf /* Keep a special section entry if it references an included function */
should_keep_special_sym(struct elf * elf,struct symbol * sym)1314dd590d4dSJosh Poimboeuf static bool should_keep_special_sym(struct elf *elf, struct symbol *sym)
1315dd590d4dSJosh Poimboeuf {
1316dd590d4dSJosh Poimboeuf struct reloc *reloc;
1317dd590d4dSJosh Poimboeuf
1318dd590d4dSJosh Poimboeuf if (is_sec_sym(sym) || !sym->sec->rsec)
1319dd590d4dSJosh Poimboeuf return false;
1320dd590d4dSJosh Poimboeuf
1321dd590d4dSJosh Poimboeuf sym_for_each_reloc(elf, sym, reloc) {
1322dd590d4dSJosh Poimboeuf if (convert_reloc_sym(elf, reloc))
1323dd590d4dSJosh Poimboeuf continue;
1324dd590d4dSJosh Poimboeuf
1325dd590d4dSJosh Poimboeuf if (is_func_sym(reloc->sym) && reloc->sym->included)
1326dd590d4dSJosh Poimboeuf return true;
1327dd590d4dSJosh Poimboeuf }
1328dd590d4dSJosh Poimboeuf
1329dd590d4dSJosh Poimboeuf return false;
1330dd590d4dSJosh Poimboeuf }
1331dd590d4dSJosh Poimboeuf
1332dd590d4dSJosh Poimboeuf /*
1333dd590d4dSJosh Poimboeuf * Klp relocations aren't allowed for __jump_table and .static_call_sites if
1334dd590d4dSJosh Poimboeuf * the referenced symbol lives in a kernel module, because such klp relocs may
1335dd590d4dSJosh Poimboeuf * be applied after static branch/call init, resulting in code corruption.
1336dd590d4dSJosh Poimboeuf *
1337dd590d4dSJosh Poimboeuf * Validate a special section entry to avoid that. Note that an inert
1338e476bb27SJosh Poimboeuf * tracepoint or pr_debug() is harmless enough, in that case just skip the
1339e476bb27SJosh Poimboeuf * entry and print a warning. Otherwise, return an error.
1340dd590d4dSJosh Poimboeuf *
1341e476bb27SJosh Poimboeuf * TODO: This is only a temporary limitation which will be fixed when livepatch
1342e476bb27SJosh Poimboeuf * adds support for submodules: fully self-contained modules which are embedded
1343e476bb27SJosh Poimboeuf * in the top-level livepatch module's data and which can be loaded on demand
1344e476bb27SJosh Poimboeuf * when their corresponding to-be-patched module gets loaded. Then klp relocs
1345e476bb27SJosh Poimboeuf * can be retired.
1346dd590d4dSJosh Poimboeuf *
1347dd590d4dSJosh Poimboeuf * Return:
1348dd590d4dSJosh Poimboeuf * -1: error: validation failed
1349e476bb27SJosh Poimboeuf * 1: warning: disabled tracepoint or pr_debug()
1350dd590d4dSJosh Poimboeuf * 0: success
1351dd590d4dSJosh Poimboeuf */
validate_special_section_klp_reloc(struct elfs * e,struct symbol * sym)1352dd590d4dSJosh Poimboeuf static int validate_special_section_klp_reloc(struct elfs *e, struct symbol *sym)
1353dd590d4dSJosh Poimboeuf {
1354dd590d4dSJosh Poimboeuf bool static_branch = !strcmp(sym->sec->name, "__jump_table");
1355dd590d4dSJosh Poimboeuf bool static_call = !strcmp(sym->sec->name, ".static_call_sites");
135611c2adcdSJosh Poimboeuf const char *code_sym = NULL;
1357dd590d4dSJosh Poimboeuf unsigned long code_offset = 0;
1358dd590d4dSJosh Poimboeuf struct reloc *reloc;
1359dd590d4dSJosh Poimboeuf int ret = 0;
1360dd590d4dSJosh Poimboeuf
1361dd590d4dSJosh Poimboeuf if (!static_branch && !static_call)
1362dd590d4dSJosh Poimboeuf return 0;
1363dd590d4dSJosh Poimboeuf
1364dd590d4dSJosh Poimboeuf sym_for_each_reloc(e->patched, sym, reloc) {
1365dd590d4dSJosh Poimboeuf const char *sym_modname;
1366dd590d4dSJosh Poimboeuf struct export *export;
1367dd590d4dSJosh Poimboeuf
1368f9fb44b0SJosh Poimboeuf if (convert_reloc_sym(e->patched, reloc))
1369f9fb44b0SJosh Poimboeuf continue;
1370f9fb44b0SJosh Poimboeuf
1371dd590d4dSJosh Poimboeuf /* Static branch/call keys are always STT_OBJECT */
1372dd590d4dSJosh Poimboeuf if (reloc->sym->type != STT_OBJECT) {
1373dd590d4dSJosh Poimboeuf
1374dd590d4dSJosh Poimboeuf /* Save code location which can be printed below */
1375dd590d4dSJosh Poimboeuf if (reloc->sym->type == STT_FUNC && !code_sym) {
137611c2adcdSJosh Poimboeuf code_sym = reloc->sym->name;
1377dd590d4dSJosh Poimboeuf code_offset = reloc_addend(reloc);
1378dd590d4dSJosh Poimboeuf }
1379dd590d4dSJosh Poimboeuf
1380dd590d4dSJosh Poimboeuf continue;
1381dd590d4dSJosh Poimboeuf }
1382dd590d4dSJosh Poimboeuf
1383dd590d4dSJosh Poimboeuf if (!klp_reloc_needed(reloc))
1384dd590d4dSJosh Poimboeuf continue;
1385dd590d4dSJosh Poimboeuf
1386dd590d4dSJosh Poimboeuf export = find_export(reloc->sym);
1387dd590d4dSJosh Poimboeuf if (export) {
1388dd590d4dSJosh Poimboeuf sym_modname = export->mod;
1389dd590d4dSJosh Poimboeuf } else {
1390dd590d4dSJosh Poimboeuf sym_modname = find_modname(e);
1391dd590d4dSJosh Poimboeuf if (!sym_modname)
1392dd590d4dSJosh Poimboeuf return -1;
1393dd590d4dSJosh Poimboeuf }
1394dd590d4dSJosh Poimboeuf
1395dd590d4dSJosh Poimboeuf /* vmlinux keys are ok */
1396dd590d4dSJosh Poimboeuf if (!strcmp(sym_modname, "vmlinux"))
1397dd590d4dSJosh Poimboeuf continue;
1398dd590d4dSJosh Poimboeuf
139911c2adcdSJosh Poimboeuf if (!code_sym)
140011c2adcdSJosh Poimboeuf code_sym = "<unknown>";
140111c2adcdSJosh Poimboeuf
1402dd590d4dSJosh Poimboeuf if (static_branch) {
1403dd590d4dSJosh Poimboeuf if (strstarts(reloc->sym->name, "__tracepoint_")) {
1404dd590d4dSJosh Poimboeuf WARN("%s: disabling unsupported tracepoint %s",
140511c2adcdSJosh Poimboeuf code_sym, reloc->sym->name + 13);
1406dd590d4dSJosh Poimboeuf ret = 1;
1407dd590d4dSJosh Poimboeuf continue;
1408dd590d4dSJosh Poimboeuf }
1409dd590d4dSJosh Poimboeuf
1410e476bb27SJosh Poimboeuf if (strstr(reloc->sym->name, "__UNIQUE_ID_ddebug_")) {
1411e476bb27SJosh Poimboeuf WARN("%s: disabling unsupported pr_debug()",
141211c2adcdSJosh Poimboeuf code_sym);
1413e476bb27SJosh Poimboeuf ret = 1;
1414e476bb27SJosh Poimboeuf continue;
1415e476bb27SJosh Poimboeuf }
1416e476bb27SJosh Poimboeuf
1417dd590d4dSJosh Poimboeuf ERROR("%s+0x%lx: unsupported static branch key %s. Use static_key_enabled() instead",
141811c2adcdSJosh Poimboeuf code_sym, code_offset, reloc->sym->name);
1419dd590d4dSJosh Poimboeuf return -1;
1420dd590d4dSJosh Poimboeuf }
1421dd590d4dSJosh Poimboeuf
1422dd590d4dSJosh Poimboeuf /* static call */
1423dd590d4dSJosh Poimboeuf if (strstarts(reloc->sym->name, "__SCK__tp_func_")) {
1424dd590d4dSJosh Poimboeuf ret = 1;
1425dd590d4dSJosh Poimboeuf continue;
1426dd590d4dSJosh Poimboeuf }
1427dd590d4dSJosh Poimboeuf
1428dd590d4dSJosh Poimboeuf ERROR("%s()+0x%lx: unsupported static call key %s. Use KLP_STATIC_CALL() instead",
142911c2adcdSJosh Poimboeuf code_sym, code_offset, reloc->sym->name);
1430dd590d4dSJosh Poimboeuf return -1;
1431dd590d4dSJosh Poimboeuf }
1432dd590d4dSJosh Poimboeuf
1433dd590d4dSJosh Poimboeuf return ret;
1434dd590d4dSJosh Poimboeuf }
1435dd590d4dSJosh Poimboeuf
clone_special_section(struct elfs * e,struct section * patched_sec)1436dd590d4dSJosh Poimboeuf static int clone_special_section(struct elfs *e, struct section *patched_sec)
1437dd590d4dSJosh Poimboeuf {
1438dd590d4dSJosh Poimboeuf struct symbol *patched_sym;
1439dd590d4dSJosh Poimboeuf
1440dd590d4dSJosh Poimboeuf /*
1441dd590d4dSJosh Poimboeuf * Extract all special section symbols (and their dependencies) which
1442dd590d4dSJosh Poimboeuf * reference included functions.
1443dd590d4dSJosh Poimboeuf */
1444dd590d4dSJosh Poimboeuf sec_for_each_sym(patched_sec, patched_sym) {
1445dd590d4dSJosh Poimboeuf int ret;
1446dd590d4dSJosh Poimboeuf
1447dd590d4dSJosh Poimboeuf if (!is_object_sym(patched_sym))
1448dd590d4dSJosh Poimboeuf continue;
1449dd590d4dSJosh Poimboeuf
1450dd590d4dSJosh Poimboeuf if (!should_keep_special_sym(e->patched, patched_sym))
1451dd590d4dSJosh Poimboeuf continue;
1452dd590d4dSJosh Poimboeuf
1453dd590d4dSJosh Poimboeuf ret = validate_special_section_klp_reloc(e, patched_sym);
1454dd590d4dSJosh Poimboeuf if (ret < 0)
1455dd590d4dSJosh Poimboeuf return -1;
1456dd590d4dSJosh Poimboeuf if (ret > 0)
1457dd590d4dSJosh Poimboeuf continue;
1458dd590d4dSJosh Poimboeuf
1459dd590d4dSJosh Poimboeuf if (!clone_symbol(e, patched_sym, true))
1460dd590d4dSJosh Poimboeuf return -1;
1461dd590d4dSJosh Poimboeuf }
1462dd590d4dSJosh Poimboeuf
1463dd590d4dSJosh Poimboeuf return 0;
1464dd590d4dSJosh Poimboeuf }
1465dd590d4dSJosh Poimboeuf
1466dd590d4dSJosh Poimboeuf /* Extract only the needed bits from special sections */
clone_special_sections(struct elfs * e)1467dd590d4dSJosh Poimboeuf static int clone_special_sections(struct elfs *e)
1468dd590d4dSJosh Poimboeuf {
1469dd590d4dSJosh Poimboeuf struct section *patched_sec;
1470dd590d4dSJosh Poimboeuf
1471dd590d4dSJosh Poimboeuf for_each_sec(e->patched, patched_sec) {
1472dd590d4dSJosh Poimboeuf if (is_special_section(patched_sec)) {
1473dd590d4dSJosh Poimboeuf if (clone_special_section(e, patched_sec))
1474dd590d4dSJosh Poimboeuf return -1;
1475dd590d4dSJosh Poimboeuf }
1476dd590d4dSJosh Poimboeuf }
1477dd590d4dSJosh Poimboeuf
1478dd590d4dSJosh Poimboeuf return 0;
1479dd590d4dSJosh Poimboeuf }
1480dd590d4dSJosh Poimboeuf
1481dd590d4dSJosh Poimboeuf /*
1482b525fcafSPetr Pavlu * Create .init.klp_objects and .init.klp_funcs sections which are intermediate
1483dd590d4dSJosh Poimboeuf * sections provided as input to the patch module's init code for building the
1484dd590d4dSJosh Poimboeuf * klp_patch, klp_object and klp_func structs for the livepatch API.
1485dd590d4dSJosh Poimboeuf */
create_klp_sections(struct elfs * e)1486dd590d4dSJosh Poimboeuf static int create_klp_sections(struct elfs *e)
1487dd590d4dSJosh Poimboeuf {
1488dd590d4dSJosh Poimboeuf size_t obj_size = sizeof(struct klp_object_ext);
1489dd590d4dSJosh Poimboeuf size_t func_size = sizeof(struct klp_func_ext);
1490dd590d4dSJosh Poimboeuf struct section *obj_sec, *funcs_sec, *str_sec;
1491dd590d4dSJosh Poimboeuf struct symbol *funcs_sym, *str_sym, *sym;
1492dd590d4dSJosh Poimboeuf char sym_name[SYM_NAME_LEN];
1493dd590d4dSJosh Poimboeuf unsigned int nr_funcs = 0;
1494dd590d4dSJosh Poimboeuf const char *modname;
1495dd590d4dSJosh Poimboeuf void *obj_data;
1496dd590d4dSJosh Poimboeuf s64 addend;
1497dd590d4dSJosh Poimboeuf
1498dd590d4dSJosh Poimboeuf obj_sec = elf_create_section_pair(e->out, KLP_OBJECTS_SEC, obj_size, 0, 0);
1499dd590d4dSJosh Poimboeuf if (!obj_sec)
1500dd590d4dSJosh Poimboeuf return -1;
1501dd590d4dSJosh Poimboeuf
1502dd590d4dSJosh Poimboeuf funcs_sec = elf_create_section_pair(e->out, KLP_FUNCS_SEC, func_size, 0, 0);
1503dd590d4dSJosh Poimboeuf if (!funcs_sec)
1504dd590d4dSJosh Poimboeuf return -1;
1505dd590d4dSJosh Poimboeuf
1506dd590d4dSJosh Poimboeuf funcs_sym = elf_create_section_symbol(e->out, funcs_sec);
1507dd590d4dSJosh Poimboeuf if (!funcs_sym)
1508dd590d4dSJosh Poimboeuf return -1;
1509dd590d4dSJosh Poimboeuf
1510dd590d4dSJosh Poimboeuf str_sec = elf_create_section(e->out, KLP_STRINGS_SEC, 0, 0,
1511dd590d4dSJosh Poimboeuf SHT_PROGBITS, 1,
1512dd590d4dSJosh Poimboeuf SHF_ALLOC | SHF_STRINGS | SHF_MERGE);
1513dd590d4dSJosh Poimboeuf if (!str_sec)
1514dd590d4dSJosh Poimboeuf return -1;
1515dd590d4dSJosh Poimboeuf
1516dd590d4dSJosh Poimboeuf if (elf_add_string(e->out, str_sec, "") == -1)
1517dd590d4dSJosh Poimboeuf return -1;
1518dd590d4dSJosh Poimboeuf
1519dd590d4dSJosh Poimboeuf str_sym = elf_create_section_symbol(e->out, str_sec);
1520dd590d4dSJosh Poimboeuf if (!str_sym)
1521dd590d4dSJosh Poimboeuf return -1;
1522dd590d4dSJosh Poimboeuf
1523dd590d4dSJosh Poimboeuf /* allocate klp_object_ext */
1524dd590d4dSJosh Poimboeuf obj_data = elf_add_data(e->out, obj_sec, NULL, obj_size);
1525dd590d4dSJosh Poimboeuf if (!obj_data)
1526dd590d4dSJosh Poimboeuf return -1;
1527dd590d4dSJosh Poimboeuf
1528dd590d4dSJosh Poimboeuf modname = find_modname(e);
1529dd590d4dSJosh Poimboeuf if (!modname)
1530dd590d4dSJosh Poimboeuf return -1;
1531dd590d4dSJosh Poimboeuf
1532dd590d4dSJosh Poimboeuf /* klp_object_ext.name */
1533dd590d4dSJosh Poimboeuf if (strcmp(modname, "vmlinux")) {
1534dd590d4dSJosh Poimboeuf addend = elf_add_string(e->out, str_sec, modname);
1535dd590d4dSJosh Poimboeuf if (addend == -1)
1536dd590d4dSJosh Poimboeuf return -1;
1537dd590d4dSJosh Poimboeuf
1538dd590d4dSJosh Poimboeuf if (!elf_create_reloc(e->out, obj_sec,
1539dd590d4dSJosh Poimboeuf offsetof(struct klp_object_ext, name),
1540dd590d4dSJosh Poimboeuf str_sym, addend, R_ABS64))
1541dd590d4dSJosh Poimboeuf return -1;
1542dd590d4dSJosh Poimboeuf }
1543dd590d4dSJosh Poimboeuf
1544dd590d4dSJosh Poimboeuf /* klp_object_ext.funcs */
1545dd590d4dSJosh Poimboeuf if (!elf_create_reloc(e->out, obj_sec, offsetof(struct klp_object_ext, funcs),
1546dd590d4dSJosh Poimboeuf funcs_sym, 0, R_ABS64))
1547dd590d4dSJosh Poimboeuf return -1;
1548dd590d4dSJosh Poimboeuf
1549dd590d4dSJosh Poimboeuf for_each_sym(e->out, sym) {
1550dd590d4dSJosh Poimboeuf unsigned long offset = nr_funcs * func_size;
1551dd590d4dSJosh Poimboeuf unsigned long sympos;
1552dd590d4dSJosh Poimboeuf void *func_data;
1553dd590d4dSJosh Poimboeuf
1554dd590d4dSJosh Poimboeuf if (!is_func_sym(sym) || sym->cold || !sym->clone || !sym->clone->changed)
1555dd590d4dSJosh Poimboeuf continue;
1556dd590d4dSJosh Poimboeuf
1557dd590d4dSJosh Poimboeuf /* allocate klp_func_ext */
1558dd590d4dSJosh Poimboeuf func_data = elf_add_data(e->out, funcs_sec, NULL, func_size);
1559dd590d4dSJosh Poimboeuf if (!func_data)
1560dd590d4dSJosh Poimboeuf return -1;
1561dd590d4dSJosh Poimboeuf
1562dd590d4dSJosh Poimboeuf /* klp_func_ext.old_name */
1563dd590d4dSJosh Poimboeuf addend = elf_add_string(e->out, str_sec, sym->clone->twin->name);
1564dd590d4dSJosh Poimboeuf if (addend == -1)
1565dd590d4dSJosh Poimboeuf return -1;
1566dd590d4dSJosh Poimboeuf
1567dd590d4dSJosh Poimboeuf if (!elf_create_reloc(e->out, funcs_sec,
1568dd590d4dSJosh Poimboeuf offset + offsetof(struct klp_func_ext, old_name),
1569dd590d4dSJosh Poimboeuf str_sym, addend, R_ABS64))
1570dd590d4dSJosh Poimboeuf return -1;
1571dd590d4dSJosh Poimboeuf
1572dd590d4dSJosh Poimboeuf /* klp_func_ext.new_func */
1573dd590d4dSJosh Poimboeuf if (!elf_create_reloc(e->out, funcs_sec,
1574dd590d4dSJosh Poimboeuf offset + offsetof(struct klp_func_ext, new_func),
1575dd590d4dSJosh Poimboeuf sym, 0, R_ABS64))
1576dd590d4dSJosh Poimboeuf return -1;
1577dd590d4dSJosh Poimboeuf
1578dd590d4dSJosh Poimboeuf /* klp_func_ext.sympos */
1579dd590d4dSJosh Poimboeuf BUILD_BUG_ON(sizeof(sympos) != sizeof_field(struct klp_func_ext, sympos));
1580dd590d4dSJosh Poimboeuf sympos = find_sympos(e->orig, sym->clone->twin);
1581dd590d4dSJosh Poimboeuf if (sympos == ULONG_MAX)
1582dd590d4dSJosh Poimboeuf return -1;
1583dd590d4dSJosh Poimboeuf memcpy(func_data + offsetof(struct klp_func_ext, sympos), &sympos,
1584dd590d4dSJosh Poimboeuf sizeof_field(struct klp_func_ext, sympos));
1585dd590d4dSJosh Poimboeuf
1586dd590d4dSJosh Poimboeuf nr_funcs++;
1587dd590d4dSJosh Poimboeuf }
1588dd590d4dSJosh Poimboeuf
1589dd590d4dSJosh Poimboeuf /* klp_object_ext.nr_funcs */
1590dd590d4dSJosh Poimboeuf BUILD_BUG_ON(sizeof(nr_funcs) != sizeof_field(struct klp_object_ext, nr_funcs));
1591dd590d4dSJosh Poimboeuf memcpy(obj_data + offsetof(struct klp_object_ext, nr_funcs), &nr_funcs,
1592dd590d4dSJosh Poimboeuf sizeof_field(struct klp_object_ext, nr_funcs));
1593dd590d4dSJosh Poimboeuf
1594dd590d4dSJosh Poimboeuf /*
1595dd590d4dSJosh Poimboeuf * Find callback pointers created by KLP_PRE_PATCH_CALLBACK() and
1596dd590d4dSJosh Poimboeuf * friends, and add them to the klp object.
1597dd590d4dSJosh Poimboeuf */
1598dd590d4dSJosh Poimboeuf
1599dd590d4dSJosh Poimboeuf if (snprintf_check(sym_name, SYM_NAME_LEN, KLP_PRE_PATCH_PREFIX "%s", modname))
1600dd590d4dSJosh Poimboeuf return -1;
1601dd590d4dSJosh Poimboeuf
1602dd590d4dSJosh Poimboeuf sym = find_symbol_by_name(e->out, sym_name);
1603dd590d4dSJosh Poimboeuf if (sym) {
1604dd590d4dSJosh Poimboeuf struct reloc *reloc;
1605dd590d4dSJosh Poimboeuf
1606dd590d4dSJosh Poimboeuf reloc = find_reloc_by_dest(e->out, sym->sec, sym->offset);
1607dd590d4dSJosh Poimboeuf
1608dd590d4dSJosh Poimboeuf if (!elf_create_reloc(e->out, obj_sec,
1609dd590d4dSJosh Poimboeuf offsetof(struct klp_object_ext, callbacks) +
1610dd590d4dSJosh Poimboeuf offsetof(struct klp_callbacks, pre_patch),
1611dd590d4dSJosh Poimboeuf reloc->sym, reloc_addend(reloc), R_ABS64))
1612dd590d4dSJosh Poimboeuf return -1;
1613dd590d4dSJosh Poimboeuf }
1614dd590d4dSJosh Poimboeuf
1615dd590d4dSJosh Poimboeuf if (snprintf_check(sym_name, SYM_NAME_LEN, KLP_POST_PATCH_PREFIX "%s", modname))
1616dd590d4dSJosh Poimboeuf return -1;
1617dd590d4dSJosh Poimboeuf
1618dd590d4dSJosh Poimboeuf sym = find_symbol_by_name(e->out, sym_name);
1619dd590d4dSJosh Poimboeuf if (sym) {
1620dd590d4dSJosh Poimboeuf struct reloc *reloc;
1621dd590d4dSJosh Poimboeuf
1622dd590d4dSJosh Poimboeuf reloc = find_reloc_by_dest(e->out, sym->sec, sym->offset);
1623dd590d4dSJosh Poimboeuf
1624dd590d4dSJosh Poimboeuf if (!elf_create_reloc(e->out, obj_sec,
1625dd590d4dSJosh Poimboeuf offsetof(struct klp_object_ext, callbacks) +
1626dd590d4dSJosh Poimboeuf offsetof(struct klp_callbacks, post_patch),
1627dd590d4dSJosh Poimboeuf reloc->sym, reloc_addend(reloc), R_ABS64))
1628dd590d4dSJosh Poimboeuf return -1;
1629dd590d4dSJosh Poimboeuf }
1630dd590d4dSJosh Poimboeuf
1631dd590d4dSJosh Poimboeuf if (snprintf_check(sym_name, SYM_NAME_LEN, KLP_PRE_UNPATCH_PREFIX "%s", modname))
1632dd590d4dSJosh Poimboeuf return -1;
1633dd590d4dSJosh Poimboeuf
1634dd590d4dSJosh Poimboeuf sym = find_symbol_by_name(e->out, sym_name);
1635dd590d4dSJosh Poimboeuf if (sym) {
1636dd590d4dSJosh Poimboeuf struct reloc *reloc;
1637dd590d4dSJosh Poimboeuf
1638dd590d4dSJosh Poimboeuf reloc = find_reloc_by_dest(e->out, sym->sec, sym->offset);
1639dd590d4dSJosh Poimboeuf
1640dd590d4dSJosh Poimboeuf if (!elf_create_reloc(e->out, obj_sec,
1641dd590d4dSJosh Poimboeuf offsetof(struct klp_object_ext, callbacks) +
1642dd590d4dSJosh Poimboeuf offsetof(struct klp_callbacks, pre_unpatch),
1643dd590d4dSJosh Poimboeuf reloc->sym, reloc_addend(reloc), R_ABS64))
1644dd590d4dSJosh Poimboeuf return -1;
1645dd590d4dSJosh Poimboeuf }
1646dd590d4dSJosh Poimboeuf
1647dd590d4dSJosh Poimboeuf if (snprintf_check(sym_name, SYM_NAME_LEN, KLP_POST_UNPATCH_PREFIX "%s", modname))
1648dd590d4dSJosh Poimboeuf return -1;
1649dd590d4dSJosh Poimboeuf
1650dd590d4dSJosh Poimboeuf sym = find_symbol_by_name(e->out, sym_name);
1651dd590d4dSJosh Poimboeuf if (sym) {
1652dd590d4dSJosh Poimboeuf struct reloc *reloc;
1653dd590d4dSJosh Poimboeuf
1654dd590d4dSJosh Poimboeuf reloc = find_reloc_by_dest(e->out, sym->sec, sym->offset);
1655dd590d4dSJosh Poimboeuf
1656dd590d4dSJosh Poimboeuf if (!elf_create_reloc(e->out, obj_sec,
1657dd590d4dSJosh Poimboeuf offsetof(struct klp_object_ext, callbacks) +
1658dd590d4dSJosh Poimboeuf offsetof(struct klp_callbacks, post_unpatch),
1659dd590d4dSJosh Poimboeuf reloc->sym, reloc_addend(reloc), R_ABS64))
1660dd590d4dSJosh Poimboeuf return -1;
1661dd590d4dSJosh Poimboeuf }
1662dd590d4dSJosh Poimboeuf
1663dd590d4dSJosh Poimboeuf return 0;
1664dd590d4dSJosh Poimboeuf }
1665dd590d4dSJosh Poimboeuf
1666dd590d4dSJosh Poimboeuf /*
1667dd590d4dSJosh Poimboeuf * Copy all .modinfo import_ns= tags to ensure all namespaced exported symbols
1668dd590d4dSJosh Poimboeuf * can be accessed via normal relocs.
1669dd590d4dSJosh Poimboeuf */
copy_import_ns(struct elfs * e)1670dd590d4dSJosh Poimboeuf static int copy_import_ns(struct elfs *e)
1671dd590d4dSJosh Poimboeuf {
1672dd590d4dSJosh Poimboeuf struct section *patched_sec, *out_sec = NULL;
1673dd590d4dSJosh Poimboeuf char *import_ns, *data_end;
1674dd590d4dSJosh Poimboeuf
1675dd590d4dSJosh Poimboeuf patched_sec = find_section_by_name(e->patched, ".modinfo");
1676dd590d4dSJosh Poimboeuf if (!patched_sec)
1677dd590d4dSJosh Poimboeuf return 0;
1678dd590d4dSJosh Poimboeuf
1679dd590d4dSJosh Poimboeuf import_ns = patched_sec->data->d_buf;
1680dd590d4dSJosh Poimboeuf if (!import_ns)
1681dd590d4dSJosh Poimboeuf return 0;
1682dd590d4dSJosh Poimboeuf
1683dd590d4dSJosh Poimboeuf for (data_end = import_ns + sec_size(patched_sec);
1684dd590d4dSJosh Poimboeuf import_ns < data_end;
1685dd590d4dSJosh Poimboeuf import_ns += strlen(import_ns) + 1) {
1686dd590d4dSJosh Poimboeuf
1687dd590d4dSJosh Poimboeuf import_ns = memmem(import_ns, data_end - import_ns, "import_ns=", 10);
1688dd590d4dSJosh Poimboeuf if (!import_ns)
1689dd590d4dSJosh Poimboeuf return 0;
1690dd590d4dSJosh Poimboeuf
1691dd590d4dSJosh Poimboeuf if (!out_sec) {
1692dd590d4dSJosh Poimboeuf out_sec = find_section_by_name(e->out, ".modinfo");
1693dd590d4dSJosh Poimboeuf if (!out_sec) {
1694dd590d4dSJosh Poimboeuf out_sec = elf_create_section(e->out, ".modinfo", 0,
1695dd590d4dSJosh Poimboeuf patched_sec->sh.sh_entsize,
1696dd590d4dSJosh Poimboeuf patched_sec->sh.sh_type,
1697dd590d4dSJosh Poimboeuf patched_sec->sh.sh_addralign,
1698dd590d4dSJosh Poimboeuf patched_sec->sh.sh_flags);
1699dd590d4dSJosh Poimboeuf if (!out_sec)
1700dd590d4dSJosh Poimboeuf return -1;
1701dd590d4dSJosh Poimboeuf }
1702dd590d4dSJosh Poimboeuf }
1703dd590d4dSJosh Poimboeuf
1704dd590d4dSJosh Poimboeuf if (!elf_add_data(e->out, out_sec, import_ns, strlen(import_ns) + 1))
1705dd590d4dSJosh Poimboeuf return -1;
1706dd590d4dSJosh Poimboeuf }
1707dd590d4dSJosh Poimboeuf
1708dd590d4dSJosh Poimboeuf return 0;
1709dd590d4dSJosh Poimboeuf }
1710dd590d4dSJosh Poimboeuf
cmd_klp_diff(int argc,const char ** argv)1711dd590d4dSJosh Poimboeuf int cmd_klp_diff(int argc, const char **argv)
1712dd590d4dSJosh Poimboeuf {
1713dd590d4dSJosh Poimboeuf struct elfs e = {0};
1714dd590d4dSJosh Poimboeuf
1715dd590d4dSJosh Poimboeuf argc = parse_options(argc, argv, klp_diff_options, klp_diff_usage, 0);
1716dd590d4dSJosh Poimboeuf if (argc != 3)
1717dd590d4dSJosh Poimboeuf usage_with_options(klp_diff_usage, klp_diff_options);
1718dd590d4dSJosh Poimboeuf
1719dd590d4dSJosh Poimboeuf objname = argv[0];
1720dd590d4dSJosh Poimboeuf
1721dd590d4dSJosh Poimboeuf e.orig = elf_open_read(argv[0], O_RDONLY);
1722dd590d4dSJosh Poimboeuf e.patched = elf_open_read(argv[1], O_RDONLY);
1723dd590d4dSJosh Poimboeuf e.out = NULL;
1724dd590d4dSJosh Poimboeuf
1725dd590d4dSJosh Poimboeuf if (!e.orig || !e.patched)
1726dd590d4dSJosh Poimboeuf return -1;
1727dd590d4dSJosh Poimboeuf
1728dd590d4dSJosh Poimboeuf if (read_exports())
1729dd590d4dSJosh Poimboeuf return -1;
1730dd590d4dSJosh Poimboeuf
1731dd590d4dSJosh Poimboeuf if (read_sym_checksums(e.orig))
1732dd590d4dSJosh Poimboeuf return -1;
1733dd590d4dSJosh Poimboeuf
1734dd590d4dSJosh Poimboeuf if (read_sym_checksums(e.patched))
1735dd590d4dSJosh Poimboeuf return -1;
1736dd590d4dSJosh Poimboeuf
1737dd590d4dSJosh Poimboeuf if (correlate_symbols(&e))
1738dd590d4dSJosh Poimboeuf return -1;
1739dd590d4dSJosh Poimboeuf
1740dd590d4dSJosh Poimboeuf if (mark_changed_functions(&e))
1741dd590d4dSJosh Poimboeuf return 0;
1742dd590d4dSJosh Poimboeuf
1743dd590d4dSJosh Poimboeuf e.out = elf_create_file(&e.orig->ehdr, argv[2]);
1744dd590d4dSJosh Poimboeuf if (!e.out)
1745dd590d4dSJosh Poimboeuf return -1;
1746dd590d4dSJosh Poimboeuf
1747f2dba603SJosh Poimboeuf /*
1748f2dba603SJosh Poimboeuf * Special section fake symbols are needed so that individual special
1749f2dba603SJosh Poimboeuf * section entries can be extracted by clone_special_sections().
1750f2dba603SJosh Poimboeuf *
1751f2dba603SJosh Poimboeuf * Note the fake symbols are also needed by clone_included_functions()
1752f2dba603SJosh Poimboeuf * because __WARN_printf() call sites add references to bug table
1753f2dba603SJosh Poimboeuf * entries in the calling functions.
1754f2dba603SJosh Poimboeuf */
1755f2dba603SJosh Poimboeuf if (create_fake_symbols(e.patched))
1756f2dba603SJosh Poimboeuf return -1;
1757f2dba603SJosh Poimboeuf
1758dd590d4dSJosh Poimboeuf if (clone_included_functions(&e))
1759dd590d4dSJosh Poimboeuf return -1;
1760dd590d4dSJosh Poimboeuf
1761dd590d4dSJosh Poimboeuf if (clone_special_sections(&e))
1762dd590d4dSJosh Poimboeuf return -1;
1763dd590d4dSJosh Poimboeuf
1764dd590d4dSJosh Poimboeuf if (create_klp_sections(&e))
1765dd590d4dSJosh Poimboeuf return -1;
1766dd590d4dSJosh Poimboeuf
1767dd590d4dSJosh Poimboeuf if (copy_import_ns(&e))
1768dd590d4dSJosh Poimboeuf return -1;
1769dd590d4dSJosh Poimboeuf
1770dd590d4dSJosh Poimboeuf if (elf_write(e.out))
1771dd590d4dSJosh Poimboeuf return -1;
1772dd590d4dSJosh Poimboeuf
1773dd590d4dSJosh Poimboeuf return elf_close(e.out);
1774dd590d4dSJosh Poimboeuf }
1775