xref: /linux/tools/objtool/klp-diff.c (revision dea622e183d34e6a4f90acfee9abb605885432bf)
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