RepoMirrors/kpatch
1
0
Fork 0
mirror of https://github.com/dynup/kpatch synced 2024-12-11 16:04:40 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #180 from ryanmiao/print_funcs

Browse Source 
export funcs via sysfs
This commit is contained in:
Seth Jennings 2014-05-27 23:24:49 -05:00
commit 2225bd83cc
1 changed files with 128 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

129
kmod/patch/kpatch-patch-hook.c
View File

@ -22,6 +22,7 @@
#include <linux/module.h>
#include <linux/printk.h>
#include <linux/slab.h>
#include <linux/kallsyms.h>
#include "kpatch.h"
static bool replace;
@ -33,6 +34,15 @@ extern char __kpatch_dynrelas, __kpatch_dynrelas_end;
static struct kpatch_module kpmod;
static struct kobject *patch_kobj;
static struct kobject *functions_kobj;
struct kpatch_func_obj {
struct kobject kobj;
struct kpatch_patch *patch;
char name[KSYM_NAME_LEN];
};
static struct kpatch_func_obj **funcs = NULL;
static ssize_t patch_enabled_show(struct kobject *kobj,
struct kobj_attribute *attr, char *buf)
@ -67,12 +77,83 @@ static ssize_t patch_enabled_store(struct kobject *kobj,
return count;
}
static struct kobj_attribute patch_enabled_attr =
__ATTR(enabled, 0644, patch_enabled_show, patch_enabled_store);
static ssize_t func_old_addr_show(struct kobject *kobj,
struct kobj_attribute *attr, char *buf)
{
struct kpatch_func_obj *func =
container_of(kobj, struct kpatch_func_obj, kobj);
return sprintf(buf, "0x%lx\n", func->patch->old_addr);
}
static ssize_t func_new_addr_show(struct kobject *kobj,
struct kobj_attribute *attr, char *buf)
{
struct kpatch_func_obj *func =
container_of(kobj, struct kpatch_func_obj, kobj);
return sprintf(buf, "0x%lx\n", func->patch->new_addr);
}
static struct kobj_attribute old_addr_attr =
__ATTR(old_addr, S_IRUGO, func_old_addr_show, NULL);
static struct kobj_attribute new_addr_attr =
__ATTR(new_addr, S_IRUGO, func_new_addr_show, NULL);
static void func_kobj_free(struct kobject *kobj)
{
struct kpatch_func_obj *func =
container_of(kobj, struct kpatch_func_obj, kobj);
kfree(func);
}
static struct attribute *func_kobj_attrs[] = {
&old_addr_attr.attr,
&new_addr_attr.attr,
NULL,
};
static ssize_t func_kobj_show(struct kobject *kobj,
struct attribute *attr, char *buf)
{
struct kobj_attribute *func_attr =
container_of(attr, struct kobj_attribute, attr);
return func_attr->show(kobj, func_attr, buf);
}
static const struct sysfs_ops func_sysfs_ops = {
.show = func_kobj_show,
};
static struct kobj_type func_ktype = {
.release = func_kobj_free,
.sysfs_ops = &func_sysfs_ops,
.default_attrs = func_kobj_attrs,
};
static struct kpatch_func_obj *func_kobj_alloc(void)
{
struct kpatch_func_obj *func;
func = kzalloc(sizeof(*func), GFP_KERNEL);
if (!func)
return NULL;
kobject_init(&func->kobj, &func_ktype);
return func;
}
static int __init patch_init(void)
{
int ret;
int i = 0;
struct kpatch_func_obj *func = NULL;
kpmod.mod = THIS_MODULE;
kpmod.patches = (struct kpatch_patch *)&__kpatch_patches;
@ -82,6 +163,13 @@ static int __init patch_init(void)
kpmod.dynrelas_nr = (&__kpatch_dynrelas_end - &__kpatch_dynrelas) /
sizeof(*kpmod.dynrelas);
funcs = kzalloc(kpmod.patches_nr * sizeof(struct kpatch_func_obj*),
GFP_KERNEL);
if (!funcs) {
ret = -ENOMEM;
goto err_ret;
}
patch_kobj = kobject_create_and_add(THIS_MODULE->name,
kpatch_patches_kobj);
if (!patch_kobj) {
@ -93,23 +181,62 @@ static int __init patch_init(void)
if (ret)
goto err_put;
functions_kobj = kobject_create_and_add("functions", patch_kobj);
if (!functions_kobj) {
ret = -ENOMEM;
goto err_sysfs;
}
for (i = 0; i < kpmod.patches_nr; i++) {
func = func_kobj_alloc();
if (!func) {
ret = -ENOMEM;
goto err_sysfs2;
}
funcs[i] = func;
sprint_symbol_no_offset(func->name, kpmod.patches[i].old_addr);
ret = kobject_add(&func->kobj, functions_kobj,
"%s", func->name);
if (ret)
goto err_sysfs2;
func->patch = &kpmod.patches[i];
}
ret = kpatch_register(&kpmod, replace);
if (ret)
goto err_sysfs;
goto err_sysfs2;
return 0;
err_sysfs2:
for (i = 0; i < kpmod.patches_nr; i++) {
if (funcs[i] != NULL)
kobject_put(&funcs[i]->kobj);
}
kobject_put(functions_kobj);
err_sysfs:
sysfs_remove_file(patch_kobj, &patch_enabled_attr.attr);
err_put:
kobject_put(patch_kobj);
err_free:
kfree(funcs);
err_ret:
return ret;
}
static void __exit patch_exit(void)
{
int i;
WARN_ON(kpmod.enabled);
for (i = 0; i < kpmod.patches_nr; i++) {
kobject_put(&funcs[i]->kobj);
}
kfree(funcs);
kobject_put(functions_kobj);
sysfs_remove_file(patch_kobj, &patch_enabled_attr.attr);
kobject_put(patch_kobj);
}