Commit | Line | Data |
---|---|---|
1ccea77e | 1 | // SPDX-License-Identifier: GPL-2.0-or-later |
442f04c3 JP |
2 | /* |
3 | * Copyright (C) 2015 Josh Poimboeuf <jpoimboe@redhat.com> | |
442f04c3 JP |
4 | */ |
5 | ||
442f04c3 JP |
6 | #include <stdio.h> |
7 | #include <stdbool.h> | |
8 | #include <string.h> | |
9 | #include <stdlib.h> | |
8ad15c69 | 10 | #include <unistd.h> |
442f04c3 JP |
11 | #include <subcmd/exec-cmd.h> |
12 | #include <subcmd/pager.h> | |
00614597 | 13 | #include <linux/kernel.h> |
442f04c3 | 14 | |
7786032e VG |
15 | #include <objtool/builtin.h> |
16 | #include <objtool/objtool.h> | |
17 | #include <objtool/warn.h> | |
442f04c3 | 18 | |
442f04c3 JP |
19 | bool help; |
20 | ||
6545eb03 JT |
21 | const char *objname; |
22 | static struct objtool_file file; | |
23 | ||
8ad15c69 PZ |
24 | static bool objtool_create_backup(const char *_objname) |
25 | { | |
26 | int len = strlen(_objname); | |
27 | char *buf, *base, *name = malloc(len+6); | |
28 | int s, d, l, t; | |
29 | ||
30 | if (!name) { | |
31 | perror("failed backup name malloc"); | |
32 | return false; | |
33 | } | |
34 | ||
35 | strcpy(name, _objname); | |
36 | strcpy(name + len, ".orig"); | |
37 | ||
38 | d = open(name, O_CREAT|O_WRONLY|O_TRUNC, 0644); | |
39 | if (d < 0) { | |
40 | perror("failed to create backup file"); | |
41 | return false; | |
42 | } | |
43 | ||
44 | s = open(_objname, O_RDONLY); | |
45 | if (s < 0) { | |
46 | perror("failed to open orig file"); | |
47 | return false; | |
48 | } | |
49 | ||
50 | buf = malloc(4096); | |
51 | if (!buf) { | |
52 | perror("failed backup data malloc"); | |
53 | return false; | |
54 | } | |
55 | ||
56 | while ((l = read(s, buf, 4096)) > 0) { | |
57 | base = buf; | |
58 | do { | |
59 | t = write(d, base, l); | |
60 | if (t < 0) { | |
61 | perror("failed backup write"); | |
62 | return false; | |
63 | } | |
64 | base += t; | |
65 | l -= t; | |
66 | } while (l); | |
67 | } | |
68 | ||
69 | if (l < 0) { | |
70 | perror("failed backup read"); | |
71 | return false; | |
72 | } | |
73 | ||
74 | free(name); | |
75 | free(buf); | |
76 | close(d); | |
77 | close(s); | |
78 | ||
79 | return true; | |
80 | } | |
81 | ||
6545eb03 JT |
82 | struct objtool_file *objtool_open_read(const char *_objname) |
83 | { | |
84 | if (objname) { | |
85 | if (strcmp(objname, _objname)) { | |
86 | WARN("won't handle more than one file at a time"); | |
87 | return NULL; | |
88 | } | |
89 | return &file; | |
90 | } | |
91 | objname = _objname; | |
92 | ||
93 | file.elf = elf_open_read(objname, O_RDWR); | |
94 | if (!file.elf) | |
95 | return NULL; | |
96 | ||
2daf7fab | 97 | if (opts.backup && !objtool_create_backup(objname)) { |
8ad15c69 PZ |
98 | WARN("can't create backup file"); |
99 | return NULL; | |
100 | } | |
101 | ||
6545eb03 | 102 | hash_init(file.insn_hash); |
43d5430a | 103 | INIT_LIST_HEAD(&file.retpoline_call_list); |
d9e9d230 | 104 | INIT_LIST_HEAD(&file.return_thunk_list); |
6545eb03 | 105 | INIT_LIST_HEAD(&file.static_call_list); |
99d00215 | 106 | INIT_LIST_HEAD(&file.mcount_loc_list); |
89bc853e | 107 | INIT_LIST_HEAD(&file.endbr_list); |
00abd384 | 108 | INIT_LIST_HEAD(&file.call_list); |
2daf7fab | 109 | file.ignore_unreachables = opts.no_unreachable; |
6545eb03 JT |
110 | file.hints = false; |
111 | ||
112 | return &file; | |
113 | } | |
114 | ||
db2b0c5d PZ |
115 | void objtool_pv_add(struct objtool_file *f, int idx, struct symbol *func) |
116 | { | |
2daf7fab | 117 | if (!opts.noinstr) |
db2b0c5d PZ |
118 | return; |
119 | ||
120 | if (!f->pv_ops) { | |
121 | WARN("paravirt confusion"); | |
122 | return; | |
123 | } | |
124 | ||
125 | /* | |
126 | * These functions will be patched into native code, | |
127 | * see paravirt_patch(). | |
128 | */ | |
129 | if (!strcmp(func->name, "_paravirt_nop") || | |
130 | !strcmp(func->name, "_paravirt_ident_64")) | |
131 | return; | |
132 | ||
988f0168 PZ |
133 | /* already added this function */ |
134 | if (!list_empty(&func->pv_target)) | |
135 | return; | |
136 | ||
db2b0c5d PZ |
137 | list_add(&func->pv_target, &f->pv_ops[idx].targets); |
138 | f->pv_ops[idx].clean = false; | |
139 | } | |
140 | ||
442f04c3 JP |
141 | int main(int argc, const char **argv) |
142 | { | |
143 | static const char *UNUSED = "OBJTOOL_NOT_IMPLEMENTED"; | |
144 | ||
145 | /* libsubcmd init */ | |
146 | exec_cmd_init("objtool", UNUSED, UNUSED, UNUSED); | |
147 | pager_init(UNUSED); | |
148 | ||
e959c279 | 149 | return objtool_run(argc, argv); |
442f04c3 | 150 | } |