wait queue 6

wait queue 还可以用于等待某个atomic变量为0，例如下下例中等待cookie->n_active == 0

wait_on_atomic_t(&cookie->n_active, fscache_wait_atomic_t, TASK_UNINTERRUPTIBLE);

其中fscache_wait_atomic_t 仅仅就是通过schedule() 让出cpu.

int fscache_wait_atomic_t(atomic_t *p)

{

    schedule();

    return 0;

}

和wait_on_atomic_t 对应的wakeup函数如下：

void wake_up_atomic_t(atomic_t *p)

{

    __wake_up_bit(atomic_t_waitqueue(p), p, WAIT_ATOMIC_T_BIT_NR);

}

wake_up_atomic_t 就是简单的调用__wake_up_bit而已。

我们重点看看wait_on_atomic_t的实现

static inline

int wait_on_atomic_t(atomic_t *val, int (*action)(atomic_t *), unsigned mode)

{

    might_sleep();

    if (atomic_read(val) == 0)

        return 0;

    return out_of_line_wait_on_atomic_t(val, action, mode);

}

首先判断val 是否为0，如果已经是0了，就不用等待了，否则调用out_of_line_wait_on_atomic_t

__sched int out_of_line_wait_on_atomic_t(atomic_t *p, int (*action)(atomic_t *),

                     unsigned mode)

{

    wait_queue_head_t *wq = atomic_t_waitqueue(p);

    DEFINE_WAIT_ATOMIC_T(wait, p);

    return __wait_on_atomic_t(wq, &wait, action, mode);

}

out_of_line_wait_on_atomic_t 首先得到wait_queue_head_t

static inline wait_queue_head_t *atomic_t_waitqueue(atomic_t *p)

{

    if (BITS_PER_LONG == 64) {

        unsigned long q = (unsigned long)p;

        return bit_waitqueue((void *)(q & ~1), q & 1);

    }

    return bit_waitqueue(p, 0);

}

这个函数通过bit_waitqueue得到wait_queue_head_t。64 和 32 bit的参数会有点区别.

out_of_line_wait_on_atomic_t 通过 DEFINE_WAIT_ATOMIC_T得到 wait queue

#define DEFINE_WAIT_ATOMIC_T(name, p)                    \

    struct wait_bit_queue name = {                    \

        .key = __WAIT_ATOMIC_T_KEY_INITIALIZER(p),        \

        .wait    = {                        \

            .private    = current,            \

            .func        = wake_atomic_t_function,    \

            .task_list    =                \

                LIST_HEAD_INIT((name).wait.task_list),    \

        },                            \

    }

这个和之前非atomic的却别是func不一样wake_atomic_t_function

wake_atomic_t_function 和之前的autoremove_wake_function 相比也就多了key.flags 和 key.bit_nr 的判断.看起来是wakeup的时候

key.flags 和 key.bit_nr和之前注册到wait_queue_head_t 里面的可能会变化.

static int wake_atomic_t_function(wait_queue_t *wait, unsigned mode, int sync,

                  void *arg)

{

    struct wait_bit_key *key = arg;

    struct wait_bit_queue *wait_bit

        = container_of(wait, struct wait_bit_queue, wait);

    atomic_t *val = key->flags;

    if (wait_bit->key.flags != key->flags ||

        wait_bit->key.bit_nr != key->bit_nr ||

        atomic_read(val) != 0)

        return 0;

    return autoremove_wake_function(wait, mode, sync, key);

}

out_of_line_wait_on_atomic_t 最后调用__wait_on_atomic_t

static __sched

int __wait_on_atomic_t(wait_queue_head_t *wq, struct wait_bit_queue *q,

               int (*action)(atomic_t *), unsigned mode)

{

    atomic_t *val;

    int ret = 0;

    do {

        prepare_to_wait(wq, &q->wait, mode);

        val = q->key.flags;

        if (atomic_read(val) == 0)

            break;

        ret = (*action)(val);

    } while (!ret && atomic_read(val) != 0);

    finish_wait(wq, &q->wait);

    return ret;

}

这个和之前的___wait_event 相比也就是多了对atomic_read(val) 值的判断.