Re: [PATCH 06/13] tracing/kprobes: Factor out struct trace_probe

From: Masami Hiramatsu
Date: Fri Aug 09 2013 - 05:35:05 EST


(2013/08/09 17:45), Namhyung Kim wrote:
> From: Namhyung Kim <namhyung.kim@xxxxxxx>
>
> There are functions that can be shared to both of kprobes and uprobes.
> Separate common data structure to struct trace_probe and use it from
> the shared functions.
>

Acked-by: Masami Hiramatsu <masami.hiramatsu.pt@xxxxxxxxxxx>

Thank you!

> Cc: Srikar Dronamraju <srikar@xxxxxxxxxxxxxxxxxx>
> Cc: Oleg Nesterov <oleg@xxxxxxxxxx>
> Cc: zhangwei(Jovi) <jovi.zhangwei@xxxxxxxxxx>
> Cc: Arnaldo Carvalho de Melo <acme@xxxxxxxxxxxxxxxxxx>
> Signed-off-by: Namhyung Kim <namhyung@xxxxxxxxxx>
> ---
> kernel/trace/trace_kprobe.c | 396 +++++++++++++++++++++-----------------------
> kernel/trace/trace_probe.h | 20 +++
> 2 files changed, 213 insertions(+), 203 deletions(-)
>
> diff --git a/kernel/trace/trace_kprobe.c b/kernel/trace/trace_kprobe.c
> index d2af84b2d346..d3f60d29c765 100644
> --- a/kernel/trace/trace_kprobe.c
> +++ b/kernel/trace/trace_kprobe.c
> @@ -27,18 +27,12 @@
> /**
> * Kprobe event core functions
> */
> -struct trace_probe {
> +struct trace_kprobe {
> struct list_head list;
> struct kretprobe rp; /* Use rp.kp for kprobe use */
> unsigned long nhit;
> - unsigned int flags; /* For TP_FLAG_* */
> const char *symbol; /* symbol name */
> - struct ftrace_event_class class;
> - struct ftrace_event_call call;
> - struct list_head files;
> - ssize_t size; /* trace entry size */
> - unsigned int nr_args;
> - struct probe_arg args[];
> + struct trace_probe p;
> };
>
> struct event_file_link {
> @@ -46,56 +40,46 @@ struct event_file_link {
> struct list_head list;
> };
>
> -#define SIZEOF_TRACE_PROBE(n) \
> - (offsetof(struct trace_probe, args) + \
> +#define SIZEOF_TRACE_PROBE(n) \
> + (offsetof(struct trace_kprobe, p.args) + \
> (sizeof(struct probe_arg) * (n)))
>
>
> -static __kprobes bool trace_probe_is_return(struct trace_probe *tp)
> +static __kprobes bool trace_kprobe_is_return(struct trace_kprobe *tk)
> {
> - return tp->rp.handler != NULL;
> + return tk->rp.handler != NULL;
> }
>
> -static __kprobes const char *trace_probe_symbol(struct trace_probe *tp)
> +static __kprobes const char *trace_kprobe_symbol(struct trace_kprobe *tk)
> {
> - return tp->symbol ? tp->symbol : "unknown";
> + return tk->symbol ? tk->symbol : "unknown";
> }
>
> -static __kprobes unsigned long trace_probe_offset(struct trace_probe *tp)
> +static __kprobes unsigned long trace_kprobe_offset(struct trace_kprobe *tk)
> {
> - return tp->rp.kp.offset;
> + return tk->rp.kp.offset;
> }
>
> -static __kprobes bool trace_probe_is_enabled(struct trace_probe *tp)
> +static __kprobes bool trace_kprobe_has_gone(struct trace_kprobe *tk)
> {
> - return !!(tp->flags & (TP_FLAG_TRACE | TP_FLAG_PROFILE));
> + return !!(kprobe_gone(&tk->rp.kp));
> }
>
> -static __kprobes bool trace_probe_is_registered(struct trace_probe *tp)
> -{
> - return !!(tp->flags & TP_FLAG_REGISTERED);
> -}
> -
> -static __kprobes bool trace_probe_has_gone(struct trace_probe *tp)
> -{
> - return !!(kprobe_gone(&tp->rp.kp));
> -}
> -
> -static __kprobes bool trace_probe_within_module(struct trace_probe *tp,
> - struct module *mod)
> +static __kprobes bool trace_kprobe_within_module(struct trace_kprobe *tk,
> + struct module *mod)
> {
> int len = strlen(mod->name);
> - const char *name = trace_probe_symbol(tp);
> + const char *name = trace_kprobe_symbol(tk);
> return strncmp(mod->name, name, len) == 0 && name[len] == ':';
> }
>
> -static __kprobes bool trace_probe_is_on_module(struct trace_probe *tp)
> +static __kprobes bool trace_kprobe_is_on_module(struct trace_kprobe *tk)
> {
> - return !!strchr(trace_probe_symbol(tp), ':');
> + return !!strchr(trace_kprobe_symbol(tk), ':');
> }
>
> -static int register_probe_event(struct trace_probe *tp);
> -static void unregister_probe_event(struct trace_probe *tp);
> +static int register_kprobe_event(struct trace_kprobe *tk);
> +static void unregister_kprobe_event(struct trace_kprobe *tk);
>
> static DEFINE_MUTEX(probe_lock);
> static LIST_HEAD(probe_list);
> @@ -107,14 +91,14 @@ static int kretprobe_dispatcher(struct kretprobe_instance *ri,
> /*
> * Allocate new trace_probe and initialize it (including kprobes).
> */
> -static struct trace_probe *alloc_trace_probe(const char *group,
> +static struct trace_kprobe *alloc_trace_kprobe(const char *group,
> const char *event,
> void *addr,
> const char *symbol,
> unsigned long offs,
> int nargs, bool is_return)
> {
> - struct trace_probe *tp;
> + struct trace_kprobe *tp;
> int ret = -ENOMEM;
>
> tp = kzalloc(SIZEOF_TRACE_PROBE(nargs), GFP_KERNEL);
> @@ -140,9 +124,9 @@ static struct trace_probe *alloc_trace_probe(const char *group,
> goto error;
> }
>
> - tp->call.class = &tp->class;
> - tp->call.name = kstrdup(event, GFP_KERNEL);
> - if (!tp->call.name)
> + tp->p.call.class = &tp->p.class;
> + tp->p.call.name = kstrdup(event, GFP_KERNEL);
> + if (!tp->p.call.name)
> goto error;
>
> if (!group || !is_good_name(group)) {
> @@ -150,41 +134,41 @@ static struct trace_probe *alloc_trace_probe(const char *group,
> goto error;
> }
>
> - tp->class.system = kstrdup(group, GFP_KERNEL);
> - if (!tp->class.system)
> + tp->p.class.system = kstrdup(group, GFP_KERNEL);
> + if (!tp->p.class.system)
> goto error;
>
> INIT_LIST_HEAD(&tp->list);
> - INIT_LIST_HEAD(&tp->files);
> + INIT_LIST_HEAD(&tp->p.files);
> return tp;
> error:
> - kfree(tp->call.name);
> + kfree(tp->p.call.name);
> kfree(tp->symbol);
> kfree(tp);
> return ERR_PTR(ret);
> }
>
> -static void free_trace_probe(struct trace_probe *tp)
> +static void free_trace_kprobe(struct trace_kprobe *tp)
> {
> int i;
>
> - for (i = 0; i < tp->nr_args; i++)
> - traceprobe_free_probe_arg(&tp->args[i]);
> + for (i = 0; i < tp->p.nr_args; i++)
> + traceprobe_free_probe_arg(&tp->p.args[i]);
>
> - kfree(tp->call.class->system);
> - kfree(tp->call.name);
> + kfree(tp->p.call.class->system);
> + kfree(tp->p.call.name);
> kfree(tp->symbol);
> kfree(tp);
> }
>
> -static struct trace_probe *find_trace_probe(const char *event,
> - const char *group)
> +static struct trace_kprobe *find_trace_kprobe(const char *event,
> + const char *group)
> {
> - struct trace_probe *tp;
> + struct trace_kprobe *tp;
>
> list_for_each_entry(tp, &probe_list, list)
> - if (strcmp(tp->call.name, event) == 0 &&
> - strcmp(tp->call.class->system, group) == 0)
> + if (strcmp(tp->p.call.name, event) == 0 &&
> + strcmp(tp->p.call.class->system, group) == 0)
> return tp;
> return NULL;
> }
> @@ -194,7 +178,7 @@ static struct trace_probe *find_trace_probe(const char *event,
> * if the file is NULL, enable "perf" handler, or enable "trace" handler.
> */
> static int
> -enable_trace_probe(struct trace_probe *tp, struct ftrace_event_file *file)
> +enable_trace_kprobe(struct trace_kprobe *tp, struct ftrace_event_file *file)
> {
> int ret = 0;
>
> @@ -208,14 +192,14 @@ enable_trace_probe(struct trace_probe *tp, struct ftrace_event_file *file)
> }
>
> link->file = file;
> - list_add_tail_rcu(&link->list, &tp->files);
> + list_add_tail_rcu(&link->list, &tp->p.files);
>
> - tp->flags |= TP_FLAG_TRACE;
> + tp->p.flags |= TP_FLAG_TRACE;
> } else
> - tp->flags |= TP_FLAG_PROFILE;
> + tp->p.flags |= TP_FLAG_PROFILE;
>
> - if (trace_probe_is_registered(tp) && !trace_probe_has_gone(tp)) {
> - if (trace_probe_is_return(tp))
> + if (trace_probe_is_registered(&tp->p) && !trace_kprobe_has_gone(tp)) {
> + if (trace_kprobe_is_return(tp))
> ret = enable_kretprobe(&tp->rp);
> else
> ret = enable_kprobe(&tp->rp.kp);
> @@ -241,14 +225,14 @@ find_event_file_link(struct trace_probe *tp, struct ftrace_event_file *file)
> * if the file is NULL, disable "perf" handler, or disable "trace" handler.
> */
> static int
> -disable_trace_probe(struct trace_probe *tp, struct ftrace_event_file *file)
> +disable_trace_kprobe(struct trace_kprobe *tp, struct ftrace_event_file *file)
> {
> struct event_file_link *link = NULL;
> int wait = 0;
> int ret = 0;
>
> if (file) {
> - link = find_event_file_link(tp, file);
> + link = find_event_file_link(&tp->p, file);
> if (!link) {
> ret = -EINVAL;
> goto out;
> @@ -256,15 +240,15 @@ disable_trace_probe(struct trace_probe *tp, struct ftrace_event_file *file)
>
> list_del_rcu(&link->list);
> wait = 1;
> - if (!list_empty(&tp->files))
> + if (!list_empty(&tp->p.files))
> goto out;
>
> - tp->flags &= ~TP_FLAG_TRACE;
> + tp->p.flags &= ~TP_FLAG_TRACE;
> } else
> - tp->flags &= ~TP_FLAG_PROFILE;
> + tp->p.flags &= ~TP_FLAG_PROFILE;
>
> - if (!trace_probe_is_enabled(tp) && trace_probe_is_registered(tp)) {
> - if (trace_probe_is_return(tp))
> + if (!trace_probe_is_enabled(&tp->p) && trace_probe_is_registered(&tp->p)) {
> + if (trace_kprobe_is_return(tp))
> disable_kretprobe(&tp->rp);
> else
> disable_kprobe(&tp->rp.kp);
> @@ -288,33 +272,33 @@ disable_trace_probe(struct trace_probe *tp, struct ftrace_event_file *file)
> }
>
> /* Internal register function - just handle k*probes and flags */
> -static int __register_trace_probe(struct trace_probe *tp)
> +static int __register_trace_kprobe(struct trace_kprobe *tp)
> {
> int i, ret;
>
> - if (trace_probe_is_registered(tp))
> + if (trace_probe_is_registered(&tp->p))
> return -EINVAL;
>
> - for (i = 0; i < tp->nr_args; i++)
> - traceprobe_update_arg(&tp->args[i]);
> + for (i = 0; i < tp->p.nr_args; i++)
> + traceprobe_update_arg(&tp->p.args[i]);
>
> /* Set/clear disabled flag according to tp->flag */
> - if (trace_probe_is_enabled(tp))
> + if (trace_probe_is_enabled(&tp->p))
> tp->rp.kp.flags &= ~KPROBE_FLAG_DISABLED;
> else
> tp->rp.kp.flags |= KPROBE_FLAG_DISABLED;
>
> - if (trace_probe_is_return(tp))
> + if (trace_kprobe_is_return(tp))
> ret = register_kretprobe(&tp->rp);
> else
> ret = register_kprobe(&tp->rp.kp);
>
> if (ret == 0)
> - tp->flags |= TP_FLAG_REGISTERED;
> + tp->p.flags |= TP_FLAG_REGISTERED;
> else {
> pr_warning("Could not insert probe at %s+%lu: %d\n",
> - trace_probe_symbol(tp), trace_probe_offset(tp), ret);
> - if (ret == -ENOENT && trace_probe_is_on_module(tp)) {
> + trace_kprobe_symbol(tp), trace_kprobe_offset(tp), ret);
> + if (ret == -ENOENT && trace_kprobe_is_on_module(tp)) {
> pr_warning("This probe might be able to register after"
> "target module is loaded. Continue.\n");
> ret = 0;
> @@ -330,14 +314,14 @@ static int __register_trace_probe(struct trace_probe *tp)
> }
>
> /* Internal unregister function - just handle k*probes and flags */
> -static void __unregister_trace_probe(struct trace_probe *tp)
> +static void __unregister_trace_kprobe(struct trace_kprobe *tp)
> {
> - if (trace_probe_is_registered(tp)) {
> - if (trace_probe_is_return(tp))
> + if (trace_probe_is_registered(&tp->p)) {
> + if (trace_kprobe_is_return(tp))
> unregister_kretprobe(&tp->rp);
> else
> unregister_kprobe(&tp->rp.kp);
> - tp->flags &= ~TP_FLAG_REGISTERED;
> + tp->p.flags &= ~TP_FLAG_REGISTERED;
> /* Cleanup kprobe for reuse */
> if (tp->rp.kp.symbol_name)
> tp->rp.kp.addr = NULL;
> @@ -345,47 +329,47 @@ static void __unregister_trace_probe(struct trace_probe *tp)
> }
>
> /* Unregister a trace_probe and probe_event: call with locking probe_lock */
> -static int unregister_trace_probe(struct trace_probe *tp)
> +static int unregister_trace_kprobe(struct trace_kprobe *tp)
> {
> /* Enabled event can not be unregistered */
> - if (trace_probe_is_enabled(tp))
> + if (trace_probe_is_enabled(&tp->p))
> return -EBUSY;
>
> - __unregister_trace_probe(tp);
> + __unregister_trace_kprobe(tp);
> list_del(&tp->list);
> - unregister_probe_event(tp);
> + unregister_kprobe_event(tp);
>
> return 0;
> }
>
> /* Register a trace_probe and probe_event */
> -static int register_trace_probe(struct trace_probe *tp)
> +static int register_trace_kprobe(struct trace_kprobe *tp)
> {
> - struct trace_probe *old_tp;
> + struct trace_kprobe *old_tp;
> int ret;
>
> mutex_lock(&probe_lock);
>
> /* Delete old (same name) event if exist */
> - old_tp = find_trace_probe(tp->call.name, tp->call.class->system);
> + old_tp = find_trace_kprobe(tp->p.call.name, tp->p.call.class->system);
> if (old_tp) {
> - ret = unregister_trace_probe(old_tp);
> + ret = unregister_trace_kprobe(old_tp);
> if (ret < 0)
> goto end;
> - free_trace_probe(old_tp);
> + free_trace_kprobe(old_tp);
> }
>
> /* Register new event */
> - ret = register_probe_event(tp);
> + ret = register_kprobe_event(tp);
> if (ret) {
> pr_warning("Failed to register probe event(%d)\n", ret);
> goto end;
> }
>
> /* Register k*probe */
> - ret = __register_trace_probe(tp);
> + ret = __register_trace_kprobe(tp);
> if (ret < 0)
> - unregister_probe_event(tp);
> + unregister_kprobe_event(tp);
> else
> list_add_tail(&tp->list, &probe_list);
>
> @@ -395,11 +379,11 @@ end:
> }
>
> /* Module notifier call back, checking event on the module */
> -static int trace_probe_module_callback(struct notifier_block *nb,
> +static int trace_kprobe_module_callback(struct notifier_block *nb,
> unsigned long val, void *data)
> {
> struct module *mod = data;
> - struct trace_probe *tp;
> + struct trace_kprobe *tp;
> int ret;
>
> if (val != MODULE_STATE_COMING)
> @@ -408,14 +392,14 @@ static int trace_probe_module_callback(struct notifier_block *nb,
> /* Update probes on coming module */
> mutex_lock(&probe_lock);
> list_for_each_entry(tp, &probe_list, list) {
> - if (trace_probe_within_module(tp, mod)) {
> + if (trace_kprobe_within_module(tp, mod)) {
> /* Don't need to check busy - this should have gone. */
> - __unregister_trace_probe(tp);
> - ret = __register_trace_probe(tp);
> + __unregister_trace_kprobe(tp);
> + ret = __register_trace_kprobe(tp);
> if (ret)
> pr_warning("Failed to re-register probe %s on"
> "%s: %d\n",
> - tp->call.name, mod->name, ret);
> + tp->p.call.name, mod->name, ret);
> }
> }
> mutex_unlock(&probe_lock);
> @@ -423,12 +407,12 @@ static int trace_probe_module_callback(struct notifier_block *nb,
> return NOTIFY_DONE;
> }
>
> -static struct notifier_block trace_probe_module_nb = {
> - .notifier_call = trace_probe_module_callback,
> +static struct notifier_block trace_kprobe_module_nb = {
> + .notifier_call = trace_kprobe_module_callback,
> .priority = 1 /* Invoked after kprobe module callback */
> };
>
> -static int create_trace_probe(int argc, char **argv)
> +static int create_trace_kprobe(int argc, char **argv)
> {
> /*
> * Argument syntax:
> @@ -448,7 +432,7 @@ static int create_trace_probe(int argc, char **argv)
> * Type of args:
> * FETCHARG:TYPE : use TYPE instead of unsigned long.
> */
> - struct trace_probe *tp;
> + struct trace_kprobe *tp;
> int i, ret = 0;
> bool is_return = false, is_delete = false;
> char *symbol = NULL, *event = NULL, *group = NULL;
> @@ -495,16 +479,16 @@ static int create_trace_probe(int argc, char **argv)
> return -EINVAL;
> }
> mutex_lock(&probe_lock);
> - tp = find_trace_probe(event, group);
> + tp = find_trace_kprobe(event, group);
> if (!tp) {
> mutex_unlock(&probe_lock);
> pr_info("Event %s/%s doesn't exist.\n", group, event);
> return -ENOENT;
> }
> /* delete an event */
> - ret = unregister_trace_probe(tp);
> + ret = unregister_trace_kprobe(tp);
> if (ret == 0)
> - free_trace_probe(tp);
> + free_trace_kprobe(tp);
> mutex_unlock(&probe_lock);
> return ret;
> }
> @@ -551,7 +535,7 @@ static int create_trace_probe(int argc, char **argv)
> is_return ? 'r' : 'p', addr);
> event = buf;
> }
> - tp = alloc_trace_probe(group, event, addr, symbol, offset, argc,
> + tp = alloc_trace_kprobe(group, event, addr, symbol, offset, argc,
> is_return);
> if (IS_ERR(tp)) {
> pr_info("Failed to allocate trace_probe.(%d)\n",
> @@ -562,36 +546,38 @@ static int create_trace_probe(int argc, char **argv)
> /* parse arguments */
> ret = 0;
> for (i = 0; i < argc && i < MAX_TRACE_ARGS; i++) {
> + struct probe_arg *parg = &tp->p.args[i];
> +
> /* Increment count for freeing args in error case */
> - tp->nr_args++;
> + tp->p.nr_args++;
>
> /* Parse argument name */
> arg = strchr(argv[i], '=');
> if (arg) {
> *arg++ = '\0';
> - tp->args[i].name = kstrdup(argv[i], GFP_KERNEL);
> + parg->name = kstrdup(argv[i], GFP_KERNEL);
> } else {
> arg = argv[i];
> /* If argument name is omitted, set "argN" */
> snprintf(buf, MAX_EVENT_NAME_LEN, "arg%d", i + 1);
> - tp->args[i].name = kstrdup(buf, GFP_KERNEL);
> + parg->name = kstrdup(buf, GFP_KERNEL);
> }
>
> - if (!tp->args[i].name) {
> + if (!parg->name) {
> pr_info("Failed to allocate argument[%d] name.\n", i);
> ret = -ENOMEM;
> goto error;
> }
>
> - if (!is_good_name(tp->args[i].name)) {
> + if (!is_good_name(parg->name)) {
> pr_info("Invalid argument[%d] name: %s\n",
> - i, tp->args[i].name);
> + i, parg->name);
> ret = -EINVAL;
> goto error;
> }
>
> - if (traceprobe_conflict_field_name(tp->args[i].name,
> - tp->args, i)) {
> + if (traceprobe_conflict_field_name(parg->name,
> + tp->p.args, i)) {
> pr_info("Argument[%d] name '%s' conflicts with "
> "another field.\n", i, argv[i]);
> ret = -EINVAL;
> @@ -599,7 +585,7 @@ static int create_trace_probe(int argc, char **argv)
> }
>
> /* Parse fetch argument */
> - ret = traceprobe_parse_probe_arg(arg, &tp->size, &tp->args[i],
> + ret = traceprobe_parse_probe_arg(arg, &tp->p.size, parg,
> is_return, true);
> if (ret) {
> pr_info("Parse error at argument[%d]. (%d)\n", i, ret);
> @@ -607,33 +593,33 @@ static int create_trace_probe(int argc, char **argv)
> }
> }
>
> - ret = register_trace_probe(tp);
> + ret = register_trace_kprobe(tp);
> if (ret)
> goto error;
> return 0;
>
> error:
> - free_trace_probe(tp);
> + free_trace_kprobe(tp);
> return ret;
> }
>
> -static int release_all_trace_probes(void)
> +static int release_all_trace_kprobes(void)
> {
> - struct trace_probe *tp;
> + struct trace_kprobe *tp;
> int ret = 0;
>
> mutex_lock(&probe_lock);
> /* Ensure no probe is in use. */
> list_for_each_entry(tp, &probe_list, list)
> - if (trace_probe_is_enabled(tp)) {
> + if (trace_probe_is_enabled(&tp->p)) {
> ret = -EBUSY;
> goto end;
> }
> /* TODO: Use batch unregistration */
> while (!list_empty(&probe_list)) {
> - tp = list_entry(probe_list.next, struct trace_probe, list);
> - unregister_trace_probe(tp);
> - free_trace_probe(tp);
> + tp = list_entry(probe_list.next, struct trace_kprobe, list);
> + unregister_trace_kprobe(tp);
> + free_trace_kprobe(tp);
> }
>
> end:
> @@ -661,22 +647,22 @@ static void probes_seq_stop(struct seq_file *m, void *v)
>
> static int probes_seq_show(struct seq_file *m, void *v)
> {
> - struct trace_probe *tp = v;
> + struct trace_kprobe *tp = v;
> int i;
>
> - seq_printf(m, "%c", trace_probe_is_return(tp) ? 'r' : 'p');
> - seq_printf(m, ":%s/%s", tp->call.class->system, tp->call.name);
> + seq_printf(m, "%c", trace_kprobe_is_return(tp) ? 'r' : 'p');
> + seq_printf(m, ":%s/%s", tp->p.call.class->system, tp->p.call.name);
>
> if (!tp->symbol)
> seq_printf(m, " 0x%p", tp->rp.kp.addr);
> else if (tp->rp.kp.offset)
> - seq_printf(m, " %s+%u", trace_probe_symbol(tp),
> + seq_printf(m, " %s+%u", trace_kprobe_symbol(tp),
> tp->rp.kp.offset);
> else
> - seq_printf(m, " %s", trace_probe_symbol(tp));
> + seq_printf(m, " %s", trace_kprobe_symbol(tp));
>
> - for (i = 0; i < tp->nr_args; i++)
> - seq_printf(m, " %s=%s", tp->args[i].name, tp->args[i].comm);
> + for (i = 0; i < tp->p.nr_args; i++)
> + seq_printf(m, " %s=%s", tp->p.args[i].name, tp->p.args[i].comm);
> seq_printf(m, "\n");
>
> return 0;
> @@ -694,7 +680,7 @@ static int probes_open(struct inode *inode, struct file *file)
> int ret;
>
> if ((file->f_mode & FMODE_WRITE) && (file->f_flags & O_TRUNC)) {
> - ret = release_all_trace_probes();
> + ret = release_all_trace_kprobes();
> if (ret < 0)
> return ret;
> }
> @@ -706,7 +692,7 @@ static ssize_t probes_write(struct file *file, const char __user *buffer,
> size_t count, loff_t *ppos)
> {
> return traceprobe_probes_write(file, buffer, count, ppos,
> - create_trace_probe);
> + create_trace_kprobe);
> }
>
> static const struct file_operations kprobe_events_ops = {
> @@ -721,9 +707,9 @@ static const struct file_operations kprobe_events_ops = {
> /* Probes profiling interfaces */
> static int probes_profile_seq_show(struct seq_file *m, void *v)
> {
> - struct trace_probe *tp = v;
> + struct trace_kprobe *tp = v;
>
> - seq_printf(m, " %-44s %15lu %15lu\n", tp->call.name, tp->nhit,
> + seq_printf(m, " %-44s %15lu %15lu\n", tp->p.call.name, tp->nhit,
> tp->rp.kp.nmissed);
>
> return 0;
> @@ -968,7 +954,7 @@ static __kprobes void store_trace_args(int ent_size, struct trace_probe *tp,
>
> /* Kprobe handler */
> static __kprobes void
> -__kprobe_trace_func(struct trace_probe *tp, struct pt_regs *regs,
> +__kprobe_trace_func(struct trace_kprobe *tp, struct pt_regs *regs,
> struct ftrace_event_file *ftrace_file)
> {
> struct kprobe_trace_entry_head *entry;
> @@ -976,7 +962,7 @@ __kprobe_trace_func(struct trace_probe *tp, struct pt_regs *regs,
> struct ring_buffer *buffer;
> int size, dsize, pc;
> unsigned long irq_flags;
> - struct ftrace_event_call *call = &tp->call;
> + struct ftrace_event_call *call = &tp->p.call;
>
> WARN_ON(call != ftrace_file->event_call);
>
> @@ -986,8 +972,8 @@ __kprobe_trace_func(struct trace_probe *tp, struct pt_regs *regs,
> local_save_flags(irq_flags);
> pc = preempt_count();
>
> - dsize = __get_data_size(tp, regs);
> - size = sizeof(*entry) + tp->size + dsize;
> + dsize = __get_data_size(&tp->p, regs);
> + size = sizeof(*entry) + tp->p.size + dsize;
>
> event = trace_event_buffer_lock_reserve(&buffer, ftrace_file,
> call->event.type,
> @@ -997,7 +983,7 @@ __kprobe_trace_func(struct trace_probe *tp, struct pt_regs *regs,
>
> entry = ring_buffer_event_data(event);
> entry->ip = (unsigned long)tp->rp.kp.addr;
> - store_trace_args(sizeof(*entry), tp, regs, (u8 *)&entry[1], dsize);
> + store_trace_args(sizeof(*entry), &tp->p, regs, (u8 *)&entry[1], dsize);
>
> if (!filter_current_check_discard(buffer, call, entry, event))
> trace_buffer_unlock_commit_regs(buffer, event,
> @@ -1005,17 +991,17 @@ __kprobe_trace_func(struct trace_probe *tp, struct pt_regs *regs,
> }
>
> static __kprobes void
> -kprobe_trace_func(struct trace_probe *tp, struct pt_regs *regs)
> +kprobe_trace_func(struct trace_kprobe *tp, struct pt_regs *regs)
> {
> struct event_file_link *link;
>
> - list_for_each_entry_rcu(link, &tp->files, list)
> + list_for_each_entry_rcu(link, &tp->p.files, list)
> __kprobe_trace_func(tp, regs, link->file);
> }
>
> /* Kretprobe handler */
> static __kprobes void
> -__kretprobe_trace_func(struct trace_probe *tp, struct kretprobe_instance *ri,
> +__kretprobe_trace_func(struct trace_kprobe *tp, struct kretprobe_instance *ri,
> struct pt_regs *regs,
> struct ftrace_event_file *ftrace_file)
> {
> @@ -1024,7 +1010,7 @@ __kretprobe_trace_func(struct trace_probe *tp, struct kretprobe_instance *ri,
> struct ring_buffer *buffer;
> int size, pc, dsize;
> unsigned long irq_flags;
> - struct ftrace_event_call *call = &tp->call;
> + struct ftrace_event_call *call = &tp->p.call;
>
> WARN_ON(call != ftrace_file->event_call);
>
> @@ -1034,8 +1020,8 @@ __kretprobe_trace_func(struct trace_probe *tp, struct kretprobe_instance *ri,
> local_save_flags(irq_flags);
> pc = preempt_count();
>
> - dsize = __get_data_size(tp, regs);
> - size = sizeof(*entry) + tp->size + dsize;
> + dsize = __get_data_size(&tp->p, regs);
> + size = sizeof(*entry) + tp->p.size + dsize;
>
> event = trace_event_buffer_lock_reserve(&buffer, ftrace_file,
> call->event.type,
> @@ -1046,7 +1032,7 @@ __kretprobe_trace_func(struct trace_probe *tp, struct kretprobe_instance *ri,
> entry = ring_buffer_event_data(event);
> entry->func = (unsigned long)tp->rp.kp.addr;
> entry->ret_ip = (unsigned long)ri->ret_addr;
> - store_trace_args(sizeof(*entry), tp, regs, (u8 *)&entry[1], dsize);
> + store_trace_args(sizeof(*entry), &tp->p, regs, (u8 *)&entry[1], dsize);
>
> if (!filter_current_check_discard(buffer, call, entry, event))
> trace_buffer_unlock_commit_regs(buffer, event,
> @@ -1054,12 +1040,12 @@ __kretprobe_trace_func(struct trace_probe *tp, struct kretprobe_instance *ri,
> }
>
> static __kprobes void
> -kretprobe_trace_func(struct trace_probe *tp, struct kretprobe_instance *ri,
> +kretprobe_trace_func(struct trace_kprobe *tp, struct kretprobe_instance *ri,
> struct pt_regs *regs)
> {
> struct event_file_link *link;
>
> - list_for_each_entry_rcu(link, &tp->files, list)
> + list_for_each_entry_rcu(link, &tp->p.files, list)
> __kretprobe_trace_func(tp, ri, regs, link->file);
> }
>
> @@ -1147,16 +1133,18 @@ static int kprobe_event_define_fields(struct ftrace_event_call *event_call)
> {
> int ret, i;
> struct kprobe_trace_entry_head field;
> - struct trace_probe *tp = (struct trace_probe *)event_call->data;
> + struct trace_kprobe *tp = (struct trace_kprobe *)event_call->data;
>
> DEFINE_FIELD(unsigned long, ip, FIELD_STRING_IP, 0);
> /* Set argument names as fields */
> - for (i = 0; i < tp->nr_args; i++) {
> - ret = trace_define_field(event_call, tp->args[i].type->fmttype,
> - tp->args[i].name,
> - sizeof(field) + tp->args[i].offset,
> - tp->args[i].type->size,
> - tp->args[i].type->is_signed,
> + for (i = 0; i < tp->p.nr_args; i++) {
> + struct probe_arg *parg = &tp->p.args[i];
> +
> + ret = trace_define_field(event_call, parg->type->fmttype,
> + parg->name,
> + sizeof(field) + parg->offset,
> + parg->type->size,
> + parg->type->is_signed,
> FILTER_OTHER);
> if (ret)
> return ret;
> @@ -1168,17 +1156,19 @@ static int kretprobe_event_define_fields(struct ftrace_event_call *event_call)
> {
> int ret, i;
> struct kretprobe_trace_entry_head field;
> - struct trace_probe *tp = (struct trace_probe *)event_call->data;
> + struct trace_kprobe *tp = (struct trace_kprobe *)event_call->data;
>
> DEFINE_FIELD(unsigned long, func, FIELD_STRING_FUNC, 0);
> DEFINE_FIELD(unsigned long, ret_ip, FIELD_STRING_RETIP, 0);
> /* Set argument names as fields */
> - for (i = 0; i < tp->nr_args; i++) {
> - ret = trace_define_field(event_call, tp->args[i].type->fmttype,
> - tp->args[i].name,
> - sizeof(field) + tp->args[i].offset,
> - tp->args[i].type->size,
> - tp->args[i].type->is_signed,
> + for (i = 0; i < tp->p.nr_args; i++) {
> + struct probe_arg *parg = &tp->p.args[i];
> +
> + ret = trace_define_field(event_call, parg->type->fmttype,
> + parg->name,
> + sizeof(field) + parg->offset,
> + parg->type->size,
> + parg->type->is_signed,
> FILTER_OTHER);
> if (ret)
> return ret;
> @@ -1186,14 +1176,14 @@ static int kretprobe_event_define_fields(struct ftrace_event_call *event_call)
> return 0;
> }
>
> -static int __set_print_fmt(struct trace_probe *tp, char *buf, int len)
> +static int __set_print_fmt(struct trace_kprobe *tp, char *buf, int len)
> {
> int i;
> int pos = 0;
>
> const char *fmt, *arg;
>
> - if (!trace_probe_is_return(tp)) {
> + if (!trace_kprobe_is_return(tp)) {
> fmt = "(%lx)";
> arg = "REC->" FIELD_STRING_IP;
> } else {
> @@ -1206,21 +1196,21 @@ static int __set_print_fmt(struct trace_probe *tp, char *buf, int len)
>
> pos += snprintf(buf + pos, LEN_OR_ZERO, "\"%s", fmt);
>
> - for (i = 0; i < tp->nr_args; i++) {
> + for (i = 0; i < tp->p.nr_args; i++) {
> pos += snprintf(buf + pos, LEN_OR_ZERO, " %s=%s",
> - tp->args[i].name, tp->args[i].type->fmt);
> + tp->p.args[i].name, tp->p.args[i].type->fmt);
> }
>
> pos += snprintf(buf + pos, LEN_OR_ZERO, "\", %s", arg);
>
> - for (i = 0; i < tp->nr_args; i++) {
> - if (strcmp(tp->args[i].type->name, "string") == 0)
> + for (i = 0; i < tp->p.nr_args; i++) {
> + if (strcmp(tp->p.args[i].type->name, "string") == 0)
> pos += snprintf(buf + pos, LEN_OR_ZERO,
> ", __get_str(%s)",
> - tp->args[i].name);
> + tp->p.args[i].name);
> else
> pos += snprintf(buf + pos, LEN_OR_ZERO, ", REC->%s",
> - tp->args[i].name);
> + tp->p.args[i].name);
> }
>
> #undef LEN_OR_ZERO
> @@ -1229,7 +1219,7 @@ static int __set_print_fmt(struct trace_probe *tp, char *buf, int len)
> return pos;
> }
>
> -static int set_print_fmt(struct trace_probe *tp)
> +static int set_print_fmt(struct trace_kprobe *tp)
> {
> int len;
> char *print_fmt;
> @@ -1242,7 +1232,7 @@ static int set_print_fmt(struct trace_probe *tp)
>
> /* Second: actually write the @print_fmt */
> __set_print_fmt(tp, print_fmt, len + 1);
> - tp->call.print_fmt = print_fmt;
> + tp->p.call.print_fmt = print_fmt;
>
> return 0;
> }
> @@ -1251,9 +1241,9 @@ static int set_print_fmt(struct trace_probe *tp)
>
> /* Kprobe profile handler */
> static __kprobes void
> -kprobe_perf_func(struct trace_probe *tp, struct pt_regs *regs)
> +kprobe_perf_func(struct trace_kprobe *tp, struct pt_regs *regs)
> {
> - struct ftrace_event_call *call = &tp->call;
> + struct ftrace_event_call *call = &tp->p.call;
> struct kprobe_trace_entry_head *entry;
> struct hlist_head *head;
> int size, __size, dsize;
> @@ -1263,8 +1253,8 @@ kprobe_perf_func(struct trace_probe *tp, struct pt_regs *regs)
> if (hlist_empty(head))
> return;
>
> - dsize = __get_data_size(tp, regs);
> - __size = sizeof(*entry) + tp->size + dsize;
> + dsize = __get_data_size(&tp->p, regs);
> + __size = sizeof(*entry) + tp->p.size + dsize;
> size = ALIGN(__size + sizeof(u32), sizeof(u64));
> size -= sizeof(u32);
>
> @@ -1274,16 +1264,16 @@ kprobe_perf_func(struct trace_probe *tp, struct pt_regs *regs)
>
> entry->ip = (unsigned long)tp->rp.kp.addr;
> memset(&entry[1], 0, dsize);
> - store_trace_args(sizeof(*entry), tp, regs, (u8 *)&entry[1], dsize);
> + store_trace_args(sizeof(*entry), &tp->p, regs, (u8 *)&entry[1], dsize);
> perf_trace_buf_submit(entry, size, rctx, 0, 1, regs, head, NULL);
> }
>
> /* Kretprobe profile handler */
> static __kprobes void
> -kretprobe_perf_func(struct trace_probe *tp, struct kretprobe_instance *ri,
> +kretprobe_perf_func(struct trace_kprobe *tp, struct kretprobe_instance *ri,
> struct pt_regs *regs)
> {
> - struct ftrace_event_call *call = &tp->call;
> + struct ftrace_event_call *call = &tp->p.call;
> struct kretprobe_trace_entry_head *entry;
> struct hlist_head *head;
> int size, __size, dsize;
> @@ -1293,8 +1283,8 @@ kretprobe_perf_func(struct trace_probe *tp, struct kretprobe_instance *ri,
> if (hlist_empty(head))
> return;
>
> - dsize = __get_data_size(tp, regs);
> - __size = sizeof(*entry) + tp->size + dsize;
> + dsize = __get_data_size(&tp->p, regs);
> + __size = sizeof(*entry) + tp->p.size + dsize;
> size = ALIGN(__size + sizeof(u32), sizeof(u64));
> size -= sizeof(u32);
>
> @@ -1304,7 +1294,7 @@ kretprobe_perf_func(struct trace_probe *tp, struct kretprobe_instance *ri,
>
> entry->func = (unsigned long)tp->rp.kp.addr;
> entry->ret_ip = (unsigned long)ri->ret_addr;
> - store_trace_args(sizeof(*entry), tp, regs, (u8 *)&entry[1], dsize);
> + store_trace_args(sizeof(*entry), &tp->p, regs, (u8 *)&entry[1], dsize);
> perf_trace_buf_submit(entry, size, rctx, 0, 1, regs, head, NULL);
> }
> #endif /* CONFIG_PERF_EVENTS */
> @@ -1319,20 +1309,20 @@ static __kprobes
> int kprobe_register(struct ftrace_event_call *event,
> enum trace_reg type, void *data)
> {
> - struct trace_probe *tp = (struct trace_probe *)event->data;
> + struct trace_kprobe *tp = (struct trace_kprobe *)event->data;
> struct ftrace_event_file *file = data;
>
> switch (type) {
> case TRACE_REG_REGISTER:
> - return enable_trace_probe(tp, file);
> + return enable_trace_kprobe(tp, file);
> case TRACE_REG_UNREGISTER:
> - return disable_trace_probe(tp, file);
> + return disable_trace_kprobe(tp, file);
>
> #ifdef CONFIG_PERF_EVENTS
> case TRACE_REG_PERF_REGISTER:
> - return enable_trace_probe(tp, NULL);
> + return enable_trace_kprobe(tp, NULL);
> case TRACE_REG_PERF_UNREGISTER:
> - return disable_trace_probe(tp, NULL);
> + return disable_trace_kprobe(tp, NULL);
> case TRACE_REG_PERF_OPEN:
> case TRACE_REG_PERF_CLOSE:
> case TRACE_REG_PERF_ADD:
> @@ -1346,14 +1336,14 @@ int kprobe_register(struct ftrace_event_call *event,
> static __kprobes
> int kprobe_dispatcher(struct kprobe *kp, struct pt_regs *regs)
> {
> - struct trace_probe *tp = container_of(kp, struct trace_probe, rp.kp);
> + struct trace_kprobe *tp = container_of(kp, struct trace_kprobe, rp.kp);
>
> tp->nhit++;
>
> - if (tp->flags & TP_FLAG_TRACE)
> + if (tp->p.flags & TP_FLAG_TRACE)
> kprobe_trace_func(tp, regs);
> #ifdef CONFIG_PERF_EVENTS
> - if (tp->flags & TP_FLAG_PROFILE)
> + if (tp->p.flags & TP_FLAG_PROFILE)
> kprobe_perf_func(tp, regs);
> #endif
> return 0; /* We don't tweek kernel, so just return 0 */
> @@ -1362,14 +1352,14 @@ int kprobe_dispatcher(struct kprobe *kp, struct pt_regs *regs)
> static __kprobes
> int kretprobe_dispatcher(struct kretprobe_instance *ri, struct pt_regs *regs)
> {
> - struct trace_probe *tp = container_of(ri->rp, struct trace_probe, rp);
> + struct trace_kprobe *tp = container_of(ri->rp, struct trace_kprobe, rp);
>
> tp->nhit++;
>
> - if (tp->flags & TP_FLAG_TRACE)
> + if (tp->p.flags & TP_FLAG_TRACE)
> kretprobe_trace_func(tp, ri, regs);
> #ifdef CONFIG_PERF_EVENTS
> - if (tp->flags & TP_FLAG_PROFILE)
> + if (tp->p.flags & TP_FLAG_PROFILE)
> kretprobe_perf_func(tp, ri, regs);
> #endif
> return 0; /* We don't tweek kernel, so just return 0 */
> @@ -1383,14 +1373,14 @@ static struct trace_event_functions kprobe_funcs = {
> .trace = print_kprobe_event
> };
>
> -static int register_probe_event(struct trace_probe *tp)
> +static int register_kprobe_event(struct trace_kprobe *tp)
> {
> - struct ftrace_event_call *call = &tp->call;
> + struct ftrace_event_call *call = &tp->p.call;
> int ret;
>
> /* Initialize ftrace_event_call */
> INIT_LIST_HEAD(&call->class->fields);
> - if (trace_probe_is_return(tp)) {
> + if (trace_kprobe_is_return(tp)) {
> call->event.funcs = &kretprobe_funcs;
> call->class->define_fields = kretprobe_event_define_fields;
> } else {
> @@ -1416,11 +1406,11 @@ static int register_probe_event(struct trace_probe *tp)
> return ret;
> }
>
> -static void unregister_probe_event(struct trace_probe *tp)
> +static void unregister_kprobe_event(struct trace_kprobe *tp)
> {
> /* tp->event is unregistered in trace_remove_event_call() */
> - trace_remove_event_call(&tp->call);
> - kfree(tp->call.print_fmt);
> + trace_remove_event_call(&tp->p.call);
> + kfree(tp->p.call.print_fmt);
> }
>
> /* Make a debugfs interface for controlling probe points */
> @@ -1429,7 +1419,7 @@ static __init int init_kprobe_trace(void)
> struct dentry *d_tracer;
> struct dentry *entry;
>
> - if (register_module_notifier(&trace_probe_module_nb))
> + if (register_module_notifier(&trace_kprobe_module_nb))
> return -EINVAL;
>
> d_tracer = tracing_init_dentry();
> @@ -1469,12 +1459,12 @@ static __used int kprobe_trace_selftest_target(int a1, int a2, int a3,
> }
>
> static struct ftrace_event_file *
> -find_trace_probe_file(struct trace_probe *tp, struct trace_array *tr)
> +find_trace_probe_file(struct trace_kprobe *tp, struct trace_array *tr)
> {
> struct ftrace_event_file *file;
>
> list_for_each_entry(file, &tr->events, list)
> - if (file->event_call == &tp->call)
> + if (file->event_call == &tp->p.call)
> return file;
>
> return NULL;
> @@ -1488,7 +1478,7 @@ static __init int kprobe_trace_self_tests_init(void)
> {
> int ret, warn = 0;
> int (*target)(int, int, int, int, int, int);
> - struct trace_probe *tp;
> + struct trace_kprobe *tp;
> struct ftrace_event_file *file;
>
> target = kprobe_trace_selftest_target;
> diff --git a/kernel/trace/trace_probe.h b/kernel/trace/trace_probe.h
> index 9ac7bdf607cc..63e5da4e3073 100644
> --- a/kernel/trace/trace_probe.h
> +++ b/kernel/trace/trace_probe.h
> @@ -159,6 +159,26 @@ struct probe_arg {
> const struct fetch_type *type; /* Type of this argument */
> };
>
> +struct trace_probe {
> + unsigned int flags; /* For TP_FLAG_* */
> + struct ftrace_event_class class;
> + struct ftrace_event_call call;
> + struct list_head files;
> + ssize_t size; /* trace entry size */
> + unsigned int nr_args;
> + struct probe_arg args[];
> +};
> +
> +static inline bool trace_probe_is_enabled(struct trace_probe *tp)
> +{
> + return !!(tp->flags & (TP_FLAG_TRACE | TP_FLAG_PROFILE));
> +}
> +
> +static inline bool trace_probe_is_registered(struct trace_probe *tp)
> +{
> + return !!(tp->flags & TP_FLAG_REGISTERED);
> +}
> +
> #define FETCH_FUNC_NAME(method, type) fetch_##method##_##type
>
> #define DECLARE_FETCH_FUNC(method, type) \
>


--
Masami HIRAMATSU
IT Management Research Dept. Linux Technology Center
Hitachi, Ltd., Yokohama Research Laboratory
E-mail: masami.hiramatsu.pt@xxxxxxxxxxx


--
To unsubscribe from this list: send the line "unsubscribe linux-kernel" in
the body of a message to majordomo@xxxxxxxxxxxxxxx
More majordomo info at http://vger.kernel.org/majordomo-info.html
Please read the FAQ at http://www.tux.org/lkml/