Add reference count of current module

This prevent module from unexpected unload
vi · Mar 15, 2023 · 7fb44f8 · 7fb44f8
1 parent 17320b1
commit 7fb44f8
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/virtual_touchscreen.c b/virtual_touchscreen.c
@@ -88,12 +88,14 @@ static int __init virt_ts_init(void)
 
 static int device_open(struct inode *inode, struct file *file) {
     if (Device_Open) return -EBUSY;
+    try_module_get(THIS_MODULE);
     ++Device_Open;
     return 0;
 }
 
 static int device_release(struct inode *inode, struct file *file) {
     --Device_Open;
+    module_put(THIS_MODULE);
     return 0;
 }