Linux 原子操作

        所谓原子操作，就是该操作绝不会在执行完毕前被任何其他任务或事件打断，也就说，它的最小的执行单位，不可能有比它更小的执行单位，因此这里的原子实际是使用了物理学里的物质微粒的概念。

 

　　原子操作需要硬件的支持，因此是架构相关的，其API和原子类型的定义都定义在内核源码树的include/asm/atomic.h文件中，它们都使用汇编语言实现，因为C语言并不能实现这样的操作。

　　原子操作主要用于实现资源计数，很多引用计数(refcnt)就是通过原子操作实现的。原子类型定义如下：

 

typedef struct { volatile int counter; } atomic_t;

　　volatile修饰字段告诉gcc不要对该类型的数据做优化处理，对它的访问都是对内存的访问，而不是对寄存器的访问。

 

　　原子操作API包括：

atomic_read(atomic_t * v);

　　该函数对原子类型的变量进行原子读操作，它返回原子类型的变量v的值。

 

 

atomic_set(atomic_t * v, int i);

　　该函数设置原子类型的变量v的值为i。

 

 

void atomic_add(int i, atomic_t *v);

　　该函数给原子类型的变量v增加值i。

 

 

atomic_sub(int i, atomic_t *v);

　　该函数从原子类型的变量v中减去i。

 

 

int atomic_sub_and_test(int i, atomic_t *v);

　　该函数从原子类型的变量v中减去i，并判断结果是否为0，如果为0，返回真，否则返回假。

 

 

void atomic_inc(atomic_t *v);

　　该函数对原子类型变量v原子地增加1。

 

 

void atomic_dec(atomic_t *v);

　　该函数对原子类型的变量v原子地减1。

 

 

int atomic_dec_and_test(atomic_t *v);

　　该函数对原子类型的变量v原子地减1，并判断结果是否为0，如果为0，返回真，否则返回假。

 

 

int atomic_inc_and_test(atomic_t *v);

　　该函数对原子类型的变量v原子地增加1，并判断结果是否为0，如果为0，返回真，否则返回假。

 

 

int atomic_add_negative(int i, atomic_t *v);

　　该函数对原子类型的变量v原子地增加I，并判断结果是否为负数，如果是，返回真，否则返回假。

 

 

int atomic_add_return(int i, atomic_t *v);

　　该函数对原子类型的变量v原子地增加i，并且返回指向v的指针。

 

 

int atomic_sub_return(int i, atomic_t *v);

　　该函数从原子类型的变量v中减去i，并且返回指向v的指针。

 

 

int atomic_inc_return(atomic_t * v);

　　该函数对原子类型的变量v原子地增加1并且返回指向v的指针。

 

 

int atomic_dec_return(atomic_t * v);

　　该函数对原子类型的变量v原子地减1并且返回指向v的指针。

　　原子操作通常用于实现资源的引用计数，在TCP/IP协议栈的IP碎片处理中，就使用了引用计数，碎片队列结构struct ipq描述了一个IP碎片，字段refcnt就是引用计数器，它的类型为atomic_t，当创建IP碎片时（在函数ip_frag_create中），使用atomic_set函数把它设置为1，当引用该IP碎片时，就使用函数atomic_inc把引用计数加1。

　　当不需要引用该IP碎片时，就使用函数ipq_put来释放该IP碎片，ipq_put使用函数atomic_dec_and_test把引用计数减1并判断引用计数是否为0，如果是就释放IP碎片。函数ipq_kill把IP碎片从ipq队列中删除，并把该删除的IP碎片的引用计数减1（通过使用函数atomic_dec实现）。

        原子操作就是执行时不会被中断的指令。两个原子操作不可能同时作用于同一个变量，这个是原子操作的一个保证，至于处理器或者操作系统如何实现那是另外一回事。

        内核提供了两个原子操作接口：一个是针对整型，其实也就是一个字，另一个针对单独的位。这些接口在不同的架构有不同的实现，大多数是直接支持简单的原子操作或者锁住内存总线以确保其他操作不能同时进行。可以想象下，简单的原子操作对应单处理器系统，对于多处理器系统就有必要对内存总线加锁了。另外有些架构如SPARC对原子操作指令的支持不足。

   针对atomic_t而不是int来进行整型的原子操作的原因是：

      1. 确保原子操作只接收atomic_t类型的参数确保原子操作只对该类型有效，同时也保证该类型不会被传给非原子操作函数。

      2. 避免编译器自作聪明的优化，原子操作应该总是接收正确的内存地址。

      3. 可以隐藏架构独特性，如同pthread_t，或者反过来。

   虽然atomic_t是个整型，也就是32位，但是以前一般只假定atomic_t不超过24位，这个跟SPARC有关，不过现在这个问题已经解决。

   定义一个atomic_t变量很简单：

      Atomic_t v;

      Atomic_t  u=ATOMIC_INIT(0);

   下面是原子操作列表：

      ATOMIC_INIT(int  i)                  申明变量时初始化为i

      Int    atomic_read(atomic_t  *v)    读取变量v

      Void  atomic_set(atomic_t  *v,int  i)    把i赋给v

      Void  atomic_add(int  i,atomic_t  *v)v+i

      Void  atomic_sub(int  i,atomic_t  *v)v-i

      Void  atomic_inc(atomic_t  *v)++v

      Void  atomic_dec(atomic_t  *v)--v

      Int  atomic_sub_and_test(int  i,atomic_t  *v)

                v-i, 如果结果为0，返回真(也就是1），否则返回假（也就是0）。呃。。。。好吧，俺啥也没说。

      Int  atomic_add_negative(int  i,atommic_t  *v)v+i, 如果结果为负数返回真，否则返回假

      Int  atomic_dec_and_test(atomic  *v)--v, 如果为0返回真，否则返回假。

      Int  atomic_inc_and_test(atomic  *v)++v, 如果为0返回真，否则返回假

    这些原子操作通常是由inline汇编语句写的Inline函数实现的，通常也会是个宏。

   需要区分下原子性与有序性：

   读操作总是原子的，在大多数正常的架构中，如果一个整型初值为42然后被设置为365，对这个数的读操作要么返回42，要么返回365，而不可能是其他一个什么不知道是啥的数字，这就是原子性。

   假如你需要一个读操作必须要在一个pending的写操作之前进行，这种行为就是有序性

   原子性保证指令不会被中断并且要么完成要么失败。有序性保证几个指令的顺序，即便这些指令在不同的线程中。

  通常原子操作总是优先选择的，因为简单。