自旋锁实现机理 spin_lock

自旋锁的概念

自旋锁（spin lock）是一种典型的对临界资源进行互斥访问的手段，它是基于系统原子操作为基础，自旋锁最多只能被一个可执行线程持有，如果一个执行线程试图获得一个被已经持有（争用）的自旋锁，那么该线程就会一直进行忙循环-旋转-等待锁重新可用，要是锁未被争用，请求锁的执行线程就可以立即得到它，继续执行。

先看一下自旋锁实现的数据结构：

typedef struct {
	union {
		u32 slock;
		struct __raw_tickets {
			u16 owner;  /*原子变量*/
			u16 next;   /*原子变量*/
		} tickets;
	};

spinlock的核心思想是基于tickets的机制：
1） 每个锁的数据结构arch_spinlock_t中维护两个字段：next和owner，只有当next和owner相等时才能获取锁；
2） 每个进程在获取锁的时候，next值会增加，当进程在释放锁的时候owner值会增加；
3） 如果有多个进程在争抢锁的时候，看起来就像是一个排队系统，FIFO ticket spinlock；

 

static inline void arch_spin_lock(arch_spinlock_t *lock)
{
	unsigned long tmp;
	u32 newval;
	arch_spinlock_t lockval;

	prefetchw(&lock->slock);


	while (lockval.tickets.next != lockval.tickets.owner) {
		wfe();
		lockval.tickets.owner = ACCESS_ONCE(lock->tickets.owner);
	}

	smp_mb();
}

当tickets中的next和owner不相等的时候，说明临界区在忙， 需要等待，然后cpu会执行wfe指令。当其他cpu忙完之后，会更新owner的值，如果owner的值如果与next值相同，那到next号的cpu执行。

补充：

WFE

WFE的一个典型使用场景，是用在spinlock中（可参考arch_spin_lock，对arm64来说，位于arm64/include/asm/spinlock.h中）。spinlock的功能，是在不同CPU core之间，保护共享资源。使用WFE的流程是：

a）资源空闲

b）Core1访问资源，acquire lock，获得资源

c）Core2访问资源，此时资源不空闲，执行WFE指令，让core进入low-power state

d）Core1释放资源，release lock，释放资源，同时执行SEV指令，唤醒Core2

e）Core2获得资源

以往的spinlock，在获得不到资源时，让Core进入busy loop，而通过插入WFE指令，可以节省功耗，也算是因祸（损失了性能）得福（降低了功耗）吧。