1 files changed, 3 insertions, 1 deletions

diff --git a/kernel/trace/trace_kprobe.c b/kernel/trace/trace_kprobe.c
index 79ce6a2bd74f..82e85836d05e 100644
--- a/kernel/trace/trace_kprobe.c
+++ b/kernel/trace/trace_kprobe.c
@@ -704,10 +704,12 @@ static int create_trace_probe(int argc, char **argv)
                 ret = parse_probe_arg(arg, &tp->args[i].fetch, is_return);
                 if (ret) {
                         pr_info("Parse error at argument%d. (%d)\n", i, ret);
+                        kfree(tp->args[i].name);
                         goto error;
                 }
+
+                tp->nr_args++;
         }
-        tp->nr_args = i;
 
         ret = register_trace_probe(tp);
         if (ret)