Linux下的内核抢占

2017-03-03

　　内核抢占在一定程度上减少了对某种事件的响应延迟，这也是内核抢占被引入的目的。之前的内核中，除了显示调用系统调度器的某些点，内核其他地方是不允许中断的，如果内核在做一些比较复杂的工作，就会造成某些急于处理的事得不到及时响应。针对内核抢占其实本质上也是对当前进程而言（不知道这么描述是否合适），因为内核是为用户程序提供服务，换言之，其本身不会主动的去执行某个动作。这里内核抢占，重点在于用户程序请求内核服务时，CPU切换到内核态在执行某个系统服务期间，被抢占。

　　当然，即使支持内核抢占，也不是什么时候都可以的，还是要考虑对临界区的保护。类似于多处理器架构，如果进程A陷入到内核模式访问某个临界资源，而在访问期间，进程B也要访问临界区，如果这种抢占被允许，那么就发生了临界区被重入。所以，在访问临界资源时需要禁止内核抢占，在出临界区则要开启内核抢占。

　　为了支持内核抢占，在进程结构体的thread_info结构中有个preempt_count字段，用以记录当前内核（活动）是否可以被抢占。当该值为0时，允许被抢占；否则，不允许。

　　关于内核抢占有几个函数：
　　

#define preempt_disable() do {     inc_preempt_count();     barrier(); } while (0)

#define preempt_enable() do {     preempt_enable_no_resched();     barrier();     preempt_check_resched(); } while (0)

#define inc_preempt_count() add_preempt_count(1)
#define dec_preempt_count() sub_preempt_count(1)

# define add_preempt_count(val)    do { preempt_count() += (val); } while (0)
# define sub_preempt_count(val)    do { preempt_count() -= (val); } while (0)

上面两个函数是禁止和启用内核抢占。其实就是一个宏定义。禁止内核抢占本质上就是把当前进程的thread_info结构中的preempt_count字段加1，而启用内核抢占就是减1.注意这里启用内核抢占之后，调用了preempt_check_resched检查当前是否需要重新调度，这也是一个宏，实现如下：

#define preempt_check_resched() do {     if (unlikely(test_thread_flag(TIF_NEED_RESCHED)))         preempt_schedule(); } while (0)

就是在开启内核抢占之后，检查下此时是否有比较重要的进程等待执行，如果有，则调用preempt_schedule函数执行调度，可见在显示开启内核抢占之后，是触发内核抢占的一个时机。而调度过程本身是不允许被抢占的。preempt_schedule函数如下

asmlinkage void __sched notrace preempt_schedule(void)
{
    struct thread_info *ti = current_thread_info();

    /*
     * If there is a non-zero preempt_count or interrupts are disabled,
     * we do not want to preempt the current task. Just return..
     */
    if (likely(ti->preempt_count || irqs_disabled()))
        return;

    do {
        add_preempt_count_notrace(PREEMPT_ACTIVE);
        __schedule();
        sub_preempt_count_notrace(PREEMPT_ACTIVE);

        /*
         * Check again in case we missed a preemption opportunity
         * between schedule and now.
         */
        barrier();
    } while (need_resched());
}

如果抢占计数器不为0或者当前处于关闭硬件中断状态均是不可以被抢占的。

此外在硬件中断请求被处理后，有可能会检查调度标识和抢占计数器。

参考资料：

1、linux 3.10.1内核

2、深入linux内核架构