|
@@ -514,8 +514,8 @@ static int kprobe_dispatcher(struct kprobe *kp, struct pt_regs *regs);
|
|
static int kretprobe_dispatcher(struct kretprobe_instance *ri,
|
|
static int kretprobe_dispatcher(struct kretprobe_instance *ri,
|
|
struct pt_regs *regs);
|
|
struct pt_regs *regs);
|
|
|
|
|
|
-/* Check the name is good for event/group */
|
|
|
|
-static int check_event_name(const char *name)
|
|
|
|
|
|
+/* Check the name is good for event/group/fields */
|
|
|
|
+static int is_good_name(const char *name)
|
|
{
|
|
{
|
|
if (!isalpha(*name) && *name != '_')
|
|
if (!isalpha(*name) && *name != '_')
|
|
return 0;
|
|
return 0;
|
|
@@ -557,7 +557,7 @@ static struct trace_probe *alloc_trace_probe(const char *group,
|
|
else
|
|
else
|
|
tp->rp.kp.pre_handler = kprobe_dispatcher;
|
|
tp->rp.kp.pre_handler = kprobe_dispatcher;
|
|
|
|
|
|
- if (!event || !check_event_name(event)) {
|
|
|
|
|
|
+ if (!event || !is_good_name(event)) {
|
|
ret = -EINVAL;
|
|
ret = -EINVAL;
|
|
goto error;
|
|
goto error;
|
|
}
|
|
}
|
|
@@ -567,7 +567,7 @@ static struct trace_probe *alloc_trace_probe(const char *group,
|
|
if (!tp->call.name)
|
|
if (!tp->call.name)
|
|
goto error;
|
|
goto error;
|
|
|
|
|
|
- if (!group || !check_event_name(group)) {
|
|
|
|
|
|
+ if (!group || !is_good_name(group)) {
|
|
ret = -EINVAL;
|
|
ret = -EINVAL;
|
|
goto error;
|
|
goto error;
|
|
}
|
|
}
|
|
@@ -883,7 +883,7 @@ static int create_trace_probe(int argc, char **argv)
|
|
int i, ret = 0;
|
|
int i, ret = 0;
|
|
int is_return = 0, is_delete = 0;
|
|
int is_return = 0, is_delete = 0;
|
|
char *symbol = NULL, *event = NULL, *group = NULL;
|
|
char *symbol = NULL, *event = NULL, *group = NULL;
|
|
- char *arg, *tmp;
|
|
|
|
|
|
+ char *arg;
|
|
unsigned long offset = 0;
|
|
unsigned long offset = 0;
|
|
void *addr = NULL;
|
|
void *addr = NULL;
|
|
char buf[MAX_EVENT_NAME_LEN];
|
|
char buf[MAX_EVENT_NAME_LEN];
|
|
@@ -992,26 +992,36 @@ static int create_trace_probe(int argc, char **argv)
|
|
/* parse arguments */
|
|
/* parse arguments */
|
|
ret = 0;
|
|
ret = 0;
|
|
for (i = 0; i < argc && i < MAX_TRACE_ARGS; i++) {
|
|
for (i = 0; i < argc && i < MAX_TRACE_ARGS; i++) {
|
|
|
|
+ /* Increment count for freeing args in error case */
|
|
|
|
+ tp->nr_args++;
|
|
|
|
+
|
|
/* Parse argument name */
|
|
/* Parse argument name */
|
|
arg = strchr(argv[i], '=');
|
|
arg = strchr(argv[i], '=');
|
|
- if (arg)
|
|
|
|
|
|
+ if (arg) {
|
|
*arg++ = '\0';
|
|
*arg++ = '\0';
|
|
- else
|
|
|
|
|
|
+ tp->args[i].name = kstrdup(argv[i], GFP_KERNEL);
|
|
|
|
+ } else {
|
|
arg = argv[i];
|
|
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);
|
|
|
|
+ }
|
|
|
|
|
|
- tp->args[i].name = kstrdup(argv[i], GFP_KERNEL);
|
|
|
|
if (!tp->args[i].name) {
|
|
if (!tp->args[i].name) {
|
|
- pr_info("Failed to allocate argument%d name '%s'.\n",
|
|
|
|
- i, argv[i]);
|
|
|
|
|
|
+ pr_info("Failed to allocate argument[%d] name.\n", i);
|
|
ret = -ENOMEM;
|
|
ret = -ENOMEM;
|
|
goto error;
|
|
goto error;
|
|
}
|
|
}
|
|
- tmp = strchr(tp->args[i].name, ':');
|
|
|
|
- if (tmp)
|
|
|
|
- *tmp = '_'; /* convert : to _ */
|
|
|
|
|
|
+
|
|
|
|
+ if (!is_good_name(tp->args[i].name)) {
|
|
|
|
+ pr_info("Invalid argument[%d] name: %s\n",
|
|
|
|
+ i, tp->args[i].name);
|
|
|
|
+ ret = -EINVAL;
|
|
|
|
+ goto error;
|
|
|
|
+ }
|
|
|
|
|
|
if (conflict_field_name(tp->args[i].name, tp->args, i)) {
|
|
if (conflict_field_name(tp->args[i].name, tp->args, i)) {
|
|
- pr_info("Argument%d name '%s' conflicts with "
|
|
|
|
|
|
+ pr_info("Argument[%d] name '%s' conflicts with "
|
|
"another field.\n", i, argv[i]);
|
|
"another field.\n", i, argv[i]);
|
|
ret = -EINVAL;
|
|
ret = -EINVAL;
|
|
goto error;
|
|
goto error;
|
|
@@ -1020,12 +1030,9 @@ static int create_trace_probe(int argc, char **argv)
|
|
/* Parse fetch argument */
|
|
/* Parse fetch argument */
|
|
ret = parse_probe_arg(arg, tp, &tp->args[i], is_return);
|
|
ret = parse_probe_arg(arg, tp, &tp->args[i], is_return);
|
|
if (ret) {
|
|
if (ret) {
|
|
- pr_info("Parse error at argument%d. (%d)\n", i, ret);
|
|
|
|
- kfree(tp->args[i].name);
|
|
|
|
|
|
+ pr_info("Parse error at argument[%d]. (%d)\n", i, ret);
|
|
goto error;
|
|
goto error;
|
|
}
|
|
}
|
|
-
|
|
|
|
- tp->nr_args++;
|
|
|
|
}
|
|
}
|
|
|
|
|
|
ret = register_trace_probe(tp);
|
|
ret = register_trace_probe(tp);
|