// SPDX-License-Identifier: GPL-2.0-or-later

/*

* Module livepatch support

*

* Copyright (C) 2016 Jessica Yu <jeyu@redhat.com>

*/

#include <linux/module.h>

#include <linux/string.h>

#include <linux/slab.h>

#include "internal.h"

/*

* Persist ELF information about a module. Copy the ELF header,

* section header table, section string table, and symtab section

* index from info to mod->klp_info.

*/

int copy_module_elf(struct module *mod, struct load_info *info)

{

unsigned int size, symndx;

int ret;

size = sizeof(*mod->klp_info);

mod->klp_info = kmalloc(size, GFP_KERNEL);

if (!mod->klp_info)

return -ENOMEM;

/* ELF header */

size = sizeof(mod->klp_info->hdr);

memcpy(&mod->klp_info->hdr, info->hdr, size);

/* ELF section header table */

size = sizeof(*info->sechdrs) * info->hdr->e_shnum;

mod->klp_info->sechdrs = kmemdup(info->sechdrs, size, GFP_KERNEL);

if (!mod->klp_info->sechdrs) {

ret = -ENOMEM;

goto free_info;

}

/* ELF section name string table */

size = info->sechdrs[info->hdr->e_shstrndx].sh_size;

mod->klp_info->secstrings = kmemdup(info->secstrings, size, GFP_KERNEL);

if (!mod->klp_info->secstrings) {

ret = -ENOMEM;

goto free_sechdrs;

}

/* ELF symbol section index */

symndx = info->index.sym;

mod->klp_info->symndx = symndx;

/*

* For livepatch modules, core_kallsyms.symtab is a complete

* copy of the original symbol table. Adjust sh_addr to point

* to core_kallsyms.symtab since the copy of the symtab in module

* init memory is freed at the end of do_init_module().

*/

mod->klp_info->sechdrs[symndx].sh_addr = (unsigned long)mod->core_kallsyms.symtab;

return 0;

free_sechdrs:

kfree(mod->klp_info->sechdrs);

free_info:

kfree(mod->klp_info);

return ret;

}

void free_module_elf(struct module *mod)

{

kfree(mod->klp_info->sechdrs);

kfree(mod->klp_info->secstrings);

kfree(mod->klp_info);

}