内核事件机制

简介: 内核事件机制

引言

  • 本章节我们将来实现事件机制,听起来比较抽象,引用生活中的例子,比如我们想要吃饭,需要等待饭做好了还要等待菜炒好才能开始吃饭
  • 从任务的角度来理解:任务 A 煮饭,任务 B 炒菜,等到饭菜都做好后,任务 C 才能开始吃饭
  • 而我们当前的任务调度策略中,各任务之间并没有什么关系,任务是并行执行的(当然了,微观上依旧是串行执行),我们有办法去控制不同任务中的代码执行顺序吗?

用互斥锁实现事件

  • 其实以我们目前实现的系统,是可以实现控制不同任务中的代码执行顺序的,用互斥锁就可以达到这种效果
  • 直接看下面的示例代码:
void TaskA(void)  
{
    gMutexRice = MutexCreat();
    SetCursorPos(0, 8);
    print("cook rice:");
    MutexLock(gMutexRice);
    while (1)
    {
        static U32 cnt = 0;
        if(cnt < 20)
        {
            SetCursorPos(12, 8);
            print("%d\n", cnt++);
            Delay(999999);
        }
        else
            break;
    }
    MutexUnLock(gMutexRice);
}
void TaskB(void)  
{
    gMutexDish = MutexCreat();
    SetCursorPos(0, 10);
    print("cook dish:");
    MutexLock(gMutexDish);
    while (1)
    {
        static U32 cnt = 0;
        if(cnt < 30)
        {
            SetCursorPos(12, 10);
            print("%d\n", cnt++);
            Delay(999999);
        }
        else
            break;
    }
    MutexUnLock(gMutexDish);
}
void TaskC(void)  
{
    MutexLock(gMutexRice);
    MutexLock(gMutexDish);
    SetCursorPos(0, 12);
    print("have dinner\n");
    MutexUnLock(gMutexRice);
    MutexUnLock(gMutexDish);
}
  • 完整代码见:app.c
  • 运行效果,可以看到,任务 C 在等待任务 A 和任务 B 都结束后才打印出 "have dinner"

  • 从上面的实验代码可以看出,使用互斥锁貌似也可以实现事件,然而,这么做合适吗?

实现事件机制的系统调用

  • 使用互斥锁也能实现我们想要的等待效果,但是,上面的任务中,我们想保护的临界资源是什么?貌似并没有需要保护的临界资源。所以使用互斥锁实现等待显然是不合理的,那么接下来我们就专门实现事件机制吧
  • 既然使用互斥锁能实现等待的效果,那么我们完全可以参见互斥锁的做法来实现等待机制嘛
  • 创建 “event.c” 和 “event.h” 文件,用于实现等待机制
  • 先实现与用户层到内核层的系统调用
内核 用户 功能
SYS_EventCreat EventCreat 创建事件
SYS_WaitEvent WaitEvent 事件
SYS_SetEvent SetEvent 设置事件
SYS_ClearEvent ClearEvent 清除事件
SYS_EventDestory EventDestory 销毁事件
  • 关于系统调用的实现前面已经讲过好几遍了,这里就不再详细描述了,具体详见:再论系统调用
  • 内核层现在并没有实现具体功能,仅打印函数名替代,用以检测系统的调用实现。具体实现见:event.cevent.hsyscall.cu_syscall.cu_syscall.h
  • 在 “app.c” 中测试一下等待相关函数系统调用是否成功,具体见:app.c
EVENT* gEvent = NULL;
void TaskA(void)  
{
    gEvent = EventCreat();
    SetEvent(gEvent);
    WaitEvent(gEvent);
    ClearEvent(gEvent);
    EventDestory(gEvent);
    while (1);
}
  • 测试结果如下:

事件机制的具体实现

  • 接下来自然就是要具体实现事件机制的相关功能函数了
  • 具体实现的相关代码见:event.cschedule.cschedule.h
  • 由于事件机制的实现与互斥锁的实现很相似,所以大部分代码可以直接参考互斥锁的实现,比如 EventInit()、SYS_EventCreat()、CheckEvent()、SYS_EventDestory() 这几个函数可以说是跟互斥锁中对应的函数一样,这里就不做详细说明了,直接给出源码
