linux gcc原子操作

本文深入探讨了GCC从4.1.2版本提供的__sync_*系列内置函数,包括加减、逻辑运算的原子操作及比较和交换功能。同时解释了内存屏障的作用及其类型,如acquirebarrier、releasebarrier和fullbarrier,并通过示例程序展示了如何使用这些函数以确保正确的内存操作顺序。

gcc从4.1.2提供了__sync_*系列的built-in函数,用于提供加减和逻辑运算的原子操作。

其声明如下:
type __sync_fetch_and_add (type *ptr, type value, ...)
type __sync_fetch_and_sub (type *ptr, type value, ...)
type __sync_fetch_and_or (type *ptr, type value, ...)
type __sync_fetch_and_and (type *ptr, type value, ...)
type __sync_fetch_and_xor (type *ptr, type value, ...)
type __sync_fetch_and_nand (type *ptr, type value, ...)


type __sync_add_and_fetch (type *ptr, type value, ...)
type __sync_sub_and_fetch (type *ptr, type value, ...)
type __sync_or_and_fetch (type *ptr, type value, ...)
type __sync_and_and_fetch (type *ptr, type value, ...)
type __sync_xor_and_fetch (type *ptr, type value, ...)
type __sync_nand_and_fetch (type *ptr, type value, ...)

这两组函数的区别在于第一组返回更新前的值,第二组返回更新后的值。

type可以是1,2,4或8字节长度的int类型,即:
int8_t / uint8_t
int16_t / uint16_t
int32_t / uint32_t
int64_t / uint64_t
后面的可扩展参数(...)用来指出哪些变量需要memory barrier,因为目前gcc实现的是full barrier(类似于linux kernel 中的mb(),表示这个操作之前的所有内存操作不会被重排序到这个操作之后),所以可以略掉这个参数。

bool __sync_bool_compare_and_swap (type *ptr, type oldval type newval, ...)
type __sync_val_compare_and_swap (type *ptr, type oldval type newval, ...)


这两个函数提供原子的比较和交换,如果*ptr == oldval,就将newval写入*ptr,
第一个函数在相等并写入的情况下返回true.
第二个函数在返回操作之前的值。


__sync_synchronize (...)

发出一个full barrier.

关于memory barrier,cpu会对我们的指令进行排序,一般说来会提高程序的效率,但有时候可能造成我们不希望得到的结果,举一个例子,比如我们有一个硬件设备,它有4个寄存器,当你发出一个操作指令的时候,一个寄存器存的是你的操作指令(比如READ),两个寄存器存的是参数(比如是地址和size),最后一个寄存器是控制寄存器,在所有的参数都设置好之后向其发出指令,设备开始读取参数,执行命令,程序可能如下:

    write1(dev.register_size,size);
    write1(dev.register_addr,addr);
    write1(dev.register_cmd,READ);
    write1(dev.register_control,GO);

如果最后一条write1被换到了前几条语句之前,那么肯定不是我们所期望的,这时候我们可以在最后一条语句之前加入一个memory barrier,强制cpu执行完前面的写入以后再执行最后一条:
    write1(dev.register_size,size);
    write1(dev.register_addr,addr);
    write1(dev.register_cmd,READ);
    __sync_synchronize();
    write1(dev.register_control,GO);
memory barrier有几种类型:
    acquire barrier : 不允许将barrier之后的内存读取指令移到barrier之前(linux kernel中的wmb())。
    release barrier : 不允许将barrier之前的内存读取指令移到barrier之后 (linux kernel中的rmb())。
    full barrier    : 以上两种barrier的合集(linux kernel中的mb())。
还有两个函数:

type __sync_lock_test_and_set (type *ptr, type value, ...)
   将*ptr设为value并返回*ptr操作之前的值。

void __sync_lock_release (type *ptr, ...)
     将*ptr置0
示例程序:
#include <stdio.h>
#include <pthread.h>
#include <stdlib.h>

static int count = 0;


void *test_func(void *arg)
{
        int i=0;
        for(i=0;i<20000;++i){
                __sync_fetch_and_add(&count,1);
        }
        return NULL;
}

int main(int argc, const char *argv[])
{
        pthread_t id[20];
        int i = 0;

        for(i=0;i<20;++i){
                pthread_create(&id[i],NULL,test_func,NULL);
        }

        for(i=0;i<20;++i){
                pthread_join(id[i],NULL);
        }

        printf("%d\n",count);
        return 0;
}

 

Linux 系统中,原子操作的实现通常依赖于特定的处理器架构,并且其定义和实现分布在内核源码树中的多个位置。以下是一些常见的文件位置,它们与原子操作的实现密切相关: 1. **通用原子操作定义**: 在 Linux 内核源码中,通用的原子操作接口定义通常位于 `include/linux/types.h` 和 `include/linux/atomic.h` 文件中。这些头文件定义了如 `atomic_t` 类型以及基本的原子操作函数,例如 `atomic_read`、`atomic_set`、`atomic_add` 等。 2. **x86 架构相关的原子操作实现**: 对于 x86 架构,原子操作的具体实现通常位于 `arch/x86/include/asm/atomic.h` 和 `arch/x86/include/asm/atomic64_64.h` 文件中。这些文件利用了 x86 提供的 `lock` 指令前缀以及相关的原子指令(如 `xchg` 和 `cmpxchg`)来保证操作原子性 [^1]。 3. **ARM 架构相关的原子操作实现**: 对于 ARM 架构,原子操作的实现通常位于 `arch/arm/include/asm/atomic.h` 文件中。ARM 架构使用了不同的机制,如加载-存储指令(如 `ldrex` 和 `strex`)来实现原子性。 4. **位操作相关的原子实现**: Linux 内核中对位数组的原子操作(如 `set_bit`、`clear_bit` 和 `test_bit`)通常定义在 `include/asm-generic/bitops.h` 文件中。具体的实现可能因架构而异,例如 x86 的实现位于 `arch/x86/include/asm/bitops.h` [^2]。 5. **编译器相关的原子操作支持**: 在用户空间程序中,原子操作的实现可能依赖于编译器提供的原子操作支持,例如 GCC 提供的内置原子操作函数(如 `__atomic_load_n`、`__atomic_store_n`)。这些函数的定义和使用可以通过包含头文件 `<stdatomic.h>` 或 `<atomic>`(C++11 及以上版本)来实现。 6. **用户空间原子操作的库支持**: 在用户空间程序中,如果需要实现多线程环境下的原子操作,通常会使用 POSIX 线程库(如 `pthread_mutex_lock` 和 `pthread_mutex_unlock`)或者 C++ 标准库中的 `std::atomic`。这些接口的实现依赖于底层的原子操作机制,通常会调用内核提供的同步机制。 ### 示例代码 以下是一个简单的用户空间原子操作示例,使用了 C++11 标准库中的 `std::atomic`: ```cpp #include <iostream> #include <atomic> #include <thread> std::atomic<int> counter(0); void increment() { for (int i = 0; i < 1000; ++i) { counter.fetch_add(1, std::memory_order_relaxed); } } int main() { std::thread t1(increment); std::thread t2(increment); t1.join(); t2.join(); std::cout << "Counter value: " << counter << std::endl; return 0; } ``` ### 相关问题 1. Linux 内核中如何实现 `atomic_t` 类型的原子性? 2. 不同处理器架构(如 x86 和 ARM)在原子操作实现上有哪些差异? 3. 用户空间程序如何利用 C++ 标准库实现原子操作? 4. `set_bit` 和 `clear_bit` 函数在 Linux 内核中的具体实现逻辑是什么? 5. 如何通过设备树(Device Tree)描述板级信息?
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值