自旋锁实现.docVIP

大多数的并行程序都需要在底层使用锁机制进行同步，简单来讲，锁无非是一套简单的原语，它们保证程序（或进程）对某一资源的互斥访问来维持数据的一致性，如果没有锁机制作为保证，多个线程可能同时访问某一资源，假设没有精心设计的（很复杂）无锁算法保证程序正确执行，那么后果往往非常严重的。无锁算法难于使用，所以一般而言都使用锁来保证程序的一致性。 如果更新某一数据结构的操作比较缓慢，那么互斥的锁是一个比较好的选择，此时如果某一进程或线程被阻塞，操作系统会重新接管控制权，并调度其他进程（或线程）继续执行，原先被阻塞的进程处于睡眠状态。控制权的转换伴随着进程上下文的切换，而这往往是一个昂贵而耗时的操作，所以对于等待锁的时间比较短，那么应该使用其他更高效的方法。 自旋锁（spinlock） 自旋锁（Spinlock）是一种常用的互斥（Mutual Exclusion）同步原语（Synchronization Primitive），试图进入临界区（Critical Section）的线程使用忙等待（Busy Waiting）的方式检测锁的状态，若锁未被持有则尝试获取。与其他锁不同，自旋锁仅仅只是“自旋”，即不停地检查某一锁是否已经被解开，自旋锁是非常快的，所以加锁-解锁操作耗时很短，然而，自旋锁也不是万精油，当因互斥导致进程睡眠的时间很长时，使用自旋锁是不明智的选择。 下面我们考虑实现自己的自旋锁，首先我们需要一些原语，幸好GCC已经为我们提供了一些内置函数， #define atomic_xadd P, V __sync_fetch_and_add P , V #define cmpxchg P, O, N __sync_val_compare_and_swap P , O , N #define atomic_inc P __sync_add_and_fetch P , 1 #define atomic_dec P __sync_add_and_fetch P , -1 #define atomic_add P, V __sync_add_and_fetch P , V #define atomic_set_bit P, V __sync_or_and_fetch P , 1 V #define atomic_clear_bit P, V __sync_and_and_fetch P , ~ 1 V 然而，我们也需要自己实现其他的几个原子操作，如下： /* Compile read-write barrier */#define barrier asm volatile : : :memory /* Pause instruction to prevent excess processor bus usage */#define cpu_relax asm volatile pause\n: : :memory /* Atomic exchange of various sizes */static inline void *xchg_64 void *ptr, void *x __asm__ __volatile__ xchgq %0,%1 : r unsigned long long x :m * volatile long long * ptr , 0 unsigned long long x :memory ; return x; static inline unsigned xchg_32 void *ptr, unsigned x __asm__ __volatile__ xchgl %0,%1 : r unsigned x :m * volatile unsigned * ptr , 0 x :memory ; return x; static inline unsigned short xchg_16 void *ptr, unsigned short x __asm__ __volatile__ xchgw %0,%1 : r unsigned short x :m * volatile unsigned short * ptr , 0 x :memory ; return x; /* Test and set a bit */static inline char atomic_bitsetandtest void *ptr, int x char out; __asm__ __volatile__ lock; bts %2,%1\n sbb %0,%0\n : r out , m