// 初始化事件
void EventInit(void)
{
    ListInit(&EVENT_LIST);
}
// 创建事件
EVENT* SYS_EventCreat(void)
{
    EVENT* event = (EVENT *)Malloc(sizeof(EVENT));
    if(NULL == event)
        return NULL;
    event->event = 0;                             // 事件未生产状态
    ListAddHead(&EVENT_LIST, (LIST_NODE *)event); // 头插
    QueueInit(&event->wait);                      // 初始化事件中的等待队列
    return event;
}
// 检查事件是否合法有效
static BOOL CheckEvent(EVENT* event)
{
    LIST_NODE* pListNode = NULL;
    EVENT* nodeTmp = NULL;
    LIST_FOR_EACH(&EVENT_LIST, pListNode)
    {
        nodeTmp = (EVENT *)LIST_NODE(pListNode, EVENT, node);
        if(event == nodeTmp)
            return 1;
    }
    return 0;
} 
// 销毁事件
E_RET SYS_EventDestory(EVENT* event)
{
    // 检查参数合法性
    if(NULL == event || !CheckEvent(event))
        return E_ERR;
    // 删除链表节点并释放内存
    ListDelNode(&event->node);
    Free(event);
    return E_OK;
}
  • 还剩下几个函数,与互斥锁的实现稍微有点区别,然而,区别也不大,比如 SYS_WaitEvent()、SYS_SetEvent()、SYS_ClearEvent()
  • 先看 SYS_WaitEvent() 函数,其主要功能是将当前任务节点从任务就绪队列中转移到事件的等待队列中,与 MutexLock() 函数相比,其逻辑简单了很多,这里的 event->event 状态只有 0 和 1 两种,0 代表事件未发生,1代表事件发生
E_RET SYS_WaitEvent(EVENT* event)
{
    // 检查参数合法性
    if(NULL == event || !CheckEvent(event))
        return E_ERR;
    // 将当前任务节点从任务就绪队列中转移到等待时间的等待队列中
    if(0 == event->event)
        EventSuspend(event);
    return E_OK;
}
  • 再来看看 SYS_SetEvent() 函数,其主要功能是将事件 event 中的等待队列中的任务转移到任务就绪队列中
E_RET SYS_SetEvent(EVENT* event)
{
    // 检查参数合法性
    if(NULL == event || !CheckEvent(event))
        return E_ERR;
    // 将事件的状态设置为 1,表明事件发生
    event->event = 1;
    // 将事件 event 中的等待队列中的任务转移到任务就绪队列中
    EventResume(event);
    return E_OK;
}
  • 以上这两个函数分别调用了 EventSuspend() 函数和 EventResume() 函数,其功能实现与 MutexSuspend() 函数和 MutexResume() 函数的实现逻辑也是差不多的。我们把它们也放到 “schedule.c” 中实现,实现原理可以参考互斥锁相关章节,这里也不做详细描述了
// 将当前任务节点从任务就绪队列中转移到事件的等待队列中
E_RET EventSuspend(EVENT* event)
{
    QUEUE_NODE* nodeTmp = NULL;
    // 把当前任务节点从就绪任务队列中取出,当前任务节点在队列尾,取出后添加到事件 event 的 wait 等待队列中
    nodeTmp = QueueTailRemove(&TASK_READY_QUEUE);
    if(NULL == nodeTmp)
        return E_ERR;
    QueueAdd(&event->wait, nodeTmp);
    // 从就绪任务队列中取出一个任务节点并执行该任务,再将该任务节点重新添加到就绪任务队列中
    nodeTmp = QueueRemove(&TASK_READY_QUEUE);
    if(NULL == nodeTmp)
        return E_ERR;
    current_task = (volatile TASK *)QUEUE_NODE(nodeTmp, TASK, node);
    QueueAdd(&TASK_READY_QUEUE, nodeTmp);
    TSS* tss = (TSS*)(*(U32*)TSS_ENTRY_ADDR);                             // 找到 TSS
    tss->esp0 = (U32)(&current_task->reg) + sizeof(current_task->reg);    // TSS.esp0 指向任务上下文数据结构 reg 的末尾
    current_reg = (U32)(&current_task->reg);                              // current_reg 指向任务上下文数据结构 reg 的起始位置
    SWITCH_TO(current_task);
    return E_OK;
}
// 将事件 event 中的等待队列中的任务转移到任务就绪队列中
E_RET EventResume(EVENT* event)
{
    QUEUE_NODE* nodeTmp = NULL;
    // 从事件的等待队列中取出所有任务,并将其添加到就绪队列中
    while(event->wait.length)
    {
        nodeTmp = QueueTailRemove(&event->wait);
        if(NULL == nodeTmp)
            return E_ERR;
        QueueHeadAdd(&TASK_READY_QUEUE, nodeTmp);
    }
    return E_OK;
}
  • 最后还有一个 SYS_ClearEvent() 函数,其作用是将事件 event->event 的状态设置为 0,表明事件已被处理
