119 lines
2.2 KiB
C
119 lines
2.2 KiB
C
|
|
// SPDX-License-Identifier: GPL-2.0-or-later
|
||
|
|
/*
|
||
|
|
* livepatch-sample.c - Kernel Live Patching Sample Module
|
||
|
|
*
|
||
|
|
* Copyright (C) 2014 Seth Jennings <sjenning@redhat.com>
|
||
|
|
*/
|
||
|
|
|
||
|
|
#define pr_fmt(fmt) KBUILD_MODNAME ": " fmt
|
||
|
|
|
||
|
|
#include <linux/module.h>
|
||
|
|
#include <linux/kernel.h>
|
||
|
|
#include <linux/livepatch.h>
|
||
|
|
#ifdef CONFIG_PPC64
|
||
|
|
#include <asm/code-patching.h>
|
||
|
|
#endif
|
||
|
|
|
||
|
|
/*
|
||
|
|
* This (dumb) live patch overrides the function that prints the
|
||
|
|
* kernel boot cmdline when /proc/cmdline is read.
|
||
|
|
*
|
||
|
|
* Example:
|
||
|
|
*
|
||
|
|
* $ cat /proc/cmdline
|
||
|
|
* <your cmdline>
|
||
|
|
*
|
||
|
|
* $ insmod livepatch-sample.ko
|
||
|
|
* $ cat /proc/cmdline
|
||
|
|
* this has been live patched
|
||
|
|
*
|
||
|
|
* $ echo 0 > /sys/kernel/livepatch/livepatch_sample/enabled
|
||
|
|
* $ cat /proc/cmdline
|
||
|
|
* <your cmdline>
|
||
|
|
*/
|
||
|
|
|
||
|
|
#include <linux/seq_file.h>
|
||
|
|
|
||
|
|
#ifdef CONFIG_LIVEPATCH_STOP_MACHINE_CONSISTENCY
|
||
|
|
void load_hook(void)
|
||
|
|
{
|
||
|
|
pr_info("load_hook\n");
|
||
|
|
}
|
||
|
|
|
||
|
|
void unload_hook(void)
|
||
|
|
{
|
||
|
|
pr_info("unload_hook\n");
|
||
|
|
}
|
||
|
|
|
||
|
|
static struct klp_hook hooks_load[] = {
|
||
|
|
{
|
||
|
|
.hook = load_hook
|
||
|
|
}, { }
|
||
|
|
};
|
||
|
|
|
||
|
|
static struct klp_hook hooks_unload[] = {
|
||
|
|
{
|
||
|
|
.hook = unload_hook
|
||
|
|
}, { }
|
||
|
|
};
|
||
|
|
#endif
|
||
|
|
|
||
|
|
static int livepatch_cmdline_proc_show(struct seq_file *m, void *v)
|
||
|
|
{
|
||
|
|
seq_printf(m, "%s\n", "this has been live patched");
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
static struct klp_func funcs[] = {
|
||
|
|
{
|
||
|
|
#ifdef CONFIG_PPC64
|
||
|
|
.old_name = ".cmdline_proc_show",
|
||
|
|
#else
|
||
|
|
.old_name = "cmdline_proc_show",
|
||
|
|
#endif
|
||
|
|
.new_func = livepatch_cmdline_proc_show,
|
||
|
|
}, { }
|
||
|
|
};
|
||
|
|
|
||
|
|
static struct klp_object objs[] = {
|
||
|
|
{
|
||
|
|
/* name being NULL means vmlinux */
|
||
|
|
.funcs = funcs,
|
||
|
|
#ifdef CONFIG_LIVEPATCH_STOP_MACHINE_CONSISTENCY
|
||
|
|
.hooks_load = hooks_load,
|
||
|
|
.hooks_unload = hooks_unload,
|
||
|
|
#endif
|
||
|
|
}, { }
|
||
|
|
};
|
||
|
|
|
||
|
|
static struct klp_patch patch = {
|
||
|
|
.mod = THIS_MODULE,
|
||
|
|
.objs = objs,
|
||
|
|
};
|
||
|
|
|
||
|
|
static int livepatch_init(void)
|
||
|
|
{
|
||
|
|
#ifdef CONFIG_PPC64
|
||
|
|
patch.objs[0].funcs[0].new_func =
|
||
|
|
(void *)ppc_function_entry((void *)livepatch_cmdline_proc_show);
|
||
|
|
#endif
|
||
|
|
|
||
|
|
#ifdef CONFIG_LIVEPATCH_PER_TASK_CONSISTENCY
|
||
|
|
return klp_enable_patch(&patch);
|
||
|
|
#elif defined(CONFIG_LIVEPATCH_STOP_MACHINE_CONSISTENCY)
|
||
|
|
return klp_register_patch(&patch);
|
||
|
|
#endif
|
||
|
|
}
|
||
|
|
|
||
|
|
static void livepatch_exit(void)
|
||
|
|
{
|
||
|
|
#ifdef CONFIG_LIVEPATCH_STOP_MACHINE_CONSISTENCY
|
||
|
|
WARN_ON(klp_unregister_patch(&patch));
|
||
|
|
#endif
|
||
|
|
}
|
||
|
|
|
||
|
|
module_init(livepatch_init);
|
||
|
|
module_exit(livepatch_exit);
|
||
|
|
MODULE_LICENSE("GPL");
|
||
|
|
MODULE_INFO(livepatch, "Y");
|