OSTEP：wakeup/waiting race

28.14 Using Queues: Sleeping Instead Of Spinning

1 typedef struct __lock_t {
2     int flag;
3     int guard;
4     queue_t *q;
5 } lock_t;
6
7 void lock_init(lock_t *m) {
8      m->flag = 0;
9      m->guard = 0;
10     queue_init(m->q);
11 }
12
13 void lock(lock_t *m) {
14     while (TestAndSet(&m->guard, 1) == 1)
15                         ; //acquire guard lock by spinning
16     if (m->flag == 0) {
17         m->flag = 1; // lock is acquired
18         m->guard = 0;
19     } else {
20         queue_add(m->q, gettid());
21         m->guard = 0;
22         park();
23     }
24 }
25
26 void unlock(lock_t *m) {
27     while (TestAndSet(&m->guard, 1) == 1)
28                         ; //acquire guard lock by spinning
29     if (queue_empty(m->q))
30         m->flag = 0; // let go of lock; no one wants it
31     else
32         unpark(queue_remove(m->q)); // hold lock
33     // (for next thread!)
34     m->guard = 0;
35 }

在原文中有一段说明了这段代码这一种情况会出现race现象，那就当一个线程在调用park之前被操作系统切换到另一个持有锁的线程，然后该线程很快的又释放了锁，这就导致第一个线程在调用park时，它的id已经在queue中被删除，然后调用park就会进入永远的sleep，这不符合我们对于一个锁的预期，所以有了以下改进

1 queue_add(m->q, gettid());
2 setpark(); // new code
3 m->guard = 0;

通过 setpark()，一个线程表明自己马上要 park。如果刚好另一个线程被调度，并且调用了 unpark，那么后续的 park调用就会直接返回，而不是一直睡眠。