xref: /OK3568_Linux_fs/kernel/tools/objtool/check.c (revision 4882a59341e53eb6f0b4789bf948001014eff981)
1*4882a593Smuzhiyun // SPDX-License-Identifier: GPL-2.0-or-later
2*4882a593Smuzhiyun /*
3*4882a593Smuzhiyun  * Copyright (C) 2015-2017 Josh Poimboeuf <jpoimboe@redhat.com>
4*4882a593Smuzhiyun  */
5*4882a593Smuzhiyun 
6*4882a593Smuzhiyun #include <string.h>
7*4882a593Smuzhiyun #include <stdlib.h>
8*4882a593Smuzhiyun #include <inttypes.h>
9*4882a593Smuzhiyun #include <sys/mman.h>
10*4882a593Smuzhiyun 
11*4882a593Smuzhiyun #include "builtin.h"
12*4882a593Smuzhiyun #include "cfi.h"
13*4882a593Smuzhiyun #include "arch.h"
14*4882a593Smuzhiyun #include "check.h"
15*4882a593Smuzhiyun #include "special.h"
16*4882a593Smuzhiyun #include "warn.h"
17*4882a593Smuzhiyun #include "arch_elf.h"
18*4882a593Smuzhiyun 
19*4882a593Smuzhiyun #include <linux/objtool.h>
20*4882a593Smuzhiyun #include <linux/hashtable.h>
21*4882a593Smuzhiyun #include <linux/kernel.h>
22*4882a593Smuzhiyun #include <linux/static_call_types.h>
23*4882a593Smuzhiyun 
24*4882a593Smuzhiyun struct alternative {
25*4882a593Smuzhiyun 	struct list_head list;
26*4882a593Smuzhiyun 	struct instruction *insn;
27*4882a593Smuzhiyun 	bool skip_orig;
28*4882a593Smuzhiyun };
29*4882a593Smuzhiyun 
30*4882a593Smuzhiyun static unsigned long nr_cfi, nr_cfi_reused, nr_cfi_cache;
31*4882a593Smuzhiyun 
32*4882a593Smuzhiyun static struct cfi_init_state initial_func_cfi;
33*4882a593Smuzhiyun static struct cfi_state init_cfi;
34*4882a593Smuzhiyun static struct cfi_state func_cfi;
35*4882a593Smuzhiyun 
find_insn(struct objtool_file * file,struct section * sec,unsigned long offset)36*4882a593Smuzhiyun struct instruction *find_insn(struct objtool_file *file,
37*4882a593Smuzhiyun 			      struct section *sec, unsigned long offset)
38*4882a593Smuzhiyun {
39*4882a593Smuzhiyun 	struct instruction *insn;
40*4882a593Smuzhiyun 
41*4882a593Smuzhiyun 	hash_for_each_possible(file->insn_hash, insn, hash, sec_offset_hash(sec, offset)) {
42*4882a593Smuzhiyun 		if (insn->sec == sec && insn->offset == offset)
43*4882a593Smuzhiyun 			return insn;
44*4882a593Smuzhiyun 	}
45*4882a593Smuzhiyun 
46*4882a593Smuzhiyun 	return NULL;
47*4882a593Smuzhiyun }
48*4882a593Smuzhiyun 
next_insn_same_sec(struct objtool_file * file,struct instruction * insn)49*4882a593Smuzhiyun static struct instruction *next_insn_same_sec(struct objtool_file *file,
50*4882a593Smuzhiyun 					      struct instruction *insn)
51*4882a593Smuzhiyun {
52*4882a593Smuzhiyun 	struct instruction *next = list_next_entry(insn, list);
53*4882a593Smuzhiyun 
54*4882a593Smuzhiyun 	if (!next || &next->list == &file->insn_list || next->sec != insn->sec)
55*4882a593Smuzhiyun 		return NULL;
56*4882a593Smuzhiyun 
57*4882a593Smuzhiyun 	return next;
58*4882a593Smuzhiyun }
59*4882a593Smuzhiyun 
next_insn_same_func(struct objtool_file * file,struct instruction * insn)60*4882a593Smuzhiyun static struct instruction *next_insn_same_func(struct objtool_file *file,
61*4882a593Smuzhiyun 					       struct instruction *insn)
62*4882a593Smuzhiyun {
63*4882a593Smuzhiyun 	struct instruction *next = list_next_entry(insn, list);
64*4882a593Smuzhiyun 	struct symbol *func = insn->func;
65*4882a593Smuzhiyun 
66*4882a593Smuzhiyun 	if (!func)
67*4882a593Smuzhiyun 		return NULL;
68*4882a593Smuzhiyun 
69*4882a593Smuzhiyun 	if (&next->list != &file->insn_list && next->func == func)
70*4882a593Smuzhiyun 		return next;
71*4882a593Smuzhiyun 
72*4882a593Smuzhiyun 	/* Check if we're already in the subfunction: */
73*4882a593Smuzhiyun 	if (func == func->cfunc)
74*4882a593Smuzhiyun 		return NULL;
75*4882a593Smuzhiyun 
76*4882a593Smuzhiyun 	/* Move to the subfunction: */
77*4882a593Smuzhiyun 	return find_insn(file, func->cfunc->sec, func->cfunc->offset);
78*4882a593Smuzhiyun }
79*4882a593Smuzhiyun 
prev_insn_same_sym(struct objtool_file * file,struct instruction * insn)80*4882a593Smuzhiyun static struct instruction *prev_insn_same_sym(struct objtool_file *file,
81*4882a593Smuzhiyun 					       struct instruction *insn)
82*4882a593Smuzhiyun {
83*4882a593Smuzhiyun 	struct instruction *prev = list_prev_entry(insn, list);
84*4882a593Smuzhiyun 
85*4882a593Smuzhiyun 	if (&prev->list != &file->insn_list && prev->func == insn->func)
86*4882a593Smuzhiyun 		return prev;
87*4882a593Smuzhiyun 
88*4882a593Smuzhiyun 	return NULL;
89*4882a593Smuzhiyun }
90*4882a593Smuzhiyun 
91*4882a593Smuzhiyun #define func_for_each_insn(file, func, insn)				\
92*4882a593Smuzhiyun 	for (insn = find_insn(file, func->sec, func->offset);		\
93*4882a593Smuzhiyun 	     insn;							\
94*4882a593Smuzhiyun 	     insn = next_insn_same_func(file, insn))
95*4882a593Smuzhiyun 
96*4882a593Smuzhiyun #define sym_for_each_insn(file, sym, insn)				\
97*4882a593Smuzhiyun 	for (insn = find_insn(file, sym->sec, sym->offset);		\
98*4882a593Smuzhiyun 	     insn && &insn->list != &file->insn_list &&			\
99*4882a593Smuzhiyun 		insn->sec == sym->sec &&				\
100*4882a593Smuzhiyun 		insn->offset < sym->offset + sym->len;			\
101*4882a593Smuzhiyun 	     insn = list_next_entry(insn, list))
102*4882a593Smuzhiyun 
103*4882a593Smuzhiyun #define sym_for_each_insn_continue_reverse(file, sym, insn)		\
104*4882a593Smuzhiyun 	for (insn = list_prev_entry(insn, list);			\
105*4882a593Smuzhiyun 	     &insn->list != &file->insn_list &&				\
106*4882a593Smuzhiyun 		insn->sec == sym->sec && insn->offset >= sym->offset;	\
107*4882a593Smuzhiyun 	     insn = list_prev_entry(insn, list))
108*4882a593Smuzhiyun 
109*4882a593Smuzhiyun #define sec_for_each_insn_from(file, insn)				\
110*4882a593Smuzhiyun 	for (; insn; insn = next_insn_same_sec(file, insn))
111*4882a593Smuzhiyun 
112*4882a593Smuzhiyun #define sec_for_each_insn_continue(file, insn)				\
113*4882a593Smuzhiyun 	for (insn = next_insn_same_sec(file, insn); insn;		\
114*4882a593Smuzhiyun 	     insn = next_insn_same_sec(file, insn))
115*4882a593Smuzhiyun 
is_jump_table_jump(struct instruction * insn)116*4882a593Smuzhiyun static bool is_jump_table_jump(struct instruction *insn)
117*4882a593Smuzhiyun {
118*4882a593Smuzhiyun 	struct alt_group *alt_group = insn->alt_group;
119*4882a593Smuzhiyun 
120*4882a593Smuzhiyun 	if (insn->jump_table)
121*4882a593Smuzhiyun 		return true;
122*4882a593Smuzhiyun 
123*4882a593Smuzhiyun 	/* Retpoline alternative for a jump table? */
124*4882a593Smuzhiyun 	return alt_group && alt_group->orig_group &&
125*4882a593Smuzhiyun 	       alt_group->orig_group->first_insn->jump_table;
126*4882a593Smuzhiyun }
127*4882a593Smuzhiyun 
is_sibling_call(struct instruction * insn)128*4882a593Smuzhiyun static bool is_sibling_call(struct instruction *insn)
129*4882a593Smuzhiyun {
130*4882a593Smuzhiyun 	/*
131*4882a593Smuzhiyun 	 * Assume only ELF functions can make sibling calls.  This ensures
132*4882a593Smuzhiyun 	 * sibling call detection consistency between vmlinux.o and individual
133*4882a593Smuzhiyun 	 * objects.
134*4882a593Smuzhiyun 	 */
135*4882a593Smuzhiyun 	if (!insn->func)
136*4882a593Smuzhiyun 		return false;
137*4882a593Smuzhiyun 
138*4882a593Smuzhiyun 	/* An indirect jump is either a sibling call or a jump to a table. */
139*4882a593Smuzhiyun 	if (insn->type == INSN_JUMP_DYNAMIC)
140*4882a593Smuzhiyun 		return !is_jump_table_jump(insn);
141*4882a593Smuzhiyun 
142*4882a593Smuzhiyun 	/* add_jump_destinations() sets insn->call_dest for sibling calls. */
143*4882a593Smuzhiyun 	return (is_static_jump(insn) && insn->call_dest);
144*4882a593Smuzhiyun }
145*4882a593Smuzhiyun 
146*4882a593Smuzhiyun /*
147*4882a593Smuzhiyun  * This checks to see if the given function is a "noreturn" function.
148*4882a593Smuzhiyun  *
149*4882a593Smuzhiyun  * For global functions which are outside the scope of this object file, we
150*4882a593Smuzhiyun  * have to keep a manual list of them.
151*4882a593Smuzhiyun  *
152*4882a593Smuzhiyun  * For local functions, we have to detect them manually by simply looking for
153*4882a593Smuzhiyun  * the lack of a return instruction.
154*4882a593Smuzhiyun  */
__dead_end_function(struct objtool_file * file,struct symbol * func,int recursion)155*4882a593Smuzhiyun static bool __dead_end_function(struct objtool_file *file, struct symbol *func,
156*4882a593Smuzhiyun 				int recursion)
157*4882a593Smuzhiyun {
158*4882a593Smuzhiyun 	int i;
159*4882a593Smuzhiyun 	struct instruction *insn;
160*4882a593Smuzhiyun 	bool empty = true;
161*4882a593Smuzhiyun 
162*4882a593Smuzhiyun 	/*
163*4882a593Smuzhiyun 	 * Unfortunately these have to be hard coded because the noreturn
164*4882a593Smuzhiyun 	 * attribute isn't provided in ELF data.
165*4882a593Smuzhiyun 	 */
166*4882a593Smuzhiyun 	static const char * const global_noreturns[] = {
167*4882a593Smuzhiyun 		"__stack_chk_fail",
168*4882a593Smuzhiyun 		"panic",
169*4882a593Smuzhiyun 		"do_exit",
170*4882a593Smuzhiyun 		"do_task_dead",
171*4882a593Smuzhiyun 		"__module_put_and_exit",
172*4882a593Smuzhiyun 		"complete_and_exit",
173*4882a593Smuzhiyun 		"__reiserfs_panic",
174*4882a593Smuzhiyun 		"lbug_with_loc",
175*4882a593Smuzhiyun 		"fortify_panic",
176*4882a593Smuzhiyun 		"usercopy_abort",
177*4882a593Smuzhiyun 		"machine_real_restart",
178*4882a593Smuzhiyun 		"rewind_stack_do_exit",
179*4882a593Smuzhiyun 		"kunit_try_catch_throw",
180*4882a593Smuzhiyun 		"xen_start_kernel",
181*4882a593Smuzhiyun 		"cpu_bringup_and_idle",
182*4882a593Smuzhiyun 	};
183*4882a593Smuzhiyun 
184*4882a593Smuzhiyun 	if (!func)
185*4882a593Smuzhiyun 		return false;
186*4882a593Smuzhiyun 
187*4882a593Smuzhiyun 	if (func->bind == STB_WEAK)
188*4882a593Smuzhiyun 		return false;
189*4882a593Smuzhiyun 
190*4882a593Smuzhiyun 	if (func->bind == STB_GLOBAL)
191*4882a593Smuzhiyun 		for (i = 0; i < ARRAY_SIZE(global_noreturns); i++)
192*4882a593Smuzhiyun 			if (!strcmp(func->name, global_noreturns[i]))
193*4882a593Smuzhiyun 				return true;
194*4882a593Smuzhiyun 
195*4882a593Smuzhiyun 	if (!func->len)
196*4882a593Smuzhiyun 		return false;
197*4882a593Smuzhiyun 
198*4882a593Smuzhiyun 	insn = find_insn(file, func->sec, func->offset);
199*4882a593Smuzhiyun 	if (!insn->func)
200*4882a593Smuzhiyun 		return false;
201*4882a593Smuzhiyun 
202*4882a593Smuzhiyun 	func_for_each_insn(file, func, insn) {
203*4882a593Smuzhiyun 		empty = false;
204*4882a593Smuzhiyun 
205*4882a593Smuzhiyun 		if (insn->type == INSN_RETURN)
206*4882a593Smuzhiyun 			return false;
207*4882a593Smuzhiyun 	}
208*4882a593Smuzhiyun 
209*4882a593Smuzhiyun 	if (empty)
210*4882a593Smuzhiyun 		return false;
211*4882a593Smuzhiyun 
212*4882a593Smuzhiyun 	/*
213*4882a593Smuzhiyun 	 * A function can have a sibling call instead of a return.  In that
214*4882a593Smuzhiyun 	 * case, the function's dead-end status depends on whether the target
215*4882a593Smuzhiyun 	 * of the sibling call returns.
216*4882a593Smuzhiyun 	 */
217*4882a593Smuzhiyun 	func_for_each_insn(file, func, insn) {
218*4882a593Smuzhiyun 		if (is_sibling_call(insn)) {
219*4882a593Smuzhiyun 			struct instruction *dest = insn->jump_dest;
220*4882a593Smuzhiyun 
221*4882a593Smuzhiyun 			if (!dest)
222*4882a593Smuzhiyun 				/* sibling call to another file */
223*4882a593Smuzhiyun 				return false;
224*4882a593Smuzhiyun 
225*4882a593Smuzhiyun 			/* local sibling call */
226*4882a593Smuzhiyun 			if (recursion == 5) {
227*4882a593Smuzhiyun 				/*
228*4882a593Smuzhiyun 				 * Infinite recursion: two functions have
229*4882a593Smuzhiyun 				 * sibling calls to each other.  This is a very
230*4882a593Smuzhiyun 				 * rare case.  It means they aren't dead ends.
231*4882a593Smuzhiyun 				 */
232*4882a593Smuzhiyun 				return false;
233*4882a593Smuzhiyun 			}
234*4882a593Smuzhiyun 
235*4882a593Smuzhiyun 			return __dead_end_function(file, dest->func, recursion+1);
236*4882a593Smuzhiyun 		}
237*4882a593Smuzhiyun 	}
238*4882a593Smuzhiyun 
239*4882a593Smuzhiyun 	return true;
240*4882a593Smuzhiyun }
241*4882a593Smuzhiyun 
dead_end_function(struct objtool_file * file,struct symbol * func)242*4882a593Smuzhiyun static bool dead_end_function(struct objtool_file *file, struct symbol *func)
243*4882a593Smuzhiyun {
244*4882a593Smuzhiyun 	return __dead_end_function(file, func, 0);
245*4882a593Smuzhiyun }
246*4882a593Smuzhiyun 
init_cfi_state(struct cfi_state * cfi)247*4882a593Smuzhiyun static void init_cfi_state(struct cfi_state *cfi)
248*4882a593Smuzhiyun {
249*4882a593Smuzhiyun 	int i;
250*4882a593Smuzhiyun 
251*4882a593Smuzhiyun 	for (i = 0; i < CFI_NUM_REGS; i++) {
252*4882a593Smuzhiyun 		cfi->regs[i].base = CFI_UNDEFINED;
253*4882a593Smuzhiyun 		cfi->vals[i].base = CFI_UNDEFINED;
254*4882a593Smuzhiyun 	}
255*4882a593Smuzhiyun 	cfi->cfa.base = CFI_UNDEFINED;
256*4882a593Smuzhiyun 	cfi->drap_reg = CFI_UNDEFINED;
257*4882a593Smuzhiyun 	cfi->drap_offset = -1;
258*4882a593Smuzhiyun }
259*4882a593Smuzhiyun 
init_insn_state(struct insn_state * state,struct section * sec)260*4882a593Smuzhiyun static void init_insn_state(struct insn_state *state, struct section *sec)
261*4882a593Smuzhiyun {
262*4882a593Smuzhiyun 	memset(state, 0, sizeof(*state));
263*4882a593Smuzhiyun 	init_cfi_state(&state->cfi);
264*4882a593Smuzhiyun 
265*4882a593Smuzhiyun 	/*
266*4882a593Smuzhiyun 	 * We need the full vmlinux for noinstr validation, otherwise we can
267*4882a593Smuzhiyun 	 * not correctly determine insn->call_dest->sec (external symbols do
268*4882a593Smuzhiyun 	 * not have a section).
269*4882a593Smuzhiyun 	 */
270*4882a593Smuzhiyun 	if (vmlinux && noinstr && sec)
271*4882a593Smuzhiyun 		state->noinstr = sec->noinstr;
272*4882a593Smuzhiyun }
273*4882a593Smuzhiyun 
cfi_alloc(void)274*4882a593Smuzhiyun static struct cfi_state *cfi_alloc(void)
275*4882a593Smuzhiyun {
276*4882a593Smuzhiyun 	struct cfi_state *cfi = calloc(sizeof(struct cfi_state), 1);
277*4882a593Smuzhiyun 	if (!cfi) {
278*4882a593Smuzhiyun 		WARN("calloc failed");
279*4882a593Smuzhiyun 		exit(1);
280*4882a593Smuzhiyun 	}
281*4882a593Smuzhiyun 	nr_cfi++;
282*4882a593Smuzhiyun 	return cfi;
283*4882a593Smuzhiyun }
284*4882a593Smuzhiyun 
285*4882a593Smuzhiyun static int cfi_bits;
286*4882a593Smuzhiyun static struct hlist_head *cfi_hash;
287*4882a593Smuzhiyun 
cficmp(struct cfi_state * cfi1,struct cfi_state * cfi2)288*4882a593Smuzhiyun static inline bool cficmp(struct cfi_state *cfi1, struct cfi_state *cfi2)
289*4882a593Smuzhiyun {
290*4882a593Smuzhiyun 	return memcmp((void *)cfi1 + sizeof(cfi1->hash),
291*4882a593Smuzhiyun 		      (void *)cfi2 + sizeof(cfi2->hash),
292*4882a593Smuzhiyun 		      sizeof(struct cfi_state) - sizeof(struct hlist_node));
293*4882a593Smuzhiyun }
294*4882a593Smuzhiyun 
cfi_key(struct cfi_state * cfi)295*4882a593Smuzhiyun static inline u32 cfi_key(struct cfi_state *cfi)
296*4882a593Smuzhiyun {
297*4882a593Smuzhiyun 	return jhash((void *)cfi + sizeof(cfi->hash),
298*4882a593Smuzhiyun 		     sizeof(*cfi) - sizeof(cfi->hash), 0);
299*4882a593Smuzhiyun }
300*4882a593Smuzhiyun 
cfi_hash_find_or_add(struct cfi_state * cfi)301*4882a593Smuzhiyun static struct cfi_state *cfi_hash_find_or_add(struct cfi_state *cfi)
302*4882a593Smuzhiyun {
303*4882a593Smuzhiyun 	struct hlist_head *head = &cfi_hash[hash_min(cfi_key(cfi), cfi_bits)];
304*4882a593Smuzhiyun 	struct cfi_state *obj;
305*4882a593Smuzhiyun 
306*4882a593Smuzhiyun 	hlist_for_each_entry(obj, head, hash) {
307*4882a593Smuzhiyun 		if (!cficmp(cfi, obj)) {
308*4882a593Smuzhiyun 			nr_cfi_cache++;
309*4882a593Smuzhiyun 			return obj;
310*4882a593Smuzhiyun 		}
311*4882a593Smuzhiyun 	}
312*4882a593Smuzhiyun 
313*4882a593Smuzhiyun 	obj = cfi_alloc();
314*4882a593Smuzhiyun 	*obj = *cfi;
315*4882a593Smuzhiyun 	hlist_add_head(&obj->hash, head);
316*4882a593Smuzhiyun 
317*4882a593Smuzhiyun 	return obj;
318*4882a593Smuzhiyun }
319*4882a593Smuzhiyun 
cfi_hash_add(struct cfi_state * cfi)320*4882a593Smuzhiyun static void cfi_hash_add(struct cfi_state *cfi)
321*4882a593Smuzhiyun {
322*4882a593Smuzhiyun 	struct hlist_head *head = &cfi_hash[hash_min(cfi_key(cfi), cfi_bits)];
323*4882a593Smuzhiyun 
324*4882a593Smuzhiyun 	hlist_add_head(&cfi->hash, head);
325*4882a593Smuzhiyun }
326*4882a593Smuzhiyun 
cfi_hash_alloc(void)327*4882a593Smuzhiyun static void *cfi_hash_alloc(void)
328*4882a593Smuzhiyun {
329*4882a593Smuzhiyun 	cfi_bits = vmlinux ? ELF_HASH_BITS - 3 : 13;
330*4882a593Smuzhiyun 	cfi_hash = mmap(NULL, sizeof(struct hlist_head) << cfi_bits,
331*4882a593Smuzhiyun 			PROT_READ|PROT_WRITE,
332*4882a593Smuzhiyun 			MAP_PRIVATE|MAP_ANON, -1, 0);
333*4882a593Smuzhiyun 	if (cfi_hash == (void *)-1L) {
334*4882a593Smuzhiyun 		WARN("mmap fail cfi_hash");
335*4882a593Smuzhiyun 		cfi_hash = NULL;
336*4882a593Smuzhiyun 	}  else if (stats) {
337*4882a593Smuzhiyun 		printf("cfi_bits: %d\n", cfi_bits);
338*4882a593Smuzhiyun 	}
339*4882a593Smuzhiyun 
340*4882a593Smuzhiyun 	return cfi_hash;
341*4882a593Smuzhiyun }
342*4882a593Smuzhiyun 
343*4882a593Smuzhiyun static unsigned long nr_insns;
344*4882a593Smuzhiyun static unsigned long nr_insns_visited;
345*4882a593Smuzhiyun 
346*4882a593Smuzhiyun /*
347*4882a593Smuzhiyun  * Call the arch-specific instruction decoder for all the instructions and add
348*4882a593Smuzhiyun  * them to the global instruction list.
349*4882a593Smuzhiyun  */
decode_instructions(struct objtool_file * file)350*4882a593Smuzhiyun static int decode_instructions(struct objtool_file *file)
351*4882a593Smuzhiyun {
352*4882a593Smuzhiyun 	struct section *sec;
353*4882a593Smuzhiyun 	struct symbol *func;
354*4882a593Smuzhiyun 	unsigned long offset;
355*4882a593Smuzhiyun 	struct instruction *insn;
356*4882a593Smuzhiyun 	int ret;
357*4882a593Smuzhiyun 
358*4882a593Smuzhiyun 	for_each_sec(file, sec) {
359*4882a593Smuzhiyun 
360*4882a593Smuzhiyun 		if (!(sec->sh.sh_flags & SHF_EXECINSTR))
361*4882a593Smuzhiyun 			continue;
362*4882a593Smuzhiyun 
363*4882a593Smuzhiyun 		if (strcmp(sec->name, ".altinstr_replacement") &&
364*4882a593Smuzhiyun 		    strcmp(sec->name, ".altinstr_aux") &&
365*4882a593Smuzhiyun 		    strncmp(sec->name, ".discard.", 9))
366*4882a593Smuzhiyun 			sec->text = true;
367*4882a593Smuzhiyun 
368*4882a593Smuzhiyun 		if (!strcmp(sec->name, ".noinstr.text") ||
369*4882a593Smuzhiyun 		    !strcmp(sec->name, ".entry.text") ||
370*4882a593Smuzhiyun 		    !strncmp(sec->name, ".text.__x86.", 12))
371*4882a593Smuzhiyun 			sec->noinstr = true;
372*4882a593Smuzhiyun 
373*4882a593Smuzhiyun 		for (offset = 0; offset < sec->len; offset += insn->len) {
374*4882a593Smuzhiyun 			insn = malloc(sizeof(*insn));
375*4882a593Smuzhiyun 			if (!insn) {
376*4882a593Smuzhiyun 				WARN("malloc failed");
377*4882a593Smuzhiyun 				return -1;
378*4882a593Smuzhiyun 			}
379*4882a593Smuzhiyun 			memset(insn, 0, sizeof(*insn));
380*4882a593Smuzhiyun 			INIT_LIST_HEAD(&insn->alts);
381*4882a593Smuzhiyun 			INIT_LIST_HEAD(&insn->stack_ops);
382*4882a593Smuzhiyun 
383*4882a593Smuzhiyun 			insn->sec = sec;
384*4882a593Smuzhiyun 			insn->offset = offset;
385*4882a593Smuzhiyun 
386*4882a593Smuzhiyun 			ret = arch_decode_instruction(file->elf, sec, offset,
387*4882a593Smuzhiyun 						      sec->len - offset,
388*4882a593Smuzhiyun 						      &insn->len, &insn->type,
389*4882a593Smuzhiyun 						      &insn->immediate,
390*4882a593Smuzhiyun 						      &insn->stack_ops);
391*4882a593Smuzhiyun 			if (ret)
392*4882a593Smuzhiyun 				goto err;
393*4882a593Smuzhiyun 
394*4882a593Smuzhiyun 			hash_add(file->insn_hash, &insn->hash, sec_offset_hash(sec, insn->offset));
395*4882a593Smuzhiyun 			list_add_tail(&insn->list, &file->insn_list);
396*4882a593Smuzhiyun 			nr_insns++;
397*4882a593Smuzhiyun 		}
398*4882a593Smuzhiyun 
399*4882a593Smuzhiyun 		list_for_each_entry(func, &sec->symbol_list, list) {
400*4882a593Smuzhiyun 			if (func->type != STT_FUNC || func->alias != func)
401*4882a593Smuzhiyun 				continue;
402*4882a593Smuzhiyun 
403*4882a593Smuzhiyun 			if (!find_insn(file, sec, func->offset)) {
404*4882a593Smuzhiyun 				WARN("%s(): can't find starting instruction",
405*4882a593Smuzhiyun 				     func->name);
406*4882a593Smuzhiyun 				return -1;
407*4882a593Smuzhiyun 			}
408*4882a593Smuzhiyun 
409*4882a593Smuzhiyun 			sym_for_each_insn(file, func, insn)
410*4882a593Smuzhiyun 				insn->func = func;
411*4882a593Smuzhiyun 		}
412*4882a593Smuzhiyun 	}
413*4882a593Smuzhiyun 
414*4882a593Smuzhiyun 	if (stats)
415*4882a593Smuzhiyun 		printf("nr_insns: %lu\n", nr_insns);
416*4882a593Smuzhiyun 
417*4882a593Smuzhiyun 	return 0;
418*4882a593Smuzhiyun 
419*4882a593Smuzhiyun err:
420*4882a593Smuzhiyun 	free(insn);
421*4882a593Smuzhiyun 	return ret;
422*4882a593Smuzhiyun }
423*4882a593Smuzhiyun 
find_last_insn(struct objtool_file * file,struct section * sec)424*4882a593Smuzhiyun static struct instruction *find_last_insn(struct objtool_file *file,
425*4882a593Smuzhiyun 					  struct section *sec)
426*4882a593Smuzhiyun {
427*4882a593Smuzhiyun 	struct instruction *insn = NULL;
428*4882a593Smuzhiyun 	unsigned int offset;
429*4882a593Smuzhiyun 	unsigned int end = (sec->len > 10) ? sec->len - 10 : 0;
430*4882a593Smuzhiyun 
431*4882a593Smuzhiyun 	for (offset = sec->len - 1; offset >= end && !insn; offset--)
432*4882a593Smuzhiyun 		insn = find_insn(file, sec, offset);
433*4882a593Smuzhiyun 
434*4882a593Smuzhiyun 	return insn;
435*4882a593Smuzhiyun }
436*4882a593Smuzhiyun 
437*4882a593Smuzhiyun /*
438*4882a593Smuzhiyun  * Mark "ud2" instructions and manually annotated dead ends.
439*4882a593Smuzhiyun  */
add_dead_ends(struct objtool_file * file)440*4882a593Smuzhiyun static int add_dead_ends(struct objtool_file *file)
441*4882a593Smuzhiyun {
442*4882a593Smuzhiyun 	struct section *sec;
443*4882a593Smuzhiyun 	struct reloc *reloc;
444*4882a593Smuzhiyun 	struct instruction *insn;
445*4882a593Smuzhiyun 
446*4882a593Smuzhiyun 	/*
447*4882a593Smuzhiyun 	 * By default, "ud2" is a dead end unless otherwise annotated, because
448*4882a593Smuzhiyun 	 * GCC 7 inserts it for certain divide-by-zero cases.
449*4882a593Smuzhiyun 	 */
450*4882a593Smuzhiyun 	for_each_insn(file, insn)
451*4882a593Smuzhiyun 		if (insn->type == INSN_BUG)
452*4882a593Smuzhiyun 			insn->dead_end = true;
453*4882a593Smuzhiyun 
454*4882a593Smuzhiyun 	/*
455*4882a593Smuzhiyun 	 * Check for manually annotated dead ends.
456*4882a593Smuzhiyun 	 */
457*4882a593Smuzhiyun 	sec = find_section_by_name(file->elf, ".rela.discard.unreachable");
458*4882a593Smuzhiyun 	if (!sec)
459*4882a593Smuzhiyun 		goto reachable;
460*4882a593Smuzhiyun 
461*4882a593Smuzhiyun 	list_for_each_entry(reloc, &sec->reloc_list, list) {
462*4882a593Smuzhiyun 		if (reloc->sym->type != STT_SECTION) {
463*4882a593Smuzhiyun 			WARN("unexpected relocation symbol type in %s", sec->name);
464*4882a593Smuzhiyun 			return -1;
465*4882a593Smuzhiyun 		}
466*4882a593Smuzhiyun 		insn = find_insn(file, reloc->sym->sec, reloc->addend);
467*4882a593Smuzhiyun 		if (insn)
468*4882a593Smuzhiyun 			insn = list_prev_entry(insn, list);
469*4882a593Smuzhiyun 		else if (reloc->addend == reloc->sym->sec->len) {
470*4882a593Smuzhiyun 			insn = find_last_insn(file, reloc->sym->sec);
471*4882a593Smuzhiyun 			if (!insn) {
472*4882a593Smuzhiyun 				WARN("can't find unreachable insn at %s+0x%" PRIx64,
473*4882a593Smuzhiyun 				     reloc->sym->sec->name, reloc->addend);
474*4882a593Smuzhiyun 				return -1;
475*4882a593Smuzhiyun 			}
476*4882a593Smuzhiyun 		} else {
477*4882a593Smuzhiyun 			WARN("can't find unreachable insn at %s+0x%" PRIx64,
478*4882a593Smuzhiyun 			     reloc->sym->sec->name, reloc->addend);
479*4882a593Smuzhiyun 			return -1;
480*4882a593Smuzhiyun 		}
481*4882a593Smuzhiyun 
482*4882a593Smuzhiyun 		insn->dead_end = true;
483*4882a593Smuzhiyun 	}
484*4882a593Smuzhiyun 
485*4882a593Smuzhiyun reachable:
486*4882a593Smuzhiyun 	/*
487*4882a593Smuzhiyun 	 * These manually annotated reachable checks are needed for GCC 4.4,
488*4882a593Smuzhiyun 	 * where the Linux unreachable() macro isn't supported.  In that case
489*4882a593Smuzhiyun 	 * GCC doesn't know the "ud2" is fatal, so it generates code as if it's
490*4882a593Smuzhiyun 	 * not a dead end.
491*4882a593Smuzhiyun 	 */
492*4882a593Smuzhiyun 	sec = find_section_by_name(file->elf, ".rela.discard.reachable");
493*4882a593Smuzhiyun 	if (!sec)
494*4882a593Smuzhiyun 		return 0;
495*4882a593Smuzhiyun 
496*4882a593Smuzhiyun 	list_for_each_entry(reloc, &sec->reloc_list, list) {
497*4882a593Smuzhiyun 		if (reloc->sym->type != STT_SECTION) {
498*4882a593Smuzhiyun 			WARN("unexpected relocation symbol type in %s", sec->name);
499*4882a593Smuzhiyun 			return -1;
500*4882a593Smuzhiyun 		}
501*4882a593Smuzhiyun 		insn = find_insn(file, reloc->sym->sec, reloc->addend);
502*4882a593Smuzhiyun 		if (insn)
503*4882a593Smuzhiyun 			insn = list_prev_entry(insn, list);
504*4882a593Smuzhiyun 		else if (reloc->addend == reloc->sym->sec->len) {
505*4882a593Smuzhiyun 			insn = find_last_insn(file, reloc->sym->sec);
506*4882a593Smuzhiyun 			if (!insn) {
507*4882a593Smuzhiyun 				WARN("can't find reachable insn at %s+0x%" PRIx64,
508*4882a593Smuzhiyun 				     reloc->sym->sec->name, reloc->addend);
509*4882a593Smuzhiyun 				return -1;
510*4882a593Smuzhiyun 			}
511*4882a593Smuzhiyun 		} else {
512*4882a593Smuzhiyun 			WARN("can't find reachable insn at %s+0x%" PRIx64,
513*4882a593Smuzhiyun 			     reloc->sym->sec->name, reloc->addend);
514*4882a593Smuzhiyun 			return -1;
515*4882a593Smuzhiyun 		}
516*4882a593Smuzhiyun 
517*4882a593Smuzhiyun 		insn->dead_end = false;
518*4882a593Smuzhiyun 	}
519*4882a593Smuzhiyun 
520*4882a593Smuzhiyun 	return 0;
521*4882a593Smuzhiyun }
522*4882a593Smuzhiyun 
create_static_call_sections(struct objtool_file * file)523*4882a593Smuzhiyun static int create_static_call_sections(struct objtool_file *file)
524*4882a593Smuzhiyun {
525*4882a593Smuzhiyun 	struct section *sec;
526*4882a593Smuzhiyun 	struct static_call_site *site;
527*4882a593Smuzhiyun 	struct instruction *insn;
528*4882a593Smuzhiyun 	struct symbol *key_sym;
529*4882a593Smuzhiyun 	char *key_name, *tmp;
530*4882a593Smuzhiyun 	int idx;
531*4882a593Smuzhiyun 
532*4882a593Smuzhiyun 	sec = find_section_by_name(file->elf, ".static_call_sites");
533*4882a593Smuzhiyun 	if (sec) {
534*4882a593Smuzhiyun 		INIT_LIST_HEAD(&file->static_call_list);
535*4882a593Smuzhiyun 		WARN("file already has .static_call_sites section, skipping");
536*4882a593Smuzhiyun 		return 0;
537*4882a593Smuzhiyun 	}
538*4882a593Smuzhiyun 
539*4882a593Smuzhiyun 	if (list_empty(&file->static_call_list))
540*4882a593Smuzhiyun 		return 0;
541*4882a593Smuzhiyun 
542*4882a593Smuzhiyun 	idx = 0;
543*4882a593Smuzhiyun 	list_for_each_entry(insn, &file->static_call_list, call_node)
544*4882a593Smuzhiyun 		idx++;
545*4882a593Smuzhiyun 
546*4882a593Smuzhiyun 	sec = elf_create_section(file->elf, ".static_call_sites", SHF_WRITE,
547*4882a593Smuzhiyun 				 sizeof(struct static_call_site), idx);
548*4882a593Smuzhiyun 	if (!sec)
549*4882a593Smuzhiyun 		return -1;
550*4882a593Smuzhiyun 
551*4882a593Smuzhiyun 	idx = 0;
552*4882a593Smuzhiyun 	list_for_each_entry(insn, &file->static_call_list, call_node) {
553*4882a593Smuzhiyun 
554*4882a593Smuzhiyun 		site = (struct static_call_site *)sec->data->d_buf + idx;
555*4882a593Smuzhiyun 		memset(site, 0, sizeof(struct static_call_site));
556*4882a593Smuzhiyun 
557*4882a593Smuzhiyun 		/* populate reloc for 'addr' */
558*4882a593Smuzhiyun 		if (elf_add_reloc_to_insn(file->elf, sec,
559*4882a593Smuzhiyun 					  idx * sizeof(struct static_call_site),
560*4882a593Smuzhiyun 					  R_X86_64_PC32,
561*4882a593Smuzhiyun 					  insn->sec, insn->offset))
562*4882a593Smuzhiyun 			return -1;
563*4882a593Smuzhiyun 
564*4882a593Smuzhiyun 		/* find key symbol */
565*4882a593Smuzhiyun 		key_name = strdup(insn->call_dest->name);
566*4882a593Smuzhiyun 		if (!key_name) {
567*4882a593Smuzhiyun 			perror("strdup");
568*4882a593Smuzhiyun 			return -1;
569*4882a593Smuzhiyun 		}
570*4882a593Smuzhiyun 		if (strncmp(key_name, STATIC_CALL_TRAMP_PREFIX_STR,
571*4882a593Smuzhiyun 			    STATIC_CALL_TRAMP_PREFIX_LEN)) {
572*4882a593Smuzhiyun 			WARN("static_call: trampoline name malformed: %s", key_name);
573*4882a593Smuzhiyun 			return -1;
574*4882a593Smuzhiyun 		}
575*4882a593Smuzhiyun 		tmp = key_name + STATIC_CALL_TRAMP_PREFIX_LEN - STATIC_CALL_KEY_PREFIX_LEN;
576*4882a593Smuzhiyun 		memcpy(tmp, STATIC_CALL_KEY_PREFIX_STR, STATIC_CALL_KEY_PREFIX_LEN);
577*4882a593Smuzhiyun 
578*4882a593Smuzhiyun 		key_sym = find_symbol_by_name(file->elf, tmp);
579*4882a593Smuzhiyun 		if (!key_sym) {
580*4882a593Smuzhiyun 			if (!module) {
581*4882a593Smuzhiyun 				WARN("static_call: can't find static_call_key symbol: %s", tmp);
582*4882a593Smuzhiyun 				return -1;
583*4882a593Smuzhiyun 			}
584*4882a593Smuzhiyun 
585*4882a593Smuzhiyun 			/*
586*4882a593Smuzhiyun 			 * For modules(), the key might not be exported, which
587*4882a593Smuzhiyun 			 * means the module can make static calls but isn't
588*4882a593Smuzhiyun 			 * allowed to change them.
589*4882a593Smuzhiyun 			 *
590*4882a593Smuzhiyun 			 * In that case we temporarily set the key to be the
591*4882a593Smuzhiyun 			 * trampoline address.  This is fixed up in
592*4882a593Smuzhiyun 			 * static_call_add_module().
593*4882a593Smuzhiyun 			 */
594*4882a593Smuzhiyun 			key_sym = insn->call_dest;
595*4882a593Smuzhiyun 		}
596*4882a593Smuzhiyun 		free(key_name);
597*4882a593Smuzhiyun 
598*4882a593Smuzhiyun 		/* populate reloc for 'key' */
599*4882a593Smuzhiyun 		if (elf_add_reloc(file->elf, sec,
600*4882a593Smuzhiyun 				  idx * sizeof(struct static_call_site) + 4,
601*4882a593Smuzhiyun 				  R_X86_64_PC32, key_sym,
602*4882a593Smuzhiyun 				  is_sibling_call(insn) * STATIC_CALL_SITE_TAIL))
603*4882a593Smuzhiyun 			return -1;
604*4882a593Smuzhiyun 
605*4882a593Smuzhiyun 		idx++;
606*4882a593Smuzhiyun 	}
607*4882a593Smuzhiyun 
608*4882a593Smuzhiyun 	return 0;
609*4882a593Smuzhiyun }
610*4882a593Smuzhiyun 
create_retpoline_sites_sections(struct objtool_file * file)611*4882a593Smuzhiyun static int create_retpoline_sites_sections(struct objtool_file *file)
612*4882a593Smuzhiyun {
613*4882a593Smuzhiyun 	struct instruction *insn;
614*4882a593Smuzhiyun 	struct section *sec;
615*4882a593Smuzhiyun 	int idx;
616*4882a593Smuzhiyun 
617*4882a593Smuzhiyun 	sec = find_section_by_name(file->elf, ".retpoline_sites");
618*4882a593Smuzhiyun 	if (sec) {
619*4882a593Smuzhiyun 		WARN("file already has .retpoline_sites, skipping");
620*4882a593Smuzhiyun 		return 0;
621*4882a593Smuzhiyun 	}
622*4882a593Smuzhiyun 
623*4882a593Smuzhiyun 	idx = 0;
624*4882a593Smuzhiyun 	list_for_each_entry(insn, &file->retpoline_call_list, call_node)
625*4882a593Smuzhiyun 		idx++;
626*4882a593Smuzhiyun 
627*4882a593Smuzhiyun 	if (!idx)
628*4882a593Smuzhiyun 		return 0;
629*4882a593Smuzhiyun 
630*4882a593Smuzhiyun 	sec = elf_create_section(file->elf, ".retpoline_sites", 0,
631*4882a593Smuzhiyun 				 sizeof(int), idx);
632*4882a593Smuzhiyun 	if (!sec) {
633*4882a593Smuzhiyun 		WARN("elf_create_section: .retpoline_sites");
634*4882a593Smuzhiyun 		return -1;
635*4882a593Smuzhiyun 	}
636*4882a593Smuzhiyun 
637*4882a593Smuzhiyun 	idx = 0;
638*4882a593Smuzhiyun 	list_for_each_entry(insn, &file->retpoline_call_list, call_node) {
639*4882a593Smuzhiyun 
640*4882a593Smuzhiyun 		int *site = (int *)sec->data->d_buf + idx;
641*4882a593Smuzhiyun 		*site = 0;
642*4882a593Smuzhiyun 
643*4882a593Smuzhiyun 		if (elf_add_reloc_to_insn(file->elf, sec,
644*4882a593Smuzhiyun 					  idx * sizeof(int),
645*4882a593Smuzhiyun 					  R_X86_64_PC32,
646*4882a593Smuzhiyun 					  insn->sec, insn->offset)) {
647*4882a593Smuzhiyun 			WARN("elf_add_reloc_to_insn: .retpoline_sites");
648*4882a593Smuzhiyun 			return -1;
649*4882a593Smuzhiyun 		}
650*4882a593Smuzhiyun 
651*4882a593Smuzhiyun 		idx++;
652*4882a593Smuzhiyun 	}
653*4882a593Smuzhiyun 
654*4882a593Smuzhiyun 	return 0;
655*4882a593Smuzhiyun }
656*4882a593Smuzhiyun 
create_return_sites_sections(struct objtool_file * file)657*4882a593Smuzhiyun static int create_return_sites_sections(struct objtool_file *file)
658*4882a593Smuzhiyun {
659*4882a593Smuzhiyun 	struct instruction *insn;
660*4882a593Smuzhiyun 	struct section *sec;
661*4882a593Smuzhiyun 	int idx;
662*4882a593Smuzhiyun 
663*4882a593Smuzhiyun 	sec = find_section_by_name(file->elf, ".return_sites");
664*4882a593Smuzhiyun 	if (sec) {
665*4882a593Smuzhiyun 		WARN("file already has .return_sites, skipping");
666*4882a593Smuzhiyun 		return 0;
667*4882a593Smuzhiyun 	}
668*4882a593Smuzhiyun 
669*4882a593Smuzhiyun 	idx = 0;
670*4882a593Smuzhiyun 	list_for_each_entry(insn, &file->return_thunk_list, call_node)
671*4882a593Smuzhiyun 		idx++;
672*4882a593Smuzhiyun 
673*4882a593Smuzhiyun 	if (!idx)
674*4882a593Smuzhiyun 		return 0;
675*4882a593Smuzhiyun 
676*4882a593Smuzhiyun 	sec = elf_create_section(file->elf, ".return_sites", 0,
677*4882a593Smuzhiyun 				 sizeof(int), idx);
678*4882a593Smuzhiyun 	if (!sec) {
679*4882a593Smuzhiyun 		WARN("elf_create_section: .return_sites");
680*4882a593Smuzhiyun 		return -1;
681*4882a593Smuzhiyun 	}
682*4882a593Smuzhiyun 
683*4882a593Smuzhiyun 	idx = 0;
684*4882a593Smuzhiyun 	list_for_each_entry(insn, &file->return_thunk_list, call_node) {
685*4882a593Smuzhiyun 
686*4882a593Smuzhiyun 		int *site = (int *)sec->data->d_buf + idx;
687*4882a593Smuzhiyun 		*site = 0;
688*4882a593Smuzhiyun 
689*4882a593Smuzhiyun 		if (elf_add_reloc_to_insn(file->elf, sec,
690*4882a593Smuzhiyun 					  idx * sizeof(int),
691*4882a593Smuzhiyun 					  R_X86_64_PC32,
692*4882a593Smuzhiyun 					  insn->sec, insn->offset)) {
693*4882a593Smuzhiyun 			WARN("elf_add_reloc_to_insn: .return_sites");
694*4882a593Smuzhiyun 			return -1;
695*4882a593Smuzhiyun 		}
696*4882a593Smuzhiyun 
697*4882a593Smuzhiyun 		idx++;
698*4882a593Smuzhiyun 	}
699*4882a593Smuzhiyun 
700*4882a593Smuzhiyun 	return 0;
701*4882a593Smuzhiyun }
702*4882a593Smuzhiyun 
create_mcount_loc_sections(struct objtool_file * file)703*4882a593Smuzhiyun static int create_mcount_loc_sections(struct objtool_file *file)
704*4882a593Smuzhiyun {
705*4882a593Smuzhiyun 	struct section *sec;
706*4882a593Smuzhiyun 	unsigned long *loc;
707*4882a593Smuzhiyun 	struct instruction *insn;
708*4882a593Smuzhiyun 	int idx;
709*4882a593Smuzhiyun 
710*4882a593Smuzhiyun 	sec = find_section_by_name(file->elf, "__mcount_loc");
711*4882a593Smuzhiyun 	if (sec) {
712*4882a593Smuzhiyun 		INIT_LIST_HEAD(&file->mcount_loc_list);
713*4882a593Smuzhiyun 		WARN("file already has __mcount_loc section, skipping");
714*4882a593Smuzhiyun 		return 0;
715*4882a593Smuzhiyun 	}
716*4882a593Smuzhiyun 
717*4882a593Smuzhiyun 	if (list_empty(&file->mcount_loc_list))
718*4882a593Smuzhiyun 		return 0;
719*4882a593Smuzhiyun 
720*4882a593Smuzhiyun 	idx = 0;
721*4882a593Smuzhiyun 	list_for_each_entry(insn, &file->mcount_loc_list, mcount_loc_node)
722*4882a593Smuzhiyun 		idx++;
723*4882a593Smuzhiyun 
724*4882a593Smuzhiyun 	sec = elf_create_section(file->elf, "__mcount_loc", 0, sizeof(unsigned long), idx);
725*4882a593Smuzhiyun 	if (!sec)
726*4882a593Smuzhiyun 		return -1;
727*4882a593Smuzhiyun 
728*4882a593Smuzhiyun 	idx = 0;
729*4882a593Smuzhiyun 	list_for_each_entry(insn, &file->mcount_loc_list, mcount_loc_node) {
730*4882a593Smuzhiyun 
731*4882a593Smuzhiyun 		loc = (unsigned long *)sec->data->d_buf + idx;
732*4882a593Smuzhiyun 		memset(loc, 0, sizeof(unsigned long));
733*4882a593Smuzhiyun 
734*4882a593Smuzhiyun 		if (elf_add_reloc_to_insn(file->elf, sec,
735*4882a593Smuzhiyun 					  idx * sizeof(unsigned long),
736*4882a593Smuzhiyun 					  R_X86_64_64,
737*4882a593Smuzhiyun 					  insn->sec, insn->offset))
738*4882a593Smuzhiyun 			return -1;
739*4882a593Smuzhiyun 
740*4882a593Smuzhiyun 		idx++;
741*4882a593Smuzhiyun 	}
742*4882a593Smuzhiyun 
743*4882a593Smuzhiyun 	return 0;
744*4882a593Smuzhiyun }
745*4882a593Smuzhiyun 
746*4882a593Smuzhiyun /*
747*4882a593Smuzhiyun  * Warnings shouldn't be reported for ignored functions.
748*4882a593Smuzhiyun  */
add_ignores(struct objtool_file * file)749*4882a593Smuzhiyun static void add_ignores(struct objtool_file *file)
750*4882a593Smuzhiyun {
751*4882a593Smuzhiyun 	struct instruction *insn;
752*4882a593Smuzhiyun 	struct section *sec;
753*4882a593Smuzhiyun 	struct symbol *func;
754*4882a593Smuzhiyun 	struct reloc *reloc;
755*4882a593Smuzhiyun 
756*4882a593Smuzhiyun 	sec = find_section_by_name(file->elf, ".rela.discard.func_stack_frame_non_standard");
757*4882a593Smuzhiyun 	if (!sec)
758*4882a593Smuzhiyun 		return;
759*4882a593Smuzhiyun 
760*4882a593Smuzhiyun 	list_for_each_entry(reloc, &sec->reloc_list, list) {
761*4882a593Smuzhiyun 		switch (reloc->sym->type) {
762*4882a593Smuzhiyun 		case STT_FUNC:
763*4882a593Smuzhiyun 			func = reloc->sym;
764*4882a593Smuzhiyun 			break;
765*4882a593Smuzhiyun 
766*4882a593Smuzhiyun 		case STT_SECTION:
767*4882a593Smuzhiyun 			func = find_func_by_offset(reloc->sym->sec, reloc->addend);
768*4882a593Smuzhiyun 			if (!func)
769*4882a593Smuzhiyun 				continue;
770*4882a593Smuzhiyun 			break;
771*4882a593Smuzhiyun 
772*4882a593Smuzhiyun 		default:
773*4882a593Smuzhiyun 			WARN("unexpected relocation symbol type in %s: %d", sec->name, reloc->sym->type);
774*4882a593Smuzhiyun 			continue;
775*4882a593Smuzhiyun 		}
776*4882a593Smuzhiyun 
777*4882a593Smuzhiyun 		func_for_each_insn(file, func, insn)
778*4882a593Smuzhiyun 			insn->ignore = true;
779*4882a593Smuzhiyun 	}
780*4882a593Smuzhiyun }
781*4882a593Smuzhiyun 
782*4882a593Smuzhiyun /*
783*4882a593Smuzhiyun  * This is a whitelist of functions that is allowed to be called with AC set.
784*4882a593Smuzhiyun  * The list is meant to be minimal and only contains compiler instrumentation
785*4882a593Smuzhiyun  * ABI and a few functions used to implement *_{to,from}_user() functions.
786*4882a593Smuzhiyun  *
787*4882a593Smuzhiyun  * These functions must not directly change AC, but may PUSHF/POPF.
788*4882a593Smuzhiyun  */
789*4882a593Smuzhiyun static const char *uaccess_safe_builtin[] = {
790*4882a593Smuzhiyun 	/* KASAN */
791*4882a593Smuzhiyun 	"kasan_report",
792*4882a593Smuzhiyun 	"kasan_check_range",
793*4882a593Smuzhiyun 	/* KASAN out-of-line */
794*4882a593Smuzhiyun 	"__asan_loadN_noabort",
795*4882a593Smuzhiyun 	"__asan_load1_noabort",
796*4882a593Smuzhiyun 	"__asan_load2_noabort",
797*4882a593Smuzhiyun 	"__asan_load4_noabort",
798*4882a593Smuzhiyun 	"__asan_load8_noabort",
799*4882a593Smuzhiyun 	"__asan_load16_noabort",
800*4882a593Smuzhiyun 	"__asan_storeN_noabort",
801*4882a593Smuzhiyun 	"__asan_store1_noabort",
802*4882a593Smuzhiyun 	"__asan_store2_noabort",
803*4882a593Smuzhiyun 	"__asan_store4_noabort",
804*4882a593Smuzhiyun 	"__asan_store8_noabort",
805*4882a593Smuzhiyun 	"__asan_store16_noabort",
806*4882a593Smuzhiyun 	"__kasan_check_read",
807*4882a593Smuzhiyun 	"__kasan_check_write",
808*4882a593Smuzhiyun 	/* KASAN in-line */
809*4882a593Smuzhiyun 	"__asan_report_load_n_noabort",
810*4882a593Smuzhiyun 	"__asan_report_load1_noabort",
811*4882a593Smuzhiyun 	"__asan_report_load2_noabort",
812*4882a593Smuzhiyun 	"__asan_report_load4_noabort",
813*4882a593Smuzhiyun 	"__asan_report_load8_noabort",
814*4882a593Smuzhiyun 	"__asan_report_load16_noabort",
815*4882a593Smuzhiyun 	"__asan_report_store_n_noabort",
816*4882a593Smuzhiyun 	"__asan_report_store1_noabort",
817*4882a593Smuzhiyun 	"__asan_report_store2_noabort",
818*4882a593Smuzhiyun 	"__asan_report_store4_noabort",
819*4882a593Smuzhiyun 	"__asan_report_store8_noabort",
820*4882a593Smuzhiyun 	"__asan_report_store16_noabort",
821*4882a593Smuzhiyun 	/* KCSAN */
822*4882a593Smuzhiyun 	"__kcsan_check_access",
823*4882a593Smuzhiyun 	"kcsan_found_watchpoint",
824*4882a593Smuzhiyun 	"kcsan_setup_watchpoint",
825*4882a593Smuzhiyun 	"kcsan_check_scoped_accesses",
826*4882a593Smuzhiyun 	"kcsan_disable_current",
827*4882a593Smuzhiyun 	"kcsan_enable_current_nowarn",
828*4882a593Smuzhiyun 	/* KCSAN/TSAN */
829*4882a593Smuzhiyun 	"__tsan_func_entry",
830*4882a593Smuzhiyun 	"__tsan_func_exit",
831*4882a593Smuzhiyun 	"__tsan_read_range",
832*4882a593Smuzhiyun 	"__tsan_write_range",
833*4882a593Smuzhiyun 	"__tsan_read1",
834*4882a593Smuzhiyun 	"__tsan_read2",
835*4882a593Smuzhiyun 	"__tsan_read4",
836*4882a593Smuzhiyun 	"__tsan_read8",
837*4882a593Smuzhiyun 	"__tsan_read16",
838*4882a593Smuzhiyun 	"__tsan_write1",
839*4882a593Smuzhiyun 	"__tsan_write2",
840*4882a593Smuzhiyun 	"__tsan_write4",
841*4882a593Smuzhiyun 	"__tsan_write8",
842*4882a593Smuzhiyun 	"__tsan_write16",
843*4882a593Smuzhiyun 	"__tsan_read_write1",
844*4882a593Smuzhiyun 	"__tsan_read_write2",
845*4882a593Smuzhiyun 	"__tsan_read_write4",
846*4882a593Smuzhiyun 	"__tsan_read_write8",
847*4882a593Smuzhiyun 	"__tsan_read_write16",
848*4882a593Smuzhiyun 	"__tsan_atomic8_load",
849*4882a593Smuzhiyun 	"__tsan_atomic16_load",
850*4882a593Smuzhiyun 	"__tsan_atomic32_load",
851*4882a593Smuzhiyun 	"__tsan_atomic64_load",
852*4882a593Smuzhiyun 	"__tsan_atomic8_store",
853*4882a593Smuzhiyun 	"__tsan_atomic16_store",
854*4882a593Smuzhiyun 	"__tsan_atomic32_store",
855*4882a593Smuzhiyun 	"__tsan_atomic64_store",
856*4882a593Smuzhiyun 	"__tsan_atomic8_exchange",
857*4882a593Smuzhiyun 	"__tsan_atomic16_exchange",
858*4882a593Smuzhiyun 	"__tsan_atomic32_exchange",
859*4882a593Smuzhiyun 	"__tsan_atomic64_exchange",
860*4882a593Smuzhiyun 	"__tsan_atomic8_fetch_add",
861*4882a593Smuzhiyun 	"__tsan_atomic16_fetch_add",
862*4882a593Smuzhiyun 	"__tsan_atomic32_fetch_add",
863*4882a593Smuzhiyun 	"__tsan_atomic64_fetch_add",
864*4882a593Smuzhiyun 	"__tsan_atomic8_fetch_sub",
865*4882a593Smuzhiyun 	"__tsan_atomic16_fetch_sub",
866*4882a593Smuzhiyun 	"__tsan_atomic32_fetch_sub",
867*4882a593Smuzhiyun 	"__tsan_atomic64_fetch_sub",
868*4882a593Smuzhiyun 	"__tsan_atomic8_fetch_and",
869*4882a593Smuzhiyun 	"__tsan_atomic16_fetch_and",
870*4882a593Smuzhiyun 	"__tsan_atomic32_fetch_and",
871*4882a593Smuzhiyun 	"__tsan_atomic64_fetch_and",
872*4882a593Smuzhiyun 	"__tsan_atomic8_fetch_or",
873*4882a593Smuzhiyun 	"__tsan_atomic16_fetch_or",
874*4882a593Smuzhiyun 	"__tsan_atomic32_fetch_or",
875*4882a593Smuzhiyun 	"__tsan_atomic64_fetch_or",
876*4882a593Smuzhiyun 	"__tsan_atomic8_fetch_xor",
877*4882a593Smuzhiyun 	"__tsan_atomic16_fetch_xor",
878*4882a593Smuzhiyun 	"__tsan_atomic32_fetch_xor",
879*4882a593Smuzhiyun 	"__tsan_atomic64_fetch_xor",
880*4882a593Smuzhiyun 	"__tsan_atomic8_fetch_nand",
881*4882a593Smuzhiyun 	"__tsan_atomic16_fetch_nand",
882*4882a593Smuzhiyun 	"__tsan_atomic32_fetch_nand",
883*4882a593Smuzhiyun 	"__tsan_atomic64_fetch_nand",
884*4882a593Smuzhiyun 	"__tsan_atomic8_compare_exchange_strong",
885*4882a593Smuzhiyun 	"__tsan_atomic16_compare_exchange_strong",
886*4882a593Smuzhiyun 	"__tsan_atomic32_compare_exchange_strong",
887*4882a593Smuzhiyun 	"__tsan_atomic64_compare_exchange_strong",
888*4882a593Smuzhiyun 	"__tsan_atomic8_compare_exchange_weak",
889*4882a593Smuzhiyun 	"__tsan_atomic16_compare_exchange_weak",
890*4882a593Smuzhiyun 	"__tsan_atomic32_compare_exchange_weak",
891*4882a593Smuzhiyun 	"__tsan_atomic64_compare_exchange_weak",
892*4882a593Smuzhiyun 	"__tsan_atomic8_compare_exchange_val",
893*4882a593Smuzhiyun 	"__tsan_atomic16_compare_exchange_val",
894*4882a593Smuzhiyun 	"__tsan_atomic32_compare_exchange_val",
895*4882a593Smuzhiyun 	"__tsan_atomic64_compare_exchange_val",
896*4882a593Smuzhiyun 	"__tsan_atomic_thread_fence",
897*4882a593Smuzhiyun 	"__tsan_atomic_signal_fence",
898*4882a593Smuzhiyun 	/* KCOV */
899*4882a593Smuzhiyun 	"write_comp_data",
900*4882a593Smuzhiyun 	"check_kcov_mode",
901*4882a593Smuzhiyun 	"__sanitizer_cov_trace_pc",
902*4882a593Smuzhiyun 	"__sanitizer_cov_trace_const_cmp1",
903*4882a593Smuzhiyun 	"__sanitizer_cov_trace_const_cmp2",
904*4882a593Smuzhiyun 	"__sanitizer_cov_trace_const_cmp4",
905*4882a593Smuzhiyun 	"__sanitizer_cov_trace_const_cmp8",
906*4882a593Smuzhiyun 	"__sanitizer_cov_trace_cmp1",
907*4882a593Smuzhiyun 	"__sanitizer_cov_trace_cmp2",
908*4882a593Smuzhiyun 	"__sanitizer_cov_trace_cmp4",
909*4882a593Smuzhiyun 	"__sanitizer_cov_trace_cmp8",
910*4882a593Smuzhiyun 	"__sanitizer_cov_trace_switch",
911*4882a593Smuzhiyun 	/* UBSAN */
912*4882a593Smuzhiyun 	"ubsan_type_mismatch_common",
913*4882a593Smuzhiyun 	"__ubsan_handle_type_mismatch",
914*4882a593Smuzhiyun 	"__ubsan_handle_type_mismatch_v1",
915*4882a593Smuzhiyun 	"__ubsan_handle_shift_out_of_bounds",
916*4882a593Smuzhiyun 	/* misc */
917*4882a593Smuzhiyun 	"csum_partial_copy_generic",
918*4882a593Smuzhiyun 	"copy_mc_fragile",
919*4882a593Smuzhiyun 	"copy_mc_fragile_handle_tail",
920*4882a593Smuzhiyun 	"copy_mc_enhanced_fast_string",
921*4882a593Smuzhiyun 	"ftrace_likely_update", /* CONFIG_TRACE_BRANCH_PROFILING */
922*4882a593Smuzhiyun 	NULL
923*4882a593Smuzhiyun };
924*4882a593Smuzhiyun 
add_uaccess_safe(struct objtool_file * file)925*4882a593Smuzhiyun static void add_uaccess_safe(struct objtool_file *file)
926*4882a593Smuzhiyun {
927*4882a593Smuzhiyun 	struct symbol *func;
928*4882a593Smuzhiyun 	const char **name;
929*4882a593Smuzhiyun 
930*4882a593Smuzhiyun 	if (!uaccess)
931*4882a593Smuzhiyun 		return;
932*4882a593Smuzhiyun 
933*4882a593Smuzhiyun 	for (name = uaccess_safe_builtin; *name; name++) {
934*4882a593Smuzhiyun 		func = find_symbol_by_name(file->elf, *name);
935*4882a593Smuzhiyun 		if (!func)
936*4882a593Smuzhiyun 			continue;
937*4882a593Smuzhiyun 
938*4882a593Smuzhiyun 		func->uaccess_safe = true;
939*4882a593Smuzhiyun 	}
940*4882a593Smuzhiyun }
941*4882a593Smuzhiyun 
942*4882a593Smuzhiyun /*
943*4882a593Smuzhiyun  * FIXME: For now, just ignore any alternatives which add retpolines.  This is
944*4882a593Smuzhiyun  * a temporary hack, as it doesn't allow ORC to unwind from inside a retpoline.
945*4882a593Smuzhiyun  * But it at least allows objtool to understand the control flow *around* the
946*4882a593Smuzhiyun  * retpoline.
947*4882a593Smuzhiyun  */
add_ignore_alternatives(struct objtool_file * file)948*4882a593Smuzhiyun static int add_ignore_alternatives(struct objtool_file *file)
949*4882a593Smuzhiyun {
950*4882a593Smuzhiyun 	struct section *sec;
951*4882a593Smuzhiyun 	struct reloc *reloc;
952*4882a593Smuzhiyun 	struct instruction *insn;
953*4882a593Smuzhiyun 
954*4882a593Smuzhiyun 	sec = find_section_by_name(file->elf, ".rela.discard.ignore_alts");
955*4882a593Smuzhiyun 	if (!sec)
956*4882a593Smuzhiyun 		return 0;
957*4882a593Smuzhiyun 
958*4882a593Smuzhiyun 	list_for_each_entry(reloc, &sec->reloc_list, list) {
959*4882a593Smuzhiyun 		if (reloc->sym->type != STT_SECTION) {
960*4882a593Smuzhiyun 			WARN("unexpected relocation symbol type in %s", sec->name);
961*4882a593Smuzhiyun 			return -1;
962*4882a593Smuzhiyun 		}
963*4882a593Smuzhiyun 
964*4882a593Smuzhiyun 		insn = find_insn(file, reloc->sym->sec, reloc->addend);
965*4882a593Smuzhiyun 		if (!insn) {
966*4882a593Smuzhiyun 			WARN("bad .discard.ignore_alts entry");
967*4882a593Smuzhiyun 			return -1;
968*4882a593Smuzhiyun 		}
969*4882a593Smuzhiyun 
970*4882a593Smuzhiyun 		insn->ignore_alts = true;
971*4882a593Smuzhiyun 	}
972*4882a593Smuzhiyun 
973*4882a593Smuzhiyun 	return 0;
974*4882a593Smuzhiyun }
975*4882a593Smuzhiyun 
arch_is_retpoline(struct symbol * sym)976*4882a593Smuzhiyun __weak bool arch_is_retpoline(struct symbol *sym)
977*4882a593Smuzhiyun {
978*4882a593Smuzhiyun 	return false;
979*4882a593Smuzhiyun }
980*4882a593Smuzhiyun 
arch_is_rethunk(struct symbol * sym)981*4882a593Smuzhiyun __weak bool arch_is_rethunk(struct symbol *sym)
982*4882a593Smuzhiyun {
983*4882a593Smuzhiyun 	return false;
984*4882a593Smuzhiyun }
985*4882a593Smuzhiyun 
986*4882a593Smuzhiyun #define NEGATIVE_RELOC	((void *)-1L)
987*4882a593Smuzhiyun 
insn_reloc(struct objtool_file * file,struct instruction * insn)988*4882a593Smuzhiyun static struct reloc *insn_reloc(struct objtool_file *file, struct instruction *insn)
989*4882a593Smuzhiyun {
990*4882a593Smuzhiyun 	if (insn->reloc == NEGATIVE_RELOC)
991*4882a593Smuzhiyun 		return NULL;
992*4882a593Smuzhiyun 
993*4882a593Smuzhiyun 	if (!insn->reloc) {
994*4882a593Smuzhiyun 		insn->reloc = find_reloc_by_dest_range(file->elf, insn->sec,
995*4882a593Smuzhiyun 						       insn->offset, insn->len);
996*4882a593Smuzhiyun 		if (!insn->reloc) {
997*4882a593Smuzhiyun 			insn->reloc = NEGATIVE_RELOC;
998*4882a593Smuzhiyun 			return NULL;
999*4882a593Smuzhiyun 		}
1000*4882a593Smuzhiyun 	}
1001*4882a593Smuzhiyun 
1002*4882a593Smuzhiyun 	return insn->reloc;
1003*4882a593Smuzhiyun }
1004*4882a593Smuzhiyun 
remove_insn_ops(struct instruction * insn)1005*4882a593Smuzhiyun static void remove_insn_ops(struct instruction *insn)
1006*4882a593Smuzhiyun {
1007*4882a593Smuzhiyun 	struct stack_op *op, *tmp;
1008*4882a593Smuzhiyun 
1009*4882a593Smuzhiyun 	list_for_each_entry_safe(op, tmp, &insn->stack_ops, list) {
1010*4882a593Smuzhiyun 		list_del(&op->list);
1011*4882a593Smuzhiyun 		free(op);
1012*4882a593Smuzhiyun 	}
1013*4882a593Smuzhiyun }
1014*4882a593Smuzhiyun 
annotate_call_site(struct objtool_file * file,struct instruction * insn,bool sibling)1015*4882a593Smuzhiyun static void annotate_call_site(struct objtool_file *file,
1016*4882a593Smuzhiyun 			       struct instruction *insn, bool sibling)
1017*4882a593Smuzhiyun {
1018*4882a593Smuzhiyun 	struct reloc *reloc = insn_reloc(file, insn);
1019*4882a593Smuzhiyun 	struct symbol *sym = insn->call_dest;
1020*4882a593Smuzhiyun 
1021*4882a593Smuzhiyun 	if (!sym)
1022*4882a593Smuzhiyun 		sym = reloc->sym;
1023*4882a593Smuzhiyun 
1024*4882a593Smuzhiyun 	/*
1025*4882a593Smuzhiyun 	 * Alternative replacement code is just template code which is
1026*4882a593Smuzhiyun 	 * sometimes copied to the original instruction. For now, don't
1027*4882a593Smuzhiyun 	 * annotate it. (In the future we might consider annotating the
1028*4882a593Smuzhiyun 	 * original instruction if/when it ever makes sense to do so.)
1029*4882a593Smuzhiyun 	 */
1030*4882a593Smuzhiyun 	if (!strcmp(insn->sec->name, ".altinstr_replacement"))
1031*4882a593Smuzhiyun 		return;
1032*4882a593Smuzhiyun 
1033*4882a593Smuzhiyun 	if (sym->static_call_tramp) {
1034*4882a593Smuzhiyun 		list_add_tail(&insn->call_node, &file->static_call_list);
1035*4882a593Smuzhiyun 		return;
1036*4882a593Smuzhiyun 	}
1037*4882a593Smuzhiyun 
1038*4882a593Smuzhiyun 	if (sym->retpoline_thunk) {
1039*4882a593Smuzhiyun 		list_add_tail(&insn->call_node, &file->retpoline_call_list);
1040*4882a593Smuzhiyun 		return;
1041*4882a593Smuzhiyun 	}
1042*4882a593Smuzhiyun 
1043*4882a593Smuzhiyun 	/*
1044*4882a593Smuzhiyun 	 * Many compilers cannot disable KCOV with a function attribute
1045*4882a593Smuzhiyun 	 * so they need a little help, NOP out any KCOV calls from noinstr
1046*4882a593Smuzhiyun 	 * text.
1047*4882a593Smuzhiyun 	 */
1048*4882a593Smuzhiyun 	if (insn->sec->noinstr && sym->kcov) {
1049*4882a593Smuzhiyun 		if (reloc) {
1050*4882a593Smuzhiyun 			reloc->type = R_NONE;
1051*4882a593Smuzhiyun 			elf_write_reloc(file->elf, reloc);
1052*4882a593Smuzhiyun 		}
1053*4882a593Smuzhiyun 
1054*4882a593Smuzhiyun 		elf_write_insn(file->elf, insn->sec,
1055*4882a593Smuzhiyun 			       insn->offset, insn->len,
1056*4882a593Smuzhiyun 			       sibling ? arch_ret_insn(insn->len)
1057*4882a593Smuzhiyun 			               : arch_nop_insn(insn->len));
1058*4882a593Smuzhiyun 
1059*4882a593Smuzhiyun 		insn->type = sibling ? INSN_RETURN : INSN_NOP;
1060*4882a593Smuzhiyun 
1061*4882a593Smuzhiyun 		if (sibling) {
1062*4882a593Smuzhiyun 			/*
1063*4882a593Smuzhiyun 			 * We've replaced the tail-call JMP insn by two new
1064*4882a593Smuzhiyun 			 * insn: RET; INT3, except we only have a single struct
1065*4882a593Smuzhiyun 			 * insn here. Mark it retpoline_safe to avoid the SLS
1066*4882a593Smuzhiyun 			 * warning, instead of adding another insn.
1067*4882a593Smuzhiyun 			 */
1068*4882a593Smuzhiyun 			insn->retpoline_safe = true;
1069*4882a593Smuzhiyun 		}
1070*4882a593Smuzhiyun 
1071*4882a593Smuzhiyun 		return;
1072*4882a593Smuzhiyun 	}
1073*4882a593Smuzhiyun }
1074*4882a593Smuzhiyun 
add_call_dest(struct objtool_file * file,struct instruction * insn,struct symbol * dest,bool sibling)1075*4882a593Smuzhiyun static void add_call_dest(struct objtool_file *file, struct instruction *insn,
1076*4882a593Smuzhiyun 			  struct symbol *dest, bool sibling)
1077*4882a593Smuzhiyun {
1078*4882a593Smuzhiyun 	insn->call_dest = dest;
1079*4882a593Smuzhiyun 	if (!dest)
1080*4882a593Smuzhiyun 		return;
1081*4882a593Smuzhiyun 
1082*4882a593Smuzhiyun 	/*
1083*4882a593Smuzhiyun 	 * Whatever stack impact regular CALLs have, should be undone
1084*4882a593Smuzhiyun 	 * by the RETURN of the called function.
1085*4882a593Smuzhiyun 	 *
1086*4882a593Smuzhiyun 	 * Annotated intra-function calls retain the stack_ops but
1087*4882a593Smuzhiyun 	 * are converted to JUMP, see read_intra_function_calls().
1088*4882a593Smuzhiyun 	 */
1089*4882a593Smuzhiyun 	remove_insn_ops(insn);
1090*4882a593Smuzhiyun 
1091*4882a593Smuzhiyun 	annotate_call_site(file, insn, sibling);
1092*4882a593Smuzhiyun }
1093*4882a593Smuzhiyun 
add_retpoline_call(struct objtool_file * file,struct instruction * insn)1094*4882a593Smuzhiyun static void add_retpoline_call(struct objtool_file *file, struct instruction *insn)
1095*4882a593Smuzhiyun {
1096*4882a593Smuzhiyun 	/*
1097*4882a593Smuzhiyun 	 * Retpoline calls/jumps are really dynamic calls/jumps in disguise,
1098*4882a593Smuzhiyun 	 * so convert them accordingly.
1099*4882a593Smuzhiyun 	 */
1100*4882a593Smuzhiyun 	switch (insn->type) {
1101*4882a593Smuzhiyun 	case INSN_CALL:
1102*4882a593Smuzhiyun 		insn->type = INSN_CALL_DYNAMIC;
1103*4882a593Smuzhiyun 		break;
1104*4882a593Smuzhiyun 	case INSN_JUMP_UNCONDITIONAL:
1105*4882a593Smuzhiyun 		insn->type = INSN_JUMP_DYNAMIC;
1106*4882a593Smuzhiyun 		break;
1107*4882a593Smuzhiyun 	case INSN_JUMP_CONDITIONAL:
1108*4882a593Smuzhiyun 		insn->type = INSN_JUMP_DYNAMIC_CONDITIONAL;
1109*4882a593Smuzhiyun 		break;
1110*4882a593Smuzhiyun 	default:
1111*4882a593Smuzhiyun 		return;
1112*4882a593Smuzhiyun 	}
1113*4882a593Smuzhiyun 
1114*4882a593Smuzhiyun 	insn->retpoline_safe = true;
1115*4882a593Smuzhiyun 
1116*4882a593Smuzhiyun 	/*
1117*4882a593Smuzhiyun 	 * Whatever stack impact regular CALLs have, should be undone
1118*4882a593Smuzhiyun 	 * by the RETURN of the called function.
1119*4882a593Smuzhiyun 	 *
1120*4882a593Smuzhiyun 	 * Annotated intra-function calls retain the stack_ops but
1121*4882a593Smuzhiyun 	 * are converted to JUMP, see read_intra_function_calls().
1122*4882a593Smuzhiyun 	 */
1123*4882a593Smuzhiyun 	remove_insn_ops(insn);
1124*4882a593Smuzhiyun 
1125*4882a593Smuzhiyun 	annotate_call_site(file, insn, false);
1126*4882a593Smuzhiyun }
1127*4882a593Smuzhiyun 
add_return_call(struct objtool_file * file,struct instruction * insn,bool add)1128*4882a593Smuzhiyun static void add_return_call(struct objtool_file *file, struct instruction *insn, bool add)
1129*4882a593Smuzhiyun {
1130*4882a593Smuzhiyun 	/*
1131*4882a593Smuzhiyun 	 * Return thunk tail calls are really just returns in disguise,
1132*4882a593Smuzhiyun 	 * so convert them accordingly.
1133*4882a593Smuzhiyun 	 */
1134*4882a593Smuzhiyun 	insn->type = INSN_RETURN;
1135*4882a593Smuzhiyun 	insn->retpoline_safe = true;
1136*4882a593Smuzhiyun 
1137*4882a593Smuzhiyun 	/* Skip the non-text sections, specially .discard ones */
1138*4882a593Smuzhiyun 	if (add && insn->sec->text)
1139*4882a593Smuzhiyun 		list_add_tail(&insn->call_node, &file->return_thunk_list);
1140*4882a593Smuzhiyun }
1141*4882a593Smuzhiyun 
1142*4882a593Smuzhiyun /*
1143*4882a593Smuzhiyun  * CONFIG_CFI_CLANG: Check if the section is a CFI jump table or a
1144*4882a593Smuzhiyun  * compiler-generated CFI handler.
1145*4882a593Smuzhiyun  */
is_cfi_section(struct section * sec)1146*4882a593Smuzhiyun static bool is_cfi_section(struct section *sec)
1147*4882a593Smuzhiyun {
1148*4882a593Smuzhiyun 	return (sec->name &&
1149*4882a593Smuzhiyun 		(!strncmp(sec->name, ".text..L.cfi.jumptable", 22) ||
1150*4882a593Smuzhiyun 		 !strcmp(sec->name, ".text.__cfi_check")));
1151*4882a593Smuzhiyun }
1152*4882a593Smuzhiyun 
1153*4882a593Smuzhiyun /*
1154*4882a593Smuzhiyun  * CONFIG_CFI_CLANG: Ignore CFI jump tables.
1155*4882a593Smuzhiyun  */
add_cfi_jumptables(struct objtool_file * file)1156*4882a593Smuzhiyun static void add_cfi_jumptables(struct objtool_file *file)
1157*4882a593Smuzhiyun {
1158*4882a593Smuzhiyun 	struct section *sec;
1159*4882a593Smuzhiyun 	struct symbol *func;
1160*4882a593Smuzhiyun 	struct instruction *insn;
1161*4882a593Smuzhiyun 
1162*4882a593Smuzhiyun 	for_each_sec(file, sec) {
1163*4882a593Smuzhiyun 		if (!is_cfi_section(sec))
1164*4882a593Smuzhiyun 			continue;
1165*4882a593Smuzhiyun 
1166*4882a593Smuzhiyun 		list_for_each_entry(func, &sec->symbol_list, list) {
1167*4882a593Smuzhiyun 			sym_for_each_insn(file, func, insn)
1168*4882a593Smuzhiyun 				insn->ignore = true;
1169*4882a593Smuzhiyun 		}
1170*4882a593Smuzhiyun 	}
1171*4882a593Smuzhiyun }
1172*4882a593Smuzhiyun 
1173*4882a593Smuzhiyun /*
1174*4882a593Smuzhiyun  * Find the destination instructions for all jumps.
1175*4882a593Smuzhiyun  */
add_jump_destinations(struct objtool_file * file)1176*4882a593Smuzhiyun static int add_jump_destinations(struct objtool_file *file)
1177*4882a593Smuzhiyun {
1178*4882a593Smuzhiyun 	struct instruction *insn;
1179*4882a593Smuzhiyun 	struct reloc *reloc;
1180*4882a593Smuzhiyun 	struct section *dest_sec;
1181*4882a593Smuzhiyun 	unsigned long dest_off;
1182*4882a593Smuzhiyun 
1183*4882a593Smuzhiyun 	for_each_insn(file, insn) {
1184*4882a593Smuzhiyun 		if (!is_static_jump(insn))
1185*4882a593Smuzhiyun 			continue;
1186*4882a593Smuzhiyun 
1187*4882a593Smuzhiyun 		reloc = insn_reloc(file, insn);
1188*4882a593Smuzhiyun 		if (!reloc) {
1189*4882a593Smuzhiyun 			dest_sec = insn->sec;
1190*4882a593Smuzhiyun 			dest_off = arch_jump_destination(insn);
1191*4882a593Smuzhiyun 		} else if (reloc->sym->type == STT_SECTION) {
1192*4882a593Smuzhiyun 			dest_sec = reloc->sym->sec;
1193*4882a593Smuzhiyun 			dest_off = arch_dest_reloc_offset(reloc->addend);
1194*4882a593Smuzhiyun 		} else if (reloc->sym->retpoline_thunk) {
1195*4882a593Smuzhiyun 			add_retpoline_call(file, insn);
1196*4882a593Smuzhiyun 			continue;
1197*4882a593Smuzhiyun 		} else if (reloc->sym->return_thunk) {
1198*4882a593Smuzhiyun 			add_return_call(file, insn, true);
1199*4882a593Smuzhiyun 			continue;
1200*4882a593Smuzhiyun 		} else if (insn->func) {
1201*4882a593Smuzhiyun 			/* internal or external sibling call (with reloc) */
1202*4882a593Smuzhiyun 			add_call_dest(file, insn, reloc->sym, true);
1203*4882a593Smuzhiyun 			continue;
1204*4882a593Smuzhiyun 		} else if (reloc->sym->sec->idx) {
1205*4882a593Smuzhiyun 			dest_sec = reloc->sym->sec;
1206*4882a593Smuzhiyun 			dest_off = reloc->sym->sym.st_value +
1207*4882a593Smuzhiyun 				   arch_dest_reloc_offset(reloc->addend);
1208*4882a593Smuzhiyun 		} else {
1209*4882a593Smuzhiyun 			/* non-func asm code jumping to another file */
1210*4882a593Smuzhiyun 			continue;
1211*4882a593Smuzhiyun 		}
1212*4882a593Smuzhiyun 
1213*4882a593Smuzhiyun 		insn->jump_dest = find_insn(file, dest_sec, dest_off);
1214*4882a593Smuzhiyun 
1215*4882a593Smuzhiyun 		if (!insn->jump_dest && dest_sec->len == dest_off)
1216*4882a593Smuzhiyun 			insn->jump_dest = find_last_insn(file, dest_sec);
1217*4882a593Smuzhiyun 
1218*4882a593Smuzhiyun 		if (!insn->jump_dest) {
1219*4882a593Smuzhiyun 			struct symbol *sym = find_symbol_by_offset(dest_sec, dest_off);
1220*4882a593Smuzhiyun 
1221*4882a593Smuzhiyun 			/*
1222*4882a593Smuzhiyun 			 * This is a special case where an alt instruction
1223*4882a593Smuzhiyun 			 * jumps past the end of the section.  These are
1224*4882a593Smuzhiyun 			 * handled later in handle_group_alt().
1225*4882a593Smuzhiyun 			 */
1226*4882a593Smuzhiyun 			if (!strcmp(insn->sec->name, ".altinstr_replacement"))
1227*4882a593Smuzhiyun 				continue;
1228*4882a593Smuzhiyun 
1229*4882a593Smuzhiyun 			if (is_cfi_section(insn->sec))
1230*4882a593Smuzhiyun 				continue;
1231*4882a593Smuzhiyun 
1232*4882a593Smuzhiyun 			/*
1233*4882a593Smuzhiyun 			 * This is a special case for zen_untrain_ret().
1234*4882a593Smuzhiyun 			 * It jumps to __x86_return_thunk(), but objtool
1235*4882a593Smuzhiyun 			 * can't find the thunk's starting RET
1236*4882a593Smuzhiyun 			 * instruction, because the RET is also in the
1237*4882a593Smuzhiyun 			 * middle of another instruction.  Objtool only
1238*4882a593Smuzhiyun 			 * knows about the outer instruction.
1239*4882a593Smuzhiyun 			 */
1240*4882a593Smuzhiyun 			if (sym && sym->return_thunk) {
1241*4882a593Smuzhiyun 				add_return_call(file, insn, false);
1242*4882a593Smuzhiyun 				continue;
1243*4882a593Smuzhiyun 			}
1244*4882a593Smuzhiyun 
1245*4882a593Smuzhiyun 			WARN_FUNC("can't find jump dest instruction at %s+0x%lx",
1246*4882a593Smuzhiyun 				  insn->sec, insn->offset, dest_sec->name,
1247*4882a593Smuzhiyun 				  dest_off);
1248*4882a593Smuzhiyun 			return -1;
1249*4882a593Smuzhiyun 		}
1250*4882a593Smuzhiyun 
1251*4882a593Smuzhiyun 		/*
1252*4882a593Smuzhiyun 		 * Cross-function jump.
1253*4882a593Smuzhiyun 		 */
1254*4882a593Smuzhiyun 		if (insn->func && insn->jump_dest->func &&
1255*4882a593Smuzhiyun 		    insn->func != insn->jump_dest->func) {
1256*4882a593Smuzhiyun 
1257*4882a593Smuzhiyun 			/*
1258*4882a593Smuzhiyun 			 * For GCC 8+, create parent/child links for any cold
1259*4882a593Smuzhiyun 			 * subfunctions.  This is _mostly_ redundant with a
1260*4882a593Smuzhiyun 			 * similar initialization in read_symbols().
1261*4882a593Smuzhiyun 			 *
1262*4882a593Smuzhiyun 			 * If a function has aliases, we want the *first* such
1263*4882a593Smuzhiyun 			 * function in the symbol table to be the subfunction's
1264*4882a593Smuzhiyun 			 * parent.  In that case we overwrite the
1265*4882a593Smuzhiyun 			 * initialization done in read_symbols().
1266*4882a593Smuzhiyun 			 *
1267*4882a593Smuzhiyun 			 * However this code can't completely replace the
1268*4882a593Smuzhiyun 			 * read_symbols() code because this doesn't detect the
1269*4882a593Smuzhiyun 			 * case where the parent function's only reference to a
1270*4882a593Smuzhiyun 			 * subfunction is through a jump table.
1271*4882a593Smuzhiyun 			 */
1272*4882a593Smuzhiyun 			if (!strstr(insn->func->name, ".cold") &&
1273*4882a593Smuzhiyun 			    strstr(insn->jump_dest->func->name, ".cold")) {
1274*4882a593Smuzhiyun 				insn->func->cfunc = insn->jump_dest->func;
1275*4882a593Smuzhiyun 				insn->jump_dest->func->pfunc = insn->func;
1276*4882a593Smuzhiyun 
1277*4882a593Smuzhiyun 			} else if (insn->jump_dest->func->pfunc != insn->func->pfunc &&
1278*4882a593Smuzhiyun 				   insn->jump_dest->offset == insn->jump_dest->func->offset) {
1279*4882a593Smuzhiyun 				/* internal sibling call (without reloc) */
1280*4882a593Smuzhiyun 				add_call_dest(file, insn, insn->jump_dest->func, true);
1281*4882a593Smuzhiyun 			}
1282*4882a593Smuzhiyun 		}
1283*4882a593Smuzhiyun 	}
1284*4882a593Smuzhiyun 
1285*4882a593Smuzhiyun 	return 0;
1286*4882a593Smuzhiyun }
1287*4882a593Smuzhiyun 
find_call_destination(struct section * sec,unsigned long offset)1288*4882a593Smuzhiyun static struct symbol *find_call_destination(struct section *sec, unsigned long offset)
1289*4882a593Smuzhiyun {
1290*4882a593Smuzhiyun 	struct symbol *call_dest;
1291*4882a593Smuzhiyun 
1292*4882a593Smuzhiyun 	call_dest = find_func_by_offset(sec, offset);
1293*4882a593Smuzhiyun 	if (!call_dest)
1294*4882a593Smuzhiyun 		call_dest = find_symbol_by_offset(sec, offset);
1295*4882a593Smuzhiyun 
1296*4882a593Smuzhiyun 	return call_dest;
1297*4882a593Smuzhiyun }
1298*4882a593Smuzhiyun 
1299*4882a593Smuzhiyun /*
1300*4882a593Smuzhiyun  * Find the destination instructions for all calls.
1301*4882a593Smuzhiyun  */
add_call_destinations(struct objtool_file * file)1302*4882a593Smuzhiyun static int add_call_destinations(struct objtool_file *file)
1303*4882a593Smuzhiyun {
1304*4882a593Smuzhiyun 	struct instruction *insn;
1305*4882a593Smuzhiyun 	unsigned long dest_off;
1306*4882a593Smuzhiyun 	struct symbol *dest;
1307*4882a593Smuzhiyun 	struct reloc *reloc;
1308*4882a593Smuzhiyun 
1309*4882a593Smuzhiyun 	for_each_insn(file, insn) {
1310*4882a593Smuzhiyun 		if (insn->type != INSN_CALL)
1311*4882a593Smuzhiyun 			continue;
1312*4882a593Smuzhiyun 
1313*4882a593Smuzhiyun 		reloc = insn_reloc(file, insn);
1314*4882a593Smuzhiyun 		if (!reloc) {
1315*4882a593Smuzhiyun 			dest_off = arch_jump_destination(insn);
1316*4882a593Smuzhiyun 			dest = find_call_destination(insn->sec, dest_off);
1317*4882a593Smuzhiyun 
1318*4882a593Smuzhiyun 			add_call_dest(file, insn, dest, false);
1319*4882a593Smuzhiyun 
1320*4882a593Smuzhiyun 			if (insn->ignore)
1321*4882a593Smuzhiyun 				continue;
1322*4882a593Smuzhiyun 
1323*4882a593Smuzhiyun 			if (!insn->call_dest) {
1324*4882a593Smuzhiyun 				WARN_FUNC("unannotated intra-function call", insn->sec, insn->offset);
1325*4882a593Smuzhiyun 				return -1;
1326*4882a593Smuzhiyun 			}
1327*4882a593Smuzhiyun 
1328*4882a593Smuzhiyun 			if (insn->func && insn->call_dest->type != STT_FUNC) {
1329*4882a593Smuzhiyun 				WARN_FUNC("unsupported call to non-function",
1330*4882a593Smuzhiyun 					  insn->sec, insn->offset);
1331*4882a593Smuzhiyun 				return -1;
1332*4882a593Smuzhiyun 			}
1333*4882a593Smuzhiyun 
1334*4882a593Smuzhiyun 		} else if (reloc->sym->type == STT_SECTION) {
1335*4882a593Smuzhiyun 			dest_off = arch_dest_reloc_offset(reloc->addend);
1336*4882a593Smuzhiyun 			dest = find_call_destination(reloc->sym->sec, dest_off);
1337*4882a593Smuzhiyun 			if (!dest) {
1338*4882a593Smuzhiyun 				if (is_cfi_section(reloc->sym->sec))
1339*4882a593Smuzhiyun 					continue;
1340*4882a593Smuzhiyun 
1341*4882a593Smuzhiyun 				WARN_FUNC("can't find call dest symbol at %s+0x%lx",
1342*4882a593Smuzhiyun 					  insn->sec, insn->offset,
1343*4882a593Smuzhiyun 					  reloc->sym->sec->name,
1344*4882a593Smuzhiyun 					  dest_off);
1345*4882a593Smuzhiyun 				return -1;
1346*4882a593Smuzhiyun 			}
1347*4882a593Smuzhiyun 
1348*4882a593Smuzhiyun 			add_call_dest(file, insn, dest, false);
1349*4882a593Smuzhiyun 
1350*4882a593Smuzhiyun 		} else if (reloc->sym->retpoline_thunk) {
1351*4882a593Smuzhiyun 			add_retpoline_call(file, insn);
1352*4882a593Smuzhiyun 
1353*4882a593Smuzhiyun 		} else
1354*4882a593Smuzhiyun 			add_call_dest(file, insn, reloc->sym, false);
1355*4882a593Smuzhiyun 	}
1356*4882a593Smuzhiyun 
1357*4882a593Smuzhiyun 	return 0;
1358*4882a593Smuzhiyun }
1359*4882a593Smuzhiyun 
1360*4882a593Smuzhiyun /*
1361*4882a593Smuzhiyun  * The .alternatives section requires some extra special care over and above
1362*4882a593Smuzhiyun  * other special sections because alternatives are patched in place.
1363*4882a593Smuzhiyun  */
handle_group_alt(struct objtool_file * file,struct special_alt * special_alt,struct instruction * orig_insn,struct instruction ** new_insn)1364*4882a593Smuzhiyun static int handle_group_alt(struct objtool_file *file,
1365*4882a593Smuzhiyun 			    struct special_alt *special_alt,
1366*4882a593Smuzhiyun 			    struct instruction *orig_insn,
1367*4882a593Smuzhiyun 			    struct instruction **new_insn)
1368*4882a593Smuzhiyun {
1369*4882a593Smuzhiyun 	struct instruction *last_orig_insn, *last_new_insn = NULL, *insn, *nop = NULL;
1370*4882a593Smuzhiyun 	struct alt_group *orig_alt_group, *new_alt_group;
1371*4882a593Smuzhiyun 	unsigned long dest_off;
1372*4882a593Smuzhiyun 
1373*4882a593Smuzhiyun 
1374*4882a593Smuzhiyun 	orig_alt_group = malloc(sizeof(*orig_alt_group));
1375*4882a593Smuzhiyun 	if (!orig_alt_group) {
1376*4882a593Smuzhiyun 		WARN("malloc failed");
1377*4882a593Smuzhiyun 		return -1;
1378*4882a593Smuzhiyun 	}
1379*4882a593Smuzhiyun 	orig_alt_group->cfi = calloc(special_alt->orig_len,
1380*4882a593Smuzhiyun 				     sizeof(struct cfi_state *));
1381*4882a593Smuzhiyun 	if (!orig_alt_group->cfi) {
1382*4882a593Smuzhiyun 		WARN("calloc failed");
1383*4882a593Smuzhiyun 		return -1;
1384*4882a593Smuzhiyun 	}
1385*4882a593Smuzhiyun 
1386*4882a593Smuzhiyun 	last_orig_insn = NULL;
1387*4882a593Smuzhiyun 	insn = orig_insn;
1388*4882a593Smuzhiyun 	sec_for_each_insn_from(file, insn) {
1389*4882a593Smuzhiyun 		if (insn->offset >= special_alt->orig_off + special_alt->orig_len)
1390*4882a593Smuzhiyun 			break;
1391*4882a593Smuzhiyun 
1392*4882a593Smuzhiyun 		insn->alt_group = orig_alt_group;
1393*4882a593Smuzhiyun 		last_orig_insn = insn;
1394*4882a593Smuzhiyun 	}
1395*4882a593Smuzhiyun 	orig_alt_group->orig_group = NULL;
1396*4882a593Smuzhiyun 	orig_alt_group->first_insn = orig_insn;
1397*4882a593Smuzhiyun 	orig_alt_group->last_insn = last_orig_insn;
1398*4882a593Smuzhiyun 
1399*4882a593Smuzhiyun 
1400*4882a593Smuzhiyun 	new_alt_group = malloc(sizeof(*new_alt_group));
1401*4882a593Smuzhiyun 	if (!new_alt_group) {
1402*4882a593Smuzhiyun 		WARN("malloc failed");
1403*4882a593Smuzhiyun 		return -1;
1404*4882a593Smuzhiyun 	}
1405*4882a593Smuzhiyun 
1406*4882a593Smuzhiyun 	if (special_alt->new_len < special_alt->orig_len) {
1407*4882a593Smuzhiyun 		/*
1408*4882a593Smuzhiyun 		 * Insert a fake nop at the end to make the replacement
1409*4882a593Smuzhiyun 		 * alt_group the same size as the original.  This is needed to
1410*4882a593Smuzhiyun 		 * allow propagate_alt_cfi() to do its magic.  When the last
1411*4882a593Smuzhiyun 		 * instruction affects the stack, the instruction after it (the
1412*4882a593Smuzhiyun 		 * nop) will propagate the new state to the shared CFI array.
1413*4882a593Smuzhiyun 		 */
1414*4882a593Smuzhiyun 		nop = malloc(sizeof(*nop));
1415*4882a593Smuzhiyun 		if (!nop) {
1416*4882a593Smuzhiyun 			WARN("malloc failed");
1417*4882a593Smuzhiyun 			return -1;
1418*4882a593Smuzhiyun 		}
1419*4882a593Smuzhiyun 		memset(nop, 0, sizeof(*nop));
1420*4882a593Smuzhiyun 		INIT_LIST_HEAD(&nop->alts);
1421*4882a593Smuzhiyun 		INIT_LIST_HEAD(&nop->stack_ops);
1422*4882a593Smuzhiyun 
1423*4882a593Smuzhiyun 		nop->sec = special_alt->new_sec;
1424*4882a593Smuzhiyun 		nop->offset = special_alt->new_off + special_alt->new_len;
1425*4882a593Smuzhiyun 		nop->len = special_alt->orig_len - special_alt->new_len;
1426*4882a593Smuzhiyun 		nop->type = INSN_NOP;
1427*4882a593Smuzhiyun 		nop->func = orig_insn->func;
1428*4882a593Smuzhiyun 		nop->alt_group = new_alt_group;
1429*4882a593Smuzhiyun 		nop->ignore = orig_insn->ignore_alts;
1430*4882a593Smuzhiyun 	}
1431*4882a593Smuzhiyun 
1432*4882a593Smuzhiyun 	if (!special_alt->new_len) {
1433*4882a593Smuzhiyun 		*new_insn = nop;
1434*4882a593Smuzhiyun 		goto end;
1435*4882a593Smuzhiyun 	}
1436*4882a593Smuzhiyun 
1437*4882a593Smuzhiyun 	insn = *new_insn;
1438*4882a593Smuzhiyun 	sec_for_each_insn_from(file, insn) {
1439*4882a593Smuzhiyun 		struct reloc *alt_reloc;
1440*4882a593Smuzhiyun 
1441*4882a593Smuzhiyun 		if (insn->offset >= special_alt->new_off + special_alt->new_len)
1442*4882a593Smuzhiyun 			break;
1443*4882a593Smuzhiyun 
1444*4882a593Smuzhiyun 		last_new_insn = insn;
1445*4882a593Smuzhiyun 
1446*4882a593Smuzhiyun 		insn->ignore = orig_insn->ignore_alts;
1447*4882a593Smuzhiyun 		insn->func = orig_insn->func;
1448*4882a593Smuzhiyun 		insn->alt_group = new_alt_group;
1449*4882a593Smuzhiyun 
1450*4882a593Smuzhiyun 		/*
1451*4882a593Smuzhiyun 		 * Since alternative replacement code is copy/pasted by the
1452*4882a593Smuzhiyun 		 * kernel after applying relocations, generally such code can't
1453*4882a593Smuzhiyun 		 * have relative-address relocation references to outside the
1454*4882a593Smuzhiyun 		 * .altinstr_replacement section, unless the arch's
1455*4882a593Smuzhiyun 		 * alternatives code can adjust the relative offsets
1456*4882a593Smuzhiyun 		 * accordingly.
1457*4882a593Smuzhiyun 		 */
1458*4882a593Smuzhiyun 		alt_reloc = insn_reloc(file, insn);
1459*4882a593Smuzhiyun 		if (alt_reloc &&
1460*4882a593Smuzhiyun 		    !arch_support_alt_relocation(special_alt, insn, alt_reloc)) {
1461*4882a593Smuzhiyun 
1462*4882a593Smuzhiyun 			WARN_FUNC("unsupported relocation in alternatives section",
1463*4882a593Smuzhiyun 				  insn->sec, insn->offset);
1464*4882a593Smuzhiyun 			return -1;
1465*4882a593Smuzhiyun 		}
1466*4882a593Smuzhiyun 
1467*4882a593Smuzhiyun 		if (!is_static_jump(insn))
1468*4882a593Smuzhiyun 			continue;
1469*4882a593Smuzhiyun 
1470*4882a593Smuzhiyun 		if (!insn->immediate)
1471*4882a593Smuzhiyun 			continue;
1472*4882a593Smuzhiyun 
1473*4882a593Smuzhiyun 		dest_off = arch_jump_destination(insn);
1474*4882a593Smuzhiyun 		if (dest_off == special_alt->new_off + special_alt->new_len)
1475*4882a593Smuzhiyun 			insn->jump_dest = next_insn_same_sec(file, last_orig_insn);
1476*4882a593Smuzhiyun 
1477*4882a593Smuzhiyun 		if (!insn->jump_dest) {
1478*4882a593Smuzhiyun 			WARN_FUNC("can't find alternative jump destination",
1479*4882a593Smuzhiyun 				  insn->sec, insn->offset);
1480*4882a593Smuzhiyun 			return -1;
1481*4882a593Smuzhiyun 		}
1482*4882a593Smuzhiyun 	}
1483*4882a593Smuzhiyun 
1484*4882a593Smuzhiyun 	if (!last_new_insn) {
1485*4882a593Smuzhiyun 		WARN_FUNC("can't find last new alternative instruction",
1486*4882a593Smuzhiyun 			  special_alt->new_sec, special_alt->new_off);
1487*4882a593Smuzhiyun 		return -1;
1488*4882a593Smuzhiyun 	}
1489*4882a593Smuzhiyun 
1490*4882a593Smuzhiyun 	if (nop)
1491*4882a593Smuzhiyun 		list_add(&nop->list, &last_new_insn->list);
1492*4882a593Smuzhiyun end:
1493*4882a593Smuzhiyun 	new_alt_group->orig_group = orig_alt_group;
1494*4882a593Smuzhiyun 	new_alt_group->first_insn = *new_insn;
1495*4882a593Smuzhiyun 	new_alt_group->last_insn = nop ? : last_new_insn;
1496*4882a593Smuzhiyun 	new_alt_group->cfi = orig_alt_group->cfi;
1497*4882a593Smuzhiyun 	return 0;
1498*4882a593Smuzhiyun }
1499*4882a593Smuzhiyun 
1500*4882a593Smuzhiyun /*
1501*4882a593Smuzhiyun  * A jump table entry can either convert a nop to a jump or a jump to a nop.
1502*4882a593Smuzhiyun  * If the original instruction is a jump, make the alt entry an effective nop
1503*4882a593Smuzhiyun  * by just skipping the original instruction.
1504*4882a593Smuzhiyun  */
handle_jump_alt(struct objtool_file * file,struct special_alt * special_alt,struct instruction * orig_insn,struct instruction ** new_insn)1505*4882a593Smuzhiyun static int handle_jump_alt(struct objtool_file *file,
1506*4882a593Smuzhiyun 			   struct special_alt *special_alt,
1507*4882a593Smuzhiyun 			   struct instruction *orig_insn,
1508*4882a593Smuzhiyun 			   struct instruction **new_insn)
1509*4882a593Smuzhiyun {
1510*4882a593Smuzhiyun 	if (orig_insn->type == INSN_NOP)
1511*4882a593Smuzhiyun 		return 0;
1512*4882a593Smuzhiyun 
1513*4882a593Smuzhiyun 	if (orig_insn->type != INSN_JUMP_UNCONDITIONAL) {
1514*4882a593Smuzhiyun 		WARN_FUNC("unsupported instruction at jump label",
1515*4882a593Smuzhiyun 			  orig_insn->sec, orig_insn->offset);
1516*4882a593Smuzhiyun 		return -1;
1517*4882a593Smuzhiyun 	}
1518*4882a593Smuzhiyun 
1519*4882a593Smuzhiyun 	*new_insn = list_next_entry(orig_insn, list);
1520*4882a593Smuzhiyun 	return 0;
1521*4882a593Smuzhiyun }
1522*4882a593Smuzhiyun 
1523*4882a593Smuzhiyun /*
1524*4882a593Smuzhiyun  * Read all the special sections which have alternate instructions which can be
1525*4882a593Smuzhiyun  * patched in or redirected to at runtime.  Each instruction having alternate
1526*4882a593Smuzhiyun  * instruction(s) has them added to its insn->alts list, which will be
1527*4882a593Smuzhiyun  * traversed in validate_branch().
1528*4882a593Smuzhiyun  */
add_special_section_alts(struct objtool_file * file)1529*4882a593Smuzhiyun static int add_special_section_alts(struct objtool_file *file)
1530*4882a593Smuzhiyun {
1531*4882a593Smuzhiyun 	struct list_head special_alts;
1532*4882a593Smuzhiyun 	struct instruction *orig_insn, *new_insn;
1533*4882a593Smuzhiyun 	struct special_alt *special_alt, *tmp;
1534*4882a593Smuzhiyun 	struct alternative *alt;
1535*4882a593Smuzhiyun 	int ret;
1536*4882a593Smuzhiyun 
1537*4882a593Smuzhiyun 	ret = special_get_alts(file->elf, &special_alts);
1538*4882a593Smuzhiyun 	if (ret)
1539*4882a593Smuzhiyun 		return ret;
1540*4882a593Smuzhiyun 
1541*4882a593Smuzhiyun 	list_for_each_entry_safe(special_alt, tmp, &special_alts, list) {
1542*4882a593Smuzhiyun 
1543*4882a593Smuzhiyun 		orig_insn = find_insn(file, special_alt->orig_sec,
1544*4882a593Smuzhiyun 				      special_alt->orig_off);
1545*4882a593Smuzhiyun 		if (!orig_insn) {
1546*4882a593Smuzhiyun 			WARN_FUNC("special: can't find orig instruction",
1547*4882a593Smuzhiyun 				  special_alt->orig_sec, special_alt->orig_off);
1548*4882a593Smuzhiyun 			ret = -1;
1549*4882a593Smuzhiyun 			goto out;
1550*4882a593Smuzhiyun 		}
1551*4882a593Smuzhiyun 
1552*4882a593Smuzhiyun 		new_insn = NULL;
1553*4882a593Smuzhiyun 		if (!special_alt->group || special_alt->new_len) {
1554*4882a593Smuzhiyun 			new_insn = find_insn(file, special_alt->new_sec,
1555*4882a593Smuzhiyun 					     special_alt->new_off);
1556*4882a593Smuzhiyun 			if (!new_insn) {
1557*4882a593Smuzhiyun 				WARN_FUNC("special: can't find new instruction",
1558*4882a593Smuzhiyun 					  special_alt->new_sec,
1559*4882a593Smuzhiyun 					  special_alt->new_off);
1560*4882a593Smuzhiyun 				ret = -1;
1561*4882a593Smuzhiyun 				goto out;
1562*4882a593Smuzhiyun 			}
1563*4882a593Smuzhiyun 		}
1564*4882a593Smuzhiyun 
1565*4882a593Smuzhiyun 		if (special_alt->group) {
1566*4882a593Smuzhiyun 			if (!special_alt->orig_len) {
1567*4882a593Smuzhiyun 				WARN_FUNC("empty alternative entry",
1568*4882a593Smuzhiyun 					  orig_insn->sec, orig_insn->offset);
1569*4882a593Smuzhiyun 				continue;
1570*4882a593Smuzhiyun 			}
1571*4882a593Smuzhiyun 
1572*4882a593Smuzhiyun 			ret = handle_group_alt(file, special_alt, orig_insn,
1573*4882a593Smuzhiyun 					       &new_insn);
1574*4882a593Smuzhiyun 			if (ret)
1575*4882a593Smuzhiyun 				goto out;
1576*4882a593Smuzhiyun 		} else if (special_alt->jump_or_nop) {
1577*4882a593Smuzhiyun 			ret = handle_jump_alt(file, special_alt, orig_insn,
1578*4882a593Smuzhiyun 					      &new_insn);
1579*4882a593Smuzhiyun 			if (ret)
1580*4882a593Smuzhiyun 				goto out;
1581*4882a593Smuzhiyun 		}
1582*4882a593Smuzhiyun 
1583*4882a593Smuzhiyun 		alt = malloc(sizeof(*alt));
1584*4882a593Smuzhiyun 		if (!alt) {
1585*4882a593Smuzhiyun 			WARN("malloc failed");
1586*4882a593Smuzhiyun 			ret = -1;
1587*4882a593Smuzhiyun 			goto out;
1588*4882a593Smuzhiyun 		}
1589*4882a593Smuzhiyun 
1590*4882a593Smuzhiyun 		alt->insn = new_insn;
1591*4882a593Smuzhiyun 		alt->skip_orig = special_alt->skip_orig;
1592*4882a593Smuzhiyun 		orig_insn->ignore_alts |= special_alt->skip_alt;
1593*4882a593Smuzhiyun 		list_add_tail(&alt->list, &orig_insn->alts);
1594*4882a593Smuzhiyun 
1595*4882a593Smuzhiyun 		list_del(&special_alt->list);
1596*4882a593Smuzhiyun 		free(special_alt);
1597*4882a593Smuzhiyun 	}
1598*4882a593Smuzhiyun 
1599*4882a593Smuzhiyun out:
1600*4882a593Smuzhiyun 	return ret;
1601*4882a593Smuzhiyun }
1602*4882a593Smuzhiyun 
add_jump_table(struct objtool_file * file,struct instruction * insn,struct reloc * table)1603*4882a593Smuzhiyun static int add_jump_table(struct objtool_file *file, struct instruction *insn,
1604*4882a593Smuzhiyun 			    struct reloc *table)
1605*4882a593Smuzhiyun {
1606*4882a593Smuzhiyun 	struct reloc *reloc = table;
1607*4882a593Smuzhiyun 	struct instruction *dest_insn;
1608*4882a593Smuzhiyun 	struct alternative *alt;
1609*4882a593Smuzhiyun 	struct symbol *pfunc = insn->func->pfunc;
1610*4882a593Smuzhiyun 	unsigned int prev_offset = 0;
1611*4882a593Smuzhiyun 
1612*4882a593Smuzhiyun 	/*
1613*4882a593Smuzhiyun 	 * Each @reloc is a switch table relocation which points to the target
1614*4882a593Smuzhiyun 	 * instruction.
1615*4882a593Smuzhiyun 	 */
1616*4882a593Smuzhiyun 	list_for_each_entry_from(reloc, &table->sec->reloc_list, list) {
1617*4882a593Smuzhiyun 
1618*4882a593Smuzhiyun 		/* Check for the end of the table: */
1619*4882a593Smuzhiyun 		if (reloc != table && reloc->jump_table_start)
1620*4882a593Smuzhiyun 			break;
1621*4882a593Smuzhiyun 
1622*4882a593Smuzhiyun 		/* Make sure the table entries are consecutive: */
1623*4882a593Smuzhiyun 		if (prev_offset && reloc->offset != prev_offset + 8)
1624*4882a593Smuzhiyun 			break;
1625*4882a593Smuzhiyun 
1626*4882a593Smuzhiyun 		/* Detect function pointers from contiguous objects: */
1627*4882a593Smuzhiyun 		if (reloc->sym->sec == pfunc->sec &&
1628*4882a593Smuzhiyun 		    reloc->addend == pfunc->offset)
1629*4882a593Smuzhiyun 			break;
1630*4882a593Smuzhiyun 
1631*4882a593Smuzhiyun 		dest_insn = find_insn(file, reloc->sym->sec, reloc->addend);
1632*4882a593Smuzhiyun 		if (!dest_insn)
1633*4882a593Smuzhiyun 			break;
1634*4882a593Smuzhiyun 
1635*4882a593Smuzhiyun 		/* Make sure the destination is in the same function: */
1636*4882a593Smuzhiyun 		if (!dest_insn->func || dest_insn->func->pfunc != pfunc)
1637*4882a593Smuzhiyun 			break;
1638*4882a593Smuzhiyun 
1639*4882a593Smuzhiyun 		alt = malloc(sizeof(*alt));
1640*4882a593Smuzhiyun 		if (!alt) {
1641*4882a593Smuzhiyun 			WARN("malloc failed");
1642*4882a593Smuzhiyun 			return -1;
1643*4882a593Smuzhiyun 		}
1644*4882a593Smuzhiyun 
1645*4882a593Smuzhiyun 		alt->insn = dest_insn;
1646*4882a593Smuzhiyun 		list_add_tail(&alt->list, &insn->alts);
1647*4882a593Smuzhiyun 		prev_offset = reloc->offset;
1648*4882a593Smuzhiyun 	}
1649*4882a593Smuzhiyun 
1650*4882a593Smuzhiyun 	if (!prev_offset) {
1651*4882a593Smuzhiyun 		WARN_FUNC("can't find switch jump table",
1652*4882a593Smuzhiyun 			  insn->sec, insn->offset);
1653*4882a593Smuzhiyun 		return -1;
1654*4882a593Smuzhiyun 	}
1655*4882a593Smuzhiyun 
1656*4882a593Smuzhiyun 	return 0;
1657*4882a593Smuzhiyun }
1658*4882a593Smuzhiyun 
1659*4882a593Smuzhiyun /*
1660*4882a593Smuzhiyun  * find_jump_table() - Given a dynamic jump, find the switch jump table
1661*4882a593Smuzhiyun  * associated with it.
1662*4882a593Smuzhiyun  */
find_jump_table(struct objtool_file * file,struct symbol * func,struct instruction * insn)1663*4882a593Smuzhiyun static struct reloc *find_jump_table(struct objtool_file *file,
1664*4882a593Smuzhiyun 				      struct symbol *func,
1665*4882a593Smuzhiyun 				      struct instruction *insn)
1666*4882a593Smuzhiyun {
1667*4882a593Smuzhiyun 	struct reloc *table_reloc;
1668*4882a593Smuzhiyun 	struct instruction *dest_insn, *orig_insn = insn;
1669*4882a593Smuzhiyun 
1670*4882a593Smuzhiyun 	/*
1671*4882a593Smuzhiyun 	 * Backward search using the @first_jump_src links, these help avoid
1672*4882a593Smuzhiyun 	 * much of the 'in between' code. Which avoids us getting confused by
1673*4882a593Smuzhiyun 	 * it.
1674*4882a593Smuzhiyun 	 */
1675*4882a593Smuzhiyun 	for (;
1676*4882a593Smuzhiyun 	     insn && insn->func && insn->func->pfunc == func;
1677*4882a593Smuzhiyun 	     insn = insn->first_jump_src ?: prev_insn_same_sym(file, insn)) {
1678*4882a593Smuzhiyun 
1679*4882a593Smuzhiyun 		if (insn != orig_insn && insn->type == INSN_JUMP_DYNAMIC)
1680*4882a593Smuzhiyun 			break;
1681*4882a593Smuzhiyun 
1682*4882a593Smuzhiyun 		/* allow small jumps within the range */
1683*4882a593Smuzhiyun 		if (insn->type == INSN_JUMP_UNCONDITIONAL &&
1684*4882a593Smuzhiyun 		    insn->jump_dest &&
1685*4882a593Smuzhiyun 		    (insn->jump_dest->offset <= insn->offset ||
1686*4882a593Smuzhiyun 		     insn->jump_dest->offset > orig_insn->offset))
1687*4882a593Smuzhiyun 		    break;
1688*4882a593Smuzhiyun 
1689*4882a593Smuzhiyun 		table_reloc = arch_find_switch_table(file, insn);
1690*4882a593Smuzhiyun 		if (!table_reloc)
1691*4882a593Smuzhiyun 			continue;
1692*4882a593Smuzhiyun 		dest_insn = find_insn(file, table_reloc->sym->sec, table_reloc->addend);
1693*4882a593Smuzhiyun 		if (!dest_insn || !dest_insn->func || dest_insn->func->pfunc != func)
1694*4882a593Smuzhiyun 			continue;
1695*4882a593Smuzhiyun 
1696*4882a593Smuzhiyun 		return table_reloc;
1697*4882a593Smuzhiyun 	}
1698*4882a593Smuzhiyun 
1699*4882a593Smuzhiyun 	return NULL;
1700*4882a593Smuzhiyun }
1701*4882a593Smuzhiyun 
1702*4882a593Smuzhiyun /*
1703*4882a593Smuzhiyun  * First pass: Mark the head of each jump table so that in the next pass,
1704*4882a593Smuzhiyun  * we know when a given jump table ends and the next one starts.
1705*4882a593Smuzhiyun  */
mark_func_jump_tables(struct objtool_file * file,struct symbol * func)1706*4882a593Smuzhiyun static void mark_func_jump_tables(struct objtool_file *file,
1707*4882a593Smuzhiyun 				    struct symbol *func)
1708*4882a593Smuzhiyun {
1709*4882a593Smuzhiyun 	struct instruction *insn, *last = NULL;
1710*4882a593Smuzhiyun 	struct reloc *reloc;
1711*4882a593Smuzhiyun 
1712*4882a593Smuzhiyun 	func_for_each_insn(file, func, insn) {
1713*4882a593Smuzhiyun 		if (!last)
1714*4882a593Smuzhiyun 			last = insn;
1715*4882a593Smuzhiyun 
1716*4882a593Smuzhiyun 		/*
1717*4882a593Smuzhiyun 		 * Store back-pointers for unconditional forward jumps such
1718*4882a593Smuzhiyun 		 * that find_jump_table() can back-track using those and
1719*4882a593Smuzhiyun 		 * avoid some potentially confusing code.
1720*4882a593Smuzhiyun 		 */
1721*4882a593Smuzhiyun 		if (insn->type == INSN_JUMP_UNCONDITIONAL && insn->jump_dest &&
1722*4882a593Smuzhiyun 		    insn->offset > last->offset &&
1723*4882a593Smuzhiyun 		    insn->jump_dest->offset > insn->offset &&
1724*4882a593Smuzhiyun 		    !insn->jump_dest->first_jump_src) {
1725*4882a593Smuzhiyun 
1726*4882a593Smuzhiyun 			insn->jump_dest->first_jump_src = insn;
1727*4882a593Smuzhiyun 			last = insn->jump_dest;
1728*4882a593Smuzhiyun 		}
1729*4882a593Smuzhiyun 
1730*4882a593Smuzhiyun 		if (insn->type != INSN_JUMP_DYNAMIC)
1731*4882a593Smuzhiyun 			continue;
1732*4882a593Smuzhiyun 
1733*4882a593Smuzhiyun 		reloc = find_jump_table(file, func, insn);
1734*4882a593Smuzhiyun 		if (reloc) {
1735*4882a593Smuzhiyun 			reloc->jump_table_start = true;
1736*4882a593Smuzhiyun 			insn->jump_table = reloc;
1737*4882a593Smuzhiyun 		}
1738*4882a593Smuzhiyun 	}
1739*4882a593Smuzhiyun }
1740*4882a593Smuzhiyun 
add_func_jump_tables(struct objtool_file * file,struct symbol * func)1741*4882a593Smuzhiyun static int add_func_jump_tables(struct objtool_file *file,
1742*4882a593Smuzhiyun 				  struct symbol *func)
1743*4882a593Smuzhiyun {
1744*4882a593Smuzhiyun 	struct instruction *insn;
1745*4882a593Smuzhiyun 	int ret;
1746*4882a593Smuzhiyun 
1747*4882a593Smuzhiyun 	func_for_each_insn(file, func, insn) {
1748*4882a593Smuzhiyun 		if (!insn->jump_table)
1749*4882a593Smuzhiyun 			continue;
1750*4882a593Smuzhiyun 
1751*4882a593Smuzhiyun 		ret = add_jump_table(file, insn, insn->jump_table);
1752*4882a593Smuzhiyun 		if (ret)
1753*4882a593Smuzhiyun 			return ret;
1754*4882a593Smuzhiyun 	}
1755*4882a593Smuzhiyun 
1756*4882a593Smuzhiyun 	return 0;
1757*4882a593Smuzhiyun }
1758*4882a593Smuzhiyun 
1759*4882a593Smuzhiyun /*
1760*4882a593Smuzhiyun  * For some switch statements, gcc generates a jump table in the .rodata
1761*4882a593Smuzhiyun  * section which contains a list of addresses within the function to jump to.
1762*4882a593Smuzhiyun  * This finds these jump tables and adds them to the insn->alts lists.
1763*4882a593Smuzhiyun  */
add_jump_table_alts(struct objtool_file * file)1764*4882a593Smuzhiyun static int add_jump_table_alts(struct objtool_file *file)
1765*4882a593Smuzhiyun {
1766*4882a593Smuzhiyun 	struct section *sec;
1767*4882a593Smuzhiyun 	struct symbol *func;
1768*4882a593Smuzhiyun 	int ret;
1769*4882a593Smuzhiyun 
1770*4882a593Smuzhiyun 	if (!file->rodata)
1771*4882a593Smuzhiyun 		return 0;
1772*4882a593Smuzhiyun 
1773*4882a593Smuzhiyun 	for_each_sec(file, sec) {
1774*4882a593Smuzhiyun 		list_for_each_entry(func, &sec->symbol_list, list) {
1775*4882a593Smuzhiyun 			if (func->type != STT_FUNC)
1776*4882a593Smuzhiyun 				continue;
1777*4882a593Smuzhiyun 
1778*4882a593Smuzhiyun 			mark_func_jump_tables(file, func);
1779*4882a593Smuzhiyun 			ret = add_func_jump_tables(file, func);
1780*4882a593Smuzhiyun 			if (ret)
1781*4882a593Smuzhiyun 				return ret;
1782*4882a593Smuzhiyun 		}
1783*4882a593Smuzhiyun 	}
1784*4882a593Smuzhiyun 
1785*4882a593Smuzhiyun 	return 0;
1786*4882a593Smuzhiyun }
1787*4882a593Smuzhiyun 
set_func_state(struct cfi_state * state)1788*4882a593Smuzhiyun static void set_func_state(struct cfi_state *state)
1789*4882a593Smuzhiyun {
1790*4882a593Smuzhiyun 	state->cfa = initial_func_cfi.cfa;
1791*4882a593Smuzhiyun 	memcpy(&state->regs, &initial_func_cfi.regs,
1792*4882a593Smuzhiyun 	       CFI_NUM_REGS * sizeof(struct cfi_reg));
1793*4882a593Smuzhiyun 	state->stack_size = initial_func_cfi.cfa.offset;
1794*4882a593Smuzhiyun }
1795*4882a593Smuzhiyun 
read_unwind_hints(struct objtool_file * file)1796*4882a593Smuzhiyun static int read_unwind_hints(struct objtool_file *file)
1797*4882a593Smuzhiyun {
1798*4882a593Smuzhiyun 	struct cfi_state cfi = init_cfi;
1799*4882a593Smuzhiyun 	struct section *sec, *relocsec;
1800*4882a593Smuzhiyun 	struct unwind_hint *hint;
1801*4882a593Smuzhiyun 	struct instruction *insn;
1802*4882a593Smuzhiyun 	struct reloc *reloc;
1803*4882a593Smuzhiyun 	int i;
1804*4882a593Smuzhiyun 
1805*4882a593Smuzhiyun 	sec = find_section_by_name(file->elf, ".discard.unwind_hints");
1806*4882a593Smuzhiyun 	if (!sec)
1807*4882a593Smuzhiyun 		return 0;
1808*4882a593Smuzhiyun 
1809*4882a593Smuzhiyun 	relocsec = sec->reloc;
1810*4882a593Smuzhiyun 	if (!relocsec) {
1811*4882a593Smuzhiyun 		WARN("missing .rela.discard.unwind_hints section");
1812*4882a593Smuzhiyun 		return -1;
1813*4882a593Smuzhiyun 	}
1814*4882a593Smuzhiyun 
1815*4882a593Smuzhiyun 	if (sec->len % sizeof(struct unwind_hint)) {
1816*4882a593Smuzhiyun 		WARN("struct unwind_hint size mismatch");
1817*4882a593Smuzhiyun 		return -1;
1818*4882a593Smuzhiyun 	}
1819*4882a593Smuzhiyun 
1820*4882a593Smuzhiyun 	file->hints = true;
1821*4882a593Smuzhiyun 
1822*4882a593Smuzhiyun 	for (i = 0; i < sec->len / sizeof(struct unwind_hint); i++) {
1823*4882a593Smuzhiyun 		hint = (struct unwind_hint *)sec->data->d_buf + i;
1824*4882a593Smuzhiyun 
1825*4882a593Smuzhiyun 		reloc = find_reloc_by_dest(file->elf, sec, i * sizeof(*hint));
1826*4882a593Smuzhiyun 		if (!reloc) {
1827*4882a593Smuzhiyun 			WARN("can't find reloc for unwind_hints[%d]", i);
1828*4882a593Smuzhiyun 			return -1;
1829*4882a593Smuzhiyun 		}
1830*4882a593Smuzhiyun 
1831*4882a593Smuzhiyun 		insn = find_insn(file, reloc->sym->sec, reloc->addend);
1832*4882a593Smuzhiyun 		if (!insn) {
1833*4882a593Smuzhiyun 			WARN("can't find insn for unwind_hints[%d]", i);
1834*4882a593Smuzhiyun 			return -1;
1835*4882a593Smuzhiyun 		}
1836*4882a593Smuzhiyun 
1837*4882a593Smuzhiyun 		insn->hint = true;
1838*4882a593Smuzhiyun 
1839*4882a593Smuzhiyun 		if (hint->type == UNWIND_HINT_TYPE_SAVE) {
1840*4882a593Smuzhiyun 			insn->hint = false;
1841*4882a593Smuzhiyun 			insn->save = true;
1842*4882a593Smuzhiyun 			continue;
1843*4882a593Smuzhiyun 		}
1844*4882a593Smuzhiyun 
1845*4882a593Smuzhiyun 		if (hint->type == UNWIND_HINT_TYPE_RESTORE) {
1846*4882a593Smuzhiyun 			insn->restore = true;
1847*4882a593Smuzhiyun 			continue;
1848*4882a593Smuzhiyun 		}
1849*4882a593Smuzhiyun 
1850*4882a593Smuzhiyun 		if (hint->type == UNWIND_HINT_TYPE_REGS_PARTIAL) {
1851*4882a593Smuzhiyun 			struct symbol *sym = find_symbol_by_offset(insn->sec, insn->offset);
1852*4882a593Smuzhiyun 
1853*4882a593Smuzhiyun 			if (sym && sym->bind == STB_GLOBAL) {
1854*4882a593Smuzhiyun 				insn->entry = 1;
1855*4882a593Smuzhiyun 			}
1856*4882a593Smuzhiyun 		}
1857*4882a593Smuzhiyun 
1858*4882a593Smuzhiyun 		if (hint->type == UNWIND_HINT_TYPE_ENTRY) {
1859*4882a593Smuzhiyun 			hint->type = UNWIND_HINT_TYPE_CALL;
1860*4882a593Smuzhiyun 			insn->entry = 1;
1861*4882a593Smuzhiyun 		}
1862*4882a593Smuzhiyun 
1863*4882a593Smuzhiyun 		if (hint->type == UNWIND_HINT_TYPE_FUNC) {
1864*4882a593Smuzhiyun 			insn->cfi = &func_cfi;
1865*4882a593Smuzhiyun 			continue;
1866*4882a593Smuzhiyun 		}
1867*4882a593Smuzhiyun 
1868*4882a593Smuzhiyun 		if (insn->cfi)
1869*4882a593Smuzhiyun 			cfi = *(insn->cfi);
1870*4882a593Smuzhiyun 
1871*4882a593Smuzhiyun 		if (arch_decode_hint_reg(hint->sp_reg, &cfi.cfa.base)) {
1872*4882a593Smuzhiyun 			WARN_FUNC("unsupported unwind_hint sp base reg %d",
1873*4882a593Smuzhiyun 				  insn->sec, insn->offset, hint->sp_reg);
1874*4882a593Smuzhiyun 			return -1;
1875*4882a593Smuzhiyun 		}
1876*4882a593Smuzhiyun 
1877*4882a593Smuzhiyun 		cfi.cfa.offset = hint->sp_offset;
1878*4882a593Smuzhiyun 		cfi.type = hint->type;
1879*4882a593Smuzhiyun 		cfi.end = hint->end;
1880*4882a593Smuzhiyun 
1881*4882a593Smuzhiyun 		insn->cfi = cfi_hash_find_or_add(&cfi);
1882*4882a593Smuzhiyun 	}
1883*4882a593Smuzhiyun 
1884*4882a593Smuzhiyun 	return 0;
1885*4882a593Smuzhiyun }
1886*4882a593Smuzhiyun 
read_retpoline_hints(struct objtool_file * file)1887*4882a593Smuzhiyun static int read_retpoline_hints(struct objtool_file *file)
1888*4882a593Smuzhiyun {
1889*4882a593Smuzhiyun 	struct section *sec;
1890*4882a593Smuzhiyun 	struct instruction *insn;
1891*4882a593Smuzhiyun 	struct reloc *reloc;
1892*4882a593Smuzhiyun 
1893*4882a593Smuzhiyun 	sec = find_section_by_name(file->elf, ".rela.discard.retpoline_safe");
1894*4882a593Smuzhiyun 	if (!sec)
1895*4882a593Smuzhiyun 		return 0;
1896*4882a593Smuzhiyun 
1897*4882a593Smuzhiyun 	list_for_each_entry(reloc, &sec->reloc_list, list) {
1898*4882a593Smuzhiyun 		if (reloc->sym->type != STT_SECTION) {
1899*4882a593Smuzhiyun 			WARN("unexpected relocation symbol type in %s", sec->name);
1900*4882a593Smuzhiyun 			return -1;
1901*4882a593Smuzhiyun 		}
1902*4882a593Smuzhiyun 
1903*4882a593Smuzhiyun 		insn = find_insn(file, reloc->sym->sec, reloc->addend);
1904*4882a593Smuzhiyun 		if (!insn) {
1905*4882a593Smuzhiyun 			WARN("bad .discard.retpoline_safe entry");
1906*4882a593Smuzhiyun 			return -1;
1907*4882a593Smuzhiyun 		}
1908*4882a593Smuzhiyun 
1909*4882a593Smuzhiyun 		if (insn->type != INSN_JUMP_DYNAMIC &&
1910*4882a593Smuzhiyun 		    insn->type != INSN_CALL_DYNAMIC &&
1911*4882a593Smuzhiyun 		    insn->type != INSN_RETURN &&
1912*4882a593Smuzhiyun 		    insn->type != INSN_NOP) {
1913*4882a593Smuzhiyun 			WARN_FUNC("retpoline_safe hint not an indirect jump/call/ret/nop",
1914*4882a593Smuzhiyun 				  insn->sec, insn->offset);
1915*4882a593Smuzhiyun 			return -1;
1916*4882a593Smuzhiyun 		}
1917*4882a593Smuzhiyun 
1918*4882a593Smuzhiyun 		insn->retpoline_safe = true;
1919*4882a593Smuzhiyun 	}
1920*4882a593Smuzhiyun 
1921*4882a593Smuzhiyun 	return 0;
1922*4882a593Smuzhiyun }
1923*4882a593Smuzhiyun 
read_instr_hints(struct objtool_file * file)1924*4882a593Smuzhiyun static int read_instr_hints(struct objtool_file *file)
1925*4882a593Smuzhiyun {
1926*4882a593Smuzhiyun 	struct section *sec;
1927*4882a593Smuzhiyun 	struct instruction *insn;
1928*4882a593Smuzhiyun 	struct reloc *reloc;
1929*4882a593Smuzhiyun 
1930*4882a593Smuzhiyun 	sec = find_section_by_name(file->elf, ".rela.discard.instr_end");
1931*4882a593Smuzhiyun 	if (!sec)
1932*4882a593Smuzhiyun 		return 0;
1933*4882a593Smuzhiyun 
1934*4882a593Smuzhiyun 	list_for_each_entry(reloc, &sec->reloc_list, list) {
1935*4882a593Smuzhiyun 		if (reloc->sym->type != STT_SECTION) {
1936*4882a593Smuzhiyun 			WARN("unexpected relocation symbol type in %s", sec->name);
1937*4882a593Smuzhiyun 			return -1;
1938*4882a593Smuzhiyun 		}
1939*4882a593Smuzhiyun 
1940*4882a593Smuzhiyun 		insn = find_insn(file, reloc->sym->sec, reloc->addend);
1941*4882a593Smuzhiyun 		if (!insn) {
1942*4882a593Smuzhiyun 			WARN("bad .discard.instr_end entry");
1943*4882a593Smuzhiyun 			return -1;
1944*4882a593Smuzhiyun 		}
1945*4882a593Smuzhiyun 
1946*4882a593Smuzhiyun 		insn->instr--;
1947*4882a593Smuzhiyun 	}
1948*4882a593Smuzhiyun 
1949*4882a593Smuzhiyun 	sec = find_section_by_name(file->elf, ".rela.discard.instr_begin");
1950*4882a593Smuzhiyun 	if (!sec)
1951*4882a593Smuzhiyun 		return 0;
1952*4882a593Smuzhiyun 
1953*4882a593Smuzhiyun 	list_for_each_entry(reloc, &sec->reloc_list, list) {
1954*4882a593Smuzhiyun 		if (reloc->sym->type != STT_SECTION) {
1955*4882a593Smuzhiyun 			WARN("unexpected relocation symbol type in %s", sec->name);
1956*4882a593Smuzhiyun 			return -1;
1957*4882a593Smuzhiyun 		}
1958*4882a593Smuzhiyun 
1959*4882a593Smuzhiyun 		insn = find_insn(file, reloc->sym->sec, reloc->addend);
1960*4882a593Smuzhiyun 		if (!insn) {
1961*4882a593Smuzhiyun 			WARN("bad .discard.instr_begin entry");
1962*4882a593Smuzhiyun 			return -1;
1963*4882a593Smuzhiyun 		}
1964*4882a593Smuzhiyun 
1965*4882a593Smuzhiyun 		insn->instr++;
1966*4882a593Smuzhiyun 	}
1967*4882a593Smuzhiyun 
1968*4882a593Smuzhiyun 	return 0;
1969*4882a593Smuzhiyun }
1970*4882a593Smuzhiyun 
read_intra_function_calls(struct objtool_file * file)1971*4882a593Smuzhiyun static int read_intra_function_calls(struct objtool_file *file)
1972*4882a593Smuzhiyun {
1973*4882a593Smuzhiyun 	struct instruction *insn;
1974*4882a593Smuzhiyun 	struct section *sec;
1975*4882a593Smuzhiyun 	struct reloc *reloc;
1976*4882a593Smuzhiyun 
1977*4882a593Smuzhiyun 	sec = find_section_by_name(file->elf, ".rela.discard.intra_function_calls");
1978*4882a593Smuzhiyun 	if (!sec)
1979*4882a593Smuzhiyun 		return 0;
1980*4882a593Smuzhiyun 
1981*4882a593Smuzhiyun 	list_for_each_entry(reloc, &sec->reloc_list, list) {
1982*4882a593Smuzhiyun 		unsigned long dest_off;
1983*4882a593Smuzhiyun 
1984*4882a593Smuzhiyun 		if (reloc->sym->type != STT_SECTION) {
1985*4882a593Smuzhiyun 			WARN("unexpected relocation symbol type in %s",
1986*4882a593Smuzhiyun 			     sec->name);
1987*4882a593Smuzhiyun 			return -1;
1988*4882a593Smuzhiyun 		}
1989*4882a593Smuzhiyun 
1990*4882a593Smuzhiyun 		insn = find_insn(file, reloc->sym->sec, reloc->addend);
1991*4882a593Smuzhiyun 		if (!insn) {
1992*4882a593Smuzhiyun 			WARN("bad .discard.intra_function_call entry");
1993*4882a593Smuzhiyun 			return -1;
1994*4882a593Smuzhiyun 		}
1995*4882a593Smuzhiyun 
1996*4882a593Smuzhiyun 		if (insn->type != INSN_CALL) {
1997*4882a593Smuzhiyun 			WARN_FUNC("intra_function_call not a direct call",
1998*4882a593Smuzhiyun 				  insn->sec, insn->offset);
1999*4882a593Smuzhiyun 			return -1;
2000*4882a593Smuzhiyun 		}
2001*4882a593Smuzhiyun 
2002*4882a593Smuzhiyun 		/*
2003*4882a593Smuzhiyun 		 * Treat intra-function CALLs as JMPs, but with a stack_op.
2004*4882a593Smuzhiyun 		 * See add_call_destinations(), which strips stack_ops from
2005*4882a593Smuzhiyun 		 * normal CALLs.
2006*4882a593Smuzhiyun 		 */
2007*4882a593Smuzhiyun 		insn->type = INSN_JUMP_UNCONDITIONAL;
2008*4882a593Smuzhiyun 
2009*4882a593Smuzhiyun 		dest_off = insn->offset + insn->len + insn->immediate;
2010*4882a593Smuzhiyun 		insn->jump_dest = find_insn(file, insn->sec, dest_off);
2011*4882a593Smuzhiyun 		if (!insn->jump_dest) {
2012*4882a593Smuzhiyun 			WARN_FUNC("can't find call dest at %s+0x%lx",
2013*4882a593Smuzhiyun 				  insn->sec, insn->offset,
2014*4882a593Smuzhiyun 				  insn->sec->name, dest_off);
2015*4882a593Smuzhiyun 			return -1;
2016*4882a593Smuzhiyun 		}
2017*4882a593Smuzhiyun 	}
2018*4882a593Smuzhiyun 
2019*4882a593Smuzhiyun 	return 0;
2020*4882a593Smuzhiyun }
2021*4882a593Smuzhiyun 
classify_symbols(struct objtool_file * file)2022*4882a593Smuzhiyun static int classify_symbols(struct objtool_file *file)
2023*4882a593Smuzhiyun {
2024*4882a593Smuzhiyun 	struct section *sec;
2025*4882a593Smuzhiyun 	struct symbol *func;
2026*4882a593Smuzhiyun 
2027*4882a593Smuzhiyun 	for_each_sec(file, sec) {
2028*4882a593Smuzhiyun 		list_for_each_entry(func, &sec->symbol_list, list) {
2029*4882a593Smuzhiyun 			if (func->bind != STB_GLOBAL)
2030*4882a593Smuzhiyun 				continue;
2031*4882a593Smuzhiyun 
2032*4882a593Smuzhiyun 			if (!strncmp(func->name, STATIC_CALL_TRAMP_PREFIX_STR,
2033*4882a593Smuzhiyun 				     strlen(STATIC_CALL_TRAMP_PREFIX_STR)))
2034*4882a593Smuzhiyun 				func->static_call_tramp = true;
2035*4882a593Smuzhiyun 
2036*4882a593Smuzhiyun 			if (arch_is_retpoline(func))
2037*4882a593Smuzhiyun 				func->retpoline_thunk = true;
2038*4882a593Smuzhiyun 
2039*4882a593Smuzhiyun 			if (arch_is_rethunk(func))
2040*4882a593Smuzhiyun 				func->return_thunk = true;
2041*4882a593Smuzhiyun 
2042*4882a593Smuzhiyun 			if (!strcmp(func->name, "__fentry__"))
2043*4882a593Smuzhiyun 				func->fentry = true;
2044*4882a593Smuzhiyun 
2045*4882a593Smuzhiyun 			if (!strncmp(func->name, "__sanitizer_cov_", 16))
2046*4882a593Smuzhiyun 				func->kcov = true;
2047*4882a593Smuzhiyun 		}
2048*4882a593Smuzhiyun 	}
2049*4882a593Smuzhiyun 
2050*4882a593Smuzhiyun 	return 0;
2051*4882a593Smuzhiyun }
2052*4882a593Smuzhiyun 
mark_rodata(struct objtool_file * file)2053*4882a593Smuzhiyun static void mark_rodata(struct objtool_file *file)
2054*4882a593Smuzhiyun {
2055*4882a593Smuzhiyun 	struct section *sec;
2056*4882a593Smuzhiyun 	bool found = false;
2057*4882a593Smuzhiyun 
2058*4882a593Smuzhiyun 	/*
2059*4882a593Smuzhiyun 	 * Search for the following rodata sections, each of which can
2060*4882a593Smuzhiyun 	 * potentially contain jump tables:
2061*4882a593Smuzhiyun 	 *
2062*4882a593Smuzhiyun 	 * - .rodata: can contain GCC switch tables
2063*4882a593Smuzhiyun 	 * - .rodata.<func>: same, if -fdata-sections is being used
2064*4882a593Smuzhiyun 	 * - .rodata..c_jump_table: contains C annotated jump tables
2065*4882a593Smuzhiyun 	 *
2066*4882a593Smuzhiyun 	 * .rodata.str1.* sections are ignored; they don't contain jump tables.
2067*4882a593Smuzhiyun 	 */
2068*4882a593Smuzhiyun 	for_each_sec(file, sec) {
2069*4882a593Smuzhiyun 		if (!strncmp(sec->name, ".rodata", 7) &&
2070*4882a593Smuzhiyun 		    !strstr(sec->name, ".str1.")) {
2071*4882a593Smuzhiyun 			sec->rodata = true;
2072*4882a593Smuzhiyun 			found = true;
2073*4882a593Smuzhiyun 		}
2074*4882a593Smuzhiyun 	}
2075*4882a593Smuzhiyun 
2076*4882a593Smuzhiyun 	file->rodata = found;
2077*4882a593Smuzhiyun }
2078*4882a593Smuzhiyun 
decode_sections(struct objtool_file * file)2079*4882a593Smuzhiyun static int decode_sections(struct objtool_file *file)
2080*4882a593Smuzhiyun {
2081*4882a593Smuzhiyun 	int ret;
2082*4882a593Smuzhiyun 
2083*4882a593Smuzhiyun 	mark_rodata(file);
2084*4882a593Smuzhiyun 
2085*4882a593Smuzhiyun 	ret = decode_instructions(file);
2086*4882a593Smuzhiyun 	if (ret)
2087*4882a593Smuzhiyun 		return ret;
2088*4882a593Smuzhiyun 
2089*4882a593Smuzhiyun 	ret = add_dead_ends(file);
2090*4882a593Smuzhiyun 	if (ret)
2091*4882a593Smuzhiyun 		return ret;
2092*4882a593Smuzhiyun 
2093*4882a593Smuzhiyun 	add_ignores(file);
2094*4882a593Smuzhiyun 	add_uaccess_safe(file);
2095*4882a593Smuzhiyun 	add_cfi_jumptables(file);
2096*4882a593Smuzhiyun 
2097*4882a593Smuzhiyun 	ret = add_ignore_alternatives(file);
2098*4882a593Smuzhiyun 	if (ret)
2099*4882a593Smuzhiyun 		return ret;
2100*4882a593Smuzhiyun 
2101*4882a593Smuzhiyun 	/*
2102*4882a593Smuzhiyun 	 * Must be before add_{jump_call}_destination.
2103*4882a593Smuzhiyun 	 */
2104*4882a593Smuzhiyun 	ret = classify_symbols(file);
2105*4882a593Smuzhiyun 	if (ret)
2106*4882a593Smuzhiyun 		return ret;
2107*4882a593Smuzhiyun 
2108*4882a593Smuzhiyun 	/*
2109*4882a593Smuzhiyun 	 * Must be before add_special_section_alts() as that depends on
2110*4882a593Smuzhiyun 	 * jump_dest being set.
2111*4882a593Smuzhiyun 	 */
2112*4882a593Smuzhiyun 	ret = add_jump_destinations(file);
2113*4882a593Smuzhiyun 	if (ret)
2114*4882a593Smuzhiyun 		return ret;
2115*4882a593Smuzhiyun 
2116*4882a593Smuzhiyun 	ret = add_special_section_alts(file);
2117*4882a593Smuzhiyun 	if (ret)
2118*4882a593Smuzhiyun 		return ret;
2119*4882a593Smuzhiyun 
2120*4882a593Smuzhiyun 	/*
2121*4882a593Smuzhiyun 	 * Must be before add_call_destination(); it changes INSN_CALL to
2122*4882a593Smuzhiyun 	 * INSN_JUMP.
2123*4882a593Smuzhiyun 	 */
2124*4882a593Smuzhiyun 	ret = read_intra_function_calls(file);
2125*4882a593Smuzhiyun 	if (ret)
2126*4882a593Smuzhiyun 		return ret;
2127*4882a593Smuzhiyun 
2128*4882a593Smuzhiyun 	ret = add_call_destinations(file);
2129*4882a593Smuzhiyun 	if (ret)
2130*4882a593Smuzhiyun 		return ret;
2131*4882a593Smuzhiyun 
2132*4882a593Smuzhiyun 	ret = add_jump_table_alts(file);
2133*4882a593Smuzhiyun 	if (ret)
2134*4882a593Smuzhiyun 		return ret;
2135*4882a593Smuzhiyun 
2136*4882a593Smuzhiyun 	ret = read_unwind_hints(file);
2137*4882a593Smuzhiyun 	if (ret)
2138*4882a593Smuzhiyun 		return ret;
2139*4882a593Smuzhiyun 
2140*4882a593Smuzhiyun 	ret = read_retpoline_hints(file);
2141*4882a593Smuzhiyun 	if (ret)
2142*4882a593Smuzhiyun 		return ret;
2143*4882a593Smuzhiyun 
2144*4882a593Smuzhiyun 	ret = read_instr_hints(file);
2145*4882a593Smuzhiyun 	if (ret)
2146*4882a593Smuzhiyun 		return ret;
2147*4882a593Smuzhiyun 
2148*4882a593Smuzhiyun 	return 0;
2149*4882a593Smuzhiyun }
2150*4882a593Smuzhiyun 
is_fentry_call(struct instruction * insn)2151*4882a593Smuzhiyun static bool is_fentry_call(struct instruction *insn)
2152*4882a593Smuzhiyun {
2153*4882a593Smuzhiyun 	if (insn->type == INSN_CALL &&
2154*4882a593Smuzhiyun 	    insn->call_dest &&
2155*4882a593Smuzhiyun 	    insn->call_dest->fentry)
2156*4882a593Smuzhiyun 		return true;
2157*4882a593Smuzhiyun 
2158*4882a593Smuzhiyun 	return false;
2159*4882a593Smuzhiyun }
2160*4882a593Smuzhiyun 
has_modified_stack_frame(struct instruction * insn,struct insn_state * state)2161*4882a593Smuzhiyun static bool has_modified_stack_frame(struct instruction *insn, struct insn_state *state)
2162*4882a593Smuzhiyun {
2163*4882a593Smuzhiyun 	struct cfi_state *cfi = &state->cfi;
2164*4882a593Smuzhiyun 	int i;
2165*4882a593Smuzhiyun 
2166*4882a593Smuzhiyun 	if (cfi->cfa.base != initial_func_cfi.cfa.base || cfi->drap)
2167*4882a593Smuzhiyun 		return true;
2168*4882a593Smuzhiyun 
2169*4882a593Smuzhiyun 	if (cfi->cfa.offset != initial_func_cfi.cfa.offset)
2170*4882a593Smuzhiyun 		return true;
2171*4882a593Smuzhiyun 
2172*4882a593Smuzhiyun 	if (cfi->stack_size != initial_func_cfi.cfa.offset)
2173*4882a593Smuzhiyun 		return true;
2174*4882a593Smuzhiyun 
2175*4882a593Smuzhiyun 	for (i = 0; i < CFI_NUM_REGS; i++) {
2176*4882a593Smuzhiyun 		if (cfi->regs[i].base != initial_func_cfi.regs[i].base ||
2177*4882a593Smuzhiyun 		    cfi->regs[i].offset != initial_func_cfi.regs[i].offset)
2178*4882a593Smuzhiyun 			return true;
2179*4882a593Smuzhiyun 	}
2180*4882a593Smuzhiyun 
2181*4882a593Smuzhiyun 	return false;
2182*4882a593Smuzhiyun }
2183*4882a593Smuzhiyun 
has_valid_stack_frame(struct insn_state * state)2184*4882a593Smuzhiyun static bool has_valid_stack_frame(struct insn_state *state)
2185*4882a593Smuzhiyun {
2186*4882a593Smuzhiyun 	struct cfi_state *cfi = &state->cfi;
2187*4882a593Smuzhiyun 
2188*4882a593Smuzhiyun 	if (cfi->cfa.base == CFI_BP && cfi->regs[CFI_BP].base == CFI_CFA &&
2189*4882a593Smuzhiyun 	    cfi->regs[CFI_BP].offset == -16)
2190*4882a593Smuzhiyun 		return true;
2191*4882a593Smuzhiyun 
2192*4882a593Smuzhiyun 	if (cfi->drap && cfi->regs[CFI_BP].base == CFI_BP)
2193*4882a593Smuzhiyun 		return true;
2194*4882a593Smuzhiyun 
2195*4882a593Smuzhiyun 	return false;
2196*4882a593Smuzhiyun }
2197*4882a593Smuzhiyun 
update_cfi_state_regs(struct instruction * insn,struct cfi_state * cfi,struct stack_op * op)2198*4882a593Smuzhiyun static int update_cfi_state_regs(struct instruction *insn,
2199*4882a593Smuzhiyun 				  struct cfi_state *cfi,
2200*4882a593Smuzhiyun 				  struct stack_op *op)
2201*4882a593Smuzhiyun {
2202*4882a593Smuzhiyun 	struct cfi_reg *cfa = &cfi->cfa;
2203*4882a593Smuzhiyun 
2204*4882a593Smuzhiyun 	if (cfa->base != CFI_SP && cfa->base != CFI_SP_INDIRECT)
2205*4882a593Smuzhiyun 		return 0;
2206*4882a593Smuzhiyun 
2207*4882a593Smuzhiyun 	/* push */
2208*4882a593Smuzhiyun 	if (op->dest.type == OP_DEST_PUSH || op->dest.type == OP_DEST_PUSHF)
2209*4882a593Smuzhiyun 		cfa->offset += 8;
2210*4882a593Smuzhiyun 
2211*4882a593Smuzhiyun 	/* pop */
2212*4882a593Smuzhiyun 	if (op->src.type == OP_SRC_POP || op->src.type == OP_SRC_POPF)
2213*4882a593Smuzhiyun 		cfa->offset -= 8;
2214*4882a593Smuzhiyun 
2215*4882a593Smuzhiyun 	/* add immediate to sp */
2216*4882a593Smuzhiyun 	if (op->dest.type == OP_DEST_REG && op->src.type == OP_SRC_ADD &&
2217*4882a593Smuzhiyun 	    op->dest.reg == CFI_SP && op->src.reg == CFI_SP)
2218*4882a593Smuzhiyun 		cfa->offset -= op->src.offset;
2219*4882a593Smuzhiyun 
2220*4882a593Smuzhiyun 	return 0;
2221*4882a593Smuzhiyun }
2222*4882a593Smuzhiyun 
save_reg(struct cfi_state * cfi,unsigned char reg,int base,int offset)2223*4882a593Smuzhiyun static void save_reg(struct cfi_state *cfi, unsigned char reg, int base, int offset)
2224*4882a593Smuzhiyun {
2225*4882a593Smuzhiyun 	if (arch_callee_saved_reg(reg) &&
2226*4882a593Smuzhiyun 	    cfi->regs[reg].base == CFI_UNDEFINED) {
2227*4882a593Smuzhiyun 		cfi->regs[reg].base = base;
2228*4882a593Smuzhiyun 		cfi->regs[reg].offset = offset;
2229*4882a593Smuzhiyun 	}
2230*4882a593Smuzhiyun }
2231*4882a593Smuzhiyun 
restore_reg(struct cfi_state * cfi,unsigned char reg)2232*4882a593Smuzhiyun static void restore_reg(struct cfi_state *cfi, unsigned char reg)
2233*4882a593Smuzhiyun {
2234*4882a593Smuzhiyun 	cfi->regs[reg].base = initial_func_cfi.regs[reg].base;
2235*4882a593Smuzhiyun 	cfi->regs[reg].offset = initial_func_cfi.regs[reg].offset;
2236*4882a593Smuzhiyun }
2237*4882a593Smuzhiyun 
2238*4882a593Smuzhiyun /*
2239*4882a593Smuzhiyun  * A note about DRAP stack alignment:
2240*4882a593Smuzhiyun  *
2241*4882a593Smuzhiyun  * GCC has the concept of a DRAP register, which is used to help keep track of
2242*4882a593Smuzhiyun  * the stack pointer when aligning the stack.  r10 or r13 is used as the DRAP
2243*4882a593Smuzhiyun  * register.  The typical DRAP pattern is:
2244*4882a593Smuzhiyun  *
2245*4882a593Smuzhiyun  *   4c 8d 54 24 08		lea    0x8(%rsp),%r10
2246*4882a593Smuzhiyun  *   48 83 e4 c0		and    $0xffffffffffffffc0,%rsp
2247*4882a593Smuzhiyun  *   41 ff 72 f8		pushq  -0x8(%r10)
2248*4882a593Smuzhiyun  *   55				push   %rbp
2249*4882a593Smuzhiyun  *   48 89 e5			mov    %rsp,%rbp
2250*4882a593Smuzhiyun  *				(more pushes)
2251*4882a593Smuzhiyun  *   41 52			push   %r10
2252*4882a593Smuzhiyun  *				...
2253*4882a593Smuzhiyun  *   41 5a			pop    %r10
2254*4882a593Smuzhiyun  *				(more pops)
2255*4882a593Smuzhiyun  *   5d				pop    %rbp
2256*4882a593Smuzhiyun  *   49 8d 62 f8		lea    -0x8(%r10),%rsp
2257*4882a593Smuzhiyun  *   c3				retq
2258*4882a593Smuzhiyun  *
2259*4882a593Smuzhiyun  * There are some variations in the epilogues, like:
2260*4882a593Smuzhiyun  *
2261*4882a593Smuzhiyun  *   5b				pop    %rbx
2262*4882a593Smuzhiyun  *   41 5a			pop    %r10
2263*4882a593Smuzhiyun  *   41 5c			pop    %r12
2264*4882a593Smuzhiyun  *   41 5d			pop    %r13
2265*4882a593Smuzhiyun  *   41 5e			pop    %r14
2266*4882a593Smuzhiyun  *   c9				leaveq
2267*4882a593Smuzhiyun  *   49 8d 62 f8		lea    -0x8(%r10),%rsp
2268*4882a593Smuzhiyun  *   c3				retq
2269*4882a593Smuzhiyun  *
2270*4882a593Smuzhiyun  * and:
2271*4882a593Smuzhiyun  *
2272*4882a593Smuzhiyun  *   4c 8b 55 e8		mov    -0x18(%rbp),%r10
2273*4882a593Smuzhiyun  *   48 8b 5d e0		mov    -0x20(%rbp),%rbx
2274*4882a593Smuzhiyun  *   4c 8b 65 f0		mov    -0x10(%rbp),%r12
2275*4882a593Smuzhiyun  *   4c 8b 6d f8		mov    -0x8(%rbp),%r13
2276*4882a593Smuzhiyun  *   c9				leaveq
2277*4882a593Smuzhiyun  *   49 8d 62 f8		lea    -0x8(%r10),%rsp
2278*4882a593Smuzhiyun  *   c3				retq
2279*4882a593Smuzhiyun  *
2280*4882a593Smuzhiyun  * Sometimes r13 is used as the DRAP register, in which case it's saved and
2281*4882a593Smuzhiyun  * restored beforehand:
2282*4882a593Smuzhiyun  *
2283*4882a593Smuzhiyun  *   41 55			push   %r13
2284*4882a593Smuzhiyun  *   4c 8d 6c 24 10		lea    0x10(%rsp),%r13
2285*4882a593Smuzhiyun  *   48 83 e4 f0		and    $0xfffffffffffffff0,%rsp
2286*4882a593Smuzhiyun  *				...
2287*4882a593Smuzhiyun  *   49 8d 65 f0		lea    -0x10(%r13),%rsp
2288*4882a593Smuzhiyun  *   41 5d			pop    %r13
2289*4882a593Smuzhiyun  *   c3				retq
2290*4882a593Smuzhiyun  */
update_cfi_state(struct instruction * insn,struct cfi_state * cfi,struct stack_op * op)2291*4882a593Smuzhiyun static int update_cfi_state(struct instruction *insn, struct cfi_state *cfi,
2292*4882a593Smuzhiyun 			     struct stack_op *op)
2293*4882a593Smuzhiyun {
2294*4882a593Smuzhiyun 	struct cfi_reg *cfa = &cfi->cfa;
2295*4882a593Smuzhiyun 	struct cfi_reg *regs = cfi->regs;
2296*4882a593Smuzhiyun 
2297*4882a593Smuzhiyun 	/* stack operations don't make sense with an undefined CFA */
2298*4882a593Smuzhiyun 	if (cfa->base == CFI_UNDEFINED) {
2299*4882a593Smuzhiyun 		if (insn->func) {
2300*4882a593Smuzhiyun 			WARN_FUNC("undefined stack state", insn->sec, insn->offset);
2301*4882a593Smuzhiyun 			return -1;
2302*4882a593Smuzhiyun 		}
2303*4882a593Smuzhiyun 		return 0;
2304*4882a593Smuzhiyun 	}
2305*4882a593Smuzhiyun 
2306*4882a593Smuzhiyun 	if (cfi->type == UNWIND_HINT_TYPE_REGS ||
2307*4882a593Smuzhiyun 	    cfi->type == UNWIND_HINT_TYPE_REGS_PARTIAL)
2308*4882a593Smuzhiyun 		return update_cfi_state_regs(insn, cfi, op);
2309*4882a593Smuzhiyun 
2310*4882a593Smuzhiyun 	switch (op->dest.type) {
2311*4882a593Smuzhiyun 
2312*4882a593Smuzhiyun 	case OP_DEST_REG:
2313*4882a593Smuzhiyun 		switch (op->src.type) {
2314*4882a593Smuzhiyun 
2315*4882a593Smuzhiyun 		case OP_SRC_REG:
2316*4882a593Smuzhiyun 			if (op->src.reg == CFI_SP && op->dest.reg == CFI_BP &&
2317*4882a593Smuzhiyun 			    cfa->base == CFI_SP &&
2318*4882a593Smuzhiyun 			    regs[CFI_BP].base == CFI_CFA &&
2319*4882a593Smuzhiyun 			    regs[CFI_BP].offset == -cfa->offset) {
2320*4882a593Smuzhiyun 
2321*4882a593Smuzhiyun 				/* mov %rsp, %rbp */
2322*4882a593Smuzhiyun 				cfa->base = op->dest.reg;
2323*4882a593Smuzhiyun 				cfi->bp_scratch = false;
2324*4882a593Smuzhiyun 			}
2325*4882a593Smuzhiyun 
2326*4882a593Smuzhiyun 			else if (op->src.reg == CFI_SP &&
2327*4882a593Smuzhiyun 				 op->dest.reg == CFI_BP && cfi->drap) {
2328*4882a593Smuzhiyun 
2329*4882a593Smuzhiyun 				/* drap: mov %rsp, %rbp */
2330*4882a593Smuzhiyun 				regs[CFI_BP].base = CFI_BP;
2331*4882a593Smuzhiyun 				regs[CFI_BP].offset = -cfi->stack_size;
2332*4882a593Smuzhiyun 				cfi->bp_scratch = false;
2333*4882a593Smuzhiyun 			}
2334*4882a593Smuzhiyun 
2335*4882a593Smuzhiyun 			else if (op->src.reg == CFI_SP && cfa->base == CFI_SP) {
2336*4882a593Smuzhiyun 
2337*4882a593Smuzhiyun 				/*
2338*4882a593Smuzhiyun 				 * mov %rsp, %reg
2339*4882a593Smuzhiyun 				 *
2340*4882a593Smuzhiyun 				 * This is needed for the rare case where GCC
2341*4882a593Smuzhiyun 				 * does:
2342*4882a593Smuzhiyun 				 *
2343*4882a593Smuzhiyun 				 *   mov    %rsp, %rax
2344*4882a593Smuzhiyun 				 *   ...
2345*4882a593Smuzhiyun 				 *   mov    %rax, %rsp
2346*4882a593Smuzhiyun 				 */
2347*4882a593Smuzhiyun 				cfi->vals[op->dest.reg].base = CFI_CFA;
2348*4882a593Smuzhiyun 				cfi->vals[op->dest.reg].offset = -cfi->stack_size;
2349*4882a593Smuzhiyun 			}
2350*4882a593Smuzhiyun 
2351*4882a593Smuzhiyun 			else if (op->src.reg == CFI_BP && op->dest.reg == CFI_SP &&
2352*4882a593Smuzhiyun 				 cfa->base == CFI_BP) {
2353*4882a593Smuzhiyun 
2354*4882a593Smuzhiyun 				/*
2355*4882a593Smuzhiyun 				 * mov %rbp, %rsp
2356*4882a593Smuzhiyun 				 *
2357*4882a593Smuzhiyun 				 * Restore the original stack pointer (Clang).
2358*4882a593Smuzhiyun 				 */
2359*4882a593Smuzhiyun 				cfi->stack_size = -cfi->regs[CFI_BP].offset;
2360*4882a593Smuzhiyun 			}
2361*4882a593Smuzhiyun 
2362*4882a593Smuzhiyun 			else if (op->dest.reg == cfa->base) {
2363*4882a593Smuzhiyun 
2364*4882a593Smuzhiyun 				/* mov %reg, %rsp */
2365*4882a593Smuzhiyun 				if (cfa->base == CFI_SP &&
2366*4882a593Smuzhiyun 				    cfi->vals[op->src.reg].base == CFI_CFA) {
2367*4882a593Smuzhiyun 
2368*4882a593Smuzhiyun 					/*
2369*4882a593Smuzhiyun 					 * This is needed for the rare case
2370*4882a593Smuzhiyun 					 * where GCC does something dumb like:
2371*4882a593Smuzhiyun 					 *
2372*4882a593Smuzhiyun 					 *   lea    0x8(%rsp), %rcx
2373*4882a593Smuzhiyun 					 *   ...
2374*4882a593Smuzhiyun 					 *   mov    %rcx, %rsp
2375*4882a593Smuzhiyun 					 */
2376*4882a593Smuzhiyun 					cfa->offset = -cfi->vals[op->src.reg].offset;
2377*4882a593Smuzhiyun 					cfi->stack_size = cfa->offset;
2378*4882a593Smuzhiyun 
2379*4882a593Smuzhiyun 				} else {
2380*4882a593Smuzhiyun 					cfa->base = CFI_UNDEFINED;
2381*4882a593Smuzhiyun 					cfa->offset = 0;
2382*4882a593Smuzhiyun 				}
2383*4882a593Smuzhiyun 			}
2384*4882a593Smuzhiyun 
2385*4882a593Smuzhiyun 			break;
2386*4882a593Smuzhiyun 
2387*4882a593Smuzhiyun 		case OP_SRC_ADD:
2388*4882a593Smuzhiyun 			if (op->dest.reg == CFI_SP && op->src.reg == CFI_SP) {
2389*4882a593Smuzhiyun 
2390*4882a593Smuzhiyun 				/* add imm, %rsp */
2391*4882a593Smuzhiyun 				cfi->stack_size -= op->src.offset;
2392*4882a593Smuzhiyun 				if (cfa->base == CFI_SP)
2393*4882a593Smuzhiyun 					cfa->offset -= op->src.offset;
2394*4882a593Smuzhiyun 				break;
2395*4882a593Smuzhiyun 			}
2396*4882a593Smuzhiyun 
2397*4882a593Smuzhiyun 			if (op->dest.reg == CFI_SP && op->src.reg == CFI_BP) {
2398*4882a593Smuzhiyun 
2399*4882a593Smuzhiyun 				/* lea disp(%rbp), %rsp */
2400*4882a593Smuzhiyun 				cfi->stack_size = -(op->src.offset + regs[CFI_BP].offset);
2401*4882a593Smuzhiyun 				break;
2402*4882a593Smuzhiyun 			}
2403*4882a593Smuzhiyun 
2404*4882a593Smuzhiyun 			if (op->src.reg == CFI_SP && cfa->base == CFI_SP) {
2405*4882a593Smuzhiyun 
2406*4882a593Smuzhiyun 				/* drap: lea disp(%rsp), %drap */
2407*4882a593Smuzhiyun 				cfi->drap_reg = op->dest.reg;
2408*4882a593Smuzhiyun 
2409*4882a593Smuzhiyun 				/*
2410*4882a593Smuzhiyun 				 * lea disp(%rsp), %reg
2411*4882a593Smuzhiyun 				 *
2412*4882a593Smuzhiyun 				 * This is needed for the rare case where GCC
2413*4882a593Smuzhiyun 				 * does something dumb like:
2414*4882a593Smuzhiyun 				 *
2415*4882a593Smuzhiyun 				 *   lea    0x8(%rsp), %rcx
2416*4882a593Smuzhiyun 				 *   ...
2417*4882a593Smuzhiyun 				 *   mov    %rcx, %rsp
2418*4882a593Smuzhiyun 				 */
2419*4882a593Smuzhiyun 				cfi->vals[op->dest.reg].base = CFI_CFA;
2420*4882a593Smuzhiyun 				cfi->vals[op->dest.reg].offset = \
2421*4882a593Smuzhiyun 					-cfi->stack_size + op->src.offset;
2422*4882a593Smuzhiyun 
2423*4882a593Smuzhiyun 				break;
2424*4882a593Smuzhiyun 			}
2425*4882a593Smuzhiyun 
2426*4882a593Smuzhiyun 			if (cfi->drap && op->dest.reg == CFI_SP &&
2427*4882a593Smuzhiyun 			    op->src.reg == cfi->drap_reg) {
2428*4882a593Smuzhiyun 
2429*4882a593Smuzhiyun 				 /* drap: lea disp(%drap), %rsp */
2430*4882a593Smuzhiyun 				cfa->base = CFI_SP;
2431*4882a593Smuzhiyun 				cfa->offset = cfi->stack_size = -op->src.offset;
2432*4882a593Smuzhiyun 				cfi->drap_reg = CFI_UNDEFINED;
2433*4882a593Smuzhiyun 				cfi->drap = false;
2434*4882a593Smuzhiyun 				break;
2435*4882a593Smuzhiyun 			}
2436*4882a593Smuzhiyun 
2437*4882a593Smuzhiyun 			if (op->dest.reg == cfi->cfa.base) {
2438*4882a593Smuzhiyun 				WARN_FUNC("unsupported stack register modification",
2439*4882a593Smuzhiyun 					  insn->sec, insn->offset);
2440*4882a593Smuzhiyun 				return -1;
2441*4882a593Smuzhiyun 			}
2442*4882a593Smuzhiyun 
2443*4882a593Smuzhiyun 			break;
2444*4882a593Smuzhiyun 
2445*4882a593Smuzhiyun 		case OP_SRC_AND:
2446*4882a593Smuzhiyun 			if (op->dest.reg != CFI_SP ||
2447*4882a593Smuzhiyun 			    (cfi->drap_reg != CFI_UNDEFINED && cfa->base != CFI_SP) ||
2448*4882a593Smuzhiyun 			    (cfi->drap_reg == CFI_UNDEFINED && cfa->base != CFI_BP)) {
2449*4882a593Smuzhiyun 				WARN_FUNC("unsupported stack pointer realignment",
2450*4882a593Smuzhiyun 					  insn->sec, insn->offset);
2451*4882a593Smuzhiyun 				return -1;
2452*4882a593Smuzhiyun 			}
2453*4882a593Smuzhiyun 
2454*4882a593Smuzhiyun 			if (cfi->drap_reg != CFI_UNDEFINED) {
2455*4882a593Smuzhiyun 				/* drap: and imm, %rsp */
2456*4882a593Smuzhiyun 				cfa->base = cfi->drap_reg;
2457*4882a593Smuzhiyun 				cfa->offset = cfi->stack_size = 0;
2458*4882a593Smuzhiyun 				cfi->drap = true;
2459*4882a593Smuzhiyun 			}
2460*4882a593Smuzhiyun 
2461*4882a593Smuzhiyun 			/*
2462*4882a593Smuzhiyun 			 * Older versions of GCC (4.8ish) realign the stack
2463*4882a593Smuzhiyun 			 * without DRAP, with a frame pointer.
2464*4882a593Smuzhiyun 			 */
2465*4882a593Smuzhiyun 
2466*4882a593Smuzhiyun 			break;
2467*4882a593Smuzhiyun 
2468*4882a593Smuzhiyun 		case OP_SRC_POP:
2469*4882a593Smuzhiyun 		case OP_SRC_POPF:
2470*4882a593Smuzhiyun 			if (!cfi->drap && op->dest.reg == cfa->base) {
2471*4882a593Smuzhiyun 
2472*4882a593Smuzhiyun 				/* pop %rbp */
2473*4882a593Smuzhiyun 				cfa->base = CFI_SP;
2474*4882a593Smuzhiyun 			}
2475*4882a593Smuzhiyun 
2476*4882a593Smuzhiyun 			if (cfi->drap && cfa->base == CFI_BP_INDIRECT &&
2477*4882a593Smuzhiyun 			    op->dest.reg == cfi->drap_reg &&
2478*4882a593Smuzhiyun 			    cfi->drap_offset == -cfi->stack_size) {
2479*4882a593Smuzhiyun 
2480*4882a593Smuzhiyun 				/* drap: pop %drap */
2481*4882a593Smuzhiyun 				cfa->base = cfi->drap_reg;
2482*4882a593Smuzhiyun 				cfa->offset = 0;
2483*4882a593Smuzhiyun 				cfi->drap_offset = -1;
2484*4882a593Smuzhiyun 
2485*4882a593Smuzhiyun 			} else if (regs[op->dest.reg].offset == -cfi->stack_size) {
2486*4882a593Smuzhiyun 
2487*4882a593Smuzhiyun 				/* pop %reg */
2488*4882a593Smuzhiyun 				restore_reg(cfi, op->dest.reg);
2489*4882a593Smuzhiyun 			}
2490*4882a593Smuzhiyun 
2491*4882a593Smuzhiyun 			cfi->stack_size -= 8;
2492*4882a593Smuzhiyun 			if (cfa->base == CFI_SP)
2493*4882a593Smuzhiyun 				cfa->offset -= 8;
2494*4882a593Smuzhiyun 
2495*4882a593Smuzhiyun 			break;
2496*4882a593Smuzhiyun 
2497*4882a593Smuzhiyun 		case OP_SRC_REG_INDIRECT:
2498*4882a593Smuzhiyun 			if (cfi->drap && op->src.reg == CFI_BP &&
2499*4882a593Smuzhiyun 			    op->src.offset == cfi->drap_offset) {
2500*4882a593Smuzhiyun 
2501*4882a593Smuzhiyun 				/* drap: mov disp(%rbp), %drap */
2502*4882a593Smuzhiyun 				cfa->base = cfi->drap_reg;
2503*4882a593Smuzhiyun 				cfa->offset = 0;
2504*4882a593Smuzhiyun 				cfi->drap_offset = -1;
2505*4882a593Smuzhiyun 			}
2506*4882a593Smuzhiyun 
2507*4882a593Smuzhiyun 			if (cfi->drap && op->src.reg == CFI_BP &&
2508*4882a593Smuzhiyun 			    op->src.offset == regs[op->dest.reg].offset) {
2509*4882a593Smuzhiyun 
2510*4882a593Smuzhiyun 				/* drap: mov disp(%rbp), %reg */
2511*4882a593Smuzhiyun 				restore_reg(cfi, op->dest.reg);
2512*4882a593Smuzhiyun 
2513*4882a593Smuzhiyun 			} else if (op->src.reg == cfa->base &&
2514*4882a593Smuzhiyun 			    op->src.offset == regs[op->dest.reg].offset + cfa->offset) {
2515*4882a593Smuzhiyun 
2516*4882a593Smuzhiyun 				/* mov disp(%rbp), %reg */
2517*4882a593Smuzhiyun 				/* mov disp(%rsp), %reg */
2518*4882a593Smuzhiyun 				restore_reg(cfi, op->dest.reg);
2519*4882a593Smuzhiyun 			}
2520*4882a593Smuzhiyun 
2521*4882a593Smuzhiyun 			break;
2522*4882a593Smuzhiyun 
2523*4882a593Smuzhiyun 		default:
2524*4882a593Smuzhiyun 			WARN_FUNC("unknown stack-related instruction",
2525*4882a593Smuzhiyun 				  insn->sec, insn->offset);
2526*4882a593Smuzhiyun 			return -1;
2527*4882a593Smuzhiyun 		}
2528*4882a593Smuzhiyun 
2529*4882a593Smuzhiyun 		break;
2530*4882a593Smuzhiyun 
2531*4882a593Smuzhiyun 	case OP_DEST_PUSH:
2532*4882a593Smuzhiyun 	case OP_DEST_PUSHF:
2533*4882a593Smuzhiyun 		cfi->stack_size += 8;
2534*4882a593Smuzhiyun 		if (cfa->base == CFI_SP)
2535*4882a593Smuzhiyun 			cfa->offset += 8;
2536*4882a593Smuzhiyun 
2537*4882a593Smuzhiyun 		if (op->src.type != OP_SRC_REG)
2538*4882a593Smuzhiyun 			break;
2539*4882a593Smuzhiyun 
2540*4882a593Smuzhiyun 		if (cfi->drap) {
2541*4882a593Smuzhiyun 			if (op->src.reg == cfa->base && op->src.reg == cfi->drap_reg) {
2542*4882a593Smuzhiyun 
2543*4882a593Smuzhiyun 				/* drap: push %drap */
2544*4882a593Smuzhiyun 				cfa->base = CFI_BP_INDIRECT;
2545*4882a593Smuzhiyun 				cfa->offset = -cfi->stack_size;
2546*4882a593Smuzhiyun 
2547*4882a593Smuzhiyun 				/* save drap so we know when to restore it */
2548*4882a593Smuzhiyun 				cfi->drap_offset = -cfi->stack_size;
2549*4882a593Smuzhiyun 
2550*4882a593Smuzhiyun 			} else if (op->src.reg == CFI_BP && cfa->base == cfi->drap_reg) {
2551*4882a593Smuzhiyun 
2552*4882a593Smuzhiyun 				/* drap: push %rbp */
2553*4882a593Smuzhiyun 				cfi->stack_size = 0;
2554*4882a593Smuzhiyun 
2555*4882a593Smuzhiyun 			} else {
2556*4882a593Smuzhiyun 
2557*4882a593Smuzhiyun 				/* drap: push %reg */
2558*4882a593Smuzhiyun 				save_reg(cfi, op->src.reg, CFI_BP, -cfi->stack_size);
2559*4882a593Smuzhiyun 			}
2560*4882a593Smuzhiyun 
2561*4882a593Smuzhiyun 		} else {
2562*4882a593Smuzhiyun 
2563*4882a593Smuzhiyun 			/* push %reg */
2564*4882a593Smuzhiyun 			save_reg(cfi, op->src.reg, CFI_CFA, -cfi->stack_size);
2565*4882a593Smuzhiyun 		}
2566*4882a593Smuzhiyun 
2567*4882a593Smuzhiyun 		/* detect when asm code uses rbp as a scratch register */
2568*4882a593Smuzhiyun 		if (!no_fp && insn->func && op->src.reg == CFI_BP &&
2569*4882a593Smuzhiyun 		    cfa->base != CFI_BP)
2570*4882a593Smuzhiyun 			cfi->bp_scratch = true;
2571*4882a593Smuzhiyun 		break;
2572*4882a593Smuzhiyun 
2573*4882a593Smuzhiyun 	case OP_DEST_REG_INDIRECT:
2574*4882a593Smuzhiyun 
2575*4882a593Smuzhiyun 		if (cfi->drap) {
2576*4882a593Smuzhiyun 			if (op->src.reg == cfa->base && op->src.reg == cfi->drap_reg) {
2577*4882a593Smuzhiyun 
2578*4882a593Smuzhiyun 				/* drap: mov %drap, disp(%rbp) */
2579*4882a593Smuzhiyun 				cfa->base = CFI_BP_INDIRECT;
2580*4882a593Smuzhiyun 				cfa->offset = op->dest.offset;
2581*4882a593Smuzhiyun 
2582*4882a593Smuzhiyun 				/* save drap offset so we know when to restore it */
2583*4882a593Smuzhiyun 				cfi->drap_offset = op->dest.offset;
2584*4882a593Smuzhiyun 			} else {
2585*4882a593Smuzhiyun 
2586*4882a593Smuzhiyun 				/* drap: mov reg, disp(%rbp) */
2587*4882a593Smuzhiyun 				save_reg(cfi, op->src.reg, CFI_BP, op->dest.offset);
2588*4882a593Smuzhiyun 			}
2589*4882a593Smuzhiyun 
2590*4882a593Smuzhiyun 		} else if (op->dest.reg == cfa->base) {
2591*4882a593Smuzhiyun 
2592*4882a593Smuzhiyun 			/* mov reg, disp(%rbp) */
2593*4882a593Smuzhiyun 			/* mov reg, disp(%rsp) */
2594*4882a593Smuzhiyun 			save_reg(cfi, op->src.reg, CFI_CFA,
2595*4882a593Smuzhiyun 				 op->dest.offset - cfi->cfa.offset);
2596*4882a593Smuzhiyun 		}
2597*4882a593Smuzhiyun 
2598*4882a593Smuzhiyun 		break;
2599*4882a593Smuzhiyun 
2600*4882a593Smuzhiyun 	case OP_DEST_LEAVE:
2601*4882a593Smuzhiyun 		if ((!cfi->drap && cfa->base != CFI_BP) ||
2602*4882a593Smuzhiyun 		    (cfi->drap && cfa->base != cfi->drap_reg)) {
2603*4882a593Smuzhiyun 			WARN_FUNC("leave instruction with modified stack frame",
2604*4882a593Smuzhiyun 				  insn->sec, insn->offset);
2605*4882a593Smuzhiyun 			return -1;
2606*4882a593Smuzhiyun 		}
2607*4882a593Smuzhiyun 
2608*4882a593Smuzhiyun 		/* leave (mov %rbp, %rsp; pop %rbp) */
2609*4882a593Smuzhiyun 
2610*4882a593Smuzhiyun 		cfi->stack_size = -cfi->regs[CFI_BP].offset - 8;
2611*4882a593Smuzhiyun 		restore_reg(cfi, CFI_BP);
2612*4882a593Smuzhiyun 
2613*4882a593Smuzhiyun 		if (!cfi->drap) {
2614*4882a593Smuzhiyun 			cfa->base = CFI_SP;
2615*4882a593Smuzhiyun 			cfa->offset -= 8;
2616*4882a593Smuzhiyun 		}
2617*4882a593Smuzhiyun 
2618*4882a593Smuzhiyun 		break;
2619*4882a593Smuzhiyun 
2620*4882a593Smuzhiyun 	case OP_DEST_MEM:
2621*4882a593Smuzhiyun 		if (op->src.type != OP_SRC_POP && op->src.type != OP_SRC_POPF) {
2622*4882a593Smuzhiyun 			WARN_FUNC("unknown stack-related memory operation",
2623*4882a593Smuzhiyun 				  insn->sec, insn->offset);
2624*4882a593Smuzhiyun 			return -1;
2625*4882a593Smuzhiyun 		}
2626*4882a593Smuzhiyun 
2627*4882a593Smuzhiyun 		/* pop mem */
2628*4882a593Smuzhiyun 		cfi->stack_size -= 8;
2629*4882a593Smuzhiyun 		if (cfa->base == CFI_SP)
2630*4882a593Smuzhiyun 			cfa->offset -= 8;
2631*4882a593Smuzhiyun 
2632*4882a593Smuzhiyun 		break;
2633*4882a593Smuzhiyun 
2634*4882a593Smuzhiyun 	default:
2635*4882a593Smuzhiyun 		WARN_FUNC("unknown stack-related instruction",
2636*4882a593Smuzhiyun 			  insn->sec, insn->offset);
2637*4882a593Smuzhiyun 		return -1;
2638*4882a593Smuzhiyun 	}
2639*4882a593Smuzhiyun 
2640*4882a593Smuzhiyun 	return 0;
2641*4882a593Smuzhiyun }
2642*4882a593Smuzhiyun 
2643*4882a593Smuzhiyun /*
2644*4882a593Smuzhiyun  * The stack layouts of alternatives instructions can sometimes diverge when
2645*4882a593Smuzhiyun  * they have stack modifications.  That's fine as long as the potential stack
2646*4882a593Smuzhiyun  * layouts don't conflict at any given potential instruction boundary.
2647*4882a593Smuzhiyun  *
2648*4882a593Smuzhiyun  * Flatten the CFIs of the different alternative code streams (both original
2649*4882a593Smuzhiyun  * and replacement) into a single shared CFI array which can be used to detect
2650*4882a593Smuzhiyun  * conflicts and nicely feed a linear array of ORC entries to the unwinder.
2651*4882a593Smuzhiyun  */
propagate_alt_cfi(struct objtool_file * file,struct instruction * insn)2652*4882a593Smuzhiyun static int propagate_alt_cfi(struct objtool_file *file, struct instruction *insn)
2653*4882a593Smuzhiyun {
2654*4882a593Smuzhiyun 	struct cfi_state **alt_cfi;
2655*4882a593Smuzhiyun 	int group_off;
2656*4882a593Smuzhiyun 
2657*4882a593Smuzhiyun 	if (!insn->alt_group)
2658*4882a593Smuzhiyun 		return 0;
2659*4882a593Smuzhiyun 
2660*4882a593Smuzhiyun 	if (!insn->cfi) {
2661*4882a593Smuzhiyun 		WARN("CFI missing");
2662*4882a593Smuzhiyun 		return -1;
2663*4882a593Smuzhiyun 	}
2664*4882a593Smuzhiyun 
2665*4882a593Smuzhiyun 	alt_cfi = insn->alt_group->cfi;
2666*4882a593Smuzhiyun 	group_off = insn->offset - insn->alt_group->first_insn->offset;
2667*4882a593Smuzhiyun 
2668*4882a593Smuzhiyun 	if (!alt_cfi[group_off]) {
2669*4882a593Smuzhiyun 		alt_cfi[group_off] = insn->cfi;
2670*4882a593Smuzhiyun 	} else {
2671*4882a593Smuzhiyun 		if (cficmp(alt_cfi[group_off], insn->cfi)) {
2672*4882a593Smuzhiyun 			WARN_FUNC("stack layout conflict in alternatives",
2673*4882a593Smuzhiyun 				  insn->sec, insn->offset);
2674*4882a593Smuzhiyun 			return -1;
2675*4882a593Smuzhiyun 		}
2676*4882a593Smuzhiyun 	}
2677*4882a593Smuzhiyun 
2678*4882a593Smuzhiyun 	return 0;
2679*4882a593Smuzhiyun }
2680*4882a593Smuzhiyun 
handle_insn_ops(struct instruction * insn,struct insn_state * state)2681*4882a593Smuzhiyun static int handle_insn_ops(struct instruction *insn, struct insn_state *state)
2682*4882a593Smuzhiyun {
2683*4882a593Smuzhiyun 	struct stack_op *op;
2684*4882a593Smuzhiyun 
2685*4882a593Smuzhiyun 	list_for_each_entry(op, &insn->stack_ops, list) {
2686*4882a593Smuzhiyun 
2687*4882a593Smuzhiyun 		if (update_cfi_state(insn, &state->cfi, op))
2688*4882a593Smuzhiyun 			return 1;
2689*4882a593Smuzhiyun 
2690*4882a593Smuzhiyun 		if (op->dest.type == OP_DEST_PUSHF) {
2691*4882a593Smuzhiyun 			if (!state->uaccess_stack) {
2692*4882a593Smuzhiyun 				state->uaccess_stack = 1;
2693*4882a593Smuzhiyun 			} else if (state->uaccess_stack >> 31) {
2694*4882a593Smuzhiyun 				WARN_FUNC("PUSHF stack exhausted",
2695*4882a593Smuzhiyun 					  insn->sec, insn->offset);
2696*4882a593Smuzhiyun 				return 1;
2697*4882a593Smuzhiyun 			}
2698*4882a593Smuzhiyun 			state->uaccess_stack <<= 1;
2699*4882a593Smuzhiyun 			state->uaccess_stack  |= state->uaccess;
2700*4882a593Smuzhiyun 		}
2701*4882a593Smuzhiyun 
2702*4882a593Smuzhiyun 		if (op->src.type == OP_SRC_POPF) {
2703*4882a593Smuzhiyun 			if (state->uaccess_stack) {
2704*4882a593Smuzhiyun 				state->uaccess = state->uaccess_stack & 1;
2705*4882a593Smuzhiyun 				state->uaccess_stack >>= 1;
2706*4882a593Smuzhiyun 				if (state->uaccess_stack == 1)
2707*4882a593Smuzhiyun 					state->uaccess_stack = 0;
2708*4882a593Smuzhiyun 			}
2709*4882a593Smuzhiyun 		}
2710*4882a593Smuzhiyun 	}
2711*4882a593Smuzhiyun 
2712*4882a593Smuzhiyun 	return 0;
2713*4882a593Smuzhiyun }
2714*4882a593Smuzhiyun 
insn_cfi_match(struct instruction * insn,struct cfi_state * cfi2)2715*4882a593Smuzhiyun static bool insn_cfi_match(struct instruction *insn, struct cfi_state *cfi2)
2716*4882a593Smuzhiyun {
2717*4882a593Smuzhiyun 	struct cfi_state *cfi1 = insn->cfi;
2718*4882a593Smuzhiyun 	int i;
2719*4882a593Smuzhiyun 
2720*4882a593Smuzhiyun 	if (!cfi1) {
2721*4882a593Smuzhiyun 		WARN("CFI missing");
2722*4882a593Smuzhiyun 		return false;
2723*4882a593Smuzhiyun 	}
2724*4882a593Smuzhiyun 
2725*4882a593Smuzhiyun 	if (memcmp(&cfi1->cfa, &cfi2->cfa, sizeof(cfi1->cfa))) {
2726*4882a593Smuzhiyun 
2727*4882a593Smuzhiyun 		WARN_FUNC("stack state mismatch: cfa1=%d%+d cfa2=%d%+d",
2728*4882a593Smuzhiyun 			  insn->sec, insn->offset,
2729*4882a593Smuzhiyun 			  cfi1->cfa.base, cfi1->cfa.offset,
2730*4882a593Smuzhiyun 			  cfi2->cfa.base, cfi2->cfa.offset);
2731*4882a593Smuzhiyun 
2732*4882a593Smuzhiyun 	} else if (memcmp(&cfi1->regs, &cfi2->regs, sizeof(cfi1->regs))) {
2733*4882a593Smuzhiyun 		for (i = 0; i < CFI_NUM_REGS; i++) {
2734*4882a593Smuzhiyun 			if (!memcmp(&cfi1->regs[i], &cfi2->regs[i],
2735*4882a593Smuzhiyun 				    sizeof(struct cfi_reg)))
2736*4882a593Smuzhiyun 				continue;
2737*4882a593Smuzhiyun 
2738*4882a593Smuzhiyun 			WARN_FUNC("stack state mismatch: reg1[%d]=%d%+d reg2[%d]=%d%+d",
2739*4882a593Smuzhiyun 				  insn->sec, insn->offset,
2740*4882a593Smuzhiyun 				  i, cfi1->regs[i].base, cfi1->regs[i].offset,
2741*4882a593Smuzhiyun 				  i, cfi2->regs[i].base, cfi2->regs[i].offset);
2742*4882a593Smuzhiyun 			break;
2743*4882a593Smuzhiyun 		}
2744*4882a593Smuzhiyun 
2745*4882a593Smuzhiyun 	} else if (cfi1->type != cfi2->type) {
2746*4882a593Smuzhiyun 
2747*4882a593Smuzhiyun 		WARN_FUNC("stack state mismatch: type1=%d type2=%d",
2748*4882a593Smuzhiyun 			  insn->sec, insn->offset, cfi1->type, cfi2->type);
2749*4882a593Smuzhiyun 
2750*4882a593Smuzhiyun 	} else if (cfi1->drap != cfi2->drap ||
2751*4882a593Smuzhiyun 		   (cfi1->drap && cfi1->drap_reg != cfi2->drap_reg) ||
2752*4882a593Smuzhiyun 		   (cfi1->drap && cfi1->drap_offset != cfi2->drap_offset)) {
2753*4882a593Smuzhiyun 
2754*4882a593Smuzhiyun 		WARN_FUNC("stack state mismatch: drap1=%d(%d,%d) drap2=%d(%d,%d)",
2755*4882a593Smuzhiyun 			  insn->sec, insn->offset,
2756*4882a593Smuzhiyun 			  cfi1->drap, cfi1->drap_reg, cfi1->drap_offset,
2757*4882a593Smuzhiyun 			  cfi2->drap, cfi2->drap_reg, cfi2->drap_offset);
2758*4882a593Smuzhiyun 
2759*4882a593Smuzhiyun 	} else
2760*4882a593Smuzhiyun 		return true;
2761*4882a593Smuzhiyun 
2762*4882a593Smuzhiyun 	return false;
2763*4882a593Smuzhiyun }
2764*4882a593Smuzhiyun 
func_uaccess_safe(struct symbol * func)2765*4882a593Smuzhiyun static inline bool func_uaccess_safe(struct symbol *func)
2766*4882a593Smuzhiyun {
2767*4882a593Smuzhiyun 	if (func)
2768*4882a593Smuzhiyun 		return func->uaccess_safe;
2769*4882a593Smuzhiyun 
2770*4882a593Smuzhiyun 	return false;
2771*4882a593Smuzhiyun }
2772*4882a593Smuzhiyun 
call_dest_name(struct instruction * insn)2773*4882a593Smuzhiyun static inline const char *call_dest_name(struct instruction *insn)
2774*4882a593Smuzhiyun {
2775*4882a593Smuzhiyun 	if (insn->call_dest)
2776*4882a593Smuzhiyun 		return insn->call_dest->name;
2777*4882a593Smuzhiyun 
2778*4882a593Smuzhiyun 	return "{dynamic}";
2779*4882a593Smuzhiyun }
2780*4882a593Smuzhiyun 
noinstr_call_dest(struct symbol * func)2781*4882a593Smuzhiyun static inline bool noinstr_call_dest(struct symbol *func)
2782*4882a593Smuzhiyun {
2783*4882a593Smuzhiyun 	/*
2784*4882a593Smuzhiyun 	 * We can't deal with indirect function calls at present;
2785*4882a593Smuzhiyun 	 * assume they're instrumented.
2786*4882a593Smuzhiyun 	 */
2787*4882a593Smuzhiyun 	if (!func)
2788*4882a593Smuzhiyun 		return false;
2789*4882a593Smuzhiyun 
2790*4882a593Smuzhiyun 	/*
2791*4882a593Smuzhiyun 	 * If the symbol is from a noinstr section; we good.
2792*4882a593Smuzhiyun 	 */
2793*4882a593Smuzhiyun 	if (func->sec->noinstr)
2794*4882a593Smuzhiyun 		return true;
2795*4882a593Smuzhiyun 
2796*4882a593Smuzhiyun 	/*
2797*4882a593Smuzhiyun 	 * The __ubsan_handle_*() calls are like WARN(), they only happen when
2798*4882a593Smuzhiyun 	 * something 'BAD' happened. At the risk of taking the machine down,
2799*4882a593Smuzhiyun 	 * let them proceed to get the message out.
2800*4882a593Smuzhiyun 	 */
2801*4882a593Smuzhiyun 	if (!strncmp(func->name, "__ubsan_handle_", 15))
2802*4882a593Smuzhiyun 		return true;
2803*4882a593Smuzhiyun 
2804*4882a593Smuzhiyun 	return false;
2805*4882a593Smuzhiyun }
2806*4882a593Smuzhiyun 
validate_call(struct instruction * insn,struct insn_state * state)2807*4882a593Smuzhiyun static int validate_call(struct instruction *insn, struct insn_state *state)
2808*4882a593Smuzhiyun {
2809*4882a593Smuzhiyun 	if (state->noinstr && state->instr <= 0 &&
2810*4882a593Smuzhiyun 	    !noinstr_call_dest(insn->call_dest)) {
2811*4882a593Smuzhiyun 		WARN_FUNC("call to %s() leaves .noinstr.text section",
2812*4882a593Smuzhiyun 				insn->sec, insn->offset, call_dest_name(insn));
2813*4882a593Smuzhiyun 		return 1;
2814*4882a593Smuzhiyun 	}
2815*4882a593Smuzhiyun 
2816*4882a593Smuzhiyun 	if (state->uaccess && !func_uaccess_safe(insn->call_dest)) {
2817*4882a593Smuzhiyun 		WARN_FUNC("call to %s() with UACCESS enabled",
2818*4882a593Smuzhiyun 				insn->sec, insn->offset, call_dest_name(insn));
2819*4882a593Smuzhiyun 		return 1;
2820*4882a593Smuzhiyun 	}
2821*4882a593Smuzhiyun 
2822*4882a593Smuzhiyun 	if (state->df) {
2823*4882a593Smuzhiyun 		WARN_FUNC("call to %s() with DF set",
2824*4882a593Smuzhiyun 				insn->sec, insn->offset, call_dest_name(insn));
2825*4882a593Smuzhiyun 		return 1;
2826*4882a593Smuzhiyun 	}
2827*4882a593Smuzhiyun 
2828*4882a593Smuzhiyun 	return 0;
2829*4882a593Smuzhiyun }
2830*4882a593Smuzhiyun 
validate_sibling_call(struct instruction * insn,struct insn_state * state)2831*4882a593Smuzhiyun static int validate_sibling_call(struct instruction *insn, struct insn_state *state)
2832*4882a593Smuzhiyun {
2833*4882a593Smuzhiyun 	if (has_modified_stack_frame(insn, state)) {
2834*4882a593Smuzhiyun 		WARN_FUNC("sibling call from callable instruction with modified stack frame",
2835*4882a593Smuzhiyun 				insn->sec, insn->offset);
2836*4882a593Smuzhiyun 		return 1;
2837*4882a593Smuzhiyun 	}
2838*4882a593Smuzhiyun 
2839*4882a593Smuzhiyun 	return validate_call(insn, state);
2840*4882a593Smuzhiyun }
2841*4882a593Smuzhiyun 
validate_return(struct symbol * func,struct instruction * insn,struct insn_state * state)2842*4882a593Smuzhiyun static int validate_return(struct symbol *func, struct instruction *insn, struct insn_state *state)
2843*4882a593Smuzhiyun {
2844*4882a593Smuzhiyun 	if (state->noinstr && state->instr > 0) {
2845*4882a593Smuzhiyun 		WARN_FUNC("return with instrumentation enabled",
2846*4882a593Smuzhiyun 			  insn->sec, insn->offset);
2847*4882a593Smuzhiyun 		return 1;
2848*4882a593Smuzhiyun 	}
2849*4882a593Smuzhiyun 
2850*4882a593Smuzhiyun 	if (state->uaccess && !func_uaccess_safe(func)) {
2851*4882a593Smuzhiyun 		WARN_FUNC("return with UACCESS enabled",
2852*4882a593Smuzhiyun 			  insn->sec, insn->offset);
2853*4882a593Smuzhiyun 		return 1;
2854*4882a593Smuzhiyun 	}
2855*4882a593Smuzhiyun 
2856*4882a593Smuzhiyun 	if (!state->uaccess && func_uaccess_safe(func)) {
2857*4882a593Smuzhiyun 		WARN_FUNC("return with UACCESS disabled from a UACCESS-safe function",
2858*4882a593Smuzhiyun 			  insn->sec, insn->offset);
2859*4882a593Smuzhiyun 		return 1;
2860*4882a593Smuzhiyun 	}
2861*4882a593Smuzhiyun 
2862*4882a593Smuzhiyun 	if (state->df) {
2863*4882a593Smuzhiyun 		WARN_FUNC("return with DF set",
2864*4882a593Smuzhiyun 			  insn->sec, insn->offset);
2865*4882a593Smuzhiyun 		return 1;
2866*4882a593Smuzhiyun 	}
2867*4882a593Smuzhiyun 
2868*4882a593Smuzhiyun 	if (func && has_modified_stack_frame(insn, state)) {
2869*4882a593Smuzhiyun 		WARN_FUNC("return with modified stack frame",
2870*4882a593Smuzhiyun 			  insn->sec, insn->offset);
2871*4882a593Smuzhiyun 		return 1;
2872*4882a593Smuzhiyun 	}
2873*4882a593Smuzhiyun 
2874*4882a593Smuzhiyun 	if (state->cfi.bp_scratch) {
2875*4882a593Smuzhiyun 		WARN_FUNC("BP used as a scratch register",
2876*4882a593Smuzhiyun 			  insn->sec, insn->offset);
2877*4882a593Smuzhiyun 		return 1;
2878*4882a593Smuzhiyun 	}
2879*4882a593Smuzhiyun 
2880*4882a593Smuzhiyun 	return 0;
2881*4882a593Smuzhiyun }
2882*4882a593Smuzhiyun 
next_insn_to_validate(struct objtool_file * file,struct instruction * insn)2883*4882a593Smuzhiyun static struct instruction *next_insn_to_validate(struct objtool_file *file,
2884*4882a593Smuzhiyun 						 struct instruction *insn)
2885*4882a593Smuzhiyun {
2886*4882a593Smuzhiyun 	struct alt_group *alt_group = insn->alt_group;
2887*4882a593Smuzhiyun 
2888*4882a593Smuzhiyun 	/*
2889*4882a593Smuzhiyun 	 * Simulate the fact that alternatives are patched in-place.  When the
2890*4882a593Smuzhiyun 	 * end of a replacement alt_group is reached, redirect objtool flow to
2891*4882a593Smuzhiyun 	 * the end of the original alt_group.
2892*4882a593Smuzhiyun 	 */
2893*4882a593Smuzhiyun 	if (alt_group && insn == alt_group->last_insn && alt_group->orig_group)
2894*4882a593Smuzhiyun 		return next_insn_same_sec(file, alt_group->orig_group->last_insn);
2895*4882a593Smuzhiyun 
2896*4882a593Smuzhiyun 	return next_insn_same_sec(file, insn);
2897*4882a593Smuzhiyun }
2898*4882a593Smuzhiyun 
2899*4882a593Smuzhiyun /*
2900*4882a593Smuzhiyun  * Follow the branch starting at the given instruction, and recursively follow
2901*4882a593Smuzhiyun  * any other branches (jumps).  Meanwhile, track the frame pointer state at
2902*4882a593Smuzhiyun  * each instruction and validate all the rules described in
2903*4882a593Smuzhiyun  * tools/objtool/Documentation/stack-validation.txt.
2904*4882a593Smuzhiyun  */
validate_branch(struct objtool_file * file,struct symbol * func,struct instruction * insn,struct insn_state state)2905*4882a593Smuzhiyun static int validate_branch(struct objtool_file *file, struct symbol *func,
2906*4882a593Smuzhiyun 			   struct instruction *insn, struct insn_state state)
2907*4882a593Smuzhiyun {
2908*4882a593Smuzhiyun 	struct alternative *alt;
2909*4882a593Smuzhiyun 	struct instruction *next_insn, *prev_insn = NULL;
2910*4882a593Smuzhiyun 	struct section *sec;
2911*4882a593Smuzhiyun 	u8 visited;
2912*4882a593Smuzhiyun 	int ret;
2913*4882a593Smuzhiyun 
2914*4882a593Smuzhiyun 	sec = insn->sec;
2915*4882a593Smuzhiyun 
2916*4882a593Smuzhiyun 	while (1) {
2917*4882a593Smuzhiyun 		next_insn = next_insn_to_validate(file, insn);
2918*4882a593Smuzhiyun 
2919*4882a593Smuzhiyun 		if (file->c_file && func && insn->func && func != insn->func->pfunc) {
2920*4882a593Smuzhiyun 			WARN("%s() falls through to next function %s()",
2921*4882a593Smuzhiyun 			     func->name, insn->func->name);
2922*4882a593Smuzhiyun 			return 1;
2923*4882a593Smuzhiyun 		}
2924*4882a593Smuzhiyun 
2925*4882a593Smuzhiyun 		if (func && insn->ignore) {
2926*4882a593Smuzhiyun 			WARN_FUNC("BUG: why am I validating an ignored function?",
2927*4882a593Smuzhiyun 				  sec, insn->offset);
2928*4882a593Smuzhiyun 			return 1;
2929*4882a593Smuzhiyun 		}
2930*4882a593Smuzhiyun 
2931*4882a593Smuzhiyun 		visited = VISITED_BRANCH << state.uaccess;
2932*4882a593Smuzhiyun 		if (insn->visited & VISITED_BRANCH_MASK) {
2933*4882a593Smuzhiyun 			if (!insn->hint && !insn_cfi_match(insn, &state.cfi))
2934*4882a593Smuzhiyun 				return 1;
2935*4882a593Smuzhiyun 
2936*4882a593Smuzhiyun 			if (insn->visited & visited)
2937*4882a593Smuzhiyun 				return 0;
2938*4882a593Smuzhiyun 		} else {
2939*4882a593Smuzhiyun 			nr_insns_visited++;
2940*4882a593Smuzhiyun 		}
2941*4882a593Smuzhiyun 
2942*4882a593Smuzhiyun 		if (state.noinstr)
2943*4882a593Smuzhiyun 			state.instr += insn->instr;
2944*4882a593Smuzhiyun 
2945*4882a593Smuzhiyun 		if (insn->hint) {
2946*4882a593Smuzhiyun 			if (insn->restore) {
2947*4882a593Smuzhiyun 				struct instruction *save_insn, *i;
2948*4882a593Smuzhiyun 
2949*4882a593Smuzhiyun 				i = insn;
2950*4882a593Smuzhiyun 				save_insn = NULL;
2951*4882a593Smuzhiyun 
2952*4882a593Smuzhiyun 				sym_for_each_insn_continue_reverse(file, func, i) {
2953*4882a593Smuzhiyun 					if (i->save) {
2954*4882a593Smuzhiyun 						save_insn = i;
2955*4882a593Smuzhiyun 						break;
2956*4882a593Smuzhiyun 					}
2957*4882a593Smuzhiyun 				}
2958*4882a593Smuzhiyun 
2959*4882a593Smuzhiyun 				if (!save_insn) {
2960*4882a593Smuzhiyun 					WARN_FUNC("no corresponding CFI save for CFI restore",
2961*4882a593Smuzhiyun 						  sec, insn->offset);
2962*4882a593Smuzhiyun 					return 1;
2963*4882a593Smuzhiyun 				}
2964*4882a593Smuzhiyun 
2965*4882a593Smuzhiyun 				if (!save_insn->visited) {
2966*4882a593Smuzhiyun 					WARN_FUNC("objtool isn't smart enough to handle this CFI save/restore combo",
2967*4882a593Smuzhiyun 						  sec, insn->offset);
2968*4882a593Smuzhiyun 					return 1;
2969*4882a593Smuzhiyun 				}
2970*4882a593Smuzhiyun 
2971*4882a593Smuzhiyun 				insn->cfi = save_insn->cfi;
2972*4882a593Smuzhiyun 				nr_cfi_reused++;
2973*4882a593Smuzhiyun 			}
2974*4882a593Smuzhiyun 
2975*4882a593Smuzhiyun 			state.cfi = *insn->cfi;
2976*4882a593Smuzhiyun 		} else {
2977*4882a593Smuzhiyun 			/* XXX track if we actually changed state.cfi */
2978*4882a593Smuzhiyun 
2979*4882a593Smuzhiyun 			if (prev_insn && !cficmp(prev_insn->cfi, &state.cfi)) {
2980*4882a593Smuzhiyun 				insn->cfi = prev_insn->cfi;
2981*4882a593Smuzhiyun 				nr_cfi_reused++;
2982*4882a593Smuzhiyun 			} else {
2983*4882a593Smuzhiyun 				insn->cfi = cfi_hash_find_or_add(&state.cfi);
2984*4882a593Smuzhiyun 			}
2985*4882a593Smuzhiyun 		}
2986*4882a593Smuzhiyun 
2987*4882a593Smuzhiyun 		insn->visited |= visited;
2988*4882a593Smuzhiyun 
2989*4882a593Smuzhiyun 		if (propagate_alt_cfi(file, insn))
2990*4882a593Smuzhiyun 			return 1;
2991*4882a593Smuzhiyun 
2992*4882a593Smuzhiyun 		if (!insn->ignore_alts && !list_empty(&insn->alts)) {
2993*4882a593Smuzhiyun 			bool skip_orig = false;
2994*4882a593Smuzhiyun 
2995*4882a593Smuzhiyun 			list_for_each_entry(alt, &insn->alts, list) {
2996*4882a593Smuzhiyun 				if (alt->skip_orig)
2997*4882a593Smuzhiyun 					skip_orig = true;
2998*4882a593Smuzhiyun 
2999*4882a593Smuzhiyun 				ret = validate_branch(file, func, alt->insn, state);
3000*4882a593Smuzhiyun 				if (ret) {
3001*4882a593Smuzhiyun 					if (backtrace)
3002*4882a593Smuzhiyun 						BT_FUNC("(alt)", insn);
3003*4882a593Smuzhiyun 					return ret;
3004*4882a593Smuzhiyun 				}
3005*4882a593Smuzhiyun 			}
3006*4882a593Smuzhiyun 
3007*4882a593Smuzhiyun 			if (skip_orig)
3008*4882a593Smuzhiyun 				return 0;
3009*4882a593Smuzhiyun 		}
3010*4882a593Smuzhiyun 
3011*4882a593Smuzhiyun 		if (handle_insn_ops(insn, &state))
3012*4882a593Smuzhiyun 			return 1;
3013*4882a593Smuzhiyun 
3014*4882a593Smuzhiyun 		switch (insn->type) {
3015*4882a593Smuzhiyun 
3016*4882a593Smuzhiyun 		case INSN_RETURN:
3017*4882a593Smuzhiyun 			if (sls && !insn->retpoline_safe &&
3018*4882a593Smuzhiyun 			    next_insn && next_insn->type != INSN_TRAP) {
3019*4882a593Smuzhiyun 				WARN_FUNC("missing int3 after ret",
3020*4882a593Smuzhiyun 					  insn->sec, insn->offset);
3021*4882a593Smuzhiyun 			}
3022*4882a593Smuzhiyun 			return validate_return(func, insn, &state);
3023*4882a593Smuzhiyun 
3024*4882a593Smuzhiyun 		case INSN_CALL:
3025*4882a593Smuzhiyun 		case INSN_CALL_DYNAMIC:
3026*4882a593Smuzhiyun 			ret = validate_call(insn, &state);
3027*4882a593Smuzhiyun 			if (ret)
3028*4882a593Smuzhiyun 				return ret;
3029*4882a593Smuzhiyun 
3030*4882a593Smuzhiyun 			if (!no_fp && func && !is_fentry_call(insn) &&
3031*4882a593Smuzhiyun 			    !has_valid_stack_frame(&state)) {
3032*4882a593Smuzhiyun 				WARN_FUNC("call without frame pointer save/setup",
3033*4882a593Smuzhiyun 					  sec, insn->offset);
3034*4882a593Smuzhiyun 				return 1;
3035*4882a593Smuzhiyun 			}
3036*4882a593Smuzhiyun 
3037*4882a593Smuzhiyun 			if (dead_end_function(file, insn->call_dest))
3038*4882a593Smuzhiyun 				return 0;
3039*4882a593Smuzhiyun 
3040*4882a593Smuzhiyun 			break;
3041*4882a593Smuzhiyun 
3042*4882a593Smuzhiyun 		case INSN_JUMP_CONDITIONAL:
3043*4882a593Smuzhiyun 		case INSN_JUMP_UNCONDITIONAL:
3044*4882a593Smuzhiyun 			if (is_sibling_call(insn)) {
3045*4882a593Smuzhiyun 				ret = validate_sibling_call(insn, &state);
3046*4882a593Smuzhiyun 				if (ret)
3047*4882a593Smuzhiyun 					return ret;
3048*4882a593Smuzhiyun 
3049*4882a593Smuzhiyun 			} else if (insn->jump_dest) {
3050*4882a593Smuzhiyun 				ret = validate_branch(file, func,
3051*4882a593Smuzhiyun 						      insn->jump_dest, state);
3052*4882a593Smuzhiyun 				if (ret) {
3053*4882a593Smuzhiyun 					if (backtrace)
3054*4882a593Smuzhiyun 						BT_FUNC("(branch)", insn);
3055*4882a593Smuzhiyun 					return ret;
3056*4882a593Smuzhiyun 				}
3057*4882a593Smuzhiyun 			}
3058*4882a593Smuzhiyun 
3059*4882a593Smuzhiyun 			if (insn->type == INSN_JUMP_UNCONDITIONAL)
3060*4882a593Smuzhiyun 				return 0;
3061*4882a593Smuzhiyun 
3062*4882a593Smuzhiyun 			break;
3063*4882a593Smuzhiyun 
3064*4882a593Smuzhiyun 		case INSN_JUMP_DYNAMIC:
3065*4882a593Smuzhiyun 			if (sls && !insn->retpoline_safe &&
3066*4882a593Smuzhiyun 			    next_insn && next_insn->type != INSN_TRAP) {
3067*4882a593Smuzhiyun 				WARN_FUNC("missing int3 after indirect jump",
3068*4882a593Smuzhiyun 					  insn->sec, insn->offset);
3069*4882a593Smuzhiyun 			}
3070*4882a593Smuzhiyun 
3071*4882a593Smuzhiyun 			/* fallthrough */
3072*4882a593Smuzhiyun 		case INSN_JUMP_DYNAMIC_CONDITIONAL:
3073*4882a593Smuzhiyun 			if (is_sibling_call(insn)) {
3074*4882a593Smuzhiyun 				ret = validate_sibling_call(insn, &state);
3075*4882a593Smuzhiyun 				if (ret)
3076*4882a593Smuzhiyun 					return ret;
3077*4882a593Smuzhiyun 			}
3078*4882a593Smuzhiyun 
3079*4882a593Smuzhiyun 			if (insn->type == INSN_JUMP_DYNAMIC)
3080*4882a593Smuzhiyun 				return 0;
3081*4882a593Smuzhiyun 
3082*4882a593Smuzhiyun 			break;
3083*4882a593Smuzhiyun 
3084*4882a593Smuzhiyun 		case INSN_CONTEXT_SWITCH:
3085*4882a593Smuzhiyun 			if (func && (!next_insn || !next_insn->hint)) {
3086*4882a593Smuzhiyun 				WARN_FUNC("unsupported instruction in callable function",
3087*4882a593Smuzhiyun 					  sec, insn->offset);
3088*4882a593Smuzhiyun 				return 1;
3089*4882a593Smuzhiyun 			}
3090*4882a593Smuzhiyun 			return 0;
3091*4882a593Smuzhiyun 
3092*4882a593Smuzhiyun 		case INSN_STAC:
3093*4882a593Smuzhiyun 			if (state.uaccess) {
3094*4882a593Smuzhiyun 				WARN_FUNC("recursive UACCESS enable", sec, insn->offset);
3095*4882a593Smuzhiyun 				return 1;
3096*4882a593Smuzhiyun 			}
3097*4882a593Smuzhiyun 
3098*4882a593Smuzhiyun 			state.uaccess = true;
3099*4882a593Smuzhiyun 			break;
3100*4882a593Smuzhiyun 
3101*4882a593Smuzhiyun 		case INSN_CLAC:
3102*4882a593Smuzhiyun 			if (!state.uaccess && func) {
3103*4882a593Smuzhiyun 				WARN_FUNC("redundant UACCESS disable", sec, insn->offset);
3104*4882a593Smuzhiyun 				return 1;
3105*4882a593Smuzhiyun 			}
3106*4882a593Smuzhiyun 
3107*4882a593Smuzhiyun 			if (func_uaccess_safe(func) && !state.uaccess_stack) {
3108*4882a593Smuzhiyun 				WARN_FUNC("UACCESS-safe disables UACCESS", sec, insn->offset);
3109*4882a593Smuzhiyun 				return 1;
3110*4882a593Smuzhiyun 			}
3111*4882a593Smuzhiyun 
3112*4882a593Smuzhiyun 			state.uaccess = false;
3113*4882a593Smuzhiyun 			break;
3114*4882a593Smuzhiyun 
3115*4882a593Smuzhiyun 		case INSN_STD:
3116*4882a593Smuzhiyun 			if (state.df) {
3117*4882a593Smuzhiyun 				WARN_FUNC("recursive STD", sec, insn->offset);
3118*4882a593Smuzhiyun 				return 1;
3119*4882a593Smuzhiyun 			}
3120*4882a593Smuzhiyun 
3121*4882a593Smuzhiyun 			state.df = true;
3122*4882a593Smuzhiyun 			break;
3123*4882a593Smuzhiyun 
3124*4882a593Smuzhiyun 		case INSN_CLD:
3125*4882a593Smuzhiyun 			if (!state.df && func) {
3126*4882a593Smuzhiyun 				WARN_FUNC("redundant CLD", sec, insn->offset);
3127*4882a593Smuzhiyun 				return 1;
3128*4882a593Smuzhiyun 			}
3129*4882a593Smuzhiyun 
3130*4882a593Smuzhiyun 			state.df = false;
3131*4882a593Smuzhiyun 			break;
3132*4882a593Smuzhiyun 
3133*4882a593Smuzhiyun 		default:
3134*4882a593Smuzhiyun 			break;
3135*4882a593Smuzhiyun 		}
3136*4882a593Smuzhiyun 
3137*4882a593Smuzhiyun 		if (insn->dead_end)
3138*4882a593Smuzhiyun 			return 0;
3139*4882a593Smuzhiyun 
3140*4882a593Smuzhiyun 		if (!next_insn) {
3141*4882a593Smuzhiyun 			if (state.cfi.cfa.base == CFI_UNDEFINED)
3142*4882a593Smuzhiyun 				return 0;
3143*4882a593Smuzhiyun 			WARN("%s: unexpected end of section", sec->name);
3144*4882a593Smuzhiyun 			return 1;
3145*4882a593Smuzhiyun 		}
3146*4882a593Smuzhiyun 
3147*4882a593Smuzhiyun 		prev_insn = insn;
3148*4882a593Smuzhiyun 		insn = next_insn;
3149*4882a593Smuzhiyun 	}
3150*4882a593Smuzhiyun 
3151*4882a593Smuzhiyun 	return 0;
3152*4882a593Smuzhiyun }
3153*4882a593Smuzhiyun 
validate_unwind_hints(struct objtool_file * file,struct section * sec)3154*4882a593Smuzhiyun static int validate_unwind_hints(struct objtool_file *file, struct section *sec)
3155*4882a593Smuzhiyun {
3156*4882a593Smuzhiyun 	struct instruction *insn;
3157*4882a593Smuzhiyun 	struct insn_state state;
3158*4882a593Smuzhiyun 	int ret, warnings = 0;
3159*4882a593Smuzhiyun 
3160*4882a593Smuzhiyun 	if (!file->hints)
3161*4882a593Smuzhiyun 		return 0;
3162*4882a593Smuzhiyun 
3163*4882a593Smuzhiyun 	init_insn_state(&state, sec);
3164*4882a593Smuzhiyun 
3165*4882a593Smuzhiyun 	if (sec) {
3166*4882a593Smuzhiyun 		insn = find_insn(file, sec, 0);
3167*4882a593Smuzhiyun 		if (!insn)
3168*4882a593Smuzhiyun 			return 0;
3169*4882a593Smuzhiyun 	} else {
3170*4882a593Smuzhiyun 		insn = list_first_entry(&file->insn_list, typeof(*insn), list);
3171*4882a593Smuzhiyun 	}
3172*4882a593Smuzhiyun 
3173*4882a593Smuzhiyun 	while (&insn->list != &file->insn_list && (!sec || insn->sec == sec)) {
3174*4882a593Smuzhiyun 		if (insn->hint && !insn->visited) {
3175*4882a593Smuzhiyun 			ret = validate_branch(file, insn->func, insn, state);
3176*4882a593Smuzhiyun 			if (ret && backtrace)
3177*4882a593Smuzhiyun 				BT_FUNC("<=== (hint)", insn);
3178*4882a593Smuzhiyun 			warnings += ret;
3179*4882a593Smuzhiyun 		}
3180*4882a593Smuzhiyun 
3181*4882a593Smuzhiyun 		insn = list_next_entry(insn, list);
3182*4882a593Smuzhiyun 	}
3183*4882a593Smuzhiyun 
3184*4882a593Smuzhiyun 	return warnings;
3185*4882a593Smuzhiyun }
3186*4882a593Smuzhiyun 
3187*4882a593Smuzhiyun /*
3188*4882a593Smuzhiyun  * Validate rethunk entry constraint: must untrain RET before the first RET.
3189*4882a593Smuzhiyun  *
3190*4882a593Smuzhiyun  * Follow every branch (intra-function) and ensure ANNOTATE_UNRET_END comes
3191*4882a593Smuzhiyun  * before an actual RET instruction.
3192*4882a593Smuzhiyun  */
validate_entry(struct objtool_file * file,struct instruction * insn)3193*4882a593Smuzhiyun static int validate_entry(struct objtool_file *file, struct instruction *insn)
3194*4882a593Smuzhiyun {
3195*4882a593Smuzhiyun 	struct instruction *next, *dest;
3196*4882a593Smuzhiyun 	int ret, warnings = 0;
3197*4882a593Smuzhiyun 
3198*4882a593Smuzhiyun 	for (;;) {
3199*4882a593Smuzhiyun 		next = next_insn_to_validate(file, insn);
3200*4882a593Smuzhiyun 
3201*4882a593Smuzhiyun 		if (insn->visited & VISITED_ENTRY)
3202*4882a593Smuzhiyun 			return 0;
3203*4882a593Smuzhiyun 
3204*4882a593Smuzhiyun 		insn->visited |= VISITED_ENTRY;
3205*4882a593Smuzhiyun 
3206*4882a593Smuzhiyun 		if (!insn->ignore_alts && !list_empty(&insn->alts)) {
3207*4882a593Smuzhiyun 			struct alternative *alt;
3208*4882a593Smuzhiyun 			bool skip_orig = false;
3209*4882a593Smuzhiyun 
3210*4882a593Smuzhiyun 			list_for_each_entry(alt, &insn->alts, list) {
3211*4882a593Smuzhiyun 				if (alt->skip_orig)
3212*4882a593Smuzhiyun 					skip_orig = true;
3213*4882a593Smuzhiyun 
3214*4882a593Smuzhiyun 				ret = validate_entry(file, alt->insn);
3215*4882a593Smuzhiyun 				if (ret) {
3216*4882a593Smuzhiyun 				        if (backtrace)
3217*4882a593Smuzhiyun 						BT_FUNC("(alt)", insn);
3218*4882a593Smuzhiyun 					return ret;
3219*4882a593Smuzhiyun 				}
3220*4882a593Smuzhiyun 			}
3221*4882a593Smuzhiyun 
3222*4882a593Smuzhiyun 			if (skip_orig)
3223*4882a593Smuzhiyun 				return 0;
3224*4882a593Smuzhiyun 		}
3225*4882a593Smuzhiyun 
3226*4882a593Smuzhiyun 		switch (insn->type) {
3227*4882a593Smuzhiyun 
3228*4882a593Smuzhiyun 		case INSN_CALL_DYNAMIC:
3229*4882a593Smuzhiyun 		case INSN_JUMP_DYNAMIC:
3230*4882a593Smuzhiyun 		case INSN_JUMP_DYNAMIC_CONDITIONAL:
3231*4882a593Smuzhiyun 			WARN_FUNC("early indirect call", insn->sec, insn->offset);
3232*4882a593Smuzhiyun 			return 1;
3233*4882a593Smuzhiyun 
3234*4882a593Smuzhiyun 		case INSN_JUMP_UNCONDITIONAL:
3235*4882a593Smuzhiyun 		case INSN_JUMP_CONDITIONAL:
3236*4882a593Smuzhiyun 			if (!is_sibling_call(insn)) {
3237*4882a593Smuzhiyun 				if (!insn->jump_dest) {
3238*4882a593Smuzhiyun 					WARN_FUNC("unresolved jump target after linking?!?",
3239*4882a593Smuzhiyun 						  insn->sec, insn->offset);
3240*4882a593Smuzhiyun 					return -1;
3241*4882a593Smuzhiyun 				}
3242*4882a593Smuzhiyun 				ret = validate_entry(file, insn->jump_dest);
3243*4882a593Smuzhiyun 				if (ret) {
3244*4882a593Smuzhiyun 					if (backtrace) {
3245*4882a593Smuzhiyun 						BT_FUNC("(branch%s)", insn,
3246*4882a593Smuzhiyun 							insn->type == INSN_JUMP_CONDITIONAL ? "-cond" : "");
3247*4882a593Smuzhiyun 					}
3248*4882a593Smuzhiyun 					return ret;
3249*4882a593Smuzhiyun 				}
3250*4882a593Smuzhiyun 
3251*4882a593Smuzhiyun 				if (insn->type == INSN_JUMP_UNCONDITIONAL)
3252*4882a593Smuzhiyun 					return 0;
3253*4882a593Smuzhiyun 
3254*4882a593Smuzhiyun 				break;
3255*4882a593Smuzhiyun 			}
3256*4882a593Smuzhiyun 
3257*4882a593Smuzhiyun 			/* fallthrough */
3258*4882a593Smuzhiyun 		case INSN_CALL:
3259*4882a593Smuzhiyun 			dest = find_insn(file, insn->call_dest->sec,
3260*4882a593Smuzhiyun 					 insn->call_dest->offset);
3261*4882a593Smuzhiyun 			if (!dest) {
3262*4882a593Smuzhiyun 				WARN("Unresolved function after linking!?: %s",
3263*4882a593Smuzhiyun 				     insn->call_dest->name);
3264*4882a593Smuzhiyun 				return -1;
3265*4882a593Smuzhiyun 			}
3266*4882a593Smuzhiyun 
3267*4882a593Smuzhiyun 			ret = validate_entry(file, dest);
3268*4882a593Smuzhiyun 			if (ret) {
3269*4882a593Smuzhiyun 				if (backtrace)
3270*4882a593Smuzhiyun 					BT_FUNC("(call)", insn);
3271*4882a593Smuzhiyun 				return ret;
3272*4882a593Smuzhiyun 			}
3273*4882a593Smuzhiyun 			/*
3274*4882a593Smuzhiyun 			 * If a call returns without error, it must have seen UNTRAIN_RET.
3275*4882a593Smuzhiyun 			 * Therefore any non-error return is a success.
3276*4882a593Smuzhiyun 			 */
3277*4882a593Smuzhiyun 			return 0;
3278*4882a593Smuzhiyun 
3279*4882a593Smuzhiyun 		case INSN_RETURN:
3280*4882a593Smuzhiyun 			WARN_FUNC("RET before UNTRAIN", insn->sec, insn->offset);
3281*4882a593Smuzhiyun 			return 1;
3282*4882a593Smuzhiyun 
3283*4882a593Smuzhiyun 		case INSN_NOP:
3284*4882a593Smuzhiyun 			if (insn->retpoline_safe)
3285*4882a593Smuzhiyun 				return 0;
3286*4882a593Smuzhiyun 			break;
3287*4882a593Smuzhiyun 
3288*4882a593Smuzhiyun 		default:
3289*4882a593Smuzhiyun 			break;
3290*4882a593Smuzhiyun 		}
3291*4882a593Smuzhiyun 
3292*4882a593Smuzhiyun 		if (!next) {
3293*4882a593Smuzhiyun 			WARN_FUNC("teh end!", insn->sec, insn->offset);
3294*4882a593Smuzhiyun 			return -1;
3295*4882a593Smuzhiyun 		}
3296*4882a593Smuzhiyun 		insn = next;
3297*4882a593Smuzhiyun 	}
3298*4882a593Smuzhiyun 
3299*4882a593Smuzhiyun 	return warnings;
3300*4882a593Smuzhiyun }
3301*4882a593Smuzhiyun 
3302*4882a593Smuzhiyun /*
3303*4882a593Smuzhiyun  * Validate that all branches starting at 'insn->entry' encounter UNRET_END
3304*4882a593Smuzhiyun  * before RET.
3305*4882a593Smuzhiyun  */
validate_unret(struct objtool_file * file)3306*4882a593Smuzhiyun static int validate_unret(struct objtool_file *file)
3307*4882a593Smuzhiyun {
3308*4882a593Smuzhiyun 	struct instruction *insn;
3309*4882a593Smuzhiyun 	int ret, warnings = 0;
3310*4882a593Smuzhiyun 
3311*4882a593Smuzhiyun 	for_each_insn(file, insn) {
3312*4882a593Smuzhiyun 		if (!insn->entry)
3313*4882a593Smuzhiyun 			continue;
3314*4882a593Smuzhiyun 
3315*4882a593Smuzhiyun 		ret = validate_entry(file, insn);
3316*4882a593Smuzhiyun 		if (ret < 0) {
3317*4882a593Smuzhiyun 			WARN_FUNC("Failed UNRET validation", insn->sec, insn->offset);
3318*4882a593Smuzhiyun 			return ret;
3319*4882a593Smuzhiyun 		}
3320*4882a593Smuzhiyun 		warnings += ret;
3321*4882a593Smuzhiyun 	}
3322*4882a593Smuzhiyun 
3323*4882a593Smuzhiyun 	return warnings;
3324*4882a593Smuzhiyun }
3325*4882a593Smuzhiyun 
validate_retpoline(struct objtool_file * file)3326*4882a593Smuzhiyun static int validate_retpoline(struct objtool_file *file)
3327*4882a593Smuzhiyun {
3328*4882a593Smuzhiyun 	struct instruction *insn;
3329*4882a593Smuzhiyun 	int warnings = 0;
3330*4882a593Smuzhiyun 
3331*4882a593Smuzhiyun 	for_each_insn(file, insn) {
3332*4882a593Smuzhiyun 		if (insn->type != INSN_JUMP_DYNAMIC &&
3333*4882a593Smuzhiyun 		    insn->type != INSN_CALL_DYNAMIC &&
3334*4882a593Smuzhiyun 		    insn->type != INSN_RETURN)
3335*4882a593Smuzhiyun 			continue;
3336*4882a593Smuzhiyun 
3337*4882a593Smuzhiyun 		if (insn->retpoline_safe)
3338*4882a593Smuzhiyun 			continue;
3339*4882a593Smuzhiyun 
3340*4882a593Smuzhiyun 		/*
3341*4882a593Smuzhiyun 		 * .init.text code is ran before userspace and thus doesn't
3342*4882a593Smuzhiyun 		 * strictly need retpolines, except for modules which are
3343*4882a593Smuzhiyun 		 * loaded late, they very much do need retpoline in their
3344*4882a593Smuzhiyun 		 * .init.text
3345*4882a593Smuzhiyun 		 */
3346*4882a593Smuzhiyun 		if (!strcmp(insn->sec->name, ".init.text") && !module)
3347*4882a593Smuzhiyun 			continue;
3348*4882a593Smuzhiyun 
3349*4882a593Smuzhiyun 		if (insn->type == INSN_RETURN) {
3350*4882a593Smuzhiyun 			if (rethunk) {
3351*4882a593Smuzhiyun 				WARN_FUNC("'naked' return found in RETHUNK build",
3352*4882a593Smuzhiyun 					  insn->sec, insn->offset);
3353*4882a593Smuzhiyun 			} else
3354*4882a593Smuzhiyun 				continue;
3355*4882a593Smuzhiyun 		} else {
3356*4882a593Smuzhiyun 			WARN_FUNC("indirect %s found in RETPOLINE build",
3357*4882a593Smuzhiyun 				  insn->sec, insn->offset,
3358*4882a593Smuzhiyun 				  insn->type == INSN_JUMP_DYNAMIC ? "jump" : "call");
3359*4882a593Smuzhiyun 		}
3360*4882a593Smuzhiyun 
3361*4882a593Smuzhiyun 		warnings++;
3362*4882a593Smuzhiyun 	}
3363*4882a593Smuzhiyun 
3364*4882a593Smuzhiyun 	return warnings;
3365*4882a593Smuzhiyun }
3366*4882a593Smuzhiyun 
is_kasan_insn(struct instruction * insn)3367*4882a593Smuzhiyun static bool is_kasan_insn(struct instruction *insn)
3368*4882a593Smuzhiyun {
3369*4882a593Smuzhiyun 	return (insn->type == INSN_CALL &&
3370*4882a593Smuzhiyun 		!strcmp(insn->call_dest->name, "__asan_handle_no_return"));
3371*4882a593Smuzhiyun }
3372*4882a593Smuzhiyun 
is_ubsan_insn(struct instruction * insn)3373*4882a593Smuzhiyun static bool is_ubsan_insn(struct instruction *insn)
3374*4882a593Smuzhiyun {
3375*4882a593Smuzhiyun 	return (insn->type == INSN_CALL &&
3376*4882a593Smuzhiyun 		!strcmp(insn->call_dest->name,
3377*4882a593Smuzhiyun 			"__ubsan_handle_builtin_unreachable"));
3378*4882a593Smuzhiyun }
3379*4882a593Smuzhiyun 
ignore_unreachable_insn(struct objtool_file * file,struct instruction * insn)3380*4882a593Smuzhiyun static bool ignore_unreachable_insn(struct objtool_file *file, struct instruction *insn)
3381*4882a593Smuzhiyun {
3382*4882a593Smuzhiyun 	int i;
3383*4882a593Smuzhiyun 	struct instruction *prev_insn;
3384*4882a593Smuzhiyun 
3385*4882a593Smuzhiyun 	if (insn->ignore || insn->type == INSN_NOP || insn->type == INSN_TRAP)
3386*4882a593Smuzhiyun 		return true;
3387*4882a593Smuzhiyun 
3388*4882a593Smuzhiyun 	/*
3389*4882a593Smuzhiyun 	 * Ignore any unused exceptions.  This can happen when a whitelisted
3390*4882a593Smuzhiyun 	 * function has an exception table entry.
3391*4882a593Smuzhiyun 	 *
3392*4882a593Smuzhiyun 	 * Also ignore alternative replacement instructions.  This can happen
3393*4882a593Smuzhiyun 	 * when a whitelisted function uses one of the ALTERNATIVE macros.
3394*4882a593Smuzhiyun 	 */
3395*4882a593Smuzhiyun 	if (!strcmp(insn->sec->name, ".fixup") ||
3396*4882a593Smuzhiyun 	    !strcmp(insn->sec->name, ".altinstr_replacement") ||
3397*4882a593Smuzhiyun 	    !strcmp(insn->sec->name, ".altinstr_aux"))
3398*4882a593Smuzhiyun 		return true;
3399*4882a593Smuzhiyun 
3400*4882a593Smuzhiyun 	if (!insn->func)
3401*4882a593Smuzhiyun 		return false;
3402*4882a593Smuzhiyun 
3403*4882a593Smuzhiyun 	/*
3404*4882a593Smuzhiyun 	 * CONFIG_UBSAN_TRAP inserts a UD2 when it sees
3405*4882a593Smuzhiyun 	 * __builtin_unreachable().  The BUG() macro has an unreachable() after
3406*4882a593Smuzhiyun 	 * the UD2, which causes GCC's undefined trap logic to emit another UD2
3407*4882a593Smuzhiyun 	 * (or occasionally a JMP to UD2).
3408*4882a593Smuzhiyun 	 *
3409*4882a593Smuzhiyun 	 * It may also insert a UD2 after calling a __noreturn function.
3410*4882a593Smuzhiyun 	 */
3411*4882a593Smuzhiyun 	prev_insn = list_prev_entry(insn, list);
3412*4882a593Smuzhiyun 	if ((prev_insn->dead_end || dead_end_function(file, prev_insn->call_dest)) &&
3413*4882a593Smuzhiyun 	    (insn->type == INSN_BUG ||
3414*4882a593Smuzhiyun 	     (insn->type == INSN_JUMP_UNCONDITIONAL &&
3415*4882a593Smuzhiyun 	      insn->jump_dest && insn->jump_dest->type == INSN_BUG)))
3416*4882a593Smuzhiyun 		return true;
3417*4882a593Smuzhiyun 
3418*4882a593Smuzhiyun 	/*
3419*4882a593Smuzhiyun 	 * Check if this (or a subsequent) instruction is related to
3420*4882a593Smuzhiyun 	 * CONFIG_UBSAN or CONFIG_KASAN.
3421*4882a593Smuzhiyun 	 *
3422*4882a593Smuzhiyun 	 * End the search at 5 instructions to avoid going into the weeds.
3423*4882a593Smuzhiyun 	 */
3424*4882a593Smuzhiyun 	for (i = 0; i < 5; i++) {
3425*4882a593Smuzhiyun 
3426*4882a593Smuzhiyun 		if (is_kasan_insn(insn) || is_ubsan_insn(insn))
3427*4882a593Smuzhiyun 			return true;
3428*4882a593Smuzhiyun 
3429*4882a593Smuzhiyun 		if (insn->type == INSN_JUMP_UNCONDITIONAL) {
3430*4882a593Smuzhiyun 			if (insn->jump_dest &&
3431*4882a593Smuzhiyun 			    insn->jump_dest->func == insn->func) {
3432*4882a593Smuzhiyun 				insn = insn->jump_dest;
3433*4882a593Smuzhiyun 				continue;
3434*4882a593Smuzhiyun 			}
3435*4882a593Smuzhiyun 
3436*4882a593Smuzhiyun 			break;
3437*4882a593Smuzhiyun 		}
3438*4882a593Smuzhiyun 
3439*4882a593Smuzhiyun 		if (insn->offset + insn->len >= insn->func->offset + insn->func->len)
3440*4882a593Smuzhiyun 			break;
3441*4882a593Smuzhiyun 
3442*4882a593Smuzhiyun 		insn = list_next_entry(insn, list);
3443*4882a593Smuzhiyun 	}
3444*4882a593Smuzhiyun 
3445*4882a593Smuzhiyun 	return false;
3446*4882a593Smuzhiyun }
3447*4882a593Smuzhiyun 
validate_symbol(struct objtool_file * file,struct section * sec,struct symbol * sym,struct insn_state * state)3448*4882a593Smuzhiyun static int validate_symbol(struct objtool_file *file, struct section *sec,
3449*4882a593Smuzhiyun 			   struct symbol *sym, struct insn_state *state)
3450*4882a593Smuzhiyun {
3451*4882a593Smuzhiyun 	struct instruction *insn;
3452*4882a593Smuzhiyun 	int ret;
3453*4882a593Smuzhiyun 
3454*4882a593Smuzhiyun 	if (!sym->len) {
3455*4882a593Smuzhiyun 		WARN("%s() is missing an ELF size annotation", sym->name);
3456*4882a593Smuzhiyun 		return 1;
3457*4882a593Smuzhiyun 	}
3458*4882a593Smuzhiyun 
3459*4882a593Smuzhiyun 	if (sym->pfunc != sym || sym->alias != sym)
3460*4882a593Smuzhiyun 		return 0;
3461*4882a593Smuzhiyun 
3462*4882a593Smuzhiyun 	insn = find_insn(file, sec, sym->offset);
3463*4882a593Smuzhiyun 	if (!insn || insn->ignore || insn->visited)
3464*4882a593Smuzhiyun 		return 0;
3465*4882a593Smuzhiyun 
3466*4882a593Smuzhiyun 	state->uaccess = sym->uaccess_safe;
3467*4882a593Smuzhiyun 
3468*4882a593Smuzhiyun 	ret = validate_branch(file, insn->func, insn, *state);
3469*4882a593Smuzhiyun 	if (ret && backtrace)
3470*4882a593Smuzhiyun 		BT_FUNC("<=== (sym)", insn);
3471*4882a593Smuzhiyun 	return ret;
3472*4882a593Smuzhiyun }
3473*4882a593Smuzhiyun 
validate_section(struct objtool_file * file,struct section * sec)3474*4882a593Smuzhiyun static int validate_section(struct objtool_file *file, struct section *sec)
3475*4882a593Smuzhiyun {
3476*4882a593Smuzhiyun 	struct insn_state state;
3477*4882a593Smuzhiyun 	struct symbol *func;
3478*4882a593Smuzhiyun 	int warnings = 0;
3479*4882a593Smuzhiyun 
3480*4882a593Smuzhiyun 	list_for_each_entry(func, &sec->symbol_list, list) {
3481*4882a593Smuzhiyun 		if (func->type != STT_FUNC)
3482*4882a593Smuzhiyun 			continue;
3483*4882a593Smuzhiyun 
3484*4882a593Smuzhiyun 		init_insn_state(&state, sec);
3485*4882a593Smuzhiyun 		set_func_state(&state.cfi);
3486*4882a593Smuzhiyun 
3487*4882a593Smuzhiyun 		warnings += validate_symbol(file, sec, func, &state);
3488*4882a593Smuzhiyun 	}
3489*4882a593Smuzhiyun 
3490*4882a593Smuzhiyun 	return warnings;
3491*4882a593Smuzhiyun }
3492*4882a593Smuzhiyun 
validate_vmlinux_functions(struct objtool_file * file)3493*4882a593Smuzhiyun static int validate_vmlinux_functions(struct objtool_file *file)
3494*4882a593Smuzhiyun {
3495*4882a593Smuzhiyun 	struct section *sec;
3496*4882a593Smuzhiyun 	int warnings = 0;
3497*4882a593Smuzhiyun 
3498*4882a593Smuzhiyun 	sec = find_section_by_name(file->elf, ".noinstr.text");
3499*4882a593Smuzhiyun 	if (sec) {
3500*4882a593Smuzhiyun 		warnings += validate_section(file, sec);
3501*4882a593Smuzhiyun 		warnings += validate_unwind_hints(file, sec);
3502*4882a593Smuzhiyun 	}
3503*4882a593Smuzhiyun 
3504*4882a593Smuzhiyun 	sec = find_section_by_name(file->elf, ".entry.text");
3505*4882a593Smuzhiyun 	if (sec) {
3506*4882a593Smuzhiyun 		warnings += validate_section(file, sec);
3507*4882a593Smuzhiyun 		warnings += validate_unwind_hints(file, sec);
3508*4882a593Smuzhiyun 	}
3509*4882a593Smuzhiyun 
3510*4882a593Smuzhiyun 	return warnings;
3511*4882a593Smuzhiyun }
3512*4882a593Smuzhiyun 
validate_functions(struct objtool_file * file)3513*4882a593Smuzhiyun static int validate_functions(struct objtool_file *file)
3514*4882a593Smuzhiyun {
3515*4882a593Smuzhiyun 	struct section *sec;
3516*4882a593Smuzhiyun 	int warnings = 0;
3517*4882a593Smuzhiyun 
3518*4882a593Smuzhiyun 	for_each_sec(file, sec) {
3519*4882a593Smuzhiyun 		if (!(sec->sh.sh_flags & SHF_EXECINSTR))
3520*4882a593Smuzhiyun 			continue;
3521*4882a593Smuzhiyun 
3522*4882a593Smuzhiyun 		warnings += validate_section(file, sec);
3523*4882a593Smuzhiyun 	}
3524*4882a593Smuzhiyun 
3525*4882a593Smuzhiyun 	return warnings;
3526*4882a593Smuzhiyun }
3527*4882a593Smuzhiyun 
validate_reachable_instructions(struct objtool_file * file)3528*4882a593Smuzhiyun static int validate_reachable_instructions(struct objtool_file *file)
3529*4882a593Smuzhiyun {
3530*4882a593Smuzhiyun 	struct instruction *insn;
3531*4882a593Smuzhiyun 
3532*4882a593Smuzhiyun 	if (file->ignore_unreachables)
3533*4882a593Smuzhiyun 		return 0;
3534*4882a593Smuzhiyun 
3535*4882a593Smuzhiyun 	for_each_insn(file, insn) {
3536*4882a593Smuzhiyun 		if (insn->visited || ignore_unreachable_insn(file, insn))
3537*4882a593Smuzhiyun 			continue;
3538*4882a593Smuzhiyun 
3539*4882a593Smuzhiyun 		WARN_FUNC("unreachable instruction", insn->sec, insn->offset);
3540*4882a593Smuzhiyun 		return 1;
3541*4882a593Smuzhiyun 	}
3542*4882a593Smuzhiyun 
3543*4882a593Smuzhiyun 	return 0;
3544*4882a593Smuzhiyun }
3545*4882a593Smuzhiyun 
check(struct objtool_file * file)3546*4882a593Smuzhiyun int check(struct objtool_file *file)
3547*4882a593Smuzhiyun {
3548*4882a593Smuzhiyun 	int ret, warnings = 0;
3549*4882a593Smuzhiyun 
3550*4882a593Smuzhiyun 	arch_initial_func_cfi_state(&initial_func_cfi);
3551*4882a593Smuzhiyun 	init_cfi_state(&init_cfi);
3552*4882a593Smuzhiyun 	init_cfi_state(&func_cfi);
3553*4882a593Smuzhiyun 	set_func_state(&func_cfi);
3554*4882a593Smuzhiyun 
3555*4882a593Smuzhiyun 	if (!cfi_hash_alloc())
3556*4882a593Smuzhiyun 		goto out;
3557*4882a593Smuzhiyun 
3558*4882a593Smuzhiyun 	cfi_hash_add(&init_cfi);
3559*4882a593Smuzhiyun 	cfi_hash_add(&func_cfi);
3560*4882a593Smuzhiyun 
3561*4882a593Smuzhiyun 	ret = decode_sections(file);
3562*4882a593Smuzhiyun 	if (ret < 0)
3563*4882a593Smuzhiyun 		goto out;
3564*4882a593Smuzhiyun 
3565*4882a593Smuzhiyun 	warnings += ret;
3566*4882a593Smuzhiyun 
3567*4882a593Smuzhiyun 	if (list_empty(&file->insn_list))
3568*4882a593Smuzhiyun 		goto out;
3569*4882a593Smuzhiyun 
3570*4882a593Smuzhiyun 	if (vmlinux && !validate_dup) {
3571*4882a593Smuzhiyun 		ret = validate_vmlinux_functions(file);
3572*4882a593Smuzhiyun 		if (ret < 0)
3573*4882a593Smuzhiyun 			goto out;
3574*4882a593Smuzhiyun 
3575*4882a593Smuzhiyun 		warnings += ret;
3576*4882a593Smuzhiyun 		goto out;
3577*4882a593Smuzhiyun 	}
3578*4882a593Smuzhiyun 
3579*4882a593Smuzhiyun 	if (retpoline) {
3580*4882a593Smuzhiyun 		ret = validate_retpoline(file);
3581*4882a593Smuzhiyun 		if (ret < 0)
3582*4882a593Smuzhiyun 			return ret;
3583*4882a593Smuzhiyun 		warnings += ret;
3584*4882a593Smuzhiyun 	}
3585*4882a593Smuzhiyun 
3586*4882a593Smuzhiyun 	ret = validate_functions(file);
3587*4882a593Smuzhiyun 	if (ret < 0)
3588*4882a593Smuzhiyun 		goto out;
3589*4882a593Smuzhiyun 	warnings += ret;
3590*4882a593Smuzhiyun 
3591*4882a593Smuzhiyun 	ret = validate_unwind_hints(file, NULL);
3592*4882a593Smuzhiyun 	if (ret < 0)
3593*4882a593Smuzhiyun 		goto out;
3594*4882a593Smuzhiyun 	warnings += ret;
3595*4882a593Smuzhiyun 
3596*4882a593Smuzhiyun 	if (unret) {
3597*4882a593Smuzhiyun 		/*
3598*4882a593Smuzhiyun 		 * Must be after validate_branch() and friends, it plays
3599*4882a593Smuzhiyun 		 * further games with insn->visited.
3600*4882a593Smuzhiyun 		 */
3601*4882a593Smuzhiyun 		ret = validate_unret(file);
3602*4882a593Smuzhiyun 		if (ret < 0)
3603*4882a593Smuzhiyun 			return ret;
3604*4882a593Smuzhiyun 		warnings += ret;
3605*4882a593Smuzhiyun 	}
3606*4882a593Smuzhiyun 
3607*4882a593Smuzhiyun 	if (!warnings) {
3608*4882a593Smuzhiyun 		ret = validate_reachable_instructions(file);
3609*4882a593Smuzhiyun 		if (ret < 0)
3610*4882a593Smuzhiyun 			goto out;
3611*4882a593Smuzhiyun 		warnings += ret;
3612*4882a593Smuzhiyun 	}
3613*4882a593Smuzhiyun 
3614*4882a593Smuzhiyun 	ret = create_static_call_sections(file);
3615*4882a593Smuzhiyun 	if (ret < 0)
3616*4882a593Smuzhiyun 		goto out;
3617*4882a593Smuzhiyun 	warnings += ret;
3618*4882a593Smuzhiyun 
3619*4882a593Smuzhiyun 	if (mcount) {
3620*4882a593Smuzhiyun 		ret = create_mcount_loc_sections(file);
3621*4882a593Smuzhiyun 		if (ret < 0)
3622*4882a593Smuzhiyun 			goto out;
3623*4882a593Smuzhiyun 		warnings += ret;
3624*4882a593Smuzhiyun 	}
3625*4882a593Smuzhiyun 
3626*4882a593Smuzhiyun 	if (retpoline) {
3627*4882a593Smuzhiyun 		ret = create_retpoline_sites_sections(file);
3628*4882a593Smuzhiyun 		if (ret < 0)
3629*4882a593Smuzhiyun 			goto out;
3630*4882a593Smuzhiyun 		warnings += ret;
3631*4882a593Smuzhiyun 	}
3632*4882a593Smuzhiyun 
3633*4882a593Smuzhiyun 	if (rethunk) {
3634*4882a593Smuzhiyun 		ret = create_return_sites_sections(file);
3635*4882a593Smuzhiyun 		if (ret < 0)
3636*4882a593Smuzhiyun 			goto out;
3637*4882a593Smuzhiyun 		warnings += ret;
3638*4882a593Smuzhiyun 	}
3639*4882a593Smuzhiyun 
3640*4882a593Smuzhiyun 	if (stats) {
3641*4882a593Smuzhiyun 		printf("nr_insns_visited: %ld\n", nr_insns_visited);
3642*4882a593Smuzhiyun 		printf("nr_cfi: %ld\n", nr_cfi);
3643*4882a593Smuzhiyun 		printf("nr_cfi_reused: %ld\n", nr_cfi_reused);
3644*4882a593Smuzhiyun 		printf("nr_cfi_cache: %ld\n", nr_cfi_cache);
3645*4882a593Smuzhiyun 	}
3646*4882a593Smuzhiyun 
3647*4882a593Smuzhiyun out:
3648*4882a593Smuzhiyun 	/*
3649*4882a593Smuzhiyun 	 *  For now, don't fail the kernel build on fatal warnings.  These
3650*4882a593Smuzhiyun 	 *  errors are still fairly common due to the growing matrix of
3651*4882a593Smuzhiyun 	 *  supported toolchains and their recent pace of change.
3652*4882a593Smuzhiyun 	 */
3653*4882a593Smuzhiyun 	return 0;
3654*4882a593Smuzhiyun }
3655