深入 ‘Spinlock’ vs ‘Mutex’:在内核编程中,为什么在中断上下文中绝对禁止使用互斥锁?

各位同仁,各位对内核编程充满热情的开发者们,大家好。

今天,我们将深入探讨一个在操作系统内核设计中至关重要、却又常常让初学者感到困惑的话题:并发控制。特别是,我们将聚焦于两种最基本的同步原语——自旋锁(Spinlock)和互斥锁(Mutex),并剖析一个核心的、不可动摇的原则:为什么在中断上下文中,我们绝对禁止使用互斥锁?

这不仅仅是一个技术细节,它触及了内核调度、中断处理和并发模型的最深层原理。理解这一点,是迈向成为一名合格的内核开发者的基石。


一、并发的挑战:共享数据与竞态条件

在现代多核处理器系统中,操作系统内核必须同时管理和调度成千上万的线程和进程。这些执行流可能在同一时间尝试访问和修改同一块共享数据。如果没有适当的同步机制,程序的行为将变得不可预测,数据可能被破坏,系统甚至可能崩溃。这就是所谓的“竞态条件”(Race Condition)。

什么是竞态条件?
当多个执行流(线程、进程、中断处理程序等)并发地访问和修改同一个共享资源,并且至少有一个是写操作时,如果结果的正确性依赖于这些操作发生的相对顺序,那么就存在竞态条件。

让我们看一个简单的例子:一个全局计数器。

// 共享数据
static int global_counter = 0;

// 多个线程或CPU可能同时执行的函数
void increment_counter_unsafe(void) {
    int temp;
    temp = global_counter; // 1. 读取当前值
    temp = temp + 1;      // 2. 加1
    global_counter = temp; // 3. 写回新值
}

假设在双核系统上,两个CPU同时执行 increment_counter_unsafe(),并且 global_counter 初始值为 0:

CPU 0 CPU 1 global_counter
temp = global_counter; (0) 0
temp = global_counter; (0) 0
temp = temp + 1; (1) 0
temp = temp + 1; (1) 0
global_counter = temp; (1) 1
global_counter = temp; (1) 1

最终 global_counter 变成了 1,而不是我们期望的 2。这就是一个典型的竞态条件。为了避免这种问题,我们需要确保在任何给定时间点,只有一个执行流能够访问和修改共享资源。这部分代码被称为“临界区”(Critical Section)。同步原语就是用来保护临界区的。


二、同步原语的基石:自旋锁(Spinlock)

自旋锁是内核中最基本、最轻量级的同步机制之一。它的核心思想是“忙等待”(Busy-Waiting)。当一个执行流尝试获取一个已经被持有的自旋锁时,它不会放弃CPU,而是会在一个紧密的循环中反复检查锁的状态,直到锁被释放。

2.1 自旋锁的工作原理

  1. 尝试获取锁: 执行流会使用原子操作(如 test_and_setcompare_and_swap)尝试将锁状态从“未锁定”设置为“已锁定”。
  2. 成功获取: 如果原子操作成功,表示该执行流获得了锁,它可以进入临界区。
  3. 失败则自旋: 如果锁已经被其他执行流持有,原子操作会失败。此时,该执行流不会被挂起或调度出CPU,而是会进入一个循环,不断地尝试获取锁。这就是“自旋”。
  4. 释放锁: 当执行流完成临界区的工作后,它会将锁状态设置为“未锁定”,从而允许其他正在自旋的执行流获取锁。

2.2 概念代码示例 (简化版)

// 定义一个简单的自旋锁结构
typedef struct {
    volatile int locked; // 0: unlocked, 1: locked
} simple_spinlock_t;

// 初始化自旋锁
void simple_spinlock_init(simple_spinlock_t *lock) {
    lock->locked = 0;
}

// 获取自旋锁
void simple_spin_lock(simple_spinlock_t *lock) {
    // 使用原子操作尝试获取锁
    // __sync_test_and_set(address, value) 是一个GCC内置函数,
    // 它会原子地将address处的值设置为value,并返回address处的旧值。
    while (__sync_test_and_set(&lock->locked, 1)) {
        // 如果返回1,说明锁已经被持有,继续自旋
        // 在实际内核中,这里会有一些CPU指令,例如PAUSE指令,
        // 降低忙等待时的功耗和总线竞争,但本质上仍在忙等。
    }
}

