欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

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

程序员文章站 2022-07-05 10:36:05
...
#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;
}