set_current_state 应用

A: A是一个等待进程等待condition 满足过后退出死循环
A:

while(1)
{
if ( condition ) //条件成立了
goto: OUT
else{
//1:----------------------
set_current_state(TASK_UNINTERRUPTIBLE);
schedule();
set_current_state(TASK_RUNNING);
}
}
OUT:

B进程会唤醒A进程:
B:

wake_up_process(A)

写完之后,虽然测试通过了```但是老感觉有什么问题所下:
在A进程中,首先它判断条件不满足,然后进入到else中如果运行到1:------的时候B进程被抢占进来了`并将A唤醒```
之后A进程被调度回CPU然后沿着1:------以下的部份运行,它设置当前状态,自己调用schedule()错过了这次条件````此后A就会被移出运行队列,永远都睡眠着```

我开始怀疑这段代码,搜索了内核代码,发现内核中很多地方也有相类似的用法无奈之下,向"中文邮件列表"中的众位高手请教,终于发现了问题所在正确的写法应该是这样的:
//1:----------------------//
set_current_state(TASK_UNINTERRUPTIBLE);
if (!condition)
schedule();
set_current_state(TASK_RUNNING);

真是差之毫里,失之千里现把这两段代码分析一下

在第一段代码中如果在标号处被抢占,B调用 wake_up_process(A)将A唤醒,此时A的运行状态会设为TASK_RUNNING如切换到A进程运行时,之后的代码又会将其设为TASK_UNINTERRUPTIBLE
在调用schdule()的时候,会判断进程是自愿放弃CPU还是被抢占出来的,如果是自愿放弃且进程不为运行状态就会将其从运行队列中删除
这样A就永远失去了被调度的机会

在第二种情况中:
1:进程A在set_current_state(TASK_UNINTERRUPTIBLE);之前被抢占,B将其唤配,切换回A后,判断condition条件时,会是一个真值,之后又会被调会TASK_RUNNING这是正常的
2:进程A在set_current_state(TASK_UNINTERRUPTIBLE)之后被抢占,假设抢占地在判断完条件之后这样在B中会将A设为RUNING状态,A切换回来的时候,进入到schdule()后不会被踢出运行队列,只是等待下次调度而已,这也是正常的.