// 释放自旋锁
void simple_spin_unlock(simple_spinlock_t *lock) {
    // 简单地将锁状态设置为0
    // 这里需要一个内存屏障以确保在释放锁之前,临界区内的所有写操作都已完成并对其他CPU可见。
    // 在实际内核中,这通常由更复杂的原子操作或指令序列保证。
    __sync_synchronize(); // 内存屏障
    lock->locked = 0;
}

2.3 Linux内核中的自旋锁API

Linux内核提供了功能更强大、更完善的自旋锁API,它们不仅处理了多核同步,还考虑了中断上下文的问题。

#include <linux/spinlock.h>
#include <linux/interrupt.h> // 包含中断相关的API

// 定义一个自旋锁变量
DEFINE_SPINLOCK(my_spinlock);
static int shared_resource_data = 0;

// 在进程上下文中使用自旋锁
void process_context_function(void) {
    spin_lock(&my_spinlock); // 获取锁
    // 临界区:访问 shared_resource_data
    shared_resource_data++;
    spin_unlock(&my_spinlock); // 释放锁
}

// 在中断上下文中使用自旋锁(这是重点!)
irqreturn_t my_irq_handler(int irq, void *dev_id) {
    unsigned long flags; // 用于保存和恢复中断状态

    // spin_lock_irqsave() 会禁用当前CPU的中断,并获取自旋锁。
    // 这对于保护进程上下文和中断上下文之间共享的数据至关重要。
    spin_lock_irqsave(&my_spinlock, flags);
    // 临界区:访问 shared_resource_data
    shared_resource_data++;
    // spin_unlock_irqrestore() 会释放自旋锁,并恢复之前保存的中断状态。
    spin_unlock_irqrestore(&my_spinlock, flags);

    return IRQ_HANDLED;
}

2.4 自旋锁的优缺点

优点:

  • 低开销: 如果临界区很短,且锁的竞争不激烈,自旋锁的开销非常小,因为它避免了上下文切换。
  • 不休眠: 不会导致当前执行流进入睡眠状态,因此可以在不允许休眠的上下文中使用(例如中断上下文)。
  • 简单: 概念相对简单。

缺点:

  • CPU浪费: 如果锁的竞争激烈,或者临界区很长,长时间的忙等待会严重浪费CPU周期。
  • 死锁风险: 如果持有自旋锁的执行流被更高优先级的执行流抢占,并且高优先级执行流也尝试获取同一个自旋锁,就会导致死锁(尤其是在单核系统中)。在多核系统中,如果高优先级任务在另一个CPU上,则会导致无谓的自旋。
  • 中断禁用: 为了防止中断处理程序与进程上下文或同一CPU上的其他中断处理程序发生死锁,通常需要禁用中断,这会增加延迟。

三、同步原语的另一支柱:互斥锁(Mutex)

互斥锁(Mutex,Mutual Exclusion的缩写)是另一种常用的同步机制,它与自旋锁最大的不同在于:当一个执行流尝试获取一个已经被持有的互斥锁时,它不会忙等待,而是会进入睡眠状态,并将CPU让给其他任务。

3.1 互斥锁的工作原理

  1. 尝试获取锁: 执行流会尝试获取互斥锁。
  2. 成功获取: 如果锁未被持有,执行流成功获取锁并进入临界区。
  3. 失败则休眠: 如果锁已被持有,执行流不会继续执行,而是会被放入一个等待队列,并被调度器设置为睡眠状态(例如 TASK_UNINTERRUPTIBLE)。CPU会立即切换到另一个可运行的任务。
  4. 唤醒与重新调度: 当持有锁的执行流释放锁时,它会唤醒等待队列中的一个或多个(通常是一个)任务。被唤醒的任务会变为可运行状态,并在调度器下次运行时有机会重新获取CPU,然后再次尝试获取锁。
  5. 释放锁: 任务完成临界区工作后,释放锁。

3.2 概念代码示例 (高层抽象)

#include <linux/mutex.h>
#include <linux/wait.h> // 内核等待队列

