【Windows原理】线程同步-临界区

#include "stdafx.h"
#include <windows.h>

int g_nNum = 0;
CRITICAL_SECTION g_section = {0};

DWORD WINAPI ThreadProcA(LPVOID lpParam)
{
    /*
     * 判断有没有人进入临界区，如果有，我就等它返回Leave
     * 如果没人占用，我就占用，然后执行后面的代码
     * 如果我一直占用，别人不能主动释放它
     * 别的线程调用EnterCriticalSection就会一直卡死
     **/
    for (int i = 0; i < 100000;i++)
    {
        EnterCriticalSection(&g_section);
        g_nNum++;
        LeaveCriticalSection(&g_section);
    }
    return 0;
}

DWORD WINAPI ThreadProcB(LPVOID lpParam)
{
    for (int i = 0; i < 100000; i++)
    {
        EnterCriticalSection(&g_section);
        g_nNum++;
        LeaveCriticalSection(&g_section);
    }
    return 0;
}


int _tmain(int argc, _TCHAR* argv[])
{
    InitializeCriticalSection(&g_section);
    HANDLE hHandle[2] = {};
    hHandle[0] = CreateThread(NULL, 0, ThreadProcA, NULL, 0, NULL);
    hHandle[1] = CreateThread(NULL, 0, ThreadProcB, NULL, 0, NULL);
    // 等到所有线程全部返回
    // 等待的多个内核对象句柄必须放在数组里
    // 参数1：总共几个内核对象
    // 参数2：存放内核对象句柄的数组
    // 参数3：在等待时间内，释放等待所有对象都有信号才返回
    // 如果FALSE，只有有一个对象返回就完成。
    // 参数4：永久等待
    WaitForMultipleObjects(2, hHandle, TRUE, INFINITE);
    printf("%d\n", g_nNum);
    DeleteCriticalSection(&g_section);
    return 0;
}