Merge pull request #119 from carloslack/kvdev

Block a process name that starts with a number
carloslack · Oct 14, 2024 · 964fb0e · 964fb0e
2 parents fa2a4b9 + d6f4dbb
commit 964fb0e
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/kovid.c b/src/kovid.c
@@ -23,6 +23,7 @@
 #include <linux/kthread.h>
 #include <linux/kernel.h>
 #include <linux/namei.h>
+#include <linux/ctype.h>
 
 #include "lkm.h"
 #include "fs.h"
@@ -558,8 +559,9 @@ static ssize_t write_cb(struct file *fptr, const char __user *user,
             char *newname;
             pid = (pid_t)simple_strtol(s, NULL, 10);
             newname = strrchr(buf, ' ');
-            if (++newname)
+            if (newname && ++newname && !isdigit(*newname)) {
                 kv_rename_task(pid, newname);
+            }
         }
     }
     proc_timeout(PRC_RESET);