// SPDX-License-Identifier: GPL-2.0-or-later/* * Read the intermediate KLP reloc/symbol representations created by klp diff * and convert them to the proper format required by livepatch.  This needs to * run last to avoid linker wreckage.  Linkers don't tend to handle the "two * rela sections for a single base section" case very well, nor do they like * SHN_LIVEPATCH. * * This is the final tool in the livepatch module generation pipeline: * *   kernel builds -> objtool klp diff -> module link -> objtool klp post-link */ #include <fcntl.h>#include <gelf.h>#include <objtool/objtool.h>#include <objtool/warn.h>#include <objtool/klp.h>#include <objtool/util.h>#include <linux/livepatch_external.h> static int fix_klp_reloc_sec(struct elf *elf, struct section *symtab,			     struct section *klp_relocs){	/* section format: __klp_relocs.sec_objname */	const char *sec_objname = klp_relocs->name + strlen(KLP_RELOCS_SEC "."); 	for (int i = 0; i < sec_size(klp_relocs) / sizeof(struct klp_reloc); i++) {		struct klp_reloc *klp_reloc;		unsigned long klp_reloc_off;		struct section *sec, *tmp, *klp_rsec;		unsigned long offset;		struct reloc *reloc;		char rsec_name[SEC_NAME_LEN];		u64 addend;		struct symbol *sym, *klp_sym; 		klp_reloc_off = i * sizeof(*klp_reloc);		klp_reloc = klp_relocs->data->d_buf + klp_reloc_off; 		/*		 * Read __klp_relocs[i]:		 */ 		/* klp_reloc.sec_offset */		reloc = find_reloc_by_dest(elf, klp_relocs,					   klp_reloc_off + offsetof(struct klp_reloc, offset));		if (!reloc) {			ERROR("malformed %s section", klp_relocs->name);			return -1;		} 		sec = reloc->sym->sec;		offset = reloc_addend(reloc); 		/* klp_reloc.sym */		reloc = find_reloc_by_dest(elf, klp_relocs,					   klp_reloc_off + offsetof(struct klp_reloc, sym));		if (!reloc) {			ERROR("malformed %s section", klp_relocs->name);			return -1;		} 		klp_sym = reloc->sym;		addend = reloc_addend(reloc); 		/*		 * Create the KLP rela:		 */ 		/* section format: .klp.rela.sec_objname.section_name */		if (snprintf_check(rsec_name, SEC_NAME_LEN,				   KLP_RELOC_SEC_PREFIX "%s.%s",				   sec_objname, sec->name))			return -1; 		klp_rsec = find_section_by_name(elf, rsec_name);		if (!klp_rsec) {			klp_rsec = elf_create_section(elf, rsec_name, 0,						      elf_rela_size(elf),						      SHT_RELA, elf_addr_size(elf),						      SHF_ALLOC | SHF_INFO_LINK | SHF_RELA_LIVEPATCH);			if (!klp_rsec)				return -1; 			klp_rsec->sh.sh_link = symtab->idx;			klp_rsec->sh.sh_info = sec->idx;			klp_rsec->base = sec;		} 		tmp = sec->rsec;		sec->rsec = klp_rsec;		if (!elf_create_reloc(elf, sec, offset, klp_sym, addend, klp_reloc->type))			return -1;		sec->rsec = tmp; 		/*		 * Fix up the corresponding KLP symbol:		 */ 		klp_sym->sym.st_shndx = SHN_LIVEPATCH;		if (!gelf_update_sym(symtab->data, klp_sym->idx, &klp_sym->sym)) {			ERROR_ELF("gelf_update_sym");			return -1;		} 		/*		 * Disable the original non-KLP reloc by converting it to R_*_NONE:		 */ 		reloc = find_reloc_by_dest(elf, sec, offset);		sym = reloc->sym;		sym->sym.st_shndx = SHN_LIVEPATCH;		set_reloc_type(elf, reloc, 0);		if (!gelf_update_sym(symtab->data, sym->idx, &sym->sym)) {			ERROR_ELF("gelf_update_sym");			return -1;		}	} 	return 0;} static int fix_klp_relocs(struct elf *elf){	struct section *symtab, *sec; 	symtab = find_section_by_name(elf, ".symtab");	if (!symtab) {		ERROR("missing .symtab");		return -1;	} 	for_each_sec(elf, sec) {		if (strncmp(sec->name, KLP_RELOCS_SEC ".",			    strlen(KLP_RELOCS_SEC ".")))			continue; 		if (fix_klp_reloc_sec(elf, symtab, sec))			return -1;	} 	return 0;} /* * This runs on the livepatch module after all other linking has been done.  It * converts the intermediate __klp_relocs.* sections into proper KLP relocs to * be processed by livepatch.  This needs to run last to avoid linker wreckage. * Linkers don't tend to handle the "two rela sections for a single base * section" case very well, nor do they appreciate SHN_LIVEPATCH. */int cmd_klp_post_link(int argc, const char **argv){	struct elf *elf; 	argc--;	argv++; 	if (argc != 1) {		fprintf(stderr, "%d\n", argc);		fprintf(stderr, "usage: objtool link <file.ko>\n");		return -1;	} 	elf = elf_open_read(argv[0], O_RDWR);	if (!elf)		return -1; 	if (fix_klp_relocs(elf))		return -1; 	if (elf_write(elf))		return -1; 	return elf_close(elf);}