504 lines · 13.5 KB · C
RawDownload
// SPDX-License-Identifier: GPL-2.0/* * Generic dynamic event control interface * * Copyright (C) 2018 Masami Hiramatsu <mhiramat@kernel.org> */ #include <linux/debugfs.h>#include <linux/kernel.h>#include <linux/list.h>#include <linux/mm.h>#include <linux/mutex.h>#include <linux/tracefs.h> #include "trace.h"#include "trace_output.h"	/* for trace_event_sem */#include "trace_dynevent.h" DEFINE_MUTEX(dyn_event_ops_mutex);static LIST_HEAD(dyn_event_ops_list); bool trace_event_dyn_try_get_ref(struct trace_event_call *dyn_call){	struct trace_event_call *call;	bool ret = false; 	if (WARN_ON_ONCE(!(dyn_call->flags & TRACE_EVENT_FL_DYNAMIC)))		return false; 	down_read(&trace_event_sem);	list_for_each_entry(call, &ftrace_events, list) {		if (call == dyn_call) {			atomic_inc(&dyn_call->refcnt);			ret = true;		}	}	up_read(&trace_event_sem);	return ret;} void trace_event_dyn_put_ref(struct trace_event_call *call){	if (WARN_ON_ONCE(!(call->flags & TRACE_EVENT_FL_DYNAMIC)))		return; 	if (WARN_ON_ONCE(atomic_read(&call->refcnt) <= 0)) {		atomic_set(&call->refcnt, 0);		return;	} 	atomic_dec(&call->refcnt);} bool trace_event_dyn_busy(struct trace_event_call *call){	return atomic_read(&call->refcnt) != 0;} int dyn_event_register(struct dyn_event_operations *ops){	if (!ops || !ops->create || !ops->show || !ops->is_busy ||	    !ops->free || !ops->match)		return -EINVAL; 	INIT_LIST_HEAD(&ops->list);	mutex_lock(&dyn_event_ops_mutex);	list_add_tail(&ops->list, &dyn_event_ops_list);	mutex_unlock(&dyn_event_ops_mutex);	return 0;} int dyn_event_release(const char *raw_command, struct dyn_event_operations *type){	struct dyn_event *pos, *n;	char *system = NULL, *event, *p;	int argc, ret = -ENOENT;	char **argv __free(argv_free) = argv_split(GFP_KERNEL, raw_command, &argc); 	if (!argv)		return -ENOMEM; 	if (argv[0][0] == '-') {		if (argv[0][1] != ':')			return -EINVAL;		event = &argv[0][2];	} else {		event = strchr(argv[0], ':');		if (!event)			return -EINVAL;		event++;	} 	p = strchr(event, '/');	if (p) {		system = event;		event = p + 1;		*p = '\0';	}	if (!system && event[0] == '\0')		return -EINVAL; 	mutex_lock(&event_mutex);	for_each_dyn_event_safe(pos, n) {		if (type && type != pos->ops)			continue;		if (!pos->ops->match(system, event,				argc - 1, (const char **)argv + 1, pos))			continue; 		ret = pos->ops->free(pos);		if (ret)			break;	}	tracing_reset_all_online_cpus();	mutex_unlock(&event_mutex);	return ret;} static int create_dyn_event(const char *raw_command); /* * Locked version of event creation. The event creation must be protected by * dyn_event_ops_mutex because of protecting trace_probe_log. */int dyn_event_create(const char *raw_command, struct dyn_event_operations *type){	int ret; 	if (!type)		return create_dyn_event(raw_command); 	mutex_lock(&dyn_event_ops_mutex);	ret = type->create(raw_command);	mutex_unlock(&dyn_event_ops_mutex);	return ret;} static int create_dyn_event(const char *raw_command){	struct dyn_event_operations *ops;	int ret = -ENODEV; 	if (raw_command[0] == '-' || raw_command[0] == '!')		return dyn_event_release(raw_command, NULL); 	mutex_lock(&dyn_event_ops_mutex);	list_for_each_entry(ops, &dyn_event_ops_list, list) {		ret = ops->create(raw_command);		if (!ret || ret != -ECANCELED)			break;	}	if (ret == -ECANCELED) {		static const char *err_msg[] = {"No matching dynamic event type"}; 		/* Wrong dynamic event. Leave an error message. */		tracing_log_err(NULL, "dynevent", raw_command, err_msg,				0, 0);		ret = -EINVAL;	} 	mutex_unlock(&dyn_event_ops_mutex); 	return ret;} /* Protected by event_mutex */LIST_HEAD(dyn_event_list); void *dyn_event_seq_start(struct seq_file *m, loff_t *pos){	mutex_lock(&event_mutex);	return seq_list_start(&dyn_event_list, *pos);} void *dyn_event_seq_next(struct seq_file *m, void *v, loff_t *pos){	return seq_list_next(v, &dyn_event_list, pos);} void dyn_event_seq_stop(struct seq_file *m, void *v){	mutex_unlock(&event_mutex);} static int dyn_event_seq_show(struct seq_file *m, void *v){	struct dyn_event *ev = v; 	if (ev && ev->ops)		return ev->ops->show(m, ev); 	return 0;} static const struct seq_operations dyn_event_seq_op = {	.start	= dyn_event_seq_start,	.next	= dyn_event_seq_next,	.stop	= dyn_event_seq_stop,	.show	= dyn_event_seq_show}; /* * dyn_events_release_all - Release all specific events * @type:	the dyn_event_operations * which filters releasing events * * This releases all events which ->ops matches @type. If @type is NULL, * all events are released. * Return -EBUSY if any of them are in use, and return other errors when * it failed to free the given event. Except for -EBUSY, event releasing * process will be aborted at that point and there may be some other * releasable events on the list. */int dyn_events_release_all(struct dyn_event_operations *type){	struct dyn_event *ev, *tmp;	int ret = 0; 	mutex_lock(&event_mutex);	for_each_dyn_event(ev) {		if (type && ev->ops != type)			continue;		if (ev->ops->is_busy(ev)) {			ret = -EBUSY;			goto out;		}	}	for_each_dyn_event_safe(ev, tmp) {		if (type && ev->ops != type)			continue;		ret = ev->ops->free(ev);		if (ret)			break;	}out:	tracing_reset_all_online_cpus();	mutex_unlock(&event_mutex); 	return ret;} static int dyn_event_open(struct inode *inode, struct file *file){	int ret; 	ret = security_locked_down(LOCKDOWN_TRACEFS);	if (ret)		return ret; 	ret = tracing_check_open_get_tr(NULL);	if (ret)		return ret; 	if ((file->f_mode & FMODE_WRITE) && (file->f_flags & O_TRUNC)) {		ret = dyn_events_release_all(NULL);		if (ret < 0)			return ret;	} 	return seq_open(file, &dyn_event_seq_op);} static ssize_t dyn_event_write(struct file *file, const char __user *buffer,				size_t count, loff_t *ppos){	return trace_parse_run_command(file, buffer, count, ppos,				       create_dyn_event);} static const struct file_operations dynamic_events_ops = {	.owner          = THIS_MODULE,	.open           = dyn_event_open,	.read           = seq_read,	.llseek         = seq_lseek,	.release        = seq_release,	.write		= dyn_event_write,}; /* Make a tracefs interface for controlling dynamic events */static __init int init_dynamic_event(void){	int ret; 	ret = tracing_init_dentry();	if (ret)		return 0; 	trace_create_file("dynamic_events", TRACE_MODE_WRITE, NULL,			  NULL, &dynamic_events_ops); 	return 0;}fs_initcall(init_dynamic_event); /** * dynevent_arg_add - Add an arg to a dynevent_cmd * @cmd: A pointer to the dynevent_cmd struct representing the new event cmd * @arg: The argument to append to the current cmd * @check_arg: An (optional) pointer to a function checking arg sanity * * Append an argument to a dynevent_cmd.  The argument string will be * appended to the current cmd string, followed by a separator, if * applicable.  Before the argument is added, the @check_arg function, * if present, will be used to check the sanity of the current arg * string. * * The cmd string and separator should be set using the * dynevent_arg_init() before any arguments are added using this * function. * * Return: 0 if successful, error otherwise. */int dynevent_arg_add(struct dynevent_cmd *cmd,		     struct dynevent_arg *arg,		     dynevent_check_arg_fn_t check_arg){	int ret = 0; 	if (check_arg) {		ret = check_arg(arg);		if (ret)			return ret;	} 	ret = seq_buf_printf(&cmd->seq, " %s%c", arg->str, arg->separator);	if (ret) {		pr_err("String is too long: %s%c\n", arg->str, arg->separator);		return -E2BIG;	} 	return ret;} /** * dynevent_arg_pair_add - Add an arg pair to a dynevent_cmd * @cmd: A pointer to the dynevent_cmd struct representing the new event cmd * @arg_pair: The argument pair to append to the current cmd * @check_arg: An (optional) pointer to a function checking arg sanity * * Append an argument pair to a dynevent_cmd.  An argument pair * consists of a left-hand-side argument and a right-hand-side * argument separated by an operator, which can be whitespace, all * followed by a separator, if applicable.  This can be used to add * arguments of the form 'type variable_name;' or 'x+y'. * * The lhs argument string will be appended to the current cmd string, * followed by an operator, if applicable, followed by the rhs string, * followed finally by a separator, if applicable.  Before the * argument is added, the @check_arg function, if present, will be * used to check the sanity of the current arg strings. * * The cmd strings, operator, and separator should be set using the * dynevent_arg_pair_init() before any arguments are added using this * function. * * Return: 0 if successful, error otherwise. */int dynevent_arg_pair_add(struct dynevent_cmd *cmd,			  struct dynevent_arg_pair *arg_pair,			  dynevent_check_arg_fn_t check_arg){	int ret = 0; 	if (check_arg) {		ret = check_arg(arg_pair);		if (ret)			return ret;	} 	ret = seq_buf_printf(&cmd->seq, " %s%c%s%c", arg_pair->lhs,			     arg_pair->operator, arg_pair->rhs,			     arg_pair->separator);	if (ret) {		pr_err("field string is too long: %s%c%s%c\n", arg_pair->lhs,		       arg_pair->operator, arg_pair->rhs,		       arg_pair->separator);		return -E2BIG;	} 	return ret;} /** * dynevent_str_add - Add a string to a dynevent_cmd * @cmd: A pointer to the dynevent_cmd struct representing the new event cmd * @str: The string to append to the current cmd * * Append a string to a dynevent_cmd.  The string will be appended to * the current cmd string as-is, with nothing prepended or appended. * * Return: 0 if successful, error otherwise. */int dynevent_str_add(struct dynevent_cmd *cmd, const char *str){	int ret = 0; 	ret = seq_buf_puts(&cmd->seq, str);	if (ret) {		pr_err("String is too long: %s\n", str);		return -E2BIG;	} 	return ret;} /** * dynevent_cmd_init - Initialize a dynevent_cmd object * @cmd: A pointer to the dynevent_cmd struct representing the cmd * @buf: A pointer to the buffer to generate the command into * @maxlen: The length of the buffer the command will be generated into * @type: The type of the cmd, checked against further operations * @run_command: The type-specific function that will actually run the command * * Initialize a dynevent_cmd.  A dynevent_cmd is used to build up and * run dynamic event creation commands, such as commands for creating * synthetic and kprobe events.  Before calling any of the functions * used to build the command, a dynevent_cmd object should be * instantiated and initialized using this function. * * The initialization sets things up by saving a pointer to the * user-supplied buffer and its length via the @buf and @maxlen * params, and by saving the cmd-specific @type and @run_command * params which are used to check subsequent dynevent_cmd operations * and actually run the command when complete. */void dynevent_cmd_init(struct dynevent_cmd *cmd, char *buf, int maxlen,		       enum dynevent_type type,		       dynevent_create_fn_t run_command){	memset(cmd, '\0', sizeof(*cmd)); 	seq_buf_init(&cmd->seq, buf, maxlen);	cmd->type = type;	cmd->run_command = run_command;} /** * dynevent_arg_init - Initialize a dynevent_arg object * @arg: A pointer to the dynevent_arg struct representing the arg * @separator: An (optional) separator, appended after adding the arg * * Initialize a dynevent_arg object.  A dynevent_arg represents an * object used to append single arguments to the current command * string.  After the arg string is successfully appended to the * command string, the optional @separator is appended.  If no * separator was specified when initializing the arg, a space will be * appended. */void dynevent_arg_init(struct dynevent_arg *arg,		       char separator){	memset(arg, '\0', sizeof(*arg)); 	if (!separator)		separator = ' ';	arg->separator = separator;} /** * dynevent_arg_pair_init - Initialize a dynevent_arg_pair object * @arg_pair: A pointer to the dynevent_arg_pair struct representing the arg * @operator: An (optional) operator, appended after adding the first arg * @separator: An (optional) separator, appended after adding the second arg * * Initialize a dynevent_arg_pair object.  A dynevent_arg_pair * represents an object used to append argument pairs such as 'type * variable_name;' or 'x+y' to the current command string.  An * argument pair consists of a left-hand-side argument and a * right-hand-side argument separated by an operator, which can be * whitespace, all followed by a separator, if applicable.  After the * first arg string is successfully appended to the command string, * the optional @operator is appended, followed by the second arg and * optional @separator.  If no separator was specified when * initializing the arg, a space will be appended. */void dynevent_arg_pair_init(struct dynevent_arg_pair *arg_pair,			    char operator, char separator){	memset(arg_pair, '\0', sizeof(*arg_pair)); 	if (!operator)		operator = ' ';	arg_pair->operator = operator; 	if (!separator)		separator = ' ';	arg_pair->separator = separator;} /** * dynevent_create - Create the dynamic event contained in dynevent_cmd * @cmd: The dynevent_cmd object containing the dynamic event creation command * * Once a dynevent_cmd object has been successfully built up via the * dynevent_cmd_init(), dynevent_arg_add() and dynevent_arg_pair_add() * functions, this function runs the final command to actually create * the event. * * Return: 0 if the event was successfully created, error otherwise. */int dynevent_create(struct dynevent_cmd *cmd)

Lines 1–500 of 504.