| 1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774 |
- // SPDX-License-Identifier: GPL-2.0-or-later
- #define _GNU_SOURCE /* memmem() */
- #include <subcmd/parse-options.h>
- #include <stdlib.h>
- #include <string.h>
- #include <libgen.h>
- #include <stdio.h>
- #include <ctype.h>
- #include <objtool/objtool.h>
- #include <objtool/warn.h>
- #include <objtool/arch.h>
- #include <objtool/klp.h>
- #include <objtool/util.h>
- #include <arch/special.h>
- #include <linux/align.h>
- #include <linux/objtool_types.h>
- #include <linux/livepatch_external.h>
- #include <linux/stringify.h>
- #include <linux/string.h>
- #include <linux/jhash.h>
- #define sizeof_field(TYPE, MEMBER) sizeof((((TYPE *)0)->MEMBER))
- struct elfs {
- struct elf *orig, *patched, *out;
- const char *modname;
- };
- struct export {
- struct hlist_node hash;
- char *mod, *sym;
- };
- static const char * const klp_diff_usage[] = {
- "objtool klp diff [<options>] <in1.o> <in2.o> <out.o>",
- NULL,
- };
- static const struct option klp_diff_options[] = {
- OPT_GROUP("Options:"),
- OPT_BOOLEAN('d', "debug", &debug, "enable debug output"),
- OPT_END(),
- };
- static DEFINE_HASHTABLE(exports, 15);
- static inline u32 str_hash(const char *str)
- {
- return jhash(str, strlen(str), 0);
- }
- static char *escape_str(const char *orig)
- {
- size_t len = 0;
- const char *a;
- char *b, *new;
- for (a = orig; *a; a++) {
- switch (*a) {
- case '\001': len += 5; break;
- case '\n':
- case '\t': len += 2; break;
- default: len++;
- }
- }
- new = malloc(len + 1);
- if (!new)
- return NULL;
- for (a = orig, b = new; *a; a++) {
- switch (*a) {
- case '\001': memcpy(b, "<SOH>", 5); b += 5; break;
- case '\n': *b++ = '\\'; *b++ = 'n'; break;
- case '\t': *b++ = '\\'; *b++ = 't'; break;
- default: *b++ = *a;
- }
- }
- *b = '\0';
- return new;
- }
- static int read_exports(void)
- {
- const char *symvers = "Module.symvers";
- char line[1024], *path = NULL;
- unsigned int line_num = 1;
- FILE *file;
- file = fopen(symvers, "r");
- if (!file) {
- path = top_level_dir(symvers);
- if (!path) {
- ERROR("can't open '%s', \"objtool diff\" should be run from the kernel tree", symvers);
- return -1;
- }
- file = fopen(path, "r");
- if (!file) {
- ERROR_GLIBC("fopen");
- return -1;
- }
- }
- while (fgets(line, 1024, file)) {
- char *sym, *mod, *type;
- struct export *export;
- sym = strchr(line, '\t');
- if (!sym) {
- ERROR("malformed Module.symvers (sym) at line %d", line_num);
- return -1;
- }
- *sym++ = '\0';
- mod = strchr(sym, '\t');
- if (!mod) {
- ERROR("malformed Module.symvers (mod) at line %d", line_num);
- return -1;
- }
- *mod++ = '\0';
- type = strchr(mod, '\t');
- if (!type) {
- ERROR("malformed Module.symvers (type) at line %d", line_num);
- return -1;
- }
- *type++ = '\0';
- if (*sym == '\0' || *mod == '\0') {
- ERROR("malformed Module.symvers at line %d", line_num);
- return -1;
- }
- export = calloc(1, sizeof(*export));
- if (!export) {
- ERROR_GLIBC("calloc");
- return -1;
- }
- export->mod = strdup(mod);
- if (!export->mod) {
- ERROR_GLIBC("strdup");
- return -1;
- }
- export->sym = strdup(sym);
- if (!export->sym) {
- ERROR_GLIBC("strdup");
- return -1;
- }
- hash_add(exports, &export->hash, str_hash(sym));
- }
- free(path);
- fclose(file);
- return 0;
- }
- static int read_sym_checksums(struct elf *elf)
- {
- struct section *sec;
- sec = find_section_by_name(elf, ".discard.sym_checksum");
- if (!sec) {
- ERROR("'%s' missing .discard.sym_checksum section, file not processed by 'objtool --checksum'?",
- elf->name);
- return -1;
- }
- if (!sec->rsec) {
- ERROR("missing reloc section for .discard.sym_checksum");
- return -1;
- }
- if (sec_size(sec) % sizeof(struct sym_checksum)) {
- ERROR("struct sym_checksum size mismatch");
- return -1;
- }
- for (int i = 0; i < sec_size(sec) / sizeof(struct sym_checksum); i++) {
- struct sym_checksum *sym_checksum;
- struct reloc *reloc;
- struct symbol *sym;
- sym_checksum = (struct sym_checksum *)sec->data->d_buf + i;
- reloc = find_reloc_by_dest(elf, sec, i * sizeof(*sym_checksum));
- if (!reloc) {
- ERROR("can't find reloc for sym_checksum[%d]", i);
- return -1;
- }
- sym = reloc->sym;
- if (is_sec_sym(sym)) {
- ERROR("not sure how to handle section %s", sym->name);
- return -1;
- }
- if (is_func_sym(sym))
- sym->csum.checksum = sym_checksum->checksum;
- }
- return 0;
- }
- static struct symbol *first_file_symbol(struct elf *elf)
- {
- struct symbol *sym;
- for_each_sym(elf, sym) {
- if (is_file_sym(sym))
- return sym;
- }
- return NULL;
- }
- static struct symbol *next_file_symbol(struct elf *elf, struct symbol *sym)
- {
- for_each_sym_continue(elf, sym) {
- if (is_file_sym(sym))
- return sym;
- }
- return NULL;
- }
- /*
- * Certain static local variables should never be correlated. They will be
- * used in place rather than referencing the originals.
- */
- static bool is_uncorrelated_static_local(struct symbol *sym)
- {
- static const char * const vars[] = {
- "__already_done.",
- "__func__.",
- "__key.",
- "__warned.",
- "_entry.",
- "_entry_ptr.",
- "_rs.",
- "descriptor.",
- "CSWTCH.",
- };
- if (!is_object_sym(sym) || !is_local_sym(sym))
- return false;
- if (!strcmp(sym->sec->name, ".data.once"))
- return true;
- for (int i = 0; i < ARRAY_SIZE(vars); i++) {
- if (strstarts(sym->name, vars[i]))
- return true;
- }
- return false;
- }
- /*
- * Clang emits several useless .Ltmp_* code labels.
- */
- static bool is_clang_tmp_label(struct symbol *sym)
- {
- return sym->type == STT_NOTYPE &&
- is_text_sec(sym->sec) &&
- strstarts(sym->name, ".Ltmp") &&
- isdigit(sym->name[5]);
- }
- static bool is_special_section(struct section *sec)
- {
- static const char * const specials[] = {
- ".altinstructions",
- ".smp_locks",
- "__bug_table",
- "__ex_table",
- "__jump_table",
- "__mcount_loc",
- /*
- * Extract .static_call_sites here to inherit non-module
- * preferential treatment. The later static call processing
- * during klp module build will be skipped when it sees this
- * section already exists.
- */
- ".static_call_sites",
- };
- static const char * const non_special_discards[] = {
- ".discard.addressable",
- ".discard.sym_checksum",
- };
- if (is_text_sec(sec))
- return false;
- for (int i = 0; i < ARRAY_SIZE(specials); i++) {
- if (!strcmp(sec->name, specials[i]))
- return true;
- }
- /* Most .discard data sections are special */
- for (int i = 0; i < ARRAY_SIZE(non_special_discards); i++) {
- if (!strcmp(sec->name, non_special_discards[i]))
- return false;
- }
- return strstarts(sec->name, ".discard.");
- }
- /*
- * These sections are referenced by special sections but aren't considered
- * special sections themselves.
- */
- static bool is_special_section_aux(struct section *sec)
- {
- static const char * const specials_aux[] = {
- ".altinstr_replacement",
- ".altinstr_aux",
- };
- for (int i = 0; i < ARRAY_SIZE(specials_aux); i++) {
- if (!strcmp(sec->name, specials_aux[i]))
- return true;
- }
- return false;
- }
- /*
- * These symbols should never be correlated, so their local patched versions
- * are used instead of linking to the originals.
- */
- static bool dont_correlate(struct symbol *sym)
- {
- return is_file_sym(sym) ||
- is_null_sym(sym) ||
- is_sec_sym(sym) ||
- is_prefix_func(sym) ||
- is_uncorrelated_static_local(sym) ||
- is_clang_tmp_label(sym) ||
- is_string_sec(sym->sec) ||
- is_special_section(sym->sec) ||
- is_special_section_aux(sym->sec) ||
- strstarts(sym->name, "__initcall__");
- }
- /*
- * For each symbol in the original kernel, find its corresponding "twin" in the
- * patched kernel.
- */
- static int correlate_symbols(struct elfs *e)
- {
- struct symbol *file1_sym, *file2_sym;
- struct symbol *sym1, *sym2;
- file1_sym = first_file_symbol(e->orig);
- file2_sym = first_file_symbol(e->patched);
- /*
- * Correlate any locals before the first FILE symbol. This has been
- * seen when LTO inexplicably strips the initramfs_data.o FILE symbol
- * due to the file only containing data and no code.
- */
- for_each_sym(e->orig, sym1) {
- if (sym1 == file1_sym || !is_local_sym(sym1))
- break;
- if (dont_correlate(sym1))
- continue;
- for_each_sym(e->patched, sym2) {
- if (sym2 == file2_sym || !is_local_sym(sym2))
- break;
- if (sym2->twin || dont_correlate(sym2))
- continue;
- if (strcmp(sym1->demangled_name, sym2->demangled_name))
- continue;
- sym1->twin = sym2;
- sym2->twin = sym1;
- break;
- }
- }
- /* Correlate locals after the first FILE symbol */
- for (; ; file1_sym = next_file_symbol(e->orig, file1_sym),
- file2_sym = next_file_symbol(e->patched, file2_sym)) {
- if (!file1_sym && file2_sym) {
- ERROR("FILE symbol mismatch: NULL != %s", file2_sym->name);
- return -1;
- }
- if (file1_sym && !file2_sym) {
- ERROR("FILE symbol mismatch: %s != NULL", file1_sym->name);
- return -1;
- }
- if (!file1_sym)
- break;
- if (strcmp(file1_sym->name, file2_sym->name)) {
- ERROR("FILE symbol mismatch: %s != %s", file1_sym->name, file2_sym->name);
- return -1;
- }
- file1_sym->twin = file2_sym;
- file2_sym->twin = file1_sym;
- sym1 = file1_sym;
- for_each_sym_continue(e->orig, sym1) {
- if (is_file_sym(sym1) || !is_local_sym(sym1))
- break;
- if (dont_correlate(sym1))
- continue;
- sym2 = file2_sym;
- for_each_sym_continue(e->patched, sym2) {
- if (is_file_sym(sym2) || !is_local_sym(sym2))
- break;
- if (sym2->twin || dont_correlate(sym2))
- continue;
- if (strcmp(sym1->demangled_name, sym2->demangled_name))
- continue;
- sym1->twin = sym2;
- sym2->twin = sym1;
- break;
- }
- }
- }
- /* Correlate globals */
- for_each_sym(e->orig, sym1) {
- if (sym1->bind == STB_LOCAL)
- continue;
- sym2 = find_global_symbol_by_name(e->patched, sym1->name);
- if (sym2 && !sym2->twin && !strcmp(sym1->name, sym2->name)) {
- sym1->twin = sym2;
- sym2->twin = sym1;
- }
- }
- for_each_sym(e->orig, sym1) {
- if (sym1->twin || dont_correlate(sym1))
- continue;
- WARN("no correlation: %s", sym1->name);
- }
- return 0;
- }
- /* "sympos" is used by livepatch to disambiguate duplicate symbol names */
- static unsigned long find_sympos(struct elf *elf, struct symbol *sym)
- {
- bool vmlinux = str_ends_with(objname, "vmlinux.o");
- unsigned long sympos = 0, nr_matches = 0;
- bool has_dup = false;
- struct symbol *s;
- if (sym->bind != STB_LOCAL)
- return 0;
- if (vmlinux && sym->type == STT_FUNC) {
- /*
- * HACK: Unfortunately, symbol ordering can differ between
- * vmlinux.o and vmlinux due to the linker script emitting
- * .text.unlikely* before .text*. Count .text.unlikely* first.
- *
- * TODO: Disambiguate symbols more reliably (checksums?)
- */
- for_each_sym(elf, s) {
- if (strstarts(s->sec->name, ".text.unlikely") &&
- !strcmp(s->name, sym->name)) {
- nr_matches++;
- if (s == sym)
- sympos = nr_matches;
- else
- has_dup = true;
- }
- }
- for_each_sym(elf, s) {
- if (!strstarts(s->sec->name, ".text.unlikely") &&
- !strcmp(s->name, sym->name)) {
- nr_matches++;
- if (s == sym)
- sympos = nr_matches;
- else
- has_dup = true;
- }
- }
- } else {
- for_each_sym(elf, s) {
- if (!strcmp(s->name, sym->name)) {
- nr_matches++;
- if (s == sym)
- sympos = nr_matches;
- else
- has_dup = true;
- }
- }
- }
- if (!sympos) {
- ERROR("can't find sympos for %s", sym->name);
- return ULONG_MAX;
- }
- return has_dup ? sympos : 0;
- }
- static int clone_sym_relocs(struct elfs *e, struct symbol *patched_sym);
- static struct symbol *__clone_symbol(struct elf *elf, struct symbol *patched_sym,
- bool data_too)
- {
- struct section *out_sec = NULL;
- unsigned long offset = 0;
- struct symbol *out_sym;
- if (data_too && !is_undef_sym(patched_sym)) {
- struct section *patched_sec = patched_sym->sec;
- out_sec = find_section_by_name(elf, patched_sec->name);
- if (!out_sec) {
- out_sec = elf_create_section(elf, patched_sec->name, 0,
- patched_sec->sh.sh_entsize,
- patched_sec->sh.sh_type,
- patched_sec->sh.sh_addralign,
- patched_sec->sh.sh_flags);
- if (!out_sec)
- return NULL;
- }
- if (is_string_sec(patched_sym->sec)) {
- out_sym = elf_create_section_symbol(elf, out_sec);
- if (!out_sym)
- return NULL;
- goto sym_created;
- }
- if (!is_sec_sym(patched_sym))
- offset = ALIGN(sec_size(out_sec), out_sec->sh.sh_addralign);
- if (patched_sym->len || is_sec_sym(patched_sym)) {
- void *data = NULL;
- size_t size;
- /* bss doesn't have data */
- if (patched_sym->sec->data->d_buf)
- data = patched_sym->sec->data->d_buf + patched_sym->offset;
- if (is_sec_sym(patched_sym))
- size = sec_size(patched_sym->sec);
- else
- size = patched_sym->len;
- if (!elf_add_data(elf, out_sec, data, size))
- return NULL;
- }
- }
- out_sym = elf_create_symbol(elf, patched_sym->name, out_sec,
- patched_sym->bind, patched_sym->type,
- offset, patched_sym->len);
- if (!out_sym)
- return NULL;
- sym_created:
- patched_sym->clone = out_sym;
- out_sym->clone = patched_sym;
- return out_sym;
- }
- static const char *sym_type(struct symbol *sym)
- {
- switch (sym->type) {
- case STT_NOTYPE: return "NOTYPE";
- case STT_OBJECT: return "OBJECT";
- case STT_FUNC: return "FUNC";
- case STT_SECTION: return "SECTION";
- case STT_FILE: return "FILE";
- default: return "UNKNOWN";
- }
- }
- static const char *sym_bind(struct symbol *sym)
- {
- switch (sym->bind) {
- case STB_LOCAL: return "LOCAL";
- case STB_GLOBAL: return "GLOBAL";
- case STB_WEAK: return "WEAK";
- default: return "UNKNOWN";
- }
- }
- /*
- * Copy a symbol to the output object, optionally including its data and
- * relocations.
- */
- static struct symbol *clone_symbol(struct elfs *e, struct symbol *patched_sym,
- bool data_too)
- {
- struct symbol *pfx;
- if (patched_sym->clone)
- return patched_sym->clone;
- dbg_indent("%s%s", patched_sym->name, data_too ? " [+DATA]" : "");
- /* Make sure the prefix gets cloned first */
- if (is_func_sym(patched_sym) && data_too) {
- pfx = get_func_prefix(patched_sym);
- if (pfx)
- clone_symbol(e, pfx, true);
- }
- if (!__clone_symbol(e->out, patched_sym, data_too))
- return NULL;
- if (data_too && clone_sym_relocs(e, patched_sym))
- return NULL;
- return patched_sym->clone;
- }
- static void mark_included_function(struct symbol *func)
- {
- struct symbol *pfx;
- func->included = 1;
- /* Include prefix function */
- pfx = get_func_prefix(func);
- if (pfx)
- pfx->included = 1;
- /* Make sure .cold parent+child always stay together */
- if (func->cfunc && func->cfunc != func)
- func->cfunc->included = 1;
- if (func->pfunc && func->pfunc != func)
- func->pfunc->included = 1;
- }
- /*
- * Copy all changed functions (and their dependencies) from the patched object
- * to the output object.
- */
- static int mark_changed_functions(struct elfs *e)
- {
- struct symbol *sym_orig, *patched_sym;
- bool changed = false;
- /* Find changed functions */
- for_each_sym(e->orig, sym_orig) {
- if (!is_func_sym(sym_orig) || is_prefix_func(sym_orig))
- continue;
- patched_sym = sym_orig->twin;
- if (!patched_sym)
- continue;
- if (sym_orig->csum.checksum != patched_sym->csum.checksum) {
- patched_sym->changed = 1;
- mark_included_function(patched_sym);
- changed = true;
- }
- }
- /* Find added functions and print them */
- for_each_sym(e->patched, patched_sym) {
- if (!is_func_sym(patched_sym) || is_prefix_func(patched_sym))
- continue;
- if (!patched_sym->twin) {
- printf("%s: new function: %s\n", objname, patched_sym->name);
- mark_included_function(patched_sym);
- changed = true;
- }
- }
- /* Print changed functions */
- for_each_sym(e->patched, patched_sym) {
- if (patched_sym->changed)
- printf("%s: changed function: %s\n", objname, patched_sym->name);
- }
- return !changed ? -1 : 0;
- }
- static int clone_included_functions(struct elfs *e)
- {
- struct symbol *patched_sym;
- for_each_sym(e->patched, patched_sym) {
- if (patched_sym->included) {
- if (!clone_symbol(e, patched_sym, true))
- return -1;
- }
- }
- return 0;
- }
- /*
- * Determine whether a relocation should reference the section rather than the
- * underlying symbol.
- */
- static bool section_reference_needed(struct section *sec)
- {
- /*
- * String symbols are zero-length and uncorrelated. It's easier to
- * deal with them as section symbols.
- */
- if (is_string_sec(sec))
- return true;
- /*
- * .rodata has mostly anonymous data so there's no way to determine the
- * length of a needed reference. just copy the whole section if needed.
- */
- if (strstarts(sec->name, ".rodata"))
- return true;
- /* UBSAN anonymous data */
- if (strstarts(sec->name, ".data..Lubsan") || /* GCC */
- strstarts(sec->name, ".data..L__unnamed_")) /* Clang */
- return true;
- return false;
- }
- static bool is_reloc_allowed(struct reloc *reloc)
- {
- return section_reference_needed(reloc->sym->sec) == is_sec_sym(reloc->sym);
- }
- static struct export *find_export(struct symbol *sym)
- {
- struct export *export;
- hash_for_each_possible(exports, export, hash, str_hash(sym->name)) {
- if (!strcmp(export->sym, sym->name))
- return export;
- }
- return NULL;
- }
- static const char *__find_modname(struct elfs *e)
- {
- struct section *sec;
- char *name;
- sec = find_section_by_name(e->orig, ".modinfo");
- if (!sec) {
- ERROR("missing .modinfo section");
- return NULL;
- }
- name = memmem(sec->data->d_buf, sec_size(sec), "\0name=", 6);
- if (name)
- return name + 6;
- name = strdup(e->orig->name);
- if (!name) {
- ERROR_GLIBC("strdup");
- return NULL;
- }
- for (char *c = name; *c; c++) {
- if (*c == '/')
- name = c + 1;
- else if (*c == '-')
- *c = '_';
- else if (*c == '.') {
- *c = '\0';
- break;
- }
- }
- return name;
- }
- /* Get the object's module name as defined by the kernel (and klp_object) */
- static const char *find_modname(struct elfs *e)
- {
- const char *modname;
- if (e->modname)
- return e->modname;
- modname = __find_modname(e);
- e->modname = modname;
- return modname;
- }
- /*
- * Copying a function from its native compiled environment to a kernel module
- * removes its natural access to local functions/variables and unexported
- * globals. References to such symbols need to be converted to KLP relocs so
- * the kernel arch relocation code knows to apply them and where to find the
- * symbols. Particularly, duplicate static symbols need to be disambiguated.
- */
- static bool klp_reloc_needed(struct reloc *patched_reloc)
- {
- struct symbol *patched_sym = patched_reloc->sym;
- struct export *export;
- /* no external symbol to reference */
- if (dont_correlate(patched_sym))
- return false;
- /* For included functions, a regular reloc will do. */
- if (patched_sym->included)
- return false;
- /*
- * If exported by a module, it has to be a klp reloc. Thanks to the
- * clusterfunk that is late module patching, the patch module is
- * allowed to be loaded before any modules it depends on.
- *
- * If exported by vmlinux, a normal reloc will do.
- */
- export = find_export(patched_sym);
- if (export)
- return strcmp(export->mod, "vmlinux");
- if (!patched_sym->twin) {
- /*
- * Presumably the symbol and its reference were added by the
- * patch. The symbol could be defined in this .o or in another
- * .o in the patch module.
- *
- * This check needs to be *after* the export check due to the
- * possibility of the patch adding a new UNDEF reference to an
- * exported symbol.
- */
- return false;
- }
- /* Unexported symbol which lives in the original vmlinux or module. */
- return true;
- }
- static int convert_reloc_sym_to_secsym(struct elf *elf, struct reloc *reloc)
- {
- struct symbol *sym = reloc->sym;
- struct section *sec = sym->sec;
- if (!sec->sym && !elf_create_section_symbol(elf, sec))
- return -1;
- reloc->sym = sec->sym;
- set_reloc_sym(elf, reloc, sym->idx);
- set_reloc_addend(elf, reloc, sym->offset + reloc_addend(reloc));
- return 0;
- }
- static int convert_reloc_secsym_to_sym(struct elf *elf, struct reloc *reloc)
- {
- struct symbol *sym = reloc->sym;
- struct section *sec = sym->sec;
- /* If the symbol has a dedicated section, it's easy to find */
- sym = find_symbol_by_offset(sec, 0);
- if (sym && sym->len == sec_size(sec))
- goto found_sym;
- /* No dedicated section; find the symbol manually */
- sym = find_symbol_containing(sec, arch_adjusted_addend(reloc));
- if (!sym) {
- /*
- * This can happen for special section references to weak code
- * whose symbol has been stripped by the linker.
- */
- return -1;
- }
- found_sym:
- reloc->sym = sym;
- set_reloc_sym(elf, reloc, sym->idx);
- set_reloc_addend(elf, reloc, reloc_addend(reloc) - sym->offset);
- return 0;
- }
- /*
- * Convert a relocation symbol reference to the needed format: either a section
- * symbol or the underlying symbol itself.
- */
- static int convert_reloc_sym(struct elf *elf, struct reloc *reloc)
- {
- if (is_reloc_allowed(reloc))
- return 0;
- if (section_reference_needed(reloc->sym->sec))
- return convert_reloc_sym_to_secsym(elf, reloc);
- else
- return convert_reloc_secsym_to_sym(elf, reloc);
- }
- /*
- * Convert a regular relocation to a klp relocation (sort of).
- */
- static int clone_reloc_klp(struct elfs *e, struct reloc *patched_reloc,
- struct section *sec, unsigned long offset,
- struct export *export)
- {
- struct symbol *patched_sym = patched_reloc->sym;
- s64 addend = reloc_addend(patched_reloc);
- const char *sym_modname, *sym_orig_name;
- static struct section *klp_relocs;
- struct symbol *sym, *klp_sym;
- unsigned long klp_reloc_off;
- char sym_name[SYM_NAME_LEN];
- struct klp_reloc klp_reloc;
- unsigned long sympos;
- if (!patched_sym->twin) {
- ERROR("unexpected klp reloc for new symbol %s", patched_sym->name);
- return -1;
- }
- /*
- * Keep the original reloc intact for now to avoid breaking objtool run
- * which relies on proper relocations for many of its features. This
- * will be disabled later by "objtool klp post-link".
- *
- * Convert it to UNDEF (and WEAK to avoid modpost warnings).
- */
- sym = patched_sym->clone;
- if (!sym) {
- /* STB_WEAK: avoid modpost undefined symbol warnings */
- sym = elf_create_symbol(e->out, patched_sym->name, NULL,
- STB_WEAK, patched_sym->type, 0, 0);
- if (!sym)
- return -1;
- patched_sym->clone = sym;
- sym->clone = patched_sym;
- }
- if (!elf_create_reloc(e->out, sec, offset, sym, addend, reloc_type(patched_reloc)))
- return -1;
- /*
- * Create the KLP symbol.
- */
- if (export) {
- sym_modname = export->mod;
- sym_orig_name = export->sym;
- sympos = 0;
- } else {
- sym_modname = find_modname(e);
- if (!sym_modname)
- return -1;
- sym_orig_name = patched_sym->twin->name;
- sympos = find_sympos(e->orig, patched_sym->twin);
- if (sympos == ULONG_MAX)
- return -1;
- }
- /* symbol format: .klp.sym.modname.sym_name,sympos */
- if (snprintf_check(sym_name, SYM_NAME_LEN, KLP_SYM_PREFIX "%s.%s,%ld",
- sym_modname, sym_orig_name, sympos))
- return -1;
- klp_sym = find_symbol_by_name(e->out, sym_name);
- if (!klp_sym) {
- __dbg_indent("%s", sym_name);
- /* STB_WEAK: avoid modpost undefined symbol warnings */
- klp_sym = elf_create_symbol(e->out, sym_name, NULL,
- STB_WEAK, patched_sym->type, 0, 0);
- if (!klp_sym)
- return -1;
- }
- /*
- * Create the __klp_relocs entry. This will be converted to an actual
- * KLP rela by "objtool klp post-link".
- *
- * This intermediate step is necessary to prevent corruption by the
- * linker, which doesn't know how to properly handle two rela sections
- * applying to the same base section.
- */
- if (!klp_relocs) {
- klp_relocs = elf_create_section(e->out, KLP_RELOCS_SEC, 0,
- 0, SHT_PROGBITS, 8, SHF_ALLOC);
- if (!klp_relocs)
- return -1;
- }
- klp_reloc_off = sec_size(klp_relocs);
- memset(&klp_reloc, 0, sizeof(klp_reloc));
- klp_reloc.type = reloc_type(patched_reloc);
- if (!elf_add_data(e->out, klp_relocs, &klp_reloc, sizeof(klp_reloc)))
- return -1;
- /* klp_reloc.offset */
- if (!sec->sym && !elf_create_section_symbol(e->out, sec))
- return -1;
- if (!elf_create_reloc(e->out, klp_relocs,
- klp_reloc_off + offsetof(struct klp_reloc, offset),
- sec->sym, offset, R_ABS64))
- return -1;
- /* klp_reloc.sym */
- if (!elf_create_reloc(e->out, klp_relocs,
- klp_reloc_off + offsetof(struct klp_reloc, sym),
- klp_sym, addend, R_ABS64))
- return -1;
- return 0;
- }
- #define dbg_clone_reloc(sec, offset, patched_sym, addend, export, klp) \
- dbg_indent("%s+0x%lx: %s%s0x%lx [%s%s%s%s%s%s]", \
- sec->name, offset, patched_sym->name, \
- addend >= 0 ? "+" : "-", labs(addend), \
- sym_type(patched_sym), \
- patched_sym->type == STT_SECTION ? "" : " ", \
- patched_sym->type == STT_SECTION ? "" : sym_bind(patched_sym), \
- is_undef_sym(patched_sym) ? " UNDEF" : "", \
- export ? " EXPORTED" : "", \
- klp ? " KLP" : "")
- /* Copy a reloc and its symbol to the output object */
- static int clone_reloc(struct elfs *e, struct reloc *patched_reloc,
- struct section *sec, unsigned long offset)
- {
- struct symbol *patched_sym = patched_reloc->sym;
- struct export *export = find_export(patched_sym);
- long addend = reloc_addend(patched_reloc);
- struct symbol *out_sym;
- bool klp;
- if (!is_reloc_allowed(patched_reloc)) {
- ERROR_FUNC(patched_reloc->sec->base, reloc_offset(patched_reloc),
- "missing symbol for reference to %s+%ld",
- patched_sym->name, addend);
- return -1;
- }
- klp = klp_reloc_needed(patched_reloc);
- dbg_clone_reloc(sec, offset, patched_sym, addend, export, klp);
- if (klp) {
- if (clone_reloc_klp(e, patched_reloc, sec, offset, export))
- return -1;
- return 0;
- }
- /*
- * Why !export sets 'data_too':
- *
- * Unexported non-klp symbols need to live in the patch module,
- * otherwise there will be unresolved symbols. Notably, this includes:
- *
- * - New functions/data
- * - String sections
- * - Special section entries
- * - Uncorrelated static local variables
- * - UBSAN sections
- */
- out_sym = clone_symbol(e, patched_sym, patched_sym->included || !export);
- if (!out_sym)
- return -1;
- /*
- * For strings, all references use section symbols, thanks to
- * section_reference_needed(). clone_symbol() has cloned an empty
- * version of the string section. Now copy the string itself.
- */
- if (is_string_sec(patched_sym->sec)) {
- const char *str = patched_sym->sec->data->d_buf + addend;
- __dbg_indent("\"%s\"", escape_str(str));
- addend = elf_add_string(e->out, out_sym->sec, str);
- if (addend == -1)
- return -1;
- }
- if (!elf_create_reloc(e->out, sec, offset, out_sym, addend,
- reloc_type(patched_reloc)))
- return -1;
- return 0;
- }
- /* Copy all relocs needed for a symbol's contents */
- static int clone_sym_relocs(struct elfs *e, struct symbol *patched_sym)
- {
- struct section *patched_rsec = patched_sym->sec->rsec;
- struct reloc *patched_reloc;
- unsigned long start, end;
- struct symbol *out_sym;
- out_sym = patched_sym->clone;
- if (!out_sym) {
- ERROR("no clone for %s", patched_sym->name);
- return -1;
- }
- if (!patched_rsec)
- return 0;
- if (!is_sec_sym(patched_sym) && !patched_sym->len)
- return 0;
- if (is_string_sec(patched_sym->sec))
- return 0;
- if (is_sec_sym(patched_sym)) {
- start = 0;
- end = sec_size(patched_sym->sec);
- } else {
- start = patched_sym->offset;
- end = start + patched_sym->len;
- }
- for_each_reloc(patched_rsec, patched_reloc) {
- unsigned long offset;
- if (reloc_offset(patched_reloc) < start ||
- reloc_offset(patched_reloc) >= end)
- continue;
- /*
- * Skip any reloc referencing .altinstr_aux. Its code is
- * always patched by alternatives. See ALTERNATIVE_TERNARY().
- */
- if (patched_reloc->sym->sec &&
- !strcmp(patched_reloc->sym->sec->name, ".altinstr_aux"))
- continue;
- if (convert_reloc_sym(e->patched, patched_reloc)) {
- ERROR_FUNC(patched_rsec->base, reloc_offset(patched_reloc),
- "failed to convert reloc sym '%s' to its proper format",
- patched_reloc->sym->name);
- return -1;
- }
- offset = out_sym->offset + (reloc_offset(patched_reloc) - patched_sym->offset);
- if (clone_reloc(e, patched_reloc, out_sym->sec, offset))
- return -1;
- }
- return 0;
- }
- static int create_fake_symbol(struct elf *elf, struct section *sec,
- unsigned long offset, size_t size)
- {
- char name[SYM_NAME_LEN];
- unsigned int type;
- static int ctr;
- char *c;
- if (snprintf_check(name, SYM_NAME_LEN, "%s_%d", sec->name, ctr++))
- return -1;
- for (c = name; *c; c++)
- if (*c == '.')
- *c = '_';
- /*
- * STT_NOTYPE: Prevent objtool from validating .altinstr_replacement
- * while still allowing objdump to disassemble it.
- */
- type = is_text_sec(sec) ? STT_NOTYPE : STT_OBJECT;
- return elf_create_symbol(elf, name, sec, STB_LOCAL, type, offset, size) ? 0 : -1;
- }
- /*
- * Special sections (alternatives, etc) are basically arrays of structs.
- * For all the special sections, create a symbol for each struct entry. This
- * is a bit cumbersome, but it makes the extracting of the individual entries
- * much more straightforward.
- *
- * There are three ways to identify the entry sizes for a special section:
- *
- * 1) ELF section header sh_entsize: Ideally this would be used almost
- * everywhere. But unfortunately the toolchains make it difficult. The
- * assembler .[push]section directive syntax only takes entsize when
- * combined with SHF_MERGE. But Clang disallows combining SHF_MERGE with
- * SHF_WRITE. And some special sections do need to be writable.
- *
- * Another place this wouldn't work is .altinstr_replacement, whose entries
- * don't have a fixed size.
- *
- * 2) ANNOTATE_DATA_SPECIAL: This is a lightweight objtool annotation which
- * points to the beginning of each entry. The size of the entry is then
- * inferred by the location of the subsequent annotation (or end of
- * section).
- *
- * 3) Simple array of pointers: If the special section is just a basic array of
- * pointers, the entry size can be inferred by the number of relocations.
- * No annotations needed.
- *
- * Note I also tried to create per-entry symbols at the time of creation, in
- * the original [inline] asm. Unfortunately, creating uniquely named symbols
- * is trickier than one might think, especially with Clang inline asm. I
- * eventually just gave up trying to make that work, in favor of using
- * ANNOTATE_DATA_SPECIAL and creating the symbols here after the fact.
- */
- static int create_fake_symbols(struct elf *elf)
- {
- struct section *sec;
- struct reloc *reloc;
- /*
- * 1) Make symbols for all the ANNOTATE_DATA_SPECIAL entries:
- */
- sec = find_section_by_name(elf, ".discard.annotate_data");
- if (!sec || !sec->rsec)
- return 0;
- for_each_reloc(sec->rsec, reloc) {
- unsigned long offset, size;
- struct reloc *next_reloc;
- if (annotype(elf, sec, reloc) != ANNOTYPE_DATA_SPECIAL)
- continue;
- offset = reloc_addend(reloc);
- size = 0;
- next_reloc = reloc;
- for_each_reloc_continue(sec->rsec, next_reloc) {
- if (annotype(elf, sec, next_reloc) != ANNOTYPE_DATA_SPECIAL ||
- next_reloc->sym->sec != reloc->sym->sec)
- continue;
- size = reloc_addend(next_reloc) - offset;
- break;
- }
- if (!size)
- size = sec_size(reloc->sym->sec) - offset;
- if (create_fake_symbol(elf, reloc->sym->sec, offset, size))
- return -1;
- }
- /*
- * 2) Make symbols for sh_entsize, and simple arrays of pointers:
- */
- for_each_sec(elf, sec) {
- unsigned int entry_size;
- unsigned long offset;
- if (!is_special_section(sec) || find_symbol_by_offset(sec, 0))
- continue;
- if (!sec->rsec) {
- ERROR("%s: missing special section relocations", sec->name);
- return -1;
- }
- entry_size = sec->sh.sh_entsize;
- if (!entry_size) {
- entry_size = arch_reloc_size(sec->rsec->relocs);
- if (sec_size(sec) != entry_size * sec_num_entries(sec->rsec)) {
- ERROR("%s: missing special section entsize or annotations", sec->name);
- return -1;
- }
- }
- for (offset = 0; offset < sec_size(sec); offset += entry_size) {
- if (create_fake_symbol(elf, sec, offset, entry_size))
- return -1;
- }
- }
- return 0;
- }
- /* Keep a special section entry if it references an included function */
- static bool should_keep_special_sym(struct elf *elf, struct symbol *sym)
- {
- struct reloc *reloc;
- if (is_sec_sym(sym) || !sym->sec->rsec)
- return false;
- sym_for_each_reloc(elf, sym, reloc) {
- if (convert_reloc_sym(elf, reloc))
- continue;
- if (is_func_sym(reloc->sym) && reloc->sym->included)
- return true;
- }
- return false;
- }
- /*
- * Klp relocations aren't allowed for __jump_table and .static_call_sites if
- * the referenced symbol lives in a kernel module, because such klp relocs may
- * be applied after static branch/call init, resulting in code corruption.
- *
- * Validate a special section entry to avoid that. Note that an inert
- * tracepoint or pr_debug() is harmless enough, in that case just skip the
- * entry and print a warning. Otherwise, return an error.
- *
- * TODO: This is only a temporary limitation which will be fixed when livepatch
- * adds support for submodules: fully self-contained modules which are embedded
- * in the top-level livepatch module's data and which can be loaded on demand
- * when their corresponding to-be-patched module gets loaded. Then klp relocs
- * can be retired.
- *
- * Return:
- * -1: error: validation failed
- * 1: warning: disabled tracepoint or pr_debug()
- * 0: success
- */
- static int validate_special_section_klp_reloc(struct elfs *e, struct symbol *sym)
- {
- bool static_branch = !strcmp(sym->sec->name, "__jump_table");
- bool static_call = !strcmp(sym->sec->name, ".static_call_sites");
- const char *code_sym = NULL;
- unsigned long code_offset = 0;
- struct reloc *reloc;
- int ret = 0;
- if (!static_branch && !static_call)
- return 0;
- sym_for_each_reloc(e->patched, sym, reloc) {
- const char *sym_modname;
- struct export *export;
- if (convert_reloc_sym(e->patched, reloc))
- continue;
- /* Static branch/call keys are always STT_OBJECT */
- if (reloc->sym->type != STT_OBJECT) {
- /* Save code location which can be printed below */
- if (reloc->sym->type == STT_FUNC && !code_sym) {
- code_sym = reloc->sym->name;
- code_offset = reloc_addend(reloc);
- }
- continue;
- }
- if (!klp_reloc_needed(reloc))
- continue;
- export = find_export(reloc->sym);
- if (export) {
- sym_modname = export->mod;
- } else {
- sym_modname = find_modname(e);
- if (!sym_modname)
- return -1;
- }
- /* vmlinux keys are ok */
- if (!strcmp(sym_modname, "vmlinux"))
- continue;
- if (!code_sym)
- code_sym = "<unknown>";
- if (static_branch) {
- if (strstarts(reloc->sym->name, "__tracepoint_")) {
- WARN("%s: disabling unsupported tracepoint %s",
- code_sym, reloc->sym->name + 13);
- ret = 1;
- continue;
- }
- if (strstr(reloc->sym->name, "__UNIQUE_ID_ddebug_")) {
- WARN("%s: disabling unsupported pr_debug()",
- code_sym);
- ret = 1;
- continue;
- }
- ERROR("%s+0x%lx: unsupported static branch key %s. Use static_key_enabled() instead",
- code_sym, code_offset, reloc->sym->name);
- return -1;
- }
- /* static call */
- if (strstarts(reloc->sym->name, "__SCK__tp_func_")) {
- ret = 1;
- continue;
- }
- ERROR("%s()+0x%lx: unsupported static call key %s. Use KLP_STATIC_CALL() instead",
- code_sym, code_offset, reloc->sym->name);
- return -1;
- }
- return ret;
- }
- static int clone_special_section(struct elfs *e, struct section *patched_sec)
- {
- struct symbol *patched_sym;
- /*
- * Extract all special section symbols (and their dependencies) which
- * reference included functions.
- */
- sec_for_each_sym(patched_sec, patched_sym) {
- int ret;
- if (!is_object_sym(patched_sym))
- continue;
- if (!should_keep_special_sym(e->patched, patched_sym))
- continue;
- ret = validate_special_section_klp_reloc(e, patched_sym);
- if (ret < 0)
- return -1;
- if (ret > 0)
- continue;
- if (!clone_symbol(e, patched_sym, true))
- return -1;
- }
- return 0;
- }
- /* Extract only the needed bits from special sections */
- static int clone_special_sections(struct elfs *e)
- {
- struct section *patched_sec;
- for_each_sec(e->patched, patched_sec) {
- if (is_special_section(patched_sec)) {
- if (clone_special_section(e, patched_sec))
- return -1;
- }
- }
- return 0;
- }
- /*
- * Create .init.klp_objects and .init.klp_funcs sections which are intermediate
- * sections provided as input to the patch module's init code for building the
- * klp_patch, klp_object and klp_func structs for the livepatch API.
- */
- static int create_klp_sections(struct elfs *e)
- {
- size_t obj_size = sizeof(struct klp_object_ext);
- size_t func_size = sizeof(struct klp_func_ext);
- struct section *obj_sec, *funcs_sec, *str_sec;
- struct symbol *funcs_sym, *str_sym, *sym;
- char sym_name[SYM_NAME_LEN];
- unsigned int nr_funcs = 0;
- const char *modname;
- void *obj_data;
- s64 addend;
- obj_sec = elf_create_section_pair(e->out, KLP_OBJECTS_SEC, obj_size, 0, 0);
- if (!obj_sec)
- return -1;
- funcs_sec = elf_create_section_pair(e->out, KLP_FUNCS_SEC, func_size, 0, 0);
- if (!funcs_sec)
- return -1;
- funcs_sym = elf_create_section_symbol(e->out, funcs_sec);
- if (!funcs_sym)
- return -1;
- str_sec = elf_create_section(e->out, KLP_STRINGS_SEC, 0, 0,
- SHT_PROGBITS, 1,
- SHF_ALLOC | SHF_STRINGS | SHF_MERGE);
- if (!str_sec)
- return -1;
- if (elf_add_string(e->out, str_sec, "") == -1)
- return -1;
- str_sym = elf_create_section_symbol(e->out, str_sec);
- if (!str_sym)
- return -1;
- /* allocate klp_object_ext */
- obj_data = elf_add_data(e->out, obj_sec, NULL, obj_size);
- if (!obj_data)
- return -1;
- modname = find_modname(e);
- if (!modname)
- return -1;
- /* klp_object_ext.name */
- if (strcmp(modname, "vmlinux")) {
- addend = elf_add_string(e->out, str_sec, modname);
- if (addend == -1)
- return -1;
- if (!elf_create_reloc(e->out, obj_sec,
- offsetof(struct klp_object_ext, name),
- str_sym, addend, R_ABS64))
- return -1;
- }
- /* klp_object_ext.funcs */
- if (!elf_create_reloc(e->out, obj_sec, offsetof(struct klp_object_ext, funcs),
- funcs_sym, 0, R_ABS64))
- return -1;
- for_each_sym(e->out, sym) {
- unsigned long offset = nr_funcs * func_size;
- unsigned long sympos;
- void *func_data;
- if (!is_func_sym(sym) || sym->cold || !sym->clone || !sym->clone->changed)
- continue;
- /* allocate klp_func_ext */
- func_data = elf_add_data(e->out, funcs_sec, NULL, func_size);
- if (!func_data)
- return -1;
- /* klp_func_ext.old_name */
- addend = elf_add_string(e->out, str_sec, sym->clone->twin->name);
- if (addend == -1)
- return -1;
- if (!elf_create_reloc(e->out, funcs_sec,
- offset + offsetof(struct klp_func_ext, old_name),
- str_sym, addend, R_ABS64))
- return -1;
- /* klp_func_ext.new_func */
- if (!elf_create_reloc(e->out, funcs_sec,
- offset + offsetof(struct klp_func_ext, new_func),
- sym, 0, R_ABS64))
- return -1;
- /* klp_func_ext.sympos */
- BUILD_BUG_ON(sizeof(sympos) != sizeof_field(struct klp_func_ext, sympos));
- sympos = find_sympos(e->orig, sym->clone->twin);
- if (sympos == ULONG_MAX)
- return -1;
- memcpy(func_data + offsetof(struct klp_func_ext, sympos), &sympos,
- sizeof_field(struct klp_func_ext, sympos));
- nr_funcs++;
- }
- /* klp_object_ext.nr_funcs */
- BUILD_BUG_ON(sizeof(nr_funcs) != sizeof_field(struct klp_object_ext, nr_funcs));
- memcpy(obj_data + offsetof(struct klp_object_ext, nr_funcs), &nr_funcs,
- sizeof_field(struct klp_object_ext, nr_funcs));
- /*
- * Find callback pointers created by KLP_PRE_PATCH_CALLBACK() and
- * friends, and add them to the klp object.
- */
- if (snprintf_check(sym_name, SYM_NAME_LEN, KLP_PRE_PATCH_PREFIX "%s", modname))
- return -1;
- sym = find_symbol_by_name(e->out, sym_name);
- if (sym) {
- struct reloc *reloc;
- reloc = find_reloc_by_dest(e->out, sym->sec, sym->offset);
- if (!elf_create_reloc(e->out, obj_sec,
- offsetof(struct klp_object_ext, callbacks) +
- offsetof(struct klp_callbacks, pre_patch),
- reloc->sym, reloc_addend(reloc), R_ABS64))
- return -1;
- }
- if (snprintf_check(sym_name, SYM_NAME_LEN, KLP_POST_PATCH_PREFIX "%s", modname))
- return -1;
- sym = find_symbol_by_name(e->out, sym_name);
- if (sym) {
- struct reloc *reloc;
- reloc = find_reloc_by_dest(e->out, sym->sec, sym->offset);
- if (!elf_create_reloc(e->out, obj_sec,
- offsetof(struct klp_object_ext, callbacks) +
- offsetof(struct klp_callbacks, post_patch),
- reloc->sym, reloc_addend(reloc), R_ABS64))
- return -1;
- }
- if (snprintf_check(sym_name, SYM_NAME_LEN, KLP_PRE_UNPATCH_PREFIX "%s", modname))
- return -1;
- sym = find_symbol_by_name(e->out, sym_name);
- if (sym) {
- struct reloc *reloc;
- reloc = find_reloc_by_dest(e->out, sym->sec, sym->offset);
- if (!elf_create_reloc(e->out, obj_sec,
- offsetof(struct klp_object_ext, callbacks) +
- offsetof(struct klp_callbacks, pre_unpatch),
- reloc->sym, reloc_addend(reloc), R_ABS64))
- return -1;
- }
- if (snprintf_check(sym_name, SYM_NAME_LEN, KLP_POST_UNPATCH_PREFIX "%s", modname))
- return -1;
- sym = find_symbol_by_name(e->out, sym_name);
- if (sym) {
- struct reloc *reloc;
- reloc = find_reloc_by_dest(e->out, sym->sec, sym->offset);
- if (!elf_create_reloc(e->out, obj_sec,
- offsetof(struct klp_object_ext, callbacks) +
- offsetof(struct klp_callbacks, post_unpatch),
- reloc->sym, reloc_addend(reloc), R_ABS64))
- return -1;
- }
- return 0;
- }
- /*
- * Copy all .modinfo import_ns= tags to ensure all namespaced exported symbols
- * can be accessed via normal relocs.
- */
- static int copy_import_ns(struct elfs *e)
- {
- struct section *patched_sec, *out_sec = NULL;
- char *import_ns, *data_end;
- patched_sec = find_section_by_name(e->patched, ".modinfo");
- if (!patched_sec)
- return 0;
- import_ns = patched_sec->data->d_buf;
- if (!import_ns)
- return 0;
- for (data_end = import_ns + sec_size(patched_sec);
- import_ns < data_end;
- import_ns += strlen(import_ns) + 1) {
- import_ns = memmem(import_ns, data_end - import_ns, "import_ns=", 10);
- if (!import_ns)
- return 0;
- if (!out_sec) {
- out_sec = find_section_by_name(e->out, ".modinfo");
- if (!out_sec) {
- out_sec = elf_create_section(e->out, ".modinfo", 0,
- patched_sec->sh.sh_entsize,
- patched_sec->sh.sh_type,
- patched_sec->sh.sh_addralign,
- patched_sec->sh.sh_flags);
- if (!out_sec)
- return -1;
- }
- }
- if (!elf_add_data(e->out, out_sec, import_ns, strlen(import_ns) + 1))
- return -1;
- }
- return 0;
- }
- int cmd_klp_diff(int argc, const char **argv)
- {
- struct elfs e = {0};
- argc = parse_options(argc, argv, klp_diff_options, klp_diff_usage, 0);
- if (argc != 3)
- usage_with_options(klp_diff_usage, klp_diff_options);
- objname = argv[0];
- e.orig = elf_open_read(argv[0], O_RDONLY);
- e.patched = elf_open_read(argv[1], O_RDONLY);
- e.out = NULL;
- if (!e.orig || !e.patched)
- return -1;
- if (read_exports())
- return -1;
- if (read_sym_checksums(e.orig))
- return -1;
- if (read_sym_checksums(e.patched))
- return -1;
- if (correlate_symbols(&e))
- return -1;
- if (mark_changed_functions(&e))
- return 0;
- e.out = elf_create_file(&e.orig->ehdr, argv[2]);
- if (!e.out)
- return -1;
- /*
- * Special section fake symbols are needed so that individual special
- * section entries can be extracted by clone_special_sections().
- *
- * Note the fake symbols are also needed by clone_included_functions()
- * because __WARN_printf() call sites add references to bug table
- * entries in the calling functions.
- */
- if (create_fake_symbols(e.patched))
- return -1;
- if (clone_included_functions(&e))
- return -1;
- if (clone_special_sections(&e))
- return -1;
- if (create_klp_sections(&e))
- return -1;
- if (copy_import_ns(&e))
- return -1;
- if (elf_write(e.out))
- return -1;
- return elf_close(e.out);
- }
|