请问一个驱动代码逻辑问题，高手请进

jiangsheng84

/**
* srpt_set_cmd_state() - Set the state of a SCSI command.
* @new: New state to be set.
*
* Does not modify the state of aborted commands. Returns the previous command
* state.
*/
enum srpt_command_state srpt_set_cmd_state(struct srpt_ioctx *     ioctx,
                                                  enum srpt_command_state new)
{
    enum srpt_command_state previous;

    BUG_ON(!ioctx);
    WARN_ON(new == SRPT_STATE_NEW);

    do
    {
        previous = atomic_read(&ioctx->state);
    } while ((previous != SRPT_STATE_DONE)
             && atomic_cmpxchg(&ioctx->state, previous, new) != previous);

    return previous;
}
按正常说，这段代码执行应该没有什么问题，atomic_read和atomic_cmpxchg都能执行成功。
但是为什么要用do while语句来实现呢？说明确实存在循环的可能吗？会不会出现死循环呢？
因为我在中断上下文执行的时候，出了一次NMI的WATCHDOG复位，也就是在中断中执行时间太长了，
会不会是这里在某种情况下死循环呢？如果不是，为什么这里要写出do while呢，完全可以用if啊！
（当然也有可能是执行中断中的其他部分代码的时间太长导致NMI复位的）

bruceteen

看不懂，但 cmpxchg 和 循环 是一对好基友，达到非锁原子操作

jiangsheng84

谢谢楼上的回复，但是这里貌似已经调用的是atomic_cmpxchg啊！为什么还要循环呢，奇怪了

jiangsheng84

(previous != SRPT_STATE_DONE)不用管它，这个表示如果读出来是处于done的状态的话，就直接退出了，不在去改变它的状态
那就相当于变成：
    do
    {
        previous = atomic_read(&ioctx->state);
    } while ( atomic_cmpxchg(&ioctx->state, previous, new) != previous);
先原子读该状态，然后传入atomic_cmpxchg函数，该函数会重新读ioctx->state得状态，然后和previous比较，如果相等，就修改
为新的状态new，并且返回旧有的状态(即previous)。
这样看起来两次都读的同一个变量，肯定相等了，然后肯定就会修改为new的状态，并返回previous的状态，应该是一次就可以执行成功啊
为什么需要do while的循环呢？

bruceteen

仔细看了一下
关于你的问题，这正是需要atomic的地方，因为可能多个进程在同时操作，atomic_read 和 atomic_cmpxchg 之间，可能被别的进程给 atomic_cmpxchg 了。

jiangsheng84

回复 5# bruceteen


    谢谢，我看内核里面也有许多类似的应用
/**
*        sysfs_get_active - get an active reference to sysfs_dirent
*        @sd: sysfs_dirent to get an active reference to
*
*        Get an active reference of @sd.  This function is noop if @sd
*        is NULL.
*
*        RETURNS:
*        Pointer to @sd on success, NULL on failure.
*/
static struct sysfs_dirent *sysfs_get_active(struct sysfs_dirent *sd)
{
        if (unlikely(!sd))
                return NULL;

        while (1) {
                int v, t;

                v = atomic_read(&sd->s_active);
                if (unlikely(v < 0))
                        return NULL;

                t = atomic_cmpxchg(&sd->s_active, v, v + 1);
                if (likely(t == v))
                        return sd;
                if (t < 0)
                        return NULL;

                cpu_relax();
        }
}
可以看到if (likely(t == v))，既然都likely了，说明一般情况下应该是都能成功的对吧！那应该是不会出现死循环的，即使有两个线程在操作这个原子变量，应该
不可能刚好大家都在死循环来修改这个值吧！

bruceteen

不会死循环的，总会有一个进程通过
就像一群人不排队上厕所一样，马桶不可能空着，即需要上厕所的人数会一直减少，不会有人永远上不了厕所