Linux原子操作

并发问题

现代操作系统支持多任务的并发，并发在提高计算资源利用率的同时也带来了资源竞争的问题。假如一个任务正在读取文件一的情况下，此时又有另一个任务想要读取文件一，会造成读取混乱问题。可以使用原子操作实现资源互斥。

原子操作

原子操作指的是在执行过程中不会被别的代码路径所中断的操作。

1．设置原子变量的值

void atomic_set(atomic_t *v, int i); //设置原子变量的值为 i 

atomic_t v = ATOMIC_INIT(0); //定义原子变量 v 并初始化为 0 

2．获取原子变量的值

atomic_read(atomic_t *v); //返回原子变量的值

3．原子变量加/减

void atomic_add(int i, atomic_t *v); //原子变量增加 i 

void atomic_sub(int i, atomic_t *v); //原子变量减少 i 

4．原子变量自增/自减

void atomic_inc(atomic_t *v); //原子变量增加 1 

void atomic_dec(atomic_t *v); //原子变量减少 1 

5．操作并测试

int atomic_inc_and_test(atomic_t *v); 

int atomic_dec_and_test(atomic_t *v); 

int atomic_sub_and_test(int i, atomic_t *v); 

上述操作对原子变量执行自增、自减和减操作后测试其是否为 0， 为 0 则返回 true，否则返回 false。

修改代码

定义原子变量

static atomic_t demo_avaliable = ATOMIC_INIT(1);

修改open函数：如果没有进程在使用该驱动 ，原子变量值 为 1 ，将原子变量减 一 为 0 ，函数返回 true ，再 ！true 为 假 ，if 里面的代码不执行；这样打开了、并使用该驱动， 原子变量变为 0；

如果再有进程来打开驱动程序，0-1 = 负1，返回 false ，if 条件成立，运行里面的代码，将原子变量加一恢复到 0，程序返回；

static int demo_open(struct inode *node, struct file *file){
    
    printk(KERN_INFO"Enter:%s\n",__func__);
    if(!atomic_dec_and_test(&demo_avaliable)){
        printk(KERN_ERR"already open\n");
        atomic_inc(&demo_avaliable);
        return -EBUSY;
    }

    file->private_data =(void*) demo_dev; 
    return 0;
}

修改release函数

static int demo_release(struct inode *node, struct file *file){
    printk(KERN_INFO"Enter:%s\n",__func__);
    atomic_inc(&demo_avaliable);
    return 0;
}

在应用程序退出时 close 函数， 自增 恢复原子变量值为 1。