Linux内核API atomic_set

函数atomic_set()的功能是将原子类型的变量v的值设置为i

atomic_set文件包含

#include <asm/atomic.h>

atomic_set函数定义

在内核源码中的位置：linux-3.19.3/arch/x86/include/asm/atomic.h

函数定义格式：

static inline void atomic_set(atomic_t * v, int i)

atomic_set输入参数说明

• v：原子类型变量，该参数一般传递一个指针。其中，原子类型atomic_t在内核文件linux-3.19.3/include/linux/types.h中定义：

typedef struct {
        int counter;
} atomic_t;

counter为一个int变量的计数器。

原子操作是最小的执行单位，它需要硬件的支持，因此是与体系结构相关的。原子操作一般用来保护简单变量，例如计数器和位掩码（bitmask），上面结构体中counter字段即可看作一个计数器。C语言不能实现原子操作，一些对原子类型的操作函数都是通过汇编语言来实现的。在笔者的系统中，为x86体系结构，因此对本章中有关原子操作的API的分析都是基于x86体系结构的。

• i：整型值，原子类型变量的值将被设置为该值。

atomic_set返回参数说明

• 函数atomic_set()没有返回值。

atomic_set实例解析

编写测试文件atomic_set.c

头文件及全局变量声明如下：

#include <linux/types.h>
#include <linux/init.h>
#include <linux/module.h>
MODULE_LICENSE("GPL");
static int __init atomic_set_init(void);
static void __exit atomic_set_exit(void);

atomic_t my_atomic ;

模块初始化函数：

int __init atomic_set_init(void)
{
    int i = 5;
    printk("before atomic_set, my_atomic.counter = %d\n", atomic_read( &my_atomic));
    atomic_set( &my_atomic, i );
    printk("after atomic_set, my_atomic.counter = %d\n", atomic_read( &my_atomic));
    return 0;
}

模块退出函数：

void __exit atomic_set_exit(void)
{
    printk("exit! \n");
}

模块初始化及退出函数调用：

module_init(atomic_set_init);
module_exit(atomic_set_exit);

实例运行结果及分析：

首先编译模块，执行命令insmod atomic_set.ko插入模块，然后执行命令dmesg -c，会出现如图所示的结果。

结果分析：

测试程序中调用了函数atomic_read()，它是对原子类型的变量进行原子读操作，参考极客笔记网关于函数atomic_read()的分析。

首先定义一个原子类型my_atomic，然后调用函数atomic_set()将其值设置为5，并通过函数atomic_read()将原子类型my_atomic的counter字段读出。由输出信息可知，调用atomic_set()后，原子类型的值由0变为了5。