check.c 69.4 KB
Newer Older
1
// SPDX-License-Identifier: GPL-2.0-or-later
2 3 4 5 6 7 8
/*
 * Copyright (C) 2015-2017 Josh Poimboeuf <jpoimboe@redhat.com>
 */

#include <string.h>
#include <stdlib.h>

9
#include "builtin.h"
10 11
#include "cfi.h"
#include "arch.h"
12 13 14
#include "check.h"
#include "special.h"
#include "warn.h"
P
Peter Zijlstra 已提交
15
#include "arch_elf.h"
16 17 18 19

#include <linux/hashtable.h>
#include <linux/kernel.h>

20 21
#define FAKE_JUMP_OFFSET -1

22 23
#define C_JUMP_TABLE_SECTION ".rodata..c_jump_table"

24 25 26
struct alternative {
	struct list_head list;
	struct instruction *insn;
P
Peter Zijlstra 已提交
27
	bool skip_orig;
28 29 30
};

const char *objname;
31
struct cfi_init_state initial_func_cfi;
32

33 34
struct instruction *find_insn(struct objtool_file *file,
			      struct section *sec, unsigned long offset)
35 36 37
{
	struct instruction *insn;

38
	hash_for_each_possible(file->insn_hash, insn, hash, sec_offset_hash(sec, offset)) {
39 40
		if (insn->sec == sec && insn->offset == offset)
			return insn;
41
	}
42 43 44 45 46 47 48 49 50

	return NULL;
}

static struct instruction *next_insn_same_sec(struct objtool_file *file,
					      struct instruction *insn)
{
	struct instruction *next = list_next_entry(insn, list);

51
	if (!next || &next->list == &file->insn_list || next->sec != insn->sec)
52 53 54 55 56
		return NULL;

	return next;
}

57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76
static struct instruction *next_insn_same_func(struct objtool_file *file,
					       struct instruction *insn)
{
	struct instruction *next = list_next_entry(insn, list);
	struct symbol *func = insn->func;

	if (!func)
		return NULL;

	if (&next->list != &file->insn_list && next->func == func)
		return next;

	/* Check if we're already in the subfunction: */
	if (func == func->cfunc)
		return NULL;

	/* Move to the subfunction: */
	return find_insn(file, func->cfunc->sec, func->cfunc->offset);
}

77 78 79 80 81 82 83 84 85 86 87
static struct instruction *prev_insn_same_sym(struct objtool_file *file,
					       struct instruction *insn)
{
	struct instruction *prev = list_prev_entry(insn, list);

	if (&prev->list != &file->insn_list && prev->func == insn->func)
		return prev;

	return NULL;
}

88
#define func_for_each_insn(file, func, insn)				\
89 90 91 92
	for (insn = find_insn(file, func->sec, func->offset);		\
	     insn;							\
	     insn = next_insn_same_func(file, insn))

93 94
#define sym_for_each_insn(file, sym, insn)				\
	for (insn = find_insn(file, sym->sec, sym->offset);		\
95
	     insn && &insn->list != &file->insn_list &&			\
96 97
		insn->sec == sym->sec &&				\
		insn->offset < sym->offset + sym->len;			\
98 99
	     insn = list_next_entry(insn, list))

100
#define sym_for_each_insn_continue_reverse(file, sym, insn)		\
101 102
	for (insn = list_prev_entry(insn, list);			\
	     &insn->list != &file->insn_list &&				\
103
		insn->sec == sym->sec && insn->offset >= sym->offset;	\
104 105 106 107 108
	     insn = list_prev_entry(insn, list))

#define sec_for_each_insn_from(file, insn)				\
	for (; insn; insn = next_insn_same_sec(file, insn))

109 110 111
#define sec_for_each_insn_continue(file, insn)				\
	for (insn = next_insn_same_sec(file, insn); insn;		\
	     insn = next_insn_same_sec(file, insn))
112

113 114 115 116 117 118
static bool is_static_jump(struct instruction *insn)
{
	return insn->type == INSN_JUMP_CONDITIONAL ||
	       insn->type == INSN_JUMP_UNCONDITIONAL;
}

119 120 121 122 123 124
static bool is_sibling_call(struct instruction *insn)
{
	/* An indirect jump is either a sibling call or a jump to a table. */
	if (insn->type == INSN_JUMP_DYNAMIC)
		return list_empty(&insn->alts);

125
	if (!is_static_jump(insn))
126 127 128 129 130 131
		return false;

	/* add_jump_destinations() sets insn->call_dest for sibling calls. */
	return !!insn->call_dest;
}

132 133 134 135 136 137 138 139 140
/*
 * This checks to see if the given function is a "noreturn" function.
 *
 * For global functions which are outside the scope of this object file, we
 * have to keep a manual list of them.
 *
 * For local functions, we have to detect them manually by simply looking for
 * the lack of a return instruction.
 */
141 142
static bool __dead_end_function(struct objtool_file *file, struct symbol *func,
				int recursion)
143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161
{
	int i;
	struct instruction *insn;
	bool empty = true;

	/*
	 * Unfortunately these have to be hard coded because the noreturn
	 * attribute isn't provided in ELF data.
	 */
	static const char * const global_noreturns[] = {
		"__stack_chk_fail",
		"panic",
		"do_exit",
		"do_task_dead",
		"__module_put_and_exit",
		"complete_and_exit",
		"__reiserfs_panic",
		"lbug_with_loc",
		"fortify_panic",
162
		"usercopy_abort",
163
		"machine_real_restart",
164
		"rewind_stack_do_exit",
165
		"kunit_try_catch_throw",
166 167
	};

168 169 170
	if (!func)
		return false;

171
	if (func->bind == STB_WEAK)
172
		return false;
173 174 175 176

	if (func->bind == STB_GLOBAL)
		for (i = 0; i < ARRAY_SIZE(global_noreturns); i++)
			if (!strcmp(func->name, global_noreturns[i]))
177
				return true;
178

179
	if (!func->len)
180
		return false;
181

182 183
	insn = find_insn(file, func->sec, func->offset);
	if (!insn->func)
184
		return false;
185

186
	func_for_each_insn(file, func, insn) {
187 188 189
		empty = false;

		if (insn->type == INSN_RETURN)
190
			return false;
191 192 193
	}

	if (empty)
194
		return false;
195 196 197 198 199 200

	/*
	 * A function can have a sibling call instead of a return.  In that
	 * case, the function's dead-end status depends on whether the target
	 * of the sibling call returns.
	 */
201
	func_for_each_insn(file, func, insn) {
202
		if (is_sibling_call(insn)) {
203 204 205 206
			struct instruction *dest = insn->jump_dest;

			if (!dest)
				/* sibling call to another file */
207
				return false;
208

209 210 211 212 213 214 215 216
			/* local sibling call */
			if (recursion == 5) {
				/*
				 * Infinite recursion: two functions have
				 * sibling calls to each other.  This is a very
				 * rare case.  It means they aren't dead ends.
				 */
				return false;
217 218
			}

219 220
			return __dead_end_function(file, dest->func, recursion+1);
		}
221 222
	}

223
	return true;
224 225
}

226
static bool dead_end_function(struct objtool_file *file, struct symbol *func)
227 228 229 230
{
	return __dead_end_function(file, func, 0);
}

231
static void init_cfi_state(struct cfi_state *cfi)
232 233 234
{
	int i;

235
	for (i = 0; i < CFI_NUM_REGS; i++) {
236 237
		cfi->regs[i].base = CFI_UNDEFINED;
		cfi->vals[i].base = CFI_UNDEFINED;
238
	}
239 240 241 242 243
	cfi->cfa.base = CFI_UNDEFINED;
	cfi->drap_reg = CFI_UNDEFINED;
	cfi->drap_offset = -1;
}

244
static void init_insn_state(struct insn_state *state, struct section *sec)
245 246 247
{
	memset(state, 0, sizeof(*state));
	init_cfi_state(&state->cfi);
248 249 250 251 252 253 254 255

	/*
	 * We need the full vmlinux for noinstr validation, otherwise we can
	 * not correctly determine insn->call_dest->sec (external symbols do
	 * not have a section).
	 */
	if (vmlinux && sec)
		state->noinstr = sec->noinstr;
256 257
}

258 259 260 261 262 263 264 265 266 267
/*
 * Call the arch-specific instruction decoder for all the instructions and add
 * them to the global instruction list.
 */
static int decode_instructions(struct objtool_file *file)
{
	struct section *sec;
	struct symbol *func;
	unsigned long offset;
	struct instruction *insn;
P
Peter Zijlstra 已提交
268
	unsigned long nr_insns = 0;
269 270
	int ret;

271
	for_each_sec(file, sec) {
272 273 274 275

		if (!(sec->sh.sh_flags & SHF_EXECINSTR))
			continue;

276 277 278 279 280
		if (strcmp(sec->name, ".altinstr_replacement") &&
		    strcmp(sec->name, ".altinstr_aux") &&
		    strncmp(sec->name, ".discard.", 9))
			sec->text = true;

281 282
		if (!strcmp(sec->name, ".noinstr.text") ||
		    !strcmp(sec->name, ".entry.text"))
283 284
			sec->noinstr = true;

285 286
		for (offset = 0; offset < sec->len; offset += insn->len) {
			insn = malloc(sizeof(*insn));
287 288 289 290
			if (!insn) {
				WARN("malloc failed");
				return -1;
			}
291 292
			memset(insn, 0, sizeof(*insn));
			INIT_LIST_HEAD(&insn->alts);
293
			INIT_LIST_HEAD(&insn->stack_ops);
294
			init_cfi_state(&insn->cfi);
295

296 297 298 299 300 301
			insn->sec = sec;
			insn->offset = offset;

			ret = arch_decode_instruction(file->elf, sec, offset,
						      sec->len - offset,
						      &insn->len, &insn->type,
302
						      &insn->immediate,
303
						      &insn->stack_ops);
304
			if (ret)
305
				goto err;
306

307
			hash_add(file->insn_hash, &insn->hash, sec_offset_hash(sec, insn->offset));
308
			list_add_tail(&insn->list, &file->insn_list);
P
Peter Zijlstra 已提交
309
			nr_insns++;
310 311 312
		}

		list_for_each_entry(func, &sec->symbol_list, list) {
313
			if (func->type != STT_FUNC || func->alias != func)
314 315 316 317 318 319 320 321
				continue;

			if (!find_insn(file, sec, func->offset)) {
				WARN("%s(): can't find starting instruction",
				     func->name);
				return -1;
			}

322
			sym_for_each_insn(file, func, insn)
323
				insn->func = func;
324 325 326
		}
	}

P
Peter Zijlstra 已提交
327 328 329
	if (stats)
		printf("nr_insns: %lu\n", nr_insns);

330
	return 0;
331 332 333 334

err:
	free(insn);
	return ret;
335 336
}

337 338 339 340 341 342 343 344 345 346 347 348 349
static struct instruction *find_last_insn(struct objtool_file *file,
					  struct section *sec)
{
	struct instruction *insn = NULL;
	unsigned int offset;
	unsigned int end = (sec->len > 10) ? sec->len - 10 : 0;

	for (offset = sec->len - 1; offset >= end && !insn; offset--)
		insn = find_insn(file, sec, offset);

	return insn;
}

350
/*
351
 * Mark "ud2" instructions and manually annotated dead ends.
352 353 354 355
 */
static int add_dead_ends(struct objtool_file *file)
{
	struct section *sec;
M
Matt Helsley 已提交
356
	struct reloc *reloc;
357 358
	struct instruction *insn;

359 360 361 362 363 364 365 366 367 368 369
	/*
	 * By default, "ud2" is a dead end unless otherwise annotated, because
	 * GCC 7 inserts it for certain divide-by-zero cases.
	 */
	for_each_insn(file, insn)
		if (insn->type == INSN_BUG)
			insn->dead_end = true;

	/*
	 * Check for manually annotated dead ends.
	 */
370 371
	sec = find_section_by_name(file->elf, ".rela.discard.unreachable");
	if (!sec)
372
		goto reachable;
373

M
Matt Helsley 已提交
374 375
	list_for_each_entry(reloc, &sec->reloc_list, list) {
		if (reloc->sym->type != STT_SECTION) {
376 377 378
			WARN("unexpected relocation symbol type in %s", sec->name);
			return -1;
		}
M
Matt Helsley 已提交
379
		insn = find_insn(file, reloc->sym->sec, reloc->addend);
380 381
		if (insn)
			insn = list_prev_entry(insn, list);
M
Matt Helsley 已提交
382 383
		else if (reloc->addend == reloc->sym->sec->len) {
			insn = find_last_insn(file, reloc->sym->sec);
384
			if (!insn) {
385
				WARN("can't find unreachable insn at %s+0x%x",
M
Matt Helsley 已提交
386
				     reloc->sym->sec->name, reloc->addend);
387 388 389 390
				return -1;
			}
		} else {
			WARN("can't find unreachable insn at %s+0x%x",
M
Matt Helsley 已提交
391
			     reloc->sym->sec->name, reloc->addend);
392 393 394 395 396 397
			return -1;
		}

		insn->dead_end = true;
	}

398 399 400 401 402 403 404 405 406 407 408
reachable:
	/*
	 * These manually annotated reachable checks are needed for GCC 4.4,
	 * where the Linux unreachable() macro isn't supported.  In that case
	 * GCC doesn't know the "ud2" is fatal, so it generates code as if it's
	 * not a dead end.
	 */
	sec = find_section_by_name(file->elf, ".rela.discard.reachable");
	if (!sec)
		return 0;

M
Matt Helsley 已提交
409 410
	list_for_each_entry(reloc, &sec->reloc_list, list) {
		if (reloc->sym->type != STT_SECTION) {
411 412 413
			WARN("unexpected relocation symbol type in %s", sec->name);
			return -1;
		}
M
Matt Helsley 已提交
414
		insn = find_insn(file, reloc->sym->sec, reloc->addend);
415 416
		if (insn)
			insn = list_prev_entry(insn, list);
M
Matt Helsley 已提交
417 418
		else if (reloc->addend == reloc->sym->sec->len) {
			insn = find_last_insn(file, reloc->sym->sec);
419
			if (!insn) {
420
				WARN("can't find reachable insn at %s+0x%x",
M
Matt Helsley 已提交
421
				     reloc->sym->sec->name, reloc->addend);
422 423 424 425
				return -1;
			}
		} else {
			WARN("can't find reachable insn at %s+0x%x",
M
Matt Helsley 已提交
426
			     reloc->sym->sec->name, reloc->addend);
427 428 429 430 431 432
			return -1;
		}

		insn->dead_end = false;
	}

433 434 435 436 437 438 439 440 441 442 443
	return 0;
}

