IOCTL_TGID

1 files changed, 10 insertions, 5 deletions

diff --git a/source/module/monitor_kernel.c b/source/module/monitor_kernel.c
index 53b056c..77ac231 100644
--- a/source/module/monitor_kernel.c
+++ b/source/module/monitor_kernel.c
@@ -44,15 +44,15 @@ static int device_release(struct inode *inode, struct file *file) {
 }
 
 typedef struct {
-  int pid;
-} ioctl_pid;
+  int id;
+} ioctl_id;
 
 static long device_ioctl(struct file *file, unsigned int ioctl_num,
                          unsigned long ioctl_param) {
   int ret = 0;
   watch_arg warg;
   ioctl_dump_param dump_param;
-  ioctl_pid wpid;
+  ioctl_id wid;
 
   printk(KERN_INFO "variable_monitor fun: %s with ioctl_num %d\n", __FUNCTION__,
          ioctl_num);
@@ -91,8 +91,13 @@ static long device_ioctl(struct file *file, unsigned int ioctl_num,
     break;
   case IOCTL_PID:
     printk(KERN_INFO "variable_monitor PID\n");
-    ret = copy_from_user(&wpid, (ioctl_pid *)ioctl_param, sizeof(ioctl_pid));
-    diag_pid(wpid.pid);
+    ret = copy_from_user(&wid, (ioctl_id *)ioctl_param, sizeof(ioctl_id));
+    diag_pid(wid.id);
+    break;
+  case IOCTL_TGID:
+    printk(KERN_INFO "variable_monitor TGID\n");
+    ret = copy_from_user(&wid, (ioctl_id *)ioctl_param, sizeof(ioctl_id));
+    diag_tgid(wid.id);
     break;
   default:
     break;