// 定义一个互斥锁结构 (简化概念,实际结构更复杂)
typedef struct {
    volatile int locked; // 0: unlocked, 1: locked
    wait_queue_head_t wait_queue; // 等待队列,用于存放睡眠的任务
} simple_mutex_t;

// 初始化互斥锁
void simple_mutex_init(simple_mutex_t *lock) {
    lock->locked = 0;
    init_waitqueue_head(&lock->wait_queue);
}

// 获取互斥锁
void simple_mutex_lock(simple_mutex_t *lock) {
    // 循环直到获取锁
    while (1) {
        // 使用原子操作尝试获取锁
        if (!__sync_test_and_set(&lock->locked, 1)) {
            break; // 成功获取锁,退出循环
        }

        // 锁已被持有,将当前任务添加到等待队列并进入睡眠
        // 实际内核中涉及更复杂的调度和任务状态管理
        // 伪代码:
        // current->state = TASK_UNINTERRUPTIBLE;
        // add_to_wait_queue(&lock->wait_queue, current);
        // schedule(); // 放弃CPU,等待被唤醒
        // remove_from_wait_queue(&lock->wait_queue, current);
        //
        // 实际上在Linux中,这由 mutex_lock() 内部处理,通常是
        // wait_event_interruptible(lock->wait_queue, !lock->locked);
        // 加上原子操作
        // 这里只是为了说明其核心原理
        wait_event_interruptible(lock->wait_queue, !lock->locked);
    }
}

// 释放互斥锁
void simple_mutex_unlock(simple_mutex_t *lock) {
    // 原子地释放锁
    __sync_synchronize(); // 内存屏障
    lock->locked = 0;

    // 唤醒等待队列中的一个或所有任务
    // 伪代码:
    // wake_up(&lock->wait_queue);
    wake_up_interruptible(&lock->wait_queue);
}

3.3 Linux内核中的互斥锁API

#include <linux/mutex.h>

// 定义一个互斥锁变量
DEFINE_MUTEX(my_mutex);
static int shared_buffer[10];
static int buffer_index = 0;

// 在进程上下文中使用互斥锁
void producer_thread_function(void) {
    mutex_lock(&my_mutex); // 获取锁
    // 临界区:写入共享缓冲区
    if (buffer_index < 10) {
        shared_buffer[buffer_index++] = 123;
        pr_info("Producer: Wrote data. Index: %dn", buffer_index);
    }
    mutex_unlock(&my_mutex); // 释放锁
}

void consumer_thread_function(void) {
    mutex_lock(&my_mutex); // 获取锁
    // 临界区:读取共享缓冲区
    if (buffer_index > 0) {
        buffer_index--;
        pr_info("Consumer: Read data. Index: %d, Value: %dn", buffer_index, shared_buffer[buffer_index]);
    }
    mutex_unlock(&my_mutex); // 释放锁
}

3.4 互斥锁的优缺点

优点:

  • 无CPU浪费: 当锁被持有并有竞争者时,不会进行忙等待,而是让出CPU,提高了CPU的利用率。
  • 适用于长临界区: 对于临界区较长或竞争激烈的场景,互斥锁的性能通常优于自旋锁。
  • 避免优先级反转: 现代操作系统的互斥锁实现通常会包含优先级继承(Priority Inheritance)或优先级上限(Priority Ceiling)等机制,以减少优先级反转的风险。

缺点:

  • 上下文切换开销: 获取或释放锁可能涉及上下文切换,这会带来较大的开销。
  • 不能在原子上下文使用: 这是最关键的一点! 由于互斥锁可能导致任务睡眠,因此它不能在不允许睡眠的原子上下文(如中断上下文)中使用。

四、核心概念:进程上下文与中断上下文

在深入探讨为何互斥锁不能在中断上下文中使用之前,我们必须清晰地理解内核中的两种主要执行上下文。

