【线程安全】临界区实现线程互斥

问题概述：

在多线程程序开发的过程中，当多个线程对同一共享资源进行修改时，就会产生线程安全问题。例如如下情况就会产生线程安全问题：

#include <Windows.h>
#include <stdio.h>

int g_count = 0;

DWORD WINAPI ThreadProc(LPVOID)
{
	for (int i = 0; i < 1000000; i++)
		g_count++;
	return 0;
}

int main()
{
	HANDLE h_thread[2];
	h_thread[0] = CreateThread(NULL, 0, ThreadProc, NULL, NULL, NULL);
	h_thread[1] = CreateThread(NULL, 0, ThreadProc, NULL, NULL, NULL);
	WaitForMultipleObjects(2, h_thread, TRUE, INFINITE);
	CloseHandle(h_thread[0]);
	CloseHandle(h_thread[1]);

	printf("count--->%d\n", g_count);

	getchar();
	return 0;
}

运行结果：原本应该为2000000，这里却为1278840

问题分析：

切换到VS自带的反汇编界面查看，g_count++由这三句汇编语句构成

这里列举一种产生线程安全的情况：

假设此时g_count=0
当线程A执行完第一句汇编指令 mov eax,g_count 后eax=0
此时线程切换，线程B开始执行，线程B一次执行完三条汇编语句，结果g_count=1
此时再次发生线程切换，线程A继续执行
执行第二句汇编指令 add eax,1 后eax=1
执行第三句汇编指令 mov g_count,eax 后g_count=1
此时产生的问题是，两个线程都对g_count进行了一次加操作，但是g_count最后的结果却为1，相当于只进行了一次加操作。

解决方案：

这里采用Win32系统提供的临界区来保证多线程对统一资源访问的安全，如下为临界区相关函数：

//定义临界区对象
CRITICAL_SECTION cs;
//初始化临界区对象
InitializeCriticalSection(&cs);
//进入临界区
EnterCriticalSection(&cs);
//离开临界区
LeaveCriticalSection(&cs);
//销毁临界区
DeleteCriticalSection(&cs);

实现代码：

#include <Windows.h>
#include <stdio.h>

CRITICAL_SECTION g_cs;
int g_count = 0;

DWORD WINAPI ThreadProc(LPVOID)
{
	EnterCriticalSection(&g_cs);
	for (int i = 0; i < 1000000; i++)
		g_count++;
	LeaveCriticalSection(&g_cs);
	return 0;
}

int main()
{
	InitializeCriticalSection(&g_cs);

	HANDLE h_thread[2];
	h_thread[0] = CreateThread(NULL, 0, ThreadProc, NULL, NULL, NULL);
	h_thread[1] = CreateThread(NULL, 0, ThreadProc, NULL, NULL, NULL);
	WaitForMultipleObjects(2, h_thread, TRUE, INFINITE);
	CloseHandle(h_thread[0]);
	CloseHandle(h_thread[1]);

	DeleteCriticalSection(&g_cs);
	printf("count--->%d\n", g_count);

	getchar();
	return 0;
}

运行结果：