1 files changed, 10 insertions, 3 deletions

diff --git a/kernel/trace/trace_uprobe.c b/kernel/trace/trace_uprobe.c
index eb7e06b54741..a88c692e3b8a 100644
--- a/kernel/trace/trace_uprobe.c
+++ b/kernel/trace/trace_uprobe.c
@@ -443,10 +443,17 @@ static int trace_uprobe_create(int argc, const char **argv)
         ret = 0;
         ref_ctr_offset = 0;
 
-        /* argc must be >= 1 */
-        if (argv[0][0] == 'r')
+        switch (argv[0][0]) {
+        case 'r':
                 is_return = true;
-        else if (argv[0][0] != 'p' || argc < 2)
+                break;
+        case 'p':
+                break;
+        default:
+                return -ECANCELED;
+        }
+
+        if (argc < 2)
                 return -ECANCELED;
 
         if (argv[0][1] == ':')