4.1 进程上下文 (Process Context)

  • 定义: 当内核代码代表一个用户空间进程执行时,我们称之为在进程上下文。这包括系统调用处理、内核线程、设备驱动程序中的大部分代码等。
  • 特性:
    • 与特定进程关联: 存在一个明确的 current 任务结构体(task_struct),代表当前正在执行的进程。
    • 可以睡眠/阻塞: 进程上下文中的代码可以调用 schedule() 函数,主动放弃CPU,进入睡眠状态(例如,等待I/O完成、等待互斥锁释放、等待信号量等)。
    • 可抢占: 进程上下文是可抢占的,即一个更高优先级的进程或中断处理程序可以打断当前进程的执行。
    • 可以进行用户态/内核态切换: 这是其存在的根本目的。
  • 示例:
    • 用户程序调用 read() 系统调用,内核执行 sys_read()
    • 内核线程(如 kworkerksoftirqd)。
    • 设备驱动程序中处理用户请求的 open()read()write() 等函数。

4.2 中断上下文 (Interrupt Context)

  • 定义: 当硬件设备(如网卡、磁盘、定时器)触发中断时,CPU会暂停当前正在执行的任务,跳转到预先注册的中断处理程序(Interrupt Service Routine, ISR)的入口点。ISR以及由ISR调度执行的软中断(Softirq)、任务队列(Tasklet)等都运行在中断上下文。
  • 特性:
    • 与任何进程无关: 中断上下文不属于任何特定的进程。它是一个异步事件,与当前运行的进程无关。虽然中断可能打断某个进程的执行,但中断处理程序本身并不是该进程的一部分。
    • 绝对不能睡眠/阻塞: 这是其最核心且最严格的约束。中断处理程序必须尽快完成,并返回到被打断的进程。它不能调用任何可能导致睡眠的函数(例如 mutex_lock()msleep()wait_event() 等)。
    • 有限的栈空间: 中断上下文通常有较小的栈空间(例如几KB),不能执行复杂、耗时或需要大量栈空间的操作。
    • 高优先级: 通常比进程上下文具有更高的优先级。
    • 不可抢占(通常): 在Linux中,一个中断处理程序本身通常不能被另一个进程抢占。它可以被更高优先级的中断抢占(如果中断嵌套被允许),但在一个中断处理程序执行期间,调度器不会去调度其他进程。
  • 示例:
    • 网卡接收到数据包,触发中断,执行网卡驱动的 ethX_interrupt()
    • 定时器中断,执行时钟中断处理程序。
    • 软中断(softirq)、任务队列(tasklet)等“下半部”机制。

五、为什么互斥锁在中断上下文中绝对禁止使用?

现在,我们终于来到了本次讲座的核心问题。理解了进程上下文和中断上下文的区别后,答案就变得显而易见。

核心原因:中断上下文不能睡眠!

让我们详细分析一下:

  1. 互斥锁的工作机制: 正如我们前面讨论的,当一个执行流尝试获取一个已经被持有的互斥锁时,它会执行以下操作:

    • 将自己添加到互斥锁的等待队列中。
    • 将当前任务的状态设置为睡眠(例如 TASK_UNINTERRUPTIBLE)。
    • 调用调度器 schedule(),放弃CPU,让调度器选择另一个可运行的任务来执行。
  2. 中断上下文的本质: 中断处理程序是异步的、时间敏感的,并且与任何特定进程无关。它的设计目标是迅速响应硬件事件,并尽快将控制权返回给被打断的进程。

  3. 冲突的发生:

    • 假设一个中断处理程序(ISR)在执行过程中需要访问一个共享资源,并尝试使用 mutex_lock() 来保护它。
    • 如果这个互斥锁当前没有被其他执行流持有,ISR会成功获取锁,完成临界区操作,然后释放锁并返回。这看起来没有问题。
    • 但是,如果这个互斥锁已经被其他进程上下文(或另一个ISR,尽管这不常见且应避免)持有呢?
  4. 灾难性后果:

    • ISR 调用 mutex_lock()
    • 发现锁已被持有。
    • 按照互斥锁的逻辑,ISR会尝试将自己(但它没有对应的 task_struct!)添加到等待队列,并尝试将自身状态设置为睡眠。
    • 接着,ISR会尝试调用 schedule() 函数,放弃CPU。

    此时,将发生内核恐慌(Kernel Panic)!

    • 没有 task_struct ISR 没有关联的 task_struct 结构体来代表自己。调度器无法将一个中断上下文放入等待队列,也无法改变它的“状态”。
    • 无法调度: schedule() 函数是为进程上下文设计的。它需要保存当前进程的上下文,选择下一个进程,并加载其上下文。中断上下文没有这些机制。中断处理程序不能被“调度出”CPU去等待一个锁。它必须完成其执行并返回。
    • 系统挂死/崩溃: 如果一个ISR试图睡眠,它会陷入一个非法状态。内核中通常会有断言(BUG_ONWARN_ON),检测到在原子上下文(如中断上下文)中调用了可能导致睡眠的函数,并立即触发内核恐慌,导致系统崩溃。