E_RET SYS_ClearEvent(EVENT* event)
{
    // 检查参数合法性
    if(NULL == event || !CheckEvent(event))
        return E_ERR;
    // 将事件的状态设置为 0,表明事件已被处理
    event->event = 0;
    return E_OK;
}
  • 最后的最后,在 “main.c” 中,不要忘记初始化事件
S32 main(void)
{
    ...
    EventInit();                            // 初始化事件
    ...
}

实验一

  • 既然已经实现了事件机制,那么我们还是回到 “吃饭问题” 上,替换掉互斥锁,重新用事件机制来解决 “吃饭问题”
  • 实验代码见:app.c
  • 其中, A、B、C 三个任务实现如下:
EVENT* gEventRice = NULL;
EVENT* gEventDish = NULL;
void TaskA(void)  
{
    gEventRice = EventCreat();
    SetCursorPos(0, 8);
    print("cook rice:");
    while (1)
    {
        static U32 cnt = 0;
        if(cnt < 20)
        {
            SetCursorPos(12, 8);
            print("%d\n", cnt++);
            Delay(999999);
        }
        else
        {
            break;
        }
    }
    SetEvent(gEventRice);
}
void TaskB(void)  
{
    gEventDish = EventCreat();
    SetCursorPos(0, 10);
    print("cook dish:");
    while (1)
    {
        static U32 cnt = 0;
        if(cnt < 30)
        {
            SetCursorPos(12, 10);
            print("%d\n", cnt++);
            Delay(999999);
        }
        else
        {
            break;
        }
    }
    SetEvent(gEventDish);
}
void TaskC(void)  
{
    WaitEvent(gEventRice);
    WaitEvent(gEventDish);
    SetCursorPos(0, 12);
    print("have dinner\n");
}
  • 由于效果跟用互斥锁实现的一样,所以这里就不放成果展示了

实验二

  • 再来一个实验测试一下我们实现的事件机制
  • 实验目标:任务 A 每打印计数累加 20 次,任务 B 打印计数加 1
  • 完整代码见:app.c
  • 实验核心代码如下,当任务 A 调用 SetEvent(),这时候任务 B 就会被从等待队列中重新放入就绪队列调度执行,但其会被一直调度执行,想要任务 B 只执行一遍,可以调用 ClearEvent() 函数清掉标志,当任务 B 再次循环执行到 WaitEvent() 函数后就会被挂起,从而达到只执行一次的效果
EVENT* gEvent = NULL;
void TaskA(void)  
{
    static U32 cnt = 0;
    gEvent = EventCreat();
    while (1)
    {
        SetCursorPos(0, 10);
        print("TASK A: %d\n", cnt++);
        if(cnt%20 == 0)
            SetEvent(gEvent);
        Delay(999999);
        if(cnt > 200)
            break;
    }
    EventDestory(gEvent);
}
void TaskB(void)  
{
    static U32 cnt = 0;
    while (1)
    {
        WaitEvent(gEvent);
        ClearEvent(gEvent);
        SetCursorPos(0, 12);
        print("TASK B: %d\n", cnt++);
        Delay(999999);
    }
}
  • 最终成果展示:

思考

  • 事件机制虽然跟互斥锁很相似,但是并没有像互斥锁那样严格限制,为什么?
  • 因为使用互斥锁的目的是为了保护临界资源,有些资源就不可以多任务同时访问,比如非法操作互斥锁可能会导致某些硬件工作异常;而等待机制并没有临界资源的概念,也不需要保护什么,其本质上就是在不同任务间传递一个状态变量,就算是不合理操作了,也没有什么严重的后果产生,仅仅只能造成程序逻辑上不符合预期罢了

死锁

  • 现在,我们的系统中已经实现了互斥锁以及任务之间等待机制,那么自然就引出了多任务设计中的经典死锁问题,直接看下面的代码

  • 单独看 TaskA 任务,程序首先获取 gMutexA 互斥锁,再获取 gMutexB 互斥锁,然后释放 gMutexB 互斥锁,最后释放 gMutexA 互斥锁,没啥毛病,但是,如果此时还有一个 TaskB 任务并行执行,那么问题就来了
  • 假设程序在执行 TaskA 任务的 ① 语句,TaskA 获取到 gMutexA 互斥锁,然后可能调度到 TaskB 的 ② 语句处执行,TaskB 获取到 gMutexB 互斥锁,再然后程序可能又调度回 TaskA ③ 处执行,此时由于 gMutexB 被 TaskB 加锁占用,那么 TaskA 就会阻塞挂起,再接着程序可能被调度到 ④ 处执行,由于 gMutexA 被 TasKA 加锁占用,那么 TaskB 也会被阻塞挂起,最终 TaskA 和 TaskB 这两个任务都被挂起了,没有人可以释放 gMutexA 和 gMutexB 互斥锁,于是 TaskA 和 TaskB 永远都不可能被唤醒参与调度执行,对于这种现象,我们也称之为死锁
  • 再来看一个经典死锁示例

  • TaskA 任务由于在等待 gEventB 事件,可能导致 TaskA 任务被阻塞挂起,此时 TaskB 任务又在等待 gEventA 事件,那么 TaskB 任务也有可能被阻塞挂起,TaskA 和 TaskB 任务都被阻塞挂起,于是 TaskA 无法设置 gEventA,那么就可能无法唤醒 TaskB,而在 TaskB 中又无法设置 gEventB,那么 TaskA 也同样无法被唤醒。这样子也可能造成死锁
  • 总结一下死锁现象:任务因执行所需的资源无法获得而进入无限阻塞的状态
  • 对于多任务程序来说,死锁是非常可怕的,因为死锁的现象不一定能复现,调试起来非常困难,所以我么在设计多任务程序的时候,一定要注意避免死锁的发生
  • 对此,我么可以整理一下死锁的发生条件
  • 任务获取了临界资源,但同时又在等待其它临界资源
  • 任务之间相互等待
  • 任务获取了资源后不能被抢夺,必须由自己释放
  • 怎样避免死锁呢?这个就只能靠程序员小心、精心设计程序了
