偶然發(fā)現(xiàn)Windows API函數(shù)LeaveCriticalSection存在一個(gè)問(wèn)題,那就是在調(diào)用它離開臨界區(qū)時(shí),會(huì)把臨界區(qū)引用計(jì)數(shù)減一,而不判斷當(dāng)前線程是不是擁有該臨界區(qū),這樣就可能引發(fā)問(wèn)題,比如A線程擁有該臨界區(qū),但B線程可以調(diào)用LeaveCriticalSection來(lái)使得其它等待該臨界區(qū)的線程擁有該臨界區(qū),或者調(diào)用LeaveCriticalSection次數(shù)大于EnterCriticalSection的次數(shù)也可能出現(xiàn)誤操作而導(dǎo)致不可預(yù)知的后果。
把EnterCriticalSection換下面這一段代碼可以有效的防止占用鎖的線程已經(jīng)退出而導(dǎo)致死鎖的情況(親測(cè)有效):
//嘗試進(jìn)入臨界區(qū)
while(!TryEnterCriticalSection(&aoSection)){
//進(jìn)入失敗,判斷當(dāng)前占有臨界區(qū)的線程是否還存活
if(aoSection.RecursionCount > 0){
DWORD result = WaitForSingleObject(aoSection.OwningThread, 0);
if(result == WAIT_FAILED){
//如果線程已經(jīng)退出,則代表線程被異常結(jié)束或代碼問(wèn)題導(dǎo)致鎖沒有釋放,就利用此bug強(qiáng)制釋放鎖
long lRecursionCount = aoSection.RecursionCount;
for(int i = 0; i < lRecursionCount; i++){
LeaveCriticalSection(&moSection);
}
}
}
}