/*
 * Warnings shouldn't be reported for ignored functions.
 */
static void add_ignores(struct objtool_file *file)
{
	struct instruction *insn;
	struct section *sec;
	struct symbol *func;
M
Matt Helsley 已提交
444
	struct reloc *reloc;
445

P
Peter Zijlstra 已提交
446 447 448
	sec = find_section_by_name(file->elf, ".rela.discard.func_stack_frame_non_standard");
	if (!sec)
		return;
449

M
Matt Helsley 已提交
450 451
	list_for_each_entry(reloc, &sec->reloc_list, list) {
		switch (reloc->sym->type) {
P
Peter Zijlstra 已提交
452
		case STT_FUNC:
M
Matt Helsley 已提交
453
			func = reloc->sym;
P
Peter Zijlstra 已提交
454 455 456
			break;

		case STT_SECTION:
M
Matt Helsley 已提交
457
			func = find_func_by_offset(reloc->sym->sec, reloc->addend);
458
			if (!func)
459
				continue;
P
Peter Zijlstra 已提交
460
			break;
461

P
Peter Zijlstra 已提交
462
		default:
M
Matt Helsley 已提交
463
			WARN("unexpected relocation symbol type in %s: %d", sec->name, reloc->sym->type);
P
Peter Zijlstra 已提交
464
			continue;
465
		}
P
Peter Zijlstra 已提交
466

467
		func_for_each_insn(file, func, insn)
P
Peter Zijlstra 已提交
468
			insn->ignore = true;
469 470 471
	}
}

P
Peter Zijlstra 已提交
472 473 474 475 476 477 478 479 480 481 482 483 484 485 486 487 488 489 490 491 492 493 494 495 496 497 498 499 500 501 502 503 504 505 506 507 508
/*
 * This is a whitelist of functions that is allowed to be called with AC set.
 * The list is meant to be minimal and only contains compiler instrumentation
 * ABI and a few functions used to implement *_{to,from}_user() functions.
 *
 * These functions must not directly change AC, but may PUSHF/POPF.
 */
static const char *uaccess_safe_builtin[] = {
	/* KASAN */
	"kasan_report",
	"check_memory_region",
	/* KASAN out-of-line */
	"__asan_loadN_noabort",
	"__asan_load1_noabort",
	"__asan_load2_noabort",
	"__asan_load4_noabort",
	"__asan_load8_noabort",
	"__asan_load16_noabort",
	"__asan_storeN_noabort",
	"__asan_store1_noabort",
	"__asan_store2_noabort",
	"__asan_store4_noabort",
	"__asan_store8_noabort",
	"__asan_store16_noabort",
	/* KASAN in-line */
	"__asan_report_load_n_noabort",
	"__asan_report_load1_noabort",
	"__asan_report_load2_noabort",
	"__asan_report_load4_noabort",
	"__asan_report_load8_noabort",
	"__asan_report_load16_noabort",
	"__asan_report_store_n_noabort",
	"__asan_report_store1_noabort",
	"__asan_report_store2_noabort",
	"__asan_report_store4_noabort",
	"__asan_report_store8_noabort",
	"__asan_report_store16_noabort",
509
	/* KCSAN */
510
	"__kcsan_check_access",
511 512
	"kcsan_found_watchpoint",
	"kcsan_setup_watchpoint",
513
	"kcsan_check_scoped_accesses",
514 515
	"kcsan_disable_current",
	"kcsan_enable_current_nowarn",
516 517 518 519 520 521 522 523 524 525 526 527 528 529 530
	/* KCSAN/TSAN */
	"__tsan_func_entry",
	"__tsan_func_exit",
	"__tsan_read_range",
	"__tsan_write_range",
	"__tsan_read1",
	"__tsan_read2",
	"__tsan_read4",
	"__tsan_read8",
	"__tsan_read16",
	"__tsan_write1",
	"__tsan_write2",
	"__tsan_write4",
	"__tsan_write8",
	"__tsan_write16",
531 532 533 534 535
	"__tsan_read_write1",
	"__tsan_read_write2",
	"__tsan_read_write4",
	"__tsan_read_write8",
	"__tsan_read_write16",
536 537 538 539 540 541 542 543 544 545 546 547 548 549 550 551 552 553 554 555 556 557 558 559 560 561 562 563 564 565 566 567 568 569 570 571 572 573 574 575 576 577 578 579 580 581 582 583 584 585
	"__tsan_atomic8_load",
	"__tsan_atomic16_load",
	"__tsan_atomic32_load",
	"__tsan_atomic64_load",
	"__tsan_atomic8_store",
	"__tsan_atomic16_store",
	"__tsan_atomic32_store",
	"__tsan_atomic64_store",
	"__tsan_atomic8_exchange",
	"__tsan_atomic16_exchange",
	"__tsan_atomic32_exchange",
	"__tsan_atomic64_exchange",
	"__tsan_atomic8_fetch_add",
	"__tsan_atomic16_fetch_add",
	"__tsan_atomic32_fetch_add",
	"__tsan_atomic64_fetch_add",
	"__tsan_atomic8_fetch_sub",
	"__tsan_atomic16_fetch_sub",
	"__tsan_atomic32_fetch_sub",
	"__tsan_atomic64_fetch_sub",
	"__tsan_atomic8_fetch_and",
	"__tsan_atomic16_fetch_and",
	"__tsan_atomic32_fetch_and",
	"__tsan_atomic64_fetch_and",
	"__tsan_atomic8_fetch_or",
	"__tsan_atomic16_fetch_or",
	"__tsan_atomic32_fetch_or",
	"__tsan_atomic64_fetch_or",
	"__tsan_atomic8_fetch_xor",
	"__tsan_atomic16_fetch_xor",
	"__tsan_atomic32_fetch_xor",
	"__tsan_atomic64_fetch_xor",
	"__tsan_atomic8_fetch_nand",
	"__tsan_atomic16_fetch_nand",
	"__tsan_atomic32_fetch_nand",
	"__tsan_atomic64_fetch_nand",
	"__tsan_atomic8_compare_exchange_strong",
	"__tsan_atomic16_compare_exchange_strong",
	"__tsan_atomic32_compare_exchange_strong",
	"__tsan_atomic64_compare_exchange_strong",
	"__tsan_atomic8_compare_exchange_weak",
	"__tsan_atomic16_compare_exchange_weak",
	"__tsan_atomic32_compare_exchange_weak",
	"__tsan_atomic64_compare_exchange_weak",
	"__tsan_atomic8_compare_exchange_val",
	"__tsan_atomic16_compare_exchange_val",
	"__tsan_atomic32_compare_exchange_val",
	"__tsan_atomic64_compare_exchange_val",
	"__tsan_atomic_thread_fence",
	"__tsan_atomic_signal_fence",
P
Peter Zijlstra 已提交
586 587
	/* KCOV */
	"write_comp_data",
588
	"check_kcov_mode",
P
Peter Zijlstra 已提交
589 590 591 592 593 594 595 596 597
	"__sanitizer_cov_trace_pc",
	"__sanitizer_cov_trace_const_cmp1",
	"__sanitizer_cov_trace_const_cmp2",
	"__sanitizer_cov_trace_const_cmp4",
	"__sanitizer_cov_trace_const_cmp8",
	"__sanitizer_cov_trace_cmp1",
	"__sanitizer_cov_trace_cmp2",
	"__sanitizer_cov_trace_cmp4",
	"__sanitizer_cov_trace_cmp8",
598
	"__sanitizer_cov_trace_switch",
P
Peter Zijlstra 已提交
599 600 601 602
	/* UBSAN */
	"ubsan_type_mismatch_common",
	"__ubsan_handle_type_mismatch",
	"__ubsan_handle_type_mismatch_v1",
603
	"__ubsan_handle_shift_out_of_bounds",
P
Peter Zijlstra 已提交
604 605 606
	/* misc */
	"csum_partial_copy_generic",
	"__memcpy_mcsafe",
607
	"mcsafe_handle_tail",
P
Peter Zijlstra 已提交
608 609 610 611 612 613 614 615 616 617 618 619 620 621 622 623 624
	"ftrace_likely_update", /* CONFIG_TRACE_BRANCH_PROFILING */
	NULL
};

static void add_uaccess_safe(struct objtool_file *file)
{
	struct symbol *func;
	const char **name;

	if (!uaccess)
		return;

	for (name = uaccess_safe_builtin; *name; name++) {
		func = find_symbol_by_name(file->elf, *name);
		if (!func)
			continue;

625
		func->uaccess_safe = true;
626 627 628
	}
}

629 630 631 632 633 634
/*
 * FIXME: For now, just ignore any alternatives which add retpolines.  This is
 * a temporary hack, as it doesn't allow ORC to unwind from inside a retpoline.
 * But it at least allows objtool to understand the control flow *around* the
 * retpoline.
 */
635
static int add_ignore_alternatives(struct objtool_file *file)
636 637
{
	struct section *sec;
M
Matt Helsley 已提交
638
	struct reloc *reloc;
639 640
	struct instruction *insn;

641
	sec = find_section_by_name(file->elf, ".rela.discard.ignore_alts");
642 643 644
	if (!sec)
		return 0;

M
Matt Helsley 已提交
645 646
	list_for_each_entry(reloc, &sec->reloc_list, list) {
		if (reloc->sym->type != STT_SECTION) {
647 648 649 650
			WARN("unexpected relocation symbol type in %s", sec->name);
			return -1;
		}

M
Matt Helsley 已提交
651
		insn = find_insn(file, reloc->sym->sec, reloc->addend);
652
		if (!insn) {
653
			WARN("bad .discard.ignore_alts entry");
654 655 656 657 658 659 660 661 662
			return -1;
		}

		insn->ignore_alts = true;
	}

	return 0;
}

663 664 665 666 667 668
/*
 * Find the destination instructions for all jumps.
 */
static int add_jump_destinations(struct objtool_file *file)
{
	struct instruction *insn;
M
Matt Helsley 已提交
669
	struct reloc *reloc;
670 671 672 673
	struct section *dest_sec;
	unsigned long dest_off;

	for_each_insn(file, insn) {
674
		if (!is_static_jump(insn))
675 676
			continue;

677
		if (insn->ignore || insn->offset == FAKE_JUMP_OFFSET)
678 679
			continue;

M
Matt Helsley 已提交
680
		reloc = find_reloc_by_dest_range(file->elf, insn->sec,
681
					       insn->offset, insn->len);
M
Matt Helsley 已提交
682
		if (!reloc) {
683
			dest_sec = insn->sec;
684
			dest_off = arch_jump_destination(insn);
M
Matt Helsley 已提交
685 686 687 688 689 690 691 692
		} else if (reloc->sym->type == STT_SECTION) {
			dest_sec = reloc->sym->sec;
			dest_off = arch_dest_reloc_offset(reloc->addend);
		} else if (reloc->sym->sec->idx) {
			dest_sec = reloc->sym->sec;
			dest_off = reloc->sym->sym.st_value +
				   arch_dest_reloc_offset(reloc->addend);
		} else if (strstr(reloc->sym->name, "_indirect_thunk_")) {
693 694 695 696
			/*
			 * Retpoline jumps are really dynamic jumps in
			 * disguise, so convert them accordingly.
			 */
697 698 699 700 701
			if (insn->type == INSN_JUMP_UNCONDITIONAL)
				insn->type = INSN_JUMP_DYNAMIC;
			else
				insn->type = INSN_JUMP_DYNAMIC_CONDITIONAL;

702
			insn->retpoline_safe = true;
703
			continue;
704
		} else {
705
			/* external sibling call */
M
Matt Helsley 已提交
706
			insn->call_dest = reloc->sym;
707 708 709 710 711 712 713 714 715 716 717 718 719 720 721 722 723 724 725
			continue;
		}

		insn->jump_dest = find_insn(file, dest_sec, dest_off);
		if (!insn->jump_dest) {

			/*
			 * This is a special case where an alt instruction
			 * jumps past the end of the section.  These are
			 * handled later in handle_group_alt().
			 */
			if (!strcmp(insn->sec->name, ".altinstr_replacement"))
				continue;

			WARN_FUNC("can't find jump dest instruction at %s+0x%lx",
				  insn->sec, insn->offset, dest_sec->name,
				  dest_off);
			return -1;
		}
726 727

		/*
728
		 * Cross-function jump.
729 730
		 */
		if (insn->func && insn->jump_dest->func &&
731 732 733 734 735 736 737 738 739 740 741 742 743 744 745
		    insn->func != insn->jump_dest->func) {

			/*
			 * For GCC 8+, create parent/child links for any cold
			 * subfunctions.  This is _mostly_ redundant with a
			 * similar initialization in read_symbols().
			 *
			 * If a function has aliases, we want the *first* such
			 * function in the symbol table to be the subfunction's
			 * parent.  In that case we overwrite the
			 * initialization done in read_symbols().
			 *
			 * However this code can't completely replace the
			 * read_symbols() code because this doesn't detect the
			 * case where the parent function's only reference to a
746
			 * subfunction is through a jump table.
747 748 749 750 751 752 753 754 755
			 */
			if (!strstr(insn->func->name, ".cold.") &&
			    strstr(insn->jump_dest->func->name, ".cold.")) {
				insn->func->cfunc = insn->jump_dest->func;
				insn->jump_dest->func->pfunc = insn->func;

			} else if (insn->jump_dest->func->pfunc != insn->func->pfunc &&
				   insn->jump_dest->offset == insn->jump_dest->func->offset) {

756
				/* internal sibling call */
757 758
				insn->call_dest = insn->jump_dest->func;
			}
759
		}
760 761 762 763 764
	}

	return 0;
}

