WaitForSingleObject 的内部实现原理

VBProFan

WaitForSingleObject 是 kernel32.dll 的导出函数，WaitForSingleObject 调用了ntdll.dll 的 NtWaitForSingleObject.
NtWaitForSingleObject 又调用了 KeWaitForSingleObject

WaitForSingleObject -> NtWaitForSingleObject -> KeWaitForSingleObject

以下是KeWaitForSingleObject的部分实现代码。

1. do
   
2. {
   
3.         WaitStatus = CurrentThread->WaitStatus;
   
4.         CurrentThread->WaitBlockList = WaitBlock = &CurrentThread->WaitBlock[0];
   
5.         CurrentObject = (PDISPATCHER_HEADER)Object;
   
6. 
   
7.         if (KiIsObjectSignaled(CurrentObject, CurrentThread))
   
8.         {
   
9.                 if (CurrentObject->SignalState != MINLONG)
   
10.                 {
    
11.                         KiSatisfyObjectWait(CurrentObject, CurrentThread);
    
12.                         Status = STATUS_WAIT_0;
    
13.                         goto WaitDone;
    
14.                 }
    
15.                 else
    
16.                 {
    
17.                         if (CurrentObject->Type == MutantObject)
    
18.                         {
    
19.                                 KeReleaseDispatcherDatabaseLock(CurrentThread->WaitIrql);
    
20.                                 ExRaiseStatus(STATUS_MUTANT_LIMIT_EXCEEDED);
    
21.                         }
    
22.                 }
    
23.         }
    
24. 
    
25.         WaitBlock->Object = CurrentObject;
    
26.         WaitBlock->Thread = CurrentThread;
    
27.         WaitBlock->WaitKey = (USHORT)(STATUS_WAIT_0);
    
28.         WaitBlock->WaitType = WaitAny;
    
29.         WaitBlock->NextWaitBlock = NULL;
    
30. 
    
31.         KiCheckAlertability(Alertable, CurrentThread, WaitMode, &Status);
    
32. 
    
33.         CurrentThread->WaitStatus = Status;
    
34. 
    
35.         if (Timeout != NULL)
    
36.         {
    
37.                 //略.有超时设置的情况
    
38.         }
    
39. 
    
40.         InsertTailList(&CurrentObject->WaitListHead, &WaitBlock->WaitListEntry);
    
41. 
    
42.         if (CurrentThread->Queue)
    
43.         {
    
44.                 DPRINT("Waking Queue\n");
    
45.                 KiWakeQueue(CurrentThread->Queue);
    
46.         }
    
47. 
    
48.         PsBlockThread(&Status, Alertable, WaitMode, (UCHAR)WaitReason);
    
49. 
    
50.         if (Status != STATUS_KERNEL_APC)
    
51.         {
    
52.                 return Status;
    
53.         }
    
54. 
    
55.         DPRINT("Looping Again\n");
    
56.         CurrentThread->WaitIrql = KeAcquireDispatcherDatabaseLock();
    
57. 
    
58. }while (TRUE);
    
59. 
    
60. WaitDone:

复制代码

[ 本帖最后由 VBProFan 于 2008-11-6 11:49 编辑 ]