场景模拟:

时间 CPU 0 事件 共享资源状态 结果
T0 进程 A 执行,获取 my_mutex my_mutex 被进程 A 持有 my_mutex locked=1
T1 进程 A 继续执行临界区代码。 my_mutex 被进程 A 持有
T2 硬件中断发生。CPU 0 暂停进程 A,跳转到中断处理程序 my_irq_handler my_mutex 被进程 A 持有 中断上下文开始执行。
T3 my_irq_handler 尝试获取 my_mutex (mutex_lock(&my_mutex))。 my_mutex 被进程 A 持有 mutex_lock() 发现锁已被持有。
T4 my_irq_handler 试图将自己添加到等待队列并调用 schedule() my_mutex 被进程 A 持有 内核恐慌! 中断上下文不能睡眠,没有 task_struct,无法进行调度。系统崩溃。

这个场景清晰地表明了互斥锁在中断上下文中的致命性。


六、中断上下文的正确同步方式:自旋锁与中断禁用

既然互斥锁不能用,那么在中断上下文中如何安全地访问共享数据呢?答案是:自旋锁,并且通常伴随着本地CPU中断的禁用。

6.1 为什么自旋锁可以在中断上下文中使用?

因为自旋锁不会导致睡眠。当中断处理程序尝试获取一个被持有的自旋锁时,它会忙等待,而不是放弃CPU。中断处理程序会持续自旋,直到锁被释放。一旦锁被释放,ISR就能立即获取它并继续执行。

6.2 为什么需要禁用中断?

禁用中断是自旋锁在中断上下文以及与中断上下文共享数据时的一个关键步骤,主要出于以下两个原因:

  1. 防止自死锁(Deadlock on the same CPU):

    • 假设在 CPU0 上,进程 A 已经获取了一个自旋锁 mylock
    • 此时,一个硬件中断发生,my_irq_handler 开始执行。
    • my_irq_handler 也尝试获取 mylock
    • 如果 mylock 已经被进程 A 持有,my_irq_handler 会进入自旋等待。
    • 问题来了:进程 A 已经被中断打断了,它无法继续执行并释放 mylock
    • 结果:my_irq_handler 会无限自旋,导致 CPU0 挂死。
    • 解决方案: 在获取自旋锁之前,禁用当前 CPU 的本地中断。这样,当进程 A 持有 mylock 并被中断打断时,my_irq_handler 仍然会尝试获取 mylock。但如果 mylock 已经被进程 A 持有,而 my_irq_handler 禁用了中断,那么进程 A 就不会被中断打断,它会继续执行直到释放 mylock,然后 my_irq_handler 才能获取锁。
  2. 保护共享数据免受中断和进程的竞态:

    • 当一个数据结构既可能被进程上下文访问,又可能被中断上下文访问时,仅仅使用 spin_lock() 是不够的。
    • spin_lock() 只能防止多个CPU同时访问。但如果进程 A 在 CPU0 上持有 mylock 并访问数据,此时 CPU0 上发生中断,my_irq_handler 也尝试访问相同数据,那么即使 my_irq_handler 也会尝试获取 mylock,但它可能会被进程 A 持有的锁阻塞,如上所述导致自死锁。
    • 禁用本地中断,可以确保在临界区内,任何中断都不会在当前CPU上打断临界区的执行,从而保证了临界区的原子性。

6.3 Linux内核中的自旋锁变体

