如何立即取消Linux內核模塊中工作隊列的工作項目？

Question

內核模塊代碼：

#include <linux/delay.h> 
#include <linux/kernel.h> 
#include <linux/module.h> 
#include <linux/workqueue.h> 

MODULE_LICENSE("GPL"); 

static struct workqueue_struct *queue; 

static void work_func(struct work_struct *work) 
{ 
    int i = 0; 
    while (i < 5) { 
     printk(KERN_INFO "%d\n", i); 
     usleep_range(1000000, 1000001); 
     i++; 
    } 
} 

DECLARE_WORK(work, work_func); 

int init_module(void) 
{ 
    queue = create_workqueue("myworkqueue"); 
    queue_work(queue, &work); 
    return 0; 
} 

void cleanup_module(void) 
{ 
    cancel_work_sync(&work);  
    destroy_workqueue(queue); 
} 

如果我做的：

insmod mymod.ko 
rmmod mymod 

rmmod掛起上cancel_work_sync，爲工作首先等待完成這，直到計數結束。

是否可以立即取消該工作項目？

最小可運行示例here。

在Linux內核4.9測試。

Answer 1

原子能控制變量

我無法找到一種方式，工作隊列停止工作，但是使用一個簡單的控制變量是一個可能的解決方案。

#include <linux/delay.h> /* usleep_range */ 
#include <linux/kernel.h> 
#include <linux/module.h> 
#include <linux/types.h> /* atomic_t */ 
#include <linux/workqueue.h> 

MODULE_LICENSE("GPL"); 

static struct workqueue_struct *queue; 
static atomic_t run = ATOMIC_INIT(1); 

static void work_func(struct work_struct *work) 
{ 
    int i = 0; 
    while (atomic_read(&run)) { 
     printk(KERN_INFO "%d\n", i); 
     usleep_range(1000000, 1000001); 
     i++; 
     if (i == 10) 
      i = 0; 
    } 
} 

DECLARE_WORK(work, work_func); 

int init_module(void) 
{ 
    queue = create_workqueue("myworkqueue"); 
    queue_work(queue, &work); 
    return 0; 
} 

void cleanup_module(void) 
{ 
    atomic_set(&run, 0); 
    destroy_workqueue(queue); 
} 

KTHREAD kthread_stop

工作隊列根據kthreads和工作隊列在該示例中基本沒用，所以我們可以直接使用kthreads。

kthread_stop等待線程返回。

參見：

• Proper way of handling threads in kernel?
• How to wait for a linux kernel thread (kthread)to exit?

信號在kthreads處理似乎一直是一個爭論話題，現在是不可能的：https://unix.stackexchange.com/questions/355280/how-signals-are-handled-in-kernel

#include <linux/delay.h> /* usleep_range */ 
#include <linux/kernel.h> 
#include <linux/kthread.h> 
#include <linux/module.h> 

MODULE_LICENSE("GPL"); 

static struct task_struct *kthread; 

static int work_func(void *data) 
{ 
    int i = 0; 
    while (!kthread_should_stop()) { 
     printk(KERN_INFO "%d\n", i); 
     usleep_range(1000000, 1000001); 
     i++; 
     if (i == 10) 
      i = 0; 
    } 
    return 0; 
} 

int init_module(void) 
{ 
    kthread = kthread_create(work_func, NULL, "mykthread"); 
    wake_up_process(kthread); 
    return 0; 
} 

void cleanup_module(void) 
{ 
    kthread_stop(kthread); 
} 

定時器

直接在中斷上下文中運行，所以更準確但更受限制。

參見：How to use timers in Linux kernel device drivers?

#include <linux/jiffies.h> 
#include <linux/kernel.h> 
#include <linux/module.h> 
#include <linux/timer.h> 

MODULE_LICENSE("GPL"); 

static void callback(unsigned long data); 
static unsigned long onesec; 

DEFINE_TIMER(mytimer, callback, 0, 0); 

static void callback(unsigned long data) 
{ 
    pr_info("%u\n", (unsigned)jiffies); 
    mod_timer(&mytimer, jiffies + onesec); 
} 

int init_module(void) 
{ 
    onesec = msecs_to_jiffies(1000); 
    mod_timer(&mytimer, jiffies + onesec); 
    return 0; 
} 

void cleanup_module(void) 
{ 
    del_timer(&mytimer); 
}