2. 原子操作：原子整型操作API、原子位操作API、内存屏障与顺序保证、ARM64与x86架构下的原子指令差异

原子操作是Linux内核并发控制中最基础的构建块。与自旋锁或互斥体不同，原子操作直接在单条指令或指令序列中完成“读-改-写”过程，不会被其他执行流中断。理解原子操作API、内存屏障语义以及不同架构下的指令差异，是调试竞态条件的前提。

2.1 原子整型操作API

Linux内核提供了一套与架构无关的原子整型操作接口，定义在 <linux/atomic.h> 中。核心数据类型为 atomic_t，它是一个封装了 int 的结构体，强制开发者只能通过专用API访问，避免直接赋值导致的竞态。

典型使用场景——引用计数：

struct my_device {
    atomic_t refcount;
    // ... 其他字段
};

// 初始化
atomic_set(&dev->refcount, 1);

// 获取引用
void get_device(struct my_device *dev) {
    atomic_inc(&dev->refcount);
}

// 释放引用，返回 true 表示可销毁
int put_device(struct my_device *dev) {
    return atomic_dec_and_test(&dev->refcount);
}

2.2 原子位操作API

除了整型变量，内核还提供了对内存中任意位的原子操作，常用于设备寄存器标志位或状态字的管理。这些API定义在 <linux/bitops.h> 中，操作对象是 unsigned long * 指针。

位操作典型用法——设备状态标志：

#define DEVICE_BUSY   0
#define DEVICE_ERROR  1
#define DEVICE_DONE   2

unsigned long dev_flags = 0;

// 原子标记设备忙
if (test_and_set_bit(DEVICE_BUSY, &dev_flags)) {
    pr_info("设备正忙，无法操作\n");
    return -EBUSY;
}

// 中断处理中标记完成
test_and_set_bit(DEVICE_DONE, &dev_flags);

// 清除忙标志
clear_bit(DEVICE_BUSY, &dev_flags);

2.3 内存屏障与顺序保证

原子操作保证了操作的不可分割性，但不保证操作的顺序。现代CPU为了性能会对内存访问进行重排序（Reordering）。内存屏障（Memory Barrier）就是用来约束这种重排序的指令。

Linux内核提供以下屏障原语：

经典生产者-消费者模式中的屏障使用：

// 生产者（CPU 0）
data = 100;
smp_wmb();              // 保证 data 写入先于 flag 写入
atomic_set(&flag, 1);

// 消费者（CPU 1）
while (atomic_read(&flag) != 1)
    cpu_relax();
smp_rmb();              // 保证 flag 读取先于 data 读取
printk("data = %d\n", data);  // 安全：一定看到 100

2.4 ARM64与x86架构下的原子指令差异

不同CPU架构的原子指令实现差异巨大。理解这些差异对于编写可移植的内核代码和调试跨平台竞态问题至关重要。

ARM64 原子操作的内核实现示例（简化）：

// ARM64 atomic_add_return (简化伪代码)
static inline int atomic_add_return(int i, atomic_t *v)
{
    unsigned long tmp;
    int result;

    asm volatile(
        "1: ldaxr   %w0, %2\n"    // 独占加载（acquire语义）
        "   add     %w0, %w0, %w3\n" // 加法
        "   stlxr   %w1, %w0, %2\n" // 独占存储（release语义）
        "   cbnz    %w1, 1b\n"     // 如果存储失败，重试
        : "=&r" (result), "=&r" (tmp), "+Q" (v->counter)
        : "Ir" (i)
        : "memory");

    return result;
}

x86 原子操作的内核实现示例（简化）：

// x86 atomic_add_return (简化伪代码)
static inline int atomic_add_return(int i, atomic_t *v)
{
    int __i = i;
    asm volatile(
        "lock; xaddl %0, %1\n"   // LOCK前缀 + XADD（交换并相加）
        : "+r" (__i), "+m" (v->counter)
        :
        : "memory");
    return __i + i;
}

2.5 实战调试要点

• 检查原子操作是否带屏障：在ARM64上，atomic_inc 和 atomic_add 不带屏障，而 atomic_inc_return 带完整屏障。误用可能导致可见性问题。
• 使用 READ_ONCE / WRITE_ONCE：对于非原子变量的共享访问，使用这两个宏可以防止编译器优化导致的撕裂读/写，但不提供硬件屏障。
• KCSAN 工具：内核的KCSAN（Kernel Concurrency Sanitizer）可以动态检测数据竞争，包括原子操作使用不当的情况。
• 调试打印中的原子性：在 printk 中读取共享变量时，应使用 atomic_read 或 READ_ONCE，避免读取到撕裂值。