Linux中的READ_ONCE、WRITE_ONCE

本文解析READ_ONCE和WRITE_ONCE宏的原理,讨论它们如何防止编译器优化,以及为何需要配合内存屏障解决并发问题。涉及volatile、内存屏障指令和编译参数-fstrict-aliasing。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

READ_ONCE宏

#define READ_ONCE(x) __READ_ONCE(x, 1)

#define __READ_ONCE(x, check)                       \
({                                  \
    union { typeof(x) __val; char __c[1]; } __u;            \
    if (check)                          \
        __read_once_size(&(x), __u.__c, sizeof(x));     \
    else                                \
        __read_once_size_nocheck(&(x), __u.__c, sizeof(x)); \
    __u.__val;                          \
})

static __always_inline
void __read_once_size(const volatile void *p, void *res, int size)
{
    __READ_ONCE_SIZE;
}

static __always_inline
void __read_once_size_nocheck(const volatile void *p, void *res, int size)
{
    __READ_ONCE_SIZE;
}

#define __READ_ONCE_SIZE                        \
({                                  \
    switch (size) {                         \
    case 1: *(__u8 *)res = *(volatile __u8 *)p; break;      \
    case 2: *(__u16 *)res = *(volatile __u16 *)p; break;        \
    case 4: *(__u32 *)res = *(volatile __u32 *)p; break;        \
    case 8: *(__u64 *)res = *(volatile __u64 *)p; break;        \
    default:                            \
        barrier();                      \
        __builtin_memcpy((void *)res, (const void *)p, size);   \
        barrier();                      \
    }                               \
})

barrier() __asm__ __volatile__("": : :"memory")

WRITE_ONCE宏

#define WRITE_ONCE(x, val) \
({                          \
    union { typeof(x) __val; char __c[1]; } __u =   \
        { .__val = (__force typeof(x)) (val) }; \
    __write_once_size(&(x), __u.__c, sizeof(x));    \
    __u.__val;                  \
})

static __always_inline void __write_once_size(volatile void *p, void *res, int size)
{
    switch (size) {
    case 1: *(volatile __u8 *)p = *(__u8 *)res; break;
    case 2: *(volatile __u16 *)p = *(__u16 *)res; break;
    case 4: *(volatile __u32 *)p = *(__u32 *)res; break;
    case 8: *(volatile __u64 *)p = *(__u64 *)res; break;
    default:
        barrier();
        __builtin_memcpy((void *)p, (const void *)res, size);
        barrier();
    }
}

知识点 barrier宏
barrier()宏往C语言代码中插入一条汇编指令,告诉编译器,
不要将这条汇编指令前的内存读写指令,优化到这条汇编指令之后,
同时也不能将这条汇编指令之后的内存读写指令,优化到这条汇编指令之前。

知识点 volatile 防止编译优化:
这里并不是把变量定义为 volatile 而是赋值操作的指针类型定义为 volatile。原因见下。

知识点 volatile 作恶:
[RFC/PATCH] doc: volatile considered evil
Nine ways to break your systems code using volatile

知识点 编译参数 -fstrict-aliasing
gcc手册
__may_alias__

知识点 内存屏障
READ_ONCE和WRITE_ONCE宏只能保证读写操作不被编译器优化掉,造成多线程执行中出问题,但是它并不能干预CPU执行编译出来的程序,也就是不能解决CPU重排序的问题缓存一致性的问题,这类问题还是需要使用内存屏障来解决。

这几尊佛的坑位太大了,需要深挖。。。

参考链接
READ_ONCE和WRITE_ONCE
内存屏障

还有谢宝友老师也写过这方面的资料。深入理解RCU、深入理解并行编程、DIM-SUM自研操作系统。
 
 
完。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值