Linux内核 - Linux ARM atomic 原子读写

背景

在Linux Arm kernel实现原子读写64位数据；

接口代码

#include <linux/atomic.h>
atomic64_t my_atomic_var = ATOMIC64_INIT(0);

void atomic_write_64(atomic64_t *v, int64_t new_value) {
    atomic64_set(v, new_value);
}

int64_t atomic_read_64(atomic64_t *v) {
    return atomic64_read(v);
}

使用自旋锁

#include <linux/spinlock.h>

spinlock_t my_lock = SPIN_LOCK_UNLOCKED;
uint64_t my_shared_var = 0;

void write_64(uint64_t new_value) {
    spin_lock(&my_lock);
    my_shared_var = new_value;
    spin_unlock(&my_lock);
}

uint64_t read_64(void) {
    uint64_t value;
    spin_lock(&my_lock);
    value = my_shared_var;
    spin_unlock(&my_lock);
    return value;
}

使用读写锁

#include <linux/rwlock.h>

rwlock_t my_rwlock = RW_LOCK_UNLOCKED;
uint64_t my_shared_var = 0;

void write_64(uint64_t new_value) {
    write_lock(&my_rwlock);
    my_shared_var = new_value;
    write_unlock(&my_rwlock);
}

uint64_t read_64(void) {
    uint64_t value;
    read_lock(&my_rwlock);
    value = my_shared_var;
    read_unlock(&my_rwlock);
    return value;
}

32位的原子操作

#include <linux/module.h>
#include <linux/kernel.h>
#include <linux/init.h>
#include <linux/atomic.h>

static atomic_t my_atomic_var = ATOMIC_INIT(0);

static int __init my_module_init(void)
{
    printk(KERN_INFO "Loading my module...\n");

    // 初始化原子变量
    atomic_set(&my_atomic_var, 0);
    printk(KERN_INFO "Initial value: %d\n", atomic_read(&my_atomic_var));

    // 增加原子变量
    atomic_inc(&my_atomic_var);
    printk(KERN_INFO "After increment: %d\n", atomic_read(&my_atomic_var));

    // 减少原子变量
    atomic_dec(&my_atomic_var);
    printk(KERN_INFO "After decrement: %d\n", atomic_read(&my_atomic_var));

    // 增加原子变量并返回新值
    printk(KERN_INFO "Add and return new value: %d\n", atomic_add_return(5, &my_atomic_var));

    // 测试并设置原子变量
    if (atomic_cmpxchg(&my_atomic_var, 5, 10) == 5) {
        printk(KERN_INFO "Compare and exchange succeeded, new value: %d\n", atomic_read(&my_atomic_var));
    } else {
        printk(KERN_INFO "Compare and exchange failed, current value: %d\n", atomic_read(&my_atomic_var));
    }

    return 0;
}

static void __exit my_module_exit(void)
{
    printk(KERN_INFO "Unloading my module...\n");
}

module_init(my_module_init);
module_exit(my_module_exit);

MODULE_LICENSE("GPL");
MODULE_AUTHOR("Your Name");
MODULE_DESCRIPTION("A simple example of using atomic variables in the Linux kernel");