765 766 767 768 769 770 771 772 773 774
static void remove_insn_ops(struct instruction *insn)
{
	struct stack_op *op, *tmp;

	list_for_each_entry_safe(op, tmp, &insn->stack_ops, list) {
		list_del(&op->list);
		free(op);
	}
}

775 776 777 778 779 780 781
/*
 * Find the destination instructions for all calls.
 */
static int add_call_destinations(struct objtool_file *file)
{
	struct instruction *insn;
	unsigned long dest_off;
M
Matt Helsley 已提交
782
	struct reloc *reloc;
783 784 785 786 787

	for_each_insn(file, insn) {
		if (insn->type != INSN_CALL)
			continue;

M
Matt Helsley 已提交
788
		reloc = find_reloc_by_dest_range(file->elf, insn->sec,
789
					       insn->offset, insn->len);
M
Matt Helsley 已提交
790
		if (!reloc) {
791
			dest_off = arch_jump_destination(insn);
792 793 794
			insn->call_dest = find_func_by_offset(insn->sec, dest_off);
			if (!insn->call_dest)
				insn->call_dest = find_symbol_by_offset(insn->sec, dest_off);
795

796 797 798 799
			if (insn->ignore)
				continue;

			if (!insn->call_dest) {
800
				WARN_FUNC("unannotated intra-function call", insn->sec, insn->offset);
801 802
				return -1;
			}
803

804 805 806 807 808 809
			if (insn->func && insn->call_dest->type != STT_FUNC) {
				WARN_FUNC("unsupported call to non-function",
					  insn->sec, insn->offset);
				return -1;
			}

M
Matt Helsley 已提交
810 811 812
		} else if (reloc->sym->type == STT_SECTION) {
			dest_off = arch_dest_reloc_offset(reloc->addend);
			insn->call_dest = find_func_by_offset(reloc->sym->sec,
813
							      dest_off);
814
			if (!insn->call_dest) {
815
				WARN_FUNC("can't find call dest symbol at %s+0x%lx",
816
					  insn->sec, insn->offset,
M
Matt Helsley 已提交
817
					  reloc->sym->sec->name,
818
					  dest_off);
819 820 821
				return -1;
			}
		} else
M
Matt Helsley 已提交
822
			insn->call_dest = reloc->sym;
823

P
Peter Zijlstra 已提交
824 825 826 827 828 829 830
		/*
		 * Many compilers cannot disable KCOV with a function attribute
		 * so they need a little help, NOP out any KCOV calls from noinstr
		 * text.
		 */
		if (insn->sec->noinstr &&
		    !strncmp(insn->call_dest->name, "__sanitizer_cov_", 16)) {
831 832 833
			if (reloc) {
				reloc->type = R_NONE;
				elf_write_reloc(file->elf, reloc);
P
Peter Zijlstra 已提交
834 835 836 837 838 839 840 841
			}

			elf_write_insn(file->elf, insn->sec,
				       insn->offset, insn->len,
				       arch_nop_insn(insn->len));
			insn->type = INSN_NOP;
		}

842 843 844 845 846 847 848 849
		/*
		 * Whatever stack impact regular CALLs have, should be undone
		 * by the RETURN of the called function.
		 *
		 * Annotated intra-function calls retain the stack_ops but
		 * are converted to JUMP, see read_intra_function_calls().
		 */
		remove_insn_ops(insn);
850 851 852 853 854 855 856 857 858 859 860 861 862 863 864 865 866 867 868 869 870 871 872 873 874 875 876
	}

	return 0;
}

/*
 * The .alternatives section requires some extra special care, over and above
 * what other special sections require:
 *
 * 1. Because alternatives are patched in-place, we need to insert a fake jump
 *    instruction at the end so that validate_branch() skips all the original
 *    replaced instructions when validating the new instruction path.
 *
 * 2. An added wrinkle is that the new instruction length might be zero.  In
 *    that case the old instructions are replaced with noops.  We simulate that
 *    by creating a fake jump as the only new instruction.
 *
 * 3. In some cases, the alternative section includes an instruction which
 *    conditionally jumps to the _end_ of the entry.  We have to modify these
 *    jumps' destinations to point back to .text rather than the end of the
 *    entry in .altinstr_replacement.
 */
static int handle_group_alt(struct objtool_file *file,
			    struct special_alt *special_alt,
			    struct instruction *orig_insn,
			    struct instruction **new_insn)
{
877
	static unsigned int alt_group_next_index = 1;
878
	struct instruction *last_orig_insn, *last_new_insn, *insn, *fake_jump = NULL;
879
	unsigned int alt_group = alt_group_next_index++;
880 881 882 883 884 885 886 887
	unsigned long dest_off;

	last_orig_insn = NULL;
	insn = orig_insn;
	sec_for_each_insn_from(file, insn) {
		if (insn->offset >= special_alt->orig_off + special_alt->orig_len)
			break;

888
		insn->alt_group = alt_group;
889 890 891
		last_orig_insn = insn;
	}

892 893 894 895 896 897 898 899
	if (next_insn_same_sec(file, last_orig_insn)) {
		fake_jump = malloc(sizeof(*fake_jump));
		if (!fake_jump) {
			WARN("malloc failed");
			return -1;
		}
		memset(fake_jump, 0, sizeof(*fake_jump));
		INIT_LIST_HEAD(&fake_jump->alts);
900
		INIT_LIST_HEAD(&fake_jump->stack_ops);
901
		init_cfi_state(&fake_jump->cfi);
902 903

		fake_jump->sec = special_alt->new_sec;
904
		fake_jump->offset = FAKE_JUMP_OFFSET;
905 906
		fake_jump->type = INSN_JUMP_UNCONDITIONAL;
		fake_jump->jump_dest = list_next_entry(last_orig_insn, list);
907
		fake_jump->func = orig_insn->func;
908 909 910
	}

	if (!special_alt->new_len) {
911 912 913 914 915 916
		if (!fake_jump) {
			WARN("%s: empty alternative at end of section",
			     special_alt->orig_sec->name);
			return -1;
		}

917 918 919 920 921
		*new_insn = fake_jump;
		return 0;
	}

	last_new_insn = NULL;
922
	alt_group = alt_group_next_index++;
923 924 925 926 927 928 929
	insn = *new_insn;
	sec_for_each_insn_from(file, insn) {
		if (insn->offset >= special_alt->new_off + special_alt->new_len)
			break;

		last_new_insn = insn;

930
		insn->ignore = orig_insn->ignore_alts;
931
		insn->func = orig_insn->func;
932
		insn->alt_group = alt_group;
933

934 935 936 937 938 939 940 941 942 943 944 945 946 947
		/*
		 * Since alternative replacement code is copy/pasted by the
		 * kernel after applying relocations, generally such code can't
		 * have relative-address relocation references to outside the
		 * .altinstr_replacement section, unless the arch's
		 * alternatives code can adjust the relative offsets
		 * accordingly.
		 *
		 * The x86 alternatives code adjusts the offsets only when it
		 * encounters a branch instruction at the very beginning of the
		 * replacement group.
		 */
		if ((insn->offset != special_alt->new_off ||
		    (insn->type != INSN_CALL && !is_static_jump(insn))) &&
M
Matt Helsley 已提交
948
		    find_reloc_by_dest_range(file->elf, insn->sec, insn->offset, insn->len)) {
949 950 951 952 953 954

			WARN_FUNC("unsupported relocation in alternatives section",
				  insn->sec, insn->offset);
			return -1;
		}

955
		if (!is_static_jump(insn))
956 957 958 959 960
			continue;

		if (!insn->immediate)
			continue;

961
		dest_off = arch_jump_destination(insn);
962 963 964 965 966 967
		if (dest_off == special_alt->new_off + special_alt->new_len) {
			if (!fake_jump) {
				WARN("%s: alternative jump to end of section",
				     special_alt->orig_sec->name);
				return -1;
			}
968
			insn->jump_dest = fake_jump;
969
		}
970 971 972 973 974 975 976 977 978 979 980 981 982 983

		if (!insn->jump_dest) {
			WARN_FUNC("can't find alternative jump destination",
				  insn->sec, insn->offset);
			return -1;
		}
	}

	if (!last_new_insn) {
		WARN_FUNC("can't find last new alternative instruction",
			  special_alt->new_sec, special_alt->new_off);
		return -1;
	}

984 985
	if (fake_jump)
		list_add(&fake_jump->list, &last_new_insn->list);
986 987 988 989 990 991 992 993 994 995 996 997 998 999 1000 1001 1002 1003 1004 1005 1006 1007 1008 1009 1010 1011 1012 1013 1014 1015 1016 1017 1018 1019 1020 1021 1022 1023 1024 1025 1026 1027 1028 1029 1030 1031 1032 1033 1034 1035 1036 1037 1038 1039 1040 1041 1042 1043 1044 1045 1046 1047 1048 1049 1050 1051 1052 1053 1054 1055

	return 0;
}

/*
 * A jump table entry can either convert a nop to a jump or a jump to a nop.
 * If the original instruction is a jump, make the alt entry an effective nop
 * by just skipping the original instruction.
 */
static int handle_jump_alt(struct objtool_file *file,
			   struct special_alt *special_alt,
			   struct instruction *orig_insn,
			   struct instruction **new_insn)
{
	if (orig_insn->type == INSN_NOP)
		return 0;

	if (orig_insn->type != INSN_JUMP_UNCONDITIONAL) {
		WARN_FUNC("unsupported instruction at jump label",
			  orig_insn->sec, orig_insn->offset);
		return -1;
	}

	*new_insn = list_next_entry(orig_insn, list);
	return 0;
}

/*
 * Read all the special sections which have alternate instructions which can be
 * patched in or redirected to at runtime.  Each instruction having alternate
 * instruction(s) has them added to its insn->alts list, which will be
 * traversed in validate_branch().
 */
static int add_special_section_alts(struct objtool_file *file)
{
	struct list_head special_alts;
	struct instruction *orig_insn, *new_insn;
	struct special_alt *special_alt, *tmp;
	struct alternative *alt;
	int ret;

	ret = special_get_alts(file->elf, &special_alts);
	if (ret)
		return ret;

	list_for_each_entry_safe(special_alt, tmp, &special_alts, list) {

		orig_insn = find_insn(file, special_alt->orig_sec,
				      special_alt->orig_off);
		if (!orig_insn) {
			WARN_FUNC("special: can't find orig instruction",
				  special_alt->orig_sec, special_alt->orig_off);
			ret = -1;
			goto out;
		}

		new_insn = NULL;
		if (!special_alt->group || special_alt->new_len) {
			new_insn = find_insn(file, special_alt->new_sec,
					     special_alt->new_off);
			if (!new_insn) {
				WARN_FUNC("special: can't find new instruction",
					  special_alt->new_sec,
					  special_alt->new_off);
				ret = -1;
				goto out;
			}
		}

		if (special_alt->group) {
1056 1057 1058 1059 1060 1061
			if (!special_alt->orig_len) {
				WARN_FUNC("empty alternative entry",
					  orig_insn->sec, orig_insn->offset);
				continue;
			}

1062 1063 1064 1065 1066 1067 1068 1069 1070 1071 1072
			ret = handle_group_alt(file, special_alt, orig_insn,
					       &new_insn);
			if (ret)
				goto out;
		} else if (special_alt->jump_or_nop) {
			ret = handle_jump_alt(file, special_alt, orig_insn,
					      &new_insn);
			if (ret)
				goto out;
		}

1073 1074 1075 1076 1077 1078 1079
		alt = malloc(sizeof(*alt));
		if (!alt) {
			WARN("malloc failed");
			ret = -1;
			goto out;
		}

1080
		alt->insn = new_insn;
P
Peter Zijlstra 已提交
1081
		alt->skip_orig = special_alt->skip_orig;
P
Peter Zijlstra 已提交
1082
		orig_insn->ignore_alts |= special_alt->skip_alt;
1083 1084 1085 1086 1087 1088 1089 1090 1091 1092
		list_add_tail(&alt->list, &orig_insn->alts);

		list_del(&special_alt->list);
		free(special_alt);
	}

out:
	return ret;
}