Linux内核提供了多种自旋锁API,以适应不同的场景:

  • spin_lock_init(lock) / DEFINE_SPINLOCK(lock): 初始化自旋锁。
  • spin_lock(lock) / spin_unlock(lock): 最基本的自旋锁,只进行锁操作,不影响中断状态。主要用于保护只被进程上下文访问的共享数据,或者在中断已经被禁用的上下文中使用。
  • spin_lock_irq(lock) / spin_unlock_irq(lock): 获取锁之前禁用本地CPU所有中断。释放锁时启用本地CPU所有中断。适用于确定需要禁用中断且不关心之前中断状态的场景。
  • spin_lock_irqsave(lock, flags) / spin_unlock_irqrestore(lock, flags): 最常用和推荐的变体。 获取锁之前保存当前CPU的中断状态到 flags 变量,然后禁用本地CPU所有中断。释放锁时,恢复 flags 中保存的中断状态。这确保了中断状态的正确恢复,避免了不必要的副作用。它用于保护被进程上下文和中断上下文(包括上半部和下半部)共享的数据。
  • spin_lock_bh(lock) / spin_unlock_bh(lock): 获取锁之前禁用本地CPU的下半部(softirq、tasklet)。释放锁时启用本地CPU的下半部。用于保护被进程上下文和下半部共享的数据。

示例:spin_lock_irqsave 的使用

#include <linux/spinlock.h>
#include <linux/interrupt.h> // For irqreturn_t, irq_handler_t

static DEFINE_SPINLOCK(my_shared_data_lock);
static unsigned long shared_counter = 0;

// 假设这是一个设备驱动的进程上下文函数
void my_driver_write(unsigned long value) {
    // 保护 shared_counter,防止与其他进程或中断发生竞态
    // 这里使用 spin_lock_irqsave 是为了同时处理与中断的同步
    unsigned long flags;
    spin_lock_irqsave(&my_shared_data_lock, flags);
    shared_counter += value;
    spin_unlock_irqrestore(&my_shared_data_lock, flags);
}

// 假设这是一个设备中断处理程序 (上半部)
irqreturn_t my_device_irq_handler(int irq, void *dev_id) {
    // 在中断上下文中,只能使用自旋锁。
    // 并且由于 shared_counter 也可能被进程上下文访问,
    // 为了防止上面提到的自死锁以及保护数据,需要禁用中断。
    unsigned long flags;
    spin_lock_irqsave(&my_shared_data_lock, flags);
    shared_counter++; // 中断处理程序修改共享数据
    spin_unlock_irqrestore(&my_shared_data_lock, flags);

    // 进行其他中断处理...
    return IRQ_HANDLED;
}

// 假设这是一个软中断处理程序 (下半部,例如一个tasklet)
void my_tasklet_function(unsigned long data) {
    // 软中断也运行在原子上下文,不能睡眠。
    // 如果共享数据还可能被进程上下文访问,同样需要禁用软中断(bh)。
    // 如果共享数据只被软中断和进程上下文访问,可以使用 spin_lock_bh。
    // 如果还可能被上半部中断访问,则仍需 spin_lock_irqsave。
    // 这里为了演示,假设只和进程上下文有共享,且软中断可能发生抢占。
    // 但更安全的做法通常是 spin_lock_irqsave
    unsigned long flags; // 即使是BH,也常常用irqsave以覆盖更广的情况
    spin_lock_irqsave(&my_shared_data_lock, flags);
    shared_counter += 10; // 软中断修改共享数据
    spin_unlock_irqrestore(&my_shared_data_lock, flags);
}

七、自旋锁 vs. 互斥锁:综合比较

为了更好地理解这两种同步原语的选择,我们用一个表格进行比较:

特性 自旋锁 (Spinlock) 互斥锁 (Mutex)
基本机制 忙等待 (Busy-waiting):当锁被持有,请求者在循环中反复检查。 睡眠/阻塞 (Sleeping/Blocking):当锁被持有,请求者进入睡眠,放弃CPU。
CPU利用率 锁竞争激烈时,CPU会持续忙等,效率低。 锁竞争激烈时,CPU会让给其他任务,效率高。
开销 锁竞争不激烈时,开销极小,不涉及上下文切换。 可能涉及上下文切换,开销相对较大。
适用场景 1. 临界区极短 (通常在几十到几百条指令)。
2. 不允许睡眠的上下文 (如中断上下文、软中断、任务队列)。
3. 多CPU之间,防止并发访问。
1. 临界区较长
2. 允许睡眠的上下文 (如进程上下文、内核线程)。
3. 避免CPU浪费。
中断禁用 通常需要配合禁用本地CPU中断 (spin_lock_irqsave),以防止自死锁和保护共享数据。 不需要也不允许 禁用中断,因为它会睡眠。
抢占 在获取锁后,通常禁用本地中断,防止当前CPU上的抢占。 允许抢占,因为任务会睡眠并等待调度。
优先级反转 存在优先级反转风险(高优先级任务等待低优先级任务释放锁)。 通过优先级继承等机制可缓解优先级反转。
可以睡眠? ,绝对不能睡眠。 ,会主动进入睡眠。
内核API spin_lock(), spin_lock_irqsave(), spin_lock_bh() mutex_lock(), mutex_unlock(), mutex_trylock()
典型用途 设备寄存器访问、中断计数器、短数据结构更新。 文件系统结构、网络协议栈、长数据结构管理。

八、高级考量与最佳实践

理解了自旋锁和互斥锁的基本原理和使用场景后,在实际内核编程中还需要考虑一些高级因素:

  1. 锁粒度:

    • 粗粒度锁: 保护大块数据或整个子系统。优点是实现简单,缺点是并发度低,可能导致大量等待。
    • 细粒度锁: 保护小块数据或数据结构中的特定字段。优点是并发度高,缺点是实现复杂,可能引入更多的锁开销和死锁风险。
    • 选择合适的锁粒度是性能和正确性之间的权衡。
  2. 死锁:

    • 当两个或多个执行流互相等待对方释放资源时,就会发生死锁。
    • 避免死锁的关键原则:
      • 一致的锁获取顺序: 总是按照相同的顺序获取多个锁。
      • 避免循环等待: 不允许一个执行流等待另一个执行流所持有的锁,而另一个执行流又在等待第一个执行流所持有的锁。
      • 避免同时持有多个锁并尝试获取更多锁。
      • 尽量减少锁的持有时间。
  3. 读写锁(RW-lock):

    • 当共享数据通常被多个读者同时访问,但只有少数写者修改时,读写锁可以提供更好的并发性。
    • 允许多个读者同时持有读锁,但写者必须独占写锁。
    • Linux内核提供了 rwlock_t 及其相关的 read_lock(), write_lock() 等API。
  4. 原子操作(Atomic Operations):

    • 对于简单的整型计数器或位操作,直接使用原子操作 (atomic_t, test_and_set_bit 等) 比使用锁更高效,因为它避免了锁的开销。
    • 例如:atomic_inc(&my_atomic_counter);
  5. 每CPU数据(Per-CPU Data):

    • 如果每个CPU都有自己的私有数据副本,那么访问这些数据就不需要锁。这是最高效的并发策略。
    • Linux内核提供了 DEFINE_PER_CPU 宏来定义每CPU变量。
  6. 内存屏障(Memory Barriers):

    • 在多核系统中,CPU和编译器可能会对内存操作进行重排序,以优化性能。内存屏障用于强制这些操作按照程序员指定的顺序执行,确保数据在不同CPU之间正确同步。
    • 锁机制通常内部包含了必要的内存屏障,但在某些无锁编程或自定义同步中,需要手动插入。

九、健壮内核同步的原则

通过今天的探讨,我们深入理解了自旋锁与互斥锁的异同,以及在内核编程中一个不可逾越的红线:中断上下文中绝对禁止使用互斥锁,因为中断处理程序不能睡眠。

健壮的内核同步建立在对执行上下文的深刻理解之上。选择正确的同步原语,不仅关系到程序的正确性,也直接影响到系统的性能和稳定性。

核心原则可以概括为:

  • 能不加锁就不加锁。 考虑原子操作、每CPU数据等无锁或低锁方案。
  • 需要加锁时,根据上下文选择。 进程上下文优先考虑互斥锁,中断上下文(包括上半部和下半部)则必须使用自旋锁。
  • 保护进程与中断共享的数据,务必使用 spin_lock_irqsave 这能同时解决多CPU竞争和单CPU中断抢占的问题。
  • 始终警惕死锁的风险,并遵循一致的锁获取顺序。

希望今天的讲座能帮助大家在内核编程的道路上走得更稳、更远。谢谢大家!

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注