linux下内核多线程的简单实现

多线程 专栏收录该内容
1 篇文章 0 订阅

内核态多线程的 学习


  • 方法、结构介绍
  1. task_struct // 用户定义 j 进程描述符,linux中把并不对进程和线程做强制区分。
  2. kthread_run() //用户创建一个线程并运行函数原型如下 kthread_run(threadfn, data, namefmt, …), threadfn 是线程被唤醒后执行的方法。
  3. kthead_stop() //用于结束一个线程的运行,需要注意的是调用此方法时,该线程必须不能已经结束,否则后果严重。
  4. kthread_should_stop() //用户返回结束标志
  5. wait_event_interruptible_on_timeout() //中断一个线程,知道满足条件或者超时为止

代码:

#include <linux/init.h>
#include <linux/module.h>
#include <linux/kernel.h>
#include <linux/delay.h>
#include <linux/wait.h>
#include <linux/kthread.h>
#include <linux/spinlock.h>

static struct task_struct * task1;
static struct task_struct * task2;
static int number = 0;
int i, j;
int num = 0;
DEFINE_SPINLOCK(mylock);
static wait_queue_head_t wait_queue;

static int thread_fun1(void* data)
{
    init_waitqueue_head(&wait_queue);
//    printk(KERN_INFO"thread1: number = %d \n", number);
    for(i = 0; i < 10; i++)
    {
        spin_lock(&mylock);
        number++;
        spin_unlock(&mylock);
        printk("thread1: number= %d \n", number);
        msleep(1000);
    }
    j = 1;
    while(!kthread_should_stop())
    {
        wait_event_interruptible_timeout(wait_queue, false, HZ);
        printk("thread 1 sleep .. %d \n", j++);
    }
    return 0;
}

static int thread_fun2(void *data){
    //printk(KERN_INFO"thread2: number = %d\n",number);
     init_waitqueue_head(&wait_queue);
    for(i=0;i<10;i++){
        spin_lock(&mylock);  //加锁以保证同步
        number++;
        spin_unlock(&mylock);
        printk("thread2: number = %d\n",number);
        msleep(1000);
        }
    j=1;
    while(!kthread_should_stop()){
        wait_event_interruptible_timeout(wait_queue,false,HZ);
        printk("thread 2 sleeping..%d\n", j++);
    }
    return 0;
}

static int __init hello_init(void){
    task1 = kthread_run(thread_fun1,NULL,"mythread1");     // run a kthread return 进程描述符
    if(IS_ERR(task1)){
        printk("thread1 create failed!\n");
    }else{
        printk("thread1 create success!\n");
    }
    task2 = kthread_run(thread_fun2,NULL,"mythread2");
    if(IS_ERR(task2)){
        printk("thread2 create failed!\n");
    }else{
        printk("thread2 create success\n");
    }
    return 0;
}

static void __exit hello_exit(void){
    if(!IS_ERR(task1)){     //这里判断指针是否正常
        kthread_stop(task1);
        printk("thread1 finished!\n");
    }
    if(!IS_ERR(task2)){
        kthread_stop(task2);
        printk("thread2 finished!\n");
    }
}
module_init(hello_init);
module_exit(hello_exit);
MODULE_LICENSE("GPL");

在这里插入图片描述

分析结果:

  • i 是全局变量,线程1执行时赋值 为 0,在线程2执行时又赋值为0 所以number打印到 11。

ps -ef 查看线程:
在这里插入图片描述
rmmod 卸载模块:
在这里插入图片描述

一个非常简单的内核级多线程程序学习!

  • 0
    点赞
  • 1
    评论
  • 4
    收藏
  • 一键三连
    一键三连
  • 扫一扫,分享海报

©️2021 CSDN 皮肤主题: 技术黑板 设计师:CSDN官方博客 返回首页
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、C币套餐、付费专栏及课程。

余额充值