1093
static int add_jump_table(struct objtool_file *file, struct instruction *insn,
M
Matt Helsley 已提交
1094
			    struct reloc *table)
1095
{
M
Matt Helsley 已提交
1096
	struct reloc *reloc = table;
1097
	struct instruction *dest_insn;
1098
	struct alternative *alt;
1099 1100
	struct symbol *pfunc = insn->func->pfunc;
	unsigned int prev_offset = 0;
1101

1102
	/*
M
Matt Helsley 已提交
1103
	 * Each @reloc is a switch table relocation which points to the target
1104 1105
	 * instruction.
	 */
M
Matt Helsley 已提交
1106
	list_for_each_entry_from(reloc, &table->sec->reloc_list, list) {
1107 1108

		/* Check for the end of the table: */
M
Matt Helsley 已提交
1109
		if (reloc != table && reloc->jump_table_start)
1110 1111
			break;

1112
		/* Make sure the table entries are consecutive: */
M
Matt Helsley 已提交
1113
		if (prev_offset && reloc->offset != prev_offset + 8)
1114 1115 1116
			break;

		/* Detect function pointers from contiguous objects: */
M
Matt Helsley 已提交
1117 1118
		if (reloc->sym->sec == pfunc->sec &&
		    reloc->addend == pfunc->offset)
1119 1120
			break;

M
Matt Helsley 已提交
1121
		dest_insn = find_insn(file, reloc->sym->sec, reloc->addend);
1122
		if (!dest_insn)
1123 1124
			break;

1125
		/* Make sure the destination is in the same function: */
1126
		if (!dest_insn->func || dest_insn->func->pfunc != pfunc)
1127
			break;
1128 1129 1130 1131 1132 1133 1134

		alt = malloc(sizeof(*alt));
		if (!alt) {
			WARN("malloc failed");
			return -1;
		}

1135
		alt->insn = dest_insn;
1136
		list_add_tail(&alt->list, &insn->alts);
M
Matt Helsley 已提交
1137
		prev_offset = reloc->offset;
1138 1139 1140 1141 1142 1143
	}

	if (!prev_offset) {
		WARN_FUNC("can't find switch jump table",
			  insn->sec, insn->offset);
		return -1;
1144 1145 1146 1147 1148 1149
	}

	return 0;
}

/*
1150
 * find_jump_table() - Given a dynamic jump, find the switch jump table in
1151 1152 1153 1154 1155 1156 1157 1158 1159 1160 1161 1162 1163 1164 1165 1166 1167 1168 1169 1170 1171 1172 1173 1174 1175 1176 1177 1178 1179 1180 1181 1182
 * .rodata associated with it.
 *
 * There are 3 basic patterns:
 *
 * 1. jmpq *[rodata addr](,%reg,8)
 *
 *    This is the most common case by far.  It jumps to an address in a simple
 *    jump table which is stored in .rodata.
 *
 * 2. jmpq *[rodata addr](%rip)
 *
 *    This is caused by a rare GCC quirk, currently only seen in three driver
 *    functions in the kernel, only with certain obscure non-distro configs.
 *
 *    As part of an optimization, GCC makes a copy of an existing switch jump
 *    table, modifies it, and then hard-codes the jump (albeit with an indirect
 *    jump) to use a single entry in the table.  The rest of the jump table and
 *    some of its jump targets remain as dead code.
 *
 *    In such a case we can just crudely ignore all unreachable instruction
 *    warnings for the entire object file.  Ideally we would just ignore them
 *    for the function, but that would require redesigning the code quite a
 *    bit.  And honestly that's just not worth doing: unreachable instruction
 *    warnings are of questionable value anyway, and this is such a rare issue.
 *
 * 3. mov [rodata addr],%reg1
 *    ... some instructions ...
 *    jmpq *(%reg1,%reg2,8)
 *
 *    This is a fairly uncommon pattern which is new for GCC 6.  As of this
 *    writing, there are 11 occurrences of it in the allmodconfig kernel.
 *
1183 1184 1185 1186
 *    As of GCC 7 there are quite a few more of these and the 'in between' code
 *    is significant. Esp. with KASAN enabled some of the code between the mov
 *    and jmpq uses .rodata itself, which can confuse things.
 *
1187 1188
 *    TODO: Once we have DWARF CFI and smarter instruction decoding logic,
 *    ensure the same register is used in the mov and jump instructions.
1189 1190
 *
 *    NOTE: RETPOLINE made it harder still to decode dynamic jumps.
1191
 */
M
Matt Helsley 已提交
1192
static struct reloc *find_jump_table(struct objtool_file *file,
1193 1194 1195
				      struct symbol *func,
				      struct instruction *insn)
{
M
Matt Helsley 已提交
1196
	struct reloc *text_reloc, *table_reloc;
1197
	struct instruction *dest_insn, *orig_insn = insn;
1198
	struct section *table_sec;
1199
	unsigned long table_offset;
1200

1201 1202 1203 1204 1205
	/*
	 * Backward search using the @first_jump_src links, these help avoid
	 * much of the 'in between' code. Which avoids us getting confused by
	 * it.
	 */
1206
	for (;
1207 1208
	     insn && insn->func && insn->func->pfunc == func;
	     insn = insn->first_jump_src ?: prev_insn_same_sym(file, insn)) {
1209

1210
		if (insn != orig_insn && insn->type == INSN_JUMP_DYNAMIC)
1211 1212 1213 1214 1215 1216 1217 1218 1219 1220
			break;

		/* allow small jumps within the range */
		if (insn->type == INSN_JUMP_UNCONDITIONAL &&
		    insn->jump_dest &&
		    (insn->jump_dest->offset <= insn->offset ||
		     insn->jump_dest->offset > orig_insn->offset))
		    break;

		/* look for a relocation which references .rodata */
M
Matt Helsley 已提交
1221
		text_reloc = find_reloc_by_dest_range(file->elf, insn->sec,
1222
						    insn->offset, insn->len);
M
Matt Helsley 已提交
1223 1224
		if (!text_reloc || text_reloc->sym->type != STT_SECTION ||
		    !text_reloc->sym->sec->rodata)
1225 1226
			continue;

M
Matt Helsley 已提交
1227 1228
		table_offset = text_reloc->addend;
		table_sec = text_reloc->sym->sec;
1229

M
Matt Helsley 已提交
1230
		if (text_reloc->type == R_X86_64_PC32)
1231 1232
			table_offset += 4;

1233 1234
		/*
		 * Make sure the .rodata address isn't associated with a
1235 1236 1237 1238 1239 1240
		 * symbol.  GCC jump tables are anonymous data.
		 *
		 * Also support C jump tables which are in the same format as
		 * switch jump tables.  For objtool to recognize them, they
		 * need to be placed in the C_JUMP_TABLE_SECTION section.  They
		 * have symbols associated with them.
1241
		 */
1242 1243
		if (find_symbol_containing(table_sec, table_offset) &&
		    strcmp(table_sec->name, C_JUMP_TABLE_SECTION))
1244 1245
			continue;

1246
		/*
M
Matt Helsley 已提交
1247
		 * Each table entry has a reloc associated with it.  The reloc
1248 1249 1250
		 * should reference text in the same function as the original
		 * instruction.
		 */
M
Matt Helsley 已提交
1251 1252
		table_reloc = find_reloc_by_dest(file->elf, table_sec, table_offset);
		if (!table_reloc)
1253
			continue;
M
Matt Helsley 已提交
1254
		dest_insn = find_insn(file, table_reloc->sym->sec, table_reloc->addend);
1255 1256
		if (!dest_insn || !dest_insn->func || dest_insn->func->pfunc != func)
			continue;
1257

1258 1259 1260 1261 1262
		/*
		 * Use of RIP-relative switch jumps is quite rare, and
		 * indicates a rare GCC quirk/bug which can leave dead code
		 * behind.
		 */
M
Matt Helsley 已提交
1263
		if (text_reloc->type == R_X86_64_PC32)
1264 1265
			file->ignore_unreachables = true;

M
Matt Helsley 已提交
1266
		return table_reloc;
1267 1268 1269 1270 1271
	}

	return NULL;
}

1272 1273 1274 1275 1276 1277
/*
 * First pass: Mark the head of each jump table so that in the next pass,
 * we know when a given jump table ends and the next one starts.
 */
static void mark_func_jump_tables(struct objtool_file *file,
				    struct symbol *func)
1278
{
1279
	struct instruction *insn, *last = NULL;
M
Matt Helsley 已提交
1280
	struct reloc *reloc;
1281

1282
	func_for_each_insn(file, func, insn) {
1283 1284 1285 1286 1287
		if (!last)
			last = insn;

		/*
		 * Store back-pointers for unconditional forward jumps such
1288
		 * that find_jump_table() can back-track using those and
1289 1290 1291 1292 1293 1294 1295 1296 1297 1298 1299
		 * avoid some potentially confusing code.
		 */
		if (insn->type == INSN_JUMP_UNCONDITIONAL && insn->jump_dest &&
		    insn->offset > last->offset &&
		    insn->jump_dest->offset > insn->offset &&
		    !insn->jump_dest->first_jump_src) {

			insn->jump_dest->first_jump_src = insn;
			last = insn->jump_dest;
		}

1300 1301 1302
		if (insn->type != INSN_JUMP_DYNAMIC)
			continue;

M
Matt Helsley 已提交
1303 1304 1305 1306
		reloc = find_jump_table(file, func, insn);
		if (reloc) {
			reloc->jump_table_start = true;
			insn->jump_table = reloc;
1307 1308
		}
	}
1309 1310 1311 1312 1313 1314 1315 1316
}

static int add_func_jump_tables(struct objtool_file *file,
				  struct symbol *func)
{
	struct instruction *insn;
	int ret;

1317
	func_for_each_insn(file, func, insn) {
1318 1319
		if (!insn->jump_table)
			continue;
1320

1321
		ret = add_jump_table(file, insn, insn->jump_table);
1322 1323 1324 1325 1326 1327 1328 1329 1330 1331 1332 1333
		if (ret)
			return ret;
	}

	return 0;
}

/*
 * For some switch statements, gcc generates a jump table in the .rodata
 * section which contains a list of addresses within the function to jump to.
 * This finds these jump tables and adds them to the insn->alts lists.
 */
1334
static int add_jump_table_alts(struct objtool_file *file)
1335 1336 1337 1338 1339
{
	struct section *sec;
	struct symbol *func;
	int ret;

1340
	if (!file->rodata)
1341 1342
		return 0;

1343
	for_each_sec(file, sec) {
1344 1345 1346 1347
		list_for_each_entry(func, &sec->symbol_list, list) {
			if (func->type != STT_FUNC)
				continue;

1348
			mark_func_jump_tables(file, func);
1349
			ret = add_func_jump_tables(file, func);
1350 1351 1352 1353 1354 1355 1356 1357
			if (ret)
				return ret;
		}
	}

	return 0;
}

1358 1359
static int read_unwind_hints(struct objtool_file *file)
{
M
Matt Helsley 已提交
1360 1361
	struct section *sec, *relocsec;
	struct reloc *reloc;
1362 1363 1364 1365 1366 1367 1368 1369 1370
	struct unwind_hint *hint;
	struct instruction *insn;
	struct cfi_reg *cfa;
	int i;

	sec = find_section_by_name(file->elf, ".discard.unwind_hints");
	if (!sec)
		return 0;

M
Matt Helsley 已提交
1371 1372
	relocsec = sec->reloc;
	if (!relocsec) {
1373 1374 1375 1376 1377 1378 1379 1380 1381 1382 1383 1384 1385 1386
		WARN("missing .rela.discard.unwind_hints section");
		return -1;
	}

	if (sec->len % sizeof(struct unwind_hint)) {
		WARN("struct unwind_hint size mismatch");
		return -1;
	}

	file->hints = true;

	for (i = 0; i < sec->len / sizeof(struct unwind_hint); i++) {
		hint = (struct unwind_hint *)sec->data->d_buf + i;

M
Matt Helsley 已提交
1387 1388 1389
		reloc = find_reloc_by_dest(file->elf, sec, i * sizeof(*hint));
		if (!reloc) {
			WARN("can't find reloc for unwind_hints[%d]", i);
1390 1391 1392
			return -1;
		}

M
Matt Helsley 已提交
1393
		insn = find_insn(file, reloc->sym->sec, reloc->addend);
1394 1395 1396 1397 1398
		if (!insn) {
			WARN("can't find insn for unwind_hints[%d]", i);
			return -1;
		}

1399
		cfa = &insn->cfi.cfa;
1400

1401
		if (hint->type == UNWIND_HINT_TYPE_RET_OFFSET) {
1402
			insn->ret_offset = hint->sp_offset;
1403 1404 1405 1406 1407 1408 1409 1410 1411 1412 1413 1414 1415 1416 1417 1418 1419 1420 1421 1422 1423 1424 1425 1426 1427 1428 1429 1430 1431 1432 1433 1434 1435 1436 1437 1438 1439
			continue;
		}

		insn->hint = true;

		switch (hint->sp_reg) {
		case ORC_REG_UNDEFINED:
			cfa->base = CFI_UNDEFINED;
			break;
		case ORC_REG_SP:
			cfa->base = CFI_SP;
			break;
		case ORC_REG_BP:
			cfa->base = CFI_BP;
			break;
		case ORC_REG_SP_INDIRECT:
			cfa->base = CFI_SP_INDIRECT;
			break;
		case ORC_REG_R10:
			cfa->base = CFI_R10;
			break;
		case ORC_REG_R13:
			cfa->base = CFI_R13;
			break;
		case ORC_REG_DI:
			cfa->base = CFI_DI;
			break;
		case ORC_REG_DX:
			cfa->base = CFI_DX;
			break;
		default:
			WARN_FUNC("unsupported unwind_hint sp base reg %d",
				  insn->sec, insn->offset, hint->sp_reg);
			return -1;
		}

		cfa->offset = hint->sp_offset;
1440 1441
		insn->cfi.type = hint->type;
		insn->cfi.end = hint->end;
1442 1443 1444 1445 1446
	}

	return 0;
}