目录
相关文章
|
3月前
|
安全 数据处理 C++
【Qt 底层之事件驱动系统】深入理解 Qt 事件机制:主事件循环与工作线程的交互探究,包括 QML 的视角
【Qt 底层之事件驱动系统】深入理解 Qt 事件机制:主事件循环与工作线程的交互探究,包括 QML 的视角
766 3
|
3月前
|
监控 安全 API
7.3 Windows驱动开发:内核监视LoadImage映像回调
在笔者上一篇文章`《内核注册并监控对象回调》`介绍了如何运用`ObRegisterCallbacks`注册`进程与线程`回调,并通过该回调实现了`拦截`指定进行运行的效果,本章`LyShark`将带大家继续探索一个新的回调注册函数,`PsSetLoadImageNotifyRoutine`常用于注册`LoadImage`映像监视,当有模块被系统加载时则可以第一时间获取到加载模块信息,需要注意的是该回调函数内无法进行拦截,如需要拦截则需写入返回指令这部分内容将在下一章进行讲解,本章将主要实现对模块的监视功能。
77 0
7.3 Windows驱动开发:内核监视LoadImage映像回调
|
8月前
|
监控 安全 API
7.1 Windows驱动开发:内核监控进程与线程回调
在前面的文章中`LyShark`一直在重复的实现对系统底层模块的枚举,今天我们将展开一个新的话题,内核监控,我们以`监控进程线程`创建为例,在`Win10`系统中监控进程与线程可以使用微软提供给我们的两个新函数来实现,此类函数的原理是创建一个回调事件,当有进程或线程被创建或者注销时,系统会通过回调机制将该进程相关信息优先返回给我们自己的函数待处理结束后再转向系统层。
95 0
7.1 Windows驱动开发:内核监控进程与线程回调
|
Linux
Linux信号:进程间通讯与事件处理的基石
在Linux操作系统中通过信号,一个进程可以向另一个进程发送某种类型的通知,以便触发特定的动作。本文将深入探讨Linux信号的概念、种类、处理方式以及代码示例。
89 0
驱动开发:摘除InlineHook内核钩子
在笔者上一篇文章`《驱动开发:内核层InlineHook挂钩函数》`中介绍了通过替换`函数`头部代码的方式实现`Hook`挂钩,对于ARK工具来说实现扫描与摘除`InlineHook`钩子也是最基本的功能,此类功能的实现一般可在应用层进行,而驱动层只需要保留一个`读写字节`的函数即可,将复杂的流程放在应用层实现是一个非常明智的选择,与`《驱动开发:内核实现进程反汇编》`中所使用的读写驱动基本一致,本篇文章中的驱动只保留两个功能,控制信号`IOCTL_GET_CUR_CODE`用于读取函数的前16个字节的内存,信号`IOCTL_SET_ORI_CODE`则用于设置前16个字节的内存。
228 0
|
算法 调度 开发者
操作系统的定时器是干什么的?底层原理是什么?
操作系统的定时器是干什么的?底层原理是什么?
684 0
驱动开发:内核枚举进程与线程ObCall回调
在笔者上一篇文章`《驱动开发:内核枚举Registry注册表回调》`中我们通过特征码定位实现了对注册表回调的枚举,本篇文章`LyShark`将教大家如何枚举系统中的`ProcessObCall`进程回调以及`ThreadObCall`线程回调,之所以放在一起来讲解是因为这两中回调在枚举是都需要使用通用结构体`_OB_CALLBACK`以及`_OBJECT_TYPE`所以放在一起来讲解最好不过。
296 0
驱动开发:内核枚举进程与线程ObCall回调
|
监控
驱动开发:内核监控进程与线程回调
在前面的文章中`LyShark`一直在重复的实现对系统底层模块的枚举,今天我们将展开一个新的话题,内核监控,我们以`监控进程线程`创建为例,在`Win10`系统中监控进程与线程可以使用微软提供给我们的两个新函数来实现,此类函数的原理是创建一个回调事件,当有进程或线程被创建或者注销时,系统会通过回调机制将该进程相关信息优先返回给我们自己的函数待处理结束后再转向系统层。
355 0
驱动开发:内核监控进程与线程回调
|
监控
驱动开发:内核监视LoadImage映像回调
在笔者上一篇文章`《驱动开发:内核注册并监控对象回调》`介绍了如何运用`ObRegisterCallbacks`注册`进程与线程`回调,并通过该回调实现了`拦截`指定进行运行的效果,本章`LyShark`将带大家继续探索一个新的回调注册函数,`PsSetLoadImageNotifyRoutine`常用于注册`LoadImage`映像监视,当有模块被系统加载时则可以第一时间获取到加载模块信息,需要注意的是该回调函数内无法进行拦截,如需要拦截则需写入返回指令这部分内容将在下一章进行讲解,本章将主要实现对模块的监视功能。
329 0
驱动开发:内核监视LoadImage映像回调
|
监控
驱动开发:内核监控FileObject文件回调
本篇文章与上一篇文章`《驱动开发:内核注册并监控对象回调》`所使用的方式是一样的都是使用`ObRegisterCallbacks`注册回调事件,只不过上一篇博文中`LyShark`将回调结构体`OB_OPERATION_REGISTRATION`中的`ObjectType`填充为了`PsProcessType`和`PsThreadType`格式从而实现监控进程与线程,本章我们需要将该结构填充为`IoFileObjectType`以此来实现对文件的监控,文件过滤驱动不仅仅可以用来监控文件的打开,还可以用它实现对文件的保护,一旦驱动加载则文件是不可被删除和改动的。
349 0
驱动开发:内核监控FileObject文件回调