1447 1448
static int read_retpoline_hints(struct objtool_file *file)
{
J
Josh Poimboeuf 已提交
1449
	struct section *sec;
1450
	struct instruction *insn;
M
Matt Helsley 已提交
1451
	struct reloc *reloc;
1452

J
Josh Poimboeuf 已提交
1453
	sec = find_section_by_name(file->elf, ".rela.discard.retpoline_safe");
1454 1455 1456
	if (!sec)
		return 0;

M
Matt Helsley 已提交
1457 1458
	list_for_each_entry(reloc, &sec->reloc_list, list) {
		if (reloc->sym->type != STT_SECTION) {
J
Josh Poimboeuf 已提交
1459
			WARN("unexpected relocation symbol type in %s", sec->name);
1460 1461 1462
			return -1;
		}

M
Matt Helsley 已提交
1463
		insn = find_insn(file, reloc->sym->sec, reloc->addend);
1464
		if (!insn) {
J
Josh Poimboeuf 已提交
1465
			WARN("bad .discard.retpoline_safe entry");
1466 1467 1468 1469 1470
			return -1;
		}

		if (insn->type != INSN_JUMP_DYNAMIC &&
		    insn->type != INSN_CALL_DYNAMIC) {
J
Josh Poimboeuf 已提交
1471
			WARN_FUNC("retpoline_safe hint not an indirect jump/call",
1472 1473 1474 1475 1476 1477 1478 1479 1480 1481
				  insn->sec, insn->offset);
			return -1;
		}

		insn->retpoline_safe = true;
	}

	return 0;
}

1482 1483 1484 1485
static int read_instr_hints(struct objtool_file *file)
{
	struct section *sec;
	struct instruction *insn;
M
Matt Helsley 已提交
1486
	struct reloc *reloc;
1487 1488 1489 1490 1491

	sec = find_section_by_name(file->elf, ".rela.discard.instr_end");
	if (!sec)
		return 0;

M
Matt Helsley 已提交
1492 1493
	list_for_each_entry(reloc, &sec->reloc_list, list) {
		if (reloc->sym->type != STT_SECTION) {
1494 1495 1496 1497
			WARN("unexpected relocation symbol type in %s", sec->name);
			return -1;
		}

M
Matt Helsley 已提交
1498
		insn = find_insn(file, reloc->sym->sec, reloc->addend);
1499 1500 1501 1502 1503 1504 1505 1506 1507 1508 1509 1510
		if (!insn) {
			WARN("bad .discard.instr_end entry");
			return -1;
		}

		insn->instr--;
	}

	sec = find_section_by_name(file->elf, ".rela.discard.instr_begin");
	if (!sec)
		return 0;

M
Matt Helsley 已提交
1511 1512
	list_for_each_entry(reloc, &sec->reloc_list, list) {
		if (reloc->sym->type != STT_SECTION) {
1513 1514 1515 1516
			WARN("unexpected relocation symbol type in %s", sec->name);
			return -1;
		}

M
Matt Helsley 已提交
1517
		insn = find_insn(file, reloc->sym->sec, reloc->addend);
1518 1519 1520 1521 1522 1523 1524 1525 1526 1527 1528
		if (!insn) {
			WARN("bad .discard.instr_begin entry");
			return -1;
		}

		insn->instr++;
	}

	return 0;
}

1529 1530 1531 1532
static int read_intra_function_calls(struct objtool_file *file)
{
	struct instruction *insn;
	struct section *sec;
M
Matt Helsley 已提交
1533
	struct reloc *reloc;
1534 1535 1536 1537 1538

	sec = find_section_by_name(file->elf, ".rela.discard.intra_function_calls");
	if (!sec)
		return 0;

M
Matt Helsley 已提交
1539
	list_for_each_entry(reloc, &sec->reloc_list, list) {
1540 1541
		unsigned long dest_off;

M
Matt Helsley 已提交
1542
		if (reloc->sym->type != STT_SECTION) {
1543 1544 1545 1546 1547
			WARN("unexpected relocation symbol type in %s",
			     sec->name);
			return -1;
		}

M
Matt Helsley 已提交
1548
		insn = find_insn(file, reloc->sym->sec, reloc->addend);
1549 1550 1551 1552 1553 1554 1555 1556 1557 1558 1559 1560 1561 1562 1563 1564 1565 1566 1567 1568 1569 1570 1571 1572 1573 1574 1575 1576 1577 1578 1579
		if (!insn) {
			WARN("bad .discard.intra_function_call entry");
			return -1;
		}

		if (insn->type != INSN_CALL) {
			WARN_FUNC("intra_function_call not a direct call",
				  insn->sec, insn->offset);
			return -1;
		}

		/*
		 * Treat intra-function CALLs as JMPs, but with a stack_op.
		 * See add_call_destinations(), which strips stack_ops from
		 * normal CALLs.
		 */
		insn->type = INSN_JUMP_UNCONDITIONAL;

		dest_off = insn->offset + insn->len + insn->immediate;
		insn->jump_dest = find_insn(file, insn->sec, dest_off);
		if (!insn->jump_dest) {
			WARN_FUNC("can't find call dest at %s+0x%lx",
				  insn->sec, insn->offset,
				  insn->sec->name, dest_off);
			return -1;
		}
	}

	return 0;
}

1580 1581 1582 1583 1584 1585
static void mark_rodata(struct objtool_file *file)
{
	struct section *sec;
	bool found = false;

	/*
1586 1587 1588 1589 1590 1591 1592 1593
	 * Search for the following rodata sections, each of which can
	 * potentially contain jump tables:
	 *
	 * - .rodata: can contain GCC switch tables
	 * - .rodata.<func>: same, if -fdata-sections is being used
	 * - .rodata..c_jump_table: contains C annotated jump tables
	 *
	 * .rodata.str1.* sections are ignored; they don't contain jump tables.
1594 1595
	 */
	for_each_sec(file, sec) {
1596 1597
		if (!strncmp(sec->name, ".rodata", 7) &&
		    !strstr(sec->name, ".str1.")) {
1598 1599 1600 1601 1602 1603 1604 1605
			sec->rodata = true;
			found = true;
		}
	}

	file->rodata = found;
}

1606 1607 1608 1609
static int decode_sections(struct objtool_file *file)
{
	int ret;

1610 1611
	mark_rodata(file);

1612 1613 1614 1615 1616 1617 1618 1619 1620
	ret = decode_instructions(file);
	if (ret)
		return ret;

	ret = add_dead_ends(file);
	if (ret)
		return ret;

	add_ignores(file);
P
Peter Zijlstra 已提交
1621
	add_uaccess_safe(file);
1622

1623
	ret = add_ignore_alternatives(file);
1624 1625 1626
	if (ret)
		return ret;

1627 1628 1629 1630
	ret = add_jump_destinations(file);
	if (ret)
		return ret;

1631
	ret = add_special_section_alts(file);
1632 1633 1634
	if (ret)
		return ret;

1635 1636 1637 1638
	ret = read_intra_function_calls(file);
	if (ret)
		return ret;

1639
	ret = add_call_destinations(file);
1640 1641 1642
	if (ret)
		return ret;

1643
	ret = add_jump_table_alts(file);
1644 1645 1646
	if (ret)
		return ret;

1647 1648 1649 1650
	ret = read_unwind_hints(file);
	if (ret)
		return ret;

1651 1652 1653 1654
	ret = read_retpoline_hints(file);
	if (ret)
		return ret;

1655 1656 1657 1658
	ret = read_instr_hints(file);
	if (ret)
		return ret;

1659 1660 1661 1662 1663
	return 0;
}

static bool is_fentry_call(struct instruction *insn)
{
1664
	if (insn->type == INSN_CALL && insn->call_dest &&
1665 1666 1667 1668 1669 1670 1671
	    insn->call_dest->type == STT_NOTYPE &&
	    !strcmp(insn->call_dest->name, "__fentry__"))
		return true;

	return false;
}

1672
static bool has_modified_stack_frame(struct instruction *insn, struct insn_state *state)
1673
{
1674
	u8 ret_offset = insn->ret_offset;
1675
	struct cfi_state *cfi = &state->cfi;
1676 1677
	int i;

1678
	if (cfi->cfa.base != initial_func_cfi.cfa.base || cfi->drap)
1679 1680
		return true;

1681
	if (cfi->cfa.offset != initial_func_cfi.cfa.offset + ret_offset)
1682 1683
		return true;

1684
	if (cfi->stack_size != initial_func_cfi.cfa.offset + ret_offset)
1685 1686
		return true;

1687 1688 1689 1690 1691 1692 1693 1694
	/*
	 * If there is a ret offset hint then don't check registers
	 * because a callee-saved register might have been pushed on
	 * the stack.
	 */
	if (ret_offset)
		return false;

1695
	for (i = 0; i < CFI_NUM_REGS; i++) {
1696 1697
		if (cfi->regs[i].base != initial_func_cfi.regs[i].base ||
		    cfi->regs[i].offset != initial_func_cfi.regs[i].offset)
1698
			return true;
1699
	}
1700 1701 1702 1703 1704 1705

	return false;
}

static bool has_valid_stack_frame(struct insn_state *state)
{
1706 1707 1708 1709
	struct cfi_state *cfi = &state->cfi;

	if (cfi->cfa.base == CFI_BP && cfi->regs[CFI_BP].base == CFI_CFA &&
	    cfi->regs[CFI_BP].offset == -16)
1710 1711
		return true;

1712
	if (cfi->drap && cfi->regs[CFI_BP].base == CFI_BP)
1713 1714 1715
		return true;

	return false;
1716 1717
}

1718 1719
static int update_cfi_state_regs(struct instruction *insn,
				  struct cfi_state *cfi,
1720
				  struct stack_op *op)
1721
{
1722
	struct cfi_reg *cfa = &cfi->cfa;
1723

1724
	if (cfa->base != CFI_SP && cfa->base != CFI_SP_INDIRECT)
1725 1726 1727
		return 0;

	/* push */
P
Peter Zijlstra 已提交
1728
	if (op->dest.type == OP_DEST_PUSH || op->dest.type == OP_DEST_PUSHF)
1729 1730 1731
		cfa->offset += 8;

	/* pop */
P
Peter Zijlstra 已提交
1732
	if (op->src.type == OP_SRC_POP || op->src.type == OP_SRC_POPF)
1733 1734 1735 1736 1737 1738 1739 1740 1741 1742
		cfa->offset -= 8;

	/* add immediate to sp */
	if (op->dest.type == OP_DEST_REG && op->src.type == OP_SRC_ADD &&
	    op->dest.reg == CFI_SP && op->src.reg == CFI_SP)
		cfa->offset -= op->src.offset;

	return 0;
}

1743
static void save_reg(struct cfi_state *cfi, unsigned char reg, int base, int offset)
1744
{
1745
	if (arch_callee_saved_reg(reg) &&
1746 1747 1748
	    cfi->regs[reg].base == CFI_UNDEFINED) {
		cfi->regs[reg].base = base;
		cfi->regs[reg].offset = offset;
1749
	}
1750 1751
}

1752
static void restore_reg(struct cfi_state *cfi, unsigned char reg)
1753
{
1754 1755
	cfi->regs[reg].base = initial_func_cfi.regs[reg].base;
	cfi->regs[reg].offset = initial_func_cfi.regs[reg].offset;
1756 1757 1758 1759 1760 1761 1762 1763 1764 1765 1766 1767 1768 1769 1770 1771 1772 1773 1774 1775 1776 1777 1778 1779 1780 1781 1782 1783 1784 1785 1786 1787 1788 1789 1790 1791 1792 1793 1794 1795 1796 1797 1798 1799 1800 1801 1802 1803 1804 1805 1806 1807 1808 1809 1810
}

/*
 * A note about DRAP stack alignment:
 *
 * GCC has the concept of a DRAP register, which is used to help keep track of
 * the stack pointer when aligning the stack.  r10 or r13 is used as the DRAP
 * register.  The typical DRAP pattern is:
 *
 *   4c 8d 54 24 08		lea    0x8(%rsp),%r10
 *   48 83 e4 c0		and    $0xffffffffffffffc0,%rsp
 *   41 ff 72 f8		pushq  -0x8(%r10)
 *   55				push   %rbp
 *   48 89 e5			mov    %rsp,%rbp
 *				(more pushes)
 *   41 52			push   %r10
 *				...
 *   41 5a			pop    %r10
 *				(more pops)
 *   5d				pop    %rbp
 *   49 8d 62 f8		lea    -0x8(%r10),%rsp
 *   c3				retq
 *
 * There are some variations in the epilogues, like:
 *
 *   5b				pop    %rbx
 *   41 5a			pop    %r10
 *   41 5c			pop    %r12
 *   41 5d			pop    %r13
 *   41 5e			pop    %r14
 *   c9				leaveq
 *   49 8d 62 f8		lea    -0x8(%r10),%rsp
 *   c3				retq
 *
 * and:
 *
 *   4c 8b 55 e8		mov    -0x18(%rbp),%r10
 *   48 8b 5d e0		mov    -0x20(%rbp),%rbx
 *   4c 8b 65 f0		mov    -0x10(%rbp),%r12
 *   4c 8b 6d f8		mov    -0x8(%rbp),%r13
 *   c9				leaveq
 *   49 8d 62 f8		lea    -0x8(%r10),%rsp
 *   c3				retq
 *
 * Sometimes r13 is used as the DRAP register, in which case it's saved and
 * restored beforehand:
 *
 *   41 55			push   %r13
 *   4c 8d 6c 24 10		lea    0x10(%rsp),%r13
 *   48 83 e4 f0		and    $0xfffffffffffffff0,%rsp
 *				...
 *   49 8d 65 f0		lea    -0x10(%r13),%rsp
 *   41 5d			pop    %r13
 *   c3				retq
 */
1811
static int update_cfi_state(struct instruction *insn, struct cfi_state *cfi,
1812
			     struct stack_op *op)
1813
{
1814 1815
	struct cfi_reg *cfa = &cfi->cfa;
	struct cfi_reg *regs = cfi->regs;
1816 1817 1818 1819 1820 1821 1822 1823 1824 1825

	/* stack operations don't make sense with an undefined CFA */
	if (cfa->base == CFI_UNDEFINED) {
		if (insn->func) {
			WARN_FUNC("undefined stack state", insn->sec, insn->offset);
			return -1;
		}
		return 0;
	}

1826 1827
	if (cfi->type == ORC_TYPE_REGS || cfi->type == ORC_TYPE_REGS_IRET)
		return update_cfi_state_regs(insn, cfi, op);
1828

1829 1830 1831 1832 1833 1834
	switch (op->dest.type) {

	case OP_DEST_REG:
		switch (op->src.type) {

		case OP_SRC_REG:
1835 1836 1837 1838 1839 1840 1841
			if (op->src.reg == CFI_SP && op->dest.reg == CFI_BP &&
			    cfa->base == CFI_SP &&
			    regs[CFI_BP].base == CFI_CFA &&
			    regs[CFI_BP].offset == -cfa->offset) {

				/* mov %rsp, %rbp */
				cfa->base = op->dest.reg;
1842
				cfi->bp_scratch = false;
1843
			}
1844

1845
			else if (op->src.reg == CFI_SP &&
1846
				 op->dest.reg == CFI_BP && cfi->drap) {
1847

1848 1849
				/* drap: mov %rsp, %rbp */
				regs[CFI_BP].base = CFI_BP;
1850 1851
				regs[CFI_BP].offset = -cfi->stack_size;
				cfi->bp_scratch = false;
1852
			}
1853

1854 1855 1856 1857 1858 1859 1860 1861 1862 1863 1864 1865
			else if (op->src.reg == CFI_SP && cfa->base == CFI_SP) {

				/*
				 * mov %rsp, %reg
				 *
				 * This is needed for the rare case where GCC
				 * does:
				 *
				 *   mov    %rsp, %rax
				 *   ...
				 *   mov    %rax, %rsp
				 */
1866 1867
				cfi->vals[op->dest.reg].base = CFI_CFA;
				cfi->vals[op->dest.reg].offset = -cfi->stack_size;
1868 1869
			}

J
Josh Poimboeuf 已提交
1870 1871 1872 1873 1874 1875 1876 1877
			else if (op->src.reg == CFI_BP && op->dest.reg == CFI_SP &&
				 cfa->base == CFI_BP) {

				/*
				 * mov %rbp, %rsp
				 *
				 * Restore the original stack pointer (Clang).
				 */
1878
				cfi->stack_size = -cfi->regs[CFI_BP].offset;
J
Josh Poimboeuf 已提交
1879 1880
			}

1881 1882 1883 1884
			else if (op->dest.reg == cfa->base) {

				/* mov %reg, %rsp */
				if (cfa->base == CFI_SP &&
1885
				    cfi->vals[op->src.reg].base == CFI_CFA) {
1886 1887 1888 1889 1890 1891 1892 1893 1894

					/*
					 * This is needed for the rare case
					 * where GCC does something dumb like:
					 *
					 *   lea    0x8(%rsp), %rcx
					 *   ...
					 *   mov    %rcx, %rsp
					 */
1895 1896
					cfa->offset = -cfi->vals[op->src.reg].offset;
					cfi->stack_size = cfa->offset;
1897 1898 1899 1900 1901

				} else {
					cfa->base = CFI_UNDEFINED;
					cfa->offset = 0;
				}
1902 1903 1904 1905 1906 1907 1908 1909
			}

			break;

		case OP_SRC_ADD:
			if (op->dest.reg == CFI_SP && op->src.reg == CFI_SP) {

				/* add imm, %rsp */
1910
				cfi->stack_size -= op->src.offset;
1911 1912 1913 1914 1915 1916 1917 1918
				if (cfa->base == CFI_SP)
					cfa->offset -= op->src.offset;
				break;
			}

			if (op->dest.reg == CFI_SP && op->src.reg == CFI_BP) {

				/* lea disp(%rbp), %rsp */
1919
				cfi->stack_size = -(op->src.offset + regs[CFI_BP].offset);
1920 1921 1922
				break;
			}

1923
			if (op->src.reg == CFI_SP && cfa->base == CFI_SP) {
1924 1925

				/* drap: lea disp(%rsp), %drap */
1926
				cfi->drap_reg = op->dest.reg;
1927 1928 1929 1930 1931 1932 1933 1934 1935 1936 1937

				/*
				 * lea disp(%rsp), %reg
				 *
				 * This is needed for the rare case where GCC
				 * does something dumb like:
				 *
				 *   lea    0x8(%rsp), %rcx
				 *   ...
				 *   mov    %rcx, %rsp
				 */
1938 1939 1940
				cfi->vals[op->dest.reg].base = CFI_CFA;
				cfi->vals[op->dest.reg].offset = \
					-cfi->stack_size + op->src.offset;
1941

1942 1943 1944
				break;
			}

1945 1946
			if (cfi->drap && op->dest.reg == CFI_SP &&
			    op->src.reg == cfi->drap_reg) {
1947 1948 1949

				 /* drap: lea disp(%drap), %rsp */
				cfa->base = CFI_SP;
1950 1951 1952
				cfa->offset = cfi->stack_size = -op->src.offset;
				cfi->drap_reg = CFI_UNDEFINED;
				cfi->drap = false;
1953 1954 1955
				break;
			}

1956
			if (op->dest.reg == cfi->cfa.base) {
1957 1958 1959 1960 1961 1962 1963 1964 1965
				WARN_FUNC("unsupported stack register modification",
					  insn->sec, insn->offset);
				return -1;
			}

			break;

		case OP_SRC_AND:
			if (op->dest.reg != CFI_SP ||
1966 1967
			    (cfi->drap_reg != CFI_UNDEFINED && cfa->base != CFI_SP) ||
			    (cfi->drap_reg == CFI_UNDEFINED && cfa->base != CFI_BP)) {
1968 1969 1970 1971 1972
				WARN_FUNC("unsupported stack pointer realignment",
					  insn->sec, insn->offset);
				return -1;
			}

1973
			if (cfi->drap_reg != CFI_UNDEFINED) {
1974
				/* drap: and imm, %rsp */
1975 1976 1977
				cfa->base = cfi->drap_reg;
				cfa->offset = cfi->stack_size = 0;
				cfi->drap = true;
1978 1979 1980 1981 1982 1983 1984 1985 1986 1987
			}

			/*
			 * Older versions of GCC (4.8ish) realign the stack
			 * without DRAP, with a frame pointer.
			 */

			break;

		case OP_SRC_POP:
P
Peter Zijlstra 已提交
1988
		case OP_SRC_POPF:
1989
			if (!cfi->drap && op->dest.reg == cfa->base) {
1990 1991 1992 1993 1994

				/* pop %rbp */
				cfa->base = CFI_SP;
			}

1995 1996 1997
			if (cfi->drap && cfa->base == CFI_BP_INDIRECT &&
			    op->dest.reg == cfi->drap_reg &&
			    cfi->drap_offset == -cfi->stack_size) {
1998

1999
				/* drap: pop %drap */
2000
				cfa->base = cfi->drap_reg;
2001
				cfa->offset = 0;
2002
				cfi->drap_offset = -1;
2003

2004
			} else if (regs[op->dest.reg].offset == -cfi->stack_size) {
2005

2006
				/* pop %reg */
2007
				restore_reg(cfi, op->dest.reg);
2008 2009
			}

2010
			cfi->stack_size -= 8;
2011 2012 2013 2014 2015 2016
			if (cfa->base == CFI_SP)
				cfa->offset -= 8;

			break;

		case OP_SRC_REG_INDIRECT:
2017 2018
			if (cfi->drap && op->src.reg == CFI_BP &&
			    op->src.offset == cfi->drap_offset) {
2019 2020

				/* drap: mov disp(%rbp), %drap */
2021
				cfa->base = cfi->drap_reg;
2022
				cfa->offset = 0;
2023
				cfi->drap_offset = -1;
2024 2025
			}

2026
			if (cfi->drap && op->src.reg == CFI_BP &&
2027 2028 2029
			    op->src.offset == regs[op->dest.reg].offset) {

				/* drap: mov disp(%rbp), %reg */
2030
				restore_reg(cfi, op->dest.reg);
2031 2032 2033 2034 2035 2036

			} else if (op->src.reg == cfa->base &&
			    op->src.offset == regs[op->dest.reg].offset + cfa->offset) {

				/* mov disp(%rbp), %reg */
				/* mov disp(%rsp), %reg */
2037
				restore_reg(cfi, op->dest.reg);
2038 2039 2040 2041 2042 2043 2044 2045 2046 2047 2048 2049 2050
			}

			break;

		default:
			WARN_FUNC("unknown stack-related instruction",
				  insn->sec, insn->offset);
			return -1;
		}

		break;

	case OP_DEST_PUSH:
P
Peter Zijlstra 已提交
2051
	case OP_DEST_PUSHF:
2052
		cfi->stack_size += 8;
2053 2054 2055 2056 2057 2058
		if (cfa->base == CFI_SP)
			cfa->offset += 8;

		if (op->src.type != OP_SRC_REG)
			break;

2059 2060
		if (cfi->drap) {
			if (op->src.reg == cfa->base && op->src.reg == cfi->drap_reg) {
2061 2062 2063

				/* drap: push %drap */
				cfa->base = CFI_BP_INDIRECT;
2064
				cfa->offset = -cfi->stack_size;
2065

2066
				/* save drap so we know when to restore it */
2067
				cfi->drap_offset = -cfi->stack_size;
2068

2069
			} else if (op->src.reg == CFI_BP && cfa->base == cfi->drap_reg) {
2070 2071

				/* drap: push %rbp */
2072
				cfi->stack_size = 0;
2073 2074 2075 2076

			} else if (regs[op->src.reg].base == CFI_UNDEFINED) {

				/* drap: push %reg */
2077
				save_reg(cfi, op->src.reg, CFI_BP, -cfi->stack_size);
2078 2079 2080 2081 2082
			}

		} else {

			/* push %reg */
2083
			save_reg(cfi, op->src.reg, CFI_CFA, -cfi->stack_size);
2084 2085 2086
		}

		/* detect when asm code uses rbp as a scratch register */
2087
		if (!no_fp && insn->func && op->src.reg == CFI_BP &&
2088
		    cfa->base != CFI_BP)
2089
			cfi->bp_scratch = true;
2090 2091 2092 2093
		break;

	case OP_DEST_REG_INDIRECT:

2094 2095
		if (cfi->drap) {
			if (op->src.reg == cfa->base && op->src.reg == cfi->drap_reg) {
2096 2097 2098 2099 2100

				/* drap: mov %drap, disp(%rbp) */
				cfa->base = CFI_BP_INDIRECT;
				cfa->offset = op->dest.offset;

2101
				/* save drap offset so we know when to restore it */
2102
				cfi->drap_offset = op->dest.offset;
2103 2104 2105 2106 2107
			}

			else if (regs[op->src.reg].base == CFI_UNDEFINED) {

				/* drap: mov reg, disp(%rbp) */
2108
				save_reg(cfi, op->src.reg, CFI_BP, op->dest.offset);
2109 2110 2111 2112 2113 2114
			}

		} else if (op->dest.reg == cfa->base) {

			/* mov reg, disp(%rbp) */
			/* mov reg, disp(%rsp) */
2115 2116
			save_reg(cfi, op->src.reg, CFI_CFA,
				 op->dest.offset - cfi->cfa.offset);
2117 2118 2119 2120 2121
		}

		break;

	case OP_DEST_LEAVE:
2122 2123
		if ((!cfi->drap && cfa->base != CFI_BP) ||
		    (cfi->drap && cfa->base != cfi->drap_reg)) {
2124 2125 2126 2127 2128 2129 2130
			WARN_FUNC("leave instruction with modified stack frame",
				  insn->sec, insn->offset);
			return -1;
		}

		/* leave (mov %rbp, %rsp; pop %rbp) */

2131 2132
		cfi->stack_size = -cfi->regs[CFI_BP].offset - 8;
		restore_reg(cfi, CFI_BP);
2133

2134
		if (!cfi->drap) {
2135 2136 2137 2138 2139 2140 2141
			cfa->base = CFI_SP;
			cfa->offset -= 8;
		}

		break;

	case OP_DEST_MEM:
P
Peter Zijlstra 已提交
2142
		if (op->src.type != OP_SRC_POP && op->src.type != OP_SRC_POPF) {
2143 2144 2145 2146 2147 2148
			WARN_FUNC("unknown stack-related memory operation",
				  insn->sec, insn->offset);
			return -1;
		}

		/* pop mem */
2149
		cfi->stack_size -= 8;
2150 2151 2152 2153 2154 2155 2156 2157 2158 2159 2160 2161 2162 2163
		if (cfa->base == CFI_SP)
			cfa->offset -= 8;

		break;

	default:
		WARN_FUNC("unknown stack-related instruction",
			  insn->sec, insn->offset);
		return -1;
	}

	return 0;
}

2164 2165 2166 2167 2168
static int handle_insn_ops(struct instruction *insn, struct insn_state *state)
{
	struct stack_op *op;

	list_for_each_entry(op, &insn->stack_ops, list) {
2169
		struct cfi_state old_cfi = state->cfi;
2170 2171
		int res;

2172
		res = update_cfi_state(insn, &state->cfi, op);
2173 2174 2175
		if (res)
			return res;

2176 2177 2178 2179 2180
		if (insn->alt_group && memcmp(&state->cfi, &old_cfi, sizeof(struct cfi_state))) {
			WARN_FUNC("alternative modifies stack", insn->sec, insn->offset);
			return -1;
		}

2181 2182 2183 2184 2185 2186 2187 2188 2189 2190 2191 2192 2193 2194 2195 2196 2197 2198 2199 2200 2201 2202 2203 2204 2205
		if (op->dest.type == OP_DEST_PUSHF) {
			if (!state->uaccess_stack) {
				state->uaccess_stack = 1;
			} else if (state->uaccess_stack >> 31) {
				WARN_FUNC("PUSHF stack exhausted",
					  insn->sec, insn->offset);
				return 1;
			}
			state->uaccess_stack <<= 1;
			state->uaccess_stack  |= state->uaccess;
		}

		if (op->src.type == OP_SRC_POPF) {
			if (state->uaccess_stack) {
				state->uaccess = state->uaccess_stack & 1;
				state->uaccess_stack >>= 1;
				if (state->uaccess_stack == 1)
					state->uaccess_stack = 0;
			}
		}
	}

	return 0;
}

2206
static bool insn_cfi_match(struct instruction *insn, struct cfi_state *cfi2)
2207
{
2208
	struct cfi_state *cfi1 = &insn->cfi;
2209 2210
	int i;

2211 2212
	if (memcmp(&cfi1->cfa, &cfi2->cfa, sizeof(cfi1->cfa))) {

2213 2214
		WARN_FUNC("stack state mismatch: cfa1=%d%+d cfa2=%d%+d",
			  insn->sec, insn->offset,
2215 2216
			  cfi1->cfa.base, cfi1->cfa.offset,
			  cfi2->cfa.base, cfi2->cfa.offset);
2217

2218
	} else if (memcmp(&cfi1->regs, &cfi2->regs, sizeof(cfi1->regs))) {
2219
		for (i = 0; i < CFI_NUM_REGS; i++) {
2220
			if (!memcmp(&cfi1->regs[i], &cfi2->regs[i],
2221 2222 2223 2224 2225
				    sizeof(struct cfi_reg)))
				continue;

			WARN_FUNC("stack state mismatch: reg1[%d]=%d%+d reg2[%d]=%d%+d",
				  insn->sec, insn->offset,
2226 2227
				  i, cfi1->regs[i].base, cfi1->regs[i].offset,
				  i, cfi2->regs[i].base, cfi2->regs[i].offset);
2228 2229 2230
			break;
		}

2231 2232
	} else if (cfi1->type != cfi2->type) {

2233
		WARN_FUNC("stack state mismatch: type1=%d type2=%d",
2234 2235 2236 2237 2238
			  insn->sec, insn->offset, cfi1->type, cfi2->type);

	} else if (cfi1->drap != cfi2->drap ||
		   (cfi1->drap && cfi1->drap_reg != cfi2->drap_reg) ||
		   (cfi1->drap && cfi1->drap_offset != cfi2->drap_offset)) {
2239

2240
		WARN_FUNC("stack state mismatch: drap1=%d(%d,%d) drap2=%d(%d,%d)",
2241
			  insn->sec, insn->offset,
2242 2243
			  cfi1->drap, cfi1->drap_reg, cfi1->drap_offset,
			  cfi2->drap, cfi2->drap_reg, cfi2->drap_offset);
2244 2245 2246 2247 2248

	} else
		return true;

	return false;
2249 2250
}

P
Peter Zijlstra 已提交
2251 2252 2253
static inline bool func_uaccess_safe(struct symbol *func)
{
	if (func)
2254
		return func->uaccess_safe;
P
Peter Zijlstra 已提交
2255 2256 2257 2258

	return false;
}

2259
static inline const char *call_dest_name(struct instruction *insn)
P
Peter Zijlstra 已提交
2260 2261 2262 2263 2264 2265 2266
{
	if (insn->call_dest)
		return insn->call_dest->name;

	return "{dynamic}";
}

2267 2268 2269 2270 2271 2272 2273 2274 2275 2276 2277 2278 2279 2280 2281 2282 2283 2284 2285 2286 2287 2288 2289 2290 2291 2292
static inline bool noinstr_call_dest(struct symbol *func)
{
	/*
	 * We can't deal with indirect function calls at present;
	 * assume they're instrumented.
	 */
	if (!func)
		return false;

	/*
	 * If the symbol is from a noinstr section; we good.
	 */
	if (func->sec->noinstr)
		return true;

	/*
	 * The __ubsan_handle_*() calls are like WARN(), they only happen when
	 * something 'BAD' happened. At the risk of taking the machine down,
	 * let them proceed to get the message out.
	 */
	if (!strncmp(func->name, "__ubsan_handle_", 15))
		return true;

	return false;
}

P
Peter Zijlstra 已提交
2293 2294
static int validate_call(struct instruction *insn, struct insn_state *state)
{
2295
	if (state->noinstr && state->instr <= 0 &&
2296
	    !noinstr_call_dest(insn->call_dest)) {
2297 2298 2299 2300 2301
		WARN_FUNC("call to %s() leaves .noinstr.text section",
				insn->sec, insn->offset, call_dest_name(insn));
		return 1;
	}

P
Peter Zijlstra 已提交
2302 2303
	if (state->uaccess && !func_uaccess_safe(insn->call_dest)) {
		WARN_FUNC("call to %s() with UACCESS enabled",
2304
				insn->sec, insn->offset, call_dest_name(insn));
P
Peter Zijlstra 已提交
2305 2306 2307
		return 1;
	}

2308 2309
	if (state->df) {
		WARN_FUNC("call to %s() with DF set",
2310
				insn->sec, insn->offset, call_dest_name(insn));
2311 2312 2313
		return 1;
	}

P
Peter Zijlstra 已提交
2314 2315 2316
	return 0;
}

2317 2318
static int validate_sibling_call(struct instruction *insn, struct insn_state *state)
{
2319
	if (has_modified_stack_frame(insn, state)) {
2320 2321 2322 2323 2324
		WARN_FUNC("sibling call from callable instruction with modified stack frame",
				insn->sec, insn->offset);
		return 1;
	}

P
Peter Zijlstra 已提交
2325
	return validate_call(insn, state);
2326 2327
}

2328 2329
static int validate_return(struct symbol *func, struct instruction *insn, struct insn_state *state)
{
2330 2331 2332 2333 2334 2335
	if (state->noinstr && state->instr > 0) {
		WARN_FUNC("return with instrumentation enabled",
			  insn->sec, insn->offset);
		return 1;
	}

2336 2337 2338 2339 2340 2341 2342 2343 2344 2345 2346 2347 2348 2349 2350 2351 2352 2353
	if (state->uaccess && !func_uaccess_safe(func)) {
		WARN_FUNC("return with UACCESS enabled",
			  insn->sec, insn->offset);
		return 1;
	}

	if (!state->uaccess && func_uaccess_safe(func)) {
		WARN_FUNC("return with UACCESS disabled from a UACCESS-safe function",
			  insn->sec, insn->offset);
		return 1;
	}

	if (state->df) {
		WARN_FUNC("return with DF set",
			  insn->sec, insn->offset);
		return 1;
	}

2354
	if (func && has_modified_stack_frame(insn, state)) {
2355 2356 2357 2358 2359
		WARN_FUNC("return with modified stack frame",
			  insn->sec, insn->offset);
		return 1;
	}

2360
	if (state->cfi.bp_scratch) {
2361 2362
		WARN_FUNC("BP used as a scratch register",
			  insn->sec, insn->offset);
2363 2364 2365 2366 2367 2368
		return 1;
	}

	return 0;
}

2369 2370 2371 2372 2373 2374 2375 2376 2377 2378 2379 2380 2381 2382 2383 2384 2385 2386 2387 2388 2389 2390 2391 2392
/*
 * Alternatives should not contain any ORC entries, this in turn means they
 * should not contain any CFI ops, which implies all instructions should have
 * the same same CFI state.
 *
 * It is possible to constuct alternatives that have unreachable holes that go
 * unreported (because they're NOPs), such holes would result in CFI_UNDEFINED
 * states which then results in ORC entries, which we just said we didn't want.
 *
 * Avoid them by copying the CFI entry of the first instruction into the whole
 * alternative.
 */
static void fill_alternative_cfi(struct objtool_file *file, struct instruction *insn)
{
	struct instruction *first_insn = insn;
	int alt_group = insn->alt_group;

	sec_for_each_insn_continue(file, insn) {
		if (insn->alt_group != alt_group)
			break;
		insn->cfi = first_insn->cfi;
	}
}

2393 2394 2395 2396 2397 2398
/*
 * Follow the branch starting at the given instruction, and recursively follow
 * any other branches (jumps).  Meanwhile, track the frame pointer state at
 * each instruction and validate all the rules described in
 * tools/objtool/Documentation/stack-validation.txt.
 */
2399
static int validate_branch(struct objtool_file *file, struct symbol *func,
P
Peter Zijlstra 已提交
2400
			   struct instruction *insn, struct insn_state state)
2401 2402
{
	struct alternative *alt;
P
Peter Zijlstra 已提交
2403
	struct instruction *next_insn;
2404
	struct section *sec;
2405
	u8 visited;
2406 2407 2408 2409 2410
	int ret;

	sec = insn->sec;

	while (1) {
2411 2412
		next_insn = next_insn_same_sec(file, insn);

2413
		if (file->c_file && func && insn->func && func != insn->func->pfunc) {
2414 2415 2416
			WARN("%s() falls through to next function %s()",
			     func->name, insn->func->name);
			return 1;
2417 2418
		}

2419 2420 2421
		if (func && insn->ignore) {
			WARN_FUNC("BUG: why am I validating an ignored function?",
				  sec, insn->offset);
2422
			return 1;
2423 2424
		}

2425
		visited = 1 << state.uaccess;
2426
		if (insn->visited) {
2427
			if (!insn->hint && !insn_cfi_match(insn, &state.cfi))
2428 2429
				return 1;

2430
			if (insn->visited & visited)
P
Peter Zijlstra 已提交
2431
				return 0;
2432 2433
		}

2434 2435 2436
		if (state.noinstr)
			state.instr += insn->instr;

2437
		if (insn->hint)
2438
			state.cfi = insn->cfi;
2439
		else
2440
			insn->cfi = state.cfi;
2441

2442
		insn->visited |= visited;
2443

2444
		if (!insn->ignore_alts && !list_empty(&insn->alts)) {
P
Peter Zijlstra 已提交
2445 2446
			bool skip_orig = false;

2447
			list_for_each_entry(alt, &insn->alts, list) {
P
Peter Zijlstra 已提交
2448 2449 2450
				if (alt->skip_orig)
					skip_orig = true;

2451
				ret = validate_branch(file, func, alt->insn, state);
2452 2453 2454 2455 2456
				if (ret) {
					if (backtrace)
						BT_FUNC("(alt)", insn);
					return ret;
				}
2457
			}
P
Peter Zijlstra 已提交
2458

2459 2460 2461
			if (insn->alt_group)
				fill_alternative_cfi(file, insn);

P
Peter Zijlstra 已提交
2462 2463
			if (skip_orig)
				return 0;
2464 2465
		}

2466 2467 2468
		if (handle_insn_ops(insn, &state))
			return 1;

2469 2470 2471
		switch (insn->type) {

		case INSN_RETURN:
2472
			return validate_return(func, insn, &state);
2473 2474

		case INSN_CALL:
P
Peter Zijlstra 已提交
2475 2476 2477 2478
		case INSN_CALL_DYNAMIC:
			ret = validate_call(insn, &state);
			if (ret)
				return ret;
2479

2480 2481
			if (!no_fp && func && !is_fentry_call(insn) &&
			    !has_valid_stack_frame(&state)) {
2482 2483 2484 2485
				WARN_FUNC("call without frame pointer save/setup",
					  sec, insn->offset);
				return 1;
			}
2486 2487 2488 2489

			if (dead_end_function(file, insn->call_dest))
				return 0;

2490 2491 2492 2493
			break;

		case INSN_JUMP_CONDITIONAL:
		case INSN_JUMP_UNCONDITIONAL:
2494
			if (func && is_sibling_call(insn)) {
2495
				ret = validate_sibling_call(insn, &state);
2496
				if (ret)
2497
					return ret;
2498

2499
			} else if (insn->jump_dest) {
2500 2501
				ret = validate_branch(file, func,
						      insn->jump_dest, state);
2502 2503 2504 2505 2506
				if (ret) {
					if (backtrace)
						BT_FUNC("(branch)", insn);
					return ret;
				}
2507
			}
2508 2509 2510 2511 2512 2513 2514

			if (insn->type == INSN_JUMP_UNCONDITIONAL)
				return 0;

			break;

		case INSN_JUMP_DYNAMIC:
2515
		case INSN_JUMP_DYNAMIC_CONDITIONAL:
2516
			if (func && is_sibling_call(insn)) {
2517 2518 2519
				ret = validate_sibling_call(insn, &state);
				if (ret)
					return ret;
2520 2521
			}

2522 2523 2524 2525
			if (insn->type == INSN_JUMP_DYNAMIC)
				return 0;

			break;
2526

2527 2528 2529 2530 2531 2532 2533 2534
		case INSN_CONTEXT_SWITCH:
			if (func && (!next_insn || !next_insn->hint)) {
				WARN_FUNC("unsupported instruction in callable function",
					  sec, insn->offset);
				return 1;
			}
			return 0;

P
Peter Zijlstra 已提交
2535 2536 2537 2538 2539 2540 2541 2542 2543 2544
		case INSN_STAC:
			if (state.uaccess) {
				WARN_FUNC("recursive UACCESS enable", sec, insn->offset);
				return 1;
			}

			state.uaccess = true;
			break;

		case INSN_CLAC:
2545
			if (!state.uaccess && func) {
P
Peter Zijlstra 已提交
2546 2547 2548 2549 2550 2551 2552 2553 2554 2555
				WARN_FUNC("redundant UACCESS disable", sec, insn->offset);
				return 1;
			}

			if (func_uaccess_safe(func) && !state.uaccess_stack) {
				WARN_FUNC("UACCESS-safe disables UACCESS", sec, insn->offset);
				return 1;
			}

			state.uaccess = false;
2556 2557
			break;

2558 2559 2560 2561 2562 2563 2564 2565
		case INSN_STD:
			if (state.df)
				WARN_FUNC("recursive STD", sec, insn->offset);

			state.df = true;
			break;

		case INSN_CLD:
2566
			if (!state.df && func)
2567 2568 2569
				WARN_FUNC("redundant CLD", sec, insn->offset);

			state.df = false;
2570 2571
			break;

2572 2573 2574 2575 2576 2577 2578
		default:
			break;
		}

		if (insn->dead_end)
			return 0;

2579
		if (!next_insn) {
2580
			if (state.cfi.cfa.base == CFI_UNDEFINED)
2581
				return 0;
2582 2583 2584
			WARN("%s: unexpected end of section", sec->name);
			return 1;
		}
2585 2586

		insn = next_insn;
2587 2588 2589 2590 2591
	}

	return 0;
}

2592
static int validate_unwind_hints(struct objtool_file *file, struct section *sec)
2593 2594 2595
{
	struct instruction *insn;
	struct insn_state state;
2596
	int ret, warnings = 0;
2597 2598 2599 2600

	if (!file->hints)
		return 0;

2601
	init_insn_state(&state, sec);
2602

2603 2604 2605 2606 2607 2608 2609 2610 2611
	if (sec) {
		insn = find_insn(file, sec, 0);
		if (!insn)
			return 0;
	} else {
		insn = list_first_entry(&file->insn_list, typeof(*insn), list);
	}

	while (&insn->list != &file->insn_list && (!sec || insn->sec == sec)) {
2612
		if (insn->hint && !insn->visited) {
2613
			ret = validate_branch(file, insn->func, insn, state);
2614 2615
			if (ret && backtrace)
				BT_FUNC("<=== (hint)", insn);
2616 2617
			warnings += ret;
		}
2618 2619

		insn = list_next_entry(insn, list);
2620 2621 2622 2623 2624
	}

	return warnings;
}

2625 2626 2627 2628 2629 2630 2631 2632 2633 2634 2635 2636 2637
static int validate_retpoline(struct objtool_file *file)
{
	struct instruction *insn;
	int warnings = 0;

	for_each_insn(file, insn) {
		if (insn->type != INSN_JUMP_DYNAMIC &&
		    insn->type != INSN_CALL_DYNAMIC)
			continue;

		if (insn->retpoline_safe)
			continue;

2638 2639 2640 2641 2642 2643 2644 2645 2646
		/*
		 * .init.text code is ran before userspace and thus doesn't
		 * strictly need retpolines, except for modules which are
		 * loaded late, they very much do need retpoline in their
		 * .init.text
		 */
		if (!strcmp(insn->sec->name, ".init.text") && !module)
			continue;

2647 2648 2649 2650 2651 2652 2653 2654 2655 2656
		WARN_FUNC("indirect %s found in RETPOLINE build",
			  insn->sec, insn->offset,
			  insn->type == INSN_JUMP_DYNAMIC ? "jump" : "call");

		warnings++;
	}

	return warnings;
}

2657 2658 2659 2660 2661 2662 2663 2664 2665 2666 2667 2668 2669
static bool is_kasan_insn(struct instruction *insn)
{
	return (insn->type == INSN_CALL &&
		!strcmp(insn->call_dest->name, "__asan_handle_no_return"));
}

static bool is_ubsan_insn(struct instruction *insn)
{
	return (insn->type == INSN_CALL &&
		!strcmp(insn->call_dest->name,
			"__ubsan_handle_builtin_unreachable"));
}

2670
static bool ignore_unreachable_insn(struct instruction *insn)
2671 2672 2673
{
	int i;

2674 2675 2676 2677 2678 2679
	if (insn->ignore || insn->type == INSN_NOP)
		return true;

	/*
	 * Ignore any unused exceptions.  This can happen when a whitelisted
	 * function has an exception table entry.
2680 2681 2682
	 *
	 * Also ignore alternative replacement instructions.  This can happen
	 * when a whitelisted function uses one of the ALTERNATIVE macros.
2683
	 */
2684 2685 2686
	if (!strcmp(insn->sec->name, ".fixup") ||
	    !strcmp(insn->sec->name, ".altinstr_replacement") ||
	    !strcmp(insn->sec->name, ".altinstr_aux"))
2687 2688
		return true;

2689 2690 2691 2692 2693 2694 2695 2696 2697 2698 2699 2700 2701 2702 2703
	if (!insn->func)
		return false;

	/*
	 * CONFIG_UBSAN_TRAP inserts a UD2 when it sees
	 * __builtin_unreachable().  The BUG() macro has an unreachable() after
	 * the UD2, which causes GCC's undefined trap logic to emit another UD2
	 * (or occasionally a JMP to UD2).
	 */
	if (list_prev_entry(insn, list)->dead_end &&
	    (insn->type == INSN_BUG ||
	     (insn->type == INSN_JUMP_UNCONDITIONAL &&
	      insn->jump_dest && insn->jump_dest->type == INSN_BUG)))
		return true;

2704 2705 2706 2707 2708 2709 2710 2711 2712 2713 2714
	/*
	 * Check if this (or a subsequent) instruction is related to
	 * CONFIG_UBSAN or CONFIG_KASAN.
	 *
	 * End the search at 5 instructions to avoid going into the weeds.
	 */
	for (i = 0; i < 5; i++) {

		if (is_kasan_insn(insn) || is_ubsan_insn(insn))
			return true;

2715 2716 2717 2718 2719 2720 2721 2722
		if (insn->type == INSN_JUMP_UNCONDITIONAL) {
			if (insn->jump_dest &&
			    insn->jump_dest->func == insn->func) {
				insn = insn->jump_dest;
				continue;
			}

			break;
2723 2724
		}

2725
		if (insn->offset + insn->len >= insn->func->offset + insn->func->len)
2726
			break;
2727

2728 2729 2730 2731 2732 2733
		insn = list_next_entry(insn, list);
	}

	return false;
}

2734 2735
static int validate_symbol(struct objtool_file *file, struct section *sec,
			   struct symbol *sym, struct insn_state *state)
2736 2737
{
	struct instruction *insn;
2738 2739 2740 2741 2742 2743 2744 2745 2746 2747 2748 2749 2750 2751 2752 2753 2754 2755 2756 2757 2758 2759 2760 2761
	int ret;

	if (!sym->len) {
		WARN("%s() is missing an ELF size annotation", sym->name);
		return 1;
	}

	if (sym->pfunc != sym || sym->alias != sym)
		return 0;

	insn = find_insn(file, sec, sym->offset);
	if (!insn || insn->ignore || insn->visited)
		return 0;

	state->uaccess = sym->uaccess_safe;

	ret = validate_branch(file, insn->func, insn, *state);
	if (ret && backtrace)
		BT_FUNC("<=== (sym)", insn);
	return ret;
}

static int validate_section(struct objtool_file *file, struct section *sec)
{
2762
	struct insn_state state;
2763 2764
	struct symbol *func;
	int warnings = 0;
2765

2766 2767 2768
	list_for_each_entry(func, &sec->symbol_list, list) {
		if (func->type != STT_FUNC)
			continue;
2769

2770
		init_insn_state(&state, sec);
2771 2772
		state.cfi.cfa = initial_func_cfi.cfa;
		memcpy(&state.cfi.regs, &initial_func_cfi.regs,
2773
		       CFI_NUM_REGS * sizeof(struct cfi_reg));
2774
		state.cfi.stack_size = initial_func_cfi.cfa.offset;
2775

2776
		warnings += validate_symbol(file, sec, func, &state);
2777 2778 2779 2780 2781
	}

	return warnings;
}

2782 2783 2784
static int validate_vmlinux_functions(struct objtool_file *file)
{
	struct section *sec;
2785
	int warnings = 0;
2786 2787

	sec = find_section_by_name(file->elf, ".noinstr.text");
2788 2789 2790 2791
	if (sec) {
		warnings += validate_section(file, sec);
		warnings += validate_unwind_hints(file, sec);
	}
2792

2793 2794 2795 2796 2797
	sec = find_section_by_name(file->elf, ".entry.text");
	if (sec) {
		warnings += validate_section(file, sec);
		warnings += validate_unwind_hints(file, sec);
	}
2798 2799

	return warnings;
2800 2801
}

2802 2803 2804 2805 2806
static int validate_functions(struct objtool_file *file)
{
	struct section *sec;
	int warnings = 0;

2807 2808 2809 2810
	for_each_sec(file, sec) {
		if (!(sec->sh.sh_flags & SHF_EXECINSTR))
			continue;

2811
		warnings += validate_section(file, sec);
2812
	}
2813 2814 2815 2816

	return warnings;
}

2817
static int validate_reachable_instructions(struct objtool_file *file)
2818 2819
{
	struct instruction *insn;
2820 2821 2822

	if (file->ignore_unreachables)
		return 0;
2823 2824

	for_each_insn(file, insn) {
2825 2826 2827 2828 2829
		if (insn->visited || ignore_unreachable_insn(insn))
			continue;

		WARN_FUNC("unreachable instruction", insn->sec, insn->offset);
		return 1;
2830 2831
	}

2832
	return 0;
2833 2834
}

2835 2836
static struct objtool_file file;

2837
int check(const char *_objname, bool orc)
2838 2839 2840 2841 2842
{
	int ret, warnings = 0;

	objname = _objname;

2843
	file.elf = elf_open_read(objname, O_RDWR);
2844
	if (!file.elf)
2845 2846 2847 2848
		return 1;

	INIT_LIST_HEAD(&file.insn_list);
	hash_init(file.insn_hash);
2849
	file.c_file = !vmlinux && find_section_by_name(file.elf, ".comment");
2850
	file.ignore_unreachables = no_unreachable;
2851
	file.hints = false;
2852

2853 2854
	arch_initial_func_cfi_state(&initial_func_cfi);

2855 2856 2857 2858 2859
	ret = decode_sections(&file);
	if (ret < 0)
		goto out;
	warnings += ret;

2860
	if (list_empty(&file.insn_list))
2861 2862
		goto out;

2863 2864 2865 2866 2867 2868 2869 2870 2871
	if (vmlinux && !validate_dup) {
		ret = validate_vmlinux_functions(&file);
		if (ret < 0)
			goto out;

		warnings += ret;
		goto out;
	}

2872 2873 2874 2875 2876 2877 2878
	if (retpoline) {
		ret = validate_retpoline(&file);
		if (ret < 0)
			return ret;
		warnings += ret;
	}

2879
	ret = validate_functions(&file);
2880 2881 2882 2883
	if (ret < 0)
		goto out;
	warnings += ret;

2884
	ret = validate_unwind_hints(&file, NULL);
2885 2886 2887 2888
	if (ret < 0)
		goto out;
	warnings += ret;

2889 2890 2891 2892 2893 2894 2895
	if (!warnings) {
		ret = validate_reachable_instructions(&file);
		if (ret < 0)
			goto out;
		warnings += ret;
	}

2896 2897 2898 2899 2900 2901 2902 2903
	if (orc) {
		ret = create_orc(&file);
		if (ret < 0)
			goto out;

		ret = create_orc_sections(&file);
		if (ret < 0)
			goto out;
2904
	}
2905

2906
	if (file.elf->changed) {
2907 2908 2909 2910 2911
		ret = elf_write(file.elf);
		if (ret < 0)
			goto out;
	}

2912
out:
2913 2914 2915 2916 2917 2918 2919 2920 2921
	if (ret < 0) {
		/*
		 *  Fatal error.  The binary is corrupt or otherwise broken in
		 *  some way, or objtool itself is broken.  Fail the kernel
		 *  build.
		 */
		return ret;
	}

2922 2923
	return 0;
}